34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz/***********************************************************************
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* This software is part of the ast package *
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner* Copyright (c) 1992-2010 AT&T Intellectual Property *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* and is licensed under the *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* Common Public License, Version 1.0 *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* by AT&T Intellectual Property *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* A copy of the License is available at *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* http://www.opensource.org/licenses/cpl1.0.txt *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9) *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* Information and Software Systems Research *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* AT&T Research *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* Florham Park NJ *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* Glenn Fowler <gsf@research.att.com> *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* David Korn <dgk@research.att.com> *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz* *
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz***********************************************************************/
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz#pragma prototyped
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainzstatic const char usage[] =
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner"[-?\n@(#)$Id: mktemp (AT&T Research) 2010-03-05 $\n]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland MainzUSAGE_LICENSE
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[+NAME?mktemp - make temporary file or directory]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[+DESCRIPTION?\bmktemp\b creates a temporary file with optional base "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "name prefix \aprefix\a. If \aprefix\a is omitted then \btmp_\b is used "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "and \b--tmp\b is implied. If \aprefix\a contains a directory prefix "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "then that directory overrides any of the directories described below. A "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "temporary file will have mode \brw-------\b and a temporary directory "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "will have mode \brwx------\b, subject to \bumask\b(1). Generated paths "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "have these attributes:]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "{"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "[+*?Lower case to avoid clashes on case ignorant filesystems.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "[+*?Pseudo-random part to deter denial of service attacks.]"
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner "[+*?Default pseudo-random part (no specific \bX...\b template) "
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner "formatted to accomodate 8.3 filesystems.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "}"
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner"[+?A consecutive trailing sequence of \bX\b's in \aprefix\a is replaced "
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner "by the pseudo-random part. If there are no \bX\b's then the "
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner "pseudo-random part is appended to the prefix.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[d:directory?Create a directory instead of a regular file.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[m:mode]:[mode?Set the mode of the created temporary to \amode\a. "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "\amode\a is symbolic or octal mode as in \bchmod\b(1). Relative modes "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "assume an initial mode of \bu=rwx\b.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[p:default?Use \adirectory\a if the \bTMPDIR\b environment variable is "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "not defined. Implies \b--tmp\b.]:[directory]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[q:quiet?Suppress file and directory error diagnostics.]"
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner"[R:regress?The pseudo random generator is seeded with \aseed\a instead "
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner "of process/system specific transient data. Use for testing "
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner "only. A seed of \b0\b is silently changed to \b1\b.]#[seed]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[t:tmp|temporary-directory?Create a path rooted in a temporary "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "directory.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[u:unsafe|dry-run?Check for file/directory existence but do not create. "
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner "Use this for testing only.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"\n"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"\n[ prefix ]\n"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"\n"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[+SEE ALSO?\bmkdir\b(1), \bpathtemp\b(3), \bmktemp\b(3)]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz#include <cmd.h>
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz#include <ls.h>
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainzint
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainzb_mktemp(int argc, char** argv, void* context)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz{
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode_t mode = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode_t mask;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz int fd;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz int i;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz int quiet = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz int unsafe = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz int* fdp = &fd;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz char* dir = "";
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz char* pfx;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz char* t;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz char path[PATH_MAX];
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz cmdinit(argc, argv, context, ERROR_CATALOG, ERROR_NOTIFY);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz for (;;)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz switch (optget(argv, usage))
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 0:
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz break;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'd':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz fdp = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'm':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode = strperm(pfx = opt_info.arg, &opt_info.arg, S_IRWXU);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (*opt_info.arg)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error(ERROR_exit(0), "%s: invalid mode", pfx);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'p':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if ((t = getenv("TMPDIR")) && *t)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz dir = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz else
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz dir = opt_info.arg;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'q':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz quiet = 1;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 't':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz dir = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'u':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz unsafe = 1;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz fdp = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner case 'R':
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner if (!pathtemp(NiL, 0, opt_info.arg, "/seed", NiL))
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner error(2, "%s: regression test initializtion failed", opt_info.arg);
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case ':':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error(2, "%s", opt_info.arg);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case '?':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error(ERROR_usage(2), "%s", opt_info.arg);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz break;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz argv += opt_info.index;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (error_info.errors || (pfx = *argv++) && *argv)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error(ERROR_usage(2), "%s", optusage(NiL));
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mask = umask(0);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!mode)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode = (fdp ? (S_IRUSR|S_IWUSR) : S_IRWXU) & ~mask;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz umask(~mode & (S_IRWXU|S_IRWXG|S_IRWXO));
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!pfx)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz pfx = "tmp_";
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (dir && !*dir)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz dir = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (t = strrchr(pfx, '/'))
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz i = ++t - pfx;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz dir = fmtbuf(i);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz memcpy(dir, pfx, i);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz dir[i] = 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz pfx = t;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz for (;;)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!pathtemp(path, sizeof(path), dir, pfx, fdp))
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (quiet)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error_info.errors++;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz else
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error(ERROR_SYSTEM|2, "cannot create temporary path");
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz break;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (fdp || unsafe || !mkdir(path, mode))
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (fdp)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz close(*fdp);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz sfputr(sfstdout, path, '\n');
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz break;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (sh_checksig(context))
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error_info.errors++;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz break;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz umask(mask);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz return error_info.errors != 0;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz}