Kernel: Add STORAGE_DEVICE_GET_SIZE ioctl request

This ioctl request makes it possible to get the size of a storage device
that has not yet been mounted.

1 files changed, 14 insertions, 0 deletions

diff --git a/Kernel/Storage/StorageDevice.cpp b/Kernel/Storage/StorageDevice.cpp
index 217fc51810..4588f67617 100644
--- a/Kernel/Storage/StorageDevice.cpp
+++ b/Kernel/Storage/StorageDevice.cpp
@@ -10,6 +10,7 @@
 #include <Kernel/FileSystem/OpenFileDescription.h>
 #include <Kernel/Storage/StorageDevice.h>
 #include <Kernel/Storage/StorageManagement.h>
+#include <LibC/sys/ioctl_numbers.h>
 
 namespace Kernel {
 
@@ -187,4 +188,17 @@ bool StorageDevice::can_write(const OpenFileDescription&, size_t offset) const
     return offset < (max_addressable_block() * block_size());
 }
 
+KResult StorageDevice::ioctl(OpenFileDescription&, unsigned request, Userspace<void*> arg)
+{
+    switch (request) {
+    case STORAGE_DEVICE_GET_SIZE: {
+        size_t disk_size = m_max_addressable_block * block_size();
+        return copy_to_user(Userspace<size_t*>(arg), &disk_size);
+        break;
+    }
+    default:
+        return EINVAL;
+    }
+}
+
 }