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