chore: add simple mkdwarfs fuzzer (wip)

This commit is contained in:
Marcus Holland-Moritz 2024-01-10 23:04:00 +01:00
parent a83bea3852
commit 7389b56f22
2 changed files with 68 additions and 1 deletions

View File

@ -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 "$<LINK_LIBRARY:WHOLE_ARCHIVE,dwarfs_categorizer>")
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)

63
test/fuzz_mkdwarfs.cpp Normal file
View File

@ -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 <https://www.gnu.org/licenses/>.
*/
#include <filesystem>
#include <folly/FileUtil.h>
#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<test::os_access_mock> 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<test::test_file_access> fa{
std::make_shared<test::test_file_access>()};
std::unique_ptr<test::test_iolayer> iol{
std::make_unique<test::test_iolayer>(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;
}