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
54 IP(src=self.pg1.remote_ip4, dst=self.pg0.remote_ip4)
55 / UDP(sport=1234, dport=1234)
59 rxs = self.send_and_expect(self.pg1, p * N_PKTS, self.pg0)
61 self.assertEqual(rx[Ether].dst, self.pg0.remote_mac)
62 self.assertEqual(rx[IP].dst, self.pg0.remote_ip4)
66 IPv6(src=self.pg2.remote_ip6, dst=self.pg0.remote_ip6)
67 / UDP(sport=1234, dport=1234)
71 rxs = self.send_and_expect(self.pg2, p * N_PKTS, self.pg0)
73 self.assertEqual(rx[Ether].dst, self.pg0.remote_mac)
74 self.assertEqual(rx[IPv6].dst, self.pg0.remote_ip6)
78 Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac)
79 / IP(src=self.pg0.remote_ip4, dst=self.pg1.remote_ip4)
80 / UDP(sport=1234, dport=1234)
84 rxs = self.send_and_expect(self.pg0, p * N_PKTS, self.pg1)
87 self.assertFalse(rx.haslayer(Ether))
88 self.assertEqual(rx[IP].dst, self.pg1.remote_ip4)
92 Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac)
93 / IPv6(src=self.pg0.remote_ip6, dst=self.pg2.remote_ip6)
94 / UDP(sport=1234, dport=1234)
98 rxs = self.send_and_expect(self.pg0, p * N_PKTS, self.pg2)
101 self.assertFalse(rx.haslayer(Ether))
102 self.assertEqual(rx[IPv6].dst, self.pg2.remote_ip6)
105 if __name__ == "__main__":
106 unittest.main(testRunner=VppTestRunner)