517 lines
15 KiB
Go
517 lines
15 KiB
Go
package itests
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"strings"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
logging "github.com/ipfs/go-log/v2"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/itests/kit"
|
|
"github.com/filecoin-project/lotus/node"
|
|
"github.com/filecoin-project/lotus/node/impl"
|
|
"github.com/filecoin-project/lotus/node/repo"
|
|
"github.com/filecoin-project/lotus/storage/paths"
|
|
sealing "github.com/filecoin-project/lotus/storage/pipeline"
|
|
"github.com/filecoin-project/lotus/storage/sealer/sealtasks"
|
|
"github.com/filecoin-project/lotus/storage/sealer/storiface"
|
|
"github.com/filecoin-project/lotus/storage/wdpost"
|
|
)
|
|
|
|
func TestWorkerPledge(t *testing.T) {
|
|
ctx := context.Background()
|
|
_, miner, worker, ens := kit.EnsembleWorker(t, kit.WithAllSubsystems(), kit.ThroughRPC(), kit.WithNoLocalSealing(true),
|
|
kit.WithSealWorkerTasks) // no mock proofs
|
|
|
|
ens.InterconnectAll().BeginMining(50 * time.Millisecond)
|
|
|
|
e, err := worker.Enabled(ctx)
|
|
require.NoError(t, err)
|
|
require.True(t, e)
|
|
|
|
miner.PledgeSectors(ctx, 1, 0, nil)
|
|
}
|
|
|
|
func TestWorkerPledgeSpread(t *testing.T) {
|
|
ctx := context.Background()
|
|
_, miner, worker, ens := kit.EnsembleWorker(t, kit.WithAllSubsystems(), kit.ThroughRPC(),
|
|
kit.WithSealWorkerTasks,
|
|
kit.WithAssigner("spread"),
|
|
) // no mock proofs
|
|
|
|
ens.InterconnectAll().BeginMining(50 * time.Millisecond)
|
|
|
|
e, err := worker.Enabled(ctx)
|
|
require.NoError(t, err)
|
|
require.True(t, e)
|
|
|
|
miner.PledgeSectors(ctx, 1, 0, nil)
|
|
}
|
|
|
|
func TestWorkerPledgeLocalFin(t *testing.T) {
|
|
ctx := context.Background()
|
|
_, miner, worker, ens := kit.EnsembleWorker(t, kit.WithAllSubsystems(), kit.ThroughRPC(),
|
|
kit.WithSealWorkerTasks,
|
|
kit.WithDisallowRemoteFinalize(true),
|
|
) // no mock proofs
|
|
|
|
ens.InterconnectAll().BeginMining(50 * time.Millisecond)
|
|
|
|
e, err := worker.Enabled(ctx)
|
|
require.NoError(t, err)
|
|
require.True(t, e)
|
|
|
|
miner.PledgeSectors(ctx, 1, 0, nil)
|
|
}
|
|
|
|
func TestWorkerDataCid(t *testing.T) {
|
|
ctx := context.Background()
|
|
_, miner, worker, _ := kit.EnsembleWorker(t, kit.WithAllSubsystems(), kit.ThroughRPC(), kit.WithNoLocalSealing(true),
|
|
kit.WithTaskTypes([]sealtasks.TaskType{sealtasks.TTFetch, sealtasks.TTCommit1, sealtasks.TTFinalize, sealtasks.TTDataCid, sealtasks.TTAddPiece, sealtasks.TTPreCommit1, sealtasks.TTPreCommit2, sealtasks.TTCommit2, sealtasks.TTUnseal})) // no mock proofs
|
|
|
|
e, err := worker.Enabled(ctx)
|
|
require.NoError(t, err)
|
|
require.True(t, e)
|
|
|
|
pi, err := miner.ComputeDataCid(ctx, 1016, strings.NewReader(strings.Repeat("a", 1016)))
|
|
require.NoError(t, err)
|
|
require.Equal(t, abi.PaddedPieceSize(1024), pi.Size)
|
|
require.Equal(t, "baga6ea4seaqlhznlutptgfwhffupyer6txswamerq5fc2jlwf2lys2mm5jtiaeq", pi.PieceCID.String())
|
|
|
|
bigPiece := abi.PaddedPieceSize(16 << 20).Unpadded()
|
|
pi, err = miner.ComputeDataCid(ctx, bigPiece, strings.NewReader(strings.Repeat("a", int(bigPiece))))
|
|
require.NoError(t, err)
|
|
require.Equal(t, bigPiece.Padded(), pi.Size)
|
|
require.Equal(t, "baga6ea4seaqmhoxl2ybw5m2wyd3pt3h4zmp7j52yumzu2rar26twns3uocq7yfa", pi.PieceCID.String())
|
|
|
|
nonFullPiece := abi.PaddedPieceSize(10 << 20).Unpadded()
|
|
pi, err = miner.ComputeDataCid(ctx, bigPiece, strings.NewReader(strings.Repeat("a", int(nonFullPiece))))
|
|
require.NoError(t, err)
|
|
require.Equal(t, bigPiece.Padded(), pi.Size)
|
|
require.Equal(t, "baga6ea4seaqbxib4pdxs5cqdn3fmtj4rcxk6rx6ztiqmrx7fcpo3ymuxbp2rodi", pi.PieceCID.String())
|
|
}
|
|
|
|
func TestWinningPostWorker(t *testing.T) {
|
|
prevIns := build.InsecurePoStValidation
|
|
build.InsecurePoStValidation = false
|
|
defer func() {
|
|
build.InsecurePoStValidation = prevIns
|
|
}()
|
|
|
|
ctx := context.Background()
|
|
client, _, worker, ens := kit.EnsembleWorker(t, kit.WithAllSubsystems(), kit.ThroughRPC(),
|
|
kit.WithTaskTypes([]sealtasks.TaskType{sealtasks.TTGenerateWinningPoSt})) // no mock proofs
|
|
|
|
ens.InterconnectAll().BeginMining(50 * time.Millisecond)
|
|
|
|
e, err := worker.Enabled(ctx)
|
|
require.NoError(t, err)
|
|
require.True(t, e)
|
|
|
|
client.WaitTillChain(ctx, kit.HeightAtLeast(6))
|
|
}
|
|
|
|
func TestWindowPostWorker(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
_ = logging.SetLogLevel("storageminer", "INFO")
|
|
|
|
sectors := 2 * 48 * 2
|
|
|
|
client, miner, _, ens := kit.EnsembleWorker(t,
|
|
kit.PresealSectors(sectors), // 2 sectors per partition, 2 partitions in all 48 deadlines
|
|
kit.LatestActorsAt(-1),
|
|
kit.ThroughRPC(),
|
|
kit.WithTaskTypes([]sealtasks.TaskType{sealtasks.TTGenerateWindowPoSt}))
|
|
|
|
maddr, err := miner.ActorAddress(ctx)
|
|
require.NoError(t, err)
|
|
|
|
di, err := client.StateMinerProvingDeadline(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
bm := ens.InterconnectAll().BeginMining(2 * time.Millisecond)[0]
|
|
|
|
di = di.NextNotElapsed()
|
|
|
|
t.Log("Running one proving period")
|
|
waitUntil := di.Open + di.WPoStChallengeWindow*2 + wdpost.SubmitConfidence
|
|
client.WaitTillChain(ctx, kit.HeightAtLeast(waitUntil))
|
|
|
|
t.Log("Waiting for post message")
|
|
bm.Stop()
|
|
|
|
var lastPending []*types.SignedMessage
|
|
for i := 0; i < 500; i++ {
|
|
lastPending, err = client.MpoolPending(ctx, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
if len(lastPending) > 0 {
|
|
break
|
|
}
|
|
time.Sleep(40 * time.Millisecond)
|
|
}
|
|
|
|
require.Greater(t, len(lastPending), 0)
|
|
|
|
t.Log("post message landed")
|
|
|
|
bm.MineBlocks(ctx, 2*time.Millisecond)
|
|
|
|
waitUntil = di.Open + di.WPoStChallengeWindow*3
|
|
t.Logf("End for head.Height > %d", waitUntil)
|
|
|
|
ts := client.WaitTillChain(ctx, kit.HeightAtLeast(waitUntil))
|
|
t.Logf("Now head.Height = %d", ts.Height())
|
|
|
|
p, err := client.StateMinerPower(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
ssz, err := miner.ActorSectorSize(ctx, maddr)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, p.MinerPower, p.TotalPower)
|
|
require.Equal(t, p.MinerPower.RawBytePower, types.NewInt(uint64(ssz)*uint64(sectors)))
|
|
|
|
mid, err := address.IDFromAddress(maddr)
|
|
require.NoError(t, err)
|
|
|
|
di, err = client.StateMinerProvingDeadline(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
// Remove one sector in the next deadline (so it's skipped)
|
|
{
|
|
parts, err := client.StateMinerPartitions(ctx, maddr, di.Index+1, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
require.Greater(t, len(parts), 0)
|
|
|
|
secs := parts[0].AllSectors
|
|
n, err := secs.Count()
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(2), n)
|
|
|
|
// Drop the sector
|
|
sid, err := secs.First()
|
|
require.NoError(t, err)
|
|
|
|
t.Logf("Drop sector %d; dl %d part %d", sid, di.Index+1, 0)
|
|
|
|
err = miner.BaseAPI.(*impl.StorageMinerAPI).IStorageMgr.Remove(ctx, storiface.SectorRef{
|
|
ID: abi.SectorID{
|
|
Miner: abi.ActorID(mid),
|
|
Number: abi.SectorNumber(sid),
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
waitUntil = di.Close + di.WPoStChallengeWindow
|
|
ts = client.WaitTillChain(ctx, kit.HeightAtLeast(waitUntil))
|
|
t.Logf("Now head.Height = %d", ts.Height())
|
|
|
|
p, err = client.StateMinerPower(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, p.MinerPower, p.TotalPower)
|
|
require.Equal(t, p.MinerPower.RawBytePower, types.NewInt(uint64(ssz)*uint64(sectors-1)))
|
|
}
|
|
|
|
type badWorkerStorage struct {
|
|
paths.Store
|
|
|
|
badsector *uint64
|
|
notBadCount int
|
|
}
|
|
|
|
func (bs *badWorkerStorage) GenerateSingleVanillaProof(ctx context.Context, minerID abi.ActorID, si storiface.PostSectorChallenge, ppt abi.RegisteredPoStProof) ([]byte, error) {
|
|
if atomic.LoadUint64(bs.badsector) == uint64(si.SectorNumber) {
|
|
bs.notBadCount--
|
|
if bs.notBadCount < 0 {
|
|
return nil, xerrors.New("no proof for you")
|
|
}
|
|
}
|
|
return bs.Store.GenerateSingleVanillaProof(ctx, minerID, si, ppt)
|
|
}
|
|
|
|
func TestWindowPostWorkerSkipBadSector(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
_ = logging.SetLogLevel("storageminer", "INFO")
|
|
|
|
sectors := 2 * 48 * 2
|
|
|
|
var badsector uint64 = 100000
|
|
|
|
client, miner, _, ens := kit.EnsembleWorker(t,
|
|
kit.PresealSectors(sectors), // 2 sectors per partition, 2 partitions in all 48 deadlines
|
|
kit.LatestActorsAt(-1),
|
|
kit.ThroughRPC(),
|
|
kit.WithTaskTypes([]sealtasks.TaskType{sealtasks.TTGenerateWindowPoSt}),
|
|
kit.WithWorkerStorage(func(store paths.Store) paths.Store {
|
|
return &badWorkerStorage{
|
|
Store: store,
|
|
badsector: &badsector,
|
|
}
|
|
}),
|
|
kit.ConstructorOpts(node.ApplyIf(node.IsType(repo.StorageMiner),
|
|
node.Override(new(paths.Store), func(store *paths.Remote) paths.Store {
|
|
return &badWorkerStorage{
|
|
Store: store,
|
|
badsector: &badsector,
|
|
notBadCount: 1,
|
|
}
|
|
}))))
|
|
|
|
maddr, err := miner.ActorAddress(ctx)
|
|
require.NoError(t, err)
|
|
|
|
di, err := client.StateMinerProvingDeadline(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
bm := ens.InterconnectAll().BeginMiningMustPost(2 * time.Millisecond)[0]
|
|
|
|
di = di.NextNotElapsed()
|
|
|
|
t.Log("Running one proving period")
|
|
waitUntil := di.Open + di.WPoStChallengeWindow*2 + wdpost.SubmitConfidence
|
|
client.WaitTillChain(ctx, kit.HeightAtLeast(waitUntil))
|
|
|
|
t.Log("Waiting for post message")
|
|
bm.Stop()
|
|
|
|
var lastPending []*types.SignedMessage
|
|
for i := 0; i < 500; i++ {
|
|
lastPending, err = client.MpoolPending(ctx, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
if len(lastPending) > 0 {
|
|
break
|
|
}
|
|
time.Sleep(40 * time.Millisecond)
|
|
}
|
|
|
|
require.Greater(t, len(lastPending), 0)
|
|
|
|
t.Log("post message landed")
|
|
|
|
bm.MineBlocksMustPost(ctx, 2*time.Millisecond)
|
|
|
|
waitUntil = di.Open + di.WPoStChallengeWindow*3
|
|
t.Logf("End for head.Height > %d", waitUntil)
|
|
|
|
ts := client.WaitTillChain(ctx, kit.HeightAtLeast(waitUntil))
|
|
t.Logf("Now head.Height = %d", ts.Height())
|
|
|
|
p, err := client.StateMinerPower(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
ssz, err := miner.ActorSectorSize(ctx, maddr)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, p.MinerPower, p.TotalPower)
|
|
require.Equal(t, p.MinerPower.RawBytePower, types.NewInt(uint64(ssz)*uint64(sectors)))
|
|
|
|
di, err = client.StateMinerProvingDeadline(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
// Remove one sector in the next deadline (so it's skipped)
|
|
{
|
|
parts, err := client.StateMinerPartitions(ctx, maddr, di.Index+1, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
require.Greater(t, len(parts), 0)
|
|
|
|
secs := parts[0].AllSectors
|
|
n, err := secs.Count()
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(2), n)
|
|
|
|
// Drop the sector
|
|
sid, err := secs.First()
|
|
require.NoError(t, err)
|
|
|
|
t.Logf("Drop sector %d; dl %d part %d", sid, di.Index+1, 0)
|
|
|
|
atomic.StoreUint64(&badsector, sid)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
waitUntil = di.Close + di.WPoStChallengeWindow
|
|
ts = client.WaitTillChain(ctx, kit.HeightAtLeast(waitUntil))
|
|
t.Logf("Now head.Height = %d", ts.Height())
|
|
|
|
p, err = client.StateMinerPower(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, p.MinerPower, p.TotalPower)
|
|
require.Equal(t, p.MinerPower.RawBytePower, types.NewInt(uint64(ssz)*uint64(sectors-1)))
|
|
}
|
|
|
|
func TestWindowPostWorkerManualPoSt(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
_ = logging.SetLogLevel("storageminer", "INFO")
|
|
|
|
sectors := 2 * 48 * 2
|
|
|
|
client, miner, _, ens := kit.EnsembleWorker(t,
|
|
kit.PresealSectors(sectors), // 2 sectors per partition, 2 partitions in all 48 deadlines
|
|
kit.LatestActorsAt(-1),
|
|
kit.ThroughRPC(),
|
|
kit.WithTaskTypes([]sealtasks.TaskType{sealtasks.TTGenerateWindowPoSt}))
|
|
|
|
maddr, err := miner.ActorAddress(ctx)
|
|
require.NoError(t, err)
|
|
|
|
di, err := client.StateMinerProvingDeadline(ctx, maddr, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
|
|
bm := ens.InterconnectAll().BeginMiningMustPost(2 * time.Millisecond)[0]
|
|
|
|
di = di.NextNotElapsed()
|
|
|
|
t.Log("Running one proving period")
|
|
waitUntil := di.Open + di.WPoStChallengeWindow*2 - 2
|
|
client.WaitTillChain(ctx, kit.HeightAtLeast(waitUntil))
|
|
|
|
t.Log("Waiting for post message")
|
|
bm.Stop()
|
|
|
|
tryDl := func(dl uint64) {
|
|
p, err := miner.ComputeWindowPoSt(ctx, dl, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
require.Len(t, p, 1)
|
|
require.Equal(t, dl, p[0].Deadline)
|
|
}
|
|
tryDl(0)
|
|
tryDl(40)
|
|
tryDl(di.Index + 4)
|
|
|
|
lastPending, err := client.MpoolPending(ctx, types.EmptyTSK)
|
|
require.NoError(t, err)
|
|
require.Len(t, lastPending, 0)
|
|
}
|
|
|
|
func TestSchedulerRemoveRequest(t *testing.T) {
|
|
ctx := context.Background()
|
|
_, miner, worker, ens := kit.EnsembleWorker(t, kit.WithAllSubsystems(), kit.ThroughRPC(), kit.WithNoLocalSealing(true),
|
|
kit.WithTaskTypes([]sealtasks.TaskType{sealtasks.TTFetch, sealtasks.TTCommit1, sealtasks.TTFinalize, sealtasks.TTDataCid, sealtasks.TTAddPiece, sealtasks.TTPreCommit1, sealtasks.TTCommit2, sealtasks.TTUnseal})) // no mock proofs
|
|
|
|
ens.InterconnectAll().BeginMining(50 * time.Millisecond)
|
|
|
|
e, err := worker.Enabled(ctx)
|
|
require.NoError(t, err)
|
|
require.True(t, e)
|
|
|
|
type info struct {
|
|
CallToWork struct {
|
|
} `json:"CallToWork"`
|
|
EarlyRet interface{} `json:"EarlyRet"`
|
|
ReturnedWork interface{} `json:"ReturnedWork"`
|
|
SchedInfo struct {
|
|
OpenWindows []string `json:"OpenWindows"`
|
|
Requests []struct {
|
|
Priority int `json:"Priority"`
|
|
SchedID string `json:"SchedId"`
|
|
Sector struct {
|
|
Miner int `json:"Miner"`
|
|
Number int `json:"Number"`
|
|
} `json:"Sector"`
|
|
TaskType string `json:"TaskType"`
|
|
} `json:"Requests"`
|
|
} `json:"SchedInfo"`
|
|
Waiting interface{} `json:"Waiting"`
|
|
}
|
|
|
|
tocheck := miner.StartPledge(ctx, 1, 0, nil)
|
|
var sn abi.SectorNumber
|
|
for n := range tocheck {
|
|
sn = n
|
|
}
|
|
// Keep checking till sector state is PC2, the request should get stuck as worker cannot process PC2
|
|
for {
|
|
st, err := miner.SectorsStatus(ctx, sn, false)
|
|
require.NoError(t, err)
|
|
if st.State == api.SectorState(sealing.PreCommit2) {
|
|
break
|
|
}
|
|
time.Sleep(time.Second)
|
|
}
|
|
|
|
// Dump current scheduler info
|
|
schedb, err := miner.SealingSchedDiag(ctx, false)
|
|
require.NoError(t, err)
|
|
|
|
j, err := json.MarshalIndent(&schedb, "", " ")
|
|
require.NoError(t, err)
|
|
|
|
var b info
|
|
err = json.Unmarshal(j, &b)
|
|
require.NoError(t, err)
|
|
|
|
var schedidb uuid.UUID
|
|
|
|
// cast scheduler info and get the request UUID. Call the SealingRemoveRequest()
|
|
require.Len(t, b.SchedInfo.Requests, 1)
|
|
require.Equal(t, "seal/v0/precommit/2", b.SchedInfo.Requests[0].TaskType)
|
|
|
|
schedidb, err = uuid.Parse(b.SchedInfo.Requests[0].SchedID)
|
|
require.NoError(t, err)
|
|
|
|
err = miner.SealingRemoveRequest(ctx, schedidb)
|
|
require.NoError(t, err)
|
|
|
|
// Dump the schduler again and compare the UUID if a request is present
|
|
// If no request present then pass the test
|
|
scheda, err := miner.SealingSchedDiag(ctx, false)
|
|
require.NoError(t, err)
|
|
|
|
k, err := json.MarshalIndent(&scheda, "", " ")
|
|
require.NoError(t, err)
|
|
|
|
var a info
|
|
err = json.Unmarshal(k, &a)
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, a.SchedInfo.Requests, 0)
|
|
}
|
|
|
|
func TestWorkerName(t *testing.T) {
|
|
name := "thisstringisprobablynotahostnameihope"
|
|
|
|
ctx := context.Background()
|
|
_, miner, worker, ens := kit.EnsembleWorker(t, kit.WithAllSubsystems(), kit.ThroughRPC(), kit.WithWorkerName(name))
|
|
|
|
ens.InterconnectAll().BeginMining(50 * time.Millisecond)
|
|
|
|
e, err := worker.Info(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, name, e.Hostname)
|
|
|
|
ws, err := miner.WorkerStats(ctx)
|
|
require.NoError(t, err)
|
|
|
|
var found bool
|
|
for _, stats := range ws {
|
|
if stats.Info.Hostname == name {
|
|
found = true
|
|
}
|
|
}
|
|
|
|
require.True(t, found)
|
|
}
|