101 lines
2.3 KiB
Go
101 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
type ResolvError struct {
|
|
qname, net string
|
|
nameservers []string
|
|
}
|
|
|
|
func (e ResolvError) Error() string {
|
|
errmsg := fmt.Sprintf("%s resolv failed on %s (%s)", e.qname, strings.Join(e.nameservers, "; "), e.net)
|
|
return errmsg
|
|
}
|
|
|
|
type Resolver struct {
|
|
config *dns.ClientConfig
|
|
}
|
|
|
|
// Lookup will ask each nameserver in top-to-bottom fashion, starting a new request
|
|
// in every second, and return as early as possbile (have an answer).
|
|
// It returns an error if no request has succeeded.
|
|
func (r *Resolver) Lookup(net string, req *dns.Msg) (message *dns.Msg, err error) {
|
|
c := &dns.Client{
|
|
Net: net,
|
|
ReadTimeout: r.Timeout(),
|
|
WriteTimeout: r.Timeout(),
|
|
}
|
|
|
|
qname := req.Question[0].Name
|
|
|
|
res := make(chan *dns.Msg, 1)
|
|
var wg sync.WaitGroup
|
|
L := func(nameserver string) {
|
|
defer wg.Done()
|
|
r, rtt, err := c.Exchange(req, nameserver)
|
|
if err != nil {
|
|
Debug("%s socket error on %s", qname, nameserver)
|
|
Debug("error:%s", err.Error())
|
|
return
|
|
}
|
|
if r != nil && r.Rcode != dns.RcodeSuccess {
|
|
Debug("%s failed to get an valid answer on %s", qname, nameserver)
|
|
return
|
|
}
|
|
Debug("%s resolv on %s (%s) ttl: %d", UnFqdn(qname), nameserver, net, rtt)
|
|
select {
|
|
case res <- r:
|
|
default:
|
|
}
|
|
}
|
|
|
|
ticker := time.NewTicker(time.Second)
|
|
defer ticker.Stop()
|
|
// Start lookup on each nameserver top-down, in every second
|
|
for _, nameserver := range r.Nameservers() {
|
|
wg.Add(1)
|
|
go L(nameserver)
|
|
// but exit early, if we have an answer
|
|
select {
|
|
case r := <-res:
|
|
return r, nil
|
|
case <-ticker.C:
|
|
continue
|
|
}
|
|
}
|
|
// wait for all the namservers to finish
|
|
wg.Wait()
|
|
select {
|
|
case r := <-res:
|
|
return r, nil
|
|
default:
|
|
return nil, ResolvError{qname, net, r.Nameservers()}
|
|
}
|
|
|
|
}
|
|
|
|
// Namservers return the array of nameservers, with port number appended.
|
|
// '#' in the name is treated as port separator, as with dnsmasq.
|
|
func (r *Resolver) Nameservers() (ns []string) {
|
|
for _, server := range r.config.Servers {
|
|
if i := strings.IndexByte(server, '#'); i > 0 {
|
|
server = server[:i] + ":" + server[i+1:]
|
|
} else {
|
|
server = server + ":" + r.config.Port
|
|
}
|
|
ns = append(ns, server)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (r *Resolver) Timeout() time.Duration {
|
|
return time.Duration(r.config.Timeout) * time.Second
|
|
}
|