fix computeKey account hash (#2795)

Oops. Discovered as part of making the code use the actual production
database types in the key computation test ;)
This commit is contained in:
Jacek Sieka 2024-10-28 19:14:28 +01:00 committed by GitHub
parent 43e08d08c7
commit 1406feab5f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 57 additions and 86 deletions

View File

@ -367,7 +367,7 @@ proc computeLeafKeysImpl(
nonce: vtx.lData.account.nonce, nonce: vtx.lData.account.nonce,
balance: vtx.lData.account.balance, balance: vtx.lData.account.balance,
# Accounts with storage filtered out above # Accounts with storage filtered out above
storageRoot: default(Hash32), storageRoot: EMPTY_ROOT_HASH,
codeHash: vtx.lData.account.codeHash, codeHash: vtx.lData.account.codeHash,
) )
writer2.finish() writer2.finish()

View File

@ -15,8 +15,16 @@ import
stew/byteutils, stew/byteutils,
unittest2, unittest2,
../../nimbus/db/aristo/[ ../../nimbus/db/aristo/[
aristo_check, aristo_compute, aristo_delete, aristo_get, aristo_merge, aristo_desc, aristo_check,
aristo_utils, aristo_serialise, aristo_init, aristo_tx/tx_stow, aristo_compute,
aristo_delete,
aristo_get,
aristo_merge,
aristo_desc,
aristo_utils,
aristo_serialise,
aristo_init,
aristo_tx/tx_stow,
] ]
func x(s: string): seq[byte] = func x(s: string): seq[byte] =
@ -25,79 +33,48 @@ func k(s: string): HashKey =
HashKey.fromBytes(s.x).value HashKey.fromBytes(s.x).value
let samples = [ let samples = [
# From InvalidBlocks/bc4895-withdrawals/twoIdenticalIndex.json
@[
(
"80".x,
"da808094c94f5374fce5edbc8e2a8697c15331677e6ebf0b822710".x,
hash32"27f166f1d7c789251299535cb176ba34116e44894476a7886fe5d73d9be5c973",
VOID_HASH_KEY,
),
(
"01".x,
"da028094c94f5374fce5edbc8e2a8697c15331677e6ebf0b822710".x,
hash32"81eac5f476f48feb289af40ee764015f6b49036760438ea45df90d5342b6ae61",
VOID_HASH_KEY,
),
(
"02".x,
"da018094c94f5374fce5edbc8e2a8697c15331677e6ebf0b822710".x,
hash32"463769ae507fcc6d6231c8888425191c5622f330fdd4b78a7b24c4521137b573",
VOID_HASH_KEY,
),
(
"03".x,
"da028094c94f5374fce5edbc8e2a8697c15331677e6ebf0b822710".x,
hash32"a95b9a7b58a6b3cb4001eb0be67951c5517141cb0183a255b5cae027a7b10b36",
VOID_HASH_KEY,
),
],
# Somew on-the-fly provided stuff # Somew on-the-fly provided stuff
@[ @[
# Create leaf node
( (
"0000".x, hash32"0000000000000000000000000000000000000000000000000000000000000001",
"0000".x, AristoAccount(balance: 0.u256, codeHash: EMPTY_CODE_HASH),
hash32"69a4785bd4f5a1590e329138d4248b6f887fa37e41bfc510a55f21b44f98be61", hash32"69b5c560f84dde1ecb0584976f4ebbe78e34bb6f32410777309a8693424bb563",
"c783200000820000".k, ),
# Overwrite existing leaf
(
hash32"0000000000000000000000000000000000000000000000000000000000000001",
AristoAccount(balance: 1.u256, codeHash: EMPTY_CODE_HASH),
hash32"5ce3c539427b494d97d1fc89080118370f173d29c7dec55a292e6c00a08c4465",
),
# Split leaf node with extension
(
hash32"0000000000000000000000000000000000000000000000000000000000000002",
AristoAccount(balance: 1.u256, codeHash: EMPTY_CODE_HASH),
hash32"6f28eee5fe67fba78c5bb42cbf6303574c4139ad97631002e07466d2f98c0d35",
), ),
( (
"0000".x, hash32"0000000000000000000000000000000000000000000000000000000000000003",
"0001".x, AristoAccount(balance: 0.u256, codeHash: EMPTY_CODE_HASH),
hash32"910fa1155b667666abe7b4b1cb4864c1dc91c57c9528e1c5f5f9f95e003afece", hash32"5dacbc38677935c135b911e8c786444e4dc297db1f0c77775ce47ffb8ce81dca",
"c783200000820001".k, ),
# Split extension
(
hash32"0100000000000000000000000000000000000000000000000000000000000000",
AristoAccount(balance: 1.u256, codeHash: EMPTY_CODE_HASH),
hash32"57dd53adbbd1969204c0b3435df8c22e0aadadad50871ce7ab4d802b77da2dd3",
), ),
( (
"0001".x, hash32"0100000000000000000000000000000000000000000000000000000000000001",
"0001".x, AristoAccount(balance: 2.u256, codeHash: EMPTY_CODE_HASH),
hash32"d082d2bfe8586142d6f40df0245e56365043819e51e2c9799c660558eeea0db5", hash32"67ebbac82cc2a55e0758299f63b785fbd3d1f17197b99c78ffd79d73d3026827",
"dd821000d9c420820001c420820001808080808080808080808080808080".k,
), ),
( (
"0002".x, hash32"0200000000000000000000000000000000000000000000000000000000000000",
"0000".x, AristoAccount(balance: 3.u256, codeHash: EMPTY_CODE_HASH),
hash32"d56ea5154fbad18e0ff1eaeafa2310d0879b59adf189c12ff1b2701e54db07b2", hash32"e7d6a8f7fb3e936eff91a5f62b96177817f2f45a105b729ab54819a99a353325",
VOID_HASH_KEY,
), ),
( ]
"0100".x,
"0100".x,
hash32"d1c0699fe7928a536e0183c6400ae34eb1174ce6b21f9d117b061385034743ad",
VOID_HASH_KEY,
),
(
"0101".x,
"0101".x,
hash32"74ddb98cb56e2dd7e8fa090b9ce740c3de589b72403b20136af75fb6168b1d19",
VOID_HASH_KEY,
),
(
"0200".x,
"0200".x,
hash32"2e777f06ab2de1a460a8412b8691f10fdcb162077ab5cbb1865636668bcb6471",
VOID_HASH_KEY,
),
],
] ]
suite "Aristo compute": suite "Aristo compute":
@ -105,29 +82,24 @@ suite "Aristo compute":
test "Add and delete entries " & $n: test "Add and delete entries " & $n:
let let
db = AristoDbRef.init VoidBackendRef db = AristoDbRef.init VoidBackendRef
root = VertexID(2) root = VertexID(1)
for inx, (k, v, r, s) in sample: for (k, v, r) in sample:
checkpoint("k = " & k.toHex & ", v = " & v.toHex()) checkpoint("k = " & k.toHex & ", v = " & $v)
check: check:
db.mergeGenericData(root, k, v) == Result[bool, AristoError].ok(true) db.mergeAccountRecord(k, v) == Result[bool, AristoError].ok(true)
# Check state against expected value # Check state against expected value
let w = db.computeKey((root, root)).expect("no errors") let w = db.computeKey((root, root)).expect("no errors")
check r == w.to(Hash32) check r == w.to(Hash32)
# Check raw node if given, check nor ref against expected value
if s.isValid:
let z = db.getVtx((root, root)).toNode(root, db).value.digestTo(HashKey)
check s == z
let rc = db.check let rc = db.check
check rc == typeof(rc).ok() check rc == typeof(rc).ok()
# Reverse run deleting entries # Reverse run deleting entries
var deletedKeys: HashSet[seq[byte]] var deletedKeys: HashSet[Hash32]
for iny, (k, v, r, s) in sample.reversed: for iny, (k, v, r) in sample.reversed:
# Check whether key was already deleted # Check whether key was already deleted
if k in deletedKeys: if k in deletedKeys:
continue continue
@ -138,13 +110,8 @@ suite "Aristo compute":
check r == w check r == w
# Check raw node if given, check nor ref against expected value
if s.isValid:
let z = db.getVtx((root, root)).toNode(root, db).value.digestTo(HashKey)
check s == z
check: check:
db.deleteGenericData(root, k).isOk db.deleteAccountRecord(k).isOk
let rc = db.check let rc = db.check
check rc == typeof(rc).ok() check rc == typeof(rc).ok()
@ -153,11 +120,15 @@ suite "Aristo compute":
# TODO use mainnet genesis in this test? # TODO use mainnet genesis in this test?
let let
db = AristoDbRef.init MemBackendRef db = AristoDbRef.init MemBackendRef
root = VertexID(2) root = VertexID(1)
for inx, (k, v, r, s) in samples[^1]: for (k, v, r) in samples[^1]:
check: check:
db.mergeGenericData(root, keccak256(k).data, v) == Result[bool, AristoError].ok(true) db.mergeAccountRecord(k, v) == Result[bool, AristoError].ok(true)
check db.txStow(1, true).isOk() check db.txStow(1, true).isOk()
check db.computeKeys(root).isOk() check db.computeKeys(root).isOk()
let w = db.computeKey((root, root)).value.to(Hash32)
check w == samples[^1][^1][2]