Fix some race conditions by using atomic operations. (#1420)

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

1 files changed, 3 insertions, 3 deletions

diff --git a/src/lib/ndpi_main.c b/src/lib/ndpi_main.c
index 54f03916d..3cfc2b2ce 100644
--- a/src/lib/ndpi_main.c
+++ b/src/lib/ndpi_main.c
@@ -145,13 +145,13 @@ static u_int32_t ndpi_tot_allocated_memory;
 /* ****************************************** */
 
 u_int32_t ndpi_get_tot_allocated_memory() {
-  return(ndpi_tot_allocated_memory);
+  return(__sync_fetch_and_add(&ndpi_tot_allocated_memory, 0));
 }
 
 /* ****************************************** */
 
 void *ndpi_malloc(size_t size) {
-  ndpi_tot_allocated_memory += size;
+  __sync_fetch_and_add(&ndpi_tot_allocated_memory, size);
   return(_ndpi_malloc ? _ndpi_malloc(size) : malloc(size));
 }
 
@@ -169,7 +169,7 @@ void *ndpi_calloc(unsigned long count, size_t size) {
 
   if(p) {
     memset(p, 0, len);
-    ndpi_tot_allocated_memory += size;
+    __sync_fetch_and_add(&ndpi_tot_allocated_memory, size);
   }
 
   return(p);