1a113757aafb418ac4391c2f1da59fb32f1d8275
[kopensolaris-gnu/glibc.git] / sysdeps / posix / tempname.c
1 /* Copyright (C) 1991,92,93,94,95,96,97,98 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3
4    The GNU C Library is free software; you can redistribute it and/or
5    modify it under the terms of the GNU Library General Public License as
6    published by the Free Software Foundation; either version 2 of the
7    License, or (at your option) any later version.
8
9    The GNU C Library is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    Library General Public License for more details.
13
14    You should have received a copy of the GNU Library General Public
15    License along with the GNU C Library; see the file COPYING.LIB.  If not,
16    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17    Boston, MA 02111-1307, USA.  */
18
19 #include <stddef.h>
20 #include <stdint.h>
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <errno.h>
25 #include <sys/types.h>
26 #include <sys/stat.h>
27 #include <fcntl.h>
28 #include <unistd.h>
29 #include <sys/time.h>
30
31 /* Return nonzero if DIR is an existent directory.  */
32 static int
33 direxists (const char *dir)
34 {
35   struct stat buf;
36   return __xstat (_STAT_VER, dir, &buf) == 0 && S_ISDIR (buf.st_mode);
37 }
38
39 /* Path search algorithm, for tmpnam, tmpfile, etc.  If DIR is
40    non-null and exists, uses it; otherwise uses the first of $TMPDIR,
41    P_tmpdir, /tmp that exists.  Copies into TMPL a template suitable
42    for use with mk[s]temp.  Will fail (-1) if DIR is non-null and
43    doesn't exist, none of the searched dirs exists, or there's not
44    enough space in TMPL. */
45 int
46 __path_search (char *tmpl, size_t tmpl_len, const char *dir, const char *pfx,
47                int try_tmpdir)
48 {
49   const char *d;
50   size_t dlen, plen;
51
52   if (!pfx || !pfx[0])
53     {
54       pfx = "file";
55       plen = 4;
56     }
57   else
58     {
59       plen = strlen (pfx);
60       if (plen > 5)
61         plen = 5;
62     }
63
64   if (try_tmpdir)
65     {
66       d = __secure_getenv ("TMPDIR");
67       if (d != NULL && direxists (d))
68         dir = d;
69       else if (dir != NULL && direxists (dir))
70         /* nothing */ ;
71     }
72   if (direxists (P_tmpdir))
73     dir = P_tmpdir;
74   else if (strcmp (P_tmpdir, "/tmp") != 0 && direxists ("/tmp"))
75     dir = "/tmp";
76   else
77     {
78       __set_errno (ENOENT);
79       return -1;
80     }
81
82   dlen = strlen (dir);
83   while (dlen > 1 && dir[dlen - 1] == '/')
84     dlen--;                     /* remove trailing slashes */
85
86   /* check we have room for "${dir}/${pfx}XXXXXX\0" */
87   if (tmpl_len < dlen + 1 + plen + 6 + 1)
88     {
89       __set_errno (EINVAL);
90       return -1;
91     }
92
93   sprintf (tmpl, "%*s/%*sXXXXXX", dlen, dir, plen, pfx);
94   return 0;
95 }
96
97 /* These are the characters used in temporary filenames.  */
98 static const char letters[] =
99 "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
100
101 /* Generate a temporary file name based on TMPL.  TMPL must match the
102    rules for mk[s]temp (i.e. end in "XXXXXX").  The name constructed
103    does not exist at the time of the call to __gen_tempname.  TMPL is
104    overwritten with the result.  If OPENIT is nonzero, creates the
105    file and returns a read-write fd; the file is mode 0600 modulo
106    umask.  If LARGEFILE is nonzero, uses open64() instead of open().
107
108    We use a clever algorithm to get hard-to-predict names. */
109 int
110 __gen_tempname (char *tmpl, int openit, int largefile)
111 {
112   int len;
113   char *XXXXXX;
114   static uint64_t value;
115   struct timeval tv;
116   int count, fd;
117   int save_errno = errno;
118
119   len = strlen (tmpl);
120   if (len < 6 || strcmp (&tmpl[len - 6], "XXXXXX"))
121     {
122       __set_errno (EINVAL);
123       return -1;
124     }
125
126   /* This is where the Xs start.  */
127   XXXXXX = &tmpl[len - 6];
128
129   /* Get some more or less random data.  */
130   __gettimeofday (&tv, NULL);
131   value += ((uint64_t) tv.tv_usec << 16) ^ tv.tv_sec ^ __getpid ();
132
133   for (count = 0; count < TMP_MAX; value += 7777, ++count)
134     {
135       uint64_t v = value;
136
137       /* Fill in the random bits.  */
138       XXXXXX[0] = letters[v % 62];
139       v /= 62;
140       XXXXXX[1] = letters[v % 62];
141       v /= 62;
142       XXXXXX[2] = letters[v % 62];
143       v /= 62;
144       XXXXXX[3] = letters[v % 62];
145       v /= 62;
146       XXXXXX[4] = letters[v % 62];
147       v /= 62;
148       XXXXXX[5] = letters[v % 62];
149
150       if (openit)
151         {
152           fd = (largefile
153                 ? __open (tmpl, O_RDWR | O_CREAT | O_EXCL, 0666)
154                 : __open64 (tmpl, O_RDWR | O_CREAT | O_EXCL, 0666));
155           if (fd >= 0)
156             {
157               __set_errno (save_errno);
158               return fd;
159             }
160           else if (errno != EEXIST)
161             /* Any other error will apply also to other names we might
162                try, and there are 2^32 or so of them, so give up now. */
163             return -1;
164         }
165       else
166         {
167           struct stat st;
168           if (__xstat (_STAT_VER, tmpl, &st) < 0)
169             {
170               if (errno == ENOENT)
171                 {
172                   __set_errno (save_errno);
173                   return 0;
174                 }
175               else
176                 /* Give up now. */
177                 return -1;
178             }
179         }
180     }
181
182   /* We got out of the loop because we ran out of combinations to try.  */
183   __set_errno (EEXIST);
184   return -1;
185 }