c++ 查找d:盘 中.exe的文件

 

 

#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; 
}