compiler.h 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. #ifndef _TOOLS_LINUX_COMPILER_H_
  2. #define _TOOLS_LINUX_COMPILER_H_
  3. /* Optimization barrier */
  4. /* The "volatile" is due to gcc bugs */
  5. #define barrier() __asm__ __volatile__("": : :"memory")
  6. #ifndef __always_inline
  7. # define __always_inline inline __attribute__((always_inline))
  8. #endif
  9. #define __user
  10. #ifndef __attribute_const__
  11. # define __attribute_const__
  12. #endif
  13. #ifndef __maybe_unused
  14. # define __maybe_unused __attribute__((unused))
  15. #endif
  16. #ifndef __packed
  17. # define __packed __attribute__((__packed__))
  18. #endif
  19. #ifndef __force
  20. # define __force
  21. #endif
  22. #ifndef __weak
  23. # define __weak __attribute__((weak))
  24. #endif
  25. #ifndef likely
  26. # define likely(x) __builtin_expect(!!(x), 1)
  27. #endif
  28. #ifndef unlikely
  29. # define unlikely(x) __builtin_expect(!!(x), 0)
  30. #endif
  31. #define ACCESS_ONCE(x) (*(volatile typeof(x) *)&(x))
  32. #include <linux/types.h>
  33. /*
  34. * Following functions are taken from kernel sources and
  35. * break aliasing rules in their original form.
  36. *
  37. * While kernel is compiled with -fno-strict-aliasing,
  38. * perf uses -Wstrict-aliasing=3 which makes build fail
  39. * under gcc 4.4.
  40. *
  41. * Using extra __may_alias__ type to allow aliasing
  42. * in this case.
  43. */
  44. typedef __u8 __attribute__((__may_alias__)) __u8_alias_t;
  45. typedef __u16 __attribute__((__may_alias__)) __u16_alias_t;
  46. typedef __u32 __attribute__((__may_alias__)) __u32_alias_t;
  47. typedef __u64 __attribute__((__may_alias__)) __u64_alias_t;
  48. static __always_inline void __read_once_size(const volatile void *p, void *res, int size)
  49. {
  50. switch (size) {
  51. case 1: *(__u8_alias_t *) res = *(volatile __u8_alias_t *) p; break;
  52. case 2: *(__u16_alias_t *) res = *(volatile __u16_alias_t *) p; break;
  53. case 4: *(__u32_alias_t *) res = *(volatile __u32_alias_t *) p; break;
  54. case 8: *(__u64_alias_t *) res = *(volatile __u64_alias_t *) p; break;
  55. default:
  56. barrier();
  57. __builtin_memcpy((void *)res, (const void *)p, size);
  58. barrier();
  59. }
  60. }
  61. static __always_inline void __write_once_size(volatile void *p, void *res, int size)
  62. {
  63. switch (size) {
  64. case 1: *(volatile __u8_alias_t *) p = *(__u8_alias_t *) res; break;
  65. case 2: *(volatile __u16_alias_t *) p = *(__u16_alias_t *) res; break;
  66. case 4: *(volatile __u32_alias_t *) p = *(__u32_alias_t *) res; break;
  67. case 8: *(volatile __u64_alias_t *) p = *(__u64_alias_t *) res; break;
  68. default:
  69. barrier();
  70. __builtin_memcpy((void *)p, (const void *)res, size);
  71. barrier();
  72. }
  73. }
  74. /*
  75. * Prevent the compiler from merging or refetching reads or writes. The
  76. * compiler is also forbidden from reordering successive instances of
  77. * READ_ONCE, WRITE_ONCE and ACCESS_ONCE (see below), but only when the
  78. * compiler is aware of some particular ordering. One way to make the
  79. * compiler aware of ordering is to put the two invocations of READ_ONCE,
  80. * WRITE_ONCE or ACCESS_ONCE() in different C statements.
  81. *
  82. * In contrast to ACCESS_ONCE these two macros will also work on aggregate
  83. * data types like structs or unions. If the size of the accessed data
  84. * type exceeds the word size of the machine (e.g., 32 bits or 64 bits)
  85. * READ_ONCE() and WRITE_ONCE() will fall back to memcpy and print a
  86. * compile-time warning.
  87. *
  88. * Their two major use cases are: (1) Mediating communication between
  89. * process-level code and irq/NMI handlers, all running on the same CPU,
  90. * and (2) Ensuring that the compiler does not fold, spindle, or otherwise
  91. * mutilate accesses that either do not require ordering or that interact
  92. * with an explicit memory barrier or atomic instruction that provides the
  93. * required ordering.
  94. */
  95. #define READ_ONCE(x) \
  96. ({ union { typeof(x) __val; char __c[1]; } __u; __read_once_size(&(x), __u.__c, sizeof(x)); __u.__val; })
  97. #define WRITE_ONCE(x, val) \
  98. ({ union { typeof(x) __val; char __c[1]; } __u = { .__val = (val) }; __write_once_size(&(x), __u.__c, sizeof(x)); __u.__val; })
  99. #ifndef __fallthrough
  100. # if defined(__GNUC__) && __GNUC__ >= 7
  101. # define __fallthrough __attribute__ ((fallthrough))
  102. # else
  103. # define __fallthrough
  104. # endif
  105. #endif
  106. #endif /* _TOOLS_LINUX_COMPILER_H */