From 1d3762ee53b17122f8ca33db4f38094345be5c83 Mon Sep 17 00:00:00 2001 From: Sami Kerola Date: Sat, 13 Apr 2013 20:54:38 +0100 Subject: [PATCH] cfdisk: check writing to a file was successful Signed-off-by: Sami Kerola --- fdisks/cfdisk.c | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/fdisks/cfdisk.c b/fdisks/cfdisk.c index 136ff9d2f6..a68f855a4e 100644 --- a/fdisks/cfdisk.c +++ b/fdisks/cfdisk.c @@ -1849,7 +1849,11 @@ print_raw_table(void) { if (to_file) { if (!print_only) - fclose(fp); + if (close_stream(fp) != 0) { + char errstr[LINE_LENGTH]; + snprintf(errstr, sizeof(errstr), _("write failed: %s"), fname); + print_warning(errstr); + } } else { menuContinue(); } @@ -1966,7 +1970,11 @@ print_p_info(void) { if (to_file) { if (!print_only) - fclose(fp); + if (close_stream(fp) != 0) { + char errstr[LINE_LENGTH]; + snprintf(errstr, sizeof(errstr), _("write failed: %s"), fname); + print_warning(errstr); + } } else { menuContinue(); } @@ -2060,7 +2068,11 @@ print_part_table(void) { if (to_file) { if (!print_only) - fclose(fp); + if (close_stream(fp) != 0) { + char errstr[LINE_LENGTH]; + snprintf(errstr, sizeof(errstr), _("write failed: %s"), fname); + print_warning(errstr); + } } else { menuContinue(); } -- 2.47.2