major improvements to temp file name generator

use current time in nanoseconds and some potentially-random (if aslr
is enabled) pointer values for the initial tempfile name generation,
and step via a cheap linear prng on collisions. limit the number of
retry attempts to prevent denial of service attacks even if an
attacker can guess the filenames.
This commit is contained in:
Rich Felker 2011-02-18 17:04:56 -05:00
commit 446b4207cc
3 changed files with 21 additions and 16 deletions

View file

@ -12,7 +12,8 @@ char *__mktemp(char *);
char *mkdtemp(char *template) char *mkdtemp(char *template)
{ {
for (;;) { int retries = 100;
while (retries--) {
if (!__mktemp(template)) return 0; if (!__mktemp(template)) return 0;
if (!mkdir(template, 0700)) return template; if (!mkdir(template, 0700)) return template;
if (errno != EEXIST) return 0; if (errno != EEXIST) return 0;
@ -20,4 +21,5 @@ char *mkdtemp(char *template)
* that we have a valid template string */ * that we have a valid template string */
strcpy(template+strlen(template)-6, "XXXXXX"); strcpy(template+strlen(template)-6, "XXXXXX");
} }
return 0;
} }

View file

@ -11,8 +11,8 @@ char *__mktemp(char *);
int mkstemp(char *template) int mkstemp(char *template)
{ {
int fd; int fd, retries = 100;
for (;;) { while (retries--) {
if (!__mktemp(template)) return 0; if (!__mktemp(template)) return 0;
if ((fd = open(template, O_RDWR | O_CREAT | O_EXCL, 0600))>=0) if ((fd = open(template, O_RDWR | O_CREAT | O_EXCL, 0600))>=0)
return fd; return fd;
@ -21,6 +21,7 @@ int mkstemp(char *template)
* that we have a valid template string */ * that we have a valid template string */
strcpy(template+strlen(template)-6, "XXXXXX"); strcpy(template+strlen(template)-6, "XXXXXX");
} }
return -1;
} }
LFS64(mkstemp); LFS64(mkstemp);

View file

@ -4,28 +4,30 @@
#include <fcntl.h> #include <fcntl.h>
#include <unistd.h> #include <unistd.h>
#include <errno.h> #include <errno.h>
#include <time.h>
#include <stdint.h>
#include "libc.h" #include "libc.h"
char *__mktemp(char *template) char *__mktemp(char *template)
{ {
static int lock; struct timespec ts;
static int index; size_t l = strlen(template);
int l = strlen(template); int retries = 10000;
unsigned long r;
if (l < 6 || strcmp(template+l-6, "XXXXXX")) { if (l < 6 || strcmp(template+l-6, "XXXXXX")) {
errno = EINVAL; errno = EINVAL;
return NULL; return 0;
} }
LOCK(&lock); clock_gettime(CLOCK_REALTIME, &ts);
for (; index < 1000000; index++) { r = ts.tv_nsec + (uintptr_t)&ts / 16 + (uintptr_t)template;
snprintf(template+l-6, 6, "%06d", index); while (retries--) {
if (access(template, F_OK) != 0) { snprintf(template+l-6, 7, "%06lX", r & 0xffffff);
UNLOCK(&lock); if (access(template, F_OK) < 0) return template;
return template; r = r * 1103515245 + 12345;
}
} }
UNLOCK(&lock); errno = EEXIST;
return NULL; return 0;
} }
weak_alias(__mktemp, mktemp); weak_alias(__mktemp, mktemp);