forked from mirrors/gecko-dev
c8c32ddc29
These registrars can hold channels alive, which can hold listeners alive, which can hold cycle collected things alive. By clearing the registrars before the final CC instead of after the final CC, we can avoid spurious leaks. It would make more logical sense to do this earlier, at net teardown, but I've made the minimal change to try to avoid stirring up weird shutdown issues. Differential Revision: https://phabricator.services.mozilla.com/D177182
99 lines
2.8 KiB
C++
99 lines
2.8 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 "BackgroundChannelRegistrar.h"
|
|
|
|
#include "mozilla/ClearOnShutdown.h"
|
|
#include "mozilla/StaticPtr.h"
|
|
#include "HttpBackgroundChannelParent.h"
|
|
#include "HttpChannelParent.h"
|
|
#include "nsXULAppAPI.h"
|
|
|
|
namespace {
|
|
mozilla::StaticRefPtr<mozilla::net::BackgroundChannelRegistrar> gSingleton;
|
|
}
|
|
|
|
namespace mozilla {
|
|
namespace net {
|
|
|
|
NS_IMPL_ISUPPORTS(BackgroundChannelRegistrar, nsIBackgroundChannelRegistrar)
|
|
|
|
BackgroundChannelRegistrar::BackgroundChannelRegistrar() {
|
|
// BackgroundChannelRegistrar is a main-thread-only object.
|
|
// All the operations should be run on main thread.
|
|
// It should be used on chrome process only.
|
|
MOZ_ASSERT(XRE_IsParentProcess());
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
}
|
|
|
|
BackgroundChannelRegistrar::~BackgroundChannelRegistrar() {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
}
|
|
|
|
// static
|
|
already_AddRefed<nsIBackgroundChannelRegistrar>
|
|
BackgroundChannelRegistrar::GetOrCreate() {
|
|
if (!gSingleton) {
|
|
gSingleton = new BackgroundChannelRegistrar();
|
|
ClearOnShutdown(&gSingleton);
|
|
}
|
|
return do_AddRef(gSingleton);
|
|
}
|
|
|
|
void BackgroundChannelRegistrar::NotifyChannelLinked(
|
|
HttpChannelParent* aChannelParent, HttpBackgroundChannelParent* aBgParent) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(aChannelParent);
|
|
MOZ_ASSERT(aBgParent);
|
|
|
|
aBgParent->LinkToChannel(aChannelParent);
|
|
aChannelParent->OnBackgroundParentReady(aBgParent);
|
|
}
|
|
|
|
// nsIBackgroundChannelRegistrar
|
|
void BackgroundChannelRegistrar::DeleteChannel(uint64_t aKey) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
|
|
mChannels.Remove(aKey);
|
|
mBgChannels.Remove(aKey);
|
|
}
|
|
|
|
void BackgroundChannelRegistrar::LinkHttpChannel(uint64_t aKey,
|
|
HttpChannelParent* aChannel) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(aChannel);
|
|
|
|
RefPtr<HttpBackgroundChannelParent> bgParent;
|
|
bool found = mBgChannels.Remove(aKey, getter_AddRefs(bgParent));
|
|
|
|
if (!found) {
|
|
mChannels.InsertOrUpdate(aKey, RefPtr{aChannel});
|
|
return;
|
|
}
|
|
|
|
MOZ_ASSERT(bgParent);
|
|
NotifyChannelLinked(aChannel, bgParent);
|
|
}
|
|
|
|
void BackgroundChannelRegistrar::LinkBackgroundChannel(
|
|
uint64_t aKey, HttpBackgroundChannelParent* aBgChannel) {
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
MOZ_ASSERT(aBgChannel);
|
|
|
|
RefPtr<HttpChannelParent> parent;
|
|
bool found = mChannels.Remove(aKey, getter_AddRefs(parent));
|
|
|
|
if (!found) {
|
|
mBgChannels.InsertOrUpdate(aKey, RefPtr{aBgChannel});
|
|
return;
|
|
}
|
|
|
|
MOZ_ASSERT(parent);
|
|
NotifyChannelLinked(parent, aBgChannel);
|
|
}
|
|
|
|
} // namespace net
|
|
} // namespace mozilla
|