feat!: migrate to allocate previously allocated rewards, and distribute unaccounted denoms

app/upgrades/v21/upgrades.go

@@ -2,15 +2,39 @@
 
 import (
 	"context"
+	"fmt"
 
+	providerkeeper "github.com/cosmos/interchain-security/v6/x/ccv/provider/keeper"
+	providertypes "github.com/cosmos/interchain-security/v6/x/ccv/provider/types"
+
+	errorsmod "cosmossdk.io/errors"
+	"cosmossdk.io/math"
 	upgradetypes "cosmossdk.io/x/upgrade/types"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
+	authkeeper "github.com/cosmos/cosmos-sdk/x/auth/keeper"
+	bankkeeper "github.com/cosmos/cosmos-sdk/x/bank/keeper"
 
 	"github.com/cosmos/gaia/v21/app/keepers"
 )
 
+// Neutron and Stride denoms that were not whitelisted but the consumer rewards pool contains amounts of those denoms.
+// Price in $ for each denom corresponds to an approximation fo the current amount stored in the consumer rewards pool
+// as of 27.09.2024. Only denoms with amounts more than $10 are included.
+const (
+	NeutronUusdc = "ibc/4E0D0854C0F846150FA8389D75EA5B5129B17703D7F4992D0356B4FE7C013D42" // ~$40
+	NeutronUtia  = "ibc/7054742D02E4F28B7DB5B44D97A496CF5AD16C2AE6948028A5FD57DCE7C5E271" // ~$300
+
+	StrideStutia  = "ibc/17DABEBAC71C388DA064A3D54FB7E68BAF0687965EC39DEADA1FB78C0F1447E6" // ~$18,000
+	StrideStadym  = "ibc/3F0A41ECB6FAF27E315583DBF39B5B69A7149D23959A0E4B319F7EF5C618DCD7" // ~$800
+	StrideStaISLM = "ibc/61A6F21D6AFF9835F66056461F1CAE24AA3323820259856B485FE7C063CA4FA6" // ~$1650
+	StrideStuband = "ibc/E9401AC885592AC2023E0FB9BA7C8BC66D346CEE04CED8E9F545F3C25290708A" // ~$300
+	StrideStadydx = "ibc/EEFD952A6DE346F2649039E99A16430B05FFEDF628A4DE99F34BB4B5F6A9346E" // ~$21,000
+	StrideStusaga = "ibc/F918765AC289257B35DECC52BD92EBCDBA3C139658BD6F2670D70A6E10B97F58" // ~$300
+)
+
+// CreateUpgradeHandler returns an upgrade handler for Gaia v21.
 func CreateUpgradeHandler(
 	mm *module.Manager,
 	configurator module.Configurator,
@@ -22,10 +46,76 @@
 
 		vm, err := mm.RunMigrations(ctx, configurator, vm)
 		if err != nil {
-			return vm, err
+			return vm, errorsmod.Wrapf(err, "running module migrations")
+		}
+
+		ctx.Logger().Info("allocating rewards of Neutron and Stride unaccounted denoms")
+		err = AllocateNeutronAndStrideUnaccountedDenoms(ctx, keepers.ProviderKeeper, keepers.BankKeeper, keepers.AccountKeeper)
+		if err != nil {
+			return vm, errorsmod.Wrapf(err, "could not allocate rewards of Neutron and Stride unaccounted denoms")
 		}
 
 		ctx.Logger().Info("Upgrade v21 complete")
 		return vm, nil
 	}
 }
