6 from vpp_object import VppObject
7 from vpp_ip import VppIpAddress
8 from vpp_lo_interface import VppLoInterface
9 from vpp_papi import MACAddress
27 def find_bridge_domain(test, bd_id):
28 bds = test.vapi.bridge_domain_dump(bd_id)
32 def find_bridge_domain_port(test, bd_id, sw_if_index):
33 bds = test.vapi.bridge_domain_dump(bd_id)
35 for p in bd.sw_if_details:
36 if p.sw_if_index == sw_if_index:
41 def find_bridge_domain_arp_entry(test, bd_id, mac, ip):
42 vmac = MACAddress(mac)
43 vip = VppIpAddress(ip)
50 arps = test.vapi.bd_ip_mac_dump(bd_id)
52 # do IP addr comparison too once .api is fixed...
53 if vmac.packed == arp.mac_address and \
54 vip.bytes == arp.ip_address[:n]:
59 def find_l2_fib_entry(test, bd_id, mac, sw_if_index):
60 vmac = MACAddress(mac)
61 lfs = test.vapi.l2_fib_table_dump(bd_id)
63 if vmac.packed == lf.mac and sw_if_index == lf.sw_if_index:
68 class VppBridgeDomain(VppObject):
70 def __init__(self, test, bd_id,
71 flood=1, uu_flood=1, forward=1,
76 self.uu_flood = uu_flood
77 self.forward = forward
79 self.arp_term = arp_term
81 def add_vpp_config(self):
82 self._test.vapi.bridge_domain_add_del(bd_id=self.bd_id,
84 uu_flood=self.uu_flood,
87 arp_term=self.arp_term, is_add=1)
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 self.object_id()
100 return "bridge-domain-%d" % (self.bd_id)
103 class VppBridgeDomainPort(VppObject):
105 def __init__(self, test, bd, itf,
106 port_type=L2_PORT_TYPE.NORMAL):
110 self.port_type = port_type
112 def add_vpp_config(self):
113 self._test.vapi.sw_interface_set_l2_bridge(
114 rx_sw_if_index=self.itf.sw_if_index, bd_id=self.bd.bd_id,
115 port_type=self.port_type, enable=1)
116 self._test.registry.register(self, self._test.logger)
118 def remove_vpp_config(self):
119 self._test.vapi.sw_interface_set_l2_bridge(
120 rx_sw_if_index=self.itf.sw_if_index, bd_id=self.bd.bd_id,
121 port_type=self.port_type, enable=0)
123 def query_vpp_config(self):
124 return find_bridge_domain_port(self._test,
126 self.itf.sw_if_index)
129 return self.object_id()
132 return "BD-Port-%s-%s" % (self.bd, self.itf)
135 class VppBridgeDomainArpEntry(VppObject):
137 def __init__(self, test, bd, mac, ip):
140 self.mac = MACAddress(mac)
141 self.ip = VppIpAddress(ip)
143 def add_vpp_config(self):
144 self._test.vapi.bd_ip_mac_add_del(bd_id=self.bd.bd_id, is_add=1,
147 self._test.registry.register(self, self._test.logger)
149 def remove_vpp_config(self):
150 self._test.vapi.bd_ip_mac_add_del(bd_id=self.bd.bd_id, is_add=0,
154 def query_vpp_config(self):
155 return find_bridge_domain_arp_entry(self._test,
161 return self.object_id()
164 return "BD-Arp-Entry-%s-%s-%s" % (self.bd, self.mac, self.ip.address)
167 class VppL2FibEntry(VppObject):
169 def __init__(self, test, bd, mac, itf,
170 static_mac=0, filter_mac=0, bvi_mac=-1):
173 self.mac = MACAddress(mac)
175 self.static_mac = static_mac
176 self.filter_mac = filter_mac
178 self.bvi_mac = isinstance(self.itf, VppLoInterface)
180 self.bvi_mac = bvi_mac
182 def add_vpp_config(self):
183 self._test.vapi.l2fib_add_del(
186 self.itf.sw_if_index,
188 static_mac=self.static_mac,
189 filter_mac=self.filter_mac,
190 bvi_mac=self.bvi_mac)
191 self._test.registry.register(self, self._test.logger)
193 def remove_vpp_config(self):
194 self._test.vapi.l2fib_add_del(
197 self.itf.sw_if_index,
200 def query_vpp_config(self):
201 return find_l2_fib_entry(self._test,
204 self.itf.sw_if_index)
207 return self.object_id()
210 return "L2-Fib-Entry-%s-%s-%s" % (self.bd, self.mac, self.itf)
213 class VppL2Vtr(VppObject):
215 def __init__(self, test, itf, op):
220 def add_vpp_config(self):
221 self.itf.set_vtr(self.op)
222 self._test.registry.register(self, self._test.logger)
224 def remove_vpp_config(self):
225 self.itf.set_vtr(L2_VTR_OP.L2_DISABLED)
227 def query_vpp_config(self):
228 ds = self._test.vapi.sw_interface_dump()
229 d = self.itf.get_interface_config_from_dump(ds)
232 return (d.vtr_op == self.op)
236 return self.object_id()
239 return "L2-vtr-%s-%d" % (str(self.itf), self.op)