* Refactor fees function * Implement suggestions * wip test * checksenderbalance tests * wip deductcost test * Tests fixes * Merge master and fix conflicts * Merging account and bank interfaces * Update x/evm/keeper/utils.go Co-authored-by: Hanchon <guillermo.paoletti@gmail.com> Co-authored-by: Federico Kunze Küllmer <31522760+fedekunze@users.noreply.github.com>
148 lines
4.8 KiB
Go
148 lines
4.8 KiB
Go
package ante
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime/debug"
|
|
|
|
"github.com/palantir/stacktrace"
|
|
|
|
tmlog "github.com/tendermint/tendermint/libs/log"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
"github.com/cosmos/cosmos-sdk/types/tx/signing"
|
|
authante "github.com/cosmos/cosmos-sdk/x/auth/ante"
|
|
authsigning "github.com/cosmos/cosmos-sdk/x/auth/signing"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
|
|
channelkeeper "github.com/cosmos/ibc-go/modules/core/04-channel/keeper"
|
|
ibcante "github.com/cosmos/ibc-go/modules/core/ante"
|
|
|
|
"github.com/tharsis/ethermint/crypto/ethsecp256k1"
|
|
evmtypes "github.com/tharsis/ethermint/x/evm/types"
|
|
)
|
|
|
|
const (
|
|
secp256k1VerifyCost uint64 = 21000
|
|
)
|
|
|
|
// NewAnteHandler returns an ante handler responsible for attempting to route an
|
|
// Ethereum or SDK transaction to an internal ante handler for performing
|
|
// transaction-level processing (e.g. fee payment, signature verification) before
|
|
// being passed onto it's respective handler.
|
|
func NewAnteHandler(
|
|
ak evmtypes.AccountKeeper,
|
|
bankKeeper evmtypes.BankKeeper,
|
|
evmKeeper EVMKeeper,
|
|
feeGrantKeeper authante.FeegrantKeeper,
|
|
channelKeeper channelkeeper.Keeper,
|
|
signModeHandler authsigning.SignModeHandler,
|
|
) sdk.AnteHandler {
|
|
return func(
|
|
ctx sdk.Context, tx sdk.Tx, sim bool,
|
|
) (newCtx sdk.Context, err error) {
|
|
var anteHandler sdk.AnteHandler
|
|
|
|
defer Recover(ctx.Logger(), &err)
|
|
|
|
txWithExtensions, ok := tx.(authante.HasExtensionOptionsTx)
|
|
if ok {
|
|
opts := txWithExtensions.GetExtensionOptions()
|
|
if len(opts) > 0 {
|
|
switch typeURL := opts[0].GetTypeUrl(); typeURL {
|
|
case "/ethermint.evm.v1.ExtensionOptionsEthereumTx":
|
|
// handle as *evmtypes.MsgEthereumTx
|
|
|
|
anteHandler = sdk.ChainAnteDecorators(
|
|
NewEthSetUpContextDecorator(), // outermost AnteDecorator. SetUpContext must be called first
|
|
authante.NewMempoolFeeDecorator(),
|
|
authante.NewTxTimeoutHeightDecorator(),
|
|
authante.NewValidateMemoDecorator(ak),
|
|
NewEthValidateBasicDecorator(),
|
|
NewEthSigVerificationDecorator(evmKeeper),
|
|
NewEthAccountVerificationDecorator(ak, bankKeeper, evmKeeper),
|
|
NewEthNonceVerificationDecorator(ak),
|
|
NewEthGasConsumeDecorator(ak, bankKeeper, evmKeeper),
|
|
NewCanTransferDecorator(evmKeeper),
|
|
NewEthIncrementSenderSequenceDecorator(ak), // innermost AnteDecorator.
|
|
)
|
|
|
|
default:
|
|
return ctx, stacktrace.Propagate(
|
|
sdkerrors.Wrap(sdkerrors.ErrUnknownExtensionOptions, typeURL),
|
|
"rejecting tx with unsupported extension option",
|
|
)
|
|
}
|
|
|
|
return anteHandler(ctx, tx, sim)
|
|
}
|
|
}
|
|
|
|
// handle as totally normal Cosmos SDK tx
|
|
|
|
switch tx.(type) {
|
|
case sdk.Tx:
|
|
anteHandler = sdk.ChainAnteDecorators(
|
|
authante.NewSetUpContextDecorator(), // outermost AnteDecorator. SetUpContext must be called first
|
|
authante.NewRejectExtensionOptionsDecorator(),
|
|
authante.NewMempoolFeeDecorator(),
|
|
authante.NewValidateBasicDecorator(),
|
|
authante.NewTxTimeoutHeightDecorator(),
|
|
authante.NewValidateMemoDecorator(ak),
|
|
ibcante.NewAnteDecorator(channelKeeper),
|
|
authante.NewConsumeGasForTxSizeDecorator(ak),
|
|
authante.NewSetPubKeyDecorator(ak), // SetPubKeyDecorator must be called before all signature verification decorators
|
|
authante.NewValidateSigCountDecorator(ak),
|
|
authante.NewDeductFeeDecorator(ak, bankKeeper, feeGrantKeeper),
|
|
authante.NewSigGasConsumeDecorator(ak, DefaultSigVerificationGasConsumer),
|
|
authante.NewSigVerificationDecorator(ak, signModeHandler),
|
|
authante.NewIncrementSequenceDecorator(ak), // innermost AnteDecorator
|
|
)
|
|
default:
|
|
return ctx, stacktrace.Propagate(
|
|
sdkerrors.Wrapf(sdkerrors.ErrUnknownRequest, "invalid transaction type: %T", tx),
|
|
"transaction is not an SDK tx",
|
|
)
|
|
}
|
|
|
|
return anteHandler(ctx, tx, sim)
|
|
}
|
|
}
|
|
|
|
func Recover(logger tmlog.Logger, err *error) {
|
|
if r := recover(); r != nil {
|
|
*err = sdkerrors.Wrapf(sdkerrors.ErrPanic, "%v", r)
|
|
|
|
if e, ok := r.(error); ok {
|
|
logger.Error(
|
|
"ante handler panicked",
|
|
"error", e,
|
|
"stack trace", string(debug.Stack()),
|
|
)
|
|
} else {
|
|
logger.Error(
|
|
"ante handler panicked",
|
|
"recover", fmt.Sprintf("%v", r),
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
var _ authante.SignatureVerificationGasConsumer = DefaultSigVerificationGasConsumer
|
|
|
|
// DefaultSigVerificationGasConsumer is the default implementation of SignatureVerificationGasConsumer. It consumes gas
|
|
// for signature verification based upon the public key type. The cost is fetched from the given params and is matched
|
|
// by the concrete type.
|
|
func DefaultSigVerificationGasConsumer(
|
|
meter sdk.GasMeter, sig signing.SignatureV2, params authtypes.Params,
|
|
) error {
|
|
// support for ethereum ECDSA secp256k1 keys
|
|
_, ok := sig.PubKey.(*ethsecp256k1.PubKey)
|
|
if ok {
|
|
meter.ConsumeGas(secp256k1VerifyCost, "ante verify: eth_secp256k1")
|
|
return nil
|
|
}
|
|
|
|
return authante.DefaultSigVerificationGasConsumer(meter, sig, params)
|
|
}
|