kapi: add find overload to find composed interfaces

2 files changed, 36 insertions, 2 deletions

diff --git a/kernel/kapi/devices/interface_registry.cpp b/kernel/kapi/devices/interface_registry.cpp
index da3d080a..d959c857 100644
--- a/kernel/kapi/devices/interface_registry.cpp
+++ b/kernel/kapi/devices/interface_registry.cpp
@@ -97,6 +97,20 @@ namespace kapi::devices
     return *found;
   }
 
+  auto interface_registry::find(interface interface, device const & device) const -> kstd::result<entry>
+  {
+    auto found = std::ranges::find_if(m_entries, [&](auto e) {
+      auto locked_device = e.device();
+      return e.interface() == interface && locked_device && locked_device.get() == &device;
+    });
+    if (found == m_entries.cend())
+    {
+      return kstd::failure(make_error_code(kstd::errc::no_such_device));
+    }
+
+    return *found;
+  }
+
 }  // namespace kapi::devices
 
 namespace kapi::test_support::devices
diff --git a/kernel/kernel/filesystem/device_inode.cpp b/kernel/kernel/filesystem/device_inode.cpp
index 33037a52..0ff8dce6 100644
--- a/kernel/kernel/filesystem/device_inode.cpp
+++ b/kernel/kernel/filesystem/device_inode.cpp
@@ -16,6 +16,26 @@
 
 #include <optional>
 
+namespace
+{
+  template<typename Interface>
+  auto resolve(kapi::devices::device & device) -> Interface *
+  {
+    if (auto * direct = device.as<Interface>())
+    {
+      return direct;
+    }
+
+    auto found = kapi::devices::interface_registry::get().find(Interface::id, device);
+    if (!found)
+    {
+      return nullptr;
+    }
+
+    return found->template as<Interface>();
+  }
+}  // namespace
+
 namespace kernel::filesystem
 {
   device_inode::device_inode(kstd::shared_ptr<kapi::devices::device> const & device)
@@ -41,7 +61,7 @@ namespace kernel::filesystem
   auto device_inode::read(void * buffer, kstd::units::bytes offset, kstd::units::bytes size) const
       -> kstd::result<kstd::units::bytes>
   {
-    if (auto block_device = m_device->as<kapi::devices::block_device>())
+    if (auto block_device = resolve<kapi::devices::block_device>(*m_device))
     {
       return devices::block_device_utils::read(*block_device, buffer, offset, size);
     }
@@ -52,7 +72,7 @@ namespace kernel::filesystem
   auto device_inode::write(void const * buffer, kstd::units::bytes offset, kstd::units::bytes size)
       -> kstd::result<kstd::units::bytes>
   {
-    if (auto block_device = m_device->as<kapi::devices::block_device>())
+    if (auto block_device = resolve<kapi::devices::block_device>(*m_device))
     {
       return devices::block_device_utils::write(*block_device, buffer, offset, size);
     }