diff options
Diffstat (limited to 'kvmd/helpers/otgmsd')
| -rw-r--r-- | kvmd/helpers/otgmsd/__init__.py | 20 | ||||
| -rw-r--r-- | kvmd/helpers/otgmsd/remount/__init__.py | 107 | ||||
| -rw-r--r-- | kvmd/helpers/otgmsd/remount/__main__.py | 24 | ||||
| -rw-r--r-- | kvmd/helpers/otgmsd/unlock/__init__.py | 58 | ||||
| -rw-r--r-- | kvmd/helpers/otgmsd/unlock/__main__.py | 24 |
5 files changed, 0 insertions, 233 deletions
diff --git a/kvmd/helpers/otgmsd/__init__.py b/kvmd/helpers/otgmsd/__init__.py deleted file mode 100644 index 4a6824c1..00000000 --- a/kvmd/helpers/otgmsd/__init__.py +++ /dev/null @@ -1,20 +0,0 @@ -# ========================================================================== # -# # -# KVMD - The main PiKVM daemon. # -# # -# Copyright (C) 2018-2022 Maxim Devaev <[email protected]> # -# # -# 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/>. # -# # -# ========================================================================== # diff --git a/kvmd/helpers/otgmsd/remount/__init__.py b/kvmd/helpers/otgmsd/remount/__init__.py deleted file mode 100644 index c66a5a17..00000000 --- a/kvmd/helpers/otgmsd/remount/__init__.py +++ /dev/null @@ -1,107 +0,0 @@ -# ========================================================================== # -# # -# KVMD - The main PiKVM daemon. # -# # -# Copyright (C) 2018-2022 Maxim Devaev <[email protected]> # -# # -# 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/>. # -# # -# ========================================================================== # - - -import sys -import os -import re -import shutil -import dataclasses -import subprocess - - -# ==== -_MOUNT_PATH = "/bin/mount" -_FSTAB_PATH = "/etc/fstab" - - -# ===== [email protected](frozen=True) -class _Storage: - mount_path: str - root_path: str - user: str - - -# ===== -def _log(msg: str) -> None: - print(msg, file=sys.stderr) - - -def _find_storage() -> _Storage: - with open(_FSTAB_PATH) as fstab_file: - for line in fstab_file.read().split("\n"): - line = line.strip() - if line and not line.startswith("#"): - parts = line.split() - if len(parts) == 6: - options = dict(re.findall(r"X-kvmd\.otgmsd-(root|user)=([^,]+)", parts[3])) - if options: - return _Storage( - mount_path=parts[1], - root_path=options.get("root", ""), - user=options.get("user", ""), - ) - raise RuntimeError(f"Can't find MSD mountpoint in {_FSTAB_PATH}") - - -def _remount(path: str, rw: bool) -> None: - mode = ("rw" if rw else "ro") - _log(f"Remounting {path} to {mode.upper()}-mode ...") - try: - subprocess.check_call([_MOUNT_PATH, "--options", f"remount,{mode}", path]) - except subprocess.CalledProcessError as err: - raise SystemExit(f"Can't remount: {err}") - - -def _mkdir(path: str) -> None: - if not os.path.exists(path): - _log(f"MKDIR --- {path}") - try: - os.mkdir(path) - except Exception as err: - raise SystemExit(f"Can't create directory: {err}") - - -def _chown(path: str, user: str) -> None: - _log(f"CHOWN --- {user} - {path}") - try: - shutil.chown(path, user) - except Exception as err: - raise SystemExit(f"Can't change ownership: {err}") - - -# ===== -def main() -> None: - if len(sys.argv) != 2 or sys.argv[1] not in ["ro", "rw"]: - raise SystemExit(f"Usage: {sys.argv[0]} [ro|rw]") - - rw = (sys.argv[1] == "rw") - - storage = _find_storage() - _remount(storage.mount_path, rw) - if rw: - if storage.root_path: - for name in ["images", "meta"]: - path = os.path.join(storage.root_path, name) - _mkdir(path) - if storage.user: - _chown(path, storage.user) diff --git a/kvmd/helpers/otgmsd/remount/__main__.py b/kvmd/helpers/otgmsd/remount/__main__.py deleted file mode 100644 index 3849d1b9..00000000 --- a/kvmd/helpers/otgmsd/remount/__main__.py +++ /dev/null @@ -1,24 +0,0 @@ -# ========================================================================== # -# # -# KVMD - The main PiKVM daemon. # -# # -# Copyright (C) 2018-2022 Maxim Devaev <[email protected]> # -# # -# 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 . import main -main() diff --git a/kvmd/helpers/otgmsd/unlock/__init__.py b/kvmd/helpers/otgmsd/unlock/__init__.py deleted file mode 100644 index 140e0e7c..00000000 --- a/kvmd/helpers/otgmsd/unlock/__init__.py +++ /dev/null @@ -1,58 +0,0 @@ -# ========================================================================== # -# # -# KVMD - The main PiKVM daemon. # -# # -# Copyright (C) 2018-2022 Maxim Devaev <[email protected]> # -# # -# 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/>. # -# # -# ========================================================================== # - - -import sys -import signal - -import psutil - - -# ===== -_PROCESS_NAME = "file-storage" - - -# ===== -def _log(msg: str) -> None: - print(msg, file=sys.stderr) - - -def _unlock() -> None: - # https://github.com/torvalds/linux/blob/3039fad/drivers/usb/gadget/function/f_mass_storage.c#L2924 - found = False - for proc in psutil.process_iter(): - attrs = proc.as_dict(attrs=["name", "exe", "pid"]) - if attrs.get("name") == _PROCESS_NAME and not attrs.get("exe"): - _log(f"Sending SIGUSR1 to MSD {_PROCESS_NAME!r} kernel thread with pid={attrs['pid']} ...") - try: - proc.send_signal(signal.SIGUSR1) - found = True - except Exception as err: - raise SystemExit(f"Can't send SIGUSR1 to MSD kernel thread with pid={attrs['pid']}: {err}") - if not found: - raise SystemExit(f"Can't find MSD kernel thread {_PROCESS_NAME!r}") - - -# ===== -def main() -> None: - if len(sys.argv) != 2 or sys.argv[1] != "unlock": - raise SystemExit(f"Usage: {sys.argv[0]} [unlock]") - _unlock() diff --git a/kvmd/helpers/otgmsd/unlock/__main__.py b/kvmd/helpers/otgmsd/unlock/__main__.py deleted file mode 100644 index 3849d1b9..00000000 --- a/kvmd/helpers/otgmsd/unlock/__main__.py +++ /dev/null @@ -1,24 +0,0 @@ -# ========================================================================== # -# # -# KVMD - The main PiKVM daemon. # -# # -# Copyright (C) 2018-2022 Maxim Devaev <[email protected]> # -# # -# 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 . import main -main() |