71 lines
1.6 KiB
C++
71 lines
1.6 KiB
C++
#include <iostream>
|
|
|
|
#include "ns3/internet-node.h"
|
|
#include "ns3/simulator.h"
|
|
#include "ns3/socket-factory.h"
|
|
#include "ns3/socket.h"
|
|
#include "ns3/inet-socket-address.h"
|
|
#include "ns3/nstime.h"
|
|
#include "ns3/packet.h"
|
|
|
|
using namespace ns3;
|
|
|
|
static void
|
|
GenerateTraffic (Ptr<Socket> socket, uint32_t size)
|
|
{
|
|
std::cout << "at=" << Simulator::Now ().GetSeconds () << "s, tx bytes=" << size << std::endl;
|
|
socket->Send (Create<Packet> (size));
|
|
if (size > 0)
|
|
{
|
|
Simulator::Schedule (Seconds (0.5), &GenerateTraffic, socket, size - 50);
|
|
}
|
|
else
|
|
{
|
|
socket->Close ();
|
|
}
|
|
}
|
|
|
|
static void
|
|
SocketPrinter (Ptr<Socket> socket, Ptr<Packet> packet, const Address &from)
|
|
{
|
|
std::cout << "at=" << Simulator::Now ().GetSeconds () << "s, rx bytes=" << packet->GetSize () << std::endl;
|
|
}
|
|
|
|
static void
|
|
PrintTraffic (Ptr<Socket> socket)
|
|
{
|
|
socket->SetRecvCallback (MakeCallback (&SocketPrinter));
|
|
}
|
|
|
|
void
|
|
RunSimulation (void)
|
|
{
|
|
Ptr<Node> a = CreateObject<InternetNode> ();
|
|
|
|
TypeId tid = TypeId::LookupByName ("Udp");
|
|
Ptr<SocketFactory> socketFactory = a->GetObject<SocketFactory> (tid);
|
|
|
|
Ptr<Socket> sink = socketFactory->CreateSocket ();
|
|
InetSocketAddress local = InetSocketAddress (Ipv4Address::GetAny (), 80);
|
|
sink->Bind (local);
|
|
|
|
Ptr<Socket> source = socketFactory->CreateSocket ();
|
|
InetSocketAddress remote = InetSocketAddress (Ipv4Address::GetLoopback (), 80);
|
|
source->Connect (remote);
|
|
|
|
GenerateTraffic (source, 500);
|
|
PrintTraffic (sink);
|
|
|
|
|
|
Simulator::Run ();
|
|
|
|
Simulator::Destroy ();
|
|
}
|
|
|
|
int main (int argc, char *argv[])
|
|
{
|
|
RunSimulation ();
|
|
|
|
return 0;
|
|
}
|