mirror of
https://github.com/maride/pancap.git
synced 2024-11-22 08:54:24 +00:00
76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
package dns
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/google/gopacket/layers"
|
|
"github.com/maride/pancap/common"
|
|
"golang.org/x/net/publicsuffix"
|
|
"log"
|
|
)
|
|
|
|
var (
|
|
numQuestions int
|
|
questionDomains []string
|
|
questionBaseDomains []string
|
|
questionPrivateDomains []string
|
|
questionType = make(map[layers.DNSType]int)
|
|
)
|
|
|
|
// Called on every DNS packet to process questions
|
|
func (p *Protocol) processDNSQuestion(questions []layers.DNSQuestion) {
|
|
// Iterate over all questions
|
|
for _, question := range questions {
|
|
// Raise stats
|
|
numQuestions++
|
|
|
|
// Add question to questions array
|
|
name := string(question.Name)
|
|
basename, basenameErr := publicsuffix.EffectiveTLDPlusOne(name)
|
|
|
|
if basenameErr != nil {
|
|
// Encountered error while checking for the basename
|
|
log.Printf("Encountered error while checking '%s' domain for its basename: %s", name, basenameErr.Error())
|
|
continue
|
|
}
|
|
|
|
// Process type questions
|
|
p.processType(questionType, question.Type)
|
|
|
|
// Append full domain and base domain
|
|
questionDomains = common.AppendIfUnique(name, questionDomains)
|
|
|
|
// Check if we need to add the base name to the private list
|
|
_, icannManaged := publicsuffix.PublicSuffix(name)
|
|
if icannManaged {
|
|
// TLD is managed by ICANN, add to the base list
|
|
questionBaseDomains = common.AppendIfUnique(basename, questionBaseDomains)
|
|
} else {
|
|
// it's not managed by ICANN, so it's private - add it to the private list
|
|
questionPrivateDomains = common.AppendIfUnique(name, questionPrivateDomains)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Generates a summary of all DNS questions
|
|
func (p *Protocol) generateDNSQuestionSummary() string {
|
|
summary := ""
|
|
|
|
// Overall question stats
|
|
summary = fmt.Sprintf("%s%d DNS questions in total\n", summary, numQuestions)
|
|
summary = fmt.Sprintf("%s%s records\n", summary, p.generateDNSTypeSummary(questionType))
|
|
summary = fmt.Sprintf("%s%d unique domains of %d base domains, of which are %d private (non-ICANN) TLDs.\n", summary, len(questionDomains), len(questionBaseDomains), len(questionPrivateDomains))
|
|
|
|
// Output base domains asked for
|
|
if len(questionBaseDomains) > 0 {
|
|
summary = fmt.Sprintf("%sAsked for these base domains:\n%s", summary, common.GenerateTree(questionBaseDomains))
|
|
}
|
|
|
|
// Output private domains
|
|
if len(questionPrivateDomains) > 0 {
|
|
summary = fmt.Sprintf("%sAsked for these private (non-ICANN managed) domains:\n%s", summary, common.GenerateTree(questionPrivateDomains))
|
|
}
|
|
|
|
// And return summary
|
|
return summary
|
|
}
|