futex.h 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. /*
  2. * Glibc independent futex library for testing kernel functionality.
  3. * Shamelessly stolen from Darren Hart <dvhltc@us.ibm.com>
  4. * http://git.kernel.org/cgit/linux/kernel/git/dvhart/futextest.git/
  5. */
  6. #ifndef _FUTEX_H
  7. #define _FUTEX_H
  8. #include <unistd.h>
  9. #include <sys/syscall.h>
  10. #include <sys/types.h>
  11. #include <linux/futex.h>
  12. /**
  13. * futex() - SYS_futex syscall wrapper
  14. * @uaddr: address of first futex
  15. * @op: futex op code
  16. * @val: typically expected value of uaddr, but varies by op
  17. * @timeout: typically an absolute struct timespec (except where noted
  18. * otherwise). Overloaded by some ops
  19. * @uaddr2: address of second futex for some ops\
  20. * @val3: varies by op
  21. * @opflags: flags to be bitwise OR'd with op, such as FUTEX_PRIVATE_FLAG
  22. *
  23. * futex() is used by all the following futex op wrappers. It can also be
  24. * used for misuse and abuse testing. Generally, the specific op wrappers
  25. * should be used instead. It is a macro instead of an static inline function as
  26. * some of the types over overloaded (timeout is used for nr_requeue for
  27. * example).
  28. *
  29. * These argument descriptions are the defaults for all
  30. * like-named arguments in the following wrappers except where noted below.
  31. */
  32. #define futex(uaddr, op, val, timeout, uaddr2, val3, opflags) \
  33. syscall(SYS_futex, uaddr, op | opflags, val, timeout, uaddr2, val3)
  34. /**
  35. * futex_wait() - block on uaddr with optional timeout
  36. * @timeout: relative timeout
  37. */
  38. static inline int
  39. futex_wait(u_int32_t *uaddr, u_int32_t val, struct timespec *timeout, int opflags)
  40. {
  41. return futex(uaddr, FUTEX_WAIT, val, timeout, NULL, 0, opflags);
  42. }
  43. /**
  44. * futex_wake() - wake one or more tasks blocked on uaddr
  45. * @nr_wake: wake up to this many tasks
  46. */
  47. static inline int
  48. futex_wake(u_int32_t *uaddr, int nr_wake, int opflags)
  49. {
  50. return futex(uaddr, FUTEX_WAKE, nr_wake, NULL, NULL, 0, opflags);
  51. }
  52. /**
  53. * futex_lock_pi() - block on uaddr as a PI mutex
  54. * @detect: whether (1) or not (0) to perform deadlock detection
  55. */
  56. static inline int
  57. futex_lock_pi(u_int32_t *uaddr, struct timespec *timeout, int detect,
  58. int opflags)
  59. {
  60. return futex(uaddr, FUTEX_LOCK_PI, detect, timeout, NULL, 0, opflags);
  61. }
  62. /**
  63. * futex_unlock_pi() - release uaddr as a PI mutex, waking the top waiter
  64. */
  65. static inline int
  66. futex_unlock_pi(u_int32_t *uaddr, int opflags)
  67. {
  68. return futex(uaddr, FUTEX_UNLOCK_PI, 0, NULL, NULL, 0, opflags);
  69. }
  70. /**
  71. * futex_cmp_requeue() - requeue tasks from uaddr to uaddr2
  72. * @nr_wake: wake up to this many tasks
  73. * @nr_requeue: requeue up to this many tasks
  74. */
  75. static inline int
  76. futex_cmp_requeue(u_int32_t *uaddr, u_int32_t val, u_int32_t *uaddr2, int nr_wake,
  77. int nr_requeue, int opflags)
  78. {
  79. return futex(uaddr, FUTEX_CMP_REQUEUE, nr_wake, nr_requeue, uaddr2,
  80. val, opflags);
  81. }
  82. #ifndef HAVE_PTHREAD_ATTR_SETAFFINITY_NP
  83. #include <pthread.h>
  84. static inline int pthread_attr_setaffinity_np(pthread_attr_t *attr,
  85. size_t cpusetsize,
  86. cpu_set_t *cpuset)
  87. {
  88. attr = attr;
  89. cpusetsize = cpusetsize;
  90. cpuset = cpuset;
  91. return 0;
  92. }
  93. #endif
  94. #endif /* _FUTEX_H */