You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
779 B
35 lines
779 B
package realm
|
|
|
|
import "github.com/miekg/dns"
|
|
|
|
type Server struct {
|
|
server *dns.Server
|
|
zone *Zone
|
|
}
|
|
|
|
func NewServer(listen string, zone *Zone) *Server {
|
|
var server *Server = &Server{zone: zone}
|
|
server.server = &dns.Server{
|
|
Addr: listen,
|
|
Net: "udp",
|
|
Handler: server,
|
|
}
|
|
return server
|
|
}
|
|
|
|
func (server *Server) ListenAndServe() error {
|
|
return server.server.ListenAndServe()
|
|
}
|
|
|
|
func (server *Server) ServeDNS(w dns.ResponseWriter, request *dns.Msg) {
|
|
var response *dns.Msg = &dns.Msg{}
|
|
response.SetReply(request)
|
|
response.Compress = true
|
|
|
|
for _, question := range request.Question {
|
|
var records []dns.RR = server.zone.Lookup(question.Name, question.Qtype, question.Qclass)
|
|
response.Answer = append(response.Answer, records...)
|
|
}
|
|
|
|
w.WriteMsg(response)
|
|
}
|