* DiscImageChef.Devices/Structs.cs:

* DiscImageChef.Devices/DiscImageChef.Devices.csproj:
	  Added structs with ATA registers.

	* DiscImageChef.Devices/Command.cs:
	* DiscImageChef.Devices/Device/Commands.cs:
	  Added ATA commands.

	* DiscImageChef.Devices/Enums.cs:
	  Added SCSI's ATA Command Pass-Through enumerations.

	* DiscImageChef.Devices/Linux/Command.cs:
	  Added ATA commands using libATA's SATL.
This commit is contained in:
2015-10-14 02:53:46 +01:00
parent d7546d4e88
commit 8d121d22b7
7 changed files with 574 additions and 0 deletions

View File

@@ -1,3 +1,19 @@
2015-10-14 Natalia Portillo <claunia@claunia.com>
* Structs.cs:
* DiscImageChef.Devices.csproj:
Added structs with ATA registers.
* Command.cs:
* Device/Commands.cs:
Added ATA commands.
* Enums.cs:
Added SCSI's ATA Command Pass-Through enumerations.
* Linux/Command.cs:
Added ATA commands using libATA's SATL.
2015-10-14 Natalia Portillo <claunia@claunia.com> 2015-10-14 Natalia Portillo <claunia@claunia.com>
* Linux/Enums.cs: * Linux/Enums.cs:

View File

@@ -128,6 +128,102 @@ namespace DiscImageChef.Devices
throw new InvalidOperationException(String.Format("Platform {0} not yet supported.", ptID)); throw new InvalidOperationException(String.Format("Platform {0} not yet supported.", ptID));
} }
} }
public static int SendAtaCommand(object fd, Structs.AtaRegistersCHS registers,
out Structs.AtaErrorRegistersCHS errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
Interop.PlatformID ptID = DetectOS.GetRealPlatformID();
return SendAtaCommand(ptID, fd, registers, out errorRegisters, protocol,
transferRegister, ref buffer, timeout, transferBlocks, out duration, out sense);
}
public static int SendAtaCommand(Interop.PlatformID ptID, object fd, Structs.AtaRegistersCHS registers,
out Structs.AtaErrorRegistersCHS errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
switch (ptID)
{
case Interop.PlatformID.Win32NT:
{
throw new NotImplementedException();
}
case Interop.PlatformID.Linux:
{
return Linux.Command.SendAtaCommand((int)fd, registers, out errorRegisters, protocol,
transferRegister, ref buffer, timeout, transferBlocks, out duration, out sense);
}
default:
throw new InvalidOperationException(String.Format("Platform {0} not yet supported.", ptID));
}
}
public static int SendAtaCommand(object fd, Structs.AtaRegistersLBA28 registers,
out Structs.AtaErrorRegistersLBA28 errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
Interop.PlatformID ptID = DetectOS.GetRealPlatformID();
return SendAtaCommand(ptID, fd, registers, out errorRegisters, protocol,
transferRegister, ref buffer, timeout, transferBlocks, out duration, out sense);
}
public static int SendAtaCommand(Interop.PlatformID ptID, object fd, Structs.AtaRegistersLBA28 registers,
out Structs.AtaErrorRegistersLBA28 errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
switch (ptID)
{
case Interop.PlatformID.Win32NT:
{
throw new NotImplementedException();
}
case Interop.PlatformID.Linux:
{
return Linux.Command.SendAtaCommand((int)fd, registers, out errorRegisters, protocol,
transferRegister, ref buffer, timeout, transferBlocks, out duration, out sense);
}
default:
throw new InvalidOperationException(String.Format("Platform {0} not yet supported.", ptID));
}
}
public static int SendAtaCommand(object fd, Structs.AtaRegistersLBA48 registers,
out Structs.AtaErrorRegistersLBA48 errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
Interop.PlatformID ptID = DetectOS.GetRealPlatformID();
return SendAtaCommand(ptID, fd, registers, out errorRegisters, protocol,
transferRegister, ref buffer, timeout, transferBlocks, out duration, out sense);
}
public static int SendAtaCommand(Interop.PlatformID ptID, object fd, Structs.AtaRegistersLBA48 registers,
out Structs.AtaErrorRegistersLBA48 errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
switch (ptID)
{
case Interop.PlatformID.Win32NT:
{
throw new NotImplementedException();
}
case Interop.PlatformID.Linux:
{
return Linux.Command.SendAtaCommand((int)fd, registers, out errorRegisters, protocol,
transferRegister, ref buffer, timeout, transferBlocks, out duration, out sense);
}
default:
throw new InvalidOperationException(String.Format("Platform {0} not yet supported.", ptID));
}
}
} }
} }

