aboutsummaryrefslogtreecommitdiff
path: root/plugin
diff options
context:
space:
mode:
Diffstat (limited to 'plugin')
-rw-r--r--plugin/forward/fuzz.go34
1 files changed, 34 insertions, 0 deletions
diff --git a/plugin/forward/fuzz.go b/plugin/forward/fuzz.go
new file mode 100644
index 000000000..eb4f78246
--- /dev/null
+++ b/plugin/forward/fuzz.go
@@ -0,0 +1,34 @@
+// +build gofuzz
+
+package forward
+
+import (
+ "github.com/coredns/coredns/plugin/pkg/dnstest"
+ "github.com/coredns/coredns/plugin/pkg/fuzz"
+
+ "github.com/miekg/dns"
+)
+
+var f *Forward
+
+// abuse init to setup a environment to test against. This start another server to that will
+// reflect responses.
+func init() {
+ f = New()
+ s := dnstest.NewServer(r{}.reflectHandler)
+ f.proxies = append(f.proxies, NewProxy(s.Addr, "tcp"))
+ f.proxies = append(f.proxies, NewProxy(s.Addr, "udp"))
+}
+
+// Fuzz fuzzes forward.
+func Fuzz(data []byte) int {
+ return fuzz.Do(f, nil, data)
+}
+
+type r struct{}
+
+func (r r) reflectHandler(w dns.ResponseWriter, req *dns.Msg) {
+ m := new(dns.Msg)
+ m.SetReply(req)
+ w.WriteMsg(m)
+}