]> wimlib.net Git - wimlib/blobdiff - src/unix_apply.c
unix_apply.c: support applying extended attributes
[wimlib] / src / unix_apply.c
index 672f39e7788d3a336a2ec5ab7b8ed216e8b5e3f4..3b31e194287fedf85326c7d42a95c857241a7f33 100644 (file)
@@ -29,6 +29,9 @@
 #include <sys/stat.h>
 #include <sys/time.h>
 #include <sys/types.h>
+#ifdef HAVE_SYS_XATTR_H
+#  include <sys/xattr.h>
+#endif
 #include <unistd.h>
 
 #include "wimlib/apply.h"
@@ -40,6 +43,7 @@
 #include "wimlib/reparse.h"
 #include "wimlib/timestamp.h"
 #include "wimlib/unix_data.h"
+#include "wimlib/xattr.h"
 
 /* We don't require O_NOFOLLOW, but the advantage of having it is that if we
  * need to extract a file to a location at which there exists a symbolic link,
@@ -54,11 +58,15 @@ static int
 unix_get_supported_features(const char *target,
                            struct wim_features *supported_features)
 {
+       supported_features->sparse_files = 1;
        supported_features->hard_links = 1;
        supported_features->symlink_reparse_points = 1;
        supported_features->unix_data = 1;
        supported_features->timestamps = 1;
        supported_features->case_sensitive_filenames = 1;
+#ifdef HAVE_XATTR_SUPPORT
+       supported_features->linux_xattrs = 1;
+#endif
        return 0;
 }
 
@@ -81,6 +89,13 @@ struct unix_apply_ctx {
         * the beginning of the array.  */
        unsigned num_open_fds;
 
+       /* For each currently open file, whether we're writing to it in "sparse"
+        * mode or not.  */
+       bool is_sparse_file[MAX_OPEN_FILES];
+
+       /* Whether is_sparse_file[] is true for any currently open file  */
+       bool any_sparse_files;
+
        /* Buffer for reading reparse point data into memory  */
        u8 reparse_data[REPARSE_DATA_MAX_SIZE];
 
@@ -253,6 +268,109 @@ unix_set_mode(int fd, const char *path, mode_t mode)
        return WIMLIB_ERR_SET_SECURITY;
 }
 
