#include <err.h>
#ifdef __APPLE__
#include <limits.h>
#endif
#include <paths.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#ifndef lint
static const char rcsid[] =
"$FreeBSD$";
#endif
static void usage(void);
int
main(int argc, char **argv)
{
int c, fd, ret;
char *tmpdir;
const char *prefix;
char *name;
int dflag, qflag, tflag, uflag;
#ifdef __APPLE__
char tmpbuf[PATH_MAX];
size_t len;
#endif
ret = dflag = qflag = tflag = uflag = 0;
prefix = "mktemp";
name = NULL;
while ((c = getopt(argc, argv, "dqt:u")) != -1)
switch (c) {
case 'd':
dflag++;
break;
case 'q':
qflag++;
break;
case 't':
prefix = optarg;
tflag++;
break;
case 'u':
uflag++;
break;
default:
usage();
}
argc -= optind;
argv += optind;
if (!tflag && argc < 1) {
tflag = 1;
prefix = "tmp";
}
if (tflag) {
#ifdef __APPLE__
if (confstr(_CS_DARWIN_USER_TEMP_DIR, tmpbuf, sizeof(tmpbuf)) > 0) {
tmpdir = tmpbuf;
} else {
tmpdir = getenv("TMPDIR");
}
if (tmpdir == NULL) {
tmpdir = _PATH_TMP;
}
len = strlen(tmpdir);
if (len > 0 && tmpdir[len - 1] == '/') {
asprintf(&name, "%s%s.XXXXXXXX", tmpdir, prefix);
} else {
asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
}
#else
tmpdir = getenv("TMPDIR");
if (tmpdir == NULL)
asprintf(&name, "%s%s.XXXXXXXX", _PATH_TMP, prefix);
else
asprintf(&name, "%s/%s.XXXXXXXX", tmpdir, prefix);
#endif
if (name == NULL) {
if (qflag)
return (1);
else
errx(1, "cannot generate template");
}
}
while (name != NULL || argc > 0) {
if (name == NULL) {
name = strdup(argv[0]);
argv++;
argc--;
}
if (dflag) {
if (mkdtemp(name) == NULL) {
ret = 1;
if (!qflag)
warn("mkdtemp failed on %s", name);
} else {
printf("%s\n", name);
if (uflag)
rmdir(name);
}
} else {
fd = mkstemp(name);
if (fd < 0) {
ret = 1;
if (!qflag)
warn("mkstemp failed on %s", name);
} else {
close(fd);
if (uflag)
unlink(name);
printf("%s\n", name);
}
}
if (name)
free(name);
name = NULL;
}
return (ret);
}
static void
usage(void)
{
fprintf(stderr,
"usage: mktemp [-d] [-q] [-t prefix] [-u] template ...\n");
fprintf(stderr,
" mktemp [-d] [-q] [-u] -t prefix \n");
exit (1);
}