X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/c4d08ac5260bb26db10bcfd9dc6fd2db60efebe9..2804ceee4ce65d49e9d52d0ed817317c0ac9efdc:/lunaix-os/hal/ahci/ahci.c diff --git a/lunaix-os/hal/ahci/ahci.c b/lunaix-os/hal/ahci/ahci.c index 41ebc86..f6f2718 100644 --- a/lunaix-os/hal/ahci/ahci.c +++ b/lunaix-os/hal/ahci/ahci.c @@ -16,6 +16,7 @@ #include #include #include +#include #include #include #include @@ -26,21 +27,37 @@ #define HBA_FIS_SIZE 256 #define HBA_CLB_SIZE 1024 +#define HBA_MY_IE (HBA_PxINTR_DHR | HBA_PxINTR_TFEE) + // #define DO_HBA_FULL_RESET LOG_MODULE("AHCI") -static struct ahci_hba hba; +struct ahci_hba hba; -void +static char sata_ifs[][20] = { "Not detected", + "SATA I (1.5Gbps)", + "SATA II (3.0Gbps)", + "SATA III (6.0Gbps)" }; + +extern void +ahci_fsexport(struct block_dev* bdev, void* fs_node); + +extern void __ahci_hba_isr(const isr_param* param); +extern void +__ahci_blkio_handler(struct blkio_req* req); + int ahci_init_device(struct hba_port* port); void achi_register_ops(struct hba_port* port); +void +ahci_register_device(struct hba_device* hbadev); + unsigned int ahci_get_port_usage() { @@ -78,9 +95,8 @@ ahci_init() struct pci_device* ahci_dev = pci_get_device_by_class(AHCI_HBA_CLASS); assert_msg(ahci_dev, "AHCI: Not found."); - uintptr_t bar6, size; - size = pci_bar_sizing(ahci_dev, &bar6, 6); - assert_msg(bar6 && PCI_BAR_MMIO(bar6), "AHCI: BAR#6 is not MMIO."); + struct pci_base_addr* bar6 = &ahci_dev->bar[5]; + assert_msg(bar6->type & BAR_TYPE_MMIO, "AHCI: BAR#6 is not MMIO."); pci_reg_t cmd = pci_read_cspace(ahci_dev->cspace_base, PCI_REG_STATUS_CMD); @@ -89,12 +105,11 @@ ahci_init() pci_write_cspace(ahci_dev->cspace_base, PCI_REG_STATUS_CMD, cmd); - pci_setup_msi(ahci_dev, AHCI_HBA_IV); - intr_subscribe(AHCI_HBA_IV, __ahci_hba_isr); + pci_setup_msi(ahci_dev, isrm_ivexalloc(__ahci_hba_isr)); memset(&hba, 0, sizeof(hba)); - hba.base = (hba_reg_t*)ioremap(PCI_BAR_ADDR_MM(bar6), size); + hba.base = (hba_reg_t*)ioremap(bar6->start, bar6->size); #ifdef DO_HBA_FULL_RESET // 重置HBA @@ -174,26 +189,27 @@ ahci_init() continue; } - kprintf(KINFO "sata%d: %s (%s)\n", + struct hba_device* hbadev = port->device; + kprintf(KINFO "sata%d: %s, blk_size=%d, blk=0..%d\n", i, - port->device->model, - port->device->serial_num); + hbadev->model, + hbadev->block_size, + (uint32_t)hbadev->max_lba); - block_mount_disk(port->device); + ahci_register_device(hbadev); } } -char sata_ifs[][20] = { "Not detected", - "SATA I (1.5Gbps)", - "SATA II (3.0Gbps)", - "SATA III (6.0Gbps)" }; - void -__ahci_hba_isr(const isr_param* param) +ahci_register_device(struct hba_device* hbadev) { - // TODO: clear the interrupt status - // TODO: I/O-operation scheduler should be here - kprintf(KDEBUG "HBA INTR\n"); + struct block_dev* bdev = + block_alloc_dev(hbadev->model, hbadev, __ahci_blkio_handler); + + bdev->end_lba = hbadev->max_lba; + bdev->blk_size = hbadev->block_size; + + block_mount(bdev, ahci_fsexport); } void @@ -270,16 +286,41 @@ sata_create_fis(struct sata_reg_fis* cmd_fis, cmd_fis->count = sector_count; } +int +hba_bind_sbuf(struct hba_cmdh* cmdh, struct hba_cmdt* cmdt, struct membuf mbuf) +{ + assert_msg(mbuf.size <= 0x400000, "HBA: Buffer too big"); + cmdh->prdt_len = 1; + cmdt->entries[0] = (struct hba_prdte){ .data_base = vmm_v2p(mbuf.buffer), + .byte_count = mbuf.size - 1 }; +} + +int +hba_bind_vbuf(struct hba_cmdh* cmdh, struct hba_cmdt* cmdt, struct vecbuf* vbuf) +{ + size_t i = 0; + struct vecbuf* pos = vbuf; + + do { + assert_msg(i < HBA_MAX_PRDTE, "HBA: Too many PRDTEs"); + assert_msg(pos->buf.size <= 0x400000, "HBA: Buffer too big"); + + cmdt->entries[i++] = + (struct hba_prdte){ .data_base = vmm_v2p(pos->buf.buffer), + .byte_count = pos->buf.size - 1 }; + pos = list_entry(pos->components.next, struct vecbuf, components); + } while (pos != vbuf); + + cmdh->prdt_len = i + 1; +} + int hba_prepare_cmd(struct hba_port* port, struct hba_cmdt** cmdt, - struct hba_cmdh** cmdh, - void* buffer, - unsigned int size) + struct hba_cmdh** cmdh) { int slot = __get_free_slot(port); assert_msg(slot >= 0, "HBA: No free slot"); - assert_msg(size <= 0x400000, "HBA: buffer too big"); // 构建命令头（Command Header）和命令表（Command Table） struct hba_cmdh* cmd_header = &port->cmdlst[slot]; @@ -292,13 +333,6 @@ hba_prepare_cmd(struct hba_port* port, cmd_header->options = HBA_CMDH_FIS_LEN(sizeof(struct sata_reg_fis)) | HBA_CMDH_CLR_BUSY; - if (buffer) { - cmd_header->prdt_len = 1; - cmd_table->entries[0] = - (struct hba_prdte){ .data_base = vmm_v2p(buffer), - .byte_count = size - 1 }; - } - *cmdh = cmd_header; *cmdt = cmd_table; @@ -313,12 +347,14 @@ ahci_init_device(struct hba_port* port) struct hba_cmdh* cmd_header; // mask DHR interrupt - port->regs[HBA_RPxIE] &= ~HBA_PxINTR_DHR; + port->regs[HBA_RPxIE] &= ~HBA_MY_IE; // 预备DMA接收缓存，用于存放HBA传回的数据 uint16_t* data_in = (uint16_t*)valloc_dma(512); - int slot = hba_prepare_cmd(port, &cmd_table, &cmd_header, data_in, 512); + int slot = hba_prepare_cmd(port, &cmd_table, &cmd_header); + hba_bind_sbuf( + cmd_header, cmd_table, (struct membuf){ .buffer = data_in, .size = 512 }); port->device = vzalloc(sizeof(struct hba_device)); port->device->port = port; @@ -400,7 +436,7 @@ ahci_init_device(struct hba_port* port) done: // reset interrupt status and unmask D2HR interrupt - port->regs[HBA_RPxIE] |= HBA_PxINTR_DHR; + port->regs[HBA_RPxIE] |= HBA_MY_IE; achi_register_ops(port); vfree_dma(data_in); @@ -409,7 +445,7 @@ done: return 1; fail: - port->regs[HBA_RPxIE] |= HBA_PxINTR_DHR; + port->regs[HBA_RPxIE] |= HBA_MY_IE; vfree_dma(data_in); vfree_dma(cmd_table); @@ -429,10 +465,8 @@ achi_register_ops(struct hba_port* port) { port->device->ops.identify = ahci_identify_device; if (!(port->device->flags & HBA_DEV_FATAPI)) { - port->device->ops.read_buffer = sata_read_buffer; - port->device->ops.write_buffer = sata_write_buffer; + port->device->ops.submit = sata_submit; } else { - port->device->ops.read_buffer = scsi_read_buffer; - port->device->ops.write_buffer = scsi_write_buffer; + port->device->ops.submit = scsi_submit; } } \ No newline at end of file