Commit b1b52426 authored by Miek Gieben's avatar Miek Gieben

Fix rewrite

parent a412255a
...@@ -32,13 +32,13 @@ type Rewrite struct { ...@@ -32,13 +32,13 @@ type Rewrite struct {
// ServeHTTP implements the middleware.Handler interface. // ServeHTTP implements the middleware.Handler interface.
func (rw Rewrite) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { func (rw Rewrite) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
wr := NewResponseReverter(w, r)
for _, rule := range rw.Rules { for _, rule := range rw.Rules {
switch result := rule.Rewrite(r); result { switch result := rule.Rewrite(r); result {
case RewriteDone: case RewriteDone:
if rw.noRevert { if rw.noRevert {
return rw.Next.ServeDNS(ctx, w, r) return rw.Next.ServeDNS(ctx, w, r)
} }
wr := NewResponseReverter(w, r)
return rw.Next.ServeDNS(ctx, wr, r) return rw.Next.ServeDNS(ctx, wr, r)
case RewriteIgnored: case RewriteIgnored:
break 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