mktemp.c revision 200463 
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
45static const char rcsid[] =
46	"$FreeBSD: head/usr.bin/mktemp/mktemp.c 200463 2009-12-13 03:18:37Z delphij $";
47#endif /* not lint */
48
49static void usage(void);
50
51int
52main(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) {
91		tmpdir = getenv("TMPDIR");
92		if (tmpdir == NULL)
93			asprintf(&name, "%s%s.XXXXXXXX", _PATH_TMP, prefix);
94		else
95			asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
96		/* if this fails, the program is in big trouble already */
97		if (name == NULL) {
98			if (qflag)
99				return (1);
100			else
101				errx(1, "cannot generate template");
102		}
103	} else if (argc < 1) {
104		usage();
105	}
106
107	/* generate all requested files */
108	while (name != NULL || argc > 0) {
109		if (name == NULL) {
110			name = strdup(argv[0]);
111			argv++;
112			argc--;
113		}
114
115		if (dflag) {
116			if (mkdtemp(name) == NULL) {
117				ret = 1;
118				if (!qflag)
119					warn("mkdtemp failed on %s", name);
120			} else {
121				printf("%s\n", name);
122				if (uflag)
123					rmdir(name);
124			}
125		} else {
126			fd = mkstemp(name);
127			if (fd < 0) {
128				ret = 1;
129				if (!qflag)
130					warn("mkstemp failed on %s", name);
131			} else {
132				close(fd);
133				if (uflag)
134					unlink(name);
135				printf("%s\n", name);
136			}
137		}
138		if (name)
139			free(name);
140		name = NULL;
141	}
142	return (ret);
143}
144
145static void
146usage(void)
147{
148	fprintf(stderr,
149		"usage: mktemp [-d] [-q] [-t prefix] [-u] template ...\n");
150	fprintf(stderr,
151		"       mktemp [-d] [-q] [-u] -t prefix \n");
152	exit (1);
153}
154