2 files changed, 60 insertions, 0 deletions
diff --git a/src/util/memory.c b/src/util/memory.c
new file mode 100644
index 0000000..9724c81
--- /dev/null
+++ b/src/util/memory.c
@@ -0,0 +1,28 @@
+/*
+ * ecgen, tool for generating Elliptic curve domain parameters
+ * Copyright (C) 2017 J08nY
+ */
+#include "memory.h"
+#include <pari/pari.h>
+
+void *alloc(void *(*fun)(size_t), size_t size) {
+	void *result = fun(size);
+	if (!result) {
+		perror("Couldn't alloc.");
+		exit(1);
+	}
+	return result;
+}
+
+void *try_malloc(size_t size) { return alloc(pari_malloc, size); }
+
+void *try_calloc(size_t size) { return alloc(pari_calloc, size); }
+
+void *try_realloc(void *ptr, size_t size) {
+	void *result = pari_realloc(ptr, size);
+	if (!result) {
+		perror("Couldn't alloc.");
+		exit(1);
+	}
+	return result;
+}
+\ No newline at end of file
diff --git a/src/util/memory.h b/src/util/memory.h
new file mode 100644
index 0000000..fef237e
--- /dev/null
+++ b/src/util/memory.h
@@ -0,0 +1,32 @@
+/*
+ * ecgen, tool for generating Elliptic curve domain parameters
+ * Copyright (C) 2017 J08nY
+ */
+#include <stddef.h>
+
+#ifndef ECGEN_MEMORY_H
+#define ECGEN_MEMORY_H
+
+/**
+ * @brief
+ * @param size
+ * @return
+ */
+void *try_malloc(size_t size);
+
+/**
+ * @brief
+ * @param size
+ * @return
+ */
+void *try_calloc(size_t size);
+
+/**
+ * @brief
+ * @param ptr
+ * @param size
+ * @return
+ */
+void *try_realloc(void *ptr, size_t size);
+
+#endif  // ECGEN_MEMORY_H