mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 14:14:22 +01:00
88532f7345
The input types for Blob.encrypt and EncryptedBlob.decrypt were not consistent. The former was in hex whereas the later was in bytes. Format the later in hex for consistency.
102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
import binascii
|
|
from hashlib import sha256
|
|
|
|
from pisa import logging, bitcoin_cli
|
|
from pisa.utils.auth_proxy import JSONRPCException
|
|
|
|
|
|
class BlockProcessor:
|
|
@staticmethod
|
|
def get_block(block_hash):
|
|
block = None
|
|
|
|
try:
|
|
block = bitcoin_cli.getblock(block_hash)
|
|
|
|
except JSONRPCException as e:
|
|
logging.error("[BlockProcessor] couldn't get block from bitcoind. Error code {}".format(e))
|
|
|
|
return block
|
|
|
|
@staticmethod
|
|
def get_best_block_hash():
|
|
block_hash = None
|
|
|
|
try:
|
|
block_hash = bitcoin_cli.getbestblockhash()
|
|
|
|
except JSONRPCException as e:
|
|
logging.error("[BlockProcessor] couldn't get block hash. Error code {}".format(e))
|
|
|
|
return block_hash
|
|
|
|
@staticmethod
|
|
def get_block_count():
|
|
block_count = None
|
|
|
|
try:
|
|
block_count = bitcoin_cli.getblockcount()
|
|
|
|
except JSONRPCException as e:
|
|
logging.error("[BlockProcessor] couldn't get block block count. Error code {}".format(e))
|
|
|
|
return block_count
|
|
|
|
@staticmethod
|
|
def get_potential_matches(txids, locator_uuid_map):
|
|
potential_locators = {sha256(binascii.unhexlify(txid)).hexdigest(): txid for txid in txids}
|
|
|
|
# Check is any of the tx_ids in the received block is an actual match
|
|
intersection = set(locator_uuid_map.keys()).intersection(potential_locators.keys())
|
|
potential_matches = {locator: potential_locators[locator] for locator in intersection}
|
|
|
|
if len(potential_matches) > 0:
|
|
logging.info("[BlockProcessor] list of potential matches: {}".format(potential_matches))
|
|
|
|
else:
|
|
logging.info("[BlockProcessor] no potential matches found")
|
|
|
|
@staticmethod
|
|
def get_matches(potential_matches, locator_uuid_map, appointments):
|
|
matches = []
|
|
|
|
for locator, dispute_txid in potential_matches.items():
|
|
for uuid in locator_uuid_map[locator]:
|
|
try:
|
|
# ToDo: #20-test-tx-decrypting-edge-cases
|
|
justice_rawtx = appointments[uuid].encrypted_blob.decrypt(dispute_txid)
|
|
justice_txid = bitcoin_cli.decoderawtransaction(justice_rawtx).get('txid')
|
|
matches.append((locator, uuid, dispute_txid, justice_txid, justice_rawtx))
|
|
|
|
logging.info("[BlockProcessor] match found for locator {} (uuid: {}): {}".format(
|
|
locator, uuid, justice_txid))
|
|
|
|
except JSONRPCException as e:
|
|
# Tx decode failed returns error code -22, maybe we should be more strict here. Leaving it simple
|
|
# for the POC
|
|
logging.error("[BlockProcessor] can't build transaction from decoded data. Error code {}".format(e))
|
|
|
|
return matches
|
|
|
|
@staticmethod
|
|
def check_confirmations(txs, unconfirmed_txs, tx_job_map, missed_confirmations):
|
|
|
|
for tx in txs:
|
|
if tx in tx_job_map and tx in unconfirmed_txs:
|
|
unconfirmed_txs.remove(tx)
|
|
|
|
logging.info("[Responder] confirmation received for tx {}".format(tx))
|
|
|
|
elif tx in unconfirmed_txs:
|
|
if tx in missed_confirmations:
|
|
missed_confirmations[tx] += 1
|
|
|
|
else:
|
|
missed_confirmations[tx] = 1
|
|
|
|
logging.info("[Responder] tx {} missed a confirmation (total missed: {})"
|
|
.format(tx, missed_confirmations[tx]))
|
|
|
|
return unconfirmed_txs, missed_confirmations
|
|
|