Add malloc wrapper

Abort on allocation failure.  This is mostly done as a formality, as
Linux tends to over-commit memory anyway.  In the event of most
failures, we won't have a reasonable recovery either.

Signed-off-by: Malfurious <m@lfurio.us>

2 files changed, 11 insertions, 0 deletions

diff --git a/helpers.c b/helpers.c
index c93b774..2cd119d 100644
--- a/helpers.c
+++ b/helpers.c
@@ -1,8 +1,16 @@
+#include <assert.h>
 #include <locale.h>
 #include <stdarg.h>
+#include <stdlib.h>
 
 #include "helpers.h"
 
+void *xmalloc(size_t size) {
+    void *ptr = malloc((size ? size : 1));
+    assert(ptr != NULL);
+    return ptr;
+}
+
 void cursinit(void) {
     setlocale(LC_ALL, "");
     initscr();
diff --git a/helpers.h b/helpers.h
index 7b85319..e0cd8c5 100644
--- a/helpers.h
+++ b/helpers.h
@@ -2,6 +2,9 @@
 
 #include <ncurses.h>
 #include <panel.h>
+#include <stddef.h>
+
+extern void *xmalloc(size_t size);
 
 extern void cursinit(void);
 extern void cursupdate(void);