Add can_modify_wim(), can_delete_from_wim()

[wimlib] / src / add_image.c

diff --git a/src/add_image.c b/src/add_image.c
index 7dcee10b01a4a55ac2ded3dad802dd2007fe5c73..d40dfa097487d6efb1108775d9a30847e50f6339 100644 (file)
--- a/src/add_image.c
+++ b/src/add_image.c
@@ -80,11 +80,9 @@ wimlib_add_empty_image(WIMStruct *wim, const tchar *name, int *new_idx_ret)
        if (name == NULL)
                name = T("");
 
-       if (wim->hdr.total_parts != 1) {
-               ERROR("Cannot add an image to a split WIM");
-               ret = WIMLIB_ERR_SPLIT_UNSUPPORTED;
+       ret = can_modify_wim(wim);
+       if (ret)
                goto out;
-       }
 
        if (wimlib_image_name_in_use(wim, name)) {
                ERROR("There is already an image named \"%"TS"\" in the WIM!",