Skip to content

Commit 104908e

Browse files
committed
improvements #566 #570
- Bump schemaVersion of PKI Distribution Point model - Add base64 format validation for OTAChecksum of Model Version model Signed-off-by: Abdulbois <abdulbois.tursunov@dsr-corporation.com> Signed-off-by: Abdulbois <abdulbois123@gmail.com>
1 parent 65d202d commit 104908e

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed

x/model/handler_test.go

+15
Original file line numberDiff line numberDiff line change
@@ -812,6 +812,21 @@ func TestHandler_AddModelVersion(t *testing.T) {
812812
require.Equal(t, schemaVersion, receivedModelVersion.SchemaVersion)
813813
}
814814

815+
func TestHandler_AddModelVersionWithInvalidOtaChecksum(t *testing.T) {
816+
setup := Setup(t)
817+
818+
// add new model
819+
msgCreateModel := NewMsgCreateModel(setup.Vendor)
820+
_, err := setup.Handler(setup.Ctx, msgCreateModel)
821+
require.NoError(t, err)
822+
823+
// add new model version
824+
msgCreateModelVersion := NewMsgCreateModelVersion(setup.Vendor, testconstants.SoftwareVersion)
825+
msgCreateModelVersion.OtaChecksum = "not_base64_encoded"
826+
_, err = setup.Handler(setup.Ctx, msgCreateModelVersion)
827+
require.ErrorIs(t, err, types.ErrOtaChecksumIsNotValid)
828+
}
829+
815830
func TestHandler_AddMultipleModelVersions(t *testing.T) {
816831
setup := Setup(t)
817832

x/model/keeper/msg_server_model_version.go

+6
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package keeper
22

33
import (
44
"context"
5+
"encoding/base64"
56

67
"cosmossdk.io/errors"
78
sdk "github.com/cosmos/cosmos-sdk/types"
@@ -44,6 +45,11 @@ func (k msgServer) CreateModelVersion(goCtx context.Context, msg *types.MsgCreat
4445
return nil, types.NewErrModelVersionAlreadyExists(msg.Vid, msg.Pid, msg.SoftwareVersion)
4546
}
4647

48+
_, err = base64.StdEncoding.DecodeString(msg.OtaChecksum)
49+
if err != nil {
50+
return nil, types.NewErrOtaChecksumIsNotValid(msg.OtaChecksum)
51+
}
52+
4753
modelVersion := types.ModelVersion{
4854
Creator: msg.Creator,
4955
Vid: msg.Vid,

0 commit comments

Comments
 (0)