1 from vpp_object import VppObject
15 def find_igmp_state(states, itf, gaddr, saddr):
18 s.sw_if_index == itf.sw_if_index
19 and str(s.gaddr) == gaddr
20 and str(s.saddr) == saddr
26 def wait_for_igmp_event(test, timeout, itf, gaddr, saddr, ff):
27 ev = test.vapi.wait_for_event(timeout, "igmp_event")
29 ev.sw_if_index == itf.sw_if_index
30 and str(ev.gaddr) == gaddr
31 and str(ev.saddr) == saddr
39 def __init__(self, gaddr, saddrs):
41 self.gaddr_p = socket.inet_pton(socket.AF_INET, gaddr)
44 self.saddrs_encoded = []
46 ss = socket.inet_pton(socket.AF_INET, s)
47 self.saddrs_p.append(ss)
48 self.saddrs_encoded.append(ss)
52 def __init__(self, sg, type):
57 class VppHostState(VppObject):
58 def __init__(self, test, filter, sw_if_index, sg):
60 self.sw_if_index = sw_if_index
64 def add_vpp_config(self):
65 self._test.vapi.igmp_listen(
66 self.filter, self.sw_if_index, self.sg.saddrs_encoded, self.sg.gaddr_p
69 def remove_vpp_config(self):
70 self._test.vapi.igmp_listen(self.filter, self.sw_if_index, [], self.sg.gaddr_p)
73 return "%s:%d" % (self.sg, self.sw_if_index)
75 def query_vpp_config(self):
76 return self._test.vapi.igmp_dump()