diff options
author | Karel Zak | 2014-09-10 12:51:25 +0200 |
---|---|---|
committer | Karel Zak | 2014-10-07 14:55:31 +0200 |
commit | 2d12903265367101c51d1b47d98a1457d93cd8ed (patch) | |
tree | eac4b9a52d758fdba187e3f0bff1e536bbc9bf19 | |
parent | sfdisk: improve UI (diff) | |
download | kernel-qcow2-util-linux-2d12903265367101c51d1b47d98a1457d93cd8ed.tar.gz kernel-qcow2-util-linux-2d12903265367101c51d1b47d98a1457d93cd8ed.tar.xz kernel-qcow2-util-linux-2d12903265367101c51d1b47d98a1457d93cd8ed.zip |
libfdisk: make yes/no dialogs more robust
Signed-off-by: Karel Zak <kzak@redhat.com>
-rw-r--r-- | disk-utils/fdisk.c | 16 | ||||
-rw-r--r-- | libfdisk/src/ask.c | 2 |
2 files changed, 13 insertions, 5 deletions
diff --git a/disk-utils/fdisk.c b/disk-utils/fdisk.c index 5fc4a4d81..ab4830302 100644 --- a/disk-utils/fdisk.c +++ b/disk-utils/fdisk.c @@ -338,10 +338,18 @@ int ask_callback(struct fdisk_context *cxt, struct fdisk_ask *ask, break; case FDISK_ASKTYPE_YESNO: fputc('\n', stdout); - fputs(fdisk_ask_get_query(ask), stdout); - rc = get_user_reply(cxt, _(" [Y]es/[N]o: "), buf, sizeof(buf)); - if (rc == 0) - fdisk_ask_yesno_set_result(ask, rpmatch(buf)); + do { + int x; + fputs(fdisk_ask_get_query(ask), stdout); + rc = get_user_reply(cxt, _(" [Y]es/[N]o: "), buf, sizeof(buf)); + if (rc) + break; + x = rpmatch(buf); + if (x == 1 || x == 0) { + fdisk_ask_yesno_set_result(ask, x); + break; + } + } while(1); DBG(ASK, ul_debug("yes-no ask: reply '%s' [rc=%d]", buf, rc)); break; case FDISK_ASKTYPE_STRING: diff --git a/libfdisk/src/ask.c b/libfdisk/src/ask.c index eb386ace9..c6e02f348 100644 --- a/libfdisk/src/ask.c +++ b/libfdisk/src/ask.c @@ -489,7 +489,7 @@ int fdisk_ask_yesno(struct fdisk_context *cxt, if (!rc) rc = fdisk_do_ask(cxt, ask); if (!rc) - *result = fdisk_ask_yesno_get_result(ask); + *result = fdisk_ask_yesno_get_result(ask) == 1 ? 1 : 0; DBG(ASK, ul_debugobj(ask, "result: %d [rc=%d]\n", *result, rc)); fdisk_free_ask(ask); |