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

meson.build

@@ -190,6 +190,23 @@ conf.set(
     ),
     description: 'Does struct tm have a tm_gmtoff field?'
 )
+conf.set(
+    'HAVE_WEAK_MALLOC',
+    cc.links(
+        '''
+           #include <malloc.h>
+           void *malloc(size_t size)
+           {
+                return NULL;
+           }
+           int main(void) {
+                malloc(1);
+           }
+        ''',
+        name: 'weak_malloc'
+    ),
+    description: 'Does struct tm have a tm_gmtoff field?'
+)
 
 if cc.has_function_attribute('fallthrough')
   conf.set('fallthrough', '__attribute__((__fallthrough__))')

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,22 @@ void nvme_free_huge(struct nvme_mem_huge *mh)
 	mh->len = 0;
 	mh->p = NULL;
 }
+
+#ifdef HAVE_WEAK_MALLOC
+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;
+}
+#endif /* HAVE_WEAK_MALLOC */