어쩌다보니

// win32fileiterator.h 
//////////////////////////////////////////////////////////////////////////

#ifndef __WIN32_FILE_ITERATOR_H_F3019690_E6A6_4997_AB83_E0FDDC109DE2__
#define __WIN32_FILE_ITERATOR_H_F3019690_E6A6_4997_AB83_E0FDDC109DE2__

#ifndef _WIN32
#error 32-bit Windows required.
#endif

#include <windows.h>

#include <iterator>
#include <string>

//////////////////////////////////////////////////////////////////////////
// initial class : codeproject.com

// (http://www.codeproject.com/file/win32fileiter.asp)
// updated by j2doll
//  v0.1 (10, May, 2005) : unicode / mbcs stl support 
//                       it's tested on windows xp sp2 / visual c++ 2005 sp1  
//
//////////////////////////////////////////////////////////////////////////

class win32_file_iterator : public std::iterator<std::input_iterator_tag,
#ifdef _UNICODE
  std::wstring>
#else
  std::string>
#endif
{

private:
  class internal_handle_data{
  public:
    internal_handle_data():_h(NULL), _ref(0){}
    void setHandle(HANDLE handle){ _h = handle; }
    HANDLE getHandle(){ return _h; }
    void incRef(){ _ref++; }
    unsigned decRef(){ return --_ref; }
    operator HANDLE(){ return _h; }
  private:
    HANDLE _h;
    unsigned _ref;
  };

public:
  win32_file_iterator(
#ifdef _UNICODE
    std::wstring strfilter,
#else
    std::string strfilter,
#endif
    bool bFullPath = false,
    DWORD flag = FILE_ATTRIBUTE_ARCHIVE|FILE_ATTRIBUTE_NORMAL|FILE_ATTRIBUTE_DIRECTORY):
    _bEnd(false), _bFullPath(bFullPath), _flag(flag){
        HANDLE h = ::FindFirstFile(strfilter.c_str(), &_wfd);
    _handle.setHandle(h);

    if(h == INVALID_HANDLE_VALUE){
      _bEnd = true;
    }else{
      _handle.incRef();
#ifdef _UNICODE
      std::wstring::size_type n1 = strfilter.find_last_of(L"\\");
#else
      std::string::size_type n1 = strfilter.find_last_of("\\");
#endif
      _strroot = strfilter.substr(0,n1+1);
      _chkvalid(_wfd);
    }
  }

  win32_file_iterator():_bEnd(true){}

  win32_file_iterator(win32_file_iterator& rhs){
   
    _handle = rhs._handle;
    _handle.incRef();

    _flag = rhs._flag;
    _bFullPath = rhs._bFullPath;
    _bEnd = rhs._bEnd;
    _wfd = rhs._wfd;
    _strfname = rhs._strfname;
    _strroot = rhs._strroot;   
  }

  ~win32_file_iterator(){
    if(_handle.decRef() == 0 && _handle.getHandle() != NULL ){
      FindClose(_handle);
    }
  }

  reference operator*(){
    return _strfname;
  }

  bool operator==(const win32_file_iterator& rhs) const{
    return (_bEnd == rhs._bEnd);
  }
  bool operator!=(const win32_file_iterator& rhs) const{
    return (_bEnd != rhs._bEnd);
  }

  //Prefix
  win32_file_iterator& operator++(){
    _findnext();
    return *this;
  }
  //Posfix
  win32_file_iterator& operator++(int){
    _findnext();
    return *this;
  }

private:
  void _findnext(){
    BOOL b = ::FindNextFile(_handle, &_wfd);
    if(b){
      _chkvalid(_wfd);
    }else{
      _bEnd = true;
    }
  }

  void _chkvalid(WIN32_FIND_DATA& _wfd){
    if(_wfd.dwFileAttributes & _flag){
      _getval(_wfd);
    }
    else{
      _findnext();
    }
  }
  void _getval(WIN32_FIND_DATA& wfd){
    if(_bFullPath)
      _strfname = _strroot+ wfd.cFileName;
    else
      _strfname = wfd.cFileName;
  }

private:
  DWORD _flag;
  bool _bFullPath;
  bool _bEnd;
  internal_handle_data _handle;
  WIN32_FIND_DATA _wfd;
#ifdef _UNICODE
  std::wstring _strroot;
  std::wstring _strfname;
#else
  std::string _strroot;
  std::string _strfname;
#endif

};

#endif

 

// test function

void test1()
{

#ifdef _UNICODE

  std::wstring pathstr = L"C:\\*.*";
  // DWORD fileopt = FILE_ATTRIBUTE_ARCHIVE|FILE_ATTRIBUTE_NORMAL; 
  DWORD fileopt = FILE_ATTRIBUTE_DIRECTORY;  // only directory 
  win32_file_iterator itBegin(pathstr, false, fileopt), itEnd;
  std::vector<std::wstring> vec(itBegin, itEnd);

  std::wstring strtemp;
  std::vector<std::wstring>::size_type i;
  for ( i = 0 ; i < vec.size(); i ++ )
  {
    strtemp = vec.at( i );

    wprintf( L"%s\r\n", strtemp.c_str() );
  }

#else

  std::string pathstr = "C:\\*.*";
  // DWORD fileopt = FILE_ATTRIBUTE_ARCHIVE|FILE_ATTRIBUTE_NORMAL;
  DWORD fileopt = FILE_ATTRIBUTE_DIRECTORY; // only directory 
  win32_file_iterator itBegin(pathstr, false, fileopt), itEnd; 
  std::vector<std::string> vec(itBegin, itEnd);

  std::string strtemp;
  std::vector<std::string>::size_type i;
  for ( i = 0 ; i < vec.size(); i ++ )
  {
    strtemp = vec.at( i );

    printf( "[DIR]%s\r\n", strtemp.c_str() ); 
  }

#endif
}