mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2024-12-27 06:21:25 +00:00
c2b76eee36
Change-Id: I2db557669ada6e031140a09b3a92bd901220f8f3 Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/580975 LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com> Reviewed-by: Lasse Folger <lassefolger@google.com> Reviewed-by: Michael Stapelberg <stapelberg@google.com> Auto-Submit: Michael Stapelberg <stapelberg@google.com>
45 lines
961 B
Go
45 lines
961 B
Go
// Copyright 2019 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package fuzztest contains a common fuzzer test.
|
|
package fuzztest
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"testing"
|
|
)
|
|
|
|
var corpus = flag.String("corpus", "corpus", "directory containing the fuzzer corpus")
|
|
|
|
// Test executes a fuzz function for every entry in the corpus.
|
|
func Test(t *testing.T, fuzz func(b []byte) int) {
|
|
dir, err := os.Open(*corpus)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
infos, err := dir.Readdir(0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
|
|
}
|
|
var names []string
|
|
for _, info := range infos {
|
|
names = append(names, info.Name())
|
|
}
|
|
sort.Strings(names)
|
|
for _, name := range names {
|
|
t.Run(name, func(t *testing.T) {
|
|
b, err := os.ReadFile(filepath.Join(*corpus, name))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
b = b[:len(b):len(b)] // set cap to len
|
|
fuzz(b)
|
|
})
|
|
}
|
|
}
|