mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-06 17:15:38 -08:00
128 lines
4.7 KiB
Python
Executable File
128 lines
4.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from __future__ import unicode_literals
|
|
import pathlib
|
|
import subprocess
|
|
import argparse
|
|
import os
|
|
import sys
|
|
|
|
# Register NetAlertX directories
|
|
INSTALL_PATH="/app"
|
|
sys.path.extend([f"{INSTALL_PATH}/front/plugins", f"{INSTALL_PATH}/server"])
|
|
|
|
from plugin_helper import Plugin_Object, Plugin_Objects, decodeBase64, handleEmpty, normalize_mac
|
|
from logger import mylog, Logger
|
|
from helper import get_setting_value
|
|
from const import logPath, applicationPath
|
|
import conf
|
|
from pytz import timezone
|
|
|
|
# Make sure the TIMEZONE for logging is correct
|
|
conf.tz = timezone(get_setting_value('TIMEZONE'))
|
|
|
|
# Make sure log level is initialized correctly
|
|
Logger(get_setting_value('LOG_LEVEL'))
|
|
|
|
pluginName = "SNMPDSC"
|
|
|
|
LOG_PATH = logPath + '/plugins'
|
|
RESULT_FILE = os.path.join(LOG_PATH, f'last_result.{pluginName}.log')
|
|
|
|
# Workflow
|
|
|
|
def main():
|
|
mylog('verbose', ['[SNMPDSC] In script '])
|
|
|
|
# init global variables
|
|
global snmpWalkCmds
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='This plugin is used to discover devices via the arp table(s) of a RFC1213 compliant router or switch.')
|
|
parser.add_argument('routers', action="store", help="IP(s) of routers, separated by comma (,) if passing multiple")
|
|
values = parser.parse_args()
|
|
|
|
timeoutSetting = get_setting_value("SNMPDSC_RUN_TIMEOUT")
|
|
|
|
plugin_objects = Plugin_Objects(RESULT_FILE)
|
|
|
|
if values.routers:
|
|
snmpWalkCmds = values.routers.split('=')[1].replace('\'','')
|
|
|
|
|
|
if ',' in snmpWalkCmds:
|
|
commands = snmpWalkCmds.split(',')
|
|
else:
|
|
commands = [snmpWalkCmds]
|
|
|
|
for cmd in commands:
|
|
mylog('verbose', ['[SNMPDSC] Router snmpwalk command: ', cmd])
|
|
# split the string, remove white spaces around each item, and exclude any empty strings
|
|
snmpwalkArgs = [arg.strip() for arg in cmd.split(' ') if arg.strip()]
|
|
|
|
# Execute N probes and insert in list
|
|
probes = 1 # N probes
|
|
|
|
for _ in range(probes):
|
|
output = subprocess.check_output (snmpwalkArgs, universal_newlines=True, stderr=subprocess.STDOUT, timeout=(timeoutSetting))
|
|
|
|
mylog('verbose', ['[SNMPDSC] output: ', output])
|
|
|
|
lines = output.split('\n')
|
|
|
|
for line in lines:
|
|
|
|
tmpSplt = line.split('"')
|
|
|
|
if len(tmpSplt) == 3:
|
|
|
|
ipStr = tmpSplt[0].split('.')[-4:] # Get the last 4 elements to extract the IP
|
|
macStr = tmpSplt[1].strip().split(' ') # Remove leading/trailing spaces from MAC
|
|
|
|
if len(ipStr) == 4:
|
|
macAddress = ':'.join(macStr)
|
|
ipAddress = '.'.join(ipStr)
|
|
|
|
mylog('verbose', [f'[SNMPDSC] IP: {ipAddress} MAC: {macAddress}'])
|
|
|
|
plugin_objects.add_object(
|
|
primaryId = handleEmpty(macAddress),
|
|
secondaryId = handleEmpty(ipAddress.strip()), # Remove leading/trailing spaces from IP
|
|
watched1 = '(unknown)',
|
|
watched2 = handleEmpty(snmpwalkArgs[6]), # router IP
|
|
extra = handleEmpty(line),
|
|
foreignKey = handleEmpty(macAddress) # Use the primary ID as the foreign key
|
|
)
|
|
else:
|
|
mylog('verbose', [f'[SNMPDSC] ipStr does not seem to contain a valid IP:', ipStr])
|
|
|
|
|
|
elif line.startswith('ipNetToMediaPhysAddress'):
|
|
# Format: snmpwalk -OXsq output
|
|
parts = line.split()
|
|
if len(parts) == 2:
|
|
|
|
ipAddress = parts[0].split('[')[-1][:-1]
|
|
macAddress = normalize_mac(parts[1])
|
|
|
|
mylog('verbose', [f'[SNMPDSC] IP: {ipAddress} MAC: {macAddress}'])
|
|
|
|
plugin_objects.add_object(
|
|
primaryId = handleEmpty(macAddress),
|
|
secondaryId = handleEmpty(ipAddress.strip()),
|
|
watched1 = '(unknown)',
|
|
watched2 = handleEmpty(snmpwalkArgs[6]),
|
|
extra = handleEmpty(line),
|
|
foreignKey = handleEmpty(macAddress)
|
|
)
|
|
|
|
mylog('verbose', ['[SNMPDSC] Entries found: ', len(plugin_objects)])
|
|
|
|
plugin_objects.write_result_file()
|
|
|
|
|
|
|
|
# BEGIN
|
|
if __name__ == '__main__':
|
|
main()
|