+#ifdef HAVE_XATTR_SUPPORT
+/* Apply extended attributes to a file */
+static int
+apply_linux_xattrs(int fd, const struct wim_inode *inode,
+                  const char *path, struct unix_apply_ctx *ctx,
+                  const void *entries, size_t entries_size)
+{
+       const void * const entries_end = entries + entries_size;
+       char name[XATTR_NAME_MAX + 1];
+
+       for (const struct wimlib_xattr_entry *entry = entries;
+            (void *)entry < entries_end; entry = xattr_entry_next(entry))
+       {
+               u16 name_len;
+               const void *value;
+               u32 value_len;
+               int res;
+
+               if (!valid_xattr_entry(entry, entries_end - (void *)entry)) {
+                       if (!path) {
+                               path = unix_build_inode_extraction_path(inode,
+                                                                       ctx);
+                       }
+                       ERROR("\"%s\": extended attribute is corrupt", path);
+                       return WIMLIB_ERR_INVALID_XATTR;
+               }
+               name_len = le16_to_cpu(entry->name_len);
+               memcpy(name, entry->name, name_len);
+               name[name_len] = '\0';
+
+               value = entry->name + name_len;
+               value_len = le32_to_cpu(entry->value_len);
+
+               if (fd >= 0)
+                       res = fsetxattr(fd, name, value, value_len, 0);
+               else
+                       res = lsetxattr(path, name, value, value_len, 0);
+
+               if (unlikely(res != 0)) {
+                       if (!path) {
+                               path = unix_build_inode_extraction_path(inode,
+                                                                       ctx);
+                       }
+                       if (is_security_xattr(name) &&
+                           (ctx->common.extract_flags &
+                            WIMLIB_EXTRACT_FLAG_STRICT_ACLS))
+                       {
+                               ERROR_WITH_ERRNO("\"%s\": unable to set extended attribute \"%s\"",
+                                                path, name);
+                               return WIMLIB_ERR_SET_XATTR;
+                       }
+                       WARNING_WITH_ERRNO("\"%s\": unable to set extended attribute \"%s\"",
+                                          path, name);
+               }
+       }
+       return 0;
+}
+#endif /* HAVE_XATTR_SUPPORT */
+
+/* Apply standard UNIX permissions (uid, gid, and mode) to a file */
+static int
+apply_unix_permissions(int fd, const struct wim_inode *inode,
+                      const char *path, struct unix_apply_ctx *ctx,
+                      const struct wimlib_unix_data *dat)
+{
+       int ret;
+
+       ret = unix_set_owner_and_group(fd, path, dat->uid, dat->gid);
+       if (ret) {
+               if (!path)
+                       path = unix_build_inode_extraction_path(inode, ctx);
+               if (ctx->common.extract_flags &
+                   WIMLIB_EXTRACT_FLAG_STRICT_ACLS)
+               {
+                       ERROR_WITH_ERRNO("\"%s\": unable to set uid=%"PRIu32" and gid=%"PRIu32,
+                                        path, dat->uid, dat->gid);
+                       return ret;
+               }
+               WARNING_WITH_ERRNO("\"%s\": unable to set uid=%"PRIu32" and gid=%"PRIu32,
+                                  path, dat->uid, dat->gid);
+       }
+
+       if (!inode_is_symlink(inode)) {
+               ret = unix_set_mode(fd, path, dat->mode);
+               if (ret) {
+                       if (!path)
+                               path = unix_build_inode_extraction_path(inode,
+                                                                       ctx);
+                       if (ctx->common.extract_flags &
+                           WIMLIB_EXTRACT_FLAG_STRICT_ACLS)
+                       {
+                               ERROR_WITH_ERRNO("\"%s\": unable to set mode=0%"PRIo32,
+                                                path, dat->mode);
+                               return ret;
+                       }
+                       WARNING_WITH_ERRNO("\"%s\": unable to set mode=0%"PRIo32,
+                                          path, dat->mode);
+               }
+       }
+
+       return 0;
+}
+
 /*
  * Set metadata on an extracted file.
  *
@@ -266,57 +384,33 @@ unix_set_metadata(int fd, const struct wim_inode *inode,
                  const char *path, struct unix_apply_ctx *ctx)
 {
        int ret;
-       struct wimlib_unix_data unix_data;
 
        if (fd < 0 && !path)
                path = unix_build_inode_extraction_path(inode, ctx);
 
-       if ((ctx->common.extract_flags & WIMLIB_EXTRACT_FLAG_UNIX_DATA)
-           && inode_get_unix_data(inode, &unix_data))
-       {
-               u32 uid = unix_data.uid;
-               u32 gid = unix_data.gid;
-               u32 mode = unix_data.mode;
+       if (ctx->common.extract_flags & WIMLIB_EXTRACT_FLAG_UNIX_DATA) {
+               struct wimlib_unix_data dat;
+       #ifdef HAVE_XATTR_SUPPORT
+               const void *entries;
+               u32 entries_size;
 
-               ret = unix_set_owner_and_group(fd, path, uid, gid);
-               if (ret) {
-                       if (!path)
-                               path = unix_build_inode_extraction_path(inode, ctx);
-                       if (ctx->common.extract_flags &
-                           WIMLIB_EXTRACT_FLAG_STRICT_ACLS)
-                       {
-                               ERROR_WITH_ERRNO("Can't set uid=%"PRIu32" and "
-                                                "gid=%"PRIu32" on \"%s\"",
-                                                uid, gid, path);
+               entries = inode_get_linux_xattrs(inode, &entries_size);
+               if (entries) {
+                       ret = apply_linux_xattrs(fd, inode, path, ctx,
+                                                entries, entries_size);
+                       if (ret)
                                return ret;
-                       } else {
-                               WARNING_WITH_ERRNO("Can't set uid=%"PRIu32" and "
-                                                  "gid=%"PRIu32" on \"%s\"",
-                                                  uid, gid, path);
-                       }
                }
-
-               ret = 0;
-               if (!inode_is_symlink(inode))
-                       ret = unix_set_mode(fd, path, mode);
-               if (ret) {
-                       if (!path)
-                               path = unix_build_inode_extraction_path(inode, ctx);
-                       if (ctx->common.extract_flags &
-                           WIMLIB_EXTRACT_FLAG_STRICT_ACLS)
-                       {
-                               ERROR_WITH_ERRNO("Can't set mode=0%"PRIo32" "
-                                                "on \"%s\"", mode, path);
+       #endif
+               if (inode_get_unix_data(inode, &dat)) {
+                       ret = apply_unix_permissions(fd, inode, path, ctx,
+                                                    &dat);
+                       if (ret)
                                return ret;
-                       } else {
-                               WARNING_WITH_ERRNO("Can't set mode=0%"PRIo32" "
-                                                  "on \"%s\"", mode, path);
-                       }
                }
        }
 
-       ret = unix_set_timestamps(fd, path,
-                                 inode->i_last_access_time,
+       ret = unix_set_timestamps(fd, path, inode->i_last_access_time,
                                  inode->i_last_write_time);
        if (ret) {
                if (!path)
@@ -324,12 +418,12 @@ unix_set_metadata(int fd, const struct wim_inode *inode,
                if (ctx->common.extract_flags &
                    WIMLIB_EXTRACT_FLAG_STRICT_TIMESTAMPS)
                {
-                       ERROR_WITH_ERRNO("Can't set timestamps on \"%s\"", path);
+                       ERROR_WITH_ERRNO("\"%s\": unable to set timestamps", path);
                        return ret;
-               } else {
-                       WARNING_WITH_ERRNO("Can't set timestamps on \"%s\"", path);
                }
+               WARNING_WITH_ERRNO("\"%s\": unable to set timestamps", path);
        }
+
        return 0;
 }
 
@@ -541,6 +635,7 @@ unix_cleanup_open_fds(struct unix_apply_ctx *ctx, unsigned offset)
        for (unsigned i = offset; i < ctx->num_open_fds; i++)
                filedes_close(&ctx->open_fds[i]);
        ctx->num_open_fds = 0;
+       ctx->any_sparse_files = false;
 }
 
 static int
@@ -584,6 +679,15 @@ retry_create:
                ERROR_WITH_ERRNO("Can't create regular file \"%s\"", first_path);
                return WIMLIB_ERR_OPEN;
        }
+       if (inode->i_attributes & FILE_ATTRIBUTE_SPARSE_FILE) {
+               ctx->is_sparse_file[ctx->num_open_fds] = true;
+               ctx->any_sparse_files = true;
+       } else {
+               ctx->is_sparse_file[ctx->num_open_fds] = false;
+#ifdef HAVE_POSIX_FALLOCATE
+               posix_fallocate(fd, 0, blob->size);
+#endif
+       }
        filedes_init(&ctx->open_fds[ctx->num_open_fds++], fd);
        return unix_create_hardlinks(inode, first_dentry, first_path, ctx);
 }
@@ -615,18 +719,37 @@ unix_extract_chunk(const struct blob_descriptor *blob, u64 offset,
                   const void *chunk, size_t size, void *_ctx)
 {
        struct unix_apply_ctx *ctx = _ctx;
+       const void * const end = chunk + size;
+       const void *p;
+       bool zeroes;
+       size_t len;
+       unsigned i;
        int ret;
 
-       for (unsigned i = 0; i < ctx->num_open_fds; i++) {
-               ret = full_write(&ctx->open_fds[i], chunk, size);
-               if (ret) {
-                       ERROR_WITH_ERRNO("Error writing data to filesystem");
-                       return ret;
+       /*
+        * For sparse files, only write nonzero regions.  This lets the
+        * filesystem use holes to represent zero regions.
+        */
+       for (p = chunk; p != end; p += len, offset += len) {
+               zeroes = maybe_detect_sparse_region(p, end - p, &len,
+                                                   ctx->any_sparse_files);
+               for (i = 0; i < ctx->num_open_fds; i++) {
+                       if (!zeroes || !ctx->is_sparse_file[i]) {
+                               ret = full_pwrite(&ctx->open_fds[i],
+                                                 p, len, offset);
+                               if (ret)
+                                       goto err;
+                       }
                }
        }
+
        if (ctx->reparse_ptr)
                ctx->reparse_ptr = mempcpy(ctx->reparse_ptr, chunk, size);
        return 0;
+
+err:
+       ERROR_WITH_ERRNO("Error writing data to filesystem");
+       return ret;
 }
 
 /* Called when a blob has been fully read for extraction  */
@@ -666,10 +789,17 @@ unix_end_extract_blob(struct blob_descriptor *blob, int status, void *_ctx)
                        if (ret)
                                break;
                } else {
-                       /* Set metadata on regular file just before closing it.
-                        */
                        struct filedes *fd = &ctx->open_fds[j];
 
+                       /* If the file is sparse, extend it to its final size. */
+                       if (ctx->is_sparse_file[j] && ftruncate(fd->fd, blob->size)) {
+                               ERROR_WITH_ERRNO("Error extending \"%s\" to final size",
+                                                unix_build_inode_extraction_path(inode, ctx));
+                               ret = WIMLIB_ERR_WRITE;
+                               break;
+                       }
+
+                       /* Set metadata on regular file just before closing.  */
                        ret = unix_set_metadata(fd->fd, inode, NULL, ctx);
                        if (ret)
                                break;