1 files changed, 57 insertions, 0 deletions
diff --git a/gfx/ipc/CanvasShutdownManager.h b/gfx/ipc/CanvasShutdownManager.h
new file mode 100644
index 0000000000..803d6a1eb8
--- /dev/null
+++ b/gfx/ipc/CanvasShutdownManager.h
@@ -0,0 +1,57 @@
+/* -*- 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 _include_gfx_ipc_CanvasShutdownManager_h__
+#define _include_gfx_ipc_CanvasShutdownManager_h__
+
+#include "mozilla/RefPtr.h"
+#include "mozilla/StaticMutex.h"
+#include "mozilla/ThreadLocal.h"
+#include <set>
+
+namespace mozilla {
+namespace dom {
+class CanvasRenderingContext2D;
+class StrongWorkerRef;
+class ThreadSafeWorkerRef;
+}  // namespace dom
+
+namespace gfx {
+
+class CanvasShutdownManager final {
+ public:
+  static CanvasShutdownManager* Get();
+  static CanvasShutdownManager* MaybeGet();
+  static void Shutdown();
+
+  dom::ThreadSafeWorkerRef* GetWorkerRef() const { return mWorkerRef; }
+  void AddShutdownObserver(dom::CanvasRenderingContext2D* aCanvas);
+  void RemoveShutdownObserver(dom::CanvasRenderingContext2D* aCanvas);
+
+  static void OnCompositorManagerRestored();
+
+  void OnRemoteCanvasLost();
+  void OnRemoteCanvasRestored();
+
+ private:
+  explicit CanvasShutdownManager(dom::StrongWorkerRef* aWorkerRef);
+  CanvasShutdownManager();
+  ~CanvasShutdownManager();
+  void Destroy();
+
+  static void MaybeRestoreRemoteCanvas();
+
+  RefPtr<dom::ThreadSafeWorkerRef> mWorkerRef;
+  std::set<dom::CanvasRenderingContext2D*> mActiveCanvas;
+  static MOZ_THREAD_LOCAL(CanvasShutdownManager*) sLocalManager;
+
+  static StaticMutex sManagersMutex;
+  static std::set<CanvasShutdownManager*> sManagers;
+};
+
+}  // namespace gfx
+}  // namespace mozilla
+
+#endif  // _include_gfx_ipc_CanvasShutdownManager_h__