GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2025-12-12 17:20:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

kvmd-otgconf: reset delay

Browse Source
This commit is contained in:
Maxim Devaev 2022-04-01 06:51:57 +03:00
parent 3b56100fe2
commit 7ba7f08be7
1 changed files with 45 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
kvmd/apps/otgconf/__init__.py
View File

@ -24,6 +24,7 @@ import os
import json import json
import contextlib import contextlib
import argparse import argparse
import time
from typing import List from typing import List
from typing import Generator from typing import Generator
@ -37,10 +38,17 @@ from .. import init
# ===== # =====
@contextlib.contextmanager class _GadgetControl:
def _udc_stopped(gadget: str, udc: str) -> Generator[None, None, None]: def __init__(self, meta_path: str, gadget: str, udc: str, init_delay: float) -> None:
udc = usb.find_udc(udc) self.__meta_path = meta_path
udc_path = usb.get_gadget_path(gadget, usb.G_UDC) self.__gadget = gadget
self.__udc = udc
self.__init_delay = init_delay
@contextlib.contextmanager
def __udc_stopped(self) -> Generator[None, None, None]:
udc = usb.find_udc(self.__udc)
udc_path = usb.get_gadget_path(self.__gadget, usb.G_UDC)
with open(udc_path) as udc_file: with open(udc_path) as udc_file:
enabled = bool(udc_file.read().strip()) enabled = bool(udc_file.read().strip())
if enabled: if enabled:
@ -50,33 +58,30 @@ def _udc_stopped(gadget: str, udc: str) -> Generator[None, None, None]:
yield yield
finally: finally:
if enabled: if enabled:
time.sleep(self.__init_delay)
with open(udc_path, "w") as udc_file: with open(udc_path, "w") as udc_file:
udc_file.write(udc) udc_file.write(udc)
def enable_function(self, func: str) -> None:
def _enable_function(gadget: str, udc: str, func: str) -> None: with self.__udc_stopped():
with _udc_stopped(gadget, udc):
os.symlink( os.symlink(
usb.get_gadget_path(gadget, usb.G_FUNCTIONS, func), usb.get_gadget_path(self.__gadget, usb.G_FUNCTIONS, func),
usb.get_gadget_path(gadget, usb.G_PROFILE, func), usb.get_gadget_path(self.__gadget, usb.G_PROFILE, func),
) )
def disable_function(self, func: str) -> None:
with self.__udc_stopped():
os.unlink(usb.get_gadget_path(self.__gadget, usb.G_PROFILE, func))
def _disable_function(gadget: str, udc: str, func: str) -> None: def list_functions(self) -> None:
with _udc_stopped(gadget, udc): for meta_name in sorted(os.listdir(self.__meta_path)):
os.unlink(usb.get_gadget_path(gadget, usb.G_PROFILE, func)) with open(os.path.join(self.__meta_path, meta_name)) as meta_file:
def _list_functions(gadget: str, meta_path: str) -> None:
for meta_name in sorted(os.listdir(meta_path)):
with open(os.path.join(meta_path, meta_name)) as meta_file:
meta = json.loads(meta_file.read()) meta = json.loads(meta_file.read())
enabled = os.path.exists(usb.get_gadget_path(gadget, usb.G_PROFILE, meta["func"])) enabled = os.path.exists(usb.get_gadget_path(self.__gadget, usb.G_PROFILE, meta["func"]))
print(f"{'+' if enabled else '-'} {meta['func']} # {meta['name']}") print(f"{'+' if enabled else '-'} {meta['func']} # {meta['name']}")
def reset(self) -> None:
def _reset_gadget(gadget: str, udc: str) -> None: with self.__udc_stopped():
with _udc_stopped(gadget, udc):
pass pass
@ -99,11 +104,12 @@ def main(argv: Optional[List[str]]=None) -> None:
parser.add_argument("-r", "--reset-gadget", action="store_true", help="Reset gadget") parser.add_argument("-r", "--reset-gadget", action="store_true", help="Reset gadget")
options = parser.parse_args(argv[1:]) options = parser.parse_args(argv[1:])
gc = _GadgetControl(config.otg.meta, config.otg.gadget, config.otg.udc, config.otg.init_delay)
if options.reset_gadget: if options.reset_gadget:
_reset_gadget(config.otg.gadget, config.otg.udc) gc.reset()
return return
elif options.enable_function: elif options.enable_function:
_enable_function(config.otg.gadget, config.otg.udc, options.enable_function) gc.enable_function(options.enable_function)
elif options.disable_function: elif options.disable_function:
_disable_function(config.otg.gadget, config.otg.udc, options.disable_function) gc.disable_function(options.disable_function)
_list_functions(config.otg.gadget, config.otg.meta) gc.list_functions()