/* -*- 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 nsThreadUtils_h__ #define nsThreadUtils_h__ #include "prthread.h" #include "prinrval.h" #include "MainThreadUtils.h" #include "nsICancelableRunnable.h" #include "nsIIdlePeriod.h" #include "nsIIdleRunnable.h" #include "nsINamed.h" #include "nsIRunnable.h" #include "nsIThreadManager.h" #include "nsITimer.h" #include "nsIThread.h" #include "nsStringGlue.h" #include "nsCOMPtr.h" #include "nsAutoPtr.h" #include "mozilla/Atomics.h" #include "mozilla/IndexSequence.h" #include "mozilla/Likely.h" #include "mozilla/Move.h" #include "mozilla/TimeStamp.h" #include "mozilla/Tuple.h" #include "mozilla/TypeTraits.h" //----------------------------------------------------------------------------- // These methods are alternatives to the methods on nsIThreadManager, provided // for convenience. /** * Create a new thread, and optionally provide an initial event for the thread. * * @param aResult * The resulting nsIThread object. * @param aInitialEvent * The initial event to run on this thread. This parameter may be null. * @param aStackSize * The size in bytes to reserve for the thread's stack. * * @returns NS_ERROR_INVALID_ARG * Indicates that the given name is not unique. */ extern nsresult NS_NewThread(nsIThread** aResult, nsIRunnable* aInitialEvent = nullptr, uint32_t aStackSize = nsIThreadManager::DEFAULT_STACK_SIZE); /** * Creates a named thread, otherwise the same as NS_NewThread */ extern nsresult NS_NewNamedThread(const nsACString& aName, nsIThread** aResult, nsIRunnable* aInitialEvent = nullptr, uint32_t aStackSize = nsIThreadManager::DEFAULT_STACK_SIZE); template inline nsresult NS_NewNamedThread(const char (&aName)[LEN], nsIThread** aResult, nsIRunnable* aInitialEvent = nullptr, uint32_t aStackSize = nsIThreadManager::DEFAULT_STACK_SIZE) { static_assert(LEN <= 16, "Thread name must be no more than 16 characters"); return NS_NewNamedThread(nsDependentCString(aName, LEN - 1), aResult, aInitialEvent, aStackSize); } /** * Get a reference to the current thread. * * @param aResult * The resulting nsIThread object. */ extern nsresult NS_GetCurrentThread(nsIThread** aResult); /** * Dispatch the given event to the current thread. * * @param aEvent * The event to dispatch. * * @returns NS_ERROR_INVALID_ARG * If event is null. */ extern nsresult NS_DispatchToCurrentThread(nsIRunnable* aEvent); extern nsresult NS_DispatchToCurrentThread(already_AddRefed&& aEvent); /** * Dispatch the given event to the main thread. * * @param aEvent * The event to dispatch. * @param aDispatchFlags * The flags to pass to the main thread's dispatch method. * * @returns NS_ERROR_INVALID_ARG * If event is null. */ extern nsresult NS_DispatchToMainThread(nsIRunnable* aEvent, uint32_t aDispatchFlags = NS_DISPATCH_NORMAL); extern nsresult NS_DispatchToMainThread(already_AddRefed&& aEvent, uint32_t aDispatchFlags = NS_DISPATCH_NORMAL); extern nsresult NS_DelayedDispatchToCurrentThread( already_AddRefed&& aEvent, uint32_t aDelayMs); /** * Dispatch the given event to the idle queue of the current thread. * * @param aEvent * The event to dispatch. * * @returns NS_ERROR_INVALID_ARG * If event is null. * @returns NS_ERROR_UNEXPECTED * If the thread is shutting down. */ extern nsresult NS_IdleDispatchToCurrentThread(already_AddRefed&& aEvent); /** * Dispatch the given event to the idle queue of the current thread. * * @param aEvent The event to dispatch. If the event implements * nsIIdleRunnable, it will receive a call on * nsIIdleRunnable::SetTimer when dispatched, with the value of * aTimeout. * * @param aTimeout The time in milliseconds until the event should be * moved from the idle queue to the regular queue, if it hasn't been * executed. If aEvent is also an nsIIdleRunnable, it is expected * that it should handle the timeout itself, after a call to * nsIIdleRunnable::SetTimer. * * @returns NS_ERROR_INVALID_ARG * If event is null. * @returns NS_ERROR_UNEXPECTED * If the thread is shutting down. */ extern nsresult NS_IdleDispatchToCurrentThread(already_AddRefed&& aEvent, uint32_t aTimeout); /** * Dispatch the given event to the idle queue of a thread. * * @param aEvent The event to dispatch. * * @param aThread The target thread for the dispatch. * * @returns NS_ERROR_INVALID_ARG * If event is null. * @returns NS_ERROR_UNEXPECTED * If the thread is shutting down. */ extern nsresult NS_IdleDispatchToThread(already_AddRefed&& aEvent, nsIThread* aThread); /** * Dispatch the given event to the idle queue of a thread. * * @param aEvent The event to dispatch. If the event implements * nsIIdleRunnable, it will receive a call on * nsIIdleRunnable::SetTimer when dispatched, with the value of * aTimeout. * * @param aTimeout The time in milliseconds until the event should be * moved from the idle queue to the regular queue, if it hasn't been * executed. If aEvent is also an nsIIdleRunnable, it is expected * that it should handle the timeout itself, after a call to * nsIIdleRunnable::SetTimer. * * @param aThread The target thread for the dispatch. * * @returns NS_ERROR_INVALID_ARG * If event is null. * @returns NS_ERROR_UNEXPECTED * If the thread is shutting down. */ extern nsresult NS_IdleDispatchToThread(already_AddRefed&& aEvent, uint32_t aTimeout, nsIThread* aThread); #ifndef XPCOM_GLUE_AVOID_NSPR /** * Process all pending events for the given thread before returning. This * method simply calls ProcessNextEvent on the thread while HasPendingEvents * continues to return true and the time spent in NS_ProcessPendingEvents * does not exceed the given timeout value. * * @param aThread * The thread object for which to process pending events. If null, then * events will be processed for the current thread. * @param aTimeout * The maximum number of milliseconds to spend processing pending events. * Events are not pre-empted to honor this timeout. Rather, the timeout * value is simply used to determine whether or not to process another event. * Pass PR_INTERVAL_NO_TIMEOUT to specify no timeout. */ extern nsresult NS_ProcessPendingEvents(nsIThread* aThread, PRIntervalTime aTimeout = PR_INTERVAL_NO_TIMEOUT); #endif /** * Shortcut for nsIThread::HasPendingEvents. * * It is an error to call this function when the given thread is not the * current thread. This function will return false if called from some * other thread. * * @param aThread * The current thread or null. * * @returns * A boolean value that if "true" indicates that there are pending events * in the current thread's event queue. */ extern bool NS_HasPendingEvents(nsIThread* aThread = nullptr); /** * Shortcut for nsIThread::ProcessNextEvent. * * It is an error to call this function when the given thread is not the * current thread. This function will simply return false if called * from some other thread. * * @param aThread * The current thread or null. * @param aMayWait * A boolean parameter that if "true" indicates that the method may block * the calling thread to wait for a pending event. * * @returns * A boolean value that if "true" indicates that an event from the current * thread's event queue was processed. */ extern bool NS_ProcessNextEvent(nsIThread* aThread = nullptr, bool aMayWait = true); // A wrapper for nested event loops. // // This function is intended to make code more obvious (do you remember // what NS_ProcessNextEvent(nullptr, true) means?) and slightly more // efficient, as people often pass nullptr or NS_GetCurrentThread to // NS_ProcessNextEvent, which results in needless querying of the current // thread every time through the loop. // // You should use this function in preference to NS_ProcessNextEvent inside // a loop unless one of the following is true: // // * You need to pass `false` to NS_ProcessNextEvent; or // * You need to do unusual things around the call to NS_ProcessNextEvent, // such as unlocking mutexes that you are holding. // // If you *do* need to call NS_ProcessNextEvent manually, please do call // NS_GetCurrentThread() outside of your loop and pass the returned pointer // into NS_ProcessNextEvent for a tiny efficiency win. namespace mozilla { // You should normally not need to deal with this template parameter. If // you enjoy esoteric event loop details, read on. // // If you specify that NS_ProcessNextEvent wait for an event, it is possible // for NS_ProcessNextEvent to return false, i.e. to indicate that an event // was not processed. This can only happen when the thread has been shut // down by another thread, but is still attempting to process events outside // of a nested event loop. // // This behavior is admittedly strange. The scenario it deals with is the // following: // // * The current thread has been shut down by some owner thread. // * The current thread is spinning an event loop waiting for some condition // to become true. // * Said condition is actually being fulfilled by another thread, so there // are timing issues in play. // // Thus, there is a small window where the current thread's event loop // spinning can check the condition, find it false, and call // NS_ProcessNextEvent to wait for another event. But we don't actually // want it to wait indefinitely, because there might not be any other events // in the event loop, and the current thread can't accept dispatched events // because it's being shut down. Thus, actually blocking would hang the // thread, which is bad. The solution, then, is to detect such a scenario // and not actually block inside NS_ProcessNextEvent. // // But this is a problem, because we want to return the status of // NS_ProcessNextEvent to the caller of SpinEventLoopUntil if possible. In // the above scenario, however, we'd stop spinning prematurely and cause // all sorts of havoc. We therefore have this template parameter to // control whether errors are ignored or passed out to the caller of // SpinEventLoopUntil. The latter is the default; if you find yourself // wanting to use the former, you should think long and hard before doing // so, and write a comment like this defending your choice. enum class ProcessFailureBehavior { IgnoreAndContinue, ReportToCaller, }; template bool SpinEventLoopUntil(Pred&& aPredicate, nsIThread* aThread = nullptr) { nsIThread* thread = aThread ? aThread : NS_GetCurrentThread(); while (!aPredicate()) { bool didSomething = NS_ProcessNextEvent(thread, true); if (Behavior == ProcessFailureBehavior::IgnoreAndContinue) { // Don't care what happened, continue on. continue; } else if (!didSomething) { return false; } } return true; } } // namespace mozilla /** * Returns true if we're in the compositor thread. * * We declare this here because the headers required to invoke * CompositorThreadHolder::IsInCompositorThread() also pull in a bunch of system * headers that #define various tokens in a way that can break the build. */ extern bool NS_IsInCompositorThread(); //----------------------------------------------------------------------------- // Helpers that work with nsCOMPtr: inline already_AddRefed do_GetCurrentThread() { nsIThread* thread = nullptr; NS_GetCurrentThread(&thread); return already_AddRefed(thread); } inline already_AddRefed do_GetMainThread() { nsIThread* thread = nullptr; NS_GetMainThread(&thread); return already_AddRefed(thread); } //----------------------------------------------------------------------------- #ifdef MOZILLA_INTERNAL_API // Fast access to the current thread. Do not release the returned pointer! If // you want to use this pointer from some other thread, then you will need to // AddRef it. Otherwise, you should only consider this pointer valid from code // running on the current thread. extern nsIThread* NS_GetCurrentThread(); /** * Set the name of the current thread. Prefer this function over * PR_SetCurrentThreadName() if possible. The name will also be included in the * crash report. * * @param aName * Name of the thread. A C language null-terminated string. */ extern void NS_SetCurrentThreadName(const char* aName); #endif //----------------------------------------------------------------------------- #ifndef XPCOM_GLUE_AVOID_NSPR namespace mozilla { // This class is designed to be subclassed. class IdlePeriod : public nsIIdlePeriod { public: NS_DECL_THREADSAFE_ISUPPORTS NS_DECL_NSIIDLEPERIOD IdlePeriod() {} protected: virtual ~IdlePeriod() {} private: IdlePeriod(const IdlePeriod&) = delete; IdlePeriod& operator=(const IdlePeriod&) = delete; IdlePeriod& operator=(const IdlePeriod&&) = delete; }; // Cancelable runnable methods implement nsICancelableRunnable, and // Idle and IdleWithTimer also nsIIdleRunnable. enum RunnableKind { Standard, Cancelable, Idle, IdleWithTimer }; // This class is designed to be subclassed. class Runnable : public nsIRunnable, public nsINamed { public: NS_DECL_THREADSAFE_ISUPPORTS NS_DECL_NSIRUNNABLE NS_DECL_NSINAMED Runnable() = delete; #ifdef RELEASE_OR_BETA explicit Runnable(const char* aName) {} #else explicit Runnable(const char* aName) : mName(aName) {} #endif protected: virtual ~Runnable() {} #ifndef RELEASE_OR_BETA const char* mName = nullptr; #endif private: Runnable(const Runnable&) = delete; Runnable& operator=(const Runnable&) = delete; Runnable& operator=(const Runnable&&) = delete; }; // This class is designed to be subclassed. class CancelableRunnable : public Runnable, public nsICancelableRunnable { public: NS_DECL_ISUPPORTS_INHERITED // nsICancelableRunnable virtual nsresult Cancel() override; CancelableRunnable() = delete; explicit CancelableRunnable(const char* aName) : Runnable(aName) {} protected: virtual ~CancelableRunnable() {} private: CancelableRunnable(const CancelableRunnable&) = delete; CancelableRunnable& operator=(const CancelableRunnable&) = delete; CancelableRunnable& operator=(const CancelableRunnable&&) = delete; }; // This class is designed to be subclassed. class IdleRunnable : public CancelableRunnable, public nsIIdleRunnable { public: NS_DECL_ISUPPORTS_INHERITED IdleRunnable() : CancelableRunnable("IdleRunnable") { } explicit IdleRunnable(const char* aName) : CancelableRunnable(aName) {} protected: virtual ~IdleRunnable() {} private: IdleRunnable(const IdleRunnable&) = delete; IdleRunnable& operator=(const IdleRunnable&) = delete; IdleRunnable& operator=(const IdleRunnable&&) = delete; }; namespace detail { // An event that can be used to call a C++11 functions or function objects, // including lambdas. The function must have no required arguments, and must // return void. template class RunnableFunction : public Runnable { public: template explicit RunnableFunction(F&& aFunction) : Runnable("RunnableFunction") , mFunction(Forward(aFunction)) { } NS_IMETHOD Run() override { static_assert(IsVoid::value, "The lambda must return void!"); mFunction(); return NS_OK; } private: StoredFunction mFunction; }; // Type alias for NS_NewRunnableFunction template using RunnableFunctionImpl = // Make sure we store a non-reference in nsRunnableFunction. typename detail::RunnableFunction::Type>; template inline already_AddRefed SetRunnableName(const char* aName, T* aObj) { #ifdef RELEASE_OR_BETA return do_AddRef(aObj); #else MOZ_RELEASE_ASSERT(aName); RefPtr ref(aObj); ref->SetName(aName); return ref.forget(); #endif } } // namespace detail namespace detail { template struct IsRefcountedSmartPointerHelper : FalseType {}; template struct IsRefcountedSmartPointerHelper> : TrueType {}; template struct IsRefcountedSmartPointerHelper> : TrueType {}; } // namespace detail template struct IsRefcountedSmartPointer : detail::IsRefcountedSmartPointerHelper::Type> {}; namespace detail { template struct RemoveSmartPointerHelper { typedef T Type; }; template struct RemoveSmartPointerHelper> { typedef Pointee Type; }; template struct RemoveSmartPointerHelper> { typedef Pointee Type; }; } // namespace detail template struct RemoveSmartPointer : detail::RemoveSmartPointerHelper::Type> {}; namespace detail { template struct RemoveRawOrSmartPointerHelper { typedef T Type; }; template struct RemoveRawOrSmartPointerHelper { typedef Pointee Type; }; template struct RemoveRawOrSmartPointerHelper> { typedef Pointee Type; }; template struct RemoveRawOrSmartPointerHelper> { typedef Pointee Type; }; } // namespace detail template struct RemoveRawOrSmartPointer : detail::RemoveRawOrSmartPointerHelper::Type> {}; } // namespace mozilla inline nsISupports* ToSupports(mozilla::Runnable *p) { return static_cast(p); } template already_AddRefed NS_NewRunnableFunction(const char* aName, Function&& aFunction) { // We store a non-reference in RunnableFunction, but still forward aFunction // to move if possible. return mozilla::detail::SetRunnableName( aName, new mozilla::detail::RunnableFunctionImpl( mozilla::Forward(aFunction))); } namespace mozilla { namespace detail { already_AddRefed CreateTimer(); template class TimerBehaviour { public: nsITimer* GetTimer() { return nullptr; } void CancelTimer() {} protected: ~TimerBehaviour() {} }; template <> class TimerBehaviour { public: nsITimer* GetTimer() { if (!mTimer) { mTimer = CreateTimer(); } return mTimer; } void CancelTimer() { if (mTimer) { mTimer->Cancel(); } } protected: ~TimerBehaviour() { CancelTimer(); } private: nsCOMPtr mTimer; }; } // namespace detail } // namespace mozilla // An event that can be used to call a method on a class. The class type must // support reference counting. This event supports Revoke for use // with nsRevocableEventPtr. template class nsRunnableMethod : public mozilla::Conditional::Type>::Type, protected mozilla::detail::TimerBehaviour { using BaseType = typename mozilla::Conditional::Type>::Type; public: nsRunnableMethod() : BaseType("nsRunnableMethod") {} virtual void Revoke() = 0; // These ReturnTypeEnforcer classes set up a blacklist for return types that // we know are not safe. The default ReturnTypeEnforcer compiles just fine but // already_AddRefed will not. template class ReturnTypeEnforcer { public: typedef int ReturnTypeIsSafe; }; template class ReturnTypeEnforcer> { // No ReturnTypeIsSafe makes this illegal! }; // Make sure this return type is safe. typedef typename ReturnTypeEnforcer::ReturnTypeIsSafe check; }; template struct nsRunnableMethodReceiver { RefPtr mObj; explicit nsRunnableMethodReceiver(ClassType* aObj) : mObj(aObj) {} ~nsRunnableMethodReceiver() { Revoke(); } ClassType* Get() const { return mObj.get(); } void Revoke() { mObj = nullptr; } }; template struct nsRunnableMethodReceiver { ClassType* MOZ_NON_OWNING_REF mObj; explicit nsRunnableMethodReceiver(ClassType* aObj) : mObj(aObj) {} ClassType* Get() const { return mObj; } void Revoke() { mObj = nullptr; } }; static inline constexpr bool IsIdle(mozilla::RunnableKind aKind) { return aKind == mozilla::Idle || aKind == mozilla::IdleWithTimer; } template struct nsRunnableMethodTraits; template struct nsRunnableMethodTraits { typedef typename mozilla::RemoveRawOrSmartPointer::Type class_type; static_assert(mozilla::IsBaseOf::value, "Stored class must inherit from method's class"); typedef R return_type; typedef nsRunnableMethod base_type; static const bool can_cancel = Kind == mozilla::Cancelable; }; template struct nsRunnableMethodTraits { typedef const typename mozilla::RemoveRawOrSmartPointer::Type class_type; static_assert(mozilla::IsBaseOf::value, "Stored class must inherit from method's class"); typedef R return_type; typedef nsRunnableMethod base_type; static const bool can_cancel = Kind == mozilla::Cancelable; }; #ifdef NS_HAVE_STDCALL template struct nsRunnableMethodTraits { typedef typename mozilla::RemoveRawOrSmartPointer::Type class_type; static_assert(mozilla::IsBaseOf::value, "Stored class must inherit from method's class"); typedef R return_type; typedef nsRunnableMethod base_type; static const bool can_cancel = Kind == mozilla::Cancelable; }; template struct nsRunnableMethodTraits { typedef typename mozilla::RemoveRawOrSmartPointer::Type class_type; static_assert(mozilla::IsBaseOf::value, "Stored class must inherit from method's class"); typedef R return_type; typedef nsRunnableMethod base_type; static const bool can_cancel = Kind == mozilla::Cancelable; }; template struct nsRunnableMethodTraits { typedef const typename mozilla::RemoveRawOrSmartPointer::Type class_type; static_assert(mozilla::IsBaseOf::value, "Stored class must inherit from method's class"); typedef R return_type; typedef nsRunnableMethod base_type; static const bool can_cancel = Kind == mozilla::Cancelable; }; template struct nsRunnableMethodTraits { typedef const typename mozilla::RemoveRawOrSmartPointer::Type class_type; static_assert(mozilla::IsBaseOf::value, "Stored class must inherit from method's class"); typedef R return_type; typedef nsRunnableMethod base_type; static const bool can_cancel = Kind == mozilla::Cancelable; }; #endif // IsParameterStorageClass::value is true if T is a parameter-storage class // that will be recognized by NS_New[NonOwning]RunnableMethodWithArg[s] to // force a specific storage&passing strategy (instead of inferring one, // see ParameterStorage). // When creating a new storage class, add a specialization for it to be // recognized. template struct IsParameterStorageClass : public mozilla::FalseType {}; // StoreXPassByY structs used to inform nsRunnableMethodArguments how to // store arguments, and how to pass them to the target method. template struct StoreCopyPassByValue { typedef T stored_type; typedef T passed_type; stored_type m; template MOZ_IMPLICIT StoreCopyPassByValue(A&& a) : m(mozilla::Forward(a)) {} passed_type PassAsParameter() { return m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreCopyPassByConstLRef { typedef T stored_type; typedef const T& passed_type; stored_type m; template MOZ_IMPLICIT StoreCopyPassByConstLRef(A&& a) : m(mozilla::Forward(a)) {} passed_type PassAsParameter() { return m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreCopyPassByLRef { typedef T stored_type; typedef T& passed_type; stored_type m; template MOZ_IMPLICIT StoreCopyPassByLRef(A&& a) : m(mozilla::Forward(a)) {} passed_type PassAsParameter() { return m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreCopyPassByRRef { typedef T stored_type; typedef T&& passed_type; stored_type m; template MOZ_IMPLICIT StoreCopyPassByRRef(A&& a) : m(mozilla::Forward(a)) {} passed_type PassAsParameter() { return mozilla::Move(m); } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreRefPassByLRef { typedef T& stored_type; typedef T& passed_type; stored_type m; template MOZ_IMPLICIT StoreRefPassByLRef(A& a) : m(a) {} passed_type PassAsParameter() { return m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreConstRefPassByConstLRef { typedef const T& stored_type; typedef const T& passed_type; stored_type m; template MOZ_IMPLICIT StoreConstRefPassByConstLRef(const A& a) : m(a) {} passed_type PassAsParameter() { return m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreRefPtrPassByPtr { typedef RefPtr stored_type; typedef T* passed_type; stored_type m; template MOZ_IMPLICIT StoreRefPtrPassByPtr(A&& a) : m(mozilla::Forward(a)) {} passed_type PassAsParameter() { return m.get(); } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StorePtrPassByPtr { typedef T* stored_type; typedef T* passed_type; stored_type m; template MOZ_IMPLICIT StorePtrPassByPtr(A a) : m(a) {} passed_type PassAsParameter() { return m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreConstPtrPassByConstPtr { typedef const T* stored_type; typedef const T* passed_type; stored_type m; template MOZ_IMPLICIT StoreConstPtrPassByConstPtr(A a) : m(a) {} passed_type PassAsParameter() { return m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreCopyPassByConstPtr { typedef T stored_type; typedef const T* passed_type; stored_type m; template MOZ_IMPLICIT StoreCopyPassByConstPtr(A&& a) : m(mozilla::Forward(a)) {} passed_type PassAsParameter() { return &m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; template struct StoreCopyPassByPtr { typedef T stored_type; typedef T* passed_type; stored_type m; template MOZ_IMPLICIT StoreCopyPassByPtr(A&& a) : m(mozilla::Forward(a)) {} passed_type PassAsParameter() { return &m; } }; template struct IsParameterStorageClass> : public mozilla::TrueType {}; namespace detail { template struct SFINAE1True : mozilla::TrueType {}; template static auto HasRefCountMethodsTest(int) -> SFINAE1True().AddRef(), mozilla::DeclVal().Release())>; template static auto HasRefCountMethodsTest(long) -> mozilla::FalseType; template struct HasRefCountMethods : decltype(HasRefCountMethodsTest(0)) {}; template struct NonnsISupportsPointerStorageClass : mozilla::Conditional::value, StoreConstPtrPassByConstPtr< typename mozilla::RemoveConst::Type>, StorePtrPassByPtr> {}; template struct PointerStorageClass : mozilla::Conditional::value, StoreRefPtrPassByPtr, typename NonnsISupportsPointerStorageClass< TWithoutPointer >::Type> {}; template struct LValueReferenceStorageClass : mozilla::Conditional::value, StoreConstRefPassByConstLRef< typename mozilla::RemoveConst::Type>, StoreRefPassByLRef> {}; template struct SmartPointerStorageClass : mozilla::Conditional::value, StoreRefPtrPassByPtr< typename mozilla::RemoveSmartPointer::Type>, StoreCopyPassByConstLRef> {}; template struct NonLValueReferenceStorageClass : mozilla::Conditional::value, StoreCopyPassByRRef< typename mozilla::RemoveReference::Type>, typename SmartPointerStorageClass::Type> {}; template struct NonPointerStorageClass : mozilla::Conditional::value, typename LValueReferenceStorageClass< typename mozilla::RemoveReference::Type >::Type, typename NonLValueReferenceStorageClass::Type> {}; template struct NonParameterStorageClass : mozilla::Conditional::value, typename PointerStorageClass< typename mozilla::RemovePointer::Type >::Type, typename NonPointerStorageClass::Type> {}; // Choose storage&passing strategy based on preferred storage type: // - If IsParameterStorageClass::value is true, use as-is. // - RC* -> StoreRefPtrPassByPtr : Store RefPtr, pass RC* // ^^ RC quacks like a ref-counted type (i.e., has AddRef and Release methods) // - const T* -> StoreConstPtrPassByConstPtr : Store const T*, pass const T* // - T* -> StorePtrPassByPtr : Store T*, pass T*. // - const T& -> StoreConstRefPassByConstLRef: Store const T&, pass const T&. // - T& -> StoreRefPassByLRef : Store T&, pass T&. // - T&& -> StoreCopyPassByRRef : Store T, pass Move(T). // - RefPtr, nsCOMPtr // -> StoreRefPtrPassByPtr : Store RefPtr, pass T* // - Other T -> StoreCopyPassByConstLRef : Store T, pass const T&. // Other available explicit options: // - StoreCopyPassByValue : Store T, pass T. // - StoreCopyPassByLRef : Store T, pass T& (of copy!) // - StoreCopyPassByConstPtr : Store T, pass const T* // - StoreCopyPassByPtr : Store T, pass T* (of copy!) // Or create your own class with PassAsParameter() method, optional // clean-up in destructor, and with associated IsParameterStorageClass<>. template struct ParameterStorage : mozilla::Conditional::value, T, typename NonParameterStorageClass::Type> {}; template static auto HasSetDeadlineTest(int) -> SFINAE1True().SetDeadline(mozilla::DeclVal()))>; template static auto HasSetDeadlineTest(long) -> mozilla::FalseType; template struct HasSetDeadline : decltype(HasSetDeadlineTest(0)) {}; template typename mozilla::EnableIf<::detail::HasSetDeadline::value>::Type SetDeadlineImpl(T* aObj, mozilla::TimeStamp aTimeStamp) { aObj->SetDeadline(aTimeStamp); } template typename mozilla::EnableIf::value>::Type SetDeadlineImpl(T* aObj, mozilla::TimeStamp aTimeStamp) { } } /* namespace detail */ namespace mozilla { namespace detail { // struct used to store arguments and later apply them to a method. template struct RunnableMethodArguments final { Tuple::Type...> mArguments; template explicit RunnableMethodArguments(As&&... aArguments) : mArguments(Forward(aArguments)...) {} template static auto applyImpl(C* o, M m, Tuple& args, IndexSequence) -> decltype(((*o).*m)(Get(args).PassAsParameter()...)) { return ((*o).*m)(Get(args).PassAsParameter()...); } template auto apply(C* o, M m) -> decltype(applyImpl(o, m, mArguments, typename IndexSequenceFor::Type())) { return applyImpl(o, m, mArguments, typename IndexSequenceFor::Type()); } }; template class RunnableMethodImpl final : public ::nsRunnableMethodTraits::base_type { typedef typename ::nsRunnableMethodTraits Traits; typedef typename Traits::class_type ClassType; typedef typename Traits::base_type BaseType; ::nsRunnableMethodReceiver mReceiver; Method mMethod; RunnableMethodArguments mArgs; using BaseType::GetTimer; using BaseType::CancelTimer; private: virtual ~RunnableMethodImpl() { Revoke(); }; static void TimedOut(nsITimer* aTimer, void* aClosure) { static_assert(IsIdle(Kind), "Don't use me!"); RefPtr r = static_cast(aClosure); r->SetDeadline(TimeStamp()); r->Run(); r->Cancel(); } public: template explicit RunnableMethodImpl(ForwardedPtrType&& aObj, Method aMethod, Args&&... aArgs) : mReceiver(Forward(aObj)) , mMethod(aMethod) , mArgs(Forward(aArgs)...) { static_assert(sizeof...(Storages) == sizeof...(Args), "Storages and Args should have equal sizes"); } NS_IMETHOD Run() { CancelTimer(); if (MOZ_LIKELY(mReceiver.Get())) { mArgs.apply(mReceiver.Get(), mMethod); } return NS_OK; } nsresult Cancel() { static_assert(Kind >= Cancelable, "Don't use me!"); Revoke(); return NS_OK; } void Revoke() { CancelTimer(); mReceiver.Revoke(); } void SetDeadline(TimeStamp aDeadline) { if (MOZ_LIKELY(mReceiver.Get())) { ::detail::SetDeadlineImpl(mReceiver.Get(), aDeadline); } } void SetTimer(uint32_t aDelay, nsIEventTarget* aTarget) { MOZ_ASSERT(aTarget); if (nsCOMPtr timer = GetTimer()) { timer->Cancel(); timer->SetTarget(aTarget); timer->InitWithNamedFuncCallback(TimedOut, this, aDelay, nsITimer::TYPE_ONE_SHOT, "detail::RunnableMethodImpl::SetTimer"); } } }; // Type aliases for NewRunnableMethod. template using OwningRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, true, Standard>::base_type; template using OwningRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, true, Standard, Storages...>; // Type aliases for NewCancelableRunnableMethod. template using CancelableRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, true, Cancelable>::base_type; template using CancelableRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, true, Cancelable, Storages...>; // Type aliases for NewIdleRunnableMethod. template using IdleRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, true, Idle>::base_type; template using IdleRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, true, Idle, Storages...>; // Type aliases for NewIdleRunnableMethodWithTimer. template using IdleRunnableMethodWithTimer = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, true, IdleWithTimer>::base_type; template using IdleRunnableMethodWithTimerImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, true, IdleWithTimer, Storages...>; // Type aliases for NewNonOwningRunnableMethod. template using NonOwningRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, false, Standard>::base_type; template using NonOwningRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, false, Standard, Storages...>; // Type aliases for NonOwningCancelableRunnableMethod template using NonOwningCancelableRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, false, Cancelable>::base_type; template using NonOwningCancelableRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, false, Cancelable, Storages...>; // Type aliases for NonOwningIdleRunnableMethod template using NonOwningIdleRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, false, Idle>::base_type; template using NonOwningIdleRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, false, Idle, Storages...>; // Type aliases for NewIdleRunnableMethodWithTimer. template using NonOwningIdleRunnableMethodWithTimer = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, false, IdleWithTimer>::base_type; template using NonOwningIdleRunnableMethodWithTimerImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, false, IdleWithTimer, Storages...>; } // namespace detail // NewRunnableMethod and friends // // Very often in Gecko, you'll find yourself in a situation where you want // to invoke a method (with or without arguments) asynchronously. You // could write a small helper class inheriting from nsRunnable to handle // all these details, or you could let NewRunnableMethod take care of all // those details for you. // // The simplest use of NewRunnableMethod looks like: // // nsCOMPtr event = // mozilla::NewRunnableMethod("description", myObject, &MyClass::HandleEvent); // NS_DispatchToCurrentThread(event); // // Statically enforced constraints: // - myObject must be of (or implicitly convertible to) type MyClass // - MyClass must define AddRef and Release methods // // The "description" string should specify a human-readable name for the // runnable; the provided string is used by various introspection tools // in the browser. // // The created runnable will take a strong reference to `myObject`. For // non-refcounted objects, or refcounted objects with unusual refcounting // requirements, and if and only if you are 110% certain that `myObject` // will live long enough, you can use NewNonOwningRunnableMethod instead, // which will, as its name implies, take a non-owning reference. If you // find yourself having to use this function, you should accompany your use // with a proof comment describing why the runnable will not lead to // use-after-frees. // // (If you find yourself writing contorted code to Release() an object // asynchronously on a different thread, you should use the // NS_ProxyRelease function.) // // Invoking a method with arguments takes a little more care. The // natural extension of the above: // // nsCOMPtr event = // mozilla::NewRunnableMethod("description", myObject, &MyClass::HandleEvent, // arg1, arg2, ...); // // can lead to security hazards (e.g. passing in raw pointers to refcounted // objects and storing those raw pointers in the runnable). We therefore // require you to specify the storage types used by the runnable, just as // you would if you were writing out the class by hand: // // nsCOMPtr event = // mozilla::NewRunnableMethod, nsTArray> // ("description", myObject, &MyClass::HandleEvent, arg1, arg2); // // Please note that you do not have to pass the same argument type as you // specify in the template arguments. For example, if you want to transfer // ownership to a runnable, you can write: // // RefPtr ptr = ...; // nsTArray array = ...; // nsCOMPtr event = // mozilla::NewRunnableMethod, nsTArray> // ("description", myObject, &MyClass::DoSomething, // Move(ptr), Move(array)); // // and there will be no extra AddRef/Release traffic, or copying of the array. // // Each type that you specify as a template argument to NewRunnableMethod // comes with its own style of storage in the runnable and its own style // of argument passing to the invoked method. See the comment for // ParameterStorage above for more details. // // If you need to customize the storage type and/or argument passing type, // you can write your own class to use as a template argument to // NewRunnableMethod. If you find yourself having to do that frequently, // please file a bug in Core::XPCOM about adding the custom type to the // core code in this file, and/or for custom rules for ParameterStorage // to select that strategy. // // For places that require you to use cancelable runnables, such as // workers, there's also NewCancelableRunnableMethod and its non-owning // counterpart. The runnables returned by these methods additionally // implement nsICancelableRunnable. // // Finally, all of the functions discussed above have additional overloads // that do not take a `const char*` as their first parameter; you may see // these in older code. The `const char*` overload is preferred and // should be used in new code exclusively. template already_AddRefed> NewRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::OwningRunnableMethodImpl( Forward(aPtr), aMethod)); } template already_AddRefed> NewCancelableRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::CancelableRunnableMethodImpl( Forward(aPtr), aMethod)); } template already_AddRefed> NewIdleRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::IdleRunnableMethodImpl( Forward(aPtr), aMethod)); } template already_AddRefed> NewIdleRunnableMethodWithTimer(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::IdleRunnableMethodWithTimerImpl( Forward(aPtr), aMethod)); } template already_AddRefed> NewNonOwningRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::NonOwningRunnableMethodImpl( Forward(aPtr), aMethod)); } template already_AddRefed> NewNonOwningCancelableRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::NonOwningCancelableRunnableMethodImpl( Forward(aPtr), aMethod)); } template already_AddRefed> NewNonOwningIdleRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::NonOwningIdleRunnableMethodImpl( Forward(aPtr), aMethod)); } template already_AddRefed> NewNonOwningIdleRunnableMethodWithTimer(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( aName, new detail::NonOwningIdleRunnableMethodWithTimerImpl( Forward(aPtr), aMethod)); } // Similar to NewRunnableMethod. Call like so: // nsCOMPtr event = // NewRunnableMethod(myObject, &MyClass::HandleEvent, myArg1,...); // 'Types' are the stored type for each argument, see ParameterStorage for details. template already_AddRefed> NewRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return detail::SetRunnableName( aName, new detail::OwningRunnableMethodImpl( Forward(aPtr), aMethod, mozilla::Forward(aArgs)...)); } template already_AddRefed> NewNonOwningRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return detail::SetRunnableName( aName, new detail::NonOwningRunnableMethodImpl( Forward(aPtr), aMethod, mozilla::Forward(aArgs)...)); } template already_AddRefed> NewCancelableRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return detail::SetRunnableName( aName, new detail::CancelableRunnableMethodImpl( Forward(aPtr), aMethod, mozilla::Forward(aArgs)...)); } template already_AddRefed> NewNonOwningCancelableRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return detail::SetRunnableName( aName, new detail::NonOwningCancelableRunnableMethodImpl( Forward(aPtr), aMethod, mozilla::Forward(aArgs)...)); } template already_AddRefed> NewIdleRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return detail::SetRunnableName( aName, new detail::IdleRunnableMethodImpl( Forward(aPtr), aMethod, mozilla::Forward(aArgs)...)); } template already_AddRefed> NewNonOwningIdleRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return detail::SetRunnableName( aName, new detail::NonOwningIdleRunnableMethodImpl( Forward(aPtr), aMethod, mozilla::Forward(aArgs)...)); } } // namespace mozilla #endif // XPCOM_GLUE_AVOID_NSPR // This class is designed to be used when you have an event class E that has a // pointer back to resource class R. If R goes away while E is still pending, // then it is important to "revoke" E so that it does not try use R after R has // been destroyed. nsRevocableEventPtr makes it easy for R to manage such // situations: // // class R; // // class E : public mozilla::Runnable { // public: // void Revoke() { // mResource = nullptr; // } // private: // R *mResource; // }; // // class R { // public: // void EventHandled() { // mEvent.Forget(); // } // private: // nsRevocableEventPtr mEvent; // }; // // void R::PostEvent() { // // Make sure any pending event is revoked. // mEvent->Revoke(); // // nsCOMPtr event = new E(); // if (NS_SUCCEEDED(NS_DispatchToCurrentThread(event))) { // // Keep pointer to event so we can revoke it. // mEvent = event; // } // } // // NS_IMETHODIMP E::Run() { // if (!mResource) // return NS_OK; // ... // mResource->EventHandled(); // return NS_OK; // } // template class nsRevocableEventPtr { public: nsRevocableEventPtr() : mEvent(nullptr) {} ~nsRevocableEventPtr() { Revoke(); } const nsRevocableEventPtr& operator=(T* aEvent) { if (mEvent != aEvent) { Revoke(); mEvent = aEvent; } return *this; } const nsRevocableEventPtr& operator=(already_AddRefed aEvent) { RefPtr event = aEvent; if (mEvent != event) { Revoke(); mEvent = event.forget(); } return *this; } void Revoke() { if (mEvent) { mEvent->Revoke(); mEvent = nullptr; } } void Forget() { mEvent = nullptr; } bool IsPending() { return mEvent != nullptr; } T* get() { return mEvent; } private: // Not implemented nsRevocableEventPtr(const nsRevocableEventPtr&); nsRevocableEventPtr& operator=(const nsRevocableEventPtr&); RefPtr mEvent; }; /** * A simple helper to suffix thread pool name * with incremental numbers. */ class nsThreadPoolNaming { public: nsThreadPoolNaming() : mCounter(0) {} /** * Returns a thread name as " #" and increments the counter. */ nsCString GetNextThreadName(const nsACString& aPoolName); template nsCString GetNextThreadName(const char (&aPoolName)[LEN]) { return GetNextThreadName(nsDependentCString(aPoolName, LEN - 1)); } private: mozilla::Atomic mCounter; nsThreadPoolNaming(const nsThreadPoolNaming&) = delete; void operator=(const nsThreadPoolNaming&) = delete; }; /** * Thread priority in most operating systems affect scheduling, not IO. This * helper is used to set the current thread to low IO priority for the lifetime * of the created object. You can only use this low priority IO setting within * the context of the current thread. */ class MOZ_STACK_CLASS nsAutoLowPriorityIO { public: nsAutoLowPriorityIO(); ~nsAutoLowPriorityIO(); private: bool lowIOPrioritySet; #if defined(XP_MACOSX) int oldPriority; #endif }; void NS_SetMainThread(); /** * Return the expiration time of the next timer to run on the current * thread. If that expiration time is greater than aDefault, then * return aDefault. aSearchBound specifies a maximum number of timers * to examine to find a timer on the current thread. If no timer that * will run on the current thread is found after examining * aSearchBound timers, return the highest seen expiration time as a * best effort guess. * * Timers with either the type nsITimer::TYPE_ONE_SHOT_LOW_PRIORITY or * nsITIMER::TYPE_REPEATING_SLACK_LOW_PRIORITY will be skipped when * searching for the next expiration time. This enables timers to * have lower priority than callbacks dispatched from * nsIThread::IdleDispatch. */ extern mozilla::TimeStamp NS_GetTimerDeadlineHintOnCurrentThread(mozilla::TimeStamp aDefault, uint32_t aSearchBound); namespace mozilla { /** * Cooperative thread scheduling is governed by two rules: * - Only one thread in the pool of cooperatively scheduled threads runs at a * time. * - Thread switching happens at well-understood safe points. * * In some cases we may want to treat all the threads in a cooperative pool as a * single thread, while other parts of the code may want to view them as separate * threads. GetCurrentVirtualThread() will return the same value for all * threads in a cooperative thread pool. GetCurrentPhysicalThread will return a * different value for each thread in the pool. * * Thread safety assertions are a concrete example where GetCurrentVirtualThread * should be used. An object may want to assert that it only can be used on the * thread that created it. Such assertions would normally prevent the object * from being used on different cooperative threads. However, the object might * really only care that it's used atomically. Cooperative scheduling guarantees * that it will be (assuming we don't yield in the middle of modifying the * object). So we can weaken the assertion to compare the virtual thread the * object was created on to the virtual thread on which it's being used. This * assertion allows the object to be used across threads in a cooperative thread * pool while preventing accesses across preemptively scheduled threads (which * would be unsafe). */ // Returns the PRThread on which this code is running. PRThread* GetCurrentPhysicalThread(); // Returns a "virtual" PRThread that should only be used for comparison with // other calls to GetCurrentVirtualThread. Two threads in the same cooperative // thread pool will return the same virtual thread. Threads that are not // cooperatively scheduled will have their own unique virtual PRThread (which // will be equal to their physical PRThread). // // The return value of GetCurrentVirtualThread() is guaranteed not to change // throughout the lifetime of a thread. // // Note that the original main thread (the first one created in the process) is // considered as part of the pool of cooperative threads, so the return value of // GetCurrentVirtualThread() for this thread (throughout its lifetime, even // during shutdown) is the same as the return value from any other thread in the // cooperative pool. PRThread* GetCurrentVirtualThread(); // These functions return event targets that can be used to dispatch to the // current or main thread. They can also be used to test if you're on those // threads (via IsOnCurrentThread). These functions should be used in preference // to the nsIThread-based NS_Get{Current,Main}Thread functions since they will // return more useful answers in the case of threads sharing an event loop. nsIEventTarget* GetCurrentThreadEventTarget(); nsIEventTarget* GetMainThreadEventTarget(); // These variants of the above functions assert that the given thread has a // serial event target (i.e., that it's not part of a thread pool) and returns // that. nsISerialEventTarget* GetCurrentThreadSerialEventTarget(); nsISerialEventTarget* GetMainThreadSerialEventTarget(); } // namespace mozilla #endif // nsThreadUtils_h__