From 20bf6ddcce3bc3b295f73e72c164049e684d6a4a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=2E=20Cemal=20=C3=96zt=C3=BCrk?= <info@yigitovski.com>
Date: Sun, 20 Aug 2023 18:08:15 +0000
Subject: [PATCH] Add MBR parser (#17)

---
 Cargo.toml      |  4 ---
 src/read_mbr.rs | 79 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 79 insertions(+), 4 deletions(-)
 create mode 100644 src/read_mbr.rs

diff --git a/Cargo.toml b/Cargo.toml
index 3da500a..9e01d05 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -3,10 +3,6 @@ name = "cuteloader"
 version = "0.1.0"
 edition = "2021"
 
-# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
-
-[dependencies]
-
 
 # panic behaviour
 
diff --git a/src/read_mbr.rs b/src/read_mbr.rs
new file mode 100644
index 0000000..534bd3a
--- /dev/null
+++ b/src/read_mbr.rs
@@ -0,0 +1,79 @@
+use x86_64::instructions::port::Port;
+
+#[derive(Debug)]
+struct PartitionEntry {
+    bootable: u8,
+    start_chs: [u8; 3],
+    partition_type: u8,
+    end_chs: [u8; 3],
+    start_lba: u32,
+    size_sectors: u32,
+}
+
+impl PartitionEntry {
+    fn new(buffer: &[u8]) -> Self {
+        Self {
+            bootable: buffer[0],
+            start_chs: [buffer[1], buffer[2], buffer[3]],
+            partition_type: buffer[4],
+            end_chs: [buffer[5], buffer[6], buffer[7]],
+            start_lba: u32::from_le_bytes([
+                buffer[8],
+                buffer[9],
+                buffer[10],
+                buffer[11],
+            ]),
+            size_sectors: u32::from_le_bytes([
+                buffer[12],
+                buffer[13],
+                buffer[14],
+                buffer[15],
+            ]),
+        }
+    }
+}
+
+fn read_disk_sector(buffer: &mut [u8], sector: u32) {
+    let ah = 0x02; // AH = 2 (Read sector)
+    let al = buffer.len() as u8;
+    let ch = (sector >> 8) as u8;
+    let cl = (sector & 0xFF) as u8;
+    let dh = 0; // DH = 0 (Head number)
+    let dl = 0x80; // DL = 0x80 (Drive number: 0x80 for first hard disk)
+
+    unsafe {
+        asm!(
+            "int 0x13",
+            in("ah") ah,
+            in("al") al,
+            in("ch") ch,
+            in("cl") cl,
+            in("dh") dh,
+            in("dl") dl,
+            inout("rcx") buffer.len() => _,
+            inout("rdi") buffer.as_mut_ptr() => _,
+        );
+    }
+}
+
+fn fetch_bootable_partitions(sector: u32) {
+    let mut mbr_buffer: [u8; 512] = [0; 512];
+    read_disk_sector(&mut mbr_buffer, sector);
+
+    let mut partition_entries: Vec<PartitionEntry> = Vec::new();
+    let mut bootable_parts: Vec<PartitionEntry> = Vec::new();
+    for i in 0..4 {
+        let offset = 446 + i * 16;
+        let entry_buffer = &mbr_buffer[offset..offset + 16];
+        let partition_entry = PartitionEntry::new(entry_buffer);
+        partition_entries.push(partition_entry);
+    }
+
+    for (i, entry) in partition_entries.iter().enumerate() {
+        if entry.bootable == 0x80 {
+		bootable_parts.push(entry)
+	}
+    }
+
+    bootable_parts
+}