diff options
Diffstat (limited to 'fuzz/fuzz_readerutils_workflow.cpp')
-rw-r--r-- | fuzz/fuzz_readerutils_workflow.cpp | 21 |
1 files changed, 9 insertions, 12 deletions
diff --git a/fuzz/fuzz_readerutils_workflow.cpp b/fuzz/fuzz_readerutils_workflow.cpp index a92877aaa..919d87f25 100644 --- a/fuzz/fuzz_readerutils_workflow.cpp +++ b/fuzz/fuzz_readerutils_workflow.cpp @@ -25,12 +25,11 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { struct ndpi_workflow_prefs prefs; pcap_t *pcap_handle; ndpi_serialization_format serialization_format; - NDPI_PROTOCOL_BITMASK enabled_bitmask; ndpi_risk flow_risk; struct ndpi_flow_info *flow = NULL; /* unused */ const u_char *pkt; struct pcap_pkthdr *header; - int r, rc; + int r; char errbuf[PCAP_ERRBUF_SIZE]; FILE *fd; u_int8_t debug_protos_index; @@ -90,19 +89,17 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) { g_ctx = ndpi_global_init(); - w = ndpi_workflow_init(&prefs, pcap_handle, 1, serialization_format, g_ctx); + + + w = ndpi_workflow_init(&prefs, pcap_handle, 1, serialization_format, g_ctx, NULL); if(w) { - NDPI_BITMASK_SET_ALL(enabled_bitmask); - rc = ndpi_set_protocol_detection_bitmask2(w->ndpi_struct, &enabled_bitmask); - if(rc == 0) { - ndpi_finalize_initialization(w->ndpi_struct); + ndpi_finalize_initialization(w->ndpi_struct); - header = NULL; + header = NULL; + r = pcap_next_ex(pcap_handle, &header, &pkt); + while (r > 0) { + ndpi_workflow_process_packet(w, header, pkt, &flow_risk, &flow); r = pcap_next_ex(pcap_handle, &header, &pkt); - while (r > 0) { - ndpi_workflow_process_packet(w, header, pkt, &flow_risk, &flow); - r = pcap_next_ex(pcap_handle, &header, &pkt); - } } ndpi_workflow_free(w); |