mirror of https://github.com/daffainfo/nuclei.git
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package executer
|
|
|
|
import (
|
|
"strings"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/projectdiscovery/gologger"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/matchers"
|
|
)
|
|
|
|
// writeOutputDNS writes dns output to streams
|
|
func (e *DNSExecuter) writeOutputDNS(domain string, matcher *matchers.Matcher, extractorResults []string) {
|
|
if e.jsonOutput {
|
|
output := jsonOutput{
|
|
Template: e.template.ID,
|
|
Type: "dns",
|
|
Matched: domain,
|
|
Severity: e.template.Info.Severity,
|
|
Author: e.template.Info.Author,
|
|
Description: e.template.Info.Description,
|
|
}
|
|
if matcher != nil && len(matcher.Name) > 0 {
|
|
output.MatcherName = matcher.Name
|
|
}
|
|
if len(extractorResults) > 0 {
|
|
output.ExtractedResults = extractorResults
|
|
}
|
|
data, err := jsoniter.Marshal(output)
|
|
if err != nil {
|
|
gologger.Warningf("Could not marshal json output: %s\n", err)
|
|
}
|
|
|
|
gologger.Silentf("%s", string(data))
|
|
|
|
if e.writer != nil {
|
|
e.outputMutex.Lock()
|
|
e.writer.Write(data)
|
|
e.writer.WriteRune('\n')
|
|
e.outputMutex.Unlock()
|
|
}
|
|
return
|
|
}
|
|
|
|
builder := &strings.Builder{}
|
|
builder.WriteRune('[')
|
|
builder.WriteString(e.template.ID)
|
|
if matcher != nil && len(matcher.Name) > 0 {
|
|
builder.WriteString(":")
|
|
builder.WriteString(matcher.Name)
|
|
}
|
|
builder.WriteString("] [dns] ")
|
|
|
|
builder.WriteString(domain)
|
|
|
|
// If any extractors, write the results
|
|
if len(extractorResults) > 0 {
|
|
builder.WriteString(" [")
|
|
for i, result := range extractorResults {
|
|
builder.WriteString(result)
|
|
if i != len(extractorResults)-1 {
|
|
builder.WriteRune(',')
|
|
}
|
|
}
|
|
builder.WriteString("]")
|
|
}
|
|
builder.WriteRune('\n')
|
|
|
|
// Write output to screen as well as any output file
|
|
message := builder.String()
|
|
gologger.Silentf("%s", message)
|
|
|
|
if e.writer != nil {
|
|
e.outputMutex.Lock()
|
|
e.writer.WriteString(message)
|
|
e.outputMutex.Unlock()
|
|
}
|
|
}
|