yyc12345/BasaltMeter
1
0
Fork 0
Code Activity

revert bsstring length transmit type

Browse Source
This commit is contained in:
yyc12345
2026-01-10 19:38:58 +08:00
parent 2a52c8ed7d
commit 0640e7bb25
2 changed files with 16 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
BasaltTrainer/pipe_operator.py
View File

@@ -1,7 +1,7 @@
import os
import sys
import struct
from typing import Optional, Any, ClassVar
from typing import Optional, Any
IS_WINDOWS: bool = sys.platform == "win32"
IS_LITTLE_ENDIAN: bool = sys.byteorder == "little"
@@ -13,6 +13,9 @@ if IS_WINDOWS:
else:
import errno
BSSTR_LEN_PACKER: struct.Struct = struct.Struct("N")
BSCHAR_SIZE: int = struct.calcsize("H") if IS_WINDOWS else struct.calcsize("c")
class PipeOperator:
"""
@@ -160,26 +163,24 @@ class PipeOperator:
def write_pod(self, pattern: struct.Struct, *args) -> None:
self.write(pattern.pack(*args))
STR_LEN_PACKER: ClassVar[struct.Struct] = struct.Struct("=I")
def read_string(self) -> str:
(length,) = self.read_pod(PipeOperator.STR_LEN_PACKER)
(length,) = self.read_pod(BSSTR_LEN_PACKER)
str_bytes = self.read(length)
return str_bytes.decode("utf-8")
def write_string(self, s: str) -> None:
str_bytes = s.encode("utf-8")
self.write_pod(PipeOperator.STR_LEN_PACKER, len(str_bytes))
self.write_pod(BSSTR_LEN_PACKER, len(str_bytes))
self.write(str_bytes)
def read_bsstring(self) -> str:
(length,) = self.read_pod(PipeOperator.STR_LEN_PACKER)
str_bytes = self.read(length * 2)
(length,) = self.read_pod(BSSTR_LEN_PACKER)
str_bytes = self.read(length * BSCHAR_SIZE)
return self.__decode_bsstring(str_bytes)
def write_bsstring(self, s: str) -> None:
str_bytes = self.__encode_bsstring(s)
self.write_pod(PipeOperator.STR_LEN_PACKER, len(str_bytes) // 2)
self.write_pod(BSSTR_LEN_PACKER, len(str_bytes) // BSCHAR_SIZE)
self.write(str_bytes)
def __encode_bsstring(self, s: str) -> bytes: