2015-09-11 08:12:09 -04:00
|
|
|
package socks
|
|
|
|
|
|
|
|
import (
|
2015-09-19 09:35:20 -04:00
|
|
|
"bytes"
|
2015-11-01 15:32:08 -05:00
|
|
|
"fmt"
|
2015-09-19 09:35:20 -04:00
|
|
|
"io/ioutil"
|
|
|
|
"net"
|
2015-09-11 08:12:09 -04:00
|
|
|
"testing"
|
|
|
|
|
2015-09-19 09:35:20 -04:00
|
|
|
"golang.org/x/net/proxy"
|
|
|
|
|
2015-10-14 08:56:11 -04:00
|
|
|
"github.com/v2ray/v2ray-core/app/point"
|
2015-11-01 15:32:08 -05:00
|
|
|
v2nettesting "github.com/v2ray/v2ray-core/common/net/testing"
|
2015-10-29 19:11:29 -04:00
|
|
|
"github.com/v2ray/v2ray-core/proxy/common/connhandler"
|
2015-10-06 17:11:08 -04:00
|
|
|
"github.com/v2ray/v2ray-core/proxy/socks/config/json"
|
2015-11-01 15:15:08 -05:00
|
|
|
proxymocks "github.com/v2ray/v2ray-core/proxy/testing/mocks"
|
2015-09-19 09:35:20 -04:00
|
|
|
"github.com/v2ray/v2ray-core/testing/mocks"
|
|
|
|
"github.com/v2ray/v2ray-core/testing/unit"
|
2015-09-11 08:12:09 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestSocksTcpConnect(t *testing.T) {
|
2015-09-19 09:35:20 -04:00
|
|
|
assert := unit.Assert(t)
|
2015-11-01 15:32:08 -05:00
|
|
|
port := v2nettesting.PickPort()
|
2015-09-19 09:35:20 -04:00
|
|
|
|
2015-11-01 15:15:08 -05:00
|
|
|
connInput := []byte("The data to be returned to socks server.")
|
|
|
|
connOutput := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
och := &proxymocks.OutboundConnectionHandler{
|
|
|
|
ConnOutput: connOutput,
|
|
|
|
ConnInput: bytes.NewReader(connInput),
|
2015-09-19 09:35:20 -04:00
|
|
|
}
|
|
|
|
|
2015-10-29 19:11:29 -04:00
|
|
|
connhandler.RegisterOutboundConnectionHandlerFactory("mock_och", och)
|
2015-09-19 09:35:20 -04:00
|
|
|
|
|
|
|
config := mocks.Config{
|
|
|
|
PortValue: port,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "socks",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: &json.SocksConfig{
|
|
|
|
AuthMethod: "noauth",
|
|
|
|
},
|
2015-09-19 09:35:20 -04:00
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_och",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: nil,
|
2015-09-19 09:35:20 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2015-10-14 08:56:11 -04:00
|
|
|
point, err := point.NewPoint(&config)
|
2015-09-19 09:35:20 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = point.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
2015-11-01 15:32:08 -05:00
|
|
|
socks5Client, err := proxy.SOCKS5("tcp", fmt.Sprintf("127.0.0.1:%d", port), nil, proxy.Direct)
|
2015-09-19 09:35:20 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
2015-09-19 09:42:46 -04:00
|
|
|
targetServer := "google.com:80"
|
|
|
|
conn, err := socks5Client.Dial("tcp", targetServer)
|
2015-09-19 09:35:20 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
data2Send := "The data to be sent to remote server."
|
|
|
|
conn.Write([]byte(data2Send))
|
|
|
|
if tcpConn, ok := conn.(*net.TCPConn); ok {
|
|
|
|
tcpConn.CloseWrite()
|
|
|
|
}
|
|
|
|
|
|
|
|
dataReturned, err := ioutil.ReadAll(conn)
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
conn.Close()
|
|
|
|
|
2015-11-01 15:15:08 -05:00
|
|
|
assert.Bytes([]byte(data2Send)).Equals(connOutput.Bytes())
|
|
|
|
assert.Bytes(dataReturned).Equals(connInput)
|
2015-09-20 10:03:12 -04:00
|
|
|
assert.String(targetServer).Equals(och.Destination.Address().String())
|
2015-09-19 09:42:46 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestSocksTcpConnectWithUserPass(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
2015-11-01 15:32:08 -05:00
|
|
|
port := v2nettesting.PickPort()
|
2015-09-19 09:42:46 -04:00
|
|
|
|
2015-11-01 15:15:08 -05:00
|
|
|
connInput := []byte("The data to be returned to socks server.")
|
|
|
|
connOutput := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
och := &proxymocks.OutboundConnectionHandler{
|
|
|
|
ConnInput: bytes.NewReader(connInput),
|
|
|
|
ConnOutput: connOutput,
|
2015-09-19 09:42:46 -04:00
|
|
|
}
|
|
|
|
|
2015-10-29 19:11:29 -04:00
|
|
|
connhandler.RegisterOutboundConnectionHandlerFactory("mock_och", och)
|
2015-09-19 09:42:46 -04:00
|
|
|
|
|
|
|
config := mocks.Config{
|
|
|
|
PortValue: port,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "socks",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: &json.SocksConfig{
|
2015-10-10 16:31:25 -04:00
|
|
|
AuthMethod: "password",
|
2015-11-03 16:09:07 -05:00
|
|
|
Accounts: json.SocksAccountMap{
|
|
|
|
"userx": "passy",
|
2015-10-10 09:51:35 -04:00
|
|
|
},
|
2015-10-06 17:11:08 -04:00
|
|
|
},
|
2015-09-19 09:42:46 -04:00
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_och",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: nil,
|
2015-09-19 09:42:46 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2015-10-14 08:56:11 -04:00
|
|
|
point, err := point.NewPoint(&config)
|
2015-09-19 09:42:46 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = point.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
2015-11-01 15:32:08 -05:00
|
|
|
socks5Client, err := proxy.SOCKS5("tcp", fmt.Sprintf("127.0.0.1:%d", port), &proxy.Auth{"userx", "passy"}, proxy.Direct)
|
2015-09-19 09:42:46 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
targetServer := "1.2.3.4:443"
|
|
|
|
conn, err := socks5Client.Dial("tcp", targetServer)
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
data2Send := "The data to be sent to remote server."
|
|
|
|
conn.Write([]byte(data2Send))
|
|
|
|
if tcpConn, ok := conn.(*net.TCPConn); ok {
|
|
|
|
tcpConn.CloseWrite()
|
|
|
|
}
|
|
|
|
|
|
|
|
dataReturned, err := ioutil.ReadAll(conn)
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
conn.Close()
|
|
|
|
|
2015-11-01 15:15:08 -05:00
|
|
|
assert.Bytes([]byte(data2Send)).Equals(connOutput.Bytes())
|
|
|
|
assert.Bytes(dataReturned).Equals(connInput)
|
2015-09-20 10:03:12 -04:00
|
|
|
assert.String(targetServer).Equals(och.Destination.Address().String())
|
2015-09-11 08:12:09 -04:00
|
|
|
}
|
2015-10-03 18:44:27 -04:00
|
|
|
|
2015-10-13 15:29:27 -04:00
|
|
|
func TestSocksTcpConnectWithWrongUserPass(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
2015-11-01 15:32:08 -05:00
|
|
|
port := v2nettesting.PickPort()
|
2015-10-13 15:29:27 -04:00
|
|
|
|
2015-11-01 15:15:08 -05:00
|
|
|
connInput := []byte("The data to be returned to socks server.")
|
|
|
|
connOutput := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
och := &proxymocks.OutboundConnectionHandler{
|
|
|
|
ConnInput: bytes.NewReader(connInput),
|
|
|
|
ConnOutput: connOutput,
|
2015-10-13 15:29:27 -04:00
|
|
|
}
|
|
|
|
|
2015-10-29 19:11:29 -04:00
|
|
|
connhandler.RegisterOutboundConnectionHandlerFactory("mock_och", och)
|
2015-10-13 15:29:27 -04:00
|
|
|
|
|
|
|
config := mocks.Config{
|
|
|
|
PortValue: port,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "socks",
|
|
|
|
SettingsValue: &json.SocksConfig{
|
|
|
|
AuthMethod: "password",
|
2015-11-03 16:09:07 -05:00
|
|
|
Accounts: json.SocksAccountMap{
|
|
|
|
"userx": "passy",
|
2015-10-13 15:29:27 -04:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_och",
|
|
|
|
SettingsValue: nil,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2015-10-14 08:56:11 -04:00
|
|
|
point, err := point.NewPoint(&config)
|
2015-10-13 15:29:27 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = point.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
2015-11-01 15:32:08 -05:00
|
|
|
socks5Client, err := proxy.SOCKS5("tcp", fmt.Sprintf("127.0.0.1:%d", port), &proxy.Auth{"userx", "passz"}, proxy.Direct)
|
2015-10-13 15:29:27 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
targetServer := "1.2.3.4:443"
|
|
|
|
_, err = socks5Client.Dial("tcp", targetServer)
|
|
|
|
assert.Error(err).IsNotNil()
|
|
|
|
}
|
|
|
|
|
2015-10-13 15:55:12 -04:00
|
|
|
func TestSocksTcpConnectWithWrongAuthMethod(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
2015-11-01 15:32:08 -05:00
|
|
|
port := v2nettesting.PickPort()
|
2015-10-13 15:55:12 -04:00
|
|
|
|
2015-11-01 15:15:08 -05:00
|
|
|
connInput := []byte("The data to be returned to socks server.")
|
|
|
|
connOutput := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
och := &proxymocks.OutboundConnectionHandler{
|
|
|
|
ConnInput: bytes.NewReader(connInput),
|
|
|
|
ConnOutput: connOutput,
|
2015-10-13 15:55:12 -04:00
|
|
|
}
|
|
|
|
|
2015-10-29 19:11:29 -04:00
|
|
|
connhandler.RegisterOutboundConnectionHandlerFactory("mock_och", och)
|
2015-10-13 15:55:12 -04:00
|
|
|
|
|
|
|
config := mocks.Config{
|
|
|
|
PortValue: port,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "socks",
|
|
|
|
SettingsValue: &json.SocksConfig{
|
|
|
|
AuthMethod: "password",
|
2015-11-03 16:09:07 -05:00
|
|
|
Accounts: json.SocksAccountMap{
|
|
|
|
"userx": "passy",
|
2015-10-13 15:55:12 -04:00
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_och",
|
|
|
|
SettingsValue: nil,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2015-10-14 08:56:11 -04:00
|
|
|
point, err := point.NewPoint(&config)
|
2015-10-13 15:55:12 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = point.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
2015-11-01 15:32:08 -05:00
|
|
|
socks5Client, err := proxy.SOCKS5("tcp", fmt.Sprintf("127.0.0.1:%d", port), nil, proxy.Direct)
|
2015-10-13 15:55:12 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
targetServer := "1.2.3.4:443"
|
|
|
|
_, err = socks5Client.Dial("tcp", targetServer)
|
|
|
|
assert.Error(err).IsNotNil()
|
|
|
|
}
|
|
|
|
|
2015-10-03 18:44:27 -04:00
|
|
|
func TestSocksUdpSend(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
2015-11-01 15:32:08 -05:00
|
|
|
port := v2nettesting.PickPort()
|
2015-10-03 18:44:27 -04:00
|
|
|
|
2015-11-01 15:15:08 -05:00
|
|
|
connInput := []byte("The data to be returned to socks server.")
|
|
|
|
connOutput := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
|
|
och := &proxymocks.OutboundConnectionHandler{
|
|
|
|
ConnInput: bytes.NewReader(connInput),
|
|
|
|
ConnOutput: connOutput,
|
2015-10-03 18:44:27 -04:00
|
|
|
}
|
|
|
|
|
2015-10-29 19:11:29 -04:00
|
|
|
connhandler.RegisterOutboundConnectionHandlerFactory("mock_och", och)
|
2015-10-03 18:44:27 -04:00
|
|
|
|
|
|
|
config := mocks.Config{
|
|
|
|
PortValue: port,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "socks",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: &json.SocksConfig{
|
|
|
|
AuthMethod: "noauth",
|
|
|
|
UDPEnabled: true,
|
|
|
|
},
|
2015-10-03 18:44:27 -04:00
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_och",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: nil,
|
2015-10-03 18:44:27 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2015-10-14 08:56:11 -04:00
|
|
|
point, err := point.NewPoint(&config)
|
2015-10-03 18:44:27 -04:00
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = point.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
conn, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
|
|
|
IP: []byte{127, 0, 0, 1},
|
|
|
|
Port: int(port),
|
|
|
|
Zone: "",
|
|
|
|
})
|
|
|
|
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
data2Send := []byte("Fake DNS request")
|
|
|
|
|
|
|
|
buffer := make([]byte, 0, 1024)
|
|
|
|
buffer = append(buffer, 0, 0, 0)
|
|
|
|
buffer = append(buffer, 1, 8, 8, 4, 4, 0, 53)
|
|
|
|
buffer = append(buffer, data2Send...)
|
|
|
|
|
|
|
|
conn.Write(buffer)
|
|
|
|
|
|
|
|
response := make([]byte, 1024)
|
|
|
|
nBytes, err := conn.Read(response)
|
|
|
|
|
|
|
|
assert.Error(err).IsNil()
|
2015-11-01 15:15:08 -05:00
|
|
|
assert.Bytes(response[10:nBytes]).Equals(connInput)
|
|
|
|
assert.Bytes(data2Send).Equals(connOutput.Bytes())
|
2015-10-03 18:44:27 -04:00
|
|
|
assert.String(och.Destination.String()).Equals("udp:8.8.4.4:53")
|
|
|
|
}
|