Add xmalloc

2 files changed, 24 insertions, 0 deletions

diff --git a/mem.c b/mem.c
new file mode 100644
index 0000000..ff76b08
--- /dev/null
+++ b/mem.c
@@ -0,0 +1,16 @@
+#include "mem.h"
+
+#include <curses.h>
+#include <stddef.h>
+#include <stdlib.h>
+
+#include "err.h"
+
+void *xmalloc(size_t size) {
+    void *rtn = malloc(size);
+    if (!rtn) {
+        endwin();
+        die("Out of memory");
+    }
+    return rtn;
+}
diff --git a/mem.h b/mem.h
new file mode 100644
index 0000000..8c1c7e2
--- /dev/null
+++ b/mem.h
@@ -0,0 +1,8 @@
+#ifndef SS_MEM_H
+#define SS_MEM_H
+
+#include <stddef.h>
+
+void* xmalloc(size_t size);
+
+#endif /* SS_MEM_H */