Add PowerDNS tags

dmachard · Apr 8, 2022 · dd26921 · dd26921
1 parent 45f5ad6
commit dd26921
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/dnsutils/message.go b/dnsutils/message.go
@@ -94,12 +94,17 @@ type DnsTap struct {
 	LatencySec       string  `json:"latency" msgpack:"latency"`
 }
 
+type PowerDns struct {
+	Tags []string `json:"tags" msgpack:"tags"`
+}
+
 type DnsMessage struct {
 	NetworkInfo DnsNetInfo  `json:"network" msgpack:"network"`
 	DNS         Dns         `json:"dns" msgpack:"dns"`
 	EDNS        DnsExtended `json:"edns" msgpack:"edns"`
 	DnsTap      DnsTap      `json:"dnstap" msgpack:"dnstap"`
 	Geo         DnsGeo      `json:"geo" msgpack:"geo"`
+	PowerDns    PowerDns    `json:"pdns" msgpack:"pdns"`
 }
 
 func (dm *DnsMessage) Init() {
@@ -146,6 +151,10 @@ func (dm *DnsMessage) Init() {
 		City:           "-",
 		Continent:      "-",
 	}
+
+	dm.PowerDns = PowerDns{
+		Tags: []string{},
+	}
 }
 
 func (dm *DnsMessage) Bytes(format []string, delimiter string) []byte {
@@ -263,6 +272,14 @@ func (dm *DnsMessage) Bytes(format []string, delimiter string) []byte {
 			} else {
 				s.WriteString("-")
 			}
+		case "tags":
+			if len(dm.PowerDns.Tags) > 0 {
+				for _, tag := range dm.PowerDns.Tags {
+					s.WriteString(tag)
+				}
+			} else {
+				s.WriteString("-")
+			}
 		default:
 			log.Fatalf("unsupport directive for text format: %s", word)
 		}

diff --git a/subprocessors/powerdns.go b/subprocessors/powerdns.go
@@ -145,6 +145,12 @@ func (d *PdnsProcessor) Run(sendTo []chan dnsutils.DnsMessage) {
 		// get query type
 		dm.DNS.Qtype = dnsutils.RdatatypeToString(int(pbdm.Question.GetQType()))
 
+		// get PowerDNS tags
+		pdns := dnsutils.PowerDns{
+			Tags: pbdm.Response.GetTags(),
+		}
+		dm.PowerDns = pdns
+
 		// decode answers
 		answers := []dnsutils.DnsAnswer{}
 		RRs := pbdm.GetResponse().GetRrs()