1 files changed, 12 insertions, 4 deletions
diff --git a/src/lib/ndpi_main.c b/src/lib/ndpi_main.c
index a9df32e00..0d5d376a2 100644
--- a/src/lib/ndpi_main.c
+++ b/src/lib/ndpi_main.c
@@ -10972,13 +10972,21 @@ u_int16_t ndpi_get_protocol_id(struct ndpi_detection_module_struct *ndpi_str, ch
 int ndpi_get_category_id(struct ndpi_detection_module_struct *ndpi_str, char *cat) {
   int i;
 
-  if(!ndpi_str) return(-1);
+  if(!ndpi_str || !cat)
+    return(-1);
+
+  if (*cat == '\0')
+   return(-1);
+
+  /* Cache the lowercased first character of 'cat' */
+  const unsigned char fc = tolower((unsigned char)*cat);
 
   for(i = 0; i < NDPI_PROTOCOL_NUM_CATEGORIES; i++) {
     const char *name = ndpi_category_get_name(ndpi_str, i);
-
-    if(strcasecmp(cat, name) == 0)
-      return(i);
+    if(name && tolower((unsigned char)*name) == fc) {
+      if(strcasecmp(name + 1, cat + 1) == 0)
+        return(i);
+    }
   }
 
   return(-1);