exit with error if fflush() fail

Tested with closed stdout. Mentioned in #2.
lnde · Sep 19, 2023 · 70343ab · 70343ab
1 parent 778b542
commit 70343ab
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/output.c b/output.c
@@ -16,9 +16,11 @@
 
 #include <sys/time.h>
 
+#include <errno.h>
 #include <inttypes.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
 #include <time.h>
 
 #include <pcap.h>
@@ -203,7 +205,10 @@ print_stats(int unused)
 	}
 
 	fprintf(stdout, "\n");
-	fflush(stdout);
+	if (fflush(stdout) == EOF) {
+		fprintf(stderr, "Error: Couldn't write to stdout: %s\n", strerror(errno));
+		exit(EXIT_FAILURE);
+	}
 
 	if (announceCount == FLG_CNT)
 		exit(EXIT_SUCCESS);