Changelog.md

@@ -1,12 +1,5 @@
 # Changelog
 
-## Version 2.6
-
-Use a more robust mutex (by using `std::recursive_mutex` instead of
-`pthread_mutex`).
-
-Minor refactorings.
-
 ## Version 2.5
 
 Added support for realloc().

src/another_memcheck.cc

@@ -3,7 +3,6 @@
 #include <iostream>
 #include <cstring>
 #include <cstdio>
-#include <mutex>
 
 // Local includes.
 #include "another_memcheck.h"
@@ -76,7 +75,7 @@ namespace SC_AM_Internal {
     return false;
   }
 
-  Stats::Stats() : malloced_list_head(nullptr), malloced_list_tail(nullptr), deferred_node(nullptr), recursive_mutex(nullptr) {
+  Stats::Stats() : malloced_list_head(nullptr), malloced_list_tail(nullptr), deferred_node(nullptr), pthread_mutex{.__align=0} {
   }
 
   void Stats::initialize() {
@@ -89,9 +88,9 @@ namespace SC_AM_Internal {
     malloced_list_head->data = nullptr;
     malloced_list_tail->data = nullptr;
 
-    recursive_mutex = real_malloc(sizeof(std::recursive_mutex));
-    void *unused = new(recursive_mutex) std::recursive_mutex{};
-    (void)unused;
+    deferred_node = nullptr;
+
+    pthread_mutex_init(&pthread_mutex, nullptr);
 
     on_exit([] ([[maybe_unused]] int status, void *ptr) {
       Stats *stats = reinterpret_cast<Stats*>(ptr);
@@ -101,14 +100,15 @@ namespace SC_AM_Internal {
   }
 
   void Stats::cleanup() {
-    using std_recursive_mutex = std::recursive_mutex;
-    ((std::recursive_mutex*)recursive_mutex)->~std_recursive_mutex();
-    real_free(recursive_mutex);
+    pthread_mutex_destroy(&pthread_mutex);
     // TODO maybe cleanup list, but it is the end of the program.
   }
 
   void *Stats::do_malloc(std::size_t size) {
-    ((std::recursive_mutex*)recursive_mutex)->lock();
+    if(int ret = pthread_mutex_lock(&pthread_mutex); ret == EINVAL) {
+      std::clog << "ERROR: pthread mutex not properly initialized!\n";
+      return nullptr;
+    }
 
     void *address = real_malloc(size);
     if (address != nullptr) {
@@ -122,12 +122,15 @@ namespace SC_AM_Internal {
       ListNode::add_to_list(malloced_list_tail, data);
     }
 
-    ((std::recursive_mutex*)recursive_mutex)->unlock();
+    pthread_mutex_unlock(&pthread_mutex);
     return address;
   }
 
   void *Stats::do_calloc(std::size_t n, std::size_t size) {
-    ((std::recursive_mutex*)recursive_mutex)->lock();
+    if(int ret = pthread_mutex_lock(&pthread_mutex); ret == EINVAL) {
+      std::clog << "ERROR: pthread mutex not properly initialized!\n";
+      return nullptr;
+    }
 
     void *address = real_calloc(n, size);
     if (address != nullptr) {
@@ -141,12 +144,15 @@ namespace SC_AM_Internal {
       ListNode::add_to_list(malloced_list_tail, data);
     }
 
-    ((std::recursive_mutex*)recursive_mutex)->unlock();
+    pthread_mutex_unlock(&pthread_mutex);
     return address;
   }
 
   void *Stats::do_realloc(void *ptr, std::size_t size) {
-    ((std::recursive_mutex*)recursive_mutex)->lock();
+    if(int ret = pthread_mutex_lock(&pthread_mutex); ret == EINVAL) {
+      std::clog << "ERROR: pthread mutex not properly initialized!\n";
+      return nullptr;
+    }
 
     if (ptr) {
       if (!ListNode::remove_from_list(malloced_list_head, ptr, this)) {
@@ -177,13 +183,16 @@ namespace SC_AM_Internal {
       }
     }
 
-    ((std::recursive_mutex*)recursive_mutex)->unlock();
+    pthread_mutex_unlock(&pthread_mutex);
     return address;
   }
 
   bool Stats::do_free(void *ptr) {
     bool result = false;
-    ((std::recursive_mutex*)recursive_mutex)->lock();
+    if(int ret = pthread_mutex_lock(&pthread_mutex); ret == EINVAL) {
+      std::clog << "ERROR: pthread mutex not properly initialized!\n";
+      return result;
+    }
 
     if (ptr) {
       if (!ListNode::remove_from_list(malloced_list_head, ptr)) {
@@ -194,7 +203,7 @@ namespace SC_AM_Internal {
       }
     }
 
-    ((std::recursive_mutex*)recursive_mutex)->unlock();
+    pthread_mutex_unlock(&pthread_mutex);
     return result;
   }
 

src/another_memcheck.h

@@ -1,6 +1,10 @@
 #ifndef SEODISPARATE_COM_ANOTHER_MEMCHECK_H
 #define SEODISPARATE_COM_ANOTHER_MEMCHECK_H
 
+extern "C" {
+#include <pthread.h>
+}
+
 namespace SC_AM_Internal {
   // Forward declaration.
   struct Stats;
@@ -52,7 +56,7 @@ namespace SC_AM_Internal {
     ListNode *malloced_list_head;
     ListNode *malloced_list_tail;
     ListNode *deferred_node;
-    void *recursive_mutex;
+    pthread_mutex_t pthread_mutex;
 
     void initialize();
     void cleanup();