vppinfra: fix memory trace

Type: fix

reset the memory trace if mem trace is turned on

Signed-off-by: Leung Lai Yung <benkerbuild@gmail.com>
Change-Id: Ib99355b9ed42ff66c720bbea5cbbf03c65820d12
diff --git a/src/vppinfra/mem_dlmalloc.c b/src/vppinfra/mem_dlmalloc.c
index ffffec8..25014c8 100644
--- a/src/vppinfra/mem_dlmalloc.c
+++ b/src/vppinfra/mem_dlmalloc.c
@@ -698,6 +698,11 @@
       mspace_realloc_in_place (h->mspace, p, new_size))
     {
       clib_mem_unpoison (p, new_size);
+      if (PREDICT_FALSE (h->flags & CLIB_MEM_HEAP_F_TRACED))
+	{
+	  mheap_put_trace (pointer_to_uword (p), old_alloc_size);
+	  mheap_get_trace (pointer_to_uword (p), clib_mem_size (p));
+	}
     }
   else
     {