Update coding style of ATA driver.
This commit is contained in:
Jonas 'Sortie' Termansen
parent
c9b3002e43
commit
4f60d028ef
613
sortix/ata.cpp
613
sortix/ata.cpp
|
|
@ -33,333 +33,334 @@
|
||||||
// TODO: Use the PCI to detect ATA devices instead of relying on them being on
|
// TODO: Use the PCI to detect ATA devices instead of relying on them being on
|
||||||
// standard locations.
|
// standard locations.
|
||||||
|
|
||||||
namespace Sortix
|
namespace Sortix {
|
||||||
|
|
||||||
|
const uint16_t PRIMARY_BUS_OFFSET = 0x1F0;
|
||||||
|
const uint16_t SECONDARY_BUS_OFFSET = 0x170;
|
||||||
|
const uint16_t DATA = 0x0;
|
||||||
|
const uint16_t FEATURE = 0x1;
|
||||||
|
const uint16_t ERROR = 0x1;
|
||||||
|
const uint16_t SECTOR_COUNT = 0x2;
|
||||||
|
const uint16_t LBA_LOW = 0x3;
|
||||||
|
const uint16_t LBA_MID = 0x4;
|
||||||
|
const uint16_t LBA_HIGH = 0x5;
|
||||||
|
const uint16_t DRIVE_SELECT = 0x6;
|
||||||
|
const uint16_t COMMAND = 0x7;
|
||||||
|
const uint16_t STATUS = 0x7;
|
||||||
|
const uint8_t CMD_READ = 0x20;
|
||||||
|
const uint8_t CMD_READ_EXT = 0x24;
|
||||||
|
const uint8_t CMD_WRITE = 0x30;
|
||||||
|
const uint8_t CMD_WRITE_EXT = 0x34;
|
||||||
|
const uint8_t CMD_FLUSH_CACHE = 0xE7;
|
||||||
|
const uint8_t CMD_IDENTIFY = 0xEC;
|
||||||
|
const uint8_t STATUS_ERROR = (1<<0);
|
||||||
|
const uint8_t STATUS_DATAREADY = (1<<3);
|
||||||
|
const uint8_t STATUS_DRIVEFAULT = (1<<5);
|
||||||
|
const uint8_t STATUS_BUSY = (1<<7);
|
||||||
|
const uint8_t CTL_NO_INTERRUPT = (1<<1);
|
||||||
|
const uint8_t CTL_RESET = (1<<2);
|
||||||
|
|
||||||
|
namespace ATA {
|
||||||
|
|
||||||
|
void DetectDrive(unsigned busid, ATABus* bus, unsigned driveid)
|
||||||
{
|
{
|
||||||
const uint16_t PRIMARY_BUS_OFFSET = 0x1F0;
|
unsigned ataid = busid*2 + driveid;
|
||||||
const uint16_t SECONDARY_BUS_OFFSET = 0x170;
|
ATADrive* drive = bus->Instatiate(driveid);
|
||||||
const uint16_t DATA = 0x0;
|
if ( !drive ) { return; }
|
||||||
const uint16_t FEATURE = 0x1;
|
DeviceFS::RegisterATADrive(ataid, drive);
|
||||||
const uint16_t ERROR = 0x1;
|
}
|
||||||
const uint16_t SECTOR_COUNT = 0x2;
|
|
||||||
const uint16_t LBA_LOW = 0x3;
|
|
||||||
const uint16_t LBA_MID = 0x4;
|
|
||||||
const uint16_t LBA_HIGH = 0x5;
|
|
||||||
const uint16_t DRIVE_SELECT = 0x6;
|
|
||||||
const uint16_t COMMAND = 0x7;
|
|
||||||
const uint16_t STATUS = 0x7;
|
|
||||||
const uint8_t CMD_READ = 0x20;
|
|
||||||
const uint8_t CMD_READ_EXT = 0x24;
|
|
||||||
const uint8_t CMD_WRITE = 0x30;
|
|
||||||
const uint8_t CMD_WRITE_EXT = 0x34;
|
|
||||||
const uint8_t CMD_FLUSH_CACHE = 0xE7;
|
|
||||||
const uint8_t CMD_IDENTIFY = 0xEC;
|
|
||||||
const uint8_t STATUS_ERROR = (1<<0);
|
|
||||||
const uint8_t STATUS_DATAREADY = (1<<3);
|
|
||||||
const uint8_t STATUS_DRIVEFAULT = (1<<5);
|
|
||||||
const uint8_t STATUS_BUSY = (1<<7);
|
|
||||||
const uint8_t CTL_NO_INTERRUPT = (1<<1);
|
|
||||||
const uint8_t CTL_RESET = (1<<2);
|
|
||||||
|
|
||||||
namespace ATA
|
void DetectBus(unsigned busid, uint16_t ioport, uint16_t altio)
|
||||||
|
{
|
||||||
|
ATABus* bus = ATA::CreateBus(ioport, altio);
|
||||||
|
if ( !bus )
|
||||||
|
return;
|
||||||
|
DetectDrive(busid, bus, 0);
|
||||||
|
DetectDrive(busid, bus, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Init()
|
||||||
|
{
|
||||||
|
DetectBus(0, 0x1F0, 0x3F6);
|
||||||
|
DetectBus(1, 0x170, 0x366);
|
||||||
|
}
|
||||||
|
|
||||||
|
ATABus* CreateBus(uint16_t portoffset, uint16_t altport)
|
||||||
|
{
|
||||||
|
unsigned status = CPU::InPortB(portoffset + STATUS);
|
||||||
|
// Detect if there is no such bus.
|
||||||
|
if ( status == 0xFF )
|
||||||
{
|
{
|
||||||
void DetectDrive(unsigned busid, ATABus* bus, unsigned driveid)
|
errno = ENODEV;
|
||||||
{
|
return NULL;
|
||||||
unsigned ataid = busid*2 + driveid;
|
|
||||||
ATADrive* drive = bus->Instatiate(driveid);
|
|
||||||
if ( !drive ) { return; }
|
|
||||||
DeviceFS::RegisterATADrive(ataid, drive);
|
|
||||||
}
|
|
||||||
|
|
||||||
void DetectBus(unsigned busid, uint16_t ioport, uint16_t altio)
|
|
||||||
{
|
|
||||||
ATABus* bus = ATA::CreateBus(ioport, altio);
|
|
||||||
if ( !bus )
|
|
||||||
return;
|
|
||||||
DetectDrive(busid, bus, 0);
|
|
||||||
DetectDrive(busid, bus, 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
void Init()
|
|
||||||
{
|
|
||||||
DetectBus(0, 0x1F0, 0x3F6);
|
|
||||||
DetectBus(1, 0x170, 0x366);
|
|
||||||
}
|
|
||||||
|
|
||||||
ATABus* CreateBus(uint16_t portoffset, uint16_t altport)
|
|
||||||
{
|
|
||||||
unsigned status = CPU::InPortB(portoffset + STATUS);
|
|
||||||
// Detect if there is no such bus.
|
|
||||||
if ( status == 0xFF )
|
|
||||||
{
|
|
||||||
errno = ENODEV;
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
return new ATABus(portoffset, altport);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
return new ATABus(portoffset, altport);
|
||||||
|
}
|
||||||
|
|
||||||
void Wait400NSecs(uint16_t iobase)
|
} // namespace ATA
|
||||||
|
|
||||||
|
void Wait400NSecs(uint16_t iobase)
|
||||||
|
{
|
||||||
|
// Now wait 400 ns for the drive to be ready.
|
||||||
|
for ( unsigned i = 0; i < 4; i++ ) { CPU::InPortB(iobase + STATUS); }
|
||||||
|
}
|
||||||
|
|
||||||
|
ATABus::ATABus(uint16_t portoffset, uint16_t altport)
|
||||||
|
{
|
||||||
|
this->iobase = portoffset;
|
||||||
|
this->altport = altport;
|
||||||
|
this->curdriveid = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
ATABus::~ATABus()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
ATADrive* ATABus::Instatiate(unsigned driveid)
|
||||||
|
{
|
||||||
|
if ( 1 < driveid ) { errno = EINVAL; return false; }
|
||||||
|
curdriveid = 0;
|
||||||
|
|
||||||
|
uint8_t drivemagic = 0xA0 | (driveid << 4);
|
||||||
|
CPU::OutPortB(iobase + DRIVE_SELECT, drivemagic);
|
||||||
|
CPU::OutPortB(iobase + SECTOR_COUNT, 0);
|
||||||
|
CPU::OutPortB(iobase + LBA_LOW, 0);
|
||||||
|
CPU::OutPortB(iobase + LBA_MID, 0);
|
||||||
|
CPU::OutPortB(iobase + LBA_HIGH, 0);
|
||||||
|
CPU::OutPortB(iobase + COMMAND, CMD_IDENTIFY);
|
||||||
|
uint8_t status;
|
||||||
|
while ( true )
|
||||||
{
|
{
|
||||||
// Now wait 400 ns for the drive to be ready.
|
status = CPU::InPortB(iobase + STATUS);
|
||||||
for ( unsigned i = 0; i < 4; i++ ) { CPU::InPortB(iobase + STATUS); }
|
if ( !status || status == 0xFF ) { errno = ENODEV; return false; }
|
||||||
|
if ( !(status & STATUS_BUSY) ) { break; }
|
||||||
}
|
}
|
||||||
|
if ( CPU::InPortB(iobase + LBA_MID) || CPU::InPortB(iobase + LBA_MID) )
|
||||||
ATABus::ATABus(uint16_t portoffset, uint16_t altport)
|
|
||||||
{
|
{
|
||||||
this->iobase = portoffset;
|
errno = ENODEV; return false; // ATAPI device not following spec.
|
||||||
this->altport = altport;
|
|
||||||
this->curdriveid = 0;
|
|
||||||
}
|
}
|
||||||
|
while ( !(status & STATUS_DATAREADY) && !(status & STATUS_ERROR) )
|
||||||
ATABus::~ATABus()
|
|
||||||
{
|
{
|
||||||
|
status = CPU::InPortB(iobase + STATUS);
|
||||||
}
|
}
|
||||||
|
if ( status & STATUS_ERROR )
|
||||||
ATADrive* ATABus::Instatiate(unsigned driveid)
|
|
||||||
{
|
{
|
||||||
if ( 1 < driveid ) { errno = EINVAL; return false; }
|
unsigned mid = CPU::InPortB(iobase + LBA_MID);
|
||||||
curdriveid = 0;
|
unsigned high = CPU::InPortB(iobase + LBA_HIGH);
|
||||||
|
if ( mid == 0x14 && high == 0xEB )
|
||||||
uint8_t drivemagic = 0xA0 | (driveid << 4);
|
|
||||||
CPU::OutPortB(iobase + DRIVE_SELECT, drivemagic);
|
|
||||||
CPU::OutPortB(iobase + SECTOR_COUNT, 0);
|
|
||||||
CPU::OutPortB(iobase + LBA_LOW, 0);
|
|
||||||
CPU::OutPortB(iobase + LBA_MID, 0);
|
|
||||||
CPU::OutPortB(iobase + LBA_HIGH, 0);
|
|
||||||
CPU::OutPortB(iobase + COMMAND, CMD_IDENTIFY);
|
|
||||||
uint8_t status;
|
|
||||||
while ( true )
|
|
||||||
{
|
{
|
||||||
status = CPU::InPortB(iobase + STATUS);
|
//Log::PrintF("Found ATAPI device instead of ATA\n");
|
||||||
if ( !status || status == 0xFF ) { errno = ENODEV; return false; }
|
|
||||||
if ( !(status & STATUS_BUSY) ) { break; }
|
|
||||||
}
|
}
|
||||||
if ( CPU::InPortB(iobase + LBA_MID) || CPU::InPortB(iobase + LBA_MID) )
|
else if ( mid == 0x3C && high == 0xC3 )
|
||||||
{
|
{
|
||||||
errno = ENODEV; return false; // ATAPI device not following spec.
|
//Log::PrintF("Found SATA device instead of ATA\n");
|
||||||
}
|
}
|
||||||
while ( !(status & STATUS_DATAREADY) && !(status & STATUS_ERROR) )
|
else if ( mid || high )
|
||||||
{
|
{
|
||||||
status = CPU::InPortB(iobase + STATUS);
|
//Log::PrintF("Found unknown device instead of ATA\n");
|
||||||
}
|
|
||||||
if ( status & STATUS_ERROR )
|
|
||||||
{
|
|
||||||
unsigned mid = CPU::InPortB(iobase + LBA_MID);
|
|
||||||
unsigned high = CPU::InPortB(iobase + LBA_HIGH);
|
|
||||||
if ( mid == 0x14 && high == 0xEB )
|
|
||||||
{
|
|
||||||
//Log::PrintF("Found ATAPI device instead of ATA\n");
|
|
||||||
}
|
|
||||||
else if ( mid == 0x3C && high == 0xC3 )
|
|
||||||
{
|
|
||||||
//Log::PrintF("Found SATA device instead of ATA\n");
|
|
||||||
}
|
|
||||||
else if ( mid || high )
|
|
||||||
{
|
|
||||||
//Log::PrintF("Found unknown device instead of ATA\n");
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
//Log::PrintF("Error status during identify\n");
|
|
||||||
}
|
|
||||||
errno = EIO;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
ATADrive* drive = new ATADrive(this, driveid, iobase, altport);
|
|
||||||
return drive;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool ATABus::SelectDrive(unsigned driveid)
|
|
||||||
{
|
|
||||||
if ( driveid == curdriveid ) { return true; }
|
|
||||||
if ( 1 < driveid ) { errno = EINVAL; return false; }
|
|
||||||
|
|
||||||
uint8_t drivemagic = 0xA0 | (driveid << 4);
|
|
||||||
CPU::OutPortB(iobase + DRIVE_SELECT, drivemagic);
|
|
||||||
Wait400NSecs(iobase);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
const size_t META_LBA28 = 60;
|
|
||||||
const size_t META_FLAGS = 83;
|
|
||||||
const size_t META_LBA48 = 100;
|
|
||||||
const uint16_t FLAG_LBA48 = (1<<10);
|
|
||||||
|
|
||||||
ATADrive::ATADrive(ATABus* bus, unsigned driveid, uint16_t portoffset, uint16_t altport)
|
|
||||||
{
|
|
||||||
this->atalock = KTHREAD_MUTEX_INITIALIZER;
|
|
||||||
this->bus = bus;
|
|
||||||
this->driveid = driveid;
|
|
||||||
this->iobase = portoffset;
|
|
||||||
this->altport = altport;
|
|
||||||
for ( size_t i = 0; i < 256; i++ )
|
|
||||||
{
|
|
||||||
meta[i] = CPU::InPortW(iobase + DATA);
|
|
||||||
}
|
|
||||||
lba48 = meta[META_FLAGS] & FLAG_LBA48;
|
|
||||||
if ( lba48 )
|
|
||||||
{
|
|
||||||
numsectors = *((uint64_t*) (meta + META_LBA48));
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
numsectors = *((uint32_t*) (meta + META_LBA28));
|
//Log::PrintF("Error status during identify\n");
|
||||||
}
|
}
|
||||||
sectorsize = 512; // TODO: Detect this!
|
errno = EIO;
|
||||||
Initialize();
|
return false;
|
||||||
}
|
|
||||||
|
|
||||||
ATADrive::~ATADrive()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
off_t ATADrive::GetSectorSize()
|
|
||||||
{
|
|
||||||
return sectorsize;
|
|
||||||
}
|
|
||||||
|
|
||||||
off_t ATADrive::GetNumSectors()
|
|
||||||
{
|
|
||||||
return numsectors;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool ATADrive::PrepareIO(bool write, off_t sector)
|
|
||||||
{
|
|
||||||
if ( numsectors <= sector ) { errno = EINVAL; return false; }
|
|
||||||
if ( write && !ENABLE_DISKWRITE )
|
|
||||||
{
|
|
||||||
errno = EPERM;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
bus->SelectDrive(driveid);
|
|
||||||
uint8_t mode = (lba48) ? 0x40 : 0xE0;
|
|
||||||
mode |= driveid << 4;
|
|
||||||
mode |= (lba48) ? 0 : (sector >> 24) & 0x0F;
|
|
||||||
CPU::OutPortB(iobase + DRIVE_SELECT, mode);
|
|
||||||
uint16_t sectorcount = 1;
|
|
||||||
uint8_t sectorcountlow = sectorcount & 0xFF;
|
|
||||||
uint8_t sectorcounthigh = (sectorcount >> 8) & 0xFF;
|
|
||||||
if ( lba48 )
|
|
||||||
{
|
|
||||||
CPU::OutPortB(iobase + SECTOR_COUNT, sectorcounthigh);
|
|
||||||
CPU::OutPortB(iobase + LBA_LOW, (sector >> 24) & 0xFF);
|
|
||||||
CPU::OutPortB(iobase + LBA_MID, (sector >> 32) & 0xFF);
|
|
||||||
CPU::OutPortB(iobase + LBA_HIGH, (sector >> 40) & 0xFF);
|
|
||||||
}
|
|
||||||
CPU::OutPortB(iobase + SECTOR_COUNT, sectorcountlow);
|
|
||||||
CPU::OutPortB(iobase + LBA_LOW, sector & 0xFF);
|
|
||||||
CPU::OutPortB(iobase + LBA_MID, (sector >> 8) & 0xFF);
|
|
||||||
CPU::OutPortB(iobase + LBA_HIGH, (sector >> 16) & 0xFF);
|
|
||||||
uint8_t command = (write) ? CMD_WRITE : CMD_READ;
|
|
||||||
if ( lba48 ) { command = (write) ? CMD_WRITE_EXT : CMD_READ_EXT; }
|
|
||||||
CPU::OutPortB(iobase + COMMAND, command);
|
|
||||||
while ( true )
|
|
||||||
{
|
|
||||||
uint8_t status = CPU::InPortB(iobase + STATUS);
|
|
||||||
if ( status & STATUS_BUSY ) { continue; }
|
|
||||||
if ( status & STATUS_DATAREADY ) { break; }
|
|
||||||
if ( status & STATUS_ERROR ) { errno = EIO; return false; }
|
|
||||||
if ( status & STATUS_DRIVEFAULT ) { errno = EIO; return false; }
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool ATADrive::ReadSector(off_t sector, uint8_t* dest)
|
|
||||||
{
|
|
||||||
ScopedLock lock(&atalock);
|
|
||||||
if ( !PrepareIO(false, sector) ) { return false; }
|
|
||||||
uint16_t* destword = (uint16_t*) dest;
|
|
||||||
for ( size_t i = 0; i < sectorsize/2; i++ )
|
|
||||||
{
|
|
||||||
destword[i] = CPU::InPortW(iobase + DATA);
|
|
||||||
}
|
|
||||||
Wait400NSecs(iobase);
|
|
||||||
uint8_t status = CPU::InPortB(iobase + STATUS);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool ATADrive::WriteSector(off_t sector, const uint8_t* src)
|
|
||||||
{
|
|
||||||
ScopedLock lock(&atalock);
|
|
||||||
if ( !PrepareIO(true, sector) ) { return false; }
|
|
||||||
const uint16_t* srcword = (const uint16_t*) src;
|
|
||||||
for ( size_t i = 0; i < sectorsize/2; i++ )
|
|
||||||
{
|
|
||||||
CPU::OutPortW(iobase + DATA, srcword[i]);
|
|
||||||
}
|
|
||||||
Wait400NSecs(iobase);
|
|
||||||
CPU::OutPortB(iobase + COMMAND, CMD_FLUSH_CACHE);
|
|
||||||
while ( true )
|
|
||||||
{
|
|
||||||
uint8_t status = CPU::InPortB(iobase + STATUS);
|
|
||||||
if ( status & STATUS_ERROR ) { errno = EIO; return false; }
|
|
||||||
if ( status & STATUS_DRIVEFAULT ) { errno = EIO; return false; }
|
|
||||||
if ( !(status & STATUS_BUSY) ) { break; }
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
size_t ATADrive::Read(off_t byteoffset, uint8_t* dest, size_t numbytes)
|
|
||||||
{
|
|
||||||
size_t sofar = 0;
|
|
||||||
size_t leadingbytes = byteoffset % sectorsize;
|
|
||||||
if ( leadingbytes || numbytes < sectorsize )
|
|
||||||
{
|
|
||||||
size_t wanted = sectorsize - leadingbytes;
|
|
||||||
if ( numbytes < wanted ) { wanted = numbytes; }
|
|
||||||
uint8_t temp[512 /*sectorsize*/];
|
|
||||||
if ( !ReadSector(byteoffset/sectorsize, temp) ) { return sofar; }
|
|
||||||
memcpy(dest + sofar, temp + leadingbytes, wanted);
|
|
||||||
sofar += wanted;
|
|
||||||
numbytes -= wanted;
|
|
||||||
byteoffset += wanted;
|
|
||||||
}
|
|
||||||
|
|
||||||
while ( sectorsize <= numbytes )
|
|
||||||
{
|
|
||||||
if ( !ReadSector(byteoffset/sectorsize, dest + sofar) ) { return sofar; }
|
|
||||||
sofar += sectorsize;
|
|
||||||
numbytes -= sectorsize;
|
|
||||||
byteoffset += sectorsize;
|
|
||||||
}
|
|
||||||
|
|
||||||
if ( numbytes ) { return sofar + Read(byteoffset, dest + sofar, numbytes); }
|
|
||||||
return sofar;
|
|
||||||
}
|
|
||||||
|
|
||||||
size_t ATADrive::Write(off_t byteoffset, const uint8_t* src, size_t numbytes)
|
|
||||||
{
|
|
||||||
size_t sofar = 0;
|
|
||||||
size_t leadingbytes = byteoffset % sectorsize;
|
|
||||||
if ( leadingbytes || numbytes < sectorsize )
|
|
||||||
{
|
|
||||||
size_t wanted = sectorsize - leadingbytes;
|
|
||||||
if ( numbytes < wanted ) { wanted = numbytes; }
|
|
||||||
uint8_t temp[512 /*sectorsize*/];
|
|
||||||
if ( !ReadSector(byteoffset/sectorsize, temp) ) { return sofar; }
|
|
||||||
memcpy(temp + leadingbytes, src + sofar, wanted);
|
|
||||||
if ( !WriteSector(byteoffset/sectorsize, temp) ) { return sofar; }
|
|
||||||
sofar += wanted;
|
|
||||||
numbytes -= wanted;
|
|
||||||
byteoffset += wanted;
|
|
||||||
}
|
|
||||||
|
|
||||||
while ( sectorsize <= numbytes )
|
|
||||||
{
|
|
||||||
if ( !WriteSector(byteoffset/sectorsize, src + sofar) ) { return sofar; }
|
|
||||||
sofar += sectorsize;
|
|
||||||
numbytes -= sectorsize;
|
|
||||||
byteoffset += sectorsize;
|
|
||||||
}
|
|
||||||
|
|
||||||
if ( numbytes ) { return sofar + Write(byteoffset, src + sofar, numbytes); }
|
|
||||||
return sofar;
|
|
||||||
}
|
|
||||||
|
|
||||||
void ATADrive::Initialize()
|
|
||||||
{
|
|
||||||
bus->SelectDrive(driveid);
|
|
||||||
CPU::OutPortB(iobase + COMMAND, CTL_NO_INTERRUPT);
|
|
||||||
}
|
}
|
||||||
|
ATADrive* drive = new ATADrive(this, driveid, iobase, altport);
|
||||||
|
return drive;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool ATABus::SelectDrive(unsigned driveid)
|
||||||
|
{
|
||||||
|
if ( driveid == curdriveid ) { return true; }
|
||||||
|
if ( 1 < driveid ) { errno = EINVAL; return false; }
|
||||||
|
|
||||||
|
uint8_t drivemagic = 0xA0 | (driveid << 4);
|
||||||
|
CPU::OutPortB(iobase + DRIVE_SELECT, drivemagic);
|
||||||
|
Wait400NSecs(iobase);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
const size_t META_LBA28 = 60;
|
||||||
|
const size_t META_FLAGS = 83;
|
||||||
|
const size_t META_LBA48 = 100;
|
||||||
|
const uint16_t FLAG_LBA48 = (1<<10);
|
||||||
|
|
||||||
|
ATADrive::ATADrive(ATABus* bus, unsigned driveid, uint16_t portoffset, uint16_t altport)
|
||||||
|
{
|
||||||
|
this->atalock = KTHREAD_MUTEX_INITIALIZER;
|
||||||
|
this->bus = bus;
|
||||||
|
this->driveid = driveid;
|
||||||
|
this->iobase = portoffset;
|
||||||
|
this->altport = altport;
|
||||||
|
for ( size_t i = 0; i < 256; i++ )
|
||||||
|
{
|
||||||
|
meta[i] = CPU::InPortW(iobase + DATA);
|
||||||
|
}
|
||||||
|
lba48 = meta[META_FLAGS] & FLAG_LBA48;
|
||||||
|
if ( lba48 )
|
||||||
|
{
|
||||||
|
numsectors = *((uint64_t*) (meta + META_LBA48));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
numsectors = *((uint32_t*) (meta + META_LBA28));
|
||||||
|
}
|
||||||
|
sectorsize = 512; // TODO: Detect this!
|
||||||
|
Initialize();
|
||||||
|
}
|
||||||
|
|
||||||
|
ATADrive::~ATADrive()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
off_t ATADrive::GetSectorSize()
|
||||||
|
{
|
||||||
|
return sectorsize;
|
||||||
|
}
|
||||||
|
|
||||||
|
off_t ATADrive::GetNumSectors()
|
||||||
|
{
|
||||||
|
return numsectors;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ATADrive::PrepareIO(bool write, off_t sector)
|
||||||
|
{
|
||||||
|
if ( numsectors <= sector ) { errno = EINVAL; return false; }
|
||||||
|
if ( write && !ENABLE_DISKWRITE )
|
||||||
|
{
|
||||||
|
errno = EPERM;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
bus->SelectDrive(driveid);
|
||||||
|
uint8_t mode = (lba48) ? 0x40 : 0xE0;
|
||||||
|
mode |= driveid << 4;
|
||||||
|
mode |= (lba48) ? 0 : (sector >> 24) & 0x0F;
|
||||||
|
CPU::OutPortB(iobase + DRIVE_SELECT, mode);
|
||||||
|
uint16_t sectorcount = 1;
|
||||||
|
uint8_t sectorcountlow = sectorcount & 0xFF;
|
||||||
|
uint8_t sectorcounthigh = (sectorcount >> 8) & 0xFF;
|
||||||
|
if ( lba48 )
|
||||||
|
{
|
||||||
|
CPU::OutPortB(iobase + SECTOR_COUNT, sectorcounthigh);
|
||||||
|
CPU::OutPortB(iobase + LBA_LOW, (sector >> 24) & 0xFF);
|
||||||
|
CPU::OutPortB(iobase + LBA_MID, (sector >> 32) & 0xFF);
|
||||||
|
CPU::OutPortB(iobase + LBA_HIGH, (sector >> 40) & 0xFF);
|
||||||
|
}
|
||||||
|
CPU::OutPortB(iobase + SECTOR_COUNT, sectorcountlow);
|
||||||
|
CPU::OutPortB(iobase + LBA_LOW, sector & 0xFF);
|
||||||
|
CPU::OutPortB(iobase + LBA_MID, (sector >> 8) & 0xFF);
|
||||||
|
CPU::OutPortB(iobase + LBA_HIGH, (sector >> 16) & 0xFF);
|
||||||
|
uint8_t command = (write) ? CMD_WRITE : CMD_READ;
|
||||||
|
if ( lba48 ) { command = (write) ? CMD_WRITE_EXT : CMD_READ_EXT; }
|
||||||
|
CPU::OutPortB(iobase + COMMAND, command);
|
||||||
|
while ( true )
|
||||||
|
{
|
||||||
|
uint8_t status = CPU::InPortB(iobase + STATUS);
|
||||||
|
if ( status & STATUS_BUSY ) { continue; }
|
||||||
|
if ( status & STATUS_DATAREADY ) { break; }
|
||||||
|
if ( status & STATUS_ERROR ) { errno = EIO; return false; }
|
||||||
|
if ( status & STATUS_DRIVEFAULT ) { errno = EIO; return false; }
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ATADrive::ReadSector(off_t sector, uint8_t* dest)
|
||||||
|
{
|
||||||
|
ScopedLock lock(&atalock);
|
||||||
|
if ( !PrepareIO(false, sector) ) { return false; }
|
||||||
|
uint16_t* destword = (uint16_t*) dest;
|
||||||
|
for ( size_t i = 0; i < sectorsize/2; i++ )
|
||||||
|
{
|
||||||
|
destword[i] = CPU::InPortW(iobase + DATA);
|
||||||
|
}
|
||||||
|
Wait400NSecs(iobase);
|
||||||
|
uint8_t status = CPU::InPortB(iobase + STATUS);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ATADrive::WriteSector(off_t sector, const uint8_t* src)
|
||||||
|
{
|
||||||
|
ScopedLock lock(&atalock);
|
||||||
|
if ( !PrepareIO(true, sector) ) { return false; }
|
||||||
|
const uint16_t* srcword = (const uint16_t*) src;
|
||||||
|
for ( size_t i = 0; i < sectorsize/2; i++ )
|
||||||
|
{
|
||||||
|
CPU::OutPortW(iobase + DATA, srcword[i]);
|
||||||
|
}
|
||||||
|
Wait400NSecs(iobase);
|
||||||
|
CPU::OutPortB(iobase + COMMAND, CMD_FLUSH_CACHE);
|
||||||
|
while ( true )
|
||||||
|
{
|
||||||
|
uint8_t status = CPU::InPortB(iobase + STATUS);
|
||||||
|
if ( status & STATUS_ERROR ) { errno = EIO; return false; }
|
||||||
|
if ( status & STATUS_DRIVEFAULT ) { errno = EIO; return false; }
|
||||||
|
if ( !(status & STATUS_BUSY) ) { break; }
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t ATADrive::Read(off_t byteoffset, uint8_t* dest, size_t numbytes)
|
||||||
|
{
|
||||||
|
size_t sofar = 0;
|
||||||
|
size_t leadingbytes = byteoffset % sectorsize;
|
||||||
|
if ( leadingbytes || numbytes < sectorsize )
|
||||||
|
{
|
||||||
|
size_t wanted = sectorsize - leadingbytes;
|
||||||
|
if ( numbytes < wanted ) { wanted = numbytes; }
|
||||||
|
uint8_t temp[512 /*sectorsize*/];
|
||||||
|
if ( !ReadSector(byteoffset/sectorsize, temp) ) { return sofar; }
|
||||||
|
memcpy(dest + sofar, temp + leadingbytes, wanted);
|
||||||
|
sofar += wanted;
|
||||||
|
numbytes -= wanted;
|
||||||
|
byteoffset += wanted;
|
||||||
|
}
|
||||||
|
|
||||||
|
while ( sectorsize <= numbytes )
|
||||||
|
{
|
||||||
|
if ( !ReadSector(byteoffset/sectorsize, dest + sofar) ) { return sofar; }
|
||||||
|
sofar += sectorsize;
|
||||||
|
numbytes -= sectorsize;
|
||||||
|
byteoffset += sectorsize;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ( numbytes ) { return sofar + Read(byteoffset, dest + sofar, numbytes); }
|
||||||
|
return sofar;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t ATADrive::Write(off_t byteoffset, const uint8_t* src, size_t numbytes)
|
||||||
|
{
|
||||||
|
size_t sofar = 0;
|
||||||
|
size_t leadingbytes = byteoffset % sectorsize;
|
||||||
|
if ( leadingbytes || numbytes < sectorsize )
|
||||||
|
{
|
||||||
|
size_t wanted = sectorsize - leadingbytes;
|
||||||
|
if ( numbytes < wanted ) { wanted = numbytes; }
|
||||||
|
uint8_t temp[512 /*sectorsize*/];
|
||||||
|
if ( !ReadSector(byteoffset/sectorsize, temp) ) { return sofar; }
|
||||||
|
memcpy(temp + leadingbytes, src + sofar, wanted);
|
||||||
|
if ( !WriteSector(byteoffset/sectorsize, temp) ) { return sofar; }
|
||||||
|
sofar += wanted;
|
||||||
|
numbytes -= wanted;
|
||||||
|
byteoffset += wanted;
|
||||||
|
}
|
||||||
|
|
||||||
|
while ( sectorsize <= numbytes )
|
||||||
|
{
|
||||||
|
if ( !WriteSector(byteoffset/sectorsize, src + sofar) ) { return sofar; }
|
||||||
|
sofar += sectorsize;
|
||||||
|
numbytes -= sectorsize;
|
||||||
|
byteoffset += sectorsize;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ( numbytes ) { return sofar + Write(byteoffset, src + sofar, numbytes); }
|
||||||
|
return sofar;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ATADrive::Initialize()
|
||||||
|
{
|
||||||
|
bus->SelectDrive(driveid);
|
||||||
|
CPU::OutPortB(iobase + COMMAND, CTL_NO_INTERRUPT);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace Sortix
|
||||||
|
|
|
||||||
100
sortix/ata.h
100
sortix/ata.h
|
|
@ -27,65 +27,67 @@
|
||||||
|
|
||||||
#include <sortix/kernel/kthread.h>
|
#include <sortix/kernel/kthread.h>
|
||||||
|
|
||||||
namespace Sortix
|
namespace Sortix {
|
||||||
|
|
||||||
|
class ATABus;
|
||||||
|
class ATADrive;
|
||||||
|
|
||||||
|
class ATABus
|
||||||
{
|
{
|
||||||
class ATABus;
|
public:
|
||||||
class ATADrive;
|
ATABus(uint16_t portoffset, uint16_t altport);
|
||||||
|
~ATABus();
|
||||||
|
|
||||||
class ATABus
|
public:
|
||||||
{
|
ATADrive* Instatiate(unsigned driveid);
|
||||||
public:
|
bool SelectDrive(unsigned driveid);
|
||||||
ATABus(uint16_t portoffset, uint16_t altport);
|
|
||||||
~ATABus();
|
|
||||||
|
|
||||||
public:
|
private:
|
||||||
ATADrive* Instatiate(unsigned driveid);
|
unsigned curdriveid;
|
||||||
bool SelectDrive(unsigned driveid);
|
uint16_t iobase;
|
||||||
|
uint16_t altport;
|
||||||
|
|
||||||
private:
|
};
|
||||||
unsigned curdriveid;
|
|
||||||
uint16_t iobase;
|
|
||||||
uint16_t altport;
|
|
||||||
|
|
||||||
};
|
class ATADrive
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
off_t GetSectorSize();
|
||||||
|
off_t GetNumSectors();
|
||||||
|
off_t GetSize() { return GetSectorSize() * GetNumSectors(); }
|
||||||
|
bool ReadSector(off_t sector, uint8_t* dest);
|
||||||
|
bool WriteSector(off_t sector, const uint8_t* src);
|
||||||
|
size_t Read(off_t byteoffset, uint8_t* dest, size_t numbytes);
|
||||||
|
size_t Write(off_t byteoffset, const uint8_t* src, size_t numbytes);
|
||||||
|
|
||||||
class ATADrive
|
public:
|
||||||
{
|
ATADrive(ATABus* bus, unsigned driveid, uint16_t portoffset, uint16_t altport);
|
||||||
public:
|
~ATADrive();
|
||||||
off_t GetSectorSize();
|
|
||||||
off_t GetNumSectors();
|
|
||||||
off_t GetSize() { return GetSectorSize() * GetNumSectors(); }
|
|
||||||
bool ReadSector(off_t sector, uint8_t* dest);
|
|
||||||
bool WriteSector(off_t sector, const uint8_t* src);
|
|
||||||
size_t Read(off_t byteoffset, uint8_t* dest, size_t numbytes);
|
|
||||||
size_t Write(off_t byteoffset, const uint8_t* src, size_t numbytes);
|
|
||||||
|
|
||||||
public:
|
private:
|
||||||
ATADrive(ATABus* bus, unsigned driveid, uint16_t portoffset, uint16_t altport);
|
void Initialize();
|
||||||
~ATADrive();
|
bool PrepareIO(bool write, off_t sector);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void Initialize();
|
kthread_mutex_t atalock;
|
||||||
bool PrepareIO(bool write, off_t sector);
|
unsigned driveid;
|
||||||
|
uint16_t meta[256];
|
||||||
|
uint16_t iobase;
|
||||||
|
uint16_t altport;
|
||||||
|
ATABus* bus;
|
||||||
|
bool lba48;
|
||||||
|
size_t sectorsize;
|
||||||
|
off_t numsectors;
|
||||||
|
|
||||||
private:
|
};
|
||||||
kthread_mutex_t atalock;
|
|
||||||
unsigned driveid;
|
|
||||||
uint16_t meta[256];
|
|
||||||
uint16_t iobase;
|
|
||||||
uint16_t altport;
|
|
||||||
ATABus* bus;
|
|
||||||
bool lba48;
|
|
||||||
size_t sectorsize;
|
|
||||||
off_t numsectors;
|
|
||||||
|
|
||||||
};
|
namespace ATA {
|
||||||
|
|
||||||
namespace ATA
|
void Init();
|
||||||
{
|
ATABus* CreateBus(uint16_t portoffset, uint16_t altport);
|
||||||
void Init();
|
|
||||||
ATABus* CreateBus(uint16_t portoffset, uint16_t altport);
|
} // namespace ATA
|
||||||
}
|
|
||||||
}
|
} // namespace Sortix
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue