mirror of
https://github.com/coredns/coredns.git
synced 2025-10-29 01:04:15 -04:00
e08d3335b0
This can't be used in its current form; revert the entire PR.
Revert "fuzzing: allow setup function to be called (#3175)"
This reverts commit 62451fd3eb.
32 lines
808 B
Go
32 lines
808 B
Go
// Package fuzz contains functions that enable fuzzing of plugins.
|
|
package fuzz
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/coredns/coredns/plugin"
|
|
"github.com/coredns/coredns/plugin/test"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// Do will fuzz p - used by gofuzz. See Makefile.fuzz for comments and context.
|
|
func Do(p plugin.Handler, data []byte) int {
|
|
ctx := context.TODO()
|
|
r := new(dns.Msg)
|
|
if err := r.Unpack(data); err != nil {
|
|
return 0 // plugin will never be called when this happens.
|
|
}
|
|
// If the data unpack into a dns msg, but does not have a proper question section discard it.
|
|
// The server parts make sure this is true before calling the plugins; mimic this behavior.
|
|
if len(r.Question) == 0 {
|
|
return 0
|
|
}
|
|
|
|
if _, err := p.ServeDNS(ctx, &test.ResponseWriter{}, r); err != nil {
|
|
return 1
|
|
}
|
|
|
|
return 0
|
|
}
|