blob: 8f9479fe8ed1d4e8d7ee971f78737632ddc9ce79
| 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 /data/local/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 | #ifdef __BIONIC__ |
| 66 | #include <android.h> |
| 67 | #define mktemp(s) bb_mktemp(s) |
| 68 | #endif |
| 69 | |
| 70 | int mktemp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
| 71 | int mktemp_main(int argc UNUSED_PARAM, char **argv) |
| 72 | { |
| 73 | const char *path; |
| 74 | char *chp; |
| 75 | unsigned opts; |
| 76 | enum { |
| 77 | OPT_d = 1 << 0, |
| 78 | OPT_q = 1 << 1, |
| 79 | OPT_t = 1 << 2, |
| 80 | OPT_p = 1 << 3, |
| 81 | OPT_u = 1 << 4, |
| 82 | }; |
| 83 | |
| 84 | path = getenv("TMPDIR"); |
| 85 | if (!path || path[0] == '\0') |
| 86 | path = "/data/local/tmp"; |
| 87 | |
| 88 | opt_complementary = "?1"; /* 1 argument max */ |
| 89 | opts = getopt32(argv, "dqtp:u", &path); |
| 90 | |
| 91 | chp = argv[optind]; |
| 92 | if (!chp) { |
| 93 | /* GNU coreutils 8.4: |
| 94 | * bare "mktemp" -> "mktemp -t tmp.XXXXXX" |
| 95 | */ |
| 96 | chp = xstrdup("tmp.XXXXXX"); |
| 97 | opts |= OPT_t; |
| 98 | } |
| 99 | #if 0 |
| 100 | /* Don't allow directory separator in template */ |
| 101 | if ((opts & OPT_t) && bb_basename(chp) != chp) { |
| 102 | errno = EINVAL; |
| 103 | goto error; |
| 104 | } |
| 105 | #endif |
| 106 | if (opts & (OPT_t|OPT_p)) |
| 107 | chp = concat_path_file(path, chp); |
| 108 | |
| 109 | if (opts & OPT_u) { |
| 110 | chp = mktemp(chp); |
| 111 | /*chp = mkstemp(chp);*/ |
| 112 | if (chp[0] == '\0') |
| 113 | goto error; |
| 114 | } else if (opts & OPT_d) { |
| 115 | if (mkdtemp(chp) == NULL) |
| 116 | goto error; |
| 117 | } else { |
| 118 | if (mkstemp(chp) < 0) |
| 119 | goto error; |
| 120 | } |
| 121 | puts(chp); |
| 122 | return EXIT_SUCCESS; |
| 123 | error: |
| 124 | if (opts & OPT_q) |
| 125 | return EXIT_FAILURE; |
| 126 | /* don't use chp as it gets mangled in case of error */ |
| 127 | bb_perror_nomsg_and_die(); |
| 128 | } |
| 129 |