More test functions

Minor tweaks for timeouts in client.go
2011-09-20 18:15:36 +02:00 · 2011-09-20 18:15:36 +02:00 · 2f74133d6a
parent 956ec9b962
commit 2f74133d6a
3 changed files with 98 additions and 66 deletions
--- a/_examples/q/dns.go
+++ b/_examples/q/dns.go
@ -7,7 +7,7 @@ import (
 // Check if the server responds at all
 func dnsAlive(l *lexer) stateFn {
 	l.verbose("Alive")
-	l.setString("QUERY,NOERROR,qr,aa,tc,rd,ra,ad,cd,z,1,0,0,0,do,0")
+	l.setString("QUERY,NOERROR,qr,aa,tc,rd,ra,ad,cd,z,0,0,0,0,do,0")
 	l.setQuestion(".", dns.TypeNS, dns.ClassINET)

 	f := l.probe()
@ -22,7 +22,7 @@ func dnsAlive(l *lexer) stateFn {
 // Check if the server returns the DO-bit when set in the request.                                                                          
 func dnsDoBitMirror(l *lexer) stateFn {
 	l.verbose("DoBitMirror")
-	l.setString("QUERY,NOERROR,qr,aa,tc,RD,ra,ad,cd,z,1,0,0,0,DO,0")
+	l.setString("QUERY,NOERROR,qr,aa,tc,RD,ra,ad,cd,z,0,0,0,0,DO,0")
 	l.setQuestion(".", dns.TypeNS, dns.ClassINET)

 	f := l.probe()
@ -36,7 +36,7 @@ func dnsDoBitMirror(l *lexer) stateFn {

 func dnsEDNS0Mangler(l *lexer) stateFn {
 	l.verbose("EDNS0Mangler")
-	l.setString("NOTIFY,NOERROR,qr,aa,tc,RD,ra,ad,cd,z,1,0,0,0,do,0")
+	l.setString("NOTIFY,NOERROR,qr,aa,tc,RD,ra,ad,cd,z,0,0,0,0,do,0")
 	l.setQuestion("012345678901234567890123456789012345678901234567890123456789012.012345678901234567890123456789012345678901234567890123456789012.012345678901234567890123456789012345678901234567890123456789012.0123456789012345678901234567890123456789012345678901234567890.", dns.TypeA, dns.ClassINET)
 	f := l.probe()
        // MaraDNS does not set the QR bit in the reply... but only with this question is seems
@ -49,39 +49,80 @@ func dnsEDNS0Mangler(l *lexer) stateFn {

 func dnsTcEnable(l *lexer) stateFn {
 	l.verbose("TcEnable")
-	l.setString("QUERY,NOERROR,qr,aa,TC,rd,ra,ad,cd,z,1,0,0,0,do,0")
+	l.setString("QUERY,NOERROR,qr,aa,TC,rd,ra,ad,cd,z,0,0,0,0,do,0")
 	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
-
-	f := l.probe()
-        f = f
+	l.probe()
 	return dnsUDPSize
 }

 func dnsUDPSize(l *lexer) stateFn {
 	l.verbose("UDPSize")
-	l.setString("QUERY,NOERROR,qr,aa,tc,rd,ra,ad,cd,z,1,0,0,0,DO,4097")
+	l.setString("QUERY,NOERROR,qr,aa,tc,rd,ra,ad,cd,z,0,0,0,0,DO,4097")
 	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
-	f := l.probe()
-        f = f
+	l.probe()
 	return dnsZero
 }

 func dnsZero(l *lexer) stateFn {
 	l.verbose("Zero")
 	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
-	l.setString("QUERY,NOERROR,qr,aa,tc,rd,ra,ad,cd,Z,1,0,0,0,do,0")
-	f := l.probe()
-        f = f
+	l.setString("QUERY,NOERROR,qr,aa,tc,rd,ra,ad,cd,Z,0,0,0,0,do,0")
+	l.probe()
 	return dnsAll
 }

 func dnsAll(l *lexer) stateFn {
 	l.verbose("All")
-	l.setString("NOTIFY,NOERROR,qr,AA,TC,RD,RA,AD,CD,Z,1,0,0,0,DO,0")
+	l.setString("QUERY,NOERROR,qr,AA,TC,RD,RA,AD,CD,Z,0,0,0,0,DO,8192")
 	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
-	f := l.probe()
-        f = f
-	return nil
+	l.probe()
+	return dnsIquery
 }

+func dnsIquery(l *lexer) stateFn {
+	l.verbose("Iquery")
+	l.setString("IQUERY,NOERROR,qr,aa,tc,rd,ra,ad,cd,Z,0,0,0,0,do,0")
+	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
+	l.probe()
+	return dnsUpdate
+}

+func dnsUpdate(l *lexer) stateFn {
+	l.verbose("Update")
+	l.setString("UPDATE,NOERROR,qr,aa,tc,rd,ra,ad,cd,Z,0,0,0,0,do,0")
+	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
+	l.probe()
+	return dnsStatus
+}
+
+func dnsStatus(l *lexer) stateFn {
+	l.verbose("Status")
+	l.setString("STATUS,NOERROR,qr,aa,tc,rd,ra,ad,cd,Z,0,0,0,0,do,0")
+	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
+	l.probe()
+	return dnsOpcodeWhacky
+}
+
+func dnsOpcodeWhacky(l *lexer) stateFn {
+	l.verbose("OpcodeWhacky")
+	l.setString("12,NOERROR,qr,aa,tc,rd,ra,ad,cd,Z,0,0,0,0,do,0")
+	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
+	l.probe()
+	return dnsRcodeWhacky
+}
+
+func dnsRcodeWhacky(l *lexer) stateFn {
+	l.verbose("RcodeWhacky")
+	l.setString("QUERY,31,qr,aa,tc,rd,ra,ad,cd,Z,0,0,0,0,do,0")
+	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
+	l.probe()
+	return dnsRcodeNotZone
+}
+
+func dnsRcodeNotZone(l *lexer) stateFn {
+	l.verbose("RcodeNotZone")
+	l.setString("QUERY,NOTZONE,qr,aa,tc,rd,ra,ad,cd,Z,0,0,0,0,do,0")
+	l.setQuestion(".", dns.TypeNS, dns.ClassINET)
+	l.probe()
+	return nil
+}
--- a/_examples/q/fp.go
+++ b/_examples/q/fp.go
@ -84,8 +84,19 @@ func (f *fingerprint) String() string {
 		return "<nil>"
 	}
 	// Use the same order as in Perl's fpdns. But use more flags.
-	s := dns.Opcode_str[f.Opcode]
-	s += "," + dns.Rcode_str[f.Rcode]
+        var s string
+        if op, ok := dns.Opcode_str[f.Opcode]; ok {
+                s = op
+        } else {        // number
+                s = valueOfInt(f.Opcode)
+        }
+
+        if op, ok := dns.Rcode_str[f.Rcode]; ok {
+                s += "," + op
+        } else {        // number
+                s += "," + valueOfInt(f.Rcode)
+        }
+
 	s += valueOfBool(f.Response, ",qr")
 	s += valueOfBool(f.Authoritative, ",aa")
 	s += valueOfBool(f.Truncated, ",tc")
@ -95,13 +106,13 @@ func (f *fingerprint) String() string {
 	s += valueOfBool(f.CheckingDisabled, ",cd")
 	s += valueOfBool(f.Zero, ",z")

-	s += valueOfInt(f.Question)
-	s += valueOfInt(f.Answer)
-	s += valueOfInt(f.Ns)
-	s += valueOfInt(f.Extra)
+	s += ","+valueOfInt(f.Question)
+	s += ","+valueOfInt(f.Answer)
+	s += ","+valueOfInt(f.Ns)
+	s += ","+valueOfInt(f.Extra)

 	s += valueOfBool(f.Do, ",do")
-	s += valueOfInt(f.UDPSize)
+	s += ","+valueOfInt(f.UDPSize)
 	return s
 }

@ -110,58 +121,38 @@ func (f *fingerprint) setString(str string) {
 	for i, s := range strings.Split(str, ",") {
 		switch i {
 		case 0:
-			f.Opcode = dns.Str_opcode[s]
+                        if op, ok := dns.Str_opcode[s]; ok {
+			        f.Opcode = op
+                        } else { // number
+                                f.Opcode = valueOfString(s)
+                        }
 		case 1:
-			f.Rcode = dns.Str_rcode[s]
+                        if op, ok := dns.Str_rcode[s]; ok {
+			        f.Rcode = op
+                        } else { // number
+                                f.Rcode = valueOfString(s)
+                        }
 		case 2:
-			f.Response = false
-			if s == strings.ToUpper("qr") {
-				f.Response = true
-			}
+			f.Response = s == strings.ToUpper("qr")
 		case 3:
-			f.Authoritative = false
-			if s == strings.ToUpper("aa") {
-				f.Authoritative = true
-			}
+			f.Authoritative = s == strings.ToUpper("aa")
 		case 4:
-			f.Truncated = false
-			if s == strings.ToUpper("tc") {
-				f.Truncated = true
-			}
+			f.Truncated = s == strings.ToUpper("tc")
 		case 5:
-			f.RecursionDesired = false
-			if s == strings.ToUpper("rd") {
-				f.RecursionDesired = true
-			}
+			f.RecursionDesired = s == strings.ToUpper("rd")
 		case 6:
-			f.RecursionAvailable = false
-			if s == strings.ToUpper("ra") {
-				f.RecursionAvailable = true
-			}
+			f.RecursionAvailable = s == strings.ToUpper("ra")
 		case 7:
-			f.AuthenticatedData = false
-			if s == strings.ToUpper("ad") {
-				f.AuthenticatedData = true
-			}
+			f.AuthenticatedData = s == strings.ToUpper("ad")
 		case 8:
-			f.CheckingDisabled = false
-			if s == strings.ToUpper("cd") {
-				f.CheckingDisabled = true
-			}
+			f.CheckingDisabled = s == strings.ToUpper("cd")
 		case 9:
-			f.Zero = false
-			if s == strings.ToUpper("z") {
-				f.Zero = true
-			}
+			f.Zero =  s == strings.ToUpper("z")
 		case 10, 11, 12, 13:
 			// Can not set content of the message
 		case 14:
-			f.Do = false
-			if s == strings.ToUpper("do") {
-				f.Do = true
-			}
+			f.Do = s == strings.ToUpper("do")
 		case 15:
-			f.UDPSize = 0
 			f.UDPSize = valueOfString(s)
 		default:
 			panic("unhandled fingerprint")
@ -257,7 +248,7 @@ func valueOfBool(b bool, w string) string {
 }

 func valueOfInt(i int) string {
-	return "," + strconv.Itoa(i)
+	return strconv.Itoa(i)
 }

 func valueOfString(s string) int {
--- a/client.go
+++ b/client.go
@ -142,8 +142,8 @@ func NewClient() *Client {
 	c.Attempts = 1
        c.ReplyChan = DefaultReplyChan
        c.QueryChan = DefaultQueryChan
-	c.ReadTimeout = 5000
-	c.WriteTimeout = 5000
+	c.ReadTimeout = 0.5 * 1e9
+	c.WriteTimeout = 0.5 * 1e9
 	return c
 }