diff --git a/dns/dns.go b/dns/dns.go index 9ba0d3e..53b2c2a 100644 --- a/dns/dns.go +++ b/dns/dns.go @@ -8,8 +8,6 @@ import ( "strconv" "time" - "github.com/likexian/doh" - dohDns "github.com/likexian/doh/dns" "github.com/miekg/dns" log "github.com/sirupsen/logrus" "github.com/xvzc/SpoofDPI/util" @@ -37,17 +35,17 @@ func (d *DnsResolver) Lookup(domain string, useSystemDns bool) (string, error) { } if useSystemDns { - log.Debug("[DNS] ", domain, " resolving with system dns") + log.Debug("[DNS] ", domain, " resolving with system dns") return systemLookup(domain) } if d.enableDoh { - log.Debug("[DNS] ", domain, " resolving with dns over https") + log.Debug("[DNS] ", domain, " resolving with dns over https") return dohLookup(domain) } - log.Debug("[DNS] ", domain, " resolving with custom dns") - return customLookup(d.host, d.port, domain) + log.Debug("[DNS] ", domain, " resolving with custom dns") + return customLookup(d.host, d.port, domain) } func customLookup(host string, port string, domain string) (string, error) { @@ -91,26 +89,18 @@ func systemLookup(domain string) (string, error) { func dohLookup(domain string) (string, error) { ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() - c := doh.Use(doh.CloudflareProvider, doh.GoogleProvider) + log.Debug("[DoH] ", domain, " resolving with dns over https") - rsp, err := c.Query(ctx, dohDns.Domain(domain), dohDns.TypeA) - if err != nil { - return "", errors.New("could not resolve the domain(doh)") - } - // doh dns answer - answer := rsp.Answer - - // print all answer - for _, a := range answer { - if a.Type != 1 { // Type == 1 -> A Record - continue + dnsUpstream := util.GetConfig().DnsAddr + client := GetDoHClient(*dnsUpstream) + resp, err := client.Resolve(ctx, domain, []uint16{dns.TypeA, dns.TypeAAAA}) + if err == nil { + if len(resp) == 0 { // yes this happens + return "", errors.New("no record found(doh)") } - return a.Data, nil + return resp[0], nil } - // close the client - c.Close() - - return "", errors.New("no record found(doh)") + return "", errors.New("could not resolve the domain(doh)") } diff --git a/dns/doh.go b/dns/doh.go new file mode 100644 index 0000000..43e52bc --- /dev/null +++ b/dns/doh.go @@ -0,0 +1,124 @@ +package dns + +import ( + "bytes" + "context" + "encoding/base64" + "fmt" + "net" + "net/http" + "strings" + "sync" + "time" + + "github.com/miekg/dns" + log "github.com/sirupsen/logrus" +) + +type DoHClient struct { + upstream string + c *http.Client +} + +var client *DoHClient +var clientOnce sync.Once + +func GetDoHClient(upstream string) *DoHClient { + clientOnce.Do(func() { + if client == nil { + if !strings.HasPrefix(upstream, "https://") { + upstream = "https://" + upstream + } + + if !strings.HasSuffix(upstream, "/dns-query") { + upstream = upstream + "/dns-query" + } + + c := &http.Client{ + Timeout: 5 * time.Second, + Transport: &http.Transport{ + DialContext: (&net.Dialer{ + Timeout: 3 * time.Second, + KeepAlive: 30 * time.Second, + }).DialContext, + TLSHandshakeTimeout: 5 * time.Second, + MaxIdleConnsPerHost: 100, + MaxIdleConns: 100, + }, + } + + client = &DoHClient{ + upstream: upstream, + c: c, + } + } + }) + + return client +} + +func (d *DoHClient) doGetRequest(ctx context.Context, msg *dns.Msg) (*dns.Msg, error) { + pack, err := msg.Pack() + if err != nil { + return nil, err + } + + url := fmt.Sprintf("%s?dns=%s", d.upstream, base64.RawStdEncoding.EncodeToString(pack)) + req, err := http.NewRequest("GET", url, nil) + if err != nil { + return nil, err + } + + req = req.WithContext(ctx) + req.Header.Set("Accept", "application/dns-message") + + resp, err := d.c.Do(req) + if err != nil { + return nil, err + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + log.Debug("[DoH] Error while resolving ", url, " : ", resp.Status) + } + + buf := bytes.Buffer{} + buf.ReadFrom(resp.Body) + + ret_msg := new(dns.Msg) + err = ret_msg.Unpack(buf.Bytes()) + if err != nil { + return nil, err + } + + return ret_msg, nil +} + +func (d *DoHClient) Resolve(ctx context.Context, domain string, dnsTypes []uint16) ([]string, error) { + var ret []string + + for _, dnsType := range dnsTypes { + msg := new(dns.Msg) + msg.SetQuestion(dns.Fqdn(domain), dnsType) + + resp, err := d.doGetRequest(ctx, msg) + if err != nil { + return nil, err + } + + if resp.Rcode != dns.RcodeSuccess { + continue + } + + for _, answer := range resp.Answer { + if t, ok := answer.(*dns.A); ok { + ret = append(ret, t.A.String()) + } + if t, ok := answer.(*dns.AAAA); ok { + ret = append(ret, t.AAAA.String()) + } + } + } + + return ret, nil +} diff --git a/go.mod b/go.mod index 4028aac..1b45c00 100644 --- a/go.mod +++ b/go.mod @@ -5,7 +5,6 @@ go 1.21 toolchain go1.21.5 require ( - github.com/likexian/doh v0.7.1 github.com/miekg/dns v1.1.61 github.com/pterm/pterm v0.12.79 github.com/sirupsen/logrus v1.9.3 @@ -17,7 +16,6 @@ require ( atomicgo.dev/schedule v0.1.0 // indirect github.com/containerd/console v1.0.3 // indirect github.com/gookit/color v1.5.4 // indirect - github.com/likexian/gokit v0.25.15 // indirect github.com/lithammer/fuzzysearch v1.1.8 // indirect github.com/mattn/go-runewidth v0.0.15 // indirect github.com/rivo/uniseg v0.4.4 // indirect diff --git a/go.sum b/go.sum index 0f0fe2d..c98a4e4 100644 --- a/go.sum +++ b/go.sum @@ -33,10 +33,6 @@ github.com/klauspost/cpuid/v2 v2.2.3/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8t github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= -github.com/likexian/doh v0.7.1 h1:0p75orMxXrfGNAOnOEtl2xZ5T9Ctqi++i/WhN6P6hs8= -github.com/likexian/doh v0.7.1/go.mod h1:aIEOK197o6nC82iRo5+5L5ws+X7xRCmuer3M4In91WA= -github.com/likexian/gokit v0.25.15 h1:QjospM1eXhdMMHwZRpMKKAHY/Wig9wgcREmLtf9NslY= -github.com/likexian/gokit v0.25.15/go.mod h1:S2QisdsxLEHWeD/XI0QMVeggp+jbxYqUxMvSBil7MRg= github.com/lithammer/fuzzysearch v1.1.8 h1:/HIuJnjHuXS8bKaiTMeeDlW2/AyIWk2brx1V8LFgLN4= github.com/lithammer/fuzzysearch v1.1.8/go.mod h1:IdqeyBClc3FFqSzYq/MXESsS4S0FsZ5ajtkr5xPLts4= github.com/mattn/go-runewidth v0.0.13/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=