]> andersk Git - openssh.git/blame - bsd-mktemp.c
- (djm) NeXT tweaks from Ben Lindstrom <mouring@pconline.com>
[openssh.git] / bsd-mktemp.c
CommitLineData
8efc0c15 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
beb43d31 37#include "config.h"
38
39#ifndef HAVE_MKDTEMP
40
8efc0c15 41#if defined(LIBC_SCCS) && !defined(lint)
42static 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
5ab44a92 54#include "bsd-misc.h"
c345cf9d 55#include "bsd-arc4random.h"
0183ea1c 56
4cca272e 57static int _gettemp(char *, int *, int, int);
8efc0c15 58
59int
60mkstemps(path, slen)
61 char *path;
62 int slen;
63{
64 int fd;
65
66 return (_gettemp(path, &fd, 0, slen) ? fd : -1);
67}
68
69int
70mkstemp(path)
71 char *path;
72{
73 int fd;
74
75 return (_gettemp(path, &fd, 0, 0) ? fd : -1);
76}
77
78char *
79mkdtemp(path)
80 char *path;
81{
82 return(_gettemp(path, (int *)NULL, 1, 0) ? path : (char *)NULL);
83}
84
85static 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}
fb723d82 189
190#endif /* !HAVE_MKDTEMP */
This page took 0.102272 seconds and 5 git commands to generate.