mirror of
https://github.com/aaru-dps/Aaru.git
synced 2025-12-16 19:24:25 +00:00
409 lines
14 KiB
C#
409 lines
14 KiB
C#
// /***************************************************************************
|
|
// Aaru Data Preservation Suite
|
|
// ----------------------------------------------------------------------------
|
|
//
|
|
// Filename : Device.cs
|
|
// Author(s) : Natalia Portillo <claunia@claunia.com>
|
|
//
|
|
// Component : Windows direct device access.
|
|
//
|
|
// --[ Description ] ----------------------------------------------------------
|
|
//
|
|
// Prepares a Windows device for direct access.
|
|
//
|
|
// --[ License ] --------------------------------------------------------------
|
|
//
|
|
// This library is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as
|
|
// published by the Free Software Foundation; either version 2.1 of the
|
|
// License, or (at your option) any later version.
|
|
//
|
|
// This library 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
|
|
// Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public
|
|
// License along with this library; if not, see <http://www.gnu.org/licenses/>.
|
|
//
|
|
// ----------------------------------------------------------------------------
|
|
// Copyright © 2011-2025 Natalia Portillo
|
|
// ****************************************************************************/
|
|
|
|
using System;
|
|
using System.Runtime.InteropServices;
|
|
using System.Runtime.Versioning;
|
|
using Aaru.CommonTypes.Enums;
|
|
using Aaru.CommonTypes.Interop;
|
|
using Aaru.CommonTypes.Structs.Devices.SCSI;
|
|
using Aaru.Decoders.SecureDigital;
|
|
using Microsoft.Win32.SafeHandles;
|
|
|
|
namespace Aaru.Devices.Windows;
|
|
|
|
/// <inheritdoc />
|
|
[SupportedOSPlatform("windows")]
|
|
partial class Device : Devices.Device, IDisposable
|
|
{
|
|
private const nuint ALIGNMENT = 64;
|
|
private nuint _capacity;
|
|
/// <summary>Gets the file handle representing this device</summary>
|
|
/// <value>The file handle</value>
|
|
SafeFileHandle _fileHandle;
|
|
private nuint _nativeBuffer;
|
|
|
|
Device() {}
|
|
|
|
#region IDisposable Members
|
|
|
|
public new unsafe void Dispose()
|
|
{
|
|
if(_nativeBuffer == 0) return;
|
|
NativeMemory.AlignedFree((void*)_nativeBuffer);
|
|
_nativeBuffer = 0;
|
|
_capacity = 0;
|
|
}
|
|
|
|
#endregion
|
|
|
|
private unsafe void EnsureCapacityAligned(nuint size)
|
|
{
|
|
if(size <= _capacity) return;
|
|
|
|
if(_nativeBuffer != 0) NativeMemory.AlignedFree((void*)_nativeBuffer);
|
|
|
|
_nativeBuffer = (nuint)NativeMemory.AlignedAlloc(size, ALIGNMENT);
|
|
|
|
_capacity = size;
|
|
}
|
|
|
|
internal new static Device Create(string devicePath, out ErrorNumber errno)
|
|
{
|
|
errno = ErrorNumber.NoError;
|
|
|
|
var dev = new Device
|
|
{
|
|
PlatformId = DetectOS.GetRealPlatformID(),
|
|
Timeout = 15,
|
|
Error = false,
|
|
IsRemovable = false,
|
|
_fileHandle = Extern.CreateFile(devicePath,
|
|
FileAccess.GenericRead | FileAccess.GenericWrite,
|
|
FileShare.Read | FileShare.Write,
|
|
IntPtr.Zero,
|
|
FileMode.OpenExisting,
|
|
FileAttributes.Normal,
|
|
IntPtr.Zero)
|
|
};
|
|
|
|
if(dev._fileHandle.IsInvalid)
|
|
{
|
|
dev.Error = true;
|
|
dev.LastError = Marshal.GetLastWin32Error();
|
|
}
|
|
|
|
if(dev.Error)
|
|
{
|
|
errno = (ErrorNumber)dev.LastError;
|
|
|
|
return null;
|
|
}
|
|
|
|
dev.Type = DeviceType.Unknown;
|
|
dev.ScsiType = PeripheralDeviceTypes.UnknownDevice;
|
|
|
|
if(dev.Error)
|
|
{
|
|
errno = (ErrorNumber)dev.LastError;
|
|
|
|
return null;
|
|
}
|
|
|
|
// Windows is answering SCSI INQUIRY for all device types so it needs to be detected first
|
|
var query = new StoragePropertyQuery
|
|
{
|
|
PropertyId = StoragePropertyId.Device,
|
|
QueryType = StorageQueryType.Standard,
|
|
AdditionalParameters = new byte[1]
|
|
};
|
|
|
|
IntPtr descriptorPtr = Marshal.AllocHGlobal(1000);
|
|
byte[] descriptorB = new byte[1000];
|
|
|
|
uint returned = 0;
|
|
int error = 0;
|
|
|
|
bool hasError = !Extern.DeviceIoControlStorageQuery(dev._fileHandle,
|
|
WindowsIoctl.IoctlStorageQueryProperty,
|
|
ref query,
|
|
(uint)Marshal.SizeOf(query),
|
|
descriptorPtr,
|
|
1000,
|
|
ref returned,
|
|
IntPtr.Zero);
|
|
|
|
if(hasError) error = Marshal.GetLastWin32Error();
|
|
|
|
Marshal.Copy(descriptorPtr, descriptorB, 0, 1000);
|
|
|
|
if(!hasError && error == 0)
|
|
{
|
|
var descriptor = new StorageDeviceDescriptor
|
|
{
|
|
Version = BitConverter.ToUInt32(descriptorB, 0),
|
|
Size = BitConverter.ToUInt32(descriptorB, 4),
|
|
DeviceType = descriptorB[8],
|
|
DeviceTypeModifier = descriptorB[9],
|
|
RemovableMedia = descriptorB[10] > 0,
|
|
CommandQueueing = descriptorB[11] > 0,
|
|
VendorIdOffset = BitConverter.ToInt32(descriptorB, 12),
|
|
ProductIdOffset = BitConverter.ToInt32(descriptorB, 16),
|
|
ProductRevisionOffset = BitConverter.ToInt32(descriptorB, 20),
|
|
SerialNumberOffset = BitConverter.ToInt32(descriptorB, 24),
|
|
BusType = (StorageBusType)BitConverter.ToUInt32(descriptorB, 28),
|
|
RawPropertiesLength = BitConverter.ToUInt32(descriptorB, 32)
|
|
};
|
|
|
|
descriptor.RawDeviceProperties = new byte[descriptor.RawPropertiesLength];
|
|
|
|
Array.Copy(descriptorB, 36, descriptor.RawDeviceProperties, 0, descriptor.RawPropertiesLength);
|
|
|
|
switch(descriptor.BusType)
|
|
{
|
|
case StorageBusType.SCSI:
|
|
case StorageBusType.SSA:
|
|
case StorageBusType.Fibre:
|
|
case StorageBusType.iSCSI:
|
|
case StorageBusType.SAS:
|
|
dev.Type = DeviceType.SCSI;
|
|
|
|
break;
|
|
case StorageBusType.FireWire:
|
|
dev.IsFireWire = true;
|
|
dev.Type = DeviceType.SCSI;
|
|
|
|
break;
|
|
case StorageBusType.USB:
|
|
dev.IsUsb = true;
|
|
dev.Type = DeviceType.SCSI;
|
|
|
|
break;
|
|
case StorageBusType.ATAPI:
|
|
dev.Type = DeviceType.ATAPI;
|
|
|
|
break;
|
|
case StorageBusType.ATA:
|
|
case StorageBusType.SATA:
|
|
dev.Type = DeviceType.ATA;
|
|
|
|
break;
|
|
case StorageBusType.MultiMediaCard:
|
|
dev.Type = DeviceType.MMC;
|
|
|
|
break;
|
|
case StorageBusType.SecureDigital:
|
|
dev.Type = DeviceType.SecureDigital;
|
|
|
|
break;
|
|
case StorageBusType.NVMe:
|
|
dev.Type = DeviceType.NVMe;
|
|
|
|
break;
|
|
}
|
|
|
|
switch(dev.Type)
|
|
{
|
|
case DeviceType.ATA:
|
|
bool atapiSense = dev.AtapiIdentify(out byte[] _, out _);
|
|
|
|
if(!atapiSense)
|
|
dev.Type = DeviceType.ATAPI;
|
|
else
|
|
dev.Manufacturer = "ATA";
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
Marshal.FreeHGlobal(descriptorPtr);
|
|
|
|
if(IsSdhci(dev._fileHandle))
|
|
{
|
|
byte[] sdBuffer = new byte[16];
|
|
|
|
dev.LastError = dev.SendMmcCommand(MmcCommands.SendCsd,
|
|
false,
|
|
false,
|
|
MmcFlags.ResponseSpiR2 | MmcFlags.ResponseR2 | MmcFlags.CommandAc,
|
|
0,
|
|
16,
|
|
1,
|
|
ref sdBuffer,
|
|
out _,
|
|
out _,
|
|
out bool sense);
|
|
|
|
if(!sense)
|
|
{
|
|
dev.CachedCsd = new byte[16];
|
|
Array.Copy(sdBuffer, 0, dev.CachedCsd, 0, 16);
|
|
}
|
|
|
|
sdBuffer = new byte[16];
|
|
|
|
dev.LastError = dev.SendMmcCommand(MmcCommands.SendCid,
|
|
false,
|
|
false,
|
|
MmcFlags.ResponseSpiR2 | MmcFlags.ResponseR2 | MmcFlags.CommandAc,
|
|
0,
|
|
16,
|
|
1,
|
|
ref sdBuffer,
|
|
out _,
|
|
out _,
|
|
out sense);
|
|
|
|
if(!sense)
|
|
{
|
|
dev.CachedCid = new byte[16];
|
|
Array.Copy(sdBuffer, 0, dev.CachedCid, 0, 16);
|
|
}
|
|
|
|
sdBuffer = new byte[8];
|
|
|
|
dev.LastError = dev.SendMmcCommand((MmcCommands)SecureDigitalCommands.SendScr,
|
|
false,
|
|
true,
|
|
MmcFlags.ResponseSpiR1 | MmcFlags.ResponseR1 | MmcFlags.CommandAdtc,
|
|
0,
|
|
8,
|
|
1,
|
|
ref sdBuffer,
|
|
out _,
|
|
out _,
|
|
out sense);
|
|
|
|
if(!sense)
|
|
{
|
|
dev.CachedScr = new byte[8];
|
|
Array.Copy(sdBuffer, 0, dev.CachedScr, 0, 8);
|
|
}
|
|
|
|
sdBuffer = new byte[4];
|
|
|
|
dev.LastError =
|
|
dev.SendMmcCommand(dev.CachedScr != null
|
|
? (MmcCommands)SecureDigitalCommands.SendOperatingCondition
|
|
: MmcCommands.SendOpCond,
|
|
false,
|
|
true,
|
|
MmcFlags.ResponseSpiR3 | MmcFlags.ResponseR3 | MmcFlags.CommandBcr,
|
|
0,
|
|
4,
|
|
1,
|
|
ref sdBuffer,
|
|
out _,
|
|
out _,
|
|
out sense);
|
|
|
|
if(!sense)
|
|
{
|
|
dev.CachedScr = new byte[4];
|
|
Array.Copy(sdBuffer, 0, dev.CachedScr, 0, 4);
|
|
}
|
|
}
|
|
|
|
#region SecureDigital / MultiMediaCard
|
|
|
|
if(dev.CachedCid != null)
|
|
{
|
|
dev.ScsiType = PeripheralDeviceTypes.DirectAccess;
|
|
dev.IsRemovable = false;
|
|
|
|
if(dev.CachedScr != null)
|
|
{
|
|
dev.Type = DeviceType.SecureDigital;
|
|
CID decoded = Decoders.SecureDigital.Decoders.DecodeCID(dev.CachedCid);
|
|
dev.Manufacturer = VendorString.Prettify(decoded.Manufacturer);
|
|
dev.Model = decoded.ProductName;
|
|
|
|
dev.FirmwareRevision =
|
|
$"{(decoded.ProductRevision & 0xF0) >> 4:X2}.{decoded.ProductRevision & 0x0F:X2}";
|
|
|
|
dev.Serial = $"{decoded.ProductSerialNumber}";
|
|
}
|
|
else
|
|
{
|
|
dev.Type = DeviceType.MMC;
|
|
Decoders.MMC.CID decoded = Decoders.MMC.Decoders.DecodeCID(dev.CachedCid);
|
|
dev.Manufacturer = Decoders.MMC.VendorString.Prettify(decoded.Manufacturer);
|
|
dev.Model = decoded.ProductName;
|
|
|
|
dev.FirmwareRevision =
|
|
$"{(decoded.ProductRevision & 0xF0) >> 4:X2}.{decoded.ProductRevision & 0x0F:X2}";
|
|
|
|
dev.Serial = $"{decoded.ProductSerialNumber}";
|
|
}
|
|
|
|
return dev;
|
|
}
|
|
|
|
#endregion SecureDigital / MultiMediaCard
|
|
|
|
#region USB
|
|
|
|
Usb.UsbDevice usbDevice = null;
|
|
|
|
// I have to search for USB disks, floppies and CD-ROMs as separate device types
|
|
foreach(string devGuid in new[]
|
|
{
|
|
Usb.GUID_DEVINTERFACE_FLOPPY, Usb.GUID_DEVINTERFACE_CDROM, Usb.GUID_DEVINTERFACE_DISK,
|
|
Usb.GUID_DEVINTERFACE_TAPE
|
|
})
|
|
{
|
|
usbDevice = Usb.FindDrivePath(devicePath, devGuid);
|
|
|
|
if(usbDevice != null) break;
|
|
}
|
|
|
|
if(usbDevice != null)
|
|
{
|
|
dev.UsbDescriptors = usbDevice.BinaryDescriptors;
|
|
dev.UsbVendor = (ushort)usbDevice.DeviceDescriptor.idVendor;
|
|
dev.UsbProduct = (ushort)usbDevice.DeviceDescriptor.idProduct;
|
|
dev.UsbManufacturerString = usbDevice.Manufacturer;
|
|
dev.UsbProductString = usbDevice.Product;
|
|
|
|
dev.UsbSerialString =
|
|
usbDevice.SerialNumber; // This is incorrect filled by Windows with SCSI/ATA serial number
|
|
}
|
|
|
|
#endregion USB
|
|
|
|
#region FireWire
|
|
|
|
// TODO: Implement
|
|
|
|
dev.IsFireWire = false;
|
|
|
|
#endregion FireWire
|
|
|
|
#region PCMCIA
|
|
|
|
// TODO: Implement
|
|
|
|
#endregion PCMCIA
|
|
|
|
return dev;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override void Close()
|
|
{
|
|
if(_fileHandle == null) return;
|
|
|
|
_fileHandle?.Close();
|
|
|
|
_fileHandle = null;
|
|
}
|
|
} |