From e0024d9ea688ee91a77abc0e28c5ea24b13ca67d Mon Sep 17 00:00:00 2001
From: Amlal <amlal.elmahrouss@icloud.com>
Date: Mon, 28 Oct 2024 07:01:58 +0100
Subject: IMP: Refactor whole source code to make it even.

	- That is because previously the source was both in lowercase and lettercase.

Signed-off-by: Amlal <amlal.elmahrouss@icloud.com>
---
 dev/ZBAKit/BootKit/HW/ATA.h  | 58 ++++++++++++++++++++++++++++++++++++++++++++
 dev/ZBAKit/BootKit/HW/SATA.h | 46 +++++++++++++++++++++++++++++++++++
 2 files changed, 104 insertions(+)
 create mode 100644 dev/ZBAKit/BootKit/HW/ATA.h
 create mode 100644 dev/ZBAKit/BootKit/HW/SATA.h

(limited to 'dev/ZBAKit/BootKit/HW')

diff --git a/dev/ZBAKit/BootKit/HW/ATA.h b/dev/ZBAKit/BootKit/HW/ATA.h
new file mode 100644
index 00000000..8c7a9c40
--- /dev/null
+++ b/dev/ZBAKit/BootKit/HW/ATA.h
@@ -0,0 +1,58 @@
+/* -------------------------------------------
+
+	Copyright ZKA Web Services Co.
+
+------------------------------------------- */
+
+#pragma once
+
+#include <Modules/ATA/ATA.h>
+#include <BootKit/Device.h>
+
+using namespace Kernel;
+
+class BootDeviceATA final : public Device
+{
+public:
+	enum
+	{
+		kPrimary   = ATA_PRIMARY_IO,
+		kSecondary = ATA_SECONDARY_IO,
+	};
+
+	explicit BootDeviceATA() noexcept;
+	~BootDeviceATA() = default;
+
+	ZKA_COPY_DELETE(BootDeviceATA);
+
+	enum
+	{
+		kSectorSize = kATASectorSize
+	};
+
+	struct ATATrait final : public Device::Trait
+	{
+		UInt16	mBus{kPrimary};
+		UInt8	mMaster{0};
+		Boolean mErr{false};
+
+		operator bool()
+		{
+			return !mErr;
+		}
+	};
+
+public:
+	operator bool();
+
+	SizeT GetSectorsCount() noexcept;
+	SizeT GetDiskSize() noexcept;
+
+	BootDeviceATA& Read(Char* Buf, const SizeT& SecCount) override;
+	BootDeviceATA& Write(Char* Buf, const SizeT& SecCount) override;
+
+	ATATrait& Leak() override;
+
+private:
+	ATATrait mTrait;
+};
diff --git a/dev/ZBAKit/BootKit/HW/SATA.h b/dev/ZBAKit/BootKit/HW/SATA.h
new file mode 100644
index 00000000..f0af4650
--- /dev/null
+++ b/dev/ZBAKit/BootKit/HW/SATA.h
@@ -0,0 +1,46 @@
+/* -------------------------------------------
+
+	Copyright ZKA Web Services Co.
+
+------------------------------------------- */
+
+#pragma once
+
+#include <CompilerKit/CompilerKit.h>
+#include <Modules/AHCI/AHCI.h>
+
+class BootDeviceSATA final
+{
+public:
+	explicit BootDeviceSATA() noexcept;
+	~BootDeviceSATA() = default;
+
+	ZKA_COPY_DEFAULT(BootDeviceSATA);
+
+	struct SATATrait final
+	{
+		Kernel::SizeT	mBase{1024};
+		Kernel::Boolean mErr{false};
+		Kernel::Boolean mDetected{false};
+
+		operator bool()
+		{
+			return !this->mErr;
+		}
+	};
+
+	operator bool()
+	{
+		return this->Leak().mDetected;
+	}
+
+	BootDeviceSATA& Read(Kernel::WideChar* Buf, const Kernel::SizeT& SecCount);
+	BootDeviceSATA& Write(Kernel::WideChar* Buf, const Kernel::SizeT& SecCount);
+
+	SATATrait& Leak();
+
+private:
+	SATATrait mTrait;
+};
+
+#define kAHCISectorSz 4096
-- 
cgit v1.2.3