uint64

sei-protocol · Dec 12, 2024 · e891b96 · e891b96
1 parent b0118d1
commit e891b96
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 16 deletions.
diff --git a/proto/confidentialtransfers/params.proto b/proto/confidentialtransfers/params.proto
@@ -6,5 +6,5 @@ option go_package = "github.com/sei-protocol/sei-chain/x/confidentialtransfers/t
 // Params defines the parameters for the confidential tokens module.
 message Params {
   bool enable_ct_module = 1;
-  uint32 range_proof_gas_cost = 2;
+  uint64 range_proof_gas_cost = 2;
 }
diff --git a/x/confidentialtransfers/keeper/keeper.go b/x/confidentialtransfers/keeper/keeper.go
@@ -29,7 +29,7 @@ type Keeper interface {
 	SetParams(ctx sdk.Context, params types.Params)
 
 	IsCtModuleEnabled(ctx sdk.Context) bool
-	GetRangeProofGasCost(ctx sdk.Context) uint32
+	GetRangeProofGasCost(ctx sdk.Context) uint64
 
 	BankKeeper() types.BankKeeper
 
@@ -186,8 +186,8 @@ func (k BaseKeeper) IsCtModuleEnabled(ctx sdk.Context) bool {
 }
 
 // GetRangeProofGasCost retrieves the value of the RangeProofGasCost param from the parameter store
-func (k BaseKeeper) GetRangeProofGasCost(ctx sdk.Context) uint32 {
-	var rangeProofGas uint32
+func (k BaseKeeper) GetRangeProofGasCost(ctx sdk.Context) uint64 {
+	var rangeProofGas uint64
 	k.paramSpace.Get(ctx, types.KeyRangeProofGas, &rangeProofGas)
 	return rangeProofGas
 }

diff --git a/x/confidentialtransfers/keeper/msg_server.go b/x/confidentialtransfers/keeper/msg_server.go
@@ -224,7 +224,7 @@ func (m msgServer) Withdraw(goCtx context.Context, req *types.MsgWithdraw) (*typ
 	// Consume additional gas as range proofs are computationally expensive.
 	cost := m.Keeper.GetRangeProofGasCost(ctx)
 	if cost > 0 {
-		ctx.GasMeter().ConsumeGas(uint64(cost), "range proof verification")
+		ctx.GasMeter().ConsumeGas(cost, "range proof verification")
 	}
 
 	verified, _ := zkproofs.VerifyRangeProof(instruction.Proofs.RemainingBalanceRangeProof, instruction.RemainingBalanceCommitment, 128, m.CachedRangeVerifierFactory)
@@ -443,7 +443,7 @@ func (m msgServer) Transfer(goCtx context.Context, req *types.MsgTransfer) (*typ
 	}
 
 	// Validate proofs
-	rangeProofGasCost := uint64(m.Keeper.GetRangeProofGasCost(ctx))
+	rangeProofGasCost := m.Keeper.GetRangeProofGasCost(ctx)
 
 	// Consume additional gas as range proofs are computationally expensive.
 	if rangeProofGasCost > 0 {

diff --git a/x/confidentialtransfers/types/params.go b/x/confidentialtransfers/types/params.go
@@ -10,7 +10,7 @@ import (
 const DefaultEnableCtModule = true
 
 // DefaultRangeProofGasCost is the default value for RangeProofGasCost param.
-const DefaultRangeProofGasCost = uint32(1000000)
+const DefaultRangeProofGasCost = uint64(1000000)
 
 // ParamKeyTable ParamTable for confidential transfers module.
 func ParamKeyTable() paramtypes.KeyTable {
@@ -57,7 +57,7 @@ func validateEnableCtModule(i interface{}) error {
 
 // Validator for the parameter.
 func validateRangeProofGasCost(i interface{}) error {
-	_, ok := i.(uint32)
+	_, ok := i.(uint64)
 	if !ok {
 		return fmt.Errorf("invalid parameter type: %T", i)
 	}

diff --git a/x/confidentialtransfers/types/params.pb.go b/x/confidentialtransfers/types/params.pb.go
diff --git a/x/confidentialtransfers/types/params_test.go b/x/confidentialtransfers/types/params_test.go
@@ -34,7 +34,7 @@ func TestDefaultParams(t *testing.T) {
 func TestParams_Validate(t *testing.T) {
 	type fields struct {
 		EnableCtModule    bool
-		RangeProofGasCost uint32
+		RangeProofGasCost uint64
 	}
 	tests := []struct {
 		name    string