mirror of
https://github.com/miekg/dns.git
synced 2025-11-03 20:21:24 +01:00
107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package responder
|
|
|
|
import (
|
|
"testing"
|
|
"dns"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
type myserv Server
|
|
|
|
func createpkg(id uint16, tcp bool, remove net.Addr) []byte {
|
|
m := new(dns.Msg)
|
|
m.MsgHdr.Id = id
|
|
m.MsgHdr.Authoritative = true
|
|
m.MsgHdr.AuthenticatedData = false
|
|
m.MsgHdr.RecursionAvailable = true
|
|
m.MsgHdr.Response = true
|
|
m.MsgHdr.Opcode = dns.OpcodeQuery
|
|
m.MsgHdr.Rcode = dns.RcodeSuccess
|
|
m.Question = make([]dns.Question, 1)
|
|
m.Question[0] = dns.Question{"miek.nl.", dns.TypeTXT, dns.ClassINET}
|
|
m.Answer = make([]dns.RR, 1)
|
|
t := new(dns.RR_TXT)
|
|
t.Hdr = dns.RR_Header{Name: "miek.nl.", Rrtype: dns.TypeTXT, Class: dns.ClassINET, Ttl: 3600}
|
|
if tcp {
|
|
t.Txt = "Dit is iets anders TCP"
|
|
} else {
|
|
t.Txt = "Dit is iets anders UDP"
|
|
}
|
|
m.Answer[0] = t
|
|
out, _ := m.Pack()
|
|
return out
|
|
}
|
|
|
|
func (s *myserv) ResponderUDP(c *net.UDPConn, a net.Addr, in []byte) {
|
|
inmsg := new(dns.Msg)
|
|
inmsg.Unpack(in)
|
|
if inmsg.MsgHdr.Response == true {
|
|
// Uh... answering to an response??
|
|
// dont think so
|
|
return
|
|
}
|
|
out := createpkg(inmsg.MsgHdr.Id, false, a)
|
|
SendUDP(out, c, a)
|
|
// Meta.QLen/RLen/QueryStart/QueryEnd can be filled in at
|
|
// this point for logging purposses or anything else
|
|
}
|
|
|
|
func (s *myserv) ResponderTCP(c *net.TCPConn, in []byte) {
|
|
inmsg := new(dns.Msg)
|
|
inmsg.Unpack(in)
|
|
if inmsg.MsgHdr.Response == true {
|
|
// Uh... answering to an response??
|
|
// dont think so
|
|
return
|
|
}
|
|
out := createpkg(inmsg.MsgHdr.Id, true, c.RemoteAddr())
|
|
SendTCP(out, c)
|
|
}
|
|
|
|
func TestResponder(t *testing.T) {
|
|
/* udp servertje */
|
|
su := new(Server)
|
|
su.Address = "127.0.0.1"
|
|
su.Port = "8053"
|
|
var us *myserv
|
|
uch := make(chan bool)
|
|
go su.NewResponder(us, uch)
|
|
|
|
/* tcp servertje */
|
|
st := new(Server)
|
|
st.Address = "127.0.0.1"
|
|
st.Port = "8053"
|
|
st.Tcp = true
|
|
var ts *myserv
|
|
tch := make(chan bool)
|
|
go st.NewResponder(ts, tch)
|
|
time.Sleep(1 * 1e9)
|
|
uch <- true
|
|
tch <- true
|
|
<-uch
|
|
<-tch
|
|
}
|
|
|
|
func TestReflectorResponder(t *testing.T) {
|
|
stop := make(chan bool)
|
|
s := new(Server)
|
|
s.Port = "8053"
|
|
s.Address = "127.0.0.1"
|
|
|
|
stoptcp := make(chan bool)
|
|
stcp := new(Server)
|
|
stcp.Port = "8053"
|
|
stcp.Address = "127.0.0.1"
|
|
stcp.Tcp = true
|
|
|
|
go stcp.NewResponder(Reflector, stoptcp)
|
|
go s.NewResponder(Reflector, stop)
|
|
|
|
time.Sleep(1 * 1e9)
|
|
stop <- true
|
|
stoptcp <- true
|
|
<-stop
|
|
<-stoptcp
|
|
}
|