6 from vpp_object import VppObject
7 from vpp_lo_interface import VppLoInterface
8 from vpp_papi import MACAddress
9 from vpp_sub_interface import L2_VTR_OP
32 def find_bridge_domain(test, bd_id):
33 bds = test.vapi.bridge_domain_dump(bd_id)
37 def find_bridge_domain_port(test, bd_id, sw_if_index):
38 bds = test.vapi.bridge_domain_dump(bd_id)
40 for p in bd.sw_if_details:
41 if p.sw_if_index == sw_if_index:
46 def find_bridge_domain_arp_entry(test, bd_id, mac, ip):
47 arps = test.vapi.bd_ip_mac_dump(bd_id)
49 # do IP addr comparison too once .api is fixed...
50 if mac == str(arp.entry.mac) and \
51 ip == str(arp.entry.ip):
56 def find_l2_fib_entry(test, bd_id, mac, sw_if_index):
57 vmac = MACAddress(mac)
58 lfs = test.vapi.l2_fib_table_dump(bd_id)
60 if vmac.packed == lf.mac and sw_if_index == lf.sw_if_index:
65 class VppBridgeDomain(VppObject):
67 def __init__(self, test, bd_id,
68 flood=1, uu_flood=1, forward=1,
69 learn=1, arp_term=1, arp_ufwd=0):
73 self.uu_flood = uu_flood
74 self.forward = forward
76 self.arp_term = arp_term
77 self.arp_ufwd = arp_ufwd
79 def add_vpp_config(self):
80 self._test.vapi.bridge_domain_add_del(bd_id=self.bd_id,
82 uu_flood=self.uu_flood,
85 arp_term=self.arp_term,
86 arp_ufwd=self.arp_ufwd,
88 self._test.registry.register(self, self._test.logger)
90 def remove_vpp_config(self):
91 self._test.vapi.bridge_domain_add_del(bd_id=self.bd_id, is_add=0)
93 def query_vpp_config(self):
94 return find_bridge_domain(self._test, self.bd_id)
97 return "bridge-domain-%d" % (self.bd_id)
100 class VppBridgeDomainPort(VppObject):
102 def __init__(self, test, bd, itf,
103 port_type=L2_PORT_TYPE.NORMAL):
107 self.port_type = port_type
109 def add_vpp_config(self):
110 self._test.vapi.sw_interface_set_l2_bridge(
111 rx_sw_if_index=self.itf.sw_if_index, bd_id=self.bd.bd_id,
112 port_type=self.port_type, enable=1)
113 self._test.registry.register(self, self._test.logger)
115 def remove_vpp_config(self):
116 self._test.vapi.sw_interface_set_l2_bridge(
117 rx_sw_if_index=self.itf.sw_if_index, bd_id=self.bd.bd_id,
118 port_type=self.port_type, enable=0)
120 def query_vpp_config(self):
121 return find_bridge_domain_port(self._test,
123 self.itf.sw_if_index)
126 return "BD-Port-%s-%s" % (self.bd, self.itf)
129 class VppBridgeDomainArpEntry(VppObject):
131 def __init__(self, test, bd, mac, ip):
137 def add_vpp_config(self):
138 self._test.vapi.bd_ip_mac_add_del(is_add=1,
140 'bd_id': self.bd.bd_id,
143 self._test.registry.register(self, self._test.logger)
145 def remove_vpp_config(self):
146 self._test.vapi.bd_ip_mac_add_del(is_add=0,
148 'bd_id': self.bd.bd_id,
152 def query_vpp_config(self):
153 return find_bridge_domain_arp_entry(self._test,
159 return "BD-Arp-Entry-%s-%s-%s" % (self.bd, self.mac, self.ip)
162 class VppL2FibEntry(VppObject):
164 def __init__(self, test, bd, mac, itf,
165 static_mac=0, filter_mac=0, bvi_mac=-1):
168 self.mac = MACAddress(mac)
170 self.static_mac = static_mac
171 self.filter_mac = filter_mac
173 self.bvi_mac = isinstance(self.itf, VppLoInterface)
175 self.bvi_mac = bvi_mac
177 def add_vpp_config(self):
178 self._test.vapi.l2fib_add_del(
181 self.itf.sw_if_index,
183 static_mac=self.static_mac,
184 filter_mac=self.filter_mac,
185 bvi_mac=self.bvi_mac)
186 self._test.registry.register(self, self._test.logger)
188 def remove_vpp_config(self):
189 self._test.vapi.l2fib_add_del(
192 self.itf.sw_if_index,
195 def query_vpp_config(self):
196 return find_l2_fib_entry(self._test,
199 self.itf.sw_if_index)
202 return "L2-Fib-Entry-%s-%s-%s" % (self.bd, self.mac, self.itf)
205 class VppL2Vtr(VppObject):
207 def __init__(self, test, itf, op):
212 def add_vpp_config(self):
213 self.itf.set_vtr(self.op)
214 self._test.registry.register(self, self._test.logger)
216 def remove_vpp_config(self):
217 self.itf.set_vtr(L2_VTR_OP.L2_DISABLED)
219 def query_vpp_config(self):
220 ds = self._test.vapi.sw_interface_dump()
221 d = self.itf.get_interface_config_from_dump(ds)
224 return (d.vtr_op == self.op)
228 return "L2-vtr-%s-%d" % (str(self.itf), self.op)