View File

@@ -56,6 +56,30 @@ namespace DiscImageChef.Devices
{ {
return Command.SendScsiCommand(platformID, fd, cdb, ref buffer, out senseBuffer, timeout, direction, out duration, out sense); return Command.SendScsiCommand(platformID, fd, cdb, ref buffer, out senseBuffer, timeout, direction, out duration, out sense);
} }
public int SendAtaCommand(Structs.AtaRegistersCHS registers, out Structs.AtaErrorRegistersCHS errorRegisters,
Enums.AtaProtocol protocol, Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
return Command.SendAtaCommand(platformID, fd, registers, out errorRegisters, protocol, transferRegister,
ref buffer, timeout, transferBlocks, out duration, out sense);
}
public int SendAtaCommand(Structs.AtaRegistersLBA28 registers, out Structs.AtaErrorRegistersLBA28 errorRegisters,
Enums.AtaProtocol protocol, Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
return Command.SendAtaCommand(platformID, fd, registers, out errorRegisters, protocol, transferRegister,
ref buffer, timeout, transferBlocks, out duration, out sense);
}
public int SendAtaCommand(Structs.AtaRegistersLBA48 registers, out Structs.AtaErrorRegistersLBA48 errorRegisters,
Enums.AtaProtocol protocol, Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
return Command.SendAtaCommand(platformID, fd, registers, out errorRegisters, protocol, transferRegister,
ref buffer, timeout, transferBlocks, out duration, out sense);
}
} }
} }

View File

@@ -50,6 +50,7 @@
<Compile Include="Device\Destructor.cs" /> <Compile Include="Device\Destructor.cs" />
<Compile Include="Device\Commands.cs" /> <Compile Include="Device\Commands.cs" />
<Compile Include="Device\ScsiCommands.cs" /> <Compile Include="Device\ScsiCommands.cs" />
<Compile Include="Structs.cs" />
</ItemGroup> </ItemGroup>
<Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" /> <Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" />
<ItemGroup> <ItemGroup>

View File

@@ -2398,6 +2398,87 @@ namespace DiscImageChef.Devices
Unspecified Unspecified
} }
#region SCSI's ATA Command Pass-Through
public enum AtaProtocol : byte
{
/// <summary>
/// Requests a device hard reset (pin 1)
/// </summary>
HardReset = 0,
/// <summary>
/// Requests a device soft reset (COMRESET issue)
/// </summary>
SoftReset = 1,
/// <summary>
/// No data is to be transfered
/// </summary>
NonData = 3,
/// <summary>
/// Requests a host->device transfer using PIO
/// </summary>
PioIn = 4,
/// <summary>
/// Requests a device->host transfer using PIO
/// </summary>
PioOut = 5,
/// <summary>
/// Requests a DMA transfer
/// </summary>
Dma = 6,
/// <summary>
/// Requests to queue a DMA transfer
/// </summary>
DmaQueued = 7,
/// <summary>
/// Requests device diagnostics
/// </summary>
DeviceDiagnostic = 8,
/// <summary>
/// Requets device reset
/// </summary>
DeviceReset = 9,
/// <summary>
/// Requests a host->device transfer using UltraDMA
/// </summary>
UDmaIn = 10,
/// <summary>
/// Requests a device->host transfer using UltraDMA
/// </summary>
UDmaOut = 11,
/// <summary>
/// Unknown Serial ATA
/// </summary>
FPDma = 12,
/// <summary>
/// Requests the Extended ATA Status Return Descriptor
/// </summary>
ReturnResponse = 15
}
/// <summary>
/// Indicates the STL which ATA register contains the length of data to
/// be transfered
/// </summary>
public enum AtaTransferRegister : byte
{
/// <summary>
/// There is no transfer
/// </summary>
NoTransfer = 0,
/// <summary>
/// FEATURE register contains the data length
/// </summary>
Feature = 1,
/// <summary>
/// SECTOR_COUNT register contains the data length
/// </summary>
SectorCount = 2,
/// <summary>
/// The STPSIU contains the data length
/// </summary>
SPTSIU = 3
}
#endregion SCSI's ATA Command Pass-Through
} }
} }

