mem: add malloc function to abort for failure

The __malloc_hook deprecated and malloc function is weak function.
Then we can add malloc function to abort when failed to allocate.

Signed-off-by: Tokunori Ikegami <[email protected]>

Author: ikegami-t

util/mem.c

@@ -3,6 +3,7 @@
 #include <unistd.h>
 #include <malloc.h>
 #include <string.h>
+#include <dlfcn.h>
 #include <sys/mman.h>
 
 #include "mem.h"
@@ -107,3 +108,20 @@ void nvme_free_huge(struct nvme_mem_huge *mh)
 	mh->len = 0;
 	mh->p = NULL;
 }
+
+void *malloc(size_t size)
+{
+	static void *(*malloc_sym)(size_t size);
+	void *result = NULL;
+
+	if (!malloc_sym)
+		malloc_sym = dlsym(RTLD_NEXT, "malloc");
+
+	if (malloc_sym)
+		result = malloc_sym(size);
+
+	if (!result)
+		abort();
+
+	return result;
+}