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
23 static const char usage[] =
24 "[-?\n@(#)$Id: mktemp (AT&T Research) 2010-03-05 $\n]"
25 USAGE_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
66 int
b_mktemp(int argc,char ** argv,void * context)67 b_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