mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-20 17:58:18 +03:00
79fa9765ca
We now use AK::Error and AK::ErrorOr<T> in both kernel and userspace! This was a slightly tedious refactoring that took a long time, so it's not unlikely that some bugs crept in. Nevertheless, it does pass basic functionality testing, and it's just real nice to finally see the same pattern in all contexts. :^)
66 lines
2.3 KiB
C++
66 lines
2.3 KiB
C++
/*
|
|
* Copyright (c) 2018-2021, Andreas Kling <kling@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
// KBuffer: Memory buffer backed by a kernel region.
|
|
//
|
|
// The memory is allocated via the global kernel-only page allocator, rather than via
|
|
// kmalloc() which is what ByteBuffer/Vector/etc will use.
|
|
//
|
|
// This makes KBuffer a little heavier to allocate, but much better for large and/or
|
|
// long-lived allocations, since they don't put all that weight and pressure on the
|
|
// severely limited kmalloc heap.
|
|
|
|
#include <AK/Assertions.h>
|
|
#include <AK/Memory.h>
|
|
#include <AK/StringView.h>
|
|
#include <Kernel/Memory/MemoryManager.h>
|
|
|
|
namespace Kernel {
|
|
|
|
class [[nodiscard]] KBuffer {
|
|
public:
|
|
static ErrorOr<NonnullOwnPtr<KBuffer>> try_create_with_size(size_t size, Memory::Region::Access access = Memory::Region::Access::ReadWrite, StringView name = "KBuffer", AllocationStrategy strategy = AllocationStrategy::Reserve)
|
|
{
|
|
auto region = TRY(MM.allocate_kernel_region(Memory::page_round_up(size), name, access, strategy));
|
|
return TRY(adopt_nonnull_own_or_enomem(new (nothrow) KBuffer { size, move(region) }));
|
|
}
|
|
|
|
static ErrorOr<NonnullOwnPtr<KBuffer>> try_create_with_bytes(ReadonlyBytes bytes, Memory::Region::Access access = Memory::Region::Access::ReadWrite, StringView name = "KBuffer", AllocationStrategy strategy = AllocationStrategy::Reserve)
|
|
{
|
|
auto buffer = TRY(try_create_with_size(bytes.size(), access, name, strategy));
|
|
memcpy(buffer->data(), bytes.data(), bytes.size());
|
|
return buffer;
|
|
}
|
|
|
|
[[nodiscard]] u8* data() { return m_region->vaddr().as_ptr(); }
|
|
[[nodiscard]] u8 const* data() const { return m_region->vaddr().as_ptr(); }
|
|
[[nodiscard]] size_t size() const { return m_size; }
|
|
[[nodiscard]] size_t capacity() const { return m_region->size(); }
|
|
|
|
[[nodiscard]] ReadonlyBytes bytes() const { return { data(), size() }; }
|
|
[[nodiscard]] Bytes bytes() { return { data(), size() }; }
|
|
|
|
void set_size(size_t size)
|
|
{
|
|
VERIFY(size <= capacity());
|
|
m_size = size;
|
|
}
|
|
|
|
private:
|
|
explicit KBuffer(size_t size, NonnullOwnPtr<Memory::Region> region)
|
|
: m_size(size)
|
|
, m_region(move(region))
|
|
{
|
|
}
|
|
|
|
size_t m_size { 0 };
|
|
NonnullOwnPtr<Memory::Region> m_region;
|
|
};
|
|
|
|
}
|