Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.bin/mktemp/mktemp.c	Lines:	29	63	46.0 %
Date:	2017-11-13	Branches:	24	47	51.1 %


/*	$OpenBSD: mktemp.c,v 1.22 2015/10/09 01:37:08 deraadt Exp $	*/

/*
 * Copyright (c) 1996, 1997, 2001-2003, 2013
 *	Todd C. Miller <Todd.Miller@courtesan.com>
 *
 * 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 <err.h>
#include <paths.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

__dead void usage(void);
__dead void fatal(const char *, ...) __attribute__((__format__(printf, 1, 2)));
__dead void fatalx(const char *, ...) __attribute__((__format__(printf, 1, 2)));

static int quiet;

int
main(int argc, char *argv[])
{
	int ch, fd, uflag = 0, tflag = 0, makedir = 0;
	char *cp, *template, *tempfile, *prefix = _PATH_TMP;
	size_t len;

	if (pledge("stdio rpath wpath cpath flock", NULL) == -1)
		err(1, "pledge");

	while ((ch = getopt(argc, argv, "dp:qtu")) != -1)
		switch(ch) {
		case 'd':
			makedir = 1;
			break;
		case 'p':
			prefix = optarg;
			tflag = 1;
			break;
		case 'q':
			quiet = 1;
			break;
		case 't':
			tflag = 1;
			break;
		case 'u':
			uflag = 1;
			break;
		default:
			usage();
	}

	/* If no template specified use a default one (implies -t mode) */
	switch (argc - optind) {
	case 1:
		template = argv[optind];
		break;
	case 0:
		template = "tmp.XXXXXXXXXX";
		tflag = 1;
		break;
	default:
		usage();
	}

	len = strlen(template);
	if (len < 6 || strcmp(&template[len - 6], "XXXXXX")) {
		fatalx("insufficient number of Xs in template `%s'",
		    template);
	}
	if (tflag) {
		if (strchr(template, '/')) {
			fatalx("template must not contain directory "
			    "separators in -t mode");
		}

		cp = getenv("TMPDIR");
		if (cp != NULL && *cp != '\0')
			prefix = cp;
		len = strlen(prefix);
		while (len != 0 && prefix[len - 1] == '/')
			len--;

		if (asprintf(&tempfile, "%.*s/%s", (int)len, prefix, template) < 0)
			tempfile = NULL;
	} else
		tempfile = strdup(template);

	if (tempfile == NULL)
		fatalx("cannot allocate memory");

	if (makedir) {
		if (mkdtemp(tempfile) == NULL)
			fatal("cannot make temp dir %s", tempfile);
		if (uflag)
			(void)rmdir(tempfile);
	} else {
		if ((fd = mkstemp(tempfile)) < 0)
			fatal("cannot make temp file %s", tempfile);
		(void)close(fd);
		if (uflag)
			(void)unlink(tempfile);
	}

	(void)puts(tempfile);
	free(tempfile);

	exit(EXIT_SUCCESS);
}

__dead void
fatal(const char *fmt, ...)
{
	if (!quiet) {
		va_list ap;

		va_start(ap, fmt);
		vwarn(fmt, ap);
		va_end(ap);
	}
	exit(EXIT_FAILURE);
}

__dead void
fatalx(const char *fmt, ...)
{
	if (!quiet) {
		va_list ap;

		va_start(ap, fmt);
		vwarnx(fmt, ap);
		va_end(ap);
	}
	exit(EXIT_FAILURE);
}

