Merge pull request #285 from AkihiroSuda/auto-assign-ssh-localport

Assign `ssh.localPort` automatically to an available port

Author: AkihiroSuda

.github/workflows/test.yml

@@ -90,7 +90,7 @@ jobs:
   integration:
     name: Integration tests
     runs-on: macos-11
-    timeout-minutes: 60
+    timeout-minutes: 120
     strategy:
       matrix:
         # GHA macOS is slow and flaky, so we only test a few YAMLS here.
@@ -165,7 +165,7 @@ jobs:
       - name: Test
         uses: nick-invision/retry@v2
         with:
-          timeout_minutes: 15
+          timeout_minutes: 30
           retry_on: error
           max_attempts: 3
           command: ./hack/test-example.sh examples/${{ matrix.example }}

cmd/limactl/hostagent.go

@@ -4,11 +4,16 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"net"
+	"net/http"
 	"os"
 	"os/signal"
 	"strconv"
 
+	"github.com/gorilla/mux"
 	"github.com/lima-vm/lima/pkg/hostagent"
+	"github.com/lima-vm/lima/pkg/hostagent/api/server"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -21,6 +26,7 @@ func newHostagentCommand() *cobra.Command {
 		Hidden: true,
 	}
 	hostagentCommand.Flags().StringP("pidfile", "p", "", "write pid to file")
+	hostagentCommand.Flags().String("socket", "", "hostagent socket")
 	return hostagentCommand
 }
 
@@ -38,6 +44,13 @@ func hostagentAction(cmd *cobra.Command, args []string) error {
 		}
 		defer os.RemoveAll(pidfile)
 	}
+	socket, err := cmd.Flags().GetString("socket")
+	if err != nil {
+		return err
+	}
+	if socket == "" {
+		return fmt.Errorf("socket must be specified (limactl version mismatch?)")
+	}
 
 	instName := args[0]
 
@@ -51,6 +64,28 @@ func hostagentAction(cmd *cobra.Command, args []string) error {
 	if err != nil {
 		return err
 	}
+
+	backend := &server.Backend{
+		Agent: ha,
+	}
+	r := mux.NewRouter()
+	server.AddRoutes(r, backend)
+	srv := &http.Server{Handler: r}
+	err = os.RemoveAll(socket)
+	if err != nil {
+		return err
+	}
+	l, err := net.Listen("unix", socket)
+	if err != nil {
+		return err
+	}
+	go func() {
+		defer os.RemoveAll(socket)
+		defer srv.Close()
+		if serveErr := srv.Serve(l); serveErr != nil {
+			logrus.WithError(serveErr).Warn("hostagent API server exited with an error")
+		}
+	}()
 	return ha.Run(cmd.Context())
 }
 

docs/internal.md

@@ -50,6 +50,7 @@ Guest agent:
 
 Host agent:
 - `ha.pid`: hostagent PID
+- `ha.sock`: hostagent REST API
 - `ha.stdout.log`: hostagent stdout (JSON lines, see `pkg/hostagent/events.Event`)
 - `ha.stderr.log`: hostagent stderr (human-readable messages)
 

examples/alpine.yaml

@@ -1,3 +1,4 @@
+# This example requires Lima v0.7.0 or later.
 images:
 - location: https://github.com/lima-vm/alpine-lima/releases/download/v0.1.8/alpine-lima-std-3.13.5-x86_64.iso
   arch: "x86_64"
@@ -9,11 +10,6 @@ mounts:
 - location: "/tmp/lima"
   writable: true
 
-ssh:
-  # localPort is changed from 60022 to avoid conflicting with the default.
-  # (TODO: assign localPort automatically)
-  localPort: 60020
-
 firmware:
   legacyBIOS: true
 

examples/archlinux.yaml

@@ -1,4 +1,4 @@
-# This example requires Lima v0.5.0 or later
+# This example requires Lima v0.7.0 or later
 arch: "x86_64"
 images:
   # NOTE: the image is periodically rotated, if you face 404, see https://mirror.pkgbuild.com/images/ to find the latest image.
@@ -10,9 +10,5 @@ mounts:
     writable: false
   - location: "/tmp/lima"
     writable: true
-ssh:
-  # localPort is changed from 60022 to avoid conflicting with the default.
-  # (TODO: assign localPort automatically)
-  localPort: 60050
 firmware:
   legacyBIOS: true

examples/debian.yaml

@@ -1,3 +1,4 @@
+# This example requires Lima v0.7.0 or later
 images:
   - location: "https://cloud.debian.org/images/cloud/bullseye/daily/20211005-786/debian-11-generic-amd64-daily-20211005-786.qcow2"
     arch: "x86_64"
@@ -10,7 +11,3 @@ mounts:
     writable: false
   - location: "/tmp/lima"
     writable: true
-ssh:
-  # localPort is changed from 60022 to avoid conflicting with the default.
-  # (TODO: assign localPort automatically)
-  localPort: 60030

examples/fedora.yaml

@@ -1,3 +1,4 @@
+# This example requires Lima v0.7.0 or later.
 arch: "x86_64"
 images:
   - location: "https://download.fedoraproject.org/pub/fedora/linux/releases/34/Cloud/x86_64/images/Fedora-Cloud-Base-34-1.2.x86_64.qcow2"
@@ -8,10 +9,5 @@ mounts:
     writable: false
   - location: "/tmp/lima"
     writable: true
-ssh:
-  # localPort is changed from 60022 to avoid conflicting with the default.
-  # (TODO: assign localPort automatically)
-  localPort: 60024
-
 firmware:
   legacyBIOS: true

examples/k3s.yaml

@@ -8,6 +8,8 @@
 # $ kubectl get no
 # NAME       STATUS   ROLES                  AGE   VERSION
 # lima-k3s   Ready    control-plane,master   69s   v1.21.1+k3s1
+#
+# This example requires Lima v0.7.0 or later.
 
 images:
   # Hint: run `limactl prune` to invalidate the "current" cache
@@ -19,9 +21,6 @@ images:
 # Mounts are disabled in this example, but can be enabled optionally.
 mounts: []
 
-ssh:
-  localPort: 60022
-
 # containerd is managed by k3s, not by Lima, so the values are set to false here.
 containerd:
   system: false

examples/opensuse.yaml

@@ -1,4 +1,4 @@
-# This example requires Lima v0.5.0 or later
+# This example requires Lima v0.7.0 or later
 images:
   # Hint: run `limactl prune` to invalidate the "Current" cache
   - location: https://download.opensuse.org/distribution/leap/15.3/appliances/openSUSE-Leap-15.3-JeOS.x86_64-15.3-OpenStack-Cloud-Current.qcow2
@@ -9,7 +9,3 @@ mounts:
     writable: false
   - location: "/tmp/lima"
     writable: true
-ssh:
-  # localPort is changed from 60022 to avoid conflicting with the default.
-  # (TODO: assign localPort automatically)
-  localPort: 60044

examples/ubuntu.yaml

@@ -1,3 +1,4 @@
+# This example requires Lima v0.7.0 or later.
 images:
   # Hint: run `limactl prune` to invalidate the "current" cache
   - location: "https://cloud-images.ubuntu.com/hirsute/current/hirsute-server-cloudimg-amd64.img"
@@ -9,7 +10,3 @@ mounts:
     writable: false
   - location: "/tmp/lima"
     writable: true
-ssh:
-  # localPort is changed from 60022 to avoid conflicting with the default.
-  # (TODO: assign localPort automatically)
-  localPort: 60023