windows的磁盘操作之七——获取当前所有的物理磁盘号

迈不过友情╰ 2022-05-10 12:33 428阅读 0赞

转载自：[http://blog.51cto.com/cutebunny/625577][http_blog.51cto.com_cutebunny_625577]

有了前几节的基础后，本节给出一个更复杂但却非常实用的例子。

很多情况下，我们想知道当前系统下安装了多少块磁盘，他们的物理驱动器号都是多少，每一块磁盘上有多少个分区，分区号怎么分布，每个分区大小是多少。这就类似于我们打开windows的磁盘管理看到的那种非常清晰的列表。对于后几个问题，我们根据物理驱动器号调用第五节[http://cutebunny.blog.51cto.com/301216/624567][http_cutebunny.blog.51cto.com_301216_624567]中的GetPartitionLetterFromPhysicalDrive函数，以及第三节[http://cutebunny.blog.51cto.com/301216/624079][http_cutebunny.blog.51cto.com_301216_624079]中的GetDiskDriveLayout函数即可搞定。那么我们这一节的重点放在如何获得当前所有物理驱动器号上。

先引入一个新的概念，设备GUID，它是同类设备统一并且唯一的标识码。对于磁盘，GUID为GUID\_DEVINTERFACE\_DISK，具体值为\{53F56307-B6BF-11D0-94F2-00A0C91EFB8B\}。windows提供一组API，可以通过GUID枚举出所有该类型的设备。先给出几个相关API的简要介绍

HDEVINFO

SetupDiGetClassDevs(

IN LPGUID  ClassGuid,  OPTIONAL

IN PCTSTR  Enumerator,  OPTIONAL

IN HWND  hwndParent,  OPTIONAL

IN DWORD  Flags

);

其中，ClassGuid填入我们感兴趣的设备GUID，该函数返回满足查询条件的一组设备的信息集合的句柄，该句柄就是获取设备信息的关键钥匙。

WINSETUPAPI BOOL WINAPI

SetupDiEnumDeviceInterfaces(

IN HDEVINFO  DeviceInfoSet,

IN PSP\_DEVINFO\_DATA  DeviceInfoData,  OPTIONAL

IN LPGUID  InterfaceClassGuid,

IN DWORD  MemberIndex,

OUT PSP\_DEVICE\_INTERFACE\_DATA  DeviceInterfaceData

);

该函数枚举SetupDiGetClassDevs获得的句柄中包含的所有设备。参数DeviceInfoSet填入我们上一步中获得的句柄，InterfaceClassGuid仍旧是我们感兴趣的GUID，MemberIndex为设备在集合中的索引，从0开始计数，最后DeviceInterfaceData是输出参数，存储枚举出的设备接口，后续可通过此接口获得详细的设备信息。

注意，参数DeviceInterfaceData.cbSize在调用前必须初始化为sizeof(SP\_DEVICE\_INTERFACE\_DATA)，这是函数的强制要求。

WINSETUPAPI BOOL WINAPI

SetupDiGetDeviceInterfaceDetail(

IN HDEVINFO  DeviceInfoSet,

IN PSP\_DEVICE\_INTERFACE\_DATA  DeviceInterfaceData,

OUT PSP\_DEVICE\_INTERFACE\_DETAIL\_DATA  DeviceInterfaceDetailData,  OPTIONAL

IN DWORD  DeviceInterfaceDetailDataSize,

OUT PDWORD  RequiredSize,  OPTIONAL

OUT PSP\_DEVINFO\_DATA  DeviceInfoData  OPTIONAL

);

该函数根据上两步中的句柄和接口获取设备的详细信息数据。参数DeviceInfoSet和DeviceInterfaceData在上两步中获得。输出参数DeviceInterfaceDetailData存储着设备信息数据，这个结构体中的成员DevicePath就是我们辛辛苦苦找寻的东西了。用它可以作为设备名调用CreateFile函数打开设备，之后的操作，嘿嘿，你懂的…

下面是具体代码

/\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*

\* Function: get device path from GUID

\* input: lpGuid, GUID pointer

\* output: pszDevicePath, device paths

\* return: Succeed, the amount of found device paths

\*         Fail, -1

\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*/

DWORD GetDevicePath(LPGUID lpGuid, CHAR \*\*pszDevicePath)

HDEVINFO hDevInfoSet;

SP\_DEVICE\_INTERFACE\_DATA ifdata;

PSP\_DEVICE\_INTERFACE\_DETAIL\_DATA pDetail;

DWORD nCount;

BOOL result;

//get a handle to a device information set

hDevInfoSet = SetupDiGetClassDevs(

lpGuid,      // class GUID

NULL,        // Enumerator

NULL,        // hwndParent

DIGCF\_PRESENT | DIGCF\_DEVICEINTERFACE    // present devices

);

//fail...

if (hDevInfoSet == INVALID\_HANDLE\_VALUE)

fprintf(stderr, "IOCTL\_STORAGE\_GET\_DEVICE\_NUMBER Error: %ld\\n", GetLastError());

return (DWORD)-1;

pDetail = (PSP\_DEVICE\_INTERFACE\_DETAIL\_DATA)malloc(INTERFACE\_DETAIL\_SIZE);

if (pDetail == NULL)

return (DWORD)-1;

pDetail->cbSize = sizeof(SP\_DEVICE\_INTERFACE\_DETAIL\_DATA);

nCount = 0;

result = TRUE;

// device index = 0, 1, 2... test the device interface one by one

while (result)

ifdata.cbSize = sizeof(ifdata);

//enumerates the device interfaces that are contained in a device information set

result = SetupDiEnumDeviceInterfaces(

hDevInfoSet,     // DeviceInfoSet

NULL,            // DeviceInfoData

lpGuid,          // GUID

nCount,   // MemberIndex

&ifdata        // DeviceInterfaceData

);

if (result)

// get details about a device interface

result = SetupDiGetDeviceInterfaceDetail(

hDevInfoSet,    // DeviceInfoSet

&ifdata,        // DeviceInterfaceData

pDetail,        // DeviceInterfaceDetailData

INTERFACE\_DETAIL\_SIZE,    // DeviceInterfaceDetailDataSize

NULL,           // RequiredSize

NULL          // DeviceInfoData

);

if (result)

// copy the path to output buffer

strcpy(pszDevicePath\[nCount\], pDetail->DevicePath);

//printf("%s\\n", pDetail->DevicePath);

nCount++;

free(pDetail);

(void)SetupDiDestroyDeviceInfoList(hDevInfoSet);

return nCount;

执行完毕后，所有满足条件的磁盘设备名称都存储在字符串数组pszDevicePath中。有了这个关键的数组，后面就可以为所欲为了。

以下是获得所有物理磁盘号的完整代码

/\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*

\* Function: get all present disks' physical number

\* input: N/A

\* output: ppDisks, array of disks' physical number

\* return: Succeed, the amount of present disks

\*         Fail, -1

\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*\*/

DWORD GetAllPresentDisks(DWORD \*\*ppDisks)

CHAR \*szDevicePath\[MAX\_DEVICE\];        // device path

DWORD nDevice;

HANDLE hDevice;

STORAGE\_DEVICE\_NUMBER number;

BOOL result;

DWORD readed;

WORD i, j;

for (i = 0; i < MAX\_DEVICE; i++)

szDevicePath\[i\] = (CHAR \*)malloc(INTERFACE\_DETAIL\_SIZE);

if (NULL == szDevicePath\[i\])

for (j = 0; j < i; j++)

free(szDevicePath\[i\]);

return (DWORD)-1;

// get the device paths

nDevice = GetDevicePath(const\_cast<LPGUID>(&GUID\_DEVINTERFACE\_DISK), szDevicePath);

if ((DWORD)-1 == nDevice)

for (i = 0; i < MAX\_DEVICE; i++)

free(szDevicePath\[i\]);

return (DWORD)-1;

\*ppDisks = (DWORD \*)malloc(sizeof(DWORD) \* nDevice);

// get the disk's physical number one by one

for (i = 0; i < nDevice; i++)

hDevice = CreateFile(

szDevicePath\[i\], // drive to open

GENERIC\_READ | GENERIC\_WRITE,     // access to the drive

FILE\_SHARE\_READ | FILE\_SHARE\_WRITE, //share mode

NULL,             // default security attributes

OPEN\_EXISTING,    // disposition

0,                // file attributes

NULL            // do not copy file attribute

);

if (hDevice == INVALID\_HANDLE\_VALUE) // cannot open the drive

for (j = 0; j < MAX\_DEVICE; j++)

free(szDevicePath\[j\]);

free(\*ppDisks);

fprintf(stderr, "CreateFile() Error: %ld\\n", GetLastError());

return DWORD(-1);

result = DeviceIoControl(

hDevice,                // handle to device

IOCTL\_STORAGE\_GET\_DEVICE\_NUMBER, // dwIoControlCode

NULL,                            // lpInBuffer

0,                               // nInBufferSize

&number,           // output buffer

sizeof(number),         // size of output buffer

&readed,       // number of bytes returned

NULL      // OVERLAPPED structure

);

if (!result) // fail

fprintf(stderr, "IOCTL\_STORAGE\_GET\_DEVICE\_NUMBER Error: %ld\\n", GetLastError());

for (j = 0; j < MAX\_DEVICE; j++)

free(szDevicePath\[j\]);

free(\*ppDisks);

(void)CloseHandle(hDevice);

return (DWORD)-1;

\*(\*ppDisks + i) = number.DeviceNumber;

(void)CloseHandle(hDevice);

for (i = 0; i < MAX\_DEVICE; i++)

free(szDevicePath\[i\]);

return nDevice;

代码说明：

1． 调用函数GetDevicePath获得前面所说的磁盘设备名称数组。

2． 对每一个磁盘设备，调用CreateFile打开并获得设备句柄。

3． 调用操作码为IOCTL\_STORAGE\_GET\_DEVICE\_NUMBER的DeviceIoControl函数获得磁盘物理驱动器号。

4． 将所有物理磁盘号存入数组返回。

大功告成了。可能有朋友会问，GetDevicePath不是已经获得了磁盘路径么，你前面说过，这个路径不是\\\\.\\PhysicalDriveX就是\\\\.\\X: ，那我们解析一下这个字符串不就可以获得磁盘号或者盘符了么。很可惜，这里的磁盘路径出现了第三种形式，而且是毫无章法的形式。打开函数GetDevicePath中的注释行//printf("%s\\n", pDetail->DevicePath);将这种形式的路径打印出来，可以看到类似为

\\\\?\\ide\#diskwdc\_wd1600aajs-08b4a0\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_01.03a01\#5&245a6b6d&0&0.0.0\#

\{53f56307-b6bf-11d0-94f2-00a0c91efb8b\}

\\\\?\\ide\#diskwdc\_wd1600aajs-08b4a0\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_\_01.03a01\#5&37141c12&0&0.1.0\#

\{53f56307-b6bf-11d0-94f2-00a0c91efb8b\}

所以，没办法，我们还是得用DeviceIoControl找出磁盘号。

[http_blog.51cto.com_cutebunny_625577]: http://blog.51cto.com/cutebunny/625577
[http_cutebunny.blog.51cto.com_301216_624567]: http://cutebunny.blog.51cto.com/301216/624567
[http_cutebunny.blog.51cto.com_301216_624079]: http://cutebunny.blog.51cto.com/301216/624079