1 files changed, 115 insertions, 0 deletions
diff --git a/dom/clients/manager/ClientHandleParent.cpp b/dom/clients/manager/ClientHandleParent.cpp
new file mode 100644
index 0000000000..c3432dc2b4
--- /dev/null
+++ b/dom/clients/manager/ClientHandleParent.cpp
@@ -0,0 +1,115 @@
+/* -*- 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 "ClientHandleParent.h"
+
+#include "ClientHandleOpParent.h"
+#include "ClientManagerService.h"
+#include "ClientPrincipalUtils.h"
+#include "ClientSourceParent.h"
+#include "mozilla/dom/ClientIPCTypes.h"
+#include "mozilla/Unused.h"
+
+namespace mozilla::dom {
+
+using mozilla::ipc::IPCResult;
+
+IPCResult ClientHandleParent::RecvTeardown() {
+  Unused << Send__delete__(this);
+  return IPC_OK();
+}
+
+void ClientHandleParent::ActorDestroy(ActorDestroyReason aReason) {
+  if (mSource) {
+    mSource->DetachHandle(this);
+    mSource = nullptr;
+  } else {
+    if (!mSourcePromiseHolder.IsEmpty()) {
+      CopyableErrorResult rv;
+      rv.ThrowAbortError("Client aborted");
+      mSourcePromiseHolder.Reject(rv, __func__);
+    }
+
+    mSourcePromiseRequestHolder.DisconnectIfExists();
+  }
+}
+
+PClientHandleOpParent* ClientHandleParent::AllocPClientHandleOpParent(
+    const ClientOpConstructorArgs& aArgs) {
+  return new ClientHandleOpParent();
+}
+
+bool ClientHandleParent::DeallocPClientHandleOpParent(
+    PClientHandleOpParent* aActor) {
+  delete aActor;
+  return true;
+}
+
+IPCResult ClientHandleParent::RecvPClientHandleOpConstructor(
+    PClientHandleOpParent* aActor, const ClientOpConstructorArgs& aArgs) {
+  auto actor = static_cast<ClientHandleOpParent*>(aActor);
+  actor->Init(std::move(const_cast<ClientOpConstructorArgs&>(aArgs)));
+  return IPC_OK();
+}
+
+ClientHandleParent::ClientHandleParent()
+    : mService(ClientManagerService::GetOrCreateInstance()), mSource(nullptr) {}
+
+ClientHandleParent::~ClientHandleParent() { MOZ_DIAGNOSTIC_ASSERT(!mSource); }
+
+void ClientHandleParent::Init(const IPCClientInfo& aClientInfo) {
+  mClientId = aClientInfo.id();
+  mPrincipalInfo = aClientInfo.principalInfo();
+
+  // Callbacks are disconnected in ActorDestroy, so capturing `this` is safe.
+  mService->FindSource(aClientInfo.id(), aClientInfo.principalInfo())
+      ->Then(
+          GetCurrentSerialEventTarget(), __func__,
+          [self = RefPtr{this}](bool) {
+            self->mSourcePromiseRequestHolder.Complete();
+            ClientSourceParent* source = self->mService->FindExistingSource(
+                self->mClientId, self->mPrincipalInfo);
+            if (source) {
+              self->FoundSource(source);
+            }
+          },
+          [self = RefPtr{this}](const CopyableErrorResult&) {
+            self->mSourcePromiseRequestHolder.Complete();
+            Unused << Send__delete__(self);
+          })
+      ->Track(mSourcePromiseRequestHolder);
+}
+
+ClientSourceParent* ClientHandleParent::GetSource() const { return mSource; }
+
+RefPtr<SourcePromise> ClientHandleParent::EnsureSource() {
+  if (mSource) {
+    return SourcePromise::CreateAndResolve(mSource, __func__);
+  }
+
+  return mSourcePromiseHolder.Ensure(__func__);
+}
+
+void ClientHandleParent::FoundSource(ClientSourceParent* aSource) {
+  MOZ_ASSERT(aSource);
+  MOZ_ASSERT(aSource->Info().Id() == mClientId);
+  if (!ClientMatchPrincipalInfo(aSource->Info().PrincipalInfo(),
+                                mPrincipalInfo)) {
+    if (mSourcePromiseHolder.IsEmpty()) {
+      CopyableErrorResult rv;
+      rv.ThrowAbortError("Client aborted");
+      mSourcePromiseHolder.Reject(rv, __func__);
+    }
+    Unused << Send__delete__(this);
+    return;
+  }
+
+  mSource = aSource;
+  mSource->AttachHandle(this);
+  mSourcePromiseHolder.ResolveIfExists(true, __func__);
+}
+
+}  // namespace mozilla::dom