fix: ics20 memo handler denom

x/uibc/ics20.go

@@ -3,9 +3,38 @@
 import (
 	"github.com/cosmos/cosmos-sdk/codec/types"
 	"github.com/cosmos/cosmos-sdk/types/tx"
+	ics20types "github.com/cosmos/ibc-go/v7/modules/apps/transfer/types"
+	ibcexported "github.com/cosmos/ibc-go/v7/modules/core/exported"
 )
 
 // UnpackInterfaces implements UnpackInterfacesMessage.UnpackInterfaces
 func (m ICS20Memo) UnpackInterfaces(unpacker types.AnyUnpacker) error {
 	return tx.UnpackInterfaces(unpacker, m.Messages)
 }
+
+// MustExtractDenomFromPacketOnRecv takes a packet with a valid ICS20 token data in the Data field and returns the
+// denom as represented in the local chain.
+// If the data cannot be unmarshalled this function will panic
+func MustExtractDenomFromPacketOnRecv(packet ibcexported.PacketI, denom string) string {
+	if ics20types.ReceiverChainIsSource(packet.GetSourcePort(), packet.GetSourceChannel(), denom) {
+		// if we receive back a token, that was originally sent from UMEE, then we need to remove
+		// prefix added by the sender chain: port/channel/base_denom -> base_denom.
+
+		voucherPrefix := ics20types.GetDenomPrefix(packet.GetSourcePort(), packet.GetSourceChannel())
+		unprefixedDenom := denom[len(voucherPrefix):]
+
+		// coin denomination used in sending from the escrow address
+		denom = unprefixedDenom
+
+		// The denomination used to send the coins is either the native denom or the hash of the path
+		// if the denomination is not native.
+		denomTrace := ics20types.ParseDenomTrace(unprefixedDenom)
+		if !denomTrace.IsNativeDenom() {
+			denom = denomTrace.IBCDenom()
+		}
+	} else {
+		prefixedDenom := ics20types.GetDenomPrefix(packet.GetDestPort(), packet.GetDestChannel()) + denom
+		denom = ics20types.ParseDenomTrace(prefixedDenom).IBCDenom()
+	}
+	return denom
+}

x/uibc/quota/quota.go

@@ -6,7 +6,6 @@
 
 	sdkmath "cosmossdk.io/math"
 	sdk "github.com/cosmos/cosmos-sdk/types"
-	ics20types "github.com/cosmos/ibc-go/v7/modules/apps/transfer/types"
 	channeltypes "github.com/cosmos/ibc-go/v7/modules/core/04-channel/types"
 	"github.com/cosmos/ibc-go/v7/modules/core/exported"
 
@@ -231,30 +230,7 @@
 // RecordIBCInflow will save the inflow amount if token is registered otherwise it will skip
 func (k Keeper) RecordIBCInflow(packet channeltypes.Packet, denom, amount string,
 ) exported.Acknowledgement {
-	// if chain is recevier and sender chain is source then we need create ibc_denom (ibc/hash(channel,denom)).
-	if ics20types.SenderChainIsSource(packet.GetSourcePort(), packet.GetSourceChannel(), denom) {
-		// SendPacket did not prefix the denom, so we must prefix denom here
-		// NOTE: sourcePrefix already contains the trailing "/"
-		sourcePrefix := ics20types.GetDenomPrefix(packet.GetDestPort(), packet.GetDestChannel())
-		prefixedDenom := sourcePrefix + denom
-		denom = ics20types.ParseDenomTrace(prefixedDenom).IBCDenom()
-	} else {
-		// if we receive back a token, that was originally sent from UMEE, then we need to fetch the native denom
-		// receive denom(port/channel/base_denom) to base_denom
-
-		// remove prefix added by the sender chain
-		voucherPrefix := ics20types.GetDenomPrefix(packet.GetSourcePort(), packet.GetSourceChannel())
-		unprefixedDenom := denom[len(voucherPrefix):]
-		// coin denomination used in sending from the escrow address
-		denom = unprefixedDenom
-		// The denomination used to send the coins is either the native denom or the hash of the path
-		// if the denomination is not native.
-		denomTrace := ics20types.ParseDenomTrace(unprefixedDenom)
-		if !denomTrace.IsNativeDenom() {
-			denom = denomTrace.IBCDenom()
-		}
-	}
-
+	denom = uibc.MustExtractDenomFromPacketOnRecv(packet, denom)
 	ts, err := k.leverage.GetTokenSettings(*k.ctx, denom)
 	if err != nil {
 		if ltypes.ErrNotRegisteredToken.Is(err) {

x/uibc/uics20/ibc_module.go

@@ -57,7 +57,7 @@
 	if ftData.Memo != "" {
 		logger := recvPacketLogger(&ctx)
 		mh := MemoHandler{im.cdc, im.leverage}
-		if err := mh.onRecvPacket(&ctx, ftData); err != nil {
+		if err := mh.onRecvPacket(&ctx, packet, ftData); err != nil {
 			logger.Error("can't handle ICS20 memo", "err", err)
 		}
 	}

x/uibc/uics20/memo_handler.go

@@ -9,6 +9,7 @@
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/tx"
 	ics20types "github.com/cosmos/ibc-go/v7/modules/apps/transfer/types"
+	ibcexported "github.com/cosmos/ibc-go/v7/modules/core/exported"
 
 	ltypes "github.com/umee-network/umee/v6/x/leverage/types"
 	"github.com/umee-network/umee/v6/x/uibc"
@@ -19,7 +20,7 @@
 	leverage ltypes.MsgServer
 }
 
-func (mh MemoHandler) onRecvPacket(ctx *sdk.Context, ftData ics20types.FungibleTokenPacketData) error {
+func (mh MemoHandler) onRecvPacket(ctx *sdk.Context, packet ibcexported.PacketI, ftData ics20types.FungibleTokenPacketData) error {
 	msgs, err := deserializeMemoMsgs(mh.cdc, []byte(ftData.Memo))
 	if err != nil {
 		recvPacketLogger(ctx).Debug("Can't deserialize ICS20 memo for hook execution", "err", err)
@@ -36,7 +37,8 @@
 	if !ok {
 		return fmt.Errorf("can't parse transfer amount: %s [%w]", ftData.Amount, err)
 	}
-	return mh.dispatchMemoMsgs(ctx, receiver, sdk.NewCoin(ftData.Denom, amount), msgs)
+	ibcDenom := uibc.MustExtractDenomFromPacketOnRecv(packet, ftData.Denom)
+	return mh.dispatchMemoMsgs(ctx, receiver, sdk.NewCoin(ibcDenom, amount), msgs)
 }
 
 // runs messages encoded in the ICS20 memo.