rt: Track backtraces of all allocations with RUSTRT_TRACK_ALLOCATIONS=3

1 files changed, 17 insertions, 0 deletions

diff --git a/src/rt/memory_region.cpp b/src/rt/memory_region.cpp
index e4130e80f5a..770691e88ea 100644
--- a/src/rt/memory_region.cpp
+++ b/src/rt/memory_region.cpp
@@ -1,6 +1,10 @@
 #include "rust_internal.h"
 #include "memory_region.h"
 
+#if RUSTRT_TRACK_ALLOCATIONS >= 3
+#include <execinfo.h>
+#endif
+
 #if RUSTRT_TRACK_ALLOCATIONS >= 1
 // For some platforms, 16 byte alignment is required.
 #  define PTR_SIZE 16
@@ -148,6 +152,13 @@ memory_region::~memory_region() {
                        header->tag,
                        (uintptr_t) get_data(header));
                 ++leak_count;
+
+#               if RUSTRT_TRACK_ALLOCATIONS >= 3
+                if (_detailed_leaks) {
+                    backtrace_symbols_fd(header->bt + 1,
+                                         header->btframes - 1, 2);
+                }
+#               endif
             }
         }
         assert(leak_count == _live_allocations);
@@ -199,6 +210,12 @@ memory_region::claim_alloc(void *mem) {
     if (_synchronized) { _lock.unlock(); }
 #   endif
 
+#   if RUSTRT_TRACK_ALLOCATIONS >= 3
+    if (_detailed_leaks) {
+        alloc->btframes = ::backtrace(alloc->bt, 32);
+    }
+#   endif
+
     add_alloc();
 }