mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-07 01:26:11 -08:00
137 lines
4.5 KiB
Python
Executable File
137 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import sqlite3
|
|
|
|
# Register NetAlertX directories
|
|
INSTALL_PATH = os.getenv('NETALERTX_APP', '/app')
|
|
sys.path.extend([f"{INSTALL_PATH}/front/plugins", f"{INSTALL_PATH}/server"])
|
|
|
|
from plugin_helper import Plugin_Objects, handleEmpty # noqa: E402 [flake8 lint suppression]
|
|
from logger import mylog, Logger # noqa: E402 [flake8 lint suppression]
|
|
from helper import get_setting_value # noqa: E402 [flake8 lint suppression]
|
|
from const import logPath, applicationPath, fullDbPath # noqa: E402 [flake8 lint suppression]
|
|
from scan.device_handling import query_MAC_vendor # noqa: E402 [flake8 lint suppression]
|
|
import conf # noqa: E402 [flake8 lint suppression]
|
|
from pytz import timezone # noqa: E402 [flake8 lint suppression]
|
|
|
|
# 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 = 'VNDRPDT'
|
|
|
|
LOG_PATH = logPath + '/plugins'
|
|
LOG_FILE = os.path.join(LOG_PATH, f'script.{pluginName}.log')
|
|
RESULT_FILE = os.path.join(LOG_PATH, f'last_result.{pluginName}.log')
|
|
|
|
|
|
def main():
|
|
|
|
mylog('verbose', ['[VNDRPDT] In script'])
|
|
|
|
# Get newest DB
|
|
update_vendor_database()
|
|
|
|
# Resolve missing vendors
|
|
plugin_objects = Plugin_Objects(RESULT_FILE)
|
|
|
|
plugin_objects = update_vendors(fullDbPath, plugin_objects)
|
|
|
|
plugin_objects.write_result_file()
|
|
|
|
mylog('verbose', ['[VNDRPDT] Update complete'])
|
|
|
|
return 0
|
|
|
|
|
|
# ===============================================================================
|
|
# Update device vendors database
|
|
# ===============================================================================
|
|
def update_vendor_database():
|
|
|
|
# Update vendors DB (iab oui)
|
|
mylog('verbose', [' Updating vendors DB (iab & oui)'])
|
|
update_args = ['sh', applicationPath + '/services/update_vendors.sh']
|
|
|
|
# Execute command
|
|
try:
|
|
# try runnning a subprocess safely
|
|
subprocess.check_output(update_args)
|
|
except subprocess.CalledProcessError as e:
|
|
# An error occured, handle it
|
|
mylog('verbose', [' FAILED: Updating vendors DB, set LOG_LEVEL=debug for more info'])
|
|
mylog('verbose', [e.output])
|
|
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# resolve missing vendors
|
|
def update_vendors(dbPath, plugin_objects):
|
|
|
|
# Connect to the App SQLite database
|
|
conn = sqlite3.connect(dbPath)
|
|
sql = conn.cursor()
|
|
|
|
# Initialize variables
|
|
ignored = 0
|
|
notFound = 0
|
|
|
|
mylog('verbose', [' Searching devices vendor'])
|
|
|
|
# Get devices without a vendor
|
|
sql.execute("""SELECT
|
|
devMac,
|
|
devLastIP,
|
|
devName,
|
|
devVendor
|
|
FROM Devices
|
|
WHERE devVendor = '(unknown)'
|
|
OR devVendor = '(Unknown)'
|
|
OR devVendor = ''
|
|
OR devVendor IS NULL
|
|
""")
|
|
devices = sql.fetchall()
|
|
conn.commit()
|
|
|
|
# Close the database connection
|
|
conn.close()
|
|
|
|
# All devices loop
|
|
for device in devices:
|
|
# Search vendor in HW Vendors DB
|
|
vendor = query_MAC_vendor(device[0])
|
|
if vendor == -1 :
|
|
notFound += 1
|
|
elif vendor == -2 :
|
|
ignored += 1
|
|
else :
|
|
plugin_objects.add_object(
|
|
primaryId = handleEmpty(device[0]), # MAC (Device Name)
|
|
secondaryId = handleEmpty(device[1]), # IP Address (always 0.0.0.0)
|
|
watched1 = handleEmpty(vendor),
|
|
watched2 = handleEmpty(device[2]), # Device name
|
|
watched3 = "",
|
|
watched4 = "",
|
|
extra = "",
|
|
foreignKey = handleEmpty(device[0])
|
|
)
|
|
|
|
# Print log
|
|
mylog('verbose', [" Devices Ignored : ", ignored])
|
|
mylog('verbose', [" Devices with missing vendor : ", len(devices)])
|
|
mylog('verbose', [" Vendors Not Found : ", notFound])
|
|
mylog('verbose', [" Vendors updated : ", len(plugin_objects)])
|
|
|
|
return plugin_objects
|
|
|
|
|
|
# ===============================================================================
|
|
# BEGIN
|
|
# ===============================================================================
|
|
if __name__ == '__main__':
|
|
main()
|