/* * Copyright (c) 2020, Andreas Kling * * SPDX-License-Identifier: BSD-2-Clause */ #pragma once #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace JS { class Heap { AK_MAKE_NONCOPYABLE(Heap); AK_MAKE_NONMOVABLE(Heap); public: explicit Heap(VM&); ~Heap(); template T* allocate_without_global_object(Args&&... args) { auto* memory = allocate_cell(sizeof(T)); new (memory) T(forward(args)...); return static_cast(memory); } template T* allocate(GlobalObject& global_object, Args&&... args) { auto* memory = allocate_cell(sizeof(T)); new (memory) T(forward(args)...); auto* cell = static_cast(memory); constexpr bool is_object = IsBaseOf; if constexpr (is_object) static_cast(cell)->disable_transitions(); cell->initialize(global_object); if constexpr (is_object) { static_cast(cell)->enable_transitions(); static_cast(cell)->set_initialized(Badge {}); } return cell; } enum class CollectionType { CollectGarbage, CollectEverything, }; void collect_garbage(CollectionType = CollectionType::CollectGarbage, bool print_report = false); VM& vm() { return m_vm; } bool should_collect_on_every_allocation() const { return m_should_collect_on_every_allocation; } void set_should_collect_on_every_allocation(bool b) { m_should_collect_on_every_allocation = b; } void set_zombify_dead_cells(bool b) { m_zombify_dead_cells = b; } void did_create_handle(Badge, HandleImpl&); void did_destroy_handle(Badge, HandleImpl&); void did_create_marked_value_list(Badge, MarkedValueList&); void did_destroy_marked_value_list(Badge, MarkedValueList&); void did_create_weak_container(Badge, WeakContainer&); void did_destroy_weak_container(Badge, WeakContainer&); void defer_gc(Badge); void undefer_gc(Badge); BlockAllocator& block_allocator() { return m_block_allocator; } void uproot_cell(Cell* cell); private: Cell* allocate_cell(size_t); void gather_roots(HashTable&); void gather_conservative_roots(HashTable&); void mark_live_cells(const HashTable& live_cells); void sweep_dead_cells(bool print_report, const Core::ElapsedTimer&); CellAllocator& allocator_for_size(size_t); template void for_each_block(Callback callback) { for (auto& allocator : m_allocators) { if (allocator->for_each_block(callback) == IterationDecision::Break) return; } } size_t m_max_allocations_between_gc { 10000 }; size_t m_allocations_since_last_gc { 0 }; bool m_should_collect_on_every_allocation { false }; VM& m_vm; Vector> m_allocators; HandleImpl::List m_handles; MarkedValueList::List m_marked_value_lists; WeakContainer::List m_weak_containers; Vector m_uprooted_cells; BlockAllocator m_block_allocator; size_t m_gc_deferrals { 0 }; bool m_should_gc_when_deferral_ends { false }; bool m_collecting_garbage { false }; bool m_zombify_dead_cells { false }; }; }