diff options
author | 2020-11-19 09:14:30 +0100 | |
---|---|---|
committer | 2020-11-19 00:14:30 -0800 | |
commit | 07ea81ba3e65b66467aefce2e844bba5881be691 (patch) | |
tree | 1c93ba3e42a1fe2e2f8e4ac0e72f14f7c628caff /infra/base-images/base-builder/ossfuzz_coverage_runner.go | |
parent | 43f768df012e4c0e52f8ae0a7171cd748ec3f000 (diff) |
[infra] Add code coverage report generation for Go projects (#3142)
* Golang coverage report
* Enables golang coverage report for gonids and go-dns
* Generates summary for golang coverage reports
* Performance profile for golang projects
Diffstat (limited to 'infra/base-images/base-builder/ossfuzz_coverage_runner.go')
-rw-r--r-- | infra/base-images/base-builder/ossfuzz_coverage_runner.go | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/infra/base-images/base-builder/ossfuzz_coverage_runner.go b/infra/base-images/base-builder/ossfuzz_coverage_runner.go new file mode 100644 index 00000000..d433da24 --- /dev/null +++ b/infra/base-images/base-builder/ossfuzz_coverage_runner.go @@ -0,0 +1,69 @@ +// Copyright 2020 Google LLC +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package mypackagebeingfuzzed + +import ( + "io/ioutil" + "os" + "runtime/pprof" + "testing" +) + +func TestFuzzCorpus(t *testing.T) { + dir := os.Getenv("FUZZ_CORPUS_DIR") + if dir == "" { + t.Logf("No fuzzing corpus directory set") + return + } + infos, err := ioutil.ReadDir(dir) + if err != nil { + t.Logf("Not fuzzing corpus directory %s", err) + return + } + filename := "" + defer func() { + if r := recover(); r != nil { + t.Error("Fuzz panicked in "+filename, r) + } + }() + profname := os.Getenv("FUZZ_PROFILE_NAME") + if profname != "" { + f, err := os.Create(profname + ".cpu.prof") + if err != nil { + t.Logf("error creating profile file %s\n", err) + } else { + _ = pprof.StartCPUProfile(f) + } + } + for i := range infos { + filename = dir + infos[i].Name() + data, err := ioutil.ReadFile(filename) + if err != nil { + t.Error("Failed to read corpus file", err) + } + FuzzFunction(data) + } + if profname != "" { + pprof.StopCPUProfile() + f, err := os.Create(profname + ".heap.prof") + if err != nil { + t.Logf("error creating heap profile file %s\n", err) + } + if err = pprof.WriteHeapProfile(f); err != nil { + t.Logf("error writing heap profile file %s\n", err) + } + f.Close() + } +} |