add

Author: aljo242

modules/core/04-channel/v2/keeper/genesis.go

@@ -0,0 +1,58 @@
+package keeper
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
+	"github.com/cosmos/ibc-go/v9/modules/core/04-channel/v2/types"
+)
+
+// InitGenesis sets the genesis state in the store.
+func (k *Keeper) InitGenesis(ctx sdk.Context, data types.GenesisState) {
+	// set acks
+	for _, ack := range data.Acknowledgements {
+		k.SetPacketAcknowledgement(ctx, ack.ClientId, ack.Sequence, ack.Data)
+	}
+
+	// set commits
+	for _, commitment := range data.Commitments {
+		k.SetPacketCommitment(ctx, commitment.ClientId, commitment.Sequence, commitment.Data)
+	}
+
+	// set receipts
+	for _, receipt := range data.Receipts {
+		k.SetPacketReceipt(ctx, receipt.ClientId, receipt.Sequence)
+	}
+
+	// set send sequences
+	for _, seq := range data.SendSequences {
+		k.SetNextSequenceSend(ctx, seq.ClientId, seq.Sequence)
+	}
+}
+
+// ExportGenesis exports the current state to a genesis state.
+func (k *Keeper) ExportGenesis(ctx sdk.Context) types.GenesisState {
+	clientStates := k.ClientKeeper.GetAllGenesisClients(ctx)
+	gs := types.GenesisState{
+		Acknowledgements: make([]types.PacketState, 0),
+		Commitments:      make([]types.PacketState, 0),
+		Receipts:         make([]types.PacketState, 0),
+		SendSequences:    make([]types.PacketSequence, 0),
+	}
+	for _, clientState := range clientStates {
+		acks := k.GetAllPacketAcknowledgementsForClient(ctx, clientState.ClientId)
+		gs.Acknowledgements = append(gs.Acknowledgements, acks...)
+
+		comms := k.GetAllPacketCommitmentsForClient(ctx, clientState.ClientId)
+		gs.Commitments = append(gs.Commitments, comms...)
+
+		receipts := k.GetAllPacketReceiptsForClient(ctx, clientState.ClientId)
+		gs.Receipts = append(gs.Receipts, receipts...)
+
+		seq, ok := k.GetNextSequenceSend(ctx, clientState.ClientId)
+		if ok {
+			gs.SendSequences = append(gs.SendSequences, types.NewPacketSequence(clientState.ClientId, seq))
+		}
+	}
+
+	return gs
+}

modules/core/04-channel/v2/keeper/genesis_test.go

@@ -0,0 +1,55 @@
+package keeper_test
+
+import (
+	"github.com/cosmos/ibc-go/v9/modules/core/04-channel/v2/types"
+	ibctesting "github.com/cosmos/ibc-go/v9/testing"
+)
+
+// TestInitExportGenesis tests the import and export flow for the channel v2 keeper.
+func (suite *KeeperTestSuite) TestInitExportGenesis() {
+	path := ibctesting.NewPath(suite.chainA, suite.chainB)
+	path.SetupV2()
+
+	app := suite.chainA.App
+
+	emptyGenesis := types.DefaultGenesisState()
+
+	// create a valid genesis state that uses the client keepers existing client IDs
+	clientStates := app.GetIBCKeeper().ClientKeeper.GetAllGenesisClients(suite.chainA.GetContext())
+	validGs := types.DefaultGenesisState()
+	for i, clientState := range clientStates {
+		ack := types.NewPacketState(clientState.ClientId, uint64(i+1), []byte("ack"))
+		receipt := types.NewPacketState(clientState.ClientId, uint64(i+1), []byte{byte(0x2)})
+		commitment := types.NewPacketState(clientState.ClientId, uint64(i+1), []byte("commit_hash"))
+		seq := types.NewPacketSequence(clientState.ClientId, uint64(i+1))
+
+		validGs.Acknowledgements = append(validGs.Acknowledgements, ack)
+		validGs.Receipts = append(validGs.Receipts, receipt)
+		validGs.Commitments = append(validGs.Commitments, commitment)
+		validGs.SendSequences = append(validGs.SendSequences, seq)
+		emptyGenesis.SendSequences = append(emptyGenesis.SendSequences, seq)
+	}
+
+	tests := []struct {
+		name     string
+		genState types.GenesisState
+	}{
+		{
+			name:     "no modifications genesis",
+			genState: emptyGenesis,
+		},
+		{
+			name:     "valid",
+			genState: validGs,
+		},
+	}
+
+	for _, tt := range tests {
+		suite.Run(tt.name, func() {
+			app.GetIBCKeeper().ChannelKeeperV2.InitGenesis(suite.chainA.GetContext(), tt.genState)
+
+			exported := app.GetIBCKeeper().ChannelKeeperV2.ExportGenesis(suite.chainA.GetContext())
+			suite.Require().Equal(tt.genState, exported)
+		})
+	}
+}

modules/core/04-channel/v2/keeper/keeper.go

