Revert "plugin_block_name: make the blocking code reusable"

This reverts commit f76e0fd8cf.

dnscrypt-proxy/plugin_block_name.go

@@ -13,17 +13,13 @@ import (
 	lumberjack "gopkg.in/natefinch/lumberjack.v2"
 )
 
-type BlockedNames struct {
+type PluginBlockName struct {
 	allWeeklyRanges *map[string]WeeklyRanges
 	patternMatcher  *PatternMatcher
 	logger          *lumberjack.Logger
 	format          string
 }
 
-type PluginBlockName struct {
-	blockedNames *BlockedNames
-}
-
 func (plugin *PluginBlockName) Name() string {
 	return "block_name"
 }
@@ -38,10 +34,8 @@ func (plugin *PluginBlockName) Init(proxy *Proxy) error {
 	if err != nil {
 		return err
 	}
-	blockedNames := BlockedNames{
+	plugin.allWeeklyRanges = proxy.allWeeklyRanges
-		allWeeklyRanges: proxy.allWeeklyRanges,
+	plugin.patternMatcher = NewPatternPatcher()
-		patternMatcher:  NewPatternPatcher(),
-	}
 	for lineNo, line := range strings.Split(string(bin), "\n") {
 		line = strings.TrimFunc(line, unicode.IsSpace)
 		if len(line) == 0 || strings.HasPrefix(line, "#") {
@@ -58,14 +52,14 @@ func (plugin *PluginBlockName) Init(proxy *Proxy) error {
 		}
 		var weeklyRanges *WeeklyRanges
 		if len(timeRangeName) > 0 {
-			weeklyRangesX, ok := (*blockedNames.allWeeklyRanges)[timeRangeName]
+			weeklyRangesX, ok := (*plugin.allWeeklyRanges)[timeRangeName]
 			if !ok {
 				dlog.Errorf("Time range [%s] not found at line %d", timeRangeName, 1+lineNo)
 			} else {
 				weeklyRanges = &weeklyRangesX
 			}
 		}
-		if err := blockedNames.patternMatcher.Add(line, weeklyRanges, lineNo+1); err != nil {
+		if err := plugin.patternMatcher.Add(line, weeklyRanges, lineNo+1); err != nil {
 			dlog.Error(err)
 			continue
 		}
@@ -73,9 +67,8 @@ func (plugin *PluginBlockName) Init(proxy *Proxy) error {
 	if len(proxy.blockNameLogFile) == 0 {
 		return nil
 	}
-	blockedNames.logger = &lumberjack.Logger{LocalTime: true, MaxSize: proxy.logMaxSize, MaxAge: proxy.logMaxAge, MaxBackups: proxy.logMaxBackups, Filename: proxy.blockNameLogFile, Compress: true}
+	plugin.logger = &lumberjack.Logger{LocalTime: true, MaxSize: proxy.logMaxSize, MaxAge: proxy.logMaxAge, MaxBackups: proxy.logMaxBackups, Filename: proxy.blockNameLogFile, Compress: true}
-	blockedNames.format = proxy.blockNameFormat
+	plugin.format = proxy.blockNameFormat
-	plugin.blockedNames = &blockedNames
 
 	return nil
 }
@@ -97,11 +90,7 @@ func (plugin *PluginBlockName) Eval(pluginsState *PluginsState, msg *dns.Msg) er
 		return nil
 	}
 	qName := strings.ToLower(StripTrailingDot(questions[0].Name))
-	return plugin.blockedNames.check(pluginsState, qName)
+	reject, reason, xweeklyRanges := plugin.patternMatcher.Eval(qName)
-}
-
-func (blockedNames *BlockedNames) check(pluginsState *PluginsState, qName string) error {
-	reject, reason, xweeklyRanges := blockedNames.patternMatcher.Eval(qName)
 	var weeklyRanges *WeeklyRanges
 	if xweeklyRanges != nil {
 		weeklyRanges = xweeklyRanges.(*WeeklyRanges)
@@ -111,12 +100,10 @@ func (blockedNames *BlockedNames) check(pluginsState *PluginsState, qName string
 			reject = false
 		}
 	}
-	if !reject {
+	if reject {
-		return nil
-	}
 		pluginsState.action = PluginsActionReject
 		pluginsState.returnCode = PluginsReturnCodeReject
-	if blockedNames.logger != nil {
+		if plugin.logger != nil {
 			var clientIPStr string
 			if pluginsState.clientProto == "udp" {
 				clientIPStr = (*pluginsState.clientAddr).(*net.UDPAddr).IP.String()
@@ -124,21 +111,22 @@ func (blockedNames *BlockedNames) check(pluginsState *PluginsState, qName string
 				clientIPStr = (*pluginsState.clientAddr).(*net.TCPAddr).IP.String()
 			}
 			var line string
-		if blockedNames.format == "tsv" {
+			if plugin.format == "tsv" {
 				now := time.Now()
 				year, month, day := now.Date()
 				hour, minute, second := now.Clock()
 				tsStr := fmt.Sprintf("[%d-%02d-%02d %02d:%02d:%02d]", year, int(month), day, hour, minute, second)
 				line = fmt.Sprintf("%s\t%s\t%s\t%s\n", tsStr, clientIPStr, StringQuote(qName), StringQuote(reason))
-		} else if blockedNames.format == "ltsv" {
+			} else if plugin.format == "ltsv" {
 				line = fmt.Sprintf("time:%d\thost:%s\tqname:%s\tmessage:%s\n", time.Now().Unix(), clientIPStr, StringQuote(qName), StringQuote(reason))
 			} else {
-			dlog.Fatalf("Unexpected log format: [%s]", blockedNames.format)
+				dlog.Fatalf("Unexpected log format: [%s]", plugin.format)
 			}
-		if blockedNames.logger == nil {
+			if plugin.logger == nil {
 				return errors.New("Log file not initialized")
 			}
-		blockedNames.logger.Write([]byte(line))
+			plugin.logger.Write([]byte(line))
+		}
 	}
 	return nil
 }