1 files changed, 4 insertions, 1 deletions
diff --git a/src/lib/ndpi_domain_classify.c b/src/lib/ndpi_domain_classify.c
index f1a319067..385796e08 100644
--- a/src/lib/ndpi_domain_classify.c
+++ b/src/lib/ndpi_domain_classify.c
@@ -35,7 +35,10 @@
 ndpi_domain_classify* ndpi_domain_classify_alloc() {
   int i;
   ndpi_domain_classify *cat = (ndpi_domain_classify*)ndpi_malloc(sizeof(ndpi_domain_classify));
-  
+
+  if(!cat)
+    return NULL;
+
   for(i=0; i<MAX_NUM_NDPI_DOMAIN_CLASSIFICATIONS; i++)
     cat->classes[i].class_id = 0, cat->classes[i].domains = NULL;