2021-05-06 11:23:48 -07:00
|
|
|
package keeper
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/cosmos/cosmos-sdk/internal/conv"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/types/address"
|
2021-07-08 04:29:12 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/types/kv"
|
2021-05-06 11:23:48 -07:00
|
|
|
"github.com/cosmos/cosmos-sdk/x/authz"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Keys for store prefixes
|
|
|
|
var (
|
|
|
|
GrantKey = []byte{0x01} // prefix for each key
|
|
|
|
)
|
|
|
|
|
|
|
|
// StoreKey is the store key string for authz
|
|
|
|
const StoreKey = authz.ModuleName
|
|
|
|
|
|
|
|
// grantStoreKey - return authorization store key
|
|
|
|
// Items are stored with the following key: values
|
|
|
|
//
|
|
|
|
// - 0x01<granterAddressLen (1 Byte)><granterAddress_Bytes><granteeAddressLen (1 Byte)><granteeAddress_Bytes><msgType_Bytes>: Grant
|
|
|
|
func grantStoreKey(grantee sdk.AccAddress, granter sdk.AccAddress, msgType string) []byte {
|
|
|
|
m := conv.UnsafeStrToBytes(msgType)
|
|
|
|
granter = address.MustLengthPrefix(granter)
|
|
|
|
grantee = address.MustLengthPrefix(grantee)
|
|
|
|
|
|
|
|
l := 1 + len(grantee) + len(granter) + len(m)
|
|
|
|
var key = make([]byte, l)
|
|
|
|
copy(key, GrantKey)
|
|
|
|
copy(key[1:], granter)
|
|
|
|
copy(key[1+len(granter):], grantee)
|
|
|
|
copy(key[l-len(m):], m)
|
|
|
|
// fmt.Println(">>>> len", l, key)
|
|
|
|
return key
|
|
|
|
}
|
|
|
|
|
|
|
|
// addressesFromGrantStoreKey - split granter & grantee address from the authorization key
|
|
|
|
func addressesFromGrantStoreKey(key []byte) (granterAddr, granteeAddr sdk.AccAddress) {
|
|
|
|
// key is of format:
|
|
|
|
// 0x01<granterAddressLen (1 Byte)><granterAddress_Bytes><granteeAddressLen (1 Byte)><granteeAddress_Bytes><msgType_Bytes>
|
2021-07-08 04:29:12 -07:00
|
|
|
kv.AssertKeyAtLeastLength(key, 2)
|
2021-05-06 11:23:48 -07:00
|
|
|
granterAddrLen := key[1] // remove prefix key
|
2021-07-08 04:29:12 -07:00
|
|
|
kv.AssertKeyAtLeastLength(key, int(3+granterAddrLen))
|
2021-05-06 11:23:48 -07:00
|
|
|
granterAddr = sdk.AccAddress(key[2 : 2+granterAddrLen])
|
|
|
|
granteeAddrLen := int(key[2+granterAddrLen])
|
2021-07-08 04:29:12 -07:00
|
|
|
kv.AssertKeyAtLeastLength(key, 4+int(granterAddrLen+byte(granteeAddrLen)))
|
2021-05-06 11:23:48 -07:00
|
|
|
granteeAddr = sdk.AccAddress(key[3+granterAddrLen : 3+granterAddrLen+byte(granteeAddrLen)])
|
|
|
|
|
|
|
|
return granterAddr, granteeAddr
|
|
|
|
}
|