Add a --bind flag for local port

.gitignore

@@ -1,3 +1,4 @@
 vendor
 bin
 .vscode
+sshcode

main.go

@@ -37,6 +37,8 @@ type rootCmd struct {
 	skipSync     bool
 	syncBack     bool
 	printVersion bool
+	noOpen       bool
+	bindAddr     string
 	sshFlags     string
 }
 
@@ -52,6 +54,8 @@ func (c *rootCmd) RegisterFlags(fl *flag.FlagSet) {
 	fl.BoolVar(&c.skipSync, "skipsync", false, "skip syncing local settings and extensions to remote host")
 	fl.BoolVar(&c.syncBack, "b", false, "sync extensions back on termination")
 	fl.BoolVar(&c.printVersion, "version", false, "print version information and exit")
+	fl.BoolVar(&c.noOpen, "no-open", false, "do not open web browser")
+	fl.StringVar(&c.bindAddr, "bind", "", "local bind address for ssh tunnel")
 	fl.StringVar(&c.sshFlags, "ssh-flags", "", "custom SSH flags")
 }
 
@@ -76,6 +80,8 @@ func (c *rootCmd) Run(fl *flag.FlagSet) {
 	err := sshCode(host, dir, options{
 		skipSync: c.skipSync,
 		sshFlags: c.sshFlags,
+		bindAddr: c.bindAddr,
+		noOpen:   c.noOpen,
 		syncBack: c.syncBack,
 	})
 

sshcode.go

@@ -25,7 +25,7 @@ type options struct {
 	skipSync   bool
 	syncBack   bool
 	noOpen     bool
-	localPort  string
+	bindAddr   string
 	remotePort string
 	sshFlags   string
 }
@@ -79,9 +79,17 @@ func sshCode(host, dir string, o options) error {
 
 	flog.Info("starting code-server...")
 
-	if o.localPort == "" {
-		o.localPort, err = randomPort()
+	var bindHost string
+	var bindPort string
+	bindHost, bindPort, err = net.SplitHostPort(o.bindAddr)
+
+	if err != nil {
+		return xerrors.Errorf("failed to parse bind address: %w", err)
+	}
+	if bindPort == "" {
+		bindPort, err = randomPort()
 	}
+
 	if err != nil {
 		return xerrors.Errorf("failed to find available local port: %w", err)
 	}
@@ -93,11 +101,12 @@ func sshCode(host, dir string, o options) error {
 		return xerrors.Errorf("failed to find available remote port: %w", err)
 	}
 
-	flog.Info("Tunneling local port %v to remote port %v", o.localPort, o.remotePort)
+	flog.Info("Tunneling local host %v:%v to remote port %v", bindHost, bindPort, o.remotePort)
 
-	sshCmdStr = fmt.Sprintf("ssh -tt -q -L %v %v %v 'cd %v; %v --host 127.0.0.1 --allow-http --no-auth --port=%v'",
-		o.localPort+":localhost:"+o.remotePort, o.sshFlags, host, dir, codeServerPath, o.remotePort,
-	)
+	sshCmdStr =
+		fmt.Sprintf("ssh -tt -q -L %v:%v:localhost:%v %v %v 'cd %v; %v --host 127.0.0.1 --allow-http --no-auth --port=%v'",
+			bindHost, bindPort, o.remotePort, o.sshFlags, host, dir, codeServerPath, o.remotePort,
+		)
 
 	// Starts code-server and forwards the remote port.
 	sshCmd = exec.Command("sh", "-c", sshCmdStr)
@@ -109,7 +118,7 @@ func sshCode(host, dir string, o options) error {
 		return xerrors.Errorf("failed to start code-server: %w", err)
 	}
 
-	url := "http://127.0.0.1:" + o.localPort
+	url := fmt.Sprintf("http://%s", o.bindAddr)
 	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
 	defer cancel()
 

sshcode_test.go

@@ -37,7 +37,7 @@ func TestSSHCode(t *testing.T) {
 		defer wg.Done()
 		err := sshCode("foo@127.0.0.1", "", options{
 			sshFlags:   testSSHArgs(sshPort),
-			localPort:  localPort,
+			bindAddr:   fmt.Sprintf("127.0.0.1:%v", localPort),
 			remotePort: remotePort,
 			noOpen:     true,
 		})