如何识别移动硬盘

Ring 3下的方法：
童鞋们应该已经发现了，GetDriveType函数只能识别出U盘，无法识别移动硬盘。移动硬盘和本地硬盘将返回相同的类型DRIVE_FIXED。MSDN上的建议是用一个蛋疼无比的函数SetupDiGetDeviceRegistryProperty来识别USB设备。

其实有个更方便的方法，可以通过DeviceIoControl函数查询总线类型来识别USB设备。贴代码：

BOOL IsUsbDriver(TCHAR tchDrvName) 

/*++

Routine Description:

    判断某分区是否是USB设备。



Arguments:

    tchDrvName - 输入参数，驱动器号。



Return Value:

    是USB设备则返回TRUE，否则返回FALSE，

    查询失败也返回FALSE。



Author:

    Fypher

    http://hi.baidu.com/nmn714

    2011/5/1

--*/ 

{ 

TCHAR tcsDrvName[8] = TEXT("////.//X:"); 

tcsDrvName[4] = tchDrvName; 

HANDLE hDevice = CreateFile( tcsDrvName,

                                 GENERIC_READ,

                                 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL,

                                 OPEN_EXISTING,

                                 FILE_ATTRIBUTE_NORMAL, NULL ); 

if ( hDevice == INVALID_HANDLE_VALUE ) 

{ 

　　return FALSE; 

} 

STORAGE_PROPERTY_QUERY StoragePropertyQuery; 

StoragePropertyQuery.PropertyId = StorageDeviceProperty; 

StoragePropertyQuery.QueryType = PropertyStandardQuery; 

BYTE buff[1024] = {0}; 

PSTORAGE_DEVICE_DESCRIPTOR pDevDesc = (PSTORAGE_DEVICE_DESCRIPTOR)buff; 

pDevDesc->Size = sizeof(buff); 

DWORD dwOutLen; 

BOOL res = DeviceIoControl( hDevice,

                                IOCTL_STORAGE_QUERY_PROPERTY, &StoragePropertyQuery, sizeof(STORAGE_PROPERTY_QUERY),

                                pDevDesc,

                                pDevDesc->Size, &dwOutLen, NULL ); 

CloseHandle(hDevice); 



if (res) { // 返回结果 

　　return pDevDesc->BusType == BusTypeUsb; 

} 

else

{ // 查询失败 

　　return FALSE; 

} 

} 

Ring 0下的方法:

在Ring 0下，通过DiskDeviceObject的名字在一定概率上能判断该DiskDeviceObject对应的卷设备是否是USB设备。即名称为HardDiskVolumeXXX的DiskDeviceObject通常是本地硬盘。不过这个方法其实也很不靠谱，尤其是Win7下，插个U盘都是HardDiskVolumeXXX。

其实最好的方法同样是查询总线类型。贴代码：

BOOLEAN IsUsbDriver(IN PDEVICE_OBJECT pDeviceObject) 

/*++

Routine Description:

    判断某磁盘设备是否是USB设备。



Arguments:

    pDeviceObject - 输入参数，卷设备对应的 DiskDeviceObject。

                       卷设备与 DiskDeviceObject 之间的关系可以参考：

                       http://hi.baidu.com/nmn714/blog/item/3d2ccccad529e994c8176842.html



Return Value:

    是USB设备则返回TRUE，否则返回FALSE，

    查询失败也返回FALSE。

--*/ 

{ 

PIRP pNewIrp; 

PSTORAGE_DEVICE_DESCRIPTOR pStorageDeviceDescriptor; 

STORAGE_PROPERTY_QUERY StoragePropertyQuery; 

CHAR buff[128]; 

KEVENT WaitEvent; 

NTSTATUS Status; 

IO_STATUS_BLOCK IoStatus; 

StoragePropertyQuery.PropertyId = StorageDeviceProperty; 

StoragePropertyQuery.QueryType = PropertyStandardQuery; 

KeInitializeEvent(&WaitEvent, NotificationEvent, FALSE); 

pNewIrp = IoBuildDeviceIoControlRequest( IOCTL_STORAGE_QUERY_PROPERTY,pDeviceObject, (PVOID)&StoragePropertyQuery, sizeof(StoragePropertyQuery), (PVOID)buff, sizeof(buff), FALSE, &WaitEvent, &IoStatus ); 

if (!pNewIrp) 

{

　　 // 创建 IRP 失败 

　　return FALSE; 

} 

　　Status = IoCallDriver(pDeviceObject, pNewIrp); 

if (Status == STATUS_PENDING) 

{ 

　　Status = KeWaitForSingleObject(&WaitEvent, Executive, KernelMode, FALSE, NULL); 

　　Status = IoStatus.Status; 

} 

if (!NT_SUCCESS(Status)) 

{ 　　// 查询失败 

　　return FALSE; 

} 

　　pStorageDeviceDescriptor = (PSTORAGE_DEVICE_DESCRIPTOR)buff; // 返回结果 

　　return pStorageDeviceDescriptor->BusType == BusTypeUsb; 

}