脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 软件编程 → C 语言 → C++磁盘

C/C++实现磁盘相关操作的示例代码

 更新时间：2023年11月20日 10:29:06   作者：微软技术分享  
这篇文章主要为大家详细介绍了C/C++如何实现磁盘相关操作,例如遍历磁盘容量、实现磁盘格式化、移除指定磁盘等,感兴趣的小伙伴可以跟随小编一起学习一下

遍历磁盘容量

如下代码实现了在Windows系统中获取所有磁盘驱动器的信息。具体包括两个函数，一个用于获取驱动器类型，另一个用于获取驱动器空间信息。主函数则调用这两个函数来遍历所有逻辑驱动器并输出相应的信息。在输出驱动器空间信息时，会输出该驱动器的总大小、已用空间以及可用空间。

#include <stdio.h>
#include <Windows.h>

void GetDrivesType(const char* lpRootPathName)
{
  UINT uDriverType = GetDriveType(lpRootPathName);
  switch (uDriverType)
  {
    case DRIVE_UNKNOWN:puts("未知磁盘"); break;
    case DRIVE_NO_ROOT_DIR: puts("路径无效"); break;
    case DRIVE_REMOVABLE: puts("可移动磁盘"); break;
    case DRIVE_FIXED: puts("固定磁盘"); break;
    case DRIVE_REMOTE: puts("网络磁盘"); break;
    case DRIVE_CDROM: puts("光驱"); break;
    case DRIVE_RAMDISK: puts("内存映射盘"); break;
    default: break;
  }
}

void GetDrivesFreeSpace(const char* lpRootPathName)
{
  unsigned long long available, total, free;
  if (GetDiskFreeSpaceEx(lpRootPathName, (ULARGE_INTEGER*)&available, 
    (ULARGE_INTEGER*)&total, (ULARGE_INTEGER*)&free))
  {
    printf("磁盘: %s | 总计: %lld MB 已用: %lld MB 剩余: %lld MB \n",
      lpRootPathName, total >> 20, available >> 20, free >> 20);
  }
}

int main(int argc,char *argv[])
{
  DWORD dwSize = MAX_PATH;
    char szLogicalDrives[MAX_PATH] = {0};

    // 获取逻辑驱动器号字符串
  DWORD dwResult = GetLogicalDriveStringsA(dwSize, szLogicalDrives);
  
  if (dwResult > 0 && dwResult <= MAX_PATH)
  {
      // 从缓冲区起始地址开始
    char* szSingleDrive = szLogicalDrives;
    while (*szSingleDrive) {
      //printf("Drive: %s\n", szSingleDrive);      // 输出单个驱动器的驱动器号
      // GetDrivesType(szSingleDrive);
      GetDrivesFreeSpace(szSingleDrive);
      // 获取下一个驱动器地址
      szSingleDrive += strlen(szSingleDrive) + 1;
    }
  }

  system("pause");
  return 0;
}

遍历盘符并存储

循环遍历盘符分区,并将所有盘符结构存储到std::vector<MyDriver>定义的容器中.

#include <windows.h>
#include <iostream>
#include <vector>
#include <string>

// 将字节转换为GB单位显示的宏定义
#define ToGB(x) (x.HighPart << 2) + (x.LowPart >> 20) / 1024.0

// 定义基础结构
typedef struct
{
    double available_space;
    double free_space;
    double total_space;
}DriverInfo;

// 定义完整结构
typedef struct
{
    char driver_name[128];
    char driver_type[128];
    double available_space;
    double free_space;
    double total_space;
}MyDriver;

using namespace std;

// 获取驱动器数量
int GutDrivesCount()
{
    DWORD drivers;
    int count = 0;

    //获取驱动器数
    drivers = GetLogicalDrives();
    while (drivers != 0)
    {
        if (drivers & 1 != 0)
        {
            count++;
        }
        drivers >>= 1;
    }
    return count;
}

// 获取驱动器类型
std::string GetDrivesType(const char* lpRootPathName)
{
    UINT uDriverType = GetDriveType(lpRootPathName);
    switch (uDriverType)
    {
    case DRIVE_UNKNOWN:
        return "未知类型"; break;
    case DRIVE_NO_ROOT_DIR:
        return "路径无效"; break;
    case DRIVE_REMOVABLE:
        return "可移动磁盘"; break;
    case DRIVE_FIXED:
        return "固定磁盘"; break;

    case DRIVE_REMOTE:
        return "网络磁盘"; break;
    case DRIVE_CDROM:
        return "光驱设备"; break;
    case DRIVE_RAMDISK:
        return "内存映射盘"; break;
    default:
        break;
    }
    return "错误参数";
}

