Implement setting of Windows-specific XML information

[wimlib] / src / update_image.c

diff --git a/src/update_image.c b/src/update_image.c
index c76ae97131d728935db67c62d8f3377af2d8b312..ac07b78ac8dbfe0c40eb338efafb97fb9e57f0d0 100644 (file)
--- a/src/update_image.c
+++ b/src/update_image.c
@@ -64,6 +64,7 @@
 #include "wimlib/metadata.h"
 #include "wimlib/paths.h"
 #include "wimlib/progress.h"
+#include "wimlib/xml_windows.h"
 
 /* Saved specification of a "primitive" update operation that was performed.  */
 struct update_primitive {
@@ -869,6 +870,14 @@ execute_add_command(struct update_command_journal *j,
                        goto out_destroy_config;
        }
 
+       if (WIMLIB_IS_WIM_ROOT_PATH(wim_target_path) &&
+           get_dentry(wim, T("Windows"), WIMLIB_CASE_INSENSITIVE))
+       {
+               ret = set_windows_specific_info(wim);
+               if (ret)
+                       goto out_destroy_config;
+       }
+
        ret = 0;
 out_destroy_config:
        destroy_capture_config(&config);