1 files changed, 98 insertions, 0 deletions
diff --git a/plugin/dnstap/setup.go b/plugin/dnstap/setup.go
new file mode 100644
index 000000000..a57873470
--- /dev/null
+++ b/plugin/dnstap/setup.go
@@ -0,0 +1,98 @@
+package dnstap
+
+import (
+	"fmt"
+	"io"
+	"log"
+	"strings"
+
+	"github.com/coredns/coredns/core/dnsserver"
+	"github.com/coredns/coredns/plugin"
+	"github.com/coredns/coredns/plugin/dnstap/out"
+	"github.com/coredns/coredns/plugin/pkg/dnsutil"
+
+	"github.com/mholt/caddy"
+	"github.com/mholt/caddy/caddyfile"
+)
+
+func init() {
+	caddy.RegisterPlugin("dnstap", caddy.Plugin{
+		ServerType: "dns",
+		Action:     wrapSetup,
+	})
+}
+
+func wrapSetup(c *caddy.Controller) error {
+	if err := setup(c); err != nil {
+		return plugin.Error("dnstap", err)
+	}
+	return nil
+}
+
+type config struct {
+	target string
+	socket bool
+	full   bool
+}
+
+func parseConfig(d *caddyfile.Dispenser) (c config, err error) {
+	d.Next() // directive name
+
+	if !d.Args(&c.target) {
+		return c, d.ArgErr()
+	}
+
+	if strings.HasPrefix(c.target, "tcp://") {
+		// remote IP endpoint
+		servers, err := dnsutil.ParseHostPortOrFile(c.target[6:])
+		if err != nil {
+			return c, d.ArgErr()
+		}
+		c.target = servers[0]
+	} else {
+		// default to UNIX socket
+		if strings.HasPrefix(c.target, "unix://") {
+			c.target = c.target[7:]
+		}
+		c.socket = true
+	}
+
+	c.full = d.NextArg() && d.Val() == "full"
+
+	return
+}
+
+func setup(c *caddy.Controller) error {
+	conf, err := parseConfig(&c.Dispenser)
+	if err != nil {
+		return err
+	}
+
+	dnstap := Dnstap{Pack: conf.full}
+
+	var o io.WriteCloser
+	if conf.socket {
+		o, err = out.NewSocket(conf.target)
+		if err != nil {
+			log.Printf("[WARN] Can't connect to %s at the moment: %s", conf.target, err)
+		}
+	} else {
+		o = out.NewTCP(conf.target)
+	}
+	dnstap.Out = o
+
+	c.OnShutdown(func() error {
+		if err := o.Close(); err != nil {
+			return fmt.Errorf("output: %s", err)
+		}
+		return nil
+	})
+
+	dnsserver.GetConfig(c).AddPlugin(
+		func(next plugin.Handler) plugin.Handler {
+			dnstap.Next = next
+			return dnstap
+		})
+
+	return nil
+}