gecko-dev/xpcom/threads/ThreadEventTarget.h
Andreas Pehrson 9717c27492 Bug 1695580 - In xpcom, cancel pending DelayedRunnable timers on shutdown. r=KrisWright
Because DelayedRunnables are fire-and-forget, there is no way for a targeted
EventTarget to clean them up on shutdown. Thus if a timer fires after
EventTarget shutdown it will fail to dispatch the timer event, and avoid
releasing the timer callback because it's not on the targeted thread. This
causes a leak as there is a ref-cycle between nsTimerImpl::mCallback and
DelayedRunnable::mTimer.

This patch adds nsIDelayedRunnableObserver for a target to observe which
DelayedRunnables are relying on their timer to run them. This allows the target
to schedule a shutdown task to cancel those timers and release the runnables on
the target thread.

Supported DelayedRunnable targets with this patch are TaskQueues,
eventqueue-based nsThreads and XPCOMThreadWrappers that wrap a supported
nsThread.

An assertion makes sure at runtime that future new uses of DelayedRunnable
target nsIDelayedRunnableObserver-supported event targets.

Differential Revision: https://phabricator.services.mozilla.com/D109781
2021-04-06 20:15:11 +00:00

71 lines
2.4 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_ThreadEventTarget_h
#define mozilla_ThreadEventTarget_h
#include "mozilla/MemoryReporting.h"
#include "mozilla/Mutex.h"
#include "mozilla/SynchronizedEventQueue.h" // for ThreadTargetSink
#include "nsIDelayedRunnableObserver.h"
#include "nsISerialEventTarget.h"
namespace mozilla {
class DelayedRunnable;
// ThreadEventTarget handles the details of posting an event to a thread. It can
// be used with any ThreadTargetSink implementation.
class ThreadEventTarget final : public nsISerialEventTarget,
public nsIDelayedRunnableObserver {
public:
ThreadEventTarget(ThreadTargetSink* aSink, bool aIsMainThread);
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIEVENTTARGET_FULL
// nsIDelayedRunnableObserver
void OnDelayedRunnableCreated(DelayedRunnable* aRunnable) override;
void OnDelayedRunnableScheduled(DelayedRunnable* aRunnable) override;
void OnDelayedRunnableRan(DelayedRunnable* aRunnable) override;
// Notification from, and on, the owner thread that it is shutting down.
// Cancels any scheduled DelayedRunnables.
void NotifyShutdown();
// Disconnects the target so that it can no longer post events.
void Disconnect(const MutexAutoLock& aProofOfLock) {
mSink->Disconnect(aProofOfLock);
}
// Sets the thread for which IsOnCurrentThread returns true to the current
// thread.
void SetCurrentThread(PRThread* aThread);
// Call ClearCurrentThread() before the PRThread is deleted on thread join.
void ClearCurrentThread();
size_t SizeOfIncludingThis(mozilla::MallocSizeOf aMallocSizeOf) const {
size_t n = 0;
if (mSink) {
n += mSink->SizeOfIncludingThis(aMallocSizeOf);
}
return aMallocSizeOf(this) + n;
}
private:
~ThreadEventTarget();
RefPtr<ThreadTargetSink> mSink;
bool mIsMainThread;
// DelayedRunnables (from DelayedDispatch) that are managed by their
// respective timers, but have not yet run. Accessed only on this nsThread.
nsTArray<RefPtr<mozilla::DelayedRunnable>> mScheduledDelayedRunnables;
};
} // namespace mozilla
#endif // mozilla_ThreadEventTarget_h