lib/dec_and_lock.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include <linux/export.h>
   3 #include <linux/spinlock.h>
   4 #include <linux/atomic.h>
   5
   6 /*
   7  * This is an implementation of the notion of "decrement a
   8  * reference count, and return locked if it decremented to zero".
   9  *
  10  * NOTE NOTE NOTE! This is _not_ equivalent to
  11  *
  12  *      if (atomic_dec_and_test(&atomic)) {
  13  *              spin_lock(&lock);
  14  *              return 1;
  15  *      }
  16  *      return 0;
  17  *
  18  * because the spin-lock and the decrement must be
  19  * "atomic".
  20  */
  21 int _atomic_dec_and_lock(atomic_t *atomic, spinlock_t *lock)
  22 {
  23         /* Subtract 1 from counter unless that drops it to 0 (ie. it was 1) */
  24         if (atomic_add_unless(atomic, -1, 1))
  25                 return 0;
  26
  27         /* Otherwise do it the slow way */
  28         spin_lock(lock);
  29         if (atomic_dec_and_test(atomic))
  30                 return 1;
  31         spin_unlock(lock);
  32         return 0;
  33 }
  34
  35 EXPORT_SYMBOL(_atomic_dec_and_lock);
  36
  37 int _atomic_dec_and_lock_irqsave(atomic_t *atomic, spinlock_t *lock,
  38                                  unsigned long *flags)
  39 {
  40         /* Subtract 1 from counter unless that drops it to 0 (ie. it was 1) */
  41         if (atomic_add_unless(atomic, -1, 1))
  42                 return 0;
  43
  44         /* Otherwise do it the slow way */
  45         spin_lock_irqsave(lock, *flags);
  46         if (atomic_dec_and_test(atomic))
  47                 return 1;
  48         spin_unlock_irqrestore(lock, *flags);
  49         return 0;
  50 }
  51 EXPORT_SYMBOL(_atomic_dec_and_lock_irqsave);