]> wimlib.net Git - wimlib/blobdiff - programs/imagex.c
wimlib-imagex: Include WIMBoot options in usage printout
[wimlib] / programs / imagex.c
index 0e3c6e8b0016a8d70dee0e0c012fc9080999701d..2a474d7e172f6594eda8b25063490a0c500e2841 100644 (file)
@@ -6,7 +6,7 @@
  */
 
 /*
- * Copyright (C) 2012, 2013 Eric Biggers
+ * Copyright (C) 2012, 2013, 2014 Eric Biggers
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -167,15 +167,18 @@ enum {
        IMAGEX_LAZY_OPTION,
        IMAGEX_LOOKUP_TABLE_OPTION,
        IMAGEX_METADATA_OPTION,
+       IMAGEX_NEW_IMAGE_OPTION,
        IMAGEX_NOCHECK_OPTION,
        IMAGEX_NORPFIX_OPTION,
        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,
@@ -197,6 +200,8 @@ enum {
        IMAGEX_UNIX_DATA_OPTION,
        IMAGEX_UPDATE_OF_OPTION,
        IMAGEX_VERBOSE_OPTION,
+       IMAGEX_WIMBOOT_OPTION,
+       IMAGEX_WIMBOOT_CONFIG_OPTION,
        IMAGEX_XML_OPTION,
 };
 
@@ -217,6 +222,7 @@ static const struct option apply_options[] = {
 
        /* --resume is undocumented for now as it needs improvement.  */
        {T("resume"),      no_argument,       NULL, IMAGEX_RESUME_OPTION},
+       {T("wimboot"),     no_argument,       NULL, IMAGEX_WIMBOOT_OPTION},
        {NULL, 0, NULL, 0},
 };
 
@@ -230,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},
@@ -249,6 +257,7 @@ static const struct option capture_or_append_options[] = {
        {T("not-pipable"), no_argument,       NULL, IMAGEX_NOT_PIPABLE_OPTION},
        {T("update-of"),   required_argument, NULL, IMAGEX_UPDATE_OF_OPTION},
        {T("delta-from"),  required_argument, NULL, IMAGEX_DELTA_FROM_OPTION},
+       {T("wimboot"),     no_argument,       NULL, IMAGEX_WIMBOOT_OPTION},
        {NULL, 0, NULL, 0},
 };
 
@@ -271,11 +280,14 @@ static const struct option export_options[] = {
        {T("nocheck"),     no_argument,       NULL, IMAGEX_NOCHECK_OPTION},
        {T("no-check"),    no_argument,       NULL, IMAGEX_NOCHECK_OPTION},
        {T("compress"),    required_argument, NULL, IMAGEX_COMPRESS_OPTION},
+       {T("compress-slow"), no_argument,     NULL, IMAGEX_COMPRESS_SLOW_OPTION},
        {T("pack-streams"),no_argument,       NULL, IMAGEX_PACK_STREAMS_OPTION},
        {T("solid"),       no_argument,       NULL, IMAGEX_PACK_STREAMS_OPTION},
        {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},
@@ -299,6 +311,7 @@ static const struct option extract_options[] = {
        {T("no-wildcards"), no_argument,      NULL, IMAGEX_NO_WILDCARDS_OPTION},
        {T("nullglob"),     no_argument,      NULL, IMAGEX_NULLGLOB_OPTION},
        {T("preserve-dir-structure"), no_argument, NULL, IMAGEX_PRESERVE_DIR_STRUCTURE_OPTION},
+       {T("wimboot"),     no_argument,       NULL, IMAGEX_WIMBOOT_OPTION},
        {NULL, 0, NULL, 0},
 };
 
@@ -342,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},
@@ -360,6 +375,7 @@ static const struct option unmount_options[] = {
        {T("check"),   no_argument, NULL, IMAGEX_CHECK_OPTION},
        {T("rebuild"), no_argument, NULL, IMAGEX_REBUILD_OPTION},
        {T("lazy"),    no_argument, NULL, IMAGEX_LAZY_OPTION},
+       {T("new-image"), no_argument, NULL, IMAGEX_NEW_IMAGE_OPTION},
        {NULL, 0, NULL, 0},
 };
 
@@ -372,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},
@@ -387,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},
 };
