374 lines
10 KiB
C
374 lines
10 KiB
C
/*
|
|
* ATA Disk Driver
|
|
*
|
|
* Provides raw block access to an (Parallel) ATA drive.
|
|
*/
|
|
|
|
#include <system.h>
|
|
#include <logging.h>
|
|
#include <module.h>
|
|
#include <fs.h>
|
|
#include <printf.h>
|
|
|
|
/* TODO: Move this to mod/ata.h */
|
|
#include <ata.h>
|
|
|
|
static char ata_drive_char = 'a';
|
|
|
|
struct ata_device {
|
|
int io_base;
|
|
int control;
|
|
int slave;
|
|
ata_identify_t identity;
|
|
};
|
|
|
|
/* TODO support other sector sizes */
|
|
#define ATA_SECTOR_SIZE 512
|
|
|
|
static void ata_device_read_sector(struct ata_device * dev, uint32_t lba, uint8_t * buf);
|
|
static void ata_device_write_sector_retry(struct ata_device * dev, uint32_t lba, uint8_t * buf);
|
|
static uint32_t read_ata(fs_node_t *node, uint32_t offset, uint32_t size, uint8_t *buffer);
|
|
static uint32_t write_ata(fs_node_t *node, uint32_t offset, uint32_t size, uint8_t *buffer);
|
|
static void open_ata(fs_node_t *node, unsigned int flags);
|
|
static void close_ata(fs_node_t *node);
|
|
|
|
static uint32_t read_ata(fs_node_t *node, uint32_t offset, uint32_t size, uint8_t *buffer) {
|
|
|
|
struct ata_device * dev = (struct ata_device *)node->device;
|
|
|
|
unsigned int start_block = offset / ATA_SECTOR_SIZE;
|
|
unsigned int end_block = (offset + size - 1) / ATA_SECTOR_SIZE;
|
|
|
|
unsigned int x_offset = 0;
|
|
|
|
if (offset > dev->identity.sectors_48 * ATA_SECTOR_SIZE) {
|
|
return 0;
|
|
}
|
|
|
|
if (offset + size > dev->identity.sectors_48 * ATA_SECTOR_SIZE) {
|
|
unsigned int i = dev->identity.sectors_48 * ATA_SECTOR_SIZE - offset;
|
|
size = i;
|
|
}
|
|
|
|
if (offset % ATA_SECTOR_SIZE) {
|
|
unsigned int prefix_size = (ATA_SECTOR_SIZE - (offset % ATA_SECTOR_SIZE));
|
|
char * tmp = malloc(ATA_SECTOR_SIZE);
|
|
ata_device_read_sector(dev, start_block, (uint8_t *)tmp);
|
|
|
|
memcpy(buffer, (void *)((uintptr_t)tmp + (offset % ATA_SECTOR_SIZE)), prefix_size);
|
|
|
|
free(tmp);
|
|
|
|
x_offset += prefix_size;
|
|
start_block++;
|
|
}
|
|
|
|
if ((offset + size) % ATA_SECTOR_SIZE && start_block < end_block) {
|
|
unsigned int postfix_size = (offset + size) % ATA_SECTOR_SIZE;
|
|
char * tmp = malloc(ATA_SECTOR_SIZE);
|
|
ata_device_read_sector(dev, end_block, (uint8_t *)tmp);
|
|
|
|
memcpy((void *)((uintptr_t)buffer + size - postfix_size), tmp, postfix_size);
|
|
|
|
free(tmp);
|
|
|
|
end_block--;
|
|
}
|
|
|
|
while (start_block <= end_block) {
|
|
ata_device_read_sector(dev, start_block, (uint8_t *)((uintptr_t)buffer + x_offset));
|
|
x_offset += ATA_SECTOR_SIZE;
|
|
start_block++;
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
static uint32_t write_ata(fs_node_t *node, uint32_t offset, uint32_t size, uint8_t *buffer) {
|
|
struct ata_device * dev = (struct ata_device *)node->device;
|
|
|
|
unsigned int start_block = offset / ATA_SECTOR_SIZE;
|
|
unsigned int end_block = (offset + size - 1) / ATA_SECTOR_SIZE;
|
|
|
|
unsigned int x_offset = 0;
|
|
|
|
if (offset > dev->identity.sectors_48 * ATA_SECTOR_SIZE) {
|
|
return 0;
|
|
}
|
|
|
|
if (offset + size > dev->identity.sectors_48 * ATA_SECTOR_SIZE) {
|
|
unsigned int i = dev->identity.sectors_48 * ATA_SECTOR_SIZE - offset;
|
|
size = i;
|
|
}
|
|
|
|
if (offset % ATA_SECTOR_SIZE) {
|
|
unsigned int prefix_size = (ATA_SECTOR_SIZE - (offset % ATA_SECTOR_SIZE));
|
|
|
|
char * tmp = malloc(ATA_SECTOR_SIZE);
|
|
ata_device_read_sector(dev, start_block, (uint8_t *)tmp);
|
|
|
|
debug_print(NOTICE, "Writing first block");
|
|
|
|
memcpy((void *)((uintptr_t)tmp + (offset % ATA_SECTOR_SIZE)), buffer, prefix_size);
|
|
ata_device_write_sector_retry(dev, start_block, (uint8_t *)tmp);
|
|
|
|
free(tmp);
|
|
x_offset += prefix_size;
|
|
start_block++;
|
|
}
|
|
|
|
if ((offset + size) % ATA_SECTOR_SIZE && start_block < end_block) {
|
|
unsigned int postfix_size = (offset + size) % ATA_SECTOR_SIZE;
|
|
|
|
char * tmp = malloc(ATA_SECTOR_SIZE);
|
|
ata_device_read_sector(dev, end_block, (uint8_t *)tmp);
|
|
|
|
debug_print(NOTICE, "Writing last block");
|
|
|
|
memcpy(tmp, (void *)((uintptr_t)buffer + size - postfix_size), postfix_size);
|
|
|
|
ata_device_write_sector_retry(dev, end_block, (uint8_t *)tmp);
|
|
|
|
free(tmp);
|
|
end_block--;
|
|
}
|
|
|
|
while (start_block <= end_block) {
|
|
ata_device_write_sector_retry(dev, start_block, (uint8_t *)((uintptr_t)buffer + x_offset));
|
|
x_offset += ATA_SECTOR_SIZE;
|
|
start_block++;
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
static void open_ata(fs_node_t * node, unsigned int flags) {
|
|
return;
|
|
}
|
|
|
|
static void close_ata(fs_node_t * node) {
|
|
return;
|
|
}
|
|
|
|
static fs_node_t * ata_device_create(struct ata_device * device) {
|
|
fs_node_t * fnode = malloc(sizeof(fs_node_t));
|
|
memset(fnode, 0x00, sizeof(fs_node_t));
|
|
fnode->inode = 0;
|
|
sprintf(fnode->name, "atadev%d", ata_drive_char - 'a');
|
|
fnode->device = device;
|
|
fnode->uid = 0;
|
|
fnode->gid = 0;
|
|
fnode->length = device->identity.sectors_48 * ATA_SECTOR_SIZE; /* TODO */
|
|
fnode->flags = FS_BLOCKDEVICE;
|
|
fnode->read = read_ata;
|
|
fnode->write = write_ata;
|
|
fnode->open = open_ata;
|
|
fnode->close = close_ata;
|
|
fnode->readdir = NULL;
|
|
fnode->finddir = NULL;
|
|
fnode->ioctl = NULL; /* TODO, identify, etc? */
|
|
return fnode;
|
|
}
|
|
|
|
static void ata_io_wait(struct ata_device * dev) {
|
|
inportb(dev->io_base + ATA_REG_ALTSTATUS);
|
|
inportb(dev->io_base + ATA_REG_ALTSTATUS);
|
|
inportb(dev->io_base + ATA_REG_ALTSTATUS);
|
|
inportb(dev->io_base + ATA_REG_ALTSTATUS);
|
|
}
|
|
|
|
|
|
static int ata_wait(struct ata_device * dev, int advanced) {
|
|
uint8_t status = 0;
|
|
|
|
ata_io_wait(dev);
|
|
|
|
while ((status = inportb(dev->io_base + ATA_REG_STATUS)) & ATA_SR_BSY);
|
|
|
|
if (advanced) {
|
|
status = inportb(dev->io_base + ATA_REG_STATUS);
|
|
if (status & ATA_SR_ERR) return 1;
|
|
if (status & ATA_SR_DF) return 1;
|
|
if (!(status & ATA_SR_DRQ)) return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void ata_soft_reset(struct ata_device * dev) {
|
|
outportb(dev->control, 0x04);
|
|
outportb(dev->control, 0x00);
|
|
}
|
|
|
|
static void ata_device_init(struct ata_device * dev) {
|
|
|
|
debug_print(NOTICE, "Initializing IDE device on bus %d", dev->io_base);
|
|
|
|
outportb(dev->io_base + 1, 1);
|
|
outportb(dev->control, 0);
|
|
|
|
outportb(dev->io_base + ATA_REG_HDDEVSEL, 0xA0 | dev->slave << 4);
|
|
ata_io_wait(dev);
|
|
|
|
outportb(dev->io_base + ATA_REG_COMMAND, ATA_CMD_IDENTIFY);
|
|
ata_io_wait(dev);
|
|
|
|
int status = inportb(dev->io_base + ATA_REG_COMMAND);
|
|
debug_print(INFO, "Device status: %d", status);
|
|
|
|
ata_wait(dev, 0);
|
|
|
|
uint16_t * buf = (uint16_t *)&dev->identity;
|
|
|
|
for (int i = 0; i < 256; ++i) {
|
|
buf[i] = inports(dev->io_base);
|
|
}
|
|
|
|
uint8_t * ptr = (uint8_t *)&dev->identity.model;
|
|
for (int i = 0; i < 39; i+=2) {
|
|
uint8_t tmp = ptr[i+1];
|
|
ptr[i+1] = ptr[i];
|
|
ptr[i] = tmp;
|
|
}
|
|
|
|
debug_print(NOTICE, "Device Name: %s", dev->identity.model);
|
|
debug_print(NOTICE, "Sectors (48): %d", (uint32_t)dev->identity.sectors_48);
|
|
debug_print(NOTICE, "Sectors (24): %d", dev->identity.sectors_28);
|
|
|
|
outportb(dev->io_base + ATA_REG_CONTROL, 0x02);
|
|
}
|
|
|
|
static int ata_device_detect(struct ata_device * dev) {
|
|
ata_soft_reset(dev);
|
|
outportb(dev->io_base + ATA_REG_HDDEVSEL, 0xA0 | dev->slave << 4);
|
|
ata_io_wait(dev);
|
|
|
|
unsigned char cl = inportb(dev->io_base + ATA_REG_LBA1); /* CYL_LO */
|
|
unsigned char ch = inportb(dev->io_base + ATA_REG_LBA2); /* CYL_HI */
|
|
|
|
debug_print(NOTICE, "Device detected: 0x%2x 0x%2x", cl, ch);
|
|
if (cl == 0xFF && ch == 0xFF) {
|
|
/* Nothing here */
|
|
return 0;
|
|
}
|
|
if (cl == 0x00 && ch == 0x00) {
|
|
/* Parallel ATA device */
|
|
|
|
char devname[64];
|
|
sprintf((char *)&devname, "/dev/hd%c", ata_drive_char);
|
|
fs_node_t * node = ata_device_create(dev);
|
|
vfs_mount(devname, node);
|
|
ata_drive_char++;
|
|
|
|
ata_device_init(dev);
|
|
|
|
return 1;
|
|
}
|
|
|
|
/* TODO: ATAPI, SATA, SATAPI */
|
|
return 0;
|
|
}
|
|
|
|
static void ata_device_read_sector(struct ata_device * dev, uint32_t lba, uint8_t * buf) {
|
|
uint16_t bus = dev->io_base;
|
|
uint8_t slave = dev->slave;
|
|
|
|
int errors = 0;
|
|
try_again:
|
|
outportb(bus + ATA_REG_CONTROL, 0x02);
|
|
|
|
ata_wait(dev, 0);
|
|
|
|
outportb(bus + ATA_REG_HDDEVSEL, 0xe0 | slave << 4 | (lba & 0x0f000000) >> 24);
|
|
outportb(bus + ATA_REG_FEATURES, 0x00);
|
|
outportb(bus + ATA_REG_SECCOUNT0, 1);
|
|
outportb(bus + ATA_REG_LBA0, (lba & 0x000000ff) >> 0);
|
|
outportb(bus + ATA_REG_LBA1, (lba & 0x0000ff00) >> 8);
|
|
outportb(bus + ATA_REG_LBA2, (lba & 0x00ff0000) >> 16);
|
|
outportb(bus + ATA_REG_COMMAND, ATA_CMD_READ_PIO);
|
|
|
|
if (ata_wait(dev, 1)) {
|
|
debug_print(WARNING, "Error during ATA read of lba block %d", lba);
|
|
errors++;
|
|
if (errors > 4) {
|
|
debug_print(WARNING, "-- Too many errors trying to read this block. Bailing.");
|
|
return;
|
|
}
|
|
goto try_again;
|
|
}
|
|
|
|
int size = 256;
|
|
inportsm(bus,buf,size);
|
|
ata_wait(dev, 0);
|
|
}
|
|
|
|
static void ata_device_write_sector(struct ata_device * dev, uint32_t lba, uint8_t * buf) {
|
|
uint16_t bus = dev->io_base;
|
|
uint8_t slave = dev->slave;
|
|
|
|
outportb(bus + ATA_REG_CONTROL, 0x02);
|
|
|
|
ata_wait(dev, 0);
|
|
outportb(bus + ATA_REG_HDDEVSEL, 0xe0 | slave << 4 | (lba & 0x0f000000) >> 24);
|
|
ata_wait(dev, 0);
|
|
|
|
outportb(bus + ATA_REG_FEATURES, 0x00);
|
|
outportb(bus + ATA_REG_SECCOUNT0, 0x01);
|
|
outportb(bus + ATA_REG_LBA0, (lba & 0x000000ff) >> 0);
|
|
outportb(bus + ATA_REG_LBA1, (lba & 0x0000ff00) >> 8);
|
|
outportb(bus + ATA_REG_LBA2, (lba & 0x00ff0000) >> 16);
|
|
outportb(bus + ATA_REG_COMMAND, ATA_CMD_WRITE_PIO);
|
|
ata_wait(dev, 0);
|
|
int size = ATA_SECTOR_SIZE / 2;
|
|
outportsm(bus,buf,size);
|
|
outportb(bus + 0x07, ATA_CMD_CACHE_FLUSH);
|
|
ata_wait(dev, 0);
|
|
}
|
|
|
|
static int buffer_compare(uint32_t * ptr1, uint32_t * ptr2, size_t size) {
|
|
assert(!(size % 4));
|
|
size_t i = 0;
|
|
while (i < size) {
|
|
if (*ptr1 != *ptr2) return 1;
|
|
ptr1++;
|
|
ptr2++;
|
|
i += sizeof(uint32_t);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static void ata_device_write_sector_retry(struct ata_device * dev, uint32_t lba, uint8_t * buf) {
|
|
uint8_t * read_buf = malloc(ATA_SECTOR_SIZE);
|
|
IRQ_OFF;
|
|
do {
|
|
ata_device_write_sector(dev, lba, buf);
|
|
ata_device_read_sector(dev, lba, read_buf);
|
|
} while (buffer_compare((uint32_t *)buf, (uint32_t *)read_buf, ATA_SECTOR_SIZE));
|
|
IRQ_RES;
|
|
free(read_buf);
|
|
}
|
|
|
|
static struct ata_device ata_primary_master = {.io_base = 0x1F0, .control = 0x3F6, .slave = 0};
|
|
static struct ata_device ata_primary_slave = {.io_base = 0x1F0, .control = 0x3F6, .slave = 1};
|
|
static struct ata_device ata_secondary_master = {.io_base = 0x170, .control = 0x376, .slave = 0};
|
|
static struct ata_device ata_secondary_slave = {.io_base = 0x170, .control = 0x376, .slave = 1};
|
|
|
|
|
|
static int ata_initialize(void) {
|
|
/* Detect drives and mount them */
|
|
|
|
ata_device_detect(&ata_primary_master);
|
|
ata_device_detect(&ata_primary_slave);
|
|
ata_device_detect(&ata_secondary_master);
|
|
ata_device_detect(&ata_secondary_slave);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int ata_finalize(void) {
|
|
|
|
return 0;
|
|
}
|
|
|
|
MODULE_DEF(ata, ata_initialize, ata_finalize);
|