feat(nexus): add the wasm message route (#2023)

app/app.go

@@ -300,7 +300,6 @@ func NewAxelarApp(
 	messageRouter := nexusTypes.NewMessageRouter().
 		AddRoute(evmTypes.ModuleName, evmKeeper.NewMessageRoute()).
 		AddRoute(axelarnetTypes.ModuleName, axelarnetKeeper.NewMessageRoute(getKeeper[axelarnetKeeper.Keeper](keepers), getKeeper[axelarnetKeeper.IBCKeeper](keepers), getKeeper[feegrantkeeper.Keeper](keepers), axelarbankkeeper.NewBankKeeper(getKeeper[bankkeeper.BaseKeeper](keepers)), getKeeper[nexusKeeper.Keeper](keepers), getKeeper[authkeeper.AccountKeeper](keepers)))
-	getKeeperAsRef[nexusKeeper.Keeper](keepers).SetMessageRouter(messageRouter)
 
 	axelarnetModule := axelarnet.NewAppModule(getKeeper[axelarnetKeeper.Keeper](keepers), getKeeper[nexusKeeper.Keeper](keepers), axelarbankkeeper.NewBankKeeper(getKeeper[bankkeeper.BaseKeeper](keepers)), getKeeper[authkeeper.AccountKeeper](keepers), getKeeper[axelarnetKeeper.IBCKeeper](keepers), transferStack, rateLimiter, logger)
 
@@ -355,10 +354,15 @@ func NewAxelarApp(
 		ibcRouter.AddRoute(wasm.ModuleName, wasmStack)
 
 		// set the contract keeper for the Ics20WasmHooks
-		wasmHooks.ContractKeeper = wasmkeeper.NewDefaultPermissionKeeper(wasmK)
+		contractKeeper := wasmkeeper.NewDefaultPermissionKeeper(wasmK)
+		wasmHooks.ContractKeeper = contractKeeper
 		setKeeper(keepers, wasmK)
+
+		messageRouter.AddRoute(wasm.ModuleName, nexusKeeper.NewMessageRoute(getKeeperAsRef[nexusKeeper.Keeper](keepers), getKeeper[authkeeper.AccountKeeper](keepers), contractKeeper))
 	}
 
+	getKeeperAsRef[nexusKeeper.Keeper](keepers).SetMessageRouter(messageRouter)
+
 	// Finalize the IBC router
 	getKeeperAsRef[ibckeeper.Keeper](keepers).SetRouter(ibcRouter)
 

x/axelarnet/keeper/message_route.go

@@ -15,6 +15,7 @@ import (
 // for IBC execution
 const gasCost = storetypes.Gas(1000000)
 
+// NewMessageRoute creates a new message route
 func NewMessageRoute(
 	keeper Keeper,
 	ibcK types.IBCKeeper,

x/evm/keeper/message_route.go

@@ -10,6 +10,7 @@ import (
 // for commands approval
 const gasCost = storetypes.Gas(10000000)
 
+// NewMessageRoute creates a new message route
 func NewMessageRoute() nexus.MessageRoute {
 	return func(ctx sdk.Context, _ nexus.RoutingContext, _ nexus.GeneralMessage) error {
 		ctx.GasMeter().ConsumeGas(gasCost, "execute-message")

x/nexus/exported/types.go

@@ -336,3 +336,16 @@ func (m GeneralMessage) Type() MessageType {
 
 	return TypeGeneralMessageWithToken
 }
+
+// FromGeneralMessage returns a WasmMessage from a GeneralMessage
+func FromGeneralMessage(msg GeneralMessage) WasmMessage {
+	return WasmMessage{
+		SourceChain:        msg.GetSourceChain(),
+		SourceAddress:      msg.GetSourceAddress(),
+		DestinationChain:   msg.GetDestinationChain(),
+		DestinationAddress: msg.GetDestinationAddress(),
+		PayloadHash:        msg.PayloadHash,
+		SourceTxID:         msg.SourceTxID,
+		SourceTxIndex:      msg.SourceTxIndex,
+	}
+}

x/nexus/keeper/general_message.go

@@ -8,6 +8,7 @@ import (
 	"github.com/CosmWasm/wasmd/x/wasm"
 	"github.com/cosmos/cosmos-sdk/store/prefix"
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
 	"github.com/cosmos/cosmos-sdk/types/query"
 
 	"github.com/axelarnetwork/axelar-core/utils"
@@ -244,5 +245,11 @@ func (k Keeper) RouteMessage(ctx sdk.Context, id string, routingCtx ...exported.
 	if len(routingCtx) == 0 {
 		routingCtx = []exported.RoutingContext{{}}
 	}
-	return k.getMessageRouter().Route(ctx, routingCtx[0], funcs.MustOk(k.GetMessage(ctx, id)))
+
+	msg := funcs.MustOk(k.GetMessage(ctx, id))
+	if err := k.getMessageRouter().Route(ctx, routingCtx[0], msg); err != nil {
+		return sdkerrors.Wrapf(err, "failed to route message %s to the %s module", id, msg.Recipient.Chain.Module)
+	}
+
+	return nil
 }

x/nexus/keeper/msg_dispatcher.go

@@ -19,8 +19,6 @@ import (
 
 var _ wasmkeeper.Messenger = (*Messenger)(nil)
 
-type request = exported.WasmMessage
-
 type Messenger struct {
 	types.Nexus
 }
@@ -32,14 +30,14 @@ func NewMessenger(nexus types.Nexus) Messenger {
 
 // DispatchMsg decodes the messages from the cosmowasm gateway and routes them to the nexus module if possible
 func (m Messenger) DispatchMsg(ctx sdk.Context, contractAddr sdk.AccAddress, _ string, msg wasmvmtypes.CosmosMsg) (events []sdk.Event, data [][]byte, err error) {
-	req := request{}
+	req := exported.WasmMessage{}
 	if err := json.Unmarshal(msg.Custom, &req); err != nil {
 		return nil, nil, sdkerrors.Wrap(wasmtypes.ErrUnknownMsg, err.Error())
 	}
 
 	gateway := m.GetParams(ctx).Gateway
 
-	if len(gateway) == 0 {
+	if gateway.Empty() {
 		return nil, nil, fmt.Errorf("gateway is not set")
 	}
 

x/nexus/keeper/wasm_message_route.go

@@ -0,0 +1,40 @@
+package keeper
+
+import (
+	"encoding/json"
+	"fmt"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
+	"github.com/axelarnetwork/axelar-core/x/nexus/exported"
+	types "github.com/axelarnetwork/axelar-core/x/nexus/types"
+)
+
+type request struct {
+	RouteMessages []exported.WasmMessage `json:"route_messages"`
+}
+
+// NewMessageRoute creates a new message route
+func NewMessageRoute(nexus types.Nexus, account types.AccountKeeper, wasm types.WasmKeeper) exported.MessageRoute {
+	return func(ctx sdk.Context, _ exported.RoutingContext, msg exported.GeneralMessage) error {
+		if msg.Asset != nil {
+			return fmt.Errorf("asset transfer is not supported")
+		}
+
+		gateway := nexus.GetParams(ctx).Gateway
+		if gateway.Empty() {
+			return fmt.Errorf("gateway is not set")
+		}
+
+		bz, err := json.Marshal(request{RouteMessages: []exported.WasmMessage{exported.FromGeneralMessage(msg)}})
+		if err != nil {
+			return nil
+		}
+
+		if _, err := wasm.Execute(ctx, gateway, account.GetModuleAddress(types.ModuleName), bz, sdk.NewCoins()); err != nil {
+			return err
+		}
+
+		return nil
+	}
+}

x/nexus/keeper/wasm_message_route_test.go

@@ -0,0 +1,101 @@
+package keeper_test
+
+import (
+	"encoding/json"
+	"testing"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/stretchr/testify/assert"
+	"github.com/tendermint/tendermint/libs/log"
+	tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
+
+	"github.com/axelarnetwork/axelar-core/testutils/fake"
+	"github.com/axelarnetwork/axelar-core/testutils/rand"
+	nexus "github.com/axelarnetwork/axelar-core/x/nexus/exported"
+	"github.com/axelarnetwork/axelar-core/x/nexus/keeper"
+	"github.com/axelarnetwork/axelar-core/x/nexus/types"
+	"github.com/axelarnetwork/axelar-core/x/nexus/types/mock"
+	. "github.com/axelarnetwork/utils/test"
+)
+
+func TestNewMessageRoute(t *testing.T) {
+	var (
+		ctx   sdk.Context
+		route nexus.MessageRoute
+		msg   nexus.GeneralMessage
+
+		nexusK   *mock.NexusMock
+		accountK *mock.AccountKeeperMock
+		wasmK    *mock.WasmKeeperMock
+		gateway  sdk.AccAddress
+	)
+
+	givenMessageRoute := Given("the message route", func() {
+		ctx = sdk.NewContext(fake.NewMultiStore(), tmproto.Header{}, false, log.TestingLogger())
+
+		nexusK = &mock.NexusMock{}
+		accountK = &mock.AccountKeeperMock{}
+		wasmK = &mock.WasmKeeperMock{}
+
+		route = keeper.NewMessageRoute(nexusK, accountK, wasmK)
+	})
+
+	givenMessageRoute.
+		When("the gateway is not set", func() {
+			nexusK.GetParamsFunc = func(ctx sdk.Context) types.Params { return types.DefaultParams() }
+		}).
+		Then("should return error", func(t *testing.T) {
+			assert.ErrorContains(t, route(ctx, nexus.RoutingContext{}, msg), "gateway is not set")
+		}).
+		Run(t)
+
+	givenMessageRoute.
+		When("the gateway is set", func() {
+			nexusK.GetParamsFunc = func(ctx sdk.Context) types.Params {
+				gateway = rand.AccAddr()
+
+				params := types.DefaultParams()
+				params.Gateway = gateway
+
+				return params
+			}
+		}).
+		Branch(
+			When("the message has an asset", func() {
+				msg = randMsg(nexus.Processing, true)
+			}).
+				Then("should return error", func(t *testing.T) {
+					assert.ErrorContains(t, route(ctx, nexus.RoutingContext{}, msg), "asset transfer is not supported")
+				}),
+
+			When("the message has no asset", func() {
+				msg = randMsg(nexus.Processing)
+			}).
+				Then("should execute the wasm message", func(t *testing.T) {
+					moduleAddr := rand.AccAddr()
+					accountK.GetModuleAddressFunc = func(_ string) sdk.AccAddress { return moduleAddr }
+
+					wasmK.ExecuteFunc = func(_ sdk.Context, _, _ sdk.AccAddress, _ []byte, _ sdk.Coins) ([]byte, error) {
+						return nil, nil
+					}
+
+					assert.NoError(t, route(ctx, nexus.RoutingContext{}, msg))
+
+					assert.Len(t, wasmK.ExecuteCalls(), 1)
+					assert.Equal(t, wasmK.ExecuteCalls()[0].ContractAddress, gateway)
+					assert.Equal(t, wasmK.ExecuteCalls()[0].Caller, moduleAddr)
+					assert.Empty(t, wasmK.ExecuteCalls()[0].Coins)
+
+					type req struct {
+						RouteMessages []nexus.WasmMessage `json:"route_messages"`
+					}
+
+					var actual req
+					assert.NoError(t, json.Unmarshal(wasmK.ExecuteCalls()[0].Msg, &actual))
+					assert.Len(t, actual.RouteMessages, 1)
+					assert.Equal(t, nexus.FromGeneralMessage(msg), actual.RouteMessages[0])
+				}),
+		).
+		Run(t)
+
+}

x/nexus/types/expected_keepers.go

@@ -3,6 +3,7 @@ package types
 import (
 	"time"
 
+	wasmtypes "github.com/CosmWasm/wasmd/x/wasm/types"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
 	"github.com/tendermint/tendermint/libs/log"
@@ -13,7 +14,7 @@ import (
 	snapshot "github.com/axelarnetwork/axelar-core/x/snapshot/exported"
 )
 
-//go:generate moq -out ./mock/expected_keepers.go -pkg mock . Nexus Snapshotter AxelarnetKeeper RewardKeeper SlashingKeeper
+//go:generate moq -out ./mock/expected_keepers.go -pkg mock . Nexus Snapshotter AxelarnetKeeper RewardKeeper SlashingKeeper WasmKeeper AccountKeeper
 
 // Nexus provides functionality to manage cross-chain transfers
 type Nexus interface {
@@ -73,3 +74,13 @@ type RewardKeeper interface {
 type SlashingKeeper interface {
 	IsTombstoned(ctx sdk.Context, consAddr sdk.ConsAddress) bool
 }
+
+// WasmKeeper provides functionality to manage wasm contracts
+type WasmKeeper interface {
+	wasmtypes.ContractOpsKeeper
+}
+
+// AccountKeeper provides functionality to get account keeper
+type AccountKeeper interface {
+	GetModuleAddress(moduleName string) sdk.AccAddress
+}