malloc: add myMalloc() and myFree()

[model-checker.git] / malloc.c

diff --git a/malloc.c b/malloc.c
new file mode 100644 (file)
index 0000000..196be43
--- /dev/null
+++ b/malloc.c
@@ -0,0 +1,38 @@
+#include "common.h"
+
+/* for RTLD_NEXT */
+#ifndef __USE_GNU
+#define __USE_GNU
+#endif
+
+#include <dlfcn.h>
+
+static void * (*real_malloc)(size_t) = NULL;
+static void (*real_free)(void *ptr) = NULL;
+
+static void __my_alloc_init(void)
+{
+
+       real_malloc = (void *(*)(size_t))dlsym(RTLD_NEXT, "malloc");
+       real_free = (void (*)(void *))dlsym(RTLD_NEXT, "free");
+       if (real_malloc == NULL || real_free == NULL) {
+               fprintf(stderr, "Error in `dlsym`: %s\n", dlerror());
+               return;
+       }
+}
+
+void *myMalloc(size_t size)
+{
+       if (real_malloc == NULL)
+               __my_alloc_init();
+
+       return real_malloc(size);
+}
+
+void myFree(void *ptr)
+{
+       if (real_free == NULL)
+               __my_alloc_init();
+
+       real_free(ptr);
+}