// 获取盘符容量
DriverInfo GetDrivesFreeSpace(const char* lpRootPathName)
{
    // ULARGE_INTEGER 64位无符号整型值
    ULARGE_INTEGER available, total, free;
    DriverInfo ref;

    // 获取分区数据并返回DriversInfo结构体
    if (GetDiskFreeSpaceEx(lpRootPathName, (ULARGE_INTEGER*)&available, (ULARGE_INTEGER*)&total, (ULARGE_INTEGER*)&free))
    {
        ref.total_space = ToGB(total);
        ref.free_space = ToGB(available);
        ref.available_space = ref.total_space - ref.free_space;
    }
    return ref;
}

std::vector<MyDriver> GetDriveForVector()
{
    DWORD count = GutDrivesCount();
    std::cout << "驱动器个数: " << count << std::endl;

    DWORD dwSize = MAX_PATH;
    char szLogicalDrives[MAX_PATH] = { 0 };

    // 获取逻辑驱动器号字符串
    DWORD dwResult = GetLogicalDriveStrings(dwSize, szLogicalDrives);

    // 处理获取到的结果
    if (dwResult > 0 && dwResult <= MAX_PATH)
    {
        // 定义两个结构, MyDriver 临时存储单个结构,ref存储所有磁盘的容器
        MyDriver my_driver_ptr;
        std::vector<MyDriver> ref;

        // 从缓冲区起始地址开始
        char* szSingleDrive = szLogicalDrives;

        while (*szSingleDrive)
        {
            // 逻辑驱动器类型
            std::string type = GetDrivesType(szSingleDrive);

            // 获取磁盘空间信息并存入 DriverInfo 结构
            DriverInfo ptr;
            ptr = GetDrivesFreeSpace(szSingleDrive);

            // 填充结构数据
            strcpy(my_driver_ptr.driver_name, szSingleDrive);
            strcpy(my_driver_ptr.driver_type, type.c_str());
            my_driver_ptr.total_space = ptr.total_space;
            my_driver_ptr.free_space = ptr.free_space;
            my_driver_ptr.available_space = ptr.available_space;

            // 加入到容器中
            ref.push_back(my_driver_ptr);
            /*
            std::cout
                << "盘符: " << szSingleDrive
                << " 类型: " << type
                << " 总容量: " << ptr.total_space
                << " 可用空间: " << ptr.free_space
                << " 已使用: " << ptr.available_space
                << std::endl;
            */
            // 获取下一个驱动器号起始地址
            szSingleDrive += strlen(szSingleDrive) + 1;
        }
        return ref;
    }
}

int main(int argc,char *argv[])
{
    std::vector<MyDriver> ptr = GetDriveForVector();

    // 循环输出vector容器
    for (int x = 0; x < ptr.size(); x++)
    {
        std::cout
            << "盘符: " << ptr[x].driver_name
            << " 类型: " << ptr[x].driver_type
            << " 总容量: " << ptr[x].total_space
            << " 可用空间: " << ptr[x].free_space
            << " 已使用: " << ptr[x].available_space
            << std::endl;
    }

    std::system("pause");
    return 0;
}

实现磁盘格式化

如下代码定义了一个函数FormatDisk，用于格式化由指定为字符串的驱动器号标识的磁盘。该函数使用Shell2.dll模块中的SHFormatDrive()这个未导出函数实现对特定磁盘的格式化。

FormatDisk函数采用std:：string参数strDisk，该参数指定要格式化的磁盘的驱动器号。该函数首先使用LoadLibraryA加载Shell32.dll库，然后使用GetProcAddress检索SHFormatDrive函数的地址。使用控制台应用程序的窗口句柄、要格式化的磁盘的驱动器ID（根据驱动器号计算）以及指定格式选项的标志来调用SHFormatDrive函数。

#include <iostream>
#include <string>
#include <Windows.h>
#include <ShlObj.h>

#pragma comment(lib, "Shell32.lib")

// 格式化磁盘
void FormatDisk(std::string strDisk)
{
  HINSTANCE hInstance = ::LoadLibraryA("Shell32.dll");
  if (NULL == hInstance)
  {
    return;
  }
  typedef DWORD(*PSHFORMATDRIVE)(HWND, UINT, UINT, UINT);
  PSHFORMATDRIVE SHFormatDrive = (PSHFORMATDRIVE)::GetProcAddress(hInstance, "SHFormatDrive");
  if (NULL == SHFormatDrive)
  {
    return;
  }
  UINT uiID = strDisk[0] - 'A';

  // 获取控制台程序窗口句柄
  HWND hWnd = ::GetConsoleWindow();

  SHFormatDrive((HWND)hWnd, uiID, 0xFFFF, 0x0001);
  ::FreeLibrary(hInstance);
}

