2010-12-31 02:38:16 +11:00
|
|
|
package main
|
2011-02-09 06:22:43 +11:00
|
|
|
|
2010-12-31 02:38:16 +11:00
|
|
|
import (
|
|
|
|
"net"
|
|
|
|
"dns"
|
|
|
|
"os"
|
|
|
|
"flag"
|
|
|
|
"fmt"
|
2010-12-31 04:48:29 +11:00
|
|
|
"strings"
|
2010-12-31 02:38:16 +11:00
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
2010-12-31 06:00:26 +11:00
|
|
|
var dnssec *bool = flag.Bool("dnssec", false, "Request DNSSEC records")
|
2010-12-31 04:48:29 +11:00
|
|
|
var port *string = flag.String("port", "53", "Set the query port")
|
|
|
|
var aa *bool = flag.Bool("aa", false, "Set AA flag in query")
|
|
|
|
var ad *bool = flag.Bool("ad", false, "Set AD flag in query")
|
|
|
|
var cd *bool = flag.Bool("cd", false, "Set CD flag in query")
|
|
|
|
var rd *bool = flag.Bool("rd", true, "Unset RD flag in query")
|
2010-12-31 04:56:12 +11:00
|
|
|
var tcp *bool = flag.Bool("tcp", false, "TCP mode")
|
2011-01-17 20:51:33 +11:00
|
|
|
var nsid *bool = flag.Bool("nsid", false, "Ask for the NSID")
|
2010-12-31 04:48:29 +11:00
|
|
|
flag.Usage = func() {
|
|
|
|
fmt.Fprintf(os.Stderr, "Usage: %s [@server] [qtype] [qclass] [name ...]\n", os.Args[0])
|
|
|
|
flag.PrintDefaults()
|
|
|
|
}
|
2010-12-31 03:15:59 +11:00
|
|
|
|
2010-12-31 04:48:29 +11:00
|
|
|
nameserver := "@127.0.0.1" // Default nameserver
|
2010-12-31 02:38:16 +11:00
|
|
|
qtype := uint16(dns.TypeA) // Default qtype
|
|
|
|
qclass := uint16(dns.ClassINET) // Default qclass
|
|
|
|
var qname []string
|
|
|
|
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
FLAGS:
|
|
|
|
for i := 0; i < flag.NArg(); i++ {
|
|
|
|
// If it starts with @ it is a nameserver
|
|
|
|
if flag.Arg(i)[0] == '@' {
|
|
|
|
nameserver = flag.Arg(i)
|
|
|
|
continue FLAGS
|
|
|
|
}
|
2011-01-19 01:44:30 +11:00
|
|
|
// First class, then type, to make ANY queries possible
|
|
|
|
// And if it looks like type, it is a type
|
|
|
|
for k, v := range dns.Rr_str {
|
2010-12-31 02:44:32 +11:00
|
|
|
if v == strings.ToUpper(flag.Arg(i)) {
|
2011-01-19 01:44:30 +11:00
|
|
|
qtype = k
|
2010-12-31 02:38:16 +11:00
|
|
|
continue FLAGS
|
|
|
|
}
|
|
|
|
}
|
2011-01-19 01:44:30 +11:00
|
|
|
// If it looks like a class, it is a class
|
|
|
|
for k, v := range dns.Class_str {
|
2010-12-31 02:44:32 +11:00
|
|
|
if v == strings.ToUpper(flag.Arg(i)) {
|
2011-01-19 01:44:30 +11:00
|
|
|
qclass = k
|
2010-12-31 02:38:16 +11:00
|
|
|
continue FLAGS
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Anything else is a qname
|
|
|
|
qname = append(qname, flag.Arg(i))
|
|
|
|
}
|
2011-02-09 06:22:43 +11:00
|
|
|
r := new(dns.Resolver)
|
2011-01-17 20:38:25 +11:00
|
|
|
r.FromFile("/etc/resolv.conf")
|
2010-12-31 04:48:29 +11:00
|
|
|
r.Timeout = 2
|
|
|
|
r.Port = *port
|
2010-12-31 04:56:12 +11:00
|
|
|
r.Tcp = *tcp
|
2010-12-31 04:48:29 +11:00
|
|
|
r.Attempts = 1
|
2010-12-31 02:38:16 +11:00
|
|
|
// @server may be a name, resolv that
|
|
|
|
var err os.Error
|
2010-12-31 04:48:29 +11:00
|
|
|
nameserver = string([]byte(nameserver)[1:]) // chop off @
|
2010-12-31 04:42:40 +11:00
|
|
|
_, addr, err := net.LookupHost(nameserver)
|
2010-12-31 04:48:29 +11:00
|
|
|
if err == nil {
|
|
|
|
r.Servers = addr
|
|
|
|
} else {
|
|
|
|
r.Servers = []string{nameserver}
|
|
|
|
}
|
2010-12-31 02:38:16 +11:00
|
|
|
|
|
|
|
m := new(dns.Msg)
|
2010-12-31 04:48:29 +11:00
|
|
|
m.MsgHdr.Authoritative = *aa
|
|
|
|
m.MsgHdr.AuthenticatedData = *ad
|
|
|
|
m.MsgHdr.CheckingDisabled = *cd
|
|
|
|
m.MsgHdr.RecursionDesired = *rd
|
2010-12-31 02:38:16 +11:00
|
|
|
m.Question = make([]dns.Question, 1)
|
2011-01-17 20:51:33 +11:00
|
|
|
if *dnssec || *nsid {
|
2010-12-31 04:48:29 +11:00
|
|
|
opt := new(dns.RR_OPT)
|
|
|
|
opt.Hdr = dns.RR_Header{Name: "", Rrtype: dns.TypeOPT}
|
2011-01-08 23:46:51 +11:00
|
|
|
opt.SetVersion(0)
|
|
|
|
opt.SetDo()
|
2011-01-27 19:40:18 +11:00
|
|
|
opt.SetUDPSize(dns.DefaultMsgSize)
|
2011-01-17 20:51:33 +11:00
|
|
|
if *nsid {
|
|
|
|
opt.Option = make([]dns.Option, 1)
|
|
|
|
opt.Option[0].Code = dns.OptionCodeNSID
|
|
|
|
opt.Option[0].Data = ""
|
|
|
|
}
|
2010-12-31 03:15:59 +11:00
|
|
|
m.Extra = make([]dns.RR, 1)
|
2010-12-31 04:48:29 +11:00
|
|
|
m.Extra[0] = opt
|
2010-12-31 03:15:59 +11:00
|
|
|
}
|
2011-01-17 20:51:33 +11:00
|
|
|
|
2010-12-31 02:38:16 +11:00
|
|
|
for _, v := range qname {
|
|
|
|
m.Question[0] = dns.Question{v, qtype, qclass}
|
2011-01-03 21:29:04 +11:00
|
|
|
m.SetId()
|
2011-02-09 06:22:43 +11:00
|
|
|
in, err := r.Query(m)
|
|
|
|
if in != nil {
|
|
|
|
if m.Id != in.Id {
|
2011-01-03 22:18:17 +11:00
|
|
|
fmt.Printf("Id mismatch\n")
|
|
|
|
}
|
2011-02-09 06:22:43 +11:00
|
|
|
fmt.Printf("%v\n", in)
|
2011-01-03 22:18:17 +11:00
|
|
|
} else {
|
2011-02-09 06:22:43 +11:00
|
|
|
fmt.Printf("%v\n", err.String())
|
2011-01-03 22:18:17 +11:00
|
|
|
}
|
2010-12-31 02:38:16 +11:00
|
|
|
}
|
|
|
|
}
|
2011-02-09 06:15:54 +11:00
|
|
|
/*
|
|
|
|
41 func (m *Meta) String() string {
|
|
|
|
42 s := ";; Query time: " + strconv.Itoa(int(m.QueryEnd-m.QueryStart)) + " nsec"
|
|
|
|
43 s += "\n;; MSG SIZE rcvd: " + strconv.Itoa(m.RLen) + ", sent: " + strconv.Itoa(m.QLen)
|
|
|
|
44 rf := float32(m.RLen)
|
|
|
|
45 qf := float32(m.QLen)
|
|
|
|
46 if qf != 0 {
|
|
|
|
47 s += " (" + strconv.Ftoa32(rf/qf, 'f', 2) + ":1)"
|
|
|
|
48 }
|
|
|
|
49 // WHEN??
|
|
|
|
50 return s
|
|
|
|
51 }
|
|
|
|
*/
|