用递归写Win32的文件夹遍历 _企业安全

文件夹遍历技术是一种非常有用的技术，在文件的搜索以及杀毒软件中都使用了这种技术。以下我将讨论如何在win32下实现这种技术。

文件夹遍历技术的核心就是使用递归算法，关于递归算法，我这里就不多介绍了，不明白的朋友请找出谭先生的《c程序设计》参阅相关内容。

以下是我的算法伪代码：

void function( lpctstr lpszpath )
{
    开始查找;
    if ( 没有找到文件 )
        return;
    do
    {
        if ( 找到的文件是目录 )
            function( 找到的目录 );
        else
            对文件进行操作;
    } while ( 查找下一个文件并成功 );
}

实现这个算法所需要用到的api函数以及结构体有：

findfirstfile；
findnextfile；
win32_find_data。

在此我假定你已经明白了以上函数及结构体的用法，now let's begin。

现在我来编写开始查找的代码。在这之前，我先假定函数参数lpszpath传入的路径格式为x:（根目录）或x:dir（非根目录），因为win32程序设计中通常使用的就是这种路径格式。你一定注意到了，如果路径是根目录，它的后边有一个路径分隔符“”，反之则没有。那么我在写代码的时候必须对这两种情况分别处理。这段代码如下：

tchar szfind[max_path];
lstrcpy( szfind, lpszpath );
if ( !isroot( szfind ) ) // isroot是我自己编写的函数，若参数是根目录，则返回true
    lstrcat( szfind, "\" );
lstrcat( szfind, "*.*" ); // 找所有文件
win32_find_data wfd;
handle hfind = findfirstfile( szfind, &wfd );
if ( invalid_handle_value == hfind ) // 如果没有找到或查找失败
    return;

下面我将讨论的是如果找到了文件，该怎么办。不过在此之前，请你进入ms-dos方式，并输入dir回车，你看到了什么？

是的，dos是不会说假话的，不像windows一样总把重要的东西隐藏起来不让你看到——如果你所在的不是根目录，你将会看到“.”与“..”这两个目录——这是在资源管理器中看不到的。从dos时代走过的朋友们大抵都明白吧，一个点代表的是当前目录，两个点代表的是上一级目录。那么我在处理信息时，就一定得把它们两个过滤出去，原因我在下面解释。do-while段的代码如下：

do
{
    if ( lstrcmp( wfd.cfilename, _t(".") ) == 0 || lstrcmp ( wfd.cfilename, _t("..") ) == 0 )
        continue; // 过滤这两个目录
    if ( wfd.dwfileattributes & file_attribute_directory )
    {
        tchar szfile[max_path];
        if ( isroot( lpszpath ) )
            wsprintf( szfile, "%s%s", lpszpath, wfd.cfilename );
        else
            wsprintf( szfile, "%s\%s", lpszpath, wfd.cfilename );
        function( szfile ); // 如果找到的是目录，则进入此目录进行递归
    }
    else
    {
        // 对文件进行操作
    }
} while ( findnextfile( hfind, &wfd ) );
findclose( hfind ); // 关闭查找句柄

现在我来解释为什么要把那两个带点的目录过滤出去。如你所见，如果找到的是目录，那么进入此目录进行递归——那么若是当前目录呢？答案很明显，如果不对其进行过滤，那么程序将进入“当前目录”进行递归。是了，如是将导致递归无休止地进行下去。

算法就这么多了，由于这是一个极其耗费系统资源的算法，因此你在程序中用到它的时候，最好将其放到一个单独的线程中运行，否则将会导致你的程序在查找过程中没有任何响应。

如果你还有更好的算法，请贴在下面我们一起讨论。

附：isroot函数源码

bool isroot( lpctstr lpszpath )
{
    tchar szroot[4];
    wsprintf( szroot, "%c:\", lpszpath[0] );
    return ( lstrcmp( szroot, lpszpath ) == 0 );
}

用递归写Win32的文件夹遍历

2009年05月24日 • 企业安全 •我要评论

相关文章:

发表评论


验证码：