mirrors/SpoofDPI
1
0
Fork 0
mirror of https://github.com/xvzc/SpoofDPI.git synced 2024-12-22 14:26:31 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: refactor doh

Browse Source
This commit is contained in:
xvzc 2024-08-13 07:53:37 +09:00
parent 76b16da2e8
commit 464afe937d
2 changed files with 9 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dns/dns.go
View File

@ -41,7 +41,7 @@ func (d *DnsResolver) Lookup(domain string, useSystemDns bool) (string, error) {
if d.enableDoh {
log.Debug("[DNS] ", domain, " resolving with dns over https")
return dohLookup(domain)
return dohLookup(d.host, domain)
}
log.Debug("[DNS] ", domain, " resolving with custom dns")
@ -86,16 +86,16 @@ func systemLookup(domain string) (string, error) {
return "", errors.New("no record found(system)")
}
func dohLookup(domain string) (string, error) {
func dohLookup(host string, domain string) (string, error) {
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
client := GetDOHClient(*util.GetConfig().DnsAddr)
client := getDOHClient(host)
msg := new(dns.Msg)
msg.SetQuestion(dns.Fqdn(domain), dns.TypeA)
response, err := client.Exchange(ctx, domain, msg)
response, err := client.dohExchange(ctx, msg)
if err != nil {
return "", errors.New("could not resolve the domain(doh)")
}

19
dns/doh.go
View File

@ -8,7 +8,6 @@ import (
"fmt"
"net"
"net/http"
"strings"
"sync"
"time"
@ -23,17 +22,9 @@ type DOHClient struct {
var client *DOHClient
var clientOnce sync.Once
func GetDOHClient(upstream string) *DOHClient {
func getDOHClient(host 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{
@ -48,7 +39,7 @@ func GetDOHClient(upstream string) *DOHClient {
}
client = &DOHClient{
upstream: upstream,
upstream: "https://" + host + "/dns-query",
client: c,
}
}
@ -57,7 +48,7 @@ func GetDOHClient(upstream string) *DOHClient {
return client
}
func (d *DOHClient) query(ctx context.Context, msg *dns.Msg) (*dns.Msg, error) {
func (d *DOHClient) dohQuery(ctx context.Context, msg *dns.Msg) (*dns.Msg, error) {
pack, err := msg.Pack()
if err != nil {
return nil, err
@ -97,8 +88,8 @@ func (d *DOHClient) query(ctx context.Context, msg *dns.Msg) (*dns.Msg, error) {
return resultMsg, nil
}
func (d *DOHClient) Exchange(ctx context.Context, domain string, msg *dns.Msg) (*dns.Msg, error) {
res, err := d.query(ctx, msg)
func (d *DOHClient) dohExchange(ctx context.Context, msg *dns.Msg) (*dns.Msg, error) {
res, err := d.dohQuery(ctx, msg)
if err != nil {
return nil, err
}