2017-12-08 16:28:42 +01:00
|
|
|
|
|
|
|
from vpp_object import VppObject
|
2018-06-07 23:48:20 -07:00
|
|
|
import socket
|
|
|
|
|
|
|
|
|
|
|
|
class IGMP_MODE:
|
|
|
|
ROUTER = 0
|
|
|
|
HOST = 1
|
|
|
|
|
|
|
|
|
|
|
|
class IGMP_FILTER:
|
|
|
|
INCLUDE = 1
|
|
|
|
EXCLUDE = 0
|
|
|
|
|
|
|
|
|
|
|
|
def find_igmp_state(states, itf, gaddr, saddr):
|
|
|
|
for s in states:
|
|
|
|
if s.sw_if_index == itf.sw_if_index and \
|
2018-12-11 13:04:01 +01:00
|
|
|
str(s.gaddr) == gaddr and str(s.saddr) == saddr:
|
2018-06-07 23:48:20 -07:00
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
|
|
def wait_for_igmp_event(test, timeout, itf, gaddr, saddr, ff):
|
|
|
|
ev = test.vapi.wait_for_event(timeout, "igmp_event")
|
|
|
|
if ev.sw_if_index == itf.sw_if_index and \
|
2018-12-11 13:04:01 +01:00
|
|
|
str(ev.gaddr) == gaddr and str(ev.saddr) == saddr and \
|
2018-06-07 23:48:20 -07:00
|
|
|
ev.filter == ff:
|
|
|
|
return True
|
|
|
|
return False
|
2017-12-08 16:28:42 +01:00
|
|
|
|
|
|
|
|
|
|
|
class IgmpSG():
|
2018-06-07 23:48:20 -07:00
|
|
|
def __init__(self, gaddr, saddrs):
|
2017-12-08 16:28:42 +01:00
|
|
|
self.gaddr = gaddr
|
2018-06-07 23:48:20 -07:00
|
|
|
self.gaddr_p = socket.inet_pton(socket.AF_INET, gaddr)
|
|
|
|
self.saddrs = saddrs
|
|
|
|
self.saddrs_p = []
|
|
|
|
self.saddrs_encoded = []
|
|
|
|
for s in saddrs:
|
|
|
|
ss = socket.inet_pton(socket.AF_INET, s)
|
|
|
|
self.saddrs_p.append(ss)
|
2018-11-27 10:05:23 +01:00
|
|
|
self.saddrs_encoded.append(ss)
|
2018-06-07 23:48:20 -07:00
|
|
|
|
2017-12-08 16:28:42 +01:00
|
|
|
|
2018-06-07 23:48:20 -07:00
|
|
|
class IgmpRecord():
|
|
|
|
def __init__(self, sg, type):
|
|
|
|
self.sg = sg
|
|
|
|
self.type = type
|
2017-12-08 16:28:42 +01:00
|
|
|
|
2018-06-07 23:48:20 -07:00
|
|
|
|
|
|
|
class VppHostState(VppObject):
|
|
|
|
def __init__(self, test, filter, sw_if_index, sg):
|
2017-12-08 16:28:42 +01:00
|
|
|
self._test = test
|
|
|
|
self.sw_if_index = sw_if_index
|
2018-06-07 23:48:20 -07:00
|
|
|
self.filter = filter
|
|
|
|
self.sg = sg
|
2017-12-08 16:28:42 +01:00
|
|
|
|
|
|
|
def add_vpp_config(self):
|
2018-06-07 23:48:20 -07:00
|
|
|
self._test.vapi.igmp_listen(
|
|
|
|
self.filter, self.sw_if_index,
|
|
|
|
self.sg.saddrs_encoded, self.sg.gaddr_p)
|
2017-12-08 16:28:42 +01:00
|
|
|
|
|
|
|
def remove_vpp_config(self):
|
2018-06-07 23:48:20 -07:00
|
|
|
self._test.vapi.igmp_listen(
|
|
|
|
self.filter,
|
|
|
|
self.sw_if_index,
|
|
|
|
[],
|
|
|
|
self.sg.gaddr_p)
|
2017-12-08 16:28:42 +01:00
|
|
|
|
|
|
|
def object_id(self):
|
2018-06-07 23:48:20 -07:00
|
|
|
return "%s:%d" % (self.sg, self.sw_if_index)
|
2017-12-08 16:28:42 +01:00
|
|
|
|
|
|
|
def query_vpp_config(self):
|
|
|
|
return self._test.vapi.igmp_dump()
|