监控磁盘大小c语言

Posted 2023-03-12

这篇文章主要介绍了C/C++ 监控磁盘与目录操作的示例，帮助大家更好的理解和学习C/C++编程，感兴趣的朋友可以了解下
遍历磁盘容量：
#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;

遍历文件特定路径:
循环遍历文件路径,并将文件后缀为.exe的路径筛选出来.
#include <stdio.h>
#include <windows.h>
#include <tlhelp32.h>

void SearchFile(char *pszDirectory)

// 搜索指定类型文件
char *pszFileName = NULL;
char *pTempSrc = NULL;
WIN32_FIND_DATA FileData = 0 ;

// 申请动态内存
pszFileName = new char[2048];
pTempSrc = new char[2048];

// 构造搜索文件类型字符串 *.* 表示搜索所有文件类型
wsprintf(pszFileName, "%s\\*.*", pszDirectory);

HANDLE hFile = ::FindFirstFile(pszFileName, &FileData);
if (INVALID_HANDLE_VALUE != hFile)

do

// 过滤掉当前目录"." 和上一层目录".."
if ('.' == FileData.cFileName[0])
continue;

// 拼接文件路径
wsprintf(pTempSrc, "%s\\%s", pszDirectory, FileData.cFileName);
// 判断是否是目录还是文件
if (FileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
SearchFile(pTempSrc); // 如果是目录则继续递归
else

char drive[_MAX_DRIVE], dir[_MAX_DIR], fname[_MAX_FNAME], ext[_MAX_EXT];
_splitpath(pTempSrc, drive, dir, fname, ext);
// 如果是文件并且后缀为.exe则输出具体路径
if (strcmp(ext, ".exe") == 0)
printf("%s \n", pTempSrc);

while (::FindNextFile(hFile, &FileData));

FindClose(hFile);
delete[]pTempSrc;
delete[]pszFileName;


int main(int argc, char * argv[])

SearchFile("c:\\MinGW7");
system("pause");
return 0;

监控文件目录变化：
#include <stdio.h>
#include <windows.h>
#include <tlhelp32.h>

UINT MonitorFileThreadProc(LPVOID lpVoid)

char *pszDirectory = (char *)lpVoid;

// 打开目录, 获取文件句柄
HANDLE hDirectory = CreateFile(pszDirectory, FILE_LIST_DIRECTORY, FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
if (INVALID_HANDLE_VALUE == hDirectory)
return 1;

char szFileName[MAX_PATH] = 0 ;
BOOL bRet = FALSE;
DWORD dwRet = 0;
DWORD dwBufferSize = 2048;

// 申请一个足够大的缓冲区
BYTE *pBuf = new BYTE[dwBufferSize];
if (NULL == pBuf)
return 2;

FILE_NOTIFY_INFORMATION *pFileNotifyInfo = (FILE_NOTIFY_INFORMATION *)pBuf;

// 开始循环设置监控
do

RtlZeroMemory(pFileNotifyInfo, dwBufferSize);
// 设置监控目录
bRet = ReadDirectoryChangesW(hDirectory, pFileNotifyInfo, dwBufferSize, TRUE,
FILE_NOTIFY_CHANGE_FILE_NAME | // 修改文件名
FILE_NOTIFY_CHANGE_ATTRIBUTES | // 修改文件属性
FILE_NOTIFY_CHANGE_LAST_WRITE, // 最后一次写入
&dwRet, NULL, NULL);
if (FALSE == bRet)
break;

// 将宽字符转换成窄字符,宽字节字符串转多字节字符串
WideCharToMultiByte(CP_ACP, 0, (wchar_t *)(&pFileNotifyInfo->FileName),
(pFileNotifyInfo->FileNameLength / 2),szFileName,MAX_PATH,NULL,NULL);

// 将路径与文件连接成完整文件路径
char FullFilePath[1024] = 0 ;
strncpy(FullFilePath, pszDirectory, strlen(pszDirectory));
strcat(FullFilePath, szFileName);

// 判断操作类型并显示
switch (pFileNotifyInfo->Action)

case FILE_ACTION_ADDED:
printf("文件被 [创建]: %s \n", FullFilePath); break;
case FILE_ACTION_REMOVED:
printf("文件被 [删除]: %s \n", FullFilePath); break;
case FILE_ACTION_MODIFIED:
printf("文件被 [修改]: %s \n", FullFilePath); break;
case FILE_ACTION_RENAMED_OLD_NAME:
printf("文件被 [重命名]: %s \n", FullFilePath); break;

while (bRet);

CloseHandle(hDirectory);
delete[] pBuf;
pBuf = NULL;
return 0;


int main(int argc, char * argv[])

char *pszDirectory = "C:\\";
// 创建线程开始监控
CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)MonitorFileThreadProc, pszDirectory, 0, NULL);
while (1)

