mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-13 18:27:35 +00:00
2ae60bead7
Bug 1723674 added a new nsID::GenerateUUID() static factory function to generate UUIDs without the overhead of querying and instantiating an nsIUUIDGenerator object. nsContentUtils::GenerateUUID() is a utility function that amortizes that overhead by holding an nsIUUIDGenerator singleton. That's no longer necessary because code that calls nsContentUtils::GenerateUUID() can now just call nsID::GenerateUUID(). No nsIUUDGenerator is needed. Differential Revision: https://phabricator.services.mozilla.com/D132866
162 lines
4.5 KiB
C++
162 lines
4.5 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 "RefMessageBodyService.h"
|
|
|
|
#include <cstdint>
|
|
#include <cstdlib>
|
|
#include "mozilla/ErrorResult.h"
|
|
#include "mozilla/RefPtr.h"
|
|
#include "mozilla/dom/ipc/StructuredCloneData.h"
|
|
#include "nsBaseHashtable.h"
|
|
#include "nsDebug.h"
|
|
|
|
namespace mozilla::dom {
|
|
|
|
// Guards sService and its members.
|
|
StaticMutex sRefMessageBodyServiceMutex;
|
|
|
|
// Raw pointer because the service is kept alive by other objects.
|
|
// See the CTOR and the DTOR of this object.
|
|
RefMessageBodyService* sService;
|
|
|
|
// static
|
|
already_AddRefed<RefMessageBodyService> RefMessageBodyService::GetOrCreate() {
|
|
StaticMutexAutoLock lock(sRefMessageBodyServiceMutex);
|
|
|
|
RefPtr<RefMessageBodyService> service = GetOrCreateInternal(lock);
|
|
return service.forget();
|
|
}
|
|
|
|
// static
|
|
RefMessageBodyService* RefMessageBodyService::GetOrCreateInternal(
|
|
const StaticMutexAutoLock& aProofOfLock) {
|
|
if (!sService) {
|
|
sService = new RefMessageBodyService(aProofOfLock);
|
|
}
|
|
return sService;
|
|
}
|
|
|
|
RefMessageBodyService::RefMessageBodyService(
|
|
const StaticMutexAutoLock& aProofOfLock) {
|
|
MOZ_DIAGNOSTIC_ASSERT(sService == nullptr);
|
|
}
|
|
|
|
RefMessageBodyService::~RefMessageBodyService() {
|
|
StaticMutexAutoLock lock(sRefMessageBodyServiceMutex);
|
|
MOZ_DIAGNOSTIC_ASSERT(sService == this);
|
|
sService = nullptr;
|
|
}
|
|
|
|
const nsID RefMessageBodyService::Register(
|
|
already_AddRefed<RefMessageBody> aBody, ErrorResult& aRv) {
|
|
RefPtr<RefMessageBody> body = aBody;
|
|
MOZ_ASSERT(body);
|
|
|
|
nsID uuid = {};
|
|
aRv = nsID::GenerateUUIDInPlace(uuid);
|
|
if (NS_WARN_IF(aRv.Failed())) {
|
|
return nsID();
|
|
}
|
|
|
|
StaticMutexAutoLock lock(sRefMessageBodyServiceMutex);
|
|
GetOrCreateInternal(lock)->mMessages.InsertOrUpdate(uuid, std::move(body));
|
|
return uuid;
|
|
}
|
|
|
|
already_AddRefed<RefMessageBody> RefMessageBodyService::Steal(const nsID& aID) {
|
|
StaticMutexAutoLock lock(sRefMessageBodyServiceMutex);
|
|
if (!sService) {
|
|
return nullptr;
|
|
}
|
|
|
|
RefPtr<RefMessageBody> body;
|
|
sService->mMessages.Remove(aID, getter_AddRefs(body));
|
|
|
|
return body.forget();
|
|
}
|
|
|
|
already_AddRefed<RefMessageBody> RefMessageBodyService::GetAndCount(
|
|
const nsID& aID) {
|
|
StaticMutexAutoLock lock(sRefMessageBodyServiceMutex);
|
|
if (!sService) {
|
|
return nullptr;
|
|
}
|
|
|
|
RefPtr<RefMessageBody> body = sService->mMessages.Get(aID);
|
|
if (!body) {
|
|
return nullptr;
|
|
}
|
|
|
|
++body->mCount;
|
|
|
|
MOZ_ASSERT_IF(body->mMaxCount.isSome(),
|
|
body->mCount <= body->mMaxCount.value());
|
|
if (body->mMaxCount.isSome() && body->mCount >= body->mMaxCount.value()) {
|
|
sService->mMessages.Remove(aID);
|
|
}
|
|
|
|
return body.forget();
|
|
}
|
|
|
|
void RefMessageBodyService::SetMaxCount(const nsID& aID, uint32_t aMaxCount) {
|
|
StaticMutexAutoLock lock(sRefMessageBodyServiceMutex);
|
|
if (!sService) {
|
|
return;
|
|
}
|
|
|
|
RefPtr<RefMessageBody> body = sService->mMessages.Get(aID);
|
|
if (!body) {
|
|
return;
|
|
}
|
|
|
|
MOZ_ASSERT(body->mMaxCount.isNothing());
|
|
body->mMaxCount.emplace(aMaxCount);
|
|
|
|
MOZ_ASSERT(body->mCount <= body->mMaxCount.value());
|
|
if (body->mCount >= body->mMaxCount.value()) {
|
|
sService->mMessages.Remove(aID);
|
|
}
|
|
}
|
|
|
|
void RefMessageBodyService::ForgetPort(const nsID& aPortID) {
|
|
StaticMutexAutoLock lock(sRefMessageBodyServiceMutex);
|
|
if (!sService) {
|
|
return;
|
|
}
|
|
|
|
for (auto iter = sService->mMessages.Iter(); !iter.Done(); iter.Next()) {
|
|
if (iter.UserData()->PortID() == aPortID) {
|
|
iter.Remove();
|
|
}
|
|
}
|
|
}
|
|
|
|
RefMessageBody::RefMessageBody(const nsID& aPortID,
|
|
UniquePtr<ipc::StructuredCloneData>&& aCloneData)
|
|
: mPortID(aPortID),
|
|
mMutex("RefMessageBody::mMutex"),
|
|
mCloneData(std::move(aCloneData)),
|
|
mMaxCount(Nothing()),
|
|
mCount(0) {}
|
|
|
|
RefMessageBody::~RefMessageBody() = default;
|
|
|
|
void RefMessageBody::Read(JSContext* aCx, JS::MutableHandle<JS::Value> aValue,
|
|
const JS::CloneDataPolicy& aCloneDataPolicy,
|
|
ErrorResult& aRv) {
|
|
MutexAutoLock lock(mMutex);
|
|
mCloneData->Read(aCx, aValue, aCloneDataPolicy, aRv);
|
|
}
|
|
|
|
bool RefMessageBody::TakeTransferredPortsAsSequence(
|
|
Sequence<OwningNonNull<mozilla::dom::MessagePort>>& aPorts) {
|
|
MOZ_ASSERT(mMaxCount.isNothing());
|
|
return mCloneData->TakeTransferredPortsAsSequence(aPorts);
|
|
}
|
|
|
|
} // namespace mozilla::dom
|