Skip to content

Commit

Permalink
Merge pull request #4878 from cakebaker/mktemp_4874
Browse files Browse the repository at this point in the history
mktemp: prioritize TMPDIR over -p when using -t
  • Loading branch information
sylvestre authored May 22, 2023
2 parents c915207 + 698fafa commit e5419ad
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 1 deletion.
9 changes: 8 additions & 1 deletion src/uu/mktemp/src/mktemp.rs
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,11 @@ static OPT_T: &str = "t";

static ARG_TEMPLATE: &str = "template";

#[cfg(not(windows))]
const TMPDIR_ENV_VAR: &str = "TMPDIR";
#[cfg(windows)]
const TMPDIR_ENV_VAR: &str = "TMP";

#[derive(Debug)]
enum MkTempError {
PersistError(PathBuf),
Expand Down Expand Up @@ -191,7 +196,9 @@ impl Options {
(tmpdir, template.to_string())
}
Some(template) => {
let tmpdir = if matches.contains_id(OPT_TMPDIR) {
let tmpdir = if env::var(TMPDIR_ENV_VAR).is_ok() && matches.get_flag(OPT_T) {
env::var(TMPDIR_ENV_VAR).ok()
} else if matches.contains_id(OPT_TMPDIR) {
matches.get_one::<String>(OPT_TMPDIR).map(String::from)
} else if matches.get_flag(OPT_T) {
// mktemp -t foo.xxx should export in TMPDIR
Expand Down
17 changes: 17 additions & 0 deletions tests/by-util/test_mktemp.rs
Original file line number Diff line number Diff line change
Expand Up @@ -884,3 +884,20 @@ fn test_default_issue_4821_t_tmpdir_p() {
println!("stdout = {stdout}");
assert!(stdout.contains(&pathname));
}

#[test]
fn test_t_ensure_tmpdir_has_higher_priority_than_p() {
let scene = TestScenario::new(util_name!());
let pathname = scene.fixtures.as_string();
let result = scene
.ucmd()
.env(TMPDIR, &pathname)
.arg("-t")
.arg("-p")
.arg("should_not_attempt_to_write_in_this_nonexisting_dir")
.arg("foo.XXXX")
.succeeds();
let stdout = result.stdout_str();
println!("stdout = {stdout}");
assert!(stdout.contains(&pathname));
}

0 comments on commit e5419ad

Please sign in to comment.