Add debug logging with time, refactor allocation

1 files changed, 28 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