如何获取具有特定扩展名的驱动器中所有文件的列表?

时间:2017-10-11 00:09:10

标签: c++ winapi path-finding

大家好,我是新手。经过长时间无望的搜索,我想发布这个问题:

我正在开发一个C ++程序,用于处理搜索特定文件类型的所有文件夹和子文件夹。

首先,函数FindFiles(string, string, bool)的工作方式与魅力相似,但第二种形式FindFiles(struct var)不能正常工作:它不会遍历所有文件夹和子文件夹。< / p>

实际上我需要第二种形式,因为搜索可能太长,我需要创建一个包含API CreateThread的线程,并将arguments struct作为LPVOID传递给它}。

#include <windows.h>
#include <string>
#include <iostream>
using namespace std;



// Arguments struct
struct args{
    string strDir;
    string strFilter;
    bool bRecurse;
};


void FindFiles(string strDir, string strFilter, bool bRecurse);
void FindFiles(args);


int main(){

//  FindFiles("D:", "*.mp3", true); // works fine

    args ar;
    ar.bRecurse  = true;
    ar.strDir    = "D:";
    ar.strFilter = "*.mp3";

    FindFiles(ar); // doesn't work fine

    cout << endl;
    return 0;
}

void FindFiles(string strDir, string strFilter, bool bRecurse = true){
    if(bRecurse)
        FindFiles(strDir, strFilter, false);
    strDir += "\\";
    WIN32_FIND_DATA wfd;

    string strFileFilter = strDir + (bRecurse ? "*" : strFilter);
    HANDLE hFile = FindFirstFile(strFileFilter.c_str(), &wfd);

    if(INVALID_HANDLE_VALUE == hFile)
        return;
    else{
        if(!bRecurse)
            cout << strDir + string(wfd.cFileName) << endl;
        while(FindNextFile(hFile, &wfd) ){
            if(!bRecurse)
                cout << strDir + string(wfd.cFileName) << endl;
            else{
                if( (wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) > 0 &&
                    wfd.cFileName[0] != '.')
                    FindFiles(strDir + string(wfd.cFileName), strFilter, true);
            }   
        }
    //  FindClose(hFile);
    }
}

void FindFiles(args ar){
    if(ar.bRecurse){
        ar.bRecurse = false;
        FindFiles(ar);
    }
    ar.strDir += "\\";
    WIN32_FIND_DATA wfd;

    string strFileFilter = ar.strDir + (ar.bRecurse ? "*" : ar.strFilter);
    HANDLE hFile = FindFirstFile(strFileFilter.c_str(), &wfd);

    if(INVALID_HANDLE_VALUE == hFile)
        return;
    else{
        if(!ar.bRecurse)
            cout << ar.strDir + string(wfd.cFileName) << endl;
        while(FindNextFile(hFile, &wfd) ){
            if(!ar.bRecurse)
                cout << ar.strDir + string(wfd.cFileName) << endl;
            else{
                if( (wfd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) > 0 &&
                    wfd.cFileName[0] != '.'){
                        ar.strDir += string(wfd.cFileName);
                        ar.bRecurse = true;
                        FindFiles(ar);
                }
            }   
        }
        FindClose(hFile);
    }
}

请看第二张表格,因为它是我的。我认为那里有一些愚蠢的错误。

我真的非常感谢任何帮助。

1 个答案:

答案 0 :(得分:4)

{1}的1-param版本没有递归,因为它正在修改FindFiles()标志,因此在搜索子文件夹时它始终为false:

ar.bRecurse

void FindFiles(args ar){ if(ar.bRecurse){ ar.bRecurse = false; // <-- modified! FindFiles(ar); } ... // ar.bRecurse is still false here! } 的3-params版本没有这样做:

FindFiles()

最简单的解决方案是在初始void FindFiles(string strDir, string strFilter, bool bRecurse = true){ if(bRecurse) FindFiles(strDir, strFilter, false); // <-- bRecurse is not modified here! ... // bRecurse is still the original value here! } 电话后将ar.bRecurse重置为true:

FindFiles()

或者为该调用使用临时结构:

void FindFiles(args ar){
    if(ar.bRecurse){
        ar.bRecurse = false; // <-- modified!
        FindFiles(ar);
        ar.bRecurse = true; // <-- reset here!
    }
    ...
}

现在,说到这一点,你根本不需要在函数顶部进行void FindFiles(args ar){ if(ar.bRecurse){ args arTmp = ar; // <-- temp copy! arTmp.bRecurse = false; // <-- modify the copy! FindFiles(arTmp); } ... // ar.bRecurse is still the original value here! } 检查。所有它真正做的是扫描输入文件夹以仅匹配文件,然后再次扫描文件夹以仅查找子文件夹。您可以通过使用单个搜索并在本地缓存子文件夹进行下一次搜索来完成相同的逻辑,例如:

bRecurse