/* closeout.c - close standard output
- Copyright (C) 1998, 1999 Free Software Foundation, Inc.
+ Copyright (C) 1998, 1999, 2000 Free Software Foundation, Inc.
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
# define EXIT_FAILURE 1
#endif
+#include <stdio.h>
+
#include <errno.h>
#ifndef errno
extern int errno;
#endif
-#include <stdio.h>
#include "closeout.h"
#include "error.h"
+#include "quotearg.h"
+
+static int default_exit_status = EXIT_FAILURE;
+static const char *file_name;
+
+/* Set the value to be used for the exit status when close_stdout is called.
+ This is useful when it is not convenient to call close_stdout_status,
+ e.g., when close_stdout is called via atexit. */
+void
+close_stdout_set_status (int status)
+{
+ default_exit_status = status;
+}
+
+/* Set the file name to be reported in the event an error is detected
+ by close_stdout_status. */
+void
+close_stdout_set_file_name (const char *file)
+{
+ file_name = file;
+}
/* Close standard output, exiting with status STATUS on failure.
If a program writes *anything* to stdout, that program should `fflush'
void
close_stdout_status (int status)
{
- if (ferror (stdout))
- error (status, 0, _("write error"));
+ int e = ferror (stdout) ? 0 : -1;
+
if (fclose (stdout) != 0)
- error (status, errno, _("write error"));
+ e = errno;
+
+ if (0 <= e)
+ {
+ char const *write_error = _("write error");
+ if (file_name)
+ error (status, e, "%s: %s", quotearg_colon (file_name), write_error);
+ else
+ error (status, e, "%s", write_error);
+ }
}
/* Close standard output, exiting with status EXIT_FAILURE on failure. */
void
close_stdout (void)
{
- close_stdout_status (EXIT_FAILURE);
+ close_stdout_status (default_exit_status);
}