|  | @@ -0,0 +1,129 @@
 | 
	
		
			
				|  |  | +package unixtest
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +import (
 | 
	
		
			
				|  |  | +	"testing"
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	"v2ray.com/core"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/app/log"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/app/proxyman"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/common"
 | 
	
		
			
				|  |  | +	clog "v2ray.com/core/common/log"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/common/net"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/common/serial"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/proxy/dokodemo"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/proxy/freedom"
 | 
	
		
			
				|  |  | +	. "v2ray.com/ext/assert"
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	"v2ray.com/core/testing/servers/tcp"
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	"v2ray.com/core/testing/scenarios"
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	"v2ray.com/core/transport/internet/domainsocket"
 | 
	
		
			
				|  |  | +)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func xor(b []byte) []byte {
 | 
	
		
			
				|  |  | +	r := make([]byte, len(b))
 | 
	
		
			
				|  |  | +	for i, v := range b {
 | 
	
		
			
				|  |  | +		r[i] = v ^ 'c'
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +	return r
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func pickPort() net.Port {
 | 
	
		
			
				|  |  | +	listener, err := net.Listen("tcp4", "127.0.0.1:0")
 | 
	
		
			
				|  |  | +	common.Must(err)
 | 
	
		
			
				|  |  | +	defer listener.Close()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	addr := listener.Addr().(*net.TCPAddr)
 | 
	
		
			
				|  |  | +	return net.Port(addr.Port)
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func TestUnix2tcpS(t *testing.T) {
 | 
	
		
			
				|  |  | +	assert := With(t)
 | 
	
		
			
				|  |  | +	tcpServer := tcp.Server{
 | 
	
		
			
				|  |  | +		MsgProcessor: xor,
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +	dest, err := tcpServer.Start()
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +	defer tcpServer.Close()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	Server := &core.Config{
 | 
	
		
			
				|  |  | +		App: []*serial.TypedMessage{
 | 
	
		
			
				|  |  | +			serial.ToTypedMessage(&log.Config{
 | 
	
		
			
				|  |  | +				ErrorLogLevel: clog.Severity_Debug,
 | 
	
		
			
				|  |  | +				ErrorLogType:  log.LogType_Console,
 | 
	
		
			
				|  |  | +			}),
 | 
	
		
			
				|  |  | +		},
 | 
	
		
			
				|  |  | +		Inbound: []*core.InboundHandlerConfig{
 | 
	
		
			
				|  |  | +			{
 | 
	
		
			
				|  |  | +				ReceiverSettings: serial.ToTypedMessage(&proxyman.UnixReceiverConfig{
 | 
	
		
			
				|  |  | +					DomainSockSettings: &domainsocket.DomainSocketSettings{Path: "\x00v2raytest"},
 | 
	
		
			
				|  |  | +				}),
 | 
	
		
			
				|  |  | +				ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
 | 
	
		
			
				|  |  | +					Address: net.NewIPOrDomain(dest.Address),
 | 
	
		
			
				|  |  | +					Port:    uint32(dest.Port),
 | 
	
		
			
				|  |  | +					NetworkList: &net.NetworkList{
 | 
	
		
			
				|  |  | +						Network: []net.Network{net.Network_TCP},
 | 
	
		
			
				|  |  | +					},
 | 
	
		
			
				|  |  | +				}),
 | 
	
		
			
				|  |  | +			},
 | 
	
		
			
				|  |  | +		},
 | 
	
		
			
				|  |  | +		Outbound: []*core.OutboundHandlerConfig{
 | 
	
		
			
				|  |  | +			{ProxySettings: serial.ToTypedMessage(&freedom.Config{})},
 | 
	
		
			
				|  |  | +		},
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +	serverPort := pickPort()
 | 
	
		
			
				|  |  | +	Client := &core.Config{
 | 
	
		
			
				|  |  | +		App: []*serial.TypedMessage{
 | 
	
		
			
				|  |  | +			serial.ToTypedMessage(&log.Config{
 | 
	
		
			
				|  |  | +				ErrorLogLevel: clog.Severity_Debug,
 | 
	
		
			
				|  |  | +				ErrorLogType:  log.LogType_Console,
 | 
	
		
			
				|  |  | +			}),
 | 
	
		
			
				|  |  | +		},
 | 
	
		
			
				|  |  | +		Outbound: []*core.OutboundHandlerConfig{
 | 
	
		
			
				|  |  | +			{
 | 
	
		
			
				|  |  | +				SenderSettings: serial.ToTypedMessage(&proxyman.UnixSenderConfig{
 | 
	
		
			
				|  |  | +					DomainSockSettings: &domainsocket.DomainSocketSettings{Path: "\x00v2raytest"},
 | 
	
		
			
				|  |  | +				}),
 | 
	
		
			
				|  |  | +				ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
 | 
	
		
			
				|  |  | +			},
 | 
	
		
			
				|  |  | +		},
 | 
	
		
			
				|  |  | +		Inbound: []*core.InboundHandlerConfig{
 | 
	
		
			
				|  |  | +			{ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
 | 
	
		
			
				|  |  | +				Address: net.NewIPOrDomain(dest.Address),
 | 
	
		
			
				|  |  | +				Port:    uint32(dest.Port),
 | 
	
		
			
				|  |  | +				NetworkList: &net.NetworkList{
 | 
	
		
			
				|  |  | +					Network: []net.Network{net.Network_TCP},
 | 
	
		
			
				|  |  | +				},
 | 
	
		
			
				|  |  | +			}),
 | 
	
		
			
				|  |  | +				ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
 | 
	
		
			
				|  |  | +					PortRange: net.SinglePortRange(serverPort),
 | 
	
		
			
				|  |  | +					Listen:    net.NewIPOrDomain(net.LocalHostIP),
 | 
	
		
			
				|  |  | +				}),
 | 
	
		
			
				|  |  | +			},
 | 
	
		
			
				|  |  | +		},
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	servers, err := scenarios.InitializeServerConfigs(Server, Client)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	port := serverPort
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	conn, err := net.DialTCP("tcp", nil, &net.TCPAddr{
 | 
	
		
			
				|  |  | +		IP:   []byte{127, 0, 0, 1},
 | 
	
		
			
				|  |  | +		Port: int(port),
 | 
	
		
			
				|  |  | +	})
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	payload := "dokodemo request."
 | 
	
		
			
				|  |  | +	nBytes, err := conn.Write([]byte(payload))
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +	assert(nBytes, Equals, len(payload))
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	response := make([]byte, 1024)
 | 
	
		
			
				|  |  | +	nBytes, err = conn.Read(response)
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +	assert(response[:nBytes], Equals, xor([]byte(payload)))
 | 
	
		
			
				|  |  | +	assert(conn.Close(), IsNil)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	scenarios.CloseAllServers(servers)
 | 
	
		
			
				|  |  | +}
 |