diff options
Diffstat (limited to 'src/lib')
-rw-r--r-- | src/lib/ndpi_main.c | 13 | ||||
-rw-r--r-- | src/lib/protocols/hangout.c | 2 | ||||
-rw-r--r-- | src/lib/protocols/stun.c | 5 |
3 files changed, 20 insertions, 0 deletions
diff --git a/src/lib/ndpi_main.c b/src/lib/ndpi_main.c index a84d650bc..efd8a18a5 100644 --- a/src/lib/ndpi_main.c +++ b/src/lib/ndpi_main.c @@ -1968,6 +1968,15 @@ static const char *categories[] = { /* ******************************************************************** */ +#ifdef TEST_LRU_HANDLER +void test_lru_handler(ndpi_lru_cache_type cache_type, u_int32_t proto, u_int32_t app_proto) { + + printf("[test_lru_handler] %u / %u / %u\n", cache_type, proto, app_proto); +} +#endif + +/* ******************************************************************** */ + struct ndpi_detection_module_struct *ndpi_init_detection_module(ndpi_init_prefs prefs) { struct ndpi_detection_module_struct *ndpi_str = ndpi_malloc(sizeof(struct ndpi_detection_module_struct)); int i; @@ -1982,6 +1991,10 @@ struct ndpi_detection_module_struct *ndpi_init_detection_module(ndpi_init_prefs memset(ndpi_str, 0, sizeof(struct ndpi_detection_module_struct)); +#ifdef TEST_LRU_HANDLER + ndpi_str->ndpi_notify_lru_add_handler_ptr = test_lru_handler; +#endif + #ifdef NDPI_ENABLE_DEBUG_MESSAGES set_ndpi_debug_function(ndpi_str, (ndpi_debug_function_ptr) ndpi_debug_printf); NDPI_BITMASK_RESET(ndpi_str->debug_bitmask); diff --git a/src/lib/protocols/hangout.c b/src/lib/protocols/hangout.c index fe0d3cb05..acddfed7f 100644 --- a/src/lib/protocols/hangout.c +++ b/src/lib/protocols/hangout.c @@ -108,6 +108,8 @@ void ndpi_search_hangout(struct ndpi_detection_module_struct *ndpi_struct, #endif ndpi_lru_add_to_cache(ndpi_struct->stun_cache, key, NDPI_PROTOCOL_HANGOUT_DUO); + if(ndpi_struct->ndpi_notify_lru_add_handler_ptr) + ndpi_struct->ndpi_notify_lru_add_handler_ptr(ndpi_hangout_cache, key, NDPI_PROTOCOL_HANGOUT_DUO); } ndpi_set_detected_protocol(ndpi_struct, flow, NDPI_PROTOCOL_HANGOUT_DUO, diff --git a/src/lib/protocols/stun.c b/src/lib/protocols/stun.c index 87d090daf..703b46be7 100644 --- a/src/lib/protocols/stun.c +++ b/src/lib/protocols/stun.c @@ -88,7 +88,12 @@ void ndpi_int_stun_add_connection(struct ndpi_detection_module_struct *ndpi_stru #endif ndpi_lru_add_to_cache(ndpi_struct->stun_cache, key, app_proto); + if(ndpi_struct->ndpi_notify_lru_add_handler_ptr) + ndpi_struct->ndpi_notify_lru_add_handler_ptr(ndpi_stun_cache, key, app_proto); + ndpi_lru_add_to_cache(ndpi_struct->stun_cache, key_rev, app_proto); + if(ndpi_struct->ndpi_notify_lru_add_handler_ptr) + ndpi_struct->ndpi_notify_lru_add_handler_ptr(ndpi_stun_cache, key_rev, app_proto); } } } |