1package dns
2
3import (
4	"fmt"
5	"io/ioutil"
6	"os"
7	"path/filepath"
8	"strings"
9	"testing"
10)
11
12func TestGenerateRangeGuard(t *testing.T) {
13	tmpdir, err := ioutil.TempDir("", "dns")
14	if err != nil {
15		t.Fatalf("could not create tmpdir for test: %v", err)
16	}
17	defer os.RemoveAll(tmpdir)
18
19	for i := 0; i <= 1; i++ {
20		path := filepath.Join(tmpdir, fmt.Sprintf("%04d.conf", i))
21		data := []byte(fmt.Sprintf("dhcp-%04d A 10.0.0.%d", i, i))
22
23		if err := ioutil.WriteFile(path, data, 0644); err != nil {
24			t.Fatalf("could not create tmpfile for test: %v", err)
25		}
26	}
27
28	var tests = [...]struct {
29		zone string
30		fail bool
31	}{
32		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
33$GENERATE 0-1 dhcp-${0,4,d} A 10.0.0.$
34`, false},
35		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
36$GENERATE 0-1 dhcp-${0,0,x} A 10.0.0.$
37`, false},
38		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
39$GENERATE 128-129 dhcp-${-128,4,d} A 10.0.0.$
40`, false},
41		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
42$GENERATE 128-129 dhcp-${-129,4,d} A 10.0.0.$
43`, true},
44		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
45$GENERATE 0-2 dhcp-${2147483647,4,d} A 10.0.0.$
46`, true},
47		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
48$GENERATE 0-1 dhcp-${2147483646,4,d} A 10.0.0.$
49`, false},
50		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
51$GENERATE 0-1/step dhcp-${0,4,d} A 10.0.0.$
52`, true},
53		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
54$GENERATE 0-1/ dhcp-${0,4,d} A 10.0.0.$
55`, true},
56		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
57$GENERATE 0-10/2 dhcp-${0,4,d} A 10.0.0.$
58`, false},
59		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
60$GENERATE 0-1/0 dhcp-${0,4,d} A 10.0.0.$
61`, true},
62		{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
63$GENERATE 0-1 $$INCLUDE ` + tmpdir + string(filepath.Separator) + `${0,4,d}.conf
64`, false},
65{`@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
66$GENERATE 0-1 dhcp-${0,4,d} A 10.0.0.$
67$GENERATE 0-2 dhcp-${0,4,d} A 10.1.0.$
68`, false},
69	}
70Outer:
71	for i := range tests {
72		for tok := range ParseZone(strings.NewReader(tests[i].zone), "test.", "test") {
73			if tok.Error != nil {
74				if !tests[i].fail {
75					t.Errorf("expected \n\n%s\nto be parsed, but got %v", tests[i].zone, tok.Error)
76				}
77				continue Outer
78			}
79		}
80		if tests[i].fail {
81			t.Errorf("expected \n\n%s\nto fail, but got no error", tests[i].zone)
82		}
83	}
84}
85
86func TestGenerateIncludeDepth(t *testing.T) {
87	tmpfile, err := ioutil.TempFile("", "dns")
88	if err != nil {
89		t.Fatalf("could not create tmpfile for test: %v", err)
90	}
91	defer os.Remove(tmpfile.Name())
92
93	zone := `@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
94$GENERATE 0-1 $$INCLUDE ` + tmpfile.Name() + `
95`
96	if _, err := tmpfile.WriteString(zone); err != nil {
97		t.Fatalf("could not write to tmpfile for test: %v", err)
98	}
99	if err := tmpfile.Close(); err != nil {
100		t.Fatalf("could not close tmpfile for test: %v", err)
101	}
102
103	zp := NewZoneParser(strings.NewReader(zone), ".", tmpfile.Name())
104	zp.SetIncludeAllowed(true)
105
106	for _, ok := zp.Next(); ok; _, ok = zp.Next() {
107	}
108
109	const expected = "too deeply nested $INCLUDE"
110	if err := zp.Err(); err == nil || !strings.Contains(err.Error(), expected) {
111		t.Errorf("expected error to include %q, got %v", expected, err)
112	}
113}
114
115func TestGenerateIncludeDisallowed(t *testing.T) {
116	const zone = `@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
117$GENERATE 0-1 $$INCLUDE test.conf
118`
119	zp := NewZoneParser(strings.NewReader(zone), ".", "")
120
121	for _, ok := zp.Next(); ok; _, ok = zp.Next() {
122	}
123
124	const expected = "$INCLUDE directive not allowed"
125	if err := zp.Err(); err == nil || !strings.Contains(err.Error(), expected) {
126		t.Errorf("expected error to include %q, got %v", expected, err)
127	}
128}
129
130func TestGenerateSurfacesErrors(t *testing.T) {
131	const zone = `@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
132$GENERATE 0-1 dhcp-${0,4,dd} A 10.0.0.$
133`
134	zp := NewZoneParser(strings.NewReader(zone), ".", "test")
135
136	for _, ok := zp.Next(); ok; _, ok = zp.Next() {
137	}
138
139	const expected = `test: dns: bad base in $GENERATE: "${0,4,dd}" at line: 2:20`
140	if err := zp.Err(); err == nil || err.Error() != expected {
141		t.Errorf("expected specific error, wanted %q, got %v", expected, err)
142	}
143}
144
145func TestGenerateSurfacesLexerErrors(t *testing.T) {
146	const zone = `@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
147$GENERATE 0-1 dhcp-${0,4,d} A 10.0.0.$ )
148`
149	zp := NewZoneParser(strings.NewReader(zone), ".", "test")
150
151	for _, ok := zp.Next(); ok; _, ok = zp.Next() {
152	}
153
154	const expected = `test: dns: bad data in $GENERATE directive: "extra closing brace" at line: 2:40`
155	if err := zp.Err(); err == nil || err.Error() != expected {
156		t.Errorf("expected specific error, wanted %q, got %v", expected, err)
157	}
158}
159
160func TestGenerateModToPrintf(t *testing.T) {
161	tests := []struct {
162		mod        string
163		wantFmt    string
164		wantOffset int
165		wantErr    bool
166	}{
167		{"0,0,d", "%d", 0, false},
168		{"0,0", "%d", 0, false},
169		{"0", "%d", 0, false},
170		{"3,2,d", "%02d", 3, false},
171		{"3,2", "%02d", 3, false},
172		{"3", "%d", 3, false},
173		{"0,0,o", "%o", 0, false},
174		{"0,0,x", "%x", 0, false},
175		{"0,0,X", "%X", 0, false},
176		{"0,0,z", "", 0, true},
177		{"0,0,0,d", "", 0, true},
178		{"-100,0,d", "%d", -100, false},
179	}
180	for _, test := range tests {
181		gotFmt, gotOffset, errMsg := modToPrintf(test.mod)
182		switch {
183		case errMsg != "" && !test.wantErr:
184			t.Errorf("modToPrintf(%q) - expected empty-error, but got %v", test.mod, errMsg)
185		case errMsg == "" && test.wantErr:
186			t.Errorf("modToPrintf(%q) - expected error, but got empty-error", test.mod)
187		case gotFmt != test.wantFmt:
188			t.Errorf("modToPrintf(%q) - expected format %q, but got %q", test.mod, test.wantFmt, gotFmt)
189		case gotOffset != test.wantOffset:
190			t.Errorf("modToPrintf(%q) - expected offset %d, but got %d", test.mod, test.wantOffset, gotOffset)
191		}
192	}
193}
194
195func BenchmarkGenerate(b *testing.B) {
196	const zone = `@ IN SOA ns.test. hostmaster.test. ( 1 8h 2h 7d 1d )
197$GENERATE 32-158 dhcp-${-32,4,d} A 10.0.0.$
198`
199
200	for n := 0; n < b.N; n++ {
201		zp := NewZoneParser(strings.NewReader(zone), ".", "")
202
203		for _, ok := zp.Next(); ok; _, ok = zp.Next() {
204		}
205
206		if err := zp.Err(); err != nil {
207			b.Fatal(err)
208		}
209	}
210}
211
212func TestCrasherString(t *testing.T) {
213	tests := []struct {
214		in  string
215		err string
216	}{
217		{"$GENERATE 0-300103\"$$GENERATE 2-2", "bad range in $GENERATE"},
218		{"$GENERATE 0-5414137360", "bad range in $GENERATE"},
219		{"$GENERATE       11522-3668518066406258", "bad range in $GENERATE"},
220		{"$GENERATE 0-200\"(;00000000000000\n$$GENERATE 0-0", "dns: garbage after $GENERATE range: \"\\\"\" at line: 1:16"},
221		{"$GENERATE 6-2048 $$GENERATE 6-036160 $$$$ORIGIN \\$", `dns: nested $GENERATE directive not allowed: "6-036160" at line: 1:19`},
222	}
223	for _, tc := range tests {
224		t.Run(tc.in, func(t *testing.T) {
225			_, err := NewRR(tc.in)
226			if err == nil {
227				t.Errorf("Expecting error for crasher line %s", tc.in)
228			}
229			if !strings.Contains(err.Error(), tc.err) {
230				t.Errorf("Expecting error %s, got %s", tc.err, err.Error())
231			}
232		})
233	}
234}
235