mirror of
https://github.com/danbulant/Cosmos
synced 2026-05-30 21:00:30 +00:00
349 lines
15 KiB
C#
349 lines
15 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using Cosmos.HAL;
|
|
using Cosmos.Debug.Kernel;
|
|
using Cosmos.HAL.BlockDevice;
|
|
using Cosmos.HAL.BlockDevice.Ports;
|
|
using Cosmos.HAL.BlockDevice.Registers;
|
|
using Cosmos.Core;
|
|
using Cosmos.Core.Memory.Old;
|
|
|
|
namespace Cosmos.HAL.BlockDevice
|
|
{
|
|
public class AHCI
|
|
{
|
|
internal static Debugger mAHCIDebugger = new Debugger("HAL", "AHCI");
|
|
internal static PCIDevice xDevice = HAL.PCI.GetDevice(0x01, 0x06);
|
|
|
|
private static List<StoragePort> mPorts = new List<StoragePort>();
|
|
private static GenericRegisters mGeneric;
|
|
private static ulong mABAR;
|
|
|
|
// Capabilities
|
|
#region Capabilities
|
|
private static bool Supports64bitAddressing;
|
|
private static bool SupportsNativeCommandQueuing;
|
|
private static bool SupportsSNotificationRegister;
|
|
private static bool SupportsMechanicalPresenceSwitch;
|
|
private static bool SupportsStaggeredSpinup;
|
|
private static bool SupportsAggressiveLinkPowerManagement;
|
|
private static bool SupportsActivityLED;
|
|
private static bool SupportsCommandListOverride;
|
|
private static uint InterfaceSpeedSupport;
|
|
private static bool SupportsAHCIModeOnly;
|
|
private static bool SupportsPortMutliplier;
|
|
private static bool FISBasedSwitchingSupported;
|
|
private static bool PIOMultipleDRQBlock;
|
|
private static bool SlumberStateCapable;
|
|
private static bool PartialStateCapable;
|
|
private static uint NumOfCommandSlots;
|
|
private static bool CommandCompletionCoalsecingSupported;
|
|
private static bool EnclosureManagementSupported;
|
|
private static bool SupportsExternalSATA;
|
|
private static uint NumOfPorts;
|
|
#endregion
|
|
|
|
// Constants
|
|
public const ulong RegularSectorSize = 512UL;
|
|
|
|
// Informations
|
|
public string SerialNo;
|
|
public string Version
|
|
{
|
|
get => ((byte)mGeneric.AHCIVersion >> 24) + (byte)(mGeneric.AHCIVersion >> 16) + "." + (byte)(mGeneric.AHCIVersion >> 8) + ((byte)(mGeneric.AHCIVersion) > 0 ? "." + (byte)mGeneric.AHCIVersion : "");
|
|
}
|
|
|
|
internal static void InitDriver()
|
|
{
|
|
if (xDevice != null)
|
|
{
|
|
AHCI Driver = new AHCI(xDevice);
|
|
}
|
|
}
|
|
|
|
internal override PCIDevice GetDevice() => xDevice;
|
|
|
|
public AHCI(PCIDevice aAHCIDevice)
|
|
{
|
|
aAHCIDevice.EnableBusMaster(true);
|
|
aAHCIDevice.EnableMemory(true);
|
|
|
|
mABAR = aAHCIDevice.BaseAddressBar[5].BaseAddress;
|
|
mGeneric = new GenericRegisters(aAHCIDevice.BaseAddressBar[5].BaseAddress);
|
|
mGeneric.GlobalHostControl |= (1U << 31);
|
|
|
|
GetCapabilities();
|
|
mPorts.Capacity = (int)NumOfPorts;
|
|
GetPorts();
|
|
|
|
foreach(StoragePort xPort in mPorts)
|
|
{
|
|
if(xPort.mPortType == PortType.SATA)
|
|
{
|
|
mAHCIDebugger.Send($"{xPort.mPortName} Port [0:{xPort.mPortNumber}]");
|
|
var xMBRData = new byte[512];
|
|
xPort.ReadBlock(0UL, 1U, xMBRData);
|
|
var xMBR = new MBR(xMBRData);
|
|
|
|
if (xMBR.EBRLocation != 0)
|
|
{
|
|
// EBR Detected!
|
|
var xEBRData = new byte[512];
|
|
xPort.ReadBlock(xMBR.EBRLocation, 1U, xEBRData);
|
|
var xEBR = new EBR(xEBRData);
|
|
for (int i = 0; i < xEBR.Partitions.Count; i++)
|
|
{
|
|
//var xPart = xEBR.Partitions[i];
|
|
//var xPartDevice = new Partition(xSATA, xPart.StartSector, xPart.SectorCount);
|
|
//Devices.Add(xPartDevice);
|
|
}
|
|
}
|
|
|
|
mAHCIDebugger.Send($"Number of MBR partitions in port [0:{xPort.mPortNumber}]: ");
|
|
mAHCIDebugger.SendNumber(xMBR.Partitions.Count);
|
|
for (int i = 0; i < xMBR.Partitions.Count; i++)
|
|
{
|
|
var xPart = xMBR.Partitions[i];
|
|
if (xPart == null)
|
|
{
|
|
Console.WriteLine("Null partition found at idx: " + i);
|
|
}
|
|
else
|
|
{
|
|
var xPartDevice = new Partition(xPort, xPart.StartSector, xPart.SectorCount);
|
|
BlockDevice.Devices.Add(xPartDevice);
|
|
Console.WriteLine("Found partition at idx: " + i);
|
|
}
|
|
}
|
|
}
|
|
else if (xPort.mPortType == PortType.SATAPI)
|
|
{
|
|
mAHCIDebugger.Send($"{xPort.mPortName} Port 0:{xPort.mPortNumber}");
|
|
// TODO: Implement ISO-9660 or UDF and Fix SATAPI
|
|
}
|
|
}
|
|
}
|
|
|
|
private void Wait(int microsecondsTimeout)
|
|
{
|
|
byte xVoid;
|
|
for (int i = 0; i < microsecondsTimeout; i++)
|
|
{
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
xVoid = Core.Global.BaseIOGroups.TextScreen.Data1.Byte;
|
|
}
|
|
}
|
|
|
|
private void GetCapabilities()
|
|
{
|
|
NumOfPorts = mGeneric.Capabilities & 0x1F;
|
|
SupportsExternalSATA = Convert.ToBoolean(mGeneric.Capabilities >> 5 & 1);
|
|
EnclosureManagementSupported = Convert.ToBoolean(mGeneric.Capabilities >> 6 & 1);
|
|
CommandCompletionCoalsecingSupported = Convert.ToBoolean(mGeneric.Capabilities >> 7 & 1);
|
|
NumOfCommandSlots = mGeneric.Capabilities >> 8 & 0x1F;
|
|
PartialStateCapable = Convert.ToBoolean(mGeneric.Capabilities >> 13 & 1);
|
|
SlumberStateCapable = Convert.ToBoolean(mGeneric.Capabilities >> 14 & 1);
|
|
PIOMultipleDRQBlock = Convert.ToBoolean(mGeneric.Capabilities >> 15 & 1);
|
|
FISBasedSwitchingSupported = Convert.ToBoolean(mGeneric.Capabilities >> 16 & 1);
|
|
SupportsPortMutliplier = Convert.ToBoolean(mGeneric.Capabilities >> 17 & 1);
|
|
SupportsAHCIModeOnly = Convert.ToBoolean(mGeneric.Capabilities >> 18 & 1);
|
|
InterfaceSpeedSupport = mGeneric.Capabilities >> 20 & 0x0F;
|
|
SupportsCommandListOverride = Convert.ToBoolean(mGeneric.Capabilities >> 24 & 1);
|
|
SupportsActivityLED = Convert.ToBoolean(mGeneric.Capabilities >> 25 & 1);
|
|
SupportsAggressiveLinkPowerManagement = Convert.ToBoolean(mGeneric.Capabilities >> 26 & 1);
|
|
SupportsStaggeredSpinup = Convert.ToBoolean(mGeneric.Capabilities >> 27 & 1);
|
|
SupportsMechanicalPresenceSwitch = Convert.ToBoolean(mGeneric.Capabilities >> 28 & 1);
|
|
SupportsSNotificationRegister = Convert.ToBoolean(mGeneric.Capabilities >> 29 & 1);
|
|
SupportsNativeCommandQueuing = Convert.ToBoolean(mGeneric.Capabilities >> 30 & 1);
|
|
Supports64bitAddressing = Convert.ToBoolean(mGeneric.Capabilities >> 31 & 1);
|
|
}
|
|
|
|
private void GetPorts()
|
|
{
|
|
// Search for disks
|
|
var xImplementedPort = mGeneric.ImplementedPorts;
|
|
var xPort = 0;
|
|
while (xPort < NumOfPorts)
|
|
{
|
|
if ((xImplementedPort & 1) != 0)
|
|
{
|
|
PortRegisters xPortReg = new PortRegisters((uint)mABAR + 0x100 + (uint)(0x80 * xPort), (uint)xPort);
|
|
PortType PortType = CheckPortType(xPortReg);
|
|
xPortReg.mPortType = PortType;
|
|
var xPortString = "0:" + ((xPort.ToString().Length <= 1) ? xPort.ToString().PadLeft(1, '0') : xPort.ToString());
|
|
if (PortType == PortType.SATA) // If Port Type was SATA.
|
|
{
|
|
Console.WriteLine("Initializing Port " + xPortString + " with type SATA");
|
|
UInt16[] xSectorData = new UInt16[256];
|
|
PortRebase(xPortReg, (uint)xPort);
|
|
SATA xSATAPort = new SATA(xPortReg);
|
|
mPorts.Add(xSATAPort);
|
|
}
|
|
else if (PortType == PortType.SATAPI) // If Port Type was SATAPI.
|
|
{
|
|
mAHCIDebugger.Send("SATAPI");
|
|
Console.WriteLine("Initializing Port " + xPortString + " with type Serial ATAPI");
|
|
//PortRebase(xPortReg, (uint)xPort);
|
|
//SATAPI xSATAPIPort = new SATAPI(xPortReg);
|
|
//mPorts.Add(xSATAPIPort);
|
|
}
|
|
else if (PortType == PortType.SEMB) // If Port Type was SEMB.
|
|
{
|
|
Console.WriteLine("SEMB Drive at port " + xPortString + " found, which is not supported yet!");
|
|
}
|
|
else if (PortType == PortType.PM) // If Port Type was Port Mulitplier.
|
|
{
|
|
Console.WriteLine("Port Multiplier Drive at port " + xPortString + " found, which is not supported yet!");
|
|
}
|
|
else if (PortType == PortType.Nothing)
|
|
mAHCIDebugger.Send("No drive found at port: " + xPortString);
|
|
else
|
|
mAHCIDebugger.Send("Unknown drive found with signature: 0x" + xPortReg.SIG);// If Implemented Port value was not zero and non of the above.
|
|
}
|
|
xImplementedPort >>= 1;
|
|
xPort++;
|
|
}
|
|
}
|
|
|
|
private PortType CheckPortType(PortRegisters aPort)
|
|
{
|
|
var xIPM = (InterfacePowerManagementStatus)((aPort.SSTS >> 8) & 0x0F);
|
|
var xSPD = (CurrentInterfaceSpeedStatus)((aPort.SSTS >> 4) & 0x0F);
|
|
var xDET = (DeviceDetectionStatus)(aPort.SSTS & 0x0F);
|
|
var xSignature = aPort.SIG;
|
|
//var LBALow = (byte)(Port.SIG >> 08);
|
|
//var LBAMedium = (byte)(Port.SIG >> 16);
|
|
//var LBAHigh = (byte)(Port.SIG >> 24);
|
|
|
|
if (xIPM != InterfacePowerManagementStatus.Active)
|
|
return PortType.Nothing;
|
|
if (xDET != DeviceDetectionStatus.DeviceDetectedWithPhy)
|
|
return PortType.Nothing;
|
|
|
|
switch (xSignature >> 16)
|
|
{
|
|
case 0xEB14: return PortType.SATAPI;
|
|
case 0xC33C: return PortType.SEMB;
|
|
case 0x9669: return PortType.PM;
|
|
case 0xFFFF: return PortType.Nothing;
|
|
default: return PortType.SATA;
|
|
}
|
|
}
|
|
|
|
private void PortRebase(PortRegisters aPort, uint aPortNumber)
|
|
{
|
|
mAHCIDebugger.Send("Stop");
|
|
if (!StopCMD(aPort)) aPort.SCTL = 1;
|
|
|
|
ulong mCLBAddress = Heap.MemAlloc(1024);
|
|
aPort.CLB = (uint)mCLBAddress & 0xFFFFFFFF;
|
|
|
|
var mFBAddress = Heap.MemAlloc(256);
|
|
aPort.FB = (uint)mFBAddress & 0xFFFFFFFF;
|
|
|
|
aPort.SERR = 1;
|
|
aPort.IS = 0;
|
|
aPort.IE = 0;
|
|
|
|
new MemoryBlock(aPort.CLB, 1024).Fill(0);
|
|
new MemoryBlock(aPort.FB, 256).Fill(0);
|
|
|
|
GetCommandHeader(aPort); // Rebasing Command Header
|
|
|
|
StartCMD(aPort);
|
|
aPort.IS = 0;
|
|
aPort.IE = 0xFFFFFFFF;
|
|
}
|
|
|
|
private static HBACommandHeader[] GetCommandHeader(PortRegisters aPort)
|
|
{
|
|
var xCTBAAddress = Heap.MemAlloc(256);
|
|
HBACommandHeader[] xCMDHeader = new HBACommandHeader[32];
|
|
for (uint i = 0; i < xCMDHeader.Length; i++)
|
|
{
|
|
xCMDHeader[i] = new HBACommandHeader(aPort.CLB, i)
|
|
{
|
|
PRDTL = 8, // 8 prdt entries per command table
|
|
// 256 bytes per command table, 64+16+48+16*8
|
|
CTBA = xCTBAAddress + (256 * i),
|
|
//TODO: Use the below register as the upper base address if HBA
|
|
// supports S64A (64-bit Addressing)
|
|
CTBAU = 0
|
|
};
|
|
new MemoryBlock(xCMDHeader[i].CTBA, 256).Fill(0);
|
|
}
|
|
return xCMDHeader;
|
|
}
|
|
|
|
private void StartCMD(PortRegisters aPort)
|
|
{
|
|
int xSpin;
|
|
for (xSpin = 0; xSpin < 101; xSpin++)
|
|
{
|
|
if ((aPort.CMD & (uint)CommandAndStatus.CMDListRunning) == 0) break;
|
|
Wait(5000);
|
|
}
|
|
if (xSpin == 101) return;
|
|
|
|
aPort.CMD |= (1 << 4);
|
|
aPort.CMD |= (1 << 0);
|
|
}
|
|
|
|
private bool StopCMD(PortRegisters aPort)
|
|
{
|
|
int xSpin;
|
|
aPort.CMD &= ~(1U << 0); //Bit 0
|
|
|
|
for (xSpin = 0; xSpin < 101; xSpin++)
|
|
{
|
|
if ((aPort.CMD & (uint)CommandAndStatus.CMDListRunning) == 0) break;
|
|
Wait(5000);
|
|
}
|
|
if (xSpin == 101) return false;
|
|
|
|
for (xSpin = 0; xSpin < 101; xSpin++)
|
|
{
|
|
if ((aPort.CI == 0)) break;
|
|
Wait(50);
|
|
}
|
|
if (xSpin == 101) return false;
|
|
|
|
aPort.CMD &= ~(1U << 4); //Bit 4
|
|
|
|
if (SupportsCommandListOverride)
|
|
{
|
|
if ((aPort.TFD & (uint)ATADeviceStatus.Busy) != 0)
|
|
{
|
|
aPort.CMD |= (1U << 3);
|
|
}
|
|
}
|
|
|
|
for (xSpin = 0; xSpin < 101; xSpin++)
|
|
{
|
|
if ((aPort.CMD & (uint)CommandAndStatus.CMDListRunning) == 0 &&
|
|
(aPort.CMD & (uint)CommandAndStatus.FISRecieveRunning) == 0 &&
|
|
(aPort.CMD & (uint)CommandAndStatus.StartProccess) == 0 &&
|
|
(aPort.CMD & (uint)CommandAndStatus.FISRecieveEnable) == 0) break;
|
|
Wait(5000);
|
|
}
|
|
if (xSpin == 101)
|
|
{
|
|
if (SupportsCommandListOverride)
|
|
aPort.CMD |= (1U << 3);
|
|
else
|
|
aPort.CMD &= ~(1U << 3);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|