磁盘设备类型获取函数

//将该部分内容保存到.cpp文件中可直接编译运行，用于辨别驱动器的类型！

#define MEDIA_INFO_SIZE    sizeof(GET_MEDIA_TYPES)+15*sizeof(DEVICE_MEDIA_INFO)
#define IOCTL_STORAGE_QUERY_PROPERTY   CTL_CODE(IOCTL_STORAGE_BASE, 0x0500, METHOD_BUFFERED, FILE_ANY_ACCESS)

typedef enum _STORAGE_BUS_TYPE { 
 BusTypeUnknown = 0x00, 
  BusTypeScsi, 
  BusTypeAtapi, 
  BusTypeAta, 
  BusType1394, 
  BusTypeSsa, 
  BusTypeFibre, 
  BusTypeUsb, 
  BusTypeRAID, 
  BusTypeiScsi, 
  BusTypeSas, 
  BusTypeSata, 
  BusTypeSd, 
  BusTypeMmc, 
  BusTypeMax, 
  BusTypeMaxReserved = 0x7F
} STORAGE_BUS_TYPE, *PSTORAGE_BUS_TYPE;

typedef enum _STORAGE_PROPERTY_ID {
 StorageDeviceProperty = 0,
  StorageAdapterProperty,
  StorageDeviceIdProperty
} STORAGE_PROPERTY_ID, *PSTORAGE_PROPERTY_ID;

typedef enum _STORAGE_QUERY_TYPE {
 PropertyStandardQuery = 0,
  PropertyExistsQuery,
  PropertyMaskQuery,
  PropertyQueryMaxDefined
} STORAGE_QUERY_TYPE, *PSTORAGE_QUERY_TYPE;

typedef struct _STORAGE_PROPERTY_QUERY {
 STORAGE_PROPERTY_ID  PropertyId;
 STORAGE_QUERY_TYPE  QueryType;
 unsigned char  AdditionalParameters[1];
} STORAGE_PROPERTY_QUERY, *PSTORAGE_PROPERTY_QUERY;

typedef struct _STORAGE_DEVICE_DESCRIPTOR {
 unsigned long  Version;
 unsigned long  Size;
 unsigned char DeviceType;
 unsigned char DeviceTypeModifier;
 unsigned char  RemovableMedia;
 unsigned char  CommandQueueing;
 unsigned long  VendorIdOffset;
 unsigned long  ProductIdOffset;
 unsigned long  ProductRevisionOffset;
 unsigned long  SerialNumberOffset;
 STORAGE_BUS_TYPE  BusType;
 unsigned long  RawPropertiesLength;
 unsigned char RawDeviceProperties[1];
} STORAGE_DEVICE_DESCRIPTOR, *PSTORAGE_DEVICE_DESCRIPTOR;

#include <windows.h>
#include <winioctl.h>
#include <iostream.h>
#include <string.h>
#include <tchar.h>

BOOL GetDriveGeometry(const TCHAR * filename, DISK_GEOMETRY * pdg)
{
 HANDLE hDevice;         // 设备句柄
 BOOL bResult;           // DeviceIoControl的返回结果
 GET_MEDIA_TYPES *pmt;   // 内部用的输出缓冲区
 DWORD dwOutBytes;       // 输出数据长度
 
 // 打开设备
 hDevice = ::CreateFile(filename,           // 文件名
  GENERIC_READ,                          // 软驱需要读盘
  FILE_SHARE_READ | FILE_SHARE_WRITE,    // 共享方式
  NULL,                                  // 默认的安全描述符
  OPEN_EXISTING,                         // 创建方式
  0,                                     // 不需设置文件属性
  NULL);                                 // 不需参照模板文件
 
 if (hDevice == INVALID_HANDLE_VALUE)
 {
  // 设备无法打开...
  return FALSE;
 }
 
 // 用IOCTL_DISK_GET_DRIVE_GEOMETRY取磁盘参数
 bResult = ::DeviceIoControl(hDevice,       // 设备句柄
  IOCTL_DISK_GET_DRIVE_GEOMETRY,         // 取磁盘参数
  NULL, 0,                               // 不需要输入数据
  pdg, sizeof(DISK_GEOMETRY),            // 输出数据缓冲区
  &dwOutBytes,                           // 输出数据长度
  (LPOVERLAPPED)NULL);                   // 用同步I/O
 
 // 如果失败，再用IOCTL_STORAGE_GET_MEDIA_TYPES_EX取介质类型参数
 if (!bResult)
 {
  pmt = (GET_MEDIA_TYPES *)new BYTE[MEDIA_INFO_SIZE];
  
  bResult = ::DeviceIoControl(hDevice,    // 设备句柄
   IOCTL_STORAGE_GET_MEDIA_TYPES_EX,   // 取介质类型参数
   NULL, 0,                            // 不需要输入数据
   pmt, MEDIA_INFO_SIZE,               // 输出数据缓冲区
   &dwOutBytes,                        // 输出数据长度
   (LPOVERLAPPED)NULL);                // 用同步I/O
  
  if (bResult)
  {
   // 注意到结构DEVICE_MEDIA_INFO是在结构DISK_GEOMETRY的基础上扩充的
   // 为简化程序，用memcpy代替如下多条赋值语句：
   // pdg->MediaType = (MEDIA_TYPE)pmt->MediaInfo[0].DeviceSpecific.DiskInfo.MediaType;
   // pdg->Cylinders = pmt->MediaInfo[0].DeviceSpecific.DiskInfo.Cylinders;
   // pdg->TracksPerCylinder = pmt->MediaInfo[0].DeviceSpecific.DiskInfo.TracksPerCylinder;
   // ... ...
   ::memcpy(pdg, pmt->MediaInfo, sizeof(DISK_GEOMETRY));
  }
  
  delete pmt;
 }
 
 // 关闭设备句柄
 ::CloseHandle(hDevice);
 
 return (bResult);
 
}

