Randy Reddig
b2d7c26edb
This change adds DialContext to ssh.Client, which opens a TCP-IP connection tunneled over the SSH connection. This is useful for proxying network connections, e.g. setting (net/http.Transport).DialContext. Fixes golang/go#20288. Change-Id: I110494c00962424ea803065535ebe2209364ac27 GitHub-Last-Rev: 3176984a71a9a1422702e3a071340ecfff71ff62 GitHub-Pull-Request: golang/crypto#260 Reviewed-on: https://go-review.googlesource.com/c/crypto/+/504735 Run-TryBot: Nicola Murino <nicola.murino@gmail.com> Run-TryBot: Han-Wen Nienhuys <hanwen@google.com> Auto-Submit: Nicola Murino <nicola.murino@gmail.com> Reviewed-by: Han-Wen Nienhuys <hanwen@google.com> Reviewed-by: Dmitri Shuralyov <dmitshur@google.com> TryBot-Result: Gopher Robot <gobot@golang.org> Reviewed-by: Nicola Murino <nicola.murino@gmail.com> Commit-Queue: Nicola Murino <nicola.murino@gmail.com>
132 lines
3.2 KiB
Go
132 lines
3.2 KiB
Go
// Copyright 2012 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
//go:build !windows && !js && !wasip1
|
|
|
|
package test
|
|
|
|
// direct-tcpip and direct-streamlocal functional tests
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type dialTester interface {
|
|
TestServerConn(t *testing.T, c net.Conn)
|
|
TestClientConn(t *testing.T, c net.Conn)
|
|
}
|
|
|
|
func testDial(t *testing.T, n, listenAddr string, x dialTester) {
|
|
server := newServer(t)
|
|
sshConn := server.Dial(clientConfig())
|
|
defer sshConn.Close()
|
|
|
|
l, err := net.Listen(n, listenAddr)
|
|
if err != nil {
|
|
t.Fatalf("Listen: %v", err)
|
|
}
|
|
defer l.Close()
|
|
|
|
testData := fmt.Sprintf("hello from %s, %s", n, listenAddr)
|
|
go func() {
|
|
for {
|
|
c, err := l.Accept()
|
|
if err != nil {
|
|
break
|
|
}
|
|
x.TestServerConn(t, c)
|
|
|
|
io.WriteString(c, testData)
|
|
c.Close()
|
|
}
|
|
}()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
conn, err := sshConn.DialContext(ctx, n, l.Addr().String())
|
|
// Canceling the context after dial should have no effect
|
|
// on the opened connection.
|
|
cancel()
|
|
if err != nil {
|
|
t.Fatalf("Dial: %v", err)
|
|
}
|
|
x.TestClientConn(t, conn)
|
|
defer conn.Close()
|
|
b, err := io.ReadAll(conn)
|
|
if err != nil {
|
|
t.Fatalf("ReadAll: %v", err)
|
|
}
|
|
t.Logf("got %q", string(b))
|
|
if string(b) != testData {
|
|
t.Fatalf("expected %q, got %q", testData, string(b))
|
|
}
|
|
}
|
|
|
|
type tcpDialTester struct {
|
|
listenAddr string
|
|
}
|
|
|
|
func (x *tcpDialTester) TestServerConn(t *testing.T, c net.Conn) {
|
|
host := strings.Split(x.listenAddr, ":")[0]
|
|
prefix := host + ":"
|
|
if !strings.HasPrefix(c.LocalAddr().String(), prefix) {
|
|
t.Fatalf("expected to start with %q, got %q", prefix, c.LocalAddr().String())
|
|
}
|
|
if !strings.HasPrefix(c.RemoteAddr().String(), prefix) {
|
|
t.Fatalf("expected to start with %q, got %q", prefix, c.RemoteAddr().String())
|
|
}
|
|
}
|
|
|
|
func (x *tcpDialTester) TestClientConn(t *testing.T, c net.Conn) {
|
|
// we use zero addresses. see *Client.Dial.
|
|
if c.LocalAddr().String() != "0.0.0.0:0" {
|
|
t.Fatalf("expected \"0.0.0.0:0\", got %q", c.LocalAddr().String())
|
|
}
|
|
if c.RemoteAddr().String() != "0.0.0.0:0" {
|
|
t.Fatalf("expected \"0.0.0.0:0\", got %q", c.RemoteAddr().String())
|
|
}
|
|
}
|
|
|
|
func TestDialTCP(t *testing.T) {
|
|
x := &tcpDialTester{
|
|
listenAddr: "127.0.0.1:0",
|
|
}
|
|
testDial(t, "tcp", x.listenAddr, x)
|
|
}
|
|
|
|
type unixDialTester struct {
|
|
listenAddr string
|
|
}
|
|
|
|
func (x *unixDialTester) TestServerConn(t *testing.T, c net.Conn) {
|
|
if c.LocalAddr().String() != x.listenAddr {
|
|
t.Fatalf("expected %q, got %q", x.listenAddr, c.LocalAddr().String())
|
|
}
|
|
if c.RemoteAddr().String() != "@" && c.RemoteAddr().String() != "" {
|
|
t.Fatalf("expected \"@\" or \"\", got %q", c.RemoteAddr().String())
|
|
}
|
|
}
|
|
|
|
func (x *unixDialTester) TestClientConn(t *testing.T, c net.Conn) {
|
|
if c.RemoteAddr().String() != x.listenAddr {
|
|
t.Fatalf("expected %q, got %q", x.listenAddr, c.RemoteAddr().String())
|
|
}
|
|
if c.LocalAddr().String() != "@" {
|
|
t.Fatalf("expected \"@\", got %q", c.LocalAddr().String())
|
|
}
|
|
}
|
|
|
|
func TestDialUnix(t *testing.T) {
|
|
addr, cleanup := newTempSocket(t)
|
|
defer cleanup()
|
|
x := &unixDialTester{
|
|
listenAddr: addr,
|
|
}
|
|
testDial(t, "unix", x.listenAddr, x)
|
|
}
|