154 lines
4.0 KiB
Go
154 lines
4.0 KiB
Go
package guardiand
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
publicrpcv1 "github.com/certusone/wormhole/node/pkg/proto/publicrpc/v1"
|
|
"github.com/certusone/wormhole/node/pkg/vaa"
|
|
"github.com/spf13/cobra"
|
|
"log"
|
|
"os"
|
|
"sort"
|
|
"text/tabwriter"
|
|
"time"
|
|
)
|
|
|
|
// How to test in container:
|
|
// kubectl exec guardian-0 -- /guardiand admin list-nodes --socket /tmp/admin.sock
|
|
|
|
var (
|
|
showDetails bool
|
|
)
|
|
|
|
func init() {
|
|
AdminClientListNodes.Flags().BoolVar(&showDetails, "showDetails", false, "Show error counter and contract addresses")
|
|
}
|
|
|
|
var AdminClientListNodes = &cobra.Command{
|
|
Use: "list-nodes",
|
|
Short: "Fetches an aggregated list of guardian nodes",
|
|
Run: runListNodes,
|
|
}
|
|
|
|
func runListNodes(cmd *cobra.Command, args []string) {
|
|
ctx := context.Background()
|
|
conn, err, c := getPublicRPCServiceClient(ctx, *clientSocketPath)
|
|
defer conn.Close()
|
|
if err != nil {
|
|
log.Fatalf("failed to get publicrpc client: %v", err)
|
|
}
|
|
|
|
lastHeartbeats, err := c.GetLastHeartbeats(ctx, &publicrpcv1.GetLastHeartbeatsRequest{})
|
|
if err != nil {
|
|
log.Fatalf("failed to list nodes: %v", err)
|
|
}
|
|
|
|
gs, err := c.GetCurrentGuardianSet(ctx, &publicrpcv1.GetCurrentGuardianSetRequest{})
|
|
if err != nil {
|
|
log.Fatalf("failed to list current guardian get: %v", err)
|
|
}
|
|
|
|
log.Printf("current guardian set index: %d (%d guardians)",
|
|
gs.GuardianSet.Index, len(gs.GuardianSet.Addresses))
|
|
|
|
nodes := lastHeartbeats.Entries
|
|
|
|
sort.Slice(nodes, func(i, j int) bool {
|
|
if nodes[i].RawHeartbeat == nil || nodes[j].RawHeartbeat == nil {
|
|
return false
|
|
}
|
|
return nodes[i].RawHeartbeat.NodeName < nodes[j].RawHeartbeat.NodeName
|
|
})
|
|
|
|
log.Printf("%d nodes in guardian state set", len(nodes))
|
|
|
|
w := tabwriter.NewWriter(os.Stdout, 0, 8, 2, ' ', 0)
|
|
|
|
if showDetails {
|
|
_, _ = w.Write([]byte("Node key\tGuardian key\tNode name\tVersion\tLast seen\tUptime\tSolana\tEthereum\tTerra\tBSC\tPolygon\n"))
|
|
} else {
|
|
_, _ = w.Write([]byte("Node key\tGuardian key\tNode name\tVersion\tLast seen\tSolana\tEthereum\tTerra\tBSC\tPolygon\n"))
|
|
}
|
|
|
|
for _, h := range nodes {
|
|
if h.RawHeartbeat == nil {
|
|
continue
|
|
}
|
|
|
|
last := time.Unix(0, h.RawHeartbeat.Timestamp)
|
|
boot := time.Unix(0, h.RawHeartbeat.BootTimestamp)
|
|
|
|
heights := map[vaa.ChainID]int64{}
|
|
truncAddrs := make(map[vaa.ChainID]string)
|
|
errors := map[vaa.ChainID]uint64{}
|
|
for _, n := range h.RawHeartbeat.Networks {
|
|
heights[vaa.ChainID(n.Id)] = n.Height
|
|
errors[vaa.ChainID(n.Id)] = n.ErrorCount
|
|
if len(n.ContractAddress) >= 16 {
|
|
truncAddrs[vaa.ChainID(n.Id)] = n.ContractAddress[:16]
|
|
} else {
|
|
truncAddrs[vaa.ChainID(n.Id)] = "INVALID"
|
|
}
|
|
}
|
|
|
|
if showDetails {
|
|
fmt.Fprintf(w,
|
|
"%s\t%s\t%s\t%s\t%s\t%s\t%s %d (%d)\t%s %d (%d)\t%s %d (%d)\t%s %d (%d)\t%s %d (%d)\n",
|
|
h.P2PNodeAddr,
|
|
h.RawHeartbeat.GuardianAddr,
|
|
h.RawHeartbeat.NodeName,
|
|
h.RawHeartbeat.Version,
|
|
time.Since(last),
|
|
time.Since(boot),
|
|
truncAddrs[vaa.ChainIDSolana],
|
|
heights[vaa.ChainIDSolana],
|
|
errors[vaa.ChainIDSolana],
|
|
truncAddrs[vaa.ChainIDEthereum],
|
|
heights[vaa.ChainIDEthereum],
|
|
errors[vaa.ChainIDEthereum],
|
|
truncAddrs[vaa.ChainIDTerra],
|
|
heights[vaa.ChainIDTerra],
|
|
errors[vaa.ChainIDTerra],
|
|
truncAddrs[vaa.ChainIDBSC],
|
|
heights[vaa.ChainIDBSC],
|
|
errors[vaa.ChainIDBSC],
|
|
truncAddrs[vaa.ChainIDPolygon],
|
|
heights[vaa.ChainIDPolygon],
|
|
errors[vaa.ChainIDPolygon],
|
|
)
|
|
} else {
|
|
fmt.Fprintf(w,
|
|
"%s\t%s\t%s\t%s\t%s\t%d\t%d\t%d\t%d\t%d\n",
|
|
h.P2PNodeAddr,
|
|
h.RawHeartbeat.GuardianAddr,
|
|
h.RawHeartbeat.NodeName,
|
|
h.RawHeartbeat.Version,
|
|
time.Since(last),
|
|
heights[vaa.ChainIDSolana],
|
|
heights[vaa.ChainIDEthereum],
|
|
heights[vaa.ChainIDTerra],
|
|
heights[vaa.ChainIDBSC],
|
|
heights[vaa.ChainIDPolygon],
|
|
)
|
|
}
|
|
}
|
|
|
|
w.Flush()
|
|
fmt.Print("\n")
|
|
|
|
for _, addr := range gs.GuardianSet.Addresses {
|
|
var found bool
|
|
for _, h := range nodes {
|
|
if h.VerifiedGuardianAddr == addr {
|
|
found = true
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
fmt.Printf("Missing guardian: %s\n", addr)
|
|
}
|
|
}
|
|
|
|
fmt.Println("\n[do not parse - use the gRPC or REST API for scripting]")
|
|
}
|