aboutsummaryrefslogtreecommitdiff
path: root/src/lib/protocols/imo.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib/protocols/imo.c')
-rw-r--r--src/lib/protocols/imo.c16
1 files changed, 6 insertions, 10 deletions
diff --git a/src/lib/protocols/imo.c b/src/lib/protocols/imo.c
index 3807e5010..9b60e251f 100644
--- a/src/lib/protocols/imo.c
+++ b/src/lib/protocols/imo.c
@@ -61,21 +61,17 @@ static void ndpi_search_imo(struct ndpi_detection_module_struct *ndpi_struct, st
ndpi_int_imo_add_connection(ndpi_struct, flow);
} else {
if(flow->num_processed_pkts > 5)
- NDPI_EXCLUDE_PROTO(ndpi_struct, flow);
+ NDPI_EXCLUDE_DISSECTOR(ndpi_struct, flow);
else
flow->l4.udp.imo_last_one_byte_pkt = 0;
}
}
-void init_imo_dissector(struct ndpi_detection_module_struct *ndpi_struct,
- u_int32_t *id) {
- ndpi_set_bitmask_protocol_detection("IMO", ndpi_struct, *id,
- NDPI_PROTOCOL_IMO,
- ndpi_search_imo,
- NDPI_SELECTION_BITMASK_PROTOCOL_V4_V6_UDP_WITH_PAYLOAD,
- SAVE_DETECTION_BITMASK_AS_UNKNOWN,
- ADD_TO_DETECTION_BITMASK);
- *id += 1;
+void init_imo_dissector(struct ndpi_detection_module_struct *ndpi_struct) {
+ register_dissector("IMO", ndpi_struct,
+ ndpi_search_imo,
+ NDPI_SELECTION_BITMASK_PROTOCOL_V4_V6_UDP_WITH_PAYLOAD,
+ 1, NDPI_PROTOCOL_IMO);
}