Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	lib/libc/stdio/mktemp.c	Lines:	21	45	46.7 %
Date:	2017-11-07	Branches:	13	36	36.1 %


/*	$OpenBSD: mktemp.c,v 1.38 2015/09/13 08:31:47 guenther Exp $ */
/*
 * Copyright (c) 1996-1998, 2008 Theo de Raadt
 * Copyright (c) 1997, 2008-2009 Todd C. Miller
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

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

#define MKTEMP_NAME	0
#define MKTEMP_FILE	1
#define MKTEMP_DIR	2

#define TEMPCHARS	"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
#define NUM_CHARS	(sizeof(TEMPCHARS) - 1)
#define MIN_X		6

#define MKOTEMP_FLAGS	(O_APPEND | O_CLOEXEC | O_DSYNC | O_RSYNC | O_SYNC)

#ifndef nitems
#define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
#endif

static int
mktemp_internal(char *path, int slen, int mode, int flags)
{
	char *start, *cp, *ep;
	const char tempchars[] = TEMPCHARS;
	unsigned int tries;
	struct stat sb;
	size_t len;
	int fd;

	len = strlen(path);
	if (len < MIN_X || slen < 0 || (size_t)slen > len - MIN_X) {
		errno = EINVAL;
		return(-1);
	}
	ep = path + len - slen;

	for (start = ep; start > path && start[-1] == 'X'; start--)
		;
	if (ep - start < MIN_X) {
		errno = EINVAL;
		return(-1);
	}

	if (flags & ~MKOTEMP_FLAGS) {
		errno = EINVAL;
		return(-1);
	}
	flags |= O_CREAT | O_EXCL | O_RDWR;

	tries = INT_MAX;
	do {
		cp = start;
		do {
			unsigned short rbuf[16];
			unsigned int i;

			/*
			 * Avoid lots of arc4random() calls by using
			 * a buffer sized for up to 16 Xs at a time.
			 */
			arc4random_buf(rbuf, sizeof(rbuf));
			for (i = 0; i < nitems(rbuf) && cp != ep; i++)
				*cp++ = tempchars[rbuf[i] % NUM_CHARS];
		} while (cp != ep);

		switch (mode) {
		case MKTEMP_NAME:
			if (lstat(path, &sb) != 0)
				return(errno == ENOENT ? 0 : -1);
			break;
		case MKTEMP_FILE:
			fd = open(path, flags, S_IRUSR|S_IWUSR);
			if (fd != -1 || errno != EEXIST)
				return(fd);
			break;
		case MKTEMP_DIR:
			if (mkdir(path, S_IRUSR|S_IWUSR|S_IXUSR) == 0)
				return(0);
			if (errno != EEXIST)
				return(-1);
			break;
		}
	} while (--tries);

	errno = EEXIST;
	return(-1);
}

char *
_mktemp(char *path)
{
	if (mktemp_internal(path, 0, MKTEMP_NAME, 0) == -1)
		return(NULL);
	return(path);
}

__warn_references(mktemp,
    "warning: mktemp() possibly used unsafely; consider using mkstemp()");

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

int
mkostemps(char *path, int slen, int flags)
{
	return(mktemp_internal(path, slen, MKTEMP_FILE, flags));
}

int
mkstemp(char *path)
{
	return(mktemp_internal(path, 0, MKTEMP_FILE, 0));
}
DEF_WEAK(mkstemp);

int
mkostemp(char *path, int flags)
{
	return(mktemp_internal(path, 0, MKTEMP_FILE, flags));
}
DEF_WEAK(mkostemp);

int
mkstemps(char *path, int slen)
{
	return(mktemp_internal(path, slen, MKTEMP_FILE, 0));
}