+
+// AllocateRewards allocates all the `denoms` that reside in the  `address` and are meant for the chain with `consumerId`
+func AllocateRewards(ctx sdk.Context, providerKeeper providerkeeper.Keeper, bankKeeper bankkeeper.Keeper, address sdk.AccAddress, consumerId string, denoms []string) error {
+	for _, denom := range denoms {
+		coinRewards := bankKeeper.GetBalance(ctx, address, denom)
+		decCoinRewards := sdk.DecCoins{sdk.DecCoin{Denom: coinRewards.Denom, Amount: math.LegacyNewDecFromInt(coinRewards.Amount)}}
+		consumerRewardsAllocation := providertypes.ConsumerRewardsAllocation{Rewards: decCoinRewards}
+
+		err := providerKeeper.SetConsumerRewardsAllocationByDenom(ctx, consumerId, denom, consumerRewardsAllocation)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// HasExpectedChainIdSanityCheck returns true if the chain with the provided `consumerId` is of a chain with the `expectedChainId`
+func HasExpectedChainIdSanityCheck(ctx sdk.Context, providerKeeper providerkeeper.Keeper, consumerId string, expectedChainId string) bool {
+	actualChainId, err := providerKeeper.GetConsumerChainId(ctx, consumerId)
+	if err != nil {
+		return false
+	}
+	if expectedChainId != actualChainId {
+		return false
+	}
+	return true
+}
+
+// AllocateNeutronAndStrideUnaccountedDenoms allocates previously unaccounted denoms to the Stride and Neutron consumer chains
+func AllocateNeutronAndStrideUnaccountedDenoms(ctx sdk.Context, providerKeeper providerkeeper.Keeper, bankKeeper bankkeeper.Keeper, accountKeeper authkeeper.AccountKeeper) error {
+	consumerRewardsPoolAddress := accountKeeper.GetModuleAccount(ctx, providertypes.ConsumerRewardsPool).GetAddress()
+
+	const NeutronConsumerId = "0"
+	const NeutronChainId = "neutron-1"
+
+	if !HasExpectedChainIdSanityCheck(ctx, providerKeeper, NeutronConsumerId, NeutronChainId) {
+		return fmt.Errorf("failed sanity check: consumer id (%s) does not correspond to chain id (%s)", NeutronConsumerId, NeutronChainId)
+	}
+
+	neutronUnaccountedDenoms := []string{NeutronUusdc, NeutronUtia}
+	err := AllocateRewards(ctx, providerKeeper, bankKeeper, consumerRewardsPoolAddress, NeutronConsumerId, neutronUnaccountedDenoms)
+	if err != nil {
+		return fmt.Errorf("cannot allocate rewards for consumer id (%s): %w", NeutronConsumerId, err)
+	}
+
+	const StrideConsumerId = "1"
+	const StrideChainId = "stride-1"
+
+	if !HasExpectedChainIdSanityCheck(ctx, providerKeeper, StrideConsumerId, StrideChainId) {
+		return fmt.Errorf("failed sanity check: consumer id (%s) does not correspond to chain id (%s)", StrideConsumerId, StrideChainId)
+	}
+
+	strideUnaccountedDenoms := []string{StrideStutia, StrideStadym, StrideStaISLM, StrideStuband, StrideStadydx, StrideStusaga}
+	err = AllocateRewards(ctx, providerKeeper, bankKeeper, consumerRewardsPoolAddress, StrideConsumerId, strideUnaccountedDenoms)
+	if err != nil {
+		return fmt.Errorf("cannot allocate rewards for consumer id (%s): %w", StrideConsumerId, err)
+	}
+
+	return nil
+}

app/upgrades/v21/upgrades_test.go

@@ -0,0 +1,30 @@
+package v21_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
+
+	"github.com/cosmos/gaia/v21/app/helpers"
+	v21 "github.com/cosmos/gaia/v21/app/upgrades/v21"
+)
+
+func TestHasExpectedChainIdSanityCheck(t *testing.T) {
+	gaiaApp := helpers.Setup(t)
+	ctx := gaiaApp.NewUncachedContext(true, tmproto.Header{})
+
+	pk := gaiaApp.ProviderKeeper
+
+	// no such consumer chain
+	consumerId := "0"
+	require.False(t, v21.HasExpectedChainIdSanityCheck(ctx, pk, consumerId, "chain-1"))
+
+	// consumer chain does not have `chain-1` id
+	pk.SetConsumerChainId(ctx, consumerId, "chain-2")
+	require.False(t, v21.HasExpectedChainIdSanityCheck(ctx, pk, consumerId, "chain-1"))
+
+	pk.SetConsumerChainId(ctx, consumerId, "chain-1")
+	require.True(t, v21.HasExpectedChainIdSanityCheck(ctx, pk, consumerId, "chain-1"))
+}

go.mod

@@ -28,7 +28,7 @@ require (
 	github.com/cosmos/ibc-apps/modules/rate-limiting/v8 v8.0.0
 	github.com/cosmos/ibc-go/modules/capability v1.0.1
 	github.com/cosmos/ibc-go/v8 v8.5.1
-	github.com/cosmos/interchain-security/v6 v6.1.0
+	github.com/cosmos/interchain-security/v6 v6.0.0-20241003083905-f56af4c46157
 	github.com/google/gofuzz v1.2.0
 	github.com/gorilla/mux v1.8.1
 	github.com/ory/dockertest/v3 v3.11.0

go.sum

@@ -443,6 +443,8 @@ github.com/cosmos/ibc-go/v8 v8.5.1 h1:3JleEMKBjRKa3FeTKt4fjg22za/qygLBo7mDkoYTNB
 github.com/cosmos/ibc-go/v8 v8.5.1/go.mod h1:P5hkAvq0Qbg0h18uLxDVA9q1kOJ0l36htMsskiNwXbo=
 github.com/cosmos/ics23/go v0.11.0 h1:jk5skjT0TqX5e5QJbEnwXIS2yI2vnmLOgpQPeM5RtnU=
 github.com/cosmos/ics23/go v0.11.0/go.mod h1:A8OjxPE67hHST4Icw94hOxxFEJMBG031xIGF/JHNIY0=
+github.com/cosmos/interchain-security/v6 v6.0.0-20241003083905-f56af4c46157 h1:KzH6KYAF3+psJ7KJKQihuY0oXL3NSgfS5VPAN525bZ8=
+github.com/cosmos/interchain-security/v6 v6.0.0-20241003083905-f56af4c46157/go.mod h1:8Ke3cbEBBMImBkWPfF1bxPFg5ZqYOxZPjpnh0ItxQMQ=
 github.com/cosmos/interchain-security/v6 v6.1.0 h1:ycTpT+If90nSEvRVu86ThPJxNtcmnOMjJmFC9ptd/yo=
 github.com/cosmos/interchain-security/v6 v6.1.0/go.mod h1:+5zIZEzkL4yNHB/UWXCu75t6GeEgEmWHbz5OnBWiL0o=
 github.com/cosmos/keyring v1.2.0 h1:8C1lBP9xhImmIabyXW4c3vFjjLiBdGCmfLUfeZlV1Yo=

tests/interchain/cosmwasm_test.go

@@ -80,7 +80,6 @@ func (s *CosmWasmSuite) TestCantInstantiateWithoutProp() {
 }
 
 func (s *CosmWasmSuite) TestCreateNewContract() {
-
 	_, contractAddr := s.storeInstantiateProposal(initState)
 
 	count := s.getContractCount(contractAddr)

tests/interchain/permissionless_test.go

@@ -360,8 +360,8 @@ func (s *PermissionlessConsumersSuite) TestChangePowerShaping() {
 		s.Require().NoError(err)
 		s.Require().Equal(consumer.ValidatorWallets[i].ValConsAddress, valCons)
 	}
-
 }
+
 func (s *PermissionlessConsumersSuite) TestConsumerCommissionRate() {
 	s.UpgradeChain()
 	cfg := s.consumerCfg