mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
5569a68452
* Value assigned to a variable is never read before being overwritten * The result of append is not used anywhere * Suspicious assignment of range-loop vars detected * Unused method receiver detected * Revert "Auxiliary commit to revert individual files from 54edcb445484a2e5d79612e19af8e949b8861253" This reverts commit bbd1e1beabf7b0c5cfc4f514dcc820062ad6c063. * Method modifies receiver * Fix test * Duplicate imports detected * Incorrectly formatted error string * Types of function parameters can be combined * One more "Unused method receiver detected" * Unused parameter detected in function
93 lines
2.7 KiB
Go
93 lines
2.7 KiB
Go
package protoarray
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/prysmaticlabs/prysm/config/params"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// This computes validator balance delta from validator votes.
|
|
// It returns a list of deltas that represents the difference between old balances and new balances.
|
|
func computeDeltas(
|
|
ctx context.Context,
|
|
blockIndices map[[32]byte]uint64,
|
|
votes []Vote,
|
|
oldBalances, newBalances []uint64,
|
|
) ([]int, []Vote, error) {
|
|
_, span := trace.StartSpan(ctx, "protoArrayForkChoice.computeDeltas")
|
|
defer span.End()
|
|
|
|
deltas := make([]int, len(blockIndices))
|
|
|
|
for validatorIndex, vote := range votes {
|
|
oldBalance := uint64(0)
|
|
newBalance := uint64(0)
|
|
|
|
// Skip if validator has never voted for current root and next root (ie. if the
|
|
// votes are zero hash aka genesis block), there's nothing to compute.
|
|
if vote.currentRoot == params.BeaconConfig().ZeroHash && vote.nextRoot == params.BeaconConfig().ZeroHash {
|
|
continue
|
|
}
|
|
|
|
// If the validator index did not exist in `oldBalance` or `newBalance` list above, the balance is just 0.
|
|
if validatorIndex < len(oldBalances) {
|
|
oldBalance = oldBalances[validatorIndex]
|
|
}
|
|
if validatorIndex < len(newBalances) {
|
|
newBalance = newBalances[validatorIndex]
|
|
}
|
|
|
|
// Perform delta only if the validator's balance or vote has changed.
|
|
if vote.currentRoot != vote.nextRoot || oldBalance != newBalance {
|
|
// Ignore the vote if it's not known in `blockIndices`,
|
|
// that means we have not seen the block before.
|
|
nextDeltaIndex, ok := blockIndices[vote.nextRoot]
|
|
if ok {
|
|
// Protection against out of bound, the `nextDeltaIndex` which defines
|
|
// the block location in the dag can not exceed the total `delta` length.
|
|
if int(nextDeltaIndex) >= len(deltas) {
|
|
return nil, nil, errInvalidNodeDelta
|
|
}
|
|
deltas[nextDeltaIndex] += int(newBalance)
|
|
}
|
|
|
|
currentDeltaIndex, ok := blockIndices[vote.currentRoot]
|
|
if ok {
|
|
// Protection against out of bound (same as above)
|
|
if int(currentDeltaIndex) >= len(deltas) {
|
|
return nil, nil, errInvalidNodeDelta
|
|
}
|
|
deltas[currentDeltaIndex] -= int(oldBalance)
|
|
}
|
|
}
|
|
|
|
// Rotate the validator vote.
|
|
vote.currentRoot = vote.nextRoot
|
|
votes[validatorIndex] = vote
|
|
}
|
|
|
|
return deltas, votes, nil
|
|
}
|
|
|
|
// This return a copy of the proto array node object.
|
|
func copyNode(node *Node) *Node {
|
|
if node == nil {
|
|
return &Node{}
|
|
}
|
|
|
|
copiedRoot := [32]byte{}
|
|
copy(copiedRoot[:], node.root[:])
|
|
|
|
return &Node{
|
|
slot: node.slot,
|
|
root: copiedRoot,
|
|
parent: node.parent,
|
|
justifiedEpoch: node.justifiedEpoch,
|
|
finalizedEpoch: node.finalizedEpoch,
|
|
weight: node.weight,
|
|
bestChild: node.bestChild,
|
|
bestDescendant: node.bestDescendant,
|
|
}
|
|
}
|