utils.h

   1 /*
   2  * utils - misc libubox utility functions
   3  *
   4  * Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
   5  *
   6  * Permission to use, copy, modify, and/or distribute this software for any
   7  * purpose with or without fee is hereby granted, provided that the above
   8  * copyright notice and this permission notice appear in all copies.
   9  *
  10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  17  */
  18
  19 #ifndef __LIBUBOX_UTILS_H
  20 #define __LIBUBOX_UTILS_H
  21
  22 #include <sys/types.h>
  23 #include <sys/time.h>
  24 #include <stdint.h>
  25 #include <stdbool.h>
  26 #include <unistd.h>
  27 #include <time.h>
  28
  29 /*
  30  * calloc_a(size_t len, [void **addr, size_t len,...], NULL)
  31  *
  32  * allocate a block of memory big enough to hold multiple aligned objects.
  33  * the pointer to the full object (starting with the first chunk) is returned,
  34  * all other pointers are stored in the locations behind extra addr arguments.
  35  * the last argument needs to be a NULL pointer
  36  */
  37
  38 #define calloc_a(len, ...) __calloc_a(len, ##__VA_ARGS__, NULL)
  39
  40 void *__calloc_a(size_t len, ...);
  41
  42 #ifndef ARRAY_SIZE
  43 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
  44 #endif
  45
  46 #define __BUILD_BUG_ON(condition) ((void)sizeof(char[1 - 2*!!(condition)]))
  47
  48 #ifdef __OPTIMIZE__
  49 extern int __BUILD_BUG_ON_CONDITION_FAILED;
  50 #define BUILD_BUG_ON(condition)                                 \
  51         do {                                                    \
  52                 __BUILD_BUG_ON(condition);                      \
  53                 if (condition)                                  \
  54                         __BUILD_BUG_ON_CONDITION_FAILED = 1;    \
  55         } while(0)
  56 #else
  57 #define BUILD_BUG_ON __BUILD_BUG_ON
  58 #endif
  59
  60 #if defined(__APPLE__) && !defined(CLOCK_MONOTONIC)
  61 #define LIBUBOX_COMPAT_CLOCK_GETTIME
  62
  63 #include <mach/clock_types.h>
  64 #define CLOCK_REALTIME  CALENDAR_CLOCK
  65 #define CLOCK_MONOTONIC SYSTEM_CLOCK
  66
  67 int clock_gettime(int type, struct timespec *tv);
  68
  69 #endif
  70
  71 #ifdef __GNUC__
  72 #define _GNUC_MIN_VER(maj, min) (((__GNUC__ << 8) + __GNUC_MINOR__) >= (((maj) << 8) + (min)))
  73 #else
  74 #define _GNUC_MIN_VER(maj, min) 0
  75 #endif
  76
  77 #if defined(__linux__) || defined(__CYGWIN__)
  78 #include <byteswap.h>
  79 #include <endian.h>
  80
  81 #elif defined(__APPLE__)
  82 #include <machine/endian.h>
  83 #include <machine/byte_order.h>
  84 #define bswap_32(x) OSSwapInt32(x)
  85 #define bswap_64(x) OSSwapInt64(x)
  86 #elif defined(__FreeBSD__)
  87 #include <sys/endian.h>
  88 #define bswap_32(x) bswap32(x)
  89 #define bswap_64(x) bswap64(x)
  90 #else
  91 #include <machine/endian.h>
  92 #define bswap_32(x) swap32(x)
  93 #define bswap_64(x) swap64(x)
  94 #endif
  95
  96 #ifndef __BYTE_ORDER
  97 #define __BYTE_ORDER BYTE_ORDER
  98 #endif
  99 #ifndef __BIG_ENDIAN
 100 #define __BIG_ENDIAN BIG_ENDIAN
 101 #endif
 102 #ifndef __LITTLE_ENDIAN
 103 #define __LITTLE_ENDIAN LITTLE_ENDIAN
 104 #endif
 105
 106 static inline uint16_t __u_bswap16(uint16_t val)
 107 {
 108         return ((val >> 8) & 0xffu) | ((val & 0xffu) << 8);
 109 }
 110
 111 #if _GNUC_MIN_VER(4, 2)
 112 #define __u_bswap32(x) __builtin_bswap32(x)
 113 #define __u_bswap64(x) __builtin_bswap64(x)
 114 #else
 115 #define __u_bswap32(x) bswap_32(x)
 116 #define __u_bswap64(x) bswap_64(x)
 117 #endif
 118
 119 #if __BYTE_ORDER == __LITTLE_ENDIAN
 120
 121 #define cpu_to_be64(x) __u_bswap64(x)
 122 #define cpu_to_be32(x) __u_bswap32(x)
 123 #define cpu_to_be16(x) __u_bswap16((uint16_t) (x))
 124
 125 #define be64_to_cpu(x) __u_bswap64(x)
 126 #define be32_to_cpu(x) __u_bswap32(x)
 127 #define be16_to_cpu(x) __u_bswap16((uint16_t) (x))
 128
 129 #define cpu_to_le64(x) (x)
 130 #define cpu_to_le32(x) (x)
 131 #define cpu_to_le16(x) (x)
 132
 133 #define le64_to_cpu(x) (x)
 134 #define le32_to_cpu(x) (x)
 135 #define le16_to_cpu(x) (x)
 136
 137 #else /* __BYTE_ORDER == __LITTLE_ENDIAN */
 138
 139 #define cpu_to_le64(x) __u_bswap64(x)
 140 #define cpu_to_le32(x) __u_bswap32(x)
 141 #define cpu_to_le16(x) __u_bswap16((uint16_t) (x))
 142
 143 #define le64_to_cpu(x) __u_bswap64(x)
 144 #define le32_to_cpu(x) __u_bswap32(x)
 145 #define le16_to_cpu(x) __u_bswap16((uint16_t) (x))
 146
 147 #define cpu_to_be64(x) (x)
 148 #define cpu_to_be32(x) (x)
 149 #define cpu_to_be16(x) (x)
 150
 151 #define be64_to_cpu(x) (x)
 152 #define be32_to_cpu(x) (x)
 153 #define be16_to_cpu(x) (x)
 154
 155 #endif
 156
 157 #ifndef __packed
 158 #define __packed __attribute__((packed))
 159 #endif
 160
 161 #ifndef __constructor
 162 #define __constructor __attribute__((constructor))
 163 #endif
 164
 165 #ifndef __destructor
 166 #define __destructor __attribute__((destructor))
 167 #endif
 168
 169 #ifndef __hidden
 170 #define __hidden __attribute__((visibility("hidden")))
 171 #endif
 172
 173 int b64_encode(const void *src, size_t src_len,
 174                void *dest, size_t dest_len);
 175
 176 int b64_decode(const void *src, void *dest, size_t dest_len);
 177
 178 #define B64_ENCODE_LEN(_len)    ((((_len) + 2) / 3) * 4 + 1)
 179 #define B64_DECODE_LEN(_len)    (((_len) / 4) * 3 + 1)
 180
 181 static inline unsigned int cbuf_order(unsigned int x)
 182 {
 183         return 32 - __builtin_clz(x - 1);
 184 }
 185
 186 static inline unsigned long cbuf_size(int order)
 187 {
 188         unsigned long page_size = sysconf(_SC_PAGESIZE);
 189         unsigned long ret = 1ULL << order;
 190
 191         if (ret < page_size)
 192                 ret = page_size;
 193
 194         return ret;
 195 }
 196
 197 void *cbuf_alloc(unsigned int order);
 198 void cbuf_free(void *ptr, unsigned int order);
 199
 200 #endif