Factor out Win32 => NT path conversion and use in win32_capture.c

[wimlib] / src / win32_common.c

diff --git a/src/win32_common.c b/src/win32_common.c
index 95d41b87c07d8f1ade69b886da9b9e3720c24f97..412421ea28ce1480cb56bcea651c2b6896e038e4 100644 (file)
--- a/src/win32_common.c
+++ b/src/win32_common.c
@@ -371,28 +371,22 @@ win32_release_capture_and_apply_privileges(void)
        win32_modify_apply_privileges(false);
 }
 
-HANDLE
-win32_open_existing_file(const wchar_t *path, DWORD dwDesiredAccess)
-{
-       return CreateFile(path,
-                         dwDesiredAccess,
-                         FILE_SHARE_READ,
-                         NULL, /* lpSecurityAttributes */
-                         OPEN_EXISTING,
-                         FILE_FLAG_BACKUP_SEMANTICS |
-                               FILE_FLAG_OPEN_REPARSE_POINT,
-                         NULL /* hTemplateFile */);
-}
-
 /* 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_NtCreateFile)(PHANDLE FileHandle,
+                                    ACCESS_MASK DesiredAccess,
+                                    POBJECT_ATTRIBUTES ObjectAttributes,
+                                    PIO_STATUS_BLOCK IoStatusBlock,
+                                    PLARGE_INTEGER AllocationSize,
+                                    ULONG FileAttributes,
+                                    ULONG ShareAccess,
+                                    ULONG CreateDisposition,
+                                    ULONG CreateOptions,
+                                    PVOID EaBuffer,
+                                    ULONG EaLength);
+
 NTSTATUS (WINAPI *func_NtOpenFile) (PHANDLE FileHandle,
                                    ACCESS_MASK DesiredAccess,
                                    POBJECT_ATTRIBUTES ObjectAttributes,
@@ -410,12 +404,15 @@ NTSTATUS (WINAPI *func_NtReadFile) (HANDLE FileHandle,
                                    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_NtWriteFile) (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_NtQueryInformationFile)(HANDLE FileHandle,
                                               PIO_STATUS_BLOCK IoStatusBlock,
@@ -447,14 +444,43 @@ NTSTATUS (WINAPI *func_NtQueryVolumeInformationFile) (HANDLE FileHandle,
                                                      ULONG Length,
                                                      FS_INFORMATION_CLASS FsInformationClass);
 
+NTSTATUS (WINAPI *func_NtSetInformationFile)(HANDLE FileHandle,
+                                            PIO_STATUS_BLOCK IoStatusBlock,
+                                            PVOID FileInformation,
+                                            ULONG Length,
+                                            FILE_INFORMATION_CLASS FileInformationClass);
+
 NTSTATUS (WINAPI *func_NtSetSecurityObject)(HANDLE Handle,
                                            SECURITY_INFORMATION SecurityInformation,
                                            PSECURITY_DESCRIPTOR SecurityDescriptor);
 
+NTSTATUS (WINAPI *func_NtFsControlFile) (HANDLE FileHandle,
+                                        HANDLE Event,
+                                        PIO_APC_ROUTINE ApcRoutine,
+                                        PVOID ApcContext,
+                                        PIO_STATUS_BLOCK IoStatusBlock,
+                                        ULONG FsControlCode,
+                                        PVOID InputBuffer,
+                                        ULONG InputBufferLength,
+                                        PVOID OutputBuffer,
+                                        ULONG OutputBufferLength);
+
 NTSTATUS (WINAPI *func_NtClose) (HANDLE Handle);
 
 DWORD (WINAPI *func_RtlNtStatusToDosError)(NTSTATUS status);
 
+BOOLEAN (WINAPI *func_RtlDosPathNameToNtPathName_U)
+                 (IN PCWSTR DosName,
+                  OUT PUNICODE_STRING NtName,
+                  OUT PCWSTR *PartName,
+                  OUT PRTL_RELATIVE_NAME_U RelativeName);
+
+NTSTATUS (WINAPI *func_RtlDosPathNameToNtPathName_U_WithStatus)
+               (IN PCWSTR DosName,
+                OUT PUNICODE_STRING NtName,
+                OUT PCWSTR *PartName,
+                OUT PRTL_RELATIVE_NAME_U RelativeName);
+
 NTSTATUS (WINAPI *func_RtlCreateSystemVolumeInformationFolder)
                (PCUNICODE_STRING VolumeRootPath);
 
@@ -492,24 +518,22 @@ struct dll_spec {
 struct dll_spec ntdll_spec = {
        .name = L"ntdll.dll",
        .syms = {
+               DLL_SYM(NtCreateFile, true),
                DLL_SYM(NtOpenFile, true),
                DLL_SYM(NtReadFile, true),
+               DLL_SYM(NtWriteFile, true),
                DLL_SYM(NtQueryInformationFile, true),
                DLL_SYM(NtQuerySecurityObject, true),
                DLL_SYM(NtQueryDirectoryFile, true),
                DLL_SYM(NtQueryVolumeInformationFile, true),
+               DLL_SYM(NtSetInformationFile, true),
                DLL_SYM(NtSetSecurityObject, true),
+               DLL_SYM(NtFsControlFile, 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),
+               DLL_SYM(RtlDosPathNameToNtPathName_U, true),
+               DLL_SYM(RtlDosPathNameToNtPathName_U_WithStatus, false), /* Not present on XP  */
                {NULL, NULL},
        },
 };
