kylefang/stacks-puppet-node
1
0
Fork 0
mirror of https://github.com/alexgo-io/stacks-puppet-node.git synced 2026-04-24 03:45:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

accept a dict of expected consensus hashes while verifying the database,

Browse Source 
and check each block along the way (instead of just the last one)
This commit is contained in:
Jude Nelson
2016-09-28 19:53:15 -04:00
parent 78f773db00
commit 8b5c554a93
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
blockstack/lib/consensus.py
View File

@@ -220,12 +220,16 @@ def block_to_virtualchain_ops( block_id, working_db, untrusted_db ):
return virtualchain_ops
def rebuild_database( target_block_id, untrusted_db_path, working_db_path=None, resume_dir=None, start_block=None ):
def rebuild_database( target_block_id, untrusted_db_path, working_db_path=None, resume_dir=None, start_block=None, expected_snapshots={} ):
"""
Given a target block ID and a path to an (untrusted) db, reconstruct it in a temporary directory by
replaying all the nameops it contains.
Optionally check that the snapshots in @expected_snapshots match up as we verify.
@expected_snapshots maps str(block_id) to str(consensus hash)
Return the consensus hash calculated at the target block.
Return None on verification failure (i.e. we got a different consensus hash than one for the same block in expected_snapshots)
"""
# reconfigure the virtualchain to use a temporary directory,
@@ -275,12 +279,17 @@ def rebuild_database( target_block_id, untrusted_db_path, working_db_path=None,
log.debug("VERIFY CONSENSUS(%s): %s" % (block_id, consensus_hash))
consensus_hashes[block_id] = consensus_hash
if str(block_id) in expected_snapshots:
if expected_snapshots[str(block_id)] != consensus_hash:
log.error("DATABASE IS NOT CONSISTENT AT %s: %s != %s" % (block_id, expected_snashots[str(block_id)], consensus_hash))
return None
# final consensus hash
return consensus_hashes[ target_block_id ]
def verify_database( trusted_consensus_hash, consensus_block_id, untrusted_db_path, working_db_path=None, start_block=None ):
def verify_database( trusted_consensus_hash, consensus_block_id, untrusted_db_path, working_db_path=None, start_block=None, expected_snapshots={} ):
"""
Verify that a database is consistent with a
known-good consensus hash.
@@ -292,10 +301,10 @@ def verify_database( trusted_consensus_hash, consensus_block_id, untrusted_db_pa
database.
"""
final_consensus_hash = rebuild_database( consensus_block_id, untrusted_db_path, working_db_path=working_db_path, start_block=start_block )
final_consensus_hash = rebuild_database( consensus_block_id, untrusted_db_path, working_db_path=working_db_path, start_block=start_block, expected_snapshots=expected_snapshots )
# did we reach the consensus hash we expected?
if final_consensus_hash == trusted_consensus_hash:
if final_consensus_hash is not None and final_consensus_hash == trusted_consensus_hash:
return True
else: