| 1 | /* vi: set sw=4 ts=4: */
|
|---|
| 2 | /*
|
|---|
| 3 | * Mini mktemp implementation for busybox
|
|---|
| 4 | *
|
|---|
| 5 | *
|
|---|
| 6 | * Copyright (C) 2000 by Daniel Jacobowitz
|
|---|
| 7 | * Written by Daniel Jacobowitz <dan@debian.org>
|
|---|
| 8 | *
|
|---|
| 9 | * Licensed under GPLv2 or later, see file LICENSE in this source tree.
|
|---|
| 10 | */
|
|---|
| 11 |
|
|---|
| 12 | /* Coreutils 6.12 man page says:
|
|---|
| 13 | * mktemp [OPTION]... [TEMPLATE]
|
|---|
| 14 | * Create a temporary file or directory, safely, and print its name. If
|
|---|
| 15 | * TEMPLATE is not specified, use tmp.XXXXXXXXXX.
|
|---|
| 16 | * -d, --directory
|
|---|
| 17 | * create a directory, not a file
|
|---|
| 18 | * -q, --quiet
|
|---|
| 19 | * suppress diagnostics about file/dir-creation failure
|
|---|
| 20 | * -u, --dry-run
|
|---|
| 21 | * do not create anything; merely print a name (unsafe)
|
|---|
| 22 | * --tmpdir[=DIR]
|
|---|
| 23 | * interpret TEMPLATE relative to DIR. If DIR is not specified,
|
|---|
| 24 | * use $TMPDIR if set, else /tmp. With this option, TEMPLATE must
|
|---|
| 25 | * not be an absolute name. Unlike with -t, TEMPLATE may contain
|
|---|
| 26 | * slashes, but even here, mktemp still creates only the final com-
|
|---|
| 27 | * ponent.
|
|---|
| 28 | * -p DIR use DIR as a prefix; implies -t [deprecated]
|
|---|
| 29 | * -t interpret TEMPLATE as a single file name component, relative to
|
|---|
| 30 | * a directory: $TMPDIR, if set; else the directory specified via
|
|---|
| 31 | * -p; else /tmp [deprecated]
|
|---|
| 32 | */
|
|---|
| 33 | //config:config MKTEMP
|
|---|
| 34 | //config: bool "mktemp"
|
|---|
| 35 | //config: default y
|
|---|
| 36 | //config: help
|
|---|
| 37 | //config: mktemp is used to create unique temporary files
|
|---|
| 38 |
|
|---|
| 39 | //applet:IF_MKTEMP(APPLET(mktemp, BB_DIR_BIN, BB_SUID_DROP))
|
|---|
| 40 |
|
|---|
| 41 | //kbuild:lib-$(CONFIG_MKTEMP) += mktemp.o
|
|---|
| 42 |
|
|---|
| 43 | //usage:#define mktemp_trivial_usage
|
|---|
| 44 | //usage: "[-dt] [-p DIR] [TEMPLATE]"
|
|---|
| 45 | //usage:#define mktemp_full_usage "\n\n"
|
|---|
| 46 | //usage: "Create a temporary file with name based on TEMPLATE and print its name.\n"
|
|---|
| 47 | //usage: "TEMPLATE must end with XXXXXX (e.g. [/dir/]nameXXXXXX).\n"
|
|---|
| 48 | //usage: "Without TEMPLATE, -t tmp.XXXXXX is assumed.\n"
|
|---|
| 49 | //usage: "\n -d Make directory, not file"
|
|---|
| 50 | //usage: "\n -q Fail silently on errors"
|
|---|
| 51 | //usage: "\n -t Prepend base directory name to TEMPLATE"
|
|---|
| 52 | //usage: "\n -p DIR Use DIR as a base directory (implies -t)"
|
|---|
| 53 | //usage: "\n -u Do not create anything; print a name"
|
|---|
| 54 | //usage: "\n"
|
|---|
| 55 | //usage: "\nBase directory is: -p DIR, else $TMPDIR, else /tmp"
|
|---|
| 56 | //usage:
|
|---|
| 57 | //usage:#define mktemp_example_usage
|
|---|
| 58 | //usage: "$ mktemp /tmp/temp.XXXXXX\n"
|
|---|
| 59 | //usage: "/tmp/temp.mWiLjM\n"
|
|---|
| 60 | //usage: "$ ls -la /tmp/temp.mWiLjM\n"
|
|---|
| 61 | //usage: "-rw------- 1 andersen andersen 0 Apr 25 17:10 /tmp/temp.mWiLjM\n"
|
|---|
| 62 |
|
|---|
| 63 | #include "libbb.h"
|
|---|
| 64 |
|
|---|
| 65 | int mktemp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
|
|---|
| 66 | int mktemp_main(int argc UNUSED_PARAM, char **argv)
|
|---|
| 67 | {
|
|---|
| 68 | const char *path;
|
|---|
| 69 | char *chp;
|
|---|
| 70 | unsigned opts;
|
|---|
| 71 | enum {
|
|---|
| 72 | OPT_d = 1 << 0,
|
|---|
| 73 | OPT_q = 1 << 1,
|
|---|
| 74 | OPT_t = 1 << 2,
|
|---|
| 75 | OPT_p = 1 << 3,
|
|---|
| 76 | OPT_u = 1 << 4,
|
|---|
| 77 | };
|
|---|
| 78 |
|
|---|
| 79 | path = getenv("TMPDIR");
|
|---|
| 80 | if (!path || path[0] == '\0')
|
|---|
| 81 | path = "/tmp";
|
|---|
| 82 |
|
|---|
| 83 | opt_complementary = "?1"; /* 1 argument max */
|
|---|
| 84 | opts = getopt32(argv, "dqtp:u", &path);
|
|---|
| 85 |
|
|---|
| 86 | chp = argv[optind];
|
|---|
| 87 | if (!chp) {
|
|---|
| 88 | /* GNU coreutils 8.4:
|
|---|
| 89 | * bare "mktemp" -> "mktemp -t tmp.XXXXXX"
|
|---|
| 90 | */
|
|---|
| 91 | chp = xstrdup("tmp.XXXXXX");
|
|---|
| 92 | opts |= OPT_t;
|
|---|
| 93 | }
|
|---|
| 94 | #if 0
|
|---|
| 95 | /* Don't allow directory separator in template */
|
|---|
| 96 | if ((opts & OPT_t) && bb_basename(chp) != chp) {
|
|---|
| 97 | errno = EINVAL;
|
|---|
| 98 | goto error;
|
|---|
| 99 | }
|
|---|
| 100 | #endif
|
|---|
| 101 | if (opts & (OPT_t|OPT_p))
|
|---|
| 102 | chp = concat_path_file(path, chp);
|
|---|
| 103 |
|
|---|
| 104 | if (opts & OPT_u) {
|
|---|
| 105 | chp = mktemp(chp);
|
|---|
| 106 | if (chp[0] == '\0')
|
|---|
| 107 | goto error;
|
|---|
| 108 | } else if (opts & OPT_d) {
|
|---|
| 109 | if (mkdtemp(chp) == NULL)
|
|---|
| 110 | goto error;
|
|---|
| 111 | } else {
|
|---|
| 112 | if (mkstemp(chp) < 0)
|
|---|
| 113 | goto error;
|
|---|
| 114 | }
|
|---|
| 115 | puts(chp);
|
|---|
| 116 | return EXIT_SUCCESS;
|
|---|
| 117 | error:
|
|---|
| 118 | if (opts & OPT_q)
|
|---|
| 119 | return EXIT_FAILURE;
|
|---|
| 120 | /* don't use chp as it gets mangled in case of error */
|
|---|
| 121 | bb_perror_nomsg_and_die();
|
|---|
| 122 | }
|
|---|
