forked from mirrors/gecko-dev
142 lines
4.3 KiB
C++
142 lines
4.3 KiB
C++
/* -*- 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 "ContentProcessManager.h"
|
|
#include "ContentParent.h"
|
|
#include "mozilla/AppShutdown.h"
|
|
#include "mozilla/dom/BrowserParent.h"
|
|
#include "mozilla/dom/BrowsingContextGroup.h"
|
|
#include "mozilla/dom/CanonicalBrowsingContext.h"
|
|
|
|
#include "mozilla/StaticPtr.h"
|
|
#include "mozilla/ClearOnShutdown.h"
|
|
|
|
#include "nsPrintfCString.h"
|
|
|
|
namespace mozilla::dom {
|
|
|
|
/* static */
|
|
StaticAutoPtr<ContentProcessManager> ContentProcessManager::sSingleton;
|
|
|
|
/* static */
|
|
ContentProcessManager* ContentProcessManager::GetSingleton() {
|
|
MOZ_ASSERT(XRE_IsParentProcess());
|
|
|
|
if (!sSingleton &&
|
|
!AppShutdown::IsInOrBeyond(ShutdownPhase::XPCOMShutdownFinal)) {
|
|
sSingleton = new ContentProcessManager();
|
|
ClearOnShutdown(&sSingleton);
|
|
}
|
|
return sSingleton;
|
|
}
|
|
|
|
void ContentProcessManager::AddContentProcess(ContentParent* aChildCp) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(aChildCp);
|
|
|
|
mContentParentMap.WithEntryHandle(aChildCp->ChildID(), [&](auto&& entry) {
|
|
MOZ_ASSERT_IF(entry, entry.Data() == aChildCp);
|
|
entry.OrInsert(aChildCp);
|
|
});
|
|
}
|
|
|
|
void ContentProcessManager::RemoveContentProcess(
|
|
const ContentParentId& aChildCpId) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
MOZ_ALWAYS_TRUE(mContentParentMap.Remove(aChildCpId));
|
|
}
|
|
|
|
ContentParent* ContentProcessManager::GetContentProcessById(
|
|
const ContentParentId& aChildCpId) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
ContentParent* contentParent = mContentParentMap.Get(aChildCpId);
|
|
if (NS_WARN_IF(!contentParent)) {
|
|
return nullptr;
|
|
}
|
|
return contentParent;
|
|
}
|
|
|
|
bool ContentProcessManager::RegisterRemoteFrame(BrowserParent* aChildBp) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(aChildBp);
|
|
|
|
return mBrowserParentMap.WithEntryHandle(
|
|
aChildBp->GetTabId(), [&](auto&& entry) {
|
|
if (entry) {
|
|
MOZ_ASSERT(entry.Data() == aChildBp);
|
|
return false;
|
|
}
|
|
|
|
// Ensure that this BrowserParent's BrowsingContextGroup is kept alive
|
|
// until the BrowserParent has been unregistered, ensuring the group
|
|
// isn't destroyed while this BrowserParent can still send messages.
|
|
aChildBp->GetBrowsingContext()->Group()->AddKeepAlive();
|
|
entry.Insert(aChildBp);
|
|
return true;
|
|
});
|
|
}
|
|
|
|
void ContentProcessManager::UnregisterRemoteFrame(const TabId& aChildTabId) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
auto childBp = mBrowserParentMap.Extract(aChildTabId);
|
|
MOZ_DIAGNOSTIC_ASSERT(childBp);
|
|
|
|
// Clear the corresponding keepalive which was added in `RegisterRemoteFrame`.
|
|
(*childBp)->GetBrowsingContext()->Group()->RemoveKeepAlive();
|
|
}
|
|
|
|
ContentParentId ContentProcessManager::GetTabProcessId(
|
|
const TabId& aChildTabId) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
if (BrowserParent* browserParent = mBrowserParentMap.Get(aChildTabId)) {
|
|
return browserParent->Manager()->ChildID();
|
|
}
|
|
return ContentParentId(0);
|
|
}
|
|
|
|
uint32_t ContentProcessManager::GetBrowserParentCountByProcessId(
|
|
const ContentParentId& aChildCpId) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
ContentParent* contentParent = mContentParentMap.Get(aChildCpId);
|
|
if (NS_WARN_IF(!contentParent)) {
|
|
return 0;
|
|
}
|
|
return contentParent->ManagedPBrowserParent().Count();
|
|
}
|
|
|
|
already_AddRefed<BrowserParent>
|
|
ContentProcessManager::GetBrowserParentByProcessAndTabId(
|
|
const ContentParentId& aChildCpId, const TabId& aChildTabId) {
|
|
RefPtr<BrowserParent> browserParent = mBrowserParentMap.Get(aChildTabId);
|
|
if (NS_WARN_IF(!browserParent)) {
|
|
return nullptr;
|
|
}
|
|
|
|
if (NS_WARN_IF(browserParent->Manager()->ChildID() != aChildCpId)) {
|
|
return nullptr;
|
|
}
|
|
|
|
return browserParent.forget();
|
|
}
|
|
|
|
already_AddRefed<BrowserParent>
|
|
ContentProcessManager::GetTopLevelBrowserParentByProcessAndTabId(
|
|
const ContentParentId& aChildCpId, const TabId& aChildTabId) {
|
|
RefPtr<BrowserParent> browserParent =
|
|
GetBrowserParentByProcessAndTabId(aChildCpId, aChildTabId);
|
|
while (browserParent && browserParent->GetBrowserBridgeParent()) {
|
|
browserParent = browserParent->GetBrowserBridgeParent()->Manager();
|
|
}
|
|
|
|
return browserParent.forget();
|
|
}
|
|
|
|
} // namespace mozilla::dom
|