From 86115a8a65c98d0665100b5ae85cc661d1404783 Mon Sep 17 00:00:00 2001 From: Nardi Ivan Date: Tue, 3 Oct 2023 09:13:43 +0200 Subject: fuzz: extend fuzzing coverage --- fuzz/fuzz_readerutils_workflow.cpp | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) (limited to 'fuzz/fuzz_readerutils_workflow.cpp') diff --git a/fuzz/fuzz_readerutils_workflow.cpp b/fuzz/fuzz_readerutils_workflow.cpp index 90d0ac094..f753560f2 100644 --- a/fuzz/fuzz_readerutils_workflow.cpp +++ b/fuzz/fuzz_readerutils_workflow.cpp @@ -31,7 +31,7 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { ndpi_risk flow_risk; const u_char *pkt; struct pcap_pkthdr *header; - int r; + int r, rc; char errbuf[PCAP_ERRBUF_SIZE]; FILE *fd; u_int8_t debug_protos_index; @@ -91,14 +91,16 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { w = ndpi_workflow_init(&prefs, pcap_handle, 1, serialization_format); if(w) { NDPI_BITMASK_SET_ALL(enabled_bitmask); - ndpi_set_protocol_detection_bitmask2(w->ndpi_struct, &enabled_bitmask); - ndpi_finalize_initialization(w->ndpi_struct); + rc = ndpi_set_protocol_detection_bitmask2(w->ndpi_struct, &enabled_bitmask); + if(rc == 0) { + ndpi_finalize_initialization(w->ndpi_struct); - header = NULL; - r = pcap_next_ex(pcap_handle, &header, &pkt); - while (r > 0) { - ndpi_workflow_process_packet(w, header, pkt, &flow_risk); + header = NULL; r = pcap_next_ex(pcap_handle, &header, &pkt); + while (r > 0) { + ndpi_workflow_process_packet(w, header, pkt, &flow_risk); + r = pcap_next_ex(pcap_handle, &header, &pkt); + } } ndpi_workflow_free(w); -- cgit v1.2.3