/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2013 Universita' di Firenze

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Author: Tommaso Pecorella <tommaso.pecorella@unifi.it>

 */

#include "ns3/log.h"

#include "ns3/nstime.h"

#include "ns3/packet-socket-address.h"

#include "ns3/packet-socket.h"

#include "ns3/packet-socket-factory.h"

#include "ns3/socket.h"

#include "ns3/simulator.h"

#include "ns3/socket-factory.h"

#include "ns3/packet.h"

#include "ns3/uinteger.h"

#include "ns3/abort.h"

#include "simple-recv-app.h"

#include <cstdlib>

#include <cstdio>

namespace ns3 {

NS_LOG_COMPONENT_DEFINE ("SimpleRecvApp")

  ;

NS_OBJECT_ENSURE_REGISTERED (SimpleRecvApp)

  ;

TypeId

SimpleRecvApp::GetTypeId (void)

{

  static TypeId tid = TypeId ("ns3::SimpleRecvApp")

    .SetParent<Application> ()

    .AddConstructor<SimpleRecvApp> ()

    .AddTraceSource ("Rx", "A packet has been received",

                     MakeTraceSourceAccessor (&SimpleRecvApp::m_rxTrace))

  ;

  return tid;

}

SimpleRecvApp::SimpleRecvApp ()

{

  NS_LOG_FUNCTION (this);

  m_pktRx = 0;

  m_bytesRx = 0;

  m_socket = 0;

}

SimpleRecvApp::~SimpleRecvApp ()

{

  NS_LOG_FUNCTION (this);

}

void

SimpleRecvApp::DoDispose (void)

{

  NS_LOG_FUNCTION (this);

  Application::DoDispose ();

}

void

SimpleRecvApp::StartApplication (void)

{

  NS_LOG_FUNCTION (this);

  if (m_socket == 0)

    {

      TypeId tid = TypeId::LookupByName ("ns3::PacketSocketFactory");

      m_socket = Socket::CreateSocket (GetNode (), tid);

      m_socket->Bind ();

    }

  m_socket->SetRecvCallback (MakeCallback (&SimpleRecvApp::HandleRead, this));

}

void

SimpleRecvApp::StopApplication (void)

{

  NS_LOG_FUNCTION (this);

  m_socket->SetRecvCallback (MakeNullCallback<void, Ptr<Socket> > ());

  m_socket->Close ();

}

void

SimpleRecvApp::HandleRead (Ptr<Socket> socket)

{

  NS_LOG_FUNCTION (this << socket);

  Ptr<Packet> packet;

  Address from;

  while ((packet = socket->RecvFrom (from)))

    {

      if (packet->GetSize () == 0)

        { //EOF

          break;

        }

      m_pktRx ++;

      m_bytesRx += packet->GetSize ();

      if (PacketSocketAddress::IsMatchingType (from))

        {

          NS_LOG_INFO ("At time " << Simulator::Now ().GetSeconds ()

                       << "s packet sink received "

                       << packet->GetSize () << " bytes from "

                       << PacketSocketAddress::ConvertFrom(from)

                       << " total Rx " << m_pktRx << " packets"

                       << " and " << m_bytesRx << " bytes");

        }

      m_rxTrace (packet, from);

    }

}

} // Namespace ns3

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2013 Universita' di Firenze

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Author: Tommaso Pecorella <tommaso.pecorella@unifi.it>

 */

#ifndef SIMPLE_RECV_APP_H

#define SIMPLE_RECV_APP_H

#include "ns3/application.h"

#include "ns3/event-id.h"

#include "ns3/ptr.h"

#include "ns3/packet-socket-address.h"

namespace ns3 {

class Socket;

class Packet;

/**

 * \ingroup socket

 *

 * \brief A simple client.

 *

 * Sends packets using PacketSocket. It does not require (or use) IP.

 *

 */

class SimpleRecvApp : public Application

{

public:

  /**

   * \brief Get the type ID.

   * \return the object TypeId

   */

  static TypeId GetTypeId (void);

  SimpleRecvApp ();

  virtual ~SimpleRecvApp ();

protected:

  virtual void DoDispose (void);

private:

  virtual void StartApplication (void);

  virtual void StopApplication (void);

  /**

   * \brief Handle a packet received by the application

   * \param socket the receiving socket

   */

  void HandleRead (Ptr<Socket> socket);

  uint32_t m_pktRx;    //!< The number of received packets

  uint32_t m_bytesRx;  //!< Total bytes received

  Ptr<Socket> m_socket; //!< Socket

  /// Traced Callback: received packets, source address.

  TracedCallback<Ptr<const Packet>, const Address &> m_rxTrace;

};

} // namespace ns3

#endif /* SIMPLE_RECV_APP_H */

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2013 Universita' di Firenze

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Author: Tommaso Pecorella <tommaso.pecorella@unifi.it>

 */

#include "ns3/log.h"

#include "ns3/nstime.h"

#include "ns3/packet-socket-address.h"

#include "ns3/packet-socket.h"

#include "ns3/packet-socket-factory.h"

#include "ns3/socket.h"

#include "ns3/simulator.h"

#include "ns3/socket-factory.h"

#include "ns3/packet.h"

#include "ns3/uinteger.h"

#include "ns3/abort.h"

#include "simple-send-app.h"

#include <cstdlib>

#include <cstdio>

namespace ns3 {

NS_LOG_COMPONENT_DEFINE ("SimpleSendApp")

  ;

NS_OBJECT_ENSURE_REGISTERED (SimpleSendApp)

