Merge pull request #5669 from oasisprotocol/peternose/trivial/undo-fi…

…x-gas-charge

go/consensus/cometbft/apps/keymanager/secrets: Undo fix gas charge

go/consensus/cometbft/apps/keymanager/secrets/txs.go

@@ -21,20 +21,6 @@ func (ext *secretsExt) updatePolicy(
 	state *secretsState.MutableState,
 	sigPol *secrets.SignedPolicySGX,
 ) error {
-	// Charge gas for this operation.
-	kmParams, err := state.ConsensusParameters(ctx)
-	if err != nil {
-		return err
-	}
-	if err = ctx.Gas().UseGas(1, secrets.GasOpUpdatePolicy, kmParams.GasCosts); err != nil {
-		return err
-	}
-
-	// Return early if simulating since this is just estimating gas.
-	if ctx.IsSimulation() {
-		return nil
-	}
-
 	// Ensure that the runtime exists and is a key manager.
 	regState := registryState.NewMutableState(ctx.State())
 	kmRt, err := common.KeyManagerRuntime(ctx, sigPol.Policy.ID)
@@ -70,6 +56,20 @@ func (ext *secretsExt) updatePolicy(
 		return nil
 	}
 
+	// Charge gas for this operation.
+	kmParams, err := state.ConsensusParameters(ctx)
+	if err != nil {
+		return err
+	}
+	if err = ctx.Gas().UseGas(1, secrets.GasOpUpdatePolicy, kmParams.GasCosts); err != nil {
+		return err
+	}
+
+	// Return early if simulating since this is just estimating gas.
+	if ctx.IsSimulation() {
+		return nil
+	}
+
 	// Ok, as far as we can tell the new policy is valid, apply it.
 	//
 	// Note: The key manager cohort responsible for servicing this ID
@@ -133,20 +133,6 @@ func (ext *secretsExt) publishMasterSecret(
 	state *secretsState.MutableState,
 	secret *secrets.SignedEncryptedMasterSecret,
 ) error {
-	// Charge gas for this operation.
-	kmParams, err := state.ConsensusParameters(ctx)
-	if err != nil {
-		return err
-	}
-	if err = ctx.Gas().UseGas(1, secrets.GasOpPublishMasterSecret, kmParams.GasCosts); err != nil {
-		return err
-	}
-
-	// Return early if simulating since this is just estimating gas.
-	if ctx.IsSimulation() {
-		return nil
-	}
-
 	// Ensure that the runtime exists and is a key manager.
 	kmRt, err := common.KeyManagerRuntime(ctx, secret.Secret.ID)
 	if err != nil {
@@ -198,6 +184,20 @@ func (ext *secretsExt) publishMasterSecret(
 		return nil
 	}
 
+	// Charge gas for this operation.
+	kmParams, err := state.ConsensusParameters(ctx)
+	if err != nil {
+		return err
+	}
+	if err = ctx.Gas().UseGas(1, secrets.GasOpPublishMasterSecret, kmParams.GasCosts); err != nil {
+		return err
+	}
+
+	// Return early if simulating since this is just estimating gas.
+	if ctx.IsSimulation() {
+		return nil
+	}
+
 	// Ok, as far as we can tell the secret is valid, save it.
 	if err := state.SetMasterSecret(ctx, secret); err != nil {
 		ctx.Logger().Error("keymanager: failed to set key manager master secret",
@@ -231,20 +231,6 @@ func (ext *secretsExt) publishEphemeralSecret(
 	state *secretsState.MutableState,
 	secret *secrets.SignedEncryptedEphemeralSecret,
 ) error {
-	// Charge gas for this operation.
-	kmParams, err := state.ConsensusParameters(ctx)
-	if err != nil {
-		return err
-	}
-	if err = ctx.Gas().UseGas(1, secrets.GasOpPublishEphemeralSecret, kmParams.GasCosts); err != nil {
-		return err
-	}
-
-	// Return early if simulating since this is just estimating gas.
-	if ctx.IsSimulation() {
-		return nil
-	}
-
 	// Ensure that the runtime exists and is a key manager.
 	kmRt, err := common.KeyManagerRuntime(ctx, secret.Secret.ID)
 	if err != nil {
@@ -289,6 +275,20 @@ func (ext *secretsExt) publishEphemeralSecret(
 		return nil
 	}
 
+	// Charge gas for this operation.
+	kmParams, err := state.ConsensusParameters(ctx)
+	if err != nil {
+		return err
+	}
+	if err = ctx.Gas().UseGas(1, secrets.GasOpPublishEphemeralSecret, kmParams.GasCosts); err != nil {
+		return err
+	}
+
+	// Return early if simulating since this is just estimating gas.
+	if ctx.IsSimulation() {
+		return nil
+	}
+
 	// Ok, as far as we can tell the secret is valid, save it.
 	if err := state.SetEphemeralSecret(ctx, secret); err != nil {
 		ctx.Logger().Error("keymanager: failed to set key manager ephemeral secret",