1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3 * Copyright (c) 2009 University of Washington
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 */
19
20 #include "ns3/test.h"
21 #include "ns3/ipv6-list-routing.h"
22 #include "ns3/ipv6-route.h"
23 #include "ns3/ipv6-routing-protocol.h"
24
25 using namespace ns3;
26
27 /**
28 * \ingroup internet-test
29 * \ingroup tests
30 *
31 * \brief IPv6 dummy routing class (A)
32 */
33 class Ipv6ARouting : public Ipv6RoutingProtocol {
34 public:
RouteOutput(Ptr<Packet> p,const Ipv6Header & header,Ptr<NetDevice> oif,Socket::SocketErrno & sockerr)35 Ptr<Ipv6Route> RouteOutput (Ptr<Packet> p, const Ipv6Header &header, Ptr<NetDevice> oif, Socket::SocketErrno &sockerr) { return 0; }
RouteInput(Ptr<const Packet> p,const Ipv6Header & header,Ptr<const NetDevice> idev,UnicastForwardCallback ucb,MulticastForwardCallback mcb,LocalDeliverCallback lcb,ErrorCallback ecb)36 bool RouteInput (Ptr<const Packet> p, const Ipv6Header &header, Ptr<const NetDevice> idev,
37 UnicastForwardCallback ucb, MulticastForwardCallback mcb,
38 LocalDeliverCallback lcb, ErrorCallback ecb) { return false; }
NotifyInterfaceUp(uint32_t interface)39 void NotifyInterfaceUp (uint32_t interface) {}
NotifyInterfaceDown(uint32_t interface)40 void NotifyInterfaceDown (uint32_t interface) {}
NotifyAddAddress(uint32_t interface,Ipv6InterfaceAddress address)41 void NotifyAddAddress (uint32_t interface, Ipv6InterfaceAddress address) {}
NotifyRemoveAddress(uint32_t interface,Ipv6InterfaceAddress address)42 void NotifyRemoveAddress (uint32_t interface, Ipv6InterfaceAddress address) {}
NotifyAddRoute(Ipv6Address dst,Ipv6Prefix mask,Ipv6Address nextHop,uint32_t interface,Ipv6Address prefixToUse=Ipv6Address::GetZero ())43 void NotifyAddRoute (Ipv6Address dst, Ipv6Prefix mask, Ipv6Address nextHop, uint32_t interface, Ipv6Address prefixToUse = Ipv6Address::
44 GetZero ()) {}
NotifyRemoveRoute(Ipv6Address dst,Ipv6Prefix mask,Ipv6Address nextHop,uint32_t interface,Ipv6Address prefixToUse)45 void NotifyRemoveRoute (Ipv6Address dst, Ipv6Prefix mask, Ipv6Address nextHop, uint32_t interface, Ipv6Address prefixToUse) {}
SetIpv6(Ptr<Ipv6> ipv6)46 void SetIpv6 (Ptr<Ipv6> ipv6) {}
PrintRoutingTable(Ptr<OutputStreamWrapper> stream,Time::Unit unit) const47 virtual void PrintRoutingTable (Ptr<OutputStreamWrapper> stream, Time::Unit unit) const {};
48 };
49
50 /**
51 * \ingroup internet-test
52 * \ingroup tests
53 *
54 * \brief IPv6 dummy routing class (B)
55 */
56 class Ipv6BRouting : public Ipv6RoutingProtocol {
57 public:
RouteOutput(Ptr<Packet> p,const Ipv6Header & header,Ptr<NetDevice> oif,Socket::SocketErrno & sockerr)58 Ptr<Ipv6Route> RouteOutput (Ptr<Packet> p, const Ipv6Header &header, Ptr<NetDevice> oif, Socket::SocketErrno &sockerr) { return 0; }
RouteInput(Ptr<const Packet> p,const Ipv6Header & header,Ptr<const NetDevice> idev,UnicastForwardCallback ucb,MulticastForwardCallback mcb,LocalDeliverCallback lcb,ErrorCallback ecb)59 bool RouteInput (Ptr<const Packet> p, const Ipv6Header &header, Ptr<const NetDevice> idev,
60 UnicastForwardCallback ucb, MulticastForwardCallback mcb,
61 LocalDeliverCallback lcb, ErrorCallback ecb) { return false; }
NotifyInterfaceUp(uint32_t interface)62 void NotifyInterfaceUp (uint32_t interface) {}
NotifyInterfaceDown(uint32_t interface)63 void NotifyInterfaceDown (uint32_t interface) {}
NotifyAddAddress(uint32_t interface,Ipv6InterfaceAddress address)64 void NotifyAddAddress (uint32_t interface, Ipv6InterfaceAddress address) {}
NotifyRemoveAddress(uint32_t interface,Ipv6InterfaceAddress address)65 void NotifyRemoveAddress (uint32_t interface, Ipv6InterfaceAddress address) {}
NotifyAddRoute(Ipv6Address dst,Ipv6Prefix mask,Ipv6Address nextHop,uint32_t interface,Ipv6Address prefixToUse=Ipv6Address::GetZero ())66 void NotifyAddRoute (Ipv6Address dst, Ipv6Prefix mask, Ipv6Address nextHop, uint32_t interface, Ipv6Address prefixToUse = Ipv6Address::
67 GetZero ()) {}
NotifyRemoveRoute(Ipv6Address dst,Ipv6Prefix mask,Ipv6Address nextHop,uint32_t interface,Ipv6Address prefixToUse)68 void NotifyRemoveRoute (Ipv6Address dst, Ipv6Prefix mask, Ipv6Address nextHop, uint32_t interface, Ipv6Address prefixToUse) {}
SetIpv6(Ptr<Ipv6> ipv6)69 void SetIpv6 (Ptr<Ipv6> ipv6) {}
PrintRoutingTable(Ptr<OutputStreamWrapper> stream,Time::Unit unit) const70 virtual void PrintRoutingTable (Ptr<OutputStreamWrapper> stream, Time::Unit unit) const {};
71 };
72
73 /**
74 * \ingroup internet-test
75 * \ingroup tests
76 *
77 * \brief IPv6 ListRouting negative test.
78 */
79 class Ipv6ListRoutingNegativeTestCase : public TestCase
80 {
81 public:
82 Ipv6ListRoutingNegativeTestCase();
83 virtual void DoRun (void);
84 };
85
Ipv6ListRoutingNegativeTestCase()86 Ipv6ListRoutingNegativeTestCase::Ipv6ListRoutingNegativeTestCase()
87 : TestCase ("Check negative priorities")
88 {
89 }
90 void
DoRun(void)91 Ipv6ListRoutingNegativeTestCase::DoRun (void)
92 {
93 Ptr<Ipv6ListRouting> lr = CreateObject<Ipv6ListRouting> ();
94 Ptr<Ipv6RoutingProtocol> aRouting = CreateObject<Ipv6ARouting> ();
95 Ptr<Ipv6RoutingProtocol> bRouting = CreateObject<Ipv6BRouting> ();
96 // The Ipv6BRouting should be added with higher priority (larger integer value)
97 lr->AddRoutingProtocol (aRouting, -10);
98 lr->AddRoutingProtocol (bRouting, -5);
99 int16_t first = 3;
100 uint32_t num = lr->GetNRoutingProtocols ();
101 NS_TEST_ASSERT_MSG_EQ (num, 2, "100");
102 Ptr<Ipv6RoutingProtocol> firstRp = lr->GetRoutingProtocol (0, first);
103 NS_TEST_ASSERT_MSG_EQ (-5, first, "101");
104 NS_TEST_ASSERT_MSG_EQ (firstRp, bRouting, "102");
105 }
106
107 /**
108 * \ingroup internet-test
109 * \ingroup tests
110 *
111 * \brief IPv6 ListRouting positive test.
112 */
113 class Ipv6ListRoutingPositiveTestCase : public TestCase
114 {
115 public:
116 Ipv6ListRoutingPositiveTestCase();
117 virtual void DoRun (void);
118 };
119
Ipv6ListRoutingPositiveTestCase()120 Ipv6ListRoutingPositiveTestCase::Ipv6ListRoutingPositiveTestCase()
121 : TestCase ("Check positive priorities")
122 {
123 }
124 void
DoRun(void)125 Ipv6ListRoutingPositiveTestCase::DoRun (void)
126 {
127 Ptr<Ipv6ListRouting> lr = CreateObject<Ipv6ListRouting> ();
128 Ptr<Ipv6RoutingProtocol> aRouting = CreateObject<Ipv6ARouting> ();
129 Ptr<Ipv6RoutingProtocol> bRouting = CreateObject<Ipv6BRouting> ();
130 // The Ipv6ARouting should be added with higher priority (larger integer
131 // value) and will be fetched first below
132 lr->AddRoutingProtocol (aRouting, 10);
133 lr->AddRoutingProtocol (bRouting, 5);
134 int16_t first = 3;
135 int16_t second = 3;
136 uint32_t num = lr->GetNRoutingProtocols ();
137 NS_TEST_ASSERT_MSG_EQ (num, 2, "200");
138 Ptr<Ipv6RoutingProtocol> firstRp = lr->GetRoutingProtocol (0, first);
139 NS_TEST_ASSERT_MSG_EQ (10, first, "201");
140 NS_TEST_ASSERT_MSG_EQ (firstRp, aRouting, "202");
141 Ptr<Ipv6RoutingProtocol> secondRp = lr->GetRoutingProtocol (1, second);
142 NS_TEST_ASSERT_MSG_EQ (5, second, "203");
143 NS_TEST_ASSERT_MSG_EQ (secondRp, bRouting, "204");
144 }
145
146 /**
147 * \ingroup internet-test
148 * \ingroup tests
149 *
150 * \brief IPv6 ListRouting TestSuite
151 */
152 class Ipv6ListRoutingTestSuite : public TestSuite
153 {
154 public:
Ipv6ListRoutingTestSuite()155 Ipv6ListRoutingTestSuite()
156 : TestSuite ("ipv6-list-routing", UNIT)
157 {
158 AddTestCase (new Ipv6ListRoutingPositiveTestCase (), TestCase::QUICK);
159 AddTestCase (new Ipv6ListRoutingNegativeTestCase (), TestCase::QUICK);
160 }
161
162 };
163
164 static Ipv6ListRoutingTestSuite g_ipv6ListRoutingTestSuite; //!< Static variable for test initialization
165