forked from petertodd/checklocktimeverify-demos
-
Notifications
You must be signed in to change notification settings - Fork 0
/
hodl.py
executable file
·158 lines (121 loc) · 4.82 KB
/
hodl.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
#!/usr/bin/python3
# Copyright (C) 2015 Peter Todd <pete@petertodd.org>
#
# This file is subject to the license terms in the LICENSE file found in the
# top-level directory of this distribution.
import argparse
import binascii
import bitcoin
import bitcoin.rpc
import logging
import math
import bitcoin.rpc
from bitcoin.core import (
b2x, b2lx, lx,
str_money_value, COIN,
COutPoint, CTxIn, CTxOut, CTransaction,
)
from bitcoin.core.script import (
OP_NOP2, OP_DROP, OP_CHECKSIG,
CScript,
SignatureHash, SIGHASH_ALL,
)
from bitcoin.wallet import P2SHBitcoinAddress, CBitcoinSecret, CBitcoinAddress
parser = argparse.ArgumentParser(description="hodl your bitcoins with CHECKLOCKTIMEVERIFY")
parser.add_argument('-v', action='store_true',
dest='verbose',
help='Verbose')
parser.add_argument('-t', action='store_true',
dest='testnet',
help='Enable testnet')
parser.add_argument('privkey', action='store',
help='Private key')
parser.add_argument('nLockTime', action='store', type=int,
help='nLockTime')
subparsers = parser.add_subparsers(title='Subcommands',
description='All operations are done through subcommands:')
def hodl_redeemScript(privkey, nLockTime):
return CScript([nLockTime, OP_NOP2, OP_DROP,
privkey.pub, OP_CHECKSIG])
def spend_hodl_redeemScript(privkey, nLockTime, unsigned_tx, n):
"""Spend a hodl output
Returns the complete scriptSig
"""
redeemScript = hodl_redeemScript(privkey, nLockTime)
sighash = SignatureHash(redeemScript, unsigned_tx, n, SIGHASH_ALL)
sig = privkey.sign(sighash) + bytes([SIGHASH_ALL])
return CScript([sig, redeemScript])
# ----- create -----
parser_create = subparsers.add_parser('create',
help='Create an address for hodling')
def create_command(args):
redeemScript = hodl_redeemScript(args.privkey, args.nLockTime)
logging.debug('redeemScript: %s' % b2x(redeemScript))
addr = P2SHBitcoinAddress.from_redeemScript(redeemScript)
print(addr)
parser_create.set_defaults(cmd_func=create_command)
# ----- spend -----
parser_spend = subparsers.add_parser('spend',
help='Spend (all) your hodled coins')
parser_spend.add_argument('prevouts', nargs='+',
metavar='txid:n',
help='Transaction output')
parser_spend.add_argument('addr', action='store',
help='Address to send the funds too')
def spend_command(args):
args.addr = CBitcoinAddress(args.addr)
redeemScript = hodl_redeemScript(args.privkey, args.nLockTime)
scriptPubKey = redeemScript.to_p2sh_scriptPubKey()
proxy = bitcoin.rpc.Proxy()
prevouts = []
for prevout in args.prevouts:
try:
txid,n = prevout.split(':')
txid = lx(txid)
n = int(n)
outpoint = COutPoint(txid, n)
except ValueError:
args.parser.error('Invalid output: %s' % prevout)
try:
prevout = proxy.gettxout(outpoint)
except IndexError:
args.parser.error('Outpoint %s not found' % outpoint)
prevout = prevout['txout']
if prevout.scriptPubKey != scriptPubKey:
args.parser.error('Outpoint not correct scriptPubKey')
prevouts.append((outpoint, prevout))
sum_in = sum(prev_txout.nValue for outpoint,prev_txout in prevouts)
tx_size = (4 + # version field
2 + # # of txins
len(prevouts) * 153 + # txins, including sigs
1 + # # of txouts
34 + # txout
4 # nLockTime field
)
feerate = int(proxy._call('estimatefee', 1) * COIN) # satoshi's per KB
if feerate <= 0:
feerate = 10000
fees = int(tx_size / 1000 * feerate)
unsigned_tx = CTransaction([CTxIn(outpoint, nSequence=0) for outpoint, prevout in prevouts],
[CTxOut(sum_in - fees,
args.addr.to_scriptPubKey())],
args.nLockTime)
signed_tx = CTransaction(
[CTxIn(txin.prevout,
spend_hodl_redeemScript(args.privkey, args.nLockTime, unsigned_tx, i),
nSequence=0)
for i, txin in enumerate(unsigned_tx.vin)],
unsigned_tx.vout,
unsigned_tx.nLockTime)
print(b2x(signed_tx.serialize()))
parser_spend.set_defaults(cmd_func=spend_command)
args = parser.parse_args()
args.parser = parser
if args.verbose:
logging.root.setLevel('DEBUG')
if args.testnet:
bitcoin.SelectParams('testnet')
args.privkey = CBitcoinSecret(args.privkey)
if not hasattr(args, 'cmd_func'):
parser.error('No command specified')
args.cmd_func(args)