sbase/mktemp.c

77 lines
1.4 KiB
C
Raw Normal View History

2013-11-12 11:52:28 +00:00
/* See LICENSE file for copyright and license details. */
#include <libgen.h>
2013-11-12 11:52:28 +00:00
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
2013-11-12 11:52:28 +00:00
#include "util.h"
static void
usage(void)
{
2013-11-13 12:10:49 +00:00
eprintf("usage: %s [-dq] [template]\n", argv0);
2013-11-12 11:52:28 +00:00
}
static int dflag = 0;
2013-11-13 12:10:49 +00:00
static int qflag = 0;
2013-11-12 11:52:28 +00:00
int
main(int argc, char *argv[])
{
char *template = "tmp.XXXXXXXXXX";
2013-11-14 19:46:21 +00:00
char *tmpdir = "/tmp", *p;
char path[PATH_MAX], tmp[PATH_MAX];
int fd;
2013-11-12 11:52:28 +00:00
ARGBEGIN {
case 'd':
dflag = 1;
break;
2013-11-13 12:10:49 +00:00
case 'q':
qflag = 1;
break;
2013-11-12 11:52:28 +00:00
default:
usage();
} ARGEND;
if (argc > 1)
usage();
else if (argc == 1)
template = argv[0];
2013-11-14 19:46:21 +00:00
if ((p = getenv("TMPDIR")))
tmpdir = p;
if (strlcpy(tmp, template, sizeof(tmp)) >= sizeof(tmp))
eprintf("path too long\n");
p = dirname(tmp);
if (p[0] != '.') {
if (strlcpy(path, template, sizeof(path)) >= sizeof(path))
eprintf("path too long\n");
} else {
if (strlcpy(path, tmpdir, sizeof(path)) >= sizeof(path))
eprintf("path too long\n");
if (strlcat(path, "/", sizeof(path)) >= sizeof(path))
eprintf("path too long\n");
if (strlcat(path, template, sizeof(path)) >= sizeof(path))
eprintf("path too long\n");
}
2013-11-12 11:52:28 +00:00
if (dflag) {
if (!mkdtemp(path)) {
2013-11-13 12:10:49 +00:00
if (!qflag)
eprintf("mkdtemp %s:", path);
2014-10-02 22:46:04 +00:00
exit(1);
2013-11-13 12:10:49 +00:00
}
2013-11-12 11:52:28 +00:00
} else {
if ((fd = mkstemp(path)) < 0) {
2013-11-13 12:10:49 +00:00
if (!qflag)
eprintf("mkstemp %s:", path);
2014-10-02 22:46:04 +00:00
exit(1);
2013-11-13 12:10:49 +00:00
}
2013-11-12 11:52:28 +00:00
close(fd);
}
puts(path);
2014-10-02 22:46:04 +00:00
return 0;
2013-11-12 11:52:28 +00:00
}