diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_ipip.py | 66 |
1 files changed, 54 insertions, 12 deletions
diff --git a/test/test_ipip.py b/test/test_ipip.py index 582ab5be84c..8d99f83e6c2 100644 --- a/test/test_ipip.py +++ b/test/test_ipip.py @@ -3,7 +3,7 @@ import unittest from scapy.layers.inet6 import IPv6, Ether, IP, UDP -from scapy.all import fragment +from scapy.all import fragment, RandShort from framework import VppTestCase, VppTestRunner from vpp_ip_route import VppIpRoute, VppRoutePath, DpoProto from socket import AF_INET, AF_INET6, inet_pton @@ -60,6 +60,18 @@ class TestIPIP(VppTestCase): def validate(self, rx, expected): self.assertEqual(rx, expected.__class__(str(expected))) + def generate_frags(self, payload_length, fragment_size): + p_ether = Ether(src=self.pg1.remote_mac, dst=self.pg1.local_mac) + p_payload = UDP(sport=1234, dport=1234) / self.payload(payload_length) + p_ip4 = IP(src="1.2.3.4", dst=self.pg0.remote_ip4) + outer_ip4 = (p_ether / IP(src=self.pg1.remote_ip4, + id=RandShort(), + dst=self.pg0.local_ip4) / p_ip4 / p_payload) + frags = fragment(outer_ip4, fragment_size) + p4_reply = (p_ip4 / p_payload) + p4_reply.ttl -= 1 + return frags, p4_reply + def test_ipip4(self): """ ip{v4,v6} over ip4 test """ p_ether = Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac) @@ -142,29 +154,48 @@ class TestIPIP(VppTestCase): for p in rx: self.validate(p[1], p6_reply) + # # Fragmentation / Reassembly and Re-fragmentation + # rv = self.vapi.ip_reassembly_enable_disable( sw_if_index=self.pg1.sw_if_index, enable_ip4=1) - # Decapsulation - p_ether = Ether(src=self.pg1.remote_mac, dst=self.pg1.local_mac) - p_payload = UDP(sport=1234, dport=1234) / self.payload(3123) - p_ip4 = IP(src="1.2.3.4", dst=self.pg0.remote_ip4) - outer_ip4 = (p_ether / IP(src=self.pg1.remote_ip4, - dst=self.pg0.local_ip4) / p_ip4 / p_payload) - frags = fragment(outer_ip4, 1400) - p4_reply = (p_ip4 / p_payload) - p4_reply.ttl -= 1 + # Send lots of fragments, verify reassembled packet + frags, p4_reply = self.generate_frags(3131, 1400) + f = [] + for i in range(0, 1000): + f.extend(frags) + self.pg1.add_stream(f) self.pg_enable_capture() - self.pg1.add_stream(frags) self.pg_start() - rx = self.pg0.get_capture(1) + rx = self.pg0.get_capture(1000) + for p in rx: self.validate(p[1], p4_reply) + f = [] + r = [] + for i in range(1, 90): + frags, p4_reply = self.generate_frags(i * 100, 1000) + f.extend(frags) + r.extend(p4_reply) + self.pg_enable_capture() + self.pg1.add_stream(f) + self.pg_start() + rx = self.pg0.get_capture(89) + i = 0 + for p in rx: + self.validate(p[1], r[i]) + i += 1 + # Now try with re-fragmentation + # + # Send fragments to tunnel head-end, for the tunnel head end + # to reassemble and then refragment + # self.vapi.sw_interface_set_mtu(self.pg0.sw_if_index, [576, 0, 0, 0]) + frags, p4_reply = self.generate_frags(3123, 1200) self.pg_enable_capture() self.pg1.add_stream(frags) self.pg_start() @@ -174,6 +205,17 @@ class TestIPIP(VppTestCase): p4_reply.id = 256 self.validate(reass_pkt, p4_reply) + self.vapi.sw_interface_set_mtu(self.pg0.sw_if_index, [1600, 0, 0, 0]) + frags, p4_reply = self.generate_frags(3123, 1200) + self.pg_enable_capture() + self.pg1.add_stream(frags) + self.pg_start() + rx = self.pg0.get_capture(2) + reass_pkt = reassemble(rx) + p4_reply.ttl -= 1 + p4_reply.id = 512 + self.validate(reass_pkt, p4_reply) + def test_ipip6(self): """ ip{v4,v6} over ip6 test """ p_ether = Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac) |