mx_util: add mx_malloc_forever

mariux · Oct 16, 2015 · 03703e1 · 03703e1
1 parent f83e67a
commit 03703e1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/mx_util.c b/mx_util.c
@@ -553,6 +553,16 @@ int mx_strtoi64(char *str, int64_t *to)
     return 0;
 }
 
+void *mx_malloc_forever(size_t size)
+{
+    void *ret;
+    do {
+        ret=malloc(size);
+        assert(ret || (!ret && errno == ENOMEM));
+   } while (!ret);
+   return ret ;
+}
+
 char *mx_strdup_forever(char *str)
 {
     char *dup;

diff --git a/mx_util.h b/mx_util.h
@@ -146,6 +146,7 @@ int mx_strtoi16(char *str, int16_t *to);
 int mx_strtoi32(char *str, int32_t *to);
 int mx_strtoi64(char *str, int64_t *to);
 
+void *mx_malloc_forever(size_t size);
 char *mx_strdup_forever(char *str);
 int mx_vasprintf_forever(char **strp, const char *fmt, va_list ap);
 int mx_asprintf_forever(char **strp, const char *fmt, ...)  __attribute__ ((format(printf, 2, 3)));