18b92e1117
* WIP: make CoreDNS DoH Server * It works * Fix tests * Review from Tom - on diff. PR * correct mime type * Cleanups and use the pkg/nonwriter * rename and updates * implement get * implement GET * Code review comments * correct context * tweaks * code review
34 lines
936 B
Go
34 lines
936 B
Go
// Package nonwriter implements a dns.ResponseWriter that never writes, but captures the dns.Msg being written.
|
|
package nonwriter
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// Writer is a type of ResponseWriter that captures the message, but never writes to the client.
|
|
type Writer struct {
|
|
dns.ResponseWriter
|
|
Msg *dns.Msg
|
|
|
|
// 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
|
|
}
|
|
|
|
// New makes and returns a new NonWriter.
|
|
func New(w dns.ResponseWriter) *Writer { return &Writer{ResponseWriter: w} }
|
|
|
|
// WriteMsg records the message, but doesn't write it itself.
|
|
func (w *Writer) WriteMsg(res *dns.Msg) error {
|
|
w.Msg = res
|
|
return nil
|
|
}
|
|
|
|
// RemoteAddr returns the remote address.
|
|
func (w *Writer) RemoteAddr() net.Addr { return w.Raddr }
|
|
|
|
// LocalAddr returns the local address.
|
|
func (w *Writer) LocalAddr() net.Addr { return w.Laddr }
|