1 files changed, 131 insertions, 0 deletions
diff --git a/att script/4(v6 DDoS)/code/辅助权威服务器/plugin/dnstap/setup.go b/att script/4(v6 DDoS)/code/辅助权威服务器/plugin/dnstap/setup.go
new file mode 100644
index 0000000..6e81ac1
--- /dev/null
+++ b/att script/4(v6 DDoS)/code/辅助权威服务器/plugin/dnstap/setup.go
@@ -0,0 +1,131 @@
+package dnstap
+
+import (
+	"net/url"
+	"ohmydns2/core/dnsserver"
+	"ohmydns2/plugin"
+	olog "ohmydns2/plugin/pkg/log"
+	"os"
+	"strings"
+
+	"github.com/coredns/caddy"
+)
+
+var log = olog.NewWithPlugin("dnstap")
+
+func init() { plugin.Register("dnstap", setup) }
+
+func parseConfig(c *caddy.Controller) ([]*Dnstap, error) {
+	dnstaps := []*Dnstap{}
+
+	for c.Next() { // directive name
+		d := Dnstap{}
+		endpoint := ""
+
+		args := c.RemainingArgs()
+
+		if len(args) == 0 {
+			return nil, c.ArgErr()
+		}
+
+		endpoint = args[0]
+
+		var dio *dio
+		if strings.HasPrefix(endpoint, "tls://") {
+			// remote network endpoint
+			endpointURL, err := url.Parse(endpoint)
+			if err != nil {
+				return nil, c.ArgErr()
+			}
+			dio = newIO("tls", endpointURL.Host)
+			d = Dnstap{io: dio}
+		} else if strings.HasPrefix(endpoint, "tcp://") {
+			// remote network endpoint
+			endpointURL, err := url.Parse(endpoint)
+			if err != nil {
+				return nil, c.ArgErr()
+			}
+			dio = newIO("tcp", endpointURL.Host)
+			d = Dnstap{io: dio}
+		} else {
+			endpoint = strings.TrimPrefix(endpoint, "unix://")
+			dio = newIO("unix", endpoint)
+			d = Dnstap{io: dio}
+		}
+
+		d.IncludeRawMessage = len(args) == 2 && args[1] == "full"
+
+		hostname, _ := os.Hostname()
+		d.Identity = []byte(hostname)
+		d.Version = []byte(caddy.AppName + "-" + caddy.AppVersion)
+
+		for c.NextBlock() {
+			switch c.Val() {
+			case "skipverify":
+				{
+					dio.skipVerify = true
+				}
+			case "identity":
+				{
+					if !c.NextArg() {
+						return nil, c.ArgErr()
+					}
+					d.Identity = []byte(c.Val())
+				}
+			case "version":
+				{
+					if !c.NextArg() {
+						return nil, c.ArgErr()
+					}
+					d.Version = []byte(c.Val())
+				}
+			}
+		}
+		dnstaps = append(dnstaps, &d)
+	}
+	return dnstaps, nil
+}
+
+func setup(c *caddy.Controller) error {
+	dnstaps, err := parseConfig(c)
+	if err != nil {
+		return plugin.Error("dnstap", err)
+	}
+
+	for i := range dnstaps {
+		dnstap := dnstaps[i]
+		c.OnStartup(func() error {
+			if err := dnstap.io.(*dio).connect(); err != nil {
+				log.Errorf("No connection to dnstap endpoint: %s", err)
+			}
+			return nil
+		})
+
+		c.OnRestart(func() error {
+			dnstap.io.(*dio).close()
+			return nil
+		})
+
+		c.OnFinalShutdown(func() error {
+			dnstap.io.(*dio).close()
+			return nil
+		})
+
+		if i == len(dnstaps)-1 {
+			// last dnstap plugin in block: point next to next plugin
+			dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
+				dnstap.Next = next
+				return dnstap
+			})
+		} else {
+			// not last dnstap plugin in block: point next to next dnstap
+			nextDnstap := dnstaps[i+1]
+			dnsserver.GetConfig(c).AddPlugin(func(plugin.Handler) plugin.Handler {
+				dnstap.Next = nextDnstap
+				return dnstap
+			})
+		}
+	}
+
+	return nil
+}