Mercurial > public > bitcaviar-plus

annotate src/bitcaviar_plus/block.py @ 23:32061555853c

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
refactor code
author Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
date Sun, 14 Nov 2021 17:35:18 +0100
parents 905b6fdc5e1b
children 30535f42d0ff
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
23
32061555853c refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 19
diff changeset
1
32061555853c refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 19
diff changeset
2 """
32061555853c refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 19
diff changeset
3 Deserialize methods for blocks
32061555853c refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 19
diff changeset
4 """
32061555853c refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 19
diff changeset
5
12
6a0a8cce058e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 10
diff changeset
6 from bitcaviar_plus.block_structure import *
6a0a8cce058e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 10
diff changeset
7 from bitcaviar_plus.helpers import __get_var_int
6a0a8cce058e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 10
diff changeset
8 from bitcaviar_plus.helpers import __compute_hash
19
905b6fdc5e1b handle invalid magic bytes
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 12
diff changeset
9 from bitcaviar_plus.errors import InvalidMagicBytes
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
10
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
11
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
12 def deserialize_block(f):
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
13 """
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
14 Deserialize block
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
15 :param f: buffer, required
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
16 :return: dict
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
17 """
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
18
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
19 block = Block()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
20 block.magic_number = f.read(4).hex()
19
905b6fdc5e1b handle invalid magic bytes
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 12
diff changeset
21
905b6fdc5e1b handle invalid magic bytes
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 12
diff changeset
22 if block.magic_number != 'f9beb4d9':
905b6fdc5e1b handle invalid magic bytes
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 12
diff changeset
23 raise InvalidMagicBytes(block.magic_number)
905b6fdc5e1b handle invalid magic bytes
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 12
diff changeset
24
9
e218f70e19e9 refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 8
diff changeset
25 block.size = f.read(4)[::-1].hex()
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
26 block_header, block.id = __deserialize_header(f)
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
27 block.transaction_count = __get_var_int(f)
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
28
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
29 transactions = []
9
e218f70e19e9 refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 8
diff changeset
30 for transaction_number in range(int(block.transaction_count, 16)):
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
31 transactions.append(__deserialize_transaction_data(f))
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
32
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
33 block_dict = block.__dict__
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
34 block_dict['header'] = block_header
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
35 block_dict['transactions'] = transactions
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
36
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
37 return block_dict
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
38
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
39
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
40 def __deserialize_header(f):
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
41 """
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
42 Deserialize block header
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
43 More info: https://learnmeabitcoin.com/technical/block-header
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
44 :param f: buffer, required
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
45 :return: (dict, string)
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
46 """
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
47
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
48 # Compute block hash
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
49 before = f.tell()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
50 header = f.read(80)
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
51 block_hash = __compute_hash(header)
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
52 f.seek(before)
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
53
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
54 header = Header()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
55 header.version = f.read(4)[::-1].hex()
9
e218f70e19e9 refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 8
diff changeset
56 header.previous_block_id = f.read(32)[::-1].hex()
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
57 header.merkle_root = f.read(32)[::-1].hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
58 header.time = f.read(4)[::-1].hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
59 header.bits = f.read(4)[::-1].hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
60 header.nonce = f.read(4)[::-1].hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
61
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
62 return header.__dict__, block_hash
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
63
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
64
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
65 def __deserialize_transaction_data(f):
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
66 """
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
67 Deserialize transaction data
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
68 More info: https://learnmeabitcoin.com/technical/transaction-data
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
69 :param f: buffer, required
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
70 :return: dict
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
71 """
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
72
9
e218f70e19e9 refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 8
diff changeset
73 transaction = Transaction()
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
74 start_transaction_data = f.tell()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
75 transaction.version = f.read(4)[::-1].hex()
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
76 transaction.input_count = __get_var_int(f)
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
77
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
78 transaction_inputs = []
9
e218f70e19e9 refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 8
diff changeset
79 for input_number in range(int(transaction.input_count, 16)):
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
80 transaction_input = TransactionInput()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
81 transaction_input.id = f.read(32)[::-1].hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
82 transaction_input.vout = f.read(4)[::-1].hex()
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
83 transaction_input.script_sig_size = __get_var_int(f)
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
84 transaction_input.script_sig = f.read(int(transaction_input.script_sig_size, 16)).hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
85 transaction_input.sequence = f.read(4)[::-1].hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
86 transaction_inputs.append(transaction_input.__dict__)
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
87
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
88 transaction.output_count = __get_var_int(f)
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
89
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
90 transaction_outputs = []
9
e218f70e19e9 refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 8
diff changeset
91 for output_number in range(int(transaction.output_count, 16)):
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
92 transaction_output = TransactionOutput()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
93 transaction_output.value = f.read(8)[::-1].hex()
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
94 transaction_output.script_pub_key_size = __get_var_int(f)
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
95 transaction_output.script_pub_key = f.read(int(transaction_output.script_pub_key_size, 16)).hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
96 transaction_outputs.append(transaction_output.__dict__)
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
97
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
98 transaction.lock_time = f.read(4)[::-1].hex()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
99
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
100 # Compute transaction id
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
101 end_transaction_data = f.tell()
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
102 transaction_data_size = end_transaction_data - start_transaction_data
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
103 f.seek(start_transaction_data)
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
104 transaction_data = f.read(transaction_data_size)
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
105 f.seek(end_transaction_data)
10
6e9f420c117e refactor code
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents: 9
diff changeset
106 transaction.id = __compute_hash(transaction_data)
8
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
107
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
108 transaction_dict = transaction.__dict__
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
109 transaction_dict['inputs'] = transaction_inputs
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
110 transaction_dict['outputs'] = transaction_outputs
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
111
4d259e84160d fix OverFlow bug
Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
parents:
diff changeset
112 return transaction_dict