From e3d15ef4e7f896ca62773cbd9b207967055d96b0 Mon Sep 17 00:00:00 2001 From: Vito Piserchia Date: Thu, 3 Nov 2016 15:23:06 +0100 Subject: Prevent Segfault when building with -DNDPI_ENABLE_DEBUG_MESSAGES --- src/lib/ndpi_main.c | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'src/lib/ndpi_main.c') diff --git a/src/lib/ndpi_main.c b/src/lib/ndpi_main.c index c6a39394d..c8d2d3d7d 100644 --- a/src/lib/ndpi_main.c +++ b/src/lib/ndpi_main.c @@ -1716,7 +1716,10 @@ void set_ndpi_free(void (*__ndpi_free)(void *ptr)) { _ndpi_free = __ndpi_ void set_ndpi_debug_function(struct ndpi_detection_module_struct *ndpi_str, ndpi_debug_function_ptr ndpi_debug_printf) { #ifdef NDPI_ENABLE_DEBUG_MESSAGES - ndpi_str->ndpi_debug_printf = ndpi_debug_printf; + if (ndpi_debug_printf != NULL) + ndpi_str->ndpi_debug_printf = ndpi_debug_printf; + else + ndpi_str->ndpi_debug_printf = (ndpi_debug_function_ptr) printf; #endif } @@ -1727,7 +1730,7 @@ struct ndpi_detection_module_struct *ndpi_init_detection_module(void) { if(ndpi_str == NULL) { #ifdef NDPI_ENABLE_DEBUG_MESSAGES - ndpi_debug_printf(0, NULL, NDPI_LOG_DEBUG, "ndpi_init_detection_module initial malloc failed\n"); + printf(0, NULL, NDPI_LOG_DEBUG, "ndpi_init_detection_module initial malloc failed\n"); #endif return NULL; } -- cgit v1.2.3