src/util.h

   1 /*
   2  * Copyright (c) 2007, Novell Inc.
   3  *
   4  * This program is licensed under the BSD license, read LICENSE.BSD
   5  * for further information
   6  */
   7
   8 /*
   9  * util.h
  10  *
  11  */
  12
  13 #ifndef SATSOLVER_UTIL_H
  14 #define SATSOLVER_UTIL_H
  15
  16 #include <stddef.h>
  17 #include <string.h>
  18
  19 /**
  20  * malloc
  21  * exits with error message on error
  22  */
  23 extern void *sat_malloc(size_t);
  24 extern void *sat_malloc2(size_t, size_t);
  25 extern void *sat_calloc(size_t, size_t);
  26 extern void *sat_realloc(void *, size_t);
  27 extern void *sat_realloc2(void *, size_t, size_t);
  28 extern void *sat_free(void *);
  29 extern void sat_oom(size_t, size_t);
  30 extern unsigned int sat_timems(unsigned int subtract);
  31 extern void sat_sort(void *base, size_t nmemb, size_t size, int (*compar)(const void *, const void *, void *), void *compard);
  32 extern char *sat_dupjoin(const char *str1, const char *str2, const char *str3);
  33
  34 static inline void *sat_extend(void *buf, size_t len, size_t nmemb, size_t size, size_t block)
  35 {
  36   if (nmemb == 1)
  37     {
  38       if ((len & block) == 0)
  39         buf = sat_realloc2(buf, len + (1 + block), size);
  40     }
  41   else
  42     {
  43       if (((len - 1) | block) != ((len + nmemb - 1) | block))
  44         buf = sat_realloc2(buf, (len + (nmemb + block)) & ~block, size);
  45     }
  46   return buf;
  47 }
  48
  49 /**
  50  * extend an array by reallocation and zero's the new section
  51  * buf old pointer
  52  * len current size
  53  * nmbemb number of elements to add
  54  * size size of each element
  55  * block block size used to allocate the elements
  56  */
  57 static inline void *sat_zextend(void *buf, size_t len, size_t nmemb, size_t size, size_t block)
  58 {
  59   buf = sat_extend(buf, len, nmemb, size, block);
  60   memset((char *)buf + len * size, 0, nmemb * size);
  61   return buf;
  62 }
  63
  64 static inline void *sat_extend_resize(void *buf, size_t len, size_t size, size_t block)
  65 {
  66   if (len)
  67     buf = sat_realloc2(buf, (len + block) & ~block, size);
  68   return buf;
  69 }
  70
  71 static inline void *sat_calloc_block(size_t len, size_t size, size_t block)
  72 {
  73   void *buf;
  74   if (!len)
  75     return 0;
  76   buf = sat_malloc2((len + block) & ~block, size);
  77   memset(buf, 0, ((len + block) & ~block) * size);
  78   return buf;
  79 }
  80 #endif /* SATSOLVER_UTIL_H */