67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
package sectorbuilder_test
|
|
|
|
import (
|
|
"io"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/lib/sectorbuilder"
|
|
)
|
|
|
|
const sectorSize = 1024
|
|
|
|
func TestSealAndVerify(t *testing.T) {
|
|
//t.Skip("this is slow")
|
|
//os.Setenv("BELLMAN_NO_GPU", "1")
|
|
|
|
build.SectorSizes = []uint64{sectorSize}
|
|
|
|
if err := build.GetParams(true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
sb, cleanup, err := sectorbuilder.TempSectorbuilder(sectorSize)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer cleanup()
|
|
|
|
dlen := sectorbuilder.UserBytesForSectorSize(sectorSize)
|
|
|
|
r := io.LimitReader(rand.New(rand.NewSource(42)), int64(dlen))
|
|
sid, err := sb.AddPiece("foo", dlen, r)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
ticket := sectorbuilder.SealTicket{
|
|
BlockHeight: 5,
|
|
TicketBytes: [32]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 1, 2},
|
|
}
|
|
|
|
pco, err := sb.SealPreCommit(sid, ticket)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
seed := sectorbuilder.SealSeed{
|
|
BlockHeight: 15,
|
|
TicketBytes: [32]byte{0, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0, 9, 8, 7, 6, 45, 3, 2, 1, 0, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0, 9},
|
|
}
|
|
|
|
sco, err := sb.SealCommit(sid, seed)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
ok, err := sectorbuilder.VerifySeal(sectorSize, pco.CommR[:], pco.CommD[:], sb.Miner, ticket.TicketBytes[:], seed.TicketBytes[:], sid, sco.Proof)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !ok {
|
|
t.Fatal("proof failed to validate")
|
|
}
|
|
}
|