mirror of
https://github.com/XTLS/Xray-core.git
synced 2026-10-03 12:27:58 +03:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
bd60b01576 | ||
|
|
f49d8b89c8 |
@@ -67,7 +67,9 @@ jobs:
|
|||||||
check-latest: true
|
check-latest: true
|
||||||
cache: false
|
cache: false
|
||||||
- name: Check Format
|
- name: Check Format
|
||||||
run: go run ./infra/vformat/main.go -mode check -pwd ./
|
run: |
|
||||||
|
go install -v mvdan.cc/gofumpt@latest
|
||||||
|
go run ./infra/vformat/main.go -mode check -pwd ./
|
||||||
|
|
||||||
test:
|
test:
|
||||||
needs: check-assets
|
needs: check-assets
|
||||||
|
|||||||
@@ -470,9 +470,6 @@ func (d *DefaultDispatcher) routedDispatch(ctx context.Context, link *transport.
|
|||||||
return // DO NOT CHANGE: the traffic shouldn't be processed by default outbound if the specified outbound tag doesn't exist (yet), e.g., VLESS Reverse Proxy
|
return // DO NOT CHANGE: the traffic shouldn't be processed by default outbound if the specified outbound tag doesn't exist (yet), e.g., VLESS Reverse Proxy
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if err != common.ErrNoClue {
|
|
||||||
errors.LogErrorInner(ctx, err, "failed to pick route for ", destination)
|
|
||||||
}
|
|
||||||
errors.LogInfo(ctx, "default route for ", destination)
|
errors.LogInfo(ctx, "default route for ", destination)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+6
-25
@@ -93,7 +93,6 @@ type NameServer struct {
|
|||||||
UnexpectedIp []*geodata.IPRule `protobuf:"bytes,13,rep,name=unexpected_ip,json=unexpectedIp,proto3" json:"unexpected_ip,omitempty"`
|
UnexpectedIp []*geodata.IPRule `protobuf:"bytes,13,rep,name=unexpected_ip,json=unexpectedIp,proto3" json:"unexpected_ip,omitempty"`
|
||||||
ActUnprior bool `protobuf:"varint,14,opt,name=actUnprior,proto3" json:"actUnprior,omitempty"`
|
ActUnprior bool `protobuf:"varint,14,opt,name=actUnprior,proto3" json:"actUnprior,omitempty"`
|
||||||
PolicyID uint32 `protobuf:"varint,17,opt,name=policyID,proto3" json:"policyID,omitempty"`
|
PolicyID uint32 `protobuf:"varint,17,opt,name=policyID,proto3" json:"policyID,omitempty"`
|
||||||
Id string `protobuf:"bytes,18,opt,name=id,proto3" json:"id,omitempty"`
|
|
||||||
unknownFields protoimpl.UnknownFields
|
unknownFields protoimpl.UnknownFields
|
||||||
sizeCache protoimpl.SizeCache
|
sizeCache protoimpl.SizeCache
|
||||||
}
|
}
|
||||||
@@ -240,13 +239,6 @@ func (x *NameServer) GetPolicyID() uint32 {
|
|||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *NameServer) GetId() string {
|
|
||||||
if x != nil {
|
|
||||||
return x.Id
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
// NameServer list used by this DNS client.
|
// NameServer list used by this DNS client.
|
||||||
@@ -266,10 +258,8 @@ type Config struct {
|
|||||||
DisableFallback bool `protobuf:"varint,10,opt,name=disableFallback,proto3" json:"disableFallback,omitempty"`
|
DisableFallback bool `protobuf:"varint,10,opt,name=disableFallback,proto3" json:"disableFallback,omitempty"`
|
||||||
DisableFallbackIfMatch bool `protobuf:"varint,11,opt,name=disableFallbackIfMatch,proto3" json:"disableFallbackIfMatch,omitempty"`
|
DisableFallbackIfMatch bool `protobuf:"varint,11,opt,name=disableFallbackIfMatch,proto3" json:"disableFallbackIfMatch,omitempty"`
|
||||||
EnableParallelQuery bool `protobuf:"varint,14,opt,name=enableParallelQuery,proto3" json:"enableParallelQuery,omitempty"`
|
EnableParallelQuery bool `protobuf:"varint,14,opt,name=enableParallelQuery,proto3" json:"enableParallelQuery,omitempty"`
|
||||||
// Absolute path to the Lua DNS query script.
|
unknownFields protoimpl.UnknownFields
|
||||||
Script string `protobuf:"bytes,15,opt,name=script,proto3" json:"script,omitempty"`
|
sizeCache protoimpl.SizeCache
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *Config) Reset() {
|
func (x *Config) Reset() {
|
||||||
@@ -379,13 +369,6 @@ func (x *Config) GetEnableParallelQuery() bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *Config) GetScript() string {
|
|
||||||
if x != nil {
|
|
||||||
return x.Script
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
type Config_HostMapping struct {
|
type Config_HostMapping struct {
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
Domain *geodata.DomainRule `protobuf:"bytes,2,opt,name=domain,proto3" json:"domain,omitempty"`
|
Domain *geodata.DomainRule `protobuf:"bytes,2,opt,name=domain,proto3" json:"domain,omitempty"`
|
||||||
@@ -452,7 +435,7 @@ var File_app_dns_config_proto protoreflect.FileDescriptor
|
|||||||
|
|
||||||
const file_app_dns_config_proto_rawDesc = "" +
|
const file_app_dns_config_proto_rawDesc = "" +
|
||||||
"\n" +
|
"\n" +
|
||||||
"\x14app/dns/config.proto\x12\fxray.app.dns\x1a\x1ccommon/net/destination.proto\x1a\x1bcommon/geodata/geodat.proto\"\xee\x05\n" +
|
"\x14app/dns/config.proto\x12\fxray.app.dns\x1a\x1ccommon/net/destination.proto\x1a\x1bcommon/geodata/geodat.proto\"\xde\x05\n" +
|
||||||
"\n" +
|
"\n" +
|
||||||
"NameServer\x123\n" +
|
"NameServer\x123\n" +
|
||||||
"\aaddress\x18\x01 \x01(\v2\x19.xray.common.net.EndpointR\aaddress\x12\x1b\n" +
|
"\aaddress\x18\x01 \x01(\v2\x19.xray.common.net.EndpointR\aaddress\x12\x1b\n" +
|
||||||
@@ -478,11 +461,10 @@ const file_app_dns_config_proto_rawDesc = "" +
|
|||||||
"\n" +
|
"\n" +
|
||||||
"actUnprior\x18\x0e \x01(\bR\n" +
|
"actUnprior\x18\x0e \x01(\bR\n" +
|
||||||
"actUnprior\x12\x1a\n" +
|
"actUnprior\x12\x1a\n" +
|
||||||
"\bpolicyID\x18\x11 \x01(\rR\bpolicyID\x12\x0e\n" +
|
"\bpolicyID\x18\x11 \x01(\rR\bpolicyIDB\x0f\n" +
|
||||||
"\x02id\x18\x12 \x01(\tR\x02idB\x0f\n" +
|
|
||||||
"\r_disableCacheB\r\n" +
|
"\r_disableCacheB\r\n" +
|
||||||
"\v_serveStaleB\x12\n" +
|
"\v_serveStaleB\x12\n" +
|
||||||
"\x10_serveExpiredTTLJ\x04\b\x04\x10\x05\"\x9a\x05\n" +
|
"\x10_serveExpiredTTLJ\x04\b\x04\x10\x05\"\x82\x05\n" +
|
||||||
"\x06Config\x129\n" +
|
"\x06Config\x129\n" +
|
||||||
"\vname_server\x18\x05 \x03(\v2\x18.xray.app.dns.NameServerR\n" +
|
"\vname_server\x18\x05 \x03(\v2\x18.xray.app.dns.NameServerR\n" +
|
||||||
"nameServer\x12\x1b\n" +
|
"nameServer\x12\x1b\n" +
|
||||||
@@ -498,8 +480,7 @@ const file_app_dns_config_proto_rawDesc = "" +
|
|||||||
"\x0fdisableFallback\x18\n" +
|
"\x0fdisableFallback\x18\n" +
|
||||||
" \x01(\bR\x0fdisableFallback\x126\n" +
|
" \x01(\bR\x0fdisableFallback\x126\n" +
|
||||||
"\x16disableFallbackIfMatch\x18\v \x01(\bR\x16disableFallbackIfMatch\x120\n" +
|
"\x16disableFallbackIfMatch\x18\v \x01(\bR\x16disableFallbackIfMatch\x120\n" +
|
||||||
"\x13enableParallelQuery\x18\x0e \x01(\bR\x13enableParallelQuery\x12\x16\n" +
|
"\x13enableParallelQuery\x18\x0e \x01(\bR\x13enableParallelQuery\x1a}\n" +
|
||||||
"\x06script\x18\x0f \x01(\tR\x06script\x1a}\n" +
|
|
||||||
"\vHostMapping\x127\n" +
|
"\vHostMapping\x127\n" +
|
||||||
"\x06domain\x18\x02 \x01(\v2\x1f.xray.common.geodata.DomainRuleR\x06domain\x12\x0e\n" +
|
"\x06domain\x18\x02 \x01(\v2\x1f.xray.common.geodata.DomainRuleR\x06domain\x12\x0e\n" +
|
||||||
"\x02ip\x18\x03 \x03(\fR\x02ip\x12%\n" +
|
"\x02ip\x18\x03 \x03(\fR\x02ip\x12%\n" +
|
||||||
|
|||||||
@@ -27,7 +27,6 @@ message NameServer {
|
|||||||
repeated xray.common.geodata.IPRule unexpected_ip = 13;
|
repeated xray.common.geodata.IPRule unexpected_ip = 13;
|
||||||
bool actUnprior = 14;
|
bool actUnprior = 14;
|
||||||
uint32 policyID = 17;
|
uint32 policyID = 17;
|
||||||
string id = 18;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
enum QueryStrategy {
|
enum QueryStrategy {
|
||||||
@@ -74,7 +73,4 @@ message Config {
|
|||||||
bool disableFallbackIfMatch = 11;
|
bool disableFallbackIfMatch = 11;
|
||||||
|
|
||||||
bool enableParallelQuery = 14;
|
bool enableParallelQuery = 14;
|
||||||
|
|
||||||
// Absolute path to the Lua DNS query script.
|
|
||||||
string script = 15;
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -31,8 +31,6 @@ type DNS struct {
|
|||||||
domainMatcher geodata.DomainMatcher
|
domainMatcher geodata.DomainMatcher
|
||||||
matcherInfos []*DomainMatcherInfo
|
matcherInfos []*DomainMatcherInfo
|
||||||
checkSystem bool
|
checkSystem bool
|
||||||
script *scriptEngine
|
|
||||||
scriptPath string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// DomainMatcherInfo contains information attached to index returned by Server.domainMatcher.
|
// DomainMatcherInfo contains information attached to index returned by Server.domainMatcher.
|
||||||
@@ -182,7 +180,6 @@ func New(ctx context.Context, config *Config) (*DNS, error) {
|
|||||||
disableFallbackIfMatch: config.DisableFallbackIfMatch,
|
disableFallbackIfMatch: config.DisableFallbackIfMatch,
|
||||||
enableParallelQuery: config.EnableParallelQuery,
|
enableParallelQuery: config.EnableParallelQuery,
|
||||||
checkSystem: checkSystem,
|
checkSystem: checkSystem,
|
||||||
scriptPath: config.Script,
|
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -193,21 +190,11 @@ func (*DNS) Type() interface{} {
|
|||||||
|
|
||||||
// Start implements common.Runnable.
|
// Start implements common.Runnable.
|
||||||
func (s *DNS) Start() error {
|
func (s *DNS) Start() error {
|
||||||
if s.scriptPath != "" {
|
|
||||||
engine, err := newScriptEngine(s.scriptPath, s)
|
|
||||||
if err != nil {
|
|
||||||
return errors.New("failed to initialize DNS script").Base(err)
|
|
||||||
}
|
|
||||||
s.script = engine
|
|
||||||
}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Close implements common.Closable.
|
// Close implements common.Closable.
|
||||||
func (s *DNS) Close() error {
|
func (s *DNS) Close() error {
|
||||||
if s.script != nil {
|
|
||||||
s.script.close()
|
|
||||||
}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -270,9 +257,6 @@ func (s *DNS) LookupIP(domain string, option dns.IPOption) ([]net.IP, uint32, er
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Name servers lookup
|
// Name servers lookup
|
||||||
if s.script != nil {
|
|
||||||
return s.script.query(domain, option)
|
|
||||||
}
|
|
||||||
if s.enableParallelQuery {
|
if s.enableParallelQuery {
|
||||||
return s.parallelQuery(domain, option)
|
return s.parallelQuery(domain, option)
|
||||||
} else {
|
} else {
|
||||||
|
|||||||
-201
@@ -1,201 +0,0 @@
|
|||||||
package dns
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"math"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
featureDNS "github.com/xtls/xray-core/features/dns"
|
|
||||||
"github.com/xtls/xray-core/features/dns/localdns"
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
// luaDNSServer adapts configured and local DNS to the same Lua API.
|
|
||||||
type luaDNSServer struct {
|
|
||||||
id string
|
|
||||||
name string
|
|
||||||
query func(context.Context, string, featureDNS.IPOption) ([]net.IP, uint32, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
// RegisterLua makes xray.dns available to scripts backed by client.
|
|
||||||
func RegisterLua(L *lua.LState, client featureDNS.Client) {
|
|
||||||
var servers []luaDNSServer
|
|
||||||
switch client := client.(type) {
|
|
||||||
case *DNS:
|
|
||||||
servers = luaServers(client)
|
|
||||||
case *localdns.Client:
|
|
||||||
servers = []luaDNSServer{{
|
|
||||||
id: "localhost",
|
|
||||||
name: "localhost",
|
|
||||||
query: func(_ context.Context, domain string, option featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
return client.LookupIP(domain, option)
|
|
||||||
},
|
|
||||||
}}
|
|
||||||
}
|
|
||||||
registerLua(L, servers, client)
|
|
||||||
}
|
|
||||||
|
|
||||||
// RegisterLua makes xray.dns available to DNS scripts.
|
|
||||||
func (s *DNS) RegisterLua(L *lua.LState) {
|
|
||||||
registerLua(L, luaServers(s), nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
func luaServers(s *DNS) []luaDNSServer {
|
|
||||||
servers := make([]luaDNSServer, len(s.clients))
|
|
||||||
for i, client := range s.clients {
|
|
||||||
servers[i] = luaDNSServer{id: client.id, name: client.Name(), query: client.QueryIP}
|
|
||||||
}
|
|
||||||
return servers
|
|
||||||
}
|
|
||||||
|
|
||||||
func registerLua(L *lua.LState, servers []luaDNSServer, client featureDNS.Client) {
|
|
||||||
L.PreloadModule("xray.dns", func(L *lua.LState) int {
|
|
||||||
serverList := L.NewTable()
|
|
||||||
for i, client := range servers {
|
|
||||||
server := L.NewTable()
|
|
||||||
|
|
||||||
server.RawSetString("ID", lua.LString(client.id))
|
|
||||||
|
|
||||||
server.RawSetString("Query", L.NewFunction(func(L *lua.LState) int {
|
|
||||||
domain, ok := L.Get(2).(lua.LString)
|
|
||||||
if !ok {
|
|
||||||
L.RaiseError("server:Query requires a domain")
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
option := featureDNS.IPOption{
|
|
||||||
IPv4Enable: L.CheckBool(3),
|
|
||||||
IPv6Enable: L.CheckBool(4),
|
|
||||||
FakeEnable: L.CheckBool(5),
|
|
||||||
}
|
|
||||||
ctx := L.Context()
|
|
||||||
if ctx == nil {
|
|
||||||
L.RaiseError("server:Query requires an active DNS query")
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
var ips []net.IP
|
|
||||||
var ttl uint32
|
|
||||||
var err error
|
|
||||||
if !option.FakeEnable && strings.EqualFold(client.name, "FakeDNS") {
|
|
||||||
err = featureDNS.ErrEmptyResponse
|
|
||||||
} else {
|
|
||||||
ips, ttl, err = client.query(ctx, string(domain), option)
|
|
||||||
}
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = ips
|
|
||||||
L.Push(addresses)
|
|
||||||
L.Push(lua.LNumber(ttl))
|
|
||||||
if err != nil {
|
|
||||||
ud := L.NewUserData()
|
|
||||||
ud.Value = err
|
|
||||||
L.Push(ud)
|
|
||||||
} else {
|
|
||||||
L.Push(lua.LNil)
|
|
||||||
}
|
|
||||||
return 3
|
|
||||||
}))
|
|
||||||
serverList.RawSetInt(i+1, server)
|
|
||||||
}
|
|
||||||
|
|
||||||
module := L.NewTable()
|
|
||||||
if servers != nil {
|
|
||||||
module.RawSetString("Servers", serverList)
|
|
||||||
}
|
|
||||||
if client != nil {
|
|
||||||
module.RawSetString("Query", newLuaClientQuery(L, client))
|
|
||||||
}
|
|
||||||
L.Push(module)
|
|
||||||
return 1
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func newLuaClientQuery(L *lua.LState, client featureDNS.Client) *lua.LFunction {
|
|
||||||
return L.NewFunction(func(L *lua.LState) int {
|
|
||||||
domain, ok := L.Get(1).(lua.LString)
|
|
||||||
if !ok {
|
|
||||||
L.RaiseError("dns.Query requires a domain")
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
option := featureDNS.IPOption{
|
|
||||||
IPv4Enable: L.CheckBool(2),
|
|
||||||
IPv6Enable: L.CheckBool(3),
|
|
||||||
FakeEnable: L.CheckBool(4),
|
|
||||||
}
|
|
||||||
if L.Context() == nil {
|
|
||||||
L.RaiseError("dns.Query requires an active DNS query")
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
ips, ttl, err := client.LookupIP(string(domain), option)
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = ips
|
|
||||||
L.Push(addresses)
|
|
||||||
L.Push(lua.LNumber(ttl))
|
|
||||||
if err != nil {
|
|
||||||
ud := L.NewUserData()
|
|
||||||
ud.Value = err
|
|
||||||
L.Push(ud)
|
|
||||||
} else {
|
|
||||||
L.Push(lua.LNil)
|
|
||||||
}
|
|
||||||
return 3
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// CallLuaHook invokes HandleDNSQuery in the supplied state.
|
|
||||||
// Returned slices and IP bytes may share storage with DNS caches or matcher inputs.
|
|
||||||
func (s *DNS) CallLuaHook(L *lua.LState, ctx context.Context, domain string, option featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
previous, top := L.Context(), L.GetTop()
|
|
||||||
L.SetContext(ctx)
|
|
||||||
defer func() {
|
|
||||||
L.SetTop(top)
|
|
||||||
if previous == nil {
|
|
||||||
L.RemoveContext()
|
|
||||||
} else {
|
|
||||||
L.SetContext(previous)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
fn := L.GetGlobal("HandleDNSQuery")
|
|
||||||
if fn.Type() != lua.LTFunction {
|
|
||||||
return nil, 0, errors.New("DNS script must define HandleDNSQuery(domain, ipv4, ipv6, fake)")
|
|
||||||
}
|
|
||||||
if err := L.CallByParam(lua.P{Fn: fn, NRet: 3, Protect: true},
|
|
||||||
lua.LString(strings.ToLower(domain)), lua.LBool(option.IPv4Enable),
|
|
||||||
lua.LBool(option.IPv6Enable), lua.LBool(option.FakeEnable)); err != nil {
|
|
||||||
return nil, 0, err
|
|
||||||
}
|
|
||||||
return readLuaDNSResult(L.Get(-3), L.Get(-2), L.Get(-1))
|
|
||||||
}
|
|
||||||
|
|
||||||
func readLuaDNSResult(addresses, ttlValue, errorValue lua.LValue) ([]net.IP, uint32, error) {
|
|
||||||
if errorValue != lua.LNil {
|
|
||||||
if ud, ok := errorValue.(*lua.LUserData); ok {
|
|
||||||
if err, ok := ud.Value.(error); ok {
|
|
||||||
return nil, 0, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if s, ok := errorValue.(lua.LString); ok {
|
|
||||||
return nil, 0, errors.New(string(s))
|
|
||||||
}
|
|
||||||
return nil, 0, errors.New("DNS script error must be an error or string")
|
|
||||||
}
|
|
||||||
ttl, ok := ttlValue.(lua.LNumber)
|
|
||||||
if !ok || ttl < 0 || ttl > math.MaxUint32 || math.Trunc(float64(ttl)) != float64(ttl) {
|
|
||||||
return nil, 0, errors.New("DNS script returned invalid TTL")
|
|
||||||
}
|
|
||||||
if addresses == lua.LNil {
|
|
||||||
return nil, 0, featureDNS.ErrEmptyResponse
|
|
||||||
}
|
|
||||||
ud, ok := addresses.(*lua.LUserData)
|
|
||||||
if !ok {
|
|
||||||
return nil, 0, errors.New("DNS script IPs must be native IP slice userdata")
|
|
||||||
}
|
|
||||||
ips, ok := ud.Value.([]net.IP)
|
|
||||||
if !ok {
|
|
||||||
return nil, 0, errors.New("DNS script IPs must be native IP slice userdata")
|
|
||||||
}
|
|
||||||
if len(ips) == 0 {
|
|
||||||
return nil, 0, featureDNS.ErrEmptyResponse
|
|
||||||
}
|
|
||||||
return ips, uint32(ttl), nil
|
|
||||||
}
|
|
||||||
@@ -1,296 +0,0 @@
|
|||||||
package dns
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
go_errors "errors"
|
|
||||||
"math"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/geodata"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
featureDNS "github.com/xtls/xray-core/features/dns"
|
|
||||||
"github.com/xtls/xray-core/features/dns/localdns"
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestReadLuaDNSResult(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
want := []net.IP{net.ParseIP("8.8.8.8"), {127, 0, 0, 1}, net.ParseIP("::1")}
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = want
|
|
||||||
ips, ttl, err := readLuaDNSResult(addresses, lua.LNumber(45), lua.LNil)
|
|
||||||
if err != nil || ttl != 45 || len(ips) != len(want) {
|
|
||||||
t.Fatalf("readLuaDNSResult() = %v, TTL %d, %v", ips, ttl, err)
|
|
||||||
}
|
|
||||||
for i := range want {
|
|
||||||
if !ips[i].Equal(want[i]) {
|
|
||||||
t.Fatalf("IP %d = %v, want %v", i, ips[i], want[i])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadLuaDNSResultValidation(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name string
|
|
||||||
change func(*[3]lua.LValue)
|
|
||||||
want string
|
|
||||||
}{
|
|
||||||
{"fractional TTL", func(v *[3]lua.LValue) { v[1] = lua.LNumber(1.5) }, "invalid TTL"},
|
|
||||||
{"oversized TTL", func(v *[3]lua.LValue) { v[1] = lua.LNumber(4294967296) }, "invalid TTL"},
|
|
||||||
{"negative TTL", func(v *[3]lua.LValue) { v[1] = lua.LNumber(-1) }, "invalid TTL"},
|
|
||||||
{"NaN TTL", func(v *[3]lua.LValue) { v[1] = lua.LNumber(math.NaN()) }, "invalid TTL"},
|
|
||||||
{"missing TTL", func(v *[3]lua.LValue) { v[1] = lua.LNil }, "invalid TTL"},
|
|
||||||
{"string IPs", func(v *[3]lua.LValue) { v[0] = lua.LString("127.0.0.1") }, "native IP slice"},
|
|
||||||
{"wrong userdata", func(v *[3]lua.LValue) { v[0].(*lua.LUserData).Value = net.ParseIP("127.0.0.1") }, "native IP slice"},
|
|
||||||
{"script error", func(v *[3]lua.LValue) { v[2] = lua.LString("blocked by script") }, "blocked by script"},
|
|
||||||
{"invalid error", func(v *[3]lua.LValue) { v[2] = lua.LTrue }, "error or string"},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = []net.IP{net.ParseIP("127.0.0.1")}
|
|
||||||
values := [3]lua.LValue{addresses, lua.LNumber(60), lua.LNil}
|
|
||||||
tc.change(&values)
|
|
||||||
_, _, err := readLuaDNSResult(values[0], values[1], values[2])
|
|
||||||
if err == nil || !strings.Contains(err.Error(), tc.want) {
|
|
||||||
t.Fatalf("readLuaDNSResult error = %v, want %q", err, tc.want)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = []net.IP(nil)
|
|
||||||
for _, empty := range []lua.LValue{addresses, lua.LNil} {
|
|
||||||
if _, _, err := readLuaDNSResult(empty, lua.LNumber(0), lua.LNil); !go_errors.Is(err, featureDNS.ErrEmptyResponse) {
|
|
||||||
t.Fatalf("empty result error = %v, want ErrEmptyResponse", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
wantErr := go_errors.New("upstream failed")
|
|
||||||
errorValue := L.NewUserData()
|
|
||||||
errorValue.Value = wantErr
|
|
||||||
if _, _, err := readLuaDNSResult(lua.LNil, lua.LNil, errorValue); err != wantErr {
|
|
||||||
t.Fatalf("upstream error = %v, want original error %v", err, wantErr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCallLuaHookCancellation(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
if err := L.DoString(`function HandleDNSQuery(domain, ipv4, ipv6, fake) while true do end end`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
|
|
||||||
defer cancel()
|
|
||||||
_, _, err := (&DNS{}).CallLuaHook(L, ctx, "example.com", featureDNS.IPOption{IPv4Enable: true})
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("CallLuaHook did not stop after context cancellation")
|
|
||||||
}
|
|
||||||
if L.Context() != nil {
|
|
||||||
t.Fatal("CallLuaHook left the canceled context on the Lua state")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCallLuaHookNormalizesDomain(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = []net.IP{net.ParseIP("127.0.0.1")}
|
|
||||||
L.SetGlobal("ips", addresses)
|
|
||||||
if err := L.DoString(`
|
|
||||||
function HandleDNSQuery(domain, ipv4, ipv6, fake)
|
|
||||||
assert(domain == "example.com")
|
|
||||||
assert(ipv4 and not ipv6 and not fake)
|
|
||||||
return ips, 60, nil
|
|
||||||
end
|
|
||||||
`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
s := &DNS{}
|
|
||||||
if _, _, err := s.CallLuaHook(L, context.Background(), "ExAmPlE.CoM", featureDNS.IPOption{IPv4Enable: true}); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestCallLuaHookRestoresState(t *testing.T) {
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name string
|
|
||||||
body string
|
|
||||||
wantErr bool
|
|
||||||
}{
|
|
||||||
{"success", `return ips, 60`, false},
|
|
||||||
{"error", `error("failed")`, true},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = []net.IP{net.ParseIP("127.0.0.1")}
|
|
||||||
L.SetGlobal("ips", addresses)
|
|
||||||
if err := L.DoString("function HandleDNSQuery() " + tc.body + " end"); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
previous, cancel := context.WithCancel(context.Background())
|
|
||||||
defer cancel()
|
|
||||||
L.SetContext(previous)
|
|
||||||
L.Push(lua.LTrue)
|
|
||||||
_, _, err := (&DNS{}).CallLuaHook(L, context.Background(), "example.com", featureDNS.IPOption{IPv4Enable: true})
|
|
||||||
if (err != nil) != tc.wantErr {
|
|
||||||
t.Fatalf("hook error = %v, want error %t", err, tc.wantErr)
|
|
||||||
}
|
|
||||||
if L.Context() != previous || L.GetTop() != 1 || L.Get(1) != lua.LTrue {
|
|
||||||
t.Fatal("hook did not restore the previous context and stack")
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaDNSServerQuery(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
geodata.RegisterLua(L)
|
|
||||||
option := featureDNS.IPOption{IPv4Enable: true}
|
|
||||||
ips := []net.IP{net.ParseIP("127.0.0.1"), net.ParseIP("8.8.8.8")}
|
|
||||||
server := &DNS{clients: []*Client{{server: &benchmarkLuaNameServer{ips: ips}, ipOption: &option, timeoutMs: time.Second}}}
|
|
||||||
server.RegisterLua(L)
|
|
||||||
if err := L.DoString(`
|
|
||||||
local server = require("xray.dns").Servers[1]
|
|
||||||
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.0/8")
|
|
||||||
function HandleDNSQuery(domain, ipv4, ipv6, fake)
|
|
||||||
local ips, ttl, err = server:Query(domain, ipv4, ipv6, fake)
|
|
||||||
assert(type(ips) == "userdata" and not err)
|
|
||||||
assert(matcher:AnyMatch(ips))
|
|
||||||
local matched = matcher:FilterIPs(ips)
|
|
||||||
return matched, ttl, err
|
|
||||||
end
|
|
||||||
`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
got, ttl, err := server.CallLuaHook(L, context.Background(), "example.com", option)
|
|
||||||
if err != nil || ttl != 60 || len(got) != 1 || !got[0].Equal(ips[0]) {
|
|
||||||
t.Fatalf("server query = %v, TTL %d, %v", got, ttl, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type luaDNSClient struct {
|
|
||||||
featureDNS.Client
|
|
||||||
lookup func(string, featureDNS.IPOption) ([]net.IP, uint32, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *luaDNSClient) LookupIP(domain string, option featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
return c.lookup(domain, option)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaDNSClientQuery(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
L.SetContext(context.Background())
|
|
||||||
defer L.RemoveContext()
|
|
||||||
geodata.RegisterLua(L)
|
|
||||||
want := []net.IP{{127, 0, 0, 1}}
|
|
||||||
client := &luaDNSClient{lookup: func(domain string, option featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
if domain != "MiXeD.Example." || !option.IPv4Enable || option.IPv6Enable || !option.FakeEnable {
|
|
||||||
t.Fatalf("dns.Query arguments = %q, %+v", domain, option)
|
|
||||||
}
|
|
||||||
return want, 42, nil
|
|
||||||
}}
|
|
||||||
RegisterLua(L, client)
|
|
||||||
if err := L.DoString(`
|
|
||||||
local dns = require("xray.dns")
|
|
||||||
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.1")
|
|
||||||
assert(dns.Servers == nil)
|
|
||||||
ips, ttl, err = dns.Query("MiXeD.Example.", true, false, true)
|
|
||||||
assert(not err and ttl == 42 and matcher:AnyMatch(ips))
|
|
||||||
`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
got := L.GetGlobal("ips").(*lua.LUserData).Value.([]net.IP)
|
|
||||||
if &got[0] != &want[0] {
|
|
||||||
t.Fatal("dns.Query copied the IP slice")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaDNSLocalClient(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
L.SetContext(context.Background())
|
|
||||||
defer L.RemoveContext()
|
|
||||||
RegisterLua(L, localdns.New())
|
|
||||||
if err := L.DoString(`
|
|
||||||
local dns = require("xray.dns")
|
|
||||||
assert(dns.Servers[1].ID == "localhost")
|
|
||||||
serverIPs, _, serverErr = dns.Servers[1]:Query("127.0.0.1", true, false, false)
|
|
||||||
clientIPs, _, clientErr = dns.Query("127.0.0.1", true, false, false)
|
|
||||||
assert(not serverErr and not clientErr)
|
|
||||||
`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
for _, name := range []string{"serverIPs", "clientIPs"} {
|
|
||||||
ips := L.GetGlobal(name).(*lua.LUserData).Value.([]net.IP)
|
|
||||||
if len(ips) != 1 || !ips[0].Equal(net.ParseIP("127.0.0.1")) {
|
|
||||||
t.Fatalf("%s = %v", name, ips)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type benchmarkLuaNameServer struct {
|
|
||||||
ips []net.IP
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*benchmarkLuaNameServer) Name() string { return "benchmark" }
|
|
||||||
func (*benchmarkLuaNameServer) IsDisableCache() bool { return true }
|
|
||||||
func (s *benchmarkLuaNameServer) QueryIP(context.Context, string, featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
return s.ips, 60, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// BenchmarkLuaDNSHookCall isolates a preloaded Lua hook and its server:Query bridge.
|
|
||||||
// The direct case measures the same DNS client without Lua.
|
|
||||||
func BenchmarkLuaDNSHookCall(b *testing.B) {
|
|
||||||
option := featureDNS.IPOption{IPv4Enable: true}
|
|
||||||
ip := net.ParseIP("127.0.0.1")
|
|
||||||
upstream := &benchmarkLuaNameServer{ips: []net.IP{ip}}
|
|
||||||
client := &Client{server: upstream, ipOption: &option, timeoutMs: time.Second}
|
|
||||||
server := &DNS{clients: []*Client{client}}
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
server.RegisterLua(L)
|
|
||||||
if err := L.DoString(`
|
|
||||||
local server = require("xray.dns").Servers[1]
|
|
||||||
function HandleDNSQuery(domain, ipv4, ipv6, fake)
|
|
||||||
return server:Query(domain, ipv4, ipv6, fake)
|
|
||||||
end
|
|
||||||
`); err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
for _, bench := range []struct {
|
|
||||||
name string
|
|
||||||
query func() ([]net.IP, uint32, error)
|
|
||||||
}{
|
|
||||||
{"direct", func() ([]net.IP, uint32, error) { return client.QueryIP(ctx, "example.com", option) }},
|
|
||||||
{"lua_hook", func() ([]net.IP, uint32, error) { return server.CallLuaHook(L, ctx, "example.com", option) }},
|
|
||||||
} {
|
|
||||||
b.Run(bench.name, func(b *testing.B) {
|
|
||||||
b.ReportAllocs()
|
|
||||||
b.ResetTimer()
|
|
||||||
var ips []net.IP
|
|
||||||
var ttl uint32
|
|
||||||
var err error
|
|
||||||
for i := 0; i < b.N; i++ {
|
|
||||||
ips, ttl, err = bench.query()
|
|
||||||
if err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
b.StopTimer()
|
|
||||||
if ttl != 60 || len(ips) != 1 || !ips[0].Equal(ip) {
|
|
||||||
b.Fatalf("query() = %v, TTL %d; want %v, TTL 60", ips, ttl, ip)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -29,7 +29,6 @@ type Server interface {
|
|||||||
|
|
||||||
// Client is the interface for DNS client.
|
// Client is the interface for DNS client.
|
||||||
type Client struct {
|
type Client struct {
|
||||||
id string
|
|
||||||
server Server
|
server Server
|
||||||
skipFallback bool
|
skipFallback bool
|
||||||
expectedIPs geodata.IPMatcher
|
expectedIPs geodata.IPMatcher
|
||||||
@@ -98,7 +97,7 @@ func NewClient(
|
|||||||
ipOption dns.IPOption,
|
ipOption dns.IPOption,
|
||||||
updateRules func(bool),
|
updateRules func(bool),
|
||||||
) (*Client, error) {
|
) (*Client, error) {
|
||||||
client := &Client{id: ns.Id}
|
client := &Client{}
|
||||||
err := core.RequireFeatures(ctx, func(dispatcher routing.Dispatcher) error {
|
err := core.RequireFeatures(ctx, func(dispatcher routing.Dispatcher) error {
|
||||||
// Create a new server for each client for now
|
// Create a new server for each client for now
|
||||||
server, err := NewServer(ctx, ns.Address.AsDestination(), dispatcher, disableCache, serveStale, serveExpiredTTL, clientIP)
|
server, err := NewServer(ctx, ns.Address.AsDestination(), dispatcher, disableCache, serveStale, serveExpiredTTL, clientIP)
|
||||||
|
|||||||
@@ -49,5 +49,5 @@ func NewLocalNameServer() *LocalNameServer {
|
|||||||
|
|
||||||
// NewLocalDNSClient creates localdns client object for directly lookup in system DNS.
|
// NewLocalDNSClient creates localdns client object for directly lookup in system DNS.
|
||||||
func NewLocalDNSClient(ipOption dns.IPOption) *Client {
|
func NewLocalDNSClient(ipOption dns.IPOption) *Client {
|
||||||
return &Client{id: "localhost", server: NewLocalNameServer(), ipOption: &ipOption}
|
return &Client{server: NewLocalNameServer(), ipOption: &ipOption}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,73 +0,0 @@
|
|||||||
package dns
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/geodata"
|
|
||||||
"github.com/xtls/xray-core/common/log"
|
|
||||||
luamgr "github.com/xtls/xray-core/common/lua"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/features/dns"
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
const scriptExecutionTimeout = 10 * time.Second
|
|
||||||
|
|
||||||
type scriptEngine struct {
|
|
||||||
dns *DNS
|
|
||||||
pool *luamgr.Pool
|
|
||||||
}
|
|
||||||
|
|
||||||
func newScriptEngine(path string, server *DNS) (*scriptEngine, error) {
|
|
||||||
program, err := luamgr.CompileFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
e := &scriptEngine{dns: server}
|
|
||||||
e.pool, err = luamgr.NewPool(server.ctx, func(poolCtx context.Context) (*lua.LState, error) {
|
|
||||||
initCtx, cancel := context.WithTimeout(poolCtx, scriptExecutionTimeout)
|
|
||||||
defer cancel()
|
|
||||||
L, err := program.NewState(initCtx, func(L *lua.LState) {
|
|
||||||
geodata.RegisterLua(L)
|
|
||||||
log.RegisterLua(L)
|
|
||||||
server.RegisterLua(L)
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if L.GetGlobal("HandleDNSQuery").Type() != lua.LTFunction {
|
|
||||||
L.Close()
|
|
||||||
return nil, errors.New("DNS script must define HandleDNSQuery(domain, ipv4, ipv6, fake)")
|
|
||||||
}
|
|
||||||
return L, nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
errors.LogInfo(server.ctx, "DNS script initialized from ", path)
|
|
||||||
return e, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e *scriptEngine) close() {
|
|
||||||
e.pool.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e *scriptEngine) query(domain string, option dns.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
L, err := e.pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
return nil, 0, err
|
|
||||||
}
|
|
||||||
reusable := false
|
|
||||||
defer func() {
|
|
||||||
e.pool.Release(L, reusable)
|
|
||||||
}()
|
|
||||||
queryCtx, cancel := context.WithTimeout(e.pool.Context(), scriptExecutionTimeout)
|
|
||||||
defer cancel()
|
|
||||||
ips, ttl, err := e.dns.CallLuaHook(L, queryCtx, domain, option)
|
|
||||||
if err == nil {
|
|
||||||
reusable = true
|
|
||||||
}
|
|
||||||
return ips, ttl, err
|
|
||||||
}
|
|
||||||
@@ -1,197 +0,0 @@
|
|||||||
package dns
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
featureDNS "github.com/xtls/xray-core/features/dns"
|
|
||||||
)
|
|
||||||
|
|
||||||
type geoIPScriptNameServer struct {
|
|
||||||
name string
|
|
||||||
answers map[string]net.IP
|
|
||||||
ttl uint32
|
|
||||||
calls int
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *geoIPScriptNameServer) Name() string { return s.name }
|
|
||||||
func (s *geoIPScriptNameServer) IsDisableCache() bool { return true }
|
|
||||||
|
|
||||||
func (s *geoIPScriptNameServer) QueryIP(ctx context.Context, domain string, _ featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
if err := ctx.Err(); err != nil {
|
|
||||||
return nil, 0, err
|
|
||||||
}
|
|
||||||
s.calls++
|
|
||||||
ip, ok := s.answers[domain]
|
|
||||||
if !ok {
|
|
||||||
return nil, 0, featureDNS.ErrEmptyResponse
|
|
||||||
}
|
|
||||||
return []net.IP{ip}, s.ttl, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDNSScriptGeoIPFallback(t *testing.T) {
|
|
||||||
t.Setenv("xray.location.asset", filepath.Join("..", "..", "resources"))
|
|
||||||
script := `
|
|
||||||
local servers = require("xray.dns").Servers
|
|
||||||
local us_ips = require("xray.geodata").BuildIPMatcher("geoip:us")
|
|
||||||
|
|
||||||
local by_id = {}
|
|
||||||
for _, server in ipairs(servers) do
|
|
||||||
by_id[server.ID] = server
|
|
||||||
end
|
|
||||||
assert(by_id.primary and by_id.fallback, "primary and fallback DNS servers are required")
|
|
||||||
|
|
||||||
function HandleDNSQuery(domain, ipv4, ipv6, fake)
|
|
||||||
local ips, ttl, err = by_id.primary:Query(domain, ipv4, ipv6, fake)
|
|
||||||
if not err and us_ips:AnyMatch(ips) then
|
|
||||||
return ips, ttl, nil
|
|
||||||
end
|
|
||||||
return by_id.fallback:Query(domain, ipv4, ipv6, fake)
|
|
||||||
end
|
|
||||||
`
|
|
||||||
scriptPath := filepath.Join(t.TempDir(), "geoip_fallback.lua")
|
|
||||||
if err := os.WriteFile(scriptPath, []byte(script), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
primary := &geoIPScriptNameServer{
|
|
||||||
name: "primary",
|
|
||||||
answers: map[string]net.IP{
|
|
||||||
"us.example": net.ParseIP("2001:4860:4860::8888"),
|
|
||||||
"other.example": net.ParseIP("127.0.0.1"),
|
|
||||||
},
|
|
||||||
ttl: 30,
|
|
||||||
}
|
|
||||||
fallback := &geoIPScriptNameServer{
|
|
||||||
name: "fallback",
|
|
||||||
answers: map[string]net.IP{"other.example": net.ParseIP("9.9.9.9")},
|
|
||||||
ttl: 60,
|
|
||||||
}
|
|
||||||
option := featureDNS.IPOption{IPv4Enable: true, IPv6Enable: true}
|
|
||||||
hosts, err := NewStaticHosts(nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
server := &DNS{
|
|
||||||
ctx: context.Background(),
|
|
||||||
hosts: hosts,
|
|
||||||
ipOption: &option,
|
|
||||||
scriptPath: scriptPath,
|
|
||||||
clients: []*Client{
|
|
||||||
{id: "primary", server: primary, ipOption: &option, timeoutMs: 2 * time.Second},
|
|
||||||
{id: "fallback", server: fallback, ipOption: &option, timeoutMs: 2 * time.Second},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
if err := server.Start(); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer server.Close()
|
|
||||||
|
|
||||||
for _, tc := range []struct {
|
|
||||||
domain string
|
|
||||||
ip net.IP
|
|
||||||
ttl uint32
|
|
||||||
}{
|
|
||||||
{"Us.Example.", net.ParseIP("2001:4860:4860::8888"), 30},
|
|
||||||
{"other.example", net.ParseIP("9.9.9.9"), 60},
|
|
||||||
} {
|
|
||||||
ips, ttl, err := server.LookupIP(tc.domain, option)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("LookupIP(%q): %v", tc.domain, err)
|
|
||||||
}
|
|
||||||
if ttl != tc.ttl || len(ips) != 1 || !ips[0].Equal(tc.ip) {
|
|
||||||
t.Fatalf("LookupIP(%q) = %v, TTL %d; want %v, TTL %d", tc.domain, ips, ttl, tc.ip, tc.ttl)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if primary.calls != 2 || fallback.calls != 1 {
|
|
||||||
t.Fatalf("upstream calls: primary %d, fallback %d; want 2 and 1", primary.calls, fallback.calls)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDNSScriptRejectsInvalidStartup(t *testing.T) {
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name string
|
|
||||||
script string
|
|
||||||
}{
|
|
||||||
{"syntax", "function HandleDNSQuery("},
|
|
||||||
{"missing hook", "value = 1"},
|
|
||||||
{"top-level error", `error("setup failed")`},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
path := filepath.Join(t.TempDir(), "script.lua")
|
|
||||||
if err := os.WriteFile(path, []byte(tc.script), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
server := &DNS{ctx: context.Background(), scriptPath: path}
|
|
||||||
if err := server.Start(); err == nil {
|
|
||||||
t.Fatal("Start accepted an invalid DNS script")
|
|
||||||
}
|
|
||||||
if server.script != nil {
|
|
||||||
t.Fatal("Start retained a script engine after failure")
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDNSScriptHookErrorAndFakeDNSOption(t *testing.T) {
|
|
||||||
path := filepath.Join(t.TempDir(), "script.lua")
|
|
||||||
script := `
|
|
||||||
local server = require("xray.dns").Servers[1]
|
|
||||||
local log = require("xray.log")
|
|
||||||
log.Info("DNS script loaded")
|
|
||||||
function HandleDNSQuery(domain, ipv4, ipv6, fake)
|
|
||||||
log.Debug("DNS query: ", domain)
|
|
||||||
if domain == "bad.example" then error("script failure") end
|
|
||||||
local ips, ttl, err = server:Query(domain, ipv4, ipv6, fake)
|
|
||||||
if err then log.Error("DNS failed: ", err) end
|
|
||||||
return ips, ttl, err
|
|
||||||
end
|
|
||||||
`
|
|
||||||
if err := os.WriteFile(path, []byte(script), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
option := featureDNS.IPOption{IPv4Enable: true}
|
|
||||||
hosts, err := NewStaticHosts(nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
upstream := &geoIPScriptNameServer{
|
|
||||||
name: "FakeDNS",
|
|
||||||
answers: map[string]net.IP{"good.example": net.ParseIP("198.18.0.1")},
|
|
||||||
ttl: 30,
|
|
||||||
}
|
|
||||||
server := &DNS{
|
|
||||||
ctx: context.Background(),
|
|
||||||
hosts: hosts,
|
|
||||||
ipOption: &option,
|
|
||||||
scriptPath: path,
|
|
||||||
clients: []*Client{{id: "fake", server: upstream, ipOption: &option, timeoutMs: time.Second}},
|
|
||||||
}
|
|
||||||
if err := server.Start(); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer server.Close()
|
|
||||||
|
|
||||||
if _, _, err := server.LookupIP("bad.example", option); err == nil || !strings.Contains(err.Error(), "script failure") {
|
|
||||||
t.Fatalf("hook failure = %v, want script failure", err)
|
|
||||||
}
|
|
||||||
if _, _, err := server.LookupIP("good.example", option); err != featureDNS.ErrEmptyResponse {
|
|
||||||
t.Fatalf("FakeDNS without FakeEnable = %v, want ErrEmptyResponse", err)
|
|
||||||
}
|
|
||||||
if upstream.calls != 0 {
|
|
||||||
t.Fatalf("FakeDNS was queried without FakeEnable: %d calls", upstream.calls)
|
|
||||||
}
|
|
||||||
withFake := featureDNS.IPOption{IPv4Enable: true, FakeEnable: true}
|
|
||||||
ips, ttl, err := server.LookupIP("good.example", withFake)
|
|
||||||
if err != nil || ttl != 30 || len(ips) != 1 || !ips[0].Equal(net.ParseIP("198.18.0.1")) {
|
|
||||||
t.Fatalf("FakeDNS with FakeEnable = %v, TTL %d, %v", ips, ttl, err)
|
|
||||||
}
|
|
||||||
if upstream.calls != 1 {
|
|
||||||
t.Fatalf("FakeDNS query count = %d, want 1", upstream.calls)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
+22
-11
@@ -330,6 +330,7 @@ type SenderConfig struct {
|
|||||||
// Send traffic through the given IP. Only IP is allowed.
|
// Send traffic through the given IP. Only IP is allowed.
|
||||||
Via *net.IPOrDomain `protobuf:"bytes,1,opt,name=via,proto3" json:"via,omitempty"`
|
Via *net.IPOrDomain `protobuf:"bytes,1,opt,name=via,proto3" json:"via,omitempty"`
|
||||||
StreamSettings *internet.StreamConfig `protobuf:"bytes,2,opt,name=stream_settings,json=streamSettings,proto3" json:"stream_settings,omitempty"`
|
StreamSettings *internet.StreamConfig `protobuf:"bytes,2,opt,name=stream_settings,json=streamSettings,proto3" json:"stream_settings,omitempty"`
|
||||||
|
ProxySettings *internet.ProxyConfig `protobuf:"bytes,3,opt,name=proxy_settings,json=proxySettings,proto3" json:"proxy_settings,omitempty"`
|
||||||
MultiplexSettings *MultiplexingConfig `protobuf:"bytes,4,opt,name=multiplex_settings,json=multiplexSettings,proto3" json:"multiplex_settings,omitempty"`
|
MultiplexSettings *MultiplexingConfig `protobuf:"bytes,4,opt,name=multiplex_settings,json=multiplexSettings,proto3" json:"multiplex_settings,omitempty"`
|
||||||
ViaCidr string `protobuf:"bytes,5,opt,name=via_cidr,json=viaCidr,proto3" json:"via_cidr,omitempty"`
|
ViaCidr string `protobuf:"bytes,5,opt,name=via_cidr,json=viaCidr,proto3" json:"via_cidr,omitempty"`
|
||||||
TargetStrategy internet.DomainStrategy `protobuf:"varint,6,opt,name=target_strategy,json=targetStrategy,proto3,enum=xray.transport.internet.DomainStrategy" json:"target_strategy,omitempty"`
|
TargetStrategy internet.DomainStrategy `protobuf:"varint,6,opt,name=target_strategy,json=targetStrategy,proto3,enum=xray.transport.internet.DomainStrategy" json:"target_strategy,omitempty"`
|
||||||
@@ -381,6 +382,13 @@ func (x *SenderConfig) GetStreamSettings() *internet.StreamConfig {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (x *SenderConfig) GetProxySettings() *internet.ProxyConfig {
|
||||||
|
if x != nil {
|
||||||
|
return x.ProxySettings
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (x *SenderConfig) GetMultiplexSettings() *MultiplexingConfig {
|
func (x *SenderConfig) GetMultiplexSettings() *MultiplexingConfig {
|
||||||
if x != nil {
|
if x != nil {
|
||||||
return x.MultiplexSettings
|
return x.MultiplexSettings
|
||||||
@@ -498,13 +506,14 @@ const file_app_proxyman_config_proto_rawDesc = "" +
|
|||||||
"\x03tag\x18\x01 \x01(\tR\x03tag\x12M\n" +
|
"\x03tag\x18\x01 \x01(\tR\x03tag\x12M\n" +
|
||||||
"\x11receiver_settings\x18\x02 \x01(\v2 .xray.common.serial.TypedMessageR\x10receiverSettings\x12G\n" +
|
"\x11receiver_settings\x18\x02 \x01(\v2 .xray.common.serial.TypedMessageR\x10receiverSettings\x12G\n" +
|
||||||
"\x0eproxy_settings\x18\x03 \x01(\v2 .xray.common.serial.TypedMessageR\rproxySettings\"\x10\n" +
|
"\x0eproxy_settings\x18\x03 \x01(\v2 .xray.common.serial.TypedMessageR\rproxySettings\"\x10\n" +
|
||||||
"\x0eOutboundConfig\"\xd6\x02\n" +
|
"\x0eOutboundConfig\"\x9d\x03\n" +
|
||||||
"\fSenderConfig\x12-\n" +
|
"\fSenderConfig\x12-\n" +
|
||||||
"\x03via\x18\x01 \x01(\v2\x1b.xray.common.net.IPOrDomainR\x03via\x12N\n" +
|
"\x03via\x18\x01 \x01(\v2\x1b.xray.common.net.IPOrDomainR\x03via\x12N\n" +
|
||||||
"\x0fstream_settings\x18\x02 \x01(\v2%.xray.transport.internet.StreamConfigR\x0estreamSettings\x12T\n" +
|
"\x0fstream_settings\x18\x02 \x01(\v2%.xray.transport.internet.StreamConfigR\x0estreamSettings\x12K\n" +
|
||||||
|
"\x0eproxy_settings\x18\x03 \x01(\v2$.xray.transport.internet.ProxyConfigR\rproxySettings\x12T\n" +
|
||||||
"\x12multiplex_settings\x18\x04 \x01(\v2%.xray.app.proxyman.MultiplexingConfigR\x11multiplexSettings\x12\x19\n" +
|
"\x12multiplex_settings\x18\x04 \x01(\v2%.xray.app.proxyman.MultiplexingConfigR\x11multiplexSettings\x12\x19\n" +
|
||||||
"\bvia_cidr\x18\x05 \x01(\tR\aviaCidr\x12P\n" +
|
"\bvia_cidr\x18\x05 \x01(\tR\aviaCidr\x12P\n" +
|
||||||
"\x0ftarget_strategy\x18\x06 \x01(\x0e2'.xray.transport.internet.DomainStrategyR\x0etargetStrategyJ\x04\b\x03\x10\x04\"\xa4\x01\n" +
|
"\x0ftarget_strategy\x18\x06 \x01(\x0e2'.xray.transport.internet.DomainStrategyR\x0etargetStrategy\"\xa4\x01\n" +
|
||||||
"\x12MultiplexingConfig\x12\x18\n" +
|
"\x12MultiplexingConfig\x12\x18\n" +
|
||||||
"\aenabled\x18\x01 \x01(\bR\aenabled\x12 \n" +
|
"\aenabled\x18\x01 \x01(\bR\aenabled\x12 \n" +
|
||||||
"\vconcurrency\x18\x02 \x01(\x05R\vconcurrency\x12(\n" +
|
"\vconcurrency\x18\x02 \x01(\x05R\vconcurrency\x12(\n" +
|
||||||
@@ -539,7 +548,8 @@ var file_app_proxyman_config_proto_goTypes = []any{
|
|||||||
(*net.IPOrDomain)(nil), // 10: xray.common.net.IPOrDomain
|
(*net.IPOrDomain)(nil), // 10: xray.common.net.IPOrDomain
|
||||||
(*internet.StreamConfig)(nil), // 11: xray.transport.internet.StreamConfig
|
(*internet.StreamConfig)(nil), // 11: xray.transport.internet.StreamConfig
|
||||||
(*serial.TypedMessage)(nil), // 12: xray.common.serial.TypedMessage
|
(*serial.TypedMessage)(nil), // 12: xray.common.serial.TypedMessage
|
||||||
(internet.DomainStrategy)(0), // 13: xray.transport.internet.DomainStrategy
|
(*internet.ProxyConfig)(nil), // 13: xray.transport.internet.ProxyConfig
|
||||||
|
(internet.DomainStrategy)(0), // 14: xray.transport.internet.DomainStrategy
|
||||||
}
|
}
|
||||||
var file_app_proxyman_config_proto_depIdxs = []int32{
|
var file_app_proxyman_config_proto_depIdxs = []int32{
|
||||||
7, // 0: xray.app.proxyman.SniffingConfig.domains_excluded:type_name -> xray.common.geodata.DomainRule
|
7, // 0: xray.app.proxyman.SniffingConfig.domains_excluded:type_name -> xray.common.geodata.DomainRule
|
||||||
@@ -552,13 +562,14 @@ var file_app_proxyman_config_proto_depIdxs = []int32{
|
|||||||
12, // 7: xray.app.proxyman.InboundHandlerConfig.proxy_settings:type_name -> xray.common.serial.TypedMessage
|
12, // 7: xray.app.proxyman.InboundHandlerConfig.proxy_settings:type_name -> xray.common.serial.TypedMessage
|
||||||
10, // 8: xray.app.proxyman.SenderConfig.via:type_name -> xray.common.net.IPOrDomain
|
10, // 8: xray.app.proxyman.SenderConfig.via:type_name -> xray.common.net.IPOrDomain
|
||||||
11, // 9: xray.app.proxyman.SenderConfig.stream_settings:type_name -> xray.transport.internet.StreamConfig
|
11, // 9: xray.app.proxyman.SenderConfig.stream_settings:type_name -> xray.transport.internet.StreamConfig
|
||||||
6, // 10: xray.app.proxyman.SenderConfig.multiplex_settings:type_name -> xray.app.proxyman.MultiplexingConfig
|
13, // 10: xray.app.proxyman.SenderConfig.proxy_settings:type_name -> xray.transport.internet.ProxyConfig
|
||||||
13, // 11: xray.app.proxyman.SenderConfig.target_strategy:type_name -> xray.transport.internet.DomainStrategy
|
6, // 11: xray.app.proxyman.SenderConfig.multiplex_settings:type_name -> xray.app.proxyman.MultiplexingConfig
|
||||||
12, // [12:12] is the sub-list for method output_type
|
14, // 12: xray.app.proxyman.SenderConfig.target_strategy:type_name -> xray.transport.internet.DomainStrategy
|
||||||
12, // [12:12] is the sub-list for method input_type
|
13, // [13:13] is the sub-list for method output_type
|
||||||
12, // [12:12] is the sub-list for extension type_name
|
13, // [13:13] is the sub-list for method input_type
|
||||||
12, // [12:12] is the sub-list for extension extendee
|
13, // [13:13] is the sub-list for extension type_name
|
||||||
0, // [0:12] is the sub-list for field type_name
|
13, // [13:13] is the sub-list for extension extendee
|
||||||
|
0, // [0:13] is the sub-list for field type_name
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() { file_app_proxyman_config_proto_init() }
|
func init() { file_app_proxyman_config_proto_init() }
|
||||||
|
|||||||
@@ -57,7 +57,7 @@ message SenderConfig {
|
|||||||
// Send traffic through the given IP. Only IP is allowed.
|
// Send traffic through the given IP. Only IP is allowed.
|
||||||
xray.common.net.IPOrDomain via = 1;
|
xray.common.net.IPOrDomain via = 1;
|
||||||
xray.transport.internet.StreamConfig stream_settings = 2;
|
xray.transport.internet.StreamConfig stream_settings = 2;
|
||||||
reserved 3;
|
xray.transport.internet.ProxyConfig proxy_settings = 3;
|
||||||
MultiplexingConfig multiplex_settings = 4;
|
MultiplexingConfig multiplex_settings = 4;
|
||||||
string via_cidr = 5;
|
string via_cidr = 5;
|
||||||
xray.transport.internet.DomainStrategy target_strategy = 6;
|
xray.transport.internet.DomainStrategy target_strategy = 6;
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ import (
|
|||||||
"github.com/xtls/xray-core/common/errors"
|
"github.com/xtls/xray-core/common/errors"
|
||||||
"github.com/xtls/xray-core/common/mux"
|
"github.com/xtls/xray-core/common/mux"
|
||||||
"github.com/xtls/xray-core/common/net"
|
"github.com/xtls/xray-core/common/net"
|
||||||
|
"github.com/xtls/xray-core/common/net/cnc"
|
||||||
"github.com/xtls/xray-core/common/serial"
|
"github.com/xtls/xray-core/common/serial"
|
||||||
"github.com/xtls/xray-core/common/session"
|
"github.com/xtls/xray-core/common/session"
|
||||||
"github.com/xtls/xray-core/core"
|
"github.com/xtls/xray-core/core"
|
||||||
@@ -25,6 +26,8 @@ import (
|
|||||||
"github.com/xtls/xray-core/transport"
|
"github.com/xtls/xray-core/transport"
|
||||||
"github.com/xtls/xray-core/transport/internet"
|
"github.com/xtls/xray-core/transport/internet"
|
||||||
"github.com/xtls/xray-core/transport/internet/stat"
|
"github.com/xtls/xray-core/transport/internet/stat"
|
||||||
|
"github.com/xtls/xray-core/transport/internet/tls"
|
||||||
|
"github.com/xtls/xray-core/transport/pipe"
|
||||||
"google.golang.org/protobuf/proto"
|
"google.golang.org/protobuf/proto"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -60,6 +63,7 @@ type Handler struct {
|
|||||||
streamSettings *internet.MemoryStreamConfig
|
streamSettings *internet.MemoryStreamConfig
|
||||||
proxyConfig proto.Message
|
proxyConfig proto.Message
|
||||||
proxy proxy.Outbound
|
proxy proxy.Outbound
|
||||||
|
outboundManager outbound.Manager
|
||||||
mux *mux.ClientManager
|
mux *mux.ClientManager
|
||||||
xudp *mux.ClientManager
|
xudp *mux.ClientManager
|
||||||
udp443 string
|
udp443 string
|
||||||
@@ -73,6 +77,7 @@ func NewHandler(ctx context.Context, config *core.OutboundHandlerConfig) (outbou
|
|||||||
uplinkCounter, downlinkCounter := getStatCounter(v, config.Tag)
|
uplinkCounter, downlinkCounter := getStatCounter(v, config.Tag)
|
||||||
h := &Handler{
|
h := &Handler{
|
||||||
tag: config.Tag,
|
tag: config.Tag,
|
||||||
|
outboundManager: v.GetFeature(outbound.ManagerType()).(outbound.Manager),
|
||||||
uplinkCounter: uplinkCounter,
|
uplinkCounter: uplinkCounter,
|
||||||
downlinkCounter: downlinkCounter,
|
downlinkCounter: downlinkCounter,
|
||||||
}
|
}
|
||||||
@@ -103,11 +108,9 @@ func NewHandler(ctx context.Context, config *core.OutboundHandlerConfig) (outbou
|
|||||||
|
|
||||||
ctx = session.ContextWithFullHandler(ctx, h)
|
ctx = session.ContextWithFullHandler(ctx, h)
|
||||||
|
|
||||||
if h.streamSettings != nil {
|
newCtx := session.ContextWithStreamSettings(ctx, h.streamSettings)
|
||||||
ctx = session.ContextWithStreamSettings(ctx, h.streamSettings)
|
|
||||||
}
|
|
||||||
|
|
||||||
rawProxyHandler, err := common.CreateObject(ctx, proxyConfig)
|
rawProxyHandler, err := common.CreateObject(newCtx, proxyConfig)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -194,6 +197,7 @@ func (h *Handler) Dispatch(ctx context.Context, link *transport.Link) {
|
|||||||
common.Interrupt(link.Reader)
|
common.Interrupt(link.Reader)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
} else {
|
} else {
|
||||||
unchangedDomain := ob.Target.Address.Domain()
|
unchangedDomain := ob.Target.Address.Domain()
|
||||||
ob.Target.Address = net.IPAddress(ips[dice.Roll(len(ips))])
|
ob.Target.Address = net.IPAddress(ips[dice.Roll(len(ips))])
|
||||||
@@ -266,26 +270,66 @@ func (h *Handler) DestIpAddress() net.IP {
|
|||||||
|
|
||||||
// Dial implements internet.Dialer.
|
// Dial implements internet.Dialer.
|
||||||
func (h *Handler) Dial(ctx context.Context, dest net.Destination) (stat.Connection, error) {
|
func (h *Handler) Dial(ctx context.Context, dest net.Destination) (stat.Connection, error) {
|
||||||
if h.senderSettings != nil && h.senderSettings.Via != nil {
|
if h.senderSettings != nil {
|
||||||
outbounds := session.OutboundsFromContext(ctx)
|
|
||||||
ob := outbounds[len(outbounds)-1]
|
if h.senderSettings.ProxySettings.HasTag() {
|
||||||
h.SetOutboundGateway(ctx, ob)
|
|
||||||
|
tag := h.senderSettings.ProxySettings.Tag
|
||||||
|
handler := h.outboundManager.GetHandler(tag)
|
||||||
|
if handler != nil {
|
||||||
|
errors.LogDebug(ctx, "proxying to ", tag, " for dest ", dest)
|
||||||
|
outbounds := session.OutboundsFromContext(ctx)
|
||||||
|
ctx = session.ContextWithOutbounds(ctx, append(outbounds, &session.Outbound{
|
||||||
|
Target: dest,
|
||||||
|
Tag: tag,
|
||||||
|
})) // add another outbound in session ctx
|
||||||
|
opts := pipe.OptionsFromContext(ctx)
|
||||||
|
uplinkReader, uplinkWriter := pipe.New(opts...)
|
||||||
|
downlinkReader, downlinkWriter := pipe.New(opts...)
|
||||||
|
|
||||||
|
go handler.Dispatch(ctx, &transport.Link{Reader: uplinkReader, Writer: downlinkWriter})
|
||||||
|
conn := cnc.NewConnection(cnc.ConnectionInputMulti(uplinkWriter), cnc.ConnectionOutputMulti(downlinkReader))
|
||||||
|
|
||||||
|
if config := tls.ConfigFromStreamSettings(h.streamSettings); config != nil {
|
||||||
|
tlsConfig := config.GetTLSConfig(tls.WithDestination(dest))
|
||||||
|
conn = tls.Client(conn, tlsConfig)
|
||||||
|
}
|
||||||
|
|
||||||
|
return h.getStatCouterConnection(conn), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
errors.LogError(ctx, "failed to get outbound handler with tag: ", tag)
|
||||||
|
return nil, errors.New("failed to get outbound handler with tag: " + tag)
|
||||||
|
}
|
||||||
|
|
||||||
|
if h.senderSettings.Via != nil {
|
||||||
|
outbounds := session.OutboundsFromContext(ctx)
|
||||||
|
ob := outbounds[len(outbounds)-1]
|
||||||
|
h.SetOutboundGateway(ctx, ob)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
conn, err := internet.Dial(ctx, dest, h.streamSettings)
|
conn, err := internet.Dial(ctx, dest, h.streamSettings)
|
||||||
conn = h.getStatCouterConnection(conn)
|
conn = h.getStatCouterConnection(conn)
|
||||||
|
outbounds := session.OutboundsFromContext(ctx)
|
||||||
|
if outbounds != nil {
|
||||||
|
ob := outbounds[len(outbounds)-1]
|
||||||
|
ob.Conn = conn
|
||||||
|
} else {
|
||||||
|
// for Vision's pre-connect
|
||||||
|
}
|
||||||
return conn, err
|
return conn, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Handler) SetOutboundGateway(ctx context.Context, ob *session.Outbound) {
|
func (h *Handler) SetOutboundGateway(ctx context.Context, ob *session.Outbound) {
|
||||||
if ob.Gateway == nil && h.senderSettings != nil && h.senderSettings.Via != nil &&
|
if ob.Gateway == nil && h.senderSettings != nil && h.senderSettings.Via != nil && !h.senderSettings.ProxySettings.HasTag() && (h.streamSettings.SocketSettings == nil || len(h.streamSettings.SocketSettings.DialerProxy) == 0) {
|
||||||
(h.streamSettings.SocketSettings == nil || len(h.streamSettings.SocketSettings.DialerProxy) == 0) {
|
|
||||||
var domain string
|
var domain string
|
||||||
addr := h.senderSettings.Via.AsAddress()
|
addr := h.senderSettings.Via.AsAddress()
|
||||||
domain = h.senderSettings.Via.GetDomain()
|
domain = h.senderSettings.Via.GetDomain()
|
||||||
switch {
|
switch {
|
||||||
case h.senderSettings.ViaCidr != "":
|
case h.senderSettings.ViaCidr != "":
|
||||||
ob.Gateway = ParseRandomIP(addr, h.senderSettings.ViaCidr)
|
ob.Gateway = ParseRandomIP(addr, h.senderSettings.ViaCidr)
|
||||||
|
|
||||||
case domain == "origin":
|
case domain == "origin":
|
||||||
if inbound := session.InboundFromContext(ctx); inbound != nil {
|
if inbound := session.InboundFromContext(ctx); inbound != nil {
|
||||||
if inbound.Local.IsValid() && inbound.Local.Address.Family().IsIP() {
|
if inbound.Local.IsValid() && inbound.Local.Address.Family().IsIP() {
|
||||||
@@ -300,9 +344,11 @@ func (h *Handler) SetOutboundGateway(ctx context.Context, ob *session.Outbound)
|
|||||||
errors.LogDebug(ctx, "use inbound source ip as sendthrough: ", inbound.Source.Address.String())
|
errors.LogDebug(ctx, "use inbound source ip as sendthrough: ", inbound.Source.Address.String())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
default: // case addr.Family().IsDomain():
|
// case addr.Family().IsDomain():
|
||||||
|
default:
|
||||||
ob.Gateway = addr
|
ob.Gateway = addr
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
+4
-14
@@ -587,10 +587,8 @@ type Config struct {
|
|||||||
DomainStrategy Config_DomainStrategy `protobuf:"varint,1,opt,name=domain_strategy,json=domainStrategy,proto3,enum=xray.app.router.Config_DomainStrategy" json:"domain_strategy,omitempty"`
|
DomainStrategy Config_DomainStrategy `protobuf:"varint,1,opt,name=domain_strategy,json=domainStrategy,proto3,enum=xray.app.router.Config_DomainStrategy" json:"domain_strategy,omitempty"`
|
||||||
Rule []*RoutingRule `protobuf:"bytes,2,rep,name=rule,proto3" json:"rule,omitempty"`
|
Rule []*RoutingRule `protobuf:"bytes,2,rep,name=rule,proto3" json:"rule,omitempty"`
|
||||||
BalancingRule []*BalancingRule `protobuf:"bytes,3,rep,name=balancing_rule,json=balancingRule,proto3" json:"balancing_rule,omitempty"`
|
BalancingRule []*BalancingRule `protobuf:"bytes,3,rep,name=balancing_rule,json=balancingRule,proto3" json:"balancing_rule,omitempty"`
|
||||||
// Absolute path to the Lua routing script.
|
unknownFields protoimpl.UnknownFields
|
||||||
Script string `protobuf:"bytes,4,opt,name=script,proto3" json:"script,omitempty"`
|
sizeCache protoimpl.SizeCache
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *Config) Reset() {
|
func (x *Config) Reset() {
|
||||||
@@ -644,13 +642,6 @@ func (x *Config) GetBalancingRule() []*BalancingRule {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *Config) GetScript() string {
|
|
||||||
if x != nil {
|
|
||||||
return x.Script
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
var File_app_router_config_proto protoreflect.FileDescriptor
|
var File_app_router_config_proto protoreflect.FileDescriptor
|
||||||
|
|
||||||
const file_app_router_config_proto_rawDesc = "" +
|
const file_app_router_config_proto_rawDesc = "" +
|
||||||
@@ -708,12 +699,11 @@ const file_app_router_config_proto_rawDesc = "" +
|
|||||||
"\tbaselines\x18\x03 \x03(\x03R\tbaselines\x12\x1a\n" +
|
"\tbaselines\x18\x03 \x03(\x03R\tbaselines\x12\x1a\n" +
|
||||||
"\bexpected\x18\x04 \x01(\x05R\bexpected\x12\x16\n" +
|
"\bexpected\x18\x04 \x01(\x05R\bexpected\x12\x16\n" +
|
||||||
"\x06maxRTT\x18\x05 \x01(\x03R\x06maxRTT\x12\x1c\n" +
|
"\x06maxRTT\x18\x05 \x01(\x03R\x06maxRTT\x12\x1c\n" +
|
||||||
"\ttolerance\x18\x06 \x01(\x02R\ttolerance\"\xae\x02\n" +
|
"\ttolerance\x18\x06 \x01(\x02R\ttolerance\"\x96\x02\n" +
|
||||||
"\x06Config\x12O\n" +
|
"\x06Config\x12O\n" +
|
||||||
"\x0fdomain_strategy\x18\x01 \x01(\x0e2&.xray.app.router.Config.DomainStrategyR\x0edomainStrategy\x120\n" +
|
"\x0fdomain_strategy\x18\x01 \x01(\x0e2&.xray.app.router.Config.DomainStrategyR\x0edomainStrategy\x120\n" +
|
||||||
"\x04rule\x18\x02 \x03(\v2\x1c.xray.app.router.RoutingRuleR\x04rule\x12E\n" +
|
"\x04rule\x18\x02 \x03(\v2\x1c.xray.app.router.RoutingRuleR\x04rule\x12E\n" +
|
||||||
"\x0ebalancing_rule\x18\x03 \x03(\v2\x1e.xray.app.router.BalancingRuleR\rbalancingRule\x12\x16\n" +
|
"\x0ebalancing_rule\x18\x03 \x03(\v2\x1e.xray.app.router.BalancingRuleR\rbalancingRule\"B\n" +
|
||||||
"\x06script\x18\x04 \x01(\tR\x06script\"B\n" +
|
|
||||||
"\x0eDomainStrategy\x12\b\n" +
|
"\x0eDomainStrategy\x12\b\n" +
|
||||||
"\x04AsIs\x10\x00\x12\x10\n" +
|
"\x04AsIs\x10\x00\x12\x10\n" +
|
||||||
"\fIpIfNonMatch\x10\x02\x12\x0e\n" +
|
"\fIpIfNonMatch\x10\x02\x12\x0e\n" +
|
||||||
|
|||||||
@@ -110,6 +110,4 @@ message Config {
|
|||||||
DomainStrategy domain_strategy = 1;
|
DomainStrategy domain_strategy = 1;
|
||||||
repeated RoutingRule rule = 2;
|
repeated RoutingRule rule = 2;
|
||||||
repeated BalancingRule balancing_rule = 3;
|
repeated BalancingRule balancing_rule = 3;
|
||||||
// Absolute path to the Lua routing script.
|
|
||||||
string script = 4;
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,206 +0,0 @@
|
|||||||
package router
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"runtime"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/features/routing"
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
luaContextType = "xray.router.Context"
|
|
||||||
luaAttributesType = "xray.router.Attributes"
|
|
||||||
)
|
|
||||||
|
|
||||||
// RegisterLua makes xray.router available to routing scripts.
|
|
||||||
func (r *Router) RegisterLua(L *lua.LState) {
|
|
||||||
registerLuaContext(L)
|
|
||||||
|
|
||||||
L.PreloadModule("xray.router", func(L *lua.LState) int {
|
|
||||||
module := L.NewTable()
|
|
||||||
|
|
||||||
module.RawSetString("NetworkUnknown", lua.LNumber(net.Network_Unknown))
|
|
||||||
module.RawSetString("NetworkTCP", lua.LNumber(net.Network_TCP))
|
|
||||||
module.RawSetString("NetworkUDP", lua.LNumber(net.Network_UDP))
|
|
||||||
module.RawSetString("NetworkUNIX", lua.LNumber(net.Network_UNIX))
|
|
||||||
module.RawSetString("LocalOS", lua.LString(runtime.GOOS))
|
|
||||||
|
|
||||||
module.RawSetString("PickOutbound", L.NewFunction(func(L *lua.LState) int {
|
|
||||||
tag, ok := L.Get(2).(lua.LString)
|
|
||||||
if !ok {
|
|
||||||
L.ArgError(2, "balancer tag must be a string")
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
balancer, found := (*r.balancers.Load())[string(tag)]
|
|
||||||
if !found {
|
|
||||||
L.Push(lua.LNil)
|
|
||||||
pushLuaError(L, errors.New("balancer ", tag, " not found"))
|
|
||||||
return 2
|
|
||||||
}
|
|
||||||
outboundTag, err := balancer.PickOutbound()
|
|
||||||
L.Push(lua.LString(outboundTag))
|
|
||||||
pushLuaError(L, err)
|
|
||||||
return 2
|
|
||||||
}))
|
|
||||||
|
|
||||||
module.RawSetString("FindProcess", L.NewFunction(func(L *lua.LState) int {
|
|
||||||
pid, name, path, err := findProcess(checkLuaContext(L), net.FindProcess)
|
|
||||||
L.Push(lua.LNumber(pid))
|
|
||||||
L.Push(lua.LString(name))
|
|
||||||
L.Push(lua.LString(path))
|
|
||||||
pushLuaError(L, err)
|
|
||||||
return 4
|
|
||||||
}))
|
|
||||||
|
|
||||||
L.Push(module)
|
|
||||||
return 1
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func registerLuaContext(L *lua.LState) {
|
|
||||||
attributes := L.NewTypeMetatable(luaAttributesType)
|
|
||||||
L.SetField(attributes, "__index", L.NewFunction(func(L *lua.LState) int {
|
|
||||||
values := L.CheckUserData(1).Value.(map[string]string)
|
|
||||||
key := L.CheckString(2)
|
|
||||||
if value, found := values[key]; found {
|
|
||||||
L.Push(lua.LString(value))
|
|
||||||
} else {
|
|
||||||
L.Push(lua.LNil)
|
|
||||||
}
|
|
||||||
return 1
|
|
||||||
}))
|
|
||||||
methods := L.NewTable()
|
|
||||||
L.SetFuncs(methods, map[string]lua.LGFunction{
|
|
||||||
"GetSourceIPs": func(L *lua.LState) int {
|
|
||||||
return pushLuaIPs(L, checkLuaContext(L).GetSourceIPs())
|
|
||||||
},
|
|
||||||
"GetTargetIPs": func(L *lua.LState) int {
|
|
||||||
return pushLuaIPs(L, checkLuaContext(L).GetTargetIPs())
|
|
||||||
},
|
|
||||||
"GetLocalIPs": func(L *lua.LState) int {
|
|
||||||
return pushLuaIPs(L, checkLuaContext(L).GetLocalIPs())
|
|
||||||
},
|
|
||||||
"GetAttributes": func(L *lua.LState) int {
|
|
||||||
values := L.NewUserData()
|
|
||||||
values.Value = checkLuaContext(L).GetAttributes()
|
|
||||||
L.SetMetatable(values, attributes)
|
|
||||||
L.Push(values)
|
|
||||||
return 1
|
|
||||||
},
|
|
||||||
})
|
|
||||||
L.SetField(L.NewTypeMetatable(luaContextType), "__index", methods)
|
|
||||||
}
|
|
||||||
|
|
||||||
func checkLuaContext(L *lua.LState) routing.Context {
|
|
||||||
ctx, ok := L.CheckUserData(1).Value.(routing.Context)
|
|
||||||
if !ok {
|
|
||||||
L.ArgError(1, "routing context expected")
|
|
||||||
}
|
|
||||||
return ctx
|
|
||||||
}
|
|
||||||
|
|
||||||
func pushLuaIPs(L *lua.LState, ips []net.IP) int {
|
|
||||||
addresses := L.NewUserData()
|
|
||||||
addresses.Value = ips
|
|
||||||
L.Push(addresses)
|
|
||||||
return 1
|
|
||||||
}
|
|
||||||
|
|
||||||
func pushLuaError(L *lua.LState, err error) {
|
|
||||||
if err == nil {
|
|
||||||
L.Push(lua.LNil)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
value := L.NewUserData()
|
|
||||||
value.Value = err
|
|
||||||
L.Push(value)
|
|
||||||
}
|
|
||||||
|
|
||||||
// CallLuaHook invokes HandleRoute in the supplied state.
|
|
||||||
func (r *Router) CallLuaHook(L *lua.LState, ctx context.Context, routeCtx routing.Context) (string, string, error) {
|
|
||||||
previous, top := L.Context(), L.GetTop()
|
|
||||||
L.SetContext(ctx)
|
|
||||||
defer func() {
|
|
||||||
L.SetTop(top)
|
|
||||||
if previous == nil {
|
|
||||||
L.RemoveContext()
|
|
||||||
} else {
|
|
||||||
L.SetContext(previous)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
fn := L.GetGlobal("HandleRoute")
|
|
||||||
if fn.Type() != lua.LTFunction {
|
|
||||||
return "", "", errors.New("routing script must define HandleRoute(...)")
|
|
||||||
}
|
|
||||||
value := L.NewUserData()
|
|
||||||
value.Value = routeCtx
|
|
||||||
L.SetMetatable(value, L.GetTypeMetatable(luaContextType))
|
|
||||||
if err := L.CallByParam(lua.P{Fn: fn, NRet: 3, Protect: true},
|
|
||||||
value, lua.LString(routeCtx.GetInboundTag()), lua.LNumber(routeCtx.GetSourcePort()),
|
|
||||||
lua.LNumber(routeCtx.GetTargetPort()), lua.LNumber(routeCtx.GetLocalPort()),
|
|
||||||
lua.LString(routeCtx.GetTargetDomain()), lua.LNumber(routeCtx.GetNetwork()),
|
|
||||||
lua.LString(routeCtx.GetProtocol()), lua.LString(routeCtx.GetUser()),
|
|
||||||
lua.LNumber(routeCtx.GetVlessRoute()), lua.LBool(routeCtx.GetSkipDNSResolve())); err != nil {
|
|
||||||
return "", "", err
|
|
||||||
}
|
|
||||||
return readLuaRouteResult(L.Get(-3), L.Get(-2), L.Get(-1))
|
|
||||||
}
|
|
||||||
|
|
||||||
func readLuaRouteResult(tagValue, ruleValue, errorValue lua.LValue) (string, string, error) {
|
|
||||||
if errorValue != lua.LNil {
|
|
||||||
if value, ok := errorValue.(*lua.LUserData); ok {
|
|
||||||
if err, ok := value.Value.(error); ok {
|
|
||||||
return "", "", err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if value, ok := errorValue.(lua.LString); ok {
|
|
||||||
return "", "", errors.New(string(value))
|
|
||||||
}
|
|
||||||
return "", "", errors.New("routing script error must be an error or string")
|
|
||||||
}
|
|
||||||
if tagValue == lua.LNil {
|
|
||||||
return "", "", nil
|
|
||||||
}
|
|
||||||
tag, ok := tagValue.(lua.LString)
|
|
||||||
if !ok {
|
|
||||||
return "", "", errors.New("routing script outboundTag must be a string or nil")
|
|
||||||
}
|
|
||||||
if tag == "" {
|
|
||||||
return "", "", nil
|
|
||||||
}
|
|
||||||
var ruleTag string
|
|
||||||
if ruleValue != lua.LNil {
|
|
||||||
value, ok := ruleValue.(lua.LString)
|
|
||||||
if !ok {
|
|
||||||
return "", "", errors.New("routing script ruleTag must be a string")
|
|
||||||
}
|
|
||||||
ruleTag = string(value)
|
|
||||||
}
|
|
||||||
return string(tag), ruleTag, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type processFinder func(string, string, uint16, string, uint16) (int, string, string, error)
|
|
||||||
|
|
||||||
func findProcess(ctx routing.Context, finder processFinder) (int, string, string, error) {
|
|
||||||
sources := ctx.GetSourceIPs()
|
|
||||||
if len(sources) == 0 {
|
|
||||||
return 0, "", "", errors.New("process lookup requires a source IP")
|
|
||||||
}
|
|
||||||
var network string
|
|
||||||
switch ctx.GetNetwork() {
|
|
||||||
case net.Network_TCP:
|
|
||||||
network = "tcp"
|
|
||||||
case net.Network_UDP:
|
|
||||||
network = "udp"
|
|
||||||
default:
|
|
||||||
return 0, "", "", errors.New("process lookup requires TCP or UDP")
|
|
||||||
}
|
|
||||||
targetIP, targetPort := "", uint16(0)
|
|
||||||
if targets := ctx.GetTargetIPs(); len(targets) > 0 {
|
|
||||||
targetIP, targetPort = targets[0].String(), uint16(ctx.GetTargetPort())
|
|
||||||
}
|
|
||||||
return finder(network, sources[0].String(), uint16(ctx.GetSourcePort()), targetIP, targetPort)
|
|
||||||
}
|
|
||||||
@@ -1,292 +0,0 @@
|
|||||||
package router
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
go_errors "errors"
|
|
||||||
"runtime"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/geodata"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/common/protocol"
|
|
||||||
"github.com/xtls/xray-core/common/session"
|
|
||||||
"github.com/xtls/xray-core/features/routing"
|
|
||||||
routing_session "github.com/xtls/xray-core/features/routing/session"
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
type luaRouteTestContext struct {
|
|
||||||
*routing_session.Context
|
|
||||||
sourceIPs, targetIPs, localIPs []net.IP
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *luaRouteTestContext) GetSourceIPs() []net.IP { return c.sourceIPs }
|
|
||||||
func (c *luaRouteTestContext) GetTargetIPs() []net.IP { return c.targetIPs }
|
|
||||||
func (c *luaRouteTestContext) GetLocalIPs() []net.IP { return c.localIPs }
|
|
||||||
|
|
||||||
func newLuaRouteTestContext() *luaRouteTestContext {
|
|
||||||
return &luaRouteTestContext{
|
|
||||||
Context: &routing_session.Context{
|
|
||||||
Inbound: &session.Inbound{
|
|
||||||
Tag: "in", VlessRoute: 4321,
|
|
||||||
Source: net.TCPDestination(net.LocalHostIP, 1234),
|
|
||||||
Local: net.TCPDestination(net.LocalHostIP, 5678),
|
|
||||||
User: &protocol.MemoryUser{Email: "user@example.com"},
|
|
||||||
},
|
|
||||||
Outbound: &session.Outbound{
|
|
||||||
Target: net.TCPDestination(net.LocalHostIP, 443),
|
|
||||||
RouteTarget: net.TCPDestination(net.DomainAddress("MiXeD.Example."), 443),
|
|
||||||
},
|
|
||||||
Content: &session.Content{
|
|
||||||
Protocol: "tls", Attributes: map[string]string{"key": "value"}, SkipDNSResolve: true,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
sourceIPs: []net.IP{{127, 0, 0, 2}},
|
|
||||||
targetIPs: []net.IP{{127, 0, 0, 3}},
|
|
||||||
localIPs: []net.IP{{127, 0, 0, 1}},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func newLuaRouterState(t *testing.T, script string) (*Router, *lua.LState) {
|
|
||||||
t.Helper()
|
|
||||||
r := new(Router)
|
|
||||||
if err := r.Init(context.Background(), &Config{}, nil, nil, nil); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
L := lua.NewState()
|
|
||||||
t.Cleanup(L.Close)
|
|
||||||
r.RegisterLua(L)
|
|
||||||
geodata.RegisterLua(L)
|
|
||||||
if err := L.DoString(script); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
return r, L
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaRouteBinding(t *testing.T) {
|
|
||||||
r, L := newLuaRouterState(t, `
|
|
||||||
local router = require("xray.router")
|
|
||||||
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.0/8")
|
|
||||||
assert(router.NetworkUnknown == 0 and router.NetworkTCP == 2)
|
|
||||||
assert(router.NetworkUDP == 3 and router.NetworkUNIX == 4)
|
|
||||||
assert(router.BuildIPMatcher == nil and router.BuildDomainMatcher == nil)
|
|
||||||
function HandleRoute(ctx, inboundTag, sourcePort, targetPort, localPort,
|
|
||||||
targetDomain, network, protocol, user, vlessRoute, skipDNSResolve, ...)
|
|
||||||
assert(select("#", ...) == 0)
|
|
||||||
assert(inboundTag == "in" and sourcePort == 1234 and targetPort == 443 and localPort == 5678)
|
|
||||||
assert(targetDomain == "MiXeD.Example." and network == router.NetworkTCP)
|
|
||||||
assert(protocol == "tls" and user == "user@example.com" and vlessRoute == 4321 and skipDNSResolve)
|
|
||||||
assert(ctx.GetNetwork == nil and ctx.Context == nil)
|
|
||||||
savedContext = ctx
|
|
||||||
sourceIPs, targetIPs, localIPs = ctx:GetSourceIPs(), ctx:GetTargetIPs(), ctx:GetLocalIPs()
|
|
||||||
attributes = ctx:GetAttributes()
|
|
||||||
assert(matcher:AnyMatch(sourceIPs) and matcher:AnyMatch(targetIPs) and matcher:AnyMatch(localIPs))
|
|
||||||
assert(attributes.key == "value" and attributes.missing == nil)
|
|
||||||
assert(not pcall(function() attributes.key = "changed" end))
|
|
||||||
return "out", "rule"
|
|
||||||
end`)
|
|
||||||
|
|
||||||
ctx := newLuaRouteTestContext()
|
|
||||||
tag, rule, err := r.CallLuaHook(L, context.Background(), ctx)
|
|
||||||
if err != nil || tag != "out" || rule != "rule" {
|
|
||||||
t.Fatalf("hook = %q, %q, %v", tag, rule, err)
|
|
||||||
}
|
|
||||||
if L.GetGlobal("savedContext").(*lua.LUserData).Value != ctx {
|
|
||||||
t.Fatal("routing context was copied")
|
|
||||||
}
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name string
|
|
||||||
want []net.IP
|
|
||||||
}{
|
|
||||||
{"sourceIPs", ctx.sourceIPs},
|
|
||||||
{"targetIPs", ctx.targetIPs},
|
|
||||||
{"localIPs", ctx.localIPs},
|
|
||||||
} {
|
|
||||||
got := L.GetGlobal(tc.name).(*lua.LUserData).Value.([]net.IP)
|
|
||||||
if &got[0] != &tc.want[0] {
|
|
||||||
t.Fatalf("%s storage was copied", tc.name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
ctx.Content.Attributes["key"] = "updated"
|
|
||||||
L.SetGlobal("expectedOS", lua.LString(runtime.GOOS))
|
|
||||||
if err := L.DoString(`
|
|
||||||
assert(attributes.key == "updated")
|
|
||||||
assert(require("xray.router").LocalOS == expectedOS)`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaRouteResult(t *testing.T) {
|
|
||||||
nativeErr := go_errors.New("native failure")
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name, body, tag, rule, wantErr string
|
|
||||||
native bool
|
|
||||||
}{
|
|
||||||
{name: "route", body: `return "out", "rule"`, tag: "out", rule: "rule"},
|
|
||||||
{name: "no match", body: `return nil`},
|
|
||||||
{name: "empty tag", body: `return ""`},
|
|
||||||
{name: "invalid tag", body: `return 1`, wantErr: "outboundTag"},
|
|
||||||
{name: "invalid rule", body: `return "out", false`, wantErr: "ruleTag"},
|
|
||||||
{name: "string error", body: `return nil, nil, "script failure"`, wantErr: "script failure"},
|
|
||||||
{name: "native error", body: `return nil, nil, nativeError`, native: true},
|
|
||||||
{name: "runtime error", body: `error("runtime failure")`, wantErr: "runtime failure"},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
r, L := newLuaRouterState(t, "function HandleRoute() "+tc.body+" end")
|
|
||||||
value := L.NewUserData()
|
|
||||||
value.Value = nativeErr
|
|
||||||
L.SetGlobal("nativeError", value)
|
|
||||||
previous := context.WithValue(context.Background(), struct{}{}, true)
|
|
||||||
L.SetContext(previous)
|
|
||||||
L.Push(lua.LTrue)
|
|
||||||
|
|
||||||
tag, rule, err := r.CallLuaHook(L, context.Background(), &routing_session.Context{})
|
|
||||||
if tag != tc.tag || rule != tc.rule {
|
|
||||||
t.Fatalf("result = %q, %q, %v", tag, rule, err)
|
|
||||||
}
|
|
||||||
switch {
|
|
||||||
case tc.native:
|
|
||||||
if err != nativeErr {
|
|
||||||
t.Fatalf("error = %v, want original error", err)
|
|
||||||
}
|
|
||||||
case tc.wantErr != "":
|
|
||||||
if err == nil || !strings.Contains(err.Error(), tc.wantErr) {
|
|
||||||
t.Fatalf("error = %v, want %q", err, tc.wantErr)
|
|
||||||
}
|
|
||||||
case err != nil:
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if L.Context() != previous || L.GetTop() != 1 || L.Get(1) != lua.LTrue {
|
|
||||||
t.Fatal("hook did not restore the previous context and stack")
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaRouteCancellation(t *testing.T) {
|
|
||||||
r, L := newLuaRouterState(t, `function HandleRoute() while true do end end`)
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
|
||||||
cancel()
|
|
||||||
if _, _, err := r.CallLuaHook(L, ctx, &routing_session.Context{}); err == nil {
|
|
||||||
t.Fatal("CallLuaHook did not stop after context cancellation")
|
|
||||||
}
|
|
||||||
if L.Context() != nil || L.GetTop() != 0 {
|
|
||||||
t.Fatal("CallLuaHook did not restore the Lua state")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFindProcess(t *testing.T) {
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name, network, target string
|
|
||||||
targetPort uint16
|
|
||||||
modify func(*luaRouteTestContext)
|
|
||||||
wantErr bool
|
|
||||||
}{
|
|
||||||
{name: "TCP", network: "tcp", target: "127.0.0.3", targetPort: 443},
|
|
||||||
{name: "UDP", network: "udp", target: "127.0.0.3", targetPort: 443, modify: func(c *luaRouteTestContext) {
|
|
||||||
c.Outbound.Target.Network = net.Network_UDP
|
|
||||||
}},
|
|
||||||
{name: "domain target", network: "tcp", modify: func(c *luaRouteTestContext) { c.targetIPs = nil }},
|
|
||||||
{name: "missing source", modify: func(c *luaRouteTestContext) { c.sourceIPs = nil }, wantErr: true},
|
|
||||||
{name: "unsupported network", modify: func(c *luaRouteTestContext) {
|
|
||||||
c.Outbound.Target.Network = net.Network_UNIX
|
|
||||||
}, wantErr: true},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
ctx := newLuaRouteTestContext()
|
|
||||||
if tc.modify != nil {
|
|
||||||
tc.modify(ctx)
|
|
||||||
}
|
|
||||||
called := false
|
|
||||||
pid, name, path, err := findProcess(ctx, func(network, source string, sourcePort uint16, target string, targetPort uint16) (int, string, string, error) {
|
|
||||||
called = true
|
|
||||||
if network != tc.network || source != "127.0.0.2" || sourcePort != 1234 || target != tc.target || targetPort != tc.targetPort {
|
|
||||||
t.Fatalf("endpoints = %s %s:%d -> %s:%d", network, source, sourcePort, target, targetPort)
|
|
||||||
}
|
|
||||||
return 42, "process", "/path/process", nil
|
|
||||||
})
|
|
||||||
if tc.wantErr {
|
|
||||||
if err == nil || called {
|
|
||||||
t.Fatalf("findProcess = %d, %q, %q, %v", pid, name, path, err)
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if err != nil || !called || pid != 42 || name != "process" || path != "/path/process" {
|
|
||||||
t.Fatalf("findProcess = %d, %q, %q, %v", pid, name, path, err)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// BenchmarkLuaRouteHookCall isolates a preloaded Lua hook and its routing context bridge.
|
|
||||||
// The direct case runs an equivalent native routing rule.
|
|
||||||
func BenchmarkLuaRouteHookCall(b *testing.B) {
|
|
||||||
r := new(Router)
|
|
||||||
if err := r.Init(context.Background(), &Config{Rule: []*RoutingRule{{
|
|
||||||
TargetTag: &RoutingRule_Tag{Tag: "out"},
|
|
||||||
RuleTag: "rule",
|
|
||||||
InboundTag: []string{"in"},
|
|
||||||
Networks: []net.Network{net.Network_TCP},
|
|
||||||
Ip: []*geodata.IPRule{{
|
|
||||||
Value: &geodata.IPRule_Custom{Custom: &geodata.CIDRRule{
|
|
||||||
Cidr: &geodata.CIDR{Ip: []byte{127, 0, 0, 0}, Prefix: 8},
|
|
||||||
}},
|
|
||||||
}},
|
|
||||||
}}}, nil, nil, nil); err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
r.RegisterLua(L)
|
|
||||||
geodata.RegisterLua(L)
|
|
||||||
if err := L.DoString(`
|
|
||||||
local router = require("xray.router")
|
|
||||||
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.0/8")
|
|
||||||
function HandleRoute(ctx, inboundTag, sourcePort, targetPort, localPort,
|
|
||||||
targetDomain, network, protocol, user, vlessRoute, skipDNSResolve)
|
|
||||||
if inboundTag == "in" and network == router.NetworkTCP and matcher:AnyMatch(ctx:GetTargetIPs()) then
|
|
||||||
return "out", "rule"
|
|
||||||
end
|
|
||||||
end
|
|
||||||
`); err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
routeCtx := newLuaRouteTestContext()
|
|
||||||
for _, benchmark := range []struct {
|
|
||||||
name string
|
|
||||||
route func() (string, string, error)
|
|
||||||
}{
|
|
||||||
{"direct", func() (string, string, error) {
|
|
||||||
route, err := r.PickRoute(routeCtx)
|
|
||||||
if err != nil {
|
|
||||||
return "", "", err
|
|
||||||
}
|
|
||||||
return route.GetOutboundTag(), route.GetRuleTag(), nil
|
|
||||||
}},
|
|
||||||
{"lua_hook", func() (string, string, error) {
|
|
||||||
return r.CallLuaHook(L, ctx, routeCtx)
|
|
||||||
}},
|
|
||||||
} {
|
|
||||||
b.Run(benchmark.name, func(b *testing.B) {
|
|
||||||
b.ReportAllocs()
|
|
||||||
b.ResetTimer()
|
|
||||||
var tag, rule string
|
|
||||||
var err error
|
|
||||||
for i := 0; i < b.N; i++ {
|
|
||||||
tag, rule, err = benchmark.route()
|
|
||||||
if err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
b.StopTimer()
|
|
||||||
if tag != "out" || rule != "rule" {
|
|
||||||
b.Fatalf("route() = %q, %q; want out, rule", tag, rule)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var _ routing.Context = (*luaRouteTestContext)(nil)
|
|
||||||
@@ -20,8 +20,6 @@ import (
|
|||||||
type Router struct {
|
type Router struct {
|
||||||
domainStrategy Config_DomainStrategy
|
domainStrategy Config_DomainStrategy
|
||||||
rules atomic.Pointer[[]*Rule]
|
rules atomic.Pointer[[]*Rule]
|
||||||
scriptPath string
|
|
||||||
script *scriptEngine
|
|
||||||
balancers atomic.Pointer[map[string]*Balancer]
|
balancers atomic.Pointer[map[string]*Balancer]
|
||||||
dns dns.Client
|
dns dns.Client
|
||||||
|
|
||||||
@@ -42,7 +40,6 @@ type Route struct {
|
|||||||
// Init initializes the Router.
|
// Init initializes the Router.
|
||||||
func (r *Router) Init(ctx context.Context, config *Config, d dns.Client, ohm outbound.Manager, dispatcher routing.Dispatcher) error {
|
func (r *Router) Init(ctx context.Context, config *Config, d dns.Client, ohm outbound.Manager, dispatcher routing.Dispatcher) error {
|
||||||
r.domainStrategy = config.DomainStrategy
|
r.domainStrategy = config.DomainStrategy
|
||||||
r.scriptPath = config.Script
|
|
||||||
r.dns = d
|
r.dns = d
|
||||||
r.ctx = ctx
|
r.ctx = ctx
|
||||||
r.ohm = ohm
|
r.ohm = ohm
|
||||||
@@ -55,10 +52,6 @@ func (r *Router) Init(ctx context.Context, config *Config, d dns.Client, ohm out
|
|||||||
|
|
||||||
// PickRoute implements routing.Router.
|
// PickRoute implements routing.Router.
|
||||||
func (r *Router) PickRoute(ctx routing.Context) (routing.Route, error) {
|
func (r *Router) PickRoute(ctx routing.Context) (routing.Route, error) {
|
||||||
if r.script != nil {
|
|
||||||
return r.script.pickRoute(ctx)
|
|
||||||
}
|
|
||||||
|
|
||||||
originalCtx := ctx
|
originalCtx := ctx
|
||||||
rule, ctx, err := r.pickRouteInternal(ctx)
|
rule, ctx, err := r.pickRouteInternal(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -228,13 +221,6 @@ func (r *Router) pickRouteInternal(ctx routing.Context) (*Rule, routing.Context,
|
|||||||
|
|
||||||
// Start implements common.Runnable.
|
// Start implements common.Runnable.
|
||||||
func (r *Router) Start() error {
|
func (r *Router) Start() error {
|
||||||
if r.scriptPath != "" {
|
|
||||||
engine, err := newScriptEngine(r.scriptPath, r)
|
|
||||||
if err != nil {
|
|
||||||
return errors.New("failed to initialize routing script").Base(err)
|
|
||||||
}
|
|
||||||
r.script = engine
|
|
||||||
}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -249,9 +235,6 @@ func closeWebhooks(rules []*Rule) {
|
|||||||
|
|
||||||
// Close implements common.Closable.
|
// Close implements common.Closable.
|
||||||
func (r *Router) Close() error {
|
func (r *Router) Close() error {
|
||||||
if r.script != nil {
|
|
||||||
r.script.close()
|
|
||||||
}
|
|
||||||
r.mu.Lock()
|
r.mu.Lock()
|
||||||
defer r.mu.Unlock()
|
defer r.mu.Unlock()
|
||||||
closeWebhooks(*r.rules.Load())
|
closeWebhooks(*r.rules.Load())
|
||||||
|
|||||||
@@ -1,79 +0,0 @@
|
|||||||
package router
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/app/dns"
|
|
||||||
"github.com/xtls/xray-core/common"
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/geodata"
|
|
||||||
"github.com/xtls/xray-core/common/log"
|
|
||||||
luamgr "github.com/xtls/xray-core/common/lua"
|
|
||||||
"github.com/xtls/xray-core/features/routing"
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
const scriptExecutionTimeout = 10 * time.Second
|
|
||||||
|
|
||||||
type scriptEngine struct {
|
|
||||||
router *Router
|
|
||||||
pool *luamgr.Pool
|
|
||||||
}
|
|
||||||
|
|
||||||
func newScriptEngine(path string, router *Router) (*scriptEngine, error) {
|
|
||||||
program, err := luamgr.CompileFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
e := &scriptEngine{router: router}
|
|
||||||
e.pool, err = luamgr.NewPool(router.ctx, func(poolCtx context.Context) (*lua.LState, error) {
|
|
||||||
initCtx, cancel := context.WithTimeout(poolCtx, scriptExecutionTimeout)
|
|
||||||
defer cancel()
|
|
||||||
L, err := program.NewState(initCtx, func(L *lua.LState) {
|
|
||||||
geodata.RegisterLua(L)
|
|
||||||
log.RegisterLua(L)
|
|
||||||
router.RegisterLua(L)
|
|
||||||
dns.RegisterLua(L, router.dns)
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if L.GetGlobal("HandleRoute").Type() != lua.LTFunction {
|
|
||||||
L.Close()
|
|
||||||
return nil, errors.New("routing script must define HandleRoute(...)")
|
|
||||||
}
|
|
||||||
return L, nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
errors.LogInfo(router.ctx, "routing script initialized from ", path)
|
|
||||||
return e, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e *scriptEngine) close() {
|
|
||||||
e.pool.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e *scriptEngine) pickRoute(ctx routing.Context) (routing.Route, error) {
|
|
||||||
L, err := e.pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
reusable := false
|
|
||||||
defer func() {
|
|
||||||
e.pool.Release(L, reusable)
|
|
||||||
}()
|
|
||||||
callCtx, cancel := context.WithTimeout(e.pool.Context(), scriptExecutionTimeout)
|
|
||||||
defer cancel()
|
|
||||||
tag, ruleTag, err := e.router.CallLuaHook(L, callCtx, ctx)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
reusable = true
|
|
||||||
if tag == "" {
|
|
||||||
return nil, common.ErrNoClue
|
|
||||||
}
|
|
||||||
return &Route{Context: ctx, outboundTag: tag, ruleTag: ruleTag}, nil
|
|
||||||
}
|
|
||||||
@@ -1,372 +0,0 @@
|
|||||||
package router
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
stdnet "net"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"sync"
|
|
||||||
"sync/atomic"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
wireDNS "github.com/miekg/dns"
|
|
||||||
"github.com/xtls/xray-core/app/dispatcher"
|
|
||||||
appdns "github.com/xtls/xray-core/app/dns"
|
|
||||||
"github.com/xtls/xray-core/app/proxyman"
|
|
||||||
_ "github.com/xtls/xray-core/app/proxyman/outbound"
|
|
||||||
"github.com/xtls/xray-core/common"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/common/serial"
|
|
||||||
"github.com/xtls/xray-core/core"
|
|
||||||
featureDNS "github.com/xtls/xray-core/features/dns"
|
|
||||||
"github.com/xtls/xray-core/features/outbound"
|
|
||||||
"github.com/xtls/xray-core/features/routing"
|
|
||||||
routing_session "github.com/xtls/xray-core/features/routing/session"
|
|
||||||
"github.com/xtls/xray-core/proxy/blackhole"
|
|
||||||
"github.com/xtls/xray-core/proxy/freedom"
|
|
||||||
)
|
|
||||||
|
|
||||||
type luaRouteDNSClient struct {
|
|
||||||
featureDNS.Client
|
|
||||||
lookup func(string, featureDNS.IPOption) ([]net.IP, uint32, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *luaRouteDNSClient) LookupIP(domain string, option featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
return d.lookup(domain, option)
|
|
||||||
}
|
|
||||||
|
|
||||||
type luaRouteOutboundManager struct{ outbound.Manager }
|
|
||||||
|
|
||||||
func (*luaRouteOutboundManager) Select(selectors []string) []string { return selectors }
|
|
||||||
|
|
||||||
func writeRouteScript(t *testing.T, script string) string {
|
|
||||||
t.Helper()
|
|
||||||
path := filepath.Join(t.TempDir(), "route.lua")
|
|
||||||
if err := os.WriteFile(path, []byte(script), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
return path
|
|
||||||
}
|
|
||||||
|
|
||||||
func startLuaRouter(t *testing.T, script string, d featureDNS.Client, config *Config) *Router {
|
|
||||||
t.Helper()
|
|
||||||
if config == nil {
|
|
||||||
config = &Config{}
|
|
||||||
}
|
|
||||||
config.Script = writeRouteScript(t, script)
|
|
||||||
r := new(Router)
|
|
||||||
if err := r.Init(context.Background(), config, d, &luaRouteOutboundManager{}, nil); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if err := r.Start(); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
t.Cleanup(func() {
|
|
||||||
if err := r.Close(); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
return r
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRouterScriptStartup(t *testing.T) {
|
|
||||||
for _, tc := range []struct{ name, script string }{
|
|
||||||
{"syntax error", "function HandleRoute("},
|
|
||||||
{"missing hook", "value = 1"},
|
|
||||||
{"initialization error", `error("setup failed")`},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
r := new(Router)
|
|
||||||
if err := r.Init(context.Background(), &Config{Script: writeRouteScript(t, tc.script)}, nil, nil, nil); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer r.Close()
|
|
||||||
if err := r.Start(); err == nil {
|
|
||||||
t.Fatal("Start accepted an invalid routing script")
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRouterScriptRouting(t *testing.T) {
|
|
||||||
var dnsCalls atomic.Int32
|
|
||||||
d := &luaRouteDNSClient{lookup: func(string, featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
dnsCalls.Add(1)
|
|
||||||
return []net.IP{{1, 2, 3, 4}}, 60, nil
|
|
||||||
}}
|
|
||||||
r := startLuaRouter(t, `
|
|
||||||
function HandleRoute(ctx, inbound)
|
|
||||||
if inbound == "miss" then return nil end
|
|
||||||
return "lua-out", "lua-rule"
|
|
||||||
end`, d, &Config{
|
|
||||||
DomainStrategy: Config_IpOnDemand,
|
|
||||||
Rule: []*RoutingRule{{
|
|
||||||
TargetTag: &RoutingRule_Tag{Tag: "json-out"},
|
|
||||||
Networks: []net.Network{net.Network_TCP},
|
|
||||||
}},
|
|
||||||
})
|
|
||||||
ctx := newLuaRouteTestContext()
|
|
||||||
ctx.Content.SkipDNSResolve = false
|
|
||||||
route, err := r.PickRoute(ctx)
|
|
||||||
if err != nil || route.GetOutboundTag() != "lua-out" || route.GetRuleTag() != "lua-rule" || route.(*Route).Context != ctx {
|
|
||||||
t.Fatalf("route = %v, %v", route, err)
|
|
||||||
}
|
|
||||||
ctx.Inbound.Tag = "miss"
|
|
||||||
if route, err := r.PickRoute(ctx); route != nil || err != common.ErrNoClue {
|
|
||||||
t.Fatalf("miss = %v, %v", route, err)
|
|
||||||
}
|
|
||||||
if dnsCalls.Load() != 0 {
|
|
||||||
t.Fatal("script routing implicitly resolved DNS")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRouterScriptModules(t *testing.T) {
|
|
||||||
ips := []net.IP{{127, 0, 0, 7}}
|
|
||||||
calls := 0
|
|
||||||
d := &luaRouteDNSClient{lookup: func(domain string, option featureDNS.IPOption) ([]net.IP, uint32, error) {
|
|
||||||
calls++
|
|
||||||
if domain != "MiXeD.Example." || !option.IPv4Enable || option.IPv6Enable || !option.FakeEnable {
|
|
||||||
t.Fatalf("dns.Query arguments = %q, %+v", domain, option)
|
|
||||||
}
|
|
||||||
return ips, 17, nil
|
|
||||||
}}
|
|
||||||
r := startLuaRouter(t, `
|
|
||||||
local dns = require("xray.dns")
|
|
||||||
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.0/8")
|
|
||||||
assert(dns.Servers == nil and type(dns.Query) == "function")
|
|
||||||
assert(type(require("xray.log").Info) == "function")
|
|
||||||
function HandleRoute(ctx, inbound, sourcePort, targetPort, localPort, domain)
|
|
||||||
local ips, ttl, err = dns.Query(domain, true, false, true)
|
|
||||||
assert(not err and ttl == 17)
|
|
||||||
assert(matcher:AnyMatch(ips) and matcher:AnyMatch(ctx:GetTargetIPs()))
|
|
||||||
return "out"
|
|
||||||
end`, d, nil)
|
|
||||||
if _, err := r.PickRoute(newLuaRouteTestContext()); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if calls != 1 {
|
|
||||||
t.Fatalf("DNS calls = %d, want 1", calls)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRouterScriptBalancerReload(t *testing.T) {
|
|
||||||
config := func(tag string) *Config {
|
|
||||||
return &Config{BalancingRule: []*BalancingRule{{
|
|
||||||
Tag: "balance", Strategy: "roundrobin", OutboundSelector: []string{tag},
|
|
||||||
}}}
|
|
||||||
}
|
|
||||||
r := startLuaRouter(t, `
|
|
||||||
local router = require("xray.router")
|
|
||||||
function HandleRoute()
|
|
||||||
local tag, err = router:PickOutbound("balance")
|
|
||||||
return tag, "balanced", err
|
|
||||||
end`, nil, config("old"))
|
|
||||||
pick := func(want string) {
|
|
||||||
t.Helper()
|
|
||||||
route, err := r.PickRoute(&routing_session.Context{})
|
|
||||||
if err != nil || route.GetOutboundTag() != want || route.GetRuleTag() != "balanced" {
|
|
||||||
t.Fatalf("route = %v, %v, want %q", route, err, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pick("old")
|
|
||||||
if err := r.SetOverrideTarget("balance", "override"); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
pick("override")
|
|
||||||
if err := r.SetOverrideTarget("balance", ""); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if err := r.ReloadRules(config("new"), false); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
pick("new")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRouterScriptConcurrentBalancerReload(t *testing.T) {
|
|
||||||
config := func(tag string) *Config {
|
|
||||||
return &Config{BalancingRule: []*BalancingRule{{
|
|
||||||
Tag: "balance", Strategy: "roundrobin", OutboundSelector: []string{tag},
|
|
||||||
}}}
|
|
||||||
}
|
|
||||||
r := startLuaRouter(t, `
|
|
||||||
local router = require("xray.router")
|
|
||||||
function HandleRoute()
|
|
||||||
local tag, err = router:PickOutbound("balance")
|
|
||||||
return tag, nil, err
|
|
||||||
end`, nil, config("a"))
|
|
||||||
|
|
||||||
var wg sync.WaitGroup
|
|
||||||
for range 4 {
|
|
||||||
wg.Go(func() {
|
|
||||||
for range 20 {
|
|
||||||
route, err := r.PickRoute(&routing_session.Context{})
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("PickRoute: %v", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if tag := route.GetOutboundTag(); tag != "a" && tag != "b" {
|
|
||||||
t.Errorf("unexpected tag %q", tag)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
wg.Go(func() {
|
|
||||||
for range 20 {
|
|
||||||
for _, tag := range []string{"a", "b"} {
|
|
||||||
if err := r.ReloadRules(config(tag), false); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
wg.Wait()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRouterScriptStateReuse(t *testing.T) {
|
|
||||||
r := startLuaRouter(t, `
|
|
||||||
local calls = 0
|
|
||||||
function HandleRoute(ctx, inbound)
|
|
||||||
calls = calls + 1
|
|
||||||
if inbound == "miss" then return nil end
|
|
||||||
if inbound == "fail" then error("failed") end
|
|
||||||
return tostring(calls)
|
|
||||||
end`, nil, nil)
|
|
||||||
ctx := newLuaRouteTestContext()
|
|
||||||
pick := func(want string) {
|
|
||||||
t.Helper()
|
|
||||||
route, err := r.PickRoute(ctx)
|
|
||||||
if err != nil || route.GetOutboundTag() != want {
|
|
||||||
t.Fatalf("route = %v, %v, want %q", route, err, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pick("1")
|
|
||||||
ctx.Inbound.Tag = "miss"
|
|
||||||
if _, err := r.PickRoute(ctx); err != common.ErrNoClue {
|
|
||||||
t.Fatalf("miss = %v", err)
|
|
||||||
}
|
|
||||||
ctx.Inbound.Tag = "in"
|
|
||||||
pick("3")
|
|
||||||
ctx.Inbound.Tag = "fail"
|
|
||||||
if _, err := r.PickRoute(ctx); err == nil {
|
|
||||||
t.Fatal("script error was ignored")
|
|
||||||
}
|
|
||||||
ctx.Inbound.Tag = "in"
|
|
||||||
pick("1")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestRouterScriptDNSDispatcherReentry(t *testing.T) {
|
|
||||||
conn, err := stdnet.ListenPacket("udp4", "127.0.0.1:0")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
port := conn.LocalAddr().(*stdnet.UDPAddr).Port
|
|
||||||
ready, stopped := make(chan struct{}), make(chan error, 1)
|
|
||||||
var queries atomic.Int32
|
|
||||||
server := &wireDNS.Server{
|
|
||||||
PacketConn: conn,
|
|
||||||
NotifyStartedFunc: func() {
|
|
||||||
close(ready)
|
|
||||||
},
|
|
||||||
Handler: wireDNS.HandlerFunc(func(w wireDNS.ResponseWriter, query *wireDNS.Msg) {
|
|
||||||
queries.Add(1)
|
|
||||||
response := new(wireDNS.Msg).SetReply(query)
|
|
||||||
for _, question := range query.Question {
|
|
||||||
if question.Name == "nested.example." && question.Qtype == wireDNS.TypeA {
|
|
||||||
response.Answer = append(response.Answer, &wireDNS.A{
|
|
||||||
Hdr: wireDNS.RR_Header{Name: question.Name, Rrtype: wireDNS.TypeA, Class: wireDNS.ClassINET, Ttl: 60},
|
|
||||||
A: stdnet.IP{127, 0, 0, 7},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if err := w.WriteMsg(response); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
}),
|
|
||||||
}
|
|
||||||
go func() { stopped <- server.ActivateAndServe() }()
|
|
||||||
defer func() {
|
|
||||||
server.Shutdown()
|
|
||||||
select {
|
|
||||||
case err := <-stopped:
|
|
||||||
if err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
case <-time.After(3 * time.Second):
|
|
||||||
t.Error("DNS server did not stop")
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
select {
|
|
||||||
case <-ready:
|
|
||||||
case err := <-stopped:
|
|
||||||
t.Fatalf("DNS server startup: %v", err)
|
|
||||||
case <-time.After(3 * time.Second):
|
|
||||||
t.Fatal("DNS server did not start")
|
|
||||||
}
|
|
||||||
|
|
||||||
dnsScript := writeRouteScript(t, `
|
|
||||||
local server = require("xray.dns").Servers[1]
|
|
||||||
function HandleDNSQuery(domain, ipv4, ipv6, fake)
|
|
||||||
return server:Query(domain, ipv4, ipv6, fake)
|
|
||||||
end`)
|
|
||||||
routerScript := writeRouteScript(t, `
|
|
||||||
local router = require("xray.router")
|
|
||||||
local dns = require("xray.dns")
|
|
||||||
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.7")
|
|
||||||
local active = false
|
|
||||||
function HandleRoute(ctx, inbound, sourcePort, targetPort, localPort, domain, network,
|
|
||||||
protocol, user, vlessRoute, skipDNSResolve)
|
|
||||||
assert(not active, "borrowed Router VM reentered")
|
|
||||||
if inbound == "dns" then
|
|
||||||
assert(network == router.NetworkUDP and skipDNSResolve == false)
|
|
||||||
return "direct", "dns-route"
|
|
||||||
end
|
|
||||||
active = true
|
|
||||||
local ips, ttl, err = dns.Query("nested.example", true, false, false)
|
|
||||||
assert(not err and matcher:AnyMatch(ips) and active)
|
|
||||||
active = false
|
|
||||||
return "direct", "outer-route"
|
|
||||||
end`)
|
|
||||||
instance, err := core.New(&core.Config{
|
|
||||||
App: []*serial.TypedMessage{
|
|
||||||
serial.ToTypedMessage(&appdns.Config{
|
|
||||||
Tag: "dns", Script: dnsScript, DisableCache: true,
|
|
||||||
NameServer: []*appdns.NameServer{{
|
|
||||||
Id: "upstream", TimeoutMs: 1000,
|
|
||||||
Address: &net.Endpoint{
|
|
||||||
Network: net.Network_UDP,
|
|
||||||
Address: &net.IPOrDomain{Address: &net.IPOrDomain_Ip{Ip: []byte{127, 0, 0, 1}}},
|
|
||||||
Port: uint32(port),
|
|
||||||
},
|
|
||||||
}},
|
|
||||||
}),
|
|
||||||
serial.ToTypedMessage(&Config{Script: routerScript}),
|
|
||||||
serial.ToTypedMessage(&dispatcher.Config{}),
|
|
||||||
serial.ToTypedMessage(&proxyman.OutboundConfig{}),
|
|
||||||
},
|
|
||||||
Outbound: []*core.OutboundHandlerConfig{
|
|
||||||
{Tag: "default", ProxySettings: serial.ToTypedMessage(&blackhole.Config{})},
|
|
||||||
{Tag: "direct", ProxySettings: serial.ToTypedMessage(&freedom.Config{
|
|
||||||
FinalRules: []*freedom.FinalRuleConfig{{Action: freedom.RuleAction_Allow}},
|
|
||||||
})},
|
|
||||||
},
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer instance.Close()
|
|
||||||
if err := instance.Start(); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
r := instance.GetFeature(routing.RouterType()).(*Router)
|
|
||||||
route, err := r.PickRoute(newLuaRouteTestContext())
|
|
||||||
if err != nil || route.GetOutboundTag() != "direct" || route.GetRuleTag() != "outer-route" {
|
|
||||||
t.Fatalf("nested DNS routing = %v, %v", route, err)
|
|
||||||
}
|
|
||||||
if queries.Load() == 0 {
|
|
||||||
t.Fatal("DNS query did not pass through the dispatcher")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -5,8 +5,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type windowsReader struct {
|
type windowsReader struct {
|
||||||
bufs []syscall.WSABuf
|
bufs []syscall.WSABuf
|
||||||
ready bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *windowsReader) Init(bs []*Buffer) {
|
func (r *windowsReader) Init(bs []*Buffer) {
|
||||||
@@ -16,7 +15,6 @@ func (r *windowsReader) Init(bs []*Buffer) {
|
|||||||
for _, b := range bs {
|
for _, b := range bs {
|
||||||
r.bufs = append(r.bufs, syscall.WSABuf{Len: uint32(Size), Buf: &b.v[0]})
|
r.bufs = append(r.bufs, syscall.WSABuf{Len: uint32(Size), Buf: &b.v[0]})
|
||||||
}
|
}
|
||||||
r.ready = false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *windowsReader) Clear() {
|
func (r *windowsReader) Clear() {
|
||||||
@@ -27,14 +25,6 @@ func (r *windowsReader) Clear() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *windowsReader) Read(fd uintptr) int32 {
|
func (r *windowsReader) Read(fd uintptr) int32 {
|
||||||
// On the first invocation, we return -1 to indicate "not ready"
|
|
||||||
// to make rawConn.Read wait for readability using the runtime's own mechanism
|
|
||||||
// because syscall.WSARecv() is a blocking call when used with nil OVERLAPPED
|
|
||||||
if !r.ready {
|
|
||||||
r.ready = true
|
|
||||||
return -1
|
|
||||||
}
|
|
||||||
|
|
||||||
var nBytes uint32
|
var nBytes uint32
|
||||||
var flags uint32
|
var flags uint32
|
||||||
err := syscall.WSARecv(syscall.Handle(fd), &r.bufs[0], uint32(len(r.bufs)), &nBytes, &flags, nil, nil)
|
err := syscall.WSARecv(syscall.Handle(fd), &r.bufs[0], uint32(len(r.bufs)), &nBytes, &flags, nil, nil)
|
||||||
|
|||||||
@@ -118,9 +118,7 @@ func (w *BufferedWriter) Write(b []byte) (int, error) {
|
|||||||
|
|
||||||
nBytes, err := w.buffer.Write(b)
|
nBytes, err := w.buffer.Write(b)
|
||||||
totalBytes += nBytes
|
totalBytes += nBytes
|
||||||
|
if err != nil {
|
||||||
// ErrBufferFull means a partial write, so flush below and continue
|
|
||||||
if err != nil && err != ErrBufferFull {
|
|
||||||
return totalBytes, err
|
return totalBytes, err
|
||||||
}
|
}
|
||||||
if !w.buffered || w.buffer.IsFull() {
|
if !w.buffered || w.buffer.IsFull() {
|
||||||
|
|||||||
@@ -10,12 +10,12 @@ import (
|
|||||||
|
|
||||||
// [,)
|
// [,)
|
||||||
func RandBetween(from int64, to int64) int64 {
|
func RandBetween(from int64, to int64) int64 {
|
||||||
|
if from == to {
|
||||||
|
return from
|
||||||
|
}
|
||||||
if from > to {
|
if from > to {
|
||||||
from, to = to, from
|
from, to = to, from
|
||||||
}
|
}
|
||||||
if d := to - from; d == 0 || d == 1 {
|
|
||||||
return from
|
|
||||||
}
|
|
||||||
bigInt, _ := rand.Int(rand.Reader, big.NewInt(to-from))
|
bigInt, _ := rand.Int(rand.Reader, big.NewInt(to-from))
|
||||||
return from + bigInt.Int64()
|
return from + bigInt.Int64()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,58 +0,0 @@
|
|||||||
package geodata
|
|
||||||
|
|
||||||
import (
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
luar "layeh.com/gopher-luar"
|
|
||||||
)
|
|
||||||
|
|
||||||
// RegisterLua makes xray.geodata available to require in an LState.
|
|
||||||
func RegisterLua(L *lua.LState) {
|
|
||||||
L.PreloadModule("xray.geodata", func(L *lua.LState) int {
|
|
||||||
module := L.NewTable()
|
|
||||||
|
|
||||||
module.RawSetString("BuildDomainMatcher", L.NewFunction(func(L *lua.LState) int {
|
|
||||||
parsed, err := ParseDomainRules(luaRules(L), Domain_Domain)
|
|
||||||
if err != nil {
|
|
||||||
L.RaiseError("%v", err)
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
matcher, err := DomainReg.BuildDomainMatcher(parsed)
|
|
||||||
if err != nil {
|
|
||||||
L.RaiseError("%v", err)
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
L.Push(luar.New(L, matcher))
|
|
||||||
return 1
|
|
||||||
}))
|
|
||||||
|
|
||||||
module.RawSetString("BuildIPMatcher", L.NewFunction(func(L *lua.LState) int {
|
|
||||||
parsed, err := ParseIPRules(luaRules(L))
|
|
||||||
if err != nil {
|
|
||||||
L.RaiseError("%v", err)
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
matcher, err := IPReg.BuildIPMatcher(parsed)
|
|
||||||
if err != nil {
|
|
||||||
L.RaiseError("%v", err)
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
L.Push(luar.New(L, matcher))
|
|
||||||
return 1
|
|
||||||
}))
|
|
||||||
L.Push(module)
|
|
||||||
return 1
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func luaRules(L *lua.LState) []string {
|
|
||||||
rules := make([]string, L.GetTop())
|
|
||||||
for i := range rules {
|
|
||||||
value, ok := L.Get(i + 1).(lua.LString)
|
|
||||||
if !ok {
|
|
||||||
L.RaiseError("geodata rules must be strings")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
rules[i] = string(value)
|
|
||||||
}
|
|
||||||
return rules
|
|
||||||
}
|
|
||||||
@@ -1,66 +0,0 @@
|
|||||||
package geodata
|
|
||||||
|
|
||||||
import (
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestLuaIPMatcher(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
RegisterLua(L)
|
|
||||||
ip := L.NewUserData()
|
|
||||||
ip.Value = net.ParseIP("127.0.0.1")
|
|
||||||
L.SetGlobal("ip", ip)
|
|
||||||
ips := L.NewUserData()
|
|
||||||
ips.Value = []net.IP{ip.Value.(net.IP), net.ParseIP("8.8.8.8")}
|
|
||||||
L.SetGlobal("ips", ips)
|
|
||||||
if err := L.DoString(`
|
|
||||||
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.0/8", "::1")
|
|
||||||
assert(matcher:Match(ip))
|
|
||||||
assert(matcher:AnyMatch(ips))
|
|
||||||
assert(not matcher:Matches(ips))
|
|
||||||
local matched, unmatched = matcher:FilterIPs(ips)
|
|
||||||
assert(type(matched) == "userdata" and type(unmatched) == "userdata")
|
|
||||||
assert(#matched == 1 and #unmatched == 1)
|
|
||||||
`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaDomainMatcher(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
RegisterLua(L)
|
|
||||||
if err := L.DoString(`
|
|
||||||
local matcher = require("xray.geodata").BuildDomainMatcher("example.com", "full:other.com")
|
|
||||||
assert(matcher:MatchAny("example.com"))
|
|
||||||
assert(matcher:MatchAny("www.example.com"))
|
|
||||||
assert(matcher:MatchAny("other.com"))
|
|
||||||
assert(not matcher:MatchAny("www.other.com"))
|
|
||||||
assert(#(matcher:Match("www.example.com")) == 1)
|
|
||||||
`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaMatchersRejectInvalidRules(t *testing.T) {
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name string
|
|
||||||
script string
|
|
||||||
}{
|
|
||||||
{"IP rule", `require("xray.geodata").BuildIPMatcher("not-an-ip")`},
|
|
||||||
{"non-string domain rule", `require("xray.geodata").BuildDomainMatcher("example.com", true)`},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
RegisterLua(L)
|
|
||||||
if err := L.DoString(tc.script); err == nil {
|
|
||||||
t.Fatal("invalid geodata rule was accepted")
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,50 +0,0 @@
|
|||||||
package log
|
|
||||||
|
|
||||||
import (
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
// RegisterLua makes xray.log available to require in an LState.
|
|
||||||
func RegisterLua(L *lua.LState) {
|
|
||||||
L.PreloadModule("xray.log", func(L *lua.LState) int {
|
|
||||||
module := L.NewTable()
|
|
||||||
for name, severity := range map[string]Severity{
|
|
||||||
"Debug": Severity_Debug,
|
|
||||||
"Info": Severity_Info,
|
|
||||||
"Warning": Severity_Warning,
|
|
||||||
"Error": Severity_Error,
|
|
||||||
} {
|
|
||||||
module.RawSetString(name, L.NewFunction(func(L *lua.LState) int {
|
|
||||||
var content strings.Builder
|
|
||||||
// Prefix with the calling script's filename.
|
|
||||||
if caller, ok := L.GetStack(1); ok {
|
|
||||||
if _, err := L.GetInfo("S", caller, lua.LNil); err == nil && caller.Source != "" {
|
|
||||||
content.WriteString(filepath.Base(strings.TrimPrefix(caller.Source, "@")))
|
|
||||||
content.WriteString(": ")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for i := 1; i <= L.GetTop(); i++ {
|
|
||||||
value := L.Get(i)
|
|
||||||
// Use Error() for Go errors in userdata.
|
|
||||||
if ud, ok := value.(*lua.LUserData); ok {
|
|
||||||
if err, ok := ud.Value.(error); ok {
|
|
||||||
content.WriteString(err.Error())
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
content.WriteString(L.ToStringMeta(value).String())
|
|
||||||
}
|
|
||||||
Record(&GeneralMessage{
|
|
||||||
Severity: severity,
|
|
||||||
Content: content.String(),
|
|
||||||
})
|
|
||||||
return 0
|
|
||||||
}))
|
|
||||||
}
|
|
||||||
L.Push(module)
|
|
||||||
return 1
|
|
||||||
})
|
|
||||||
}
|
|
||||||
@@ -1,92 +0,0 @@
|
|||||||
package log
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
lua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
type luaLogHandler struct {
|
|
||||||
messages []Message
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h *luaLogHandler) Handle(msg Message) {
|
|
||||||
h.messages = append(h.messages, msg)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLuaLog(t *testing.T) {
|
|
||||||
logHandler.RLock()
|
|
||||||
previous := logHandler.Handler
|
|
||||||
logHandler.RUnlock()
|
|
||||||
t.Cleanup(func() { RegisterHandler(previous) })
|
|
||||||
handler := &luaLogHandler{}
|
|
||||||
RegisterHandler(handler)
|
|
||||||
|
|
||||||
L := lua.NewState()
|
|
||||||
defer L.Close()
|
|
||||||
RegisterLua(L)
|
|
||||||
nativeError := L.NewUserData()
|
|
||||||
nativeError.Value = fmt.Errorf("lookup failed: %w", errors.New("upstream timeout"))
|
|
||||||
L.SetGlobal("nativeError", nativeError)
|
|
||||||
path := filepath.Join(t.TempDir(), "logging.lua")
|
|
||||||
if err := os.WriteFile(path, []byte(`
|
|
||||||
local log = require("xray.log")
|
|
||||||
assert(log == require("xray.log"))
|
|
||||||
log.Debug("query: ", "example.com")
|
|
||||||
log.Info("count=", 42, ", enabled=", true, ", value=", nil)
|
|
||||||
log.Warning(setmetatable({}, {
|
|
||||||
__tostring = function() return "fallback" end
|
|
||||||
}))
|
|
||||||
assert(select("#", log.Error("failed")) == 0)
|
|
||||||
log.Error("DNS failed: ", nativeError)
|
|
||||||
log.Warning(nativeError)
|
|
||||||
local ok, err = pcall(function() error("Lua failure", 0) end)
|
|
||||||
assert(not ok)
|
|
||||||
log.Error(err)
|
|
||||||
function logHook()
|
|
||||||
log.Info("hook")
|
|
||||||
end
|
|
||||||
`), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if err := L.DoFile(path); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if err := L.DoString(`
|
|
||||||
logHook()
|
|
||||||
require("xray.log").Info("anonymous")
|
|
||||||
`); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
want := []struct {
|
|
||||||
severity Severity
|
|
||||||
message string
|
|
||||||
}{
|
|
||||||
{Severity_Debug, "[Debug] logging.lua: query: example.com"},
|
|
||||||
{Severity_Info, "[Info] logging.lua: count=42, enabled=true, value=nil"},
|
|
||||||
{Severity_Warning, "[Warning] logging.lua: fallback"},
|
|
||||||
{Severity_Error, "[Error] logging.lua: failed"},
|
|
||||||
{Severity_Error, "[Error] logging.lua: DNS failed: lookup failed: upstream timeout"},
|
|
||||||
{Severity_Warning, "[Warning] logging.lua: lookup failed: upstream timeout"},
|
|
||||||
{Severity_Error, "[Error] logging.lua: Lua failure"},
|
|
||||||
{Severity_Info, "[Info] logging.lua: hook"},
|
|
||||||
{Severity_Info, "[Info] <string>: anonymous"},
|
|
||||||
}
|
|
||||||
if len(handler.messages) != len(want) {
|
|
||||||
t.Fatalf("logged %d messages, want %d", len(handler.messages), len(want))
|
|
||||||
}
|
|
||||||
for i, expected := range want {
|
|
||||||
msg, ok := handler.messages[i].(*GeneralMessage)
|
|
||||||
if !ok {
|
|
||||||
t.Fatalf("message %d has type %T, want *GeneralMessage", i, handler.messages[i])
|
|
||||||
}
|
|
||||||
if msg.Severity != expected.severity || msg.String() != expected.message {
|
|
||||||
t.Errorf("message %d = %q with severity %v, want %q with severity %v", i, msg.String(), msg.Severity, expected.message, expected.severity)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,138 +0,0 @@
|
|||||||
package lua
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"errors"
|
|
||||||
"sync"
|
|
||||||
|
|
||||||
glua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
const maxIdleStates = 16
|
|
||||||
|
|
||||||
// LStateFactory must initialize a state fully and observe ctx while doing so.
|
|
||||||
// The pool owns any non-nil state it returns, even when it also returns an error.
|
|
||||||
type LStateFactory func(ctx context.Context) (*glua.LState, error)
|
|
||||||
|
|
||||||
// Pool lends each state to one caller at a time. It grows on contention and
|
|
||||||
// keeps up to maxIdleStates idle states until Close. Callers decide whether a
|
|
||||||
// state is reusable.
|
|
||||||
type Pool struct {
|
|
||||||
ctx context.Context
|
|
||||||
cancel context.CancelFunc
|
|
||||||
|
|
||||||
factory LStateFactory
|
|
||||||
idle []*glua.LState
|
|
||||||
|
|
||||||
mu sync.Mutex
|
|
||||||
active sync.WaitGroup
|
|
||||||
closed bool
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewPool initializes one state before returning, so top-level errors surface at startup.
|
|
||||||
func NewPool(ctx context.Context, factory LStateFactory) (*Pool, error) {
|
|
||||||
poolCtx, cancel := context.WithCancel(ctx)
|
|
||||||
|
|
||||||
// Create one state now to catch factory errors at startup.
|
|
||||||
state, err := factory(poolCtx)
|
|
||||||
if err != nil {
|
|
||||||
cancel()
|
|
||||||
if state != nil {
|
|
||||||
state.Close()
|
|
||||||
}
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if state == nil {
|
|
||||||
cancel()
|
|
||||||
return nil, errors.New("Lua state factory returned nil")
|
|
||||||
}
|
|
||||||
if err := poolCtx.Err(); err != nil {
|
|
||||||
state.Close()
|
|
||||||
cancel()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return &Pool{ctx: poolCtx, cancel: cancel, factory: factory, idle: []*glua.LState{state}}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Context is cancelled by Close. Query contexts should derive from it.
|
|
||||||
func (p *Pool) Context() context.Context {
|
|
||||||
return p.ctx
|
|
||||||
}
|
|
||||||
|
|
||||||
// Acquire returns an initialized exclusive state, growing the pool if necessary.
|
|
||||||
func (p *Pool) Acquire() (*glua.LState, error) {
|
|
||||||
p.mu.Lock()
|
|
||||||
if p.closed || p.ctx.Err() != nil {
|
|
||||||
p.mu.Unlock()
|
|
||||||
return nil, p.ctx.Err()
|
|
||||||
}
|
|
||||||
|
|
||||||
p.active.Add(1)
|
|
||||||
|
|
||||||
n := len(p.idle)
|
|
||||||
if n != 0 {
|
|
||||||
state := p.idle[n-1]
|
|
||||||
p.idle = p.idle[:n-1]
|
|
||||||
p.mu.Unlock()
|
|
||||||
return state, nil
|
|
||||||
}
|
|
||||||
p.mu.Unlock()
|
|
||||||
|
|
||||||
// TODO: Limit the total number of states. When the limit is reached, wait
|
|
||||||
// for a Release instead of creating another state; allow the wait to be
|
|
||||||
// cancelled by the caller or by Close.
|
|
||||||
state, err := p.factory(p.ctx)
|
|
||||||
if err == nil && state == nil {
|
|
||||||
err = errors.New("Lua state factory returned nil")
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
if state != nil {
|
|
||||||
state.Close()
|
|
||||||
}
|
|
||||||
p.active.Done()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if err := p.ctx.Err(); err != nil {
|
|
||||||
state.Close()
|
|
||||||
p.active.Done()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return state, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Release returns a healthy state to the pool and closes a failed or cancelled one.
|
|
||||||
func (p *Pool) Release(state *glua.LState, reusable bool) {
|
|
||||||
if reusable {
|
|
||||||
p.mu.Lock()
|
|
||||||
if !p.closed && p.ctx.Err() == nil && len(p.idle) < maxIdleStates {
|
|
||||||
p.idle = append(p.idle, state)
|
|
||||||
} else {
|
|
||||||
reusable = false
|
|
||||||
}
|
|
||||||
p.mu.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
if !reusable {
|
|
||||||
state.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
p.active.Done()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close cancels active work, closes idle states, and waits for borrowed states.
|
|
||||||
func (p *Pool) Close() {
|
|
||||||
p.mu.Lock()
|
|
||||||
if !p.closed {
|
|
||||||
p.closed = true
|
|
||||||
p.cancel()
|
|
||||||
for _, state := range p.idle {
|
|
||||||
state.Close()
|
|
||||||
}
|
|
||||||
p.idle = nil
|
|
||||||
}
|
|
||||||
p.mu.Unlock()
|
|
||||||
|
|
||||||
p.active.Wait()
|
|
||||||
}
|
|
||||||
@@ -1,197 +0,0 @@
|
|||||||
package lua
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"errors"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
glua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestPoolFactoryFailureClosesReturnedState(t *testing.T) {
|
|
||||||
failure := errors.New("factory failed")
|
|
||||||
state := glua.NewState()
|
|
||||||
_, err := NewPool(context.Background(), func(context.Context) (*glua.LState, error) {
|
|
||||||
return state, failure
|
|
||||||
})
|
|
||||||
if !errors.Is(err, failure) || !state.IsClosed() {
|
|
||||||
t.Fatalf("NewPool error = %v, state closed = %t", err, state.IsClosed())
|
|
||||||
}
|
|
||||||
|
|
||||||
var failedState *glua.LState
|
|
||||||
calls := 0
|
|
||||||
pool, err := NewPool(context.Background(), func(context.Context) (*glua.LState, error) {
|
|
||||||
calls++
|
|
||||||
if calls == 1 {
|
|
||||||
return glua.NewState(), nil
|
|
||||||
}
|
|
||||||
failedState = glua.NewState()
|
|
||||||
return failedState, failure
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer pool.Close()
|
|
||||||
borrowed, err := pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer pool.Release(borrowed, true)
|
|
||||||
_, err = pool.Acquire()
|
|
||||||
if !errors.Is(err, failure) || !failedState.IsClosed() {
|
|
||||||
t.Fatalf("Acquire error = %v, state closed = %t", err, failedState.IsClosed())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPoolReusesStatesAndLimitsIdle(t *testing.T) {
|
|
||||||
created := 0
|
|
||||||
pool, err := NewPool(context.Background(), func(context.Context) (*glua.LState, error) {
|
|
||||||
created++
|
|
||||||
return glua.NewState(), nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer pool.Close()
|
|
||||||
|
|
||||||
states := make([]*glua.LState, maxIdleStates+3)
|
|
||||||
for i := range states {
|
|
||||||
states[i], err = pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for _, state := range states {
|
|
||||||
pool.Release(state, true)
|
|
||||||
}
|
|
||||||
for i, state := range states {
|
|
||||||
if got, want := state.IsClosed(), i >= maxIdleStates; got != want {
|
|
||||||
t.Fatalf("state %d closed = %t, want %t", i, got, want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
borrowed, err := pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if created != len(states) {
|
|
||||||
t.Fatalf("Acquire created %d states, want %d", created, len(states))
|
|
||||||
}
|
|
||||||
pool.Release(borrowed, true)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPoolCloseCancelsAndWaitsForBorrowedState(t *testing.T) {
|
|
||||||
pool, err := NewPool(context.Background(), func(context.Context) (*glua.LState, error) {
|
|
||||||
return glua.NewState(), nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
state, err := pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
done := make(chan struct{})
|
|
||||||
go func() {
|
|
||||||
pool.Close()
|
|
||||||
close(done)
|
|
||||||
}()
|
|
||||||
select {
|
|
||||||
case <-pool.Context().Done():
|
|
||||||
case <-time.After(time.Second):
|
|
||||||
pool.Release(state, false)
|
|
||||||
t.Fatal("Close did not cancel the pool context")
|
|
||||||
}
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
pool.Release(state, false)
|
|
||||||
t.Fatal("Close returned while a state was borrowed")
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
pool.Release(state, true)
|
|
||||||
select {
|
|
||||||
case <-done:
|
|
||||||
case <-time.After(time.Second):
|
|
||||||
t.Fatal("Close did not finish after Release")
|
|
||||||
}
|
|
||||||
if !state.IsClosed() {
|
|
||||||
t.Fatal("borrowed state was not closed")
|
|
||||||
}
|
|
||||||
if _, err := pool.Acquire(); !errors.Is(err, context.Canceled) {
|
|
||||||
t.Fatalf("Acquire after Close = %v, want context.Canceled", err)
|
|
||||||
}
|
|
||||||
pool.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestPoolCloseCancelsStateCreation(t *testing.T) {
|
|
||||||
started := make(chan struct{})
|
|
||||||
first := true
|
|
||||||
pool, err := NewPool(context.Background(), func(ctx context.Context) (*glua.LState, error) {
|
|
||||||
if first {
|
|
||||||
first = false
|
|
||||||
return glua.NewState(), nil
|
|
||||||
}
|
|
||||||
close(started)
|
|
||||||
<-ctx.Done()
|
|
||||||
return nil, ctx.Err()
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
borrowed, err := pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
acquireDone := make(chan error, 1)
|
|
||||||
go func() {
|
|
||||||
_, err := pool.Acquire()
|
|
||||||
acquireDone <- err
|
|
||||||
}()
|
|
||||||
<-started
|
|
||||||
closeDone := make(chan struct{})
|
|
||||||
go func() {
|
|
||||||
pool.Close()
|
|
||||||
close(closeDone)
|
|
||||||
}()
|
|
||||||
select {
|
|
||||||
case err := <-acquireDone:
|
|
||||||
if !errors.Is(err, context.Canceled) {
|
|
||||||
t.Fatalf("Acquire during Close = %v, want context.Canceled", err)
|
|
||||||
}
|
|
||||||
case <-time.After(time.Second):
|
|
||||||
t.Fatal("state creation did not stop after Close")
|
|
||||||
}
|
|
||||||
select {
|
|
||||||
case <-closeDone:
|
|
||||||
pool.Release(borrowed, false)
|
|
||||||
t.Fatal("Close returned while the initial state was borrowed")
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
pool.Release(borrowed, true)
|
|
||||||
select {
|
|
||||||
case <-closeDone:
|
|
||||||
case <-time.After(time.Second):
|
|
||||||
t.Fatal("Close did not finish after Release")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func BenchmarkPoolAcquireRelease(b *testing.B) {
|
|
||||||
pool, err := NewPool(context.Background(), func(context.Context) (*glua.LState, error) {
|
|
||||||
return glua.NewState(), nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
defer pool.Close()
|
|
||||||
|
|
||||||
b.ReportAllocs()
|
|
||||||
b.ResetTimer()
|
|
||||||
for i := 0; i < b.N; i++ {
|
|
||||||
state, err := pool.Acquire()
|
|
||||||
if err != nil {
|
|
||||||
b.Fatal(err)
|
|
||||||
}
|
|
||||||
pool.Release(state, true)
|
|
||||||
}
|
|
||||||
b.StopTimer()
|
|
||||||
}
|
|
||||||
@@ -1,56 +0,0 @@
|
|||||||
// Package lua provides shared GopherLua programs and state management for Xray scripts.
|
|
||||||
package lua
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
"context"
|
|
||||||
"os"
|
|
||||||
|
|
||||||
glua "github.com/yuin/gopher-lua"
|
|
||||||
"github.com/yuin/gopher-lua/parse"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Program holds immutable bytecode that can be run by independent LStates.
|
|
||||||
type Program struct {
|
|
||||||
proto *glua.FunctionProto
|
|
||||||
}
|
|
||||||
|
|
||||||
// CompileFile reads and compiles a Lua file once.
|
|
||||||
func CompileFile(path string) (*Program, error) {
|
|
||||||
f, err := os.Open(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer f.Close()
|
|
||||||
chunk, err := parse.Parse(bufio.NewReader(f), path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
proto, err := glua.Compile(chunk, path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &Program{proto: proto}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewState creates a VM, makes modules available, and executes the file top level.
|
|
||||||
// Module loaders run only when Lua calls require. Each state gets its own globals.
|
|
||||||
// The caller owns the returned state.
|
|
||||||
func (p *Program) NewState(ctx context.Context, register func(*glua.LState)) (*glua.LState, error) {
|
|
||||||
L := glua.NewState()
|
|
||||||
if register != nil {
|
|
||||||
register(L)
|
|
||||||
}
|
|
||||||
L.SetContext(ctx)
|
|
||||||
L.Push(L.NewFunctionFromProto(p.proto))
|
|
||||||
err := L.PCall(0, 0, nil)
|
|
||||||
L.RemoveContext()
|
|
||||||
if err == nil {
|
|
||||||
err = ctx.Err()
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
L.Close()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return L, nil
|
|
||||||
}
|
|
||||||
@@ -1,55 +0,0 @@
|
|||||||
package lua
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
glua "github.com/yuin/gopher-lua"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestProgramStatesAreIndependent(t *testing.T) {
|
|
||||||
path := filepath.Join(t.TempDir(), "state.lua")
|
|
||||||
if err := os.WriteFile(path, []byte("value = (value or 0) + 1"), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
program, err := CompileFile(path)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
first, err := program.NewState(context.Background(), nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer first.Close()
|
|
||||||
first.SetGlobal("value", glua.LNumber(42))
|
|
||||||
second, err := program.NewState(context.Background(), nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
defer second.Close()
|
|
||||||
if got := second.GetGlobal("value"); got != glua.LNumber(1) {
|
|
||||||
t.Fatalf("second state value = %v, want 1", got)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestProgramInitializationObservesCancellation(t *testing.T) {
|
|
||||||
path := filepath.Join(t.TempDir(), "loop.lua")
|
|
||||||
if err := os.WriteFile(path, []byte("while true do end"), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
program, err := CompileFile(path)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
|
||||||
cancel()
|
|
||||||
state, err := program.NewState(ctx, nil)
|
|
||||||
if err == nil || state != nil {
|
|
||||||
if state != nil {
|
|
||||||
state.Close()
|
|
||||||
}
|
|
||||||
t.Fatalf("NewState with canceled context = %v, %v; want nil state and error", state, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,8 +1,6 @@
|
|||||||
package platform // import "github.com/xtls/xray-core/common/platform"
|
package platform // import "github.com/xtls/xray-core/common/platform"
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strconv"
|
"strconv"
|
||||||
@@ -92,49 +90,3 @@ func GetConfDirPath() string {
|
|||||||
configPath := NewEnvFlag(ConfdirLocation).GetValue(func() string { return "" })
|
configPath := NewEnvFlag(ConfdirLocation).GetValue(func() string { return "" })
|
||||||
return configPath
|
return configPath
|
||||||
}
|
}
|
||||||
|
|
||||||
// ResolveLuaFile finds a local Lua script and returns its absolute path.
|
|
||||||
// Relative paths: XRAY_LOCATION_CONFDIR > XRAY_LOCATION_CONFIG > working dir > executable dir.
|
|
||||||
func ResolveLuaFile(path string) (string, error) {
|
|
||||||
if path == "" {
|
|
||||||
return "", errors.New("Lua file path is empty")
|
|
||||||
}
|
|
||||||
paths := []string{path}
|
|
||||||
if !filepath.IsAbs(path) {
|
|
||||||
paths = nil
|
|
||||||
for _, dir := range []string{
|
|
||||||
GetConfDirPath(),
|
|
||||||
NewEnvFlag(ConfigLocation).GetValue(func() string { return "" }),
|
|
||||||
".",
|
|
||||||
getExecutableDir(),
|
|
||||||
} {
|
|
||||||
if dir != "" {
|
|
||||||
paths = append(paths, filepath.Join(dir, path))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return resolveFile(paths)
|
|
||||||
}
|
|
||||||
|
|
||||||
func resolveFile(paths []string) (string, error) {
|
|
||||||
var tried []string
|
|
||||||
for _, path := range paths {
|
|
||||||
path, err := filepath.Abs(path)
|
|
||||||
if err != nil {
|
|
||||||
return "", fmt.Errorf("failed to resolve file path: %w", err)
|
|
||||||
}
|
|
||||||
tried = append(tried, path)
|
|
||||||
info, err := os.Stat(path)
|
|
||||||
if errors.Is(err, os.ErrNotExist) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return "", fmt.Errorf("failed to inspect file %q: %w", path, err)
|
|
||||||
}
|
|
||||||
if !info.Mode().IsRegular() {
|
|
||||||
return "", fmt.Errorf("file is not a regular file: %s", path)
|
|
||||||
}
|
|
||||||
return path, nil
|
|
||||||
}
|
|
||||||
return "", fmt.Errorf("file not found; tried %q: %w", tried, os.ErrNotExist)
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
package platform_test
|
package platform_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
@@ -65,53 +64,3 @@ func TestGetAssetLocation(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestResolveLuaFile(t *testing.T) {
|
|
||||||
workingDir := t.TempDir()
|
|
||||||
t.Chdir(workingDir)
|
|
||||||
executable, err := os.Executable()
|
|
||||||
common.Must(err)
|
|
||||||
file, err := os.CreateTemp(filepath.Dir(executable), "lua-*.lua")
|
|
||||||
common.Must(err)
|
|
||||||
common.Must(file.Close())
|
|
||||||
defer os.Remove(file.Name())
|
|
||||||
|
|
||||||
name := filepath.Base(file.Name())
|
|
||||||
paths := []string{
|
|
||||||
filepath.Join(t.TempDir(), name),
|
|
||||||
filepath.Join(t.TempDir(), name),
|
|
||||||
filepath.Join(workingDir, name),
|
|
||||||
file.Name(),
|
|
||||||
}
|
|
||||||
t.Setenv(ConfdirLocation, filepath.Dir(paths[0]))
|
|
||||||
t.Setenv(ConfigLocation, filepath.Dir(paths[1]))
|
|
||||||
for _, path := range paths[:3] {
|
|
||||||
common.Must(os.WriteFile(path, nil, 0o600))
|
|
||||||
}
|
|
||||||
if got, err := ResolveLuaFile(paths[2]); err != nil || got != paths[2] {
|
|
||||||
t.Fatalf("absolute path = %q, %v; want %q", got, err, paths[2])
|
|
||||||
}
|
|
||||||
for i, want := range paths {
|
|
||||||
if i == 2 {
|
|
||||||
t.Setenv(ConfdirLocation, "")
|
|
||||||
t.Setenv(ConfigLocation, "")
|
|
||||||
}
|
|
||||||
if got, err := ResolveLuaFile(name); err != nil || got != want {
|
|
||||||
t.Fatalf("resolved path = %q, %v; want %q", got, err, want)
|
|
||||||
}
|
|
||||||
common.Must(os.Remove(want))
|
|
||||||
}
|
|
||||||
if _, err := ResolveLuaFile(name); !errors.Is(err, os.ErrNotExist) {
|
|
||||||
t.Fatalf("missing file error = %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
t.Setenv(ConfdirLocation, filepath.Dir(paths[0]))
|
|
||||||
t.Setenv(ConfigLocation, filepath.Dir(paths[1]))
|
|
||||||
common.Must(os.Mkdir(paths[0], 0o700))
|
|
||||||
common.Must(os.WriteFile(paths[1], nil, 0o600))
|
|
||||||
for _, path := range []string{"", name, filepath.Join(t.TempDir(), name)} {
|
|
||||||
if _, err := ResolveLuaFile(path); err == nil {
|
|
||||||
t.Fatalf("accepted invalid path %q", path)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -70,6 +70,8 @@ type Outbound struct {
|
|||||||
Tag string
|
Tag string
|
||||||
// Name of the outbound proxy that handles the connection.
|
// Name of the outbound proxy that handles the connection.
|
||||||
Name string
|
Name string
|
||||||
|
// Unused. Conn is actually internet.Connection. May be nil. It is currently nil for outbound with proxySettings
|
||||||
|
Conn net.Conn
|
||||||
// CanSpliceCopy is a property for this connection
|
// CanSpliceCopy is a property for this connection
|
||||||
// 1 = can, 2 = after processing protocol info should be able to, 3 = cannot
|
// 1 = can, 2 = after processing protocol info should be able to, 3 = cannot
|
||||||
CanSpliceCopy int
|
CanSpliceCopy int
|
||||||
|
|||||||
+2
-2
@@ -19,8 +19,8 @@ import (
|
|||||||
|
|
||||||
var (
|
var (
|
||||||
Version_x byte = 26
|
Version_x byte = 26
|
||||||
Version_y byte = 9
|
Version_y byte = 7
|
||||||
Version_z byte = 9
|
Version_z byte = 28
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
module github.com/xtls/xray-core
|
module github.com/xtls/xray-core
|
||||||
|
|
||||||
go 1.27
|
go 1.26
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/apernet/quic-go v0.61.1-0.20260806010916-184d081eef3e
|
github.com/apernet/quic-go v0.61.1-0.20260806010916-184d081eef3e
|
||||||
@@ -22,24 +22,21 @@ require (
|
|||||||
github.com/sagernet/sing-shadowsocks v0.2.7
|
github.com/sagernet/sing-shadowsocks v0.2.7
|
||||||
github.com/stretchr/testify v1.12.1
|
github.com/stretchr/testify v1.12.1
|
||||||
github.com/vishvananda/netlink v1.3.1
|
github.com/vishvananda/netlink v1.3.1
|
||||||
github.com/xtls/reality v0.0.0-20260908062103-8cdf7bf9c7f0
|
github.com/xtls/reality v0.0.0-20260322125925-9234c772ba8f
|
||||||
github.com/yuin/gopher-lua v1.1.2
|
|
||||||
go4.org/netipx v0.0.0-20231129151722-fdeea329fbba
|
go4.org/netipx v0.0.0-20231129151722-fdeea329fbba
|
||||||
golang.org/x/crypto v0.57.0
|
golang.org/x/crypto v0.55.0
|
||||||
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842
|
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842
|
||||||
golang.org/x/net v0.59.0
|
golang.org/x/net v0.58.0
|
||||||
golang.org/x/sync v0.23.0
|
golang.org/x/sync v0.22.0
|
||||||
golang.org/x/sys v0.48.0
|
golang.org/x/sys v0.47.0
|
||||||
golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2
|
golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2
|
||||||
golang.zx2c4.com/wireguard v0.0.0-20250521234502-f333402bd9cb
|
golang.zx2c4.com/wireguard v0.0.0-20250521234502-f333402bd9cb
|
||||||
golang.zx2c4.com/wireguard/windows v1.1.1
|
golang.zx2c4.com/wireguard/windows v1.0.1
|
||||||
google.golang.org/grpc v1.84.0
|
google.golang.org/grpc v1.83.1
|
||||||
google.golang.org/protobuf v1.36.12
|
google.golang.org/protobuf v1.36.12
|
||||||
gvisor.dev/gvisor v0.0.0-20260122175437-89a5d21be8f0
|
gvisor.dev/gvisor v0.0.0-20260122175437-89a5d21be8f0
|
||||||
h12.io/socks v1.0.3
|
h12.io/socks v1.0.3
|
||||||
layeh.com/gopher-luar v1.0.11
|
|
||||||
lukechampine.com/blake3 v1.4.1
|
lukechampine.com/blake3 v1.4.1
|
||||||
mvdan.cc/gofumpt v0.12.0
|
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
@@ -51,6 +48,7 @@ require (
|
|||||||
github.com/juju/ratelimit v1.0.2 // indirect
|
github.com/juju/ratelimit v1.0.2 // indirect
|
||||||
github.com/klauspost/compress v1.17.4 // indirect
|
github.com/klauspost/compress v1.17.4 // indirect
|
||||||
github.com/koron/go-ssdp v0.0.4 // indirect
|
github.com/koron/go-ssdp v0.0.4 // indirect
|
||||||
|
github.com/kr/text v0.2.0 // indirect
|
||||||
github.com/libp2p/go-netroute v0.2.1 // indirect
|
github.com/libp2p/go-netroute v0.2.1 // indirect
|
||||||
github.com/pion/dtls/v3 v3.1.5 // indirect
|
github.com/pion/dtls/v3 v3.1.5 // indirect
|
||||||
github.com/pion/logging v0.2.4 // indirect
|
github.com/pion/logging v0.2.4 // indirect
|
||||||
@@ -59,9 +57,8 @@ require (
|
|||||||
github.com/vishvananda/netns v0.0.5 // indirect
|
github.com/vishvananda/netns v0.0.5 // indirect
|
||||||
github.com/wlynxg/anet v0.0.5 // indirect
|
github.com/wlynxg/anet v0.0.5 // indirect
|
||||||
go.yaml.in/yaml/v3 v3.0.5 // indirect
|
go.yaml.in/yaml/v3 v3.0.5 // indirect
|
||||||
golang.org/x/text v0.42.0 // indirect
|
golang.org/x/text v0.41.0 // indirect
|
||||||
golang.org/x/time v0.14.0 // indirect
|
golang.org/x/time v0.14.0 // indirect
|
||||||
golang.org/x/tools v0.49.0 // indirect
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20260526163538-3dc84a4a5aaa // indirect
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20260706201446-f0a921348800 // indirect
|
|
||||||
gopkg.in/yaml.v2 v2.4.0 // indirect
|
gopkg.in/yaml.v2 v2.4.0 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -2,15 +2,17 @@ github.com/andybalholm/brotli v1.0.6 h1:Yf9fFpf49Zrxb9NlQaluyE92/+X7UVHlhMNJN2sx
|
|||||||
github.com/andybalholm/brotli v1.0.6/go.mod h1:fO7iG3H7G2nSZ7m0zPUDn85XEX2GTukHGRSepvi9Eig=
|
github.com/andybalholm/brotli v1.0.6/go.mod h1:fO7iG3H7G2nSZ7m0zPUDn85XEX2GTukHGRSepvi9Eig=
|
||||||
github.com/apernet/quic-go v0.61.1-0.20260806010916-184d081eef3e h1:5mgtR5gwIgBKMiGI1QdXldZZ+SNor06Nbu1wCBulQBg=
|
github.com/apernet/quic-go v0.61.1-0.20260806010916-184d081eef3e h1:5mgtR5gwIgBKMiGI1QdXldZZ+SNor06Nbu1wCBulQBg=
|
||||||
github.com/apernet/quic-go v0.61.1-0.20260806010916-184d081eef3e/go.mod h1:x7qxEvX6MCVtDuBKHj3E+88+BtrbEMuAL5qGUKItjW8=
|
github.com/apernet/quic-go v0.61.1-0.20260806010916-184d081eef3e/go.mod h1:x7qxEvX6MCVtDuBKHj3E+88+BtrbEMuAL5qGUKItjW8=
|
||||||
github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
|
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||||
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
|
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||||
github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
|
|
||||||
github.com/cloudflare/circl v1.6.5 h1:O64F26HEqNhznd/hrC5KZXVKYuKM2rx4deZDTc4ihQA=
|
github.com/cloudflare/circl v1.6.5 h1:O64F26HEqNhznd/hrC5KZXVKYuKM2rx4deZDTc4ihQA=
|
||||||
github.com/cloudflare/circl v1.6.5/go.mod h1:h5LNyxAc5nTue9DS5jT+48en2PSDYt3zdGnz5OstK6c=
|
github.com/cloudflare/circl v1.6.5/go.mod h1:h5LNyxAc5nTue9DS5jT+48en2PSDYt3zdGnz5OstK6c=
|
||||||
|
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
|
||||||
github.com/ghodss/yaml v1.0.1-0.20220118164431-d8423dcdf344 h1:Arcl6UOIS/kgO2nW3A65HN+7CMjSDP/gofXL4CZt1V4=
|
github.com/ghodss/yaml v1.0.1-0.20220118164431-d8423dcdf344 h1:Arcl6UOIS/kgO2nW3A65HN+7CMjSDP/gofXL4CZt1V4=
|
||||||
github.com/ghodss/yaml v1.0.1-0.20220118164431-d8423dcdf344/go.mod h1:GIjDIg/heH5DOkXY3YJ/wNhfHsQHoXGjl8G8amsYQ1I=
|
github.com/ghodss/yaml v1.0.1-0.20220118164431-d8423dcdf344/go.mod h1:GIjDIg/heH5DOkXY3YJ/wNhfHsQHoXGjl8G8amsYQ1I=
|
||||||
github.com/go-quicktest/qt v1.102.0 h1:HSQxCeh5YZH3EL3W39ixjtyaEhcWSXQHtHnMBzSs474=
|
github.com/go-logr/logr v1.4.3 h1:CjnDlHq8ikf6E492q6eKboGOC0T8CDaOvkHCIg8idEI=
|
||||||
github.com/go-quicktest/qt v1.102.0/go.mod h1:p4lGIVX+8Wa6ZPNDvqcxq36XpUDLh42FLetFU7odllI=
|
github.com/go-logr/logr v1.4.3/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
|
||||||
|
github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
|
||||||
|
github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=
|
||||||
github.com/golang/mock v1.7.0-rc.1 h1:YojYx61/OLFsiv6Rw1Z96LpldJIy31o+UHmwAUMJ6/U=
|
github.com/golang/mock v1.7.0-rc.1 h1:YojYx61/OLFsiv6Rw1Z96LpldJIy31o+UHmwAUMJ6/U=
|
||||||
github.com/golang/mock v1.7.0-rc.1/go.mod h1:s42URUywIqd+OcERslBJvOjepvNymP31m3q8d/GkuRs=
|
github.com/golang/mock v1.7.0-rc.1/go.mod h1:s42URUywIqd+OcERslBJvOjepvNymP31m3q8d/GkuRs=
|
||||||
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
||||||
@@ -71,8 +73,8 @@ github.com/refraction-networking/utls v1.8.3-0.20260301010127-aa6edf4b11af h1:er
|
|||||||
github.com/refraction-networking/utls v1.8.3-0.20260301010127-aa6edf4b11af/go.mod h1:jkSOEkLqn+S/jtpEHPOsVv/4V4EVnelwbMQl4vCWXAM=
|
github.com/refraction-networking/utls v1.8.3-0.20260301010127-aa6edf4b11af/go.mod h1:jkSOEkLqn+S/jtpEHPOsVv/4V4EVnelwbMQl4vCWXAM=
|
||||||
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
|
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
|
||||||
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
|
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
|
||||||
github.com/rogpeppe/go-internal v1.16.0 h1:O9DK+vNMDVGLr2BeZqmpLeMjiMNkuXfcqntWbZV6S5g=
|
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
||||||
github.com/rogpeppe/go-internal v1.16.0/go.mod h1:DrUVZyrJU+txYW5/1kwtXQSMFio52ZOxX7yM1VHvnxs=
|
github.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog=
|
||||||
github.com/sagernet/sing v0.5.1 h1:mhL/MZVq0TjuvHcpYcFtmSD1BFOxZ/+8ofbNZcg1k1Y=
|
github.com/sagernet/sing v0.5.1 h1:mhL/MZVq0TjuvHcpYcFtmSD1BFOxZ/+8ofbNZcg1k1Y=
|
||||||
github.com/sagernet/sing v0.5.1/go.mod h1:ARkL0gM13/Iv5VCZmci/NuoOlePoIsW0m7BWfln/Hak=
|
github.com/sagernet/sing v0.5.1/go.mod h1:ARkL0gM13/Iv5VCZmci/NuoOlePoIsW0m7BWfln/Hak=
|
||||||
github.com/sagernet/sing-shadowsocks v0.2.7 h1:zaopR1tbHEw5Nk6FAkM05wCslV6ahVegEZaKMv9ipx8=
|
github.com/sagernet/sing-shadowsocks v0.2.7 h1:zaopR1tbHEw5Nk6FAkM05wCslV6ahVegEZaKMv9ipx8=
|
||||||
@@ -85,12 +87,21 @@ github.com/vishvananda/netns v0.0.5 h1:DfiHV+j8bA32MFM7bfEunvT8IAqQ/NzSJHtcmW5zd
|
|||||||
github.com/vishvananda/netns v0.0.5/go.mod h1:SpkAiCQRtJ6TvvxPnOSyH3BMl6unz3xZlaprSwhNNJM=
|
github.com/vishvananda/netns v0.0.5/go.mod h1:SpkAiCQRtJ6TvvxPnOSyH3BMl6unz3xZlaprSwhNNJM=
|
||||||
github.com/wlynxg/anet v0.0.5 h1:J3VJGi1gvo0JwZ/P1/Yc/8p63SoW98B5dHkYDmpgvvU=
|
github.com/wlynxg/anet v0.0.5 h1:J3VJGi1gvo0JwZ/P1/Yc/8p63SoW98B5dHkYDmpgvvU=
|
||||||
github.com/wlynxg/anet v0.0.5/go.mod h1:eay5PRQr7fIVAMbTbchTnO9gG65Hg/uYGdc7mguHxoA=
|
github.com/wlynxg/anet v0.0.5/go.mod h1:eay5PRQr7fIVAMbTbchTnO9gG65Hg/uYGdc7mguHxoA=
|
||||||
github.com/xtls/reality v0.0.0-20260908062103-8cdf7bf9c7f0 h1:rb+fKQFhz+5I2PPuQsNYxI5mUU840XWYtRF0ZBjvkws=
|
github.com/xtls/reality v0.0.0-20260322125925-9234c772ba8f h1:iy2JRioxmUpoJ3SzbFPyTxHZMbR/rSHP7dOOgYaq1O8=
|
||||||
github.com/xtls/reality v0.0.0-20260908062103-8cdf7bf9c7f0/go.mod h1:DsJblcWDGt76+FVqBVwbwRhxyyNJsGV48gJLch0OOWI=
|
github.com/xtls/reality v0.0.0-20260322125925-9234c772ba8f/go.mod h1:DsJblcWDGt76+FVqBVwbwRhxyyNJsGV48gJLch0OOWI=
|
||||||
github.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
|
github.com/yuin/goldmark v1.4.1/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
|
||||||
github.com/yuin/gopher-lua v0.0.0-20190206043414-8bfc7677f583/go.mod h1:gqRgreBUhTSL0GeU64rtZ3Uq3wtjOa/TB2YfrtkCbVQ=
|
go.opentelemetry.io/auto/sdk v1.2.1 h1:jXsnJ4Lmnqd11kwkBV2LgLoFMZKizbCi5fNZ/ipaZ64=
|
||||||
github.com/yuin/gopher-lua v1.1.2 h1:yF/FjE3hD65tBbt0VXLE13HWS9h34fdzJmrWRXwobGA=
|
go.opentelemetry.io/auto/sdk v1.2.1/go.mod h1:KRTj+aOaElaLi+wW1kO/DZRXwkF4C5xPbEe3ZiIhN7Y=
|
||||||
github.com/yuin/gopher-lua v1.1.2/go.mod h1:7aRmXIWl37SqRf0koeyylBEzJ+aPt8A+mmkQ4f1ntR8=
|
go.opentelemetry.io/otel v1.44.0 h1:JjwHmHpA4iZ3wBxluu2fbbE7j4kqlE8jXyAyPXH7HqU=
|
||||||
|
go.opentelemetry.io/otel v1.44.0/go.mod h1:BMgjTHL9WPRlRjL2oZCBTL4whCGtXch2H4BhOPIAyYc=
|
||||||
|
go.opentelemetry.io/otel/metric v1.44.0 h1:1w0gILTcHdr3YI+ixLyjemwrVnsMURbTZFrSYCdDdmc=
|
||||||
|
go.opentelemetry.io/otel/metric v1.44.0/go.mod h1:8O7hanEPBNgEMmybD3s2VBKcgWOCsA6tzHBPODAiquo=
|
||||||
|
go.opentelemetry.io/otel/sdk v1.44.0 h1:nHYwb9lK+fJPU/dnT6s7W7Z8itMWyqrnVfbheVYrZ58=
|
||||||
|
go.opentelemetry.io/otel/sdk v1.44.0/go.mod h1:Osuydd3Se74nqjAKxid74N5eC+jfEqfTegHRnq58oK0=
|
||||||
|
go.opentelemetry.io/otel/sdk/metric v1.44.0 h1:3LlKgI+VjbVsjNRFZJZAJ30WjXC5VkNRks6si09iEfI=
|
||||||
|
go.opentelemetry.io/otel/sdk/metric v1.44.0/go.mod h1:5B5pMARnXxKhltooO4xUuCBorl65a4EpnTalObqOigA=
|
||||||
|
go.opentelemetry.io/otel/trace v1.44.0 h1:jxF5CsGYCe74MCRx2X4g7WsY/VBKRqqpNvXlX/6gtIk=
|
||||||
|
go.opentelemetry.io/otel/trace v1.44.0/go.mod h1:oLl1jrMQAVo6v3GAggN+1VH9VIz9iUSvW53sW1Q8PIE=
|
||||||
go.uber.org/mock v0.5.2 h1:LbtPTcP8A5k9WPXj54PPPbjcI4Y6lhyOZXn+VS7wNko=
|
go.uber.org/mock v0.5.2 h1:LbtPTcP8A5k9WPXj54PPPbjcI4Y6lhyOZXn+VS7wNko=
|
||||||
go.uber.org/mock v0.5.2/go.mod h1:wLlUxC2vVTPTaE3UD51E0BGOAElKrILxhVSDYQLld5o=
|
go.uber.org/mock v0.5.2/go.mod h1:wLlUxC2vVTPTaE3UD51E0BGOAElKrILxhVSDYQLld5o=
|
||||||
go.yaml.in/yaml/v3 v3.0.5 h1:N6y/pJk8buWs9NY5ERU2HSMfm+IuD/OtfdAnq6kESPw=
|
go.yaml.in/yaml/v3 v3.0.5 h1:N6y/pJk8buWs9NY5ERU2HSMfm+IuD/OtfdAnq6kESPw=
|
||||||
@@ -99,8 +110,8 @@ go4.org/netipx v0.0.0-20231129151722-fdeea329fbba h1:0b9z3AuHCjxk0x/opv64kcgZLBs
|
|||||||
go4.org/netipx v0.0.0-20231129151722-fdeea329fbba/go.mod h1:PLyyIXexvUFg3Owu6p/WfdlivPbZJsZdgWZlrGope/Y=
|
go4.org/netipx v0.0.0-20231129151722-fdeea329fbba/go.mod h1:PLyyIXexvUFg3Owu6p/WfdlivPbZJsZdgWZlrGope/Y=
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
golang.org/x/crypto v0.57.0 h1:3ZVCjf8Ggz7zneR/EHRVx68Ctf+2pmIMP2UFhh9cC6M=
|
golang.org/x/crypto v0.55.0 h1:+KWHjbgOaAQ66dh/YlkZKHlz9ZUlq61AFirAR9ntP8M=
|
||||||
golang.org/x/crypto v0.57.0/go.mod h1:Fdz0i5U6CoizGwLda9DttjSk6qlZo25zYNtR+ycvuZA=
|
golang.org/x/crypto v0.55.0/go.mod h1:uq0V9dE/fzQuJtbnL+2EhWOE63vo164FY8xqEnV9xis=
|
||||||
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842 h1:vr/HnozRka3pE4EsMEg1lgkXJkTFJCVUX+S/ZT6wYzM=
|
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842 h1:vr/HnozRka3pE4EsMEg1lgkXJkTFJCVUX+S/ZT6wYzM=
|
||||||
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842/go.mod h1:XtvwrStGgqGPLc4cjQfWqZHG1YFdYs6swckp8vpsjnc=
|
golang.org/x/exp v0.0.0-20240506185415-9bf2ced13842/go.mod h1:XtvwrStGgqGPLc4cjQfWqZHG1YFdYs6swckp8vpsjnc=
|
||||||
golang.org/x/lint v0.0.0-20200302205851-738671d3881b/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=
|
golang.org/x/lint v0.0.0-20200302205851-738671d3881b/go.mod h1:3xt1FjdF8hUf6vQPIChWIBhFzV8gjjsPE/fR3IyQdNY=
|
||||||
@@ -109,13 +120,12 @@ golang.org/x/mod v0.5.1/go.mod h1:5OXOZSfqPIIbmVBIIKWRFfZjPR0E5r58TLhUjH0a2Ro=
|
|||||||
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
golang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
golang.org/x/net v0.0.0-20211015210444-4f30a5c0130f/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
|
||||||
golang.org/x/net v0.59.0 h1:5zfYln+w5XCxwrnMMJPufRgNoXEaGxl0wo5GqPXyues=
|
golang.org/x/net v0.58.0 h1:ynWG7rqYi4ccpTEuPZ2QGWHktVEM9DMCj9yzDE0Q7To=
|
||||||
golang.org/x/net v0.59.0/go.mod h1:2DA/G1UfVbCpQPeWTmMPGY7Cs2PkBkwu743bVX5PIVg=
|
golang.org/x/net v0.58.0/go.mod h1:YwCddHnFlT7eLQqVprV19OnhLGtc5xOKgE0RyqgfWAU=
|
||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.23.0 h1:KameEIfc1IkluZyXWLn39Wd4tURc6GbCiISGiZm2bQk=
|
golang.org/x/sync v0.22.0 h1:SZjpbeLmrCk4xhRSZFNZW5gFUeCeFgjekvI/+gfScek=
|
||||||
golang.org/x/sync v0.23.0/go.mod h1:sUUOizhqBxiL6pEWpqNLUiaJn1ShEbZ6BBqskPbjZm0=
|
golang.org/x/sync v0.22.0/go.mod h1:9xrNwdLfx4jkKbNva9FpL6vEN7evnE43NNNJQ2LF3+0=
|
||||||
golang.org/x/sys v0.0.0-20190204203706-41f3e6584952/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
@@ -123,22 +133,20 @@ golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7w
|
|||||||
golang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20211019181941-9d821ace8654/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.48.0 h1:bbX/i/6MgT9BVLM9RT1thmxL04yeTAhbEz4SyadbXoo=
|
golang.org/x/sys v0.47.0 h1:o7XGOvZQCADBQQ4Y7VNq2dRWQR7JmOUW8Kxx4ZsNgWs=
|
||||||
golang.org/x/sys v0.48.0/go.mod h1:hNLxWAXmnKAxqDtdwIYC4bM9oQPEecfsnNMuSxOs3og=
|
golang.org/x/sys v0.47.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
|
||||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
golang.org/x/text v0.42.0 h1:JbOZXgfeCPU9gacVtYliJqOhD+zhrEqK4LfdpmlUZqI=
|
golang.org/x/text v0.41.0 h1:vz/seA0lnX87Othu2f/0L24RcgrXD9/YFTSuGjj3rH8=
|
||||||
golang.org/x/text v0.42.0/go.mod h1:ojzP1Z+2QtioaF8DTtO8K5q7JWVVYwZKenzujK0Zd0E=
|
golang.org/x/text v0.41.0/go.mod h1:jvf1O8ajNzZqhSrQBPbutR/EB83Cc0CFrezNQIwbb5M=
|
||||||
golang.org/x/time v0.14.0 h1:MRx4UaLrDotUKUdCIqzPC48t1Y9hANFKIRpNx+Te8PI=
|
golang.org/x/time v0.14.0 h1:MRx4UaLrDotUKUdCIqzPC48t1Y9hANFKIRpNx+Te8PI=
|
||||||
golang.org/x/time v0.14.0/go.mod h1:eL/Oa2bBBK0TkX57Fyni+NgnyQQN4LitPmob2Hjnqw4=
|
golang.org/x/time v0.14.0/go.mod h1:eL/Oa2bBBK0TkX57Fyni+NgnyQQN4LitPmob2Hjnqw4=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
golang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
|
golang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
|
||||||
golang.org/x/tools v0.1.8/go.mod h1:nABZi5QlRsZVlzPpHl034qft6wpY4eDcsTt5AaioBiU=
|
golang.org/x/tools v0.1.8/go.mod h1:nABZi5QlRsZVlzPpHl034qft6wpY4eDcsTt5AaioBiU=
|
||||||
golang.org/x/tools v0.49.0 h1:3NI7VXzL9+1WZD52Dx2ttoPwD5DWrFGpl9mFZDlmisI=
|
|
||||||
golang.org/x/tools v0.49.0/go.mod h1:SJNXV9DBKT0UbdttsQjbfJlAE/q+y36++zo3uL3N0Oo=
|
|
||||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
@@ -146,14 +154,14 @@ golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2 h1:B82qJJgjvYKsXS9jeu
|
|||||||
golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2/go.mod h1:deeaetjYA+DHMHg+sMSMI58GrEteJUUzzw7en6TJQcI=
|
golang.zx2c4.com/wintun v0.0.0-20230126152724-0fa3db229ce2/go.mod h1:deeaetjYA+DHMHg+sMSMI58GrEteJUUzzw7en6TJQcI=
|
||||||
golang.zx2c4.com/wireguard v0.0.0-20250521234502-f333402bd9cb h1:whnFRlWMcXI9d+ZbWg+4sHnLp52d5yiIPUxMBSt4X9A=
|
golang.zx2c4.com/wireguard v0.0.0-20250521234502-f333402bd9cb h1:whnFRlWMcXI9d+ZbWg+4sHnLp52d5yiIPUxMBSt4X9A=
|
||||||
golang.zx2c4.com/wireguard v0.0.0-20250521234502-f333402bd9cb/go.mod h1:rpwXGsirqLqN2L0JDJQlwOboGHmptD5ZD6T2VmcqhTw=
|
golang.zx2c4.com/wireguard v0.0.0-20250521234502-f333402bd9cb/go.mod h1:rpwXGsirqLqN2L0JDJQlwOboGHmptD5ZD6T2VmcqhTw=
|
||||||
golang.zx2c4.com/wireguard/windows v1.1.1 h1:8/H97U1v1PNDNcBsMZgU3KFuND9MQdTsU2NOwmCXArE=
|
golang.zx2c4.com/wireguard/windows v1.0.1 h1:eOxiDVbywPC+ZQqvdCK7x+ZwWXKbYv50TtH8ysFIbw8=
|
||||||
golang.zx2c4.com/wireguard/windows v1.1.1/go.mod h1:+fbT3FFdX4zzYDLwJh5+HPEcNN/3HyNdzhNSVsQM+zs=
|
golang.zx2c4.com/wireguard/windows v1.0.1/go.mod h1:+fbT3FFdX4zzYDLwJh5+HPEcNN/3HyNdzhNSVsQM+zs=
|
||||||
gonum.org/v1/gonum v0.17.0 h1:VbpOemQlsSMrYmn7T2OUvQ4dqxQXU+ouZFQsZOx50z4=
|
gonum.org/v1/gonum v0.17.0 h1:VbpOemQlsSMrYmn7T2OUvQ4dqxQXU+ouZFQsZOx50z4=
|
||||||
gonum.org/v1/gonum v0.17.0/go.mod h1:El3tOrEuMpv2UdMrbNlKEh9vd86bmQ6vqIcDwxEOc1E=
|
gonum.org/v1/gonum v0.17.0/go.mod h1:El3tOrEuMpv2UdMrbNlKEh9vd86bmQ6vqIcDwxEOc1E=
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20260706201446-f0a921348800 h1:qEHAMpSaUhtD0p3NbEEI83HwNGFxEwaSJ1G9PLnCBZE=
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20260526163538-3dc84a4a5aaa h1:mZHHdPZl0dbGHCflZgAq/Q468DWVFcU2whhB2KAo8fk=
|
||||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20260706201446-f0a921348800/go.mod h1:4Hqkh8ycfw05ld/3BWL7rJOSfebL2Q+DVDeRgYgxUU8=
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20260526163538-3dc84a4a5aaa/go.mod h1:4Hqkh8ycfw05ld/3BWL7rJOSfebL2Q+DVDeRgYgxUU8=
|
||||||
google.golang.org/grpc v1.84.0 h1:soMyaPJ8pAak5PIQ0DGBUir0XRo2fRoMqhNWMLlLxO0=
|
google.golang.org/grpc v1.83.1 h1:HIO0+BEtBP6soyqvqC8sNUjZ7bTs+0hFQuFF+RAy++Y=
|
||||||
google.golang.org/grpc v1.84.0/go.mod h1:ljCht0DrxQrXBDRTZp52Qxh3Ffk8CdYm2sj4O2QN2C0=
|
google.golang.org/grpc v1.83.1/go.mod h1:kDyl6SKsiHKt0uylY5gtn5cEjkrIOhQOGDgIc4JGwzQ=
|
||||||
google.golang.org/protobuf v1.36.12 h1:pJOKDDOyeXErUroCihFAd5LQuwXBSpVnKGrj5o/fwxc=
|
google.golang.org/protobuf v1.36.12 h1:pJOKDDOyeXErUroCihFAd5LQuwXBSpVnKGrj5o/fwxc=
|
||||||
google.golang.org/protobuf v1.36.12/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
google.golang.org/protobuf v1.36.12/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
@@ -166,9 +174,5 @@ gvisor.dev/gvisor v0.0.0-20260122175437-89a5d21be8f0 h1:Lk6hARj5UPY47dBep70OD/TI
|
|||||||
gvisor.dev/gvisor v0.0.0-20260122175437-89a5d21be8f0/go.mod h1:QkHjoMIBaYtpVufgwv3keYAbln78mBoCuShZrPrer1Q=
|
gvisor.dev/gvisor v0.0.0-20260122175437-89a5d21be8f0/go.mod h1:QkHjoMIBaYtpVufgwv3keYAbln78mBoCuShZrPrer1Q=
|
||||||
h12.io/socks v1.0.3 h1:Ka3qaQewws4j4/eDQnOdpr4wXsC//dXtWvftlIcCQUo=
|
h12.io/socks v1.0.3 h1:Ka3qaQewws4j4/eDQnOdpr4wXsC//dXtWvftlIcCQUo=
|
||||||
h12.io/socks v1.0.3/go.mod h1:AIhxy1jOId/XCz9BO+EIgNL2rQiPTBNnOfnVnQ+3Eck=
|
h12.io/socks v1.0.3/go.mod h1:AIhxy1jOId/XCz9BO+EIgNL2rQiPTBNnOfnVnQ+3Eck=
|
||||||
layeh.com/gopher-luar v1.0.11 h1:8zJudpKI6HWkoh9eyyNFaTM79PY6CAPcIr6X/KTiliw=
|
|
||||||
layeh.com/gopher-luar v1.0.11/go.mod h1:TPnIVCZ2RJBndm7ohXyaqfhzjlZ+OA2SZR/YwL8tECk=
|
|
||||||
lukechampine.com/blake3 v1.4.1 h1:I3Smz7gso8w4/TunLKec6K2fn+kyKtDxr/xcQEN84Wg=
|
lukechampine.com/blake3 v1.4.1 h1:I3Smz7gso8w4/TunLKec6K2fn+kyKtDxr/xcQEN84Wg=
|
||||||
lukechampine.com/blake3 v1.4.1/go.mod h1:QFosUxmjB8mnrWFSNwKmvxHpfY72bmD2tQ0kBMM3kwo=
|
lukechampine.com/blake3 v1.4.1/go.mod h1:QFosUxmjB8mnrWFSNwKmvxHpfY72bmD2tQ0kBMM3kwo=
|
||||||
mvdan.cc/gofumpt v0.12.0 h1:1Lbudkz2kpM9Cjz2pL4M19u7q+GaEhCTNf7N9mfpcho=
|
|
||||||
mvdan.cc/gofumpt v0.12.0/go.mod h1:SmBHHrljiZu/uoypeKup3rFzP6eoC9UwCp2iH5E3jZA=
|
|
||||||
|
|||||||
@@ -14,11 +14,9 @@ import (
|
|||||||
"github.com/xtls/xray-core/common/errors"
|
"github.com/xtls/xray-core/common/errors"
|
||||||
"github.com/xtls/xray-core/common/geodata"
|
"github.com/xtls/xray-core/common/geodata"
|
||||||
"github.com/xtls/xray-core/common/net"
|
"github.com/xtls/xray-core/common/net"
|
||||||
"github.com/xtls/xray-core/common/platform"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type NameServerConfig struct {
|
type NameServerConfig struct {
|
||||||
ID string `json:"id"`
|
|
||||||
Address *Address `json:"address"`
|
Address *Address `json:"address"`
|
||||||
ClientIP *Address `json:"clientIp"`
|
ClientIP *Address `json:"clientIp"`
|
||||||
Port uint16 `json:"port"`
|
Port uint16 `json:"port"`
|
||||||
@@ -45,7 +43,6 @@ func (c *NameServerConfig) UnmarshalJSON(data []byte) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var advanced struct {
|
var advanced struct {
|
||||||
ID string `json:"id"`
|
|
||||||
Address *Address `json:"address"`
|
Address *Address `json:"address"`
|
||||||
ClientIP *Address `json:"clientIp"`
|
ClientIP *Address `json:"clientIp"`
|
||||||
Port uint16 `json:"port"`
|
Port uint16 `json:"port"`
|
||||||
@@ -63,7 +60,6 @@ func (c *NameServerConfig) UnmarshalJSON(data []byte) error {
|
|||||||
UnexpectedIPs StringList `json:"unexpectedIPs"`
|
UnexpectedIPs StringList `json:"unexpectedIPs"`
|
||||||
}
|
}
|
||||||
if err := json.Unmarshal(data, &advanced); err == nil {
|
if err := json.Unmarshal(data, &advanced); err == nil {
|
||||||
c.ID = advanced.ID
|
|
||||||
c.Address = advanced.Address
|
c.Address = advanced.Address
|
||||||
c.ClientIP = advanced.ClientIP
|
c.ClientIP = advanced.ClientIP
|
||||||
c.Port = advanced.Port
|
c.Port = advanced.Port
|
||||||
@@ -138,7 +134,6 @@ func (c *NameServerConfig) Build() (*dns.NameServer, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return &dns.NameServer{
|
return &dns.NameServer{
|
||||||
Id: c.ID,
|
|
||||||
Address: &net.Endpoint{
|
Address: &net.Endpoint{
|
||||||
Network: net.Network_UDP,
|
Network: net.Network_UDP,
|
||||||
Address: c.Address.Build(),
|
Address: c.Address.Build(),
|
||||||
@@ -164,7 +159,6 @@ func (c *NameServerConfig) Build() (*dns.NameServer, error) {
|
|||||||
// DNSConfig is a JSON serializable object for dns.Config
|
// DNSConfig is a JSON serializable object for dns.Config
|
||||||
type DNSConfig struct {
|
type DNSConfig struct {
|
||||||
Servers []*NameServerConfig `json:"servers"`
|
Servers []*NameServerConfig `json:"servers"`
|
||||||
Script string `json:"script"`
|
|
||||||
Hosts *HostsWrapper `json:"hosts"`
|
Hosts *HostsWrapper `json:"hosts"`
|
||||||
ClientIP *Address `json:"clientIp"`
|
ClientIP *Address `json:"clientIp"`
|
||||||
Tag string `json:"tag"`
|
Tag string `json:"tag"`
|
||||||
@@ -284,14 +278,6 @@ func (c *DNSConfig) Build() (*dns.Config, error) {
|
|||||||
QueryStrategy: resolveQueryStrategy(c.QueryStrategy),
|
QueryStrategy: resolveQueryStrategy(c.QueryStrategy),
|
||||||
}
|
}
|
||||||
|
|
||||||
if c.Script != "" {
|
|
||||||
path, err := platform.ResolveLuaFile(c.Script)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("failed to resolve DNS script: ", c.Script).Base(err)
|
|
||||||
}
|
|
||||||
config.Script = path
|
|
||||||
}
|
|
||||||
|
|
||||||
if c.ClientIP != nil {
|
if c.ClientIP != nil {
|
||||||
if !c.ClientIP.Family().IsIP() {
|
if !c.ClientIP.Family().IsIP() {
|
||||||
return nil, errors.New("not an IP address:", c.ClientIP.String())
|
return nil, errors.New("not an IP address:", c.ClientIP.String())
|
||||||
|
|||||||
@@ -2,8 +2,6 @@ package conf_test
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/google/go-cmp/cmp"
|
"github.com/google/go-cmp/cmp"
|
||||||
@@ -124,51 +122,3 @@ func TestDNSConfigParsing(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDNSScriptConfig(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
t.Setenv("xray.location.confdir", dir)
|
|
||||||
path := filepath.Join(dir, "lookup.lua")
|
|
||||||
if err := os.WriteFile(path, []byte("function HandleDNSQuery(domain, ipv4, ipv6, fake) end"), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name string
|
|
||||||
script string
|
|
||||||
wantError bool
|
|
||||||
}{
|
|
||||||
{"relative", "lookup.lua", false},
|
|
||||||
{"absolute", path, false},
|
|
||||||
{"missing", "missing.lua", true},
|
|
||||||
{"directory", dir, true},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
built, err := (&DNSConfig{Script: tc.script}).Build()
|
|
||||||
if tc.wantError {
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("Build accepted an invalid script path")
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if built.Script != path {
|
|
||||||
t.Fatalf("script path = %q, want %q", built.Script, path)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
var parsed DNSConfig
|
|
||||||
if err := json.Unmarshal([]byte(`{"servers":[{"id":"primary","address":"1.1.1.1"}]}`), &parsed); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
built, err := parsed.Build()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if len(built.NameServer) != 1 || built.NameServer[0].Id != "primary" {
|
|
||||||
t.Fatalf("nameserver IDs = %v, want primary", built.NameServer)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,37 +0,0 @@
|
|||||||
package conf
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/netip"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/protocol"
|
|
||||||
"github.com/xtls/xray-core/proxy/masque"
|
|
||||||
"google.golang.org/protobuf/proto"
|
|
||||||
)
|
|
||||||
|
|
||||||
type MasqueClientConfig struct {
|
|
||||||
Address *Address `json:"address"`
|
|
||||||
Port uint16 `json:"port"`
|
|
||||||
RemoteDNS []string `json:"remoteDNS"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *MasqueClientConfig) Build() (proto.Message, error) {
|
|
||||||
if c.Address == nil {
|
|
||||||
return nil, errors.New(`MASQUE: "address" is not set`)
|
|
||||||
}
|
|
||||||
if c.Port == 0 {
|
|
||||||
return nil, errors.New(`MASQUE: "port" is not set`)
|
|
||||||
}
|
|
||||||
for _, s := range c.RemoteDNS {
|
|
||||||
if _, err := netip.ParseAddr(s); err != nil {
|
|
||||||
return nil, errors.New(`MASQUE: invalid "remoteDNS" `, s).Base(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return &masque.ClientConfig{
|
|
||||||
Server: &protocol.ServerEndpoint{
|
|
||||||
Address: c.Address.Build(),
|
|
||||||
Port: uint32(c.Port),
|
|
||||||
},
|
|
||||||
RemoteDns: c.RemoteDNS,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
@@ -1,85 +0,0 @@
|
|||||||
package conf_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
. "github.com/xtls/xray-core/infra/conf"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/masque"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestMasqueConfig(t *testing.T) {
|
|
||||||
creator := func() Buildable {
|
|
||||||
return new(MasqueConfig)
|
|
||||||
}
|
|
||||||
|
|
||||||
runMultiTestCase(t, []TestCase{
|
|
||||||
{
|
|
||||||
Input: `{}`,
|
|
||||||
Parser: loadJSON(creator),
|
|
||||||
Output: &masque.Config{Path: "/.well-known/masque/ip/*/*/"},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
Input: `{
|
|
||||||
"host": "example.com:8443",
|
|
||||||
"path": "/.well-known/masque/ip/{target}/{ipproto}/",
|
|
||||||
"headers": {"Authorization": "Basic dTpw"}
|
|
||||||
}`,
|
|
||||||
Parser: loadJSON(creator),
|
|
||||||
Output: &masque.Config{
|
|
||||||
Host: "example.com:8443",
|
|
||||||
Path: "/.well-known/masque/ip/*/*/",
|
|
||||||
Headers: map[string]string{"Authorization": "Basic dTpw"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{
|
|
||||||
Input: `{"path": "/masque/ip{?target,ipproto}"}`,
|
|
||||||
Parser: loadJSON(creator),
|
|
||||||
Output: &masque.Config{Path: "/masque/ip?target=*&ipproto=*"},
|
|
||||||
},
|
|
||||||
})
|
|
||||||
|
|
||||||
for _, input := range []string{
|
|
||||||
`{"path": "/masque/{target}/{ipproto}/{dns}"}`,
|
|
||||||
`{"path": "masque"}`,
|
|
||||||
`{"host": "example.com/path"}`,
|
|
||||||
`{"headers": {"host": "example.com"}}`,
|
|
||||||
`{"headers": {"Capsule-Protocol": "?0"}}`,
|
|
||||||
`{"headers": {"X Token": "a"}}`,
|
|
||||||
`{"headers": {"X-Token": "a\r\nb"}}`,
|
|
||||||
} {
|
|
||||||
if _, err := loadJSON(creator)(input); err == nil {
|
|
||||||
t.Errorf("expected an error for %s", input)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestMasqueOutboundConfig(t *testing.T) {
|
|
||||||
build := func(s string) error {
|
|
||||||
c := new(OutboundDetourConfig)
|
|
||||||
if err := json.Unmarshal([]byte(s), c); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err := c.Build()
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := build(`{
|
|
||||||
"protocol": "masque",
|
|
||||||
"settings": {"address": "example.com", "port": 443},
|
|
||||||
"streamSettings": {"network": "masque", "security": "tls"},
|
|
||||||
"mux": {"enabled": false, "concurrency": -1}
|
|
||||||
}`); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
for _, input := range []string{
|
|
||||||
`{"protocol": "masque", "settings": {"address": "example.com"}, "streamSettings": {"network": "masque", "security": "tls"}}`,
|
|
||||||
`{"protocol": "masque", "settings": {"address": "example.com", "port": 443}, "streamSettings": {"network": "masque", "security": "tls"}, "mux": {"enabled": true}}`,
|
|
||||||
`{"protocol": "masque", "settings": {"address": "example.com", "port": 443}, "streamSettings": {"network": "masque", "security": "tls"}, "mux": {"enabled": true, "concurrency": -1}}`,
|
|
||||||
`{"protocol": "freedom", "streamSettings": {"network": "masque", "security": "tls"}}`,
|
|
||||||
} {
|
|
||||||
if err := build(input); err == nil {
|
|
||||||
t.Errorf("expected an error for %s", input)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -7,7 +7,6 @@ import (
|
|||||||
"github.com/xtls/xray-core/app/router"
|
"github.com/xtls/xray-core/app/router"
|
||||||
"github.com/xtls/xray-core/common/errors"
|
"github.com/xtls/xray-core/common/errors"
|
||||||
"github.com/xtls/xray-core/common/geodata"
|
"github.com/xtls/xray-core/common/geodata"
|
||||||
"github.com/xtls/xray-core/common/platform"
|
|
||||||
"github.com/xtls/xray-core/common/serial"
|
"github.com/xtls/xray-core/common/serial"
|
||||||
|
|
||||||
"google.golang.org/protobuf/proto"
|
"google.golang.org/protobuf/proto"
|
||||||
@@ -73,7 +72,6 @@ type RouterConfig struct {
|
|||||||
RuleList []json.RawMessage `json:"rules"`
|
RuleList []json.RawMessage `json:"rules"`
|
||||||
DomainStrategy *string `json:"domainStrategy"`
|
DomainStrategy *string `json:"domainStrategy"`
|
||||||
Balancers []*BalancingRule `json:"balancers"`
|
Balancers []*BalancingRule `json:"balancers"`
|
||||||
Script string `json:"script"`
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *RouterConfig) getDomainStrategy() router.Config_DomainStrategy {
|
func (c *RouterConfig) getDomainStrategy() router.Config_DomainStrategy {
|
||||||
@@ -94,15 +92,6 @@ func (c *RouterConfig) getDomainStrategy() router.Config_DomainStrategy {
|
|||||||
|
|
||||||
func (c *RouterConfig) Build() (*router.Config, error) {
|
func (c *RouterConfig) Build() (*router.Config, error) {
|
||||||
config := new(router.Config)
|
config := new(router.Config)
|
||||||
|
|
||||||
if c.Script != "" {
|
|
||||||
path, err := platform.ResolveLuaFile(c.Script)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("failed to resolve routing script").Base(err)
|
|
||||||
}
|
|
||||||
config.Script = path
|
|
||||||
}
|
|
||||||
|
|
||||||
config.DomainStrategy = c.getDomainStrategy()
|
config.DomainStrategy = c.getDomainStrategy()
|
||||||
|
|
||||||
var rawRuleList []json.RawMessage
|
var rawRuleList []json.RawMessage
|
||||||
|
|||||||
@@ -2,8 +2,6 @@ package conf_test
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
_ "unsafe"
|
_ "unsafe"
|
||||||
@@ -238,39 +236,3 @@ func TestRouterConfig(t *testing.T) {
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestRouterScriptConfig(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
t.Setenv("xray.location.confdir", dir)
|
|
||||||
path := filepath.Join(dir, "route.lua")
|
|
||||||
if err := os.WriteFile(path, []byte("function HandleRoute() end"), 0o600); err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tc := range []struct {
|
|
||||||
name string
|
|
||||||
script string
|
|
||||||
wantError bool
|
|
||||||
}{
|
|
||||||
{"relative", "route.lua", false},
|
|
||||||
{"absolute", path, false},
|
|
||||||
{"missing", "missing.lua", true},
|
|
||||||
{"directory", dir, true},
|
|
||||||
} {
|
|
||||||
t.Run(tc.name, func(t *testing.T) {
|
|
||||||
built, err := (&RouterConfig{Script: tc.script}).Build()
|
|
||||||
if tc.wantError {
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("Build accepted invalid script path")
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
if built.Script != path {
|
|
||||||
t.Fatalf("script path = %q, want %q", built.Script, path)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -23,7 +23,6 @@ import (
|
|||||||
"github.com/xtls/xray-core/transport/internet/finalmask/realm"
|
"github.com/xtls/xray-core/transport/internet/finalmask/realm"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/salamander"
|
"github.com/xtls/xray-core/transport/internet/finalmask/salamander"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/sudoku"
|
"github.com/xtls/xray-core/transport/internet/finalmask/sudoku"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/udphop"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/xdns"
|
"github.com/xtls/xray-core/transport/internet/finalmask/xdns"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/xicmp"
|
"github.com/xtls/xray-core/transport/internet/finalmask/xicmp"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/xmc"
|
"github.com/xtls/xray-core/transport/internet/finalmask/xmc"
|
||||||
@@ -84,7 +83,6 @@ var (
|
|||||||
"xdns": func() interface{} { return new(Xdns) },
|
"xdns": func() interface{} { return new(Xdns) },
|
||||||
"xicmp": func() interface{} { return new(Xicmp) },
|
"xicmp": func() interface{} { return new(Xicmp) },
|
||||||
"realm": func() interface{} { return new(Realm) },
|
"realm": func() interface{} { return new(Realm) },
|
||||||
"udphop": func() interface{} { return new(UDPHop) },
|
|
||||||
}, "type", "settings")
|
}, "type", "settings")
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -907,52 +905,6 @@ func (c *Realm) Build() (proto.Message, error) {
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type UDPHop struct {
|
|
||||||
Mode string `json:"mode"`
|
|
||||||
Interval Int32Range `json:"interval"`
|
|
||||||
RemoteIPs []string `json:"remoteIPs"`
|
|
||||||
RemotePorts PortList `json:"remotePorts"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *UDPHop) Build() (proto.Message, error) {
|
|
||||||
var local, remote, remoteOnce bool
|
|
||||||
for _, mode := range strings.Split(c.Mode, ",") {
|
|
||||||
switch strings.ToLower(mode) {
|
|
||||||
case "intervallocal":
|
|
||||||
local = true
|
|
||||||
case "intervalremote":
|
|
||||||
remote = true
|
|
||||||
case "perconnremote":
|
|
||||||
remoteOnce = true
|
|
||||||
default:
|
|
||||||
return nil, errors.New("invalid mode ", mode)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
var remoteIPs []string
|
|
||||||
for _, ip := range c.RemoteIPs {
|
|
||||||
prefix, err := netip.ParsePrefix(ip)
|
|
||||||
if err == nil {
|
|
||||||
remoteIPs = append(remoteIPs, prefix.String())
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
addr, err := netip.ParseAddr(ip)
|
|
||||||
if err == nil {
|
|
||||||
remoteIPs = append(remoteIPs, netip.PrefixFrom(addr, addr.BitLen()).String())
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
return nil, errors.New("invalid ip ", ip)
|
|
||||||
}
|
|
||||||
return &udphop.Config{
|
|
||||||
Local: local,
|
|
||||||
Remote: remote,
|
|
||||||
RemoteOnce: remoteOnce,
|
|
||||||
IntervalMin: int64(c.Interval.From),
|
|
||||||
IntervalMax: int64(c.Interval.To),
|
|
||||||
RemoteIPs: remoteIPs,
|
|
||||||
RemotePorts: c.RemotePorts.Build().Ports(),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type Mask struct {
|
type Mask struct {
|
||||||
Type string `json:"type"`
|
Type string `json:"type"`
|
||||||
Settings *json.RawMessage `json:"settings"`
|
Settings *json.RawMessage `json:"settings"`
|
||||||
@@ -986,6 +938,7 @@ type QuicParamsConfig struct {
|
|||||||
BrutalUp Bandwidth `json:"brutalUp"`
|
BrutalUp Bandwidth `json:"brutalUp"`
|
||||||
BrutalDown Bandwidth `json:"brutalDown"`
|
BrutalDown Bandwidth `json:"brutalDown"`
|
||||||
BrutalDisableLossCompensation bool `json:"brutalDisableLossCompensation"`
|
BrutalDisableLossCompensation bool `json:"brutalDisableLossCompensation"`
|
||||||
|
UdpHop UdpHop `json:"udpHop"`
|
||||||
InitStreamReceiveWindow uint64 `json:"initStreamReceiveWindow"`
|
InitStreamReceiveWindow uint64 `json:"initStreamReceiveWindow"`
|
||||||
MaxStreamReceiveWindow uint64 `json:"maxStreamReceiveWindow"`
|
MaxStreamReceiveWindow uint64 `json:"maxStreamReceiveWindow"`
|
||||||
InitConnectionReceiveWindow uint64 `json:"initConnectionReceiveWindow"`
|
InitConnectionReceiveWindow uint64 `json:"initConnectionReceiveWindow"`
|
||||||
|
|||||||
@@ -36,10 +36,6 @@ func (p TransportProtocol) Build() (string, error) {
|
|||||||
return "", errors.PrintRemovedFeatureError("QUIC transport (without web service, etc.)", "XHTTP stream-one H3")
|
return "", errors.PrintRemovedFeatureError("QUIC transport (without web service, etc.)", "XHTTP stream-one H3")
|
||||||
case "hysteria":
|
case "hysteria":
|
||||||
return "hysteria", nil
|
return "hysteria", nil
|
||||||
case "masque":
|
|
||||||
return "masque", nil
|
|
||||||
case "xdrive":
|
|
||||||
return "xdrive", nil
|
|
||||||
default:
|
default:
|
||||||
return "", errors.New("Config: unknown transport protocol: ", p)
|
return "", errors.New("Config: unknown transport protocol: ", p)
|
||||||
}
|
}
|
||||||
@@ -63,8 +59,6 @@ type StreamConfig struct {
|
|||||||
WSSettings *WebSocketConfig `json:"wsSettings"`
|
WSSettings *WebSocketConfig `json:"wsSettings"`
|
||||||
HTTPUPGRADESettings *HttpUpgradeConfig `json:"httpupgradeSettings"`
|
HTTPUPGRADESettings *HttpUpgradeConfig `json:"httpupgradeSettings"`
|
||||||
HysteriaSettings *HysteriaConfig `json:"hysteriaSettings"`
|
HysteriaSettings *HysteriaConfig `json:"hysteriaSettings"`
|
||||||
MASQUESettings *MasqueConfig `json:"masqueSettings"`
|
|
||||||
XDRIVESettings *XDriveConfig `json:"xdriveSettings"`
|
|
||||||
SocketSettings *SocketConfig `json:"sockopt"`
|
SocketSettings *SocketConfig `json:"sockopt"`
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -198,26 +192,6 @@ func (c *StreamConfig) Build() (*internet.StreamConfig, error) {
|
|||||||
Settings: serial.ToTypedMessage(hs),
|
Settings: serial.ToTypedMessage(hs),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
if c.MASQUESettings != nil {
|
|
||||||
ms, err := c.MASQUESettings.Build()
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("Failed to build MASQUE config.").Base(err)
|
|
||||||
}
|
|
||||||
config.TransportSettings = append(config.TransportSettings, &internet.TransportConfig{
|
|
||||||
ProtocolName: "masque",
|
|
||||||
Settings: serial.ToTypedMessage(ms),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if c.XDRIVESettings != nil {
|
|
||||||
xs, err := c.XDRIVESettings.Build()
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("Failed to build XDRIVE config.").Base(err)
|
|
||||||
}
|
|
||||||
config.TransportSettings = append(config.TransportSettings, &internet.TransportConfig{
|
|
||||||
ProtocolName: "xdrive",
|
|
||||||
Settings: serial.ToTypedMessage(xs),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
if c.SocketSettings != nil {
|
if c.SocketSettings != nil {
|
||||||
ss, err := c.SocketSettings.Build()
|
ss, err := c.SocketSettings.Build()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -279,6 +253,10 @@ func (c *StreamConfig) Build() (*internet.StreamConfig, error) {
|
|||||||
return nil, errors.New("unknown congestion control: ", c.FinalMask.QuicParams.Congestion, ", valid values: reno, bbr, brutal, force-brutal")
|
return nil, errors.New("unknown congestion control: ", c.FinalMask.QuicParams.Congestion, ", valid values: reno, bbr, brutal, force-brutal")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (c.FinalMask.QuicParams.UdpHop.Interval.From != 0 && c.FinalMask.QuicParams.UdpHop.Interval.From < 5) || (c.FinalMask.QuicParams.UdpHop.Interval.To != 0 && c.FinalMask.QuicParams.UdpHop.Interval.To < 5) {
|
||||||
|
return nil, errors.New("Interval must be at least 5")
|
||||||
|
}
|
||||||
|
|
||||||
if c.FinalMask.QuicParams.InitStreamReceiveWindow > 0 && c.FinalMask.QuicParams.InitStreamReceiveWindow < 16384 {
|
if c.FinalMask.QuicParams.InitStreamReceiveWindow > 0 && c.FinalMask.QuicParams.InitStreamReceiveWindow < 16384 {
|
||||||
return nil, errors.New("InitStreamReceiveWindow must be at least 16384")
|
return nil, errors.New("InitStreamReceiveWindow must be at least 16384")
|
||||||
}
|
}
|
||||||
@@ -312,20 +290,43 @@ func (c *StreamConfig) Build() (*internet.StreamConfig, error) {
|
|||||||
BrutalUp: up,
|
BrutalUp: up,
|
||||||
BrutalDown: down,
|
BrutalDown: down,
|
||||||
BrutalDisableLossCompensation: c.FinalMask.QuicParams.BrutalDisableLossCompensation,
|
BrutalDisableLossCompensation: c.FinalMask.QuicParams.BrutalDisableLossCompensation,
|
||||||
InitStreamReceiveWindow: c.FinalMask.QuicParams.InitStreamReceiveWindow,
|
UdpHop: &internet.UdpHop{
|
||||||
MaxStreamReceiveWindow: c.FinalMask.QuicParams.MaxStreamReceiveWindow,
|
Ports: c.FinalMask.QuicParams.UdpHop.PortList.Build().Ports(),
|
||||||
InitConnReceiveWindow: c.FinalMask.QuicParams.InitConnectionReceiveWindow,
|
IntervalMin: int64(c.FinalMask.QuicParams.UdpHop.Interval.From),
|
||||||
MaxConnReceiveWindow: c.FinalMask.QuicParams.MaxConnectionReceiveWindow,
|
IntervalMax: int64(c.FinalMask.QuicParams.UdpHop.Interval.To),
|
||||||
MaxIdleTimeout: c.FinalMask.QuicParams.MaxIdleTimeout,
|
},
|
||||||
KeepAlivePeriod: c.FinalMask.QuicParams.KeepAlivePeriod,
|
InitStreamReceiveWindow: c.FinalMask.QuicParams.InitStreamReceiveWindow,
|
||||||
DisablePathMtuDiscovery: c.FinalMask.QuicParams.DisablePathMTUDiscovery,
|
MaxStreamReceiveWindow: c.FinalMask.QuicParams.MaxStreamReceiveWindow,
|
||||||
DisableChromeParrot: c.FinalMask.QuicParams.DisableChromeParrot,
|
InitConnReceiveWindow: c.FinalMask.QuicParams.InitConnectionReceiveWindow,
|
||||||
DisableGSO: c.FinalMask.QuicParams.DisableGSO,
|
MaxConnReceiveWindow: c.FinalMask.QuicParams.MaxConnectionReceiveWindow,
|
||||||
MaxIncomingStreams: c.FinalMask.QuicParams.MaxIncomingStreams,
|
MaxIdleTimeout: c.FinalMask.QuicParams.MaxIdleTimeout,
|
||||||
DisableStatelessReset: c.FinalMask.QuicParams.DisableStatelessReset,
|
KeepAlivePeriod: c.FinalMask.QuicParams.KeepAlivePeriod,
|
||||||
|
DisablePathMtuDiscovery: c.FinalMask.QuicParams.DisablePathMTUDiscovery,
|
||||||
|
DisableChromeParrot: c.FinalMask.QuicParams.DisableChromeParrot,
|
||||||
|
DisableGSO: c.FinalMask.QuicParams.DisableGSO,
|
||||||
|
MaxIncomingStreams: c.FinalMask.QuicParams.MaxIncomingStreams,
|
||||||
|
DisableStatelessReset: c.FinalMask.QuicParams.DisableStatelessReset,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return config, nil
|
return config, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ProxyConfig struct {
|
||||||
|
Tag string `json:"tag"`
|
||||||
|
|
||||||
|
// TransportLayerProxy: For compatibility.
|
||||||
|
TransportLayerProxy bool `json:"transportLayer"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build implements Buildable.
|
||||||
|
func (v *ProxyConfig) Build() (*internet.ProxyConfig, error) {
|
||||||
|
if v.Tag == "" {
|
||||||
|
return nil, errors.New("Proxy tag is not set.")
|
||||||
|
}
|
||||||
|
return &internet.ProxyConfig{
|
||||||
|
Tag: v.Tag,
|
||||||
|
TransportLayerProxy: v.TransportLayerProxy,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
package conf
|
package conf
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"math/big"
|
"math/big"
|
||||||
"net/url"
|
"net/url"
|
||||||
@@ -20,12 +21,9 @@ import (
|
|||||||
"github.com/xtls/xray-core/transport/internet/httpupgrade"
|
"github.com/xtls/xray-core/transport/internet/httpupgrade"
|
||||||
"github.com/xtls/xray-core/transport/internet/hysteria"
|
"github.com/xtls/xray-core/transport/internet/hysteria"
|
||||||
"github.com/xtls/xray-core/transport/internet/kcp"
|
"github.com/xtls/xray-core/transport/internet/kcp"
|
||||||
"github.com/xtls/xray-core/transport/internet/masque"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/splithttp"
|
"github.com/xtls/xray-core/transport/internet/splithttp"
|
||||||
"github.com/xtls/xray-core/transport/internet/tcp"
|
"github.com/xtls/xray-core/transport/internet/tcp"
|
||||||
"github.com/xtls/xray-core/transport/internet/websocket"
|
"github.com/xtls/xray-core/transport/internet/websocket"
|
||||||
"github.com/xtls/xray-core/transport/internet/xdrive"
|
|
||||||
"golang.org/x/net/http/httpguts"
|
|
||||||
"google.golang.org/protobuf/proto"
|
"google.golang.org/protobuf/proto"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -536,6 +534,10 @@ type KCPConfig struct {
|
|||||||
|
|
||||||
// Build implements Buildable.
|
// Build implements Buildable.
|
||||||
func (c *KCPConfig) Build() (proto.Message, error) {
|
func (c *KCPConfig) Build() (proto.Message, error) {
|
||||||
|
if c.HeaderConfig != nil || c.Seed != nil {
|
||||||
|
return nil, errors.PrintRemovedFeatureError("mkcp header & seed", "finalmask/udp header-* & mkcp-original & mkcp-aes128gcm")
|
||||||
|
}
|
||||||
|
|
||||||
config := common.Must2(internet.CreateTransportConfig(kcp.ProtocolName)).(*kcp.Config)
|
config := common.Must2(internet.CreateTransportConfig(kcp.ProtocolName)).(*kcp.Config)
|
||||||
|
|
||||||
if c.Mtu != nil {
|
if c.Mtu != nil {
|
||||||
@@ -558,16 +560,16 @@ func (c *KCPConfig) Build() (proto.Message, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if config.Mtu < 21 {
|
if config.Mtu < 21 {
|
||||||
return nil, errors.New("MTU must be at least 21")
|
return nil, errors.New("Mtu must be at least 21").AtError()
|
||||||
}
|
}
|
||||||
if config.Tti < 10 || config.Tti > 1000 {
|
if config.Tti < 10 || config.Tti > 1000 {
|
||||||
return nil, errors.New("TTI must be between 10 and 1000")
|
return nil, errors.New("invalid mKCP TTI: ", c.Tti).AtError()
|
||||||
}
|
}
|
||||||
if config.CwndMultiplier < 1 {
|
if config.CwndMultiplier < 1 {
|
||||||
return nil, errors.New("CwndMultiplier must be at least 1")
|
return nil, errors.New("CwndMultiplier must be at least 1").AtError()
|
||||||
}
|
}
|
||||||
if config.GetSendingBufferSize() == 0 {
|
if config.GetSendingBufferSize() == 0 {
|
||||||
return nil, errors.New("MaxSendingWindow must be at least ", config.Mtu)
|
return nil, errors.New("MaxSendingWindow must be >= Mtu").AtError()
|
||||||
}
|
}
|
||||||
|
|
||||||
return config, nil
|
return config, nil
|
||||||
@@ -737,6 +739,11 @@ func (b Bandwidth) Bps() (uint64, error) {
|
|||||||
return uint64(val*float64(mul)) / 8, nil
|
return uint64(val*float64(mul)) / 8, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type UdpHop struct {
|
||||||
|
PortList PortList `json:"ports"`
|
||||||
|
Interval Int32Range `json:"interval"`
|
||||||
|
}
|
||||||
|
|
||||||
type Masquerade struct {
|
type Masquerade struct {
|
||||||
Type string `json:"type"`
|
Type string `json:"type"`
|
||||||
|
|
||||||
@@ -753,8 +760,14 @@ type Masquerade struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type HysteriaConfig struct {
|
type HysteriaConfig struct {
|
||||||
Version int32 `json:"version"`
|
Version int32 `json:"version"`
|
||||||
Auth string `json:"auth"`
|
Auth string `json:"auth"`
|
||||||
|
|
||||||
|
Congestion *string `json:"congestion"`
|
||||||
|
Up *Bandwidth `json:"up"`
|
||||||
|
Down *Bandwidth `json:"down"`
|
||||||
|
UdpHop *UdpHop `json:"udphop"`
|
||||||
|
|
||||||
UdpIdleTimeout int64 `json:"udpIdleTimeout"`
|
UdpIdleTimeout int64 `json:"udpIdleTimeout"`
|
||||||
Masquerade Masquerade `json:"masquerade"`
|
Masquerade Masquerade `json:"masquerade"`
|
||||||
}
|
}
|
||||||
@@ -764,6 +777,10 @@ func (c *HysteriaConfig) Build() (proto.Message, error) {
|
|||||||
return nil, errors.New("version != 2")
|
return nil, errors.New("version != 2")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if c.Congestion != nil || c.Up != nil || c.Down != nil || c.UdpHop != nil {
|
||||||
|
errors.LogWarning(context.Background(), "congestion & up & down & udphop move to finalmask/quicParams")
|
||||||
|
}
|
||||||
|
|
||||||
if c.UdpIdleTimeout != 0 && (c.UdpIdleTimeout < 2 || c.UdpIdleTimeout > 600) {
|
if c.UdpIdleTimeout != 0 && (c.UdpIdleTimeout < 2 || c.UdpIdleTimeout > 600) {
|
||||||
return nil, errors.New("UdpIdleTimeout must be between 2 and 600")
|
return nil, errors.New("UdpIdleTimeout must be between 2 and 600")
|
||||||
}
|
}
|
||||||
@@ -788,46 +805,6 @@ func (c *HysteriaConfig) Build() (proto.Message, error) {
|
|||||||
return config, nil
|
return config, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type MasqueConfig struct {
|
|
||||||
Host string `json:"host"`
|
|
||||||
Path string `json:"path"`
|
|
||||||
Headers map[string]string `json:"headers"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *MasqueConfig) Build() (proto.Message, error) {
|
|
||||||
path := c.Path
|
|
||||||
if path == "" {
|
|
||||||
path = masque.DefaultPath
|
|
||||||
}
|
|
||||||
path = strings.NewReplacer(
|
|
||||||
"{target}", "*", "{ipproto}", "*",
|
|
||||||
"{?target,ipproto}", "?target=*&ipproto=*", "{?ipproto,target}", "?ipproto=*&target=*",
|
|
||||||
"{&target,ipproto}", "&target=*&ipproto=*", "{&ipproto,target}", "&ipproto=*&target=*",
|
|
||||||
).Replace(path)
|
|
||||||
if !strings.HasPrefix(path, "/") || strings.ContainsAny(path, "{}") {
|
|
||||||
return nil, errors.New(`invalid "path": `, path, `, only the variables {target} and {ipproto} are supported`)
|
|
||||||
}
|
|
||||||
if c.Host != "" {
|
|
||||||
if u, err := url.Parse("https://" + c.Host); err != nil || u.Host != c.Host {
|
|
||||||
return nil, errors.New(`invalid "host": `, c.Host)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for k, v := range c.Headers {
|
|
||||||
if !httpguts.ValidHeaderFieldName(k) || !httpguts.ValidHeaderFieldValue(v) {
|
|
||||||
return nil, errors.New(`invalid header in "headers": `, strconv.Quote(k))
|
|
||||||
}
|
|
||||||
switch strings.ToLower(k) {
|
|
||||||
case "host", "capsule-protocol":
|
|
||||||
return nil, errors.New(`"headers" can't contain "`, k, `"`)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return &masque.Config{
|
|
||||||
Host: c.Host,
|
|
||||||
Path: path,
|
|
||||||
Headers: c.Headers,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func readFileOrString(f string, s []string) ([]byte, error) {
|
func readFileOrString(f string, s []string) ([]byte, error) {
|
||||||
if len(f) > 0 {
|
if len(f) > 0 {
|
||||||
return filesystem.ReadCert(f)
|
return filesystem.ReadCert(f)
|
||||||
@@ -837,50 +814,3 @@ func readFileOrString(f string, s []string) ([]byte, error) {
|
|||||||
}
|
}
|
||||||
return nil, errors.New("both file and bytes are empty.")
|
return nil, errors.New("both file and bytes are empty.")
|
||||||
}
|
}
|
||||||
|
|
||||||
type XDriveConfig struct {
|
|
||||||
RemoteFolder string `json:"remoteFolder"`
|
|
||||||
Service string `json:"service"`
|
|
||||||
Secrets []string `json:"secrets"`
|
|
||||||
SegmentBytes uint32 `json:"segmentBytes"`
|
|
||||||
FlushIntervalMs uint32 `json:"flushIntervalMs"`
|
|
||||||
PollIntervalMs uint32 `json:"pollIntervalMs"`
|
|
||||||
MaxPollIntervalMs uint32 `json:"maxPollIntervalMs"`
|
|
||||||
SessionTTLSeconds uint32 `json:"sessionTtlSeconds"`
|
|
||||||
Concurrency uint32 `json:"concurrency"`
|
|
||||||
EagerWindowMs uint32 `json:"eagerWindowMs"`
|
|
||||||
HoleTimeoutMs uint32 `json:"holeTimeoutMs"`
|
|
||||||
Template json.RawMessage `json:"template"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Build implements Buildable.
|
|
||||||
func (c *XDriveConfig) Build() (proto.Message, error) {
|
|
||||||
switch c.Service {
|
|
||||||
case "local":
|
|
||||||
case "Google Drive":
|
|
||||||
if len(c.Secrets) != 3 {
|
|
||||||
return nil, errors.New("Google Drive needs 3 secrets in order of ClientID, ClientSecret, RefreshToken")
|
|
||||||
}
|
|
||||||
case "template":
|
|
||||||
if len(c.Template) == 0 {
|
|
||||||
return nil, errors.New(`service "template" needs a "template" object`)
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
return nil, errors.New("unsupported service")
|
|
||||||
}
|
|
||||||
config := &xdrive.Config{
|
|
||||||
RemoteFolder: c.RemoteFolder,
|
|
||||||
Service: c.Service,
|
|
||||||
Secrets: c.Secrets,
|
|
||||||
SegmentBytes: c.SegmentBytes,
|
|
||||||
FlushIntervalMs: c.FlushIntervalMs,
|
|
||||||
PollIntervalMs: c.PollIntervalMs,
|
|
||||||
MaxPollIntervalMs: c.MaxPollIntervalMs,
|
|
||||||
SessionTtlSeconds: c.SessionTTLSeconds,
|
|
||||||
Concurrency: c.Concurrency,
|
|
||||||
EagerWindowMs: c.EagerWindowMs,
|
|
||||||
HoleTimeoutMs: c.HoleTimeoutMs,
|
|
||||||
Template: string(c.Template),
|
|
||||||
}
|
|
||||||
return config, nil
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -113,10 +113,10 @@ func (c *REALITYConfig) Build() (proto.Message, error) {
|
|||||||
config.MinClientVer[i] = byte(u)
|
config.MinClientVer[i] = byte(u)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// errors.LogWarning(context.Background(), `REALITY: Changing "minClientVer" will increase the likelihood of your server's IP being blocked by the GFW`)
|
errors.LogWarning(context.Background(), `REALITY: Changing "minClientVer" will increase the likelihood of your server's IP being blocked by the GFW`)
|
||||||
} else {
|
} else {
|
||||||
// config.MinClientVer = []byte{26, 3, 27} // change it at your own risk: https://github.com/XTLS/Xray-core/commit/af7eb68028732a8ee3c0e5d6ab2b8a657bb2e770
|
config.MinClientVer = []byte{26, 3, 27} // change it at your own risk: https://github.com/XTLS/Xray-core/commit/af7eb68028732a8ee3c0e5d6ab2b8a657bb2e770
|
||||||
// errors.LogWarning(context.Background(), `REALITY: The default minimal client version is Xray-core v26.3.27, other clients may be refused to connect`)
|
errors.LogWarning(context.Background(), `REALITY: The default minimal client version is Xray-core v26.3.27, other clients may be refused to connect`)
|
||||||
}
|
}
|
||||||
if c.MaxClientVer != "" {
|
if c.MaxClientVer != "" {
|
||||||
config.MaxClientVer = make([]byte, 3)
|
config.MaxClientVer = make([]byte, 3)
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type CustomSockoptConfig struct {
|
type CustomSockoptConfig struct {
|
||||||
System string `json:"system"`
|
Syetem string `json:"system"`
|
||||||
Network string `json:"network"`
|
Network string `json:"network"`
|
||||||
Level string `json:"level"`
|
Level string `json:"level"`
|
||||||
Opt string `json:"opt"`
|
Opt string `json:"opt"`
|
||||||
@@ -124,7 +124,7 @@ func (c *SocketConfig) Build() (*internet.SocketConfig, error) {
|
|||||||
|
|
||||||
for _, copt := range c.CustomSockopt {
|
for _, copt := range c.CustomSockopt {
|
||||||
customSockopt := &internet.CustomSockopt{
|
customSockopt := &internet.CustomSockopt{
|
||||||
System: copt.System,
|
System: copt.Syetem,
|
||||||
Network: copt.Network,
|
Network: copt.Network,
|
||||||
Level: copt.Level,
|
Level: copt.Level,
|
||||||
Opt: copt.Opt,
|
Opt: copt.Opt,
|
||||||
|
|||||||
@@ -291,76 +291,3 @@ func TestHeaderCustomUDPBuildRejectsExprWithoutArgs(t *testing.T) {
|
|||||||
t.Fatalf("expected transform arg rejection, got %v", err)
|
t.Fatalf("expected transform arg rejection, got %v", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestXDriveStreamConfig(t *testing.T) {
|
|
||||||
config := new(StreamConfig)
|
|
||||||
if err := json.Unmarshal([]byte(`{
|
|
||||||
"method": "xdrive",
|
|
||||||
"xdriveSettings": {
|
|
||||||
"remoteFolder": "/tmp/xdrive",
|
|
||||||
"service": "local"
|
|
||||||
}
|
|
||||||
}`), config); err != nil {
|
|
||||||
t.Fatalf("Unmarshal: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
built, err := config.Build()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Build: %v", err)
|
|
||||||
}
|
|
||||||
if built.ProtocolName != "xdrive" {
|
|
||||||
t.Fatalf("ProtocolName is %q, want %q", built.ProtocolName, "xdrive")
|
|
||||||
}
|
|
||||||
if len(built.TransportSettings) != 1 || built.TransportSettings[0].ProtocolName != "xdrive" {
|
|
||||||
t.Fatalf("TransportSettings is %v, want a single xdrive entry", built.TransportSettings)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestXDriveRejectsUnknownService(t *testing.T) {
|
|
||||||
config := new(XDriveConfig)
|
|
||||||
if err := json.Unmarshal([]byte(`{"remoteFolder": "/tmp/xdrive", "service": "Dropbox"}`), config); err != nil {
|
|
||||||
t.Fatalf("Unmarshal: %v", err)
|
|
||||||
}
|
|
||||||
if _, err := config.Build(); err == nil {
|
|
||||||
t.Fatal("Build accepted an unsupported service")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestXDriveTemplateStreamConfig(t *testing.T) {
|
|
||||||
config := new(StreamConfig)
|
|
||||||
if err := json.Unmarshal([]byte(`{
|
|
||||||
"method": "xdrive",
|
|
||||||
"xdriveSettings": {
|
|
||||||
"remoteFolder": "folder",
|
|
||||||
"service": "template",
|
|
||||||
"secrets": ["user", "pass"],
|
|
||||||
"template": {
|
|
||||||
"flatten": true,
|
|
||||||
"auth": {"type": "basic", "username": "{secret0}", "password": "{secret1}"},
|
|
||||||
"put": {"method": "PUT", "url": "https://dav.example/{folder}/{name}"},
|
|
||||||
"get": {"method": "GET", "url": "https://dav.example/{folder}/{name}"},
|
|
||||||
"delete": {"method": "DELETE", "url": "https://dav.example/{folder}/{name}"},
|
|
||||||
"list": {"method": "PROPFIND", "url": "https://dav.example/{folder}/", "namesRegex": "<d:href>/folder/([^<]+)</d:href>"}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}`), config); err != nil {
|
|
||||||
t.Fatalf("Unmarshal: %v", err)
|
|
||||||
}
|
|
||||||
built, err := config.Build()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Build: %v", err)
|
|
||||||
}
|
|
||||||
if built.ProtocolName != "xdrive" {
|
|
||||||
t.Fatalf("ProtocolName is %q, want xdrive", built.ProtocolName)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestXDriveTemplateNeedsTemplate(t *testing.T) {
|
|
||||||
config := new(XDriveConfig)
|
|
||||||
if err := json.Unmarshal([]byte(`{"remoteFolder": "f", "service": "template"}`), config); err != nil {
|
|
||||||
t.Fatalf("Unmarshal: %v", err)
|
|
||||||
}
|
|
||||||
if _, err := config.Build(); err == nil {
|
|
||||||
t.Fatal("Build accepted a template service without a template")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -312,9 +312,6 @@ func (c *VLessOutboundConfig) Build() (proto.Message, error) {
|
|||||||
if err := json.Unmarshal(rawUser, account); err != nil {
|
if err := json.Unmarshal(rawUser, account); err != nil {
|
||||||
return nil, errors.New(`VLESS users: invalid user`).Base(err)
|
return nil, errors.New(`VLESS users: invalid user`).Base(err)
|
||||||
}
|
}
|
||||||
// validateOutboundTransportSecurity needs to see these
|
|
||||||
c.Encryption = account.Encryption
|
|
||||||
c.Address = rec.Address
|
|
||||||
if account.Reverse != nil { // may not be reached: error json unmarshal
|
if account.Reverse != nil { // may not be reached: error json unmarshal
|
||||||
return nil, errors.New(`VLESS users: please use simplified outbound's config style to use "reverse"`)
|
return nil, errors.New(`VLESS users: please use simplified outbound's config style to use "reverse"`)
|
||||||
}
|
}
|
||||||
|
|||||||
+23
-7
@@ -59,13 +59,14 @@ func (c *WireGuardPeerConfig) Build() (*wireguard.PeerConfig, error) {
|
|||||||
type WireGuardConfig struct {
|
type WireGuardConfig struct {
|
||||||
IsClient bool `json:""`
|
IsClient bool `json:""`
|
||||||
|
|
||||||
NoKernelTun bool `json:"noKernelTun"`
|
NoKernelTun bool `json:"noKernelTun"`
|
||||||
SecretKey string `json:"secretKey"`
|
SecretKey string `json:"secretKey"`
|
||||||
Address []string `json:"address"`
|
Address []string `json:"address"`
|
||||||
Peers []*WireGuardPeerConfig `json:"peers"`
|
Peers []*WireGuardPeerConfig `json:"peers"`
|
||||||
MTU int32 `json:"mtu"`
|
MTU int32 `json:"mtu"`
|
||||||
Reserved []byte `json:"reserved"`
|
Reserved []byte `json:"reserved"`
|
||||||
DNS []string `json:"remoteDNS"`
|
DomainStrategy string `json:"domainStrategy"`
|
||||||
|
DNS []string `json:"remoteDNS"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *WireGuardConfig) Build() (proto.Message, error) {
|
func (c *WireGuardConfig) Build() (proto.Message, error) {
|
||||||
@@ -124,6 +125,21 @@ func (c *WireGuardConfig) Build() (proto.Message, error) {
|
|||||||
}
|
}
|
||||||
config.Reserved = c.Reserved
|
config.Reserved = c.Reserved
|
||||||
|
|
||||||
|
switch strings.ToLower(c.DomainStrategy) {
|
||||||
|
case "forceip", "":
|
||||||
|
config.DomainStrategy = wireguard.DeviceConfig_FORCE_IP
|
||||||
|
case "forceipv4":
|
||||||
|
config.DomainStrategy = wireguard.DeviceConfig_FORCE_IP4
|
||||||
|
case "forceipv6":
|
||||||
|
config.DomainStrategy = wireguard.DeviceConfig_FORCE_IP6
|
||||||
|
case "forceipv4v6":
|
||||||
|
config.DomainStrategy = wireguard.DeviceConfig_FORCE_IP46
|
||||||
|
case "forceipv6v4":
|
||||||
|
config.DomainStrategy = wireguard.DeviceConfig_FORCE_IP64
|
||||||
|
default:
|
||||||
|
return nil, errors.New("unsupported domain strategy: ", c.DomainStrategy)
|
||||||
|
}
|
||||||
|
|
||||||
config.IsClient = c.IsClient
|
config.IsClient = c.IsClient
|
||||||
config.NoKernelTun = c.NoKernelTun
|
config.NoKernelTun = c.NoKernelTun
|
||||||
config.DNS = c.DNS
|
config.DNS = c.DNS
|
||||||
|
|||||||
+35
-42
@@ -15,8 +15,6 @@ import (
|
|||||||
"github.com/xtls/xray-core/common/net"
|
"github.com/xtls/xray-core/common/net"
|
||||||
"github.com/xtls/xray-core/common/serial"
|
"github.com/xtls/xray-core/common/serial"
|
||||||
core "github.com/xtls/xray-core/core"
|
core "github.com/xtls/xray-core/core"
|
||||||
"github.com/xtls/xray-core/proxy/freedom"
|
|
||||||
"github.com/xtls/xray-core/proxy/masque"
|
|
||||||
"github.com/xtls/xray-core/transport/internet"
|
"github.com/xtls/xray-core/transport/internet"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -49,7 +47,6 @@ var (
|
|||||||
"vmess": func() interface{} { return new(VMessOutboundConfig) },
|
"vmess": func() interface{} { return new(VMessOutboundConfig) },
|
||||||
"trojan": func() interface{} { return new(TrojanClientConfig) },
|
"trojan": func() interface{} { return new(TrojanClientConfig) },
|
||||||
"hysteria": func() interface{} { return new(HysteriaClientConfig) },
|
"hysteria": func() interface{} { return new(HysteriaClientConfig) },
|
||||||
"masque": func() interface{} { return new(MasqueClientConfig) },
|
|
||||||
"dns": func() interface{} { return new(DNSOutboundConfig) },
|
"dns": func() interface{} { return new(DNSOutboundConfig) },
|
||||||
"wireguard": func() interface{} { return &WireGuardConfig{IsClient: true} },
|
"wireguard": func() interface{} { return &WireGuardConfig{IsClient: true} },
|
||||||
}, "protocol", "settings")
|
}, "protocol", "settings")
|
||||||
@@ -219,11 +216,21 @@ type OutboundDetourConfig struct {
|
|||||||
Tag string `json:"tag"`
|
Tag string `json:"tag"`
|
||||||
Settings *json.RawMessage `json:"settings"`
|
Settings *json.RawMessage `json:"settings"`
|
||||||
StreamSetting *StreamConfig `json:"streamSettings"`
|
StreamSetting *StreamConfig `json:"streamSettings"`
|
||||||
ProxySettings *json.RawMessage `json:"proxySettings"`
|
ProxySettings *ProxyConfig `json:"proxySettings"`
|
||||||
MuxSettings *MuxConfig `json:"mux"`
|
MuxSettings *MuxConfig `json:"mux"`
|
||||||
TargetStrategy string `json:"targetStrategy"`
|
TargetStrategy string `json:"targetStrategy"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *OutboundDetourConfig) checkChainProxyConfig() error {
|
||||||
|
if c.StreamSetting == nil || c.ProxySettings == nil || c.StreamSetting.SocketSettings == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if len(c.ProxySettings.Tag) > 0 && len(c.StreamSetting.SocketSettings.DialerProxy) > 0 {
|
||||||
|
return errors.New("proxySettings.tag is conflicted with sockopt.dialerProxy").AtWarning()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func requiresTransportSecurity(address *Address) bool {
|
func requiresTransportSecurity(address *Address) bool {
|
||||||
if address == nil || address.Address == nil {
|
if address == nil || address.Address == nil {
|
||||||
return false
|
return false
|
||||||
@@ -244,7 +251,7 @@ func validateOutboundTransportSecurity(rawConfig interface{}, senderSettings *pr
|
|||||||
if vlessCfg.Encryption != "" && vlessCfg.Encryption != "none" {
|
if vlessCfg.Encryption != "" && vlessCfg.Encryption != "none" {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
if requiresTransportSecurity(vlessCfg.Address) {
|
if requiresTransportSecurity(vlessCfg.Vnext[0].Address) {
|
||||||
return errors.New("vless without TLS or other encryption is prohibited unless the server address is a private IP or domain")
|
return errors.New("vless without TLS or other encryption is prohibited unless the server address is a private IP or domain")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -260,10 +267,6 @@ func validateOutboundTransportSecurity(rawConfig interface{}, senderSettings *pr
|
|||||||
|
|
||||||
// Build implements Buildable.
|
// Build implements Buildable.
|
||||||
func (c *OutboundDetourConfig) Build() (*core.OutboundHandlerConfig, error) {
|
func (c *OutboundDetourConfig) Build() (*core.OutboundHandlerConfig, error) {
|
||||||
if c.ProxySettings != nil {
|
|
||||||
return nil, errors.PrintRemovedFeatureError(`outbound "proxySettings"`, `"streamSettings.sockopt.dialerProxy"`)
|
|
||||||
}
|
|
||||||
|
|
||||||
senderSettings := &proxyman.SenderConfig{}
|
senderSettings := &proxyman.SenderConfig{}
|
||||||
switch strings.ToLower(c.TargetStrategy) {
|
switch strings.ToLower(c.TargetStrategy) {
|
||||||
case "asis", "":
|
case "asis", "":
|
||||||
@@ -291,6 +294,9 @@ func (c *OutboundDetourConfig) Build() (*core.OutboundHandlerConfig, error) {
|
|||||||
default:
|
default:
|
||||||
return nil, errors.New("unsupported target domain strategy: ", c.TargetStrategy)
|
return nil, errors.New("unsupported target domain strategy: ", c.TargetStrategy)
|
||||||
}
|
}
|
||||||
|
if err := c.checkChainProxyConfig(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
if c.SendThrough != nil {
|
if c.SendThrough != nil {
|
||||||
address := ParseSendThough(c.SendThrough)
|
address := ParseSendThough(c.SendThrough)
|
||||||
@@ -316,6 +322,26 @@ func (c *OutboundDetourConfig) Build() (*core.OutboundHandlerConfig, error) {
|
|||||||
senderSettings.StreamSettings = ss
|
senderSettings.StreamSettings = ss
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if c.ProxySettings != nil {
|
||||||
|
ps, err := c.ProxySettings.Build()
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.New("invalid outbound detour proxy settings").Base(err)
|
||||||
|
}
|
||||||
|
if ps.TransportLayerProxy {
|
||||||
|
if senderSettings.StreamSettings != nil {
|
||||||
|
if senderSettings.StreamSettings.SocketSettings != nil {
|
||||||
|
senderSettings.StreamSettings.SocketSettings.DialerProxy = ps.Tag
|
||||||
|
} else {
|
||||||
|
senderSettings.StreamSettings.SocketSettings = &internet.SocketConfig{DialerProxy: ps.Tag}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
senderSettings.StreamSettings = &internet.StreamConfig{SocketSettings: &internet.SocketConfig{DialerProxy: ps.Tag}}
|
||||||
|
}
|
||||||
|
ps = nil
|
||||||
|
}
|
||||||
|
senderSettings.ProxySettings = ps
|
||||||
|
}
|
||||||
|
|
||||||
if c.MuxSettings != nil {
|
if c.MuxSettings != nil {
|
||||||
ms, err := c.MuxSettings.Build()
|
ms, err := c.MuxSettings.Build()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -340,39 +366,6 @@ func (c *OutboundDetourConfig) Build() (*core.OutboundHandlerConfig, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, ok := ts.(*masque.ClientConfig); ok {
|
|
||||||
if ms := senderSettings.MultiplexSettings; ms != nil && ms.Enabled {
|
|
||||||
return nil, errors.New(`masque outbound does not support "mux"`)
|
|
||||||
}
|
|
||||||
} else if senderSettings.StreamSettings != nil && senderSettings.StreamSettings.ProtocolName == "masque" {
|
|
||||||
return nil, errors.New("the masque transport can only be used by the masque outbound")
|
|
||||||
}
|
|
||||||
|
|
||||||
if fc, ok := ts.(*freedom.Config); ok {
|
|
||||||
if senderSettings.StreamSettings != nil &&
|
|
||||||
senderSettings.StreamSettings.SocketSettings != nil &&
|
|
||||||
senderSettings.StreamSettings.SocketSettings.AddressPortStrategy != internet.AddressPortStrategy_None {
|
|
||||||
return nil, errors.New(`freedom outbound does not support "sockopt.addressPortStrategy"`)
|
|
||||||
}
|
|
||||||
|
|
||||||
var strategy internet.DomainStrategy
|
|
||||||
if strategy = senderSettings.TargetStrategy; strategy != internet.DomainStrategy_AS_IS {
|
|
||||||
errors.LogWarning(context.Background(), `The "outbound.targetStrategy" setting is not supported directly by freedom and has been automatically migrated to "sockopt.domainStrategy" with no behavior change.`)
|
|
||||||
senderSettings.TargetStrategy = internet.DomainStrategy_AS_IS
|
|
||||||
} else if strategy = fc.DomainStrategy; strategy != internet.DomainStrategy_AS_IS {
|
|
||||||
errors.LogWarning(context.Background(), `The "freedom.domainStrategy" setting is deprecated and will be removed. For compatibility, its value has been automatically migrated to "sockopt.domainStrategy". Please update your config before removal.`)
|
|
||||||
}
|
|
||||||
if strategy != internet.DomainStrategy_AS_IS {
|
|
||||||
if senderSettings.StreamSettings == nil {
|
|
||||||
senderSettings.StreamSettings = &internet.StreamConfig{}
|
|
||||||
}
|
|
||||||
if senderSettings.StreamSettings.SocketSettings == nil {
|
|
||||||
senderSettings.StreamSettings.SocketSettings = &internet.SocketConfig{}
|
|
||||||
}
|
|
||||||
senderSettings.StreamSettings.SocketSettings.DomainStrategy = strategy
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return &core.OutboundHandlerConfig{
|
return &core.OutboundHandlerConfig{
|
||||||
SenderSettings: serial.ToTypedMessage(senderSettings),
|
SenderSettings: serial.ToTypedMessage(senderSettings),
|
||||||
Tag: c.Tag,
|
Tag: c.Tag,
|
||||||
|
|||||||
+138
-272
@@ -1,18 +1,15 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"errors"
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"go/build"
|
||||||
"os"
|
"os"
|
||||||
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
"sort"
|
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
|
||||||
"sync/atomic"
|
|
||||||
|
|
||||||
"mvdan.cc/gofumpt/format"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -26,27 +23,101 @@ var (
|
|||||||
isFormat bool
|
isFormat bool
|
||||||
)
|
)
|
||||||
|
|
||||||
func getModuleInfo(pwd string) (modPath, langVersion string, err error) {
|
// envFile returns the name of the Go environment configuration file.
|
||||||
data, err := os.ReadFile(filepath.Join(pwd, "go.mod"))
|
// Copy from https://github.com/golang/go/blob/c4f2a9788a7be04daf931ac54382fbe2cb754938/src/cmd/go/internal/cfg/cfg.go#L150-L166
|
||||||
if err != nil {
|
func envFile() (string, error) {
|
||||||
return "", "", err
|
if file := os.Getenv("GOENV"); file != "" {
|
||||||
}
|
if file == "off" {
|
||||||
for _, line := range strings.Split(string(data), "\n") {
|
return "", errors.New("GOENV=off")
|
||||||
fields := strings.Fields(line)
|
|
||||||
if len(fields) >= 2 {
|
|
||||||
switch fields[0] {
|
|
||||||
case "module":
|
|
||||||
modPath = fields[1]
|
|
||||||
case "go":
|
|
||||||
langVersion = "go" + strings.TrimPrefix(fields[1], "go")
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
return file, nil
|
||||||
}
|
}
|
||||||
return modPath, langVersion, nil
|
dir, err := os.UserConfigDir()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if dir == "" {
|
||||||
|
return "", errors.New("missing user-config dir")
|
||||||
|
}
|
||||||
|
return filepath.Join(dir, "go", "env"), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func formatGoSource(src []byte, opts format.Options) ([]byte, error) {
|
// GetRuntimeEnv returns the value of runtime environment variable,
|
||||||
return format.Source(src, opts)
|
// that is set by running following command: `go env -w key=value`.
|
||||||
|
func GetRuntimeEnv(key string) (string, error) {
|
||||||
|
file, err := envFile()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if file == "" {
|
||||||
|
return "", errors.New("missing runtime env file")
|
||||||
|
}
|
||||||
|
var data []byte
|
||||||
|
var runtimeEnv string
|
||||||
|
data, readErr := os.ReadFile(file)
|
||||||
|
if readErr != nil {
|
||||||
|
return "", readErr
|
||||||
|
}
|
||||||
|
envStrings := strings.Split(string(data), "\n")
|
||||||
|
for _, envItem := range envStrings {
|
||||||
|
envItem = strings.TrimSuffix(envItem, "\r")
|
||||||
|
envKeyValue := strings.Split(envItem, "=")
|
||||||
|
if len(envKeyValue) == 2 && strings.TrimSpace(envKeyValue[0]) == key {
|
||||||
|
runtimeEnv = strings.TrimSpace(envKeyValue[1])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return runtimeEnv, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetGOBIN returns GOBIN environment variable as a string. It will NOT be empty.
|
||||||
|
func GetGOBIN() string {
|
||||||
|
// The one set by user explicitly by `export GOBIN=/path` or `env GOBIN=/path command`
|
||||||
|
GOBIN := os.Getenv("GOBIN")
|
||||||
|
if GOBIN == "" {
|
||||||
|
var err error
|
||||||
|
// The one set by user by running `go env -w GOBIN=/path`
|
||||||
|
GOBIN, err = GetRuntimeEnv("GOBIN")
|
||||||
|
if err != nil {
|
||||||
|
// The default one that Golang uses
|
||||||
|
return filepath.Join(build.Default.GOPATH, "bin")
|
||||||
|
}
|
||||||
|
if GOBIN == "" {
|
||||||
|
return filepath.Join(build.Default.GOPATH, "bin")
|
||||||
|
}
|
||||||
|
return GOBIN
|
||||||
|
}
|
||||||
|
return GOBIN
|
||||||
|
}
|
||||||
|
|
||||||
|
func Run(binary string, args []string) ([]byte, error) {
|
||||||
|
cmd := exec.Command(binary, args...)
|
||||||
|
cmd.Env = append(cmd.Env, os.Environ()...)
|
||||||
|
output, cmdErr := cmd.CombinedOutput()
|
||||||
|
if cmdErr != nil {
|
||||||
|
return nil, cmdErr
|
||||||
|
}
|
||||||
|
return output, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func RunMany(binary string, args, files []string) bool {
|
||||||
|
fmt.Println("Processing with", binary, args, "...")
|
||||||
|
|
||||||
|
formatRequired := false
|
||||||
|
maxTasks := make(chan struct{}, runtime.NumCPU())
|
||||||
|
for _, file := range files {
|
||||||
|
maxTasks <- struct{}{}
|
||||||
|
go func(file string) {
|
||||||
|
output, err := Run(binary, append(args, file))
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
} else if len(output) > 0 {
|
||||||
|
fmt.Println(string(output))
|
||||||
|
formatRequired = true
|
||||||
|
}
|
||||||
|
<-maxTasks
|
||||||
|
}(file)
|
||||||
|
}
|
||||||
|
return formatRequired
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
@@ -79,76 +150,26 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pwd := *directory
|
pwd := *directory
|
||||||
modPath, langVersion, modErr := getModuleInfo(pwd)
|
GOBIN := GetGOBIN()
|
||||||
if modErr != nil {
|
binPath := os.Getenv("PATH")
|
||||||
fmt.Println("Error reading go.mod:", modErr)
|
pathSlice := []string{pwd, GOBIN, binPath}
|
||||||
|
binPath = strings.Join(pathSlice, string(os.PathListSeparator))
|
||||||
|
os.Setenv("PATH", binPath)
|
||||||
|
|
||||||
|
suffix := ""
|
||||||
|
if runtime.GOOS == "windows" {
|
||||||
|
suffix = ".exe"
|
||||||
|
}
|
||||||
|
gofmt := "gofumpt" + suffix
|
||||||
|
|
||||||
|
if gofmtPath, err := exec.LookPath(gofmt); err != nil {
|
||||||
|
fmt.Println("Can not find", gofmt, "in system path or current working directory.")
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
} else {
|
||||||
opts := format.Options{
|
gofmt = gofmtPath
|
||||||
LangVersion: langVersion,
|
|
||||||
ModulePath: modPath,
|
|
||||||
}
|
|
||||||
|
|
||||||
if isFormat {
|
|
||||||
fmt.Println("Formatting Go source files...")
|
|
||||||
} else if isCheck {
|
|
||||||
fmt.Println("Checking files thar are not properly formatted...")
|
|
||||||
}
|
|
||||||
|
|
||||||
jobs := make(chan string, runtime.NumCPU())
|
|
||||||
var wg sync.WaitGroup
|
|
||||||
var formatRequired atomic.Bool
|
|
||||||
var hasErrors atomic.Bool
|
|
||||||
|
|
||||||
for i := 0; i < runtime.NumCPU(); i++ {
|
|
||||||
wg.Go(func() {
|
|
||||||
for path := range jobs {
|
|
||||||
src, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "Error reading %s: %v\n", path, err)
|
|
||||||
hasErrors.Store(true)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
formatted, err := formatGoSource(src, opts)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "Error formatting %s: %v\n", path, err)
|
|
||||||
hasErrors.Store(true)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if !bytes.Equal(src, formatted) {
|
|
||||||
var diffText []byte
|
|
||||||
if isDryrun {
|
|
||||||
newName := filepath.ToSlash(path)
|
|
||||||
oldName := newName + ".orig"
|
|
||||||
diffText = diff(oldName, src, newName, formatted)
|
|
||||||
}
|
|
||||||
if isFormat {
|
|
||||||
info, statErr := os.Stat(path)
|
|
||||||
if statErr != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "Error stating %s: %v\n", path, statErr)
|
|
||||||
hasErrors.Store(true)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if writeErr := os.WriteFile(path, formatted, info.Mode().Perm()); writeErr != nil {
|
|
||||||
fmt.Fprintf(os.Stderr, "Error writing %s: %v\n", path, writeErr)
|
|
||||||
hasErrors.Store(true)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
formatRequired.Store(true)
|
|
||||||
if isDryrun && len(diffText) > 0 {
|
|
||||||
fmt.Printf("%s\n%s", path, diffText)
|
|
||||||
} else {
|
|
||||||
fmt.Println(path)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
rawFilesSlice := make([]string, 0, 1000)
|
||||||
walkErr := filepath.Walk(pwd, func(path string, info os.FileInfo, err error) error {
|
walkErr := filepath.Walk(pwd, func(path string, info os.FileInfo, err error) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
@@ -165,206 +186,51 @@ func main() {
|
|||||||
!strings.HasSuffix(filename, ".pb.go") &&
|
!strings.HasSuffix(filename, ".pb.go") &&
|
||||||
!strings.Contains(dir, filepath.Join("testing", "mocks")) &&
|
!strings.Contains(dir, filepath.Join("testing", "mocks")) &&
|
||||||
!strings.Contains(path, filepath.Join("main", "distro", "all", "all.go")) {
|
!strings.Contains(path, filepath.Join("main", "distro", "all", "all.go")) {
|
||||||
jobs <- path
|
rawFilesSlice = append(rawFilesSlice, path)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
})
|
})
|
||||||
close(jobs)
|
|
||||||
wg.Wait()
|
|
||||||
|
|
||||||
if walkErr != nil {
|
if walkErr != nil {
|
||||||
fmt.Println(walkErr)
|
fmt.Println(walkErr)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
if hasErrors.Load() {
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
if isFormat {
|
if isFormat {
|
||||||
if formatRequired.Load() {
|
gofmtArgs := []string{
|
||||||
fmt.Println("Do NOT forget to commit file changes.")
|
"-l", "-e", "-w",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fmt.Println("Formatting Go source files...")
|
||||||
|
RunMany(gofmt, gofmtArgs, rawFilesSlice)
|
||||||
|
fmt.Println("Do NOT forget to commit file changes.")
|
||||||
}
|
}
|
||||||
|
|
||||||
if isCheck {
|
if isCheck {
|
||||||
if formatRequired.Load() {
|
gofmtListArgs := []string{
|
||||||
|
"-l", "-e",
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println("Checking files thar are not properly formatted...")
|
||||||
|
formatRequired := RunMany(gofmt, gofmtListArgs, rawFilesSlice)
|
||||||
|
if formatRequired {
|
||||||
fmt.Println("Format problem(s) found.")
|
fmt.Println("Format problem(s) found.")
|
||||||
fmt.Println("Please run 'go run ./infra/vformat/main.go' to format the Go source files.")
|
}
|
||||||
|
|
||||||
|
if isDryrun {
|
||||||
|
if formatRequired {
|
||||||
|
gofmtShowArgs := []string{
|
||||||
|
"-d", "-e",
|
||||||
|
}
|
||||||
|
RunMany(gofmt, gofmtShowArgs, rawFilesSlice)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if formatRequired {
|
||||||
|
fmt.Println("Please run 'go install -v mvdan.cc/gofumpt@latest', then run 'go run ./infra/vformat/main.go' to format the Go source files.")
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
} else {
|
} else {
|
||||||
fmt.Println("All Go source file format check has been passed.")
|
fmt.Println("All Go source file format check has been passed.")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// diff algorithm copied from mvdan.cc/gofumpt/internal/govendor/diff
|
|
||||||
type pair struct{ x, y int }
|
|
||||||
|
|
||||||
func diff(oldName string, old []byte, newName string, new []byte) []byte {
|
|
||||||
if bytes.Equal(old, new) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
x := diffLines(old)
|
|
||||||
y := diffLines(new)
|
|
||||||
|
|
||||||
var out bytes.Buffer
|
|
||||||
fmt.Fprintf(&out, "diff %s %s\n", oldName, newName)
|
|
||||||
fmt.Fprintf(&out, "--- %s\n", oldName)
|
|
||||||
fmt.Fprintf(&out, "+++ %s\n", newName)
|
|
||||||
|
|
||||||
var (
|
|
||||||
done pair
|
|
||||||
chunk pair
|
|
||||||
count pair
|
|
||||||
ctext []string
|
|
||||||
)
|
|
||||||
for _, m := range diffTgs(x, y) {
|
|
||||||
if m.x < done.x {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
start := m
|
|
||||||
for start.x > done.x && start.y > done.y && x[start.x-1] == y[start.y-1] {
|
|
||||||
start.x--
|
|
||||||
start.y--
|
|
||||||
}
|
|
||||||
end := m
|
|
||||||
for end.x < len(x) && end.y < len(y) && x[end.x] == y[end.y] {
|
|
||||||
end.x++
|
|
||||||
end.y++
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, s := range x[done.x:start.x] {
|
|
||||||
ctext = append(ctext, "-"+s)
|
|
||||||
count.x++
|
|
||||||
}
|
|
||||||
for _, s := range y[done.y:start.y] {
|
|
||||||
ctext = append(ctext, "+"+s)
|
|
||||||
count.y++
|
|
||||||
}
|
|
||||||
|
|
||||||
const C = 3
|
|
||||||
if (end.x < len(x) || end.y < len(y)) &&
|
|
||||||
(end.x-start.x < C || (len(ctext) > 0 && end.x-start.x < 2*C)) {
|
|
||||||
for _, s := range x[start.x:end.x] {
|
|
||||||
ctext = append(ctext, " "+s)
|
|
||||||
count.x++
|
|
||||||
count.y++
|
|
||||||
}
|
|
||||||
done = end
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(ctext) > 0 {
|
|
||||||
n := end.x - start.x
|
|
||||||
if n > C {
|
|
||||||
n = C
|
|
||||||
}
|
|
||||||
for _, s := range x[start.x : start.x+n] {
|
|
||||||
ctext = append(ctext, " "+s)
|
|
||||||
count.x++
|
|
||||||
count.y++
|
|
||||||
}
|
|
||||||
done = pair{start.x + n, start.y + n}
|
|
||||||
|
|
||||||
if count.x > 0 {
|
|
||||||
chunk.x++
|
|
||||||
}
|
|
||||||
if count.y > 0 {
|
|
||||||
chunk.y++
|
|
||||||
}
|
|
||||||
fmt.Fprintf(&out, "@@ -%d,%d +%d,%d @@\n", chunk.x, count.x, chunk.y, count.y)
|
|
||||||
for _, s := range ctext {
|
|
||||||
out.WriteString(s)
|
|
||||||
}
|
|
||||||
count.x = 0
|
|
||||||
count.y = 0
|
|
||||||
ctext = ctext[:0]
|
|
||||||
}
|
|
||||||
|
|
||||||
if end.x >= len(x) && end.y >= len(y) {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
chunk = pair{end.x - C, end.y - C}
|
|
||||||
for _, s := range x[chunk.x:end.x] {
|
|
||||||
ctext = append(ctext, " "+s)
|
|
||||||
count.x++
|
|
||||||
count.y++
|
|
||||||
}
|
|
||||||
done = end
|
|
||||||
}
|
|
||||||
|
|
||||||
return out.Bytes()
|
|
||||||
}
|
|
||||||
|
|
||||||
func diffLines(x []byte) []string {
|
|
||||||
l := strings.SplitAfter(string(x), "\n")
|
|
||||||
if l[len(l)-1] == "" {
|
|
||||||
l = l[:len(l)-1]
|
|
||||||
} else {
|
|
||||||
l[len(l)-1] += "\n\\ No newline at end of file\n"
|
|
||||||
}
|
|
||||||
return l
|
|
||||||
}
|
|
||||||
|
|
||||||
func diffTgs(x, y []string) []pair {
|
|
||||||
m := make(map[string]int)
|
|
||||||
for _, s := range x {
|
|
||||||
if c := m[s]; c > -2 {
|
|
||||||
m[s] = c - 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for _, s := range y {
|
|
||||||
if c := m[s]; c > -8 {
|
|
||||||
m[s] = c - 4
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var xi, yi, inv []int
|
|
||||||
for i, s := range y {
|
|
||||||
if m[s] == -5 {
|
|
||||||
m[s] = len(yi)
|
|
||||||
yi = append(yi, i)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for i, s := range x {
|
|
||||||
if j, ok := m[s]; ok && j >= 0 {
|
|
||||||
xi = append(xi, i)
|
|
||||||
inv = append(inv, j)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
J := inv
|
|
||||||
n := len(xi)
|
|
||||||
T := make([]int, n)
|
|
||||||
L := make([]int, n)
|
|
||||||
for i := range T {
|
|
||||||
T[i] = n + 1
|
|
||||||
}
|
|
||||||
for i := 0; i < n; i++ {
|
|
||||||
k := sort.Search(n, func(k int) bool {
|
|
||||||
return T[k] >= J[i]
|
|
||||||
})
|
|
||||||
T[k] = J[i]
|
|
||||||
L[i] = k + 1
|
|
||||||
}
|
|
||||||
k := 0
|
|
||||||
for _, v := range L {
|
|
||||||
if k < v {
|
|
||||||
k = v
|
|
||||||
}
|
|
||||||
}
|
|
||||||
seq := make([]pair, 2+k)
|
|
||||||
seq[1+k] = pair{len(x), len(y)}
|
|
||||||
lastj := n
|
|
||||||
for i := n - 1; i >= 0; i-- {
|
|
||||||
if L[i] == k && J[i] < lastj {
|
|
||||||
seq[k] = pair{xi[i], yi[J[i]]}
|
|
||||||
k--
|
|
||||||
}
|
|
||||||
}
|
|
||||||
seq[0] = pair{0, 0}
|
|
||||||
return seq
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -41,7 +41,6 @@ import (
|
|||||||
_ "github.com/xtls/xray-core/proxy/freedom"
|
_ "github.com/xtls/xray-core/proxy/freedom"
|
||||||
_ "github.com/xtls/xray-core/proxy/http"
|
_ "github.com/xtls/xray-core/proxy/http"
|
||||||
_ "github.com/xtls/xray-core/proxy/loopback"
|
_ "github.com/xtls/xray-core/proxy/loopback"
|
||||||
_ "github.com/xtls/xray-core/proxy/masque"
|
|
||||||
_ "github.com/xtls/xray-core/proxy/shadowsocks"
|
_ "github.com/xtls/xray-core/proxy/shadowsocks"
|
||||||
_ "github.com/xtls/xray-core/proxy/socks"
|
_ "github.com/xtls/xray-core/proxy/socks"
|
||||||
_ "github.com/xtls/xray-core/proxy/trojan"
|
_ "github.com/xtls/xray-core/proxy/trojan"
|
||||||
@@ -55,14 +54,12 @@ import (
|
|||||||
_ "github.com/xtls/xray-core/transport/internet/grpc"
|
_ "github.com/xtls/xray-core/transport/internet/grpc"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/httpupgrade"
|
_ "github.com/xtls/xray-core/transport/internet/httpupgrade"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/kcp"
|
_ "github.com/xtls/xray-core/transport/internet/kcp"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/masque"
|
|
||||||
_ "github.com/xtls/xray-core/transport/internet/reality"
|
_ "github.com/xtls/xray-core/transport/internet/reality"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/splithttp"
|
_ "github.com/xtls/xray-core/transport/internet/splithttp"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/tcp"
|
_ "github.com/xtls/xray-core/transport/internet/tcp"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/tls"
|
_ "github.com/xtls/xray-core/transport/internet/tls"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/udp"
|
_ "github.com/xtls/xray-core/transport/internet/udp"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/websocket"
|
_ "github.com/xtls/xray-core/transport/internet/websocket"
|
||||||
_ "github.com/xtls/xray-core/transport/internet/xdrive"
|
|
||||||
|
|
||||||
// Transport headers
|
// Transport headers
|
||||||
_ "github.com/xtls/xray-core/transport/internet/headers/http"
|
_ "github.com/xtls/xray-core/transport/internet/headers/http"
|
||||||
|
|||||||
+102
-96
@@ -53,10 +53,6 @@ func reloadEnvSettings() error {
|
|||||||
func init() {
|
func init() {
|
||||||
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
|
common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
|
||||||
h := new(Handler)
|
h := new(Handler)
|
||||||
if streamSettings, ok := session.StreamSettingsFromContext(ctx).(*internet.MemoryStreamConfig); ok && streamSettings.SocketSettings != nil {
|
|
||||||
h.resolveStrategy = streamSettings.SocketSettings.DomainStrategy
|
|
||||||
h.usesDialerProxy = len(streamSettings.SocketSettings.DialerProxy) > 0
|
|
||||||
}
|
|
||||||
if err := core.RequireFeatures(ctx, func(pm policy.Manager) error {
|
if err := core.RequireFeatures(ctx, func(pm policy.Manager) error {
|
||||||
return h.Init(config.(*Config), pm)
|
return h.Init(config.(*Config), pm)
|
||||||
}); err != nil {
|
}); err != nil {
|
||||||
@@ -93,11 +89,9 @@ type FinalRule struct {
|
|||||||
|
|
||||||
// Handler handles Freedom connections.
|
// Handler handles Freedom connections.
|
||||||
type Handler struct {
|
type Handler struct {
|
||||||
policyManager policy.Manager
|
policyManager policy.Manager
|
||||||
config *Config
|
config *Config
|
||||||
finalRules []*FinalRule
|
finalRules []*FinalRule
|
||||||
resolveStrategy internet.DomainStrategy
|
|
||||||
usesDialerProxy bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func buildFinalRule(config *FinalRuleConfig) (*FinalRule, error) {
|
func buildFinalRule(config *FinalRuleConfig) (*FinalRule, error) {
|
||||||
@@ -174,6 +168,22 @@ func getDefaultFinalRule(inbound *session.Inbound) *FinalRule {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (h *Handler) shouldResolveDomainBeforeFinalRules(dialDest net.Destination, defaultRule *FinalRule) bool {
|
||||||
|
if !dialDest.Address.Family().IsDomain() {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if len(h.finalRules) > 0 {
|
||||||
|
rule := h.finalRules[0]
|
||||||
|
if rule.action == RuleAction_Allow && rule.network[dialDest.Network] && len(rule.port) == 0 && rule.ip == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if defaultRule != nil || len(h.finalRules) > 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
func (h *Handler) matchFinalRule(network net.Network, address net.Address, port net.Port, defaultRule *FinalRule) *FinalRule {
|
func (h *Handler) matchFinalRule(network net.Network, address net.Address, port net.Port, defaultRule *FinalRule) *FinalRule {
|
||||||
for _, rule := range h.finalRules {
|
for _, rule := range h.finalRules {
|
||||||
if rule.Apply(network, address, port) {
|
if rule.Apply(network, address, port) {
|
||||||
@@ -186,16 +196,17 @@ func (h *Handler) matchFinalRule(network net.Network, address net.Address, port
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (h *Handler) applyFinalRules(network net.Network, address net.Address, port net.Port, defaultRule *FinalRule) RuleAction {
|
||||||
|
if rule := h.matchFinalRule(network, address, port, defaultRule); rule != nil {
|
||||||
|
return rule.action
|
||||||
|
}
|
||||||
|
return RuleAction_Allow
|
||||||
|
}
|
||||||
|
|
||||||
// Init initializes the Handler with necessary parameters.
|
// Init initializes the Handler with necessary parameters.
|
||||||
func (h *Handler) Init(config *Config, pm policy.Manager) error {
|
func (h *Handler) Init(config *Config, pm policy.Manager) error {
|
||||||
h.config = config
|
h.config = config
|
||||||
h.policyManager = pm
|
h.policyManager = pm
|
||||||
if h.usesDialerProxy { // freedom is not the final outbound, final rules do not apply
|
|
||||||
if len(config.FinalRules) > 0 {
|
|
||||||
errors.LogWarning(context.Background(), `The "finalRules" setting is ignored when "sockopt.dialerProxy" is set, since freedom is not the final outbound.`)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
h.finalRules = make([]*FinalRule, 0, len(config.FinalRules))
|
h.finalRules = make([]*FinalRule, 0, len(config.FinalRules))
|
||||||
for _, rc := range config.FinalRules {
|
for _, rc := range config.FinalRules {
|
||||||
rule, err := buildFinalRule(rc)
|
rule, err := buildFinalRule(rc)
|
||||||
@@ -226,20 +237,6 @@ func (h *Handler) blockDelay(rule *FinalRule) time.Duration {
|
|||||||
return time.Duration(min+uint64(dice.Roll(int(span+1)))) * time.Second
|
return time.Duration(min+uint64(dice.Roll(int(span+1)))) * time.Second
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Handler) blackhole(ctx context.Context, input buf.Reader, output buf.Writer, rule *FinalRule, dest *net.Destination) error {
|
|
||||||
delay := h.blockDelay(rule)
|
|
||||||
errors.LogInfo(ctx, "blocked target: ", *dest, ", blackholing connection for ", delay)
|
|
||||||
timer := time.AfterFunc(delay, func() {
|
|
||||||
common.Interrupt(input)
|
|
||||||
common.Interrupt(output)
|
|
||||||
errors.LogInfo(ctx, "closed blackholed connection to blocked target: ", *dest)
|
|
||||||
})
|
|
||||||
defer timer.Stop()
|
|
||||||
defer common.Close(output)
|
|
||||||
_ = buf.Copy(input, buf.Discard)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func isValidAddress(addr *net.IPOrDomain) bool {
|
func isValidAddress(addr *net.IPOrDomain) bool {
|
||||||
if addr == nil {
|
if addr == nil {
|
||||||
return false
|
return false
|
||||||
@@ -259,10 +256,7 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
ob.Name = "freedom"
|
ob.Name = "freedom"
|
||||||
ob.CanSpliceCopy = 1
|
ob.CanSpliceCopy = 1
|
||||||
inbound := session.InboundFromContext(ctx)
|
inbound := session.InboundFromContext(ctx)
|
||||||
var defaultRule *FinalRule
|
defaultRule := getDefaultFinalRule(inbound)
|
||||||
if !h.usesDialerProxy { // freedom is not the final outbound, final rules do not apply (and the domain is not resolved)
|
|
||||||
defaultRule = getDefaultFinalRule(inbound)
|
|
||||||
}
|
|
||||||
|
|
||||||
destination := ob.Target
|
destination := ob.Target
|
||||||
origTargetAddr := ob.OriginalTarget.Address
|
origTargetAddr := ob.OriginalTarget.Address
|
||||||
@@ -290,53 +284,61 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
var conn stat.Connection
|
var conn stat.Connection
|
||||||
var blockedDest *net.Destination
|
var blockedDest *net.Destination
|
||||||
var blockedRule *FinalRule
|
var blockedRule *FinalRule
|
||||||
|
firstResolve := true
|
||||||
err := retry.ExponentialBackoff(5, 100).On(func() error {
|
err := retry.ExponentialBackoff(5, 100).On(func() error {
|
||||||
if destination.Address.Family().IsDomain() {
|
dialDest := destination
|
||||||
if defaultRule != nil || len(h.finalRules) > 0 {
|
if h.config.DomainStrategy.HasStrategy() && dialDest.Address.Family().IsDomain() {
|
||||||
if strategy := h.resolveStrategy; strategy.HasStrategy() {
|
strategy := h.config.DomainStrategy
|
||||||
ips, err := internet.LookupForIP(destination.Address.Domain(), strategy, outGateway)
|
if destination.Network == net.Network_UDP && origTargetAddr != nil && outGateway == nil {
|
||||||
if err != nil { // non-force may still dial with system DNS
|
strategy = strategy.GetDynamicStrategy(origTargetAddr.Family())
|
||||||
errors.LogInfoInner(ctx, err, "failed to get IP address for domain ", destination.Address.Domain())
|
|
||||||
if strategy.ForceIP() {
|
|
||||||
return err // retry
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for _, ip := range ips {
|
|
||||||
if addr := net.IPAddress(ip); addr != nil {
|
|
||||||
if rule := h.matchFinalRule(destination.Network, addr, destination.Port, defaultRule); rule != nil && rule.action == RuleAction_Block {
|
|
||||||
blockedDest = &destination
|
|
||||||
blockedDest.Address = addr
|
|
||||||
blockedRule = rule
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
addrs, err := net.DefaultResolver.LookupIPAddr(ctx, destination.Address.Domain())
|
|
||||||
if err != nil { // dialer may retry DNS
|
|
||||||
errors.LogInfoInner(ctx, err, "failed to get IP address for domain ", destination.Address.Domain())
|
|
||||||
}
|
|
||||||
for _, addr := range addrs {
|
|
||||||
if ipAddr := net.IPAddress(addr.IP); ipAddr != nil {
|
|
||||||
if rule := h.matchFinalRule(destination.Network, ipAddr, destination.Port, defaultRule); rule != nil && rule.action == RuleAction_Block {
|
|
||||||
blockedDest = &destination
|
|
||||||
blockedDest.Address = ipAddr
|
|
||||||
blockedRule = rule
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
} else {
|
ips, err := internet.LookupForIP(dialDest.Address.Domain(), strategy, outGateway)
|
||||||
if rule := h.matchFinalRule(destination.Network, destination.Address, destination.Port, defaultRule); rule != nil && rule.action == RuleAction_Block {
|
if err != nil {
|
||||||
blockedDest = &destination
|
errors.LogInfoInner(ctx, err, "failed to get IP address for domain ", dialDest.Address.Domain())
|
||||||
blockedRule = rule
|
if h.config.DomainStrategy.ForceIP() || h.shouldResolveDomainBeforeFinalRules(dialDest, defaultRule) {
|
||||||
return nil
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
dialDest = net.Destination{
|
||||||
|
Network: dialDest.Network,
|
||||||
|
Address: net.IPAddress(ips[dice.Roll(len(ips))]),
|
||||||
|
Port: dialDest.Port,
|
||||||
|
}
|
||||||
|
errors.LogInfo(ctx, "dialing to ", dialDest)
|
||||||
|
}
|
||||||
|
} else if h.shouldResolveDomainBeforeFinalRules(dialDest, defaultRule) { // asis + domain + hasrules
|
||||||
|
domain := dialDest.Address.Domain()
|
||||||
|
var ips []net.IP
|
||||||
|
if firstResolve {
|
||||||
|
firstResolve = false
|
||||||
|
supportIPv4, supportIPv6 := utils.CheckRoutes()
|
||||||
|
if supportIPv4 {
|
||||||
|
ips, _ = net.DefaultResolver.LookupIP(ctx, "ip4", domain)
|
||||||
|
}
|
||||||
|
if len(ips) == 0 && supportIPv6 {
|
||||||
|
ips, _ = net.DefaultResolver.LookupIP(ctx, "ip6", domain)
|
||||||
|
}
|
||||||
|
if len(ips) == 0 {
|
||||||
|
return errors.New("failed to get IP address for domain ", domain)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
ips, _ = net.DefaultResolver.LookupIP(ctx, "ip", domain)
|
||||||
|
}
|
||||||
|
if len(ips) == 0 { // SRV/TXT, lookup failed
|
||||||
|
return errors.New("failed to get IP address for domain ", domain)
|
||||||
|
}
|
||||||
|
if addr := net.IPAddress(ips[dice.Roll(len(ips))]); addr != nil {
|
||||||
|
dialDest.Address = addr
|
||||||
|
errors.LogInfo(ctx, "dialing to ", dialDest)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if rule := h.matchFinalRule(dialDest.Network, dialDest.Address, dialDest.Port, defaultRule); rule != nil && rule.action == RuleAction_Block {
|
||||||
|
blockedDest = &dialDest
|
||||||
|
blockedRule = rule
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
rawConn, err := dialer.Dial(ctx, destination)
|
rawConn, err := dialer.Dial(ctx, dialDest)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -348,17 +350,20 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
return errors.New("failed to open connection to ", destination).Base(err)
|
return errors.New("failed to open connection to ", destination).Base(err)
|
||||||
}
|
}
|
||||||
if blockedDest != nil {
|
if blockedDest != nil {
|
||||||
return h.blackhole(ctx, input, output, blockedRule, blockedDest)
|
delay := h.blockDelay(blockedRule)
|
||||||
}
|
errors.LogInfo(ctx, "blocked target: ", *blockedDest, ", blackholing connection for ", delay)
|
||||||
if destination.Address.Family().IsDomain() && (defaultRule != nil || len(h.finalRules) > 0) {
|
timer := time.AfterFunc(delay, func() {
|
||||||
// pre-check may fail or dialer may select another IP
|
common.Interrupt(input)
|
||||||
remoteDest := net.DestinationFromAddr(conn.RemoteAddr())
|
common.Interrupt(output)
|
||||||
if rule := h.matchFinalRule(remoteDest.Network, remoteDest.Address, remoteDest.Port, defaultRule); rule != nil && rule.action == RuleAction_Block {
|
errors.LogInfo(ctx, "closed blackholed connection to blocked target: ", *blockedDest)
|
||||||
conn.Close()
|
})
|
||||||
return h.blackhole(ctx, input, output, rule, &remoteDest)
|
defer timer.Stop()
|
||||||
|
defer common.Close(output)
|
||||||
|
if err := buf.Copy(input, buf.Discard); err != nil {
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if h.config.ProxyProtocol > 0 && h.config.ProxyProtocol <= 2 {
|
if h.config.ProxyProtocol > 0 && h.config.ProxyProtocol <= 2 {
|
||||||
version := byte(h.config.ProxyProtocol)
|
version := byte(h.config.ProxyProtocol)
|
||||||
srcAddr := inbound.Source.RawNetAddr()
|
srcAddr := inbound.Source.RawNetAddr()
|
||||||
@@ -403,7 +408,7 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
writer = buf.NewWriter(conn)
|
writer = buf.NewWriter(conn)
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
writer = NewPacketWriter(conn, h, defaultRule, UDPOverride, destination, outGateway)
|
writer = NewPacketWriter(conn, h, defaultRule, UDPOverride, destination)
|
||||||
if h.config.Noises != nil {
|
if h.config.Noises != nil {
|
||||||
errors.LogDebug(ctx, "NOISE", h.config.Noises)
|
errors.LogDebug(ctx, "NOISE", h.config.Noises)
|
||||||
writer = &NoisePacketWriter{
|
writer = &NoisePacketWriter{
|
||||||
@@ -507,7 +512,7 @@ func (r *PacketReader) ReadMultiBuffer() (buf.MultiBuffer, error) {
|
|||||||
}
|
}
|
||||||
udpAddr := d.(*net.UDPAddr)
|
udpAddr := d.(*net.UDPAddr)
|
||||||
sourceAddr := net.IPAddress(udpAddr.IP)
|
sourceAddr := net.IPAddress(udpAddr.IP)
|
||||||
if rule := r.Handler.matchFinalRule(net.Network_UDP, sourceAddr, net.Port(udpAddr.Port), r.DefaultRule); rule != nil && rule.action == RuleAction_Block {
|
if r.Handler.applyFinalRules(net.Network_UDP, sourceAddr, net.Port(udpAddr.Port), r.DefaultRule) == RuleAction_Block {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
b.Resize(0, int32(n))
|
b.Resize(0, int32(n))
|
||||||
@@ -532,7 +537,7 @@ func (r *PacketReader) ReadMultiBuffer() (buf.MultiBuffer, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// DialDest means the dial target used in the dialer when creating conn
|
// DialDest means the dial target used in the dialer when creating conn
|
||||||
func NewPacketWriter(conn net.Conn, h *Handler, defaultRule *FinalRule, UDPOverride net.Destination, DialDest net.Destination, outGateway net.Address) buf.Writer {
|
func NewPacketWriter(conn net.Conn, h *Handler, defaultRule *FinalRule, UDPOverride net.Destination, DialDest net.Destination) buf.Writer {
|
||||||
iConn := conn
|
iConn := conn
|
||||||
statConn, ok := iConn.(*stat.CounterConnection)
|
statConn, ok := iConn.(*stat.CounterConnection)
|
||||||
if ok {
|
if ok {
|
||||||
@@ -556,8 +561,9 @@ func NewPacketWriter(conn net.Conn, h *Handler, defaultRule *FinalRule, UDPOverr
|
|||||||
DefaultRule: defaultRule,
|
DefaultRule: defaultRule,
|
||||||
UDPOverride: UDPOverride,
|
UDPOverride: UDPOverride,
|
||||||
ResolvedUDPAddr: resolvedUDPAddr,
|
ResolvedUDPAddr: resolvedUDPAddr,
|
||||||
OutGateway: outGateway,
|
LocalAddr: net.DestinationFromAddr(conn.LocalAddr()).Address,
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
return &buf.SequentialWriter{Writer: conn}
|
return &buf.SequentialWriter{Writer: conn}
|
||||||
}
|
}
|
||||||
@@ -574,7 +580,7 @@ type PacketWriter struct {
|
|||||||
// Resulting in these packets being sent to many different IPs randomly
|
// Resulting in these packets being sent to many different IPs randomly
|
||||||
// So, cache and keep the resolve result
|
// So, cache and keep the resolve result
|
||||||
ResolvedUDPAddr *utils.TypedSyncMap[string, net.Address]
|
ResolvedUDPAddr *utils.TypedSyncMap[string, net.Address]
|
||||||
OutGateway net.Address
|
LocalAddr net.Address
|
||||||
}
|
}
|
||||||
|
|
||||||
func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
||||||
@@ -597,21 +603,21 @@ func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
|||||||
if ip, ok := w.ResolvedUDPAddr.Load(b.UDP.Address.Domain()); ok {
|
if ip, ok := w.ResolvedUDPAddr.Load(b.UDP.Address.Domain()); ok {
|
||||||
b.UDP.Address = ip
|
b.UDP.Address = ip
|
||||||
} else {
|
} else {
|
||||||
shouldUseSystemResolver := true
|
ShouldUseSystemResolver := true
|
||||||
if strategy := w.Handler.resolveStrategy; strategy.HasStrategy() {
|
if w.Handler.config.DomainStrategy.HasStrategy() {
|
||||||
ips, err := internet.LookupForIP(b.UDP.Address.Domain(), strategy, w.OutGateway)
|
ips, err := internet.LookupForIP(b.UDP.Address.Domain(), w.Handler.config.DomainStrategy, w.LocalAddr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// drop packet if resolve failed when forceIP
|
// drop packet if resolve failed when forceIP
|
||||||
if strategy.ForceIP() {
|
if w.Handler.config.DomainStrategy.ForceIP() {
|
||||||
b.Release()
|
b.Release()
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
ip = net.IPAddress(ips[dice.Roll(len(ips))])
|
ip = net.IPAddress(ips[dice.Roll(len(ips))])
|
||||||
shouldUseSystemResolver = false
|
ShouldUseSystemResolver = false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if shouldUseSystemResolver {
|
if ShouldUseSystemResolver {
|
||||||
udpAddr, err := net.ResolveUDPAddr("udp", b.UDP.NetAddr())
|
udpAddr, err := net.ResolveUDPAddr("udp", b.UDP.NetAddr())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
b.Release()
|
b.Release()
|
||||||
@@ -625,7 +631,7 @@ func (w *PacketWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if rule := w.matchFinalRule(net.Network_UDP, b.UDP.Address, b.UDP.Port, w.DefaultRule); rule != nil && rule.action == RuleAction_Block {
|
if w.applyFinalRules(net.Network_UDP, b.UDP.Address, b.UDP.Port, w.DefaultRule) == RuleAction_Block {
|
||||||
b.Release()
|
b.Release()
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -332,7 +332,7 @@ func readResponseAndHandle100Continue(r *bufio.Reader, req *http.Request, writer
|
|||||||
return nil, errors.New("failed to read http 1xx response").Base(err)
|
return nil, errors.New("failed to read http 1xx response").Base(err)
|
||||||
}
|
}
|
||||||
ResponseHeader1xx = append(ResponseHeader1xx, data...)
|
ResponseHeader1xx = append(ResponseHeader1xx, data...)
|
||||||
if len(ResponseHeader1xx) >= 4 && bytes.Equal(ResponseHeader1xx[len(ResponseHeader1xx)-4:], []byte{'\r', '\n', '\r', '\n'}) {
|
if bytes.Equal(ResponseHeader1xx[len(ResponseHeader1xx)-4:], []byte{'\r', '\n', '\r', '\n'}) {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
if len(ResponseHeader1xx) > 1024 {
|
if len(ResponseHeader1xx) > 1024 {
|
||||||
|
|||||||
@@ -1,49 +0,0 @@
|
|||||||
package http
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bufio"
|
|
||||||
"bytes"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
// A malformed upstream response containing a bare '\n' before the real
|
|
||||||
// status line used to crash readResponseAndHandle100Continue: the first
|
|
||||||
// ReadSlice('\n') returns fewer than 4 bytes, and slicing
|
|
||||||
// ResponseHeader1xx[len(ResponseHeader1xx)-4:] panicked with a negative
|
|
||||||
// index instead of returning an error.
|
|
||||||
func TestReadResponseAndHandle100ContinueDoesNotPanicOnEarlyNewline(t *testing.T) {
|
|
||||||
payload := "X\nHTTP/1.1 100 Continue\r\n\r\n" + strings.Repeat("A", 40)
|
|
||||||
r := bufio.NewReader(bytes.NewReader([]byte(payload)))
|
|
||||||
req, err := http.NewRequest("GET", "http://example.com/", nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Must not panic; a parse error for the garbage trailing bytes is fine.
|
|
||||||
_, _ = readResponseAndHandle100Continue(r, req, io.Discard)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadResponseAndHandle100ContinueForwardsAndParsesFinalResponse(t *testing.T) {
|
|
||||||
payload := "HTTP/1.1 100 Continue\r\n\r\n" +
|
|
||||||
"HTTP/1.1 200 OK\r\nContent-Length: 5\r\n\r\nhello"
|
|
||||||
r := bufio.NewReader(bytes.NewReader([]byte(payload)))
|
|
||||||
req, err := http.NewRequest("GET", "http://example.com/", nil)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var forwarded bytes.Buffer
|
|
||||||
resp, err := readResponseAndHandle100Continue(r, req, &forwarded)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("unexpected error: %v", err)
|
|
||||||
}
|
|
||||||
if resp.StatusCode != 200 {
|
|
||||||
t.Fatalf("expected status 200, got %d", resp.StatusCode)
|
|
||||||
}
|
|
||||||
if !strings.Contains(forwarded.String(), "100 Continue") {
|
|
||||||
t.Fatalf("expected 1xx response to be forwarded, got %q", forwarded.String())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -236,14 +236,14 @@ type UDPReader struct {
|
|||||||
|
|
||||||
func (r *UDPReader) ReadFrom(p []byte) (n int, addr *net.Destination, err error) {
|
func (r *UDPReader) ReadFrom(p []byte) (n int, addr *net.Destination, err error) {
|
||||||
for {
|
for {
|
||||||
var packet [1500]byte
|
var buf [hysteria.MaxDatagramFrameSize]byte
|
||||||
|
|
||||||
n, err := r.reader.Read(packet[:])
|
n, err := r.reader.Read(buf[:])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, nil, err
|
return 0, nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
msg, err := ParseUDPMessage(packet[:n])
|
msg, err := ParseUDPMessage(buf[:n])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,328 +0,0 @@
|
|||||||
package masque
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
go_errors "errors"
|
|
||||||
"io"
|
|
||||||
"net/netip"
|
|
||||||
"slices"
|
|
||||||
"sync"
|
|
||||||
"sync/atomic"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"golang.zx2c4.com/wireguard/tun"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common"
|
|
||||||
"github.com/xtls/xray-core/common/buf"
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/common/protocol"
|
|
||||||
"github.com/xtls/xray-core/common/session"
|
|
||||||
"github.com/xtls/xray-core/common/signal"
|
|
||||||
"github.com/xtls/xray-core/common/task"
|
|
||||||
"github.com/xtls/xray-core/core"
|
|
||||||
"github.com/xtls/xray-core/features/policy"
|
|
||||||
"github.com/xtls/xray-core/proxy/wireguard"
|
|
||||||
"github.com/xtls/xray-core/transport"
|
|
||||||
"github.com/xtls/xray-core/transport/internet"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/masque"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/stat"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/tls"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
establishTimeout = 10 * time.Second
|
|
||||||
retryInterval = time.Second
|
|
||||||
)
|
|
||||||
|
|
||||||
type Client struct {
|
|
||||||
server *protocol.ServerSpec
|
|
||||||
policyManager policy.Manager
|
|
||||||
remoteDNS []netip.Addr
|
|
||||||
|
|
||||||
ctx context.Context
|
|
||||||
cancel context.CancelFunc
|
|
||||||
|
|
||||||
tunnel atomic.Pointer[tunnel]
|
|
||||||
|
|
||||||
mu sync.Mutex
|
|
||||||
lastErr error
|
|
||||||
lastErrAt time.Time
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewClient(ctx context.Context, config *ClientConfig) (*Client, error) {
|
|
||||||
v := core.MustFromContext(ctx)
|
|
||||||
p := v.GetFeature(policy.ManagerType()).(policy.Manager)
|
|
||||||
|
|
||||||
streamSettings := session.StreamSettingsFromContext(ctx).(*internet.MemoryStreamConfig)
|
|
||||||
if _, ok := streamSettings.ProtocolSettings.(*masque.Config); !ok {
|
|
||||||
return nil, errors.New("not masque transport")
|
|
||||||
}
|
|
||||||
if tls.ConfigFromStreamSettings(streamSettings) == nil {
|
|
||||||
return nil, errors.New(`MASQUE requires "security": "tls"`)
|
|
||||||
}
|
|
||||||
if config.Server == nil {
|
|
||||||
return nil, errors.New(`no target server found`)
|
|
||||||
}
|
|
||||||
server, err := protocol.NewServerSpecFromPB(config.Server)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("failed to get server spec").Base(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
dns := config.RemoteDns
|
|
||||||
if len(dns) == 0 {
|
|
||||||
dns = []string{"1.1.1.1", "1.0.0.1", "2606:4700:4700::1111", "2606:4700:4700::1001"}
|
|
||||||
}
|
|
||||||
remoteDNS := make([]netip.Addr, 0, len(dns))
|
|
||||||
for _, s := range dns {
|
|
||||||
addr, err := netip.ParseAddr(s)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("invalid remote DNS server ", s).Base(err)
|
|
||||||
}
|
|
||||||
remoteDNS = append(remoteDNS, addr)
|
|
||||||
}
|
|
||||||
|
|
||||||
c := &Client{
|
|
||||||
server: server,
|
|
||||||
policyManager: p,
|
|
||||||
remoteDNS: remoteDNS,
|
|
||||||
}
|
|
||||||
c.ctx, c.cancel = context.WithCancel(context.Background())
|
|
||||||
return c, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) Process(ctx context.Context, link *transport.Link, dialer internet.Dialer) error {
|
|
||||||
outbounds := session.OutboundsFromContext(ctx)
|
|
||||||
ob := outbounds[len(outbounds)-1]
|
|
||||||
if !ob.Target.IsValid() {
|
|
||||||
return errors.New("target not specified")
|
|
||||||
}
|
|
||||||
ob.Name = "masque"
|
|
||||||
ob.CanSpliceCopy = 3
|
|
||||||
|
|
||||||
t, err := c.getTunnel(ctx, dialer)
|
|
||||||
if err != nil {
|
|
||||||
return errors.New("failed to establish CONNECT-IP tunnel").Base(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var newCtx context.Context
|
|
||||||
var newCancel context.CancelFunc
|
|
||||||
if session.TimeoutOnlyFromContext(ctx) {
|
|
||||||
newCtx, newCancel = context.WithCancel(context.Background())
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionPolicy := c.policyManager.ForLevel(0)
|
|
||||||
ctx, cancel := context.WithCancel(ctx)
|
|
||||||
timer := signal.CancelAfterInactivity(ctx, func() {
|
|
||||||
cancel()
|
|
||||||
if newCancel != nil {
|
|
||||||
newCancel()
|
|
||||||
}
|
|
||||||
}, sessionPolicy.Timeouts.ConnectionIdle)
|
|
||||||
|
|
||||||
if newCtx != nil {
|
|
||||||
ctx = newCtx
|
|
||||||
}
|
|
||||||
|
|
||||||
var reader buf.Reader
|
|
||||||
var writer buf.Writer
|
|
||||||
|
|
||||||
switch ob.Target.Network {
|
|
||||||
case net.Network_TCP:
|
|
||||||
var conn net.Conn
|
|
||||||
var err error
|
|
||||||
if sessionPolicy.Timeouts.Handshake != 0 {
|
|
||||||
timeoutCtx, timeoutCancel := context.WithTimeout(ctx, sessionPolicy.Timeouts.Handshake)
|
|
||||||
conn, err = t.tnet.DialContext(timeoutCtx, "tcp", ob.Target.NetAddr())
|
|
||||||
timeoutCancel()
|
|
||||||
} else {
|
|
||||||
conn, err = t.tnet.Dial("tcp", ob.Target.NetAddr())
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return errors.New("failed to create TCP connection").Base(err)
|
|
||||||
}
|
|
||||||
defer conn.Close()
|
|
||||||
reader = buf.NewReader(conn)
|
|
||||||
writer = buf.NewWriter(conn)
|
|
||||||
case net.Network_UDP:
|
|
||||||
conn, err := t.tnet.Dial("udp", ob.Target.NetAddr())
|
|
||||||
if err != nil {
|
|
||||||
return errors.New("failed to create UDP connection").Base(err)
|
|
||||||
}
|
|
||||||
defer conn.Close()
|
|
||||||
uc := &wireguard.UDPConnClient{
|
|
||||||
PacketConn: conn.(*internet.PacketConnWrapper).PacketConn,
|
|
||||||
Dest: conn.RemoteAddr().(*net.UDPAddr),
|
|
||||||
}
|
|
||||||
reader = uc
|
|
||||||
writer = uc
|
|
||||||
default:
|
|
||||||
panic(ob.Target.Network)
|
|
||||||
}
|
|
||||||
|
|
||||||
requestFunc := func() error {
|
|
||||||
defer timer.SetTimeout(sessionPolicy.Timeouts.DownlinkOnly)
|
|
||||||
return buf.Copy(link.Reader, writer, buf.UpdateActivity(timer))
|
|
||||||
}
|
|
||||||
|
|
||||||
responseFunc := func() error {
|
|
||||||
defer timer.SetTimeout(sessionPolicy.Timeouts.UplinkOnly)
|
|
||||||
return buf.Copy(reader, link.Writer, buf.UpdateActivity(timer))
|
|
||||||
}
|
|
||||||
|
|
||||||
responseDonePost := task.OnSuccess(responseFunc, task.Close(link.Writer))
|
|
||||||
if err := task.Run(ctx, requestFunc, responseDonePost); err != nil {
|
|
||||||
common.Interrupt(link.Reader)
|
|
||||||
common.Interrupt(link.Writer)
|
|
||||||
return errors.New("connection ends").Base(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) getTunnel(ctx context.Context, dialer internet.Dialer) (*tunnel, error) {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
if c.ctx.Err() != nil {
|
|
||||||
return nil, errors.New("closed")
|
|
||||||
}
|
|
||||||
if t := c.tunnel.Load(); t != nil {
|
|
||||||
select {
|
|
||||||
case <-t.done:
|
|
||||||
default:
|
|
||||||
return t, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if err := ctx.Err(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if c.lastErr != nil && time.Since(c.lastErrAt) < retryInterval {
|
|
||||||
return nil, c.lastErr
|
|
||||||
}
|
|
||||||
|
|
||||||
t, err := c.establish(ctx, dialer)
|
|
||||||
if err != nil {
|
|
||||||
c.lastErr, c.lastErrAt = err, time.Now()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
c.lastErr = nil
|
|
||||||
c.tunnel.Store(t)
|
|
||||||
if c.ctx.Err() != nil {
|
|
||||||
if c.tunnel.CompareAndSwap(t, nil) {
|
|
||||||
t.close()
|
|
||||||
}
|
|
||||||
return nil, errors.New("closed")
|
|
||||||
}
|
|
||||||
return t, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) establish(ctx context.Context, dialer internet.Dialer) (*tunnel, error) {
|
|
||||||
ctx, cancel := context.WithTimeout(context.WithoutCancel(ctx), establishTimeout)
|
|
||||||
defer cancel()
|
|
||||||
defer context.AfterFunc(c.ctx, cancel)()
|
|
||||||
conn, err := dialer.Dial(ctx, c.server.Destination)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
mconn, ok := stat.TryUnwrapStatsConn(conn).(*masque.Conn)
|
|
||||||
if !ok {
|
|
||||||
conn.Close()
|
|
||||||
return nil, errors.New("not a CONNECT-IP connection")
|
|
||||||
}
|
|
||||||
t, err := newTunnel(conn, mconn.LocalAddrs(), c.remoteDNS)
|
|
||||||
if err != nil {
|
|
||||||
conn.Close()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
errors.LogInfo(ctx, "MASQUE: tunnel established from ", mconn.LocalAddrs())
|
|
||||||
return t, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) Close() error {
|
|
||||||
c.cancel()
|
|
||||||
if t := c.tunnel.Swap(nil); t != nil {
|
|
||||||
t.close()
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type tunnel struct {
|
|
||||||
conn stat.Connection
|
|
||||||
dev tun.Device
|
|
||||||
tnet *wireguard.Net
|
|
||||||
done chan struct{}
|
|
||||||
closeOnce sync.Once
|
|
||||||
}
|
|
||||||
|
|
||||||
func newTunnel(conn stat.Connection, local []netip.Addr, remoteDNS []netip.Addr) (*tunnel, error) {
|
|
||||||
var dns []netip.Addr
|
|
||||||
for _, addr := range remoteDNS {
|
|
||||||
if slices.ContainsFunc(local, func(l netip.Addr) bool { return l.Is4() == addr.Is4() }) {
|
|
||||||
dns = append(dns, addr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(dns) == 0 {
|
|
||||||
errors.LogWarning(context.Background(), "MASQUE: no remote DNS server is reachable from the assigned addresses ", local, ", domain names will fail to resolve")
|
|
||||||
dns = remoteDNS
|
|
||||||
}
|
|
||||||
|
|
||||||
dev, tnet, _, err := wireguard.CreateNetTUN(local, dns, masque.MinPacketSize, true)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
t := &tunnel{
|
|
||||||
conn: conn,
|
|
||||||
dev: dev,
|
|
||||||
tnet: tnet,
|
|
||||||
done: make(chan struct{}),
|
|
||||||
}
|
|
||||||
go t.readFromTunnel()
|
|
||||||
go t.writeToTunnel()
|
|
||||||
return t, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *tunnel) readFromTunnel() {
|
|
||||||
defer t.close()
|
|
||||||
b := make([]byte, buf.Size)
|
|
||||||
for {
|
|
||||||
n, err := t.conn.Read(b)
|
|
||||||
if err != nil {
|
|
||||||
if go_errors.Is(err, io.ErrShortBuffer) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
errors.LogInfoInner(context.Background(), err, "MASQUE: tunnel closed")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
t.dev.Write([][]byte{b[:n]}, 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *tunnel) writeToTunnel() {
|
|
||||||
bufs := [][]byte{make([]byte, masque.MinPacketSize)}
|
|
||||||
sizes := []int{0}
|
|
||||||
for {
|
|
||||||
if _, err := t.dev.Read(bufs, sizes, 0); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if _, err := t.conn.Write(bufs[0][:sizes[0]]); err != nil {
|
|
||||||
var ptb *masque.PacketTooBigError
|
|
||||||
if go_errors.As(err, &ptb) {
|
|
||||||
go t.dev.Write([][]byte{ptb.ICMP}, 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *tunnel) close() {
|
|
||||||
t.closeOnce.Do(func() {
|
|
||||||
close(t.done)
|
|
||||||
t.conn.Close()
|
|
||||||
t.dev.Close()
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
common.Must(common.RegisterConfig((*ClientConfig)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
|
|
||||||
return NewClient(ctx, config.(*ClientConfig))
|
|
||||||
}))
|
|
||||||
}
|
|
||||||
@@ -1,136 +0,0 @@
|
|||||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|
||||||
// versions:
|
|
||||||
// protoc-gen-go v1.36.11
|
|
||||||
// protoc v6.33.5
|
|
||||||
// source: proxy/masque/config.proto
|
|
||||||
|
|
||||||
package masque
|
|
||||||
|
|
||||||
import (
|
|
||||||
protocol "github.com/xtls/xray-core/common/protocol"
|
|
||||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
|
||||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
|
||||||
reflect "reflect"
|
|
||||||
sync "sync"
|
|
||||||
unsafe "unsafe"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
// Verify that this generated code is sufficiently up-to-date.
|
|
||||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
|
||||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
|
||||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
|
||||||
)
|
|
||||||
|
|
||||||
type ClientConfig struct {
|
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
|
||||||
Server *protocol.ServerEndpoint `protobuf:"bytes,1,opt,name=server,proto3" json:"server,omitempty"`
|
|
||||||
RemoteDns []string `protobuf:"bytes,2,rep,name=remote_dns,json=remoteDns,proto3" json:"remote_dns,omitempty"`
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *ClientConfig) Reset() {
|
|
||||||
*x = ClientConfig{}
|
|
||||||
mi := &file_proxy_masque_config_proto_msgTypes[0]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *ClientConfig) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*ClientConfig) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *ClientConfig) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_proxy_masque_config_proto_msgTypes[0]
|
|
||||||
if x != nil {
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
if ms.LoadMessageInfo() == nil {
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
return ms
|
|
||||||
}
|
|
||||||
return mi.MessageOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deprecated: Use ClientConfig.ProtoReflect.Descriptor instead.
|
|
||||||
func (*ClientConfig) Descriptor() ([]byte, []int) {
|
|
||||||
return file_proxy_masque_config_proto_rawDescGZIP(), []int{0}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *ClientConfig) GetServer() *protocol.ServerEndpoint {
|
|
||||||
if x != nil {
|
|
||||||
return x.Server
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *ClientConfig) GetRemoteDns() []string {
|
|
||||||
if x != nil {
|
|
||||||
return x.RemoteDns
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var File_proxy_masque_config_proto protoreflect.FileDescriptor
|
|
||||||
|
|
||||||
const file_proxy_masque_config_proto_rawDesc = "" +
|
|
||||||
"\n" +
|
|
||||||
"\x19proxy/masque/config.proto\x12\x11xray.proxy.masque\x1a!common/protocol/server_spec.proto\"k\n" +
|
|
||||||
"\fClientConfig\x12<\n" +
|
|
||||||
"\x06server\x18\x01 \x01(\v2$.xray.common.protocol.ServerEndpointR\x06server\x12\x1d\n" +
|
|
||||||
"\n" +
|
|
||||||
"remote_dns\x18\x02 \x03(\tR\tremoteDnsBU\n" +
|
|
||||||
"\x15com.xray.proxy.masqueP\x01Z&github.com/xtls/xray-core/proxy/masque\xaa\x02\x11Xray.Proxy.Masqueb\x06proto3"
|
|
||||||
|
|
||||||
var (
|
|
||||||
file_proxy_masque_config_proto_rawDescOnce sync.Once
|
|
||||||
file_proxy_masque_config_proto_rawDescData []byte
|
|
||||||
)
|
|
||||||
|
|
||||||
func file_proxy_masque_config_proto_rawDescGZIP() []byte {
|
|
||||||
file_proxy_masque_config_proto_rawDescOnce.Do(func() {
|
|
||||||
file_proxy_masque_config_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_proxy_masque_config_proto_rawDesc), len(file_proxy_masque_config_proto_rawDesc)))
|
|
||||||
})
|
|
||||||
return file_proxy_masque_config_proto_rawDescData
|
|
||||||
}
|
|
||||||
|
|
||||||
var file_proxy_masque_config_proto_msgTypes = make([]protoimpl.MessageInfo, 1)
|
|
||||||
var file_proxy_masque_config_proto_goTypes = []any{
|
|
||||||
(*ClientConfig)(nil), // 0: xray.proxy.masque.ClientConfig
|
|
||||||
(*protocol.ServerEndpoint)(nil), // 1: xray.common.protocol.ServerEndpoint
|
|
||||||
}
|
|
||||||
var file_proxy_masque_config_proto_depIdxs = []int32{
|
|
||||||
1, // 0: xray.proxy.masque.ClientConfig.server:type_name -> xray.common.protocol.ServerEndpoint
|
|
||||||
1, // [1:1] is the sub-list for method output_type
|
|
||||||
1, // [1:1] is the sub-list for method input_type
|
|
||||||
1, // [1:1] is the sub-list for extension type_name
|
|
||||||
1, // [1:1] is the sub-list for extension extendee
|
|
||||||
0, // [0:1] is the sub-list for field type_name
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() { file_proxy_masque_config_proto_init() }
|
|
||||||
func file_proxy_masque_config_proto_init() {
|
|
||||||
if File_proxy_masque_config_proto != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
type x struct{}
|
|
||||||
out := protoimpl.TypeBuilder{
|
|
||||||
File: protoimpl.DescBuilder{
|
|
||||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
|
||||||
RawDescriptor: unsafe.Slice(unsafe.StringData(file_proxy_masque_config_proto_rawDesc), len(file_proxy_masque_config_proto_rawDesc)),
|
|
||||||
NumEnums: 0,
|
|
||||||
NumMessages: 1,
|
|
||||||
NumExtensions: 0,
|
|
||||||
NumServices: 0,
|
|
||||||
},
|
|
||||||
GoTypes: file_proxy_masque_config_proto_goTypes,
|
|
||||||
DependencyIndexes: file_proxy_masque_config_proto_depIdxs,
|
|
||||||
MessageInfos: file_proxy_masque_config_proto_msgTypes,
|
|
||||||
}.Build()
|
|
||||||
File_proxy_masque_config_proto = out.File
|
|
||||||
file_proxy_masque_config_proto_goTypes = nil
|
|
||||||
file_proxy_masque_config_proto_depIdxs = nil
|
|
||||||
}
|
|
||||||
@@ -1,14 +0,0 @@
|
|||||||
syntax = "proto3";
|
|
||||||
|
|
||||||
package xray.proxy.masque;
|
|
||||||
option csharp_namespace = "Xray.Proxy.Masque";
|
|
||||||
option go_package = "github.com/xtls/xray-core/proxy/masque";
|
|
||||||
option java_package = "com.xray.proxy.masque";
|
|
||||||
option java_multiple_files = true;
|
|
||||||
|
|
||||||
import "common/protocol/server_spec.proto";
|
|
||||||
|
|
||||||
message ClientConfig {
|
|
||||||
xray.common.protocol.ServerEndpoint server = 1;
|
|
||||||
repeated string remote_dns = 2;
|
|
||||||
}
|
|
||||||
@@ -82,7 +82,6 @@ func (o *Outbound) Process(ctx context.Context, link *transport.Link, dialer int
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.New("failed to connect to server").Base(err)
|
return errors.New("failed to connect to server").Base(err)
|
||||||
}
|
}
|
||||||
defer connection.Close()
|
|
||||||
|
|
||||||
if session.TimeoutOnlyFromContext(ctx) {
|
if session.TimeoutOnlyFromContext(ctx) {
|
||||||
ctx, _ = context.WithCancel(context.Background())
|
ctx, _ = context.WithCancel(context.Background())
|
||||||
|
|||||||
+4
-33
@@ -37,25 +37,6 @@ type Handler struct {
|
|||||||
downlinkCounter stats.Counter
|
downlinkCounter stats.Counter
|
||||||
}
|
}
|
||||||
|
|
||||||
type tunUDPStatsWriter struct {
|
|
||||||
writer buf.Writer
|
|
||||||
counter stats.Counter
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *tunUDPStatsWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
|
||||||
for len(mb) > 0 {
|
|
||||||
remaining, packet := buf.SplitFirst(mb)
|
|
||||||
packetSize := packet.Len()
|
|
||||||
if err := w.writer.WriteMultiBuffer(buf.MultiBuffer{packet}); err != nil {
|
|
||||||
buf.ReleaseMulti(remaining)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
w.counter.Add(int64(packetSize))
|
|
||||||
mb = remaining
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// ConnectionHandler interface with the only method that stack is going to push new connections to
|
// ConnectionHandler interface with the only method that stack is going to push new connections to
|
||||||
type ConnectionHandler interface {
|
type ConnectionHandler interface {
|
||||||
HandleConnection(conn net.Conn, destination net.Destination)
|
HandleConnection(conn net.Conn, destination net.Destination)
|
||||||
@@ -123,7 +104,7 @@ func (t *Handler) Start() error {
|
|||||||
iface := updater.Get()
|
iface := updater.Get()
|
||||||
if iface == nil {
|
if iface == nil {
|
||||||
errors.LogInfo(context.Background(), "[tun] falied to set interface > iface == nil")
|
errors.LogInfo(context.Background(), "[tun] falied to set interface > iface == nil")
|
||||||
return errors.New("iface not found")
|
return nil
|
||||||
}
|
}
|
||||||
return c.Control(func(fd uintptr) {
|
return c.Control(func(fd uintptr) {
|
||||||
addrPort, _ := netip.ParseAddrPort(address)
|
addrPort, _ := netip.ParseAddrPort(address)
|
||||||
@@ -190,8 +171,7 @@ func (t *Handler) HandleConnection(conn net.Conn, destination net.Destination) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
source := net.DestinationFromAddr(remote)
|
source := net.DestinationFromAddr(remote)
|
||||||
isUDP := destination.Network == net.Network_UDP
|
if t.uplinkCounter != nil || t.downlinkCounter != nil {
|
||||||
if !isUDP && (t.uplinkCounter != nil || t.downlinkCounter != nil) {
|
|
||||||
conn = &stat.CounterConnection{
|
conn = &stat.CounterConnection{
|
||||||
Connection: conn,
|
Connection: conn,
|
||||||
ReadCounter: t.uplinkCounter,
|
ReadCounter: t.uplinkCounter,
|
||||||
@@ -223,18 +203,9 @@ func (t *Handler) HandleConnection(conn net.Conn, destination net.Destination) {
|
|||||||
})
|
})
|
||||||
errors.LogInfo(ctx, "processing from ", source, " to ", destination)
|
errors.LogInfo(ctx, "processing from ", source, " to ", destination)
|
||||||
|
|
||||||
reader := &buf.TimeoutWrapperReader{Reader: buf.NewReader(conn)}
|
|
||||||
writer := buf.NewWriter(conn)
|
|
||||||
if isUDP {
|
|
||||||
reader.Counter = t.uplinkCounter
|
|
||||||
if t.downlinkCounter != nil {
|
|
||||||
writer = &tunUDPStatsWriter{writer: writer, counter: t.downlinkCounter}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
link := &transport.Link{
|
link := &transport.Link{
|
||||||
Reader: reader,
|
Reader: &buf.TimeoutWrapperReader{Reader: buf.NewReader(conn)},
|
||||||
Writer: writer,
|
Writer: buf.NewWriter(conn),
|
||||||
}
|
}
|
||||||
if err := t.dispatcher.DispatchLink(ctx, destination, link); err != nil {
|
if err := t.dispatcher.DispatchLink(ctx, destination, link); err != nil {
|
||||||
errors.LogError(ctx, errors.New("connection closed").Base(err))
|
errors.LogError(ctx, errors.New("connection closed").Base(err))
|
||||||
|
|||||||
+134
-142
@@ -3,14 +3,14 @@
|
|||||||
package tun
|
package tun
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"crypto/md5"
|
"crypto/md5"
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
go_errors "errors"
|
go_errors "errors"
|
||||||
"net"
|
"net"
|
||||||
"net/netip"
|
"net/netip"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
|
||||||
"unsafe"
|
"unsafe"
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
"github.com/xtls/xray-core/common/errors"
|
||||||
@@ -31,14 +31,13 @@ func procyield(cycles uint32)
|
|||||||
type WindowsTun struct {
|
type WindowsTun struct {
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
|
|
||||||
options *Config
|
options *Config
|
||||||
adapter *wintun.Adapter
|
adapter *wintun.Adapter
|
||||||
session wintun.Session
|
session wintun.Session
|
||||||
readWait windows.Handle
|
readWait windows.Handle
|
||||||
luid winipcfg.LUID
|
luid winipcfg.LUID
|
||||||
cbr winipcfg.ChangeCallback
|
changeCallback winipcfg.ChangeCallback
|
||||||
cbi winipcfg.ChangeCallback
|
closed bool
|
||||||
closed bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// WindowsTun implements Tun
|
// WindowsTun implements Tun
|
||||||
@@ -86,37 +85,23 @@ func open(name, desc string) (*wintun.Adapter, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *WindowsTun) Start() (err error) {
|
func (t *WindowsTun) Start() error {
|
||||||
var address4, address6 bool
|
var has4, has6 bool
|
||||||
addresses := make([]netip.Prefix, 0, len(t.options.Gateway))
|
allowedIPs := make([]netip.Prefix, 0, len(t.options.AutoSystemRoutingTable))
|
||||||
for _, cidr := range t.options.Gateway {
|
for _, route := range t.options.AutoSystemRoutingTable {
|
||||||
prefix := netip.MustParsePrefix(cidr)
|
allowedIPs = append(allowedIPs, netip.MustParsePrefix(route))
|
||||||
if prefix.Addr().Is4() {
|
|
||||||
address4 = true
|
|
||||||
} else {
|
|
||||||
address6 = true
|
|
||||||
}
|
|
||||||
addresses = append(addresses, prefix)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
dns := make([]netip.Addr, 0, len(t.options.DNS))
|
|
||||||
for _, ip := range t.options.DNS {
|
|
||||||
dns = append(dns, netip.MustParseAddr(ip))
|
|
||||||
}
|
|
||||||
|
|
||||||
var route4, route6 bool
|
|
||||||
routesMap := make(map[winipcfg.RouteData]struct{})
|
routesMap := make(map[winipcfg.RouteData]struct{})
|
||||||
for _, cidr := range t.options.AutoSystemRoutingTable {
|
for _, ip := range allowedIPs {
|
||||||
prefix := netip.MustParsePrefix(cidr)
|
|
||||||
route := winipcfg.RouteData{
|
route := winipcfg.RouteData{
|
||||||
Destination: prefix.Masked(),
|
Destination: ip.Masked(),
|
||||||
Metric: 0,
|
Metric: 0,
|
||||||
}
|
}
|
||||||
if prefix.Addr().Is4() {
|
if ip.Addr().Is4() {
|
||||||
route4 = true
|
has4 = true
|
||||||
route.NextHop = netip.IPv4Unspecified()
|
route.NextHop = netip.IPv4Unspecified()
|
||||||
} else {
|
} else {
|
||||||
route6 = true
|
has6 = true
|
||||||
route.NextHop = netip.IPv6Unspecified()
|
route.NextHop = netip.IPv6Unspecified()
|
||||||
}
|
}
|
||||||
routesMap[route] = struct{}{}
|
routesMap[route] = struct{}{}
|
||||||
@@ -126,40 +111,24 @@ func (t *WindowsTun) Start() (err error) {
|
|||||||
r := route
|
r := route
|
||||||
routesData = append(routesData, &r)
|
routesData = append(routesData, &r)
|
||||||
}
|
}
|
||||||
|
err := t.luid.SetRoutes(routesData)
|
||||||
var retryTimes int
|
if err != nil {
|
||||||
var firstErr error
|
return errors.New("unable to set routes").Base(err)
|
||||||
startOver:
|
|
||||||
if retryTimes > 0 {
|
|
||||||
if retryTimes > 15 {
|
|
||||||
return windows.ERROR_NOT_FOUND
|
|
||||||
}
|
|
||||||
errors.LogErrorInner(context.Background(), firstErr, "Interface configuration failed, retrying attempt ", retryTimes, "/15")
|
|
||||||
time.Sleep(time.Second)
|
|
||||||
}
|
}
|
||||||
retryTimes++
|
|
||||||
for _, family := range []winipcfg.AddressFamily{windows.AF_INET, windows.AF_INET6} {
|
if len(t.options.Gateway) > 0 {
|
||||||
if family == windows.AF_INET && route4 || family == windows.AF_INET6 && route6 {
|
addresses := make([]netip.Prefix, 0, len(t.options.Gateway))
|
||||||
err = t.luid.SetRoutesForFamily(family, routesData)
|
for _, address := range t.options.Gateway {
|
||||||
if err != nil {
|
addresses = append(addresses, netip.MustParsePrefix(address))
|
||||||
firstErr = errors.New("unable to set routes").Base(err)
|
|
||||||
if err == windows.ERROR_NOT_FOUND {
|
|
||||||
goto startOver
|
|
||||||
}
|
|
||||||
return firstErr
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
if family == windows.AF_INET && address4 || family == windows.AF_INET6 && address6 {
|
err := t.luid.SetIPAddresses(addresses)
|
||||||
err = t.luid.SetIPAddressesForFamily(family, addresses)
|
if err != nil {
|
||||||
if err != nil {
|
return errors.New("unable to set ips").Base(err)
|
||||||
firstErr = errors.New("unable to set ips").Base(err)
|
|
||||||
if err == windows.ERROR_NOT_FOUND {
|
|
||||||
goto startOver
|
|
||||||
}
|
|
||||||
return firstErr
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
ipif, err := t.luid.IPInterface(family)
|
}
|
||||||
|
|
||||||
|
if has4 {
|
||||||
|
ipif, err := t.luid.IPInterface(windows.AF_INET)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -167,45 +136,56 @@ startOver:
|
|||||||
ipif.DadTransmits = 0
|
ipif.DadTransmits = 0
|
||||||
ipif.ManagedAddressConfigurationSupported = false
|
ipif.ManagedAddressConfigurationSupported = false
|
||||||
ipif.OtherStatefulConfigurationSupported = false
|
ipif.OtherStatefulConfigurationSupported = false
|
||||||
if family == windows.AF_INET && (address4 || route4) || family == windows.AF_INET6 && (address6 || route6) {
|
ipif.NLMTU = t.options.MTU
|
||||||
ipif.NLMTU = t.options.MTU
|
ipif.UseAutomaticMetric = false
|
||||||
}
|
ipif.Metric = 0
|
||||||
if family == windows.AF_INET && route4 || family == windows.AF_INET6 && route6 {
|
|
||||||
ipif.UseAutomaticMetric = false
|
|
||||||
ipif.Metric = 0
|
|
||||||
}
|
|
||||||
err = ipif.Set()
|
err = ipif.Set()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
firstErr = errors.New("unable to set metric and MTU").Base(err)
|
return err
|
||||||
if err == windows.ERROR_NOT_FOUND {
|
|
||||||
goto startOver
|
|
||||||
}
|
|
||||||
return firstErr
|
|
||||||
}
|
}
|
||||||
err = t.luid.SetDNS(family, dns, nil)
|
}
|
||||||
|
if has6 {
|
||||||
|
ipif, err := t.luid.IPInterface(windows.AF_INET6)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
firstErr = errors.New("unable to set DNS").Base(err)
|
return err
|
||||||
if err == windows.ERROR_NOT_FOUND {
|
}
|
||||||
goto startOver
|
ipif.RouterDiscoveryBehavior = winipcfg.RouterDiscoveryDisabled
|
||||||
}
|
ipif.DadTransmits = 0
|
||||||
return firstErr
|
ipif.ManagedAddressConfigurationSupported = false
|
||||||
|
ipif.OtherStatefulConfigurationSupported = false
|
||||||
|
ipif.NLMTU = t.options.MTU
|
||||||
|
ipif.UseAutomaticMetric = false
|
||||||
|
ipif.Metric = 0
|
||||||
|
err = ipif.Set()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(t.options.DNS) > 0 {
|
||||||
|
dns := make([]netip.Addr, 0, len(t.options.DNS))
|
||||||
|
for _, ip := range t.options.DNS {
|
||||||
|
dns = append(dns, netip.MustParseAddr(ip))
|
||||||
|
}
|
||||||
|
err := t.luid.SetDNS(windows.AF_INET, dns, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = t.luid.SetDNS(windows.AF_INET6, dns, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if updater != nil {
|
if updater != nil {
|
||||||
t.cbr, err = winipcfg.RegisterRouteChangeCallback(func(notificationType winipcfg.MibNotificationType, route *winipcfg.MibIPforwardRow2) {
|
t.changeCallback, err = winipcfg.RegisterInterfaceChangeCallback(func(notificationType winipcfg.MibNotificationType, iface *winipcfg.MibIPInterfaceRow) {
|
||||||
updater.Update()
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
t.cbi, err = winipcfg.RegisterInterfaceChangeCallback(func(notificationType winipcfg.MibNotificationType, iface *winipcfg.MibIPInterfaceRow) {
|
|
||||||
updater.Update()
|
updater.Update()
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -217,26 +197,12 @@ func (t *WindowsTun) Close() error {
|
|||||||
}
|
}
|
||||||
t.closed = true
|
t.closed = true
|
||||||
|
|
||||||
if t.cbr != nil {
|
if t.changeCallback != nil {
|
||||||
t.cbr.Unregister()
|
t.changeCallback.Unregister()
|
||||||
}
|
|
||||||
if t.cbi != nil {
|
|
||||||
t.cbi.Unregister()
|
|
||||||
}
|
|
||||||
if t.luid != 0 {
|
|
||||||
t.luid.FlushRoutes(windows.AF_INET)
|
|
||||||
t.luid.FlushIPAddresses(windows.AF_INET)
|
|
||||||
t.luid.FlushDNS(windows.AF_INET)
|
|
||||||
t.luid.FlushRoutes(windows.AF_INET6)
|
|
||||||
t.luid.FlushIPAddresses(windows.AF_INET6)
|
|
||||||
t.luid.FlushDNS(windows.AF_INET6)
|
|
||||||
}
|
|
||||||
if t.session != (wintun.Session{}) {
|
|
||||||
t.session.End()
|
|
||||||
}
|
|
||||||
if t.adapter != nil {
|
|
||||||
t.adapter.Close()
|
|
||||||
}
|
}
|
||||||
|
t.session.End()
|
||||||
|
_ = t.adapter.Close()
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -345,49 +311,75 @@ func setinterface(network, address string, fd uintptr, iface *net.Interface) err
|
|||||||
}
|
}
|
||||||
|
|
||||||
func findOutboundInterface(tunIndex int, fixedName string) (*net.Interface, error) {
|
func findOutboundInterface(tunIndex int, fixedName string) (*net.Interface, error) {
|
||||||
if fixedName != "" {
|
interfaces, err := net.Interfaces()
|
||||||
return net.InterfaceByName(fixedName)
|
|
||||||
}
|
|
||||||
|
|
||||||
r, err := winipcfg.GetIPForwardTable2(windows.AF_UNSPEC)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
lowestMetric := ^uint32(0)
|
|
||||||
index := uint32(0)
|
|
||||||
lowestMetricWifi := ^uint32(0)
|
|
||||||
indexWifi := uint32(0)
|
|
||||||
for i := range r {
|
|
||||||
if r[i].DestinationPrefix.PrefixLength != 0 || r[i].InterfaceIndex == uint32(tunIndex) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
ifrow, err := r[i].InterfaceLUID.Interface()
|
|
||||||
if err != nil || ifrow.OperStatus != winipcfg.IfOperStatusUp {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
iface, err := r[i].InterfaceLUID.IPInterface(windows.AF_INET)
|
if fixedName != "" {
|
||||||
if err != nil {
|
for _, iface := range interfaces {
|
||||||
iface, err = r[i].InterfaceLUID.IPInterface(windows.AF_INET6)
|
if iface.Index != tunIndex && iface.Name == fixedName {
|
||||||
if err != nil {
|
return &iface, nil
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
if ifrow.Type == windows.IF_TYPE_IEEE80211 {
|
var candidates []struct {
|
||||||
if r[i].Metric+iface.Metric < lowestMetricWifi {
|
index int
|
||||||
lowestMetricWifi = r[i].Metric + iface.Metric
|
score int
|
||||||
indexWifi = r[i].InterfaceIndex
|
}
|
||||||
}
|
for i, iface := range interfaces {
|
||||||
|
if iface.Index == tunIndex {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if r[i].Metric+iface.Metric < lowestMetric {
|
if strings.Contains(iface.Name, "vEthernet") {
|
||||||
lowestMetric = r[i].Metric + iface.Metric
|
continue
|
||||||
index = r[i].InterfaceIndex
|
|
||||||
}
|
}
|
||||||
|
if iface.Flags&net.FlagUp == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if iface.Flags&net.FlagLoopback != 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
addrs, err := iface.Addrs()
|
||||||
|
if err != nil || len(addrs) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
candidates = append(candidates, struct {
|
||||||
|
index int
|
||||||
|
score int
|
||||||
|
}{i, scoreWindowsInterface(&iface, addrs)})
|
||||||
}
|
}
|
||||||
if indexWifi != 0 {
|
|
||||||
index = indexWifi
|
sort.Slice(candidates, func(i, j int) bool {
|
||||||
|
if candidates[i].score != candidates[j].score {
|
||||||
|
return candidates[i].score > candidates[j].score
|
||||||
|
}
|
||||||
|
return interfaces[candidates[i].index].Name < interfaces[candidates[j].index].Name
|
||||||
|
})
|
||||||
|
if len(candidates) == 0 {
|
||||||
|
return nil, nil
|
||||||
}
|
}
|
||||||
return net.InterfaceByIndex(int(index))
|
|
||||||
|
iface := interfaces[candidates[0].index]
|
||||||
|
return &iface, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func scoreWindowsInterface(iface *net.Interface, addrs []net.Addr) int {
|
||||||
|
score := 0
|
||||||
|
|
||||||
|
name := strings.ToLower(iface.Name)
|
||||||
|
if strings.Contains(name, "wlan") || strings.Contains(name, "wi-fi") {
|
||||||
|
score += 2
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, addr := range addrs {
|
||||||
|
if strings.HasPrefix(addr.String(), "192.168.") {
|
||||||
|
score++
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return score
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -202,6 +202,8 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
|
ob.Conn = conn // for Vision's pre-connect
|
||||||
|
|
||||||
iConn := stat.TryUnwrapStatsConn(conn)
|
iConn := stat.TryUnwrapStatsConn(conn)
|
||||||
target := ob.Target
|
target := ob.Target
|
||||||
errors.LogInfo(ctx, "tunneling request to ", target, " via ", rec.Destination.NetAddr())
|
errors.LogInfo(ctx, "tunneling request to ", target, " via ", rec.Destination.NetAddr())
|
||||||
|
|||||||
+140
-125
@@ -3,6 +3,7 @@ package wireguard
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
gonet "net"
|
||||||
"net/netip"
|
"net/netip"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
@@ -27,10 +28,14 @@ import (
|
|||||||
"github.com/xtls/xray-core/features/stats"
|
"github.com/xtls/xray-core/features/stats"
|
||||||
"github.com/xtls/xray-core/transport"
|
"github.com/xtls/xray-core/transport"
|
||||||
"github.com/xtls/xray-core/transport/internet"
|
"github.com/xtls/xray-core/transport/internet"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
"golang.zx2c4.com/wireguard/device"
|
"golang.zx2c4.com/wireguard/device"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type entry struct {
|
||||||
|
got []net.IP
|
||||||
|
time time.Time
|
||||||
|
}
|
||||||
|
|
||||||
type Handler struct {
|
type Handler struct {
|
||||||
conf *DeviceConfig
|
conf *DeviceConfig
|
||||||
policyManager policy.Manager
|
policyManager policy.Manager
|
||||||
@@ -44,6 +49,11 @@ type Handler struct {
|
|||||||
tnet *Net
|
tnet *Net
|
||||||
dev *device.Device
|
dev *device.Device
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
|
|
||||||
|
// TODO: cache cleanup loop
|
||||||
|
local bool
|
||||||
|
cache map[string]entry
|
||||||
|
cacheMu sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewClient(ctx context.Context, conf *DeviceConfig) (*Handler, error) {
|
func NewClient(ctx context.Context, conf *DeviceConfig) (*Handler, error) {
|
||||||
@@ -99,10 +109,15 @@ func NewClient(ctx context.Context, conf *DeviceConfig) (*Handler, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
local := false
|
||||||
dns := conf.DNS
|
dns := conf.DNS
|
||||||
if len(dns) == 0 {
|
if len(dns) == 0 {
|
||||||
dns = []string{"1.1.1.1", "1.0.0.1", "2606:4700:4700::1111", "2606:4700:4700::1001"}
|
dns = []string{"1.1.1.1", "1.0.0.1", "2606:4700:4700::1111", "2606:4700:4700::1001"}
|
||||||
}
|
}
|
||||||
|
if len(dns) == 1 && dns[0] == "local" {
|
||||||
|
local = true
|
||||||
|
dns = nil
|
||||||
|
}
|
||||||
dnses := make([]netip.Addr, 0, len(dns))
|
dnses := make([]netip.Addr, 0, len(dns))
|
||||||
for _, dns := range dns {
|
for _, dns := range dns {
|
||||||
dnses = append(dnses, netip.MustParseAddr(dns))
|
dnses = append(dnses, netip.MustParseAddr(dns))
|
||||||
@@ -136,6 +151,9 @@ func NewClient(ctx context.Context, conf *DeviceConfig) (*Handler, error) {
|
|||||||
|
|
||||||
tun: tun,
|
tun: tun,
|
||||||
tnet: tnet,
|
tnet: tnet,
|
||||||
|
|
||||||
|
local: local,
|
||||||
|
cache: make(map[string]entry),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -154,6 +172,22 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var addr netip.Addr
|
||||||
|
if ob.Target.Address.Family().IsDomain() {
|
||||||
|
ip, err := h.resolveRemote(ob.Target.Address.String())
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("failed to resolve domain").Base(err)
|
||||||
|
}
|
||||||
|
addr, _ = netip.AddrFromSlice(ip)
|
||||||
|
} else {
|
||||||
|
addr, _ = netip.AddrFromSlice(ob.Target.Address.IP())
|
||||||
|
}
|
||||||
|
|
||||||
|
addrPort := netip.AddrPortFrom(addr, ob.Target.Port.Value())
|
||||||
|
if !addrPort.IsValid() {
|
||||||
|
return errors.New("invalid target ", ob.Target)
|
||||||
|
}
|
||||||
|
|
||||||
var newCtx context.Context
|
var newCtx context.Context
|
||||||
var newCancel context.CancelFunc
|
var newCancel context.CancelFunc
|
||||||
if session.TimeoutOnlyFromContext(ctx) {
|
if session.TimeoutOnlyFromContext(ctx) {
|
||||||
@@ -182,10 +216,10 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
var err error
|
var err error
|
||||||
if sessionPolicy.Timeouts.Handshake != 0 {
|
if sessionPolicy.Timeouts.Handshake != 0 {
|
||||||
timeoutCtx, timeoutCancel := context.WithTimeout(ctx, sessionPolicy.Timeouts.Handshake)
|
timeoutCtx, timeoutCancel := context.WithTimeout(ctx, sessionPolicy.Timeouts.Handshake)
|
||||||
conn, err = h.tnet.DialContext(timeoutCtx, "tcp", ob.Target.NetAddr())
|
conn, err = h.tnet.DialContextTCPAddrPort(timeoutCtx, addrPort)
|
||||||
timeoutCancel()
|
timeoutCancel()
|
||||||
} else {
|
} else {
|
||||||
conn, err = h.tnet.Dial("tcp", ob.Target.NetAddr())
|
conn, err = h.tnet.DialContextTCPAddrPort(ctx, addrPort)
|
||||||
}
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.New("failed to create TCP connection").Base(err)
|
return errors.New("failed to create TCP connection").Base(err)
|
||||||
@@ -194,14 +228,15 @@ func (h *Handler) Process(ctx context.Context, link *transport.Link, dialer inte
|
|||||||
reader = buf.NewReader(conn)
|
reader = buf.NewReader(conn)
|
||||||
writer = buf.NewWriter(conn)
|
writer = buf.NewWriter(conn)
|
||||||
case net.Network_UDP:
|
case net.Network_UDP:
|
||||||
conn, err := h.tnet.Dial("udp", ob.Target.NetAddr())
|
conn, err := h.tnet.DialUDPAddrPort(netip.AddrPort{}, addrPort)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.New("failed to create UDP connection").Base(err)
|
return errors.New("failed to create UDP connection").Base(err)
|
||||||
}
|
}
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
c := &UDPConnClient{
|
c := &udpConnClient{
|
||||||
PacketConn: conn.(*internet.PacketConnWrapper).PacketConn,
|
PacketConn: conn.(*internet.PacketConnWrapper).PacketConn,
|
||||||
Dest: conn.RemoteAddr().(*net.UDPAddr),
|
resolveFunc: h.resolveRemote,
|
||||||
|
dest: gonet.UDPAddrFromAddrPort(addrPort),
|
||||||
}
|
}
|
||||||
reader = c
|
reader = c
|
||||||
writer = c
|
writer = c
|
||||||
@@ -258,26 +293,26 @@ func (h *Handler) init(ctx context.Context) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
conn, err := internet.DialSystem(ctx, dest, h.streamSettings.SocketSettings)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
var pktConn net.PacketConn
|
var pktConn net.PacketConn
|
||||||
if h.streamSettings.FinalMask != nil {
|
switch c := conn.(type) {
|
||||||
conn, err := h.streamSettings.FinalMask.DialUDP(ctx, dest)
|
case *internet.PacketConnWrapper:
|
||||||
|
pktConn = c.PacketConn
|
||||||
|
case *cnc.Connection:
|
||||||
|
pktConn = &internet.FakePacketConn{Conn: c}
|
||||||
|
default:
|
||||||
|
panic(reflect.TypeOf(c))
|
||||||
|
}
|
||||||
|
if h.streamSettings.UdpmaskManager != nil {
|
||||||
|
newConn, err := h.streamSettings.UdpmaskManager.WrapPacketConnClient(pktConn)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, errors.New("failed to dial to dest").Base(err)
|
pktConn.Close()
|
||||||
}
|
return nil, errors.New("mask err").Base(err)
|
||||||
pktConn = conn.(*finalmask.PacketConnWrapper).PacketConn
|
|
||||||
} else {
|
|
||||||
conn, err := internet.DialSystem(ctx, dest, h.streamSettings.SocketSettings)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errors.New("failed to dial to dest").Base(err)
|
|
||||||
}
|
|
||||||
switch c := conn.(type) {
|
|
||||||
case *internet.PacketConnWrapper:
|
|
||||||
pktConn = c.PacketConn
|
|
||||||
case *cnc.Connection:
|
|
||||||
pktConn = &internet.FakePacketConn{Conn: c}
|
|
||||||
default:
|
|
||||||
panic(reflect.TypeOf(c))
|
|
||||||
}
|
}
|
||||||
|
pktConn = newConn
|
||||||
}
|
}
|
||||||
if h.uplinkCounter != nil || h.downlinkCounter != nil {
|
if h.uplinkCounter != nil || h.downlinkCounter != nil {
|
||||||
pktConn = &PacketCounterConnection{
|
pktConn = &PacketCounterConnection{
|
||||||
@@ -336,54 +371,90 @@ func (h *Handler) init(ctx context.Context) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (h *Handler) resolveLocal(host string) (net.IP, error) {
|
func (h *Handler) resolveLocal(host string) (net.IP, error) {
|
||||||
|
return h.resolveDomain(host, h.conf.DomainStrategy, func(host string) ([]net.IP, uint32, error) {
|
||||||
|
return h.dns.LookupIP(host, dns.IPOption{IPv4Enable: true, IPv6Enable: true})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) resolveRemote(host string) (net.IP, error) {
|
||||||
|
return h.resolveDomain(host, h.conf.DomainStrategy, func(host string) ([]net.IP, uint32, error) {
|
||||||
|
if h.local {
|
||||||
|
return h.dns.LookupIP(host, dns.IPOption{IPv4Enable: true, IPv6Enable: true})
|
||||||
|
}
|
||||||
|
return h.tnet.LookupHost(host)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) resolveDomain(host string, strategy DeviceConfig_DomainStrategy, lookupIP func(host string) ([]net.IP, uint32, error)) (net.IP, error) {
|
||||||
if ip := net.ParseIP(host); ip != nil {
|
if ip := net.ParseIP(host); ip != nil {
|
||||||
return ip, nil
|
return ip, nil
|
||||||
}
|
}
|
||||||
ips, _, err := h.dns.LookupIP(host, dns.IPOption{IPv4Enable: true, IPv6Enable: true})
|
h.cacheMu.Lock()
|
||||||
|
if entry, ok := h.cache[host]; ok {
|
||||||
|
if time.Now().Before(entry.time) {
|
||||||
|
h.cacheMu.Unlock()
|
||||||
|
return entry.got[dice.Roll(len(entry.got))], nil
|
||||||
|
}
|
||||||
|
delete(h.cache, host)
|
||||||
|
}
|
||||||
|
h.cacheMu.Unlock()
|
||||||
|
ips, ttl, err := lookupIP(host)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
got := ips
|
if len(ips) == 0 {
|
||||||
if h.streamSettings.SocketSettings != nil {
|
return nil, dns.ErrEmptyResponse
|
||||||
var got4, got6 []net.IP
|
}
|
||||||
for _, ip := range ips {
|
var got4, got6 []net.IP
|
||||||
if ip.To4() != nil {
|
for _, ip := range ips {
|
||||||
got4 = append(got4, ip)
|
if ip.To4() != nil {
|
||||||
} else {
|
got4 = append(got4, ip)
|
||||||
got6 = append(got6, ip)
|
} else {
|
||||||
}
|
got6 = append(got6, ip)
|
||||||
}
|
|
||||||
switch h.streamSettings.SocketSettings.DomainStrategy {
|
|
||||||
case internet.DomainStrategy_AS_IS, internet.DomainStrategy_USE_IP, internet.DomainStrategy_FORCE_IP:
|
|
||||||
got = ips
|
|
||||||
case internet.DomainStrategy_USE_IP4, internet.DomainStrategy_FORCE_IP4:
|
|
||||||
got = got4
|
|
||||||
case internet.DomainStrategy_USE_IP6, internet.DomainStrategy_FORCE_IP6:
|
|
||||||
got = got6
|
|
||||||
case internet.DomainStrategy_USE_IP46, internet.DomainStrategy_FORCE_IP46:
|
|
||||||
got = got4
|
|
||||||
if len(got) == 0 {
|
|
||||||
got = got6
|
|
||||||
}
|
|
||||||
case internet.DomainStrategy_USE_IP64, internet.DomainStrategy_FORCE_IP64:
|
|
||||||
got = got6
|
|
||||||
if len(got) == 0 {
|
|
||||||
got = got4
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(got) == 0 {
|
|
||||||
return nil, dns.ErrEmptyResponse
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
var got []net.IP
|
||||||
|
switch strategy {
|
||||||
|
case DeviceConfig_FORCE_IP:
|
||||||
|
got = ips
|
||||||
|
return ips[dice.Roll(len(ips))], nil
|
||||||
|
case DeviceConfig_FORCE_IP4:
|
||||||
|
got = got4
|
||||||
|
case DeviceConfig_FORCE_IP6:
|
||||||
|
got = got6
|
||||||
|
case DeviceConfig_FORCE_IP46:
|
||||||
|
got = got4
|
||||||
|
if len(got) == 0 {
|
||||||
|
got = got6
|
||||||
|
}
|
||||||
|
case DeviceConfig_FORCE_IP64:
|
||||||
|
got = got6
|
||||||
|
if len(got) == 0 {
|
||||||
|
got = got4
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
panic(strategy)
|
||||||
|
}
|
||||||
|
if len(got) == 0 {
|
||||||
|
return nil, dns.ErrEmptyResponse
|
||||||
|
}
|
||||||
|
entry := entry{
|
||||||
|
got: got,
|
||||||
|
time: time.Now().Add(time.Duration(ttl) * time.Second),
|
||||||
|
}
|
||||||
|
h.cacheMu.Lock()
|
||||||
|
h.cache[host] = entry
|
||||||
|
h.cacheMu.Unlock()
|
||||||
return got[dice.Roll(len(got))], nil
|
return got[dice.Roll(len(got))], nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type UDPConnClient struct {
|
type udpConnClient struct {
|
||||||
net.PacketConn
|
net.PacketConn
|
||||||
Dest *net.UDPAddr
|
resolveFunc func(host string) (net.IP, error)
|
||||||
|
dest *net.UDPAddr
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPConnClient) ReadMultiBuffer() (buf.MultiBuffer, error) {
|
func (c *udpConnClient) ReadMultiBuffer() (buf.MultiBuffer, error) {
|
||||||
b := buf.New()
|
b := buf.New()
|
||||||
b.Resize(0, buf.Size)
|
b.Resize(0, buf.Size)
|
||||||
n, addr, err := c.PacketConn.ReadFrom(b.Bytes())
|
n, addr, err := c.PacketConn.ReadFrom(b.Bytes())
|
||||||
@@ -402,13 +473,20 @@ func (c *UDPConnClient) ReadMultiBuffer() (buf.MultiBuffer, error) {
|
|||||||
return buf.MultiBuffer{b}, nil
|
return buf.MultiBuffer{b}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPConnClient) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
func (c *udpConnClient) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
||||||
for i, b := range mb {
|
for i, b := range mb {
|
||||||
dst := c.Dest
|
dst := c.dest
|
||||||
if b.UDP != nil {
|
if b.UDP != nil {
|
||||||
if b.UDP.Address.Family().IsDomain() {
|
if b.UDP.Address.Family().IsDomain() {
|
||||||
if b.UDP.Port != net.Port(dst.Port) {
|
ip, err := c.resolveFunc(b.UDP.Address.String())
|
||||||
dst = &net.UDPAddr{IP: dst.IP, Port: int(b.UDP.Port)}
|
if err != nil {
|
||||||
|
errors.LogErrorInner(context.Background(), err, "drop packet to ", b.UDP, " with size ", len(b.Bytes()))
|
||||||
|
b.Release()
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
dst = &net.UDPAddr{
|
||||||
|
IP: ip,
|
||||||
|
Port: int(b.UDP.Port),
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
dst = b.UDP.RawNetAddr().(*net.UDPAddr)
|
dst = b.UDP.RawNetAddr().(*net.UDPAddr)
|
||||||
@@ -445,66 +523,3 @@ func (c *PacketCounterConnection) WriteTo(p []byte, addr net.Addr) (n int, err e
|
|||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
type entry struct {
|
|
||||||
saddr []string
|
|
||||||
deadline time.Time
|
|
||||||
}
|
|
||||||
|
|
||||||
type cache struct {
|
|
||||||
running bool
|
|
||||||
m map[string]entry
|
|
||||||
mu sync.Mutex
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *cache) run() {
|
|
||||||
if c.running {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
c.running = true
|
|
||||||
if c.m == nil {
|
|
||||||
c.m = make(map[string]entry)
|
|
||||||
}
|
|
||||||
go c.gc()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *cache) gc() {
|
|
||||||
ticker := time.NewTicker(time.Minute)
|
|
||||||
defer ticker.Stop()
|
|
||||||
for now := range ticker.C {
|
|
||||||
c.mu.Lock()
|
|
||||||
for key, entry := range c.m {
|
|
||||||
if now.After(entry.deadline) {
|
|
||||||
delete(c.m, key)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(c.m) == 0 {
|
|
||||||
c.running = false
|
|
||||||
c.mu.Unlock()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
c.mu.Unlock()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *cache) LookupHost(host string) []string {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
c.run()
|
|
||||||
if entry, ok := c.m[host]; ok {
|
|
||||||
if time.Now().Before(entry.deadline) {
|
|
||||||
return entry.saddr
|
|
||||||
}
|
|
||||||
delete(c.m, host)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *cache) Cache(host string, saddr []string, ttl uint32) {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
c.m[host] = entry{
|
|
||||||
saddr: saddr,
|
|
||||||
deadline: time.Now().Add(time.Second * time.Duration(ttl)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
+102
-26
@@ -22,6 +22,61 @@ const (
|
|||||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type DeviceConfig_DomainStrategy int32
|
||||||
|
|
||||||
|
const (
|
||||||
|
DeviceConfig_FORCE_IP DeviceConfig_DomainStrategy = 0
|
||||||
|
DeviceConfig_FORCE_IP4 DeviceConfig_DomainStrategy = 1
|
||||||
|
DeviceConfig_FORCE_IP6 DeviceConfig_DomainStrategy = 2
|
||||||
|
DeviceConfig_FORCE_IP46 DeviceConfig_DomainStrategy = 3
|
||||||
|
DeviceConfig_FORCE_IP64 DeviceConfig_DomainStrategy = 4
|
||||||
|
)
|
||||||
|
|
||||||
|
// Enum value maps for DeviceConfig_DomainStrategy.
|
||||||
|
var (
|
||||||
|
DeviceConfig_DomainStrategy_name = map[int32]string{
|
||||||
|
0: "FORCE_IP",
|
||||||
|
1: "FORCE_IP4",
|
||||||
|
2: "FORCE_IP6",
|
||||||
|
3: "FORCE_IP46",
|
||||||
|
4: "FORCE_IP64",
|
||||||
|
}
|
||||||
|
DeviceConfig_DomainStrategy_value = map[string]int32{
|
||||||
|
"FORCE_IP": 0,
|
||||||
|
"FORCE_IP4": 1,
|
||||||
|
"FORCE_IP6": 2,
|
||||||
|
"FORCE_IP46": 3,
|
||||||
|
"FORCE_IP64": 4,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func (x DeviceConfig_DomainStrategy) Enum() *DeviceConfig_DomainStrategy {
|
||||||
|
p := new(DeviceConfig_DomainStrategy)
|
||||||
|
*p = x
|
||||||
|
return p
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x DeviceConfig_DomainStrategy) String() string {
|
||||||
|
return protoimpl.X.EnumStringOf(x.Descriptor(), protoreflect.EnumNumber(x))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (DeviceConfig_DomainStrategy) Descriptor() protoreflect.EnumDescriptor {
|
||||||
|
return file_proxy_wireguard_config_proto_enumTypes[0].Descriptor()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (DeviceConfig_DomainStrategy) Type() protoreflect.EnumType {
|
||||||
|
return &file_proxy_wireguard_config_proto_enumTypes[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x DeviceConfig_DomainStrategy) Number() protoreflect.EnumNumber {
|
||||||
|
return protoreflect.EnumNumber(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use DeviceConfig_DomainStrategy.Descriptor instead.
|
||||||
|
func (DeviceConfig_DomainStrategy) EnumDescriptor() ([]byte, []int) {
|
||||||
|
return file_proxy_wireguard_config_proto_rawDescGZIP(), []int{1, 0}
|
||||||
|
}
|
||||||
|
|
||||||
type PeerConfig struct {
|
type PeerConfig struct {
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
PublicKey string `protobuf:"bytes,1,opt,name=public_key,json=publicKey,proto3" json:"public_key,omitempty"`
|
PublicKey string `protobuf:"bytes,1,opt,name=public_key,json=publicKey,proto3" json:"public_key,omitempty"`
|
||||||
@@ -99,18 +154,19 @@ func (x *PeerConfig) GetAllowedIps() []string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type DeviceConfig struct {
|
type DeviceConfig struct {
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
SecretKey string `protobuf:"bytes,1,opt,name=secret_key,json=secretKey,proto3" json:"secret_key,omitempty"`
|
SecretKey string `protobuf:"bytes,1,opt,name=secret_key,json=secretKey,proto3" json:"secret_key,omitempty"`
|
||||||
Endpoint []string `protobuf:"bytes,2,rep,name=endpoint,proto3" json:"endpoint,omitempty"`
|
Endpoint []string `protobuf:"bytes,2,rep,name=endpoint,proto3" json:"endpoint,omitempty"`
|
||||||
Peers []*PeerConfig `protobuf:"bytes,3,rep,name=peers,proto3" json:"peers,omitempty"`
|
Peers []*PeerConfig `protobuf:"bytes,3,rep,name=peers,proto3" json:"peers,omitempty"`
|
||||||
Users []*protocol.User `protobuf:"bytes,5,rep,name=users,proto3" json:"users,omitempty"`
|
Users []*protocol.User `protobuf:"bytes,5,rep,name=users,proto3" json:"users,omitempty"`
|
||||||
Mtu int32 `protobuf:"varint,4,opt,name=mtu,proto3" json:"mtu,omitempty"`
|
Mtu int32 `protobuf:"varint,4,opt,name=mtu,proto3" json:"mtu,omitempty"`
|
||||||
Reserved []byte `protobuf:"bytes,6,opt,name=reserved,proto3" json:"reserved,omitempty"`
|
Reserved []byte `protobuf:"bytes,6,opt,name=reserved,proto3" json:"reserved,omitempty"`
|
||||||
IsClient bool `protobuf:"varint,8,opt,name=is_client,json=isClient,proto3" json:"is_client,omitempty"`
|
DomainStrategy DeviceConfig_DomainStrategy `protobuf:"varint,7,opt,name=domain_strategy,json=domainStrategy,proto3,enum=xray.proxy.wireguard.DeviceConfig_DomainStrategy" json:"domain_strategy,omitempty"`
|
||||||
NoKernelTun bool `protobuf:"varint,9,opt,name=no_kernel_tun,json=noKernelTun,proto3" json:"no_kernel_tun,omitempty"`
|
IsClient bool `protobuf:"varint,8,opt,name=is_client,json=isClient,proto3" json:"is_client,omitempty"`
|
||||||
DNS []string `protobuf:"bytes,10,rep,name=DNS,proto3" json:"DNS,omitempty"`
|
NoKernelTun bool `protobuf:"varint,9,opt,name=no_kernel_tun,json=noKernelTun,proto3" json:"no_kernel_tun,omitempty"`
|
||||||
unknownFields protoimpl.UnknownFields
|
DNS []string `protobuf:"bytes,10,rep,name=DNS,proto3" json:"DNS,omitempty"`
|
||||||
sizeCache protoimpl.SizeCache
|
unknownFields protoimpl.UnknownFields
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *DeviceConfig) Reset() {
|
func (x *DeviceConfig) Reset() {
|
||||||
@@ -185,6 +241,13 @@ func (x *DeviceConfig) GetReserved() []byte {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (x *DeviceConfig) GetDomainStrategy() DeviceConfig_DomainStrategy {
|
||||||
|
if x != nil {
|
||||||
|
return x.DomainStrategy
|
||||||
|
}
|
||||||
|
return DeviceConfig_FORCE_IP
|
||||||
|
}
|
||||||
|
|
||||||
func (x *DeviceConfig) GetIsClient() bool {
|
func (x *DeviceConfig) GetIsClient() bool {
|
||||||
if x != nil {
|
if x != nil {
|
||||||
return x.IsClient
|
return x.IsClient
|
||||||
@@ -220,7 +283,7 @@ const file_proxy_wireguard_config_proto_rawDesc = "" +
|
|||||||
"\n" +
|
"\n" +
|
||||||
"keep_alive\x18\x04 \x01(\tR\tkeepAlive\x12\x1f\n" +
|
"keep_alive\x18\x04 \x01(\tR\tkeepAlive\x12\x1f\n" +
|
||||||
"\vallowed_ips\x18\x05 \x03(\tR\n" +
|
"\vallowed_ips\x18\x05 \x03(\tR\n" +
|
||||||
"allowedIps\"\xb4\x02\n" +
|
"allowedIps\"\xee\x03\n" +
|
||||||
"\fDeviceConfig\x12\x1d\n" +
|
"\fDeviceConfig\x12\x1d\n" +
|
||||||
"\n" +
|
"\n" +
|
||||||
"secret_key\x18\x01 \x01(\tR\tsecretKey\x12\x1a\n" +
|
"secret_key\x18\x01 \x01(\tR\tsecretKey\x12\x1a\n" +
|
||||||
@@ -228,11 +291,20 @@ const file_proxy_wireguard_config_proto_rawDesc = "" +
|
|||||||
"\x05peers\x18\x03 \x03(\v2 .xray.proxy.wireguard.PeerConfigR\x05peers\x120\n" +
|
"\x05peers\x18\x03 \x03(\v2 .xray.proxy.wireguard.PeerConfigR\x05peers\x120\n" +
|
||||||
"\x05users\x18\x05 \x03(\v2\x1a.xray.common.protocol.UserR\x05users\x12\x10\n" +
|
"\x05users\x18\x05 \x03(\v2\x1a.xray.common.protocol.UserR\x05users\x12\x10\n" +
|
||||||
"\x03mtu\x18\x04 \x01(\x05R\x03mtu\x12\x1a\n" +
|
"\x03mtu\x18\x04 \x01(\x05R\x03mtu\x12\x1a\n" +
|
||||||
"\breserved\x18\x06 \x01(\fR\breserved\x12\x1b\n" +
|
"\breserved\x18\x06 \x01(\fR\breserved\x12Z\n" +
|
||||||
|
"\x0fdomain_strategy\x18\a \x01(\x0e21.xray.proxy.wireguard.DeviceConfig.DomainStrategyR\x0edomainStrategy\x12\x1b\n" +
|
||||||
"\tis_client\x18\b \x01(\bR\bisClient\x12\"\n" +
|
"\tis_client\x18\b \x01(\bR\bisClient\x12\"\n" +
|
||||||
"\rno_kernel_tun\x18\t \x01(\bR\vnoKernelTun\x12\x10\n" +
|
"\rno_kernel_tun\x18\t \x01(\bR\vnoKernelTun\x12\x10\n" +
|
||||||
"\x03DNS\x18\n" +
|
"\x03DNS\x18\n" +
|
||||||
" \x03(\tR\x03DNSB^\n" +
|
" \x03(\tR\x03DNS\"\\\n" +
|
||||||
|
"\x0eDomainStrategy\x12\f\n" +
|
||||||
|
"\bFORCE_IP\x10\x00\x12\r\n" +
|
||||||
|
"\tFORCE_IP4\x10\x01\x12\r\n" +
|
||||||
|
"\tFORCE_IP6\x10\x02\x12\x0e\n" +
|
||||||
|
"\n" +
|
||||||
|
"FORCE_IP46\x10\x03\x12\x0e\n" +
|
||||||
|
"\n" +
|
||||||
|
"FORCE_IP64\x10\x04B^\n" +
|
||||||
"\x18com.xray.proxy.wireguardP\x01Z)github.com/xtls/xray-core/proxy/wireguard\xaa\x02\x14Xray.Proxy.WireGuardb\x06proto3"
|
"\x18com.xray.proxy.wireguardP\x01Z)github.com/xtls/xray-core/proxy/wireguard\xaa\x02\x14Xray.Proxy.WireGuardb\x06proto3"
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -247,20 +319,23 @@ func file_proxy_wireguard_config_proto_rawDescGZIP() []byte {
|
|||||||
return file_proxy_wireguard_config_proto_rawDescData
|
return file_proxy_wireguard_config_proto_rawDescData
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var file_proxy_wireguard_config_proto_enumTypes = make([]protoimpl.EnumInfo, 1)
|
||||||
var file_proxy_wireguard_config_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
|
var file_proxy_wireguard_config_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
|
||||||
var file_proxy_wireguard_config_proto_goTypes = []any{
|
var file_proxy_wireguard_config_proto_goTypes = []any{
|
||||||
(*PeerConfig)(nil), // 0: xray.proxy.wireguard.PeerConfig
|
(DeviceConfig_DomainStrategy)(0), // 0: xray.proxy.wireguard.DeviceConfig.DomainStrategy
|
||||||
(*DeviceConfig)(nil), // 1: xray.proxy.wireguard.DeviceConfig
|
(*PeerConfig)(nil), // 1: xray.proxy.wireguard.PeerConfig
|
||||||
(*protocol.User)(nil), // 2: xray.common.protocol.User
|
(*DeviceConfig)(nil), // 2: xray.proxy.wireguard.DeviceConfig
|
||||||
|
(*protocol.User)(nil), // 3: xray.common.protocol.User
|
||||||
}
|
}
|
||||||
var file_proxy_wireguard_config_proto_depIdxs = []int32{
|
var file_proxy_wireguard_config_proto_depIdxs = []int32{
|
||||||
0, // 0: xray.proxy.wireguard.DeviceConfig.peers:type_name -> xray.proxy.wireguard.PeerConfig
|
1, // 0: xray.proxy.wireguard.DeviceConfig.peers:type_name -> xray.proxy.wireguard.PeerConfig
|
||||||
2, // 1: xray.proxy.wireguard.DeviceConfig.users:type_name -> xray.common.protocol.User
|
3, // 1: xray.proxy.wireguard.DeviceConfig.users:type_name -> xray.common.protocol.User
|
||||||
2, // [2:2] is the sub-list for method output_type
|
0, // 2: xray.proxy.wireguard.DeviceConfig.domain_strategy:type_name -> xray.proxy.wireguard.DeviceConfig.DomainStrategy
|
||||||
2, // [2:2] is the sub-list for method input_type
|
3, // [3:3] is the sub-list for method output_type
|
||||||
2, // [2:2] is the sub-list for extension type_name
|
3, // [3:3] is the sub-list for method input_type
|
||||||
2, // [2:2] is the sub-list for extension extendee
|
3, // [3:3] is the sub-list for extension type_name
|
||||||
0, // [0:2] is the sub-list for field type_name
|
3, // [3:3] is the sub-list for extension extendee
|
||||||
|
0, // [0:3] is the sub-list for field type_name
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() { file_proxy_wireguard_config_proto_init() }
|
func init() { file_proxy_wireguard_config_proto_init() }
|
||||||
@@ -273,13 +348,14 @@ func file_proxy_wireguard_config_proto_init() {
|
|||||||
File: protoimpl.DescBuilder{
|
File: protoimpl.DescBuilder{
|
||||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
||||||
RawDescriptor: unsafe.Slice(unsafe.StringData(file_proxy_wireguard_config_proto_rawDesc), len(file_proxy_wireguard_config_proto_rawDesc)),
|
RawDescriptor: unsafe.Slice(unsafe.StringData(file_proxy_wireguard_config_proto_rawDesc), len(file_proxy_wireguard_config_proto_rawDesc)),
|
||||||
NumEnums: 0,
|
NumEnums: 1,
|
||||||
NumMessages: 2,
|
NumMessages: 2,
|
||||||
NumExtensions: 0,
|
NumExtensions: 0,
|
||||||
NumServices: 0,
|
NumServices: 0,
|
||||||
},
|
},
|
||||||
GoTypes: file_proxy_wireguard_config_proto_goTypes,
|
GoTypes: file_proxy_wireguard_config_proto_goTypes,
|
||||||
DependencyIndexes: file_proxy_wireguard_config_proto_depIdxs,
|
DependencyIndexes: file_proxy_wireguard_config_proto_depIdxs,
|
||||||
|
EnumInfos: file_proxy_wireguard_config_proto_enumTypes,
|
||||||
MessageInfos: file_proxy_wireguard_config_proto_msgTypes,
|
MessageInfos: file_proxy_wireguard_config_proto_msgTypes,
|
||||||
}.Build()
|
}.Build()
|
||||||
File_proxy_wireguard_config_proto = out.File
|
File_proxy_wireguard_config_proto = out.File
|
||||||
|
|||||||
@@ -17,6 +17,13 @@ message PeerConfig {
|
|||||||
}
|
}
|
||||||
|
|
||||||
message DeviceConfig {
|
message DeviceConfig {
|
||||||
|
enum DomainStrategy {
|
||||||
|
FORCE_IP = 0;
|
||||||
|
FORCE_IP4 = 1;
|
||||||
|
FORCE_IP6 = 2;
|
||||||
|
FORCE_IP46 = 3;
|
||||||
|
FORCE_IP64 = 4;
|
||||||
|
}
|
||||||
string secret_key = 1;
|
string secret_key = 1;
|
||||||
repeated string endpoint = 2;
|
repeated string endpoint = 2;
|
||||||
repeated PeerConfig peers = 3;
|
repeated PeerConfig peers = 3;
|
||||||
@@ -24,6 +31,7 @@ message DeviceConfig {
|
|||||||
int32 mtu = 4;
|
int32 mtu = 4;
|
||||||
|
|
||||||
bytes reserved = 6;
|
bytes reserved = 6;
|
||||||
|
DomainStrategy domain_strategy = 7;
|
||||||
bool is_client = 8;
|
bool is_client = 8;
|
||||||
bool no_kernel_tun = 9;
|
bool no_kernel_tun = 9;
|
||||||
repeated string DNS = 10;
|
repeated string DNS = 10;
|
||||||
|
|||||||
+14
-159
@@ -15,8 +15,6 @@ import (
|
|||||||
"net"
|
"net"
|
||||||
"net/netip"
|
"net/netip"
|
||||||
"os"
|
"os"
|
||||||
"regexp"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
@@ -44,7 +42,6 @@ type netTun struct {
|
|||||||
events chan tun.Event
|
events chan tun.Event
|
||||||
notifyHandle *channel.NotificationHandle
|
notifyHandle *channel.NotificationHandle
|
||||||
incomingPacket chan *buffer.View
|
incomingPacket chan *buffer.View
|
||||||
closed chan struct{}
|
|
||||||
mtu int
|
mtu int
|
||||||
dnsServers []netip.Addr
|
dnsServers []netip.Addr
|
||||||
hasV4, hasV6 bool
|
hasV4, hasV6 bool
|
||||||
@@ -61,7 +58,6 @@ func CreateNetTUN(localAddresses, dnsServers []netip.Addr, mtu int, handleLocal
|
|||||||
stack: stack.New(opts),
|
stack: stack.New(opts),
|
||||||
events: make(chan tun.Event, 10),
|
events: make(chan tun.Event, 10),
|
||||||
incomingPacket: make(chan *buffer.View),
|
incomingPacket: make(chan *buffer.View),
|
||||||
closed: make(chan struct{}),
|
|
||||||
dnsServers: dnsServers,
|
dnsServers: dnsServers,
|
||||||
mtu: mtu,
|
mtu: mtu,
|
||||||
}
|
}
|
||||||
@@ -128,15 +124,12 @@ func (tun *netTun) Events() <-chan tun.Event {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (tun *netTun) Read(buf [][]byte, sizes []int, offset int) (int, error) {
|
func (tun *netTun) Read(buf [][]byte, sizes []int, offset int) (int, error) {
|
||||||
var view *buffer.View
|
view, ok := <-tun.incomingPacket
|
||||||
select {
|
if !ok {
|
||||||
case view = <-tun.incomingPacket:
|
|
||||||
case <-tun.closed:
|
|
||||||
return 0, os.ErrClosed
|
return 0, os.ErrClosed
|
||||||
}
|
}
|
||||||
|
|
||||||
n, err := view.Read(buf[0][offset:])
|
n, err := view.Read(buf[0][offset:])
|
||||||
view.Release()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
@@ -173,11 +166,7 @@ func (tun *netTun) WriteNotify() {
|
|||||||
view := pkt.ToView()
|
view := pkt.ToView()
|
||||||
pkt.DecRef()
|
pkt.DecRef()
|
||||||
|
|
||||||
select {
|
tun.incomingPacket <- view
|
||||||
case tun.incomingPacket <- view:
|
|
||||||
case <-tun.closed:
|
|
||||||
view.Release()
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tun *netTun) Close() error {
|
func (tun *netTun) Close() error {
|
||||||
@@ -190,9 +179,8 @@ func (tun *netTun) Close() error {
|
|||||||
close(tun.events)
|
close(tun.events)
|
||||||
}
|
}
|
||||||
|
|
||||||
// we don't close incomingPacket, because WriteNotify may be mid-send on it (DNS lookup) and would panic.
|
if tun.incomingPacket != nil {
|
||||||
if tun.closed != nil {
|
close(tun.incomingPacket)
|
||||||
close(tun.closed)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
@@ -231,7 +219,6 @@ type Net struct {
|
|||||||
DialUDPAddrPort func(laddr, raddr netip.AddrPort) (net.Conn, error)
|
DialUDPAddrPort func(laddr, raddr netip.AddrPort) (net.Conn, error)
|
||||||
dnsServers []netip.Addr
|
dnsServers []netip.Addr
|
||||||
hasV4, hasV6 bool
|
hasV4, hasV6 bool
|
||||||
cache cache
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func convertToFullAddr(endpoint netip.AddrPort) (tcpip.FullAddress, tcpip.NetworkProtocolNumber) {
|
func convertToFullAddr(endpoint netip.AddrPort) (tcpip.FullAddress, tcpip.NetworkProtocolNumber) {
|
||||||
@@ -259,12 +246,9 @@ var (
|
|||||||
errServerTemporarilyMisbehaving = errors.New("server misbehaving")
|
errServerTemporarilyMisbehaving = errors.New("server misbehaving")
|
||||||
errCanceled = errors.New("operation was canceled")
|
errCanceled = errors.New("operation was canceled")
|
||||||
errTimeout = errors.New("i/o timeout")
|
errTimeout = errors.New("i/o timeout")
|
||||||
errNumericPort = errors.New("port must be numeric")
|
|
||||||
errNoSuitableAddress = errors.New("no suitable address found")
|
|
||||||
errMissingAddress = errors.New("missing address")
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func (net *Net) LookupHost(host string) (addrs []string, err error) {
|
func (net *Net) LookupHost(host string) (addrs []net.IP, ttl uint32, err error) {
|
||||||
return net.LookupContextHost(context.Background(), host)
|
return net.LookupContextHost(context.Background(), host)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -583,12 +567,9 @@ func (tnet *Net) tryOneName(ctx context.Context, name string, qtype dnsmessage.T
|
|||||||
return dnsmessage.Parser{}, "", lastErr
|
return dnsmessage.Parser{}, "", lastErr
|
||||||
}
|
}
|
||||||
|
|
||||||
func (tnet *Net) LookupContextHost(ctx context.Context, host string) ([]string, error) {
|
func (tnet *Net) LookupContextHost(ctx context.Context, host string) ([]net.IP, uint32, error) {
|
||||||
if saddr := tnet.cache.LookupHost(host); saddr != nil {
|
|
||||||
return saddr, nil
|
|
||||||
}
|
|
||||||
if host == "" || (!tnet.hasV6 && !tnet.hasV4) {
|
if host == "" || (!tnet.hasV6 && !tnet.hasV4) {
|
||||||
return nil, &net.DNSError{Err: errNoSuchHost.Error(), Name: host, IsNotFound: true}
|
return nil, 0, &net.DNSError{Err: errNoSuchHost.Error(), Name: host, IsNotFound: true}
|
||||||
}
|
}
|
||||||
zlen := len(host)
|
zlen := len(host)
|
||||||
if strings.IndexByte(host, ':') != -1 {
|
if strings.IndexByte(host, ':') != -1 {
|
||||||
@@ -597,11 +578,11 @@ func (tnet *Net) LookupContextHost(ctx context.Context, host string) ([]string,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
if ip, err := netip.ParseAddr(host[:zlen]); err == nil {
|
if ip, err := netip.ParseAddr(host[:zlen]); err == nil {
|
||||||
return []string{ip.String()}, nil
|
return []net.IP{ip.AsSlice()}, 0, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if !isDomainName(host) {
|
if !isDomainName(host) {
|
||||||
return nil, &net.DNSError{Err: errNoSuchHost.Error(), Name: host, IsNotFound: true}
|
return nil, 0, &net.DNSError{Err: errNoSuchHost.Error(), Name: host, IsNotFound: true}
|
||||||
}
|
}
|
||||||
type result struct {
|
type result struct {
|
||||||
p dnsmessage.Parser
|
p dnsmessage.Parser
|
||||||
@@ -702,137 +683,11 @@ func (tnet *Net) LookupContextHost(ctx context.Context, host string) ([]string,
|
|||||||
}
|
}
|
||||||
|
|
||||||
if len(addrs) == 0 && lastErr != nil {
|
if len(addrs) == 0 && lastErr != nil {
|
||||||
return nil, lastErr
|
return nil, 0, lastErr
|
||||||
}
|
}
|
||||||
saddrs := make([]string, 0, len(addrs))
|
ips := make([]net.IP, 0, len(addrs))
|
||||||
for _, ip := range addrs {
|
for _, ip := range addrs {
|
||||||
saddrs = append(saddrs, ip.String())
|
ips = append(ips, ip.AsSlice())
|
||||||
}
|
}
|
||||||
tnet.cache.Cache(host, saddrs, ttl)
|
return ips, ttl, nil
|
||||||
return saddrs, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func partialDeadline(now, deadline time.Time, addrsRemaining int) (time.Time, error) {
|
|
||||||
if deadline.IsZero() {
|
|
||||||
return deadline, nil
|
|
||||||
}
|
|
||||||
timeRemaining := deadline.Sub(now)
|
|
||||||
if timeRemaining <= 0 {
|
|
||||||
return time.Time{}, errTimeout
|
|
||||||
}
|
|
||||||
timeout := timeRemaining / time.Duration(addrsRemaining)
|
|
||||||
const saneMinimum = 2 * time.Second
|
|
||||||
if timeout < saneMinimum {
|
|
||||||
if timeRemaining < saneMinimum {
|
|
||||||
timeout = timeRemaining
|
|
||||||
} else {
|
|
||||||
timeout = saneMinimum
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return now.Add(timeout), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var protoSplitter = regexp.MustCompile(`^(tcp|udp|ping)(4|6)?$`)
|
|
||||||
|
|
||||||
func (tnet *Net) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
|
|
||||||
if ctx == nil {
|
|
||||||
panic("nil context")
|
|
||||||
}
|
|
||||||
var acceptV4, acceptV6 bool
|
|
||||||
matches := protoSplitter.FindStringSubmatch(network)
|
|
||||||
if matches == nil {
|
|
||||||
return nil, &net.OpError{Op: "dial", Err: net.UnknownNetworkError(network)}
|
|
||||||
} else if len(matches[2]) == 0 {
|
|
||||||
acceptV4 = true
|
|
||||||
acceptV6 = true
|
|
||||||
} else {
|
|
||||||
acceptV4 = matches[2][0] == '4'
|
|
||||||
acceptV6 = !acceptV4
|
|
||||||
}
|
|
||||||
var host string
|
|
||||||
var port int
|
|
||||||
if matches[1] == "ping" {
|
|
||||||
host = address
|
|
||||||
} else {
|
|
||||||
var sport string
|
|
||||||
var err error
|
|
||||||
host, sport, err = net.SplitHostPort(address)
|
|
||||||
if err != nil {
|
|
||||||
return nil, &net.OpError{Op: "dial", Err: err}
|
|
||||||
}
|
|
||||||
port, err = strconv.Atoi(sport)
|
|
||||||
if err != nil || port < 0 || port > 65535 {
|
|
||||||
return nil, &net.OpError{Op: "dial", Err: errNumericPort}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
allAddr, err := tnet.LookupContextHost(ctx, host)
|
|
||||||
if err != nil {
|
|
||||||
return nil, &net.OpError{Op: "dial", Err: err}
|
|
||||||
}
|
|
||||||
var addrs []netip.AddrPort
|
|
||||||
for _, addr := range allAddr {
|
|
||||||
ip, err := netip.ParseAddr(addr)
|
|
||||||
if err == nil && ((ip.Is4() && acceptV4) || (ip.Is6() && acceptV6)) {
|
|
||||||
addrs = append(addrs, netip.AddrPortFrom(ip, uint16(port)))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(addrs) == 0 && len(allAddr) != 0 {
|
|
||||||
return nil, &net.OpError{Op: "dial", Err: errNoSuitableAddress}
|
|
||||||
}
|
|
||||||
|
|
||||||
var firstErr error
|
|
||||||
for i, addr := range addrs {
|
|
||||||
select {
|
|
||||||
case <-ctx.Done():
|
|
||||||
err := ctx.Err()
|
|
||||||
if err == context.Canceled {
|
|
||||||
err = errCanceled
|
|
||||||
} else if err == context.DeadlineExceeded {
|
|
||||||
err = errTimeout
|
|
||||||
}
|
|
||||||
return nil, &net.OpError{Op: "dial", Err: err}
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
|
|
||||||
dialCtx := ctx
|
|
||||||
if deadline, hasDeadline := ctx.Deadline(); hasDeadline {
|
|
||||||
partialDeadline, err := partialDeadline(time.Now(), deadline, len(addrs)-i)
|
|
||||||
if err != nil {
|
|
||||||
if firstErr == nil {
|
|
||||||
firstErr = &net.OpError{Op: "dial", Err: err}
|
|
||||||
}
|
|
||||||
break
|
|
||||||
}
|
|
||||||
if partialDeadline.Before(deadline) {
|
|
||||||
var cancel context.CancelFunc
|
|
||||||
dialCtx, cancel = context.WithDeadline(ctx, partialDeadline)
|
|
||||||
defer cancel()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var c net.Conn
|
|
||||||
switch matches[1] {
|
|
||||||
case "tcp":
|
|
||||||
c, err = tnet.DialContextTCPAddrPort(dialCtx, addr)
|
|
||||||
case "udp":
|
|
||||||
c, err = tnet.DialUDPAddrPort(netip.AddrPort{}, addr)
|
|
||||||
case "ping":
|
|
||||||
err = errors.New("not support")
|
|
||||||
// c, err = tnet.DialPingAddr(netip.Addr{}, addr.Addr())
|
|
||||||
}
|
|
||||||
if err == nil {
|
|
||||||
return c, nil
|
|
||||||
}
|
|
||||||
if firstErr == nil {
|
|
||||||
firstErr = err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if firstErr == nil {
|
|
||||||
firstErr = &net.OpError{Op: "dial", Err: errMissingAddress}
|
|
||||||
}
|
|
||||||
return nil, firstErr
|
|
||||||
}
|
|
||||||
|
|
||||||
func (tnet *Net) Dial(network, address string) (net.Conn, error) {
|
|
||||||
return tnet.DialContext(context.Background(), network, address)
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -258,16 +258,18 @@ func (s *Server) Start() error {
|
|||||||
return errors.New("address is domain")
|
return errors.New("address is domain")
|
||||||
}
|
}
|
||||||
listenFunc := func() (net.PacketConn, error) {
|
listenFunc := func() (net.PacketConn, error) {
|
||||||
var pktConn net.PacketConn
|
pktConn, err := internet.ListenSystemPacket(context.Background(), &net.UDPAddr{IP: s.src.Address.IP(), Port: int(s.src.Port)}, s.streamSettings.SocketSettings)
|
||||||
var err error
|
|
||||||
if s.streamSettings.FinalMask != nil {
|
|
||||||
pktConn, err = s.streamSettings.FinalMask.ListenPacket(context.Background(), &net.UDPAddr{IP: s.src.Address.IP(), Port: int(s.src.Port)})
|
|
||||||
} else {
|
|
||||||
pktConn, err = internet.ListenSystemPacket(context.Background(), &net.UDPAddr{IP: s.src.Address.IP(), Port: int(s.src.Port)}, s.streamSettings.SocketSettings)
|
|
||||||
}
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
if s.streamSettings.UdpmaskManager != nil {
|
||||||
|
newConn, err := s.streamSettings.UdpmaskManager.WrapPacketConnServer(pktConn)
|
||||||
|
if err != nil {
|
||||||
|
pktConn.Close()
|
||||||
|
return nil, errors.New("mask err").Base(err)
|
||||||
|
}
|
||||||
|
pktConn = newConn
|
||||||
|
}
|
||||||
if s.uplinkCounter != nil || s.downlinkCounter != nil {
|
if s.uplinkCounter != nil || s.downlinkCounter != nil {
|
||||||
pktConn = &PacketCounterConnection{
|
pktConn = &PacketCounterConnection{
|
||||||
PacketConn: pktConn,
|
PacketConn: pktConn,
|
||||||
|
|||||||
@@ -82,14 +82,9 @@ func TestResolveIP(t *testing.T) {
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
Tag: "direct",
|
Tag: "direct",
|
||||||
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
|
ProxySettings: serial.ToTypedMessage(&freedom.Config{
|
||||||
StreamSettings: &internet.StreamConfig{
|
DomainStrategy: internet.DomainStrategy_USE_IP,
|
||||||
SocketSettings: &internet.SocketConfig{
|
|
||||||
DomainStrategy: internet.DomainStrategy_USE_IP,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}),
|
}),
|
||||||
ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -92,7 +92,7 @@ func TestPassiveConnection(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDialerProxy(t *testing.T) {
|
func TestProxy(t *testing.T) {
|
||||||
tcpServer := tcp.Server{
|
tcpServer := tcp.Server{
|
||||||
MsgProcessor: xor,
|
MsgProcessor: xor,
|
||||||
}
|
}
|
||||||
@@ -187,10 +187,8 @@ func TestDialerProxy(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}),
|
}),
|
||||||
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
|
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
|
||||||
StreamSettings: &internet.StreamConfig{
|
ProxySettings: &internet.ProxyConfig{
|
||||||
SocketSettings: &internet.SocketConfig{
|
Tag: "proxy",
|
||||||
DialerProxy: "proxy",
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
}),
|
}),
|
||||||
},
|
},
|
||||||
@@ -220,7 +218,7 @@ func TestDialerProxy(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDialerProxyOverKCP(t *testing.T) {
|
func TestProxyOverKCP(t *testing.T) {
|
||||||
tcpServer := tcp.Server{
|
tcpServer := tcp.Server{
|
||||||
MsgProcessor: xor,
|
MsgProcessor: xor,
|
||||||
}
|
}
|
||||||
@@ -323,11 +321,11 @@ func TestDialerProxyOverKCP(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}),
|
}),
|
||||||
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
|
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
|
||||||
|
ProxySettings: &internet.ProxyConfig{
|
||||||
|
Tag: "proxy",
|
||||||
|
},
|
||||||
StreamSettings: &internet.StreamConfig{
|
StreamSettings: &internet.StreamConfig{
|
||||||
ProtocolName: "mkcp",
|
ProtocolName: "mkcp",
|
||||||
SocketSettings: &internet.SocketConfig{
|
|
||||||
DialerProxy: "proxy",
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
}),
|
}),
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -1,275 +0,0 @@
|
|||||||
package scenarios
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
gotls "crypto/tls"
|
|
||||||
"crypto/x509"
|
|
||||||
go_errors "errors"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"net/netip"
|
|
||||||
"sync/atomic"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/apernet/quic-go"
|
|
||||||
"github.com/apernet/quic-go/http3"
|
|
||||||
"golang.org/x/sync/errgroup"
|
|
||||||
"gvisor.dev/gvisor/pkg/tcpip"
|
|
||||||
"gvisor.dev/gvisor/pkg/tcpip/adapters/gonet"
|
|
||||||
"gvisor.dev/gvisor/pkg/tcpip/network/ipv4"
|
|
||||||
"gvisor.dev/gvisor/pkg/tcpip/network/ipv6"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/app/log"
|
|
||||||
"github.com/xtls/xray-core/app/proxyman"
|
|
||||||
"github.com/xtls/xray-core/common"
|
|
||||||
clog "github.com/xtls/xray-core/common/log"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/common/protocol"
|
|
||||||
"github.com/xtls/xray-core/common/protocol/tls/cert"
|
|
||||||
"github.com/xtls/xray-core/common/serial"
|
|
||||||
core "github.com/xtls/xray-core/core"
|
|
||||||
"github.com/xtls/xray-core/proxy/dokodemo"
|
|
||||||
"github.com/xtls/xray-core/proxy/masque"
|
|
||||||
"github.com/xtls/xray-core/proxy/wireguard"
|
|
||||||
"github.com/xtls/xray-core/testing/servers/tcp"
|
|
||||||
"github.com/xtls/xray-core/testing/servers/udp"
|
|
||||||
"github.com/xtls/xray-core/transport/internet"
|
|
||||||
transmasque "github.com/xtls/xray-core/transport/internet/masque"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/masque/connectip"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/tls"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
masqueServerV4 = netip.MustParseAddr("10.13.0.1")
|
|
||||||
masqueServerV6 = netip.MustParseAddr("fd13::1")
|
|
||||||
masqueClientV4 = netip.MustParsePrefix("10.13.0.2/32")
|
|
||||||
masqueClientV6 = netip.MustParsePrefix("fd13::2/128")
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
masqueEchoPort = 7
|
|
||||||
masqueAuthorization = "Basic dTpw"
|
|
||||||
)
|
|
||||||
|
|
||||||
func startMasqueServer(t *testing.T) (net.Port, [32]byte) {
|
|
||||||
dev, _, gstack, err := wireguard.CreateNetTUN([]netip.Addr{masqueServerV4, masqueServerV6}, nil, transmasque.MinPacketSize, false)
|
|
||||||
common.Must(err)
|
|
||||||
t.Cleanup(func() { dev.Close() })
|
|
||||||
|
|
||||||
for _, addr := range []netip.Addr{masqueServerV4, masqueServerV6} {
|
|
||||||
proto := ipv4.ProtocolNumber
|
|
||||||
if addr.Is6() {
|
|
||||||
proto = ipv6.ProtocolNumber
|
|
||||||
}
|
|
||||||
local := tcpip.FullAddress{NIC: 1, Addr: tcpip.AddrFromSlice(addr.AsSlice()), Port: masqueEchoPort}
|
|
||||||
l, err := gonet.ListenTCP(gstack, local, proto)
|
|
||||||
common.Must(err)
|
|
||||||
go func() {
|
|
||||||
for {
|
|
||||||
c, err := l.Accept()
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
go func() {
|
|
||||||
defer c.Close()
|
|
||||||
b := make([]byte, 2048)
|
|
||||||
for {
|
|
||||||
n, err := c.Read(b)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if _, err := c.Write(xor(b[:n])); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
u, err := gonet.DialUDP(gstack, &local, nil, proto)
|
|
||||||
common.Must(err)
|
|
||||||
go func() {
|
|
||||||
b := make([]byte, 2048)
|
|
||||||
for {
|
|
||||||
n, addr, err := u.ReadFrom(b)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
u.WriteTo(xor(b[:n]), addr)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
var current atomic.Pointer[connectip.Conn]
|
|
||||||
go func() {
|
|
||||||
bufs := [][]byte{make([]byte, transmasque.MinPacketSize)}
|
|
||||||
sizes := []int{0}
|
|
||||||
for {
|
|
||||||
if _, err := dev.Read(bufs, sizes, 0); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if conn := current.Load(); conn != nil {
|
|
||||||
if icmp, _ := conn.WritePacket(bufs[0][:sizes[0]]); len(icmp) > 0 {
|
|
||||||
go dev.Write([][]byte{icmp}, 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
handler := func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
if r.URL.Path != transmasque.DefaultPath {
|
|
||||||
w.WriteHeader(http.StatusNotFound)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if r.Header.Get("Authorization") != masqueAuthorization {
|
|
||||||
w.WriteHeader(http.StatusUnauthorized)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
req, err := connectip.ParseProxyRequest(r)
|
|
||||||
if err != nil {
|
|
||||||
var perr *connectip.ProxyRequestParseError
|
|
||||||
if go_errors.As(err, &perr) {
|
|
||||||
w.WriteHeader(perr.HTTPStatus)
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
conn, err := (&connectip.Proxy{}).Proxy(w, req)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer conn.Close()
|
|
||||||
common.Must(conn.AssignAddresses([]netip.Prefix{masqueClientV4, masqueClientV6}))
|
|
||||||
common.Must(conn.AdvertiseRoute([]connectip.IPRoute{
|
|
||||||
{StartIP: netip.IPv4Unspecified(), EndIP: netip.AddrFrom4([4]byte{255, 255, 255, 255})},
|
|
||||||
{StartIP: netip.IPv6Unspecified(), EndIP: netip.AddrFrom16([16]byte{0: 0xff, 1: 0xff, 2: 0xff, 3: 0xff, 4: 0xff, 5: 0xff, 6: 0xff, 7: 0xff, 8: 0xff, 9: 0xff, 10: 0xff, 11: 0xff, 12: 0xff, 13: 0xff, 14: 0xff, 15: 0xff})},
|
|
||||||
}))
|
|
||||||
go func() {
|
|
||||||
for {
|
|
||||||
ar, err := conn.ReceiveAddressRequest(context.Background())
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
assigned := make([]netip.Prefix, len(ar.Prefixes))
|
|
||||||
for i, p := range ar.Prefixes {
|
|
||||||
if p.Addr().Is4() {
|
|
||||||
assigned[i] = masqueClientV4
|
|
||||||
} else {
|
|
||||||
assigned[i] = masqueClientV6
|
|
||||||
}
|
|
||||||
}
|
|
||||||
ar.Respond(assigned, nil)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
current.Store(conn)
|
|
||||||
b := make([]byte, 2048)
|
|
||||||
for {
|
|
||||||
n, err := conn.ReadPacket(b)
|
|
||||||
if err != nil {
|
|
||||||
if go_errors.Is(err, io.ErrShortBuffer) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
dev.Write([][]byte{b[:n]}, 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
certificate, certHash := cert.MustGenerate(nil, cert.CommonName("localhost"))
|
|
||||||
key := common.Must2(x509.ParsePKCS8PrivateKey(certificate.PrivateKey))
|
|
||||||
tlsConfig := &gotls.Config{
|
|
||||||
Certificates: []gotls.Certificate{{Certificate: [][]byte{certificate.Certificate}, PrivateKey: key}},
|
|
||||||
NextProtos: []string{http3.NextProtoH3},
|
|
||||||
}
|
|
||||||
pktConn := common.Must2(net.ListenUDP("udp", &net.UDPAddr{IP: net.LocalHostIP.IP()}))
|
|
||||||
tr := &quic.Transport{Conn: pktConn}
|
|
||||||
ln := common.Must2(tr.ListenEarly(tlsConfig, &quic.Config{EnableDatagrams: true, InitialPacketSize: 1350}))
|
|
||||||
server := &http3.Server{Handler: http.HandlerFunc(handler), EnableDatagrams: true}
|
|
||||||
go server.ServeListener(ln)
|
|
||||||
t.Cleanup(func() {
|
|
||||||
server.Close()
|
|
||||||
ln.Close()
|
|
||||||
tr.Close()
|
|
||||||
pktConn.Close()
|
|
||||||
})
|
|
||||||
|
|
||||||
return net.Port(pktConn.LocalAddr().(*net.UDPAddr).Port), certHash
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestMasque(t *testing.T) {
|
|
||||||
serverPort, certHash := startMasqueServer(t)
|
|
||||||
|
|
||||||
tcpPort := tcp.PickPort()
|
|
||||||
tcp6Port := tcp.PickPort()
|
|
||||||
udpPort := udp.PickPort()
|
|
||||||
dokodemoTo := func(port net.Port, addr netip.Addr, network net.Network) *core.InboundHandlerConfig {
|
|
||||||
return &core.InboundHandlerConfig{
|
|
||||||
ReceiverSettings: serial.ToTypedMessage(&proxyman.ReceiverConfig{
|
|
||||||
PortList: &net.PortList{Range: []*net.PortRange{net.SinglePortRange(port)}},
|
|
||||||
Listen: net.NewIPOrDomain(net.LocalHostIP),
|
|
||||||
}),
|
|
||||||
ProxySettings: serial.ToTypedMessage(&dokodemo.Config{
|
|
||||||
RewriteAddress: net.NewIPOrDomain(net.IPAddress(addr.AsSlice())),
|
|
||||||
RewritePort: masqueEchoPort,
|
|
||||||
AllowedNetworks: []net.Network{network},
|
|
||||||
}),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
clientConfig := &core.Config{
|
|
||||||
App: []*serial.TypedMessage{
|
|
||||||
serial.ToTypedMessage(&log.Config{
|
|
||||||
ErrorLogLevel: clog.Severity_Debug,
|
|
||||||
ErrorLogType: log.LogType_Console,
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
Inbound: []*core.InboundHandlerConfig{
|
|
||||||
dokodemoTo(tcpPort, masqueServerV4, net.Network_TCP),
|
|
||||||
dokodemoTo(tcp6Port, masqueServerV6, net.Network_TCP),
|
|
||||||
dokodemoTo(udpPort, masqueServerV4, net.Network_UDP),
|
|
||||||
},
|
|
||||||
Outbound: []*core.OutboundHandlerConfig{
|
|
||||||
{
|
|
||||||
ProxySettings: serial.ToTypedMessage(&masque.ClientConfig{
|
|
||||||
Server: &protocol.ServerEndpoint{
|
|
||||||
Address: net.NewIPOrDomain(net.LocalHostIP),
|
|
||||||
Port: uint32(serverPort),
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{
|
|
||||||
StreamSettings: &internet.StreamConfig{
|
|
||||||
ProtocolName: "masque",
|
|
||||||
TransportSettings: []*internet.TransportConfig{
|
|
||||||
{
|
|
||||||
ProtocolName: "masque",
|
|
||||||
Settings: serial.ToTypedMessage(&transmasque.Config{
|
|
||||||
Path: transmasque.DefaultPath,
|
|
||||||
Headers: map[string]string{"Authorization": masqueAuthorization},
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
SecurityType: serial.GetMessageType(&tls.Config{}),
|
|
||||||
SecuritySettings: []*serial.TypedMessage{
|
|
||||||
serial.ToTypedMessage(&tls.Config{
|
|
||||||
ServerName: "localhost",
|
|
||||||
PinnedPeerCertSha256: [][]byte{certHash[:]},
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
servers, err := InitializeServerConfigs(clientConfig)
|
|
||||||
common.Must(err)
|
|
||||||
defer CloseAllServers(servers)
|
|
||||||
|
|
||||||
var errg errgroup.Group
|
|
||||||
for range 3 {
|
|
||||||
errg.Go(testTCPConn(tcpPort, 1024*1024, time.Second*20))
|
|
||||||
}
|
|
||||||
errg.Go(testTCPConn(tcp6Port, 1024*1024, time.Second*20))
|
|
||||||
errg.Go(testUDPConn(udpPort, 1024, time.Second*5))
|
|
||||||
if err := errg.Wait(); err != nil {
|
|
||||||
t.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -509,7 +509,7 @@ func TestVlessXtlsVisionReality(t *testing.T) {
|
|||||||
|
|
||||||
// This testing test all known utls fingerprint in tls.PresetFingerprints that support reality (expect unsafe and random*)
|
// This testing test all known utls fingerprint in tls.PresetFingerprints that support reality (expect unsafe and random*)
|
||||||
// Beacuse figerprint support may be broken after utls/reality update
|
// Beacuse figerprint support may be broken after utls/reality update
|
||||||
// Known working fingerprint: chrome, firefox, safari
|
// Known broken fingerprint: android, 360
|
||||||
func TestVlessRealityFingerprints(t *testing.T) {
|
func TestVlessRealityFingerprints(t *testing.T) {
|
||||||
TestFingerprint := func(fingerprint string) error {
|
TestFingerprint := func(fingerprint string) error {
|
||||||
tcpServer := tcp.Server{
|
tcpServer := tcp.Server{
|
||||||
@@ -641,7 +641,7 @@ func TestVlessRealityFingerprints(t *testing.T) {
|
|||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
fingerPrints := []string{"chrome", "firefox", "safari"}
|
fingerPrints := []string{"chrome", "firefox", "safari", "ios", "edge", "qq"}
|
||||||
wg := sync.WaitGroup{}
|
wg := sync.WaitGroup{}
|
||||||
wg.Add(len(fingerPrints))
|
wg.Add(len(fingerPrints))
|
||||||
for _, fp := range fingerPrints {
|
for _, fp := range fingerPrints {
|
||||||
|
|||||||
@@ -65,7 +65,6 @@ func TestWireguard(t *testing.T) {
|
|||||||
ProxySettings: serial.ToTypedMessage(&freedom.Config{
|
ProxySettings: serial.ToTypedMessage(&freedom.Config{
|
||||||
FinalRules: []*freedom.FinalRuleConfig{{Action: freedom.RuleAction_Allow}},
|
FinalRules: []*freedom.FinalRuleConfig{{Action: freedom.RuleAction_Allow}},
|
||||||
}),
|
}),
|
||||||
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{}),
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
@@ -105,7 +104,6 @@ func TestWireguard(t *testing.T) {
|
|||||||
AllowedIps: []string{"0.0.0.0/0", "::0/0"},
|
AllowedIps: []string{"0.0.0.0/0", "::0/0"},
|
||||||
}},
|
}},
|
||||||
}),
|
}),
|
||||||
SenderSettings: serial.ToTypedMessage(&proxyman.SenderConfig{}),
|
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ type ConfigCreator func() interface{}
|
|||||||
|
|
||||||
var globalTransportConfigCreatorCache = make(map[string]ConfigCreator)
|
var globalTransportConfigCreatorCache = make(map[string]ConfigCreator)
|
||||||
|
|
||||||
var strategy = [11][3]byte{
|
var strategy = [][]byte{
|
||||||
// name strategy, prefer, fallback
|
// name strategy, prefer, fallback
|
||||||
{0, 0, 0}, // AsIs none, /, /
|
{0, 0, 0}, // AsIs none, /, /
|
||||||
{1, 0, 0}, // UseIP use, both, none
|
{1, 0, 0}, // UseIP use, both, none
|
||||||
@@ -25,6 +25,8 @@ var strategy = [11][3]byte{
|
|||||||
{2, 6, 4}, // ForceIPv6v4 force, 6, 4
|
{2, 6, 4}, // ForceIPv6v4 force, 6, 4
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const unknownProtocol = "unknown"
|
||||||
|
|
||||||
func RegisterProtocolConfigCreator(name string, creator ConfigCreator) error {
|
func RegisterProtocolConfigCreator(name string, creator ConfigCreator) error {
|
||||||
if _, found := globalTransportConfigCreatorCache[name]; found {
|
if _, found := globalTransportConfigCreatorCache[name]; found {
|
||||||
return errors.New("protocol ", name, " is already registered").AtError()
|
return errors.New("protocol ", name, " is already registered").AtError()
|
||||||
@@ -89,6 +91,10 @@ func (c *StreamConfig) HasSecuritySettings() bool {
|
|||||||
return len(c.SecuritySettings) > 0
|
return len(c.SecuritySettings) > 0
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *ProxyConfig) HasTag() bool {
|
||||||
|
return c != nil && len(c.Tag) > 0
|
||||||
|
}
|
||||||
|
|
||||||
func (m SocketConfig_TProxyMode) IsEnabled() bool {
|
func (m SocketConfig_TProxyMode) IsEnabled() bool {
|
||||||
return m != SocketConfig_Off
|
return m != SocketConfig_Off
|
||||||
}
|
}
|
||||||
|
|||||||
+194
-63
@@ -206,7 +206,7 @@ func (x SocketConfig_TProxyMode) Number() protoreflect.EnumNumber {
|
|||||||
|
|
||||||
// Deprecated: Use SocketConfig_TProxyMode.Descriptor instead.
|
// Deprecated: Use SocketConfig_TProxyMode.Descriptor instead.
|
||||||
func (SocketConfig_TProxyMode) EnumDescriptor() ([]byte, []int) {
|
func (SocketConfig_TProxyMode) EnumDescriptor() ([]byte, []int) {
|
||||||
return file_transport_internet_config_proto_rawDescGZIP(), []int{4, 0}
|
return file_transport_internet_config_proto_rawDescGZIP(), []int{6, 0}
|
||||||
}
|
}
|
||||||
|
|
||||||
type TransportConfig struct {
|
type TransportConfig struct {
|
||||||
@@ -382,6 +382,66 @@ func (x *StreamConfig) GetSocketSettings() *SocketConfig {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type UdpHop struct {
|
||||||
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
|
Ports []uint32 `protobuf:"varint,1,rep,packed,name=ports,proto3" json:"ports,omitempty"`
|
||||||
|
IntervalMin int64 `protobuf:"varint,2,opt,name=interval_min,json=intervalMin,proto3" json:"interval_min,omitempty"`
|
||||||
|
IntervalMax int64 `protobuf:"varint,3,opt,name=interval_max,json=intervalMax,proto3" json:"interval_max,omitempty"`
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UdpHop) Reset() {
|
||||||
|
*x = UdpHop{}
|
||||||
|
mi := &file_transport_internet_config_proto_msgTypes[2]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UdpHop) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*UdpHop) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *UdpHop) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_transport_internet_config_proto_msgTypes[2]
|
||||||
|
if x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use UdpHop.ProtoReflect.Descriptor instead.
|
||||||
|
func (*UdpHop) Descriptor() ([]byte, []int) {
|
||||||
|
return file_transport_internet_config_proto_rawDescGZIP(), []int{2}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UdpHop) GetPorts() []uint32 {
|
||||||
|
if x != nil {
|
||||||
|
return x.Ports
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UdpHop) GetIntervalMin() int64 {
|
||||||
|
if x != nil {
|
||||||
|
return x.IntervalMin
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *UdpHop) GetIntervalMax() int64 {
|
||||||
|
if x != nil {
|
||||||
|
return x.IntervalMax
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
type QuicParams struct {
|
type QuicParams struct {
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
Congestion string `protobuf:"bytes,1,opt,name=congestion,proto3" json:"congestion,omitempty"`
|
Congestion string `protobuf:"bytes,1,opt,name=congestion,proto3" json:"congestion,omitempty"`
|
||||||
@@ -389,24 +449,25 @@ type QuicParams struct {
|
|||||||
BrutalUp uint64 `protobuf:"varint,3,opt,name=brutal_up,json=brutalUp,proto3" json:"brutal_up,omitempty"`
|
BrutalUp uint64 `protobuf:"varint,3,opt,name=brutal_up,json=brutalUp,proto3" json:"brutal_up,omitempty"`
|
||||||
BrutalDown uint64 `protobuf:"varint,4,opt,name=brutal_down,json=brutalDown,proto3" json:"brutal_down,omitempty"`
|
BrutalDown uint64 `protobuf:"varint,4,opt,name=brutal_down,json=brutalDown,proto3" json:"brutal_down,omitempty"`
|
||||||
BrutalDisableLossCompensation bool `protobuf:"varint,5,opt,name=brutal_disable_loss_compensation,json=brutalDisableLossCompensation,proto3" json:"brutal_disable_loss_compensation,omitempty"`
|
BrutalDisableLossCompensation bool `protobuf:"varint,5,opt,name=brutal_disable_loss_compensation,json=brutalDisableLossCompensation,proto3" json:"brutal_disable_loss_compensation,omitempty"`
|
||||||
InitStreamReceiveWindow uint64 `protobuf:"varint,6,opt,name=init_stream_receive_window,json=initStreamReceiveWindow,proto3" json:"init_stream_receive_window,omitempty"`
|
UdpHop *UdpHop `protobuf:"bytes,6,opt,name=udp_hop,json=udpHop,proto3" json:"udp_hop,omitempty"`
|
||||||
MaxStreamReceiveWindow uint64 `protobuf:"varint,7,opt,name=max_stream_receive_window,json=maxStreamReceiveWindow,proto3" json:"max_stream_receive_window,omitempty"`
|
InitStreamReceiveWindow uint64 `protobuf:"varint,7,opt,name=init_stream_receive_window,json=initStreamReceiveWindow,proto3" json:"init_stream_receive_window,omitempty"`
|
||||||
InitConnReceiveWindow uint64 `protobuf:"varint,8,opt,name=init_conn_receive_window,json=initConnReceiveWindow,proto3" json:"init_conn_receive_window,omitempty"`
|
MaxStreamReceiveWindow uint64 `protobuf:"varint,8,opt,name=max_stream_receive_window,json=maxStreamReceiveWindow,proto3" json:"max_stream_receive_window,omitempty"`
|
||||||
MaxConnReceiveWindow uint64 `protobuf:"varint,9,opt,name=max_conn_receive_window,json=maxConnReceiveWindow,proto3" json:"max_conn_receive_window,omitempty"`
|
InitConnReceiveWindow uint64 `protobuf:"varint,9,opt,name=init_conn_receive_window,json=initConnReceiveWindow,proto3" json:"init_conn_receive_window,omitempty"`
|
||||||
MaxIdleTimeout int64 `protobuf:"varint,10,opt,name=max_idle_timeout,json=maxIdleTimeout,proto3" json:"max_idle_timeout,omitempty"`
|
MaxConnReceiveWindow uint64 `protobuf:"varint,10,opt,name=max_conn_receive_window,json=maxConnReceiveWindow,proto3" json:"max_conn_receive_window,omitempty"`
|
||||||
KeepAlivePeriod int64 `protobuf:"varint,11,opt,name=keep_alive_period,json=keepAlivePeriod,proto3" json:"keep_alive_period,omitempty"`
|
MaxIdleTimeout int64 `protobuf:"varint,11,opt,name=max_idle_timeout,json=maxIdleTimeout,proto3" json:"max_idle_timeout,omitempty"`
|
||||||
DisablePathMtuDiscovery bool `protobuf:"varint,12,opt,name=disable_path_mtu_discovery,json=disablePathMtuDiscovery,proto3" json:"disable_path_mtu_discovery,omitempty"`
|
KeepAlivePeriod int64 `protobuf:"varint,12,opt,name=keep_alive_period,json=keepAlivePeriod,proto3" json:"keep_alive_period,omitempty"`
|
||||||
DisableChromeParrot bool `protobuf:"varint,13,opt,name=disable_chrome_parrot,json=disableChromeParrot,proto3" json:"disable_chrome_parrot,omitempty"`
|
DisablePathMtuDiscovery bool `protobuf:"varint,13,opt,name=disable_path_mtu_discovery,json=disablePathMtuDiscovery,proto3" json:"disable_path_mtu_discovery,omitempty"`
|
||||||
DisableGSO bool `protobuf:"varint,14,opt,name=disableGSO,proto3" json:"disableGSO,omitempty"`
|
DisableChromeParrot bool `protobuf:"varint,14,opt,name=disable_chrome_parrot,json=disableChromeParrot,proto3" json:"disable_chrome_parrot,omitempty"`
|
||||||
MaxIncomingStreams int64 `protobuf:"varint,15,opt,name=max_incoming_streams,json=maxIncomingStreams,proto3" json:"max_incoming_streams,omitempty"`
|
DisableGSO bool `protobuf:"varint,15,opt,name=disableGSO,proto3" json:"disableGSO,omitempty"`
|
||||||
DisableStatelessReset bool `protobuf:"varint,16,opt,name=disable_stateless_reset,json=disableStatelessReset,proto3" json:"disable_stateless_reset,omitempty"`
|
MaxIncomingStreams int64 `protobuf:"varint,16,opt,name=max_incoming_streams,json=maxIncomingStreams,proto3" json:"max_incoming_streams,omitempty"`
|
||||||
|
DisableStatelessReset bool `protobuf:"varint,17,opt,name=disable_stateless_reset,json=disableStatelessReset,proto3" json:"disable_stateless_reset,omitempty"`
|
||||||
unknownFields protoimpl.UnknownFields
|
unknownFields protoimpl.UnknownFields
|
||||||
sizeCache protoimpl.SizeCache
|
sizeCache protoimpl.SizeCache
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *QuicParams) Reset() {
|
func (x *QuicParams) Reset() {
|
||||||
*x = QuicParams{}
|
*x = QuicParams{}
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[2]
|
mi := &file_transport_internet_config_proto_msgTypes[3]
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
ms.StoreMessageInfo(mi)
|
ms.StoreMessageInfo(mi)
|
||||||
}
|
}
|
||||||
@@ -418,7 +479,7 @@ func (x *QuicParams) String() string {
|
|||||||
func (*QuicParams) ProtoMessage() {}
|
func (*QuicParams) ProtoMessage() {}
|
||||||
|
|
||||||
func (x *QuicParams) ProtoReflect() protoreflect.Message {
|
func (x *QuicParams) ProtoReflect() protoreflect.Message {
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[2]
|
mi := &file_transport_internet_config_proto_msgTypes[3]
|
||||||
if x != nil {
|
if x != nil {
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
if ms.LoadMessageInfo() == nil {
|
if ms.LoadMessageInfo() == nil {
|
||||||
@@ -431,7 +492,7 @@ func (x *QuicParams) ProtoReflect() protoreflect.Message {
|
|||||||
|
|
||||||
// Deprecated: Use QuicParams.ProtoReflect.Descriptor instead.
|
// Deprecated: Use QuicParams.ProtoReflect.Descriptor instead.
|
||||||
func (*QuicParams) Descriptor() ([]byte, []int) {
|
func (*QuicParams) Descriptor() ([]byte, []int) {
|
||||||
return file_transport_internet_config_proto_rawDescGZIP(), []int{2}
|
return file_transport_internet_config_proto_rawDescGZIP(), []int{3}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *QuicParams) GetCongestion() string {
|
func (x *QuicParams) GetCongestion() string {
|
||||||
@@ -469,6 +530,13 @@ func (x *QuicParams) GetBrutalDisableLossCompensation() bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (x *QuicParams) GetUdpHop() *UdpHop {
|
||||||
|
if x != nil {
|
||||||
|
return x.UdpHop
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (x *QuicParams) GetInitStreamReceiveWindow() uint64 {
|
func (x *QuicParams) GetInitStreamReceiveWindow() uint64 {
|
||||||
if x != nil {
|
if x != nil {
|
||||||
return x.InitStreamReceiveWindow
|
return x.InitStreamReceiveWindow
|
||||||
@@ -546,6 +614,58 @@ func (x *QuicParams) GetDisableStatelessReset() bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ProxyConfig struct {
|
||||||
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
|
Tag string `protobuf:"bytes,1,opt,name=tag,proto3" json:"tag,omitempty"`
|
||||||
|
TransportLayerProxy bool `protobuf:"varint,2,opt,name=transportLayerProxy,proto3" json:"transportLayerProxy,omitempty"`
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *ProxyConfig) Reset() {
|
||||||
|
*x = ProxyConfig{}
|
||||||
|
mi := &file_transport_internet_config_proto_msgTypes[4]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *ProxyConfig) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*ProxyConfig) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *ProxyConfig) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_transport_internet_config_proto_msgTypes[4]
|
||||||
|
if x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use ProxyConfig.ProtoReflect.Descriptor instead.
|
||||||
|
func (*ProxyConfig) Descriptor() ([]byte, []int) {
|
||||||
|
return file_transport_internet_config_proto_rawDescGZIP(), []int{4}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *ProxyConfig) GetTag() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.Tag
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *ProxyConfig) GetTransportLayerProxy() bool {
|
||||||
|
if x != nil {
|
||||||
|
return x.TransportLayerProxy
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
type CustomSockopt struct {
|
type CustomSockopt struct {
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
state protoimpl.MessageState `protogen:"open.v1"`
|
||||||
System string `protobuf:"bytes,1,opt,name=system,proto3" json:"system,omitempty"`
|
System string `protobuf:"bytes,1,opt,name=system,proto3" json:"system,omitempty"`
|
||||||
@@ -560,7 +680,7 @@ type CustomSockopt struct {
|
|||||||
|
|
||||||
func (x *CustomSockopt) Reset() {
|
func (x *CustomSockopt) Reset() {
|
||||||
*x = CustomSockopt{}
|
*x = CustomSockopt{}
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[3]
|
mi := &file_transport_internet_config_proto_msgTypes[5]
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
ms.StoreMessageInfo(mi)
|
ms.StoreMessageInfo(mi)
|
||||||
}
|
}
|
||||||
@@ -572,7 +692,7 @@ func (x *CustomSockopt) String() string {
|
|||||||
func (*CustomSockopt) ProtoMessage() {}
|
func (*CustomSockopt) ProtoMessage() {}
|
||||||
|
|
||||||
func (x *CustomSockopt) ProtoReflect() protoreflect.Message {
|
func (x *CustomSockopt) ProtoReflect() protoreflect.Message {
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[3]
|
mi := &file_transport_internet_config_proto_msgTypes[5]
|
||||||
if x != nil {
|
if x != nil {
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
if ms.LoadMessageInfo() == nil {
|
if ms.LoadMessageInfo() == nil {
|
||||||
@@ -585,7 +705,7 @@ func (x *CustomSockopt) ProtoReflect() protoreflect.Message {
|
|||||||
|
|
||||||
// Deprecated: Use CustomSockopt.ProtoReflect.Descriptor instead.
|
// Deprecated: Use CustomSockopt.ProtoReflect.Descriptor instead.
|
||||||
func (*CustomSockopt) Descriptor() ([]byte, []int) {
|
func (*CustomSockopt) Descriptor() ([]byte, []int) {
|
||||||
return file_transport_internet_config_proto_rawDescGZIP(), []int{3}
|
return file_transport_internet_config_proto_rawDescGZIP(), []int{5}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *CustomSockopt) GetSystem() string {
|
func (x *CustomSockopt) GetSystem() string {
|
||||||
@@ -665,7 +785,7 @@ type SocketConfig struct {
|
|||||||
|
|
||||||
func (x *SocketConfig) Reset() {
|
func (x *SocketConfig) Reset() {
|
||||||
*x = SocketConfig{}
|
*x = SocketConfig{}
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[4]
|
mi := &file_transport_internet_config_proto_msgTypes[6]
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
ms.StoreMessageInfo(mi)
|
ms.StoreMessageInfo(mi)
|
||||||
}
|
}
|
||||||
@@ -677,7 +797,7 @@ func (x *SocketConfig) String() string {
|
|||||||
func (*SocketConfig) ProtoMessage() {}
|
func (*SocketConfig) ProtoMessage() {}
|
||||||
|
|
||||||
func (x *SocketConfig) ProtoReflect() protoreflect.Message {
|
func (x *SocketConfig) ProtoReflect() protoreflect.Message {
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[4]
|
mi := &file_transport_internet_config_proto_msgTypes[6]
|
||||||
if x != nil {
|
if x != nil {
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
if ms.LoadMessageInfo() == nil {
|
if ms.LoadMessageInfo() == nil {
|
||||||
@@ -690,7 +810,7 @@ func (x *SocketConfig) ProtoReflect() protoreflect.Message {
|
|||||||
|
|
||||||
// Deprecated: Use SocketConfig.ProtoReflect.Descriptor instead.
|
// Deprecated: Use SocketConfig.ProtoReflect.Descriptor instead.
|
||||||
func (*SocketConfig) Descriptor() ([]byte, []int) {
|
func (*SocketConfig) Descriptor() ([]byte, []int) {
|
||||||
return file_transport_internet_config_proto_rawDescGZIP(), []int{4}
|
return file_transport_internet_config_proto_rawDescGZIP(), []int{6}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *SocketConfig) GetMark() int32 {
|
func (x *SocketConfig) GetMark() int32 {
|
||||||
@@ -852,7 +972,7 @@ type HappyEyeballsConfig struct {
|
|||||||
|
|
||||||
func (x *HappyEyeballsConfig) Reset() {
|
func (x *HappyEyeballsConfig) Reset() {
|
||||||
*x = HappyEyeballsConfig{}
|
*x = HappyEyeballsConfig{}
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[5]
|
mi := &file_transport_internet_config_proto_msgTypes[7]
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
ms.StoreMessageInfo(mi)
|
ms.StoreMessageInfo(mi)
|
||||||
}
|
}
|
||||||
@@ -864,7 +984,7 @@ func (x *HappyEyeballsConfig) String() string {
|
|||||||
func (*HappyEyeballsConfig) ProtoMessage() {}
|
func (*HappyEyeballsConfig) ProtoMessage() {}
|
||||||
|
|
||||||
func (x *HappyEyeballsConfig) ProtoReflect() protoreflect.Message {
|
func (x *HappyEyeballsConfig) ProtoReflect() protoreflect.Message {
|
||||||
mi := &file_transport_internet_config_proto_msgTypes[5]
|
mi := &file_transport_internet_config_proto_msgTypes[7]
|
||||||
if x != nil {
|
if x != nil {
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
if ms.LoadMessageInfo() == nil {
|
if ms.LoadMessageInfo() == nil {
|
||||||
@@ -877,7 +997,7 @@ func (x *HappyEyeballsConfig) ProtoReflect() protoreflect.Message {
|
|||||||
|
|
||||||
// Deprecated: Use HappyEyeballsConfig.ProtoReflect.Descriptor instead.
|
// Deprecated: Use HappyEyeballsConfig.ProtoReflect.Descriptor instead.
|
||||||
func (*HappyEyeballsConfig) Descriptor() ([]byte, []int) {
|
func (*HappyEyeballsConfig) Descriptor() ([]byte, []int) {
|
||||||
return file_transport_internet_config_proto_rawDescGZIP(), []int{5}
|
return file_transport_internet_config_proto_rawDescGZIP(), []int{7}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (x *HappyEyeballsConfig) GetPrioritizeIpv6() bool {
|
func (x *HappyEyeballsConfig) GetPrioritizeIpv6() bool {
|
||||||
@@ -928,7 +1048,11 @@ const file_transport_internet_config_proto_rawDesc = "" +
|
|||||||
"\btcpmasks\x18\v \x03(\v2 .xray.common.serial.TypedMessageR\btcpmasks\x12D\n" +
|
"\btcpmasks\x18\v \x03(\v2 .xray.common.serial.TypedMessageR\btcpmasks\x12D\n" +
|
||||||
"\vquic_params\x18\f \x01(\v2#.xray.transport.internet.QuicParamsR\n" +
|
"\vquic_params\x18\f \x01(\v2#.xray.transport.internet.QuicParamsR\n" +
|
||||||
"quicParams\x12N\n" +
|
"quicParams\x12N\n" +
|
||||||
"\x0fsocket_settings\x18\x06 \x01(\v2%.xray.transport.internet.SocketConfigR\x0esocketSettings\"\x8d\x06\n" +
|
"\x0fsocket_settings\x18\x06 \x01(\v2%.xray.transport.internet.SocketConfigR\x0esocketSettings\"d\n" +
|
||||||
|
"\x06UdpHop\x12\x14\n" +
|
||||||
|
"\x05ports\x18\x01 \x03(\rR\x05ports\x12!\n" +
|
||||||
|
"\finterval_min\x18\x02 \x01(\x03R\vintervalMin\x12!\n" +
|
||||||
|
"\finterval_max\x18\x03 \x01(\x03R\vintervalMax\"\xc7\x06\n" +
|
||||||
"\n" +
|
"\n" +
|
||||||
"QuicParams\x12\x1e\n" +
|
"QuicParams\x12\x1e\n" +
|
||||||
"\n" +
|
"\n" +
|
||||||
@@ -939,21 +1063,25 @@ const file_transport_internet_config_proto_rawDesc = "" +
|
|||||||
"\tbrutal_up\x18\x03 \x01(\x04R\bbrutalUp\x12\x1f\n" +
|
"\tbrutal_up\x18\x03 \x01(\x04R\bbrutalUp\x12\x1f\n" +
|
||||||
"\vbrutal_down\x18\x04 \x01(\x04R\n" +
|
"\vbrutal_down\x18\x04 \x01(\x04R\n" +
|
||||||
"brutalDown\x12G\n" +
|
"brutalDown\x12G\n" +
|
||||||
" brutal_disable_loss_compensation\x18\x05 \x01(\bR\x1dbrutalDisableLossCompensation\x12;\n" +
|
" brutal_disable_loss_compensation\x18\x05 \x01(\bR\x1dbrutalDisableLossCompensation\x128\n" +
|
||||||
"\x1ainit_stream_receive_window\x18\x06 \x01(\x04R\x17initStreamReceiveWindow\x129\n" +
|
"\audp_hop\x18\x06 \x01(\v2\x1f.xray.transport.internet.UdpHopR\x06udpHop\x12;\n" +
|
||||||
"\x19max_stream_receive_window\x18\a \x01(\x04R\x16maxStreamReceiveWindow\x127\n" +
|
"\x1ainit_stream_receive_window\x18\a \x01(\x04R\x17initStreamReceiveWindow\x129\n" +
|
||||||
"\x18init_conn_receive_window\x18\b \x01(\x04R\x15initConnReceiveWindow\x125\n" +
|
"\x19max_stream_receive_window\x18\b \x01(\x04R\x16maxStreamReceiveWindow\x127\n" +
|
||||||
"\x17max_conn_receive_window\x18\t \x01(\x04R\x14maxConnReceiveWindow\x12(\n" +
|
"\x18init_conn_receive_window\x18\t \x01(\x04R\x15initConnReceiveWindow\x125\n" +
|
||||||
"\x10max_idle_timeout\x18\n" +
|
"\x17max_conn_receive_window\x18\n" +
|
||||||
" \x01(\x03R\x0emaxIdleTimeout\x12*\n" +
|
" \x01(\x04R\x14maxConnReceiveWindow\x12(\n" +
|
||||||
"\x11keep_alive_period\x18\v \x01(\x03R\x0fkeepAlivePeriod\x12;\n" +
|
"\x10max_idle_timeout\x18\v \x01(\x03R\x0emaxIdleTimeout\x12*\n" +
|
||||||
"\x1adisable_path_mtu_discovery\x18\f \x01(\bR\x17disablePathMtuDiscovery\x122\n" +
|
"\x11keep_alive_period\x18\f \x01(\x03R\x0fkeepAlivePeriod\x12;\n" +
|
||||||
"\x15disable_chrome_parrot\x18\r \x01(\bR\x13disableChromeParrot\x12\x1e\n" +
|
"\x1adisable_path_mtu_discovery\x18\r \x01(\bR\x17disablePathMtuDiscovery\x122\n" +
|
||||||
|
"\x15disable_chrome_parrot\x18\x0e \x01(\bR\x13disableChromeParrot\x12\x1e\n" +
|
||||||
"\n" +
|
"\n" +
|
||||||
"disableGSO\x18\x0e \x01(\bR\n" +
|
"disableGSO\x18\x0f \x01(\bR\n" +
|
||||||
"disableGSO\x120\n" +
|
"disableGSO\x120\n" +
|
||||||
"\x14max_incoming_streams\x18\x0f \x01(\x03R\x12maxIncomingStreams\x126\n" +
|
"\x14max_incoming_streams\x18\x10 \x01(\x03R\x12maxIncomingStreams\x126\n" +
|
||||||
"\x17disable_stateless_reset\x18\x10 \x01(\bR\x15disableStatelessReset\"\x93\x01\n" +
|
"\x17disable_stateless_reset\x18\x11 \x01(\bR\x15disableStatelessReset\"Q\n" +
|
||||||
|
"\vProxyConfig\x12\x10\n" +
|
||||||
|
"\x03tag\x18\x01 \x01(\tR\x03tag\x120\n" +
|
||||||
|
"\x13transportLayerProxy\x18\x02 \x01(\bR\x13transportLayerProxy\"\x93\x01\n" +
|
||||||
"\rCustomSockopt\x12\x16\n" +
|
"\rCustomSockopt\x12\x16\n" +
|
||||||
"\x06system\x18\x01 \x01(\tR\x06system\x12\x18\n" +
|
"\x06system\x18\x01 \x01(\tR\x06system\x12\x18\n" +
|
||||||
"\anetwork\x18\x02 \x01(\tR\anetwork\x12\x14\n" +
|
"\anetwork\x18\x02 \x01(\tR\anetwork\x12\x14\n" +
|
||||||
@@ -1037,39 +1165,42 @@ func file_transport_internet_config_proto_rawDescGZIP() []byte {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var file_transport_internet_config_proto_enumTypes = make([]protoimpl.EnumInfo, 3)
|
var file_transport_internet_config_proto_enumTypes = make([]protoimpl.EnumInfo, 3)
|
||||||
var file_transport_internet_config_proto_msgTypes = make([]protoimpl.MessageInfo, 6)
|
var file_transport_internet_config_proto_msgTypes = make([]protoimpl.MessageInfo, 8)
|
||||||
var file_transport_internet_config_proto_goTypes = []any{
|
var file_transport_internet_config_proto_goTypes = []any{
|
||||||
(DomainStrategy)(0), // 0: xray.transport.internet.DomainStrategy
|
(DomainStrategy)(0), // 0: xray.transport.internet.DomainStrategy
|
||||||
(AddressPortStrategy)(0), // 1: xray.transport.internet.AddressPortStrategy
|
(AddressPortStrategy)(0), // 1: xray.transport.internet.AddressPortStrategy
|
||||||
(SocketConfig_TProxyMode)(0), // 2: xray.transport.internet.SocketConfig.TProxyMode
|
(SocketConfig_TProxyMode)(0), // 2: xray.transport.internet.SocketConfig.TProxyMode
|
||||||
(*TransportConfig)(nil), // 3: xray.transport.internet.TransportConfig
|
(*TransportConfig)(nil), // 3: xray.transport.internet.TransportConfig
|
||||||
(*StreamConfig)(nil), // 4: xray.transport.internet.StreamConfig
|
(*StreamConfig)(nil), // 4: xray.transport.internet.StreamConfig
|
||||||
(*QuicParams)(nil), // 5: xray.transport.internet.QuicParams
|
(*UdpHop)(nil), // 5: xray.transport.internet.UdpHop
|
||||||
(*CustomSockopt)(nil), // 6: xray.transport.internet.CustomSockopt
|
(*QuicParams)(nil), // 6: xray.transport.internet.QuicParams
|
||||||
(*SocketConfig)(nil), // 7: xray.transport.internet.SocketConfig
|
(*ProxyConfig)(nil), // 7: xray.transport.internet.ProxyConfig
|
||||||
(*HappyEyeballsConfig)(nil), // 8: xray.transport.internet.HappyEyeballsConfig
|
(*CustomSockopt)(nil), // 8: xray.transport.internet.CustomSockopt
|
||||||
(*serial.TypedMessage)(nil), // 9: xray.common.serial.TypedMessage
|
(*SocketConfig)(nil), // 9: xray.transport.internet.SocketConfig
|
||||||
(*net.IPOrDomain)(nil), // 10: xray.common.net.IPOrDomain
|
(*HappyEyeballsConfig)(nil), // 10: xray.transport.internet.HappyEyeballsConfig
|
||||||
|
(*serial.TypedMessage)(nil), // 11: xray.common.serial.TypedMessage
|
||||||
|
(*net.IPOrDomain)(nil), // 12: xray.common.net.IPOrDomain
|
||||||
}
|
}
|
||||||
var file_transport_internet_config_proto_depIdxs = []int32{
|
var file_transport_internet_config_proto_depIdxs = []int32{
|
||||||
9, // 0: xray.transport.internet.TransportConfig.settings:type_name -> xray.common.serial.TypedMessage
|
11, // 0: xray.transport.internet.TransportConfig.settings:type_name -> xray.common.serial.TypedMessage
|
||||||
10, // 1: xray.transport.internet.StreamConfig.address:type_name -> xray.common.net.IPOrDomain
|
12, // 1: xray.transport.internet.StreamConfig.address:type_name -> xray.common.net.IPOrDomain
|
||||||
3, // 2: xray.transport.internet.StreamConfig.transport_settings:type_name -> xray.transport.internet.TransportConfig
|
3, // 2: xray.transport.internet.StreamConfig.transport_settings:type_name -> xray.transport.internet.TransportConfig
|
||||||
9, // 3: xray.transport.internet.StreamConfig.security_settings:type_name -> xray.common.serial.TypedMessage
|
11, // 3: xray.transport.internet.StreamConfig.security_settings:type_name -> xray.common.serial.TypedMessage
|
||||||
9, // 4: xray.transport.internet.StreamConfig.udpmasks:type_name -> xray.common.serial.TypedMessage
|
11, // 4: xray.transport.internet.StreamConfig.udpmasks:type_name -> xray.common.serial.TypedMessage
|
||||||
9, // 5: xray.transport.internet.StreamConfig.tcpmasks:type_name -> xray.common.serial.TypedMessage
|
11, // 5: xray.transport.internet.StreamConfig.tcpmasks:type_name -> xray.common.serial.TypedMessage
|
||||||
5, // 6: xray.transport.internet.StreamConfig.quic_params:type_name -> xray.transport.internet.QuicParams
|
6, // 6: xray.transport.internet.StreamConfig.quic_params:type_name -> xray.transport.internet.QuicParams
|
||||||
7, // 7: xray.transport.internet.StreamConfig.socket_settings:type_name -> xray.transport.internet.SocketConfig
|
9, // 7: xray.transport.internet.StreamConfig.socket_settings:type_name -> xray.transport.internet.SocketConfig
|
||||||
2, // 8: xray.transport.internet.SocketConfig.tproxy:type_name -> xray.transport.internet.SocketConfig.TProxyMode
|
5, // 8: xray.transport.internet.QuicParams.udp_hop:type_name -> xray.transport.internet.UdpHop
|
||||||
0, // 9: xray.transport.internet.SocketConfig.domain_strategy:type_name -> xray.transport.internet.DomainStrategy
|
2, // 9: xray.transport.internet.SocketConfig.tproxy:type_name -> xray.transport.internet.SocketConfig.TProxyMode
|
||||||
6, // 10: xray.transport.internet.SocketConfig.customSockopt:type_name -> xray.transport.internet.CustomSockopt
|
0, // 10: xray.transport.internet.SocketConfig.domain_strategy:type_name -> xray.transport.internet.DomainStrategy
|
||||||
1, // 11: xray.transport.internet.SocketConfig.address_port_strategy:type_name -> xray.transport.internet.AddressPortStrategy
|
8, // 11: xray.transport.internet.SocketConfig.customSockopt:type_name -> xray.transport.internet.CustomSockopt
|
||||||
8, // 12: xray.transport.internet.SocketConfig.happy_eyeballs:type_name -> xray.transport.internet.HappyEyeballsConfig
|
1, // 12: xray.transport.internet.SocketConfig.address_port_strategy:type_name -> xray.transport.internet.AddressPortStrategy
|
||||||
13, // [13:13] is the sub-list for method output_type
|
10, // 13: xray.transport.internet.SocketConfig.happy_eyeballs:type_name -> xray.transport.internet.HappyEyeballsConfig
|
||||||
13, // [13:13] is the sub-list for method input_type
|
14, // [14:14] is the sub-list for method output_type
|
||||||
13, // [13:13] is the sub-list for extension type_name
|
14, // [14:14] is the sub-list for method input_type
|
||||||
13, // [13:13] is the sub-list for extension extendee
|
14, // [14:14] is the sub-list for extension type_name
|
||||||
0, // [0:13] is the sub-list for field type_name
|
14, // [14:14] is the sub-list for extension extendee
|
||||||
|
0, // [0:14] is the sub-list for field type_name
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() { file_transport_internet_config_proto_init() }
|
func init() { file_transport_internet_config_proto_init() }
|
||||||
@@ -1083,7 +1214,7 @@ func file_transport_internet_config_proto_init() {
|
|||||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
||||||
RawDescriptor: unsafe.Slice(unsafe.StringData(file_transport_internet_config_proto_rawDesc), len(file_transport_internet_config_proto_rawDesc)),
|
RawDescriptor: unsafe.Slice(unsafe.StringData(file_transport_internet_config_proto_rawDesc), len(file_transport_internet_config_proto_rawDesc)),
|
||||||
NumEnums: 3,
|
NumEnums: 3,
|
||||||
NumMessages: 6,
|
NumMessages: 8,
|
||||||
NumExtensions: 0,
|
NumExtensions: 0,
|
||||||
NumServices: 0,
|
NumServices: 0,
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -64,23 +64,35 @@ message StreamConfig {
|
|||||||
SocketConfig socket_settings = 6;
|
SocketConfig socket_settings = 6;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
message UdpHop {
|
||||||
|
repeated uint32 ports = 1;
|
||||||
|
int64 interval_min = 2;
|
||||||
|
int64 interval_max = 3;
|
||||||
|
}
|
||||||
|
|
||||||
message QuicParams {
|
message QuicParams {
|
||||||
string congestion = 1;
|
string congestion = 1;
|
||||||
string bbr_profile = 2;
|
string bbr_profile = 2;
|
||||||
uint64 brutal_up = 3;
|
uint64 brutal_up = 3;
|
||||||
uint64 brutal_down = 4;
|
uint64 brutal_down = 4;
|
||||||
bool brutal_disable_loss_compensation = 5;
|
bool brutal_disable_loss_compensation = 5;
|
||||||
uint64 init_stream_receive_window = 6;
|
UdpHop udp_hop = 6;
|
||||||
uint64 max_stream_receive_window = 7;
|
uint64 init_stream_receive_window = 7;
|
||||||
uint64 init_conn_receive_window = 8;
|
uint64 max_stream_receive_window = 8;
|
||||||
uint64 max_conn_receive_window = 9;
|
uint64 init_conn_receive_window = 9;
|
||||||
int64 max_idle_timeout = 10;
|
uint64 max_conn_receive_window = 10;
|
||||||
int64 keep_alive_period = 11;
|
int64 max_idle_timeout = 11;
|
||||||
bool disable_path_mtu_discovery = 12;
|
int64 keep_alive_period = 12;
|
||||||
bool disable_chrome_parrot = 13;
|
bool disable_path_mtu_discovery = 13;
|
||||||
bool disableGSO = 14;
|
bool disable_chrome_parrot = 14;
|
||||||
int64 max_incoming_streams = 15;
|
bool disableGSO = 15;
|
||||||
bool disable_stateless_reset = 16;
|
int64 max_incoming_streams = 16;
|
||||||
|
bool disable_stateless_reset = 17;
|
||||||
|
}
|
||||||
|
|
||||||
|
message ProxyConfig {
|
||||||
|
string tag = 1;
|
||||||
|
bool transportLayerProxy = 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
message CustomSockopt {
|
message CustomSockopt {
|
||||||
|
|||||||
@@ -2,291 +2,106 @@ package finalmask
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"net"
|
||||||
"slices"
|
"slices"
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/buf"
|
"github.com/xtls/xray-core/common/buf"
|
||||||
"github.com/xtls/xray-core/common/errors"
|
"github.com/xtls/xray-core/common/errors"
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Dialer struct {
|
type Udpmask interface {
|
||||||
DialTCP func(net.Destination) (net.Conn, error)
|
UDP()
|
||||||
DialUDP func(net.Destination) (net.Conn, error)
|
|
||||||
|
WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error)
|
||||||
|
WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
type ListenConfig struct {
|
type UdpmaskManager struct {
|
||||||
Listen func(net.Addr) (net.Listener, error)
|
udpmasks []Udpmask
|
||||||
ListenPacket func(net.Addr) (net.PacketConn, error)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type TCPMask interface {
|
func NewUdpmaskManager(udpmasks []Udpmask) *UdpmaskManager {
|
||||||
WrapConnClient(net.Conn, *net.Destination, *Dialer) (net.Conn, error)
|
return &UdpmaskManager{
|
||||||
WrapConnServer(net.Conn) (net.Conn, error)
|
udpmasks: udpmasks,
|
||||||
// Listen(net.Listener) (net.Listener, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
type UDPMask interface {
|
|
||||||
WrapPacketConnClient(net.PacketConn, *net.Destination, *Dialer) (net.PacketConn, error)
|
|
||||||
WrapPacketConnServer(net.PacketConn, net.Addr, *ListenConfig) (net.PacketConn, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
type FinalMask struct {
|
|
||||||
tcpMasks []TCPMask
|
|
||||||
udpMasks []UDPMask
|
|
||||||
dialTCP func(context.Context, net.Destination) (net.Conn, error)
|
|
||||||
listen func(context.Context, net.Addr) (net.Listener, error)
|
|
||||||
dialUDP func(context.Context, net.Destination) (net.PacketConn, net.Addr, error)
|
|
||||||
listenPacket func(context.Context, net.Addr) (net.PacketConn, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewFinalMask(tcpMasks []TCPMask, udpMasks []UDPMask, dialTCP func(context.Context, net.Destination) (net.Conn, error), listen func(context.Context, net.Addr) (net.Listener, error), dialUDP func(context.Context, net.Destination) (net.PacketConn, net.Addr, error), listenPacket func(context.Context, net.Addr) (net.PacketConn, error)) *FinalMask {
|
|
||||||
slices.Reverse(tcpMasks)
|
|
||||||
slices.Reverse(udpMasks)
|
|
||||||
return &FinalMask{
|
|
||||||
tcpMasks: tcpMasks,
|
|
||||||
udpMasks: udpMasks,
|
|
||||||
dialTCP: dialTCP,
|
|
||||||
dialUDP: dialUDP,
|
|
||||||
listen: listen,
|
|
||||||
listenPacket: listenPacket,
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (fm *FinalMask) DialTCP(ctx context.Context, dest net.Destination) (net.Conn, error) {
|
func (m *UdpmaskManager) WrapPacketConnClient(raw net.PacketConn) (net.PacketConn, error) {
|
||||||
if len(fm.tcpMasks) == 0 {
|
|
||||||
return fm.dialTCP(ctx, dest)
|
|
||||||
}
|
|
||||||
for i := range fm.tcpMasks {
|
|
||||||
if i > 0 {
|
|
||||||
if _, ok := fm.tcpMasks[i].(interface{ HandleDial() }); ok {
|
|
||||||
return nil, fmt.Errorf("incorrect index: %d %T", i, fm.tcpMasks[i])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
var conn net.Conn
|
|
||||||
var err error
|
|
||||||
if _, ok := fm.tcpMasks[0].(interface{ HandleDial() }); !ok {
|
|
||||||
conn, err = fm.dialTCP(ctx, dest)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
dialer := &Dialer{
|
|
||||||
DialTCP: func(dest net.Destination) (net.Conn, error) {
|
|
||||||
return fm.dialTCP(ctx, dest)
|
|
||||||
},
|
|
||||||
DialUDP: func(dest net.Destination) (net.Conn, error) {
|
|
||||||
conn, addr, err := fm.dialUDP(ctx, dest)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &PacketConnWrapper{PacketConn: conn, udpAddr: addr}, err
|
|
||||||
},
|
|
||||||
}
|
|
||||||
for i := range fm.tcpMasks {
|
|
||||||
var newConn net.Conn
|
|
||||||
newConn, err = fm.tcpMasks[i].WrapConnClient(conn, &dest, dialer)
|
|
||||||
if err != nil {
|
|
||||||
_ = conn.Close()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
conn = newConn
|
|
||||||
}
|
|
||||||
return conn, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (fm *FinalMask) Listen(ctx context.Context, addr net.Addr) (net.Listener, error) {
|
|
||||||
if len(fm.tcpMasks) == 0 {
|
|
||||||
return fm.listen(ctx, addr)
|
|
||||||
}
|
|
||||||
off := 0
|
|
||||||
listener, err := fm.listen(ctx, addr)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
for i := range fm.tcpMasks {
|
|
||||||
if _, ok := fm.tcpMasks[i].(interface {
|
|
||||||
Listen(net.Listener) (net.Listener, error)
|
|
||||||
}); ok {
|
|
||||||
if i-off == 0 {
|
|
||||||
l, err := fm.tcpMasks[i].(interface {
|
|
||||||
Listen(net.Listener) (net.Listener, error)
|
|
||||||
}).Listen(listener)
|
|
||||||
if err != nil {
|
|
||||||
listener.Close()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
listener = l
|
|
||||||
} else {
|
|
||||||
l, err := fm.tcpMasks[i].(interface {
|
|
||||||
Listen(net.Listener) (net.Listener, error)
|
|
||||||
}).Listen(&TCPListener{Listener: listener, tcpMasks: fm.tcpMasks[off:i]})
|
|
||||||
if err != nil {
|
|
||||||
listener.Close()
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
listener = l
|
|
||||||
}
|
|
||||||
off = i + 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if off < len(fm.tcpMasks) {
|
|
||||||
return &TCPListener{Listener: listener, tcpMasks: fm.tcpMasks[off:]}, nil
|
|
||||||
}
|
|
||||||
return listener, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (fm *FinalMask) DialUDP(ctx context.Context, dest net.Destination) (net.Conn, error) {
|
|
||||||
if len(fm.udpMasks) == 0 {
|
|
||||||
conn, addr, err := fm.dialUDP(ctx, dest)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &PacketConnWrapper{PacketConn: conn, udpAddr: addr}, nil
|
|
||||||
}
|
|
||||||
for i := range fm.udpMasks {
|
|
||||||
if i > 0 {
|
|
||||||
if _, ok := fm.udpMasks[i].(interface{ HandleDial() }); ok {
|
|
||||||
return nil, fmt.Errorf("incorrect index: %d %T", i, fm.udpMasks[i])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
var conn net.PacketConn
|
|
||||||
var addr net.Addr
|
|
||||||
var err error
|
|
||||||
if _, ok := fm.udpMasks[0].(interface{ HandleDial() }); !ok {
|
|
||||||
conn, addr, err = fm.dialUDP(ctx, dest)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
dialer := &Dialer{
|
|
||||||
DialTCP: func(dest net.Destination) (net.Conn, error) {
|
|
||||||
return fm.dialTCP(ctx, dest)
|
|
||||||
},
|
|
||||||
DialUDP: func(dest net.Destination) (net.Conn, error) {
|
|
||||||
conn, addr, err := fm.dialUDP(ctx, dest)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &PacketConnWrapper{PacketConn: conn, udpAddr: addr}, err
|
|
||||||
},
|
|
||||||
}
|
|
||||||
var sizes []int
|
var sizes []int
|
||||||
var conns []net.PacketConn
|
var conns []net.PacketConn
|
||||||
for i := range fm.udpMasks {
|
for i, mask := range slices.Backward(m.udpmasks) {
|
||||||
var newConn net.PacketConn
|
if _, ok := mask.(headerConn); ok {
|
||||||
if _, ok := fm.udpMasks[i].(interface{ HeaderConn() }); ok {
|
conn, err := mask.WrapPacketConnClient(nil, i, len(m.udpmasks)-1)
|
||||||
newConn, err = fm.udpMasks[i].WrapPacketConnClient(nil, nil, nil)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_ = conn.Close()
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
sizes = append(sizes, newConn.(interface{ Size() int }).Size())
|
sizes = append(sizes, conn.(headerSize).Size())
|
||||||
conns = append(conns, newConn)
|
conns = append(conns, conn)
|
||||||
} else {
|
} else {
|
||||||
if len(conns) > 0 {
|
if len(conns) > 0 {
|
||||||
conn = &headerManagerConn{PacketConn: conn, sizes: sizes, conns: conns}
|
raw = &headerManagerConn{sizes: sizes, conns: conns, PacketConn: raw}
|
||||||
sizes = nil
|
sizes = nil
|
||||||
conns = nil
|
conns = nil
|
||||||
}
|
}
|
||||||
newConn, err = fm.udpMasks[i].WrapPacketConnClient(conn, &dest, dialer)
|
var err error
|
||||||
|
raw, err = mask.WrapPacketConnClient(raw, i, len(m.udpmasks)-1)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_ = conn.Close()
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
conn = newConn
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(conns) > 0 {
|
if len(conns) > 0 {
|
||||||
conn = &headerManagerConn{PacketConn: conn, sizes: sizes, conns: conns}
|
raw = &headerManagerConn{sizes: sizes, conns: conns, PacketConn: raw}
|
||||||
sizes = nil
|
sizes = nil
|
||||||
conns = nil
|
conns = nil
|
||||||
}
|
}
|
||||||
if addr == nil {
|
return raw, nil
|
||||||
addr = &net.UDPAddr{IP: []byte{0, 0, 0, 0}}
|
|
||||||
}
|
|
||||||
return &PacketConnWrapper{PacketConn: conn, udpAddr: addr}, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (fm *FinalMask) ListenPacket(ctx context.Context, addr net.Addr) (net.PacketConn, error) {
|
func (m *UdpmaskManager) WrapPacketConnServer(raw net.PacketConn) (net.PacketConn, error) {
|
||||||
if len(fm.udpMasks) == 0 {
|
|
||||||
return fm.listenPacket(ctx, addr)
|
|
||||||
}
|
|
||||||
for i := range fm.udpMasks {
|
|
||||||
if i > 0 {
|
|
||||||
if _, ok := fm.udpMasks[i].(interface{ HandleListen() }); ok {
|
|
||||||
return nil, fmt.Errorf("incorrect index: %d %T", i, fm.udpMasks[i])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
var conn net.PacketConn
|
|
||||||
var err error
|
|
||||||
if _, ok := fm.udpMasks[0].(interface{ HandleListen() }); !ok {
|
|
||||||
conn, err = fm.listenPacket(ctx, addr)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
lc := &ListenConfig{
|
|
||||||
Listen: func(addr net.Addr) (net.Listener, error) { return fm.listen(ctx, addr) },
|
|
||||||
ListenPacket: func(addr net.Addr) (net.PacketConn, error) { return fm.listenPacket(ctx, addr) },
|
|
||||||
}
|
|
||||||
var sizes []int
|
var sizes []int
|
||||||
var conns []net.PacketConn
|
var conns []net.PacketConn
|
||||||
for i := range fm.udpMasks {
|
for i, mask := range slices.Backward(m.udpmasks) {
|
||||||
var newConn net.PacketConn
|
if _, ok := mask.(headerConn); ok {
|
||||||
if _, ok := fm.udpMasks[i].(interface{ HeaderConn() }); ok {
|
conn, err := mask.WrapPacketConnServer(nil, i, len(m.udpmasks)-1)
|
||||||
newConn, err = fm.udpMasks[i].WrapPacketConnServer(nil, nil, nil)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_ = conn.Close()
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
sizes = append(sizes, newConn.(interface{ Size() int }).Size())
|
sizes = append(sizes, conn.(headerSize).Size())
|
||||||
conns = append(conns, newConn)
|
conns = append(conns, conn)
|
||||||
} else {
|
} else {
|
||||||
if len(conns) > 0 {
|
if len(conns) > 0 {
|
||||||
conn = &headerManagerConn{PacketConn: conn, sizes: sizes, conns: conns}
|
raw = &headerManagerConn{sizes: sizes, conns: conns, PacketConn: raw}
|
||||||
sizes = nil
|
sizes = nil
|
||||||
conns = nil
|
conns = nil
|
||||||
}
|
}
|
||||||
newConn, err = fm.udpMasks[i].WrapPacketConnServer(conn, addr, lc)
|
var err error
|
||||||
|
raw, err = mask.WrapPacketConnServer(raw, i, len(m.udpmasks)-1)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_ = conn.Close()
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
conn = newConn
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(conns) > 0 {
|
if len(conns) > 0 {
|
||||||
conn = &headerManagerConn{PacketConn: conn, sizes: sizes, conns: conns}
|
raw = &headerManagerConn{sizes: sizes, conns: conns, PacketConn: raw}
|
||||||
sizes = nil
|
sizes = nil
|
||||||
conns = nil
|
conns = nil
|
||||||
}
|
}
|
||||||
return conn, nil
|
return raw, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
UDPSize = 4096
|
UDPSize = 4096
|
||||||
)
|
)
|
||||||
|
|
||||||
type PacketConnWrapper struct {
|
type headerConn interface {
|
||||||
net.PacketConn
|
HeaderConn()
|
||||||
udpAddr net.Addr
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *PacketConnWrapper) RemoteAddr() net.Addr {
|
type headerSize interface {
|
||||||
return c.udpAddr
|
Size() int
|
||||||
}
|
|
||||||
|
|
||||||
func (c *PacketConnWrapper) Read(b []byte) (n int, err error) {
|
|
||||||
n, _, err = c.PacketConn.ReadFrom(b)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *PacketConnWrapper) Write(b []byte) (n int, err error) {
|
|
||||||
return c.PacketConn.WriteTo(b, c.udpAddr)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type headerManagerConn struct {
|
type headerManagerConn struct {
|
||||||
@@ -379,27 +194,75 @@ func (c *headerManagerConn) WriteTo(p []byte, addr net.Addr) (n int, err error)
|
|||||||
return len(p), nil
|
return len(p), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type TCPListener struct {
|
type Tcpmask interface {
|
||||||
net.Listener
|
TCP()
|
||||||
tcpMasks []TCPMask
|
|
||||||
|
WrapConnClient(net.Conn) (net.Conn, error)
|
||||||
|
WrapConnServer(net.Conn) (net.Conn, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *TCPListener) Accept() (net.Conn, error) {
|
type TcpmaskManager struct {
|
||||||
|
tcpmasks []Tcpmask
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewTcpmaskManager(tcpmasks []Tcpmask) *TcpmaskManager {
|
||||||
|
return &TcpmaskManager{
|
||||||
|
tcpmasks: tcpmasks,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TcpmaskManager) WrapConnClient(raw net.Conn) (net.Conn, error) {
|
||||||
|
var err error
|
||||||
|
for _, mask := range slices.Backward(m.tcpmasks) {
|
||||||
|
raw, err = mask.WrapConnClient(raw)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return raw, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TcpmaskManager) WrapConnServer(raw net.Conn) (net.Conn, error) {
|
||||||
|
var err error
|
||||||
|
for _, mask := range slices.Backward(m.tcpmasks) {
|
||||||
|
raw, err = mask.WrapConnServer(raw)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return raw, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *TcpmaskManager) WrapListener(l net.Listener) (net.Listener, error) {
|
||||||
|
return NewTcpListener(m, l)
|
||||||
|
}
|
||||||
|
|
||||||
|
type tcpListener struct {
|
||||||
|
m *TcpmaskManager
|
||||||
|
net.Listener
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewTcpListener(m *TcpmaskManager, l net.Listener) (net.Listener, error) {
|
||||||
|
return &tcpListener{
|
||||||
|
m: m,
|
||||||
|
Listener: l,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *tcpListener) Accept() (net.Conn, error) {
|
||||||
conn, err := l.Listener.Accept()
|
conn, err := l.Listener.Accept()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return conn, err
|
return conn, err
|
||||||
}
|
}
|
||||||
|
|
||||||
for i := range l.tcpMasks {
|
newConn, err := l.m.WrapConnServer(conn)
|
||||||
var newConn net.Conn
|
if err != nil {
|
||||||
newConn, err = l.tcpMasks[i].WrapConnServer(conn)
|
errors.LogDebugInner(context.Background(), err, "mask err")
|
||||||
if err != nil {
|
_ = conn.Close()
|
||||||
_ = conn.Close()
|
return nil, err
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
conn = newConn
|
|
||||||
}
|
}
|
||||||
return conn, nil
|
|
||||||
|
return newConn, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type TcpMaskConn interface {
|
type TcpMaskConn interface {
|
||||||
|
|||||||
@@ -1,14 +1,14 @@
|
|||||||
package fragment
|
package fragment
|
||||||
|
|
||||||
import (
|
import "net"
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (c *Config) WrapConnClient(conn net.Conn, dest *net.Destination, dialer *finalmask.Dialer) (net.Conn, error) {
|
func (c *Config) TCP() {
|
||||||
return NewConnClient(c, conn, false)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapConnServer(conn net.Conn) (net.Conn, error) {
|
func (c *Config) WrapConnClient(raw net.Conn) (net.Conn, error) {
|
||||||
return NewConnServer(c, conn, true)
|
return NewConnClient(c, raw, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Config) WrapConnServer(raw net.Conn) (net.Conn, error) {
|
||||||
|
return NewConnServer(c, raw, true)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,30 +1,35 @@
|
|||||||
package custom
|
package custom
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func (c *TCPConfig) WrapConnClient(conn net.Conn, dest *net.Destination, dialer *finalmask.Dialer) (net.Conn, error) {
|
func (c *TCPConfig) TCP() {}
|
||||||
return NewConnClientTCP(c, conn)
|
|
||||||
|
func (c *TCPConfig) WrapConnClient(raw net.Conn) (net.Conn, error) {
|
||||||
|
return NewConnClientTCP(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *TCPConfig) WrapConnServer(conn net.Conn) (net.Conn, error) {
|
func (c *TCPConfig) WrapConnServer(raw net.Conn) (net.Conn, error) {
|
||||||
return NewConnServerTCP(c, conn)
|
return NewConnServerTCP(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPConfig) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *UDPConfig) UDP() {}
|
||||||
return NewConnClientUDP(c, conn)
|
|
||||||
|
func (c *UDPConfig) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
|
return NewConnClientUDP(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPConfig) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *UDPConfig) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServerUDP(c, conn)
|
return NewConnServerUDP(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPStandaloneConfig) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *UDPStandaloneConfig) UDP() {}
|
||||||
return NewConnClientUDPStandalone(c, conn)
|
|
||||||
|
func (c *UDPStandaloneConfig) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
|
return NewConnClientUDPStandalone(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *UDPStandaloneConfig) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *UDPStandaloneConfig) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServerUDPStandalone(c, conn)
|
return NewConnServerUDPStandalone(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,6 +9,8 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/xtls/xray-core/transport/internet/finalmask"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestMetadataEvaluatorRejectsUnknownName(t *testing.T) {
|
func TestMetadataEvaluatorRejectsUnknownName(t *testing.T) {
|
||||||
@@ -154,7 +156,7 @@ func TestMetadataUDPStandaloneWriteUsesRemotePort(t *testing.T) {
|
|||||||
}
|
}
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := cfg.WrapPacketConnClient(clientRaw, nil, nil)
|
client, err := finalmask.NewUdpmaskManager([]finalmask.Udpmask{cfg}).WrapPacketConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -299,7 +301,7 @@ func TestMetadataTCPHandshakeUsesEndpointPorts(t *testing.T) {
|
|||||||
}
|
}
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := clientCfg.WrapConnClient(clientRaw, nil, nil)
|
client, err := clientCfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,6 +5,8 @@ import (
|
|||||||
"net"
|
"net"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/xtls/xray-core/transport/internet/finalmask"
|
||||||
)
|
)
|
||||||
|
|
||||||
func mustSendRecvUDP(t *testing.T, from net.PacketConn, to net.PacketConn, msg []byte) {
|
func mustSendRecvUDP(t *testing.T, from net.PacketConn, to net.PacketConn, msg []byte) {
|
||||||
@@ -46,6 +48,7 @@ func TestStateUDPResponseReusesPriorCapturedValues(t *testing.T) {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
maskManager := finalmask.NewUdpmaskManager([]finalmask.Udpmask{cfg})
|
||||||
|
|
||||||
clientRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
clientRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -59,11 +62,11 @@ func TestStateUDPResponseReusesPriorCapturedValues(t *testing.T) {
|
|||||||
}
|
}
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := cfg.WrapPacketConnClient(clientRaw, nil, nil)
|
client, err := maskManager.WrapPacketConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
server, err := cfg.WrapPacketConnServer(serverRaw, nil, nil)
|
server, err := maskManager.WrapPacketConnServer(serverRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ func TestDSLTCPHandshakeReusesCapturedValue(t *testing.T) {
|
|||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
client, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -117,7 +117,7 @@ func TestDSLTCPClientRejectsMismatchedResponseSequence(t *testing.T) {
|
|||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := clientCfg.WrapConnClient(clientRaw, nil, nil)
|
client, err := clientCfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,16 +1,17 @@
|
|||||||
package aes128gcm
|
package aes128gcm
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func (c *Config) UDP() {}
|
||||||
|
|
||||||
func (c *Config) HeaderConn() {}
|
func (c *Config) HeaderConn() {}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnClient(c, conn)
|
return NewConnClient(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServer(c, conn)
|
return NewConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,16 +1,17 @@
|
|||||||
package header
|
package header
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func (c *Config) UDP() {}
|
||||||
|
|
||||||
func (c *Config) HeaderConn() {}
|
func (c *Config) HeaderConn() {}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnClient(c, conn)
|
return NewConnClient(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServer(c, conn)
|
return NewConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,16 +1,17 @@
|
|||||||
package original
|
package original
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func (c *Config) UDP() {}
|
||||||
|
|
||||||
func (c *Config) HeaderConn() {}
|
func (c *Config) HeaderConn() {}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnClient(c, conn)
|
return NewConnClient(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServer(c, conn)
|
return NewConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,14 +1,14 @@
|
|||||||
package noise
|
package noise
|
||||||
|
|
||||||
import (
|
import "net"
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) UDP() {
|
||||||
return NewConnClient(c, conn)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServer(c, conn)
|
return NewConnClient(c, raw)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
|
return NewConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,14 +1,27 @@
|
|||||||
package realm
|
package realm
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
|
"github.com/xtls/xray-core/common/errors"
|
||||||
|
"github.com/xtls/xray-core/transport/internet"
|
||||||
|
"github.com/xtls/xray-core/transport/internet/hysteria/udphop"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) UDP() {}
|
||||||
return NewConnClient(c, conn)
|
|
||||||
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
|
_, ok1 := raw.(*internet.FakePacketConn)
|
||||||
|
_, ok2 := raw.(*udphop.UdpHopPacketConn)
|
||||||
|
if level != 0 || ok1 || ok2 {
|
||||||
|
return nil, errors.New("realm requires being at the outermost level")
|
||||||
|
}
|
||||||
|
return NewConnClient(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServer(c, conn)
|
if level != 0 {
|
||||||
|
return nil, errors.New("realm requires being at the outermost level")
|
||||||
|
}
|
||||||
|
return NewConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,24 +1,27 @@
|
|||||||
package salamander
|
package salamander
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func (c *Config) UDP() {}
|
||||||
|
|
||||||
func (c *Config) HeaderConn() {}
|
func (c *Config) HeaderConn() {}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewSalamanderConnClient(c, conn)
|
return NewSalamanderConnClient(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewSalamanderConnServer(c, conn)
|
return NewSalamanderConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *GeckoConfig) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *GeckoConfig) UDP() {}
|
||||||
return NewGeckoConnClient(c, conn)
|
|
||||||
|
func (c *GeckoConfig) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
|
return NewGeckoConnClient(c, raw)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *GeckoConfig) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *GeckoConfig) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewGeckoConnServer(c, conn)
|
return NewGeckoConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,18 +1,25 @@
|
|||||||
package sudoku
|
package sudoku
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
|
"github.com/xtls/xray-core/common/errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func (c *Config) TCP() {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Config) UDP() {
|
||||||
|
}
|
||||||
|
|
||||||
// Sudoku in finalmask mode is a pure appearance transform with no standalone handshake.
|
// Sudoku in finalmask mode is a pure appearance transform with no standalone handshake.
|
||||||
// TCP always keeps classic sudoku on uplink and uses packed downlink optimization on server writes.
|
// TCP always keeps classic sudoku on uplink and uses packed downlink optimization on server writes.
|
||||||
func (c *Config) WrapConnClient(conn net.Conn, dest *net.Destination, dialer *finalmask.Dialer) (net.Conn, error) {
|
func (c *Config) WrapConnClient(raw net.Conn) (net.Conn, error) {
|
||||||
return newPackedDirectionalConn(conn, c, true)
|
return newPackedDirectionalConn(raw, c, true)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapConnServer(conn net.Conn) (net.Conn, error) {
|
func (c *Config) WrapConnServer(raw net.Conn) (net.Conn, error) {
|
||||||
return newPackedDirectionalConn(conn, c, false)
|
return newPackedDirectionalConn(raw, c, false)
|
||||||
}
|
}
|
||||||
|
|
||||||
func newPackedDirectionalConn(raw net.Conn, config *Config, readPacked bool) (net.Conn, error) {
|
func newPackedDirectionalConn(raw net.Conn, config *Config, readPacked bool) (net.Conn, error) {
|
||||||
@@ -35,10 +42,16 @@ func newPackedDirectionalConn(raw net.Conn, config *Config, readPacked bool) (ne
|
|||||||
return newWrappedConn(raw, reader, writer), nil
|
return newWrappedConn(raw, reader, writer), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewUDPConn(conn, c)
|
if level != levelCount {
|
||||||
|
return nil, errors.New("sudoku udp mask must be the innermost mask in chain")
|
||||||
|
}
|
||||||
|
return NewUDPConn(raw, c)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewUDPConn(conn, c)
|
if level != levelCount {
|
||||||
|
return nil, errors.New("sudoku udp mask must be the innermost mask in chain")
|
||||||
|
}
|
||||||
|
return NewUDPConn(raw, c)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,14 +2,12 @@ package finalmask_test
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"context"
|
|
||||||
"io"
|
"io"
|
||||||
gonet "net"
|
"net"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
"github.com/xtls/xray-core/transport/internet/finalmask"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/header/custom"
|
"github.com/xtls/xray-core/transport/internet/finalmask/header/custom"
|
||||||
)
|
)
|
||||||
@@ -22,14 +20,11 @@ func mustSendRecvTcp(
|
|||||||
) {
|
) {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
|
|
||||||
waitCh := make(chan error)
|
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
_, err := from.Write(msg)
|
_, err := from.Write(msg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Error(err)
|
||||||
}
|
}
|
||||||
close(waitCh)
|
|
||||||
}()
|
}()
|
||||||
|
|
||||||
buf := make([]byte, 1024)
|
buf := make([]byte, 1024)
|
||||||
@@ -45,23 +40,18 @@ func mustSendRecvTcp(
|
|||||||
if !bytes.Equal(buf[:n], msg) {
|
if !bytes.Equal(buf[:n], msg) {
|
||||||
t.Fatalf("unexpected data %q", buf[:n])
|
t.Fatalf("unexpected data %q", buf[:n])
|
||||||
}
|
}
|
||||||
|
|
||||||
<-waitCh
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type layerMaskTcp struct {
|
type layerMaskTcp struct {
|
||||||
name string
|
name string
|
||||||
mask finalmask.TCPMask
|
mask finalmask.Tcpmask
|
||||||
}
|
}
|
||||||
|
|
||||||
type failingWrapMask struct{}
|
type failingWrapMask struct{}
|
||||||
|
|
||||||
func (failingWrapMask) TCP() {}
|
func (failingWrapMask) TCP() {}
|
||||||
func (f failingWrapMask) WrapConnClient(conn net.Conn, dest *net.Destination, dialer *finalmask.Dialer) (net.Conn, error) {
|
func (f failingWrapMask) WrapConnClient(raw net.Conn) (net.Conn, error) { return raw, nil }
|
||||||
return conn, nil
|
func (f failingWrapMask) WrapConnServer(raw net.Conn) (net.Conn, error) {
|
||||||
}
|
|
||||||
|
|
||||||
func (f failingWrapMask) WrapConnServer(conn net.Conn) (net.Conn, error) {
|
|
||||||
return nil, io.ErrClosedPipe
|
return nil, io.ErrClosedPipe
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -102,31 +92,32 @@ func TestConnReadWrite(t *testing.T) {
|
|||||||
t.Run(c.name, func(t *testing.T) {
|
t.Run(c.name, func(t *testing.T) {
|
||||||
mask := c.mask
|
mask := c.mask
|
||||||
|
|
||||||
dialTCP := func(ctx context.Context, dest net.Destination) (net.Conn, error) {
|
maskManager := finalmask.NewTcpmaskManager([]finalmask.Tcpmask{mask})
|
||||||
return net.Dial("tcp", dest.NetAddr())
|
|
||||||
}
|
|
||||||
listen := func(ctx context.Context, addr net.Addr) (net.Listener, error) {
|
|
||||||
return net.Listen("tcp", addr.String())
|
|
||||||
}
|
|
||||||
finalMask := finalmask.NewFinalMask([]finalmask.TCPMask{mask}, nil, dialTCP, listen, nil, nil)
|
|
||||||
|
|
||||||
listener, err := finalMask.Listen(context.Background(), &net.TCPAddr{IP: net.LocalHostIP.IP()})
|
ln, err := net.Listen("tcp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
t.Cleanup(func() { listener.Close() })
|
|
||||||
|
|
||||||
client, err := finalMask.DialTCP(context.Background(), net.TCPDestination(net.IPAddress(listener.Addr().(*net.TCPAddr).IP), net.Port(listener.Addr().(*net.TCPAddr).Port)))
|
client, err := net.Dial("tcp", ln.Addr().String())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
t.Cleanup(func() { client.Close() })
|
|
||||||
|
|
||||||
server, err := listener.Accept()
|
client, err = maskManager.WrapConnClient(client)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
server, err := ln.Accept()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
server, err = maskManager.WrapConnServer(server)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
t.Cleanup(func() { server.Close() })
|
|
||||||
|
|
||||||
_ = client.SetDeadline(time.Now().Add(time.Second))
|
_ = client.SetDeadline(time.Now().Add(time.Second))
|
||||||
_ = server.SetDeadline(time.Now().Add(time.Second))
|
_ = server.SetDeadline(time.Now().Add(time.Second))
|
||||||
@@ -159,32 +150,34 @@ func TestTCPcustomStaticHandshakeRoundTrip(t *testing.T) {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
maskManager := finalmask.NewTcpmaskManager([]finalmask.Tcpmask{cfg})
|
||||||
|
|
||||||
dialTCP := func(ctx context.Context, dest net.Destination) (net.Conn, error) {
|
ln, err := net.Listen("tcp", "127.0.0.1:0")
|
||||||
return net.Dial("tcp", dest.NetAddr())
|
|
||||||
}
|
|
||||||
listen := func(ctx context.Context, addr net.Addr) (net.Listener, error) {
|
|
||||||
return net.Listen("tcp", addr.String())
|
|
||||||
}
|
|
||||||
finalMask := finalmask.NewFinalMask([]finalmask.TCPMask{cfg}, nil, dialTCP, listen, nil, nil)
|
|
||||||
|
|
||||||
listener, err := finalMask.Listen(context.Background(), &net.TCPAddr{IP: net.LocalHostIP.IP()})
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer listener.Close()
|
defer ln.Close()
|
||||||
|
|
||||||
client, err := finalMask.DialTCP(context.Background(), net.TCPDestination(net.IPAddress(listener.Addr().(*net.TCPAddr).IP), net.Port(listener.Addr().(*net.TCPAddr).Port)))
|
clientRaw, err := net.Dial("tcp", ln.Addr().String())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer client.Close()
|
defer clientRaw.Close()
|
||||||
|
|
||||||
server, err := listener.Accept()
|
serverRaw, err := ln.Accept()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer serverRaw.Close()
|
||||||
|
|
||||||
|
client, err := maskManager.WrapConnClient(clientRaw)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
server, err := maskManager.WrapConnServer(serverRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer server.Close()
|
|
||||||
|
|
||||||
_ = client.SetDeadline(time.Now().Add(time.Second))
|
_ = client.SetDeadline(time.Now().Add(time.Second))
|
||||||
_ = server.SetDeadline(time.Now().Add(time.Second))
|
_ = server.SetDeadline(time.Now().Add(time.Second))
|
||||||
@@ -227,11 +220,11 @@ func TestTCPcustomClientRejectsMismatchedServerSequence(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := clientCfg.WrapConnClient(clientRaw, nil, nil)
|
client, err := clientCfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -264,37 +257,42 @@ func TestTCPcustomClientRejectsMismatchedServerSequence(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestTCPWrapListenerRejectsImmediateWrapErrors(t *testing.T) {
|
func TestTCPWrapListenerRejectsImmediateWrapErrors(t *testing.T) {
|
||||||
dialTCP := func(ctx context.Context, dest net.Destination) (net.Conn, error) {
|
clientManager := finalmask.NewTcpmaskManager([]finalmask.Tcpmask{failingWrapMask{}})
|
||||||
return net.Dial("tcp", dest.NetAddr())
|
serverManager := finalmask.NewTcpmaskManager([]finalmask.Tcpmask{failingWrapMask{}})
|
||||||
}
|
|
||||||
listen := func(ctx context.Context, addr net.Addr) (net.Listener, error) {
|
|
||||||
return net.Listen("tcp", addr.String())
|
|
||||||
}
|
|
||||||
finalMask := finalmask.NewFinalMask([]finalmask.TCPMask{failingWrapMask{}}, nil, dialTCP, listen, nil, nil)
|
|
||||||
|
|
||||||
listener, err := finalMask.Listen(context.Background(), &net.TCPAddr{IP: net.LocalHostIP.IP()})
|
rawLn, err := net.Listen("tcp", "127.0.0.1:0")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer rawLn.Close()
|
||||||
|
|
||||||
|
ln, err := serverManager.WrapListener(rawLn)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer listener.Close()
|
|
||||||
|
|
||||||
accepted := make(chan struct {
|
accepted := make(chan struct {
|
||||||
conn net.Conn
|
conn net.Conn
|
||||||
err error
|
err error
|
||||||
}, 1)
|
}, 1)
|
||||||
go func() {
|
go func() {
|
||||||
conn, err := listener.Accept()
|
conn, err := ln.Accept()
|
||||||
accepted <- struct {
|
accepted <- struct {
|
||||||
conn net.Conn
|
conn net.Conn
|
||||||
err error
|
err error
|
||||||
}{conn: conn, err: err}
|
}{conn: conn, err: err}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
client, err := finalMask.DialTCP(context.Background(), net.TCPDestination(net.IPAddress(listener.Addr().(*net.TCPAddr).IP), net.Port(listener.Addr().(*net.TCPAddr).Port)))
|
clientRaw, err := net.Dial("tcp", rawLn.Addr().String())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer clientRaw.Close()
|
||||||
|
|
||||||
|
client, err := clientManager.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer client.Close()
|
|
||||||
|
|
||||||
_ = client.SetDeadline(time.Now().Add(time.Second))
|
_ = client.SetDeadline(time.Now().Add(time.Second))
|
||||||
|
|
||||||
|
|||||||
@@ -2,15 +2,13 @@ package finalmask_test
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"context"
|
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
"io"
|
"io"
|
||||||
gonet "net"
|
"net"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/proxy"
|
"github.com/xtls/xray-core/proxy"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
"github.com/xtls/xray-core/transport/internet/finalmask"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask/header/custom"
|
"github.com/xtls/xray-core/transport/internet/finalmask/header/custom"
|
||||||
@@ -53,7 +51,7 @@ func mustSendRecv(
|
|||||||
|
|
||||||
type layerMask struct {
|
type layerMask struct {
|
||||||
name string
|
name string
|
||||||
mask finalmask.UDPMask
|
mask finalmask.Udpmask
|
||||||
layers int
|
layers int
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -215,23 +213,25 @@ func newStandaloneStunLikeUDPServerConfig() *custom.UDPStandaloneConfig {
|
|||||||
func newUDPClientServerPair(t *testing.T, cfg *custom.UDPStandaloneConfig) (net.PacketConn, net.PacketConn, net.PacketConn, net.PacketConn) {
|
func newUDPClientServerPair(t *testing.T, cfg *custom.UDPStandaloneConfig) (net.PacketConn, net.PacketConn, net.PacketConn, net.PacketConn) {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
|
|
||||||
clientRaw, err := gonet.ListenPacket("udp", "127.0.0.1:0")
|
clientRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
t.Cleanup(func() { _ = clientRaw.Close() })
|
t.Cleanup(func() { _ = clientRaw.Close() })
|
||||||
|
|
||||||
serverRaw, err := gonet.ListenPacket("udp", "127.0.0.1:0")
|
serverRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
t.Cleanup(func() { _ = serverRaw.Close() })
|
t.Cleanup(func() { _ = serverRaw.Close() })
|
||||||
|
|
||||||
client, err := cfg.WrapPacketConnClient(clientRaw, nil, nil)
|
maskManager := finalmask.NewUdpmaskManager([]finalmask.Udpmask{cfg})
|
||||||
|
|
||||||
|
client, err := maskManager.WrapPacketConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
server, err := cfg.WrapPacketConnServer(serverRaw, nil, nil)
|
server, err := maskManager.WrapPacketConnServer(serverRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -348,39 +348,31 @@ func TestPacketConnReadWrite(t *testing.T) {
|
|||||||
if layers <= 0 {
|
if layers <= 0 {
|
||||||
layers = 1
|
layers = 1
|
||||||
}
|
}
|
||||||
masks := make([]finalmask.UDPMask, 0, layers)
|
masks := make([]finalmask.Udpmask, 0, layers)
|
||||||
for i := 0; i < layers; i++ {
|
for i := 0; i < layers; i++ {
|
||||||
masks = append(masks, mask)
|
masks = append(masks, mask)
|
||||||
}
|
}
|
||||||
|
maskManager := finalmask.NewUdpmaskManager(masks)
|
||||||
|
|
||||||
dialUDP := func(ctx context.Context, dest net.Destination) (net.PacketConn, net.Addr, error) {
|
client, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
udpAddr, err := net.ResolveUDPAddr("udp", dest.NetAddr())
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
conn, err := gonet.ListenPacket("udp", "127.0.0.1:0")
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
return conn, udpAddr, nil
|
|
||||||
}
|
|
||||||
listenPacket := func(ctx context.Context, addr net.Addr) (net.PacketConn, error) {
|
|
||||||
return gonet.ListenPacket(addr.Network(), addr.String())
|
|
||||||
}
|
|
||||||
finalMask := finalmask.NewFinalMask(nil, masks, nil, nil, dialUDP, listenPacket)
|
|
||||||
|
|
||||||
server, err := finalMask.ListenPacket(context.Background(), &net.UDPAddr{IP: net.LocalHostIP.IP()})
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
t.Cleanup(func() { server.Close() })
|
|
||||||
|
|
||||||
clientConn, err := finalMask.DialUDP(context.Background(), net.UDPDestination(net.IPAddress(server.LocalAddr().(*net.UDPAddr).IP), net.Port(server.LocalAddr().(*net.UDPAddr).Port)))
|
client, err = maskManager.WrapPacketConnClient(client)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
server, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
server, err = maskManager.WrapPacketConnServer(server)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
t.Cleanup(func() { clientConn.Close() })
|
|
||||||
client := clientConn.(*finalmask.PacketConnWrapper).PacketConn
|
|
||||||
|
|
||||||
_ = client.SetDeadline(time.Now().Add(time.Second))
|
_ = client.SetDeadline(time.Now().Add(time.Second))
|
||||||
_ = server.SetDeadline(time.Now().Add(time.Second))
|
_ = server.SetDeadline(time.Now().Add(time.Second))
|
||||||
@@ -405,20 +397,21 @@ func TestUDPcustomStaticHeaderWireShape(t *testing.T) {
|
|||||||
{Rand: 1, RandMin: 0x30, RandMax: 0x40},
|
{Rand: 1, RandMin: 0x30, RandMax: 0x40},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
maskManager := finalmask.NewUdpmaskManager([]finalmask.Udpmask{cfg})
|
||||||
|
|
||||||
clientRaw, err := gonet.ListenPacket("udp", "127.0.0.1:0")
|
clientRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
|
|
||||||
serverRaw, err := gonet.ListenPacket("udp", "127.0.0.1:0")
|
serverRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := cfg.WrapPacketConnClient(clientRaw, nil, nil)
|
client, err := maskManager.WrapPacketConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -649,11 +642,11 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
Ascii: "prefer_ascii",
|
Ascii: "prefer_ascii",
|
||||||
}
|
}
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -690,11 +683,11 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
PaddingMax: 0,
|
PaddingMax: 0,
|
||||||
}
|
}
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -745,10 +738,10 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
countWireBytes := func(wrapServer func(net.Conn, *sudoku.Config) (net.Conn, error), cfg *sudoku.Config) int64 {
|
countWireBytes := func(wrapServer func(net.Conn, *sudoku.Config) (net.Conn, error), cfg *sudoku.Config) int64 {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
watchedServerRaw := &countingConn{Conn: serverRaw}
|
watchedServerRaw := &countingConn{Conn: serverRaw}
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -800,11 +793,11 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
CustomTables: []string{"xpxvvpvv", "vxpvxvvp"},
|
CustomTables: []string{"xpxvvpvv", "vxpvxvvp"},
|
||||||
}
|
}
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -842,11 +835,11 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
PaddingMax: 0,
|
PaddingMax: 0,
|
||||||
}
|
}
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -875,6 +868,19 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
t.Run("GivenSudokuUDPMask_WhenNotInnermost_ThenWrapFails", func(t *testing.T) {
|
||||||
|
cfg := &sudoku.Config{Password: "sudoku-udp"}
|
||||||
|
raw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer raw.Close()
|
||||||
|
|
||||||
|
if _, err := cfg.WrapPacketConnClient(raw, 0, 1); err == nil {
|
||||||
|
t.Fatal("expected innermost check failure")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
t.Run("GivenSudokuMultiTableUDPMask_WhenClientSendsMultipleDatagrams_ThenPayloadMatches", func(t *testing.T) {
|
t.Run("GivenSudokuMultiTableUDPMask_WhenClientSendsMultipleDatagrams_ThenPayloadMatches", func(t *testing.T) {
|
||||||
cfg := &sudoku.Config{
|
cfg := &sudoku.Config{
|
||||||
Password: "sudoku-udp-multi",
|
Password: "sudoku-udp-multi",
|
||||||
@@ -883,24 +889,25 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
PaddingMin: 0,
|
PaddingMin: 0,
|
||||||
PaddingMax: 0,
|
PaddingMax: 0,
|
||||||
}
|
}
|
||||||
|
maskManager := finalmask.NewUdpmaskManager([]finalmask.Udpmask{cfg})
|
||||||
|
|
||||||
clientRaw, err := gonet.ListenPacket("udp", "127.0.0.1:0")
|
clientRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
|
|
||||||
serverRaw, err := gonet.ListenPacket("udp", "127.0.0.1:0")
|
serverRaw, err := net.ListenPacket("udp", "127.0.0.1:0")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
client, err := cfg.WrapPacketConnClient(clientRaw, nil, nil)
|
client, err := maskManager.WrapPacketConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
server, err := cfg.WrapPacketConnServer(serverRaw, nil, nil)
|
server, err := maskManager.WrapPacketConnServer(serverRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -954,7 +961,7 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
}
|
}
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -1001,11 +1008,11 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
Ascii: "prefer_entropy",
|
Ascii: "prefer_entropy",
|
||||||
}
|
}
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -1025,11 +1032,11 @@ func TestSudokuBDD(t *testing.T) {
|
|||||||
Ascii: "prefer_entropy",
|
Ascii: "prefer_entropy",
|
||||||
}
|
}
|
||||||
|
|
||||||
clientRaw, serverRaw := gonet.Pipe()
|
clientRaw, serverRaw := net.Pipe()
|
||||||
defer clientRaw.Close()
|
defer clientRaw.Close()
|
||||||
defer serverRaw.Close()
|
defer serverRaw.Close()
|
||||||
|
|
||||||
clientConn, err := cfg.WrapConnClient(clientRaw, nil, nil)
|
clientConn, err := cfg.WrapConnClient(clientRaw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,17 +0,0 @@
|
|||||||
package udphop
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (c *Config) HandleDial() {}
|
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
|
||||||
return NewUDPHopConn(c, dest, dialer)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
|
||||||
return nil, errors.New("udphop: client only")
|
|
||||||
}
|
|
||||||
@@ -1,178 +0,0 @@
|
|||||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
|
||||||
// versions:
|
|
||||||
// protoc-gen-go v1.36.11
|
|
||||||
// protoc v6.33.5
|
|
||||||
// source: transport/internet/finalmask/udphop/config.proto
|
|
||||||
|
|
||||||
package udphop
|
|
||||||
|
|
||||||
import (
|
|
||||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
|
||||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
|
||||||
reflect "reflect"
|
|
||||||
sync "sync"
|
|
||||||
unsafe "unsafe"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
// Verify that this generated code is sufficiently up-to-date.
|
|
||||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
|
||||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
|
||||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
|
||||||
)
|
|
||||||
|
|
||||||
type Config struct {
|
|
||||||
state protoimpl.MessageState `protogen:"open.v1"`
|
|
||||||
Local bool `protobuf:"varint,2,opt,name=local,proto3" json:"local,omitempty"`
|
|
||||||
Remote bool `protobuf:"varint,3,opt,name=remote,proto3" json:"remote,omitempty"`
|
|
||||||
RemoteOnce bool `protobuf:"varint,4,opt,name=remote_once,json=remoteOnce,proto3" json:"remote_once,omitempty"`
|
|
||||||
IntervalMin int64 `protobuf:"varint,5,opt,name=interval_min,json=intervalMin,proto3" json:"interval_min,omitempty"`
|
|
||||||
IntervalMax int64 `protobuf:"varint,6,opt,name=interval_max,json=intervalMax,proto3" json:"interval_max,omitempty"`
|
|
||||||
RemoteIPs []string `protobuf:"bytes,7,rep,name=remoteIPs,proto3" json:"remoteIPs,omitempty"`
|
|
||||||
RemotePorts []uint32 `protobuf:"varint,8,rep,packed,name=remote_ports,json=remotePorts,proto3" json:"remote_ports,omitempty"`
|
|
||||||
unknownFields protoimpl.UnknownFields
|
|
||||||
sizeCache protoimpl.SizeCache
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) Reset() {
|
|
||||||
*x = Config{}
|
|
||||||
mi := &file_transport_internet_finalmask_udphop_config_proto_msgTypes[0]
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) String() string {
|
|
||||||
return protoimpl.X.MessageStringOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (*Config) ProtoMessage() {}
|
|
||||||
|
|
||||||
func (x *Config) ProtoReflect() protoreflect.Message {
|
|
||||||
mi := &file_transport_internet_finalmask_udphop_config_proto_msgTypes[0]
|
|
||||||
if x != nil {
|
|
||||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
|
||||||
if ms.LoadMessageInfo() == nil {
|
|
||||||
ms.StoreMessageInfo(mi)
|
|
||||||
}
|
|
||||||
return ms
|
|
||||||
}
|
|
||||||
return mi.MessageOf(x)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Deprecated: Use Config.ProtoReflect.Descriptor instead.
|
|
||||||
func (*Config) Descriptor() ([]byte, []int) {
|
|
||||||
return file_transport_internet_finalmask_udphop_config_proto_rawDescGZIP(), []int{0}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) GetLocal() bool {
|
|
||||||
if x != nil {
|
|
||||||
return x.Local
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) GetRemote() bool {
|
|
||||||
if x != nil {
|
|
||||||
return x.Remote
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) GetRemoteOnce() bool {
|
|
||||||
if x != nil {
|
|
||||||
return x.RemoteOnce
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) GetIntervalMin() int64 {
|
|
||||||
if x != nil {
|
|
||||||
return x.IntervalMin
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) GetIntervalMax() int64 {
|
|
||||||
if x != nil {
|
|
||||||
return x.IntervalMax
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) GetRemoteIPs() []string {
|
|
||||||
if x != nil {
|
|
||||||
return x.RemoteIPs
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (x *Config) GetRemotePorts() []uint32 {
|
|
||||||
if x != nil {
|
|
||||||
return x.RemotePorts
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var File_transport_internet_finalmask_udphop_config_proto protoreflect.FileDescriptor
|
|
||||||
|
|
||||||
const file_transport_internet_finalmask_udphop_config_proto_rawDesc = "" +
|
|
||||||
"\n" +
|
|
||||||
"0transport/internet/finalmask/udphop/config.proto\x12(xray.transport.internet.finalmask.udphop\"\xe4\x01\n" +
|
|
||||||
"\x06Config\x12\x14\n" +
|
|
||||||
"\x05local\x18\x02 \x01(\bR\x05local\x12\x16\n" +
|
|
||||||
"\x06remote\x18\x03 \x01(\bR\x06remote\x12\x1f\n" +
|
|
||||||
"\vremote_once\x18\x04 \x01(\bR\n" +
|
|
||||||
"remoteOnce\x12!\n" +
|
|
||||||
"\finterval_min\x18\x05 \x01(\x03R\vintervalMin\x12!\n" +
|
|
||||||
"\finterval_max\x18\x06 \x01(\x03R\vintervalMax\x12\x1c\n" +
|
|
||||||
"\tremoteIPs\x18\a \x03(\tR\tremoteIPs\x12!\n" +
|
|
||||||
"\fremote_ports\x18\b \x03(\rR\vremotePortsJ\x04\b\x01\x10\x02B\x9a\x01\n" +
|
|
||||||
",com.xray.transport.internet.finalmask.udphopP\x01Z=github.com/xtls/xray-core/transport/internet/finalmask/udphop\xaa\x02(Xray.Transport.Internet.Finalmask.Udphopb\x06proto3"
|
|
||||||
|
|
||||||
var (
|
|
||||||
file_transport_internet_finalmask_udphop_config_proto_rawDescOnce sync.Once
|
|
||||||
file_transport_internet_finalmask_udphop_config_proto_rawDescData []byte
|
|
||||||
)
|
|
||||||
|
|
||||||
func file_transport_internet_finalmask_udphop_config_proto_rawDescGZIP() []byte {
|
|
||||||
file_transport_internet_finalmask_udphop_config_proto_rawDescOnce.Do(func() {
|
|
||||||
file_transport_internet_finalmask_udphop_config_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_transport_internet_finalmask_udphop_config_proto_rawDesc), len(file_transport_internet_finalmask_udphop_config_proto_rawDesc)))
|
|
||||||
})
|
|
||||||
return file_transport_internet_finalmask_udphop_config_proto_rawDescData
|
|
||||||
}
|
|
||||||
|
|
||||||
var file_transport_internet_finalmask_udphop_config_proto_msgTypes = make([]protoimpl.MessageInfo, 1)
|
|
||||||
var file_transport_internet_finalmask_udphop_config_proto_goTypes = []any{
|
|
||||||
(*Config)(nil), // 0: xray.transport.internet.finalmask.udphop.Config
|
|
||||||
}
|
|
||||||
var file_transport_internet_finalmask_udphop_config_proto_depIdxs = []int32{
|
|
||||||
0, // [0:0] is the sub-list for method output_type
|
|
||||||
0, // [0:0] is the sub-list for method input_type
|
|
||||||
0, // [0:0] is the sub-list for extension type_name
|
|
||||||
0, // [0:0] is the sub-list for extension extendee
|
|
||||||
0, // [0:0] is the sub-list for field type_name
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() { file_transport_internet_finalmask_udphop_config_proto_init() }
|
|
||||||
func file_transport_internet_finalmask_udphop_config_proto_init() {
|
|
||||||
if File_transport_internet_finalmask_udphop_config_proto != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
type x struct{}
|
|
||||||
out := protoimpl.TypeBuilder{
|
|
||||||
File: protoimpl.DescBuilder{
|
|
||||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
|
||||||
RawDescriptor: unsafe.Slice(unsafe.StringData(file_transport_internet_finalmask_udphop_config_proto_rawDesc), len(file_transport_internet_finalmask_udphop_config_proto_rawDesc)),
|
|
||||||
NumEnums: 0,
|
|
||||||
NumMessages: 1,
|
|
||||||
NumExtensions: 0,
|
|
||||||
NumServices: 0,
|
|
||||||
},
|
|
||||||
GoTypes: file_transport_internet_finalmask_udphop_config_proto_goTypes,
|
|
||||||
DependencyIndexes: file_transport_internet_finalmask_udphop_config_proto_depIdxs,
|
|
||||||
MessageInfos: file_transport_internet_finalmask_udphop_config_proto_msgTypes,
|
|
||||||
}.Build()
|
|
||||||
File_transport_internet_finalmask_udphop_config_proto = out.File
|
|
||||||
file_transport_internet_finalmask_udphop_config_proto_goTypes = nil
|
|
||||||
file_transport_internet_finalmask_udphop_config_proto_depIdxs = nil
|
|
||||||
}
|
|
||||||
@@ -1,19 +0,0 @@
|
|||||||
syntax = "proto3";
|
|
||||||
|
|
||||||
package xray.transport.internet.finalmask.udphop;
|
|
||||||
option csharp_namespace = "Xray.Transport.Internet.Finalmask.Udphop";
|
|
||||||
option go_package = "github.com/xtls/xray-core/transport/internet/finalmask/udphop";
|
|
||||||
option java_package = "com.xray.transport.internet.finalmask.udphop";
|
|
||||||
option java_multiple_files = true;
|
|
||||||
|
|
||||||
message Config {
|
|
||||||
reserved 1;
|
|
||||||
bool local = 2;
|
|
||||||
bool remote = 3;
|
|
||||||
bool remote_once = 4;
|
|
||||||
int64 interval_min = 5;
|
|
||||||
int64 interval_max = 6;
|
|
||||||
repeated string remoteIPs = 7;
|
|
||||||
repeated uint32 remote_ports = 8;
|
|
||||||
}
|
|
||||||
|
|
||||||
@@ -1,289 +0,0 @@
|
|||||||
package udphop
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"crypto/rand"
|
|
||||||
goerrors "errors"
|
|
||||||
"io"
|
|
||||||
mrand "math/rand"
|
|
||||||
"net/netip"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/xtls/xray-core/common"
|
|
||||||
"github.com/xtls/xray-core/common/crypto"
|
|
||||||
"github.com/xtls/xray-core/common/errors"
|
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
|
||||||
|
|
||||||
var pool = sync.Pool{
|
|
||||||
New: func() any {
|
|
||||||
return make([]byte, finalmask.UDPSize)
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
type packet struct {
|
|
||||||
p []byte
|
|
||||||
addr net.Addr
|
|
||||||
err error
|
|
||||||
}
|
|
||||||
|
|
||||||
type udpHopConn struct {
|
|
||||||
dialer *finalmask.Dialer
|
|
||||||
local bool
|
|
||||||
remote bool
|
|
||||||
|
|
||||||
intervalMin int64
|
|
||||||
intervalMax int64
|
|
||||||
remoteIPs []netip.Prefix
|
|
||||||
remotePorts []uint32
|
|
||||||
|
|
||||||
deadline time.Time
|
|
||||||
readDeadline time.Time
|
|
||||||
writeDeadline time.Time
|
|
||||||
|
|
||||||
pre net.PacketConn
|
|
||||||
cur net.PacketConn
|
|
||||||
addr *net.UDPAddr
|
|
||||||
readCh chan packet
|
|
||||||
closeCh chan struct{}
|
|
||||||
wg sync.WaitGroup
|
|
||||||
mu sync.RWMutex
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewUDPHopConn(c *Config, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
|
||||||
if c.IntervalMin < 5 || c.IntervalMax < 5 {
|
|
||||||
return nil, errors.New("invalid interval")
|
|
||||||
}
|
|
||||||
remoteIPs := make([]netip.Prefix, 0, len(c.RemoteIPs))
|
|
||||||
for _, ip := range c.RemoteIPs {
|
|
||||||
remoteIPs = append(remoteIPs, netip.MustParsePrefix(ip))
|
|
||||||
}
|
|
||||||
remotePorts := c.RemotePorts
|
|
||||||
if c.Remote || c.RemoteOnce {
|
|
||||||
if len(remoteIPs) > 0 {
|
|
||||||
dest.Address = net.IPAddress(randPrefix(remoteIPs[mrand.Intn(len(remoteIPs))]))
|
|
||||||
}
|
|
||||||
if len(remotePorts) > 0 {
|
|
||||||
dest.Port = net.Port(remotePorts[mrand.Intn(len(remotePorts))])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
conn, err := dialer.DialUDP(*dest)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
cur := conn.(*finalmask.PacketConnWrapper).PacketConn
|
|
||||||
addr := conn.RemoteAddr().(*net.UDPAddr)
|
|
||||||
client := &udpHopConn{
|
|
||||||
dialer: dialer,
|
|
||||||
local: c.Local,
|
|
||||||
remote: c.Remote,
|
|
||||||
|
|
||||||
intervalMin: c.IntervalMin,
|
|
||||||
intervalMax: c.IntervalMax,
|
|
||||||
remoteIPs: remoteIPs,
|
|
||||||
remotePorts: remotePorts,
|
|
||||||
|
|
||||||
cur: cur,
|
|
||||||
addr: addr,
|
|
||||||
readCh: make(chan packet),
|
|
||||||
closeCh: make(chan struct{}),
|
|
||||||
}
|
|
||||||
go client.run()
|
|
||||||
client.wg.Add(1)
|
|
||||||
go client.recv(client.cur)
|
|
||||||
return client, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) closed() bool {
|
|
||||||
select {
|
|
||||||
case <-c.closeCh:
|
|
||||||
return true
|
|
||||||
default:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) run() {
|
|
||||||
ticker := time.NewTicker(time.Second * time.Duration(crypto.RandBetween(c.intervalMin, c.intervalMax+1)))
|
|
||||||
defer ticker.Stop()
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-c.closeCh:
|
|
||||||
return
|
|
||||||
case <-ticker.C:
|
|
||||||
ticker.Reset(time.Second * time.Duration(crypto.RandBetween(c.intervalMin, c.intervalMax+1)))
|
|
||||||
c.hop()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) hop() {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
if c.closed() {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
oldIP := c.addr.IP
|
|
||||||
oldPort := c.addr.Port
|
|
||||||
if c.remote {
|
|
||||||
if len(c.remoteIPs) > 0 {
|
|
||||||
c.addr.IP = randPrefix(c.remoteIPs[mrand.Intn(len(c.remoteIPs))])
|
|
||||||
}
|
|
||||||
if len(c.remotePorts) > 0 {
|
|
||||||
c.addr.Port = int(c.remotePorts[mrand.Intn(len(c.remotePorts))])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if c.local {
|
|
||||||
conn, err := c.dialer.DialUDP(net.UDPDestination(net.IPAddress(c.addr.IP), net.Port(c.addr.Port)))
|
|
||||||
if err != nil {
|
|
||||||
c.addr.IP = oldIP
|
|
||||||
c.addr.Port = oldPort
|
|
||||||
errors.LogErrorInner(context.Background(), err, "hop err")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
conn.SetDeadline(c.deadline)
|
|
||||||
conn.SetReadDeadline(c.readDeadline)
|
|
||||||
conn.SetWriteDeadline(c.writeDeadline)
|
|
||||||
if c.pre != nil {
|
|
||||||
_ = c.pre.Close()
|
|
||||||
}
|
|
||||||
c.pre = c.cur
|
|
||||||
c.cur = conn.(*finalmask.PacketConnWrapper).PacketConn
|
|
||||||
c.wg.Add(1)
|
|
||||||
go c.recv(c.cur)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) recv(conn net.PacketConn) {
|
|
||||||
defer c.wg.Done()
|
|
||||||
|
|
||||||
for {
|
|
||||||
p := pool.Get().([]byte)
|
|
||||||
n, addr, err := conn.ReadFrom(p)
|
|
||||||
if err != nil {
|
|
||||||
pool.Put(p[:cap(p)])
|
|
||||||
if c.closed() {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
var netErr net.Error
|
|
||||||
if goerrors.As(err, &netErr) && netErr.Timeout() {
|
|
||||||
select {
|
|
||||||
case c.readCh <- packet{err: err}:
|
|
||||||
case <-c.closeCh:
|
|
||||||
return
|
|
||||||
}
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
errors.LogErrorInner(context.Background(), err, "recv err")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
select {
|
|
||||||
case c.readCh <- packet{p: p[:n], addr: addr}:
|
|
||||||
case <-c.closeCh:
|
|
||||||
pool.Put(p[:cap(p)])
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
|
|
||||||
packet, ok := <-c.readCh
|
|
||||||
if ok {
|
|
||||||
if packet.p != nil {
|
|
||||||
n = copy(p, packet.p)
|
|
||||||
pool.Put(packet.p[:cap(packet.p)])
|
|
||||||
}
|
|
||||||
return n, packet.addr, packet.err
|
|
||||||
}
|
|
||||||
return 0, nil, io.ErrClosedPipe
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) WriteTo(p []byte, addr net.Addr) (n int, err error) {
|
|
||||||
c.mu.RLock()
|
|
||||||
defer c.mu.RUnlock()
|
|
||||||
_, err = c.cur.WriteTo(p, c.addr)
|
|
||||||
if err != nil {
|
|
||||||
errors.LogErrorInner(context.Background(), err, "send err")
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
return len(p), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) Close() error {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
if c.closed() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
close(c.closeCh)
|
|
||||||
if c.pre != nil {
|
|
||||||
_ = c.pre.Close()
|
|
||||||
}
|
|
||||||
_ = c.cur.Close()
|
|
||||||
c.wg.Wait()
|
|
||||||
select {
|
|
||||||
case packet := <-c.readCh:
|
|
||||||
if packet.p != nil {
|
|
||||||
pool.Put(packet.p[:cap(packet.p)])
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
close(c.readCh)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) LocalAddr() net.Addr {
|
|
||||||
c.mu.RLock()
|
|
||||||
defer c.mu.RUnlock()
|
|
||||||
return c.cur.LocalAddr()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) SetDeadline(t time.Time) error {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
c.deadline = t
|
|
||||||
if c.pre != nil {
|
|
||||||
_ = c.pre.SetDeadline(t)
|
|
||||||
}
|
|
||||||
return c.cur.SetDeadline(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) SetReadDeadline(t time.Time) error {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
c.readDeadline = t
|
|
||||||
if c.pre != nil {
|
|
||||||
_ = c.pre.SetReadDeadline(t)
|
|
||||||
}
|
|
||||||
return c.cur.SetReadDeadline(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *udpHopConn) SetWriteDeadline(t time.Time) error {
|
|
||||||
c.mu.Lock()
|
|
||||||
defer c.mu.Unlock()
|
|
||||||
c.writeDeadline = t
|
|
||||||
if c.pre != nil {
|
|
||||||
_ = c.pre.SetWriteDeadline(t)
|
|
||||||
}
|
|
||||||
return c.cur.SetWriteDeadline(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func randPrefix(p netip.Prefix) []byte {
|
|
||||||
if p.IsSingleIP() {
|
|
||||||
return p.Addr().AsSlice()
|
|
||||||
}
|
|
||||||
b := p.Addr().AsSlice()
|
|
||||||
prefix := p.Bits()
|
|
||||||
var new [16]byte
|
|
||||||
common.Must2(rand.Read(new[:len(b)]))
|
|
||||||
i := prefix / 8
|
|
||||||
j := prefix % 8
|
|
||||||
if i+1 < len(b) {
|
|
||||||
copy(b[i+1:], new[i+1:])
|
|
||||||
}
|
|
||||||
mask := byte(0xff << (8 - j))
|
|
||||||
b[i] = (b[i] & mask) | (new[i] &^ mask)
|
|
||||||
return b
|
|
||||||
}
|
|
||||||
@@ -1,14 +1,24 @@
|
|||||||
package xdns
|
package xdns
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/xtls/xray-core/common/net"
|
"net"
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnClient(conn net.PacketConn, dest *net.Destination, dialer *finalmask.Dialer) (net.PacketConn, error) {
|
func (c *Config) UDP() {
|
||||||
return NewConnClient(c, conn)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Config) WrapPacketConnServer(conn net.PacketConn, addr net.Addr, lc *finalmask.ListenConfig) (net.PacketConn, error) {
|
func (c *Config) WrapPacketConnClient(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
return NewConnServer(c, conn)
|
// _, ok1 := raw.(*internet.FakePacketConn)
|
||||||
|
// _, ok2 := raw.(*udphop.UdpHopPacketConn)
|
||||||
|
// if level != 0 || ok1 || ok2 {
|
||||||
|
// return nil, errors.New("xdns requires being at the outermost level")
|
||||||
|
// }
|
||||||
|
return NewConnClient(c, raw)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Config) WrapPacketConnServer(raw net.PacketConn, level int, levelCount int) (net.PacketConn, error) {
|
||||||
|
// if level != 0 {
|
||||||
|
// return nil, errors.New("xdns requires being at the outermost level")
|
||||||
|
// }
|
||||||
|
return NewConnServer(c, raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,7 +8,8 @@ import (
|
|||||||
goerrors "errors"
|
goerrors "errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
mrand "math/rand"
|
mathrand "math/rand"
|
||||||
|
"net"
|
||||||
"net/netip"
|
"net/netip"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
@@ -16,7 +17,6 @@ import (
|
|||||||
|
|
||||||
"github.com/xtls/xray-core/common"
|
"github.com/xtls/xray-core/common"
|
||||||
"github.com/xtls/xray-core/common/errors"
|
"github.com/xtls/xray-core/common/errors"
|
||||||
"github.com/xtls/xray-core/common/net"
|
|
||||||
"github.com/xtls/xray-core/transport/internet/finalmask"
|
"github.com/xtls/xray-core/transport/internet/finalmask"
|
||||||
"golang.org/x/net/icmp"
|
"golang.org/x/net/icmp"
|
||||||
"golang.org/x/net/ipv4"
|
"golang.org/x/net/ipv4"
|
||||||
@@ -36,21 +36,20 @@ type packet struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type xicmpConnClient struct {
|
type xicmpConnClient struct {
|
||||||
|
conn net.PacketConn
|
||||||
icmp4 *icmp.PacketConn
|
icmp4 *icmp.PacketConn
|
||||||
icmp6 *icmp.PacketConn
|
icmp6 *icmp.PacketConn
|
||||||
udp bool
|
udp bool
|
||||||
ips []netip.Addr
|
ips []netip.Addr
|
||||||
ip net.IP
|
|
||||||
clientID [8]byte
|
clientID [8]byte
|
||||||
id int
|
id int
|
||||||
seq int
|
seq int
|
||||||
readCh chan packet
|
readCh chan packet
|
||||||
closeCh chan struct{}
|
closedCh chan struct{}
|
||||||
wg sync.WaitGroup
|
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewConnClient(c *Config, dest *net.Destination) (net.PacketConn, error) {
|
func NewConnClient(c *Config, raw net.PacketConn) (net.PacketConn, error) {
|
||||||
var icmp4, icmp6 *icmp.PacketConn
|
var icmp4, icmp6 *icmp.PacketConn
|
||||||
var err4, err6 error
|
var err4, err6 error
|
||||||
if c.DGRAM {
|
if c.DGRAM {
|
||||||
@@ -69,39 +68,35 @@ func NewConnClient(c *Config, dest *net.Destination) (net.PacketConn, error) {
|
|||||||
ips = append(ips, netip.MustParseAddr(ip))
|
ips = append(ips, netip.MustParseAddr(ip))
|
||||||
}
|
}
|
||||||
|
|
||||||
var ip net.IP
|
|
||||||
if len(ips) > 0 {
|
|
||||||
ip = ips[mrand.Intn(len(ips))].AsSlice()
|
|
||||||
} else {
|
|
||||||
ip = dest.Address.IP()
|
|
||||||
}
|
|
||||||
|
|
||||||
var clientID [8]byte
|
var clientID [8]byte
|
||||||
common.Must2(rand.Read(clientID[:]))
|
common.Must2(rand.Read(clientID[:]))
|
||||||
|
|
||||||
conn := &xicmpConnClient{
|
conn := &xicmpConnClient{
|
||||||
|
conn: raw,
|
||||||
icmp4: icmp4,
|
icmp4: icmp4,
|
||||||
icmp6: icmp6,
|
icmp6: icmp6,
|
||||||
udp: c.DGRAM,
|
udp: c.DGRAM,
|
||||||
ips: ips,
|
ips: ips,
|
||||||
ip: ip,
|
|
||||||
clientID: clientID,
|
clientID: clientID,
|
||||||
id: mrand.Intn(65536),
|
id: mathrand.Intn(65536),
|
||||||
seq: 1,
|
seq: 1,
|
||||||
readCh: make(chan packet),
|
readCh: make(chan packet),
|
||||||
closeCh: make(chan struct{}),
|
closedCh: make(chan struct{}),
|
||||||
}
|
}
|
||||||
|
|
||||||
conn.wg.Add(2)
|
|
||||||
go conn.recv4()
|
go conn.recv4()
|
||||||
go conn.recv6()
|
go conn.recv6()
|
||||||
|
|
||||||
return conn, nil
|
return conn, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *xicmpConnClient) ring(a, b uint16) uint16 {
|
||||||
|
return min(a-b, b-a)
|
||||||
|
}
|
||||||
|
|
||||||
func (c *xicmpConnClient) closed() bool {
|
func (c *xicmpConnClient) closed() bool {
|
||||||
select {
|
select {
|
||||||
case <-c.closeCh:
|
case <-c.closedCh:
|
||||||
return true
|
return true
|
||||||
default:
|
default:
|
||||||
return false
|
return false
|
||||||
@@ -109,28 +104,26 @@ func (c *xicmpConnClient) closed() bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *xicmpConnClient) recv4() {
|
func (c *xicmpConnClient) recv4() {
|
||||||
defer c.wg.Done()
|
|
||||||
|
|
||||||
var b [finalmask.UDPSize]byte
|
var b [finalmask.UDPSize]byte
|
||||||
|
|
||||||
for {
|
for {
|
||||||
|
if c.closed() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
n, addr, err := c.icmp4.ReadFrom(b[:])
|
n, addr, err := c.icmp4.ReadFrom(b[:])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if c.closed() {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
var netErr net.Error
|
var netErr net.Error
|
||||||
if goerrors.As(err, &netErr) && netErr.Timeout() {
|
if goerrors.As(err, &netErr) && netErr.Timeout() {
|
||||||
select {
|
select {
|
||||||
case c.readCh <- packet{
|
case c.readCh <- packet{
|
||||||
err: err,
|
err: err,
|
||||||
}:
|
}:
|
||||||
case <-c.closeCh:
|
case <-c.closedCh:
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
errors.LogErrorInner(context.Background(), err, "recv err 4")
|
continue
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
msg, err := icmp.ParseMessage(1, b[:n])
|
msg, err := icmp.ParseMessage(1, b[:n])
|
||||||
@@ -153,6 +146,10 @@ func (c *xicmpConnClient) recv4() {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if c.ring(uint16(echo.Seq), uint16(c.seq)) > 1000 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
if len(echo.Data) > 8 && bytes.Equal(echo.Data[:8], c.clientID[:]) {
|
if len(echo.Data) > 8 && bytes.Equal(echo.Data[:8], c.clientID[:]) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -169,7 +166,7 @@ func (c *xicmpConnClient) recv4() {
|
|||||||
p: p,
|
p: p,
|
||||||
addr: addr,
|
addr: addr,
|
||||||
}:
|
}:
|
||||||
case <-c.closeCh:
|
case <-c.closedCh:
|
||||||
pool.Put(p)
|
pool.Put(p)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -177,28 +174,26 @@ func (c *xicmpConnClient) recv4() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *xicmpConnClient) recv6() {
|
func (c *xicmpConnClient) recv6() {
|
||||||
defer c.wg.Done()
|
|
||||||
|
|
||||||
var b [finalmask.UDPSize]byte
|
var b [finalmask.UDPSize]byte
|
||||||
|
|
||||||
for {
|
for {
|
||||||
|
if c.closed() {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
n, addr, err := c.icmp6.ReadFrom(b[:])
|
n, addr, err := c.icmp6.ReadFrom(b[:])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if c.closed() {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
var netErr net.Error
|
var netErr net.Error
|
||||||
if goerrors.As(err, &netErr) && netErr.Timeout() {
|
if goerrors.As(err, &netErr) && netErr.Timeout() {
|
||||||
select {
|
select {
|
||||||
case c.readCh <- packet{
|
case c.readCh <- packet{
|
||||||
err: err,
|
err: err,
|
||||||
}:
|
}:
|
||||||
case <-c.closeCh:
|
case <-c.closedCh:
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
errors.LogErrorInner(context.Background(), err, "recv err 6")
|
continue
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
msg, err := icmp.ParseMessage(58, b[:n])
|
msg, err := icmp.ParseMessage(58, b[:n])
|
||||||
@@ -221,6 +216,10 @@ func (c *xicmpConnClient) recv6() {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if c.ring(uint16(echo.Seq), uint16(c.seq)) > 1000 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
if len(echo.Data) > 8 && bytes.Equal(echo.Data[:8], c.clientID[:]) {
|
if len(echo.Data) > 8 && bytes.Equal(echo.Data[:8], c.clientID[:]) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -237,7 +236,7 @@ func (c *xicmpConnClient) recv6() {
|
|||||||
p: p,
|
p: p,
|
||||||
addr: addr,
|
addr: addr,
|
||||||
}:
|
}:
|
||||||
case <-c.closeCh:
|
case <-c.closedCh:
|
||||||
pool.Put(p)
|
pool.Put(p)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -245,15 +244,16 @@ func (c *xicmpConnClient) recv6() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *xicmpConnClient) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
|
func (c *xicmpConnClient) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
|
||||||
packet, ok := <-c.readCh
|
select {
|
||||||
if ok {
|
case packet := <-c.readCh:
|
||||||
if packet.p != nil {
|
if packet.p != nil {
|
||||||
n = copy(p, packet.p)
|
n = copy(p, packet.p)
|
||||||
pool.Put(packet.p)
|
pool.Put(packet.p)
|
||||||
}
|
}
|
||||||
return n, packet.addr, packet.err
|
return n, packet.addr, packet.err
|
||||||
|
case <-c.closedCh:
|
||||||
|
return 0, nil, io.EOF
|
||||||
}
|
}
|
||||||
return 0, nil, io.EOF
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *xicmpConnClient) WriteTo(p []byte, addr net.Addr) (n int, err error) {
|
func (c *xicmpConnClient) WriteTo(p []byte, addr net.Addr) (n int, err error) {
|
||||||
@@ -268,9 +268,9 @@ func (c *xicmpConnClient) WriteTo(p []byte, addr net.Addr) (n int, err error) {
|
|||||||
c.seq %= 65536
|
c.seq %= 65536
|
||||||
c.mu.Unlock()
|
c.mu.Unlock()
|
||||||
|
|
||||||
ip := c.ip
|
ip := addr.(*net.UDPAddr).IP
|
||||||
if len(c.ips) > 0 {
|
if len(c.ips) > 0 {
|
||||||
ip = c.ips[mrand.Intn(len(c.ips))].AsSlice()
|
ip = c.ips[mathrand.Intn(len(c.ips))].AsSlice()
|
||||||
}
|
}
|
||||||
|
|
||||||
if c.udp {
|
if c.udp {
|
||||||
@@ -294,9 +294,10 @@ func (c *xicmpConnClient) WriteTo(p []byte, addr net.Addr) (n int, err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
errors.LogErrorInner(context.Background(), err, "send err")
|
errors.LogErrorInner(context.Background(), err, "xicmp write")
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return len(p), nil
|
return len(p), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -306,23 +307,15 @@ func (c *xicmpConnClient) Close() error {
|
|||||||
if c.closed() {
|
if c.closed() {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
close(c.closeCh)
|
close(c.closedCh)
|
||||||
_ = c.icmp4.Close()
|
_ = c.icmp4.Close()
|
||||||
_ = c.icmp6.Close()
|
_ = c.icmp6.Close()
|
||||||
c.wg.Wait()
|
_ = c.conn.Close()
|
||||||
select {
|
|
||||||
case p := <-c.readCh:
|
|
||||||
if p.p != nil {
|
|
||||||
pool.Put(p.p)
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
close(c.readCh)
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *xicmpConnClient) LocalAddr() net.Addr {
|
func (c *xicmpConnClient) LocalAddr() net.Addr {
|
||||||
return &net.UDPAddr{IP: []byte{0, 0, 0, 0}}
|
return c.conn.LocalAddr()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *xicmpConnClient) SetDeadline(t time.Time) error {
|
func (c *xicmpConnClient) SetDeadline(t time.Time) error {
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user