一个C++类实现文件全盘搜索

  #include <windows.h> 
#include <shlwapi.h> 
#include <iostream> 
 
#pragma comment(lib, "shlwapi.lib") 
 
using  namespace std; 
 
//定义过滤器的最大数量 
#define CONST_MAX_FILTER 16 
//链表的数据结构 
typedef  struct tagList{ 
  TCHAR szFile[MAX_PATH]; 
   struct tagList *NextFile; 
}FileList, *PFileList; 
//主体类 
class CHunter{ 
public: 
  CHunter(); 
  ~CHunter(); 
 
   void AddFilter( TCHAR *szFilter ); //添加过滤器 
   void CHunter::Hunt( TCHAR *szPath ); 
  TCHAR *GetFile(); //取得链表中的文件 
  DWORD    GetFileCount(); //取得文件的数量 
 
private: 
  PFileList headNode; //链表头 
  PFileList currNode; 
 
   void AddFile( TCHAR *szFile ); 
   void HuntFile(  char *lpPath ) ; 
 
  TCHAR    szFilter[CONST_MAX_FILTER][5] ; 
  DWORD    dwFilterCount ; 
  DWORD    dwFileCount ; 
}; 
 
CHunter::CHunter():dwFilterCount(0),dwFileCount(0) 
{ 
  headNode = (FileList *)malloc(  sizeof(FileList) ); 
  headNode->NextFile = NULL; 
  currNode = headNode; 
   for( int i=0; i< CONST_MAX_FILTER; i++) 
    ZeroMemory( szFilter[i], 5 ) ; 
} 
 
CHunter::~CHunter() 
{ 
  PFileList next, tmp; 
  tmp = headNode; 
 
   while( tmp->NextFile != NULL ) 
  { 
    next = tmp->NextFile ; 
    free(tmp); 
    tmp = next; 
  } 
  free(tmp); 
} 
 
//添加过滤器，比如.txt，为了简省，没有考虑文件扩展名长度大于4的情况 
//请自行修改 
void CHunter::AddFilter( TCHAR *szInp ) 
{ 
   if( strlen(szInp) > 4 ) 
     return; 
  strncpy( szFilter[dwFilterCount++], szInp, 5 ); 
} 
 
void CHunter::AddFile( TCHAR *szFile ) 
{ 
  currNode->NextFile = (FileList *)malloc(  sizeof(FileList) ); 
  currNode = currNode->NextFile;    
  ZeroMemory(currNode->szFile, MAX_PATH ); 
  currNode->NextFile = NULL ; 
  strncpy( currNode->szFile, szFile, MAX_PATH ); 
  dwFileCount++; 
} 
 
//这是入口函数，调用它即可开始搜索，这个路径的尾部不应有反斜杠 
void CHunter::Hunt( TCHAR *szPath ) 
{ 
  HuntFile( szPath ); 
  currNode = headNode->NextFile; 
} 
 
DWORD CHunter::GetFileCount() 
{ 
   return  this->dwFileCount; 
} 
 
//这个函数依次遍历链表中的文件,并返回一个文件名 
TCHAR *CHunter::GetFile() 
{ 
  TCHAR *szRet; 
  szRet = currNode->szFile; 
  currNode = currNode->NextFile; 
 
   return szRet; 
} 
 
void CHunter::HuntFile( char * lpPath) 
{ 
         char szFind[MAX_PATH]; 
   char szFile[MAX_PATH]; 
        WIN32_FIND_DATA FindFileData; 
 
  ZeroMemory(szFind,MAX_PATH); 
        ZeroMemory(szFile,MAX_PATH); 
  strcpy(szFind,lpPath); 
        strcat(szFind, "\\*.*"); 
 
        HANDLE hFind=::FindFirstFile(szFind,&FindFileData); 
         if(INVALID_HANDLE_VALUE == hFind)         return; 
         
         while(TRUE) 
        { 
                 if(FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) 
                { 
                         if(FindFileData.cFileName[0]!='.') 
                        { 
                                strcpy(szFile,lpPath); 
                                strcat(szFile,"\\"); 
                                strcat(szFile,FindFileData.cFileName); 
                                HuntFile(szFile); 
                        } 
                } 
                 else 
                { 
      strcpy(szFile,lpPath); 
                        strcat(szFile,"\\"); 
      strcat(szFile,FindFileData.cFileName); 
                         //cout << szFile << FindFileData.cFileName << endl; 
       for(  int i=0; i< dwFilterCount; i++ ) 
         if( strncmp(szFilter[i], PathFindExtension(szFile), 5) == 0 ) 
           this->AddFile( szFile ); 
                } 
                 if(!FindNextFile(hFind,&FindFileData))         break; 
        } 
        FindClose(hFind); 
} 
 
//示例,搜索D盘所有的exe文件 
int main( int argc,  char* argv[]) 
{ 
        CHunter hunter; 
  hunter.AddFilter( ".exe"); 
  hunter.Hunt( "D:"); 
 
  cout<< hunter.GetFileCount() << endl; 
   for( int i=0; i< hunter.GetFileCount(); i++) 
    cout<< hunter.GetFile() << endl; 
   return 0; 
}