Source code

Revision control

Other Tools

/* 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 "SessionStorageService.h"
#include "mozilla/Services.h"
#include "mozilla/Unused.h"
#include "nsIObserverService.h"
#include "nsXULAppAPI.h"
#include <cstring>
namespace mozilla {
namespace dom {
namespace {
const char* const kContentProcessShutdownTopic = "content-child-will-shutdown";
}
RefPtr<SessionStorageService> SessionStorageService::sService = nullptr;
bool SessionStorageService::sShutdown = false;
NS_IMPL_ISUPPORTS(SessionStorageService, nsIObserver)
SessionStorageService::SessionStorageService() {
if (const nsCOMPtr<nsIObserverService> observerService =
services::GetObserverService()) {
Unused << observerService->AddObserver(this, kContentProcessShutdownTopic,
false);
}
}
SessionStorageService::~SessionStorageService() {
if (const nsCOMPtr<nsIObserverService> observerService =
services::GetObserverService()) {
Unused << observerService->RemoveObserver(this,
kContentProcessShutdownTopic);
}
}
SessionStorageService* SessionStorageService::Get() {
if (sShutdown) {
return nullptr;
}
if (XRE_IsParentProcess()) {
ShutDown();
return nullptr;
}
if (!sService) {
sService = new SessionStorageService();
}
return sService;
}
NS_IMETHODIMP SessionStorageService::Observe(nsISupports* const aSubject,
const char* const aTopic,
const char16_t* const aData) {
if (std::strcmp(aTopic, kContentProcessShutdownTopic) == 0) {
SendSessionStorageDataToParentProcess();
ShutDown();
}
return NS_OK;
}
void SessionStorageService::RegisterSessionStorageManager(
SessionStorageManager* aManager) {
mManagers.PutEntry(aManager);
}
void SessionStorageService::UnregisterSessionStorageManager(
SessionStorageManager* aManager) {
if (const auto entry = mManagers.GetEntry(aManager)) {
mManagers.RemoveEntry(entry);
}
}
void SessionStorageService::SendSessionStorageDataToParentProcess() {
for (auto iter = mManagers.Iter(); !iter.Done(); iter.Next()) {
iter.Get()->GetKey()->SendSessionStorageDataToParentProcess();
}
}
void SessionStorageService::ShutDown() {
sShutdown = true;
sService = nullptr;
}
} // namespace dom
} // namespace mozilla