学步园

　　平时写程序时经常会用到的遍历子目录枚举文件的功能，用 Windows API 函数 FindFirstFile() 和 FindNextFile() 直接实现起来会相当繁琐，有许多细节需要注意和记忆，要在短时间内写出可以正常工作的、没有BUG的、能够递归遍历多层子目录并枚举其中所有文件的程序代码，不是一件轻松的事情。以下这个 doFileEnumeration() 函数，是我（liigo）在 Windows API 函数 FindFirstFile() 和 FindNextFile() 的基础上封装实现的，可以轻松自如的处理遍历子目录枚举文件的任务，使用起来非常简单，具有较高的实用价值。这个函数其实是我之前发布的易语言“辅助调试支持库”（已开源至googlecode）中“枚举文件”“枚举子目录”这两条命令的底层实现函数。

　　doFileEnumeration() 函数功能是实现枚举文件和枚举子目录，支持递归处理多层子目录嵌套的情况。它有五个参数：第一个参数lpPath指定欲遍历的路径（文件夹）；第二个参数bRecursion指定是否递归处理子目录；第三个参数bEnumFiles指定是枚举文件还是枚举子目录；第四个参数pFunc为用户回调函数，枚举过程中每遇到一个文件或子目录，都会调用它，并传入这个文件或子目录的完整路径；第五个参数pUserData为用户任意指定的数据，它也将被传入用户回调函数。用户回调函数(EnumerateFunc)有两个参数，一个是文件或子目录的完整路径(lpFileOrPath)，一个是用户自定义数据(pUserData)，它被自动调用，用户需在此函数中编码处理代码。

　　这个函数经过我(liigo)的仔细设计，接口很简洁，使用也方便。有意去除了 FindFirstFile(Ex)/FindNextFile 中过滤文件名称或属性的部分功能，也是出于简化接口的考虑，——在用户回调函数中处理这些事情也是很容易的（在易语言中更容易）。在实现的细节上，重点是对子目录和递归的处理。以下是完整的源代码：

#include <windows.h></p> <p>typedef BOOL (WINAPI *EnumerateFunc) (LPCSTR lpFileOrPath, void* pUserData);</p> <p>void doFileEnumeration(LPSTR lpPath, BOOL bRecursion, BOOL bEnumFiles, EnumerateFunc pFunc, void* pUserData)<br /> {</p> <p> static BOOL s_bUserBreak = FALSE;<br /> try{<br /> //-------------------------------------------------------------------------<br /> if(s_bUserBreak) return;</p> <p> int len = strlen(lpPath);<br /> if(lpPath==NULL || len<=0) return;</p> <p> //NotifySys(NRS_DO_EVENTS, 0,0);</p> <p> char path[MAX_PATH];<br /> strcpy(path, lpPath);<br /> if(lpPath[len-1] != '//') strcat(path, "//");<br /> strcat(path, "*");</p> <p> WIN32_FIND_DATA fd;<br /> HANDLE hFindFile = FindFirstFile(path, &fd);<br /> if(hFindFile == INVALID_HANDLE_VALUE)<br /> {<br /> ::FindClose(hFindFile); return;<br /> }</p> <p> char tempPath[MAX_PATH]; BOOL bUserReture=TRUE; BOOL bIsDirectory;</p> <p> BOOL bFinish = FALSE;<br /> while(!bFinish)<br /> {<br /> strcpy(tempPath, lpPath);<br /> if(lpPath[len-1] != '//') strcat(tempPath, "//");<br /> strcat(tempPath, fd.cFileName);</p> <p> bIsDirectory = ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0);</p> <p> //如果是.或..<br /> if( bIsDirectory<br /> && (strcmp(fd.cFileName, ".")==0 || strcmp(fd.cFileName, "..")==0))<br /> {<br /> bFinish = (FindNextFile(hFindFile, &fd) == FALSE);<br /> continue;<br /> }</p> <p> if(pFunc && bEnumFiles!=bIsDirectory)<br /> {<br /> bUserReture = pFunc(tempPath, pUserData);<br /> if(bUserReture==FALSE)<br /> {<br /> s_bUserBreak = TRUE; ::FindClose(hFindFile); return;<br /> }<br /> }</p> <p> //NotifySys(NRS_DO_EVENTS, 0,0);</p> <p> if(bIsDirectory && bRecursion) //是子目录<br /> {<br /> doFileEnumeration(tempPath, bRecursion, bEnumFiles, pFunc, pUserData);<br /> }</p> <p> bFinish = (FindNextFile(hFindFile, &fd) == FALSE);<br /> }</p> <p> ::FindClose(hFindFile);</p> <p> //-------------------------------------------------------------------------<br /> }catch(...){ ASSERT(0); return; }<br /> }

以下是应用示例代码：

BOOL WINAPI myEnumerateFunc(LPCSTR lpFileOrPath, void* pUserData)<br /> {<br /> char* pdot;<br /> if((pdot = strrchr(lpFileOrPath, '.')) && stricmp(pdot, ".mp3") == 0)<br /> {<br /> printf("%s/n", lpFileOrPath);<br /> }<br /> return TRUE;<br /> }</p> <p>int main()<br /> {<br /> doFileEnumeration("C://Music", TRUE, TRUE, myEnumerateFunc, NULL);<br /> return 0;<br /> }