mutex_32.h 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. /*
  2. * Assembly implementation of the mutex fastpath, based on atomic
  3. * decrement/increment.
  4. *
  5. * started by Ingo Molnar:
  6. *
  7. * Copyright (C) 2004, 2005, 2006 Red Hat, Inc., Ingo Molnar <mingo@redhat.com>
  8. */
  9. #ifndef _ASM_MUTEX_H
  10. #define _ASM_MUTEX_H
  11. #include "asm/alternative.h"
  12. /**
  13. * __mutex_fastpath_lock - try to take the lock by moving the count
  14. * from 1 to a 0 value
  15. * @count: pointer of type atomic_t
  16. * @fn: function to call if the original value was not 1
  17. *
  18. * Change the count from 1 to a value lower than 1, and call <fn> if it
  19. * wasn't 1 originally. This function MUST leave the value lower than 1
  20. * even when the "1" assertion wasn't true.
  21. */
  22. #define __mutex_fastpath_lock(count, fail_fn) \
  23. do { \
  24. unsigned int dummy; \
  25. \
  26. typecheck(atomic_t *, count); \
  27. typecheck_fn(fastcall void (*)(atomic_t *), fail_fn); \
  28. \
  29. __asm__ __volatile__( \
  30. LOCK_PREFIX " decl (%%eax) \n" \
  31. " jns 1f \n" \
  32. " call "#fail_fn" \n" \
  33. "1: \n" \
  34. \
  35. :"=a" (dummy) \
  36. : "a" (count) \
  37. : "memory", "ecx", "edx"); \
  38. } while (0)
  39. /**
  40. * __mutex_fastpath_lock_retval - try to take the lock by moving the count
  41. * from 1 to a 0 value
  42. * @count: pointer of type atomic_t
  43. * @fail_fn: function to call if the original value was not 1
  44. *
  45. * Change the count from 1 to a value lower than 1, and call <fail_fn> if it
  46. * wasn't 1 originally. This function returns 0 if the fastpath succeeds,
  47. * or anything the slow path function returns
  48. */
  49. static inline int
  50. __mutex_fastpath_lock_retval(atomic_t *count,
  51. int fastcall (*fail_fn)(atomic_t *))
  52. {
  53. if (unlikely(atomic_dec_return(count) < 0))
  54. return fail_fn(count);
  55. else
  56. return 0;
  57. }
  58. /**
  59. * __mutex_fastpath_unlock - try to promote the mutex from 0 to 1
  60. * @count: pointer of type atomic_t
  61. * @fail_fn: function to call if the original value was not 0
  62. *
  63. * try to promote the mutex from 0 to 1. if it wasn't 0, call <fail_fn>.
  64. * In the failure case, this function is allowed to either set the value
  65. * to 1, or to set it to a value lower than 1.
  66. *
  67. * If the implementation sets it to a value of lower than 1, the
  68. * __mutex_slowpath_needs_to_unlock() macro needs to return 1, it needs
  69. * to return 0 otherwise.
  70. */
  71. #define __mutex_fastpath_unlock(count, fail_fn) \
  72. do { \
  73. unsigned int dummy; \
  74. \
  75. typecheck(atomic_t *, count); \
  76. typecheck_fn(fastcall void (*)(atomic_t *), fail_fn); \
  77. \
  78. __asm__ __volatile__( \
  79. LOCK_PREFIX " incl (%%eax) \n" \
  80. " jg 1f \n" \
  81. " call "#fail_fn" \n" \
  82. "1: \n" \
  83. \
  84. :"=a" (dummy) \
  85. : "a" (count) \
  86. : "memory", "ecx", "edx"); \
  87. } while (0)
  88. #define __mutex_slowpath_needs_to_unlock() 1
  89. /**
  90. * __mutex_fastpath_trylock - try to acquire the mutex, without waiting
  91. *
  92. * @count: pointer of type atomic_t
  93. * @fail_fn: fallback function
  94. *
  95. * Change the count from 1 to a value lower than 1, and return 0 (failure)
  96. * if it wasn't 1 originally, or return 1 (success) otherwise. This function
  97. * MUST leave the value lower than 1 even when the "1" assertion wasn't true.
  98. * Additionally, if the value was < 0 originally, this function must not leave
  99. * it to 0 on failure.
  100. */
  101. static inline int
  102. __mutex_fastpath_trylock(atomic_t *count, int (*fail_fn)(atomic_t *))
  103. {
  104. /*
  105. * We have two variants here. The cmpxchg based one is the best one
  106. * because it never induce a false contention state. It is included
  107. * here because architectures using the inc/dec algorithms over the
  108. * xchg ones are much more likely to support cmpxchg natively.
  109. *
  110. * If not we fall back to the spinlock based variant - that is
  111. * just as efficient (and simpler) as a 'destructive' probing of
  112. * the mutex state would be.
  113. */
  114. #ifdef __HAVE_ARCH_CMPXCHG
  115. if (likely(atomic_cmpxchg(count, 1, 0) == 1))
  116. return 1;
  117. return 0;
  118. #else
  119. return fail_fn(count);
  120. #endif
  121. }
  122. #endif