| |
| #include <filesystem> |
| #include <string> |
| |
| #include "binary_annotator.h" |
| #include "test_init.h" |
| |
| static std::filesystem::path exe_path_; |
| static const uint8_t *schema_bfbs_; |
| static size_t schema_bfbs_length_; |
| |
| bool TestFileExists(std::filesystem::path file_path) { |
| if (file_path.has_filename() && std::filesystem::exists(file_path)) |
| return true; |
| |
| TEST_OUTPUT_LINE("@DEBUG: file '%s' not found", file_path.string().c_str()); |
| for (const auto &entry : |
| std::filesystem::directory_iterator(file_path.parent_path())) { |
| TEST_OUTPUT_LINE("@DEBUG: parent path entry: '%s'", |
| entry.path().string().c_str()); |
| } |
| return false; |
| } |
| |
| std::string LoadBinarySchema(const char *file_name) { |
| const auto file_path = exe_path_.parent_path() / file_name; |
| TEST_EQ(true, TestFileExists(file_path)); |
| std::string schemafile; |
| TEST_EQ(true, |
| flatbuffers::LoadFile(file_path.string().c_str(), true, &schemafile)); |
| |
| flatbuffers::Verifier verifier( |
| reinterpret_cast<const uint8_t *>(schemafile.c_str()), schemafile.size()); |
| TEST_EQ(true, reflection::VerifySchemaBuffer(verifier)); |
| return schemafile; |
| } |
| |
| extern "C" int LLVMFuzzerInitialize(int *, char ***argv) { |
| exe_path_ = (*argv)[0]; |
| static const std::string schema_file = |
| LoadBinarySchema("annotated_binary.bfbs"); |
| schema_bfbs_ = reinterpret_cast<const uint8_t *>(schema_file.c_str()); |
| schema_bfbs_length_ = schema_file.size(); |
| return 0; |
| } |
| |
| extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { |
| flatbuffers::BinaryAnnotator annotator(schema_bfbs_, schema_bfbs_length_, |
| data, size, false); |
| |
| annotator.Annotate(); |
| return 0; |
| } |