Stop hardcoding the default port for ssh

In case the Port is given in the ssh config already... The ssh command will know the default port, otherwise. Signed-off-by: Anders F Björklund <[email protected]>
lima-vm · Dec 30, 2021 · b89aa7e · b89aa7e
1 parent 611af10
commit b89aa7e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/cmd/sshocker/run.go b/cmd/sshocker/run.go
@@ -53,7 +53,7 @@ var (
 func parseHost(s string) (string, int, error) {
 	if !strings.Contains(s, ":") {
 		// FIXME: this check is not valid for IPv6!
-		return s, 22, nil
+		return s, 0, nil
 	}
 	host, portStr, err := net.SplitHostPort(s)
 	if err != nil {

diff --git a/pkg/reversesshfs/reversesshfs.go b/pkg/reversesshfs/reversesshfs.go
@@ -33,7 +33,11 @@ func (rsf *ReverseSSHFS) Prepare() error {
 	if !filepath.IsAbs(rsf.RemotePath) {
 		return errors.Errorf("unexpected relative path: %q", rsf.RemotePath)
 	}
-	sshArgs = append(sshArgs, "-p", strconv.Itoa(rsf.Port), rsf.Host, "--", "mkdir", "-p", rsf.RemotePath)
+	if rsf.Port != 0 {
+		sshArgs = append(sshArgs, "-p", strconv.Itoa(rsf.Port))
+	}
+	sshArgs = append(sshArgs, rsf.Host, "--")
+	sshArgs = append(sshArgs, "mkdir", "-p", rsf.RemotePath)
 	sshCmd := exec.Command(sshBinary, sshArgs...)
 	logrus.Debugf("executing ssh for preparing sshfs: %s %v", sshCmd.Path, sshCmd.Args)
 	out, err := sshCmd.CombinedOutput()
@@ -52,7 +56,11 @@ func (rsf *ReverseSSHFS) Start() error {
 	if !filepath.IsAbs(rsf.RemotePath) {
 		return errors.Errorf("unexpected relative path: %q", rsf.RemotePath)
 	}
-	sshArgs = append(sshArgs, "-p", strconv.Itoa(rsf.Port), rsf.Host, "--", "sshfs", ":"+rsf.LocalPath, rsf.RemotePath, "-o", "slave")
+	if rsf.Port != 0 {
+		sshArgs = append(sshArgs, "-p", strconv.Itoa(rsf.Port))
+	}
+	sshArgs = append(sshArgs, rsf.Host, "--")
+	sshArgs = append(sshArgs, "sshfs", ":"+rsf.LocalPath, rsf.RemotePath, "-o", "slave")
 	if rsf.Readonly {
 		sshArgs = append(sshArgs, "-o", "ro")
 	}

diff --git a/pkg/sshocker/sshocker.go b/pkg/sshocker/sshocker.go
@@ -31,7 +31,10 @@ func (x *Sshocker) Run() error {
 	for _, l := range x.LForwards {
 		args = append(args, "-L", l)
 	}
-	args = append(args, "-p", strconv.Itoa(x.Port), x.Host, "--")
+	if x.Port != 0 {
+		args = append(args, "-p", strconv.Itoa(x.Port))
+	}
+	args = append(args, x.Host, "--")
 	if len(x.Command) > 0 {
 		args = append(args, x.Command...)
 	}