2015-09-19 15:56:09 -04:00
|
|
|
package vmess
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/v2ray/v2ray-core"
|
2015-10-08 08:46:18 -04:00
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
2015-09-19 17:54:36 -04:00
|
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
2015-09-19 15:56:09 -04:00
|
|
|
"github.com/v2ray/v2ray-core/testing/mocks"
|
|
|
|
"github.com/v2ray/v2ray-core/testing/unit"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestVMessInAndOut(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
|
|
|
|
|
|
|
data2Send := "The data to be send to outbound server."
|
|
|
|
|
|
|
|
portA := uint16(17392)
|
|
|
|
ich := &mocks.InboundConnectionHandler{
|
|
|
|
Data2Send: []byte(data2Send),
|
|
|
|
DataReturned: bytes.NewBuffer(make([]byte, 0, 1024)),
|
|
|
|
}
|
|
|
|
|
|
|
|
core.RegisterInboundConnectionHandlerFactory("mock_ich", ich)
|
|
|
|
|
|
|
|
configA := mocks.Config{
|
|
|
|
PortValue: portA,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_ich",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: nil,
|
2015-09-19 15:56:09 -04:00
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "vmess",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: &VMessOutboundConfig{
|
|
|
|
[]VNextConfig{
|
|
|
|
VNextConfig{
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Port: 13829,
|
|
|
|
Network: "tcp",
|
|
|
|
Users: []VMessUser{
|
|
|
|
VMessUser{Id: "ad937d9d-6e23-4a5a-ba23-bce5092a7c51"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2015-09-19 15:56:09 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
pointA, err := core.NewPoint(&configA)
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = pointA.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
portB := uint16(13829)
|
|
|
|
|
|
|
|
och := &mocks.OutboundConnectionHandler{
|
|
|
|
Data2Send: bytes.NewBuffer(make([]byte, 0, 1024)),
|
|
|
|
Data2Return: []byte("The data to be returned to inbound server."),
|
|
|
|
}
|
|
|
|
|
|
|
|
core.RegisterOutboundConnectionHandlerFactory("mock_och", och)
|
|
|
|
|
|
|
|
configB := mocks.Config{
|
|
|
|
PortValue: portB,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "vmess",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: &VMessInboundConfig{
|
|
|
|
AllowedClients: []VMessUser{
|
|
|
|
VMessUser{Id: "ad937d9d-6e23-4a5a-ba23-bce5092a7c51"},
|
|
|
|
},
|
|
|
|
},
|
2015-09-19 15:56:09 -04:00
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_och",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: nil,
|
2015-09-19 15:56:09 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
pointB, err := core.NewPoint(&configB)
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = pointB.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
2015-09-20 12:22:29 -04:00
|
|
|
dest := v2net.NewTCPDestination(v2net.IPAddress([]byte{1, 2, 3, 4}, 80))
|
2015-10-02 09:32:26 -04:00
|
|
|
ich.Communicate(v2net.NewPacket(dest, nil, true))
|
2015-09-19 15:56:09 -04:00
|
|
|
assert.Bytes([]byte(data2Send)).Equals(och.Data2Send.Bytes())
|
|
|
|
assert.Bytes(ich.DataReturned.Bytes()).Equals(och.Data2Return)
|
|
|
|
}
|
2015-10-03 05:34:01 -04:00
|
|
|
|
|
|
|
func TestVMessInAndOutUDP(t *testing.T) {
|
|
|
|
assert := unit.Assert(t)
|
|
|
|
|
|
|
|
data2Send := "The data to be send to outbound server."
|
|
|
|
|
|
|
|
portA := uint16(17394)
|
|
|
|
ich := &mocks.InboundConnectionHandler{
|
|
|
|
Data2Send: []byte(data2Send),
|
|
|
|
DataReturned: bytes.NewBuffer(make([]byte, 0, 1024)),
|
|
|
|
}
|
|
|
|
|
|
|
|
core.RegisterInboundConnectionHandlerFactory("mock_ich", ich)
|
|
|
|
|
|
|
|
configA := mocks.Config{
|
|
|
|
PortValue: portA,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_ich",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: nil,
|
2015-10-03 05:34:01 -04:00
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "vmess",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: &VMessOutboundConfig{
|
|
|
|
[]VNextConfig{
|
|
|
|
VNextConfig{
|
|
|
|
Address: "127.0.0.1",
|
|
|
|
Port: 13841,
|
|
|
|
Network: "udp",
|
|
|
|
Users: []VMessUser{
|
|
|
|
VMessUser{Id: "ad937d9d-6e23-4a5a-ba23-bce5092a7c51"},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2015-10-03 05:34:01 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
pointA, err := core.NewPoint(&configA)
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = pointA.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
portB := uint16(13841)
|
|
|
|
|
|
|
|
och := &mocks.OutboundConnectionHandler{
|
|
|
|
Data2Send: bytes.NewBuffer(make([]byte, 0, 1024)),
|
|
|
|
Data2Return: []byte("The data to be returned to inbound server."),
|
|
|
|
}
|
|
|
|
|
|
|
|
core.RegisterOutboundConnectionHandlerFactory("mock_och", och)
|
|
|
|
|
|
|
|
configB := mocks.Config{
|
|
|
|
PortValue: portB,
|
|
|
|
InboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "vmess",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: &VMessInboundConfig{
|
|
|
|
AllowedClients: []VMessUser{
|
|
|
|
VMessUser{Id: "ad937d9d-6e23-4a5a-ba23-bce5092a7c51"},
|
|
|
|
},
|
|
|
|
UDPEnabled: true,
|
|
|
|
},
|
2015-10-03 05:34:01 -04:00
|
|
|
},
|
|
|
|
OutboundConfigValue: &mocks.ConnectionConfig{
|
|
|
|
ProtocolValue: "mock_och",
|
2015-10-06 17:11:08 -04:00
|
|
|
SettingsValue: nil,
|
2015-10-03 05:34:01 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
pointB, err := core.NewPoint(&configB)
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
|
|
|
err = pointB.Start()
|
|
|
|
assert.Error(err).IsNil()
|
|
|
|
|
2015-10-08 08:46:18 -04:00
|
|
|
data2SendBuffer := alloc.NewBuffer()
|
|
|
|
data2SendBuffer.Clear()
|
|
|
|
data2SendBuffer.Append([]byte(data2Send))
|
2015-10-03 05:34:01 -04:00
|
|
|
dest := v2net.NewUDPDestination(v2net.IPAddress([]byte{1, 2, 3, 4}, 80))
|
2015-10-08 08:46:18 -04:00
|
|
|
ich.Communicate(v2net.NewPacket(dest, data2SendBuffer, false))
|
2015-10-03 05:34:01 -04:00
|
|
|
assert.Bytes([]byte(data2Send)).Equals(och.Data2Send.Bytes())
|
|
|
|
assert.Bytes(ich.DataReturned.Bytes()).Equals(och.Data2Return)
|
|
|
|
}
|