Move lm operations to separate command

cmd/lm/main.go

@@ -0,0 +1,47 @@
+//go:build windows
+
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/Microsoft/hcsshim/internal/shimdiag"
+	"github.com/urfave/cli"
+)
+
+func main() {
+	app := cli.NewApp()
+	app.Name = "lm"
+	app.Usage = "Interact with runhcs shim live migration interface"
+	app.Commands = []cli.Command{
+		lmPrepareCommand,
+		lmListenCommand,
+		lmAcceptCommand,
+		lmDialCommand,
+		lmTransferCommand,
+		lmFinalizeCommand,
+		json2pbCommand,
+		pb2jsonCommand,
+	}
+	if err := app.Run(os.Args); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}
+
+func getPid(shimName string) (int32, error) {
+	shim, err := shimdiag.GetShim(shimName)
+	if err != nil {
+		return 0, err
+	}
+	defer shim.Close()
+
+	svc := shimdiag.NewShimDiagClient(shim)
+	resp, err := svc.DiagPid(context.Background(), &shimdiag.PidRequest{})
+	if err != nil {
+		return 0, err
+	}
+	return resp.Pid, nil
+}

cmd/shimdiag/shimdiag.go

@@ -21,14 +21,6 @@ func main() {
 		stacksCommand,
 		tasksCommand,
 		shareCommand,
-		lmPrepareCommand,
-		lmListenCommand,
-		lmAcceptCommand,
-		lmDialCommand,
-		lmTransferCommand,
-		lmFinalizeCommand,
-		json2pbCommand,
-		pb2jsonCommand,
 	}
 	if err := app.Run(os.Args); err != nil {
 		fmt.Fprintln(os.Stderr, err)