ICU-20734 Improve fuzzer_driver

See #1204
This commit is contained in:
Frank Tang 2020-07-31 21:32:31 +00:00 committed by Frank Yung-Fong Tang
parent 41d1d57af0
commit 7ddc231195
2 changed files with 47 additions and 7 deletions

View file

@ -73,7 +73,9 @@ distclean-local: clean-local
$(RMV) Makefile
check-local: all-local
$(foreach trgt,$(FUZZER_TARGETS), echo $(trgt); $(INVOKE) ./$(trgt) \
# Create a dummy test case file with content "abc123"
echo "abc123" > dummytestcase
$(foreach trgt,$(FUZZER_TARGETS), echo $(trgt); $(INVOKE) ./$(trgt) dummytestcase -q \
$(TEST_OUTPUT_OPTS) || exit \
$(IOTEST_OPTS);)

View file

@ -1,8 +1,12 @@
// © 2019 and later: Unicode, Inc. and others.
// License & terms of use: http://www.unicode.org/copyright.html
#include <fstream>
#include <iostream>
#include <sstream>
#include <stddef.h>
#include <stdint.h>
#include <string>
#include "cmemory.h"
@ -10,11 +14,45 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size);
int main(int argc, char* argv[])
{
(void) argc;
(void) argv;
const char *fuzzer_data = "abc123";
LLVMFuzzerTestOneInput((const uint8_t *) fuzzer_data, strlen(fuzzer_data));
bool show_warning = true;
bool show_error = true;
#if UPRV_HAS_FEATURE(address_sanitizer)
show_warning = false;
#endif
#if UPRV_HAS_FEATURE(memory_sanitizer)
show_warning = false;
#endif
if (argc > 2 && strcmp(argv[2], "-q") == 0) {
show_warning = false;
show_error = false;
}
if (show_warning) {
std::cerr << "WARNING: This binary work only under build configure with" << std::endl
<< " CFLAGS=\"-fsanitize=$SANITIZE\""
<< " CXXFLAGS=\"-fsanitize=$SANITIZE\""
<< " ./runConfigureICU ... " << std::endl
<< " where $SANITIZE is 'address' or 'memory'" << std::endl
<< "Plesae run the above step and make tests to rebuild" << std::endl;
// Do not return -1 here so we will pass the unit test.
}
if (argc < 2) {
if (show_error) {
std::cerr << "Usage: " << argv[0] << " testcasefile [-q]" << std::endl
<< " -q : quiet while error" << std::endl;
}
return -1;
}
const char *path = argv[1];
std::ifstream file(path, std::ios::binary);
if (!file.is_open()) {
if (show_error) {
std::cerr << "Cannot open testcase file " << path << std::endl;
}
return -1;
}
std::ostringstream ostrm;
ostrm << file.rdbuf();
LLVMFuzzerTestOneInput((const uint8_t *) ostrm.str().c_str(), ostrm.str().size());
return 0;
return 0;
}