IO.hpp

   1 #ifndef __DFTPD__IO_HPP__
   2 #define __DFTPD__IO_HPP__
   3
   4 #ifdef _WIN32
   5 #include <windows.h>
   6 #define S_ISDIR(mode) (((mode) & _S_IFMT) == (_S_IFDIR))
   7 #define S_ISREG(mode) (((mode) & _S_IFMT) == (_S_IFREG))
   8 #define localtime_r(a,b) localtime_s(b,a)
   9 #else
  10 #include <dirent.h>
  11 #endif
  12
  13 #include <string>
  14
  15 class IO
  16 {
  17 public:
  18         static bool MkDir( const std::string& dir );
  19         static bool RmDir( const std::string& dir );
  20
  21 private:
  22         IO() {};
  23 };
  24
  25 class Directory
  26 {
  27 public:
  28         Directory();
  29         ~Directory();
  30
  31         bool Open( const std::string& dir );
  32
  33         Directory& operator++();
  34         operator bool();
  35
  36         std::string GetName();
  37
  38 private:
  39 #ifdef _WIN32
  40         HANDLE m_handle;
  41         WIN32_FIND_DATA m_ffd;
  42         std::string m_path;
  43 #else
  44         DIR* m_dir;
  45         dirent* m_dirent;
  46 #endif
  47 };
  48
  49 #endif