compat/win32/dirent.c

   1 #include "../git-compat-util.h"
   2 #include "dirent.h"
   3
   4 struct DIR {
   5         struct dirent dd_dir; /* includes d_type */
   6         HANDLE dd_handle;     /* FindFirstFile handle */
   7         int dd_stat;          /* 0-based index */
   8         char dd_name[1];      /* extend struct */
   9 };
  10
  11 DIR *opendir(const char *name)
  12 {
  13         DWORD attrs = GetFileAttributesA(name);
  14         int len;
  15         DIR *p;
  16
  17         /* check for valid path */
  18         if (attrs == INVALID_FILE_ATTRIBUTES) {
  19                 errno = ENOENT;
  20                 return NULL;
  21         }
  22
  23         /* check if it's a directory */
  24         if (!(attrs & FILE_ATTRIBUTE_DIRECTORY)) {
  25                 errno = ENOTDIR;
  26                 return NULL;
  27         }
  28
  29         /* check that the pattern won't be too long for FindFirstFileA */
  30         len = strlen(name);
  31         if (is_dir_sep(name[len - 1]))
  32                 len--;
  33         if (len + 2 >= MAX_PATH) {
  34                 errno = ENAMETOOLONG;
  35                 return NULL;
  36         }
  37
  38         p = malloc(sizeof(DIR) + len + 2);
  39         if (!p)
  40                 return NULL;
  41
  42         memset(p, 0, sizeof(DIR) + len + 2);
  43         strcpy(p->dd_name, name);
  44         p->dd_name[len] = '/';
  45         p->dd_name[len+1] = '*';
  46
  47         p->dd_handle = INVALID_HANDLE_VALUE;
  48         return p;
  49 }
  50
  51 struct dirent *readdir(DIR *dir)
  52 {
  53         WIN32_FIND_DATAA buf;
  54         HANDLE handle;
  55
  56         if (!dir || !dir->dd_handle) {
  57                 errno = EBADF; /* No set_errno for mingw */
  58                 return NULL;
  59         }
  60
  61         if (dir->dd_handle == INVALID_HANDLE_VALUE && dir->dd_stat == 0) {
  62                 DWORD lasterr;
  63                 handle = FindFirstFileA(dir->dd_name, &buf);
  64                 lasterr = GetLastError();
  65                 dir->dd_handle = handle;
  66                 if (handle == INVALID_HANDLE_VALUE && (lasterr != ERROR_NO_MORE_FILES)) {
  67                         errno = err_win_to_posix(lasterr);
  68                         return NULL;
  69                 }
  70         } else if (dir->dd_handle == INVALID_HANDLE_VALUE) {
  71                 return NULL;
  72         } else if (!FindNextFileA(dir->dd_handle, &buf)) {
  73                 DWORD lasterr = GetLastError();
  74                 FindClose(dir->dd_handle);
  75                 dir->dd_handle = INVALID_HANDLE_VALUE;
  76                 /* POSIX says you shouldn't set errno when readdir can't
  77                    find any more files; so, if another error we leave it set. */
  78                 if (lasterr != ERROR_NO_MORE_FILES)
  79                         errno = err_win_to_posix(lasterr);
  80                 return NULL;
  81         }
  82
  83         /* We get here if `buf' contains valid data.  */
  84         strcpy(dir->dd_dir.d_name, buf.cFileName);
  85         ++dir->dd_stat;
  86
  87         /* Set file type, based on WIN32_FIND_DATA */
  88         dir->dd_dir.d_type = 0;
  89         if (buf.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
  90                 dir->dd_dir.d_type |= DT_DIR;
  91         else
  92                 dir->dd_dir.d_type |= DT_REG;
  93
  94         return &dir->dd_dir;
  95 }
  96
  97 int closedir(DIR *dir)
  98 {
  99         if (!dir) {
 100                 errno = EBADF;
 101                 return -1;
 102         }
 103
 104         if (dir->dd_handle != INVALID_HANDLE_VALUE)
 105                 FindClose(dir->dd_handle);
 106         free(dir);
 107         return 0;
 108 }