Sleep(10000);

system("pause");
return 0;

监控目录文件变化:
可以将其改为一个简单的文件防篡改程序,也可以用来监控病毒的行为.
#include <stdio.h>
#include <Windows.h>
#include <tlhelp32.h>

DWORD WINAPI MonitorFileThreadProc(LPVOID lParam)

char *pszDirectory = (char *)lParam;
BOOL bRet = FALSE;
BYTE Buffer[1024] = 0 ;

FILE_NOTIFY_INFORMATION *pBuffer = (FILE_NOTIFY_INFORMATION *)Buffer;
DWORD dwByteReturn = 0;
HANDLE hFile = CreateFile(pszDirectory, FILE_LIST_DIRECTORY, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,
NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL);
if (INVALID_HANDLE_VALUE == hFile)
return 1;

while (TRUE)

ZeroMemory(Buffer, sizeof(Buffer));
// 设置监控目录回调函数
bRet = ReadDirectoryChangesW(hFile,&Buffer,sizeof(Buffer),TRUE,
FILE_NOTIFY_CHANGE_FILE_NAME | // 修改文件名
FILE_NOTIFY_CHANGE_ATTRIBUTES | // 修改文件属性
FILE_NOTIFY_CHANGE_LAST_WRITE, // 最后一次写入
&dwByteReturn, NULL, NULL);
if (TRUE == bRet)

char szFileName[MAX_PATH] = 0 ;

// 将宽字符转换成窄字符,宽字节字符串转多字节字符串
WideCharToMultiByte(CP_ACP,0,pBuffer->FileName,(pBuffer->FileNameLength / 2),
szFileName,MAX_PATH,NULL,NULL);

// 将路径与文件连接成完整文件路径
char FullFilePath[1024] = 0 ;
strncpy(FullFilePath, pszDirectory, strlen(pszDirectory));
strcat(FullFilePath, szFileName);

switch (pBuffer->Action)

case FILE_ACTION_ADDED:

printf("添加: %s \n", FullFilePath); break;

case FILE_ACTION_REMOVED:

printf("删除: %s \n", FullFilePath); break;

case FILE_ACTION_MODIFIED:

printf("修改: %s \n", FullFilePath); break;

case FILE_ACTION_RENAMED_OLD_NAME:

printf("重命名: %s", szFileName);
if (0 != pBuffer->NextEntryOffset)

FILE_NOTIFY_INFORMATION *tmpBuffer = (FILE_NOTIFY_INFORMATION *)
((DWORD)pBuffer + pBuffer->NextEntryOffset);
switch (tmpBuffer->Action)

case FILE_ACTION_RENAMED_NEW_NAME:

ZeroMemory(szFileName, MAX_PATH);
WideCharToMultiByte(CP_ACP,0,tmpBuffer->FileName,
(tmpBuffer->FileNameLength / 2),
szFileName,MAX_PATH,NULL,NULL);
printf(" -> %s \n", szFileName);
break;



break;

case FILE_ACTION_RENAMED_NEW_NAME:

printf("重命名(new): %s \n", FullFilePath); break;




CloseHandle(hFile);
return 0;


int main(int argc, char * argv[])

char *pszDirectory = "C:\\";

HANDLE hThread = CreateThread(NULL, 0, MonitorFileThreadProc, pszDirectory, 0, NULL);
WaitForSingleObject(hThread, INFINITE);
CloseHandle(hThread);
return 0;

