6 from scapy.packet import Raw
7 from scapy.layers.l2 import Ether
8 from scapy.layers.inet import IP, UDP
9 from scapy.layers.inet6 import IPv6
11 from framework import VppTestCase, VppTestRunner
14 class TestPgTun(VppTestCase):
18 super(TestPgTun, self).setUp()
20 # create 3 pg interfaces - one each ethernet, ip4-tun, ip6-tun.
21 self.create_pg_interfaces(range(0, 1))
22 self.pg_interfaces += self.create_pg_ip4_interfaces(range(1, 2))
23 self.pg_interfaces += self.create_pg_ip6_interfaces(range(2, 3))
25 for i in self.pg_interfaces:
28 for i in [self.pg0, self.pg1]:
31 for i in [self.pg0, self.pg2]:
34 self.pg0.resolve_arp()
35 self.pg0.resolve_ndp()
38 for i in self.pg_interfaces:
41 super(TestPgTun, self).tearDown()
43 def test_pg_tun(self):
44 """ IP[46] Tunnel Mode PG """
47 # test that we can send and receive IP encap'd packets on the
53 p = (IP(src=self.pg1.remote_ip4, dst=self.pg0.remote_ip4) /
54 UDP(sport=1234, dport=1234) /
57 rxs = self.send_and_expect(self.pg1, p * N_PKTS, self.pg0)
59 self.assertEqual(rx[Ether].dst, self.pg0.remote_mac)
60 self.assertEqual(rx[IP].dst, self.pg0.remote_ip4)
63 p = (IPv6(src=self.pg2.remote_ip6, dst=self.pg0.remote_ip6) /
64 UDP(sport=1234, dport=1234) /
67 rxs = self.send_and_expect(self.pg2, p * N_PKTS, self.pg0)
69 self.assertEqual(rx[Ether].dst, self.pg0.remote_mac)
70 self.assertEqual(rx[IPv6].dst, self.pg0.remote_ip6)
73 p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
74 IP(src=self.pg0.remote_ip4, dst=self.pg1.remote_ip4) /
75 UDP(sport=1234, dport=1234) /
78 rxs = self.send_and_expect(self.pg0, p * N_PKTS, self.pg1)
81 self.assertFalse(rx.haslayer(Ether))
82 self.assertEqual(rx[IP].dst, self.pg1.remote_ip4)
85 p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) /
86 IPv6(src=self.pg0.remote_ip6, dst=self.pg2.remote_ip6) /
87 UDP(sport=1234, dport=1234) /
90 rxs = self.send_and_expect(self.pg0, p * N_PKTS, self.pg2)
93 self.assertFalse(rx.haslayer(Ether))
94 self.assertEqual(rx[IPv6].dst, self.pg2.remote_ip6)
97 if __name__ == '__main__':
98 unittest.main(testRunner=VppTestRunner)