dns/_examples/q/lex.go
2011-09-19 22:27:56 +02:00

55 lines
1.4 KiB
Go

package main
import (
"dns"
)
type itemType int
// We handle the checking as a lexing program.
// We use the lexer like Rob Pike lectures about in this
// clip: http://www.youtube.com/watch?v=HxaD_trXwRE
type item struct {
typ itemType
val string
}
const (
itemError itemType = iota
itemVender // software vendor
itemSoftware // the name of the DNS server software
itemVersion // the version of the software (empty if not determined)
)
// stateFn represents the state of the scanner as a function that returns the next state.
type stateFn func(*lexer) stateFn
type lexer struct {
client *dns.Client // client used.
addr string // addr of the server being scanned.
fp *fingerprint // fingerprint to test.
q dns.Question // question to ask.
items chan item // channel of scanned items.
state stateFn // the next function to enter.
}
func (l *lexer) probe() *fingerprint {
return sendProbe(l.client, l.addr, l.fp, l.q)
}
func (l *lexer) emit() {
// send item on the channel
}
// "Lexer" functions
// Check if the server returns the DO-bit when set in the request.
func lexDoBitMirorred(l *lexer) stateFn {
// The important part here is that the DO bit is on
l.fp.SetString("QUERY,NOERROR,qr,aa,tc,RD,ad,cd,z,1,0,0,0,DO,0")
l.q = dns.Question{".", dns.TypeNS, dns.ClassINET}
l.probe()
return nil
}