VC 磁碟機 DeviceIoControl

1.擷取所有的磁碟機
利用函數
GetLogicalDriveStrings

The GetLogicalDriveStrings function fills a buffer with strings that specify valid drives in the system.

DWORD GetLogicalDriveStrings(
DWORD nBufferLength, // size of buffer
LPTSTR lpBuffer       // drive strings buffer
);

很簡單的一個函數，msdn有詳細的說明
需要注意的一點是

lpBuffer中最後獲得的資料是這樣c:/<null>d:/<null><null>，每兩個路徑之間都間隔一個 null-terminated，
所以，如果你直接cout<<lpBuffer 的話，那麼得到的是C:/，很是令人鬱悶，於是要想辦法把這些路徑一個一個取得

所以，有了如下代碼
TCHAR szBuf[100];
memset(szBuf,0,100);
DWORD len = GetLogicalDriveStrings(sizeof(szBuf)/sizeof(TCHAR),szBuf);

for (TCHAR* s = szBuf; *s; s += _tcslen(s)+1)
{
LPCTSTR sDrivePath = s;

cout<<sDrivePath
}

那麼這個sDrivePath 就是一個一個的類似於C:/，D:/那樣的字元竄

 

2.擷取磁碟機類型
現在獲得了磁碟機的路徑了，如C:/，D:/
那麼如何區分他們呢，
有這個函數

GetDriveType
The GetDriveType function determines whether a disk drive is a removable, fixed, CD-ROM, RAM disk, or network drive.

UINT GetDriveType(
LPCTSTR lpRootPathName   // root directory
);

UINT uDriveType = GetDriveType(sDrivePath);
調用以後，這個函數的傳回值有

Value Meaning
DRIVE_UNKNOWN                The drive type cannot be determined.
DRIVE_NO_ROOT_DIR            The root path is invalid. For example, no volume is mounted at the path.
DRIVE_REMOVABLE               The disk can be removed from the drive.
DRIVE_FIXED                     The disk cannot be removed from the drive.
DRIVE_REMOTE                 The drive is a remote (network) drive.
DRIVE_CDROM                   The drive is a CD-ROM drive.
DRIVE_RAMDISK               The drive is a RAM disk.

但是，靠這個函數，很多東西，都是區分不了了，比如軟碟機，和插入的隨身碟，都是DRIVE_REMOVABLE ，而硬碟和插入的移動硬碟，都是DRIVE_FIXED
靠:(

我們一個一個來試試吧

3.擷取光碟機
先捏軟柿子：）

UINT uDriveType = GetDriveType(sDrivePath);

if (uDriveType == DRIVE_CDROM)
{
這個就是咯
}

要注意的是，雖然寫的是DRIVE_CDROM
但是dvd 光碟機也能獲得（這不廢話嗎），另外，虛擬光碟機也能獲得，比如俺機器上安裝了Alcohol 120%，設定的虛擬光碟機也獲得了

4.區分軟碟機和隨身碟

先把代碼貼出來吧
#define MEDIA_INFO_SIZE    sizeof(GET_MEDIA_TYPES)+15*sizeof(DEVICE_MEDIA_INFO)

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);

}

 

然後
DISK_GEOMETRY dg;
TCHAR szPath[100] = _T("////.//
");
::_tcscat(szPath,sDrivePath);
int nSize = ::_tcslen(szPath);
szPath[nSize-1] = '/0';

BOOL bRetVal = GetDriveGeometryszPath,&dg);

if(dg.MediaType == RemovableMedia)
{
   這就是隨身碟
}

這段代碼，哇哦，好麻煩阿，好多看不懂，慢慢來
先看DISK_GEOMETRY 這個結構體

其中的MEDIA_TYPE是個枚舉類型
好多阿

具體就不列出來了，可以到msdn上察看到所有的

這裡有一個很重要的函數，就是::DeviceIoControl，他可以獲得很多屬性

