X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/202a8b1f31fa458e96ad7f20ae056d200b31b029..0172e34a3d2f50358f1bd6ebbbb79555afb082bc:/lunaix-os/hal/ahci/utils.c?ds=sidebyside

diff --git a/lunaix-os/hal/ahci/utils.c b/lunaix-os/hal/ahci/utils.c
index 9af9cfe..e101efd 100644
--- a/lunaix-os/hal/ahci/utils.c
+++ b/lunaix-os/hal/ahci/utils.c
@@ -2,24 +2,39 @@
 #include <hal/ahci/utils.h>
 #include <klibc/string.h>
 
-#define IDDEV_MAXLBA_OFFSET 60
-#define IDDEV_LSECSIZE_OFFSET 117
-#define IDDEV_WWN_OFFSET 108
-#define IDDEV_SERIALNUM_OFFSET 10
-#define IDDEV_MODELNUM_OFFSET 27
+#define IDDEV_OFFMAXLBA 60
+#define IDDEV_OFFMAXLBA_EXT 230
+#define IDDEV_OFFLSECSIZE 117
+#define IDDEV_OFFWWN 108
+#define IDDEV_OFFSERIALNUM 10
+#define IDDEV_OFFMODELNUM 27
+#define IDDEV_OFFADDSUPPORT 69
+#define IDDEV_OFFALIGN 209
+#define IDDEV_OFFLPP 106
+#define IDDEV_OFFCAPABILITIES 49
 
 void
 ahci_parse_dev_info(struct hba_device* dev_info, uint16_t* data)
 {
-    dev_info->max_lba = *((uint32_t*)(data + IDDEV_MAXLBA_OFFSET));
-    dev_info->block_size = *((uint32_t*)(data + IDDEV_LSECSIZE_OFFSET));
-    dev_info->cbd_size = (*data & 0x3) ? 12 : 16;
-    memcpy(&dev_info->wwn, (uint8_t*)(data + IDDEV_WWN_OFFSET), 8);
+    dev_info->max_lba = *((uint32_t*)(data + IDDEV_OFFMAXLBA));
+    dev_info->block_size = *((uint32_t*)(data + IDDEV_OFFLSECSIZE));
+    dev_info->cbd_size = (*data & 0x3) ? 16 : 12;
+    dev_info->wwn = *(uint64_t*)(data + IDDEV_OFFWWN);
+    dev_info->block_per_sec = 1 << (*(data + IDDEV_OFFLPP) & 0xf);
+    dev_info->alignment_offset = *(data + IDDEV_OFFALIGN) & 0x3fff;
+    dev_info->capabilities = *((uint32_t*)(data + IDDEV_OFFCAPABILITIES));
+
     if (!dev_info->block_size) {
         dev_info->block_size = 512;
     }
-    ahci_parsestr(&dev_info->serial_num, data + IDDEV_SERIALNUM_OFFSET, 10);
-    ahci_parsestr(&dev_info->model, data + IDDEV_MODELNUM_OFFSET, 20);
+
+    if ((*(data + IDDEV_OFFADDSUPPORT) & 0x8)) {
+        dev_info->max_lba = *((uint64_t*)(data + IDDEV_OFFMAXLBA_EXT));
+        dev_info->flags |= HBA_DEV_FEXTLBA;
+    }
+
+    ahci_parsestr(&dev_info->serial_num, data + IDDEV_OFFSERIALNUM, 10);
+    ahci_parsestr(&dev_info->model, data + IDDEV_OFFMODELNUM, 20);
 }
 
 void