def getMibRevision(mibDir, mibFile):
mibCompiler = MibCompiler(
mibParser,
codeGenerator,
fileWriter
)
mibCompiler.addSources(
FileReader(mibDir, recursive=False, ignoreErrors=ignoreErrorsFlag),
*getReadersFromUrls(*mibSources)
)
try:
processed = mibCompiler.compile(
mibFile, **dict(noDeps=True, rebuild=True, fuzzyMatching=False, ignoreErrors=ignoreErrorsFlag)
)
except error.PySmiError:
sys.stderr.write('ERROR: %s\r\n' % sys.exc_info()[1])
sys.exit(EX_SOFTWARE)
for canonicalMibName in processed:
if (processed[canonicalMibName] == 'compiled' and
processed[canonicalMibName].path == 'file://' + os.path.join(mibDir, mibFile)):
try:
revision = datetime.strptime(processed[canonicalMibName].revision, '%Y-%m-%d %H:%M')
except Exception:
revision = datetime.fromtimestamp(0)
return canonicalMibName, revision
raise error.PySmiError('Can\'t read or parse MIB "%s"' % os.path.join(mibDir, mibFile))
def __init__(self, mib_list, src_dir_list, http_sources=None):
if "BFT_DEBUG" in os.environ:
self.dbg = os.environ.get('BFT_DEBUG')
else:
self.dbg = ""
if "yy" in self.dbg:
# VERY verbose, but essential for spotting
# possible ASN.1 errors
from pysmi import debug
debug.setLogger(debug.Debug('reader', 'compiler'))
# Initialize compiler infrastructure
mibCompiler = MibCompiler(SmiStarParser(), JsonCodeGen(),
CallbackWriter(self.callback_func))
# search for source MIBs here
mibCompiler.addSources(*[FileReader(x) for x in src_dir_list])
if http_sources:
# search for source MIBs at Web sites
mibCompiler.addSources(*[HttpReader(*x) for x in http_sources])
# never recompile MIBs with MACROs
mibCompiler.addSearchers(StubSearcher(*JsonCodeGen.baseMibs))
# run recursive MIB compilation
mibCompiler.compile(*mib_list)
def _compile_mib_to_json(mib, source_mib_directories, destination_directory):
from pysmi.codegen import JsonCodeGen
from pysmi.compiler import MibCompiler
from pysmi.parser import SmiV1CompatParser
from pysmi.reader import FileReader, HttpReader
from pysmi.searcher import AnyFileSearcher, StubSearcher
from pysmi.writer import FileWriter
mib_stubs = JsonCodeGen.baseMibs
compile_documentation = True
# Compiler infrastructure
searchers = [
AnyFileSearcher(destination_directory).setOptions(exts=['.json']),
StubSearcher(*mib_stubs)
]
code_generator = JsonCodeGen()
file_writer = FileWriter(destination_directory).setOptions(suffix='.json')
mib_compiler = MibCompiler(SmiV1CompatParser(tempdir=''), code_generator,
file_writer)
# use source_mib_directories as mibs source
for source_directory in source_mib_directories:
mib_compiler.addSources(FileReader(source_directory))
# use snmp mibs repo as mibs source
reader = HttpReader('raw.githubusercontent.com', 80,
'/projx/snmp-mibs/master/@mib@')
mib_compiler.addSources(reader)
mib_compiler.addSearchers(*searchers)
processed = mib_compiler.compile(
mib,
**dict(
noDeps=False,
rebuild=False,
dryRun=False,
dstTemplate=None,
genTexts=compile_documentation,
textFilter=False and (lambda symbol, text: text) or None,
writeMibs=True,
ignoreErrors=False,
))
return processed
def __init__(self, mib_list, src_dir_list, http_sources=None):
if "BFT_DEBUG" in os.environ:
self.dbg = os.environ.get('BFT_DEBUG')
else:
self.dbg = ""
if "yy" in self.dbg:
# VERY verbose, but essential for spotting
# possible ASN.1 errors
from pysmi import debug
debug.setLogger(debug.Debug('reader', 'compiler'))
# Initialize compiler infrastructure
mibCompiler = MibCompiler(SmiStarParser(), JsonCodeGen(),
CallbackWriter(self.callback_func))
# search for source MIBs here
mibCompiler.addSources(*[FileReader(x) for x in src_dir_list])
if http_sources:
# search for source MIBs at Web sites
mibCompiler.addSources(*[HttpReader(*x) for x in http_sources])
# never recompile MIBs with MACROs
mibCompiler.addSearchers(StubSearcher(*JsonCodeGen.baseMibs))
# run recursive MIB compilation
mib_dict = mibCompiler.compile(*mib_list)
err = False
if mib_dict is None or mib_dict == {}:
print(
"ERROR: failed on mib compilation (mibCompiler.compile returned an empty dictionary)"
)
err = True
for key, value in mib_dict.iteritems():
if value == 'unprocessed':
print("ERROR: failed on mib compilation: " + key + ": " +
value)
err = True
if err:
raise Exception("SnmpMibs failed to initialize.")
elif 'BFT_DEBUG' in os.environ:
print('# %d MIB modules compiled' % len(mib_dict))
def transform_mibs_to_json(self,
specific_mibs: object = None,
exclude_mibs: object = []) -> object:
mib_names = []
if specific_mibs is None:
for filename in os.listdir(self.local_mib_dir):
try:
filename.rindex('.my', -3)
except:
continue
mib_names.append(filename)
else:
mib_names = specific_mibs.copy()
logging.info('Compiling MIBs to JSON.')
try:
mib_compiler = MibCompiler(
SmiV1CompatParser(), JsonCodeGen(),
FileWriter(self.local_json_dir).setOptions(suffix='.json'))
mib_compiler.addSources(
FileReader(self.local_mib_dir, recursive=True))
mib_stubs = JsonCodeGen.baseMibs
searchers = [
AnyFileSearcher(
self.local_json_dir).setOptions(exts=['.json']),
StubSearcher(*mib_stubs)
]
mib_compiler.addSearchers(*searchers)
if not os.path.isdir(self.local_json_dir):
os.makedirs(self.local_json_dir)
processed = mib_compiler.compile(
*mib_names,
**dict(noDeps=False,
rebuild=True,
genTexts=True,
writeMibs=True,
ignoreErrors=True))
mib_compiler.buildIndex(processed, ignoreErrors=True)
except error.PySmiError:
logging.error('ERROR: %s', str(sys.exc_info()[1]))
sys.exit(1)
logging.info('Finished compiling MIBs to JSON.')
inputMibs = ['IF-MIB', 'IP-MIB']
srcDirectories = ['/usr/share/snmp/mibs']
httpSources = [('mibs.snmplabs.com', 80, '/asn1/@mib@')]
def printOut(mibName, jsonDoc, cbCtx):
print('\n\n# MIB module %s' % mibName)
print(jsonDoc)
# Initialize compiler infrastructure
mibCompiler = MibCompiler(SmiStarParser(), JsonCodeGen(),
CallbackWriter(printOut))
# search for source MIBs here
mibCompiler.addSources(*[FileReader(x) for x in srcDirectories])
# search for source MIBs at Web sites
mibCompiler.addSources(*[HttpReader(*x) for x in httpSources])
# never recompile MIBs with MACROs
mibCompiler.addSearchers(StubSearcher(*JsonCodeGen.baseMibs))
# run recursive MIB compilation
results = mibCompiler.compile(*inputMibs)
print('\n# Results: %s' %
', '.join(['%s:%s' % (x, results[x]) for x in results]))
def __init__(self, host, port, raw_mibs, compiled_mibs):
self.oid_mapping = {}
self.databus_mediator = DatabusMediator(self.oid_mapping)
# mapping between OID and databus keys
# Create SNMP engine
self.snmpEngine = engine.SnmpEngine()
# Configure SNMP compiler
mib_builder = self.snmpEngine.getMibBuilder()
addMibCompiler(mib_builder, destination=compiled_mibs)
mib_builder.getMibCompiler().addSources(FileReader(raw_mibs))
mib_builder.getMibCompiler().addSources(
FtpReader("ftp.cisco.com", "/pub/mibs/v2/@mib@", 80))
# Transport setup
udp_sock = gevent.socket.socket(gevent.socket.AF_INET,
gevent.socket.SOCK_DGRAM)
udp_sock.setsockopt(gevent.socket.SOL_SOCKET,
gevent.socket.SO_BROADCAST, 1)
udp_sock.bind((host, port))
self.server_port = udp_sock.getsockname()[1]
# UDP over IPv4
self.addSocketTransport(self.snmpEngine, udp.domainName, udp_sock)
# SNMPv1
config.addV1System(self.snmpEngine, "public-read", "public")
# SNMPv3/USM setup
# user: usr-md5-des, auth: MD5, priv DES
config.addV3User(
self.snmpEngine,
"usr-md5-des",
config.usmHMACMD5AuthProtocol,
"authkey1",
config.usmDESPrivProtocol,
"privkey1",
)
# user: usr-sha-none, auth: SHA, priv NONE
config.addV3User(self.snmpEngine, "usr-sha-none",
config.usmHMACSHAAuthProtocol, "authkey1")
# user: usr-sha-aes128, auth: SHA, priv AES/128
config.addV3User(
self.snmpEngine,
"usr-sha-aes128",
config.usmHMACSHAAuthProtocol,
"authkey1",
config.usmAesCfb128Protocol,
"privkey1",
)
# Allow full MIB access for each user at VACM
config.addVacmUser(
self.snmpEngine,
1,
"public-read",
"noAuthNoPriv",
readSubTree=(1, 3, 6, 1, 2, 1),
writeSubTree=(1, 3, 6, 1, 2, 1),
)
config.addVacmUser(
self.snmpEngine,
2,
"public-read",
"noAuthNoPriv",
readSubTree=(1, 3, 6, 1, 2, 1),
writeSubTree=(1, 3, 6, 1, 2, 1),
)
config.addVacmUser(
self.snmpEngine,
3,
"usr-md5-des",
"authPriv",
readSubTree=(1, 3, 6, 1, 2, 1),
writeSubTree=(1, 3, 6, 1, 2, 1),
)
config.addVacmUser(
self.snmpEngine,
3,
"usr-sha-none",
"authNoPriv",
readSubTree=(1, 3, 6, 1, 2, 1),
writeSubTree=(1, 3, 6, 1, 2, 1),
)
config.addVacmUser(
self.snmpEngine,
3,
"usr-sha-aes128",
"authPriv",
readSubTree=(1, 3, 6, 1, 2, 1),
writeSubTree=(1, 3, 6, 1, 2, 1),
)
# Get default SNMP context this SNMP engine serves
snmpContext = context.SnmpContext(self.snmpEngine)
# Register SNMP Applications at the SNMP engine for particular SNMP context
self.resp_app_get = conpot_cmdrsp.c_GetCommandResponder(
self.snmpEngine, snmpContext, self.databus_mediator, host, port)
self.resp_app_set = conpot_cmdrsp.c_SetCommandResponder(
self.snmpEngine, snmpContext, self.databus_mediator, host, port)
self.resp_app_next = conpot_cmdrsp.c_NextCommandResponder(
self.snmpEngine, snmpContext, self.databus_mediator, host, port)
self.resp_app_bulk = conpot_cmdrsp.c_BulkCommandResponder(
self.snmpEngine, snmpContext, self.databus_mediator, host, port)
