mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-06-08 05:27:14 +09:00
285bc005cb
Before this change each built-in iterator object has a boolean `m_next_method_was_redefined`. If user code later changed the iterator’s prototype (e.g. `Object.setPrototypeOf()`), we still believed the built-in fast-path was safe and skipped the user supplied override, producing wrong results. With this change `BuiltinIterator::as_builtin_iterator_if_next_is_not_redefined()` looks up the current `next` property and verifies that it is still the built-in native function.
78 lines
2.1 KiB
C++
78 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2021, Idan Horowitz <idan.horowitz@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <LibJS/Runtime/Array.h>
|
|
#include <LibJS/Runtime/MapIterator.h>
|
|
#include <LibJS/Runtime/MapIteratorPrototype.h>
|
|
#include <LibJS/Runtime/NativeFunction.h>
|
|
|
|
namespace JS {
|
|
|
|
GC_DEFINE_ALLOCATOR(MapIterator);
|
|
|
|
GC::Ref<MapIterator> MapIterator::create(Realm& realm, Map& map, Object::PropertyKind iteration_kind)
|
|
{
|
|
return realm.create<MapIterator>(map, iteration_kind, realm.intrinsics().map_iterator_prototype());
|
|
}
|
|
|
|
MapIterator::MapIterator(Map& map, Object::PropertyKind iteration_kind, Object& prototype)
|
|
: Object(ConstructWithPrototypeTag::Tag, prototype)
|
|
, m_map(map)
|
|
, m_iteration_kind(iteration_kind)
|
|
, m_iterator(static_cast<Map const&>(map).begin())
|
|
{
|
|
}
|
|
|
|
void MapIterator::visit_edges(Cell::Visitor& visitor)
|
|
{
|
|
Base::visit_edges(visitor);
|
|
visitor.visit(m_map);
|
|
}
|
|
|
|
BuiltinIterator* MapIterator::as_builtin_iterator_if_next_is_not_redefined(IteratorRecord const& iterator_record)
|
|
{
|
|
if (iterator_record.next_method.is_object()) {
|
|
auto& next_function = iterator_record.next_method.as_object();
|
|
if (next_function.is_native_function()) {
|
|
auto const& native_function = static_cast<NativeFunction const&>(next_function);
|
|
if (native_function.is_map_prototype_next_builtin())
|
|
return this;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
ThrowCompletionOr<void> MapIterator::next(VM& vm, bool& done, Value& value)
|
|
{
|
|
if (m_done) {
|
|
done = true;
|
|
value = js_undefined();
|
|
return {};
|
|
}
|
|
|
|
if (m_iterator.is_end()) {
|
|
m_done = true;
|
|
done = true;
|
|
value = js_undefined();
|
|
return {};
|
|
}
|
|
|
|
auto entry = *m_iterator;
|
|
++m_iterator;
|
|
if (m_iteration_kind == Object::PropertyKind::Key) {
|
|
value = entry.key;
|
|
return {};
|
|
}
|
|
if (m_iteration_kind == Object::PropertyKind::Value) {
|
|
value = entry.value;
|
|
return {};
|
|
}
|
|
|
|
value = Array::create_from(*vm.current_realm(), { entry.key, entry.value });
|
|
return {};
|
|
}
|
|
|
|
}
|