From f20cec4985e5334dddd22f4c07df46694f000456 Mon Sep 17 00:00:00 2001
From: Ivan Nardi <12729895+IvanNardi@users.noreply.github.com>
Date: Wed, 11 Dec 2024 16:41:35 +0100
Subject: fuzz: improve fuzzing coverage (#2642)

Updtae pl7m code (Fix swap-direction mutation)
---
 fuzz/fuzz_config.cpp    | 10 ++++++++++
 fuzz/fuzz_ndpi_reader.c |  1 +
 2 files changed, 11 insertions(+)

(limited to 'fuzz')

diff --git a/fuzz/fuzz_config.cpp b/fuzz/fuzz_config.cpp
index 61fbce5ad..c8d8d7a65 100644
--- a/fuzz/fuzz_config.cpp
+++ b/fuzz/fuzz_config.cpp
@@ -377,6 +377,16 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
     snprintf(cfg_value, sizeof(cfg_value), "%d", value);
     ndpi_set_config(ndpi_info_mod, NULL, "dpi.guess_ip_before_port", cfg_value);
   }
+  if(fuzzed_data.ConsumeBool()) {
+    value = fuzzed_data.ConsumeIntegralInRange(0, 1 + 1);
+    snprintf(cfg_value, sizeof(cfg_value), "%d", value);
+    ndpi_set_config(ndpi_info_mod, NULL, "dpi.compute_entropy", cfg_value);
+  }
+  if(fuzzed_data.ConsumeBool()) {
+    value = fuzzed_data.ConsumeIntegralInRange(0, 16777215 + 1);
+    snprintf(cfg_value, sizeof(cfg_value), "%d", value);
+    ndpi_set_config(ndpi_info_mod, NULL, "dpi.address_cache_size", cfg_value);
+  }
   if(fuzzed_data.ConsumeBool()) {
     value = fuzzed_data.ConsumeIntegralInRange(0, 1 + 1);
     snprintf(cfg_value, sizeof(cfg_value), "%d", value);
diff --git a/fuzz/fuzz_ndpi_reader.c b/fuzz/fuzz_ndpi_reader.c
index 4e736909d..de38f95ac 100644
--- a/fuzz/fuzz_ndpi_reader.c
+++ b/fuzz/fuzz_ndpi_reader.c
@@ -98,6 +98,7 @@ int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
     ndpi_set_config(workflow->ndpi_struct, "tls", "dpi.heuristics", "0x07");
     ndpi_set_config(workflow->ndpi_struct, "tls", "dpi.heuristics.max_packets_extra_dissection", "40");
     ndpi_set_config(workflow->ndpi_struct, "stun", "monitoring", "1");
+    ndpi_set_config(workflow->ndpi_struct, NULL, "dpi.address_cache_size", "8192");
 
     ndpi_finalize_initialization(workflow->ndpi_struct);
 
-- 
cgit v1.2.3