wimexport: Add --wimboot flag

[wimlib] / src / export_image.c

diff --git a/src/export_image.c b/src/export_image.c
index 8304ea01a8fda9af7cc622456679899e148a8fd4..9c25be732eb8d9424bd30ecd905ec50159de271d 100644 (file)
--- a/src/export_image.c
+++ b/src/export_image.c
@@ -124,7 +124,8 @@ wimlib_export_image(WIMStruct *src_wim,
        if (export_flags & ~(WIMLIB_EXPORT_FLAG_BOOT |
                             WIMLIB_EXPORT_FLAG_NO_NAMES |
                             WIMLIB_EXPORT_FLAG_NO_DESCRIPTIONS |
-                            WIMLIB_EXPORT_FLAG_GIFT))
+                            WIMLIB_EXPORT_FLAG_GIFT |
+                            WIMLIB_EXPORT_FLAG_WIMBOOT))
                return WIMLIB_ERR_INVALID_PARAM;
 
        if (src_wim == NULL || dest_wim == NULL)
@@ -263,6 +264,13 @@ wimlib_export_image(WIMStruct *src_wim,
                        dest_wim->hdr.boot_idx = dest_wim->hdr.image_count;
                }
 
+               /* Possibly set WIMBoot flag  */
+               if (export_flags & WIMLIB_EXPORT_FLAG_WIMBOOT) {
+                       wim_info_set_wimboot(dest_wim->wim_info,
+                                            dest_wim->hdr.image_count,
+                                            true);
+               }
+
        }
        /* Set the reparse point fixup flag on the destination WIM if the flag
         * is set on the source WIM. */