prysm-pulse/slasher/detection/attestations/mock_spanner.go
Ivan Martinez 31b1e6a7a8
Add double vote detection to spanner (#4954)
* Add double vote detection to spanner

* Add documentation

* Update slasher/detection/attestations/spanner.go

* Gazelle

* Fix filter output
2020-02-27 17:21:05 -05:00

78 lines
2.9 KiB
Go

package attestations
import (
"context"
"sync"
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
"github.com/prysmaticlabs/prysm/slasher/detection/attestations/iface"
"github.com/prysmaticlabs/prysm/slasher/detection/attestations/types"
)
var _ = iface.SpanDetector(&MockSpanDetector{})
// MockSpanDetector defines a struct which can detect slashable
// attestation offenses by tracking validator min-max
// spans from validators.
type MockSpanDetector struct {
// Slice of epochs for valindex => min-max span.
spans []map[uint64][2]uint16
lock sync.RWMutex
}
// DetectSlashingForValidator mocks a detected slashing, if the sent attestation data
// has a source epoch of 0, nothing will be detected. If the sent attestation data has a target
// epoch equal to or greater than 6, it will "detect" a surrounded vote for the target epoch + 1.
// If the target epoch is greater than 12, it will "detect" a surrounding vote for target epoch - 1.
// Lastly, if it has a target epoch less than 6, it will "detect" a double vote for the target epoch.
func (s *MockSpanDetector) DetectSlashingForValidator(
ctx context.Context,
validatorIdx uint64,
attData *ethpb.AttestationData,
) (*types.DetectionResult, error) {
switch {
// If the source epoch is 0, don't find a slashing.
case attData.Source.Epoch == 0:
return nil, nil
// If the target epoch is > 12, it will "detect" a surrounded saved attestation.
case attData.Target.Epoch > 12:
return &types.DetectionResult{
Kind: types.SurroundVote,
SlashableEpoch: attData.Target.Epoch - 1,
}, nil
// If the target epoch is >= 6 < 12, it will "detect" a surrounding saved attestation.
case attData.Target.Epoch >= 6:
return &types.DetectionResult{
Kind: types.SurroundVote,
SlashableEpoch: attData.Target.Epoch + 1,
}, nil
// If the target epoch is less than 6, it will "detect" a double vote.
default:
return &types.DetectionResult{
Kind: types.DoubleVote,
SlashableEpoch: attData.Target.Epoch,
}, nil
}
}
// SpanForEpochByValidator returns the specific min-max span for a
// validator index in a given epoch.
func (s *MockSpanDetector) SpanForEpochByValidator(ctx context.Context, valIdx uint64, epoch uint64) ([3]uint16, error) {
return [3]uint16{0, 0, 0}, nil
}
// ValidatorSpansByEpoch returns a list of all validator spans in a given epoch.
func (s *MockSpanDetector) ValidatorSpansByEpoch(ctx context.Context, epoch uint64) map[uint64][3]uint16 {
return make(map[uint64][3]uint16, 0)
}
// DeleteValidatorSpansByEpoch mocks the delete spans by epoch function.
func (s *MockSpanDetector) DeleteValidatorSpansByEpoch(ctx context.Context, validatorIdx uint64, epoch uint64) error {
return nil
}
// UpdateSpans is a mock for updating the spans for a given attestation..
func (s *MockSpanDetector) UpdateSpans(ctx context.Context, att *ethpb.IndexedAttestation) error {
return nil
}