c : add bit_array, bit_compose, malloc_hook

1 files changed, 37 insertions, 0 deletions

diff --git a/c/malloc_hook.c b/c/malloc_hook.c
new file mode 100644
index 0000000..aa22379
--- /dev/null
+++ b/c/malloc_hook.c
@@ -0,0 +1,37 @@
+#include <stdlib.h>
+#include <stdio.h>
+
+#define __USE_GNU
+#include <dlfcn.h>
+
+typedef void* (*real_malloc_t) (size_t);
+
+static real_malloc_t real_malloc = NULL;
+
+static void __mtrace_init(void)
+{
+    real_malloc = (real_malloc_t) dlsym(RTLD_NEXT, "malloc");
+    if (NULL == real_malloc) {
+        fprintf(stderr, "Error in `dlsym`: %s\n", dlerror());
+        return;
+    }
+}
+
+void *malloc(size_t size)
+{
+    if(real_malloc == NULL)
+        __mtrace_init();
+
+    void *p = NULL;
+    fprintf(stderr, "malloc(%zu) = ", size);
+    p = real_malloc(size);
+    fprintf(stderr, "%p\n", p);
+    return p;
+}
+
+
+int main(int argc, char **argv, char **env)
+{
+    void * nothing = malloc(666);
+    free(nothing);
+}