ladybird/Userland/Libraries/LibJS/Runtime/FinalizationRegistry.h
Aliaksandr Kalenik a3b4c2a30f LibJS+LibWeb: Change JobCallback to be GC-allocated
Fixes leak caused by mutual dependency when JS::Handle<JobCallback> is
owned by GC-allocated PromiseReaction.
2024-03-16 14:27:06 +01:00

57 lines
1.6 KiB
C++

/*
* Copyright (c) 2021-2022, Idan Horowitz <idan.horowitz@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include <AK/SinglyLinkedList.h>
#include <LibJS/Heap/GCPtr.h>
#include <LibJS/Runtime/FunctionObject.h>
#include <LibJS/Runtime/GlobalObject.h>
#include <LibJS/Runtime/JobCallback.h>
#include <LibJS/Runtime/Object.h>
#include <LibJS/Runtime/Value.h>
#include <LibJS/Runtime/WeakContainer.h>
namespace JS {
class FinalizationRegistry final
: public Object
, public WeakContainer {
JS_OBJECT(FinalizationRegistry, Object);
JS_DECLARE_ALLOCATOR(FinalizationRegistry);
public:
virtual ~FinalizationRegistry() override = default;
void add_finalization_record(Cell& target, Value held_value, Cell* unregister_token);
bool remove_by_token(Cell& unregister_token);
ThrowCompletionOr<void> cleanup(GCPtr<JobCallback> = {});
virtual void remove_dead_cells(Badge<Heap>) override;
Realm& realm() { return *m_realm; }
Realm const& realm() const { return *m_realm; }
JobCallback& cleanup_callback() { return *m_cleanup_callback; }
JobCallback const& cleanup_callback() const { return *m_cleanup_callback; }
private:
FinalizationRegistry(Realm&, NonnullGCPtr<JobCallback>, Object& prototype);
virtual void visit_edges(Visitor& visitor) override;
NonnullGCPtr<Realm> m_realm;
NonnullGCPtr<JobCallback> m_cleanup_callback;
struct FinalizationRecord {
GCPtr<Cell> target;
Value held_value;
GCPtr<Cell> unregister_token;
};
SinglyLinkedList<FinalizationRecord> m_records;
};
}