571 lines
14 KiB
Go
571 lines
14 KiB
Go
|
package miner
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"errors"
|
||
|
|
||
|
"github.com/filecoin-project/go-address"
|
||
|
"github.com/filecoin-project/go-bitfield"
|
||
|
rle "github.com/filecoin-project/go-bitfield/rle"
|
||
|
"github.com/filecoin-project/go-state-types/abi"
|
||
|
"github.com/filecoin-project/go-state-types/dline"
|
||
|
"github.com/ipfs/go-cid"
|
||
|
"github.com/libp2p/go-libp2p-core/peer"
|
||
|
cbg "github.com/whyrusleeping/cbor-gen"
|
||
|
"golang.org/x/xerrors"
|
||
|
|
||
|
"github.com/filecoin-project/lotus/chain/actors/adt"
|
||
|
|
||
|
builtin7 "github.com/filecoin-project/specs-actors/v7/actors/builtin"
|
||
|
|
||
|
miner7 "github.com/filecoin-project/specs-actors/v7/actors/builtin/miner"
|
||
|
adt7 "github.com/filecoin-project/specs-actors/v7/actors/util/adt"
|
||
|
)
|
||
|
|
||
|
var _ State = (*state7)(nil)
|
||
|
|
||
|
func load7(store adt.Store, root cid.Cid) (State, error) {
|
||
|
out := state7{store: store}
|
||
|
err := store.Get(store.Context(), root, &out)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return &out, nil
|
||
|
}
|
||
|
|
||
|
func make7(store adt.Store) (State, error) {
|
||
|
out := state7{store: store}
|
||
|
out.State = miner7.State{}
|
||
|
return &out, nil
|
||
|
}
|
||
|
|
||
|
type state7 struct {
|
||
|
miner7.State
|
||
|
store adt.Store
|
||
|
}
|
||
|
|
||
|
type deadline7 struct {
|
||
|
miner7.Deadline
|
||
|
store adt.Store
|
||
|
}
|
||
|
|
||
|
type partition7 struct {
|
||
|
miner7.Partition
|
||
|
store adt.Store
|
||
|
}
|
||
|
|
||
|
func (s *state7) AvailableBalance(bal abi.TokenAmount) (available abi.TokenAmount, err error) {
|
||
|
defer func() {
|
||
|
if r := recover(); r != nil {
|
||
|
err = xerrors.Errorf("failed to get available balance: %w", r)
|
||
|
available = abi.NewTokenAmount(0)
|
||
|
}
|
||
|
}()
|
||
|
// this panics if the miner doesnt have enough funds to cover their locked pledge
|
||
|
available, err = s.GetAvailableBalance(bal)
|
||
|
return available, err
|
||
|
}
|
||
|
|
||
|
func (s *state7) VestedFunds(epoch abi.ChainEpoch) (abi.TokenAmount, error) {
|
||
|
return s.CheckVestedFunds(s.store, epoch)
|
||
|
}
|
||
|
|
||
|
func (s *state7) LockedFunds() (LockedFunds, error) {
|
||
|
return LockedFunds{
|
||
|
VestingFunds: s.State.LockedFunds,
|
||
|
InitialPledgeRequirement: s.State.InitialPledge,
|
||
|
PreCommitDeposits: s.State.PreCommitDeposits,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) FeeDebt() (abi.TokenAmount, error) {
|
||
|
return s.State.FeeDebt, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) InitialPledge() (abi.TokenAmount, error) {
|
||
|
return s.State.InitialPledge, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) PreCommitDeposits() (abi.TokenAmount, error) {
|
||
|
return s.State.PreCommitDeposits, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) GetSector(num abi.SectorNumber) (*SectorOnChainInfo, error) {
|
||
|
info, ok, err := s.State.GetSector(s.store, num)
|
||
|
if !ok || err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
ret := fromV7SectorOnChainInfo(*info)
|
||
|
return &ret, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) FindSector(num abi.SectorNumber) (*SectorLocation, error) {
|
||
|
dlIdx, partIdx, err := s.State.FindSector(s.store, num)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return &SectorLocation{
|
||
|
Deadline: dlIdx,
|
||
|
Partition: partIdx,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) NumLiveSectors() (uint64, error) {
|
||
|
dls, err := s.State.LoadDeadlines(s.store)
|
||
|
if err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
var total uint64
|
||
|
if err := dls.ForEach(s.store, func(dlIdx uint64, dl *miner7.Deadline) error {
|
||
|
total += dl.LiveSectors
|
||
|
return nil
|
||
|
}); err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
return total, nil
|
||
|
}
|
||
|
|
||
|
// GetSectorExpiration returns the effective expiration of the given sector.
|
||
|
//
|
||
|
// If the sector does not expire early, the Early expiration field is 0.
|
||
|
func (s *state7) GetSectorExpiration(num abi.SectorNumber) (*SectorExpiration, error) {
|
||
|
dls, err := s.State.LoadDeadlines(s.store)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
// NOTE: this can be optimized significantly.
|
||
|
// 1. If the sector is non-faulty, it will either expire on-time (can be
|
||
|
// learned from the sector info), or in the next quantized expiration
|
||
|
// epoch (i.e., the first element in the partition's expiration queue.
|
||
|
// 2. If it's faulty, it will expire early within the first 14 entries
|
||
|
// of the expiration queue.
|
||
|
stopErr := errors.New("stop")
|
||
|
out := SectorExpiration{}
|
||
|
err = dls.ForEach(s.store, func(dlIdx uint64, dl *miner7.Deadline) error {
|
||
|
partitions, err := dl.PartitionsArray(s.store)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
quant := s.State.QuantSpecForDeadline(dlIdx)
|
||
|
var part miner7.Partition
|
||
|
return partitions.ForEach(&part, func(partIdx int64) error {
|
||
|
if found, err := part.Sectors.IsSet(uint64(num)); err != nil {
|
||
|
return err
|
||
|
} else if !found {
|
||
|
return nil
|
||
|
}
|
||
|
if found, err := part.Terminated.IsSet(uint64(num)); err != nil {
|
||
|
return err
|
||
|
} else if found {
|
||
|
// already terminated
|
||
|
return stopErr
|
||
|
}
|
||
|
|
||
|
q, err := miner7.LoadExpirationQueue(s.store, part.ExpirationsEpochs, quant, miner7.PartitionExpirationAmtBitwidth)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
var exp miner7.ExpirationSet
|
||
|
return q.ForEach(&exp, func(epoch int64) error {
|
||
|
if early, err := exp.EarlySectors.IsSet(uint64(num)); err != nil {
|
||
|
return err
|
||
|
} else if early {
|
||
|
out.Early = abi.ChainEpoch(epoch)
|
||
|
return nil
|
||
|
}
|
||
|
if onTime, err := exp.OnTimeSectors.IsSet(uint64(num)); err != nil {
|
||
|
return err
|
||
|
} else if onTime {
|
||
|
out.OnTime = abi.ChainEpoch(epoch)
|
||
|
return stopErr
|
||
|
}
|
||
|
return nil
|
||
|
})
|
||
|
})
|
||
|
})
|
||
|
if err == stopErr {
|
||
|
err = nil
|
||
|
}
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
if out.Early == 0 && out.OnTime == 0 {
|
||
|
return nil, xerrors.Errorf("failed to find sector %d", num)
|
||
|
}
|
||
|
return &out, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) GetPrecommittedSector(num abi.SectorNumber) (*SectorPreCommitOnChainInfo, error) {
|
||
|
info, ok, err := s.State.GetPrecommittedSector(s.store, num)
|
||
|
if !ok || err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
ret := fromV7SectorPreCommitOnChainInfo(*info)
|
||
|
|
||
|
return &ret, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) ForEachPrecommittedSector(cb func(SectorPreCommitOnChainInfo) error) error {
|
||
|
precommitted, err := adt7.AsMap(s.store, s.State.PreCommittedSectors, builtin7.DefaultHamtBitwidth)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
var info miner7.SectorPreCommitOnChainInfo
|
||
|
if err := precommitted.ForEach(&info, func(_ string) error {
|
||
|
return cb(fromV7SectorPreCommitOnChainInfo(info))
|
||
|
}); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) LoadSectors(snos *bitfield.BitField) ([]*SectorOnChainInfo, error) {
|
||
|
sectors, err := miner7.LoadSectors(s.store, s.State.Sectors)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
// If no sector numbers are specified, load all.
|
||
|
if snos == nil {
|
||
|
infos := make([]*SectorOnChainInfo, 0, sectors.Length())
|
||
|
var info7 miner7.SectorOnChainInfo
|
||
|
if err := sectors.ForEach(&info7, func(_ int64) error {
|
||
|
info := fromV7SectorOnChainInfo(info7)
|
||
|
infos = append(infos, &info)
|
||
|
return nil
|
||
|
}); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return infos, nil
|
||
|
}
|
||
|
|
||
|
// Otherwise, load selected.
|
||
|
infos7, err := sectors.Load(*snos)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
infos := make([]*SectorOnChainInfo, len(infos7))
|
||
|
for i, info7 := range infos7 {
|
||
|
info := fromV7SectorOnChainInfo(*info7)
|
||
|
infos[i] = &info
|
||
|
}
|
||
|
return infos, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) loadAllocatedSectorNumbers() (bitfield.BitField, error) {
|
||
|
var allocatedSectors bitfield.BitField
|
||
|
err := s.store.Get(s.store.Context(), s.State.AllocatedSectors, &allocatedSectors)
|
||
|
return allocatedSectors, err
|
||
|
}
|
||
|
|
||
|
func (s *state7) IsAllocated(num abi.SectorNumber) (bool, error) {
|
||
|
allocatedSectors, err := s.loadAllocatedSectorNumbers()
|
||
|
if err != nil {
|
||
|
return false, err
|
||
|
}
|
||
|
|
||
|
return allocatedSectors.IsSet(uint64(num))
|
||
|
}
|
||
|
|
||
|
func (s *state7) GetProvingPeriodStart() (abi.ChainEpoch, error) {
|
||
|
return s.State.ProvingPeriodStart, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) UnallocatedSectorNumbers(count int) ([]abi.SectorNumber, error) {
|
||
|
allocatedSectors, err := s.loadAllocatedSectorNumbers()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
allocatedRuns, err := allocatedSectors.RunIterator()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
unallocatedRuns, err := rle.Subtract(
|
||
|
&rle.RunSliceIterator{Runs: []rle.Run{{Val: true, Len: abi.MaxSectorNumber}}},
|
||
|
allocatedRuns,
|
||
|
)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
iter, err := rle.BitsFromRuns(unallocatedRuns)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
sectors := make([]abi.SectorNumber, 0, count)
|
||
|
for iter.HasNext() && len(sectors) < count {
|
||
|
nextNo, err := iter.Next()
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
sectors = append(sectors, abi.SectorNumber(nextNo))
|
||
|
}
|
||
|
|
||
|
return sectors, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) GetAllocatedSectors() (*bitfield.BitField, error) {
|
||
|
var allocatedSectors bitfield.BitField
|
||
|
if err := s.store.Get(s.store.Context(), s.State.AllocatedSectors, &allocatedSectors); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return &allocatedSectors, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) LoadDeadline(idx uint64) (Deadline, error) {
|
||
|
dls, err := s.State.LoadDeadlines(s.store)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
dl, err := dls.LoadDeadline(s.store, idx)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return &deadline7{*dl, s.store}, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) ForEachDeadline(cb func(uint64, Deadline) error) error {
|
||
|
dls, err := s.State.LoadDeadlines(s.store)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
return dls.ForEach(s.store, func(i uint64, dl *miner7.Deadline) error {
|
||
|
return cb(i, &deadline7{*dl, s.store})
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (s *state7) NumDeadlines() (uint64, error) {
|
||
|
return miner7.WPoStPeriodDeadlines, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) DeadlinesChanged(other State) (bool, error) {
|
||
|
other7, ok := other.(*state7)
|
||
|
if !ok {
|
||
|
// treat an upgrade as a change, always
|
||
|
return true, nil
|
||
|
}
|
||
|
|
||
|
return !s.State.Deadlines.Equals(other7.Deadlines), nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) MinerInfoChanged(other State) (bool, error) {
|
||
|
other0, ok := other.(*state7)
|
||
|
if !ok {
|
||
|
// treat an upgrade as a change, always
|
||
|
return true, nil
|
||
|
}
|
||
|
return !s.State.Info.Equals(other0.State.Info), nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) Info() (MinerInfo, error) {
|
||
|
info, err := s.State.GetInfo(s.store)
|
||
|
if err != nil {
|
||
|
return MinerInfo{}, err
|
||
|
}
|
||
|
|
||
|
var pid *peer.ID
|
||
|
if peerID, err := peer.IDFromBytes(info.PeerId); err == nil {
|
||
|
pid = &peerID
|
||
|
}
|
||
|
|
||
|
mi := MinerInfo{
|
||
|
Owner: info.Owner,
|
||
|
Worker: info.Worker,
|
||
|
ControlAddresses: info.ControlAddresses,
|
||
|
|
||
|
NewWorker: address.Undef,
|
||
|
WorkerChangeEpoch: -1,
|
||
|
|
||
|
PeerId: pid,
|
||
|
Multiaddrs: info.Multiaddrs,
|
||
|
WindowPoStProofType: info.WindowPoStProofType,
|
||
|
SectorSize: info.SectorSize,
|
||
|
WindowPoStPartitionSectors: info.WindowPoStPartitionSectors,
|
||
|
ConsensusFaultElapsed: info.ConsensusFaultElapsed,
|
||
|
}
|
||
|
|
||
|
if info.PendingWorkerKey != nil {
|
||
|
mi.NewWorker = info.PendingWorkerKey.NewWorker
|
||
|
mi.WorkerChangeEpoch = info.PendingWorkerKey.EffectiveAt
|
||
|
}
|
||
|
|
||
|
return mi, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) DeadlineInfo(epoch abi.ChainEpoch) (*dline.Info, error) {
|
||
|
return s.State.RecordedDeadlineInfo(epoch), nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) DeadlineCronActive() (bool, error) {
|
||
|
return s.State.DeadlineCronActive, nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) sectors() (adt.Array, error) {
|
||
|
return adt7.AsArray(s.store, s.Sectors, miner7.SectorsAmtBitwidth)
|
||
|
}
|
||
|
|
||
|
func (s *state7) decodeSectorOnChainInfo(val *cbg.Deferred) (SectorOnChainInfo, error) {
|
||
|
var si miner7.SectorOnChainInfo
|
||
|
err := si.UnmarshalCBOR(bytes.NewReader(val.Raw))
|
||
|
if err != nil {
|
||
|
return SectorOnChainInfo{}, err
|
||
|
}
|
||
|
|
||
|
return fromV7SectorOnChainInfo(si), nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) precommits() (adt.Map, error) {
|
||
|
return adt7.AsMap(s.store, s.PreCommittedSectors, builtin7.DefaultHamtBitwidth)
|
||
|
}
|
||
|
|
||
|
func (s *state7) decodeSectorPreCommitOnChainInfo(val *cbg.Deferred) (SectorPreCommitOnChainInfo, error) {
|
||
|
var sp miner7.SectorPreCommitOnChainInfo
|
||
|
err := sp.UnmarshalCBOR(bytes.NewReader(val.Raw))
|
||
|
if err != nil {
|
||
|
return SectorPreCommitOnChainInfo{}, err
|
||
|
}
|
||
|
|
||
|
return fromV7SectorPreCommitOnChainInfo(sp), nil
|
||
|
}
|
||
|
|
||
|
func (s *state7) EraseAllUnproven() error {
|
||
|
|
||
|
dls, err := s.State.LoadDeadlines(s.store)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
err = dls.ForEach(s.store, func(dindx uint64, dl *miner7.Deadline) error {
|
||
|
ps, err := dl.PartitionsArray(s.store)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
var part miner7.Partition
|
||
|
err = ps.ForEach(&part, func(pindx int64) error {
|
||
|
_ = part.ActivateUnproven()
|
||
|
err = ps.Set(uint64(pindx), &part)
|
||
|
return nil
|
||
|
})
|
||
|
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
dl.Partitions, err = ps.Root()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return dls.UpdateDeadline(s.store, dindx, dl)
|
||
|
})
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return s.State.SaveDeadlines(s.store, dls)
|
||
|
|
||
|
}
|
||
|
|
||
|
func (d *deadline7) LoadPartition(idx uint64) (Partition, error) {
|
||
|
p, err := d.Deadline.LoadPartition(d.store, idx)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return &partition7{*p, d.store}, nil
|
||
|
}
|
||
|
|
||
|
func (d *deadline7) ForEachPartition(cb func(uint64, Partition) error) error {
|
||
|
ps, err := d.Deadline.PartitionsArray(d.store)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
var part miner7.Partition
|
||
|
return ps.ForEach(&part, func(i int64) error {
|
||
|
return cb(uint64(i), &partition7{part, d.store})
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (d *deadline7) PartitionsChanged(other Deadline) (bool, error) {
|
||
|
other7, ok := other.(*deadline7)
|
||
|
if !ok {
|
||
|
// treat an upgrade as a change, always
|
||
|
return true, nil
|
||
|
}
|
||
|
|
||
|
return !d.Deadline.Partitions.Equals(other7.Deadline.Partitions), nil
|
||
|
}
|
||
|
|
||
|
func (d *deadline7) PartitionsPoSted() (bitfield.BitField, error) {
|
||
|
return d.Deadline.PartitionsPoSted, nil
|
||
|
}
|
||
|
|
||
|
func (d *deadline7) DisputableProofCount() (uint64, error) {
|
||
|
|
||
|
ops, err := d.OptimisticProofsSnapshotArray(d.store)
|
||
|
if err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
|
||
|
return ops.Length(), nil
|
||
|
|
||
|
}
|
||
|
|
||
|
func (p *partition7) AllSectors() (bitfield.BitField, error) {
|
||
|
return p.Partition.Sectors, nil
|
||
|
}
|
||
|
|
||
|
func (p *partition7) FaultySectors() (bitfield.BitField, error) {
|
||
|
return p.Partition.Faults, nil
|
||
|
}
|
||
|
|
||
|
func (p *partition7) RecoveringSectors() (bitfield.BitField, error) {
|
||
|
return p.Partition.Recoveries, nil
|
||
|
}
|
||
|
|
||
|
func (p *partition7) UnprovenSectors() (bitfield.BitField, error) {
|
||
|
return p.Partition.Unproven, nil
|
||
|
}
|
||
|
|
||
|
func fromV7SectorOnChainInfo(v7 miner7.SectorOnChainInfo) SectorOnChainInfo {
|
||
|
|
||
|
return SectorOnChainInfo{
|
||
|
SectorNumber: v7.SectorNumber,
|
||
|
SealProof: v7.SealProof,
|
||
|
SealedCID: v7.SealedCID,
|
||
|
DealIDs: v7.DealIDs,
|
||
|
Activation: v7.Activation,
|
||
|
Expiration: v7.Expiration,
|
||
|
DealWeight: v7.DealWeight,
|
||
|
VerifiedDealWeight: v7.VerifiedDealWeight,
|
||
|
InitialPledge: v7.InitialPledge,
|
||
|
ExpectedDayReward: v7.ExpectedDayReward,
|
||
|
ExpectedStoragePledge: v7.ExpectedStoragePledge,
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
func fromV7SectorPreCommitOnChainInfo(v7 miner7.SectorPreCommitOnChainInfo) SectorPreCommitOnChainInfo {
|
||
|
|
||
|
return SectorPreCommitOnChainInfo{
|
||
|
Info: (SectorPreCommitInfo)(v7.Info),
|
||
|
PreCommitDeposit: v7.PreCommitDeposit,
|
||
|
PreCommitEpoch: v7.PreCommitEpoch,
|
||
|
DealWeight: v7.DealWeight,
|
||
|
VerifiedDealWeight: v7.VerifiedDealWeight,
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
func (s *state7) GetState() interface{} {
|
||
|
return &s.State
|
||
|
}
|