遍历(广度优先)windows系统某一目录

遍历目录的类(广度优先)，本来想找一段代码了事，后来没发现合适的，自己写了一个，分享给大家。其中用到一个线程类，也是自己封装的，与主题无关不贴出来，有需要的话去我的资源里找吧。

directory_traversal.h文件

#ifndef DIRECTORY_TRAVERSAL_H__
#define DIRECTORY_TRAVERSAL_H__
#include <windows.h>
#include <queue>
#include "hthread.h"

struct FileInfo{
  FileInfo();
  FileInfo(const FileInfo&);
  FileInfo& operator=(const FileInfo&);
  WIN32_FIND_DATAW   file_data;
  wchar_t           path[MAX_PATH];
  size_t            level;
};

int MakePath(wchar_t* file_path, size_t size, const wchar_t* parent_path, const wchar_t* file_name);
FileInfo MakeFileInfo(const wchar_t* parent_path, size_t level, const WIN32_FIND_DATA& cur_fd);


class DirectoryTraversal
{
public:
  typedef bool (*callback_find)(const FileInfo* , void*);
  typedef void (*callback_complete)(bool , void*);
  DirectoryTraversal();
  virtual ~DirectoryTraversal();

  // 设置遍历参数
  void SetCallback(callback_find cb_find, void* p1, callback_complete cb_complete, void* p2);
  // 启动遍历线程
  int Start(const wchar_t* begin_path, size_t max_level);

  int BreadthFirstTravers(const wchar_t* begin_path, size_t max_level);

  void SetStopFlag();
private:
  static unsigned int __stdcall ThreadTravers(void* lp);
  // 用递归方法，遍历d:\\w导致栈溢出
  //int BreadthFirstTravers(const wchar_t* path, size_t level);

  /*
  如果cb_find返回bool型, 一般应返回true, 如果返回false且当前文件是目录, 则不遍历此目录下的文件
  p作为cb_find的参数.
  */
  

  HANDLE FindFirstSubFile(const wchar_t* path, WIN32_FIND_DATA* fd);

protected:
  wchar_t       begin_path_[MAX_PATH];
  size_t        max_level_; // 遍历级数
  bool          stop_;
  callback_find cb_find_;
  void*         find_para_;
  callback_complete cb_complete_;
  void*         complete_para_;
  HThread       thread_;

  std::queue<FileInfo> folder_queue_;
};

#endif

directory_traversal.cpp文件

#include <Shlwapi.h>
#include <string.h>
#include <assert.h>
#include <wchar.h>
#include "directory_traversal.h"

#pragma comment(lib,"shlwapi.lib")

FileInfo::FileInfo()
{
  level = 0;
  memset(path, 0, sizeof(path));
}

FileInfo::FileInfo(const FileInfo& r)
{
  this->file_data = r.file_data;
  wcscpy_s(this->path, MAX_PATH, r.path);
  this->level = r.level;
}

FileInfo& FileInfo::operator=(const FileInfo& r)
{
  if(&r != this)
  {
    this->file_data = r.file_data;
    wcscpy_s(this->path, MAX_PATH, r.path);
    this->level = r.level;
  }
  return *this;
}

int MakePath(wchar_t* file_path, size_t size, const wchar_t* parent_path, const wchar_t* file_name)
{
  int ret = wcscpy_s(file_path, size, parent_path);
  if (0 == ret)
    ret = PathAppendW(file_path, file_name) ? 0 : -1;
  return ret;
}

FileInfo MakeFileInfo(const wchar_t* parent_path, size_t level, const WIN32_FIND_DATAW& cur_fd)
{
  FileInfo cur_folder; 
  MakePath(cur_folder.path, MAX_PATH, parent_path, cur_fd.cFileName);
  cur_folder.file_data = cur_fd;
  cur_folder.level = level;
  return cur_folder;
}

DirectoryTraversal::DirectoryTraversal() : stop_(false)
{}

DirectoryTraversal::~DirectoryTraversal()
{
  thread_.WaitExit();
}

void DirectoryTraversal::SetCallback(callback_find cb_find, void* p1, callback_complete cb_complete, void* p2)
{
  cb_find_ = cb_find;
  find_para_ = p1;

  cb_complete_ = cb_complete;
  complete_para_ = p2;
}

int DirectoryTraversal::Start(const wchar_t* begin_path, size_t max_level)
{
  stop_ = false;
  wcscpy_s(begin_path_, MAX_PATH, begin_path);  // 线程启动前要保存传入的数据
  max_level_ = max_level;
  thread_.Start(&DirectoryTraversal::ThreadTravers, this);
  return 0;
}

unsigned int __stdcall DirectoryTraversal::ThreadTravers(void* lp)
{
  DirectoryTraversal* self = (DirectoryTraversal*)lp;
  if(wcslen(self->begin_path_) == 0)
    return -1;
  return self->BreadthFirstTravers(self->begin_path_, self->max_level_);
}

int DirectoryTraversal::BreadthFirstTravers(const wchar_t* begin_path, size_t max_level)
{
  while(!folder_queue_.empty())
    folder_queue_.pop();

  HANDLE find_file = INVALID_HANDLE_VALUE;
    WIN32_FIND_DATAW fd;

  wchar_t cur_path[MAX_PATH];
  wcscpy_s(cur_path, MAX_PATH, begin_path);
  size_t cur_level = 1;
  do {
    if (max_level > 0 && cur_level > max_level) {
      if (cb_complete_)
        (*cb_complete_)(stop_, complete_para_);
      return 0;
	  }

    find_file = FindFirstSubFile(cur_path, &fd);
    if (INVALID_HANDLE_VALUE == find_file) 
    {
      assert(false);
      if (cb_complete_)
        (*cb_complete_)(stop_, complete_para_);
      return -1;
    }

    FileInfo new_find_file;
    do {      
      if (!(fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
        new_find_file = MakeFileInfo(cur_path, cur_level, fd);
        if(cb_find_)
          (*cb_find_)(&new_find_file, find_para_);
      }
      else {
        if(!(fd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) && 
          0 != wcscmp(fd.cFileName, L".") && 0 != wcscmp(fd.cFileName, L"..")) {
            new_find_file = MakeFileInfo(cur_path, cur_level, fd);
            if(cb_find_){
              if((*cb_find_)(&new_find_file, find_para_))
                folder_queue_.push(new_find_file);
            }
            else
              folder_queue_.push(new_find_file);
        }
      }  
	  if (stop_) break;    
    } while(FindNextFileW(find_file, &fd));

    FindClose(find_file);

    if(stop_ || folder_queue_.empty()) {
      if (cb_complete_)
        (*cb_complete_)(stop_, complete_para_);
      return 0;
	  }

    FileInfo next_folder = folder_queue_.front();
    folder_queue_.pop();

    cur_level = next_folder.level + 1; 

    wcscpy_s(cur_path, MAX_PATH, next_folder.path);
  } while(true);
}

void DirectoryTraversal::SetStopFlag()
{
  stop_ = true;
}

  HANDLE DirectoryTraversal::FindFirstSubFile(const wchar_t* path, WIN32_FIND_DATAW* fd)
  {
    wchar_t buffer[MAX_PATH];
    wcscpy_s(buffer, MAX_PATH, path);
    if(!PathAppendW(buffer, L"*.*"))
      return NULL;
    return FindFirstFileW(buffer, fd);    
  }