1 files changed, 11 insertions, 12 deletions
diff --git a/kernel/devices/src/storage/RAMDisk/RAMDiskDevice.cpp b/kernel/devices/src/storage/RAMDisk/RAMDiskDevice.cpp
index 1e650d8..82a3cdf 100644
--- a/kernel/devices/src/storage/RAMDisk/RAMDiskDevice.cpp
+++ b/kernel/devices/src/storage/RAMDisk/RAMDiskDevice.cpp
@@ -1,6 +1,6 @@
 #include "devices/storage/RAMDisk/RAMDiskDevice.hpp"
 
-#include "kapi/memory.hpp"
+#include "kapi/boot_module/boot_module.hpp"
 #include "kapi/system.hpp"
 
 #include <kstd/print>
@@ -9,26 +9,25 @@
 
 namespace devices::storage::ram_disk
 {
-  ram_disk_device::ram_disk_device(kapi::memory::linear_address data_start, size_t data_size)
-      : m_data_start(data_start)
-      , m_data_size(data_size)
+  ram_disk_device::ram_disk_device(kapi::boot_modules::boot_module const & module)
+      : m_boot_module(module)
   {}
 
-  auto ram_disk_device::read_block(size_t block_index, void * /*buffer*/) const -> void
+  auto ram_disk_device::read_block(size_t block_index, void * buffer) const -> void
   {
-    // if (buffer == nullptr)
-    // {
-    //   kapi::system::panic("[RAM DISK DEVICE] read_block called with null buffer.");
-    // }
+    if (buffer == nullptr)
+    {
+      kapi::system::panic("[RAM DISK DEVICE] read_block called with null buffer.");
+    }
 
     auto const offset = block_index * block_size;
-    if (offset + block_size >
-        m_data_size)  // TODO BA-FS26 really correct, what if block_size doesn't divide m_data_size?
+    // TODO BA-FS26 really correct, what if block_size doesn't divide m_boot_module.size?
+    if (offset + block_size > m_boot_module.size)
     {
       kapi::system::panic("[RAM DISK DEVICE] read_block out of bounds.");
     }
 
-    auto const source = static_cast<std::byte *>(m_data_start) + offset;
+    auto const source = static_cast<std::byte *>(m_boot_module.start_address) + offset;
     for (size_t i = 0; i < block_size; ++i)
     {
       kstd::println("address: {}, value: {}", source + i, std::to_integer<unsigned int>(*(source + i)));