mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-02 16:21:42 +00:00
Add regression test for verify_header_signature fork block bug
Co-authored-by: michaelsproul <4452260+michaelsproul@users.noreply.github.com>
This commit is contained in:
@@ -115,3 +115,104 @@ async fn rpc_columns_with_invalid_header_signature() {
|
||||
BlockError::InvalidSignature(InvalidSignature::ProposerSignature)
|
||||
));
|
||||
}
|
||||
|
||||
// Regression test for verify_header_signature bug: it uses head_fork() which is wrong for fork blocks
|
||||
#[tokio::test]
|
||||
async fn verify_header_signature_fork_block_bug() {
|
||||
// Create a spec with all forks enabled at genesis except Fulu which is at epoch 1
|
||||
// This allows us to easily create the scenario where the head is at Electra
|
||||
// but we're trying to verify a block from Fulu epoch
|
||||
let mut spec = test_spec::<E>();
|
||||
spec.altair_fork_epoch = Some(Epoch::new(0));
|
||||
spec.bellatrix_fork_epoch = Some(Epoch::new(0));
|
||||
spec.capella_fork_epoch = Some(Epoch::new(0));
|
||||
spec.deneb_fork_epoch = Some(Epoch::new(0));
|
||||
spec.electra_fork_epoch = Some(Epoch::new(0));
|
||||
let fulu_fork_epoch = Epoch::new(1);
|
||||
spec.fulu_fork_epoch = Some(fulu_fork_epoch);
|
||||
|
||||
// Only run this test if columns are enabled.
|
||||
if !spec.is_fulu_scheduled() {
|
||||
return;
|
||||
}
|
||||
|
||||
let spec = Arc::new(spec);
|
||||
let harness = get_harness(VALIDATOR_COUNT, spec.clone(), NodeCustodyType::Supernode);
|
||||
harness.execution_block_generator().set_min_blob_count(1);
|
||||
|
||||
// Add some blocks in epoch 0 (Electra)
|
||||
harness
|
||||
.extend_chain(
|
||||
E::slots_per_epoch() as usize - 1,
|
||||
BlockStrategy::OnCanonicalHead,
|
||||
AttestationStrategy::AllValidators,
|
||||
)
|
||||
.await;
|
||||
|
||||
// Verify we're still in epoch 0 (Electra)
|
||||
let pre_fork_state = harness.get_current_state();
|
||||
assert_eq!(pre_fork_state.current_epoch(), Epoch::new(0));
|
||||
assert!(matches!(pre_fork_state, BeaconState::Electra(_)));
|
||||
|
||||
// Now produce a block at the first slot of epoch 1 (Fulu fork).
|
||||
// make_block will advance the state which will trigger the Electra->Fulu upgrade.
|
||||
let fork_slot = fulu_fork_epoch.start_slot(E::slots_per_epoch());
|
||||
let ((signed_block, opt_blobs), _state_root) = harness
|
||||
.make_block(pre_fork_state.clone(), fork_slot)
|
||||
.await;
|
||||
let (_, blobs) = opt_blobs.expect("Blobs should be present");
|
||||
assert!(!blobs.is_empty(), "Block should have blobs");
|
||||
let block_root = signed_block.canonical_root();
|
||||
|
||||
// Process the block WITHOUT blobs to make it unavailable.
|
||||
// The block will be accepted but won't become the head because it's not fully available.
|
||||
// This keeps the head at the pre-fork state (Electra).
|
||||
harness.advance_slot();
|
||||
let rpc_block = harness
|
||||
.build_rpc_block_from_blobs(block_root, signed_block.clone(), None)
|
||||
.expect("Should build RPC block");
|
||||
let availability = harness
|
||||
.chain
|
||||
.process_block(
|
||||
block_root,
|
||||
rpc_block,
|
||||
NotifyExecutionLayer::Yes,
|
||||
BlockImportSource::RangeSync,
|
||||
|| Ok(()),
|
||||
)
|
||||
.await
|
||||
.expect("Block should be processed");
|
||||
assert_eq!(
|
||||
availability,
|
||||
AvailabilityProcessingStatus::MissingComponents(fork_slot, block_root),
|
||||
"Block should be pending availability"
|
||||
);
|
||||
|
||||
// The head should still be in epoch 0 (Electra) because the fork block isn't available
|
||||
let current_head_state = harness.get_current_state();
|
||||
assert_eq!(current_head_state.current_epoch(), Epoch::new(0));
|
||||
assert!(matches!(current_head_state, BeaconState::Electra(_)));
|
||||
|
||||
// Now try to process columns for the fork block.
|
||||
// The bug: verify_header_signature uses head_fork() which gets the fork from
|
||||
// the head state (still Electra fork), but the block was signed with the Fulu fork version.
|
||||
// This causes incorrect signature verification failure.
|
||||
let data_column_sidecars =
|
||||
generate_data_column_sidecars_from_block(&signed_block, &harness.chain.spec);
|
||||
|
||||
let err = harness
|
||||
.chain
|
||||
.process_rpc_custody_columns(data_column_sidecars)
|
||||
.await
|
||||
.expect_err("Should fail due to verify_header_signature bug");
|
||||
|
||||
// The bug causes this to fail with InvalidSignature even though the signature is actually valid
|
||||
assert!(
|
||||
matches!(
|
||||
err,
|
||||
BlockError::InvalidSignature(InvalidSignature::ProposerSignature)
|
||||
),
|
||||
"Expected InvalidSignature(ProposerSignature) error due to fork mismatch bug, got: {:?}",
|
||||
err
|
||||
);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user