diff options
Diffstat (limited to 'src/lib/protocols/natpmp.c')
| -rw-r--r-- | src/lib/protocols/natpmp.c | 18 |
1 files changed, 6 insertions, 12 deletions
diff --git a/src/lib/protocols/natpmp.c b/src/lib/protocols/natpmp.c index 4c259c91c..bdd5b308d 100644 --- a/src/lib/protocols/natpmp.c +++ b/src/lib/protocols/natpmp.c @@ -173,7 +173,7 @@ static void ndpi_search_natpmp(struct ndpi_detection_module_struct *ndpi_struct, if (natpmp_is_valid(packet, &natpmp_type) == 0) { - NDPI_EXCLUDE_PROTO(ndpi_struct, flow); + NDPI_EXCLUDE_DISSECTOR(ndpi_struct, flow); return; } @@ -188,16 +188,10 @@ static void ndpi_search_natpmp(struct ndpi_detection_module_struct *ndpi_struct, } } -void init_natpmp_dissector(struct ndpi_detection_module_struct *ndpi_struct, - u_int32_t *id) +void init_natpmp_dissector(struct ndpi_detection_module_struct *ndpi_struct) { - ndpi_set_bitmask_protocol_detection("NAT-PMP", ndpi_struct, *id, - NDPI_PROTOCOL_NATPMP, - ndpi_search_natpmp, - NDPI_SELECTION_BITMASK_PROTOCOL_V4_V6_UDP_WITH_PAYLOAD, - SAVE_DETECTION_BITMASK_AS_UNKNOWN, - ADD_TO_DETECTION_BITMASK - ); - - *id += 1; + register_dissector("NAT-PMP", ndpi_struct, + ndpi_search_natpmp, + NDPI_SELECTION_BITMASK_PROTOCOL_V4_V6_UDP_WITH_PAYLOAD, + 1, NDPI_PROTOCOL_NATPMP); } |