67 lines
2.3 KiB
Go
67 lines
2.3 KiB
Go
package ante
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
clientexported "github.com/cosmos/cosmos-sdk/x/ibc/02-client/exported"
|
|
clientkeeper "github.com/cosmos/cosmos-sdk/x/ibc/02-client/keeper"
|
|
channelkeeper "github.com/cosmos/cosmos-sdk/x/ibc/04-channel/keeper"
|
|
channeltypes "github.com/cosmos/cosmos-sdk/x/ibc/04-channel/types"
|
|
)
|
|
|
|
// ProofVerificationDecorator handles messages that contains application specific packet types,
|
|
// including MsgPacket, MsgAcknowledgement, MsgTimeout.
|
|
// MsgUpdateClients are also handled here to perform atomic multimsg transaction
|
|
type ProofVerificationDecorator struct {
|
|
clientKeeper clientkeeper.Keeper
|
|
channelKeeper channelkeeper.Keeper
|
|
}
|
|
|
|
// NewProofVerificationDecorator constructs new ProofverificationDecorator
|
|
func NewProofVerificationDecorator(clientKeeper clientkeeper.Keeper, channelKeeper channelkeeper.Keeper) ProofVerificationDecorator {
|
|
return ProofVerificationDecorator{
|
|
clientKeeper: clientKeeper,
|
|
channelKeeper: channelKeeper,
|
|
}
|
|
}
|
|
|
|
// AnteHandle executes MsgUpdateClient, MsgPacket, MsgAcknowledgement, MsgTimeout.
|
|
// The packet execution messages are then passed to the respective application handlers.
|
|
func (pvr ProofVerificationDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simulate bool, next sdk.AnteHandler) (sdk.Context, error) {
|
|
for _, msg := range tx.GetMsgs() {
|
|
var err error
|
|
switch msg := msg.(type) {
|
|
case clientexported.MsgUpdateClient:
|
|
_, err = pvr.clientKeeper.UpdateClient(ctx, msg.GetClientID(), msg.GetHeader())
|
|
case *channeltypes.MsgPacket:
|
|
err = pvr.channelKeeper.RecvPacket(ctx, msg.Packet, msg.Proof, msg.ProofHeight)
|
|
case *channeltypes.MsgAcknowledgement:
|
|
err = pvr.channelKeeper.AcknowledgePacket(ctx, msg.Packet, msg.Acknowledgement, msg.Proof, msg.ProofHeight)
|
|
case *channeltypes.MsgTimeout:
|
|
err = pvr.channelKeeper.TimeoutPacket(ctx, msg.Packet, msg.Proof, msg.ProofHeight, msg.NextSequenceRecv)
|
|
default:
|
|
// don't emit sender event for other msg types
|
|
continue
|
|
}
|
|
|
|
attributes := make([]sdk.Attribute, len(msg.GetSigners()))
|
|
|
|
for i, signer := range msg.GetSigners() {
|
|
attributes[i] = sdk.NewAttribute(sdk.AttributeKeySender, signer.String())
|
|
}
|
|
|
|
ctx.EventManager().EmitEvents(sdk.Events{
|
|
sdk.NewEvent(
|
|
sdk.EventTypeMessage,
|
|
attributes...,
|
|
),
|
|
})
|
|
|
|
if err != nil {
|
|
return ctx, err
|
|
}
|
|
|
|
}
|
|
|
|
return next(ctx, tx, simulate)
|
|
}
|