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

otg msd: handle functions switching

Browse Source
This commit is contained in:
Maxim Devaev
2022-03-31 11:54:31 +03:00
parent 03a4c13291
commit 6828a0e6aa
2 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
kvmd/plugins/msd/otg/drive.py
View File

@@ -23,6 +23,8 @@
import os
import errno
from typing import List
from .... import env
from .. import MsdOperationError
@@ -37,14 +39,23 @@ class MsdDriveLockedError(MsdOperationError):
# =====
class Drive:
def __init__(self, gadget: str, instance: int, lun: int) -> None:
self.__path = os.path.join(
self.__lun_path = os.path.join(
f"{env.SYSFS_PREFIX}/sys/kernel/config/usb_gadget",
gadget,
f"functions/mass_storage.usb{instance}/lun.{lun}",
)
self.__configs_root_path = os.path.join(
f"{env.SYSFS_PREFIX}/sys/kernel/config/usb_gadget",
gadget,
"configs/c.1",
)
self.__func_path = os.path.join(self.__configs_root_path, f"mass_storage.usb{instance}")
def get_sysfs_path(self) -> str:
return self.__path
def is_enabled(self) -> bool:
return os.path.exists(self.__func_path)
def get_watchable_paths(self) -> List[str]:
return [self.__lun_path, self.__configs_root_path]
# =====
@@ -69,12 +80,12 @@ class Drive:
# =====
def __get_param(self, param: str) -> str:
with open(os.path.join(self.__path, param)) as param_file:
with open(os.path.join(self.__lun_path, param)) as param_file:
return param_file.read().strip()
def __set_param(self, param: str, value: str) -> None:
try:
with open(os.path.join(self.__path, param), "w") as param_file:
with open(os.path.join(self.__lun_path, param), "w") as param_file:
param_file.write(value + "\n")
except OSError as err:
if err.errno == errno.EBUSY: