yyc12345/BasaltMeter
1
0
Fork 0
Code Activity

finish representer protocol

Browse Source
This commit is contained in:
yyc12345
2026-01-10 17:10:14 +08:00
parent 1abafeb0ca
commit 2a52c8ed7d
7 changed files with 75 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
BasaltTrainer/pipe_operator.py
View File

@@ -160,7 +160,7 @@ class PipeOperator:
def write_pod(self, pattern: struct.Struct, *args) -> None:
self.write(pattern.pack(*args))
STR_LEN_PACKER: ClassVar[struct.Struct] = struct.Struct("=N")
STR_LEN_PACKER: ClassVar[struct.Struct] = struct.Struct("=I")
def read_string(self) -> str:
(length,) = self.read_pod(PipeOperator.STR_LEN_PACKER)
@@ -174,22 +174,28 @@ class PipeOperator:
def read_bsstring(self) -> str:
(length,) = self.read_pod(PipeOperator.STR_LEN_PACKER)
str_bytes = self.read(length)
str_bytes = self.read(length * 2)
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))
self.write_pod(PipeOperator.STR_LEN_PACKER, len(str_bytes) // 2)
self.write(str_bytes)
def __encode_bsstring(self, s: str) -> bytes:
if IS_LITTLE_ENDIAN:
return s.encode("utf_16_le")
if IS_WINDOWS:
if IS_LITTLE_ENDIAN:
return s.encode("utf_16_le")
else:
return s.encode("utf_16_be")
else:
return s.encode("utf_16_be")
return s.encode("utf-8")
def __decode_bsstring(self, b: bytes) -> str:
if IS_LITTLE_ENDIAN:
return b.decode("utf_16_le")
if IS_WINDOWS:
if IS_LITTLE_ENDIAN:
return b.decode("utf_16_le")
else:
return b.decode("utf_16_be")
else:
return b.decode("utf_16_be")
return b.decode("utf-8")