mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-26 05:17:22 +00:00
6bea17cb54
* Update libp2p to support go 1.19 * gaz * go mod tidy * Only update the minimum deps * go mod tidy * revert .bazelrc * Update go-libp2p to v0.22.0 and update import paths (#11440) * Fix import paths * Fix go-libp2p-peerstore import * Bazel updates * fix * revert some comments changes * revert some comment stuff * fix dependency issues * vendor problematic library * use your brain * remove * tests Co-authored-by: Marco Munizaga <marco@marcopolo.io> Co-authored-by: Nishant Das <nishdas93@gmail.com>
275 lines
9.2 KiB
Go
275 lines
9.2 KiB
Go
//go:build go1.18
|
|
|
|
package sync
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
|
pb "github.com/libp2p/go-libp2p-pubsub/pb"
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
gcache "github.com/patrickmn/go-cache"
|
|
mock "github.com/prysmaticlabs/prysm/v3/beacon-chain/blockchain/testing"
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/core/signing"
|
|
dbtest "github.com/prysmaticlabs/prysm/v3/beacon-chain/db/testing"
|
|
doublylinkedtree "github.com/prysmaticlabs/prysm/v3/beacon-chain/forkchoice/doubly-linked-tree"
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/p2p"
|
|
p2ptest "github.com/prysmaticlabs/prysm/v3/beacon-chain/p2p/testing"
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/state/stategen"
|
|
mockSync "github.com/prysmaticlabs/prysm/v3/beacon-chain/sync/initial-sync/testing"
|
|
lruwrpr "github.com/prysmaticlabs/prysm/v3/cache/lru"
|
|
"github.com/prysmaticlabs/prysm/v3/config/params"
|
|
ethpb "github.com/prysmaticlabs/prysm/v3/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/util"
|
|
)
|
|
|
|
func FuzzValidateBeaconBlockPubSub_Phase0(f *testing.F) {
|
|
db := dbtest.SetupDB(f)
|
|
p := p2ptest.NewFuzzTestP2P()
|
|
ctx := context.Background()
|
|
beaconState, privKeys := util.DeterministicGenesisState(f, 100)
|
|
parentBlock := util.NewBeaconBlock()
|
|
util.SaveBlock(f, ctx, db, parentBlock)
|
|
bRoot, err := parentBlock.Block.HashTreeRoot()
|
|
require.NoError(f, err)
|
|
require.NoError(f, db.SaveState(ctx, beaconState, bRoot))
|
|
require.NoError(f, db.SaveStateSummary(ctx, ðpb.StateSummary{Root: bRoot[:]}))
|
|
copied := beaconState.Copy()
|
|
require.NoError(f, copied.SetSlot(1))
|
|
proposerIdx, err := helpers.BeaconProposerIndex(ctx, copied)
|
|
require.NoError(f, err)
|
|
msg := util.NewBeaconBlock()
|
|
msg.Block.ParentRoot = bRoot[:]
|
|
msg.Block.Slot = 1
|
|
msg.Block.ProposerIndex = proposerIdx
|
|
msg.Signature, err = signing.ComputeDomainAndSign(beaconState, 0, msg.Block, params.BeaconConfig().DomainBeaconProposer, privKeys[proposerIdx])
|
|
require.NoError(f, err)
|
|
|
|
stateGen := stategen.New(db, doublylinkedtree.New())
|
|
chainService := &mock.ChainService{Genesis: time.Unix(time.Now().Unix()-int64(params.BeaconConfig().SecondsPerSlot), 0),
|
|
State: beaconState,
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
Root: make([]byte, 32),
|
|
},
|
|
DB: db,
|
|
}
|
|
r := &Service{
|
|
cfg: &config{
|
|
beaconDB: db,
|
|
p2p: p,
|
|
initialSync: &mockSync.Sync{IsSyncing: false},
|
|
chain: chainService,
|
|
blockNotifier: chainService.BlockNotifier(),
|
|
stateGen: stateGen,
|
|
},
|
|
seenBlockCache: lruwrpr.New(10),
|
|
badBlockCache: lruwrpr.New(10),
|
|
slotToPendingBlocks: gcache.New(time.Second, 2*time.Second),
|
|
seenPendingBlocks: make(map[[32]byte]bool),
|
|
}
|
|
buf := new(bytes.Buffer)
|
|
_, err = p.Encoding().EncodeGossip(buf, msg)
|
|
require.NoError(f, err)
|
|
topic := p2p.GossipTypeMapping[reflect.TypeOf(msg)]
|
|
digest, err := r.currentForkDigest()
|
|
assert.NoError(f, err)
|
|
topic = r.addDigestToTopic(topic, digest)
|
|
|
|
f.Add("junk", []byte("junk"), buf.Bytes(), []byte(topic))
|
|
f.Fuzz(func(t *testing.T, pid string, from, data, topic []byte) {
|
|
r.cfg.p2p = p2ptest.NewFuzzTestP2P()
|
|
r.rateLimiter = newRateLimiter(r.cfg.p2p)
|
|
cService := &mock.ChainService{
|
|
Genesis: time.Unix(time.Now().Unix()-int64(params.BeaconConfig().SecondsPerSlot*10000000), 0),
|
|
State: beaconState,
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
Root: make([]byte, 32),
|
|
},
|
|
DB: db,
|
|
}
|
|
r.cfg.chain = cService
|
|
r.cfg.blockNotifier = cService.BlockNotifier()
|
|
strTop := string(topic)
|
|
msg := &pubsub.Message{
|
|
Message: &pb.Message{
|
|
From: from,
|
|
Data: data,
|
|
Topic: &strTop,
|
|
},
|
|
}
|
|
_, err := r.validateBeaconBlockPubSub(ctx, peer.ID(pid), msg)
|
|
_ = err
|
|
})
|
|
}
|
|
|
|
func FuzzValidateBeaconBlockPubSub_Altair(f *testing.F) {
|
|
db := dbtest.SetupDB(f)
|
|
p := p2ptest.NewFuzzTestP2P()
|
|
ctx := context.Background()
|
|
beaconState, privKeys := util.DeterministicGenesisStateAltair(f, 100)
|
|
parentBlock := util.NewBeaconBlockAltair()
|
|
util.SaveBlock(f, ctx, db, parentBlock)
|
|
bRoot, err := parentBlock.Block.HashTreeRoot()
|
|
require.NoError(f, err)
|
|
require.NoError(f, db.SaveState(ctx, beaconState, bRoot))
|
|
require.NoError(f, db.SaveStateSummary(ctx, ðpb.StateSummary{Root: bRoot[:]}))
|
|
copied := beaconState.Copy()
|
|
require.NoError(f, copied.SetSlot(1))
|
|
proposerIdx, err := helpers.BeaconProposerIndex(ctx, copied)
|
|
require.NoError(f, err)
|
|
msg := util.NewBeaconBlock()
|
|
msg.Block.ParentRoot = bRoot[:]
|
|
msg.Block.Slot = 1
|
|
msg.Block.ProposerIndex = proposerIdx
|
|
msg.Signature, err = signing.ComputeDomainAndSign(beaconState, 0, msg.Block, params.BeaconConfig().DomainBeaconProposer, privKeys[proposerIdx])
|
|
require.NoError(f, err)
|
|
|
|
stateGen := stategen.New(db, doublylinkedtree.New())
|
|
chainService := &mock.ChainService{Genesis: time.Unix(time.Now().Unix()-int64(params.BeaconConfig().SecondsPerSlot), 0),
|
|
State: beaconState,
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
Root: make([]byte, 32),
|
|
},
|
|
DB: db,
|
|
}
|
|
r := &Service{
|
|
cfg: &config{
|
|
beaconDB: db,
|
|
p2p: p,
|
|
initialSync: &mockSync.Sync{IsSyncing: false},
|
|
chain: chainService,
|
|
blockNotifier: chainService.BlockNotifier(),
|
|
stateGen: stateGen,
|
|
},
|
|
seenBlockCache: lruwrpr.New(10),
|
|
badBlockCache: lruwrpr.New(10),
|
|
slotToPendingBlocks: gcache.New(time.Second, 2*time.Second),
|
|
seenPendingBlocks: make(map[[32]byte]bool),
|
|
}
|
|
buf := new(bytes.Buffer)
|
|
_, err = p.Encoding().EncodeGossip(buf, msg)
|
|
require.NoError(f, err)
|
|
topic := p2p.GossipTypeMapping[reflect.TypeOf(msg)]
|
|
digest, err := r.currentForkDigest()
|
|
assert.NoError(f, err)
|
|
topic = r.addDigestToTopic(topic, digest)
|
|
|
|
f.Add("junk", []byte("junk"), buf.Bytes(), []byte(topic))
|
|
f.Fuzz(func(t *testing.T, pid string, from, data, topic []byte) {
|
|
r.cfg.p2p = p2ptest.NewFuzzTestP2P()
|
|
r.rateLimiter = newRateLimiter(r.cfg.p2p)
|
|
cService := &mock.ChainService{
|
|
Genesis: time.Unix(time.Now().Unix()-int64(params.BeaconConfig().SecondsPerSlot*10000000), 0),
|
|
State: beaconState,
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
Root: make([]byte, 32),
|
|
},
|
|
DB: db,
|
|
}
|
|
r.cfg.chain = cService
|
|
r.cfg.blockNotifier = cService.BlockNotifier()
|
|
strTop := string(topic)
|
|
msg := &pubsub.Message{
|
|
Message: &pb.Message{
|
|
From: from,
|
|
Data: data,
|
|
Topic: &strTop,
|
|
},
|
|
}
|
|
_, err := r.validateBeaconBlockPubSub(ctx, peer.ID(pid), msg)
|
|
_ = err
|
|
})
|
|
}
|
|
|
|
func FuzzValidateBeaconBlockPubSub_Bellatrix(f *testing.F) {
|
|
db := dbtest.SetupDB(f)
|
|
p := p2ptest.NewFuzzTestP2P()
|
|
ctx := context.Background()
|
|
beaconState, privKeys := util.DeterministicGenesisStateBellatrix(f, 100)
|
|
parentBlock := util.NewBeaconBlockBellatrix()
|
|
util.SaveBlock(f, ctx, db, parentBlock)
|
|
bRoot, err := parentBlock.Block.HashTreeRoot()
|
|
require.NoError(f, err)
|
|
require.NoError(f, db.SaveState(ctx, beaconState, bRoot))
|
|
require.NoError(f, db.SaveStateSummary(ctx, ðpb.StateSummary{Root: bRoot[:]}))
|
|
copied := beaconState.Copy()
|
|
require.NoError(f, copied.SetSlot(1))
|
|
proposerIdx, err := helpers.BeaconProposerIndex(ctx, copied)
|
|
require.NoError(f, err)
|
|
msg := util.NewBeaconBlock()
|
|
msg.Block.ParentRoot = bRoot[:]
|
|
msg.Block.Slot = 1
|
|
msg.Block.ProposerIndex = proposerIdx
|
|
msg.Signature, err = signing.ComputeDomainAndSign(beaconState, 0, msg.Block, params.BeaconConfig().DomainBeaconProposer, privKeys[proposerIdx])
|
|
require.NoError(f, err)
|
|
|
|
stateGen := stategen.New(db, doublylinkedtree.New())
|
|
chainService := &mock.ChainService{Genesis: time.Unix(time.Now().Unix()-int64(params.BeaconConfig().SecondsPerSlot), 0),
|
|
State: beaconState,
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
Root: make([]byte, 32),
|
|
},
|
|
DB: db,
|
|
}
|
|
r := &Service{
|
|
cfg: &config{
|
|
beaconDB: db,
|
|
p2p: p,
|
|
initialSync: &mockSync.Sync{IsSyncing: false},
|
|
chain: chainService,
|
|
blockNotifier: chainService.BlockNotifier(),
|
|
stateGen: stateGen,
|
|
},
|
|
seenBlockCache: lruwrpr.New(10),
|
|
badBlockCache: lruwrpr.New(10),
|
|
slotToPendingBlocks: gcache.New(time.Second, 2*time.Second),
|
|
seenPendingBlocks: make(map[[32]byte]bool),
|
|
}
|
|
buf := new(bytes.Buffer)
|
|
_, err = p.Encoding().EncodeGossip(buf, msg)
|
|
require.NoError(f, err)
|
|
topic := p2p.GossipTypeMapping[reflect.TypeOf(msg)]
|
|
digest, err := r.currentForkDigest()
|
|
assert.NoError(f, err)
|
|
topic = r.addDigestToTopic(topic, digest)
|
|
|
|
f.Add("junk", []byte("junk"), buf.Bytes(), []byte(topic))
|
|
f.Fuzz(func(t *testing.T, pid string, from, data, topic []byte) {
|
|
r.cfg.p2p = p2ptest.NewFuzzTestP2P()
|
|
r.rateLimiter = newRateLimiter(r.cfg.p2p)
|
|
cService := &mock.ChainService{
|
|
Genesis: time.Unix(time.Now().Unix()-int64(params.BeaconConfig().SecondsPerSlot*10000000), 0),
|
|
State: beaconState,
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
Root: make([]byte, 32),
|
|
},
|
|
DB: db,
|
|
}
|
|
r.cfg.chain = cService
|
|
r.cfg.blockNotifier = cService.BlockNotifier()
|
|
strTop := string(topic)
|
|
msg := &pubsub.Message{
|
|
Message: &pb.Message{
|
|
From: from,
|
|
Data: data,
|
|
Topic: &strTop,
|
|
},
|
|
}
|
|
_, err := r.validateBeaconBlockPubSub(ctx, peer.ID(pid), msg)
|
|
_ = err
|
|
})
|
|
}
|