src/plugins/srv6-mobile/test/test_srv6_end.py

   1 #!/usr/bin/env python
   2
   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 *
   8
   9
  10 class TestSRv6EndMGTP4E(VppTestCase):
  11     """ SRv6 End.M.GTP4.E (SRv6 -> GTP-U) """
  12
  13     @classmethod
  14     def setUpClass(cls):
  15         super(TestSRv6EndMGTP4E, cls).setUpClass()
  16         try:
  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]
  20
  21             cls.pg_if_i.config_ip6()
  22             cls.pg_if_o.config_ip4()
  23
  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"
  27
  28             for pg_if in cls.pg_interfaces:
  29                 pg_if.admin_up()
  30                 pg_if.resolve_arp()
  31
  32         except Exception:
  33             super(TestSRv6EndMGTP4E, cls).tearDownClass()
  34             raise
  35
  36     def create_packets(self, inner):
  37
  38         ip4_dst = IPv4Address(str(self.ip4_dst))
  39         # 32bit prefix + 32bit IPv4 DA + 8bit + 32bit TEID + 24bit
  40         dst = b'\xaa' * 4 + ip4_dst.packed + \
  41             b'\x11' + b'\xbb' * 4 + b'\x11' * 3
  42         ip6_dst = IPv6Address(dst)
  43
  44         ip4_src = IPv4Address(str(self.ip4_src))
  45         # 64bit prefix + 32bit IPv4 SA + 16 bit port + 16bit
  46         src = b'\xcc' * 8 + ip4_src.packed + \
  47             b'\xdd' * 2 + b'\x11' * 2
  48         ip6_src = IPv6Address(src)
  49
  50         self.logger.info("ip4 dst: {}".format(ip4_dst))
  51         self.logger.info("ip4 src: {}".format(ip4_src))
  52         self.logger.info("ip6 dst (remote srgw): {}".format(ip6_dst))
  53         self.logger.info("ip6 src (local  srgw): {}".format(ip6_src))
  54
  55         pkts = list()
  56         for d, s in inner:
  57             pkt = (Ether() /
  58                    IPv6(dst=str(ip6_dst), src=str(ip6_src)) /
  59                    IPv6ExtHdrSegmentRouting() /
  60                    IPv6(dst=d, src=s) /
  61                    UDP(sport=1000, dport=23))
  62             self.logger.info(pkt.show2(dump=True))
  63             pkts.append(pkt)
  64
  65         return pkts
  66
  67     def test_srv6_end(self):
  68         """ test_srv6_end """
  69         pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")])
  70
  71         self.vapi.cli(
  72             "sr localsid address {} behavior end.m.gtp4.e v4src_position 64"
  73             .format(pkts[0]['IPv6'].dst))
  74         self.logger.info(self.vapi.cli("show sr localsids"))
  75
  76         self.vapi.cli("clear errors")
  77
  78         self.pg0.add_stream(pkts)
  79         self.pg_enable_capture(self.pg_interfaces)
  80         self.pg_start()
  81
  82         self.logger.info(self.vapi.cli("show errors"))
  83         self.logger.info(self.vapi.cli("show int address"))
  84
  85         capture = self.pg1.get_capture(len(pkts))
  86
  87         for pkt in capture:
  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)
  92
  93
  94 class TestSRv6TMTmap(VppTestCase):
  95     """ SRv6 T.M.Tmap (GTP-U -> SRv6) """
  96
  97     @classmethod
  98     def setUpClass(cls):
  99         super(TestSRv6TMTmap, cls).setUpClass()
 100         try:
 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]
 104
 105             cls.pg_if_i.config_ip4()
 106             cls.pg_if_o.config_ip6()
 107
 108             for pg_if in cls.pg_interfaces:
 109                 pg_if.admin_up()
 110                 pg_if.resolve_arp()
 111
 112         except Exception:
 113             super(TestSRv6TMTmap, cls).tearDownClass()
 114             raise
 115
 116
 117 class TestSRv6EndMGTP6E(VppTestCase):
 118     """ SRv6 End.M.GTP6.E """
 119
 120     @classmethod
 121     def setUpClass(cls):
 122         super(TestSRv6EndMGTP6E, cls).setUpClass()
 123         try:
 124             cls.create_pg_interfaces(range(2))
 125             cls.pg_if_i = cls.pg_interfaces[0]
 126             cls.pg_if_o = cls.pg_interfaces[1]
 127
 128             cls.pg_if_i.config_ip4()
 129             cls.pg_if_o.config_ip6()
 130
 131             for pg_if in cls.pg_interfaces:
 132                 pg_if.admin_up()
 133                 pg_if.resolve_arp()
 134
 135         except Exception:
 136             super(TestSRv6EndMGTP6E, cls).tearDownClass()
 137             raise
 138
 139
 140 class TestSRv6EndMGTP6D(VppTestCase):
 141     """ SRv6 End.M.GTP6.D """
 142
 143     @classmethod
 144     def setUpClass(cls):
 145         super(TestSRv6EndMGTP6D, cls).setUpClass()
 146         try:
 147             cls.create_pg_interfaces(range(2))
 148             cls.pg_if_i = cls.pg_interfaces[0]
 149             cls.pg_if_o = cls.pg_interfaces[1]
 150
 151             cls.pg_if_i.config_ip4()
 152             cls.pg_if_o.config_ip6()
 153
 154             for pg_if in cls.pg_interfaces:
 155                 pg_if.admin_up()
 156                 pg_if.resolve_arp()
 157
 158         except Exception:
 159             super(TestSRv6EndMGTP6D, cls).tearDownClass()
 160             raise