2 from vpp_object import VppObject
16 def find_igmp_state(states, itf, gaddr, saddr):
18 if s.sw_if_index == itf.sw_if_index and \
19 s.gaddr.address == socket.inet_pton(socket.AF_INET, gaddr) and \
20 s.saddr.address == socket.inet_pton(socket.AF_INET, saddr):
25 def wait_for_igmp_event(test, timeout, itf, gaddr, saddr, ff):
26 ev = test.vapi.wait_for_event(timeout, "igmp_event")
27 if ev.sw_if_index == itf.sw_if_index and \
28 ev.gaddr.address == socket.inet_pton(socket.AF_INET, gaddr) and \
29 ev.saddr.address == socket.inet_pton(socket.AF_INET, saddr) and \
36 def __init__(self, gaddr, saddrs):
38 self.gaddr_p = socket.inet_pton(socket.AF_INET, gaddr)
41 self.saddrs_encoded = []
43 ss = socket.inet_pton(socket.AF_INET, s)
44 self.saddrs_p.append(ss)
45 self.saddrs_encoded.append({'address': ss})
49 def __init__(self, sg, type):
54 class VppHostState(VppObject):
55 def __init__(self, test, filter, sw_if_index, sg):
57 self.sw_if_index = sw_if_index
61 def add_vpp_config(self):
62 self._test.vapi.igmp_listen(
63 self.filter, self.sw_if_index,
64 self.sg.saddrs_encoded, self.sg.gaddr_p)
66 def remove_vpp_config(self):
67 self._test.vapi.igmp_listen(
74 return self.object_id()
77 return "%s:%d" % (self.sg, self.sw_if_index)
79 def query_vpp_config(self):
80 return self._test.vapi.igmp_dump()