wimlib.net Git - wimlib/blob - src/win32_common.c

   1 /*
   2  * win32_common.c - Windows code common to applying and capturing images.
   3  */
   4
   5 /*
   6  * Copyright (C) 2013, 2014 Eric Biggers
   7  *
   8  * This file is part of wimlib, a library for working with WIM files.
   9  *
  10  * wimlib is free software; you can redistribute it and/or modify it under the
  11  * terms of the GNU General Public License as published by the Free
  12  * Software Foundation; either version 3 of the License, or (at your option)
  13  * any later version.
  14  *
  15  * wimlib is distributed in the hope that it will be useful, but WITHOUT ANY
  16  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
  17  * A PARTICULAR PURPOSE. See the GNU General Public License for more
  18  * details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with wimlib; if not, see http://www.gnu.org/licenses/.
  22  */
  23
  24 #ifdef __WIN32__
  25
  26 #ifdef HAVE_CONFIG_H
  27 #  include "config.h"
  28 #endif
  29
  30 #include <errno.h>
  31
  32 #include <winternl.h>
  33
  34 #include "wimlib/win32_common.h"
  35 #include "wimlib/assert.h"
  36 #include "wimlib/error.h"
  37 #include "wimlib/util.h"
  38
  39 static int
  40 win32_error_to_errno(DWORD err_code)
  41 {
  42         /* This mapping is that used in Cygwin.
  43          * Some of these choices are arbitrary. */
  44         switch (err_code) {
  45         case ERROR_ACCESS_DENIED:
  46                 return EACCES;
  47         case ERROR_ACTIVE_CONNECTIONS:
  48                 return EAGAIN;
  49         case ERROR_ALREADY_EXISTS:
  50                 return EEXIST;
  51         case ERROR_BAD_DEVICE:
  52                 return ENODEV;
  53         case ERROR_BAD_EXE_FORMAT:
  54                 return ENOEXEC;
  55         case ERROR_BAD_NETPATH:
  56                 return ENOENT;
  57         case ERROR_BAD_NET_NAME:
  58                 return ENOENT;
  59         case ERROR_BAD_NET_RESP:
  60                 return ENOSYS;
  61         case ERROR_BAD_PATHNAME:
  62                 return ENOENT;
  63         case ERROR_BAD_PIPE:
  64                 return EINVAL;
  65         case ERROR_BAD_UNIT:
  66                 return ENODEV;
  67         case ERROR_BAD_USERNAME:
  68                 return EINVAL;
  69         case ERROR_BEGINNING_OF_MEDIA:
  70                 return EIO;
  71         case ERROR_BROKEN_PIPE:
  72                 return EPIPE;
  73         case ERROR_BUSY:
  74                 return EBUSY;
  75         case ERROR_BUS_RESET:
  76                 return EIO;
  77         case ERROR_CALL_NOT_IMPLEMENTED:
  78                 return ENOSYS;
  79         case ERROR_CANNOT_MAKE:
  80                 return EPERM;
  81         case ERROR_CHILD_NOT_COMPLETE:
  82                 return EBUSY;
  83         case ERROR_COMMITMENT_LIMIT:
  84                 return EAGAIN;
  85         case ERROR_CRC:
  86                 return EIO;
  87         case ERROR_DEVICE_DOOR_OPEN:
  88                 return EIO;
  89         case ERROR_DEVICE_IN_USE:
  90                 return EAGAIN;
  91         case ERROR_DEVICE_REQUIRES_CLEANING:
  92                 return EIO;
  93         case ERROR_DIRECTORY:
  94                 return ENOTDIR;
  95         case ERROR_DIR_NOT_EMPTY:
  96                 return ENOTEMPTY;
  97         case ERROR_DISK_CORRUPT:
  98                 return EIO;
  99         case ERROR_DISK_FULL:
 100                 return ENOSPC;
 101 #ifdef ENOTUNIQ
 102         case ERROR_DUP_NAME:
 103                 return ENOTUNIQ;
 104 #endif
 105         case ERROR_EAS_DIDNT_FIT:
 106                 return ENOSPC;
 107 #ifdef ENOTSUP
 108         case ERROR_EAS_NOT_SUPPORTED:
 109                 return ENOTSUP;
 110 #endif
 111         case ERROR_EA_LIST_INCONSISTENT:
 112                 return EINVAL;
 113         case ERROR_EA_TABLE_FULL:
 114                 return ENOSPC;
 115         case ERROR_END_OF_MEDIA:
 116                 return ENOSPC;
 117         case ERROR_EOM_OVERFLOW:
 118                 return EIO;
 119         case ERROR_EXE_MACHINE_TYPE_MISMATCH:
 120                 return ENOEXEC;
 121         case ERROR_EXE_MARKED_INVALID:
 122                 return ENOEXEC;
 123         case ERROR_FILEMARK_DETECTED:
 124                 return EIO;
 125         case ERROR_FILENAME_EXCED_RANGE:
 126                 return ENAMETOOLONG;
 127         case ERROR_FILE_CORRUPT:
 128                 return EEXIST;
 129         case ERROR_FILE_EXISTS:
 130                 return EEXIST;
 131         case ERROR_FILE_INVALID:
 132                 return ENXIO;
 133         case ERROR_FILE_NOT_FOUND:
 134                 return ENOENT;
 135         case ERROR_HANDLE_DISK_FULL:
 136                 return ENOSPC;
 137 #ifdef ENODATA
 138         case ERROR_HANDLE_EOF:
 139                 return ENODATA;
 140 #endif
 141         case ERROR_INVALID_ADDRESS:
 142                 return EINVAL;
 143         case ERROR_INVALID_AT_INTERRUPT_TIME:
 144                 return EINTR;
 145         case ERROR_INVALID_BLOCK_LENGTH:
 146                 return EIO;
 147         case ERROR_INVALID_DATA:
 148                 return EINVAL;
 149         case ERROR_INVALID_DRIVE:
 150                 return ENODEV;
 151         case ERROR_INVALID_EA_NAME:
 152                 return EINVAL;
 153         case ERROR_INVALID_EXE_SIGNATURE:
 154                 return ENOEXEC;
 155 #ifdef EBADRQC
 156         case ERROR_INVALID_FUNCTION:
 157                 return EBADRQC;
 158 #endif
 159         case ERROR_INVALID_HANDLE:
 160                 return EBADF;
 161         case ERROR_INVALID_NAME:
 162                 return ENOENT;
 163         case ERROR_INVALID_PARAMETER:
 164                 return EINVAL;
 165         case ERROR_INVALID_SIGNAL_NUMBER:
 166                 return EINVAL;
 167         case ERROR_IOPL_NOT_ENABLED:
 168                 return ENOEXEC;
 169         case ERROR_IO_DEVICE:
 170                 return EIO;
 171         case ERROR_IO_INCOMPLETE:
 172                 return EAGAIN;
 173         case ERROR_IO_PENDING:
 174                 return EAGAIN;
 175         case ERROR_LOCK_VIOLATION:
 176                 return EBUSY;
 177         case ERROR_MAX_THRDS_REACHED:
 178                 return EAGAIN;
 179         case ERROR_META_EXPANSION_TOO_LONG:
 180                 return EINVAL;
 181         case ERROR_MOD_NOT_FOUND:
 182                 return ENOENT;
 183 #ifdef EMSGSIZE
 184         case ERROR_MORE_DATA:
 185                 return EMSGSIZE;
 186 #endif
 187         case ERROR_NEGATIVE_SEEK:
 188                 return EINVAL;
 189         case ERROR_NETNAME_DELETED:
 190                 return ENOENT;
 191         case ERROR_NOACCESS:
 192                 return EFAULT;
 193         case ERROR_NONE_MAPPED:
 194                 return EINVAL;
 195         case ERROR_NONPAGED_SYSTEM_RESOURCES:
 196                 return EAGAIN;
 197 #ifdef ENOLINK
 198         case ERROR_NOT_CONNECTED:
 199                 return ENOLINK;
 200 #endif
 201         case ERROR_NOT_ENOUGH_MEMORY:
 202                 return ENOMEM;
 203         case ERROR_NOT_OWNER:
 204                 return EPERM;
 205 #ifdef ENOMEDIUM
 206         case ERROR_NOT_READY:
 207                 return ENOMEDIUM;
 208 #endif
 209         case ERROR_NOT_SAME_DEVICE:
 210                 return EXDEV;
 211         case ERROR_NOT_SUPPORTED:
 212                 return ENOSYS;
 213         case ERROR_NO_DATA:
 214                 return EPIPE;
 215         case ERROR_NO_DATA_DETECTED:
 216                 return EIO;
 217 #ifdef ENOMEDIUM
 218         case ERROR_NO_MEDIA_IN_DRIVE:
 219                 return ENOMEDIUM;
 220 #endif
 221 #ifdef ENMFILE
 222         case ERROR_NO_MORE_FILES:
 223                 return ENMFILE;
 224 #endif
 225 #ifdef ENMFILE
 226         case ERROR_NO_MORE_ITEMS:
 227                 return ENMFILE;
 228 #endif
 229         case ERROR_NO_MORE_SEARCH_HANDLES:
 230                 return ENFILE;
 231         case ERROR_NO_PROC_SLOTS:
 232                 return EAGAIN;
 233         case ERROR_NO_SIGNAL_SENT:
 234                 return EIO;
 235         case ERROR_NO_SYSTEM_RESOURCES:
 236                 return EFBIG;
 237         case ERROR_NO_TOKEN:
 238                 return EINVAL;
 239         case ERROR_OPEN_FAILED:
 240                 return EIO;
 241         case ERROR_OPEN_FILES:
 242                 return EAGAIN;
 243         case ERROR_OUTOFMEMORY:
 244                 return ENOMEM;
 245         case ERROR_PAGED_SYSTEM_RESOURCES:
 246                 return EAGAIN;
 247         case ERROR_PAGEFILE_QUOTA:
 248                 return EAGAIN;
 249         case ERROR_PATH_NOT_FOUND:
 250                 return ENOENT;
 251         case ERROR_PIPE_BUSY:
 252                 return EBUSY;
 253         case ERROR_PIPE_CONNECTED:
 254                 return EBUSY;
 255 #ifdef ECOMM
 256         case ERROR_PIPE_LISTENING:
 257                 return ECOMM;
 258         case ERROR_PIPE_NOT_CONNECTED:
 259                 return ECOMM;
 260 #endif
 261         case ERROR_POSSIBLE_DEADLOCK:
 262                 return EDEADLOCK;
 263         case ERROR_PRIVILEGE_NOT_HELD:
 264                 return EPERM;
 265         case ERROR_PROCESS_ABORTED:
 266                 return EFAULT;
 267         case ERROR_PROC_NOT_FOUND:
 268                 return ESRCH;
 269 #ifdef ENONET
 270         case ERROR_REM_NOT_LIST:
 271                 return ENONET;
 272 #endif
 273         case ERROR_SECTOR_NOT_FOUND:
 274                 return EINVAL;
 275         case ERROR_SEEK:
 276                 return EINVAL;
 277         case ERROR_SETMARK_DETECTED:
 278                 return EIO;
 279         case ERROR_SHARING_BUFFER_EXCEEDED:
 280                 return ENOLCK;
 281         case ERROR_SHARING_VIOLATION:
 282                 return EBUSY;
 283         case ERROR_SIGNAL_PENDING:
 284                 return EBUSY;
 285         case ERROR_SIGNAL_REFUSED:
 286                 return EIO;
 287 #ifdef ELIBBAD
 288         case ERROR_SXS_CANT_GEN_ACTCTX:
 289                 return ELIBBAD;
 290 #endif
 291         case ERROR_THREAD_1_INACTIVE:
 292                 return EINVAL;
 293         case ERROR_TOO_MANY_LINKS:
 294                 return EMLINK;
 295         case ERROR_TOO_MANY_OPEN_FILES:
 296                 return EMFILE;
 297         case ERROR_WAIT_NO_CHILDREN:
 298                 return ECHILD;
 299         case ERROR_WORKING_SET_QUOTA:
 300                 return EAGAIN;
 301         case ERROR_WRITE_PROTECT:
 302                 return EROFS;
 303         default:
 304                 return -1;
 305         }
 306 }
 307
 308
 309 void
 310 set_errno_from_win32_error(DWORD err)
 311 {
 312         errno = win32_error_to_errno(err);
 313 }
 314
 315 void
 316 set_errno_from_GetLastError(void)
 317 {
 318         set_errno_from_win32_error(GetLastError());
 319 }
 320
 321 void
 322 set_errno_from_nt_status(NTSTATUS status)
 323 {
 324         set_errno_from_win32_error((*func_RtlNtStatusToDosError)(status));
 325 }
 326
 327 /* Given a Windows-style path, return the number of characters of the prefix
 328  * that specify the path to the root directory of a drive, or return 0 if the
 329  * drive is relative (or at least on the current drive, in the case of
 330  * absolute-but-not-really-absolute paths like \Windows\System32) */
 331 static size_t
 332 win32_path_drive_spec_len(const wchar_t *path)
 333 {
 334         size_t n = 0;
 335
 336         if (!wcsncmp(path, L"\\\\?\\", 4)) {
 337                 /* \\?\-prefixed path.  Check for following drive letter and
 338                  * path separator. */
 339                 if (path[4] != L'\0' && path[5] == L':' &&
 340                     is_any_path_separator(path[6]))
 341                         n = 7;
 342         } else {
 343                 /* Not a \\?\-prefixed path.  Check for an initial drive letter
 344                  * and path separator. */
 345                 if (path[0] != L'\0' && path[1] == L':' &&
 346                     is_any_path_separator(path[2]))
 347                         n = 3;
 348         }
 349         /* Include any additional path separators.*/
 350         if (n > 0)
 351                 while (is_any_path_separator(path[n]))
 352                         n++;
 353         return n;
 354 }
 355
 356 bool
 357 win32_path_is_root_of_drive(const wchar_t *path)
 358 {
 359         size_t drive_spec_len;
 360         wchar_t full_path[32768];
 361         DWORD ret;
 362
 363         ret = GetFullPathName(path, ARRAY_LEN(full_path), full_path, NULL);
 364         if (ret > 0 && ret < ARRAY_LEN(full_path))
 365                 path = full_path;
 366
 367         /* Explicit drive letter and path separator? */
 368         drive_spec_len = win32_path_drive_spec_len(path);
 369         if (drive_spec_len > 0 && path[drive_spec_len] == L'\0')
 370                 return true;
 371
 372         /* All path separators? */
 373         for (const wchar_t *p = path; *p != L'\0'; p++)
 374                 if (!is_any_path_separator(*p))
 375                         return false;
 376         return true;
 377 }
 378
 379
 380 /* Given a path, which may not yet exist, get a set of flags that describe the
 381  * features of the volume the path is on. */
 382 int
 383 win32_get_vol_flags(const wchar_t *path, unsigned *vol_flags_ret,
 384                     bool *supports_SetFileShortName_ret)
 385 {
 386         wchar_t *volume;
 387         BOOL bret;
 388         DWORD vol_flags;
 389         size_t drive_spec_len;
 390         wchar_t filesystem_name[MAX_PATH + 1];
 391
 392         if (supports_SetFileShortName_ret)
 393                 *supports_SetFileShortName_ret = false;
 394
 395         drive_spec_len = win32_path_drive_spec_len(path);
 396
 397         if (drive_spec_len == 0)
 398                 if (path[0] != L'\0' && path[1] == L':') /* Drive-relative path? */
 399                         drive_spec_len = 2;
 400
 401         if (drive_spec_len == 0) {
 402                 /* Path does not start with a drive letter; use the volume of
 403                  * the current working directory. */
 404                 volume = NULL;
 405         } else {
 406                 /* Path starts with a drive letter (or \\?\ followed by a drive
 407                  * letter); use it. */
 408                 volume = alloca((drive_spec_len + 2) * sizeof(wchar_t));
 409                 wmemcpy(volume, path, drive_spec_len);
 410                 /* Add trailing backslash in case this was a drive-relative
 411                  * path. */
 412                 volume[drive_spec_len] = L'\\';
 413                 volume[drive_spec_len + 1] = L'\0';
 414         }
 415         bret = GetVolumeInformation(
 416                         volume,                         /* lpRootPathName */
 417                         NULL,                           /* lpVolumeNameBuffer */
 418                         0,                              /* nVolumeNameSize */
 419                         NULL,                           /* lpVolumeSerialNumber */
 420                         NULL,                           /* lpMaximumComponentLength */
 421                         &vol_flags,                     /* lpFileSystemFlags */
 422                         filesystem_name,                /* lpFileSystemNameBuffer */
 423                         ARRAY_LEN(filesystem_name));    /* nFileSystemNameSize */
 424         if (!bret) {
 425                 set_errno_from_GetLastError();
 426                 WARNING_WITH_ERRNO("Failed to get volume information for "
 427                                    "path \"%ls\"", path);
 428                 vol_flags = 0xffffffff;
 429                 goto out;
 430         }
 431
 432         if (wcsstr(filesystem_name, L"NTFS")) {
 433                 /* FILE_SUPPORTS_HARD_LINKS is only supported on Windows 7 and later.
 434                  * Force it on anyway if filesystem is NTFS.  */
 435                 vol_flags |= FILE_SUPPORTS_HARD_LINKS;
 436
 437                 if (supports_SetFileShortName_ret)
 438                         *supports_SetFileShortName_ret = true;
 439         }
 440
 441 out:
 442         DEBUG("using vol_flags = %x", vol_flags);
 443         *vol_flags_ret = vol_flags;
 444         return 0;
 445 }
 446
 447 static bool
 448 win32_modify_privilege(const wchar_t *privilege, bool enable)
 449 {
 450         HANDLE hToken;
 451         LUID luid;
 452         TOKEN_PRIVILEGES newState;
 453         bool ret = FALSE;
 454
 455         if (!OpenProcessToken(GetCurrentProcess(),
 456                               TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,
 457                               &hToken))
 458                 goto out;
 459
 460         if (!LookupPrivilegeValue(NULL, privilege, &luid))
 461                 goto out_close_handle;
 462
 463         newState.PrivilegeCount = 1;
 464         newState.Privileges[0].Luid = luid;
 465         newState.Privileges[0].Attributes = (enable ? SE_PRIVILEGE_ENABLED : 0);
 466         SetLastError(ERROR_SUCCESS);
 467         ret = AdjustTokenPrivileges(hToken, FALSE, &newState, 0, NULL, NULL);
 468         if (ret && GetLastError() == ERROR_NOT_ALL_ASSIGNED)
 469                 ret = FALSE;
 470 out_close_handle:
 471         CloseHandle(hToken);
 472 out:
 473         return ret;
 474 }
 475
 476 static bool
 477 win32_modify_capture_privileges(bool enable)
 478 {
 479         return win32_modify_privilege(SE_BACKUP_NAME, enable)
 480             && win32_modify_privilege(SE_SECURITY_NAME, enable);
 481 }
 482
 483 static bool
 484 win32_modify_apply_privileges(bool enable)
 485 {
 486         return win32_modify_privilege(SE_RESTORE_NAME, enable)
 487             && win32_modify_privilege(SE_SECURITY_NAME, enable)
 488             && win32_modify_privilege(SE_TAKE_OWNERSHIP_NAME, enable);
 489 }
 490
 491 static void
 492 win32_release_capture_and_apply_privileges(void)
 493 {
 494         win32_modify_capture_privileges(false);
 495         win32_modify_apply_privileges(false);
 496 }
 497
 498 HANDLE
 499 win32_open_existing_file(const wchar_t *path, DWORD dwDesiredAccess)
 500 {
 501         return CreateFile(path,
 502                           dwDesiredAccess,
 503                           FILE_SHARE_READ,
 504                           NULL, /* lpSecurityAttributes */
 505                           OPEN_EXISTING,
 506                           FILE_FLAG_BACKUP_SEMANTICS |
 507                                 FILE_FLAG_OPEN_REPARSE_POINT,
 508                           NULL /* hTemplateFile */);
 509 }
 510
 511 /* Pointers to dynamically loaded functions  */
 512
 513 /* kernel32.dll:  Vista and later */
 514 BOOL (WINAPI *func_CreateSymbolicLinkW)(const wchar_t *lpSymlinkFileName,
 515                                         const wchar_t *lpTargetFileName,
 516                                         DWORD dwFlags);
 517
 518 /* ntdll.dll  */
 519
 520 NTSTATUS (WINAPI *func_NtOpenFile) (PHANDLE FileHandle,
 521                                     ACCESS_MASK DesiredAccess,
 522                                     POBJECT_ATTRIBUTES ObjectAttributes,
 523                                     PIO_STATUS_BLOCK IoStatusBlock,
 524                                     ULONG ShareAccess,
 525                                     ULONG OpenOptions);
 526
 527 NTSTATUS (WINAPI *func_NtReadFile) (HANDLE FileHandle,
 528                                     HANDLE Event,
 529                                     PIO_APC_ROUTINE ApcRoutine,
 530                                     PVOID ApcContext,
 531                                     PIO_STATUS_BLOCK IoStatusBlock,
 532                                     PVOID Buffer,
 533                                     ULONG Length,
 534                                     PLARGE_INTEGER ByteOffset,
 535                                     PULONG Key);
 536
 537 NTSTATUS (WINAPI *func_NtOpenFile) (PHANDLE FileHandle,
 538                                     ACCESS_MASK DesiredAccess,
 539                                     POBJECT_ATTRIBUTES ObjectAttributes,
 540                                     PIO_STATUS_BLOCK IoStatusBlock,
 541                                     ULONG ShareAccess,
 542                                     ULONG OpenOptions);
 543
 544 NTSTATUS (WINAPI *func_NtQueryInformationFile)(HANDLE FileHandle,
 545                                                PIO_STATUS_BLOCK IoStatusBlock,
 546                                                PVOID FileInformation,
 547                                                ULONG Length,
 548                                                FILE_INFORMATION_CLASS FileInformationClass);
 549
 550 NTSTATUS (WINAPI *func_NtQuerySecurityObject)(HANDLE handle,
 551                                               SECURITY_INFORMATION SecurityInformation,
 552                                               PSECURITY_DESCRIPTOR SecurityDescriptor,
 553                                               ULONG Length,
 554                                               PULONG LengthNeeded);
 555
 556 NTSTATUS (WINAPI *func_NtQueryDirectoryFile) (HANDLE FileHandle,
 557                                               HANDLE Event,
 558                                               PIO_APC_ROUTINE ApcRoutine,
 559                                               PVOID ApcContext,
 560                                               PIO_STATUS_BLOCK IoStatusBlock,
 561                                               PVOID FileInformation,
 562                                               ULONG Length,
 563                                               FILE_INFORMATION_CLASS FileInformationClass,
 564                                               BOOLEAN ReturnSingleEntry,
 565                                               PUNICODE_STRING FileName,
 566                                               BOOLEAN RestartScan);
 567
 568 NTSTATUS (WINAPI *func_NtSetSecurityObject)(HANDLE Handle,
 569                                             SECURITY_INFORMATION SecurityInformation,
 570                                             PSECURITY_DESCRIPTOR SecurityDescriptor);
 571
 572 NTSTATUS (WINAPI *func_NtClose) (HANDLE Handle);
 573
 574 DWORD (WINAPI *func_RtlNtStatusToDosError)(NTSTATUS status);
 575
 576 NTSTATUS (WINAPI *func_RtlCreateSystemVolumeInformationFolder)
 577                 (PCUNICODE_STRING VolumeRootPath);
 578
 579 static OSVERSIONINFO windows_version_info = {
 580         .dwOSVersionInfoSize = sizeof(OSVERSIONINFO),
 581 };
 582
 583 static bool acquired_privileges = false;
 584
 585 bool
 586 windows_version_is_at_least(unsigned major, unsigned minor)
 587 {
 588         return windows_version_info.dwMajorVersion > major ||
 589                 (windows_version_info.dwMajorVersion == major &&
 590                  windows_version_info.dwMinorVersion >= minor);
 591 }
 592
 593 struct dll_sym {
 594         void **func_ptr;
 595         const char *name;
 596         bool required;
 597 };
 598
 599 #define DLL_SYM(name, required) { (void **)&func_##name, #name, required }
 600
 601 #define for_each_sym(sym, spec) \
 602         for ((sym) = (spec)->syms; (sym)->name; (sym)++)
 603
 604 struct dll_spec {
 605         const wchar_t *name;
 606         HMODULE handle;
 607         const struct dll_sym syms[];
 608 };
 609
 610 struct dll_spec ntdll_spec = {
 611         .name = L"ntdll.dll",
 612         .syms = {
 613                 DLL_SYM(NtOpenFile, true),
 614                 DLL_SYM(NtReadFile, true),
 615                 DLL_SYM(NtQueryInformationFile, true),
 616                 DLL_SYM(NtQuerySecurityObject, true),
 617                 DLL_SYM(NtQueryDirectoryFile, true),
 618                 DLL_SYM(NtSetSecurityObject, true),
 619                 DLL_SYM(NtClose, true),
 620                 DLL_SYM(RtlNtStatusToDosError, true),
 621                 DLL_SYM(RtlCreateSystemVolumeInformationFolder, false),
 622                 {NULL, NULL},
 623         },
 624 };
 625
 626 struct dll_spec kernel32_spec = {
 627         .name = L"kernel32.dll",
 628         .syms = {
 629                 DLL_SYM(CreateSymbolicLinkW, false),
 630                 {NULL, NULL},
 631         },
 632 };
 633
 634 static int
 635 init_dll(struct dll_spec *spec)
 636 {
 637         const struct dll_sym *sym;
 638         void *addr;
 639
 640         if (!spec->handle)
 641                 spec->handle = LoadLibrary(spec->name);
 642         if (!spec->handle) {
 643                 for_each_sym(sym, spec) {
 644                         if (sym->required) {
 645                                 ERROR("%ls could not be loaded!", spec->name);
 646                                 return WIMLIB_ERR_UNSUPPORTED;
 647                         }
 648                 }
 649                 return 0;
 650         }
 651         for_each_sym(sym, spec) {
 652                 addr = (void *)GetProcAddress(spec->handle, sym->name);
 653                 if (addr) {
 654                         *(sym->func_ptr) = addr;
 655                 } else if (sym->required) {
 656                         ERROR("Can't find %s in %ls", sym->name, spec->name);
 657                         return WIMLIB_ERR_UNSUPPORTED;
 658                 }
 659         }
 660         return 0;
 661 }
 662
 663 static void
 664 cleanup_dll(struct dll_spec *spec)
 665 {
 666         const struct dll_sym *sym;
 667
 668         if (spec->handle) {
 669                 FreeLibrary(spec->handle);
 670                 spec->handle = NULL;
 671
 672                 for_each_sym(sym, spec)
 673                         *(sym->func_ptr) = NULL;
 674         }
 675 }
 676
 677 /* One-time initialization for Windows capture/apply code.  */
 678 int
 679 win32_global_init(int init_flags)
 680 {
 681         int ret;
 682
 683         /* Try to acquire useful privileges.  */
 684         if (!(init_flags & WIMLIB_INIT_FLAG_DONT_ACQUIRE_PRIVILEGES)) {
 685                 ret = WIMLIB_ERR_INSUFFICIENT_PRIVILEGES;
 686                 if (!win32_modify_capture_privileges(true))
 687                         if (init_flags & WIMLIB_INIT_FLAG_STRICT_CAPTURE_PRIVILEGES)
 688                                 goto out_drop_privs;
 689                 if (!win32_modify_apply_privileges(true))
 690                         if (init_flags & WIMLIB_INIT_FLAG_STRICT_APPLY_PRIVILEGES)
 691                                 goto out_drop_privs;
 692                 acquired_privileges = true;
 693         }
 694
 695         /* Get Windows version information.  */
 696         GetVersionEx(&windows_version_info);
 697
 698         /* Try to dynamically load some functions.  */
 699         ret = init_dll(&kernel32_spec);
 700         if (ret)
 701                 goto out_drop_privs;
 702
 703         ret = init_dll(&ntdll_spec);
 704         if (ret)
 705                 goto out_cleanup_kernel32;
 706
 707         return 0;
 708
 709 out_cleanup_kernel32:
 710         cleanup_dll(&kernel32_spec);
 711 out_drop_privs:
 712         win32_release_capture_and_apply_privileges();
 713         return ret;
 714 }
 715
 716 void
 717 win32_global_cleanup(void)
 718 {
 719         if (acquired_privileges)
 720                 win32_release_capture_and_apply_privileges();
 721
 722         cleanup_dll(&kernel32_spec);
 723         cleanup_dll(&ntdll_spec);
 724 }
 725
 726 #endif /* __WIN32__ */