add dial

Author: wzshiming

dial.go

@@ -0,0 +1,27 @@
+package commandproxy
+
+import (
+	"context"
+	"net"
+)
+
+type DialProxyCommand []string
+
+func (p *DialProxyCommand) DialContext(ctx context.Context, network string, address string) (net.Conn, error) {
+	host, port, err := net.SplitHostPort(address)
+	if err != nil {
+		return nil, err
+	}
+	m := map[byte]string{
+		'n': network,
+		'h': host,
+		'p': port,
+	}
+	proxy := make([]string, len(*p))
+	copy(proxy, *p)
+	for i := range proxy {
+		proxy[i] = ReplaceEscape(proxy[i], m)
+	}
+	cp := ProxyCommand(ctx, proxy[0], proxy[1:]...)
+	return cp.Stdio()
+}

dial_test.go

@@ -0,0 +1,28 @@
+package commandproxy
+
+import (
+	"context"
+	"net"
+	"testing"
+)
+
+func TestDialProxyCommand(t *testing.T) {
+	listener, err := net.Listen("tcp", ":0")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	command, err := SplitCommand("nc %n %p")
+	if err != nil {
+		t.Fatal(err)
+	}
+	dial := DialProxyCommand(command)
+	conn, err := dial.DialContext(context.Background(), listener.Addr().Network(), listener.Addr().String())
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = conn.Close()
+	if err != nil {
+		t.Fatal(err)
+	}
+}