lotus/chain/gen/gen.go

691 lines
18 KiB
Go
Raw Normal View History

2019-07-25 22:15:33 +00:00
package gen
import (
"bytes"
2019-07-25 22:15:33 +00:00
"context"
"crypto/sha256"
"fmt"
2019-11-29 03:38:18 +00:00
"io/ioutil"
2019-07-30 13:20:40 +00:00
"sync/atomic"
2019-07-25 22:15:33 +00:00
2020-02-27 21:45:31 +00:00
ffi "github.com/filecoin-project/filecoin-ffi"
2020-02-23 20:00:47 +00:00
"github.com/filecoin-project/go-address"
2020-02-19 19:26:11 +00:00
commcid "github.com/filecoin-project/go-fil-commcid"
2020-02-23 20:00:47 +00:00
sectorbuilder "github.com/filecoin-project/go-sectorbuilder"
2020-02-08 02:18:32 +00:00
"github.com/filecoin-project/specs-actors/actors/abi"
2020-02-12 23:52:36 +00:00
"github.com/filecoin-project/specs-actors/actors/crypto"
block "github.com/ipfs/go-block-format"
"github.com/ipfs/go-blockservice"
"github.com/ipfs/go-car"
"github.com/ipfs/go-cid"
blockstore "github.com/ipfs/go-ipfs-blockstore"
offline "github.com/ipfs/go-ipfs-exchange-offline"
2020-02-23 20:00:47 +00:00
format "github.com/ipfs/go-ipld-format"
logging "github.com/ipfs/go-log/v2"
"github.com/ipfs/go-merkledag"
"github.com/filecoin-project/lotus/api"
"github.com/filecoin-project/lotus/build"
2020-02-11 20:48:03 +00:00
genesis2 "github.com/filecoin-project/lotus/chain/gen/genesis"
"github.com/filecoin-project/lotus/chain/stmgr"
"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/chain/wallet"
2019-11-29 03:38:18 +00:00
"github.com/filecoin-project/lotus/cmd/lotus-seed/seed"
"github.com/filecoin-project/lotus/genesis"
"github.com/filecoin-project/lotus/lib/sigs"
"github.com/filecoin-project/lotus/node/repo"
"github.com/filecoin-project/lotus/storage/sbmock"
2019-07-26 12:19:27 +00:00
"go.opencensus.io/trace"
"golang.org/x/xerrors"
2019-07-25 22:15:33 +00:00
)
var log = logging.Logger("gen")
2019-10-08 00:28:13 +00:00
const msgsPerBlock = 20
2019-07-30 13:20:40 +00:00
2019-07-25 22:15:33 +00:00
type ChainGen struct {
msgsPerBlock int
bs blockstore.Blockstore
cs *store.ChainStore
sm *stmgr.StateManager
genesis *types.BlockHeader
CurTipset *store.FullTipSet
2019-07-25 22:15:33 +00:00
2020-02-08 02:18:32 +00:00
Timestamper func(*types.TipSet, abi.ChainEpoch) uint64
2019-10-10 03:04:10 +00:00
GetMessages func(*ChainGen) ([]*types.SignedMessage, error)
2019-07-30 13:20:40 +00:00
w *wallet.Wallet
2019-11-21 22:21:45 +00:00
eppProvs map[address.Address]ElectionPoStProver
Miners []address.Address
2019-07-30 13:20:40 +00:00
receivers []address.Address
banker address.Address
bankerNonce uint64
2019-07-29 19:34:09 +00:00
r repo.Repo
2019-07-29 19:34:09 +00:00
lr repo.LockedRepo
2019-07-25 22:15:33 +00:00
}
type mybs struct {
blockstore.Blockstore
}
func (m mybs) Get(c cid.Cid) (block.Block, error) {
b, err := m.Blockstore.Get(c)
if err != nil {
return nil, err
}
return b, nil
}
func NewGenerator() (*ChainGen, error) {
mr := repo.NewMemory(nil)
2019-11-12 17:59:38 +00:00
lr, err := mr.Lock(repo.StorageMiner)
2019-07-25 22:15:33 +00:00
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("taking mem-repo lock failed: %w", err)
2019-07-25 22:15:33 +00:00
}
2019-07-29 19:34:09 +00:00
ds, err := lr.Datastore("/metadata")
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("failed to get metadata datastore: %w", err)
2019-07-29 19:34:09 +00:00
}
bds, err := lr.Datastore("/blocks")
2019-07-25 22:15:33 +00:00
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("failed to get blocks datastore: %w", err)
2019-07-25 22:15:33 +00:00
}
2019-07-29 19:34:09 +00:00
bs := mybs{blockstore.NewIdStore(blockstore.NewBlockstore(bds))}
2019-07-25 22:15:33 +00:00
ks, err := lr.KeyStore()
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("getting repo keystore failed: %w", err)
2019-07-25 22:15:33 +00:00
}
w, err := wallet.NewWallet(ks)
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("creating memrepo wallet failed: %w", err)
2019-07-25 22:15:33 +00:00
}
2020-02-12 23:52:36 +00:00
banker, err := w.GenerateKey(crypto.SigTypeSecp256k1)
2019-07-25 22:15:33 +00:00
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("failed to generate banker key: %w", err)
2019-07-25 22:15:33 +00:00
}
2019-07-30 13:20:40 +00:00
receievers := make([]address.Address, msgsPerBlock)
for r := range receievers {
2020-02-12 23:52:36 +00:00
receievers[r], err = w.GenerateKey(crypto.SigTypeBLS)
2019-07-30 13:20:40 +00:00
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("failed to generate receiver key: %w", err)
2019-07-30 13:20:40 +00:00
}
}
maddr1 := genesis2.MinerAddress(0)
2019-11-29 03:38:18 +00:00
m1temp, err := ioutil.TempDir("", "preseal")
if err != nil {
return nil, err
}
2020-02-27 21:45:31 +00:00
genm1, k1, err := seed.PreSeal(maddr1, abi.RegisteredProof_StackedDRG2KiBPoSt, 0, 1, m1temp, []byte("some randomness"), nil)
2019-11-29 03:38:18 +00:00
if err != nil {
return nil, err
}
maddr2 := genesis2.MinerAddress(1)
m2temp, err := ioutil.TempDir("", "preseal")
if err != nil {
return nil, err
}
2020-02-27 21:45:31 +00:00
genm2, k2, err := seed.PreSeal(maddr2, abi.RegisteredProof_StackedDRG2KiBPoSt, 0, 1, m2temp, []byte("some randomness"), nil)
if err != nil {
return nil, err
}
mk1, err := w.Import(k1)
if err != nil {
return nil, err
}
mk2, err := w.Import(k2)
if err != nil {
return nil, err
}
sys := vm.Syscalls(&genFakeVerifier{})
tpl := genesis.Template{
2020-02-13 00:15:33 +00:00
Accounts: []genesis.Actor{
{
Type: genesis.TAccount,
Balance: types.FromFil(40000),
Meta: (&genesis.AccountMeta{Owner: mk1}).ActorMeta(),
},
{
Type: genesis.TAccount,
Balance: types.FromFil(40000),
Meta: (&genesis.AccountMeta{Owner: mk2}).ActorMeta(),
},
{
Type: genesis.TAccount,
Balance: types.FromFil(50000),
Meta: (&genesis.AccountMeta{Owner: banker}).ActorMeta(),
},
},
2020-02-13 00:15:33 +00:00
Miners: []genesis.Miner{
*genm1,
*genm2,
2019-11-29 03:38:18 +00:00
},
NetworkName: "",
Timestamp: 100000,
2019-09-06 20:03:28 +00:00
}
genb, err := genesis2.MakeGenesisBlock(context.TODO(), bs, sys, tpl)
2019-07-25 22:15:33 +00:00
if err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("make genesis block failed: %w", err)
2019-07-25 22:15:33 +00:00
}
2020-01-13 20:47:27 +00:00
cs := store.NewChainStore(bs, ds, sys)
2019-07-25 22:15:33 +00:00
genfb := &types.FullBlock{Header: genb.Genesis}
gents := store.NewFullTipSet([]*types.FullBlock{genfb})
2019-07-25 22:15:33 +00:00
2019-07-26 12:19:27 +00:00
if err := cs.SetGenesis(genb.Genesis); err != nil {
2019-08-15 18:45:32 +00:00
return nil, xerrors.Errorf("set genesis failed: %w", err)
2019-07-26 12:19:27 +00:00
}
2019-11-21 22:21:45 +00:00
mgen := make(map[address.Address]ElectionPoStProver)
for i := range tpl.Miners {
mgen[genesis2.MinerAddress(uint64(i))] = &eppProvider{}
2019-11-21 22:21:45 +00:00
}
sm := stmgr.NewStateManager(cs)
miners := []address.Address{maddr1, maddr2}
2019-07-25 22:15:33 +00:00
gen := &ChainGen{
bs: bs,
cs: cs,
sm: sm,
2019-07-25 22:15:33 +00:00
msgsPerBlock: msgsPerBlock,
genesis: genb.Genesis,
2019-07-30 13:20:40 +00:00
w: w,
GetMessages: getRandomMessages,
Miners: miners,
eppProvs: mgen,
banker: banker,
receivers: receievers,
2019-07-30 13:20:40 +00:00
CurTipset: gents,
2019-07-29 19:34:09 +00:00
r: mr,
2019-07-29 19:34:09 +00:00
lr: lr,
2019-07-25 22:15:33 +00:00
}
return gen, nil
}
func (cg *ChainGen) SetStateManager(sm *stmgr.StateManager) {
cg.sm = sm
}
func (cg *ChainGen) ChainStore() *store.ChainStore {
return cg.cs
}
2019-07-25 22:15:33 +00:00
func (cg *ChainGen) Genesis() *types.BlockHeader {
return cg.genesis
}
func (cg *ChainGen) GenesisCar() ([]byte, error) {
offl := offline.Exchange(cg.bs)
blkserv := blockservice.New(cg.bs, offl)
dserv := merkledag.NewDAGService(blkserv)
out := new(bytes.Buffer)
2020-02-21 19:27:29 +00:00
if err := car.WriteCarWithWalker(context.TODO(), dserv, []cid.Cid{cg.Genesis().Cid()}, out, CarWalkFunc); err != nil {
2019-11-21 22:21:45 +00:00
return nil, xerrors.Errorf("genesis car write car failed: %w", err)
}
return out.Bytes(), nil
}
2020-02-21 19:27:29 +00:00
func CarWalkFunc(nd format.Node) (out []*format.Link, err error) {
for _, link := range nd.Links() {
if link.Cid.Prefix().MhType == uint64(commcid.FC_SEALED_V1) || link.Cid.Prefix().MhType == uint64(commcid.FC_UNSEALED_V1) {
continue
}
out = append(out, link)
}
return out, nil
}
2019-11-21 22:21:45 +00:00
func (cg *ChainGen) nextBlockProof(ctx context.Context, pts *types.TipSet, m address.Address, round int64) (*types.EPostProof, *types.Ticket, error) {
2020-02-23 20:00:47 +00:00
mc := &mca{w: cg.w, sm: cg.sm}
2020-02-23 20:00:47 +00:00
// TODO: REVIEW: Am I doing this correctly?
ticketRand, err := mc.ChainGetRandomness(ctx, pts.Key(), crypto.DomainSeparationTag_TicketProduction, pts.Height(), m.Bytes())
if err != nil {
return nil, nil, err
}
st := pts.ParentState()
worker, err := stmgr.GetMinerWorkerRaw(ctx, cg.sm, st, m)
if err != nil {
return nil, nil, xerrors.Errorf("get miner worker: %w", err)
}
2019-08-16 04:40:59 +00:00
2020-02-23 20:00:47 +00:00
vrfout, err := ComputeVRF(ctx, cg.w.Sign, worker, ticketRand)
if err != nil {
return nil, nil, xerrors.Errorf("compute VRF: %w", err)
}
2019-08-15 02:30:21 +00:00
tick := &types.Ticket{
2019-10-09 04:38:59 +00:00
VRFProof: vrfout,
2019-08-15 02:30:21 +00:00
}
2020-02-23 20:00:47 +00:00
eproofin, err := IsRoundWinner(ctx, pts, round, m, cg.eppProvs[m], mc)
if err != nil {
return nil, nil, xerrors.Errorf("checking round winner failed: %w", err)
}
2019-12-03 21:27:07 +00:00
if eproofin == nil {
return nil, tick, nil
}
2019-12-03 18:25:56 +00:00
eproof, err := ComputeProof(ctx, cg.eppProvs[m], eproofin)
if err != nil {
return nil, nil, xerrors.Errorf("computing proof: %w", err)
}
return eproof, tick, nil
2019-07-25 22:15:33 +00:00
}
type MinedTipSet struct {
TipSet *store.FullTipSet
Messages []*types.SignedMessage
}
func (cg *ChainGen) NextTipSet() (*MinedTipSet, error) {
mts, err := cg.NextTipSetFromMiners(cg.CurTipset.TipSet(), cg.Miners)
if err != nil {
return nil, err
}
cg.CurTipset = mts.TipSet
return mts, nil
}
func (cg *ChainGen) NextTipSetFromMiners(base *types.TipSet, miners []address.Address) (*MinedTipSet, error) {
var blks []*types.FullBlock
msgs, err := cg.GetMessages(cg)
2019-07-25 22:15:33 +00:00
if err != nil {
return nil, xerrors.Errorf("get random messages: %w", err)
2019-07-25 22:15:33 +00:00
}
for round := int64(base.Height() + 1); len(blks) == 0; round++ {
for _, m := range miners {
proof, t, err := cg.nextBlockProof(context.TODO(), base, m, round)
if err != nil {
return nil, xerrors.Errorf("next block proof: %w", err)
}
if proof != nil {
2020-02-08 02:18:32 +00:00
fblk, err := cg.makeBlock(base, m, proof, t, abi.ChainEpoch(round), msgs)
if err != nil {
return nil, xerrors.Errorf("making a block for next tipset failed: %w", err)
}
2019-11-12 10:18:46 +00:00
if err := cg.cs.PersistBlockHeaders(fblk.Header); err != nil {
return nil, xerrors.Errorf("chainstore AddBlock: %w", err)
}
blks = append(blks, fblk)
}
}
}
fts := store.NewFullTipSet(blks)
return &MinedTipSet{
TipSet: fts,
Messages: msgs,
}, nil
}
2020-02-08 02:18:32 +00:00
func (cg *ChainGen) makeBlock(parents *types.TipSet, m address.Address, eproof *types.EPostProof, ticket *types.Ticket, height abi.ChainEpoch, msgs []*types.SignedMessage) (*types.FullBlock, error) {
2019-10-10 03:04:10 +00:00
var ts uint64
if cg.Timestamper != nil {
ts = cg.Timestamper(parents, height-parents.Height())
2019-10-10 03:04:10 +00:00
} else {
2020-02-08 02:18:37 +00:00
ts = parents.MinTimestamp() + uint64((height-parents.Height())*build.BlockDelay)
2019-10-10 03:04:10 +00:00
}
fblk, err := MinerCreateBlock(context.TODO(), cg.sm, cg.w, m, parents, ticket, eproof, msgs, height, ts)
if err != nil {
return nil, err
}
return fblk, err
}
// This function is awkward. It's used to deal with messages made when
// simulating forks
func (cg *ChainGen) ResyncBankerNonce(ts *types.TipSet) error {
act, err := cg.sm.GetActor(cg.banker, ts)
if err != nil {
return err
}
cg.bankerNonce = act.Nonce
return nil
}
func (cg *ChainGen) Banker() address.Address {
return cg.banker
}
func (cg *ChainGen) Wallet() *wallet.Wallet {
return cg.w
}
func getRandomMessages(cg *ChainGen) ([]*types.SignedMessage, error) {
2019-07-30 13:20:40 +00:00
msgs := make([]*types.SignedMessage, cg.msgsPerBlock)
for m := range msgs {
msg := types.Message{
To: cg.receivers[m%len(cg.receivers)],
2019-07-30 13:20:40 +00:00
From: cg.banker,
Nonce: atomic.AddUint64(&cg.bankerNonce, 1) - 1,
Value: types.NewInt(uint64(m + 1)),
Method: 0,
GasLimit: types.NewInt(10000),
GasPrice: types.NewInt(0),
}
sig, err := cg.w.Sign(context.TODO(), cg.banker, msg.Cid().Bytes())
2019-07-30 13:20:40 +00:00
if err != nil {
return nil, err
2019-07-30 13:20:40 +00:00
}
msgs[m] = &types.SignedMessage{
Message: msg,
Signature: *sig,
}
}
return msgs, nil
2019-07-25 22:15:33 +00:00
}
2019-07-29 19:34:09 +00:00
func (cg *ChainGen) YieldRepo() (repo.Repo, error) {
if err := cg.lr.Close(); err != nil {
return nil, err
}
return cg.r, nil
}
type MiningCheckAPI interface {
2020-02-23 20:00:47 +00:00
ChainGetRandomness(ctx context.Context, tsk types.TipSetKey, personalization crypto.DomainSeparationTag, randEpoch abi.ChainEpoch, entropy []byte) (abi.Randomness, error)
StateMinerPower(context.Context, address.Address, types.TipSetKey) (api.MinerPower, error)
StateMinerWorker(context.Context, address.Address, types.TipSetKey) (address.Address, error)
StateMinerSectorSize(context.Context, address.Address, types.TipSetKey) (abi.SectorSize, error)
2019-11-21 22:21:45 +00:00
StateMinerProvingSet(context.Context, address.Address, types.TipSetKey) ([]*api.ChainSectorInfo, error)
2019-11-25 04:45:13 +00:00
2020-02-12 23:52:36 +00:00
WalletSign(context.Context, address.Address, []byte) (*crypto.Signature, error)
}
type mca struct {
w *wallet.Wallet
sm *stmgr.StateManager
}
2020-02-23 20:00:47 +00:00
func (mca mca) ChainGetRandomness(ctx context.Context, tsk types.TipSetKey, personalization crypto.DomainSeparationTag, randEpoch abi.ChainEpoch, entropy []byte) (abi.Randomness, error) {
pts, err := mca.sm.ChainStore().LoadTipSet(tsk)
if err != nil {
return nil, xerrors.Errorf("loading tipset key: %w", err)
}
return mca.sm.ChainStore().GetRandomness(ctx, pts.Cids(), personalization, int64(randEpoch), entropy)
}
func (mca mca) StateMinerPower(ctx context.Context, maddr address.Address, tsk types.TipSetKey) (api.MinerPower, error) {
ts, err := mca.sm.ChainStore().LoadTipSet(tsk)
if err != nil {
return api.MinerPower{}, xerrors.Errorf("loading tipset %s: %w", tsk, err)
}
2019-09-09 20:03:10 +00:00
mpow, tpow, err := stmgr.GetPower(ctx, mca.sm, ts, maddr)
if err != nil {
return api.MinerPower{}, err
}
return api.MinerPower{
MinerPower: mpow,
TotalPower: tpow,
}, err
}
func (mca mca) StateMinerWorker(ctx context.Context, maddr address.Address, tsk types.TipSetKey) (address.Address, error) {
ts, err := mca.sm.ChainStore().LoadTipSet(tsk)
if err != nil {
return address.Undef, xerrors.Errorf("loading tipset %s: %w", tsk, err)
}
return stmgr.GetMinerWorkerRaw(ctx, mca.sm, ts.ParentState(), maddr)
}
func (mca mca) StateMinerSectorSize(ctx context.Context, maddr address.Address, tsk types.TipSetKey) (abi.SectorSize, error) {
ts, err := mca.sm.ChainStore().LoadTipSet(tsk)
if err != nil {
return 0, xerrors.Errorf("loading tipset %s: %w", tsk, err)
}
2019-11-21 22:21:45 +00:00
return stmgr.GetMinerSectorSize(ctx, mca.sm, ts, maddr)
}
func (mca mca) StateMinerProvingSet(ctx context.Context, maddr address.Address, tsk types.TipSetKey) ([]*api.ChainSectorInfo, error) {
ts, err := mca.sm.ChainStore().LoadTipSet(tsk)
if err != nil {
return nil, xerrors.Errorf("loading tipset %s: %w", tsk, err)
}
2019-11-25 04:45:13 +00:00
return stmgr.GetMinerProvingSet(ctx, mca.sm, ts, maddr)
}
2020-02-12 23:52:36 +00:00
func (mca mca) WalletSign(ctx context.Context, a address.Address, v []byte) (*crypto.Signature, error) {
return mca.w.Sign(ctx, a, v)
}
2019-11-21 22:21:45 +00:00
type ElectionPoStProver interface {
2020-02-27 21:45:31 +00:00
GenerateCandidates(context.Context, []abi.SectorInfo, abi.PoStRandomness) ([]ffi.PoStCandidateWithTicket, error)
ComputeProof(context.Context, []abi.SectorInfo, []byte, []ffi.PoStCandidateWithTicket) ([]abi.PoStProof, error)
2019-11-21 22:21:45 +00:00
}
2019-12-02 12:51:16 +00:00
type eppProvider struct{}
2019-11-21 22:21:45 +00:00
2020-02-27 21:45:31 +00:00
func (epp *eppProvider) GenerateCandidates(ctx context.Context, _ []abi.SectorInfo, eprand abi.PoStRandomness) ([]ffi.PoStCandidateWithTicket, error) {
return []ffi.PoStCandidateWithTicket{
2019-12-02 12:51:16 +00:00
{
2020-02-27 21:45:31 +00:00
Candidate: abi.PoStCandidate{
RegisteredProof: abi.RegisteredProof_StackedDRG2KiBPoSt,
SectorID: abi.SectorID{Number: 1},
PartialTicket: abi.PartialTicket{},
PrivateProof: abi.PrivatePoStCandidateProof{},
ChallengeIndex: 1,
},
2019-11-21 22:21:45 +00:00
},
}, nil
}
2020-02-27 21:45:31 +00:00
func (epp *eppProvider) ComputeProof(ctx context.Context, _ []abi.SectorInfo, eprand []byte, winners []ffi.PoStCandidateWithTicket) ([]abi.PoStProof, error) {
2020-02-27 21:45:31 +00:00
return []abi.PoStProof{{
ProofBytes: []byte("valid proof"),
}}, nil
2019-11-21 22:21:45 +00:00
}
2019-12-03 18:25:56 +00:00
type ProofInput struct {
2020-02-27 21:45:31 +00:00
sectors []abi.SectorInfo
2019-12-03 19:33:29 +00:00
hvrf []byte
2020-02-27 21:45:31 +00:00
winners []ffi.PoStCandidateWithTicket
2019-12-03 19:33:29 +00:00
vrfout []byte
2019-12-03 18:25:56 +00:00
}
2019-12-03 21:27:07 +00:00
func IsRoundWinner(ctx context.Context, ts *types.TipSet, round int64, miner address.Address, epp ElectionPoStProver, a MiningCheckAPI) (*ProofInput, error) {
2020-02-23 20:00:47 +00:00
epostRand, err := a.ChainGetRandomness(ctx, ts.Key(), crypto.DomainSeparationTag_ElectionPoStChallengeSeed, abi.ChainEpoch(round-build.EcRandomnessLookback), miner.Bytes())
if err != nil {
2019-12-03 21:27:07 +00:00
return nil, xerrors.Errorf("chain get randomness: %w", err)
}
mworker, err := a.StateMinerWorker(ctx, miner, ts.Key())
if err != nil {
2019-12-03 21:27:07 +00:00
return nil, xerrors.Errorf("failed to get miner worker: %w", err)
}
2020-02-23 20:00:47 +00:00
vrfout, err := ComputeVRF(ctx, a.WalletSign, mworker, epostRand)
if err != nil {
2019-12-03 21:27:07 +00:00
return nil, xerrors.Errorf("failed to compute VRF: %w", err)
}
pset, err := a.StateMinerProvingSet(ctx, miner, ts.Key())
2019-11-25 04:45:13 +00:00
if err != nil {
2019-12-03 21:27:07 +00:00
return nil, xerrors.Errorf("failed to load proving set for miner: %w", err)
2019-11-25 04:45:13 +00:00
}
if len(pset) == 0 {
2019-12-03 21:27:07 +00:00
return nil, nil
}
2019-11-25 04:45:13 +00:00
2020-02-27 21:45:31 +00:00
var sinfos []abi.SectorInfo
2019-11-25 04:45:13 +00:00
for _, s := range pset {
2020-02-27 21:45:31 +00:00
sinfos = append(sinfos, abi.SectorInfo{
SectorNumber: s.ID,
SealedCID: s.Info.Info.SealedCID,
2019-11-25 04:45:13 +00:00
})
}
2019-11-26 02:43:43 +00:00
hvrf := sha256.Sum256(vrfout)
2020-02-27 21:45:31 +00:00
candidates, err := epp.GenerateCandidates(ctx, sinfos, hvrf[:])
2019-11-21 22:21:45 +00:00
if err != nil {
2019-12-03 21:27:07 +00:00
return nil, xerrors.Errorf("failed to generate electionPoSt candidates: %w", err)
2019-11-21 22:21:45 +00:00
}
pow, err := a.StateMinerPower(ctx, miner, ts.Key())
if err != nil {
2019-12-03 21:27:07 +00:00
return nil, xerrors.Errorf("failed to check power: %w", err)
}
ssize, err := a.StateMinerSectorSize(ctx, miner, ts.Key())
2019-11-21 22:21:45 +00:00
if err != nil {
2019-12-03 21:27:07 +00:00
return nil, xerrors.Errorf("failed to look up miners sector size: %w", err)
2019-11-21 22:21:45 +00:00
}
2020-02-27 21:45:31 +00:00
var winners []ffi.PoStCandidateWithTicket
2019-11-21 22:21:45 +00:00
for _, c := range candidates {
2020-02-27 21:45:31 +00:00
if types.IsTicketWinner(c.Candidate.PartialTicket, ssize, uint64(len(sinfos)), pow.TotalPower) {
2019-11-21 22:21:45 +00:00
winners = append(winners, c)
}
}
// no winners, sad
if len(winners) == 0 {
2019-12-03 21:27:07 +00:00
return nil, nil
2019-11-21 22:21:45 +00:00
}
2019-12-03 21:27:07 +00:00
return &ProofInput{
2020-02-27 21:45:31 +00:00
sectors: sinfos,
2019-12-03 18:25:56 +00:00
hvrf: hvrf[:],
winners: winners,
vrfout: vrfout,
}, nil
}
func ComputeProof(ctx context.Context, epp ElectionPoStProver, pi *ProofInput) (*types.EPostProof, error) {
proof, err := epp.ComputeProof(ctx, pi.sectors, pi.hvrf, pi.winners)
2019-11-21 22:21:45 +00:00
if err != nil {
2019-12-03 18:25:56 +00:00
return nil, xerrors.Errorf("failed to compute snark for election proof: %w", err)
2019-11-21 22:21:45 +00:00
}
ept := types.EPostProof{
2020-02-27 21:45:31 +00:00
Proofs: proof,
2019-12-03 18:25:56 +00:00
PostRand: pi.vrfout,
2019-11-21 22:21:45 +00:00
}
2019-12-03 18:25:56 +00:00
for _, win := range pi.winners {
part := make([]byte, 32)
2020-02-27 21:45:31 +00:00
copy(part, win.Candidate.PartialTicket)
2019-11-28 12:46:56 +00:00
ept.Candidates = append(ept.Candidates, types.EPostTicket{
Partial: part,
2020-02-27 21:45:31 +00:00
SectorID: win.Candidate.SectorID.Number,
ChallengeIndex: uint64(win.Candidate.ChallengeIndex),
2019-11-21 22:21:45 +00:00
})
}
2019-12-03 18:25:56 +00:00
return &ept, nil
}
2020-02-12 23:52:36 +00:00
type SignFunc func(context.Context, address.Address, []byte) (*crypto.Signature, error)
2020-02-23 20:00:47 +00:00
func VerifyVRF(ctx context.Context, worker address.Address, vrfBase, vrfproof []byte) error {
2019-12-02 12:51:16 +00:00
_, span := trace.StartSpan(ctx, "VerifyVRF")
2019-11-21 22:21:45 +00:00
defer span.End()
2020-02-12 23:52:36 +00:00
sig := &crypto.Signature{
Type: crypto.SigTypeBLS,
2019-11-21 22:21:45 +00:00
Data: vrfproof,
}
if err := sigs.Verify(sig, worker, vrfBase); err != nil {
2019-11-21 22:21:45 +00:00
return xerrors.Errorf("vrf was invalid: %w", err)
}
return nil
}
2020-02-23 20:00:47 +00:00
func ComputeVRF(ctx context.Context, sign SignFunc, worker address.Address, sigInput []byte) ([]byte, error) {
2019-11-21 22:21:45 +00:00
sig, err := sign(ctx, worker, sigInput)
if err != nil {
return nil, err
}
2020-02-12 23:52:36 +00:00
if sig.Type != crypto.SigTypeBLS {
return nil, fmt.Errorf("miner worker address was not a BLS key")
}
return sig.Data, nil
}
type genFakeVerifier struct{}
var _ sectorbuilder.Verifier = (*genFakeVerifier)(nil)
2020-02-27 21:45:31 +00:00
func (m genFakeVerifier) VerifyElectionPost(ctx context.Context, pvi abi.PoStVerifyInfo) (bool, error) {
panic("nyi")
}
2020-02-27 21:45:31 +00:00
func (m genFakeVerifier) GenerateDataCommitment(ssize abi.PaddedPieceSize, pieces []abi.PieceInfo) (cid.Cid, error) {
return sbmock.MockVerifier.GenerateDataCommitment(ssize, pieces)
}
2020-02-27 21:45:31 +00:00
func (m genFakeVerifier) VerifySeal(svi abi.SealVerifyInfo) (bool, error) {
return true, nil
}
2020-02-27 21:45:31 +00:00
func (m genFakeVerifier) VerifyFallbackPost(ctx context.Context, pvi abi.PoStVerifyInfo) (bool, error) {
panic("nyi")
}