diff --git a/Kernel/Process.h b/Kernel/Process.h index a17e9da359..b2b8f961ac 100644 --- a/Kernel/Process.h +++ b/Kernel/Process.h @@ -12,6 +12,7 @@ #include #include #include +#include #include class FileDescriptor; @@ -45,7 +46,7 @@ struct DisplayInfo { byte* framebuffer; }; -class Process : public InlineLinkedListNode { +class Process : public InlineLinkedListNode, public Weakable { friend class InlineLinkedListNode; friend class WSWindowManager; // FIXME: Make a better API for allocate_region(). friend class GraphicsBitmap; // FIXME: Make a better API for allocate_region(). diff --git a/SharedGraphics/GraphicsBitmap.cpp b/SharedGraphics/GraphicsBitmap.cpp index 7121161ce1..97d8e815d4 100644 --- a/SharedGraphics/GraphicsBitmap.cpp +++ b/SharedGraphics/GraphicsBitmap.cpp @@ -16,7 +16,7 @@ RetainPtr GraphicsBitmap::create(Process& process, const Size& s GraphicsBitmap::GraphicsBitmap(Process& process, const Size& size) : m_size(size) , m_pitch(size.width() * sizeof(RGBA32)) - , m_client_process(&process) + , m_client_process(process.makeWeakPtr()) { InterruptDisabler disabler; size_t size_in_bytes = size.width() * size.height() * sizeof(RGBA32); @@ -47,7 +47,7 @@ GraphicsBitmap::GraphicsBitmap(const Size& size, RGBA32* data) GraphicsBitmap::~GraphicsBitmap() { #ifdef KERNEL - if (m_client_region) + if (m_client_region && m_client_process) m_client_process->deallocate_region(*m_client_region); if (m_server_region) WSMessageLoop::the().server_process().deallocate_region(*m_server_region); diff --git a/SharedGraphics/GraphicsBitmap.h b/SharedGraphics/GraphicsBitmap.h index a0796f92a4..de3f934ad2 100644 --- a/SharedGraphics/GraphicsBitmap.h +++ b/SharedGraphics/GraphicsBitmap.h @@ -43,7 +43,7 @@ private: size_t m_pitch { 0 }; #ifdef KERNEL - Process* m_client_process { nullptr }; + WeakPtr m_client_process; Region* m_client_region { nullptr }; Region* m_server_region { nullptr }; #endif