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
47 static const char rcsid[] =
48 "$FreeBSD: stable/12/usr.bin/mktemp/mktemp.c 326276 2017-11-27 15:37:16Z pfg $";
49 #endif /* not lint */
50
51 static void usage(void);
52
53 int
main(int argc,char ** argv)54 main(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
150 static void
usage(void)151 usage(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