blob: febd62112cf0ed14c65aba7b268cfc0716874aba [file] [log] [blame]
//
// stupid-interest-sink.cpp
// Abstraction
//
// Created by Ilya Moiseenko on 10.08.11.
// Copyright 2011 UCLA. All rights reserved.
//
#include "stupid-interest-sink.h"
#include "ns3/address.h"
#include "ns3/address-utils.h"
#include "ns3/log.h"
#include "ns3/inet-socket-address.h"
#include "ns3/node.h"
#include "ns3/socket.h"
#include "ns3/udp-socket.h"
#include "ns3/simulator.h"
#include "ns3/socket-factory.h"
#include "ns3/packet.h"
#include "ns3/trace-source-accessor.h"
#include "ns3/udp-socket-factory.h"
using namespace std;
namespace ns3 {
NS_LOG_COMPONENT_DEFINE ("StupidInterestSink");
NS_OBJECT_ENSURE_REGISTERED (StupidInterestSink);
TypeId
StupidInterestSink::GetTypeId (void)
{
static TypeId tid = TypeId ("ns3::StupidInterestSink")
.SetParent<Application> ()
.AddConstructor<StupidInterestSink> ()
.AddAttribute ("Local", "The Address on which to Bind the rx socket.",
AddressValue (),
MakeAddressAccessor (&StupidInterestSink::m_local),
MakeAddressChecker ())
.AddAttribute ("Protocol", "The type id of the protocol to use for the rx socket.",
TypeIdValue (UdpSocketFactory::GetTypeId ()),
MakeTypeIdAccessor (&StupidInterestSink::m_tid),
MakeTypeIdChecker ())
.AddTraceSource ("Rx", "A packet has been received",
MakeTraceSourceAccessor (&StupidInterestSink::m_rxTrace))
;
return tid;
}
StupidInterestSink::StupidInterestSink ()
{
NS_LOG_FUNCTION (this);
m_socket = 0;
m_totalRx = 0;
}
StupidInterestSink::~StupidInterestSink()
{
NS_LOG_FUNCTION (this);
}
uint32_t StupidInterestSink::GetTotalRx () const
{
return m_totalRx;
}
Ptr<Socket>
StupidInterestSink::GetListeningSocket (void) const
{
NS_LOG_FUNCTION (this);
return m_socket;
}
std::list<Ptr<Socket> >
StupidInterestSink::GetAcceptedSockets (void) const
{
NS_LOG_FUNCTION (this);
return m_socketList;
}
void StupidInterestSink::DoDispose (void)
{
NS_LOG_FUNCTION (this);
m_socket = 0;
m_socketList.clear ();
// chain up
Application::DoDispose ();
}
// Application Methods
void StupidInterestSink::StartApplication () // Called at time specified by Start
{
NS_LOG_FUNCTION (this);
// Create the socket if not already
if (!m_socket)
{
m_socket = Socket::CreateSocket (GetNode (), m_tid);
m_socket->Bind (m_local);
m_socket->Listen ();
m_socket->ShutdownSend ();
if (addressUtils::IsMulticast (m_local))
{
Ptr<UdpSocket> udpSocket = DynamicCast<UdpSocket> (m_socket);
if (udpSocket)
{
// equivalent to setsockopt (MCAST_JOIN_GROUP)
udpSocket->MulticastJoinGroup (0, m_local);
}
else
{
NS_FATAL_ERROR ("Error: joining multicast on a non-UDP socket");
}
}
}
m_socket->SetRecvCallback (MakeCallback (&StupidInterestSink::HandleRead, this));
m_socket->SetAcceptCallback (
MakeNullCallback<bool, Ptr<Socket>, const Address &> (),
MakeCallback (&StupidInterestSink::HandleAccept, this));
m_socket->SetCloseCallbacks (
MakeCallback (&StupidInterestSink::HandlePeerClose, this),
MakeCallback (&StupidInterestSink::HandlePeerError, this));
}
void StupidInterestSink::StopApplication () // Called at time specified by Stop
{
NS_LOG_FUNCTION (this);
while(!m_socketList.empty ()) //these are accepted sockets, close them
{
Ptr<Socket> acceptedSocket = m_socketList.front ();
m_socketList.pop_front ();
acceptedSocket->Close ();
}
if (m_socket)
{
m_socket->Close ();
m_socket->SetRecvCallback (MakeNullCallback<void, Ptr<Socket> > ());
}
}
void StupidInterestSink::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;
}
if (InetSocketAddress::IsMatchingType (from))
{
m_totalRx += packet->GetSize ();
InetSocketAddress address = InetSocketAddress::ConvertFrom (from);
NS_LOG_INFO ("Received " << packet->GetSize () << " bytes from " <<
address.GetIpv4 () << " [" << address << "]"
<< " total Rx " << m_totalRx);
//cast address to void , to suppress 'address' set but not used
//compiler warning in optimized builds
(void) address;
}
m_rxTrace (packet, from);
}
}
void StupidInterestSink::HandlePeerClose (Ptr<Socket> socket)
{
NS_LOG_INFO ("PktSink, peerClose");
}
void StupidInterestSink::HandlePeerError (Ptr<Socket> socket)
{
NS_LOG_INFO ("PktSink, peerError");
}
void StupidInterestSink::HandleAccept (Ptr<Socket> s, const Address& from)
{
NS_LOG_FUNCTION (this << s << from);
s->SetRecvCallback (MakeCallback (&StupidInterestSink::HandleRead, this));
m_socketList.push_back (s);
}
} // Namespace ns3