blob: 975a924aba340367d55f95b7e76ff89fae63304d [file] [log] [blame]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001# subprocess - Subprocesses with accessible I/O streams
2#
Tim Peterse718f612004-10-12 21:51:32 +00003# For more information about this module, see PEP 324.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00004#
Peter Astrandc26516b2005-02-21 08:13:02 +00005# This module should remain compatible with Python 2.2, see PEP 291.
6#
Peter Astrand3a708df2005-09-23 17:37:29 +00007# Copyright (c) 2003-2005 by Peter Astrand <astrand@lysator.liu.se>
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00008#
Peter Astrand69bf13f2005-02-14 08:56:32 +00009# Licensed to PSF under a Contributor Agreement.
Peter Astrand3a708df2005-09-23 17:37:29 +000010# See http://www.python.org/2.4/license for licensing details.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000011
Raymond Hettinger837dd932004-10-17 16:36:53 +000012r"""subprocess - Subprocesses with accessible I/O streams
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000013
Fredrik Lundh15aaacc2004-10-17 14:47:05 +000014This module allows you to spawn processes, connect to their
15input/output/error pipes, and obtain their return codes. This module
16intends to replace several other, older modules and functions, like:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000017
18os.system
19os.spawn*
20os.popen*
21popen2.*
22commands.*
23
24Information about how the subprocess module can be used to replace these
25modules and functions can be found below.
26
27
28
29Using the subprocess module
30===========================
31This module defines one class called Popen:
32
33class Popen(args, bufsize=0, executable=None,
34 stdin=None, stdout=None, stderr=None,
35 preexec_fn=None, close_fds=False, shell=False,
36 cwd=None, env=None, universal_newlines=False,
37 startupinfo=None, creationflags=0):
38
39
40Arguments are:
41
42args should be a string, or a sequence of program arguments. The
43program to execute is normally the first item in the args sequence or
44string, but can be explicitly set by using the executable argument.
45
46On UNIX, with shell=False (default): In this case, the Popen class
47uses os.execvp() to execute the child program. args should normally
48be a sequence. A string will be treated as a sequence with the string
49as the only item (the program to execute).
50
51On UNIX, with shell=True: If args is a string, it specifies the
52command string to execute through the shell. If args is a sequence,
53the first item specifies the command string, and any additional items
54will be treated as additional shell arguments.
55
56On Windows: the Popen class uses CreateProcess() to execute the child
57program, which operates on strings. If args is a sequence, it will be
58converted to a string using the list2cmdline method. Please note that
59not all MS Windows applications interpret the command line the same
60way: The list2cmdline is designed for applications using the same
61rules as the MS C runtime.
62
63bufsize, if given, has the same meaning as the corresponding argument
64to the built-in open() function: 0 means unbuffered, 1 means line
65buffered, any other positive value means use a buffer of
66(approximately) that size. A negative bufsize means to use the system
67default, which usually means fully buffered. The default value for
68bufsize is 0 (unbuffered).
69
70stdin, stdout and stderr specify the executed programs' standard
71input, standard output and standard error file handles, respectively.
72Valid values are PIPE, an existing file descriptor (a positive
73integer), an existing file object, and None. PIPE indicates that a
74new pipe to the child should be created. With None, no redirection
75will occur; the child's file handles will be inherited from the
76parent. Additionally, stderr can be STDOUT, which indicates that the
77stderr data from the applications should be captured into the same
78file handle as for stdout.
79
80If preexec_fn is set to a callable object, this object will be called
81in the child process just before the child is executed.
82
83If close_fds is true, all file descriptors except 0, 1 and 2 will be
84closed before the child process is executed.
85
86if shell is true, the specified command will be executed through the
87shell.
88
89If cwd is not None, the current directory will be changed to cwd
90before the child is executed.
91
92If env is not None, it defines the environment variables for the new
93process.
94
95If universal_newlines is true, the file objects stdout and stderr are
96opened as a text files, but lines may be terminated by any of '\n',
97the Unix end-of-line convention, '\r', the Macintosh convention or
98'\r\n', the Windows convention. All of these external representations
99are seen as '\n' by the Python program. Note: This feature is only
100available if Python is built with universal newline support (the
101default). Also, the newlines attribute of the file objects stdout,
102stdin and stderr are not updated by the communicate() method.
103
104The startupinfo and creationflags, if given, will be passed to the
105underlying CreateProcess() function. They can specify things such as
106appearance of the main window and priority for the new process.
107(Windows only)
108
109
110This module also defines two shortcut functions:
111
Peter Astrand5f5e1412004-12-05 20:15:36 +0000112call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000113 Run command with arguments. Wait for command to complete, then
114 return the returncode attribute.
115
116 The arguments are the same as for the Popen constructor. Example:
117
118 retcode = call(["ls", "-l"])
119
Peter Astrand454f7672005-01-01 09:36:35 +0000120check_call(*popenargs, **kwargs):
121 Run command with arguments. Wait for command to complete. If the
122 exit code was zero then return, otherwise raise
123 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000124 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000125
126 The arguments are the same as for the Popen constructor. Example:
127
128 check_call(["ls", "-l"])
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000129
130Exceptions
131----------
132Exceptions raised in the child process, before the new program has
133started to execute, will be re-raised in the parent. Additionally,
134the exception object will have one extra attribute called
135'child_traceback', which is a string containing traceback information
136from the childs point of view.
137
138The most common exception raised is OSError. This occurs, for
139example, when trying to execute a non-existent file. Applications
140should prepare for OSErrors.
141
142A ValueError will be raised if Popen is called with invalid arguments.
143
Peter Astrand7d1d4362006-07-14 14:04:45 +0000144check_call() will raise CalledProcessError, if the called process
145returns a non-zero return code.
Peter Astrand454f7672005-01-01 09:36:35 +0000146
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000147
148Security
149--------
150Unlike some other popen functions, this implementation will never call
151/bin/sh implicitly. This means that all characters, including shell
152metacharacters, can safely be passed to child processes.
153
154
155Popen objects
156=============
157Instances of the Popen class have the following methods:
158
159poll()
160 Check if child process has terminated. Returns returncode
161 attribute.
162
163wait()
164 Wait for child process to terminate. Returns returncode attribute.
165
166communicate(input=None)
167 Interact with process: Send data to stdin. Read data from stdout
168 and stderr, until end-of-file is reached. Wait for process to
Neal Norwitza186ee22006-12-29 03:01:53 +0000169 terminate. The optional input argument should be a string to be
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000170 sent to the child process, or None, if no data should be sent to
171 the child.
Tim Peterse718f612004-10-12 21:51:32 +0000172
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000173 communicate() returns a tuple (stdout, stderr).
174
175 Note: The data read is buffered in memory, so do not use this
176 method if the data size is large or unlimited.
177
178The following attributes are also available:
179
180stdin
181 If the stdin argument is PIPE, this attribute is a file object
182 that provides input to the child process. Otherwise, it is None.
183
184stdout
185 If the stdout argument is PIPE, this attribute is a file object
186 that provides output from the child process. Otherwise, it is
187 None.
188
189stderr
190 If the stderr argument is PIPE, this attribute is file object that
191 provides error output from the child process. Otherwise, it is
192 None.
193
194pid
195 The process ID of the child process.
196
197returncode
198 The child return code. A None value indicates that the process
199 hasn't terminated yet. A negative value -N indicates that the
200 child was terminated by signal N (UNIX only).
201
202
203Replacing older functions with the subprocess module
204====================================================
205In this section, "a ==> b" means that b can be used as a replacement
206for a.
207
208Note: All functions in this section fail (more or less) silently if
209the executed program cannot be found; this module raises an OSError
210exception.
211
212In the following examples, we assume that the subprocess module is
213imported with "from subprocess import *".
214
215
216Replacing /bin/sh shell backquote
217---------------------------------
218output=`mycmd myarg`
219==>
220output = Popen(["mycmd", "myarg"], stdout=PIPE).communicate()[0]
221
222
223Replacing shell pipe line
224-------------------------
225output=`dmesg | grep hda`
226==>
227p1 = Popen(["dmesg"], stdout=PIPE)
Peter Astrand6fdf3cb2004-11-30 18:06:42 +0000228p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000229output = p2.communicate()[0]
230
231
232Replacing os.system()
233---------------------
234sts = os.system("mycmd" + " myarg")
235==>
236p = Popen("mycmd" + " myarg", shell=True)
Neal Norwitz84404832006-07-10 00:05:34 +0000237pid, sts = os.waitpid(p.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000238
239Note:
240
241* Calling the program through the shell is usually not required.
242
243* It's easier to look at the returncode attribute than the
244 exitstatus.
245
246A more real-world example would look like this:
247
248try:
249 retcode = call("mycmd" + " myarg", shell=True)
250 if retcode < 0:
251 print >>sys.stderr, "Child was terminated by signal", -retcode
252 else:
253 print >>sys.stderr, "Child returned", retcode
254except OSError, e:
255 print >>sys.stderr, "Execution failed:", e
256
257
258Replacing os.spawn*
259-------------------
Tim Peterse718f612004-10-12 21:51:32 +0000260P_NOWAIT example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000261
262pid = os.spawnlp(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg")
263==>
264pid = Popen(["/bin/mycmd", "myarg"]).pid
265
266
267P_WAIT example:
268
269retcode = os.spawnlp(os.P_WAIT, "/bin/mycmd", "mycmd", "myarg")
270==>
271retcode = call(["/bin/mycmd", "myarg"])
272
273
Tim Peterse718f612004-10-12 21:51:32 +0000274Vector example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000275
276os.spawnvp(os.P_NOWAIT, path, args)
277==>
278Popen([path] + args[1:])
279
280
Tim Peterse718f612004-10-12 21:51:32 +0000281Environment example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000282
283os.spawnlpe(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg", env)
284==>
285Popen(["/bin/mycmd", "myarg"], env={"PATH": "/usr/bin"})
286
287
Tim Peterse718f612004-10-12 21:51:32 +0000288Replacing os.popen*
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000289-------------------
290pipe = os.popen(cmd, mode='r', bufsize)
291==>
292pipe = Popen(cmd, shell=True, bufsize=bufsize, stdout=PIPE).stdout
293
294pipe = os.popen(cmd, mode='w', bufsize)
295==>
296pipe = Popen(cmd, shell=True, bufsize=bufsize, stdin=PIPE).stdin
297
298
299(child_stdin, child_stdout) = os.popen2(cmd, mode, bufsize)
300==>
301p = Popen(cmd, shell=True, bufsize=bufsize,
302 stdin=PIPE, stdout=PIPE, close_fds=True)
303(child_stdin, child_stdout) = (p.stdin, p.stdout)
304
305
306(child_stdin,
307 child_stdout,
308 child_stderr) = os.popen3(cmd, mode, bufsize)
309==>
310p = Popen(cmd, shell=True, bufsize=bufsize,
311 stdin=PIPE, stdout=PIPE, stderr=PIPE, close_fds=True)
312(child_stdin,
313 child_stdout,
314 child_stderr) = (p.stdin, p.stdout, p.stderr)
315
316
317(child_stdin, child_stdout_and_stderr) = os.popen4(cmd, mode, bufsize)
318==>
319p = Popen(cmd, shell=True, bufsize=bufsize,
320 stdin=PIPE, stdout=PIPE, stderr=STDOUT, close_fds=True)
321(child_stdin, child_stdout_and_stderr) = (p.stdin, p.stdout)
322
323
324Replacing popen2.*
325------------------
326Note: If the cmd argument to popen2 functions is a string, the command
327is executed through /bin/sh. If it is a list, the command is directly
328executed.
329
330(child_stdout, child_stdin) = popen2.popen2("somestring", bufsize, mode)
331==>
332p = Popen(["somestring"], shell=True, bufsize=bufsize
333 stdin=PIPE, stdout=PIPE, close_fds=True)
334(child_stdout, child_stdin) = (p.stdout, p.stdin)
335
336
337(child_stdout, child_stdin) = popen2.popen2(["mycmd", "myarg"], bufsize, mode)
338==>
339p = Popen(["mycmd", "myarg"], bufsize=bufsize,
340 stdin=PIPE, stdout=PIPE, close_fds=True)
341(child_stdout, child_stdin) = (p.stdout, p.stdin)
342
Neal Norwitzaa87fb62007-05-11 06:23:01 +0000343The popen2.Popen3 and popen2.Popen4 basically works as subprocess.Popen,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000344except that:
345
346* subprocess.Popen raises an exception if the execution fails
347* the capturestderr argument is replaced with the stderr argument.
348* stdin=PIPE and stdout=PIPE must be specified.
349* popen2 closes all filedescriptors by default, but you have to specify
Tim Peterse718f612004-10-12 21:51:32 +0000350 close_fds=True with subprocess.Popen.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000351"""
352
353import sys
354mswindows = (sys.platform == "win32")
355
356import os
Peter Astrandc26516b2005-02-21 08:13:02 +0000357import types
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000358import traceback
Gregory P. Smith87d49792008-01-19 20:57:59 +0000359import gc
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000360
Peter Astrand454f7672005-01-01 09:36:35 +0000361# Exception classes used by this module.
Peter Astrand7d1d4362006-07-14 14:04:45 +0000362class CalledProcessError(Exception):
Peter Astrand454f7672005-01-01 09:36:35 +0000363 """This exception is raised when a process run by check_call() returns
364 a non-zero exit status. The exit status will be stored in the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000365 returncode attribute."""
366 def __init__(self, returncode, cmd):
367 self.returncode = returncode
368 self.cmd = cmd
369 def __str__(self):
370 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
Tim Peters73a9ead2006-07-18 21:55:15 +0000371
Peter Astrand454f7672005-01-01 09:36:35 +0000372
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000373if mswindows:
374 import threading
375 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000376 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000377 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000378 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
379 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
380 from win32api import GetCurrentProcess, DuplicateHandle, \
381 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000382 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000383 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000384 from win32process import CreateProcess, STARTUPINFO, \
385 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000386 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000387 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000388 else:
389 from _subprocess import *
390 class STARTUPINFO:
391 dwFlags = 0
392 hStdInput = None
393 hStdOutput = None
394 hStdError = None
Georg Brandlad624892006-06-04 22:15:37 +0000395 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000396 class pywintypes:
397 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000398else:
399 import select
400 import errno
401 import fcntl
402 import pickle
403
Peter Astrand454f7672005-01-01 09:36:35 +0000404__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000405
406try:
407 MAXFD = os.sysconf("SC_OPEN_MAX")
408except:
409 MAXFD = 256
410
411# True/False does not exist on 2.2.0
412try:
413 False
414except NameError:
415 False = 0
416 True = 1
417
418_active = []
419
420def _cleanup():
421 for inst in _active[:]:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000422 if inst.poll(_deadstate=sys.maxint) >= 0:
423 try:
424 _active.remove(inst)
425 except ValueError:
426 # This can happen if two threads create a new Popen instance.
427 # It's harmless that it was already removed, so ignore.
428 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000429
430PIPE = -1
431STDOUT = -2
432
433
Peter Astrand5f5e1412004-12-05 20:15:36 +0000434def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000435 """Run command with arguments. Wait for command to complete, then
436 return the returncode attribute.
437
438 The arguments are the same as for the Popen constructor. Example:
439
440 retcode = call(["ls", "-l"])
441 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000442 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000443
444
Peter Astrand454f7672005-01-01 09:36:35 +0000445def check_call(*popenargs, **kwargs):
446 """Run command with arguments. Wait for command to complete. If
447 the exit code was zero then return, otherwise raise
448 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000449 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000450
451 The arguments are the same as for the Popen constructor. Example:
452
453 check_call(["ls", "-l"])
454 """
455 retcode = call(*popenargs, **kwargs)
456 cmd = kwargs.get("args")
457 if cmd is None:
458 cmd = popenargs[0]
459 if retcode:
Peter Astrand7d1d4362006-07-14 14:04:45 +0000460 raise CalledProcessError(retcode, cmd)
Peter Astrand454f7672005-01-01 09:36:35 +0000461 return retcode
462
463
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000464def list2cmdline(seq):
465 """
466 Translate a sequence of arguments into a command line
467 string, using the same rules as the MS C runtime:
468
469 1) Arguments are delimited by white space, which is either a
470 space or a tab.
471
472 2) A string surrounded by double quotation marks is
473 interpreted as a single argument, regardless of white space
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000474 or pipe characters contained within. A quoted string can be
475 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000476
477 3) A double quotation mark preceded by a backslash is
478 interpreted as a literal double quotation mark.
479
480 4) Backslashes are interpreted literally, unless they
481 immediately precede a double quotation mark.
482
483 5) If backslashes immediately precede a double quotation mark,
484 every pair of backslashes is interpreted as a literal
485 backslash. If the number of backslashes is odd, the last
486 backslash escapes the next double quotation mark as
487 described in rule 3.
488 """
489
490 # See
491 # http://msdn.microsoft.com/library/en-us/vccelng/htm/progs_12.asp
492 result = []
493 needquote = False
494 for arg in seq:
495 bs_buf = []
496
497 # Add a space to separate this argument from the others
498 if result:
499 result.append(' ')
500
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000501 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000502 if needquote:
503 result.append('"')
504
505 for c in arg:
506 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000507 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000508 bs_buf.append(c)
509 elif c == '"':
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000510 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000511 result.append('\\' * len(bs_buf)*2)
512 bs_buf = []
513 result.append('\\"')
514 else:
515 # Normal char
516 if bs_buf:
517 result.extend(bs_buf)
518 bs_buf = []
519 result.append(c)
520
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000521 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000522 if bs_buf:
523 result.extend(bs_buf)
524
525 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000526 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000527 result.append('"')
528
529 return ''.join(result)
530
531
532class Popen(object):
533 def __init__(self, args, bufsize=0, executable=None,
534 stdin=None, stdout=None, stderr=None,
535 preexec_fn=None, close_fds=False, shell=False,
536 cwd=None, env=None, universal_newlines=False,
537 startupinfo=None, creationflags=0):
538 """Create new Popen instance."""
539 _cleanup()
540
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000541 self._child_created = False
Peter Astrand738131d2004-11-30 21:04:45 +0000542 if not isinstance(bufsize, (int, long)):
543 raise TypeError("bufsize must be an integer")
544
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000545 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000546 if preexec_fn is not None:
547 raise ValueError("preexec_fn is not supported on Windows "
548 "platforms")
Peter Astrand81a191b2007-05-26 22:18:20 +0000549 if close_fds and (stdin is not None or stdout is not None or
550 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000551 raise ValueError("close_fds is not supported on Windows "
Peter Astrand81a191b2007-05-26 22:18:20 +0000552 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000553 else:
554 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000555 if startupinfo is not None:
556 raise ValueError("startupinfo is only supported on Windows "
557 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000558 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000559 raise ValueError("creationflags is only supported on Windows "
560 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000561
Tim Peterse718f612004-10-12 21:51:32 +0000562 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000563 self.stdout = None
564 self.stderr = None
565 self.pid = None
566 self.returncode = None
567 self.universal_newlines = universal_newlines
568
569 # Input and output objects. The general principle is like
570 # this:
571 #
572 # Parent Child
573 # ------ -----
574 # p2cwrite ---stdin---> p2cread
575 # c2pread <--stdout--- c2pwrite
576 # errread <--stderr--- errwrite
577 #
578 # On POSIX, the child objects are file descriptors. On
579 # Windows, these are Windows file handles. The parent objects
580 # are file descriptors on both platforms. The parent objects
581 # are None when not using PIPEs. The child objects are None
582 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000583
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000584 (p2cread, p2cwrite,
585 c2pread, c2pwrite,
586 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
587
588 self._execute_child(args, executable, preexec_fn, close_fds,
589 cwd, env, universal_newlines,
590 startupinfo, creationflags, shell,
591 p2cread, p2cwrite,
592 c2pread, c2pwrite,
593 errread, errwrite)
594
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000595 # On Windows, you cannot just redirect one or two handles: You
596 # either have to redirect all three or none. If the subprocess
597 # user has only redirected one or two handles, we are
598 # automatically creating PIPEs for the rest. We should close
Tim Petersea5962f2007-03-12 18:07:52 +0000599 # these after the process is started. See bug #1124861.
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000600 if mswindows:
601 if stdin is None and p2cwrite is not None:
602 os.close(p2cwrite)
603 p2cwrite = None
604 if stdout is None and c2pread is not None:
605 os.close(c2pread)
606 c2pread = None
607 if stderr is None and errread is not None:
608 os.close(errread)
609 errread = None
610
Peter Astrandf5400032007-02-02 19:06:36 +0000611 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000612 self.stdin = os.fdopen(p2cwrite, 'wb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000613 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000614 if universal_newlines:
615 self.stdout = os.fdopen(c2pread, 'rU', bufsize)
616 else:
617 self.stdout = os.fdopen(c2pread, 'rb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000618 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000619 if universal_newlines:
620 self.stderr = os.fdopen(errread, 'rU', bufsize)
621 else:
622 self.stderr = os.fdopen(errread, 'rb', bufsize)
Tim Peterse718f612004-10-12 21:51:32 +0000623
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000624
625 def _translate_newlines(self, data):
626 data = data.replace("\r\n", "\n")
627 data = data.replace("\r", "\n")
628 return data
629
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000630
Georg Brandl24522982007-04-21 20:35:38 +0000631 def __del__(self, sys=sys):
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000632 if not self._child_created:
633 # We didn't get to successfully create a child process.
634 return
635 # In case the child hasn't been waited on, check if it's done.
636 self.poll(_deadstate=sys.maxint)
Georg Brandl13cf38c2006-07-20 16:28:39 +0000637 if self.returncode is None and _active is not None:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000638 # Child is still running, keep us alive until we can wait on it.
639 _active.append(self)
640
641
Peter Astrand23109f02005-03-03 20:28:59 +0000642 def communicate(self, input=None):
643 """Interact with process: Send data to stdin. Read data from
644 stdout and stderr, until end-of-file is reached. Wait for
645 process to terminate. The optional input argument should be a
646 string to be sent to the child process, or None, if no data
647 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000648
Peter Astrand23109f02005-03-03 20:28:59 +0000649 communicate() returns a tuple (stdout, stderr)."""
650
651 # Optimization: If we are only using one pipe, or no pipe at
652 # all, using select() or threads is unnecessary.
653 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000654 stdout = None
655 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000656 if self.stdin:
657 if input:
658 self.stdin.write(input)
659 self.stdin.close()
660 elif self.stdout:
661 stdout = self.stdout.read()
662 elif self.stderr:
663 stderr = self.stderr.read()
664 self.wait()
665 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000666
Peter Astrand23109f02005-03-03 20:28:59 +0000667 return self._communicate(input)
668
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000669
670 if mswindows:
671 #
672 # Windows methods
673 #
674 def _get_handles(self, stdin, stdout, stderr):
675 """Construct and return tupel with IO objects:
676 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
677 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000678 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000679 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000680
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000681 p2cread, p2cwrite = None, None
682 c2pread, c2pwrite = None, None
683 errread, errwrite = None, None
684
Peter Astrandd38ddf42005-02-10 08:32:50 +0000685 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000686 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000687 if p2cread is not None:
688 pass
689 elif stdin is None or stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000690 p2cread, p2cwrite = CreatePipe(None, 0)
691 # Detach and turn into fd
692 p2cwrite = p2cwrite.Detach()
693 p2cwrite = msvcrt.open_osfhandle(p2cwrite, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000694 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000695 p2cread = msvcrt.get_osfhandle(stdin)
696 else:
697 # Assuming file-like object
698 p2cread = msvcrt.get_osfhandle(stdin.fileno())
699 p2cread = self._make_inheritable(p2cread)
700
Peter Astrandd38ddf42005-02-10 08:32:50 +0000701 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000702 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000703 if c2pwrite is not None:
704 pass
705 elif stdout is None or stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000706 c2pread, c2pwrite = CreatePipe(None, 0)
707 # Detach and turn into fd
708 c2pread = c2pread.Detach()
709 c2pread = msvcrt.open_osfhandle(c2pread, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000710 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000711 c2pwrite = msvcrt.get_osfhandle(stdout)
712 else:
713 # Assuming file-like object
714 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
715 c2pwrite = self._make_inheritable(c2pwrite)
716
Peter Astrandd38ddf42005-02-10 08:32:50 +0000717 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000718 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000719 if errwrite is not None:
720 pass
721 elif stderr is None or stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000722 errread, errwrite = CreatePipe(None, 0)
723 # Detach and turn into fd
724 errread = errread.Detach()
725 errread = msvcrt.open_osfhandle(errread, 0)
726 elif stderr == STDOUT:
727 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000728 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000729 errwrite = msvcrt.get_osfhandle(stderr)
730 else:
731 # Assuming file-like object
732 errwrite = msvcrt.get_osfhandle(stderr.fileno())
733 errwrite = self._make_inheritable(errwrite)
734
735 return (p2cread, p2cwrite,
736 c2pread, c2pwrite,
737 errread, errwrite)
738
739
740 def _make_inheritable(self, handle):
741 """Return a duplicate of handle, which is inheritable"""
742 return DuplicateHandle(GetCurrentProcess(), handle,
743 GetCurrentProcess(), 0, 1,
744 DUPLICATE_SAME_ACCESS)
745
746
747 def _find_w9xpopen(self):
748 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000749 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
750 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000751 if not os.path.exists(w9xpopen):
752 # Eeek - file-not-found - possibly an embedding
753 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000754 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
755 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000756 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000757 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
758 "needed for Popen to work with your "
759 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000760 return w9xpopen
761
Tim Peterse718f612004-10-12 21:51:32 +0000762
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000763 def _execute_child(self, args, executable, preexec_fn, close_fds,
764 cwd, env, universal_newlines,
765 startupinfo, creationflags, shell,
766 p2cread, p2cwrite,
767 c2pread, c2pwrite,
768 errread, errwrite):
769 """Execute program (MS Windows version)"""
770
Peter Astrandc26516b2005-02-21 08:13:02 +0000771 if not isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000772 args = list2cmdline(args)
773
Peter Astrandc1d65362004-11-07 14:30:34 +0000774 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000775 if startupinfo is None:
Georg Brandlad624892006-06-04 22:15:37 +0000776 startupinfo = STARTUPINFO()
777 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000778 startupinfo.dwFlags |= STARTF_USESTDHANDLES
779 startupinfo.hStdInput = p2cread
780 startupinfo.hStdOutput = c2pwrite
781 startupinfo.hStdError = errwrite
782
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000783 if shell:
Georg Brandlad624892006-06-04 22:15:37 +0000784 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
785 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000786 comspec = os.environ.get("COMSPEC", "cmd.exe")
787 args = comspec + " /c " + args
Tim Peterse8374a52004-10-13 03:15:00 +0000788 if (GetVersion() >= 0x80000000L or
789 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000790 # Win9x, or using command.com on NT. We need to
791 # use the w9xpopen intermediate program. For more
792 # information, see KB Q150956
793 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
794 w9xpopen = self._find_w9xpopen()
795 args = '"%s" %s' % (w9xpopen, args)
796 # Not passing CREATE_NEW_CONSOLE has been known to
797 # cause random failures on win9x. Specifically a
798 # dialog: "Your program accessed mem currently in
799 # use at xxx" and a hopeful warning about the
800 # stability of your system. Cost is Ctrl+C wont
801 # kill children.
802 creationflags |= CREATE_NEW_CONSOLE
803
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000804 # Start the process
805 try:
806 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000807 # no special security
808 None, None,
Peter Astrand81a191b2007-05-26 22:18:20 +0000809 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000810 creationflags,
811 env,
812 cwd,
813 startupinfo)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000814 except pywintypes.error, e:
815 # Translate pywintypes.error to WindowsError, which is
816 # a subclass of OSError. FIXME: We should really
817 # translate errno using _sys_errlist (or simliar), but
818 # how can this be done from Python?
819 raise WindowsError(*e.args)
820
821 # Retain the process handle, but close the thread handle
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000822 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000823 self._handle = hp
824 self.pid = pid
825 ht.Close()
826
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000827 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000828 # handles that only the child should have open. You need
829 # to make sure that no handles to the write end of the
830 # output pipe are maintained in this process or else the
831 # pipe will not close when the child process exits and the
832 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000833 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000834 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000835 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000836 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000837 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000838 errwrite.Close()
839
Tim Peterse718f612004-10-12 21:51:32 +0000840
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000841 def poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000842 """Check if child process has terminated. Returns returncode
843 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000844 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000845 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
846 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000847 return self.returncode
848
849
850 def wait(self):
851 """Wait for child process to terminate. Returns returncode
852 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000853 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000854 obj = WaitForSingleObject(self._handle, INFINITE)
855 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000856 return self.returncode
857
858
859 def _readerthread(self, fh, buffer):
860 buffer.append(fh.read())
861
862
Peter Astrand23109f02005-03-03 20:28:59 +0000863 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000864 stdout = None # Return
865 stderr = None # Return
866
867 if self.stdout:
868 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000869 stdout_thread = threading.Thread(target=self._readerthread,
870 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000871 stdout_thread.setDaemon(True)
872 stdout_thread.start()
873 if self.stderr:
874 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000875 stderr_thread = threading.Thread(target=self._readerthread,
876 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000877 stderr_thread.setDaemon(True)
878 stderr_thread.start()
879
880 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000881 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000882 self.stdin.write(input)
883 self.stdin.close()
884
885 if self.stdout:
886 stdout_thread.join()
887 if self.stderr:
888 stderr_thread.join()
889
890 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000891 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000892 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000893 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000894 stderr = stderr[0]
895
896 # Translate newlines, if requested. We cannot let the file
897 # object do the translation: It is based on stdio, which is
898 # impossible to combine with select (unless forcing no
899 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +0000900 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000901 if stdout:
902 stdout = self._translate_newlines(stdout)
903 if stderr:
904 stderr = self._translate_newlines(stderr)
905
906 self.wait()
907 return (stdout, stderr)
908
909 else:
910 #
911 # POSIX methods
912 #
913 def _get_handles(self, stdin, stdout, stderr):
914 """Construct and return tupel with IO objects:
915 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
916 """
917 p2cread, p2cwrite = None, None
918 c2pread, c2pwrite = None, None
919 errread, errwrite = None, None
920
Peter Astrandd38ddf42005-02-10 08:32:50 +0000921 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000922 pass
923 elif stdin == PIPE:
924 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000925 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000926 p2cread = stdin
927 else:
928 # Assuming file-like object
929 p2cread = stdin.fileno()
930
Peter Astrandd38ddf42005-02-10 08:32:50 +0000931 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000932 pass
933 elif stdout == PIPE:
934 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000935 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000936 c2pwrite = stdout
937 else:
938 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000939 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000940
Peter Astrandd38ddf42005-02-10 08:32:50 +0000941 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000942 pass
943 elif stderr == PIPE:
944 errread, errwrite = os.pipe()
945 elif stderr == STDOUT:
946 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000947 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000948 errwrite = stderr
949 else:
950 # Assuming file-like object
951 errwrite = stderr.fileno()
952
953 return (p2cread, p2cwrite,
954 c2pread, c2pwrite,
955 errread, errwrite)
956
957
958 def _set_cloexec_flag(self, fd):
959 try:
960 cloexec_flag = fcntl.FD_CLOEXEC
961 except AttributeError:
962 cloexec_flag = 1
963
964 old = fcntl.fcntl(fd, fcntl.F_GETFD)
965 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
966
967
968 def _close_fds(self, but):
Georg Brandl309501a2008-01-19 20:22:13 +0000969 os.closerange(3, but)
970 os.closerange(but + 1, MAXFD)
Tim Peterse718f612004-10-12 21:51:32 +0000971
972
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000973 def _execute_child(self, args, executable, preexec_fn, close_fds,
974 cwd, env, universal_newlines,
975 startupinfo, creationflags, shell,
976 p2cread, p2cwrite,
977 c2pread, c2pwrite,
978 errread, errwrite):
979 """Execute program (POSIX version)"""
980
Peter Astrandc26516b2005-02-21 08:13:02 +0000981 if isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000982 args = [args]
Georg Brandl6c0e1e82006-10-29 09:05:04 +0000983 else:
984 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000985
986 if shell:
987 args = ["/bin/sh", "-c"] + args
988
Peter Astrandd38ddf42005-02-10 08:32:50 +0000989 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000990 executable = args[0]
991
992 # For transferring possible exec failure from child to parent
993 # The first char specifies the exception type: 0 means
994 # OSError, 1 means some other error.
995 errpipe_read, errpipe_write = os.pipe()
996 self._set_cloexec_flag(errpipe_write)
997
Gregory P. Smith87d49792008-01-19 20:57:59 +0000998 gc_was_enabled = gc.isenabled()
999 # Disable gc to avoid bug where gc -> file_dealloc ->
1000 # write to stderr -> hang. http://bugs.python.org/issue1336
1001 gc.disable()
1002 try:
1003 self.pid = os.fork()
1004 except:
1005 if gc_was_enabled:
1006 gc.enable()
1007 raise
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001008 self._child_created = True
1009 if self.pid == 0:
1010 # Child
1011 try:
1012 # Close parent's pipe ends
1013 if p2cwrite is not None:
1014 os.close(p2cwrite)
1015 if c2pread is not None:
1016 os.close(c2pread)
1017 if errread is not None:
1018 os.close(errread)
1019 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001020
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001021 # Dup fds for child
1022 if p2cread is not None:
1023 os.dup2(p2cread, 0)
1024 if c2pwrite is not None:
1025 os.dup2(c2pwrite, 1)
1026 if errwrite is not None:
1027 os.dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001028
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001029 # Close pipe fds. Make sure we don't close the same
1030 # fd more than once, or standard fds.
1031 if p2cread is not None and p2cread not in (0,):
1032 os.close(p2cread)
1033 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
1034 os.close(c2pwrite)
1035 if errwrite is not None and errwrite not in (p2cread, c2pwrite, 2):
1036 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001037
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001038 # Close all other fds, if asked for
1039 if close_fds:
1040 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001041
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001042 if cwd is not None:
1043 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001044
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001045 if preexec_fn:
1046 apply(preexec_fn)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001047
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001048 if env is None:
1049 os.execvp(executable, args)
1050 else:
1051 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001052
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001053 except:
1054 exc_type, exc_value, tb = sys.exc_info()
1055 # Save the traceback and attach it to the exception object
1056 exc_lines = traceback.format_exception(exc_type,
1057 exc_value,
1058 tb)
1059 exc_value.child_traceback = ''.join(exc_lines)
1060 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001061
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001062 # This exitcode won't be reported to applications, so it
1063 # really doesn't matter what we return.
1064 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001065
1066 # Parent
Gregory P. Smith87d49792008-01-19 20:57:59 +00001067 if gc_was_enabled:
1068 gc.enable()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001069 os.close(errpipe_write)
Peter Astrandf5400032007-02-02 19:06:36 +00001070 if p2cread is not None and p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001071 os.close(p2cread)
Peter Astrandf5400032007-02-02 19:06:36 +00001072 if c2pwrite is not None and c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001073 os.close(c2pwrite)
Peter Astrandf5400032007-02-02 19:06:36 +00001074 if errwrite is not None and errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001075 os.close(errwrite)
1076
1077 # Wait for exec to fail or succeed; possibly raising exception
1078 data = os.read(errpipe_read, 1048576) # Exceptions limited to 1 MB
1079 os.close(errpipe_read)
1080 if data != "":
Peter Astrandf791d7a2005-01-01 09:38:57 +00001081 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001082 child_exception = pickle.loads(data)
1083 raise child_exception
1084
1085
1086 def _handle_exitstatus(self, sts):
1087 if os.WIFSIGNALED(sts):
1088 self.returncode = -os.WTERMSIG(sts)
1089 elif os.WIFEXITED(sts):
1090 self.returncode = os.WEXITSTATUS(sts)
1091 else:
1092 # Should never happen
1093 raise RuntimeError("Unknown child exit status!")
1094
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001095
Martin v. Löwis17de8ff2006-04-10 15:55:37 +00001096 def poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001097 """Check if child process has terminated. Returns returncode
1098 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001099 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001100 try:
1101 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1102 if pid == self.pid:
1103 self._handle_exitstatus(sts)
1104 except os.error:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +00001105 if _deadstate is not None:
1106 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001107 return self.returncode
1108
1109
1110 def wait(self):
1111 """Wait for child process to terminate. Returns returncode
1112 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001113 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001114 pid, sts = os.waitpid(self.pid, 0)
1115 self._handle_exitstatus(sts)
1116 return self.returncode
1117
1118
Peter Astrand23109f02005-03-03 20:28:59 +00001119 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001120 read_set = []
1121 write_set = []
1122 stdout = None # Return
1123 stderr = None # Return
1124
1125 if self.stdin:
1126 # Flush stdio buffer. This might block, if the user has
1127 # been writing to .stdin in an uncontrolled fashion.
1128 self.stdin.flush()
1129 if input:
1130 write_set.append(self.stdin)
1131 else:
1132 self.stdin.close()
1133 if self.stdout:
1134 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001135 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001136 if self.stderr:
1137 read_set.append(self.stderr)
1138 stderr = []
1139
Peter Astrand1812f8c2007-01-07 14:34:16 +00001140 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001141 while read_set or write_set:
1142 rlist, wlist, xlist = select.select(read_set, write_set, [])
1143
1144 if self.stdin in wlist:
1145 # When select has indicated that the file is writable,
1146 # we can write up to PIPE_BUF bytes without risk
1147 # blocking. POSIX defines PIPE_BUF >= 512
Peter Astrand1812f8c2007-01-07 14:34:16 +00001148 bytes_written = os.write(self.stdin.fileno(), buffer(input, input_offset, 512))
Tim Petersf733abb2007-01-30 03:03:46 +00001149 input_offset += bytes_written
Peter Astrand1812f8c2007-01-07 14:34:16 +00001150 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001151 self.stdin.close()
1152 write_set.remove(self.stdin)
1153
1154 if self.stdout in rlist:
1155 data = os.read(self.stdout.fileno(), 1024)
1156 if data == "":
1157 self.stdout.close()
1158 read_set.remove(self.stdout)
1159 stdout.append(data)
1160
1161 if self.stderr in rlist:
1162 data = os.read(self.stderr.fileno(), 1024)
1163 if data == "":
1164 self.stderr.close()
1165 read_set.remove(self.stderr)
1166 stderr.append(data)
1167
1168 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +00001169 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001170 stdout = ''.join(stdout)
Peter Astrandd38ddf42005-02-10 08:32:50 +00001171 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001172 stderr = ''.join(stderr)
1173
1174 # Translate newlines, if requested. We cannot let the file
1175 # object do the translation: It is based on stdio, which is
1176 # impossible to combine with select (unless forcing no
1177 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +00001178 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001179 if stdout:
1180 stdout = self._translate_newlines(stdout)
1181 if stderr:
1182 stderr = self._translate_newlines(stderr)
1183
1184 self.wait()
1185 return (stdout, stderr)
1186
1187
1188def _demo_posix():
1189 #
1190 # Example 1: Simple redirection: Get process list
1191 #
1192 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
1193 print "Process list:"
1194 print plist
1195
1196 #
1197 # Example 2: Change uid before executing child
1198 #
1199 if os.getuid() == 0:
1200 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1201 p.wait()
1202
1203 #
1204 # Example 3: Connecting several subprocesses
1205 #
1206 print "Looking for 'hda'..."
1207 p1 = Popen(["dmesg"], stdout=PIPE)
1208 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
1209 print repr(p2.communicate()[0])
1210
1211 #
1212 # Example 4: Catch execution error
1213 #
1214 print
1215 print "Trying a weird file..."
1216 try:
1217 print Popen(["/this/path/does/not/exist"]).communicate()
1218 except OSError, e:
1219 if e.errno == errno.ENOENT:
1220 print "The file didn't exist. I thought so..."
1221 print "Child traceback:"
1222 print e.child_traceback
1223 else:
1224 print "Error", e.errno
1225 else:
1226 print >>sys.stderr, "Gosh. No error."
1227
1228
1229def _demo_windows():
1230 #
1231 # Example 1: Connecting several subprocesses
1232 #
1233 print "Looking for 'PROMPT' in set output..."
1234 p1 = Popen("set", stdout=PIPE, shell=True)
1235 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
1236 print repr(p2.communicate()[0])
1237
1238 #
1239 # Example 2: Simple execution of program
1240 #
1241 print "Executing calc..."
1242 p = Popen("calc")
1243 p.wait()
1244
1245
1246if __name__ == "__main__":
1247 if mswindows:
1248 _demo_windows()
1249 else:
1250 _demo_posix()