Added class to safely eject USB drive
This commit is contained in:
parent
ba28d2dc9b
commit
ba6a4581c4
|
|
@ -51,6 +51,7 @@
|
|||
<Reference Include="System.Xml" />
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Compile Include="ejectDrive.cs" />
|
||||
<Compile Include="splashScreen.cs">
|
||||
<SubType>Form</SubType>
|
||||
</Compile>
|
||||
|
|
|
|||
|
|
@ -0,0 +1,349 @@
|
|||
using System;
|
||||
using System.Text;
|
||||
using System.Runtime.InteropServices;
|
||||
using System.Collections;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading;
|
||||
using System.ComponentModel;
|
||||
|
||||
namespace Saviour_Backup_System
|
||||
{
|
||||
|
||||
public class RemoveDriveTools //Taken from: http://www.codeproject.com/Articles/375916/How-to-Prepare-a-USB-Drive-for-Safe-Removal
|
||||
{
|
||||
[StructLayout(LayoutKind.Sequential)]
|
||||
struct STORAGE_DEVICE_NUMBER
|
||||
{
|
||||
public int DeviceType;
|
||||
public int DeviceNumber;
|
||||
public int PartitionNumber;
|
||||
};
|
||||
|
||||
enum DriveType : uint
|
||||
{
|
||||
/// <summary>The drive type cannot be determined.</summary>
|
||||
DRIVE_UNKNOWN = 0, //DRIVE_UNKNOWN
|
||||
/// <summary>The root path is invalid, for example, no volume is mounted at the path.</summary>
|
||||
DRIVE_NO_ROOT_DIR = 1, //DRIVE_NO_ROOT_DIR
|
||||
/// <summary>The drive is a type that has removable media, for example, a floppy drive or removable hard disk.</summary>
|
||||
DRIVE_REMOVABLE = 2, //DRIVE_REMOVABLE
|
||||
/// <summary>The drive is a type that cannot be removed, for example, a fixed hard drive.</summary>
|
||||
DRIVE_FIXED = 3, //DRIVE_FIXED
|
||||
/// <summary>The drive is a remote (network) drive.</summary>
|
||||
DRIVE_REMOTE = 4, //DRIVE_REMOTE
|
||||
/// <summary>The drive is a CD-ROM drive.</summary>
|
||||
DRIVE_CDROM = 5, //DRIVE_CDROM
|
||||
/// <summary>The drive is a RAM disk.</summary>
|
||||
DRIVE_RAMDISK = 6 //DRIVE_RAMDISK
|
||||
}
|
||||
|
||||
const string GUID_DEVINTERFACE_VOLUME = "53f5630d-b6bf-11d0-94f2-00a0c91efb8b";
|
||||
const string GUID_DEVINTERFACE_DISK = "53f56307-b6bf-11d0-94f2-00a0c91efb8b";
|
||||
const string GUID_DEVINTERFACE_FLOPPY = "53f56311-b6bf-11d0-94f2-00a0c91efb8b";
|
||||
const string GUID_DEVINTERFACE_CDROM = "53f56308-b6bf-11d0-94f2-00a0c91efb8b";
|
||||
|
||||
const int INVALID_HANDLE_VALUE = -1;
|
||||
const int GENERIC_READ = unchecked((int)0x80000000);
|
||||
const int GENERIC_WRITE = unchecked((int)0x40000000);
|
||||
const int FILE_SHARE_READ = unchecked((int)0x00000001);
|
||||
const int FILE_SHARE_WRITE = unchecked((int)0x00000002);
|
||||
const int OPEN_EXISTING = unchecked((int)3);
|
||||
const int FSCTL_LOCK_VOLUME = unchecked((int)0x00090018);
|
||||
const int FSCTL_DISMOUNT_VOLUME = unchecked((int)0x00090020);
|
||||
const int IOCTL_STORAGE_EJECT_MEDIA = unchecked((int)0x002D4808);
|
||||
const int IOCTL_STORAGE_MEDIA_REMOVAL = unchecked((int)0x002D4804);
|
||||
const int IOCTL_STORAGE_GET_DEVICE_NUMBER = unchecked((int)0x002D1080);
|
||||
|
||||
const int ERROR_NO_MORE_ITEMS = 259;
|
||||
const int ERROR_INSUFFICIENT_BUFFER = 122;
|
||||
const int ERROR_INVALID_DATA = 13;
|
||||
|
||||
[DllImport("kernel32.dll")]
|
||||
static extern DriveType GetDriveType([MarshalAs(UnmanagedType.LPStr)] string lpRootPathName);
|
||||
|
||||
[DllImport("kernel32.dll")]
|
||||
static extern uint QueryDosDevice(string lpDeviceName, StringBuilder lpTargetPath, int ucchMax);
|
||||
|
||||
[DllImport("kernel32.dll", EntryPoint = "CreateFileW", CharSet = CharSet.Unicode, SetLastError = true)]
|
||||
static extern IntPtr CreateFile(
|
||||
string lpFileName,
|
||||
int dwDesiredAccess,
|
||||
int dwShareMode,
|
||||
IntPtr lpSecurityAttributes,
|
||||
int dwCreationDisposition,
|
||||
int dwFlagsAndAttributes,
|
||||
IntPtr hTemplateFile);
|
||||
|
||||
[DllImport("kernel32.dll", ExactSpelling = true, SetLastError = true)]
|
||||
static extern bool CloseHandle(IntPtr handle);
|
||||
|
||||
[DllImport("kernel32.dll", ExactSpelling = true, SetLastError = true)]
|
||||
static extern bool DeviceIoControl(
|
||||
IntPtr hDevice,
|
||||
int dwIoControlCode,
|
||||
IntPtr lpInBuffer,
|
||||
int nInBufferSize,
|
||||
IntPtr lpOutBuffer,
|
||||
int nOutBufferSize,
|
||||
out int lpBytesReturned,
|
||||
IntPtr lpOverlapped);
|
||||
|
||||
// from setupapi.h
|
||||
const int DIGCF_PRESENT = (0x00000002);
|
||||
const int DIGCF_DEVICEINTERFACE = (0x00000010);
|
||||
|
||||
[StructLayout(LayoutKind.Sequential)]
|
||||
class SP_DEVINFO_DATA
|
||||
{
|
||||
public int cbSize = Marshal.SizeOf(typeof(SP_DEVINFO_DATA));
|
||||
public Guid classGuid = Guid.Empty; // temp
|
||||
public int devInst = 0; // dumy
|
||||
public int reserved = 0;
|
||||
}
|
||||
|
||||
[StructLayout(LayoutKind.Sequential, Pack = 2)]
|
||||
struct SP_DEVICE_INTERFACE_DETAIL_DATA
|
||||
{
|
||||
public int cbSize;
|
||||
public short devicePath;
|
||||
}
|
||||
|
||||
[StructLayout(LayoutKind.Sequential)]
|
||||
class SP_DEVICE_INTERFACE_DATA
|
||||
{
|
||||
public int cbSize = Marshal.SizeOf(typeof(SP_DEVICE_INTERFACE_DATA));
|
||||
public Guid interfaceClassGuid = Guid.Empty; // temp
|
||||
public int flags = 0;
|
||||
public int reserved = 0;
|
||||
}
|
||||
|
||||
[DllImport("setupapi.dll")]
|
||||
static extern IntPtr SetupDiGetClassDevs(
|
||||
ref Guid classGuid,
|
||||
int enumerator,
|
||||
IntPtr hwndParent,
|
||||
int flags);
|
||||
|
||||
[DllImport("setupapi.dll", SetLastError = true, CharSet = CharSet.Auto)]
|
||||
static extern bool SetupDiEnumDeviceInterfaces(
|
||||
IntPtr deviceInfoSet,
|
||||
SP_DEVINFO_DATA deviceInfoData,
|
||||
ref Guid interfaceClassGuid,
|
||||
int memberIndex,
|
||||
SP_DEVICE_INTERFACE_DATA deviceInterfaceData);
|
||||
|
||||
[DllImport("setupapi.dll", SetLastError = true, CharSet = CharSet.Auto)]
|
||||
static extern bool SetupDiGetDeviceInterfaceDetail(
|
||||
IntPtr deviceInfoSet,
|
||||
SP_DEVICE_INTERFACE_DATA deviceInterfaceData,
|
||||
IntPtr deviceInterfaceDetailData,
|
||||
int deviceInterfaceDetailDataSize,
|
||||
ref int requiredSize,
|
||||
SP_DEVINFO_DATA deviceInfoData);
|
||||
|
||||
[DllImport("setupapi.dll")]
|
||||
static extern uint SetupDiDestroyDeviceInfoList(
|
||||
IntPtr deviceInfoSet);
|
||||
|
||||
[DllImport("setupapi.dll")]
|
||||
static extern int CM_Get_Parent(
|
||||
ref int pdnDevInst,
|
||||
int dnDevInst,
|
||||
int ulFlags);
|
||||
|
||||
[DllImport("setupapi.dll")]
|
||||
static extern int CM_Request_Device_Eject(
|
||||
int dnDevInst,
|
||||
out PNP_VETO_TYPE pVetoType,
|
||||
StringBuilder pszVetoName,
|
||||
int ulNameLength,
|
||||
int ulFlags);
|
||||
|
||||
[DllImport("setupapi.dll", EntryPoint = "CM_Request_Device_Eject")]
|
||||
static extern int CM_Request_Device_Eject_NoUi(
|
||||
int dnDevInst,
|
||||
IntPtr pVetoType,
|
||||
StringBuilder pszVetoName,
|
||||
int ulNameLength,
|
||||
int ulFlags);
|
||||
|
||||
enum PNP_VETO_TYPE
|
||||
{
|
||||
Ok,
|
||||
TypeUnknown,
|
||||
LegacyDevice,
|
||||
PendingClose,
|
||||
WindowsApp,
|
||||
WindowsService,
|
||||
OutstandingOpen,
|
||||
Device,
|
||||
Driver,
|
||||
IllegalDeviceRequest,
|
||||
InsufficientPower,
|
||||
NonDisableable,
|
||||
LegacyDriver,
|
||||
InsufficientRights
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Call with "X:" or similar
|
||||
/// </summary>
|
||||
/// <param name="driveCharWithColon"></param>
|
||||
/// <returns></returns>
|
||||
public static bool RemoveDrive(string driveCharWithColon)
|
||||
{
|
||||
// open the storage volume
|
||||
IntPtr hVolume = CreateFile(@"\\.\" + driveCharWithColon, 0, FILE_SHARE_READ | FILE_SHARE_WRITE, IntPtr.Zero, OPEN_EXISTING, 0, IntPtr.Zero);
|
||||
if (hVolume.ToInt32() == -1) return false;
|
||||
|
||||
// get the volume's device number
|
||||
long DeviceNumber = GetDeviceNumber(hVolume);
|
||||
if (DeviceNumber == -1) return false;
|
||||
|
||||
// get the drive type which is required to match the device numbers correctely
|
||||
string rootPath = driveCharWithColon + "\\";
|
||||
DriveType driveType = GetDriveType(rootPath);
|
||||
|
||||
// get the dos device name (like \device\floppy0) to decide if it's a floppy or not - who knows a better way?
|
||||
StringBuilder pathInformation = new StringBuilder(250);
|
||||
uint res = QueryDosDevice(driveCharWithColon, pathInformation, 250);
|
||||
if (res == 0) return false;
|
||||
|
||||
// get the device instance handle of the storage volume by means of a SetupDi enum and matching the device number
|
||||
long DevInst = GetDrivesDevInstByDeviceNumber(DeviceNumber, driveType, pathInformation.ToString());
|
||||
if (DevInst == 0) return false;
|
||||
|
||||
// get drives's parent, e.g. the USB bridge, the SATA port, an IDE channel with two drives!
|
||||
int DevInstParent = 0;
|
||||
CM_Get_Parent(ref DevInstParent, (int)DevInst, 0);
|
||||
|
||||
for (int tries = 1; tries <= 3; tries++) // sometimes we need some tries...
|
||||
{
|
||||
int r = CM_Request_Device_Eject_NoUi(DevInstParent, IntPtr.Zero, null, 0, 0);
|
||||
if (r == 0) return true;
|
||||
Thread.Sleep(500);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
static long GetDeviceNumber(IntPtr handle)
|
||||
{
|
||||
// get the volume's device number
|
||||
long DeviceNumber = -1;
|
||||
int size = 0x400; // some big size
|
||||
IntPtr buffer = Marshal.AllocHGlobal(size);
|
||||
int bytesReturned = 0;
|
||||
|
||||
try
|
||||
{
|
||||
DeviceIoControl(handle, IOCTL_STORAGE_GET_DEVICE_NUMBER, IntPtr.Zero, 0, buffer, size, out bytesReturned, IntPtr.Zero);
|
||||
}
|
||||
finally
|
||||
{
|
||||
CloseHandle(handle);
|
||||
}
|
||||
|
||||
if (bytesReturned > 0)
|
||||
{
|
||||
STORAGE_DEVICE_NUMBER sdn = (STORAGE_DEVICE_NUMBER)Marshal.PtrToStructure(buffer, typeof(STORAGE_DEVICE_NUMBER));
|
||||
DeviceNumber = sdn.DeviceNumber;
|
||||
}
|
||||
Marshal.FreeHGlobal(buffer);
|
||||
|
||||
return DeviceNumber;
|
||||
}
|
||||
|
||||
|
||||
//----------------------------------------------------------------------
|
||||
// returns the device instance handle of a storage volume or 0 on error
|
||||
//----------------------------------------------------------------------
|
||||
static long GetDrivesDevInstByDeviceNumber(long DeviceNumber, DriveType DriveType, string dosDeviceName)
|
||||
{
|
||||
bool IsFloppy = dosDeviceName.Contains("\\Floppy"); // who knows a better way?
|
||||
Guid guid;
|
||||
|
||||
switch (DriveType)
|
||||
{
|
||||
case DriveType.DRIVE_REMOVABLE:
|
||||
if (IsFloppy) guid = new Guid(GUID_DEVINTERFACE_FLOPPY);
|
||||
else guid = new Guid(GUID_DEVINTERFACE_DISK);
|
||||
break;
|
||||
case DriveType.DRIVE_FIXED:
|
||||
guid = new Guid(GUID_DEVINTERFACE_DISK);
|
||||
break;
|
||||
case DriveType.DRIVE_CDROM:
|
||||
guid = new Guid(GUID_DEVINTERFACE_CDROM);
|
||||
break;
|
||||
default:
|
||||
return 0;
|
||||
}
|
||||
|
||||
// Get device interface info set handle for all devices attached to system
|
||||
IntPtr hDevInfo = SetupDiGetClassDevs(ref guid, 0, IntPtr.Zero, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE);
|
||||
|
||||
if (hDevInfo.ToInt32() == INVALID_HANDLE_VALUE) throw new Win32Exception(Marshal.GetLastWin32Error());
|
||||
|
||||
// Retrieve a context structure for a device interface of a device information set
|
||||
int dwIndex = 0;
|
||||
|
||||
while (true)
|
||||
{
|
||||
SP_DEVICE_INTERFACE_DATA interfaceData = new SP_DEVICE_INTERFACE_DATA();
|
||||
if (!SetupDiEnumDeviceInterfaces(hDevInfo, null, ref guid, dwIndex, interfaceData))
|
||||
{
|
||||
int error = Marshal.GetLastWin32Error();
|
||||
if (error != ERROR_NO_MORE_ITEMS) throw new Win32Exception(error);
|
||||
break;
|
||||
}
|
||||
|
||||
SP_DEVINFO_DATA devData = new SP_DEVINFO_DATA();
|
||||
int size = 0;
|
||||
if (!SetupDiGetDeviceInterfaceDetail(hDevInfo, interfaceData, IntPtr.Zero, 0, ref size, devData))
|
||||
{
|
||||
int error = Marshal.GetLastWin32Error();
|
||||
if (error != ERROR_INSUFFICIENT_BUFFER) throw new Win32Exception(error);
|
||||
}
|
||||
|
||||
IntPtr buffer = Marshal.AllocHGlobal(size);
|
||||
SP_DEVICE_INTERFACE_DETAIL_DATA detailData = new SP_DEVICE_INTERFACE_DETAIL_DATA();
|
||||
detailData.cbSize = Marshal.SizeOf(typeof(SP_DEVICE_INTERFACE_DETAIL_DATA));
|
||||
Marshal.StructureToPtr(detailData, buffer, false);
|
||||
|
||||
if (!SetupDiGetDeviceInterfaceDetail(hDevInfo, interfaceData, buffer, size, ref size, devData))
|
||||
{
|
||||
Marshal.FreeHGlobal(buffer);
|
||||
throw new Win32Exception(Marshal.GetLastWin32Error());
|
||||
}
|
||||
|
||||
IntPtr pDevicePath = (IntPtr)((int)buffer + Marshal.SizeOf(typeof(int)));
|
||||
string devicePath = Marshal.PtrToStringAuto(pDevicePath);
|
||||
Marshal.FreeHGlobal(buffer);
|
||||
|
||||
// open the disk or cdrom or floppy
|
||||
IntPtr hDrive = CreateFile(devicePath, 0, FILE_SHARE_READ | FILE_SHARE_WRITE, IntPtr.Zero, OPEN_EXISTING, 0, IntPtr.Zero);
|
||||
if (hDrive.ToInt32() != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
// get its device number
|
||||
long driveDeviceNumber = GetDeviceNumber(hDrive);
|
||||
if (DeviceNumber == driveDeviceNumber) // match the given device number with the one of the current device
|
||||
{
|
||||
// CloseHandle( hDrive ); // handle hDrive was closed inside GetDeviceNumber(..)
|
||||
SetupDiDestroyDeviceInfoList(hDevInfo);
|
||||
return devData.devInst;
|
||||
}
|
||||
// CloseHandle( hDrive ); <span style="font-size: 9pt;">// handle hDrive was closed inside GetDeviceNumber(..)</span>
|
||||
|
||||
}
|
||||
dwIndex++;
|
||||
}
|
||||
|
||||
SetupDiDestroyDeviceInfoList(hDevInfo);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void Test()
|
||||
{
|
||||
bool ok = RemoveDrive("H:");
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue