diff options
author | Karel Zak | 2016-05-18 14:15:02 +0200 |
---|---|---|
committer | Karel Zak | 2016-05-18 14:15:02 +0200 |
commit | a67054f9be6998cd111ab81e4ca8931128569f56 (patch) | |
tree | c0c6596e792a39646dc4f9a40a2c531abb731e04 /libfdisk/src/script.c | |
parent | sfdisk: Add support for altering GPT size (diff) | |
download | kernel-qcow2-util-linux-a67054f9be6998cd111ab81e4ca8931128569f56.tar.gz kernel-qcow2-util-linux-a67054f9be6998cd111ab81e4ca8931128569f56.tar.xz kernel-qcow2-util-linux-a67054f9be6998cd111ab81e4ca8931128569f56.zip |
libfdisk: make table-length usage more robust
Signed-off-by: Karel Zak <kzak@redhat.com>
Diffstat (limited to 'libfdisk/src/script.c')
-rw-r--r-- | libfdisk/src/script.c | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/libfdisk/src/script.c b/libfdisk/src/script.c index fd05bf703..75e1b52de 100644 --- a/libfdisk/src/script.c +++ b/libfdisk/src/script.c @@ -1346,10 +1346,15 @@ int fdisk_apply_script_headers(struct fdisk_context *cxt, struct fdisk_script *d return rc; str = fdisk_script_get_header(dp, "table-length"); - if (str) - return fdisk_gpt_set_npartitions(cxt, strtoul(str, NULL, 0)); + if (str) { + uintmax_t sz; - return 0; + rc = parse_size(str, &sz, NULL); + if (rc == 0) + rc = fdisk_gpt_set_npartitions(cxt, sz); + } + + return rc; } /** |