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