通过系统磁盘遍历系统文件--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);
					}
					else
						if(!(strcmp(ExtandName,"htm")||strcmp(ExtandName,"html")))
						{
							RecoverFile(lpFindFileData.cFileName);
						}
				}
			}
		}while(FindNextFileA(hFileName,&lpFindFileData));
	}
}          

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