lunaix-os/hal/ahci/ata.c

   1 #include <hal/ahci/hba.h>
   2 #include <hal/ahci/sata.h>
   3
   4 #include <lunaix/mm/valloc.h>
   5 #include <lunaix/mm/vmm.h>
   6 #include <lunaix/spike.h>
   7
   8 int
   9 __sata_buffer_io(struct hba_port* port,
  10                  uint64_t lba,
  11                  void* buffer,
  12                  uint32_t size,
  13                  int write)
  14 {
  15     assert_msg(((uintptr_t)buffer & 0x3) == 0, "HBA: Bad buffer alignment");
  16
  17     struct hba_cmdh* header;
  18     struct hba_cmdt* table;
  19     int slot = hba_alloc_slot(port, &table, &header, 0);
  20     int bitmask = 1 << slot;
  21
  22     // 确保端口是空闲的
  23     wait_until(!(port->regs[HBA_RPxTFD] & (HBA_PxTFD_BSY | HBA_PxTFD_DRQ)));
  24
  25     port->regs[HBA_RPxIS] = 0;
  26
  27     table->entries[0] = (struct hba_prdte){ .byte_count = size - 1,
  28                                             .data_base = vmm_v2p(buffer) };
  29     header->prdt_len = 1;
  30     header->options |= HBA_CMDH_WRITE * (write == 1);
  31
  32     uint16_t count = ICEIL(size, port->device->block_size);
  33     struct sata_reg_fis* fis = table->command_fis;
  34
  35     if ((port->device->flags & HBA_DEV_FEXTLBA)) {
  36         // 如果该设备支持48位LBA寻址
  37         sata_create_fis(
  38           fis, write ? ATA_WRITE_DMA_EXT : ATA_READ_DMA_EXT, lba, count);
  39         fis->dev = (1 << 6) * (!write);
  40     } else {
  41         sata_create_fis(fis, write ? ATA_WRITE_DMA : ATA_READ_DMA, lba, count);
  42     }
  43
  44     int retries = 0;
  45
  46     while (retries < MAX_RETRY) {
  47         port->regs[HBA_RPxCI] = bitmask;
  48
  49         wait_until(!(port->regs[HBA_RPxCI] & bitmask));
  50
  51         if ((port->regs[HBA_RPxTFD] & HBA_PxTFD_ERR)) {
  52             // 有错误
  53             sata_read_error(port);
  54             retries++;
  55         } else {
  56             vfree_dma(table);
  57             return 1;
  58         }
  59     }
  60
  61 fail:
  62     vfree_dma(table);
  63     return 0;
  64 }
  65
  66 int
  67 sata_read_buffer(struct hba_port* port,
  68                  uint64_t lba,
  69                  void* buffer,
  70                  uint32_t size)
  71 {
  72     return __sata_buffer_io(port, lba, buffer, size, 0);
  73 }
  74
  75 int
  76 sata_write_buffer(struct hba_port* port,
  77                   uint64_t lba,
  78                   void* buffer,
  79                   uint32_t size)
  80 {
  81     return __sata_buffer_io(port, lba, buffer, size, 1);
  82 }
  83
  84 void
  85 sata_read_error(struct hba_port* port)
  86 {
  87     uint32_t tfd = port->regs[HBA_RPxTFD];
  88     port->device->last_error = (tfd >> 8) & 0xff;
  89 }