3 from framework import VppTestCase
4 from ipaddress import IPv4Address
5 from ipaddress import IPv6Address
6 from scapy.contrib.gtp import *
7 from scapy.all import *
10 class TestSRv6EndMGTP4E(VppTestCase):
11 """SRv6 End.M.GTP4.E (SRv6 -> GTP-U)"""
15 super(TestSRv6EndMGTP4E, cls).setUpClass()
17 cls.create_pg_interfaces(range(2))
18 cls.pg_if_i = cls.pg_interfaces[0]
19 cls.pg_if_o = cls.pg_interfaces[1]
21 cls.pg_if_i.config_ip6()
22 cls.pg_if_o.config_ip4()
24 cls.ip4_dst = cls.pg_if_o.remote_ip4
25 # cls.ip4_src = cls.pg_if_o.local_ip4
26 cls.ip4_src = "192.168.192.10"
28 for pg_if in cls.pg_interfaces:
33 super(TestSRv6EndMGTP4E, cls).tearDownClass()
36 def create_packets(self, inner):
37 ip4_dst = IPv4Address(str(self.ip4_dst))
38 # 32bit prefix + 32bit IPv4 DA + 8bit + 32bit TEID + 24bit
39 dst = b"\xaa" * 4 + ip4_dst.packed + b"\x11" + b"\xbb" * 4 + b"\x11" * 3
40 ip6_dst = IPv6Address(dst)
42 ip4_src = IPv4Address(str(self.ip4_src))
43 # 64bit prefix + 32bit IPv4 SA + 16 bit port + 16bit
44 src = b"\xcc" * 8 + ip4_src.packed + b"\xdd" * 2 + b"\x11" * 2
45 ip6_src = IPv6Address(src)
47 self.logger.info("ip4 dst: {}".format(ip4_dst))
48 self.logger.info("ip4 src: {}".format(ip4_src))
49 self.logger.info("ip6 dst (remote srgw): {}".format(ip6_dst))
50 self.logger.info("ip6 src (local srgw): {}".format(ip6_src))
56 / IPv6(dst=str(ip6_dst), src=str(ip6_src))
57 / IPv6ExtHdrSegmentRouting()
59 / UDP(sport=1000, dport=23)
61 self.logger.info(pkt.show2(dump=True))
66 def test_srv6_mobile(self):
67 """test_srv6_mobile"""
68 pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")])
71 "sr localsid address {} behavior end.m.gtp4.e ".format(pkts[0]["IPv6"].dst)
72 + "v4src_position 64 fib-table 0"
74 self.logger.info(self.vapi.cli("show sr localsid"))
76 self.vapi.cli("clear errors")
78 self.pg0.add_stream(pkts)
79 self.pg_enable_capture(self.pg_interfaces)
82 self.logger.info(self.vapi.cli("show errors"))
83 self.logger.info(self.vapi.cli("show int address"))
85 capture = self.pg1.get_capture(len(pkts))
88 self.logger.info(pkt.show2(dump=True))
89 self.assertEqual(pkt[IP].dst, self.ip4_dst)
90 self.assertEqual(pkt[IP].src, self.ip4_src)
91 self.assertEqual(pkt[GTP_U_Header].teid, 0xBBBBBBBB)
94 class TestSRv6TMGTP4D(VppTestCase):
95 """SRv6 T.M.GTP4.D (GTP-U -> SRv6)"""
99 super(TestSRv6TMGTP4D, cls).setUpClass()
101 cls.create_pg_interfaces(range(2))
102 cls.pg_if_i = cls.pg_interfaces[0]
103 cls.pg_if_o = cls.pg_interfaces[1]
105 cls.pg_if_i.config_ip4()
106 cls.pg_if_i.config_ip6()
107 cls.pg_if_o.config_ip4()
108 cls.pg_if_o.config_ip6()
110 cls.ip4_dst = "1.1.1.1"
111 cls.ip4_src = "2.2.2.2"
113 cls.ip6_dst = cls.pg_if_o.remote_ip6
115 for pg_if in cls.pg_interfaces:
118 pg_if.resolve_ndp(timeout=5)
121 super(TestSRv6TMGTP4D, cls).tearDownClass()
124 def create_packets(self, inner):
125 ip4_dst = IPv4Address(str(self.ip4_dst))
127 ip4_src = IPv4Address(str(self.ip4_src))
129 self.logger.info("ip4 dst: {}".format(ip4_dst))
130 self.logger.info("ip4 src: {}".format(ip4_src))
136 / IP(dst=str(ip4_dst), src=str(ip4_src))
137 / UDP(sport=2152, dport=2152)
138 / GTP_U_Header(gtp_type="g_pdu", teid=200)
140 / UDP(sport=1000, dport=23)
142 self.logger.info(pkt.show2(dump=True))
147 def test_srv6_mobile(self):
148 """test_srv6_mobile"""
149 pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")])
151 self.vapi.cli("set sr encaps source addr A1::1")
152 self.vapi.cli("sr policy add bsid D4:: next D2:: next D3::")
154 "sr policy add bsid D5:: behavior t.m.gtp4.d D4::/32 "
155 + "v6src_prefix C1::/64 nhtype ipv6 fib-table 0 drop-in"
157 self.vapi.cli("sr steer l3 {}/32 via bsid D5::".format(self.ip4_dst))
158 self.vapi.cli("ip route add D2::/32 via {}".format(self.ip6_dst))
160 self.logger.info(self.vapi.cli("show sr steer"))
161 self.logger.info(self.vapi.cli("show sr policies"))
163 self.vapi.cli("clear errors")
165 self.pg0.add_stream(pkts)
166 self.pg_enable_capture(self.pg_interfaces)
169 self.logger.info(self.vapi.cli("show errors"))
170 self.logger.info(self.vapi.cli("show int address"))
172 capture = self.pg1.get_capture(len(pkts))
175 self.logger.info(pkt.show2(dump=True))
177 "GTP4.D Address={}".format(
178 str(pkt[IPv6ExtHdrSegmentRouting].addresses[0])
182 str(pkt[IPv6ExtHdrSegmentRouting].addresses[0]), "d4:0:101:101::c800:0"
186 class TestSRv6EndMGTP6E(VppTestCase):
187 """SRv6 End.M.GTP6.E"""
191 super(TestSRv6EndMGTP6E, cls).setUpClass()
193 cls.create_pg_interfaces(range(2))
194 cls.pg_if_i = cls.pg_interfaces[0]
195 cls.pg_if_o = cls.pg_interfaces[1]
197 cls.pg_if_i.config_ip6()
198 cls.pg_if_o.config_ip6()
200 cls.ip6_nhop = cls.pg_if_o.remote_ip6
202 for pg_if in cls.pg_interfaces:
204 pg_if.resolve_ndp(timeout=5)
207 super(TestSRv6EndMGTP6E, cls).tearDownClass()
210 def create_packets(self, inner):
211 # 64bit prefix + 8bit QFI + 32bit TEID + 24bit
212 dst = b"\xaa" * 8 + b"\x00" + b"\xbb" * 4 + b"\x00" * 3
213 ip6_dst = IPv6Address(dst)
215 self.ip6_dst = ip6_dst
217 src = b"\xcc" * 8 + b"\xdd" * 4 + b"\x11" * 4
218 ip6_src = IPv6Address(src)
220 self.ip6_src = ip6_src
226 / IPv6(dst=str(ip6_dst), src=str(ip6_src))
227 / IPv6ExtHdrSegmentRouting(
228 segleft=1, lastentry=0, tag=0, addresses=["a1::1"]
231 / UDP(sport=1000, dport=23)
233 self.logger.info(pkt.show2(dump=True))
238 def test_srv6_mobile(self):
239 """test_srv6_mobile"""
240 pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")])
243 "sr localsid prefix {}/64 behavior end.m.gtp6.e fib-table 0".format(
247 self.vapi.cli("ip route add a1::/64 via {}".format(self.ip6_nhop))
248 self.logger.info(self.vapi.cli("show sr localsid"))
250 self.vapi.cli("clear errors")
252 self.pg0.add_stream(pkts)
253 self.pg_enable_capture(self.pg_interfaces)
256 self.logger.info(self.vapi.cli("show errors"))
257 self.logger.info(self.vapi.cli("show int address"))
259 capture = self.pg1.get_capture(len(pkts))
262 self.logger.info(pkt.show2(dump=True))
263 self.assertEqual(pkt[IPv6].dst, "a1::1")
264 self.assertEqual(pkt[IPv6].src, str(self.ip6_src))
265 self.assertEqual(pkt[GTP_U_Header].teid, 0xBBBBBBBB)
268 class TestSRv6EndMGTP6D(VppTestCase):
269 """SRv6 End.M.GTP6.D"""
273 super(TestSRv6EndMGTP6D, cls).setUpClass()
275 cls.create_pg_interfaces(range(2))
276 cls.pg_if_i = cls.pg_interfaces[0]
277 cls.pg_if_o = cls.pg_interfaces[1]
279 cls.pg_if_i.config_ip6()
280 cls.pg_if_o.config_ip6()
282 cls.ip6_nhop = cls.pg_if_o.remote_ip6
284 cls.ip6_dst = "2001::1"
285 cls.ip6_src = "2002::1"
287 for pg_if in cls.pg_interfaces:
289 pg_if.resolve_ndp(timeout=5)
292 super(TestSRv6EndMGTP6D, cls).tearDownClass()
295 def create_packets(self, inner):
296 ip6_dst = IPv6Address(str(self.ip6_dst))
298 ip6_src = IPv6Address(str(self.ip6_src))
300 self.logger.info("ip6 dst: {}".format(ip6_dst))
301 self.logger.info("ip6 src: {}".format(ip6_src))
307 / IPv6(dst=str(ip6_dst), src=str(ip6_src))
308 / UDP(sport=2152, dport=2152)
309 / GTP_U_Header(gtp_type="g_pdu", teid=200)
311 / UDP(sport=1000, dport=23)
313 self.logger.info(pkt.show2(dump=True))
318 def test_srv6_mobile(self):
319 """test_srv6_mobile"""
320 pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")])
322 self.vapi.cli("set sr encaps source addr A1::1")
323 self.vapi.cli("sr policy add bsid D4:: next D2:: next D3::")
325 "sr localsid prefix 2001::/64 behavior end.m.gtp6.d "
326 + "D4::/64 fib-table 0 drop-in"
328 self.vapi.cli("ip route add D2::/64 via {}".format(self.ip6_nhop))
330 self.logger.info(self.vapi.cli("show sr policies"))
331 self.logger.info(self.vapi.cli("show sr localsid"))
333 self.vapi.cli("clear errors")
335 self.pg0.add_stream(pkts)
336 self.pg_enable_capture(self.pg_interfaces)
339 self.logger.info(self.vapi.cli("show errors"))
340 self.logger.info(self.vapi.cli("show int address"))
342 capture = self.pg1.get_capture(len(pkts))
345 self.logger.info(pkt.show2(dump=True))
347 "GTP6.D SID0={}".format(str(pkt[IPv6ExtHdrSegmentRouting].addresses[0]))
350 "GTP6.D SID1={}".format(str(pkt[IPv6ExtHdrSegmentRouting].addresses[1]))
352 self.assertEqual(str(pkt[IPv6ExtHdrSegmentRouting].addresses[0]), "2001::1")
354 str(pkt[IPv6ExtHdrSegmentRouting].addresses[1]), "d4::c800:0"