3 files changed, 20 insertions, 33 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};