mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 20:07:17 +00:00
105 lines
4.1 KiB
Go
105 lines
4.1 KiB
Go
package validators
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/utils"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
)
|
|
|
|
// ShuffleValidatorRegistryToCommittees shuffles validator indices and splits them by slot and shard.
|
|
func ShuffleValidatorRegistryToCommittees(
|
|
seed [32]byte,
|
|
validators []*pb.ValidatorRecord,
|
|
crosslinkStartShard uint64,
|
|
) ([]*pb.ShardAndCommitteeArray, error) {
|
|
indices := ActiveValidatorIndices(validators)
|
|
// split the shuffled list for slot.
|
|
shuffledValidatorRegistry, err := utils.ShuffleIndices(seed, indices)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return splitBySlotShard(shuffledValidatorRegistry, crosslinkStartShard), nil
|
|
}
|
|
|
|
// InitialShardAndCommitteesForSlots initialises the committees for shards by shuffling the validators
|
|
// and assigning them to specific shards.
|
|
func InitialShardAndCommitteesForSlots(validators []*pb.ValidatorRecord) ([]*pb.ShardAndCommitteeArray, error) {
|
|
seed := [32]byte{}
|
|
committees, err := ShuffleValidatorRegistryToCommittees(seed, validators, 1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Initialize with 3 cycles of the same committees.
|
|
initialCommittees := make([]*pb.ShardAndCommitteeArray, 0, 3*params.BeaconConfig().CycleLength)
|
|
initialCommittees = append(initialCommittees, committees...)
|
|
initialCommittees = append(initialCommittees, committees...)
|
|
initialCommittees = append(initialCommittees, committees...)
|
|
return initialCommittees, nil
|
|
}
|
|
|
|
// AttesterIndices returns the validator indices that acted as attesters
|
|
// for a particular attestation.
|
|
func AttesterIndices(
|
|
shardCommittees *pb.ShardAndCommitteeArray,
|
|
attestation *pb.AggregatedAttestation,
|
|
) ([]uint32, error) {
|
|
shardCommitteesArray := shardCommittees.ArrayShardAndCommittee
|
|
for _, shardCommittee := range shardCommitteesArray {
|
|
if attestation.Shard == shardCommittee.Shard {
|
|
return shardCommittee.Committee, nil
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("unable to find committee for shard %d", attestation.Shard)
|
|
}
|
|
|
|
// splitBySlotShard splits the validator list into evenly sized committees and assigns each
|
|
// committee to a slot and a shard. If the validator set is large, multiple committees are assigned
|
|
// to a single slot and shard. See getCommitteesPerSlot for more details.
|
|
func splitBySlotShard(shuffledValidatorRegistry []uint32, crosslinkStartShard uint64) []*pb.ShardAndCommitteeArray {
|
|
committeesPerSlot := getCommitteesPerSlot(uint64(len(shuffledValidatorRegistry)))
|
|
committeBySlotAndShard := []*pb.ShardAndCommitteeArray{}
|
|
|
|
// split the validator indices by slot.
|
|
validatorsBySlot := utils.SplitIndices(shuffledValidatorRegistry, params.BeaconConfig().CycleLength)
|
|
for i, validatorsForSlot := range validatorsBySlot {
|
|
shardCommittees := []*pb.ShardAndCommittee{}
|
|
validatorsByShard := utils.SplitIndices(validatorsForSlot, committeesPerSlot)
|
|
shardStart := crosslinkStartShard + uint64(i)*committeesPerSlot
|
|
|
|
for j, validatorsForShard := range validatorsByShard {
|
|
shardID := (shardStart + uint64(j)) % params.BeaconConfig().ShardCount
|
|
shardCommittees = append(shardCommittees, &pb.ShardAndCommittee{
|
|
Shard: shardID,
|
|
Committee: validatorsForShard,
|
|
})
|
|
}
|
|
|
|
committeBySlotAndShard = append(committeBySlotAndShard, &pb.ShardAndCommitteeArray{
|
|
ArrayShardAndCommittee: shardCommittees,
|
|
})
|
|
}
|
|
return committeBySlotAndShard
|
|
}
|
|
|
|
// getCommitteesPerSlot calculates the parameters for ShuffleValidatorRegistryToCommittees.
|
|
// The minimum value for committeesPerSlot is 1.
|
|
// Otherwise, the value for committeesPerSlot is the smaller of
|
|
// numActiveValidatorRegistry / CycleLength / (MinCommitteeSize*2) + 1 or
|
|
// ShardCount / CycleLength.
|
|
func getCommitteesPerSlot(numActiveValidatorRegistry uint64) uint64 {
|
|
cycleLength := params.BeaconConfig().CycleLength
|
|
boundOnValidatorRegistry := numActiveValidatorRegistry/cycleLength/(params.BeaconConfig().TargetCommitteeSize*2) + 1
|
|
boundOnShardCount := params.BeaconConfig().ShardCount / cycleLength
|
|
// Ensure that comitteesPerSlot is at least 1.
|
|
if boundOnShardCount == 0 {
|
|
return 1
|
|
} else if boundOnValidatorRegistry > boundOnShardCount {
|
|
return boundOnShardCount
|
|
}
|
|
return boundOnValidatorRegistry
|
|
}
|