Win32下C++實(shí)現(xiàn)快速獲取硬盤(pán)分區(qū)信息
更新時(shí)間：2025年03月14日 11:05:06 作者：Flame_Cyclone
這篇文章主要為大家詳細(xì)介紹了Win32下C++如何實(shí)現(xiàn)快速獲取硬盤(pán)分區(qū)信息,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下
實(shí)現(xiàn)代碼
CDiskDriveUtils.h
#pragma once
#include <wtypesbase.h>
#include <string>
#include <tchar.h>
#include <vector>
#include <map>
#ifdef _UNICODE
using _tstring = std::wstring;
#else
using _tstring = std::string;
#endif
namespace CDiskDriveUtils
{
enum ePartitionStyle {
ePartitionMBR,
ePartitionGPT,
ePartitionRAW
};
enum eDriveType
{
eDriveUnknown,
eDriveNoRootDir,
eDriveRemovable,
eDriveFixed,
eDriveRemote,
eDriveCDROM,
eDriveRAMdisk
};
enum eStorageBusType {
eBusUnknown,
eBusScsi,
eBusAtapi,
eBusAta,
eBus1394,
eBusSsa,
eBusFibre,
eBusUsb,
eBusRAID,
eBusiScsi,
eBusSas,
eBusSata,
eBusSd,
eBusMmc,
eBusVirtual,
eBusFileBackedVirtual,
eBusSpaces,
eBusNvme,
eBusSCM,
eBusUfs,
eBusMax,
eBusMaxReserved = 0x7F
};
typedef struct _DISK_PARTION_INFO
{
_tstring PathName;
_tstring VolumeName;
uint64_t StartingOffset;
uint64_t PartitionLength;
uint64_t FreeBytes;
uint32_t PartitionNumber;
ePartitionStyle PartitionStyle;
eDriveType DriveType;
_DISK_PARTION_INFO() :
PartitionNumber(0),
PartitionLength(0),
StartingOffset(0),
FreeBytes(0),
PartitionStyle(ePartitionStyle::ePartitionRAW),
DriveType(eDriveType::eDriveUnknown)
{
}
}DISK_PARTION_INFO;
typedef struct _LOGICAL_VOLUME_INFO
{
_tstring PathName;
_tstring VolumeName;
_tstring FileSystemName;
uint64_t StartingOffset;
uint64_t ExtentLength;
uint64_t FreeBytes;
uint32_t DiskNumber;
eDriveType DriveType;
_LOGICAL_VOLUME_INFO() :
DiskNumber(0),
StartingOffset(0),
ExtentLength(0),
FreeBytes(0),
DriveType(eDriveType::eDriveUnknown)
{
}
}LOGICAL_VOLUME_INFO;
typedef struct _DISK_INFO
{
_tstring DevicePath;
_tstring ProductId;
_tstring SerialNumber;
_tstring AdapterSerialNumber;
_tstring ProductRevision;
_tstring BusTypeName;
uint64_t DiskSize;
uint64_t Attributes;
eStorageBusType BusType;
int16_t Temperature;
bool fRemovableMedia;
std::vector<DISK_PARTION_INFO> Partitions;
_DISK_INFO() :
Temperature(0),
BusType(eStorageBusType::eBusUnknown),
DiskSize(0),
Attributes(0),
fRemovableMedia(false)
{
}
}DISK_DRIVE_INFO;
std::map<uint32_t, DISK_DRIVE_INFO> GetDiskDriveInfos();
bool GetDiskDriveInfo(DWORD nIndex, DISK_DRIVE_INFO& info);
bool GetLogicalVolumeInfo(std::vector<LOGICAL_VOLUME_INFO>& vVolumeInfo);
}CDiskDriveUtils.cpp
#include "CDiskDriveUtils.h"
#include <winioctl.h>
#include <strsafe.h>
#include <cstdint>
namespace CDiskDriveUtils
{
std::string _WStrToMultiStr(UINT CodePage, const std::wstring& str)
{
int cbMultiByte = ::WideCharToMultiByte(CodePage, 0, str.c_str(), -1, NULL, 0, NULL, NULL);
std::string strResult(cbMultiByte, 0);
size_t nConverted = ::WideCharToMultiByte(CodePage, 0, str.c_str(), (int)str.size(), &strResult[0], (int)strResult.size(), NULL, NULL);
strResult.resize(nConverted);
return strResult;
}
std::wstring _MultiStrToWStr(UINT CodePage, const std::string& str)
{
int cchWideChar = ::MultiByteToWideChar(CodePage, 0, str.c_str(), -1, NULL, 0);
std::wstring strResult(cchWideChar, 0);
size_t nConverted = ::MultiByteToWideChar(CodePage, 0, str.c_str(), (int)str.size(), &strResult[0], (int)strResult.size());
strResult.resize(nConverted);
return strResult;
}
_tstring WStrToTStr(const std::wstring& str)
{
#ifdef _UNICODE
return str;
#else
return _WStrToMultiStr(CP_ACP, str);
#endif
}
_tstring AStrToTStr(const std::string& str)
{
#ifdef _UNICODE
return _MultiStrToWStr(CP_ACP, str);
#else
return str;
#endif
}
std::map<uint32_t, DISK_DRIVE_INFO> GetDiskDriveInfos()
{
std::map<uint32_t, DISK_DRIVE_INFO> result;
for (int i = 0; i < 1000; i++)
{
DISK_DRIVE_INFO info;
if (GetDiskDriveInfo(i, info))
{
result.emplace(i, info);
}
}
return result;
}
void RemoveBackSpace(_tstring& strName)
{
size_t nLength = strName.size();
for (auto it = strName.crbegin(); it != strName.crend(); it++)
{
if (_T(' ') == *it)
{
nLength--;
}
else
{
break;
}
}
strName.resize(nLength);
}
bool GetDiskDriveInfo(DWORD nIndex, DISK_DRIVE_INFO& info)
{
TCHAR szFileName[MAX_PATH] = { 0 };
HANDLE hFile = INVALID_HANDLE_VALUE;
DWORD dwBytesReturned = 0;
bool bSuccess = false;
do
{
(void)::StringCchPrintf(szFileName, _countof(szFileName), _T(R"(\\.\PhysicalDrive%d)"), nIndex);
DWORD dwDesiredAccess = GENERIC_READ | GENERIC_WRITE;
DWORD dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE;
hFile = ::CreateFile(szFileName, dwDesiredAccess, dwShareMode, NULL, OPEN_EXISTING, 0, NULL);
if (INVALID_HANDLE_VALUE == hFile)
{
break;
}
STORAGE_PROPERTY_QUERY StorageQuery = { };
StorageQuery.PropertyId = StorageDeviceProperty;
StorageQuery.QueryType = PropertyStandardQuery;
std::vector<uint8_t> OutBuffer(1024 * 64);
info.DevicePath = szFileName;
if (::DeviceIoControl(hFile, IOCTL_STORAGE_QUERY_PROPERTY, &StorageQuery, sizeof(StorageQuery), OutBuffer.data(), OutBuffer.size(), &dwBytesReturned, NULL))
{
PSTORAGE_DEVICE_DESCRIPTOR pDesc = (PSTORAGE_DEVICE_DESCRIPTOR)OutBuffer.data();
info.ProductId = AStrToTStr((LPCSTR)((LPBYTE)pDesc + pDesc->ProductIdOffset));
info.ProductRevision = AStrToTStr((LPCSTR)((LPBYTE)pDesc + pDesc->ProductRevisionOffset));
info.SerialNumber = AStrToTStr((LPCSTR)((LPBYTE)pDesc + pDesc->SerialNumberOffset));
info.BusType = (eStorageBusType)pDesc->BusType;
info.fRemovableMedia = pDesc->RemovableMedia;
RemoveBackSpace(info.ProductId);
}
StorageQuery.PropertyId = StorageDeviceTemperatureProperty;
if (::DeviceIoControl(hFile, IOCTL_STORAGE_QUERY_PROPERTY, &StorageQuery, sizeof(StorageQuery), OutBuffer.data(), OutBuffer.size(), &dwBytesReturned, NULL))
{
PSTORAGE_TEMPERATURE_DATA_DESCRIPTOR pDesc = (PSTORAGE_TEMPERATURE_DATA_DESCRIPTOR)OutBuffer.data();
info.Temperature = pDesc->TemperatureInfo->Temperature;
}
StorageQuery.PropertyId = StorageAdapterSerialNumberProperty;
if (::DeviceIoControl(hFile, IOCTL_STORAGE_QUERY_PROPERTY, &StorageQuery, sizeof(StorageQuery), OutBuffer.data(), OutBuffer.size(), &dwBytesReturned, NULL))
{
PSTORAGE_ADAPTER_SERIAL_NUMBER pDesc = (PSTORAGE_ADAPTER_SERIAL_NUMBER)OutBuffer.data();
info.AdapterSerialNumber = WStrToTStr(pDesc->SerialNumber);
}
if (::DeviceIoControl(hFile, IOCTL_DISK_GET_DRIVE_GEOMETRY_EX, NULL, 0, OutBuffer.data(), OutBuffer.size(), &dwBytesReturned, NULL))
{
PDISK_GEOMETRY_EX pDesc = (PDISK_GEOMETRY_EX)OutBuffer.data();
info.DiskSize = *(unsigned long long*) & pDesc->DiskSize;
}
if (::DeviceIoControl(hFile, IOCTL_DISK_GET_DRIVE_LAYOUT_EX, NULL, 0, OutBuffer.data(), OutBuffer.size(), &dwBytesReturned, NULL))
{
PDRIVE_LAYOUT_INFORMATION_EX pDesc = (PDRIVE_LAYOUT_INFORMATION_EX)OutBuffer.data();
for (DWORD i = 0; i < pDesc->PartitionCount; i++)
{
PPARTITION_INFORMATION_EX pPartInfo = &pDesc->PartitionEntry[i];
if (0 == pPartInfo->PartitionNumber)
{
continue;
}
DISK_PARTION_INFO partionInfo;
partionInfo.PartitionStyle = (ePartitionStyle)pPartInfo->PartitionStyle;
partionInfo.PartitionLength = pPartInfo->PartitionLength.QuadPart;
partionInfo.PartitionNumber = pPartInfo->PartitionNumber;
partionInfo.StartingOffset = pPartInfo->StartingOffset.QuadPart;
info.Partitions.push_back(partionInfo);
}
}
if (::DeviceIoControl(hFile, IOCTL_DISK_GET_DISK_ATTRIBUTES, NULL, 0, OutBuffer.data(), OutBuffer.size(), &dwBytesReturned, NULL))
{
PGET_DISK_ATTRIBUTES pDesc = (PGET_DISK_ATTRIBUTES)OutBuffer.data();
info.Attributes = pDesc->Attributes;
}
if (::DeviceIoControl(hFile, IOCTL_DISK_GET_CACHE_INFORMATION, NULL, 0, OutBuffer.data(), OutBuffer.size(), &dwBytesReturned, NULL))
{
PDISK_CACHE_INFORMATION pDesc = (PDISK_CACHE_INFORMATION)OutBuffer.data();
pDesc = (PDISK_CACHE_INFORMATION)OutBuffer.data();
}
// 從卷信息設(shè)置分區(qū)信息
std::vector<LOGICAL_VOLUME_INFO> vVolumeInfo;
(void)GetLogicalVolumeInfo(vVolumeInfo);
for (auto& Partition : info.Partitions)
{
for (const auto& volume : vVolumeInfo)
{
if (nIndex == volume.DiskNumber && Partition.StartingOffset == volume.StartingOffset)
{
Partition.PathName = volume.PathName;
Partition.VolumeName = volume.VolumeName;
Partition.FreeBytes = volume.FreeBytes;
Partition.DriveType = volume.DriveType;
break;
}
}
}
bSuccess = true;
} while (false);
if (INVALID_HANDLE_VALUE != hFile)
{
::CloseHandle(hFile);
}
return bSuccess;
}
bool GetLogicalVolumeInfo(std::vector<LOGICAL_VOLUME_INFO>& vVolumeInfo)
{
CHAR szOutBuffer[1024] = { 0 };
TCHAR szBuf[MAX_PATH] = { 0 };
DWORD dwBytesReturned = 0;
if (0 == GetLogicalDriveStrings(_countof(szBuf), szBuf))
{
return false;
}
LPCTSTR lpDriveString = szBuf;
while (_T('\0') != *lpDriveString)
{
TCHAR szDeviceBuf[MAX_PATH] = { 0 };
HANDLE hDevice = INVALID_HANDLE_VALUE;
::StringCchCopy(szDeviceBuf, _countof(szDeviceBuf), _T(R"(\\.\)"));
::StringCchCatN(szDeviceBuf, _countof(szDeviceBuf), lpDriveString, 2);
hDevice = ::CreateFile(szDeviceBuf, GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_SEQUENTIAL_SCAN,
NULL);
if (INVALID_HANDLE_VALUE == hDevice)
{
lpDriveString += 4;
continue;
}
if (::DeviceIoControl(hDevice,
IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS,
NULL,
0,
&szOutBuffer,
sizeof(szOutBuffer),
&dwBytesReturned,
NULL))
{
PVOLUME_DISK_EXTENTS pDesc = (PVOLUME_DISK_EXTENTS)szOutBuffer;
//當(dāng)用戶(hù)嘗試獲取有關(guān)沒(méi)有軟盤(pán)的軟盤(pán)驅(qū)動(dòng)器或沒(méi)有光盤(pán)的 CD-ROM 驅(qū)動(dòng)器的信息時(shí)，
//系統(tǒng)會(huì)顯示一個(gè)消息框, 防止系統(tǒng)顯示此消息框
(void)::SetErrorMode(SEM_FAILCRITICALERRORS);
//獲取卷信息 (卷中的磁盤(pán)數(shù)可以跨多個(gè)磁盤(pán)) 。
for (DWORD i = 0; i < pDesc->NumberOfDiskExtents; i++)
{
PDISK_EXTENT pDiskExtent = &pDesc->Extents[i];
TCHAR szVolumeName[MAX_PATH] = { 0 };
DWORD dwVolumeSerialNumber = 0;
DWORD dwMaximumComponentLength = 0;
DWORD dwFileSystemFlags = 0;
TCHAR szFileSystemName[MAX_PATH] = { 0 };
::GetVolumeInformation(lpDriveString,
szVolumeName,
_countof(szVolumeName),
&dwVolumeSerialNumber,
&dwMaximumComponentLength,
&dwFileSystemFlags,
szFileSystemName,
_countof(szFileSystemName)
);
ULARGE_INTEGER FreeBytesAvailableToCaller = { 0 };
ULARGE_INTEGER TotalNumberOfBytes = { 0 };
ULARGE_INTEGER TotalNumberOfFreeBytes = { 0 };
GetDiskFreeSpaceEx(lpDriveString, &FreeBytesAvailableToCaller, &TotalNumberOfBytes, &TotalNumberOfFreeBytes);
LOGICAL_VOLUME_INFO info;
info.PathName = _tstring(lpDriveString).substr(0, 2);
info.VolumeName = szVolumeName;
info.FileSystemName = szFileSystemName;
info.DiskNumber = pDiskExtent->DiskNumber;
info.StartingOffset = *(unsigned long long*) & pDiskExtent->StartingOffset;
info.ExtentLength = *(unsigned long long*) & pDiskExtent->ExtentLength;
info.FreeBytes = TotalNumberOfFreeBytes.QuadPart;
info.DriveType = (eDriveType)::GetDriveType(lpDriveString);
vVolumeInfo.push_back(info);
}
}
::CloseHandle(hDevice);
lpDriveString += 4;
}
return true;
}
}main.cpp
#include <locale.h>
#include <tchar.h>
#include "Win32Utils/CDiskDriveUtils.h"
int _tmain(int argc, LPCTSTR argv[])
{
setlocale(LC_ALL, "");
std::map<uint32_t, CDiskDriveUtils::DISK_DRIVE_INFO> diskInfoList = CDiskDriveUtils::GetDiskDriveInfos();
return 0;
}結(jié)果如下
以上就是Win32下C++實(shí)現(xiàn)快速獲取硬盤(pán)分區(qū)信息的詳細(xì)內(nèi)容，更多關(guān)于C++獲取硬盤(pán)分區(qū)信息的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
這篇文章主要為大家詳細(xì)介紹了C++實(shí)現(xiàn)俄羅斯方塊小游戲，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-10-10
C/C++實(shí)現(xiàn)的MD5哈希校驗(yàn)的示例代碼
MD5算法是一種廣泛使用的 Hash 算法,常用于確保信息傳輸?shù)耐暾耘c一致性,本文主要介紹了C/C++實(shí)現(xiàn)的MD5哈希校驗(yàn)的示例代碼,具有一定的參考價(jià)值,感興趣的可以了解一下2023-10-10
這篇文章主要介紹了二叉樹(shù)前序遍歷的非遞歸算法,需要的朋友可以參考下2014-02-02
深入解析C++設(shè)計(jì)模式編程中解釋器模式的運(yùn)用
這篇文章主要介紹了C++設(shè)計(jì)模式編程中解釋器模式的運(yùn)用,解釋器模式給定一個(gè)語(yǔ)言,定義它的文法的一種表示,并定義一個(gè)解釋器,這個(gè)解釋器使用該表示來(lái)解釋語(yǔ)言中的句子,需要的朋友可以參考下2016-03-03
C++標(biāo)準(zhǔn)之(ravalue reference) 右值引用介紹
臨時(shí)對(duì)象的產(chǎn)生和拷貝所帶來(lái)的效率折損，一直是C++所為人詬病的問(wèn)題,下面簡(jiǎn)單地介紹一下Copy Elision、RVO，對(duì)此不感興趣的可以直接跳過(guò)2012-11-11
這篇文章主要介紹了C++基礎(chǔ)——類(lèi)繼承中方法重載，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-04-04
GCC編譯過(guò)程(預(yù)處理,編譯,匯編,鏈接)及GCC命令詳解
文章詳細(xì)介紹了GCC編譯器的工作原理,包括預(yù)處理、編譯、匯編和鏈接四個(gè)主要階段,每個(gè)階段都有其特定的任務(wù)和輸出文件,文章還解釋了如何使用GCC命令選項(xiàng)來(lái)查看每個(gè)階段的輸出,以及如何通過(guò)調(diào)整編譯選項(xiàng)來(lái)優(yōu)化程序性能或調(diào)試問(wèn)題,感興趣的朋友跟隨小編一起看看吧2024-11-11
C語(yǔ)言實(shí)現(xiàn)通訊錄系統(tǒng)課程設(shè)計(jì)
這篇文章主要為大家詳細(xì)介紹了C語(yǔ)言實(shí)現(xiàn)通訊錄系統(tǒng)課程設(shè)計(jì)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-07-07
這篇文章主要介紹了簡(jiǎn)述C++的復(fù)雜性，幫助大家更好的理解和認(rèn)識(shí)c++編程語(yǔ)言，感興趣的朋友可以了解下2020-08-08
c語(yǔ)言函數(shù)如何求兩個(gè)數(shù)的最大值
這篇文章主要介紹了c語(yǔ)言函數(shù)如何求兩個(gè)數(shù)的最大值問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-12-12