1 files changed, 39 insertions, 0 deletions
diff --git a/HALKit/AMD64/PCI/Iterator.cpp b/HALKit/AMD64/PCI/Iterator.cpp
new file mode 100644
index 00000000..a5649e52
--- /dev/null
+++ b/HALKit/AMD64/PCI/Iterator.cpp
@@ -0,0 +1,39 @@
+/*
+ *	========================================================
+ *
+ *	hCore
+ * 	Copyright Mahrouss Logic, all rights reserved.
+ *
+ * 	========================================================
+ */
+
+#include <KernelKit/PCI/Iterator.hpp>
+
+#define PCI_ITERATOR_FIND_AND_UNWRAP(DEV, SZ)                                                                          \
+    if (DEV.Leak())                                                                                                    \
+        return DEV.Leak();
+
+namespace hCore::PCI {
+    Iterator::Iterator(const Types::PciDeviceKind &type) {
+        // probe devices.
+        for (int bus = 0; bus < ME_BUS_COUNT; ++bus) {
+            for (int device = 0; device < ME_DEVICE_COUNT; ++device) {
+                for (int function = 0; function < ME_FUNCTION_COUNT; ++function) {
+                    Device dev(bus, device, function, 0);
+
+                    if (dev.Class() == (UChar) type) {
+                        m_Devices[bus].Leak().Leak() = dev;
+                    }
+                }
+            }
+        }
+    }
+
+    Iterator::~Iterator() {
+    }
+
+    Ref<PCI::Device> Iterator::operator[](const Size &sz) {
+        PCI_ITERATOR_FIND_AND_UNWRAP(m_Devices[sz], sz);
+        return {};
+    }
+} // namespace hCore::PCI