diff options
| -rw-r--r-- | toys/lsb/mktemp.c | 37 |
1 files changed, 20 insertions, 17 deletions
diff --git a/toys/lsb/mktemp.c b/toys/lsb/mktemp.c index c1175fec..52e53ee6 100644 --- a/toys/lsb/mktemp.c +++ b/toys/lsb/mktemp.c @@ -12,8 +12,8 @@ config MKTEMP help usage: mktemp [-dq] [-p DIR] [TEMPLATE] - Safely create new file and print its name. Default TEMPLATE is - /tmp/tmp.XXXXXX and each trailing X is replaced with random char. + Safely create a new file and print its name. The default TEMPLATE is + tmp.XXXXXX. The default DIR is $TMPDIR, or /tmp if $TMPDIR is not set. -d, --directory Create directory instead of file -p DIR, --tmpdir=DIR Put new file in DIR @@ -29,24 +29,27 @@ GLOBALS( void mktemp_main(void) { - int d_flag = toys.optflags & FLAG_d; - char *tmp; + int d_flag = toys.optflags & FLAG_d; + char *template = *toys.optargs; + int success; - tmp = *toys.optargs; - - if (!tmp) { - if (!TT.tmpdir) TT.tmpdir = "/tmp"; - tmp = "tmp.xxxxxx"; + if (!template) { + template = "tmp.XXXXXX"; } - if (TT.tmpdir) tmp = xmprintf("%s/%s", TT.tmpdir ? TT.tmpdir : "/tmp", - *toys.optargs ? *toys.optargs : "tmp.XXXXXX"); - if (d_flag ? mkdtemp(tmp) == NULL : mkstemp(tmp) == -1) - if (toys.optflags & FLAG_q) - perror_exit("Failed to create temporary %s", - d_flag ? "directory" : "file"); + if (!TT.tmpdir) TT.tmpdir = getenv("TMPDIR"); + if (!TT.tmpdir) TT.tmpdir = "/tmp"; + + snprintf(toybuf, sizeof(toybuf), "%s/%s", TT.tmpdir, template); - xputs(tmp); + if (d_flag ? mkdtemp(toybuf) == NULL : mkstemp(toybuf) == -1) { + if (toys.optflags & FLAG_q) { + toys.exitval = 1; + } else { + perror_exit("Failed to create temporary %s with template %s/%s", + d_flag ? "directory" : "file", TT.tmpdir, template); + } + } - if (CFG_TOYBOX_FREE && TT.tmpdir) free(tmp); + xputs(toybuf); } |