Allow disabling of filtering but keeping querylog, safebrowsing, safesearch and parental working.

config.go

@@ -160,16 +160,16 @@ func writeAllConfigs() error {
 }
 
 const coreDNSConfigTemplate = `. {
-    {{if .FilteringEnabled}}dnsfilter {{.FilterFile}} {
+    dnsfilter {{if .FilteringEnabled}}{{.FilterFile}}{{end}} {
         {{if .SafeBrowsingEnabled}}safebrowsing{{end}}
         {{if .ParentalEnabled}}parental {{.ParentalSensitivity}}{{end}}
         {{if .SafeSearchEnabled}}safesearch{{end}}
         {{if .QueryLogEnabled}}querylog{{end}}
-    }{{end}}
+    }
     {{.Pprof}}
-	hosts {
+    hosts {
-		fallthrough
+        fallthrough
-	}
+    }
     {{if .UpstreamDNS}}forward . {{range .UpstreamDNS}}{{.}} {{end}}{{end}}
     {{.Cache}}
     {{.Prometheus}}

coredns_plugin/coredns_plugin.go

@@ -92,14 +92,12 @@ func setupPlugin(c *caddy.Controller) (*plug, error) {
 	p.d = dnsfilter.New()
 	p.hosts = make(map[string]net.IP)
 
-	var filterFileName string
+	filterFileNames := []string{}
 	for c.Next() {
 		args := c.RemainingArgs()
-		if len(args) == 0 {
+		if len(args) > 0 {
-			// must have at least one argument
+			filterFileNames = append(filterFileNames, args...)
-			return nil, c.ArgErr()
 		}
-		filterFileName = args[0]
 		for c.NextBlock() {
 			switch c.Val() {
 			case "safebrowsing":
@@ -139,34 +137,39 @@ func setupPlugin(c *caddy.Controller) (*plug, error) {
 		}
 	}
 
-	file, err := os.Open(filterFileName)
+	log.Printf("filterFileNames = %+v", filterFileNames)
-	if err != nil {
-		return nil, err
-	}
-	defer file.Close()
 
-	count := 0
+	for i, filterFileName := range filterFileNames {
-	scanner := bufio.NewScanner(file)
+		file, err := os.Open(filterFileName)
-	for scanner.Scan() {
-		text := scanner.Text()
-		if p.parseEtcHosts(text) {
-			continue
-		}
-		err = p.d.AddRule(text, 0)
-		if err == dnsfilter.ErrInvalidSyntax {
-			continue
-		}
 		if err != nil {
 			return nil, err
 		}
-		count++
+		defer file.Close()
-	}
-	log.Printf("Added %d rules from %s", count, filterFileName)
 
-	if err = scanner.Err(); err != nil {
+		count := 0
-		return nil, err
+		scanner := bufio.NewScanner(file)
+		for scanner.Scan() {
+			text := scanner.Text()
+			if p.parseEtcHosts(text) {
+				continue
+			}
+			err = p.d.AddRule(text, uint32(i))
+			if err == dnsfilter.ErrInvalidSyntax {
+				continue
+			}
+			if err != nil {
+				return nil, err
+			}
+			count++
+		}
+		log.Printf("Added %d rules from %s", count, filterFileName)
+
+		if err = scanner.Err(); err != nil {
+			return nil, err
+		}
 	}
 
+	var err error
 	p.upstream, err = upstream.New(nil)
 	if err != nil {
 		return nil, err

coredns_plugin/coredns_plugin_test.go

@@ -20,7 +20,8 @@ func TestSetup(t *testing.T) {
 		config  string
 		failing bool
 	}{
-		{`dnsfilter`, true},
+		{`dnsfilter`, false},
+		{`dnsfilter /dev/nonexistent/abcdef`, true},
 		{`dnsfilter ../tests/dns.txt`, false},
 		{`dnsfilter ../tests/dns.txt { safebrowsing }`, false},
 		{`dnsfilter ../tests/dns.txt { parental }`, true},