/* -*- 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 "nsIIncrementalRunnable.h" #include "nsINamed.h" #include "nsIRunnable.h" #include "nsIThreadManager.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); extern nsresult NS_IdleDispatchToCurrentThread(already_AddRefed&& aEvent); #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); /** * 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; }; // 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() {} #ifdef RELEASE_OR_BETA explicit Runnable(const char* aName) {} #else explicit Runnable(const char* aName) : mName(aName) {} #endif protected: virtual ~Runnable() {} private: Runnable(const Runnable&) = delete; Runnable& operator=(const Runnable&) = delete; Runnable& operator=(const Runnable&&) = delete; #ifndef RELEASE_OR_BETA const char* mName = nullptr; #endif }; // This class is designed to be subclassed. class CancelableRunnable : public Runnable, public nsICancelableRunnable { public: NS_DECL_ISUPPORTS_INHERITED // nsICancelableRunnable virtual nsresult Cancel() override; CancelableRunnable() {} 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 IncrementalRunnable : public CancelableRunnable, public nsIIncrementalRunnable { public: NS_DECL_ISUPPORTS_INHERITED // nsIIncrementalRunnable virtual void SetDeadline(TimeStamp aDeadline) override; IncrementalRunnable() {} protected: virtual ~IncrementalRunnable() {} private: IncrementalRunnable(const IncrementalRunnable&) = delete; IncrementalRunnable& operator=(const IncrementalRunnable&) = delete; IncrementalRunnable& operator=(const IncrementalRunnable&&) = 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) : 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(already_AddRefed&& aObj, const char* aName) { MOZ_RELEASE_ASSERT(aName); RefPtr ref(aObj); ref->SetName(aName); return ref.forget(); } } // 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(Function&& aFunction) { // We store a non-reference in RunnableFunction, but still forward aFunction // to move if possible. return do_AddRef(new mozilla::detail::RunnableFunctionImpl (mozilla::Forward(aFunction))); } template already_AddRefed NS_NewRunnableFunction(const char* aName, Function&& aFunction) { return mozilla::detail::SetRunnableName( NS_NewRunnableFunction(mozilla::Forward(aFunction)), aName); } // 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 { public: 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; } }; 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 = 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 = 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 = 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 = 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 = 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 = 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> {}; } /* 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::class_type ClassType; ::nsRunnableMethodReceiver mReceiver; Method mMethod; RunnableMethodArguments mArgs; private: virtual ~RunnableMethodImpl() { Revoke(); }; 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() { if (MOZ_LIKELY(mReceiver.Get())) { mArgs.apply(mReceiver.Get(), mMethod); } return NS_OK; } nsresult Cancel() { static_assert(Cancelable, "Don't use me!"); Revoke(); return NS_OK; } void Revoke() { mReceiver.Revoke(); } }; // Type aliases for NewRunnableMethod. template using OwningRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, true, false>::base_type; template using OwningRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, true, false, Storages...>; // Type aliases for NewCancelableRunnableMethod. template using CancelableRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, true, true>::base_type; template using CancelableRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, true, true, Storages...>; // Type aliases for NewNonOwningRunnableMethod. template using NonOwningRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, false, false>::base_type; template using NonOwningRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, false, false, Storages...>; // Type aliases for NonOwningCancelableRunnableMethod template using NonOwningCancelableRunnableMethod = typename ::nsRunnableMethodTraits< typename RemoveReference::Type, Method, false, true>::base_type; template using NonOwningCancelableRunnableMethodImpl = RunnableMethodImpl< typename RemoveReference::Type, Method, false, true, 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(PtrType&& aPtr, Method aMethod) { return do_AddRef( new detail::OwningRunnableMethodImpl (Forward(aPtr), aMethod)); } template already_AddRefed> NewRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( NewRunnableMethod(Forward(aPtr), aMethod), aName); } template already_AddRefed> NewCancelableRunnableMethod(PtrType&& aPtr, Method aMethod) { return do_AddRef( new detail::CancelableRunnableMethodImpl (Forward(aPtr), aMethod)); } template already_AddRefed> NewCancelableRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( NewCancelableRunnableMethod(Forward(aPtr), aMethod), aName); } template already_AddRefed> NewNonOwningRunnableMethod(PtrType&& aPtr, Method aMethod) { return do_AddRef( new detail::NonOwningRunnableMethodImpl (Forward(aPtr), aMethod)); } template already_AddRefed> NewNonOwningRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( NewNonOwningRunnableMethod(Forward(aPtr), aMethod), aName); } template already_AddRefed> NewNonOwningCancelableRunnableMethod(PtrType&& aPtr, Method aMethod) { return do_AddRef( new detail::NonOwningCancelableRunnableMethodImpl (Forward(aPtr), aMethod)); } template already_AddRefed> NewNonOwningCancelableRunnableMethod(const char* aName, PtrType&& aPtr, Method aMethod) { return detail::SetRunnableName( NewNonOwningCancelableRunnableMethod(Forward(aPtr), aMethod), aName); } // 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(PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return do_AddRef( new detail::OwningRunnableMethodImpl (Forward(aPtr), aMethod, mozilla::Forward(aArgs)...)); } 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( NewRunnableMethod (Forward(aPtr), aMethod, mozilla::Forward(aArgs)...), aName); } template already_AddRefed> NewNonOwningRunnableMethod(PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return do_AddRef( new detail::NonOwningRunnableMethodImpl (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( NewNonOwningRunnableMethod (Forward(aPtr), aMethod, mozilla::Forward(aArgs)...), aName); } template already_AddRefed> NewCancelableRunnableMethod(PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return do_AddRef( new detail::CancelableRunnableMethodImpl (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( NewCancelableRunnableMethod (Forward(aPtr), aMethod, mozilla::Forward(aArgs)...), aName); } template already_AddRefed> NewNonOwningCancelableRunnableMethod(PtrType&& aPtr, Method aMethod, Args&&... aArgs) { static_assert(sizeof...(Storages) == sizeof...(Args), " size should be equal to number of arguments"); return do_AddRef( new detail::NonOwningCancelableRunnableMethodImpl (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( NewNonOwningCancelableRunnableMethod (Forward(aPtr), aMethod, mozilla::Forward(aArgs)...), aName); } } // 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(); 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). PRThread* GetCurrentVirtualThread(); } // namespace mozilla #endif // nsThreadUtils_h__