summaryrefslogtreecommitdiff
path: root/Kernel/Memory/RingBuffer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'Kernel/Memory/RingBuffer.cpp')
-rw-r--r--Kernel/Memory/RingBuffer.cpp66
1 files changed, 66 insertions, 0 deletions
diff --git a/Kernel/Memory/RingBuffer.cpp b/Kernel/Memory/RingBuffer.cpp
new file mode 100644
index 0000000000..9c62a53a9d
--- /dev/null
+++ b/Kernel/Memory/RingBuffer.cpp
@@ -0,0 +1,66 @@
+/*
+ * Copyright (c) 2021, Sahan Fernando <sahan.h.fernando@gmail.com>.
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <Kernel/Memory/MemoryManager.h>
+#include <Kernel/Memory/RingBuffer.h>
+#include <Kernel/UserOrKernelBuffer.h>
+
+namespace Kernel {
+
+RingBuffer::RingBuffer(String region_name, size_t capacity)
+ : m_region(MM.allocate_contiguous_kernel_region(page_round_up(capacity), move(region_name), Region::Access::Read | Region::Access::Write))
+ , m_capacity_in_bytes(capacity)
+{
+}
+
+bool RingBuffer::copy_data_in(const UserOrKernelBuffer& buffer, size_t offset, size_t length, PhysicalAddress& start_of_copied_data, size_t& bytes_copied)
+{
+ size_t start_of_free_area = (m_start_of_used + m_num_used_bytes) % m_capacity_in_bytes;
+ bytes_copied = min(m_capacity_in_bytes - m_num_used_bytes, min(m_capacity_in_bytes - start_of_free_area, length));
+ if (bytes_copied == 0)
+ return false;
+ if (buffer.read(m_region->vaddr().offset(start_of_free_area).as_ptr(), offset, bytes_copied)) {
+ m_num_used_bytes += bytes_copied;
+ start_of_copied_data = m_region->physical_page(start_of_free_area / PAGE_SIZE)->paddr().offset(start_of_free_area % PAGE_SIZE);
+ return true;
+ }
+ return false;
+}
+
+KResultOr<size_t> RingBuffer::copy_data_out(size_t size, UserOrKernelBuffer& buffer) const
+{
+ auto start = m_start_of_used % m_capacity_in_bytes;
+ auto num_bytes = min(min(m_num_used_bytes, size), m_capacity_in_bytes - start);
+ if (!buffer.write(m_region->vaddr().offset(start).as_ptr(), num_bytes))
+ return EIO;
+ return num_bytes;
+}
+
+KResultOr<PhysicalAddress> RingBuffer::reserve_space(size_t size)
+{
+ if (m_capacity_in_bytes < m_num_used_bytes + size)
+ return ENOSPC;
+ size_t start_of_free_area = (m_start_of_used + m_num_used_bytes) % m_capacity_in_bytes;
+ m_num_used_bytes += size;
+ PhysicalAddress start_of_reserved_space = m_region->physical_page(start_of_free_area / PAGE_SIZE)->paddr().offset(start_of_free_area % PAGE_SIZE);
+ return start_of_reserved_space;
+}
+
+void RingBuffer::reclaim_space(PhysicalAddress chunk_start, size_t chunk_size)
+{
+ VERIFY(start_of_used() == chunk_start);
+ VERIFY(m_num_used_bytes >= chunk_size);
+ m_num_used_bytes -= chunk_size;
+ m_start_of_used += chunk_size;
+}
+
+PhysicalAddress RingBuffer::start_of_used() const
+{
+ size_t start = m_start_of_used % m_capacity_in_bytes;
+ return m_region->physical_page(start / PAGE_SIZE)->paddr().offset(start % PAGE_SIZE);
+}
+
+}