-
Notifications
You must be signed in to change notification settings - Fork 131
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(node-api): proof of proposer index in beacon block (#2099)
- Loading branch information
Showing
19 changed files
with
265 additions
and
38 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
82 changes: 82 additions & 0 deletions
82
mod/node-api/handlers/proof/merkle/block_proposer_index.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,82 @@ | ||
| // SPDX-License-Identifier: BUSL-1.1 | ||
| // | ||
| // Copyright (C) 2024, Berachain Foundation. All rights reserved. | ||
| // Use of this software is governed by the Business Source License included | ||
| // in the LICENSE file of this repository and at www.mariadb.com/bsl11. | ||
| // | ||
| // ANY USE OF THE LICENSED WORK IN VIOLATION OF THIS LICENSE WILL AUTOMATICALLY | ||
| // TERMINATE YOUR RIGHTS UNDER THIS LICENSE FOR THE CURRENT AND ALL OTHER | ||
| // VERSIONS OF THE LICENSED WORK. | ||
| // | ||
| // THIS LICENSE DOES NOT GRANT YOU ANY RIGHT IN ANY TRADEMARK OR LOGO OF | ||
| // LICENSOR OR ITS AFFILIATES (PROVIDED THAT YOU MAY USE A TRADEMARK OR LOGO OF | ||
| // LICENSOR AS EXPRESSLY REQUIRED BY THIS LICENSE). | ||
| // | ||
| // TO THE EXTENT PERMITTED BY APPLICABLE LAW, THE LICENSED WORK IS PROVIDED ON | ||
| // AN “AS IS” BASIS. LICENSOR HEREBY DISCLAIMS ALL WARRANTIES AND CONDITIONS, | ||
| // EXPRESS OR IMPLIED, INCLUDING (WITHOUT LIMITATION) WARRANTIES OF | ||
| // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NON-INFRINGEMENT, AND | ||
| // TITLE. | ||
|
|
||
| package merkle | ||
|
|
||
| import ( | ||
| "github.com/berachain/beacon-kit/mod/errors" | ||
| "github.com/berachain/beacon-kit/mod/node-api/handlers/proof/types" | ||
| "github.com/berachain/beacon-kit/mod/primitives/pkg/common" | ||
| "github.com/berachain/beacon-kit/mod/primitives/pkg/encoding/ssz/merkle" | ||
| ) | ||
|
|
||
| // ProveProposerIndexInBlock generates a proof for the proposer index in the | ||
| // beacon block. The proof is then verified against the beacon block root as a | ||
| // sanity check. Returns the proof along with the beacon block root. It uses | ||
| // the fastssz library to generate the proof. | ||
| func ProveProposerIndexInBlock[ | ||
| BeaconBlockHeaderT types.BeaconBlockHeader, | ||
| ](bbh BeaconBlockHeaderT) ([]common.Root, common.Root, error) { | ||
| blockProofTree, err := bbh.GetTree() | ||
| if err != nil { | ||
| return nil, common.Root{}, err | ||
| } | ||
|
|
||
| proposerIndexProof, err := blockProofTree.Prove( | ||
| ProposerIndexGIndexDenebBlock, | ||
| ) | ||
| if err != nil { | ||
| return nil, common.Root{}, err | ||
| } | ||
|
|
||
| proof := make([]common.Root, len(proposerIndexProof.Hashes)) | ||
| for i, hash := range proposerIndexProof.Hashes { | ||
| proof[i] = common.NewRootFromBytes(hash) | ||
| } | ||
|
|
||
| beaconRoot, err := verifyProposerIndexInBlock( | ||
| bbh, proof, common.NewRootFromBytes(proposerIndexProof.Leaf), | ||
| ) | ||
| if err != nil { | ||
| return nil, common.Root{}, err | ||
| } | ||
|
|
||
| return proof, beaconRoot, nil | ||
| } | ||
|
|
||
| // verifyProposerIndexInBlock verifies the proposer index proof in the block. | ||
| // | ||
| // TODO: verifying the proof is not absolutely necessary. | ||
| func verifyProposerIndexInBlock( | ||
| bbh types.BeaconBlockHeader, proof []common.Root, leaf common.Root, | ||
| ) (common.Root, error) { | ||
| beaconRoot := bbh.HashTreeRoot() | ||
| if beaconRootVerified, err := merkle.VerifyProof( | ||
| ProposerIndexGIndexDenebBlock, leaf, proof, beaconRoot, | ||
| ); err != nil { | ||
| return common.Root{}, err | ||
| } else if !beaconRootVerified { | ||
| return common.Root{}, errors.New( | ||
| "proposer index proof failed to verify against beacon root", | ||
| ) | ||
| } | ||
|
|
||
| return beaconRoot, nil | ||
| } |
84 changes: 84 additions & 0 deletions
84
mod/node-api/handlers/proof/merkle/block_proposer_index_test.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,84 @@ | ||
| // SPDX-License-Identifier: BUSL-1.1 | ||
| // | ||
| // Copyright (C) 2024, Berachain Foundation. All rights reserved. | ||
| // Use of this software is governed by the Business Source License included | ||
| // in the LICENSE file of this repository and at www.mariadb.com/bsl11. | ||
| // | ||
| // ANY USE OF THE LICENSED WORK IN VIOLATION OF THIS LICENSE WILL AUTOMATICALLY | ||
| // TERMINATE YOUR RIGHTS UNDER THIS LICENSE FOR THE CURRENT AND ALL OTHER | ||
| // VERSIONS OF THE LICENSED WORK. | ||
| // | ||
| // THIS LICENSE DOES NOT GRANT YOU ANY RIGHT IN ANY TRADEMARK OR LOGO OF | ||
| // LICENSOR OR ITS AFFILIATES (PROVIDED THAT YOU MAY USE A TRADEMARK OR LOGO OF | ||
| // LICENSOR AS EXPRESSLY REQUIRED BY THIS LICENSE). | ||
| // | ||
| // TO THE EXTENT PERMITTED BY APPLICABLE LAW, THE LICENSED WORK IS PROVIDED ON | ||
| // AN “AS IS” BASIS. LICENSOR HEREBY DISCLAIMS ALL WARRANTIES AND CONDITIONS, | ||
| // EXPRESS OR IMPLIED, INCLUDING (WITHOUT LIMITATION) WARRANTIES OF | ||
| // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NON-INFRINGEMENT, AND | ||
| // TITLE. | ||
|
|
||
| package merkle_test | ||
|
|
||
| import ( | ||
| "testing" | ||
|
|
||
| "github.com/berachain/beacon-kit/mod/consensus-types/pkg/types" | ||
| "github.com/berachain/beacon-kit/mod/node-api/handlers/proof/merkle" | ||
| "github.com/berachain/beacon-kit/mod/primitives/pkg/common" | ||
| "github.com/berachain/beacon-kit/mod/primitives/pkg/math" | ||
| "github.com/stretchr/testify/require" | ||
| ) | ||
|
|
||
| // TestBlockProposerIndexProof tests the ProveProposerIndexInBlock function | ||
| // and that the generated proof correctly verifies. | ||
| func TestBlockProposerIndexProof(t *testing.T) { | ||
| testCases := []struct { | ||
| name string | ||
| slot math.Slot | ||
| proposerIndex math.ValidatorIndex | ||
| parentBlockRoot common.Root | ||
| stateRoot common.Root | ||
| bodyRoot common.Root | ||
| expectedProofFile string | ||
| }{ | ||
| { | ||
| name: "1 Validator Set", | ||
| slot: 69, | ||
| proposerIndex: 0, | ||
| parentBlockRoot: common.Root{1, 2, 3}, | ||
| stateRoot: common.Root{4, 5, 6}, | ||
| bodyRoot: common.Root{7, 8, 9}, | ||
| expectedProofFile: "one_validator_proposer_index_proof.json", | ||
| }, | ||
| { | ||
| name: "Many Validator Set", | ||
| slot: 420, | ||
| proposerIndex: 69, | ||
| parentBlockRoot: common.Root{1, 2, 3}, | ||
| stateRoot: common.Root{4, 5, 6}, | ||
| bodyRoot: common.Root{7, 8, 9}, | ||
| expectedProofFile: "many_validators_proposer_index_proof.json", | ||
| }, | ||
| } | ||
|
|
||
| for _, tc := range testCases { | ||
| t.Run(tc.name, func(t *testing.T) { | ||
| bbh := (&types.BeaconBlockHeader{}).New( | ||
| tc.slot, | ||
| tc.proposerIndex, | ||
| tc.parentBlockRoot, | ||
| tc.stateRoot, | ||
| tc.bodyRoot, | ||
| ) | ||
|
|
||
| proof, beaconRoot, err := merkle.ProveProposerIndexInBlock(bbh) | ||
| require.NoError(t, err) | ||
|
|
||
| require.Equal(t, bbh.HashTreeRoot(), beaconRoot) | ||
|
|
||
| expectedProof := ReadProofFromFile(t, tc.expectedProofFile) | ||
| require.Equal(t, expectedProof, proof) | ||
| }) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.