A Discrete-Event Network Simulator
API
loopback.cc
Go to the documentation of this file.
1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3  * Copyright (c) 2010 IITP RAS
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation;
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17  *
18  * Authors: Pavel Boyko <boyko@iitp.ru>
19  */
20 
21 #include "ns3/test.h"
22 #include "ns3/simulator.h"
23 #include "ns3/socket-factory.h"
24 #include "ns3/udp-socket-factory.h"
25 #include "ns3/mobility-helper.h"
26 #include "ns3/double.h"
27 #include "ns3/uinteger.h"
28 #include "ns3/string.h"
29 #include "ns3/boolean.h"
30 #include "ns3/yans-wifi-helper.h"
31 #include "ns3/internet-stack-helper.h"
32 #include "ns3/ipv4-address-helper.h"
33 #include "ns3/abort.h"
34 #include "ns3/udp-echo-helper.h"
35 #include "ns3/mobility-model.h"
36 #include "ns3/pcap-file.h"
37 #include "ns3/aodv-helper.h"
38 #include "ns3/v4ping.h"
39 #include "ns3/config.h"
40 #include "ns3/constant-position-mobility-model.h"
41 #include "ns3/names.h"
42 #include <sstream>
43 
44 namespace ns3 {
45 namespace aodv {
46 
52 class LoopbackTestCase : public TestCase
53 {
54  uint32_t m_count;
58  uint16_t m_echoSendPort;
59  uint16_t m_echoReplyPort;
60 
65  void SendData (Ptr<Socket> socket);
70  void ReceivePkt (Ptr<Socket> socket);
75  void EchoData (Ptr<Socket> socket);
76 
77 public:
79  void DoRun ();
80 };
81 
83  : TestCase ("UDP Echo 127.0.0.1 test"),
84  m_count (0)
85 {
86  m_echoSendPort = 1233;
87  m_echoReplyPort = 1234;
88 }
89 
91 {
92  Ptr<Packet> receivedPacket = socket->Recv (std::numeric_limits<uint32_t>::max (), 0);
93 
94  m_count++;
95 }
96 
97 void
99 {
100  Address from;
101  Ptr<Packet> receivedPacket = socket->RecvFrom (std::numeric_limits<uint32_t>::max (), 0, from);
102 
105 
106  receivedPacket->RemoveAllPacketTags ();
107  receivedPacket->RemoveAllByteTags ();
108 
109  socket->SendTo (receivedPacket, 0, to);
110 }
111 
112 void
114 {
116  socket->SendTo (Create<Packet> (123), 0, realTo);
117 
118  Simulator::ScheduleWithContext (socket->GetNode ()->GetId (), Seconds (1.0),
119  &LoopbackTestCase::SendData, this, socket);
120 }
121 
122 void
124 {
126  nodes.Create (1);
127  Ptr<MobilityModel> m = CreateObject<ConstantPositionMobilityModel> ();
128  m->SetPosition (Vector (0, 0, 0));
129  nodes.Get (0)->AggregateObject (m);
130  // Setup WiFi
131  WifiMacHelper wifiMac;
132  wifiMac.SetType ("ns3::AdhocWifiMac");
135  wifiPhy.SetChannel (wifiChannel.Create ());
137  wifi.SetRemoteStationManager ("ns3::ConstantRateWifiManager", "DataMode", StringValue ("OfdmRate6Mbps"), "RtsCtsThreshold", StringValue ("2200"));
138  NetDeviceContainer devices = wifi.Install (wifiPhy, wifiMac, nodes);
139 
140  // Setup TCP/IP & AODV
141  AodvHelper aodv; // Use default parameters here
142  InternetStackHelper internetStack;
143  internetStack.SetRoutingHelper (aodv);
144  internetStack.Install (nodes);
146  address.SetBase ("10.1.1.0", "255.255.255.0");
148 
149  // Setup echos
150  Ptr<SocketFactory> socketFactory = nodes.Get (0)->GetObject<UdpSocketFactory> ();
151  m_rxSocket = socketFactory->CreateSocket ();
154 
155  m_echoSocket = socketFactory->CreateSocket ();
158 
159  m_txSocket = socketFactory->CreateSocket ();
160 
163 
164  // Run
165  Simulator::Stop (Seconds (5));
166  Simulator::Run ();
167 
168  m_txSocket->Close ();
169  m_echoSocket->Close ();
170  m_rxSocket->Close ();
171 
173 
174  // Check that 4 packets delivered
175  NS_TEST_ASSERT_MSG_EQ (m_count, 4, "Exactly 4 echo replies must be delivered.");
176 }
177 
185 {
186 public:
187  AodvLoopbackTestSuite () : TestSuite ("routing-aodv-loopback", SYSTEM)
188  {
189  SetDataDir (NS_TEST_SOURCEDIR);
190  // UDP Echo loopback test case
192  }
194 
195 
196 } // namespace aodv
197 } // namespace ns3
an Inet address class
This test suite implements a System Test.
Definition: test.h:1354
uint32_t GetId(void) const
Definition: node.cc:107
holds a vector of std::pair of Ptr<Ipv4> and interface index.
Hold variables of type string.
Definition: string.h:41
Make it easy to create and manage PHY objects for the YANS model.
static YansWifiChannelHelper Default(void)
Create a channel helper in a default working state.
void DoRun()
Implementation to actually run this TestCase.
Definition: loopback.cc:123
A suite of tests to run.
Definition: test.h:1343
uint16_t m_echoSendPort
echo send port;
Definition: loopback.cc:58
static void Run(void)
Run the simulation.
Definition: simulator.cc:170
AODV Loopback test suite.
Definition: loopback.cc:184
aggregate IP/TCP/UDP functionality to existing Nodes.
AODV loopback UDP echo test case.
Definition: loopback.cc:52
Ptr< Socket > m_echoSocket
echo socket;
Definition: loopback.cc:56
ns3::aodv::AodvLoopbackTestSuite g_aodvLoopbackTestSuite
the test suite
static YansWifiPhyHelper Default(void)
Create a PHY helper in a default working state.
encapsulates test code
Definition: test.h:1153
helps to create WifiNetDevice objects
Definition: wifi-helper.h:300
a polymophic address class
Definition: address.h:90
Ptr< YansWifiChannel > Create(void) const
void SetChannel(Ptr< YansWifiChannel > channel)
Ptr< Socket > m_txSocket
transmit socket;
Definition: loopback.cc:55
void RemoveAllPacketTags(void)
Remove all packet tags.
Definition: packet.cc:891
nodes
Definition: first.py:32
#define max(a, b)
Definition: 80211b.c:43
void AddTestCase(TestCase *testCase, TestDuration duration=QUICK)
Add an individual child TestCase to this test suite.
Definition: test.cc:299
void SendData(Ptr< Socket > socket)
Send data function.
Definition: loopback.cc:113
void ReceivePkt(Ptr< Socket > socket)
Receive packet function.
Definition: loopback.cc:90
#define NS_TEST_ASSERT_MSG_EQ(actual, limit, msg)
Test that an actual and expected (limit) value are equal and report and abort if not.
Definition: test.h:166
holds a vector of ns3::NetDevice pointers
Callback< R > MakeCallback(R(T::*memPtr)(void), OBJ objPtr)
Definition: callback.h:1610
void SetRecvCallback(Callback< void, Ptr< Socket > >)
Notify application when new data is available to be read.
Definition: socket.cc:128
static void Destroy(void)
Execute the events scheduled with ScheduleDestroy().
Definition: simulator.cc:134
uint32_t m_count
number of packet received;
Definition: loopback.cc:54
virtual int Bind(const Address &address)=0
Allocate a local endpoint for this socket.
Every class exported by the ns3 library is enclosed in the ns3 namespace.
static InetSocketAddress ConvertFrom(const Address &address)
Returns an InetSocketAddress which corresponds to the input Address.
keep track of a set of node pointers.
address
Definition: first.py:44
virtual Ptr< Packet > Recv(uint32_t maxSize, uint32_t flags)=0
Read data from the socket.
manage and create wifi channel objects for the YANS model.
create MAC layers for a ns3::WifiNetDevice.
static Ipv4Address GetLoopback(void)
void SetPosition(const Vector &position)
virtual void SetType(std::string type, std::string n0="", const AttributeValue &v0=EmptyAttributeValue(), std::string n1="", const AttributeValue &v1=EmptyAttributeValue(), std::string n2="", const AttributeValue &v2=EmptyAttributeValue(), std::string n3="", const AttributeValue &v3=EmptyAttributeValue(), std::string n4="", const AttributeValue &v4=EmptyAttributeValue(), std::string n5="", const AttributeValue &v5=EmptyAttributeValue(), std::string n6="", const AttributeValue &v6=EmptyAttributeValue(), std::string n7="", const AttributeValue &v7=EmptyAttributeValue(), std::string n8="", const AttributeValue &v8=EmptyAttributeValue(), std::string n9="", const AttributeValue &v9=EmptyAttributeValue(), std::string n10="", const AttributeValue &v10=EmptyAttributeValue())
Fast test.
Definition: test.h:1159
void Install(std::string nodeName) const
Aggregate implementations of the ns3::Ipv4, ns3::Ipv6, ns3::Udp, and ns3::Tcp classes onto the provid...
wifi
Definition: third.py:96
static void ScheduleWithContext(uint32_t context, Time const &delay, MEM mem_ptr, OBJ obj)
Schedule an event with the given context.
Definition: simulator.h:1484
Ipv4 addresses are stored in host order in this class.
Definition: ipv4-address.h:41
virtual Ptr< Node > GetNode(void) const =0
Return the node this socket is associated with.
static void Stop(void)
Tell the Simulator the calling event should be the last one executed.
Definition: simulator.cc:178
void EchoData(Ptr< Socket > socket)
Echo data function.
Definition: loopback.cc:98
Time Seconds(double value)
Construct a Time in the indicated unit.
Definition: nstime.h:1070
interfaces
Definition: first.py:48
void SetDataDir(std::string directory)
Set the data directory where reference trace files can be found.
Definition: test.cc:465
void RemoveAllByteTags(void)
Remove all byte tags stored in this packet.
Definition: packet.cc:371
virtual int SendTo(Ptr< Packet > p, uint32_t flags, const Address &toAddress)=0
Send data to a specified peer.
A helper class to make life easier while doing simple IPv4 address assignment in scripts.
Helper class that adds AODV routing to nodes.
Definition: aodv-helper.h:34
API to create UDP socket instances.
virtual Ptr< Packet > RecvFrom(uint32_t maxSize, uint32_t flags, Address &fromAddress)=0
Read a single packet from the socket and retrieve the sender address.
devices
Definition: first.py:39
virtual int Close(void)=0
Close a socket.
void SetRoutingHelper(const Ipv4RoutingHelper &routing)
uint16_t m_echoReplyPort
echo reply port;
Definition: loopback.cc:59
Ipv4Address GetIpv4(void) const
Ptr< Socket > m_rxSocket
receive socket;
Definition: loopback.cc:57