#pragma once

#include <Kernel/VM/MemoryManager.h>
#include <AK/OwnPtr.h>

struct SharedBuffer {
private:
    struct Reference {
        Reference(pid_t pid)
            : pid(pid)
        {
        }

        pid_t pid;
        unsigned count { 1 };
        Region* region { nullptr };
    };
public:
    SharedBuffer(int id, int size)
        : m_shared_buffer_id(id)
        , m_vmo(VMObject::create_anonymous(size))
    {
#ifdef SHARED_BUFFER_DEBUG
        dbgprintf("Created shared buffer %d of size %d\n", m_shared_buffer_id, size);
#endif
    }

    ~SharedBuffer()
    {
#ifdef SHARED_BUFFER_DEBUG
        dbgprintf("Destroyed shared buffer %d of size %d\n", m_shared_buffer_id, size());
#endif
    }

    bool is_shared_with(pid_t peer_pid);
    void* get_address(Process& process);
    void share_with(pid_t peer_pid);
    void release(Process& process);
    void disown(pid_t pid);
    size_t size() const { return m_vmo->size(); }
    void destroy_if_unused();
    void seal();

    int m_shared_buffer_id { -1 };
    bool m_writable { true };
    NonnullRefPtr<VMObject> m_vmo;
    Vector<Reference, 2> m_refs;
};

Lockable<HashMap<int, OwnPtr<SharedBuffer>>>& shared_buffers();