git / moonbite-core / test/functional/test_framework/address.py · raw

#!/usr/bin/env python3# Copyright (c) 2016-2020 The Bitcoin Core developers# Distributed under the MIT software license, see the accompanying# file COPYING or http://www.opensource.org/licenses/mit-license.php."""Encode and decode Bitcoin addresses.- base58 P2PKH and P2SH addresses.- bech32 segwit v0 P2WPKH and P2WSH addresses."""import enumimport unittestfrom .script import hash256, hash160, sha256, CScript, OP_0from .segwit_addr import encode_segwit_addressfrom .util import assert_equal, hex_str_to_bytesADDRESS_BCRT1_UNSPENDABLE = 'rmoon1qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqjymla8'ADDRESS_BCRT1_UNSPENDABLE_DESCRIPTOR = 'addr(rmoon1qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqjymla8)#7as0y5tj'# Coins sent to this address can be spent with a witness stack of just OP_TRUEADDRESS_BCRT1_P2WSH_OP_TRUE = 'rmoon1qft5p2uhsdcdc3l2ua4ap5qqfg4pjaqlp250x7us7a8qqhrxrxfsqnm677p'class AddressType(enum.Enum):    bech32 = 'bech32'    p2sh_segwit = 'p2sh-segwit'    legacy = 'legacy'  # P2PKHchars = '123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz'def byte_to_base58(b, version):    result = ''    str = b.hex()    str = chr(version).encode('latin-1').hex() + str    checksum = hash256(hex_str_to_bytes(str)).hex()    str += checksum[:8]    value = int('0x' + str, 0)    while value > 0:        result = chars[value % 58] + result        value //= 58    while (str[:2] == '00'):        result = chars[0] + result        str = str[2:]    return resultdef base58_to_byte(s):    """Converts a base58-encoded string to its data and version.    Throws if the base58 checksum is invalid."""    if not s:        return b''    n = 0    for c in s:        n *= 58        assert c in chars        digit = chars.index(c)        n += digit    h = '%x' % n    if len(h) % 2:        h = '0' + h    res = n.to_bytes((n.bit_length() + 7) // 8, 'big')    pad = 0    for c in s:        if c == chars[0]:            pad += 1        else:            break    res = b'\x00' * pad + res    # Assert if the checksum is invalid    assert_equal(hash256(res[:-4])[:4], res[-4:])    return res[1:-4], int(res[0])def keyhash_to_p2pkh(hash, main=False):    assert len(hash) == 20    version = 0 if main else 111    return byte_to_base58(hash, version)def scripthash_to_p2sh(hash, main=False):    assert len(hash) == 20    version = 5 if main else 58    return byte_to_base58(hash, version)def key_to_p2pkh(key, main=False):    key = check_key(key)    return keyhash_to_p2pkh(hash160(key), main)def script_to_p2sh(script, main=False):    script = check_script(script)    return scripthash_to_p2sh(hash160(script), main)def key_to_p2sh_p2wpkh(key, main=False):    key = check_key(key)    p2shscript = CScript([OP_0, hash160(key)])    return script_to_p2sh(p2shscript, main)def program_to_witness(version, program, main=False):    if (type(program) is str):        program = hex_str_to_bytes(program)    assert 0 <= version <= 16    assert 2 <= len(program) <= 40    assert version > 0 or len(program) in [20, 32]    return encode_segwit_address("ltc" if main else "rltc", version, program)def script_to_p2wsh(script, main=False):    script = check_script(script)    return program_to_witness(0, sha256(script), main)def key_to_p2wpkh(key, main=False):    key = check_key(key)    return program_to_witness(0, hash160(key), main)def script_to_p2sh_p2wsh(script, main=False):    script = check_script(script)    p2shscript = CScript([OP_0, sha256(script)])    return script_to_p2sh(p2shscript, main)def check_key(key):    if (type(key) is str):        key = hex_str_to_bytes(key)  # Assuming this is hex string    if (type(key) is bytes and (len(key) == 33 or len(key) == 65)):        return key    assert Falsedef check_script(script):    if (type(script) is str):        script = hex_str_to_bytes(script)  # Assuming this is hex string    if (type(script) is bytes or type(script) is CScript):        return script    assert Falseclass TestFrameworkScript(unittest.TestCase):    def test_base58encodedecode(self):        def check_base58(data, version):            self.assertEqual(base58_to_byte(byte_to_base58(data, version)), (data, version))        check_base58(bytes.fromhex('1f8ea1702a7bd4941bca0941b852c4bbfedb2e05'), 111)        check_base58(bytes.fromhex('3a0b05f4d7f66c3ba7009f453530296c845cc9cf'), 111)        check_base58(bytes.fromhex('41c1eaf111802559bad61b60d62b1f897c63928a'), 111)        check_base58(bytes.fromhex('0041c1eaf111802559bad61b60d62b1f897c63928a'), 111)        check_base58(bytes.fromhex('000041c1eaf111802559bad61b60d62b1f897c63928a'), 111)        check_base58(bytes.fromhex('00000041c1eaf111802559bad61b60d62b1f897c63928a'), 111)        check_base58(bytes.fromhex('1f8ea1702a7bd4941bca0941b852c4bbfedb2e05'), 0)        check_base58(bytes.fromhex('3a0b05f4d7f66c3ba7009f453530296c845cc9cf'), 0)        check_base58(bytes.fromhex('41c1eaf111802559bad61b60d62b1f897c63928a'), 0)        check_base58(bytes.fromhex('0041c1eaf111802559bad61b60d62b1f897c63928a'), 0)        check_base58(bytes.fromhex('000041c1eaf111802559bad61b60d62b1f897c63928a'), 0)        check_base58(bytes.fromhex('00000041c1eaf111802559bad61b60d62b1f897c63928a'), 0)