gecko-dev/xpcom/base/StaticMutex.h

136 строки
3.5 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/. */
#ifndef mozilla_StaticMutex_h
#define mozilla_StaticMutex_h
#include "mozilla/Atomics.h"
#include "mozilla/Mutex.h"
namespace mozilla {
/**
* StaticMutex is a Mutex that can (and in fact, must) be used as a
* global/static variable.
*
* The main reason to use StaticMutex as opposed to
* StaticAutoPtr<OffTheBooksMutex> is that we instantiate the StaticMutex in a
* thread-safe manner the first time it's used.
*
* The same caveats that apply to StaticAutoPtr apply to StaticMutex. In
* particular, do not use StaticMutex as a stack variable or a class instance
* variable, because this class relies on the fact that global variablies are
* initialized to 0 in order to initialize mMutex. It is only safe to use
* StaticMutex as a global or static variable.
*/
template <recordreplay::Behavior Recording>
class MOZ_ONLY_USED_TO_AVOID_STATIC_CONSTRUCTORS BaseStaticMutex
{
public:
// In debug builds, check that mMutex is initialized for us as we expect by
// the compiler. In non-debug builds, don't declare a constructor so that
// the compiler can see that the constructor is trivial.
#ifdef DEBUG
BaseStaticMutex()
{
MOZ_ASSERT(!mMutex);
}
#endif
void Lock()
{
Mutex()->Lock();
}
void Unlock()
{
Mutex()->Unlock();
}
void AssertCurrentThreadOwns()
{
#ifdef DEBUG
Mutex()->AssertCurrentThreadOwns();
#endif
}
private:
OffTheBooksMutex* Mutex()
{
if (mMutex) {
return mMutex;
}
OffTheBooksMutex* mutex = new OffTheBooksMutex("StaticMutex", Recording);
if (!mMutex.compareExchange(nullptr, mutex)) {
delete mutex;
}
return mMutex;
}
Atomic<OffTheBooksMutex*, SequentiallyConsistent, Recording> mMutex;
// Disallow copy constructor, but only in debug mode. We only define
// a default constructor in debug mode (see above); if we declared
// this constructor always, the compiler wouldn't generate a trivial
// default constructor for us in non-debug mode.
#ifdef DEBUG
BaseStaticMutex(BaseStaticMutex& aOther);
#endif
// Disallow these operators.
BaseStaticMutex& operator=(BaseStaticMutex* aRhs);
static void* operator new(size_t) CPP_THROW_NEW;
static void operator delete(void*);
};
typedef BaseStaticMutex<recordreplay::Behavior::Preserve> StaticMutex;
typedef BaseStaticMutex<recordreplay::Behavior::DontPreserve> StaticMutexNotRecorded;
// Helper for StaticMutexAutoLock/Unlock.
class MOZ_STACK_CLASS AnyStaticMutex
{
public:
MOZ_IMPLICIT AnyStaticMutex(StaticMutex& aMutex)
: mStaticMutex(&aMutex), mStaticMutexNotRecorded(nullptr)
{}
MOZ_IMPLICIT AnyStaticMutex(StaticMutexNotRecorded& aMutex)
: mStaticMutex(nullptr), mStaticMutexNotRecorded(&aMutex)
{}
void Lock()
{
if (mStaticMutex) {
mStaticMutex->Lock();
} else {
mStaticMutexNotRecorded->Lock();
}
}
void Unlock()
{
if (mStaticMutex) {
mStaticMutex->Unlock();
} else {
mStaticMutexNotRecorded->Unlock();
}
}
private:
StaticMutex* mStaticMutex;
StaticMutexNotRecorded* mStaticMutexNotRecorded;
};
typedef BaseAutoLock<AnyStaticMutex> StaticMutexAutoLock;
typedef BaseAutoUnlock<AnyStaticMutex> StaticMutexAutoUnlock;
} // namespace mozilla
#endif