aboutsummaryrefslogtreecommitdiff
path: root/plugin/pkg/fuzz/do.go
diff options
context:
space:
mode:
Diffstat (limited to 'plugin/pkg/fuzz/do.go')
-rw-r--r--plugin/pkg/fuzz/do.go7
1 files changed, 6 insertions, 1 deletions
diff --git a/plugin/pkg/fuzz/do.go b/plugin/pkg/fuzz/do.go
index 054c4298a..95a5400f1 100644
--- a/plugin/pkg/fuzz/do.go
+++ b/plugin/pkg/fuzz/do.go
@@ -11,7 +11,12 @@ import (
)
// Do will fuzz p - used by gofuzz. See Makefile.fuzz for comments and context.
-func Do(p plugin.Handler, data []byte) int {
+func Do(p plugin.Handler, fn SetupFunc, data []byte) int {
+ if fn != nil {
+ if err := fn(); err != nil {
+ panic("fuzz: " + err.Error())
+ }
+ }
ctx := context.TODO()
r := new(dns.Msg)
if err := r.Unpack(data); err != nil {