]> wimlib.net Git - wimlib/blob - src/win32_replacements.c
Add a clang-format file
[wimlib] / src / win32_replacements.c
1 /*
2  * win32_replacements.c - Replacements for various functions not available on
3  * Windows, such as fsync().
4  */
5
6 /*
7  * Copyright (C) 2013-2016 Eric Biggers
8  *
9  * This file is free software; you can redistribute it and/or modify it under
10  * the terms of the GNU Lesser General Public License as published by the Free
11  * Software Foundation; either version 3 of the License, or (at your option) any
12  * later version.
13  *
14  * This file is distributed in the hope that it will be useful, but WITHOUT
15  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
16  * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
17  * details.
18  *
19  * You should have received a copy of the GNU Lesser General Public License
20  * along with this file; if not, see https://www.gnu.org/licenses/.
21  */
22
23 #ifdef _WIN32
24
25 #ifdef HAVE_CONFIG_H
26 #  include "config.h"
27 #endif
28
29 #include <errno.h>
30 #include <io.h> /* for _get_osfhandle()  */
31 #include <fcntl.h>
32
33 #include "wimlib/win32_common.h"
34
35 #include "wimlib/assert.h"
36 #include "wimlib/glob.h"
37 #include "wimlib/error.h"
38 #include "wimlib/timestamp.h"
39 #include "wimlib/util.h"
40
41 static int
42 win32_error_to_errno(DWORD err_code)
43 {
44         /* This mapping is that used in Cygwin.
45          * Some of these choices are arbitrary. */
46         switch (err_code) {
47         case ERROR_ACCESS_DENIED:
48                 return EACCES;
49         case ERROR_ACTIVE_CONNECTIONS:
50                 return EAGAIN;
51         case ERROR_ALREADY_EXISTS:
52                 return EEXIST;
53         case ERROR_BAD_DEVICE:
54                 return ENODEV;
55         case ERROR_BAD_EXE_FORMAT:
56                 return ENOEXEC;
57         case ERROR_BAD_NETPATH:
58                 return ENOENT;
59         case ERROR_BAD_NET_NAME:
60                 return ENOENT;
61         case ERROR_BAD_NET_RESP:
62                 return ENOSYS;
63         case ERROR_BAD_PATHNAME:
64                 return ENOENT;
65         case ERROR_BAD_PIPE:
66                 return EINVAL;
67         case ERROR_BAD_UNIT:
68                 return ENODEV;
69         case ERROR_BAD_USERNAME:
70                 return EINVAL;
71         case ERROR_BEGINNING_OF_MEDIA:
72                 return EIO;
73         case ERROR_BROKEN_PIPE:
74                 return EPIPE;
75         case ERROR_BUSY:
76                 return EBUSY;
77         case ERROR_BUS_RESET:
78                 return EIO;
79         case ERROR_CALL_NOT_IMPLEMENTED:
80                 return ENOSYS;
81         case ERROR_CANNOT_MAKE:
82                 return EPERM;
83         case ERROR_CHILD_NOT_COMPLETE:
84                 return EBUSY;
85         case ERROR_COMMITMENT_LIMIT:
86                 return EAGAIN;
87         case ERROR_CRC:
88                 return EIO;
89         case ERROR_DEVICE_DOOR_OPEN:
90                 return EIO;
91         case ERROR_DEVICE_IN_USE:
92                 return EAGAIN;
93         case ERROR_DEVICE_REQUIRES_CLEANING:
94                 return EIO;
95         case ERROR_DIRECTORY:
96                 return ENOTDIR;
97         case ERROR_DIR_NOT_EMPTY:
98                 return ENOTEMPTY;
99         case ERROR_DISK_CORRUPT:
100                 return EIO;
101         case ERROR_DISK_FULL:
102                 return ENOSPC;
103 #ifdef ENOTUNIQ
104         case ERROR_DUP_NAME:
105                 return ENOTUNIQ;
106 #endif
107         case ERROR_EAS_DIDNT_FIT:
108                 return ENOSPC;
109 #ifdef ENOTSUP
110         case ERROR_EAS_NOT_SUPPORTED:
111                 return ENOTSUP;
112 #endif
113         case ERROR_EA_LIST_INCONSISTENT:
114                 return EINVAL;
115         case ERROR_EA_TABLE_FULL:
116                 return ENOSPC;
117         case ERROR_END_OF_MEDIA:
118                 return ENOSPC;
119         case ERROR_EOM_OVERFLOW:
120                 return EIO;
121         case ERROR_EXE_MACHINE_TYPE_MISMATCH:
122                 return ENOEXEC;
123         case ERROR_EXE_MARKED_INVALID:
124                 return ENOEXEC;
125         case ERROR_FILEMARK_DETECTED:
126                 return EIO;
127         case ERROR_FILENAME_EXCED_RANGE:
128                 return ENAMETOOLONG;
129         case ERROR_FILE_CORRUPT:
130                 return EEXIST;
131         case ERROR_FILE_EXISTS:
132                 return EEXIST;
133         case ERROR_FILE_INVALID:
134                 return ENXIO;
135         case ERROR_FILE_NOT_FOUND:
136                 return ENOENT;
137         case ERROR_HANDLE_DISK_FULL:
138                 return ENOSPC;
139 #ifdef ENODATA
140         case ERROR_HANDLE_EOF:
141                 return ENODATA;
142 #endif
143         case ERROR_INVALID_ADDRESS:
144                 return EINVAL;
145         case ERROR_INVALID_AT_INTERRUPT_TIME:
146                 return EINTR;
147         case ERROR_INVALID_BLOCK_LENGTH:
148                 return EIO;
149         case ERROR_INVALID_DATA:
150                 return EINVAL;
151         case ERROR_INVALID_DRIVE:
152                 return ENODEV;
153         case ERROR_INVALID_EA_NAME:
154                 return EINVAL;
155         case ERROR_INVALID_EXE_SIGNATURE:
156                 return ENOEXEC;
157 #ifdef EBADRQC
158         case ERROR_INVALID_FUNCTION:
159                 return EBADRQC;
160 #endif
161         case ERROR_INVALID_HANDLE:
162                 return EBADF;
163         case ERROR_INVALID_NAME:
164                 return ENOENT;
165         case ERROR_INVALID_PARAMETER:
166                 return EINVAL;
167         case ERROR_INVALID_SIGNAL_NUMBER:
168                 return EINVAL;
169         case ERROR_IOPL_NOT_ENABLED:
170                 return ENOEXEC;
171         case ERROR_IO_DEVICE:
172                 return EIO;
173         case ERROR_IO_INCOMPLETE:
174                 return EAGAIN;
175         case ERROR_IO_PENDING:
176                 return EAGAIN;
177         case ERROR_LOCK_VIOLATION:
178                 return EBUSY;
179         case ERROR_MAX_THRDS_REACHED:
180                 return EAGAIN;
181         case ERROR_META_EXPANSION_TOO_LONG:
182                 return EINVAL;
183         case ERROR_MOD_NOT_FOUND:
184                 return ENOENT;
185 #ifdef EMSGSIZE
186         case ERROR_MORE_DATA:
187                 return EMSGSIZE;
188 #endif
189         case ERROR_NEGATIVE_SEEK:
190                 return EINVAL;
191         case ERROR_NETNAME_DELETED:
192                 return ENOENT;
193         case ERROR_NOACCESS:
194                 return EFAULT;
195         case ERROR_NONE_MAPPED:
196                 return EINVAL;
197         case ERROR_NONPAGED_SYSTEM_RESOURCES:
198                 return EAGAIN;
199 #ifdef ENOLINK
200         case ERROR_NOT_CONNECTED:
201                 return ENOLINK;
202 #endif
203         case ERROR_NOT_ENOUGH_MEMORY:
204                 return ENOMEM;
205         case ERROR_NOT_OWNER:
206                 return EPERM;
207 #ifdef ENOMEDIUM
208         case ERROR_NOT_READY:
209                 return ENOMEDIUM;
210 #endif
211         case ERROR_NOT_SAME_DEVICE:
212                 return EXDEV;
213         case ERROR_NOT_SUPPORTED:
214                 return ENOSYS;
215         case ERROR_NO_DATA:
216                 return EPIPE;
217         case ERROR_NO_DATA_DETECTED:
218                 return EIO;
219 #ifdef ENOMEDIUM
220         case ERROR_NO_MEDIA_IN_DRIVE:
221                 return ENOMEDIUM;
222 #endif
223 #ifdef ENMFILE
224         case ERROR_NO_MORE_FILES:
225                 return ENMFILE;
226 #endif
227 #ifdef ENMFILE
228         case ERROR_NO_MORE_ITEMS:
229                 return ENMFILE;
230 #endif
231         case ERROR_NO_MORE_SEARCH_HANDLES:
232                 return ENFILE;
233         case ERROR_NO_PROC_SLOTS:
234                 return EAGAIN;
235         case ERROR_NO_SIGNAL_SENT:
236                 return EIO;
237         case ERROR_NO_SYSTEM_RESOURCES:
238                 return EFBIG;
239         case ERROR_NO_TOKEN:
240                 return EINVAL;
241         case ERROR_OPEN_FAILED:
242                 return EIO;
243         case ERROR_OPEN_FILES:
244                 return EAGAIN;
245         case ERROR_OUTOFMEMORY:
246                 return ENOMEM;
247         case ERROR_PAGED_SYSTEM_RESOURCES:
248                 return EAGAIN;
249         case ERROR_PAGEFILE_QUOTA:
250                 return EAGAIN;
251         case ERROR_PATH_NOT_FOUND:
252                 return ENOENT;
253         case ERROR_PIPE_BUSY:
254                 return EBUSY;
255         case ERROR_PIPE_CONNECTED:
256                 return EBUSY;
257 #ifdef ECOMM
258         case ERROR_PIPE_LISTENING:
259                 return ECOMM;
260         case ERROR_PIPE_NOT_CONNECTED:
261                 return ECOMM;
262 #endif
263         case ERROR_POSSIBLE_DEADLOCK:
264                 return EDEADLOCK;
265         case ERROR_PRIVILEGE_NOT_HELD:
266                 return EPERM;
267         case ERROR_PROCESS_ABORTED:
268                 return EFAULT;
269         case ERROR_PROC_NOT_FOUND:
270                 return ESRCH;
271 #ifdef ENONET
272         case ERROR_REM_NOT_LIST:
273                 return ENONET;
274 #endif
275         case ERROR_SECTOR_NOT_FOUND:
276                 return EINVAL;
277         case ERROR_SEEK:
278                 return EINVAL;
279         case ERROR_SETMARK_DETECTED:
280                 return EIO;
281         case ERROR_SHARING_BUFFER_EXCEEDED:
282                 return ENOLCK;
283         case ERROR_SHARING_VIOLATION:
284                 return EBUSY;
285         case ERROR_SIGNAL_PENDING:
286                 return EBUSY;
287         case ERROR_SIGNAL_REFUSED:
288                 return EIO;
289 #ifdef ELIBBAD
290         case ERROR_SXS_CANT_GEN_ACTCTX:
291                 return ELIBBAD;
292 #endif
293         case ERROR_THREAD_1_INACTIVE:
294                 return EINVAL;
295         case ERROR_TOO_MANY_LINKS:
296                 return EMLINK;
297         case ERROR_TOO_MANY_OPEN_FILES:
298                 return EMFILE;
299         case ERROR_WAIT_NO_CHILDREN:
300                 return ECHILD;
301         case ERROR_WORKING_SET_QUOTA:
302                 return EAGAIN;
303         case ERROR_WRITE_PROTECT:
304                 return EROFS;
305         default:
306                 return -1;
307         }
308 }
309
310 static void
311 set_errno_from_win32_error(DWORD err)
312 {
313         errno = win32_error_to_errno(err);
314 }
315
316 static void
317 set_errno_from_GetLastError(void)
318 {
319         set_errno_from_win32_error(GetLastError());
320 }
321
322 /* Replacement for POSIX fsync() */
323 int
324 fsync(int fd)
325 {
326         HANDLE h;
327
328         h = (HANDLE)_get_osfhandle(fd);
329         if (h == INVALID_HANDLE_VALUE)
330                 goto err;
331         if (!FlushFileBuffers(h))
332                 goto err_set_errno;
333         return 0;
334 err_set_errno:
335         set_errno_from_GetLastError();
336 err:
337         return -1;
338 }
339
340 /* Use the Win32 API to get the number of processors.  */
341 unsigned
342 get_available_cpus(void)
343 {
344         SYSTEM_INFO sysinfo;
345         GetSystemInfo(&sysinfo);
346         return sysinfo.dwNumberOfProcessors;
347 }
348
349 /* Use the Win32 API to get the amount of available memory.  */
350 u64
351 get_available_memory(void)
352 {
353         MEMORYSTATUSEX status = {
354                 .dwLength = sizeof(status),
355         };
356         GlobalMemoryStatusEx(&status);
357         return (u64)min(status.ullTotalPhys, status.ullTotalVirtual) * 85 / 100;
358 }
359
360 /* Replacement for POSIX-2008 realpath().  Warning: partial functionality only
361  * (resolved_path must be NULL).   Also I highly doubt that GetFullPathName
362  * really does the right thing under all circumstances. */
363 wchar_t *
364 realpath(const wchar_t *path, wchar_t *resolved_path)
365 {
366         DWORD ret;
367         DWORD err;
368         wimlib_assert(resolved_path == NULL);
369
370         ret = GetFullPathNameW(path, 0, NULL, NULL);
371         if (!ret) {
372                 err = GetLastError();
373                 goto fail_win32;
374         }
375
376         resolved_path = MALLOC(ret * sizeof(wchar_t));
377         if (!resolved_path)
378                 goto out;
379         ret = GetFullPathNameW(path, ret, resolved_path, NULL);
380         if (!ret) {
381                 err = GetLastError();
382                 FREE(resolved_path);
383                 resolved_path = NULL;
384                 goto fail_win32;
385         }
386         goto out;
387 fail_win32:
388         set_errno_from_win32_error(err);
389 out:
390         return resolved_path;
391 }
392
393 /* A quick hack to get reasonable rename() semantics on Windows, in particular
394  * deleting the destination file instead of failing with ERROR_FILE_EXISTS and
395  * working around any processes that may have the destination file open.
396  *
397  * Note: This is intended to be called when overwriting a regular file with an
398  * updated copy and is *not* a fully POSIX compliant rename().  For that you may
399  * wish to take a look at Cygwin's implementation, but be prepared...
400  *
401  * Return 0 on success, -1 on regular error, or 1 if the destination file was
402  * deleted but the source could not be renamed and therefore should not be
403  * deleted.
404  */
405 int
406 win32_rename_replacement(const wchar_t *srcpath, const wchar_t *dstpath)
407 {
408         wchar_t *tmpname;
409
410         /* Normally, MoveFileExW() with the MOVEFILE_REPLACE_EXISTING flag does
411          * what we want.  */
412
413         if (MoveFileExW(srcpath, dstpath, MOVEFILE_REPLACE_EXISTING))
414                 return 0;
415
416         /* MoveFileExW() failed.  One way this can happen is if any process has
417          * the destination file open, in which case ERROR_ACCESS_DENIED is
418          * produced.  This can commonly happen if there is a backup or antivirus
419          * program monitoring or scanning the files.  This behavior is very
420          * different from the behavior of POSIX rename(), which simply unlinks
421          * the destination file and allows other processes to keep it open!  */
422
423         if (GetLastError() != ERROR_ACCESS_DENIED)
424                 goto err_set_errno;
425
426         /* We can work around the above-mentioned problem by renaming the
427          * destination file to yet another temporary file, then "deleting" it,
428          * which on Windows will in fact not actually delete it immediately but
429          * rather mark it for deletion when the last handle to it is closed.  */
430         {
431                 static const wchar_t orig_suffix[5] = L".orig";
432                 const size_t num_rand_chars = 9;
433                 wchar_t *p;
434
435                 size_t dstlen = wcslen(dstpath);
436
437                 tmpname = alloca(sizeof(wchar_t) *
438                                  (dstlen + ARRAY_LEN(orig_suffix) + num_rand_chars + 1));
439                 p = tmpname;
440                 p = wmempcpy(p, dstpath, dstlen);
441                 p = wmempcpy(p, orig_suffix, ARRAY_LEN(orig_suffix));
442                 get_random_alnum_chars(p, num_rand_chars);
443                 p += num_rand_chars;
444                 *p = L'\0';
445         }
446
447         if (!MoveFile(dstpath, tmpname))
448                 goto err_set_errno;
449
450         if (!DeleteFile(tmpname)) {
451                 set_errno_from_GetLastError();
452                 WARNING_WITH_ERRNO("Failed to delete original file "
453                                    "(moved to \"%ls\")", tmpname);
454         }
455
456         if (!MoveFile(srcpath, dstpath)) {
457                 set_errno_from_GetLastError();
458                 WARNING_WITH_ERRNO("Atomic semantics not respected in "
459                                    "failed rename() (new file is at \"%ls\")",
460                                    srcpath);
461                 return 1;
462         }
463
464         return 0;
465
466 err_set_errno:
467         set_errno_from_GetLastError();
468         return -1;
469 }
470
471 #define MAX_IO_AMOUNT 1048576
472
473 static int
474 do_pread_or_pwrite(int fd, void *buf, size_t count, off_t offset,
475                    bool is_pwrite)
476 {
477         HANDLE h;
478         LARGE_INTEGER orig_offset;
479         DWORD result = 0xFFFFFFFF;
480         LARGE_INTEGER relative_offset;
481         OVERLAPPED overlapped;
482         BOOL bret;
483         DWORD err = 0;
484
485         h = (HANDLE)_get_osfhandle(fd);
486         if (h == INVALID_HANDLE_VALUE)
487                 goto error;
488
489         if (GetFileType(h) == FILE_TYPE_PIPE) {
490                 errno = ESPIPE;
491                 goto error;
492         }
493
494         /* Get original position */
495         relative_offset.QuadPart = 0;
496         if (!SetFilePointerEx(h, relative_offset, &orig_offset, FILE_CURRENT)) {
497                 err = GetLastError();
498                 win32_error(err, L"Failed to get original file position");
499                 goto error;
500         }
501
502         memset(&overlapped, 0, sizeof(overlapped));
503         overlapped.Offset = offset;
504         overlapped.OffsetHigh = offset >> 32;
505
506         /* Do the read or write at the specified offset */
507         count = min(count, MAX_IO_AMOUNT);
508         SetLastError(0);
509         if (is_pwrite)
510                 bret = WriteFile(h, buf, count, &result, &overlapped);
511         else
512                 bret = ReadFile(h, buf, count, &result, &overlapped);
513         if (!bret) {
514                 err = GetLastError();
515                 win32_error(err, L"Failed to %s %zu bytes at offset %"PRIu64,
516                             (is_pwrite ? "write" : "read"), count, offset);
517                 goto error;
518         }
519
520         wimlib_assert(result <= count);
521
522         /* Restore the original position */
523         if (!SetFilePointerEx(h, orig_offset, NULL, FILE_BEGIN)) {
524                 err = GetLastError();
525                 win32_error(err, L"Failed to restore file position to %"PRIu64,
526                             offset);
527                 goto error;
528         }
529
530         return result;
531
532 error:
533         if (err)
534                 set_errno_from_win32_error(err);
535         return -1;
536 }
537
538 /* Dumb Windows implementation of pread().  It temporarily changes the file
539  * offset, so it is not safe to use with readers/writers on the same file
540  * descriptor.  */
541 ssize_t
542 win32_pread(int fd, void *buf, size_t count, off_t offset)
543 {
544         return do_pread_or_pwrite(fd, buf, count, offset, false);
545 }
546
547 /* Dumb Windows implementation of pwrite().  It temporarily changes the file
548  * offset, so it is not safe to use with readers/writers on the same file
549  * descriptor. */
550 ssize_t
551 win32_pwrite(int fd, const void *buf, size_t count, off_t offset)
552 {
553         return do_pread_or_pwrite(fd, (void*)buf, count, offset, true);
554 }
555
556 /* Replacement for read() which doesn't hide the Win32 error code */
557 ssize_t
558 win32_read(int fd, void *buf, size_t count)
559 {
560         HANDLE h = (HANDLE)_get_osfhandle(fd);
561         DWORD result = 0xFFFFFFFF;
562
563         if (h == INVALID_HANDLE_VALUE)
564                 return -1;
565
566         count = min(count, MAX_IO_AMOUNT);
567         SetLastError(0);
568         if (!ReadFile(h, buf, count, &result, NULL)) {
569                 DWORD err = GetLastError();
570                 win32_error(err,
571                             L"Error reading %zu bytes from fd %d", count, fd);
572                 set_errno_from_win32_error(err);
573                 return -1;
574         }
575
576         wimlib_assert(result <= count);
577         return result;
578 }
579
580 /* Replacement for write() which doesn't hide the Win32 error code */
581 ssize_t
582 win32_write(int fd, const void *buf, size_t count)
583 {
584         HANDLE h = (HANDLE)_get_osfhandle(fd);
585         DWORD result = 0xFFFFFFFF;
586
587         if (h == INVALID_HANDLE_VALUE)
588                 return -1;
589
590         count = min(count, MAX_IO_AMOUNT);
591         SetLastError(0);
592         if (!WriteFile(h, buf, count, &result, NULL)) {
593                 DWORD err = GetLastError();
594                 win32_error(err,
595                             L"Error writing %zu bytes to fd %d", count, fd);
596                 set_errno_from_win32_error(err);
597                 return -1;
598         }
599
600         wimlib_assert(result <= count);
601         return result;
602 }
603
604 /* Replacement for glob() in Windows native builds that operates on wide
605  * characters.  */
606 int
607 win32_wglob(const wchar_t *pattern, int flags,
608             int (*errfunc)(const wchar_t *epath, int eerrno),
609             glob_t *pglob)
610 {
611         WIN32_FIND_DATAW dat;
612         DWORD err;
613         HANDLE hFind;
614         int ret;
615         size_t nspaces;
616         int errno_save;
617
618         const wchar_t *backslash, *end_slash;
619         size_t prefix_len;
620
621         backslash = wcsrchr(pattern, L'\\');
622         end_slash = wcsrchr(pattern, L'/');
623
624         if (backslash > end_slash)
625                 end_slash = backslash;
626
627         if (end_slash)
628                 prefix_len = end_slash - pattern + 1;
629         else
630                 prefix_len = 0;
631
632         /* This function does not support all functionality of the POSIX glob(),
633          * so make sure the parameters are consistent with supported
634          * functionality. */
635         wimlib_assert(errfunc == NULL);
636         wimlib_assert((flags & GLOB_ERR) == GLOB_ERR);
637         wimlib_assert((flags & ~(GLOB_NOSORT | GLOB_ERR)) == 0);
638
639         hFind = FindFirstFileW(pattern, &dat);
640         if (hFind == INVALID_HANDLE_VALUE) {
641                 err = GetLastError();
642                 if (err == ERROR_FILE_NOT_FOUND) {
643                         errno = 0;
644                         return GLOB_NOMATCH;
645                 } else {
646                         set_errno_from_win32_error(err);
647                         return GLOB_ABORTED;
648                 }
649         }
650         pglob->gl_pathc = 0;
651         pglob->gl_pathv = NULL;
652         nspaces = 0;
653         do {
654                 wchar_t *path;
655                 if (pglob->gl_pathc == nspaces) {
656                         size_t new_nspaces;
657                         wchar_t **pathv;
658
659                         new_nspaces = nspaces * 2 + 1;
660                         pathv = REALLOC(pglob->gl_pathv,
661                                         new_nspaces * sizeof(pglob->gl_pathv[0]));
662                         if (!pathv)
663                                 goto oom;
664                         pglob->gl_pathv = pathv;
665                         nspaces = new_nspaces;
666                 }
667                 size_t filename_len = wcslen(dat.cFileName);
668                 size_t len_needed = prefix_len + filename_len;
669
670                 path = MALLOC((len_needed + 1) * sizeof(wchar_t));
671                 if (!path)
672                         goto oom;
673
674                 wmemcpy(path, pattern, prefix_len);
675                 wmemcpy(path + prefix_len, dat.cFileName, filename_len + 1);
676                 pglob->gl_pathv[pglob->gl_pathc++] = path;
677         } while (FindNextFileW(hFind, &dat));
678         err = GetLastError();
679         FindClose(hFind);
680         if (err != ERROR_NO_MORE_FILES) {
681                 set_errno_from_win32_error(err);
682                 ret = GLOB_ABORTED;
683                 goto fail_globfree;
684         }
685         return 0;
686
687 oom:
688         FindClose(hFind);
689         errno = ENOMEM;
690         ret = GLOB_NOSPACE;
691 fail_globfree:
692         errno_save = errno;
693         globfree(pglob);
694         errno = errno_save;
695         return ret;
696 }
697
698 void
699 globfree(glob_t *pglob)
700 {
701         size_t i;
702         for (i = 0; i < pglob->gl_pathc; i++)
703                 FREE(pglob->gl_pathv[i]);
704         FREE(pglob->gl_pathv);
705 }
706
707 /* Replacement for fopen(path, "a") that doesn't prevent other processes from
708  * reading the file  */
709 FILE *
710 win32_open_logfile(const wchar_t *path)
711 {
712         HANDLE h;
713         int fd;
714         FILE *fp;
715
716         h = CreateFile(path, FILE_APPEND_DATA, FILE_SHARE_VALID_FLAGS,
717                        NULL, OPEN_ALWAYS, 0, NULL);
718         if (h == INVALID_HANDLE_VALUE)
719                 return NULL;
720
721         fd = _open_osfhandle((intptr_t)h, O_APPEND);
722         if (fd < 0) {
723                 CloseHandle(h);
724                 return NULL;
725         }
726
727         fp = fdopen(fd, "a");
728         if (!fp) {
729                 close(fd);
730                 return NULL;
731         }
732
733         return fp;
734 }
735
736 #define RtlGenRandom SystemFunction036
737 BOOLEAN WINAPI RtlGenRandom(PVOID RandomBuffer, ULONG RandomBufferLength);
738
739 /*
740  * Generate @n cryptographically secure random bytes (thread-safe)
741  *
742  * This is the Windows version.  It uses RtlGenRandom() (actually called
743  * SystemFunction036) from advapi32.dll.
744  */
745 void
746 get_random_bytes(void *p, size_t n)
747 {
748         while (n != 0) {
749                 u32 count = min(n, UINT32_MAX);
750
751                 if (!RtlGenRandom(p, count)) {
752                         win32_error(GetLastError(),
753                                     L"RtlGenRandom() failed (count=%u)", count);
754                         wimlib_assert(0);
755                         count = 0;
756                 }
757                 p += count;
758                 n -= count;
759         }
760 }
761
762 /* Retrieve the current time as a WIM timestamp.  */
763 u64
764 now_as_wim_timestamp(void)
765 {
766         FILETIME ft;
767
768         GetSystemTimeAsFileTime(&ft);
769
770         return ((u64)ft.dwHighDateTime << 32) | ft.dwLowDateTime;
771 }
772
773 #endif /* _WIN32 */