X-Git-Url: https://wimlib.net/git/?a=blobdiff_plain;f=src%2Fwin32_common.c;h=1768fd9bd5076371351c1e5f8fad4ac0ac4966ce;hb=06b7241e194674d56c970fc56441beb84bc258bb;hp=051720a255ef9d15f3ebda96a050c25aa39f961c;hpb=22920a29da3c42c5aa07c0ad0817d78bf26d99c3;p=wimlib diff --git a/src/win32_common.c b/src/win32_common.c index 051720a2..1768fd9b 100644 --- a/src/win32_common.c +++ b/src/win32_common.c @@ -3,7 +3,7 @@ */ /* - * Copyright (C) 2013 Eric Biggers + * Copyright (C) 2013, 2014 Eric Biggers * * This file is part of wimlib, a library for working with WIM files. * @@ -29,9 +29,7 @@ #include -#ifdef WITH_NTDLL -# include -#endif +#include #include "wimlib/win32_common.h" #include "wimlib/assert.h" @@ -320,13 +318,11 @@ set_errno_from_GetLastError(void) set_errno_from_win32_error(GetLastError()); } -#ifdef WITH_NTDLL void -set_errno_from_nt_status(DWORD status) +set_errno_from_nt_status(NTSTATUS status) { - set_errno_from_win32_error(RtlNtStatusToDosError(status)); + set_errno_from_win32_error((*func_RtlNtStatusToDosError)(status)); } -#endif /* Given a Windows-style path, return the number of characters of the prefix * that specify the path to the root directory of a drive, or return 0 if the @@ -502,43 +498,94 @@ win32_release_capture_and_apply_privileges(void) HANDLE win32_open_existing_file(const wchar_t *path, DWORD dwDesiredAccess) { - return CreateFileW(path, - dwDesiredAccess, - FILE_SHARE_READ, - NULL, /* lpSecurityAttributes */ - OPEN_EXISTING, - FILE_FLAG_BACKUP_SEMANTICS | - FILE_FLAG_OPEN_REPARSE_POINT, - NULL /* hTemplateFile */); + return CreateFile(path, + dwDesiredAccess, + FILE_SHARE_READ, + NULL, /* lpSecurityAttributes */ + OPEN_EXISTING, + FILE_FLAG_BACKUP_SEMANTICS | + FILE_FLAG_OPEN_REPARSE_POINT, + NULL /* hTemplateFile */); } -#ifndef WITH_NTDLL -/* Pointers to functions that are not available on all targetted versions of - * Windows (XP and later). NOTE: The WINAPI annotations seem to be important; I - * assume it specifies a certain calling convention. */ - -/* Vista and later */ -HANDLE (WINAPI *win32func_FindFirstStreamW)(LPCWSTR lpFileName, - STREAM_INFO_LEVELS InfoLevel, - LPVOID lpFindStreamData, - DWORD dwFlags) = NULL; - -/* Vista and later */ -BOOL (WINAPI *win32func_FindNextStreamW)(HANDLE hFindStream, - LPVOID lpFindStreamData) = NULL; -#endif /* !WITH_NTDLL */ - -/* Vista and later */ -BOOL (WINAPI *win32func_CreateSymbolicLinkW)(const wchar_t *lpSymlinkFileName, - const wchar_t *lpTargetFileName, - DWORD dwFlags) = NULL; +/* Pointers to dynamically loaded functions */ + +/* kernel32.dll: Vista and later */ +BOOL (WINAPI *func_CreateSymbolicLinkW)(const wchar_t *lpSymlinkFileName, + const wchar_t *lpTargetFileName, + DWORD dwFlags); + +/* ntdll.dll */ + +NTSTATUS (WINAPI *func_NtOpenFile) (PHANDLE FileHandle, + ACCESS_MASK DesiredAccess, + POBJECT_ATTRIBUTES ObjectAttributes, + PIO_STATUS_BLOCK IoStatusBlock, + ULONG ShareAccess, + ULONG OpenOptions); + +NTSTATUS (WINAPI *func_NtReadFile) (HANDLE FileHandle, + HANDLE Event, + PIO_APC_ROUTINE ApcRoutine, + PVOID ApcContext, + PIO_STATUS_BLOCK IoStatusBlock, + PVOID Buffer, + ULONG Length, + PLARGE_INTEGER ByteOffset, + PULONG Key); + +NTSTATUS (WINAPI *func_NtOpenFile) (PHANDLE FileHandle, + ACCESS_MASK DesiredAccess, + POBJECT_ATTRIBUTES ObjectAttributes, + PIO_STATUS_BLOCK IoStatusBlock, + ULONG ShareAccess, + ULONG OpenOptions); + +NTSTATUS (WINAPI *func_NtQueryInformationFile)(HANDLE FileHandle, + PIO_STATUS_BLOCK IoStatusBlock, + PVOID FileInformation, + ULONG Length, + FILE_INFORMATION_CLASS FileInformationClass); + +NTSTATUS (WINAPI *func_NtQuerySecurityObject)(HANDLE handle, + SECURITY_INFORMATION SecurityInformation, + PSECURITY_DESCRIPTOR SecurityDescriptor, + ULONG Length, + PULONG LengthNeeded); + +NTSTATUS (WINAPI *func_NtQueryDirectoryFile) (HANDLE FileHandle, + HANDLE Event, + PIO_APC_ROUTINE ApcRoutine, + PVOID ApcContext, + PIO_STATUS_BLOCK IoStatusBlock, + PVOID FileInformation, + ULONG Length, + FILE_INFORMATION_CLASS FileInformationClass, + BOOLEAN ReturnSingleEntry, + PUNICODE_STRING FileName, + BOOLEAN RestartScan); + +NTSTATUS (WINAPI *func_NtQueryVolumeInformationFile) (HANDLE FileHandle, + PIO_STATUS_BLOCK IoStatusBlock, + PVOID FsInformation, + ULONG Length, + FS_INFORMATION_CLASS FsInformationClass); + +NTSTATUS (WINAPI *func_NtSetSecurityObject)(HANDLE Handle, + SECURITY_INFORMATION SecurityInformation, + PSECURITY_DESCRIPTOR SecurityDescriptor); + +NTSTATUS (WINAPI *func_NtClose) (HANDLE Handle); + +DWORD (WINAPI *func_RtlNtStatusToDosError)(NTSTATUS status); + +NTSTATUS (WINAPI *func_RtlCreateSystemVolumeInformationFolder) + (PCUNICODE_STRING VolumeRootPath); static OSVERSIONINFO windows_version_info = { .dwOSVersionInfoSize = sizeof(OSVERSIONINFO), }; -static HMODULE hKernel32 = NULL; - static bool acquired_privileges = false; bool @@ -549,18 +596,106 @@ windows_version_is_at_least(unsigned major, unsigned minor) windows_version_info.dwMinorVersion >= minor); } +struct dll_sym { + void **func_ptr; + const char *name; + bool required; +}; + +#define DLL_SYM(name, required) { (void **)&func_##name, #name, required } + +#define for_each_sym(sym, spec) \ + for ((sym) = (spec)->syms; (sym)->name; (sym)++) + +struct dll_spec { + const wchar_t *name; + HMODULE handle; + const struct dll_sym syms[]; +}; + +struct dll_spec ntdll_spec = { + .name = L"ntdll.dll", + .syms = { + DLL_SYM(NtOpenFile, true), + DLL_SYM(NtReadFile, true), + DLL_SYM(NtQueryInformationFile, true), + DLL_SYM(NtQuerySecurityObject, true), + DLL_SYM(NtQueryDirectoryFile, true), + DLL_SYM(NtQueryVolumeInformationFile, true), + DLL_SYM(NtSetSecurityObject, true), + DLL_SYM(NtClose, true), + DLL_SYM(RtlNtStatusToDosError, true), + DLL_SYM(RtlCreateSystemVolumeInformationFolder, false), + {NULL, NULL}, + }, +}; + +struct dll_spec kernel32_spec = { + .name = L"kernel32.dll", + .syms = { + DLL_SYM(CreateSymbolicLinkW, false), + {NULL, NULL}, + }, +}; + +static int +init_dll(struct dll_spec *spec) +{ + const struct dll_sym *sym; + void *addr; + + if (!spec->handle) + spec->handle = LoadLibrary(spec->name); + if (!spec->handle) { + for_each_sym(sym, spec) { + if (sym->required) { + ERROR("%ls could not be loaded!", spec->name); + return WIMLIB_ERR_UNSUPPORTED; + } + } + return 0; + } + for_each_sym(sym, spec) { + addr = (void *)GetProcAddress(spec->handle, sym->name); + if (addr) { + *(sym->func_ptr) = addr; + } else if (sym->required) { + ERROR("Can't find %s in %ls", sym->name, spec->name); + return WIMLIB_ERR_UNSUPPORTED; + } + } + return 0; +} + +static void +cleanup_dll(struct dll_spec *spec) +{ + const struct dll_sym *sym; + + if (spec->handle) { + FreeLibrary(spec->handle); + spec->handle = NULL; + + for_each_sym(sym, spec) + *(sym->func_ptr) = NULL; + } +} + /* One-time initialization for Windows capture/apply code. */ int win32_global_init(int init_flags) { + int ret; + /* Try to acquire useful privileges. */ if (!(init_flags & WIMLIB_INIT_FLAG_DONT_ACQUIRE_PRIVILEGES)) { + ret = WIMLIB_ERR_INSUFFICIENT_PRIVILEGES; if (!win32_modify_capture_privileges(true)) if (init_flags & WIMLIB_INIT_FLAG_STRICT_CAPTURE_PRIVILEGES) - goto insufficient_privileges; + goto out_drop_privs; if (!win32_modify_apply_privileges(true)) if (init_flags & WIMLIB_INIT_FLAG_STRICT_APPLY_PRIVILEGES) - goto insufficient_privileges; + goto out_drop_privs; acquired_privileges = true; } @@ -568,28 +703,21 @@ win32_global_init(int init_flags) GetVersionEx(&windows_version_info); /* Try to dynamically load some functions. */ - if (hKernel32 == NULL) - hKernel32 = LoadLibrary(L"Kernel32.dll"); - - if (hKernel32) { - #ifndef WITH_NTDLL - win32func_FindFirstStreamW = (void*)GetProcAddress(hKernel32, - "FindFirstStreamW"); - if (win32func_FindFirstStreamW) { - win32func_FindNextStreamW = (void*)GetProcAddress(hKernel32, - "FindNextStreamW"); - if (!win32func_FindNextStreamW) - win32func_FindFirstStreamW = NULL; - } - #endif /* !WITH_NTDLL */ - win32func_CreateSymbolicLinkW = (void*)GetProcAddress(hKernel32, - "CreateSymbolicLinkW"); - } + ret = init_dll(&kernel32_spec); + if (ret) + goto out_drop_privs; + + ret = init_dll(&ntdll_spec); + if (ret) + goto out_cleanup_kernel32; + return 0; -insufficient_privileges: +out_cleanup_kernel32: + cleanup_dll(&kernel32_spec); +out_drop_privs: win32_release_capture_and_apply_privileges(); - return WIMLIB_ERR_INSUFFICIENT_PRIVILEGES; + return ret; } void @@ -597,10 +725,9 @@ win32_global_cleanup(void) { if (acquired_privileges) win32_release_capture_and_apply_privileges(); - if (hKernel32 != NULL) { - FreeLibrary(hKernel32); - hKernel32 = NULL; - } + + cleanup_dll(&kernel32_spec); + cleanup_dll(&ntdll_spec); } #endif /* __WIN32__ */