X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Funix_capture.c;h=dd79b658e0238cc9f15ed3d039ef1285e2dc9015;hp=f3e9f0e564866f3b7b760a39f5f64627ec5d34dc;hb=5d3d469e410dc5f4a28814ad231336fc174cba56;hpb=bb40342796df9c677f6903b596abf4e9e5769845 diff --git a/src/unix_capture.c b/src/unix_capture.c index f3e9f0e5..dd79b658 100644 --- a/src/unix_capture.c +++ b/src/unix_capture.c @@ -1,5 +1,9 @@ /* - * Copyright (C) 2013 Eric Biggers + * unix_capture.c: Capture a directory tree on UNIX. + */ + +/* + * Copyright (C) 2012, 2013 Eric Biggers * * This file is part of wimlib, a library for working with WIM files. * @@ -19,21 +23,29 @@ #ifndef __WIN32__ -#include "wimlib_internal.h" -#include "dentry.h" -#include "lookup_table.h" -#include "timestamp.h" +#ifdef HAVE_CONFIG_H +# include "config.h" +#endif #include #include -#include +#include #include +#include + +#include "wimlib/capture.h" +#include "wimlib/dentry.h" +#include "wimlib/error.h" +#include "wimlib/lookup_table.h" +#include "wimlib/paths.h" +#include "wimlib/reparse.h" +#include "wimlib/timestamp.h" static int unix_capture_regular_file(const char *path, u64 size, struct wim_inode *inode, - struct wim_lookup_table *lookup_table) + struct list_head *unhashed_streams) { inode->i_attributes = FILE_ATTRIBUTE_NORMAL; @@ -52,8 +64,8 @@ unix_capture_regular_file(const char *path, } lte->file_on_disk = file_on_disk; lte->resource_location = RESOURCE_IN_FILE_ON_DISK; - lte->resource_entry.original_size = size; - lookup_table_insert_unhashed(lookup_table, lte, inode, 0); + lte->size = size; + add_unhashed_stream(lte, inode, 0, unhashed_streams); inode->i_lte = lte; } return 0; @@ -82,7 +94,7 @@ unix_capture_directory(struct wim_dentry *dir_dentry, if (!dir) { ERROR_WITH_ERRNO("Failed to open the directory `%s'", path); - return WIMLIB_ERR_OPEN; + return WIMLIB_ERR_OPENDIR; } /* Recurse on directory contents */ @@ -116,6 +128,7 @@ unix_capture_directory(struct wim_dentry *dir_dentry, if (child) dentry_add_child(dir_dentry, child); } + path[path_len] = '\0'; closedir(dir); return ret; } @@ -133,12 +146,9 @@ unix_capture_symlink(struct wim_dentry **root_p, inode->i_attributes = FILE_ATTRIBUTE_REPARSE_POINT; inode->i_reparse_tag = WIM_IO_REPARSE_TAG_SYMLINK; - /* The idea here is to call readlink() to get the UNIX target of - * the symbolic link, then turn the target into a reparse point - * data buffer that contains a relative or absolute symbolic - * link (NOT a junction point or *full* path symbolic link with - * drive letter). - */ + /* The idea here is to call readlink() to get the UNIX target of the + * symbolic link, then turn the target into a reparse point data buffer + * that contains a relative or absolute symbolic link. */ deref_name_len = readlink(path, deref_name_buf, sizeof(deref_name_buf) - 1); if (deref_name_len >= 0) { @@ -147,20 +157,23 @@ unix_capture_symlink(struct wim_dentry **root_p, dest[deref_name_len] = '\0'; DEBUG("Read symlink `%s'", dest); - if ((params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_RPFIX) && + if ((params->add_flags & WIMLIB_ADD_FLAG_RPFIX) && dest[0] == '/') { dest = capture_fixup_absolute_symlink(dest, params->capture_root_ino, params->capture_root_dev); - if (!dest) { - WARNING("Ignoring out of tree absolute symlink " - "\"%s\" -> \"%s\"\n" - " (Use --norpfix to capture " - "absolute symlinks as-is)", - path, deref_name_buf); + if (dest == NULL) { + /* RPFIX (reparse point fixup) mode: Ignore + * absolute symbolic link that points out of the + * tree to be captured. */ free_dentry(*root_p); *root_p = NULL; + params->progress.scan.cur_path = path; + params->progress.scan.symlink_target = deref_name_buf; + do_capture_progress(params, + WIMLIB_SCAN_DENTRY_EXCLUDED_SYMLINK, + NULL); return 0; } inode->i_not_rpfixed = 0; @@ -192,52 +205,44 @@ unix_build_dentry_tree_recursive(struct wim_dentry **root_ret, struct add_image_params *params) { struct wim_dentry *root = NULL; - int ret = 0; - struct wim_inode *inode; + int ret; + struct wim_inode *inode = NULL; + struct stat stbuf; if (exclude_path(path, path_len, params->config, true)) { - if ((params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_EXCLUDE_VERBOSE) - && params->progress_func) - { - union wimlib_progress_info info; - info.scan.cur_path = path; - info.scan.excluded = true; - params->progress_func(WIMLIB_PROGRESS_MSG_SCAN_DENTRY, &info); - } - goto out; + ret = 0; + goto out_progress; } - if ((params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_VERBOSE) - && params->progress_func) - { - union wimlib_progress_info info; - info.scan.cur_path = path; - info.scan.excluded = false; - params->progress_func(WIMLIB_PROGRESS_MSG_SCAN_DENTRY, &info); - } - - struct stat stbuf; - int (*stat_fn)(const char *restrict, struct stat *restrict); - if ((params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_DEREFERENCE) || - (params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_ROOT)) - stat_fn = stat; + if ((params->add_flags & WIMLIB_ADD_FLAG_DEREFERENCE) || + (params->add_flags & WIMLIB_ADD_FLAG_ROOT)) + ret = stat(path, &stbuf); else - stat_fn = lstat; + ret = lstat(path, &stbuf); - ret = (*stat_fn)(path, &stbuf); - if (ret != 0) { - ERROR_WITH_ERRNO("Failed to stat `%s'", path); + if (ret) { + ERROR_WITH_ERRNO("Failed to stat \"%s\"", path); + ret = WIMLIB_ERR_STAT; goto out; } - if (!S_ISREG(stbuf.st_mode) && !S_ISDIR(stbuf.st_mode) - && !S_ISLNK(stbuf.st_mode)) { - ERROR("`%s' is not a regular file, directory, or symbolic link.", - path); - ret = WIMLIB_ERR_SPECIAL_FILE; + + if (!S_ISREG(stbuf.st_mode) && + !S_ISDIR(stbuf.st_mode) && + !S_ISLNK(stbuf.st_mode)) + { + if (params->add_flags & WIMLIB_ADD_FLAG_NO_UNSUPPORTED_EXCLUDE) + { + ERROR("Can't archive unsupported file \"%s\"", path); + ret = WIMLIB_ERR_UNSUPPORTED_FILE; + goto out; + } + params->progress.scan.cur_path = path; + do_capture_progress(params, WIMLIB_SCAN_DENTRY_UNSUPPORTED, NULL); + ret = 0; goto out; } - ret = inode_table_new_dentry(params->inode_table, + ret = inode_table_new_dentry(¶ms->inode_table, path_basename_with_len(path, path_len), stbuf.st_ino, stbuf.st_dev, false, &root); if (ret) @@ -245,8 +250,11 @@ unix_build_dentry_tree_recursive(struct wim_dentry **root_ret, inode = root->d_inode; - if (inode->i_nlink > 1) /* Already captured this inode? */ - goto out; + if (inode->i_nlink > 1) { + /* Already captured this inode? */ + ret = 0; + goto out_progress; + } #ifdef HAVE_STAT_NANOSECOND_PRECISION inode->i_creation_time = timespec_to_wim_timestamp(stbuf.st_mtim); @@ -258,7 +266,7 @@ unix_build_dentry_tree_recursive(struct wim_dentry **root_ret, inode->i_last_access_time = unix_timestamp_to_wim(stbuf.st_atime); #endif inode->i_resolved = 1; - if (params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_UNIX_DATA) { + if (params->add_flags & WIMLIB_ADD_FLAG_UNIX_DATA) { ret = inode_set_unix_data(inode, stbuf.st_uid, stbuf.st_gid, stbuf.st_mode, @@ -267,26 +275,39 @@ unix_build_dentry_tree_recursive(struct wim_dentry **root_ret, if (ret) goto out; } - params->add_image_flags &= ~WIMLIB_ADD_IMAGE_FLAG_ROOT; + params->add_flags &= ~WIMLIB_ADD_FLAG_ROOT; if (S_ISREG(stbuf.st_mode)) ret = unix_capture_regular_file(path, stbuf.st_size, - inode, params->lookup_table); + inode, params->unhashed_streams); else if (S_ISDIR(stbuf.st_mode)) ret = unix_capture_directory(root, path, path_len, params); - else + else { ret = unix_capture_symlink(&root, path, inode, params); -out: - if (ret == 0) - *root_ret = root; + if (root == NULL) + goto out; + } + + if (ret) + goto out; + +out_progress: + params->progress.scan.cur_path = path; + if (root == NULL) + do_capture_progress(params, WIMLIB_SCAN_DENTRY_EXCLUDED, NULL); else + do_capture_progress(params, WIMLIB_SCAN_DENTRY_OK, inode); +out: + if (ret) free_dentry_tree(root, params->lookup_table); + else + *root_ret = root; return ret; } /* * unix_build_dentry_tree(): - * Builds a tree of WIM dentries from an on-disk directory tree (UNIX - * version; no NTFS-specific data is captured). + * Builds a tree of WIM dentries from an on-disk directory tree (UNIX + * version; no NTFS-specific data is captured). * * @root_ret: Place to return a pointer to the root of the dentry tree. Only * modified if successful. Set to NULL if the file or directory was @@ -320,7 +341,7 @@ unix_build_dentry_tree(struct wim_dentry **root_ret, return WIMLIB_ERR_STAT; } - if ((params->add_image_flags & WIMLIB_ADD_IMAGE_FLAG_ROOT) && + if ((params->add_flags & WIMLIB_ADD_FLAG_ROOT) && !S_ISDIR(root_stbuf.st_mode)) { ERROR("Root of capture \"%s\" is not a directory", @@ -337,7 +358,7 @@ unix_build_dentry_tree(struct wim_dentry **root_ret, if (path_len >= path_bufsz) return WIMLIB_ERR_INVALID_PARAM; - path_buf = MALLOC(path_bufsz); + path_buf = MALLOC(path_bufsz); if (!path_buf) return WIMLIB_ERR_NOMEM; memcpy(path_buf, root_disk_path, path_len + 1);