mirror of https://github.com/v2ray/v2ray-core
130 lines
3.7 KiB
Go
130 lines
3.7 KiB
Go
package vmess_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/v2ray/v2ray-core/app"
|
|
"github.com/v2ray/v2ray-core/app/dispatcher"
|
|
"github.com/v2ray/v2ray-core/app/dns"
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
v2nettesting "github.com/v2ray/v2ray-core/common/net/testing"
|
|
"github.com/v2ray/v2ray-core/common/protocol"
|
|
"github.com/v2ray/v2ray-core/common/uuid"
|
|
"github.com/v2ray/v2ray-core/proxy"
|
|
proxytesting "github.com/v2ray/v2ray-core/proxy/testing"
|
|
proxymocks "github.com/v2ray/v2ray-core/proxy/testing/mocks"
|
|
_ "github.com/v2ray/v2ray-core/proxy/vmess/inbound"
|
|
_ "github.com/v2ray/v2ray-core/proxy/vmess/outbound"
|
|
"github.com/v2ray/v2ray-core/shell/point"
|
|
"github.com/v2ray/v2ray-core/testing/assert"
|
|
)
|
|
|
|
func TestVMessInAndOut(t *testing.T) {
|
|
assert := assert.On(t)
|
|
|
|
id, err := uuid.ParseString("ad937d9d-6e23-4a5a-ba23-bce5092a7c51")
|
|
assert.Error(err).IsNil()
|
|
|
|
testAccount := protocol.NewID(id)
|
|
|
|
portA := v2nettesting.PickPort()
|
|
portB := v2nettesting.PickPort()
|
|
|
|
ichConnInput := []byte("The data to be send to outbound server.")
|
|
ichConnOutput := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
ich := &proxymocks.InboundConnectionHandler{
|
|
ConnInput: bytes.NewReader(ichConnInput),
|
|
ConnOutput: ichConnOutput,
|
|
}
|
|
|
|
protocol, err := proxytesting.RegisterInboundConnectionHandlerCreator("mock_ich",
|
|
func(space app.Space, config interface{}, meta *proxy.InboundHandlerMeta) (proxy.InboundHandler, error) {
|
|
ich.ListeningAddress = meta.Address
|
|
ich.ListeningPort = meta.Port
|
|
ich.PacketDispatcher = space.GetApp(dispatcher.APP_ID).(dispatcher.PacketDispatcher)
|
|
return ich, nil
|
|
})
|
|
assert.Error(err).IsNil()
|
|
|
|
configA := &point.Config{
|
|
Port: portA,
|
|
DNSConfig: &dns.Config{
|
|
NameServers: []v2net.Destination{
|
|
v2net.UDPDestination(v2net.DomainAddress("localhost"), v2net.Port(53)),
|
|
},
|
|
},
|
|
InboundConfig: &point.InboundConnectionConfig{
|
|
Protocol: protocol,
|
|
ListenOn: v2net.LocalHostIP,
|
|
Settings: nil,
|
|
},
|
|
OutboundConfig: &point.OutboundConnectionConfig{
|
|
Protocol: "vmess",
|
|
Settings: []byte(`{
|
|
"vnext": [
|
|
{
|
|
"address": "127.0.0.1",
|
|
"port": ` + portB.String() + `,
|
|
"users": [
|
|
{"id": "` + testAccount.String() + `"}
|
|
]
|
|
}
|
|
]
|
|
}`),
|
|
},
|
|
}
|
|
|
|
pointA, err := point.NewPoint(configA)
|
|
assert.Error(err).IsNil()
|
|
|
|
err = pointA.Start()
|
|
assert.Error(err).IsNil()
|
|
|
|
ochConnInput := []byte("The data to be returned to inbound server.")
|
|
ochConnOutput := bytes.NewBuffer(make([]byte, 0, 1024))
|
|
och := &proxymocks.OutboundConnectionHandler{
|
|
ConnInput: bytes.NewReader(ochConnInput),
|
|
ConnOutput: ochConnOutput,
|
|
}
|
|
|
|
protocol, err = proxytesting.RegisterOutboundConnectionHandlerCreator("mock_och",
|
|
func(space app.Space, config interface{}, meta *proxy.OutboundHandlerMeta) (proxy.OutboundHandler, error) {
|
|
return och, nil
|
|
})
|
|
assert.Error(err).IsNil()
|
|
|
|
configB := &point.Config{
|
|
Port: portB,
|
|
DNSConfig: &dns.Config{
|
|
NameServers: []v2net.Destination{
|
|
v2net.UDPDestination(v2net.DomainAddress("localhost"), v2net.Port(53)),
|
|
},
|
|
},
|
|
InboundConfig: &point.InboundConnectionConfig{
|
|
Protocol: "vmess",
|
|
ListenOn: v2net.LocalHostIP,
|
|
Settings: []byte(`{
|
|
"clients": [
|
|
{"id": "` + testAccount.String() + `"}
|
|
]
|
|
}`),
|
|
},
|
|
OutboundConfig: &point.OutboundConnectionConfig{
|
|
Protocol: protocol,
|
|
Settings: nil,
|
|
},
|
|
}
|
|
|
|
pointB, err := point.NewPoint(configB)
|
|
assert.Error(err).IsNil()
|
|
|
|
err = pointB.Start()
|
|
assert.Error(err).IsNil()
|
|
|
|
dest := v2net.TCPDestination(v2net.IPAddress([]byte{1, 2, 3, 4}), 80)
|
|
ich.Communicate(dest)
|
|
assert.Bytes(ichConnInput).Equals(ochConnOutput.Bytes())
|
|
assert.Bytes(ichConnOutput.Bytes()).Equals(ochConnInput)
|
|
}
|