From 8892a1b49042e4e9f9ca011fecee31bc367b0e53 Mon Sep 17 00:00:00 2001 From: Miek Gieben Date: Tue, 5 Apr 2016 15:54:06 +0100 Subject: [PATCH] Middleware chaining fixes For prometheus use the plain value, not a pointer and change all usages. Allow AXFR to be requested over udp as well and some other more log printed when commencing an AXFR. --- core/directives.go | 2 +- core/setup/prometheus.go | 16 ++++++++-------- middleware/etcd/handler.go | 3 +-- middleware/file/file.go | 5 ++--- middleware/file/xfr.go | 7 +++---- middleware/prometheus/handler.go | 2 +- 6 files changed, 16 insertions(+), 19 deletions(-) diff --git a/core/directives.go b/core/directives.go index b679fb7e5..f3aa6af1a 100644 --- a/core/directives.go +++ b/core/directives.go @@ -51,8 +51,8 @@ var directiveOrder = []directive{ {"shutdown", setup.Shutdown}, // Directives that inject handlers (middleware) - {"log", setup.Log}, {"prometheus", setup.Prometheus}, + {"log", setup.Log}, {"rewrite", setup.Rewrite}, {"loadbalance", setup.Loadbalance}, {"file", setup.File}, diff --git a/core/setup/prometheus.go b/core/setup/prometheus.go index 3bcb907ce..db9929fdb 100644 --- a/core/setup/prometheus.go +++ b/core/setup/prometheus.go @@ -32,17 +32,17 @@ func Prometheus(c *Controller) (middleware.Middleware, error) { }, nil } -func parsePrometheus(c *Controller) (*prom.Metrics, error) { +func parsePrometheus(c *Controller) (prom.Metrics, error) { var ( - metrics *prom.Metrics + metrics prom.Metrics err error ) for c.Next() { - if metrics != nil { - return nil, c.Err("prometheus: can only have one metrics module per server") + if metrics.Addr != "" { + return prom.Metrics{}, c.Err("prometheus: can only have one metrics module per server") } - metrics = &prom.Metrics{ZoneNames: c.ServerBlockHosts} + metrics = prom.Metrics{ZoneNames: c.ServerBlockHosts} args := c.RemainingArgs() switch len(args) { @@ -50,18 +50,18 @@ func parsePrometheus(c *Controller) (*prom.Metrics, error) { case 1: metrics.Addr = args[0] default: - return nil, c.ArgErr() + return prom.Metrics{}, c.ArgErr() } for c.NextBlock() { switch c.Val() { case "address": args = c.RemainingArgs() if len(args) != 1 { - return nil, c.ArgErr() + return prom.Metrics{}, c.ArgErr() } metrics.Addr = args[0] default: - return nil, c.Errf("prometheus: unknown item: %s", c.Val()) + return prom.Metrics{}, c.Errf("prometheus: unknown item: %s", c.Val()) } } diff --git a/middleware/etcd/handler.go b/middleware/etcd/handler.go index 8ffc0c64f..4ad5be82d 100644 --- a/middleware/etcd/handler.go +++ b/middleware/etcd/handler.go @@ -1,7 +1,6 @@ package etcd import ( - "fmt" "strings" "github.com/miekg/coredns/middleware" @@ -13,7 +12,7 @@ import ( func (e Etcd) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { state := middleware.State{W: w, Req: r} if state.QClass() != dns.ClassINET { - return dns.RcodeServerFailure, fmt.Errorf("etcd: can only deal with ClassINET") + return e.Next.ServeDNS(ctx, w, r) } // We need to check stubzones first, because we may get a request for a zone we diff --git a/middleware/file/file.go b/middleware/file/file.go index 473b55574..907891783 100644 --- a/middleware/file/file.go +++ b/middleware/file/file.go @@ -1,7 +1,6 @@ package file import ( - "fmt" "io" "log" @@ -26,7 +25,7 @@ type ( func (f File) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { state := middleware.State{W: w, Req: r} if state.QClass() != dns.ClassINET { - return dns.RcodeServerFailure, fmt.Errorf("file: can only deal with ClassINET") + return f.Next.ServeDNS(ctx, w, r) } qname := state.Name() zone := middleware.Zones(f.Zones.Names).Matches(qname) @@ -63,7 +62,7 @@ func (f File) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (i return dns.RcodeServerFailure, nil } - if state.Proto() != "udp" && state.QType() == dns.TypeAXFR { + if state.QType() == dns.TypeAXFR || state.QType() == dns.TypeIXFR { xfr := Xfr{z} return xfr.ServeDNS(ctx, w, r) } diff --git a/middleware/file/xfr.go b/middleware/file/xfr.go index ed1862451..92811d7a7 100644 --- a/middleware/file/xfr.go +++ b/middleware/file/xfr.go @@ -2,6 +2,7 @@ package file import ( "fmt" + "log" "github.com/miekg/coredns/middleware" @@ -24,9 +25,6 @@ func (x Xfr) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (in if state.QType() != dns.TypeAXFR { return 0, fmt.Errorf("xfr called with non transfer type: %d", state.QType()) } - if state.Proto() == "udp" { - return 0, fmt.Errorf("xfr called with udp") - } records := x.All() if len(records) == 0 { @@ -40,6 +38,7 @@ func (x Xfr) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (in j, l := 0, 0 records = append(records, records[0]) // add closing SOA to the end + log.Printf("[INFO] Outgoing transfer of %d records of zone %s to %s started", len(records), x.name, state.IP()) for i, r := range records { l += dns.Len(r) if l > transferLength { @@ -57,4 +56,4 @@ func (x Xfr) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (in return dns.RcodeSuccess, nil } -const transferLength = 100 // Start a new envelop after message reaches this size. Intentionally small to test multi envelope parsing +const transferLength = 1000 // Start a new envelop after message reaches this size in bytes. Intentionally small to test multi envelope parsing. diff --git a/middleware/prometheus/handler.go b/middleware/prometheus/handler.go index 3c07cd942..92f82aea3 100644 --- a/middleware/prometheus/handler.go +++ b/middleware/prometheus/handler.go @@ -10,7 +10,7 @@ import ( "github.com/miekg/dns" ) -func (m *Metrics) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { +func (m Metrics) ServeDNS(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) { state := middleware.State{W: w, Req: r} qname := state.Name()