mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-09 11:51:02 -05:00
250 lines
7.8 KiB
Go
250 lines
7.8 KiB
Go
|
package dns_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/google/go-cmp/cmp"
|
||
|
|
"github.com/miekg/dns"
|
||
|
|
"google.golang.org/protobuf/types/known/anypb"
|
||
|
|
|
||
|
|
core "github.com/v2fly/v2ray-core/v5"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/app/dispatcher"
|
||
|
|
. "github.com/v2fly/v2ray-core/v5/app/dns"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/app/dns/fakedns"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/app/policy"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/app/proxyman"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/common"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/common/net"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/common/serial"
|
||
|
|
feature_dns "github.com/v2fly/v2ray-core/v5/features/dns"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/proxy/freedom"
|
||
|
|
"github.com/v2fly/v2ray-core/v5/testing/servers/udp"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFakeDNS(t *testing.T) {
|
||
|
|
port := udp.PickPort()
|
||
|
|
|
||
|
|
dnsServer := dns.Server{
|
||
|
|
Addr: "127.0.0.1:" + port.String(),
|
||
|
|
Net: "udp",
|
||
|
|
Handler: &staticHandler{},
|
||
|
|
UDPSize: 1200,
|
||
|
|
}
|
||
|
|
|
||
|
|
go dnsServer.ListenAndServe()
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
|
||
|
|
config := &core.Config{
|
||
|
|
App: []*anypb.Any{
|
||
|
|
serial.ToTypedMessage(&Config{
|
||
|
|
NameServer: []*NameServer{
|
||
|
|
{ // "fakedns"
|
||
|
|
Address: &net.Endpoint{
|
||
|
|
Network: net.Network_UDP,
|
||
|
|
Address: &net.IPOrDomain{
|
||
|
|
Address: &net.IPOrDomain_Domain{
|
||
|
|
Domain: "fakedns",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Port: uint32(53),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{ // { "address": "127.0.0.1", "port": "<port>", "domains": ["domain:google.com"], "fakedns": "198.19.0.0/16", "fallbackStrategy": "disabled" }
|
||
|
|
Address: &net.Endpoint{
|
||
|
|
Network: net.Network_UDP,
|
||
|
|
Address: &net.IPOrDomain{
|
||
|
|
Address: &net.IPOrDomain_Ip{
|
||
|
|
Ip: []byte{127, 0, 0, 1},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Port: uint32(port),
|
||
|
|
},
|
||
|
|
PrioritizedDomain: []*NameServer_PriorityDomain{
|
||
|
|
{Type: DomainMatchingType_Subdomain, Domain: "google.com"},
|
||
|
|
},
|
||
|
|
FakeDns: &fakedns.FakeDnsPoolMulti{
|
||
|
|
Pools: []*fakedns.FakeDnsPool{
|
||
|
|
{IpPool: "198.19.0.0/16", LruSize: 256},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
FallbackStrategy: FallbackStrategy_Disabled.Enum(),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
FakeDns: &fakedns.FakeDnsPoolMulti{ // "fakedns": "198.18.0.0/16"
|
||
|
|
Pools: []*fakedns.FakeDnsPool{
|
||
|
|
{IpPool: "198.18.0.0/16", LruSize: 256},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}),
|
||
|
|
serial.ToTypedMessage(&dispatcher.Config{}),
|
||
|
|
serial.ToTypedMessage(&proxyman.OutboundConfig{}),
|
||
|
|
serial.ToTypedMessage(&policy.Config{}),
|
||
|
|
},
|
||
|
|
Outbound: []*core.OutboundHandlerConfig{
|
||
|
|
{
|
||
|
|
ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
v, err := core.New(config)
|
||
|
|
common.Must(err)
|
||
|
|
common.Must(v.Start())
|
||
|
|
|
||
|
|
dnsClient := v.GetFeature(feature_dns.ClientType()).(feature_dns.Client)
|
||
|
|
fakeClient := dnsClient.(feature_dns.ClientWithFakeDNS).AsFakeDNSClient()
|
||
|
|
|
||
|
|
var fakeIPForFacebook net.IP
|
||
|
|
var fakeIPForGoogle net.IP
|
||
|
|
|
||
|
|
{ // Lookup facebook.com with Fake Client will return 198.18.0.0/16 (global fake pool)
|
||
|
|
ips, err := fakeClient.LookupIP("facebook.com")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("unexpected error: ", err)
|
||
|
|
}
|
||
|
|
for _, ip := range ips {
|
||
|
|
if !(&net.IPNet{IP: net.IP{198, 18, 0, 0}, Mask: net.CIDRMask(16, 8*net.IPv4len)}).Contains(ip) {
|
||
|
|
t.Fatal("Lookup facebook.com with fake client not in global pool 198.18.0.0/16")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fakeIPForFacebook = ips[0]
|
||
|
|
}
|
||
|
|
{ // Lookup facebook.com with Normal Client with return empty record (because UDP server matching "domain:google.com" are configured with fallback disabled)
|
||
|
|
_, err := dnsClient.LookupIP("facebook.com")
|
||
|
|
if err != feature_dns.ErrEmptyResponse {
|
||
|
|
t.Fatal("Lookup facebook.com with normal client not returning empty response")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{ // Lookup google.com with Fake Client will return 198.19.0.0/16 (local fake pool)
|
||
|
|
ips, err := fakeClient.LookupIP("google.com")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("unexpected error: ", err)
|
||
|
|
}
|
||
|
|
for _, ip := range ips {
|
||
|
|
if !(&net.IPNet{IP: net.IP{198, 19, 0, 0}, Mask: net.CIDRMask(16, 8*net.IPv4len)}).Contains(ip) {
|
||
|
|
t.Fatal("Lookup google.com with fake client not in global pool 198.19.0.0/16")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fakeIPForGoogle = ips[0]
|
||
|
|
}
|
||
|
|
{ // Lookup google.com with Normal Client will return 8.8.8.8
|
||
|
|
ips, err := dnsClient.LookupIP("google.com")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("unexpected error: ", err)
|
||
|
|
}
|
||
|
|
if r := cmp.Diff(ips, []net.IP{{8, 8, 8, 8}}); r != "" {
|
||
|
|
t.Fatal("Lookup google.com with normal client not returning 8.8.8.8")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fakeEngine := dnsClient.(feature_dns.ClientWithFakeDNS).AsFakeDNSEngine().(feature_dns.FakeDNSEngineRev0)
|
||
|
|
{
|
||
|
|
if !fakeEngine.IsIPInIPPool(net.IPAddress(fakeIPForFacebook)) {
|
||
|
|
t.Fatal("Fake IP of domain facebook.com not in FakeDNSEngine's pool.")
|
||
|
|
}
|
||
|
|
if !fakeEngine.IsIPInIPPool(net.IPAddress(fakeIPForGoogle)) {
|
||
|
|
t.Fatal("Fake IP of domain google.com not in FakeDNSEngine's pool.")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{
|
||
|
|
if domain := fakeEngine.GetDomainFromFakeDNS(net.IPAddress(fakeIPForFacebook)); domain != "facebook.com" {
|
||
|
|
t.Fatal("Recover fake IP to get domain facebook.com failed.")
|
||
|
|
}
|
||
|
|
if domain := fakeEngine.GetDomainFromFakeDNS(net.IPAddress(fakeIPForGoogle)); domain != "google.com" {
|
||
|
|
t.Fatal("Recover fake IP to get domain google.com failed.")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{
|
||
|
|
ips := fakeEngine.GetFakeIPForDomain("api.google.com")
|
||
|
|
for _, ip := range ips {
|
||
|
|
if !(&net.IPNet{IP: net.IP{198, 19, 0, 0}, Mask: net.CIDRMask(16, 8*net.IPv4len)}).Contains(ip.IP()) {
|
||
|
|
t.Fatal("Fake IP for api.google.com not in local pool 198.19.0.0/16")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{
|
||
|
|
ips := fakeEngine.GetFakeIPForDomain3("v2fly.org", true, false)
|
||
|
|
for _, ip := range ips {
|
||
|
|
if !(&net.IPNet{IP: net.IP{198, 18, 0, 0}, Mask: net.CIDRMask(16, 8*net.IPv4len)}).Contains(ip.IP()) {
|
||
|
|
t.Fatal("Fake IP for v2fly.org not in global pool 198.18.0.0/16")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFakeDNSEmptyGlobalConfig(t *testing.T) {
|
||
|
|
config := &core.Config{
|
||
|
|
App: []*anypb.Any{
|
||
|
|
serial.ToTypedMessage(&Config{
|
||
|
|
NameServer: []*NameServer{
|
||
|
|
{ // "fakedns"
|
||
|
|
Address: &net.Endpoint{
|
||
|
|
Network: net.Network_UDP,
|
||
|
|
Address: &net.IPOrDomain{
|
||
|
|
Address: &net.IPOrDomain_Domain{
|
||
|
|
Domain: "fakedns",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
QueryStrategy: QueryStrategy_USE_IP4.Enum(),
|
||
|
|
},
|
||
|
|
{ // "localhost"
|
||
|
|
Address: &net.Endpoint{
|
||
|
|
Network: net.Network_UDP,
|
||
|
|
Address: &net.IPOrDomain{
|
||
|
|
Address: &net.IPOrDomain_Domain{
|
||
|
|
Domain: "localhost",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
QueryStrategy: QueryStrategy_USE_IP6.Enum(),
|
||
|
|
PrioritizedDomain: []*NameServer_PriorityDomain{
|
||
|
|
{Type: DomainMatchingType_Subdomain, Domain: "google.com"},
|
||
|
|
},
|
||
|
|
FakeDns: &fakedns.FakeDnsPoolMulti{Pools: []*fakedns.FakeDnsPool{}}, // "fakedns": true
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}),
|
||
|
|
serial.ToTypedMessage(&dispatcher.Config{}),
|
||
|
|
serial.ToTypedMessage(&proxyman.OutboundConfig{}),
|
||
|
|
serial.ToTypedMessage(&policy.Config{}),
|
||
|
|
},
|
||
|
|
Outbound: []*core.OutboundHandlerConfig{
|
||
|
|
{
|
||
|
|
ProxySettings: serial.ToTypedMessage(&freedom.Config{}),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
v, err := core.New(config)
|
||
|
|
common.Must(err)
|
||
|
|
common.Must(v.Start())
|
||
|
|
|
||
|
|
dnsClient := v.GetFeature(feature_dns.ClientType()).(feature_dns.Client)
|
||
|
|
fakeClient := dnsClient.(feature_dns.ClientWithFakeDNS).AsFakeDNSClient()
|
||
|
|
|
||
|
|
{ // Lookup facebook.com will return 198.18.0.0/15 (default IPv4 pool)
|
||
|
|
ips, err := fakeClient.LookupIP("facebook.com")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("unexpected error: ", err)
|
||
|
|
}
|
||
|
|
for _, ip := range ips {
|
||
|
|
if !(&net.IPNet{IP: net.IP{198, 18, 0, 0}, Mask: net.CIDRMask(15, 8*net.IPv4len)}).Contains(ip) {
|
||
|
|
t.Fatal("Lookup facebook.com with fake client not in default IPv4 pool 198.18.0.0/15")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{ // Lookup google.com will return fc00::/18 (default IPv6 pool)
|
||
|
|
ips, err := fakeClient.LookupIP("google.com")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("unexpected error: ", err)
|
||
|
|
}
|
||
|
|
for _, ip := range ips {
|
||
|
|
if !(&net.IPNet{IP: net.IP{0xfc, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, Mask: net.CIDRMask(18, 8*net.IPv6len)}).Contains(ip) {
|
||
|
|
t.Fatal("Lookup google.com with fake client not in default IPv6 pool fc00::/18")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|