fixed memory leak in ndpi_cluster_bins / binUnitTest

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>

1 files changed, 3 insertions, 1 deletions

diff --git a/src/lib/ndpi_analyze.c b/src/lib/ndpi_analyze.c
index 74adbb2c9..07fc09362 100644
--- a/src/lib/ndpi_analyze.c
+++ b/src/lib/ndpi_analyze.c
@@ -486,7 +486,7 @@ int ndpi_cluster_bins(struct ndpi_bin *bins, u_int16_t num_bins,
   
   if(num_clusters > num_bins) return(-1);
 
-  if((centroids = (struct ndpi_bin*)malloc(sizeof(struct ndpi_bin)*num_clusters)) == NULL)
+  if((centroids = (struct ndpi_bin*)ndpi_malloc(sizeof(struct ndpi_bin)*num_clusters)) == NULL)
     return(-2);
   else {
     for(i=0; i<num_clusters; i++)
@@ -571,6 +571,8 @@ int ndpi_cluster_bins(struct ndpi_bin *bins, u_int16_t num_bins,
   for(i=0; i<num_clusters; i++)
     ndpi_free_bin(&centroids[i]);
 
+  ndpi_free(centroids);
+
   return(0);
 }