mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-05 07:55:36 +00:00
7eaf5e509f
Updates #5162 Change-Id: Id7bdec303b25471f69d542f8ce43805328d56c12 Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package socks5
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"testing"
|
|
|
|
"golang.org/x/net/proxy"
|
|
)
|
|
|
|
func socks5Server(listener net.Listener) {
|
|
var server Server
|
|
err := server.Serve(listener)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
listener.Close()
|
|
}
|
|
|
|
func backendServer(listener net.Listener) {
|
|
conn, err := listener.Accept()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
conn.Write([]byte("Test"))
|
|
conn.Close()
|
|
listener.Close()
|
|
}
|
|
|
|
func TestRead(t *testing.T) {
|
|
// backend server which we'll use SOCKS5 to connect to
|
|
listener, err := net.Listen("tcp", ":0")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
backendServerPort := listener.Addr().(*net.TCPAddr).Port
|
|
go backendServer(listener)
|
|
|
|
// SOCKS5 server
|
|
socks5, err := net.Listen("tcp", ":0")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
socks5Port := socks5.Addr().(*net.TCPAddr).Port
|
|
go socks5Server(socks5)
|
|
|
|
addr := fmt.Sprintf("localhost:%d", socks5Port)
|
|
socksDialer, err := proxy.SOCKS5("tcp", addr, nil, proxy.Direct)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
addr = fmt.Sprintf("localhost:%d", backendServerPort)
|
|
conn, err := socksDialer.Dial("tcp", addr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
buf := make([]byte, 4)
|
|
_, err = io.ReadFull(conn, buf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if string(buf) != "Test" {
|
|
t.Fatalf("got: %q want: Test", buf)
|
|
}
|
|
|
|
err = conn.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|