mirror of
https://github.com/sqall01/LSMS.git
synced 2024-11-17 01:28:25 +01:00
158 lines
4.1 KiB
Python
Executable file
158 lines
4.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# written by sqall
|
|
# twitter: https://twitter.com/sqall01
|
|
# blog: https://h4des.org
|
|
# github: https://github.com/sqall01
|
|
#
|
|
# Licensed under the MIT License.
|
|
|
|
"""
|
|
Short summary:
|
|
Monitor /etc/hosts for changes to detect malicious attempts to divert traffic.
|
|
|
|
Requirements:
|
|
None
|
|
"""
|
|
|
|
import os
|
|
from typing import Dict, Set
|
|
|
|
from lib.state import load_state, store_state
|
|
from lib.util import output_error, output_finding
|
|
|
|
# Read configuration.
|
|
try:
|
|
from config.config import ALERTR_FIFO, FROM_ADDR, TO_ADDR, STATE_DIR
|
|
from config.monitor_hosts_file import ACTIVATED
|
|
STATE_DIR = os.path.join(os.path.dirname(__file__), STATE_DIR, os.path.basename(__file__))
|
|
except:
|
|
ALERTR_FIFO = None
|
|
FROM_ADDR = None
|
|
TO_ADDR = None
|
|
ACTIVATED = True
|
|
STATE_DIR = os.path.join("/tmp", os.path.basename(__file__))
|
|
|
|
|
|
class MonitorHostsException(Exception):
|
|
def __init__(self, msg: str):
|
|
self._msg = msg
|
|
|
|
def __str__(self):
|
|
return self._msg
|
|
|
|
|
|
def _get_hosts() -> Dict[str, Set[str]]:
|
|
|
|
hosts_data = {}
|
|
with open("/etc/hosts", 'rt') as fp:
|
|
for line in fp:
|
|
line = line.strip()
|
|
|
|
if line == "":
|
|
continue
|
|
|
|
# Ignore comments.
|
|
if line[0] == "#":
|
|
continue
|
|
|
|
entry = line.split()
|
|
if len(entry) < 2:
|
|
raise MonitorHostsException("Not able to parse line: %s" % line)
|
|
|
|
ip = entry[0]
|
|
hosts = set(entry[1:])
|
|
if ip not in hosts_data.keys():
|
|
hosts_data[ip] = hosts
|
|
|
|
else:
|
|
for host in hosts:
|
|
hosts_data[ip].add(host)
|
|
|
|
return hosts_data
|
|
|
|
|
|
def monitor_hosts():
|
|
|
|
# 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 not ACTIVATED:
|
|
if print_output:
|
|
print("Module deactivated.")
|
|
return
|
|
|
|
stored_hosts_data = {}
|
|
try:
|
|
state_data = load_state(STATE_DIR)
|
|
|
|
# Convert list to set.
|
|
for k, v in state_data.items():
|
|
stored_hosts_data[k] = set(v)
|
|
|
|
except Exception as e:
|
|
output_error(__file__, str(e))
|
|
return
|
|
|
|
curr_hosts_data = {}
|
|
try:
|
|
curr_hosts_data = _get_hosts()
|
|
|
|
except Exception as e:
|
|
output_error(__file__, str(e))
|
|
return
|
|
|
|
# Compare stored data with current one.
|
|
for stored_entry_ip in stored_hosts_data.keys():
|
|
|
|
# Extract current entry belonging to the same ip.
|
|
if stored_entry_ip not in curr_hosts_data.keys():
|
|
message = "Host name for IP '%s' was deleted." % stored_entry_ip
|
|
|
|
output_finding(__file__, message)
|
|
|
|
continue
|
|
|
|
# Check host entry was removed.
|
|
for host in stored_hosts_data[stored_entry_ip]:
|
|
if host not in curr_hosts_data[stored_entry_ip]:
|
|
message = "Host name entry for IP '%s' was removed.\n\n" % stored_entry_ip
|
|
message += "Entry: %s" % host
|
|
|
|
output_finding(__file__, message)
|
|
|
|
# Check host entry was added.
|
|
for host in curr_hosts_data[stored_entry_ip]:
|
|
if host not in stored_hosts_data[stored_entry_ip]:
|
|
message = "Host name entry for IP '%s' was added.\n\n" % stored_entry_ip
|
|
message += "Entry: %s" % host
|
|
|
|
output_finding(__file__, message)
|
|
|
|
# Check new data was added.
|
|
for curr_entry_ip in curr_hosts_data.keys():
|
|
if curr_entry_ip not in stored_hosts_data.keys():
|
|
message = "New host name was added for IP '%s'.\n\n" % curr_entry_ip
|
|
message += "Entries:\n"
|
|
for host in curr_hosts_data[curr_entry_ip]:
|
|
message += host
|
|
message += "\n"
|
|
|
|
output_finding(__file__, message)
|
|
|
|
try:
|
|
# Convert set to list.
|
|
state_data = {}
|
|
for k, v in curr_hosts_data.items():
|
|
state_data[k] = list(v)
|
|
|
|
store_state(STATE_DIR, state_data)
|
|
|
|
except Exception as e:
|
|
output_error(__file__, str(e))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
monitor_hosts()
|