Reworked fuzzer to use mem tracer (#159)

* Rewrote fuzzer to use mem tracer

Author: Justin Boswell

tests/fuzz/fuzz_h2_decoder.c

@@ -17,23 +17,17 @@
 
 #include <aws/testing/aws_test_harness.h>
 
+#include <aws/common/allocator.h>
 #include <aws/common/logging.h>
 
 AWS_EXTERN_C_BEGIN
 
-AWS_TEST_ALLOCATOR_INIT(fuzz_h2_decoder)
-
 int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
 
     /* Setup allocator and parameters */
-    struct aws_allocator *allocator = &fuzz_h2_decoder_allocator;
-    struct memory_test_allocator *alloc_impl = &fuzz_h2_decoder_alloc_impl;
+    struct aws_allocator *allocator = aws_mem_tracer_new(aws_default_allocator(), AWS_MEMTRACE_BYTES, 0);
     struct aws_byte_cursor to_decode = aws_byte_cursor_from_array(data, size);
 
-    /* Reset the allocator's leak checker */
-    alloc_impl->allocated = 0;
-    alloc_impl->freed = 0;
-
     /* Enable logging */
     struct aws_logger logger;
     struct aws_logger_standard_options log_options = {
@@ -62,7 +56,8 @@ int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
     aws_logger_clean_up(&logger);
 
     /* Check for leaks */
-    AWS_FATAL_ASSERT(alloc_impl->allocated == alloc_impl->freed);
+    ASSERT_UINT_EQUALS(0, aws_mem_tracer_count(allocator));
+    allocator = aws_mem_tracer_destroy(allocator);
 
     return 0;
 }

tests/fuzz/fuzz_h2_frames.c

@@ -16,21 +16,16 @@
 #include <aws/http/private/h2_frames.h>
 #include <aws/http/private/hpack.h>
 
+#include <aws/common/allocator.h>
 #include <aws/testing/aws_test_harness.h>
 
 AWS_EXTERN_C_BEGIN
 
-AWS_TEST_ALLOCATOR_INIT(fuzz_h2_frames)
-
 int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
 
-    struct aws_allocator *allocator = &fuzz_h2_frames_allocator;
-    struct memory_test_allocator *alloc_impl = &fuzz_h2_frames_alloc_impl;
+    struct aws_allocator *allocator = aws_mem_tracer_new(aws_default_allocator(), AWS_MEMTRACE_BYTES, 0);
     struct aws_byte_cursor to_decode = aws_byte_cursor_from_array(data, size);
 
-    AWS_ZERO_STRUCT(*alloc_impl);
-    aws_mutex_init(&alloc_impl->mutex);
-
     aws_hpack_static_table_init(allocator);
 
     struct aws_h2_frame_decoder decoder;
@@ -108,13 +103,8 @@ int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
     aws_h2_frame_decoder_clean_up(&decoder);
     aws_hpack_static_table_clean_up();
 
-    ASSERT_UINT_EQUALS(
-        alloc_impl->allocated,
-        alloc_impl->freed,
-        "Memory Leak Detected %d bytes were allocated, "
-        "but only %d were freed.",
-        alloc_impl->allocated,
-        alloc_impl->freed);
+    ASSERT_UINT_EQUALS(0, aws_mem_tracer_count(allocator), "Memory Leak Detected");
+    allocator = aws_mem_tracer_destroy(allocator);
 
     return 0;
 }