зеркало из https://github.com/mozilla/gecko-dev.git
135 строки
3.1 KiB
C++
135 строки
3.1 KiB
C++
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
||
|
/* 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 "BroadcastChannelService.h"
|
||
|
#include "BroadcastChannelParent.h"
|
||
|
#include "mozilla/ipc/BackgroundParent.h"
|
||
|
|
||
|
namespace mozilla {
|
||
|
|
||
|
using namespace ipc;
|
||
|
|
||
|
namespace dom {
|
||
|
|
||
|
namespace {
|
||
|
|
||
|
BroadcastChannelService* sInstance = nullptr;
|
||
|
|
||
|
} // anonymous namespace
|
||
|
|
||
|
BroadcastChannelService::BroadcastChannelService()
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
|
||
|
// sInstance is a raw BroadcastChannelService*.
|
||
|
MOZ_ASSERT(!sInstance);
|
||
|
sInstance = this;
|
||
|
}
|
||
|
|
||
|
BroadcastChannelService::~BroadcastChannelService()
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
MOZ_ASSERT(sInstance == this);
|
||
|
MOZ_ASSERT(mAgents.Count() == 0);
|
||
|
|
||
|
sInstance = nullptr;
|
||
|
}
|
||
|
|
||
|
// static
|
||
|
already_AddRefed<BroadcastChannelService>
|
||
|
BroadcastChannelService::GetOrCreate()
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
|
||
|
nsRefPtr<BroadcastChannelService> instance = sInstance;
|
||
|
if (!instance) {
|
||
|
instance = new BroadcastChannelService();
|
||
|
}
|
||
|
return instance.forget();
|
||
|
}
|
||
|
|
||
|
void
|
||
|
BroadcastChannelService::RegisterActor(BroadcastChannelParent* aParent)
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
MOZ_ASSERT(aParent);
|
||
|
MOZ_ASSERT(!mAgents.Contains(aParent));
|
||
|
|
||
|
mAgents.PutEntry(aParent);
|
||
|
}
|
||
|
|
||
|
void
|
||
|
BroadcastChannelService::UnregisterActor(BroadcastChannelParent* aParent)
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
MOZ_ASSERT(aParent);
|
||
|
MOZ_ASSERT(mAgents.Contains(aParent));
|
||
|
|
||
|
mAgents.RemoveEntry(aParent);
|
||
|
}
|
||
|
|
||
|
namespace {
|
||
|
|
||
|
struct MOZ_STACK_CLASS PostMessageData MOZ_FINAL
|
||
|
{
|
||
|
PostMessageData(BroadcastChannelParent* aParent,
|
||
|
const nsAString& aMessage,
|
||
|
const nsAString& aOrigin,
|
||
|
const nsAString& aChannel)
|
||
|
: mParent(aParent)
|
||
|
, mMessage(aMessage)
|
||
|
, mOrigin(aOrigin)
|
||
|
, mChannel(aChannel)
|
||
|
{
|
||
|
MOZ_ASSERT(aParent);
|
||
|
MOZ_COUNT_CTOR(PostMessageData);
|
||
|
}
|
||
|
|
||
|
~PostMessageData()
|
||
|
{
|
||
|
MOZ_COUNT_DTOR(PostMessageData);
|
||
|
}
|
||
|
|
||
|
BroadcastChannelParent* mParent;
|
||
|
const nsString mMessage;
|
||
|
const nsString mOrigin;
|
||
|
const nsString mChannel;
|
||
|
};
|
||
|
|
||
|
PLDHashOperator
|
||
|
PostMessageEnumerator(nsPtrHashKey<BroadcastChannelParent>* aKey, void* aPtr)
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
|
||
|
auto* data = static_cast<PostMessageData*>(aPtr);
|
||
|
BroadcastChannelParent* parent = aKey->GetKey();
|
||
|
MOZ_ASSERT(parent);
|
||
|
|
||
|
if (parent != data->mParent) {
|
||
|
parent->CheckAndDeliver(data->mMessage, data->mOrigin, data->mChannel);
|
||
|
}
|
||
|
|
||
|
return PL_DHASH_NEXT;
|
||
|
}
|
||
|
|
||
|
} // anonymous namespace
|
||
|
|
||
|
void
|
||
|
BroadcastChannelService::PostMessage(BroadcastChannelParent* aParent,
|
||
|
const nsAString& aMessage,
|
||
|
const nsAString& aOrigin,
|
||
|
const nsAString& aChannel)
|
||
|
{
|
||
|
AssertIsOnBackgroundThread();
|
||
|
MOZ_ASSERT(aParent);
|
||
|
MOZ_ASSERT(mAgents.Contains(aParent));
|
||
|
|
||
|
PostMessageData data(aParent, aMessage, aOrigin, aChannel);
|
||
|
mAgents.EnumerateEntries(PostMessageEnumerator, &data);
|
||
|
}
|
||
|
|
||
|
} // dom namespace
|
||
|
} // mozilla namespace
|