Merge pull request #132 from zeux/fuzz

Improve fuzzing support

1 files changed, 8 insertions, 10 deletions

diff --git a/tests/fuzz_parse.cpp b/tests/fuzz_parse.cpp
index e758196..94c610a 100644
--- a/tests/fuzz_parse.cpp
+++ b/tests/fuzz_parse.cpp
@@ -1,16 +1,14 @@
 #include "../src/pugixml.hpp"
-#include "allocator.hpp"
 
-int main(int argc, const char** argv)
-{
-    pugi::set_memory_management_functions(memory_allocate, memory_deallocate);
+#include <stdint.h>
 
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
+{
     pugi::xml_document doc;
 
-    for (int i = 1; i < argc; ++i)
-    {
-	    doc.load_file(argv[i]);
-	    doc.load_file(argv[i], pugi::parse_minimal);
-	    doc.load_file(argv[i], pugi::parse_full);
-	}
+	doc.load_buffer(Data, Size);
+	doc.load_buffer(Data, Size, pugi::parse_minimal);
+	doc.load_buffer(Data, Size, pugi::parse_full);
+
+	return 0;
 }