[1/n] feat(deposits): merkle tree machinery to support deposit inclusion verification

beacon/blockchain/deposit.go

@@ -39,7 +39,7 @@ func (s *Service[
 	ctx context.Context,
 	blockNum math.U64,
 ) {
-	if blockNum < s.eth1FollowDistance {
+	if blockNum <= s.eth1FollowDistance {
 		s.logger.Info(
 			"depositFetcher, nothing to fetch",
 			"block num", blockNum,

beacon/blockchain/finalize_block.go

@@ -92,14 +92,12 @@ func (s *Service[
 	// STEP 4: Post Finalizations cleanups
 
 	// fetch and store the deposit for the block
-	blockNum := blk.GetBody().GetExecutionPayload().GetNumber()
-	s.depositFetcher(ctx, blockNum)
+	s.depositFetcher(ctx, blk.GetBody().GetExecutionPayload().GetNumber())
 
 	// store the finalized block in the KVStore.
-	slot := blk.GetSlot()
 	if err = s.blockStore.Set(blk); err != nil {
 		s.logger.Error(
-			"failed to store block", "slot", slot, "error", err,
+			"failed to store block", "slot", blk.GetSlot(), "error", err,
 		)
 	}
 

beacon/blockchain/pruning.go

@@ -49,7 +49,10 @@ func (s *Service[
 	return nil
 }
 
-func depositPruneRangeFn(deposits []*ctypes.Deposit, cs common.ChainSpec) (uint64, uint64) {
+func depositPruneRangeFn(
+	deposits []*ctypes.Deposit,
+	cs common.ChainSpec,
+) (uint64, uint64) {
 	if len(deposits) == 0 || cs.MaxDepositsPerBlock() == 0 {
 		return 0, 0
 	}

consensus-types/types/deposits.go

@@ -42,15 +42,15 @@ func (ds Deposits) SizeSSZ(siz *ssz.Sizer, _ bool) uint32 {
 func (ds Deposits) DefineSSZ(c *ssz.Codec) {
 	c.DefineDecoder(func(*ssz.Decoder) {
 		ssz.DefineSliceOfStaticObjectsContent(
-			c, (*[]*Deposit)(&ds), constants.MaxDepositsPerBlock)
+			c, (*[]*Deposit)(&ds), constants.MaxDeposits)
 	})
 	c.DefineEncoder(func(*ssz.Encoder) {
 		ssz.DefineSliceOfStaticObjectsContent(
-			c, (*[]*Deposit)(&ds), constants.MaxDepositsPerBlock)
+			c, (*[]*Deposit)(&ds), constants.MaxDeposits)
 	})
 	c.DefineHasher(func(*ssz.Hasher) {
 		ssz.DefineSliceOfStaticObjectsOffset(
-			c, (*[]*Deposit)(&ds), constants.MaxDepositsPerBlock)
+			c, (*[]*Deposit)(&ds), constants.MaxDeposits)
 	})
 }
 

node-api/backend/types.go

@@ -67,7 +67,10 @@ type BlockStore[BeaconBlockT any] interface {
 // DepositStore defines the interface for deposit storage.
 type DepositStore interface {
 	// GetDepositsByIndex returns `numView` expected deposits.
-	GetDepositsByIndex(startIndex uint64, numView uint64) ([]*ctypes.Deposit, error)
+	GetDepositsByIndex(
+		startIndex uint64,
+		numView uint64,
+	) ([]*ctypes.Deposit, error)
 	// Prune prunes the deposit store of [start, end)
 	Prune(start, end uint64) error
 	// EnqueueDeposits adds a list of deposits to the deposit store.

node-core/components/store.go

@@ -38,5 +38,8 @@ func ProvideKVStore[
 	ExecutionPayloadHeaderT ExecutionPayloadHeader[ExecutionPayloadHeaderT],
 ](in KVStoreInput) *beacondb.KVStore[ExecutionPayloadHeaderT] {
 	payloadCodec := &encoding.SSZInterfaceCodec[ExecutionPayloadHeaderT]{}
-	return beacondb.New[ExecutionPayloadHeaderT](in.KVStoreService, payloadCodec)
+	return beacondb.New[ExecutionPayloadHeaderT](
+		in.KVStoreService,
+		payloadCodec,
+	)
 }

primitives/constants/payload.go

@@ -31,8 +31,12 @@ const (
 	// payload.
 	MaxTxsPerPayload uint64 = 1048576
 
-	// MaxDepositsPerBlock is the maximum number of deposits per block.
-	MaxDepositsPerBlock uint64 = 16
+	// DepositContractDepth is the depth of the deposit contract merkle tree.
+	DepositContractDepth uint64 = 32
+
+	// MaxDeposits is the maximum number of deposits supported by the
+	// deposit tree (2**32).
+	MaxDeposits uint64 = 1 << DepositContractDepth
 
 	// MaxWithdrawalsPerPayload is the maximum number of withdrawals in a
 	// execution payload.

primitives/math/errors.go

@@ -28,6 +28,9 @@ var (
 	// ErrUnexpectedInputLengthBase is the base error for unexpected input
 	// length errors.
 	ErrUnexpectedInputLengthBase = errors.New("unexpected input length")
+
+	// ErrOverflow is the base error for overflow errors.
+	ErrOverflow = errors.New("integer overflow")
 )
 
 // ErrUnexpectedInputLength returns an error indicating that the input length.

primitives/math/pow/pow.go

@@ -20,6 +20,17 @@
 
 package pow
 
+// TwoToThePowerOf returns 2^n. panics if n >= 64.
+//
+//nolint:mnd // todo fix.
+func TwoToThePowerOf[U64T ~uint64](n U64T) U64T {
+	if n >= 64 {
+		panic("integer overflow")
+	}
+
+	return 1 << n
+}
+
 // PrevPowerOfTwo returns the previous power of 2 for the given input.
 //
 //nolint:mnd // todo fix.

primitives/math/u64.go

@@ -21,6 +21,7 @@
 package math
 
 import (
+	"math"
 	"math/big"
 	"strconv"
 
@@ -155,3 +156,16 @@ func GweiFromWei(i *big.Int) (Gwei, error) {
 func (u Gwei) ToWei() *U256 {
 	return (&U256{}).Mul(NewU256(uint64(u)), NewU256(GweiPerWei))
 }
+
+// ---------------------------- Helper Methods ----------------------------
+
+// Int returns the integer value of the uint64 argument. If there is an
+// overflow, then an error is
+// returned.
+func Int(u uint64) (int, error) {
+	if u > math.MaxInt {
+		return 0, ErrOverflow
+	}
+
+	return int(u), nil
+}

scripts/build/linting.mk

@@ -53,7 +53,7 @@ license-fix:
 
 nilaway:
 	@echo "--> Running nilaway"
-	(go run go.uber.org/nilaway/cmd/nilaway -exclude-errors-in-files "geth-primitives/deposit/" -v ./...) || exit 1;
+	(go run go.uber.org/nilaway/cmd/nilaway -exclude-pkgs "github.com/berachain/beacon-kit/storage/deposit/merkle" -exclude-errors-in-files "geth-primitives/deposit/" -v ./...) || exit 1;
 	@printf "Nilaway check complete\n"
 
 #################

storage/beacondb/registry.go

@@ -30,7 +30,9 @@ import (
 )
 
 // AddValidator registers a new validator in the beacon state.
-func (kv *KVStore[ExecutionPayloadHeaderT]) AddValidator(val *ctypes.Validator) error {
+func (kv *KVStore[ExecutionPayloadHeaderT]) AddValidator(
+	val *ctypes.Validator,
+) error {
 	// Get the next validator index from the sequence.
 	idx, err := kv.validatorIndex.Next(kv.ctx)
 	if err != nil {
@@ -46,7 +48,9 @@ func (kv *KVStore[ExecutionPayloadHeaderT]) AddValidator(val *ctypes.Validator)
 }
 
 // AddValidator registers a new validator in the beacon state.
-func (kv *KVStore[ExecutionPayloadHeaderT]) AddValidatorBartio(val *ctypes.Validator) error {
+func (kv *KVStore[ExecutionPayloadHeaderT]) AddValidatorBartio(
+	val *ctypes.Validator,
+) error {
 	// Get the ne
 	idx, err := kv.validatorIndex.Next(kv.ctx)
 	if err != nil {

storage/deposit/merkle/README.md

@@ -0,0 +1,7 @@
+# Deposit Merkle Tree
+
+This package implements the [EIP-4881 spec](https://eips.ethereum.org/EIPS/eip-4881) for a minimal sparse Merkle tree.
+
+The format proposed by the EIP allows for the pruning of deposits that are no longer needed to participate fully in consensus.
+
+Thanks to [Prysm](https://github.com/prysmaticlabs/prysm/blob/develop/beacon-chain/cache/depositsnapshot/deposit_tree.go) and Mark Mackey ([@ethDreamer](https://github.com/ethDreamer)) for the reference implementation.

storage/deposit/merkle/errors.go

@@ -0,0 +1,65 @@
+// 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/errors"
+
+var (
+	// ErrFinalizedNodeCannotPushLeaf may occur when attempting to push a leaf
+	// to a finalized node.
+	// When a node is finalized, it cannot be modified or changed.
+	ErrFinalizedNodeCannotPushLeaf = errors.New(
+		"can't push a leaf to a finalized node",
+	)
+
+	// ErrLeafNodeCannotPushLeaf may occur when attempting to push a leaf to a
+	// leaf node.
+	ErrLeafNodeCannotPushLeaf = errors.New("can't push a leaf to a leaf node")
+
+	// ErrZeroLevel occurs when the value of level is 0.
+	ErrZeroLevel = errors.New("level should be greater than 0")
+
+	// ErrZeroDepth occurs when the value of depth is 0.
+	ErrZeroDepth = errors.New("depth should be greater than 0")
+)
+
+var (
+	// ErrInvalidSnapshotRoot occurs when the snapshot root does not match the
+	// calculated root.
+	ErrInvalidSnapshotRoot = errors.New("snapshot root is invalid")
+
+	// ErrInvalidDepositCount occurs when the value for mix in length is 0.
+	ErrInvalidDepositCount = errors.New(
+		"deposit count should be greater than 0",
+	)
+
+	// ErrInvalidIndex occurs when the index is less than the number of
+	// finalized deposits.
+	ErrInvalidIndex = errors.New(
+		"index should be greater than finalizedDeposits - 1",
+	)
+
+	// ErrTooManyDeposits occurs when the number of deposits exceeds the
+	// capacity of the tree.
+	ErrTooManyDeposits = errors.New(
+		"number of deposits should not be greater than the capacity of the tree",
+	)
+)