mem.h 638 B

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