mirror of https://github.com/XTLS/Xray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
2.7 KiB
107 lines
2.7 KiB
package dns_test |
|
|
|
import ( |
|
"context" |
|
"net/url" |
|
"testing" |
|
"time" |
|
|
|
"github.com/google/go-cmp/cmp" |
|
. "github.com/xtls/xray-core/app/dns" |
|
"github.com/xtls/xray-core/common" |
|
"github.com/xtls/xray-core/common/net" |
|
dns_feature "github.com/xtls/xray-core/features/dns" |
|
) |
|
|
|
func TestTCPLocalNameServer(t *testing.T) { |
|
url, err := url.Parse("tcp+local://8.8.8.8") |
|
common.Must(err) |
|
s, err := NewTCPLocalNameServer(url, QueryStrategy_USE_IP) |
|
common.Must(err) |
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) |
|
ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.IPOption{ |
|
IPv4Enable: true, |
|
IPv6Enable: true, |
|
}, false) |
|
cancel() |
|
common.Must(err) |
|
if len(ips) == 0 { |
|
t.Error("expect some ips, but got 0") |
|
} |
|
} |
|
|
|
func TestTCPLocalNameServerWithCache(t *testing.T) { |
|
url, err := url.Parse("tcp+local://8.8.8.8") |
|
common.Must(err) |
|
s, err := NewTCPLocalNameServer(url, QueryStrategy_USE_IP) |
|
common.Must(err) |
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) |
|
ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.IPOption{ |
|
IPv4Enable: true, |
|
IPv6Enable: true, |
|
}, false) |
|
cancel() |
|
common.Must(err) |
|
if len(ips) == 0 { |
|
t.Error("expect some ips, but got 0") |
|
} |
|
|
|
ctx2, cancel := context.WithTimeout(context.Background(), time.Second*5) |
|
ips2, err := s.QueryIP(ctx2, "google.com", net.IP(nil), dns_feature.IPOption{ |
|
IPv4Enable: true, |
|
IPv6Enable: true, |
|
}, true) |
|
cancel() |
|
common.Must(err) |
|
if r := cmp.Diff(ips2, ips); r != "" { |
|
t.Fatal(r) |
|
} |
|
} |
|
|
|
func TestTCPLocalNameServerWithIPv4Override(t *testing.T) { |
|
url, err := url.Parse("tcp+local://8.8.8.8") |
|
common.Must(err) |
|
s, err := NewTCPLocalNameServer(url, QueryStrategy_USE_IP4) |
|
common.Must(err) |
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) |
|
ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.IPOption{ |
|
IPv4Enable: true, |
|
IPv6Enable: true, |
|
}, false) |
|
cancel() |
|
common.Must(err) |
|
|
|
if len(ips) == 0 { |
|
t.Error("expect some ips, but got 0") |
|
} |
|
|
|
for _, ip := range ips { |
|
if len(ip) != net.IPv4len { |
|
t.Error("expect only IPv4 response from DNS query") |
|
} |
|
} |
|
} |
|
|
|
func TestTCPLocalNameServerWithIPv6Override(t *testing.T) { |
|
url, err := url.Parse("tcp+local://8.8.8.8") |
|
common.Must(err) |
|
s, err := NewTCPLocalNameServer(url, QueryStrategy_USE_IP6) |
|
common.Must(err) |
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) |
|
ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.IPOption{ |
|
IPv4Enable: true, |
|
IPv6Enable: true, |
|
}, false) |
|
cancel() |
|
common.Must(err) |
|
|
|
if len(ips) == 0 { |
|
t.Error("expect some ips, but got 0") |
|
} |
|
|
|
for _, ip := range ips { |
|
if len(ip) != net.IPv6len { |
|
t.Error("expect only IPv6 response from DNS query") |
|
} |
|
} |
|
}
|
|
|