GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2026-02-01 18:41:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

improved otg msd

Browse Source
This commit is contained in:
Devaev Maxim
2019-10-21 03:40:58 +03:00
parent 7ee3d585e8
commit cae86d9302
5 changed files with 122 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
kvmd/helpers/otgmsd/unlock/__init__.py
View File

@@ -31,12 +31,17 @@ _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"])
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
@@ -49,6 +54,5 @@ def _unlock() -> None:
# =====
def main() -> None:
if len(sys.argv) != 2 or sys.argv[1] != "unlock":
raise SystemExit(f"This program interrupts all IO operations performed by OTG MSD.\n\n"
f"Usage: python -m kvmd.helpers.otgmsd.unlock [-h|--help|unlock]")
raise SystemExit(f"Usage: {sys.argv[0]} [unlock]")
_unlock()