Skip to content

Commit

Permalink
add blob validation to test_fixture_fork_choice
Browse files Browse the repository at this point in the history
Preparation for processing new tests from:

- ethereum/consensus-specs#3463
  • Loading branch information
etan-status committed Sep 18, 2023
1 parent 6344076 commit 2d94478
Showing 1 changed file with 49 additions and 15 deletions.
64 changes: 49 additions & 15 deletions tests/consensus_spec/test_fixture_fork_choice.nim
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@

import
# Status libraries
stew/results, chronicles,
stew/[byteutils, results], chronicles,
taskpools,
# Internals
../../beacon_chain/spec/[helpers, forks, state_transition_block],
Expand All @@ -28,7 +28,7 @@ import

from std/json import
JsonNode, getBool, getInt, getStr, hasKey, items, len, pairs, `$`, `[]`
from std/sequtils import toSeq
from std/sequtils import mapIt, toSeq
from std/strutils import contains

# Test format described at https://github.com/ethereum/consensus-specs/tree/v1.3.0/tests/formats/fork_choice
Expand All @@ -45,6 +45,10 @@ type
opInvalidateRoot
opChecks

BlobData = object
blobs: seq[KzgBlob]
proofs: seq[KzgProof]

Operation = object
valid: bool
# variant specific fields
Expand All @@ -55,6 +59,7 @@ type
att: Attestation
of opOnBlock:
blck: ForkedSignedBeaconBlock
blobData: Opt[BlobData]
of opOnMergeBlock:
powBlock: PowBlock
of opOnAttesterSlashing:
Expand All @@ -69,7 +74,7 @@ from ../../beacon_chain/spec/datatypes/capella import
BeaconBlock, BeaconState, SignedBeaconBlock

from ../../beacon_chain/spec/datatypes/deneb import
BeaconBlock, BeaconState, SignedBeaconBlock
KzgBlob, KzgProof, BeaconBlock, BeaconState, SignedBeaconBlock

