广度优先文件遍历非递归 (Nonrecursive EnumFiles With Breadth-First Algo)

最新推荐文章于 2019-10-31 14:43:21 发布

Cpp肥兔

最新推荐文章于 2019-10-31 14:43:21 发布

阅读量682

点赞数 1

CC 4.0 BY-SA版权

分类专栏： c++

本文链接：https://blog.youkuaiyun.com/yezigu921/article/details/8555386

c++ 专栏收录该内容

2 篇文章

订阅专栏

本文介绍了如何使用广度优先算法来遍历指定文件夹下的所有文件，特别是在Adobe Acrobat插件中实现这一功能，用于判断文件夹下是否有敏感文件，并在必要时取消操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

时间：

2013-01-30 11:18

摘要：

遍历一个指定文件夹下的所有文件，基于广度优先思想，也可认为是树形结构的层次遍历

编写背景：

在做Adobe Acrobat插件时收到一个需求，Adobe有一个菜单功能是让用户指定一个folder，然后把foler下的所有文件导入到pdf中，用户要求hook这个过程，当用户指定一个folder时，判断folder下是否有敏感文件，如果有则取消本次操作。

个人感觉对于这样的需求，使用广度优先算法，找到敏感文件的时间必少于深度优先。

代码：

bool EnumFiles_Breadth_Fisrt(string RootFolder,bool (*DoWork)(string curFile))  //Level traversal ,
{
	// need RootFolder  optimization
	//optimization RootFolder
	if (!iends_with(RootFolder,"\\")) RootFolder+="\\"; 
	DWORD dw=::GetFileAttributesA(RootFolder.c_str() );
	if(  !(dw&FILE_ATTRIBUTE_DIRECTORY) )return  false;  // Folder only
	// Currently Find *.*
	queue<string> runtimeQueue;  //each node is path+filetpye
	runtimeQueue.push(RootFolder+"*.*");
	//Breadth-first
	do 
	{
		WIN32_FIND_DATAA fd;
		string curSearchPattern(runtimeQueue.front()); // current FindFileTpye
		string curfolder=curSearchPattern.substr(0,curSearchPattern.find_last_of('\\')+1);
		//	cout<<curfolder.c_str()<<endl;
		HANDLE hFind = ::FindFirstFileA(curSearchPattern.c_str(),&fd);	
		if (INVALID_HANDLE_VALUE == hFind  )	return false;

		do
		{
			if (fd.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)
			{
				if( !iequals(fd.cFileName,".") && !iequals(fd.cFileName,"..") )
				{
					string internalfolder(curfolder+fd.cFileName+"\\"); //a sub Folder
					runtimeQueue.push(internalfolder+"*.*");	//currently find type
				}
			}
			else
			{
				if(false==(*DoWork)(curfolder+fd.cFileName)) return false;
			}
		}while(FindNextFileA(hFind,&fd));
		runtimeQueue.pop();
		FindClose(hFind);
	} while (!runtimeQueue.empty());
	return true;

}