106 lines
3.0 KiB
Go
106 lines
3.0 KiB
Go
package feegrant
|
|
|
|
import (
|
|
"github.com/gogo/protobuf/proto"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
)
|
|
|
|
// TODO: Revisit this once we have propoer gas fee framework.
|
|
// Tracking issues https://github.com/cosmos/cosmos-sdk/issues/9054, https://github.com/cosmos/cosmos-sdk/discussions/9072
|
|
const (
|
|
gasCostPerIteration = uint64(10)
|
|
)
|
|
|
|
var _ FeeAllowanceI = (*AllowedMsgAllowance)(nil)
|
|
var _ types.UnpackInterfacesMessage = (*AllowedMsgAllowance)(nil)
|
|
|
|
// UnpackInterfaces implements UnpackInterfacesMessage.UnpackInterfaces
|
|
func (a *AllowedMsgAllowance) UnpackInterfaces(unpacker types.AnyUnpacker) error {
|
|
var allowance FeeAllowanceI
|
|
return unpacker.UnpackAny(a.Allowance, &allowance)
|
|
}
|
|
|
|
// NewAllowedMsgFeeAllowance creates new filtered fee allowance.
|
|
func NewAllowedMsgAllowance(allowance FeeAllowanceI, allowedMsgs []string) (*AllowedMsgAllowance, error) {
|
|
msg, ok := allowance.(proto.Message)
|
|
if !ok {
|
|
return nil, sdkerrors.Wrapf(sdkerrors.ErrPackAny, "cannot proto marshal %T", msg)
|
|
}
|
|
any, err := types.NewAnyWithValue(msg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &AllowedMsgAllowance{
|
|
Allowance: any,
|
|
AllowedMessages: allowedMsgs,
|
|
}, nil
|
|
}
|
|
|
|
// GetAllowance returns allowed fee allowance.
|
|
func (a *AllowedMsgAllowance) GetAllowance() (FeeAllowanceI, error) {
|
|
allowance, ok := a.Allowance.GetCachedValue().(FeeAllowanceI)
|
|
if !ok {
|
|
return nil, sdkerrors.Wrap(ErrNoAllowance, "failed to get allowance")
|
|
}
|
|
|
|
return allowance, nil
|
|
}
|
|
|
|
// Accept method checks for the filtered messages has valid expiry
|
|
func (a *AllowedMsgAllowance) Accept(ctx sdk.Context, fee sdk.Coins, msgs []sdk.Msg) (bool, error) {
|
|
if !a.allMsgTypesAllowed(ctx, msgs) {
|
|
return false, sdkerrors.Wrap(ErrMessageNotAllowed, "message does not exist in allowed messages")
|
|
}
|
|
|
|
allowance, err := a.GetAllowance()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return allowance.Accept(ctx, fee, msgs)
|
|
}
|
|
|
|
func (a *AllowedMsgAllowance) allowedMsgsToMap(ctx sdk.Context) map[string]bool {
|
|
msgsMap := make(map[string]bool, len(a.AllowedMessages))
|
|
for _, msg := range a.AllowedMessages {
|
|
ctx.GasMeter().ConsumeGas(gasCostPerIteration, "check msg")
|
|
msgsMap[msg] = true
|
|
}
|
|
|
|
return msgsMap
|
|
}
|
|
|
|
func (a *AllowedMsgAllowance) allMsgTypesAllowed(ctx sdk.Context, msgs []sdk.Msg) bool {
|
|
msgsMap := a.allowedMsgsToMap(ctx)
|
|
|
|
for _, msg := range msgs {
|
|
ctx.GasMeter().ConsumeGas(gasCostPerIteration, "check msg")
|
|
if !msgsMap[sdk.MsgTypeURL(msg)] {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// ValidateBasic implements FeeAllowance and enforces basic sanity checks
|
|
func (a *AllowedMsgAllowance) ValidateBasic() error {
|
|
if a.Allowance == nil {
|
|
return sdkerrors.Wrap(ErrNoAllowance, "allowance should not be empty")
|
|
}
|
|
if len(a.AllowedMessages) == 0 {
|
|
return sdkerrors.Wrap(ErrNoMessages, "allowed messages shouldn't be empty")
|
|
}
|
|
|
|
allowance, err := a.GetAllowance()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return allowance.ValidateBasic()
|
|
}
|