package zerocomm_test import ( "bytes" "fmt" "io" "testing" commcid "github.com/filecoin-project/go-fil-commcid" abi "github.com/filecoin-project/specs-actors/actors/abi" "github.com/ipfs/go-cid" "github.com/filecoin-project/lotus/storage/sectorstorage/ffiwrapper" "github.com/filecoin-project/lotus/storage/sectorstorage/zerocomm" ) func TestComms(t *testing.T) { t.Skip("don't have enough ram") // no, but seriously, currently this needs like 3tb of /tmp var expPieceComms [zerocomm.Levels - zerocomm.Skip]cid.Cid { l2, err := ffiwrapper.GeneratePieceCIDFromFile(abi.RegisteredProof_StackedDRG2KiBPoSt, bytes.NewReader(make([]byte, 127)), 127) if err != nil { t.Fatal(err) } expPieceComms[0] = l2 } for i := 1; i < zerocomm.Levels-2; i++ { var err error sz := abi.UnpaddedPieceSize(127 << uint(i)) fmt.Println(i, sz) r := io.LimitReader(&NullReader{}, int64(sz)) expPieceComms[i], err = ffiwrapper.GeneratePieceCIDFromFile(abi.RegisteredProof_StackedDRG2KiBPoSt, r, sz) if err != nil { t.Fatal(err) } } for i, comm := range expPieceComms { c, err := commcid.CIDToPieceCommitmentV1(comm) if err != nil { t.Fatal(err) } if string(c) != string(zerocomm.PieceComms[i][:]) { t.Errorf("zero commitment %d didn't match", i) } } for _, comm := range expPieceComms { // Could do codegen, but this is good enough fmt.Printf("%#v,\n", comm) } } func TestCommsSmall(t *testing.T) { var expPieceComms [8]cid.Cid lvls := len(expPieceComms) + zerocomm.Skip { l2, err := ffiwrapper.GeneratePieceCIDFromFile(abi.RegisteredProof_StackedDRG2KiBPoSt, bytes.NewReader(make([]byte, 127)), 127) if err != nil { t.Fatal(err) } expPieceComms[0] = l2 } for i := 1; i < lvls-2; i++ { var err error sz := abi.UnpaddedPieceSize(127 << uint(i)) fmt.Println(i, sz) r := io.LimitReader(&NullReader{}, int64(sz)) expPieceComms[i], err = ffiwrapper.GeneratePieceCIDFromFile(abi.RegisteredProof_StackedDRG2KiBPoSt, r, sz) if err != nil { t.Fatal(err) } } for i, comm := range expPieceComms { c, err := commcid.CIDToPieceCommitmentV1(comm) if err != nil { t.Fatal(err) } if string(c) != string(zerocomm.PieceComms[i][:]) { t.Errorf("zero commitment %d didn't match", i) } } for _, comm := range expPieceComms { // Could do codegen, but this is good enough fmt.Printf("%#v,\n", comm) } } func TestForSise(t *testing.T) { exp, err := ffiwrapper.GeneratePieceCIDFromFile(abi.RegisteredProof_StackedDRG2KiBPoSt, bytes.NewReader(make([]byte, 1016)), 1016) if err != nil { return } actual := zerocomm.ZeroPieceCommitment(1016) if !exp.Equals(actual) { t.Errorf("zero commitment didn't match") } } type NullReader struct{} func (NullReader) Read(out []byte) (int, error) { for i := range out { out[i] = 0 } return len(out), nil }