sys-utils: verify writing to streams was successful

[thirdparty/util-linux.git] / sys-utils / flock.c

diff --git a/sys-utils/flock.c b/sys-utils/flock.c
index 50cea13920ca045b12275351ef95d08a5986fe96..1db28a294fcbf80df249e2d7c5e1be2c2e857a78 100644 (file)
--- a/sys-utils/flock.c
+++ b/sys-utils/flock.c
@@ -42,6 +42,7 @@
 #include "c.h"
 #include "nls.h"
 #include "strutils.h"
+#include "closestream.h"
 
 static void __attribute__((__noreturn__)) usage(int ex)
 {
@@ -161,6 +162,7 @@ int main(int argc, char *argv[])
        setlocale(LC_ALL, "");
        bindtextdomain(PACKAGE, LOCALEDIR);
        textdomain(PACKAGE);
+       atexit(close_stdout);
 
        if (argc < 2)
                usage(EX_USAGE);