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.3 KiB
Go
62 lines
1.3 KiB
Go
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package resolvers
|
|
|
|
import (
|
|
"net"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"git.froth.zone/sam/awl/pkg/util"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
const (
|
|
tcp = "tcp"
|
|
udp = "udp"
|
|
)
|
|
|
|
// Resolver is the main resolver interface.
|
|
type Resolver interface {
|
|
LookUp(*dns.Msg) (util.Response, error)
|
|
}
|
|
|
|
// LoadResolver loads the respective resolver for performing a DNS query.
|
|
func LoadResolver(opts util.Options) (Resolver, error) {
|
|
switch {
|
|
case opts.HTTPS:
|
|
opts.Logger.Info("loading DNS-over-HTTPS resolver")
|
|
|
|
if !strings.HasPrefix(opts.Request.Server, "https://") {
|
|
opts.Request.Server = "https://" + opts.Request.Server
|
|
}
|
|
|
|
return &HTTPSResolver{
|
|
opts: opts,
|
|
}, nil
|
|
case opts.QUIC:
|
|
opts.Logger.Info("loading DNS-over-QUIC resolver")
|
|
opts.Request.Server = net.JoinHostPort(opts.Request.Server, strconv.Itoa(opts.Request.Port))
|
|
|
|
return &QUICResolver{
|
|
opts: opts,
|
|
}, nil
|
|
case opts.DNSCrypt:
|
|
opts.Logger.Info("loading DNSCrypt resolver")
|
|
|
|
if !strings.HasPrefix(opts.Request.Server, "sdns://") {
|
|
opts.Request.Server = "sdns://" + opts.Request.Server
|
|
}
|
|
|
|
return &DNSCryptResolver{
|
|
opts: opts,
|
|
}, nil
|
|
default:
|
|
opts.Logger.Info("loading standard/DNS-over-TLS resolver")
|
|
opts.Request.Server = net.JoinHostPort(opts.Request.Server, strconv.Itoa(opts.Request.Port))
|
|
|
|
return &StandardResolver{
|
|
opts: opts,
|
|
}, nil
|
|
}
|
|
}
|