gecko-dev/xpcom/threads/RecursiveMutex.h

113 строки
2.9 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
// A lock that can be acquired multiple times on the same thread.
#ifndef mozilla_RecursiveMutex_h
#define mozilla_RecursiveMutex_h
#include "mozilla/BlockingResourceBase.h"
#ifndef XP_WIN
# include <pthread.h>
#endif
namespace mozilla {
class RecursiveMutex : public BlockingResourceBase {
public:
explicit RecursiveMutex(const char* aName);
~RecursiveMutex();
#ifdef DEBUG
void Lock();
void Unlock();
#else
void Lock() { LockInternal(); }
void Unlock() { UnlockInternal(); }
#endif
#ifdef DEBUG
/**
* AssertCurrentThreadIn
**/
void AssertCurrentThreadIn();
/**
* AssertNotCurrentThreadIn
**/
void AssertNotCurrentThreadIn() {
// Not currently implemented. See bug 476536 for discussion.
}
#else
void AssertCurrentThreadIn() {}
void AssertNotCurrentThreadIn() {}
#endif
private:
RecursiveMutex() = delete;
RecursiveMutex(const RecursiveMutex&) = delete;
RecursiveMutex& operator=(const RecursiveMutex&) = delete;
void LockInternal();
void UnlockInternal();
#ifdef DEBUG
PRThread* mOwningThread;
size_t mEntryCount;
#endif
#if !defined(XP_WIN)
pthread_mutex_t mMutex;
#else
// We eschew including windows.h and using CRITICAL_SECTION here so that files
// including us don't also pull in windows.h. Just use a type that's big
// enough for CRITICAL_SECTION, and we'll fix it up later.
void* mMutex[6];
#endif
};
class MOZ_RAII RecursiveMutexAutoLock {
public:
explicit RecursiveMutexAutoLock(RecursiveMutex& aRecursiveMutex)
: mRecursiveMutex(&aRecursiveMutex) {
NS_ASSERTION(mRecursiveMutex, "null mutex");
mRecursiveMutex->Lock();
}
~RecursiveMutexAutoLock(void) { mRecursiveMutex->Unlock(); }
private:
RecursiveMutexAutoLock() = delete;
RecursiveMutexAutoLock(const RecursiveMutexAutoLock&) = delete;
RecursiveMutexAutoLock& operator=(const RecursiveMutexAutoLock&) = delete;
static void* operator new(size_t) noexcept(true);
mozilla::RecursiveMutex* mRecursiveMutex;
};
class MOZ_RAII RecursiveMutexAutoUnlock {
public:
explicit RecursiveMutexAutoUnlock(RecursiveMutex& aRecursiveMutex)
: mRecursiveMutex(&aRecursiveMutex) {
NS_ASSERTION(mRecursiveMutex, "null mutex");
mRecursiveMutex->Unlock();
}
~RecursiveMutexAutoUnlock(void) { mRecursiveMutex->Lock(); }
private:
RecursiveMutexAutoUnlock() = delete;
RecursiveMutexAutoUnlock(const RecursiveMutexAutoUnlock&) = delete;
RecursiveMutexAutoUnlock& operator=(const RecursiveMutexAutoUnlock&) = delete;
static void* operator new(size_t) noexcept(true);
mozilla::RecursiveMutex* mRecursiveMutex;
};
} // namespace mozilla
#endif // mozilla_RecursiveMutex_h