__dead void
usage(void)
{
	extern char *__progname;

	(void)fprintf(stderr,
	    "usage: %s [-dqtu] [-p directory] [template]\n", __progname);
	exit(EXIT_FAILURE);
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: mktemp.c,v 1.22 2015/10/09 01:37:08 deraadt Exp $ */
2
3			/*
4			* Copyright (c) 1996, 1997, 2001-2003, 2013
5			* Todd C. Miller <Todd.Miller@courtesan.com>
6			*
7			* Permission to use, copy, modify, and distribute this software for any
8			* purpose with or without fee is hereby granted, provided that the above
9			* copyright notice and this permission notice appear in all copies.
10			*
11			* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12			* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13			* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14			* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15			* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16			* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17			* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18			*/
19
20			#include <err.h>
21			#include <paths.h>
22			#include <stdarg.h>
23			#include <stdio.h>
24			#include <stdlib.h>
25			#include <string.h>
26			#include <unistd.h>
27
28			__dead void usage(void);
29			__dead void fatal(const char *, ...) __attribute__((__format__(printf, 1, 2)));
30			__dead void fatalx(const char *, ...) __attribute__((__format__(printf, 1, 2)));
31
32			static int quiet;
33
34			int
35			main(int argc, char *argv[])
36			{
37			int ch, fd, uflag = 0, tflag = 0, makedir = 0;
38		524	char cp, template, tempfile, prefix = _PATH_TMP;
39			size_t len;
40
41	✗✓	262	if (pledge("stdio rpath wpath cpath flock", NULL) == -1)
42			err(1, "pledge");
43
44	✓✓	556	while ((ch = getopt(argc, argv, "dp:qtu")) != -1)
45	✓✗✓✗ ✗✗	16	switch(ch) {
46			case 'd':
47			makedir = 1;
48		12	break;
49			case 'p':
50			prefix = optarg;
51			tflag = 1;
52			break;
53			case 'q':
54		4	quiet = 1;
55		4	break;
56			case 't':
57			tflag = 1;
58			break;
59			case 'u':
60			uflag = 1;
61			break;
62			default:
63			usage();
64			}
65
66			/* If no template specified use a default one (implies -t mode) */
67	✓✓✗	262	switch (argc - optind) {
68			case 1:
69		250	template = argv[optind];
70		250	break;
71			case 0:
72			template = "tmp.XXXXXXXXXX";
73			tflag = 1;
74		12	break;
75			default:
76			usage();
77			}
78
79		262	len = strlen(template);
80	✓✗✗✓	524	if (len < 6 \|\| strcmp(&template[len - 6], "XXXXXX")) {
81			fatalx("insufficient number of Xs in template `%s'",
82			template);
83			}
84	✓✓	262	if (tflag) {
85	✗✓	12	if (strchr(template, '/')) {
86			fatalx("template must not contain directory "
87			"separators in -t mode");
88			}
89
90		12	cp = getenv("TMPDIR");
91	✗✓✗✗	12	if (cp != NULL && *cp != '\0')
92			prefix = cp;
93		12	len = strlen(prefix);
94	✓✗✓✓	96	while (len != 0 && prefix[len - 1] == '/')
95		12	len--;
96
97	✗✓	12	if (asprintf(&tempfile, "%.*s/%s", (int)len, prefix, template) < 0)
98			tempfile = NULL;
99			} else
100		250	tempfile = strdup(template);
101
102	✗✓	262	if (tempfile == NULL)
103			fatalx("cannot allocate memory");
104
105	✓✓	262	if (makedir) {
106	✗✓	12	if (mkdtemp(tempfile) == NULL)
107			fatal("cannot make temp dir %s", tempfile);
108	✗✓	12	if (uflag)
109			(void)rmdir(tempfile);
110			} else {
111	✗✓	250	if ((fd = mkstemp(tempfile)) < 0)
112			fatal("cannot make temp file %s", tempfile);
113		250	(void)close(fd);
114	✗✓	250	if (uflag)
115			(void)unlink(tempfile);
116			}
117
118			(void)puts(tempfile);
119			free(tempfile);
120
121			exit(EXIT_SUCCESS);
122			}
123
124			__dead void
125			fatal(const char *fmt, ...)
126			{
127			if (!quiet) {
128			va_list ap;
129
130			va_start(ap, fmt);
131			vwarn(fmt, ap);
132			va_end(ap);
133			}
134			exit(EXIT_FAILURE);
135			}
136
137			__dead void
138			fatalx(const char *fmt, ...)
139			{
140			if (!quiet) {
141			va_list ap;
142
143			va_start(ap, fmt);
144			vwarnx(fmt, ap);
145			va_end(ap);
146			}
147			exit(EXIT_FAILURE);
148			}
149
150			__dead void
151			usage(void)
152			{
153			extern char *__progname;
154
155			(void)fprintf(stderr,
156			"usage: %s [-dqtu] [-p directory] [template]\n", __progname);
157			exit(EXIT_FAILURE);
158			}