awl/pkg/resolvers/DNSCrypt.go
Sam Therapy d952d43705
All checks were successful
continuous-integration/drone/push Build is passing
continuous-integration/drone/pr Build is passing
refactor: Add named returns somewhere
Signed-off-by: Sam Therapy <sam@samtherapy.net>
2022-12-12 19:54:23 +01:00

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) (resp util.Response, err 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
}