posix: Sync tempname with gnulib [BZ #26648]

It syncs with gnulib commit b1268f22f443e8e4b9e.  The try_tempname_len
now uses getrandom on each iteration to get entropy and only uses the
clock plus ASLR as source of entropy if getrandom fails.

Checked on x86_64-linux-gnu and i686-linux-gnu.
This commit is contained in:
Adhemerval Zanella 2020-09-29 08:56:07 -03:00
parent 42dda89dcb
commit 4dddd7e9cb
1 changed files with 161 additions and 113 deletions

View File

@ -16,7 +16,7 @@
<https://www.gnu.org/licenses/>. */ <https://www.gnu.org/licenses/>. */
#if !_LIBC #if !_LIBC
# include <config.h> # include <libc-config.h>
# include "tempname.h" # include "tempname.h"
#endif #endif
@ -24,9 +24,6 @@
#include <assert.h> #include <assert.h>
#include <errno.h> #include <errno.h>
#ifndef __set_errno
# define __set_errno(Val) errno = (Val)
#endif
#include <stdio.h> #include <stdio.h>
#ifndef P_tmpdir #ifndef P_tmpdir
@ -50,11 +47,11 @@
#include <string.h> #include <string.h>
#include <fcntl.h> #include <fcntl.h>
#include <time.h> #include <stdalign.h>
#include <stdint.h> #include <stdint.h>
#include <unistd.h> #include <sys/random.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <time.h>
#if _LIBC #if _LIBC
# define struct_stat64 struct stat64 # define struct_stat64 struct stat64
@ -62,32 +59,38 @@
#else #else
# define struct_stat64 struct stat # define struct_stat64 struct stat
# define __gen_tempname gen_tempname # define __gen_tempname gen_tempname
# define __getpid getpid
# define __mkdir mkdir # define __mkdir mkdir
# define __open open # define __open open
# define __secure_getenv secure_getenv # define __lstat64(file, buf) lstat (file, buf)
# define __stat64(file, buf) stat (file, buf)
# define __getrandom getrandom
# define __clock_gettime64 clock_gettime
# define __timespec64 timespec
#endif #endif
#ifdef _LIBC /* Use getrandom if it works, falling back on a 64-bit linear
# include <random-bits.h> congruential generator that starts with Var's value
# define RANDOM_BITS(Var) ((Var) = random_bits ()) mixed in with a clock's low-order bits if available. */
# else typedef uint_fast64_t random_value;
# define RANDOM_BITS(Var) \ #define RANDOM_VALUE_MAX UINT_FAST64_MAX
{ \ #define BASE_62_DIGITS 10 /* 62**10 < UINT_FAST64_MAX */
struct timespec ts; \ #define BASE_62_POWER (62LL * 62 * 62 * 62 * 62 * 62 * 62 * 62 * 62 * 62)
clock_gettime (CLOCK_REALTIME, &ts); \
(Var) = ((uint64_t) tv.tv_nsec << 16) ^ tv.tv_sec; \
}
#endif
/* Use the widest available unsigned type if uint64_t is not static random_value
available. The algorithm below extracts a number less than 62**6 random_bits (random_value var)
(approximately 2**35.725) from uint64_t, so ancient hosts where {
uintmax_t is only 32 bits lose about 3.725 bits of randomness, random_value r;
which is better than not having mkstemp at all. */ /* Without GRND_NONBLOCK it can be blocked for minutes on some systems. */
#if !defined UINT64_MAX && !defined uint64_t if (__getrandom (&r, sizeof r, GRND_NONBLOCK) == sizeof r)
# define uint64_t uintmax_t return r;
#if _LIBC || (defined CLOCK_MONOTONIC && HAVE_CLOCK_GETTIME)
/* Add entropy if getrandom did not work. */
struct __timespec64 tv;
__clock_gettime64 (CLOCK_MONOTONIC, &tv);
var ^= tv.tv_nsec;
#endif #endif
return 2862933555777941757 * var + 3037000493;
}
#if _LIBC #if _LIBC
/* Return nonzero if DIR is an existent directory. */ /* Return nonzero if DIR is an existent directory. */
@ -162,14 +165,45 @@ __path_search (char *tmpl, size_t tmpl_len, const char *dir, const char *pfx,
} }
#endif /* _LIBC */ #endif /* _LIBC */
#if _LIBC
static int try_tempname_len (char *, int, void *, int (*) (char *, void *),
size_t);
#endif
static int
try_file (char *tmpl, void *flags)
{
int *openflags = flags;
return __open (tmpl,
(*openflags & ~O_ACCMODE)
| O_RDWR | O_CREAT | O_EXCL, S_IRUSR | S_IWUSR);
}
static int
try_dir (char *tmpl, void *flags _GL_UNUSED)
{
return __mkdir (tmpl, S_IRUSR | S_IWUSR | S_IXUSR);
}
static int
try_nocreate (char *tmpl, void *flags _GL_UNUSED)
{
struct_stat64 st;
if (__lstat64 (tmpl, &st) == 0 || errno == EOVERFLOW)
__set_errno (EEXIST);
return errno == ENOENT ? 0 : -1;
}
/* These are the characters used in temporary file names. */ /* These are the characters used in temporary file names. */
static const char letters[] = static const char letters[] =
"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"; "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
/* Generate a temporary file name based on TMPL. TMPL must match the /* Generate a temporary file name based on TMPL. TMPL must match the
rules for mk[s]temp (i.e. end in "XXXXXX", possibly with a suffix). rules for mk[s]temp (i.e., end in at least X_SUFFIX_LEN "X"s,
possibly with a suffix).
The name constructed does not exist at the time of the call to The name constructed does not exist at the time of the call to
__gen_tempname. TMPL is overwritten with the result. this function. TMPL is overwritten with the result.
KIND may be one of: KIND may be one of:
__GT_NOCREATE: simply verify that the name does not exist __GT_NOCREATE: simply verify that the name does not exist
@ -179,22 +213,43 @@ static const char letters[] =
__GT_DIR: create a directory, which will be mode 0700. __GT_DIR: create a directory, which will be mode 0700.
We use a clever algorithm to get hard-to-predict names. */ We use a clever algorithm to get hard-to-predict names. */
#ifdef _LIBC
static
#endif
int int
__gen_tempname (char *tmpl, int suffixlen, int flags, int kind) gen_tempname_len (char *tmpl, int suffixlen, int flags, int kind,
size_t x_suffix_len)
{ {
int len; static int (*const tryfunc[]) (char *, void *) =
{
[__GT_FILE] = try_file,
[__GT_DIR] = try_dir,
[__GT_NOCREATE] = try_nocreate
};
return try_tempname_len (tmpl, suffixlen, &flags, tryfunc[kind],
x_suffix_len);
}
#ifdef _LIBC
static
#endif
int
try_tempname_len (char *tmpl, int suffixlen, void *args,
int (*tryfunc) (char *, void *), size_t x_suffix_len)
{
size_t len;
char *XXXXXX; char *XXXXXX;
unsigned int count; unsigned int count;
int fd = -1; int fd = -1;
int save_errno = errno; int save_errno = errno;
struct_stat64 st;
/* A lower bound on the number of temporary files to attempt to /* A lower bound on the number of temporary files to attempt to
generate. The maximum total number of temporary file names that generate. The maximum total number of temporary file names that
can exist for a given template is 62**6. It should never be can exist for a given template is 62**6. It should never be
necessary to try all of these combinations. Instead if a reasonable necessary to try all of these combinations. Instead if a reasonable
number of names is tried (we define reasonable as 62**3) fail to number of names is tried (we define reasonable as 62**3) fail to
give the system administrator the chance to remove the problems. */ give the system administrator the chance to remove the problems.
This value requires that X_SUFFIX_LEN be at least 3. */
#define ATTEMPTS_MIN (62 * 62 * 62) #define ATTEMPTS_MIN (62 * 62 * 62)
/* The number of times to attempt to generate a temporary file. To /* The number of times to attempt to generate a temporary file. To
@ -205,72 +260,50 @@ __gen_tempname (char *tmpl, int suffixlen, int flags, int kind)
unsigned int attempts = ATTEMPTS_MIN; unsigned int attempts = ATTEMPTS_MIN;
#endif #endif
/* A random variable. The initial value is used only the for fallback path
on 'random_bits' on 'getrandom' failure. Its initial value tries to use
some entropy from the ASLR and ignore possible bits from the stack
alignment. */
random_value v = ((uintptr_t) &v) / alignof (max_align_t);
/* How many random base-62 digits can currently be extracted from V. */
int vdigits = 0;
/* Least unfair value for V. If V is less than this, V can generate
BASE_62_DIGITS digits fairly. Otherwise it might be biased. */
random_value const unfair_min
= RANDOM_VALUE_MAX - RANDOM_VALUE_MAX % BASE_62_POWER;
len = strlen (tmpl); len = strlen (tmpl);
if (len < 6 + suffixlen || memcmp (&tmpl[len - 6 - suffixlen], "XXXXXX", 6)) if (len < x_suffix_len + suffixlen
|| strspn (&tmpl[len - x_suffix_len - suffixlen], "X") < x_suffix_len)
{ {
__set_errno (EINVAL); __set_errno (EINVAL);
return -1; return -1;
} }
/* This is where the Xs start. */ /* This is where the Xs start. */
XXXXXX = &tmpl[len - 6 - suffixlen]; XXXXXX = &tmpl[len - x_suffix_len - suffixlen];
uint64_t pid = (uint64_t) __getpid () << 32;
for (count = 0; count < attempts; ++count) for (count = 0; count < attempts; ++count)
{ {
uint64_t v; for (size_t i = 0; i < x_suffix_len; i++)
/* Get some more or less random data. */
RANDOM_BITS (v);
v ^= pid;
/* Fill in the random bits. */
XXXXXX[0] = letters[v % 62];
v /= 62;
XXXXXX[1] = letters[v % 62];
v /= 62;
XXXXXX[2] = letters[v % 62];
v /= 62;
XXXXXX[3] = letters[v % 62];
v /= 62;
XXXXXX[4] = letters[v % 62];
v /= 62;
XXXXXX[5] = letters[v % 62];
switch (kind)
{ {
case __GT_FILE: if (vdigits == 0)
fd = __open (tmpl,
(flags & ~O_ACCMODE)
| O_RDWR | O_CREAT | O_EXCL, S_IRUSR | S_IWUSR);
break;
case __GT_DIR:
fd = __mkdir (tmpl, S_IRUSR | S_IWUSR | S_IXUSR);
break;
case __GT_NOCREATE:
/* This case is backward from the other three. __gen_tempname
succeeds if lstat fails because the name does not exist.
Note the continue to bypass the common logic at the bottom
of the loop. */
if (__lstat64 (tmpl, &st) < 0)
{ {
if (errno == ENOENT) do
{ v = random_bits (v);
__set_errno (save_errno); while (unfair_min <= v);
return 0;
}
else
/* Give up now. */
return -1;
}
continue;
default: vdigits = BASE_62_DIGITS;
assert (! "invalid KIND in __gen_tempname");
abort ();
} }
XXXXXX[i] = letters[v % 62];
v /= 62;
vdigits--;
}
fd = tryfunc (tmpl, args);
if (fd >= 0) if (fd >= 0)
{ {
__set_errno (save_errno); __set_errno (save_errno);
@ -284,3 +317,18 @@ __gen_tempname (char *tmpl, int suffixlen, int flags, int kind)
__set_errno (EEXIST); __set_errno (EEXIST);
return -1; return -1;
} }
int
__gen_tempname (char *tmpl, int suffixlen, int flags, int kind)
{
return gen_tempname_len (tmpl, suffixlen, flags, kind, 6);
}
#if !_LIBC
int
try_tempname (char *tmpl, int suffixlen, void *args,
int (*tryfunc) (char *, void *))
{
return try_tempname_len (tmpl, suffixlen, args, tryfunc, 6);
}
#endif