[apple/libc.git] / stdio / mktemp.c

/*
 * Copyright (c) 1987, 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *	This product includes software developed by the University of
 *	California, Berkeley and its contributors.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#if defined(LIBC_SCCS) && !defined(lint)
#if 0
static char sccsid[] = "@(#)mktemp.c	8.1 (Berkeley) 6/4/93";
#endif
static const char rcsid[] =
  "$FreeBSD: src/lib/libc/stdio/mktemp.c,v 1.19.2.1 2001/01/20 09:35:24 kris Exp $";
#endif /* LIBC_SCCS and not lint */

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <unistd.h>
#include <stdint.h>

char *_mktemp __P((char *));

static int _gettemp __P((char *, int *, int, int));

static const unsigned char padchar[] =
"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";

int
mkstemps(path, slen)
	char *path;
	int slen;
{
	int fd;

	return (_gettemp(path, &fd, 0, slen) ? fd : -1);
}

int
mkstemp(path)
	char *path;
{
	int fd;

	return (_gettemp(path, &fd, 0, 0) ? fd : -1);
}

char *
mkdtemp(path)
	char *path;
{
	return(_gettemp(path, (int *)NULL, 1, 0) ? path : (char *)NULL);
}

char *
_mktemp(path)
	char *path;
{
	return(_gettemp(path, (int *)NULL, 0, 0) ? path : (char *)NULL);
}

char *
mktemp(path)
	char *path;
{
	return(_mktemp(path));
}

static int
_gettemp(path, doopen, domkdir, slen)
	char *path;
	register int *doopen;
	int domkdir;
	int slen;
{
	register char *start, *trv, *suffp;
	char *pad;
	struct stat sbuf;
	int rval;
	uint32_t rand;

	if (doopen && domkdir) {
		errno = EINVAL;
		return(0);
	}

	for (trv = path; *trv; ++trv)
		;
	trv -= slen;
	suffp = trv;
	--trv;
	if (trv < path) {
		errno = EINVAL;
		return (0);
	}

	/* Fill space with random characters */
	while (trv >= path && *trv == 'X') {
		rand = arc4random() % (sizeof(padchar) - 1);
		*trv-- = padchar[rand];
	}
	start = trv + 1;

	/*
	 * check the target directory.
	 */
	if (doopen || domkdir) {
		for (;; --trv) {
			if (trv <= path)
				break;
			if (*trv == '/') {
				*trv = '\0';
				rval = stat(path, &sbuf);
				*trv = '/';
				if (rval != 0)
					return(0);
				if (!S_ISDIR(sbuf.st_mode)) {
					errno = ENOTDIR;
					return(0);
				}
				break;
			}
		}
	}

	for (;;) {
		if (doopen) {
			if ((*doopen =
			    open(path, O_CREAT|O_EXCL|O_RDWR, 0600)) >= 0)
				return(1);
			if (errno != EEXIST)
				return(0);
		} else if (domkdir) {
			if (mkdir(path, 0700) == 0)
				return(1);
			if (errno != EEXIST)
				return(0);
		} else if (lstat(path, &sbuf))
			return(errno == ENOENT ? 1 : 0);

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