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.
80 lines
3.3 KiB
80 lines
3.3 KiB
#! /usr/bin/env python |
|
'''Public and private key validity and formatting tests.''' |
|
|
|
import jmbitcoin as btc |
|
import binascii |
|
import json |
|
import pytest |
|
import os |
|
testdir = os.path.dirname(os.path.realpath(__file__)) |
|
|
|
def test_read_raw_privkeys(): |
|
badkeys = [b'', b'\x07'*31,b'\x07'*34, b'\x07'*33] |
|
for b in badkeys: |
|
with pytest.raises(Exception) as e_info: |
|
c, k = btc.read_privkey(b) |
|
goodkeys = [(b'\x07'*32, False), (b'\x07'*32 + b'\x01', True)] |
|
for g in goodkeys: |
|
c, k = btc.read_privkey(g[0]) |
|
assert c == g[1] |
|
|
|
def test_wif_privkeys_invalid(): |
|
#first try to create wif privkey from key of wrong length |
|
bad_privs = [b'\x01\x02'*17] #some silly private key but > 33 bytes |
|
|
|
#next try to create wif with correct length but wrong compression byte |
|
bad_privs.append(b'\x07'*32 + b'\x02') |
|
|
|
for priv in bad_privs: |
|
with pytest.raises(Exception) as e_info: |
|
fake_wif = btc.wif_compressed_privkey(binascii.hexlify(priv).decode('ascii')) |
|
|
|
#Create a wif with wrong length |
|
bad_wif1 = btc.bin_to_b58check(b'\x01\x02'*34, b'\x80') |
|
#Create a wif with wrong compression byte |
|
bad_wif2 = btc.bin_to_b58check(b'\x07'*33, b'\x80') |
|
for bw in [bad_wif1, bad_wif2]: |
|
with pytest.raises(Exception) as e_info: |
|
fake_priv = btc.from_wif_privkey(bw) |
|
|
|
#Some invalid b58 from bitcoin repo; |
|
#none of these are valid as any kind of key or address |
|
with open(os.path.join(testdir,"base58_keys_invalid.json"), "r") as f: |
|
json_data = f.read() |
|
invalid_key_list = json.loads(json_data) |
|
for k in invalid_key_list: |
|
bad_key = k[0] |
|
for netval in ["mainnet", "testnet"]: |
|
#if using pytest -s ; sanity check to see what's actually being tested |
|
print('testing this key: ' + bad_key) |
|
#should throw exception |
|
with pytest.raises(Exception) as e_info: |
|
from_wif_key = btc.from_wif_privkey(bad_key, |
|
btc.get_version_byte(bad_key)) |
|
#in case the b58 check encoding is valid, we should |
|
#also check if the leading version byte is in the |
|
#expected set, and throw an error if not. |
|
if chr(btc.get_version_byte(bad_key)) not in b'\x80\xef': |
|
raise Exception("Invalid version byte") |
|
|
|
def test_wif_privkeys_valid(): |
|
with open(os.path.join(testdir,"base58_keys_valid.json"), "r") as f: |
|
json_data = f.read() |
|
valid_keys_list = json.loads(json_data) |
|
for a in valid_keys_list: |
|
key, hex_key, prop_dict = a |
|
if prop_dict["isPrivkey"]: |
|
netval = "testnet" if prop_dict["isTestnet"] else "mainnet" |
|
print('testing this key: ' + key) |
|
assert btc.get_version_byte( |
|
key) in b'\x80\xef', "not valid network byte" |
|
comp = prop_dict["isCompressed"] |
|
from_wif_key = btc.from_wif_privkey( |
|
key, |
|
compressed=comp, |
|
vbyte=btc.from_int_to_byte(btc.from_byte_to_int(btc.get_version_byte(key))-128)) |
|
expected_key = hex_key |
|
if comp: expected_key += '01' |
|
assert from_wif_key == expected_key, "Incorrect key decoding: " + \ |
|
str(from_wif_key) + ", should be: " + str(expected_key) |
|
|
|
|