0bcad32b38
The API calls that use any of vl_api_address_t, vl_api_ip4_address, vl_api_ip6_address_t, vl_api_prefix_t, vl_api_ip4_prefix_t, vl_api_ip6_prefix_t now accepts either the old style dictionary, a text string (2001:db8::/32) or an ipaddress ojbect. Unless it is called with '_no_type_conversion':True, it will also return an appropriate ipaddress object. Change-Id: I84e4a1577bd57f6b5ae725f316a523988b6a955b Signed-off-by: Ole Troan <ot@cisco.com>
79 lines
1.9 KiB
Python
79 lines
1.9 KiB
Python
|
|
from vpp_object import VppObject
|
|
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 \
|
|
str(s.gaddr) == gaddr and str(s.saddr) == saddr:
|
|
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 \
|
|
str(ev.gaddr) == gaddr and str(ev.saddr) == saddr and \
|
|
ev.filter == ff:
|
|
return True
|
|
return False
|
|
|
|
|
|
class IgmpSG():
|
|
def __init__(self, gaddr, saddrs):
|
|
self.gaddr = gaddr
|
|
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)
|
|
self.saddrs_encoded.append(ss)
|
|
|
|
|
|
class IgmpRecord():
|
|
def __init__(self, sg, type):
|
|
self.sg = sg
|
|
self.type = type
|
|
|
|
|
|
class VppHostState(VppObject):
|
|
def __init__(self, test, filter, sw_if_index, sg):
|
|
self._test = test
|
|
self.sw_if_index = sw_if_index
|
|
self.filter = filter
|
|
self.sg = sg
|
|
|
|
def add_vpp_config(self):
|
|
self._test.vapi.igmp_listen(
|
|
self.filter, self.sw_if_index,
|
|
self.sg.saddrs_encoded, self.sg.gaddr_p)
|
|
|
|
def remove_vpp_config(self):
|
|
self._test.vapi.igmp_listen(
|
|
self.filter,
|
|
self.sw_if_index,
|
|
[],
|
|
self.sg.gaddr_p)
|
|
|
|
def __str__(self):
|
|
return self.object_id()
|
|
|
|
def object_id(self):
|
|
return "%s:%d" % (self.sg, self.sw_if_index)
|
|
|
|
def query_vpp_config(self):
|
|
return self._test.vapi.igmp_dump()
|