Generalized support for referencing resources in external WIMs

[wimlib] / src / dentry.c

diff --git a/src/dentry.c b/src/dentry.c
index ca9fe0b18993eedc604eac4d393665535fe1fe6d..a73828ba41c1ada58c7b9528d48033214b6c477c 100644 (file)
--- a/src/dentry.c
+++ b/src/dentry.c
@@ -2596,9 +2596,6 @@ wimlib_reference_template_image(WIMStruct *wim, int new_image, int template_imag
        int ret;
        struct wim_image_metadata *new_imd;
 
-       if (wim->hdr.part_number != 1)
-               return WIMLIB_ERR_SPLIT_UNSUPPORTED;
-
        if (new_image < 1 || new_image > wim->hdr.image_count)
                return WIMLIB_ERR_INVALID_IMAGE;
 
@@ -2608,6 +2605,9 @@ wimlib_reference_template_image(WIMStruct *wim, int new_image, int template_imag
        if (new_image == template_image)
                return WIMLIB_ERR_INVALID_PARAM;
 
+       if (!wim_has_metadata(wim))
+               return WIMLIB_ERR_METADATA_NOT_FOUND;
+
        new_imd = wim->image_metadata[new_image - 1];
        if (!new_imd->modified)
                return WIMLIB_ERR_INVALID_PARAM;