1 /*
2  * Dirent interface for Microsoft Visual Studio
3  *
4  * Copyright (C) 1998-2019 Toni Ronkko
5  * This file is part of dirent.  Dirent may be freely distributed
6  * under the MIT license.  For all details and documentation, see
7  * https://github.com/tronkko/dirent
8  */
9 #ifndef DIRENT_H
10 #define DIRENT_H
11 
12 /* Hide warnings about unreferenced local functions */
13 #if defined(__clang__)
14 #   pragma clang diagnostic ignored "-Wunused-function"
15 #elif defined(_MSC_VER)
16 #   pragma warning(disable:4505)
17 #elif defined(__GNUC__)
18 #   pragma GCC diagnostic ignored "-Wunused-function"
19 #endif
20 
21 /*
22  * Include windows.h without Windows Sockets 1.1 to prevent conflicts with
23  * Windows Sockets 2.0.
24  */
25 #ifndef WIN32_LEAN_AND_MEAN
26 #   define WIN32_LEAN_AND_MEAN
27 #endif
28 #include <windows.h>
29 
30 #include <stdio.h>
31 #include <stdarg.h>
32 #include <wchar.h>
33 #include <string.h>
34 #include <stdlib.h>
35 #include <malloc.h>
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 #include <errno.h>
39 #include <ctype.h>
40 
41 /* Indicates that d_type field is available in dirent structure */
42 #define _DIRENT_HAVE_D_TYPE
43 
44 /* Indicates that d_namlen field is available in dirent structure */
45 #define _DIRENT_HAVE_D_NAMLEN
46 
47 /* Entries missing from MSVC 6.0 */
48 #if !defined(FILE_ATTRIBUTE_DEVICE)
49 #   define FILE_ATTRIBUTE_DEVICE 0x40
50 #endif
51 
52 /* File type and permission flags for stat(), general mask */
53 #if !defined(S_IFMT)
54 #   define S_IFMT _S_IFMT
55 #endif
56 
57 /* Directory bit */
58 #if !defined(S_IFDIR)
59 #   define S_IFDIR _S_IFDIR
60 #endif
61 
62 /* Character device bit */
63 #if !defined(S_IFCHR)
64 #   define S_IFCHR _S_IFCHR
65 #endif
66 
67 /* Pipe bit */
68 #if !defined(S_IFFIFO)
69 #   define S_IFFIFO _S_IFFIFO
70 #endif
71 
72 /* Regular file bit */
73 #if !defined(S_IFREG)
74 #   define S_IFREG _S_IFREG
75 #endif
76 
77 /* Read permission */
78 #if !defined(S_IREAD)
79 #   define S_IREAD _S_IREAD
80 #endif
81 
82 /* Write permission */
83 #if !defined(S_IWRITE)
84 #   define S_IWRITE _S_IWRITE
85 #endif
86 
87 /* Execute permission */
88 #if !defined(S_IEXEC)
89 #   define S_IEXEC _S_IEXEC
90 #endif
91 
92 /* Pipe */
93 #if !defined(S_IFIFO)
94 #   define S_IFIFO _S_IFIFO
95 #endif
96 
97 /* Block device */
98 #if !defined(S_IFBLK)
99 #   define S_IFBLK 0
100 #endif
101 
102 /* Link */
103 #if !defined(S_IFLNK)
104 #   define S_IFLNK 0
105 #endif
106 
107 /* Socket */
108 #if !defined(S_IFSOCK)
109 #   define S_IFSOCK 0
110 #endif
111 
112 /* Read user permission */
113 #if !defined(S_IRUSR)
114 #   define S_IRUSR S_IREAD
115 #endif
116 
117 /* Write user permission */
118 #if !defined(S_IWUSR)
119 #   define S_IWUSR S_IWRITE
120 #endif
121 
122 /* Execute user permission */
123 #if !defined(S_IXUSR)
124 #   define S_IXUSR 0
125 #endif
126 
127 /* Read group permission */
128 #if !defined(S_IRGRP)
129 #   define S_IRGRP 0
130 #endif
131 
132 /* Write group permission */
133 #if !defined(S_IWGRP)
134 #   define S_IWGRP 0
135 #endif
136 
137 /* Execute group permission */
138 #if !defined(S_IXGRP)
139 #   define S_IXGRP 0
140 #endif
141 
142 /* Read others permission */
143 #if !defined(S_IROTH)
144 #   define S_IROTH 0
145 #endif
146 
147 /* Write others permission */
148 #if !defined(S_IWOTH)
149 #   define S_IWOTH 0
150 #endif
151 
152 /* Execute others permission */
153 #if !defined(S_IXOTH)
154 #   define S_IXOTH 0
155 #endif
156 
157 /* Maximum length of file name */
158 #if !defined(PATH_MAX)
159 #   define PATH_MAX MAX_PATH
160 #endif
161 #if !defined(FILENAME_MAX)
162 #   define FILENAME_MAX MAX_PATH
163 #endif
164 #if !defined(NAME_MAX)
165 #   define NAME_MAX FILENAME_MAX
166 #endif
167 
168 /* File type flags for d_type */
169 #define DT_UNKNOWN 0
170 #define DT_REG S_IFREG
171 #define DT_DIR S_IFDIR
172 #define DT_FIFO S_IFIFO
173 #define DT_SOCK S_IFSOCK
174 #define DT_CHR S_IFCHR
175 #define DT_BLK S_IFBLK
176 #define DT_LNK S_IFLNK
177 
178 /* Macros for converting between st_mode and d_type */
179 #define IFTODT(mode) ((mode) & S_IFMT)
180 #define DTTOIF(type) (type)
181 
182 /*
183  * File type macros.  Note that block devices, sockets and links cannot be
184  * distinguished on Windows and the macros S_ISBLK, S_ISSOCK and S_ISLNK are
185  * only defined for compatibility.  These macros should always return false
186  * on Windows.
187  */
188 #if !defined(S_ISFIFO)
189 #   define S_ISFIFO(mode) (((mode) & S_IFMT) == S_IFIFO)
190 #endif
191 #if !defined(S_ISDIR)
192 #   define S_ISDIR(mode) (((mode) & S_IFMT) == S_IFDIR)
193 #endif
194 #if !defined(S_ISREG)
195 #   define S_ISREG(mode) (((mode) & S_IFMT) == S_IFREG)
196 #endif
197 #if !defined(S_ISLNK)
198 #   define S_ISLNK(mode) (((mode) & S_IFMT) == S_IFLNK)
199 #endif
200 #if !defined(S_ISSOCK)
201 #   define S_ISSOCK(mode) (((mode) & S_IFMT) == S_IFSOCK)
202 #endif
203 #if !defined(S_ISCHR)
204 #   define S_ISCHR(mode) (((mode) & S_IFMT) == S_IFCHR)
205 #endif
206 #if !defined(S_ISBLK)
207 #   define S_ISBLK(mode) (((mode) & S_IFMT) == S_IFBLK)
208 #endif
209 
210 /* Return the exact length of the file name without zero terminator */
211 #define _D_EXACT_NAMLEN(p) ((p)->d_namlen)
212 
213 /* Return the maximum size of a file name */
214 #define _D_ALLOC_NAMLEN(p) ((PATH_MAX)+1)
215 
216 
217 #ifdef __cplusplus
218 extern "C" {
219 #endif
220 
221 
222 /* Wide-character version */
223 struct _wdirent {
224     /* Always zero */
225     long d_ino;
226 
227     /* File position within stream */
228     long d_off;
229 
230     /* Structure size */
231     unsigned short d_reclen;
232 
233     /* Length of name without \0 */
234     size_t d_namlen;
235 
236     /* File type */
237     int d_type;
238 
239     /* File name */
240     wchar_t d_name[PATH_MAX+1];
241 };
242 typedef struct _wdirent _wdirent;
243 
244 struct _WDIR {
245     /* Current directory entry */
246     struct _wdirent ent;
247 
248     /* Private file data */
249     WIN32_FIND_DATAW data;
250 
251     /* True if data is valid */
252     int cached;
253 
254     /* Win32 search handle */
255     HANDLE handle;
256 
257     /* Initial directory name */
258     wchar_t *patt;
259 };
260 typedef struct _WDIR _WDIR;
261 
262 /* Multi-byte character version */
263 struct dirent {
264     /* Always zero */
265     long d_ino;
266 
267     /* File position within stream */
268     long d_off;
269 
270     /* Structure size */
271     unsigned short d_reclen;
272 
273     /* Length of name without \0 */
274     size_t d_namlen;
275 
276     /* File type */
277     int d_type;
278 
279     /* File name */
280     char d_name[PATH_MAX+1];
281 };
282 typedef struct dirent dirent;
283 
284 struct DIR {
285     struct dirent ent;
286     struct _WDIR *wdirp;
287 };
288 typedef struct DIR DIR;
289 
290 
291 /* Dirent functions */
292 static DIR *opendir(const char *dirname);
293 static _WDIR *_wopendir(const wchar_t *dirname);
294 
295 static struct dirent *readdir(DIR *dirp);
296 static struct _wdirent *_wreaddir(_WDIR *dirp);
297 
298 static int readdir_r(
299     DIR *dirp, struct dirent *entry, struct dirent **result);
300 static int _wreaddir_r(
301     _WDIR *dirp, struct _wdirent *entry, struct _wdirent **result);
302 
303 static int closedir(DIR *dirp);
304 static int _wclosedir(_WDIR *dirp);
305 
306 static void rewinddir(DIR* dirp);
307 static void _wrewinddir(_WDIR* dirp);
308 
309 static int scandir(const char *dirname, struct dirent ***namelist,
310     int (*filter)(const struct dirent*),
311     int (*compare)(const struct dirent**, const struct dirent**));
312 
313 static int alphasort(const struct dirent **a, const struct dirent **b);
314 
315 static int versionsort(const struct dirent **a, const struct dirent **b);
316 
317 static int strverscmp(const char *a, const char *b);
318 
319 /* For compatibility with Symbian */
320 #define wdirent _wdirent
321 #define WDIR _WDIR
322 #define wopendir _wopendir
323 #define wreaddir _wreaddir
324 #define wclosedir _wclosedir
325 #define wrewinddir _wrewinddir
326 
327 /* Compatibility with older Microsoft compilers and non-Microsoft compilers */
328 #if !defined(_MSC_VER) || _MSC_VER < 1400
329 #   define wcstombs_s dirent_wcstombs_s
330 #   define mbstowcs_s dirent_mbstowcs_s
331 #endif
332 
333 /* Optimize dirent_set_errno() away on modern Microsoft compilers */
334 #if defined(_MSC_VER) && _MSC_VER >= 1400
335 #   define dirent_set_errno _set_errno
336 #endif
337 
338 
339 /* Internal utility functions */
340 static WIN32_FIND_DATAW *dirent_first(_WDIR *dirp);
341 static WIN32_FIND_DATAW *dirent_next(_WDIR *dirp);
342 
343 #if !defined(_MSC_VER) || _MSC_VER < 1400
344 static int dirent_mbstowcs_s(
345     size_t *pReturnValue, wchar_t *wcstr, size_t sizeInWords,
346     const char *mbstr, size_t count);
347 #endif
348 
349 #if !defined(_MSC_VER) || _MSC_VER < 1400
350 static int dirent_wcstombs_s(
351     size_t *pReturnValue, char *mbstr, size_t sizeInBytes,
352     const wchar_t *wcstr, size_t count);
353 #endif
354 
355 #if !defined(_MSC_VER) || _MSC_VER < 1400
356 static void dirent_set_errno(int error);
357 #endif
358 
359 
360 /*
361  * Open directory stream DIRNAME for read and return a pointer to the
362  * internal working area that is used to retrieve individual directory
363  * entries.
364  */
_wopendir(const wchar_t * dirname)365 static _WDIR *_wopendir(const wchar_t *dirname)
366 {
367     wchar_t *p;
368 
369     /* Must have directory name */
370     if (dirname == NULL || dirname[0] == '\0') {
371         dirent_set_errno(ENOENT);
372         return NULL;
373     }
374 
375     /* Allocate new _WDIR structure */
376     _WDIR *dirp = (_WDIR*) malloc(sizeof(struct _WDIR));
377     if (!dirp)
378         return NULL;
379 
380     /* Reset _WDIR structure */
381     dirp->handle = INVALID_HANDLE_VALUE;
382     dirp->patt = NULL;
383     dirp->cached = 0;
384 
385     /*
386      * Compute the length of full path plus zero terminator
387      *
388      * Note that on WinRT there's no way to convert relative paths
389      * into absolute paths, so just assume it is an absolute path.
390      */
391 #if WINAPI_FAMILY_PARTITION(WINAPI_PARTITION_DESKTOP)
392     /* Desktop */
393     DWORD n = GetFullPathNameW(dirname, 0, NULL, NULL);
394 #else
395     /* WinRT */
396     size_t n = wcslen(dirname);
397 #endif
398 
399     /* Allocate room for absolute directory name and search pattern */
400     dirp->patt = (wchar_t*) malloc(sizeof(wchar_t) * n + 16);
401     if (dirp->patt == NULL)
402         goto exit_closedir;
403 
404     /*
405      * Convert relative directory name to an absolute one.  This
406      * allows rewinddir() to function correctly even when current
407      * working directory is changed between opendir() and rewinddir().
408      *
409      * Note that on WinRT there's no way to convert relative paths
410      * into absolute paths, so just assume it is an absolute path.
411      */
412 #if WINAPI_FAMILY_PARTITION(WINAPI_PARTITION_DESKTOP)
413     /* Desktop */
414     n = GetFullPathNameW(dirname, n, dirp->patt, NULL);
415     if (n <= 0)
416         goto exit_closedir;
417 #else
418     /* WinRT */
419     wcsncpy_s(dirp->patt, n+1, dirname, n);
420 #endif
421 
422     /* Append search pattern \* to the directory name */
423     p = dirp->patt + n;
424     switch (p[-1]) {
425     case '\\':
426     case '/':
427     case ':':
428         /* Directory ends in path separator, e.g. c:\temp\ */
429         /*NOP*/;
430         break;
431 
432     default:
433         /* Directory name doesn't end in path separator */
434         *p++ = '\\';
435     }
436     *p++ = '*';
437     *p = '\0';
438 
439     /* Open directory stream and retrieve the first entry */
440     if (!dirent_first(dirp))
441         goto exit_closedir;
442 
443     /* Success */
444     return dirp;
445 
446     /* Failure */
447 exit_closedir:
448     _wclosedir(dirp);
449     return NULL;
450 }
451 
452 /*
453  * Read next directory entry.
454  *
455  * Returns pointer to static directory entry which may be overwritten by
456  * subsequent calls to _wreaddir().
457  */
_wreaddir(_WDIR * dirp)458 static struct _wdirent *_wreaddir(_WDIR *dirp)
459 {
460     /*
461      * Read directory entry to buffer.  We can safely ignore the return
462      * value as entry will be set to NULL in case of error.
463      */
464     struct _wdirent *entry;
465     (void) _wreaddir_r(dirp, &dirp->ent, &entry);
466 
467     /* Return pointer to statically allocated directory entry */
468     return entry;
469 }
470 
471 /*
472  * Read next directory entry.
473  *
474  * Returns zero on success.  If end of directory stream is reached, then sets
475  * result to NULL and returns zero.
476  */
_wreaddir_r(_WDIR * dirp,struct _wdirent * entry,struct _wdirent ** result)477 static int _wreaddir_r(
478     _WDIR *dirp, struct _wdirent *entry, struct _wdirent **result)
479 {
480     /* Read next directory entry */
481     WIN32_FIND_DATAW *datap = dirent_next(dirp);
482     if (!datap) {
483         /* Return NULL to indicate end of directory */
484         *result = NULL;
485         return /*OK*/0;
486     }
487 
488     /*
489      * Copy file name as wide-character string.  If the file name is too
490      * long to fit in to the destination buffer, then truncate file name
491      * to PATH_MAX characters and zero-terminate the buffer.
492      */
493     size_t n = 0;
494     while (n < PATH_MAX && datap->cFileName[n] != 0) {
495         entry->d_name[n] = datap->cFileName[n];
496         n++;
497     }
498     entry->d_name[n] = 0;
499 
500     /* Length of file name excluding zero terminator */
501     entry->d_namlen = n;
502 
503     /* File type */
504     DWORD attr = datap->dwFileAttributes;
505     if ((attr & FILE_ATTRIBUTE_DEVICE) != 0)
506         entry->d_type = DT_CHR;
507     else if ((attr & FILE_ATTRIBUTE_DIRECTORY) != 0)
508         entry->d_type = DT_DIR;
509     else
510         entry->d_type = DT_REG;
511 
512     /* Reset dummy fields */
513     entry->d_ino = 0;
514     entry->d_off = 0;
515     entry->d_reclen = sizeof(struct _wdirent);
516 
517     /* Set result address */
518     *result = entry;
519     return /*OK*/0;
520 }
521 
522 /*
523  * Close directory stream opened by opendir() function.  This invalidates the
524  * DIR structure as well as any directory entry read previously by
525  * _wreaddir().
526  */
_wclosedir(_WDIR * dirp)527 static int _wclosedir(_WDIR *dirp)
528 {
529     if (!dirp) {
530         dirent_set_errno(EBADF);
531         return /*failure*/-1;
532     }
533 
534     /* Release search handle */
535     if (dirp->handle != INVALID_HANDLE_VALUE)
536         FindClose(dirp->handle);
537 
538     /* Release search pattern */
539     free(dirp->patt);
540 
541     /* Release directory structure */
542     free(dirp);
543     return /*success*/0;
544 }
545 
546 /*
547  * Rewind directory stream such that _wreaddir() returns the very first
548  * file name again.
549  */
_wrewinddir(_WDIR * dirp)550 static void _wrewinddir(_WDIR* dirp)
551 {
552     if (!dirp)
553         return;
554 
555     /* Release existing search handle */
556     if (dirp->handle != INVALID_HANDLE_VALUE)
557         FindClose(dirp->handle);
558 
559     /* Open new search handle */
560     dirent_first(dirp);
561 }
562 
563 /* Get first directory entry */
dirent_first(_WDIR * dirp)564 static WIN32_FIND_DATAW *dirent_first(_WDIR *dirp)
565 {
566     if (!dirp)
567         return NULL;
568 
569     /* Open directory and retrieve the first entry */
570     dirp->handle = FindFirstFileExW(
571         dirp->patt, FindExInfoStandard, &dirp->data,
572         FindExSearchNameMatch, NULL, 0);
573     if (dirp->handle == INVALID_HANDLE_VALUE)
574         goto error;
575 
576     /* A directory entry is now waiting in memory */
577     dirp->cached = 1;
578     return &dirp->data;
579 
580 error:
581     /* Failed to open directory: no directory entry in memory */
582     dirp->cached = 0;
583 
584     /* Set error code */
585     DWORD errorcode = GetLastError();
586     switch (errorcode) {
587     case ERROR_ACCESS_DENIED:
588         /* No read access to directory */
589         dirent_set_errno(EACCES);
590         break;
591 
592     case ERROR_DIRECTORY:
593         /* Directory name is invalid */
594         dirent_set_errno(ENOTDIR);
595         break;
596 
597     case ERROR_PATH_NOT_FOUND:
598     default:
599         /* Cannot find the file */
600         dirent_set_errno(ENOENT);
601     }
602     return NULL;
603 }
604 
605 /* Get next directory entry */
dirent_next(_WDIR * dirp)606 static WIN32_FIND_DATAW *dirent_next(_WDIR *dirp)
607 {
608     /* Is the next directory entry already in cache? */
609     if (dirp->cached) {
610         /* Yes, a valid directory entry found in memory */
611         dirp->cached = 0;
612         return &dirp->data;
613     }
614 
615     /* No directory entry in cache */
616     if (dirp->handle == INVALID_HANDLE_VALUE)
617         return NULL;
618 
619     /* Read the next directory entry from stream */
620     if (FindNextFileW(dirp->handle, &dirp->data) == FALSE)
621         goto exit_close;
622 
623     /* Success */
624     return &dirp->data;
625 
626     /* Failure */
627 exit_close:
628     FindClose(dirp->handle);
629     dirp->handle = INVALID_HANDLE_VALUE;
630     return NULL;
631 }
632 
633 /* Open directory stream using plain old C-string */
opendir(const char * dirname)634 static DIR *opendir(const char *dirname)
635 {
636     /* Must have directory name */
637     if (dirname == NULL || dirname[0] == '\0') {
638         dirent_set_errno(ENOENT);
639         return NULL;
640     }
641 
642     /* Allocate memory for DIR structure */
643     struct DIR *dirp = (DIR*) malloc(sizeof(struct DIR));
644     if (!dirp)
645         return NULL;
646 
647     /* Convert directory name to wide-character string */
648     wchar_t wname[PATH_MAX + 1];
649     size_t n;
650     int error = mbstowcs_s(&n, wname, PATH_MAX + 1, dirname, PATH_MAX+1);
651     if (error)
652         goto exit_failure;
653 
654     /* Open directory stream using wide-character name */
655     dirp->wdirp = _wopendir(wname);
656     if (!dirp->wdirp)
657         goto exit_failure;
658 
659     /* Success */
660     return dirp;
661 
662     /* Failure */
663 exit_failure:
664     free(dirp);
665     return NULL;
666 }
667 
668 /* Read next directory entry */
readdir(DIR * dirp)669 static struct dirent *readdir(DIR *dirp)
670 {
671     /*
672      * Read directory entry to buffer.  We can safely ignore the return
673      * value as entry will be set to NULL in case of error.
674      */
675     struct dirent *entry;
676     (void) readdir_r(dirp, &dirp->ent, &entry);
677 
678     /* Return pointer to statically allocated directory entry */
679     return entry;
680 }
681 
682 /*
683  * Read next directory entry into called-allocated buffer.
684  *
685  * Returns zero on success.  If the end of directory stream is reached, then
686  * sets result to NULL and returns zero.
687  */
readdir_r(DIR * dirp,struct dirent * entry,struct dirent ** result)688 static int readdir_r(
689     DIR *dirp, struct dirent *entry, struct dirent **result)
690 {
691     /* Read next directory entry */
692     WIN32_FIND_DATAW *datap = dirent_next(dirp->wdirp);
693     if (!datap) {
694         /* No more directory entries */
695         *result = NULL;
696         return /*OK*/0;
697     }
698 
699     /* Attempt to convert file name to multi-byte string */
700     size_t n;
701     int error = wcstombs_s(
702         &n, entry->d_name, PATH_MAX + 1,
703         datap->cFileName, PATH_MAX + 1);
704 
705     /*
706      * If the file name cannot be represented by a multi-byte string, then
707      * attempt to use old 8+3 file name.  This allows the program to
708      * access files although file names may seem unfamiliar to the user.
709      *
710      * Be ware that the code below cannot come up with a short file name
711      * unless the file system provides one.  At least VirtualBox shared
712      * folders fail to do this.
713      */
714     if (error && datap->cAlternateFileName[0] != '\0') {
715         error = wcstombs_s(
716             &n, entry->d_name, PATH_MAX + 1,
717             datap->cAlternateFileName, PATH_MAX + 1);
718     }
719 
720     if (!error) {
721         /* Length of file name excluding zero terminator */
722         entry->d_namlen = n - 1;
723 
724         /* File attributes */
725         DWORD attr = datap->dwFileAttributes;
726         if ((attr & FILE_ATTRIBUTE_DEVICE) != 0)
727             entry->d_type = DT_CHR;
728         else if ((attr & FILE_ATTRIBUTE_DIRECTORY) != 0)
729             entry->d_type = DT_DIR;
730         else
731             entry->d_type = DT_REG;
732 
733         /* Reset dummy fields */
734         entry->d_ino = 0;
735         entry->d_off = 0;
736         entry->d_reclen = sizeof(struct dirent);
737     } else {
738         /*
739          * Cannot convert file name to multi-byte string so construct
740          * an erroneous directory entry and return that.  Note that
741          * we cannot return NULL as that would stop the processing
742          * of directory entries completely.
743          */
744         entry->d_name[0] = '?';
745         entry->d_name[1] = '\0';
746         entry->d_namlen = 1;
747         entry->d_type = DT_UNKNOWN;
748         entry->d_ino = 0;
749         entry->d_off = -1;
750         entry->d_reclen = 0;
751     }
752 
753     /* Return pointer to directory entry */
754     *result = entry;
755     return /*OK*/0;
756 }
757 
758 /* Close directory stream */
closedir(DIR * dirp)759 static int closedir(DIR *dirp)
760 {
761     int ok;
762 
763     if (!dirp)
764         goto exit_failure;
765 
766     /* Close wide-character directory stream */
767     ok = _wclosedir(dirp->wdirp);
768     dirp->wdirp = NULL;
769 
770     /* Release multi-byte character version */
771     free(dirp);
772     return ok;
773 
774 exit_failure:
775     /* Invalid directory stream */
776     dirent_set_errno(EBADF);
777     return /*failure*/-1;
778 }
779 
780 /* Rewind directory stream to beginning */
rewinddir(DIR * dirp)781 static void rewinddir(DIR* dirp)
782 {
783     if (!dirp)
784         return;
785 
786     /* Rewind wide-character string directory stream */
787     _wrewinddir(dirp->wdirp);
788 }
789 
790 /* Scan directory for entries */
scandir(const char * dirname,struct dirent *** namelist,int (* filter)(const struct dirent *),int (* compare)(const struct dirent **,const struct dirent **))791 static int scandir(
792     const char *dirname, struct dirent ***namelist,
793     int (*filter)(const struct dirent*),
794     int (*compare)(const struct dirent**, const struct dirent**))
795 {
796     int result;
797 
798     /* Open directory stream */
799     DIR *dir = opendir(dirname);
800     if (!dir) {
801         /* Cannot open directory */
802         return /*Error*/ -1;
803     }
804 
805     /* Read directory entries to memory */
806     struct dirent *tmp = NULL;
807     struct dirent **files = NULL;
808     size_t size = 0;
809     size_t allocated = 0;
810     while (1) {
811         /* Allocate room for a temporary directory entry */
812         if (!tmp) {
813             tmp = (struct dirent*) malloc(sizeof(struct dirent));
814             if (!tmp)
815                 goto exit_failure;
816         }
817 
818         /* Read directory entry to temporary area */
819         struct dirent *entry;
820         if (readdir_r(dir, tmp, &entry) != /*OK*/0)
821             goto exit_failure;
822 
823         /* Stop if we already read the last directory entry */
824         if (entry == NULL)
825             goto exit_success;
826 
827         /* Determine whether to include the entry in results */
828         if (filter && !filter(tmp))
829             continue;
830 
831         /* Enlarge pointer table to make room for another pointer */
832         if (size >= allocated) {
833             /* Compute number of entries in the new table */
834             size_t num_entries = size * 2 + 16;
835 
836             /* Allocate new pointer table or enlarge existing */
837             void *p = realloc(files, sizeof(void*) * num_entries);
838             if (!p)
839                 goto exit_failure;
840 
841             /* Got the memory */
842             files = (dirent**) p;
843             allocated = num_entries;
844         }
845 
846         /* Store the temporary entry to ptr table */
847         files[size++] = tmp;
848         tmp = NULL;
849     }
850 
851 exit_failure:
852     /* Release allocated file entries */
853     for (size_t i = 0; i < size; i++) {
854         free(files[i]);
855     }
856 
857     /* Release the pointer table */
858     free(files);
859     files = NULL;
860 
861     /* Exit with error code */
862     result = /*error*/ -1;
863     goto exit_status;
864 
865 exit_success:
866     /* Sort directory entries */
867     qsort(files, size, sizeof(void*),
868         (int (*) (const void*, const void*)) compare);
869 
870     /* Pass pointer table to caller */
871     if (namelist)
872         *namelist = files;
873 
874     /* Return the number of directory entries read */
875     result = (int) size;
876 
877 exit_status:
878     /* Release temporary directory entry, if we had one */
879     free(tmp);
880 
881     /* Close directory stream */
882     closedir(dir);
883     return result;
884 }
885 
886 /* Alphabetical sorting */
alphasort(const struct dirent ** a,const struct dirent ** b)887 static int alphasort(const struct dirent **a, const struct dirent **b)
888 {
889     return strcoll((*a)->d_name, (*b)->d_name);
890 }
891 
892 /* Sort versions */
versionsort(const struct dirent ** a,const struct dirent ** b)893 static int versionsort(const struct dirent **a, const struct dirent **b)
894 {
895     return strverscmp((*a)->d_name, (*b)->d_name);
896 }
897 
898 /* Compare strings */
strverscmp(const char * a,const char * b)899 static int strverscmp(const char *a, const char *b)
900 {
901     size_t i = 0;
902     size_t j;
903 
904     /* Find first difference */
905     while (a[i] == b[i]) {
906         if (a[i] == '\0') {
907             /* No difference */
908             return 0;
909         }
910         ++i;
911     }
912 
913     /* Count backwards and find the leftmost digit */
914     j = i;
915     while (j > 0 && isdigit(a[j-1])) {
916         --j;
917     }
918 
919     /* Determine mode of comparison */
920     if (a[j] == '0' || b[j] == '0') {
921         /* Find the next non-zero digit */
922         while (a[j] == '0' && a[j] == b[j]) {
923             j++;
924         }
925 
926         /* String with more digits is smaller, e.g 002 < 01 */
927         if (isdigit(a[j])) {
928             if (!isdigit(b[j])) {
929                 return -1;
930             }
931         } else if (isdigit(b[j])) {
932             return 1;
933         }
934     } else if (isdigit(a[j]) && isdigit(b[j])) {
935         /* Numeric comparison */
936         size_t k1 = j;
937         size_t k2 = j;
938 
939         /* Compute number of digits in each string */
940         while (isdigit(a[k1])) {
941             k1++;
942         }
943         while (isdigit(b[k2])) {
944             k2++;
945         }
946 
947         /* Number with more digits is bigger, e.g 999 < 1000 */
948         if (k1 < k2)
949             return -1;
950         else if (k1 > k2)
951             return 1;
952     }
953 
954     /* Alphabetical comparison */
955     return (int) ((unsigned char) a[i]) - ((unsigned char) b[i]);
956 }
957 
958 /* Convert multi-byte string to wide character string */
959 #if !defined(_MSC_VER) || _MSC_VER < 1400
dirent_mbstowcs_s(size_t * pReturnValue,wchar_t * wcstr,size_t sizeInWords,const char * mbstr,size_t count)960 static int dirent_mbstowcs_s(
961     size_t *pReturnValue, wchar_t *wcstr,
962     size_t sizeInWords, const char *mbstr, size_t count)
963 {
964     /* Older Visual Studio or non-Microsoft compiler */
965     size_t n = mbstowcs(wcstr, mbstr, sizeInWords);
966     if (wcstr && n >= count)
967         return /*error*/ 1;
968 
969     /* Zero-terminate output buffer */
970     if (wcstr && sizeInWords) {
971         if (n >= sizeInWords)
972             n = sizeInWords - 1;
973         wcstr[n] = 0;
974     }
975 
976     /* Length of multi-byte string with zero terminator */
977     if (pReturnValue) {
978         *pReturnValue = n + 1;
979     }
980 
981     /* Success */
982     return 0;
983 }
984 #endif
985 
986 /* Convert wide-character string to multi-byte string */
987 #if !defined(_MSC_VER) || _MSC_VER < 1400
dirent_wcstombs_s(size_t * pReturnValue,char * mbstr,size_t sizeInBytes,const wchar_t * wcstr,size_t count)988 static int dirent_wcstombs_s(
989     size_t *pReturnValue, char *mbstr,
990     size_t sizeInBytes, const wchar_t *wcstr, size_t count)
991 {
992     /* Older Visual Studio or non-Microsoft compiler */
993     size_t n = wcstombs(mbstr, wcstr, sizeInBytes);
994     if (mbstr && n >= count)
995         return /*error*/1;
996 
997     /* Zero-terminate output buffer */
998     if (mbstr && sizeInBytes) {
999         if (n >= sizeInBytes) {
1000             n = sizeInBytes - 1;
1001         }
1002         mbstr[n] = '\0';
1003     }
1004 
1005     /* Length of resulting multi-bytes string WITH zero-terminator */
1006     if (pReturnValue) {
1007         *pReturnValue = n + 1;
1008     }
1009 
1010     /* Success */
1011     return 0;
1012 }
1013 #endif
1014 
1015 /* Set errno variable */
1016 #if !defined(_MSC_VER) || _MSC_VER < 1400
dirent_set_errno(int error)1017 static void dirent_set_errno(int error)
1018 {
1019     /* Non-Microsoft compiler or older Microsoft compiler */
1020     errno = error;
1021 }
1022 #endif
1023 
1024 #ifdef __cplusplus
1025 }
1026 #endif
1027 #endif /*DIRENT_H*/
1028