Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
//go:build !confonly
// +build !confonly
 
package dns
 
import (
    "context"
 
    core "github.com/v2fly/v2ray-core/v5"
    "github.com/v2fly/v2ray-core/v5/common/net"
    "github.com/v2fly/v2ray-core/v5/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, newError("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, newError("Unable to convert IP to net ip").Base(err).AtError()
    }
 
    newError(f.Name(), " got answer: ", domain, " -> ", ips).AtInfo().WriteToLog()
 
    if len(netIP) > 0 {
        return netIP, nil
    }
    return nil, dns.ErrEmptyResponse
}