tempname.c 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249
  1. /* Copyright (C) 1991,92,93,94,95,96,97,98,99 Free Software Foundation, Inc.
  2. This file is part of the GNU C Library.
  3. The GNU C Library is free software; you can redistribute it and/or
  4. modify it under the terms of the GNU Library General Public License as
  5. published by the Free Software Foundation; either version 2 of the
  6. License, or (at your option) any later version.
  7. The GNU C Library is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  10. Library General Public License for more details.
  11. You should have received a copy of the GNU Library General Public
  12. License along with the GNU C Library; see the file COPYING.LIB. If not,
  13. write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  14. Boston, MA 02111-1307, USA. */
  15. /* March 11, 2002 Manuel Novoa III
  16. *
  17. * Modify code to remove dependency on libgcc long long arith support funcs.
  18. */
  19. /* June 6, 2004 Erik Andersen
  20. *
  21. * Don't use brain damaged getpid() based randomness.
  22. */
  23. /* April 15, 2005 Mike Frysinger
  24. *
  25. * Use brain damaged getpid() if real random fails.
  26. */
  27. #include <stddef.h>
  28. #include <stdint.h>
  29. #include <stdio.h>
  30. #include <stdlib.h>
  31. #include <string.h>
  32. #include <errno.h>
  33. #include <fcntl.h>
  34. #include <unistd.h>
  35. #include <assert.h>
  36. #include <sys/types.h>
  37. #include <sys/stat.h>
  38. #include <sys/time.h>
  39. #include "tempname.h"
  40. /* Return nonzero if DIR is an existent directory. */
  41. static int direxists (const char *dir)
  42. {
  43. struct stat buf;
  44. return stat(dir, &buf) == 0 && S_ISDIR (buf.st_mode);
  45. }
  46. /* Path search algorithm, for tmpnam, tmpfile, etc. If DIR is
  47. non-null and exists, uses it; otherwise uses the first of $TMPDIR,
  48. P_tmpdir, /tmp that exists. Copies into TMPL a template suitable
  49. for use with mk[s]temp. Will fail (-1) if DIR is non-null and
  50. doesn't exist, none of the searched dirs exists, or there's not
  51. enough space in TMPL. */
  52. int attribute_hidden ___path_search (char *tmpl, size_t tmpl_len, const char *dir,
  53. const char *pfx /*, int try_tmpdir*/)
  54. {
  55. /*const char *d; */
  56. size_t dlen, plen;
  57. if (!pfx || !pfx[0])
  58. {
  59. pfx = "file";
  60. plen = 4;
  61. }
  62. else
  63. {
  64. plen = strlen (pfx);
  65. if (plen > 5)
  66. plen = 5;
  67. }
  68. /* Disable support for $TMPDIR */
  69. #if 0
  70. if (try_tmpdir)
  71. {
  72. d = __secure_getenv ("TMPDIR");
  73. if (d != NULL && direxists (d))
  74. dir = d;
  75. else if (dir != NULL && direxists (dir))
  76. /* nothing */ ;
  77. else
  78. dir = NULL;
  79. }
  80. #endif
  81. if (dir == NULL)
  82. {
  83. if (direxists (P_tmpdir))
  84. dir = P_tmpdir;
  85. else if (strcmp (P_tmpdir, "/tmp") != 0 && direxists ("/tmp"))
  86. dir = "/tmp";
  87. else
  88. {
  89. __set_errno (ENOENT);
  90. return -1;
  91. }
  92. }
  93. dlen = strlen (dir);
  94. while (dlen > 1 && dir[dlen - 1] == '/')
  95. dlen--; /* remove trailing slashes */
  96. /* check we have room for "${dir}/${pfx}XXXXXX\0" */
  97. if (tmpl_len < dlen + 1 + plen + 6 + 1)
  98. {
  99. __set_errno (EINVAL);
  100. return -1;
  101. }
  102. sprintf (tmpl, "%.*s/%.*sXXXXXX", dlen, dir, plen, pfx);
  103. return 0;
  104. }
  105. /* These are the characters used in temporary filenames. */
  106. static const char letters[] =
  107. "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
  108. #define NUM_LETTERS (62)
  109. static unsigned int fillrand(unsigned char *buf, unsigned int len)
  110. {
  111. int fd;
  112. unsigned int result = -1;
  113. fd = open("/dev/urandom", O_RDONLY);
  114. if (fd < 0) {
  115. fd = open("/dev/random", O_RDONLY | O_NONBLOCK);
  116. }
  117. if (fd >= 0) {
  118. result = read(fd, buf, len);
  119. close(fd);
  120. }
  121. return result;
  122. }
  123. static void brain_damaged_fillrand(unsigned char *buf, unsigned int len)
  124. {
  125. unsigned int i, k;
  126. struct timeval tv;
  127. uint32_t high, low, rh;
  128. static uint64_t value;
  129. gettimeofday(&tv, NULL);
  130. value += ((uint64_t) tv.tv_usec << 16) ^ tv.tv_sec ^ getpid();
  131. low = value & UINT32_MAX;
  132. high = value >> 32;
  133. for (i = 0; i < len; ++i) {
  134. rh = high % NUM_LETTERS;
  135. high /= NUM_LETTERS;
  136. #define L ((UINT32_MAX % NUM_LETTERS + 1) % NUM_LETTERS)
  137. k = (low % NUM_LETTERS) + (L * rh);
  138. #undef L
  139. #define H ((UINT32_MAX / NUM_LETTERS) + ((UINT32_MAX % NUM_LETTERS + 1) / NUM_LETTERS))
  140. low = (low / NUM_LETTERS) + (H * rh) + (k / NUM_LETTERS);
  141. #undef H
  142. k %= NUM_LETTERS;
  143. buf[i] = letters[k];
  144. }
  145. }
  146. /* Generate a temporary file name based on TMPL. TMPL must match the
  147. rules for mk[s]temp (i.e. end in "XXXXXX"). The name constructed
  148. does not exist at the time of the call to __gen_tempname. TMPL is
  149. overwritten with the result.
  150. KIND may be one of:
  151. __GT_NOCREATE: simply verify that the name does not exist
  152. at the time of the call. mode argument is ignored.
  153. __GT_FILE: create the file using open(O_CREAT|O_EXCL)
  154. and return a read-write fd with given mode.
  155. __GT_BIGFILE: same as __GT_FILE but use open64().
  156. __GT_DIR: create a directory with given mode.
  157. */
  158. int __gen_tempname (char *tmpl, int kind, mode_t mode)
  159. {
  160. char *XXXXXX;
  161. unsigned int i;
  162. int fd, save_errno = errno;
  163. unsigned char randomness[6];
  164. size_t len;
  165. len = strlen (tmpl);
  166. /* This is where the Xs start. */
  167. XXXXXX = tmpl + len - 6;
  168. if (len < 6 || strcmp (XXXXXX, "XXXXXX"))
  169. {
  170. __set_errno (EINVAL);
  171. return -1;
  172. }
  173. for (i = 0; i < TMP_MAX; ++i) {
  174. int j;
  175. /* Get some random data. */
  176. if (fillrand(randomness, sizeof(randomness)) != sizeof(randomness)) {
  177. /* if random device nodes failed us, lets use the braindamaged ver */
  178. brain_damaged_fillrand(randomness, sizeof(randomness));
  179. }
  180. for (j = 0; j < sizeof(randomness); ++j)
  181. XXXXXX[j] = letters[randomness[j] % NUM_LETTERS];
  182. switch (kind) {
  183. case __GT_NOCREATE:
  184. {
  185. struct stat st;
  186. if (stat (tmpl, &st) < 0) {
  187. if (errno == ENOENT) {
  188. fd = 0;
  189. goto restore_and_ret;
  190. } else
  191. /* Give up now. */
  192. return -1;
  193. } else
  194. fd = 0;
  195. }
  196. case __GT_FILE:
  197. fd = open (tmpl, O_RDWR | O_CREAT | O_EXCL, mode);
  198. break;
  199. #if defined __UCLIBC_HAS_LFS__
  200. case __GT_BIGFILE:
  201. fd = open64 (tmpl, O_RDWR | O_CREAT | O_EXCL, mode);
  202. break;
  203. #endif
  204. case __GT_DIR:
  205. fd = mkdir (tmpl, mode);
  206. break;
  207. default:
  208. fd = -1;
  209. assert (! "invalid KIND in __gen_tempname");
  210. }
  211. if (fd >= 0) {
  212. restore_and_ret:
  213. __set_errno (save_errno);
  214. return fd;
  215. }
  216. else if (errno != EEXIST)
  217. /* Any other error will apply also to other names we might
  218. try, and there are 2^32 or so of them, so give up now. */
  219. return -1;
  220. }
  221. /* We got out of the loop because we ran out of combinations to try. */
  222. __set_errno (EEXIST);
  223. return -1;
  224. }