blob: 10dd0de6368a508cc0dede029995641d14768489
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 | |
34 | //usage:#define mktemp_trivial_usage |
35 | //usage: "[-dt] [-p DIR] [TEMPLATE]" |
36 | //usage:#define mktemp_full_usage "\n\n" |
37 | //usage: "Create a temporary file with name based on TEMPLATE and print its name.\n" |
38 | //usage: "TEMPLATE must end with XXXXXX (e.g. [/dir/]nameXXXXXX).\n" |
39 | //usage: "Without TEMPLATE, -t tmp.XXXXXX is assumed.\n" |
40 | //usage: "\n -d Make directory, not file" |
41 | //usage: "\n -q Fail silently on errors" |
42 | //usage: "\n -t Prepend base directory name to TEMPLATE" |
43 | //usage: "\n -p DIR Use DIR as a base directory (implies -t)" |
44 | //usage: "\n -u Do not create anything; print a name" |
45 | //usage: "\n" |
46 | //usage: "\nBase directory is: -p DIR, else $TMPDIR, else /data/local/tmp" |
47 | //usage: |
48 | //usage:#define mktemp_example_usage |
49 | //usage: "$ mktemp /tmp/temp.XXXXXX\n" |
50 | //usage: "/tmp/temp.mWiLjM\n" |
51 | //usage: "$ ls -la /tmp/temp.mWiLjM\n" |
52 | //usage: "-rw------- 1 andersen andersen 0 Apr 25 17:10 /tmp/temp.mWiLjM\n" |
53 | |
54 | #include "libbb.h" |
55 | |
56 | #ifdef __BIONIC__ |
57 | #define mktemp(s) bb_mktemp(s) |
58 | #endif |
59 | |
60 | int mktemp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
61 | int mktemp_main(int argc UNUSED_PARAM, char **argv) |
62 | { |
63 | const char *path; |
64 | char *chp; |
65 | unsigned opts; |
66 | enum { |
67 | OPT_d = 1 << 0, |
68 | OPT_q = 1 << 1, |
69 | OPT_t = 1 << 2, |
70 | OPT_p = 1 << 3, |
71 | OPT_u = 1 << 4, |
72 | }; |
73 | |
74 | path = getenv("TMPDIR"); |
75 | if (!path || path[0] == '\0') |
76 | path = "/data/local/tmp"; |
77 | |
78 | opt_complementary = "?1"; /* 1 argument max */ |
79 | opts = getopt32(argv, "dqtp:u", &path); |
80 | |
81 | chp = argv[optind]; |
82 | if (!chp) { |
83 | /* GNU coreutils 8.4: |
84 | * bare "mktemp" -> "mktemp -t tmp.XXXXXX" |
85 | */ |
86 | chp = xstrdup("tmp.XXXXXX"); |
87 | opts |= OPT_t; |
88 | } |
89 | #if 0 |
90 | /* Don't allow directory separator in template */ |
91 | if ((opts & OPT_t) && bb_basename(chp) != chp) { |
92 | errno = EINVAL; |
93 | goto error; |
94 | } |
95 | #endif |
96 | if (opts & (OPT_t|OPT_p)) |
97 | chp = concat_path_file(path, chp); |
98 | |
99 | if (opts & OPT_u) { |
100 | chp = mktemp(chp); |
101 | /*chp = mkstemp(chp);*/ |
102 | if (chp[0] == '\0') |
103 | goto error; |
104 | } else if (opts & OPT_d) { |
105 | if (mkdtemp(chp) == NULL) |
106 | goto error; |
107 | } else { |
108 | if (mkstemp(chp) < 0) |
109 | goto error; |
110 | } |
111 | puts(chp); |
112 | return EXIT_SUCCESS; |
113 | error: |
114 | if (opts & OPT_q) |
115 | return EXIT_FAILURE; |
116 | /* don't use chp as it gets mangled in case of error */ |
117 | bb_perror_nomsg_and_die(); |
118 | } |
119 |