core.h 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360
  1. #ifndef _LSUP_CORE_H
  2. #define _LSUP_CORE_H
  3. #ifndef NOCOLOR
  4. #define LOG_USE_COLOR
  5. #endif
  6. #include <ctype.h>
  7. #include <dirent.h>
  8. #include <inttypes.h>
  9. #include <limits.h>
  10. #include <stdbool.h>
  11. #include <stddef.h>
  12. #include <stdio.h>
  13. #include <stdlib.h>
  14. #include <string.h>
  15. #include <sys/stat.h>
  16. #include <uuid/uuid.h>
  17. #include "log.h"
  18. #include "xxhash.h"
  19. #define LSUP_VERSION "1.0a3"
  20. // Logging and debugging.
  21. #ifdef DEBUG
  22. // GDB breakpoints.
  23. #include <signal.h>
  24. #define BREAKPOINT raise (SIGINT)
  25. #else
  26. #define BREAKPOINT
  27. #endif
  28. #define LIKELY(x) __builtin_expect(!!(x), true)
  29. #define UNLIKELY(x) __builtin_expect(!!(x), false)
  30. #define LSUP_NS "urn:lsup:" /// Default LS namespace.
  31. // TODO Cross-platform ramdisk path.
  32. #define TMPDIR "/tmp"
  33. #define KLEN sizeof(LSUP_Key)
  34. #define DBL_KLEN sizeof(LSUP_DoubleKey)
  35. #define TRP_KLEN sizeof(LSUP_TripleKey)
  36. #define QUAD_KLEN sizeof(LSUP_QuadKey)
  37. # define UUIDSTR_SIZE 37
  38. /** @brief "NULL" triple, a value that is never user-provided.
  39. */
  40. #define NULL_TRP {NULL_KEY, NULL_KEY, NULL_KEY}
  41. /* * * RETURN CODES * * */
  42. /**
  43. * 0 is success, positive integers (>88800) are warnings, and negative integers
  44. * (>-88900) are errors.
  45. */
  46. typedef int LSUP_rc;
  47. /** @brief Generic success return code.
  48. */
  49. #define LSUP_OK 0
  50. /** @brief No action taken.
  51. *
  52. * An attempt to create or update a resource was made, but the resource already
  53. * existed substantially in the same form, so no action took place. The caller
  54. * is expected to find the resource as though the action actually took place.
  55. * If this is returned from the iteration of multiple updates, it means that
  56. * none of the iterations produced a change in state.
  57. */
  58. #define LSUP_NOACTION 88801
  59. /** @brief No result yielded.
  60. *
  61. * A read operation returned no results. If an iterator is expected to be
  62. * created, it may be created empty.
  63. */
  64. #define LSUP_NORESULT 88802
  65. /** @brief Loop end.
  66. *
  67. * End of a loop was reached. This can be used in a while() or for()
  68. * loop as a terminating condition.
  69. */
  70. #define LSUP_END 88803
  71. /** @brief Conflict warning.
  72. * An attempt to create or update a resource was made, but the resource existed
  73. * with a different form or value. The caller should find the value of the
  74. * existing resource to be different than the one that was attempted to store.
  75. * If this is returned from the iteration of multiple updates, it means that
  76. * other resources in the loop may have changed state and the operation as a
  77. * whole completed successfully.
  78. */
  79. #define LSUP_CONFLICT 88804
  80. /*
  81. * NOTE When adding new warning codes, use a value larger than the last one
  82. * in the list. Also change LSUP_MAX_WARNING.
  83. */
  84. #define LSUP_MIN_WARNING LSUP_NOACTION
  85. #define LSUP_MAX_WARNING LSUP_CONFLICT
  86. /** @brief Generic error return code.
  87. */
  88. #define LSUP_ERROR -88899
  89. /** @brief TODO
  90. */
  91. #define LSUP_PARSE_ERR -88898
  92. /** @brief TODO
  93. */
  94. #define LSUP_VALUE_ERR -88897
  95. /** @brief TODO
  96. */
  97. #define LSUP_TXN_ERR -88896
  98. /** @brief TODO
  99. */
  100. #define LSUP_DB_ERR -88895
  101. /** @brief TODO
  102. */
  103. #define LSUP_NOT_IMPL_ERR -88894
  104. /** @brief TODO
  105. */
  106. #define LSUP_IO_ERR -88893
  107. /** @brief TODO
  108. */
  109. #define LSUP_MEM_ERR -88892
  110. /** @brief Conflict error.
  111. *
  112. * A critical resource conflict happened and no resources were updated. If this
  113. * is returned from the iteration of multiple updates, it means that the
  114. * operation has been interrupted and any state change within the loop prior to
  115. * the error has been rolled back.
  116. */
  117. #define LSUP_CONFLICT_ERR -88891
  118. /** @brief TODO
  119. */
  120. #define LSUP_ENV_ERR -88890
  121. /*
  122. * NOTE When adding new error codes, use a value larger than the last one
  123. * in the list. Also change LSUP_MAX_ERROR.
  124. */
  125. #define LSUP_MIN_ERROR LSUP_ERROR
  126. #define LSUP_MAX_ERROR LSUP_ENV_ERR
  127. // Hashing stuff.
  128. #ifndef LSUP_HASH_SEED
  129. /** @brief Seed used for all hashing. Compile-time configurable.
  130. */
  131. #define LSUP_HASH_SEED 0
  132. #endif
  133. #define LSUP_HASH32(buf, size, seed) XXH32 (buf, size, seed)
  134. #define LSUP_HASH64(buf, size, seed) XXH3_64bits_withSeed (buf, size, seed)
  135. #define LSUP_HASH128(buf, size, seed) XXH3_128bits_withSeed (buf, size, seed)
  136. // TODO Add 32-bit switch.
  137. #define LSUP_HASH(buf, size, seed) LSUP_HASH64 (buf, size, seed)
  138. extern char *warning_msg[], *error_msg[];
  139. extern char *LSUP_root_path;
  140. typedef XXH32_hash_t LSUP_Hash32;
  141. typedef XXH64_hash_t LSUP_Hash64;
  142. typedef XXH128_hash_t LSUP_Hash128;
  143. // TODO Add 32-bit switch.
  144. typedef LSUP_Hash64 LSUP_Hash;
  145. typedef enum {
  146. LSUP_BOOL_UNION,
  147. LSUP_BOOL_SUBTRACTION,
  148. LSUP_BOOL_INTERSECTION,
  149. LSUP_BOOL_XOR,
  150. } LSUP_bool_op;
  151. typedef size_t LSUP_Key;
  152. typedef LSUP_Key LSUP_DoubleKey[2];
  153. typedef LSUP_Key LSUP_TripleKey[3];
  154. typedef LSUP_Key LSUP_QuadKey[4];
  155. typedef char uuid_str_t[UUIDSTR_SIZE];
  156. /** @brief Make recursive directories.
  157. *
  158. * Modified from
  159. * https://gist.github.com/JonathonReinhart/8c0d90191c38af2dcadb102c4e202950
  160. */
  161. LSUP_rc
  162. mkdir_p (const char *path, mode_t mode);
  163. /** @brief Remove a directory recursively, as in Unix "rm -r".
  164. *
  165. * @param path[in] Path of directory to remove.
  166. */
  167. LSUP_rc
  168. rm_r (const char *path);
  169. /** @brief Return an error message for a return code.
  170. */
  171. const char *
  172. LSUP_strerror (LSUP_rc rc);
  173. /** @brief Encode a code point using UTF-8.
  174. *
  175. * https://gist.github.com/MightyPork/52eda3e5677b4b03524e40c9f0ab1da5
  176. *
  177. * @author Ondřej Hruška <ondra@ondrovo.com>
  178. *
  179. * @license MIT
  180. *
  181. * @param out - output buffer (min 5 characters), will be 0-terminated
  182. * @param utf - code point 0-0x10FFFF
  183. * @return number of bytes on success, 0 on failure (also produces U+FFFD,
  184. * which uses 3 bytes)
  185. */
  186. inline int utf8_encode (const uint32_t utf, unsigned char *out)
  187. {
  188. if (utf <= 0x7F) {
  189. // Plain ASCII
  190. out[0] = (char) utf;
  191. out[1] = 0;
  192. return 1;
  193. }
  194. else if (utf <= 0x07FF) {
  195. // 2-byte unicode
  196. out[0] = (char) (((utf >> 6) & 0x1F) | 0xC0);
  197. out[1] = (char) (((utf >> 0) & 0x3F) | 0x80);
  198. out[2] = 0;
  199. return 2;
  200. }
  201. else if (utf <= 0xFFFF) {
  202. // 3-byte unicode
  203. out[0] = (char) (((utf >> 12) & 0x0F) | 0xE0);
  204. out[1] = (char) (((utf >> 6) & 0x3F) | 0x80);
  205. out[2] = (char) (((utf >> 0) & 0x3F) | 0x80);
  206. out[3] = 0;
  207. return 3;
  208. }
  209. else if (utf <= 0x10FFFF) {
  210. // 4-byte unicode
  211. out[0] = (char) (((utf >> 18) & 0x07) | 0xF0);
  212. out[1] = (char) (((utf >> 12) & 0x3F) | 0x80);
  213. out[2] = (char) (((utf >> 6) & 0x3F) | 0x80);
  214. out[3] = (char) (((utf >> 0) & 0x3F) | 0x80);
  215. out[4] = 0;
  216. return 4;
  217. }
  218. else {
  219. // error - use replacement character
  220. out[0] = (char) 0xEF;
  221. out[1] = (char) 0xBF;
  222. out[2] = (char) 0xBD;
  223. out[3] = 0;
  224. return 0;
  225. }
  226. }
  227. /** @brief Log an error or warning for return codes that are not LSUP_OK.
  228. *
  229. * Note that, if used outside of the other macros below, care must be taken
  230. * to pass it an actual return code rather than an expression, otherwise the
  231. * expression will be evaluated multiple times.
  232. */
  233. #define LOG_RC(rc) do { \
  234. if ((rc) < 0) log_error (LSUP_strerror (rc)); \
  235. else if ((rc) > 0) log_debug (LSUP_strerror (rc)); \
  236. } while (0);
  237. /// Error handling via goto.
  238. #define CHECK(exp, marker) do { \
  239. LSUP_rc _rc = (exp); \
  240. LOG_RC(_rc); \
  241. if (UNLIKELY (_rc != LSUP_OK)) goto marker; \
  242. } while (0);
  243. /// Jump if rc is negative (skip warnings).
  244. #define PCHECK(exp, marker) do { \
  245. LSUP_rc _rc = (exp); \
  246. LOG_RC(_rc); \
  247. if (UNLIKELY (_rc < LSUP_OK)) goto marker; \
  248. } while (0);
  249. /// Return rc if it is of LSUP_rc type and nonzero.
  250. #define RCCK(exp) do { \
  251. LSUP_rc _rc = (exp); \
  252. LOG_RC(_rc); \
  253. if (UNLIKELY (_rc != LSUP_OK)) return _rc; \
  254. } while (0);
  255. /// Return rc if it is of LSUP_rc type and negative (=error)
  256. #define PRCCK(exp) do { \
  257. LSUP_rc _rc = (exp); \
  258. LOG_RC(_rc); \
  259. if (UNLIKELY (_rc < LSUP_OK)) return _rc; \
  260. } while (0);
  261. /// Return NULL if RC is nonzero.
  262. #define RCNL(exp) do { \
  263. LSUP_rc _rc = (exp); \
  264. LOG_RC(_rc); \
  265. if (UNLIKELY (_rc != LSUP_OK)) return NULL; \
  266. } while (0);
  267. /// Return NULL if RC is negative (=error)
  268. #define PRCNL(exp) do { \
  269. LSUP_rc _rc = (exp); \
  270. LOG_RC(_rc); \
  271. if (UNLIKELY (_rc < LSUP_OK)) return NULL; \
  272. } while (0);
  273. /// Allocate one pointer with malloc and return rc if it fails.
  274. #define MALLOC_GUARD(var, rc) do { \
  275. (var) = malloc (sizeof *(var)); \
  276. if (UNLIKELY (var == NULL)) return (rc); \
  277. } while (0);
  278. /// Allocate one pointer with calloc and return rc if it fails.
  279. #define CALLOC_GUARD(var, rc) do { \
  280. (var) = calloc (1, sizeof *(var)); \
  281. if (UNLIKELY (var == NULL)) return (rc); \
  282. } while (0);
  283. /*
  284. #define MALLOC_GUARD_ME(var) MALLOC_GUARD((var), LSUP_MEM_ERR) \
  285. #define CALLOC_GUARD_ME(var) CALLOC_GUARD((var), LSUP_MEM_ERR) \
  286. #define MALLOC_GUARD_NL(var) MALLOC_GUARD((var), NULL) \
  287. #define CALLOC_GUARD_NL(var) CALLOC_GUARD((var), NULL) \
  288. */
  289. #endif /* _LSUP_CORE_H */