mirror of
https://github.com/LBRYFoundation/LBRY-Vault.git
synced 2025-08-23 17:47:31 +00:00
This bug is triggered if ecc_fast._libsecp256k1 and/or crypto.AES are not present. Before, if the first test would have raised an exception it would have been implicitly caught by returning from the finally block and hence this effectively masks the test's outcome. Now, the exception is properly propagated causing the test to fail if an exception is raised.
766 lines
40 KiB
Python
766 lines
40 KiB
Python
import base64
|
|
import unittest
|
|
import sys
|
|
|
|
from electrum.bitcoin import (
|
|
public_key_to_p2pkh,
|
|
bip32_root, bip32_public_derivation, bip32_private_derivation,
|
|
Hash, address_from_private_key,
|
|
is_address, is_private_key, xpub_from_xprv, is_new_seed, is_old_seed,
|
|
var_int, op_push, address_to_script,
|
|
deserialize_privkey, serialize_privkey, is_segwit_address,
|
|
is_b58_address, address_to_scripthash, is_minikey, is_compressed, is_xpub,
|
|
xpub_type, is_xprv, is_bip32_derivation, seed_type, EncodeBase58Check,
|
|
script_num_to_hex, push_script, add_number_to_script, int_to_hex, convert_bip32_path_to_list_of_uint32)
|
|
from electrum import ecc, crypto, constants
|
|
from electrum.ecc import number_to_string, string_to_number
|
|
from electrum.transaction import opcodes
|
|
from electrum.util import bfh, bh2u
|
|
from electrum.storage import WalletStorage
|
|
from electrum.keystore import xtype_from_derivation
|
|
|
|
from electrum import ecc_fast
|
|
|
|
from . import SequentialTestCase
|
|
from . import TestCaseForTestnet
|
|
from . import FAST_TESTS
|
|
|
|
|
|
try:
|
|
import ecdsa
|
|
except ImportError:
|
|
sys.exit("Error: python-ecdsa does not seem to be installed. Try 'sudo pip install ecdsa'")
|
|
|
|
|
|
def needs_test_with_all_ecc_implementations(func):
|
|
"""Function decorator to run a unit test twice:
|
|
once when libsecp256k1 is not available, once when it is.
|
|
|
|
NOTE: this is inherently sequential;
|
|
tests running in parallel would break things
|
|
"""
|
|
def run_test(*args, **kwargs):
|
|
if FAST_TESTS: # if set, only run tests once, using fastest implementation
|
|
func(*args, **kwargs)
|
|
return
|
|
ecc_fast.undo_monkey_patching_of_python_ecdsa_internals_with_libsecp256k1()
|
|
try:
|
|
# first test without libsecp
|
|
func(*args, **kwargs)
|
|
finally:
|
|
ecc_fast.do_monkey_patching_of_python_ecdsa_internals_with_libsecp256k1()
|
|
# if libsecp is not available, we are done
|
|
if not ecc_fast._libsecp256k1:
|
|
return
|
|
# if libsecp is available, test again now
|
|
func(*args, **kwargs)
|
|
return run_test
|
|
|
|
|
|
def needs_test_with_all_aes_implementations(func):
|
|
"""Function decorator to run a unit test twice:
|
|
once when pycryptodomex is not available, once when it is.
|
|
|
|
NOTE: this is inherently sequential;
|
|
tests running in parallel would break things
|
|
"""
|
|
def run_test(*args, **kwargs):
|
|
if FAST_TESTS: # if set, only run tests once, using fastest implementation
|
|
func(*args, **kwargs)
|
|
return
|
|
_aes = crypto.AES
|
|
crypto.AES = None
|
|
try:
|
|
# first test without pycryptodomex
|
|
func(*args, **kwargs)
|
|
finally:
|
|
crypto.AES = _aes
|
|
# if pycryptodomex is not available, we are done
|
|
if not _aes:
|
|
return
|
|
# if pycryptodomex is available, test again now
|
|
func(*args, **kwargs)
|
|
return run_test
|
|
|
|
|
|
class Test_bitcoin(SequentialTestCase):
|
|
|
|
def test_libsecp256k1_is_available(self):
|
|
# we want the unit testing framework to test with libsecp256k1 available.
|
|
self.assertTrue(bool(ecc_fast._libsecp256k1))
|
|
|
|
def test_pycryptodomex_is_available(self):
|
|
# we want the unit testing framework to test with pycryptodomex available.
|
|
self.assertTrue(bool(crypto.AES))
|
|
|
|
@needs_test_with_all_aes_implementations
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_crypto(self):
|
|
for message in [b"Chancellor on brink of second bailout for banks", b'\xff'*512]:
|
|
self._do_test_crypto(message)
|
|
|
|
def _do_test_crypto(self, message):
|
|
G = ecc.generator()
|
|
_r = G.order()
|
|
pvk = ecdsa.util.randrange(_r)
|
|
|
|
Pub = pvk*G
|
|
pubkey_c = Pub.get_public_key_bytes(True)
|
|
#pubkey_u = point_to_ser(Pub,False)
|
|
addr_c = public_key_to_p2pkh(pubkey_c)
|
|
|
|
#print "Private key ", '%064x'%pvk
|
|
eck = ecc.ECPrivkey(number_to_string(pvk,_r))
|
|
|
|
#print "Compressed public key ", pubkey_c.encode('hex')
|
|
enc = ecc.ECPubkey(pubkey_c).encrypt_message(message)
|
|
dec = eck.decrypt_message(enc)
|
|
self.assertEqual(message, dec)
|
|
|
|
#print "Uncompressed public key", pubkey_u.encode('hex')
|
|
#enc2 = EC_KEY.encrypt_message(message, pubkey_u)
|
|
dec2 = eck.decrypt_message(enc)
|
|
self.assertEqual(message, dec2)
|
|
|
|
signature = eck.sign_message(message, True)
|
|
#print signature
|
|
eck.verify_message_for_address(signature, message)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_ecc_sanity(self):
|
|
G = ecc.generator()
|
|
n = G.order()
|
|
self.assertEqual(ecc.CURVE_ORDER, n)
|
|
inf = n * G
|
|
self.assertEqual(ecc.point_at_infinity(), inf)
|
|
self.assertTrue(inf.is_at_infinity())
|
|
self.assertFalse(G.is_at_infinity())
|
|
self.assertEqual(11 * G, 7 * G + 4 * G)
|
|
self.assertEqual((n + 2) * G, 2 * G)
|
|
self.assertEqual((n - 2) * G, -2 * G)
|
|
A = (n - 2) * G
|
|
B = (n - 1) * G
|
|
C = n * G
|
|
D = (n + 1) * G
|
|
self.assertFalse(A.is_at_infinity())
|
|
self.assertFalse(B.is_at_infinity())
|
|
self.assertTrue(C.is_at_infinity())
|
|
self.assertTrue((C * 5).is_at_infinity())
|
|
self.assertFalse(D.is_at_infinity())
|
|
self.assertEqual(inf, C)
|
|
self.assertEqual(inf, A + 2 * G)
|
|
self.assertEqual(inf, D + (-1) * G)
|
|
self.assertNotEqual(A, B)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_msg_signing(self):
|
|
msg1 = b'Chancellor on brink of second bailout for banks'
|
|
msg2 = b'Electrum'
|
|
|
|
def sign_message_with_wif_privkey(wif_privkey, msg):
|
|
txin_type, privkey, compressed = deserialize_privkey(wif_privkey)
|
|
key = ecc.ECPrivkey(privkey)
|
|
return key.sign_message(msg, compressed)
|
|
|
|
sig1 = sign_message_with_wif_privkey(
|
|
'L1TnU2zbNaAqMoVh65Cyvmcjzbrj41Gs9iTLcWbpJCMynXuap6UN', msg1)
|
|
addr1 = '15hETetDmcXm1mM4sEf7U2KXC9hDHFMSzz'
|
|
sig2 = sign_message_with_wif_privkey(
|
|
'5Hxn5C4SQuiV6e62A1MtZmbSeQyrLFhu5uYks62pU5VBUygK2KD', msg2)
|
|
addr2 = '1GPHVTY8UD9my6jyP4tb2TYJwUbDetyNC6'
|
|
|
|
sig1_b64 = base64.b64encode(sig1)
|
|
sig2_b64 = base64.b64encode(sig2)
|
|
|
|
self.assertEqual(sig1_b64, b'H/9jMOnj4MFbH3d7t4yCQ9i7DgZU/VZ278w3+ySv2F4yIsdqjsc5ng3kmN8OZAThgyfCZOQxZCWza9V5XzlVY0Y=')
|
|
self.assertEqual(sig2_b64, b'G84dmJ8TKIDKMT9qBRhpX2sNmR0y5t+POcYnFFJCs66lJmAs3T8A6Sbpx7KA6yTQ9djQMabwQXRrDomOkIKGn18=')
|
|
|
|
self.assertTrue(ecc.verify_message_with_address(addr1, sig1, msg1))
|
|
self.assertTrue(ecc.verify_message_with_address(addr2, sig2, msg2))
|
|
|
|
self.assertFalse(ecc.verify_message_with_address(addr1, b'wrong', msg1))
|
|
self.assertFalse(ecc.verify_message_with_address(addr1, sig2, msg1))
|
|
|
|
@needs_test_with_all_aes_implementations
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_decrypt_message(self):
|
|
key = WalletStorage.get_eckey_from_password('pw123')
|
|
self.assertEqual(b'me<(s_s)>age', key.decrypt_message(b'QklFMQMDFtgT3zWSQsa+Uie8H/WvfUjlu9UN9OJtTt3KlgKeSTi6SQfuhcg1uIz9hp3WIUOFGTLr4RNQBdjPNqzXwhkcPi2Xsbiw6UCNJncVPJ6QBg=='))
|
|
self.assertEqual(b'me<(s_s)>age', key.decrypt_message(b'QklFMQKXOXbylOQTSMGfo4MFRwivAxeEEkewWQrpdYTzjPhqjHcGBJwdIhB7DyRfRQihuXx1y0ZLLv7XxLzrILzkl/H4YUtZB4uWjuOAcmxQH4i/Og=='))
|
|
self.assertEqual(b'hey_there' * 100, key.decrypt_message(b'QklFMQLOOsabsXtGQH8edAa6VOUa5wX8/DXmxX9NyHoAx1a5bWgllayGRVPeI2bf0ZdWK0tfal0ap0ZIVKbd2eOJybqQkILqT6E1/Syzq0Zicyb/AA1eZNkcX5y4gzloxinw00ubCA8M7gcUjJpOqbnksATcJ5y2YYXcHMGGfGurWu6uJ/UyrNobRidWppRMW5yR9/6utyNvT6OHIolCMEf7qLcmtneoXEiz51hkRdZS7weNf9mGqSbz9a2NL3sdh1A0feHIjAZgcCKcAvksNUSauf0/FnIjzTyPRpjRDMeDC8Ci3sGiuO3cvpWJwhZfbjcS26KmBv2CHWXfRRNFYOInHZNIXWNAoBB47Il5bGSMd+uXiGr+SQ9tNvcu+BiJNmFbxYqg+oQ8dGAl1DtvY2wJVY8k7vO9BIWSpyIxfGw7EDifhc5vnOmGe016p6a01C3eVGxgl23UYMrP7+fpjOcPmTSF4rk5U5ljEN3MSYqlf1QEv0OqlI9q1TwTK02VBCjMTYxDHsnt04OjNBkNO8v5uJ4NR+UUDBEp433z53I59uawZ+dbk4v4ZExcl8EGmKm3Gzbal/iJ/F7KQuX2b/ySEhLOFVYFWxK73X1nBvCSK2mC2/8fCw8oI5pmvzJwQhcCKTdEIrz3MMvAHqtPScDUOjzhXxInQOCb3+UBj1PPIdqkYLvZss1TEaBwYZjLkVnK2MBj7BaqT6Rp6+5A/fippUKHsnB6eYMEPR2YgDmCHL+4twxHJG6UWdP3ybaKiiAPy2OHNP6PTZ0HrqHOSJzBSDD+Z8YpaRg29QX3UEWlqnSKaan0VYAsV1VeaN0XFX46/TWO0L5tjhYVXJJYGqo6tIQJymxATLFRF6AZaD1Mwd27IAL04WkmoQoXfO6OFfwdp/shudY/1gBkDBvGPICBPtnqkvhGF+ZF3IRkuPwiFWeXmwBxKHsRx/3+aJu32Ml9+za41zVk2viaxcGqwTc5KMexQFLAUwqhv+aIik7U+5qk/gEVSuRoVkihoweFzKolNF+BknH2oB4rZdPixag5Zje3DvgjsSFlOl69W/67t/Gs8htfSAaHlsB8vWRQr9+v/lxTbrAw+O0E+sYGoObQ4qQMyQshNZEHbpPg63eWiHtJJnrVBvOeIbIHzoLDnMDsWVWZSMzAQ1vhX1H5QLgSEbRlKSliVY03kDkh/Nk/KOn+B2q37Ialq4JcRoIYFGJ8AoYEAD0tRuTqFddIclE75HzwaNG7NyKW1plsa72ciOPwsPJsdd5F0qdSQ3OSKtooTn7uf6dXOc4lDkfrVYRlZ0PX'))
|
|
|
|
@needs_test_with_all_aes_implementations
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_encrypt_message(self):
|
|
key = WalletStorage.get_eckey_from_password('secret_password77')
|
|
msgs = [
|
|
bytes([0] * 555),
|
|
b'cannot think of anything funny'
|
|
]
|
|
for plaintext in msgs:
|
|
ciphertext1 = key.encrypt_message(plaintext)
|
|
ciphertext2 = key.encrypt_message(plaintext)
|
|
self.assertEqual(plaintext, key.decrypt_message(ciphertext1))
|
|
self.assertEqual(plaintext, key.decrypt_message(ciphertext2))
|
|
self.assertNotEqual(ciphertext1, ciphertext2)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_sign_transaction(self):
|
|
eckey1 = ecc.ECPrivkey(bfh('7e1255fddb52db1729fc3ceb21a46f95b8d9fe94cc83425e936a6c5223bb679d'))
|
|
sig1 = eckey1.sign_transaction(bfh('5a548b12369a53faaa7e51b5081829474ebdd9c924b3a8230b69aa0be254cd94'))
|
|
self.assertEqual(bfh('3045022100902a288b98392254cd23c0e9a49ac6d7920f171b8249a48e484b998f1874a2010220723d844826828f092cf400cb210c4fa0b8cd1b9d1a7f21590e78e022ff6476b9'), sig1)
|
|
|
|
eckey2 = ecc.ECPrivkey(bfh('c7ce8c1462c311eec24dff9e2532ac6241e50ae57e7d1833af21942136972f23'))
|
|
sig2 = eckey2.sign_transaction(bfh('642a2e66332f507c92bda910158dfe46fc10afbf72218764899d3af99a043fac'))
|
|
self.assertEqual(bfh('30440220618513f4cfc87dde798ce5febae7634c23e7b9254a1eabf486be820f6a7c2c4702204fef459393a2b931f949e63ced06888f35e286e446dc46feb24b5b5f81c6ed52'), sig2)
|
|
|
|
@needs_test_with_all_aes_implementations
|
|
def test_aes_homomorphic(self):
|
|
"""Make sure AES is homomorphic."""
|
|
payload = u'\u66f4\u7a33\u5b9a\u7684\u4ea4\u6613\u5e73\u53f0'
|
|
password = u'secret'
|
|
enc = crypto.pw_encode(payload, password)
|
|
dec = crypto.pw_decode(enc, password)
|
|
self.assertEqual(dec, payload)
|
|
|
|
@needs_test_with_all_aes_implementations
|
|
def test_aes_encode_without_password(self):
|
|
"""When not passed a password, pw_encode is noop on the payload."""
|
|
payload = u'\u66f4\u7a33\u5b9a\u7684\u4ea4\u6613\u5e73\u53f0'
|
|
enc = crypto.pw_encode(payload, None)
|
|
self.assertEqual(payload, enc)
|
|
|
|
@needs_test_with_all_aes_implementations
|
|
def test_aes_deencode_without_password(self):
|
|
"""When not passed a password, pw_decode is noop on the payload."""
|
|
payload = u'\u66f4\u7a33\u5b9a\u7684\u4ea4\u6613\u5e73\u53f0'
|
|
enc = crypto.pw_decode(payload, None)
|
|
self.assertEqual(payload, enc)
|
|
|
|
@needs_test_with_all_aes_implementations
|
|
def test_aes_decode_with_invalid_password(self):
|
|
"""pw_decode raises an Exception when supplied an invalid password."""
|
|
payload = u"blah"
|
|
password = u"uber secret"
|
|
wrong_password = u"not the password"
|
|
enc = crypto.pw_encode(payload, password)
|
|
self.assertRaises(Exception, crypto.pw_decode, enc, wrong_password)
|
|
|
|
def test_hash(self):
|
|
"""Make sure the Hash function does sha256 twice"""
|
|
payload = u"test"
|
|
expected = b'\x95MZI\xfdp\xd9\xb8\xbc\xdb5\xd2R&x)\x95\x7f~\xf7\xfalt\xf8\x84\x19\xbd\xc5\xe8"\t\xf4'
|
|
|
|
result = Hash(payload)
|
|
self.assertEqual(expected, result)
|
|
|
|
def test_int_to_hex(self):
|
|
self.assertEqual('00', int_to_hex(0, 1))
|
|
self.assertEqual('ff', int_to_hex(-1, 1))
|
|
self.assertEqual('00000000', int_to_hex(0, 4))
|
|
self.assertEqual('01000000', int_to_hex(1, 4))
|
|
self.assertEqual('7f', int_to_hex(127, 1))
|
|
self.assertEqual('7f00', int_to_hex(127, 2))
|
|
self.assertEqual('80', int_to_hex(128, 1))
|
|
self.assertEqual('80', int_to_hex(-128, 1))
|
|
self.assertEqual('8000', int_to_hex(128, 2))
|
|
self.assertEqual('ff', int_to_hex(255, 1))
|
|
self.assertEqual('ff7f', int_to_hex(32767, 2))
|
|
self.assertEqual('0080', int_to_hex(-32768, 2))
|
|
self.assertEqual('ffff', int_to_hex(65535, 2))
|
|
with self.assertRaises(OverflowError): int_to_hex(256, 1)
|
|
with self.assertRaises(OverflowError): int_to_hex(-129, 1)
|
|
with self.assertRaises(OverflowError): int_to_hex(-257, 1)
|
|
with self.assertRaises(OverflowError): int_to_hex(65536, 2)
|
|
with self.assertRaises(OverflowError): int_to_hex(-32769, 2)
|
|
|
|
def test_var_int(self):
|
|
for i in range(0xfd):
|
|
self.assertEqual(var_int(i), "{:02x}".format(i) )
|
|
|
|
self.assertEqual(var_int(0xfd), "fdfd00")
|
|
self.assertEqual(var_int(0xfe), "fdfe00")
|
|
self.assertEqual(var_int(0xff), "fdff00")
|
|
self.assertEqual(var_int(0x1234), "fd3412")
|
|
self.assertEqual(var_int(0xffff), "fdffff")
|
|
self.assertEqual(var_int(0x10000), "fe00000100")
|
|
self.assertEqual(var_int(0x12345678), "fe78563412")
|
|
self.assertEqual(var_int(0xffffffff), "feffffffff")
|
|
self.assertEqual(var_int(0x100000000), "ff0000000001000000")
|
|
self.assertEqual(var_int(0x0123456789abcdef), "ffefcdab8967452301")
|
|
|
|
def test_op_push(self):
|
|
self.assertEqual(op_push(0x00), '00')
|
|
self.assertEqual(op_push(0x12), '12')
|
|
self.assertEqual(op_push(0x4b), '4b')
|
|
self.assertEqual(op_push(0x4c), '4c4c')
|
|
self.assertEqual(op_push(0xfe), '4cfe')
|
|
self.assertEqual(op_push(0xff), '4cff')
|
|
self.assertEqual(op_push(0x100), '4d0001')
|
|
self.assertEqual(op_push(0x1234), '4d3412')
|
|
self.assertEqual(op_push(0xfffe), '4dfeff')
|
|
self.assertEqual(op_push(0xffff), '4dffff')
|
|
self.assertEqual(op_push(0x10000), '4e00000100')
|
|
self.assertEqual(op_push(0x12345678), '4e78563412')
|
|
|
|
def test_script_num_to_hex(self):
|
|
# test vectors from https://github.com/btcsuite/btcd/blob/fdc2bc867bda6b351191b5872d2da8270df00d13/txscript/scriptnum.go#L77
|
|
self.assertEqual(script_num_to_hex(127), '7f')
|
|
self.assertEqual(script_num_to_hex(-127), 'ff')
|
|
self.assertEqual(script_num_to_hex(128), '8000')
|
|
self.assertEqual(script_num_to_hex(-128), '8080')
|
|
self.assertEqual(script_num_to_hex(129), '8100')
|
|
self.assertEqual(script_num_to_hex(-129), '8180')
|
|
self.assertEqual(script_num_to_hex(256), '0001')
|
|
self.assertEqual(script_num_to_hex(-256), '0081')
|
|
self.assertEqual(script_num_to_hex(32767), 'ff7f')
|
|
self.assertEqual(script_num_to_hex(-32767), 'ffff')
|
|
self.assertEqual(script_num_to_hex(32768), '008000')
|
|
self.assertEqual(script_num_to_hex(-32768), '008080')
|
|
|
|
def test_push_script(self):
|
|
# https://github.com/bitcoin/bips/blob/master/bip-0062.mediawiki#push-operators
|
|
self.assertEqual(push_script(''), bh2u(bytes([opcodes.OP_0])))
|
|
self.assertEqual(push_script('07'), bh2u(bytes([opcodes.OP_7])))
|
|
self.assertEqual(push_script('10'), bh2u(bytes([opcodes.OP_16])))
|
|
self.assertEqual(push_script('81'), bh2u(bytes([opcodes.OP_1NEGATE])))
|
|
self.assertEqual(push_script('11'), '0111')
|
|
self.assertEqual(push_script(75 * '42'), '4b' + 75 * '42')
|
|
self.assertEqual(push_script(76 * '42'), bh2u(bytes([opcodes.OP_PUSHDATA1]) + bfh('4c' + 76 * '42')))
|
|
self.assertEqual(push_script(100 * '42'), bh2u(bytes([opcodes.OP_PUSHDATA1]) + bfh('64' + 100 * '42')))
|
|
self.assertEqual(push_script(255 * '42'), bh2u(bytes([opcodes.OP_PUSHDATA1]) + bfh('ff' + 255 * '42')))
|
|
self.assertEqual(push_script(256 * '42'), bh2u(bytes([opcodes.OP_PUSHDATA2]) + bfh('0001' + 256 * '42')))
|
|
self.assertEqual(push_script(520 * '42'), bh2u(bytes([opcodes.OP_PUSHDATA2]) + bfh('0802' + 520 * '42')))
|
|
|
|
def test_add_number_to_script(self):
|
|
# https://github.com/bitcoin/bips/blob/master/bip-0062.mediawiki#numbers
|
|
self.assertEqual(add_number_to_script(0), bytes([opcodes.OP_0]))
|
|
self.assertEqual(add_number_to_script(7), bytes([opcodes.OP_7]))
|
|
self.assertEqual(add_number_to_script(16), bytes([opcodes.OP_16]))
|
|
self.assertEqual(add_number_to_script(-1), bytes([opcodes.OP_1NEGATE]))
|
|
self.assertEqual(add_number_to_script(-127), bfh('01ff'))
|
|
self.assertEqual(add_number_to_script(-2), bfh('0182'))
|
|
self.assertEqual(add_number_to_script(17), bfh('0111'))
|
|
self.assertEqual(add_number_to_script(127), bfh('017f'))
|
|
self.assertEqual(add_number_to_script(-32767), bfh('02ffff'))
|
|
self.assertEqual(add_number_to_script(-128), bfh('028080'))
|
|
self.assertEqual(add_number_to_script(128), bfh('028000'))
|
|
self.assertEqual(add_number_to_script(32767), bfh('02ff7f'))
|
|
self.assertEqual(add_number_to_script(-8388607), bfh('03ffffff'))
|
|
self.assertEqual(add_number_to_script(-32768), bfh('03008080'))
|
|
self.assertEqual(add_number_to_script(32768), bfh('03008000'))
|
|
self.assertEqual(add_number_to_script(8388607), bfh('03ffff7f'))
|
|
self.assertEqual(add_number_to_script(-2147483647), bfh('04ffffffff'))
|
|
self.assertEqual(add_number_to_script(-8388608 ), bfh('0400008080'))
|
|
self.assertEqual(add_number_to_script(8388608), bfh('0400008000'))
|
|
self.assertEqual(add_number_to_script(2147483647), bfh('04ffffff7f'))
|
|
|
|
def test_address_to_script(self):
|
|
# bech32 native segwit
|
|
# test vectors from BIP-0173
|
|
self.assertEqual(address_to_script('BC1QW508D6QEJXTDG4Y5R3ZARVARY0C5XW7KV8F3T4'), '0014751e76e8199196d454941c45d1b3a323f1433bd6')
|
|
self.assertEqual(address_to_script('bc1pw508d6qejxtdg4y5r3zarvary0c5xw7kw508d6qejxtdg4y5r3zarvary0c5xw7k7grplx'), '5128751e76e8199196d454941c45d1b3a323f1433bd6751e76e8199196d454941c45d1b3a323f1433bd6')
|
|
self.assertEqual(address_to_script('BC1SW50QA3JX3S'), '6002751e')
|
|
self.assertEqual(address_to_script('bc1zw508d6qejxtdg4y5r3zarvaryvg6kdaj'), '5210751e76e8199196d454941c45d1b3a323')
|
|
|
|
# base58 P2PKH
|
|
self.assertEqual(address_to_script('14gcRovpkCoGkCNBivQBvw7eso7eiNAbxG'), '76a91428662c67561b95c79d2257d2a93d9d151c977e9188ac')
|
|
self.assertEqual(address_to_script('1BEqfzh4Y3zzLosfGhw1AsqbEKVW6e1qHv'), '76a914704f4b81cadb7bf7e68c08cd3657220f680f863c88ac')
|
|
|
|
# base58 P2SH
|
|
self.assertEqual(address_to_script('35ZqQJcBQMZ1rsv8aSuJ2wkC7ohUCQMJbT'), 'a9142a84cf00d47f699ee7bbc1dea5ec1bdecb4ac15487')
|
|
self.assertEqual(address_to_script('3PyjzJ3im7f7bcV724GR57edKDqoZvH7Ji'), 'a914f47c8954e421031ad04ecd8e7752c9479206b9d387')
|
|
|
|
|
|
class Test_bitcoin_testnet(TestCaseForTestnet):
|
|
|
|
def test_address_to_script(self):
|
|
# bech32 native segwit
|
|
# test vectors from BIP-0173
|
|
self.assertEqual(address_to_script('tb1qrp33g0q5c5txsp9arysrx4k6zdkfs4nce4xj0gdcccefvpysxf3q0sl5k7'), '00201863143c14c5166804bd19203356da136c985678cd4d27a1b8c6329604903262')
|
|
self.assertEqual(address_to_script('tb1qqqqqp399et2xygdj5xreqhjjvcmzhxw4aywxecjdzew6hylgvsesrxh6hy'), '0020000000c4a5cad46221b2a187905e5266362b99d5e91c6ce24d165dab93e86433')
|
|
|
|
# base58 P2PKH
|
|
self.assertEqual(address_to_script('mutXcGt1CJdkRvXuN2xoz2quAAQYQ59bRX'), '76a9149da64e300c5e4eb4aaffc9c2fd465348d5618ad488ac')
|
|
self.assertEqual(address_to_script('miqtaRTkU3U8rzwKbEHx3g8FSz8GJtPS3K'), '76a914247d2d5b6334bdfa2038e85b20fc15264f8e5d2788ac')
|
|
|
|
# base58 P2SH
|
|
self.assertEqual(address_to_script('2N3LSvr3hv5EVdfcrxg2Yzecf3SRvqyBE4p'), 'a9146eae23d8c4a941316017946fc761a7a6c85561fb87')
|
|
self.assertEqual(address_to_script('2NE4ZdmxFmUgwu5wtfoN2gVniyMgRDYq1kk'), 'a914e4567743d378957cd2ee7072da74b1203c1a7a0b87')
|
|
|
|
|
|
class Test_xprv_xpub(SequentialTestCase):
|
|
|
|
xprv_xpub = (
|
|
# Taken from test vectors in https://en.bitcoin.it/wiki/BIP_0032_TestVectors
|
|
{'xprv': 'xprvA41z7zogVVwxVSgdKUHDy1SKmdb533PjDz7J6N6mV6uS3ze1ai8FHa8kmHScGpWmj4WggLyQjgPie1rFSruoUihUZREPSL39UNdE3BBDu76',
|
|
'xpub': 'xpub6H1LXWLaKsWFhvm6RVpEL9P4KfRZSW7abD2ttkWP3SSQvnyA8FSVqNTEcYFgJS2UaFcxupHiYkro49S8yGasTvXEYBVPamhGW6cFJodrTHy',
|
|
'xtype': 'standard'},
|
|
{'xprv': 'yprvAJEYHeNEPcyBoQYM7sGCxDiNCTX65u4ANgZuSGTrKN5YCC9MP84SBayrgaMyZV7zvkHrr3HVPTK853s2SPk4EttPazBZBmz6QfDkXeE8Zr7',
|
|
'xpub': 'ypub6XDth9u8DzXV1tcpDtoDKMf6kVMaVMn1juVWEesTshcX4zUVvfNgjPJLXrD9N7AdTLnbHFL64KmBn3SNaTe69iZYbYCqLCCNPZKbLz9niQ4',
|
|
'xtype': 'p2wpkh-p2sh'},
|
|
{'xprv': 'zprvAWgYBBk7JR8GkraNZJeEodAp2UR1VRWJTXyV1ywuUVs1awUgTiBS1ZTDtLA5F3MFDn1LZzu8dUpSKdT7ToDpvEG6PQu4bJs7zQY47Sd3sEZ',
|
|
'xpub': 'zpub6jftahH18ngZyLeqfLBFAm7YaWFVttE9pku5pNMX2qPzTjoq1FVgZMmhjecyB2nqFb31gHE9vNvbaggU6vvWpNZbXEWLLUjYjFqG95LNyT8',
|
|
'xtype': 'p2wpkh'},
|
|
)
|
|
|
|
def _do_test_bip32(self, seed, sequence):
|
|
xprv, xpub = bip32_root(bfh(seed), 'standard')
|
|
self.assertEqual("m/", sequence[0:2])
|
|
path = 'm'
|
|
sequence = sequence[2:]
|
|
for n in sequence.split('/'):
|
|
child_path = path + '/' + n
|
|
if n[-1] != "'":
|
|
xpub2 = bip32_public_derivation(xpub, path, child_path)
|
|
xprv, xpub = bip32_private_derivation(xprv, path, child_path)
|
|
if n[-1] != "'":
|
|
self.assertEqual(xpub, xpub2)
|
|
path = child_path
|
|
|
|
return xpub, xprv
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_bip32(self):
|
|
# see https://en.bitcoin.it/wiki/BIP_0032_TestVectors
|
|
xpub, xprv = self._do_test_bip32("000102030405060708090a0b0c0d0e0f", "m/0'/1/2'/2/1000000000")
|
|
self.assertEqual("xpub6H1LXWLaKsWFhvm6RVpEL9P4KfRZSW7abD2ttkWP3SSQvnyA8FSVqNTEcYFgJS2UaFcxupHiYkro49S8yGasTvXEYBVPamhGW6cFJodrTHy", xpub)
|
|
self.assertEqual("xprvA41z7zogVVwxVSgdKUHDy1SKmdb533PjDz7J6N6mV6uS3ze1ai8FHa8kmHScGpWmj4WggLyQjgPie1rFSruoUihUZREPSL39UNdE3BBDu76", xprv)
|
|
|
|
xpub, xprv = self._do_test_bip32("fffcf9f6f3f0edeae7e4e1dedbd8d5d2cfccc9c6c3c0bdbab7b4b1aeaba8a5a29f9c999693908d8a8784817e7b7875726f6c696663605d5a5754514e4b484542","m/0/2147483647'/1/2147483646'/2")
|
|
self.assertEqual("xpub6FnCn6nSzZAw5Tw7cgR9bi15UV96gLZhjDstkXXxvCLsUXBGXPdSnLFbdpq8p9HmGsApME5hQTZ3emM2rnY5agb9rXpVGyy3bdW6EEgAtqt", xpub)
|
|
self.assertEqual("xprvA2nrNbFZABcdryreWet9Ea4LvTJcGsqrMzxHx98MMrotbir7yrKCEXw7nadnHM8Dq38EGfSh6dqA9QWTyefMLEcBYJUuekgW4BYPJcr9E7j", xprv)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_xpub_from_xprv(self):
|
|
"""We can derive the xpub key from a xprv."""
|
|
for xprv_details in self.xprv_xpub:
|
|
result = xpub_from_xprv(xprv_details['xprv'])
|
|
self.assertEqual(result, xprv_details['xpub'])
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_is_xpub(self):
|
|
for xprv_details in self.xprv_xpub:
|
|
xpub = xprv_details['xpub']
|
|
self.assertTrue(is_xpub(xpub))
|
|
self.assertFalse(is_xpub('xpub1nval1d'))
|
|
self.assertFalse(is_xpub('xpub661MyMwAqRbcFWohJWt7PHsFEJfZAvw9ZxwQoDa4SoMgsDDM1T7WK3u9E4edkC4ugRnZ8E4xDZRpk8Rnts3Nbt97dPwT52WRONGBADWRONG'))
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_xpub_type(self):
|
|
for xprv_details in self.xprv_xpub:
|
|
xpub = xprv_details['xpub']
|
|
self.assertEqual(xprv_details['xtype'], xpub_type(xpub))
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_is_xprv(self):
|
|
for xprv_details in self.xprv_xpub:
|
|
xprv = xprv_details['xprv']
|
|
self.assertTrue(is_xprv(xprv))
|
|
self.assertFalse(is_xprv('xprv1nval1d'))
|
|
self.assertFalse(is_xprv('xprv661MyMwAqRbcFWohJWt7PHsFEJfZAvw9ZxwQoDa4SoMgsDDM1T7WK3u9E4edkC4ugRnZ8E4xDZRpk8Rnts3Nbt97dPwT52WRONGBADWRONG'))
|
|
|
|
def test_is_bip32_derivation(self):
|
|
self.assertTrue(is_bip32_derivation("m/0'/1"))
|
|
self.assertTrue(is_bip32_derivation("m/0'/0'"))
|
|
self.assertTrue(is_bip32_derivation("m/44'/0'/0'/0/0"))
|
|
self.assertTrue(is_bip32_derivation("m/49'/0'/0'/0/0"))
|
|
self.assertFalse(is_bip32_derivation("mmmmmm"))
|
|
self.assertFalse(is_bip32_derivation("n/"))
|
|
self.assertFalse(is_bip32_derivation(""))
|
|
self.assertFalse(is_bip32_derivation("m/q8462"))
|
|
|
|
def test_convert_bip32_path_to_list_of_uint32(self):
|
|
self.assertEqual([0, 0x80000001, 0x80000001], convert_bip32_path_to_list_of_uint32("m/0/-1/1'"))
|
|
self.assertEqual([], convert_bip32_path_to_list_of_uint32("m/"))
|
|
self.assertEqual([2147483692, 2147488889, 221], convert_bip32_path_to_list_of_uint32("m/44'/5241'/221"))
|
|
|
|
def test_xtype_from_derivation(self):
|
|
self.assertEqual('standard', xtype_from_derivation("m/44'"))
|
|
self.assertEqual('standard', xtype_from_derivation("m/44'/"))
|
|
self.assertEqual('standard', xtype_from_derivation("m/44'/0'/0'"))
|
|
self.assertEqual('standard', xtype_from_derivation("m/44'/5241'/221"))
|
|
self.assertEqual('standard', xtype_from_derivation("m/45'"))
|
|
self.assertEqual('standard', xtype_from_derivation("m/45'/56165/271'"))
|
|
self.assertEqual('p2wpkh-p2sh', xtype_from_derivation("m/49'"))
|
|
self.assertEqual('p2wpkh-p2sh', xtype_from_derivation("m/49'/134"))
|
|
self.assertEqual('p2wpkh', xtype_from_derivation("m/84'"))
|
|
self.assertEqual('p2wpkh', xtype_from_derivation("m/84'/112'/992/112/33'/0/2"))
|
|
self.assertEqual('p2wsh-p2sh', xtype_from_derivation("m/48'/0'/0'/1'"))
|
|
self.assertEqual('p2wsh-p2sh', xtype_from_derivation("m/48'/0'/0'/1'/52112/52'"))
|
|
self.assertEqual('p2wsh-p2sh', xtype_from_derivation("m/48'/9'/2'/1'"))
|
|
self.assertEqual('p2wsh', xtype_from_derivation("m/48'/0'/0'/2'"))
|
|
self.assertEqual('p2wsh', xtype_from_derivation("m/48'/1'/0'/2'/77'/0"))
|
|
|
|
def test_version_bytes(self):
|
|
xprv_headers_b58 = {
|
|
'standard': 'xprv',
|
|
'p2wpkh-p2sh': 'yprv',
|
|
'p2wsh-p2sh': 'Yprv',
|
|
'p2wpkh': 'zprv',
|
|
'p2wsh': 'Zprv',
|
|
}
|
|
xpub_headers_b58 = {
|
|
'standard': 'xpub',
|
|
'p2wpkh-p2sh': 'ypub',
|
|
'p2wsh-p2sh': 'Ypub',
|
|
'p2wpkh': 'zpub',
|
|
'p2wsh': 'Zpub',
|
|
}
|
|
for xtype, xkey_header_bytes in constants.net.XPRV_HEADERS.items():
|
|
xkey_header_bytes = bfh("%08x" % xkey_header_bytes)
|
|
xkey_bytes = xkey_header_bytes + bytes([0] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xprv_headers_b58[xtype]))
|
|
|
|
xkey_bytes = xkey_header_bytes + bytes([255] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xprv_headers_b58[xtype]))
|
|
|
|
for xtype, xkey_header_bytes in constants.net.XPUB_HEADERS.items():
|
|
xkey_header_bytes = bfh("%08x" % xkey_header_bytes)
|
|
xkey_bytes = xkey_header_bytes + bytes([0] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xpub_headers_b58[xtype]))
|
|
|
|
xkey_bytes = xkey_header_bytes + bytes([255] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xpub_headers_b58[xtype]))
|
|
|
|
|
|
class Test_xprv_xpub_testnet(TestCaseForTestnet):
|
|
|
|
def test_version_bytes(self):
|
|
xprv_headers_b58 = {
|
|
'standard': 'tprv',
|
|
'p2wpkh-p2sh': 'uprv',
|
|
'p2wsh-p2sh': 'Uprv',
|
|
'p2wpkh': 'vprv',
|
|
'p2wsh': 'Vprv',
|
|
}
|
|
xpub_headers_b58 = {
|
|
'standard': 'tpub',
|
|
'p2wpkh-p2sh': 'upub',
|
|
'p2wsh-p2sh': 'Upub',
|
|
'p2wpkh': 'vpub',
|
|
'p2wsh': 'Vpub',
|
|
}
|
|
for xtype, xkey_header_bytes in constants.net.XPRV_HEADERS.items():
|
|
xkey_header_bytes = bfh("%08x" % xkey_header_bytes)
|
|
xkey_bytes = xkey_header_bytes + bytes([0] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xprv_headers_b58[xtype]))
|
|
|
|
xkey_bytes = xkey_header_bytes + bytes([255] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xprv_headers_b58[xtype]))
|
|
|
|
for xtype, xkey_header_bytes in constants.net.XPUB_HEADERS.items():
|
|
xkey_header_bytes = bfh("%08x" % xkey_header_bytes)
|
|
xkey_bytes = xkey_header_bytes + bytes([0] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xpub_headers_b58[xtype]))
|
|
|
|
xkey_bytes = xkey_header_bytes + bytes([255] * 74)
|
|
xkey_b58 = EncodeBase58Check(xkey_bytes)
|
|
self.assertTrue(xkey_b58.startswith(xpub_headers_b58[xtype]))
|
|
|
|
|
|
class Test_keyImport(SequentialTestCase):
|
|
|
|
priv_pub_addr = (
|
|
{'priv': 'KzMFjMC2MPadjvX5Cd7b8AKKjjpBSoRKUTpoAtN6B3J9ezWYyXS6',
|
|
'exported_privkey': 'p2pkh:KzMFjMC2MPadjvX5Cd7b8AKKjjpBSoRKUTpoAtN6B3J9ezWYyXS6',
|
|
'pub': '02c6467b7e621144105ed3e4835b0b4ab7e35266a2ae1c4f8baa19e9ca93452997',
|
|
'address': '17azqT8T16coRmWKYFj3UjzJuxiYrYFRBR',
|
|
'minikey' : False,
|
|
'txin_type': 'p2pkh',
|
|
'compressed': True,
|
|
'addr_encoding': 'base58',
|
|
'scripthash': 'c9aecd1fef8d661a42c560bf75c8163e337099800b8face5ca3d1393a30508a7'},
|
|
{'priv': 'p2pkh:Kzj8VjwpZ99bQqVeUiRXrKuX9mLr1o6sWxFMCBJn1umC38BMiQTD',
|
|
'exported_privkey': 'p2pkh:Kzj8VjwpZ99bQqVeUiRXrKuX9mLr1o6sWxFMCBJn1umC38BMiQTD',
|
|
'pub': '0352d78b4b37e0f6d4e164423436f2925fa57817467178eca550a88f2821973c41',
|
|
'address': '1GXgZ5Qi6gmXTHVSpUPZLy4Ci2nbfb3ZNb',
|
|
'minikey': False,
|
|
'txin_type': 'p2pkh',
|
|
'compressed': True,
|
|
'addr_encoding': 'base58',
|
|
'scripthash': 'a9b2a76fc196c553b352186dfcca81fcf323a721cd8431328f8e9d54216818c1'},
|
|
{'priv': '5Hxn5C4SQuiV6e62A1MtZmbSeQyrLFhu5uYks62pU5VBUygK2KD',
|
|
'exported_privkey': 'p2pkh:5Hxn5C4SQuiV6e62A1MtZmbSeQyrLFhu5uYks62pU5VBUygK2KD',
|
|
'pub': '04e5fe91a20fac945845a5518450d23405ff3e3e1ce39827b47ee6d5db020a9075422d56a59195ada0035e4a52a238849f68e7a325ba5b2247013e0481c5c7cb3f',
|
|
'address': '1GPHVTY8UD9my6jyP4tb2TYJwUbDetyNC6',
|
|
'minikey': False,
|
|
'txin_type': 'p2pkh',
|
|
'compressed': False,
|
|
'addr_encoding': 'base58',
|
|
'scripthash': 'f5914651408417e1166f725a5829ff9576d0dbf05237055bf13abd2af7f79473'},
|
|
{'priv': 'p2pkh:5KhYQCe1xd5g2tqpmmGpUWDpDuTbA8vnpbiCNDwMPAx29WNQYfN',
|
|
'exported_privkey': 'p2pkh:5KhYQCe1xd5g2tqpmmGpUWDpDuTbA8vnpbiCNDwMPAx29WNQYfN',
|
|
'pub': '048f0431b0776e8210376c81280011c2b68be43194cb00bd47b7e9aa66284b713ce09556cde3fee606051a07613f3c159ef3953b8927c96ae3dae94a6ba4182e0e',
|
|
'address': '147kiRHHm9fqeMQSgqf4k35XzuWLP9fmmS',
|
|
'minikey': False,
|
|
'txin_type': 'p2pkh',
|
|
'compressed': False,
|
|
'addr_encoding': 'base58',
|
|
'scripthash': '6dd2e07ad2de9ba8eec4bbe8467eb53f8845acff0d9e6f5627391acc22ff62df'},
|
|
{'priv': 'LHJnnvRzsdrTX2j5QeWVsaBkabK7gfMNqNNqxnbBVRaJYfk24iJz',
|
|
'exported_privkey': 'p2wpkh-p2sh:Kz9XebiCXL2BZzhYJViiHDzn5iup1povWV8aqstzWU4sz1K5nVva',
|
|
'pub': '0279ad237ca0d812fb503ab86f25e15ebd5fa5dd95c193639a8a738dcd1acbad81',
|
|
'address': '3GeVJB3oKr7psgKR6BTXSxKtWUkfsHHhk7',
|
|
'minikey': False,
|
|
'txin_type': 'p2wpkh-p2sh',
|
|
'compressed': True,
|
|
'addr_encoding': 'base58',
|
|
'scripthash': 'd7b04e882fa6b13246829ac552a2b21461d9152eb00f0a6adb58457a3e63d7c5'},
|
|
{'priv': 'p2wpkh-p2sh:L3CZH1pm87X4bbE6mSGvZnAZ1KcFDRomBudUkrkBG7EZhDtBVXMW',
|
|
'exported_privkey': 'p2wpkh-p2sh:L3CZH1pm87X4bbE6mSGvZnAZ1KcFDRomBudUkrkBG7EZhDtBVXMW',
|
|
'pub': '0229da20a15b3363b2c28e3c5093c180b56c439df0b968a970366bb1f38435361e',
|
|
'address': '3C79goMwT7zSTjXnPoCg6VFGAnUpZAkyus',
|
|
'minikey': False,
|
|
'txin_type': 'p2wpkh-p2sh',
|
|
'compressed': True,
|
|
'addr_encoding': 'base58',
|
|
'scripthash': '714bf6bfe1083e69539f40d4c7a7dca85d187471b35642e55f20d7e866494cf7'},
|
|
{'priv': 'L8g5V8kFFeg2WbecahRSdobARbHz2w2STH9S8ePHVSY4fmia7Rsj',
|
|
'exported_privkey': 'p2wpkh:Kz6SuyPM5VktY5dr2d2YqdVgBA6LCWkiHqXJaC3BzxnMPSUuYzmF',
|
|
'pub': '03e9f948421aaa89415dc5f281a61b60dde12aae3181b3a76cd2d849b164fc6d0b',
|
|
'address': 'bc1qqmpt7u5e9hfznljta5gnvhyvfd2kdd0r90hwue',
|
|
'minikey': False,
|
|
'txin_type': 'p2wpkh',
|
|
'compressed': True,
|
|
'addr_encoding': 'bech32',
|
|
'scripthash': '1929acaaef3a208c715228e9f1ca0318e3a6b9394ab53c8d026137f847ecf97b'},
|
|
{'priv': 'p2wpkh:KyDWy5WbjLA58Zesh1o8m3pADGdJ3v33DKk4m7h8BD5zDKDmDFwo',
|
|
'exported_privkey': 'p2wpkh:KyDWy5WbjLA58Zesh1o8m3pADGdJ3v33DKk4m7h8BD5zDKDmDFwo',
|
|
'pub': '038c57657171c1f73e34d5b3971d05867d50221ad94980f7e87cbc2344425e6a1e',
|
|
'address': 'bc1qpakeeg4d9ydyjxd8paqrw4xy9htsg532xzxn50',
|
|
'minikey': False,
|
|
'txin_type': 'p2wpkh',
|
|
'compressed': True,
|
|
'addr_encoding': 'bech32',
|
|
'scripthash': '242f02adde84ebb2a7dd778b2f3a81b3826f111da4d8960d826d7a4b816cb261'},
|
|
# from http://bitscan.com/articles/security/spotlight-on-mini-private-keys
|
|
{'priv': 'SzavMBLoXU6kDrqtUVmffv',
|
|
'exported_privkey': 'p2pkh:5Kb8kLf9zgWQnogidDA76MzPL6TsZZY36hWXMssSzNydYXYB9KF',
|
|
'pub': '04588d202afcc1ee4ab5254c7847ec25b9a135bbda0f2bc69ee1a714749fd77dc9f88ff2a00d7e752d44cbe16e1ebcf0890b76ec7c78886109dee76ccfc8445424',
|
|
'address': '1CC3X2gu58d6wXUWMffpuzN9JAfTUWu4Kj',
|
|
'minikey': True,
|
|
'txin_type': 'p2pkh',
|
|
'compressed': False, # this is actually ambiguous... issue #2748
|
|
'addr_encoding': 'base58',
|
|
'scripthash': '5b07ddfde826f5125ee823900749103cea37808038ecead5505a766a07c34445'},
|
|
)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_public_key_from_private_key(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
txin_type, privkey, compressed = deserialize_privkey(priv_details['priv'])
|
|
result = ecc.ECPrivkey(privkey).get_public_key_hex(compressed=compressed)
|
|
self.assertEqual(priv_details['pub'], result)
|
|
self.assertEqual(priv_details['txin_type'], txin_type)
|
|
self.assertEqual(priv_details['compressed'], compressed)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_address_from_private_key(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
addr2 = address_from_private_key(priv_details['priv'])
|
|
self.assertEqual(priv_details['address'], addr2)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_is_valid_address(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
addr = priv_details['address']
|
|
self.assertFalse(is_address(priv_details['priv']))
|
|
self.assertFalse(is_address(priv_details['pub']))
|
|
self.assertTrue(is_address(addr))
|
|
|
|
is_enc_b58 = priv_details['addr_encoding'] == 'base58'
|
|
self.assertEqual(is_enc_b58, is_b58_address(addr))
|
|
|
|
is_enc_bech32 = priv_details['addr_encoding'] == 'bech32'
|
|
self.assertEqual(is_enc_bech32, is_segwit_address(addr))
|
|
|
|
self.assertFalse(is_address("not an address"))
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_is_private_key(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
self.assertTrue(is_private_key(priv_details['priv']))
|
|
self.assertTrue(is_private_key(priv_details['exported_privkey']))
|
|
self.assertFalse(is_private_key(priv_details['pub']))
|
|
self.assertFalse(is_private_key(priv_details['address']))
|
|
self.assertFalse(is_private_key("not a privkey"))
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_serialize_privkey(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
txin_type, privkey, compressed = deserialize_privkey(priv_details['priv'])
|
|
priv2 = serialize_privkey(privkey, compressed, txin_type)
|
|
self.assertEqual(priv_details['exported_privkey'], priv2)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_address_to_scripthash(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
sh = address_to_scripthash(priv_details['address'])
|
|
self.assertEqual(priv_details['scripthash'], sh)
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_is_minikey(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
minikey = priv_details['minikey']
|
|
priv = priv_details['priv']
|
|
self.assertEqual(minikey, is_minikey(priv))
|
|
|
|
@needs_test_with_all_ecc_implementations
|
|
def test_is_compressed(self):
|
|
for priv_details in self.priv_pub_addr:
|
|
self.assertEqual(priv_details['compressed'],
|
|
is_compressed(priv_details['priv']))
|
|
|
|
|
|
class Test_seeds(SequentialTestCase):
|
|
""" Test old and new seeds. """
|
|
|
|
mnemonics = {
|
|
('cell dumb heartbeat north boom tease ship baby bright kingdom rare squeeze', 'old'),
|
|
('cell dumb heartbeat north boom tease ' * 4, 'old'),
|
|
('cell dumb heartbeat north boom tease ship baby bright kingdom rare badword', ''),
|
|
('cElL DuMb hEaRtBeAt nOrTh bOoM TeAsE ShIp bAbY BrIgHt kInGdOm rArE SqUeEzE', 'old'),
|
|
(' cElL DuMb hEaRtBeAt nOrTh bOoM TeAsE ShIp bAbY BrIgHt kInGdOm rArE SqUeEzE ', 'old'),
|
|
# below seed is actually 'invalid old' as it maps to 33 hex chars
|
|
('hurry idiot prefer sunset mention mist jaw inhale impossible kingdom rare squeeze', 'old'),
|
|
('cram swing cover prefer miss modify ritual silly deliver chunk behind inform able', 'standard'),
|
|
('cram swing cover prefer miss modify ritual silly deliver chunk behind inform', ''),
|
|
('ostrich security deer aunt climb inner alpha arm mutual marble solid task', 'standard'),
|
|
('OSTRICH SECURITY DEER AUNT CLIMB INNER ALPHA ARM MUTUAL MARBLE SOLID TASK', 'standard'),
|
|
(' oStRiCh sEcUrItY DeEr aUnT ClImB InNeR AlPhA ArM MuTuAl mArBlE SoLiD TaSk ', 'standard'),
|
|
('x8', 'standard'),
|
|
('science dawn member doll dutch real can brick knife deny drive list', '2fa'),
|
|
('science dawn member doll dutch real ca brick knife deny drive list', ''),
|
|
(' sCience dawn member doll Dutch rEAl can brick knife deny drive lisT', '2fa'),
|
|
('frost pig brisk excite novel report camera enlist axis nation novel desert', 'segwit'),
|
|
(' fRoSt pig brisk excIte novel rePort CamEra enlist axis nation nOVeL dEsert ', 'segwit'),
|
|
('9dk', 'segwit'),
|
|
}
|
|
|
|
def test_new_seed(self):
|
|
seed = "cram swing cover prefer miss modify ritual silly deliver chunk behind inform able"
|
|
self.assertTrue(is_new_seed(seed))
|
|
|
|
seed = "cram swing cover prefer miss modify ritual silly deliver chunk behind inform"
|
|
self.assertFalse(is_new_seed(seed))
|
|
|
|
def test_old_seed(self):
|
|
self.assertTrue(is_old_seed(" ".join(["like"] * 12)))
|
|
self.assertFalse(is_old_seed(" ".join(["like"] * 18)))
|
|
self.assertTrue(is_old_seed(" ".join(["like"] * 24)))
|
|
self.assertFalse(is_old_seed("not a seed"))
|
|
|
|
self.assertTrue(is_old_seed("0123456789ABCDEF" * 2))
|
|
self.assertTrue(is_old_seed("0123456789ABCDEF" * 4))
|
|
|
|
def test_seed_type(self):
|
|
for seed_words, _type in self.mnemonics:
|
|
self.assertEqual(_type, seed_type(seed_words), msg=seed_words)
|