blob: 56e05a3774392c7661b599be46df23c6e3a1203a [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*
Fredrik Lundh5b3687d2004-10-12 15:26:28 +000020commands.*
21
22Information about how the subprocess module can be used to replace these
23modules and functions can be found below.
24
25
26
27Using the subprocess module
28===========================
29This module defines one class called Popen:
30
31class Popen(args, bufsize=0, executable=None,
32 stdin=None, stdout=None, stderr=None,
33 preexec_fn=None, close_fds=False, shell=False,
34 cwd=None, env=None, universal_newlines=False,
35 startupinfo=None, creationflags=0):
36
37
38Arguments are:
39
40args should be a string, or a sequence of program arguments. The
41program to execute is normally the first item in the args sequence or
42string, but can be explicitly set by using the executable argument.
43
44On UNIX, with shell=False (default): In this case, the Popen class
45uses os.execvp() to execute the child program. args should normally
46be a sequence. A string will be treated as a sequence with the string
47as the only item (the program to execute).
48
49On UNIX, with shell=True: If args is a string, it specifies the
50command string to execute through the shell. If args is a sequence,
51the first item specifies the command string, and any additional items
52will be treated as additional shell arguments.
53
54On Windows: the Popen class uses CreateProcess() to execute the child
55program, which operates on strings. If args is a sequence, it will be
56converted to a string using the list2cmdline method. Please note that
57not all MS Windows applications interpret the command line the same
58way: The list2cmdline is designed for applications using the same
59rules as the MS C runtime.
60
61bufsize, if given, has the same meaning as the corresponding argument
62to the built-in open() function: 0 means unbuffered, 1 means line
63buffered, any other positive value means use a buffer of
64(approximately) that size. A negative bufsize means to use the system
65default, which usually means fully buffered. The default value for
66bufsize is 0 (unbuffered).
67
68stdin, stdout and stderr specify the executed programs' standard
69input, standard output and standard error file handles, respectively.
70Valid values are PIPE, an existing file descriptor (a positive
71integer), an existing file object, and None. PIPE indicates that a
72new pipe to the child should be created. With None, no redirection
73will occur; the child's file handles will be inherited from the
74parent. Additionally, stderr can be STDOUT, which indicates that the
75stderr data from the applications should be captured into the same
76file handle as for stdout.
77
78If preexec_fn is set to a callable object, this object will be called
79in the child process just before the child is executed.
80
81If close_fds is true, all file descriptors except 0, 1 and 2 will be
82closed before the child process is executed.
83
84if shell is true, the specified command will be executed through the
85shell.
86
87If cwd is not None, the current directory will be changed to cwd
88before the child is executed.
89
90If env is not None, it defines the environment variables for the new
91process.
92
93If universal_newlines is true, the file objects stdout and stderr are
94opened as a text files, but lines may be terminated by any of '\n',
95the Unix end-of-line convention, '\r', the Macintosh convention or
96'\r\n', the Windows convention. All of these external representations
97are seen as '\n' by the Python program. Note: This feature is only
98available if Python is built with universal newline support (the
99default). Also, the newlines attribute of the file objects stdout,
100stdin and stderr are not updated by the communicate() method.
101
102The startupinfo and creationflags, if given, will be passed to the
103underlying CreateProcess() function. They can specify things such as
104appearance of the main window and priority for the new process.
105(Windows only)
106
107
108This module also defines two shortcut functions:
109
Peter Astrand5f5e1412004-12-05 20:15:36 +0000110call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000111 Run command with arguments. Wait for command to complete, then
112 return the returncode attribute.
113
114 The arguments are the same as for the Popen constructor. Example:
115
116 retcode = call(["ls", "-l"])
117
Peter Astrand454f7672005-01-01 09:36:35 +0000118check_call(*popenargs, **kwargs):
119 Run command with arguments. Wait for command to complete. If the
120 exit code was zero then return, otherwise raise
121 CalledProcessError. The CalledProcessError object will have the
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000122 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000123
124 The arguments are the same as for the Popen constructor. Example:
125
126 check_call(["ls", "-l"])
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000127
128Exceptions
129----------
130Exceptions raised in the child process, before the new program has
131started to execute, will be re-raised in the parent. Additionally,
132the exception object will have one extra attribute called
133'child_traceback', which is a string containing traceback information
134from the childs point of view.
135
136The most common exception raised is OSError. This occurs, for
137example, when trying to execute a non-existent file. Applications
138should prepare for OSErrors.
139
140A ValueError will be raised if Popen is called with invalid arguments.
141
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000142check_call() will raise CalledProcessError, if the called process
143returns a non-zero return code.
Peter Astrand454f7672005-01-01 09:36:35 +0000144
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000145
146Security
147--------
148Unlike some other popen functions, this implementation will never call
149/bin/sh implicitly. This means that all characters, including shell
150metacharacters, can safely be passed to child processes.
151
152
153Popen objects
154=============
155Instances of the Popen class have the following methods:
156
157poll()
158 Check if child process has terminated. Returns returncode
159 attribute.
160
161wait()
162 Wait for child process to terminate. Returns returncode attribute.
163
164communicate(input=None)
165 Interact with process: Send data to stdin. Read data from stdout
166 and stderr, until end-of-file is reached. Wait for process to
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000167 terminate. The optional input argument should be a string to be
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000168 sent to the child process, or None, if no data should be sent to
169 the child.
Tim Peterse718f612004-10-12 21:51:32 +0000170
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000171 communicate() returns a tuple (stdout, stderr).
172
173 Note: The data read is buffered in memory, so do not use this
174 method if the data size is large or unlimited.
175
176The following attributes are also available:
177
178stdin
179 If the stdin argument is PIPE, this attribute is a file object
180 that provides input to the child process. Otherwise, it is None.
181
182stdout
183 If the stdout argument is PIPE, this attribute is a file object
184 that provides output from the child process. Otherwise, it is
185 None.
186
187stderr
188 If the stderr argument is PIPE, this attribute is file object that
189 provides error output from the child process. Otherwise, it is
190 None.
191
192pid
193 The process ID of the child process.
194
195returncode
196 The child return code. A None value indicates that the process
197 hasn't terminated yet. A negative value -N indicates that the
198 child was terminated by signal N (UNIX only).
199
200
201Replacing older functions with the subprocess module
202====================================================
203In this section, "a ==> b" means that b can be used as a replacement
204for a.
205
206Note: All functions in this section fail (more or less) silently if
207the executed program cannot be found; this module raises an OSError
208exception.
209
210In the following examples, we assume that the subprocess module is
211imported with "from subprocess import *".
212
213
214Replacing /bin/sh shell backquote
215---------------------------------
216output=`mycmd myarg`
217==>
218output = Popen(["mycmd", "myarg"], stdout=PIPE).communicate()[0]
219
220
221Replacing shell pipe line
222-------------------------
223output=`dmesg | grep hda`
224==>
225p1 = Popen(["dmesg"], stdout=PIPE)
Peter Astrand6fdf3cb2004-11-30 18:06:42 +0000226p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000227output = p2.communicate()[0]
228
229
230Replacing os.system()
231---------------------
232sts = os.system("mycmd" + " myarg")
233==>
234p = Popen("mycmd" + " myarg", shell=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000235pid, sts = os.waitpid(p.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000236
237Note:
238
239* Calling the program through the shell is usually not required.
240
241* It's easier to look at the returncode attribute than the
242 exitstatus.
243
244A more real-world example would look like this:
245
246try:
247 retcode = call("mycmd" + " myarg", shell=True)
248 if retcode < 0:
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000249 print("Child was terminated by signal", -retcode, file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000250 else:
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000251 print("Child returned", retcode, file=sys.stderr)
252except OSError as e:
253 print("Execution failed:", e, file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000254
255
256Replacing os.spawn*
257-------------------
Tim Peterse718f612004-10-12 21:51:32 +0000258P_NOWAIT example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000259
260pid = os.spawnlp(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg")
261==>
262pid = Popen(["/bin/mycmd", "myarg"]).pid
263
264
265P_WAIT example:
266
267retcode = os.spawnlp(os.P_WAIT, "/bin/mycmd", "mycmd", "myarg")
268==>
269retcode = call(["/bin/mycmd", "myarg"])
270
271
Tim Peterse718f612004-10-12 21:51:32 +0000272Vector example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000273
274os.spawnvp(os.P_NOWAIT, path, args)
275==>
276Popen([path] + args[1:])
277
278
Tim Peterse718f612004-10-12 21:51:32 +0000279Environment example:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000280
281os.spawnlpe(os.P_NOWAIT, "/bin/mycmd", "mycmd", "myarg", env)
282==>
283Popen(["/bin/mycmd", "myarg"], env={"PATH": "/usr/bin"})
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000284"""
285
286import sys
287mswindows = (sys.platform == "win32")
288
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000289import io
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000290import os
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000291import traceback
292
Peter Astrand454f7672005-01-01 09:36:35 +0000293# Exception classes used by this module.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000294class CalledProcessError(Exception):
Peter Astrand454f7672005-01-01 09:36:35 +0000295 """This exception is raised when a process run by check_call() returns
296 a non-zero exit status. The exit status will be stored in the
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000297 returncode attribute."""
298 def __init__(self, returncode, cmd):
299 self.returncode = returncode
300 self.cmd = cmd
301 def __str__(self):
302 return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode)
303
Peter Astrand454f7672005-01-01 09:36:35 +0000304
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000305if mswindows:
306 import threading
307 import msvcrt
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000308 if 0: # <-- change this to use pywin32 instead of the _subprocess driver
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000309 import pywintypes
Tim Peterse8374a52004-10-13 03:15:00 +0000310 from win32api import GetStdHandle, STD_INPUT_HANDLE, \
311 STD_OUTPUT_HANDLE, STD_ERROR_HANDLE
312 from win32api import GetCurrentProcess, DuplicateHandle, \
313 GetModuleFileName, GetVersion
Peter Astrandc1d65362004-11-07 14:30:34 +0000314 from win32con import DUPLICATE_SAME_ACCESS, SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000315 from win32pipe import CreatePipe
Tim Peterse8374a52004-10-13 03:15:00 +0000316 from win32process import CreateProcess, STARTUPINFO, \
317 GetExitCodeProcess, STARTF_USESTDHANDLES, \
Peter Astrandc1d65362004-11-07 14:30:34 +0000318 STARTF_USESHOWWINDOW, CREATE_NEW_CONSOLE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000319 from win32event import WaitForSingleObject, INFINITE, WAIT_OBJECT_0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000320 else:
321 from _subprocess import *
322 class STARTUPINFO:
323 dwFlags = 0
324 hStdInput = None
325 hStdOutput = None
326 hStdError = None
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000327 wShowWindow = 0
Fredrik Lundh3e73a012004-10-13 18:19:18 +0000328 class pywintypes:
329 error = IOError
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000330else:
331 import select
332 import errno
333 import fcntl
334 import pickle
335
Peter Astrand454f7672005-01-01 09:36:35 +0000336__all__ = ["Popen", "PIPE", "STDOUT", "call", "check_call", "CalledProcessError"]
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000337
338try:
339 MAXFD = os.sysconf("SC_OPEN_MAX")
340except:
341 MAXFD = 256
342
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000343_active = []
344
345def _cleanup():
346 for inst in _active[:]:
Christian Heimesa37d4c62007-12-04 23:02:19 +0000347 res = inst.poll(_deadstate=sys.maxsize)
Guido van Rossumb5d47ef2006-08-24 02:27:45 +0000348 if res is not None and res >= 0:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000349 try:
350 _active.remove(inst)
351 except ValueError:
352 # This can happen if two threads create a new Popen instance.
353 # It's harmless that it was already removed, so ignore.
354 pass
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000355
356PIPE = -1
357STDOUT = -2
358
359
Peter Astrand5f5e1412004-12-05 20:15:36 +0000360def call(*popenargs, **kwargs):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000361 """Run command with arguments. Wait for command to complete, then
362 return the returncode attribute.
363
364 The arguments are the same as for the Popen constructor. Example:
365
366 retcode = call(["ls", "-l"])
367 """
Peter Astrand5f5e1412004-12-05 20:15:36 +0000368 return Popen(*popenargs, **kwargs).wait()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000369
370
Peter Astrand454f7672005-01-01 09:36:35 +0000371def check_call(*popenargs, **kwargs):
372 """Run command with arguments. Wait for command to complete. If
373 the exit code was zero then return, otherwise raise
374 CalledProcessError. The CalledProcessError object will have the
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000375 return code in the returncode attribute.
Peter Astrand454f7672005-01-01 09:36:35 +0000376
377 The arguments are the same as for the Popen constructor. Example:
378
379 check_call(["ls", "-l"])
380 """
381 retcode = call(*popenargs, **kwargs)
382 cmd = kwargs.get("args")
383 if cmd is None:
384 cmd = popenargs[0]
385 if retcode:
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000386 raise CalledProcessError(retcode, cmd)
Peter Astrand454f7672005-01-01 09:36:35 +0000387 return retcode
388
389
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000390def list2cmdline(seq):
391 """
392 Translate a sequence of arguments into a command line
393 string, using the same rules as the MS C runtime:
394
395 1) Arguments are delimited by white space, which is either a
396 space or a tab.
397
398 2) A string surrounded by double quotation marks is
399 interpreted as a single argument, regardless of white space
400 contained within. A quoted string can be embedded in an
401 argument.
402
403 3) A double quotation mark preceded by a backslash is
404 interpreted as a literal double quotation mark.
405
406 4) Backslashes are interpreted literally, unless they
407 immediately precede a double quotation mark.
408
409 5) If backslashes immediately precede a double quotation mark,
410 every pair of backslashes is interpreted as a literal
411 backslash. If the number of backslashes is odd, the last
412 backslash escapes the next double quotation mark as
413 described in rule 3.
414 """
415
416 # See
417 # http://msdn.microsoft.com/library/en-us/vccelng/htm/progs_12.asp
418 result = []
419 needquote = False
420 for arg in seq:
421 bs_buf = []
422
423 # Add a space to separate this argument from the others
424 if result:
425 result.append(' ')
426
Thomas Woutersfc7bb8c2007-01-15 15:49:28 +0000427 needquote = (" " in arg) or ("\t" in arg) or arg == ""
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000428 if needquote:
429 result.append('"')
430
431 for c in arg:
432 if c == '\\':
Tim Peterse718f612004-10-12 21:51:32 +0000433 # Don't know if we need to double yet.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000434 bs_buf.append(c)
435 elif c == '"':
Tim Peterse718f612004-10-12 21:51:32 +0000436 # Double backspaces.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000437 result.append('\\' * len(bs_buf)*2)
438 bs_buf = []
439 result.append('\\"')
440 else:
441 # Normal char
442 if bs_buf:
443 result.extend(bs_buf)
444 bs_buf = []
445 result.append(c)
446
Tim Peterse718f612004-10-12 21:51:32 +0000447 # Add remaining backspaces, if any.
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000448 if bs_buf:
449 result.extend(bs_buf)
450
451 if needquote:
Peter Astrand7e78ade2005-03-03 21:10:23 +0000452 result.extend(bs_buf)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000453 result.append('"')
454
455 return ''.join(result)
456
457
458class Popen(object):
459 def __init__(self, args, bufsize=0, executable=None,
460 stdin=None, stdout=None, stderr=None,
461 preexec_fn=None, close_fds=False, shell=False,
462 cwd=None, env=None, universal_newlines=False,
463 startupinfo=None, creationflags=0):
464 """Create new Popen instance."""
465 _cleanup()
466
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000467 self._child_created = False
Guido van Rossum46a05a72007-06-07 21:56:45 +0000468 if bufsize is None:
469 bufsize = 0 # Restore default
Walter Dörwaldaa97f042007-05-03 21:05:51 +0000470 if not isinstance(bufsize, int):
Peter Astrand738131d2004-11-30 21:04:45 +0000471 raise TypeError("bufsize must be an integer")
472
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000473 if mswindows:
Tim Peterse8374a52004-10-13 03:15:00 +0000474 if preexec_fn is not None:
475 raise ValueError("preexec_fn is not supported on Windows "
476 "platforms")
Guido van Rossume7ba4952007-06-06 23:52:48 +0000477 if close_fds and (stdin is not None or stdout is not None or
478 stderr is not None):
Tim Peterse8374a52004-10-13 03:15:00 +0000479 raise ValueError("close_fds is not supported on Windows "
Guido van Rossume7ba4952007-06-06 23:52:48 +0000480 "platforms if you redirect stdin/stdout/stderr")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000481 else:
482 # POSIX
Tim Peterse8374a52004-10-13 03:15:00 +0000483 if startupinfo is not None:
484 raise ValueError("startupinfo is only supported on Windows "
485 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000486 if creationflags != 0:
Tim Peterse8374a52004-10-13 03:15:00 +0000487 raise ValueError("creationflags is only supported on Windows "
488 "platforms")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000489
Tim Peterse718f612004-10-12 21:51:32 +0000490 self.stdin = None
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000491 self.stdout = None
492 self.stderr = None
493 self.pid = None
494 self.returncode = None
495 self.universal_newlines = universal_newlines
496
497 # Input and output objects. The general principle is like
498 # this:
499 #
500 # Parent Child
501 # ------ -----
502 # p2cwrite ---stdin---> p2cread
503 # c2pread <--stdout--- c2pwrite
504 # errread <--stderr--- errwrite
505 #
506 # On POSIX, the child objects are file descriptors. On
507 # Windows, these are Windows file handles. The parent objects
508 # are file descriptors on both platforms. The parent objects
509 # are None when not using PIPEs. The child objects are None
510 # when not redirecting.
Tim Peterse718f612004-10-12 21:51:32 +0000511
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000512 (p2cread, p2cwrite,
513 c2pread, c2pwrite,
514 errread, errwrite) = self._get_handles(stdin, stdout, stderr)
515
516 self._execute_child(args, executable, preexec_fn, close_fds,
517 cwd, env, universal_newlines,
518 startupinfo, creationflags, shell,
519 p2cread, p2cwrite,
520 c2pread, c2pwrite,
521 errread, errwrite)
522
Thomas Wouterscf297e42007-02-23 15:07:44 +0000523 # On Windows, you cannot just redirect one or two handles: You
524 # either have to redirect all three or none. If the subprocess
525 # user has only redirected one or two handles, we are
526 # automatically creating PIPEs for the rest. We should close
Guido van Rossumd8faa362007-04-27 19:54:29 +0000527 # these after the process is started. See bug #1124861.
Thomas Wouterscf297e42007-02-23 15:07:44 +0000528 if mswindows:
529 if stdin is None and p2cwrite is not None:
530 os.close(p2cwrite)
531 p2cwrite = None
532 if stdout is None and c2pread is not None:
533 os.close(c2pread)
534 c2pread = None
535 if stderr is None and errread is not None:
536 os.close(errread)
537 errread = None
538
Guido van Rossumc2f93dc2007-05-24 00:50:02 +0000539 if bufsize == 0:
540 bufsize = 1 # Nearly unbuffered (XXX for now)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000541 if p2cwrite is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000542 self.stdin = io.open(p2cwrite, 'wb', bufsize)
543 if self.universal_newlines:
544 self.stdin = io.TextIOWrapper(self.stdin)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000545 if c2pread is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000546 self.stdout = io.open(c2pread, 'rb', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000547 if universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000548 self.stdout = io.TextIOWrapper(self.stdout)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000549 if errread is not None:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000550 self.stderr = io.open(errread, 'rb', bufsize)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000551 if universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +0000552 self.stderr = io.TextIOWrapper(self.stderr)
Tim Peterse718f612004-10-12 21:51:32 +0000553
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000554
Guido van Rossum98297ee2007-11-06 21:34:58 +0000555 def _translate_newlines(self, data, encoding):
556 data = data.replace(b"\r\n", b"\n").replace(b"\r", b"\n")
557 return data.decode(encoding)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000558
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000559
Guido van Rossumd8faa362007-04-27 19:54:29 +0000560 def __del__(self, sys=sys):
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000561 if not self._child_created:
562 # We didn't get to successfully create a child process.
563 return
564 # In case the child hasn't been waited on, check if it's done.
Christian Heimesa37d4c62007-12-04 23:02:19 +0000565 self.poll(_deadstate=sys.maxsize)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000566 if self.returncode is None and _active is not None:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000567 # Child is still running, keep us alive until we can wait on it.
568 _active.append(self)
569
570
Peter Astrand23109f02005-03-03 20:28:59 +0000571 def communicate(self, input=None):
572 """Interact with process: Send data to stdin. Read data from
573 stdout and stderr, until end-of-file is reached. Wait for
574 process to terminate. The optional input argument should be a
575 string to be sent to the child process, or None, if no data
576 should be sent to the child.
Tim Peterseba28be2005-03-28 01:08:02 +0000577
Peter Astrand23109f02005-03-03 20:28:59 +0000578 communicate() returns a tuple (stdout, stderr)."""
579
580 # Optimization: If we are only using one pipe, or no pipe at
581 # all, using select() or threads is unnecessary.
582 if [self.stdin, self.stdout, self.stderr].count(None) >= 2:
Tim Peterseba28be2005-03-28 01:08:02 +0000583 stdout = None
584 stderr = None
Peter Astrand23109f02005-03-03 20:28:59 +0000585 if self.stdin:
586 if input:
587 self.stdin.write(input)
588 self.stdin.close()
589 elif self.stdout:
590 stdout = self.stdout.read()
591 elif self.stderr:
592 stderr = self.stderr.read()
593 self.wait()
594 return (stdout, stderr)
Tim Peterseba28be2005-03-28 01:08:02 +0000595
Peter Astrand23109f02005-03-03 20:28:59 +0000596 return self._communicate(input)
597
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000598
599 if mswindows:
600 #
601 # Windows methods
602 #
603 def _get_handles(self, stdin, stdout, stderr):
604 """Construct and return tupel with IO objects:
605 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
606 """
Peter Astrandd38ddf42005-02-10 08:32:50 +0000607 if stdin is None and stdout is None and stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000608 return (None, None, None, None, None, None)
Tim Peterse718f612004-10-12 21:51:32 +0000609
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000610 p2cread, p2cwrite = None, None
611 c2pread, c2pwrite = None, None
612 errread, errwrite = None, None
613
Peter Astrandd38ddf42005-02-10 08:32:50 +0000614 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000615 p2cread = GetStdHandle(STD_INPUT_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000616 if p2cread is not None:
617 pass
618 elif stdin is None or stdin == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000619 p2cread, p2cwrite = CreatePipe(None, 0)
620 # Detach and turn into fd
621 p2cwrite = p2cwrite.Detach()
622 p2cwrite = msvcrt.open_osfhandle(p2cwrite, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000623 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000624 p2cread = msvcrt.get_osfhandle(stdin)
625 else:
626 # Assuming file-like object
627 p2cread = msvcrt.get_osfhandle(stdin.fileno())
628 p2cread = self._make_inheritable(p2cread)
629
Peter Astrandd38ddf42005-02-10 08:32:50 +0000630 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000631 c2pwrite = GetStdHandle(STD_OUTPUT_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000632 if c2pwrite is not None:
633 pass
634 elif stdout is None or stdout == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000635 c2pread, c2pwrite = CreatePipe(None, 0)
636 # Detach and turn into fd
637 c2pread = c2pread.Detach()
638 c2pread = msvcrt.open_osfhandle(c2pread, 0)
Peter Astrandd38ddf42005-02-10 08:32:50 +0000639 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000640 c2pwrite = msvcrt.get_osfhandle(stdout)
641 else:
642 # Assuming file-like object
643 c2pwrite = msvcrt.get_osfhandle(stdout.fileno())
644 c2pwrite = self._make_inheritable(c2pwrite)
645
Peter Astrandd38ddf42005-02-10 08:32:50 +0000646 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000647 errwrite = GetStdHandle(STD_ERROR_HANDLE)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000648 if errwrite is not None:
649 pass
650 elif stderr is None or stderr == PIPE:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000651 errread, errwrite = CreatePipe(None, 0)
652 # Detach and turn into fd
653 errread = errread.Detach()
654 errread = msvcrt.open_osfhandle(errread, 0)
655 elif stderr == STDOUT:
656 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000657 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000658 errwrite = msvcrt.get_osfhandle(stderr)
659 else:
660 # Assuming file-like object
661 errwrite = msvcrt.get_osfhandle(stderr.fileno())
662 errwrite = self._make_inheritable(errwrite)
663
664 return (p2cread, p2cwrite,
665 c2pread, c2pwrite,
666 errread, errwrite)
667
668
669 def _make_inheritable(self, handle):
670 """Return a duplicate of handle, which is inheritable"""
671 return DuplicateHandle(GetCurrentProcess(), handle,
672 GetCurrentProcess(), 0, 1,
673 DUPLICATE_SAME_ACCESS)
674
675
676 def _find_w9xpopen(self):
677 """Find and return absolut path to w9xpopen.exe"""
Tim Peterse8374a52004-10-13 03:15:00 +0000678 w9xpopen = os.path.join(os.path.dirname(GetModuleFileName(0)),
679 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000680 if not os.path.exists(w9xpopen):
681 # Eeek - file-not-found - possibly an embedding
682 # situation - see if we can locate it in sys.exec_prefix
Tim Peterse8374a52004-10-13 03:15:00 +0000683 w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix),
684 "w9xpopen.exe")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000685 if not os.path.exists(w9xpopen):
Tim Peterse8374a52004-10-13 03:15:00 +0000686 raise RuntimeError("Cannot locate w9xpopen.exe, which is "
687 "needed for Popen to work with your "
688 "shell or platform.")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000689 return w9xpopen
690
Tim Peterse718f612004-10-12 21:51:32 +0000691
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000692 def _execute_child(self, args, executable, preexec_fn, close_fds,
693 cwd, env, universal_newlines,
694 startupinfo, creationflags, shell,
695 p2cread, p2cwrite,
696 c2pread, c2pwrite,
697 errread, errwrite):
698 """Execute program (MS Windows version)"""
699
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000700 if not isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000701 args = list2cmdline(args)
702
Peter Astrandc1d65362004-11-07 14:30:34 +0000703 # Process startup details
Peter Astrandd38ddf42005-02-10 08:32:50 +0000704 if startupinfo is None:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000705 startupinfo = STARTUPINFO()
706 if None not in (p2cread, c2pwrite, errwrite):
Peter Astrandc1d65362004-11-07 14:30:34 +0000707 startupinfo.dwFlags |= STARTF_USESTDHANDLES
708 startupinfo.hStdInput = p2cread
709 startupinfo.hStdOutput = c2pwrite
710 startupinfo.hStdError = errwrite
711
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000712 if shell:
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000713 startupinfo.dwFlags |= STARTF_USESHOWWINDOW
714 startupinfo.wShowWindow = SW_HIDE
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000715 comspec = os.environ.get("COMSPEC", "cmd.exe")
716 args = comspec + " /c " + args
Guido van Rossume2a383d2007-01-15 16:59:06 +0000717 if (GetVersion() >= 0x80000000 or
Tim Peterse8374a52004-10-13 03:15:00 +0000718 os.path.basename(comspec).lower() == "command.com"):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000719 # Win9x, or using command.com on NT. We need to
720 # use the w9xpopen intermediate program. For more
721 # information, see KB Q150956
722 # (http://web.archive.org/web/20011105084002/http://support.microsoft.com/support/kb/articles/Q150/9/56.asp)
723 w9xpopen = self._find_w9xpopen()
724 args = '"%s" %s' % (w9xpopen, args)
725 # Not passing CREATE_NEW_CONSOLE has been known to
726 # cause random failures on win9x. Specifically a
727 # dialog: "Your program accessed mem currently in
728 # use at xxx" and a hopeful warning about the
729 # stability of your system. Cost is Ctrl+C wont
730 # kill children.
731 creationflags |= CREATE_NEW_CONSOLE
732
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000733 # Start the process
734 try:
735 hp, ht, pid, tid = CreateProcess(executable, args,
Tim Peterse8374a52004-10-13 03:15:00 +0000736 # no special security
737 None, None,
Guido van Rossume7ba4952007-06-06 23:52:48 +0000738 int(not close_fds),
Tim Peterse8374a52004-10-13 03:15:00 +0000739 creationflags,
740 env,
741 cwd,
742 startupinfo)
Guido van Rossumb940e112007-01-10 16:19:56 +0000743 except pywintypes.error as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000744 # Translate pywintypes.error to WindowsError, which is
745 # a subclass of OSError. FIXME: We should really
746 # translate errno using _sys_errlist (or simliar), but
747 # how can this be done from Python?
748 raise WindowsError(*e.args)
749
750 # Retain the process handle, but close the thread handle
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000751 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000752 self._handle = hp
753 self.pid = pid
754 ht.Close()
755
Andrew M. Kuchling51ee66e2004-10-12 16:38:42 +0000756 # Child is launched. Close the parent's copy of those pipe
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000757 # handles that only the child should have open. You need
758 # to make sure that no handles to the write end of the
759 # output pipe are maintained in this process or else the
760 # pipe will not close when the child process exits and the
761 # ReadFile will hang.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000762 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000763 p2cread.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000764 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000765 c2pwrite.Close()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000766 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000767 errwrite.Close()
768
Tim Peterse718f612004-10-12 21:51:32 +0000769
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000770 def poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000771 """Check if child process has terminated. Returns returncode
772 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000773 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000774 if WaitForSingleObject(self._handle, 0) == WAIT_OBJECT_0:
775 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000776 return self.returncode
777
778
779 def wait(self):
780 """Wait for child process to terminate. Returns returncode
781 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +0000782 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000783 obj = WaitForSingleObject(self._handle, INFINITE)
784 self.returncode = GetExitCodeProcess(self._handle)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000785 return self.returncode
786
787
788 def _readerthread(self, fh, buffer):
789 buffer.append(fh.read())
790
791
Peter Astrand23109f02005-03-03 20:28:59 +0000792 def _communicate(self, input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000793 stdout = None # Return
794 stderr = None # Return
795
796 if self.stdout:
797 stdout = []
Tim Peterse8374a52004-10-13 03:15:00 +0000798 stdout_thread = threading.Thread(target=self._readerthread,
799 args=(self.stdout, stdout))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000800 stdout_thread.setDaemon(True)
801 stdout_thread.start()
802 if self.stderr:
803 stderr = []
Tim Peterse8374a52004-10-13 03:15:00 +0000804 stderr_thread = threading.Thread(target=self._readerthread,
805 args=(self.stderr, stderr))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000806 stderr_thread.setDaemon(True)
807 stderr_thread.start()
808
809 if self.stdin:
Peter Astrandd38ddf42005-02-10 08:32:50 +0000810 if input is not None:
Guido van Rossumc12a8132007-10-26 04:29:23 +0000811 if isinstance(input, str):
812 input = input.encode()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000813 self.stdin.write(input)
814 self.stdin.close()
815
816 if self.stdout:
817 stdout_thread.join()
818 if self.stderr:
819 stderr_thread.join()
820
821 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +0000822 if stdout is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000823 stdout = stdout[0]
Peter Astrandd38ddf42005-02-10 08:32:50 +0000824 if stderr is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000825 stderr = stderr[0]
826
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000827 self.wait()
828 return (stdout, stderr)
829
830 else:
831 #
832 # POSIX methods
833 #
834 def _get_handles(self, stdin, stdout, stderr):
835 """Construct and return tupel with IO objects:
836 p2cread, p2cwrite, c2pread, c2pwrite, errread, errwrite
837 """
838 p2cread, p2cwrite = None, None
839 c2pread, c2pwrite = None, None
840 errread, errwrite = None, None
841
Peter Astrandd38ddf42005-02-10 08:32:50 +0000842 if stdin is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000843 pass
844 elif stdin == PIPE:
845 p2cread, p2cwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000846 elif isinstance(stdin, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000847 p2cread = stdin
848 else:
849 # Assuming file-like object
850 p2cread = stdin.fileno()
851
Peter Astrandd38ddf42005-02-10 08:32:50 +0000852 if stdout is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000853 pass
854 elif stdout == PIPE:
855 c2pread, c2pwrite = os.pipe()
Peter Astrandd38ddf42005-02-10 08:32:50 +0000856 elif isinstance(stdout, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000857 c2pwrite = stdout
858 else:
859 # Assuming file-like object
Tim Peterse718f612004-10-12 21:51:32 +0000860 c2pwrite = stdout.fileno()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000861
Peter Astrandd38ddf42005-02-10 08:32:50 +0000862 if stderr is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000863 pass
864 elif stderr == PIPE:
865 errread, errwrite = os.pipe()
866 elif stderr == STDOUT:
867 errwrite = c2pwrite
Peter Astrandd38ddf42005-02-10 08:32:50 +0000868 elif isinstance(stderr, int):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000869 errwrite = stderr
870 else:
871 # Assuming file-like object
872 errwrite = stderr.fileno()
873
874 return (p2cread, p2cwrite,
875 c2pread, c2pwrite,
876 errread, errwrite)
877
878
879 def _set_cloexec_flag(self, fd):
880 try:
881 cloexec_flag = fcntl.FD_CLOEXEC
882 except AttributeError:
883 cloexec_flag = 1
884
885 old = fcntl.fcntl(fd, fcntl.F_GETFD)
886 fcntl.fcntl(fd, fcntl.F_SETFD, old | cloexec_flag)
887
888
889 def _close_fds(self, but):
Guido van Rossum805365e2007-05-07 22:24:25 +0000890 for i in range(3, MAXFD):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000891 if i == but:
892 continue
893 try:
894 os.close(i)
895 except:
896 pass
Tim Peterse718f612004-10-12 21:51:32 +0000897
898
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000899 def _execute_child(self, args, executable, preexec_fn, close_fds,
900 cwd, env, universal_newlines,
901 startupinfo, creationflags, shell,
902 p2cread, p2cwrite,
903 c2pread, c2pwrite,
904 errread, errwrite):
905 """Execute program (POSIX version)"""
906
Guido van Rossum3172c5d2007-10-16 18:12:55 +0000907 if isinstance(args, str):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000908 args = [args]
Thomas Wouters89f507f2006-12-13 04:49:30 +0000909 else:
910 args = list(args)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000911
912 if shell:
913 args = ["/bin/sh", "-c"] + args
914
Peter Astrandd38ddf42005-02-10 08:32:50 +0000915 if executable is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000916 executable = args[0]
917
918 # For transferring possible exec failure from child to parent
919 # The first char specifies the exception type: 0 means
920 # OSError, 1 means some other error.
921 errpipe_read, errpipe_write = os.pipe()
922 self._set_cloexec_flag(errpipe_write)
923
924 self.pid = os.fork()
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000925 self._child_created = True
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000926 if self.pid == 0:
927 # Child
928 try:
929 # Close parent's pipe ends
Thomas Wouterscf297e42007-02-23 15:07:44 +0000930 if p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000931 os.close(p2cwrite)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000932 if c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000933 os.close(c2pread)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000934 if errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000935 os.close(errread)
936 os.close(errpipe_read)
937
938 # Dup fds for child
Thomas Wouterscf297e42007-02-23 15:07:44 +0000939 if p2cread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000940 os.dup2(p2cread, 0)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000941 if c2pwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000942 os.dup2(c2pwrite, 1)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000943 if errwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000944 os.dup2(errwrite, 2)
945
Thomas Wouters89f507f2006-12-13 04:49:30 +0000946 # Close pipe fds. Make sure we don't close the same
947 # fd more than once, or standard fds.
Thomas Wouterscf297e42007-02-23 15:07:44 +0000948 if p2cread is not None and p2cread not in (0,):
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000949 os.close(p2cread)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000950 if c2pwrite is not None and c2pwrite not in (p2cread, 1):
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000951 os.close(c2pwrite)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000952 if (errwrite is not None and
953 errwrite not in (p2cread, c2pwrite, 2)):
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000954 os.close(errwrite)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000955
956 # Close all other fds, if asked for
957 if close_fds:
958 self._close_fds(but=errpipe_write)
959
Peter Astrandd38ddf42005-02-10 08:32:50 +0000960 if cwd is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000961 os.chdir(cwd)
962
963 if preexec_fn:
Neal Norwitzd9108552006-03-17 08:00:19 +0000964 preexec_fn()
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000965
Peter Astrandd38ddf42005-02-10 08:32:50 +0000966 if env is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000967 os.execvp(executable, args)
968 else:
969 os.execvpe(executable, args, env)
970
971 except:
972 exc_type, exc_value, tb = sys.exc_info()
973 # Save the traceback and attach it to the exception object
Tim Peterse8374a52004-10-13 03:15:00 +0000974 exc_lines = traceback.format_exception(exc_type,
975 exc_value,
976 tb)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000977 exc_value.child_traceback = ''.join(exc_lines)
978 os.write(errpipe_write, pickle.dumps(exc_value))
979
980 # This exitcode won't be reported to applications, so it
981 # really doesn't matter what we return.
982 os._exit(255)
983
984 # Parent
985 os.close(errpipe_write)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000986 if p2cread is not None and p2cwrite is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000987 os.close(p2cread)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000988 if c2pwrite is not None and c2pread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000989 os.close(c2pwrite)
Thomas Wouterscf297e42007-02-23 15:07:44 +0000990 if errwrite is not None and errread is not None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000991 os.close(errwrite)
992
993 # Wait for exec to fail or succeed; possibly raising exception
994 data = os.read(errpipe_read, 1048576) # Exceptions limited to 1 MB
995 os.close(errpipe_read)
Guido van Rossumaf2362a2007-05-15 22:32:02 +0000996 if data:
Peter Astrandf791d7a2005-01-01 09:38:57 +0000997 os.waitpid(self.pid, 0)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +0000998 child_exception = pickle.loads(data)
999 raise child_exception
1000
1001
1002 def _handle_exitstatus(self, sts):
1003 if os.WIFSIGNALED(sts):
1004 self.returncode = -os.WTERMSIG(sts)
1005 elif os.WIFEXITED(sts):
1006 self.returncode = os.WEXITSTATUS(sts)
1007 else:
1008 # Should never happen
1009 raise RuntimeError("Unknown child exit status!")
1010
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001011
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001012 def poll(self, _deadstate=None):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001013 """Check if child process has terminated. Returns returncode
1014 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001015 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001016 try:
1017 pid, sts = os.waitpid(self.pid, os.WNOHANG)
1018 if pid == self.pid:
1019 self._handle_exitstatus(sts)
1020 except os.error:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001021 if _deadstate is not None:
1022 self.returncode = _deadstate
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001023 return self.returncode
1024
1025
1026 def wait(self):
1027 """Wait for child process to terminate. Returns returncode
1028 attribute."""
Peter Astrandd38ddf42005-02-10 08:32:50 +00001029 if self.returncode is None:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001030 pid, sts = os.waitpid(self.pid, 0)
1031 self._handle_exitstatus(sts)
1032 return self.returncode
1033
1034
Peter Astrand23109f02005-03-03 20:28:59 +00001035 def _communicate(self, input):
Guido van Rossumbae07c92007-10-08 02:46:15 +00001036 if self.stdin:
1037 if isinstance(input, str): # Unicode
1038 input = input.encode("utf-8") # XXX What else?
Guido van Rossum98297ee2007-11-06 21:34:58 +00001039 input = bytes(input)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001040 read_set = []
1041 write_set = []
1042 stdout = None # Return
1043 stderr = None # Return
1044
1045 if self.stdin:
1046 # Flush stdio buffer. This might block, if the user has
1047 # been writing to .stdin in an uncontrolled fashion.
1048 self.stdin.flush()
1049 if input:
1050 write_set.append(self.stdin)
1051 else:
1052 self.stdin.close()
1053 if self.stdout:
1054 read_set.append(self.stdout)
Tim Peterse718f612004-10-12 21:51:32 +00001055 stdout = []
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001056 if self.stderr:
1057 read_set.append(self.stderr)
1058 stderr = []
1059
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001060 input_offset = 0
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001061 while read_set or write_set:
1062 rlist, wlist, xlist = select.select(read_set, write_set, [])
1063
Guido van Rossum98297ee2007-11-06 21:34:58 +00001064 # XXX Rewrite these to use non-blocking I/O on the
1065 # file objects; they are no longer using C stdio!
1066
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001067 if self.stdin in wlist:
1068 # When select has indicated that the file is writable,
1069 # we can write up to PIPE_BUF bytes without risk
1070 # blocking. POSIX defines PIPE_BUF >= 512
Guido van Rossumbae07c92007-10-08 02:46:15 +00001071 chunk = input[input_offset : input_offset + 512]
1072 bytes_written = os.write(self.stdin.fileno(), chunk)
Thomas Wouters9fe394c2007-02-05 01:24:16 +00001073 input_offset += bytes_written
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001074 if input_offset >= len(input):
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001075 self.stdin.close()
1076 write_set.remove(self.stdin)
1077
1078 if self.stdout in rlist:
1079 data = os.read(self.stdout.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001080 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001081 self.stdout.close()
1082 read_set.remove(self.stdout)
1083 stdout.append(data)
1084
1085 if self.stderr in rlist:
1086 data = os.read(self.stderr.fileno(), 1024)
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001087 if not data:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001088 self.stderr.close()
1089 read_set.remove(self.stderr)
1090 stderr.append(data)
1091
1092 # All data exchanged. Translate lists into strings.
Peter Astrandd38ddf42005-02-10 08:32:50 +00001093 if stdout is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001094 stdout = b"".join(stdout)
Peter Astrandd38ddf42005-02-10 08:32:50 +00001095 if stderr is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001096 stderr = b"".join(stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001097
Guido van Rossum98297ee2007-11-06 21:34:58 +00001098 # Translate newlines, if requested.
1099 # This also turns bytes into strings.
Guido van Rossumc9e363c2007-05-15 23:18:55 +00001100 if self.universal_newlines:
Guido van Rossumfa0054a2007-05-24 04:05:35 +00001101 if stdout is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001102 stdout = self._translate_newlines(stdout,
1103 self.stdout.encoding)
Guido van Rossumfa0054a2007-05-24 04:05:35 +00001104 if stderr is not None:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001105 stderr = self._translate_newlines(stderr,
1106 self.stderr.encoding)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001107
1108 self.wait()
1109 return (stdout, stderr)
1110
1111
1112def _demo_posix():
1113 #
1114 # Example 1: Simple redirection: Get process list
1115 #
1116 plist = Popen(["ps"], stdout=PIPE).communicate()[0]
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001117 print("Process list:")
1118 print(plist)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001119
1120 #
1121 # Example 2: Change uid before executing child
1122 #
1123 if os.getuid() == 0:
1124 p = Popen(["id"], preexec_fn=lambda: os.setuid(100))
1125 p.wait()
1126
1127 #
1128 # Example 3: Connecting several subprocesses
1129 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001130 print("Looking for 'hda'...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001131 p1 = Popen(["dmesg"], stdout=PIPE)
1132 p2 = Popen(["grep", "hda"], stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001133 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001134
1135 #
1136 # Example 4: Catch execution error
1137 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001138 print()
1139 print("Trying a weird file...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001140 try:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001141 print(Popen(["/this/path/does/not/exist"]).communicate())
Guido van Rossumb940e112007-01-10 16:19:56 +00001142 except OSError as e:
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001143 if e.errno == errno.ENOENT:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001144 print("The file didn't exist. I thought so...")
1145 print("Child traceback:")
1146 print(e.child_traceback)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001147 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001148 print("Error", e.errno)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001149 else:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001150 print("Gosh. No error.", file=sys.stderr)
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001151
1152
1153def _demo_windows():
1154 #
1155 # Example 1: Connecting several subprocesses
1156 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001157 print("Looking for 'PROMPT' in set output...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001158 p1 = Popen("set", stdout=PIPE, shell=True)
1159 p2 = Popen('find "PROMPT"', stdin=p1.stdout, stdout=PIPE)
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001160 print(repr(p2.communicate()[0]))
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001161
1162 #
1163 # Example 2: Simple execution of program
1164 #
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001165 print("Executing calc...")
Fredrik Lundh5b3687d2004-10-12 15:26:28 +00001166 p = Popen("calc")
1167 p.wait()
1168
1169
1170if __name__ == "__main__":
1171 if mswindows:
1172 _demo_windows()
1173 else:
1174 _demo_posix()