2020-04-22 11:04:19 +00:00
|
|
|
import
|
|
|
|
randutils, stew/byteutils, random,
|
2020-04-24 08:58:36 +00:00
|
|
|
eth/[common, rlp], eth/trie/[hexary, db],
|
2020-04-28 02:30:10 +00:00
|
|
|
faststreams/input_stream, nimcrypto/[utils, sysrand],
|
2020-04-22 11:04:19 +00:00
|
|
|
../stateless/[witness_from_tree, tree_from_witness]
|
|
|
|
|
2020-04-28 02:30:10 +00:00
|
|
|
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 randHash(): Hash256 =
|
|
|
|
discard randomBytes(result.data[0].addr, sizeof(result))
|
|
|
|
|
|
|
|
proc randAccount(): Account =
|
|
|
|
result.nonce = randNonce()
|
|
|
|
result.balance = randU256()
|
|
|
|
result.codeHash = randHash()
|
|
|
|
result.storageRoot = randHash()
|
|
|
|
|
2020-04-22 11:04:19 +00:00
|
|
|
proc runTest(keyBytes: int, valBytes: int, numPairs: int) =
|
|
|
|
var memDB = newMemoryDB()
|
|
|
|
var trie = initHexaryTrie(memDB)
|
|
|
|
|
|
|
|
var
|
|
|
|
keys = newSeq[Bytes](numPairs)
|
|
|
|
vals = newSeq[Bytes](numPairs)
|
|
|
|
|
|
|
|
for i in 0..<numPairs:
|
|
|
|
keys[i] = randList(byte, rng(0, 255), keyBytes, unique = false)
|
|
|
|
vals[i] = randList(byte, rng(0, 255), valBytes, unique = false)
|
|
|
|
trie.put(keys[i], vals[i])
|
|
|
|
|
|
|
|
let rootHash = trie.rootHash
|
|
|
|
|
|
|
|
var wb = initWitnessBuilder(memDB, rootHash)
|
|
|
|
var witness = wb.getBranchRecurse(keys[0])
|
2020-04-23 02:56:35 +00:00
|
|
|
var input = memoryInput(witness)
|
2020-04-22 11:04:19 +00:00
|
|
|
|
2020-04-24 08:58:36 +00:00
|
|
|
var db = newMemoryDB()
|
2020-04-23 02:56:35 +00:00
|
|
|
var tb = initTreeBuilder(input, db)
|
2020-04-22 11:04:19 +00:00
|
|
|
var root = tb.treeNode()
|
|
|
|
debugEcho "root: ", root.data.toHex
|
|
|
|
debugEcho "rootHash: ", rootHash.data.toHex
|
|
|
|
doAssert root.data == rootHash.data
|
|
|
|
|
|
|
|
proc main() =
|
|
|
|
runTest(7, 100, 50)
|
|
|
|
runTest(1, 1, 1)
|
|
|
|
runTest(5, 5, 1)
|
|
|
|
runTest(6, 7, 3)
|
|
|
|
runTest(7, 10, 7)
|
|
|
|
runTest(8, 15, 11)
|
|
|
|
runTest(9, 30, 13)
|
|
|
|
runTest(11, 40, 10)
|
|
|
|
runTest(20, 1, 15)
|
|
|
|
runTest(25, 10, 20)
|
2020-04-24 08:58:36 +00:00
|
|
|
|
2020-04-22 11:04:19 +00:00
|
|
|
randomize()
|
|
|
|
for i in 0..<30:
|
|
|
|
runTest(rand(1..30), rand(1..50), rand(1..30))
|
2020-04-24 08:58:36 +00:00
|
|
|
|
2020-04-22 11:04:19 +00:00
|
|
|
|
|
|
|
main()
|