char *
mkdtemp(char *path)
{
	int error;

	error = mktemp_internal(path, 0, MKTEMP_DIR, 0);
	return(error ? NULL : path);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: mktemp.c,v 1.38 2015/09/13 08:31:47 guenther Exp $ */
2			/*
3			* Copyright (c) 1996-1998, 2008 Theo de Raadt
4			* Copyright (c) 1997, 2008-2009 Todd C. Miller
5			*
6			* Permission to use, copy, modify, and distribute this software for any
7			* purpose with or without fee is hereby granted, provided that the above
8			* copyright notice and this permission notice appear in all copies.
9			*
10			* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11			* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12			* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13			* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14			* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15			* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16			* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17			*/
18
19			#include <sys/types.h>
20			#include <sys/stat.h>
21			#include <errno.h>
22			#include <fcntl.h>
23			#include <limits.h>
24			#include <stdio.h>
25			#include <stdlib.h>
26			#include <string.h>
27			#include <ctype.h>
28			#include <unistd.h>
29
30			#define MKTEMP_NAME 0
31			#define MKTEMP_FILE 1
32			#define MKTEMP_DIR 2
33
34			#define TEMPCHARS "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
35			#define NUM_CHARS (sizeof(TEMPCHARS) - 1)
36			#define MIN_X 6
37
38			#define MKOTEMP_FLAGS (O_APPEND \| O_CLOEXEC \| O_DSYNC \| O_RSYNC \| O_SYNC)
39
40			#ifndef nitems
41			#define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
42			#endif
43
44			static int
45			mktemp_internal(char *path, int slen, int mode, int flags)
46			{
47			char start, cp, *ep;
48			const char tempchars[] = TEMPCHARS;
49			unsigned int tries;
50		3500	struct stat sb;
51			size_t len;
52			int fd;
53
54		1750	len = strlen(path);
55	✓✗✗✓	3500	if (len < MIN_X \|\| slen < 0 \|\| (size_t)slen > len - MIN_X) {
56			errno = EINVAL;
57			return(-1);
58			}
59		1750	ep = path + len - slen;
60
61	✓✗✓✓	47250	for (start = ep; start > path && start[-1] == 'X'; start--)
62			;
63	✗✓	1750	if (ep - start < MIN_X) {
64			errno = EINVAL;
65			return(-1);
66			}
67
68	✗✓	1750	if (flags & ~MKOTEMP_FLAGS) {
69			errno = EINVAL;
70			return(-1);
71			}
72		1750	flags \|= O_CREAT \| O_EXCL \| O_RDWR;
73
74			tries = INT_MAX;
75		1750	do {
76			cp = start;
77		1750	do {
78		1750	unsigned short rbuf[16];
79			unsigned int i;
80
81			/*
82			* Avoid lots of arc4random() calls by using
83			* a buffer sized for up to 16 Xs at a time.
84			*/
85		1750	arc4random_buf(rbuf, sizeof(rbuf));
86	✓✗✓✓	47250	for (i = 0; i < nitems(rbuf) && cp != ep; i++)
87		14000	*cp++ = tempchars[rbuf[i] % NUM_CHARS];
88	✗✓	1750	} while (cp != ep);
89
90	✗✓✗✗	1750	switch (mode) {
91			case MKTEMP_NAME:
92			if (lstat(path, &sb) != 0)
93			return(errno == ENOENT ? 0 : -1);
94			break;
95			case MKTEMP_FILE:
96		1750	fd = open(path, flags, S_IRUSR\|S_IWUSR);
97	✗✓✗✗	1750	if (fd != -1 \|\| errno != EEXIST)
98		1750	return(fd);
99			break;
100			case MKTEMP_DIR:
101			if (mkdir(path, S_IRUSR\|S_IWUSR\|S_IXUSR) == 0)
102			return(0);
103			if (errno != EEXIST)
104			return(-1);
105			break;
106			}
107			} while (--tries);
108
109			errno = EEXIST;
110			return(-1);
111		1750	}
112
113			char *
114			_mktemp(char *path)
115			{
116			if (mktemp_internal(path, 0, MKTEMP_NAME, 0) == -1)
117			return(NULL);
118			return(path);
119			}
120
121			__warn_references(mktemp,
122			"warning: mktemp() possibly used unsafely; consider using mkstemp()");
123
124			char *
125			mktemp(char *path)
126			{
127			return(_mktemp(path));
128			}
129
130			int
131			mkostemps(char *path, int slen, int flags)
132			{
133			return(mktemp_internal(path, slen, MKTEMP_FILE, flags));
134			}
135
136			int
137			mkstemp(char *path)
138			{
139		3500	return(mktemp_internal(path, 0, MKTEMP_FILE, 0));
140			}
141			DEF_WEAK(mkstemp);
142
143			int
144			mkostemp(char *path, int flags)
145			{
146			return(mktemp_internal(path, 0, MKTEMP_FILE, flags));
147			}
148			DEF_WEAK(mkostemp);
149
150			int
151			mkstemps(char *path, int slen)
152			{
153			return(mktemp_internal(path, slen, MKTEMP_FILE, 0));
154			}
155
156			char *
157			mkdtemp(char *path)
158			{
159			int error;
160
161			error = mktemp_internal(path, 0, MKTEMP_DIR, 0);
162			return(error ? NULL : path);
163			}