123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
def main():
|
|
|
|
import socket
|
|
import os
|
|
import config
|
|
import datetime
|
|
import signal
|
|
import threading
|
|
import log
|
|
import sys
|
|
import select
|
|
import time
|
|
|
|
socketFile = '/etc/monutil/ip.socket'
|
|
|
|
if len(sys.argv) == 4:
|
|
|
|
hostname = socket.gethostname()
|
|
ipAddress = socket.gethostbyname(hostname)
|
|
blockedIPAddress = sys.argv[1]
|
|
jail = sys.argv[2]
|
|
live = int(sys.argv[3])
|
|
logTime = datetime.datetime.now()
|
|
|
|
clientSocket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
clientSocket.connect(socketFile)
|
|
data = f"{hostname}|{ipAddress}|{logTime}|ipBlock|{blockedIPAddress}|{jail}|{live}"
|
|
clientSocket.send(data.encode('utf-8'))
|
|
clientSocket.close()
|
|
sys.exit(0)
|
|
|
|
else:
|
|
|
|
if os.path.exists(socketFile):
|
|
os.remove(socketFile)
|
|
|
|
stopEvent = threading.Event()
|
|
failureEvent = threading.Event()
|
|
|
|
dataBuffer = []
|
|
|
|
def publishData(stopEvent):
|
|
|
|
try:
|
|
|
|
if config.loggingMode == 'rabbitmq':
|
|
|
|
import rabbitmq
|
|
rabbitmq = rabbitmq.rabbitMQClient(config.rabbitmqca,config.rabbitmqcacert,config.rabbitmqcakey,config.rabbitmqHost,config.rabbitmqPort,config.rabbitmqRoutingKey)
|
|
|
|
while not (stopEvent.is_set() and not failureEvent.is_set()):
|
|
|
|
time.sleep(1)
|
|
|
|
if dataBuffer:
|
|
|
|
data = dataBuffer.pop(0)
|
|
|
|
if config.loggingMode == 'rabbitmq':
|
|
|
|
rabbitmq.publish(f"{data}")
|
|
|
|
else:
|
|
|
|
hostname = socket.gethostname()
|
|
ipAddress = socket.gethostbyname(hostname)
|
|
data = data.split('|')
|
|
manager = log.logsManager(config.sqlServer, config.sqlDatabase, config.sqlUsername, config.sqlPassword)
|
|
manager.insertIPBlock(hostname, ipAddress, data[1], data[5], data[6], data[2], config.ipinfoAPIToken)
|
|
|
|
except Exception:
|
|
failureEvent.set()
|
|
|
|
def cleanup(signum, frame):
|
|
|
|
print("Signal received, shutting down...")
|
|
stopEvent.set()
|
|
|
|
signal.signal(signal.SIGTERM, cleanup)
|
|
signal.signal(signal.SIGINT, cleanup)
|
|
|
|
def server(stopEvent):
|
|
|
|
try:
|
|
|
|
serverSocket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
serverSocket.bind(socketFile)
|
|
serverSocket.listen(50)
|
|
|
|
while not (stopEvent.is_set() and not failureEvent.is_set()):
|
|
|
|
readable, _, _ = select.select([serverSocket], [], [], 1.0)
|
|
|
|
if readable:
|
|
clientSocket, _ = serverSocket.accept()
|
|
data = clientSocket.recv(1024).decode('utf-8')
|
|
|
|
if data:
|
|
|
|
dataBuffer.append(data)
|
|
print(data)
|
|
|
|
clientSocket.close()
|
|
|
|
except Exception:
|
|
|
|
failureEvent.set()
|
|
|
|
publishThread = threading.Thread(target=publishData, args=(stopEvent,))
|
|
serverThread = threading.Thread(target=server, args=(stopEvent,))
|
|
|
|
publishThread.start()
|
|
serverThread.start()
|
|
|
|
publishThread.join()
|
|
serverThread.join()
|
|
|
|
if failureEvent.is_set():
|
|
print("One of the threads failed. Terminating")
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|