From 2a731001cc3bb52da74f9ed21318d11c724e1de9 Mon Sep 17 00:00:00 2001
From: Nathanael Sensfelder <SpamShield0@MultiAgentSystems.org>
Date: Mon, 19 Jun 2017 16:07:36 +0200
Subject: Initial commit.

---
 jabberhive-regex.py | 142 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 142 insertions(+)
 create mode 100755 jabberhive-regex.py

diff --git a/jabberhive-regex.py b/jabberhive-regex.py
new file mode 100755
index 0000000..0d012c4
--- /dev/null
+++ b/jabberhive-regex.py
@@ -0,0 +1,142 @@
+#!/bin/env python3
+import argparse
+import re
+
+import socket
+import _thread
+
+class ClientState:
+    CLIENT_IS_SENDING_DOWNSTREAM = 1
+    CLIENT_IS_SENDING_UPSTREAM = 2
+    CLIENT_IS_CONNECTING = 3
+    CLIENT_IS_TERMINATING = 4
+
+def client_main (source, params):
+
+    pattern = re.compile(params.regex)
+
+    state = ClientState.CLIENT_IS_CONNECTING
+    source_f = source.makefile(
+        mode='r',
+        buffering=True,
+        encoding="UTF-8",
+        newline='\n'
+    )
+
+    t_connect = None
+    f_connect = None
+    current_target = None
+
+    while True:
+        if (state == ClientState.CLIENT_IS_SENDING_DOWNSTREAM):
+            up_data = source_f.readline()
+
+            if (pattern.match(up_data)):
+                if (t_connect != None):
+                    t_connect.send(up_data.encode())
+                    current_target = 't'
+                    state = ClientState.CLIENT_IS_SENDING_UPSTREAM
+                    print("[Matched] Sending upstream...")
+                else:
+                    source.send("!P \n".encode())
+                    print("Matched")
+            else:
+                if (f_connect != None):
+                    f_connect.send(up_data.encode())
+                    current_target = 'f'
+                    state = ClientState.CLIENT_IS_SENDING_UPSTREAM
+                    print("[No match] Sending upstream...")
+                else:
+                    source.send("!P \n".encode())
+                    print("Did not match")
+        elif (state == ClientState.CLIENT_IS_SENDING_UPSTREAM):
+            matched = 0
+            c = b"\0"
+
+            while (c != b"\n"):
+                if (current_target == 't'):
+                    c = t_connect.recv(1)
+                else:
+                    c = f_connect.recv(1)
+
+                source.send(c)
+
+                if ((matched == 0) and (c == b"!")):
+                    matched = 1
+                elif ((matched == 1) and ((c == b"P") or (c == b"N"))):
+                    matched = 2
+                elif ((matched == 2) and (c == b" ")):
+                    print("Sending downstream...")
+                    state = ClientState.CLIENT_IS_SENDING_DOWNSTREAM
+                else:
+                    matched = -1
+
+        elif (state == ClientState.CLIENT_IS_CONNECTING):
+            print("Connecting to downstream...")
+            if (params.destination_true != None):
+                t_connect = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+                t_connect.connect(params.destination_true)
+
+            if (params.destination_false != None):
+                f_connect = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+                f_connect.connect(params.destination_false)
+
+            print("Sending downstream...")
+            state = ClientState.CLIENT_IS_SENDING_DOWNSTREAM
+        else:
+            break
+
+################################################################################
+## ARGUMENTS HANDLING ##########################################################
+################################################################################
+
+parser = argparse.ArgumentParser(
+    description = (
+        "Generates a list of instructions to construct the Structural Level."
+    )
+)
+
+parser.add_argument(
+    '-s',
+    '--socket-name',
+    type = str,
+    required = True,
+    help = 'Name of the UNIX socket for this filter.'
+)
+
+parser.add_argument(
+    '-t',
+    '--destination-true',
+    type = str,
+    help = 'UNIX socket this filter sends to when a match is found.',
+)
+
+parser.add_argument(
+    '-f',
+    '--destination-false',
+    type = str,
+    help = 'UNIX socket this filter sends to when a match is found.',
+)
+
+parser.add_argument(
+    '-r',
+    '--regex',
+    type = str,
+    required = True,
+    help = 'The regex to test the message against.',
+)
+
+args = parser.parse_args()
+
+################################################################################
+## MAIN ########################################################################
+################################################################################
+server_socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+
+server_socket.bind(args.socket_name)
+server_socket.listen(5)
+
+while True:
+    (client, client_address) = server_socket.accept()
+    _thread.start_new_thread(client_main, (client, args))
+
-- 
cgit v1.2.3-70-g09d2