A Discrete-Event Network Simulator
API
Loading...
Searching...
No Matches
socket-options-ipv4.cc
Go to the documentation of this file.
1/*
2 * This program is free software; you can redistribute it and/or modify
3 * it under the terms of the GNU General Public License version 2 as
4 * published by the Free Software Foundation;
5 *
6 * This program is distributed in the hope that it will be useful,
7 * but WITHOUT ANY WARRANTY; without even the implied warranty of
8 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
9 * GNU General Public License for more details.
10 *
11 * You should have received a copy of the GNU General Public License
12 * along with this program; if not, write to the Free Software
13 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
14 */
15
16// Network topology
17//
18// n0 n1
19// | |
20// =================
21// LAN
22//
23// - UDP flows from n0 to n1
24
25#include "ns3/applications-module.h"
26#include "ns3/core-module.h"
27#include "ns3/csma-module.h"
28#include "ns3/internet-module.h"
29#include "ns3/network-module.h"
30
31#include <fstream>
32
33using namespace ns3;
34
35NS_LOG_COMPONENT_DEFINE("SocketOptionsIpv4");
36
37void
39{
40 NS_LOG_INFO("Received one packet!");
41 Ptr<Packet> packet = socket->Recv();
42 SocketIpTosTag tosTag;
43 if (packet->RemovePacketTag(tosTag))
44 {
45 NS_LOG_INFO(" TOS = " << (uint32_t)tosTag.GetTos());
46 }
47 SocketIpTtlTag ttlTag;
48 if (packet->RemovePacketTag(ttlTag))
49 {
50 NS_LOG_INFO(" TTL = " << (uint32_t)ttlTag.GetTtl());
51 }
52}
53
54static void
55SendPacket(Ptr<Socket> socket, uint32_t pktSize, uint32_t pktCount, Time pktInterval)
56{
57 if (pktCount > 0)
58 {
59 socket->Send(Create<Packet>(pktSize));
60 Simulator::Schedule(pktInterval, &SendPacket, socket, pktSize, pktCount - 1, pktInterval);
61 }
62 else
63 {
64 socket->Close();
65 }
66}
67
68int
69main(int argc, char* argv[])
70{
71 //
72 // Allow the user to override any of the defaults and the above Bind() at
73 // run-time, via command-line arguments
74 //
75 uint32_t packetSize = 1024;
76 uint32_t packetCount = 10;
77 double packetInterval = 1.0;
78
79 // Socket options for IPv4, currently TOS, TTL, RECVTOS, and RECVTTL
80 uint32_t ipTos = 0;
81 bool ipRecvTos = true;
82 uint32_t ipTtl = 0;
83 bool ipRecvTtl = true;
84
85 CommandLine cmd(__FILE__);
86 cmd.AddValue("PacketSize", "Packet size in bytes", packetSize);
87 cmd.AddValue("PacketCount", "Number of packets to send", packetCount);
88 cmd.AddValue("Interval", "Interval between packets", packetInterval);
89 cmd.AddValue("IP_TOS", "IP_TOS", ipTos);
90 cmd.AddValue("IP_RECVTOS", "IP_RECVTOS", ipRecvTos);
91 cmd.AddValue("IP_TTL", "IP_TTL", ipTtl);
92 cmd.AddValue("IP_RECVTTL", "IP_RECVTTL", ipRecvTtl);
93 cmd.Parse(argc, argv);
94
95 NS_LOG_INFO("Create nodes.");
97 n.Create(2);
98
99 InternetStackHelper internet;
100 internet.Install(n);
101
102 Address serverAddress;
103
104 NS_LOG_INFO("Create channels.");
106 csma.SetChannelAttribute("DataRate", DataRateValue(DataRate(5000000)));
107 csma.SetChannelAttribute("Delay", TimeValue(MilliSeconds(2)));
108 csma.SetDeviceAttribute("Mtu", UintegerValue(1400));
109 NetDeviceContainer d = csma.Install(n);
110
111 NS_LOG_INFO("Assign IP Addresses.");
113 ipv4.SetBase("10.1.1.0", "255.255.255.0");
114 Ipv4InterfaceContainer i = ipv4.Assign(d);
115 serverAddress = Address(i.GetAddress(1));
116
117 NS_LOG_INFO("Create sockets.");
118 // Receiver socket on n1
119 TypeId tid = TypeId::LookupByName("ns3::UdpSocketFactory");
120 Ptr<Socket> recvSink = Socket::CreateSocket(n.Get(1), tid);
122 recvSink->SetIpRecvTos(ipRecvTos);
123 recvSink->SetIpRecvTtl(ipRecvTtl);
124 recvSink->Bind(local);
125 recvSink->SetRecvCallback(MakeCallback(&ReceivePacket));
126
127 // Sender socket on n0
128 Ptr<Socket> source = Socket::CreateSocket(n.Get(0), tid);
130
131 // Set socket options, it is also possible to set the options after the socket has been
132 // created/connected.
133 if (ipTos > 0)
134 {
135 source->SetIpTos(ipTos);
136 }
137
138 if (ipTtl > 0)
139 {
140 source->SetIpTtl(ipTtl);
141 }
142 source->Connect(remote);
143
144 AsciiTraceHelper ascii;
145 csma.EnableAsciiAll(ascii.CreateFileStream("socket-options-ipv4.tr"));
146 csma.EnablePcapAll("socket-options-ipv4", false);
147
148 // Schedule SendPacket
149 Time interPacketInterval = Seconds(packetInterval);
150 Simulator::ScheduleWithContext(source->GetNode()->GetId(),
151 Seconds(1.0),
152 &SendPacket,
153 source,
155 packetCount,
156 interPacketInterval);
157
158 NS_LOG_INFO("Run Simulation.");
161 NS_LOG_INFO("Done.");
162
163 return 0;
164}
a polymophic address class
Definition: address.h:100
Manage ASCII trace files for device models.
Definition: trace-helper.h:173
Ptr< OutputStreamWrapper > CreateFileStream(std::string filename, std::ios::openmode filemode=std::ios::out)
Create and initialize an output stream object we'll use to write the traced bits.
Parse command-line arguments.
Definition: command-line.h:232
build a set of CsmaNetDevice objects
Definition: csma-helper.h:48
Class for representing data rates.
Definition: data-rate.h:90
AttributeValue implementation for DataRate.
an Inet address class
aggregate IP/TCP/UDP functionality to existing Nodes.
void Install(std::string nodeName) const
Aggregate implementations of the ns3::Ipv4, ns3::Ipv6, ns3::Udp, and ns3::Tcp classes onto the provid...
A helper class to make life easier while doing simple IPv4 address assignment in scripts.
void SetBase(Ipv4Address network, Ipv4Mask mask, Ipv4Address base="0.0.0.1")
Set the base network number, network mask and base address.
Ipv4InterfaceContainer Assign(const NetDeviceContainer &c)
Assign IP addresses to the net devices specified in the container based on the current network prefix...
static Ipv4Address GetAny()
holds a vector of std::pair of Ptr<Ipv4> and interface index.
Ipv4Address GetAddress(uint32_t i, uint32_t j=0) const
holds a vector of ns3::NetDevice pointers
keep track of a set of node pointers.
void Create(uint32_t n)
Create n nodes and append pointers to them to the end of this NodeContainer.
Ptr< Node > Get(uint32_t i) const
Get the Ptr<Node> stored in this container at a given index.
Smart pointer class similar to boost::intrusive_ptr.
Definition: ptr.h:78
static EventId Schedule(const Time &delay, FUNC f, Ts &&... args)
Schedule an event to expire after delay.
Definition: simulator.h:568
static void Destroy()
Execute the events scheduled with ScheduleDestroy().
Definition: simulator.cc:140
static void ScheduleWithContext(uint32_t context, const Time &delay, FUNC f, Ts &&... args)
Schedule an event with the given context.
Definition: simulator.h:587
static void Run()
Run the simulation.
Definition: simulator.cc:176
static Ptr< Socket > CreateSocket(Ptr< Node > node, TypeId tid)
This method wraps the creation of sockets that is performed on a given node by a SocketFactory specif...
Definition: socket.cc:72
indicates whether the socket has IP_TOS set.
Definition: socket.h:1269
uint8_t GetTos() const
Get the tag's TOS.
Definition: socket.cc:802
This class implements a tag that carries the socket-specific TTL of a packet to the IP layer.
Definition: socket.h:1122
uint8_t GetTtl() const
Get the tag's TTL.
Definition: socket.cc:609
Simulation virtual time values and global simulation resolution.
Definition: nstime.h:105
AttributeValue implementation for Time.
Definition: nstime.h:1423
a unique identifier for an interface.
Definition: type-id.h:60
static TypeId LookupByName(std::string name)
Get a TypeId by name.
Definition: type-id.cc:839
Hold an unsigned integer type.
Definition: uinteger.h:45
#define NS_LOG_COMPONENT_DEFINE(name)
Define a Log component with a specific name.
Definition: log.h:202
#define NS_LOG_INFO(msg)
Use NS_LOG to output a message of level LOG_INFO.
Definition: log.h:275
Time Seconds(double value)
Construct a Time in the indicated unit.
Definition: nstime.h:1336
Time MilliSeconds(uint64_t value)
Construct a Time in the indicated unit.
Definition: nstime.h:1348
Every class exported by the ns3 library is enclosed in the ns3 namespace.
Callback< R, Args... > MakeCallback(R(T::*memPtr)(Args...), OBJ objPtr)
Build Callbacks for class method members which take varying numbers of arguments and potentially retu...
Definition: callback.h:707
ns cmd
Definition: second.py:33
ns csma
Definition: second.py:56
void ReceivePacket(Ptr< Socket > socket)
static void SendPacket(Ptr< Socket > socket, uint32_t pktSize, uint32_t pktCount, Time pktInterval)
uint32_t pktSize
packet size used for the simulation (in bytes)
static const uint32_t packetSize
Packet size generated at the AP.