65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package dnsserver
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// DoHWriter is a dns.ResponseWriter that adds more specific LocalAddr and RemoteAddr methods.
|
|
type DoHWriter struct {
|
|
// raddr is the remote's address. This can be optionally set.
|
|
raddr net.Addr
|
|
// laddr is our address. This can be optionally set.
|
|
laddr net.Addr
|
|
|
|
// request is the HTTP request we're currently handling.
|
|
request *http.Request
|
|
|
|
// Msg is a response to be written to the client.
|
|
Msg *dns.Msg
|
|
}
|
|
|
|
// WriteMsg stores the message to be written to the client.
|
|
func (d *DoHWriter) WriteMsg(m *dns.Msg) error {
|
|
d.Msg = m
|
|
return nil
|
|
}
|
|
|
|
// Write stores the message to be written to the client.
|
|
func (d *DoHWriter) Write(b []byte) (int, error) {
|
|
d.Msg = new(dns.Msg)
|
|
return len(b), d.Msg.Unpack(b)
|
|
}
|
|
|
|
// RemoteAddr returns the remote address.
|
|
func (d *DoHWriter) RemoteAddr() net.Addr {
|
|
return d.raddr
|
|
}
|
|
|
|
// LocalAddr returns the local address.
|
|
func (d *DoHWriter) LocalAddr() net.Addr {
|
|
return d.laddr
|
|
}
|
|
|
|
// Request returns the HTTP request.
|
|
func (d *DoHWriter) Request() *http.Request {
|
|
return d.request
|
|
}
|
|
|
|
// Close no-op implementation.
|
|
func (d *DoHWriter) Close() error {
|
|
return nil
|
|
}
|
|
|
|
// TsigStatus no-op implementation.
|
|
func (d *DoHWriter) TsigStatus() error {
|
|
return nil
|
|
}
|
|
|
|
// TsigTimersOnly no-op implementation.
|
|
func (d *DoHWriter) TsigTimersOnly(_ bool) {}
|
|
|
|
// Hijack no-op implementation.
|
|
func (d *DoHWriter) Hijack() {}
|