From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 19 Apr 2024 02:47:55 +0200 Subject: Adding upstream version 124.0.1. Signed-off-by: Daniel Baumann --- js/src/builtin/WeakMapObject-inl.h | 97 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 97 insertions(+) create mode 100644 js/src/builtin/WeakMapObject-inl.h (limited to 'js/src/builtin/WeakMapObject-inl.h') diff --git a/js/src/builtin/WeakMapObject-inl.h b/js/src/builtin/WeakMapObject-inl.h new file mode 100644 index 0000000000..28ab1abeb2 --- /dev/null +++ b/js/src/builtin/WeakMapObject-inl.h @@ -0,0 +1,97 @@ +/* -*- 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/. */ + +#ifndef builtin_WeakMapObject_inl_h +#define builtin_WeakMapObject_inl_h + +#include "builtin/WeakMapObject.h" + +#include "js/friend/ErrorMessages.h" // js::GetErrorMessage, JSMSG_* +#include "js/Prefs.h" +#include "js/Wrapper.h" +#include "gc/WeakMap-inl.h" +#include "vm/JSObject-inl.h" + +namespace js { + +static bool TryPreserveReflector(JSContext* cx, HandleObject obj) { + if (!MaybePreserveDOMWrapper(cx, obj)) { + JS_ReportErrorNumberASCII(cx, GetErrorMessage, nullptr, + JSMSG_BAD_WEAKMAP_KEY); + return false; + } + + return true; +} + +static MOZ_ALWAYS_INLINE bool WeakCollectionPutEntryInternal( + JSContext* cx, Handle obj, HandleValue key, + HandleValue value) { + ValueValueWeakMap* map = obj->getMap(); + if (!map) { + auto newMap = cx->make_unique(cx, obj.get()); + if (!newMap) { + return false; + } + map = newMap.release(); + InitReservedSlot(obj, WeakCollectionObject::DataSlot, map, + MemoryUse::WeakMapObject); + } + + if (key.isObject()) { + RootedObject keyObj(cx, &key.toObject()); + + // Preserve wrapped native keys to prevent wrapper optimization. + if (!TryPreserveReflector(cx, keyObj)) { + return false; + } + + RootedObject delegate(cx, UncheckedUnwrapWithoutExpose(keyObj)); + if (delegate && !TryPreserveReflector(cx, delegate)) { + return false; + } + } + + MOZ_ASSERT_IF(key.isObject(), + key.toObject().compartment() == obj->compartment()); + MOZ_ASSERT_IF(value.isGCThing(), + gc::ToMarkable(value)->zoneFromAnyThread() == obj->zone() || + gc::ToMarkable(value)->zoneFromAnyThread()->isAtomsZone()); + MOZ_ASSERT_IF(value.isObject(), + value.toObject().compartment() == obj->compartment()); + if (!map->put(key, value)) { + JS_ReportOutOfMemory(cx); + return false; + } + return true; +} + +// https://tc39.es/ecma262/#sec-canbeheldweakly +static MOZ_ALWAYS_INLINE bool CanBeHeldWeakly(JSContext* cx, + HandleValue value) { + // 1. If v is an Object, return true. + if (value.isObject()) { + return true; + } + +#ifdef NIGHTLY_BUILD + bool symbolsAsWeakMapKeysEnabled = + JS::Prefs::experimental_symbols_as_weakmap_keys(); + + // 2. If v is a Symbol and KeyForSymbol(v) is undefined, return true. + if (symbolsAsWeakMapKeysEnabled && value.isSymbol() && + value.toSymbol()->code() != JS::SymbolCode::InSymbolRegistry) { + return true; + } +#endif + + // 3. Return false. + return false; +} + +} // namespace js + +#endif /* builtin_WeakMapObject_inl_h */ -- cgit v1.2.3