IMP: Major refactor of header and source files extensions.

Signed-off-by: Amlal El Mahrouss <amlal.elmahrouss@icloud.com>

1 files changed, 167 insertions, 0 deletions

diff --git a/dev/zka/src/DriveMgr.cc b/dev/zka/src/DriveMgr.cc
new file mode 100644
index 00000000..9f5201d9
--- /dev/null
+++ b/dev/zka/src/DriveMgr.cc
@@ -0,0 +1,167 @@
+/* -------------------------------------------
+
+	Copyright ZKA Web Services Co.
+
+------------------------------------------- */
+
+#include <KernelKit/DebugOutput.h>
+#include <KernelKit/DriveMgr.h>
+#include <NewKit/Utils.h>
+
+#include <modules/ATA/ATA.h>
+#include <modules/AHCI/AHCI.h>
+#include <modules/NVME/NVME.h>
+
+/***********************************************************************************/
+/// @file DriveMgr.cxx
+/// @brief Drive Manager API.
+/***********************************************************************************/
+
+namespace Kernel
+{
+	STATIC UInt16 kATAIO	 = 0U;
+	STATIC UInt8  kATAMaster = 0U;
+
+	/// @brief reads from an ATA drive.
+	/// @param pckt Packet structure (fPacketContent must be non null)
+	/// @return
+	Void io_drv_input(DriveTrait::DrivePacket* pckt)
+	{
+		if (!pckt)
+		{
+			return;
+		}
+
+#ifdef __AHCI__
+		drv_std_read(pckt->fLba, (Char*)pckt->fPacketContent.Leak(), kAHCISectorSize, pckt->fPacketSize);
+#elif defined(__ATA_PIO__) || defined(__ATA_DMA__)
+		drv_std_read(pckt->fLba, kATAIO, kATAMaster, (Char*)pckt->fPacketContent.Leak(), kATASectorSize, pckt->fPacketSize);
+#endif
+	}
+
+	/// @brief Writes to an ATA drive.
+	/// @param pckt the packet to write.
+	/// @return
+	Void io_drv_output(DriveTrait::DrivePacket* pckt)
+	{
+		if (!pckt)
+		{
+			return;
+		}
+
+#ifdef __AHCI__
+		drv_std_write(pckt->fLba, (Char*)pckt->fPacketContent.Leak(), kAHCISectorSize, pckt->fPacketSize);
+#elif defined(__ATA_PIO__) || defined(__ATA_DMA__)
+		drv_std_write(pckt->fLba, kATAIO, kATAMaster, (Char*)pckt->fPacketContent.Leak(), kATASectorSize, pckt->fPacketSize);
+#endif
+	}
+
+	/// @brief Executes a disk check on the ATA drive.
+	/// @param pckt the packet to read.
+	/// @return
+	Void io_drv_init(DriveTrait::DrivePacket* pckt)
+	{
+		if (!pckt)
+		{
+			return;
+		}
+
+		pckt->fPacketGood = false;
+
+#if defined(__ATA_PIO__) || defined(__ATA_DMA__)
+		kATAMaster = true;
+		kATAIO	   = ATA_PRIMARY_IO;
+
+		if (!drv_std_init(kATAIO, kATAMaster, kATAIO, kATAMaster))
+		    return;
+
+#elif defined(__AHCI__)
+		UInt16 pi = 0;
+
+		if (!drv_std_init(pi))
+		    return;
+#endif // if defined(__ATA_PIO__) || defined (__ATA_DMA__)
+
+		pckt->fPacketGood = true;
+	}
+
+/// @brief Gets the drive kind (ATA, SCSI, AHCI...)
+/// @param no arguments.
+/// @return no arguments.
+#ifdef __ATA_PIO__
+	const Char* io_drv_kind(Void)
+	{
+		return "ATA-PIO";
+	}
+#endif
+#ifdef __ATA_DMA__
+	const Char* io_drv_kind(Void)
+	{
+		return "ATA-DMA";
+	}
+#endif
+#ifdef __AHCI__
+	const Char* io_drv_kind(Void)
+	{
+		return "AHCI";
+	}
+#endif
+#ifdef __ZKA_MINIMAL_OS__
+	const Char* io_drv_kind(Void)
+	{
+		return "Not Loaded";
+	}
+#endif
+#ifdef __ZKA_ED__
+	const Char* io_drv_kind(Void)
+	{
+		return "C++ Code Editor";
+	}
+#endif
+
+	/// @brief Unimplemented drive function.
+	/// @param pckt the packet to read.
+	/// @return
+	Void io_drv_unimplemented(DriveTrait::DrivePacket* pckt)
+	{
+		ZKA_UNUSED(pckt);
+	}
+
+	/// @brief Makes a new drive.
+	/// @return the new blank drive.
+	DriveTrait io_construct_drive() noexcept
+	{
+		DriveTrait trait;
+
+		rt_copy_memory((VoidPtr) "\\Mount\\NUL:", trait.fName, rt_string_len("\\Mount\\NUL:"));
+		trait.fKind = kInvalidStorage;
+
+		trait.fInput	 = io_drv_unimplemented;
+		trait.fOutput	 = io_drv_unimplemented;
+		trait.fVerify	 = io_drv_unimplemented;
+		trait.fInit		 = io_drv_unimplemented;
+		trait.fDriveKind = io_drv_kind;
+
+		return trait;
+	}
+
+	/// @brief Fetches the main drive.
+	/// @return the new drive.
+	DriveTrait io_construct_main_drive() noexcept
+	{
+		DriveTrait trait;
+
+		rt_copy_memory((VoidPtr) "\\Mount\\MainDisk:", trait.fName, rt_string_len("\\Mount\\MainDisk:"));
+		trait.fKind = kMassStorage | kEPMDrive;
+
+		trait.fVerify	 = io_drv_unimplemented;
+		trait.fOutput	 = io_drv_output;
+		trait.fInput	 = io_drv_input;
+		trait.fInit		 = io_drv_init;
+		trait.fDriveKind = io_drv_kind;
+
+		kcout << "Constructed drive successfully.\r";
+
+		return trait;
+	}
+} // namespace Kernel