mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-12-27 07:59:47 +00:00
0e2304c403
Fixes https://github.com/XTLS/Xray-core/issues/4054 Fixes https://github.com/XTLS/Xray-core/issues/3338 Fixes https://github.com/XTLS/Xray-core/issues/3221
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package dns
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/xtls/xray-core/common/errors"
|
|
"github.com/xtls/xray-core/common/net"
|
|
"github.com/xtls/xray-core/features/dns"
|
|
)
|
|
|
|
type FakeDNSServer struct {
|
|
fakeDNSEngine dns.FakeDNSEngine
|
|
}
|
|
|
|
func NewFakeDNSServer(fd dns.FakeDNSEngine) *FakeDNSServer {
|
|
return &FakeDNSServer{fakeDNSEngine: fd}
|
|
}
|
|
|
|
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 {
|
|
return nil, errors.New("Unable to locate a fake DNS Engine").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
|
|
}
|