sacrebleu-dns/utils/queries.go

70 lines
2.3 KiB
Go
Raw Normal View History

2020-12-13 04:01:04 +01:00
package utils
import (
"fmt"
"strings"
"github.com/go-redis/redis/v8"
"github.com/sirupsen/logrus"
)
//GetRecord : Check the SQL and REDIS database for a Record.
2020-12-14 23:20:24 +01:00
//A Record struct is used as input and output
2020-12-13 04:01:04 +01:00
func GetRecord(entry Record) Record {
//Check for strict record in Redis cache
redisKey := entry.Fqdn + "--" + fmt.Sprint(entry.Qtype)
result, redisErr := redisCheckForRecord(redisKey, entry)
var sqlErr bool //The err returned for sqlCheckForRecord or sqlCheckForReverse6Wildcard
2020-12-13 04:01:04 +01:00
//If reverse DNS
reverseCheck := IsReverse(entry.Fqdn)
if reverseCheck > 0 {
2020-12-14 23:20:24 +01:00
//If reverse record not found in redis
2020-12-13 04:01:04 +01:00
if redisErr == redis.Nil {
2020-12-14 23:20:24 +01:00
//Check for it in the SQL database
2020-12-13 04:01:04 +01:00
logrus.Debug("QUERIES : Check for strict reverse in MySQL")
result, sqlErr = sqlCheckForRecord(redisKey, entry.Fqdn, entry)
if sqlErr {
2020-12-14 23:20:24 +01:00
//Check for wildcard reverse in the SQL
2020-12-13 04:01:04 +01:00
logrus.Debug("QUERIES : Check for wildcard reverse in MySQL")
2020-12-13 19:43:31 +01:00
result, _ = sqlCheckForReverse6Wildcard(redisKey, entry.Fqdn, entry)
2020-12-13 04:01:04 +01:00
}
}
//For dynamic reverse dns
2020-12-14 23:20:24 +01:00
//Check for it by looking for a "%s" in the record content
//If true, replace it with the formated IP
2020-12-13 04:01:04 +01:00
if strings.Contains(result.Content, "%s") {
record := ExtractAddressFromReverse(entry.Fqdn)
var recordFormated string
if reverseCheck == 1 {
recordFormated = strings.ReplaceAll(record, ".", "-")
2020-12-13 04:01:04 +01:00
} else {
recordFormated = strings.ReplaceAll(record, ":", "-")
2020-12-13 04:01:04 +01:00
}
result.Content = fmt.Sprintf(result.Content, recordFormated)
}
} else if redisErr == redis.Nil { //If strict record NOT in Redis cache & not Reverse
//Check for wildcard in Redis cache
logrus.Debug("QUERIES : Check for wildcard in redis cache")
mainDomainKey := fmt.Sprintf("*%s", entry.Fqdn[strings.Index(entry.Fqdn, "."):]) //Remove the last subdomain
redismdKey := fmt.Sprintf("%s--%v", mainDomainKey, entry.Qtype)
result, redisErr = redisCheckForRecord(redismdKey, entry)
//If none of both check in mysql
if redisErr == redis.Nil {
//Check for strict record in mysql
logrus.Debug("QUERIES : Check for strict record in MSQL")
result, sqlErr = sqlCheckForRecord(redisKey, entry.Fqdn, entry)
if sqlErr {
2020-12-13 04:01:04 +01:00
//Check for wildcard record in mysql
logrus.Debug("QUERIES : Check for wildcard in MSQL")
2020-12-13 19:43:31 +01:00
result, _ = sqlCheckForRecord(redismdKey, fmt.Sprint(mainDomainKey), entry)
2020-12-13 04:01:04 +01:00
}
}
}
return result
}