1 files changed, 71 insertions, 0 deletions
diff --git a/fuzz/fuzz_ds_ptree.cpp b/fuzz/fuzz_ds_ptree.cpp
new file mode 100644
index 000000000..ae92a0257
--- /dev/null
+++ b/fuzz/fuzz_ds_ptree.cpp
@@ -0,0 +1,71 @@
+#include "ndpi_api.h"
+#include "fuzz_common_code.h"
+
+#include <stdint.h>
+#include <stdio.h>
+#include <assert.h>
+#include "fuzzer/FuzzedDataProvider.h"
+
+
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
+  FuzzedDataProvider fuzzed_data(data, size);
+  u_int16_t i, num_iteration;
+  ndpi_ptree_t *t;
+  ndpi_ip_addr_t addr, addr_added;
+  u_int8_t bits;
+  int rc, is_added = 0;
+  u_int64_t user_data;
+
+  /* To allow memory allocation failures */
+  fuzz_set_alloc_callbacks_and_seed(size);
+
+  t = ndpi_ptree_create();
+
+  /* Random insert */
+  num_iteration = fuzzed_data.ConsumeIntegral<u_int8_t>();
+  for (i = 0; i < num_iteration; i++) {
+    if (fuzzed_data.ConsumeBool()) {
+      if(fuzzed_data.remaining_bytes() > 16) {
+	memcpy(&addr.ipv6, fuzzed_data.ConsumeBytes<u_int8_t>(16).data(), 16);
+        bits = fuzzed_data.ConsumeIntegralInRange(0, 128);
+      } else {
+        continue;
+      }
+    } else {
+      memset(&addr, '\0', sizeof(addr));
+      addr.ipv4 = fuzzed_data.ConsumeIntegral<u_int32_t>();
+      bits = fuzzed_data.ConsumeIntegralInRange(0, 32);
+    };
+
+    rc = ndpi_ptree_insert(t, &addr, bits, 0);
+    /* Keep one random node really added */
+    if (rc == 0 && is_added == 0 && fuzzed_data.ConsumeBool()) {
+      is_added = 1;
+      addr_added = addr;
+    }
+  }
+
+  /* Random search */
+  num_iteration = fuzzed_data.ConsumeIntegral<u_int8_t>();
+  for (i = 0; i < num_iteration; i++) {
+    if (fuzzed_data.ConsumeBool()) {
+      if(fuzzed_data.remaining_bytes() > 16) {
+	memcpy(&addr.ipv6, fuzzed_data.ConsumeBytes<u_int8_t>(16).data(), 16);
+      } else {
+        continue;
+      }
+    } else {
+      memset(&addr, '\0', sizeof(addr));
+      addr.ipv4 = fuzzed_data.ConsumeIntegral<u_int32_t>();
+    };
+
+    ndpi_ptree_match_addr(t, &addr, &user_data);
+  }
+  /* Search of an added node */
+  if (is_added)
+    ndpi_ptree_match_addr(t, &addr_added, &user_data);
+
+  ndpi_ptree_destroy(t);
+
+  return 0;
+}