kapi: use bytes for size in boot modules

2 files changed, 4 insertions, 3 deletions

diff --git a/kernel/kernel/drivers/storage/ram_disk.cpp b/kernel/kernel/drivers/storage/ram_disk.cpp
index 186f8cbc..383a9417 100644
--- a/kernel/kernel/drivers/storage/ram_disk.cpp
+++ b/kernel/kernel/drivers/storage/ram_disk.cpp
@@ -107,7 +107,7 @@ namespace kernel::drivers::storage
     }
 
     auto const & module = facet->module();
-    if (module.start_address.raw() == 0 || module.size == 0)
+    if (module.start_address.raw() == 0 || module.size == 0_B)
     {
       return kstd::failure(make_error_code(kstd::errc::invalid_argument));
     }
diff --git a/kernel/kernel/test_support/filesystem/storage_boot_module_fixture.cpp b/kernel/kernel/test_support/filesystem/storage_boot_module_fixture.cpp
index d5068565..b9228c69 100644
--- a/kernel/kernel/test_support/filesystem/storage_boot_module_fixture.cpp
+++ b/kernel/kernel/test_support/filesystem/storage_boot_module_fixture.cpp
@@ -10,6 +10,7 @@
 #include <kapi/memory.hpp>
 
 #include <kstd/memory.hpp>
+#include <kstd/units.hpp>
 
 #include <cstddef>
 #include <fcntl.h>
@@ -99,7 +100,7 @@ namespace kernel::tests::filesystem
       auto module = kapi::boot_modules::module{
         .name = m_module_names[i].c_str(),
         .start_address = kapi::memory::linear_address{m_module_data[i].data()},
-        .size = m_module_data[i].size(),
+        .size = kstd::bytes{m_module_data[i].size()},
       };
       m_boot_module_bus->add_child(kstd::make_shared<kapi::boot_modules::device>(i, module));
     }
@@ -142,7 +143,7 @@ namespace kernel::tests::filesystem
     auto module = kapi::boot_modules::module{
       .name = m_module_names.back().c_str(),
       .start_address = kapi::memory::linear_address{mapped_image.mapping},
-      .size = mapped_image.size,
+      .size = kstd::bytes{mapped_image.size},
     };
 
     m_boot_module_bus->add_child(kstd::make_shared<kapi::boot_modules::device>(index, module));