1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
package dnsutil
import (
"errors"
"fmt"
"sync"
"time"
"github.com/miekg/dns"
"github.com/mpolden/zdns/dns/http"
)
var (
// TypeToString contains a mapping of DNS request type to string.
TypeToString = dns.TypeToString
// RcodeToString contains a mapping of Mapping DNS response code to string.
RcodeToString = dns.RcodeToString
)
// Exchanger is the interface that wraps the Exchange method of a DNS client.
type Exchanger interface {
Exchange(*dns.Msg, string) (*dns.Msg, time.Duration, error)
}
// Client wraps a DNS client and a list of server addresses.
type Client struct {
Exchanger Exchanger
Addresses []string
}
// NewClient creates a new Client using the named network and addresses.
func NewClient(network string, timeout time.Duration, addresses ...string) *Client {
var client Exchanger
if network == "https" {
client = http.NewClient(timeout)
} else {
client = &dns.Client{Net: network, Timeout: timeout}
}
return &Client{Exchanger: client, Addresses: addresses}
}
// Exchange performs a synchronous DNS query. All addresses in Client c are queried in parallel and the first successful
// response is returned.
func (c *Client) Exchange(msg *dns.Msg) (*dns.Msg, error) {
done := make(chan bool, 1)
ch := make(chan *dns.Msg, len(c.Addresses))
var wg sync.WaitGroup
wg.Add(len(c.Addresses))
err := errors.New("addr is empty")
for _, a := range c.Addresses {
go func(addr string) {
defer wg.Done()
r, _, err1 := c.Exchanger.Exchange(msg, addr)
if err1 != nil {
err = fmt.Errorf("resolver %s failed: %w", addr, err1)
return
}
ch <- r
}(a)
}
go func() {
wg.Wait()
done <- true
}()
for {
select {
case <-done:
return nil, err
case rr := <-ch:
return rr, nil
}
}
}
// Answers returns all values in the answer section of DNS message msg.
func Answers(msg *dns.Msg) []string {
var answers []string
for _, answer := range msg.Answer {
switch v := answer.(type) {
case *dns.A:
answers = append(answers, v.A.String())
case *dns.AAAA:
answers = append(answers, v.AAAA.String())
case *dns.MX:
answers = append(answers, v.Mx)
case *dns.PTR:
answers = append(answers, v.Ptr)
case *dns.NS:
answers = append(answers, v.Ns)
case *dns.CNAME:
answers = append(answers, v.Target)
}
}
return answers
}
// MinTTL returns the lowest TTL of of answer, authority and additional sections.
func MinTTL(msg *dns.Msg) time.Duration {
var ttl uint32 = (1 << 31) - 1 // Maximum TTL from RFC 2181
for _, answer := range msg.Answer {
ttl = min(answer.Header().Ttl, ttl)
}
for _, ns := range msg.Ns {
ttl = min(ns.Header().Ttl, ttl)
}
for _, extra := range msg.Extra {
// OPT (EDNS) is a pseudo record which uses TTL field for extended RCODE and flags
if extra.Header().Rrtype == dns.TypeOPT {
continue
}
ttl = min(extra.Header().Ttl, ttl)
}
return time.Duration(ttl) * time.Second
}
func min(x, y uint32) uint32 {
if x < y {
return x
}
return y
}
|