@@ -711,7 +729,7 @@ static bool
 is_comment_line(const tchar *line, size_t len)
 {
        for (;;) {
-               if (*line == T('#'))
+               if (*line == T('#') || *line == T(';'))
                        return true;
                if (!istspace(*line) && *line != T('\0'))
                        return false;
@@ -809,144 +827,6 @@ parse_source_list(tchar **source_list_contents_p, size_t source_list_nchars,
        return sources;
 }
 
-
-enum capture_config_section {
-       CAPTURE_CONFIG_NO_SECTION,
-       CAPTURE_CONFIG_EXCLUSION_SECTION,
-       CAPTURE_CONFIG_EXCLUSION_EXCEPTION_SECTION,
-       CAPTURE_CONFIG_IGNORE_SECTION,
-};
-
-enum {
-       CAPTURE_CONFIG_INVALID_SECTION,
-       CAPTURE_CONFIG_CHANGED_SECTION,
-       CAPTURE_CONFIG_SAME_SECTION,
-};
-
-static int
-check_config_section(tchar *line, size_t len,
-                    enum capture_config_section *cur_section)
-{
-       while (istspace(*line))
-               line++;
-
-       if (*line != T('['))
-               return CAPTURE_CONFIG_SAME_SECTION;
-
-       line++;
-       tchar *endbrace = tstrrchr(line, T(']'));
-       if (!endbrace)
-               return CAPTURE_CONFIG_SAME_SECTION;
-
-       if (!tmemcmp(line, T("ExclusionList"), endbrace - line)) {
-               *cur_section = CAPTURE_CONFIG_EXCLUSION_SECTION;
-       } else if (!tmemcmp(line, T("ExclusionException"), endbrace - line)) {
-               *cur_section = CAPTURE_CONFIG_EXCLUSION_EXCEPTION_SECTION;
-       } else if (!tmemcmp(line, T("CompressionExclusionList"), endbrace - line)) {
-               *cur_section = CAPTURE_CONFIG_IGNORE_SECTION;
-               tfputs(T("WARNING: Ignoring [CompressionExclusionList] section "
-                        "of capture config file\n"),
-                      stderr);
-       } else if (!tmemcmp(line, T("AlignmentList"), endbrace - line)) {
-               *cur_section = CAPTURE_CONFIG_IGNORE_SECTION;
-               tfputs(T("WARNING: Ignoring [AlignmentList] section "
-                        "of capture config file\n"),
-                      stderr);
-       } else {
-               imagex_error(T("Invalid capture config file section \"%"TS"\""),
-                            line - 1);
-               return CAPTURE_CONFIG_INVALID_SECTION;
-       }
-       return CAPTURE_CONFIG_CHANGED_SECTION;
-}
-
-
-static bool
-pattern_list_add_pattern(struct wimlib_pattern_list *pat_list,
-                        tchar *pat)
-{
-       if (pat_list->num_pats == pat_list->num_allocated_pats) {
-               tchar **pats;
-               size_t num_allocated_pats = pat_list->num_pats + 8;
-
-               pats = realloc(pat_list->pats,
-                              num_allocated_pats * sizeof(pat_list->pats[0]));
-               if (!pats) {
-                       imagex_error(T("Out of memory!"));
-                       return false;
-               }
-               pat_list->pats = pats;
-               pat_list->num_allocated_pats = num_allocated_pats;
-       }
-       pat_list->pats[pat_list->num_pats++] = pat;
-       return true;
-}
-
-static bool
-parse_capture_config_line(tchar *line, size_t len,
-                         enum capture_config_section *cur_section,
-                         struct wimlib_capture_config *config)
-{
-       tchar *filename;
-       int ret;
-
-       ret = check_config_section(line, len, cur_section);
-       if (ret == CAPTURE_CONFIG_INVALID_SECTION)
-               return false;
-       if (ret == CAPTURE_CONFIG_CHANGED_SECTION)
-               return true;
-
-       switch (*cur_section) {
-       case CAPTURE_CONFIG_NO_SECTION:
-               imagex_error(T("Line \"%"TS"\" is not in a section "
-                              "(such as [ExclusionList]"), line);
-               return false;
-       case CAPTURE_CONFIG_EXCLUSION_SECTION:
-               if (parse_string(&line, &len, &filename) != PARSE_STRING_SUCCESS)
-                       return false;
-               return pattern_list_add_pattern(&config->exclusion_pats,
-                                               filename);
-       case CAPTURE_CONFIG_EXCLUSION_EXCEPTION_SECTION:
-               if (parse_string(&line, &len, &filename) != PARSE_STRING_SUCCESS)
-                       return false;
-               return pattern_list_add_pattern(&config->exclusion_exception_pats,
-                                               filename);
-       case CAPTURE_CONFIG_IGNORE_SECTION:
-               return true;
-       }
-       return false;
-}
-
-static int
-parse_capture_config(tchar **contents_p, size_t nchars,
-                    struct wimlib_capture_config *config)
-{
-       ssize_t nlines;
-       tchar *p;
-       size_t i;
-       enum capture_config_section cur_section;
-
-       memset(config, 0, sizeof(*config));
-
-       nlines = text_file_count_lines(contents_p, &nchars);
-       if (nlines < 0)
-               return -1;
-
-       cur_section = CAPTURE_CONFIG_NO_SECTION;
-       p = *contents_p;
-       for (i = 0; i < nlines; i++) {
-               tchar *endp = tmemchr(p, T('\n'), nchars);
-               size_t len = endp - p + 1;
-               *endp = T('\0');
-               if (!is_comment_line(p, len))
-                       if (!parse_capture_config_line(p, len, &cur_section, config))
-                               return -1;
-               p = endp + 1;
-
-       }
-       return 0;
-}
-
 /* Reads the contents of a file into memory. */
 static char *
 file_get_contents(const tchar *filename, size_t *len_ret)
@@ -1359,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;
        }
@@ -1422,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;
@@ -1600,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)
@@ -1662,6 +1548,9 @@ imagex_apply(int argc, tchar **argv, int cmd)
                case IMAGEX_RESUME_OPTION:
                        extract_flags |= WIMLIB_EXTRACT_FLAG_RESUME;
                        break;
