#ifndef builtin_WeakMapObject_inl_h
#define builtin_WeakMapObject_inl_h
#include "builtin/WeakMapObject.h"
#include "vm/ProxyObject.h"
#include "gc/WeakMap-inl.h"
namespace js {
static bool TryPreserveReflector(JSContext* cx, HandleObject obj) {
if (obj->getClass()->isWrappedNative() || obj->getClass()->isDOMClass() ||
(obj->is<ProxyObject>() && obj->as<ProxyObject>().handler()->family() ==
GetDOMProxyHandlerFamily())) {
MOZ_ASSERT(cx->runtime()->preserveWrapperCallback);
if (!cx->runtime()->preserveWrapperCallback(cx, obj)) {
JS_ReportErrorNumberASCII(cx, GetErrorMessage, nullptr,
JSMSG_BAD_WEAKMAP_KEY);
return false;
}
}
return true;
}
static MOZ_ALWAYS_INLINE bool WeakCollectionPutEntryInternal(
JSContext* cx, Handle<WeakCollectionObject*> obj, HandleObject key,
HandleValue value) {
ObjectValueMap* map = obj->getMap();
if (!map) {
auto newMap = cx->make_unique<ObjectValueMap>(cx, obj.get());
if (!newMap) {
return false;
}
map = newMap.release();
obj->setPrivate(map);
}
if (!TryPreserveReflector(cx, key)) {
return false;
}
RootedObject delegate(cx, UncheckedUnwrapWithoutExpose(key));
if (delegate && !TryPreserveReflector(cx, delegate)) {
return false;
}
MOZ_ASSERT(key->compartment() == obj->compartment());
MOZ_ASSERT_IF(value.isObject(),
value.toObject().compartment() == obj->compartment());
if (!map->put(key, value)) {
JS_ReportOutOfMemory(cx);
return false;
}
return true;
}
}
#endif