locking/percpu-rwsem: Extract __percpu_down_read_trylock()

In preparation for removing the embedded rwsem and building a custom
lock, extract the read-trylock primitive.

Signed-off-by: Peter Zijlstra (Intel) <peterz@infradead.org>
Signed-off-by: Ingo Molnar <mingo@kernel.org>
Reviewed-by: Davidlohr Bueso <dbueso@suse.de>
Acked-by: Will Deacon <will@kernel.org>
Acked-by: Waiman Long <longman@redhat.com>
Tested-by: Juri Lelli <juri.lelli@redhat.com>
Link: https://lkml.kernel.org/r/20200131151540.098485539@infradead.org
This commit is contained in:
Peter Zijlstra 2019-10-31 12:34:23 +01:00 committed by Ingo Molnar
parent 71365d4023
commit 75ff64572e
1 changed files with 13 additions and 6 deletions

View File

@ -45,7 +45,7 @@ void percpu_free_rwsem(struct percpu_rw_semaphore *sem)
} }
EXPORT_SYMBOL_GPL(percpu_free_rwsem); EXPORT_SYMBOL_GPL(percpu_free_rwsem);
bool __percpu_down_read(struct percpu_rw_semaphore *sem, bool try) static bool __percpu_down_read_trylock(struct percpu_rw_semaphore *sem)
{ {
__this_cpu_inc(*sem->read_count); __this_cpu_inc(*sem->read_count);
@ -73,11 +73,18 @@ bool __percpu_down_read(struct percpu_rw_semaphore *sem, bool try)
if (likely(!smp_load_acquire(&sem->readers_block))) if (likely(!smp_load_acquire(&sem->readers_block)))
return true; return true;
/* __this_cpu_dec(*sem->read_count);
* Per the above comment; we still have preemption disabled and
* will thus decrement on the same CPU as we incremented. /* Prod writer to re-evaluate readers_active_check() */
*/ rcuwait_wake_up(&sem->writer);
__percpu_up_read(sem);
return false;
}
bool __percpu_down_read(struct percpu_rw_semaphore *sem, bool try)
{
if (__percpu_down_read_trylock(sem))
return true;
if (try) if (try)
return false; return false;