+               case IMAGEX_WIMBOOT_OPTION:
+                       extract_flags |= WIMLIB_EXTRACT_FLAG_WIMBOOT;
+                       break;
                default:
                        goto out_usage;
                }
@@ -1796,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;
@@ -1817,9 +1707,7 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
        tchar *source;
        tchar *source_copy;
 
-       const tchar *config_file = NULL;
-       tchar *config_str;
-       struct wimlib_capture_config *config;
+       tchar *config_file = NULL;
 
        bool source_list = false;
        size_t source_list_nchars = 0;
@@ -1864,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;
@@ -1938,6 +1831,9 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
                                goto out_free_base_wimfiles;
                        write_flags |= WIMLIB_WRITE_FLAG_SKIP_EXTERNAL_WIMS;
                        break;
+               case IMAGEX_WIMBOOT_OPTION:
+                       add_image_flags |= WIMLIB_ADD_IMAGE_FLAG_WIMBOOT;
+                       break;
                default:
                        goto out_usage;
                }
@@ -1955,16 +1851,19 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
 
 
        if (compression_type == WIMLIB_COMPRESSION_TYPE_INVALID) {
-               compression_type = WIMLIB_COMPRESSION_TYPE_LZX;
-
-               if (!compress_slow) {
-                       struct wimlib_lzx_compressor_params params = {
-                               .hdr.size = sizeof(params),
-                               .algorithm = WIMLIB_LZX_ALGORITHM_FAST,
-                               .use_defaults = 1,
-                       };
-                       wimlib_set_default_compressor_params(WIMLIB_COMPRESSION_TYPE_LZX,
-                                                            &params.hdr);
+               if (add_image_flags & WIMLIB_ADD_IMAGE_FLAG_WIMBOOT) {
+                       compression_type = WIMLIB_COMPRESSION_TYPE_XPRESS;
+               } else {
+                       compression_type = WIMLIB_COMPRESSION_TYPE_LZX;
+                       if (!compress_slow && pack_ctype != WIMLIB_COMPRESSION_TYPE_LZX) {
+                               struct wimlib_lzx_compressor_params params = {
+                                       .hdr.size = sizeof(params),
+                                       .algorithm = WIMLIB_LZX_ALGORITHM_FAST,
+                                       .use_defaults = 1,
+                               };
+                               wimlib_set_default_compressor_params(WIMLIB_COMPRESSION_TYPE_LZX,
+                                                                    &params.hdr);
+                       }
                }
        }
 
@@ -2075,27 +1974,6 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
                source_list_contents = NULL;
        }
 
