refactor(x/auth/middleware)!: tx middleware to support pluggable feemarket module (#11413)
Closes: #11415 ## Description - Create `FeeMarket` interface, and move exiting static fee logic into `StaticFeeMarket` implementation. - Merged `MempoolFeeMiddleware` and `TxPriorityMiddleware` into `DeductFeeMiddleware`, so we can deduct fee based on the check result. - ~~Support extension options in `Tx.AuthInfo`, so feemarket module can extend the tx fields.~~ Keep in TxBody - No Ledger support in v0.46 --- ### Author Checklist *All items are required. Please add a note to the item if the item is not applicable and please add links to any relevant follow up issues.* I have... - [ ] included the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title - [ ] added `!` to the type prefix if API or client breaking change - [ ] targeted the correct branch (see [PR Targeting](https://github.com/cosmos/cosmos-sdk/blob/master/CONTRIBUTING.md#pr-targeting)) - [ ] provided a link to the relevant issue or specification - [ ] followed the guidelines for [building modules](https://github.com/cosmos/cosmos-sdk/blob/master/docs/building-modules) - [ ] included the necessary unit and integration [tests](https://github.com/cosmos/cosmos-sdk/blob/master/CONTRIBUTING.md#testing) - [ ] added a changelog entry to `CHANGELOG.md` - [ ] included comments for [documenting Go code](https://blog.golang.org/godoc) - [ ] updated the relevant documentation or specification - [ ] reviewed "Files changed" and left comments if necessary - [ ] confirmed all CI checks have passed ### Reviewers Checklist *All items are required. Please add a note if the item is not applicable and please add your handle next to the items reviewed if you only reviewed selected items.* I have... - [ ] confirmed the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title - [ ] confirmed `!` in the type prefix if API or client breaking change - [ ] confirmed all author checklist items have been addressed - [ ] reviewed state machine logic - [ ] reviewed API design and naming - [ ] reviewed documentation is accurate - [ ] reviewed tests and test coverage - [ ] manually tested (if applicable)
This commit is contained in:
parent
4e92b6294d
commit
e75c734c83
|
@ -148,7 +148,7 @@ Ref: https://keepachangelog.com/en/1.0.0/
|
||||||
* [\#11274](https://github.com/cosmos/cosmos-sdk/pull/11274) `types/errors.New` now is an alias for `types/errors.Register` and should only be used in initialization code.
|
* [\#11274](https://github.com/cosmos/cosmos-sdk/pull/11274) `types/errors.New` now is an alias for `types/errors.Register` and should only be used in initialization code.
|
||||||
* (authz)[\#11060](https://github.com/cosmos/cosmos-sdk/pull/11060) `authz.NewMsgGrant` `expiration` is now a pointer. When `nil` is used then no expiration will be set (grant won't expire).
|
* (authz)[\#11060](https://github.com/cosmos/cosmos-sdk/pull/11060) `authz.NewMsgGrant` `expiration` is now a pointer. When `nil` is used then no expiration will be set (grant won't expire).
|
||||||
* (x/distribution)[\#11457](https://github.com/cosmos/cosmos-sdk/pull/11457) Add amount field to `distr.MsgWithdrawDelegatorRewardResponse` and `distr.MsgWithdrawValidatorCommissionResponse`.
|
* (x/distribution)[\#11457](https://github.com/cosmos/cosmos-sdk/pull/11457) Add amount field to `distr.MsgWithdrawDelegatorRewardResponse` and `distr.MsgWithdrawValidatorCommissionResponse`.
|
||||||
|
* (x/auth/middleware) [#11413](https://github.com/cosmos/cosmos-sdk/pull/11413) Refactor tx middleware to be extensible on tx fee logic. Merged `MempoolFeeMiddleware` and `TxPriorityMiddleware` functionalities into `DeductFeeMiddleware`, make the logic extensible using the `TxFeeChecker` option, the current fee logic is preserved by the default `checkTxFeeWithValidatorMinGasPrices` implementation. Change `RejectExtensionOptionsMiddleware` to `NewExtensionOptionsMiddleware` which is extensible with the `ExtensionOptionChecker` option. Unpack the tx extension options `Any`s to interface `TxExtensionOptionI`.
|
||||||
|
|
||||||
### Client Breaking Changes
|
### Client Breaking Changes
|
||||||
|
|
||||||
|
|
|
@ -6,6 +6,7 @@ import (
|
||||||
"github.com/cosmos/cosmos-sdk/codec/types"
|
"github.com/cosmos/cosmos-sdk/codec/types"
|
||||||
sdk "github.com/cosmos/cosmos-sdk/types"
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||||||
"github.com/cosmos/cosmos-sdk/types/msgservice"
|
"github.com/cosmos/cosmos-sdk/types/msgservice"
|
||||||
|
tx "github.com/cosmos/cosmos-sdk/types/tx"
|
||||||
)
|
)
|
||||||
|
|
||||||
func NewTestInterfaceRegistry() types.InterfaceRegistry {
|
func NewTestInterfaceRegistry() types.InterfaceRegistry {
|
||||||
|
@ -31,6 +32,10 @@ func RegisterInterfaces(registry types.InterfaceRegistry) {
|
||||||
(*HasHasAnimalI)(nil),
|
(*HasHasAnimalI)(nil),
|
||||||
&HasHasAnimal{},
|
&HasHasAnimal{},
|
||||||
)
|
)
|
||||||
|
registry.RegisterImplementations(
|
||||||
|
(*tx.TxExtensionOptionI)(nil),
|
||||||
|
&Cat{},
|
||||||
|
)
|
||||||
|
|
||||||
msgservice.RegisterMsgServiceDesc(registry, &_Msg_serviceDesc)
|
msgservice.RegisterMsgServiceDesc(registry, &_Msg_serviceDesc)
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,21 @@
|
||||||
|
package tx
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/cosmos/cosmos-sdk/codec/types"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TxExtensionOptionI defines the interface for tx extension options
|
||||||
|
type TxExtensionOptionI interface{}
|
||||||
|
|
||||||
|
// unpackTxExtensionOptionsI unpacks Any's to TxExtensionOptionI's.
|
||||||
|
func unpackTxExtensionOptionsI(unpacker types.AnyUnpacker, anys []*types.Any) error {
|
||||||
|
for _, any := range anys {
|
||||||
|
var opt TxExtensionOptionI
|
||||||
|
err := unpacker.UnpackAny(any, &opt)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -173,7 +173,19 @@ func (t *Tx) UnpackInterfaces(unpacker codectypes.AnyUnpacker) error {
|
||||||
|
|
||||||
// UnpackInterfaces implements the UnpackInterfaceMessages.UnpackInterfaces method
|
// UnpackInterfaces implements the UnpackInterfaceMessages.UnpackInterfaces method
|
||||||
func (m *TxBody) UnpackInterfaces(unpacker codectypes.AnyUnpacker) error {
|
func (m *TxBody) UnpackInterfaces(unpacker codectypes.AnyUnpacker) error {
|
||||||
return UnpackInterfaces(unpacker, m.Messages)
|
if err := UnpackInterfaces(unpacker, m.Messages); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := unpackTxExtensionOptionsI(unpacker, m.ExtensionOptions); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := unpackTxExtensionOptionsI(unpacker, m.NonCriticalExtensionOptions); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// UnpackInterfaces implements the UnpackInterfaceMessages.UnpackInterfaces method
|
// UnpackInterfaces implements the UnpackInterfaceMessages.UnpackInterfaces method
|
||||||
|
@ -200,4 +212,6 @@ func RegisterInterfaces(registry codectypes.InterfaceRegistry) {
|
||||||
|
|
||||||
registry.RegisterInterface("cosmos.tx.v1beta1.Tx", (*sdk.Tx)(nil))
|
registry.RegisterInterface("cosmos.tx.v1beta1.Tx", (*sdk.Tx)(nil))
|
||||||
registry.RegisterImplementations((*sdk.Tx)(nil), &Tx{})
|
registry.RegisterImplementations((*sdk.Tx)(nil), &Tx{})
|
||||||
|
|
||||||
|
registry.RegisterInterface("cosmos.tx.v1beta1.TxExtensionOptionI", (*TxExtensionOptionI)(nil))
|
||||||
}
|
}
|
||||||
|
|
|
@ -72,7 +72,7 @@ func (s *MWTestSuite) TestBranchStore() {
|
||||||
middleware.NewTxDecoderMiddleware(s.clientCtx.TxConfig.TxDecoder()),
|
middleware.NewTxDecoderMiddleware(s.clientCtx.TxConfig.TxDecoder()),
|
||||||
middleware.GasTxMiddleware,
|
middleware.GasTxMiddleware,
|
||||||
middleware.RecoveryTxMiddleware,
|
middleware.RecoveryTxMiddleware,
|
||||||
middleware.DeductFeeMiddleware(s.app.AccountKeeper, s.app.BankKeeper, s.app.FeeGrantKeeper),
|
middleware.DeductFeeMiddleware(s.app.AccountKeeper, s.app.BankKeeper, s.app.FeeGrantKeeper, nil),
|
||||||
middleware.IncrementSequenceMiddleware(s.app.AccountKeeper),
|
middleware.IncrementSequenceMiddleware(s.app.AccountKeeper),
|
||||||
middleware.WithBranchedStore,
|
middleware.WithBranchedStore,
|
||||||
middleware.ConsumeBlockGasMiddleware,
|
middleware.ConsumeBlockGasMiddleware,
|
||||||
|
|
|
@ -14,27 +14,44 @@ type HasExtensionOptionsTx interface {
|
||||||
GetNonCriticalExtensionOptions() []*codectypes.Any
|
GetNonCriticalExtensionOptions() []*codectypes.Any
|
||||||
}
|
}
|
||||||
|
|
||||||
type rejectExtensionOptionsTxHandler struct {
|
// ExtensionOptionChecker is a function that returns true if the extension option is accepted.
|
||||||
next tx.Handler
|
type ExtensionOptionChecker func(*codectypes.Any) bool
|
||||||
|
|
||||||
|
// rejectExtensionOption is the default extension check that reject all tx
|
||||||
|
// extensions.
|
||||||
|
func rejectExtensionOption(*codectypes.Any) bool {
|
||||||
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// RejectExtensionOptionsMiddleware creates a new rejectExtensionOptionsMiddleware.
|
type rejectExtensionOptionsTxHandler struct {
|
||||||
// rejectExtensionOptionsMiddleware is a middleware that rejects all extension
|
next tx.Handler
|
||||||
// options which can optionally be included in protobuf transactions. Users that
|
checker ExtensionOptionChecker
|
||||||
// need extension options should create a custom middleware chain that handles
|
}
|
||||||
// needed extension options properly and rejects unknown ones.
|
|
||||||
func RejectExtensionOptionsMiddleware(txh tx.Handler) tx.Handler {
|
// NewExtensionOptionsMiddleware creates a new middleware that rejects all extension
|
||||||
return rejectExtensionOptionsTxHandler{
|
// options which can optionally be included in protobuf transactions that don't pass the checker.
|
||||||
next: txh,
|
// Users that need extension options should pass a custom checker that returns true for the
|
||||||
|
// needed extension options.
|
||||||
|
func NewExtensionOptionsMiddleware(checker ExtensionOptionChecker) tx.Middleware {
|
||||||
|
if checker == nil {
|
||||||
|
checker = rejectExtensionOption
|
||||||
|
}
|
||||||
|
return func(txh tx.Handler) tx.Handler {
|
||||||
|
return rejectExtensionOptionsTxHandler{
|
||||||
|
next: txh,
|
||||||
|
checker: checker,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var _ tx.Handler = rejectExtensionOptionsTxHandler{}
|
var _ tx.Handler = rejectExtensionOptionsTxHandler{}
|
||||||
|
|
||||||
func checkExtOpts(tx sdk.Tx) error {
|
func checkExtOpts(tx sdk.Tx, checker ExtensionOptionChecker) error {
|
||||||
if hasExtOptsTx, ok := tx.(HasExtensionOptionsTx); ok {
|
if hasExtOptsTx, ok := tx.(HasExtensionOptionsTx); ok {
|
||||||
if len(hasExtOptsTx.GetExtensionOptions()) != 0 {
|
for _, opt := range hasExtOptsTx.GetExtensionOptions() {
|
||||||
return sdkerrors.ErrUnknownExtensionOptions
|
if !checker(opt) {
|
||||||
|
return sdkerrors.ErrUnknownExtensionOptions
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -43,7 +60,7 @@ func checkExtOpts(tx sdk.Tx) error {
|
||||||
|
|
||||||
// CheckTx implements tx.Handler.CheckTx.
|
// CheckTx implements tx.Handler.CheckTx.
|
||||||
func (txh rejectExtensionOptionsTxHandler) CheckTx(ctx context.Context, req tx.Request, checkReq tx.RequestCheckTx) (tx.Response, tx.ResponseCheckTx, error) {
|
func (txh rejectExtensionOptionsTxHandler) CheckTx(ctx context.Context, req tx.Request, checkReq tx.RequestCheckTx) (tx.Response, tx.ResponseCheckTx, error) {
|
||||||
if err := checkExtOpts(req.Tx); err != nil {
|
if err := checkExtOpts(req.Tx, txh.checker); err != nil {
|
||||||
return tx.Response{}, tx.ResponseCheckTx{}, err
|
return tx.Response{}, tx.ResponseCheckTx{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -52,7 +69,7 @@ func (txh rejectExtensionOptionsTxHandler) CheckTx(ctx context.Context, req tx.R
|
||||||
|
|
||||||
// DeliverTx implements tx.Handler.DeliverTx.
|
// DeliverTx implements tx.Handler.DeliverTx.
|
||||||
func (txh rejectExtensionOptionsTxHandler) DeliverTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
func (txh rejectExtensionOptionsTxHandler) DeliverTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
||||||
if err := checkExtOpts(req.Tx); err != nil {
|
if err := checkExtOpts(req.Tx, txh.checker); err != nil {
|
||||||
return tx.Response{}, err
|
return tx.Response{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -61,7 +78,7 @@ func (txh rejectExtensionOptionsTxHandler) DeliverTx(ctx context.Context, req tx
|
||||||
|
|
||||||
// SimulateTx implements tx.Handler.SimulateTx method.
|
// SimulateTx implements tx.Handler.SimulateTx method.
|
||||||
func (txh rejectExtensionOptionsTxHandler) SimulateTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
func (txh rejectExtensionOptionsTxHandler) SimulateTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
||||||
if err := checkExtOpts(req.Tx); err != nil {
|
if err := checkExtOpts(req.Tx, txh.checker); err != nil {
|
||||||
return tx.Response{}, err
|
return tx.Response{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -2,6 +2,7 @@ package middleware_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/cosmos/cosmos-sdk/codec/types"
|
"github.com/cosmos/cosmos-sdk/codec/types"
|
||||||
|
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
|
||||||
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
||||||
sdk "github.com/cosmos/cosmos-sdk/types"
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||||||
typestx "github.com/cosmos/cosmos-sdk/types/tx"
|
typestx "github.com/cosmos/cosmos-sdk/types/tx"
|
||||||
|
@ -9,28 +10,45 @@ import (
|
||||||
"github.com/cosmos/cosmos-sdk/x/auth/tx"
|
"github.com/cosmos/cosmos-sdk/x/auth/tx"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (s *MWTestSuite) TestRejectExtensionOptionsMiddleware() {
|
func (s *MWTestSuite) TestExtensionOptionsMiddleware() {
|
||||||
ctx := s.SetupTest(true) // setup
|
testCases := []struct {
|
||||||
txBuilder := s.clientCtx.TxConfig.NewTxBuilder()
|
msg string
|
||||||
|
allow bool
|
||||||
txHandler := middleware.ComposeMiddlewares(noopTxHandler, middleware.RejectExtensionOptionsMiddleware)
|
}{
|
||||||
|
{"allow extension", true},
|
||||||
// no extension options should not trigger an error
|
{"reject extension", false},
|
||||||
theTx := txBuilder.GetTx()
|
|
||||||
_, _, err := txHandler.CheckTx(sdk.WrapSDKContext(ctx), typestx.Request{Tx: theTx}, typestx.RequestCheckTx{})
|
|
||||||
s.Require().NoError(err)
|
|
||||||
|
|
||||||
extOptsTxBldr, ok := txBuilder.(tx.ExtensionOptionsTxBuilder)
|
|
||||||
if !ok {
|
|
||||||
// if we can't set extension options, this middleware doesn't apply and we're done
|
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
for _, tc := range testCases {
|
||||||
|
s.Run(tc.msg, func() {
|
||||||
|
ctx := s.SetupTest(true) // setup
|
||||||
|
txBuilder := s.clientCtx.TxConfig.NewTxBuilder()
|
||||||
|
|
||||||
// setting any extension option should cause an error
|
txHandler := middleware.ComposeMiddlewares(noopTxHandler, middleware.NewExtensionOptionsMiddleware(func(_ *codectypes.Any) bool {
|
||||||
any, err := types.NewAnyWithValue(testdata.NewTestMsg())
|
return tc.allow
|
||||||
s.Require().NoError(err)
|
}))
|
||||||
extOptsTxBldr.SetExtensionOptions(any)
|
|
||||||
theTx = txBuilder.GetTx()
|
// no extension options should not trigger an error
|
||||||
_, _, err = txHandler.CheckTx(sdk.WrapSDKContext(ctx), typestx.Request{Tx: theTx}, typestx.RequestCheckTx{})
|
theTx := txBuilder.GetTx()
|
||||||
s.Require().EqualError(err, "unknown extension options")
|
_, _, err := txHandler.CheckTx(sdk.WrapSDKContext(ctx), typestx.Request{Tx: theTx}, typestx.RequestCheckTx{})
|
||||||
|
s.Require().NoError(err)
|
||||||
|
|
||||||
|
extOptsTxBldr, ok := txBuilder.(tx.ExtensionOptionsTxBuilder)
|
||||||
|
if !ok {
|
||||||
|
// if we can't set extension options, this middleware doesn't apply and we're done
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// set an extension option and check
|
||||||
|
any, err := types.NewAnyWithValue(testdata.NewTestMsg())
|
||||||
|
s.Require().NoError(err)
|
||||||
|
extOptsTxBldr.SetExtensionOptions(any)
|
||||||
|
theTx = txBuilder.GetTx()
|
||||||
|
_, _, err = txHandler.CheckTx(sdk.WrapSDKContext(ctx), typestx.Request{Tx: theTx}, typestx.RequestCheckTx{})
|
||||||
|
if tc.allow {
|
||||||
|
s.Require().NoError(err)
|
||||||
|
} else {
|
||||||
|
s.Require().EqualError(err, "unknown extension options")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,73 +10,9 @@ import (
|
||||||
"github.com/cosmos/cosmos-sdk/x/auth/types"
|
"github.com/cosmos/cosmos-sdk/x/auth/types"
|
||||||
)
|
)
|
||||||
|
|
||||||
var _ tx.Handler = mempoolFeeTxHandler{}
|
// TxFeeChecker check if the provided fee is enough and returns the effective fee and tx priority,
|
||||||
|
// the effective fee should be deducted later, and the priority should be returned in abci response.
|
||||||
type mempoolFeeTxHandler struct {
|
type TxFeeChecker func(ctx sdk.Context, tx sdk.Tx) (sdk.Coins, int64, error)
|
||||||
next tx.Handler
|
|
||||||
}
|
|
||||||
|
|
||||||
// MempoolFeeMiddleware will check if the transaction's fee is at least as large
|
|
||||||
// as the local validator's minimum gasFee (defined in validator config).
|
|
||||||
// If fee is too low, middleware returns error and tx is rejected from mempool.
|
|
||||||
// Note this only applies when ctx.CheckTx = true
|
|
||||||
// If fee is high enough or not CheckTx, then call next middleware
|
|
||||||
// CONTRACT: Tx must implement FeeTx to use MempoolFeeMiddleware
|
|
||||||
func MempoolFeeMiddleware(txh tx.Handler) tx.Handler {
|
|
||||||
return mempoolFeeTxHandler{
|
|
||||||
next: txh,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// CheckTx implements tx.Handler.CheckTx. It is responsible for determining if a
|
|
||||||
// transaction's fees meet the required minimum of the processing node. Note, a
|
|
||||||
// node can have zero fees set as the minimum. If non-zero minimum fees are set
|
|
||||||
// and the transaction does not meet the minimum, the transaction is rejected.
|
|
||||||
//
|
|
||||||
// Recall, a transaction's fee is determined by ceil(minGasPrice * gasLimit).
|
|
||||||
func (txh mempoolFeeTxHandler) CheckTx(ctx context.Context, req tx.Request, checkReq tx.RequestCheckTx) (tx.Response, tx.ResponseCheckTx, error) {
|
|
||||||
sdkCtx := sdk.UnwrapSDKContext(ctx)
|
|
||||||
|
|
||||||
feeTx, ok := req.Tx.(sdk.FeeTx)
|
|
||||||
if !ok {
|
|
||||||
return tx.Response{}, tx.ResponseCheckTx{}, sdkerrors.Wrap(sdkerrors.ErrTxDecode, "Tx must be a FeeTx")
|
|
||||||
}
|
|
||||||
|
|
||||||
feeCoins := feeTx.GetFee()
|
|
||||||
gas := feeTx.GetGas()
|
|
||||||
|
|
||||||
// Ensure that the provided fees meet a minimum threshold for the validator,
|
|
||||||
// if this is a CheckTx. This is only for local mempool purposes, and thus
|
|
||||||
// is only ran on check tx.
|
|
||||||
minGasPrices := sdkCtx.MinGasPrices()
|
|
||||||
if !minGasPrices.IsZero() {
|
|
||||||
requiredFees := make(sdk.Coins, len(minGasPrices))
|
|
||||||
|
|
||||||
// Determine the required fees by multiplying each required minimum gas
|
|
||||||
// price by the gas limit, where fee = ceil(minGasPrice * gasLimit).
|
|
||||||
glDec := sdk.NewDec(int64(gas))
|
|
||||||
for i, gp := range minGasPrices {
|
|
||||||
fee := gp.Amount.Mul(glDec)
|
|
||||||
requiredFees[i] = sdk.NewCoin(gp.Denom, fee.Ceil().RoundInt())
|
|
||||||
}
|
|
||||||
|
|
||||||
if !feeCoins.IsAnyGTE(requiredFees) {
|
|
||||||
return tx.Response{}, tx.ResponseCheckTx{}, sdkerrors.Wrapf(sdkerrors.ErrInsufficientFee, "insufficient fees; got: %s required: %s", feeCoins, requiredFees)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return txh.next.CheckTx(ctx, req, checkReq)
|
|
||||||
}
|
|
||||||
|
|
||||||
// DeliverTx implements tx.Handler.DeliverTx.
|
|
||||||
func (txh mempoolFeeTxHandler) DeliverTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
|
||||||
return txh.next.DeliverTx(ctx, req)
|
|
||||||
}
|
|
||||||
|
|
||||||
// SimulateTx implements tx.Handler.SimulateTx.
|
|
||||||
func (txh mempoolFeeTxHandler) SimulateTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
|
||||||
return txh.next.SimulateTx(ctx, req)
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ tx.Handler = deductFeeTxHandler{}
|
var _ tx.Handler = deductFeeTxHandler{}
|
||||||
|
|
||||||
|
@ -84,6 +20,7 @@ type deductFeeTxHandler struct {
|
||||||
accountKeeper AccountKeeper
|
accountKeeper AccountKeeper
|
||||||
bankKeeper types.BankKeeper
|
bankKeeper types.BankKeeper
|
||||||
feegrantKeeper FeegrantKeeper
|
feegrantKeeper FeegrantKeeper
|
||||||
|
txFeeChecker TxFeeChecker
|
||||||
next tx.Handler
|
next tx.Handler
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -91,19 +28,22 @@ type deductFeeTxHandler struct {
|
||||||
// If the first signer does not have the funds to pay for the fees, return with InsufficientFunds error
|
// If the first signer does not have the funds to pay for the fees, return with InsufficientFunds error
|
||||||
// Call next middleware if fees successfully deducted
|
// Call next middleware if fees successfully deducted
|
||||||
// CONTRACT: Tx must implement FeeTx interface to use deductFeeTxHandler
|
// CONTRACT: Tx must implement FeeTx interface to use deductFeeTxHandler
|
||||||
func DeductFeeMiddleware(ak AccountKeeper, bk types.BankKeeper, fk FeegrantKeeper) tx.Middleware {
|
func DeductFeeMiddleware(ak AccountKeeper, bk types.BankKeeper, fk FeegrantKeeper, tfc TxFeeChecker) tx.Middleware {
|
||||||
|
if tfc == nil {
|
||||||
|
tfc = checkTxFeeWithValidatorMinGasPrices
|
||||||
|
}
|
||||||
return func(txh tx.Handler) tx.Handler {
|
return func(txh tx.Handler) tx.Handler {
|
||||||
return deductFeeTxHandler{
|
return deductFeeTxHandler{
|
||||||
accountKeeper: ak,
|
accountKeeper: ak,
|
||||||
bankKeeper: bk,
|
bankKeeper: bk,
|
||||||
feegrantKeeper: fk,
|
feegrantKeeper: fk,
|
||||||
|
txFeeChecker: tfc,
|
||||||
next: txh,
|
next: txh,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dfd deductFeeTxHandler) checkDeductFee(ctx context.Context, sdkTx sdk.Tx) error {
|
func (dfd deductFeeTxHandler) checkDeductFee(ctx sdk.Context, sdkTx sdk.Tx, fee sdk.Coins) error {
|
||||||
sdkCtx := sdk.UnwrapSDKContext(ctx)
|
|
||||||
feeTx, ok := sdkTx.(sdk.FeeTx)
|
feeTx, ok := sdkTx.(sdk.FeeTx)
|
||||||
if !ok {
|
if !ok {
|
||||||
return sdkerrors.Wrap(sdkerrors.ErrTxDecode, "Tx must be a FeeTx")
|
return sdkerrors.Wrap(sdkerrors.ErrTxDecode, "Tx must be a FeeTx")
|
||||||
|
@ -113,10 +53,8 @@ func (dfd deductFeeTxHandler) checkDeductFee(ctx context.Context, sdkTx sdk.Tx)
|
||||||
return fmt.Errorf("Fee collector module account (%s) has not been set", types.FeeCollectorName)
|
return fmt.Errorf("Fee collector module account (%s) has not been set", types.FeeCollectorName)
|
||||||
}
|
}
|
||||||
|
|
||||||
fee := feeTx.GetFee()
|
|
||||||
feePayer := feeTx.FeePayer()
|
feePayer := feeTx.FeePayer()
|
||||||
feeGranter := feeTx.FeeGranter()
|
feeGranter := feeTx.FeeGranter()
|
||||||
|
|
||||||
deductFeesFrom := feePayer
|
deductFeesFrom := feePayer
|
||||||
|
|
||||||
// if feegranter set deduct fee from feegranter account.
|
// if feegranter set deduct fee from feegranter account.
|
||||||
|
@ -125,7 +63,7 @@ func (dfd deductFeeTxHandler) checkDeductFee(ctx context.Context, sdkTx sdk.Tx)
|
||||||
if dfd.feegrantKeeper == nil {
|
if dfd.feegrantKeeper == nil {
|
||||||
return sdkerrors.ErrInvalidRequest.Wrap("fee grants are not enabled")
|
return sdkerrors.ErrInvalidRequest.Wrap("fee grants are not enabled")
|
||||||
} else if !feeGranter.Equals(feePayer) {
|
} else if !feeGranter.Equals(feePayer) {
|
||||||
err := dfd.feegrantKeeper.UseGrantedFees(sdkCtx, feeGranter, feePayer, fee, sdkTx.GetMsgs())
|
err := dfd.feegrantKeeper.UseGrantedFees(ctx, feeGranter, feePayer, fee, sdkTx.GetMsgs())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return sdkerrors.Wrapf(err, "%s does not not allow to pay fees for %s", feeGranter, feePayer)
|
return sdkerrors.Wrapf(err, "%s does not not allow to pay fees for %s", feeGranter, feePayer)
|
||||||
}
|
}
|
||||||
|
@ -134,39 +72,52 @@ func (dfd deductFeeTxHandler) checkDeductFee(ctx context.Context, sdkTx sdk.Tx)
|
||||||
deductFeesFrom = feeGranter
|
deductFeesFrom = feeGranter
|
||||||
}
|
}
|
||||||
|
|
||||||
deductFeesFromAcc := dfd.accountKeeper.GetAccount(sdkCtx, deductFeesFrom)
|
deductFeesFromAcc := dfd.accountKeeper.GetAccount(ctx, deductFeesFrom)
|
||||||
if deductFeesFromAcc == nil {
|
if deductFeesFromAcc == nil {
|
||||||
return sdkerrors.ErrUnknownAddress.Wrapf("fee payer address: %s does not exist", deductFeesFrom)
|
return sdkerrors.ErrUnknownAddress.Wrapf("fee payer address: %s does not exist", deductFeesFrom)
|
||||||
}
|
}
|
||||||
|
|
||||||
// deduct the fees
|
// deduct the fees
|
||||||
if !feeTx.GetFee().IsZero() {
|
if !fee.IsZero() {
|
||||||
err := DeductFees(dfd.bankKeeper, sdkCtx, deductFeesFromAcc, feeTx.GetFee())
|
err := DeductFees(dfd.bankKeeper, ctx, deductFeesFromAcc, fee)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
events := sdk.Events{sdk.NewEvent(sdk.EventTypeTx,
|
events := sdk.Events{sdk.NewEvent(sdk.EventTypeTx,
|
||||||
sdk.NewAttribute(sdk.AttributeKeyFee, feeTx.GetFee().String()),
|
sdk.NewAttribute(sdk.AttributeKeyFee, fee.String()),
|
||||||
)}
|
)}
|
||||||
sdkCtx.EventManager().EmitEvents(events)
|
ctx.EventManager().EmitEvents(events)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// CheckTx implements tx.Handler.CheckTx.
|
// CheckTx implements tx.Handler.CheckTx.
|
||||||
func (dfd deductFeeTxHandler) CheckTx(ctx context.Context, req tx.Request, checkReq tx.RequestCheckTx) (tx.Response, tx.ResponseCheckTx, error) {
|
func (dfd deductFeeTxHandler) CheckTx(ctx context.Context, req tx.Request, checkReq tx.RequestCheckTx) (tx.Response, tx.ResponseCheckTx, error) {
|
||||||
if err := dfd.checkDeductFee(ctx, req.Tx); err != nil {
|
sdkCtx := sdk.UnwrapSDKContext(ctx)
|
||||||
|
fee, priority, err := dfd.txFeeChecker(sdkCtx, req.Tx)
|
||||||
|
if err != nil {
|
||||||
|
return tx.Response{}, tx.ResponseCheckTx{}, err
|
||||||
|
}
|
||||||
|
if err := dfd.checkDeductFee(sdkCtx, req.Tx, fee); err != nil {
|
||||||
return tx.Response{}, tx.ResponseCheckTx{}, err
|
return tx.Response{}, tx.ResponseCheckTx{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return dfd.next.CheckTx(ctx, req, checkReq)
|
res, checkRes, err := dfd.next.CheckTx(ctx, req, checkReq)
|
||||||
|
checkRes.Priority = priority
|
||||||
|
|
||||||
|
return res, checkRes, err
|
||||||
}
|
}
|
||||||
|
|
||||||
// DeliverTx implements tx.Handler.DeliverTx.
|
// DeliverTx implements tx.Handler.DeliverTx.
|
||||||
func (dfd deductFeeTxHandler) DeliverTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
func (dfd deductFeeTxHandler) DeliverTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
||||||
if err := dfd.checkDeductFee(ctx, req.Tx); err != nil {
|
sdkCtx := sdk.UnwrapSDKContext(ctx)
|
||||||
|
fee, _, err := dfd.txFeeChecker(sdkCtx, req.Tx)
|
||||||
|
if err != nil {
|
||||||
|
return tx.Response{}, err
|
||||||
|
}
|
||||||
|
if err := dfd.checkDeductFee(sdkCtx, req.Tx, fee); err != nil {
|
||||||
return tx.Response{}, err
|
return tx.Response{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -174,7 +125,12 @@ func (dfd deductFeeTxHandler) DeliverTx(ctx context.Context, req tx.Request) (tx
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dfd deductFeeTxHandler) SimulateTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
func (dfd deductFeeTxHandler) SimulateTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
||||||
if err := dfd.checkDeductFee(ctx, req.Tx); err != nil {
|
sdkCtx := sdk.UnwrapSDKContext(ctx)
|
||||||
|
fee, _, err := dfd.txFeeChecker(sdkCtx, req.Tx)
|
||||||
|
if err != nil {
|
||||||
|
return tx.Response{}, err
|
||||||
|
}
|
||||||
|
if err := dfd.checkDeductFee(sdkCtx, req.Tx, fee); err != nil {
|
||||||
return tx.Response{}, err
|
return tx.Response{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -13,14 +13,21 @@ func (s *MWTestSuite) TestEnsureMempoolFees() {
|
||||||
ctx := s.SetupTest(true) // setup
|
ctx := s.SetupTest(true) // setup
|
||||||
txBuilder := s.clientCtx.TxConfig.NewTxBuilder()
|
txBuilder := s.clientCtx.TxConfig.NewTxBuilder()
|
||||||
|
|
||||||
txHandler := middleware.ComposeMiddlewares(noopTxHandler, middleware.MempoolFeeMiddleware)
|
txHandler := middleware.ComposeMiddlewares(noopTxHandler, middleware.DeductFeeMiddleware(
|
||||||
|
s.app.AccountKeeper,
|
||||||
|
s.app.BankKeeper,
|
||||||
|
s.app.FeeGrantKeeper,
|
||||||
|
nil,
|
||||||
|
))
|
||||||
|
|
||||||
// keys and addresses
|
// keys and addresses
|
||||||
priv1, _, addr1 := testdata.KeyTestPubAddr()
|
priv1, _, addr1 := testdata.KeyTestPubAddr()
|
||||||
|
|
||||||
// msg and signatures
|
// msg and signatures
|
||||||
msg := testdata.NewTestMsg(addr1)
|
msg := testdata.NewTestMsg(addr1)
|
||||||
feeAmount := testdata.NewTestFeeAmount()
|
atomCoin := sdk.NewCoin("atom", sdk.NewInt(150))
|
||||||
|
apeCoin := sdk.NewInt64Coin("ape", 1500000)
|
||||||
|
feeAmount := sdk.NewCoins(apeCoin, atomCoin)
|
||||||
gasLimit := testdata.NewTestGasLimit()
|
gasLimit := testdata.NewTestGasLimit()
|
||||||
s.Require().NoError(txBuilder.SetMsgs(msg))
|
s.Require().NoError(txBuilder.SetMsgs(msg))
|
||||||
txBuilder.SetFeeAmount(feeAmount)
|
txBuilder.SetFeeAmount(feeAmount)
|
||||||
|
@ -39,16 +46,23 @@ func (s *MWTestSuite) TestEnsureMempoolFees() {
|
||||||
_, _, err = txHandler.CheckTx(sdk.WrapSDKContext(ctx), tx.Request{Tx: testTx}, tx.RequestCheckTx{})
|
_, _, err = txHandler.CheckTx(sdk.WrapSDKContext(ctx), tx.Request{Tx: testTx}, tx.RequestCheckTx{})
|
||||||
s.Require().NotNil(err, "Middleware should have errored on too low fee for local gasPrice")
|
s.Require().NotNil(err, "Middleware should have errored on too low fee for local gasPrice")
|
||||||
|
|
||||||
// txHandler should not error since we do not check minGasPrice in DeliverTx
|
// txHandler should fail since we also check minGasPrice in DeliverTx
|
||||||
_, err = txHandler.DeliverTx(sdk.WrapSDKContext(ctx), tx.Request{Tx: testTx})
|
_, err = txHandler.DeliverTx(sdk.WrapSDKContext(ctx), tx.Request{Tx: testTx})
|
||||||
s.Require().Nil(err, "MempoolFeeMiddleware returned error in DeliverTx")
|
s.Require().Error(err, "MempoolFeeMiddleware don't error in DeliverTx")
|
||||||
|
|
||||||
atomPrice = sdk.NewDecCoinFromDec("atom", sdk.NewDec(0).Quo(sdk.NewDec(100000)))
|
atomPrice = sdk.NewDecCoinFromDec("atom", sdk.NewDec(0).Quo(sdk.NewDec(100000)))
|
||||||
lowGasPrice := []sdk.DecCoin{atomPrice}
|
lowGasPrice := []sdk.DecCoin{atomPrice}
|
||||||
ctx = ctx.WithMinGasPrices(lowGasPrice)
|
ctx = ctx.WithMinGasPrices(lowGasPrice)
|
||||||
|
|
||||||
_, _, err = txHandler.CheckTx(sdk.WrapSDKContext(ctx), tx.Request{Tx: testTx}, tx.RequestCheckTx{})
|
// Set account with sufficient funds
|
||||||
|
acc := s.app.AccountKeeper.NewAccountWithAddress(ctx, addr1)
|
||||||
|
s.app.AccountKeeper.SetAccount(ctx, acc)
|
||||||
|
err = testutil.FundAccount(s.app.BankKeeper, ctx, addr1, feeAmount)
|
||||||
|
s.Require().NoError(err)
|
||||||
|
|
||||||
|
_, checkTxRes, err := txHandler.CheckTx(sdk.WrapSDKContext(ctx), tx.Request{Tx: testTx}, tx.RequestCheckTx{})
|
||||||
s.Require().Nil(err, "Middleware should not have errored on fee higher than local gasPrice")
|
s.Require().Nil(err, "Middleware should not have errored on fee higher than local gasPrice")
|
||||||
|
s.Require().Equal(atomCoin.Amount.Int64(), checkTxRes.Priority, "priority should be atom amount")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *MWTestSuite) TestDeductFees() {
|
func (s *MWTestSuite) TestDeductFees() {
|
||||||
|
@ -60,6 +74,7 @@ func (s *MWTestSuite) TestDeductFees() {
|
||||||
s.app.AccountKeeper,
|
s.app.AccountKeeper,
|
||||||
s.app.BankKeeper,
|
s.app.BankKeeper,
|
||||||
s.app.FeeGrantKeeper,
|
s.app.FeeGrantKeeper,
|
||||||
|
nil,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -36,6 +36,7 @@ func (s *MWTestSuite) TestDeductFeesNoDelegation() {
|
||||||
s.app.AccountKeeper,
|
s.app.AccountKeeper,
|
||||||
s.app.BankKeeper,
|
s.app.BankKeeper,
|
||||||
s.app.FeeGrantKeeper,
|
s.app.FeeGrantKeeper,
|
||||||
|
nil,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -43,11 +43,13 @@ type TxHandlerOptions struct {
|
||||||
LegacyRouter sdk.Router
|
LegacyRouter sdk.Router
|
||||||
MsgServiceRouter *MsgServiceRouter
|
MsgServiceRouter *MsgServiceRouter
|
||||||
|
|
||||||
AccountKeeper AccountKeeper
|
AccountKeeper AccountKeeper
|
||||||
BankKeeper types.BankKeeper
|
BankKeeper types.BankKeeper
|
||||||
FeegrantKeeper FeegrantKeeper
|
FeegrantKeeper FeegrantKeeper
|
||||||
SignModeHandler authsigning.SignModeHandler
|
SignModeHandler authsigning.SignModeHandler
|
||||||
SigGasConsumer func(meter sdk.GasMeter, sig signing.SignatureV2, params types.Params) error
|
SigGasConsumer func(meter sdk.GasMeter, sig signing.SignatureV2, params types.Params) error
|
||||||
|
ExtensionOptionChecker ExtensionOptionChecker
|
||||||
|
TxFeeChecker TxFeeChecker
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewDefaultTxHandler defines a TxHandler middleware stacks that should work
|
// NewDefaultTxHandler defines a TxHandler middleware stacks that should work
|
||||||
|
@ -74,6 +76,16 @@ func NewDefaultTxHandler(options TxHandlerOptions) (tx.Handler, error) {
|
||||||
sigGasConsumer = DefaultSigVerificationGasConsumer
|
sigGasConsumer = DefaultSigVerificationGasConsumer
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var extensionOptionChecker = options.ExtensionOptionChecker
|
||||||
|
if extensionOptionChecker == nil {
|
||||||
|
extensionOptionChecker = rejectExtensionOption
|
||||||
|
}
|
||||||
|
|
||||||
|
var txFeeChecker = options.TxFeeChecker
|
||||||
|
if txFeeChecker == nil {
|
||||||
|
txFeeChecker = checkTxFeeWithValidatorMinGasPrices
|
||||||
|
}
|
||||||
|
|
||||||
return ComposeMiddlewares(
|
return ComposeMiddlewares(
|
||||||
NewRunMsgsTxHandler(options.MsgServiceRouter, options.LegacyRouter),
|
NewRunMsgsTxHandler(options.MsgServiceRouter, options.LegacyRouter),
|
||||||
NewTxDecoderMiddleware(options.TxDecoder),
|
NewTxDecoderMiddleware(options.TxDecoder),
|
||||||
|
@ -89,10 +101,8 @@ func NewDefaultTxHandler(options TxHandlerOptions) (tx.Handler, error) {
|
||||||
// Choose which events to index in Tendermint. Make sure no events are
|
// Choose which events to index in Tendermint. Make sure no events are
|
||||||
// emitted outside of this middleware.
|
// emitted outside of this middleware.
|
||||||
NewIndexEventsTxMiddleware(options.IndexEvents),
|
NewIndexEventsTxMiddleware(options.IndexEvents),
|
||||||
// Reject all extension options which can optionally be included in the
|
// Reject all extension options other than the ones needed by the feemarket.
|
||||||
// tx.
|
NewExtensionOptionsMiddleware(extensionOptionChecker),
|
||||||
RejectExtensionOptionsMiddleware,
|
|
||||||
MempoolFeeMiddleware,
|
|
||||||
ValidateBasicMiddleware,
|
ValidateBasicMiddleware,
|
||||||
TxTimeoutHeightMiddleware,
|
TxTimeoutHeightMiddleware,
|
||||||
ValidateMemoMiddleware(options.AccountKeeper),
|
ValidateMemoMiddleware(options.AccountKeeper),
|
||||||
|
@ -101,8 +111,7 @@ func NewDefaultTxHandler(options TxHandlerOptions) (tx.Handler, error) {
|
||||||
// ComposeMiddlewares godoc for details.
|
// ComposeMiddlewares godoc for details.
|
||||||
// `DeductFeeMiddleware` and `IncrementSequenceMiddleware` should be put outside of `WithBranchedStore` middleware,
|
// `DeductFeeMiddleware` and `IncrementSequenceMiddleware` should be put outside of `WithBranchedStore` middleware,
|
||||||
// so their storage writes are not discarded when tx fails.
|
// so their storage writes are not discarded when tx fails.
|
||||||
DeductFeeMiddleware(options.AccountKeeper, options.BankKeeper, options.FeegrantKeeper),
|
DeductFeeMiddleware(options.AccountKeeper, options.BankKeeper, options.FeegrantKeeper, txFeeChecker),
|
||||||
TxPriorityMiddleware,
|
|
||||||
SetPubKeyMiddleware(options.AccountKeeper),
|
SetPubKeyMiddleware(options.AccountKeeper),
|
||||||
ValidateSigCountMiddleware(options.AccountKeeper),
|
ValidateSigCountMiddleware(options.AccountKeeper),
|
||||||
SigGasConsumeMiddleware(options.AccountKeeper, sigGasConsumer),
|
SigGasConsumeMiddleware(options.AccountKeeper, sigGasConsumer),
|
||||||
|
|
|
@ -1,63 +0,0 @@
|
||||||
package middleware
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
||||||
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
||||||
"github.com/cosmos/cosmos-sdk/types/tx"
|
|
||||||
)
|
|
||||||
|
|
||||||
var _ tx.Handler = txPriorityHandler{}
|
|
||||||
|
|
||||||
type txPriorityHandler struct {
|
|
||||||
next tx.Handler
|
|
||||||
}
|
|
||||||
|
|
||||||
// TxPriorityMiddleware implements tx handling middleware that determines a
|
|
||||||
// transaction's priority via a naive mechanism -- the total sum of fees provided.
|
|
||||||
// It sets the Priority in ResponseCheckTx only.
|
|
||||||
func TxPriorityMiddleware(txh tx.Handler) tx.Handler {
|
|
||||||
return txPriorityHandler{next: txh}
|
|
||||||
}
|
|
||||||
|
|
||||||
// CheckTx implements tx.Handler.CheckTx. We set the Priority of the transaction
|
|
||||||
// to be ordered in the Tendermint mempool based naively on the total sum of all
|
|
||||||
// fees included. Applications that need more sophisticated mempool ordering
|
|
||||||
// should look to implement their own fee handling middleware instead of using
|
|
||||||
// TxPriorityHandler.
|
|
||||||
func (h txPriorityHandler) CheckTx(ctx context.Context, req tx.Request, checkReq tx.RequestCheckTx) (tx.Response, tx.ResponseCheckTx, error) {
|
|
||||||
feeTx, ok := req.Tx.(sdk.FeeTx)
|
|
||||||
if !ok {
|
|
||||||
return tx.Response{}, tx.ResponseCheckTx{}, sdkerrors.Wrap(sdkerrors.ErrTxDecode, "Tx must be a FeeTx")
|
|
||||||
}
|
|
||||||
|
|
||||||
feeCoins := feeTx.GetFee()
|
|
||||||
|
|
||||||
res, checkRes, err := h.next.CheckTx(ctx, req, checkReq)
|
|
||||||
checkRes.Priority = GetTxPriority(feeCoins)
|
|
||||||
|
|
||||||
return res, checkRes, err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h txPriorityHandler) DeliverTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
|
||||||
return h.next.DeliverTx(ctx, req)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h txPriorityHandler) SimulateTx(ctx context.Context, req tx.Request) (tx.Response, error) {
|
|
||||||
return h.next.SimulateTx(ctx, req)
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetTxPriority returns a naive tx priority based on the amount of the smallest denomination of the fee
|
|
||||||
// provided in a transaction.
|
|
||||||
func GetTxPriority(fee sdk.Coins) int64 {
|
|
||||||
var priority int64
|
|
||||||
for _, c := range fee {
|
|
||||||
p := c.Amount.Int64()
|
|
||||||
if priority == 0 || p < priority {
|
|
||||||
priority = p
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return priority
|
|
||||||
}
|
|
|
@ -1,38 +0,0 @@
|
||||||
package middleware_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
|
|
||||||
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
|
||||||
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
||||||
"github.com/cosmos/cosmos-sdk/types/tx"
|
|
||||||
"github.com/cosmos/cosmos-sdk/x/auth/middleware"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (s *MWTestSuite) TestPriority() {
|
|
||||||
ctx := s.SetupTest(true) // setup
|
|
||||||
txBuilder := s.clientCtx.TxConfig.NewTxBuilder()
|
|
||||||
|
|
||||||
txHandler := middleware.ComposeMiddlewares(noopTxHandler, middleware.TxPriorityMiddleware)
|
|
||||||
|
|
||||||
// keys and addresses
|
|
||||||
priv1, _, addr1 := testdata.KeyTestPubAddr()
|
|
||||||
|
|
||||||
// msg and signatures
|
|
||||||
msg := testdata.NewTestMsg(addr1)
|
|
||||||
atomCoin := sdk.NewCoin("atom", sdk.NewInt(150))
|
|
||||||
apeCoin := sdk.NewInt64Coin("ape", 1500000)
|
|
||||||
feeAmount := sdk.NewCoins(apeCoin, atomCoin)
|
|
||||||
gasLimit := testdata.NewTestGasLimit()
|
|
||||||
s.Require().NoError(txBuilder.SetMsgs(msg))
|
|
||||||
txBuilder.SetFeeAmount(feeAmount)
|
|
||||||
txBuilder.SetGasLimit(gasLimit)
|
|
||||||
|
|
||||||
privs, accNums, accSeqs := []cryptotypes.PrivKey{priv1}, []uint64{0}, []uint64{0}
|
|
||||||
testTx, _, err := s.createTestTx(txBuilder, privs, accNums, accSeqs, ctx.ChainID())
|
|
||||||
s.Require().NoError(err)
|
|
||||||
|
|
||||||
// txHandler errors with insufficient fees
|
|
||||||
_, checkTxRes, err := txHandler.CheckTx(sdk.WrapSDKContext(ctx), tx.Request{Tx: testTx}, tx.RequestCheckTx{})
|
|
||||||
s.Require().NoError(err, "Middleware should not have errored on too low fee for local gasPrice")
|
|
||||||
s.Require().Equal(atomCoin.Amount.Int64(), checkTxRes.Priority, "priority should be atom amount")
|
|
||||||
}
|
|
|
@ -0,0 +1,59 @@
|
||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math"
|
||||||
|
|
||||||
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||||||
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
// checkTxFeeWithValidatorMinGasPrices implements the default fee logic, where the minimum price per
|
||||||
|
// unit of gas is fixed and set by each validator, can the tx priority is computed from the gas price.
|
||||||
|
func checkTxFeeWithValidatorMinGasPrices(ctx sdk.Context, tx sdk.Tx) (sdk.Coins, int64, error) {
|
||||||
|
feeTx, ok := tx.(sdk.FeeTx)
|
||||||
|
if !ok {
|
||||||
|
return nil, 0, sdkerrors.Wrap(sdkerrors.ErrTxDecode, "Tx must be a FeeTx")
|
||||||
|
}
|
||||||
|
|
||||||
|
feeCoins := feeTx.GetFee()
|
||||||
|
gas := feeTx.GetGas()
|
||||||
|
|
||||||
|
// Ensure that the provided fees meet a minimum threshold for the validator,
|
||||||
|
// This is only for local mempool purposes, if this is a DeliverTx, the `MinGasPrices` should be zero.
|
||||||
|
minGasPrices := ctx.MinGasPrices()
|
||||||
|
if !minGasPrices.IsZero() {
|
||||||
|
requiredFees := make(sdk.Coins, len(minGasPrices))
|
||||||
|
|
||||||
|
// Determine the required fees by multiplying each required minimum gas
|
||||||
|
// price by the gas limit, where fee = ceil(minGasPrice * gasLimit).
|
||||||
|
glDec := sdk.NewDec(int64(gas))
|
||||||
|
for i, gp := range minGasPrices {
|
||||||
|
fee := gp.Amount.Mul(glDec)
|
||||||
|
requiredFees[i] = sdk.NewCoin(gp.Denom, fee.Ceil().RoundInt())
|
||||||
|
}
|
||||||
|
|
||||||
|
if !feeCoins.IsAnyGTE(requiredFees) {
|
||||||
|
return nil, 0, sdkerrors.Wrapf(sdkerrors.ErrInsufficientFee, "insufficient fees; got: %s required: %s", feeCoins, requiredFees)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
priority := getTxPriority(feeCoins)
|
||||||
|
return feeCoins, priority, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// getTxPriority returns a naive tx priority based on the amount of the smallest denomination of the fee
|
||||||
|
// provided in a transaction.
|
||||||
|
func getTxPriority(fee sdk.Coins) int64 {
|
||||||
|
var priority int64
|
||||||
|
for _, c := range fee {
|
||||||
|
p := int64(math.MaxInt64)
|
||||||
|
if c.Amount.IsInt64() {
|
||||||
|
p = c.Amount.Int64()
|
||||||
|
}
|
||||||
|
if priority == 0 || p < priority {
|
||||||
|
priority = p
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return priority
|
||||||
|
}
|
Loading…
Reference in New Issue