View File

@@ -98,6 +98,248 @@ namespace DiscImageChef.Devices.Linux
return error; return error;
} }
static ScsiIoctlDirection AtaProtocolToScsiDirection(Enums.AtaProtocol protocol)
{
switch (protocol)
{
case Enums.AtaProtocol.DeviceDiagnostic:
case Enums.AtaProtocol.DeviceReset:
case Enums.AtaProtocol.HardReset:
case Enums.AtaProtocol.NonData:
case Enums.AtaProtocol.SoftReset:
case Enums.AtaProtocol.ReturnResponse:
return ScsiIoctlDirection.None;
case Enums.AtaProtocol.PioIn:
case Enums.AtaProtocol.UDmaIn:
return ScsiIoctlDirection.Out;
case Enums.AtaProtocol.PioOut:
case Enums.AtaProtocol.UDmaOut:
return ScsiIoctlDirection.In;
default:
return ScsiIoctlDirection.Unspecified;
}
}
internal static int SendAtaCommand(int fd, Structs.AtaRegistersCHS registers,
out Structs.AtaErrorRegistersCHS errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
duration = 0;
sense = false;
errorRegisters = new Structs.AtaErrorRegistersCHS();
if (buffer == null)
return -1;
byte[] cdb = new byte[12];
cdb[0] = (byte)Enums.ScsiCommands.AtaPassThrough;
cdb[1] = (byte)(((byte)protocol << 1) & 0x1E);
if (transferRegister != Enums.AtaTransferRegister.NoTransfer &&
protocol != Enums.AtaProtocol.NonData)
{
switch (protocol)
{
case Enums.AtaProtocol.PioIn:
case Enums.AtaProtocol.UDmaIn:
cdb[2] = 0x08;
break;
default:
cdb[2] = 0x00;
break;
}
if (transferBlocks)
cdb[2] |= 0x04;
cdb[2] |= (byte)((int)transferRegister & 0x03);
}
cdb[3] = registers.feature;
cdb[4] = registers.sectorCount;
cdb[5] = registers.sector;
cdb[6] = registers.cylinderHigh;
cdb[7] = registers.cylinderLow;
cdb[8] = registers.deviceHead;
cdb[9] = registers.command;
byte[] senseBuffer;
int error = SendScsiCommand(fd, cdb, ref buffer, out senseBuffer, timeout, AtaProtocolToScsiDirection(protocol), out duration, out sense);
// Now get error registers
byte[] returnCdb = new byte[12];
returnCdb[0] = (byte)Enums.ScsiCommands.AtaPassThrough;
returnCdb[1] = (byte)(((byte)Enums.AtaProtocol.ReturnResponse << 1) & 0x1E);
byte[] returnBuffer = new byte[14];
bool returnSense;
double returnDuration;
SendScsiCommand(fd, returnCdb, ref returnBuffer, out senseBuffer, timeout, ScsiIoctlDirection.In, out returnDuration, out returnSense);
if (returnBuffer[0] != 0x09 && returnBuffer[1] != 0x0C)
return error;
errorRegisters.error = returnBuffer[3];
errorRegisters.sectorCount = returnBuffer[5];
errorRegisters.sector = returnBuffer[7];
errorRegisters.cylinderHigh = returnBuffer[9];
errorRegisters.cylinderLow = returnBuffer[11];
errorRegisters.deviceHead = returnBuffer[12];
errorRegisters.status = returnBuffer[13];
sense |= error != 0;
return error;
}
internal static int SendAtaCommand(int fd, Structs.AtaRegistersLBA28 registers,
out Structs.AtaErrorRegistersLBA28 errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
duration = 0;
sense = false;
errorRegisters = new Structs.AtaErrorRegistersLBA28();
if (buffer == null)
return -1;
byte[] cdb = new byte[12];
cdb[0] = (byte)Enums.ScsiCommands.AtaPassThrough;
cdb[1] = (byte)(((byte)protocol << 1) & 0x1E);
if (transferRegister != Enums.AtaTransferRegister.NoTransfer &&
protocol != Enums.AtaProtocol.NonData)
{
switch (protocol)
{
case Enums.AtaProtocol.PioIn:
case Enums.AtaProtocol.UDmaIn:
cdb[2] = 0x08;
break;
default:
cdb[2] = 0x00;
break;
}
if (transferBlocks)
cdb[2] |= 0x04;
cdb[2] |= (byte)((int)transferRegister & 0x03);
}
cdb[3] = registers.feature;
cdb[4] = registers.sectorCount;
cdb[5] = registers.lbaLow;
cdb[6] = registers.lbaMid;
cdb[7] = registers.lbaHigh;
cdb[8] = registers.deviceHead;
cdb[9] = registers.command;
byte[] senseBuffer;
int error = SendScsiCommand(fd, cdb, ref buffer, out senseBuffer, timeout, AtaProtocolToScsiDirection(protocol), out duration, out sense);
// Now get error registers
byte[] returnCdb = new byte[12];
returnCdb[0] = (byte)Enums.ScsiCommands.AtaPassThrough;
returnCdb[1] = (byte)(((byte)Enums.AtaProtocol.ReturnResponse << 1) & 0x1E);
byte[] returnBuffer = new byte[14];
bool returnSense;
double returnDuration;
SendScsiCommand(fd, returnCdb, ref returnBuffer, out senseBuffer, timeout, ScsiIoctlDirection.In, out returnDuration, out returnSense);
if (returnBuffer[0] != 0x09 && returnBuffer[1] != 0x0C)
return error;
errorRegisters.error = returnBuffer[3];
errorRegisters.sectorCount = returnBuffer[5];
errorRegisters.lbaLow = returnBuffer[7];
errorRegisters.lbaMid = returnBuffer[9];
errorRegisters.lbaHigh = returnBuffer[11];
errorRegisters.deviceHead = returnBuffer[12];
errorRegisters.status = returnBuffer[13];
sense |= error != 0;
return error;
}
internal static int SendAtaCommand(int fd, Structs.AtaRegistersLBA48 registers,
out Structs.AtaErrorRegistersLBA48 errorRegisters, Enums.AtaProtocol protocol,
Enums.AtaTransferRegister transferRegister, ref byte[] buffer, uint timeout,
bool transferBlocks, out double duration, out bool sense)
{
duration = 0;
sense = false;
errorRegisters = new Structs.AtaErrorRegistersLBA48();
if (buffer == null)
return -1;
byte[] cdb = new byte[16];
cdb[0] = (byte)Enums.ScsiCommands.AtaPassThrough16;
cdb[1] |= 0x01;
cdb[1] = (byte)(((byte)protocol << 1) & 0x1E);
if (transferRegister != Enums.AtaTransferRegister.NoTransfer &&
protocol != Enums.AtaProtocol.NonData)
{
switch (protocol)
{
case Enums.AtaProtocol.PioIn:
case Enums.AtaProtocol.UDmaIn:
cdb[2] = 0x08;
break;
default:
cdb[2] = 0x00;
break;
}
if (transferBlocks)
cdb[2] |= 0x04;
cdb[2] |= (byte)((int)transferRegister & 0x03);
}
cdb[3] = (byte)((registers.feature & 0xFF00) >> 8);
cdb[4] = (byte)(registers.feature & 0xFF);
cdb[5] = (byte)((registers.sectorCount & 0xFF00) >> 8);
cdb[6] = (byte)(registers.sectorCount & 0xFF);
cdb[7] = (byte)((registers.lbaLow & 0xFF00) >> 8);
cdb[8] = (byte)(registers.lbaLow & 0xFF);
cdb[9] = (byte)((registers.lbaMid & 0xFF00) >> 8);
cdb[10] = (byte)(registers.lbaMid & 0xFF);
cdb[11] = (byte)((registers.lbaHigh & 0xFF00) >> 8);
cdb[12] = (byte)(registers.lbaHigh & 0xFF);
cdb[13] = registers.deviceHead;
cdb[14] = registers.command;
byte[] senseBuffer;
int error = SendScsiCommand(fd, cdb, ref buffer, out senseBuffer, timeout, AtaProtocolToScsiDirection(protocol), out duration, out sense);
// Now get error registers
byte[] returnCdb = new byte[16];
returnCdb[0] = (byte)Enums.ScsiCommands.AtaPassThrough16;
returnCdb[1] = (byte)(((byte)Enums.AtaProtocol.ReturnResponse << 1) & 0x1E);
byte[] returnBuffer = new byte[14];
bool returnSense;
double returnDuration;
SendScsiCommand(fd, returnCdb, ref returnBuffer, out senseBuffer, timeout, ScsiIoctlDirection.In, out returnDuration, out returnSense);
if (returnBuffer[0] != 0x09 && returnBuffer[1] != 0x0C)
return error;
errorRegisters.error = returnBuffer[3];
errorRegisters.sectorCount = (ushort)((returnBuffer[4] << 8) + returnBuffer[5]);
errorRegisters.lbaLow = (ushort)((returnBuffer[6] << 8) + returnBuffer[7]);
errorRegisters.lbaMid = (ushort)((returnBuffer[8] << 8) + returnBuffer[9]);
errorRegisters.lbaHigh = (ushort)((returnBuffer[10] << 8) + returnBuffer[11]);
errorRegisters.deviceHead = returnBuffer[12];
errorRegisters.status = returnBuffer[13];
sense |= error != 0;
return error;
}
} }
} }

