diff --git a/CMakeLists.txt b/CMakeLists.txt index cc08bdc2..5fdba01a 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -597,7 +597,7 @@ endif() add_custom_target(symlinks ALL DEPENDS ${SYMLINKS}) -if(WITH_TESTS OR WITH_BENCHMARKS) +if(WITH_TESTS OR WITH_BENCHMARKS OR WITH_FUZZ) add_library(test_helpers test/test_helpers.cpp test/test_iolayer.cpp test/test_strings.cpp test/loremipsum.cpp test/test_dirtree.cpp test/filter_test_data.cpp) @@ -674,6 +674,10 @@ if(WITH_FUZZ) add_executable(fuzz_categorizers test/fuzz_categorizers.cpp) target_link_libraries(fuzz_categorizers "$") list(APPEND BINARY_TARGETS fuzz_categorizers) + + add_executable(fuzz_mkdwarfs test/fuzz_mkdwarfs.cpp) + target_link_libraries(fuzz_mkdwarfs mkdwarfs_main test_helpers) + list(APPEND BINARY_TARGETS fuzz_mkdwarfs) endif() if(WITH_MAN_PAGES) diff --git a/test/fuzz_mkdwarfs.cpp b/test/fuzz_mkdwarfs.cpp new file mode 100644 index 00000000..47e20842 --- /dev/null +++ b/test/fuzz_mkdwarfs.cpp @@ -0,0 +1,63 @@ +/* vim:set ts=2 sw=2 sts=2 et: */ +/** + * \author Marcus Holland-Moritz (github@mhxnet.de) + * \copyright Copyright (c) Marcus Holland-Moritz + * + * This file is part of dwarfs. + * + * dwarfs is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * dwarfs is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with dwarfs. If not, see . + */ + +#include + +#include + +#include "dwarfs_tool_main.h" +#include "test_helpers.h" + +using namespace dwarfs; + +int main(int argc, char** argv) { + if (argc < 2) { + return 1; + } + + std::shared_ptr os{ + test::os_access_mock::create_test_instance()}; + +#ifdef __AFL_LOOP + while (__AFL_LOOP(10000)) +#endif + { + std::string cmdline; + if (!folly::readFile(argv[1], cmdline)) { + std::terminate(); + } + + std::shared_ptr fa{ + std::make_shared()}; + std::unique_ptr iol{ + std::make_unique(os, fa)}; + + auto args = test::parse_args(cmdline); + args.insert(args.begin(), "mkdwarfs"); + + try { + mkdwarfs_main(args, iol->get()); + } catch (const std::exception&) { + } + } + + return 0; +}