From 27150f3e0a769b6a403545fe5b2b411626caf662 Mon Sep 17 00:00:00 2001
From: Miek Gieben <miek@miek.nl>
Date: Fri, 13 Jan 2012 12:36:05 +0100
Subject: [PATCH] Some minor tweaks

---
 _examples/reflect/reflect.go | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/_examples/reflect/reflect.go b/_examples/reflect/reflect.go
index 561486e6..f229f29a 100644
--- a/_examples/reflect/reflect.go
+++ b/_examples/reflect/reflect.go
@@ -29,7 +29,7 @@ import (
 	"strconv"
 )
 
-const dom = "whoami.miek.nl"
+const dom = "whoami.miek.nl."
 
 func handleReflect(w dns.ResponseWriter, r *dns.Msg) {
 	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.Txt = str
 
-	m.Extra = append(m.Extra, t)
-	m.Answer = append(m.Answer, rr)
+        switch r.Question[0].Qtype {
+        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()
 	if !ok {
 		log.Print("Packing failed")