futex.h 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  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. */
  55. static inline int
  56. futex_lock_pi(u_int32_t *uaddr, struct timespec *timeout, int opflags)
  57. {
  58. return futex(uaddr, FUTEX_LOCK_PI, 0, timeout, NULL, 0, opflags);
  59. }
  60. /**
  61. * futex_unlock_pi() - release uaddr as a PI mutex, waking the top waiter
  62. */
  63. static inline int
  64. futex_unlock_pi(u_int32_t *uaddr, int opflags)
  65. {
  66. return futex(uaddr, FUTEX_UNLOCK_PI, 0, NULL, NULL, 0, opflags);
  67. }
  68. /**
  69. * futex_cmp_requeue() - requeue tasks from uaddr to uaddr2
  70. * @nr_wake: wake up to this many tasks
  71. * @nr_requeue: requeue up to this many tasks
  72. */
  73. static inline int
  74. futex_cmp_requeue(u_int32_t *uaddr, u_int32_t val, u_int32_t *uaddr2, int nr_wake,
  75. int nr_requeue, int opflags)
  76. {
  77. return futex(uaddr, FUTEX_CMP_REQUEUE, nr_wake, nr_requeue, uaddr2,
  78. val, opflags);
  79. }
  80. #ifndef HAVE_PTHREAD_ATTR_SETAFFINITY_NP
  81. #include <pthread.h>
  82. static inline int pthread_attr_setaffinity_np(pthread_attr_t *attr,
  83. size_t cpusetsize,
  84. cpu_set_t *cpuset)
  85. {
  86. attr = attr;
  87. cpusetsize = cpusetsize;
  88. cpuset = cpuset;
  89. return 0;
  90. }
  91. #endif
  92. #endif /* _FUTEX_H */