Add some common utility functions

Signed-off-by: pacien <pacien.trangirard@pacien.net>

4 files changed, 67 insertions, 0 deletions

diff --git a/include/common/error.h b/include/common/error.h
new file mode 100644
index 0000000..522d646
--- /dev/null
+++ b/include/common/error.h
@@ -0,0 +1,24 @@
+#ifndef UPEM_MORPHING_ERROR
+#define UPEM_MORPHING_ERROR
+
+/**
+ * File: error.h
+ */
+
+/**
+ * Constants: Common errors
+ *
+ * OUT_OF_MEMORY_ERROR - when memory cannot be allocated for the program
+ */
+#define OUT_OF_MEMORY_ERROR "Out of memory"
+
+/**
+ * Function: rage quit
+ * Logs the supplied error message to stderr before exiting the program with an error status code.
+ *
+ * Parameters;
+ *   *msg - error message to log
+ */
+void rage_quit(const char *msg);
+
+#endif
diff --git a/include/common/mem.h b/include/common/mem.h
new file mode 100644
index 0000000..1c46778
--- /dev/null
+++ b/include/common/mem.h
@@ -0,0 +1,23 @@
+#ifndef UPEM_MORPHING_MEM
+#define UPEM_MORPHING_MEM
+
+/**
+ * File: mem.h
+ */
+
+#include <stddef.h>
+
+/**
+ * Function: malloc_or_die
+ * Allocates a memory chunk of the requested size.
+ * Interrupts the execution of the program in case of an error.
+ *
+ * Parameters:
+ *   size - size of the chunk to allocate
+ *
+ * Returns:
+ *   A non-NULL pointer to the allocated chunk
+ */
+void *malloc_or_die(size_t size);
+
+#endif
diff --git a/src/common/error.c b/src/common/error.c
new file mode 100644
index 0000000..f28a6cd
--- /dev/null
+++ b/src/common/error.c
@@ -0,0 +1,8 @@
+#include "common/error.h"
+#include <stdlib.h>
+#include <stdio.h>
+
+void rage_quit(const char *msg) {
+  fprintf(stderr, "FATAL ERROR: %s\n", msg);
+  exit(1);
+}
diff --git a/src/common/mem.c b/src/common/mem.c
new file mode 100644
index 0000000..855e010
--- /dev/null
+++ b/src/common/mem.c
@@ -0,0 +1,12 @@
+#include "common/mem.h"
+#include <stdlib.h>
+#include "common/error.h"
+
+void *malloc_or_die(size_t size) {
+  void *ptr = malloc(size);
+
+  if (ptr == NULL)
+    rage_quit(OUT_OF_MEMORY_ERROR);
+
+  return ptr;
+}