add genesis version verifier in validateGenesis

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

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/x/upgrade/module.go b/x/upgrade/module.go
@@ -3,6 +3,7 @@ package upgrade
 import (
 	"encoding/json"
 
+	sekaitypes "github.com/KiraCore/sekai/types"
 	"github.com/KiraCore/sekai/x/upgrade/client/cli"
 	"github.com/KiraCore/sekai/x/upgrade/keeper"
 	"github.com/KiraCore/sekai/x/upgrade/types"
@@ -41,6 +42,14 @@ func (b AppModuleBasic) DefaultGenesis(cdc codec.JSONCodec) json.RawMessage { //
 }
 
 func (b AppModuleBasic) ValidateGenesis(marshaler codec.JSONCodec, config client.TxEncodingConfig, message json.RawMessage) error {
+	genesis := types.GenesisState{}
+	err := marshaler.UnmarshalJSON(message, &genesis)
+	if err != nil {
+		return err
+	}
+	if sekaitypes.SekaiVersion != genesis.Version {
+		return types.ErrInvalidGenesisVersion
+	}
 	return nil
 }
 

diff --git a/x/upgrade/types/errors.go b/x/upgrade/types/errors.go
@@ -5,5 +5,6 @@ import (
 )
 
 var (
-	ErrInvalidUpgradeTime = errors.Register(ModuleName, 1, "invalid upgrade time")
+	ErrInvalidUpgradeTime    = errors.Register(ModuleName, 1, "invalid upgrade time")
+	ErrInvalidGenesisVersion = errors.Register(ModuleName, 2, "invalid genesis version")
 )