blob: db8392d12bfb56ca92cfde1865800ecfa2f7d4b2 [file] [log] [blame]
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -07001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */
2/*
3 * Copyright (c) 2011 University of California, Los Angeles
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 * Author: Ilya Moiseenko <iliamo@cs.ucla.edu>
19 */
20
21#include "annotated-topology-reader.h"
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -070022
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -080023#include "ns3/nstime.h"
24#include "ns3/log.h"
25#include "ns3/assert.h"
26#include "ns3/names.h"
27#include "ns3/net-device-container.h"
28#include "ns3/point-to-point-helper.h"
29#include "ns3/point-to-point-net-device.h"
30#include "ns3/internet-stack-helper.h"
31#include "ns3/ipv4-address-helper.h"
32#include "ns3/ipv4-global-routing-helper.h"
33#include "ns3/drop-tail-queue.h"
34#include "ns3/ipv4-interface.h"
35#include "ns3/ipv4.h"
36#include "ns3/string.h"
37#include "ns3/pointer.h"
38#include "ns3/uinteger.h"
39#include "ns3/ipv4-address.h"
40
41#include "ns3/constant-position-mobility-model.h"
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -080042
43#include <boost/foreach.hpp>
44#include <boost/lexical_cast.hpp>
45
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -080046#include <set>
47
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -070048using namespace std;
49
50namespace ns3
51{
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -070052
53NS_LOG_COMPONENT_DEFINE ("AnnotatedTopologyReader");
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -070054
Alexander Afanasyev011b8592011-12-21 14:45:27 -080055AnnotatedTopologyReader::AnnotatedTopologyReader (const std::string &path, double scale/*=1.0*/)
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -080056 : m_path (path)
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -080057 , m_randX (0, 100.0)
58 , m_randY (0, 100.0)
Alexander Afanasyev011b8592011-12-21 14:45:27 -080059 , m_scale (scale)
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -070060{
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -080061 NS_LOG_FUNCTION (this);
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -080062
Alexander Afanasyevbdc0d982011-12-16 01:15:26 -080063 SetMobilityModel ("ns3::ConstantPositionMobilityModel");
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -070064}
65
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -080066void
67AnnotatedTopologyReader::SetBoundingBox (double ulx, double uly, double lrx, double lry)
68{
69 NS_LOG_FUNCTION (this << ulx << uly << lrx << lry);
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -080070
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -080071 m_randX = UniformVariable (ulx, lrx);
72 m_randY = UniformVariable (uly, lry);
73}
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -080074
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -080075void
76AnnotatedTopologyReader::SetMobilityModel (const std::string &model)
77{
78 NS_LOG_FUNCTION (this << model);
79 m_mobilityFactory.SetTypeId (model);
80}
81
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -070082AnnotatedTopologyReader::~AnnotatedTopologyReader ()
83{
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -080084 NS_LOG_FUNCTION (this);
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -070085}
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -080086
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -080087Ptr<Node>
88AnnotatedTopologyReader::CreateNode (const std::string name)
89{
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -080090 return CreateNode (name, m_randX.GetValue (), m_randY.GetValue ());
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -080091}
92
93Ptr<Node>
94AnnotatedTopologyReader::CreateNode (const std::string name, double posX, double posY)
95{
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -080096 NS_LOG_FUNCTION (this << name << posX << posY);
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -080097 Ptr<Node> node = CreateObject<Node> ();
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -080098 Ptr<MobilityModel> loc = DynamicCast<MobilityModel> (m_mobilityFactory.Create ());
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -080099 node->AggregateObject (loc);
100
101 loc->SetPosition (Vector (posX, posY, 0));
102
103 Names::Add (m_path, name, node);
Alexander Afanasyev5beb35a2011-12-21 16:45:13 -0800104 m_nodes.Add (node);
Alexander Afanasyevae3b7c32011-12-13 13:20:06 -0800105
106 return node;
107}
108
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -0700109NodeContainer
Alexander Afanasyev5beb35a2011-12-21 16:45:13 -0800110AnnotatedTopologyReader::GetNodes () const
111{
112 return m_nodes;
113}
114
Ilya Moiseenkoad9e8ab2012-01-11 19:58:34 -0800115std::list<TopologyReader::Link>
116AnnotatedTopologyReader::GetLinks () const
117{
118 return m_linksList;
119}
Alexander Afanasyev5beb35a2011-12-21 16:45:13 -0800120
121NodeContainer
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -0700122AnnotatedTopologyReader::Read (void)
123{
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800124 ifstream topgen;
125 topgen.open (GetFileName ().c_str ());
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -0700126
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800127 if ( !topgen.is_open () )
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700128 {
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800129 NS_LOG_ERROR ("Cannot open file " << GetFileName () << " for reading");
Alexander Afanasyev5beb35a2011-12-21 16:45:13 -0800130 return m_nodes;
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700131 }
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800132
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800133 while (!topgen.eof ())
134 {
135 string line;
136 getline (topgen, line);
137
138 if (line == "router") break;
139 }
140
141 while (!topgen.eof ())
142 {
143 string line;
144 getline (topgen,line);
145 if (line[0] == '#') continue; // comments
146 if (line=="link") break; // stop reading nodes
147
148 istringstream lineBuffer (line);
149 string name, city;
150 double latitude, longitude;
151
152 lineBuffer >> name >> city >> latitude >> longitude;
Alexander Afanasyev5beb35a2011-12-21 16:45:13 -0800153 if (name.empty ()) continue;
154
Alexander Afanasyev011b8592011-12-21 14:45:27 -0800155 Ptr<Node> node = CreateNode (name, m_scale*longitude, -m_scale*latitude);
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800156 }
157
158 map<string, set<string> > processedLinks; // to eliminate duplications
159
160 // SeekToSection ("link");
161 while (!topgen.eof ())
162 {
163 string line;
164 getline (topgen,line);
165 if (line == "") continue;
166 if (line[0] == '#') continue; // comments
167
168 // NS_LOG_DEBUG ("Input: [" << line << "]");
169
170 istringstream lineBuffer (line);
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800171 string from, to, capacity, metric, delay, maxPackets;
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800172
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800173 lineBuffer >> from >> to >> capacity >> metric >> delay >> maxPackets;
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800174
175 if (processedLinks[to].size () != 0 &&
176 processedLinks[to].find (from) != processedLinks[to].end ())
177 {
178 continue; // duplicated link
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -0700179 }
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800180 processedLinks[from].insert (to);
181
182 Ptr<Node> fromNode = Names::Find<Node> (m_path, from);
183 NS_ASSERT (fromNode != 0);
184 Ptr<Node> toNode = Names::Find<Node> (m_path, to);
185 NS_ASSERT (fromNode != 0);
186
187 Link link (fromNode, from, toNode, to);
188
189 link.SetAttribute ("DataRate", capacity);
190 link.SetAttribute ("OSPF", metric);
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -0800191
192 if (!delay.empty ())
193 link.SetAttribute ("Delay", delay);
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800194 if (!maxPackets.empty ())
195 link.SetAttribute ("MaxPackets", maxPackets);
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800196
197 AddLink (link);
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800198 NS_LOG_DEBUG ("New link " << from << " <==> " << to << " / " << capacity << " with " << metric << " metric (" << delay << ", " << maxPackets << ")");
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -0700199 }
200
Alexander Afanasyev5beb35a2011-12-21 16:45:13 -0800201 NS_LOG_INFO ("Annotated topology created with " << m_nodes.GetN () << " nodes and " << LinksSize () << " links");
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800202 topgen.close ();
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -0700203
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800204 ApplySettings ();
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800205
Alexander Afanasyev5beb35a2011-12-21 16:45:13 -0800206 return m_nodes;
Ilya Moiseenko7dd43be2011-08-18 18:57:12 -0700207}
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700208
Ilya Moiseenko58d26672011-12-08 13:48:06 -0800209void
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800210AnnotatedTopologyReader::AssignIpv4Addresses (Ipv4Address base)
Ilya Moiseenko58d26672011-12-08 13:48:06 -0800211{
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800212 Ipv4AddressHelper address (base, Ipv4Mask ("/24"));
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700213
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800214 BOOST_FOREACH (const Link &link, m_linksList)
Ilya Moiseenko58d26672011-12-08 13:48:06 -0800215 {
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800216 address.Assign (NetDeviceContainer (link.GetFromNetDevice (),
217 link.GetToNetDevice ()));
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700218
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800219 base = Ipv4Address (base.Get () + 256);
220 address.SetBase (base, Ipv4Mask ("/24"));
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800221 }
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700222
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800223 ApplyOspfMetric ();
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800224}
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700225
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800226void
227AnnotatedTopologyReader::ApplyOspfMetric ()
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800228{
Alexander Afanasyev0777f1c2011-12-12 20:22:45 -0800229 BOOST_FOREACH (const Link &link, m_linksList)
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800230 {
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -0800231 NS_LOG_DEBUG ("OSPF: " << link.GetAttribute ("OSPF"));
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800232 uint16_t metric = boost::lexical_cast<uint16_t> (link.GetAttribute ("OSPF"));
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700233
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800234 {
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800235 Ptr<Ipv4> ipv4 = link.GetFromNode ()->GetObject<Ipv4> ();
236 NS_ASSERT (ipv4 != 0);
Alexander Afanasyev0777f1c2011-12-12 20:22:45 -0800237
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800238 int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetFromNetDevice ());
239 NS_ASSERT (interfaceId >= 0);
Alexander Afanasyev0777f1c2011-12-12 20:22:45 -0800240
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800241 ipv4->SetMetric (interfaceId,metric);
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800242 }
Ilya Moiseenko1eff17d2011-08-17 10:55:53 -0700243
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800244 {
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800245 Ptr<Ipv4> ipv4 = link.GetToNode ()->GetObject<Ipv4> ();
246 NS_ASSERT (ipv4 != 0);
247
248 int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetToNetDevice ());
249 NS_ASSERT (interfaceId >= 0);
250
251 ipv4->SetMetric (interfaceId,metric);
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800252 }
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800253 }
254}
255
256void
257AnnotatedTopologyReader::ApplySettings ()
258{
259 PointToPointHelper p2p;
Ilya Moiseenko58d26672011-12-08 13:48:06 -0800260
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800261 BOOST_FOREACH (Link &link, m_linksList)
262 {
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800263 string tmp;
264
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -0800265 if (link.GetAttributeFailSafe ("DataRate", tmp))
266 {
Alexander Afanasyev8f5a9bb2011-12-18 19:49:02 -0800267 NS_LOG_INFO ("DataRate = " + link.GetAttribute("DataRate"));
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800268 p2p.SetDeviceAttribute ("DataRate", StringValue (link.GetAttribute ("DataRate")));
Alexander Afanasyev7dbdcaf2011-12-13 21:40:37 -0800269 }
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800270
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800271 if (link.GetAttributeFailSafe ("Delay", tmp))
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800272 {
Alexander Afanasyev8f5a9bb2011-12-18 19:49:02 -0800273 NS_LOG_INFO ("Delay = " + link.GetAttribute("Delay"));
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800274 p2p.SetChannelAttribute ("Delay", StringValue (link.GetAttribute ("Delay")));
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800275 }
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800276
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800277 NetDeviceContainer nd = p2p.Install(link.GetFromNode (), link.GetToNode ());
278 link.SetNetDevices (nd.Get (0), nd.Get (1));
279
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800280 if (link.GetAttributeFailSafe ("MaxPackets", tmp))
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800281 {
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800282 NS_LOG_INFO ("MaxPackets = " + link.GetAttribute ("MaxPackets"));
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800283
Alexander Afanasyev120bf312011-12-19 01:24:47 -0800284 PointerValue txQueue;
285
286 link.GetToNetDevice ()->GetAttribute ("TxQueue", txQueue);
287 NS_ASSERT (txQueue.Get<DropTailQueue> () != 0);
288 txQueue.Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute ("MaxPackets")));
289
290 link.GetFromNetDevice ()->GetAttribute ("TxQueue", txQueue);
291 NS_ASSERT (txQueue.Get<DropTailQueue> () != 0);
292 txQueue.Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute ("MaxPackets")));
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800293 }
Ilya Moiseenko7e14efa2011-12-12 17:56:22 -0800294
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800295 }
Alexander Afanasyev66e6fd72011-12-12 21:34:51 -0800296}
Alexander Afanasyev8633d5d2011-12-12 18:02:31 -0800297
298}