lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-11-16 17:38:45 -05:00
Code Issues Releases Wiki Activity

more test cases for freedom

Browse Source
This commit is contained in:
Darien Raymond 2016-02-04 10:11:11 +00:00
parent 6e8b65551c
commit b18693a9c4
1 changed files with 35 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
proxy/freedom/freedom_test.go
View File

@ -1,27 +1,22 @@
package freedom package freedom_test
import ( import (
"fmt"
"net"
"testing" "testing"
"golang.org/x/net/proxy" "github.com/v2ray/v2ray-core/common/alloc"
v2net "github.com/v2ray/v2ray-core/common/net"
v2io "github.com/v2ray/v2ray-core/common/io"
v2nettesting "github.com/v2ray/v2ray-core/common/net/testing" v2nettesting "github.com/v2ray/v2ray-core/common/net/testing"
_ "github.com/v2ray/v2ray-core/proxy/socks" . "github.com/v2ray/v2ray-core/proxy/freedom"
"github.com/v2ray/v2ray-core/shell/point"
v2testing "github.com/v2ray/v2ray-core/testing" v2testing "github.com/v2ray/v2ray-core/testing"
"github.com/v2ray/v2ray-core/testing/assert" "github.com/v2ray/v2ray-core/testing/assert"
"github.com/v2ray/v2ray-core/testing/servers/tcp" "github.com/v2ray/v2ray-core/testing/servers/tcp"
"github.com/v2ray/v2ray-core/transport/ray"
) )
func TestSocksTcpConnect(t *testing.T) { func TestSinglePacket(t *testing.T) {
v2testing.Current(t) v2testing.Current(t)
port := v2nettesting.PickPort() port := v2nettesting.PickPort()
data2Send := "Data to be sent to remote"
tcpServer := &tcp.Server{ tcpServer := &tcp.Server{
Port: port, Port: port,
MsgProcessor: func(data []byte) []byte { MsgProcessor: func(data []byte) []byte {
@ -34,40 +29,38 @@ func TestSocksTcpConnect(t *testing.T) {
_, err := tcpServer.Start() _, err := tcpServer.Start()
assert.Error(err).IsNil() assert.Error(err).IsNil()
pointPort := v2nettesting.PickPort() freedom := &FreedomConnection{}
config := &point.Config{ traffic := ray.NewRay()
Port: pointPort, data2Send := "Data to be sent to remote"
InboundConfig: &point.ConnectionConfig{ payload := alloc.NewSmallBuffer().Clear().Append([]byte(data2Send))
Protocol: "socks", packet := v2net.NewPacket(v2net.TCPDestination(v2net.IPAddress([]byte{127, 0, 0, 1}), port), payload, false)
Settings: []byte(`{"auth": "noauth"}`),
}, err = freedom.Dispatch(packet, traffic)
OutboundConfig: &point.ConnectionConfig{ assert.Error(err).IsNil()
Protocol: "freedom", close(traffic.InboundInput())
Settings: nil,
}, respPayload := <-traffic.InboundOutput()
defer respPayload.Release()
assert.Bytes(respPayload.Value).Equals([]byte("Processed: Data to be sent to remote"))
_, open := <-traffic.InboundOutput()
assert.Bool(open).IsFalse()
tcpServer.Close()
} }
point, err := point.NewPoint(config) func TestUnreachableDestination(t *testing.T) {
assert.Error(err).IsNil() v2testing.Current(t)
err = point.Start() freedom := &FreedomConnection{}
assert.Error(err).IsNil() traffic := ray.NewRay()
data2Send := "Data to be sent to remote"
payload := alloc.NewSmallBuffer().Clear().Append([]byte(data2Send))
packet := v2net.NewPacket(v2net.TCPDestination(v2net.IPAddress([]byte{127, 0, 0, 2}), 80), payload, false)
socks5Client, err := proxy.SOCKS5("tcp", fmt.Sprintf("127.0.0.1:%d", pointPort), nil, proxy.Direct) err := freedom.Dispatch(packet, traffic)
assert.Error(err).IsNil() assert.Error(err).IsNotNil()
targetServer := fmt.Sprintf("127.0.0.1:%d", port) _, open := <-traffic.InboundOutput()
conn, err := socks5Client.Dial("tcp", targetServer) assert.Bool(open).IsFalse()
assert.Error(err).IsNil()
conn.Write([]byte(data2Send))
if tcpConn, ok := conn.(*net.TCPConn); ok {
tcpConn.CloseWrite()
}
dataReturned, err := v2io.ReadFrom(conn, nil)
assert.Error(err).IsNil()
conn.Close()
assert.Bytes(dataReturned.Value).Equals([]byte("Processed: Data to be sent to remote"))
} }