wimlib-imagex: Include WIMBoot options in usage printout

[wimlib] / programs / imagex.c

diff --git a/programs/imagex.c b/programs/imagex.c
index 753206d55ecacd917fad6ffa1eae92efc3e364de..2a474d7e172f6594eda8b25063490a0c500e2841 100644 (file)
--- a/programs/imagex.c
+++ b/programs/imagex.c
@@ -173,10 +173,12 @@ enum {
        IMAGEX_NOT_PIPABLE_OPTION,
        IMAGEX_NO_ACLS_OPTION,
        IMAGEX_NO_ATTRIBUTES_OPTION,
+       IMAGEX_NO_REPLACE_OPTION,
        IMAGEX_NO_WILDCARDS_OPTION,
        IMAGEX_NULLGLOB_OPTION,
        IMAGEX_ONE_FILE_ONLY_OPTION,
        IMAGEX_PACK_CHUNK_SIZE_OPTION,
+       IMAGEX_PACK_COMPRESS_OPTION,
        IMAGEX_PACK_STREAMS_OPTION,
        IMAGEX_PATH_OPTION,
        IMAGEX_PIPABLE_OPTION,
@@ -199,6 +201,7 @@ enum {
        IMAGEX_UPDATE_OF_OPTION,
        IMAGEX_VERBOSE_OPTION,
        IMAGEX_WIMBOOT_OPTION,
+       IMAGEX_WIMBOOT_CONFIG_OPTION,
        IMAGEX_XML_OPTION,
 };
 
@@ -233,6 +236,8 @@ static const struct option capture_or_append_options[] = {
        {T("chunk-size"),  required_argument, NULL, IMAGEX_CHUNK_SIZE_OPTION},
        {T("pack-chunk-size"), required_argument, NULL, IMAGEX_PACK_CHUNK_SIZE_OPTION},
        {T("solid-chunk-size"),required_argument, NULL, IMAGEX_PACK_CHUNK_SIZE_OPTION},
+       {T("pack-compress"), required_argument, NULL, IMAGEX_PACK_COMPRESS_OPTION},
+       {T("solid-compress"),required_argument, NULL, IMAGEX_PACK_COMPRESS_OPTION},
        {T("pack-streams"), no_argument,      NULL, IMAGEX_PACK_STREAMS_OPTION},
        {T("solid"),       no_argument,      NULL, IMAGEX_PACK_STREAMS_OPTION},
        {T("config"),      required_argument, NULL, IMAGEX_CONFIG_OPTION},
@@ -281,6 +286,8 @@ static const struct option export_options[] = {
        {T("chunk-size"),  required_argument, NULL, IMAGEX_CHUNK_SIZE_OPTION},
        {T("pack-chunk-size"), required_argument, NULL, IMAGEX_PACK_CHUNK_SIZE_OPTION},
        {T("solid-chunk-size"),required_argument, NULL, IMAGEX_PACK_CHUNK_SIZE_OPTION},
+       {T("pack-compress"), required_argument, NULL, IMAGEX_PACK_COMPRESS_OPTION},
+       {T("solid-compress"),required_argument, NULL, IMAGEX_PACK_COMPRESS_OPTION},
        {T("ref"),         required_argument, NULL, IMAGEX_REF_OPTION},
        {T("threads"),     required_argument, NULL, IMAGEX_THREADS_OPTION},
        {T("rebuild"),     no_argument,       NULL, IMAGEX_REBUILD_OPTION},
@@ -348,6 +355,8 @@ static const struct option optimize_options[] = {
        {T("chunk-size"),  required_argument, NULL, IMAGEX_CHUNK_SIZE_OPTION},
        {T("pack-chunk-size"), required_argument, NULL, IMAGEX_PACK_CHUNK_SIZE_OPTION},
        {T("solid-chunk-size"),required_argument, NULL, IMAGEX_PACK_CHUNK_SIZE_OPTION},
+       {T("pack-compress"), required_argument, NULL, IMAGEX_PACK_COMPRESS_OPTION},
+       {T("solid-compress"),required_argument, NULL, IMAGEX_PACK_COMPRESS_OPTION},
        {T("pack-streams"),no_argument,       NULL, IMAGEX_PACK_STREAMS_OPTION},
        {T("solid"),       no_argument,       NULL, IMAGEX_PACK_STREAMS_OPTION},
        {T("threads"),     required_argument, NULL, IMAGEX_THREADS_OPTION},
@@ -379,6 +388,7 @@ static const struct option update_options[] = {
        {T("check"),       no_argument,       NULL, IMAGEX_CHECK_OPTION},
        {T("rebuild"),     no_argument,       NULL, IMAGEX_REBUILD_OPTION},
        {T("command"),     required_argument, NULL, IMAGEX_COMMAND_OPTION},
+       {T("wimboot-config"), required_argument, NULL, IMAGEX_WIMBOOT_CONFIG_OPTION},
 
        /* Default delete options */
        {T("force"),       no_argument,       NULL, IMAGEX_FORCE_OPTION},
@@ -394,6 +404,7 @@ static const struct option update_options[] = {
        {T("noacls"),      no_argument,       NULL, IMAGEX_NO_ACLS_OPTION},
        {T("no-acls"),     no_argument,       NULL, IMAGEX_NO_ACLS_OPTION},
        {T("strict-acls"), no_argument,       NULL, IMAGEX_STRICT_ACLS_OPTION},
+       {T("no-replace"),  no_argument,       NULL, IMAGEX_NO_REPLACE_OPTION},
 
        {NULL, 0, NULL, 0},
 };
@@ -1228,6 +1239,10 @@ imagex_progress_func(enum wimlib_progress_msg msg,
                        break;
                }
                break;
+       case WIMLIB_PROGRESS_MSG_REPLACE_FILE_IN_WIM:
+               imagex_printf(T("Updating \"%"TS"\" in WIM image\n"),
+                             info->replace.path_in_wim);
+               break;
        default:
                break;
        }
@@ -1291,6 +1306,8 @@ update_command_add_option(int op, const tchar *option,
                        cmd->add.add_flags |= WIMLIB_ADD_FLAG_STRICT_ACLS;
                else if (!tstrcmp(option, T("--dereference")))
                        cmd->add.add_flags |= WIMLIB_ADD_FLAG_DEREFERENCE;
+               else if (!tstrcmp(option, T("--no-replace")))
+                       cmd->add.add_flags |= WIMLIB_ADD_FLAG_NO_REPLACE;
                else
                        recognized = false;
                break;
@@ -1469,7 +1486,7 @@ parse_update_command_file(tchar **cmd_file_contents_p, size_t cmd_file_nchars,
        return cmds;
 }
 
-/* Apply one image, or all images, from a WIM file into a directory, OR apply
+/* Apply one image, or all images, from a WIM file to a directory, OR apply
  * one image from a WIM file to a NTFS volume.  */
 static int
 imagex_apply(int argc, tchar **argv, int cmd)
@@ -1668,6 +1685,7 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
        int compression_type = WIMLIB_COMPRESSION_TYPE_INVALID;
        uint32_t chunk_size = UINT32_MAX;
        uint32_t pack_chunk_size = UINT32_MAX;
+       int pack_ctype = WIMLIB_COMPRESSION_TYPE_INVALID;
        const tchar *wimfile;
        int wim_fd;
        const tchar *name;
@@ -1734,6 +1752,11 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
                        if (pack_chunk_size == UINT32_MAX)
                                goto out_err;
                        break;
+               case IMAGEX_PACK_COMPRESS_OPTION:
+                       pack_ctype = get_compression_type(optarg);
+                       if (pack_ctype == WIMLIB_COMPRESSION_TYPE_INVALID)
+                               goto out_err;
+                       break;
                case IMAGEX_PACK_STREAMS_OPTION:
                        write_flags |= WIMLIB_WRITE_FLAG_PACK_STREAMS;
                        break;
@@ -1832,7 +1855,7 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
                        compression_type = WIMLIB_COMPRESSION_TYPE_XPRESS;
                } else {
                        compression_type = WIMLIB_COMPRESSION_TYPE_LZX;
-                       if (!compress_slow) {
+                       if (!compress_slow && pack_ctype != WIMLIB_COMPRESSION_TYPE_LZX) {
                                struct wimlib_lzx_compressor_params params = {
                                        .hdr.size = sizeof(params),
                                        .algorithm = WIMLIB_LZX_ALGORITHM_FAST,
@@ -1842,7 +1865,6 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
                                                                     &params.hdr);
                        }
                }
-
        }
 
        if (compress_slow)
@@ -1972,6 +1994,11 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
                if (ret)
                        goto out_free_wim;
        }
+       if (pack_ctype != WIMLIB_COMPRESSION_TYPE_INVALID) {
+               ret = wimlib_set_output_pack_compression_type(wim, pack_ctype);
+               if (ret)
+                       goto out_free_wim;
+       }
        if (pack_chunk_size != UINT32_MAX) {
                ret = wimlib_set_output_pack_chunk_size(wim, pack_chunk_size);
                if (ret)
@@ -2568,6 +2595,7 @@ imagex_export(int argc, tchar **argv, int cmd)
        unsigned num_threads = 0;
        uint32_t chunk_size = UINT32_MAX;
        uint32_t pack_chunk_size = UINT32_MAX;
+       int pack_ctype = WIMLIB_COMPRESSION_TYPE_INVALID;
 
        for_opt(c, export_options) {
                switch (c) {
@@ -2603,6 +2631,11 @@ imagex_export(int argc, tchar **argv, int cmd)
                        if (pack_chunk_size == UINT32_MAX)
                                goto out_err;
                        break;
+               case IMAGEX_PACK_COMPRESS_OPTION:
+                       pack_ctype = get_compression_type(optarg);
+                       if (pack_ctype == WIMLIB_COMPRESSION_TYPE_INVALID)
+                               goto out_err;
+                       break;
                case IMAGEX_REF_OPTION:
                        ret = string_set_append(&refglobs, optarg);
                        if (ret)
@@ -2732,6 +2765,11 @@ imagex_export(int argc, tchar **argv, int cmd)
                if (ret)
                        goto out_free_dest_wim;
        }
+       if (pack_ctype != WIMLIB_COMPRESSION_TYPE_INVALID) {
+               ret = wimlib_set_output_pack_compression_type(dest_wim, pack_ctype);
+               if (ret)
+                       goto out_free_dest_wim;
+       }
        if (pack_chunk_size != UINT32_MAX) {
                ret = wimlib_set_output_pack_chunk_size(dest_wim, pack_chunk_size);
                if (ret)
@@ -3408,6 +3446,7 @@ imagex_optimize(int argc, tchar **argv, int cmd)
        int compression_type = WIMLIB_COMPRESSION_TYPE_INVALID;
        uint32_t chunk_size = UINT32_MAX;
        uint32_t pack_chunk_size = UINT32_MAX;
+       int pack_ctype = WIMLIB_COMPRESSION_TYPE_INVALID;
        int ret;
        WIMStruct *wim;
        const tchar *wimfile;
@@ -3447,6 +3486,11 @@ imagex_optimize(int argc, tchar **argv, int cmd)
                        if (pack_chunk_size == UINT32_MAX)
                                goto out_err;
                        break;
+               case IMAGEX_PACK_COMPRESS_OPTION:
+                       pack_ctype = get_compression_type(optarg);
+                       if (pack_ctype == WIMLIB_COMPRESSION_TYPE_INVALID)
+                               goto out_err;
+                       break;
                case IMAGEX_PACK_STREAMS_OPTION:
                        write_flags |= WIMLIB_WRITE_FLAG_PACK_STREAMS;
                        write_flags |= WIMLIB_WRITE_FLAG_RECOMPRESS;
@@ -3491,6 +3535,11 @@ imagex_optimize(int argc, tchar **argv, int cmd)
                if (ret)
                        goto out_wimlib_free;
        }
+       if (pack_ctype != WIMLIB_COMPRESSION_TYPE_INVALID) {
+               ret = wimlib_set_output_pack_compression_type(wim, pack_ctype);
+               if (ret)
+                       goto out_wimlib_free;
+       }
        if (pack_chunk_size != UINT32_MAX) {
                ret = wimlib_set_output_pack_chunk_size(wim, pack_chunk_size);
                if (ret)
@@ -3673,6 +3722,7 @@ imagex_update(int argc, tchar **argv, int cmd)
        size_t num_cmds;
        tchar *command_str = NULL;
        tchar *config_file = NULL;
+       tchar *wimboot_config = NULL;
 
        for_opt(c, update_options) {
                switch (c) {
@@ -3703,6 +3753,9 @@ imagex_update(int argc, tchar **argv, int cmd)
                                goto out_err;
                        }
                        break;
+               case IMAGEX_WIMBOOT_CONFIG_OPTION:
+                       wimboot_config = optarg;
+                       break;
                /* Default delete options */
                case IMAGEX_FORCE_OPTION:
                        default_delete_flags |= WIMLIB_DELETE_FLAG_FORCE;
@@ -3733,6 +3786,9 @@ imagex_update(int argc, tchar **argv, int cmd)
                case IMAGEX_STRICT_ACLS_OPTION:
                        default_add_flags |= WIMLIB_ADD_FLAG_STRICT_ACLS;
                        break;
+               case IMAGEX_NO_REPLACE_OPTION:
+                       default_add_flags |= WIMLIB_ADD_FLAG_NO_REPLACE;
+                       break;
                default:
                        goto out_usage;
                }
@@ -3776,7 +3832,11 @@ imagex_update(int argc, tchar **argv, int cmd)
                cmd_file_contents = NULL;
                cmds = parse_update_command_file(&command_str, tstrlen(command_str),
                                                 &num_cmds);
-       } else {
+               if (!cmds) {
+                       ret = -1;
+                       goto out_free_cmd_file_contents;
+               }
+       } else if (!wimboot_config) {
                if (isatty(STDIN_FILENO)) {
                        tputs(T("Reading update commands from standard input..."));
                        recommend_man_page(CMD_UPDATE, stdout);
@@ -3790,10 +3850,14 @@ imagex_update(int argc, tchar **argv, int cmd)
                /* Parse the update commands */
                cmds = parse_update_command_file(&cmd_file_contents, cmd_file_nchars,
                                                 &num_cmds);
-       }
-       if (!cmds) {
-               ret = -1;
-               goto out_free_cmd_file_contents;
+               if (!cmds) {
+                       ret = -1;
+                       goto out_free_cmd_file_contents;
+               }
+       } else {
+               cmd_file_contents = NULL;
+               cmds = NULL;
+               num_cmds = 0;
        }
 
        /* Set default flags and capture config on the update commands */
@@ -3817,6 +3881,26 @@ imagex_update(int argc, tchar **argv, int cmd)
        if (ret)
                goto out_free_cmds;
 
+       if (wimboot_config) {
+               /* --wimboot-config=FILE is short for an
+                * "add FILE /Windows/System32/WimBootCompress.ini" command.
+                */
+               struct wimlib_update_command cmd = {
+                       .op = WIMLIB_UPDATE_OP_ADD,
+                       .add = {
+                               .fs_source_path = wimboot_config,
+                               .wim_target_path =
+                                       T("/Windows/System32/WimBootCompress.ini"),
+                               .config_file = NULL,
+                               .add_flags = 0,
+                       },
+               };
+               ret = wimlib_update_image(wim, image, &cmd, 1,
+                                         update_flags, imagex_progress_func);
+               if (ret)
+                       goto out_free_cmds;
+       }
+
        /* Overwrite the updated WIM */
        ret = wimlib_overwrite(wim, write_flags, num_threads,
                               imagex_progress_func);
@@ -3874,7 +3958,7 @@ T(
 "                    [--nocheck] [--flags EDITION_ID] [--dereference]\n"
 "                    [--config=FILE] [--threads=NUM_THREADS] [--source-list]\n"
 "                    [--no-acls] [--strict-acls] [--rpfix] [--norpfix]\n"
-"                    [--update-of=[WIMFILE:]IMAGE]\n"
+"                    [--update-of=[WIMFILE:]IMAGE] [--wimboot]\n"
 ),
 [CMD_APPLY] =
 T(
@@ -3882,7 +3966,7 @@ T(
 "                    (DIRECTORY | NTFS_VOLUME) [--check] [--ref=\"GLOB\"]\n"
 "                    [--no-acls] [--strict-acls] [--no-attributes]\n"
 "                    [--rpfix] [--norpfix] [--hardlink] [--symlink]\n"
-"                    [--include-invalid-names]\n"
+"                    [--include-invalid-names] [--wimboot]\n"
 ),
 [CMD_CAPTURE] =
 T(
@@ -3892,7 +3976,7 @@ T(
 "                    [--dereference] [--config=FILE] [--threads=NUM_THREADS]\n"
 "                    [--source-list] [--no-acls] [--strict-acls] [--rpfix]\n"
 "                    [--norpfix] [--update-of=[WIMFILE:]IMAGE]\n"
-"                    [--delta-from=WIMFILE]\n"
+"                    [--delta-from=WIMFILE] [--wimboot]\n"
 ),
 [CMD_DELETE] =
 T(
@@ -3963,7 +4047,8 @@ T(
 T(
 "    %"TS" WIMFILE [IMAGE_NUM | IMAGE_NAME] [--check] [--rebuild]\n"
 "                    [--threads=NUM_THREADS] [DEFAULT_ADD_OPTIONS]\n"
-"                    [DEFAULT_DELETE_OPTIONS] [--command=STRING] [< CMDFILE]\n"
+"                    [DEFAULT_DELETE_OPTIONS] [--command=STRING]\n"
+"                    [--wimboot-config=FILE| [< CMDFILE]\n"
 ),
 };
 
@@ -4187,7 +4272,7 @@ main(int argc, char **argv)
                exit(2);
        }
 
-       /* Enable warning and error messages in wimlib be more user-friendly.
+       /* Enable warning and error messages in wimlib to be more user-friendly.
         * */
        wimlib_set_print_errors(true);