diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_igmp.py | 323 | ||||
| -rw-r--r-- | test/vpp_igmp.py | 39 | ||||
| -rw-r--r-- | test/vpp_papi_provider.py | 36 |
3 files changed, 398 insertions, 0 deletions
diff --git a/test/test_igmp.py b/test/test_igmp.py new file mode 100644 index 00000000000..62b1c9637d6 --- /dev/null +++ b/test/test_igmp.py @@ -0,0 +1,323 @@ +#!/usr/bin/env python + +import unittest +import socket + +from framework import VppTestCase, VppTestRunner, running_extended_tests +from vpp_igmp import * + +from scapy.packet import Raw +from scapy.layers.l2 import Ether +from scapy.layers.inet import IP +from scapy.contrib.igmpv3 import * +from scapy.contrib.igmp import * + + +def checkIGMPv3(): + try: + tmp = IGMPv3() + tmp = IGMPv3mr() + tmp = IGMPv3gr() + tmp = IGMPv3mq() + except NameError: + return False + return True + + +class TestIgmp(VppTestCase): + """ IGMP Test Case """ + + def setUp(self): + super(TestIgmp, self).setUp() + + self.create_pg_interfaces(range(2)) + self.sg_list = [] + self.config_list = [] + + self.ip_addr = [] + for pg in self.pg_interfaces: + pg.admin_up() + pg.config_ip4() + pg.resolve_arp() + + def tearDown(self): + for pg in self.pg_interfaces: + self.vapi.igmp_clear_interface(pg.sw_if_index) + pg.unconfig_ip4() + pg.admin_down() + super(TestIgmp, self).tearDown() + + def send(self, ti, pkts): + ti.add_stream(pkts) + self.pg_enable_capture(self.pg_interfaces) + self.pg_start() + + @unittest.skipUnless(checkIGMPv3(), "missing scapy igmpv3 implementation") + def test_igmp_parse_report(self): + """ IGMP parse Membership Report """ + + # + # VPP acts as a router + # + self.vapi.want_igmp_events(1) + + # hos sends join IGMP 'join' + p_join = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22', tos=0xc0) / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=3, maddr="224.1.1.1", srcaddrs=["10.1.1.1"])) + + self.send(self.pg0, p_join) + + # search for the corresponding state created in VPP + dump = self.vapi.igmp_dump() + self.assertEqual(len(dump), 1) + self.assertEqual(dump[0].sw_if_index, self.pg0.sw_if_index) + self.assertEqual(dump[0].gaddr, + socket.inet_pton(socket.AF_INET, + "224.1.1.1")) + self.assertEqual(dump[0].saddr, + socket.inet_pton(socket.AF_INET, + "10.1.1.1")) + + # VPP sends a notification that a new group has been joined + ev = self.vapi.wait_for_event(2, "igmp_event") + + self.assertEqual(ev.saddr, + socket.inet_pton(socket.AF_INET, + "10.1.1.1")) + self.assertEqual(ev.gaddr, + socket.inet_pton(socket.AF_INET, + "224.1.1.1")) + self.assertEqual(ev.is_join, 1) + + # host sends IGMP leave + p_leave = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22', tos=0xc0) / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=4, maddr="224.1.1.1", srcaddrs=["10.1.1.1"])) + + self.send(self.pg0, p_leave) + + # VPP sends a notification that a new group has been left + ev = self.vapi.wait_for_event(2, "igmp_event") + + self.assertEqual(ev.saddr, + socket.inet_pton(socket.AF_INET, + "10.1.1.1")) + self.assertEqual(ev.gaddr, + socket.inet_pton(socket.AF_INET, + "224.1.1.1")) + self.assertEqual(ev.is_join, 0) + + # state gone + dump = self.vapi.igmp_dump() + self.assertFalse(dump) + + # resend the join + self.send(self.pg0, p_join) + dump = self.vapi.igmp_dump() + self.assertEqual(len(dump), 1) + self.assertEqual(dump[0].sw_if_index, self.pg0.sw_if_index) + self.assertEqual(dump[0].gaddr, + socket.inet_pton(socket.AF_INET, + "224.1.1.1")) + self.assertEqual(dump[0].saddr, + socket.inet_pton(socket.AF_INET, + "10.1.1.1")) + + # IGMP block + p_block = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22', tos=0xc0) / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=6, maddr="224.1.1.1", srcaddrs=["10.1.1.1"])) + + self.send(self.pg0, p_block) + + dump = self.vapi.igmp_dump() + self.assertFalse(dump) + + def verify_general_query(self, p): + ip = p[IP] + self.assertEqual(ip.dst, "224.0.0.1") + self.assertEqual(ip.proto, 2) + igmp = p[IGMPv3] + self.assertEqual(igmp.type, 0x11) + self.assertEqual(igmp.gaddr, "0.0.0.0") + + @unittest.skipUnless(checkIGMPv3(), "missing scapy igmpv3 implementation") + def test_igmp_send_query(self): + """ IGMP send General Query """ + + # + # VPP acts as a router. + # Send a membership report so VPP builds state + # + p_mr = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22') / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=3, maddr="224.1.1.1", srcaddrs=["10.1.1.1"])) + + self.send(self.pg0, p_mr) + self.logger.info(self.vapi.cli("sh igmp config")) + + # + # wait for VPP to send out the General Query + # + capture = self.pg0.get_capture(1, timeout=61) + + self.verify_general_query(capture[0]) + + # + # the state will expire in 10 more seconds + # + self.sleep(10) + self.assertFalse(self.vapi.igmp_dump()) + + @unittest.skipUnless(checkIGMPv3(), "missing scapy igmpv3 implementation") + def test_igmp_src_exp(self): + """ IGMP per source timer """ + + # + # VPP Acts as a router + # + + # Host join for (10.1.1.1,224.1.1.1) + p_mr1 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22') / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=3, maddr="224.1.1.1", srcaddrs=["10.1.1.1"])) + + self.send(self.pg0, p_mr1) + + # VPP (router) sends General Query + capture = self.pg0.get_capture(1, timeout=61) + + self.verify_general_query(capture[0]) + + # host join for same G and another S: (10.1.1.2,224.1.1.1) + # therefore leaving (10.1.1.1,224.1.1.1) + p_mr2 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22') / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=2, maddr="224.1.1.1", srcaddrs=["10.1.1.2"])) + + self.send(self.pg0, p_mr2) + + # wait for VPP to send general query + capture = self.pg0.get_capture(1, timeout=61) + self.verify_general_query(capture[0]) + + # host leaves (10.1.1.2,224.1.1.1) + p_l = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22') / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=2, maddr="224.1.1.1", srcaddrs=["10.1.1.2"])) + + self.send(self.pg0, p_l) + + # FIXME BUG + p_l = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.22') / + IGMPv3() / + IGMPv3mr(numgrp=1) / + IGMPv3gr(rtype=2, maddr="224.1.1.1", srcaddrs=["10.1.1.1"])) + self.send(self.pg0, p_l) + + # + # host has left all groups, no state left. + # + self.sleep(10) + self.logger.error(self.vapi.cli("sh igmp config")) + self.assertFalse(self.vapi.igmp_dump()) + + @unittest.skipUnless(checkIGMPv3(), "missing scapy igmpv3 implementation") + def test_igmp_query_resp(self): + """ IGMP General Query response """ + + # + # VPP acting as a host. + # Add a listener in VPP for (10.1.1.1,244.1.1.1) + # + self.config_list.append( + VppIgmpConfig( + self, self.pg0.sw_if_index, IgmpSG( + socket.inet_pton( + socket.AF_INET, "10.1.1.1"), socket.inet_pton( + socket.AF_INET, "224.1.1.1")))) + self.config_list[0].add_vpp_config() + + # verify state exists + self.assertTrue(self.vapi.igmp_dump(self.pg0.sw_if_index)) + + # + # Send a general query (from a router) + # + p = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / + IP(src=self.pg0.remote_ip4, dst='224.0.0.1', tos=0xc0) / + IGMPv3(type=0x11, mrcode=100) / + IGMPv3mq(gaddr="0.0.0.0")) + + self.send(self.pg0, p) + + # + # expect VPP to respond with a membership report for the + # (10.1.1.1, 224.1.1.1) state + # + capture = self.pg0.get_capture(1, timeout=10) + + self.assertEqual(capture[0][IGMPv3].type, 0x22) + self.assertEqual(capture[0][IGMPv3mr].numgrp, 1) + self.assertEqual(capture[0][IGMPv3gr].rtype, 1) + self.assertEqual(capture[0][IGMPv3gr].numsrc, 1) + self.assertEqual(capture[0][IGMPv3gr].maddr, "224.1.1.1") + self.assertEqual(len(capture[0][IGMPv3gr].srcaddrs), 1) + self.assertEqual(capture[0][IGMPv3gr].srcaddrs[0], "10.1.1.1") + + @unittest.skipUnless(checkIGMPv3(), "missing scapy igmpv3 implementation") + def test_igmp_listen(self): + """ IGMP listen (S,G)s """ + + # + # VPP acts as a host + # Add IGMP group state to multiple interfaces and validate its + # presence + # + for pg in self.pg_interfaces: + self.sg_list.append(IgmpSG(socket.inet_pton( + socket.AF_INET, "10.1.1.%d" % pg._sw_if_index), + socket.inet_pton(socket.AF_INET, "224.1.1.1"))) + + for pg in self.pg_interfaces: + self.config_list.append( + VppIgmpConfig( + self, + pg._sw_if_index, + self.sg_list)) + self.config_list[-1].add_vpp_config() + + for config in self.config_list: + dump = self.vapi.igmp_dump(config.sw_if_index) + self.assertTrue(dump) + self.assertEqual(len(dump), len(config.sg_list)) + for idx, e in enumerate(dump): + self.assertEqual(e.sw_if_index, config.sw_if_index) + self.assertEqual(e.saddr, config.sg_list[idx].saddr) + self.assertEqual(e.gaddr, config.sg_list[idx].gaddr) + + for config in self.config_list: + config.remove_vpp_config() + + dump = self.vapi.igmp_dump() + self.assertFalse(dump) + + +if __name__ == '__main__': + unittest.main(testRunner=VppTestRunner) diff --git a/test/vpp_igmp.py b/test/vpp_igmp.py new file mode 100644 index 00000000000..d1a308878c5 --- /dev/null +++ b/test/vpp_igmp.py @@ -0,0 +1,39 @@ + +from vpp_object import VppObject + + +class IgmpSG(): + def __init__(self, saddr, gaddr): + self.saddr = saddr + self.gaddr = gaddr + + +class VppIgmpConfig(VppObject): + def __init__(self, test, sw_if_index, sg=None): + self._test = test + self.sw_if_index = sw_if_index + if isinstance(sg, list): + self.sg_list = sg + else: + self.sg_list = [] + self.sg_list.append(sg) + + def add_sg(self, sg): + self.sg.append(sg) + + def add_vpp_config(self): + for e in self.sg_list: + self._test.vapi.igmp_listen( + 1, self.sw_if_index, e.saddr, e.gaddr) + + def remove_vpp_config(self): + self._test.vapi.igmp_clear_interface(self.sw_if_index) + + def __str__(self): + return self.object_id() + + def object_id(self): + return "%s:%d" % (self.sg_list, self.sw_if_index) + + def query_vpp_config(self): + return self._test.vapi.igmp_dump() diff --git a/test/vpp_papi_provider.py b/test/vpp_papi_provider.py index da59bc86ca0..dd553cb473e 100644 --- a/test/vpp_papi_provider.py +++ b/test/vpp_papi_provider.py @@ -3291,3 +3291,39 @@ class VppPapiProvider(object): {'sw_if_index': sw_if_index, 'input_source': input_source, 'enable': enable}) + + def igmp_listen(self, enable, sw_if_index, saddr, gaddr): + """ Listen for new (S,G) on specified interface + + :param enable: add/del + :param sw_if_index: interface sw index + :param saddr: source ip4 addr + :param gaddr: group ip4 addr + """ + return self.api(self.papi.igmp_listen, + {'enable': enable, + 'sw_if_index': sw_if_index, + 'saddr': saddr, + 'gaddr': gaddr}) + + def igmp_dump(self, sw_if_index=None): + """ Dump all (S,G) interface configurations """ + if sw_if_index is None: + dump_all = 1 + sw_if_index = 0 + else: + dump_all = 0 + return self.api(self.papi.igmp_dump, {'sw_if_index': sw_if_index, + 'dump_all': dump_all}) + + def igmp_clear_interface(self, sw_if_index): + """ Remove all (S,G)s from specified interface + doesn't send IGMP report! + """ + return self.api( + self.papi.igmp_clear_interface, { + 'sw_if_index': sw_if_index}) + + def want_igmp_events(self, enable=1): + return self.api(self.papi.want_igmp_events, {'enable': enable, + 'pid': os.getpid()}) |