| """ |
| L2/BD Types |
| |
| """ |
| |
| from vpp_object import VppObject |
| from vpp_lo_interface import VppLoInterface |
| from vpp_papi import MACAddress |
| from vpp_sub_interface import L2_VTR_OP |
| |
| try: |
| text_type = unicode |
| except NameError: |
| text_type = str |
| |
| |
| class L2_PORT_TYPE: |
| NORMAL = 0 |
| BVI = 1 |
| UU_FWD = 2 |
| |
| |
| class BRIDGE_FLAGS: |
| NONE = 0 |
| LEARN = 1 |
| FWD = 2 |
| FLOOD = 4 |
| UU_FLOOD = 8 |
| ARP_TERM = 16 |
| ARP_UFWD = 32 |
| |
| |
| def find_bridge_domain(test, bd_id): |
| bds = test.vapi.bridge_domain_dump(bd_id) |
| return len(bds) == 1 |
| |
| |
| def find_bridge_domain_port(test, bd_id, sw_if_index): |
| bds = test.vapi.bridge_domain_dump(bd_id) |
| for bd in bds: |
| for p in bd.sw_if_details: |
| if p.sw_if_index == sw_if_index: |
| return True |
| return False |
| |
| |
| def find_bridge_domain_arp_entry(test, bd_id, mac, ip): |
| arps = test.vapi.bd_ip_mac_dump(bd_id) |
| for arp in arps: |
| # do IP addr comparison too once .api is fixed... |
| if mac == str(arp.entry.mac) and ip == str(arp.entry.ip): |
| return True |
| return False |
| |
| |
| def find_l2_fib_entry(test, bd_id, mac, sw_if_index): |
| vmac = MACAddress(mac) |
| lfs = test.vapi.l2_fib_table_dump(bd_id) |
| for lf in lfs: |
| if vmac.packed == lf.mac and sw_if_index == lf.sw_if_index: |
| return True |
| return False |
| |
| |
| class VppBridgeDomain(VppObject): |
| def __init__( |
| self, |
| test, |
| bd_id, |
| flood=1, |
| uu_flood=1, |
| forward=1, |
| learn=1, |
| arp_term=1, |
| arp_ufwd=0, |
| ): |
| self._test = test |
| self.bd_id = bd_id |
| self.flood = flood |
| self.uu_flood = uu_flood |
| self.forward = forward |
| self.learn = learn |
| self.arp_term = arp_term |
| self.arp_ufwd = arp_ufwd |
| |
| def add_vpp_config(self): |
| self._test.vapi.bridge_domain_add_del_v2( |
| bd_id=self.bd_id, |
| flood=self.flood, |
| uu_flood=self.uu_flood, |
| forward=self.forward, |
| learn=self.learn, |
| arp_term=self.arp_term, |
| arp_ufwd=self.arp_ufwd, |
| is_add=1, |
| ) |
| self._test.registry.register(self, self._test.logger) |
| return self |
| |
| def remove_vpp_config(self): |
| self._test.vapi.bridge_domain_add_del_v2(bd_id=self.bd_id, is_add=0) |
| |
| def query_vpp_config(self): |
| return find_bridge_domain(self._test, self.bd_id) |
| |
| def object_id(self): |
| return "bridge-domain-%d" % (self.bd_id) |
| |
| |
| class VppBridgeDomainPort(VppObject): |
| def __init__(self, test, bd, itf, port_type=L2_PORT_TYPE.NORMAL): |
| self._test = test |
| self.bd = bd |
| self.itf = itf |
| self.port_type = port_type |
| |
| def add_vpp_config(self): |
| self._test.vapi.sw_interface_set_l2_bridge( |
| rx_sw_if_index=self.itf.sw_if_index, |
| bd_id=self.bd.bd_id, |
| port_type=self.port_type, |
| enable=1, |
| ) |
| self._test.registry.register(self, self._test.logger) |
| return self |
| |
| def remove_vpp_config(self): |
| self._test.vapi.sw_interface_set_l2_bridge( |
| rx_sw_if_index=self.itf.sw_if_index, |
| bd_id=self.bd.bd_id, |
| port_type=self.port_type, |
| enable=0, |
| ) |
| |
| def query_vpp_config(self): |
| return find_bridge_domain_port(self._test, self.bd.bd_id, self.itf.sw_if_index) |
| |
| def object_id(self): |
| return "BD-Port-%s-%s" % (self.bd, self.itf) |
| |
| |
| class VppBridgeDomainArpEntry(VppObject): |
| def __init__(self, test, bd, mac, ip): |
| self._test = test |
| self.bd = bd |
| self.mac = mac |
| self.ip = ip |
| |
| def add_vpp_config(self): |
| self._test.vapi.bd_ip_mac_add_del( |
| is_add=1, entry={"bd_id": self.bd.bd_id, "ip": self.ip, "mac": self.mac} |
| ) |
| self._test.registry.register(self, self._test.logger) |
| return self |
| |
| def remove_vpp_config(self): |
| self._test.vapi.bd_ip_mac_add_del( |
| is_add=0, entry={"bd_id": self.bd.bd_id, "ip": self.ip, "mac": self.mac} |
| ) |
| |
| def query_vpp_config(self): |
| return find_bridge_domain_arp_entry( |
| self._test, self.bd.bd_id, self.mac, self.ip |
| ) |
| |
| def object_id(self): |
| return "BD-Arp-Entry-%s-%s-%s" % (self.bd, self.mac, self.ip) |
| |
| |
| class VppL2FibEntry(VppObject): |
| def __init__(self, test, bd, mac, itf, static_mac=0, filter_mac=0, bvi_mac=-1): |
| self._test = test |
| self.bd = bd |
| self.mac = MACAddress(mac) |
| self.itf = itf |
| self.static_mac = static_mac |
| self.filter_mac = filter_mac |
| if bvi_mac == -1: |
| self.bvi_mac = isinstance(self.itf, VppLoInterface) |
| else: |
| self.bvi_mac = bvi_mac |
| |
| def add_vpp_config(self): |
| self._test.vapi.l2fib_add_del( |
| self.mac.packed, |
| self.bd.bd_id, |
| self.itf.sw_if_index, |
| is_add=1, |
| static_mac=self.static_mac, |
| filter_mac=self.filter_mac, |
| bvi_mac=self.bvi_mac, |
| ) |
| self._test.registry.register(self, self._test.logger) |
| return self |
| |
| def remove_vpp_config(self): |
| self._test.vapi.l2fib_add_del( |
| self.mac.packed, self.bd.bd_id, self.itf.sw_if_index, is_add=0 |
| ) |
| |
| def query_vpp_config(self): |
| return find_l2_fib_entry( |
| self._test, self.bd.bd_id, self.mac.packed, self.itf.sw_if_index |
| ) |
| |
| def object_id(self): |
| return "L2-Fib-Entry-%s-%s-%s" % (self.bd, self.mac, self.itf) |
| |
| |
| class VppL2Vtr(VppObject): |
| def __init__(self, test, itf, op): |
| self._test = test |
| self.itf = itf |
| self.op = op |
| |
| def add_vpp_config(self): |
| self.itf.set_vtr(self.op) |
| self._test.registry.register(self, self._test.logger) |
| return self |
| |
| def remove_vpp_config(self): |
| self.itf.set_vtr(L2_VTR_OP.L2_DISABLED) |
| |
| def query_vpp_config(self): |
| ds = self._test.vapi.sw_interface_dump() |
| d = self.itf.get_interface_config_from_dump(ds) |
| |
| if d is not None: |
| return d.vtr_op == self.op |
| return False |
| |
| def object_id(self): |
| return "L2-vtr-%s-%d" % (str(self.itf), self.op) |