68 lines
1.8 KiB
Nim
68 lines
1.8 KiB
Nim
import
|
|
randutils, stew/byteutils, random,
|
|
eth/[common, rlp], eth/trie/[hexary, db, trie_defs],
|
|
faststreams/input_stream, nimcrypto/[utils, sysrand],
|
|
../stateless/[witness_from_tree, tree_from_witness]
|
|
|
|
type
|
|
DB = TrieDatabaseRef
|
|
|
|
proc randU256(): UInt256 =
|
|
var bytes: array[32, byte]
|
|
discard randomBytes(bytes[0].addr, sizeof(result))
|
|
result = UInt256.fromBytesBE(bytes)
|
|
|
|
proc randNonce(): AccountNonce =
|
|
discard randomBytes(result.addr, sizeof(result))
|
|
|
|
proc randCode(db: DB): Hash256 =
|
|
if rand(0..1) == 0:
|
|
result = blankStringHash
|
|
else:
|
|
let codeLen = rand(1..150)
|
|
let code = randList(byte, rng(0, 255), codeLen, unique = false)
|
|
result = hexary.keccak(code)
|
|
db.put(result.data, code)
|
|
|
|
proc randHash(): Hash256 =
|
|
discard randomBytes(result.data[0].addr, sizeof(result))
|
|
|
|
proc randAccount(db: DB): Account =
|
|
result.nonce = randNonce()
|
|
result.balance = randU256()
|
|
result.codeHash = randCode(db)
|
|
result.storageRoot = randHash()
|
|
|
|
proc runTest(numPairs: int) =
|
|
var memDB = newMemoryDB()
|
|
var trie = initSecureHexaryTrie(memDB)
|
|
var addrs = newSeq[Bytes](numPairs)
|
|
|
|
for i in 0..<numPairs:
|
|
addrs[i] = randList(byte, rng(0, 255), 20, unique = false)
|
|
let acc = randAccount(memDB)
|
|
trie.put(addrs[i], rlp.encode(acc))
|
|
|
|
let rootHash = trie.rootHash
|
|
|
|
var wb = initWitnessBuilder(memDB, rootHash)
|
|
var witness = wb.getBranchRecurse(addrs[0])
|
|
var db = newMemoryDB()
|
|
when defined(useInputStream):
|
|
var input = memoryInput(witness)
|
|
var tb = initTreeBuilder(input, db)
|
|
else:
|
|
var tb = initTreeBuilder(witness, db)
|
|
var root = tb.treeNode()
|
|
debugEcho "root: ", root.data.toHex
|
|
debugEcho "rootHash: ", rootHash.data.toHex
|
|
doAssert root.data == rootHash.data
|
|
|
|
proc main() =
|
|
randomize()
|
|
|
|
for i in 0..<30:
|
|
runTest(rand(1..30))
|
|
|
|
main()
|