1/* Temporary file handling for tests.
2 Copyright (C) 1998-2017 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <http://www.gnu.org/licenses/>. */
18
19/* This is required to get an mkstemp which can create large files on
20 some 32-bit platforms. */
21#define _FILE_OFFSET_BITS 64
22
23#include <support/temp_file.h>
24#include <support/temp_file-internal.h>
25#include <support/support.h>
26
27#include <paths.h>
28#include <stdio.h>
29#include <stdlib.h>
30#include <string.h>
31#include <unistd.h>
32
33/* List of temporary files. */
34static struct temp_name_list
35{
36 struct temp_name_list *next;
37 char *name;
38 pid_t owner;
39} *temp_name_list;
40
41/* Location of the temporary files. Set by the test skeleton via
42 support_set_test_dir. The string is not be freed. */
43static const char *test_dir = _PATH_TMP;
44
45void
46add_temp_file (const char *name)
47{
48 struct temp_name_list *newp
49 = (struct temp_name_list *) xcalloc (sizeof (*newp), 1);
50 char *newname = strdup (name);
51 if (newname != NULL)
52 {
53 newp->name = newname;
54 newp->next = temp_name_list;
55 newp->owner = getpid ();
56 temp_name_list = newp;
57 }
58 else
59 free (newp);
60}
61
62int
63create_temp_file (const char *base, char **filename)
64{
65 char *fname;
66 int fd;
67
68 fname = (char *) xmalloc (strlen (test_dir) + 1 + strlen (base)
69 + sizeof ("XXXXXX"));
70 strcpy (stpcpy (stpcpy (stpcpy (fname, test_dir), "/"), base), "XXXXXX");
71
72 fd = mkstemp (fname);
73 if (fd == -1)
74 {
75 printf ("cannot open temporary file '%s': %m\n", fname);
76 free (fname);
77 return -1;
78 }
79
80 add_temp_file (fname);
81 if (filename != NULL)
82 *filename = fname;
83 else
84 free (fname);
85
86 return fd;
87}
88
89/* Helper functions called by the test skeleton follow. */
90
91void
92support_set_test_dir (const char *path)
93{
94 test_dir = path;
95}
96
97void
98support_delete_temp_files (void)
99{
100 pid_t pid = getpid ();
101 while (temp_name_list != NULL)
102 {
103 /* Only perform the removal if the path was registed in the same
104 process, as identified by the PID. (This assumes that the
105 parent process which registered the temporary file sticks
106 around, to prevent PID reuse.) */
107 if (temp_name_list->owner == pid)
108 {
109 if (remove (temp_name_list->name) != 0)
110 printf ("warning: could not remove temporary file: %s: %m\n",
111 temp_name_list->name);
112 }
113 free (temp_name_list->name);
114
115 struct temp_name_list *next = temp_name_list->next;
116 free (temp_name_list);
117 temp_name_list = next;
118 }
119}
120
121void
122support_print_temp_files (FILE *f)
123{
124 if (temp_name_list != NULL)
125 {
126 struct temp_name_list *n;
127 fprintf (f, "temp_files=(\n");
128 for (n = temp_name_list; n != NULL; n = n->next)
129 fprintf (f, " '%s'\n", n->name);
130 fprintf (f, ")\n");
131 }
132}
133