471 lines
14 KiB
Go
471 lines
14 KiB
Go
package stmgr
|
|
|
|
import (
|
|
"context"
|
|
amt "github.com/filecoin-project/go-amt-ipld/v2"
|
|
"github.com/filecoin-project/sector-storage/ffiwrapper"
|
|
"github.com/filecoin-project/specs-actors/actors/abi"
|
|
"github.com/filecoin-project/specs-actors/actors/abi/big"
|
|
"github.com/filecoin-project/specs-actors/actors/builtin"
|
|
init_ "github.com/filecoin-project/specs-actors/actors/builtin/init"
|
|
"github.com/filecoin-project/specs-actors/actors/builtin/market"
|
|
"github.com/filecoin-project/specs-actors/actors/builtin/miner"
|
|
"github.com/filecoin-project/specs-actors/actors/builtin/power"
|
|
"github.com/filecoin-project/specs-actors/actors/crypto"
|
|
"github.com/filecoin-project/specs-actors/actors/util/adt"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/chain/state"
|
|
"github.com/filecoin-project/lotus/chain/store"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/vm"
|
|
"github.com/filecoin-project/lotus/node/modules/dtypes"
|
|
|
|
cid "github.com/ipfs/go-cid"
|
|
blockstore "github.com/ipfs/go-ipfs-blockstore"
|
|
cbor "github.com/ipfs/go-ipld-cbor"
|
|
cbg "github.com/whyrusleeping/cbor-gen"
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
func GetNetworkName(ctx context.Context, sm *StateManager, st cid.Cid) (dtypes.NetworkName, error) {
|
|
var state init_.State
|
|
_, err := sm.LoadActorStateRaw(ctx, builtin.InitActorAddr, &state, st)
|
|
if err != nil {
|
|
return "", xerrors.Errorf("(get sset) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
return dtypes.NetworkName(state.NetworkName), nil
|
|
}
|
|
|
|
func GetMinerWorkerRaw(ctx context.Context, sm *StateManager, st cid.Cid, maddr address.Address) (address.Address, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorStateRaw(ctx, maddr, &mas, st)
|
|
if err != nil {
|
|
return address.Undef, xerrors.Errorf("(get sset) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
cst := cbor.NewCborStore(sm.cs.Blockstore())
|
|
state, err := state.LoadStateTree(cst, st)
|
|
if err != nil {
|
|
return address.Undef, xerrors.Errorf("load state tree: %w", err)
|
|
}
|
|
|
|
return vm.ResolveToKeyAddr(state, cst, mas.Info.Worker)
|
|
}
|
|
|
|
func GetPower(ctx context.Context, sm *StateManager, ts *types.TipSet, maddr address.Address) (types.BigInt, types.BigInt, error) {
|
|
return getPowerRaw(ctx, sm, ts.ParentState(), maddr)
|
|
}
|
|
|
|
func getPowerRaw(ctx context.Context, sm *StateManager, st cid.Cid, maddr address.Address) (types.BigInt, types.BigInt, error) {
|
|
var ps power.State
|
|
_, err := sm.LoadActorStateRaw(ctx, builtin.StoragePowerActorAddr, &ps, st)
|
|
if err != nil {
|
|
return big.Zero(), big.Zero(), xerrors.Errorf("(get sset) failed to load power actor state: %w", err)
|
|
}
|
|
|
|
var mpow big.Int
|
|
if maddr != address.Undef {
|
|
cm, err := adt.AsMap(sm.cs.Store(ctx), ps.Claims)
|
|
if err != nil {
|
|
return types.BigInt{}, types.BigInt{}, err
|
|
}
|
|
|
|
var claim power.Claim
|
|
if _, err := cm.Get(adt.AddrKey(maddr), &claim); err != nil {
|
|
return big.Zero(), big.Zero(), err
|
|
}
|
|
|
|
mpow = claim.QualityAdjPower // TODO: is quality adjusted power what we want here?
|
|
}
|
|
|
|
return mpow, ps.TotalQualityAdjPower, nil
|
|
}
|
|
|
|
func SectorSetSizes(ctx context.Context, sm *StateManager, maddr address.Address, ts *types.TipSet) (api.MinerSectors, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorState(ctx, maddr, &mas, ts)
|
|
if err != nil {
|
|
return api.MinerSectors{}, xerrors.Errorf("(get sset) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
blks := cbor.NewCborStore(sm.ChainStore().Blockstore())
|
|
ss, err := amt.LoadAMT(ctx, blks, mas.Sectors)
|
|
if err != nil {
|
|
return api.MinerSectors{}, err
|
|
}
|
|
|
|
return api.MinerSectors{
|
|
Sset: ss.Count,
|
|
}, nil
|
|
}
|
|
|
|
func PreCommitInfo(ctx context.Context, sm *StateManager, maddr address.Address, sid abi.SectorNumber, ts *types.TipSet) (miner.SectorPreCommitOnChainInfo, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorState(ctx, maddr, &mas, ts)
|
|
if err != nil {
|
|
return miner.SectorPreCommitOnChainInfo{}, xerrors.Errorf("(get sset) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
i, ok, err := mas.GetPrecommittedSector(sm.cs.Store(ctx), sid)
|
|
if err != nil {
|
|
return miner.SectorPreCommitOnChainInfo{}, err
|
|
}
|
|
if !ok {
|
|
return miner.SectorPreCommitOnChainInfo{}, xerrors.New("precommit not found")
|
|
}
|
|
|
|
return *i, nil
|
|
}
|
|
|
|
func GetMinerSectorSet(ctx context.Context, sm *StateManager, ts *types.TipSet, maddr address.Address, filter *abi.BitField) ([]*api.ChainSectorInfo, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorState(ctx, maddr, &mas, ts)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("(get sset) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
return LoadSectorsFromSet(ctx, sm.ChainStore().Blockstore(), mas.Sectors, filter)
|
|
}
|
|
|
|
func GetSectorsForWinningPoSt(ctx context.Context, pv ffiwrapper.Verifier, sm *StateManager, ts *types.TipSet, maddr address.Address) ([]abi.SectorInfo, error) {
|
|
pts, err := sm.cs.LoadTipSet(ts.Parents()) // TODO: Review: check correct lookback for winningPost sector set
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("loading parent tipset: %w", err)
|
|
}
|
|
|
|
var mas miner.State
|
|
_, err = sm.LoadActorStateRaw(ctx, maddr, &mas, pts.ParentState())
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("(get ssize) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
// TODO: Optimization: we could avoid loaditg the whole proving set here if we had AMT.GetNth with bitfield filtering
|
|
sectorSet, err := GetProvingSetRaw(ctx, sm, mas)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("getting proving set: %w", err)
|
|
}
|
|
|
|
spt, err := ffiwrapper.SealProofTypeFromSectorSize(mas.Info.SectorSize)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("getting seal proof type: %w", err)
|
|
}
|
|
|
|
wpt, err := spt.RegisteredWinningPoStProof()
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("getting window proof type: %w", err)
|
|
}
|
|
|
|
mid, err := address.IDFromAddress(maddr)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("getting miner ID: %w", err)
|
|
}
|
|
|
|
// TODO: use the right dst, also NB: not using any 'entropy' in this call because nicola really didnt want it
|
|
rand, err := sm.cs.GetRandomness(ctx, ts.Cids(), crypto.DomainSeparationTag_ElectionPoStChallengeSeed, ts.Height() - 1, nil)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to get randomness for winning post: %w", err)
|
|
}
|
|
|
|
ids, err := pv.GenerateWinningPoStSectorChallenge(ctx, wpt, abi.ActorID(mid), rand, uint64(len(sectorSet)))
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("generating winning post challenges: %w", err)
|
|
}
|
|
|
|
out := make([]abi.SectorInfo, len(ids))
|
|
for i, n := range ids {
|
|
out[i] = abi.SectorInfo{
|
|
RegisteredProof: wpt,
|
|
SectorNumber: sectorSet[n].ID,
|
|
SealedCID: sectorSet[n].Info.Info.SealedCID,
|
|
}
|
|
}
|
|
|
|
return out, nil
|
|
}
|
|
|
|
func StateMinerInfo(ctx context.Context, sm *StateManager, ts *types.TipSet, maddr address.Address) (miner.MinerInfo, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorStateRaw(ctx, maddr, &mas, ts.ParentState())
|
|
if err != nil {
|
|
return miner.MinerInfo{}, xerrors.Errorf("(get ssize) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
return mas.Info, nil
|
|
}
|
|
|
|
func GetMinerSlashed(ctx context.Context, sm *StateManager, ts *types.TipSet, maddr address.Address) (bool, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorState(ctx, maddr, &mas, ts)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("(get miner slashed) failed to load miner actor state")
|
|
}
|
|
|
|
var spas power.State
|
|
_, err = sm.LoadActorState(ctx, builtin.StoragePowerActorAddr, &spas, ts)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("(get miner slashed) failed to load power actor state")
|
|
}
|
|
|
|
store := sm.cs.Store(ctx)
|
|
|
|
{
|
|
claims, err := adt.AsMap(store, spas.Claims)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
ok, err := claims.Get(power.AddrKey(maddr), nil)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if !ok {
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
{
|
|
detectedFaulty, err := adt.AsMap(store, spas.PoStDetectedFaultMiners)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
ok, err := detectedFaulty.Get(power.AddrKey(maddr), nil)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if ok {
|
|
return true, nil
|
|
}
|
|
}
|
|
|
|
return false, nil
|
|
}
|
|
|
|
func GetMinerDeadlines(ctx context.Context, sm *StateManager, ts *types.TipSet, maddr address.Address) (*miner.Deadlines, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorState(ctx, maddr, &mas, ts)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("(get ssize) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
return miner.LoadDeadlines(sm.cs.Store(ctx), &mas)
|
|
}
|
|
|
|
func GetMinerFaults(ctx context.Context, sm *StateManager, ts *types.TipSet, maddr address.Address) ([]abi.SectorNumber, error) {
|
|
var mas miner.State
|
|
_, err := sm.LoadActorState(ctx, maddr, &mas, ts)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("(get ssize) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
faults, err := mas.Faults.All(miner.SectorsMax)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("reading fault bit set: %w", err)
|
|
}
|
|
|
|
out := make([]abi.SectorNumber, len(faults))
|
|
for i, fault := range faults {
|
|
out[i] = abi.SectorNumber(fault)
|
|
}
|
|
|
|
return out, nil
|
|
}
|
|
|
|
func GetStorageDeal(ctx context.Context, sm *StateManager, dealId abi.DealID, ts *types.TipSet) (*api.MarketDeal, error) {
|
|
var state market.State
|
|
if _, err := sm.LoadActorState(ctx, builtin.StorageMarketActorAddr, &state, ts); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
da, err := amt.LoadAMT(ctx, cbor.NewCborStore(sm.ChainStore().Blockstore()), state.Proposals)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var dp market.DealProposal
|
|
if err := da.Get(ctx, uint64(dealId), &dp); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sa, err := market.AsDealStateArray(sm.ChainStore().Store(ctx), state.States)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
st, err := sa.Get(dealId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &api.MarketDeal{
|
|
Proposal: dp,
|
|
State: *st,
|
|
}, nil
|
|
}
|
|
|
|
func ListMinerActors(ctx context.Context, sm *StateManager, ts *types.TipSet) ([]address.Address, error) {
|
|
var state power.State
|
|
if _, err := sm.LoadActorState(ctx, builtin.StoragePowerActorAddr, &state, ts); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
m, err := adt.AsMap(sm.cs.Store(ctx), state.Claims)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var miners []address.Address
|
|
err = m.ForEach(nil, func(k string) error {
|
|
a, err := address.NewFromBytes([]byte(k))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
miners = append(miners, a)
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return miners, nil
|
|
}
|
|
|
|
func LoadSectorsFromSet(ctx context.Context, bs blockstore.Blockstore, ssc cid.Cid, filter *abi.BitField) ([]*api.ChainSectorInfo, error) {
|
|
a, err := amt.LoadAMT(ctx, cbor.NewCborStore(bs), ssc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var sset []*api.ChainSectorInfo
|
|
if err := a.ForEach(ctx, func(i uint64, v *cbg.Deferred) error {
|
|
if filter != nil {
|
|
set, err := filter.IsSet(i)
|
|
if err != nil {
|
|
return xerrors.Errorf("filter check error: %w", err)
|
|
}
|
|
if set {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
var oci miner.SectorOnChainInfo
|
|
if err := cbor.DecodeInto(v.Raw, &oci); err != nil {
|
|
return err
|
|
}
|
|
sset = append(sset, &api.ChainSectorInfo{
|
|
Info: oci,
|
|
ID: abi.SectorNumber(i),
|
|
})
|
|
return nil
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return sset, nil
|
|
}
|
|
|
|
func ComputeState(ctx context.Context, sm *StateManager, height abi.ChainEpoch, msgs []*types.Message, ts *types.TipSet) (cid.Cid, []*api.InvocResult, error) {
|
|
if ts == nil {
|
|
ts = sm.cs.GetHeaviestTipSet()
|
|
}
|
|
|
|
base, trace, err := sm.ExecutionTrace(ctx, ts)
|
|
if err != nil {
|
|
return cid.Undef, nil, err
|
|
}
|
|
|
|
fstate, err := sm.handleStateForks(ctx, base, height, ts.Height())
|
|
if err != nil {
|
|
return cid.Undef, nil, err
|
|
}
|
|
|
|
r := store.NewChainRand(sm.cs, ts.Cids(), height)
|
|
vmi, err := vm.NewVM(fstate, height, r, sm.cs.Blockstore(), sm.cs.VMSys())
|
|
if err != nil {
|
|
return cid.Undef, nil, err
|
|
}
|
|
|
|
for i, msg := range msgs {
|
|
// TODO: Use the signed message length for secp messages
|
|
ret, err := vmi.ApplyMessage(ctx, msg)
|
|
if err != nil {
|
|
return cid.Undef, nil, xerrors.Errorf("applying message %s: %w", msg.Cid(), err)
|
|
}
|
|
if ret.ExitCode != 0 {
|
|
log.Infof("compute state apply message %d failed (exit: %d): %s", i, ret.ExitCode, ret.ActorErr)
|
|
}
|
|
}
|
|
|
|
root, err := vmi.Flush(ctx)
|
|
if err != nil {
|
|
return cid.Undef, nil, err
|
|
}
|
|
|
|
return root, trace, nil
|
|
}
|
|
|
|
func GetProvingSetRaw(ctx context.Context, sm *StateManager, mas miner.State) ([]*api.ChainSectorInfo, error) {
|
|
notProving, err := abi.BitFieldUnion(mas.Faults, mas.Recoveries)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
provset, err := LoadSectorsFromSet(ctx, sm.cs.Blockstore(), mas.Sectors, notProving)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to get proving set: %w", err)
|
|
}
|
|
|
|
return provset, nil
|
|
}
|
|
|
|
func MinerGetBaseInfo(ctx context.Context, sm *StateManager, tsk types.TipSetKey, maddr address.Address) (*api.MiningBaseInfo, error) {
|
|
ts, err := sm.ChainStore().LoadTipSet(tsk)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to load tipset for mining base: %w", err)
|
|
}
|
|
|
|
st, _, err := sm.TipSetState(ctx, ts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var mas miner.State
|
|
_, err = sm.LoadActorState(ctx, maddr, &mas, ts)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("(get sset) failed to load miner actor state: %w", err)
|
|
}
|
|
|
|
provset, err := GetProvingSetRaw(ctx, sm, mas)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
mpow, tpow, err := getPowerRaw(ctx, sm, st, maddr)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to get power: %w", err)
|
|
}
|
|
|
|
prev, err := sm.ChainStore().GetLatestBeaconEntry(ts)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to get latest beacon entry: %w", err)
|
|
}
|
|
|
|
worker, err := sm.ResolveToKeyAddress(ctx, mas.GetWorker(), ts)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("resolving worker address: %w", err)
|
|
}
|
|
|
|
return &api.MiningBaseInfo{
|
|
MinerPower: mpow,
|
|
NetworkPower: tpow,
|
|
Sectors: provset,
|
|
WorkerKey: worker,
|
|
SectorSize: mas.Info.SectorSize,
|
|
PrevBeaconEntry: *prev,
|
|
}, nil
|
|
}
|