Track all the things.

Author: itamarst

filprofiler/_filpreload.c

@@ -101,7 +101,7 @@ __attribute__((visibility("default"))) void fil_dump_peak_to_flamegraph(const ch
 // Override memory-allocation functions:
 __attribute__((visibility("default"))) void *malloc(size_t size) {
   void *result = __libc_malloc(size);
-  if (size > 1000 && result != NULL && !will_i_be_reentrant && initialized) {
+  if (!will_i_be_reentrant && initialized) {
     will_i_be_reentrant = 1;
     add_allocation_hook((size_t)result, size);
     will_i_be_reentrant = 0;
@@ -112,7 +112,7 @@ __attribute__((visibility("default"))) void *malloc(size_t size) {
 __attribute__((visibility("default"))) void *calloc(size_t nmemb, size_t size) {
   void *result = __libc_calloc(nmemb, size);
   size_t allocated = nmemb * size;
-  if (allocated > 1000 && !will_i_be_reentrant && initialized) {
+  if (!will_i_be_reentrant && initialized) {
     will_i_be_reentrant = 1;
     add_allocation_hook((size_t)result, allocated);
     will_i_be_reentrant = 0;

memapi/src/memorytracking.rs

@@ -64,7 +64,7 @@ impl AllocationTracker {
         let alloc = Allocation { callstack, size };
         self.current_allocations.insert(address, alloc);
         self.current_allocated_bytes += size;
-        if self.current_allocated_bytes > self.peak_allocated_bytes {
+        if self.current_allocated_bytes > self.peak_allocated_bytes + 10000 {
             self.peak_allocated_bytes = self.current_allocated_bytes;
             self.peak_allocations = self.current_allocations.clone();
         }