Initial

1 files changed, 19 insertions, 0 deletions

diff --git a/memory.h b/memory.h
new file mode 100644
index 0000000..9175dea
--- /dev/null
+++ b/memory.h
@@ -0,0 +1,19 @@
+#pragma once
+
+#include <stdlib.h>
+
+#define malloc(num,type) \
+	((type*)check_after_allocation("malloc",num,sizeof(type),malloc((num)*sizeof(type))))
+#define calloc(num,type) \
+	((type*)check_after_allocation("calloc",num,sizeof(type),calloc((num),sizeof(type))))
+#define realloc(ptr,num,type) \
+	((type*)check_after_allocation("realloc",num,sizeof(type),realloc((ptr),(num)*sizeof(type))))
+#define strdup(str) \
+	((char*)check_after_allocation_str("strdup",strdup(str)))
+#define asprintf(...) \
+	(memory_asprintf_wrapper(__VA_ARGS__))
+
+void* check_after_allocation(const char *func,size_t num,size_t sz,void *ptr);
+void* check_after_allocation_str(const char *func,void *ptr);
+
+int memory_asprintf_wrapper(char **ret,const char *format,...) __attribute__((format (printf, 2, 3)));