Files
ladybird/Libraries/LibGC/Function.h
Andreas Kling 949889f124 LibGC: Clear functions during finalization
A finalized GC::Function should no longer be callable. Clear the
underlying AK::Function during finalization so stale references fail as
empty callbacks instead of calling through finalized capture storage.
2026-05-10 10:58:11 +02:00

63 lines
1.4 KiB
C++

/*
* Copyright (c) 2023, Andreas Kling <andreas@ladybird.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#pragma once
#include <AK/Function.h>
#include <LibGC/Cell.h>
#include <LibGC/Heap.h>
namespace GC {
template<typename T>
class Function final : public Cell {
GC_CELL(Function, Cell);
GC_DECLARE_ALLOCATOR(Function);
public:
static constexpr bool OVERRIDES_FINALIZE = true;
static Ref<Function> create(Heap& heap, ESCAPING AK::Function<T>&& function)
{
return heap.allocate<Function>(move(function));
}
virtual ~Function() override = default;
virtual void finalize() override
{
Base::finalize();
m_function = nullptr;
}
[[nodiscard]] AK::Function<T> const& function() const { return m_function; }
private:
Function(AK::Function<T>&& function)
: m_function(move(function))
{
}
virtual void visit_edges(Visitor& visitor) override
{
Base::visit_edges(visitor);
visitor.visit_possible_values(m_function.raw_capture_range());
}
AK::Function<T> m_function;
};
template<typename Callable, typename T = EquivalentFunctionType<Callable>>
static Ref<Function<T>> create_function(Heap& heap, ESCAPING Callable&& function)
{
return Function<T>::create(heap, AK::Function<T> { forward<Callable>(function) });
}
template<typename T>
GC_DEFINE_ALLOCATOR(Function<T>);
}