def __init__(self, *args, **kwargs): super(ExtremeVlan, self).__init__(*args, **kwargs) self.extremevlan = ExtremeVlanMib(self.agent) self.bridge = BridgeMib(self.agent) self.baseport_ifindex = {}
class ExtremeVlan(Plugin): """Collects 802.1q info from EXTREME-VLAN-MIB and BRIDGE-MIB""" def __init__(self, *args, **kwargs): super(ExtremeVlan, self).__init__(*args, **kwargs) self.extremevlan = ExtremeVlanMib(self.agent) self.bridge = BridgeMib(self.agent) self.baseport_ifindex = {} @classmethod def can_handle(cls, netbox): daddy_says_ok = super(ExtremeVlan, cls).can_handle(netbox) if netbox.type: vendor_id = netbox.type.get_enterprise_id() if vendor_id != VENDORID_EXTREMENETWORKS: return False return daddy_says_ok def handle(self): self._logger.debug( "Collecting Extreme Networks proprietary 802.1q VLAN information") return self._get_vlan_data() @inlineCallbacks def _get_vlan_data(self): vlan_ports = yield self.extremevlan.get_vlan_ports() if not vlan_ports: # Doesn't appear to have VLAN data in EXTREME MIBs, get out now. returnValue(None) ifindex_vlan = yield self.extremevlan.get_ifindex_vlan_map() self.baseport_ifindex = yield self.bridge.get_baseport_ifindex_map() vlan_config = self._consolidate_vlan_config(vlan_ports, ifindex_vlan) self._store_access_ports(vlan_config) self._store_trunk_ports(vlan_config) def _consolidate_vlan_config(self, vlan_ports, ifindex_vlan): config = {} for vlan_ifindex, (tagged, untagged) in vlan_ports.items(): vlanid = ifindex_vlan.get(vlan_ifindex, None) if not vlanid: continue tagged_indexes = self._portnums_to_ifindexes(tagged) untagged_indexes = self._portnums_to_ifindexes(untagged) config[vlanid] = (tagged_indexes, untagged_indexes) return config def _portnums_to_ifindexes(self, portnums): return [self.baseport_ifindex[portnum] for portnum in portnums if portnum in self.baseport_ifindex] def _store_access_ports(self, vlan_config): """Store vlan memberships for all ports.""" for vlanid, (_tagged, untagged) in vlan_config.items(): for ifindex in untagged: interface = self.containers.factory(ifindex, shadows.Interface) interface.trunk = False interface.vlan = vlanid def _store_trunk_ports(self, vlan_config): """Store the set of enabled vlans for each trunk port.""" ifindex_vlans = {} for vlanid, (tagged, _untagged) in vlan_config.items(): for ifindex in tagged: if ifindex not in ifindex_vlans: ifindex_vlans[ifindex] = set() ifindex_vlans[ifindex].add(vlanid) for ifindex, vlans in ifindex_vlans.items(): interface = self.containers.factory(ifindex, shadows.Interface) interface.trunk = True interface.vlan = None allowed = self.containers.factory(ifindex, shadows.SwPortAllowedVlan) allowed.interface = interface allowed.hex_string = vlan_list_to_hex(vlans)
def __init__(self, *args, **kwargs): super(Dot1q, self).__init__(*args, **kwargs) self.bridgemib = BridgeMib(self.agent) self.qbridgemib = QBridgeMib(self.agent)
class Dot1q(Plugin): """Collect 802.1q info from BRIDGE and Q-BRIDGE MIBs.""" baseports = {} pvids = {} def __init__(self, *args, **kwargs): super(Dot1q, self).__init__(*args, **kwargs) self.bridgemib = BridgeMib(self.agent) self.qbridgemib = QBridgeMib(self.agent) def _remap_vlan(self, vlan_ident): """Hook-in for subclasses to remap VLAN numbers if necessary""" return vlan_ident @inlineCallbacks def handle(self): """Collects VLAN configuration using Q-BRIDGE-MIB. If no PVID information can be found in Q-BRIDGE, the plugin assumes the device either doesn't support Q-BRIDGE or doesn't support 802.1q and exits. """ self._logger.debug("Collecting 802.1q VLAN information") self.baseports = yield self.bridgemib.get_baseport_ifindex_map() self.pvids = yield self.qbridgemib.get_baseport_pvid_map() if self.pvids: self._process_pvids() else: return yield self._get_vlan_names() yield self._get_tagging_info() def _process_pvids(self): """Process the list of collected port/PVID mappings. If no PVID data was found, it is assumed that this bridge either does not support Q-BRIDGE-MIB or does not support 802.1q, and the plugin exits. """ self._logger.debug("PVID mapping: %r", self.pvids) for port, pvid in self.pvids.items(): self._set_port_pvid(port, pvid) def _set_port_pvid(self, port, pvid): if port in self.baseports: ifindex = self.baseports[port] interface = self.containers.factory(ifindex, shadows.Interface) interface.vlan = pvid interface.trunk = False # default all ports to non-tagging at first else: self._logger.debug("saw reference to non-existant baseport %s", port) @inlineCallbacks def _get_tagging_info(self): """Retrieves and processes information about VLAN egress ports and tagging. The set of untagged ports is subtracted from the total set of egress ports for each vlan, resulting in a set of ports that transmit and receive tagged frames for this vlan (i.e. trunk ports). """ egress, untagged = yield self._retrieve_vlan_ports() trunkports = self._find_trunkports(egress, untagged) self._logger.debug("trunkports: %r", trunkports) self._store_trunkports(trunkports) @inlineCallbacks def _retrieve_vlan_ports(self): query = self.qbridgemib egress = yield query.get_vlan_current_egress_ports() untagged = yield query.get_vlan_current_untagged_ports() if not egress and not untagged: egress = yield query.get_vlan_static_egress_ports() untagged = yield query.get_vlan_static_untagged_ports() returnValue((egress, untagged)) def _find_trunkports(self, egress, untagged): trunkports = defaultdict(list) for vlan, (egress, untagged) in mergedicts(egress, untagged).items(): try: tagged = egress - untagged except ValueError: self._logger.error( "vlan %s subtraction mismatch between " "EgressPorts and UntaggedPorts", vlan) else: for port in tagged: trunkports[port].append(vlan) finally: self._logger.debug("vlan: %s egress: %r untagged: %r", vlan, egress, untagged) return trunkports def _store_trunkports(self, trunkports): for port, vlans in trunkports.items(): self._set_trunkport(port, vlans) def _set_trunkport(self, port, vlans): if port not in self.baseports: self._logger.debug("saw reference to non-existant baseport %s", port) return # Mark as trunk ifindex = self.baseports[port] interface = self.containers.factory(ifindex, shadows.Interface) interface.trunk = True # Store a hex string representation of enabled VLANs # in swportallowedvlan allowed = self.containers.factory(ifindex, shadows.SwPortAllowedVlan) allowed.interface = interface allowed.hex_string = vlan_list_to_hex(vlans) @inlineCallbacks def _get_vlan_names(self): names = yield self.qbridgemib.get_vlan_static_names() if names: for vlannum, name in names.items(): vlannum = self._remap_vlan(vlannum) suffix = '+{}'.format(vlannum) if name.endswith(suffix): name = name[:-len(suffix)] vlan = self.containers.factory(name, shadows.Vlan) vlan.net_type = shadows.NetType.get('lan') vlan.vlan = vlannum vlan.net_ident = name vlan.netbox = self.netbox self._logger.debug("Found vlan {}: {}".format(vlannum, name))
def __init__(self, *args, **kwargs): super(Dot1q, self).__init__(*args, **kwargs) self.bridgemib = BridgeMib(self.agent) self.qbridgemib = QBridgeMib(self.agent)
class Dot1q(Plugin): """Collect 802.1q info from BRIDGE and Q-BRIDGE MIBs.""" baseports = {} pvids = {} def __init__(self, *args, **kwargs): super(Dot1q, self).__init__(*args, **kwargs) self.bridgemib = BridgeMib(self.agent) self.qbridgemib = QBridgeMib(self.agent) @inlineCallbacks def handle(self): """Collects VLAN configuration using Q-BRIDGE-MIB. If no PVID information can be found in Q-BRIDGE, the plugin assumes the device either doesn't support Q-BRIDGE or doesn't support 802.1q and exits. """ self._logger.debug("Collecting 802.1q VLAN information") self.baseports = yield self.bridgemib.get_baseport_ifindex_map() self.pvids = yield self.qbridgemib.get_baseport_pvid_map() if self.pvids: self._process_pvids() else: return yield self._get_tagging_info() def _process_pvids(self): """Process the list of collected port/PVID mappings. If no PVID data was found, it is assumed that this bridge either does not support Q-BRIDGE-MIB or does not support 802.1q, and the plugin exits. """ self._logger.debug("PVID mapping: %r", self.pvids) for port, pvid in self.pvids.items(): self._set_port_pvid(port, pvid) def _set_port_pvid(self, port, pvid): if port in self.baseports: ifindex = self.baseports[port] interface = self.containers.factory(ifindex, shadows.Interface) interface.vlan = pvid else: self._logger.debug("saw reference to non-existant baseport %s", port) @inlineCallbacks def _get_tagging_info(self): """Retrieves and processes information about VLAN egress ports and tagging. The set of untagged ports is subtracted from the total set of egress ports for each vlan, resulting in a set of ports that transmit and receive tagged frames for this vlan (i.e. trunk ports). """ egress, untagged = yield self._retrieve_vlan_ports() trunkports = self._find_trunkports(egress, untagged) self._logger.debug("trunkports: %r", trunkports) self._store_trunkports(trunkports) @inlineCallbacks def _retrieve_vlan_ports(self): query = self.qbridgemib egress = yield query.get_vlan_current_egress_ports() untagged = yield query.get_vlan_current_untagged_ports() if not egress or not untagged: egress = yield query.get_vlan_static_egress_ports() untagged = yield query.get_vlan_static_untagged_ports() returnValue((egress, untagged)) def _find_trunkports(self, egress, untagged): trunkports = {} for vlan, (egress, untagged) in mergedicts(egress, untagged).items(): try: tagged = egress - untagged except ValueError: self._logger.error("vlan %s subtraction mismatch between " "EgressPorts and UntaggedPorts", vlan) else: for port in tagged.get_ports(): if port not in trunkports: trunkports[port] = [vlan] else: trunkports[port].append(vlan) finally: self._logger.debug("vlan: %s egress: %r untagged: %r", vlan, egress.get_ports(), untagged.get_ports()) return trunkports def _store_trunkports(self, trunkports): for port, vlans in trunkports.items(): self._set_trunkport(port, vlans) def _set_trunkport(self, port, vlans): if port not in self.baseports: self._logger.debug("saw reference to non-existant baseport %s", port) return # Mark as trunk ifindex = self.baseports[port] interface = self.containers.factory(ifindex, shadows.Interface) interface.trunk = True # Store a hex string representation of enabled VLANs # in swportallowedvlan allowed = self.containers.factory(ifindex, shadows.SwPortAllowedVlan) allowed.interface = interface allowed.hex_string = vlan_list_to_hex(vlans)