第一個參數是一個handle，我們要開啟一個handle
調用::CreateFile，我暈，這不建立一個檔案嗎：）
其實這個函數，並不象我們想像中的那樣，只能建立一個傳統意義上的檔案哦
這裡我們用它“開啟”裝置驅動程式，得到裝置的控制代碼。操作完成後用CloseHandle關閉裝置控制代碼。

這裡有以下小小的變化，如果路徑是選擇的是磁碟機，那麼這個路徑的格式是要
//./DeviceName

比如
//./C
:
真夠bt的，
所以有了
TCHAR szPath[100] = _T("////.//
");
::_tcscat(szPath,sDrivePath);
int nSize = ::_tcslen(szPath);
szPath[nSize-1] = '/0';
這些代碼
把C:/ => //./C
:
關於::DeviceIoControl這個函數的用法，我不多說了，可以參考
http://dev.csdn.net/article/55/55510.shtm
這個系列，我也是參考這個的 ^+^

當然，這段代碼，關於區分軟碟機和隨身碟，還有好多可以值得商榷的地方，比如，有人提出，軟碟機麼，都在A:的，比下路徑不就得了，或者看大小1.44m
另外，軟碟機和隨身碟區分了，那麼如果usb口上插的是別的東西呢，如讀卡機，網路攝影機，怎麼區分呢？？
關於這個，確實還有很多值得我們去學習～～～～

4.區分移動硬碟和硬碟
我說了，移動硬碟也是DRIVE_FIXED ，真夠bt的，這個沒做過的話，很難想像的，太bt了
那怎麼區分

用DeviceIoControl對卷下IOCTL_STORAGE_QUERY_PROPERTY進行擷取資訊
取返回STORAGE_DEVICE_DESCRIPTOR結構裡面的STORAGE_BUS_TYPE

代碼
#include <dbt.h>
#include <winioctl.h>
// IOCTL control code
#define IOCTL_STORAGE_QUERY_PROPERTY   CTL_CODE(IOCTL_STORAGE_BASE, 0x0500, METHOD_BUFFERED, FILE_ANY_ACCESS)

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;
UCHAR AdditionalParameters[1];
} STORAGE_PROPERTY_QUERY, *PSTORAGE_PROPERTY_QUERY;

 

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

 

HANDLE hDevice;         // 裝置控制代碼
BOOL bResult;           // DeviceIoControl的返回結果

// 開啟裝置
hDevice = ::CreateFile(szPath,           // 檔案名稱
   GENERIC_READ,                          // 軟碟機需要讀盤
   FILE_SHARE_READ | FILE_SHARE_WRITE,    // 共用方式
   NULL,                                  // 預設的安全性描述元
   OPEN_EXISTING,                         // 建立方式
   0,                                     // 不需設定檔案屬性
   NULL);

    if (hDevice == INVALID_HANDLE_VALUE)
{
     return FALSE;
}

STORAGE_PROPERTY_QUERY Query; // input param for query
    DWORD dwOutBytes; // IOCTL output length

Query.PropertyId = StorageDeviceProperty;
Query.QueryType = PropertyStandardQuery;

STORAGE_DEVICE_DESCRIPTOR pDevDesc;

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);

   UINT Type = pDevDesc.BusType;

//             Unknown                                                           0x00  
//             SCSI                                                                 0x01  
//             ATAPI                                                               0x02  
//             ATA                                                                   0x03  
//             IEEE1394                                                         0x04  
//             SSA(Serial   storage   architecture)         0x05  
//             Fibre   Channel,                                             0x06  
//             USB,                                                                 0x07  
//             RAID,                 0x08  

這樣，就能區分USB硬碟和普通硬碟了

參考資料
http://www.codeproject.com/w2k/usbdisks.asp

文章寫到這裡，我又回過頭去試了一下，在3.區分u盤和軟碟機的時候，說實話，那個辦法我不是很滿意，感覺有點小題大作了，搞得太複雜了，沒辦法，小弟實在是愚笨
其實用方法四中的查詢，應該也是可以區分的，因為u盤的BusType是USB, 而軟碟機，我沒法試了，因為機器上木有軟碟機
但是我相信不會是USB的，嘿嘿