IP Prüfung für XML-RPC begonnen. Prüft lediglich auf IP und noch keine Level

revpipyload/xrpcserver.py

@@ -0,0 +1,88 @@
+# -*- coding: utf-8 -*-
+#
+# RevPiPyLoad
+#
+# Webpage: https://revpimodio.org/revpipyplc/
+# (c) Sven Sager, License: LGPLv3
+#
+"""XML-RPC Server anpassungen fuer Absicherung."""
+from helper import IpAclManager
+from concurrent import futures
+from xmlrpc.server import SimpleXMLRPCServer, SimpleXMLRPCRequestHandler
+
+
+class SaveXMLRPCServer(SimpleXMLRPCServer):
+
+    """Erstellt einen erweiterten XMLRPCServer."""
+
+    aclmgr = IpAclManager()
+
+    def __init__(
+            self, addr, logRequests=True, allow_none=False,
+            use_builtin_types=False, acl=""):
+        """Init SaveXMLRPCServer class."""
+
+        SaveXMLRPCServer.aclmgr.acl = acl
+
+        # Vererbte Klasse instantiieren
+        super().__init__(
+            addr=addr,
+            requestHandler=SaveXMLRPCRequestHandler,
+            logRequests=logRequests,
+            allow_none=allow_none,
+            encoding="utf-8",
+            bind_and_activate=False,
+            use_builtin_types=use_builtin_types
+        )
+
+        # Klassenvariablen
+        self.tpe = futures.ThreadPoolExecutor(max_workers=1)
+        self.fut = None
+
+    def isAlive(self):
+        """Prueft ob der XML RPC Server laeuft.
+        @return True, wenn Server noch laeuft"""
+        return False if self.fut is None else self.fut.running()
+
+    def start(self):
+        """Startet den XML-RPC Server."""
+        if self.fut is None:
+            self.server_bind()
+            self.server_activate()
+            self.fut = self.tpe.submit(self.serve_forever)
+        else:
+            raise RuntimeError("savexmlrpcservers can only be started once")
+
+    def stop(self):
+        """Stoppt den XML-RPC Server."""
+        if self.fut is not None:
+            self.shutdown()
+            self.tpe.shutdown()
+            self.server_close()
+        else:
+            raise RuntimeError("save xml rpc server was not started")
+
+
+class SaveXMLRPCRequestHandler(SimpleXMLRPCRequestHandler):
+
+    """Verwaltet die XML-Requests und prueft Berechtigungen."""
+
+    def parse_request(self):
+        """Berechtigungen pruefen.
+        @return True, wenn Parsen erfolgreich war"""
+        # Request parsen und ggf. schon abbrechen
+        if not super().parse_request():
+            return False
+
+        # IP-Adresse prüfen
+        int_acl = SaveXMLRPCServer.aclmgr.get_acllevel(self.address_string())
+        if int_acl >= 0:
+            return True
+        else:
+            self.send_error(
+                401,
+                "IP '{}' not allowed with acl level '{}'"
+                "".format(self.address_string(), int_acl)
+            )
+
+        return False