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
|
package test
import (
"os"
"path/filepath"
"testing"
"github.com/miekg/dns"
)
func setupProxyTargetCoreDNS(t *testing.T, fn func(string)) {
tmpdir := t.TempDir()
content := `
example.org. IN SOA sns.dns.icann.org. noc.dns.icann.org. 1 3600 3600 3600 3600
google.com. IN SOA ns1.google.com. dns-admin.google.com. 1 3600 3600 3600 3600
google.com. IN A 172.217.25.110
`
path := filepath.Join(tmpdir, "file")
if err := os.WriteFile(path, []byte(content), 0644); err != nil {
t.Fatalf("Could not write to temp file: %s", err)
}
defer os.Remove(path)
corefile := `.:0 {
file ` + path + `
}`
i, udp, _, err := CoreDNSServerAndPorts(corefile)
if err != nil {
t.Fatalf("Could not get proxy target CoreDNS serving instance: %s", err)
}
defer i.Stop()
fn(udp)
}
func TestLookupAutoPathErratic(t *testing.T) {
setupProxyTargetCoreDNS(t, func(proxyPath string) {
corefile := `.:0 {
erratic
autopath @erratic
forward . ` + proxyPath + `
debug
}`
i, udp, _, err := CoreDNSServerAndPorts(corefile)
if err != nil {
t.Fatalf("Could not get CoreDNS serving instance: %s", err)
}
defer i.Stop()
tests := []struct {
qname string
expectedAnswer string
expectedType uint16
}{
{"google.com.a.example.org.", "google.com.a.example.org.", dns.TypeCNAME},
{"google.com.", "google.com.", dns.TypeA},
}
for i, tc := range tests {
m := new(dns.Msg)
// erratic always returns this search path: "a.example.org.", "b.example.org.", "".
m.SetQuestion(tc.qname, dns.TypeA)
r, err := dns.Exchange(m, udp)
if err != nil {
t.Fatalf("Test %d, failed to sent query: %q", i, err)
}
if len(r.Answer) == 0 {
t.Fatalf("Test %d, answer section should have RRs", i)
}
if x := r.Answer[0].Header().Name; x != tc.expectedAnswer {
t.Fatalf("Test %d, expected answer %s, got %s", i, tc.expectedAnswer, x)
}
if x := r.Answer[0].Header().Rrtype; x != tc.expectedType {
t.Fatalf("Test %d, expected answer type %d, got %d", i, tc.expectedType, x)
}
}
})
}
func TestAutoPathErraticNotLoaded(t *testing.T) {
setupProxyTargetCoreDNS(t, func(proxyPath string) {
corefile := `.:0 {
autopath @erratic
forward . ` + proxyPath + `
debug
}`
i, err := CoreDNSServer(corefile)
if err != nil {
t.Fatalf("Could not get CoreDNS serving instance: %s", err)
}
udp, _ := CoreDNSServerPorts(i, 0)
if udp == "" {
t.Fatalf("Could not get UDP listening port")
}
defer i.Stop()
m := new(dns.Msg)
m.SetQuestion("google.com.a.example.org.", dns.TypeA)
r, err := dns.Exchange(m, udp)
if err != nil {
t.Fatalf("Failed to sent query: %q", err)
}
if r.Rcode != dns.RcodeNameError {
t.Fatalf("Expected NXDOMAIN, got %d", r.Rcode)
}
})
}
|
