mirror of
https://github.com/coredns/coredns.git
synced 2025-10-31 10:13:14 -04:00
6445a3f2f0
Make the CH middleware actually work. Needs a bit of a hack to route the fake version.bind and friends zone to the correct handler. Fiddle with the order in directive.go so that CH queries get logged as well. Secondly add class rewriting to the rewrite middleware handler and also log the class by default.
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package chaos
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/miekg/coredns/middleware"
|
|
|
|
"github.com/miekg/dns"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
// Chaos allows CoreDNS to reply to CH TXT queries and return author or
|
|
// version information.
|
|
type Chaos struct {
|
|
Next middleware.Handler
|
|
Version string
|
|
Authors map[string]bool
|
|
}
|
|
|
|
func (c Chaos) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
|
|
state := middleware.State{W: w, Req: r}
|
|
if state.QClass() != dns.ClassCHAOS || state.QType() != dns.TypeTXT {
|
|
return c.Next.ServeDNS(ctx, w, r)
|
|
}
|
|
|
|
m := new(dns.Msg)
|
|
m.SetReply(r)
|
|
|
|
hdr := dns.RR_Header{Name: state.QName(), Rrtype: dns.TypeTXT, Class: dns.ClassCHAOS, Ttl: 0}
|
|
switch state.Name() {
|
|
default:
|
|
return c.Next.ServeDNS(ctx, w, r)
|
|
case "authors.bind.":
|
|
for a, _ := range c.Authors {
|
|
m.Answer = append(m.Answer, &dns.TXT{Hdr: hdr, Txt: []string{trim(a)}})
|
|
}
|
|
case "version.bind.", "version.server.":
|
|
m.Answer = []dns.RR{&dns.TXT{Hdr: hdr, Txt: []string{trim(c.Version)}}}
|
|
case "hostname.bind.", "id.server.":
|
|
hostname, err := os.Hostname()
|
|
if err != nil {
|
|
hostname = "localhost"
|
|
}
|
|
m.Answer = []dns.RR{&dns.TXT{Hdr: hdr, Txt: []string{trim(hostname)}}}
|
|
}
|
|
w.WriteMsg(m)
|
|
return 0, nil
|
|
}
|
|
|
|
func trim(s string) string {
|
|
if len(s) < 256 {
|
|
return s
|
|
}
|
|
return s[:255]
|
|
}
|