  ;

TypeId

SimpleSendApp::GetTypeId (void)

{

  static TypeId tid = TypeId ("ns3::SimpleSendApp")

    .SetParent<Application> ()

    .AddConstructor<SimpleSendApp> ()

    .AddAttribute ("MaxPackets",

                   "The maximum number of packets the application will send",

                   UintegerValue (100),

                   MakeUintegerAccessor (&SimpleSendApp::m_count),

                   MakeUintegerChecker<uint32_t> ())

    .AddAttribute ("Interval",

                   "The time to wait between packets", TimeValue (Seconds (1.0)),

                   MakeTimeAccessor (&SimpleSendApp::m_interval),

                   MakeTimeChecker ())

    .AddAttribute ("RemoteAddress",

                   "The destination Address of the outbound packets",

                   AddressValue (),

                   MakeAddressAccessor (&SimpleSendApp::m_peerAddress),

                   MakeAddressChecker ())

    .AddAttribute ("PacketSize",

                   "Size of packets generated. The minimum packet size is 12 bytes which is the size of the header carrying the sequence number and the time stamp.",

                   UintegerValue (1024),

                   MakeUintegerAccessor (&SimpleSendApp::m_size),

                   MakeUintegerChecker<uint32_t> (12,1500))

  ;

  return tid;

}

SimpleSendApp::SimpleSendApp ()

{

  NS_LOG_FUNCTION (this);

  m_sent = 0;

  m_socket = 0;

  m_sendEvent = EventId ();

}

SimpleSendApp::~SimpleSendApp ()

{

  NS_LOG_FUNCTION (this);

}

void

SimpleSendApp::SetRemote (PacketSocketAddress addr)

{

  NS_LOG_FUNCTION (this << addr);

  m_peerAddress = Address(addr);

}

void

SimpleSendApp::SetRemote (Address addr)

{

  NS_LOG_FUNCTION (this << addr);

  m_peerAddress = addr;

}

void

SimpleSendApp::DoDispose (void)

{

  NS_LOG_FUNCTION (this);

  Application::DoDispose ();

}

void

SimpleSendApp::StartApplication (void)

{

  NS_LOG_FUNCTION (this);

  if (m_socket == 0)

    {

      TypeId tid = TypeId::LookupByName ("ns3::PacketSocketFactory");

      m_socket = Socket::CreateSocket (GetNode (), tid);

      if (PacketSocketAddress::IsMatchingType(m_peerAddress) == true)

        {

          m_socket->Bind ();

          m_socket->Connect (m_peerAddress);

        }

      else

        {

          NS_ABORT_MSG ("Peer address type not matching PacketSocketAddress");

        }

    }

  m_socket->SetRecvCallback (MakeNullCallback<void, Ptr<Socket> > ());

  m_sendEvent = Simulator::Schedule (Seconds (0.0), &SimpleSendApp::Send, this);

}

void

SimpleSendApp::StopApplication (void)

{

  NS_LOG_FUNCTION (this);

  Simulator::Cancel (m_sendEvent);

  m_socket->Close ();

}

void

SimpleSendApp::Send (void)

{

  NS_LOG_FUNCTION (this);

  NS_ASSERT (m_sendEvent.IsExpired ());

  Ptr<Packet> p = Create<Packet> (m_size);

  std::stringstream peerAddressStringStream;

  peerAddressStringStream << PacketSocketAddress::ConvertFrom (m_peerAddress);

  if ((m_socket->Send (p)) >= 0)

    {

      ++m_sent;

      NS_LOG_INFO ("TraceDelay TX " << m_size << " bytes to "

                                    << peerAddressStringStream.str () << " Uid: "

                                    << p->GetUid () << " Time: "

                                    << (Simulator::Now ()).GetSeconds ());

    }

  else

    {

      NS_LOG_INFO ("Error while sending " << m_size << " bytes to "

                                          << peerAddressStringStream.str ());

    }

  if (m_sent < m_count)

    {

      m_sendEvent = Simulator::Schedule (m_interval, &SimpleSendApp::Send, this);

    }

}

} // Namespace ns3

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2013 Universita' di Firenze

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Author: Tommaso Pecorella <tommaso.pecorella@unifi.it>

 */

#ifndef SIMPLE_SEND_APP_H

#define SIMPLE_SEND_APP_H

#include "ns3/application.h"

#include "ns3/event-id.h"

#include "ns3/ptr.h"

#include "ns3/packet-socket-address.h"

namespace ns3 {

class Socket;

class Packet;

/**

 * \ingroup socket

 *

 * \brief A simple client.

 *

 * Sends packets using PacketSocket. It does not require (or use) IP.

 *

 */

class SimpleSendApp : public Application

{

public:

  /**

   * \brief Get the type ID.

   * \return the object TypeId

   */

  static TypeId GetTypeId (void);

  SimpleSendApp ();

  virtual ~SimpleSendApp ();

  /**

   * \brief set the remote address and port

   * \param addr remote address

   */

  void SetRemote (PacketSocketAddress addr);

  /**

   * \brief set the remote address and port

   * \param addr remote address

   */

  void SetRemote (Address addr);

protected:

  virtual void DoDispose (void);

private:

  virtual void StartApplication (void);

  virtual void StopApplication (void);

  /**

   * \brief Send a packet

   */

  void Send (void);

  uint32_t m_count; //!< Maximum number of packets the application will send

  Time m_interval; //!< Packet inter-send time

  uint32_t m_size; //!< Size of the sent packet

  uint32_t m_sent; //!< Counter for sent packets

  Ptr<Socket> m_socket; //!< Socket

  Address m_peerAddress; //!< Remote peer address

  EventId m_sendEvent; //!< Event to send the next packet

};

} // namespace ns3

#endif /* SIMPLE_SEND_APP_H */

        'utils/simple-send-app.cc',

        'utils/simple-recv-app.cc',

        'utils/simple-send-app.h',

        'utils/simple-recv-app.h',