PreciseLeakSanitizer: Add instrumentation functions and fix something…

… wrong

RuntimeLibrary/plsan.cpp

@@ -17,13 +17,17 @@ __attribute__((constructor)) void __plsan_init() { /* TODO: */
 /* finialization routines called after main()*/
 __attribute__((destructor)) void __plsan_fini() { delete plsan; }
 
+void __plsan_align(long size) {}
+
 void __plsan_alloc(void *addr, size_t size) {
   /* TODO: initialize references */
 }
 
-void __plsan_store(void **addr, void *value) { /* TODO: update references */
+void __plsan_store(void **lhs, void *rhs) { /* TODO: update references */
 }
 
+void __plsan_enter_func() {}
+
 void __plsan_exit_func() {
   /*
    *  TODO: check if
@@ -41,6 +45,12 @@ Plsan::Plsan() {
   handler = new PlsanHandler();
 }
 
+long Plsan::align_size(long size) { return 0; }
+
+void Plsan::init_refcnt(void *addr, size_t size) {}
+
+void Plsan::reference_count(void **lhs, void *rhs) {}
+
 void Plsan::enter_func() {
   // enter_func
 }

RuntimeLibrary/plsan.h

@@ -10,8 +10,11 @@ namespace __plsan {
 class Plsan {
 public:
   Plsan();
-  // long align_size(long size);
 
+  // Instrumentation function
+  long align_size(long size);
+  void init_refcnt(void *addr, size_t size);
+  void reference_count(void **lhs, void *rhs);
   void enter_func();
   void exit_func();
 

RuntimeLibrary/plsan_shadow.h

@@ -5,7 +5,8 @@
 
 namespace __plsan {
 
-#define MMAP_SIZE ((1L << 48) / 16)
+#define MIN_DYN_ALLOC_SIZE 16
+#define MMAP_SIZE ((1L << 48) / MIN_DYN_ALLOC_SIZE)
 
 class PlsanShadow {
 public:

RuntimeLibrary/plsan_storage.cpp

@@ -1,11 +1,18 @@
 #include "plsan_storage.h"
 
+namespace __plsan {
+
+PlsanStorage::PlsanStorage() {}
+
 void PlsanStorage::push_function() {}
 
 void PlsanStorage::pop_function() {}
 
-void PlsanStorage::add_mem_addr() {}
+// Not shadow memmory address, but origin address is needed.
+void PlsanStorage::add_mem_addr(void *addr) {}
 
 LocalDynAllocStorage PlsanStorage::get_function_stack() {
   return function_stack;
 }
+
+} // namespace __plsan

RuntimeLibrary/plsan_storage.h

@@ -3,17 +3,22 @@
 
 #include <stack>
 
+namespace __plsan {
+
 using LocalDynAllocStorage = std::stack<std::stack<void *>>;
 
 class PlsanStorage {
 public:
+  PlsanStorage();
   void push_function();
   void pop_function();
-  void add_mem_addr();
+  void add_mem_addr(void *addr);
   LocalDynAllocStorage get_function_stack();
 
 private:
   LocalDynAllocStorage function_stack;
 };
 
+} // namespace __plsan
+
 #endif