libfdisk: do not reset default if undefined by script

Signed-off-by: Karel Zak <kzak@redhat.com>

diff --git a/libfdisk/src/gpt.c b/libfdisk/src/gpt.c
index 1a086a61d8fbb891758895a1f6beb2027f812fb1..97b161312e59f0ee465840d89904278c71b33f09 100644 (file)
--- a/libfdisk/src/gpt.c
+++ b/libfdisk/src/gpt.c
@@ -758,7 +758,9 @@ static int count_first_last_lba(struct fdisk_context *cxt,
        /* Get the table length from the script, if given */
        if (cxt->script) {
                rc = get_script_u64(cxt, &nents, "table-length");
-               if (rc < 0)
+               if (rc == 1)
+                       nents = GPT_NPARTITIONS;  /* undefined by script */
+               else if (rc < 0)
                        return rc;
        }