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.
50 lines
1.2 KiB
50 lines
1.2 KiB
package dns |
|
|
|
import ( |
|
"context" |
|
|
|
"github.com/xtls/xray-core/common/errors" |
|
"github.com/xtls/xray-core/common/net" |
|
"github.com/xtls/xray-core/core" |
|
"github.com/xtls/xray-core/features/dns" |
|
) |
|
|
|
type FakeDNSServer struct { |
|
fakeDNSEngine dns.FakeDNSEngine |
|
} |
|
|
|
func NewFakeDNSServer() *FakeDNSServer { |
|
return &FakeDNSServer{} |
|
} |
|
|
|
func (FakeDNSServer) Name() string { |
|
return "FakeDNS" |
|
} |
|
|
|
func (f *FakeDNSServer) QueryIP(ctx context.Context, domain string, _ net.IP, opt dns.IPOption, _ bool) ([]net.IP, error) { |
|
if f.fakeDNSEngine == nil { |
|
if err := core.RequireFeatures(ctx, func(fd dns.FakeDNSEngine) { |
|
f.fakeDNSEngine = fd |
|
}); err != nil { |
|
return nil, errors.New("Unable to locate a fake DNS Engine").Base(err).AtError() |
|
} |
|
} |
|
var ips []net.Address |
|
if fkr0, ok := f.fakeDNSEngine.(dns.FakeDNSEngineRev0); ok { |
|
ips = fkr0.GetFakeIPForDomain3(domain, opt.IPv4Enable, opt.IPv6Enable) |
|
} else { |
|
ips = f.fakeDNSEngine.GetFakeIPForDomain(domain) |
|
} |
|
|
|
netIP, err := toNetIP(ips) |
|
if err != nil { |
|
return nil, errors.New("Unable to convert IP to net ip").Base(err).AtError() |
|
} |
|
|
|
errors.LogInfo(ctx, f.Name(), " got answer: ", domain, " -> ", ips) |
|
|
|
if len(netIP) > 0 { |
|
return netIP, nil |
|
} |
|
return nil, dns.ErrEmptyResponse |
|
}
|
|
|