1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
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 \
s.gaddr == socket.inet_pton(socket.AF_INET, gaddr) and \
s.saddr == socket.inet_pton(socket.AF_INET, 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 \
ev.gaddr == socket.inet_pton(socket.AF_INET, gaddr) and \
ev.saddr == socket.inet_pton(socket.AF_INET, 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()
|
