1 /* $NetBSD: mktemp.c,v 1.12 2012/11/03 13:34:08 christos Exp $ */
2
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 #if HAVE_NBTOOL_CONFIG_H
40 #include "nbtool_config.h"
41 #endif
42
43 #include <sys/cdefs.h>
44 #include <sys/types.h>
45 #include <err.h>
46 #include <paths.h>
47 #include <stdio.h>
48 #include <stdlib.h>
49 #include <string.h>
50 #include <unistd.h>
51
52 #if defined(__RCSID) && !defined(__lint)
53 __RCSID("$NetBSD: mktemp.c,v 1.12 2012/11/03 13:34:08 christos Exp $");
54 #endif /* !__lint */
55
56 static void usage(void) __dead;
57
58 int
main(int argc,char ** argv)59 main(int argc, char **argv)
60 {
61 int c, fd, ret;
62 char *tmpdir;
63 const char *prefix;
64 char *name;
65 int dflag, qflag, tflag, uflag;
66
67 setprogname(*argv);
68 ret = dflag = qflag = tflag = uflag = 0;
69 tmpdir = NULL;
70 prefix = "mktemp";
71 name = NULL;
72
73 while ((c = getopt(argc, argv, "dp:qt:u")) != -1)
74 switch (c) {
75 case 'd':
76 dflag++;
77 break;
78
79 case 'p':
80 tmpdir = optarg;
81 break;
82
83 case 'q':
84 qflag++;
85 break;
86
87 case 't':
88 prefix = optarg;
89 tflag++;
90 break;
91
92 case 'u':
93 uflag++;
94 break;
95
96 default:
97 usage();
98 }
99
100 argc -= optind;
101 argv += optind;
102
103 if (tflag == 0 && argc < 1)
104 tflag = 1;
105
106 if (tflag) {
107 if (tmpdir == NULL)
108 tmpdir = getenv("TMPDIR");
109 if (tmpdir == NULL)
110 (void)asprintf(&name, "%s%s.XXXXXXXX", _PATH_TMP,
111 prefix);
112 else
113 (void)asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
114 /* if this fails, the program is in big trouble already */
115 if (name == NULL) {
116 if (qflag)
117 return 1;
118 else
119 errx(1, "Cannot generate template");
120 }
121 } else if (argc < 1) {
122 usage();
123 }
124
125 /* generate all requested files */
126 while (name != NULL || argc > 0) {
127 if (name == NULL) {
128 if (tmpdir)
129 (void)asprintf(&name, "%s/%s",
130 tmpdir, argv[0]);
131 else
132 name = strdup(argv[0]);
133 argv++;
134 argc--;
135 }
136
137 if (dflag) {
138 if (mkdtemp(name) == NULL) {
139 ret = 1;
140 if (!qflag)
141 warn("mkdtemp failed on %s", name);
142 } else {
143 (void)printf("%s\n", name);
144 if (uflag)
145 (void)rmdir(name);
146 }
147 } else {
148 fd = mkstemp(name);
149 if (fd < 0) {
150 ret = 1;
151 if (!qflag)
152 warn("mkstemp failed on %s", name);
153 } else {
154 (void)close(fd);
155 if (uflag)
156 (void)unlink(name);
157 (void)printf("%s\n", name);
158 }
159 }
160 if (name)
161 free(name);
162 name = NULL;
163 }
164 return ret;
165 }
166
167 static void
usage(void)168 usage(void)
169 {
170 (void)fprintf(stderr,
171 "Usage: %s [-dqu] [-p <tmpdir>] {-t prefix | template ...}\n",
172 getprogname());
173 exit (1);
174 }
175