以上就是C/C++ 监控磁盘与目录操作的示例的详细内容 参考技术A 要解决这个问题，根据我们已经学过的知识，其实有这样一个简单的思路：这是一个监控程序，需要不断地检查磁盘剩余容量，所以主体上这是一个while(true)的无限循环：因为需要检查磁盘剩余容量，所以我们需要调用系统API得到系统的磁盘剩余容量，稍微google，就可以得到GetDiskFreeSpace()函数就是来做这件事的。最后，既然是检测，那就是当得到的剩余容量小于某个阀值时，提出警告，于是，这里需要用到条件语句对剩余容量进行判断，并采取不同的措施，如果满足剩余容量，这Sleep()一定时间后继续检查，如果不满足，则提出警告并结束整个循环。

按照上面的思路，我们可以将整个监控程序实现如下：

/*
============================================================================
diskCapacity1.c
Created on : 2013年10月31日16:28:13
Author : Bruce
Version : v1.0
Copyright : Your copyright notice
Description : disk capacity in C, Ansi-style
============================================================================
*/
#include <windows.h>
#include <stdio.h>
#include <stdbool.h>
int main()

//开始无限循环，不断见识磁盘容量
while(true)

ULARGE_INTEGER space;

//调用Windows API获得C盘的剩余容量
if(GetDiskFreeSpaceEx("C:",0,0,&space))

//将剩余字节数转换为G（千兆字节）为单位
int free = space.QuadPart/(1024*1024*1024);
//如果剩余字节数小于10千兆字节，则向用户报警，并用break关键字结束循环
if(free <10)

puts("WARNNING! The free space is less than 10G.");
break;

//休眠10分钟（60*1000），开始下一次循环
Sleep(10*60*1000);


return 0;
参考技术B 在工作过程中，很多spark在saveAsTextFile的过程中都会repartition，用于减少磁盘文件，节约内存空间；但是repartition的分区数，只能根据实际结果测试后重新进行调整或者根据自己的经验进行预估。但是当数据骤增或骤减的时候之前的经验值就不是那么可靠，而且当数据骤增时采用snapyy、gz等让当个文件过大导致task oom。

但是在实际测试、API说明中发现，SizeEstimator 只是计算了Java Heap中大小而不是磁盘落地大小。

于是使用：RDD总数／takeSample取样数*takeSample取样大小=RDD总大小；(sample 返回的数据占比不一定与参数相同)

这样有两个缺点：

有两个shuffle

结果浮动较大

所以，我们使用了累加器Accumulator，使用map获取每行的大小并进行累加，代码如下：

import org.apache.spark.SparkContext

import org.apache.spark.rdd.RDD

/**

* 通过累积器计算分区数量

*/ 参考技术C 监控磁盘大小
   ///   
        /// 获取指定驱动器的空间总大小(单位为B) 
        ///   
        ///  只需输入代表驱动器的字母即可 （大写） 
        ///    
        public static long GetHardDiskSpace(string str_HardDiskName)
       
            long totalSize = new long();
            str_HardDiskName = str_HardDiskName + ":\\";
            System.IO.DriveInfo[] drives = System.IO.DriveInfo.GetDrives();
            foreach (System.IO.DriveInfo drive in drives)
           
                if (drive.Name == str_HardDiskName)
               
                    totalSize = drive.TotalSize / (1024 * 1024 * 1024);
               
           
            return totalSize;
       

        ///   
        /// 获取指定驱动器的剩余空间占比(单位为B) 
        ///   
        ///  只需输入代表驱动器的字母即可  
        ///    
        public static float GetHardDiskFreeSpaceRate(string str_HardDiskName)
       
            long freeSpace = new long();
            float rate = 0;
            str_HardDiskName = str_HardDiskName + ":\\";
            System.IO.DriveInfo[] drives = System.IO.DriveInfo.GetDrives();
            foreach (System.IO.DriveInfo drive in drives)
           
                if (drive.Name == str_HardDiskName)
               
                    freeSpace = drive.TotalFreeSpace ;
                    rate = (float)freeSpace / drive.TotalSize;

               
           
            return rate;