-
-
Notifications
You must be signed in to change notification settings - Fork 40
/
dot.go
73 lines (58 loc) · 1.59 KB
/
dot.go
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
package config
import (
"errors"
"fmt"
"time"
"github.com/qdm12/dns/v2/pkg/provider"
"github.com/qdm12/gosettings"
"github.com/qdm12/gosettings/reader"
"github.com/qdm12/gotree"
)
type DoT struct {
// UpstreamResolvers is a list of DNS over TLS upstream
// resolvers to use.
UpstreamResolvers []string
// Timeout is the maximum duration to wait for a response from
// upstream DNS over TLS servers. If left unset, it defaults to
// 1 second.
Timeout time.Duration
}
func (d *DoT) setDefaults() {
d.UpstreamResolvers = gosettings.DefaultSlice(d.UpstreamResolvers, []string{
provider.Cloudflare().Name,
provider.Google().Name,
})
d.Timeout = gosettings.DefaultComparable(d.Timeout, time.Second)
}
var (
ErrTimeoutTooSmall = errors.New("timeout is too small")
)
func (d *DoT) validate() (err error) {
err = checkUpstreamResolverNames(d.UpstreamResolvers)
if err != nil {
return fmt.Errorf("upstream resolvers: %w", err)
}
const minTimeout = time.Millisecond
if d.Timeout < minTimeout {
return fmt.Errorf("%w: %s must be at least %s",
ErrTimeoutTooSmall, d.Timeout, minTimeout)
}
return nil
}
func (d *DoT) String() string {
return d.ToLinesNode().String()
}
func (d *DoT) ToLinesNode() (node *gotree.Node) {
node = gotree.New("DNS over TLS:")
node.Appendf("Upstream resolvers: %s", andStrings(d.UpstreamResolvers))
node.Appendf("Request timeout: %s", d.Timeout)
return node
}
func (d *DoT) read(reader *reader.Reader) (err error) {
d.UpstreamResolvers = reader.CSV("DOT_RESOLVERS")
d.Timeout, err = reader.Duration("DOT_TIMEOUT")
if err != nil {
return err
}
return nil
}