mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-06 17:15:38 -08:00
132 lines
5.2 KiB
Python
Executable File
132 lines
5.2 KiB
Python
Executable File
# !/usr/bin/env python
|
|
|
|
from __future__ import unicode_literals
|
|
import argparse
|
|
import os
|
|
import sys
|
|
import chardet
|
|
|
|
# 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, is_mac # noqa: E402 [flake8 lint suppression]
|
|
from logger import mylog, Logger # noqa: E402 [flake8 lint suppression]
|
|
from dhcp_leases import DhcpLeases # noqa: E402 [flake8 lint suppression]
|
|
from helper import get_setting_value # noqa: E402 [flake8 lint suppression]
|
|
import conf # noqa: E402 [flake8 lint suppression]
|
|
from const import logPath # 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 = 'DHCPLSS'
|
|
|
|
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', [f'[{pluginName}] In script'])
|
|
last_run_logfile = open(RESULT_FILE, 'a')
|
|
last_run_logfile.write("")
|
|
|
|
parser = argparse.ArgumentParser(description='Import devices from dhcp.leases files')
|
|
parser.add_argument(
|
|
'paths',
|
|
action="store",
|
|
help="absolute dhcp.leases file paths to check separated by ','"
|
|
)
|
|
|
|
values = parser.parse_args()
|
|
|
|
plugin_objects = Plugin_Objects(RESULT_FILE)
|
|
|
|
if values.paths:
|
|
for path in values.paths.split('=')[1].split(','):
|
|
plugin_objects = get_entries(path, plugin_objects)
|
|
mylog('verbose', [f'[{pluginName}] {len(plugin_objects)} Entries found in "{path}"'])
|
|
|
|
plugin_objects.write_result_file()
|
|
|
|
|
|
# -------------------------------------------------------------
|
|
def get_entries(path, plugin_objects):
|
|
|
|
# Check if the path exists
|
|
if not os.path.exists(path):
|
|
mylog('none', [f'[{pluginName}] ⚠ ERROR: "{path}" does not exist.'])
|
|
else:
|
|
# Detect file encoding
|
|
with open(path, 'rb') as f:
|
|
result = chardet.detect(f.read())
|
|
|
|
# Use the detected encoding
|
|
encoding = result['encoding']
|
|
|
|
# Order: MAC, IP, IsActive, NAME, Hardware
|
|
# Handle pihole-specific dhcp.leases files
|
|
if 'pihole' in path:
|
|
with open(path, 'r', encoding=encoding, errors='replace') as f:
|
|
for line in f:
|
|
row = line.rstrip().split()
|
|
if len(row) == 5:
|
|
plugin_objects.add_object(
|
|
primaryId = handleEmpty(row[1]),
|
|
secondaryId = handleEmpty(row[2]),
|
|
watched1 = handleEmpty('True'),
|
|
watched2 = handleEmpty(row[3]),
|
|
watched3 = handleEmpty(row[4]),
|
|
watched4 = handleEmpty('True'),
|
|
extra = handleEmpty(path),
|
|
foreignKey = handleEmpty(row[1])
|
|
)
|
|
elif 'dnsmasq' in path:
|
|
# [Lease expiry time] [mac address] [ip address] [hostname] [client id, if known]
|
|
# e.g.
|
|
# 1715932537 01:5c:5c:5c:5c:5c:5c 192.168.1.115 ryans-laptop 01:5c:5c:5c:5c:5c:5c
|
|
with open(path, 'r', encoding=encoding, errors='replace') as f:
|
|
for line in f:
|
|
row = line.rstrip().split()
|
|
if len(row) > 3:
|
|
plugin_objects.add_object(
|
|
primaryId = handleEmpty(row[1]),
|
|
secondaryId = handleEmpty(row[2]),
|
|
watched1 = handleEmpty('True'),
|
|
watched2 = handleEmpty(row[3]),
|
|
watched3 = '',
|
|
watched4 = handleEmpty('True'),
|
|
extra = handleEmpty(path),
|
|
foreignKey = handleEmpty(row[1])
|
|
)
|
|
else:
|
|
# Handle generic dhcp.leases files
|
|
leases = DhcpLeases(path)
|
|
leasesList = leases.get()
|
|
for lease in leasesList:
|
|
|
|
# filter out irrelevant entries (e.g. from OPNsense dhcp.leases files)
|
|
if is_mac(lease.ethernet):
|
|
|
|
plugin_objects.add_object(
|
|
primaryId = handleEmpty(lease.ethernet),
|
|
secondaryId = handleEmpty(lease.ip),
|
|
watched1 = handleEmpty(lease.active),
|
|
watched2 = handleEmpty(lease.hostname),
|
|
watched3 = handleEmpty(lease.hardware),
|
|
watched4 = handleEmpty(lease.binding_state),
|
|
extra = handleEmpty(path),
|
|
foreignKey = handleEmpty(lease.ethernet)
|
|
)
|
|
return plugin_objects
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|