fune/toolkit/components/extensions/webrequest/WebRequestService.cpp
Nicholas Nethercote d225f7151b Bug 1400460 - Rename nsIAtom as nsAtom. r=hiro.
(Path is actually r=froydnj.)

Bug 1400459 devirtualized nsIAtom so that it is no longer a subclass of
nsISupports. This means that nsAtom is now a better name for it than nsIAtom.

MozReview-Commit-ID: 91U22X2NydP

--HG--
rename : xpcom/ds/nsIAtom.h => xpcom/ds/nsAtom.h
extra : rebase_source : ac3e904a21b8b48e74534fff964f1623ee937c67
2017-10-03 09:05:19 +11:00

78 lines
2.2 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 "WebRequestService.h"
#include "mozilla/Assertions.h"
#include "mozilla/ClearOnShutdown.h"
#include "nsIChannel.h"
using namespace mozilla;
using namespace mozilla::dom;
using namespace mozilla::extensions;
static WebRequestService* sWeakWebRequestService;
WebRequestService::~WebRequestService()
{
sWeakWebRequestService = nullptr;
}
/* static */ WebRequestService&
WebRequestService::GetSingleton()
{
static RefPtr<WebRequestService> instance;
if (!sWeakWebRequestService) {
instance = new WebRequestService();
ClearOnShutdown(&instance);
// A separate weak instance that we keep a reference to as long as the
// original service is alive, even after our strong reference is cleared to
// allow the service to be destroyed.
sWeakWebRequestService = instance;
}
return *sWeakWebRequestService;
}
UniquePtr<WebRequestChannelEntry>
WebRequestService::RegisterChannel(ChannelWrapper* aChannel)
{
UniquePtr<ChannelEntry> entry(new ChannelEntry(aChannel));
auto key = mChannelEntries.LookupForAdd(entry->mChannelId);
MOZ_DIAGNOSTIC_ASSERT(!key);
key.OrInsert([&entry]() { return entry.get(); });
return Move(entry);
}
already_AddRefed<nsITraceableChannel>
WebRequestService::GetTraceableChannel(uint64_t aChannelId,
nsAtom* aAddonId,
nsIContentParent* aContentParent)
{
if (auto entry = mChannelEntries.Get(aChannelId)) {
if (entry->mChannel) {
return entry->mChannel->GetTraceableChannel(aAddonId, aContentParent);
}
}
return nullptr;
}
WebRequestChannelEntry::WebRequestChannelEntry(ChannelWrapper* aChannel)
: mChannelId(aChannel->Id())
, mChannel(aChannel)
{}
WebRequestChannelEntry::~WebRequestChannelEntry()
{
if (sWeakWebRequestService) {
sWeakWebRequestService->mChannelEntries.Remove(mChannelId);
}
}