]>
Commit | Line | Data |
---|---|---|
1a9d15db MH |
1 | #ifndef TEMPFILE_H |
2 | #define TEMPFILE_H | |
3 | ||
24d82185 | 4 | #include "list.h" |
ef3ca954 | 5 | #include "strbuf.h" |
24d82185 | 6 | |
1a9d15db MH |
7 | /* |
8 | * Handle temporary files. | |
9 | * | |
10 | * The tempfile API allows temporary files to be created, deleted, and | |
11 | * atomically renamed. Temporary files that are still active when the | |
12 | * program ends are cleaned up automatically. Lockfiles (see | |
13 | * "lockfile.h") are built on top of this API. | |
14 | * | |
15 | * | |
16 | * Calling sequence | |
17 | * ---------------- | |
18 | * | |
19 | * The caller: | |
20 | * | |
1a9d15db | 21 | * * Attempts to create a temporary file by calling |
076aa2cb JK |
22 | * `create_tempfile()`. The resources used for the temporary file are |
23 | * managed by the tempfile API. | |
1a9d15db MH |
24 | * |
25 | * * Writes new content to the file by either: | |
26 | * | |
076aa2cb | 27 | * * writing to the `tempfile->fd` file descriptor |
1a9d15db MH |
28 | * |
29 | * * calling `fdopen_tempfile()` to get a `FILE` pointer for the | |
30 | * open file and writing to the file using stdio. | |
31 | * | |
076aa2cb | 32 | * Note that the file descriptor created by create_tempfile() |
05d1ed61 BW |
33 | * is marked O_CLOEXEC, so the new contents must be written by |
34 | * the current process, not any spawned one. | |
35 | * | |
1a9d15db MH |
36 | * When finished writing, the caller can: |
37 | * | |
38 | * * Close the file descriptor and remove the temporary file by | |
39 | * calling `delete_tempfile()`. | |
40 | * | |
41 | * * Close the temporary file and rename it atomically to a specified | |
42 | * filename by calling `rename_tempfile()`. This relinquishes | |
43 | * control of the file. | |
44 | * | |
45 | * * Close the file descriptor without removing or renaming the | |
49bd0fc2 | 46 | * temporary file by calling `close_tempfile_gently()`, and later call |
1a9d15db MH |
47 | * `delete_tempfile()` or `rename_tempfile()`. |
48 | * | |
422a21c6 | 49 | * After the temporary file is renamed or deleted, the `tempfile` |
076aa2cb | 50 | * object is no longer valid and should not be reused. |
1a9d15db MH |
51 | * |
52 | * If the program exits before `rename_tempfile()` or | |
53 | * `delete_tempfile()` is called, an `atexit(3)` handler will close | |
54 | * and remove the temporary file. | |
55 | * | |
56 | * If you need to close the file descriptor yourself, do so by calling | |
49bd0fc2 | 57 | * `close_tempfile_gently()`. You should never call `close(2)` or `fclose(3)` |
1a9d15db MH |
58 | * yourself, otherwise the `struct tempfile` structure would still |
59 | * think that the file descriptor needs to be closed, and a later | |
60 | * cleanup would result in duplicate calls to `close(2)`. Worse yet, | |
61 | * if you close and then later open another file descriptor for a | |
62 | * completely different purpose, then the unrelated file descriptor | |
63 | * might get closed. | |
64 | * | |
65 | * | |
66 | * Error handling | |
67 | * -------------- | |
68 | * | |
076aa2cb JK |
69 | * `create_tempfile()` returns an allocated tempfile on success or NULL |
70 | * on failure. On errors, `errno` describes the reason for failure. | |
1a9d15db | 71 | * |
5de134ca MÅ |
72 | * `rename_tempfile()` and `close_tempfile_gently()` return 0 on success. |
73 | * On failure they set `errno` appropriately and return -1. | |
74 | * `delete_tempfile()` and `rename` (but not `close`) do their best to | |
75 | * delete the temporary file before returning. | |
1a9d15db MH |
76 | */ |
77 | ||
78 | struct tempfile { | |
24d82185 | 79 | volatile struct volatile_list_head list; |
1a9d15db MH |
80 | volatile sig_atomic_t active; |
81 | volatile int fd; | |
82 | FILE *volatile fp; | |
83 | volatile pid_t owner; | |
1a9d15db MH |
84 | struct strbuf filename; |
85 | }; | |
86 | ||
87 | /* | |
88 | * Attempt to create a temporary file at the specified `path`. Return | |
076aa2cb JK |
89 | * a tempfile (whose "fd" member can be used for writing to it), or |
90 | * NULL on error. It is an error if a file already exists at that path. | |
bef0413c TB |
91 | * Note that `mode` will be further modified by the umask, and possibly |
92 | * `core.sharedRepository`, so it is not guaranteed to have the given | |
93 | * mode. | |
1a9d15db | 94 | */ |
bef0413c TB |
95 | struct tempfile *create_tempfile_mode(const char *path, int mode); |
96 | ||
97 | static inline struct tempfile *create_tempfile(const char *path) | |
98 | { | |
99 | return create_tempfile_mode(path, 0666); | |
100 | } | |
1a9d15db | 101 | |
99397152 MH |
102 | /* |
103 | * Register an existing file as a tempfile, meaning that it will be | |
104 | * deleted when the program exits. The tempfile is considered closed, | |
105 | * but it can be worked with like any other closed tempfile (for | |
106 | * example, it can be opened using reopen_tempfile()). | |
107 | */ | |
55454427 | 108 | struct tempfile *register_tempfile(const char *path); |
99397152 | 109 | |
354ab112 MH |
110 | |
111 | /* | |
112 | * mks_tempfile functions | |
113 | * | |
114 | * The following functions attempt to create and open temporary files | |
115 | * with names derived automatically from a template, in the manner of | |
116 | * mkstemps(), and arrange for them to be deleted if the program ends | |
117 | * before they are deleted explicitly. There is a whole family of such | |
118 | * functions, named according to the following pattern: | |
119 | * | |
120 | * x?mks_tempfile_t?s?m?() | |
121 | * | |
122 | * The optional letters have the following meanings: | |
123 | * | |
124 | * x - die if the temporary file cannot be created. | |
125 | * | |
126 | * t - create the temporary file under $TMPDIR (as opposed to | |
127 | * relative to the current directory). When these variants are | |
128 | * used, template should be the pattern for the filename alone, | |
129 | * without a path. | |
130 | * | |
131 | * s - template includes a suffix that is suffixlen characters long. | |
132 | * | |
133 | * m - the temporary file should be created with the specified mode | |
134 | * (otherwise, the mode is set to 0600). | |
135 | * | |
136 | * None of these functions modify template. If the caller wants to | |
137 | * know the (absolute) path of the file that was created, it can be | |
138 | * read from tempfile->filename. | |
139 | * | |
076aa2cb JK |
140 | * On success, the functions return a tempfile whose "fd" member is open |
141 | * for writing the temporary file. On errors, they return NULL and set | |
142 | * errno appropriately (except for the "x" variants, which die() on | |
143 | * errors). | |
354ab112 MH |
144 | */ |
145 | ||
146 | /* See "mks_tempfile functions" above. */ | |
55454427 | 147 | struct tempfile *mks_tempfile_sm(const char *filename_template, |
ad6dad09 | 148 | int suffixlen, int mode); |
354ab112 MH |
149 | |
150 | /* See "mks_tempfile functions" above. */ | |
ea8ace4a | 151 | static inline struct tempfile *mks_tempfile_s(const char *filename_template, |
076aa2cb | 152 | int suffixlen) |
354ab112 | 153 | { |
ea8ace4a | 154 | return mks_tempfile_sm(filename_template, suffixlen, 0600); |
354ab112 MH |
155 | } |
156 | ||
157 | /* See "mks_tempfile functions" above. */ | |
ea8ace4a | 158 | static inline struct tempfile *mks_tempfile_m(const char *filename_template, int mode) |
354ab112 | 159 | { |
ea8ace4a | 160 | return mks_tempfile_sm(filename_template, 0, mode); |
354ab112 MH |
161 | } |
162 | ||
163 | /* See "mks_tempfile functions" above. */ | |
ea8ace4a | 164 | static inline struct tempfile *mks_tempfile(const char *filename_template) |
354ab112 | 165 | { |
ea8ace4a | 166 | return mks_tempfile_sm(filename_template, 0, 0600); |
354ab112 MH |
167 | } |
168 | ||
169 | /* See "mks_tempfile functions" above. */ | |
55454427 | 170 | struct tempfile *mks_tempfile_tsm(const char *filename_template, |
ad6dad09 | 171 | int suffixlen, int mode); |
354ab112 MH |
172 | |
173 | /* See "mks_tempfile functions" above. */ | |
ea8ace4a | 174 | static inline struct tempfile *mks_tempfile_ts(const char *filename_template, |
076aa2cb | 175 | int suffixlen) |
354ab112 | 176 | { |
ea8ace4a | 177 | return mks_tempfile_tsm(filename_template, suffixlen, 0600); |
354ab112 MH |
178 | } |
179 | ||
180 | /* See "mks_tempfile functions" above. */ | |
ea8ace4a | 181 | static inline struct tempfile *mks_tempfile_tm(const char *filename_template, int mode) |
354ab112 | 182 | { |
ea8ace4a | 183 | return mks_tempfile_tsm(filename_template, 0, mode); |
354ab112 MH |
184 | } |
185 | ||
186 | /* See "mks_tempfile functions" above. */ | |
ea8ace4a | 187 | static inline struct tempfile *mks_tempfile_t(const char *filename_template) |
354ab112 | 188 | { |
ea8ace4a | 189 | return mks_tempfile_tsm(filename_template, 0, 0600); |
354ab112 MH |
190 | } |
191 | ||
192 | /* See "mks_tempfile functions" above. */ | |
55454427 | 193 | struct tempfile *xmks_tempfile_m(const char *filename_template, int mode); |
354ab112 MH |
194 | |
195 | /* See "mks_tempfile functions" above. */ | |
ea8ace4a | 196 | static inline struct tempfile *xmks_tempfile(const char *filename_template) |
354ab112 | 197 | { |
ea8ace4a | 198 | return xmks_tempfile_m(filename_template, 0600); |
354ab112 MH |
199 | } |
200 | ||
1a9d15db MH |
201 | /* |
202 | * Associate a stdio stream with the temporary file (which must still | |
203 | * be open). Return `NULL` (*without* deleting the file) on error. The | |
49bd0fc2 | 204 | * stream is closed automatically when `close_tempfile_gently()` is called or |
1a9d15db MH |
205 | * when the file is deleted or renamed. |
206 | */ | |
55454427 | 207 | FILE *fdopen_tempfile(struct tempfile *tempfile, const char *mode); |
1a9d15db MH |
208 | |
209 | static inline int is_tempfile_active(struct tempfile *tempfile) | |
210 | { | |
f5b4dc76 | 211 | return tempfile && tempfile->active; |
1a9d15db MH |
212 | } |
213 | ||
214 | /* | |
215 | * Return the path of the lockfile. The return value is a pointer to a | |
216 | * field within the lock_file object and should not be freed. | |
217 | */ | |
55454427 | 218 | const char *get_tempfile_path(struct tempfile *tempfile); |
1a9d15db | 219 | |
55454427 DL |
220 | int get_tempfile_fd(struct tempfile *tempfile); |
221 | FILE *get_tempfile_fp(struct tempfile *tempfile); | |
1a9d15db MH |
222 | |
223 | /* | |
224 | * If the temporary file is still open, close it (and the file pointer | |
225 | * too, if it has been opened using `fdopen_tempfile()`) without | |
226 | * deleting the file. Return 0 upon success. On failure to `close(2)`, | |
49bd0fc2 JK |
227 | * return a negative value. Usually `delete_tempfile()` or `rename_tempfile()` |
228 | * should eventually be called regardless of whether `close_tempfile_gently()` | |
229 | * succeeds. | |
1a9d15db | 230 | */ |
55454427 | 231 | int close_tempfile_gently(struct tempfile *tempfile); |
1a9d15db MH |
232 | |
233 | /* | |
234 | * Re-open a temporary file that has been closed using | |
49bd0fc2 | 235 | * `close_tempfile_gently()` but not yet deleted or renamed. This can be used |
1a9d15db MH |
236 | * to implement a sequence of operations like the following: |
237 | * | |
238 | * * Create temporary file. | |
239 | * | |
49bd0fc2 | 240 | * * Write new contents to file, then `close_tempfile_gently()` to cause the |
1a9d15db MH |
241 | * contents to be written to disk. |
242 | * | |
243 | * * Pass the name of the temporary file to another program to allow | |
244 | * it (and nobody else) to inspect or even modify the file's | |
245 | * contents. | |
246 | * | |
6c003d6f JK |
247 | * * `reopen_tempfile()` to reopen the temporary file, truncating the existing |
248 | * contents. Write out the new contents. | |
1a9d15db MH |
249 | * |
250 | * * `rename_tempfile()` to move the file to its permanent location. | |
251 | */ | |
55454427 | 252 | int reopen_tempfile(struct tempfile *tempfile); |
1a9d15db MH |
253 | |
254 | /* | |
255 | * Close the file descriptor and/or file pointer and remove the | |
256 | * temporary file associated with `tempfile`. It is a NOOP to call | |
257 | * `delete_tempfile()` for a `tempfile` object that has already been | |
258 | * deleted or renamed. | |
259 | */ | |
55454427 | 260 | void delete_tempfile(struct tempfile **tempfile_p); |
1a9d15db MH |
261 | |
262 | /* | |
263 | * Close the file descriptor and/or file pointer if they are still | |
264 | * open, and atomically rename the temporary file to `path`. `path` | |
265 | * must be on the same filesystem as the lock file. Return 0 on | |
266 | * success. On failure, delete the temporary file and return -1, with | |
267 | * `errno` set to the value from the failing call to `close(2)` or | |
268 | * `rename(2)`. It is a bug to call `rename_tempfile()` for a | |
269 | * `tempfile` object that is not currently active. | |
270 | */ | |
55454427 | 271 | int rename_tempfile(struct tempfile **tempfile_p, const char *path); |
1a9d15db MH |
272 | |
273 | #endif /* TEMPFILE_H */ |