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)
42 /* Non ansi compilers may set this */
44 #define P_tmpdir "/tmp"
47 /* Creates a temporary file and stores its name in *FILE_NAME and
48 a file descriptor for it in *FD. Returns success. Caller is
49 responsible for freeing *FILE_NAME. */
51 make_temp_file (int *fd, char **file_name)
53 const char *parent_dir;
55 assert (file_name != NULL);
58 if (getenv ("TMPDIR") != NULL)
59 parent_dir = getenv ("TMPDIR");
61 parent_dir = P_tmpdir;
63 *file_name = xmalloc (strlen (parent_dir) + 32);
64 sprintf (*file_name, "%s/psppXXXXXX", parent_dir);
65 *fd = mkstemp (*file_name);
68 msg (ME, _("%s: Creating temporary file: %s."),
69 *file_name, strerror (errno));
78 /* Creates a temporary file and stores its name in *FILE_NAME and
79 a file stream for it in *FP. Returns success. Caller is
80 responsible for freeing *FILE_NAME and for closing *FP */
82 make_unique_file_stream (FILE **fp, char **file_name)
84 static int serial = 0;
85 const char *parent_dir;
89 Need to check for pre-existing file name.
90 Need also to pass in the directory instead of using /tmp
93 assert (file_name != NULL);
96 if (getenv ("TMPDIR") != NULL)
97 parent_dir = getenv ("TMPDIR");
99 parent_dir = P_tmpdir;
101 *file_name = xmalloc (strlen (parent_dir) + 32);
104 sprintf (*file_name, "%s/pspp%d.png", parent_dir, serial++);
106 *fp = fopen(*file_name, "w");
110 msg (ME, _("%s: Creating file: %s."), *file_name, strerror (errno));
126 static struct ll_list all_files = LL_INITIALIZER (all_files);
128 static void free_replace_file (struct replace_file *);
129 static void unlink_replace_files (void);
131 struct replace_file *
132 replace_file_start (const char *file_name, const char *mode,
133 mode_t permissions, FILE **fp, char **tmp_name)
135 static bool registered;
137 struct replace_file *rf;
140 /* If FILE_NAME represents a special file, write to it directly
141 instead of trying to replace it. */
142 if (stat (file_name, &s) == 0 && !S_ISREG (s.st_mode))
144 /* Open file descriptor. */
145 fd = open (file_name, O_WRONLY);
148 msg (ME, _("Opening %s for writing: %s."),
149 file_name, strerror (errno));
153 /* Open file as stream. */
154 *fp = fdopen (fd, mode);
157 msg (ME, _("Opening stream for %s: %s."),
158 file_name, strerror (errno));
163 rf = xmalloc (sizeof *rf);
164 rf->file_name = NULL;
165 rf->tmp_name = xstrdup (file_name);
166 if (tmp_name != NULL)
167 *tmp_name = rf->tmp_name;
173 at_fatal_signal (unlink_replace_files);
176 block_fatal_signals ();
178 rf = xmalloc (sizeof *rf);
179 rf->file_name = xstrdup (file_name);
182 /* Generate unique temporary file name. */
183 rf->tmp_name = xasprintf ("%s.tmpXXXXXX", file_name);
184 if (gen_tempname (rf->tmp_name, 0, 0600, GT_NOCREATE) < 0)
186 msg (ME, _("Creating temporary file to replace %s: %s."),
187 rf->file_name, strerror (errno));
191 /* Create file by that name. */
192 fd = open (rf->tmp_name, O_WRONLY | O_CREAT | O_EXCL | O_BINARY, permissions);
197 msg (ME, _("Creating temporary file %s: %s."),
198 rf->tmp_name, strerror (errno));
205 /* Open file as stream. */
206 *fp = fdopen (fd, mode);
209 msg (ME, _("Opening stream for temporary file %s: %s."),
210 rf->tmp_name, strerror (errno));
212 unlink (rf->tmp_name);
216 /* Register file for deletion. */
217 ll_push_head (&all_files, &rf->ll);
218 unblock_fatal_signals ();
220 if (tmp_name != NULL)
221 *tmp_name = rf->tmp_name;
226 unblock_fatal_signals ();
227 free_replace_file (rf);
229 if (tmp_name != NULL)
235 replace_file_commit (struct replace_file *rf)
239 if (rf->file_name != NULL)
243 block_fatal_signals ();
244 ok = rename (rf->tmp_name, rf->file_name) == 0;
247 unblock_fatal_signals ();
250 msg (ME, _("Replacing %s by %s: %s."),
251 rf->tmp_name, rf->file_name, strerror (save_errno));
255 /* Special file: no temporary file to rename. */
257 free_replace_file (rf);
263 replace_file_abort (struct replace_file *rf)
267 if (rf->file_name != NULL)
271 block_fatal_signals ();
272 ok = unlink (rf->tmp_name) == 0;
275 unblock_fatal_signals ();
278 msg (ME, _("Removing %s: %s."), rf->tmp_name, strerror (save_errno));
282 /* Special file: no temporary file to unlink. */
284 free_replace_file (rf);
290 free_replace_file (struct replace_file *rf)
292 free (rf->file_name);
298 unlink_replace_files (void)
300 struct replace_file *rf;
302 block_fatal_signals ();
303 ll_for_each (rf, struct replace_file, ll, &all_files)
305 /* We don't free_replace_file(RF) because calling free is unsafe
306 from an asynchronous signal handler. */
307 unlink (rf->tmp_name);
310 unblock_fatal_signals ();