Add register write command

tuya-cloudcutter · Nov 4, 2023 · a31d5f6 · a31d5f6
1 parent d4d14e5
commit a31d5f6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/bk7231tools/serial/cmd_chip.py b/bk7231tools/serial/cmd_chip.py
@@ -17,6 +17,7 @@
     BkReadRegResp,
     BkRebootCmnd,
     BkSetBaudRateCmnd,
+    BkWriteRegCmnd,
 )
 from .protocol import CHIP_BY_CRC, PROTOCOLS, BK7231Protocol, ProtocolType
 from .utils import fix_addr
@@ -99,6 +100,10 @@ def register_read(self, address: int) -> int:
         response: BkReadRegResp = self.command(command)
         return response.value
 
+    def register_write(self, address: int, value: int) -> None:
+        command = BkWriteRegCmnd(address, value)
+        self.command(command)
+
     def read_flash_range_crc(self, start: int, end: int) -> int:
         start = fix_addr(start)
         end = fix_addr(end)

diff --git a/bk7231tools/serial/packets.py b/bk7231tools/serial/packets.py
@@ -54,6 +54,16 @@ class BkLinkCheckResp(Packet):
 
 @dataclass
 class BkWriteRegCmnd(Packet):
+    CODE = 0x01  # CMD_WriteReg
+    FORMAT = "<II"
+    HAS_RESP_OTHER = True
+    HAS_RESP_SAME = slice(0, 8)
+    address: int
+    value: int
+
+
+@dataclass
+class BkWriteRegResp(Packet):
     CODE = 0x01  # CMD_WriteReg
     FORMAT = "<II"
     address: int
@@ -277,6 +287,7 @@ def deserialize(cls, data: bytes) -> "Packet":
 RESPONSE_TABLE: Dict[Type[Packet], Type[Packet]] = {
     # short commands
     BkLinkCheckCmnd: BkLinkCheckResp,  # 0x00 / CMD_LinkCheck
+    BkWriteRegCmnd: BkWriteRegResp,  # 0x01 / CMD_WriteReg
     BkReadRegCmnd: BkReadRegResp,  # 0x03 / CMD_ReadReg
     BkCheckCrcCmnd: BkCheckCrcResp,  # 0x10 / CMD_CheckCRC
     BkBootVersionCmnd: BkBootVersionResp,  # 0x11 / CMD_ReadBootVersion