183 lines
4.8 KiB
Go
183 lines
4.8 KiB
Go
package rpc
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/spf13/viper"
|
|
tmtypes "github.com/tendermint/tendermint/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client"
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
|
"github.com/cosmos/cosmos-sdk/client/utils"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// TODO these next two functions feel kinda hacky based on their placement
|
|
|
|
//ValidatorCommand returns the validator set for a given height
|
|
func ValidatorCommand() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "tendermint-validator-set [height]",
|
|
Short: "Get the full tendermint validator set at given height",
|
|
Args: cobra.MaximumNArgs(1),
|
|
RunE: printValidators,
|
|
}
|
|
cmd.Flags().StringP(client.FlagNode, "n", "tcp://localhost:26657", "Node to connect to")
|
|
viper.BindPFlag(client.FlagNode, cmd.Flags().Lookup(client.FlagNode))
|
|
cmd.Flags().Bool(client.FlagTrustNode, false, "Trust connected full node (don't verify proofs for responses)")
|
|
viper.BindPFlag(client.FlagTrustNode, cmd.Flags().Lookup(client.FlagTrustNode))
|
|
return cmd
|
|
}
|
|
|
|
// Validator output in bech32 format
|
|
type ValidatorOutput struct {
|
|
Address sdk.ValAddress `json:"address"` // in bech32
|
|
PubKey string `json:"pub_key"` // in bech32
|
|
ProposerPriority int64 `json:"proposer_priority"`
|
|
VotingPower int64 `json:"voting_power"`
|
|
}
|
|
|
|
// Validators at a certain height output in bech32 format
|
|
type ResultValidatorsOutput struct {
|
|
BlockHeight int64 `json:"block_height"`
|
|
Validators []ValidatorOutput `json:"validators"`
|
|
}
|
|
|
|
func bech32ValidatorOutput(validator *tmtypes.Validator) (ValidatorOutput, error) {
|
|
bechValPubkey, err := sdk.Bech32ifyConsPub(validator.PubKey)
|
|
if err != nil {
|
|
return ValidatorOutput{}, err
|
|
}
|
|
|
|
return ValidatorOutput{
|
|
Address: sdk.ValAddress(validator.Address),
|
|
PubKey: bechValPubkey,
|
|
ProposerPriority: validator.ProposerPriority,
|
|
VotingPower: validator.VotingPower,
|
|
}, nil
|
|
}
|
|
|
|
func getValidators(cliCtx context.CLIContext, height *int64) ([]byte, error) {
|
|
// get the node
|
|
node, err := cliCtx.GetNode()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
validatorsRes, err := node.Validators(height)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if !cliCtx.TrustNode {
|
|
check, err := cliCtx.Verify(validatorsRes.BlockHeight)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if !bytes.Equal(check.ValidatorsHash, tmtypes.NewValidatorSet(validatorsRes.Validators).Hash()) {
|
|
return nil, fmt.Errorf("got invalid validatorset")
|
|
}
|
|
}
|
|
|
|
outputValidatorsRes := ResultValidatorsOutput{
|
|
BlockHeight: validatorsRes.BlockHeight,
|
|
Validators: make([]ValidatorOutput, len(validatorsRes.Validators)),
|
|
}
|
|
|
|
for i := 0; i < len(validatorsRes.Validators); i++ {
|
|
outputValidatorsRes.Validators[i], err = bech32ValidatorOutput(validatorsRes.Validators[i])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
if cliCtx.Indent {
|
|
return cdc.MarshalJSONIndent(outputValidatorsRes, "", " ")
|
|
}
|
|
return cdc.MarshalJSON(outputValidatorsRes)
|
|
|
|
}
|
|
|
|
// CMD
|
|
|
|
func printValidators(cmd *cobra.Command, args []string) error {
|
|
var height *int64
|
|
// optional height
|
|
if len(args) > 0 {
|
|
h, err := strconv.Atoi(args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if h > 0 {
|
|
tmp := int64(h)
|
|
height = &tmp
|
|
}
|
|
}
|
|
|
|
output, err := getValidators(context.NewCLIContext(), height)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(string(output))
|
|
return nil
|
|
}
|
|
|
|
// REST
|
|
|
|
// Validator Set at a height REST handler
|
|
func ValidatorSetRequestHandlerFn(cliCtx context.CLIContext) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
|
|
height, err := strconv.ParseInt(vars["height"], 10, 64)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
w.Write([]byte("ERROR: Couldn't parse block height. Assumed format is '/validatorsets/{height}'."))
|
|
return
|
|
}
|
|
|
|
chainHeight, err := GetChainHeight(cliCtx)
|
|
if height > chainHeight {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte("ERROR: Requested block height is bigger then the chain length."))
|
|
return
|
|
}
|
|
|
|
output, err := getValidators(cliCtx, &height)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
utils.PostProcessResponse(w, cdc, output, cliCtx.Indent)
|
|
}
|
|
}
|
|
|
|
// Latest Validator Set REST handler
|
|
func LatestValidatorSetRequestHandlerFn(cliCtx context.CLIContext) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
height, err := GetChainHeight(cliCtx)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
|
|
output, err := getValidators(cliCtx, &height)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
utils.PostProcessResponse(w, cdc, output, cliCtx.Indent)
|
|
}
|
|
}
|