add heaptrack support

status-im · May 24, 2022 · 577ad68 · 577ad68
1 parent c6a9f27
commit 577ad68
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/lib/system/alloc.nim b/lib/system/alloc.nim
@@ -740,6 +740,15 @@ when false:
       else:
         result = nil
 
+when defined(heaptracker):
+  const heaptrackLib =
+    when defined(heaptracker_inject):
+      "libheaptrack_inject.so"
+    else:
+      "libheaptrack_preload.so"
+  proc heaptrack_malloc(a: pointer, size: int) {.cdecl, importc, dynlib:heaptrackLib.}
+  proc heaptrack_free(a: pointer) {.cdecl, importc, dynlib:heaptrackLib.}
+
 proc rawAlloc(a: var MemRegion, requestedSize: int): pointer =
   when defined(nimTypeNames):
     inc(a.allocCounter)
@@ -812,13 +821,18 @@ proc rawAlloc(a: var MemRegion, requestedSize: int): pointer =
   sysAssert(allocInv(a), "rawAlloc: end")
   when logAlloc: cprintf("var pointer_%p = alloc(%ld)\n", result, requestedSize)
 
+  when defined(heaptracker):
+    heaptrack_malloc(result, requestedSize)
+
 proc rawAlloc0(a: var MemRegion, requestedSize: int): pointer =
   result = rawAlloc(a, requestedSize)
   zeroMem(result, requestedSize)
 
 proc rawDealloc(a: var MemRegion, p: pointer) =
   when defined(nimTypeNames):
     inc(a.deallocCounter)
+  when defined(heaptracker):
+    heaptrack_free(p)
   #sysAssert(isAllocatedPtr(a, p), "rawDealloc: no allocated pointer")
   sysAssert(allocInv(a), "rawDealloc: begin")
   var c = pageAddr(p)