func fileParse(c *Controller) (file.Zones, error) { z := make(map[string]*file.Zone) names := []string{} for c.Next() { if c.Val() == "file" { // file db.file [zones...] if !c.NextArg() { return file.Zones{}, c.ArgErr() } fileName := c.Val() origins := c.ServerBlockHosts args := c.RemainingArgs() if len(args) > 0 { origins = args } reader, err := os.Open(fileName) if err != nil { // bail out return file.Zones{}, err } for i, _ := range origins { origins[i] = middleware.Host(origins[i]).Normalize() zone, err := file.Parse(reader, origins[i], fileName) if err == nil { z[origins[i]] = zone } else { return file.Zones{}, err } names = append(names, origins[i]) } noReload := false for c.NextBlock() { t, _, e := transferParse(c) if e != nil { return file.Zones{}, e } switch c.Val() { case "no_reload": noReload = true } // discard from, here, maybe check and show log when we do? for _, origin := range origins { if t != nil { z[origin].TransferTo = append(z[origin].TransferTo, t...) } z[origin].NoReload = noReload } } } } return file.Zones{Z: z, Names: names}, nil }
func TestLookupZone(t *testing.T) { zone, err := file.Parse(strings.NewReader(dbMiekNL), "miek.nl.", "stdin") if err != nil { return } fm := file.File{Next: test.ErrorHandler(), Zones: file.Zones{Z: map[string]*file.Zone{"miek.nl.": zone}, Names: []string{"miek.nl."}}} dnskey, rm1, rm2 := newKey(t) defer rm1() defer rm2() dh := NewDnssec([]string{"miek.nl."}, []*DNSKEY{dnskey}, fm) ctx := context.TODO() for _, tc := range dnsTestCases { m := tc.Msg() rec := middleware.NewResponseRecorder(&test.ResponseWriter{}) _, err := dh.ServeDNS(ctx, rec, m) if err != nil { t.Errorf("expected no error, got %v\n", err) return } resp := rec.Msg() sort.Sort(test.RRSet(resp.Answer)) sort.Sort(test.RRSet(resp.Ns)) sort.Sort(test.RRSet(resp.Extra)) if !test.Header(t, tc, resp) { t.Logf("%v\n", resp) continue } if !test.Section(t, tc, test.Answer, resp.Answer) { t.Logf("%v\n", resp) } if !test.Section(t, tc, test.Ns, resp.Ns) { t.Logf("%v\n", resp) } if !test.Section(t, tc, test.Extra, resp.Extra) { t.Logf("%v\n", resp) } } }