summaryrefslogtreecommitdiffstats
path: root/test/test_mtu.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_mtu.py')
-rw-r--r--test/test_mtu.py212
1 files changed, 212 insertions, 0 deletions
diff --git a/test/test_mtu.py b/test/test_mtu.py
new file mode 100644
index 00000000000..abf0c94b268
--- /dev/null
+++ b/test/test_mtu.py
@@ -0,0 +1,212 @@
+#!/usr/bin/env python
+"""IP4 and IP6 MTU functional tests"""
+
+#
+# Add tests for:
+# - sub interfaces
+# - Verify that adjacencies inherit MTU correctly
+# - Verify that sub-interfaces inherit MTU correctly
+# - Different types of interfaces?
+#
+import unittest
+from scapy.layers.inet6 import IPv6, Ether, IP, UDP, ICMPv6PacketTooBig
+from scapy.layers.inet import ICMP
+from framework import VppTestCase, VppTestRunner
+from vpp_ip_route import VppIpRoute, VppRoutePath, DpoProto
+from socket import AF_INET, AF_INET6, inet_pton
+import StringIO
+
+""" Test_mtu is a subclass of VPPTestCase classes.
+ MTU tests.
+"""
+
+
+def reassemble(listoffragments):
+ buffer = StringIO.StringIO()
+ first = listoffragments[0]
+ buffer.seek(20)
+ for pkt in listoffragments:
+ # pkt.show2()
+ buffer.seek(pkt[IP].frag*8)
+ buffer.write(pkt[IP].payload)
+ first.len = len(buffer.getvalue()) + 20
+ first.flags = 0
+ del(first.chksum)
+ header = str(first[IP])[:20]
+ return first[IP].__class__(header + buffer.getvalue())
+
+
+class TestMTU(VppTestCase):
+ """ MTU Test Case """
+
+ @classmethod
+ def setUpClass(cls):
+ super(TestMTU, cls).setUpClass()
+ cls.create_pg_interfaces(range(2))
+ cls.interfaces = list(cls.pg_interfaces)
+
+ def setUp(cls):
+ super(TestMTU, cls).setUp()
+ for i in cls.interfaces:
+ i.admin_up()
+ i.config_ip4()
+ i.config_ip6()
+ i.disable_ipv6_ra()
+ i.resolve_arp()
+ i.resolve_ndp()
+
+ def tearDown(self):
+ super(TestMTU, self).tearDown()
+ if not self.vpp_dead:
+ for i in self.pg_interfaces:
+ i.unconfig_ip4()
+ i.unconfig_ip6()
+ i.admin_down()
+
+ def validate(self, rx, expected):
+ self.assertEqual(rx, expected.__class__(str(expected)))
+
+ def validate_bytes(self, rx, expected):
+ self.assertEqual(rx, expected)
+
+ def payload(self, len):
+ return 'x' * len
+
+ def get_mtu(self, sw_if_index):
+ rv = self.vapi.sw_interface_dump()
+ for i in rv:
+ if i.sw_if_index == sw_if_index:
+ return i.link_mtu
+ return 0
+
+ def test_ip4_mtu(self):
+ """ IP4 MTU test """
+
+ #
+ # TODO: Link MTU is 216 bytes 'off'. Fix when L3 MTU patches committed
+ #
+ mtu_offset = 216
+ p_ether = Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
+ p_ip4 = IP(src=self.pg0.remote_ip4, dst=self.pg1.remote_ip4,
+ flags='DF')
+
+ # TODO: Re-enable when MTU fixes are committed
+ current_mtu = self.get_mtu(self.pg1.sw_if_index)
+ current_mtu -= mtu_offset
+
+ p_payload = UDP(sport=1234, dport=1234) / self.payload(
+ current_mtu - 20 - 8)
+
+ p4 = p_ether / p_ip4 / p_payload
+ p4_reply = p_ip4 / p_payload
+ p4_reply.ttl -= 1
+ rx = self.send_and_expect(self.pg0, p4*11, self.pg1)
+ for p in rx:
+ self.validate(p[1], p4_reply)
+
+ # MTU
+ self.vapi.sw_interface_set_mtu(self.pg1.sw_if_index, 576 + mtu_offset)
+ self.assertEqual(576, self.get_mtu(self.pg1.sw_if_index) - mtu_offset)
+
+ # Should fail. Too large MTU
+ p_icmp4 = ICMP(type='dest-unreach', code='fragmentation-needed',
+ nexthopmtu=576, chksum=0x2dbb)
+ icmp4_reply = (IP(src=self.pg0.local_ip4,
+ dst=self.pg0.remote_ip4,
+ ttl=254, len=576, id=0) /
+ p_icmp4 / p_ip4 / p_payload)
+ icmp4_reply[1].ttl -= 1
+ n = icmp4_reply.__class__(str(icmp4_reply))
+ s = str(icmp4_reply)
+ icmp4_reply = s[0:576]
+ rx = self.send_and_expect(self.pg0, p4*11, self.pg0)
+ for p in rx:
+ # p.show2()
+ # n.show2()
+ self.validate_bytes(str(p[1]), icmp4_reply)
+
+ '''
+ # Now with DF off. Expect fragments.
+ # First go with 1500 byte packets.
+ p_payload = UDP(sport=1234, dport=1234) / self.payload(
+ 1500 - 20 - 8)
+ p4 = p_ether / p_ip4 / p_payload
+ p4.flags = 0
+ p4_reply = p_ip4 / p_payload
+ p4_reply.ttl = 62 # check this
+ p4_reply.flags = 0
+ p4_reply.id = 256
+ self.pg_enable_capture()
+ self.pg0.add_stream(p4*1)
+ self.pg_start()
+ rx = self.pg1.get_capture(3)
+ print('RX', len(rx))
+ reass_pkt = reassemble(rx)
+ self.validate(reass_pkt, p4_reply)
+ '''
+ # Now what happens with a 9K frame
+ '''
+ p_payload = UDP(sport=1234, dport=1234) / self.payload(
+ current_mtu - 20 - 8)
+ p4 = p_ether / p_ip4 / p_payload
+ p4.flags = 0
+ p4_reply = p_ip4 / p_payload
+ p4_reply.ttl = 62 # check this
+ p4_reply.flags = 0
+ p4_reply.id = 512
+
+ self.pg_enable_capture()
+ self.pg0.add_stream(p4*1)
+ self.pg_start()
+ rx = self.pg1.get_capture(16)
+ reass_pkt = reassemble(rx)
+ reass_pkt.show2()
+ p4_reply.show2()
+ self.validate(reass_pkt, p4_reply)
+ '''
+ # Reset MTU
+ self.vapi.sw_interface_set_mtu(self.pg1.sw_if_index, current_mtu)
+
+ @unittest.skip("Enable when IPv6 fragmentation is added")
+ def test_ip6_mtu(self):
+ """ IP6 MTU test """
+
+ p_ether = Ether(src=self.pg0.remote_mac, dst=self.pg0.local_mac)
+ p_ip6 = IPv6(src=self.pg0.remote_ip6, dst=self.pg1.remote_ip6)
+
+ current_mtu = self.get_mtu(self.pg1.sw_if_index)
+ p_payload = UDP(sport=1234, dport=1234) / self.payload(
+ current_mtu - 40 - 8)
+
+ p6 = p_ether / p_ip6 / p_payload
+ p6_reply = p_ip6 / p_payload
+ p6_reply.hlim -= 1
+ rx = self.send_and_expect(self.pg0, p6*9, self.pg1)
+ for p in rx:
+ self.validate(p[1], p6_reply)
+
+ # MTU (only checked on encap)
+ self.vapi.sw_interface_set_mtu(self.pg1.sw_if_index, 1280)
+ self.assertEqual(1280, self.get_mtu(self.pg1.sw_if_index))
+
+ # Should fail. Too large MTU
+ p_icmp6 = ICMPv6PacketTooBig(mtu=1280, cksum=0x4c7a)
+ icmp6_reply = (IPv6(src=self.pg0.local_ip6,
+ dst=self.pg0.remote_ip6,
+ hlim=254, plen=1240) /
+ p_icmp6 / p_ip6 / p_payload)
+ icmp6_reply[2].hlim -= 1
+ n = icmp6_reply.__class__(str(icmp6_reply))
+ s = str(icmp6_reply)
+ icmp6_reply = s[0:1280]
+
+ rx = self.send_and_expect(self.pg0, p6*9, self.pg0)
+ for p in rx:
+ self.validate_bytes(str(p[1]), icmp6_reply)
+
+ # Reset MTU
+ self.vapi.sw_interface_set_mtu(self.pg1.sw_if_index, current_mtu)
+
+
+if __name__ == '__main__':
+ unittest.main(testRunner=VppTestRunner)