compat/win32/dirent.c

   1 #include "../../git-compat-util.h"
   2
   3 struct DIR {
   4         struct dirent dd_dir; /* includes d_type */
   5         HANDLE dd_handle;     /* FindFirstFile handle */
   6         int dd_stat;          /* 0-based index */
   7 };
   8
   9 static inline void finddata2dirent(struct dirent *ent, WIN32_FIND_DATAW *fdata)
  10 {
  11         /* convert UTF-16 name to UTF-8 */
  12         xwcstoutf(ent->d_name, fdata->cFileName, sizeof(ent->d_name));
  13
  14         /* Set file type, based on WIN32_FIND_DATA */
  15         if (fdata->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
  16                 ent->d_type = DT_DIR;
  17         else
  18                 ent->d_type = DT_REG;
  19 }
  20
  21 struct dirent *readdir(DIR *dir)
  22 {
  23         if (!dir) {
  24                 errno = EBADF; /* No set_errno for mingw */
  25                 return NULL;
  26         }
  27
  28         /* if first entry, dirent has already been set up by opendir */
  29         if (dir->dd_stat) {
  30                 /* get next entry and convert from WIN32_FIND_DATA to dirent */
  31                 WIN32_FIND_DATAW fdata;
  32                 if (FindNextFileW(dir->dd_handle, &fdata)) {
  33                         finddata2dirent(&dir->dd_dir, &fdata);
  34                 } else {
  35                         DWORD lasterr = GetLastError();
  36                         /* POSIX says you shouldn't set errno when readdir can't
  37                            find any more files; so, if another error we leave it set. */
  38                         if (lasterr != ERROR_NO_MORE_FILES)
  39                                 errno = err_win_to_posix(lasterr);
  40                         return NULL;
  41                 }
  42         }
  43
  44         ++dir->dd_stat;
  45         return &dir->dd_dir;
  46 }
  47
  48 int closedir(DIR *dir)
  49 {
  50         if (!dir) {
  51                 errno = EBADF;
  52                 return -1;
  53         }
  54
  55         FindClose(dir->dd_handle);
  56         free(dir);
  57         return 0;
  58 }
  59
  60 DIR *opendir(const char *name)
  61 {
  62         wchar_t pattern[MAX_PATH + 2]; /* + 2 for '/' '*' */
  63         WIN32_FIND_DATAW fdata;
  64         HANDLE h;
  65         int len;
  66         DIR *dir;
  67
  68         /* convert name to UTF-16 and check length < MAX_PATH */
  69         if ((len = xutftowcs_path(pattern, name)) < 0)
  70                 return NULL;
  71
  72         /* append optional '/' and wildcard '*' */
  73         if (len && !is_dir_sep(pattern[len - 1]))
  74                 pattern[len++] = '/';
  75         pattern[len++] = '*';
  76         pattern[len] = 0;
  77
  78         /* open find handle */
  79         h = FindFirstFileW(pattern, &fdata);
  80         if (h == INVALID_HANDLE_VALUE) {
  81                 DWORD err = GetLastError();
  82                 errno = (err == ERROR_DIRECTORY) ? ENOTDIR : err_win_to_posix(err);
  83                 return NULL;
  84         }
  85
  86         /* initialize DIR structure and copy first dir entry */
  87         dir = xmalloc(sizeof(DIR));
  88         dir->dd_handle = h;
  89         dir->dd_stat = 0;
  90         finddata2dirent(&dir->dd_dir, &fdata);
  91         return dir;
  92 }