mirror of https://github.com/status-im/consul.git
124 lines
3.2 KiB
Go
124 lines
3.2 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package dns
|
|
|
|
import (
|
|
"errors"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/miekg/dns"
|
|
|
|
"github.com/hashicorp/consul/ipaddr"
|
|
"github.com/hashicorp/consul/logging"
|
|
)
|
|
|
|
type recursor struct {
|
|
logger hclog.Logger
|
|
}
|
|
|
|
func newRecursor(logger hclog.Logger) *recursor {
|
|
return &recursor{
|
|
logger: logger.Named(logging.DNS),
|
|
}
|
|
}
|
|
|
|
// handle is used to process DNS queries for externally configured servers
|
|
func (r *recursor) handle(req *dns.Msg, cfgCtx *RouterDynamicConfig, remoteAddr net.Addr) (*dns.Msg, error) {
|
|
q := req.Question[0]
|
|
|
|
network := "udp"
|
|
defer func(s time.Time) {
|
|
r.logger.Trace("request served from client",
|
|
"question", q,
|
|
"network", network,
|
|
"latency", time.Since(s).String(),
|
|
"client", remoteAddr.String(),
|
|
"client_network", remoteAddr.Network(),
|
|
)
|
|
}(time.Now())
|
|
|
|
// Switch to TCP if the client is
|
|
if _, ok := remoteAddr.(*net.TCPAddr); ok {
|
|
network = "tcp"
|
|
}
|
|
|
|
// Recursively resolve
|
|
c := &dns.Client{Net: network, Timeout: cfgCtx.RecursorTimeout}
|
|
var resp *dns.Msg
|
|
var rtt time.Duration
|
|
var err error
|
|
for _, idx := range cfgCtx.RecursorStrategy.Indexes(len(cfgCtx.Recursors)) {
|
|
recurseAddr := cfgCtx.Recursors[idx]
|
|
resp, rtt, err = c.Exchange(req, recurseAddr)
|
|
// Check if the response is valid and has the desired Response code
|
|
if resp != nil && (resp.Rcode != dns.RcodeSuccess && resp.Rcode != dns.RcodeNameError) {
|
|
r.logger.Trace("recurse failed for question",
|
|
"question", q,
|
|
"rtt", rtt,
|
|
"recursor", recurseAddr,
|
|
"rcode", dns.RcodeToString[resp.Rcode],
|
|
)
|
|
// If we still have recursors to forward the query to,
|
|
// we move forward onto the next one else the loop ends
|
|
continue
|
|
} else if err == nil || (resp != nil && resp.Truncated) {
|
|
// Compress the response; we don't know if the incoming
|
|
// response was compressed or not, so by not compressing
|
|
// we might generate an invalid packet on the way out.
|
|
resp.Compress = !cfgCtx.DisableCompression
|
|
|
|
// Forward the response
|
|
r.logger.Trace("recurse succeeded for question",
|
|
"question", q,
|
|
"rtt", rtt,
|
|
"recursor", recurseAddr,
|
|
)
|
|
return resp, nil
|
|
}
|
|
r.logger.Error("recurse failed", "error", err)
|
|
}
|
|
|
|
// If all resolvers fail, return a SERVFAIL message
|
|
r.logger.Error("all resolvers failed for question from client",
|
|
"question", q,
|
|
"client", remoteAddr.String(),
|
|
"client_network", remoteAddr.Network(),
|
|
)
|
|
|
|
return nil, errRecursionFailed
|
|
}
|
|
|
|
// formatRecursorAddress is used to add a port to the recursor if omitted.
|
|
func formatRecursorAddress(recursor string) (string, error) {
|
|
_, _, err := net.SplitHostPort(recursor)
|
|
var ae *net.AddrError
|
|
if errors.As(err, &ae) {
|
|
switch ae.Err {
|
|
case "missing port in address":
|
|
recursor = ipaddr.FormatAddressPort(recursor, 53)
|
|
case "too many colons in address":
|
|
if ip := net.ParseIP(recursor); ip != nil && ip.To4() == nil {
|
|
recursor = ipaddr.FormatAddressPort(recursor, 53)
|
|
break
|
|
}
|
|
fallthrough
|
|
default:
|
|
return "", err
|
|
}
|
|
} else if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// Get the address
|
|
addr, err := net.ResolveTCPAddr("tcp", recursor)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// Return string
|
|
return addr.String(), nil
|
|
}
|