2
0

mem.h 526 B

1234567891011121314151617181920212223242526
  1. /*
  2. * 文件名: mem.h
  3. * 目标: 内存管理工具的头文件
  4. */
  5. #ifndef AFUN_MEM_H
  6. #define AFUN_MEM_H
  7. #include <stdlib.h>
  8. #define free(p) ((((p)!=NULL) ? (free(p), NULL) : NULL), (p)=NULL) // free不是可选的宏
  9. #if BUILD_MEM
  10. /* 取代calloc函数 */
  11. static void *safeCalloc(size_t n, size_t size);
  12. static void *safeCalloc(size_t n, size_t size) {
  13. void *re = calloc(n, size);
  14. if (re == NULL)
  15. exit(1);
  16. return re;
  17. }
  18. #define calloc(n, size) (safeCalloc(n, size))
  19. #endif
  20. #endif // AFUN_MEM_H