From 28f4189630f41153da34cb41a79b43fc468ead3d Mon Sep 17 00:00:00 2001 From: Miek Gieben Date: Thu, 30 Dec 2010 16:44:32 +0100 Subject: [PATCH] Add 'q' a dig-like program (only 100 lines of code) --- _examples/q/q.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/_examples/q/q.go b/_examples/q/q.go index 7fe00907..b850775a 100644 --- a/_examples/q/q.go +++ b/_examples/q/q.go @@ -7,6 +7,7 @@ import ( "os" "flag" "fmt" + "strings" ) var Usage = func() { @@ -35,14 +36,14 @@ FLAGS: } // If it looks like a class, it is a class for k, v := range dns.Class_str { - if v == flag.Arg(i) { + if v == strings.ToUpper(flag.Arg(i)) { qclass = k continue FLAGS } } // And if it looks like type, it is a type for k, v := range dns.Rr_str { - if v == flag.Arg(i) { + if v == strings.ToUpper(flag.Arg(i)) { qtype = k continue FLAGS } @@ -57,7 +58,7 @@ FLAGS: qr := resolver.NewQuerier(r) // @server may be a name, resolv that var err os.Error - _, addr, err := net.LookupHost(nameserver) + _, addr, err := net.LookupHost(string([]byte(nameserver)[1:])) //chop off @ if err == nil { r.Servers = addr } else {