Host-customized fork of https://github.com/tecnovert/basicswap/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.3 KiB
51 lines
1.3 KiB
# -*- 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]), 3 |
|
if i == 0xfe: |
|
return int.from_bytes(b[offset: offset + 4]), 5 |
|
# 0xff |
|
return int.from_bytes(b[offset: offset + 8]), 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
|
|
|