kapi: move protocol pointer to base bus

5 files changed, 21 insertions, 46 deletions

diff --git a/kernel/kapi/devices/bus.cpp b/kernel/kapi/devices/bus.cpp
index 9eecced0..dad90a15 100644
--- a/kernel/kapi/devices/bus.cpp
+++ b/kernel/kapi/devices/bus.cpp
@@ -1,6 +1,7 @@
 #include <kapi/devices/bus.hpp>
 
 #include <kapi/devices.hpp>
+#include <kapi/devices/bus_protocol.hpp>
 #include <kapi/devices/driver_registry.hpp>
 #include <kapi/system.hpp>
 
@@ -17,6 +18,11 @@ namespace kapi::devices
       : device{name}
   {}
 
+  bus::bus(kstd::string const & name, bus_protocol & protocol)
+      : device{name}
+      , m_protocol{&protocol}
+  {}
+
   auto bus::add_child(kstd::shared_ptr<device> child) -> void
   {
     child->set_parent(kstd::static_pointer_cast<bus>(shared_from_this()));
@@ -43,6 +49,10 @@ namespace kapi::devices
     {
       return this;
     }
+    else if (facet == bus_protocol::id)
+    {
+      return m_protocol;
+    }
 
     return device::query_facet(facet);
   }
diff --git a/kernel/kapi/devices/driver.tests.cpp b/kernel/kapi/devices/driver.tests.cpp
index b935cc57..623f9d4c 100644
--- a/kernel/kapi/devices/driver.tests.cpp
+++ b/kernel/kapi/devices/driver.tests.cpp
@@ -47,17 +47,9 @@ namespace
 
   struct controller_bus final : kapi::devices::bus
   {
-    using kapi::devices::bus::bus;
-
-    [[nodiscard]] auto query_facet(kapi::capabilities::facet_id facet) -> void * override
-    {
-      if (facet == kapi::devices::bus_protocol::id)
-      {
-        return static_cast<kapi::devices::bus_protocol *>(&controller_protocol_instance);
-      }
-
-      return kapi::devices::bus::query_facet(facet);
-    }
+    controller_bus(kstd::string name)
+        : bus{name, controller_protocol_instance}
+    {}
   };
 
   struct controller_device final : kapi::devices::bus, controller_identification
@@ -142,17 +134,9 @@ namespace
 
   struct leaf_bus final : kapi::devices::bus
   {
-    using kapi::devices::bus::bus;
-
-    [[nodiscard]] auto query_facet(kapi::capabilities::facet_id id) -> void * override
-    {
-      if (id == kapi::devices::bus_protocol::id)
-      {
-        return static_cast<kapi::devices::bus_protocol *>(&leaf_protocol_instance);
-      }
-
-      return kapi::devices::bus::query_facet(id);
-    }
+    leaf_bus(kstd::string name)
+        : bus{name, leaf_protocol_instance}
+    {}
   };
 
   struct controller_driver final : kapi::devices::driver, controller_driver_identification
diff --git a/kernel/kapi/devices/driver_registry.tests.cpp b/kernel/kapi/devices/driver_registry.tests.cpp
index 7375bd4d..c3889d42 100644
--- a/kernel/kapi/devices/driver_registry.tests.cpp
+++ b/kernel/kapi/devices/driver_registry.tests.cpp
@@ -4,6 +4,7 @@
 
 #include <kstd/memory.hpp>
 #include <kstd/result.hpp>
+#include <kstd/string.hpp>
 
 #include <catch2/catch_test_macros.hpp>
 
@@ -52,17 +53,9 @@ namespace
 
   struct test_bus final : kapi::devices::bus
   {
-    using kapi::devices::bus::bus;
-
-    [[nodiscard]] auto query_facet(kapi::capabilities::facet_id facet) -> void * override
-    {
-      if (facet == kapi::devices::bus_protocol::id)
-      {
-        return static_cast<kapi::devices::bus_protocol *>(&bus_protocol);
-      }
-
-      return kapi::devices::bus::query_facet(facet);
-    }
+    test_bus(kstd::string name)
+        : bus{name, bus_protocol}
+    {}
 
     unsigned match_calls{};
     test_protocol bus_protocol{&match_calls};
diff --git a/kernel/kernel/bus/boot_modules.cpp b/kernel/kernel/bus/boot_modules.cpp
index 2f5fe8dd..ed7c1e5a 100644
--- a/kernel/kernel/bus/boot_modules.cpp
+++ b/kernel/kernel/bus/boot_modules.cpp
@@ -60,17 +60,7 @@ namespace kernel::bus
   }  // namespace
 
   boot_modules::boot_modules()
-      : kapi::devices::bus{"boot_modules"}
+      : kapi::devices::bus{"boot_modules", protocol_instance}
   {}
 
-  auto boot_modules::query_facet(kapi::capabilities::facet_id id) -> void *
-  {
-    if (id == kapi::devices::bus_protocol::id)
-    {
-      return static_cast<kapi::devices::bus_protocol *>(&protocol_instance);
-    }
-
-    return kapi::devices::bus::query_facet(id);
-  }
-
 }  // namespace kernel::bus
diff --git a/kernel/kernel/bus/boot_modules.hpp b/kernel/kernel/bus/boot_modules.hpp
index 011e2b2e..852e5154 100644
--- a/kernel/kernel/bus/boot_modules.hpp
+++ b/kernel/kernel/bus/boot_modules.hpp
@@ -14,8 +14,6 @@ namespace kernel::bus
   struct boot_modules final : kapi::devices::bus
   {
     boot_modules();
-
-    [[nodiscard]] auto query_facet(kapi::capabilities::facet_id id) -> void * override;
   };
 
 }  // namespace kernel::bus