int main()
{
 TCHAR szBuf[256];
 memset(szBuf,0,sizeof(szBuf));
 DISK_GEOMETRY dg;
 TCHAR sztmpPath[100] = _T(".//");
 TCHAR szPath[100] = _T(".//");
 HANDLE hDevice;         // 设备句柄
 BOOL bResult;           // DeviceIoControl的返回结果
 STORAGE_PROPERTY_QUERY Query; // input param for query
 DWORD dwOutBytes; // IOCTL output length
 STORAGE_DEVICE_DESCRIPTOR pDevDesc;

 DWORD len = GetLogicalDriveStrings(sizeof(szBuf)/sizeof(TCHAR),szBuf);
 
 for (TCHAR* s = szBuf; *s; s += strlen(s)+1)
 {
  LPCTSTR sDrivePath = s;
  memcpy(szPath,sztmpPath,sizeof(szPath));
  ::_tcscat(szPath,sDrivePath); 
  int nSize = ::_tcslen(szPath);
  szPath[nSize-1] = '/0';  
        BOOL bResult = GetDriveGeometry(szPath,&dg);
  
  // 打开设备
  hDevice = ::CreateFile(szPath,             // 文件名
   GENERIC_READ,                          // 软驱需要读盘
   FILE_SHARE_READ | FILE_SHARE_WRITE,    // 共享方式
   NULL,                                  // 默认的安全描述符
   OPEN_EXISTING,                         // 创建方式
   0,                                     // 不需设置文件属性
   NULL);   
  if (hDevice == INVALID_HANDLE_VALUE)
  {
   return FALSE;
  }
  
  
  Query.PropertyId = StorageDeviceProperty;
  Query.QueryType = PropertyStandardQuery;
    
  pDevDesc.Size = sizeof(STORAGE_DEVICE_DESCRIPTOR);

  
  // 用 IOCTL_STORAGE_QUERY_PROPERTY  
  bResult = ::DeviceIoControl(hDevice, // device handle
   IOCTL_STORAGE_QUERY_PROPERTY, // info of device property
   &Query, sizeof(STORAGE_PROPERTY_QUERY), // input data buffer
   &pDevDesc, pDevDesc.Size, // output data buffer
   &dwOutBytes, // out's length
   (LPOVERLAPPED)NULL);

  if (bResult == FALSE)
  {
   CloseHandle(hDevice);
   return 0;
  }
  
  UINT Type = pDevDesc.BusType;
  switch(Type)
  {
  case    BusTypeUnknown:
   cout<<sDrivePath<<"  未知设备"<<'/n';
   break;

  case    BusTypeScsi:
   cout<<sDrivePath<<"  SCSI设备"<<'/n';
   break;
   
  case    BusTypeAtapi:
   cout<<sDrivePath<<"  API设备"<<'/n';
   break;
   
  case  BusTypeAta:
   cout<<sDrivePath<<"  ATA设备"<<'/n';
   break;
   
  case  BusType1394:
   cout<<sDrivePath<<"  1394设备"<<'/n';
   break;
   
  case BusTypeSsa:
   cout<<sDrivePath<<"  SSA设备"<<'/n';
   break;
   
  case  BusTypeFibre:
   cout<<sDrivePath<<"  光纤设备"<<'/n';
   break;
   
  case  BusTypeUsb:
   cout<<sDrivePath<<"  USB设备"<<'/n';
   break;
   
  case  BusTypeRAID:
   cout<<sDrivePath<<"  RAID设备"<<'/n';
   break;
   
  case BusTypeiScsi:
   cout<<sDrivePath<<"  iSCSI设备"<<'/n';
   break;
   
  case BusTypeSas:
   cout<<sDrivePath<<"  SAS设备"<<'/n';
   break;
   
  case BusTypeSata:
   cout<<sDrivePath<<"  SATA设备"<<'/n';
   break;
   
  case  BusTypeSd:
   cout<<sDrivePath<<"  SD设备"<<'/n';
   break;
   
  case BusTypeMmc:
   cout<<sDrivePath<<"  MMC设备"<<'/n';
   break;
   
  case BusTypeMax:
   cout<<sDrivePath<<"  MAX设备"<<'/n';
   break;
   
  case  BusTypeMaxReserved:
   cout<<sDrivePath<<"  ...设备"<<'/n';
   break;
  default:
   break;
  }
  CloseHandle(hDevice);
 }
};

欢迎大家加入"数据恢复技术"QQ群：30481379 

        本QQ群主要研究分析NTFS、EXT2/3、ResierFS等文件系统及数据恢复技术，同时对各种系统的磁盘阵列恢复的阵列参数智能分析算法进行技术交流，欢迎有志同道合者加入！

         由于本群主要面向数据恢复的技术研究，同时起点要高于一般的数据恢复技术，所以希望加入者在加入本群以前已经有关于这方面技术的积累会相关研究经验，希望大家能理解！

本文来自CSDN博客，转载请标明出处：http://blog.csdn.net/ddl007/archive/2008/02/28/2127981.aspx