add genesis verification in init genesis

KiraCore · May 27, 2024 · fe445d6 · fe445d6
1 parent dd1aa75
commit fe445d6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/x/upgrade/module.go b/x/upgrade/module.go
@@ -99,6 +99,10 @@ func (am AppModule) InitGenesis(
 	var genesisState types.GenesisState
 	cdc.MustUnmarshalJSON(data, &genesisState)
 
+	if genesisState.Version != sekaitypes.SekaiVersion {
+		panic(types.ErrInvalidGenesisVersion)
+	}
+
 	if genesisState.CurrentPlan != nil {
 		am.upgradeKeeper.SaveCurrentPlan(ctx, *genesisState.CurrentPlan)
 	}

diff --git a/x/upgrade/types/genesis.go b/x/upgrade/types/genesis.go
@@ -1,8 +1,13 @@
 package types
 
+import (
+	sekaitypes "github.com/KiraCore/sekai/types"
+)
+
 // DefaultGenesis returns the default CustomGo genesis state
 func DefaultGenesis() *GenesisState {
 	return &GenesisState{
+		Version:     sekaitypes.SekaiVersion,
 		CurrentPlan: nil,
 		NextPlan:    nil,
 	}