source: MondoRescue/branches/3.3/mindi-busybox/debianutils/mktemp.c@ 3865

/* 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]
 */
//config:config MKTEMP
//config:   bool "mktemp"
//config:   default y
//config:   help
//config:     mktemp is used to create unique temporary files

//applet:IF_MKTEMP(APPLET(mktemp, BB_DIR_BIN, BB_SUID_DROP))

//kbuild:lib-$(CONFIG_MKTEMP) += mktemp.o

//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"
//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";

    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 0
    /* Don't allow directory separator in template */
    if ((opts & OPT_t) && bb_basename(chp) != chp) {
        errno = EINVAL;
        goto error;
    }
#endif
    if (opts & (OPT_t|OPT_p))
        chp = concat_path_file(path, chp);

    if (opts & OPT_u) {
        chp = mktemp(chp);
        if (chp[0] == '\0')
            goto error;
    } else if (opts & OPT_d) {
        if (mkdtemp(chp) == NULL)
            goto error;
    } else {
        if (mkstemp(chp) < 0)
            goto error;
    }
    puts(chp);
    return EXIT_SUCCESS;
 error:
    if (opts & OPT_q)
        return EXIT_FAILURE;
    /* don't use chp as it gets mangled in case of error */
    bb_perror_nomsg_and_die();
}