From 8f79ea64fe7e53a14329b4c6476aa3729f679aa4 Mon Sep 17 00:00:00 2001
From: Miek Gieben <miek@miek.nl>
Date: Tue, 1 Mar 2011 13:13:31 +0100
Subject: [PATCH] add ns

---
 _examples/Makefile    |   1 +
 _examples/ns/Makefile |   8 +++
 _examples/ns/ns.go    | 129 ++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 138 insertions(+)
 create mode 100644 _examples/ns/Makefile
 create mode 100644 _examples/ns/ns.go

diff --git a/_examples/Makefile b/_examples/Makefile
index 330dccb3..627a9602 100644
--- a/_examples/Makefile
+++ b/_examples/Makefile
@@ -4,6 +4,7 @@ chaos \
 axfr \
 reflect \
 funkensturm \
+ns \
 
 all: 
 	for i in $(EXAMPLES); do gomake -C $$i; done
diff --git a/_examples/ns/Makefile b/_examples/ns/Makefile
new file mode 100644
index 00000000..676dbf01
--- /dev/null
+++ b/_examples/ns/Makefile
@@ -0,0 +1,8 @@
+# Copyright 2009 The Go Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style
+# license that can be found in the LICENSE file.
+include $(GOROOT)/src/Make.inc
+TARG=ns
+GOFILES=ns.go
+DEPS=../../
+include $(GOROOT)/src/Make.cmd
diff --git a/_examples/ns/ns.go b/_examples/ns/ns.go
new file mode 100644
index 00000000..538c89f1
--- /dev/null
+++ b/_examples/ns/ns.go
@@ -0,0 +1,129 @@
+package main
+
+import (
+	"net"
+	"dns"
+	"os"
+	"flag"
+	"fmt"
+	"strconv"
+	"strings"
+)
+
+func main() {
+	var dnssec *bool = flag.Bool("dnssec", false, "Request DNSSEC records")
+	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")
+	var tcp *bool = flag.Bool("tcp", false, "TCP mode")
+	var nsid *bool = flag.Bool("nsid", false, "Ask for the NSID")
+	flag.Usage = func() {
+		fmt.Fprintf(os.Stderr, "Usage: %s [@server] [qtype] [qclass] [name ...]\n", os.Args[0])
+		flag.PrintDefaults()
+	}
+
+	nameserver := "@127.0.0.1"      // Default nameserver
+	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
+		}
+		// 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 {
+			if v == strings.ToUpper(flag.Arg(i)) {
+				qtype = k
+				continue Flags
+			}
+		}
+		// If it looks like a class, it is a class
+		for k, v := range dns.Class_str {
+			if v == strings.ToUpper(flag.Arg(i)) {
+				qclass = k
+				continue Flags
+			}
+		}
+		// If it starts with TYPExxx it is unknown rr
+		if strings.HasPrefix(flag.Arg(i), "TYPE") {
+			i, e := strconv.Atoi(string([]byte(flag.Arg(i))[4:]))
+			if e == nil {
+				qtype = uint16(i)
+				continue Flags
+			}
+		}
+
+		// Anything else is a qname
+		qname = append(qname, flag.Arg(i))
+	}
+	r := new(dns.Resolver)
+	r.FromFile("/etc/resolv.conf")
+	r.Timeout = 2
+	r.Port = *port
+	r.Tcp = *tcp
+	r.Attempts = 1
+	// @server may be a name, resolv that 
+	var err os.Error
+	nameserver = string([]byte(nameserver)[1:]) // chop off @
+	_, addr, err := net.LookupHost(nameserver)
+	if err == nil {
+		r.Servers = addr
+	} else {
+		r.Servers = []string{nameserver}
+	}
+
+	m := new(dns.Msg)
+	m.MsgHdr.Authoritative = *aa
+	m.MsgHdr.AuthenticatedData = *ad
+	m.MsgHdr.CheckingDisabled = *cd
+	m.MsgHdr.RecursionDesired = *rd
+	m.Question = make([]dns.Question, 1)
+	if *dnssec || *nsid {
+		opt := new(dns.RR_OPT)
+		opt.Hdr = dns.RR_Header{Name: "", Rrtype: dns.TypeOPT}
+		opt.SetVersion(0)
+		opt.SetDo()
+		opt.SetUDPSize(dns.DefaultMsgSize)
+		if *nsid {
+			opt.SetNsid("")
+		}
+		m.Extra = make([]dns.RR, 1)
+		m.Extra[0] = opt
+	}
+
+	for _, v := range qname {
+		m.Question[0] = dns.Question{v, qtype, qclass}
+		m.SetId()
+		in, err := r.Query(m)
+		if in != nil {
+			if m.Id != in.Id {
+				fmt.Printf("Id mismatch\n")
+			}
+			fmt.Printf("%v\n", in)
+		} else {
+			fmt.Printf("%v\n", err.String())
+		}
+	}
+}
+/*
+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 }
+*/