int main(int argc, char* argv[])
{
  // 传入磁盘
  FormatDisk("D");

  return 0;
}

移除指定磁盘

如下代码演示了如何通过 Windows API 移除指定的磁盘驱动器，包括移除盘符和卸载卷加载点。代码首先定义了一个 DeleteVolume 函数，接收一个指向字符串的指针，表示要删除的磁盘驱动器的盘符。然后，函数将盘符转换为设备名称，使用 DefineDosDeviceA 函数将其从系统中移除。接着，函数使用 DeleteVolumeMountPointA 函数删除卷加载点。最后，main 函数调用 DeleteVolume 函数四次，移除了 C:、D:、E:、F:四个磁盘驱动器。

#include <iostream>
#include <Windows.h>

// 移除指定盘符
BOOL DeleteVolume(char* lpszDriver)
{
  // 将盘符和Dos设备路径移除
  char szDeviceName[MAX_PATH] = { 0 };
  strcpy(szDeviceName, lpszDriver);
  szDeviceName[2] = '\0';
  if (!DefineDosDeviceA(DDD_REMOVE_DEFINITION, szDeviceName, NULL))
  {
    return FALSE;
  }

  // 卸载卷加载点
  if (!DeleteVolumeMountPointA(lpszDriver))
  {
    return FALSE;
  }
  return TRUE;
}

int main(int argc, char *argv[])
{
  DeleteVolume((char*)"C:");
  DeleteVolume((char *)"D:");
  DeleteVolume((char*)"E:");
  DeleteVolume((char*)"F:");

    return 0;
}

输出磁盘分区表

如下代码，用于读取和分析Windows系统上第一个物理硬盘的主引导记录MBR。代码中定义了几个数据结构来表示MBR及其组件，包括引导记录、磁盘分区表和磁盘签名，ShowMbr使用ReadFile函数从硬盘读取MBR数据，然后以十六进制格式逐字节打印MBR数据。AnalysMbr函数提取并分析MBR数据，打印出引导记录、磁盘签名和分区表信息。

主函数中使用CreateFileA打开第一个物理硬盘，使用ShowMbr函数读取MBR数据，使用AnalystMbr函数分析MBR数据，然后使用CloseHandle函数关闭文件句柄，此段代码读者在编译时需采用64位模式编译。

#include <iostream>
#include <Windows.h>

// 定义数据结构体
#define BOOTRECORDSIZE 440
typedef struct _BOOTRECORD
{
  unsigned char BootRecore[BOOTRECORDSIZE];
}BOOTRECORD, * PBOOTRECORD;

#define DPTSIZE 64
typedef struct _DPT
{
  unsigned char Dpt[DPTSIZE];
}DPT, * PDPT;

#define DPTNUMBER 4
typedef struct _DP
{
  unsigned char BootSign;     // 引导标志
  unsigned char StartHsc[3];
  unsigned char PatitionType;   // 分区类型
  unsigned char EndHsc[3];
  ULONG SectorsPreceding;     // 本分去之前使用的扇区数
  ULONG SectorsInPatition;    // 分区的总扇区数
}DP, * PDP;

typedef struct _MBR
{
  BOOTRECORD BootRecord;      // 引导程序
  unsigned char ulSigned[4];    // windows磁盘签名
  unsigned char sReserve[2];    // 保留位
  DPT Dpt;            // 分区表
  unsigned char EndSign[2];   // 结束标志
}MBR, * PMBR;

void ShowMbr(HANDLE hFile, PMBR pMbr)
{
  DWORD dwTemp = 0;
  ::ReadFile(hFile, (LPVOID)pMbr, sizeof(MBR), &dwTemp, NULL);
  for (int i = 0; i < 512; i++)
  {
    printf("%2x ", ((BYTE*)pMbr)[i]);
    if (0 == ((i + 1) % 16))
    {
      printf("\r\n");
    }
    else if (0 == ((i + 1) % 8))
    {
      printf("   ");
    }
  }
}

