]> wimlib.net Git - wimlib/blobdiff - programs/imagex.c
NTFS apply updates
[wimlib] / programs / imagex.c
index a1b9cea29654fce6423005309277ba4ee73f0056..1fce776bcd5b1684cfd1174768a858496ba7d8cd 100644 (file)
 #include <errno.h>
 #include <sys/stat.h>
 
+#ifdef WITH_NTFS_3G
+#include <unistd.h>
+#include <sys/wait.h>
+#endif
+
 #define ARRAY_LEN(array) (sizeof(array) / sizeof(array[0]))
 
 #define swap(a, b) ({ typeof(a) __a = (a); typeof(b) __b = (b); \
@@ -39,6 +44,7 @@
 #define for_opt(c, opts) while ((c = getopt_long_only(argc, (char**)argv, "", \
                                opts, NULL)) != -1)
 
+
 enum imagex_op_type {
        APPEND,
        APPLY,
@@ -75,14 +81,14 @@ static const char *path_basename(const char *path)
 static const char *usage_strings[] = {
 [APPEND] = 
 "    imagex append DIRECTORY WIMFILE [\"IMAGE_NAME\"] [\"DESCRIPTION\"] [--boot]\n"
-"        [--check] [--config CONFIG_FILE] [--flags EDITIONID] [--verify]\n",
+"                  [--check] [--flags EDITIONID] [--dereference]\n",
 [APPLY] = 
 "    imagex apply WIMFILE [IMAGE_NUM | IMAGE_NAME | all] DIRECTORY [--check]\n"
-"        [--verify] [--hardlink] [--symlink] [--verbose]\n",
+"                 [--hardlink] [--symlink] [--verbose]\n",
 [CAPTURE] = 
 "    imagex capture DIRECTORY WIMFILE [\"IMAGE_NAME\"] [\"DESCRIPTION\"]\n"
-"        [--boot] [--check] [--compress[=TYPE]] [--config CONFIG_FILE]\n"
-"        [--flags \"EditionID\"] [--norpfix] [--verify] [--verbose]\n",
+"       l           [--boot] [--check] [--compress[=TYPE]]\n"
+"                   [--flags \"EditionID\"] [--verbose] [--dereference]\n",
 [DELETE] = 
 "    imagex delete WIMFILE (IMAGE_NUM | IMAGE_NAME | all) [--check]\n",
 [DIR] = 
@@ -90,19 +96,19 @@ static const char *usage_strings[] = {
 [EXPORT] = 
 "    imagex export SRC_WIMFILE (SRC_IMAGE_NUM | SRC_IMAGE_NAME | all ) \n"
 "        DEST_WIMFILE [\"DEST_IMAGE_NAME\"] [\"DEST_IMAGE_DESCRIPTION\"]\n"
-"        [--boot] [--check] [--compress[=TYPE]]\n",
+"                  [--boot] [--check] [--compress[=TYPE]]\n",
 [INFO] = 
 "    imagex info WIMFILE [IMAGE_NUM | IMAGE_NAME] [NEW_NAME]\n"
-"        [NEW_DESC] [--boot] [--check] [--header] [--lookup-table]\n"
-"        [--xml] [--extract-xml FILE] [--metadata]\n",
+"                [NEW_DESC] [--boot] [--check] [--header] [--lookup-table]\n"
+"                [--xml] [--extract-xml FILE] [--metadata]\n",
 [JOIN] = 
 "    imagex join [--check] WIMFILE SPLIT_WIM...\n",
 [MOUNT] = 
 "    imagex mount WIMFILE (IMAGE_NUM | IMAGE_NAME) DIRECTORY\n"
-"        [--check] [--debug]\n",
+"                 [--check] [--debug] [--stream-interface=INTERFACE]\n",
 [MOUNTRW] = 
 "    imagex mountrw WIMFILE [IMAGE_NUM | IMAGE_NAME] DIRECTORY\n"
-"        [--check] [--debug]\n",
+"                   [--check] [--debug] [--stream-interface=INTERFACE]\n",
 [SPLIT] = 
 "    imagex split WIMFILE SPLIT_WIMFILE PART_SIZE [--check]\n",
 [UNMOUNT] = 
@@ -118,14 +124,12 @@ static const struct option common_options[] = {
 static const struct option append_options[] = {
        {"boot",   no_argument,       NULL, 'b'},
        {"check",  no_argument,       NULL, 'c'},
-       {"config", required_argument, NULL, 'C'},
        {"flags",    required_argument, NULL, 'f'},
-       {"verify", no_argument,       NULL, 'V'},
+       {"dereference", no_argument, NULL, 'L'},
        {NULL, 0, NULL, 0},
 };
 static const struct option apply_options[] = {
        {"check",    no_argument,       NULL, 'c'},
-       {"verify",   no_argument,       NULL, 'V'},
        {"hardlink", no_argument,       NULL, 'h'},
        {"symlink",  no_argument,       NULL, 's'},
        {"verbose",  no_argument,       NULL, 'v'},
@@ -135,11 +139,9 @@ static const struct option capture_options[] = {
        {"boot",     no_argument,       NULL, 'b'},
        {"check",    no_argument,       NULL, 'c'},
        {"compress", optional_argument, NULL, 'x'},
-       {"config",   required_argument, NULL, 'C'},
        {"flags",    required_argument, NULL, 'f'},
-       {"norpfix",  no_argument,       NULL, 'n'},
-       {"verify",   no_argument,       NULL, 'V'},
-       {"verbose",   no_argument,       NULL, 'v'},
+       {"verbose",  no_argument,       NULL,'v'},
+       {"dereference", no_argument, NULL, 'L'},
        {NULL, 0, NULL, 0},
 };
 static const struct option delete_options[] = {
@@ -173,6 +175,7 @@ static const struct option join_options[] = {
 static const struct option mount_options[] = {
        {"check", no_argument, NULL, 'c'},
        {"debug", no_argument, NULL, 'd'},
+       {"stream-interface", required_argument, NULL, 's'},
        {NULL, 0, NULL, 0},
 };
 
@@ -275,7 +278,7 @@ static int verify_image_exists_and_is_single(int image)
 static int get_compression_type(const char *optarg)
 {
        if (!optarg)
-               return WIM_COMPRESSION_TYPE_LZX;
+               return WIM_COMPRESSION_TYPE_XPRESS;
        if (strcasecmp(optarg, "maximum") == 0 || strcasecmp(optarg, "lzx") == 0)
                return WIM_COMPRESSION_TYPE_LZX;
        else if (strcasecmp(optarg, "fast") == 0 || strcasecmp(optarg, "xpress") == 0)
@@ -312,14 +315,11 @@ static int imagex_append(int argc, const char **argv)
                        open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
                        write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
                        break;
-               case 'C':
-                       /*config = optarg;*/
-                       break;
                case 'f':
                        flags_element = optarg;
                        break;
-               case 'V':
-                       /* verify */
+               case 'L':
+                       add_image_flags |= WIMLIB_ADD_IMAGE_FLAG_DEREFERENCE;
                        break;
                default:
                        usage(APPEND);
@@ -355,9 +355,7 @@ done:
 static int imagex_apply(int argc, const char **argv)
 {
        int c;
-       int link_type = WIM_LINK_TYPE_NONE;
        int open_flags = WIMLIB_OPEN_FLAG_SHOW_PROGRESS;
-       bool verbose = false;
        int image;
        int num_images;
        WIMStruct *w;
@@ -365,23 +363,21 @@ static int imagex_apply(int argc, const char **argv)
        const char *wimfile;
        const char *dir;
        const char *image_num_or_name;
+       int extract_flags = 0;
 
        for_opt(c, apply_options) {
                switch (c) {
                case 'c':
                        open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
                        break;
-               case 'V':
-                       /* verify */
-                       break;
                case 'h':
-                       link_type = WIM_LINK_TYPE_HARD;
+                       extract_flags |= WIMLIB_EXTRACT_FLAG_HARDLINK;
                        break;
                case 's':
-                       link_type = WIM_LINK_TYPE_SYMBOLIC;
+                       extract_flags |= WIMLIB_EXTRACT_FLAG_SYMLINK;
                        break;
                case 'v':
-                       verbose = true;
+                       extract_flags |= WIMLIB_EXTRACT_FLAG_VERBOSE;
                        break;
                default:
                        usage(APPLY);
@@ -406,12 +402,12 @@ static int imagex_apply(int argc, const char **argv)
 
        ret = wimlib_open_wim(wimfile, open_flags, &w);
        if (ret != 0)
-               return ret;
+               goto out;
 
        image = wimlib_resolve_image(w, image_num_or_name);
        ret = verify_image_exists(image);
        if (ret != 0)
-               goto done;
+               goto out;
 
        num_images = wimlib_get_num_images(w);
        if (argc == 2 && num_images != 1) {
@@ -419,15 +415,31 @@ static int imagex_apply(int argc, const char **argv)
                             "(or all)", wimfile, num_images);
                usage(APPLY);
                ret = -1;
-               goto done;
+               goto out;
        }
-       ret = wimlib_set_output_dir(w, dir);
-       if (ret != 0)
-               goto done;
-       wimlib_set_verbose(w, verbose);
-       wimlib_set_link_type(w, link_type);
-       ret = wimlib_extract_image(w, image);
-done:
+
+#ifdef WITH_NTFS_3G
+       struct stat stbuf;
+
+       ret = stat(dir, &stbuf);
+       if (ret == 0) {
+               if (S_ISBLK(stbuf.st_mode) || S_ISREG(stbuf.st_mode)) {
+                       const char *ntfs_device = dir;
+                       printf("Applying image %d of `%s' to NTFS filesystem on `%s'\n",
+                              image, wimfile, ntfs_device);
+                       ret = wimlib_apply_image_to_ntfs_volume(w, image,
+                                                               ntfs_device,
+                                                               extract_flags);
+                       goto out;
+               }
+       } else {
+               if (errno != -ENOENT)
+                       imagex_error_with_errno("Failed to stat `%s'", dir);
+       }
+#endif
+
+       ret = wimlib_extract_image(w, image, dir, extract_flags);
+out:
        wimlib_free(w);
        return ret;
 }
@@ -439,9 +451,8 @@ static int imagex_capture(int argc, const char **argv)
        int c;
        int add_image_flags = 0;
        int write_flags = WIMLIB_WRITE_FLAG_SHOW_PROGRESS;
-       int compression_type = WIM_COMPRESSION_TYPE_NONE;
-       const char *flags_element    = NULL;
-       bool verbose         = false;
+       int compression_type = WIM_COMPRESSION_TYPE_XPRESS;
+       const char *flags_element = NULL;
        const char *dir;
        const char *wimfile;
        const char *name;
@@ -462,20 +473,15 @@ static int imagex_capture(int argc, const char **argv)
                        if (compression_type == WIM_COMPRESSION_TYPE_INVALID)
                                return -1;
                        break;
-               case 'C':
-                       /*config = optarg;*/
-                       break;
                case 'f':
                        flags_element = optarg;
                        break;
-               case 'n':
-                       /*norpfix = true;*/
-                       break;
                case 'v':
-                       verbose = true;
+                       add_image_flags |= WIMLIB_ADD_IMAGE_FLAG_VERBOSE;
+                       write_flags |= WIMLIB_WRITE_FLAG_VERBOSE;
                        break;
-               case 'V':
-                       /*verify = true;*/
+               case 'L':
+                       add_image_flags |= WIMLIB_ADD_IMAGE_FLAG_DEREFERENCE;
                        break;
                default:
                        usage(CAPTURE);
@@ -498,8 +504,6 @@ static int imagex_capture(int argc, const char **argv)
        if (ret != 0)
                return ret;
 
-       wimlib_set_verbose(w, verbose);
-
        ret = wimlib_add_image(w, dir, name, desc, flags_element, 
                               add_image_flags);
        if (ret != 0) {
@@ -646,7 +650,7 @@ static int imagex_export(int argc, const char **argv)
        int open_flags = WIMLIB_OPEN_FLAG_SHOW_PROGRESS;
        int export_flags = 0;
        int write_flags = WIMLIB_WRITE_FLAG_SHOW_PROGRESS;
-       int compression_type = WIM_COMPRESSION_TYPE_NONE;
+       int compression_type = WIM_COMPRESSION_TYPE_XPRESS;
        bool compression_type_specified = false;
        const char *src_wimfile;
        const char *src_image_num_or_name;
@@ -701,7 +705,7 @@ static int imagex_export(int argc, const char **argv)
        if (stat(dest_wimfile, &stbuf) == 0) {
                wim_is_new = false;
                /* Destination file exists. */
-               if (!S_ISREG(stbuf.st_mode)) {
+               if (!S_ISREG(stbuf.st_mode) && !S_ISLNK(stbuf.st_mode)) {
                        imagex_error("`%s' is not a regular file",
                                        dest_wimfile);
                        goto done;
@@ -715,6 +719,7 @@ static int imagex_export(int argc, const char **argv)
                        imagex_error("Cannot specify a compression type that is "
                                     "not the same as that used in the "
                                     "destination WIM");
+                       ret = -1;
                        goto done;
                }
                compression_type = wimlib_get_compression_type(dest_w);
@@ -1072,19 +1077,26 @@ static int imagex_mount_rw_or_ro(int argc, const char **argv)
                case 'd':
                        mount_flags |= WIMLIB_MOUNT_FLAG_DEBUG;
                        break;
+               case 's':
+                       if (strcasecmp(optarg, "none") == 0)
+                               mount_flags |= WIMLIB_MOUNT_FLAG_STREAM_INTERFACE_NONE;
+                       else if (strcasecmp(optarg, "xattr") == 0)
+                               mount_flags |= WIMLIB_MOUNT_FLAG_STREAM_INTERFACE_XATTR;
+                       else if (strcasecmp(optarg, "windows") == 0)
+                               mount_flags |= WIMLIB_MOUNT_FLAG_STREAM_INTERFACE_WINDOWS;
+                       else {
+                               imagex_error("Unknown stream interface \"%s\"", optarg);
+                               goto mount_usage;
+                       }
+                       break;
                default:
-                       usage((mount_flags & WIMLIB_MOUNT_FLAG_READWRITE)  
-                                       ? MOUNTRW : MOUNT);
-                       return -1;
+                       goto mount_usage;
                }
        }
        argc -= optind;
        argv += optind;
-       if (argc != 2 && argc != 3) {
-               usage((mount_flags & WIMLIB_MOUNT_FLAG_READWRITE)  
-                               ? MOUNTRW : MOUNT);
-               return -1;
-       }
+       if (argc != 2 && argc != 3)
+               goto mount_usage;
 
        wimfile = argv[0];
 
@@ -1122,6 +1134,10 @@ static int imagex_mount_rw_or_ro(int argc, const char **argv)
 done:
        wimlib_free(w);
        return ret;
+mount_usage:
+       usage((mount_flags & WIMLIB_MOUNT_FLAG_READWRITE)  
+                       ? MOUNTRW : MOUNT);
+       return -1;
 }
 
 /* Split a WIM into a spanned set */
@@ -1265,6 +1281,8 @@ int main(int argc, const char **argv)
                                imagex_error("Exiting with error code %d:\n"
                                             "       %s.", ret,
                                             wimlib_get_error_string(ret));
+                               if (ret == WIMLIB_ERR_NTFS_3G)
+                                       imagex_error_with_errno("errno");
                        }
                        return ret;
                }
@@ -1274,3 +1292,10 @@ int main(int argc, const char **argv)
        usage_all();
        return 1;
 }
+/*#ifndef WITH_NTFS_3G*/
+               /*ERROR("wimlib was not compiled with support for NTFS-3g, so we cannot extract");*/
+               /*ERROR("a WIM to a NTFS filesystem while preserving NTFS-specific metadata.");*/
+               /*ERROR("Please apply the WIM to a directory rather than a block device, ");*/
+               /*ERROR("and without the NTFS flag; or compile in support for NTFS-3g.");*/
+               /*return WIMLIB_ERR_UNSUPPORTED;*/
+/*#endif*/