xpcom/threads/RWLock.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/RWLock.h"
   8
   9 #ifdef XP_WIN
  10 #  include <windows.h>
  11
  12 static_assert(sizeof(SRWLOCK) <= sizeof(void*), "SRWLOCK is too big!");
  13
  14 #  define NativeHandle(m) (reinterpret_cast<SRWLOCK*>(&m))
  15 #else
  16 #  define NativeHandle(m) (&m)
  17 #endif
  18
  19 namespace mozilla {
  20
  21 RWLock::RWLock(const char* aName)
  22     : BlockingResourceBase(aName, eMutex)
  23 #ifdef DEBUG
  24       ,
  25       mOwningThread(nullptr)
  26 #endif
  27 {
  28 #ifdef XP_WIN
  29   InitializeSRWLock(NativeHandle(mRWLock));
  30 #else
  31   MOZ_RELEASE_ASSERT(pthread_rwlock_init(NativeHandle(mRWLock), nullptr) == 0,
  32                      "pthread_rwlock_init failed");
  33 #endif
  34 }
  35
  36 #ifdef DEBUG
  37 bool RWLock::LockedForWritingByCurrentThread() {
  38   return mOwningThread == PR_GetCurrentThread();
  39 }
  40 #endif
  41
  42 #ifndef XP_WIN
  43 RWLock::~RWLock() {
  44   MOZ_RELEASE_ASSERT(pthread_rwlock_destroy(NativeHandle(mRWLock)) == 0,
  45                      "pthread_rwlock_destroy failed");
  46 }
  47 #endif
  48
  49 void RWLock::ReadLockInternal() {
  50 #ifdef XP_WIN
  51   AcquireSRWLockShared(NativeHandle(mRWLock));
  52 #else
  53   MOZ_RELEASE_ASSERT(pthread_rwlock_rdlock(NativeHandle(mRWLock)) == 0,
  54                      "pthread_rwlock_rdlock failed");
  55 #endif
  56 }
  57
  58 void RWLock::ReadUnlockInternal() {
  59 #ifdef XP_WIN
  60   ReleaseSRWLockShared(NativeHandle(mRWLock));
  61 #else
  62   MOZ_RELEASE_ASSERT(pthread_rwlock_unlock(NativeHandle(mRWLock)) == 0,
  63                      "pthread_rwlock_unlock failed");
  64 #endif
  65 }
  66
  67 void RWLock::WriteLockInternal() {
  68 #ifdef XP_WIN
  69   AcquireSRWLockExclusive(NativeHandle(mRWLock));
  70 #else
  71   MOZ_RELEASE_ASSERT(pthread_rwlock_wrlock(NativeHandle(mRWLock)) == 0,
  72                      "pthread_rwlock_wrlock failed");
  73 #endif
  74 }
  75
  76 void RWLock::WriteUnlockInternal() {
  77 #ifdef XP_WIN
  78   ReleaseSRWLockExclusive(NativeHandle(mRWLock));
  79 #else
  80   MOZ_RELEASE_ASSERT(pthread_rwlock_unlock(NativeHandle(mRWLock)) == 0,
  81                      "pthread_rwlock_unlock failed");
  82 #endif
  83 }
  84
  85 }  // namespace mozilla
  86
  87 #undef NativeHandle