ableos/sysdata/programs/diskio_driver/src/main.hb

.{memory, buffer, log} := @use("../../../libraries/stn/src/lib.hb")

// Influenced by https://github.com/levex/osdev/blob/master/drivers/ata.c

main := fn(): int {
	// shuts down ableOS
	// memory.outb(0xF400, 0)

	a := memory.inb(0x4600)
	b := memory.inb(0x4700)

	// c := buffer.search("XNumber\0")
	ata := ATA.new()
	return 0
}

ATA_PRIMARY := @as(u8, @intcast(0x0))
ATA_PRIMARY_IO := @as(u8, @intcast(0x1F0))

ATA_SECONDARY := @as(u8, @intcast(0x1))
ATA_SECONDARY_IO := @as(u8, @intcast(0x170))

ATA_REG_HDDEVSEL := @as(u8, @intcast(0x6))

ATA_MASTER := @as(u8, @intcast(0x0))
IDE := struct {}

ide_select_drive := fn(bus: u8, i: u8): void {
	if bus == ATA_PRIMARY {
		if i == ATA_MASTER {
			log.info("Primary ATA selected\0")
			memory.outb(ATA_PRIMARY_IO + ATA_REG_HDDEVSEL, 0xA0)
		} else {
			memory.outb(ATA_PRIMARY_IO + ATA_REG_HDDEVSEL, 0xB0)
		}
	} else {
		if i == ATA_MASTER {
			memory.outb(ATA_SECONDARY_IO + ATA_REG_HDDEVSEL, 0xA0)
		} else {
			memory.outb(ATA_SECONDARY_IO + ATA_REG_HDDEVSEL, 0xB0)
		}
	}
}

identify := fn(bus: u8, drive: u8): bool {
	io := 0

	// BUG: Causes qemu to crash
	ide_select_drive(bus, drive)
	if bus == ATA_PRIMARY {
		io = ATA_PRIMARY_IO
	}

	return true
}

ata_probe := fn(): void {
	if identify(ATA_PRIMARY, ATA_MASTER) {
	}
}

ATA := struct {
	new := fn(): void {
		log.info("Checking for ATA Drives.\0")
		// ata_probe()
	}
}