1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
| | #include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>
#include <limits.h>
#include <errno.h>
#include "libc.h"
char *__randname(char *);
int __open_tempfile (char *template, int len, int flags)
{
if (len < 0) return EINVAL;
int l = strlen(template)-len;
if (l < 6 || strncmp(template+l-6, "XXXXXX",6)) {
errno = EINVAL;
*template = 0;
return -1;
}
/* Null terminate the template before the suffix,
and save the char for adding back the suffix */
char suffix = template[l];
template[l] = '\0';
int fd, retries = 100, t0 = *template;
while (retries--) {
if (!*__randname(template)) return -1;
/* Add back the suffix */
template[l] = suffix;
if ((fd = open(template, flags | O_CREAT | O_EXCL, 0600))>=0)
return fd;
if (errno != EEXIST) return -1;
/* this is safe because mktemp verified
* that we have a valid template string */
template[0] = t0;
strcpy(template+l-6, "XXXXXX");
}
return -1;
}
|