From dd79e7c9d0f9c0d8a6c72a86a8a96b092d59bb45 Mon Sep 17 00:00:00 2001 From: Sami Kerola Date: Sat, 13 Apr 2013 20:54:47 +0100 Subject: resizepart: check writing to a file descriptor was successful Signed-off-by: Sami Kerola --- disk-utils/resizepart.c | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'disk-utils/resizepart.c') diff --git a/disk-utils/resizepart.c b/disk-utils/resizepart.c index 5517c4ff2..765e3a406 100644 --- a/disk-utils/resizepart.c +++ b/disk-utils/resizepart.c @@ -11,6 +11,7 @@ #include "partx.h" #include "sysfs.h" #include "strutils.h" +#include "closestream.h" static void __attribute__ ((__noreturn__)) usage(FILE * out) { @@ -73,6 +74,7 @@ int main(int argc, char **argv) setlocale(LC_ALL, ""); bindtextdomain(PACKAGE, LOCALEDIR); textdomain(PACKAGE); + atexit(close_stdout); while ((c = getopt_long(argc, argv, "Vh", longopts, NULL)) != -1) switch (c) { @@ -102,5 +104,8 @@ int main(int argc, char **argv) strtou64_or_err(argv[3], _("invalid length argument")))) err(EXIT_FAILURE, _("failed to resize partition")); + if (close_fd(fd) != 0) + err(EXIT_FAILURE, _("write failed")); + return 0; } -- cgit v1.2.3-55-g7522