mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
61 lines
2.0 KiB
Go
61 lines
2.0 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db/filters"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
)
|
|
|
|
// VerifyWeakSubjectivityRoot verifies the weak subjectivity root in the service struct.
|
|
// Reference design: https://github.com/ethereum/eth2.0-specs/blob/master/specs/phase0/weak-subjectivity.md#weak-subjectivity-sync-procedure
|
|
func (s *Service) VerifyWeakSubjectivityRoot(ctx context.Context) error {
|
|
// TODO(7342): Remove the following to fully use weak subjectivity in production.
|
|
if len(s.cfg.WspBlockRoot) == 0 || s.cfg.WspEpoch == 0 {
|
|
return nil
|
|
}
|
|
|
|
// Do nothing if the weak subjectivity has previously been verified,
|
|
// or weak subjectivity epoch is higher than last finalized epoch.
|
|
if s.wsVerified {
|
|
return nil
|
|
}
|
|
if s.cfg.WspEpoch > s.finalizedCheckpt.Epoch {
|
|
return nil
|
|
}
|
|
|
|
r := bytesutil.ToBytes32(s.cfg.WspBlockRoot)
|
|
log.Infof("Performing weak subjectivity check for root %#x in epoch %d", r, s.cfg.WspEpoch)
|
|
// Save initial sync cached blocks to DB.
|
|
if err := s.cfg.BeaconDB.SaveBlocks(ctx, s.getInitSyncBlocks()); err != nil {
|
|
return err
|
|
}
|
|
// A node should have the weak subjectivity block in the DB.
|
|
if !s.cfg.BeaconDB.HasBlock(ctx, r) {
|
|
return fmt.Errorf("node does not have root in DB: %#x", r)
|
|
}
|
|
|
|
startSlot, err := helpers.StartSlot(s.cfg.WspEpoch)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// A node should have the weak subjectivity block corresponds to the correct epoch in the DB.
|
|
filter := filters.NewFilter().SetStartSlot(startSlot).SetEndSlot(startSlot + params.BeaconConfig().SlotsPerEpoch)
|
|
roots, err := s.cfg.BeaconDB.BlockRoots(ctx, filter)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, root := range roots {
|
|
if r == root {
|
|
log.Info("Weak subjectivity check has passed")
|
|
s.wsVerified = true
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return fmt.Errorf("node does not have root in db corresponding to epoch: %#x %d", r, s.cfg.WspEpoch)
|
|
}
|