rwsem.h 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. /*
  2. * The owner field of the rw_semaphore structure will be set to
  3. * RWSEM_READ_OWNED when a reader grabs the lock. A writer will clear
  4. * the owner field when it unlocks. A reader, on the other hand, will
  5. * not touch the owner field when it unlocks.
  6. *
  7. * In essence, the owner field now has the following 3 states:
  8. * 1) 0
  9. * - lock is free or the owner hasn't set the field yet
  10. * 2) RWSEM_READER_OWNED
  11. * - lock is currently or previously owned by readers (lock is free
  12. * or not set by owner yet)
  13. * 3) Other non-zero value
  14. * - a writer owns the lock
  15. */
  16. #define RWSEM_READER_OWNED ((struct task_struct *)1UL)
  17. #ifdef CONFIG_RWSEM_SPIN_ON_OWNER
  18. /*
  19. * All writes to owner are protected by WRITE_ONCE() to make sure that
  20. * store tearing can't happen as optimistic spinners may read and use
  21. * the owner value concurrently without lock. Read from owner, however,
  22. * may not need READ_ONCE() as long as the pointer value is only used
  23. * for comparison and isn't being dereferenced.
  24. */
  25. static inline void rwsem_set_owner(struct rw_semaphore *sem)
  26. {
  27. WRITE_ONCE(sem->owner, current);
  28. }
  29. static inline void rwsem_clear_owner(struct rw_semaphore *sem)
  30. {
  31. WRITE_ONCE(sem->owner, NULL);
  32. }
  33. static inline void rwsem_set_reader_owned(struct rw_semaphore *sem)
  34. {
  35. /*
  36. * We check the owner value first to make sure that we will only
  37. * do a write to the rwsem cacheline when it is really necessary
  38. * to minimize cacheline contention.
  39. */
  40. if (sem->owner != RWSEM_READER_OWNED)
  41. WRITE_ONCE(sem->owner, RWSEM_READER_OWNED);
  42. }
  43. static inline bool rwsem_owner_is_writer(struct task_struct *owner)
  44. {
  45. return owner && owner != RWSEM_READER_OWNED;
  46. }
  47. static inline bool rwsem_owner_is_reader(struct task_struct *owner)
  48. {
  49. return owner == RWSEM_READER_OWNED;
  50. }
  51. #else
  52. static inline void rwsem_set_owner(struct rw_semaphore *sem)
  53. {
  54. }
  55. static inline void rwsem_clear_owner(struct rw_semaphore *sem)
  56. {
  57. }
  58. static inline void rwsem_set_reader_owned(struct rw_semaphore *sem)
  59. {
  60. }
  61. #endif