2019-01-05 20:43:22 +00:00
|
|
|
package core_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"crypto/rand"
|
|
|
|
"io"
|
|
|
|
"testing"
|
2019-01-05 23:34:38 +00:00
|
|
|
"time"
|
2019-01-05 20:43:22 +00:00
|
|
|
|
|
|
|
"github.com/google/go-cmp/cmp"
|
2020-12-20 02:50:46 +00:00
|
|
|
"google.golang.org/protobuf/proto"
|
2019-01-05 20:43:22 +00:00
|
|
|
|
2021-02-16 20:31:50 +00:00
|
|
|
core "github.com/v2fly/v2ray-core/v4"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/app/dispatcher"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/app/proxyman"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/common"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/common/net"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/common/serial"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/proxy/freedom"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/testing/servers/tcp"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/testing/servers/udp"
|
2019-01-05 20:43:22 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func xor(b []byte) []byte {
|
|
|
|
r := make([]byte, len(b))
|
|
|
|
for i, v := range b {
|
|
|
|
r[i] = v ^ 'c'
|
|
|
|
}
|
|
|
|
return r
|
|
|
|
}
|
|
|
|
|
|
|
|
func xor2(b []byte) []byte {
|
|
|
|
r := make([]byte, len(b))
|
|
|
|
for i, v := range b {
|
|
|
|
r[i] = v ^ 'd'
|
|
|
|
}
|
|
|
|
return r
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestV2RayDial(t *testing.T) {
|
|
|
|
tcpServer := tcp.Server{
|
|
|
|
MsgProcessor: xor,
|
|
|
|
}
|
|
|
|
dest, err := tcpServer.Start()
|
|
|
|
common.Must(err)
|
|
|
|
defer tcpServer.Close()
|
|
|
|
|
|
|
|
config := &core.Config{
|
|
|
|
App: []*serial.TypedMessage{
|
|
|
|
serial.ToTypedMessage(&dispatcher.Config{}),
|
|
|
|
serial.ToTypedMessage(&proxyman.InboundConfig{}),
|
|
|
|
serial.ToTypedMessage(&proxyman.OutboundConfig{}),
|
|
|
|
},
|
|
|
|
Outbound: []*core.OutboundHandlerConfig{
|
|
|
|
{
|
|
|
|
ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
cfgBytes, err := proto.Marshal(config)
|
|
|
|
common.Must(err)
|
|
|
|
|
improve commands (#648)
* stat show json refactor
> will show {} since scripts should expect it, its the json style of blank value
* combine statistics commands to one
* code optimize
* fix runtime flag
* remove json indent
* change overridden to override
* api bi -json
* convert stdin support
code optimize
* writeRow() code optimize
add stats tittle
revert back to restartlogger
* api log -restart
* follow log
* codeql
* move -json to shared flags
* flags optimize
* update flag descriptions
* change "-v" of "api bo" to duration
* change "-expire" of "tls cert" to days
* cmds short description optimize
* fix multiple log followers
* Format loader refactor
* "infra/conf/merge" refactor
* "LoadConfig" refactor
* add "infra/conf/mergers"
* contribute to it will benifit `v2ray run`,`v2ray test`,`v2ray convert`
* easily add new formats, by just adding a converter like json.FromTOML
* default format auto, to all cmds above
* auto detect input format
* mixed formats support
* better stdin behavior
* don't wait if no content
* don't use 'stdin:' placeholder
* `v2ray test` now behaves exactly the same with `v2ray run`, including stdin reading
* api ado, adi, rmo, rmi refactor
* support folders to files resolving, mixed formats
* remove remaining 'stdin:' placeholders
* fix tests
* os.Stdin.Stat() behaves different in platforms, removed
* code optimize
Co-authored-by: loyalsoldier <10487845+Loyalsoldier@users.noreply.github.com>
2021-02-21 15:02:42 +00:00
|
|
|
server, err := core.StartInstance(core.FormatProtobuf, cfgBytes)
|
2019-01-05 20:43:22 +00:00
|
|
|
common.Must(err)
|
|
|
|
defer server.Close()
|
|
|
|
|
|
|
|
conn, err := core.Dial(context.Background(), server, dest)
|
|
|
|
common.Must(err)
|
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
const size = 10240 * 1024
|
|
|
|
payload := make([]byte, size)
|
|
|
|
common.Must2(rand.Read(payload))
|
|
|
|
|
|
|
|
if _, err := conn.Write(payload); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
receive := make([]byte, size)
|
|
|
|
if _, err := io.ReadFull(conn, receive); err != nil {
|
|
|
|
t.Fatal("failed to read all response: ", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if r := cmp.Diff(xor(receive), payload); r != "" {
|
|
|
|
t.Error(r)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-01-05 23:34:38 +00:00
|
|
|
func TestV2RayDialUDPConn(t *testing.T) {
|
|
|
|
udpServer := udp.Server{
|
|
|
|
MsgProcessor: xor,
|
|
|
|
}
|
|
|
|
dest, err := udpServer.Start()
|
|
|
|
common.Must(err)
|
|
|
|
defer udpServer.Close()
|
|
|
|
|
|
|
|
config := &core.Config{
|
|
|
|
App: []*serial.TypedMessage{
|
|
|
|
serial.ToTypedMessage(&dispatcher.Config{}),
|
|
|
|
serial.ToTypedMessage(&proxyman.InboundConfig{}),
|
|
|
|
serial.ToTypedMessage(&proxyman.OutboundConfig{}),
|
|
|
|
},
|
|
|
|
Outbound: []*core.OutboundHandlerConfig{
|
|
|
|
{
|
|
|
|
ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
cfgBytes, err := proto.Marshal(config)
|
|
|
|
common.Must(err)
|
|
|
|
|
improve commands (#648)
* stat show json refactor
> will show {} since scripts should expect it, its the json style of blank value
* combine statistics commands to one
* code optimize
* fix runtime flag
* remove json indent
* change overridden to override
* api bi -json
* convert stdin support
code optimize
* writeRow() code optimize
add stats tittle
revert back to restartlogger
* api log -restart
* follow log
* codeql
* move -json to shared flags
* flags optimize
* update flag descriptions
* change "-v" of "api bo" to duration
* change "-expire" of "tls cert" to days
* cmds short description optimize
* fix multiple log followers
* Format loader refactor
* "infra/conf/merge" refactor
* "LoadConfig" refactor
* add "infra/conf/mergers"
* contribute to it will benifit `v2ray run`,`v2ray test`,`v2ray convert`
* easily add new formats, by just adding a converter like json.FromTOML
* default format auto, to all cmds above
* auto detect input format
* mixed formats support
* better stdin behavior
* don't wait if no content
* don't use 'stdin:' placeholder
* `v2ray test` now behaves exactly the same with `v2ray run`, including stdin reading
* api ado, adi, rmo, rmi refactor
* support folders to files resolving, mixed formats
* remove remaining 'stdin:' placeholders
* fix tests
* os.Stdin.Stat() behaves different in platforms, removed
* code optimize
Co-authored-by: loyalsoldier <10487845+Loyalsoldier@users.noreply.github.com>
2021-02-21 15:02:42 +00:00
|
|
|
server, err := core.StartInstance(core.FormatProtobuf, cfgBytes)
|
2019-01-05 23:34:38 +00:00
|
|
|
common.Must(err)
|
|
|
|
defer server.Close()
|
|
|
|
|
|
|
|
conn, err := core.Dial(context.Background(), server, dest)
|
|
|
|
common.Must(err)
|
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
const size = 1024
|
|
|
|
payload := make([]byte, size)
|
|
|
|
common.Must2(rand.Read(payload))
|
|
|
|
|
|
|
|
for i := 0; i < 2; i++ {
|
|
|
|
if _, err := conn.Write(payload); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
time.Sleep(time.Millisecond * 500)
|
|
|
|
|
|
|
|
receive := make([]byte, size*2)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
|
|
n, err := conn.Read(receive)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal("expect no error, but got ", err)
|
|
|
|
}
|
|
|
|
if n != size {
|
|
|
|
t.Fatal("expect read size ", size, " but got ", n)
|
|
|
|
}
|
|
|
|
|
|
|
|
if r := cmp.Diff(xor(receive[:n]), payload); r != "" {
|
|
|
|
t.Fatal(r)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-01-05 20:43:22 +00:00
|
|
|
func TestV2RayDialUDP(t *testing.T) {
|
|
|
|
udpServer1 := udp.Server{
|
|
|
|
MsgProcessor: xor,
|
|
|
|
}
|
|
|
|
dest1, err := udpServer1.Start()
|
|
|
|
common.Must(err)
|
|
|
|
defer udpServer1.Close()
|
|
|
|
|
|
|
|
udpServer2 := udp.Server{
|
|
|
|
MsgProcessor: xor2,
|
|
|
|
}
|
|
|
|
dest2, err := udpServer2.Start()
|
|
|
|
common.Must(err)
|
|
|
|
defer udpServer2.Close()
|
|
|
|
|
|
|
|
config := &core.Config{
|
|
|
|
App: []*serial.TypedMessage{
|
|
|
|
serial.ToTypedMessage(&dispatcher.Config{}),
|
|
|
|
serial.ToTypedMessage(&proxyman.InboundConfig{}),
|
|
|
|
serial.ToTypedMessage(&proxyman.OutboundConfig{}),
|
|
|
|
},
|
|
|
|
Outbound: []*core.OutboundHandlerConfig{
|
|
|
|
{
|
|
|
|
ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
cfgBytes, err := proto.Marshal(config)
|
|
|
|
common.Must(err)
|
|
|
|
|
improve commands (#648)
* stat show json refactor
> will show {} since scripts should expect it, its the json style of blank value
* combine statistics commands to one
* code optimize
* fix runtime flag
* remove json indent
* change overridden to override
* api bi -json
* convert stdin support
code optimize
* writeRow() code optimize
add stats tittle
revert back to restartlogger
* api log -restart
* follow log
* codeql
* move -json to shared flags
* flags optimize
* update flag descriptions
* change "-v" of "api bo" to duration
* change "-expire" of "tls cert" to days
* cmds short description optimize
* fix multiple log followers
* Format loader refactor
* "infra/conf/merge" refactor
* "LoadConfig" refactor
* add "infra/conf/mergers"
* contribute to it will benifit `v2ray run`,`v2ray test`,`v2ray convert`
* easily add new formats, by just adding a converter like json.FromTOML
* default format auto, to all cmds above
* auto detect input format
* mixed formats support
* better stdin behavior
* don't wait if no content
* don't use 'stdin:' placeholder
* `v2ray test` now behaves exactly the same with `v2ray run`, including stdin reading
* api ado, adi, rmo, rmi refactor
* support folders to files resolving, mixed formats
* remove remaining 'stdin:' placeholders
* fix tests
* os.Stdin.Stat() behaves different in platforms, removed
* code optimize
Co-authored-by: loyalsoldier <10487845+Loyalsoldier@users.noreply.github.com>
2021-02-21 15:02:42 +00:00
|
|
|
server, err := core.StartInstance(core.FormatProtobuf, cfgBytes)
|
2019-01-05 20:43:22 +00:00
|
|
|
common.Must(err)
|
|
|
|
defer server.Close()
|
|
|
|
|
|
|
|
conn, err := core.DialUDP(context.Background(), server)
|
|
|
|
common.Must(err)
|
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
const size = 1024
|
|
|
|
{
|
|
|
|
payload := make([]byte, size)
|
|
|
|
common.Must2(rand.Read(payload))
|
|
|
|
|
|
|
|
if _, err := conn.WriteTo(payload, &net.UDPAddr{
|
|
|
|
IP: dest1.Address.IP(),
|
|
|
|
Port: int(dest1.Port),
|
|
|
|
}); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
receive := make([]byte, size)
|
|
|
|
if _, _, err := conn.ReadFrom(receive); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if r := cmp.Diff(xor(receive), payload); r != "" {
|
|
|
|
t.Error(r)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
{
|
|
|
|
payload := make([]byte, size)
|
|
|
|
common.Must2(rand.Read(payload))
|
|
|
|
|
|
|
|
if _, err := conn.WriteTo(payload, &net.UDPAddr{
|
|
|
|
IP: dest2.Address.IP(),
|
|
|
|
Port: int(dest2.Port),
|
|
|
|
}); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
receive := make([]byte, size)
|
|
|
|
if _, _, err := conn.ReadFrom(receive); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if r := cmp.Diff(xor2(receive), payload); r != "" {
|
|
|
|
t.Error(r)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|