diff --git a/socks5_test.go b/socks5_test.go new file mode 100644 index 0000000..5f32e28 --- /dev/null +++ b/socks5_test.go @@ -0,0 +1,103 @@ +package socks5 + +import ( + "bytes" + "encoding/binary" + "io" + "net" + "testing" + "time" +) + +func TestSOCKS5_Connect(t *testing.T) { + // Create a local listener + l, err := net.Listen("tcp", "127.0.0.1:0") + if err != nil { + t.Fatalf("err: %v", err) + } + go func() { + conn, err := l.Accept() + if err != nil { + t.Fatalf("err: %v", err) + } + defer conn.Close() + + buf := make([]byte, 4) + if _, err := io.ReadAtLeast(conn, buf, 4); err != nil { + t.Fatalf("err: %v", err) + } + + if !bytes.Equal(buf, []byte("ping")) { + t.Fatalf("bad: %v", buf) + } + conn.Write([]byte("pong")) + }() + lAddr := l.Addr().(*net.TCPAddr) + + // Create a socks server + creds := StaticCredentials{ + "foo": "bar", + } + conf := &Config{ + Credentials: creds, + } + serv, err := New(conf) + if err != nil { + t.Fatalf("err: %v", err) + } + + // Start listening + go func() { + if err := serv.ListenAndServe("tcp", "127.0.0.1:12365"); err != nil { + t.Fatalf("err: %v", err) + } + }() + time.Sleep(10 * time.Millisecond) + + // Get a local conn + conn, err := net.Dial("tcp", "127.0.0.1:12365") + if err != nil { + t.Fatalf("err: %v", err) + } + + // Connect, auth and connec to local + req := bytes.NewBuffer(nil) + req.Write([]byte{5}) + req.Write([]byte{2, noAuth, userPassAuth}) + req.Write([]byte{1, 3, 'f', 'o', 'o', 3, 'b', 'a', 'r'}) + req.Write([]byte{5, 1, 0, 1, 127, 0, 0, 1}) + + port := []byte{0, 0} + binary.BigEndian.PutUint16(port, uint16(lAddr.Port)) + req.Write(port) + + // Send a ping + req.Write([]byte("ping")) + + // Send all the bytes + conn.Write(req.Bytes()) + + // Verify response + expected := []byte{ + socks5Version, userPassAuth, + 1, authSuccess, + 5, + 0, + 0, + 1, + 127, 0, 0, 1, + port[0], + port[1], + 'p', 'o', 'n', 'g', + } + out := make([]byte, len(expected)) + + conn.SetDeadline(time.Now().Add(time.Second)) + if _, err := io.ReadAtLeast(conn, out, len(out)); err != nil { + t.Fatalf("err: %v", err) + } + + if !bytes.Equal(out, expected) { + t.Fatalf("bad: %v", out) + } +}