2 files changed, 14 insertions, 0 deletions
diff --git a/src/lib/protocols/quic.c b/src/lib/protocols/quic.c
index 61e7f6880..aafeb3397 100644
--- a/src/lib/protocols/quic.c
+++ b/src/lib/protocols/quic.c
@@ -1598,8 +1598,15 @@ static int ndpi_search_quic_extra(struct ndpi_detection_module_struct *ndpi_stru
      (packet->payload[1] == 201 || /* RTCP, Receiver Report */
       packet->payload[1] == 200 || /* RTCP, Sender Report */
       is_valid_rtp_payload_type(packet->payload[1] & 0x7F)) /* RTP */) {
+    ndpi_protocol proto;
+
     NDPI_LOG_DBG(ndpi_struct, "Found RTP/RTCP over QUIC\n");
     ndpi_set_detected_protocol(ndpi_struct, flow, NDPI_PROTOCOL_SNAPCHAT_CALL, NDPI_PROTOCOL_QUIC, NDPI_CONFIDENCE_DPI);
+    /* In "extra_eval" data path, if we change the classification, we need to update the category, too */
+    proto.master_protocol = NDPI_PROTOCOL_QUIC;
+    proto.app_protocol = NDPI_PROTOCOL_SNAPCHAT_CALL;
+    proto.category = NDPI_PROTOCOL_CATEGORY_UNSPECIFIED;
+    ndpi_fill_protocol_category(ndpi_struct, flow, &proto);
   } else {
     /* Unexpected traffic pattern: we should investigate it... */
     NDPI_LOG_INFO(ndpi_struct, "To investigate...\n");
diff --git a/src/lib/protocols/skype.c b/src/lib/protocols/skype.c
index 8fb4b195b..6fece0e63 100644
--- a/src/lib/protocols/skype.c
+++ b/src/lib/protocols/skype.c
@@ -54,7 +54,14 @@ static int ndpi_check_skype_udp_again(struct ndpi_detection_module_struct *ndpi_
     }
 
     if (detected) {
+      ndpi_protocol proto;
+
       ndpi_set_detected_protocol(ndpi_struct, flow, NDPI_PROTOCOL_SKYPE_TEAMS, NDPI_PROTOCOL_UNKNOWN, NDPI_CONFIDENCE_DPI);
+      /* In "extra_eval" data path, if we change the classification, we need to update the category, too */
+      proto.master_protocol = NDPI_PROTOCOL_UNKNOWN;
+      proto.app_protocol = NDPI_PROTOCOL_SKYPE_TEAMS;
+      proto.category = NDPI_PROTOCOL_CATEGORY_UNSPECIFIED;
+      ndpi_fill_protocol_category(ndpi_struct, flow, &proto);
       flow->extra_packets_func = NULL;
 
       /* Stop checking extra packets */