138 lines
4.6 KiB
Python
138 lines
4.6 KiB
Python
from os import mkdir
|
|
from shutil import rmtree
|
|
from ssz_proof_types import *
|
|
from convert_transactions import *
|
|
|
|
def create_transaction_proof(transactions: Transactions, tx_index: uint64) -> TransactionProof:
|
|
tx = transactions[tx_index]
|
|
return TransactionProof(
|
|
payload_root=Root(tx.payload.hash_tree_root()),
|
|
tx_hash=tx.tx_hash,
|
|
tx_index=tx_index,
|
|
tx_branch=build_proof(
|
|
transactions.get_backing(),
|
|
MAX_TRANSACTIONS_PER_PAYLOAD * 2 + tx_index,
|
|
),
|
|
)
|
|
|
|
def create_amount_proof(transactions: Transactions, tx_index: uint64) -> TransactionProof:
|
|
tx = transactions[tx_index]
|
|
return AmountProof(
|
|
tx_from=tx.payload.tx_from,
|
|
nonce=tx.payload.nonce,
|
|
tx_to=tx.payload.tx_to,
|
|
tx_value=tx.payload.tx_value,
|
|
multi_branch=[
|
|
tx.payload.get_backing().getter(gindex).merkle_root()
|
|
for gindex in AMOUNT_PROOF_HELPER_INDICES
|
|
],
|
|
tx_hash=tx.tx_hash,
|
|
tx_index=tx_index,
|
|
tx_branch=build_proof(
|
|
transactions.get_backing(),
|
|
MAX_TRANSACTIONS_PER_PAYLOAD * 2 + tx_index,
|
|
),
|
|
)
|
|
|
|
def create_sender_proof(transactions: Transactions, tx_index: uint64) -> TransactionProof:
|
|
tx = transactions[tx_index]
|
|
return SenderProof(
|
|
tx_from=tx.payload.tx_from,
|
|
nonce=tx.payload.nonce,
|
|
tx_to=tx.payload.tx_to,
|
|
tx_value=tx.payload.tx_value,
|
|
multi_branch=[
|
|
tx.payload.get_backing().getter(gindex).merkle_root()
|
|
for gindex in SENDER_PROOF_HELPER_INDICES
|
|
],
|
|
tx_hash=tx.tx_hash,
|
|
tx_index=tx_index,
|
|
tx_branch=build_proof(
|
|
transactions.get_backing(),
|
|
MAX_TRANSACTIONS_PER_PAYLOAD * 2 + tx_index,
|
|
),
|
|
)
|
|
|
|
def create_info_proof(transactions: Transactions, tx_index: uint64) -> TransactionProof:
|
|
tx = transactions[tx_index]
|
|
return InfoProof(
|
|
tx_from=tx.payload.tx_from,
|
|
nonce=tx.payload.nonce,
|
|
tx_to=tx.payload.tx_to,
|
|
tx_value=tx.payload.tx_value,
|
|
limits=tx.payload.limits,
|
|
multi_branch=[
|
|
tx.payload.get_backing().getter(gindex).merkle_root()
|
|
for gindex in INFO_PROOF_HELPER_INDICES
|
|
],
|
|
tx_hash=tx.tx_hash,
|
|
tx_index=tx_index,
|
|
tx_branch=build_proof(
|
|
transactions.get_backing(),
|
|
MAX_TRANSACTIONS_PER_PAYLOAD * 2 + tx_index,
|
|
),
|
|
)
|
|
|
|
transaction_proofs = [
|
|
create_transaction_proof(transactions, tx_index)
|
|
for tx_index in range(len(transactions))
|
|
]
|
|
amount_proofs = [
|
|
create_amount_proof(transactions, tx_index)
|
|
for tx_index in range(len(transactions))
|
|
]
|
|
sender_proofs = [
|
|
create_sender_proof(transactions, tx_index)
|
|
for tx_index in range(len(transactions))
|
|
]
|
|
info_proofs = [
|
|
create_info_proof(transactions, tx_index)
|
|
for tx_index in range(len(transactions))
|
|
]
|
|
|
|
if __name__ == '__main__':
|
|
dir = os_path.join(os_path.dirname(os_path.realpath(__file__)), 'proofs')
|
|
if os_path.exists(dir) and os_path.isdir(dir):
|
|
rmtree(dir)
|
|
mkdir(dir)
|
|
|
|
print('transactions_root')
|
|
print(f'0x{transactions_root.hex()}')
|
|
file = open(os_path.join(dir, f'transactions_root.ssz'), 'wb')
|
|
file.write(transactions_root)
|
|
file.close()
|
|
|
|
file = open(os_path.join(dir, f'nil_0.ssz'), 'wb')
|
|
file.close()
|
|
|
|
for tx_index in range(len(transactions)):
|
|
print()
|
|
|
|
encoded = transaction_proofs[tx_index].encode_bytes()
|
|
print(f'{tx_index} - TransactionProof - {len(encoded)} bytes (Snappy: {len(compress(encoded))})')
|
|
print(encoded.hex())
|
|
file = open(os_path.join(dir, f'transaction_{tx_index}.ssz'), 'wb')
|
|
file.write(encoded)
|
|
file.close()
|
|
|
|
encoded = amount_proofs[tx_index].encode_bytes()
|
|
print(f'{tx_index} - AmountProof - {len(encoded)} bytes (Snappy: {len(compress(encoded))})')
|
|
print(encoded.hex())
|
|
file = open(os_path.join(dir, f'amount_{tx_index}.ssz'), 'wb')
|
|
file.write(encoded)
|
|
file.close()
|
|
|
|
encoded = sender_proofs[tx_index].encode_bytes()
|
|
print(f'{tx_index} - SenderProof - {len(encoded)} bytes (Snappy: {len(compress(encoded))})')
|
|
print(encoded.hex())
|
|
file = open(os_path.join(dir, f'sender_{tx_index}.ssz'), 'wb')
|
|
file.write(encoded)
|
|
file.close()
|
|
|
|
encoded = info_proofs[tx_index].encode_bytes()
|
|
print(f'{tx_index} - InfoProof - {len(encoded)} bytes (Snappy: {len(compress(encoded))})')
|
|
print(encoded.hex())
|
|
file = open(os_path.join(dir, f'info_{tx_index}.ssz'), 'wb')
|
|
file.write(encoded)
|
|
file.close()
|