153b4c9107b4085e047f770eb21b78a8df1cbabe
[electrum-nvc.git] / scripts / validate_tx
1 #!/usr/bin/env python
2
3 import sys, hashlib
4 from electrum import Interface
5
6 """validate a transaction (SPV)"""
7
8
9 Hash = lambda x: hashlib.sha256(hashlib.sha256(x).digest()).digest()
10
11 def rev_hex(s):
12     return s.decode('hex')[::-1].encode('hex')
13
14 def int_to_hex(i, length=1):
15     s = hex(i)[2:].rstrip('L')
16     s = "0"*(2*length - len(s)) + s
17     return rev_hex(s)
18
19
20 i = Interface({'server':'ecdsa.org:50002:s'})
21 i.start()
22
23
24 def merkle_root(merkle):
25
26     merkle = map (lambda tx_hash: tx_hash.decode('hex')[::-1], merkle)
27     while len(merkle) != 1:
28         if len(merkle)%2: merkle.append( merkle[-1] )
29         n = []
30         while merkle:
31             n.append( Hash( merkle[0] + merkle[1] ) )
32             merkle = merkle[2:]
33
34         merkle = n
35             
36     return merkle[0][::-1].encode('hex')
37
38
39 def hash_header(res):
40     header = int_to_hex(res.get('version'),4) \
41         + rev_hex(res.get('prev_block_hash')) \
42         + rev_hex(res.get('merkle_root')) \
43         + int_to_hex(int(res.get('timestamp')),4) \
44         + int_to_hex(int(res.get('bits')),4) \
45         + int_to_hex(int(res.get('nonce')),4)
46     return rev_hex(Hash(header.decode('hex')).encode('hex'))
47
48
49 def verify_tx(tx_hash):
50     
51     res = i.synchronous_get([ ('blockchain.transaction.get_merkle',[tx_hash]) ])[0]
52     assert tx_hash in res.get('merkle')
53     assert res.get('merkle_root') == merkle_root(res['merkle'])
54     block_height = res.get('block_height')
55     print block_height
56
57     headers_requests = []
58     for height in range(block_height-10,block_height+10):
59         headers_requests.append( ('blockchain.block.get_header',[height]) )
60     res = i.synchronous_get(headers_requests)
61
62     _hash = None
63     for header in res:
64         if _hash: assert _hash == header.get('prev_block_hash')
65         _hash = hash_header(header)
66         print _hash
67         if height==block_height:
68             assert header.get('merkle_root') == res.get('merkle_root')
69         
70
71
72 try:
73     tx = sys.argv[1]
74 except:
75     tx = '587430e52af2cec98b3fd543083469ffa7a5f5dd2bd569898a7897a64e2eb031'
76
77 verify_tx(tx)
78