From f889bb249dbc1af818f479505b2703cfb301e817 Mon Sep 17 00:00:00 2001
From: acevest <zhaoyanbai@126.com>
Date: Wed, 17 May 2023 22:58:28 +0800
Subject: [PATCH] =?utf8?q?=E7=A1=AC=E7=9B=98=E6=8F=90=E7=A4=BA=E4=BF=A1?=
 =?utf8?q?=E6=81=AF=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 Makefile      |  4 +++-
 drivers/ata.c | 16 ++++++++--------
 kernel/exec.c |  4 ++--
 qemu.sh       |  1 +
 4 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/Makefile b/Makefile
index c4903f0..70fc756 100644
--- a/Makefile
+++ b/Makefile
@@ -23,7 +23,9 @@ else ifeq ($(OS), Linux)
 endif
 
 
-CFLAGS		= -g -c -fno-builtin -m32 -DBUILDER='"$(shell whoami)"' -DFIX_SYSENTER_ESP_MODE=1
+CFLAGS		= -g -c -fno-builtin -m32 -DBUILDER='"$(shell whoami)"'
+CFLAGS     += -DFIX_SYSENTER_ESP_MODE=1
+#CFLAGS     += -DPCI_RW_ALIGN_MODE
 SYSTEMMAP	= System.map
 KERNELBIN	= KERNEL.BIN
 LINKSCRIPT	= scripts/link.ld
diff --git a/drivers/ata.c b/drivers/ata.c
index 3c21625..a981df9 100644
--- a/drivers/ata.c
+++ b/drivers/ata.c
@@ -75,14 +75,13 @@ void ide_ata_init() {
 
         uint8_t status = inb(REG_STATUS(dev));
         if (status == 0 || (status & ATA_STATUS_ERR) || (status & ATA_STATUS_RDY == 0)) {
-            printk("ata[%d] not exists: %x\n", i, status);
             ide_drives[i].present = 0;
             continue;
         } else {
             ide_drives[i].present = 1;
         }
 
-        printk("ata[%d] exists: %x\n", i, status);
+        printk("ata[%d] status %x %s exists\n", i, status, ide_drives[i].present == 1 ? "" : "not");
         insl(REG_DATA(dev), identify, SECT_SIZE / sizeof(uint32_t));
 
         // 第49个word的第8个bit位表示是否支持DMA
@@ -90,17 +89,18 @@ void ide_ata_init() {
         // 第100~103个word的八个字节表示user的LBA最大值
         printk("%04x %04x %d %d\n", identify[49], 1 << 8, identify[49] & (1 << 8), (identify[49] & (1 << 8)) != 0);
         if ((identify[49] & (1 << 8)) != 0) {
-            printk("support DMA\n");
             ide_drives[i].dma = 1;
         }
 
+        u64 max_lba = *(u64 *)(identify + 100);
+
         if ((identify[83] & (1 << 10)) != 0) {
-            printk("support LBA48\n");
             ide_drives[i].lba48 = 1;
-            u64 lba = *(u64 *)(identify + 100);
-            ide_drives[i].max_lba = lba;
-            printk("hard disk size: %u MB\n", (lba * 512) >> 20);
+            ide_drives[i].max_lba = max_lba;
         }
+
+        printk("hard disk %s %s size: %u MB\n", ide_drives[i].dma == 1 ? "DMA" : "",
+               ide_drives[i].lba48 == 1 ? "LBA48" : "LBA28", (max_lba * 512) >> 20);
     }
 }
 
@@ -281,7 +281,7 @@ int ata_pio_read_ext(int dev, uint64_t pos, uint16_t count, int timeout, void *d
     // 先写扇区数的高字节
     outb((count >> 8) & 0xFF, REG_NSECTOR(dev));
 
-    // 接着写LBA48，高三个字节
+    // 接着写LBA48，高三个字节q
     outb((pos >> 24) & 0xFF, REG_LBAL(dev));
     outb((pos >> 32) & 0xFF, REG_LBAM(dev));
     outb((pos >> 40) & 0xFF, REG_LBAH(dev));
diff --git a/kernel/exec.c b/kernel/exec.c
index b2afffa..cac3df4 100644
--- a/kernel/exec.c
+++ b/kernel/exec.c
@@ -28,8 +28,8 @@ void put_paging(unsigned long vaddr, unsigned long paddr, unsigned long flags) {
     assert(PAGE_ALIGN(vaddr) == vaddr);
     assert(PAGE_ALIGN(paddr) == paddr);
 
-    unsigned int npde = get_npd(vaddr);
-    unsigned int npte = get_npt(vaddr);
+    unsigned int npde = get_npde(vaddr);
+    unsigned int npte = get_npte(vaddr);
 
     pde_t *page_dir = (pde_t *)pa2va(current->cr3);
     pte_t *page_table = (pte_t *)PAGE_ALIGN(page_dir[npde]);
diff --git a/qemu.sh b/qemu.sh
index 83cecf8..4f7cb2c 100755
--- a/qemu.sh
+++ b/qemu.sh
@@ -1,4 +1,5 @@
 qemu-system-i386 \
+    -device ich9-ahci,id=ahci \
     -boot d \
     -drive file=HD.IMG,format=raw,index=0,media=disk \
     -drive file=kernel.iso,index=1,media=cdrom \
-- 
2.44.0