1/*-
2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3 *
4 * Copyright (c) 1994, 1995, 1996, 1998 Peter Wemm <peter@netplex.com.au>
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 *
28 */
29
30/*
31 * This program was originally written long ago, originally for a non
32 * BSD-like OS without mkstemp().  It's been modified over the years
33 * to use mkstemp() rather than the original O_CREAT|O_EXCL/fstat/lstat
34 * etc style hacks.
35 * A cleanup, misc options and mkdtemp() calls were added to try and work
36 * more like the OpenBSD version - which was first to publish the interface.
37 */
38
39#include <err.h>
40#include <paths.h>
41#include <stdio.h>
42#include <stdlib.h>
43#include <string.h>
44#include <unistd.h>
45
46#ifndef lint
47static const char rcsid[] =
48	"$FreeBSD$";
49#endif /* not lint */
50
51static void usage(void);
52
53int
54main(int argc, char **argv)
55{
56	int c, fd, ret;
57	char *tmpdir;
58	const char *prefix;
59	char *name;
60	int dflag, qflag, tflag, uflag;
61
62	ret = dflag = qflag = tflag = uflag = 0;
63	prefix = "mktemp";
64	name = NULL;
65
66	while ((c = getopt(argc, argv, "dqt:u")) != -1)
67		switch (c) {
68		case 'd':
69			dflag++;
70			break;
71
72		case 'q':
73			qflag++;
74			break;
75
76		case 't':
77			prefix = optarg;
78			tflag++;
79			break;
80
81		case 'u':
82			uflag++;
83			break;
84
85		default:
86			usage();
87		}
88
89	argc -= optind;
90	argv += optind;
91
92	if (!tflag && argc < 1) {
93		tflag = 1;
94		prefix = "tmp";
95	}
96
97	if (tflag) {
98		tmpdir = getenv("TMPDIR");
99		if (tmpdir == NULL)
100			asprintf(&name, "%s%s.XXXXXXXX", _PATH_TMP, prefix);
101		else
102			asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
103		/* if this fails, the program is in big trouble already */
104		if (name == NULL) {
105			if (qflag)
106				return (1);
107			else
108				errx(1, "cannot generate template");
109		}
110	}
111
112	/* generate all requested files */
113	while (name != NULL || argc > 0) {
114		if (name == NULL) {
115			name = strdup(argv[0]);
116			argv++;
117			argc--;
118		}
119
120		if (dflag) {
121			if (mkdtemp(name) == NULL) {
122				ret = 1;
123				if (!qflag)
124					warn("mkdtemp failed on %s", name);
125			} else {
126				printf("%s\n", name);
127				if (uflag)
128					rmdir(name);
129			}
130		} else {
131			fd = mkstemp(name);
132			if (fd < 0) {
133				ret = 1;
134				if (!qflag)
135					warn("mkstemp failed on %s", name);
136			} else {
137				close(fd);
138				if (uflag)
139					unlink(name);
140				printf("%s\n", name);
141			}
142		}
143		if (name)
144			free(name);
145		name = NULL;
146	}
147	return (ret);
148}
149
150static void
151usage(void)
152{
153	fprintf(stderr,
154		"usage: mktemp [-d] [-q] [-t prefix] [-u] template ...\n");
155	fprintf(stderr,
156		"       mktemp [-d] [-q] [-u] -t prefix \n");
157	exit (1);
158}
159