1 files changed, 43 insertions, 0 deletions
diff --git a/Kernel/StorageKit/NVME.hpp b/Kernel/StorageKit/NVME.hpp
new file mode 100644
index 00000000..4f1c6362
--- /dev/null
+++ b/Kernel/StorageKit/NVME.hpp
@@ -0,0 +1,43 @@
+/* -------------------------------------------
+
+    Copyright SoftwareLabs
+
+------------------------------------------- */
+
+#pragma once
+
+#include <KernelKit/DeviceManager.hpp>
+#include <KernelKit/DriveManager.hxx>
+#include <NewKit/OwnPtr.hpp>
+
+namespace NewOS
+{
+	class NVMEDeviceInterface : public DeviceInterface<MountpointInterface*>
+	{
+	public:
+		explicit NVMEDeviceInterface(void (*Out)(MountpointInterface* outpacket),
+									 void (*In)(MountpointInterface* inpacket),
+									 void (*Cleanup)(void))
+			: DeviceInterface(Out, In), fCleanup(Cleanup)
+		{
+		}
+
+		virtual ~NVMEDeviceInterface()
+		{
+			if (fCleanup)
+				fCleanup();
+		}
+
+	public:
+		NVMEDeviceInterface& operator=(const NVMEDeviceInterface&) = default;
+		NVMEDeviceInterface(const NVMEDeviceInterface&)			   = default;
+
+		const char* Name() const override;
+
+	public:
+		OwnPtr<MountpointInterface*> operator()(UInt32 dmaLow, UInt32 dmaHigh, SizeT sz);
+
+	private:
+		void (*fCleanup)(void);
+	};
+} // namespace NewOS