1 files changed, 68 insertions, 0 deletions
diff --git a/dom/cache/StreamControl.cpp b/dom/cache/StreamControl.cpp
new file mode 100644
index 0000000000..5d6c9a9e0e
--- /dev/null
+++ b/dom/cache/StreamControl.cpp
@@ -0,0 +1,68 @@
+/* -*- 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/. */
+
+#include "mozilla/dom/cache/StreamControl.h"
+
+namespace mozilla::dom::cache {
+
+void StreamControl::AddReadStream(
+    SafeRefPtr<ReadStream::Controllable> aReadStream) {
+  AssertOwningThread();
+  MOZ_DIAGNOSTIC_ASSERT(aReadStream);
+  MOZ_ASSERT(!mReadStreamList.Contains(aReadStream));
+  mReadStreamList.AppendElement(std::move(aReadStream));
+}
+
+void StreamControl::ForgetReadStream(
+    SafeRefPtr<ReadStream::Controllable> aReadStream) {
+  AssertOwningThread();
+  MOZ_ALWAYS_TRUE(mReadStreamList.RemoveElement(aReadStream));
+}
+
+void StreamControl::NoteClosed(SafeRefPtr<ReadStream::Controllable> aReadStream,
+                               const nsID& aId) {
+  AssertOwningThread();
+  ForgetReadStream(std::move(aReadStream));
+  NoteClosedAfterForget(aId);
+}
+
+StreamControl::~StreamControl() {
+  // owning thread only, but can't call virtual AssertOwningThread in destructor
+  MOZ_DIAGNOSTIC_ASSERT(mReadStreamList.IsEmpty());
+}
+
+void StreamControl::CloseAllReadStreams() {
+  AssertOwningThread();
+
+  // A copy of mReadStreamList is necessary here for two reasons:
+  // 1. mReadStreamList is modified in StreamControl::ForgetReadStream (called
+  //    transitively)
+  // 2. the this pointer is deleted by CacheStreamControlParent::Shutdown
+  //    (called transitively)
+  auto readStreamList = mReadStreamList.Clone();
+  for (const auto& stream : readStreamList.ForwardRange()) {
+    stream->CloseStream();
+  }
+}
+
+void StreamControl::CloseAllReadStreamsWithoutReporting() {
+  AssertOwningThread();
+
+  for (const auto& stream : mReadStreamList.ForwardRange()) {
+    // Note, we cannot trigger IPC traffic here.  So use
+    // CloseStreamWithoutReporting().
+    stream->CloseStreamWithoutReporting();
+  }
+}
+
+bool StreamControl::HasEverBeenRead() const {
+  const auto range = mReadStreamList.NonObservingRange();
+  return std::any_of(range.begin(), range.end(), [](const auto& stream) {
+    return stream->HasEverBeenRead();
+  });
+}
+
+}  // namespace mozilla::dom::cache