View File

@@ -0,0 +1,114 @@
// /***************************************************************************
// The Disc Image Chef
// ----------------------------------------------------------------------------
//
// Filename : Structs.cs
// Version : 1.0
// Author(s) : Natalia Portillo
//
// Component : Component
//
// Revision : $Revision$
// Last change by : $Author$
// Date : $Date$
//
// --[ Description ] ----------------------------------------------------------
//
// Description
//
// --[ License ] --------------------------------------------------------------
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright (C) 2011-2015 Claunia.com
// ****************************************************************************/
// //$Id$
using System;
namespace DiscImageChef.Devices
{
public static class Structs
{
public struct AtaRegistersCHS
{
public byte feature;
public byte sectorCount;
public byte sector;
public byte cylinderLow;
public byte cylinderHigh;
public byte deviceHead;
public byte command;
}
public struct AtaRegistersLBA28
{
public byte feature;
public byte sectorCount;
public byte lbaLow;
public byte lbaMid;
public byte lbaHigh;
public byte deviceHead;
public byte command;
}
public struct AtaRegistersLBA48
{
public ushort feature;
public ushort sectorCount;
public ushort lbaLow;
public ushort lbaMid;
public ushort lbaHigh;
public byte deviceHead;
public byte command;
}
public struct AtaErrorRegistersCHS
{
public byte status;
public byte error;
public byte sectorCount;
public byte sector;
public byte cylinderLow;
public byte cylinderHigh;
public byte deviceHead;
public byte command;
}
public struct AtaErrorRegistersLBA28
{
public byte status;
public byte error;
public byte sectorCount;
public byte lbaLow;
public byte lbaMid;
public byte lbaHigh;
public byte deviceHead;
public byte command;
}
public struct AtaErrorRegistersLBA48
{
public byte status;
public byte error;
public ushort sectorCount;
public ushort lbaLow;
public ushort lbaMid;
public ushort lbaHigh;
public byte deviceHead;
public byte command;
}
}
}