@@ -578,19 +602,12 @@ win32_global_init(int init_flags)
        /* Get Windows version information.  */
        GetVersionEx(&windows_version_info);
 
-       /* Try to dynamically load some functions.  */
-       ret = init_dll(&kernel32_spec);
-       if (ret)
-               goto out_drop_privs;
-
        ret = init_dll(&ntdll_spec);
        if (ret)
-               goto out_cleanup_kernel32;
+               goto out_drop_privs;
 
        return 0;
 
-out_cleanup_kernel32:
-       cleanup_dll(&kernel32_spec);
 out_drop_privs:
        win32_release_capture_and_apply_privileges();
        return ret;
@@ -602,8 +619,45 @@ win32_global_cleanup(void)
        if (acquired_privileges)
                win32_release_capture_and_apply_privileges();
 
-       cleanup_dll(&kernel32_spec);
        cleanup_dll(&ntdll_spec);
 }
 
+/*
+ * Translates a Win32-namespace path into an NT-namespace path.
+ *
+ * On success, returns 0.  The NT-namespace path will be stored in the
+ * UNICODE_STRING structure pointed to by nt_path.  nt_path->Buffer will be set
+ * to a new buffer that must later be freed with HeapFree().  (Really
+ * RtlHeapFree(), but HeapFree() seems to be the same thing.)
+ *
+ * On failure, returns WIMLIB_ERR_NOMEM or WIMLIB_ERR_INVALID_PARAM.
+ */
+int
+win32_path_to_nt_path(const wchar_t *win32_path, UNICODE_STRING *nt_path)
+{
+       NTSTATUS status;
+
+       if (func_RtlDosPathNameToNtPathName_U_WithStatus) {
+               status = (*func_RtlDosPathNameToNtPathName_U_WithStatus)(win32_path,
+                                                                        nt_path,
+                                                                        NULL, NULL);
+       } else {
+               if ((*func_RtlDosPathNameToNtPathName_U)(win32_path, nt_path,
+                                                        NULL, NULL))
+                       status = STATUS_SUCCESS;
+               else
+                       status = STATUS_NO_MEMORY;
+       }
+
+       if (likely(NT_SUCCESS(status)))
+               return 0;
+
+       if (status == STATUS_NO_MEMORY)
+               return WIMLIB_ERR_NOMEM;
+
+       ERROR("\"%ls\": invalid path name (status=0x%08"PRIx32")",
+             win32_path, (u32)status);
+       return WIMLIB_ERR_INVALID_PARAM;
+}
+
 #endif /* __WIN32__ */