kvmd-otgnet

kvmd/apps/otgnet/netctl.py

@@ -0,0 +1,92 @@
+# ========================================================================== #
+#                                                                            #
+#    KVMD - The main Pi-KVM daemon.                                          #
+#                                                                            #
+#    Copyright (C) 2018  Maxim Devaev <mdevaev@gmail.com>                    #
+#                                                                            #
+#    This program is free software: you can redistribute it and/or modify    #
+#    it under the terms of the GNU General Public License as published by    #
+#    the Free Software Foundation, either version 3 of the License, or       #
+#    (at your option) any later version.                                     #
+#                                                                            #
+#    This program is distributed in the hope that it will be useful,         #
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of          #
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           #
+#    GNU General Public License for more details.                            #
+#                                                                            #
+#    You should have received a copy of the GNU General Public License       #
+#    along with this program.  If not, see <https://www.gnu.org/licenses/>.  #
+#                                                                            #
+# ========================================================================== #
+
+
+from typing import List
+from typing import Dict
+
+
+# =====
+class BaseCtl:
+    def get_command(self, direct: bool) -> List[str]:
+        raise NotImplementedError
+
+
+class IfaceUpCtl(BaseCtl):
+    def __init__(self, base_cmd: List[str], iface: str) -> None:
+        self.__base_cmd = base_cmd
+        self.__iface = iface
+
+    def get_command(self, direct: bool) -> List[str]:
+        return [*self.__base_cmd, "link", "set", self.__iface, ("up" if direct else "down")]
+
+
+class IfaceAddIpCtl(BaseCtl):
+    def __init__(self, base_cmd: List[str], iface: str, cidr: str) -> None:
+        self.__base_cmd = base_cmd
+        self.__iface = iface
+        self.__cidr = cidr
+
+    def get_command(self, direct: bool) -> List[str]:
+        return [*self.__base_cmd, "address", ("add" if direct else "del"), self.__cidr, "dev", self.__iface]
+
+
+class IptablesDropAllCtl(BaseCtl):
+    def __init__(self, base_cmd: List[str], iface: str) -> None:
+        self.__base_cmd = base_cmd
+        self.__iface = iface
+
+    def get_command(self, direct: bool) -> List[str]:
+        return [*self.__base_cmd, ("-A" if direct else "-D"), "INPUT", "-i", self.__iface, "-j", "DROP"]
+
+
+class IptablesAllowPortCtl(BaseCtl):
+    def __init__(self, base_cmd: List[str], iface: str, port: int, tcp: bool) -> None:
+        self.__base_cmd = base_cmd
+        self.__iface = iface
+        self.__port = port
+        self.__proto = ("tcp" if tcp else "udp")
+
+    def get_command(self, direct: bool) -> List[str]:
+        return [
+            *self.__base_cmd,
+            ("-A" if direct else "-D"), "INPUT", "-i", self.__iface, "-p", self.__proto,
+            "--dport", str(self.__port), "-j", "ACCEPT",
+        ]
+
+
+class CustomCtl(BaseCtl):
+    def __init__(
+        self,
+        direct_cmd: List[str],
+        reverse_cmd: List[str],
+        placeholders: Dict[str, str],
+    ) -> None:
+
+        self.__direct_cmd = direct_cmd
+        self.__reverse_cmd = reverse_cmd
+        self.__placeholders = placeholders
+
+    def get_command(self, direct: bool) -> List[str]:
+        return [
+            part.format(**self.__placeholders)
+            for part in (self.__direct_cmd if direct else self.__reverse_cmd)
+        ]