Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

make integrity check faster

Browse source
This commit is contained in:
Victor Shyba 2019-07-09 15:00:06 -03:00 committed by Lex Berezhny
parent 9990889cce
commit 2481305a1a
1 changed files with 13 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
torba/torba/client/baseheader.py
View file

@ -165,18 +165,24 @@ class BaseHeaders:
)
def repair(self):
previous_header_hash = fail = None
for height in range(self.height):
chunk = self.get_raw_header(height)
try:
# validate_chunk() is CPU bound and reads previous chunks from file system
self.validate_chunk(height, chunk)
except InvalidHeader as e:
log.warning("Header file corrupted at height %s, truncating it.", e.height)
self.io.seek((e.height) * self.header_size, os.SEEK_SET)
raw = self.get_raw_header(height)
header = self.deserialize(height, raw)
if height:
if header['prev_block_hash'] != previous_header_hash:
fail = True
else:
if self.hash_header(raw) != self.genesis_hash:
fail = True
if fail:
log.warning("Header file corrupted at height %s, truncating it.", height - 1)
self.io.seek((height - 1) * self.header_size, os.SEEK_SET)
self.io.truncate()
self.io.flush()
self._size = None
return
previous_header_hash = self.hash_header(raw)
@staticmethod
def get_proof_of_work(header_hash: bytes) -> ArithUint256: