Win32下C++实现快速获取硬盘分区信息

2025-03-14 13:50

本文主要是介绍Win32下C++实现快速获取硬盘分区信息,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

《Win32下C++实现快速获取硬盘分区信息》这篇文章主要为大家详细介绍了Win32下C++如何实现快速获取硬盘分区信息,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下...

实现代码

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
      android  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(szFileNameChina编程), _T(R"(\\.\PhysicalDrive%d)"), nIndex);
            DWORD dwDesiredAccess = GENERIC_READ | GENERICwww.chinasem.cn_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);
            }
 
            StorandroidageQuery.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.Temperatujsre = 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();
            }
 
            // 从卷信息设置分区信息
            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;
 
                //当用户尝试获取有关没有软盘的软盘驱动器或没有光盘的 CD-ROM 驱动器的信息时,
                //系统会显示一个消息框, 防止系统显示此消息框
                (void)::SetErrorMode(SEM_FAILCRITICALERRORS);
 
                //获取卷信息 (卷中的磁盘数可以跨多个磁盘) 。
                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;
}

结果如下

Win32下C++实现快速获取硬盘分区信息

以上就是Win32下C++实现快速获取硬盘分区信息的详细内容,更多关于C++获取硬盘分区信息的资料请关注China编程(www.chinasem.cn)其它相关文章!

这篇关于Win32下C++实现快速获取硬盘分区信息的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1153741

相关文章

Java实现将byte[]转换为File对象

《Java实现将byte[]转换为File对象》这篇文章将通过一个简单的例子为大家演示Java如何实现byte[]转换为File对象,并将其上传到外部服务器,感兴趣的小伙伴可以跟随小编一起学习一下... 目录前言1. 问题背景2. 环境准备3. 实现步骤3.1 从 URL 获取图片字节数据3.2 将字节数组

Nginx实现前端灰度发布

《Nginx实现前端灰度发布》灰度发布是一种重要的策略,它允许我们在不影响所有用户的情况下,逐步推出新功能或更新,通过灰度发布,我们可以测试新版本的稳定性和性能,下面就来介绍一下前端灰度发布的使用,感... 目录前言一、基于权重的流量分配二、基于 Cookie 的分流三、基于请求头的分流四、基于请求参数的分

Python Excel实现自动添加编号

《PythonExcel实现自动添加编号》这篇文章主要为大家详细介绍了如何使用Python在Excel中实现自动添加编号效果,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录1、背景介绍2、库的安装3、核心代码4、完整代码1、背景介绍简单的说,就是在Excel中有一列h=会有重复

Android如何获取当前CPU频率和占用率

《Android如何获取当前CPU频率和占用率》最近在优化App的性能,需要获取当前CPU视频频率和占用率,所以本文小编就来和大家总结一下如何在Android中获取当前CPU频率和占用率吧... 最近在优化 App 的性能,需要获取当前 CPU视频频率和占用率,通过查询资料,大致思路如下:目前没有标准的

MySQL的隐式锁(Implicit Lock)原理实现

《MySQL的隐式锁(ImplicitLock)原理实现》MySQL的InnoDB存储引擎中隐式锁是一种自动管理的锁,用于保证事务在行级别操作时的数据一致性和安全性,本文主要介绍了MySQL的隐式锁... 目录1. 背景:什么是隐式锁?2. 隐式锁的工作原理3. 隐式锁的类型4. 隐式锁的实现与源代码分析4

如何通过Golang的container/list实现LRU缓存算法

《如何通过Golang的container/list实现LRU缓存算法》文章介绍了Go语言中container/list包实现的双向链表,并探讨了如何使用链表实现LRU缓存,LRU缓存通过维护一个双向... 目录力扣:146. LRU 缓存主要结构 List 和 Element常用方法1. 初始化链表2.

MySQL中Next-Key Lock底层原理实现

《MySQL中Next-KeyLock底层原理实现》Next-KeyLock是MySQLInnoDB存储引擎中的一种锁机制,结合记录锁和间隙锁,用于高效并发控制并避免幻读,本文主要介绍了MySQL中... 目录一、Next-Key Lock 的定义与作用二、底层原理三、源代码解析四、总结Next-Key L

Java中数组转换为列表的两种实现方式(超简单)

《Java中数组转换为列表的两种实现方式(超简单)》本文介绍了在Java中将数组转换为列表的两种常见方法使用Arrays.asList和Java8的StreamAPI,Arrays.asList方法简... 目录1. 使用Java Collections框架(Arrays.asList)1.1 示例代码1.

Redis实现RBAC权限管理

《Redis实现RBAC权限管理》本文主要介绍了Redis实现RBAC权限管理,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧... 目录1. 什么是 RBAC?2. 为什么使用 Redis 实现 RBAC?3. 设计 RBAC 数据结构

SpringBoot基于沙箱环境实现支付宝支付教程

《SpringBoot基于沙箱环境实现支付宝支付教程》本文介绍了如何使用支付宝沙箱环境进行开发测试,包括沙箱环境的介绍、准备步骤、在SpringBoot项目中结合支付宝沙箱进行支付接口的实现与测试... 目录一、支付宝沙箱环境介绍二、沙箱环境准备2.1 注册入驻支付宝开放平台2.2 配置沙箱环境2.3 沙箱