From 43f7dc9ba08d8a840827646133ac3dc100b8610c Mon Sep 17 00:00:00 2001
From: Ivan Nardi <12729895+IvanNardi@users.noreply.github.com>
Date: Wed, 20 Nov 2024 13:36:41 +0100
Subject: fuzz: extend fuzzing coverage (#2626)

---
 fuzz/fuzz_config.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'fuzz/fuzz_config.cpp')

diff --git a/fuzz/fuzz_config.cpp b/fuzz/fuzz_config.cpp
index 6ca213cce..0bed755f8 100644
--- a/fuzz/fuzz_config.cpp
+++ b/fuzz/fuzz_config.cpp
@@ -36,6 +36,7 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
   char cfg_param[32];
   u_int64_t cat_userdata = 0;
   u_int16_t unused1, unused2;
+  ndpi_master_app_protocol proto1, proto2;
 
   /* To allow memory allocation failures */
   fuzz_set_alloc_callbacks_and_seed(size);
@@ -645,6 +646,12 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
   ndpi_fpc_confidence_get_name(static_cast<ndpi_fpc_confidence_t>(fuzzed_data.ConsumeIntegral<u_int8_t>()));
   ndpi_get_proto_breed_name(static_cast<ndpi_protocol_breed_t>(fuzzed_data.ConsumeIntegral<u_int8_t>()));
   ndpi_get_l4_proto_name(static_cast<ndpi_l4_proto_info>(fuzzed_data.ConsumeIntegral<u_int8_t>()));
+  proto1.app_protocol = fuzzed_data.ConsumeIntegral<u_int16_t>();
+  proto1.master_protocol = fuzzed_data.ConsumeIntegral<u_int16_t>();
+  ndpi_is_proto_unknown(proto1);
+  proto2.app_protocol = fuzzed_data.ConsumeIntegral<u_int16_t>();
+  proto2.master_protocol = fuzzed_data.ConsumeIntegral<u_int16_t>();
+  ndpi_is_proto_equals(proto1, proto2, fuzzed_data.ConsumeBool());
 
   char buf2[16];
   ndpi_entropy2str(fuzzed_data.ConsumeFloatingPoint<float>(), fuzzed_data.ConsumeBool() ? buf2 : NULL, sizeof(buf2));
-- 
cgit v1.2.3