rwsem-xadd.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756
  1. // SPDX-License-Identifier: GPL-2.0
  2. /* rwsem.c: R/W semaphores: contention handling functions
  3. *
  4. * Written by David Howells (dhowells@redhat.com).
  5. * Derived from arch/i386/kernel/semaphore.c
  6. *
  7. * Writer lock-stealing by Alex Shi <alex.shi@intel.com>
  8. * and Michel Lespinasse <walken@google.com>
  9. *
  10. * Optimistic spinning by Tim Chen <tim.c.chen@intel.com>
  11. * and Davidlohr Bueso <davidlohr@hp.com>. Based on mutexes.
  12. */
  13. #include <linux/rwsem.h>
  14. #include <linux/init.h>
  15. #include <linux/export.h>
  16. #include <linux/sched/signal.h>
  17. #include <linux/sched/rt.h>
  18. #include <linux/sched/wake_q.h>
  19. #include <linux/sched/debug.h>
  20. #include <linux/osq_lock.h>
  21. #include "rwsem.h"
  22. #ifdef CONFIG_MTK_TASK_TURBO
  23. #include <mt-plat/turbo_common.h>
  24. #endif
  25. /*
  26. * Guide to the rw_semaphore's count field for common values.
  27. * (32-bit case illustrated, similar for 64-bit)
  28. *
  29. * 0x0000000X (1) X readers active or attempting lock, no writer waiting
  30. * X = #active_readers + #readers attempting to lock
  31. * (X*ACTIVE_BIAS)
  32. *
  33. * 0x00000000 rwsem is unlocked, and no one is waiting for the lock or
  34. * attempting to read lock or write lock.
  35. *
  36. * 0xffff000X (1) X readers active or attempting lock, with waiters for lock
  37. * X = #active readers + # readers attempting lock
  38. * (X*ACTIVE_BIAS + WAITING_BIAS)
  39. * (2) 1 writer attempting lock, no waiters for lock
  40. * X-1 = #active readers + #readers attempting lock
  41. * ((X-1)*ACTIVE_BIAS + ACTIVE_WRITE_BIAS)
  42. * (3) 1 writer active, no waiters for lock
  43. * X-1 = #active readers + #readers attempting lock
  44. * ((X-1)*ACTIVE_BIAS + ACTIVE_WRITE_BIAS)
  45. *
  46. * 0xffff0001 (1) 1 reader active or attempting lock, waiters for lock
  47. * (WAITING_BIAS + ACTIVE_BIAS)
  48. * (2) 1 writer active or attempting lock, no waiters for lock
  49. * (ACTIVE_WRITE_BIAS)
  50. *
  51. * 0xffff0000 (1) There are writers or readers queued but none active
  52. * or in the process of attempting lock.
  53. * (WAITING_BIAS)
  54. * Note: writer can attempt to steal lock for this count by adding
  55. * ACTIVE_WRITE_BIAS in cmpxchg and checking the old count
  56. *
  57. * 0xfffe0001 (1) 1 writer active, or attempting lock. Waiters on queue.
  58. * (ACTIVE_WRITE_BIAS + WAITING_BIAS)
  59. *
  60. * Note: Readers attempt to lock by adding ACTIVE_BIAS in down_read and checking
  61. * the count becomes more than 0 for successful lock acquisition,
  62. * i.e. the case where there are only readers or nobody has lock.
  63. * (1st and 2nd case above).
  64. *
  65. * Writers attempt to lock by adding ACTIVE_WRITE_BIAS in down_write and
  66. * checking the count becomes ACTIVE_WRITE_BIAS for successful lock
  67. * acquisition (i.e. nobody else has lock or attempts lock). If
  68. * unsuccessful, in rwsem_down_write_failed, we'll check to see if there
  69. * are only waiters but none active (5th case above), and attempt to
  70. * steal the lock.
  71. *
  72. */
  73. /*
  74. * Initialize an rwsem:
  75. */
  76. void __init_rwsem(struct rw_semaphore *sem, const char *name,
  77. struct lock_class_key *key)
  78. {
  79. #ifdef CONFIG_DEBUG_LOCK_ALLOC
  80. /*
  81. * Make sure we are not reinitializing a held semaphore:
  82. */
  83. debug_check_no_locks_freed((void *)sem, sizeof(*sem));
  84. lockdep_init_map(&sem->dep_map, name, key, 0);
  85. #endif
  86. atomic_long_set(&sem->count, RWSEM_UNLOCKED_VALUE);
  87. raw_spin_lock_init(&sem->wait_lock);
  88. INIT_LIST_HEAD(&sem->wait_list);
  89. #ifdef CONFIG_RWSEM_SPIN_ON_OWNER
  90. sem->owner = NULL;
  91. osq_lock_init(&sem->osq);
  92. #endif
  93. #ifdef CONFIG_MTK_TASK_TURBO
  94. sem->turbo_owner = NULL;
  95. #endif
  96. }
  97. EXPORT_SYMBOL(__init_rwsem);
  98. enum rwsem_waiter_type {
  99. RWSEM_WAITING_FOR_WRITE,
  100. RWSEM_WAITING_FOR_READ
  101. };
  102. struct rwsem_waiter {
  103. struct list_head list;
  104. struct task_struct *task;
  105. enum rwsem_waiter_type type;
  106. };
  107. enum rwsem_wake_type {
  108. RWSEM_WAKE_ANY, /* Wake whatever's at head of wait list */
  109. RWSEM_WAKE_READERS, /* Wake readers only */
  110. RWSEM_WAKE_READ_OWNED /* Waker thread holds the read lock */
  111. };
  112. /*
  113. * handle the lock release when processes blocked on it that can now run
  114. * - if we come here from up_xxxx(), then:
  115. * - the 'active part' of count (&0x0000ffff) reached 0 (but may have changed)
  116. * - the 'waiting part' of count (&0xffff0000) is -ve (and will still be so)
  117. * - there must be someone on the queue
  118. * - the wait_lock must be held by the caller
  119. * - tasks are marked for wakeup, the caller must later invoke wake_up_q()
  120. * to actually wakeup the blocked task(s) and drop the reference count,
  121. * preferably when the wait_lock is released
  122. * - woken process blocks are discarded from the list after having task zeroed
  123. * - writers are only marked woken if downgrading is false
  124. */
  125. static void __rwsem_mark_wake(struct rw_semaphore *sem,
  126. enum rwsem_wake_type wake_type,
  127. struct wake_q_head *wake_q)
  128. {
  129. struct rwsem_waiter *waiter, *tmp;
  130. long oldcount, woken = 0, adjustment = 0;
  131. struct list_head wlist;
  132. /*
  133. * Take a peek at the queue head waiter such that we can determine
  134. * the wakeup(s) to perform.
  135. */
  136. waiter = list_first_entry(&sem->wait_list, struct rwsem_waiter, list);
  137. if (waiter->type == RWSEM_WAITING_FOR_WRITE) {
  138. if (wake_type == RWSEM_WAKE_ANY) {
  139. /*
  140. * Mark writer at the front of the queue for wakeup.
  141. * Until the task is actually later awoken later by
  142. * the caller, other writers are able to steal it.
  143. * Readers, on the other hand, will block as they
  144. * will notice the queued writer.
  145. */
  146. wake_q_add(wake_q, waiter->task);
  147. }
  148. return;
  149. }
  150. /*
  151. * Writers might steal the lock before we grant it to the next reader.
  152. * We prefer to do the first reader grant before counting readers
  153. * so we can bail out early if a writer stole the lock.
  154. */
  155. if (wake_type != RWSEM_WAKE_READ_OWNED) {
  156. adjustment = RWSEM_ACTIVE_READ_BIAS;
  157. try_reader_grant:
  158. oldcount = atomic_long_fetch_add(adjustment, &sem->count);
  159. if (unlikely(oldcount < RWSEM_WAITING_BIAS)) {
  160. /*
  161. * If the count is still less than RWSEM_WAITING_BIAS
  162. * after removing the adjustment, it is assumed that
  163. * a writer has stolen the lock. We have to undo our
  164. * reader grant.
  165. */
  166. if (atomic_long_add_return(-adjustment, &sem->count) <
  167. RWSEM_WAITING_BIAS)
  168. return;
  169. /* Last active locker left. Retry waking readers. */
  170. goto try_reader_grant;
  171. }
  172. /*
  173. * It is not really necessary to set it to reader-owned here,
  174. * but it gives the spinners an early indication that the
  175. * readers now have the lock.
  176. */
  177. rwsem_set_reader_owned(sem);
  178. }
  179. /*
  180. * Grant an infinite number of read locks to the readers at the front
  181. * of the queue. We know that woken will be at least 1 as we accounted
  182. * for above. Note we increment the 'active part' of the count by the
  183. * number of readers before waking any processes up.
  184. *
  185. * We have to do wakeup in 2 passes to prevent the possibility that
  186. * the reader count may be decremented before it is incremented. It
  187. * is because the to-be-woken waiter may not have slept yet. So it
  188. * may see waiter->task got cleared, finish its critical section and
  189. * do an unlock before the reader count increment.
  190. *
  191. * 1) Collect the read-waiters in a separate list, count them and
  192. * fully increment the reader count in rwsem.
  193. * 2) For each waiters in the new list, clear waiter->task and
  194. * put them into wake_q to be woken up later.
  195. */
  196. list_for_each_entry(waiter, &sem->wait_list, list) {
  197. if (waiter->type == RWSEM_WAITING_FOR_WRITE)
  198. break;
  199. woken++;
  200. }
  201. list_cut_before(&wlist, &sem->wait_list, &waiter->list);
  202. adjustment = woken * RWSEM_ACTIVE_READ_BIAS - adjustment;
  203. if (list_empty(&sem->wait_list)) {
  204. /* hit end of list above */
  205. adjustment -= RWSEM_WAITING_BIAS;
  206. }
  207. if (adjustment)
  208. atomic_long_add(adjustment, &sem->count);
  209. /* 2nd pass */
  210. list_for_each_entry_safe(waiter, tmp, &wlist, list) {
  211. struct task_struct *tsk;
  212. tsk = waiter->task;
  213. get_task_struct(tsk);
  214. /*
  215. * Ensure calling get_task_struct() before setting the reader
  216. * waiter to nil such that rwsem_down_read_failed() cannot
  217. * race with do_exit() by always holding a reference count
  218. * to the task to wakeup.
  219. */
  220. smp_store_release(&waiter->task, NULL);
  221. /*
  222. * Ensure issuing the wakeup (either by us or someone else)
  223. * after setting the reader waiter to nil.
  224. */
  225. wake_q_add(wake_q, tsk);
  226. /* wake_q_add() already take the task ref */
  227. put_task_struct(tsk);
  228. }
  229. }
  230. /*
  231. * Wait for the read lock to be granted
  232. */
  233. static inline struct rw_semaphore __sched *
  234. __rwsem_down_read_failed_common(struct rw_semaphore *sem, int state)
  235. {
  236. long count, adjustment = -RWSEM_ACTIVE_READ_BIAS;
  237. struct rwsem_waiter waiter;
  238. DEFINE_WAKE_Q(wake_q);
  239. waiter.task = current;
  240. waiter.type = RWSEM_WAITING_FOR_READ;
  241. raw_spin_lock_irq(&sem->wait_lock);
  242. if (list_empty(&sem->wait_list))
  243. adjustment += RWSEM_WAITING_BIAS;
  244. #ifdef CONFIG_MTK_TASK_TURBO
  245. rwsem_list_add(waiter.task, &waiter.list, &sem->wait_list);
  246. #else
  247. list_add_tail(&waiter.list, &sem->wait_list);
  248. #endif
  249. /* we're now waiting on the lock, but no longer actively locking */
  250. count = atomic_long_add_return(adjustment, &sem->count);
  251. /*
  252. * If there are no active locks, wake the front queued process(es).
  253. *
  254. * If there are no writers and we are first in the queue,
  255. * wake our own waiter to join the existing active readers !
  256. */
  257. if (count == RWSEM_WAITING_BIAS ||
  258. (count > RWSEM_WAITING_BIAS &&
  259. adjustment != -RWSEM_ACTIVE_READ_BIAS))
  260. __rwsem_mark_wake(sem, RWSEM_WAKE_ANY, &wake_q);
  261. #ifdef CONFIG_MTK_TASK_TURBO
  262. if (waiter.task)
  263. rwsem_start_turbo_inherit(sem);
  264. #endif
  265. raw_spin_unlock_irq(&sem->wait_lock);
  266. wake_up_q(&wake_q);
  267. /* wait to be given the lock */
  268. while (true) {
  269. set_current_state(state);
  270. if (!waiter.task)
  271. break;
  272. if (signal_pending_state(state, current)) {
  273. raw_spin_lock_irq(&sem->wait_lock);
  274. if (waiter.task)
  275. goto out_nolock;
  276. raw_spin_unlock_irq(&sem->wait_lock);
  277. break;
  278. }
  279. schedule();
  280. }
  281. __set_current_state(TASK_RUNNING);
  282. return sem;
  283. out_nolock:
  284. list_del(&waiter.list);
  285. if (list_empty(&sem->wait_list))
  286. atomic_long_add(-RWSEM_WAITING_BIAS, &sem->count);
  287. raw_spin_unlock_irq(&sem->wait_lock);
  288. __set_current_state(TASK_RUNNING);
  289. return ERR_PTR(-EINTR);
  290. }
  291. __visible struct rw_semaphore * __sched
  292. rwsem_down_read_failed(struct rw_semaphore *sem)
  293. {
  294. return __rwsem_down_read_failed_common(sem, TASK_UNINTERRUPTIBLE);
  295. }
  296. EXPORT_SYMBOL(rwsem_down_read_failed);
  297. __visible struct rw_semaphore * __sched
  298. rwsem_down_read_failed_killable(struct rw_semaphore *sem)
  299. {
  300. return __rwsem_down_read_failed_common(sem, TASK_KILLABLE);
  301. }
  302. EXPORT_SYMBOL(rwsem_down_read_failed_killable);
  303. /*
  304. * This function must be called with the sem->wait_lock held to prevent
  305. * race conditions between checking the rwsem wait list and setting the
  306. * sem->count accordingly.
  307. */
  308. static inline bool rwsem_try_write_lock(long count, struct rw_semaphore *sem)
  309. {
  310. /*
  311. * Avoid trying to acquire write lock if count isn't RWSEM_WAITING_BIAS.
  312. */
  313. if (count != RWSEM_WAITING_BIAS)
  314. return false;
  315. /*
  316. * Acquire the lock by trying to set it to ACTIVE_WRITE_BIAS. If there
  317. * are other tasks on the wait list, we need to add on WAITING_BIAS.
  318. */
  319. count = list_is_singular(&sem->wait_list) ?
  320. RWSEM_ACTIVE_WRITE_BIAS :
  321. RWSEM_ACTIVE_WRITE_BIAS + RWSEM_WAITING_BIAS;
  322. if (atomic_long_cmpxchg_acquire(&sem->count, RWSEM_WAITING_BIAS, count)
  323. == RWSEM_WAITING_BIAS) {
  324. rwsem_set_owner(sem);
  325. return true;
  326. }
  327. return false;
  328. }
  329. #ifdef CONFIG_RWSEM_SPIN_ON_OWNER
  330. /*
  331. * Try to acquire write lock before the writer has been put on wait queue.
  332. */
  333. static inline bool rwsem_try_write_lock_unqueued(struct rw_semaphore *sem)
  334. {
  335. long old, count = atomic_long_read(&sem->count);
  336. while (true) {
  337. if (!(count == 0 || count == RWSEM_WAITING_BIAS))
  338. return false;
  339. old = atomic_long_cmpxchg_acquire(&sem->count, count,
  340. count + RWSEM_ACTIVE_WRITE_BIAS);
  341. if (old == count) {
  342. rwsem_set_owner(sem);
  343. return true;
  344. }
  345. count = old;
  346. }
  347. }
  348. static inline bool rwsem_can_spin_on_owner(struct rw_semaphore *sem)
  349. {
  350. struct task_struct *owner;
  351. bool ret = true;
  352. BUILD_BUG_ON(!rwsem_has_anonymous_owner(RWSEM_OWNER_UNKNOWN));
  353. if (need_resched())
  354. return false;
  355. rcu_read_lock();
  356. owner = READ_ONCE(sem->owner);
  357. if (!owner || !is_rwsem_owner_spinnable(owner)) {
  358. ret = !owner; /* !owner is spinnable */
  359. goto done;
  360. }
  361. /*
  362. * As lock holder preemption issue, we both skip spinning if task is not
  363. * on cpu or its cpu is preempted
  364. */
  365. ret = owner->on_cpu && !vcpu_is_preempted(task_cpu(owner));
  366. done:
  367. rcu_read_unlock();
  368. return ret;
  369. }
  370. /*
  371. * Return true only if we can still spin on the owner field of the rwsem.
  372. */
  373. static noinline bool rwsem_spin_on_owner(struct rw_semaphore *sem)
  374. {
  375. struct task_struct *owner = READ_ONCE(sem->owner);
  376. if (!owner || !is_rwsem_owner_spinnable(owner))
  377. return false;
  378. while (true) {
  379. bool on_cpu, same_owner;
  380. /*
  381. * Ensure sem->owner still matches owner. If that fails,
  382. * owner might point to free()d memory, if it still matches,
  383. * the rcu_read_lock() ensures the memory stays valid.
  384. */
  385. rcu_read_lock();
  386. same_owner = sem->owner == owner;
  387. if (same_owner)
  388. on_cpu = owner->on_cpu &&
  389. !vcpu_is_preempted(task_cpu(owner));
  390. rcu_read_unlock();
  391. if (!same_owner)
  392. break;
  393. /*
  394. * abort spinning when need_resched or owner is not running or
  395. * owner's cpu is preempted.
  396. */
  397. if (!on_cpu || need_resched())
  398. return false;
  399. cpu_relax();
  400. }
  401. /*
  402. * If there is a new owner or the owner is not set, we continue
  403. * spinning.
  404. */
  405. return is_rwsem_owner_spinnable(READ_ONCE(sem->owner));
  406. }
  407. static bool rwsem_optimistic_spin(struct rw_semaphore *sem)
  408. {
  409. bool taken = false;
  410. preempt_disable();
  411. /* sem->wait_lock should not be held when doing optimistic spinning */
  412. if (!rwsem_can_spin_on_owner(sem))
  413. goto done;
  414. if (!osq_lock(&sem->osq))
  415. goto done;
  416. /*
  417. * Optimistically spin on the owner field and attempt to acquire the
  418. * lock whenever the owner changes. Spinning will be stopped when:
  419. * 1) the owning writer isn't running; or
  420. * 2) readers own the lock as we can't determine if they are
  421. * actively running or not.
  422. */
  423. while (rwsem_spin_on_owner(sem)) {
  424. /*
  425. * Try to acquire the lock
  426. */
  427. if (rwsem_try_write_lock_unqueued(sem)) {
  428. taken = true;
  429. break;
  430. }
  431. /*
  432. * When there's no owner, we might have preempted between the
  433. * owner acquiring the lock and setting the owner field. If
  434. * we're an RT task that will live-lock because we won't let
  435. * the owner complete.
  436. */
  437. if (!sem->owner && (need_resched() || rt_task(current)))
  438. break;
  439. /*
  440. * The cpu_relax() call is a compiler barrier which forces
  441. * everything in this loop to be re-loaded. We don't need
  442. * memory barriers as we'll eventually observe the right
  443. * values at the cost of a few extra spins.
  444. */
  445. cpu_relax();
  446. }
  447. osq_unlock(&sem->osq);
  448. done:
  449. preempt_enable();
  450. return taken;
  451. }
  452. /*
  453. * Return true if the rwsem has active spinner
  454. */
  455. static inline bool rwsem_has_spinner(struct rw_semaphore *sem)
  456. {
  457. return osq_is_locked(&sem->osq);
  458. }
  459. #else
  460. static bool rwsem_optimistic_spin(struct rw_semaphore *sem)
  461. {
  462. return false;
  463. }
  464. static inline bool rwsem_has_spinner(struct rw_semaphore *sem)
  465. {
  466. return false;
  467. }
  468. #endif
  469. /*
  470. * Wait until we successfully acquire the write lock
  471. */
  472. static inline struct rw_semaphore *
  473. __rwsem_down_write_failed_common(struct rw_semaphore *sem, int state)
  474. {
  475. long count;
  476. bool waiting = true; /* any queued threads before us */
  477. struct rwsem_waiter waiter;
  478. struct rw_semaphore *ret = sem;
  479. DEFINE_WAKE_Q(wake_q);
  480. /* undo write bias from down_write operation, stop active locking */
  481. count = atomic_long_sub_return(RWSEM_ACTIVE_WRITE_BIAS, &sem->count);
  482. /* do optimistic spinning and steal lock if possible */
  483. if (rwsem_optimistic_spin(sem))
  484. return sem;
  485. /*
  486. * Optimistic spinning failed, proceed to the slowpath
  487. * and block until we can acquire the sem.
  488. */
  489. waiter.task = current;
  490. waiter.type = RWSEM_WAITING_FOR_WRITE;
  491. raw_spin_lock_irq(&sem->wait_lock);
  492. /* account for this before adding a new element to the list */
  493. if (list_empty(&sem->wait_list))
  494. waiting = false;
  495. #ifdef CONFIG_MTK_TASK_TURBO
  496. rwsem_list_add(waiter.task, &waiter.list, &sem->wait_list);
  497. #else
  498. list_add_tail(&waiter.list, &sem->wait_list);
  499. #endif
  500. /* we're now waiting on the lock, but no longer actively locking */
  501. if (waiting) {
  502. count = atomic_long_read(&sem->count);
  503. /*
  504. * If there were already threads queued before us and there are
  505. * no active writers, the lock must be read owned; so we try to
  506. * wake any read locks that were queued ahead of us.
  507. */
  508. if (count > RWSEM_WAITING_BIAS) {
  509. __rwsem_mark_wake(sem, RWSEM_WAKE_READERS, &wake_q);
  510. /*
  511. * The wakeup is normally called _after_ the wait_lock
  512. * is released, but given that we are proactively waking
  513. * readers we can deal with the wake_q overhead as it is
  514. * similar to releasing and taking the wait_lock again
  515. * for attempting rwsem_try_write_lock().
  516. */
  517. wake_up_q(&wake_q);
  518. /*
  519. * Reinitialize wake_q after use.
  520. */
  521. wake_q_init(&wake_q);
  522. }
  523. } else
  524. count = atomic_long_add_return(RWSEM_WAITING_BIAS, &sem->count);
  525. #ifdef CONFIG_MTK_TASK_TURBO
  526. /* inherit if current is turbo */
  527. rwsem_start_turbo_inherit(sem);
  528. #endif
  529. /* wait until we successfully acquire the lock */
  530. set_current_state(state);
  531. while (true) {
  532. if (rwsem_try_write_lock(count, sem))
  533. break;
  534. raw_spin_unlock_irq(&sem->wait_lock);
  535. /* Block until there are no active lockers. */
  536. do {
  537. if (signal_pending_state(state, current))
  538. goto out_nolock;
  539. schedule();
  540. set_current_state(state);
  541. } while ((count = atomic_long_read(&sem->count)) & RWSEM_ACTIVE_MASK);
  542. raw_spin_lock_irq(&sem->wait_lock);
  543. }
  544. __set_current_state(TASK_RUNNING);
  545. list_del(&waiter.list);
  546. raw_spin_unlock_irq(&sem->wait_lock);
  547. return ret;
  548. out_nolock:
  549. __set_current_state(TASK_RUNNING);
  550. raw_spin_lock_irq(&sem->wait_lock);
  551. list_del(&waiter.list);
  552. if (list_empty(&sem->wait_list))
  553. atomic_long_add(-RWSEM_WAITING_BIAS, &sem->count);
  554. else
  555. __rwsem_mark_wake(sem, RWSEM_WAKE_ANY, &wake_q);
  556. raw_spin_unlock_irq(&sem->wait_lock);
  557. wake_up_q(&wake_q);
  558. return ERR_PTR(-EINTR);
  559. }
  560. __visible struct rw_semaphore * __sched
  561. rwsem_down_write_failed(struct rw_semaphore *sem)
  562. {
  563. return __rwsem_down_write_failed_common(sem, TASK_UNINTERRUPTIBLE);
  564. }
  565. EXPORT_SYMBOL(rwsem_down_write_failed);
  566. __visible struct rw_semaphore * __sched
  567. rwsem_down_write_failed_killable(struct rw_semaphore *sem)
  568. {
  569. return __rwsem_down_write_failed_common(sem, TASK_KILLABLE);
  570. }
  571. EXPORT_SYMBOL(rwsem_down_write_failed_killable);
  572. /*
  573. * handle waking up a waiter on the semaphore
  574. * - up_read/up_write has decremented the active part of count if we come here
  575. */
  576. __visible
  577. struct rw_semaphore *rwsem_wake(struct rw_semaphore *sem)
  578. {
  579. unsigned long flags;
  580. DEFINE_WAKE_Q(wake_q);
  581. /*
  582. * __rwsem_down_write_failed_common(sem)
  583. * rwsem_optimistic_spin(sem)
  584. * osq_unlock(sem->osq)
  585. * ...
  586. * atomic_long_add_return(&sem->count)
  587. *
  588. * - VS -
  589. *
  590. * __up_write()
  591. * if (atomic_long_sub_return_release(&sem->count) < 0)
  592. * rwsem_wake(sem)
  593. * osq_is_locked(&sem->osq)
  594. *
  595. * And __up_write() must observe !osq_is_locked() when it observes the
  596. * atomic_long_add_return() in order to not miss a wakeup.
  597. *
  598. * This boils down to:
  599. *
  600. * [S.rel] X = 1 [RmW] r0 = (Y += 0)
  601. * MB RMB
  602. * [RmW] Y += 1 [L] r1 = X
  603. *
  604. * exists (r0=1 /\ r1=0)
  605. */
  606. smp_rmb();
  607. /*
  608. * If a spinner is present, it is not necessary to do the wakeup.
  609. * Try to do wakeup only if the trylock succeeds to minimize
  610. * spinlock contention which may introduce too much delay in the
  611. * unlock operation.
  612. *
  613. * spinning writer up_write/up_read caller
  614. * --------------- -----------------------
  615. * [S] osq_unlock() [L] osq
  616. * MB RMB
  617. * [RmW] rwsem_try_write_lock() [RmW] spin_trylock(wait_lock)
  618. *
  619. * Here, it is important to make sure that there won't be a missed
  620. * wakeup while the rwsem is free and the only spinning writer goes
  621. * to sleep without taking the rwsem. Even when the spinning writer
  622. * is just going to break out of the waiting loop, it will still do
  623. * a trylock in rwsem_down_write_failed() before sleeping. IOW, if
  624. * rwsem_has_spinner() is true, it will guarantee at least one
  625. * trylock attempt on the rwsem later on.
  626. */
  627. if (rwsem_has_spinner(sem)) {
  628. /*
  629. * The smp_rmb() here is to make sure that the spinner
  630. * state is consulted before reading the wait_lock.
  631. */
  632. smp_rmb();
  633. if (!raw_spin_trylock_irqsave(&sem->wait_lock, flags))
  634. return sem;
  635. goto locked;
  636. }
  637. raw_spin_lock_irqsave(&sem->wait_lock, flags);
  638. locked:
  639. if (!list_empty(&sem->wait_list))
  640. __rwsem_mark_wake(sem, RWSEM_WAKE_ANY, &wake_q);
  641. raw_spin_unlock_irqrestore(&sem->wait_lock, flags);
  642. wake_up_q(&wake_q);
  643. return sem;
  644. }
  645. EXPORT_SYMBOL(rwsem_wake);
  646. /*
  647. * downgrade a write lock into a read lock
  648. * - caller incremented waiting part of count and discovered it still negative
  649. * - just wake up any readers at the front of the queue
  650. */
  651. __visible
  652. struct rw_semaphore *rwsem_downgrade_wake(struct rw_semaphore *sem)
  653. {
  654. unsigned long flags;
  655. DEFINE_WAKE_Q(wake_q);
  656. raw_spin_lock_irqsave(&sem->wait_lock, flags);
  657. if (!list_empty(&sem->wait_list))
  658. __rwsem_mark_wake(sem, RWSEM_WAKE_READ_OWNED, &wake_q);
  659. raw_spin_unlock_irqrestore(&sem->wait_lock, flags);
  660. wake_up_q(&wake_q);
  661. return sem;
  662. }
  663. EXPORT_SYMBOL(rwsem_downgrade_wake);