Source code
Revision control
Copy as Markdown
Other Tools
/* -*- 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,
#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