busybox/debianutils/mktemp.c
Denys Vlasenko 75e1e7b3d5 mktemp: add support for -u
zlib-1.2.6 Makefile uses "mktemp -u".

function                                             old     new   delta
___path_search                                         -     266    +266
mktemp_main                                          165     250     +85
tempnam                                                -      79     +79
packed_usage                                       29189   29176     -13
------------------------------------------------------------------------------
(add/remove: 3/0 grow/shrink: 1/1 up/down: 430/-13)           Total: 417 bytes

Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
2012-04-17 16:00:20 +02:00

116 lines
3.4 KiB
C

/* vi: set sw=4 ts=4: */
/*
* Mini mktemp implementation for busybox
*
*
* Copyright (C) 2000 by Daniel Jacobowitz
* Written by Daniel Jacobowitz <dan@debian.org>
*
* Licensed under GPLv2 or later, see file LICENSE in this source tree.
*/
/* Coreutils 6.12 man page says:
* mktemp [OPTION]... [TEMPLATE]
* Create a temporary file or directory, safely, and print its name. If
* TEMPLATE is not specified, use tmp.XXXXXXXXXX.
* -d, --directory
* create a directory, not a file
* -q, --quiet
* suppress diagnostics about file/dir-creation failure
* -u, --dry-run
* do not create anything; merely print a name (unsafe)
* --tmpdir[=DIR]
* interpret TEMPLATE relative to DIR. If DIR is not specified,
* use $TMPDIR if set, else /tmp. With this option, TEMPLATE must
* not be an absolute name. Unlike with -t, TEMPLATE may contain
* slashes, but even here, mktemp still creates only the final com-
* ponent.
* -p DIR use DIR as a prefix; implies -t [deprecated]
* -t interpret TEMPLATE as a single file name component, relative to
* a directory: $TMPDIR, if set; else the directory specified via
* -p; else /tmp [deprecated]
*/
//usage:#define mktemp_trivial_usage
//usage: "[-dt] [-p DIR] [TEMPLATE]"
//usage:#define mktemp_full_usage "\n\n"
//usage: "Create a temporary file with name based on TEMPLATE and print its name.\n"
//usage: "TEMPLATE must end with XXXXXX (e.g. [/dir/]nameXXXXXX).\n"
//usage: "Without TEMPLATE, -t tmp.XXXXXX is assumed.\n"
//usage: "\n -d Make directory, not file"
////usage: "\n -q Fail silently on errors" - we ignore this opt
//usage: "\n -t Prepend base directory name to TEMPLATE"
//usage: "\n -p DIR Use DIR as a base directory (implies -t)"
//usage: "\n -u Do not create anything; print a name"
//usage: "\n"
//usage: "\nBase directory is: -p DIR, else $TMPDIR, else /tmp"
//usage:
//usage:#define mktemp_example_usage
//usage: "$ mktemp /tmp/temp.XXXXXX\n"
//usage: "/tmp/temp.mWiLjM\n"
//usage: "$ ls -la /tmp/temp.mWiLjM\n"
//usage: "-rw------- 1 andersen andersen 0 Apr 25 17:10 /tmp/temp.mWiLjM\n"
#include "libbb.h"
int mktemp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int mktemp_main(int argc UNUSED_PARAM, char **argv)
{
const char *path;
char *chp;
unsigned opts;
enum {
OPT_d = 1 << 0,
OPT_q = 1 << 1,
OPT_t = 1 << 2,
OPT_p = 1 << 3,
OPT_u = 1 << 4,
};
path = getenv("TMPDIR");
if (!path || path[0] == '\0')
path = "/tmp";
/* -q is ignored */
opt_complementary = "?1"; /* 1 argument max */
opts = getopt32(argv, "dqtp:u", &path);
chp = argv[optind];
if (!chp) {
/* GNU coreutils 8.4:
* bare "mktemp" -> "mktemp -t tmp.XXXXXX"
*/
chp = xstrdup("tmp.XXXXXX");
opts |= OPT_t;
}
if (opts & OPT_u) {
/* Remove (up to) 6 X's */
unsigned len = strlen(chp);
int cnt = len > 6 ? 6 : len;
while (--cnt >= 0 && chp[--len] == 'X')
chp[len] = '\0';
chp = tempnam(opts & (OPT_t|OPT_p) ? path : "./", chp);
if (!chp)
return EXIT_FAILURE;
if (!(opts & (OPT_t|OPT_p)))
chp += 2;
goto ret;
}
if (opts & (OPT_t|OPT_p))
chp = concat_path_file(path, chp);
if (opts & OPT_d) {
if (mkdtemp(chp) == NULL)
return EXIT_FAILURE;
} else {
if (mkstemp(chp) < 0)
return EXIT_FAILURE;
}
ret:
puts(chp);
return EXIT_SUCCESS;
}