diff options
Diffstat (limited to 'plugin/pkg/fuzz/do.go')
-rw-r--r-- | plugin/pkg/fuzz/do.go | 7 |
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 { |