1 /* Licensed to the Apache Software Foundation (ASF) under one or more
2 * contributor license agreements. See the NOTICE file distributed with
3 * this work for additional information regarding copyright ownership.
4 * The ASF licenses this file to You under the Apache License, Version 2.0
5 * (the "License"); you may not use this file except in compliance with
6 * the License. You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16 /*
17 * Copyright (c) 1987, 1993
18 * The Regents of the University of California. All rights reserved.
19 *
20 * Redistribution and use in source and binary forms, with or without
21 * modification, are permitted provided that the following conditions
22 * are met:
23 * 1. Redistributions of source code must retain the above copyright
24 * notice, this list of conditions and the following disclaimer.
25 * 2. Redistributions in binary form must reproduce the above copyright
26 * notice, this list of conditions and the following disclaimer in the
27 * documentation and/or other materials provided with the distribution.
28 * 3. All advertising materials mentioning features or use of this software
29 * must display the following acknowledgement:
30 * This product includes software developed by the University of
31 * California, Berkeley and its contributors.
32 * 4. Neither the name of the University nor the names of its contributors
33 * may be used to endorse or promote products derived from this software
34 * without specific prior written permission.
35 *
36 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
37 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
38 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
39 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
40 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
41 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
42 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
43 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
44 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
45 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
46 * SUCH DAMAGE.
47 */
48
49 #include "apr_private.h"
50 #include "apr_file_io.h" /* prototype of apr_mkstemp() */
51 #include "apr_strings.h" /* prototype of apr_mkstemp() */
52 #include "apr_arch_file_io.h" /* prototype of apr_mkstemp() */
53 #include "apr_portable.h" /* for apr_os_file_put() */
54 #include "apr_arch_inherit.h"
55
56 #ifndef HAVE_MKSTEMP
57
58 #if defined(SVR4) || defined(WIN32) || defined(NETWARE)
59 #ifdef SVR4
60 #if HAVE_INTTYPES_H
61 #include <inttypes.h>
62 #endif
63 #endif
64 #define arc4random() rand()
65 #define seedrandom(a) srand(a)
66 #else
67 #if APR_HAVE_STDINT_H
68 #include <stdint.h>
69 #endif
70 #define arc4random() random()
71 #define seedrandom(a) srandom(a)
72 #endif
73
74 #if APR_HAVE_SYS_TYPES_H
75 #include <sys/types.h>
76 #endif
77 #ifdef HAVE_SYS_STAT_H
78 #include <sys/stat.h>
79 #endif
80 #if APR_HAVE_FCNTL_H
81 #include <fcntl.h>
82 #endif
83 #include <stdio.h>
84 #include <stdlib.h>
85 #include <string.h>
86 #include <ctype.h>
87 #ifdef HAVE_TIME_H
88 #include <time.h>
89 #endif
90
91 static const unsigned char padchar[] =
92 "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
93 static apr_uint32_t randseed=0;
94
gettemp(char * path,apr_file_t ** doopen,apr_int32_t flags,apr_pool_t * p)95 static int gettemp(char *path, apr_file_t **doopen, apr_int32_t flags, apr_pool_t *p)
96 {
97 register char *start, *trv, *suffp;
98 char *pad;
99 apr_finfo_t sbuf;
100 apr_status_t rv;
101 apr_uint32_t randnum;
102
103 if (randseed==0) {
104 randseed = (int)apr_time_now();
105 seedrandom(randseed);
106 }
107
108 for (trv = path; *trv; ++trv)
109 ;
110 suffp = trv;
111 --trv;
112 if (trv < path) {
113 return APR_EINVAL;
114 }
115
116 /* Fill space with random characters */
117 while (*trv == 'X') {
118 randnum = arc4random() % (sizeof(padchar) - 1);
119 *trv-- = padchar[randnum];
120 }
121 start = trv + 1;
122
123 /*
124 * check the target directory.
125 */
126 for (;; --trv) {
127 if (trv <= path)
128 break;
129 if (*trv == '/') {
130 *trv = '\0';
131 rv = apr_stat(&sbuf, path, APR_FINFO_TYPE, p);
132 *trv = '/';
133 if (rv != APR_SUCCESS)
134 return rv;
135 if (sbuf.filetype != APR_DIR) {
136 return APR_ENOTDIR;
137 }
138 break;
139 }
140 }
141
142 for (;;) {
143 if ((rv = apr_file_open(doopen, path, flags,
144 APR_UREAD | APR_UWRITE, p)) == APR_SUCCESS)
145 return APR_SUCCESS;
146 if (!APR_STATUS_IS_EEXIST(rv))
147 return rv;
148
149 /* If we have a collision, cycle through the space of filenames */
150 for (trv = start;;) {
151 if (*trv == '\0' || trv == suffp)
152 return APR_EINVAL; /* XXX: is this the correct return code? */
153 pad = strchr((char *)padchar, *trv);
154 if (pad == NULL || !*++pad) {
155 *trv++ = padchar[0];
156 }
157 else {
158 *trv++ = *pad;
159 break;
160 }
161 }
162 }
163 /*NOTREACHED*/
164 }
165
166 #else
167
168 #if APR_HAVE_STDLIB_H
169 #include <stdlib.h> /* for mkstemp() - Single Unix */
170 #endif
171 #if APR_HAVE_UNISTD_H
172 #include <unistd.h> /* for mkstemp() - FreeBSD */
173 #endif
174 #endif /* !defined(HAVE_MKSTEMP) */
175
apr_file_mktemp(apr_file_t ** fp,char * template,apr_int32_t flags,apr_pool_t * p)176 APR_DECLARE(apr_status_t) apr_file_mktemp(apr_file_t **fp, char *template, apr_int32_t flags, apr_pool_t *p)
177 {
178 #ifdef HAVE_MKSTEMP
179 int fd;
180 #endif
181 flags = (!flags) ? APR_FOPEN_CREATE | APR_FOPEN_READ | APR_FOPEN_WRITE | APR_FOPEN_EXCL |
182 APR_FOPEN_DELONCLOSE : flags;
183 #ifndef HAVE_MKSTEMP
184 return gettemp(template, fp, flags, p);
185 #else
186
187 #ifdef HAVE_MKSTEMP64
188 fd = mkstemp64(template);
189 #else
190 fd = mkstemp(template);
191 #endif
192
193 if (fd == -1) {
194 return errno;
195 }
196 /* XXX: We must reset several flags values as passed-in, since
197 * mkstemp didn't subscribe to our preference flags.
198 *
199 * We either have to unset the flags, or fix up the fd and other
200 * xthread and inherit bits appropriately. Since gettemp() above
201 * calls apr_file_open, our flags are respected in that code path.
202 */
203 apr_os_file_put(fp, &fd, flags, p);
204 (*fp)->fname = apr_pstrdup(p, template);
205
206 if (!(flags & APR_FOPEN_NOCLEANUP)) {
207 int flags;
208
209 if ((flags = fcntl(fd, F_GETFD)) == -1)
210 return errno;
211
212 flags |= FD_CLOEXEC;
213 if (fcntl(fd, F_SETFD, flags) == -1)
214 return errno;
215
216 apr_pool_cleanup_register((*fp)->pool, (void *)(*fp),
217 apr_unix_file_cleanup,
218 apr_unix_child_file_cleanup);
219 }
220 #endif
221 return APR_SUCCESS;
222 }
223
224