*/
/*
- * Copyright (C) 2012-2017 Eric Biggers
+ * Copyright (C) 2012-2018 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
IMAGEX_FORCE_OPTION,
IMAGEX_HEADER_OPTION,
IMAGEX_IMAGE_PROPERTY_OPTION,
+ IMAGEX_INCLUDE_INTEGRITY_OPTION,
IMAGEX_INCLUDE_INVALID_NAMES_OPTION,
IMAGEX_LAZY_OPTION,
IMAGEX_METADATA_OPTION,
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
{T("no-check"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
{T("nocheck"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_OPTION},
{T("compress"), required_argument, NULL, IMAGEX_COMPRESS_OPTION},
{T("chunk-size"), required_argument, NULL, IMAGEX_CHUNK_SIZE_OPTION},
{T("solid"), no_argument, NULL, IMAGEX_SOLID_OPTION},
static const struct option delete_options[] = {
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_OPTION},
{T("soft"), no_argument, NULL, IMAGEX_SOFT_OPTION},
{T("unsafe-compact"), no_argument, NULL, IMAGEX_UNSAFE_COMPACT_OPTION},
{NULL, 0, NULL, 0},
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
{T("nocheck"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
{T("no-check"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_OPTION},
{T("compress"), required_argument, NULL, IMAGEX_COMPRESS_OPTION},
{T("recompress"), no_argument, NULL, IMAGEX_RECOMPRESS_OPTION},
{T("chunk-size"), required_argument, NULL, IMAGEX_CHUNK_SIZE_OPTION},
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
{T("nocheck"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
{T("no-check"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_OPTION},
{T("extract-xml"), required_argument, NULL, IMAGEX_EXTRACT_XML_OPTION},
{T("header"), no_argument, NULL, IMAGEX_HEADER_OPTION},
{T("lookup-table"), no_argument, NULL, IMAGEX_BLOBS_OPTION},
static const struct option join_options[] = {
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_OPTION},
{NULL, 0, NULL, 0},
};
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
{T("nocheck"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
{T("no-check"), no_argument, NULL, IMAGEX_NOCHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_OPTION},
{T("compress"), required_argument, NULL, IMAGEX_COMPRESS_OPTION},
{T("recompress"), no_argument, NULL, IMAGEX_RECOMPRESS_OPTION},
{T("chunk-size"), required_argument, NULL, IMAGEX_CHUNK_SIZE_OPTION},
static const struct option split_options[] = {
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_OPTION},
{NULL, 0, NULL, 0},
};
* update_command_add_option(). */
{T("threads"), required_argument, NULL, IMAGEX_THREADS_OPTION},
{T("check"), no_argument, NULL, IMAGEX_CHECK_OPTION},
+ {T("include-integrity"), no_argument, NULL, IMAGEX_INCLUDE_INTEGRITY_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},
STRING_LIST(base_wimfiles);
WIMStruct **base_wims;
- WIMStruct *template_wim;
+ WIMStruct *template_wim = NULL;
const tchar *template_wimfile = NULL;
const tchar *template_image_name_or_num = NULL;
int template_image = WIMLIB_NO_IMAGE;
break;
case IMAGEX_CHECK_OPTION:
open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
case IMAGEX_NOCHECK_OPTION:
* open the WIM if needed and parse the image index. */
if (template_image_name_or_num) {
-
- if (base_wimfiles.num_strings == 1 &&
- template_wimfile == base_wimfiles.strings[0]) {
- template_wim = base_wims[0];
- } else if (template_wimfile == wimfile) {
+ if (cmd == CMD_APPEND && !tstrcmp(template_wimfile, wimfile)) {
template_wim = wim;
} else {
+ for (size_t i = 0; i < base_wimfiles.num_strings; i++) {
+ if (!tstrcmp(template_wimfile,
+ base_wimfiles.strings[i])) {
+ template_wim = base_wims[i];
+ break;
+ }
+ }
+ }
+
+ if (!template_wim) {
ret = wimlib_open_wim_with_progress(template_wimfile,
open_flags,
&template_wim,
template_wimfile);
if (ret)
goto out_free_template_wim;
- } else {
- template_wim = NULL;
}
ret = wimlib_add_image_multisource(wim,
write_flags, num_threads);
}
out_free_template_wim:
- /* template_wim may alias base_wims[0] or wim. */
- if ((base_wimfiles.num_strings != 1 || template_wim != base_wims[0]) &&
- template_wim != wim)
- wimlib_free(template_wim);
+ /* 'template_wim' may alias 'wim' or any of the 'base_wims' */
+ if (template_wim == wim)
+ goto out_free_base_wims;
+ for (size_t i = 0; i < base_wimfiles.num_strings; i++)
+ if (template_wim == base_wims[i])
+ goto out_free_base_wims;
+ wimlib_free(template_wim);
out_free_base_wims:
for (size_t i = 0; i < base_wimfiles.num_strings; i++)
wimlib_free(base_wims[i]);
switch (c) {
case IMAGEX_CHECK_OPTION:
open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
case IMAGEX_SOFT_OPTION:
break;
case IMAGEX_CHECK_OPTION:
open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
case IMAGEX_NOCHECK_OPTION:
{
int c;
bool boot = false;
- bool check = false;
- bool nocheck = false;
bool header = false;
bool blobs = false;
bool xml = false;
int image;
int ret;
int open_flags = 0;
+ int write_flags = 0;
struct wimlib_wim_info info;
for_opt(c, info_options) {
boot = true;
break;
case IMAGEX_CHECK_OPTION:
- check = true;
+ open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
+ write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
case IMAGEX_NOCHECK_OPTION:
- nocheck = true;
+ write_flags |= WIMLIB_WRITE_FLAG_NO_CHECK_INTEGRITY;
break;
case IMAGEX_HEADER_OPTION:
header = true;
goto out;
}
- if (check && nocheck) {
- imagex_error(T("Can't specify both --check and --nocheck"));
- goto out_err;
- }
-
- if (check)
- open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
-
ret = wimlib_open_wim_with_progress(wimfile, open_flags, &wim,
imagex_progress_func, NULL);
if (ret)
/* Only call wimlib_overwrite() if something actually needs to
* be changed. */
if (boot || any_property_changes ||
- (check && !info.has_integrity_table) ||
- (nocheck && info.has_integrity_table))
+ ((write_flags & WIMLIB_WRITE_FLAG_CHECK_INTEGRITY) &&
+ !info.has_integrity_table) ||
+ ((write_flags & WIMLIB_WRITE_FLAG_NO_CHECK_INTEGRITY) &&
+ info.has_integrity_table))
{
- int write_flags = 0;
-
- if (check)
- write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
- if (nocheck)
- write_flags |= WIMLIB_WRITE_FLAG_NO_CHECK_INTEGRITY;
ret = wimlib_overwrite(wim, write_flags, 1);
} else {
imagex_printf(T("The file \"%"TS"\" was not modified "
out_usage:
usage(CMD_INFO, stderr);
-out_err:
ret = -1;
goto out;
}
switch (c) {
case IMAGEX_CHECK_OPTION:
swm_open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
wim_write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
default:
switch (c) {
case IMAGEX_CHECK_OPTION:
open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
case IMAGEX_NOCHECK_OPTION:
switch (c) {
case IMAGEX_CHECK_OPTION:
open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
default:
break;
case IMAGEX_CHECK_OPTION:
open_flags |= WIMLIB_OPEN_FLAG_CHECK_INTEGRITY;
+ /* fall-through */
+ case IMAGEX_INCLUDE_INTEGRITY_OPTION:
write_flags |= WIMLIB_WRITE_FLAG_CHECK_INTEGRITY;
break;
case IMAGEX_REBUILD_OPTION:
static void
version(void)
{
- static const tchar * const s =
+ static const tchar * const fmt =
T(
-"wimlib-imagex (distributed with " PACKAGE " " PACKAGE_VERSION ")\n"
-"Copyright (C) 2012-2017 Eric Biggers\n"
+"wimlib-imagex " PACKAGE_VERSION " (using wimlib %"TS")\n"
+"Copyright (C) 2012-2018 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"
"\n"
"Report bugs to "PACKAGE_BUGREPORT".\n"
);
- tfputs(s, stdout);
+ tfprintf(stdout, fmt, wimlib_get_version_string());
}
-
static void
do_common_options(int *argc_p, tchar **argv, int cmd)
{