/
resolver.go
120 lines (96 loc) · 2.13 KB
/
resolver.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package newdns
import (
"net"
"github.com/miekg/dns"
)
var fakeAddr = &net.TCPAddr{
IP: net.IP{0, 0, 0, 0},
Port: 0,
}
type responseWriter struct {
msg *dns.Msg
}
func (w *responseWriter) LocalAddr() net.Addr {
return fakeAddr
}
func (w *responseWriter) RemoteAddr() net.Addr {
return fakeAddr
}
func (w *responseWriter) WriteMsg(msg *dns.Msg) error {
// check message
if w.msg != nil {
panic("message already set")
}
// set message
w.msg = msg
return nil
}
func (w *responseWriter) Write([]byte) (int, error) {
panic("not implemented")
}
func (w *responseWriter) Close() error {
return nil
}
func (w *responseWriter) TsigStatus() error {
panic("not implemented")
}
func (w *responseWriter) TsigTimersOnly(bool) {
panic("not implemented")
}
func (w *responseWriter) Hijack() {
panic("not implemented")
}
// Resolver returns a very primitive recursive resolver that uses the provided
// handler to resolve all names.
func Resolver(handler dns.Handler) dns.Handler {
return dns.HandlerFunc(func(w dns.ResponseWriter, req *dns.Msg) {
// forward query if no recursion is desired
if !req.RecursionDesired {
handler.ServeDNS(w, req)
return
}
// prepare response
res := new(dns.Msg)
res.SetReply(req)
res.RecursionAvailable = true
// query handler
var wr responseWriter
handler.ServeDNS(&wr, req)
// check response
if wr.msg == nil {
_ = w.WriteMsg(res)
return
}
// add resolved answers
res.Answer = append(res.Answer, resolve(handler, wr.msg.Answer)...)
// write response
err := w.WriteMsg(res)
if err != nil {
_ = w.Close()
}
})
}
func resolve(handler dns.Handler, records []dns.RR) []dns.RR {
// prepare result
var res []dns.RR
res = append(res, records...)
// handle records
for _, record := range records {
if cname, ok := record.(*dns.CNAME); ok {
// query handler
var wr responseWriter
handler.ServeDNS(&wr, &dns.Msg{
Question: []dns.Question{
{
Name: cname.Target,
Qtype: dns.TypeA,
Qclass: dns.ClassINET,
},
},
})
// add resolved answers
res = append(res, resolve(handler, wr.msg.Answer)...)
}
}
return res
}