Upgrade v0.1.2, network halt fix.

Addresses a critical staking PowerReduction issue by mutating validators power for accurate voting power recalibration and network integrity. Do not run before we have finished everything!

app/app.go

@@ -479,6 +479,8 @@ func New(
 		panic(err)
 	}
 
+	app.applyUpgrades()
+
 	return app
 }
 
@@ -655,3 +657,7 @@ func initParamsKeeper(
 
 	return paramsKeeper
 }
+
+func (app *App) applyUpgrades() {
+	app.applyUpgrade_v0_1_2()
+}

app/upgrade_v0_1_2.go

@@ -0,0 +1,90 @@
+// Copyright 2024 Galactica Network
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package app
+
+import (
+	"bytes"
+	"fmt"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
+
+	"github.com/Galactica-corp/galactica/app/upgrades/v0_1_2"
+)
+
+// applyUpgrade_v0_1_2 checks and applies the upgrade plan if necessary.
+func (app *App) applyUpgrade_v0_1_2() {
+	ctx, err := app.CreateQueryContext(v0_1_2.UpgradeBlockHeight, false)
+	if err != nil {
+		return
+	}
+
+	plan, err := app.UpgradeKeeper.ReadUpgradeInfoFromDisk()
+	if err != nil || plan.Height < v0_1_2.UpgradeBlockHeight {
+		app.UpgradeKeeper.SetUpgradeHandler(v0_1_2.UpgradeName, app.upgradeHandler_v0_1_2())
+		app.UpgradeKeeper.ApplyUpgrade(ctx, v0_1_2.Plan)
+	}
+}
+
+// upgradeHandler_v0_1_2 returns a handler function for processing the upgrade.
+func (app *App) upgradeHandler_v0_1_2() func(
+	ctx sdk.Context,
+	_ upgradetypes.Plan,
+	fromVM module.VersionMap,
+) (module.VersionMap, error) {
+	return func(ctx sdk.Context, _ upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+		logger := ctx.Logger().With("upgrade", v0_1_2.UpgradeName)
+		validators := app.StakingKeeper.GetAllValidators(ctx)
+
+		for _, validator := range validators {
+			if err := app.updateValidatorPowerIndex(ctx, validator); err != nil {
+				panic(fmt.Sprintf("failed to update validator power index: %v", err))
+			}
+
+			logger.Info("Validator power index updated", "validator", validator.OperatorAddress)
+		}
+		logger.Info("All validators updated successfully.")
+
+		if err := app.UpgradeKeeper.DumpUpgradeInfoToDisk(v0_1_2.UpgradeBlockHeight, v0_1_2.Plan); err != nil {
+			return nil, err
+		}
+
+		return app.ModuleManager.RunMigrations(ctx, app.Configurator(), fromVM)
+	}
+}
+
+// updateValidatorPowerIndex updates the power index for a single validator.
+func (app *App) updateValidatorPowerIndex(ctx sdk.Context, validator stakingtypes.Validator) error {
+	store := ctx.KVStore(app.GetKey(stakingtypes.StoreKey))
+	iterator := sdk.KVStorePrefixIterator(store, stakingtypes.ValidatorsByPowerIndexKey)
+	defer iterator.Close()
+
+	for ; iterator.Valid(); iterator.Next() {
+		valAddr := stakingtypes.ParseValidatorPowerRankKey(iterator.Key())
+		if bytes.Equal(valAddr, validator.GetOperator()) {
+			store.Delete(iterator.Key())
+			break // Assuming unique power index key per validator.
+		}
+	}
+
+	app.StakingKeeper.SetValidatorByPowerIndex(ctx, validator)
+	if _, err := app.StakingKeeper.ApplyAndReturnValidatorSetUpdates(ctx); err != nil {
+		return err
+	}
+
+	return nil
+}

app/upgrades/v0_1_2/upgrade.go

@@ -0,0 +1,32 @@
+// Copyright 2024 Galactica Network
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package v0_1_2
+
+import (
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
+)
+
+const (
+	UpgradeName        = "0.1.2"
+	UpgradeBlockHeight = 16951
+)
+
+// Plan defines the upgrade plan for addressing the staking PowerReduction issue.
+var Plan = upgradetypes.Plan{
+	Name:   UpgradeName,
+	Height: UpgradeBlockHeight,
+	Info: "Addresses a critical staking PowerReduction issue by mutating validators' " +
+		"power for accurate voting power recalibration and network integrity.",
+}

cmd/galacticad/cmd/config.go

@@ -15,6 +15,8 @@
 package cmd
 
 import (
+	"math/big"
+
 	"cosmossdk.io/errors"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
@@ -73,6 +75,10 @@ func initSDKConfig() {
 
 	RegisterDenoms()
 
+	sdk.DefaultPowerReduction = sdk.NewIntFromBigInt(
+		new(big.Int).Exp(big.NewInt(10), big.NewInt(BaseDenomUnit), nil),
+	)
+
 	config.Seal()
 }