Implemented partition detection

2025-07-07 05:19:17 +00:00 · 2022-11-02 04:14:06 +02:00
parent 2650202bc5
commit ca33e7b85b
4 changed files with 320 additions and 1 deletions
--- a/include/disk.hpp
+++ b/include/disk.hpp
@ -0,0 +1,137 @@
+#ifndef __FENNIX_KERNEL_DISK_H__
+#define __FENNIX_KERNEL_DISK_H__
+
+#include <types.h>
+#include <vector.hpp>
+
+namespace Disk
+{
+#define MBR_MAGIC0 0x55
+#define MBR_MAGIC1 0xAA
+
+// "EFI PART"
+#define GPT_MAGIC 0x5452415020494645ULL
+
+    enum PartitionStyle
+    {
+        Unknown,
+        MBR,
+        GPT
+    };
+
+    enum PartitionFlags
+    {
+        Present,
+        Bootable,
+        EFISystemPartition
+    };
+
+    struct MasterBootRecordPartition
+    {
+        uint8_t Flags;
+        uint8_t CHSFirst[3];
+        uint8_t Type;
+        uint8_t CHSLast[3];
+        uint32_t LBAFirst;
+        uint32_t Sectors;
+    } __attribute__((packed));
+
+    struct MasterBootRecord
+    {
+        uint8_t Bootstrap[440];
+        uint32_t UniqueID;
+        uint16_t Reserved;
+        MasterBootRecordPartition Partitions[4];
+        uint8_t Signature[2];
+    } __attribute__((packed));
+
+    struct GUIDPartitionTablePartition
+    {
+        uint64_t TypeLow;
+        uint64_t TypeHigh;
+        uint64_t GUIDLow;
+        uint64_t GUIDHigh;
+        uint64_t StartLBA;
+        uint64_t EndLBA;
+        uint64_t Attributes;
+        char Label[72];
+    } __attribute__((packed));
+
+    struct GUIDPartitionTable
+    {
+        uint64_t Signature;
+        uint32_t Revision;
+        uint32_t HdrSize;
+        uint32_t HdrCRC32;
+        uint32_t Reserved;
+        uint64_t LBA;
+        uint64_t ALTLBA;
+        uint64_t FirstBlock;
+        uint64_t LastBlock;
+        uint64_t GUIDLow;
+        uint64_t GUIDHigh;
+        uint64_t PartLBA;
+        uint32_t PartCount;
+        uint32_t EntrySize;
+        uint32_t PartCRC32;
+    } __attribute__((packed));
+
+    struct PartitionTable
+    {
+        MasterBootRecord MBR;
+        GUIDPartitionTable GPT;
+    };
+
+    class Partition
+    {
+    public:
+        char Label[72] = "Unidentified Partition";
+        uint64_t StartLBA = 0xdeadbeef;
+        uint64_t EndLBA = 0xdeadbeef;
+        uint64_t Sectors = 0xdeadbeef;
+        uint64_t Flags = 0xdeadbeef;
+        unsigned char Port = 0xdeadbeef;
+        PartitionStyle Style = PartitionStyle::Unknown;
+        size_t Index = 0;
+
+        uint64_t Read(uint64_t Offset, uint64_t Count, uint8_t *Buffer) {}
+        uint64_t Write(uint64_t Offset, uint64_t Count, uint8_t *Buffer) {}
+        Partition() {}
+        ~Partition() {}
+    };
+
+    class Drive
+    {
+    public:
+        char Name[64] = "Unidentified Drive";
+        uint8_t *Buffer = nullptr;
+        PartitionTable Table;
+        PartitionStyle Style = PartitionStyle::Unknown;
+        Vector<Partition *> Partitions;
+        bool MechanicalDisk = false;
+        uint64_t UniqueIdentifier = 0xdeadbeef;
+
+        uint64_t Read(uint64_t Offset, uint64_t Count, uint8_t *Buffer) {}
+        uint64_t Write(uint64_t Offset, uint64_t Count, uint8_t *Buffer) {}
+        Drive()
+        { // TODO: Allocate buffer
+        }
+        ~Drive() {}
+    };
+
+    class Manager
+    {
+    private:
+        unsigned char AvailablePorts = 0;
+        int BytesPerSector = 0;
+
+        Vector<Drive *> drives;
+
+    public:
+        void FetchDisks(unsigned long DriverUID);
+        Manager();
+        ~Manager();
+    };
+}
+
+#endif // !__FENNIX_KERNEL_DISK_H__