From 70814002a98ec52deda2ff61ac03613916c3efeb Mon Sep 17 00:00:00 2001
From: Nardi Ivan <nardi.ivan@gmail.com>
Date: Fri, 15 Sep 2023 20:30:30 +0200
Subject: fuzz: extend fuzzing coverage

---
 fuzz/fuzz_ds_domain_classify.cpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'fuzz/fuzz_ds_domain_classify.cpp')

diff --git a/fuzz/fuzz_ds_domain_classify.cpp b/fuzz/fuzz_ds_domain_classify.cpp
index afd43a796..9a945deff 100644
--- a/fuzz/fuzz_ds_domain_classify.cpp
+++ b/fuzz/fuzz_ds_domain_classify.cpp
@@ -33,6 +33,11 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
     }
   }
 
+  ndpi_domain_classify_add_domains(d, NDPI_PROTOCOL_UNKNOWN, "random_list.list");
+
+  if (fuzzed_data.ConsumeBool())
+    ndpi_domain_classify_finalize(d);
+
   /* "Random" search */
   num_iteration = fuzzed_data.ConsumeIntegral<u_int8_t>();
   for (i = 0; i < num_iteration; i++) {
-- 
cgit v1.2.3