grumbulon
3a0a8f015a
All checks were successful
continuous-integration/drone/push Build is passing
refactor Co-authored-by: Sam Therapy <sam@samtherapy.net> Reviewed-on: #110 Reviewed-by: Sam <sam@samtherapy.net>
62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package resolvers
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"git.froth.zone/sam/awl/pkg/util"
|
|
"github.com/ameshkov/dnscrypt/v2"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// DNSCryptResolver is for making DNSCrypt queries.
|
|
type DNSCryptResolver struct {
|
|
opts util.Options
|
|
}
|
|
|
|
var _ Resolver = (*DNSCryptResolver)(nil)
|
|
|
|
// LookUp performs a DNS query.
|
|
func (resolver *DNSCryptResolver) LookUp(msg *dns.Msg) (util.Response, error) {
|
|
client := dnscrypt.Client{
|
|
Timeout: resolver.opts.Request.Timeout,
|
|
UDPSize: 1232,
|
|
}
|
|
|
|
if resolver.opts.TCP || resolver.opts.TLS {
|
|
client.Net = tcp
|
|
} else {
|
|
client.Net = udp
|
|
}
|
|
|
|
switch {
|
|
case resolver.opts.IPv4:
|
|
client.Net += "4"
|
|
case resolver.opts.IPv6:
|
|
client.Net += "6"
|
|
}
|
|
|
|
resolver.opts.Logger.Debug("Using", client.Net, "for making the request")
|
|
|
|
resolverInf, err := client.Dial(resolver.opts.Request.Server)
|
|
if err != nil {
|
|
return util.Response{}, fmt.Errorf("dnscrypt: dial: %w", err)
|
|
}
|
|
|
|
now := time.Now()
|
|
res, err := client.Exchange(msg, resolverInf)
|
|
rtt := time.Since(now)
|
|
|
|
if err != nil {
|
|
return util.Response{}, fmt.Errorf("dnscrypt: exchange: %w", err)
|
|
}
|
|
|
|
resolver.opts.Logger.Info("Request successful")
|
|
|
|
return util.Response{
|
|
DNS: res,
|
|
RTT: rtt,
|
|
}, nil
|
|
}
|