1/***********************************************************************
2*                                                                      *
3*               This software is part of the ast package               *
4*          Copyright (c) 1992-2010 AT&T Intellectual Property          *
5*                      and is licensed under the                       *
6*                  Common Public License, Version 1.0                  *
7*                    by AT&T Intellectual Property                     *
8*                                                                      *
9*                A copy of the License is available at                 *
10*            http://www.opensource.org/licenses/cpl1.0.txt             *
11*         (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9)         *
12*                                                                      *
13*              Information and Software Systems Research               *
14*                            AT&T Research                             *
15*                           Florham Park NJ                            *
16*                                                                      *
17*                 Glenn Fowler <gsf@research.att.com>                  *
18*                  David Korn <dgk@research.att.com>                   *
19*                                                                      *
20***********************************************************************/
21#pragma prototyped
22
23static const char usage[] =
24"[-?\n@(#)$Id: mktemp (AT&T Research) 2010-03-05 $\n]"
25USAGE_LICENSE
26"[+NAME?mktemp - make temporary file or directory]"
27"[+DESCRIPTION?\bmktemp\b creates a temporary file with optional base "
28    "name prefix \aprefix\a. If \aprefix\a is omitted then \btmp_\b is used "
29    "and \b--tmp\b is implied. If \aprefix\a contains a directory prefix "
30    "then that directory overrides any of the directories described below. A "
31    "temporary file will have mode \brw-------\b and a temporary directory "
32    "will have mode \brwx------\b, subject to \bumask\b(1). Generated paths "
33    "have these attributes:]"
34    "{"
35        "[+*?Lower case to avoid clashes on case ignorant filesystems.]"
36        "[+*?Pseudo-random part to deter denial of service attacks.]"
37        "[+*?Default pseudo-random part (no specific \bX...\b template) "
38            "formatted to accomodate 8.3 filesystems.]"
39    "}"
40"[+?A consecutive trailing sequence of \bX\b's in \aprefix\a is replaced "
41    "by the pseudo-random part. If there are no \bX\b's then the "
42    "pseudo-random part is appended to the prefix.]"
43"[d:directory?Create a directory instead of a regular file.]"
44"[m:mode]:[mode?Set the mode of the created temporary to \amode\a. "
45    "\amode\a is symbolic or octal mode as in \bchmod\b(1). Relative modes "
46    "assume an initial mode of \bu=rwx\b.]"
47"[p:default?Use \adirectory\a if the \bTMPDIR\b environment variable is "
48    "not defined. Implies \b--tmp\b.]:[directory]"
49"[q:quiet?Suppress file and directory error diagnostics.]"
50"[R:regress?The pseudo random generator is seeded with \aseed\a instead "
51    "of process/system specific transient data. Use for testing "
52    "only. A seed of \b0\b is silently changed to \b1\b.]#[seed]"
53"[t:tmp|temporary-directory?Create a path rooted in a temporary "
54    "directory.]"
55"[u:unsafe|dry-run?Check for file/directory existence but do not create. "
56    "Use this for testing only.]"
57"\n"
58"\n[ prefix ]\n"
59"\n"
60"[+SEE ALSO?\bmkdir\b(1), \bpathtemp\b(3), \bmktemp\b(3)]"
61;
62
63#include <cmd.h>
64#include <ls.h>
65
66int
67b_mktemp(int argc, char** argv, void* context)
68{
69	mode_t		mode = 0;
70	mode_t		mask;
71	int		fd;
72	int		i;
73	int		quiet = 0;
74	int		unsafe = 0;
75	int*		fdp = &fd;
76	char*		dir = "";
77	char*		pfx;
78	char*		t;
79	char		path[PATH_MAX];
80
81	cmdinit(argc, argv, context, ERROR_CATALOG, ERROR_NOTIFY);
82	for (;;)
83	{
84		switch (optget(argv, usage))
85		{
86		case 0:
87			break;
88		case 'd':
89			fdp = 0;
90			continue;
91		case 'm':
92			mode = strperm(pfx = opt_info.arg, &opt_info.arg, S_IRWXU);
93			if (*opt_info.arg)
94				error(ERROR_exit(0), "%s: invalid mode", pfx);
95			continue;
96		case 'p':
97			if ((t = getenv("TMPDIR")) && *t)
98				dir = 0;
99			else
100				dir = opt_info.arg;
101			continue;
102		case 'q':
103			quiet = 1;
104			continue;
105		case 't':
106			dir = 0;
107			continue;
108		case 'u':
109			unsafe = 1;
110			fdp = 0;
111			continue;
112		case 'R':
113			if (!pathtemp(NiL, 0, opt_info.arg, "/seed", NiL))
114				error(2, "%s: regression test initializtion failed", opt_info.arg);
115			continue;
116		case ':':
117			error(2, "%s", opt_info.arg);
118			continue;
119		case '?':
120			error(ERROR_usage(2), "%s", opt_info.arg);
121			continue;
122		}
123		break;
124	}
125	argv += opt_info.index;
126	if (error_info.errors || (pfx = *argv++) && *argv)
127		error(ERROR_usage(2), "%s", optusage(NiL));
128	mask = umask(0);
129	if (!mode)
130		mode = (fdp ? (S_IRUSR|S_IWUSR) : S_IRWXU) & ~mask;
131	umask(~mode & (S_IRWXU|S_IRWXG|S_IRWXO));
132	if (!pfx)
133	{
134		pfx = "tmp_";
135		if (dir && !*dir)
136			dir = 0;
137	}
138	if (t = strrchr(pfx, '/'))
139	{
140		i = ++t - pfx;
141		dir = fmtbuf(i);
142		memcpy(dir, pfx, i);
143		dir[i] = 0;
144		pfx = t;
145	}
146	for (;;)
147	{
148		if (!pathtemp(path, sizeof(path), dir, pfx, fdp))
149		{
150			if (quiet)
151				error_info.errors++;
152			else
153				error(ERROR_SYSTEM|2, "cannot create temporary path");
154			break;
155		}
156		if (fdp || unsafe || !mkdir(path, mode))
157		{
158			if (fdp)
159				close(*fdp);
160			sfputr(sfstdout, path, '\n');
161			break;
162		}
163		if (sh_checksig(context))
164		{
165			error_info.errors++;
166			break;
167		}
168	}
169	umask(mask);
170	return error_info.errors != 0;
171}
172