ua_util.h 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. #ifndef UA_UTIL_H_
  2. #define UA_UTIL_H_
  3. #include "ua_config.h"
  4. #include <assert.h>
  5. #define UA_assert(ignore) assert(ignore)
  6. /*********************/
  7. /* Memory Management */
  8. /*********************/
  9. /* Replace the macros with functions for custom allocators if necessary */
  10. #include <stdlib.h> // malloc, free
  11. #ifdef _WIN32
  12. # include <malloc.h>
  13. #endif
  14. #ifndef UA_free
  15. # define UA_free(ptr) free(ptr)
  16. #endif
  17. #ifndef UA_malloc
  18. # define UA_malloc(size) malloc(size)
  19. #endif
  20. #ifndef UA_calloc
  21. # define UA_calloc(num, size) calloc(num, size)
  22. #endif
  23. #ifndef UA_realloc
  24. # define UA_realloc(ptr, size) realloc(ptr, size)
  25. #endif
  26. #ifndef NO_ALLOCA
  27. # ifdef __GNUC__
  28. # define UA_alloca(size) __builtin_alloca (size)
  29. # elif defined(_WIN32)
  30. # define UA_alloca(SIZE) _alloca(SIZE)
  31. # else
  32. # include <alloca.h>
  33. # define UA_alloca(SIZE) alloca(SIZE)
  34. # endif
  35. #endif
  36. #define container_of(ptr, type, member) \
  37. (type *)((uintptr_t)ptr - offsetof(type,member))
  38. /************************/
  39. /* Thread Local Storage */
  40. /************************/
  41. #ifdef UA_ENABLE_MULTITHREADING
  42. # ifdef __GNUC__
  43. # define UA_THREAD_LOCAL __thread
  44. # elif defined(_MSC_VER)
  45. # define UA_THREAD_LOCAL __declspec(thread)
  46. # else
  47. # error No thread local storage keyword defined for this compiler
  48. # endif
  49. #else
  50. # define UA_THREAD_LOCAL
  51. #endif
  52. /********************/
  53. /* System Libraries */
  54. /********************/
  55. #ifdef _WIN32
  56. # include <winsock2.h> //needed for amalgation
  57. # include <windows.h>
  58. # undef SLIST_ENTRY
  59. #endif
  60. #include <time.h>
  61. #if defined(_WIN32) && !defined(__MINGW32__)
  62. int gettimeofday(struct timeval *tp, struct timezone *tzp);
  63. #else
  64. # include <sys/time.h>
  65. #endif
  66. #if defined(__APPLE__) || defined(__MACH__)
  67. #include <mach/clock.h>
  68. #include <mach/mach.h>
  69. #endif
  70. /*************************/
  71. /* External Dependencies */
  72. /*************************/
  73. #include "queue.h"
  74. #ifdef UA_ENABLE_MULTITHREADING
  75. # define _LGPL_SOURCE
  76. # include <urcu.h>
  77. # include <urcu/wfcqueue.h>
  78. # include <urcu/uatomic.h>
  79. # include <urcu/rculfhash.h>
  80. # include <urcu/lfstack.h>
  81. # ifdef NDEBUG
  82. # define UA_RCU_LOCK() rcu_read_lock()
  83. # define UA_RCU_UNLOCK() rcu_read_unlock()
  84. # define UA_ASSERT_RCU_LOCKED()
  85. # define UA_ASSERT_RCU_UNLOCKED()
  86. # else
  87. extern UA_THREAD_LOCAL bool rcu_locked;
  88. # define UA_ASSERT_RCU_LOCKED() assert(rcu_locked)
  89. # define UA_ASSERT_RCU_UNLOCKED() assert(!rcu_locked)
  90. # define UA_RCU_LOCK() do { \
  91. UA_ASSERT_RCU_UNLOCKED(); \
  92. rcu_locked = true; \
  93. rcu_read_lock(); } while(0)
  94. # define UA_RCU_UNLOCK() do { \
  95. UA_ASSERT_RCU_LOCKED(); \
  96. rcu_locked = false; \
  97. rcu_read_lock(); } while(0)
  98. # endif
  99. #else
  100. # define UA_RCU_LOCK()
  101. # define UA_RCU_UNLOCK()
  102. # define UA_ASSERT_RCU_LOCKED()
  103. # define UA_ASSERT_RCU_UNLOCKED()
  104. #endif
  105. #endif /* UA_UTIL_H_ */