proc initialLoad(
path: string, db: BeaconChainDB,
Expand Down Expand Up @@ -134,6 +139,8 @@ proc loadOps(path: string, fork: ConsensusFork): seq[Operation] =

result = @[]
for step in steps[0]:
var numExtraFields = 0

if step.hasKey"tick":
result.add Operation(kind: opOnTick,
tick: step["tick"].getInt())
Expand All @@ -147,42 +154,58 @@ proc loadOps(path: string, fork: ConsensusFork): seq[Operation] =
att: att)
elif step.hasKey"block":
let filename = step["block"].getStr()
doAssert step.hasKey"blobs" == step.hasKey"proofs"
case fork
of ConsensusFork.Phase0:
let blck = parseTest(
path/filename & ".ssz_snappy",
SSZ, phase0.SignedBeaconBlock
)
doAssert not step.hasKey"blobs"
result.add Operation(kind: opOnBlock,
blck: ForkedSignedBeaconBlock.init(blck))
of ConsensusFork.Altair:
let blck = parseTest(
path/filename & ".ssz_snappy",
SSZ, altair.SignedBeaconBlock
)
doAssert not step.hasKey"blobs"
result.add Operation(kind: opOnBlock,
blck: ForkedSignedBeaconBlock.init(blck))
of ConsensusFork.Bellatrix:
let blck = parseTest(
path/filename & ".ssz_snappy",
SSZ, bellatrix.SignedBeaconBlock
)
doAssert not step.hasKey"blobs"
result.add Operation(kind: opOnBlock,
blck: ForkedSignedBeaconBlock.init(blck))
of ConsensusFork.Capella:
let blck = parseTest(
path/filename & ".ssz_snappy",
SSZ, capella.SignedBeaconBlock
)
doAssert not step.hasKey"blobs"
result.add Operation(kind: opOnBlock,
blck: ForkedSignedBeaconBlock.init(blck))
of ConsensusFork.Deneb:
let blck = parseTest(
path/filename & ".ssz_snappy",
SSZ, deneb.SignedBeaconBlock
)
let
blck = parseTest(
path/filename & ".ssz_snappy",
SSZ, deneb.SignedBeaconBlock)
blobData =
if step.hasKey"blobs":
numExtraFields += 2
Opt.some BlobData(
blobs: distinctBase(parseTest(
path/(step["blobs"].getStr()) & ".ssz_snappy",
SSZ, List[KzgBlob, Limit MAX_BLOBS_PER_BLOCK])),
proofs: step["proofs"].mapIt(KzgProof.fromHex(it.getStr())))
else:
Opt.none(BlobData)
result.add Operation(kind: opOnBlock,
blck: ForkedSignedBeaconBlock.init(blck))
blck: ForkedSignedBeaconBlock.init(blck),
blobData: blobData)
elif step.hasKey"attester_slashing":
let filename = step["attester_slashing"].getStr()
let attesterSlashing = parseTest(
Expand All @@ -205,10 +228,10 @@ proc loadOps(path: string, fork: ConsensusFork): seq[Operation] =
doAssert false, "Unknown test step: " & $step

if step.hasKey"valid":
doAssert step.len == 2
doAssert step.len == 2 + numExtraFields
result[^1].valid = step["valid"].getBool()
elif not step.hasKey"checks" and not step.hasKey"payload_status":
doAssert step.len == 1
doAssert step.len == 1 + numExtraFields
result[^1].valid = true

proc stepOnBlock(
Expand All @@ -218,18 +241,29 @@ proc stepOnBlock(
state: var ForkedHashedBeaconState,
stateCache: var StateCache,
signedBlock: ForkySignedBeaconBlock,
blobData: Opt[BlobData],
time: BeaconTime,
invalidatedRoots: Table[Eth2Digest, Eth2Digest]):
Result[BlockRef, VerifierError] =
# 1. Move state to proper slot.
# 1. Validate blobs
when typeof(signedBlock).toFork() >= ConsensusFork.Deneb:
let kzgCommits = signedBlock.message.body.blob_kzg_commitments.asSeq
if kzgCommits.len > 0 or blobData.isSome:
if blobData.isNone or kzgCommits.validate_blobs(
blobData.get.blobs, blobData.get.proofs).isErr:
return err(VerifierError.Invalid)
else:
doAssert blobData.isNone, "Pre-Deneb test with specified blob data"

# 2. Move state to proper slot
doAssert dag.updateState(
state,
dag.getBlockIdAtSlot(time.slotOrZero).expect("block exists"),
save = false,
stateCache
)

# 2. Add block to DAG
# 3. Add block to DAG
when signedBlock is phase0.SignedBeaconBlock:
type TrustedBlock = phase0.TrustedSignedBeaconBlock
elif signedBlock is altair.SignedBeaconBlock:
Expand Down Expand Up @@ -272,12 +306,12 @@ proc stepOnBlock(
blckRef: BlockRef, signedBlock: TrustedBlock,
epochRef: EpochRef, unrealized: FinalityCheckpoints):

# 3. Update fork choice if valid
# 4. Update fork choice if valid
let status = fkChoice[].process_block(
dag, epochRef, blckRef, unrealized, signedBlock.message, time)
doAssert status.isOk()

# 4. Update DAG with new head
# 5. Update DAG with new head
var quarantine = Quarantine.init()
let newHead = fkChoice[].get_head(dag, time).get()
dag.updateHead(dag.getBlockRef(newHead).get(), quarantine, [])
Expand Down Expand Up @@ -377,7 +411,7 @@ proc doRunTest(path: string, fork: ConsensusFork) =
let status = stepOnBlock(
stores.dag, stores.fkChoice,
verifier, state[], stateCache,
blck, time, invalidatedRoots)
blck, step.blobData, time, invalidatedRoots)
doAssert status.isOk == step.valid
of opOnAttesterSlashing:
let indices =
Expand Down

0 comments on commit 2d94478

Please sign in to comment.