1 files changed, 135 insertions, 0 deletions
diff --git a/gfx/layers/ipc/SharedSurfacesParent.h b/gfx/layers/ipc/SharedSurfacesParent.h
new file mode 100644
index 0000000000..1ccc99b2f0
--- /dev/null
+++ b/gfx/layers/ipc/SharedSurfacesParent.h
@@ -0,0 +1,135 @@
+/* -*- 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_GFX_SHAREDSURFACESPARENT_H
+#define MOZILLA_GFX_SHAREDSURFACESPARENT_H
+
+#include <stdint.h>                         // for uint32_t
+#include "mozilla/Attributes.h"             // for override
+#include "mozilla/StaticMonitor.h"          // for StaticMutex
+#include "mozilla/StaticPtr.h"              // for StaticAutoPtr
+#include "mozilla/RefPtr.h"                 // for already_AddRefed
+#include "mozilla/ipc/SharedMemory.h"       // for SharedMemory, etc
+#include "mozilla/gfx/2D.h"                 // for SurfaceFormat
+#include "mozilla/gfx/Point.h"              // for IntSize
+#include "mozilla/layers/LayersSurfaces.h"  // for SurfaceDescriptorShared
+#include "mozilla/layers/SourceSurfaceSharedData.h"
+#include "mozilla/webrender/WebRenderTypes.h"  // for wr::ExternalImageId
+#include "nsExpirationTracker.h"
+#include "nsRefPtrHashtable.h"
+
+namespace mozilla {
+namespace gfx {
+class DataSourceSurface;
+}  // namespace gfx
+
+namespace layers {
+
+class SharedSurfacesChild;
+class SharedSurfacesMemoryReport;
+
+class SharedSurfacesParent final {
+ public:
+  static void Initialize();
+  static void ShutdownRenderThread();
+  static void Shutdown();
+
+  // Get without increasing the consumer count.
+  static already_AddRefed<gfx::DataSourceSurface> Get(
+      const wr::ExternalImageId& aId);
+
+  // Get but also increase the consumer count. Must call Release after finished.
+  static already_AddRefed<gfx::DataSourceSurface> Acquire(
+      const wr::ExternalImageId& aId);
+
+  static bool Release(const wr::ExternalImageId& aId, bool aForCreator = false);
+
+  static void Add(const wr::ExternalImageId& aId,
+                  SurfaceDescriptorShared&& aDesc, base::ProcessId aPid);
+
+  static void Remove(const wr::ExternalImageId& aId);
+
+  static void DestroyProcess(base::ProcessId aPid);
+
+  static void AccumulateMemoryReport(base::ProcessId aPid,
+                                     SharedSurfacesMemoryReport& aReport);
+
+  static bool AccumulateMemoryReport(SharedSurfacesMemoryReport& aReport);
+
+  static void AddTracking(gfx::SourceSurfaceSharedDataWrapper* aSurface);
+
+  static void RemoveTracking(gfx::SourceSurfaceSharedDataWrapper* aSurface);
+
+  static bool AgeOneGeneration(
+      nsTArray<RefPtr<gfx::SourceSurfaceSharedDataWrapper>>& aExpired);
+
+  static bool AgeAndExpireOneGeneration();
+
+ private:
+  friend class SharedSurfacesChild;
+  friend class gfx::SourceSurfaceSharedDataWrapper;
+
+  SharedSurfacesParent();
+
+  static void AddSameProcess(const wr::ExternalImageId& aId,
+                             gfx::SourceSurfaceSharedData* aSurface);
+
+  static void AddTrackingLocked(gfx::SourceSurfaceSharedDataWrapper* aSurface,
+                                const StaticMonitorAutoLock& aAutoLock);
+
+  static void RemoveTrackingLocked(
+      gfx::SourceSurfaceSharedDataWrapper* aSurface,
+      const StaticMonitorAutoLock& aAutoLock);
+
+  static bool AgeOneGenerationLocked(
+      nsTArray<RefPtr<gfx::SourceSurfaceSharedDataWrapper>>& aExpired,
+      const StaticMonitorAutoLock& aAutoLock);
+
+  static void ExpireMap(
+      nsTArray<RefPtr<gfx::SourceSurfaceSharedDataWrapper>>& aExpired);
+
+  static StaticMonitor sMonitor MOZ_UNANNOTATED;
+
+  static StaticAutoPtr<SharedSurfacesParent> sInstance;
+
+  nsRefPtrHashtable<nsUint64HashKey, gfx::SourceSurfaceSharedDataWrapper>
+      mSurfaces;
+
+  class MappingTracker final
+      : public ExpirationTrackerImpl<gfx::SourceSurfaceSharedDataWrapper, 4,
+                                     StaticMonitor, StaticMonitorAutoLock> {
+   public:
+    explicit MappingTracker(uint32_t aExpirationTimeoutMS,
+                            nsIEventTarget* aEventTarget)
+        : ExpirationTrackerImpl<gfx::SourceSurfaceSharedDataWrapper, 4,
+                                StaticMonitor, StaticMonitorAutoLock>(
+              aExpirationTimeoutMS, "SharedMappingTracker", aEventTarget) {}
+
+    void TakeExpired(
+        nsTArray<RefPtr<gfx::SourceSurfaceSharedDataWrapper>>& aExpired,
+        const StaticMonitorAutoLock& aAutoLock);
+
+   protected:
+    void NotifyExpiredLocked(gfx::SourceSurfaceSharedDataWrapper* aSurface,
+                             const StaticMonitorAutoLock& aAutoLock) override;
+
+    void NotifyHandlerEndLocked(
+        const StaticMonitorAutoLock& aAutoLock) override {}
+
+    void NotifyHandlerEnd() override;
+
+    StaticMonitor& GetMutex() override { return sMonitor; }
+
+    nsTArray<RefPtr<gfx::SourceSurfaceSharedDataWrapper>> mExpired;
+  };
+
+  MappingTracker mTracker;
+};
+
+}  // namespace layers
+}  // namespace mozilla
+
+#endif