1 files changed, 184 insertions, 0 deletions
diff --git a/src/kernel/HALKit/AMD64/Storage/DMA+Generic.cpp b/src/kernel/HALKit/AMD64/Storage/DMA+Generic.cpp
new file mode 100644
index 00000000..8bf7b161
--- /dev/null
+++ b/src/kernel/HALKit/AMD64/Storage/DMA+Generic.cpp
@@ -0,0 +1,184 @@
+// Copyright 2024-2025, Amlal El Mahrouss (amlal@nekernel.org)
+// Licensed under the Apache License, Version 2.0 (see LICENSE file)
+// Official repository: https://github.com/nekernel-org/nekernel
+
+#include <ArchKit/ArchKit.h>
+#include <KernelKit/PCI/Iterator.h>
+#include <modules/ATA/ATA.h>
+
+#if defined(__ATA_DMA__)
+
+#define kATADataLen (256)
+
+using namespace Kernel;
+using namespace Kernel::HAL;
+
+/// BUGS: 0
+
+STATIC Boolean              kATADetected                  = false;
+STATIC Int32 kATADeviceType ATTRIBUTE(unused)             = kATADeviceCount;
+STATIC UInt16               kATAIdentifyData[kATADataLen] = {0};
+STATIC Kernel::PCI::Device kATADevice;
+STATIC Char                kATADiskModel[50] ATTRIBUTE(unused) = {"GENERIC DMA"};
+
+Boolean drv_std_wait_io(UInt16 IO) {
+  for (int i = 0; i < 400; i++) rt_in8(IO + ATA_REG_STATUS);
+
+ATAWaitForIO_Retry:
+  auto status_rdy = rt_in8(IO + ATA_REG_STATUS);
+
+  if ((status_rdy & ATA_SR_BSY)) goto ATAWaitForIO_Retry;
+
+ATAWaitForIO_Retry2:
+  status_rdy = rt_in8(IO + ATA_REG_STATUS);
+
+  if (status_rdy & ATA_SR_ERR) return false;
+
+  if (!(status_rdy & ATA_SR_DRDY)) goto ATAWaitForIO_Retry2;
+
+  return true;
+}
+
+Void drv_std_select(UInt16 Bus) {
+  if (Bus == ATA_PRIMARY_IO)
+    rt_out8(Bus + ATA_REG_HDDEVSEL, ATA_PRIMARY_SEL);
+  else
+    rt_out8(Bus + ATA_REG_HDDEVSEL, ATA_SECONDARY_SEL);
+}
+
+Boolean drv_std_init(UInt16 Bus, UInt8 Drive, UInt16& OutBus, UInt8& OutMaster) {
+  NE_UNUSED(Bus);
+  NE_UNUSED(Drive);
+  NE_UNUSED(OutBus);
+  NE_UNUSED(OutMaster);
+
+  PCI::Iterator iterator(Types::PciDeviceKind::MassStorageController);
+
+  for (SizeT device_index = 0; device_index < NE_BUS_COUNT; ++device_index) {
+    kATADevice = iterator[device_index].Leak();  // And then leak the reference.
+
+    /// IDE interface
+    if (kATADevice.Subclass() == 0x01) {
+      break;
+    }
+  }
+
+  return NO;
+}
+
+namespace Kernel::Detail {
+struct PRDEntry final {
+  UInt32 mAddress;
+  UInt16 mByteCount;
+  UInt16 mFlags;  /// @param PRD flags, set to 0x8000 to indicate end of prd.
+};
+}  // namespace Kernel::Detail
+
+static UIntPtr kReadAddr  = mib_cast(2);
+static UIntPtr kWriteAddr = mib_cast(2) + kib_cast(64);
+
+Void drv_std_read(UInt64 Lba, UInt16 IO, UInt8 Master, Char* Buf, SizeT SectorSz, SizeT Size) {
+  Lba /= SectorSz;
+
+  if (Size > kib_cast(64)) return;
+
+  UInt8 Command = ((!Master) ? 0xE0 : 0xF0);
+
+  rt_copy_memory((VoidPtr) Buf, (VoidPtr) kReadAddr, Size);
+
+  drv_std_select(IO);
+
+  rt_out8(IO + ATA_REG_HDDEVSEL, (Command) | (((Lba) >> 24) & 0x0F));
+
+  rt_out8(IO + ATA_REG_SEC_COUNT0, ((Size + SectorSz - 1) / SectorSz));
+
+  rt_out8(IO + ATA_REG_LBA0, (Lba) & 0xFF);
+  rt_out8(IO + ATA_REG_LBA1, (Lba) >> 8);
+  rt_out8(IO + ATA_REG_LBA2, (Lba) >> 16);
+  rt_out8(IO + ATA_REG_LBA3, (Lba) >> 24);
+
+  Kernel::Detail::PRDEntry* prd =
+      (Kernel::Detail::PRDEntry*) (kATADevice.Bar(0x20) + 4);  // The PRDEntry is not correct.
+
+  prd->mAddress   = (UInt32) (UIntPtr) kReadAddr;
+  prd->mByteCount = Size - 1;
+  prd->mFlags     = 0x8000;  // indicate the end of prd.
+
+  rt_out32(kATADevice.Bar(0x20) + 0x04, (UInt32) (UIntPtr) prd);
+
+  rt_out8(kATADevice.Bar(0x20) + ATA_REG_COMMAND, ATA_CMD_READ_DMA);
+
+  rt_out8(kATADevice.Bar(0x20) + 0x00, 0x09);  // Start DMA engine
+
+  while (rt_in8(kATADevice.Bar(0x20) + ATA_REG_STATUS) & 0x01);
+
+  rt_out8(kATADevice.Bar(0x20) + 0x00, 0x00);  // Stop DMA engine
+
+  rt_copy_memory((VoidPtr) kReadAddr, (VoidPtr) Buf, Size);
+
+  delete prd;
+  prd = nullptr;
+}
+
+Void drv_std_write(UInt64 Lba, UInt16 IO, UInt8 Master, Char* Buf, SizeT SectorSz, SizeT Size) {
+  Lba /= SectorSz;
+
+  if (Size > kib_cast(64)) return;
+
+  UInt8 Command = ((!Master) ? 0xE0 : 0xF0);
+
+  rt_copy_memory((VoidPtr) Buf, (VoidPtr) kWriteAddr, Size);
+
+  rt_out8(IO + ATA_REG_HDDEVSEL, (Command) | (((Lba) >> 24) & 0x0F));
+
+  rt_out8(IO + ATA_REG_SEC_COUNT0, ((Size + (SectorSz - 1)) / SectorSz));
+
+  rt_out8(IO + ATA_REG_LBA0, (Lba) & 0xFF);
+  rt_out8(IO + ATA_REG_LBA1, (Lba) >> 8);
+  rt_out8(IO + ATA_REG_LBA2, (Lba) >> 16);
+  rt_out8(IO + ATA_REG_LBA3, (Lba) >> 24);
+
+  Kernel::Detail::PRDEntry* prd = (Kernel::Detail::PRDEntry*) (kATADevice.Bar(0x20) + 4);
+
+  prd->mAddress   = (UInt32) (UIntPtr) kWriteAddr;
+  prd->mByteCount = Size - 1;
+  prd->mFlags     = 0x8000;
+
+  rt_out32(kATADevice.Bar(0x20) + 0x04, (UInt32) (UIntPtr) prd);
+  rt_out8(kATADevice.Bar(0x20) + ATA_REG_COMMAND, ATA_CMD_WRITE_DMA);
+
+  rt_out8(IO + 0x00, 0x09);  // Start DMA engine
+
+  while (rt_in8(kATADevice.Bar(0x20) + ATA_REG_STATUS) & 0x01);
+
+  rt_out8(kATADevice.Bar(0x20) + 0x00, 0x00);  // Stop DMA engine
+
+  delete prd;
+  prd = nullptr;
+}
+
+/***********************************************************************************/
+/// @brief Is ATA detected?
+/***********************************************************************************/
+Boolean drv_std_detected(Void) {
+  return kATADetected;
+}
+
+/***********************************************************************************/
+/***
+  @brief Gets the number of sectors inside the drive.
+  @return Number of sectors, or zero.
+*/
+/***********************************************************************************/
+Kernel::SizeT drv_std_get_sector_count() {
+  return (kATAIdentifyData[61] << 16) | kATAIdentifyData[60];
+}
+
+/***********************************************************************************/
+/// @brief Get the size of the current drive.
+/***********************************************************************************/
+Kernel::SizeT drv_std_get_size() {
+  return (drv_std_get_sector_count()) * kATASectorSize;
+}
+
+#endif /* ifdef __ATA_DMA__ */