Initial test refactoring

tests/utils/colors.py

@@ -0,0 +1,17 @@
+pref = "\033["
+reset = f"{pref}0m"
+
+class colors:
+    black = "30m"
+    red = "31m"
+    green = "32m"
+    yellow = "33m"
+    blue = "34m"
+    magenta = "35m"
+    cyan = "36m"
+    white = "37m"
+
+def puts(text, *args, color=colors.white, is_bold=False, **kwargs):
+    print(f'{pref}{1 if is_bold else 0};{color}' + text + reset, *args, **kwargs)
+
+def sep(): puts("-----------------------------------", is_bold=True)

tests/utils/firegexapi.py

@@ -0,0 +1,201 @@
+from requests import Session
+
+def verify(req):
+    try:
+        assert(req.json()["status"] == "ok")
+    except Exception:
+        return False
+    return True
+
+class BearerSession():
+    def __init__(self):
+        self.s = Session()
+        self.headers = {}
+
+    def post(self, endpoint, json={}, data=""):
+        headers = self.headers
+        if data:
+            headers["Content-Type"] = "application/x-www-form-urlencoded"
+        return self.s.post(endpoint, json=json, data=data, headers=headers)
+
+    def get(self, endpoint, json={}):
+        return self.s.get(endpoint, json=json, headers=self.headers)
+    
+    def set_token(self,token):
+         self.headers = {"Authorization": f"Bearer {token}"}
+    
+    def unset_token(self):
+        self.headers = {}
+
+class FiregexAPI:
+    def __init__(self,address):
+        self.s = BearerSession()
+        self.address = address
+    
+    #General API
+    def status(self):
+        return self.s.get(f"{self.address}api/status").json()
+    
+    def login(self,password):
+        req = self.s.post(f"{self.address}api/login", data=f"username=login&password={password}")
+        try : 
+            self.s.set_token(req.json()["access_token"])
+            return True
+        except Exception:
+            return False
+
+    def logout(self):
+        self.s.unset_token()
+        return True
+
+    def set_password(self,password):
+        req = self.s.post(f"{self.address}api/set-password", json={"password":password})
+        if verify(req):
+            self.s.set_token(req.json()["access_token"])
+            return True
+        else:
+            return False
+
+    def change_password(self,password,expire):
+        req = self.s.post(f"{self.address}api/change-password", json={"password":password, "expire":expire})
+        if verify(req):
+            self.s.set_token(req.json()["access_token"])
+            return True
+        else:
+            return False
+
+    def get_interfaces(self):
+        req = self.s.get(f"{self.address}api/interfaces")
+        return req.json()
+
+    def reset(self, delete):
+        req = self.s.post(f"{self.address}api/reset", json={"delete":delete})
+
+    #Netfilter regex
+    def nf_get_stats():
+        req = self.s.get(f"{self.address}api/nfregex/stats")
+        return req.json()
+
+    def nf_get_services(self):
+        req = self.s.get(f"{self.address}api/nfregex/services")
+        return req.json() 
+
+    def nf_get_service(self,service_id):
+        req = self.s.get(f"{self.address}api/nfregex/service/{service_id}")
+        return req.json()
+
+    def nf_stop_service(self,service_id):
+        req = self.s.get(f"{self.address}api/nfregex/service/{service_id}/stop")
+        return verify(req)
+    
+    def nf_start_service(self,service_id):
+        req = self.s.get(f"{self.address}api/nfregex/service/{service_id}/start")
+        return verify(req)
+
+    def nf_delete_service(self,service_id):
+        req = self.s.get(f"{self.address}api/nfregex/service/{service_id}/delete")
+        return verify(req)
+
+    def nf_rename_service(self,newname):
+        req = self.s.post(f"{self.address}api/nfregex/services/add" , json={"name":service_name,"port":service_port, "ip_int": ip_int, "proto": proto})
+        return verify(req)
+
+    def nf_get_service_regexes(self,service_id):
+        req = self.s.get(f"{self.address}api/nfregex/service/{service_id}/regexes")
+        return req.json()
+
+    def nf_get_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/nfregex/regex/{regex_id}")
+        return req.json()
+    
+    def nf_delete_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/nfregex/regex/{regex_id}/delete")
+        return verify(req)
+    
+    def nf_enable_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/nfregex/regex/{regex_id}/enable")
+        return verify(req)
+
+    def nf_disable_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/nfregex/regex/{regex_id}/disable")
+        return verify(req)
+
+    def nf_add_regex(self, service_id: str, regex: str, mode: str, active: bool, is_blacklist: bool, is_case_sensitive: bool):
+        req = self.s.post(f"{self.address}api/nfregex/regexes/add", 
+            json={"service_id": service_id, "regex": regex, "mode": mode, "active": active, "is_blacklist": is_blacklist, "is_case_sensitive": is_case_sensitive})
+        return verify(req)
+
+    def nf_add_service(self, name: str, port: int, proto: str, ip_int: str):
+        req = self.s.post(f"{self.address}api/nfregex/services/add" , 
+            json={"name":name,"port":port, "proto": proto, "ip_int": ip_int})
+        return req.json()["service_id"] if verify(req) else False 
+
+    #Proxy regex
+    def px_get_stats():
+        req = self.s.get(f"{self.address}api/regexproxy/stats")
+        return req.json() 
+
+    def px_get_services(self):
+        req = self.s.get(f"{self.address}api/regexproxy/services")
+        return req.json()
+
+    def px_get_service(self,service_id):
+        req = self.s.get(f"{self.address}api/regexproxy/service/{service_id}")
+        return req.json()
+
+    def px_stop_service(self,service_id):
+        req = self.s.get(f"{self.address}api/regexproxy/service/{service_id}/stop")
+        return verify(req)
+    
+    def px_pause_service(self,service_id):
+        req = self.s.get(f"{self.address}api/regexproxy/service/{service_id}/pause")
+        return verify(req)
+
+    def px_start_service(self,service_id):
+        req = self.s.get(f"{self.address}api/regexproxy/service/{service_id}/start")
+        return verify(req)
+
+    def px_delete_service(self,service_id):
+        req = self.s.get(f"{self.address}api/regexproxy/service/{service_id}/delete")
+        return verify(req)
+
+    def px_change_service_port(self,service_id, port, internalPort):
+        payload = {}
+        if port: payload["port"] = port
+        if internalPort: payload["internalPort"] = internalPort
+        req = self.s.post(f"{self.address}api/regexproxy/service/{service_id}/start", json=payload)
+        return req.json() if verify(req) else False
+
+    def px_get_service_regexes(self,service_id):
+        req = self.s.get(f"{self.address}api/regexproxy/service/{service_id}/regexes")
+        return req.json() 
+
+    def px_get_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/regexproxy/regex/{regex_id}")
+        return req.json() 
+    
+    def px_delete_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/regexproxy/regex/{regex_id}/delete")
+        return verify(req)
+    
+    def px_enable_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/regexproxy/regex/{regex_id}/enable")
+        return verify(req)
+
+    def px_disable_regex(self,regex_id):
+        req = self.s.get(f"{self.address}api/regexproxy/regex/{regex_id}/disable")
+        return verify(req)
+
+    def px_add_regex(self, service_id: str, regex: str, mode: str, active: bool, is_blacklist: bool, is_case_sensitive: bool):
+        req = self.s.post(f"{self.address}api/regexproxy/regexes/add", 
+            json={"service_id": service_id, "regex": regex, "mode": mode, "active": active, "is_blacklist": is_blacklist, "is_case_sensitive": is_case_sensitive})
+        return verify(req)
+
+    def px_rename_service(self,newname):
+        req = self.s.post(f"{self.address}api/regexproxy/services/add" , json={"name":service_name,"port":service_port, "ip_int": ip_int, "proto": proto})
+        return verify(req)
+
+    def px_add_service(self, name: str, port: int, internalPort: [int,None]):
+        req = self.s.post(f"{self.address}api/regexproxy/services/add" , 
+            json={"name":name,"port":port, "internalPort": internalPort})
+        return req.json()["service_id"] if verify(req) else False 

tests/utils/tcpserver.py

@@ -0,0 +1,31 @@
+from multiprocessing import Process
+import socket
+
+class TcpServer:
+    def __init__(self,port):
+        def _startServer(port):
+            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+            sock.bind(('127.0.0.1', port))
+            sock.listen(8)
+            while True:  
+                connection,address = sock.accept()  
+                buf = connection.recv(4096)  
+                connection.send(buf)    		
+                connection.close()
+        self.port = port
+        self.server = Process(target=_startServer,args=[port])
+
+    def start(self):
+        self.server.start()
+    
+    def stop(self):
+        self.server.terminate()
+
+    def sendCheckData(self,data):
+        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        s.connect(('localhost', self.port))
+        s.sendall(data)
+        received_data = s.recv(4096)
+        s.close()
+        return received_data == data