blob: 4cd6a040fd0b85a91f16b47cf3b6d95391c1bfa2 [file] [log] [blame]
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
3 * Copyright (c) 2011-2015 Regents of the University of California.
4 *
5 * This file is part of ndnSIM. See AUTHORS for complete list of ndnSIM authors and
6 * contributors.
7 *
8 * ndnSIM is free software: you can redistribute it and/or modify it under the terms
9 * of the GNU General Public License as published by the Free Software Foundation,
10 * either version 3 of the License, or (at your option) any later version.
11 *
12 * ndnSIM is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
13 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
14 * PURPOSE. See the GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along with
17 * ndnSIM, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
18 **/
19
20#include "helper/ndn-global-routing-helper.hpp"
21
22#include "model/ndn-global-router.hpp"
23#include "model/ndn-l3-protocol.hpp"
Alexander Afanasyevca3c67e2016-09-08 15:48:23 -070024#include "model/ndn-net-device-link-service.hpp"
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -070025
26#include "ns3/channel.h"
27#include "ns3/net-device.h"
28#include "ns3/core-module.h"
29#include "ns3/network-module.h"
30#include "ns3/ndnSIM-module.h"
31#include "ns3/point-to-point-net-device.h"
32#include "ns3/point-to-point-module.h"
33#include "ns3/point-to-point-layout-module.h"
34
35#include "../tests-common.hpp"
36
Alexander Afanasyev929b17e2015-08-13 16:54:09 -070037#include <boost/filesystem.hpp>
38
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -070039namespace ns3 {
40namespace ndn {
41
Alexander Afanasyev929b17e2015-08-13 16:54:09 -070042const boost::filesystem::path TEST_TOPO_TXT = boost::filesystem::path(TEST_CONFIG_PATH) / "topo.txt";
43
44class GlobalRoutingHelperFixture : public CleanupFixture
45{
46public:
47 GlobalRoutingHelperFixture()
48 {
49 boost::filesystem::create_directories(TEST_CONFIG_PATH);
50 }
51
52 ~GlobalRoutingHelperFixture()
53 {
54 boost::filesystem::remove(TEST_TOPO_TXT);
55 }
56};
57
58BOOST_FIXTURE_TEST_SUITE(HelperGlobalRoutingHelper, GlobalRoutingHelperFixture)
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -070059
60BOOST_AUTO_TEST_CASE(CalculateRouteCase1)
61{
Alexander Afanasyev929b17e2015-08-13 16:54:09 -070062 ofstream file1(TEST_TOPO_TXT.string().c_str());
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -070063 file1 << "router\n\n"
64 << "#node city y x mpi-partition\n"
65 << "A1 NA 1 1 1\n"
66 << "B1 NA 80 -40 1\n"
67 << "C1 NA 80 40 1\n\n"
68 << "link\n\n"
69 << "# from to capacity metric delay queue\n"
70 << "A1 B1 10Mbps 100 1ms 100\n"
71 << "A1 C1 10Mbps 50 1ms 100\n"
72 << "B1 C1 10Mbps 1 1ms 100\n";
73 file1.close();
74
75 AnnotatedTopologyReader topologyReader("");
Alexander Afanasyev929b17e2015-08-13 16:54:09 -070076 topologyReader.SetFileName(TEST_TOPO_TXT.string().c_str());
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -070077 topologyReader.Read();
78
79 // Install NDN stack on all nodes
80 ndn::StackHelper ndnHelper;
81 ndnHelper.InstallAll();
82
83 topologyReader.ApplyOspfMetric();
84
85 ndn::GlobalRoutingHelper ndnGlobalRoutingHelper;
86 BOOST_CHECK_NO_THROW(ndnGlobalRoutingHelper.InstallAll());
87
88 ndnGlobalRoutingHelper.AddOrigins("/test/prefix", Names::Find<Node>("C1"));
89 BOOST_CHECK_NO_THROW(ndn::GlobalRoutingHelper::CalculateRoutes());
90
91 auto ndn = Names::Find<Node>("A1")->GetObject<ndn::L3Protocol>();
92 for (const auto& entry : ndn->getForwarder()->getFib()) {
93 bool isFirst = true;
94 for (auto& nextHop : entry.getNextHops()) {
Alexander Afanasyevca3c67e2016-09-08 15:48:23 -070095 auto face = nextHop.getFace();
96 auto linkService = dynamic_cast<NetDeviceLinkService*>(face->getLinkService());
97 if (linkService == nullptr)
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -070098 continue;
Alexander Afanasyevca3c67e2016-09-08 15:48:23 -070099 BOOST_CHECK_EQUAL(Names::FindName(linkService->GetNetDevice()->GetChannel()->GetDevice(1)->GetNode()), "C1");
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -0700100 isFirst = false;
101 }
102 }
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -0700103}
104
105BOOST_AUTO_TEST_CASE(CalculateRouteCase2)
106{
Alexander Afanasyev929b17e2015-08-13 16:54:09 -0700107 ofstream file1(TEST_TOPO_TXT.string().c_str());
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -0700108 file1 << "router\n\n"
109 << "#node city y x mpi-partition\n"
110 << "A2 NA 1 1 1\n"
111 << "B2 NA 80 -40 1\n"
112 << "C2 NA 80 40 1\n\n"
113 << "link\n\n"
114 << "# from to capacity metric delay queue\n"
115 << "A2 B2 10Mbps 100 1ms 100\n"
116 << "A2 C2 10Mbps 500 1ms 100\n"
117 << "B2 C2 10Mbps 1 1ms 100\n";
118 file1.close();
119
120 AnnotatedTopologyReader topologyReader("");
Alexander Afanasyev929b17e2015-08-13 16:54:09 -0700121 topologyReader.SetFileName(TEST_TOPO_TXT.string().c_str());
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -0700122 topologyReader.Read();
123
124 // Install NDN stack on all nodes
125 ndn::StackHelper ndnHelper;
126 ndnHelper.InstallAll();
127
128 topologyReader.ApplyOspfMetric();
129
130 ndn::GlobalRoutingHelper ndnGlobalRoutingHelper;
131 ndnGlobalRoutingHelper.InstallAll();
132
133 ndnGlobalRoutingHelper.AddOrigins("/prefix", Names::Find<Node>("C2"));
134 ndn::GlobalRoutingHelper::CalculateRoutes();
135
136 auto ndn = Names::Find<Node>("A2")->GetObject<ndn::L3Protocol>();
137 for (const auto& entry : ndn->getForwarder()->getFib()) {
138 bool isFirst = true;
139 for (auto& nextHop : entry.getNextHops()) {
Alexander Afanasyevca3c67e2016-09-08 15:48:23 -0700140 auto face = nextHop.getFace();
141 auto linkService = dynamic_cast<NetDeviceLinkService*>(face->getLinkService());
142 if (linkService == nullptr)
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -0700143 continue;
Alexander Afanasyevca3c67e2016-09-08 15:48:23 -0700144 BOOST_CHECK_EQUAL(Names::FindName(linkService->GetNetDevice()->GetChannel()->GetDevice(1)->GetNode()), "B2");
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -0700145 isFirst = false;
146 }
147 }
Yuanzhi Gaodd516fe2015-04-23 04:18:24 -0700148}
149
150BOOST_AUTO_TEST_SUITE_END()
151
152} // namespace ndn
153} // namespace ns3