Some minor tweaks

This commit is contained in:
Miek Gieben 2012-01-13 12:36:05 +01:00
parent 6e3385b868
commit 27150f3e0a
1 changed files with 12 additions and 3 deletions

View File

@ -29,7 +29,7 @@ import (
"strconv" "strconv"
) )
const dom = "whoami.miek.nl" const dom = "whoami.miek.nl."
func handleReflect(w dns.ResponseWriter, r *dns.Msg) { func handleReflect(w dns.ResponseWriter, r *dns.Msg) {
var ( var (
@ -65,8 +65,17 @@ func handleReflect(w dns.ResponseWriter, r *dns.Msg) {
t.Hdr = dns.RR_Header{Name: dom, Rrtype: dns.TypeTXT, Class: dns.ClassINET, Ttl: 0} t.Hdr = dns.RR_Header{Name: dom, Rrtype: dns.TypeTXT, Class: dns.ClassINET, Ttl: 0}
t.Txt = str t.Txt = str
m.Extra = append(m.Extra, t) switch r.Question[0].Qtype {
m.Answer = append(m.Answer, rr) case dns.TypeTXT:
m.Answer = append(m.Answer, t)
m.Extra = append(m.Extra, rr)
default: fallthrough
case dns.TypeAAAA, dns.TypeA:
m.Answer = append(m.Answer, rr)
m.Extra = append(m.Extra, t)
}
b, ok := m.Pack() b, ok := m.Pack()
if !ok { if !ok {
log.Print("Packing failed") log.Print("Packing failed")