aboutsummaryrefslogtreecommitdiff
path: root/plugin/header/handler.go
diff options
context:
space:
mode:
authorGravatar Qasim Sarfraz <syed.qasim.sarfraz@gmail.com> 2021-07-15 09:32:39 +0200
committerGravatar GitHub <noreply@github.com> 2021-07-15 09:32:39 +0200
commit00128bda4e2fa56ec4bf66e321bf3e00263d1188 (patch)
tree777af2d1db0a5133046f00e38eb41de98685fa91 /plugin/header/handler.go
parentfaa10f61aff9cce538375586b71a9bdd0fab6422 (diff)
downloadcoredns-00128bda4e2fa56ec4bf66e321bf3e00263d1188.tar.gz
coredns-00128bda4e2fa56ec4bf66e321bf3e00263d1188.tar.zst
coredns-00128bda4e2fa56ec4bf66e321bf3e00263d1188.zip
plugin/header: Introduce header plugin (#4752)
* Add header plugin Signed-off-by: MQasimSarfraz <syed.qasim.sarfraz@gmail.com> * fix import format * improve README.md * Add codeowners for header plugin
Diffstat (limited to 'plugin/header/handler.go')
-rw-r--r--plugin/header/handler.go24
1 files changed, 24 insertions, 0 deletions
diff --git a/plugin/header/handler.go b/plugin/header/handler.go
new file mode 100644
index 000000000..a002c4fb1
--- /dev/null
+++ b/plugin/header/handler.go
@@ -0,0 +1,24 @@
+package header
+
+import (
+ "context"
+
+ "github.com/coredns/coredns/plugin"
+
+ "github.com/miekg/dns"
+)
+
+// Header modifies dns.MsgHdr in the responses
+type Header struct {
+ Rules []Rule
+ Next plugin.Handler
+}
+
+// ServeDNS implements the plugin.Handler interface.
+func (h Header) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
+ wr := ResponseHeaderWriter{ResponseWriter: w, Rules: h.Rules}
+ return plugin.NextOrFailure(h.Name(), h.Next, ctx, &wr, r)
+}
+
+// Name implements the plugin.Handler interface.
+func (h Header) Name() string { return "header" }