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

VC++实现文件全盘搜索
很久没更新了，今天决定写点东西。
这是我以前写的一个C++的类，可以在硬盘上全盘搜索指定的文件（可以用通配符），还可以添加过滤器，以便搜索特定的文件。该类使用链表储存搜索结果（学了那么多数据结构，总算可以用一把了），性能还是可以的。虽说没什么技术含量，但也挺有用的
下面就是这个类的代码，包括测试程序，原本我没有写注释，现在加上了一些。

这个博客居然不支持C++的代码插入，只好用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;
}

转载于:https://blog.51cto.com/spinlock/170515