mirror of
https://github.com/sqall01/LSMS.git
synced 2024-11-19 02:20:33 +01:00
86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
|
import os
|
||
|
|
import socket
|
||
|
|
|
||
|
|
from .alerts import raise_alert_alertr, raise_alert_mail
|
||
|
|
|
||
|
|
try:
|
||
|
|
from config.config import ALERTR_FIFO, FROM_ADDR, TO_ADDR, STATE_DIR
|
||
|
|
|
||
|
|
except:
|
||
|
|
ALERTR_FIFO = None
|
||
|
|
FROM_ADDR = None
|
||
|
|
TO_ADDR = None
|
||
|
|
|
||
|
|
|
||
|
|
def output_error(file_name: str, msg: str):
|
||
|
|
|
||
|
|
base_name = os.path.basename(file_name)
|
||
|
|
|
||
|
|
# Decide where to output results.
|
||
|
|
print_output = False
|
||
|
|
if ALERTR_FIFO is None and FROM_ADDR is None and TO_ADDR is None:
|
||
|
|
print_output = True
|
||
|
|
|
||
|
|
if print_output:
|
||
|
|
message = "#" * 80
|
||
|
|
message += "\nError in '%s':\n%s" % (base_name, msg)
|
||
|
|
print(message)
|
||
|
|
|
||
|
|
else:
|
||
|
|
hostname = socket.gethostname()
|
||
|
|
message = "Error in '%s' on host '%s':\n%s" \
|
||
|
|
% (base_name, hostname, msg)
|
||
|
|
|
||
|
|
if ALERTR_FIFO:
|
||
|
|
optional_data = dict()
|
||
|
|
optional_data["error"] = True
|
||
|
|
optional_data["script"] = base_name
|
||
|
|
optional_data["message"] = message
|
||
|
|
|
||
|
|
raise_alert_alertr(ALERTR_FIFO,
|
||
|
|
optional_data)
|
||
|
|
|
||
|
|
if FROM_ADDR is not None and TO_ADDR is not None:
|
||
|
|
mail_subject = "[Security] Error in '%s' on host '%s'" % (base_name, socket.gethostname())
|
||
|
|
raise_alert_mail(FROM_ADDR,
|
||
|
|
TO_ADDR,
|
||
|
|
mail_subject,
|
||
|
|
message)
|
||
|
|
|
||
|
|
|
||
|
|
def output_finding(file_name: str, msg: str):
|
||
|
|
|
||
|
|
base_name = os.path.basename(file_name)
|
||
|
|
|
||
|
|
# Decide where to output results.
|
||
|
|
print_output = False
|
||
|
|
if ALERTR_FIFO is None and FROM_ADDR is None and TO_ADDR is None:
|
||
|
|
print_output = True
|
||
|
|
|
||
|
|
if print_output:
|
||
|
|
message = "#" * 80
|
||
|
|
message += "\nFinding in '%s':\n%s" % (base_name, msg)
|
||
|
|
|
||
|
|
print(message)
|
||
|
|
|
||
|
|
else:
|
||
|
|
hostname = socket.gethostname()
|
||
|
|
message = "Finding in '%s' on host '%s':\n%s" \
|
||
|
|
% (base_name, hostname, msg)
|
||
|
|
|
||
|
|
if ALERTR_FIFO:
|
||
|
|
optional_data = dict()
|
||
|
|
optional_data["finding"] = True
|
||
|
|
optional_data["script"] = base_name
|
||
|
|
optional_data["message"] = message
|
||
|
|
|
||
|
|
raise_alert_alertr(ALERTR_FIFO,
|
||
|
|
optional_data)
|
||
|
|
|
||
|
|
if FROM_ADDR is not None and TO_ADDR is not None:
|
||
|
|
mail_subject = "[Security] Finding in '%s' on host '%s'" % (base_name, socket.gethostname())
|
||
|
|
raise_alert_mail(FROM_ADDR,
|
||
|
|
TO_ADDR,
|
||
|
|
mail_subject,
|
||
|
|
message)
|