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.
102 lines
3.8 KiB
102 lines
3.8 KiB
#!/usr/bin/env python3 |
|
# Copyright (c) 2018 The Bitcoin Core developers |
|
# Distributed under the MIT software license, see the accompanying |
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php. |
|
|
|
import os |
|
import sys |
|
import argparse |
|
import json |
|
|
|
def perform_pre_checks(): |
|
mock_result_path = os.path.join(os.getcwd(), "mock_result") |
|
if(os.path.isfile(mock_result_path)): |
|
with open(mock_result_path, "r", encoding="utf8") as f: |
|
mock_result = f.read() |
|
if mock_result[0]: |
|
sys.stdout.write(mock_result[2:]) |
|
sys.exit(int(mock_result[0])) |
|
|
|
def enumerate(args): |
|
sys.stdout.write(json.dumps([{"fingerprint": "00000001", "type": "trezor", "model": "trezor_t"}, {"fingerprint": "00000002"}])) |
|
|
|
def getdescriptors(args): |
|
xpub = "tpubD6NzVbkrYhZ4WaWSyoBvQwbpLkojyoTZPRsgXELWz3Popb3qkjcJyJUGLnL4qHHoQvao8ESaAstxYSnhyswJ76uZPStJRJCTKvosUCJZL5B" |
|
|
|
sys.stdout.write(json.dumps({ |
|
"receive": [ |
|
"pkh([00000001/44'/1'/" + args.account + "']" + xpub + "/0/*)#vt6w3l3j", |
|
"sh(wpkh([00000001/49'/1'/" + args.account + "']" + xpub + "/0/*))#r0grqw5x", |
|
"wpkh([00000001/84'/1'/" + args.account + "']" + xpub + "/0/*)#x30uthjs" |
|
], |
|
"internal": [ |
|
"pkh([00000001/44'/1'/" + args.account + "']" + xpub + "/1/*)#all0v2p2", |
|
"sh(wpkh([00000001/49'/1'/" + args.account + "']" + xpub + "/1/*))#kwx4c3pe", |
|
"wpkh([00000001/84'/1'/" + args.account + "']" + xpub + "/1/*)#h92akzzg" |
|
] |
|
})) |
|
|
|
|
|
def displayaddress(args): |
|
# Several descriptor formats are acceptable, so allowing for potential |
|
# changes to InferDescriptor: |
|
if args.fingerprint != "00000001": |
|
return sys.stdout.write(json.dumps({"error": "Unexpected fingerprint", "fingerprint": args.fingerprint})) |
|
|
|
expected_desc = [ |
|
"wpkh([00000001/84'/1'/0'/0/0]02c97dc3f4420402e01a113984311bf4a1b8de376cac0bdcfaf1b3ac81f13433c7)#0yneg42r" |
|
] |
|
if args.desc not in expected_desc: |
|
return sys.stdout.write(json.dumps({"error": "Unexpected descriptor", "desc": args.desc})) |
|
|
|
return sys.stdout.write(json.dumps({"address": "bcrt1qm90ugl4d48jv8n6e5t9ln6t9zlpm5th68x4f8g"})) |
|
|
|
def signtx(args): |
|
if args.fingerprint != "00000001": |
|
return sys.stdout.write(json.dumps({"error": "Unexpected fingerprint", "fingerprint": args.fingerprint})) |
|
|
|
with open(os.path.join(os.getcwd(), "mock_psbt"), "r", encoding="utf8") as f: |
|
mock_psbt = f.read() |
|
|
|
if args.fingerprint == "00000001" : |
|
sys.stdout.write(json.dumps({ |
|
"psbt": mock_psbt, |
|
"complete": True |
|
})) |
|
else: |
|
sys.stdout.write(json.dumps({"psbt": args.psbt})) |
|
|
|
parser = argparse.ArgumentParser(prog='./signer.py', description='External signer mock') |
|
parser.add_argument('--fingerprint') |
|
parser.add_argument('--chain', default='main') |
|
parser.add_argument('--stdin', action='store_true') |
|
|
|
subparsers = parser.add_subparsers(description='Commands', dest='command') |
|
subparsers.required = True |
|
|
|
parser_enumerate = subparsers.add_parser('enumerate', help='list available signers') |
|
parser_enumerate.set_defaults(func=enumerate) |
|
|
|
parser_getdescriptors = subparsers.add_parser('getdescriptors') |
|
parser_getdescriptors.set_defaults(func=getdescriptors) |
|
parser_getdescriptors.add_argument('--account', metavar='account') |
|
|
|
parser_displayaddress = subparsers.add_parser('displayaddress', help='display address on signer') |
|
parser_displayaddress.add_argument('--desc', metavar='desc') |
|
parser_displayaddress.set_defaults(func=displayaddress) |
|
|
|
parser_signtx = subparsers.add_parser('signtx') |
|
parser_signtx.add_argument('psbt', metavar='psbt') |
|
|
|
parser_signtx.set_defaults(func=signtx) |
|
|
|
if not sys.stdin.isatty(): |
|
buffer = sys.stdin.read() |
|
if buffer and buffer.rstrip() != "": |
|
sys.argv.extend(buffer.rstrip().split(" ")) |
|
|
|
args = parser.parse_args() |
|
|
|
perform_pre_checks() |
|
|
|
args.func(args)
|
|
|