复制文件

mem/mem.c

@@ -1,11 +1,6 @@
-#undef malloc
-#undef free
-#undef realloc
-#undef memcpy
-
 void *safe_malloc(size_t size){
     void *tmp;
-    tmp = calloc(size,1);
+    tmp = calloc(size, 1);
     if(tmp != NULL){
         return tmp;
     }
@@ -47,4 +42,4 @@ void *safe_memcpy(void *p1, const void *p2, size_t size){
         puts("memcpy error");
         exit(1);
     }
-}
+}

mem/mem.h

@@ -0,0 +1,4 @@
+#define malloc(size) safe_malloc(size)
+#define free(p) p=safe_free(p)
+#define realloc(p,size) safe_realloc(p,size)
+#define memcpy(p1,p2,size) safe_memcpy(p1,p2,size)