rwlock.c 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. #include "config.h"
  2. #include "rwlock.h"
  3. #include "bool.h"
  4. #include "atomic.h"
  5. #include "threads.h"
  6. /* A simple spinlock. Yield the thread while the given integer is set by
  7. * another. Could probably be improved... */
  8. #define LOCK(l) do { \
  9. while(ATOMIC_EXCHANGE(int, &(l), true) == true) \
  10. althrd_yield(); \
  11. } while(0)
  12. #define UNLOCK(l) ATOMIC_STORE(&(l), false)
  13. void RWLockInit(RWLock *lock)
  14. {
  15. InitRef(&lock->read_count, 0);
  16. InitRef(&lock->write_count, 0);
  17. ATOMIC_INIT(&lock->read_lock, false);
  18. ATOMIC_INIT(&lock->read_entry_lock, false);
  19. ATOMIC_INIT(&lock->write_lock, false);
  20. }
  21. void ReadLock(RWLock *lock)
  22. {
  23. LOCK(lock->read_entry_lock);
  24. LOCK(lock->read_lock);
  25. if(IncrementRef(&lock->read_count) == 1)
  26. LOCK(lock->write_lock);
  27. UNLOCK(lock->read_lock);
  28. UNLOCK(lock->read_entry_lock);
  29. }
  30. void ReadUnlock(RWLock *lock)
  31. {
  32. if(DecrementRef(&lock->read_count) == 0)
  33. UNLOCK(lock->write_lock);
  34. }
  35. void WriteLock(RWLock *lock)
  36. {
  37. if(IncrementRef(&lock->write_count) == 1)
  38. LOCK(lock->read_lock);
  39. LOCK(lock->write_lock);
  40. }
  41. void WriteUnlock(RWLock *lock)
  42. {
  43. UNLOCK(lock->write_lock);
  44. if(DecrementRef(&lock->write_count) == 0)
  45. UNLOCK(lock->read_lock);
  46. }