feat: lotus-shed get remote peer hello message

cli/net.go

@@ -150,7 +150,7 @@ var NetPing = &cli.Command{
 
 		ctx := ReqContext(cctx)
 
-		pis, err := addrInfoFromArg(ctx, cctx)
+		pis, err := AddrInfoFromArg(ctx, cctx)
 		if err != nil {
 			return err
 		}
@@ -271,7 +271,7 @@ var NetConnect = &cli.Command{
 		defer closer()
 		ctx := ReqContext(cctx)
 
-		pis, err := addrInfoFromArg(ctx, cctx)
+		pis, err := AddrInfoFromArg(ctx, cctx)
 		if err != nil {
 			return err
 		}
@@ -290,7 +290,7 @@ var NetConnect = &cli.Command{
 	},
 }
 
-func addrInfoFromArg(ctx context.Context, cctx *cli.Context) ([]peer.AddrInfo, error) {
+func AddrInfoFromArg(ctx context.Context, cctx *cli.Context) ([]peer.AddrInfo, error) {
 	pis, err := addrutil.ParseAddresses(ctx, cctx.Args().Slice())
 	if err != nil {
 		a, perr := address.NewFromString(cctx.Args().First())

cmd/lotus-shed/hello.go

@@ -0,0 +1,120 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"os"
+	"time"
+
+	cborutil "github.com/filecoin-project/go-cbor-util"
+	"github.com/filecoin-project/lotus/blockstore"
+	"github.com/filecoin-project/lotus/chain/consensus/filcns"
+	"github.com/filecoin-project/lotus/chain/store"
+	lcli "github.com/filecoin-project/lotus/cli"
+	"github.com/filecoin-project/lotus/node/hello"
+	"github.com/ipfs/go-datastore"
+	"github.com/libp2p/go-libp2p"
+	"github.com/libp2p/go-libp2p-core/host"
+	inet "github.com/libp2p/go-libp2p-core/network"
+	"github.com/libp2p/go-libp2p-core/peer"
+	"github.com/urfave/cli/v2"
+	"golang.org/x/xerrors"
+)
+
+var resultCh chan bool
+
+var helloCmd = &cli.Command{
+	Name:        "hello",
+	Description: "Get remote peer hello message by multiaddr",
+	Flags: []cli.Flag{
+		&cli.StringFlag{
+			Name:     "multiaddr",
+			Usage:    "remote peer multiaddr",
+			Required: true,
+		},
+		&cli.StringFlag{
+			Name:  "genesis-path",
+			Usage: "genesis car file path",
+		},
+	},
+
+	Action: func(cctx *cli.Context) error {
+		ctx := lcli.ReqContext(cctx)
+		resultCh = make(chan bool, 1)
+		cf := cctx.String("genesis-path")
+		f, err := os.OpenFile(cf, os.O_RDONLY, 0664)
+		if err != nil {
+			return xerrors.Errorf("opening the car file: %w", err)
+		}
+		bs := blockstore.FromDatastore(datastore.NewMapDatastore())
+		cs := store.NewChainStore(bs, bs, datastore.NewMapDatastore(), filcns.Weight, nil)
+		ts, err := cs.Import(cctx.Context, f)
+		if err != nil {
+			return err
+		}
+		pis, err := lcli.AddrInfoFromArg(ctx, cctx)
+		if err != nil {
+			return err
+		}
+		h, err := libp2p.New()
+		if err != nil {
+			return err
+		}
+		err = h.Connect(ctx, pis[0])
+		if err != nil {
+			return err
+		}
+		h.SetStreamHandler(hello.ProtocolID, HandleStream)
+		weight, err := cs.Weight(ctx, ts)
+		if err != nil {
+			return err
+		}
+		gen := ts.Blocks()[0]
+		hmsg := &hello.HelloMessage{
+			HeaviestTipSet:       ts.Cids(),
+			HeaviestTipSetHeight: ts.Height(),
+			HeaviestTipSetWeight: weight,
+			GenesisHash:          gen.Cid(),
+		}
+		err = SayHello(ctx, pis[0].ID, h, hmsg)
+		if err != nil {
+			return err
+		}
+		ctx, done := context.WithTimeout(context.Background(), 5*time.Second)
+		defer done()
+		select {
+		case <-resultCh:
+		case <-ctx.Done():
+			fmt.Println("can't get hello message, please try again")
+		}
+		return nil
+	},
+}
+
+func HandleStream(s inet.Stream) {
+	var hmsg hello.HelloMessage
+	if err := cborutil.ReadCborRPC(s, &hmsg); err != nil {
+		log.Infow("failed to read hello message, disconnecting", "error", err)
+		_ = s.Conn().Close()
+		return
+	}
+	data, err := json.Marshal(hmsg)
+	if err != nil {
+		return
+	}
+	fmt.Println(string(data))
+	resultCh <- true
+}
+
+func SayHello(ctx context.Context, pid peer.ID, h host.Host, hmsg *hello.HelloMessage) error {
+	s, err := h.NewStream(ctx, pid, hello.ProtocolID)
+	if err != nil {
+		return xerrors.Errorf("error opening stream: %w", err)
+	}
+	if err := cborutil.WriteCborRPC(s, hmsg); err != nil {
+		return xerrors.Errorf("writing rpc to peer: %w", err)
+	}
+
+	return nil
+}

cmd/lotus-shed/main.go

@@ -39,6 +39,7 @@ func main() {
 		marketCmd,
 		miscCmd,
 		mpoolCmd,
+		helloCmd,
 		genesisVerifyCmd,
 		mathCmd,
 		minerCmd,