mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-24 20:37:17 +00:00
02966e64d8
* Add Wrapper to LRU Cache to handle Invalid Parameters #9461 * Regenerate BUILD.bazel and simplify tests using lru.Cache * Fix: fuzz_exports.go build error * Fix: block_fuzz.go * Revert lru.Cache interface * Remove redundant err check in pending_attestations_queue_test.go * Add tests for lru wrapper * Use lru package in prysm/shared instead of lruwrpr * Fix: goimports * Fix: BUILD.bazel Co-authored-by: Nishant Das <nishdas93@gmail.com>
46 lines
1.4 KiB
Go
46 lines
1.4 KiB
Go
// +build libfuzzer
|
|
|
|
package sync
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
|
gcache "github.com/patrickmn/go-cache"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// NewRegularSyncFuzz service without registering handlers.
|
|
func NewRegularSyncFuzz(cfg *Config) *Service {
|
|
rLimiter := newRateLimiter(cfg.P2P)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
r := &Service{
|
|
cfg: cfg,
|
|
ctx: ctx,
|
|
cancel: cancel,
|
|
slotToPendingBlocks: gcache.New(time.Second, 2*time.Second),
|
|
seenPendingBlocks: make(map[[32]byte]bool),
|
|
blkRootToPendingAtts: make(map[[32]byte][]*ethpb.SignedAggregateAttestationAndProof),
|
|
rateLimiter: rLimiter,
|
|
}
|
|
|
|
return r
|
|
}
|
|
|
|
// FuzzValidateBeaconBlockPubSub exports private method validateBeaconBlockPubSub for fuzz testing.
|
|
func (s *Service) FuzzValidateBeaconBlockPubSub(ctx context.Context, pid peer.ID, msg *pubsub.Message) pubsub.ValidationResult {
|
|
return s.validateBeaconBlockPubSub(ctx, pid, msg)
|
|
}
|
|
|
|
// FuzzBeaconBlockSubscriber exports private method beaconBlockSubscriber for fuzz testing.
|
|
func (s *Service) FuzzBeaconBlockSubscriber(ctx context.Context, msg proto.Message) error {
|
|
return s.beaconBlockSubscriber(ctx, msg)
|
|
}
|
|
|
|
func (s *Service) InitCaches() {
|
|
s.initCaches()
|
|
}
|