diff options
Diffstat (limited to 'src/lib/ndpi_main.c')
| -rw-r--r-- | src/lib/ndpi_main.c | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/src/lib/ndpi_main.c b/src/lib/ndpi_main.c index 59f21c555..80b9f6d42 100644 --- a/src/lib/ndpi_main.c +++ b/src/lib/ndpi_main.c @@ -329,8 +329,10 @@ u_int16_t ndpi_map_user_proto_id_to_ndpi_id(struct ndpi_detection_module_struct u_int16_t user_proto_id) { #ifdef NDPI_ENABLE_DEBUG_MESSAGES +#if 0 /* Too much verbose... */ NDPI_LOG_DBG2(ndpi_str, "[DEBUG] ***** %s(%u)\n", __FUNCTION__, user_proto_id); #endif +#endif if(user_proto_id < NDPI_MAX_SUPPORTED_PROTOCOLS) return(user_proto_id); @@ -356,8 +358,10 @@ u_int16_t ndpi_map_user_proto_id_to_ndpi_id(struct ndpi_detection_module_struct u_int16_t ndpi_map_ndpi_id_to_user_proto_id(struct ndpi_detection_module_struct *ndpi_str, u_int16_t ndpi_proto_id) { #ifdef NDPI_ENABLE_DEBUG_MESSAGES +#if 0 /* Too much verbose... */ NDPI_LOG_DBG2(ndpi_str, "[DEBUG] ***** %s(%u)\n", __FUNCTION__, ndpi_proto_id); #endif +#endif if(ndpi_proto_id < NDPI_MAX_SUPPORTED_PROTOCOLS) return(ndpi_proto_id); @@ -2996,6 +3000,9 @@ struct ndpi_detection_module_struct *ndpi_init_detection_module(ndpi_init_prefs ndpi_str->opportunistic_tls_pop_enabled = 1; ndpi_str->opportunistic_tls_ftp_enabled = 1; + ndpi_str->monitoring_stun_pkts_to_process = 4; + ndpi_str->monitoring_stun_flags = 0; + ndpi_str->aggressiveness_ookla = NDPI_AGGRESSIVENESS_OOKLA_TLS; if(prefs & ndpi_enable_tcp_ack_payload_heuristic) @@ -9774,6 +9781,42 @@ int ndpi_seen_flow_beginning(const struct ndpi_flow_struct *flow) /* ******************************************************************** */ +int ndpi_set_monitoring_state(struct ndpi_detection_module_struct *ndpi_struct, + u_int16_t proto, u_int32_t num_pkts, u_int32_t flags) +{ + if(!ndpi_struct || num_pkts > 0xFFFF) + return -1; + + switch(proto) { + case NDPI_PROTOCOL_STUN: + ndpi_struct->monitoring_stun_pkts_to_process = num_pkts; + ndpi_struct->monitoring_stun_flags = flags; + return 0; + default: + return -1; + } +} + +/* ******************************************************************** */ + +int ndpi_get_monitoring_state(struct ndpi_detection_module_struct *ndpi_struct, + u_int16_t proto, u_int32_t *num_pkts, u_int32_t *flags) +{ + if(!ndpi_struct || !num_pkts || !flags) + return -1; + + switch(proto) { + case NDPI_PROTOCOL_STUN: + *num_pkts = ndpi_struct->monitoring_stun_pkts_to_process; + *flags = ndpi_struct->monitoring_stun_flags; + return 0; + default: + return -1; + } +} + +/* ******************************************************************** */ + int ndpi_set_opportunistic_tls(struct ndpi_detection_module_struct *ndpi_struct, u_int16_t proto, int value) { |