AdGuardHome/querylog/search_criteria.go

package querylog

import (
	"strings"

	"github.com/AdguardTeam/AdGuardHome/dnsfilter"
)

type criteriaType int

const (
	ctDomain          criteriaType = iota // domain name
	ctClient                              // client IP address
	ctQuestionType                        // question type
	ctFilteringStatus                     // filtering status
)

const (
	filteringStatusAll      = "all"
	filteringStatusFiltered = "filtered" // all kinds of filtering

	filteringStatusBlocked             = "blocked"              // blocked or blocked service
	filteringStatusBlockedSafebrowsing = "blocked_safebrowsing" // blocked by safebrowsing
	filteringStatusBlockedParental     = "blocked_parental"     // blocked by parental control
	filteringStatusWhitelisted         = "whitelisted"          // whitelisted
	filteringStatusRewritten           = "rewritten"            // all kinds of rewrites
	filteringStatusSafeSearch          = "safe_search"          // enforced safe search
)

// filteringStatusValues -- array with all possible filteringStatus values
var filteringStatusValues = []string{
	filteringStatusAll, filteringStatusFiltered, filteringStatusBlocked,
	filteringStatusBlockedSafebrowsing, filteringStatusBlockedParental,
	filteringStatusWhitelisted, filteringStatusRewritten, filteringStatusSafeSearch,
}

// searchCriteria - every search request may contain a list of different search criteria
// we use each of them to match the query
type searchCriteria struct {
	criteriaType criteriaType // type of the criteria
	strict       bool         // should we strictly match (equality) or not (indexOf)
	value        string       // search criteria value
}

// quickMatch - quickly checks if the log entry matches this search criteria
// the reason is to do it as quickly as possible without de-serializing the entry
func (c *searchCriteria) quickMatch(line string) bool {
	// note that we do this only for a limited set of criteria

	switch c.criteriaType {
	case ctDomain:
		return c.quickMatchJSONValue(line, "QH")
	case ctClient:
		return c.quickMatchJSONValue(line, "IP")
	case ctQuestionType:
		return c.quickMatchJSONValue(line, "QT")
	default:
		return true
	}
}

// quickMatchJSONValue - helper used by quickMatch
func (c *searchCriteria) quickMatchJSONValue(line string, propertyName string) bool {
	val := readJSONValue(line, propertyName)
	if len(val) == 0 {
		return false
	}

	if c.strict && c.value == val {
		return true
	}
	if !c.strict && strings.Contains(val, c.value) {
		return true
	}

	return false
}

// match - checks if the log entry matches this search criteria
// nolint (gocyclo)
func (c *searchCriteria) match(entry *logEntry) bool {
	switch c.criteriaType {
	case ctDomain:
		if c.strict && entry.QHost == c.value {
			return true
		}
		if !c.strict && strings.Contains(entry.QHost, c.value) {
			return true
		}
		return false
	case ctClient:
		if c.strict && entry.IP == c.value {
			return true
		}
		if !c.strict && strings.Contains(entry.IP, c.value) {
			return true
		}
		return false
	case ctQuestionType:
		if c.strict && entry.QType == c.value {
			return true
		}
		if !c.strict && strings.Contains(entry.QType, c.value) {
			return true
		}
	case ctFilteringStatus:
		res := entry.Result

		switch c.value {
		case filteringStatusAll:
			return true
		case filteringStatusFiltered:
			return res.IsFiltered
		case filteringStatusBlocked:
			return res.IsFiltered &&
				(res.Reason == dnsfilter.FilteredBlackList ||
					res.Reason == dnsfilter.FilteredBlockedService)
		case filteringStatusBlockedParental:
			return res.IsFiltered && res.Reason == dnsfilter.FilteredParental
		case filteringStatusBlockedSafebrowsing:
			return res.IsFiltered && res.Reason == dnsfilter.FilteredSafeBrowsing
		case filteringStatusWhitelisted:
			return res.IsFiltered && res.Reason == dnsfilter.NotFilteredWhiteList
		case filteringStatusRewritten:
			return res.IsFiltered &&
				(res.Reason == dnsfilter.ReasonRewrite ||
					res.Reason == dnsfilter.RewriteEtcHosts)
		case filteringStatusSafeSearch:
			return res.IsFiltered && res.Reason == dnsfilter.FilteredSafeSearch
		default:
			return false
		}

	default:
		return false
	}

	return false
}