a
/
badguardhome
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

This commit is contained in:
Eugene Bujak 2018-09-25 19:26:26 +03:00
parent 50b188a086
commit 09a39cce03
3 changed files with 36 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
config.go
View File

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

15
coredns_plugin/coredns_plugin.go
View File

@ -92,14 +92,12 @@ func setupPlugin(c *caddy.Controller) (*plug, error) {
p.d = dnsfilter.New() p.d = dnsfilter.New()
p.hosts = make(map[string]net.IP) p.hosts = make(map[string]net.IP)
var filterFileName string filterFileNames := []string{}
for c.Next() { for c.Next() {
args := c.RemainingArgs() 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() { for c.NextBlock() {
switch c.Val() { switch c.Val() {
case "safebrowsing": case "safebrowsing":
@ -139,6 +137,9 @@ func setupPlugin(c *caddy.Controller) (*plug, error) {
} }
} }
log.Printf("filterFileNames = %+v", filterFileNames)
for i, filterFileName := range filterFileNames {
file, err := os.Open(filterFileName) file, err := os.Open(filterFileName)
if err != nil { if err != nil {
return nil, err return nil, err
@ -152,7 +153,7 @@ func setupPlugin(c *caddy.Controller) (*plug, error) {
if p.parseEtcHosts(text) { if p.parseEtcHosts(text) {
continue continue
} }
err = p.d.AddRule(text, 0) err = p.d.AddRule(text, uint32(i))
if err == dnsfilter.ErrInvalidSyntax { if err == dnsfilter.ErrInvalidSyntax {
continue continue
} }
@ -166,7 +167,9 @@ func setupPlugin(c *caddy.Controller) (*plug, error) {
if err = scanner.Err(); err != nil { if err = scanner.Err(); err != nil {
return nil, err return nil, err
} }
}
var err error
p.upstream, err = upstream.New(nil) p.upstream, err = upstream.New(nil)
if err != nil { if err != nil {
return nil, err return nil, err

3
coredns_plugin/coredns_plugin_test.go
View File

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