diff --git a/coredns_plugin/coredns_plugin.go b/coredns_plugin/coredns_plugin.go index 520859b..7ca4539 100644 --- a/coredns_plugin/coredns_plugin.go +++ b/coredns_plugin/coredns_plugin.go @@ -75,7 +75,7 @@ func setupPlugin(c *caddy.Controller) (*plug, error) { // create new Plugin and copy default values p := &plug{ settings: defaultPluginSettings, - d: dnsfilter.New(), + d: dnsfilter.New(nil), } log.Println("Initializing the CoreDNS plugin") @@ -86,7 +86,7 @@ func setupPlugin(c *caddy.Controller) (*plug, error) { switch blockValue { case "safebrowsing": log.Println("Browsing security service is enabled") - p.d.EnableSafeBrowsing() + p.d.SafeBrowsingEnabled = true if c.NextArg() { if len(c.Val()) == 0 { return nil, c.ArgErr() @@ -95,7 +95,7 @@ func setupPlugin(c *caddy.Controller) (*plug, error) { } case "safesearch": log.Println("Safe search is enabled") - p.d.EnableSafeSearch() + p.d.SafeSearchEnabled = true case "parental": if !c.NextArg() { return nil, c.ArgErr() @@ -106,10 +106,11 @@ func setupPlugin(c *caddy.Controller) (*plug, error) { } log.Println("Parental control is enabled") - err = p.d.EnableParental(sensitivity) - if err != nil { - return nil, c.ArgErr() + if !dnsfilter.IsParentalSensitivityValid(sensitivity) { + return nil, dnsfilter.ErrInvalidParental } + p.d.ParentalEnabled = true + p.d.ParentalSensitivity = sensitivity if c.NextArg() { if len(c.Val()) == 0 { return nil, c.ArgErr()