-
-
Notifications
You must be signed in to change notification settings - Fork 36
/
doh.go
68 lines (54 loc) · 1.52 KB
/
doh.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
package config
import (
"fmt"
"time"
"github.com/qdm12/dns/v2/pkg/provider"
"github.com/qdm12/gosettings"
"github.com/qdm12/gosettings/reader"
"github.com/qdm12/gotree"
)
type DoH struct {
// UpstreamResolvers is a list of DNS over HTTPS upstream
// resolvers to use.
UpstreamResolvers []string
// Timeout is the maximum duration to wait for a response from
// upstream DNS over HTTPS servers. If left unset, it defaults
// to 1 second.
Timeout time.Duration
}
func (d *DoH) setDefaults() {
d.UpstreamResolvers = gosettings.DefaultSlice(d.UpstreamResolvers, []string{
provider.Cloudflare().Name,
provider.Google().Name,
})
d.Timeout = gosettings.DefaultComparable(d.Timeout, time.Second)
}
func (d *DoH) 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 *DoH) String() string {
return d.ToLinesNode().String()
}
func (d *DoH) ToLinesNode() (node *gotree.Node) {
node = gotree.New("DNS over HTTPs:")
node.Appendf("Upstream resolvers: %s", andStrings(d.UpstreamResolvers))
node.Appendf("Query timeout: %s", d.Timeout)
return node
}
func (d *DoH) read(reader *reader.Reader) (err error) {
d.UpstreamResolvers = reader.CSV("DOH_RESOLVERS")
d.Timeout, err = reader.Duration("DOH_TIMEOUT")
if err != nil {
return err
}
return nil
}