void AnalysMbr(MBR Mbr)
{
  printf("\r\n引导记录:\r\n");

  for (int i = 0; i < BOOTRECORDSIZE; i++)
  {
    printf("%2x ", Mbr.BootRecord.BootRecore[i]);
    if (0 == ((i + 1) % 16))
    {
      printf("\r\n");
    }
    else if (0 == ((i + 1) % 8))
    {
      printf("   ");
    }
  }

  printf("\r\n磁盘签名:\r\n");
  for (int i = 0; i < 4; i++)
  {
    printf("%02x ", Mbr.ulSigned[i]);
  }

  printf("\r\n解析分区表:\r\n");
  for (int i = 0; i < DPTSIZE; i++)
  {
    printf("%02x ", Mbr.Dpt.Dpt[i]);
    if (0 == ((i + 1) % 16))
    {
      printf("\r\n");
    }
    else if (0 == ((i + 1) % 8))
    {
      printf("   ");
    }
  }

  printf("\r\n");

  PDP pDp = (PDP) & (Mbr.Dpt.Dpt);
  for (int i = 0; i < DPTNUMBER; i++)
  {
    printf("引导标识:%02x ", pDp[i].BootSign);
    printf("分区类型:%02x ", pDp[i].PatitionType);
    printf("\r\n");
    printf("本分区之前扇区数:%d ", pDp[i].SectorsPreceding);
    printf("本分区的总扇区数:%d", pDp[i].SectorsInPatition);
    printf("\r\n");
    printf("该分区的大小:%f\r\n", (double)pDp[i].SectorsInPatition / 1024 / 1024 * 512 / 1024 / 1024);
    printf("\r\n");
  }

  printf("结束标志:\r\n");
  for (int i = 0; i < 2; i++)
  {
    printf("%02x ", Mbr.EndSign[i]);
  }
  printf("\r\n");
}

int main(int argc, char* argv[])
{
  // 打开物理硬盘设备
  HANDLE hFile = ::CreateFileA("\\\\.\\PhysicalDrive0",
    GENERIC_READ | GENERIC_WRITE,FILE_SHARE_READ | FILE_SHARE_WRITE,
    NULL,OPEN_EXISTING,FILE_ATTRIBUTE_ARCHIVE,NULL);
  if (INVALID_HANDLE_VALUE == hFile)
  {
    return -1;
  }

  MBR Mbr = { 0 };
  ShowMbr(hFile, &Mbr);
  AnalysMbr(Mbr);
  CloseHandle(hFile);
  return 0;
}

到此这篇关于C/C++实现磁盘相关操作的示例代码的文章就介绍到这了,更多相关C++磁盘内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • C语言实践设计开发飞机游戏

    C语言实践设计开发飞机游戏

    飞机大战想必是很多人童年时期的经典游戏,我们依旧能记得抱个老人机娱乐的场景,下面这篇文章主要给大家介绍了关于如何利用C语言写一个简单的飞机大战小游戏的相关资料,需要的朋友可以参考下
    2022-02-02
  • C语言中的链接编写教程

    C语言中的链接编写教程

    这篇文章主要介绍了C语言中的链接编写教程,是C语言入门学习中的基础知识,需要的朋友可以参考下
    2015-08-08
  • C++设计模式之解释器模式

    C++设计模式之解释器模式

    这篇文章主要介绍了C++设计模式之解释器模式,本文讲解了什么是解释器模式、文法规则和抽象语法树、解释器模式的使用场合等内容,需要的朋友可以参考下
    2014-10-10
  • C++ ffmpeg实现将视频帧转换成jpg或png等图片

    C++ ffmpeg实现将视频帧转换成jpg或png等图片

    有时播放实时流的时候有截图的需求，需要将解码出来的图片保存本地或上传服务器，这时就需要将avframe中的数据编码成png、jpg等格式的图片，我们使用ffmpeg的相关编码器就可以实现功能，下面就来讲讲具体实现方法吧
    2023-03-03
  • Qt线程QThread开启和安全退出的实现

    Qt线程QThread开启和安全退出的实现

    本文主要介绍了Qt线程QThread开启和安全退出的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-06-06
  • 详解约瑟夫环问题及其相关的C语言算法实现

    详解约瑟夫环问题及其相关的C语言算法实现

    这篇文章主要介绍了详解约瑟夫环问题及其相关的C语言算法实现,也是ACM当中经常会引用到的基础题目,文中共介绍了三种C语言解答,需要的朋友可以参考下
    2015-08-08
  • C++关于引用作为函数的用法

    C++关于引用作为函数的用法

    今天小编就为大家分享一篇关于C++关于引用作为函数的用法，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧
    2018-12-12
  • 详解C++中常量的类型与定义

    详解C++中常量的类型与定义

    这篇文章主要介绍了详解C++中常量的类型与定义,使用#define与const来定义常量是C++入门学习中的基础知识,需要的朋友可以参考下
    2016-05-05
  • C语言实现家庭理财系统

    C语言实现家庭理财系统

    这篇文章主要为大家详细介绍了C语言实现家庭理财系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2021-11-11
  • 详解C++中static的用法

    详解C++中static的用法

    这篇文章主要介绍了c++中static的用法详解,C 语言的 static 关键字有三种用途，具体内容详情大家跟随脚本之家小编一起学习吧
    2018-05-05

最新评论