1 files changed, 94 insertions, 0 deletions
diff --git a/widget/windows/filedialog/WinFileDialogParent.cpp b/widget/windows/filedialog/WinFileDialogParent.cpp
new file mode 100644
index 0000000000..2c256a1506
--- /dev/null
+++ b/widget/windows/filedialog/WinFileDialogParent.cpp
@@ -0,0 +1,94 @@
+/* -*- 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/widget/filedialog/WinFileDialogParent.h"
+
+#include "mozilla/Logging.h"
+#include "mozilla/Result.h"
+#include "mozilla/SpinEventLoopUntil.h"
+#include "mozilla/ipc/UtilityProcessManager.h"
+#include "nsISupports.h"
+
+namespace mozilla::widget::filedialog {
+
+// Count of currently-open file dialogs (not just open-file dialogs).
+static size_t sOpenDialogActors = 0;
+
+WinFileDialogParent::WinFileDialogParent() {
+  MOZ_LOG(sLogFileDialog, LogLevel::Debug,
+          ("%s %p", __PRETTY_FUNCTION__, this));
+}
+
+WinFileDialogParent::~WinFileDialogParent() {
+  MOZ_LOG(sLogFileDialog, LogLevel::Debug,
+          ("%s %p", __PRETTY_FUNCTION__, this));
+}
+
+PWinFileDialogParent::nsresult WinFileDialogParent::BindToUtilityProcess(
+    mozilla::ipc::UtilityProcessParent* aUtilityParent) {
+  Endpoint<PWinFileDialogParent> parentEnd;
+  Endpoint<PWinFileDialogChild> childEnd;
+  nsresult rv = PWinFileDialog::CreateEndpoints(base::GetCurrentProcId(),
+                                                aUtilityParent->OtherPid(),
+                                                &parentEnd, &childEnd);
+
+  if (NS_FAILED(rv)) {
+    MOZ_ASSERT(false, "Protocol endpoints failure");
+    return NS_ERROR_FAILURE;
+  }
+
+  if (!aUtilityParent->SendStartWinFileDialogService(std::move(childEnd))) {
+    MOZ_ASSERT(false, "SendStartWinFileDialogService failed");
+    return NS_ERROR_FAILURE;
+  }
+
+  if (!parentEnd.Bind(this)) {
+    MOZ_ASSERT(false, "parentEnd.Bind failed");
+    return NS_ERROR_FAILURE;
+  }
+
+  sOpenDialogActors++;
+  return NS_OK;
+}
+
+void WinFileDialogParent::ProcessingError(Result aCode, const char* aReason) {
+  detail::LogProcessingError(sLogFileDialog, this, aCode, aReason);
+}
+
+ProcessProxy::ProcessProxy(RefPtr<WFDP>&& obj)
+    : data(MakeRefPtr<Contents>(std::move(obj))) {}
+
+ProcessProxy::Contents::Contents(RefPtr<WFDP>&& obj) : ptr(std::move(obj)) {}
+
+ProcessProxy::Contents::~Contents() {
+  AssertIsOnMainThread();
+
+  // destroy the actor...
+  ptr->Close();
+
+  // ... and possibly the process
+  if (!--sOpenDialogActors) {
+    StopProcess();
+  }
+}
+
+void ProcessProxy::Contents::StopProcess() {
+  auto const upm = ipc::UtilityProcessManager::GetSingleton();
+  if (!upm) {
+    // This is only possible when the UtilityProcessManager has shut down -- in
+    // which case the file-dialog process has also already been directed to shut
+    // down, and there's nothing we need to do here.
+    return;
+  }
+
+  MOZ_LOG(sLogFileDialog, LogLevel::Debug,
+          ("%s: killing the WINDOWS_FILE_DIALOG process (no more live "
+           "actors)",
+           __PRETTY_FUNCTION__));
+  upm->CleanShutdown(ipc::SandboxingKind::WINDOWS_FILE_DIALOG);
+}
+
+}  // namespace mozilla::widget::filedialog