refactoring read and write block calculations

1 files changed, 13 insertions, 1 deletions

diff --git a/kernel/devices/src/BlockDevice.cpp b/kernel/devices/src/BlockDevice.cpp
index 6d43073..edc0109 100644
--- a/kernel/devices/src/BlockDevice.cpp
+++ b/kernel/devices/src/BlockDevice.cpp
@@ -6,7 +6,19 @@
 
 namespace devices
 {
-  block_device::block_device(size_t major, size_t minor)
+  block_device::block_device(size_t major, size_t minor, size_t block_size)
       : device(major, minor)
+      , m_block_size(block_size)
   {}
+
+  auto block_device::calculate_transfer(size_t block_index) const -> transfer_info
+  {
+    size_t const offset = block_index * m_block_size;
+    size_t const limit = size();
+
+    size_t const available = (offset < limit) ? (limit - offset) : 0;
+    size_t const to_transfer = (available < m_block_size) ? available : m_block_size;
+
+    return {offset, to_transfer, m_block_size - to_transfer};
+  }
 }  // namespace devices
\ No newline at end of file