Commit b1b52426 authored by Miek Gieben's avatar Miek Gieben

Fix rewrite

parent a412255a
......@@ -32,13 +32,13 @@ type Rewrite struct {
// ServeHTTP implements the middleware.Handler interface.
func (rw Rewrite) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
wr := NewResponseReverter(w, r)
for _, rule := range rw.Rules {
switch result := rule.Rewrite(r); result {
case RewriteDone:
if rw.noRevert {
return rw.Next.ServeDNS(ctx, w, r)
}
wr := NewResponseReverter(w, r)
return rw.Next.ServeDNS(ctx, wr, r)
case RewriteIgnored:
break
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment