blob: 9eb192135c4d9e924e6eabe138e230aed8b44ec9 [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
Christian Heimese74c8f22008-04-19 02:23:57 +0000360import signal
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000361
Peter Astrand454f7672005-01-01 09:36:35 +0000362# Exception classes used by this module.
Peter Astrand7d1d4362006-07-14 14:04:45 +0000363class CalledProcessError(Exception):
Peter Astrand454f7672005-01-01 09:36:35 +0000364 """This exception is raised when a process run by check_call() returns
365 a non-zero exit status. The exit status will be stored in the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000366 returncode attribute."""
367 def __init__(self, returncode, cmd):
368 self.returncode = returncode
369 self.cmd = cmd
370 def __str__(self):
371 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
Tim Peters73a9ead2006-07-18 21:55:15 +0000372
Peter Astrand454f7672005-01-01 09:36:35 +0000373
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000374if mswindows:
375 import threading
376 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000377 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000378 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000379 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
380 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
381 from win32api import GetCurrentProcess, DuplicateHandle, \
382 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000383 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000384 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000385 from win32process import CreateProcess, STARTUPINFO, \
386 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000387 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Christian Heimese74c8f22008-04-19 02:23:57 +0000388 from win32process import TerminateProcess
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000389 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000390 else:
391 from _subprocess import *
392 class STARTUPINFO:
393 dwFlags = 0
394 hStdInput = None
395 hStdOutput = None
396 hStdError = None
Georg Brandlad624892006-06-04 22:15:37 +0000397 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000398 class pywintypes:
399 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000400else:
401 import select
402 import errno
403 import fcntl
404 import pickle
405
Peter Astrand454f7672005-01-01 09:36:35 +0000406__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000407
408try:
409 MAXFD = os.sysconf("SC_OPEN_MAX")
410except:
411 MAXFD = 256
412
413# True/False does not exist on 2.2.0
Brett Cannon03446c42008-08-08 04:19:32 +0000414#try:
415# False
416#except NameError:
417# False = 0
418# True = 1
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000419
420_active = []
421
422def _cleanup():
423 for inst in _active[:]:
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000424 if inst._internal_poll(_deadstate=sys.maxint) >= 0:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000425 try:
426 _active.remove(inst)
427 except ValueError:
428 # This can happen if two threads create a new Popen instance.
429 # It's harmless that it was already removed, so ignore.
430 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000431
432PIPE = -1
433STDOUT = -2
434
435
Peter Astrand5f5e1412004-12-05 20:15:36 +0000436def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000437 """Run command with arguments. Wait for command to complete, then
438 return the returncode attribute.
439
440 The arguments are the same as for the Popen constructor. Example:
441
442 retcode = call(["ls", "-l"])
443 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000444 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000445
446
Peter Astrand454f7672005-01-01 09:36:35 +0000447def check_call(*popenargs, **kwargs):
448 """Run command with arguments. Wait for command to complete. If
449 the exit code was zero then return, otherwise raise
450 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000451 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000452
453 The arguments are the same as for the Popen constructor. Example:
454
455 check_call(["ls", "-l"])
456 """
457 retcode = call(*popenargs, **kwargs)
458 cmd = kwargs.get("args")
459 if cmd is None:
460 cmd = popenargs[0]
461 if retcode:
Peter Astrand7d1d4362006-07-14 14:04:45 +0000462 raise CalledProcessError(retcode, cmd)
Peter Astrand454f7672005-01-01 09:36:35 +0000463 return retcode
464
465
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000466def list2cmdline(seq):
467 """
468 Translate a sequence of arguments into a command line
469 string, using the same rules as the MS C runtime:
470
471 1) Arguments are delimited by white space, which is either a
472 space or a tab.
473
474 2) A string surrounded by double quotation marks is
475 interpreted as a single argument, regardless of white space
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000476 or pipe characters contained within. A quoted string can be
477 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000478
479 3) A double quotation mark preceded by a backslash is
480 interpreted as a literal double quotation mark.
481
482 4) Backslashes are interpreted literally, unless they
483 immediately precede a double quotation mark.
484
485 5) If backslashes immediately precede a double quotation mark,
486 every pair of backslashes is interpreted as a literal
487 backslash. If the number of backslashes is odd, the last
488 backslash escapes the next double quotation mark as
489 described in rule 3.
490 """
491
492 # See
493 # http://msdn.microsoft.com/library/en-us/vccelng/htm/progs_12.asp
494 result = []
495 needquote = False
496 for arg in seq:
497 bs_buf = []
498
499 # Add a space to separate this argument from the others
500 if result:
501 result.append(' ')
502
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000503 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000504 if needquote:
505 result.append('"')
506
507 for c in arg:
508 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000509 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000510 bs_buf.append(c)
511 elif c == '"':
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000512 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000513 result.append('\\' * len(bs_buf)*2)
514 bs_buf = []
515 result.append('\\"')
516 else:
517 # Normal char
518 if bs_buf:
519 result.extend(bs_buf)
520 bs_buf = []
521 result.append(c)
522
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000523 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000524 if bs_buf:
525 result.extend(bs_buf)
526
527 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000528 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000529 result.append('"')
530
531 return ''.join(result)
532
533
534class Popen(object):
535 def __init__(self, args, bufsize=0, executable=None,
536 stdin=None, stdout=None, stderr=None,
537 preexec_fn=None, close_fds=False, shell=False,
538 cwd=None, env=None, universal_newlines=False,
539 startupinfo=None, creationflags=0):
540 """Create new Popen instance."""
541 _cleanup()
542
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000543 self._child_created = False
Peter Astrand738131d2004-11-30 21:04:45 +0000544 if not isinstance(bufsize, (int, long)):
545 raise TypeError("bufsize must be an integer")
546
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000547 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000548 if preexec_fn is not None:
549 raise ValueError("preexec_fn is not supported on Windows "
550 "platforms")
Peter Astrand81a191b2007-05-26 22:18:20 +0000551 if close_fds and (stdin is not None or stdout is not None or
552 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000553 raise ValueError("close_fds is not supported on Windows "
Peter Astrand81a191b2007-05-26 22:18:20 +0000554 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000555 else:
556 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000557 if startupinfo is not None:
558 raise ValueError("startupinfo is only supported on Windows "
559 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000560 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000561 raise ValueError("creationflags is only supported on Windows "
562 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000563
Tim Peterse718f612004-10-12 21:51:32 +0000564 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000565 self.stdout = None
566 self.stderr = None
567 self.pid = None
568 self.returncode = None
569 self.universal_newlines = universal_newlines
570
571 # Input and output objects. The general principle is like
572 # this:
573 #
574 # Parent Child
575 # ------ -----
576 # p2cwrite ---stdin---> p2cread
577 # c2pread <--stdout--- c2pwrite
578 # errread <--stderr--- errwrite
579 #
580 # On POSIX, the child objects are file descriptors. On
581 # Windows, these are Windows file handles. The parent objects
582 # are file descriptors on both platforms. The parent objects
583 # are None when not using PIPEs. The child objects are None
584 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000585
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000586 (p2cread, p2cwrite,
587 c2pread, c2pwrite,
588 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
589
590 self._execute_child(args, executable, preexec_fn, close_fds,
591 cwd, env, universal_newlines,
592 startupinfo, creationflags, shell,
593 p2cread, p2cwrite,
594 c2pread, c2pwrite,
595 errread, errwrite)
596
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000597 if mswindows:
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000598 if p2cwrite is not None:
599 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0)
600 if c2pread is not None:
601 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0)
602 if errread is not None:
603 errread = msvcrt.open_osfhandle(errread.Detach(), 0)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000604
Peter Astrandf5400032007-02-02 19:06:36 +0000605 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000606 self.stdin = os.fdopen(p2cwrite, 'wb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000607 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000608 if universal_newlines:
609 self.stdout = os.fdopen(c2pread, 'rU', bufsize)
610 else:
611 self.stdout = os.fdopen(c2pread, 'rb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000612 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000613 if universal_newlines:
614 self.stderr = os.fdopen(errread, 'rU', bufsize)
615 else:
616 self.stderr = os.fdopen(errread, 'rb', bufsize)
Tim Peterse718f612004-10-12 21:51:32 +0000617
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000618
619 def _translate_newlines(self, data):
620 data = data.replace("\r\n", "\n")
621 data = data.replace("\r", "\n")
622 return data
623
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000624
Georg Brandl24522982007-04-21 20:35:38 +0000625 def __del__(self, sys=sys):
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000626 if not self._child_created:
627 # We didn't get to successfully create a child process.
628 return
629 # In case the child hasn't been waited on, check if it's done.
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000630 self._internal_poll(_deadstate=sys.maxint)
Georg Brandl13cf38c2006-07-20 16:28:39 +0000631 if self.returncode is None and _active is not None:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000632 # Child is still running, keep us alive until we can wait on it.
633 _active.append(self)
634
635
Peter Astrand23109f02005-03-03 20:28:59 +0000636 def communicate(self, input=None):
637 """Interact with process: Send data to stdin. Read data from
638 stdout and stderr, until end-of-file is reached. Wait for
639 process to terminate. The optional input argument should be a
640 string to be sent to the child process, or None, if no data
641 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000642
Peter Astrand23109f02005-03-03 20:28:59 +0000643 communicate() returns a tuple (stdout, stderr)."""
644
645 # Optimization: If we are only using one pipe, or no pipe at
646 # all, using select() or threads is unnecessary.
647 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000648 stdout = None
649 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000650 if self.stdin:
651 if input:
652 self.stdin.write(input)
653 self.stdin.close()
654 elif self.stdout:
655 stdout = self.stdout.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000656 self.stdout.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000657 elif self.stderr:
658 stderr = self.stderr.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000659 self.stderr.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000660 self.wait()
661 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000662
Peter Astrand23109f02005-03-03 20:28:59 +0000663 return self._communicate(input)
664
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000665
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000666 def poll(self):
667 return self._internal_poll()
668
669
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000670 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)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000687 if p2cread is None:
688 p2cread, _ = CreatePipe(None, 0)
689 elif stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000690 p2cread, p2cwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000691 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000692 p2cread = msvcrt.get_osfhandle(stdin)
693 else:
694 # Assuming file-like object
695 p2cread = msvcrt.get_osfhandle(stdin.fileno())
696 p2cread = self._make_inheritable(p2cread)
697
Peter Astrandd38ddf42005-02-10 08:32:50 +0000698 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000699 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000700 if c2pwrite is None:
701 _, c2pwrite = CreatePipe(None, 0)
702 elif stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000703 c2pread, c2pwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000704 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000705 c2pwrite = msvcrt.get_osfhandle(stdout)
706 else:
707 # Assuming file-like object
708 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
709 c2pwrite = self._make_inheritable(c2pwrite)
710
Peter Astrandd38ddf42005-02-10 08:32:50 +0000711 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000712 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000713 if errwrite is None:
714 _, errwrite = CreatePipe(None, 0)
715 elif stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000716 errread, errwrite = CreatePipe(None, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000717 elif stderr == STDOUT:
718 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000719 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000720 errwrite = msvcrt.get_osfhandle(stderr)
721 else:
722 # Assuming file-like object
723 errwrite = msvcrt.get_osfhandle(stderr.fileno())
724 errwrite = self._make_inheritable(errwrite)
725
726 return (p2cread, p2cwrite,
727 c2pread, c2pwrite,
728 errread, errwrite)
729
730
731 def _make_inheritable(self, handle):
732 """Return a duplicate of handle, which is inheritable"""
733 return DuplicateHandle(GetCurrentProcess(), handle,
734 GetCurrentProcess(), 0, 1,
735 DUPLICATE_SAME_ACCESS)
736
737
738 def _find_w9xpopen(self):
739 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000740 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
741 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000742 if not os.path.exists(w9xpopen):
743 # Eeek - file-not-found - possibly an embedding
744 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000745 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
746 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000747 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000748 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
749 "needed for Popen to work with your "
750 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000751 return w9xpopen
752
Tim Peterse718f612004-10-12 21:51:32 +0000753
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000754 def _execute_child(self, args, executable, preexec_fn, close_fds,
755 cwd, env, universal_newlines,
756 startupinfo, creationflags, shell,
757 p2cread, p2cwrite,
758 c2pread, c2pwrite,
759 errread, errwrite):
760 """Execute program (MS Windows version)"""
761
Peter Astrandc26516b2005-02-21 08:13:02 +0000762 if not isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000763 args = list2cmdline(args)
764
Peter Astrandc1d65362004-11-07 14:30:34 +0000765 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000766 if startupinfo is None:
Georg Brandlad624892006-06-04 22:15:37 +0000767 startupinfo = STARTUPINFO()
768 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000769 startupinfo.dwFlags |= STARTF_USESTDHANDLES
770 startupinfo.hStdInput = p2cread
771 startupinfo.hStdOutput = c2pwrite
772 startupinfo.hStdError = errwrite
773
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000774 if shell:
Georg Brandlad624892006-06-04 22:15:37 +0000775 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
776 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000777 comspec = os.environ.get("COMSPEC", "cmd.exe")
778 args = comspec + " /c " + args
Tim Peterse8374a52004-10-13 03:15:00 +0000779 if (GetVersion() >= 0x80000000L or
780 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000781 # Win9x, or using command.com on NT. We need to
782 # use the w9xpopen intermediate program. For more
783 # information, see KB Q150956
784 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
785 w9xpopen = self._find_w9xpopen()
786 args = '"%s" %s' % (w9xpopen, args)
787 # Not passing CREATE_NEW_CONSOLE has been known to
788 # cause random failures on win9x. Specifically a
789 # dialog: "Your program accessed mem currently in
790 # use at xxx" and a hopeful warning about the
791 # stability of your system. Cost is Ctrl+C wont
792 # kill children.
793 creationflags |= CREATE_NEW_CONSOLE
794
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000795 # Start the process
796 try:
797 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000798 # no special security
799 None, None,
Peter Astrand81a191b2007-05-26 22:18:20 +0000800 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000801 creationflags,
802 env,
803 cwd,
804 startupinfo)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000805 except pywintypes.error, e:
806 # Translate pywintypes.error to WindowsError, which is
807 # a subclass of OSError. FIXME: We should really
808 # translate errno using _sys_errlist (or simliar), but
809 # how can this be done from Python?
810 raise WindowsError(*e.args)
811
812 # Retain the process handle, but close the thread handle
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000813 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000814 self._handle = hp
815 self.pid = pid
816 ht.Close()
817
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000818 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000819 # handles that only the child should have open. You need
820 # to make sure that no handles to the write end of the
821 # output pipe are maintained in this process or else the
822 # pipe will not close when the child process exits and the
823 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000824 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000825 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000826 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000827 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000828 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000829 errwrite.Close()
830
Tim Peterse718f612004-10-12 21:51:32 +0000831
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000832 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000833 """Check if child process has terminated. Returns returncode
834 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000835 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000836 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
837 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000838 return self.returncode
839
840
841 def wait(self):
842 """Wait for child process to terminate. 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 obj = WaitForSingleObject(self._handle, INFINITE)
846 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000847 return self.returncode
848
849
850 def _readerthread(self, fh, buffer):
851 buffer.append(fh.read())
852
853
Peter Astrand23109f02005-03-03 20:28:59 +0000854 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000855 stdout = None # Return
856 stderr = None # Return
857
858 if self.stdout:
859 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000860 stdout_thread = threading.Thread(target=self._readerthread,
861 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000862 stdout_thread.setDaemon(True)
863 stdout_thread.start()
864 if self.stderr:
865 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000866 stderr_thread = threading.Thread(target=self._readerthread,
867 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000868 stderr_thread.setDaemon(True)
869 stderr_thread.start()
870
871 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000872 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000873 self.stdin.write(input)
874 self.stdin.close()
875
876 if self.stdout:
877 stdout_thread.join()
878 if self.stderr:
879 stderr_thread.join()
880
881 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000882 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000883 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000884 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000885 stderr = stderr[0]
886
887 # Translate newlines, if requested. We cannot let the file
888 # object do the translation: It is based on stdio, which is
889 # impossible to combine with select (unless forcing no
890 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +0000891 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000892 if stdout:
893 stdout = self._translate_newlines(stdout)
894 if stderr:
895 stderr = self._translate_newlines(stderr)
896
897 self.wait()
898 return (stdout, stderr)
899
Christian Heimese74c8f22008-04-19 02:23:57 +0000900 def send_signal(self, sig):
901 """Send a signal to the process
902 """
903 if sig == signal.SIGTERM:
904 self.terminate()
905 else:
906 raise ValueError("Only SIGTERM is supported on Windows")
907
908 def terminate(self):
909 """Terminates the process
910 """
911 TerminateProcess(self._handle, 1)
912
913 kill = terminate
914
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000915 else:
916 #
917 # POSIX methods
918 #
919 def _get_handles(self, stdin, stdout, stderr):
920 """Construct and return tupel with IO objects:
921 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
922 """
923 p2cread, p2cwrite = None, None
924 c2pread, c2pwrite = None, None
925 errread, errwrite = None, None
926
Peter Astrandd38ddf42005-02-10 08:32:50 +0000927 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000928 pass
929 elif stdin == PIPE:
930 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000931 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000932 p2cread = stdin
933 else:
934 # Assuming file-like object
935 p2cread = stdin.fileno()
936
Peter Astrandd38ddf42005-02-10 08:32:50 +0000937 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000938 pass
939 elif stdout == PIPE:
940 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000941 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000942 c2pwrite = stdout
943 else:
944 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000945 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000946
Peter Astrandd38ddf42005-02-10 08:32:50 +0000947 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000948 pass
949 elif stderr == PIPE:
950 errread, errwrite = os.pipe()
951 elif stderr == STDOUT:
952 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000953 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000954 errwrite = stderr
955 else:
956 # Assuming file-like object
957 errwrite = stderr.fileno()
958
959 return (p2cread, p2cwrite,
960 c2pread, c2pwrite,
961 errread, errwrite)
962
963
964 def _set_cloexec_flag(self, fd):
965 try:
966 cloexec_flag = fcntl.FD_CLOEXEC
967 except AttributeError:
968 cloexec_flag = 1
969
970 old = fcntl.fcntl(fd, fcntl.F_GETFD)
971 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
972
973
974 def _close_fds(self, but):
Georg Brandl309501a2008-01-19 20:22:13 +0000975 os.closerange(3, but)
976 os.closerange(but + 1, MAXFD)
Tim Peterse718f612004-10-12 21:51:32 +0000977
978
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000979 def _execute_child(self, args, executable, preexec_fn, close_fds,
980 cwd, env, universal_newlines,
981 startupinfo, creationflags, shell,
982 p2cread, p2cwrite,
983 c2pread, c2pwrite,
984 errread, errwrite):
985 """Execute program (POSIX version)"""
986
Peter Astrandc26516b2005-02-21 08:13:02 +0000987 if isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000988 args = [args]
Georg Brandl6c0e1e82006-10-29 09:05:04 +0000989 else:
990 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000991
992 if shell:
993 args = ["/bin/sh", "-c"] + args
994
Peter Astrandd38ddf42005-02-10 08:32:50 +0000995 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000996 executable = args[0]
997
998 # For transferring possible exec failure from child to parent
999 # The first char specifies the exception type: 0 means
1000 # OSError, 1 means some other error.
1001 errpipe_read, errpipe_write = os.pipe()
1002 self._set_cloexec_flag(errpipe_write)
1003
Gregory P. Smith87d49792008-01-19 20:57:59 +00001004 gc_was_enabled = gc.isenabled()
1005 # Disable gc to avoid bug where gc -> file_dealloc ->
1006 # write to stderr -> hang. http://bugs.python.org/issue1336
1007 gc.disable()
1008 try:
1009 self.pid = os.fork()
1010 except:
1011 if gc_was_enabled:
1012 gc.enable()
1013 raise
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001014 self._child_created = True
1015 if self.pid == 0:
1016 # Child
1017 try:
1018 # Close parent's pipe ends
1019 if p2cwrite is not None:
1020 os.close(p2cwrite)
1021 if c2pread is not None:
1022 os.close(c2pread)
1023 if errread is not None:
1024 os.close(errread)
1025 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001026
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001027 # Dup fds for child
1028 if p2cread is not None:
1029 os.dup2(p2cread, 0)
1030 if c2pwrite is not None:
1031 os.dup2(c2pwrite, 1)
1032 if errwrite is not None:
1033 os.dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001034
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001035 # Close pipe fds. Make sure we don't close the same
1036 # fd more than once, or standard fds.
1037 if p2cread is not None and p2cread not in (0,):
1038 os.close(p2cread)
1039 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
1040 os.close(c2pwrite)
1041 if errwrite is not None and errwrite not in (p2cread, c2pwrite, 2):
1042 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001043
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001044 # Close all other fds, if asked for
1045 if close_fds:
1046 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001047
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001048 if cwd is not None:
1049 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001050
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001051 if preexec_fn:
Brett Cannon03446c42008-08-08 04:19:32 +00001052 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001053
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001054 if env is None:
1055 os.execvp(executable, args)
1056 else:
1057 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001058
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001059 except:
1060 exc_type, exc_value, tb = sys.exc_info()
1061 # Save the traceback and attach it to the exception object
1062 exc_lines = traceback.format_exception(exc_type,
1063 exc_value,
1064 tb)
1065 exc_value.child_traceback = ''.join(exc_lines)
1066 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001067
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001068 # This exitcode won't be reported to applications, so it
1069 # really doesn't matter what we return.
1070 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001071
1072 # Parent
Gregory P. Smith87d49792008-01-19 20:57:59 +00001073 if gc_was_enabled:
1074 gc.enable()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001075 os.close(errpipe_write)
Peter Astrandf5400032007-02-02 19:06:36 +00001076 if p2cread is not None and p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001077 os.close(p2cread)
Peter Astrandf5400032007-02-02 19:06:36 +00001078 if c2pwrite is not None and c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001079 os.close(c2pwrite)
Peter Astrandf5400032007-02-02 19:06:36 +00001080 if errwrite is not None and errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001081 os.close(errwrite)
1082
1083 # Wait for exec to fail or succeed; possibly raising exception
1084 data = os.read(errpipe_read, 1048576) # Exceptions limited to 1 MB
1085 os.close(errpipe_read)
1086 if data != "":
Peter Astrandf791d7a2005-01-01 09:38:57 +00001087 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001088 child_exception = pickle.loads(data)
Georg Brandl78162da2009-02-14 17:04:26 +00001089 for fd in (p2cwrite, c2pread, errread):
1090 if fd is not None:
1091 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001092 raise child_exception
1093
1094
1095 def _handle_exitstatus(self, sts):
1096 if os.WIFSIGNALED(sts):
1097 self.returncode = -os.WTERMSIG(sts)
1098 elif os.WIFEXITED(sts):
1099 self.returncode = os.WEXITSTATUS(sts)
1100 else:
1101 # Should never happen
1102 raise RuntimeError("Unknown child exit status!")
1103
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001104
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +00001105 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001106 """Check if child process has terminated. Returns returncode
1107 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001108 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001109 try:
1110 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1111 if pid == self.pid:
1112 self._handle_exitstatus(sts)
1113 except os.error:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +00001114 if _deadstate is not None:
1115 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001116 return self.returncode
1117
1118
1119 def wait(self):
1120 """Wait for child process to terminate. Returns returncode
1121 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001122 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001123 pid, sts = os.waitpid(self.pid, 0)
1124 self._handle_exitstatus(sts)
1125 return self.returncode
1126
1127
Peter Astrand23109f02005-03-03 20:28:59 +00001128 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001129 read_set = []
1130 write_set = []
1131 stdout = None # Return
1132 stderr = None # Return
1133
1134 if self.stdin:
1135 # Flush stdio buffer. This might block, if the user has
1136 # been writing to .stdin in an uncontrolled fashion.
1137 self.stdin.flush()
1138 if input:
1139 write_set.append(self.stdin)
1140 else:
1141 self.stdin.close()
1142 if self.stdout:
1143 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001144 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001145 if self.stderr:
1146 read_set.append(self.stderr)
1147 stderr = []
1148
Peter Astrand1812f8c2007-01-07 14:34:16 +00001149 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001150 while read_set or write_set:
Gregory P. Smithf4140642008-07-06 07:16:40 +00001151 try:
1152 rlist, wlist, xlist = select.select(read_set, write_set, [])
1153 except select.error, e:
1154 if e.args[0] == errno.EINTR:
1155 continue
1156 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001157
1158 if self.stdin in wlist:
1159 # When select has indicated that the file is writable,
1160 # we can write up to PIPE_BUF bytes without risk
1161 # blocking. POSIX defines PIPE_BUF >= 512
Brett Cannon03446c42008-08-08 04:19:32 +00001162 chunk = input[input_offset : input_offset + 512]
1163 bytes_written = os.write(self.stdin.fileno(), chunk)
Tim Petersf733abb2007-01-30 03:03:46 +00001164 input_offset += bytes_written
Peter Astrand1812f8c2007-01-07 14:34:16 +00001165 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001166 self.stdin.close()
1167 write_set.remove(self.stdin)
1168
1169 if self.stdout in rlist:
1170 data = os.read(self.stdout.fileno(), 1024)
1171 if data == "":
1172 self.stdout.close()
1173 read_set.remove(self.stdout)
1174 stdout.append(data)
1175
1176 if self.stderr in rlist:
1177 data = os.read(self.stderr.fileno(), 1024)
1178 if data == "":
1179 self.stderr.close()
1180 read_set.remove(self.stderr)
1181 stderr.append(data)
1182
1183 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +00001184 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001185 stdout = ''.join(stdout)
Peter Astrandd38ddf42005-02-10 08:32:50 +00001186 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001187 stderr = ''.join(stderr)
1188
1189 # Translate newlines, if requested. We cannot let the file
1190 # object do the translation: It is based on stdio, which is
1191 # impossible to combine with select (unless forcing no
1192 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +00001193 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001194 if stdout:
1195 stdout = self._translate_newlines(stdout)
1196 if stderr:
1197 stderr = self._translate_newlines(stderr)
1198
1199 self.wait()
1200 return (stdout, stderr)
1201
Christian Heimese74c8f22008-04-19 02:23:57 +00001202 def send_signal(self, sig):
1203 """Send a signal to the process
1204 """
1205 os.kill(self.pid, sig)
1206
1207 def terminate(self):
1208 """Terminate the process with SIGTERM
1209 """
1210 self.send_signal(signal.SIGTERM)
1211
1212 def kill(self):
1213 """Kill the process with SIGKILL
1214 """
1215 self.send_signal(signal.SIGKILL)
1216
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001217
1218def _demo_posix():
1219 #
1220 # Example 1: Simple redirection: Get process list
1221 #
1222 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
1223 print "Process list:"
1224 print plist
1225
1226 #
1227 # Example 2: Change uid before executing child
1228 #
1229 if os.getuid() == 0:
1230 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1231 p.wait()
1232
1233 #
1234 # Example 3: Connecting several subprocesses
1235 #
1236 print "Looking for 'hda'..."
1237 p1 = Popen(["dmesg"], stdout=PIPE)
1238 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
1239 print repr(p2.communicate()[0])
1240
1241 #
1242 # Example 4: Catch execution error
1243 #
1244 print
1245 print "Trying a weird file..."
1246 try:
1247 print Popen(["/this/path/does/not/exist"]).communicate()
1248 except OSError, e:
1249 if e.errno == errno.ENOENT:
1250 print "The file didn't exist. I thought so..."
1251 print "Child traceback:"
1252 print e.child_traceback
1253 else:
1254 print "Error", e.errno
1255 else:
1256 print >>sys.stderr, "Gosh. No error."
1257
1258
1259def _demo_windows():
1260 #
1261 # Example 1: Connecting several subprocesses
1262 #
1263 print "Looking for 'PROMPT' in set output..."
1264 p1 = Popen("set", stdout=PIPE, shell=True)
1265 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
1266 print repr(p2.communicate()[0])
1267
1268 #
1269 # Example 2: Simple execution of program
1270 #
1271 print "Executing calc..."
1272 p = Popen("calc")
1273 p.wait()
1274
1275
1276if __name__ == "__main__":
1277 if mswindows:
1278 _demo_windows()
1279 else:
1280 _demo_posix()