1 /* PSPP - a program for statistical analysis.
2 Copyright (C) 2004, 2010 Free Software Foundation, Inc.
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation, either version 3 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>. */
19 #include "data/make-file.h"
30 #include "data/file-name.h"
31 #include "libpspp/ll.h"
32 #include "libpspp/message.h"
34 #include "gl/fatal-signal.h"
35 #include "gl/tempname.h"
36 #include "gl/xalloc.h"
37 #include "gl/xvasprintf.h"
40 #define _(msgid) gettext (msgid)
49 static struct ll_list all_files = LL_INITIALIZER (all_files);
51 static void free_replace_file (struct replace_file *);
52 static void unlink_replace_files (void);
55 replace_file_start (const char *file_name, const char *mode,
56 mode_t permissions, FILE **fp, char **tmp_name)
58 static bool registered;
60 struct replace_file *rf;
62 int saved_errno = errno;
64 /* If FILE_NAME represents a special file, write to it directly
65 instead of trying to replace it. */
66 if (stat (file_name, &s) == 0 && !S_ISREG (s.st_mode))
68 /* Open file descriptor. */
69 fd = open (file_name, O_WRONLY);
73 msg (ME, _("Opening %s for writing: %s."),
74 file_name, strerror (saved_errno));
79 /* Open file as stream. */
80 *fp = fdopen (fd, mode);
84 msg (ME, _("Opening stream for %s: %s."),
85 file_name, strerror (saved_errno));
91 rf = xmalloc (sizeof *rf);
93 rf->tmp_name = xstrdup (file_name);
95 *tmp_name = rf->tmp_name;
101 at_fatal_signal (unlink_replace_files);
104 block_fatal_signals ();
106 rf = xmalloc (sizeof *rf);
107 rf->file_name = xstrdup (file_name);
110 /* Generate unique temporary file name. */
111 rf->tmp_name = xasprintf ("%s.tmpXXXXXX", file_name);
112 if (gen_tempname (rf->tmp_name, 0, 0600, GT_NOCREATE) < 0)
114 msg (ME, _("Creating temporary file to replace %s: %s."),
115 rf->file_name, strerror (errno));
120 /* Create file by that name. */
121 fd = open (rf->tmp_name, O_WRONLY | O_CREAT | O_EXCL | O_BINARY, permissions);
126 msg (ME, _("Creating temporary file %s: %s."),
127 rf->tmp_name, strerror (errno));
135 /* Open file as stream. */
136 *fp = fdopen (fd, mode);
139 msg (ME, _("Opening stream for temporary file %s: %s."),
140 rf->tmp_name, strerror (errno));
142 unlink (rf->tmp_name);
147 /* Register file for deletion. */
148 ll_push_head (&all_files, &rf->ll);
149 unblock_fatal_signals ();
151 if (tmp_name != NULL)
152 *tmp_name = rf->tmp_name;
157 unblock_fatal_signals ();
158 free_replace_file (rf);
160 if (tmp_name != NULL)
167 replace_file_commit (struct replace_file *rf)
171 if (rf->file_name != NULL)
175 block_fatal_signals ();
176 ok = rename (rf->tmp_name, rf->file_name) == 0;
179 unblock_fatal_signals ();
182 msg (ME, _("Replacing %s by %s: %s."),
183 rf->tmp_name, rf->file_name, strerror (save_errno));
187 /* Special file: no temporary file to rename. */
189 free_replace_file (rf);
195 replace_file_abort (struct replace_file *rf)
199 if (rf->file_name != NULL)
203 block_fatal_signals ();
204 ok = unlink (rf->tmp_name) == 0;
207 unblock_fatal_signals ();
210 msg (ME, _("Removing %s: %s."), rf->tmp_name, strerror (save_errno));
214 /* Special file: no temporary file to unlink. */
216 free_replace_file (rf);
222 free_replace_file (struct replace_file *rf)
224 free (rf->file_name);
230 unlink_replace_files (void)
232 struct replace_file *rf;
234 block_fatal_signals ();
235 ll_for_each (rf, struct replace_file, ll, &all_files)
237 /* We don't free_replace_file(RF) because calling free is unsafe
238 from an asynchronous signal handler. */
239 unlink (rf->tmp_name);
241 unblock_fatal_signals ();