1 files changed, 73 insertions, 0 deletions
diff --git a/comp/lucas-standen-NEA/code/global/util.c b/comp/lucas-standen-NEA/code/global/util.c
new file mode 100644
index 0000000..736619b
--- /dev/null
+++ b/comp/lucas-standen-NEA/code/global/util.c
@@ -0,0 +1,73 @@
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <stdbool.h>
+#include <errno.h>
+#include <error.h>
+
+// functions for user
+void Die(); // brings down the program
+void *CheckedMalloc(long size); // malloc checked with autofree
+void *CheckedRealloc(void *out, long size); // realloc checked with autofree
+int CheckedFree(void *ptr); // frees a pointer if it is in the array MEMptrs
+void CheckedFreeALL(); // frees all pointers in the array MEMptrs
+
+#define MAXPTRS 30000 // maximum allocs done by user
+
+void *MEMptrs[MAXPTRS] = { NULL };
+size_t currentPtr = 0;
+
+void Die(){
+	perror("zpy parser");
+	exit(errno);
+}
+
+void *CheckedMalloc(long size){
+	void *out = malloc(size);
+	if (out == NULL)
+		Die();
+	MEMptrs[currentPtr] = out;	
+	currentPtr++;
+	if (currentPtr > MAXPTRS){
+		printf("used %d ptrs!\n", MAXPTRS);
+		Die();
+	}
+	return out;
+}
+
+void *CheckedRealloc(void *orig, long size){
+	void *out = realloc(orig, size);
+	if (out == NULL)
+		Die();
+	MEMptrs[currentPtr] = out;	
+	currentPtr++;
+	if (currentPtr > MAXPTRS){
+		printf("used %d ptrs!\n", MAXPTRS);
+		Die();
+	}
+
+	for (int i = 0; i < MAXPTRS; i++)
+		if (MEMptrs[i] == orig && MEMptrs[i] != NULL) MEMptrs[i] = NULL;
+
+	return out;
+}
+
+int CheckedFree(void *ptr){
+	if (ptr == NULL) return 1;
+	for (int i = 0; i < MAXPTRS; i++){
+		if (MEMptrs[i] == ptr){
+			free(MEMptrs[i]);
+			MEMptrs[i] = NULL;
+			return 0;
+		}
+	}
+	return 1;
+}
+
+void CheckedFreeALL(){
+	for (int i = 0; i < MAXPTRS; i++){
+		if (MEMptrs[i] != NULL){
+			free(MEMptrs[i]);
+		}
+	}
+}