mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-12 15:27:16 -05:00
140 lines
3.3 KiB
Go
140 lines
3.3 KiB
Go
package dokodemo
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
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/shell/point"
|
|
"github.com/v2ray/v2ray-core/shell/point/config/testing/mocks"
|
|
"github.com/v2ray/v2ray-core/testing/servers/tcp"
|
|
"github.com/v2ray/v2ray-core/testing/servers/udp"
|
|
"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]))
|
|
}
|
|
|
|
func TestDokodemoUDP(t *testing.T) {
|
|
assert := unit.Assert(t)
|
|
|
|
port := v2nettesting.PickPort()
|
|
|
|
data2Send := "Data to be sent to remote."
|
|
|
|
udpServer := &udp.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 := udpServer.Start()
|
|
assert.Error(err).IsNil()
|
|
|
|
pointPort := v2nettesting.PickPort()
|
|
networkList := v2netjson.NetworkList([]string{"udp"})
|
|
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()
|
|
|
|
udpClient, err := net.DialUDP("udp", nil, &net.UDPAddr{
|
|
IP: []byte{127, 0, 0, 1},
|
|
Port: int(pointPort),
|
|
Zone: "",
|
|
})
|
|
assert.Error(err).IsNil()
|
|
|
|
udpClient.Write([]byte(data2Send))
|
|
|
|
response := make([]byte, 1024)
|
|
nBytes, err := udpClient.Read(response)
|
|
assert.Error(err).IsNil()
|
|
udpClient.Close()
|
|
|
|
assert.String("Processed: " + data2Send).Equals(string(response[:nBytes]))
|
|
}
|