diff options
Diffstat (limited to 'plugin/rewrite/rewrite.go')
| -rw-r--r-- | plugin/rewrite/rewrite.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/plugin/rewrite/rewrite.go b/plugin/rewrite/rewrite.go index 9b61ee123..422ebd9c6 100644 --- a/plugin/rewrite/rewrite.go +++ b/plugin/rewrite/rewrite.go @@ -42,7 +42,7 @@ type Rewrite struct { 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(w, r); result { + switch result := rule.Rewrite(ctx, w, r); result { case RewriteDone: respRule := rule.GetResponseRule() if respRule.Active == true { @@ -76,7 +76,7 @@ func (rw Rewrite) Name() string { return "rewrite" } // Rule describes a rewrite rule. type Rule interface { // Rewrite rewrites the current request. - Rewrite(dns.ResponseWriter, *dns.Msg) Result + Rewrite(context.Context, dns.ResponseWriter, *dns.Msg) Result // Mode returns the processing mode stop or continue. Mode() string // GetResponseRule returns the rule to rewrite response with, if any. |