X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Fwin32_capture.c;h=4bccfa1412f6c4902f8551f4cc0e5f806db754fa;hp=27b3794fd436073d23d120800272ec1f50623941;hb=db5f5eee38b76db7f19367d0f4225b36f001325a;hpb=daf72364fc96380cdc0d8ceeeda640d82930f1b9 diff --git a/src/win32_capture.c b/src/win32_capture.c index 27b3794f..4bccfa14 100644 --- a/src/win32_capture.c +++ b/src/win32_capture.c @@ -406,11 +406,18 @@ winnt_recurse_directory(HANDLE h, info->FileName[1] == L'.')) { wchar_t *p; + wchar_t *filename; struct wim_dentry *child; p = full_path + full_path_nchars; - *p++ = L'\\'; - p = wmempcpy(p, info->FileName, + /* Only add a backslash if we don't already have + * one. This prevents a duplicate backslash + * from being added when the path to the capture + * dir had a trailing backslash. */ + if (*(p - 1) != L'\\') + *p++ = L'\\'; + filename = p; + p = wmempcpy(filename, info->FileName, info->FileNameLength / 2); *p = '\0'; @@ -419,7 +426,7 @@ winnt_recurse_directory(HANDLE h, h, full_path, p - full_path, - full_path + full_path_nchars + 1, + filename, info->FileNameLength / 2, params, stats, @@ -453,9 +460,6 @@ out_free_buf: /* Reparse point fixup status code */ enum rp_status { - /* Reparse point should be excluded from capture */ - RP_EXCLUDED = -0, - /* Reparse point will be captured literally (no fixup) */ RP_NOT_FIXED = -1, @@ -650,14 +654,13 @@ winnt_try_rpfix(u8 *rpbuf, u16 *rpbuflen_p, capture_root_ino, capture_root_dev); if (!rel_target) { - /* Target points outside of the tree being captured. Exclude - * this reparse point from the capture (but inform the library - * user). */ + /* Target points outside of the tree being captured. Don't + * adjust it. */ ret = winnt_rpfix_progress(params, path, &rpdata, - WIMLIB_SCAN_DENTRY_EXCLUDED_SYMLINK); + WIMLIB_SCAN_DENTRY_NOT_FIXED_SYMLINK); if (ret) return ret; - return RP_EXCLUDED; + return RP_NOT_FIXED; } if (rel_target == rpdata.substitute_name) { @@ -722,7 +725,7 @@ winnt_try_rpfix(u8 *rpbuf, u16 *rpbuflen_p, * On success, the length of the reparse point buffer in bytes is written * to this location. * - * On success, returns a nonpositive `enum rp_status' value. + * On success, returns a negative `enum rp_status' value. * On failure, returns a positive error code. */ static int @@ -1094,10 +1097,7 @@ winnt_build_dentry_tree_recursive(struct wim_dentry **root_ret, if (should_exclude_path(full_path + params->capture_root_nchars, full_path_nchars - params->capture_root_nchars, params->config)) - { - ret = 0; goto out_progress; - } /* Open the file. */ status = winnt_openat(cur_dir, @@ -1110,11 +1110,19 @@ winnt_build_dentry_tree_recursive(struct wim_dentry **root_ret, SYNCHRONIZE, &h); if (unlikely(!NT_SUCCESS(status))) { - set_errno_from_nt_status(status); - ERROR_WITH_ERRNO("\"%ls\": Can't open file " - "(status=0x%08"PRIx32")", - printable_path(full_path), (u32)status); - ret = WIMLIB_ERR_OPEN; + if (status == STATUS_DELETE_PENDING) { + WARNING("\"%ls\": Deletion pending; skipping file", + printable_path(full_path)); + ret = 0; + } else { + set_errno_from_nt_status(status); + ERROR_WITH_ERRNO("\"%ls\": Can't open file " + "(status=0x%08"PRIx32")", + printable_path(full_path), (u32)status); + ret = WIMLIB_ERR_OPEN; + } + /* XXX: Provide option to exclude files that fail with + * STATUS_SHARING_VIOLATION? */ goto out; } @@ -1204,9 +1212,6 @@ winnt_build_dentry_tree_recursive(struct wim_dentry **root_ret, ret = winnt_get_reparse_data(h, full_path, params, rpbuf, &rpbuflen); switch (ret) { - case RP_EXCLUDED: - ret = 0; - goto out; case RP_FIXED: not_rpfixed = 0; break; @@ -1256,7 +1261,6 @@ winnt_build_dentry_tree_recursive(struct wim_dentry **root_ret, if (inode->i_nlink > 1) { /* Shared inode (hard link); skip reading per-inode information. */ - ret = 0; goto out_progress; } @@ -1389,9 +1393,8 @@ win32_build_dentry_tree(struct wim_dentry **root_ret, struct add_image_params *params) { wchar_t *path; - DWORD dret; - size_t path_nchars; int ret; + UNICODE_STRING ntpath; struct winnt_scan_stats stats; /* WARNING: There is no check for overflow later when this buffer is @@ -1401,33 +1404,31 @@ win32_build_dentry_tree(struct wim_dentry **root_ret, if (!path) return WIMLIB_ERR_NOMEM; - /* Translate into full path. */ - dret = GetFullPathName(root_disk_path, WINDOWS_NT_MAX_PATH - 3, - &path[4], NULL); + ret = win32_path_to_nt_path(root_disk_path, &ntpath); + if (ret) + goto out_free_path; - if (unlikely(dret == 0 || dret >= WINDOWS_NT_MAX_PATH - 3)) { - ERROR("Can't get full path name for \"%ls\"", root_disk_path); - return WIMLIB_ERR_UNSUPPORTED; + if (ntpath.Length < 4 * sizeof(wchar_t) || + ntpath.Length > WINDOWS_NT_MAX_PATH * sizeof(wchar_t) || + wmemcmp(ntpath.Buffer, L"\\??\\", 4)) + { + ERROR("\"%ls\": unrecognized path format", root_disk_path); + ret = WIMLIB_ERR_INVALID_PARAM; + goto out_free_path; } - /* Add \??\ prefix to form the NT namespace path. */ - wmemcpy(path, L"\\??\\", 4); - path_nchars = dret + 4; - - /* Strip trailing slashes. If we don't do this, we may create a path - * with multiple consecutive backslashes, which for some reason causes - * Windows to report that the file cannot be found. */ - while (unlikely(path[path_nchars - 1] == L'\\' && - path[path_nchars - 2] != L':')) - path[--path_nchars] = L'\0'; + params->capture_root_nchars = ntpath.Length / sizeof(wchar_t); + wmemcpy(path, ntpath.Buffer, params->capture_root_nchars); + path[params->capture_root_nchars] = L'\0'; - params->capture_root_nchars = path_nchars; + HeapFree(GetProcessHeap(), 0, ntpath.Buffer); memset(&stats, 0, sizeof(stats)); ret = winnt_build_dentry_tree_recursive(root_ret, NULL, - path, path_nchars, L"", 0, - params, &stats, 0); + path, params->capture_root_nchars, + L"", 0, params, &stats, 0); +out_free_path: FREE(path); if (ret == 0) winnt_do_scan_warnings(root_disk_path, &stats);