lunaix-os/hal/ahci/utils.c

   1 #include <hal/ahci/ahci.h>
   2 #include <hal/ahci/hba.h>
   3 #include <hal/ahci/sata.h>
   4 #include <hal/ahci/scsi.h>
   5 #include <klibc/string.h>
   6
   7 #include <lunaix/spike.h>
   8
   9 #define IDDEV_OFFMAXLBA 60
  10 #define IDDEV_OFFMAXLBA_EXT 230
  11 #define IDDEV_OFFLSECSIZE 117
  12 #define IDDEV_OFFWWN 108
  13 #define IDDEV_OFFSERIALNUM 10
  14 #define IDDEV_OFFMODELNUM 27
  15 #define IDDEV_OFFADDSUPPORT 69
  16 #define IDDEV_OFFA48SUPPORT 83
  17 #define IDDEV_OFFALIGN 209
  18 #define IDDEV_OFFLPP 106
  19 #define IDDEV_OFFCAPABILITIES 49
  20
  21 static u32_t cdb_size[] = { SCSI_CDB12, SCSI_CDB16, 0, 0 };
  22
  23 void
  24 ahci_parse_dev_info(struct hba_device* dev_info, uint16_t* data)
  25 {
  26     dev_info->max_lba = *((u32_t*)(data + IDDEV_OFFMAXLBA));
  27     dev_info->block_size = *((u32_t*)(data + IDDEV_OFFLSECSIZE));
  28     dev_info->cbd_size = cdb_size[(*data & 0x3)];
  29     dev_info->wwn = *(uint64_t*)(data + IDDEV_OFFWWN);
  30     dev_info->block_per_sec = 1 << (*(data + IDDEV_OFFLPP) & 0xf);
  31     dev_info->alignment_offset = *(data + IDDEV_OFFALIGN) & 0x3fff;
  32     dev_info->capabilities = *((u32_t*)(data + IDDEV_OFFCAPABILITIES));
  33
  34     if (!dev_info->block_size) {
  35         dev_info->block_size = 512;
  36     }
  37
  38     if ((*(data + IDDEV_OFFADDSUPPORT) & 0x8) &&
  39         (*(data + IDDEV_OFFA48SUPPORT) & 0x400)) {
  40         dev_info->max_lba = *((uint64_t*)(data + IDDEV_OFFMAXLBA_EXT));
  41         dev_info->flags |= HBA_DEV_FEXTLBA;
  42     }
  43
  44     ahci_parsestr(dev_info->serial_num, data + IDDEV_OFFSERIALNUM, 10);
  45     ahci_parsestr(dev_info->model, data + IDDEV_OFFMODELNUM, 20);
  46 }
  47
  48 void
  49 ahci_parsestr(char* str, uint16_t* reg_start, int size_word)
  50 {
  51     int j = 0;
  52     for (int i = 0; i < size_word; i++, j += 2) {
  53         uint16_t reg = *(reg_start + i);
  54         str[j] = (char)(reg >> 8);
  55         str[j + 1] = (char)(reg & 0xff);
  56     }
  57     str[j - 1] = '\0';
  58     strrtrim(str);
  59 }
  60
  61 int
  62 ahci_try_send(struct hba_port* port, int slot)
  63 {
  64     int retries = 0, bitmask = 1 << slot;
  65
  66     // 确保端口是空闲的
  67     wait_until(!(port->regs[HBA_RPxTFD] & (HBA_PxTFD_BSY | HBA_PxTFD_DRQ)));
  68
  69     hba_clear_reg(port->regs[HBA_RPxIS]);
  70
  71     while (retries < MAX_RETRY) {
  72         // PxCI寄存器置位，告诉HBA这儿有个数据需要发送到SATA端口
  73         port->regs[HBA_RPxCI] = bitmask;
  74
  75         wait_until_expire(!(port->regs[HBA_RPxCI] & bitmask), 1000000);
  76
  77         port->regs[HBA_RPxCI] &= ~bitmask; // ensure CI bit is cleared
  78         if ((port->regs[HBA_RPxTFD] & HBA_PxTFD_ERR)) {
  79             // 有错误
  80             sata_read_error(port);
  81             retries++;
  82         } else {
  83             break;
  84         }
  85     }
  86
  87     hba_clear_reg(port->regs[HBA_RPxIS]);
  88
  89     return retries < MAX_RETRY;
  90 }
  91
  92 void
  93 ahci_post(struct hba_port* port, struct hba_cmd_state* state, int slot)
  94 {
  95     int bitmask = 1 << slot;
  96
  97     // 确保端口是空闲的
  98     wait_until(!(port->regs[HBA_RPxTFD] & (HBA_PxTFD_BSY | HBA_PxTFD_DRQ)));
  99
 100     hba_clear_reg(port->regs[HBA_RPxIS]);
 101
 102     port->cmdctx.issued[slot] = state;
 103     port->cmdctx.tracked_ci |= bitmask;
 104     port->regs[HBA_RPxCI] |= bitmask;
 105 }