160 lines
3.3 KiB
Go
160 lines
3.3 KiB
Go
package dns_query
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
"github.com/miekg/dns"
|
|
"net"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type DnsQuery struct {
|
|
// Domains or subdomains to query
|
|
Domains []string
|
|
|
|
// Server to query
|
|
Servers []string
|
|
|
|
// Record type
|
|
RecordType string `toml:"record_type"`
|
|
|
|
// DNS server port number
|
|
Port int
|
|
|
|
// Dns query timeout in seconds. 0 means no timeout
|
|
Timeout int
|
|
}
|
|
|
|
var sampleConfig = `
|
|
### servers to query
|
|
servers = ["8.8.8.8"] # required
|
|
|
|
### Domains or subdomains to query. "."(root) is default
|
|
domains = ["mjasion.pl"] #optional
|
|
|
|
### Query record type. Posible values: A, AAAA, CNAME, MX, NS, PTR, TXT, SOA, SPF, SRV. Default is "NS"
|
|
recordType = "A" # optional
|
|
|
|
### Dns server port. 53 is default
|
|
port = 53 # optional
|
|
|
|
### Query timeout in seconds. Default is 2 seconds
|
|
timeout = 2 # optional
|
|
`
|
|
|
|
func (d *DnsQuery) SampleConfig() string {
|
|
return sampleConfig
|
|
}
|
|
|
|
func (d *DnsQuery) Description() string {
|
|
return "Query given DNS server and gives statistics"
|
|
}
|
|
func (d *DnsQuery) Gather(acc telegraf.Accumulator) error {
|
|
d.setDefaultValues()
|
|
for _, domain := range d.Domains {
|
|
for _, server := range d.Servers {
|
|
dnsQueryTime, err := d.getDnsQueryTime(domain, server)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tags := map[string]string{
|
|
"server": server,
|
|
"domain": domain,
|
|
"record_type": d.RecordType,
|
|
}
|
|
|
|
fields := map[string]interface{}{"query_time_ms": dnsQueryTime}
|
|
acc.AddFields("dns_query", fields, tags)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *DnsQuery) setDefaultValues() {
|
|
if len(d.RecordType) == 0 {
|
|
d.RecordType = "NS"
|
|
}
|
|
|
|
if len(d.Domains) == 0 {
|
|
d.Domains = []string{"."}
|
|
d.RecordType = "NS"
|
|
}
|
|
|
|
if d.Port == 0 {
|
|
d.Port = 53
|
|
}
|
|
|
|
if d.Timeout == 0 {
|
|
d.Timeout = 2
|
|
}
|
|
}
|
|
|
|
func (d *DnsQuery) getDnsQueryTime(domain string, server string) (float64, error) {
|
|
dnsQueryTime := float64(0)
|
|
|
|
c := new(dns.Client)
|
|
c.ReadTimeout = time.Duration(d.Timeout) * time.Second
|
|
|
|
m := new(dns.Msg)
|
|
recordType, err := d.parseRecordType()
|
|
if err != nil {
|
|
return dnsQueryTime, err
|
|
}
|
|
m.SetQuestion(dns.Fqdn(domain), recordType)
|
|
m.RecursionDesired = true
|
|
|
|
r, rtt, err := c.Exchange(m, net.JoinHostPort(server, strconv.Itoa(d.Port)))
|
|
if err != nil {
|
|
return dnsQueryTime, err
|
|
}
|
|
if r.Rcode != dns.RcodeSuccess {
|
|
return dnsQueryTime, errors.New(fmt.Sprintf("Invalid answer name %s after %s query for %s\n", domain, d.RecordType, domain))
|
|
}
|
|
dnsQueryTime = float64(rtt.Nanoseconds()) / 1e6
|
|
return dnsQueryTime, nil
|
|
}
|
|
|
|
func (d *DnsQuery) parseRecordType() (uint16, error) {
|
|
var recordType uint16
|
|
var error error
|
|
|
|
switch d.RecordType {
|
|
case "A":
|
|
recordType = dns.TypeA
|
|
case "AAAA":
|
|
recordType = dns.TypeAAAA
|
|
case "ANY":
|
|
recordType = dns.TypeANY
|
|
case "CNAME":
|
|
recordType = dns.TypeCNAME
|
|
case "MX":
|
|
recordType = dns.TypeMX
|
|
case "NS":
|
|
recordType = dns.TypeNS
|
|
case "PTR":
|
|
recordType = dns.TypePTR
|
|
case "SOA":
|
|
recordType = dns.TypeSOA
|
|
case "SPF":
|
|
recordType = dns.TypeSPF
|
|
case "SRV":
|
|
recordType = dns.TypeSRV
|
|
case "TXT":
|
|
recordType = dns.TypeTXT
|
|
default:
|
|
error = errors.New(fmt.Sprintf("Record type %s not recognized", d.RecordType))
|
|
}
|
|
|
|
return recordType, error
|
|
}
|
|
|
|
func init() {
|
|
inputs.Add("dns_query", func() telegraf.Input {
|
|
return &DnsQuery{}
|
|
})
|
|
}
|