arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-22 05:10:56 +00:00
Code Issues Releases Wiki Activity

protobuf: fix uvarint dumping

Browse Source 
In python3, chr() is not a proper way to pack an int to bytes.
This commit is contained in:
Jan Pochyla 2016-10-05 12:24:55 +02:00
parent 6be657114e
commit ec412c6da3
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/lib/protobuf.py
View File

@ -43,17 +43,21 @@ def print_protobuf_message(message_type):
print('CLOSE', message_type) print('CLOSE', message_type)
_UVARINT_DUMP_BUFFER = bytearray(1)
class UVarintType: class UVarintType:
WIRE_TYPE = 0 WIRE_TYPE = 0
@staticmethod @staticmethod
def dump(target, value): async def dump(target, value):
shifted_value = True shifted = True
while shifted_value: while shifted:
shifted_value = value >> 7 shifted = value >> 7
yield from target.write(chr((value & 0x7F) | ( _UVARINT_DUMP_BUFFER[0] = (value & 0x7F) | (
0x80 if shifted_value != 0 else 0x00))) 0x80 if shifted else 0x00)
value = shifted_value await target.write(_UVARINT_DUMP_BUFFER)
value = shifted
@staticmethod @staticmethod
def load(source): def load(source):