blob: e9c9a0ef463f00d704cd13d7b37475a88a5d9e64 [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
Thomas Wouters0e3f5912006-08-11 14:57:12 +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
Thomas Wouters0e3f5912006-08-11 14:57:12 +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
Thomas Wouters902d6eb2007-01-09 23:18:33 +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)
Thomas Wouters0e3f5912006-08-11 14:57:12 +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
343The popen2.Popen3 and popen3.Popen4 basically works as subprocess.Popen,
344except 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
353"""
354
355import sys
356mswindows = (sys.platform == "win32")
357
358import os
Peter Astrandc26516b2005-02-21 08:13:02 +0000359import types
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000360import traceback
361
Peter Astrand454f7672005-01-01 09:36:35 +0000362# Exception classes used by this module.
Thomas Wouters0e3f5912006-08-11 14:57:12 +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
Thomas Wouters0e3f5912006-08-11 14:57:12 +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)
372
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
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000388 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000389 else:
390 from _subprocess import *
391 class STARTUPINFO:
392 dwFlags = 0
393 hStdInput = None
394 hStdOutput = None
395 hStdError = None
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000396 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000397 class pywintypes:
398 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000399else:
400 import select
401 import errno
402 import fcntl
403 import pickle
404
Peter Astrand454f7672005-01-01 09:36:35 +0000405__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000406
407try:
408 MAXFD = os.sysconf("SC_OPEN_MAX")
409except:
410 MAXFD = 256
411
412# True/False does not exist on 2.2.0
413try:
414 False
415except NameError:
416 False = 0
417 True = 1
418
419_active = []
420
421def _cleanup():
422 for inst in _active[:]:
Guido van Rossumb5d47ef2006-08-24 02:27:45 +0000423 res = inst.poll(_deadstate=sys.maxint)
424 if res is not None and res >= 0:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +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
Thomas Wouters0e3f5912006-08-11 14:57:12 +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:
Thomas Wouters0e3f5912006-08-11 14:57:12 +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
476 contained within. A quoted string can be embedded in an
477 argument.
478
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
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000503 needquote = (" " in arg) or ("\t" in arg) or 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 == '"':
Tim Peterse718f612004-10-12 21:51:32 +0000512 # Double backspaces.
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
Tim Peterse718f612004-10-12 21:51:32 +0000523 # Add remaining backspaces, 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
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000543 self._child_created = False
Guido van Rossume2a383d2007-01-15 16:59:06 +0000544 if not isinstance(bufsize, (int, int)):
Peter Astrand738131d2004-11-30 21:04:45 +0000545 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")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000551 if close_fds:
Tim Peterse8374a52004-10-13 03:15:00 +0000552 raise ValueError("close_fds is not supported on Windows "
553 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000554 else:
555 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000556 if startupinfo is not None:
557 raise ValueError("startupinfo is only supported on Windows "
558 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000559 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000560 raise ValueError("creationflags is only supported on Windows "
561 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000562
Tim Peterse718f612004-10-12 21:51:32 +0000563 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000564 self.stdout = None
565 self.stderr = None
566 self.pid = None
567 self.returncode = None
568 self.universal_newlines = universal_newlines
569
570 # Input and output objects. The general principle is like
571 # this:
572 #
573 # Parent Child
574 # ------ -----
575 # p2cwrite ---stdin---> p2cread
576 # c2pread <--stdout--- c2pwrite
577 # errread <--stderr--- errwrite
578 #
579 # On POSIX, the child objects are file descriptors. On
580 # Windows, these are Windows file handles. The parent objects
581 # are file descriptors on both platforms. The parent objects
582 # are None when not using PIPEs. The child objects are None
583 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000584
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000585 (p2cread, p2cwrite,
586 c2pread, c2pwrite,
587 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
588
589 self._execute_child(args, executable, preexec_fn, close_fds,
590 cwd, env, universal_newlines,
591 startupinfo, creationflags, shell,
592 p2cread, p2cwrite,
593 c2pread, c2pwrite,
594 errread, errwrite)
595
Thomas Wouterscf297e42007-02-23 15:07:44 +0000596 # On Windows, you cannot just redirect one or two handles: You
597 # either have to redirect all three or none. If the subprocess
598 # user has only redirected one or two handles, we are
599 # automatically creating PIPEs for the rest. We should close
600 # these after the process is started. See bug #1124861.
601 if mswindows:
602 if stdin is None and p2cwrite is not None:
603 os.close(p2cwrite)
604 p2cwrite = None
605 if stdout is None and c2pread is not None:
606 os.close(c2pread)
607 c2pread = None
608 if stderr is None and errread is not None:
609 os.close(errread)
610 errread = None
611
612 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000613 self.stdin = os.fdopen(p2cwrite, 'wb', bufsize)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000614 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000615 if universal_newlines:
616 self.stdout = os.fdopen(c2pread, 'rU', bufsize)
617 else:
618 self.stdout = os.fdopen(c2pread, 'rb', bufsize)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000619 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000620 if universal_newlines:
621 self.stderr = os.fdopen(errread, 'rU', bufsize)
622 else:
623 self.stderr = os.fdopen(errread, 'rb', bufsize)
Tim Peterse718f612004-10-12 21:51:32 +0000624
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000625
626 def _translate_newlines(self, data):
627 data = data.replace("\r\n", "\n")
628 data = data.replace("\r", "\n")
629 return data
630
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000631
632 def __del__(self):
633 if not self._child_created:
634 # We didn't get to successfully create a child process.
635 return
636 # In case the child hasn't been waited on, check if it's done.
637 self.poll(_deadstate=sys.maxint)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000638 if self.returncode is None and _active is not None:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000639 # Child is still running, keep us alive until we can wait on it.
640 _active.append(self)
641
642
Peter Astrand23109f02005-03-03 20:28:59 +0000643 def communicate(self, input=None):
644 """Interact with process: Send data to stdin. Read data from
645 stdout and stderr, until end-of-file is reached. Wait for
646 process to terminate. The optional input argument should be a
647 string to be sent to the child process, or None, if no data
648 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000649
Peter Astrand23109f02005-03-03 20:28:59 +0000650 communicate() returns a tuple (stdout, stderr)."""
651
652 # Optimization: If we are only using one pipe, or no pipe at
653 # all, using select() or threads is unnecessary.
654 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000655 stdout = None
656 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000657 if self.stdin:
658 if input:
659 self.stdin.write(input)
660 self.stdin.close()
661 elif self.stdout:
662 stdout = self.stdout.read()
663 elif self.stderr:
664 stderr = self.stderr.read()
665 self.wait()
666 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000667
Peter Astrand23109f02005-03-03 20:28:59 +0000668 return self._communicate(input)
669
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000670
671 if mswindows:
672 #
673 # Windows methods
674 #
675 def _get_handles(self, stdin, stdout, stderr):
676 """Construct and return tupel with IO objects:
677 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
678 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000679 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000680 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000681
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000682 p2cread, p2cwrite = None, None
683 c2pread, c2pwrite = None, None
684 errread, errwrite = None, None
685
Peter Astrandd38ddf42005-02-10 08:32:50 +0000686 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000687 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000688 if p2cread is not None:
689 pass
690 elif stdin is None or stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000691 p2cread, p2cwrite = CreatePipe(None, 0)
692 # Detach and turn into fd
693 p2cwrite = p2cwrite.Detach()
694 p2cwrite = msvcrt.open_osfhandle(p2cwrite, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000695 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000696 p2cread = msvcrt.get_osfhandle(stdin)
697 else:
698 # Assuming file-like object
699 p2cread = msvcrt.get_osfhandle(stdin.fileno())
700 p2cread = self._make_inheritable(p2cread)
701
Peter Astrandd38ddf42005-02-10 08:32:50 +0000702 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000703 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000704 if c2pwrite is not None:
705 pass
706 elif stdout is None or stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000707 c2pread, c2pwrite = CreatePipe(None, 0)
708 # Detach and turn into fd
709 c2pread = c2pread.Detach()
710 c2pread = msvcrt.open_osfhandle(c2pread, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000711 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000712 c2pwrite = msvcrt.get_osfhandle(stdout)
713 else:
714 # Assuming file-like object
715 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
716 c2pwrite = self._make_inheritable(c2pwrite)
717
Peter Astrandd38ddf42005-02-10 08:32:50 +0000718 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000719 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000720 if errwrite is not None:
721 pass
722 elif stderr is None or stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000723 errread, errwrite = CreatePipe(None, 0)
724 # Detach and turn into fd
725 errread = errread.Detach()
726 errread = msvcrt.open_osfhandle(errread, 0)
727 elif stderr == STDOUT:
728 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000729 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000730 errwrite = msvcrt.get_osfhandle(stderr)
731 else:
732 # Assuming file-like object
733 errwrite = msvcrt.get_osfhandle(stderr.fileno())
734 errwrite = self._make_inheritable(errwrite)
735
736 return (p2cread, p2cwrite,
737 c2pread, c2pwrite,
738 errread, errwrite)
739
740
741 def _make_inheritable(self, handle):
742 """Return a duplicate of handle, which is inheritable"""
743 return DuplicateHandle(GetCurrentProcess(), handle,
744 GetCurrentProcess(), 0, 1,
745 DUPLICATE_SAME_ACCESS)
746
747
748 def _find_w9xpopen(self):
749 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000750 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
751 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000752 if not os.path.exists(w9xpopen):
753 # Eeek - file-not-found - possibly an embedding
754 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000755 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
756 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000757 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000758 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
759 "needed for Popen to work with your "
760 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000761 return w9xpopen
762
Tim Peterse718f612004-10-12 21:51:32 +0000763
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000764 def _execute_child(self, args, executable, preexec_fn, close_fds,
765 cwd, env, universal_newlines,
766 startupinfo, creationflags, shell,
767 p2cread, p2cwrite,
768 c2pread, c2pwrite,
769 errread, errwrite):
770 """Execute program (MS Windows version)"""
771
Peter Astrandc26516b2005-02-21 08:13:02 +0000772 if not isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000773 args = list2cmdline(args)
774
Peter Astrandc1d65362004-11-07 14:30:34 +0000775 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000776 if startupinfo is None:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000777 startupinfo = STARTUPINFO()
778 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000779 startupinfo.dwFlags |= STARTF_USESTDHANDLES
780 startupinfo.hStdInput = p2cread
781 startupinfo.hStdOutput = c2pwrite
782 startupinfo.hStdError = errwrite
783
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000784 if shell:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000785 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
786 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000787 comspec = os.environ.get("COMSPEC", "cmd.exe")
788 args = comspec + " /c " + args
Guido van Rossume2a383d2007-01-15 16:59:06 +0000789 if (GetVersion() >= 0x80000000 or
Tim Peterse8374a52004-10-13 03:15:00 +0000790 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000791 # Win9x, or using command.com on NT. We need to
792 # use the w9xpopen intermediate program. For more
793 # information, see KB Q150956
794 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
795 w9xpopen = self._find_w9xpopen()
796 args = '"%s" %s' % (w9xpopen, args)
797 # Not passing CREATE_NEW_CONSOLE has been known to
798 # cause random failures on win9x. Specifically a
799 # dialog: "Your program accessed mem currently in
800 # use at xxx" and a hopeful warning about the
801 # stability of your system. Cost is Ctrl+C wont
802 # kill children.
803 creationflags |= CREATE_NEW_CONSOLE
804
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000805 # Start the process
806 try:
807 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000808 # no special security
809 None, None,
810 # must inherit handles to pass std
811 # handles
812 1,
813 creationflags,
814 env,
815 cwd,
816 startupinfo)
Guido van Rossumb940e112007-01-10 16:19:56 +0000817 except pywintypes.error as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000818 # Translate pywintypes.error to WindowsError, which is
819 # a subclass of OSError. FIXME: We should really
820 # translate errno using _sys_errlist (or simliar), but
821 # how can this be done from Python?
822 raise WindowsError(*e.args)
823
824 # Retain the process handle, but close the thread handle
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000825 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000826 self._handle = hp
827 self.pid = pid
828 ht.Close()
829
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000830 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000831 # handles that only the child should have open. You need
832 # to make sure that no handles to the write end of the
833 # output pipe are maintained in this process or else the
834 # pipe will not close when the child process exits and the
835 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000836 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000837 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000838 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000839 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000840 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000841 errwrite.Close()
842
Tim Peterse718f612004-10-12 21:51:32 +0000843
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000844 def poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000845 """Check if child process has terminated. Returns returncode
846 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000847 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000848 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
849 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000850 return self.returncode
851
852
853 def wait(self):
854 """Wait for child process to terminate. Returns returncode
855 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000856 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000857 obj = WaitForSingleObject(self._handle, INFINITE)
858 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000859 return self.returncode
860
861
862 def _readerthread(self, fh, buffer):
863 buffer.append(fh.read())
864
865
Peter Astrand23109f02005-03-03 20:28:59 +0000866 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000867 stdout = None # Return
868 stderr = None # Return
869
870 if self.stdout:
871 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000872 stdout_thread = threading.Thread(target=self._readerthread,
873 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000874 stdout_thread.setDaemon(True)
875 stdout_thread.start()
876 if self.stderr:
877 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000878 stderr_thread = threading.Thread(target=self._readerthread,
879 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000880 stderr_thread.setDaemon(True)
881 stderr_thread.start()
882
883 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000884 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000885 self.stdin.write(input)
886 self.stdin.close()
887
888 if self.stdout:
889 stdout_thread.join()
890 if self.stderr:
891 stderr_thread.join()
892
893 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000894 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000895 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000896 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000897 stderr = stderr[0]
898
899 # Translate newlines, if requested. We cannot let the file
900 # object do the translation: It is based on stdio, which is
901 # impossible to combine with select (unless forcing no
902 # buffering).
Thomas Wouters477c8d52006-05-27 19:21:47 +0000903 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000904 if stdout:
905 stdout = self._translate_newlines(stdout)
906 if stderr:
907 stderr = self._translate_newlines(stderr)
908
909 self.wait()
910 return (stdout, stderr)
911
912 else:
913 #
914 # POSIX methods
915 #
916 def _get_handles(self, stdin, stdout, stderr):
917 """Construct and return tupel with IO objects:
918 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
919 """
920 p2cread, p2cwrite = None, None
921 c2pread, c2pwrite = None, None
922 errread, errwrite = None, None
923
Peter Astrandd38ddf42005-02-10 08:32:50 +0000924 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000925 pass
926 elif stdin == PIPE:
927 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000928 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000929 p2cread = stdin
930 else:
931 # Assuming file-like object
932 p2cread = stdin.fileno()
933
Peter Astrandd38ddf42005-02-10 08:32:50 +0000934 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000935 pass
936 elif stdout == PIPE:
937 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000938 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000939 c2pwrite = stdout
940 else:
941 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000942 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000943
Peter Astrandd38ddf42005-02-10 08:32:50 +0000944 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000945 pass
946 elif stderr == PIPE:
947 errread, errwrite = os.pipe()
948 elif stderr == STDOUT:
949 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000950 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000951 errwrite = stderr
952 else:
953 # Assuming file-like object
954 errwrite = stderr.fileno()
955
956 return (p2cread, p2cwrite,
957 c2pread, c2pwrite,
958 errread, errwrite)
959
960
961 def _set_cloexec_flag(self, fd):
962 try:
963 cloexec_flag = fcntl.FD_CLOEXEC
964 except AttributeError:
965 cloexec_flag = 1
966
967 old = fcntl.fcntl(fd, fcntl.F_GETFD)
968 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
969
970
971 def _close_fds(self, but):
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000972 for i in xrange(3, MAXFD):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000973 if i == but:
974 continue
975 try:
976 os.close(i)
977 except:
978 pass
Tim Peterse718f612004-10-12 21:51:32 +0000979
980
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000981 def _execute_child(self, args, executable, preexec_fn, close_fds,
982 cwd, env, universal_newlines,
983 startupinfo, creationflags, shell,
984 p2cread, p2cwrite,
985 c2pread, c2pwrite,
986 errread, errwrite):
987 """Execute program (POSIX version)"""
988
Peter Astrandc26516b2005-02-21 08:13:02 +0000989 if isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000990 args = [args]
Thomas Wouters89f507f2006-12-13 04:49:30 +0000991 else:
992 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000993
994 if shell:
995 args = ["/bin/sh", "-c"] + args
996
Peter Astrandd38ddf42005-02-10 08:32:50 +0000997 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000998 executable = args[0]
999
1000 # For transferring possible exec failure from child to parent
1001 # The first char specifies the exception type: 0 means
1002 # OSError, 1 means some other error.
1003 errpipe_read, errpipe_write = os.pipe()
1004 self._set_cloexec_flag(errpipe_write)
1005
1006 self.pid = os.fork()
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001007 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001008 if self.pid == 0:
1009 # Child
1010 try:
1011 # Close parent's pipe ends
Thomas Wouterscf297e42007-02-23 15:07:44 +00001012 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001013 os.close(p2cwrite)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001014 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001015 os.close(c2pread)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001016 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001017 os.close(errread)
1018 os.close(errpipe_read)
1019
1020 # Dup fds for child
Thomas Wouterscf297e42007-02-23 15:07:44 +00001021 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001022 os.dup2(p2cread, 0)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001023 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001024 os.dup2(c2pwrite, 1)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001025 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001026 os.dup2(errwrite, 2)
1027
Thomas Wouters89f507f2006-12-13 04:49:30 +00001028 # Close pipe fds. Make sure we don't close the same
1029 # fd more than once, or standard fds.
Thomas Wouterscf297e42007-02-23 15:07:44 +00001030 if p2cread is not None and p2cread not in (0,):
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001031 os.close(p2cread)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001032 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001033 os.close(c2pwrite)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001034 if errwrite is not None and errwrite not in (p2cread, c2pwrite, 2):
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001035 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001036
1037 # Close all other fds, if asked for
1038 if close_fds:
1039 self._close_fds(but=errpipe_write)
1040
Peter Astrandd38ddf42005-02-10 08:32:50 +00001041 if cwd is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001042 os.chdir(cwd)
1043
1044 if preexec_fn:
Neal Norwitzd9108552006-03-17 08:00:19 +00001045 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001046
Peter Astrandd38ddf42005-02-10 08:32:50 +00001047 if env is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001048 os.execvp(executable, args)
1049 else:
1050 os.execvpe(executable, args, env)
1051
1052 except:
1053 exc_type, exc_value, tb = sys.exc_info()
1054 # Save the traceback and attach it to the exception object
Tim Peterse8374a52004-10-13 03:15:00 +00001055 exc_lines = traceback.format_exception(exc_type,
1056 exc_value,
1057 tb)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001058 exc_value.child_traceback = ''.join(exc_lines)
1059 os.write(errpipe_write, pickle.dumps(exc_value))
1060
1061 # This exitcode won't be reported to applications, so it
1062 # really doesn't matter what we return.
1063 os._exit(255)
1064
1065 # Parent
1066 os.close(errpipe_write)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001067 if p2cread is not None and p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001068 os.close(p2cread)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001069 if c2pwrite is not None and c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001070 os.close(c2pwrite)
Thomas Wouterscf297e42007-02-23 15:07:44 +00001071 if errwrite is not None and errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001072 os.close(errwrite)
1073
1074 # Wait for exec to fail or succeed; possibly raising exception
1075 data = os.read(errpipe_read, 1048576) # Exceptions limited to 1 MB
1076 os.close(errpipe_read)
1077 if data != "":
Peter Astrandf791d7a2005-01-01 09:38:57 +00001078 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001079 child_exception = pickle.loads(data)
1080 raise child_exception
1081
1082
1083 def _handle_exitstatus(self, sts):
1084 if os.WIFSIGNALED(sts):
1085 self.returncode = -os.WTERMSIG(sts)
1086 elif os.WIFEXITED(sts):
1087 self.returncode = os.WEXITSTATUS(sts)
1088 else:
1089 # Should never happen
1090 raise RuntimeError("Unknown child exit status!")
1091
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001092
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001093 def poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001094 """Check if child process has terminated. Returns returncode
1095 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001096 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001097 try:
1098 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1099 if pid == self.pid:
1100 self._handle_exitstatus(sts)
1101 except os.error:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001102 if _deadstate is not None:
1103 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001104 return self.returncode
1105
1106
1107 def wait(self):
1108 """Wait for child process to terminate. Returns returncode
1109 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001110 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001111 pid, sts = os.waitpid(self.pid, 0)
1112 self._handle_exitstatus(sts)
1113 return self.returncode
1114
1115
Peter Astrand23109f02005-03-03 20:28:59 +00001116 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001117 read_set = []
1118 write_set = []
1119 stdout = None # Return
1120 stderr = None # Return
1121
1122 if self.stdin:
1123 # Flush stdio buffer. This might block, if the user has
1124 # been writing to .stdin in an uncontrolled fashion.
1125 self.stdin.flush()
1126 if input:
1127 write_set.append(self.stdin)
1128 else:
1129 self.stdin.close()
1130 if self.stdout:
1131 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001132 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001133 if self.stderr:
1134 read_set.append(self.stderr)
1135 stderr = []
1136
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001137 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001138 while read_set or write_set:
1139 rlist, wlist, xlist = select.select(read_set, write_set, [])
1140
1141 if self.stdin in wlist:
1142 # When select has indicated that the file is writable,
1143 # we can write up to PIPE_BUF bytes without risk
1144 # blocking. POSIX defines PIPE_BUF >= 512
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001145 bytes_written = os.write(self.stdin.fileno(), buffer(input, input_offset, 512))
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001146 input_offset += bytes_written
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001147 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001148 self.stdin.close()
1149 write_set.remove(self.stdin)
1150
1151 if self.stdout in rlist:
1152 data = os.read(self.stdout.fileno(), 1024)
1153 if data == "":
1154 self.stdout.close()
1155 read_set.remove(self.stdout)
1156 stdout.append(data)
1157
1158 if self.stderr in rlist:
1159 data = os.read(self.stderr.fileno(), 1024)
1160 if data == "":
1161 self.stderr.close()
1162 read_set.remove(self.stderr)
1163 stderr.append(data)
1164
1165 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +00001166 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001167 stdout = ''.join(stdout)
Peter Astrandd38ddf42005-02-10 08:32:50 +00001168 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001169 stderr = ''.join(stderr)
1170
1171 # Translate newlines, if requested. We cannot let the file
1172 # object do the translation: It is based on stdio, which is
1173 # impossible to combine with select (unless forcing no
1174 # buffering).
Thomas Wouters477c8d52006-05-27 19:21:47 +00001175 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001176 if stdout:
1177 stdout = self._translate_newlines(stdout)
1178 if stderr:
1179 stderr = self._translate_newlines(stderr)
1180
1181 self.wait()
1182 return (stdout, stderr)
1183
1184
1185def _demo_posix():
1186 #
1187 # Example 1: Simple redirection: Get process list
1188 #
1189 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001190 print("Process list:")
1191 print(plist)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001192
1193 #
1194 # Example 2: Change uid before executing child
1195 #
1196 if os.getuid() == 0:
1197 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1198 p.wait()
1199
1200 #
1201 # Example 3: Connecting several subprocesses
1202 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001203 print("Looking for 'hda'...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001204 p1 = Popen(["dmesg"], stdout=PIPE)
1205 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001206 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001207
1208 #
1209 # Example 4: Catch execution error
1210 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001211 print()
1212 print("Trying a weird file...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001213 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001214 print(Popen(["/this/path/does/not/exist"]).communicate())
Guido van Rossumb940e112007-01-10 16:19:56 +00001215 except OSError as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001216 if e.errno == errno.ENOENT:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001217 print("The file didn't exist. I thought so...")
1218 print("Child traceback:")
1219 print(e.child_traceback)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001220 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001221 print("Error", e.errno)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001222 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001223 print("Gosh. No error.", file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001224
1225
1226def _demo_windows():
1227 #
1228 # Example 1: Connecting several subprocesses
1229 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001230 print("Looking for 'PROMPT' in set output...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001231 p1 = Popen("set", stdout=PIPE, shell=True)
1232 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001233 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001234
1235 #
1236 # Example 2: Simple execution of program
1237 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001238 print("Executing calc...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001239 p = Popen("calc")
1240 p.wait()
1241
1242
1243if __name__ == "__main__":
1244 if mswindows:
1245 _demo_windows()
1246 else:
1247 _demo_posix()