/* vim: noexpandtab tabstop=4 shiftwidth=4
 *
 * ToAruOS PCI Initialization
 */

#include <system.h>
#include <ata.h>

ide_channel_regs_t ide_channels[2];
ide_device_t ide_devices[4];
uint8_t ide_buf[2048] = {0};
uint8_t ide_irq_invoked = 0;
uint8_t atapi_packet[12] = {0xA8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};

void ide_detect() {
	
}

void ide_read_sector(uint16_t bus, uint8_t slave, uint32_t lba, char * buf) {
	__asm__ __volatile__ ("cli");
	outportb(bus + ATA_REG_FEATURES, 0x00);
	outportb(bus + ATA_REG_SECCOUNT0, 1);
	outportb(bus + ATA_REG_HDDEVSEL,  0xe0 | slave << 4 | 
								 (lba & 0x0f000000) >> 24);
	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);
	uint8_t status = 0;
	while ((status = inportb(bus + 0x07)) & 0x80);
	for (volatile int i = 0; i < 512; i += 2) {
		uint16_t s = inports(0x1f0);
		buf[i] = s & 0xFF;
		buf[i+1] = (s & 0xFF00) >> 8;
	}
	outportb(0x177, 0xe7);
	__asm__ __volatile__ ("sti");
}

void ide_write_sector(uint16_t bus, uint8_t slave, uint32_t lba, char * buf) {
	__asm__ __volatile__ ("cli");
	outportb(bus + ATA_REG_FEATURES, 0x00);
	outportb(bus + ATA_REG_SECCOUNT0, 1);
	outportb(bus + ATA_REG_HDDEVSEL,  0xe0 | slave << 4 | 
								 (lba & 0x0f000000) >> 24);
	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);
	uint8_t status = 0;
	while ((status = inportb(bus + 0x07)) & 0x80);
	for (volatile int i = 0; i < 512; i+=2) {
		uint16_t s = (buf[i+1] << 8) + (buf[i]);
		outports(bus, s);
	}
	outportb(bus + 0x07, ATA_CMD_CACHE_FLUSH);
	outportb(0x177, 0xe7);
	__asm__ __volatile__ ("sti");
}