238 lines
6.2 KiB
Go
238 lines
6.2 KiB
Go
package authz
|
|
|
|
import (
|
|
authzcodec "github.com/cosmos/cosmos-sdk/x/authz/codec"
|
|
"time"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
|
|
cdctypes "github.com/cosmos/cosmos-sdk/codec/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/migrations/legacytx"
|
|
)
|
|
|
|
var (
|
|
_ sdk.Msg = &MsgGrant{}
|
|
_ sdk.Msg = &MsgRevoke{}
|
|
_ sdk.Msg = &MsgExec{}
|
|
|
|
// For amino support.
|
|
_ legacytx.LegacyMsg = &MsgGrant{}
|
|
_ legacytx.LegacyMsg = &MsgRevoke{}
|
|
_ legacytx.LegacyMsg = &MsgExec{}
|
|
|
|
_ cdctypes.UnpackInterfacesMessage = &MsgGrant{}
|
|
_ cdctypes.UnpackInterfacesMessage = &MsgExec{}
|
|
)
|
|
|
|
// NewMsgGrant creates a new MsgGrant
|
|
//nolint:interfacer
|
|
func NewMsgGrant(granter sdk.AccAddress, grantee sdk.AccAddress, a Authorization, expiration *time.Time) (*MsgGrant, error) {
|
|
m := &MsgGrant{
|
|
Granter: granter.String(),
|
|
Grantee: grantee.String(),
|
|
Grant: Grant{Expiration: expiration},
|
|
}
|
|
err := m.SetAuthorization(a)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// GetSigners implements Msg
|
|
func (msg MsgGrant) GetSigners() []sdk.AccAddress {
|
|
granter, _ := sdk.AccAddressFromBech32(msg.Granter)
|
|
return []sdk.AccAddress{granter}
|
|
}
|
|
|
|
// ValidateBasic implements Msg
|
|
func (msg MsgGrant) ValidateBasic() error {
|
|
granter, err := sdk.AccAddressFromBech32(msg.Granter)
|
|
if err != nil {
|
|
return sdkerrors.ErrInvalidAddress.Wrapf("invalid granter address: %s", err)
|
|
}
|
|
grantee, err := sdk.AccAddressFromBech32(msg.Grantee)
|
|
if err != nil {
|
|
return sdkerrors.ErrInvalidAddress.Wrapf("invalid grantee address: %s", err)
|
|
}
|
|
|
|
if granter.Equals(grantee) {
|
|
return ErrGranteeIsGranter
|
|
}
|
|
return msg.Grant.ValidateBasic()
|
|
}
|
|
|
|
// Type implements the LegacyMsg.Type method.
|
|
func (msg MsgGrant) Type() string {
|
|
return sdk.MsgTypeURL(&msg)
|
|
}
|
|
|
|
// Route implements the LegacyMsg.Route method.
|
|
func (msg MsgGrant) Route() string {
|
|
return sdk.MsgTypeURL(&msg)
|
|
}
|
|
|
|
// GetSignBytes implements the LegacyMsg.GetSignBytes method.
|
|
func (msg MsgGrant) GetSignBytes() []byte {
|
|
return sdk.MustSortJSON(authzcodec.ModuleCdc.MustMarshalJSON(&msg))
|
|
}
|
|
|
|
// GetAuthorization returns the cache value from the MsgGrant.Authorization if present.
|
|
func (msg *MsgGrant) GetAuthorization() (Authorization, error) {
|
|
return msg.Grant.GetAuthorization()
|
|
}
|
|
|
|
// SetAuthorization converts Authorization to any and adds it to MsgGrant.Authorization.
|
|
func (msg *MsgGrant) SetAuthorization(a Authorization) error {
|
|
m, ok := a.(proto.Message)
|
|
if !ok {
|
|
return sdkerrors.ErrPackAny.Wrapf("can't proto marshal %T", m)
|
|
}
|
|
any, err := cdctypes.NewAnyWithValue(m)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msg.Grant.Authorization = any
|
|
return nil
|
|
}
|
|
|
|
// UnpackInterfaces implements UnpackInterfacesMessage.UnpackInterfaces
|
|
func (msg MsgExec) UnpackInterfaces(unpacker cdctypes.AnyUnpacker) error {
|
|
for _, x := range msg.Msgs {
|
|
var msgExecAuthorized sdk.Msg
|
|
err := unpacker.UnpackAny(x, &msgExecAuthorized)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UnpackInterfaces implements UnpackInterfacesMessage.UnpackInterfaces
|
|
func (msg MsgGrant) UnpackInterfaces(unpacker cdctypes.AnyUnpacker) error {
|
|
return msg.Grant.UnpackInterfaces(unpacker)
|
|
}
|
|
|
|
// NewMsgRevoke creates a new MsgRevoke
|
|
//nolint:interfacer
|
|
func NewMsgRevoke(granter sdk.AccAddress, grantee sdk.AccAddress, msgTypeURL string) MsgRevoke {
|
|
return MsgRevoke{
|
|
Granter: granter.String(),
|
|
Grantee: grantee.String(),
|
|
MsgTypeUrl: msgTypeURL,
|
|
}
|
|
}
|
|
|
|
// GetSigners implements Msg
|
|
func (msg MsgRevoke) GetSigners() []sdk.AccAddress {
|
|
granter, _ := sdk.AccAddressFromBech32(msg.Granter)
|
|
return []sdk.AccAddress{granter}
|
|
}
|
|
|
|
// ValidateBasic implements MsgRequest.ValidateBasic
|
|
func (msg MsgRevoke) ValidateBasic() error {
|
|
granter, err := sdk.AccAddressFromBech32(msg.Granter)
|
|
if err != nil {
|
|
return sdkerrors.ErrInvalidAddress.Wrapf("invalid granter address: %s", err)
|
|
}
|
|
grantee, err := sdk.AccAddressFromBech32(msg.Grantee)
|
|
if err != nil {
|
|
return sdkerrors.ErrInvalidAddress.Wrapf("invalid grantee address: %s", err)
|
|
}
|
|
|
|
if granter.Equals(grantee) {
|
|
return ErrGranteeIsGranter
|
|
}
|
|
|
|
if msg.MsgTypeUrl == "" {
|
|
return sdkerrors.ErrInvalidRequest.Wrap("missing method name")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Type implements the LegacyMsg.Type method.
|
|
func (msg MsgRevoke) Type() string {
|
|
return sdk.MsgTypeURL(&msg)
|
|
}
|
|
|
|
// Route implements the LegacyMsg.Route method.
|
|
func (msg MsgRevoke) Route() string {
|
|
return sdk.MsgTypeURL(&msg)
|
|
}
|
|
|
|
// GetSignBytes implements the LegacyMsg.GetSignBytes method.
|
|
func (msg MsgRevoke) GetSignBytes() []byte {
|
|
return sdk.MustSortJSON(authzcodec.ModuleCdc.MustMarshalJSON(&msg))
|
|
}
|
|
|
|
// NewMsgExec creates a new MsgExecAuthorized
|
|
//nolint:interfacer
|
|
func NewMsgExec(grantee sdk.AccAddress, msgs []sdk.Msg) MsgExec {
|
|
msgsAny := make([]*cdctypes.Any, len(msgs))
|
|
for i, msg := range msgs {
|
|
any, err := cdctypes.NewAnyWithValue(msg)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
msgsAny[i] = any
|
|
}
|
|
|
|
return MsgExec{
|
|
Grantee: grantee.String(),
|
|
Msgs: msgsAny,
|
|
}
|
|
}
|
|
|
|
// GetMessages returns the cache values from the MsgExecAuthorized.Msgs if present.
|
|
func (msg MsgExec) GetMessages() ([]sdk.Msg, error) {
|
|
msgs := make([]sdk.Msg, len(msg.Msgs))
|
|
for i, msgAny := range msg.Msgs {
|
|
msg, ok := msgAny.GetCachedValue().(sdk.Msg)
|
|
if !ok {
|
|
return nil, sdkerrors.ErrInvalidRequest.Wrapf("messages contains %T which is not a sdk.MsgRequest", msgAny)
|
|
}
|
|
msgs[i] = msg
|
|
}
|
|
|
|
return msgs, nil
|
|
}
|
|
|
|
// GetSigners implements Msg
|
|
func (msg MsgExec) GetSigners() []sdk.AccAddress {
|
|
grantee, _ := sdk.AccAddressFromBech32(msg.Grantee)
|
|
return []sdk.AccAddress{grantee}
|
|
}
|
|
|
|
// ValidateBasic implements Msg
|
|
func (msg MsgExec) ValidateBasic() error {
|
|
if _, err := sdk.AccAddressFromBech32(msg.Grantee); err != nil {
|
|
return sdkerrors.ErrInvalidAddress.Wrapf("invalid grantee address: %s", err)
|
|
}
|
|
|
|
if len(msg.Msgs) == 0 {
|
|
return sdkerrors.ErrInvalidRequest.Wrapf("messages cannot be empty")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Type implements the LegacyMsg.Type method.
|
|
func (msg MsgExec) Type() string {
|
|
return sdk.MsgTypeURL(&msg)
|
|
}
|
|
|
|
// Route implements the LegacyMsg.Route method.
|
|
func (msg MsgExec) Route() string {
|
|
return sdk.MsgTypeURL(&msg)
|
|
}
|
|
|
|
// GetSignBytes implements the LegacyMsg.GetSignBytes method.
|
|
func (msg MsgExec) GetSignBytes() []byte {
|
|
return sdk.MustSortJSON(authzcodec.ModuleCdc.MustMarshalJSON(&msg))
|
|
}
|