From f3af39ee42b954ec0486986c7cfac9ee44cd63e4 Mon Sep 17 00:00:00 2001 From: Luca Deri Date: Sun, 16 Jan 2022 12:47:33 +0100 Subject: Added performance tests tools --- src/lib/ndpi_main.c | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) (limited to 'src/lib') diff --git a/src/lib/ndpi_main.c b/src/lib/ndpi_main.c index 8d88b5c2e..20df63de6 100644 --- a/src/lib/ndpi_main.c +++ b/src/lib/ndpi_main.c @@ -139,9 +139,23 @@ static inline uint8_t flow_is_proto(struct ndpi_flow_struct *flow, u_int16_t p) /* ****************************************** */ +static u_int32_t ndpi_tot_allocated_memory; + +/* ****************************************** */ + +u_int32_t ndpi_get_tot_allocated_memory() { + return(ndpi_tot_allocated_memory); +} + +/* ****************************************** */ + void *ndpi_malloc(size_t size) { + ndpi_tot_allocated_memory += size; return(_ndpi_malloc ? _ndpi_malloc(size) : malloc(size)); } + +/* ****************************************** */ + void *ndpi_flow_malloc(size_t size) { return(_ndpi_flow_malloc ? _ndpi_flow_malloc(size) : ndpi_malloc(size)); } @@ -152,9 +166,11 @@ void *ndpi_calloc(unsigned long count, size_t size) { size_t len = count * size; void *p = ndpi_malloc(len); - if(p) + if(p) { memset(p, 0, len); - + ndpi_tot_allocated_memory += size; + } + return(p); } -- cgit v1.2.3