通过系统磁盘遍历系统文件--C++实现

本文主要是介绍通过系统磁盘遍历系统文件--C++实现，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

#include<stdio.h>
#include<windows.h>
#include<string.h>
#include<shlwapi.h>//包含头文件
#pragma comment(lib,"shlwapi.lib");//包含库文件
int Dlength;//获得存储磁盘字符串的数组的字符长度
char FilePath;
char DriveString[20];//存放所有的磁盘字符串
void GetDrives();//该函数的作用是获取当前系统的所有磁盘，并且循环遍历磁盘下的文件（传参）
void LookForFile(char *);
void RecoverPE(char *);
void RecoverFile(char *);
void LookForSpecialProcess();
int main()
{LookForSpecialProcess();GetDrives();
}
void GetDrives()
{char drive[7];int length=0;//记录磁盘字符串的长度char Type=0;//记录磁盘类型int i=0;GetLogicalDriveStrings(20, DriveString);while(1){if(DriveString[length]!='\0'){/*if(DriveString[length+1]=='\0'){length++;}*/drive[i]=DriveString[length];i++;//DriveString[length+1]='\0';length+=1;}else{drive[i]=0x00;Type=GetDriveTypeA(drive);if(Type==DRIVE_REMOVABLE||Type==DRIVE_FIXED){drive[i]='*';drive[i+1]='.';drive[i+2]='*';drive[i+3]='\0';LookForFile(drive);//确定磁盘类型之后，这里的代码段就是遍历相应的磁盘}i=0;length++;if(DriveString[length]=='\0'&&DriveString[length+1]=='\0'){break;}}}Dlength=length;
}void LookForFile(char * pDrive)
{WIN32_FIND_DATA  lpFindFileData;HANDLE           hFileName;PTSTR            ExtandName;hFileName=FindFirstFileA(pDrive,&lpFindFileData);//获取C盘下的第一个目录信息if(hFileName!=INVALID_HANDLE_VALUE){do{if(strcmp(lpFindFileData.cFileName,".")&&strcmp(lpFindFileData.cFileName,".."))//判断当前的目录是不是上一层目录或者是不是当前目录{if(lpFindFileData.dwFileAttributes==FILE_ATTRIBUTE_DIRECTORY)//获取当前获取到的文件信息的属性，判断是不是目录，不是目录的话应该就是文件，如果是目录的话就递归进去寻找文件{//如果是目录的话就递归进去，查找文件strcat(lpFindFileData.cFileName,"*.*");FilePath=*lpFindFileData.cFileName;LookForFile(&FilePath);}else{//如果不是目录，那么就进行判断文件的后缀是不是目标后缀ExtandName=PathFindExtensionA(lpFindFileData.cFileName);if(!(strcmp(ExtandName,"exe")||strcmp(ExtandName,"dll"))){RecoverPE(lpFindFileData.cFileName);}elseif(!(strcmp(ExtandName,"htm")||strcmp(ExtandName,"html"))){RecoverFile(lpFindFileData.cFileName);}}}}while(FindNextFileA(hFileName,&lpFindFileData));}
}          

在实际编译的时候，如果只包含头文件，不包含库文件，会报错（#pragma comment(lib,“shlwapi.lib”);//包含库文件）必须要有

这篇关于通过系统磁盘遍历系统文件--C++实现的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---