forked from JoinMarket-Org/joinmarket
/
patientsendpayment.py
249 lines (222 loc) · 8.87 KB
/
patientsendpayment.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
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
from __future__ import absolute_import
import sys
import threading
import time
from datetime import timedelta
from optparse import OptionParser
# data_dir = os.path.dirname(os.path.realpath(__file__))
# sys.path.insert(0, os.path.join(data_dir, 'joinmarket'))
from joinmarket import Maker, Taker, load_program_config, IRCMessageChannel
from joinmarket import validate_address, jm_single
from joinmarket import random_nick
from joinmarket import get_log, choose_orders, weighted_order_choose, \
debug_dump_object
from joinmarket import Wallet
log = get_log()
class TakerThread(threading.Thread):
def __init__(self, tmaker):
threading.Thread.__init__(self)
self.daemon = True
self.tmaker = tmaker
self.finished = False
def finishcallback(self, coinjointx):
self.tmaker.msgchan.shutdown()
def run(self):
# TODO this thread doesnt wake up for what could be hours
# need a loop that periodically checks self.finished
# TODO another issue is, what if the bot has run out of utxos and
# needs to wait for some tx to confirm before it can trade
# presumably it needs to wait here until the tx confirms
time.sleep(self.tmaker.waittime)
if self.finished:
return
print 'giving up waiting'
# cancel the remaining order
self.tmaker.modify_orders([0], [])
orders, total_cj_fee = choose_orders(self.tmaker.db, self.tmaker.amount,
self.tmaker.makercount,
weighted_order_choose)
print 'chosen orders to fill ' + str(orders) + ' totalcjfee=' + str(
total_cj_fee)
total_amount = self.tmaker.amount + total_cj_fee + self.tmaker.txfee
print 'total amount spent = ' + str(total_amount)
utxos = self.tmaker.wallet.select_utxos(self.tmaker.mixdepth,
total_amount)
self.tmaker.start_cj(
self.tmaker.wallet, self.tmaker.amount, orders, utxos,
self.tmaker.destaddr,
self.tmaker.wallet.get_internal_addr(self.tmaker.mixdepth),
self.tmaker.txfee, self.finishcallback)
class PatientSendPayment(Maker, Taker):
def __init__(self, msgchan, wallet, destaddr, amount, makercount, txfee,
cjfee, waittime, mixdepth):
self.destaddr = destaddr
self.amount = amount
self.makercount = makercount
self.txfee = txfee
self.cjfee = cjfee
self.waittime = waittime
self.mixdepth = mixdepth
Maker.__init__(self, msgchan, wallet)
Taker.__init__(self, msgchan)
def get_crypto_box_from_nick(self, nick):
if self.cjtx:
return Taker.get_crypto_box_from_nick(self, nick)
else:
return Maker.get_crypto_box_from_nick(self, nick)
def on_welcome(self):
Maker.on_welcome(self)
Taker.on_welcome(self)
self.takerthread = TakerThread(self)
self.takerthread.start()
def create_my_orders(self):
# choose an absolute fee order to discourage people from
# mixing smaller amounts
order = {'oid': 0,
'ordertype': 'absorder',
'minsize': 0,
'maxsize': self.amount,
'txfee': self.txfee,
'cjfee': self.cjfee}
return [order]
def oid_to_order(self, cjorder, oid, amount):
# TODO race condition (kinda)
# if an order arrives and before it finishes another order arrives
# its possible this bot will end up paying to the destaddr more than it
# intended
utxos = self.wallet.select_utxos(self.mixdepth, amount)
return utxos, self.destaddr, self.wallet.get_internal_addr(self.mixdepth)
def on_tx_unconfirmed(self, cjorder, balance, removed_utxos):
self.amount -= cjorder.cj_amount
if self.amount == 0:
self.takerthread.finished = True
print 'finished sending, exiting..'
self.msgchan.shutdown()
return [], []
available_balance = self.wallet.get_balance_by_mixdepth()[self.mixdepth]
if available_balance >= self.amount:
order = {'oid': 0,
'ordertype': 'absorder',
'minsize': 0,
'maxsize': self.amount,
'txfee': self.txfee,
'cjfee': self.cjfee}
return [], [order]
else:
log.debug('not enough money left, have to wait until tx confirms')
return [0], []
def on_tx_confirmed(self, cjorder, confirmations, txid, balance):
if len(self.orderlist) == 0:
order = {'oid': 0,
'ordertype': 'absorder',
'minsize': 0,
'maxsize': self.amount,
'txfee': self.txfee,
'cjfee': self.cjfee}
return [], [order]
else:
return [], []
def main():
parser = OptionParser(
usage=
'usage: %prog [options] [wallet file / fromaccount] [amount] [destaddr]',
description='Sends a payment from your wallet to an given address' +
' using coinjoin. First acts as a maker, announcing an order and ' +
'waiting for someone to fill it. After a set period of time, gives' +
' up waiting and acts as a taker and coinjoins any remaining coins')
parser.add_option(
'-f',
'--txfee',
action='store',
type='int',
dest='txfee',
default=10000,
help='miner fee contribution, in satoshis, default=10000')
parser.add_option(
'-N',
'--makercount',
action='store',
type='int',
dest='makercount',
help=
'how many makers to coinjoin with when taking liquidity, default=2',
default=2)
parser.add_option(
'-w',
'--wait-time',
action='store',
type='float',
dest='waittime',
help='wait time in hours as a maker before becoming a taker, default=8',
default=8)
parser.add_option(
'-c',
'--cjfee',
action='store',
type='int',
dest='cjfee',
help=
'coinjoin fee asked for when being a maker, in satoshis per order filled, default=50000',
default=50000)
parser.add_option(
'-m',
'--mixdepth',
action='store',
type='int',
dest='mixdepth',
help='mixing depth to spend from, default=0',
default=0)
parser.add_option(
'--rpcwallet',
action='store_true',
dest='userpcwallet',
default=False,
help=
'Use the Bitcoin Core wallet through json rpc, instead of the internal joinmarket '
+ 'wallet. Requires blockchain_source=json-rpc')
(options, args) = parser.parse_args()
if len(args) < 3:
parser.error('Needs a wallet, amount and destination address')
sys.exit(0)
wallet_name = args[0]
amount = int(args[1])
destaddr = args[2]
load_program_config()
addr_valid, errormsg = validate_address(destaddr)
if not addr_valid:
print 'ERROR: Address invalid. ' + errormsg
return
waittime = timedelta(hours=options.waittime).total_seconds()
print 'txfee=%d cjfee=%d waittime=%s makercount=%d' % (
options.txfee, options.cjfee, str(timedelta(hours=options.waittime)),
options.makercount)
# todo: this section doesn't make a lot of sense
if not options.userpcwallet:
wallet = Wallet(wallet_name, options.mixdepth + 1)
else:
print 'not implemented yet'
sys.exit(0)
# wallet = BitcoinCoreWallet(fromaccount=wallet_name)
jm_single().bc_interface.sync_wallet(wallet)
available_balance = wallet.get_balance_by_mixdepth()[options.mixdepth]
if available_balance < amount:
print 'not enough money at mixdepth=%d, exiting' % options.mixdepth
return
jm_single().nickname = random_nick()
log.debug('Running patient sender of a payment')
irc = IRCMessageChannel(jm_single().nickname)
PatientSendPayment(irc, wallet, destaddr, amount, options.makercount,
options.txfee, options.cjfee, waittime,
options.mixdepth)
try:
irc.run()
except:
log.debug('CRASHING, DUMPING EVERYTHING')
debug_dump_object(wallet, ['addr_cache', 'keys', 'seed'])
# todo: looks wrong. dump on the class object?
# debug_dump_object(taker)
import traceback
traceback.print_exc()
if __name__ == "__main__":
main()
print('done')