From 86115a8a65c98d0665100b5ae85cc661d1404783 Mon Sep 17 00:00:00 2001
From: Nardi Ivan <nardi.ivan@gmail.com>
Date: Tue, 3 Oct 2023 09:13:43 +0200
Subject: fuzz: extend fuzzing coverage

---
 fuzz/fuzz_ds_domain_classify.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(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 9a945deff..40acd9b18 100644
--- a/fuzz/fuzz_ds_domain_classify.cpp
+++ b/fuzz/fuzz_ds_domain_classify.cpp
@@ -33,7 +33,9 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
     }
   }
 
-  ndpi_domain_classify_add_domains(d, NDPI_PROTOCOL_UNKNOWN, "random_list.list");
+  ndpi_domain_classify_add_domains(d,
+				   fuzzed_data.ConsumeIntegralInRange(0, NDPI_LAST_IMPLEMENTED_PROTOCOL - 1),
+				   (char *)"random_list.list");
 
   if (fuzzed_data.ConsumeBool())
     ndpi_domain_classify_finalize(d);
-- 
cgit v1.2.3