mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
1.8 KiB
78 lines
1.8 KiB
package dokodemo
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/v2ray/v2ray-core/app/point"
|
|
v2netjson "github.com/v2ray/v2ray-core/common/net/json"
|
|
v2nettesting "github.com/v2ray/v2ray-core/common/net/testing"
|
|
"github.com/v2ray/v2ray-core/proxy/dokodemo/config/json"
|
|
_ "github.com/v2ray/v2ray-core/proxy/freedom"
|
|
"github.com/v2ray/v2ray-core/testing/mocks"
|
|
"github.com/v2ray/v2ray-core/testing/servers/tcp"
|
|
"github.com/v2ray/v2ray-core/testing/unit"
|
|
)
|
|
|
|
func TestDokodemoTCP(t *testing.T) {
|
|
assert := unit.Assert(t)
|
|
|
|
port := v2nettesting.PickPort()
|
|
|
|
data2Send := "Data to be sent to remote."
|
|
|
|
tcpServer := &tcp.Server{
|
|
Port: port,
|
|
MsgProcessor: func(data []byte) []byte {
|
|
buffer := make([]byte, 0, 2048)
|
|
buffer = append(buffer, []byte("Processed: ")...)
|
|
buffer = append(buffer, data...)
|
|
return buffer
|
|
},
|
|
}
|
|
_, err := tcpServer.Start()
|
|
assert.Error(err).IsNil()
|
|
|
|
pointPort := v2nettesting.PickPort()
|
|
networkList := v2netjson.NetworkList([]string{"tcp"})
|
|
config := mocks.Config{
|
|
PortValue: pointPort,
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
ProtocolValue: "dokodemo-door",
|
|
SettingsValue: &json.DokodemoConfig{
|
|
Host: "127.0.0.1",
|
|
Port: int(port),
|
|
Network: &networkList,
|
|
Timeout: 0,
|
|
},
|
|
},
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
ProtocolValue: "freedom",
|
|
SettingsValue: nil,
|
|
},
|
|
}
|
|
|
|
point, err := point.NewPoint(&config)
|
|
assert.Error(err).IsNil()
|
|
|
|
err = point.Start()
|
|
assert.Error(err).IsNil()
|
|
|
|
tcpClient, err := net.DialTCP("tcp", nil, &net.TCPAddr{
|
|
IP: []byte{127, 0, 0, 1},
|
|
Port: int(pointPort),
|
|
Zone: "",
|
|
})
|
|
assert.Error(err).IsNil()
|
|
|
|
tcpClient.Write([]byte(data2Send))
|
|
tcpClient.CloseWrite()
|
|
|
|
response := make([]byte, 1024)
|
|
nBytes, err := tcpClient.Read(response)
|
|
assert.Error(err).IsNil()
|
|
tcpClient.Close()
|
|
|
|
assert.String("Processed: " + data2Send).Equals(string(response[:nBytes]))
|
|
}
|