mirror of
https://github.com/basicswap/basicswap.git
synced 2025-11-05 18:38:09 +01:00
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (c) 2024 tecnovert
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
|
|
def decode_compactsize(b: bytes, offset: int = 0) -> (int, int):
|
|
i = b[offset]
|
|
if i < 0xFD:
|
|
return i, 1
|
|
offset += 1
|
|
if i == 0xFD:
|
|
return int.from_bytes(b[offset : offset + 2], "little"), 3
|
|
if i == 0xFE:
|
|
return int.from_bytes(b[offset : offset + 4], "little"), 5
|
|
# 0xff
|
|
return int.from_bytes(b[offset : offset + 8], "little"), 9
|
|
|
|
|
|
def encode_compactsize(i: int) -> bytes:
|
|
if i < 0xFD:
|
|
return bytes((i,))
|
|
if i <= 0xFFFF:
|
|
return bytes((0xFD,)) + i.to_bytes(2, "little")
|
|
if i <= 0xFFFFFFFF:
|
|
return bytes((0xFE,)) + i.to_bytes(4, "little")
|
|
return bytes((0xFF,)) + i.to_bytes(8, "little")
|
|
|
|
|
|
def decode_varint(b: bytes, offset: int = 0) -> (int, int):
|
|
i: int = 0
|
|
num_bytes: int = 0
|
|
while True:
|
|
c = b[offset + num_bytes]
|
|
i += (c & 0x7F) << (num_bytes * 7)
|
|
num_bytes += 1
|
|
if not c & 0x80:
|
|
break
|
|
if num_bytes > 8:
|
|
raise ValueError("Too many bytes")
|
|
return i, num_bytes
|
|
|
|
|
|
def encode_varint(i: int) -> bytes:
|
|
b = bytearray()
|
|
while i > 0x7F:
|
|
b += bytes(((i & 0x7F) | 0x80,))
|
|
i = i >> 7
|
|
b += bytes((i,))
|
|
return b
|