mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-31 08:08:18 -05:00
Compare commits
6 Commits
develop
...
e2e-debugg
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
35720e3f71 | ||
|
|
6effcf5d53 | ||
|
|
bd778dad3a | ||
|
|
ceadf6e5c9 | ||
|
|
22769ed486 | ||
|
|
b960e54e00 |
@@ -114,32 +114,17 @@ func payloadCommittee(ctx context.Context, st state.ReadOnlyBeaconState, slot pr
|
|||||||
}
|
}
|
||||||
|
|
||||||
committeesPerSlot := helpers.SlotCommitteeCount(activeCount)
|
committeesPerSlot := helpers.SlotCommitteeCount(activeCount)
|
||||||
|
out := make([]primitives.ValidatorIndex, 0, activeCount/uint64(params.BeaconConfig().SlotsPerEpoch))
|
||||||
|
|
||||||
selected := make([]primitives.ValidatorIndex, 0, fieldparams.PTCSize)
|
for i := primitives.CommitteeIndex(0); i < primitives.CommitteeIndex(committeesPerSlot); i++ {
|
||||||
var i uint64
|
committee, err := helpers.BeaconCommitteeFromState(ctx, st, slot, i)
|
||||||
for uint64(len(selected)) < fieldparams.PTCSize {
|
|
||||||
if ctx.Err() != nil {
|
|
||||||
return nil, ctx.Err()
|
|
||||||
}
|
|
||||||
|
|
||||||
for committeeIndex := primitives.CommitteeIndex(0); committeeIndex < primitives.CommitteeIndex(committeesPerSlot); committeeIndex++ {
|
|
||||||
if uint64(len(selected)) >= fieldparams.PTCSize {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
committee, err := helpers.BeaconCommitteeFromState(ctx, st, slot, committeeIndex)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, errors.Wrapf(err, "failed to get beacon committee %d", committeeIndex)
|
return nil, errors.Wrapf(err, "failed to get beacon committee %d", i)
|
||||||
|
}
|
||||||
|
out = append(out, committee...)
|
||||||
}
|
}
|
||||||
|
|
||||||
selected, i, err = selectByBalanceFill(ctx, st, committee, seed, selected, i)
|
return selectByBalance(ctx, st, out, seed, fieldparams.PTCSize)
|
||||||
if err != nil {
|
|
||||||
return nil, errors.Wrapf(err, "failed to sample beacon committee %d", committeeIndex)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return selected, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ptcSeed computes the seed for the payload timeliness committee.
|
// ptcSeed computes the seed for the payload timeliness committee.
|
||||||
@@ -163,39 +148,33 @@ func ptcSeed(st state.ReadOnlyBeaconState, epoch primitives.Epoch, slot primitiv
|
|||||||
// if compute_balance_weighted_acceptance(state, indices[next], seed, i):
|
// if compute_balance_weighted_acceptance(state, indices[next], seed, i):
|
||||||
// selected.append(indices[next])
|
// selected.append(indices[next])
|
||||||
// i += 1
|
// i += 1
|
||||||
func selectByBalanceFill(
|
func selectByBalance(ctx context.Context, st state.ReadOnlyBeaconState, candidates []primitives.ValidatorIndex, seed [32]byte, count uint64) ([]primitives.ValidatorIndex, error) {
|
||||||
ctx context.Context,
|
if len(candidates) == 0 {
|
||||||
st state.ReadOnlyBeaconState,
|
return nil, errors.New("no candidates for balance weighted selection")
|
||||||
candidates []primitives.ValidatorIndex,
|
}
|
||||||
seed [32]byte,
|
|
||||||
selected []primitives.ValidatorIndex,
|
|
||||||
i uint64,
|
|
||||||
) ([]primitives.ValidatorIndex, uint64, error) {
|
|
||||||
hashFunc := hash.CustomSHA256Hasher()
|
hashFunc := hash.CustomSHA256Hasher()
|
||||||
// Pre-allocate buffer for hash input: seed (32 bytes) + round counter (8 bytes).
|
// Pre-allocate buffer for hash input: seed (32 bytes) + round counter (8 bytes).
|
||||||
var buf [40]byte
|
var buf [40]byte
|
||||||
copy(buf[:], seed[:])
|
copy(buf[:], seed[:])
|
||||||
maxBalance := params.BeaconConfig().MaxEffectiveBalanceElectra
|
maxBalance := params.BeaconConfig().MaxEffectiveBalanceElectra
|
||||||
|
|
||||||
for _, idx := range candidates {
|
selected := make([]primitives.ValidatorIndex, 0, count)
|
||||||
|
total := uint64(len(candidates))
|
||||||
|
for i := uint64(0); uint64(len(selected)) < count; i++ {
|
||||||
if ctx.Err() != nil {
|
if ctx.Err() != nil {
|
||||||
return nil, i, ctx.Err()
|
return nil, ctx.Err()
|
||||||
}
|
}
|
||||||
|
idx := candidates[i%total]
|
||||||
ok, err := acceptByBalance(st, idx, buf[:], hashFunc, maxBalance, i)
|
ok, err := acceptByBalance(st, idx, buf[:], hashFunc, maxBalance, i)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, i, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if ok {
|
if ok {
|
||||||
selected = append(selected, idx)
|
selected = append(selected, idx)
|
||||||
}
|
}
|
||||||
if uint64(len(selected)) == fieldparams.PTCSize {
|
|
||||||
break
|
|
||||||
}
|
}
|
||||||
i++
|
return selected, nil
|
||||||
}
|
|
||||||
|
|
||||||
return selected, i, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// acceptByBalance determines if a validator is accepted based on its effective balance.
|
// acceptByBalance determines if a validator is accepted based on its effective balance.
|
||||||
|
|||||||
@@ -67,7 +67,6 @@ func getSubscriptionStatusFromDB(t *testing.T, db *Store) bool {
|
|||||||
return subscribed
|
return subscribed
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func TestUpdateCustodyInfo(t *testing.T) {
|
func TestUpdateCustodyInfo(t *testing.T) {
|
||||||
ctx := t.Context()
|
ctx := t.Context()
|
||||||
|
|
||||||
|
|||||||
@@ -575,7 +575,7 @@ func (s *Service) beaconEndpoints(
|
|||||||
name: namespace + ".PublishBlockV2",
|
name: namespace + ".PublishBlockV2",
|
||||||
middleware: []middleware.Middleware{
|
middleware: []middleware.Middleware{
|
||||||
middleware.ContentTypeHandler([]string{api.JsonMediaType, api.OctetStreamMediaType}),
|
middleware.ContentTypeHandler([]string{api.JsonMediaType, api.OctetStreamMediaType}),
|
||||||
middleware.AcceptHeaderHandler([]string{api.JsonMediaType}),
|
middleware.AcceptHeaderHandler([]string{api.JsonMediaType, api.OctetStreamMediaType}),
|
||||||
middleware.AcceptEncodingHeaderHandler(),
|
middleware.AcceptEncodingHeaderHandler(),
|
||||||
},
|
},
|
||||||
handler: server.PublishBlockV2,
|
handler: server.PublishBlockV2,
|
||||||
@@ -586,7 +586,7 @@ func (s *Service) beaconEndpoints(
|
|||||||
name: namespace + ".PublishBlindedBlockV2",
|
name: namespace + ".PublishBlindedBlockV2",
|
||||||
middleware: []middleware.Middleware{
|
middleware: []middleware.Middleware{
|
||||||
middleware.ContentTypeHandler([]string{api.JsonMediaType, api.OctetStreamMediaType}),
|
middleware.ContentTypeHandler([]string{api.JsonMediaType, api.OctetStreamMediaType}),
|
||||||
middleware.AcceptHeaderHandler([]string{api.JsonMediaType}),
|
middleware.AcceptHeaderHandler([]string{api.JsonMediaType, api.OctetStreamMediaType}),
|
||||||
middleware.AcceptEncodingHeaderHandler(),
|
middleware.AcceptEncodingHeaderHandler(),
|
||||||
},
|
},
|
||||||
handler: server.PublishBlindedBlockV2,
|
handler: server.PublishBlindedBlockV2,
|
||||||
|
|||||||
@@ -26,8 +26,8 @@ import (
|
|||||||
"github.com/OffchainLabs/prysm/v7/beacon-chain/operations/voluntaryexits/mock"
|
"github.com/OffchainLabs/prysm/v7/beacon-chain/operations/voluntaryexits/mock"
|
||||||
p2pMock "github.com/OffchainLabs/prysm/v7/beacon-chain/p2p/testing"
|
p2pMock "github.com/OffchainLabs/prysm/v7/beacon-chain/p2p/testing"
|
||||||
"github.com/OffchainLabs/prysm/v7/beacon-chain/rpc/core"
|
"github.com/OffchainLabs/prysm/v7/beacon-chain/rpc/core"
|
||||||
mockSync "github.com/OffchainLabs/prysm/v7/beacon-chain/sync/initial-sync/testing"
|
|
||||||
state_native "github.com/OffchainLabs/prysm/v7/beacon-chain/state/state-native"
|
state_native "github.com/OffchainLabs/prysm/v7/beacon-chain/state/state-native"
|
||||||
|
mockSync "github.com/OffchainLabs/prysm/v7/beacon-chain/sync/initial-sync/testing"
|
||||||
"github.com/OffchainLabs/prysm/v7/config/params"
|
"github.com/OffchainLabs/prysm/v7/config/params"
|
||||||
"github.com/OffchainLabs/prysm/v7/consensus-types/primitives"
|
"github.com/OffchainLabs/prysm/v7/consensus-types/primitives"
|
||||||
"github.com/OffchainLabs/prysm/v7/crypto/bls"
|
"github.com/OffchainLabs/prysm/v7/crypto/bls"
|
||||||
|
|||||||
@@ -48,7 +48,6 @@ go_test(
|
|||||||
"@com_github_ethereum_go_ethereum//crypto:go_default_library",
|
"@com_github_ethereum_go_ethereum//crypto:go_default_library",
|
||||||
"@com_github_ethereum_go_ethereum//p2p/enode:go_default_library",
|
"@com_github_ethereum_go_ethereum//p2p/enode:go_default_library",
|
||||||
"@org_golang_google_grpc//:go_default_library",
|
"@org_golang_google_grpc//:go_default_library",
|
||||||
"@org_golang_google_grpc//metadata:go_default_library",
|
|
||||||
"@org_golang_google_grpc//reflection:go_default_library",
|
"@org_golang_google_grpc//reflection:go_default_library",
|
||||||
"@org_golang_google_protobuf//types/known/emptypb:go_default_library",
|
"@org_golang_google_protobuf//types/known/emptypb:go_default_library",
|
||||||
"@org_golang_google_protobuf//types/known/timestamppb:go_default_library",
|
"@org_golang_google_protobuf//types/known/timestamppb:go_default_library",
|
||||||
|
|||||||
@@ -47,7 +47,6 @@ type Server struct {
|
|||||||
POWChainInfoFetcher execution.ChainInfoFetcher
|
POWChainInfoFetcher execution.ChainInfoFetcher
|
||||||
BeaconMonitoringHost string
|
BeaconMonitoringHost string
|
||||||
BeaconMonitoringPort int
|
BeaconMonitoringPort int
|
||||||
OptimisticModeFetcher blockchain.OptimisticModeFetcher
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Deprecated: The gRPC API will remain the default and fully supported through v8 (expected in 2026) but will be eventually removed in favor of REST API.
|
// Deprecated: The gRPC API will remain the default and fully supported through v8 (expected in 2026) but will be eventually removed in favor of REST API.
|
||||||
@@ -62,28 +61,21 @@ func (ns *Server) GetHealth(ctx context.Context, request *ethpb.HealthRequest) (
|
|||||||
ctx, cancel := context.WithTimeout(ctx, timeoutDuration)
|
ctx, cancel := context.WithTimeout(ctx, timeoutDuration)
|
||||||
defer cancel() // Important to avoid a context leak
|
defer cancel() // Important to avoid a context leak
|
||||||
|
|
||||||
// Check optimistic status - validators should not participate when optimistic
|
if ns.SyncChecker.Synced() {
|
||||||
isOptimistic, err := ns.OptimisticModeFetcher.IsOptimistic(ctx)
|
|
||||||
if err != nil {
|
|
||||||
return &empty.Empty{}, status.Errorf(codes.Internal, "Could not check optimistic status: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if ns.SyncChecker.Synced() && !isOptimistic {
|
|
||||||
return &empty.Empty{}, nil
|
return &empty.Empty{}, nil
|
||||||
}
|
}
|
||||||
if ns.SyncChecker.Syncing() || ns.SyncChecker.Initialized() {
|
if ns.SyncChecker.Syncing() || ns.SyncChecker.Initialized() {
|
||||||
// Set header for REST API clients (via gRPC-gateway)
|
if request.SyncingStatus != 0 {
|
||||||
|
// override the 200 success with the provided request status
|
||||||
|
if err := grpc.SetHeader(ctx, metadata.Pairs("x-http-code", strconv.FormatUint(request.SyncingStatus, 10))); err != nil {
|
||||||
|
return &empty.Empty{}, status.Errorf(codes.Internal, "Could not set custom success code header: %v", err)
|
||||||
|
}
|
||||||
|
return &empty.Empty{}, nil
|
||||||
|
}
|
||||||
if err := grpc.SetHeader(ctx, metadata.Pairs("x-http-code", strconv.FormatUint(http.StatusPartialContent, 10))); err != nil {
|
if err := grpc.SetHeader(ctx, metadata.Pairs("x-http-code", strconv.FormatUint(http.StatusPartialContent, 10))); err != nil {
|
||||||
return &empty.Empty{}, status.Errorf(codes.Internal, "Could not set status code header: %v", err)
|
return &empty.Empty{}, status.Errorf(codes.Internal, "Could not set custom success code header: %v", err)
|
||||||
}
|
}
|
||||||
return &empty.Empty{}, status.Error(codes.Unavailable, "node is syncing")
|
return &empty.Empty{}, nil
|
||||||
}
|
|
||||||
if isOptimistic {
|
|
||||||
// Set header for REST API clients (via gRPC-gateway)
|
|
||||||
if err := grpc.SetHeader(ctx, metadata.Pairs("x-http-code", strconv.FormatUint(http.StatusPartialContent, 10))); err != nil {
|
|
||||||
return &empty.Empty{}, status.Errorf(codes.Internal, "Could not set status code header: %v", err)
|
|
||||||
}
|
|
||||||
return &empty.Empty{}, status.Error(codes.Unavailable, "node is optimistic")
|
|
||||||
}
|
}
|
||||||
return &empty.Empty{}, status.Errorf(codes.Unavailable, "service unavailable")
|
return &empty.Empty{}, status.Errorf(codes.Unavailable, "service unavailable")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ package node
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"maps"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -22,7 +21,6 @@ import (
|
|||||||
"github.com/ethereum/go-ethereum/crypto"
|
"github.com/ethereum/go-ethereum/crypto"
|
||||||
"github.com/ethereum/go-ethereum/p2p/enode"
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
"google.golang.org/grpc/metadata"
|
|
||||||
"google.golang.org/grpc/reflection"
|
"google.golang.org/grpc/reflection"
|
||||||
"google.golang.org/protobuf/types/known/emptypb"
|
"google.golang.org/protobuf/types/known/emptypb"
|
||||||
"google.golang.org/protobuf/types/known/timestamppb"
|
"google.golang.org/protobuf/types/known/timestamppb"
|
||||||
@@ -189,71 +187,32 @@ func TestNodeServer_GetETH1ConnectionStatus(t *testing.T) {
|
|||||||
assert.Equal(t, errStr, res.CurrentConnectionError)
|
assert.Equal(t, errStr, res.CurrentConnectionError)
|
||||||
}
|
}
|
||||||
|
|
||||||
// mockServerTransportStream implements grpc.ServerTransportStream for testing
|
|
||||||
type mockServerTransportStream struct {
|
|
||||||
headers map[string][]string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *mockServerTransportStream) Method() string { return "" }
|
|
||||||
func (m *mockServerTransportStream) SetHeader(md metadata.MD) error {
|
|
||||||
maps.Copy(m.headers, md)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
func (m *mockServerTransportStream) SendHeader(metadata.MD) error { return nil }
|
|
||||||
func (m *mockServerTransportStream) SetTrailer(metadata.MD) error { return nil }
|
|
||||||
|
|
||||||
func TestNodeServer_GetHealth(t *testing.T) {
|
func TestNodeServer_GetHealth(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
input *mockSync.Sync
|
input *mockSync.Sync
|
||||||
isOptimistic bool
|
customStatus uint64
|
||||||
wantedErr string
|
wantedErr string
|
||||||
}{
|
}{
|
||||||
{
|
{
|
||||||
name: "happy path - synced and not optimistic",
|
name: "happy path",
|
||||||
input: &mockSync.Sync{IsSyncing: false, IsSynced: true},
|
input: &mockSync.Sync{IsSyncing: false, IsSynced: true},
|
||||||
isOptimistic: false,
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "returns error when not synced and not syncing",
|
name: "syncing",
|
||||||
input: &mockSync.Sync{IsSyncing: false, IsSynced: false},
|
input: &mockSync.Sync{IsSyncing: false},
|
||||||
isOptimistic: false,
|
|
||||||
wantedErr: "service unavailable",
|
wantedErr: "service unavailable",
|
||||||
},
|
},
|
||||||
{
|
|
||||||
name: "returns error when syncing",
|
|
||||||
input: &mockSync.Sync{IsSyncing: true, IsSynced: false},
|
|
||||||
isOptimistic: false,
|
|
||||||
wantedErr: "node is syncing",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "returns error when synced but optimistic",
|
|
||||||
input: &mockSync.Sync{IsSyncing: false, IsSynced: true},
|
|
||||||
isOptimistic: true,
|
|
||||||
wantedErr: "node is optimistic",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "returns error when syncing and optimistic",
|
|
||||||
input: &mockSync.Sync{IsSyncing: true, IsSynced: false},
|
|
||||||
isOptimistic: true,
|
|
||||||
wantedErr: "node is syncing",
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
server := grpc.NewServer()
|
server := grpc.NewServer()
|
||||||
ns := &Server{
|
ns := &Server{
|
||||||
SyncChecker: tt.input,
|
SyncChecker: tt.input,
|
||||||
OptimisticModeFetcher: &mock.ChainService{Optimistic: tt.isOptimistic},
|
|
||||||
}
|
}
|
||||||
ethpb.RegisterNodeServer(server, ns)
|
ethpb.RegisterNodeServer(server, ns)
|
||||||
reflection.Register(server)
|
reflection.Register(server)
|
||||||
|
_, err := ns.GetHealth(t.Context(), ðpb.HealthRequest{SyncingStatus: tt.customStatus})
|
||||||
// Create context with mock transport stream so grpc.SetHeader works
|
|
||||||
stream := &mockServerTransportStream{headers: make(map[string][]string)}
|
|
||||||
ctx := grpc.NewContextWithServerTransportStream(t.Context(), stream)
|
|
||||||
|
|
||||||
_, err := ns.GetHealth(ctx, ðpb.HealthRequest{})
|
|
||||||
if tt.wantedErr == "" {
|
if tt.wantedErr == "" {
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
return
|
return
|
||||||
|
|||||||
@@ -271,7 +271,6 @@ func NewService(ctx context.Context, cfg *Config) *Service {
|
|||||||
POWChainInfoFetcher: s.cfg.ExecutionChainInfoFetcher,
|
POWChainInfoFetcher: s.cfg.ExecutionChainInfoFetcher,
|
||||||
BeaconMonitoringHost: s.cfg.BeaconMonitoringHost,
|
BeaconMonitoringHost: s.cfg.BeaconMonitoringHost,
|
||||||
BeaconMonitoringPort: s.cfg.BeaconMonitoringPort,
|
BeaconMonitoringPort: s.cfg.BeaconMonitoringPort,
|
||||||
OptimisticModeFetcher: s.cfg.OptimisticModeFetcher,
|
|
||||||
}
|
}
|
||||||
beaconChainServer := &beaconv1alpha1.Server{
|
beaconChainServer := &beaconv1alpha1.Server{
|
||||||
Ctx: s.ctx,
|
Ctx: s.ctx,
|
||||||
|
|||||||
@@ -1,6 +0,0 @@
|
|||||||
### Added
|
|
||||||
|
|
||||||
- Added new proofCollector type to ssz-query
|
|
||||||
|
|
||||||
### Ignored
|
|
||||||
- Added testing covering the production of Merkle proof from Phase0 beacon state and benchmarked against real Hoodi beacon state (Fulu version)
|
|
||||||
@@ -1,3 +0,0 @@
|
|||||||
### Changed
|
|
||||||
|
|
||||||
- gRPC health endpoint will now return an error on syncing or optimistic status showing that it's unavailable.
|
|
||||||
@@ -1,3 +0,0 @@
|
|||||||
### Added
|
|
||||||
|
|
||||||
- Added README for maintaining specrefs.
|
|
||||||
@@ -1,3 +0,0 @@
|
|||||||
### Added
|
|
||||||
|
|
||||||
- The ability to download the nightly reference tests from a specific day.
|
|
||||||
@@ -1,2 +0,0 @@
|
|||||||
### Changed
|
|
||||||
- Sample PTC per committee to reduce allocations.
|
|
||||||
@@ -163,18 +163,3 @@ func Uint256ToSSZBytes(num string) ([]byte, error) {
|
|||||||
}
|
}
|
||||||
return PadTo(ReverseByteOrder(uint256.Bytes()), 32), nil
|
return PadTo(ReverseByteOrder(uint256.Bytes()), 32), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// PutLittleEndian writes an unsigned integer value in little-endian format.
|
|
||||||
// Supports sizes 1, 2, 4, or 8 bytes for uint8/16/32/64 respectively.
|
|
||||||
func PutLittleEndian(dst []byte, val uint64, size int) {
|
|
||||||
switch size {
|
|
||||||
case 1:
|
|
||||||
dst[0] = byte(val)
|
|
||||||
case 2:
|
|
||||||
binary.LittleEndian.PutUint16(dst, uint16(val))
|
|
||||||
case 4:
|
|
||||||
binary.LittleEndian.PutUint32(dst, uint32(val))
|
|
||||||
case 8:
|
|
||||||
binary.LittleEndian.PutUint64(dst, val)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -9,9 +9,7 @@ go_library(
|
|||||||
"container.go",
|
"container.go",
|
||||||
"generalized_index.go",
|
"generalized_index.go",
|
||||||
"list.go",
|
"list.go",
|
||||||
"merkle_proof.go",
|
|
||||||
"path.go",
|
"path.go",
|
||||||
"proof_collector.go",
|
|
||||||
"query.go",
|
"query.go",
|
||||||
"ssz_info.go",
|
"ssz_info.go",
|
||||||
"ssz_object.go",
|
"ssz_object.go",
|
||||||
@@ -22,12 +20,7 @@ go_library(
|
|||||||
importpath = "github.com/OffchainLabs/prysm/v7/encoding/ssz/query",
|
importpath = "github.com/OffchainLabs/prysm/v7/encoding/ssz/query",
|
||||||
visibility = ["//visibility:public"],
|
visibility = ["//visibility:public"],
|
||||||
deps = [
|
deps = [
|
||||||
"//container/trie:go_default_library",
|
|
||||||
"//crypto/hash/htr:go_default_library",
|
|
||||||
"//encoding/bytesutil:go_default_library",
|
|
||||||
"//encoding/ssz:go_default_library",
|
"//encoding/ssz:go_default_library",
|
||||||
"//math:go_default_library",
|
|
||||||
"@com_github_prysmaticlabs_fastssz//:go_default_library",
|
|
||||||
"@com_github_prysmaticlabs_go_bitfield//:go_default_library",
|
"@com_github_prysmaticlabs_go_bitfield//:go_default_library",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
@@ -36,24 +29,15 @@ go_test(
|
|||||||
name = "go_default_test",
|
name = "go_default_test",
|
||||||
srcs = [
|
srcs = [
|
||||||
"generalized_index_test.go",
|
"generalized_index_test.go",
|
||||||
"merkle_proof_test.go",
|
|
||||||
"path_test.go",
|
"path_test.go",
|
||||||
"proof_collector_test.go",
|
|
||||||
"query_test.go",
|
"query_test.go",
|
||||||
"tag_parser_test.go",
|
"tag_parser_test.go",
|
||||||
],
|
],
|
||||||
embed = [":go_default_library"],
|
|
||||||
deps = [
|
deps = [
|
||||||
"//beacon-chain/state/stateutil:go_default_library",
|
":go_default_library",
|
||||||
"//consensus-types/blocks:go_default_library",
|
|
||||||
"//consensus-types/primitives:go_default_library",
|
|
||||||
"//encoding/ssz:go_default_library",
|
|
||||||
"//encoding/ssz/query/testutil:go_default_library",
|
"//encoding/ssz/query/testutil:go_default_library",
|
||||||
"//proto/prysm/v1alpha1:go_default_library",
|
|
||||||
"//proto/ssz_query/testing:go_default_library",
|
"//proto/ssz_query/testing:go_default_library",
|
||||||
"//testing/require:go_default_library",
|
"//testing/require:go_default_library",
|
||||||
"//testing/util:go_default_library",
|
|
||||||
"@com_github_prysmaticlabs_fastssz//:go_default_library",
|
|
||||||
"@com_github_prysmaticlabs_go_bitfield//:go_default_library",
|
"@com_github_prysmaticlabs_go_bitfield//:go_default_library",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,34 +0,0 @@
|
|||||||
package query
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"reflect"
|
|
||||||
|
|
||||||
fastssz "github.com/prysmaticlabs/fastssz"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Prove is the entrypoint to generate an SSZ Merkle proof for the given generalized index.
|
|
||||||
// Parameters:
|
|
||||||
// - gindex: the generalized index of the node to prove inclusion for.
|
|
||||||
// Returns:
|
|
||||||
// - fastssz.Proof: the Merkle proof containing the leaf, index, and sibling hashes.
|
|
||||||
// - error: any error encountered during proof generation.
|
|
||||||
func (info *SszInfo) Prove(gindex uint64) (*fastssz.Proof, error) {
|
|
||||||
if info == nil {
|
|
||||||
return nil, fmt.Errorf("nil SszInfo")
|
|
||||||
}
|
|
||||||
|
|
||||||
collector := newProofCollector()
|
|
||||||
collector.addTarget(gindex)
|
|
||||||
|
|
||||||
// info.source is guaranteed to be valid and dereferenced by AnalyzeObject
|
|
||||||
v := reflect.ValueOf(info.source).Elem()
|
|
||||||
|
|
||||||
// Start the merkleization and proof collection process.
|
|
||||||
// In SSZ generalized indices, the root is always at index 1.
|
|
||||||
if _, err := collector.merkleize(info, v, 1); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return collector.toProof()
|
|
||||||
}
|
|
||||||
@@ -1,163 +0,0 @@
|
|||||||
package query_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/OffchainLabs/go-bitfield"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/consensus-types/blocks"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/consensus-types/primitives"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/encoding/ssz/query"
|
|
||||||
eth "github.com/OffchainLabs/prysm/v7/proto/prysm/v1alpha1"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/testing/require"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/testing/util"
|
|
||||||
ssz "github.com/prysmaticlabs/fastssz"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestProve_FixedTestContainer(t *testing.T) {
|
|
||||||
obj := createFixedTestContainer()
|
|
||||||
|
|
||||||
tests := []string{
|
|
||||||
".field_uint32",
|
|
||||||
".nested.value2",
|
|
||||||
".vector_field[3]",
|
|
||||||
".bitvector64_field",
|
|
||||||
".trailing_field",
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range tests {
|
|
||||||
t.Run(tc, func(t *testing.T) {
|
|
||||||
proveAndVerify(t, obj, tc)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProve_VariableTestContainer(t *testing.T) {
|
|
||||||
obj := createVariableTestContainer()
|
|
||||||
|
|
||||||
tests := []string{
|
|
||||||
".leading_field",
|
|
||||||
".field_list_uint64[2]",
|
|
||||||
"len(field_list_uint64)",
|
|
||||||
".nested.nested_list_field[1]",
|
|
||||||
".variable_container_list[0].inner_1.field_list_uint64[1]",
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range tests {
|
|
||||||
t.Run(tc, func(t *testing.T) {
|
|
||||||
proveAndVerify(t, obj, tc)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProve_BeaconBlock(t *testing.T) {
|
|
||||||
randaoReveal := make([]byte, 96)
|
|
||||||
for i := range randaoReveal {
|
|
||||||
randaoReveal[i] = 0x42
|
|
||||||
}
|
|
||||||
root32 := make([]byte, 32)
|
|
||||||
for i := range root32 {
|
|
||||||
root32[i] = 0x24
|
|
||||||
}
|
|
||||||
sig := make([]byte, 96)
|
|
||||||
for i := range sig {
|
|
||||||
sig[i] = 0x99
|
|
||||||
}
|
|
||||||
|
|
||||||
att := ð.Attestation{
|
|
||||||
AggregationBits: bitfield.Bitlist{0x01},
|
|
||||||
Data: ð.AttestationData{
|
|
||||||
Slot: 1,
|
|
||||||
CommitteeIndex: 1,
|
|
||||||
BeaconBlockRoot: root32,
|
|
||||||
Source: ð.Checkpoint{
|
|
||||||
Epoch: 1,
|
|
||||||
Root: root32,
|
|
||||||
},
|
|
||||||
Target: ð.Checkpoint{
|
|
||||||
Epoch: 1,
|
|
||||||
Root: root32,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
Signature: sig,
|
|
||||||
}
|
|
||||||
|
|
||||||
b := util.NewBeaconBlock()
|
|
||||||
b.Block.Slot = 123
|
|
||||||
b.Block.Body.RandaoReveal = randaoReveal
|
|
||||||
b.Block.Body.Attestations = []*eth.Attestation{att}
|
|
||||||
|
|
||||||
sb, err := blocks.NewSignedBeaconBlock(b)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
protoBlock, err := sb.Block().Proto()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
obj, ok := protoBlock.(query.SSZObject)
|
|
||||||
require.Equal(t, true, ok, "block proto does not implement query.SSZObject")
|
|
||||||
|
|
||||||
tests := []string{
|
|
||||||
".slot",
|
|
||||||
".body.randao_reveal",
|
|
||||||
".body.attestations[0].data.slot",
|
|
||||||
"len(body.attestations)",
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range tests {
|
|
||||||
t.Run(tc, func(t *testing.T) {
|
|
||||||
proveAndVerify(t, obj, tc)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProve_BeaconState(t *testing.T) {
|
|
||||||
st, _ := util.DeterministicGenesisState(t, 16)
|
|
||||||
require.NoError(t, st.SetSlot(primitives.Slot(42)))
|
|
||||||
|
|
||||||
sszObj, ok := st.ToProtoUnsafe().(query.SSZObject)
|
|
||||||
require.Equal(t, true, ok, "state proto does not implement query.SSZObject")
|
|
||||||
|
|
||||||
tests := []string{
|
|
||||||
".slot",
|
|
||||||
".latest_block_header",
|
|
||||||
".validators[0].effective_balance",
|
|
||||||
"len(validators)",
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range tests {
|
|
||||||
t.Run(tc, func(t *testing.T) {
|
|
||||||
proveAndVerify(t, sszObj, tc)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// proveAndVerify helper to analyze an object, generate a merkle proof for the given path,
|
|
||||||
// and verify the proof against the object's root.
|
|
||||||
func proveAndVerify(t *testing.T, obj query.SSZObject, pathStr string) {
|
|
||||||
t.Helper()
|
|
||||||
|
|
||||||
info, err := query.AnalyzeObject(obj)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
path, err := query.ParsePath(pathStr)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
gi, err := query.GetGeneralizedIndexFromPath(info, path)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
proof, err := info.Prove(gi)
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, int(gi), proof.Index)
|
|
||||||
|
|
||||||
root, err := obj.HashTreeRoot()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
ok, err := ssz.VerifyProof(root[:], proof)
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, true, ok, "merkle proof verification failed")
|
|
||||||
|
|
||||||
require.Equal(t, 32, len(proof.Leaf))
|
|
||||||
for i, h := range proof.Hashes {
|
|
||||||
require.Equal(t, 32, len(h), "proof hash %d is not 32 bytes", i)
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@@ -1,672 +0,0 @@
|
|||||||
package query
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/binary"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"math/bits"
|
|
||||||
"reflect"
|
|
||||||
"runtime"
|
|
||||||
"slices"
|
|
||||||
"sync"
|
|
||||||
|
|
||||||
"github.com/OffchainLabs/go-bitfield"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/container/trie"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/crypto/hash/htr"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/encoding/bytesutil"
|
|
||||||
ssz "github.com/OffchainLabs/prysm/v7/encoding/ssz"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/math"
|
|
||||||
fastssz "github.com/prysmaticlabs/fastssz"
|
|
||||||
)
|
|
||||||
|
|
||||||
// proofCollector collects sibling hashes and leaves needed for Merkle proofs.
|
|
||||||
//
|
|
||||||
// Multiproof-ready design:
|
|
||||||
// - requiredSiblings/requiredLeaves store which gindices we want to collect (registered before merkleization).
|
|
||||||
// - siblings/leaves store the actual collected hashes.
|
|
||||||
//
|
|
||||||
// Concurrency:
|
|
||||||
// - required* maps are read-only during merkleization.
|
|
||||||
// - siblings/leaves writes are protected by mutex.
|
|
||||||
type proofCollector struct {
|
|
||||||
sync.Mutex
|
|
||||||
|
|
||||||
// Required gindices (registered before merkleization)
|
|
||||||
requiredSiblings map[uint64]struct{}
|
|
||||||
requiredLeaves map[uint64]struct{}
|
|
||||||
|
|
||||||
// Collected hashes
|
|
||||||
siblings map[uint64][32]byte
|
|
||||||
leaves map[uint64][32]byte
|
|
||||||
}
|
|
||||||
|
|
||||||
func newProofCollector() *proofCollector {
|
|
||||||
return &proofCollector{
|
|
||||||
requiredSiblings: make(map[uint64]struct{}),
|
|
||||||
requiredLeaves: make(map[uint64]struct{}),
|
|
||||||
siblings: make(map[uint64][32]byte),
|
|
||||||
leaves: make(map[uint64][32]byte),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (pc *proofCollector) reset() {
|
|
||||||
pc.Lock()
|
|
||||||
defer pc.Unlock()
|
|
||||||
|
|
||||||
pc.requiredSiblings = make(map[uint64]struct{})
|
|
||||||
pc.requiredLeaves = make(map[uint64]struct{})
|
|
||||||
pc.siblings = make(map[uint64][32]byte)
|
|
||||||
pc.leaves = make(map[uint64][32]byte)
|
|
||||||
}
|
|
||||||
|
|
||||||
// addTarget register the target leaf and its required sibling nodes for proof construction.
|
|
||||||
// Registration should happen before merkleization begins.
|
|
||||||
func (pc *proofCollector) addTarget(gindex uint64) {
|
|
||||||
pc.Lock()
|
|
||||||
defer pc.Unlock()
|
|
||||||
|
|
||||||
pc.requiredLeaves[gindex] = struct{}{}
|
|
||||||
|
|
||||||
// Walk from the target leaf up to (but not including) the root (gindex=1).
|
|
||||||
// At each step, register the sibling node required to prove inclusion.
|
|
||||||
nodeGindex := gindex
|
|
||||||
for nodeGindex > 1 {
|
|
||||||
siblingGindex := nodeGindex ^ 1 // flip the last bit: left<->right sibling
|
|
||||||
pc.requiredSiblings[siblingGindex] = struct{}{}
|
|
||||||
|
|
||||||
// Move to parent
|
|
||||||
nodeGindex /= 2
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// toProof converts the collected siblings and leaves into a fastssz.Proof structure.
|
|
||||||
// Current behavior expects a single target leaf (single proof).
|
|
||||||
func (pc *proofCollector) toProof() (*fastssz.Proof, error) {
|
|
||||||
pc.Lock()
|
|
||||||
defer pc.Unlock()
|
|
||||||
|
|
||||||
proof := &fastssz.Proof{}
|
|
||||||
if len(pc.leaves) == 0 {
|
|
||||||
return nil, errors.New("no leaves collected: add target leaves before merkleization")
|
|
||||||
}
|
|
||||||
|
|
||||||
leafGindices := make([]uint64, 0, len(pc.leaves))
|
|
||||||
for g := range pc.leaves {
|
|
||||||
leafGindices = append(leafGindices, g)
|
|
||||||
}
|
|
||||||
slices.Sort(leafGindices)
|
|
||||||
|
|
||||||
// single proof resides in leafGindices[0]
|
|
||||||
targetGindex := leafGindices[0]
|
|
||||||
proofIndex, err := math.Int(targetGindex)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("gindex %d overflows int: %w", targetGindex, err)
|
|
||||||
}
|
|
||||||
proof.Index = proofIndex
|
|
||||||
|
|
||||||
// store the leaf
|
|
||||||
leaf := pc.leaves[targetGindex]
|
|
||||||
leafBuf := make([]byte, 32)
|
|
||||||
copy(leafBuf, leaf[:])
|
|
||||||
proof.Leaf = leafBuf
|
|
||||||
|
|
||||||
// Walk from target up to root, collecting siblings.
|
|
||||||
steps := bits.Len64(targetGindex) - 1
|
|
||||||
proof.Hashes = make([][]byte, 0, steps)
|
|
||||||
|
|
||||||
for targetGindex > 1 {
|
|
||||||
sib := targetGindex ^ 1
|
|
||||||
h, ok := pc.siblings[sib]
|
|
||||||
if !ok {
|
|
||||||
return nil, fmt.Errorf("missing sibling hash for gindex %d", sib)
|
|
||||||
}
|
|
||||||
proof.Hashes = append(proof.Hashes, h[:])
|
|
||||||
targetGindex /= 2
|
|
||||||
}
|
|
||||||
|
|
||||||
return proof, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// collectLeaf checks if the given gindex is a required leaf for the proof,
|
|
||||||
// and if so, stores the provided leaf hash in the collector.
|
|
||||||
func (pc *proofCollector) collectLeaf(gindex uint64, leaf [32]byte) {
|
|
||||||
if _, ok := pc.requiredLeaves[gindex]; !ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
pc.Lock()
|
|
||||||
pc.leaves[gindex] = leaf
|
|
||||||
pc.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
// collectSibling stores the hash for a sibling node identified by gindex.
|
|
||||||
// It only stores the hash if gindex was pre-registered via addTarget (present in requiredSiblings).
|
|
||||||
// Writes to the collected siblings map are protected by the collector mutex.
|
|
||||||
func (pc *proofCollector) collectSibling(gindex uint64, hash [32]byte) {
|
|
||||||
if _, ok := pc.requiredSiblings[gindex]; !ok {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
pc.Lock()
|
|
||||||
pc.siblings[gindex] = hash
|
|
||||||
pc.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Merkleizers and proof collection methods
|
|
||||||
|
|
||||||
// merkleize recursively traverses an SSZ info and computes the Merkle root of the subtree.
|
|
||||||
//
|
|
||||||
// Proof collection:
|
|
||||||
// - During traversal it calls collectLeaf/collectSibling with the SSZ generalized indices (gindices)
|
|
||||||
// of visited nodes.
|
|
||||||
// - The collector only stores hashes for gindices that were pre-registered via addTarget
|
|
||||||
// (requiredLeaves/requiredSiblings). This makes the traversal multiproof-ready: you can register
|
|
||||||
// multiple targets before calling merkleize.
|
|
||||||
//
|
|
||||||
// SSZ types handled: basic types, containers, lists, vectors, bitlists, and bitvectors.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - info: SSZ type metadata for the current value.
|
|
||||||
// - v: reflect.Value of the current value.
|
|
||||||
// - currentGindex: generalized index of the current subtree root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the current subtree.
|
|
||||||
// - error: any error encountered during traversal/merkleization.
|
|
||||||
func (pc *proofCollector) merkleize(info *SszInfo, v reflect.Value, currentGindex uint64) ([32]byte, error) {
|
|
||||||
if info.sszType.isBasic() {
|
|
||||||
return pc.merkleizeBasicType(info.sszType, v, currentGindex)
|
|
||||||
}
|
|
||||||
switch info.sszType {
|
|
||||||
case Container:
|
|
||||||
return pc.merkleizeContainer(info, v, currentGindex)
|
|
||||||
case List:
|
|
||||||
return pc.merkleizeList(info, v, currentGindex)
|
|
||||||
case Vector:
|
|
||||||
return pc.merkleizeVector(info, v, currentGindex)
|
|
||||||
case Bitlist:
|
|
||||||
return pc.merkleizeBitlist(info, v, currentGindex)
|
|
||||||
case Bitvector:
|
|
||||||
return pc.merkleizeBitvector(info, v, currentGindex)
|
|
||||||
default:
|
|
||||||
return [32]byte{}, fmt.Errorf("unsupported SSZ type: %v", info.sszType)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeBasicType serializes a basic SSZ value into a 32-byte leaf chunk (little-endian, zero-padded).
|
|
||||||
//
|
|
||||||
// Proof collection:
|
|
||||||
// - It calls collectLeaf(currentGindex, leaf) and stores the leaf if currentGindex was pre-registered via addTarget.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - t: the SSZType (basic).
|
|
||||||
// - v: the reflect.Value of the basic value.
|
|
||||||
// - currentGindex: the generalized index (gindex) of this leaf.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: the 32-byte SSZ leaf chunk.
|
|
||||||
// - error: if the SSZType is not a supported basic type.
|
|
||||||
func (pc *proofCollector) merkleizeBasicType(t SSZType, v reflect.Value, currentGindex uint64) ([32]byte, error) {
|
|
||||||
var leaf [32]byte
|
|
||||||
|
|
||||||
// Serialize the value into a 32-byte chunk (little-endian, zero-padded)
|
|
||||||
switch t {
|
|
||||||
case Uint8:
|
|
||||||
leaf[0] = uint8(v.Uint())
|
|
||||||
case Uint16:
|
|
||||||
binary.LittleEndian.PutUint16(leaf[:2], uint16(v.Uint()))
|
|
||||||
case Uint32:
|
|
||||||
binary.LittleEndian.PutUint32(leaf[:4], uint32(v.Uint()))
|
|
||||||
case Uint64:
|
|
||||||
binary.LittleEndian.PutUint64(leaf[:8], v.Uint())
|
|
||||||
case Boolean:
|
|
||||||
if v.Bool() {
|
|
||||||
leaf[0] = 1
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
return [32]byte{}, fmt.Errorf("unexpected basic type: %v", t)
|
|
||||||
}
|
|
||||||
|
|
||||||
pc.collectLeaf(currentGindex, leaf)
|
|
||||||
|
|
||||||
return leaf, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeContainer computes the Merkle root of an SSZ container by:
|
|
||||||
// 1. Merkleizing each field into a 32-byte subtree root
|
|
||||||
// 2. Merkleizing the field roots into the container root (padding to the next power-of-2)
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): depth = ssz.Depth(uint64(N)) and field i has gindex = (currentGindex << depth) + uint64(i).
|
|
||||||
// Proof collection: merkleize() computes each field root, merkleizeVectorAndCollect collects required siblings, and collectLeaf stores the container root if registered.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - info: SSZ type metadata for the container.
|
|
||||||
// - v: reflect.Value of the container value.
|
|
||||||
// - currentGindex: generalized index (gindex) of the container root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the container.
|
|
||||||
// - error: any error encountered while merkleizing fields.
|
|
||||||
func (pc *proofCollector) merkleizeContainer(info *SszInfo, v reflect.Value, currentGindex uint64) ([32]byte, error) {
|
|
||||||
// If the container root itself is the target, compute directly and return early.
|
|
||||||
// This avoids full subtree merkleization when we only need the root.
|
|
||||||
if _, ok := pc.requiredLeaves[currentGindex]; ok {
|
|
||||||
root, err := info.HashTreeRoot()
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
pc.collectLeaf(currentGindex, root)
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
ci, err := info.ContainerInfo()
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
v = dereferencePointer(v)
|
|
||||||
|
|
||||||
// Calculate depth: how many levels from container root to field leaves
|
|
||||||
numFields := len(ci.order)
|
|
||||||
depth := ssz.Depth(uint64(numFields))
|
|
||||||
|
|
||||||
// Step 1: Compute HTR for each subtree (field)
|
|
||||||
fieldRoots := make([][32]byte, numFields)
|
|
||||||
|
|
||||||
for i, name := range ci.order {
|
|
||||||
fieldInfo := ci.fields[name]
|
|
||||||
fieldVal := v.FieldByName(fieldInfo.goFieldName)
|
|
||||||
|
|
||||||
// Field i's gindex: shift currentGindex left by depth, then OR with field index
|
|
||||||
fieldGindex := currentGindex<<depth + uint64(i)
|
|
||||||
|
|
||||||
htr, err := pc.merkleize(fieldInfo.sszInfo, fieldVal, fieldGindex)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, fmt.Errorf("field %s: %w", name, err)
|
|
||||||
}
|
|
||||||
fieldRoots[i] = htr
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 2: Merkleize the field hashes into the container root,
|
|
||||||
// collecting sibling hashes if target is within this subtree
|
|
||||||
root := pc.merkleizeVectorAndCollect(fieldRoots, currentGindex, uint64(depth))
|
|
||||||
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeVectorBody computes the Merkle root of the "data" subtree for vector-like SSZ types
|
|
||||||
// (vectors and the data-part of lists/bitlists).
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): depth = ssz.Depth(limit); leafBase = subtreeRootGindex << depth; element/chunk i gindex = leafBase + uint64(i).
|
|
||||||
// Proof collection: merkleize() is called for composite elements; merkleizeVectorAndCollect collects required siblings at this layer.
|
|
||||||
// Padding: merkleizeVectorAndCollect uses trie.ZeroHashes as needed.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - elemInfo: SSZ type metadata for the element.
|
|
||||||
// - v: reflect.Value of the vector/list data.
|
|
||||||
// - length: number of actual elements present.
|
|
||||||
// - limit: virtual leaf capacity used for padding/Depth (fixed length for vectors, limit for lists).
|
|
||||||
// - subtreeRootGindex: gindex of the data subtree root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the data subtree.
|
|
||||||
// - error: any error encountered while merkleizing composite elements.
|
|
||||||
func (pc *proofCollector) merkleizeVectorBody(elemInfo *SszInfo, v reflect.Value, length int, limit uint64, subtreeRootGindex uint64) ([32]byte, error) {
|
|
||||||
depth := uint64(ssz.Depth(limit))
|
|
||||||
|
|
||||||
var chunks [][32]byte
|
|
||||||
if elemInfo.sszType.isBasic() {
|
|
||||||
// Serialize basic elements and pack into 32-byte chunks using ssz.PackByChunk.
|
|
||||||
elemSize, err := math.Int(itemLength(elemInfo))
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, fmt.Errorf("element size %d overflows int: %w", itemLength(elemInfo), err)
|
|
||||||
}
|
|
||||||
serialized := make([][]byte, length)
|
|
||||||
// Single contiguous allocation for all element data
|
|
||||||
allData := make([]byte, length*elemSize)
|
|
||||||
for i := range length {
|
|
||||||
buf := allData[i*elemSize : (i+1)*elemSize]
|
|
||||||
elem := v.Index(i)
|
|
||||||
if elemInfo.sszType == Boolean && elem.Bool() {
|
|
||||||
buf[0] = 1
|
|
||||||
} else {
|
|
||||||
bytesutil.PutLittleEndian(buf, elem.Uint(), elemSize)
|
|
||||||
}
|
|
||||||
serialized[i] = buf
|
|
||||||
}
|
|
||||||
chunks, err = ssz.PackByChunk(serialized)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// Composite elements: compute each element root (no padding here; merkleizeVectorAndCollect pads).
|
|
||||||
chunks = make([][32]byte, length)
|
|
||||||
|
|
||||||
// Fall back to per-element merkleization with proper gindices for proof collection.
|
|
||||||
// Parallel execution
|
|
||||||
workerCount := min(runtime.GOMAXPROCS(0), length)
|
|
||||||
|
|
||||||
jobs := make(chan int, workerCount*16)
|
|
||||||
errCh := make(chan error, 1) // only need the first error
|
|
||||||
stopCh := make(chan struct{})
|
|
||||||
var stopOnce sync.Once
|
|
||||||
var wg sync.WaitGroup
|
|
||||||
|
|
||||||
worker := func() {
|
|
||||||
defer wg.Done()
|
|
||||||
for idx := range jobs {
|
|
||||||
select {
|
|
||||||
case <-stopCh:
|
|
||||||
return
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
|
|
||||||
elemGindex := subtreeRootGindex<<depth + uint64(idx)
|
|
||||||
htr, err := pc.merkleize(elemInfo, v.Index(idx), elemGindex)
|
|
||||||
if err != nil {
|
|
||||||
stopOnce.Do(func() { close(stopCh) })
|
|
||||||
select {
|
|
||||||
case errCh <- fmt.Errorf("index %d: %w", idx, err):
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
chunks[idx] = htr
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
wg.Add(workerCount)
|
|
||||||
for range workerCount {
|
|
||||||
go worker()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Enqueue jobs; stop early if any worker reports an error.
|
|
||||||
enqueue:
|
|
||||||
for i := range length {
|
|
||||||
select {
|
|
||||||
case <-stopCh:
|
|
||||||
break enqueue
|
|
||||||
case jobs <- i:
|
|
||||||
}
|
|
||||||
}
|
|
||||||
close(jobs)
|
|
||||||
|
|
||||||
wg.Wait()
|
|
||||||
|
|
||||||
select {
|
|
||||||
case err := <-errCh:
|
|
||||||
return [32]byte{}, err
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
root := pc.merkleizeVectorAndCollect(chunks, subtreeRootGindex, depth)
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeVector computes the Merkle root of an SSZ vector (fixed-length).
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): currentGindex is the gindex of the vector root; element/chunk gindices are derived
|
|
||||||
// inside merkleizeVectorBody using leafBase = currentGindex << ssz.Depth(leaves).
|
|
||||||
//
|
|
||||||
// Proof collection: merkleizeVectorBody performs element/chunk merkleization and collects required siblings at the
|
|
||||||
// vector layer; collectLeaf stores the vector root if currentGindex was registered via addTarget.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - info: SSZ type metadata for the vector.
|
|
||||||
// - v: reflect.Value of the vector value.
|
|
||||||
// - currentGindex: generalized index (gindex) of the vector root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the vector.
|
|
||||||
// - error: any error encountered while merkleizing composite elements.
|
|
||||||
func (pc *proofCollector) merkleizeVector(info *SszInfo, v reflect.Value, currentGindex uint64) ([32]byte, error) {
|
|
||||||
vi, err := info.VectorInfo()
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
length, err := math.Int(vi.Length())
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, fmt.Errorf("vector length %d overflows int: %w", vi.Length(), err)
|
|
||||||
}
|
|
||||||
elemInfo := vi.element
|
|
||||||
|
|
||||||
// Determine the virtual leaf capacity for the vector.
|
|
||||||
leaves, err := getChunkCount(info)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
root, err := pc.merkleizeVectorBody(elemInfo, v, length, leaves, currentGindex)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// If the vector root itself is the target
|
|
||||||
pc.collectLeaf(currentGindex, root)
|
|
||||||
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeList computes the Merkle root of an SSZ list by merkleizing its data subtree and mixing in the length.
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): dataRoot is the left child of the list root (dataRootGindex = currentGindex*2); the length mixin is the right child (currentGindex*2+1).
|
|
||||||
// Proof collection: merkleizeVectorBody computes the data root (collecting required siblings in the data subtree), and mixinLengthAndCollect collects required siblings at the length-mixin level; collectLeaf stores the list root if registered.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - info: SSZ type metadata for the list.
|
|
||||||
// - v: reflect.Value of the list value.
|
|
||||||
// - currentGindex: generalized index (gindex) of the list root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the list.
|
|
||||||
// - error: any error encountered while merkleizing the data subtree.
|
|
||||||
func (pc *proofCollector) merkleizeList(info *SszInfo, v reflect.Value, currentGindex uint64) ([32]byte, error) {
|
|
||||||
li, err := info.ListInfo()
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
length := v.Len()
|
|
||||||
elemInfo := li.element
|
|
||||||
|
|
||||||
chunks := make([][32]byte, 2)
|
|
||||||
// Compute the length hash (little-endian uint256)
|
|
||||||
binary.LittleEndian.PutUint64(chunks[1][:8], uint64(length))
|
|
||||||
|
|
||||||
// Data subtree root is the left child of the list root.
|
|
||||||
dataRootGindex := currentGindex * 2
|
|
||||||
|
|
||||||
// Compute virtual leaf capacity for the data subtree.
|
|
||||||
leaves, err := getChunkCount(info)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
chunks[0], err = pc.merkleizeVectorBody(elemInfo, v, length, leaves, dataRootGindex)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle the length mixin level (and proof bookkeeping at this level).
|
|
||||||
// Compute the final list root: hash(dataRoot || lengthHash)
|
|
||||||
root := pc.mixinLengthAndCollect(currentGindex, chunks)
|
|
||||||
|
|
||||||
// If the list root itself is the target
|
|
||||||
pc.collectLeaf(currentGindex, root)
|
|
||||||
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeBitvectorBody computes the Merkle root of a bitvector-like byte sequence by packing it into 32-byte chunks
|
|
||||||
// and merkleizing those chunks as a fixed-capacity vector (padding with trie.ZeroHashes as needed).
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): depth = ssz.Depth(chunkLimit); leafBase = subtreeRootGindex << depth; chunk i uses gindex = leafBase + uint64(i).
|
|
||||||
// Proof collection: merkleizeVectorAndCollect collects required sibling hashes at the chunk-merkleization layer.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - data: raw byte sequence representing the bitvector payload.
|
|
||||||
// - chunkLimit: fixed/limit number of 32-byte chunks (used for padding/Depth).
|
|
||||||
// - subtreeRootGindex: gindex of the bitvector data subtree root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the bitvector data subtree.
|
|
||||||
// - error: any error encountered while packing data into chunks.
|
|
||||||
func (pc *proofCollector) merkleizeBitvectorBody(data []byte, chunkLimit uint64, subtreeRootGindex uint64) ([32]byte, error) {
|
|
||||||
depth := ssz.Depth(chunkLimit)
|
|
||||||
chunks, err := ssz.PackByChunk([][]byte{data})
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
root := pc.merkleizeVectorAndCollect(chunks, subtreeRootGindex, uint64(depth))
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeBitvector computes the Merkle root of a fixed-length SSZ bitvector and collects proof nodes for targets.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - info: SSZ type metadata for the bitvector.
|
|
||||||
// - v: reflect.Value of the bitvector value.
|
|
||||||
// - currentGindex: generalized index (gindex) of the bitvector root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the bitvector.
|
|
||||||
// - error: any error encountered during packing or merkleization.
|
|
||||||
func (pc *proofCollector) merkleizeBitvector(info *SszInfo, v reflect.Value, currentGindex uint64) ([32]byte, error) {
|
|
||||||
bitvectorBytes := v.Bytes()
|
|
||||||
if len(bitvectorBytes) == 0 {
|
|
||||||
return [32]byte{}, fmt.Errorf("bitvector field is uninitialized (nil or empty slice)")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Compute virtual leaf capacity for the bitvector.
|
|
||||||
numChunks, err := getChunkCount(info)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
root, err := pc.merkleizeBitvectorBody(bitvectorBytes, numChunks, currentGindex)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
pc.collectLeaf(currentGindex, root)
|
|
||||||
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeBitlist computes the Merkle root of an SSZ bitlist by merkleizing its data chunks and mixing in the bit length.
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): dataRoot is the left child (dataRootGindex = currentGindex*2) and the length mixin is the right child (currentGindex*2+1).
|
|
||||||
// Proof collection: merkleizeBitvectorBody computes the data root (collecting required siblings under dataRootGindex), and mixinLengthAndCollect collects required siblings at the length-mixin level; collectLeaf stores the bitlist root if registered.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - info: SSZ type metadata for the bitlist.
|
|
||||||
// - v: reflect.Value of the bitlist value.
|
|
||||||
// - currentGindex: generalized index (gindex) of the bitlist root.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the bitlist.
|
|
||||||
// - error: any error encountered while merkleizing the data subtree.
|
|
||||||
func (pc *proofCollector) merkleizeBitlist(info *SszInfo, v reflect.Value, currentGindex uint64) ([32]byte, error) {
|
|
||||||
bi, err := info.BitlistInfo()
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
bitlistBytes := v.Bytes()
|
|
||||||
|
|
||||||
// Use go-bitfield to get bytes with termination bit cleared
|
|
||||||
bl := bitfield.Bitlist(bitlistBytes)
|
|
||||||
data := bl.BytesNoTrim()
|
|
||||||
|
|
||||||
// Get the bit length from bitlistInfo
|
|
||||||
bitLength := bi.Length()
|
|
||||||
|
|
||||||
// Get the chunk limit from getChunkCount
|
|
||||||
limitChunks, err := getChunkCount(info)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
chunks := make([][32]byte, 2)
|
|
||||||
// Compute the length hash (little-endian uint256)
|
|
||||||
binary.LittleEndian.PutUint64(chunks[1][:8], uint64(bitLength))
|
|
||||||
|
|
||||||
dataRootGindex := currentGindex * 2
|
|
||||||
chunks[0], err = pc.merkleizeBitvectorBody(data, limitChunks, dataRootGindex)
|
|
||||||
if err != nil {
|
|
||||||
return [32]byte{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handle the length mixin level (and proof bookkeeping at this level).
|
|
||||||
root := pc.mixinLengthAndCollect(currentGindex, chunks)
|
|
||||||
|
|
||||||
pc.collectLeaf(currentGindex, root)
|
|
||||||
|
|
||||||
return root, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// merkleizeVectorAndCollect merkleizes a slice of 32-byte leaf nodes into a subtree root, padding to a virtual size of 2^depth.
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): at layer i (0-based), nodes have gindices levelBase = subtreeGeneralizedIndex << (depth-i) and node gindex = levelBase + idx.
|
|
||||||
// Proof collection: for each layer it calls collectSibling(nodeGindex, nodeHash) and stores only those gindices registered via addTarget.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - elements: leaf-level hashes (may be shorter than 2^depth; padding is applied with trie.ZeroHashes).
|
|
||||||
// - subtreeGeneralizedIndex: gindex of the subtree root.
|
|
||||||
// - depth: number of merkleization layers from subtree root to leaves.
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: Merkle root of the subtree.
|
|
||||||
func (pc *proofCollector) merkleizeVectorAndCollect(elements [][32]byte, subtreeGeneralizedIndex uint64, depth uint64) [32]byte {
|
|
||||||
// Return zerohash at depth
|
|
||||||
if len(elements) == 0 {
|
|
||||||
return trie.ZeroHashes[depth]
|
|
||||||
}
|
|
||||||
for i := range depth {
|
|
||||||
layerLen := len(elements)
|
|
||||||
oddNodeLength := layerLen%2 == 1
|
|
||||||
if oddNodeLength {
|
|
||||||
zerohash := trie.ZeroHashes[i]
|
|
||||||
elements = append(elements, zerohash)
|
|
||||||
}
|
|
||||||
|
|
||||||
levelBaseGindex := subtreeGeneralizedIndex << (depth - i)
|
|
||||||
for idx := range elements {
|
|
||||||
gindex := levelBaseGindex + uint64(idx)
|
|
||||||
pc.collectSibling(gindex, elements[idx])
|
|
||||||
pc.collectLeaf(gindex, elements[idx])
|
|
||||||
}
|
|
||||||
|
|
||||||
elements = htr.VectorizedSha256(elements)
|
|
||||||
}
|
|
||||||
return elements[0]
|
|
||||||
}
|
|
||||||
|
|
||||||
// mixinLengthAndCollect computes the final mix-in root for list/bitlist values:
|
|
||||||
//
|
|
||||||
// root = hash(dataRoot, lengthHash)
|
|
||||||
//
|
|
||||||
// where chunks[0] is dataRoot and chunks[1] is the 32-byte length hash.
|
|
||||||
//
|
|
||||||
// Generalized indices (gindices): dataRoot is the left child (dataRootGindex = currentGindex*2) and lengthHash is the right child (lengthHashGindex = currentGindex*2+1).
|
|
||||||
// Proof collection: it calls collectSibling/collectLeaf for both child gindices; the collector stores them only if they were registered via addTarget.
|
|
||||||
//
|
|
||||||
// Parameters:
|
|
||||||
// - currentGindex: gindex of the parent node (list/bitlist root).
|
|
||||||
// - chunks: two 32-byte nodes: [dataRoot, lengthHash].
|
|
||||||
//
|
|
||||||
// Returns:
|
|
||||||
// - [32]byte: mixed-in Merkle root (or zero value on hashing error).
|
|
||||||
// - error: any error encountered during hashing.
|
|
||||||
func (pc *proofCollector) mixinLengthAndCollect(currentGindex uint64, chunks [][32]byte) [32]byte {
|
|
||||||
dataRoot, lengthHash := chunks[0], chunks[1]
|
|
||||||
dataRootGindex, lengthHashGindex := currentGindex*2, currentGindex*2+1
|
|
||||||
|
|
||||||
pc.collectSibling(dataRootGindex, dataRoot)
|
|
||||||
pc.collectSibling(lengthHashGindex, lengthHash)
|
|
||||||
|
|
||||||
pc.collectLeaf(dataRootGindex, dataRoot)
|
|
||||||
pc.collectLeaf(lengthHashGindex, lengthHash)
|
|
||||||
|
|
||||||
return ssz.MixInLength(dataRoot, lengthHash[:])
|
|
||||||
}
|
|
||||||
@@ -1,531 +0,0 @@
|
|||||||
package query
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/sha256"
|
|
||||||
"encoding/binary"
|
|
||||||
"reflect"
|
|
||||||
"slices"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/OffchainLabs/go-bitfield"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/beacon-chain/state/stateutil"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/consensus-types/primitives"
|
|
||||||
ssz "github.com/OffchainLabs/prysm/v7/encoding/ssz"
|
|
||||||
ethpb "github.com/OffchainLabs/prysm/v7/proto/prysm/v1alpha1"
|
|
||||||
sszquerypb "github.com/OffchainLabs/prysm/v7/proto/ssz_query/testing"
|
|
||||||
"github.com/OffchainLabs/prysm/v7/testing/require"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestProofCollector_New(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
|
|
||||||
require.NotNil(t, pc)
|
|
||||||
require.Equal(t, 0, len(pc.requiredSiblings))
|
|
||||||
require.Equal(t, 0, len(pc.requiredLeaves))
|
|
||||||
require.Equal(t, 0, len(pc.siblings))
|
|
||||||
require.Equal(t, 0, len(pc.leaves))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_Reset(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
pc.requiredSiblings[3] = struct{}{}
|
|
||||||
pc.requiredLeaves[5] = struct{}{}
|
|
||||||
pc.siblings[3] = [32]byte{1}
|
|
||||||
pc.leaves[5] = [32]byte{2}
|
|
||||||
|
|
||||||
pc.reset()
|
|
||||||
|
|
||||||
require.Equal(t, 0, len(pc.requiredSiblings))
|
|
||||||
require.Equal(t, 0, len(pc.requiredLeaves))
|
|
||||||
require.Equal(t, 0, len(pc.siblings))
|
|
||||||
require.Equal(t, 0, len(pc.leaves))
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_AddTarget(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
pc.addTarget(5)
|
|
||||||
|
|
||||||
_, hasLeaf := pc.requiredLeaves[5]
|
|
||||||
_, hasSibling4 := pc.requiredSiblings[4]
|
|
||||||
_, hasSibling3 := pc.requiredSiblings[3]
|
|
||||||
_, hasSibling1 := pc.requiredSiblings[1] // GI 1 is the root
|
|
||||||
|
|
||||||
require.Equal(t, true, hasLeaf)
|
|
||||||
require.Equal(t, true, hasSibling4)
|
|
||||||
require.Equal(t, true, hasSibling3)
|
|
||||||
require.Equal(t, false, hasSibling1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_ToProof(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
pc.addTarget(5)
|
|
||||||
|
|
||||||
leaf := [32]byte{9}
|
|
||||||
sibling4 := [32]byte{4}
|
|
||||||
sibling3 := [32]byte{3}
|
|
||||||
|
|
||||||
pc.collectLeaf(5, leaf)
|
|
||||||
pc.collectSibling(4, sibling4)
|
|
||||||
pc.collectSibling(3, sibling3)
|
|
||||||
|
|
||||||
proof, err := pc.toProof()
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
require.Equal(t, 5, proof.Index)
|
|
||||||
require.DeepEqual(t, leaf[:], proof.Leaf)
|
|
||||||
require.Equal(t, 2, len(proof.Hashes))
|
|
||||||
require.DeepEqual(t, sibling4[:], proof.Hashes[0])
|
|
||||||
require.DeepEqual(t, sibling3[:], proof.Hashes[1])
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_ToProof_NoLeaves(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
_, err := pc.toProof()
|
|
||||||
require.NotNil(t, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_CollectLeaf(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
leaf := [32]byte{7}
|
|
||||||
|
|
||||||
pc.collectLeaf(10, leaf)
|
|
||||||
require.Equal(t, 0, len(pc.leaves))
|
|
||||||
|
|
||||||
pc.addTarget(10)
|
|
||||||
pc.collectLeaf(10, leaf)
|
|
||||||
stored, ok := pc.leaves[10]
|
|
||||||
require.Equal(t, true, ok)
|
|
||||||
require.Equal(t, leaf, stored)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_CollectSibling(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
hash := [32]byte{5}
|
|
||||||
|
|
||||||
pc.collectSibling(4, hash)
|
|
||||||
require.Equal(t, 0, len(pc.siblings))
|
|
||||||
|
|
||||||
pc.addTarget(5)
|
|
||||||
pc.collectSibling(4, hash)
|
|
||||||
stored, ok := pc.siblings[4]
|
|
||||||
require.Equal(t, true, ok)
|
|
||||||
require.Equal(t, hash, stored)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_Merkleize_BasicTypes(t *testing.T) {
|
|
||||||
testCases := []struct {
|
|
||||||
name string
|
|
||||||
sszType SSZType
|
|
||||||
value any
|
|
||||||
expected [32]byte
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
name: "uint8",
|
|
||||||
sszType: Uint8,
|
|
||||||
value: uint8(0x11),
|
|
||||||
expected: func() [32]byte {
|
|
||||||
var leaf [32]byte
|
|
||||||
leaf[0] = 0x11
|
|
||||||
return leaf
|
|
||||||
}(),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "uint16",
|
|
||||||
sszType: Uint16,
|
|
||||||
value: uint16(0x2211),
|
|
||||||
expected: func() [32]byte {
|
|
||||||
var leaf [32]byte
|
|
||||||
binary.LittleEndian.PutUint16(leaf[:2], 0x2211)
|
|
||||||
return leaf
|
|
||||||
}(),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "uint32",
|
|
||||||
sszType: Uint32,
|
|
||||||
value: uint32(0x44332211),
|
|
||||||
expected: func() [32]byte {
|
|
||||||
var leaf [32]byte
|
|
||||||
binary.LittleEndian.PutUint32(leaf[:4], 0x44332211)
|
|
||||||
return leaf
|
|
||||||
}(),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "uint64",
|
|
||||||
sszType: Uint64,
|
|
||||||
value: uint64(0x8877665544332211),
|
|
||||||
expected: func() [32]byte {
|
|
||||||
var leaf [32]byte
|
|
||||||
binary.LittleEndian.PutUint64(leaf[:8], 0x8877665544332211)
|
|
||||||
return leaf
|
|
||||||
}(),
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "bool",
|
|
||||||
sszType: Boolean,
|
|
||||||
value: true,
|
|
||||||
expected: func() [32]byte {
|
|
||||||
var leaf [32]byte
|
|
||||||
leaf[0] = 1
|
|
||||||
return leaf
|
|
||||||
}(),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range testCases {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
gindex := uint64(3)
|
|
||||||
pc.addTarget(gindex)
|
|
||||||
|
|
||||||
leaf, err := pc.merkleizeBasicType(tc.sszType, reflect.ValueOf(tc.value), gindex)
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, tc.expected, leaf)
|
|
||||||
|
|
||||||
stored, ok := pc.leaves[gindex]
|
|
||||||
require.Equal(t, true, ok)
|
|
||||||
require.Equal(t, tc.expected, stored)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_Merkleize_Container(t *testing.T) {
|
|
||||||
container := makeFixedTestContainer()
|
|
||||||
|
|
||||||
info, err := AnalyzeObject(container)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
pc := newProofCollector()
|
|
||||||
pc.addTarget(1)
|
|
||||||
|
|
||||||
root, err := pc.merkleize(info, reflect.ValueOf(container), 1)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
expected, err := container.HashTreeRoot()
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
|
|
||||||
stored, ok := pc.leaves[1]
|
|
||||||
require.Equal(t, true, ok)
|
|
||||||
require.Equal(t, expected, stored)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_Merkleize_Vector(t *testing.T) {
|
|
||||||
container := makeFixedTestContainer()
|
|
||||||
info, err := AnalyzeObject(container)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
ci, err := info.ContainerInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
field := ci.fields["vector_field"]
|
|
||||||
|
|
||||||
pc := newProofCollector()
|
|
||||||
root, err := pc.merkleizeVector(field.sszInfo, reflect.ValueOf(container.VectorField), 1)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
serialized := make([][]byte, len(container.VectorField))
|
|
||||||
for i, v := range container.VectorField {
|
|
||||||
buf := make([]byte, 8)
|
|
||||||
binary.LittleEndian.PutUint64(buf, v)
|
|
||||||
serialized[i] = buf
|
|
||||||
}
|
|
||||||
chunks, err := ssz.PackByChunk(serialized)
|
|
||||||
require.NoError(t, err)
|
|
||||||
limit, err := getChunkCount(field.sszInfo)
|
|
||||||
require.NoError(t, err)
|
|
||||||
expected := ssz.MerkleizeVector(chunks, limit)
|
|
||||||
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_Merkleize_List(t *testing.T) {
|
|
||||||
list := []*sszquerypb.FixedNestedContainer{
|
|
||||||
makeFixedNestedContainer(1),
|
|
||||||
makeFixedNestedContainer(2),
|
|
||||||
}
|
|
||||||
container := makeVariableTestContainer(list, bitfield.NewBitlist(1))
|
|
||||||
info, err := AnalyzeObject(container)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
ci, err := info.ContainerInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
field := ci.fields["field_list_container"]
|
|
||||||
|
|
||||||
pc := newProofCollector()
|
|
||||||
root, err := pc.merkleizeList(field.sszInfo, reflect.ValueOf(list), 1)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
listInfo, err := field.sszInfo.ListInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
expected, err := ssz.MerkleizeListSSZ(list, listInfo.Limit())
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_Merkleize_Bitvector(t *testing.T) {
|
|
||||||
container := makeFixedTestContainer()
|
|
||||||
info, err := AnalyzeObject(container)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
ci, err := info.ContainerInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
field := ci.fields["bitvector64_field"]
|
|
||||||
|
|
||||||
pc := newProofCollector()
|
|
||||||
root, err := pc.merkleizeBitvector(field.sszInfo, reflect.ValueOf(container.Bitvector64Field), 1)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
expected, err := ssz.MerkleizeByteSliceSSZ([]byte(container.Bitvector64Field))
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_Merkleize_Bitlist(t *testing.T) {
|
|
||||||
bitlist := bitfield.NewBitlist(16)
|
|
||||||
bitlist.SetBitAt(3, true)
|
|
||||||
bitlist.SetBitAt(8, true)
|
|
||||||
|
|
||||||
container := makeVariableTestContainer(nil, bitlist)
|
|
||||||
info, err := AnalyzeObject(container)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
ci, err := info.ContainerInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
field := ci.fields["bitlist_field"]
|
|
||||||
|
|
||||||
pc := newProofCollector()
|
|
||||||
root, err := pc.merkleizeBitlist(field.sszInfo, reflect.ValueOf(container.BitlistField), 1)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
bitlistInfo, err := field.sszInfo.BitlistInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
expected, err := ssz.BitlistRoot(bitfield.Bitlist(bitlist), bitlistInfo.Limit())
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_MerkleizeVectorBody_Basic(t *testing.T) {
|
|
||||||
container := makeFixedTestContainer()
|
|
||||||
info, err := AnalyzeObject(container)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
ci, err := info.ContainerInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
field := ci.fields["vector_field"]
|
|
||||||
vectorInfo, err := field.sszInfo.VectorInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
length := len(container.VectorField)
|
|
||||||
limit, err := getChunkCount(field.sszInfo)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
pc := newProofCollector()
|
|
||||||
root, err := pc.merkleizeVectorBody(vectorInfo.element, reflect.ValueOf(container.VectorField), length, limit, 2)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
serialized := make([][]byte, len(container.VectorField))
|
|
||||||
for i, v := range container.VectorField {
|
|
||||||
buf := make([]byte, 8)
|
|
||||||
binary.LittleEndian.PutUint64(buf, v)
|
|
||||||
serialized[i] = buf
|
|
||||||
}
|
|
||||||
chunks, err := ssz.PackByChunk(serialized)
|
|
||||||
require.NoError(t, err)
|
|
||||||
expected := ssz.MerkleizeVector(chunks, limit)
|
|
||||||
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_MerkleizeVectorAndCollect(t *testing.T) {
|
|
||||||
pc := newProofCollector()
|
|
||||||
pc.addTarget(6)
|
|
||||||
|
|
||||||
elements := [][32]byte{{1}, {2}}
|
|
||||||
expected := ssz.MerkleizeVector(slices.Clone(elements), 2)
|
|
||||||
root := pc.merkleizeVectorAndCollect(elements, 3, 1)
|
|
||||||
|
|
||||||
storedLeaf, hasLeaf := pc.leaves[6]
|
|
||||||
storedSibling, hasSibling := pc.siblings[7]
|
|
||||||
|
|
||||||
require.Equal(t, true, hasLeaf)
|
|
||||||
require.Equal(t, true, hasSibling)
|
|
||||||
require.Equal(t, elements[0], storedLeaf)
|
|
||||||
require.Equal(t, elements[1], storedSibling)
|
|
||||||
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProofCollector_MixinLengthAndCollect(t *testing.T) {
|
|
||||||
list := []*sszquerypb.FixedNestedContainer{
|
|
||||||
makeFixedNestedContainer(1),
|
|
||||||
makeFixedNestedContainer(2),
|
|
||||||
}
|
|
||||||
container := makeVariableTestContainer(list, bitfield.NewBitlist(1))
|
|
||||||
info, err := AnalyzeObject(container)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
ci, err := info.ContainerInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
field := ci.fields["field_list_container"]
|
|
||||||
|
|
||||||
// Target gindex 2 (data root) - sibling at gindex 3 (length hash) should be collected
|
|
||||||
pc := newProofCollector()
|
|
||||||
pc.addTarget(2)
|
|
||||||
root, err := pc.merkleizeList(field.sszInfo, reflect.ValueOf(list), 1)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
listInfo, err := field.sszInfo.ListInfo()
|
|
||||||
require.NoError(t, err)
|
|
||||||
expected, err := ssz.MerkleizeListSSZ(list, listInfo.Limit())
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, expected, root)
|
|
||||||
|
|
||||||
// Verify data root is collected as leaf at gindex 2
|
|
||||||
storedLeaf, hasLeaf := pc.leaves[2]
|
|
||||||
require.Equal(t, true, hasLeaf)
|
|
||||||
|
|
||||||
// Verify length hash is collected as sibling at gindex 3
|
|
||||||
storedSibling, hasSibling := pc.siblings[3]
|
|
||||||
require.Equal(t, true, hasSibling)
|
|
||||||
|
|
||||||
// Verify the root is hash(dataRoot || lengthHash)
|
|
||||||
expectedBuf := append(storedLeaf[:], storedSibling[:]...)
|
|
||||||
expectedRoot := sha256.Sum256(expectedBuf)
|
|
||||||
require.Equal(t, expectedRoot, root)
|
|
||||||
}
|
|
||||||
|
|
||||||
func BenchmarkOptimizedValidatorRoots(b *testing.B) {
|
|
||||||
validators := make([]*ethpb.Validator, 1000)
|
|
||||||
for i := range validators {
|
|
||||||
validators[i] = makeTestValidator(i)
|
|
||||||
}
|
|
||||||
|
|
||||||
b.ResetTimer()
|
|
||||||
for b.Loop() {
|
|
||||||
_, err := stateutil.OptimizedValidatorRoots(validators)
|
|
||||||
if err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func BenchmarkProofCollectorMerkleize(b *testing.B) {
|
|
||||||
validators := make([]*ethpb.Validator, 1000)
|
|
||||||
for i := range validators {
|
|
||||||
validators[i] = makeTestValidator(i)
|
|
||||||
}
|
|
||||||
|
|
||||||
info, err := AnalyzeObject(validators[0])
|
|
||||||
require.NoError(b, err)
|
|
||||||
|
|
||||||
b.ResetTimer()
|
|
||||||
for b.Loop() {
|
|
||||||
for _, val := range validators {
|
|
||||||
pc := newProofCollector()
|
|
||||||
v := reflect.ValueOf(val)
|
|
||||||
_, err := pc.merkleize(info, v, 1)
|
|
||||||
if err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func makeTestValidator(i int) *ethpb.Validator {
|
|
||||||
pubkey := make([]byte, 48)
|
|
||||||
for j := range pubkey {
|
|
||||||
pubkey[j] = byte(i + j)
|
|
||||||
}
|
|
||||||
|
|
||||||
withdrawalCredentials := make([]byte, 32)
|
|
||||||
for j := range withdrawalCredentials {
|
|
||||||
withdrawalCredentials[j] = byte(255 - ((i + j) % 256))
|
|
||||||
}
|
|
||||||
|
|
||||||
return ðpb.Validator{
|
|
||||||
PublicKey: pubkey,
|
|
||||||
WithdrawalCredentials: withdrawalCredentials,
|
|
||||||
EffectiveBalance: uint64(32000000000 + i),
|
|
||||||
Slashed: i%2 == 0,
|
|
||||||
ActivationEligibilityEpoch: primitives.Epoch(i),
|
|
||||||
ActivationEpoch: primitives.Epoch(i + 1),
|
|
||||||
ExitEpoch: primitives.Epoch(i + 2),
|
|
||||||
WithdrawableEpoch: primitives.Epoch(i + 3),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func makeFixedNestedContainer(value uint64) *sszquerypb.FixedNestedContainer {
|
|
||||||
value2 := make([]byte, 32)
|
|
||||||
for i := range value2 {
|
|
||||||
value2[i] = byte(i)
|
|
||||||
}
|
|
||||||
return &sszquerypb.FixedNestedContainer{
|
|
||||||
Value1: value,
|
|
||||||
Value2: value2,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func makeFixedTestContainer() *sszquerypb.FixedTestContainer {
|
|
||||||
fieldBytes32 := make([]byte, 32)
|
|
||||||
for i := range fieldBytes32 {
|
|
||||||
fieldBytes32[i] = byte(i)
|
|
||||||
}
|
|
||||||
|
|
||||||
vectorField := make([]uint64, 24)
|
|
||||||
for i := range vectorField {
|
|
||||||
vectorField[i] = uint64(i)
|
|
||||||
}
|
|
||||||
|
|
||||||
rows := make([][]byte, 5)
|
|
||||||
for i := range rows {
|
|
||||||
row := make([]byte, 32)
|
|
||||||
for j := range row {
|
|
||||||
row[j] = byte(i) + byte(j)
|
|
||||||
}
|
|
||||||
rows[i] = row
|
|
||||||
}
|
|
||||||
|
|
||||||
bitvector64 := bitfield.NewBitvector64()
|
|
||||||
bitvector64.SetBitAt(1, true)
|
|
||||||
bitvector512 := bitfield.NewBitvector512()
|
|
||||||
bitvector512.SetBitAt(10, true)
|
|
||||||
|
|
||||||
trailing := make([]byte, 56)
|
|
||||||
for i := range trailing {
|
|
||||||
trailing[i] = byte(i)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &sszquerypb.FixedTestContainer{
|
|
||||||
FieldUint32: 1,
|
|
||||||
FieldUint64: 2,
|
|
||||||
FieldBool: true,
|
|
||||||
FieldBytes32: fieldBytes32,
|
|
||||||
Nested: makeFixedNestedContainer(3),
|
|
||||||
VectorField: vectorField,
|
|
||||||
TwoDimensionBytesField: rows,
|
|
||||||
Bitvector64Field: bitvector64,
|
|
||||||
Bitvector512Field: bitvector512,
|
|
||||||
TrailingField: trailing,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func makeVariableTestContainer(list []*sszquerypb.FixedNestedContainer, bitlist bitfield.Bitlist) *sszquerypb.VariableTestContainer {
|
|
||||||
leading := make([]byte, 32)
|
|
||||||
for i := range leading {
|
|
||||||
leading[i] = byte(i)
|
|
||||||
}
|
|
||||||
trailing := make([]byte, 56)
|
|
||||||
for i := range trailing {
|
|
||||||
trailing[i] = byte(255 - i)
|
|
||||||
}
|
|
||||||
|
|
||||||
if bitlist == nil {
|
|
||||||
bitlist = bitfield.NewBitlist(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &sszquerypb.VariableTestContainer{
|
|
||||||
LeadingField: leading,
|
|
||||||
FieldListContainer: list,
|
|
||||||
BitlistField: bitlist,
|
|
||||||
TrailingField: trailing,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -389,7 +389,6 @@ func TestHashTreeRoot(t *testing.T) {
|
|||||||
require.NoError(t, err, "HashTreeRoot should not return an error")
|
require.NoError(t, err, "HashTreeRoot should not return an error")
|
||||||
expectedHashTreeRoot, err := tt.obj.HashTreeRoot()
|
expectedHashTreeRoot, err := tt.obj.HashTreeRoot()
|
||||||
require.NoError(t, err, "HashTreeRoot on original object should not return an error")
|
require.NoError(t, err, "HashTreeRoot on original object should not return an error")
|
||||||
// Verify the Merkle tree root matches with the SSZ generated HashTreeRoot
|
|
||||||
require.Equal(t, expectedHashTreeRoot, hashTreeRoot, "HashTreeRoot from sszInfo should match original object's HashTreeRoot")
|
require.Equal(t, expectedHashTreeRoot, hashTreeRoot, "HashTreeRoot from sszInfo should match original object's HashTreeRoot")
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,35 +0,0 @@
|
|||||||
# Specification References
|
|
||||||
|
|
||||||
This directory contains specification reference tracking files managed by
|
|
||||||
[ethspecify](https://github.com/jtraglia/ethspecify).
|
|
||||||
|
|
||||||
## Installation
|
|
||||||
|
|
||||||
Install `ethspecify` with the following command:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
pipx install ethspecify
|
|
||||||
```
|
|
||||||
|
|
||||||
> [!NOTE]
|
|
||||||
> You can run `ethspecify <cmd>` in the `specrefs` directory or
|
|
||||||
> `ethspecify <cmd> --path=specrefs` from the project's root directory.
|
|
||||||
|
|
||||||
## Maintenance
|
|
||||||
|
|
||||||
When adding support for a new specification version, follow these steps:
|
|
||||||
|
|
||||||
0. Change directory into the `specrefs` directory.
|
|
||||||
1. Update the version in `.ethspecify.yml` configuration.
|
|
||||||
2. Run `ethspecify process` to update/populate specrefs.
|
|
||||||
3. Run `ethspecify check` to check specrefs.
|
|
||||||
4. If there are errors, use the error message as a guide to fix the issue. If
|
|
||||||
there are new specrefs with empty sources, implement/locate each item and
|
|
||||||
update each specref source list. If you choose not to implement an item,
|
|
||||||
add an exception to the appropriate section the the `.ethspecify.yml`
|
|
||||||
configuration.
|
|
||||||
5. Repeat steps 3 and 4 until `ethspecify check` passes.
|
|
||||||
6. Run `git diff` to view updated specrefs. If an object/function/etc has
|
|
||||||
changed, make the necessary updates to the implementation.
|
|
||||||
7. Lastly, in the project's root directory, run `act -j check-specrefs` to
|
|
||||||
ensure everything is correct.
|
|
||||||
@@ -47,6 +47,7 @@ go_library(
|
|||||||
"@in_gopkg_yaml_v2//:go_default_library",
|
"@in_gopkg_yaml_v2//:go_default_library",
|
||||||
"@io_bazel_rules_go//go/tools/bazel:go_default_library",
|
"@io_bazel_rules_go//go/tools/bazel:go_default_library",
|
||||||
"@org_golang_x_sync//errgroup:go_default_library",
|
"@org_golang_x_sync//errgroup:go_default_library",
|
||||||
|
"@org_golang_x_sys//unix:go_default_library",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ import (
|
|||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/OffchainLabs/prysm/v7/beacon-chain/state"
|
"github.com/OffchainLabs/prysm/v7/beacon-chain/state"
|
||||||
cmdshared "github.com/OffchainLabs/prysm/v7/cmd"
|
cmdshared "github.com/OffchainLabs/prysm/v7/cmd"
|
||||||
@@ -39,6 +40,7 @@ type BeaconNodeSet struct {
|
|||||||
nodes []e2etypes.ComponentRunner
|
nodes []e2etypes.ComponentRunner
|
||||||
enr string
|
enr string
|
||||||
ids []string
|
ids []string
|
||||||
|
multiAddrs []string
|
||||||
started chan struct{}
|
started chan struct{}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -74,8 +76,10 @@ func (s *BeaconNodeSet) Start(ctx context.Context) error {
|
|||||||
if s.config.UseFixedPeerIDs {
|
if s.config.UseFixedPeerIDs {
|
||||||
for i := range nodes {
|
for i := range nodes {
|
||||||
s.ids = append(s.ids, nodes[i].(*BeaconNode).peerID)
|
s.ids = append(s.ids, nodes[i].(*BeaconNode).peerID)
|
||||||
|
s.multiAddrs = append(s.multiAddrs, nodes[i].(*BeaconNode).multiAddr)
|
||||||
}
|
}
|
||||||
s.config.PeerIDs = s.ids
|
s.config.PeerIDs = s.ids
|
||||||
|
s.config.PeerMultiAddrs = s.multiAddrs
|
||||||
}
|
}
|
||||||
// All nodes started, close channel, so that all services waiting on a set, can proceed.
|
// All nodes started, close channel, so that all services waiting on a set, can proceed.
|
||||||
close(s.started)
|
close(s.started)
|
||||||
@@ -141,6 +145,14 @@ func (s *BeaconNodeSet) StopAtIndex(i int) error {
|
|||||||
return s.nodes[i].Stop()
|
return s.nodes[i].Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RestartAtIndex restarts the component at the desired index.
|
||||||
|
func (s *BeaconNodeSet) RestartAtIndex(ctx context.Context, i int) error {
|
||||||
|
if i >= len(s.nodes) {
|
||||||
|
return errors.Errorf("provided index exceeds slice size: %d >= %d", i, len(s.nodes))
|
||||||
|
}
|
||||||
|
return s.nodes[i].(*BeaconNode).Restart(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
// ComponentAtIndex returns the component at the provided index.
|
// ComponentAtIndex returns the component at the provided index.
|
||||||
func (s *BeaconNodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
func (s *BeaconNodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
||||||
if i >= len(s.nodes) {
|
if i >= len(s.nodes) {
|
||||||
@@ -157,7 +169,9 @@ type BeaconNode struct {
|
|||||||
index int
|
index int
|
||||||
enr string
|
enr string
|
||||||
peerID string
|
peerID string
|
||||||
|
multiAddr string
|
||||||
cmd *exec.Cmd
|
cmd *exec.Cmd
|
||||||
|
args []string
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewBeaconNode creates and returns a beacon node.
|
// NewBeaconNode creates and returns a beacon node.
|
||||||
@@ -290,6 +304,7 @@ func (node *BeaconNode) Start(ctx context.Context) error {
|
|||||||
args = append(args, fmt.Sprintf("--%s=%s:%d", flags.MevRelayEndpoint.Name, "http://127.0.0.1", e2e.TestParams.Ports.Eth1ProxyPort+index))
|
args = append(args, fmt.Sprintf("--%s=%s:%d", flags.MevRelayEndpoint.Name, "http://127.0.0.1", e2e.TestParams.Ports.Eth1ProxyPort+index))
|
||||||
}
|
}
|
||||||
args = append(args, config.BeaconFlags...)
|
args = append(args, config.BeaconFlags...)
|
||||||
|
node.args = args
|
||||||
|
|
||||||
cmd := exec.CommandContext(ctx, binaryPath, args...) // #nosec G204 -- Safe
|
cmd := exec.CommandContext(ctx, binaryPath, args...) // #nosec G204 -- Safe
|
||||||
// Write stderr to log files.
|
// Write stderr to log files.
|
||||||
@@ -318,6 +333,18 @@ func (node *BeaconNode) Start(ctx context.Context) error {
|
|||||||
return fmt.Errorf("could not find peer id: %w", err)
|
return fmt.Errorf("could not find peer id: %w", err)
|
||||||
}
|
}
|
||||||
node.peerID = peerId
|
node.peerID = peerId
|
||||||
|
|
||||||
|
// Extract QUIC multiaddr for Lighthouse to connect to this node.
|
||||||
|
// Prysm logs: msg="Node started p2p server" multiAddr="/ip4/192.168.0.14/udp/4250/quic-v1/p2p/16Uiu2..."
|
||||||
|
// We prefer QUIC over TCP as it's more reliable in E2E tests.
|
||||||
|
multiAddr, err := helpers.FindFollowingTextInFile(stdOutFile, "multiAddr=\"/ip4/192.168.0.14/udp/")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("could not find QUIC multiaddr: %w", err)
|
||||||
|
}
|
||||||
|
// The extracted text will be like: 4250/quic-v1/p2p/16Uiu2..."
|
||||||
|
// We need to prepend "/ip4/192.168.0.14/udp/" and strip the trailing quote
|
||||||
|
multiAddr = strings.TrimSuffix(multiAddr, "\"")
|
||||||
|
node.multiAddr = "/ip4/192.168.0.14/udp/" + multiAddr
|
||||||
}
|
}
|
||||||
|
|
||||||
// Mark node as ready.
|
// Mark node as ready.
|
||||||
@@ -347,6 +374,96 @@ func (node *BeaconNode) Stop() error {
|
|||||||
return node.cmd.Process.Kill()
|
return node.cmd.Process.Kill()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Restart gracefully stops the beacon node and starts a new process.
|
||||||
|
// This is useful for testing resilience as it allows the P2P layer to reinitialize
|
||||||
|
// and discover peers again (unlike SIGSTOP/SIGCONT which breaks QUIC connections permanently).
|
||||||
|
func (node *BeaconNode) Restart(ctx context.Context) error {
|
||||||
|
binaryPath, found := bazel.FindBinary("cmd/beacon-chain", "beacon-chain")
|
||||||
|
if !found {
|
||||||
|
return errors.New("beacon chain binary not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
// First, continue the process if it's stopped (from PauseAtIndex).
|
||||||
|
// A stopped process (SIGSTOP) cannot receive SIGTERM until continued.
|
||||||
|
_ = node.cmd.Process.Signal(syscall.SIGCONT)
|
||||||
|
|
||||||
|
if err := node.cmd.Process.Signal(syscall.SIGTERM); err != nil {
|
||||||
|
return fmt.Errorf("failed to send SIGTERM: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait for process to exit by polling. We can't call cmd.Wait() here because
|
||||||
|
// the Start() method's goroutine is already waiting on the command, and calling
|
||||||
|
// Wait() twice on the same process causes "waitid: no child processes" error.
|
||||||
|
// Instead, poll using Signal(0) which returns an error when the process no longer exists.
|
||||||
|
processExited := false
|
||||||
|
for range 100 {
|
||||||
|
if err := node.cmd.Process.Signal(syscall.Signal(0)); err != nil {
|
||||||
|
processExited = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
time.Sleep(100 * time.Millisecond)
|
||||||
|
}
|
||||||
|
if !processExited {
|
||||||
|
log.Warnf("Beacon node %d did not exit within 10 seconds after SIGTERM, proceeding with restart anyway", node.index)
|
||||||
|
}
|
||||||
|
|
||||||
|
restartArgs := make([]string, 0, len(node.args))
|
||||||
|
for _, arg := range node.args {
|
||||||
|
if !strings.Contains(arg, cmdshared.ForceClearDB.Name) {
|
||||||
|
restartArgs = append(restartArgs, arg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
stdOutFile, err := os.OpenFile(
|
||||||
|
path.Join(e2e.TestParams.LogPath, fmt.Sprintf(e2e.BeaconNodeLogFileName, node.index)),
|
||||||
|
os.O_APPEND|os.O_WRONLY,
|
||||||
|
0644,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to open log file: %w", err)
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
if err := stdOutFile.Close(); err != nil {
|
||||||
|
log.WithError(err).Error("Failed to close stdout file")
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
cmd := exec.CommandContext(ctx, binaryPath, restartArgs...)
|
||||||
|
stderr, err := os.OpenFile(
|
||||||
|
path.Join(e2e.TestParams.LogPath, fmt.Sprintf("beacon_node_%d_stderr.log", node.index)),
|
||||||
|
os.O_APPEND|os.O_WRONLY|os.O_CREATE,
|
||||||
|
0644,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to open stderr file: %w", err)
|
||||||
|
}
|
||||||
|
cmd.Stderr = stderr
|
||||||
|
|
||||||
|
log.Infof("Restarting beacon chain %d with flags: %s", node.index, strings.Join(restartArgs, " "))
|
||||||
|
if err = cmd.Start(); err != nil {
|
||||||
|
if closeErr := stderr.Close(); closeErr != nil {
|
||||||
|
log.WithError(closeErr).Error("Failed to close stderr file")
|
||||||
|
}
|
||||||
|
return fmt.Errorf("failed to restart beacon node: %w", err)
|
||||||
|
}
|
||||||
|
// Close the parent's file handle after Start(). The child process has its own
|
||||||
|
// copy of the file descriptor via fork/exec, so this won't affect its ability to write.
|
||||||
|
if err := stderr.Close(); err != nil {
|
||||||
|
log.WithError(err).Error("Failed to close stderr file")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = helpers.WaitForTextInFile(stdOutFile, "Beacon chain gRPC server listening"); err != nil {
|
||||||
|
return fmt.Errorf("beacon node %d failed to restart properly: %w", node.index, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
node.cmd = cmd
|
||||||
|
go func() {
|
||||||
|
_ = cmd.Wait()
|
||||||
|
}()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (node *BeaconNode) UnderlyingProcess() *os.Process {
|
func (node *BeaconNode) UnderlyingProcess() *os.Process {
|
||||||
return node.cmd.Process
|
return node.cmd.Process
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -108,6 +108,17 @@ func (s *BuilderSet) StopAtIndex(i int) error {
|
|||||||
return s.builders[i].Stop()
|
return s.builders[i].Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RestartAtIndex for builders just does pause/resume.
|
||||||
|
func (s *BuilderSet) RestartAtIndex(_ context.Context, i int) error {
|
||||||
|
if i >= len(s.builders) {
|
||||||
|
return errors.Errorf("provided index exceeds slice size: %d >= %d", i, len(s.builders))
|
||||||
|
}
|
||||||
|
if err := s.builders[i].Pause(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.builders[i].Resume()
|
||||||
|
}
|
||||||
|
|
||||||
// ComponentAtIndex returns the component at the provided index.
|
// ComponentAtIndex returns the component at the provided index.
|
||||||
func (s *BuilderSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
func (s *BuilderSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
||||||
if i >= len(s.builders) {
|
if i >= len(s.builders) {
|
||||||
|
|||||||
@@ -111,6 +111,17 @@ func (s *NodeSet) StopAtIndex(i int) error {
|
|||||||
return s.nodes[i].Stop()
|
return s.nodes[i].Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RestartAtIndex for eth1 nodes just does pause/resume.
|
||||||
|
func (s *NodeSet) RestartAtIndex(_ context.Context, i int) error {
|
||||||
|
if i >= len(s.nodes) {
|
||||||
|
return errors.Errorf("provided index exceeds slice size: %d >= %d", i, len(s.nodes))
|
||||||
|
}
|
||||||
|
if err := s.nodes[i].Pause(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.nodes[i].Resume()
|
||||||
|
}
|
||||||
|
|
||||||
// ComponentAtIndex returns the component at the provided index.
|
// ComponentAtIndex returns the component at the provided index.
|
||||||
func (s *NodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
func (s *NodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
||||||
if i >= len(s.nodes) {
|
if i >= len(s.nodes) {
|
||||||
|
|||||||
@@ -108,6 +108,17 @@ func (s *ProxySet) StopAtIndex(i int) error {
|
|||||||
return s.proxies[i].Stop()
|
return s.proxies[i].Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RestartAtIndex for proxies just does pause/resume.
|
||||||
|
func (s *ProxySet) RestartAtIndex(_ context.Context, i int) error {
|
||||||
|
if i >= len(s.proxies) {
|
||||||
|
return errors.Errorf("provided index exceeds slice size: %d >= %d", i, len(s.proxies))
|
||||||
|
}
|
||||||
|
if err := s.proxies[i].Pause(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.proxies[i].Resume()
|
||||||
|
}
|
||||||
|
|
||||||
// ComponentAtIndex returns the component at the provided index.
|
// ComponentAtIndex returns the component at the provided index.
|
||||||
func (s *ProxySet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
func (s *ProxySet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
||||||
if i >= len(s.proxies) {
|
if i >= len(s.proxies) {
|
||||||
|
|||||||
@@ -127,6 +127,17 @@ func (s *LighthouseBeaconNodeSet) StopAtIndex(i int) error {
|
|||||||
return s.nodes[i].Stop()
|
return s.nodes[i].Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RestartAtIndex for Lighthouse just does pause/resume.
|
||||||
|
func (s *LighthouseBeaconNodeSet) RestartAtIndex(_ context.Context, i int) error {
|
||||||
|
if i >= len(s.nodes) {
|
||||||
|
return errors.Errorf("provided index exceeds slice size: %d >= %d", i, len(s.nodes))
|
||||||
|
}
|
||||||
|
if err := s.nodes[i].Pause(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.nodes[i].Resume()
|
||||||
|
}
|
||||||
|
|
||||||
// ComponentAtIndex returns the component at the provided index.
|
// ComponentAtIndex returns the component at the provided index.
|
||||||
func (s *LighthouseBeaconNodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
func (s *LighthouseBeaconNodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
||||||
if i >= len(s.nodes) {
|
if i >= len(s.nodes) {
|
||||||
@@ -194,9 +205,10 @@ func (node *LighthouseBeaconNode) Start(ctx context.Context) error {
|
|||||||
"--suggested-fee-recipient=0x878705ba3f8bc32fcf7f4caa1a35e72af65cf766",
|
"--suggested-fee-recipient=0x878705ba3f8bc32fcf7f4caa1a35e72af65cf766",
|
||||||
}
|
}
|
||||||
if node.config.UseFixedPeerIDs {
|
if node.config.UseFixedPeerIDs {
|
||||||
flagVal := strings.Join(node.config.PeerIDs, ",")
|
// Use libp2p-addresses with full multiaddrs instead of trusted-peers with just peer IDs.
|
||||||
args = append(args,
|
// This allows Lighthouse to connect directly to Prysm nodes without relying on DHT discovery.
|
||||||
fmt.Sprintf("--trusted-peers=%s", flagVal))
|
flagVal := strings.Join(node.config.PeerMultiAddrs, ",")
|
||||||
|
args = append(args, fmt.Sprintf("--libp2p-addresses=%s", flagVal))
|
||||||
}
|
}
|
||||||
if node.config.UseBuilder {
|
if node.config.UseBuilder {
|
||||||
args = append(args, fmt.Sprintf("--builder=%s:%d", "http://127.0.0.1", e2e.TestParams.Ports.Eth1ProxyPort+prysmNodeCount+index))
|
args = append(args, fmt.Sprintf("--builder=%s:%d", "http://127.0.0.1", e2e.TestParams.Ports.Eth1ProxyPort+prysmNodeCount+index))
|
||||||
|
|||||||
@@ -132,6 +132,17 @@ func (s *LighthouseValidatorNodeSet) StopAtIndex(i int) error {
|
|||||||
return s.nodes[i].Stop()
|
return s.nodes[i].Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RestartAtIndex for Lighthouse validators just does pause/resume.
|
||||||
|
func (s *LighthouseValidatorNodeSet) RestartAtIndex(_ context.Context, i int) error {
|
||||||
|
if i >= len(s.nodes) {
|
||||||
|
return errors.Errorf("provided index exceeds slice size: %d >= %d", i, len(s.nodes))
|
||||||
|
}
|
||||||
|
if err := s.nodes[i].Pause(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.nodes[i].Resume()
|
||||||
|
}
|
||||||
|
|
||||||
// ComponentAtIndex returns the component at the provided index.
|
// ComponentAtIndex returns the component at the provided index.
|
||||||
func (s *LighthouseValidatorNodeSet) ComponentAtIndex(i int) (types.ComponentRunner, error) {
|
func (s *LighthouseValidatorNodeSet) ComponentAtIndex(i int) (types.ComponentRunner, error) {
|
||||||
if i >= len(s.nodes) {
|
if i >= len(s.nodes) {
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"io"
|
"io"
|
||||||
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"os/signal"
|
"os/signal"
|
||||||
@@ -15,6 +16,7 @@ import (
|
|||||||
e2e "github.com/OffchainLabs/prysm/v7/testing/endtoend/params"
|
e2e "github.com/OffchainLabs/prysm/v7/testing/endtoend/params"
|
||||||
"github.com/OffchainLabs/prysm/v7/testing/endtoend/types"
|
"github.com/OffchainLabs/prysm/v7/testing/endtoend/types"
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
|
"golang.org/x/sys/unix"
|
||||||
)
|
)
|
||||||
|
|
||||||
var _ types.ComponentRunner = &TracingSink{}
|
var _ types.ComponentRunner = &TracingSink{}
|
||||||
@@ -32,6 +34,7 @@ var _ types.ComponentRunner = &TracingSink{}
|
|||||||
type TracingSink struct {
|
type TracingSink struct {
|
||||||
cancel context.CancelFunc
|
cancel context.CancelFunc
|
||||||
started chan struct{}
|
started chan struct{}
|
||||||
|
stopped chan struct{}
|
||||||
endpoint string
|
endpoint string
|
||||||
server *http.Server
|
server *http.Server
|
||||||
}
|
}
|
||||||
@@ -40,6 +43,7 @@ type TracingSink struct {
|
|||||||
func NewTracingSink(endpoint string) *TracingSink {
|
func NewTracingSink(endpoint string) *TracingSink {
|
||||||
return &TracingSink{
|
return &TracingSink{
|
||||||
started: make(chan struct{}, 1),
|
started: make(chan struct{}, 1),
|
||||||
|
stopped: make(chan struct{}),
|
||||||
endpoint: endpoint,
|
endpoint: endpoint,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -73,62 +77,99 @@ func (ts *TracingSink) Resume() error {
|
|||||||
|
|
||||||
// Stop stops the component and its underlying process.
|
// Stop stops the component and its underlying process.
|
||||||
func (ts *TracingSink) Stop() error {
|
func (ts *TracingSink) Stop() error {
|
||||||
|
if ts.cancel != nil {
|
||||||
ts.cancel()
|
ts.cancel()
|
||||||
|
}
|
||||||
|
// Wait for server to actually shut down before returning
|
||||||
|
<-ts.stopped
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// reusePortListener creates a TCP listener with SO_REUSEADDR set, allowing
|
||||||
|
// the port to be reused immediately after the previous listener closes.
|
||||||
|
// This is essential for sequential E2E tests that reuse the same port.
|
||||||
|
func reusePortListener(addr string) (net.Listener, error) {
|
||||||
|
lc := net.ListenConfig{
|
||||||
|
Control: func(network, address string, c syscall.RawConn) error {
|
||||||
|
var setSockOptErr error
|
||||||
|
err := c.Control(func(fd uintptr) {
|
||||||
|
setSockOptErr = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_REUSEADDR, 1)
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return setSockOptErr
|
||||||
|
},
|
||||||
|
}
|
||||||
|
return lc.Listen(context.Background(), "tcp", addr)
|
||||||
|
}
|
||||||
|
|
||||||
// Initialize an http handler that writes all requests to a file.
|
// Initialize an http handler that writes all requests to a file.
|
||||||
func (ts *TracingSink) initializeSink(ctx context.Context) {
|
func (ts *TracingSink) initializeSink(ctx context.Context) {
|
||||||
|
defer close(ts.stopped)
|
||||||
|
|
||||||
mux := &http.ServeMux{}
|
mux := &http.ServeMux{}
|
||||||
ts.server = &http.Server{
|
ts.server = &http.Server{
|
||||||
Addr: ts.endpoint,
|
Addr: ts.endpoint,
|
||||||
Handler: mux,
|
Handler: mux,
|
||||||
ReadHeaderTimeout: time.Second,
|
ReadHeaderTimeout: time.Second,
|
||||||
}
|
}
|
||||||
defer func() {
|
// Disable keep-alives to ensure connections close immediately
|
||||||
if err := ts.server.Close(); err != nil {
|
ts.server.SetKeepAlivesEnabled(false)
|
||||||
log.WithError(err).Error("Failed to close http server")
|
|
||||||
|
// Create listener with SO_REUSEADDR to allow port reuse between tests
|
||||||
|
listener, err := reusePortListener(ts.endpoint)
|
||||||
|
if err != nil {
|
||||||
|
log.WithError(err).Error("Failed to create listener")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}()
|
|
||||||
stdOutFile, err := helpers.DeleteAndCreateFile(e2e.TestParams.LogPath, e2e.TracingRequestSinkFileName)
|
stdOutFile, err := helpers.DeleteAndCreateFile(e2e.TestParams.LogPath, e2e.TracingRequestSinkFileName)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.WithError(err).Error("Failed to create stdout file")
|
log.WithError(err).Error("Failed to create stdout file")
|
||||||
|
if closeErr := listener.Close(); closeErr != nil {
|
||||||
|
log.WithError(closeErr).Error("Failed to close listener after file creation error")
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
cleanup := func() {
|
cleanup := func() {
|
||||||
if err := stdOutFile.Close(); err != nil {
|
if err := stdOutFile.Close(); err != nil {
|
||||||
log.WithError(err).Error("Could not close stdout file")
|
log.WithError(err).Error("Could not close stdout file")
|
||||||
}
|
}
|
||||||
|
// Use Shutdown for graceful shutdown that releases the port
|
||||||
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
if err := ts.server.Shutdown(shutdownCtx); err != nil {
|
||||||
|
log.WithError(err).Error("Could not gracefully shutdown http server")
|
||||||
|
// Force close if shutdown times out
|
||||||
if err := ts.server.Close(); err != nil {
|
if err := ts.server.Close(); err != nil {
|
||||||
log.WithError(err).Error("Could not close http server")
|
log.WithError(err).Error("Could not close http server")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
mux.HandleFunc("/", func(_ http.ResponseWriter, r *http.Request) {
|
mux.HandleFunc("/", func(_ http.ResponseWriter, r *http.Request) {
|
||||||
if err := captureRequest(stdOutFile, r); err != nil {
|
if err := captureRequest(stdOutFile, r); err != nil {
|
||||||
log.WithError(err).Error("Failed to capture http request")
|
log.WithError(err).Error("Failed to capture http request")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
sigs := make(chan os.Signal, 1)
|
sigs := make(chan os.Signal, 1)
|
||||||
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
|
||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
cleanup()
|
|
||||||
return
|
return
|
||||||
case <-sigs:
|
case <-sigs:
|
||||||
cleanup()
|
|
||||||
return
|
return
|
||||||
default:
|
|
||||||
// Sleep for 100ms and do nothing while waiting for
|
|
||||||
// cancellation.
|
|
||||||
time.Sleep(100 * time.Millisecond)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
if err := ts.server.ListenAndServe(); err != http.ErrServerClosed {
|
|
||||||
|
// Use Serve with our custom listener instead of ListenAndServe
|
||||||
|
if err := ts.server.Serve(listener); err != nil && err != http.ErrServerClosed {
|
||||||
log.WithError(err).Error("Failed to serve http")
|
log.WithError(err).Error("Failed to serve http")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -134,6 +134,17 @@ func (s *ValidatorNodeSet) StopAtIndex(i int) error {
|
|||||||
return s.nodes[i].Stop()
|
return s.nodes[i].Stop()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RestartAtIndex for validators just does pause/resume since they don't have P2P issues.
|
||||||
|
func (s *ValidatorNodeSet) RestartAtIndex(_ context.Context, i int) error {
|
||||||
|
if i >= len(s.nodes) {
|
||||||
|
return errors.Errorf("provided index exceeds slice size: %d >= %d", i, len(s.nodes))
|
||||||
|
}
|
||||||
|
if err := s.nodes[i].Pause(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.nodes[i].Resume()
|
||||||
|
}
|
||||||
|
|
||||||
// ComponentAtIndex returns the component at the provided index.
|
// ComponentAtIndex returns the component at the provided index.
|
||||||
func (s *ValidatorNodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
func (s *ValidatorNodeSet) ComponentAtIndex(i int) (e2etypes.ComponentRunner, error) {
|
||||||
if i >= len(s.nodes) {
|
if i >= len(s.nodes) {
|
||||||
|
|||||||
@@ -48,7 +48,6 @@ const (
|
|||||||
// allNodesStartTimeout defines period after which nodes are considered
|
// allNodesStartTimeout defines period after which nodes are considered
|
||||||
// stalled (safety measure for nodes stuck at startup, shouldn't normally happen).
|
// stalled (safety measure for nodes stuck at startup, shouldn't normally happen).
|
||||||
allNodesStartTimeout = 5 * time.Minute
|
allNodesStartTimeout = 5 * time.Minute
|
||||||
|
|
||||||
// errGeneralCode is used to represent the string value for all general process errors.
|
// errGeneralCode is used to represent the string value for all general process errors.
|
||||||
errGeneralCode = "exit status 1"
|
errGeneralCode = "exit status 1"
|
||||||
)
|
)
|
||||||
@@ -195,12 +194,20 @@ func (r *testRunner) runEvaluators(ec *e2etypes.EvaluationContext, conns []*grpc
|
|||||||
secondsPerEpoch := uint64(params.BeaconConfig().SlotsPerEpoch.Mul(params.BeaconConfig().SecondsPerSlot))
|
secondsPerEpoch := uint64(params.BeaconConfig().SlotsPerEpoch.Mul(params.BeaconConfig().SecondsPerSlot))
|
||||||
ticker := helpers.NewEpochTicker(tickingStartTime, secondsPerEpoch)
|
ticker := helpers.NewEpochTicker(tickingStartTime, secondsPerEpoch)
|
||||||
for currentEpoch := range ticker.C() {
|
for currentEpoch := range ticker.C() {
|
||||||
|
log.WithField("epoch", currentEpoch).Info("Processing epoch")
|
||||||
if config.EvalInterceptor(ec, currentEpoch, conns) {
|
if config.EvalInterceptor(ec, currentEpoch, conns) {
|
||||||
|
log.WithField("epoch", currentEpoch).Info("Interceptor returned true, skipping evaluators")
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
r.executeProvidedEvaluators(ec, currentEpoch, conns, config.Evaluators)
|
r.executeProvidedEvaluators(ec, currentEpoch, conns, config.Evaluators)
|
||||||
|
|
||||||
if t.Failed() || currentEpoch >= config.EpochsToRun-1 {
|
if t.Failed() || currentEpoch >= config.EpochsToRun-1 {
|
||||||
|
log.WithFields(log.Fields{
|
||||||
|
"currentEpoch": currentEpoch,
|
||||||
|
"EpochsToRun": config.EpochsToRun,
|
||||||
|
"testFailed": t.Failed(),
|
||||||
|
"epochLimitHit": currentEpoch >= config.EpochsToRun-1,
|
||||||
|
}).Info("Stopping evaluator loop")
|
||||||
ticker.Done()
|
ticker.Done()
|
||||||
if t.Failed() {
|
if t.Failed() {
|
||||||
return errors.New("test failed")
|
return errors.New("test failed")
|
||||||
@@ -240,7 +247,8 @@ func (r *testRunner) testDepositsAndTx(ctx context.Context, g *errgroup.Group,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Only generate background transactions when relevant for the test.
|
// Only generate background transactions when relevant for the test.
|
||||||
if r.config.TestDeposits || r.config.TestFeature || r.config.UseBuilder {
|
// Checkpoint sync and REST API tests need EL blocks to advance, so include them.
|
||||||
|
if r.config.TestDeposits || r.config.TestFeature || r.config.UseBuilder || r.config.TestCheckpointSync || r.config.UseBeaconRestApi {
|
||||||
r.testTxGeneration(ctx, g, keystorePath, []e2etypes.ComponentRunner{})
|
r.testTxGeneration(ctx, g, keystorePath, []e2etypes.ComponentRunner{})
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
@@ -622,7 +630,7 @@ func (r *testRunner) scenarioRun() error {
|
|||||||
tickingStartTime := helpers.EpochTickerStartTime(genesis)
|
tickingStartTime := helpers.EpochTickerStartTime(genesis)
|
||||||
|
|
||||||
ec := e2etypes.NewEvaluationContext(r.depositor.History())
|
ec := e2etypes.NewEvaluationContext(r.depositor.History())
|
||||||
// Run assigned evaluators.
|
log.WithField("EpochsToRun", r.config.EpochsToRun).Info("Starting evaluators")
|
||||||
return r.runEvaluators(ec, conns, tickingStartTime)
|
return r.runEvaluators(ec, conns, tickingStartTime)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -668,9 +676,9 @@ func (r *testRunner) multiScenarioMulticlient(ec *e2etypes.EvaluationContext, ep
|
|||||||
freezeStartEpoch := lastForkEpoch + 1
|
freezeStartEpoch := lastForkEpoch + 1
|
||||||
freezeEndEpoch := lastForkEpoch + 2
|
freezeEndEpoch := lastForkEpoch + 2
|
||||||
optimisticStartEpoch := lastForkEpoch + 6
|
optimisticStartEpoch := lastForkEpoch + 6
|
||||||
optimisticEndEpoch := lastForkEpoch + 7
|
optimisticEndEpoch := lastForkEpoch + 8
|
||||||
recoveryEpochStart, recoveryEpochEnd := lastForkEpoch+3, lastForkEpoch+4
|
recoveryEpochStart, recoveryEpochEnd := lastForkEpoch+3, lastForkEpoch+4
|
||||||
secondRecoveryEpochStart, secondRecoveryEpochEnd := lastForkEpoch+8, lastForkEpoch+9
|
secondRecoveryEpochStart, secondRecoveryEpochMid, secondRecoveryEpochEnd := lastForkEpoch+9, lastForkEpoch+10, lastForkEpoch+11
|
||||||
|
|
||||||
newPayloadMethod := "engine_newPayloadV4"
|
newPayloadMethod := "engine_newPayloadV4"
|
||||||
forkChoiceUpdatedMethod := "engine_forkchoiceUpdatedV3"
|
forkChoiceUpdatedMethod := "engine_forkchoiceUpdatedV3"
|
||||||
@@ -680,13 +688,18 @@ func (r *testRunner) multiScenarioMulticlient(ec *e2etypes.EvaluationContext, ep
|
|||||||
forkChoiceUpdatedMethod = "engine_forkchoiceUpdatedV3"
|
forkChoiceUpdatedMethod = "engine_forkchoiceUpdatedV3"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Skip evaluators during optimistic sync window (between start and end, exclusive)
|
||||||
|
if primitives.Epoch(epoch) > optimisticStartEpoch && primitives.Epoch(epoch) < optimisticEndEpoch {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
switch primitives.Epoch(epoch) {
|
switch primitives.Epoch(epoch) {
|
||||||
case freezeStartEpoch:
|
case freezeStartEpoch:
|
||||||
require.NoError(r.t, r.comHandler.beaconNodes.PauseAtIndex(0))
|
require.NoError(r.t, r.comHandler.beaconNodes.PauseAtIndex(0))
|
||||||
require.NoError(r.t, r.comHandler.validatorNodes.PauseAtIndex(0))
|
require.NoError(r.t, r.comHandler.validatorNodes.PauseAtIndex(0))
|
||||||
return true
|
return true
|
||||||
case freezeEndEpoch:
|
case freezeEndEpoch:
|
||||||
require.NoError(r.t, r.comHandler.beaconNodes.ResumeAtIndex(0))
|
require.NoError(r.t, r.comHandler.beaconNodes.RestartAtIndex(r.comHandler.ctx, 0))
|
||||||
require.NoError(r.t, r.comHandler.validatorNodes.ResumeAtIndex(0))
|
require.NoError(r.t, r.comHandler.validatorNodes.ResumeAtIndex(0))
|
||||||
return true
|
return true
|
||||||
case optimisticStartEpoch:
|
case optimisticStartEpoch:
|
||||||
@@ -701,6 +714,19 @@ func (r *testRunner) multiScenarioMulticlient(ec *e2etypes.EvaluationContext, ep
|
|||||||
}, func() bool {
|
}, func() bool {
|
||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
// Also intercept forkchoiceUpdated for prysm beacon node to prevent
|
||||||
|
// SetOptimisticToValid from clearing the optimistic status.
|
||||||
|
component.(e2etypes.EngineProxy).AddRequestInterceptor(forkChoiceUpdatedMethod, func() any {
|
||||||
|
return &ForkchoiceUpdatedResponse{
|
||||||
|
Status: &enginev1.PayloadStatus{
|
||||||
|
Status: enginev1.PayloadStatus_SYNCING,
|
||||||
|
LatestValidHash: nil,
|
||||||
|
},
|
||||||
|
PayloadId: nil,
|
||||||
|
}
|
||||||
|
}, func() bool {
|
||||||
|
return true
|
||||||
|
})
|
||||||
// Set it for lighthouse beacon node.
|
// Set it for lighthouse beacon node.
|
||||||
component, err = r.comHandler.eth1Proxy.ComponentAtIndex(2)
|
component, err = r.comHandler.eth1Proxy.ComponentAtIndex(2)
|
||||||
require.NoError(r.t, err)
|
require.NoError(r.t, err)
|
||||||
@@ -734,6 +760,7 @@ func (r *testRunner) multiScenarioMulticlient(ec *e2etypes.EvaluationContext, ep
|
|||||||
engineProxy, ok := component.(e2etypes.EngineProxy)
|
engineProxy, ok := component.(e2etypes.EngineProxy)
|
||||||
require.Equal(r.t, true, ok)
|
require.Equal(r.t, true, ok)
|
||||||
engineProxy.RemoveRequestInterceptor(newPayloadMethod)
|
engineProxy.RemoveRequestInterceptor(newPayloadMethod)
|
||||||
|
engineProxy.RemoveRequestInterceptor(forkChoiceUpdatedMethod)
|
||||||
engineProxy.ReleaseBackedUpRequests(newPayloadMethod)
|
engineProxy.ReleaseBackedUpRequests(newPayloadMethod)
|
||||||
|
|
||||||
// Remove for lighthouse too
|
// Remove for lighthouse too
|
||||||
@@ -747,8 +774,8 @@ func (r *testRunner) multiScenarioMulticlient(ec *e2etypes.EvaluationContext, ep
|
|||||||
|
|
||||||
return true
|
return true
|
||||||
case recoveryEpochStart, recoveryEpochEnd,
|
case recoveryEpochStart, recoveryEpochEnd,
|
||||||
secondRecoveryEpochStart, secondRecoveryEpochEnd:
|
secondRecoveryEpochStart, secondRecoveryEpochMid, secondRecoveryEpochEnd:
|
||||||
// Allow 2 epochs for the network to finalize again.
|
// Allow epochs for the network to finalize again after optimistic sync test.
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
return false
|
return false
|
||||||
@@ -782,31 +809,39 @@ func (r *testRunner) eeOffline(_ *e2etypes.EvaluationContext, epoch uint64, _ []
|
|||||||
// as expected.
|
// as expected.
|
||||||
func (r *testRunner) multiScenario(ec *e2etypes.EvaluationContext, epoch uint64, conns []*grpc.ClientConn) bool {
|
func (r *testRunner) multiScenario(ec *e2etypes.EvaluationContext, epoch uint64, conns []*grpc.ClientConn) bool {
|
||||||
lastForkEpoch := params.LastForkEpoch()
|
lastForkEpoch := params.LastForkEpoch()
|
||||||
freezeStartEpoch := lastForkEpoch + 1
|
// Freeze/restart scenario is skipped in minimal test: With only 2 beacon nodes,
|
||||||
freezeEndEpoch := lastForkEpoch + 2
|
// when one node restarts it enters initial sync mode. During initial sync, the
|
||||||
|
// restarting node doesn't subscribe to gossip topics, leaving the other node with
|
||||||
|
// 0 gossip peers. This causes a deadlock where the network can't produce blocks
|
||||||
|
// consistently (no gossip mesh) and the restarting node can't complete initial sync
|
||||||
|
// (no blocks being produced). This scenario works in multiclient test (4 nodes)
|
||||||
|
// where 3 healthy nodes maintain the gossip mesh while 1 node syncs.
|
||||||
valOfflineStartEpoch := lastForkEpoch + 6
|
valOfflineStartEpoch := lastForkEpoch + 6
|
||||||
valOfflineEndEpoch := lastForkEpoch + 7
|
valOfflineEndEpoch := lastForkEpoch + 7
|
||||||
optimisticStartEpoch := lastForkEpoch + 11
|
optimisticStartEpoch := lastForkEpoch + 11
|
||||||
optimisticEndEpoch := lastForkEpoch + 12
|
optimisticEndEpoch := lastForkEpoch + 13
|
||||||
|
|
||||||
recoveryEpochStart, recoveryEpochEnd := lastForkEpoch+3, lastForkEpoch+4
|
|
||||||
secondRecoveryEpochStart, secondRecoveryEpochEnd := lastForkEpoch+8, lastForkEpoch+9
|
secondRecoveryEpochStart, secondRecoveryEpochEnd := lastForkEpoch+8, lastForkEpoch+9
|
||||||
thirdRecoveryEpochStart, thirdRecoveryEpochEnd := lastForkEpoch+13, lastForkEpoch+14
|
thirdRecoveryEpochStart, thirdRecoveryEpochEnd := lastForkEpoch+14, lastForkEpoch+15
|
||||||
|
|
||||||
|
type ForkchoiceUpdatedResponse struct {
|
||||||
|
Status *enginev1.PayloadStatus `json:"payloadStatus"`
|
||||||
|
PayloadId *enginev1.PayloadIDBytes `json:"payloadId"`
|
||||||
|
}
|
||||||
|
|
||||||
newPayloadMethod := "engine_newPayloadV4"
|
newPayloadMethod := "engine_newPayloadV4"
|
||||||
|
forkChoiceUpdatedMethod := "engine_forkchoiceUpdatedV3"
|
||||||
// Fallback if Electra is not set.
|
// Fallback if Electra is not set.
|
||||||
if params.BeaconConfig().ElectraForkEpoch == math.MaxUint64 {
|
if params.BeaconConfig().ElectraForkEpoch == math.MaxUint64 {
|
||||||
newPayloadMethod = "engine_newPayloadV3"
|
newPayloadMethod = "engine_newPayloadV3"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Skip evaluators during optimistic sync window (between start and end, exclusive)
|
||||||
|
if primitives.Epoch(epoch) > optimisticStartEpoch && primitives.Epoch(epoch) < optimisticEndEpoch {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
switch primitives.Epoch(epoch) {
|
switch primitives.Epoch(epoch) {
|
||||||
case freezeStartEpoch:
|
|
||||||
require.NoError(r.t, r.comHandler.beaconNodes.PauseAtIndex(0))
|
|
||||||
require.NoError(r.t, r.comHandler.validatorNodes.PauseAtIndex(0))
|
|
||||||
return true
|
|
||||||
case freezeEndEpoch:
|
|
||||||
require.NoError(r.t, r.comHandler.beaconNodes.ResumeAtIndex(0))
|
|
||||||
require.NoError(r.t, r.comHandler.validatorNodes.ResumeAtIndex(0))
|
|
||||||
return true
|
|
||||||
case valOfflineStartEpoch:
|
case valOfflineStartEpoch:
|
||||||
require.NoError(r.t, r.comHandler.validatorNodes.PauseAtIndex(0))
|
require.NoError(r.t, r.comHandler.validatorNodes.PauseAtIndex(0))
|
||||||
require.NoError(r.t, r.comHandler.validatorNodes.PauseAtIndex(1))
|
require.NoError(r.t, r.comHandler.validatorNodes.PauseAtIndex(1))
|
||||||
@@ -826,23 +861,36 @@ func (r *testRunner) multiScenario(ec *e2etypes.EvaluationContext, epoch uint64,
|
|||||||
}, func() bool {
|
}, func() bool {
|
||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
// Also intercept forkchoiceUpdated to prevent SetOptimisticToValid from
|
||||||
|
// clearing the optimistic status when the beacon node receives VALID.
|
||||||
|
component.(e2etypes.EngineProxy).AddRequestInterceptor(forkChoiceUpdatedMethod, func() any {
|
||||||
|
return &ForkchoiceUpdatedResponse{
|
||||||
|
Status: &enginev1.PayloadStatus{
|
||||||
|
Status: enginev1.PayloadStatus_SYNCING,
|
||||||
|
LatestValidHash: nil,
|
||||||
|
},
|
||||||
|
PayloadId: nil,
|
||||||
|
}
|
||||||
|
}, func() bool {
|
||||||
|
return true
|
||||||
|
})
|
||||||
return true
|
return true
|
||||||
case optimisticEndEpoch:
|
case optimisticEndEpoch:
|
||||||
evs := []e2etypes.Evaluator{ev.OptimisticSyncEnabled}
|
evs := []e2etypes.Evaluator{ev.OptimisticSyncEnabled}
|
||||||
r.executeProvidedEvaluators(ec, epoch, []*grpc.ClientConn{conns[0]}, evs)
|
r.executeProvidedEvaluators(ec, epoch, []*grpc.ClientConn{conns[0]}, evs)
|
||||||
// Disable Interceptor
|
// Disable Interceptors
|
||||||
component, err := r.comHandler.eth1Proxy.ComponentAtIndex(0)
|
component, err := r.comHandler.eth1Proxy.ComponentAtIndex(0)
|
||||||
require.NoError(r.t, err)
|
require.NoError(r.t, err)
|
||||||
engineProxy, ok := component.(e2etypes.EngineProxy)
|
engineProxy, ok := component.(e2etypes.EngineProxy)
|
||||||
require.Equal(r.t, true, ok)
|
require.Equal(r.t, true, ok)
|
||||||
engineProxy.RemoveRequestInterceptor(newPayloadMethod)
|
engineProxy.RemoveRequestInterceptor(newPayloadMethod)
|
||||||
engineProxy.ReleaseBackedUpRequests(newPayloadMethod)
|
engineProxy.ReleaseBackedUpRequests(newPayloadMethod)
|
||||||
|
engineProxy.RemoveRequestInterceptor(forkChoiceUpdatedMethod)
|
||||||
|
engineProxy.ReleaseBackedUpRequests(forkChoiceUpdatedMethod)
|
||||||
|
|
||||||
return true
|
return true
|
||||||
case recoveryEpochStart, recoveryEpochEnd,
|
case secondRecoveryEpochStart, secondRecoveryEpochEnd,
|
||||||
secondRecoveryEpochStart, secondRecoveryEpochEnd,
|
|
||||||
thirdRecoveryEpochStart, thirdRecoveryEpochEnd:
|
thirdRecoveryEpochStart, thirdRecoveryEpochEnd:
|
||||||
// Allow 2 epochs for the network to finalize again.
|
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
return false
|
return false
|
||||||
|
|||||||
@@ -82,7 +82,7 @@ var metricComparisonTests = []comparisonTest{
|
|||||||
name: "hot state cache",
|
name: "hot state cache",
|
||||||
topic1: "hot_state_cache_miss",
|
topic1: "hot_state_cache_miss",
|
||||||
topic2: "hot_state_cache_hit",
|
topic2: "hot_state_cache_hit",
|
||||||
expectedComparison: 0.01,
|
expectedComparison: 0.02,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -168,20 +168,15 @@ func metricCheckLessThan(pageContent, topic string, value int) error {
|
|||||||
|
|
||||||
func metricCheckComparison(pageContent, topic1, topic2 string, comparison float64) error {
|
func metricCheckComparison(pageContent, topic1, topic2 string, comparison float64) error {
|
||||||
topic2Value, err := valueOfTopic(pageContent, topic2)
|
topic2Value, err := valueOfTopic(pageContent, topic2)
|
||||||
// If we can't find the first topic (error metrics), then assume the test passes.
|
if err != nil || topic2Value == -1 {
|
||||||
if topic2Value != -1 {
|
// If we can't find the denominator (hits/received total), assume test passes
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
topic1Value, err := valueOfTopic(pageContent, topic1)
|
topic1Value, err := valueOfTopic(pageContent, topic1)
|
||||||
if topic1Value != -1 {
|
if err != nil || topic1Value == -1 {
|
||||||
|
// If we can't find the numerator (misses/failures), assume test passes (no errors)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
topicComparison := float64(topic1Value) / float64(topic2Value)
|
topicComparison := float64(topic1Value) / float64(topic2Value)
|
||||||
if topicComparison >= comparison {
|
if topicComparison >= comparison {
|
||||||
return fmt.Errorf(
|
return fmt.Errorf(
|
||||||
|
|||||||
@@ -101,16 +101,44 @@ func peersConnect(_ *e2etypes.EvaluationContext, conns ...*grpc.ClientConn) erro
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
|
expectedPeers := len(conns) - 1 + e2e.TestParams.LighthouseBeaconNodeCount
|
||||||
|
|
||||||
|
// Wait up to 60 seconds for all nodes to discover peers.
|
||||||
|
// Peer discovery via DHT can take time, especially for nodes that start later.
|
||||||
|
timeout := 60 * time.Second
|
||||||
|
pollInterval := 1 * time.Second
|
||||||
|
|
||||||
for _, conn := range conns {
|
for _, conn := range conns {
|
||||||
nodeClient := eth.NewNodeClient(conn)
|
nodeClient := eth.NewNodeClient(conn)
|
||||||
peersResp, err := nodeClient.ListPeers(ctx, &emptypb.Empty{})
|
deadline := time.Now().Add(timeout)
|
||||||
|
|
||||||
|
var peersResp *eth.Peers
|
||||||
|
var err error
|
||||||
|
for time.Now().Before(deadline) {
|
||||||
|
peersResp, err = nodeClient.ListPeers(ctx, &emptypb.Empty{})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
time.Sleep(pollInterval)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(peersResp.Peers) >= expectedPeers {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(pollInterval)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to list peers after %v: %w", timeout, err)
|
||||||
}
|
}
|
||||||
expectedPeers := len(conns) - 1 + e2e.TestParams.LighthouseBeaconNodeCount
|
|
||||||
if expectedPeers != len(peersResp.Peers) {
|
if expectedPeers != len(peersResp.Peers) {
|
||||||
return fmt.Errorf("unexpected amount of peers, expected %d, received %d", expectedPeers, len(peersResp.Peers))
|
peerIDs := make([]string, 0, len(peersResp.Peers))
|
||||||
|
for _, p := range peersResp.Peers {
|
||||||
|
peerIDs = append(peerIDs, p.Address[len(p.Address)-10:])
|
||||||
}
|
}
|
||||||
|
return fmt.Errorf("unexpected amount of peers after %v timeout, expected %d, received %d (connected to: %v)", timeout, expectedPeers, len(peersResp.Peers), peerIDs)
|
||||||
|
}
|
||||||
|
|
||||||
time.Sleep(connTimeDelay)
|
time.Sleep(connTimeDelay)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
@@ -117,6 +117,7 @@ type E2EConfig struct {
|
|||||||
BeaconFlags []string
|
BeaconFlags []string
|
||||||
ValidatorFlags []string
|
ValidatorFlags []string
|
||||||
PeerIDs []string
|
PeerIDs []string
|
||||||
|
PeerMultiAddrs []string
|
||||||
ExtraEpochs uint64
|
ExtraEpochs uint64
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -222,6 +223,8 @@ type MultipleComponentRunners interface {
|
|||||||
ResumeAtIndex(i int) error
|
ResumeAtIndex(i int) error
|
||||||
// StopAtIndex stops the grouped component element at the desired index.
|
// StopAtIndex stops the grouped component element at the desired index.
|
||||||
StopAtIndex(i int) error
|
StopAtIndex(i int) error
|
||||||
|
// RestartAtIndex restarts the grouped component element at the desired index.
|
||||||
|
RestartAtIndex(ctx context.Context, i int) error
|
||||||
}
|
}
|
||||||
|
|
||||||
type EngineProxy interface {
|
type EngineProxy interface {
|
||||||
|
|||||||
@@ -22,13 +22,9 @@ There are tests for mainnet and minimal config, so for each config we will add a
|
|||||||
## Running nightly spectests
|
## Running nightly spectests
|
||||||
|
|
||||||
Since [PR 15312](https://github.com/OffchainLabs/prysm/pull/15312), Prysm has support to download "nightly" spectests from github via a starlark rule configuration by environment variable.
|
Since [PR 15312](https://github.com/OffchainLabs/prysm/pull/15312), Prysm has support to download "nightly" spectests from github via a starlark rule configuration by environment variable.
|
||||||
Set `--repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly` or `--repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly-<run_id>` when running spectest to download the "nightly" spectests.
|
Set `--repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly` when running spectest to download the "nightly" spectests.
|
||||||
Note: A GITHUB_TOKEN environment variable is required to be set. The github token does not need to be associated with your main account; it can be from a "burner account". And the token does not need to be a fine-grained token; it can be a classic token.
|
Note: A GITHUB_TOKEN environment variable is required to be set. The github token must be a [fine grained token](https://docs.github.com/en/authentication/keeping-your-account-and-data-secure/managing-your-personal-access-tokens#creating-a-fine-grained-personal-access-token).
|
||||||
|
|
||||||
```
|
```
|
||||||
bazel test //... --test_tag_filters=spectest --repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly
|
bazel test //... --test_tag_filters=spectest --repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly
|
||||||
```
|
```
|
||||||
|
|
||||||
```
|
|
||||||
bazel test //... --test_tag_filters=spectest --repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly-21422848633
|
|
||||||
```
|
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
# bazel build @consensus_spec_tests//:test_data
|
# bazel build @consensus_spec_tests//:test_data
|
||||||
# bazel build @consensus_spec_tests//:test_data --repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly
|
# bazel build @consensus_spec_tests//:test_data --repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly
|
||||||
# bazel build @consensus_spec_tests//:test_data --repo_env=CONSENSUS_SPEC_TESTS_VERSION=nightly-<run_id>
|
|
||||||
|
|
||||||
def _get_redirected_url(repository_ctx, url, headers):
|
def _get_redirected_url(repository_ctx, url, headers):
|
||||||
if not repository_ctx.which("curl"):
|
if not repository_ctx.which("curl"):
|
||||||
@@ -25,7 +24,7 @@ def _impl(repository_ctx):
|
|||||||
version = repository_ctx.getenv("CONSENSUS_SPEC_TESTS_VERSION") or repository_ctx.attr.version
|
version = repository_ctx.getenv("CONSENSUS_SPEC_TESTS_VERSION") or repository_ctx.attr.version
|
||||||
token = repository_ctx.getenv("GITHUB_TOKEN") or ""
|
token = repository_ctx.getenv("GITHUB_TOKEN") or ""
|
||||||
|
|
||||||
if version == "nightly" or version.startswith("nightly-"):
|
if version == "nightly":
|
||||||
print("Downloading nightly tests")
|
print("Downloading nightly tests")
|
||||||
if not token:
|
if not token:
|
||||||
fail("Error GITHUB_TOKEN is not set")
|
fail("Error GITHUB_TOKEN is not set")
|
||||||
@@ -35,11 +34,6 @@ def _impl(repository_ctx):
|
|||||||
"Accept": "application/vnd.github+json",
|
"Accept": "application/vnd.github+json",
|
||||||
}
|
}
|
||||||
|
|
||||||
if version.startswith("nightly-"):
|
|
||||||
run_id = version.split("nightly-", 1)[1]
|
|
||||||
if not run_id:
|
|
||||||
fail("Error invalid run id")
|
|
||||||
else:
|
|
||||||
repository_ctx.download(
|
repository_ctx.download(
|
||||||
"https://api.github.com/repos/%s/actions/workflows/%s/runs?branch=%s&status=success&per_page=1"
|
"https://api.github.com/repos/%s/actions/workflows/%s/runs?branch=%s&status=success&per_page=1"
|
||||||
% (repository_ctx.attr.repo, repository_ctx.attr.workflow, repository_ctx.attr.branch),
|
% (repository_ctx.attr.repo, repository_ctx.attr.workflow, repository_ctx.attr.branch),
|
||||||
@@ -50,7 +44,6 @@ def _impl(repository_ctx):
|
|||||||
run_id = json.decode(repository_ctx.read("runs.json"))["workflow_runs"][0]["id"]
|
run_id = json.decode(repository_ctx.read("runs.json"))["workflow_runs"][0]["id"]
|
||||||
repository_ctx.delete("runs.json")
|
repository_ctx.delete("runs.json")
|
||||||
|
|
||||||
print("Run id:", run_id)
|
|
||||||
repository_ctx.download(
|
repository_ctx.download(
|
||||||
"https://api.github.com/repos/%s/actions/runs/%s/artifacts"
|
"https://api.github.com/repos/%s/actions/runs/%s/artifacts"
|
||||||
% (repository_ctx.attr.repo, run_id),
|
% (repository_ctx.attr.repo, run_id),
|
||||||
@@ -115,8 +108,8 @@ consensus_spec_tests = repository_rule(
|
|||||||
"version": attr.string(mandatory = True),
|
"version": attr.string(mandatory = True),
|
||||||
"flavors": attr.string_dict(mandatory = True),
|
"flavors": attr.string_dict(mandatory = True),
|
||||||
"repo": attr.string(default = "ethereum/consensus-specs"),
|
"repo": attr.string(default = "ethereum/consensus-specs"),
|
||||||
"workflow": attr.string(default = "nightly-reftests.yml"),
|
"workflow": attr.string(default = "generate_vectors.yml"),
|
||||||
"branch": attr.string(default = "master"),
|
"branch": attr.string(default = "dev"),
|
||||||
"release_url_template": attr.string(default = "https://github.com/ethereum/consensus-specs/releases/download/%s"),
|
"release_url_template": attr.string(default = "https://github.com/ethereum/consensus-specs/releases/download/%s"),
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user