2014-10-09 16:02:46 -07:00
|
|
|
#!/usr/bin/python
|
|
|
|
#
|
|
|
|
# Copyright 2014 Cumulus Networks, Inc. All rights reserved.
|
|
|
|
# Author: Roopa Prabhu, roopa@cumulusnetworks.com
|
|
|
|
#
|
|
|
|
|
2016-04-29 21:21:45 +02:00
|
|
|
from cache import MSTPAttrsCache
|
2014-10-09 16:02:46 -07:00
|
|
|
from utilsbase import *
|
|
|
|
from ifupdown.iface import *
|
2016-05-13 19:52:57 +02:00
|
|
|
from ifupdown.utils import utils
|
2014-10-09 16:02:46 -07:00
|
|
|
from cache import *
|
|
|
|
import re
|
2015-11-29 20:22:06 -05:00
|
|
|
import json
|
2014-10-09 16:02:46 -07:00
|
|
|
|
|
|
|
class mstpctlutil(utilsBase):
|
|
|
|
""" This class contains helper methods to interact with mstpd using
|
|
|
|
mstputils commands """
|
|
|
|
|
|
|
|
_cache_fill_done = False
|
|
|
|
|
|
|
|
_bridgeattrmap = {'bridgeid' : 'bridge-id',
|
|
|
|
'maxage' : 'max-age',
|
|
|
|
'fdelay' : 'forward-delay',
|
|
|
|
'txholdcount' : 'tx-hold-count',
|
|
|
|
'maxhops' : 'max-hops',
|
|
|
|
'ageing' : 'ageing-time',
|
|
|
|
'hello' : 'hello-time',
|
|
|
|
'forcevers' : 'force-protocol-version'}
|
|
|
|
|
|
|
|
_bridgeportattrmap = {'portadminedge' : 'admin-edge-port',
|
2014-11-09 13:54:33 -08:00
|
|
|
'portp2p' : 'admin-point-to-point',
|
2014-10-09 16:02:46 -07:00
|
|
|
'portrestrrole' : 'restricted-role',
|
|
|
|
'portrestrtcn' : 'restricted-TCN',
|
|
|
|
'bpduguard' : 'bpdu-guard-port',
|
|
|
|
'portautoedge' : 'auto-edge-port',
|
|
|
|
'portnetwork' : 'network-port',
|
|
|
|
'portbpdufilter' : 'bpdufilter-port'}
|
|
|
|
|
|
|
|
def __init__(self, *args, **kargs):
|
|
|
|
utilsBase.__init__(self, *args, **kargs)
|
|
|
|
|
|
|
|
def is_mstpd_running(self):
|
|
|
|
try:
|
2016-05-13 19:52:57 +02:00
|
|
|
utils.exec_command('/bin/pidof mstpd')
|
2014-10-09 16:02:46 -07:00
|
|
|
except:
|
|
|
|
return False
|
|
|
|
else:
|
|
|
|
return True
|
|
|
|
|
|
|
|
def get_bridgeport_attr(self, bridgename, portname, attrname):
|
|
|
|
try:
|
2016-05-13 19:52:57 +02:00
|
|
|
cmdl = ['/sbin/mstpctl', 'showportdetail', bridgename, portname,
|
|
|
|
self._bridgeportattrmap[attrname]]
|
|
|
|
return utils.exec_commandl(cmdl).strip('\n')
|
2014-10-09 16:02:46 -07:00
|
|
|
except Exception, e:
|
|
|
|
pass
|
|
|
|
return None
|
|
|
|
|
|
|
|
def get_bridgeport_attrs(self, bridgename, portname):
|
|
|
|
bridgeattrs = {}
|
|
|
|
try:
|
|
|
|
bridgeattrs = dict((k, self.get_bridgeport_attr(bridgename, v))
|
|
|
|
for k, v in self._bridgeattrmap.items())
|
|
|
|
bridgeattrs['treeprio'] = int(bridgeattrs.get('bridgeid',
|
|
|
|
'').split('.')[0], base=16) * 4096
|
|
|
|
except Exception, e:
|
2016-04-19 15:18:35 +02:00
|
|
|
self.logger.warn(str(e))
|
2014-10-09 16:02:46 -07:00
|
|
|
pass
|
|
|
|
return bridgeattrs
|
|
|
|
|
2016-04-29 21:21:45 +02:00
|
|
|
def _get_mstpctl_bridgeport_attr_from_cache(self, bridgename):
|
|
|
|
attrs = MSTPAttrsCache.get(bridgename)
|
|
|
|
if not attrs:
|
|
|
|
try:
|
|
|
|
cmd = ['/sbin/mstpctl', 'showportdetail', bridgename, 'json']
|
2016-05-13 19:52:57 +02:00
|
|
|
output = utils.exec_commandl(cmd)
|
2016-04-29 21:21:45 +02:00
|
|
|
if not output:
|
|
|
|
return None
|
|
|
|
except Exception as e:
|
|
|
|
self.logger.info(str(e))
|
|
|
|
return None
|
|
|
|
mstpctl_bridgeport_attrs_dict = {}
|
|
|
|
try:
|
|
|
|
mstpctl_bridge_cache = json.loads(output.strip('\n'))
|
|
|
|
for portname in mstpctl_bridge_cache.keys():
|
|
|
|
# we will ignore the portid for now and just index
|
|
|
|
# by bridgename, portname, and json attribute
|
|
|
|
for portid in mstpctl_bridge_cache[portname].keys():
|
|
|
|
mstpctl_bridgeport_attrs_dict[portname] = {}
|
|
|
|
for jsonAttr in mstpctl_bridge_cache[portname][portid].keys():
|
|
|
|
jsonVal = mstpctl_bridge_cache[portname][portid][jsonAttr]
|
|
|
|
mstpctl_bridgeport_attrs_dict[portname][jsonAttr] = str(jsonVal)
|
|
|
|
MSTPAttrsCache.set(bridgename, mstpctl_bridgeport_attrs_dict)
|
|
|
|
return mstpctl_bridgeport_attrs_dict
|
|
|
|
except Exception as e:
|
|
|
|
self.logger.info('%s: cannot fetch mstpctl bridge port attributes: %s', str(e))
|
|
|
|
return attrs
|
2015-11-29 20:22:06 -05:00
|
|
|
|
2016-04-29 21:21:45 +02:00
|
|
|
def get_mstpctl_bridgeport_attr(self, bridgename, portname, attr):
|
|
|
|
attrs = self._get_mstpctl_bridgeport_attr_from_cache(bridgename)
|
|
|
|
if not attrs:
|
|
|
|
return 'no'
|
|
|
|
else:
|
|
|
|
val = attrs.get(portname,{}).get(attr, 'no')
|
|
|
|
if val == 'True':
|
|
|
|
val = 'yes'
|
|
|
|
return str(val)
|
2015-11-29 20:22:06 -05:00
|
|
|
|
2014-10-09 16:02:46 -07:00
|
|
|
def set_bridgeport_attrs(self, bridgename, bridgeportname, attrdict,
|
|
|
|
check=True):
|
|
|
|
for k, v in attrdict.iteritems():
|
|
|
|
if not v:
|
|
|
|
continue
|
|
|
|
try:
|
2016-05-05 22:09:49 -07:00
|
|
|
self.set_bridgeport_attr(bridgename, bridgeportname,
|
2014-10-09 16:02:46 -07:00
|
|
|
k, v, check)
|
|
|
|
except Exception, e:
|
|
|
|
self.logger.warn(str(e))
|
|
|
|
|
|
|
|
def set_bridgeport_attr(self, bridgename, bridgeportname, attrname,
|
|
|
|
attrvalue, check=True):
|
|
|
|
if check:
|
|
|
|
attrvalue_curr = self.get_bridgeport_attr(bridgename,
|
|
|
|
bridgeportname, attrname)
|
|
|
|
if attrvalue_curr and attrvalue_curr == attrvalue:
|
|
|
|
return
|
|
|
|
if attrname == 'treeportcost' or attrname == 'treeportprio':
|
2016-05-13 19:52:57 +02:00
|
|
|
utils.exec_commandl(['/sbin/mstpctl', 'set%s' % attrname,
|
|
|
|
'%s' % bridgename, '%s' % bridgeportname, '0',
|
|
|
|
'%s' % attrvalue])
|
2014-10-09 16:02:46 -07:00
|
|
|
else:
|
2016-05-13 19:52:57 +02:00
|
|
|
utils.exec_commandl(['/sbin/mstpctl', 'set%s' % attrname,
|
|
|
|
'%s' % bridgename, '%s' % bridgeportname,
|
|
|
|
'%s' % attrvalue])
|
2014-10-09 16:02:46 -07:00
|
|
|
|
|
|
|
def get_bridge_attrs(self, bridgename):
|
|
|
|
bridgeattrs = {}
|
|
|
|
try:
|
|
|
|
bridgeattrs = dict((k, self.get_bridge_attr(bridgename, k))
|
|
|
|
for k in self._bridgeattrmap.keys())
|
|
|
|
bridgeattrs['treeprio'] = '%d' %(int(bridgeattrs.get('bridgeid',
|
|
|
|
'').split('.')[0], base=16) * 4096)
|
|
|
|
del bridgeattrs['bridgeid']
|
|
|
|
except Exception, e:
|
2014-11-09 13:54:33 -08:00
|
|
|
self.logger.debug(bridgeattrs)
|
|
|
|
self.logger.debug(str(e))
|
2014-10-09 16:02:46 -07:00
|
|
|
pass
|
|
|
|
return bridgeattrs
|
|
|
|
|
|
|
|
def get_bridge_attr(self, bridgename, attrname):
|
|
|
|
try:
|
2016-05-13 19:52:57 +02:00
|
|
|
cmdl = ['/sbin/mstpctl', 'showbridge', bridgename,
|
|
|
|
self._bridgeattrmap[attrname]]
|
|
|
|
return utils.exec_commandl(cmdl).strip('\n')
|
2014-10-09 16:02:46 -07:00
|
|
|
except Exception, e:
|
|
|
|
pass
|
|
|
|
return None
|
|
|
|
|
|
|
|
def set_bridge_attr(self, bridgename, attrname, attrvalue, check=True):
|
|
|
|
|
|
|
|
if check:
|
|
|
|
attrvalue_curr = self.get_bridge_attr(bridgename, attrname)
|
|
|
|
if attrvalue_curr and attrvalue_curr == attrvalue:
|
|
|
|
return
|
|
|
|
if attrname == 'treeprio':
|
2016-05-13 19:52:57 +02:00
|
|
|
utils.exec_commandl(['/sbin/mstpctl', 'set%s' % attrname,
|
|
|
|
'%s' % bridgename, '0', '%s' % attrvalue],
|
|
|
|
stdout=False, stderr=None)
|
2014-10-09 16:02:46 -07:00
|
|
|
else:
|
2016-05-13 19:52:57 +02:00
|
|
|
utils.exec_commandl(['/sbin/mstpctl', 'set%s' % attrname,
|
|
|
|
'%s' % bridgename, '%s' % attrvalue],
|
|
|
|
stdout=False, stderr=None)
|
2014-10-09 16:02:46 -07:00
|
|
|
|
|
|
|
def set_bridge_attrs(self, bridgename, attrdict, check=True):
|
|
|
|
for k, v in attrdict.iteritems():
|
|
|
|
if not v:
|
|
|
|
continue
|
|
|
|
try:
|
|
|
|
self.set_bridge_attr(bridgename, k, v, check)
|
|
|
|
except Exception, e:
|
|
|
|
self.logger.warn('%s: %s' %(bridgename, str(e)))
|
|
|
|
pass
|
|
|
|
|
|
|
|
def get_bridge_treeprio(self, bridgename):
|
|
|
|
try:
|
2016-05-13 19:52:57 +02:00
|
|
|
cmdl = ['/sbin/mstpctl',
|
|
|
|
'showbridge',
|
|
|
|
bridgename,
|
|
|
|
self._bridgeattrmap['bridgeid']]
|
|
|
|
|
|
|
|
bridgeid = utils.exec_commandl(cmdl).strip('\n')
|
2014-10-09 16:02:46 -07:00
|
|
|
return '%d' %(int(bridgeid.split('.')[0], base=16) * 4096)
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
return None
|
|
|
|
|
|
|
|
def set_bridge_treeprio(self, bridgename, attrvalue, check=True):
|
|
|
|
if check:
|
|
|
|
attrvalue_curr = self.get_bridge_treeprio(bridgename)
|
|
|
|
if attrvalue_curr and attrvalue_curr == attrvalue:
|
|
|
|
return
|
2016-05-13 19:52:57 +02:00
|
|
|
utils.exec_commandl(['/sbin/mstpctl', 'settreeprio', bridgename, '0',
|
|
|
|
str(attrvalue)])
|
2014-10-09 16:02:46 -07:00
|
|
|
|
|
|
|
def showbridge(self, bridgename=None):
|
|
|
|
if bridgename:
|
2016-05-13 19:52:57 +02:00
|
|
|
return utils.exec_command('/sbin/mstpctl showbridge %s' % bridgename)
|
2014-10-09 16:02:46 -07:00
|
|
|
else:
|
2016-05-13 19:52:57 +02:00
|
|
|
return utils.exec_command('/sbin/mstpctl showbridge')
|
2014-10-09 16:02:46 -07:00
|
|
|
|
|
|
|
def showportdetail(self, bridgename):
|
2016-05-13 19:52:57 +02:00
|
|
|
return utils.exec_command('/sbin/mstpctl showportdetail %s' % bridgename)
|
2014-10-09 16:02:46 -07:00
|
|
|
|
|
|
|
def mstpbridge_exists(self, bridgename):
|
|
|
|
try:
|
2016-05-13 19:52:57 +02:00
|
|
|
utils.exec_command('mstpctl showbridge %s' % bridgename, stdout=False)
|
2014-10-09 16:02:46 -07:00
|
|
|
return True
|
|
|
|
except:
|
|
|
|
return False
|