fdisk: verify writing to streams was successful

[thirdparty/util-linux.git] / fdisk / gpt.c

diff --git a/fdisk/gpt.c b/fdisk/gpt.c
index aeacd662456997f3251d13bdac48fa63987aca83..bb6911a218d37fd02e5275490f1204a7d0e9ca6b 100644 (file)
--- a/fdisk/gpt.c
+++ b/fdisk/gpt.c
@@ -39,6 +39,7 @@
 #include "gpt.h"
 #include "blkdev.h"
 #include "bitops.h"
+#include "closestream.h"
 
 #define GPT_HEADER_SIGNATURE 0x5452415020494645LL
 #define GPT_PRIMARY_PARTITION_TABLE_LBA 1
@@ -202,6 +203,7 @@ gpt_probe_signature_devname(char *devname)
 int
 main(int argc, char **argv)
 {
+       atexit(close_stdout);
        if (argc!=2)
        {
                fprintf(stderr, "usage: %s <dev>\n", argv[0]);