1/* $NetBSD: mktemp.c,v 1.10 2007/12/15 19:44:52 perry 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.10 2007/12/15 19:44:52 perry Exp $");
54#endif /* !__lint */
55
56static void usage(void) __dead;
57
58int
59main(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) {
104		if (tmpdir == NULL)
105			tmpdir = getenv("TMPDIR");
106		if (tmpdir == NULL)
107			(void)asprintf(&name, "%s%s.XXXXXXXX", _PATH_TMP,
108			    prefix);
109		else
110			(void)asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
111		/* if this fails, the program is in big trouble already */
112		if (name == NULL) {
113			if (qflag)
114				return 1;
115			else
116				errx(1, "Cannot generate template");
117		}
118	} else if (argc < 1) {
119		usage();
120	}
121
122	/* generate all requested files */
123	while (name != NULL || argc > 0) {
124		if (name == NULL) {
125			if (tmpdir)
126				(void)asprintf(&name, "%s/%s",
127				    tmpdir, argv[0]);
128			else
129				name = strdup(argv[0]);
130			argv++;
131			argc--;
132		}
133
134		if (dflag) {
135			if (mkdtemp(name) == NULL) {
136				ret = 1;
137				if (!qflag)
138					warn("mkdtemp failed on %s", name);
139			} else {
140				(void)printf("%s\n", name);
141				if (uflag)
142					(void)rmdir(name);
143			}
144		} else {
145			fd = mkstemp(name);
146			if (fd < 0) {
147				ret = 1;
148				if (!qflag)
149					warn("mkstemp failed on %s", name);
150			} else {
151				(void)close(fd);
152				if (uflag)
153					(void)unlink(name);
154				(void)printf("%s\n", name);
155			}
156		}
157		if (name)
158			free(name);
159		name = NULL;
160	}
161	return ret;
162}
163
164static void
165usage(void)
166{
167	(void)fprintf(stderr,
168	    "Usage: %s [-dqu] [-p <tmpdir>] {-t prefix | template ...}\n",
169	    getprogname());
170	exit (1);
171}
172