summaryrefslogtreecommitdiffhomepage
path: root/dev/kernel/src/DriveMgr.cc
diff options
context:
space:
mode:
Diffstat (limited to 'dev/kernel/src/DriveMgr.cc')
-rw-r--r--dev/kernel/src/DriveMgr.cc318
1 files changed, 148 insertions, 170 deletions
diff --git a/dev/kernel/src/DriveMgr.cc b/dev/kernel/src/DriveMgr.cc
index b5a8e9e4..763096b3 100644
--- a/dev/kernel/src/DriveMgr.cc
+++ b/dev/kernel/src/DriveMgr.cc
@@ -1,16 +1,16 @@
/* -------------------------------------------
- Copyright (C) 2024-2025, Amlal El Mahrouss, all rights reserved.
+ Copyright (C) 2024-2025, Amlal El Mahrouss, all rights reserved.
------------------------------------------- */
+#include <FirmwareKit/EPM.h>
+#include <FirmwareKit/GPT.h>
#include <KernelKit/DebugOutput.h>
#include <KernelKit/DriveMgr.h>
#include <NewKit/Utils.h>
-#include <FirmwareKit/EPM.h>
-#include <FirmwareKit/GPT.h>
-#include <modules/ATA/ATA.h>
#include <modules/AHCI/AHCI.h>
+#include <modules/ATA/ATA.h>
#include <modules/NVME/NVME.h>
/***********************************************************************************/
@@ -18,233 +18,211 @@
/// @brief Drive Manager of kernel.
/***********************************************************************************/
-namespace Kernel
-{
+namespace Kernel {
#if defined(__ATA_PIO__) || defined(__ATA_DMA__)
- STATIC UInt16 kATAIO = 0U;
- STATIC UInt8 kATAMaster = 0U;
+STATIC UInt16 kATAIO = 0U;
+STATIC UInt8 kATAMaster = 0U;
#endif
#if defined(__AHCI__)
- STATIC UInt16 kAHCIPortsImplemented = 0UL;
+STATIC UInt16 kAHCIPortsImplemented = 0UL;
#endif
- /// @brief reads from an ATA drive.
- /// @param pckt Packet structure (fPacketContent must be non null)
- /// @return
- Void io_drv_input(DriveTrait::DrivePacket pckt)
- {
+/// @brief reads from an ATA drive.
+/// @param pckt Packet structure (fPacketContent must be non null)
+/// @return
+Void io_drv_input(DriveTrait::DrivePacket pckt) {
#ifdef __AHCI__
- drv_std_read(pckt.fPacketLba, (Char*)pckt.fPacketContent, kAHCISectorSize, pckt.fPacketSize);
+ drv_std_read(pckt.fPacketLba, (Char*) pckt.fPacketContent, kAHCISectorSize, pckt.fPacketSize);
#elif defined(__ATA_PIO__) || defined(__ATA_DMA__)
- drv_std_read(pckt.fPacketLba, kATAIO, kATAMaster, (Char*)pckt.fPacketContent, kATASectorSize, pckt.fPacketSize);
+ drv_std_read(pckt.fPacketLba, kATAIO, kATAMaster, (Char*) pckt.fPacketContent, 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.fPacketReadOnly)
- {
- pckt.fPacketGood = NO;
- return;
- }
+}
+
+/// @brief Writes to an ATA drive.
+/// @param pckt the packet to write.
+/// @return
+Void io_drv_output(DriveTrait::DrivePacket pckt) {
+ if (pckt.fPacketReadOnly) {
+ pckt.fPacketGood = NO;
+ return;
+ }
#ifdef __AHCI__
- drv_std_write(pckt.fPacketLba, (Char*)pckt.fPacketContent, kAHCISectorSize, pckt.fPacketSize);
+ drv_std_write(pckt.fPacketLba, (Char*) pckt.fPacketContent, kAHCISectorSize, pckt.fPacketSize);
#elif defined(__ATA_PIO__) || defined(__ATA_DMA__)
- drv_std_write(pckt.fPacketLba, kATAIO, kATAMaster, (Char*)pckt.fPacketContent, kATASectorSize, pckt.fPacketSize);
+ drv_std_write(pckt.fPacketLba, kATAIO, kATAMaster, (Char*) pckt.fPacketContent, 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)
- {
- NE_UNUSED(pckt);
+/// @brief Executes a disk check on the ATA drive.
+/// @param pckt the packet to read.
+/// @return
+Void io_drv_init(DriveTrait::DrivePacket pckt) {
+ NE_UNUSED(pckt);
#if defined(__ATA_PIO__) || defined(__ATA_DMA__)
- kATAMaster = 0;
- kATAIO = 0;
+ kATAMaster = 0;
+ kATAIO = 0;
- kATAMaster = YES;
- kATAIO = ATA_PRIMARY_IO;
+ kATAMaster = YES;
+ kATAIO = ATA_PRIMARY_IO;
- if (drv_std_init(kATAIO, kATAMaster, kATAIO, kATAMaster))
- {
- pckt.fPacketGood = YES;
- return;
- }
+ if (drv_std_init(kATAIO, kATAMaster, kATAIO, kATAMaster)) {
+ pckt.fPacketGood = YES;
+ return;
+ }
- kATAMaster = NO;
- kATAIO = ATA_SECONDARY_IO;
+ kATAMaster = NO;
+ kATAIO = ATA_SECONDARY_IO;
- if (!drv_std_init(kATAIO, kATAMaster, kATAIO, kATAMaster))
- {
- pckt.fPacketGood = YES;
- return;
- }
+ if (!drv_std_init(kATAIO, kATAMaster, kATAIO, kATAMaster)) {
+ pckt.fPacketGood = YES;
+ return;
+ }
- pckt.fPacketGood = YES;
+ pckt.fPacketGood = YES;
#elif defined(__AHCI__)
- kAHCIPortsImplemented = 0;
+ kAHCIPortsImplemented = 0;
- if (drv_std_init(kAHCIPortsImplemented))
- {
- pckt.fPacketGood = YES;
- }
-#endif // if defined(__ATA_PIO__) || defined (__ATA_DMA__)
- }
+ if (drv_std_init(kAHCIPortsImplemented)) {
+ pckt.fPacketGood = YES;
+ }
+#endif // if defined(__ATA_PIO__) || defined (__ATA_DMA__)
+}
/// @brief Gets the drive kind (ATA, SCSI, AHCI...)
/// @param void no arguments.
/// @return no arguments.
#ifdef __ATA_PIO__
- const Char* io_drv_kind(Void)
- {
- return "ATA-PIO";
- }
+const Char* io_drv_kind(Void) {
+ return "ATA-PIO";
+}
#elif defined(__ATA_DMA__)
- const Char* io_drv_kind(Void)
- {
- return "ATA-DMA";
- }
+const Char* io_drv_kind(Void) {
+ return "ATA-DMA";
+}
#elif defined(__AHCI__)
- const Char* io_drv_kind(Void)
- {
- return "AHCI";
- }
+const Char* io_drv_kind(Void) {
+ return "AHCI";
+}
#else
- const Char* io_drv_kind(Void)
- {
- return "null";
- }
+const Char* io_drv_kind(Void) {
+ return "null";
+}
#endif
- /// @brief Unimplemented drive function.
- /// @param pckt the packet to read.
- Void io_drv_unimplemented(DriveTrait::DrivePacket pckt) noexcept
- {
- NE_UNUSED(pckt);
- }
+/// @brief Unimplemented drive function.
+/// @param pckt the packet to read.
+Void io_drv_unimplemented(DriveTrait::DrivePacket pckt) noexcept {
+ NE_UNUSED(pckt);
+}
- /// @brief Makes a new drive.
- /// @return the new blank drive.
- DriveTrait io_construct_blank_drive() noexcept
- {
- DriveTrait trait;
+/// @brief Makes a new drive.
+/// @return the new blank drive.
+DriveTrait io_construct_blank_drive() noexcept {
+ DriveTrait trait;
- constexpr auto kBlankDrive = "/media/blank/";
+ constexpr auto kBlankDrive = "/media/blank/";
- rt_copy_memory((VoidPtr)kBlankDrive, trait.fName, rt_string_len(kBlankDrive));
- trait.fKind = kInvalidDrive;
+ rt_copy_memory((VoidPtr) kBlankDrive, trait.fName, rt_string_len(kBlankDrive));
+ trait.fKind = kInvalidDrive;
- trait.fInput = io_drv_unimplemented;
- trait.fOutput = io_drv_unimplemented;
- trait.fVerify = io_drv_unimplemented;
- trait.fInit = io_drv_unimplemented;
- trait.fProtocol = io_drv_kind;
+ trait.fInput = io_drv_unimplemented;
+ trait.fOutput = io_drv_unimplemented;
+ trait.fVerify = io_drv_unimplemented;
+ trait.fInit = io_drv_unimplemented;
+ trait.fProtocol = io_drv_kind;
- kout << "Construct: " << trait.fName << "\r";
+ kout << "Construct: " << trait.fName << "\r";
- return trait;
- }
+ return trait;
+}
- namespace Detail
- {
- Void io_detect_drive(DriveTrait& trait)
- {
- trait.fInit(trait.fPacket);
+namespace Detail {
+ Void io_detect_drive(DriveTrait& trait) {
+ trait.fInit(trait.fPacket);
- EPM_PART_BLOCK block_struct;
+ EPM_PART_BLOCK block_struct;
- trait.fPacket.fPacketLba = kEPMBootBlockLba;
- trait.fPacket.fPacketSize = sizeof(EPM_PART_BLOCK);
- trait.fPacket.fPacketContent = &block_struct;
+ trait.fPacket.fPacketLba = kEPMBootBlockLba;
+ trait.fPacket.fPacketSize = sizeof(EPM_PART_BLOCK);
+ trait.fPacket.fPacketContent = &block_struct;
- rt_copy_memory((VoidPtr) "fs/detect-packet", trait.fPacket.fPacketMime,
- rt_string_len("fs/detect-packet"));
+ rt_copy_memory((VoidPtr) "fs/detect-packet", trait.fPacket.fPacketMime,
+ rt_string_len("fs/detect-packet"));
- trait.fInput(trait.fPacket);
+ trait.fInput(trait.fPacket);
- if (rt_string_cmp(block_struct.Magic, kEPMMagic, kEPMMagicLength) == 0)
- {
- trait.fPacket.fPacketReadOnly = NO;
- trait.fKind = kMassStorageDrive | kEPMDrive;
+ if (rt_string_cmp(block_struct.Magic, kEPMMagic, kEPMMagicLength) == 0) {
+ trait.fPacket.fPacketReadOnly = NO;
+ trait.fKind = kMassStorageDrive | kEPMDrive;
- kout << "Disk is EPM formatted.\r";
+ kout << "Disk is EPM formatted.\r";
- trait.fSectorSz = block_struct.SectorSz;
- trait.fLbaEnd = block_struct.LbaEnd;
- trait.fLbaStart = block_struct.LbaStart;
- }
- else
- {
- GPT_PARTITION_TABLE gpt_struct;
+ trait.fSectorSz = block_struct.SectorSz;
+ trait.fLbaEnd = block_struct.LbaEnd;
+ trait.fLbaStart = block_struct.LbaStart;
+ } else {
+ GPT_PARTITION_TABLE gpt_struct;
- trait.fPacket.fPacketLba = kEPMBootBlockLba;
- trait.fPacket.fPacketSize = sizeof(GPT_PARTITION_TABLE);
- trait.fPacket.fPacketContent = &gpt_struct;
+ trait.fPacket.fPacketLba = kEPMBootBlockLba;
+ trait.fPacket.fPacketSize = sizeof(GPT_PARTITION_TABLE);
+ trait.fPacket.fPacketContent = &gpt_struct;
- rt_copy_memory((VoidPtr) "fs/detect-packet", trait.fPacket.fPacketMime,
- rt_string_len("fs/detect-packet"));
+ rt_copy_memory((VoidPtr) "fs/detect-packet", trait.fPacket.fPacketMime,
+ rt_string_len("fs/detect-packet"));
- trait.fInput(trait.fPacket);
+ trait.fInput(trait.fPacket);
- if (rt_string_cmp(gpt_struct.Signature, kMagicGPT, kMagicLenGPT) == 0)
- {
- trait.fPacket.fPacketReadOnly = NO;
- trait.fKind = kMassStorageDrive | kGPTDrive;
+ if (rt_string_cmp(gpt_struct.Signature, kMagicGPT, kMagicLenGPT) == 0) {
+ trait.fPacket.fPacketReadOnly = NO;
+ trait.fKind = kMassStorageDrive | kGPTDrive;
- kout << "Disk is GPT formatted.\r";
+ kout << "Disk is GPT formatted.\r";
- trait.fSectorSz = gpt_struct.SizeOfEntries;
- trait.fLbaEnd = gpt_struct.LastGPTEntry;
- trait.fLbaStart = gpt_struct.FirstGPTEntry;
- }
- else
- {
- kout << "Disk is unformatted.\r";
+ trait.fSectorSz = gpt_struct.SizeOfEntries;
+ trait.fLbaEnd = gpt_struct.LastGPTEntry;
+ trait.fLbaStart = gpt_struct.FirstGPTEntry;
+ } else {
+ kout << "Disk is unformatted.\r";
- trait.fPacket.fPacketReadOnly = YES;
- trait.fKind = kMassStorageDrive | kUnformattedDrive | kReadOnlyDrive;
- }
- }
+ trait.fPacket.fPacketReadOnly = YES;
+ trait.fKind = kMassStorageDrive | kUnformattedDrive | kReadOnlyDrive;
+ }
+ }
- rt_copy_memory((VoidPtr) "*/*", trait.fPacket.fPacketMime,
- rt_string_len("*/*"));
+ rt_copy_memory((VoidPtr) "*/*", trait.fPacket.fPacketMime, rt_string_len("*/*"));
- trait.fPacket.fPacketLba = 0;
- trait.fPacket.fPacketSize = 0UL;
- trait.fPacket.fPacketContent = nullptr;
- }
- } // namespace Detail
+ trait.fPacket.fPacketLba = 0;
+ trait.fPacket.fPacketSize = 0UL;
+ trait.fPacket.fPacketContent = nullptr;
+ }
+} // namespace Detail
- /// @brief Fetches the main drive.
- /// @return the new drive. (returns kEPMDrive if EPM formatted)
- DriveTrait io_construct_main_drive() noexcept
- {
- DriveTrait trait;
+/// @brief Fetches the main drive.
+/// @return the new drive. (returns kEPMDrive if EPM formatted)
+DriveTrait io_construct_main_drive() noexcept {
+ DriveTrait trait;
- constexpr auto kMainDrive = "/media/main/";
+ constexpr auto kMainDrive = "/media/main/";
- rt_copy_memory((VoidPtr)kMainDrive, trait.fName, rt_string_len(kMainDrive));
+ rt_copy_memory((VoidPtr) kMainDrive, trait.fName, rt_string_len(kMainDrive));
- MUST_PASS(trait.fName[0] != 0);
+ MUST_PASS(trait.fName[0] != 0);
- trait.fVerify = io_drv_unimplemented;
- trait.fOutput = io_drv_output;
- trait.fInput = io_drv_input;
- trait.fInit = io_drv_init;
- trait.fProtocol = io_drv_kind;
+ trait.fVerify = io_drv_unimplemented;
+ trait.fOutput = io_drv_output;
+ trait.fInput = io_drv_input;
+ trait.fInit = io_drv_init;
+ trait.fProtocol = io_drv_kind;
- kout << "Detecting partition scheme of: " << trait.fName << ".\r";
+ kout << "Detecting partition scheme of: " << trait.fName << ".\r";
- Detail::io_detect_drive(trait);
+ Detail::io_detect_drive(trait);
- return trait;
- }
-} // namespace Kernel
+ return trait;
+}
+} // namespace Kernel
generated by cgit v1.2.3 (git 2.39.1) at 2026-03-13 04:55:42 +0000