63 lines
1.8 KiB
Nim
63 lines
1.8 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2021 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
# at your option.
|
|
# This file may not be copied, modified, or distributed except according to
|
|
# those terms.
|
|
|
|
import
|
|
chronicles, eth/[common, rlp], stew/io2,
|
|
./p2p/chain, ./db/[db_chain, select_backend]
|
|
|
|
type
|
|
# trick the rlp decoder
|
|
# so we can separate the body and header
|
|
EthHeader = object
|
|
header: BlockHeader
|
|
|
|
proc importRlpBlock*(blocksRlp: openArray[byte]; chainDB: BaseChainDB; importFile: string = ""): bool =
|
|
var
|
|
# the encoded rlp can contains one or more blocks
|
|
rlp = rlpFromBytes(blocksRlp)
|
|
chain = newChain(chainDB, extraValidation = true)
|
|
errorCount = 0
|
|
let
|
|
head = chainDB.getCanonicalHead()
|
|
|
|
while rlp.hasData:
|
|
try:
|
|
let
|
|
header = rlp.read(EthHeader).header
|
|
body = rlp.readRecordType(BlockBody, false)
|
|
if header.blockNumber > head.blockNumber:
|
|
if chain.persistBlocks([header], [body]) == ValidationResult.Error:
|
|
# register one more error and continue
|
|
errorCount.inc
|
|
except RlpError as e:
|
|
# terminate if there was a decoding error
|
|
error "rlp error",
|
|
fileName = importFile,
|
|
msg = e.msg,
|
|
exception = e.name
|
|
return false
|
|
except CatchableError as e:
|
|
# otherwise continue
|
|
error "import error",
|
|
fileName = importFile,
|
|
msg = e.msg,
|
|
exception = e.name
|
|
errorCount.inc
|
|
|
|
return errorCount == 0
|
|
|
|
proc importRlpBlock*(importFile: string; chainDB: BaseChainDB): bool =
|
|
let res = io2.readAllBytes(importFile)
|
|
if res.isErr:
|
|
error "failed to import",
|
|
fileName = importFile
|
|
return false
|
|
|
|
importRlpBlock(res.get, chainDB, importFile)
|
|
|