vc讀取硬盤

 

VC下獲取各個驅動器路徑,分類及其獲取其信息的資料整理
1.獲取所有的驅動器
利用函數
GetLogicalDriveStrings

size_t szAllDriveStrings = GetLogicalDriveStrings(0,NULL);

//返回字符串的長度


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.


但是,靠這個函數,很多東西,都是區分不了了,比如軟驅,和插入的U盤,都是DRIVE_REMOVABLE ,而硬盤和插入的移動硬盤,都是DRIVE_FIXED
靠:(

我們一個一個來試試吧


3.獲取光驅
先捏軟柿子:)

UINT uDriveType = GetDriveType(sDrivePath);

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

要注意的是,雖然寫的是DRIVE_CDROM
但是dvd 光驅也能獲得(這不廢話嗎),另外,虛擬光驅也能獲得,比如俺機器上安裝了Alcohol 120%,設置的虛擬光驅也獲得了


4.區分軟驅和U盤

先把代碼貼出來吧
#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)

   
 
 
2 VC下獲取各個驅動器路徑
 
{
   // 設備無法打開...
   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)
{
    這就是U盤
}

這段代碼,哇哦,好麻煩阿,好多看不懂,慢慢來
先看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這個系列,我也是參考這個的 ^+^

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


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

用DeviceIoControl對卷下IOCTL_STORAGE_QUERY_PROPERTY進行獲取信息
   
 
 
3 VC下獲取各個驅動器路徑
 
取返回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硬盤和普通硬盤了
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章