134f9b3eeSRoland Mainz /***********************************************************************
234f9b3eeSRoland Mainz * *
334f9b3eeSRoland Mainz * This software is part of the ast package *
4*b30d1939SAndy Fiddaman * Copyright (c) 1992-2012 AT&T Intellectual Property *
534f9b3eeSRoland Mainz * and is licensed under the *
6*b30d1939SAndy Fiddaman * Eclipse Public License, Version 1.0 *
734f9b3eeSRoland Mainz * by AT&T Intellectual Property *
834f9b3eeSRoland Mainz * *
934f9b3eeSRoland Mainz * A copy of the License is available at *
10*b30d1939SAndy Fiddaman * http://www.eclipse.org/org/documents/epl-v10.html *
11*b30d1939SAndy Fiddaman * (with md5 checksum b35adb5213ca9657e911e9befb180842) *
1234f9b3eeSRoland Mainz * *
1334f9b3eeSRoland Mainz * Information and Software Systems Research *
1434f9b3eeSRoland Mainz * AT&T Research *
1534f9b3eeSRoland Mainz * Florham Park NJ *
1634f9b3eeSRoland Mainz * *
1734f9b3eeSRoland Mainz * Glenn Fowler <gsf@research.att.com> *
1834f9b3eeSRoland Mainz * David Korn <dgk@research.att.com> *
1934f9b3eeSRoland Mainz * *
2034f9b3eeSRoland Mainz ***********************************************************************/
2134f9b3eeSRoland Mainz #pragma prototyped
2234f9b3eeSRoland Mainz
2334f9b3eeSRoland Mainz static const char usage[] =
243e14f97fSRoger A. Faulkner "[-?\n@(#)$Id: mktemp (AT&T Research) 2010-03-05 $\n]"
2534f9b3eeSRoland Mainz USAGE_LICENSE
2634f9b3eeSRoland Mainz "[+NAME?mktemp - make temporary file or directory]"
2734f9b3eeSRoland Mainz "[+DESCRIPTION?\bmktemp\b creates a temporary file with optional base "
2834f9b3eeSRoland Mainz "name prefix \aprefix\a. If \aprefix\a is omitted then \btmp_\b is used "
2934f9b3eeSRoland Mainz "and \b--tmp\b is implied. If \aprefix\a contains a directory prefix "
3034f9b3eeSRoland Mainz "then that directory overrides any of the directories described below. A "
3134f9b3eeSRoland Mainz "temporary file will have mode \brw-------\b and a temporary directory "
3234f9b3eeSRoland Mainz "will have mode \brwx------\b, subject to \bumask\b(1). Generated paths "
3334f9b3eeSRoland Mainz "have these attributes:]"
3434f9b3eeSRoland Mainz "{"
3534f9b3eeSRoland Mainz "[+*?Lower case to avoid clashes on case ignorant filesystems.]"
3634f9b3eeSRoland Mainz "[+*?Pseudo-random part to deter denial of service attacks.]"
373e14f97fSRoger A. Faulkner "[+*?Default pseudo-random part (no specific \bX...\b template) "
383e14f97fSRoger A. Faulkner "formatted to accomodate 8.3 filesystems.]"
3934f9b3eeSRoland Mainz "}"
403e14f97fSRoger A. Faulkner "[+?A consecutive trailing sequence of \bX\b's in \aprefix\a is replaced "
413e14f97fSRoger A. Faulkner "by the pseudo-random part. If there are no \bX\b's then the "
423e14f97fSRoger A. Faulkner "pseudo-random part is appended to the prefix.]"
4334f9b3eeSRoland Mainz "[d:directory?Create a directory instead of a regular file.]"
4434f9b3eeSRoland Mainz "[m:mode]:[mode?Set the mode of the created temporary to \amode\a. "
4534f9b3eeSRoland Mainz "\amode\a is symbolic or octal mode as in \bchmod\b(1). Relative modes "
4634f9b3eeSRoland Mainz "assume an initial mode of \bu=rwx\b.]"
4734f9b3eeSRoland Mainz "[p:default?Use \adirectory\a if the \bTMPDIR\b environment variable is "
4834f9b3eeSRoland Mainz "not defined. Implies \b--tmp\b.]:[directory]"
4934f9b3eeSRoland Mainz "[q:quiet?Suppress file and directory error diagnostics.]"
503e14f97fSRoger A. Faulkner "[R:regress?The pseudo random generator is seeded with \aseed\a instead "
513e14f97fSRoger A. Faulkner "of process/system specific transient data. Use for testing "
523e14f97fSRoger A. Faulkner "only. A seed of \b0\b is silently changed to \b1\b.]#[seed]"
5334f9b3eeSRoland Mainz "[t:tmp|temporary-directory?Create a path rooted in a temporary "
5434f9b3eeSRoland Mainz "directory.]"
5534f9b3eeSRoland Mainz "[u:unsafe|dry-run?Check for file/directory existence but do not create. "
563e14f97fSRoger A. Faulkner "Use this for testing only.]"
5734f9b3eeSRoland Mainz "\n"
5834f9b3eeSRoland Mainz "\n[ prefix ]\n"
5934f9b3eeSRoland Mainz "\n"
6034f9b3eeSRoland Mainz "[+SEE ALSO?\bmkdir\b(1), \bpathtemp\b(3), \bmktemp\b(3)]"
6134f9b3eeSRoland Mainz ;
6234f9b3eeSRoland Mainz
6334f9b3eeSRoland Mainz #include <cmd.h>
6434f9b3eeSRoland Mainz #include <ls.h>
6534f9b3eeSRoland Mainz
6634f9b3eeSRoland Mainz int
b_mktemp(int argc,char ** argv,Shbltin_t * context)67*b30d1939SAndy Fiddaman b_mktemp(int argc, char** argv, Shbltin_t* context)
6834f9b3eeSRoland Mainz {
6934f9b3eeSRoland Mainz mode_t mode = 0;
7034f9b3eeSRoland Mainz mode_t mask;
7134f9b3eeSRoland Mainz int fd;
7234f9b3eeSRoland Mainz int i;
7334f9b3eeSRoland Mainz int quiet = 0;
7434f9b3eeSRoland Mainz int unsafe = 0;
7534f9b3eeSRoland Mainz int* fdp = &fd;
7634f9b3eeSRoland Mainz char* dir = "";
7734f9b3eeSRoland Mainz char* pfx;
7834f9b3eeSRoland Mainz char* t;
7934f9b3eeSRoland Mainz char path[PATH_MAX];
8034f9b3eeSRoland Mainz
8134f9b3eeSRoland Mainz cmdinit(argc, argv, context, ERROR_CATALOG, ERROR_NOTIFY);
8234f9b3eeSRoland Mainz for (;;)
8334f9b3eeSRoland Mainz {
8434f9b3eeSRoland Mainz switch (optget(argv, usage))
8534f9b3eeSRoland Mainz {
8634f9b3eeSRoland Mainz case 'd':
8734f9b3eeSRoland Mainz fdp = 0;
8834f9b3eeSRoland Mainz continue;
8934f9b3eeSRoland Mainz case 'm':
9034f9b3eeSRoland Mainz mode = strperm(pfx = opt_info.arg, &opt_info.arg, S_IRWXU);
9134f9b3eeSRoland Mainz if (*opt_info.arg)
9234f9b3eeSRoland Mainz error(ERROR_exit(0), "%s: invalid mode", pfx);
9334f9b3eeSRoland Mainz continue;
9434f9b3eeSRoland Mainz case 'p':
9534f9b3eeSRoland Mainz if ((t = getenv("TMPDIR")) && *t)
9634f9b3eeSRoland Mainz dir = 0;
9734f9b3eeSRoland Mainz else
9834f9b3eeSRoland Mainz dir = opt_info.arg;
9934f9b3eeSRoland Mainz continue;
10034f9b3eeSRoland Mainz case 'q':
10134f9b3eeSRoland Mainz quiet = 1;
10234f9b3eeSRoland Mainz continue;
10334f9b3eeSRoland Mainz case 't':
10434f9b3eeSRoland Mainz dir = 0;
10534f9b3eeSRoland Mainz continue;
10634f9b3eeSRoland Mainz case 'u':
10734f9b3eeSRoland Mainz unsafe = 1;
10834f9b3eeSRoland Mainz fdp = 0;
10934f9b3eeSRoland Mainz continue;
1103e14f97fSRoger A. Faulkner case 'R':
1113e14f97fSRoger A. Faulkner if (!pathtemp(NiL, 0, opt_info.arg, "/seed", NiL))
1123e14f97fSRoger A. Faulkner error(2, "%s: regression test initializtion failed", opt_info.arg);
1133e14f97fSRoger A. Faulkner continue;
11434f9b3eeSRoland Mainz case ':':
11534f9b3eeSRoland Mainz error(2, "%s", opt_info.arg);
116*b30d1939SAndy Fiddaman break;
11734f9b3eeSRoland Mainz case '?':
11834f9b3eeSRoland Mainz error(ERROR_usage(2), "%s", opt_info.arg);
119*b30d1939SAndy Fiddaman break;
12034f9b3eeSRoland Mainz }
12134f9b3eeSRoland Mainz break;
12234f9b3eeSRoland Mainz }
12334f9b3eeSRoland Mainz argv += opt_info.index;
12434f9b3eeSRoland Mainz if (error_info.errors || (pfx = *argv++) && *argv)
12534f9b3eeSRoland Mainz error(ERROR_usage(2), "%s", optusage(NiL));
12634f9b3eeSRoland Mainz mask = umask(0);
12734f9b3eeSRoland Mainz if (!mode)
12834f9b3eeSRoland Mainz mode = (fdp ? (S_IRUSR|S_IWUSR) : S_IRWXU) & ~mask;
12934f9b3eeSRoland Mainz umask(~mode & (S_IRWXU|S_IRWXG|S_IRWXO));
13034f9b3eeSRoland Mainz if (!pfx)
13134f9b3eeSRoland Mainz {
13234f9b3eeSRoland Mainz pfx = "tmp_";
13334f9b3eeSRoland Mainz if (dir && !*dir)
13434f9b3eeSRoland Mainz dir = 0;
13534f9b3eeSRoland Mainz }
13634f9b3eeSRoland Mainz if (t = strrchr(pfx, '/'))
13734f9b3eeSRoland Mainz {
13834f9b3eeSRoland Mainz i = ++t - pfx;
13934f9b3eeSRoland Mainz dir = fmtbuf(i);
14034f9b3eeSRoland Mainz memcpy(dir, pfx, i);
14134f9b3eeSRoland Mainz dir[i] = 0;
14234f9b3eeSRoland Mainz pfx = t;
14334f9b3eeSRoland Mainz }
14434f9b3eeSRoland Mainz for (;;)
14534f9b3eeSRoland Mainz {
14634f9b3eeSRoland Mainz if (!pathtemp(path, sizeof(path), dir, pfx, fdp))
14734f9b3eeSRoland Mainz {
14834f9b3eeSRoland Mainz if (quiet)
14934f9b3eeSRoland Mainz error_info.errors++;
15034f9b3eeSRoland Mainz else
15134f9b3eeSRoland Mainz error(ERROR_SYSTEM|2, "cannot create temporary path");
15234f9b3eeSRoland Mainz break;
15334f9b3eeSRoland Mainz }
15434f9b3eeSRoland Mainz if (fdp || unsafe || !mkdir(path, mode))
15534f9b3eeSRoland Mainz {
15634f9b3eeSRoland Mainz if (fdp)
15734f9b3eeSRoland Mainz close(*fdp);
15834f9b3eeSRoland Mainz sfputr(sfstdout, path, '\n');
15934f9b3eeSRoland Mainz break;
16034f9b3eeSRoland Mainz }
16134f9b3eeSRoland Mainz if (sh_checksig(context))
16234f9b3eeSRoland Mainz {
16334f9b3eeSRoland Mainz error_info.errors++;
16434f9b3eeSRoland Mainz break;
16534f9b3eeSRoland Mainz }
16634f9b3eeSRoland Mainz }
16734f9b3eeSRoland Mainz umask(mask);
16834f9b3eeSRoland Mainz return error_info.errors != 0;
16934f9b3eeSRoland Mainz }
170