PreciseLeakSanitizer: Add destructor

hygoni · Jan 19, 2024 · 2933467 · 2933467
1 parent 8c465c9
commit 2933467
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/RuntimeLibrary/plsan.cpp b/RuntimeLibrary/plsan.cpp
@@ -7,9 +7,16 @@
 
 #include <cstddef>
 
+__plsan::Plsan *plsan;
+
 /* Initialization routines called before main() */
 __attribute__((constructor)) void __plsan_init() { /* TODO: */
-  __plsan::Plsan *plsan;
+  plsan = new __plsan::Plsan();
+}
+
+/* finialization routines called after main()*/
+__attribute__((destructor)) void __plsan_fini() {
+    delete plsan;
 }
 
 void __plsan_alloc(void *addr, size_t size) {

diff --git a/RuntimeLibrary/plsan_shadow.cpp b/RuntimeLibrary/plsan_shadow.cpp
@@ -11,6 +11,10 @@ PlsanShadow::PlsanShadow() {
     throw "mmap failed\n";
 }
 
+PlsanShadow::~PlsanShadow() {
+  munmap(shadow_addr, MMAP_SIZE);
+}
+
 void PlsanShadow::alloc_shadow(void *addr, size_t size) {}
 
 void PlsanShadow::update_reference(void **lhs, void *rhs) {}

diff --git a/RuntimeLibrary/plsan_shadow.h b/RuntimeLibrary/plsan_shadow.h
@@ -10,6 +10,7 @@ namespace __plsan {
 class PlsanShadow {
 public:
   PlsanShadow();
+  ~PlsanShadow();
   void alloc_shadow(void *addr, size_t size);
   void update_reference(void **lhs, void *rhs);