lib/transaction.py

   1 #!/usr/bin/env python
   2 #
   3 # Electrum - lightweight Bitcoin client
   4 # Copyright (C) 2011 thomasv@gitorious
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18
  19
  20 # Note: The deserialization code originally comes from ABE.
  21
  22
  23 from bitcoin import *
  24 from util import print_error
  25 import time
  26 import struct
  27
  28 #
  29 # Workalike python implementation of Bitcoin's CDataStream class.
  30 #
  31 import struct
  32 import StringIO
  33 import mmap
  34
  35 class SerializationError(Exception):
  36     """ Thrown when there's a problem deserializing or serializing """
  37
  38 class BCDataStream(object):
  39     def __init__(self):
  40         self.input = None
  41         self.read_cursor = 0
  42
  43     def clear(self):
  44         self.input = None
  45         self.read_cursor = 0
  46
  47     def write(self, bytes):  # Initialize with string of bytes
  48         if self.input is None:
  49             self.input = bytes
  50         else:
  51             self.input += bytes
  52
  53     def map_file(self, file, start):  # Initialize with bytes from file
  54         self.input = mmap.mmap(file.fileno(), 0, access=mmap.ACCESS_READ)
  55         self.read_cursor = start
  56
  57     def seek_file(self, position):
  58         self.read_cursor = position
  59
  60     def close_file(self):
  61         self.input.close()
  62
  63     def read_string(self):
  64         # Strings are encoded depending on length:
  65         # 0 to 252 :  1-byte-length followed by bytes (if any)
  66         # 253 to 65,535 : byte'253' 2-byte-length followed by bytes
  67         # 65,536 to 4,294,967,295 : byte '254' 4-byte-length followed by bytes
  68         # ... and the Bitcoin client is coded to understand:
  69         # greater than 4,294,967,295 : byte '255' 8-byte-length followed by bytes of string
  70         # ... but I don't think it actually handles any strings that big.
  71         if self.input is None:
  72             raise SerializationError("call write(bytes) before trying to deserialize")
  73
  74         try:
  75             length = self.read_compact_size()
  76         except IndexError:
  77             raise SerializationError("attempt to read past end of buffer")
  78
  79         return self.read_bytes(length)
  80
  81     def write_string(self, string):
  82         # Length-encoded as with read-string
  83         self.write_compact_size(len(string))
  84         self.write(string)
  85
  86     def read_bytes(self, length):
  87         try:
  88             result = self.input[self.read_cursor:self.read_cursor+length]
  89             self.read_cursor += length
  90             return result
  91         except IndexError:
  92             raise SerializationError("attempt to read past end of buffer")
  93
  94         return ''
  95
  96     def read_boolean(self): return self.read_bytes(1)[0] != chr(0)
  97     def read_int16(self): return self._read_num('<h')
  98     def read_uint16(self): return self._read_num('<H')
  99     def read_int32(self): return self._read_num('<i')
 100     def read_uint32(self): return self._read_num('<I')
 101     def read_int64(self): return self._read_num('<q')
 102     def read_uint64(self): return self._read_num('<Q')
 103
 104     def write_boolean(self, val): return self.write(chr(1) if val else chr(0))
 105     def write_int16(self, val): return self._write_num('<h', val)
 106     def write_uint16(self, val): return self._write_num('<H', val)
 107     def write_int32(self, val): return self._write_num('<i', val)
 108     def write_uint32(self, val): return self._write_num('<I', val)
 109     def write_int64(self, val): return self._write_num('<q', val)
 110     def write_uint64(self, val): return self._write_num('<Q', val)
 111
 112     def read_compact_size(self):
 113         size = ord(self.input[self.read_cursor])
 114         self.read_cursor += 1
 115         if size == 253:
 116             size = self._read_num('<H')
 117         elif size == 254:
 118             size = self._read_num('<I')
 119         elif size == 255:
 120             size = self._read_num('<Q')
 121         return size
 122
 123     def write_compact_size(self, size):
 124         if size < 0:
 125             raise SerializationError("attempt to write size < 0")
 126         elif size < 253:
 127             self.write(chr(size))
 128         elif size < 2**16:
 129             self.write('\xfd')
 130             self._write_num('<H', size)
 131         elif size < 2**32:
 132             self.write('\xfe')
 133             self._write_num('<I', size)
 134         elif size < 2**64:
 135             self.write('\xff')
 136             self._write_num('<Q', size)
 137
 138     def _read_num(self, format):
 139         (i,) = struct.unpack_from(format, self.input, self.read_cursor)
 140         self.read_cursor += struct.calcsize(format)
 141         return i
 142
 143     def _write_num(self, format, num):
 144         s = struct.pack(format, num)
 145         self.write(s)
 146
 147 #
 148 # enum-like type
 149 # From the Python Cookbook, downloaded from http://code.activestate.com/recipes/67107/
 150 #
 151 import types, string, exceptions
 152
 153 class EnumException(exceptions.Exception):
 154     pass
 155
 156 class Enumeration:
 157     def __init__(self, name, enumList):
 158         self.__doc__ = name
 159         lookup = { }
 160         reverseLookup = { }
 161         i = 0
 162         uniqueNames = [ ]
 163         uniqueValues = [ ]
 164         for x in enumList:
 165             if type(x) == types.TupleType:
 166                 x, i = x
 167             if type(x) != types.StringType:
 168                 raise EnumException, "enum name is not a string: " + x
 169             if type(i) != types.IntType:
 170                 raise EnumException, "enum value is not an integer: " + i
 171             if x in uniqueNames:
 172                 raise EnumException, "enum name is not unique: " + x
 173             if i in uniqueValues:
 174                 raise EnumException, "enum value is not unique for " + x
 175             uniqueNames.append(x)
 176             uniqueValues.append(i)
 177             lookup[x] = i
 178             reverseLookup[i] = x
 179             i = i + 1
 180         self.lookup = lookup
 181         self.reverseLookup = reverseLookup
 182     def __getattr__(self, attr):
 183         if not self.lookup.has_key(attr):
 184             raise AttributeError
 185         return self.lookup[attr]
 186     def whatis(self, value):
 187         return self.reverseLookup[value]
 188
 189
 190 # This function comes from bitcointools, bct-LICENSE.txt.
 191 def long_hex(bytes):
 192     return bytes.encode('hex_codec')
 193
 194 # This function comes from bitcointools, bct-LICENSE.txt.
 195 def short_hex(bytes):
 196     t = bytes.encode('hex_codec')
 197     if len(t) < 11:
 198         return t
 199     return t[0:4]+"..."+t[-4:]
 200
 201
 202
 203
 204 def parse_redeemScript(bytes):
 205     dec = [ x for x in script_GetOp(bytes.decode('hex')) ]
 206
 207     # 2 of 2
 208     match = [ opcodes.OP_2, opcodes.OP_PUSHDATA4, opcodes.OP_PUSHDATA4, opcodes.OP_2, opcodes.OP_CHECKMULTISIG ]
 209     if match_decoded(dec, match):
 210         pubkeys = [ dec[1][1].encode('hex'), dec[2][1].encode('hex') ]
 211         return 2, pubkeys
 212
 213     # 2 of 3
 214     match = [ opcodes.OP_2, opcodes.OP_PUSHDATA4, opcodes.OP_PUSHDATA4, opcodes.OP_PUSHDATA4, opcodes.OP_3, opcodes.OP_CHECKMULTISIG ]
 215     if match_decoded(dec, match):
 216         pubkeys = [ dec[1][1].encode('hex'), dec[2][1].encode('hex'), dec[3][1].encode('hex') ]
 217         return 2, pubkeys
 218
 219
 220
 221 opcodes = Enumeration("Opcodes", [
 222     ("OP_0", 0), ("OP_PUSHDATA1",76), "OP_PUSHDATA2", "OP_PUSHDATA4", "OP_1NEGATE", "OP_RESERVED",
 223     "OP_1", "OP_2", "OP_3", "OP_4", "OP_5", "OP_6", "OP_7",
 224     "OP_8", "OP_9", "OP_10", "OP_11", "OP_12", "OP_13", "OP_14", "OP_15", "OP_16",
 225     "OP_NOP", "OP_VER", "OP_IF", "OP_NOTIF", "OP_VERIF", "OP_VERNOTIF", "OP_ELSE", "OP_ENDIF", "OP_VERIFY",
 226     "OP_RETURN", "OP_TOALTSTACK", "OP_FROMALTSTACK", "OP_2DROP", "OP_2DUP", "OP_3DUP", "OP_2OVER", "OP_2ROT", "OP_2SWAP",
 227     "OP_IFDUP", "OP_DEPTH", "OP_DROP", "OP_DUP", "OP_NIP", "OP_OVER", "OP_PICK", "OP_ROLL", "OP_ROT",
 228     "OP_SWAP", "OP_TUCK", "OP_CAT", "OP_SUBSTR", "OP_LEFT", "OP_RIGHT", "OP_SIZE", "OP_INVERT", "OP_AND",
 229     "OP_OR", "OP_XOR", "OP_EQUAL", "OP_EQUALVERIFY", "OP_RESERVED1", "OP_RESERVED2", "OP_1ADD", "OP_1SUB", "OP_2MUL",
 230     "OP_2DIV", "OP_NEGATE", "OP_ABS", "OP_NOT", "OP_0NOTEQUAL", "OP_ADD", "OP_SUB", "OP_MUL", "OP_DIV",
 231     "OP_MOD", "OP_LSHIFT", "OP_RSHIFT", "OP_BOOLAND", "OP_BOOLOR",
 232     "OP_NUMEQUAL", "OP_NUMEQUALVERIFY", "OP_NUMNOTEQUAL", "OP_LESSTHAN",
 233     "OP_GREATERTHAN", "OP_LESSTHANOREQUAL", "OP_GREATERTHANOREQUAL", "OP_MIN", "OP_MAX",
 234     "OP_WITHIN", "OP_RIPEMD160", "OP_SHA1", "OP_SHA256", "OP_HASH160",
 235     "OP_HASH256", "OP_CODESEPARATOR", "OP_CHECKSIG", "OP_CHECKSIGVERIFY", "OP_CHECKMULTISIG",
 236     "OP_CHECKMULTISIGVERIFY",
 237     ("OP_SINGLEBYTE_END", 0xF0),
 238     ("OP_DOUBLEBYTE_BEGIN", 0xF000),
 239     "OP_PUBKEY", "OP_PUBKEYHASH",
 240     ("OP_INVALIDOPCODE", 0xFFFF),
 241 ])
 242
 243
 244 def script_GetOp(bytes):
 245     i = 0
 246     while i < len(bytes):
 247         vch = None
 248         opcode = ord(bytes[i])
 249         i += 1
 250         if opcode >= opcodes.OP_SINGLEBYTE_END:
 251             opcode <<= 8
 252             opcode |= ord(bytes[i])
 253             i += 1
 254
 255         if opcode <= opcodes.OP_PUSHDATA4:
 256             nSize = opcode
 257             if opcode == opcodes.OP_PUSHDATA1:
 258                 nSize = ord(bytes[i])
 259                 i += 1
 260             elif opcode == opcodes.OP_PUSHDATA2:
 261                 (nSize,) = struct.unpack_from('<H', bytes, i)
 262                 i += 2
 263             elif opcode == opcodes.OP_PUSHDATA4:
 264                 (nSize,) = struct.unpack_from('<I', bytes, i)
 265                 i += 4
 266             vch = bytes[i:i+nSize]
 267             i += nSize
 268
 269         yield (opcode, vch, i)
 270
 271
 272 def script_GetOpName(opcode):
 273     return (opcodes.whatis(opcode)).replace("OP_", "")
 274
 275
 276 def decode_script(bytes):
 277     result = ''
 278     for (opcode, vch, i) in script_GetOp(bytes):
 279         if len(result) > 0: result += " "
 280         if opcode <= opcodes.OP_PUSHDATA4:
 281             result += "%d:"%(opcode,)
 282             result += short_hex(vch)
 283         else:
 284             result += script_GetOpName(opcode)
 285     return result
 286
 287
 288 def match_decoded(decoded, to_match):
 289     if len(decoded) != len(to_match):
 290         return False;
 291     for i in range(len(decoded)):
 292         if to_match[i] == opcodes.OP_PUSHDATA4 and decoded[i][0] <= opcodes.OP_PUSHDATA4 and decoded[i][0]>0:
 293             continue  # Opcodes below OP_PUSHDATA4 all just push data onto stack, and are equivalent.
 294         if to_match[i] != decoded[i][0]:
 295             return False
 296     return True
 297
 298 def get_address_from_input_script(bytes):
 299     try:
 300         decoded = [ x for x in script_GetOp(bytes) ]
 301     except Exception:
 302         # coinbase transactions raise an exception
 303         print_error("cannot find address in input script", bytes.encode('hex'))
 304         return [], {}, "(None)"
 305
 306     # payto_pubkey
 307     match = [ opcodes.OP_PUSHDATA4 ]
 308     if match_decoded(decoded, match):
 309         return None, {}, "(pubkey)"
 310
 311     # non-generated TxIn transactions push a signature
 312     # (seventy-something bytes) and then their public key
 313     # (65 bytes) onto the stack:
 314     match = [ opcodes.OP_PUSHDATA4, opcodes.OP_PUSHDATA4 ]
 315     if match_decoded(decoded, match):
 316         sig = decoded[0][1].encode('hex')
 317         pubkey = decoded[1][1].encode('hex')
 318         if sig[-2:] == '01':
 319             sig = sig[:-2]
 320             return [pubkey], {pubkey:sig}, public_key_to_bc_address(pubkey.decode('hex'))
 321         else:
 322             print_error("cannot find address in input script", bytes.encode('hex'))
 323             return [], {}, "(None)"
 324
 325
 326     # p2sh transaction, 2 of n
 327     match = [ opcodes.OP_0 ]
 328     while len(match) < len(decoded):
 329         match.append(opcodes.OP_PUSHDATA4)
 330
 331     if match_decoded(decoded, match):
 332
 333         redeemScript = decoded[-1][1]
 334         num = len(match) - 2
 335         signatures = map(lambda x:x[1][:-1].encode('hex'), decoded[1:-1])
 336
 337         dec2 = [ x for x in script_GetOp(redeemScript) ]
 338
 339         # 2 of 2
 340         match2 = [ opcodes.OP_2, opcodes.OP_PUSHDATA4, opcodes.OP_PUSHDATA4, opcodes.OP_2, opcodes.OP_CHECKMULTISIG ]
 341         if match_decoded(dec2, match2):
 342             pubkeys = [ dec2[1][1].encode('hex'), dec2[2][1].encode('hex') ]
 343             return pubkeys, signatures, hash_160_to_bc_address(hash_160(redeemScript), 5)
 344
 345         # 2 of 3
 346         match2 = [ opcodes.OP_2, opcodes.OP_PUSHDATA4, opcodes.OP_PUSHDATA4, opcodes.OP_PUSHDATA4, opcodes.OP_3, opcodes.OP_CHECKMULTISIG ]
 347         if match_decoded(dec2, match2):
 348             pubkeys = [ dec2[1][1].encode('hex'), dec2[2][1].encode('hex'), dec2[3][1].encode('hex') ]
 349             return pubkeys, signatures, hash_160_to_bc_address(hash_160(redeemScript), 5)
 350
 351     print_error("cannot find address in input script", bytes.encode('hex'))
 352     return [], {}, "(None)"
 353
 354
 355
 356 def get_address_from_output_script(bytes):
 357     decoded = [ x for x in script_GetOp(bytes) ]
 358
 359     # The Genesis Block, self-payments, and pay-by-IP-address payments look like:
 360     # 65 BYTES:... CHECKSIG
 361     match = [ opcodes.OP_PUSHDATA4, opcodes.OP_CHECKSIG ]
 362     if match_decoded(decoded, match):
 363         return True, public_key_to_bc_address(decoded[0][1])
 364
 365     # Pay-by-Bitcoin-address TxOuts look like:
 366     # DUP HASH160 20 BYTES:... EQUALVERIFY CHECKSIG
 367     match = [ opcodes.OP_DUP, opcodes.OP_HASH160, opcodes.OP_PUSHDATA4, opcodes.OP_EQUALVERIFY, opcodes.OP_CHECKSIG ]
 368     if match_decoded(decoded, match):
 369         return False, hash_160_to_bc_address(decoded[2][1])
 370
 371     # p2sh
 372     match = [ opcodes.OP_HASH160, opcodes.OP_PUSHDATA4, opcodes.OP_EQUAL ]
 373     if match_decoded(decoded, match):
 374         return False, hash_160_to_bc_address(decoded[1][1],5)
 375
 376     return False, "(None)"
 377
 378
 379 class Transaction:
 380
 381     def __init__(self, raw):
 382         self.raw = raw
 383         self.deserialize()
 384         self.inputs = self.d['inputs']
 385         self.outputs = self.d['outputs']
 386         self.outputs = map(lambda x: (x['address'],x['value']), self.outputs)
 387         self.locktime = self.d['lockTime']
 388
 389     def __str__(self):
 390         return self.raw
 391
 392     @classmethod
 393     def from_io(klass, inputs, outputs):
 394         raw = klass.serialize(inputs, outputs, for_sig = None) # for_sig=-1 means do not sign
 395         self = klass(raw)
 396         self.inputs = inputs
 397         self.outputs = outputs
 398         return self
 399
 400     @classmethod
 401     def sweep(klass, privkeys, network, to_address, fee):
 402         inputs = []
 403         for privkey in privkeys:
 404             pubkey = public_key_from_private_key(privkey)
 405             address = address_from_private_key(privkey)
 406             u = network.synchronous_get([ ('blockchain.address.listunspent',[address])])[0]
 407             pay_script = klass.pay_script(address)
 408             for item in u:
 409                 item['scriptPubKey'] = pay_script
 410                 item['redeemPubkey'] = pubkey
 411                 item['address'] = address
 412                 item['prevout_hash'] = item['tx_hash']
 413                 item['prevout_n'] = item['tx_pos']
 414             inputs += u
 415
 416         if not inputs:
 417             return
 418
 419         total = sum( map(lambda x:int(x.get('value')), inputs) ) - fee
 420         outputs = [(to_address, total)]
 421         self = klass.from_io(inputs, outputs)
 422         self.sign({ pubkey:privkey })
 423         return self
 424
 425     @classmethod
 426     def multisig_script(klass, public_keys, num=None):
 427         n = len(public_keys)
 428         if num is None: num = n
 429         # supports only "2 of 2", and "2 of 3" transactions
 430         assert num <= n and n in [2,3]
 431
 432         if num==2:
 433             s = '52'
 434         elif num == 3:
 435             s = '53'
 436         else:
 437             raise
 438
 439         for k in public_keys:
 440             s += var_int(len(k)/2)
 441             s += k
 442         if n==2:
 443             s += '52'
 444         elif n==3:
 445             s += '53'
 446         else:
 447             raise
 448         s += 'ae'
 449
 450         return s
 451
 452
 453     @classmethod
 454     def pay_script(self, addr):
 455         addrtype, hash_160 = bc_address_to_hash_160(addr)
 456         if addrtype == 0:
 457             script = '76a9'                                      # op_dup, op_hash_160
 458             script += '14'                                       # push 0x14 bytes
 459             script += hash_160.encode('hex')
 460             script += '88ac'                                     # op_equalverify, op_checksig
 461         elif addrtype == 5:
 462             script = 'a9'                                        # op_hash_160
 463             script += '14'                                       # push 0x14 bytes
 464             script += hash_160.encode('hex')
 465             script += '87'                                       # op_equal
 466         else:
 467             raise
 468         return script
 469
 470
 471     @classmethod
 472     def serialize( klass, inputs, outputs, for_sig = None ):
 473
 474         push_script = lambda x: op_push(len(x)/2) + x
 475         s  = int_to_hex(1,4)                                         # version
 476         s += var_int( len(inputs) )                                  # number of inputs
 477         for i in range(len(inputs)):
 478             txin = inputs[i]
 479             s += txin['prevout_hash'].decode('hex')[::-1].encode('hex')   # prev hash
 480             s += int_to_hex(txin['prevout_n'],4)                          # prev index
 481
 482             signatures = txin.get('signatures', {})
 483             if for_sig is None and not signatures:
 484                 script = ''
 485
 486             elif for_sig is None:
 487                 pubkeys = txin['pubkeys']
 488                 sig_list = ''
 489                 for pubkey in pubkeys:
 490                     sig = signatures.get(pubkey)
 491                     if not sig:
 492                         continue
 493                     sig = sig + '01'
 494                     sig_list += push_script(sig)
 495
 496                 if not txin.get('redeemScript'):
 497                     script = sig_list
 498                     script += push_script(pubkeys[0])
 499                 else:
 500                     script = '00'                                    # op_0
 501                     script += sig_list
 502                     redeem_script = klass.multisig_script(pubkeys,2)
 503                     assert redeem_script == txin.get('redeemScript')
 504                     script += push_script(redeem_script)
 505
 506             elif for_sig==i:
 507                 if txin.get('redeemScript'):
 508                     script = txin['redeemScript']                    # p2sh uses the inner script
 509                 else:
 510                     script = txin['scriptPubKey']                    # scriptsig
 511             else:
 512                 script = ''
 513             s += var_int( len(script)/2 )                            # script length
 514             s += script
 515             s += "ffffffff"                                          # sequence
 516
 517         s += var_int( len(outputs) )                                 # number of outputs
 518         for output in outputs:
 519             addr, amount = output
 520             s += int_to_hex( amount, 8)                              # amount
 521             script = klass.pay_script(addr)
 522             s += var_int( len(script)/2 )                           #  script length
 523             s += script                                             #  script
 524         s += int_to_hex(0,4)                                        #  lock time
 525         if for_sig is not None and for_sig != -1:
 526             s += int_to_hex(1, 4)                                   #  hash type
 527         return s
 528
 529
 530     def tx_for_sig(self,i):
 531         return self.serialize(self.inputs, self.outputs, for_sig = i)
 532
 533
 534     def hash(self):
 535         return Hash(self.raw.decode('hex') )[::-1].encode('hex')
 536
 537     def add_signature(self, i, pubkey, sig):
 538         txin = self.inputs[i]
 539         signatures = txin.get("signatures",{})
 540         signatures[pubkey] = sig
 541         txin["signatures"] = signatures
 542         self.inputs[i] = txin
 543         print_error("adding signature for", pubkey)
 544         self.raw = self.serialize( self.inputs, self.outputs )
 545
 546
 547     def is_complete(self):
 548         for i, txin in enumerate(self.inputs):
 549             redeem_script = txin.get('redeemScript')
 550             num, redeem_pubkeys = parse_redeemScript(redeem_script) if redeem_script else (1, [txin.get('redeemPubkey')])
 551             signatures = txin.get("signatures",{})
 552             if len(signatures) == num:
 553                 continue
 554             else:
 555                 return False
 556         return True
 557
 558
 559
 560     def sign(self, keypairs):
 561         print_error("tx.sign(), keypairs:", keypairs)
 562
 563         for i, txin in enumerate(self.inputs):
 564
 565             # if the input is multisig, parse redeem script
 566             redeem_script = txin.get('redeemScript')
 567             num, redeem_pubkeys = parse_redeemScript(redeem_script) if redeem_script else (1, [txin.get('redeemPubkey')])
 568
 569             # add pubkeys
 570             txin["pubkeys"] = redeem_pubkeys
 571             # get list of already existing signatures
 572             signatures = txin.get("signatures",{})
 573             # continue if this txin is complete
 574             if len(signatures) == num:
 575                 continue
 576
 577             for_sig = Hash(self.tx_for_sig(i).decode('hex'))
 578             for pubkey in redeem_pubkeys:
 579                 if pubkey in keypairs.keys():
 580                     # add signature
 581                     sec = keypairs[pubkey]
 582                     pkey = regenerate_key(sec)
 583                     secexp = pkey.secret
 584                     private_key = ecdsa.SigningKey.from_secret_exponent( secexp, curve = SECP256k1 )
 585                     public_key = private_key.get_verifying_key()
 586                     sig = private_key.sign_digest_deterministic( for_sig, hashfunc=hashlib.sha256, sigencode = ecdsa.util.sigencode_der )
 587                     assert public_key.verify_digest( sig, for_sig, sigdecode = ecdsa.util.sigdecode_der)
 588                     self.add_signature(i, pubkey, sig.encode('hex'))
 589
 590
 591         print_error("is_complete", self.is_complete())
 592         self.raw = self.serialize( self.inputs, self.outputs )
 593
 594
 595
 596     def deserialize(self):
 597         vds = BCDataStream()
 598         vds.write(self.raw.decode('hex'))
 599         d = {}
 600         start = vds.read_cursor
 601         d['version'] = vds.read_int32()
 602         n_vin = vds.read_compact_size()
 603         d['inputs'] = []
 604         for i in xrange(n_vin):
 605             d['inputs'].append(self.parse_input(vds))
 606         n_vout = vds.read_compact_size()
 607         d['outputs'] = []
 608         for i in xrange(n_vout):
 609             d['outputs'].append(self.parse_output(vds, i))
 610         d['lockTime'] = vds.read_uint32()
 611         self.d = d
 612         return self.d
 613
 614
 615     def parse_input(self, vds):
 616         d = {}
 617         prevout_hash = hash_encode(vds.read_bytes(32))
 618         prevout_n = vds.read_uint32()
 619         scriptSig = vds.read_bytes(vds.read_compact_size())
 620         sequence = vds.read_uint32()
 621
 622         if prevout_hash == '00'*32:
 623             d['is_coinbase'] = True
 624         else:
 625             d['is_coinbase'] = False
 626             d['prevout_hash'] = prevout_hash
 627             d['prevout_n'] = prevout_n
 628             d['sequence'] = sequence
 629             if scriptSig:
 630                 pubkeys, signatures, address = get_address_from_input_script(scriptSig)
 631             else:
 632                 pubkeys = []
 633                 signatures = {}
 634                 address = None
 635             d['address'] = address
 636             d['pubkeys'] = pubkeys
 637             d['signatures'] = signatures
 638         return d
 639
 640
 641     def parse_output(self, vds, i):
 642         d = {}
 643         d['value'] = vds.read_int64()
 644         scriptPubKey = vds.read_bytes(vds.read_compact_size())
 645         is_pubkey, address = get_address_from_output_script(scriptPubKey)
 646         d['is_pubkey'] = is_pubkey
 647         d['address'] = address
 648         d['scriptPubKey'] = scriptPubKey.encode('hex')
 649         d['prevout_n'] = i
 650         return d
 651
 652
 653     def add_extra_addresses(self, txlist):
 654         for i in self.inputs:
 655             if i.get("address") == "(pubkey)":
 656                 prev_tx = txlist.get(i.get('prevout_hash'))
 657                 if prev_tx:
 658                     address, value = prev_tx.outputs[i.get('prevout_n')]
 659                     print_error("found pay-to-pubkey address:", address)
 660                     i["address"] = address
 661
 662
 663     def has_address(self, addr):
 664         found = False
 665         for txin in self.inputs:
 666             if addr == txin.get('address'):
 667                 found = True
 668                 break
 669         for txout in self.outputs:
 670             if addr == txout[0]:
 671                 found = True
 672                 break
 673         return found
 674
 675
 676     def get_value(self, addresses, prevout_values):
 677         # return the balance for that tx
 678         is_relevant = False
 679         is_send = False
 680         is_pruned = False
 681         is_partial = False
 682         v_in = v_out = v_out_mine = 0
 683
 684         for item in self.inputs:
 685             addr = item.get('address')
 686             if addr in addresses:
 687                 is_send = True
 688                 is_relevant = True
 689                 key = item['prevout_hash']  + ':%d'%item['prevout_n']
 690                 value = prevout_values.get( key )
 691                 if value is None:
 692                     is_pruned = True
 693                 else:
 694                     v_in += value
 695             else:
 696                 is_partial = True
 697
 698         if not is_send: is_partial = False
 699
 700         for item in self.outputs:
 701             addr, value = item
 702             v_out += value
 703             if addr in addresses:
 704                 v_out_mine += value
 705                 is_relevant = True
 706
 707         if is_pruned:
 708             # some inputs are mine:
 709             fee = None
 710             if is_send:
 711                 v = v_out_mine - v_out
 712             else:
 713                 # no input is mine
 714                 v = v_out_mine
 715
 716         else:
 717             v = v_out_mine - v_in
 718
 719             if is_partial:
 720                 # some inputs are mine, but not all
 721                 fee = None
 722                 is_send = v < 0
 723             else:
 724                 # all inputs are mine
 725                 fee = v_out - v_in
 726
 727         return is_relevant, is_send, v, fee
 728
 729
 730     def get_input_info(self):
 731         keys = ['prevout_hash', 'prevout_n', 'address', 'KeyID', 'scriptPubKey', 'redeemScript', 'redeemPubkey', 'pubkeys', 'signatures', 'is_coinbase']
 732         info = []
 733         for i in self.inputs:
 734             item = {}
 735             for k in keys:
 736                 v = i.get(k)
 737                 if v is not None:
 738                     item[k] = v
 739             info.append(item)
 740         return info
 741
 742
 743     def as_dict(self):
 744         import json
 745         out = {
 746             "hex":self.raw,
 747             "complete":self.is_complete()
 748             }
 749
 750         if not self.is_complete():
 751             input_info = self.get_input_info()
 752             out['input_info'] = json.dumps(input_info).replace(' ','')
 753
 754         return out
 755
 756
 757     def requires_fee(self, verifier):
 758         # see https://en.bitcoin.it/wiki/Transaction_fees
 759         threshold = 57600000
 760         size = len(self.raw)/2
 761         if size >= 10000:
 762             return True
 763
 764         for o in self.outputs:
 765             value = o[1]
 766             if value < 1000000:
 767                 return True
 768         sum = 0
 769         for i in self.inputs:
 770             age = verifier.get_confirmations(i["prevout_hash"])[0]
 771             sum += i["value"] * age
 772         priority = sum / size
 773         print_error(priority, threshold)
 774         return priority < threshold
 775
 776
 777
 778     def add_input_info(self, input_info):
 779         for i, txin in enumerate(self.inputs):
 780             item = input_info[i]
 781             txin['scriptPubKey'] = item['scriptPubKey']
 782             txin['redeemScript'] = item.get('redeemScript')
 783             txin['redeemPubkey'] = item.get('redeemPubkey')
 784             txin['KeyID'] = item.get('KeyID')
 785             txin['signatures'] = item.get('signatures',{})