-       if (config_file) {
-               /* Read and parse capture configuration file.  */
-               size_t config_len;
-
-               config_str = file_get_text_contents(config_file, &config_len);
-               if (!config_str) {
-                       ret = -1;
-                       goto out_free_capture_sources;
-               }
-
-               config = alloca(sizeof(*config));
-               ret = parse_capture_config(&config_str, config_len, config);
-               if (ret)
-                       goto out_free_config;
-       } else {
-               /* No capture configuration file specified; use default
-                * configuration for capturing Windows operating systems.  */
-               config = NULL;
-               add_image_flags |= WIMLIB_ADD_FLAG_WINCONFIG;
-       }
-
        /* Open the existing WIM, or create a new one.  */
        if (cmd == CMD_APPEND)
                ret = wimlib_open_wim(wimfile, open_flags, &wim,
@@ -2103,13 +1981,23 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
        else
                ret = wimlib_create_new_wim(compression_type, &wim);
        if (ret)
-               goto out_free_config;
+               goto out_free_capture_sources;
 
        /* Set chunk size if non-default.  */
        if (chunk_size != UINT32_MAX) {
                ret = wimlib_set_output_chunk_size(wim, chunk_size);
                if (ret)
                        goto out_free_wim;
+       } else if ((add_image_flags & WIMLIB_ADD_IMAGE_FLAG_WIMBOOT) &&
+                  compression_type == WIMLIB_COMPRESSION_TYPE_XPRESS) {
+               ret = wimlib_set_output_chunk_size(wim, 4096);
+               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);
@@ -2238,7 +2126,7 @@ imagex_capture_or_append(int argc, tchar **argv, int cmd)
                                           capture_sources,
                                           num_sources,
                                           name,
-                                          config,
+                                          config_file,
                                           add_image_flags,
                                           imagex_progress_func);
        if (ret)
@@ -2308,12 +2196,6 @@ out_free_base_wims:
        free(base_wims);
 out_free_wim:
        wimlib_free(wim);
-out_free_config:
-       if (config) {
-               free(config->exclusion_pats.pats);
-               free(config->exclusion_exception_pats.pats);
-               free(config_str);
-       }
 out_free_capture_sources:
        if (capture_sources_malloced)
                free(capture_sources);
@@ -2713,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) {
@@ -2731,6 +2614,10 @@ imagex_export(int argc, tchar **argv, int cmd)
                        if (compression_type == WIMLIB_COMPRESSION_TYPE_INVALID)
                                goto out_err;
                        break;
+               case IMAGEX_COMPRESS_SLOW_OPTION:
+                       write_flags |= WIMLIB_WRITE_FLAG_RECOMPRESS;
+                       set_compress_slow();
+                       break;
                case IMAGEX_PACK_STREAMS_OPTION:
                        write_flags |= WIMLIB_WRITE_FLAG_PACK_STREAMS;
                        break;
@@ -2744,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)
@@ -2873,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)
@@ -2946,7 +2843,6 @@ imagex_extract(int argc, tchar **argv, int cmd)
        int ret;
        const tchar *wimfile;
        const tchar *image_num_or_name;
-       const tchar *pathlist;
        tchar *dest_dir = T(".");
        int extract_flags = WIMLIB_EXTRACT_FLAG_NORPFIX |
                            WIMLIB_EXTRACT_FLAG_GLOB_PATHS |
@@ -2955,8 +2851,6 @@ imagex_extract(int argc, tchar **argv, int cmd)
 
        STRING_SET(refglobs);
 
-       struct wimlib_extract_command *cmds;
-       size_t num_cmds;
        tchar *root_path = T("");
 
        for_opt(c, extract_options) {
@@ -3005,6 +2899,9 @@ imagex_extract(int argc, tchar **argv, int cmd)
                case IMAGEX_PRESERVE_DIR_STRUCTURE_OPTION:
                        notlist_extract_flags &= ~WIMLIB_EXTRACT_FLAG_NO_PRESERVE_DIR_STRUCTURE;
                        break;
+               case IMAGEX_WIMBOOT_OPTION:
+                       extract_flags |= WIMLIB_EXTRACT_FLAG_WIMBOOT;
+                       break;
                default:
                        goto out_usage;
                }
@@ -3549,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;
@@ -3588,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;
@@ -3632,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)
@@ -3753,6 +3661,9 @@ imagex_unmount(int argc, tchar **argv, int cmd)
                case IMAGEX_LAZY_OPTION:
                        unmount_flags |= WIMLIB_UNMOUNT_FLAG_LAZY;
                        break;
+               case IMAGEX_NEW_IMAGE_OPTION:
+                       unmount_flags |= WIMLIB_UNMOUNT_FLAG_NEW_IMAGE;
+                       break;
                default:
                        goto out_usage;
                }
@@ -3762,6 +3673,15 @@ imagex_unmount(int argc, tchar **argv, int cmd)
        if (argc != 1)
                goto out_usage;
 
+       if (unmount_flags & WIMLIB_UNMOUNT_FLAG_NEW_IMAGE) {
+               if (!(unmount_flags & WIMLIB_UNMOUNT_FLAG_COMMIT)) {
+                       imagex_error(T("--new-image is meaningless "
+                                      "without --commit also specified!"));
+                       goto out_err;
+               }
+               imagex_printf(T("Committing changes as new image...\n"));
+       }
+
        ret = wimlib_unmount_image(argv[0], unmount_flags,
                                   imagex_progress_func);
        if (ret)
