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
|
package dnsserver
import (
"fmt"
"testing"
"github.com/miekg/dns"
"blitiri.com.ar/go/dnss/internal/testutil"
)
// Tests for the DNS server.
func TestServe(t *testing.T) {
res := testutil.NewTestResolver()
res.Response = &dns.Msg{
Answer: []dns.RR{testutil.NewRR(t, "response.test. A 1.1.1.1")},
}
unqUpstreamAddr := testutil.GetFreePort()
go testutil.ServeTestDNSServer(unqUpstreamAddr,
testutil.MakeStaticHandler(t, "unq. A 2.2.2.2"))
overrideAddr3 := testutil.GetFreePort()
go testutil.ServeTestDNSServer(overrideAddr3,
testutil.MakeStaticHandler(t, "a.ov3. A 3.3.3.3"))
overrideAddr4 := testutil.GetFreePort()
go testutil.ServeTestDNSServer(overrideAddr4,
testutil.MakeStaticHandler(t, "b.ov4. A 4.4.4.4"))
overrides := DomainMap{
"ov3.": overrideAddr3,
"a.ov4.": overrideAddr4,
}
srv := New(testutil.GetFreePort(), res, unqUpstreamAddr, overrides)
go srv.ListenAndServe()
testutil.WaitForDNSServer(srv.Addr)
query(t, srv.Addr, "response.test.", "1.1.1.1")
query(t, srv.Addr, "unqualified.", "2.2.2.2")
query(t, srv.Addr, "ov3.", "3.3.3.3")
query(t, srv.Addr, "x.ov3.", "3.3.3.3")
query(t, srv.Addr, "y.x.OV3.", "3.3.3.3")
query(t, srv.Addr, "A.ov4.", "4.4.4.4")
query(t, srv.Addr, "z.a.ov4.", "4.4.4.4")
query(t, srv.Addr, "b.ov4.", "1.1.1.1") // Not overridden.
}
func query(t *testing.T, srv, domain, expected string) {
_, rr, err := testutil.DNSQuery(srv, domain, dns.TypeA)
if err != nil {
t.Errorf("error querying %q: %v", domain, err)
return
}
result := rr.(*dns.A).A.String()
if result != expected {
t.Errorf("query %q: expected %q but got %q", domain, expected, result)
}
}
func TestBadUpstreams(t *testing.T) {
res := testutil.NewTestResolver()
res.RespError = fmt.Errorf("response error for testing")
// Get addresses but don't start the servers, so we get an error when
// trying to reach them.
unqUpstreamAddr := testutil.GetFreePort()
overrideAddr1 := testutil.GetFreePort()
overrides := DomainMap{
"ov1.": overrideAddr1,
}
srv := New(testutil.GetFreePort(), res, unqUpstreamAddr, overrides)
go srv.ListenAndServe()
testutil.WaitForDNSServer(srv.Addr)
queryFailure(t, srv.Addr, "response.test.")
queryFailure(t, srv.Addr, "unqualified.")
queryFailure(t, srv.Addr, "ov1.")
}
func queryFailure(t *testing.T, srv, domain string) {
m, _, err := testutil.DNSQuery(srv, domain, dns.TypeA)
if err != nil {
t.Errorf("error querying %q: %v", domain, err)
}
if m.Rcode != dns.RcodeServerFailure {
t.Errorf("query %q: expected SERVFAIL, got message: %v", domain, m)
}
}
|
