summaryrefslogtreecommitdiffstats
path: root/toolkit/components/extensions/webrequest/WebRequestService.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'toolkit/components/extensions/webrequest/WebRequestService.cpp')
-rw-r--r--toolkit/components/extensions/webrequest/WebRequestService.cpp55
1 files changed, 55 insertions, 0 deletions
diff --git a/toolkit/components/extensions/webrequest/WebRequestService.cpp b/toolkit/components/extensions/webrequest/WebRequestService.cpp
new file mode 100644
index 0000000000..7ec2433ac5
--- /dev/null
+++ b/toolkit/components/extensions/webrequest/WebRequestService.cpp
@@ -0,0 +1,55 @@
+/* -*- 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"
+
+using namespace mozilla;
+using namespace mozilla::dom;
+using namespace mozilla::extensions;
+
+static StaticRefPtr<WebRequestService> sWebRequestService;
+
+/* static */ WebRequestService& WebRequestService::GetSingleton() {
+ if (!sWebRequestService) {
+ sWebRequestService = new WebRequestService();
+ ClearOnShutdown(&sWebRequestService);
+ }
+ return *sWebRequestService;
+}
+
+UniquePtr<WebRequestChannelEntry> WebRequestService::RegisterChannel(
+ ChannelWrapper* aChannel) {
+ UniquePtr<ChannelEntry> entry(new ChannelEntry(aChannel));
+
+ mChannelEntries.WithEntryHandle(entry->mChannelId, [&](auto&& key) {
+ MOZ_DIAGNOSTIC_ASSERT(!key);
+ key.Insert(entry.get());
+ });
+
+ return entry;
+}
+
+already_AddRefed<nsITraceableChannel> WebRequestService::GetTraceableChannel(
+ uint64_t aChannelId, nsAtom* aAddonId, ContentParent* 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 (sWebRequestService) {
+ sWebRequestService->mChannelEntries.Remove(mChannelId);
+ }
+}