mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-13 13:43:30 +00:00
69 lines
2.3 KiB
Go
69 lines
2.3 KiB
Go
package epoch
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/types"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
)
|
|
|
|
// Attestations returns the pending attestations of slots in the epoch
|
|
// (state.slot-EPOCH_LENGTH...state.slot-1), not attestations that got
|
|
// included in the chain during the epoch.
|
|
//
|
|
// Spec pseudocode definition:
|
|
// [a for a in state.latest_attestations if state.slot - EPOCH_LENGTH <=
|
|
// a.data.slot < state.slot]
|
|
func Attestations(state *pb.BeaconState) []*pb.PendingAttestationRecord {
|
|
epochLength := params.BeaconConfig().EpochLength
|
|
var thisEpochAttestations []*pb.PendingAttestationRecord
|
|
var earliestSlot uint64
|
|
|
|
for _, attestation := range state.LatestAttestations {
|
|
|
|
// If the state slot is less than epochLength, then the earliestSlot would
|
|
// result in a negative number. Therefore we should default to
|
|
// earliestSlot = 0 in this case.
|
|
if state.Slot > epochLength {
|
|
earliestSlot = state.Slot - epochLength
|
|
}
|
|
|
|
if earliestSlot <= attestation.GetData().Slot && attestation.GetData().Slot < state.Slot {
|
|
thisEpochAttestations = append(thisEpochAttestations, attestation)
|
|
}
|
|
}
|
|
return thisEpochAttestations
|
|
}
|
|
|
|
// BoundaryAttestations returns the pending attestations from
|
|
// the epoch boundary block.
|
|
//
|
|
// Spec pseudocode definition:
|
|
// [a for a in this_epoch_attestations if a.data.epoch_boundary_root ==
|
|
// get_block_root(state, state.slot-EPOCH_LENGTH) and a.justified_slot ==
|
|
// state.justified_slot]
|
|
func BoundaryAttestations(
|
|
state *pb.BeaconState,
|
|
thisEpochAttestations []*pb.PendingAttestationRecord,
|
|
) ([]*pb.PendingAttestationRecord, error) {
|
|
epochLength := params.BeaconConfig().EpochLength
|
|
var boundaryAttestations []*pb.PendingAttestationRecord
|
|
|
|
for _, attestation := range thisEpochAttestations {
|
|
|
|
boundaryBlockRoot, err := types.BlockRoot(state, state.Slot-epochLength)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
attestationData := attestation.GetData()
|
|
sameRoot := bytes.Equal(attestationData.JustifiedBlockRootHash32, boundaryBlockRoot)
|
|
sameSlotNum := attestationData.JustifiedSlot == state.JustifiedSlot
|
|
if sameRoot && sameSlotNum {
|
|
boundaryAttestations = append(boundaryAttestations, attestation)
|
|
}
|
|
}
|
|
return boundaryAttestations, nil
|
|
}
|