compat/win32/dirent.c

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