def __init__(self, mibBuilder):
self.mibBuilder = mibBuilder
self.lastBuildId = -1
self.__mibSymbolsIdx = OrderedDict()
def indexMib(self):
if self.lastBuildId == self.mibBuilder.lastBuildId:
return
debug.logger & debug.flagMIB and debug.logger(
'indexMib: re-indexing MIB view')
MibScalarInstance, = self.mibBuilder.importSymbols(
'SNMPv2-SMI', 'MibScalarInstance')
#
# Create indices
#
# Module name -> module-scope indices
self.__mibSymbolsIdx = OrderedDict()
# Oid <-> label indices
# This is potentionally ambiguous mapping. Sort modules in
# ascending age for resolution
def __sortFun(x, b=self.mibBuilder):
if b.moduleID in b.mibSymbols[x]:
m = b.mibSymbols[x][b.moduleID]
r = m.getRevisions()
if r:
return r[0]
else:
return "1970-01-01 00:00"
modNames = list(self.mibBuilder.mibSymbols.keys())
modNames.sort(key=__sortFun)
# Index modules names
for modName in [''] + modNames:
# Modules index
self.__mibSymbolsIdx[modName] = mibMod = {
'oidToLabelIdx': OidOrderedDict(),
'labelToOidIdx': {},
'varToNameIdx': {},
'typeToModIdx': OrderedDict(),
'oidToModIdx': {}
}
if not modName:
globMibMod = mibMod
continue
# Types & MIB vars indices
for n, v in self.mibBuilder.mibSymbols[modName].items():
if n == self.mibBuilder.moduleID: # do not index this
continue # special symbol
if isinstance(v, classTypes):
if n in mibMod['typeToModIdx']:
raise error.SmiError(
'Duplicate SMI type %s::%s, has %s' % \
(modName, n, mibMod['typeToModIdx'][n])
)
globMibMod['typeToModIdx'][n] = modName
mibMod['typeToModIdx'][n] = modName
elif isinstance(v, instanceTypes):
if isinstance(v, MibScalarInstance):
continue
if n in mibMod['varToNameIdx']:
raise error.SmiError(
'Duplicate MIB variable %s::%s has %s' % \
(modName, n, mibMod['varToNameIdx'][n])
)
globMibMod['varToNameIdx'][n] = v.name
mibMod['varToNameIdx'][n] = v.name
# Potentionally ambiguous mapping ahead
globMibMod['oidToModIdx'][v.name] = modName
mibMod['oidToModIdx'][v.name] = modName
globMibMod['oidToLabelIdx'][v.name] = (n, )
mibMod['oidToLabelIdx'][v.name] = (n, )
else:
raise error.SmiError('Unexpected object %s::%s' %
(modName, n))
# Build oid->long-label index
oidToLabelIdx = self.__mibSymbolsIdx['']['oidToLabelIdx']
labelToOidIdx = self.__mibSymbolsIdx['']['labelToOidIdx']
if oidToLabelIdx:
prevOid = oidToLabelIdx.keys()[0]
else:
prevOid = ()
baseLabel = ()
for key in oidToLabelIdx.keys():
keydiff = len(key) - len(prevOid)
if keydiff > 0:
baseLabel = oidToLabelIdx[prevOid]
if keydiff > 1:
baseLabel = baseLabel + key[-keydiff:-1]
if keydiff < 0:
keyLen = len(key)
i = keyLen - 1
while i:
k = key[:i]
if k in oidToLabelIdx:
baseLabel = oidToLabelIdx[k]
if i != keyLen - 1:
baseLabel = baseLabel + key[i:-1]
break
i = i - 1
# Build oid->long-label index
oidToLabelIdx[key] = baseLabel + oidToLabelIdx[key]
# Build label->oid index
labelToOidIdx[oidToLabelIdx[key]] = key
prevOid = key
# Build module-scope oid->long-label index
for mibMod in self.__mibSymbolsIdx.values():
for oid in mibMod['oidToLabelIdx'].keys():
mibMod['oidToLabelIdx'][oid] = oidToLabelIdx[oid]
mibMod['labelToOidIdx'][oidToLabelIdx[oid]] = oid
self.lastBuildId = self.mibBuilder.lastBuildId
def indexMib(self):
if self.lastBuildId == self.mibBuilder.lastBuildId:
return
debug.logger & debug.FLAG_MIB and debug.logger(
'indexMib: re-indexing MIB view')
MibScalarInstance, = self.mibBuilder.importSymbols(
'SNMPv2-SMI', 'MibScalarInstance')
#
# Create indices
#
# Module name -> module-scope indices
self._mibSymbolsIdx.clear()
globMibMod = {
'oidToLabelIdx': OidOrderedDict(),
'labelToOidIdx': {},
'varToNameIdx': {},
'typeToModIdx': OrderedDict(),
'oidToModIdx': {}
}
self._mibSymbolsIdx[''] = globMibMod
# Oid <-> label indices
modNames = sorted(self.mibBuilder.mibSymbols, key=self._sortFun)
# Index modules names
for modName in modNames:
mibMod = {
'oidToLabelIdx': OidOrderedDict(),
'labelToOidIdx': {},
'varToNameIdx': {},
'typeToModIdx': OrderedDict(),
'oidToModIdx': {}
}
self._mibSymbolsIdx[modName] = mibMod
# Types & MIB vars indices
for n, v in self.mibBuilder.mibSymbols[modName].items():
if n == self.mibBuilder.moduleID: # do not index this
continue # special symbol
if isinstance(v, classTypes):
if n in mibMod['typeToModIdx']:
raise error.SmiError(
'Duplicate SMI type %s::%s, has '
'%s' % (modName, n, mibMod['typeToModIdx'][n]))
globMibMod['typeToModIdx'][n] = modName
mibMod['typeToModIdx'][n] = modName
elif isinstance(v, instanceTypes):
if isinstance(v, MibScalarInstance):
continue
if n in mibMod['varToNameIdx']:
raise error.SmiError(
'Duplicate MIB variable %s::%s has '
'%s' % (modName, n, mibMod['varToNameIdx'][n]))
globMibMod['varToNameIdx'][n] = v.name
mibMod['varToNameIdx'][n] = v.name
# Potentially ambiguous mapping ahead
globMibMod['oidToModIdx'][v.name] = modName
mibMod['oidToModIdx'][v.name] = modName
globMibMod['oidToLabelIdx'][v.name] = (n, )
mibMod['oidToLabelIdx'][v.name] = (n, )
else:
raise error.SmiError('Unexpected object %s::%s' %
(modName, n))
# Build oid->long-label index
oidToLabelIdx = self._mibSymbolsIdx['']['oidToLabelIdx']
labelToOidIdx = self._mibSymbolsIdx['']['labelToOidIdx']
prevOid = ()
baseLabel = ()
for key in oidToLabelIdx:
keydiff = len(key) - len(prevOid)
if keydiff > 0:
if prevOid:
if keydiff == 1:
baseLabel = oidToLabelIdx[prevOid]
else:
baseLabel += key[-keydiff:-1]
else:
baseLabel = ()
elif keydiff < 0:
baseLabel = ()
keyLen = len(key)
i = keyLen - 1
while i:
k = key[:i]
if k in oidToLabelIdx:
baseLabel = oidToLabelIdx[k]
if i != keyLen - 1:
baseLabel += key[i:-1]
break
i -= 1
# Build oid->long-label index
oidToLabelIdx[key] = baseLabel + oidToLabelIdx[key]
# Build label->oid index
labelToOidIdx[oidToLabelIdx[key]] = key
prevOid = key
# Build module-scope oid->long-label index
for mibMod in self._mibSymbolsIdx.values():
for oid in mibMod['oidToLabelIdx']:
mibMod['oidToLabelIdx'][oid] = oidToLabelIdx[oid]
mibMod['labelToOidIdx'][oidToLabelIdx[oid]] = oid
self.lastBuildId = self.mibBuilder.lastBuildId
