Password Authentication

backend/proxy/__init__.py

@@ -0,0 +1,87 @@
+import subprocess, re
+
+#c++ -o proxy proxy.cpp
+
+class Filter:
+    def __init__(self, regex, is_blacklist=True, c_to_s=False, s_to_c=False ):
+        self.regex = regex
+        self.is_blacklist = is_blacklist
+        if c_to_s == s_to_c: c_to_s = s_to_c = True # (False, False) == (True, True)
+        self.c_to_s = c_to_s
+        self.s_to_c = s_to_c
+        self.blocked = 0
+    
+    def compile(self):
+        if isinstance(self.regex, str): self.regex = self.regex.encode()
+        if not isinstance(self.regex, bytes): raise Exception("Invalid Regex Paramether")
+        re.compile(self.regex) # raise re.error if is invalid!
+        if self.c_to_s:
+            yield "C"+self.regex.hex() if self.is_blacklist else "c"+self.regex.hex()
+        if self.s_to_c:
+            yield "S"+self.regex.hex() if self.is_blacklist else "s"+self.regex.hex()
+
+
+
+class Proxy:
+    def __init__(self, internal_port, public_port, filters=None, public_host="0.0.0.0", internal_host="127.0.0.1"):
+        self.public_host = public_host
+        self.public_port = public_port
+        self.internal_host = internal_host
+        self.internal_port = internal_port
+        self.filters = set(filters) if filters else set([])
+        self.process = None
+    
+    def start(self, callback=None):
+        if self.process is None:
+            filter_map = self.compile_filters()
+            filters_codes = list(filter_map.keys())
+            self.process = subprocess.Popen(
+                ["./proxy", str(self.public_host), str(self.public_port), str(self.internal_host), str(self.internal_port), *filters_codes],
+                stdout=subprocess.PIPE, universal_newlines=True
+            )
+            for stdout_line in iter(self.process.stdout.readline, ""):
+                if stdout_line.startswith("BLOCKED"):
+                    regex_id = stdout_line.split()[1]
+                    filter_map[regex_id].blocked+=1
+                    if callback: callback(filter_map[regex_id])
+            self.process.stdout.close()
+            return self.process.wait()
+    
+    def stop(self):
+        if self.process:
+            self.process.terminate()
+            try:
+                self.process.wait(timeout=3)
+                return True
+            except Exception:
+                self.process.kill()
+                return False
+            finally:
+                self.process = None
+        return True
+
+    def restart(self):
+        status = self.stop()
+        self.start()
+        return status
+    
+    def reload(self):
+        if self.process: self.restart()
+
+    def compile_filters(self):
+        res = {}
+        for filter_obj in self.filters:
+            raw_filters = filter_obj.compile()
+            for filter in raw_filters:
+                res[filter] = filter_obj
+        return res
+
+    def add_filter(self, filter):
+        self.filters.add(filter)
+        self.reload()
+
+    def remove_filter(self, filter):
+        try:
+            del self.filters[self.filters.remove(filter)]
+        except ValueError: return
+        self.reload()