X-Git-Url: https://wimlib.net/git/?a=blobdiff_plain;f=src%2Fwin32_replacements.c;h=ebf6e20e6cc2d4414b6270df65f7a3a214893982;hb=dd3f804966f577b104eda649f572e8e765edc7ef;hp=d3757a3f35ed843039e8c9f085f28a90386a9f3d;hpb=2159b8808998d2a75380ef6339147e48a39e3bae;p=wimlib diff --git a/src/win32_replacements.c b/src/win32_replacements.c index d3757a3f..ebf6e20e 100644 --- a/src/win32_replacements.c +++ b/src/win32_replacements.c @@ -6,20 +6,18 @@ /* * Copyright (C) 2013 Eric Biggers * - * This file is part of wimlib, a library for working with WIM files. + * This file is free software; you can redistribute it and/or modify it under + * the terms of the GNU Lesser General Public License as published by the Free + * Software Foundation; either version 3 of the License, or (at your option) any + * later version. * - * wimlib is free software; you can redistribute it and/or modify it under the - * terms of the GNU General Public License as published by the Free - * Software Foundation; either version 3 of the License, or (at your option) - * any later version. - * - * wimlib is distributed in the hope that it will be useful, but WITHOUT ANY - * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR - * A PARTICULAR PURPOSE. See the GNU General Public License for more + * This file is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more * details. * - * You should have received a copy of the GNU General Public License - * along with wimlib; if not, see http://www.gnu.org/licenses/. + * You should have received a copy of the GNU Lesser General Public License + * along with this file; if not, see http://www.gnu.org/licenses/. */ #ifdef __WIN32__ @@ -30,11 +28,12 @@ #include #include -#include /* for PathMatchSpecW() */ +#include /* for _get_osfhandle() */ +#include + #include "wimlib/win32_common.h" #include "wimlib/assert.h" -#include "wimlib/file_io.h" #include "wimlib/glob.h" #include "wimlib/error.h" #include "wimlib/util.h" @@ -57,7 +56,7 @@ err: return -1; } -/* Use the Win32 API to get the number of processors */ +/* Use the Win32 API to get the number of processors. */ unsigned win32_get_number_of_processors(void) { @@ -66,6 +65,17 @@ win32_get_number_of_processors(void) return sysinfo.dwNumberOfProcessors; } +/* Use the Win32 API to get the amount of available memory. */ +u64 +win32_get_avail_memory(void) +{ + MEMORYSTATUSEX status = { + .dwLength = sizeof(status), + }; + GlobalMemoryStatusEx(&status); + return (u64)min(status.ullTotalPhys, status.ullTotalVirtual) * 85 / 100; +} + /* Replacement for POSIX-2008 realpath(). Warning: partial functionality only * (resolved_path must be NULL). Also I highly doubt that GetFullPathName * really does the right thing under all circumstances. */ @@ -88,7 +98,7 @@ realpath(const wchar_t *path, wchar_t *resolved_path) ret = GetFullPathNameW(path, ret, resolved_path, NULL); if (!ret) { err = GetLastError(); - free(resolved_path); + FREE(resolved_path); resolved_path = NULL; goto fail_win32; } @@ -177,51 +187,10 @@ err_set_errno: return -1; } -/* Replacement for POSIX fnmatch() (partial functionality only) */ -int -fnmatch(const wchar_t *pattern, const wchar_t *string, int flags) -{ - if (PathMatchSpecW(string, pattern)) - return 0; - else - return FNM_NOMATCH; -} - -/* truncate() replacement */ -int -win32_truncate_replacement(const wchar_t *path, off_t size) -{ - DWORD err = NO_ERROR; - LARGE_INTEGER liOffset; - - HANDLE h = win32_open_existing_file(path, GENERIC_WRITE); - if (h == INVALID_HANDLE_VALUE) - goto fail; - - liOffset.QuadPart = size; - if (!SetFilePointerEx(h, liOffset, NULL, FILE_BEGIN)) - goto fail_close_handle; - - if (!SetEndOfFile(h)) - goto fail_close_handle; - CloseHandle(h); - return 0; - -fail_close_handle: - err = GetLastError(); - CloseHandle(h); -fail: - if (err == NO_ERROR) - err = GetLastError(); - set_errno_from_win32_error(err); - return -1; -} - - /* This really could be replaced with _wcserror_s, but this doesn't seem to * actually be available in MSVCRT.DLL on Windows XP (perhaps it's statically * linked in by Visual Studio...?). */ -extern int +int win32_strerror_r_replacement(int errnum, wchar_t *buf, size_t buflen) { static pthread_mutex_t strerror_lock = PTHREAD_MUTEX_INITIALIZER; @@ -299,61 +268,6 @@ pwrite(int fd, const void *buf, size_t count, off_t offset) return do_pread_or_pwrite(fd, (void*)buf, count, offset, true); } -/* Dumb Windows implementation of writev(). It writes the vectors one at a - * time. */ -ssize_t writev(int fd, const struct iovec *iov, int iovcnt) -{ - ssize_t total_bytes_written = 0; - - if (iovcnt <= 0) { - errno = EINVAL; - return -1; - } - for (int i = 0; i < iovcnt; i++) { - ssize_t bytes_written; - - bytes_written = write(fd, iov[i].iov_base, iov[i].iov_len); - if (bytes_written >= 0) - total_bytes_written += bytes_written; - if (bytes_written != iov[i].iov_len) { - if (total_bytes_written == 0) - total_bytes_written = -1; - break; - } - } - return total_bytes_written; -} - -int -win32_get_file_and_vol_ids(const wchar_t *path, u64 *ino_ret, u64 *dev_ret) -{ - HANDLE h; - BY_HANDLE_FILE_INFORMATION file_info; - int ret; - DWORD err; - - h = win32_open_existing_file(path, FILE_READ_ATTRIBUTES); - if (h == INVALID_HANDLE_VALUE) { - ret = WIMLIB_ERR_OPEN; - goto out; - } - - if (!GetFileInformationByHandle(h, &file_info)) { - ret = WIMLIB_ERR_STAT; - } else { - *ino_ret = ((u64)file_info.nFileIndexHigh << 32) | - (u64)file_info.nFileIndexLow; - *dev_ret = file_info.dwVolumeSerialNumber; - ret = 0; - } - err = GetLastError(); - CloseHandle(h); - SetLastError(err); -out: - set_errno_from_GetLastError(); - return ret; -} - /* Replacement for glob() in Windows native builds that operates on wide * characters. */ int @@ -366,6 +280,7 @@ win32_wglob(const wchar_t *pattern, int flags, HANDLE hFind; int ret; size_t nspaces; + int errno_save; const wchar_t *backslash, *end_slash; size_t prefix_len; @@ -395,9 +310,7 @@ win32_wglob(const wchar_t *pattern, int flags, errno = 0; return GLOB_NOMATCH; } else { - /* The other possible error codes for FindFirstFile() - * are undocumented. */ - errno = EIO; + set_errno_from_win32_error(err); return GLOB_ABORTED; } } @@ -431,22 +344,21 @@ win32_wglob(const wchar_t *pattern, int flags, } while (FindNextFileW(hFind, &dat)); err = GetLastError(); CloseHandle(hFind); - if (err == ERROR_NO_MORE_FILES) { - errno = 0; - return 0; - } else { - /* Other possible error codes for FindNextFile() are - * undocumented */ - errno = EIO; + if (err != ERROR_NO_MORE_FILES) { + set_errno_from_win32_error(err); ret = GLOB_ABORTED; goto fail_globfree; } + return 0; + oom: CloseHandle(hFind); errno = ENOMEM; ret = GLOB_NOSPACE; fail_globfree: + errno_save = errno; globfree(pglob); + errno = errno_save; return ret; } @@ -455,8 +367,37 @@ globfree(glob_t *pglob) { size_t i; for (i = 0; i < pglob->gl_pathc; i++) - free(pglob->gl_pathv[i]); - free(pglob->gl_pathv); + FREE(pglob->gl_pathv[i]); + FREE(pglob->gl_pathv); +} + +/* Replacement for fopen(path, "a") that doesn't prevent other processes from + * reading the file */ +FILE * +win32_open_logfile(const wchar_t *path) +{ + HANDLE h; + int fd; + FILE *fp; + + h = CreateFile(path, FILE_APPEND_DATA, FILE_SHARE_VALID_FLAGS, + NULL, OPEN_ALWAYS, 0, NULL); + if (h == INVALID_HANDLE_VALUE) + return NULL; + + fd = _open_osfhandle((intptr_t)h, O_APPEND); + if (fd < 0) { + CloseHandle(h); + return NULL; + } + + fp = fdopen(fd, "a"); + if (!fp) { + close(fd); + return NULL; + } + + return fp; } #endif /* __WIN32__ */