This commit implements Feature Request #59

client/client.go

@@ -2,6 +2,7 @@ package chclient
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"io"
 	"net"
@@ -73,12 +74,18 @@ func NewClient(config *Config) (*Client, error) {
 	//swap to websockets scheme
 	u.Scheme = strings.Replace(u.Scheme, "http", "ws", 1)
 	shared := &chshare.Config{}
+	stdioCnt := 0
 	createSocksServer := false
 	for _, s := range config.Remotes {
 		r, err := chshare.DecodeRemote(s)
 		if err != nil {
 			return nil, fmt.Errorf("Failed to decode remote '%s': %s", s, err)
 		}
+		if r.Stdio {
+			stdioCnt++
+		}
+		if stdioCnt > 1 {
+			return nil, errors.New("Only one stdio is allowed")
 		if r.Socks && r.Reverse {
 			createSocksServer = true
 		}

main.go

@@ -253,6 +253,7 @@ var clientHelp = `
       socks
       5000:socks
       R:2222:localhost:22
+      stdio:example.com:22
       R:socks
       R:5000:socks
 
@@ -270,6 +271,13 @@ var clientHelp = `
     default socks port (1080) and terminate the connection at the
     client's internal SOCKS5 proxy.
 
+    When stdio is used as local-host, the tunnel will connect standard
+    input/output of this program with the remote. This is useful when 
+    combined with ssh ProxyCommand. You can use
+      ssh -o ProxyCommand='chisel client chiselserver stdio:%h:%p' \
+          [email protected]
+    to connect to an SSH server through the tunnel.
+
   Options:
 
     --fingerprint, A *strongly recommended* fingerprint string

share/logger.go

@@ -20,7 +20,7 @@ func NewLogger(prefix string) *Logger {
 func NewLoggerFlag(prefix string, flag int) *Logger {
 	l := &Logger{
 		prefix: prefix,
-		logger: log.New(os.Stdout, "", flag),
+		logger: log.New(os.Stderr, "", flag),
 		Info:   false,
 		Debug:  false,
 	}

share/proxy.go

@@ -31,11 +31,25 @@ func NewTCPProxy(logger *Logger, ssh GetSSHConn, index int, remote *Remote) *TCP
 }
 
 func (p *TCPProxy) Start(ctx context.Context) error {
-	l, err := net.Listen("tcp4", p.remote.LocalHost+":"+p.remote.LocalPort)
-	if err != nil {
-		return fmt.Errorf("%s: %s", p.Logger.Prefix(), err)
+	if p.remote.Stdio {
+		go func() {
+			for {
+				p.accept(Stdio)
+				select {
+				case <-ctx.Done():
+					return
+				default:
+					// the connection is not ready yet
+				}
+			}
+		}()
+	} else {
+		l, err := net.Listen("tcp4", p.remote.LocalHost+":"+p.remote.LocalPort)
+		if err != nil {
+			return fmt.Errorf("%s: %s", p.Logger.Prefix(), err)
+		}
+		go p.listen(ctx, l)
 	}
-	go p.listen(ctx, l)
 	return nil
 }
 

share/remote.go

@@ -20,10 +20,16 @@ import (
 //   192.168.0.1:3000:google.com:80 ->
 //     local  192.168.0.1:3000
 //     remote google.com:80
+//   127.0.0.1:1080:socks
+//     local  127.0.0.1:1080
+//     remote socks
+//   stdio:example.com:22
+//     local  stdio
+//     remote example.com:22
 
 type Remote struct {
 	LocalHost, LocalPort, RemoteHost, RemotePort string
-	Socks, Reverse                               bool
+	Socks, Reverse, Stdio                        bool
 }
 
 const revPrefix = "R:"
@@ -46,6 +52,10 @@ func DecodeRemote(s string) (*Remote, error) {
 			r.Socks = true
 			continue
 		}
+		if i == 0 && p == "stdio" {
+			r.Stdio = true
+			continue
+		}
 		if isPort(p) {
 			if !r.Socks && r.RemotePort == "" {
 				r.RemotePort = p
@@ -80,6 +90,9 @@ func DecodeRemote(s string) (*Remote, error) {
 	if !r.Socks && r.RemoteHost == "" {
 		r.RemoteHost = "0.0.0.0"
 	}
+	if r.Stdio && r.Reverse {
+		return nil, errors.New("stdio cannot be used with reverse tunnel")
+	}
 	return r, nil
 }
 
@@ -106,7 +119,14 @@ func (r *Remote) String() string {
 	if r.Reverse {
 		tag = revPrefix
 	}
-	return tag + r.LocalHost + ":" + r.LocalPort + "=>" + r.Remote()
+	return tag + r.Local() + "=>" + r.Remote()
+}
+
+func (r *Remote) Local() string {
+	if r.Stdio {
+		return "stdio"
+	}
+	return r.LocalHost + ":" + r.LocalPort
 }
 
 func (r *Remote) Remote() string {

share/stdio.go

@@ -0,0 +1,23 @@
+package chshare
+
+import (
+	"os"
+)
+
+
+type stdio struct {}
+
+var Stdio = &stdio{}
+
+func (t *stdio) Read(b []byte) (n int, err error) {
+	return os.Stdin.Read(b)
+}
+
+func (t *stdio) Write(b []byte) (n int, err error) {
+	return os.Stdout.Write(b)
+}
+
+// We do not close stdio
+func (t *stdio) Close() error {
+	return nil
+}