]> andersk Git - openssh.git/blob - bsd-mktemp.c
- (djm) OpenBSD CVS changes:
[openssh.git] / bsd-mktemp.c
1 /* THIS FILE HAS BEEN MODIFIED FROM THE ORIGINAL OPENBSD SOURCE */
2 /* Changes: Removed mktemp */
3
4 /*
5  * Copyright (c) 1987, 1993
6  *      The Regents of the University of California.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. All advertising materials mentioning features or use of this software
17  *    must display the following acknowledgement:
18  *      This product includes software developed by the University of
19  *      California, Berkeley and its contributors.
20  * 4. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  */
36
37 #include "config.h"
38
39 #ifndef HAVE_MKDTEMP
40
41 #if defined(LIBC_SCCS) && !defined(lint)
42 static char rcsid[] = "$OpenBSD: mktemp.c,v 1.13 1998/06/30 23:03:13 deraadt Exp $";
43 #endif /* LIBC_SCCS and not lint */
44
45 #include <sys/types.h>
46 #include <sys/stat.h>
47 #include <fcntl.h>
48 #include <errno.h>
49 #include <stdio.h>
50 #include <stdlib.h>
51 #include <ctype.h>
52 #include <unistd.h>
53
54 #include "bsd-misc.h"
55 #include "bsd-arc4random.h"
56
57 static int _gettemp(char *, int *, int, int);
58
59 int
60 mkstemps(path, slen)
61         char *path;
62         int slen;
63 {
64         int fd;
65
66         return (_gettemp(path, &fd, 0, slen) ? fd : -1);
67 }
68
69 int
70 mkstemp(path)
71         char *path;
72 {
73         int fd;
74
75         return (_gettemp(path, &fd, 0, 0) ? fd : -1);
76 }
77
78 char *
79 mkdtemp(path)
80         char *path;
81 {
82         return(_gettemp(path, (int *)NULL, 1, 0) ? path : (char *)NULL);
83 }
84
85 static int
86 _gettemp(path, doopen, domkdir, slen)
87         char *path;
88         register int *doopen;
89         int domkdir;
90         int slen;
91 {
92         register char *start, *trv, *suffp;
93         struct stat sbuf;
94         int pid, rval;
95
96         if (doopen && domkdir) {
97                 errno = EINVAL;
98                 return(0);
99         }
100
101         for (trv = path; *trv; ++trv)
102                 ;
103         trv -= slen;
104         suffp = trv;
105         --trv;
106         if (trv < path) {
107                 errno = EINVAL;
108                 return (0);
109         }
110         pid = getpid();
111         while (*trv == 'X' && pid != 0) {
112                 *trv-- = (pid % 10) + '0';
113                 pid /= 10;
114         }
115         while (*trv == 'X') {
116                 char c;
117
118                 pid = (arc4random() & 0xffff) % (26+26);
119                 if (pid < 26)
120                         c = pid + 'A';
121                 else
122                         c = (pid - 26) + 'a';
123                 *trv-- = c;
124         }
125         start = trv + 1;
126
127         /*
128          * check the target directory; if you have six X's and it
129          * doesn't exist this runs for a *very* long time.
130          */
131         if (doopen || domkdir) {
132                 for (;; --trv) {
133                         if (trv <= path)
134                                 break;
135                         if (*trv == '/') {
136                                 *trv = '\0';
137                                 rval = stat(path, &sbuf);
138                                 *trv = '/';
139                                 if (rval != 0)
140                                         return(0);
141                                 if (!S_ISDIR(sbuf.st_mode)) {
142                                         errno = ENOTDIR;
143                                         return(0);
144                                 }
145                                 break;
146                         }
147                 }
148         }
149
150         for (;;) {
151                 if (doopen) {
152                         if ((*doopen =
153                             open(path, O_CREAT|O_EXCL|O_RDWR, 0600)) >= 0)
154                                 return(1);
155                         if (errno != EEXIST)
156                                 return(0);
157                 } else if (domkdir) {
158                         if (mkdir(path, 0700) == 0)
159                                 return(1);
160                         if (errno != EEXIST)
161                                 return(0);
162                 } else if (lstat(path, &sbuf))
163                         return(errno == ENOENT ? 1 : 0);
164
165                 /* tricky little algorithm for backward compatibility */
166                 for (trv = start;;) {
167                         if (!*trv)
168                                 return (0);
169                         if (*trv == 'Z') {
170                                 if (trv == suffp)
171                                         return (0);
172                                 *trv++ = 'a';
173                         } else {
174                                 if (isdigit(*trv))
175                                         *trv = 'a';
176                                 else if (*trv == 'z')   /* inc from z to A */
177                                         *trv = 'A';
178                                 else {
179                                         if (trv == suffp)
180                                                 return (0);
181                                         ++*trv;
182                                 }
183                                 break;
184                         }
185                 }
186         }
187         /*NOTREACHED*/
188 }
189
190 #endif /* !HAVE_MKDTEMP */
This page took 0.048235 seconds and 5 git commands to generate.