From e308e59002539a989e87c72bdb462b9652e304b3 Mon Sep 17 00:00:00 2001
From: Toni Uhlig <matzeton@googlemail.com>
Date: Tue, 7 Jul 2020 17:07:20 +0200
Subject: fixed memory leak in ndpi_cluster_bins / binUnitTest

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
---
 example/ndpiReader.c   | 2 ++
 src/lib/ndpi_analyze.c | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/example/ndpiReader.c b/example/ndpiReader.c
index 39e36f248..a3e35bd4f 100644
--- a/example/ndpiReader.c
+++ b/example/ndpiReader.c
@@ -3163,6 +3163,8 @@ static void binUnitTest() {
 
   for(i=0; i<num_bins; i++)
     ndpi_free_bin(&bins[i]);
+
+  free(bins);
 }
 
 /* *********************************************** */
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);
 }
 
-- 
cgit v1.2.3