Merge branch 'vel21ripn-dev' into dev

1 files changed, 8 insertions, 9 deletions

diff --git a/src/lib/protocols/stealthnet.c b/src/lib/protocols/stealthnet.c
index 09e6c18d4..8bd75b1a3 100644
--- a/src/lib/protocols/stealthnet.c
+++ b/src/lib/protocols/stealthnet.c
@@ -22,11 +22,14 @@
  * 
  */
 
-
-#include "ndpi_protocols.h"
+#include "ndpi_protocol_ids.h"
 
 #ifdef NDPI_PROTOCOL_STEALTHNET
 
+#define NDPI_CURRENT_PROTO NDPI_PROTOCOL_STEALTHNET
+
+#include "ndpi_api.h"
+
 
 static void ndpi_int_stealthnet_add_connection(struct ndpi_detection_module_struct
 					       *ndpi_struct, struct ndpi_flow_struct *flow)
@@ -39,20 +42,16 @@ void ndpi_search_stealthnet(struct ndpi_detection_module_struct
 {
   struct ndpi_packet_struct *packet = &flow->packet;
 	
-
-  //  struct ndpi_id_struct *src = flow->src;
-  //  struct ndpi_id_struct *dst = flow->dst;
-
+  NDPI_LOG_DBG(ndpi_struct, "search stealthnet\n");
 
   if (packet->payload_packet_len > 40
       && memcmp(packet->payload, "LARS REGENSBURGER'S FILE SHARING PROTOCOL", 41) == 0) {
-    NDPI_LOG(NDPI_PROTOCOL_STEALTHNET, ndpi_struct, NDPI_LOG_DEBUG, "found stealthnet\n");
+    NDPI_LOG_INFO(ndpi_struct, "found stealthnet\n");
     ndpi_int_stealthnet_add_connection(ndpi_struct, flow);
     return;
   }
 
-  NDPI_LOG(NDPI_PROTOCOL_STEALTHNET, ndpi_struct, NDPI_LOG_DEBUG, "exclude stealthnet.\n");
-  NDPI_ADD_PROTOCOL_TO_BITMASK(flow->excluded_protocol_bitmask, NDPI_PROTOCOL_STEALTHNET);
+  NDPI_EXCLUDE_PROTO(ndpi_struct, flow);
 
 }