src/libstd/sys/windows/process2.rs

   1 // Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 use prelude::v1::*;
  12
  13 use ascii::*;
  14 use collections::HashMap;
  15 use collections;
  16 use env;
  17 use ffi::{OsString, OsStr};
  18 use fmt;
  19 use fs;
  20 use io::{self, Error};
  21 use libc::{self, c_void};
  22 use os::windows::ffi::OsStrExt;
  23 use ptr;
  24 use sync::{StaticMutex, MUTEX_INIT};
  25 use sys::handle::Handle;
  26 use sys::pipe2::AnonPipe;
  27 use sys::{self, cvt};
  28 use sys_common::{AsInner, FromInner};
  29
  30 ////////////////////////////////////////////////////////////////////////////////
  31 // Command
  32 ////////////////////////////////////////////////////////////////////////////////
  33
  34 fn mk_key(s: &OsStr) -> OsString {
  35     FromInner::from_inner(sys::os_str::Buf {
  36         inner: s.as_inner().inner.to_ascii_uppercase()
  37     })
  38 }
  39
  40 #[derive(Clone)]
  41 pub struct Command {
  42     pub program: OsString,
  43     pub args: Vec<OsString>,
  44     pub env: Option<HashMap<OsString, OsString>>,
  45     pub cwd: Option<OsString>,
  46     pub detach: bool, // not currently exposed in std::process
  47 }
  48
  49 impl Command {
  50     pub fn new(program: &OsStr) -> Command {
  51         Command {
  52             program: program.to_os_string(),
  53             args: Vec::new(),
  54             env: None,
  55             cwd: None,
  56             detach: false,
  57         }
  58     }
  59
  60     pub fn arg(&mut self, arg: &OsStr) {
  61         self.args.push(arg.to_os_string())
  62     }
  63     pub fn args<'a, I: Iterator<Item = &'a OsStr>>(&mut self, args: I) {
  64         self.args.extend(args.map(OsStr::to_os_string))
  65     }
  66     fn init_env_map(&mut self){
  67         if self.env.is_none() {
  68             self.env = Some(env::vars_os().map(|(key, val)| {
  69                 (mk_key(&key), val)
  70             }).collect());
  71         }
  72     }
  73     pub fn env(&mut self, key: &OsStr, val: &OsStr) {
  74         self.init_env_map();
  75         self.env.as_mut().unwrap().insert(mk_key(key), val.to_os_string());
  76     }
  77     pub fn env_remove(&mut self, key: &OsStr) {
  78         self.init_env_map();
  79         self.env.as_mut().unwrap().remove(&mk_key(key));
  80     }
  81     pub fn env_clear(&mut self) {
  82         self.env = Some(HashMap::new())
  83     }
  84     pub fn cwd(&mut self, dir: &OsStr) {
  85         self.cwd = Some(dir.to_os_string())
  86     }
  87 }
  88
  89 ////////////////////////////////////////////////////////////////////////////////
  90 // Processes
  91 ////////////////////////////////////////////////////////////////////////////////
  92
  93 // `CreateProcess` is racy!
  94 // http://support.microsoft.com/kb/315939
  95 static CREATE_PROCESS_LOCK: StaticMutex = MUTEX_INIT;
  96
  97 /// A value representing a child process.
  98 ///
  99 /// The lifetime of this value is linked to the lifetime of the actual
 100 /// process - the Process destructor calls self.finish() which waits
 101 /// for the process to terminate.
 102 pub struct Process {
 103     /// A HANDLE to the process, which will prevent the pid being
 104     /// re-used until the handle is closed.
 105     handle: Handle,
 106 }
 107
 108 pub enum Stdio {
 109     Inherit,
 110     Piped(AnonPipe),
 111     None,
 112 }
 113
 114 impl Process {
 115     #[allow(deprecated)]
 116     pub fn spawn(cfg: &Command,
 117                  in_fd: Stdio,
 118                  out_fd: Stdio,
 119                  err_fd: Stdio) -> io::Result<Process>
 120     {
 121         use libc::types::os::arch::extra::{DWORD, HANDLE, STARTUPINFO};
 122         use libc::consts::os::extra::{
 123             TRUE, FALSE,
 124             STARTF_USESTDHANDLES,
 125             INVALID_HANDLE_VALUE,
 126             DUPLICATE_SAME_ACCESS
 127         };
 128         use libc::funcs::extra::kernel32::{
 129             GetCurrentProcess,
 130             DuplicateHandle,
 131             CloseHandle,
 132             CreateProcessW
 133         };
 134
 135         use env::split_paths;
 136         use mem;
 137         use iter::Iterator;
 138
 139         // To have the spawning semantics of unix/windows stay the same, we need to
 140         // read the *child's* PATH if one is provided. See #15149 for more details.
 141         let program = cfg.env.as_ref().and_then(|env| {
 142             for (key, v) in env {
 143                 if OsStr::new("PATH") != &**key { continue }
 144
 145                 // Split the value and test each path to see if the
 146                 // program exists.
 147                 for path in split_paths(&v) {
 148                     let path = path.join(cfg.program.to_str().unwrap())
 149                                    .with_extension(env::consts::EXE_EXTENSION);
 150                     if fs::metadata(&path).is_ok() {
 151                         return Some(path.into_os_string())
 152                     }
 153                 }
 154                 break
 155             }
 156             None
 157         });
 158
 159         unsafe {
 160             let mut si = zeroed_startupinfo();
 161             si.cb = mem::size_of::<STARTUPINFO>() as DWORD;
 162             si.dwFlags = STARTF_USESTDHANDLES;
 163
 164             let cur_proc = GetCurrentProcess();
 165
 166             let set_fd = |fd: &Stdio, slot: &mut HANDLE,
 167                           is_stdin: bool| {
 168                 match *fd {
 169                     Stdio::Inherit => {}
 170
 171                     // Similarly to unix, we don't actually leave holes for the
 172                     // stdio file descriptors, but rather open up /dev/null
 173                     // equivalents. These equivalents are drawn from libuv's
 174                     // windows process spawning.
 175                     Stdio::None => {
 176                         let access = if is_stdin {
 177                             libc::FILE_GENERIC_READ
 178                         } else {
 179                             libc::FILE_GENERIC_WRITE | libc::FILE_READ_ATTRIBUTES
 180                         };
 181                         let size = mem::size_of::<libc::SECURITY_ATTRIBUTES>();
 182                         let mut sa = libc::SECURITY_ATTRIBUTES {
 183                             nLength: size as libc::DWORD,
 184                             lpSecurityDescriptor: ptr::null_mut(),
 185                             bInheritHandle: 1,
 186                         };
 187                         let mut filename: Vec<u16> = "NUL".utf16_units().collect();
 188                         filename.push(0);
 189                         *slot = libc::CreateFileW(filename.as_ptr(),
 190                                                   access,
 191                                                   libc::FILE_SHARE_READ |
 192                                                       libc::FILE_SHARE_WRITE,
 193                                                   &mut sa,
 194                                                   libc::OPEN_EXISTING,
 195                                                   0,
 196                                                   ptr::null_mut());
 197                         if *slot == INVALID_HANDLE_VALUE {
 198                             return Err(Error::last_os_error())
 199                         }
 200                     }
 201                     Stdio::Piped(ref pipe) => {
 202                         let orig = pipe.handle().raw();
 203                         if DuplicateHandle(cur_proc, orig, cur_proc, slot,
 204                                            0, TRUE, DUPLICATE_SAME_ACCESS) == FALSE {
 205                             return Err(Error::last_os_error())
 206                         }
 207                     }
 208                 }
 209                 Ok(())
 210             };
 211
 212             try!(set_fd(&in_fd, &mut si.hStdInput, true));
 213             try!(set_fd(&out_fd, &mut si.hStdOutput, false));
 214             try!(set_fd(&err_fd, &mut si.hStdError, false));
 215
 216             let mut cmd_str = make_command_line(program.as_ref().unwrap_or(&cfg.program),
 217                                             &cfg.args);
 218             cmd_str.push(0); // add null terminator
 219
 220             let mut pi = zeroed_process_information();
 221             let mut create_err = None;
 222
 223             // stolen from the libuv code.
 224             let mut flags = libc::CREATE_UNICODE_ENVIRONMENT;
 225             if cfg.detach {
 226                 flags |= libc::DETACHED_PROCESS | libc::CREATE_NEW_PROCESS_GROUP;
 227             }
 228
 229             with_envp(cfg.env.as_ref(), |envp| {
 230                 with_dirp(cfg.cwd.as_ref(), |dirp| {
 231                     let _lock = CREATE_PROCESS_LOCK.lock().unwrap();
 232                     let created = CreateProcessW(ptr::null(),
 233                                                  cmd_str.as_mut_ptr(),
 234                                                  ptr::null_mut(),
 235                                                  ptr::null_mut(),
 236                                                  TRUE,
 237                                                  flags, envp, dirp,
 238                                                  &mut si, &mut pi);
 239                     if created == FALSE {
 240                         create_err = Some(Error::last_os_error());
 241                     }
 242                 })
 243             });
 244
 245             if !in_fd.inherited() {
 246                 assert!(CloseHandle(si.hStdInput) != 0);
 247             }
 248             if !out_fd.inherited() {
 249                 assert!(CloseHandle(si.hStdOutput) != 0);
 250             }
 251             if !err_fd.inherited() {
 252                 assert!(CloseHandle(si.hStdError) != 0);
 253             }
 254
 255             match create_err {
 256                 Some(err) => return Err(err),
 257                 None => {}
 258             }
 259
 260             // We close the thread handle because we don't care about keeping the
 261             // thread id valid, and we aren't keeping the thread handle around to be
 262             // able to close it later. We don't close the process handle however
 263             // because std::we want the process id to stay valid at least until the
 264             // calling code closes the process handle.
 265             assert!(CloseHandle(pi.hThread) != 0);
 266
 267             Ok(Process {
 268                 handle: Handle::new(pi.hProcess)
 269             })
 270         }
 271     }
 272
 273     pub unsafe fn kill(&self) -> io::Result<()> {
 274         try!(cvt(libc::TerminateProcess(self.handle.raw(), 1)));
 275         Ok(())
 276     }
 277
 278     pub fn wait(&self) -> io::Result<ExitStatus> {
 279         use libc::consts::os::extra::{
 280             FALSE,
 281             STILL_ACTIVE,
 282             INFINITE,
 283             WAIT_OBJECT_0,
 284         };
 285         use libc::funcs::extra::kernel32::{
 286             GetExitCodeProcess,
 287             WaitForSingleObject,
 288         };
 289
 290         unsafe {
 291             loop {
 292                 let mut status = 0;
 293                 if GetExitCodeProcess(self.handle.raw(), &mut status) == FALSE {
 294                     let err = Err(Error::last_os_error());
 295                     return err;
 296                 }
 297                 if status != STILL_ACTIVE {
 298                     return Ok(ExitStatus(status as i32));
 299                 }
 300                 match WaitForSingleObject(self.handle.raw(), INFINITE) {
 301                     WAIT_OBJECT_0 => {}
 302                     _ => {
 303                         let err = Err(Error::last_os_error());
 304                         return err
 305                     }
 306                 }
 307             }
 308         }
 309     }
 310 }
 311
 312 impl Stdio {
 313     fn inherited(&self) -> bool {
 314         match *self { Stdio::Inherit => true, _ => false }
 315     }
 316 }
 317
 318 #[derive(PartialEq, Eq, Clone, Copy, Debug)]
 319 pub struct ExitStatus(i32);
 320
 321 impl ExitStatus {
 322     pub fn success(&self) -> bool {
 323         self.0 == 0
 324     }
 325     pub fn code(&self) -> Option<i32> {
 326         Some(self.0)
 327     }
 328 }
 329
 330 impl fmt::Display for ExitStatus {
 331     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
 332         write!(f, "exit code: {}", self.0)
 333     }
 334 }
 335
 336 fn zeroed_startupinfo() -> libc::types::os::arch::extra::STARTUPINFO {
 337     libc::types::os::arch::extra::STARTUPINFO {
 338         cb: 0,
 339         lpReserved: ptr::null_mut(),
 340         lpDesktop: ptr::null_mut(),
 341         lpTitle: ptr::null_mut(),
 342         dwX: 0,
 343         dwY: 0,
 344         dwXSize: 0,
 345         dwYSize: 0,
 346         dwXCountChars: 0,
 347         dwYCountCharts: 0,
 348         dwFillAttribute: 0,
 349         dwFlags: 0,
 350         wShowWindow: 0,
 351         cbReserved2: 0,
 352         lpReserved2: ptr::null_mut(),
 353         hStdInput: libc::INVALID_HANDLE_VALUE,
 354         hStdOutput: libc::INVALID_HANDLE_VALUE,
 355         hStdError: libc::INVALID_HANDLE_VALUE,
 356     }
 357 }
 358
 359 fn zeroed_process_information() -> libc::types::os::arch::extra::PROCESS_INFORMATION {
 360     libc::types::os::arch::extra::PROCESS_INFORMATION {
 361         hProcess: ptr::null_mut(),
 362         hThread: ptr::null_mut(),
 363         dwProcessId: 0,
 364         dwThreadId: 0
 365     }
 366 }
 367
 368 // Produces a wide string *without terminating null*
 369 fn make_command_line(prog: &OsStr, args: &[OsString]) -> Vec<u16> {
 370     // Encode the command and arguments in a command line string such
 371     // that the spawned process may recover them using CommandLineToArgvW.
 372     let mut cmd: Vec<u16> = Vec::new();
 373     append_arg(&mut cmd, prog);
 374     for arg in args {
 375         cmd.push(' ' as u16);
 376         append_arg(&mut cmd, arg);
 377     }
 378     return cmd;
 379
 380     fn append_arg(cmd: &mut Vec<u16>, arg: &OsStr) {
 381         // If an argument has 0 characters then we need to quote it to ensure
 382         // that it actually gets passed through on the command line or otherwise
 383         // it will be dropped entirely when parsed on the other end.
 384         let arg_bytes = &arg.as_inner().inner.as_inner();
 385         let quote = arg_bytes.iter().any(|c| *c == b' ' || *c == b'\t')
 386             || arg_bytes.is_empty();
 387         if quote {
 388             cmd.push('"' as u16);
 389         }
 390
 391         let mut iter = arg.encode_wide();
 392         let mut backslashes: usize = 0;
 393         while let Some(x) = iter.next() {
 394             if x == '\\' as u16 {
 395                 backslashes += 1;
 396             } else {
 397                 if x == '"' as u16 {
 398                     // Add n+1 backslashes to total 2n+1 before internal '"'.
 399                     for _ in 0..(backslashes+1) {
 400                         cmd.push('\\' as u16);
 401                     }
 402                 }
 403                 backslashes = 0;
 404             }
 405             cmd.push(x);
 406         }
 407
 408         if quote {
 409             // Add n backslashes to total 2n before ending '"'.
 410             for _ in 0..backslashes {
 411                 cmd.push('\\' as u16);
 412             }
 413             cmd.push('"' as u16);
 414         }
 415     }
 416 }
 417
 418 fn with_envp<F, T>(env: Option<&collections::HashMap<OsString, OsString>>, cb: F) -> T
 419     where F: FnOnce(*mut c_void) -> T,
 420 {
 421     // On Windows we pass an "environment block" which is not a char**, but
 422     // rather a concatenation of null-terminated k=v\0 sequences, with a final
 423     // \0 to terminate.
 424     match env {
 425         Some(env) => {
 426             let mut blk = Vec::new();
 427
 428             for pair in env {
 429                 blk.extend(pair.0.encode_wide());
 430                 blk.push('=' as u16);
 431                 blk.extend(pair.1.encode_wide());
 432                 blk.push(0);
 433             }
 434             blk.push(0);
 435             cb(blk.as_mut_ptr() as *mut c_void)
 436         }
 437         _ => cb(ptr::null_mut())
 438     }
 439 }
 440
 441 fn with_dirp<T, F>(d: Option<&OsString>, cb: F) -> T where
 442     F: FnOnce(*const u16) -> T,
 443 {
 444     match d {
 445       Some(dir) => {
 446           let mut dir_str: Vec<u16> = dir.encode_wide().collect();
 447           dir_str.push(0);
 448           cb(dir_str.as_ptr())
 449       },
 450       None => cb(ptr::null())
 451     }
 452 }
 453
 454 #[cfg(test)]
 455 mod tests {
 456     use prelude::v1::*;
 457     use str;
 458     use ffi::{OsStr, OsString};
 459     use super::make_command_line;
 460
 461     #[test]
 462     fn test_make_command_line() {
 463         fn test_wrapper(prog: &str, args: &[&str]) -> String {
 464             String::from_utf16(
 465                 &make_command_line(OsStr::new(prog),
 466                                    &args.iter()
 467                                         .map(|a| OsString::from(a))
 468                                         .collect::<Vec<OsString>>())).unwrap()
 469         }
 470
 471         assert_eq!(
 472             test_wrapper("prog", &["aaa", "bbb", "ccc"]),
 473             "prog aaa bbb ccc"
 474         );
 475
 476         assert_eq!(
 477             test_wrapper("C:\\Program Files\\blah\\blah.exe", &["aaa"]),
 478             "\"C:\\Program Files\\blah\\blah.exe\" aaa"
 479         );
 480         assert_eq!(
 481             test_wrapper("C:\\Program Files\\test", &["aa\"bb"]),
 482             "\"C:\\Program Files\\test\" aa\\\"bb"
 483         );
 484         assert_eq!(
 485             test_wrapper("echo", &["a b c"]),
 486             "echo \"a b c\""
 487         );
 488         assert_eq!(
 489             test_wrapper("echo", &["\" \\\" \\", "\\"]),
 490             "echo \"\\\" \\\\\\\" \\\\\" \\"
 491         );
 492         assert_eq!(
 493             test_wrapper("\u{03c0}\u{042f}\u{97f3}\u{00e6}\u{221e}", &[]),
 494             "\u{03c0}\u{042f}\u{97f3}\u{00e6}\u{221e}"
 495         );
 496     }
 497 }