1 files changed, 25 insertions, 24 deletions

diff --git a/src/lib/protocols/bfcp.c b/src/lib/protocols/bfcp.c
index b2bd52bdb..8e7917b28 100644
--- a/src/lib/protocols/bfcp.c
+++ b/src/lib/protocols/bfcp.c
@@ -43,43 +43,44 @@ static void ndpi_search_bfcp(struct ndpi_detection_module_struct *ndpi_struct,
   }
 
   u_int8_t version = (packet->payload[0] >> 5) & 0x07;
-  u_int8_t reserved = (packet->payload[0] >> 3) & 0x01;
+  u_int8_t reserved = (packet->payload[0] & 0x03);
 
-  if (version != 1 || reserved != 0) {
+  /* RFC4582: 1
+     RFC8855: 1 on TCP, 2 on UDP */
+  if (!(version == 1 ||
+        (version == 2 && flow->l4_proto == IPPROTO_UDP))) {
+    goto not_bfcp;
+  }
+  if (reserved != 0) {
     goto not_bfcp;
   }
 
   u_int8_t primitive = packet->payload[1];
-  if (primitive < 1 || primitive > 17) {
+  if (primitive < 1 || primitive > 18) {
     goto not_bfcp;
   }
 
-  u_int32_t conference_id = ntohl(get_u_int32_t(packet->payload, 4));
-  if (!flow->bfcp_stage) {
-    flow->bfcp_conference_id = conference_id;
-    flow->bfcp_stage = 1;
-    return;
+  u_int16_t length = ntohs(get_u_int16_t(packet->payload, 2));
+  if (12 + length * 4 != packet->payload_packet_len) {
+    goto not_bfcp;
   }
 
-  if (flow->bfcp_stage && flow->bfcp_conference_id == conference_id) {
-    NDPI_LOG_INFO(ndpi_struct, "found BFCP\n");
-    ndpi_set_detected_protocol(ndpi_struct, flow, NDPI_PROTOCOL_BFCP,
-                               NDPI_PROTOCOL_UNKNOWN, NDPI_CONFIDENCE_DPI);
-    return;
-  }
+  NDPI_LOG_INFO(ndpi_struct, "found BFCP\n");
+  ndpi_set_detected_protocol(ndpi_struct, flow, NDPI_PROTOCOL_BFCP,
+                                               NDPI_PROTOCOL_UNKNOWN, NDPI_CONFIDENCE_DPI);
+
+  flow->protos.bfcp.conference_id = ntohl(get_u_int32_t(packet->payload, 4));
+  flow->protos.bfcp.user_id = ntohs(get_u_int16_t(packet->payload, 10));
+  return;
 
 not_bfcp:
-  NDPI_EXCLUDE_PROTO(ndpi_struct, flow);
+  NDPI_EXCLUDE_DISSECTOR(ndpi_struct, flow);
 }
 
-void init_bfcp_dissector(struct ndpi_detection_module_struct *ndpi_struct, u_int32_t *id)
+void init_bfcp_dissector(struct ndpi_detection_module_struct *ndpi_struct)
 {
-  ndpi_set_bitmask_protocol_detection("BFCP", ndpi_struct, *id,
-                                      NDPI_PROTOCOL_BFCP,
-                                      ndpi_search_bfcp,
-                                      NDPI_SELECTION_BITMASK_PROTOCOL_V4_V6_TCP_OR_UDP_WITH_PAYLOAD_WITHOUT_RETRANSMISSION,
-                                      SAVE_DETECTION_BITMASK_AS_UNKNOWN,
-                                      ADD_TO_DETECTION_BITMASK);
-
-  *id += 1;
+  register_dissector("BFCP", ndpi_struct,
+                     ndpi_search_bfcp,
+                     NDPI_SELECTION_BITMASK_PROTOCOL_V4_V6_TCP_OR_UDP_WITH_PAYLOAD_WITHOUT_RETRANSMISSION,
+                     1, NDPI_PROTOCOL_BFCP);
 }