74 lines
2.6 KiB
Rust
74 lines
2.6 KiB
Rust
// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
// no-prefer-dynamic
|
|
// ignore-cross-compile
|
|
|
|
#![feature(rustc_private)]
|
|
|
|
extern crate rustc_back;
|
|
|
|
use std::env;
|
|
use std::fs;
|
|
use std::process;
|
|
use std::str;
|
|
use rustc_back::tempdir::TempDir;
|
|
|
|
fn main() {
|
|
// If we're the child, make sure we were invoked correctly
|
|
let args: Vec<String> = env::args().collect();
|
|
if args.len() > 1 && args[1] == "child" {
|
|
// FIXME: This should check the whole `args[0]` instead of just
|
|
// checking that it ends_with the executable name. This
|
|
// is needed because of Windows, which has a different behavior.
|
|
// See #15149 for more info.
|
|
return assert!(args[0].ends_with(&format!("mytest{}",
|
|
env::consts::EXE_SUFFIX)));
|
|
}
|
|
|
|
test();
|
|
}
|
|
|
|
fn test() {
|
|
// If we're the parent, copy our own binary to a new directory.
|
|
let my_path = env::current_exe().unwrap();
|
|
let my_dir = my_path.parent().unwrap();
|
|
|
|
let child_dir = TempDir::new_in(&my_dir, "issue-15140-child").unwrap();
|
|
let child_dir = child_dir.path();
|
|
|
|
let child_path = child_dir.join(&format!("mytest{}",
|
|
env::consts::EXE_SUFFIX));
|
|
fs::copy(&my_path, &child_path).unwrap();
|
|
|
|
// Append the new directory to our own PATH.
|
|
let path = {
|
|
let mut paths: Vec<_> = env::split_paths(&env::var_os("PATH").unwrap()).collect();
|
|
paths.push(child_dir.to_path_buf());
|
|
env::join_paths(paths).unwrap()
|
|
};
|
|
|
|
let child_output = process::Command::new("mytest").env("PATH", &path)
|
|
.arg("child")
|
|
.output().unwrap();
|
|
|
|
assert!(child_output.status.success(),
|
|
format!("child assertion failed\n child stdout:\n {}\n child stderr:\n {}",
|
|
str::from_utf8(&child_output.stdout).unwrap(),
|
|
str::from_utf8(&child_output.stderr).unwrap()));
|
|
|
|
let res = fs::remove_dir_all(&child_dir);
|
|
if res.is_err() {
|
|
// On Windows deleting just executed mytest.exe can fail because it's still locked
|
|
std::thread::sleep_ms(1000);
|
|
fs::remove_dir_all(&child_dir).unwrap();
|
|
}
|
|
}
|