94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
def main():
|
|
|
|
import socket
|
|
import os
|
|
import config
|
|
import datetime
|
|
import signal
|
|
import threading
|
|
from log import logsManager
|
|
import sys
|
|
import select
|
|
|
|
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)
|
|
|
|
stop_event = threading.Event()
|
|
dataBuffer = []
|
|
|
|
def publishData(stop_event):
|
|
|
|
if config.loggingMode = 'rabbitmq':
|
|
|
|
import rabbitmq
|
|
rabbitmq = rabbitmq.rabbitMQClient(config.rabbitmqca,config.rabbitmqcacert,config.rabbitmqcakey,config.rabbitmqHost,config.rabbitmqPort,config.rabbitmqRoutingKey)
|
|
|
|
while not (stop_event.is_set()):
|
|
if dataBuffer:
|
|
data = dataBuffer.pop(0)
|
|
|
|
if config.loggingMode == 'rabbitmq':
|
|
rabbitmq.publish(f"{data}")
|
|
else:
|
|
print("Not yet implemented")
|
|
|
|
def cleanup(signum, frame):
|
|
|
|
print("Signal received, shutting down...")
|
|
stop_event.set()
|
|
|
|
signal.signal(signal.SIGTERM, cleanup)
|
|
signal.signal(signal.SIGINT, cleanup)
|
|
|
|
def server(stop_event):
|
|
|
|
serverSocket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
serverSocket.bind(socketFile)
|
|
serverSocket.listen(50)
|
|
|
|
while not (stop_event.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()
|
|
|
|
publishThread = threading.Thread(target=publishData, args=(stop_event,))
|
|
serverThread = threading.Thread(target=server, args=(stop_event,))
|
|
|
|
publishThread.start()
|
|
serverThread.start()
|
|
|
|
publishThread.join()
|
|
serverThread.join()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|