mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-20 15:05:21 +00:00
3375484f79
* Use cosmossdk.io/errors for deprecated error methods * Update error registration with cosmossdk.io/errors * Use cosmossdk.io/math for deprecated sdk.Int alias * Fix modified proto file * Update sdk.Int usage in swap hooks * Update e2e test deprecated method usage
60 lines
2.1 KiB
Go
60 lines
2.1 KiB
Go
package ante
|
|
|
|
import (
|
|
errorsmod "cosmossdk.io/errors"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
authsigning "github.com/cosmos/cosmos-sdk/x/auth/signing"
|
|
)
|
|
|
|
// AddressFetcher is a type signature for functions used by the AuthenticatedMempoolDecorator to get authorized addresses.
|
|
type AddressFetcher func(sdk.Context) []sdk.AccAddress
|
|
|
|
var _ sdk.AnteDecorator = AuthenticatedMempoolDecorator{}
|
|
|
|
// AuthenticatedMempoolDecorator blocks all txs from reaching the mempool unless they're signed by one of the authorzed addresses.
|
|
// It only runs before entry to mempool (CheckTx), and not in consensus (DeliverTx)
|
|
type AuthenticatedMempoolDecorator struct {
|
|
addressFetchers []AddressFetcher
|
|
}
|
|
|
|
func NewAuthenticatedMempoolDecorator(fetchers ...AddressFetcher) AuthenticatedMempoolDecorator {
|
|
return AuthenticatedMempoolDecorator{
|
|
addressFetchers: fetchers,
|
|
}
|
|
}
|
|
|
|
func (amd AuthenticatedMempoolDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simulate bool, next sdk.AnteHandler) (newCtx sdk.Context, err error) {
|
|
// This is only for local mempool purposes, and thus is only run on check tx.
|
|
if ctx.IsCheckTx() && !simulate {
|
|
sigTx, ok := tx.(authsigning.SigVerifiableTx)
|
|
if !ok {
|
|
return ctx, errorsmod.Wrap(sdkerrors.ErrTxDecode, "tx must be sig verifiable tx")
|
|
}
|
|
if !commonAddressesExist(sigTx.GetSigners(), amd.fetchAuthorizedAddresses(ctx)) {
|
|
return ctx, errorsmod.Wrap(sdkerrors.ErrUnauthorized, "tx contains no signers authorized for this mempool")
|
|
}
|
|
}
|
|
return next(ctx, tx, simulate)
|
|
}
|
|
|
|
func (amd AuthenticatedMempoolDecorator) fetchAuthorizedAddresses(ctx sdk.Context) []sdk.AccAddress {
|
|
addrs := []sdk.AccAddress{}
|
|
for _, fetch := range amd.addressFetchers {
|
|
addrs = append(addrs, fetch(ctx)...)
|
|
}
|
|
return addrs
|
|
}
|
|
|
|
// commonAddressesExist checks if there is any intersection between two lists of addresses
|
|
func commonAddressesExist(addresses1, addresses2 []sdk.AccAddress) bool {
|
|
for _, a1 := range addresses1 {
|
|
for _, a2 := range addresses2 {
|
|
if a1.Equals(a2) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
return false
|
|
}
|