blob: 4c6b763cb02a6505ea1b73f2970dcb9adf76aa05 [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-------------------
Philip Jenvey78652962009-09-29 19:18:11 +0000290pipe = os.popen("cmd", mode='r', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000291==>
Philip Jenvey78652962009-09-29 19:18:11 +0000292pipe = Popen("cmd", shell=True, bufsize=bufsize, stdout=PIPE).stdout
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000293
Philip Jenvey78652962009-09-29 19:18:11 +0000294pipe = os.popen("cmd", mode='w', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000295==>
Philip Jenvey78652962009-09-29 19:18:11 +0000296pipe = Popen("cmd", shell=True, bufsize=bufsize, stdin=PIPE).stdin
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000297
298
Philip Jenvey78652962009-09-29 19:18:11 +0000299(child_stdin, child_stdout) = os.popen2("cmd", mode, bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000300==>
Philip Jenvey78652962009-09-29 19:18:11 +0000301p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000302 stdin=PIPE, stdout=PIPE, close_fds=True)
303(child_stdin, child_stdout) = (p.stdin, p.stdout)
304
305
306(child_stdin,
307 child_stdout,
Philip Jenvey78652962009-09-29 19:18:11 +0000308 child_stderr) = os.popen3("cmd", mode, bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000309==>
Philip Jenvey78652962009-09-29 19:18:11 +0000310p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000311 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
Philip Jenvey78652962009-09-29 19:18:11 +0000317(child_stdin, child_stdout_and_stderr) = os.popen4("cmd", mode,
318 bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000319==>
Philip Jenvey78652962009-09-29 19:18:11 +0000320p = Popen("cmd", shell=True, bufsize=bufsize,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000321 stdin=PIPE, stdout=PIPE, stderr=STDOUT, close_fds=True)
322(child_stdin, child_stdout_and_stderr) = (p.stdin, p.stdout)
323
Philip Jenvey78652962009-09-29 19:18:11 +0000324On Unix, os.popen2, os.popen3 and os.popen4 also accept a sequence as
325the command to execute, in which case arguments will be passed
326directly to the program without shell intervention. This usage can be
327replaced as follows:
328
329(child_stdin, child_stdout) = os.popen2(["/bin/ls", "-l"], mode,
330 bufsize)
331==>
332p = Popen(["/bin/ls", "-l"], bufsize=bufsize, stdin=PIPE, stdout=PIPE)
333(child_stdin, child_stdout) = (p.stdin, p.stdout)
334
335Return code handling translates as follows:
336
337pipe = os.popen("cmd", 'w')
338...
339rc = pipe.close()
340if rc != None and rc % 256:
341 print "There were some errors"
342==>
343process = Popen("cmd", 'w', shell=True, stdin=PIPE)
344...
345process.stdin.close()
346if process.wait() != 0:
347 print "There were some errors"
348
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000349
350Replacing popen2.*
351------------------
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000352(child_stdout, child_stdin) = popen2.popen2("somestring", bufsize, mode)
353==>
354p = Popen(["somestring"], shell=True, bufsize=bufsize
355 stdin=PIPE, stdout=PIPE, close_fds=True)
356(child_stdout, child_stdin) = (p.stdout, p.stdin)
357
Philip Jenvey78652962009-09-29 19:18:11 +0000358On Unix, popen2 also accepts a sequence as the command to execute, in
359which case arguments will be passed directly to the program without
360shell intervention. This usage can be replaced as follows:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000361
Philip Jenvey78652962009-09-29 19:18:11 +0000362(child_stdout, child_stdin) = popen2.popen2(["mycmd", "myarg"], bufsize,
363 mode)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000364==>
365p = Popen(["mycmd", "myarg"], bufsize=bufsize,
366 stdin=PIPE, stdout=PIPE, close_fds=True)
367(child_stdout, child_stdin) = (p.stdout, p.stdin)
368
Neal Norwitzaa87fb62007-05-11 06:23:01 +0000369The popen2.Popen3 and popen2.Popen4 basically works as subprocess.Popen,
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000370except that:
371
372* subprocess.Popen raises an exception if the execution fails
373* the capturestderr argument is replaced with the stderr argument.
374* stdin=PIPE and stdout=PIPE must be specified.
375* popen2 closes all filedescriptors by default, but you have to specify
Tim Peterse718f612004-10-12 21:51:32 +0000376 close_fds=True with subprocess.Popen.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000377"""
378
379import sys
380mswindows = (sys.platform == "win32")
381
382import os
Peter Astrandc26516b2005-02-21 08:13:02 +0000383import types
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000384import traceback
Gregory P. Smith87d49792008-01-19 20:57:59 +0000385import gc
Christian Heimese74c8f22008-04-19 02:23:57 +0000386import signal
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000387
Peter Astrand454f7672005-01-01 09:36:35 +0000388# Exception classes used by this module.
Peter Astrand7d1d4362006-07-14 14:04:45 +0000389class CalledProcessError(Exception):
Peter Astrand454f7672005-01-01 09:36:35 +0000390 """This exception is raised when a process run by check_call() returns
391 a non-zero exit status. The exit status will be stored in the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000392 returncode attribute."""
393 def __init__(self, returncode, cmd):
394 self.returncode = returncode
395 self.cmd = cmd
396 def __str__(self):
397 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
Tim Peters73a9ead2006-07-18 21:55:15 +0000398
Peter Astrand454f7672005-01-01 09:36:35 +0000399
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000400if mswindows:
401 import threading
402 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000403 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000404 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000405 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
406 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
407 from win32api import GetCurrentProcess, DuplicateHandle, \
408 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000409 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000410 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000411 from win32process import CreateProcess, STARTUPINFO, \
412 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000413 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Christian Heimese74c8f22008-04-19 02:23:57 +0000414 from win32process import TerminateProcess
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000415 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000416 else:
417 from _subprocess import *
418 class STARTUPINFO:
419 dwFlags = 0
420 hStdInput = None
421 hStdOutput = None
422 hStdError = None
Georg Brandlad624892006-06-04 22:15:37 +0000423 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000424 class pywintypes:
425 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000426else:
427 import select
428 import errno
429 import fcntl
430 import pickle
431
Peter Astrand454f7672005-01-01 09:36:35 +0000432__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000433
434try:
435 MAXFD = os.sysconf("SC_OPEN_MAX")
436except:
437 MAXFD = 256
438
439# True/False does not exist on 2.2.0
Brett Cannon03446c42008-08-08 04:19:32 +0000440#try:
441# False
442#except NameError:
443# False = 0
444# True = 1
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000445
446_active = []
447
448def _cleanup():
449 for inst in _active[:]:
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000450 if inst._internal_poll(_deadstate=sys.maxint) >= 0:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000451 try:
452 _active.remove(inst)
453 except ValueError:
454 # This can happen if two threads create a new Popen instance.
455 # It's harmless that it was already removed, so ignore.
456 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000457
458PIPE = -1
459STDOUT = -2
460
461
Peter Astrand5f5e1412004-12-05 20:15:36 +0000462def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000463 """Run command with arguments. Wait for command to complete, then
464 return the returncode attribute.
465
466 The arguments are the same as for the Popen constructor. Example:
467
468 retcode = call(["ls", "-l"])
469 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000470 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000471
472
Peter Astrand454f7672005-01-01 09:36:35 +0000473def check_call(*popenargs, **kwargs):
474 """Run command with arguments. Wait for command to complete. If
475 the exit code was zero then return, otherwise raise
476 CalledProcessError. The CalledProcessError object will have the
Peter Astrand7d1d4362006-07-14 14:04:45 +0000477 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000478
479 The arguments are the same as for the Popen constructor. Example:
480
481 check_call(["ls", "-l"])
482 """
483 retcode = call(*popenargs, **kwargs)
484 cmd = kwargs.get("args")
485 if cmd is None:
486 cmd = popenargs[0]
487 if retcode:
Peter Astrand7d1d4362006-07-14 14:04:45 +0000488 raise CalledProcessError(retcode, cmd)
Peter Astrand454f7672005-01-01 09:36:35 +0000489 return retcode
490
491
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000492def list2cmdline(seq):
493 """
494 Translate a sequence of arguments into a command line
495 string, using the same rules as the MS C runtime:
496
497 1) Arguments are delimited by white space, which is either a
498 space or a tab.
499
500 2) A string surrounded by double quotation marks is
501 interpreted as a single argument, regardless of white space
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000502 or pipe characters contained within. A quoted string can be
503 embedded in an argument.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000504
505 3) A double quotation mark preceded by a backslash is
506 interpreted as a literal double quotation mark.
507
508 4) Backslashes are interpreted literally, unless they
509 immediately precede a double quotation mark.
510
511 5) If backslashes immediately precede a double quotation mark,
512 every pair of backslashes is interpreted as a literal
513 backslash. If the number of backslashes is odd, the last
514 backslash escapes the next double quotation mark as
515 described in rule 3.
516 """
517
518 # See
Eric Smithf4b581e2009-11-09 15:20:16 +0000519 # http://msdn.microsoft.com/en-us/library/17w5ykft.aspx
520 # or search http://msdn.microsoft.com for
521 # "Parsing C++ Command-Line Arguments"
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000522 result = []
523 needquote = False
524 for arg in seq:
525 bs_buf = []
526
527 # Add a space to separate this argument from the others
528 if result:
529 result.append(' ')
530
Gregory P. Smith70eb2f92008-01-19 22:49:37 +0000531 needquote = (" " in arg) or ("\t" in arg) or ("|" in arg) or not arg
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000532 if needquote:
533 result.append('"')
534
535 for c in arg:
536 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000537 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000538 bs_buf.append(c)
539 elif c == '"':
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000540 # Double backslashes.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000541 result.append('\\' * len(bs_buf)*2)
542 bs_buf = []
543 result.append('\\"')
544 else:
545 # Normal char
546 if bs_buf:
547 result.extend(bs_buf)
548 bs_buf = []
549 result.append(c)
550
Gregory P. Smithe047e6d2008-01-19 20:49:02 +0000551 # Add remaining backslashes, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000552 if bs_buf:
553 result.extend(bs_buf)
554
555 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000556 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000557 result.append('"')
558
559 return ''.join(result)
560
561
562class Popen(object):
563 def __init__(self, args, bufsize=0, executable=None,
564 stdin=None, stdout=None, stderr=None,
565 preexec_fn=None, close_fds=False, shell=False,
566 cwd=None, env=None, universal_newlines=False,
567 startupinfo=None, creationflags=0):
568 """Create new Popen instance."""
569 _cleanup()
570
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000571 self._child_created = False
Peter Astrand738131d2004-11-30 21:04:45 +0000572 if not isinstance(bufsize, (int, long)):
573 raise TypeError("bufsize must be an integer")
574
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000575 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000576 if preexec_fn is not None:
577 raise ValueError("preexec_fn is not supported on Windows "
578 "platforms")
Peter Astrand81a191b2007-05-26 22:18:20 +0000579 if close_fds and (stdin is not None or stdout is not None or
580 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000581 raise ValueError("close_fds is not supported on Windows "
Peter Astrand81a191b2007-05-26 22:18:20 +0000582 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000583 else:
584 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000585 if startupinfo is not None:
586 raise ValueError("startupinfo is only supported on Windows "
587 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000588 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000589 raise ValueError("creationflags is only supported on Windows "
590 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000591
Tim Peterse718f612004-10-12 21:51:32 +0000592 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000593 self.stdout = None
594 self.stderr = None
595 self.pid = None
596 self.returncode = None
597 self.universal_newlines = universal_newlines
598
599 # Input and output objects. The general principle is like
600 # this:
601 #
602 # Parent Child
603 # ------ -----
604 # p2cwrite ---stdin---> p2cread
605 # c2pread <--stdout--- c2pwrite
606 # errread <--stderr--- errwrite
607 #
608 # On POSIX, the child objects are file descriptors. On
609 # Windows, these are Windows file handles. The parent objects
610 # are file descriptors on both platforms. The parent objects
611 # are None when not using PIPEs. The child objects are None
612 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000613
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000614 (p2cread, p2cwrite,
615 c2pread, c2pwrite,
616 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
617
618 self._execute_child(args, executable, preexec_fn, close_fds,
619 cwd, env, universal_newlines,
620 startupinfo, creationflags, shell,
621 p2cread, p2cwrite,
622 c2pread, c2pwrite,
623 errread, errwrite)
624
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000625 if mswindows:
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000626 if p2cwrite is not None:
627 p2cwrite = msvcrt.open_osfhandle(p2cwrite.Detach(), 0)
628 if c2pread is not None:
629 c2pread = msvcrt.open_osfhandle(c2pread.Detach(), 0)
630 if errread is not None:
631 errread = msvcrt.open_osfhandle(errread.Detach(), 0)
Peter Astrand5f9c6ae2007-02-06 15:37:50 +0000632
Peter Astrandf5400032007-02-02 19:06:36 +0000633 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000634 self.stdin = os.fdopen(p2cwrite, 'wb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000635 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000636 if universal_newlines:
637 self.stdout = os.fdopen(c2pread, 'rU', bufsize)
638 else:
639 self.stdout = os.fdopen(c2pread, 'rb', bufsize)
Peter Astrandf5400032007-02-02 19:06:36 +0000640 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000641 if universal_newlines:
642 self.stderr = os.fdopen(errread, 'rU', bufsize)
643 else:
644 self.stderr = os.fdopen(errread, 'rb', bufsize)
Tim Peterse718f612004-10-12 21:51:32 +0000645
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000646
647 def _translate_newlines(self, data):
648 data = data.replace("\r\n", "\n")
649 data = data.replace("\r", "\n")
650 return data
651
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000652
Georg Brandl24522982007-04-21 20:35:38 +0000653 def __del__(self, sys=sys):
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000654 if not self._child_created:
655 # We didn't get to successfully create a child process.
656 return
657 # In case the child hasn't been waited on, check if it's done.
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000658 self._internal_poll(_deadstate=sys.maxint)
Georg Brandl13cf38c2006-07-20 16:28:39 +0000659 if self.returncode is None and _active is not None:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000660 # Child is still running, keep us alive until we can wait on it.
661 _active.append(self)
662
663
Peter Astrand23109f02005-03-03 20:28:59 +0000664 def communicate(self, input=None):
665 """Interact with process: Send data to stdin. Read data from
666 stdout and stderr, until end-of-file is reached. Wait for
667 process to terminate. The optional input argument should be a
668 string to be sent to the child process, or None, if no data
669 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000670
Peter Astrand23109f02005-03-03 20:28:59 +0000671 communicate() returns a tuple (stdout, stderr)."""
672
673 # Optimization: If we are only using one pipe, or no pipe at
674 # all, using select() or threads is unnecessary.
675 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000676 stdout = None
677 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000678 if self.stdin:
679 if input:
680 self.stdin.write(input)
681 self.stdin.close()
682 elif self.stdout:
683 stdout = self.stdout.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000684 self.stdout.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000685 elif self.stderr:
686 stderr = self.stderr.read()
Gregory P. Smith4036fd42008-05-26 20:22:14 +0000687 self.stderr.close()
Peter Astrand23109f02005-03-03 20:28:59 +0000688 self.wait()
689 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000690
Peter Astrand23109f02005-03-03 20:28:59 +0000691 return self._communicate(input)
692
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000693
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000694 def poll(self):
695 return self._internal_poll()
696
697
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000698 if mswindows:
699 #
700 # Windows methods
701 #
702 def _get_handles(self, stdin, stdout, stderr):
703 """Construct and return tupel with IO objects:
704 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
705 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000706 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000707 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000708
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000709 p2cread, p2cwrite = None, None
710 c2pread, c2pwrite = None, None
711 errread, errwrite = None, None
712
Peter Astrandd38ddf42005-02-10 08:32:50 +0000713 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000714 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000715 if p2cread is None:
716 p2cread, _ = CreatePipe(None, 0)
717 elif stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000718 p2cread, p2cwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000719 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000720 p2cread = msvcrt.get_osfhandle(stdin)
721 else:
722 # Assuming file-like object
723 p2cread = msvcrt.get_osfhandle(stdin.fileno())
724 p2cread = self._make_inheritable(p2cread)
725
Peter Astrandd38ddf42005-02-10 08:32:50 +0000726 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000727 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000728 if c2pwrite is None:
729 _, c2pwrite = CreatePipe(None, 0)
730 elif stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000731 c2pread, c2pwrite = CreatePipe(None, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000732 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000733 c2pwrite = msvcrt.get_osfhandle(stdout)
734 else:
735 # Assuming file-like object
736 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
737 c2pwrite = self._make_inheritable(c2pwrite)
738
Peter Astrandd38ddf42005-02-10 08:32:50 +0000739 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000740 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Hirokazu Yamamoto857c00e2009-03-03 22:55:00 +0000741 if errwrite is None:
742 _, errwrite = CreatePipe(None, 0)
743 elif stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000744 errread, errwrite = CreatePipe(None, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000745 elif stderr == STDOUT:
746 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000747 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000748 errwrite = msvcrt.get_osfhandle(stderr)
749 else:
750 # Assuming file-like object
751 errwrite = msvcrt.get_osfhandle(stderr.fileno())
752 errwrite = self._make_inheritable(errwrite)
753
754 return (p2cread, p2cwrite,
755 c2pread, c2pwrite,
756 errread, errwrite)
757
758
759 def _make_inheritable(self, handle):
760 """Return a duplicate of handle, which is inheritable"""
761 return DuplicateHandle(GetCurrentProcess(), handle,
762 GetCurrentProcess(), 0, 1,
763 DUPLICATE_SAME_ACCESS)
764
765
766 def _find_w9xpopen(self):
767 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000768 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
769 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000770 if not os.path.exists(w9xpopen):
771 # Eeek - file-not-found - possibly an embedding
772 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000773 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
774 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000775 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000776 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
777 "needed for Popen to work with your "
778 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000779 return w9xpopen
780
Tim Peterse718f612004-10-12 21:51:32 +0000781
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000782 def _execute_child(self, args, executable, preexec_fn, close_fds,
783 cwd, env, universal_newlines,
784 startupinfo, creationflags, shell,
785 p2cread, p2cwrite,
786 c2pread, c2pwrite,
787 errread, errwrite):
788 """Execute program (MS Windows version)"""
789
Peter Astrandc26516b2005-02-21 08:13:02 +0000790 if not isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000791 args = list2cmdline(args)
792
Peter Astrandc1d65362004-11-07 14:30:34 +0000793 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000794 if startupinfo is None:
Georg Brandlad624892006-06-04 22:15:37 +0000795 startupinfo = STARTUPINFO()
796 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000797 startupinfo.dwFlags |= STARTF_USESTDHANDLES
798 startupinfo.hStdInput = p2cread
799 startupinfo.hStdOutput = c2pwrite
800 startupinfo.hStdError = errwrite
801
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000802 if shell:
Georg Brandlad624892006-06-04 22:15:37 +0000803 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
804 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000805 comspec = os.environ.get("COMSPEC", "cmd.exe")
806 args = comspec + " /c " + args
Tim Peterse8374a52004-10-13 03:15:00 +0000807 if (GetVersion() >= 0x80000000L or
808 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000809 # Win9x, or using command.com on NT. We need to
810 # use the w9xpopen intermediate program. For more
811 # information, see KB Q150956
812 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
813 w9xpopen = self._find_w9xpopen()
814 args = '"%s" %s' % (w9xpopen, args)
815 # Not passing CREATE_NEW_CONSOLE has been known to
816 # cause random failures on win9x. Specifically a
817 # dialog: "Your program accessed mem currently in
818 # use at xxx" and a hopeful warning about the
819 # stability of your system. Cost is Ctrl+C wont
820 # kill children.
821 creationflags |= CREATE_NEW_CONSOLE
822
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000823 # Start the process
824 try:
825 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000826 # no special security
827 None, None,
Peter Astrand81a191b2007-05-26 22:18:20 +0000828 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000829 creationflags,
830 env,
831 cwd,
832 startupinfo)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000833 except pywintypes.error, e:
834 # Translate pywintypes.error to WindowsError, which is
835 # a subclass of OSError. FIXME: We should really
836 # translate errno using _sys_errlist (or simliar), but
837 # how can this be done from Python?
838 raise WindowsError(*e.args)
839
840 # Retain the process handle, but close the thread handle
Martin v. Löwis17de8ff2006-04-10 15:55:37 +0000841 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000842 self._handle = hp
843 self.pid = pid
844 ht.Close()
845
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000846 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000847 # handles that only the child should have open. You need
848 # to make sure that no handles to the write end of the
849 # output pipe are maintained in this process or else the
850 # pipe will not close when the child process exits and the
851 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000852 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000853 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000854 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000855 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000856 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000857 errwrite.Close()
858
Tim Peterse718f612004-10-12 21:51:32 +0000859
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +0000860 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000861 """Check if child process has terminated. Returns returncode
862 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000863 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000864 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
865 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000866 return self.returncode
867
868
869 def wait(self):
870 """Wait for child process to terminate. Returns returncode
871 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000872 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000873 obj = WaitForSingleObject(self._handle, INFINITE)
874 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000875 return self.returncode
876
877
878 def _readerthread(self, fh, buffer):
879 buffer.append(fh.read())
880
881
Peter Astrand23109f02005-03-03 20:28:59 +0000882 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000883 stdout = None # Return
884 stderr = None # Return
885
886 if self.stdout:
887 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000888 stdout_thread = threading.Thread(target=self._readerthread,
889 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000890 stdout_thread.setDaemon(True)
891 stdout_thread.start()
892 if self.stderr:
893 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000894 stderr_thread = threading.Thread(target=self._readerthread,
895 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000896 stderr_thread.setDaemon(True)
897 stderr_thread.start()
898
899 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000900 if input is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000901 self.stdin.write(input)
902 self.stdin.close()
903
904 if self.stdout:
905 stdout_thread.join()
906 if self.stderr:
907 stderr_thread.join()
908
909 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000910 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000911 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000912 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000913 stderr = stderr[0]
914
915 # Translate newlines, if requested. We cannot let the file
916 # object do the translation: It is based on stdio, which is
917 # impossible to combine with select (unless forcing no
918 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +0000919 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000920 if stdout:
921 stdout = self._translate_newlines(stdout)
922 if stderr:
923 stderr = self._translate_newlines(stderr)
924
925 self.wait()
926 return (stdout, stderr)
927
Christian Heimese74c8f22008-04-19 02:23:57 +0000928 def send_signal(self, sig):
929 """Send a signal to the process
930 """
931 if sig == signal.SIGTERM:
932 self.terminate()
933 else:
934 raise ValueError("Only SIGTERM is supported on Windows")
935
936 def terminate(self):
937 """Terminates the process
938 """
939 TerminateProcess(self._handle, 1)
940
941 kill = terminate
942
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000943 else:
944 #
945 # POSIX methods
946 #
947 def _get_handles(self, stdin, stdout, stderr):
948 """Construct and return tupel with IO objects:
949 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
950 """
951 p2cread, p2cwrite = None, None
952 c2pread, c2pwrite = None, None
953 errread, errwrite = None, None
954
Peter Astrandd38ddf42005-02-10 08:32:50 +0000955 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000956 pass
957 elif stdin == PIPE:
958 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000959 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000960 p2cread = stdin
961 else:
962 # Assuming file-like object
963 p2cread = stdin.fileno()
964
Peter Astrandd38ddf42005-02-10 08:32:50 +0000965 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000966 pass
967 elif stdout == PIPE:
968 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000969 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000970 c2pwrite = stdout
971 else:
972 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000973 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000974
Peter Astrandd38ddf42005-02-10 08:32:50 +0000975 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000976 pass
977 elif stderr == PIPE:
978 errread, errwrite = os.pipe()
979 elif stderr == STDOUT:
980 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000981 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000982 errwrite = stderr
983 else:
984 # Assuming file-like object
985 errwrite = stderr.fileno()
986
987 return (p2cread, p2cwrite,
988 c2pread, c2pwrite,
989 errread, errwrite)
990
991
992 def _set_cloexec_flag(self, fd):
993 try:
994 cloexec_flag = fcntl.FD_CLOEXEC
995 except AttributeError:
996 cloexec_flag = 1
997
998 old = fcntl.fcntl(fd, fcntl.F_GETFD)
999 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
1000
1001
1002 def _close_fds(self, but):
Georg Brandl309501a2008-01-19 20:22:13 +00001003 os.closerange(3, but)
1004 os.closerange(but + 1, MAXFD)
Tim Peterse718f612004-10-12 21:51:32 +00001005
1006
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001007 def _execute_child(self, args, executable, preexec_fn, close_fds,
1008 cwd, env, universal_newlines,
1009 startupinfo, creationflags, shell,
1010 p2cread, p2cwrite,
1011 c2pread, c2pwrite,
1012 errread, errwrite):
1013 """Execute program (POSIX version)"""
1014
Peter Astrandc26516b2005-02-21 08:13:02 +00001015 if isinstance(args, types.StringTypes):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001016 args = [args]
Georg Brandl6c0e1e82006-10-29 09:05:04 +00001017 else:
1018 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001019
1020 if shell:
1021 args = ["/bin/sh", "-c"] + args
1022
Peter Astrandd38ddf42005-02-10 08:32:50 +00001023 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001024 executable = args[0]
1025
1026 # For transferring possible exec failure from child to parent
1027 # The first char specifies the exception type: 0 means
1028 # OSError, 1 means some other error.
1029 errpipe_read, errpipe_write = os.pipe()
Gregory P. Smith87d49792008-01-19 20:57:59 +00001030 try:
Gregory P. Smith92ffc632008-01-19 22:23:56 +00001031 try:
Facundo Batista07638b92009-06-19 19:51:26 +00001032 self._set_cloexec_flag(errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001033
Facundo Batista07638b92009-06-19 19:51:26 +00001034 gc_was_enabled = gc.isenabled()
1035 # Disable gc to avoid bug where gc -> file_dealloc ->
1036 # write to stderr -> hang. http://bugs.python.org/issue1336
1037 gc.disable()
1038 try:
1039 self.pid = os.fork()
1040 except:
1041 if gc_was_enabled:
1042 gc.enable()
1043 raise
1044 self._child_created = True
1045 if self.pid == 0:
1046 # Child
1047 try:
1048 # Close parent's pipe ends
1049 if p2cwrite is not None:
1050 os.close(p2cwrite)
1051 if c2pread is not None:
1052 os.close(c2pread)
1053 if errread is not None:
1054 os.close(errread)
1055 os.close(errpipe_read)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001056
Facundo Batista07638b92009-06-19 19:51:26 +00001057 # Dup fds for child
1058 if p2cread is not None:
1059 os.dup2(p2cread, 0)
1060 if c2pwrite is not None:
1061 os.dup2(c2pwrite, 1)
1062 if errwrite is not None:
1063 os.dup2(errwrite, 2)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001064
Facundo Batista07638b92009-06-19 19:51:26 +00001065 # Close pipe fds. Make sure we don't close the same
1066 # fd more than once, or standard fds.
1067 if p2cread is not None and p2cread not in (0,):
1068 os.close(p2cread)
1069 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
1070 os.close(c2pwrite)
1071 if errwrite is not None and errwrite not in (p2cread, c2pwrite, 2):
1072 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001073
Facundo Batista07638b92009-06-19 19:51:26 +00001074 # Close all other fds, if asked for
1075 if close_fds:
1076 self._close_fds(but=errpipe_write)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001077
Facundo Batista07638b92009-06-19 19:51:26 +00001078 if cwd is not None:
1079 os.chdir(cwd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001080
Facundo Batista07638b92009-06-19 19:51:26 +00001081 if preexec_fn:
1082 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001083
Facundo Batista07638b92009-06-19 19:51:26 +00001084 if env is None:
1085 os.execvp(executable, args)
1086 else:
1087 os.execvpe(executable, args, env)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001088
Facundo Batista07638b92009-06-19 19:51:26 +00001089 except:
1090 exc_type, exc_value, tb = sys.exc_info()
1091 # Save the traceback and attach it to the exception object
1092 exc_lines = traceback.format_exception(exc_type,
1093 exc_value,
1094 tb)
1095 exc_value.child_traceback = ''.join(exc_lines)
1096 os.write(errpipe_write, pickle.dumps(exc_value))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001097
Facundo Batista07638b92009-06-19 19:51:26 +00001098 # This exitcode won't be reported to applications, so it
1099 # really doesn't matter what we return.
1100 os._exit(255)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001101
Facundo Batista07638b92009-06-19 19:51:26 +00001102 # Parent
1103 if gc_was_enabled:
1104 gc.enable()
1105 finally:
1106 # be sure the FD is closed no matter what
1107 os.close(errpipe_write)
1108
1109 if p2cread is not None and p2cwrite is not None:
1110 os.close(p2cread)
1111 if c2pwrite is not None and c2pread is not None:
1112 os.close(c2pwrite)
1113 if errwrite is not None and errread is not None:
1114 os.close(errwrite)
1115
1116 # Wait for exec to fail or succeed; possibly raising exception
1117 data = os.read(errpipe_read, 1048576) # Exceptions limited to 1 MB
1118 finally:
1119 # be sure the FD is closed no matter what
1120 os.close(errpipe_read)
1121
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001122 if data != "":
Peter Astrandf791d7a2005-01-01 09:38:57 +00001123 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001124 child_exception = pickle.loads(data)
Georg Brandl78162da2009-02-14 17:04:26 +00001125 for fd in (p2cwrite, c2pread, errread):
1126 if fd is not None:
1127 os.close(fd)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001128 raise child_exception
1129
1130
1131 def _handle_exitstatus(self, sts):
1132 if os.WIFSIGNALED(sts):
1133 self.returncode = -os.WTERMSIG(sts)
1134 elif os.WIFEXITED(sts):
1135 self.returncode = os.WEXITSTATUS(sts)
1136 else:
1137 # Should never happen
1138 raise RuntimeError("Unknown child exit status!")
1139
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001140
Gregory P. Smitha36f8fe2008-08-04 00:13:29 +00001141 def _internal_poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001142 """Check if child process has terminated. Returns returncode
1143 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001144 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001145 try:
1146 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1147 if pid == self.pid:
1148 self._handle_exitstatus(sts)
1149 except os.error:
Martin v. Löwis17de8ff2006-04-10 15:55:37 +00001150 if _deadstate is not None:
1151 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001152 return self.returncode
1153
1154
1155 def wait(self):
1156 """Wait for child process to terminate. Returns returncode
1157 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001158 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001159 pid, sts = os.waitpid(self.pid, 0)
1160 self._handle_exitstatus(sts)
1161 return self.returncode
1162
1163
Peter Astrand23109f02005-03-03 20:28:59 +00001164 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001165 read_set = []
1166 write_set = []
1167 stdout = None # Return
1168 stderr = None # Return
1169
1170 if self.stdin:
1171 # Flush stdio buffer. This might block, if the user has
1172 # been writing to .stdin in an uncontrolled fashion.
1173 self.stdin.flush()
1174 if input:
1175 write_set.append(self.stdin)
1176 else:
1177 self.stdin.close()
1178 if self.stdout:
1179 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001180 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001181 if self.stderr:
1182 read_set.append(self.stderr)
1183 stderr = []
1184
Peter Astrand1812f8c2007-01-07 14:34:16 +00001185 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001186 while read_set or write_set:
Gregory P. Smithf4140642008-07-06 07:16:40 +00001187 try:
1188 rlist, wlist, xlist = select.select(read_set, write_set, [])
1189 except select.error, e:
1190 if e.args[0] == errno.EINTR:
1191 continue
1192 raise
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001193
1194 if self.stdin in wlist:
1195 # When select has indicated that the file is writable,
1196 # we can write up to PIPE_BUF bytes without risk
1197 # blocking. POSIX defines PIPE_BUF >= 512
Brett Cannon03446c42008-08-08 04:19:32 +00001198 chunk = input[input_offset : input_offset + 512]
1199 bytes_written = os.write(self.stdin.fileno(), chunk)
Tim Petersf733abb2007-01-30 03:03:46 +00001200 input_offset += bytes_written
Peter Astrand1812f8c2007-01-07 14:34:16 +00001201 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001202 self.stdin.close()
1203 write_set.remove(self.stdin)
1204
1205 if self.stdout in rlist:
1206 data = os.read(self.stdout.fileno(), 1024)
1207 if data == "":
1208 self.stdout.close()
1209 read_set.remove(self.stdout)
1210 stdout.append(data)
1211
1212 if self.stderr in rlist:
1213 data = os.read(self.stderr.fileno(), 1024)
1214 if data == "":
1215 self.stderr.close()
1216 read_set.remove(self.stderr)
1217 stderr.append(data)
1218
1219 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +00001220 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001221 stdout = ''.join(stdout)
Peter Astrandd38ddf42005-02-10 08:32:50 +00001222 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001223 stderr = ''.join(stderr)
1224
1225 # Translate newlines, if requested. We cannot let the file
1226 # object do the translation: It is based on stdio, which is
1227 # impossible to combine with select (unless forcing no
1228 # buffering).
Neal Norwitza6d01ce2006-05-02 06:23:22 +00001229 if self.universal_newlines and hasattr(file, 'newlines'):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001230 if stdout:
1231 stdout = self._translate_newlines(stdout)
1232 if stderr:
1233 stderr = self._translate_newlines(stderr)
1234
1235 self.wait()
1236 return (stdout, stderr)
1237
Christian Heimese74c8f22008-04-19 02:23:57 +00001238 def send_signal(self, sig):
1239 """Send a signal to the process
1240 """
1241 os.kill(self.pid, sig)
1242
1243 def terminate(self):
1244 """Terminate the process with SIGTERM
1245 """
1246 self.send_signal(signal.SIGTERM)
1247
1248 def kill(self):
1249 """Kill the process with SIGKILL
1250 """
1251 self.send_signal(signal.SIGKILL)
1252
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001253
1254def _demo_posix():
1255 #
1256 # Example 1: Simple redirection: Get process list
1257 #
1258 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
1259 print "Process list:"
1260 print plist
1261
1262 #
1263 # Example 2: Change uid before executing child
1264 #
1265 if os.getuid() == 0:
1266 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1267 p.wait()
1268
1269 #
1270 # Example 3: Connecting several subprocesses
1271 #
1272 print "Looking for 'hda'..."
1273 p1 = Popen(["dmesg"], stdout=PIPE)
1274 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
1275 print repr(p2.communicate()[0])
1276
1277 #
1278 # Example 4: Catch execution error
1279 #
1280 print
1281 print "Trying a weird file..."
1282 try:
1283 print Popen(["/this/path/does/not/exist"]).communicate()
1284 except OSError, e:
1285 if e.errno == errno.ENOENT:
1286 print "The file didn't exist. I thought so..."
1287 print "Child traceback:"
1288 print e.child_traceback
1289 else:
1290 print "Error", e.errno
1291 else:
1292 print >>sys.stderr, "Gosh. No error."
1293
1294
1295def _demo_windows():
1296 #
1297 # Example 1: Connecting several subprocesses
1298 #
1299 print "Looking for 'PROMPT' in set output..."
1300 p1 = Popen("set", stdout=PIPE, shell=True)
1301 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
1302 print repr(p2.communicate()[0])
1303
1304 #
1305 # Example 2: Simple execution of program
1306 #
1307 print "Executing calc..."
1308 p = Popen("calc")
1309 p.wait()
1310
1311
1312if __name__ == "__main__":
1313 if mswindows:
1314 _demo_windows()
1315 else:
1316 _demo_posix()