From 10b9d3fa969edc7d8c341e11faf8c9c52891a47d Mon Sep 17 00:00:00 2001
From: Geoff Stuart <geoff.vball@gmail.com>
Date: Fri, 14 Oct 2022 16:03:28 -0400
Subject: [PATCH] add lotus-shed invariance method

---
 cmd/lotus-shed/invariants.go | 145 +++++++++++++++++++++++++++++++++++
 cmd/lotus-shed/main.go       |   1 +
 2 files changed, 146 insertions(+)
 create mode 100644 cmd/lotus-shed/invariants.go

diff --git a/cmd/lotus-shed/invariants.go b/cmd/lotus-shed/invariants.go
new file mode 100644
index 000000000..b409b7452
--- /dev/null
+++ b/cmd/lotus-shed/invariants.go
@@ -0,0 +1,145 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"time"
+
+	"github.com/ipfs/go-cid"
+	"github.com/urfave/cli/v2"
+	"golang.org/x/xerrors"
+
+	actorstypes "github.com/filecoin-project/go-state-types/actors"
+	"github.com/filecoin-project/go-state-types/builtin"
+	v8 "github.com/filecoin-project/go-state-types/builtin/v8"
+	v9 "github.com/filecoin-project/go-state-types/builtin/v9"
+
+	"github.com/filecoin-project/lotus/blockstore"
+	"github.com/filecoin-project/lotus/chain/actors"
+	"github.com/filecoin-project/lotus/chain/consensus/filcns"
+	"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"
+	lcli "github.com/filecoin-project/lotus/cli"
+	"github.com/filecoin-project/lotus/node/repo"
+	"github.com/filecoin-project/lotus/storage/sealer/ffiwrapper"
+)
+
+var invariantsCmd = &cli.Command{
+	Name:        "check-invariants",
+	Description: "Check state invariants",
+	ArgsUsage:   "[block to look back from]",
+	Flags: []cli.Flag{
+		&cli.StringFlag{
+			Name:  "repo",
+			Value: "~/.lotus",
+		},
+	},
+	Action: func(cctx *cli.Context) error {
+		ctx := context.TODO()
+
+		if cctx.NArg() != 1 {
+			return lcli.IncorrectNumArgs(cctx)
+		}
+
+		blkCid, err := cid.Decode(cctx.Args().First())
+		if err != nil {
+			return fmt.Errorf("failed to parse input: %w", err)
+		}
+
+		fsrepo, err := repo.NewFS(cctx.String("repo"))
+		if err != nil {
+			return err
+		}
+
+		lkrepo, err := fsrepo.Lock(repo.FullNode)
+		if err != nil {
+			return err
+		}
+
+		defer lkrepo.Close() //nolint:errcheck
+
+		bs, err := lkrepo.Blockstore(ctx, repo.UniversalBlockstore)
+		if err != nil {
+			return fmt.Errorf("failed to open blockstore: %w", err)
+		}
+
+		defer func() {
+			if c, ok := bs.(io.Closer); ok {
+				if err := c.Close(); err != nil {
+					log.Warnf("failed to close blockstore: %s", err)
+				}
+			}
+		}()
+
+		mds, err := lkrepo.Datastore(context.Background(), "/metadata")
+		if err != nil {
+			return err
+		}
+
+		cs := store.NewChainStore(bs, bs, mds, filcns.Weight, nil)
+		defer cs.Close() //nolint:errcheck
+
+		sm, err := stmgr.NewStateManager(cs, filcns.NewTipSetExecutor(), vm.Syscalls(ffiwrapper.ProofVerifier), filcns.DefaultUpgradeSchedule(), nil)
+		if err != nil {
+			return err
+		}
+
+		blk, err := cs.GetBlock(ctx, blkCid)
+		if err != nil {
+			return err
+		}
+
+		ts, err := cs.LoadTipSet(ctx, types.NewTipSetKey(blk.Parents...))
+		if err != nil {
+			return err
+		}
+
+		nv := sm.GetNetworkVersion(ctx, ts.Height())
+		fmt.Println("Network Version ", nv)
+
+		av, err := actorstypes.VersionForNetwork(nv)
+		fmt.Println("Actors Version ", av)
+
+		actorCodeCids, err := actors.GetActorCodeIDs(av)
+		if err != nil {
+			return err
+		}
+
+		actorStore := store.ActorStore(ctx, blockstore.NewTieredBstore(bs, blockstore.NewMemorySync()))
+
+		// Load the state root.
+		var stateRoot types.StateRoot
+		if err := actorStore.Get(ctx, ts.ParentState(), &stateRoot); err != nil {
+			return xerrors.Errorf("failed to decode state root: %w", err)
+		}
+
+		actorTree, err := builtin.LoadTree(actorStore, stateRoot.Actors)
+
+		startTime := time.Now()
+
+		var messages *builtin.MessageAccumulator
+		switch av {
+		case actorstypes.Version8:
+			messages, err = v8.CheckStateInvariants(actorTree, ts.Height()-1, actorCodeCids)
+			if err != nil {
+				return xerrors.Errorf("checking state invariants: %w", err)
+			}
+		case actorstypes.Version9:
+			messages, err = v9.CheckStateInvariants(actorTree, ts.Height()-1, actorCodeCids)
+			if err != nil {
+				return xerrors.Errorf("checking state invariants: %w", err)
+			}
+		}
+
+		fmt.Println("completed, took ", time.Since(startTime))
+
+		for _, message := range messages.Messages() {
+			fmt.Println("got the following error: ", message)
+		}
+
+		return nil
+	},
+}
diff --git a/cmd/lotus-shed/main.go b/cmd/lotus-shed/main.go
index 0f10a617f..c661b04ef 100644
--- a/cmd/lotus-shed/main.go
+++ b/cmd/lotus-shed/main.go
@@ -73,6 +73,7 @@ func main() {
 		migrationsCmd,
 		diffCmd,
 		itestdCmd,
+		invariantsCmd,
 	}
 
 	app := &cli.App{