@@ -1,12 +1,15 @@
 package keeper
 
 import (
+	"bytes"
 	"context"
 
 	"cosmossdk.io/core/appmodule"
 	"cosmossdk.io/log"
+	storetypes "cosmossdk.io/store/types"
 
 	"github.com/cosmos/cosmos-sdk/codec"
+	"github.com/cosmos/cosmos-sdk/runtime"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
 	connectionkeeper "github.com/cosmos/ibc-go/v9/modules/core/03-connection/keeper"
@@ -50,7 +53,7 @@ func NewKeeper(
 }
 
 // Logger returns a module-specific logger.
-func (Keeper) Logger(ctx context.Context) log.Logger {
+func (*Keeper) Logger(ctx context.Context) log.Logger {
 	sdkCtx := sdk.UnwrapSDKContext(ctx) // TODO: https://github.com/cosmos/ibc-go/issues/5917
 	return sdkCtx.Logger().With("module", "x/"+exported.ModuleName+"/"+types.SubModuleName)
 }
@@ -194,3 +197,57 @@ func (k *Keeper) DeleteAsyncPacket(ctx context.Context, clientID string, sequenc
 		panic(err)
 	}
 }
+
+// GetAllPacketCommitmentsForClient returns all stored PacketCommitments objects for a specified
+// client ID.
+func (k *Keeper) GetAllPacketCommitmentsForClient(ctx context.Context, clientID string) []types.PacketState {
+	store := runtime.KVStoreAdapter(k.KVStoreService.OpenKVStore(ctx))
+	storePrefix := hostv2.PacketCommitmentPrefixKey(clientID)
+	iterator := storetypes.KVStorePrefixIterator(store, storePrefix)
+
+	var commitments []types.PacketState
+	for ; iterator.Valid(); iterator.Next() {
+		sequenceBz := bytes.TrimPrefix(iterator.Key(), storePrefix)
+		sequence := sdk.BigEndianToUint64(sequenceBz)
+		state := types.NewPacketState(clientID, sequence, iterator.Value())
+
+		commitments = append(commitments, state)
+	}
+	return commitments
+}
+
+// GetAllPacketAcknowledgementsForClient returns all stored PacketAcknowledgements objects for a specified
+// client ID.
+func (k *Keeper) GetAllPacketAcknowledgementsForClient(ctx context.Context, clientID string) []types.PacketState {
+	store := runtime.KVStoreAdapter(k.KVStoreService.OpenKVStore(ctx))
+	storePrefix := hostv2.PacketAcknowledgementPrefixKey(clientID)
+	iterator := storetypes.KVStorePrefixIterator(store, storePrefix)
+
+	var acknowledgements []types.PacketState
+	for ; iterator.Valid(); iterator.Next() {
+		sequenceBz := bytes.TrimPrefix(iterator.Key(), storePrefix)
+		sequence := sdk.BigEndianToUint64(sequenceBz)
+		state := types.NewPacketState(clientID, sequence, iterator.Value())
+
+		acknowledgements = append(acknowledgements, state)
+	}
+	return acknowledgements
+}
+
+// GetAllPacketReceiptsForClient returns all stored PacketReceipts objects for a specified
+// client ID.
+func (k *Keeper) GetAllPacketReceiptsForClient(ctx context.Context, clientID string) []types.PacketState {
+	store := runtime.KVStoreAdapter(k.KVStoreService.OpenKVStore(ctx))
+	storePrefix := hostv2.PacketReceiptPrefixKey(clientID)
+	iterator := storetypes.KVStorePrefixIterator(store, storePrefix)
+
+	var receipts []types.PacketState
+	for ; iterator.Valid(); iterator.Next() {
+		sequenceBz := bytes.TrimPrefix(iterator.Key(), storePrefix)
+		sequence := sdk.BigEndianToUint64(sequenceBz)
+		state := types.NewPacketState(clientID, sequence, iterator.Value())
+
+		receipts = append(receipts, state)
+	}
+	return receipts
+}

modules/core/04-channel/v2/module.go

@@ -1,23 +1,99 @@
 package client
 
 import (
+	"context"
+	"encoding/json"
+	"fmt"
+
 	"github.com/spf13/cobra"
 
+	"cosmossdk.io/core/appmodule"
+
+	"github.com/cosmos/cosmos-sdk/codec"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/module"
+
 	"github.com/cosmos/ibc-go/v9/modules/core/04-channel/v2/client/cli"
+	"github.com/cosmos/ibc-go/v9/modules/core/04-channel/v2/keeper"
 	"github.com/cosmos/ibc-go/v9/modules/core/04-channel/v2/types"
 )
 
+var (
+	_ appmodule.AppModule = (*AppModule)(nil)
+
+	_ module.AppModule  = (*AppModule)(nil)
+	_ module.HasGenesis = (*AppModule)(nil)
+)
+
+// AppModule represents the AppModule for this module
+type AppModule struct {
+	cdc    codec.Codec
+	keeper *keeper.Keeper
+}
+
+func (AppModule) IsAppModule() {}
+
+func (AppModule) IsOnePerModuleType() {}
+
+func (m AppModule) DefaultGenesis() json.RawMessage {
+	gs := types.DefaultGenesisState()
+	return m.cdc.MustMarshalJSON(&gs)
+}
+
+func (m AppModule) ValidateGenesis(data json.RawMessage) error {
+	gs := &types.GenesisState{}
+	err := m.cdc.UnmarshalJSON(data, gs)
+	if err != nil {
+		return err
+	}
+
+	return gs.Validate()
+}
+
+func (m AppModule) InitGenesis(ctx context.Context, data json.RawMessage) error {
+	gs := &types.GenesisState{}
+	err := m.cdc.UnmarshalJSON(data, gs)
+	if err != nil {
+		return fmt.Errorf("failed to unmarshal genesis state: %w", err)
+	}
+
+	sdkCtx := sdk.UnwrapSDKContext(ctx)
+	m.keeper.InitGenesis(sdkCtx, *gs)
+
+	return nil
+}
+
+func (m AppModule) ExportGenesis(ctx context.Context) (json.RawMessage, error) {
+	sdkCtx := sdk.UnwrapSDKContext(ctx)
+	gs := m.keeper.ExportGenesis(sdkCtx)
+	return json.Marshal(gs)
+}
+
 // Name returns the IBC channel/v2 name
-func Name() string {
+func (AppModule) Name() string {
 	return types.SubModuleName
 }
 
+func Name() string {
+	return AppModule{}.Name()
+}
+
 // GetQueryCmd returns the root query command for IBC channels v2.
-func GetQueryCmd() *cobra.Command {
+func (AppModule) GetQueryCmd() *cobra.Command {
 	return cli.GetQueryCmd()
 }
 
+// GetQueryCmd returns the root query command for IBC channels v2.
+func GetQueryCmd() *cobra.Command {
+	return AppModule{}.GetQueryCmd()
+}
+
 // GetTxCmd returns the root tx command for IBC channels v2.
-func GetTxCmd() *cobra.Command {
+func (AppModule) GetTxCmd() *cobra.Command {
 	return cli.NewTxCmd()
 }
+
+// GetTxCmd returns the root tx command for IBC channels v2.
+func GetTxCmd() *cobra.Command {
+	return AppModule{}.GetTxCmd()
+}

modules/core/04-channel/v2/types/expected_keepers.go

@@ -25,4 +25,6 @@ type ClientKeeper interface {
 	GetClientConsensusState(ctx context.Context, clientID string, height exported.Height) (exported.ConsensusState, bool)
 	// GetClientCounterparty returns the counterpartyInfo given a clientID
 	GetClientCounterparty(ctx context.Context, clientID string) (clienttypes.CounterpartyInfo, bool)
+	// GetAllGenesisClients returns all the clients in state with their client ids returned as IdentifiedClientState
+	GetAllGenesisClients(ctx context.Context) clienttypes.IdentifiedClientStates
 }

modules/core/24-host/v2/packet_keys.go

@@ -6,22 +6,28 @@ import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
+const (
+	PacketCommitmentBasePrefix      = byte(1)
+	PacketReceiptBasePrefix         = byte(2)
+	PacketAcknowledgementBasePrefix = byte(3)
+)
+
 // PacketCommitmentPrefixKey returns the store key prefix under which packet commitments for a particular channel are stored.
 // channelID must be a generated identifier, not provided externally so key collisions are not possible.
 func PacketCommitmentPrefixKey(channelID string) []byte {
-	return append([]byte(channelID), byte(1))
+	return append([]byte(channelID), PacketCommitmentBasePrefix)
 }
 
 // PacketCommitmentKey returns the store key of under which a packet commitment is stored.
 // channelID must be a generated identifier, not provided externally so key collisions are not possible.
 func PacketCommitmentKey(channelID string, sequence uint64) []byte {
-	return append(PacketCommitmentPrefixKey((channelID)), sdk.Uint64ToBigEndian(sequence)...)
+	return append(PacketCommitmentPrefixKey(channelID), sdk.Uint64ToBigEndian(sequence)...)
 }
 
 // PacketReceiptPrefixKey returns the store key prefix under which packet receipts for a particular channel are stored.
 // channelID must be a generated identifier, not provided externally so key collisions are not possible.
 func PacketReceiptPrefixKey(channelID string) []byte {
-	return append([]byte(channelID), byte(2))
+	return append([]byte(channelID), PacketReceiptBasePrefix)
 }
 
 // PacketReceiptKey returns the store key of under which a packet receipt is stored.
@@ -33,7 +39,7 @@ func PacketReceiptKey(channelID string, sequence uint64) []byte {
 // PacketAcknowledgementPrefixKey returns the store key prefix under which packet acknowledgements for a particular channel are stored.
 // channelID must be a generated identifier, not provided externally so key collisions are not possible.
 func PacketAcknowledgementPrefixKey(channelID string) []byte {
-	return append([]byte(channelID), byte(3))
+	return append([]byte(channelID), PacketAcknowledgementBasePrefix)
 }
 
 // PacketAcknowledgementKey returns the store key of under which a packet acknowledgement is stored.