2d92803b9f
* Add new account bech32 prefixes with godocs * Restructure spacing of existing account code * Update account godocs * More account godoc updates + new tm pub/addr helpers * Update validator type to use new account types/bech32 prefixes * Fix account documentation errors * Update Bech32 prefix for consensus nodes * Update Bech32 spec doc * Fix account type tests * Add missing account consensus functions, clear up godocs, and fix tests * Add to TestRandBech32PubkeyConsistency check * Update initialization of validator public keys * Update query signing info command * Implement new ConsAddress type with associated unit tests * [WIP] Update stake and slashing parameters * Update all calls to MustBech32ifyValPub * [WIP] Validator operator API updates * [WIP] Fix and update unit tests * Fix gov logs (helping to debug failing tests) * Fix gov tally * Fix all broken x/ unit tests * Update gaia app genesis address logic * Fix linting errors * Fix broken LCD tests * Fix broken CLI tests * Implement command to get validator address and pubkey from key name * Add support for getting validator key information via REST endpoint * Update PENDING log * Update docs * Revert GaiaGenTx.PubKey bech32 prefix * Fix broken docs and cli tests * Update genesis to use correct Bech32 (cons) prefix for pubkeys * Update docs and unit tests to reflect new cosmos account bech32 prefix * minor formatting
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
package assoc
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/wire"
|
|
)
|
|
|
|
// ValidatorSet defines
|
|
type ValidatorSet struct {
|
|
sdk.ValidatorSet
|
|
|
|
store sdk.KVStore
|
|
cdc *wire.Codec
|
|
|
|
maxAssoc int
|
|
addrLen int
|
|
}
|
|
|
|
var _ sdk.ValidatorSet = ValidatorSet{}
|
|
|
|
// NewValidatorSet returns new ValidatorSet with underlying ValidatorSet
|
|
func NewValidatorSet(cdc *wire.Codec, store sdk.KVStore, valset sdk.ValidatorSet, maxAssoc int, addrLen int) ValidatorSet {
|
|
if maxAssoc < 0 || addrLen < 0 {
|
|
panic("Cannot use negative integer for NewValidatorSet")
|
|
}
|
|
return ValidatorSet{
|
|
ValidatorSet: valset,
|
|
|
|
store: store,
|
|
cdc: cdc,
|
|
|
|
maxAssoc: maxAssoc,
|
|
addrLen: addrLen,
|
|
}
|
|
}
|
|
|
|
// Implements sdk.ValidatorSet
|
|
func (valset ValidatorSet) Validator(ctx sdk.Context, addr sdk.ValAddress) (res sdk.Validator) {
|
|
base := valset.store.Get(GetBaseKey(addr))
|
|
res = valset.ValidatorSet.Validator(ctx, base)
|
|
if res == nil {
|
|
res = valset.ValidatorSet.Validator(ctx, addr)
|
|
}
|
|
return
|
|
}
|
|
|
|
// GetBaseKey :: sdk.ValAddress -> sdk.ValAddress
|
|
func GetBaseKey(addr sdk.ValAddress) []byte {
|
|
return append([]byte{0x00}, addr...)
|
|
}
|
|
|
|
// GetAssocPrefix :: sdk.ValAddress -> (sdk.ValAddress -> byte)
|
|
func GetAssocPrefix(base sdk.ValAddress) []byte {
|
|
return append([]byte{0x01}, base...)
|
|
}
|
|
|
|
// GetAssocKey :: (sdk.ValAddress, sdk.ValAddress) -> byte
|
|
func GetAssocKey(base sdk.ValAddress, assoc sdk.ValAddress) []byte {
|
|
return append(append([]byte{0x01}, base...), assoc...)
|
|
}
|
|
|
|
// Associate associates new address with validator address
|
|
func (valset ValidatorSet) Associate(ctx sdk.Context, base sdk.ValAddress, assoc sdk.ValAddress) bool {
|
|
if len(base) != valset.addrLen || len(assoc) != valset.addrLen {
|
|
return false
|
|
}
|
|
// If someone already owns the associated address
|
|
if valset.store.Get(GetBaseKey(assoc)) != nil {
|
|
return false
|
|
}
|
|
valset.store.Set(GetBaseKey(assoc), base)
|
|
valset.store.Set(GetAssocKey(base, assoc), []byte{0x00})
|
|
return true
|
|
}
|
|
|
|
// Dissociate removes association between addresses
|
|
func (valset ValidatorSet) Dissociate(ctx sdk.Context, base sdk.ValAddress, assoc sdk.ValAddress) bool {
|
|
if len(base) != valset.addrLen || len(assoc) != valset.addrLen {
|
|
return false
|
|
}
|
|
// No associated address found for given validator
|
|
if !bytes.Equal(valset.store.Get(GetBaseKey(assoc)), base) {
|
|
return false
|
|
}
|
|
valset.store.Delete(GetBaseKey(assoc))
|
|
valset.store.Delete(GetAssocKey(base, assoc))
|
|
return true
|
|
}
|
|
|
|
// Associations returns all associated addresses with a validator
|
|
func (valset ValidatorSet) Associations(ctx sdk.Context, base sdk.ValAddress) (res []sdk.ValAddress) {
|
|
res = make([]sdk.ValAddress, valset.maxAssoc)
|
|
iter := sdk.KVStorePrefixIterator(valset.store, GetAssocPrefix(base))
|
|
defer iter.Close()
|
|
i := 0
|
|
for ; iter.Valid(); iter.Next() {
|
|
key := iter.Key()
|
|
res[i] = key[len(key)-valset.addrLen:]
|
|
i++
|
|
}
|
|
return res[:i]
|
|
}
|