]>
Commit | Line | Data |
---|---|---|
c23de0aa | 1 | /* Temporary file handling for tests. |
d614a753 | 2 | Copyright (C) 1998-2020 Free Software Foundation, Inc. |
c23de0aa FW |
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 | |
5a82c748 | 17 | <https://www.gnu.org/licenses/>. */ |
c23de0aa FW |
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> | |
c23de0aa FW |
28 | #include <stdio.h> |
29 | #include <stdlib.h> | |
30 | #include <string.h> | |
c22553ef | 31 | #include <unistd.h> |
c23de0aa FW |
32 | |
33 | /* List of temporary files. */ | |
34 | static struct temp_name_list | |
35 | { | |
706256af | 36 | struct temp_name_list *next; |
c23de0aa | 37 | char *name; |
c22553ef | 38 | pid_t owner; |
c23de0aa FW |
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. */ | |
43 | static const char *test_dir = _PATH_TMP; | |
44 | ||
45 | void | |
46 | add_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; | |
706256af | 54 | newp->next = temp_name_list; |
c22553ef | 55 | newp->owner = getpid (); |
706256af | 56 | temp_name_list = newp; |
c23de0aa FW |
57 | } |
58 | else | |
59 | free (newp); | |
60 | } | |
61 | ||
62 | int | |
63 | create_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 | ||
1ffe1ccb FW |
89 | char * |
90 | support_create_temp_directory (const char *base) | |
91 | { | |
8adfb0ee FW |
92 | char *path = xasprintf ("%s/%sXXXXXX", test_dir, base); |
93 | if (mkdtemp (path) == NULL) | |
1ffe1ccb | 94 | { |
8adfb0ee | 95 | printf ("error: mkdtemp (\"%s\"): %m", path); |
1ffe1ccb FW |
96 | exit (1); |
97 | } | |
8adfb0ee FW |
98 | add_temp_file (path); |
99 | return path; | |
1ffe1ccb FW |
100 | } |
101 | ||
c23de0aa FW |
102 | /* Helper functions called by the test skeleton follow. */ |
103 | ||
104 | void | |
105 | support_set_test_dir (const char *path) | |
106 | { | |
107 | test_dir = path; | |
108 | } | |
109 | ||
110 | void | |
111 | support_delete_temp_files (void) | |
112 | { | |
c22553ef | 113 | pid_t pid = getpid (); |
c23de0aa FW |
114 | while (temp_name_list != NULL) |
115 | { | |
c22553ef FW |
116 | /* Only perform the removal if the path was registed in the same |
117 | process, as identified by the PID. (This assumes that the | |
118 | parent process which registered the temporary file sticks | |
119 | around, to prevent PID reuse.) */ | |
120 | if (temp_name_list->owner == pid) | |
121 | { | |
122 | if (remove (temp_name_list->name) != 0) | |
123 | printf ("warning: could not remove temporary file: %s: %m\n", | |
124 | temp_name_list->name); | |
125 | } | |
c23de0aa FW |
126 | free (temp_name_list->name); |
127 | ||
706256af | 128 | struct temp_name_list *next = temp_name_list->next; |
c23de0aa FW |
129 | free (temp_name_list); |
130 | temp_name_list = next; | |
131 | } | |
132 | } | |
133 | ||
134 | void | |
135 | support_print_temp_files (FILE *f) | |
136 | { | |
137 | if (temp_name_list != NULL) | |
138 | { | |
139 | struct temp_name_list *n; | |
140 | fprintf (f, "temp_files=(\n"); | |
706256af | 141 | for (n = temp_name_list; n != NULL; n = n->next) |
c23de0aa FW |
142 | fprintf (f, " '%s'\n", n->name); |
143 | fprintf (f, ")\n"); | |
144 | } | |
145 | } |