Problem: no validate for malformed e2ee key in register

CHANGELOG.md

@@ -16,6 +16,7 @@
 * (versiondb) [#1387](https://github.com/crypto-org-chain/cronos/pull/1387) Add dedicated config section for versiondb, prepare for sdk 0.50 integration.
 * (e2ee)[#1413](https://github.com/crypto-org-chain/cronos/pull/1413) Add custom keyring implementation for e2ee module.
 * (e2ee)[#1415](https://github.com/crypto-org-chain/cronos/pull/1415) Add batch keys query for e2ee module.
+* (e2ee)[#1421](https://github.com/crypto-org-chain/cronos/pull/1421) Validate e2ee key when register.
 
 ### Bug Fixes
 

integration_tests/configs/default.jsonnet

@@ -90,7 +90,7 @@
         e2ee: {
           keys: [{
             address: 'crc16z0herz998946wr659lr84c8c556da55dc34hh',
-            key: std.base64('key'),
+            key: 'age1k3mpspxytgvx6e0jja0xgrtzz7vw2p00c2a3xmq5ygfzhwh4wg0s35z4c8',
           }],
         },
         gov: {

integration_tests/cosmoscli.py

@@ -1852,7 +1852,7 @@ def query_e2ee_key(self, address):
                 home=self.data_dir,
                 output="json",
             )
-        )["key"]
+        ).get("key")
 
     def query_e2ee_keys(self, *addresses):
         return json.loads(
@@ -1864,7 +1864,7 @@ def query_e2ee_keys(self, *addresses):
                 home=self.data_dir,
                 output="json",
             )
-        )["keys"]
+        ).get("keys")
 
     def register_e2ee_key(self, key, **kwargs):
         kwargs.setdefault("gas_prices", DEFAULT_GAS_PRICE)

integration_tests/test_e2ee.py

@@ -1,3 +1,10 @@
+def test_register(cronos):
+    cli = cronos.cosmos_cli()
+    pubkey0 = cli.keygen(keyring_name="key0")
+    cli.register_e2ee_key(pubkey0 + "malformed", _from="validator")
+    assert not cli.query_e2ee_key(cli.address("validator"))
+
+
 def test_encrypt_decrypt(cronos):
     cli = cronos.cosmos_cli()
 

x/e2ee/keeper/keeper.go

@@ -32,6 +32,9 @@
 	ctx context.Context,
 	req *types.MsgRegisterEncryptionKey,
 ) (*types.MsgRegisterEncryptionKeyResponse, error) {
+	if err := req.ValidateBasic(); err != nil {
+		return nil, err
+	}
 	bz, err := k.addressCodec.StringToBytes(req.Address)
 	if err != nil {
 		return nil, err

x/e2ee/types/msg.go

@@ -3,6 +3,7 @@
 import (
 	fmt "fmt"
 
+	"filippo.io/age"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
@@ -11,7 +12,11 @@
 		return fmt.Errorf("address cannot be empty")
 	}
 	if len(m.Key) == 0 {
-		return fmt.Errorf("key cannot be nil")
+		return fmt.Errorf("key cannot be empty")
+	}
+
+	if _, err := age.ParseX25519Recipient(m.Key); err != nil {
+		return err
 	}
 	// validate bech32 format of Address
 	if _, err := sdk.AccAddressFromBech32(m.Address); err != nil {