Use __xstat instead of __stat.
[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 {
48   const char *d;
49   size_t dlen, plen;
50
51   if (!pfx || !pfx[0])
52     {
53       pfx = "file";
54       plen = 4;
55     }
56   else
57     {
58       plen = strlen (pfx);
59       if (plen > 5)
60         plen = 5;
61     }
62
63   d = __secure_getenv ("TMPDIR");
64   if (d != NULL && direxists (d))
65     dir = d;
66   else if (dir != NULL && direxists (dir))
67     /* nothing */ ;
68   else if (direxists (P_tmpdir))
69     dir = P_tmpdir;
70   else if (direxists ("/tmp"))
71     dir = "/tmp";
72   else
73     {
74       __set_errno (ENOENT);
75       return -1;
76     }
77
78   dlen = strlen (dir);
79   while (dlen > 1 && dir[dlen - 1] == '/')
80     dlen--;                     /* remove trailing slashes */
81
82   /* check we have room for "${dir}/${pfx}XXXXXX\0" */
83   if (tmpl_len < dlen + 1 + plen + 6 + 1)
84     {
85       __set_errno (EINVAL);
86       return -1;
87     }
88
89   sprintf (tmpl, "%*s/%*sXXXXXX", dlen, dir, plen, pfx);
90   return 0;
91 }
92
93 /* These are the characters used in temporary filenames.  */
94 static const char letters[] =
95 "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
96
97 /* Generate a temporary file name based on TMPL.  TMPL must match the
98    rules for mk[s]temp (i.e. end in "XXXXXX").  The name constructed
99    does not exist at the time of the call to __gen_tempname.  TMPL is
100    overwritten with the result.  If OPENIT is nonzero, creates the
101    file and returns a read-write fd; the file is mode 0600 modulo
102    umask.  If LARGEFILE is nonzero, uses open64() instead of open().
103
104    We use a clever algorithm to get hard-to-predict names. */
105 int
106 __gen_tempname (char *tmpl, int openit, int largefile)
107 {
108   int len;
109   char *XXXXXX;
110   static uint64_t value;
111   struct timeval tv;
112   int count, fd;
113   int save_errno = errno;
114
115   len = strlen (tmpl);
116   if (len < 6 || strcmp (&tmpl[len - 6], "XXXXXX"))
117     {
118       __set_errno (EINVAL);
119       return -1;
120     }
121
122   /* This is where the Xs start.  */
123   XXXXXX = &tmpl[len - 6];
124
125   /* Get some more or less random data.  */
126   __gettimeofday (&tv, NULL);
127   value += ((uint64_t) tv.tv_usec << 16) ^ tv.tv_sec ^ __getpid ();
128
129   for (count = 0; count < TMP_MAX; value += 7777, ++count)
130     {
131       uint64_t v = value;
132
133       /* Fill in the random bits.  */
134       XXXXXX[0] = letters[v % 62];
135       v /= 62;
136       XXXXXX[1] = letters[v % 62];
137       v /= 62;
138       XXXXXX[2] = letters[v % 62];
139       v /= 62;
140       XXXXXX[3] = letters[v % 62];
141       v /= 62;
142       XXXXXX[4] = letters[v % 62];
143       v /= 62;
144       XXXXXX[5] = letters[v % 62];
145
146       if (openit)
147         {
148           fd = (largefile
149                 ? __open (tmpl, O_RDWR | O_CREAT | O_EXCL, 0666)
150                 : __open64 (tmpl, O_RDWR | O_CREAT | O_EXCL, 0666));
151           if (fd >= 0)
152             {
153               __set_errno (save_errno);
154               return fd;
155             }
156           else if (errno != EEXIST)
157             /* Any other error will apply also to other names we might
158                try, and there are 2^32 or so of them, so give up now. */
159             return -1;
160         }
161       else
162         {
163           struct stat st;
164           if (__xstat (_STAT_VER, tmpl, &st) < 0)
165             {
166               if (errno == ENOENT)
167                 {
168                   __set_errno (save_errno);
169                   return 0;
170                 }
171               else
172                 /* Give up now. */
173                 return -1;
174             }
175         }
176     }
177
178   /* We got out of the loop because we ran out of combinations to try.  */
179   __set_errno (EEXIST);
180   return -1;
181 }