Backport test Genesis assertion on nitro init to 3.7.x (#3596)

* Test Genesis assertion on nitro init

* revert go.mod changes

Author: joshuacolvin0

cmd/conf/init.go

@@ -41,6 +41,7 @@ type InitConfig struct {
 	ReorgToBatch                  int64         `koanf:"reorg-to-batch"`
 	ReorgToMessageBatch           int64         `koanf:"reorg-to-message-batch"`
 	ReorgToBlockBatch             int64         `koanf:"reorg-to-block-batch"`
+	ValidateGenesisAssertion      bool          `koanf:"validate-genesis-assertion"`
 }
 
 var InitConfigDefault = InitConfig{
@@ -71,6 +72,7 @@ var InitConfigDefault = InitConfig{
 	ReorgToBatch:                  -1,
 	ReorgToMessageBatch:           -1,
 	ReorgToBlockBatch:             -1,
+	ValidateGenesisAssertion:      true,
 }
 
 func InitConfigAddOptions(prefix string, f *pflag.FlagSet) {
@@ -105,6 +107,7 @@ func InitConfigAddOptions(prefix string, f *pflag.FlagSet) {
 		"\"force\"- force rebuilding which would commence rebuilding despite the status of previous attempts,\n"+
 		"\"false\"- do not rebuild on startup",
 	)
+	f.Bool(prefix+".validate-genesis-assertion", InitConfigDefault.ValidateGenesisAssertion, "tests genesis assertion posted on parent chain against the genesis block created on init")
 }
 
 func (c *InitConfig) Validate() error {

cmd/nitro/init.go

@@ -25,6 +25,7 @@ import (
 	"github.com/cavaliergopher/grab/v3"
 	"github.com/codeclysm/extract/v3"
 
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core"
 	"github.com/ethereum/go-ethereum/core/rawdb"
@@ -36,6 +37,7 @@ import (
 	"github.com/ethereum/go-ethereum/node"
 	"github.com/ethereum/go-ethereum/params"
 
+	protocol "github.com/offchainlabs/bold/chain-abstraction"
 	"github.com/offchainlabs/nitro/arbnode"
 	"github.com/offchainlabs/nitro/arbos/arbosState"
 	"github.com/offchainlabs/nitro/arbos/arbostypes"
@@ -44,10 +46,13 @@ import (
 	"github.com/offchainlabs/nitro/cmd/pruning"
 	"github.com/offchainlabs/nitro/cmd/staterecovery"
 	"github.com/offchainlabs/nitro/execution/gethexec"
+	"github.com/offchainlabs/nitro/solgen/go/rollupgen"
+	"github.com/offchainlabs/nitro/staker/bold"
 	"github.com/offchainlabs/nitro/statetransfer"
 	"github.com/offchainlabs/nitro/util"
 	"github.com/offchainlabs/nitro/util/arbmath"
 	"github.com/offchainlabs/nitro/util/dbutil"
+	"github.com/offchainlabs/nitro/util/headerreader"
 )
 
 var notFoundError = errors.New("file not found")
@@ -806,6 +811,14 @@ func openInitializeChainDb(ctx context.Context, stack *node.Node, config *NodeCo
 		}
 		testUpdateTxIndex(chainDb, chainConfig, &txIndexWg)
 	} else {
+		var initDataReaderHasAccounts bool
+		if config.Init.ValidateGenesisAssertion {
+			accountsReader, err := initDataReader.GetAccountDataReader()
+			if err != nil {
+				return chainDb, nil, err
+			}
+			initDataReaderHasAccounts = accountsReader.More()
+		}
 		genesisBlockNr, err := initDataReader.GetNextBlockNumber()
 		if err != nil {
 			return chainDb, nil, err
@@ -895,6 +908,14 @@ func openInitializeChainDb(ctx context.Context, stack *node.Node, config *NodeCo
 		if err != nil {
 			return chainDb, nil, err
 		}
+		if config.Init.ValidateGenesisAssertion {
+			if err := validateGenesisAssertion(ctx, rollupAddrs.Rollup, l1Client, l2BlockChain.Genesis().Hash(), initDataReaderHasAccounts); err != nil {
+				if !config.Init.Force {
+					return chainDb, nil, fmt.Errorf("error testing genesis assertion: %w", err)
+				}
+				log.Error("Error testing genesis assertions", "err", err)
+			}
+		}
 	}
 
 	txIndexWg.Wait()
@@ -916,6 +937,39 @@ func openInitializeChainDb(ctx context.Context, stack *node.Node, config *NodeCo
 	return rebuildLocalWasm(ctx, &config.Execution, l2BlockChain, chainDb, wasmDb, config.Init.RebuildLocalWasm)
 }
 
+func validateGenesisAssertion(ctx context.Context, rollupAddress common.Address, l1Client *ethclient.Client, genesisBlockHash common.Hash, initDataReaderHasAccounts bool) error {
+	userLogic, err := rollupgen.NewRollupUserLogic(rollupAddress, l1Client)
+	if err != nil {
+		return err
+	}
+	_, err = userLogic.ChallengeGracePeriodBlocks(&bind.CallOpts{Context: ctx})
+	if err != nil {
+		if !headerreader.IsExecutionReverted(err) {
+			return err
+		}
+		log.Warn("Genesis Assertion is not tested") // not a bold chain
+		return nil
+	}
+	genesisAssertionHash, err := userLogic.GenesisAssertionHash(&bind.CallOpts{Context: context.Background()})
+	if err != nil {
+		return err
+	}
+	genesisAssertionCreationInfo, err := bold.ReadBoldAssertionCreationInfo(ctx, userLogic, l1Client, rollupAddress, genesisAssertionHash)
+	if err != nil {
+		return err
+	}
+	beforeGlobalState := protocol.GoGlobalStateFromSolidity(genesisAssertionCreationInfo.BeforeState.GlobalState)
+	afterGlobalState := protocol.GoGlobalStateFromSolidity(genesisAssertionCreationInfo.AfterState.GlobalState)
+	isNullAssertion := beforeGlobalState.Batch == afterGlobalState.Batch && beforeGlobalState.PosInBatch == afterGlobalState.PosInBatch
+	if isNullAssertion && initDataReaderHasAccounts {
+		return errors.New("genesis assertion is null but there are accounts in the init data")
+	}
+	if !isNullAssertion && afterGlobalState.BlockHash != genesisBlockHash {
+		return errors.New("genesis assertion is non null and its afterGlobalState.BlockHash doesn't match the genesis blockHash")
+	}
+	return nil
+}
+
 func testTxIndexUpdated(chainDb ethdb.Database, lastBlock uint64) bool {
 	var transactions types.Transactions
 	blockHash := rawdb.ReadCanonicalHash(chainDb, lastBlock)

cmd/nitro/init_test.go

@@ -429,6 +429,7 @@ func TestOpenInitializeChainDbIncompatibleStateScheme(t *testing.T) {
 	nodeConfig.Node = *arbnode.ConfigDefaultL2Test()
 	nodeConfig.Init.DevInit = true
 	nodeConfig.Init.DevInitAddress = "0x3f1Eae7D46d88F08fc2F8ed27FCb2AB183EB2d0E"
+	nodeConfig.Init.ValidateGenesisAssertion = false
 
 	l1Client := ethclient.NewClient(stack.Attach())
 
@@ -696,6 +697,7 @@ func TestOpenInitializeChainDbEmptyInit(t *testing.T) {
 	nodeConfig.Chain.ID = 42161
 	nodeConfig.Node = *arbnode.ConfigDefaultL2Test()
 	nodeConfig.Init.Empty = true
+	nodeConfig.Init.ValidateGenesisAssertion = false
 
 	l1Client := ethclient.NewClient(stack.Attach())