util.h

   1 #ifndef ZZGO_UTIL_H
   2 #define ZZGO_UTIL_H
   3
   4 #include <stdlib.h>
   5
   6 /* Misc. definitions. */
   7
   8 #define likely(x) __builtin_expect(!!(x), 1)
   9 #define unlikely(x) __builtin_expect((x), 0)
  10
  11 static inline void *
  12 checked_malloc(size_t size, char *filename, unsigned int line, const char *func)
  13 {
  14         void *p = malloc(size);
  15         if (!p) {
  16                 fprintf(stderr, "%s:%u: %s: OUT OF MEMORY malloc(%zu)\n",
  17                         filename, line, func, size);
  18                 exit(1);
  19         }
  20         return p;
  21 }
  22
  23 static inline void *
  24 checked_calloc(size_t nmemb, size_t size, char *filename, unsigned int line, const char *func)
  25 {
  26         void *p = calloc(nmemb, size);
  27         if (!p) {
  28                 fprintf(stderr, "%s:%u: %s: OUT OF MEMORY calloc(%zu, %zu)\n",
  29                         filename, line, func, nmemb, size);
  30                 exit(1);
  31         }
  32         return p;
  33 }
  34
  35 #define malloc2(size)        checked_malloc((size), __FILE__, __LINE__, __func__)
  36 #define calloc2(nmemb, size) checked_calloc((nmemb), (size), __FILE__, __LINE__, __func__)
  37
  38 #endif