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,21 @@ 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,21 @@ 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);
+
+	CHECK_NULL_PTR(result);
+
+	return result;
+}
+#endif /* HAVE_WEAK_MALLOC */

util/mem.h

@@ -5,6 +5,12 @@
 #include <stddef.h>
 #include <stdbool.h>
 
+#define CHECK_NULL_PTR(ptr) \
+	do { \
+		if (!ptr) \
+			abort(); \
+	} while (false)
+
 void *nvme_alloc(size_t len);
 void *nvme_realloc(void *p, size_t len);