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

suppressed serial hid spam

Browse Source
This commit is contained in:
Devaev Maxim 2020-06-05 05:25:27 +03:00
parent 9e60f24c54
commit b9e53a00f8
1 changed files with 90 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

92
kvmd/plugins/hid/serial.py
View File

@ -30,6 +30,7 @@ import struct
import errno
import time
from typing import List
from typing import Dict
from typing import AsyncGenerator
@ -47,6 +48,7 @@ from ... import gpio
from ...yamlconf import Option
from ...validators.basic import valid_bool
from ...validators.basic import valid_int_f0
from ...validators.basic import valid_int_f1
from ...validators.basic import valid_float_f01
@ -58,6 +60,22 @@ from ...validators.hw import valid_gpio_pin
from . import BaseHid
# =====
class _RequestError(Exception):
def __init__(self, msg: str, online: bool=False) -> None:
super().__init__(msg)
self.msg = msg
self.online = online
class _FatalRequestError(_RequestError):
pass
class _TempRequestError(_RequestError):
pass
# =====
class _BaseEvent:
def make_command(self) -> bytes:
@ -141,6 +159,7 @@ class Plugin(BaseHid, multiprocessing.Process): # pylint: disable=too-many-inst
read_retries: int,
common_retries: int,
retries_delay: float,
errors_threshold: int,
noop: bool,
) -> None:
@ -155,6 +174,7 @@ class Plugin(BaseHid, multiprocessing.Process): # pylint: disable=too-many-inst
self.__read_retries = read_retries
self.__common_retries = common_retries
self.__retries_delay = retries_delay
self.__errors_threshold = errors_threshold
self.__noop = noop
self.__reset_wip = False
@ -183,6 +203,7 @@ class Plugin(BaseHid, multiprocessing.Process): # pylint: disable=too-many-inst
"read_retries": Option(10, type=valid_int_f1),
"common_retries": Option(100, type=valid_int_f1),
"retries_delay": Option(0.1, type=valid_float_f01),
"errors_threshold": Option(5, type=valid_int_f0),
"noop": Option(False, type=valid_bool),
}
@ -311,46 +332,34 @@ class Plugin(BaseHid, multiprocessing.Process): # pylint: disable=too-many-inst
def __process_request(self, tty: serial.Serial, request: bytes) -> None: # pylint: disable=too-many-branches
logger = get_logger()
errors: List[str] = []
runtime_errors = False
common_error_occured = False
common_retries = self.__common_retries
read_retries = self.__read_retries
while common_retries and read_retries:
if not self.__noop:
if tty.in_waiting:
tty.read(tty.in_waiting)
assert tty.write(request) == len(request)
response = tty.read(4)
else:
response = b"\x33\x20" # Magic + OK
response += struct.pack(">H", self.__make_crc16(response))
response = self.__send_request(tty, request)
try:
if len(response) < 4:
logger.error("No response from HID: request=%r", request)
read_retries -= 1
else:
raise _TempRequestError(f"No response from HID: request={request!r}")
assert len(response) == 4, response
if self.__make_crc16(response[-4:-2]) != struct.unpack(">H", response[-2:])[0]:
logger.error("Invalid response CRC; requesting response again ...")
request = self.__make_request(b"\x02\x00\x00\x00\x00") # Repeat an answer
else:
raise _TempRequestError("Invalid response CRC; requesting response again ...")
code = response[1]
if code == 0x48: # Request timeout
logger.error("Got request timeout from HID: request=%r", request)
if code == 0x48: # Request timeout # pylint: disable=no-else-raise
raise _TempRequestError(f"Got request timeout from HID: request={request!r}")
elif code == 0x40: # CRC Error
logger.error("Got CRC error of request from HID: request=%r", request)
raise _TempRequestError(f"Got CRC error of request from HID: request={request!r}")
elif code == 0x45: # Unknown command
logger.error("HID did not recognize the request=%r", request)
self.__state_flags.update(online=True)
return
raise _FatalRequestError(f"HID did not recognize the request={request!r}", online=True)
elif code == 0x24: # Rebooted?
logger.error("No previous command state inside HID, seems it was rebooted")
self.__state_flags.update(online=True)
return
raise _FatalRequestError("No previous command state inside HID, seems it was rebooted", online=True)
elif code == 0x20: # Done
if common_error_occured:
logger.info("Success!")
self.__state_flags.update(online=True)
return
elif code & 0x80: # Pong with leds
@ -362,18 +371,43 @@ class Plugin(BaseHid, multiprocessing.Process): # pylint: disable=too-many-inst
)
return
else:
logger.error("Invalid response from HID: request=%r; code=0x%x", request, code)
raise _TempRequestError(f"Invalid response from HID: request={request!r}; code=0x{code:02X}")
common_error_occured = True
except _RequestError as err:
common_retries -= 1
self.__state_flags.update(online=False)
self.__state_flags.update(online=err.online)
if runtime_errors:
logger.error(err.msg)
else:
errors.append(err.msg)
if len(errors) > self.__errors_threshold:
for msg in errors:
logger.error(msg)
errors = []
runtime_errors = True
if isinstance(err, _FatalRequestError):
break
if common_retries and read_retries:
logger.error("Retries left: common_retries=%d; read_retries=%d", common_retries, read_retries)
time.sleep(self.__retries_delay)
for msg in errors:
logger.error(msg)
if not (common_retries and read_retries):
logger.error("Can't process HID request due many errors: %r", request)
def __send_request(self, tty: serial.Serial, request: bytes) -> bytes:
if not self.__noop:
if tty.in_waiting:
tty.read(tty.in_waiting)
assert tty.write(request) == len(request)
response = tty.read(4)
else:
response = b"\x33\x20" # Magic + OK
response += struct.pack(">H", self.__make_crc16(response))
return response
def __make_request(self, command: bytes) -> bytes:
request = b"\x33" + command
request += struct.pack(">H", self.__make_crc16(request))