xuemingqiang/mongo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SERVER-51335 Persist fuzzer corpora for each branch and variant

Browse Source
This commit is contained in:
Ben Caimano
2021-03-10 19:42:36 +00:00
committed by Evergreen Agent
parent d24fbc3f38
commit 2d48577820
6 changed files with 125 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
buildscripts/resmokelib/testing/testcases/cpp_libfuzzer_test.py
View File

@@ -2,7 +2,6 @@
import datetime
import os
import subprocess
from buildscripts.resmokelib import core
from buildscripts.resmokelib import utils
@@ -16,42 +15,31 @@ class CPPLibfuzzerTestCase(interface.ProcessTestCase):
REGISTERED_NAME = "cpp_libfuzzer_test"
DEFAULT_TIMEOUT = datetime.timedelta(hours=1)
def __init__(self, logger, program_executable, program_options=None):
def __init__( # pylint: disable=too-many-arguments
self, logger, program_executable, program_options=None, runs=1000000,
corpus_directory_stem="corpora"):
"""Initialize the CPPLibfuzzerTestCase with the executable to run."""
interface.ProcessTestCase.__init__(self, logger, "C++ libfuzzer test", program_executable)
self.program_executable = program_executable
self.program_options = utils.default_if_none(program_options, {}).copy()
self.corpus_directory = "corpus/corpus-" + self.short_name()
self.runs = runs
self.corpus_directory = f"{corpus_directory_stem}/corpus-{self.short_name()}"
self.merged_corpus_directory = f"{corpus_directory_stem}-merged/corpus-{self.short_name()}"
os.makedirs(self.corpus_directory, exist_ok=True)
def _make_process(self):
default_args = [
self.program_executable, self.corpus_directory, "-max_len=100000", "-rss_limit_mb=5000"
self.program_executable,
"-max_len=100000",
"-rss_limit_mb=5000",
f"-runs={self.runs}",
self.corpus_directory,
]
self.program_options["job_num"] = self.fixture.job_num
self.program_options["test_id"] = self._id
return core.programs.make_process(self.logger, default_args, **self.program_options)
def _execute(self, process):
"""Run the specified process."""
self.logger.info("Starting Libfuzzer Test %s...\n%s", self.short_description(),
process.as_command())
process.start()
self.logger.info("%s started with pid %s.", self.short_description(), process.pid)
try:
self.return_code = process.wait(self.DEFAULT_TIMEOUT.total_seconds())
except subprocess.TimeoutExpired:
# If the test timeout, then no errors were detected. Thus, the return code should be 0.
process.stop(mode=fixture_interface.TeardownMode.KILL)
process.wait()
self.logger.info("%s timed out. No errors were found.", self.short_description())
self.return_code = 0
if self.return_code != 0:
self.logger.info("Failed %s", self.return_code)
raise self.failureException("%s failed" % (self.short_description()))
self.logger.info("%s finished.", self.short_description())