@@ -3771,6 +3691,7 @@ out:
 
 out_usage:
        usage(CMD_UNMOUNT, stderr);
+out_err:
        ret = -1;
        goto out;
 }
@@ -3790,7 +3711,8 @@ imagex_update(int argc, tchar **argv, int cmd)
        int write_flags = 0;
        int update_flags = WIMLIB_UPDATE_FLAG_SEND_PROGRESS;
        int default_add_flags = WIMLIB_ADD_FLAG_EXCLUDE_VERBOSE |
-                               WIMLIB_ADD_FLAG_VERBOSE;
+                               WIMLIB_ADD_FLAG_VERBOSE |
+                               WIMLIB_ADD_FLAG_WINCONFIG;
        int default_delete_flags = 0;
        unsigned num_threads = 0;
        int c;
@@ -3799,10 +3721,8 @@ imagex_update(int argc, tchar **argv, int cmd)
        struct wimlib_update_command *cmds;
        size_t num_cmds;
        tchar *command_str = NULL;
-
-       const tchar *config_file = NULL;
-       tchar *config_str;
-       struct wimlib_capture_config *config;
+       tchar *config_file = NULL;
+       tchar *wimboot_config = NULL;
 
        for_opt(c, update_options) {
                switch (c) {
@@ -3833,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;
@@ -3863,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;
                }
@@ -3900,32 +3826,17 @@ imagex_update(int argc, tchar **argv, int cmd)
                image = 1;
        }
 
-       /* Parse capture configuration file if specified */
-       if (config_file) {
-               size_t config_len;
-
-               config_str = file_get_text_contents(config_file, &config_len);
-               if (!config_str) {
-                       ret = -1;
-                       goto out_wimlib_free;
-               }
-
-               config = alloca(sizeof(*config));
-               ret = parse_capture_config(&config_str, config_len, config);
-               if (ret)
-                       goto out_free_config;
-       } else {
-               config = NULL;
-               default_add_flags |= WIMLIB_ADD_FLAG_WINCONFIG;
-       }
-
        /* Read update commands from standard input, or the command string if
         * specified.  */
        if (command_str) {
                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);
@@ -3933,16 +3844,20 @@ imagex_update(int argc, tchar **argv, int cmd)
                cmd_file_contents = stdin_get_text_contents(&cmd_file_nchars);
                if (!cmd_file_contents) {
                        ret = -1;
-                       goto out_free_config;
+                       goto out_wimlib_free;
                }
 
                /* 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 */
@@ -3950,7 +3865,7 @@ imagex_update(int argc, tchar **argv, int cmd)
                switch (cmds[i].op) {
                case WIMLIB_UPDATE_OP_ADD:
                        cmds[i].add.add_flags |= default_add_flags;
-                       cmds[i].add.config = config;
+                       cmds[i].add.config_file = config_file;
                        break;
                case WIMLIB_UPDATE_OP_DELETE:
                        cmds[i].delete_.delete_flags |= default_delete_flags;
@@ -3966,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);
@@ -3973,12 +3908,6 @@ out_free_cmds:
        free(cmds);
 out_free_cmd_file_contents:
        free(cmd_file_contents);
-out_free_config:
-       if (config) {
-               free(config->exclusion_pats.pats);
-               free(config->exclusion_exception_pats.pats);
-               free(config_str);
-       }
 out_wimlib_free:
        wimlib_free(wim);
 out_free_command_str:
@@ -4029,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(
@@ -4037,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(
@@ -4047,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(
@@ -4111,13 +4040,15 @@ T(
 [CMD_UNMOUNT] =
 T(
 "    %"TS" DIRECTORY [--commit] [--check] [--rebuild] [--lazy]\n"
+"                    [--new-image]\n"
 ),
 #endif
 [CMD_UPDATE] =
 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"
 ),
 };
 
@@ -4148,8 +4079,8 @@ version(void)
 {
        static const tchar *s =
        T(
-IMAGEX_PROGNAME " (" PACKAGE ") " PACKAGE_VERSION "\n"
-"Copyright (C) 2012, 2013 Eric Biggers\n"
+IMAGEX_PROGNAME " (distributed with " PACKAGE " " PACKAGE_VERSION ")\n"
+"Copyright (C) 2012, 2013, 2014 Eric Biggers\n"
 "License GPLv3+; GNU GPL version 3 or later <http://gnu.org/licenses/gpl.html>.\n"
 "This is free software: you are free to change and redistribute it.\n"
 "There is NO WARRANTY, to the extent permitted by law.\n"
@@ -4341,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);