]>
Commit | Line | Data |
---|---|---|
1f780e48 RD |
1 | #!/usr/bin/env python |
2 | # Copyright (c) 2002-2003 ActiveState | |
3 | # See LICENSE.txt for license details. | |
4 | """ Contents of LICENSE.txt: | |
5 | Permission is hereby granted, free of charge, to any person obtaining a | |
6 | copy of this software and associated documentation files (the | |
7 | "Software"), to deal in the Software without restriction, including | |
8 | without limitation the rights to use, copy, modify, merge, publish, | |
9 | distribute, sublicense, and/or sell copies of the Software, and to | |
10 | permit persons to whom the Software is furnished to do so, subject to | |
11 | the following conditions: | |
12 | ||
13 | The above copyright notice and this permission notice shall be included | |
14 | in all copies or substantial portions of the Software. | |
15 | ||
16 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS | |
17 | OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | |
18 | MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. | |
19 | IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY | |
20 | CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, | |
21 | TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE | |
22 | SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. | |
23 | """ | |
24 | ||
25 | r""" | |
26 | Python interface for process control. | |
27 | ||
28 | This module defines three Process classes for spawning, | |
29 | communicating and control processes. They are: Process, ProcessOpen, | |
30 | ProcessProxy. All of the classes allow one to specify the command (cmd), | |
31 | starting working directory (cwd), and environment to create for the | |
32 | new process (env) and to "wait" for termination of the child and | |
33 | "kill" the child. | |
34 | ||
35 | Process: | |
36 | Use this class to simply launch a process (either a GUI app or a | |
37 | console app in a new console) with which you do not intend to | |
38 | communicate via it std handles. | |
39 | ||
40 | ProcessOpen: | |
41 | Think of this as a super version of Python's os.popen3() method. | |
42 | This spawns the given command and sets up pipes for | |
43 | stdin/stdout/stderr which can then be used to communicate with | |
44 | the child. | |
45 | ||
46 | ProcessProxy: | |
47 | This is a heavy-weight class that, similar to ProcessOpen, | |
48 | spawns the given commands and sets up pipes to the child's | |
49 | stdin/stdout/stderr. However, it also starts three threads to | |
50 | proxy communication between each of the child's and parent's std | |
51 | handles. At the parent end of this communication are, by | |
52 | default, IOBuffer objects. You may specify your own objects here | |
53 | (usually sub-classing from IOBuffer, which handles some | |
54 | synchronization issues for you). The result is that it is | |
55 | possible to have your own IOBuffer instance that gets, say, a | |
56 | .write() "event" for every write that the child does on its | |
57 | stdout. | |
58 | ||
59 | Understanding ProcessProxy is pretty complex. Some examples | |
60 | below attempt to help show some uses. Here is a diagram of the | |
61 | comminucation: | |
62 | ||
63 | <parent process> | |
64 | ,---->->->------' ^ `------>->->----, | |
65 | | | v | |
66 | IOBuffer IOBuffer IOBuffer | |
67 | (p.stdout) (p.stderr) (p.stdin) | |
68 | | | | | |
69 | _OutFileProxy _OutFileProxy _InFileProxy | |
70 | thread thread thread | |
71 | | ^ | | |
72 | `----<-<-<------, | ,------<-<-<----' | |
73 | <child process> | |
74 | ||
75 | Usage: | |
76 | import process | |
77 | p = process.<Process class>(cmd='echo hi', ...) | |
78 | #... use the various methods and attributes | |
79 | ||
80 | Examples: | |
81 | A simple 'hello world': | |
82 | >>> import process | |
83 | >>> p = process.ProcessOpen(['echo', 'hello']) | |
84 | >>> p.stdout.read() | |
85 | 'hello\r\n' | |
86 | >>> p.wait() # .wait() returns the child's exit status | |
87 | 0 | |
88 | ||
89 | Redirecting the stdout handler: | |
90 | >>> import sys | |
91 | >>> p = process.ProcessProxy(['echo', 'hello'], stdout=sys.stdout) | |
92 | hello | |
93 | ||
94 | Using stdin (need to use ProcessProxy here because it defaults to | |
95 | text-mode translation on Windows, ProcessOpen does not support | |
96 | this): | |
97 | >>> p = process.ProcessProxy(['sort']) | |
98 | >>> p.stdin.write('5\n') | |
99 | >>> p.stdin.write('2\n') | |
100 | >>> p.stdin.write('7\n') | |
101 | >>> p.stdin.close() | |
102 | >>> p.stdout.read() | |
103 | '2\n5\n7\n' | |
104 | ||
105 | Specifying environment variables: | |
106 | >>> p = process.ProcessOpen(['perl', '-e', 'print $ENV{FOO}']) | |
107 | >>> p.stdout.read() | |
108 | '' | |
109 | >>> p = process.ProcessOpen(['perl', '-e', 'print $ENV{FOO}'], | |
110 | ... env={'FOO':'bar'}) | |
111 | >>> p.stdout.read() | |
112 | 'bar' | |
113 | ||
114 | Killing a long running process (On Linux, to poll you must use | |
115 | p.wait(os.WNOHANG)): | |
116 | >>> p = ProcessOpen(['perl', '-e', 'while (1) {}']) | |
117 | >>> try: | |
118 | ... p.wait(os.WNOHANG) # poll to see if is process still running | |
119 | ... except ProcessError, ex: | |
120 | ... if ex.errno == ProcessProxy.WAIT_TIMEOUT: | |
121 | ... print "process is still running" | |
122 | ... | |
123 | process is still running | |
124 | >>> p.kill(42) | |
125 | >>> p.wait() | |
126 | 42 | |
127 | ||
128 | Providing objects for stdin/stdout/stderr: | |
129 | XXX write this, mention IOBuffer subclassing. | |
130 | """ | |
131 | #TODO: | |
132 | # - Discuss the decision to NOT have the stdout/stderr _OutFileProxy's | |
133 | # wait for process termination before closing stdin. It will just | |
134 | # close stdin when stdout is seen to have been closed. That is | |
135 | # considered Good Enough (tm). Theoretically it would be nice to | |
136 | # only abort the stdin proxying when the process terminates, but | |
137 | # watching for process termination in any of the parent's thread | |
138 | # adds the undesired condition that the parent cannot exit with the | |
139 | # child still running. That sucks. | |
140 | # XXX Note that I don't even know if the current stdout proxy even | |
141 | # closes the stdin proxy at all. | |
142 | # - DavidA: if I specify "unbuffered" for my stdin handler (in the | |
143 | # ProcessProxy constructor) then the stdin IOBuffer should do a | |
144 | # fparent.read() rather than a fparent.readline(). TrentM: can I do | |
145 | # that? What happens? | |
146 | # | |
147 | ||
148 | import os | |
149 | import sys | |
150 | import threading | |
151 | import types | |
152 | import pprint | |
153 | if sys.platform.startswith("win"): | |
154 | import msvcrt | |
155 | import win32api | |
156 | import win32file | |
157 | import win32pipe | |
158 | import pywintypes | |
159 | import win32process | |
160 | import win32event | |
161 | # constants pulled from win32con to save memory | |
162 | VER_PLATFORM_WIN32_WINDOWS = 1 | |
163 | CTRL_BREAK_EVENT = 1 | |
164 | SW_SHOWDEFAULT = 10 | |
165 | WM_CLOSE = 0x10 | |
166 | DUPLICATE_SAME_ACCESS = 2 | |
167 | ||
168 | else: | |
169 | import signal | |
170 | ||
171 | ||
172 | #---- exceptions | |
173 | ||
174 | class ProcessError(Exception): | |
175 | def __init__(self, msg, errno=-1): | |
176 | Exception.__init__(self, msg) | |
177 | self.errno = errno | |
178 | ||
179 | ||
180 | #---- internal logging facility | |
181 | ||
182 | class Logger: | |
183 | DEBUG, INFO, WARN, ERROR, FATAL = range(5) | |
184 | def __init__(self, name, level=None, streamOrFileName=sys.stderr): | |
185 | self.name = name | |
186 | if level is None: | |
187 | self.level = self.WARN | |
188 | else: | |
189 | self.level = level | |
190 | if type(streamOrFileName) == types.StringType: | |
191 | self.stream = open(streamOrFileName, 'w') | |
192 | self._opennedStream = 1 | |
193 | else: | |
194 | self.stream = streamOrFileName | |
195 | self._opennedStream = 0 | |
196 | def __del__(self): | |
197 | if self._opennedStream: | |
198 | self.stream.close() | |
199 | def _getLevelName(self, level): | |
200 | levelNameMap = { | |
201 | self.DEBUG: "DEBUG", | |
202 | self.INFO: "INFO", | |
203 | self.WARN: "WARN", | |
204 | self.ERROR: "ERROR", | |
205 | self.FATAL: "FATAL", | |
206 | } | |
207 | return levelNameMap[level] | |
208 | def log(self, level, msg, *args): | |
209 | if level < self.level: | |
210 | return | |
211 | message = "%s: %s:" % (self.name, self._getLevelName(level).lower()) | |
212 | message = message + (msg % args) + "\n" | |
213 | self.stream.write(message) | |
214 | self.stream.flush() | |
215 | def debug(self, msg, *args): | |
216 | self.log(self.DEBUG, msg, *args) | |
217 | def info(self, msg, *args): | |
218 | self.log(self.INFO, msg, *args) | |
219 | def warn(self, msg, *args): | |
220 | self.log(self.WARN, msg, *args) | |
221 | def error(self, msg, *args): | |
222 | self.log(self.ERROR, msg, *args) | |
223 | def fatal(self, msg, *args): | |
224 | self.log(self.FATAL, msg, *args) | |
225 | ||
226 | # Loggers: | |
227 | # - 'log' to log normal process handling | |
228 | # - 'logres' to track system resource life | |
229 | # - 'logfix' to track wait/kill proxying in _ThreadFixer | |
230 | if 1: # normal/production usage | |
231 | log = Logger("process", Logger.WARN) | |
232 | else: # development/debugging usage | |
233 | log = Logger("process", Logger.DEBUG, sys.stdout) | |
234 | if 1: # normal/production usage | |
235 | logres = Logger("process.res", Logger.WARN) | |
236 | else: # development/debugging usage | |
237 | logres = Logger("process.res", Logger.DEBUG, sys.stdout) | |
238 | if 1: # normal/production usage | |
239 | logfix = Logger("process.waitfix", Logger.WARN) | |
240 | else: # development/debugging usage | |
241 | logfix = Logger("process.waitfix", Logger.DEBUG, sys.stdout) | |
242 | ||
243 | ||
244 | ||
245 | #---- globals | |
246 | ||
247 | _version_ = (0, 5, 0) | |
248 | ||
249 | # List of registered processes (see _(un)registerProcess). | |
250 | _processes = [] | |
251 | ||
252 | ||
253 | ||
254 | #---- internal support routines | |
255 | ||
256 | def _escapeArg(arg): | |
257 | """Escape the given command line argument for the shell.""" | |
258 | #XXX There is a probably more that we should escape here. | |
259 | return arg.replace('"', r'\"') | |
260 | ||
261 | ||
262 | def _joinArgv(argv): | |
263 | r"""Join an arglist to a string appropriate for running. | |
264 | ||
265 | >>> import os | |
266 | >>> _joinArgv(['foo', 'bar "baz']) | |
267 | 'foo "bar \\"baz"' | |
268 | """ | |
269 | cmdstr = "" | |
270 | for arg in argv: | |
271 | if ' ' in arg or ';' in arg: | |
272 | cmdstr += '"%s"' % _escapeArg(arg) | |
273 | else: | |
274 | cmdstr += _escapeArg(arg) | |
275 | cmdstr += ' ' | |
276 | if cmdstr.endswith(' '): cmdstr = cmdstr[:-1] # strip trailing space | |
277 | return cmdstr | |
278 | ||
279 | ||
280 | def _getPathFromEnv(env): | |
281 | """Return the PATH environment variable or None. | |
282 | ||
283 | Do the right thing for case sensitivity per platform. | |
284 | XXX Icky. This guarantee of proper case sensitivity of environment | |
285 | variables should be done more fundamentally in this module. | |
286 | """ | |
287 | if sys.platform.startswith("win"): | |
288 | for key in env.keys(): | |
289 | if key.upper() == "PATH": | |
290 | return env[key] | |
291 | else: | |
292 | return None | |
293 | else: | |
294 | if env.has_key("PATH"): | |
295 | return env["PATH"] | |
296 | else: | |
297 | return None | |
298 | ||
299 | ||
300 | def _whichFirstArg(cmd, env=None): | |
301 | """Return the given command ensuring that the first arg (the command to | |
302 | launch) is a full path to an existing file. | |
303 | ||
304 | Raise a ProcessError if no such executable could be found. | |
305 | """ | |
306 | # Parse out the first arg. | |
307 | if cmd.startswith('"'): | |
308 | # The .replace() is to ensure it does not mistakenly find the | |
309 | # second '"' in, say (escaped quote): | |
310 | # "C:\foo\"bar" arg1 arg2 | |
311 | idx = cmd.replace('\\"', 'XX').find('"', 1) | |
312 | if idx == -1: | |
313 | raise ProcessError("Malformed command: %r" % cmd) | |
314 | first, rest = cmd[1:idx], cmd[idx+1:] | |
315 | rest = rest.lstrip() | |
316 | else: | |
317 | if ' ' in cmd: | |
318 | first, rest = cmd.split(' ', 1) | |
319 | else: | |
320 | first, rest = cmd, "" | |
321 | ||
322 | # Ensure the first arg is a valid path to the appropriate file. | |
323 | import which | |
324 | if os.sep in first: | |
325 | altpath = [os.path.dirname(first)] | |
326 | firstbase = os.path.basename(first) | |
327 | candidates = list(which.which(firstbase, path=altpath)) | |
328 | elif env: | |
329 | altpath = _getPathFromEnv(env) | |
330 | if altpath: | |
331 | candidates = list(which.which(first, altpath.split(os.pathsep))) | |
332 | else: | |
333 | candidates = list(which.which(first)) | |
334 | else: | |
335 | candidates = list(which.which(first)) | |
336 | if candidates: | |
337 | return _joinArgv( [candidates[0]] ) + ' ' + rest | |
338 | else: | |
339 | raise ProcessError("Could not find an appropriate leading command "\ | |
340 | "for: %r" % cmd) | |
341 | ||
342 | ||
343 | if sys.platform.startswith("win"): | |
344 | def _SaferCreateProcess(appName, # app name | |
345 | cmd, # command line | |
346 | processSA, # process security attributes | |
347 | threadSA, # thread security attributes | |
348 | inheritHandles, # are handles are inherited | |
349 | creationFlags, # creation flags | |
350 | env, # environment | |
351 | cwd, # current working directory | |
352 | si): # STARTUPINFO pointer | |
353 | """If CreateProcess fails from environment type inconsistency then | |
354 | fix that and try again. | |
355 | ||
356 | win32process.CreateProcess requires that all environment keys and | |
357 | values either be all ASCII or all unicode. Try to remove this burden | |
358 | from the user of process.py. | |
359 | """ | |
360 | isWin9x = win32api.GetVersionEx()[3] == VER_PLATFORM_WIN32_WINDOWS | |
361 | # On Win9x all keys and values of 'env' must be ASCII (XXX | |
362 | # Actually this is probably only true if the Unicode support | |
363 | # libraries, which are not installed by default, are not | |
364 | # installed). On other Windows flavours all keys and values of | |
365 | # 'env' must all be ASCII *or* all Unicode. We will try to | |
366 | # automatically convert to the appropriate type, issuing a | |
367 | # warning if such an automatic conversion is necessary. | |
368 | ||
369 | #XXX Komodo 2.0 Beta 1 hack. This requirement should be | |
370 | # pushed out to Komodo code using process.py. Or should it? | |
371 | if isWin9x and env: | |
372 | aenv = {} | |
373 | for key, value in env.items(): | |
374 | aenv[str(key)] = str(value) | |
375 | env = aenv | |
376 | ||
377 | log.debug("""\ | |
378 | _SaferCreateProcess(appName=%r, | |
379 | cmd=%r, | |
380 | env=%r, | |
381 | cwd=%r) | |
382 | os.getcwd(): %r | |
383 | """, appName, cmd, env, cwd, os.getcwd()) | |
384 | try: | |
385 | hProcess, hThread, processId, threadId\ | |
386 | = win32process.CreateProcess(appName, cmd, processSA, | |
387 | threadSA, inheritHandles, | |
388 | creationFlags, env, cwd, si) | |
389 | except TypeError, ex: | |
390 | if ex.args == ('All dictionary items must be strings, or all must be unicode',): | |
391 | # Try again with an all unicode environment. | |
392 | #XXX Would be nice if didn't have to depend on the error | |
393 | # string to catch this. | |
394 | #XXX Removing this warning for 2.3 release. See bug | |
395 | # 23215. The right fix is to correct the PHPAppInfo | |
396 | # stuff to heed the warning. | |
397 | #import warnings | |
398 | #warnings.warn('env: ' + str(ex), stacklevel=4) | |
399 | if isWin9x and env: | |
400 | aenv = {} | |
401 | try: | |
402 | for key, value in env.items(): | |
403 | aenv[str(key)] = str(value) | |
404 | except UnicodeError, ex: | |
405 | raise ProcessError(str(ex)) | |
406 | env = aenv | |
407 | elif env: | |
408 | uenv = {} | |
409 | for key, val in env.items(): | |
410 | uenv[unicode(key)] = unicode(val) | |
411 | env = uenv | |
412 | hProcess, hThread, processId, threadId\ | |
413 | = win32process.CreateProcess(appName, cmd, processSA, | |
414 | threadSA, inheritHandles, | |
415 | creationFlags, env, cwd, | |
416 | si) | |
417 | else: | |
418 | raise | |
419 | return hProcess, hThread, processId, threadId | |
420 | ||
421 | ||
422 | # Maintain references to all spawned ProcessProxy objects to avoid hangs. | |
423 | # Otherwise, if the user lets the a ProcessProxy object go out of | |
424 | # scope before the process has terminated, it is possible to get a | |
425 | # hang (at least it *used* to be so when we had the | |
426 | # win32api.CloseHandle(<stdin handle>) call in the __del__() method). | |
427 | # XXX Is this hang possible on Linux as well? | |
428 | # A reference is removed from this list when the process's .wait or | |
429 | # .kill method is called. | |
430 | # XXX Should an atexit() handler be registered to kill all curently | |
431 | # running processes? Else *could* get hangs, n'est ce pas? | |
432 | def _registerProcess(process): | |
433 | global _processes | |
434 | log.info("_registerprocess(process=%r)", process) | |
435 | ||
436 | # Clean up zombie processes. | |
437 | # If the user does not call .wait() or .kill() on processes then | |
438 | # the ProcessProxy object will not get cleaned up until Python | |
439 | # exits and _processes goes out of scope. Under heavy usage that | |
440 | # is a big memory waste. Cleaning up here alleviates that. | |
441 | for p in _processes[:]: # use copy of _process, because we may modifiy it | |
442 | try: | |
443 | # poll to see if is process still running | |
444 | if sys.platform.startswith("win"): | |
445 | timeout = 0 | |
446 | else: | |
447 | timeout = os.WNOHANG | |
448 | p.wait(timeout) | |
449 | _unregisterProcess(p) | |
450 | except ProcessError, ex: | |
451 | if ex.errno == ProcessProxy.WAIT_TIMEOUT: | |
452 | pass | |
453 | else: | |
454 | raise | |
455 | ||
456 | _processes.append(process) | |
457 | ||
458 | def _unregisterProcess(process): | |
459 | global _processes | |
460 | log.info("_unregisterProcess(process=%r)", process) | |
461 | try: | |
462 | _processes.remove(process) | |
463 | del process | |
464 | except ValueError: | |
465 | pass | |
466 | ||
467 | ||
468 | def _fixupCommand(cmd, env=None): | |
469 | """Fixup the command string so it is launchable via CreateProcess. | |
470 | ||
471 | One cannot just launch, say "python", via CreateProcess. A full path | |
472 | to an executable is required. In general there are two choices: | |
473 | 1. Launch the command string via the shell. The shell will find | |
474 | the fullpath to the appropriate executable. This shell will | |
475 | also be able to execute special shell commands, like "dir", | |
476 | which don't map to an actual executable. | |
477 | 2. Find the fullpath to the appropriate executable manually and | |
478 | launch that exe. | |
479 | ||
480 | Option (1) is preferred because you don't have to worry about not | |
481 | exactly duplicating shell behaviour and you get the added bonus of | |
482 | being able to launch "dir" and friends. | |
483 | ||
484 | However, (1) is not always an option. Doing so when the shell is | |
485 | command.com (as on all Win9x boxes) or when using WinNT's cmd.exe, | |
486 | problems are created with .kill() because these shells seem to eat | |
487 | up Ctrl-C's and Ctrl-Break's sent via | |
488 | win32api.GenerateConsoleCtrlEvent(). Strangely this only happens | |
489 | when spawn via this Python interface. For example, Ctrl-C get | |
490 | through to hang.exe here: | |
491 | C:\> ...\w9xpopen.exe "C:\WINDOWS\COMMAND.COM /c hang.exe" | |
492 | ^C | |
493 | but not here: | |
494 | >>> p = ProcessOpen('hang.exe') | |
495 | # This results in the same command to CreateProcess as | |
496 | # above. | |
497 | >>> p.kill() | |
498 | ||
499 | Hence, for these platforms we fallback to option (2). Cons: | |
500 | - cannot spawn shell commands like 'dir' directly | |
501 | - cannot spawn batch files | |
502 | """ | |
503 | if sys.platform.startswith("win"): | |
504 | # Fixup the command string to spawn. (Lifted from | |
505 | # posixmodule.c::_PyPopenCreateProcess() with some modifications) | |
506 | comspec = os.environ.get("COMSPEC", None) | |
507 | win32Version = win32api.GetVersion() | |
508 | if comspec is None: | |
509 | raise ProcessError("Cannot locate a COMSPEC environment "\ | |
510 | "variable to use as the shell") | |
511 | # Explicitly check if we are using COMMAND.COM. If we | |
512 | # are then use the w9xpopen hack. | |
513 | elif (win32Version & 0x80000000L == 0) and\ | |
514 | (win32Version & 0x5L >= 5) and\ | |
515 | os.path.basename(comspec).lower() != "command.com": | |
516 | # 2000/XP and not using command.com. | |
517 | if '"' in cmd or "'" in cmd: | |
518 | cmd = comspec + ' /c "%s"' % cmd | |
519 | else: | |
520 | cmd = comspec + ' /c ' + cmd | |
521 | elif (win32Version & 0x80000000L == 0) and\ | |
522 | (win32Version & 0x5L < 5) and\ | |
523 | os.path.basename(comspec).lower() != "command.com": | |
524 | # NT and not using command.com. | |
525 | try: | |
526 | cmd = _whichFirstArg(cmd, env) | |
527 | except ProcessError: | |
528 | raise ProcessError("Could not find a suitable executable "\ | |
529 | "to launch for '%s'. On WinNT you must manually prefix "\ | |
530 | "shell commands and batch files with 'cmd.exe /c' to "\ | |
531 | "have the shell run them." % cmd) | |
532 | else: | |
533 | # Oh gag, we're on Win9x and/or using COMMAND.COM. Use the | |
534 | # workaround listed in KB: Q150956 | |
535 | w9xpopen = os.path.join( | |
536 | os.path.dirname(win32api.GetModuleFileName(0)), | |
537 | 'w9xpopen.exe') | |
538 | if not os.path.exists(w9xpopen): | |
539 | # Eeek - file-not-found - possibly an embedding | |
540 | # situation - see if we can locate it in sys.exec_prefix | |
541 | w9xpopen = os.path.join(os.path.dirname(sys.exec_prefix), | |
542 | 'w9xpopen.exe') | |
543 | if not os.path.exists(w9xpopen): | |
544 | raise ProcessError(\ | |
545 | "Can not locate 'w9xpopen.exe' which is needed "\ | |
546 | "for ProcessOpen to work with your shell or "\ | |
547 | "platform.") | |
548 | ## This would be option (1): | |
549 | #cmd = '%s "%s /c %s"'\ | |
550 | # % (w9xpopen, comspec, cmd.replace('"', '\\"')) | |
551 | try: | |
552 | cmd = _whichFirstArg(cmd, env) | |
553 | except ProcessError: | |
554 | raise ProcessError("Could not find a suitable executable "\ | |
555 | "to launch for '%s'. On Win9x you must manually prefix "\ | |
556 | "shell commands and batch files with 'command.com /c' "\ | |
557 | "to have the shell run them." % cmd) | |
558 | cmd = '%s "%s"' % (w9xpopen, cmd.replace('"', '\\"')) | |
559 | return cmd | |
560 | ||
561 | class _FileWrapper: | |
562 | """Wrap a system file object, hiding some nitpicky details. | |
563 | ||
564 | This class provides a Python file-like interface to either a Python | |
565 | file object (pretty easy job), a file descriptor, or an OS-specific | |
566 | file handle (e.g. Win32 handles to file objects on Windows). Any or | |
567 | all of these object types may be passed to this wrapper. If more | |
568 | than one is specified this wrapper prefers to work with certain one | |
569 | in this order: | |
570 | - file descriptor (because usually this allows for | |
571 | return-immediately-on-read-if-anything-available semantics and | |
572 | also provides text mode translation on Windows) | |
573 | - OS-specific handle (allows for the above read semantics) | |
574 | - file object (buffering can cause difficulty for interacting | |
575 | with spawned programs) | |
576 | ||
577 | It also provides a place where related such objects can be kept | |
578 | alive together to prevent premature ref-counted collection. (E.g. on | |
579 | Windows a Python file object may be associated with a Win32 file | |
580 | handle. If the file handle is not kept alive the Python file object | |
581 | will cease to function.) | |
582 | """ | |
583 | def __init__(self, file=None, descriptor=None, handle=None): | |
584 | self._file = file | |
585 | self._descriptor = descriptor | |
586 | self._handle = handle | |
587 | self._closed = 0 | |
588 | if self._descriptor is not None or self._handle is not None: | |
589 | self._lineBuf = "" # to support .readline() | |
590 | ||
591 | def __del__(self): | |
592 | self.close() | |
593 | ||
594 | def __getattr__(self, name): | |
595 | """Forward to the underlying file object.""" | |
596 | if self._file is not None: | |
597 | return getattr(self._file, name) | |
598 | else: | |
599 | raise ProcessError("no file object to pass '%s' attribute to" | |
600 | % name) | |
601 | ||
602 | def _win32Read(self, nBytes): | |
603 | try: | |
604 | log.info("[%s] _FileWrapper.read: waiting for read on pipe", | |
605 | id(self)) | |
606 | errCode, text = win32file.ReadFile(self._handle, nBytes) | |
607 | except pywintypes.error, ex: | |
608 | # Ignore errors for now, like "The pipe is being closed.", | |
609 | # etc. XXX There *may* be errors we don't want to avoid. | |
610 | log.info("[%s] _FileWrapper.read: error reading from pipe: %s", | |
611 | id(self), ex) | |
612 | return "" | |
613 | assert errCode == 0,\ | |
614 | "Why is 'errCode' from ReadFile non-zero? %r" % errCode | |
615 | if not text: | |
616 | # Empty text signifies that the pipe has been closed on | |
617 | # the parent's end. | |
618 | log.info("[%s] _FileWrapper.read: observed close of parent", | |
619 | id(self)) | |
620 | # Signal the child so it knows to stop listening. | |
621 | self.close() | |
622 | return "" | |
623 | else: | |
624 | log.info("[%s] _FileWrapper.read: read %d bytes from pipe: %r", | |
625 | id(self), len(text), text) | |
626 | return text | |
627 | ||
628 | def read(self, nBytes=-1): | |
629 | # nBytes <= 0 means "read everything" | |
630 | # Note that we are changing the "read everything" cue to | |
631 | # include 0, because actually doing | |
632 | # win32file.ReadFile(<handle>, 0) results in every subsequent | |
633 | # read returning 0, i.e. it shuts down the pipe. | |
634 | if self._descriptor is not None: | |
635 | if nBytes <= 0: | |
636 | text, self._lineBuf = self._lineBuf, "" | |
637 | while 1: | |
638 | t = os.read(self._descriptor, 4092) | |
639 | if not t: | |
640 | break | |
641 | else: | |
642 | text += t | |
643 | else: | |
644 | if len(self._lineBuf) >= nBytes: | |
645 | text, self._lineBuf =\ | |
646 | self._lineBuf[:nBytes], self._lineBuf[nBytes:] | |
647 | else: | |
648 | nBytesToGo = nBytes - len(self._lineBuf) | |
649 | text = self._lineBuf + os.read(self._descriptor, | |
650 | nBytesToGo) | |
651 | self._lineBuf = "" | |
652 | return text | |
653 | elif self._handle is not None: | |
654 | if nBytes <= 0: | |
655 | text, self._lineBuf = self._lineBuf, "" | |
656 | while 1: | |
657 | t = self._win32Read(4092) | |
658 | if not t: | |
659 | break | |
660 | else: | |
661 | text += t | |
662 | else: | |
663 | if len(self._lineBuf) >= nBytes: | |
664 | text, self._lineBuf =\ | |
665 | self._lineBuf[:nBytes], self._lineBuf[nBytes:] | |
666 | else: | |
667 | nBytesToGo = nBytes - len(self._lineBuf) | |
668 | text, self._lineBuf =\ | |
669 | self._lineBuf + self._win32Read(nBytesToGo), "" | |
670 | return text | |
671 | elif self._file is not None: | |
672 | return self._file.read(nBytes) | |
673 | else: | |
674 | raise "FileHandle.read: no handle to read with" | |
675 | ||
676 | def readline(self): | |
677 | if self._descriptor is not None or self._handle is not None: | |
678 | while 1: | |
679 | #XXX This is not portable to the Mac. | |
680 | idx = self._lineBuf.find('\n') | |
681 | if idx != -1: | |
682 | line, self._lineBuf =\ | |
683 | self._lineBuf[:idx+1], self._lineBuf[idx+1:] | |
684 | break | |
685 | else: | |
686 | lengthBefore = len(self._lineBuf) | |
687 | t = self.read(4092) | |
688 | if len(t) <= lengthBefore: # no new data was read | |
689 | line, self._lineBuf = self._lineBuf, "" | |
690 | break | |
691 | else: | |
692 | self._lineBuf += t | |
693 | return line | |
694 | elif self._file is not None: | |
695 | return self._file.readline() | |
696 | else: | |
697 | raise "FileHandle.readline: no handle to read with" | |
698 | ||
699 | def readlines(self): | |
700 | if self._descriptor is not None or self._handle is not None: | |
701 | lines = [] | |
702 | while 1: | |
703 | line = self.readline() | |
704 | if line: | |
705 | lines.append(line) | |
706 | else: | |
707 | break | |
708 | return lines | |
709 | elif self._file is not None: | |
710 | return self._file.readlines() | |
711 | else: | |
712 | raise "FileHandle.readline: no handle to read with" | |
713 | ||
714 | def write(self, text): | |
715 | if self._descriptor is not None: | |
716 | os.write(self._descriptor, text) | |
717 | elif self._handle is not None: | |
718 | try: | |
719 | errCode, nBytesWritten = win32file.WriteFile(self._handle, text) | |
720 | except pywintypes.error, ex: | |
721 | # Ingore errors like "The pipe is being closed.", for | |
722 | # now. | |
723 | log.info("[%s] _FileWrapper.write: error writing to pipe, "\ | |
724 | "ignored", id(self)) | |
725 | return | |
726 | assert errCode == 0,\ | |
727 | "Why is 'errCode' from WriteFile non-zero? %r" % errCode | |
728 | if not nBytesWritten: | |
729 | # No bytes written signifies that the pipe has been | |
730 | # closed on the child's end. | |
731 | log.info("[%s] _FileWrapper.write: observed close of pipe", | |
732 | id(self)) | |
733 | return | |
734 | else: | |
735 | log.info("[%s] _FileWrapper.write: wrote %d bytes to pipe: %r", | |
736 | id(self), len(text), text) | |
737 | elif self._file is not None: | |
738 | self._file.write(text) | |
739 | else: | |
740 | raise "FileHandle.write: nothing to write with" | |
741 | ||
742 | def close(self): | |
743 | """Close all associated file objects and handles.""" | |
744 | log.debug("[%s] _FileWrapper.close()", id(self)) | |
745 | if not self._closed: | |
746 | self._closed = 1 | |
747 | if self._file is not None: | |
748 | log.debug("[%s] _FileWrapper.close: close file", id(self)) | |
749 | self._file.close() | |
750 | log.debug("[%s] _FileWrapper.close: done file close", id(self)) | |
751 | if self._descriptor is not None: | |
752 | try: | |
753 | os.close(self._descriptor) | |
754 | except OSError, ex: | |
755 | if ex.errno == 9: | |
756 | # Ignore: OSError: [Errno 9] Bad file descriptor | |
757 | # XXX *Should* we be ignoring this? It appears very | |
758 | # *in*frequently in test_wait.py. | |
759 | log.debug("[%s] _FileWrapper.close: closing "\ | |
760 | "descriptor raised OSError", id(self)) | |
761 | else: | |
762 | raise | |
763 | if self._handle is not None: | |
764 | log.debug("[%s] _FileWrapper.close: close handle", id(self)) | |
765 | try: | |
766 | win32api.CloseHandle(self._handle) | |
767 | except win32api.error: | |
768 | log.debug("[%s] _FileWrapper.close: closing handle raised", | |
769 | id(self)) | |
770 | pass | |
771 | log.debug("[%s] _FileWrapper.close: done closing handle", | |
772 | id(self)) | |
773 | ||
774 | def __repr__(self): | |
775 | return "<_FileWrapper: file:%r fd:%r os_handle:%r>"\ | |
776 | % (self._file, self._descriptor, self._handle) | |
777 | ||
778 | ||
779 | class _CountingCloser: | |
780 | """Call .close() on the given object after own .close() is called | |
781 | the precribed number of times. | |
782 | """ | |
783 | def __init__(self, objectsToClose, count): | |
784 | """ | |
785 | "objectsToClose" is a list of object on which to call .close(). | |
786 | "count" is the number of times this object's .close() method | |
787 | must be called before .close() is called on the given objects. | |
788 | """ | |
789 | self.objectsToClose = objectsToClose | |
790 | self.count = count | |
791 | if self.count <= 0: | |
792 | raise ProcessError("illegal 'count' value: %s" % self.count) | |
793 | ||
794 | def close(self): | |
795 | self.count -= 1 | |
796 | log.debug("[%d] _CountingCloser.close(): count=%d", id(self), | |
797 | self.count) | |
798 | if self.count == 0: | |
799 | for objectToClose in self.objectsToClose: | |
800 | objectToClose.close() | |
801 | ||
802 | ||
803 | ||
804 | #---- public interface | |
805 | ||
806 | class Process: | |
807 | """Create a process. | |
808 | ||
809 | One can optionally specify the starting working directory, the | |
810 | process environment, and std handles to have the child process | |
811 | inherit (all defaults are the parent's current settings). 'wait' and | |
812 | 'kill' method allow for control of the child's termination. | |
813 | """ | |
814 | # TODO: | |
815 | # - Rename this or merge it with ProcessOpen somehow. | |
816 | # | |
817 | if sys.platform.startswith("win"): | |
818 | # .wait() argument constants | |
819 | INFINITE = win32event.INFINITE | |
820 | # .wait() return error codes | |
821 | WAIT_FAILED = win32event.WAIT_FAILED | |
822 | WAIT_TIMEOUT = win32event.WAIT_TIMEOUT | |
823 | # creation "flags" constants | |
824 | # XXX Should drop these and just document usage of | |
825 | # win32process.CREATE_* constants on windows. | |
826 | CREATE_NEW_CONSOLE = win32process.CREATE_NEW_CONSOLE | |
827 | else: | |
828 | # .wait() argument constants | |
829 | INFINITE = 0 | |
830 | # .wait() return error codes | |
831 | WAIT_TIMEOUT = 258 | |
832 | WAIT_FAILED = -1 | |
833 | # creation "flags" constants | |
834 | CREATE_NEW_CONSOLE = 0x10 # same as win32process.CREATE_NEW_CONSOLE | |
835 | ||
836 | def __init__(self, cmd, cwd=None, env=None, flags=0): | |
837 | """Create a child process. | |
838 | ||
839 | "cmd" is a command string or argument vector to spawn. | |
840 | "cwd" is a working directory in which to start the child process. | |
841 | "env" is an environment dictionary for the child. | |
842 | "flags" are system-specific process creation flags. On Windows | |
843 | this can be a bitwise-OR of any of the win32process.CREATE_* | |
844 | constants (Note: win32process.CREATE_NEW_PROCESS_GROUP is always | |
845 | OR'd in). On Unix, this is currently ignored. | |
846 | """ | |
847 | log.info("Process.__init__(cmd=%r, cwd=%r, env=%r, flags=%r)", | |
848 | cmd, cwd, env, flags) | |
849 | self._cmd = cmd | |
850 | if not self._cmd: | |
851 | raise ProcessError("You must specify a command.") | |
852 | self._cwd = cwd | |
853 | self._env = env | |
854 | self._flags = flags | |
855 | if sys.platform.startswith("win"): | |
856 | self._flags |= win32process.CREATE_NEW_PROCESS_GROUP | |
857 | ||
858 | if sys.platform.startswith("win"): | |
859 | self._startOnWindows() | |
860 | else: | |
861 | self.__retvalCache = None | |
862 | self._startOnUnix() | |
863 | ||
864 | def _runChildOnUnix(self): | |
865 | #XXX Errors running the child do *not* get communicated back. | |
866 | ||
867 | #XXX Perhaps we should *always* prefix with '/bin/sh -c'? There is a | |
868 | # disparity btwn how this works on Linux and Windows. | |
869 | if isinstance(self._cmd, types.StringTypes): | |
870 | # This is easier than trying to reproduce shell interpretation to | |
871 | # separate the arguments. | |
872 | cmd = ['/bin/sh', '-c', self._cmd] | |
873 | else: | |
874 | cmd = self._cmd | |
875 | ||
876 | # Close all file descriptors (except std*) inherited from the parent. | |
877 | MAXFD = 256 # Max number of file descriptors (os.getdtablesize()???) | |
878 | for i in range(3, MAXFD): | |
879 | try: | |
880 | os.close(i) | |
881 | except OSError: | |
882 | pass | |
883 | ||
884 | try: | |
885 | if self._env: | |
886 | os.execvpe(cmd[0], cmd, self._env) | |
887 | else: | |
888 | os.execvp(cmd[0], cmd) | |
889 | finally: | |
890 | os._exit(1) # Should never get here. | |
891 | ||
892 | def _forkAndExecChildOnUnix(self): | |
893 | """Fork and start the child process. | |
894 | ||
895 | Sets self._pid as a side effect. | |
896 | """ | |
897 | pid = os.fork() | |
898 | if pid == 0: # child | |
899 | self._runChildOnUnix() | |
900 | # parent | |
901 | self._pid = pid | |
902 | ||
903 | def _startOnUnix(self): | |
904 | if self._cwd: | |
905 | oldDir = os.getcwd() | |
906 | try: | |
907 | os.chdir(self._cwd) | |
908 | except OSError, ex: | |
909 | raise ProcessError(msg=str(ex), errno=ex.errno) | |
910 | self._forkAndExecChildOnUnix() | |
911 | ||
912 | # parent | |
913 | if self._cwd: | |
914 | os.chdir(oldDir) | |
915 | ||
916 | def _startOnWindows(self): | |
917 | if type(self._cmd) in (types.ListType, types.TupleType): | |
918 | # And arg vector was passed in. | |
919 | cmd = _joinArgv(self._cmd) | |
920 | else: | |
921 | cmd = self._cmd | |
922 | ||
923 | si = win32process.STARTUPINFO() | |
924 | si.dwFlags = win32process.STARTF_USESHOWWINDOW | |
925 | si.wShowWindow = SW_SHOWDEFAULT | |
926 | ||
927 | if not (self._flags & self.CREATE_NEW_CONSOLE): | |
928 | #XXX This is hacky. | |
929 | # We cannot then use _fixupCommand because this will cause a | |
930 | # shell to be openned as the command is launched. Therefore need | |
931 | # to ensure be have the full path to the executable to launch. | |
932 | try: | |
933 | cmd = _whichFirstArg(cmd, self._env) | |
934 | except ProcessError: | |
935 | # Could not find the command, perhaps it is an internal | |
936 | # shell command -- fallback to _fixupCommand | |
937 | cmd = _fixupCommand(cmd, self._env) | |
938 | else: | |
939 | cmd = _fixupCommand(cmd, self._env) | |
940 | log.debug("cmd = %r", cmd) | |
941 | ||
942 | # Start the child process. | |
943 | try: | |
944 | self._hProcess, self._hThread, self._processId, self._threadId\ | |
945 | = _SaferCreateProcess( | |
946 | None, # app name | |
947 | cmd, # command line | |
948 | None, # process security attributes | |
949 | None, # primary thread security attributes | |
950 | 0, # handles are inherited | |
951 | self._flags, # creation flags | |
952 | self._env, # environment | |
953 | self._cwd, # current working directory | |
954 | si) # STARTUPINFO pointer | |
955 | win32api.CloseHandle(self._hThread) | |
956 | except win32api.error, ex: | |
957 | raise ProcessError(msg="Error creating process for '%s': %s"\ | |
958 | % (cmd, ex.args[2]), | |
959 | errno=ex.args[0]) | |
960 | ||
961 | def wait(self, timeout=None): | |
962 | """Wait for the started process to complete. | |
963 | ||
964 | "timeout" (on Windows) is a floating point number of seconds after | |
965 | which to timeout. Default is win32event.INFINITE. | |
966 | "timeout" (on Unix) is akin to the os.waitpid() "options" argument | |
967 | (os.WNOHANG may be used to return immediately if the process has | |
968 | not exited). Default is 0, i.e. wait forever. | |
969 | ||
970 | If the wait time's out it will raise a ProcessError. Otherwise it | |
971 | will return the child's exit value (on Windows) or the child's exit | |
972 | status excoded as per os.waitpid() (on Linux): | |
973 | "a 16-bit number, whose low byte is the signal number that killed | |
974 | the process, and whose high byte is the exit status (if the | |
975 | signal number is zero); the high bit of the low byte is set if a | |
976 | core file was produced." | |
977 | In the latter case, use the os.W*() methods to interpret the return | |
978 | value. | |
979 | """ | |
980 | # XXX Or should returning the exit value be move out to another | |
981 | # function as on Win32 process control? If so, then should | |
982 | # perhaps not make WaitForSingleObject semantic transformation. | |
983 | if sys.platform.startswith("win"): | |
984 | if timeout is None: | |
985 | timeout = win32event.INFINITE | |
986 | else: | |
987 | timeout = timeout * 1000.0 # Win32 API's timeout is in millisecs | |
988 | ||
989 | rc = win32event.WaitForSingleObject(self._hProcess, timeout) | |
990 | if rc == win32event.WAIT_FAILED: | |
991 | raise ProcessError("'WAIT_FAILED' when waiting for process to "\ | |
992 | "terminate: %r" % self._cmd, rc) | |
993 | elif rc == win32event.WAIT_TIMEOUT: | |
994 | raise ProcessError("'WAIT_TIMEOUT' when waiting for process to "\ | |
995 | "terminate: %r" % self._cmd, rc) | |
996 | ||
997 | retval = win32process.GetExitCodeProcess(self._hProcess) | |
998 | else: | |
999 | # os.waitpid() will raise: | |
1000 | # OSError: [Errno 10] No child processes | |
1001 | # on subsequent .wait() calls. Change these semantics to have | |
1002 | # subsequent .wait() calls return the exit status and return | |
1003 | # immediately without raising an exception. | |
1004 | # (XXX It would require synchronization code to handle the case | |
1005 | # of multiple simultaneous .wait() requests, however we can punt | |
1006 | # on that because it is moot while Linux still has the problem | |
1007 | # for which _ThreadFixer() exists.) | |
1008 | if self.__retvalCache is not None: | |
1009 | retval = self.__retvalCache | |
1010 | else: | |
1011 | if timeout is None: | |
1012 | timeout = 0 | |
1013 | pid, sts = os.waitpid(self._pid, timeout) | |
1014 | if pid == self._pid: | |
1015 | self.__retvalCache = retval = sts | |
1016 | else: | |
1017 | raise ProcessError("Wait for process timed out.", | |
1018 | self.WAIT_TIMEOUT) | |
1019 | return retval | |
1020 | ||
1021 | def kill(self, exitCode=0, gracePeriod=1.0, sig=None): | |
1022 | """Kill process. | |
1023 | ||
1024 | "exitCode" [deprecated, not supported] (Windows only) is the | |
1025 | code the terminated process should exit with. | |
1026 | "gracePeriod" (Windows only) is a number of seconds the process is | |
1027 | allowed to shutdown with a WM_CLOSE signal before a hard | |
1028 | terminate is called. | |
1029 | "sig" (Unix only) is the signal to use to kill the process. Defaults | |
1030 | to signal.SIGKILL. See os.kill() for more information. | |
1031 | ||
1032 | Windows: | |
1033 | Try for an orderly shutdown via WM_CLOSE. If still running | |
1034 | after gracePeriod (1 sec. default), terminate. | |
1035 | """ | |
1036 | if sys.platform.startswith("win"): | |
1037 | import win32gui | |
1038 | # Send WM_CLOSE to windows in this process group. | |
1039 | win32gui.EnumWindows(self._close_, 0) | |
1040 | ||
1041 | # Send Ctrl-Break signal to all processes attached to this | |
1042 | # console. This is supposed to trigger shutdown handlers in | |
1043 | # each of the processes. | |
1044 | try: | |
1045 | win32api.GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, | |
1046 | self._processId) | |
1047 | except AttributeError: | |
1048 | log.warn("The win32api module does not have "\ | |
1049 | "GenerateConsoleCtrlEvent(). This may mean that "\ | |
1050 | "parts of this process group have NOT been killed.") | |
1051 | except win32api.error, ex: | |
1052 | if ex.args[0] not in (6, 87): | |
1053 | # Ignore the following: | |
1054 | # api_error: (87, 'GenerateConsoleCtrlEvent', 'The parameter is incorrect.') | |
1055 | # api_error: (6, 'GenerateConsoleCtrlEvent', 'The handle is invalid.') | |
1056 | # Get error 6 if there is no console. | |
1057 | raise | |
1058 | ||
1059 | # Last resort: call TerminateProcess if it has not yet. | |
1060 | retval = 0 | |
1061 | try: | |
1062 | self.wait(gracePeriod) | |
1063 | except ProcessError, ex: | |
1064 | log.info("[%s] Process.kill: calling TerminateProcess", id(self)) | |
1065 | win32process.TerminateProcess(self._hProcess, -1) | |
1066 | win32api.Sleep(100) # wait for resources to be released | |
1067 | ||
1068 | else: | |
1069 | if sig is None: | |
1070 | sig = signal.SIGKILL | |
1071 | try: | |
1072 | os.kill(self._pid, sig) | |
1073 | except OSError, ex: | |
1074 | if ex.errno != 3: | |
1075 | # Ignore: OSError: [Errno 3] No such process | |
1076 | raise | |
1077 | ||
1078 | def _close_(self, hwnd, dummy): | |
1079 | """Callback used by .kill() on Windows. | |
1080 | ||
1081 | EnumWindows callback - sends WM_CLOSE to any window owned by this | |
1082 | process. | |
1083 | """ | |
1084 | threadId, processId = win32process.GetWindowThreadProcessId(hwnd) | |
1085 | if processId == self._processId: | |
1086 | import win32gui | |
1087 | win32gui.PostMessage(hwnd, WM_CLOSE, 0, 0) | |
1088 | ||
1089 | ||
1090 | class ProcessOpen(Process): | |
1091 | """Create a process and setup pipes to it standard handles. | |
1092 | ||
1093 | This is a super popen3. | |
1094 | """ | |
1095 | # TODO: | |
1096 | # - Share some implementation with Process and ProcessProxy. | |
1097 | # | |
1098 | ||
1099 | def __init__(self, cmd, mode='t', cwd=None, env=None): | |
1100 | """Create a Process with proxy threads for each std handle. | |
1101 | ||
1102 | "cmd" is the command string or argument vector to run. | |
1103 | "mode" (Windows only) specifies whether the pipes used to communicate | |
1104 | with the child are openned in text, 't', or binary, 'b', mode. | |
1105 | This is ignored on platforms other than Windows. Default is 't'. | |
1106 | "cwd" optionally specifies the directory in which the child process | |
1107 | should be started. Default is None, a.k.a. inherits the cwd from | |
1108 | the parent. | |
1109 | "env" is optionally a mapping specifying the environment in which to | |
1110 | start the child. Default is None, a.k.a. inherits the environment | |
1111 | of the parent. | |
1112 | """ | |
1113 | # Keep a reference to ensure it is around for this object's destruction. | |
1114 | self.__log = log | |
1115 | log.info("ProcessOpen.__init__(cmd=%r, mode=%r, cwd=%r, env=%r)", | |
1116 | cmd, mode, cwd, env) | |
1117 | self._cmd = cmd | |
1118 | if not self._cmd: | |
1119 | raise ProcessError("You must specify a command.") | |
1120 | self._cwd = cwd | |
1121 | self._env = env | |
1122 | self._mode = mode | |
1123 | if self._mode not in ('t', 'b'): | |
1124 | raise ProcessError("'mode' must be 't' or 'b'.") | |
1125 | self._closed = 0 | |
1126 | ||
1127 | if sys.platform.startswith("win"): | |
1128 | self._startOnWindows() | |
1129 | else: | |
1130 | self.__retvalCache = None | |
1131 | self._startOnUnix() | |
1132 | ||
1133 | _registerProcess(self) | |
1134 | ||
1135 | def __del__(self): | |
1136 | #XXX Should probably not rely upon this. | |
1137 | logres.info("[%s] ProcessOpen.__del__()", id(self)) | |
1138 | self.close() | |
1139 | del self.__log # drop reference | |
1140 | ||
1141 | def close(self): | |
1142 | if not self._closed: | |
1143 | self.__log.info("[%s] ProcessOpen.close()" % id(self)) | |
1144 | ||
1145 | # Ensure that all IOBuffer's are closed. If they are not, these | |
1146 | # can cause hangs. | |
1147 | try: | |
1148 | self.__log.info("[%s] ProcessOpen: closing stdin (%r)."\ | |
1149 | % (id(self), self.stdin)) | |
1150 | self.stdin.close() | |
1151 | except AttributeError: | |
1152 | # May not have gotten far enough in the __init__ to set | |
1153 | # self.stdin, etc. | |
1154 | pass | |
1155 | try: | |
1156 | self.__log.info("[%s] ProcessOpen: closing stdout (%r)."\ | |
1157 | % (id(self), self.stdout)) | |
1158 | self.stdout.close() | |
1159 | except AttributeError: | |
1160 | # May not have gotten far enough in the __init__ to set | |
1161 | # self.stdout, etc. | |
1162 | pass | |
1163 | try: | |
1164 | self.__log.info("[%s] ProcessOpen: closing stderr (%r)."\ | |
1165 | % (id(self), self.stderr)) | |
1166 | self.stderr.close() | |
1167 | except AttributeError: | |
1168 | # May not have gotten far enough in the __init__ to set | |
1169 | # self.stderr, etc. | |
1170 | pass | |
1171 | ||
1172 | self._closed = 1 | |
1173 | ||
1174 | def _forkAndExecChildOnUnix(self, fdChildStdinRd, fdChildStdoutWr, | |
1175 | fdChildStderrWr): | |
1176 | """Fork and start the child process. | |
1177 | ||
1178 | Sets self._pid as a side effect. | |
1179 | """ | |
1180 | pid = os.fork() | |
1181 | if pid == 0: # child | |
1182 | os.dup2(fdChildStdinRd, 0) | |
1183 | os.dup2(fdChildStdoutWr, 1) | |
1184 | os.dup2(fdChildStderrWr, 2) | |
1185 | self._runChildOnUnix() | |
1186 | # parent | |
1187 | self._pid = pid | |
1188 | ||
1189 | def _startOnUnix(self): | |
1190 | # Create pipes for std handles. | |
1191 | fdChildStdinRd, fdChildStdinWr = os.pipe() | |
1192 | fdChildStdoutRd, fdChildStdoutWr = os.pipe() | |
1193 | fdChildStderrRd, fdChildStderrWr = os.pipe() | |
1194 | ||
1195 | if self._cwd: | |
1196 | oldDir = os.getcwd() | |
1197 | try: | |
1198 | os.chdir(self._cwd) | |
1199 | except OSError, ex: | |
1200 | raise ProcessError(msg=str(ex), errno=ex.errno) | |
1201 | self._forkAndExecChildOnUnix(fdChildStdinRd, fdChildStdoutWr, | |
1202 | fdChildStderrWr) | |
1203 | if self._cwd: | |
1204 | os.chdir(oldDir) | |
1205 | ||
1206 | os.close(fdChildStdinRd) | |
1207 | os.close(fdChildStdoutWr) | |
1208 | os.close(fdChildStderrWr) | |
1209 | ||
1210 | self.stdin = _FileWrapper(descriptor=fdChildStdinWr) | |
1211 | logres.info("[%s] ProcessOpen._start(): create child stdin: %r", | |
1212 | id(self), self.stdin) | |
1213 | self.stdout = _FileWrapper(descriptor=fdChildStdoutRd) | |
1214 | logres.info("[%s] ProcessOpen._start(): create child stdout: %r", | |
1215 | id(self), self.stdout) | |
1216 | self.stderr = _FileWrapper(descriptor=fdChildStderrRd) | |
1217 | logres.info("[%s] ProcessOpen._start(): create child stderr: %r", | |
1218 | id(self), self.stderr) | |
1219 | ||
1220 | def _startOnWindows(self): | |
1221 | if type(self._cmd) in (types.ListType, types.TupleType): | |
1222 | # An arg vector was passed in. | |
1223 | cmd = _joinArgv(self._cmd) | |
1224 | else: | |
1225 | cmd = self._cmd | |
1226 | ||
1227 | # Create pipes for std handles. | |
1228 | # (Set the bInheritHandle flag so pipe handles are inherited.) | |
1229 | saAttr = pywintypes.SECURITY_ATTRIBUTES() | |
1230 | saAttr.bInheritHandle = 1 | |
1231 | #XXX Should maybe try with os.pipe. Dunno what that does for | |
1232 | # inheritability though. | |
1233 | hChildStdinRd, hChildStdinWr = win32pipe.CreatePipe(saAttr, 0) | |
1234 | hChildStdoutRd, hChildStdoutWr = win32pipe.CreatePipe(saAttr, 0) | |
1235 | hChildStderrRd, hChildStderrWr = win32pipe.CreatePipe(saAttr, 0) | |
1236 | ||
1237 | try: | |
1238 | # Duplicate the parent ends of the pipes so they are not | |
1239 | # inherited. | |
1240 | hChildStdinWrDup = win32api.DuplicateHandle( | |
1241 | win32api.GetCurrentProcess(), | |
1242 | hChildStdinWr, | |
1243 | win32api.GetCurrentProcess(), | |
1244 | 0, | |
1245 | 0, # not inherited | |
1246 | DUPLICATE_SAME_ACCESS) | |
1247 | win32api.CloseHandle(hChildStdinWr) | |
1248 | self._hChildStdinWr = hChildStdinWrDup | |
1249 | hChildStdoutRdDup = win32api.DuplicateHandle( | |
1250 | win32api.GetCurrentProcess(), | |
1251 | hChildStdoutRd, | |
1252 | win32api.GetCurrentProcess(), | |
1253 | 0, | |
1254 | 0, # not inherited | |
1255 | DUPLICATE_SAME_ACCESS) | |
1256 | win32api.CloseHandle(hChildStdoutRd) | |
1257 | self._hChildStdoutRd = hChildStdoutRdDup | |
1258 | hChildStderrRdDup = win32api.DuplicateHandle( | |
1259 | win32api.GetCurrentProcess(), | |
1260 | hChildStderrRd, | |
1261 | win32api.GetCurrentProcess(), | |
1262 | 0, | |
1263 | 0, # not inherited | |
1264 | DUPLICATE_SAME_ACCESS) | |
1265 | win32api.CloseHandle(hChildStderrRd) | |
1266 | self._hChildStderrRd = hChildStderrRdDup | |
1267 | ||
1268 | # Set the translation mode and buffering. | |
1269 | if self._mode == 't': | |
1270 | flags = os.O_TEXT | |
1271 | else: | |
1272 | flags = 0 | |
1273 | fdChildStdinWr = msvcrt.open_osfhandle(self._hChildStdinWr, flags) | |
1274 | fdChildStdoutRd = msvcrt.open_osfhandle(self._hChildStdoutRd, flags) | |
1275 | fdChildStderrRd = msvcrt.open_osfhandle(self._hChildStderrRd, flags) | |
1276 | ||
1277 | self.stdin = _FileWrapper(descriptor=fdChildStdinWr, | |
1278 | handle=self._hChildStdinWr) | |
1279 | logres.info("[%s] ProcessOpen._start(): create child stdin: %r", | |
1280 | id(self), self.stdin) | |
1281 | self.stdout = _FileWrapper(descriptor=fdChildStdoutRd, | |
1282 | handle=self._hChildStdoutRd) | |
1283 | logres.info("[%s] ProcessOpen._start(): create child stdout: %r", | |
1284 | id(self), self.stdout) | |
1285 | self.stderr = _FileWrapper(descriptor=fdChildStderrRd, | |
1286 | handle=self._hChildStderrRd) | |
1287 | logres.info("[%s] ProcessOpen._start(): create child stderr: %r", | |
1288 | id(self), self.stderr) | |
1289 | ||
1290 | # Start the child process. | |
1291 | si = win32process.STARTUPINFO() | |
1292 | si.dwFlags = win32process.STARTF_USESHOWWINDOW | |
1293 | si.wShowWindow = 0 # SW_HIDE | |
1294 | si.hStdInput = hChildStdinRd | |
1295 | si.hStdOutput = hChildStdoutWr | |
1296 | si.hStdError = hChildStderrWr | |
1297 | si.dwFlags |= win32process.STARTF_USESTDHANDLES | |
1298 | ||
1299 | cmd = _fixupCommand(cmd, self._env) | |
1300 | ||
1301 | creationFlags = win32process.CREATE_NEW_PROCESS_GROUP | |
1302 | try: | |
1303 | self._hProcess, hThread, self._processId, threadId\ | |
1304 | = _SaferCreateProcess( | |
1305 | None, # app name | |
1306 | cmd, # command line | |
1307 | None, # process security attributes | |
1308 | None, # primary thread security attributes | |
1309 | 1, # handles are inherited | |
1310 | creationFlags, # creation flags | |
1311 | self._env, # environment | |
1312 | self._cwd, # current working directory | |
1313 | si) # STARTUPINFO pointer | |
1314 | except win32api.error, ex: | |
1315 | raise ProcessError(msg=ex.args[2], errno=ex.args[0]) | |
1316 | win32api.CloseHandle(hThread) | |
1317 | ||
1318 | finally: | |
1319 | # Close child ends of pipes on the parent's side (the | |
1320 | # parent's ends of the pipe are closed in the _FileWrappers.) | |
1321 | win32file.CloseHandle(hChildStdinRd) | |
1322 | win32file.CloseHandle(hChildStdoutWr) | |
1323 | win32file.CloseHandle(hChildStderrWr) | |
1324 | ||
1325 | def wait(self, timeout=None): | |
1326 | """Wait for the started process to complete. | |
1327 | ||
1328 | "timeout" (on Windows) is a floating point number of seconds after | |
1329 | which to timeout. Default is win32event.INFINITE. | |
1330 | "timeout" (on Unix) is akin to the os.waitpid() "options" argument | |
1331 | (os.WNOHANG may be used to return immediately if the process has | |
1332 | not exited). Default is 0, i.e. wait forever. | |
1333 | ||
1334 | If the wait time's out it will raise a ProcessError. Otherwise it | |
1335 | will return the child's exit value (on Windows) or the child's exit | |
1336 | status excoded as per os.waitpid() (on Linux): | |
1337 | "a 16-bit number, whose low byte is the signal number that killed | |
1338 | the process, and whose high byte is the exit status (if the | |
1339 | signal number is zero); the high bit of the low byte is set if a | |
1340 | core file was produced." | |
1341 | In the latter case, use the os.W*() methods to interpret the return | |
1342 | value. | |
1343 | """ | |
1344 | # XXX Or should returning the exit value be move out to another | |
1345 | # function as on Win32 process control? If so, then should | |
1346 | # perhaps not make WaitForSingleObject semantic | |
1347 | # transformation. | |
1348 | # TODO: | |
1349 | # - Need to rationalize the .wait() API for Windows vs. Unix. | |
1350 | # It is a real pain in the current situation. | |
1351 | if sys.platform.startswith("win"): | |
1352 | if timeout is None: | |
1353 | timeout = win32event.INFINITE | |
1354 | else: | |
1355 | timeout = timeout * 1000.0 # Win32 API's timeout is in millisecs | |
1356 | ||
1357 | #rc = win32event.WaitForSingleObject(self._hProcess, timeout) | |
1358 | rc = win32event.WaitForSingleObject(self._hProcess, int(timeout)) # MATT -- Making timeout an integer | |
1359 | if rc == win32event.WAIT_FAILED: | |
1360 | raise ProcessError("'WAIT_FAILED' when waiting for process to "\ | |
1361 | "terminate: %r" % self._cmd, rc) | |
1362 | elif rc == win32event.WAIT_TIMEOUT: | |
1363 | raise ProcessError("'WAIT_TIMEOUT' when waiting for process to "\ | |
1364 | "terminate: %r" % self._cmd, rc) | |
1365 | ||
1366 | retval = win32process.GetExitCodeProcess(self._hProcess) | |
1367 | else: | |
1368 | # os.waitpid() will raise: | |
1369 | # OSError: [Errno 10] No child processes | |
1370 | # on subsequent .wait() calls. Change these semantics to have | |
1371 | # subsequent .wait() calls return the exit status and return | |
1372 | # immediately without raising an exception. | |
1373 | # (XXX It would require synchronization code to handle the case | |
1374 | # of multiple simultaneous .wait() requests, however we can punt | |
1375 | # on that because it is moot while Linux still has the problem | |
1376 | # for which _ThreadFixer() exists.) | |
1377 | if self.__retvalCache is not None: | |
1378 | retval = self.__retvalCache | |
1379 | else: | |
1380 | if timeout is None: | |
1381 | timeout = 0 | |
1382 | pid, sts = os.waitpid(self._pid, timeout) | |
1383 | if pid == self._pid: | |
1384 | self.__retvalCache = retval = sts | |
1385 | else: | |
1386 | raise ProcessError("Wait for process timed out.", | |
1387 | self.WAIT_TIMEOUT) | |
1388 | _unregisterProcess(self) | |
1389 | return retval | |
1390 | ||
1391 | def kill(self, exitCode=0, gracePeriod=1.0, sig=None): | |
1392 | """Kill process. | |
1393 | ||
1394 | "exitCode" [deprecated, not supported] (Windows only) is the | |
1395 | code the terminated process should exit with. | |
1396 | "gracePeriod" (Windows only) is a number of seconds the process is | |
1397 | allowed to shutdown with a WM_CLOSE signal before a hard | |
1398 | terminate is called. | |
1399 | "sig" (Unix only) is the signal to use to kill the process. Defaults | |
1400 | to signal.SIGKILL. See os.kill() for more information. | |
1401 | ||
1402 | Windows: | |
1403 | Try for an orderly shutdown via WM_CLOSE. If still running | |
1404 | after gracePeriod (1 sec. default), terminate. | |
1405 | """ | |
1406 | if sys.platform.startswith("win"): | |
1407 | import win32gui | |
1408 | # Send WM_CLOSE to windows in this process group. | |
1409 | win32gui.EnumWindows(self._close_, 0) | |
1410 | ||
1411 | # Send Ctrl-Break signal to all processes attached to this | |
1412 | # console. This is supposed to trigger shutdown handlers in | |
1413 | # each of the processes. | |
1414 | try: | |
1415 | win32api.GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, | |
1416 | self._processId) | |
1417 | except AttributeError: | |
1418 | log.warn("The win32api module does not have "\ | |
1419 | "GenerateConsoleCtrlEvent(). This may mean that "\ | |
1420 | "parts of this process group have NOT been killed.") | |
1421 | except win32api.error, ex: | |
1422 | if ex.args[0] not in (6, 87): | |
1423 | # Ignore the following: | |
1424 | # api_error: (87, 'GenerateConsoleCtrlEvent', 'The parameter is incorrect.') | |
1425 | # api_error: (6, 'GenerateConsoleCtrlEvent', 'The handle is invalid.') | |
1426 | # Get error 6 if there is no console. | |
1427 | raise | |
1428 | ||
1429 | # Last resort: call TerminateProcess if it has not yet. | |
1430 | retval = 0 | |
1431 | try: | |
1432 | self.wait(gracePeriod) | |
1433 | except ProcessError, ex: | |
1434 | log.info("[%s] Process.kill: calling TerminateProcess", id(self)) | |
1435 | win32process.TerminateProcess(self._hProcess, -1) | |
1436 | win32api.Sleep(100) # wait for resources to be released | |
1437 | ||
1438 | else: | |
1439 | if sig is None: | |
1440 | sig = signal.SIGKILL | |
1441 | try: | |
1442 | os.kill(self._pid, sig) | |
1443 | except OSError, ex: | |
1444 | if ex.errno != 3: | |
1445 | # Ignore: OSError: [Errno 3] No such process | |
1446 | raise | |
1447 | ||
1448 | _unregisterProcess(self) | |
1449 | ||
1450 | def _close_(self, hwnd, dummy): | |
1451 | """Callback used by .kill() on Windows. | |
1452 | ||
1453 | EnumWindows callback - sends WM_CLOSE to any window owned by this | |
1454 | process. | |
1455 | """ | |
1456 | threadId, processId = win32process.GetWindowThreadProcessId(hwnd) | |
1457 | if processId == self._processId: | |
1458 | import win32gui | |
1459 | win32gui.PostMessage(hwnd, WM_CLOSE, 0, 0) | |
1460 | ||
1461 | ||
1462 | class ProcessProxy(Process): | |
1463 | """Create a process and proxy communication via the standard handles. | |
1464 | """ | |
1465 | #XXX To add to docstring: | |
1466 | # - stdout/stderr proxy handling | |
1467 | # - stdin proxy handling | |
1468 | # - termination | |
1469 | # - how to .start(), i.e. basic usage rules | |
1470 | # - mention that pased in stdin/stdout/stderr objects have to | |
1471 | # implement at least .write (is .write correct for stdin)? | |
1472 | # - if you pass in stdin, stdout, and/or stderr streams it is the | |
1473 | # user's responsibility to close them afterwards. | |
1474 | # - 'cmd' arg can be a command string or an arg vector | |
1475 | # - etc. | |
1476 | #TODO: | |
1477 | # - .suspend() and .resume()? See Win32::Process Perl module. | |
1478 | # | |
1479 | def __init__(self, cmd, mode='t', cwd=None, env=None, | |
1480 | stdin=None, stdout=None, stderr=None): | |
1481 | """Create a Process with proxy threads for each std handle. | |
1482 | ||
1483 | "cmd" is the command string or argument vector to run. | |
1484 | "mode" (Windows only) specifies whether the pipes used to communicate | |
1485 | with the child are openned in text, 't', or binary, 'b', mode. | |
1486 | This is ignored on platforms other than Windows. Default is 't'. | |
1487 | "cwd" optionally specifies the directory in which the child process | |
1488 | should be started. Default is None, a.k.a. inherits the cwd from | |
1489 | the parent. | |
1490 | "env" is optionally a mapping specifying the environment in which to | |
1491 | start the child. Default is None, a.k.a. inherits the environment | |
1492 | of the parent. | |
1493 | "stdin", "stdout", "stderr" can be used to specify objects with | |
1494 | file-like interfaces to handle read (stdout/stderr) and write | |
1495 | (stdin) events from the child. By default a process.IOBuffer | |
1496 | instance is assigned to each handler. IOBuffer may be | |
1497 | sub-classed. See the IOBuffer doc string for more information. | |
1498 | """ | |
1499 | # Keep a reference to ensure it is around for this object's destruction. | |
1500 | self.__log = log | |
1501 | log.info("ProcessProxy.__init__(cmd=%r, mode=%r, cwd=%r, env=%r, "\ | |
1502 | "stdin=%r, stdout=%r, stderr=%r)", | |
1503 | cmd, mode, cwd, env, stdin, stdout, stderr) | |
1504 | self._cmd = cmd | |
1505 | if not self._cmd: | |
1506 | raise ProcessError("You must specify a command.") | |
1507 | self._mode = mode | |
1508 | if self._mode not in ('t', 'b'): | |
1509 | raise ProcessError("'mode' must be 't' or 'b'.") | |
1510 | self._cwd = cwd | |
1511 | self._env = env | |
1512 | if stdin is None: | |
1513 | self.stdin = IOBuffer(name='<stdin>') | |
1514 | else: | |
1515 | self.stdin = stdin | |
1516 | if stdout is None: | |
1517 | self.stdout = IOBuffer(name='<stdout>') | |
1518 | else: | |
1519 | self.stdout = stdout | |
1520 | if stderr is None: | |
1521 | self.stderr = IOBuffer(name='<stderr>') | |
1522 | else: | |
1523 | self.stderr = stderr | |
1524 | self._closed = 0 | |
1525 | ||
1526 | if sys.platform.startswith("win"): | |
1527 | self._startOnWindows() | |
1528 | else: | |
1529 | self.__retvalCache = None | |
1530 | self._startOnUnix() | |
1531 | ||
1532 | _registerProcess(self) | |
1533 | ||
1534 | def __del__(self): | |
1535 | #XXX Should probably not rely upon this. | |
1536 | logres.info("[%s] ProcessProxy.__del__()", id(self)) | |
1537 | self.close() | |
1538 | del self.__log # drop reference | |
1539 | ||
1540 | def close(self): | |
1541 | if not self._closed: | |
1542 | self.__log.info("[%s] ProcessProxy.close()" % id(self)) | |
1543 | ||
1544 | # Ensure that all IOBuffer's are closed. If they are not, these | |
1545 | # can cause hangs. | |
1546 | self.__log.info("[%s] ProcessProxy: closing stdin (%r)."\ | |
1547 | % (id(self), self.stdin)) | |
1548 | try: | |
1549 | self.stdin.close() | |
1550 | self._stdinProxy.join() | |
1551 | except AttributeError: | |
1552 | # May not have gotten far enough in the __init__ to set | |
1553 | # self.stdin, etc. | |
1554 | pass | |
1555 | self.__log.info("[%s] ProcessProxy: closing stdout (%r)."\ | |
1556 | % (id(self), self.stdout)) | |
1557 | try: | |
1558 | self.stdout.close() | |
1559 | if self._stdoutProxy is not threading.currentThread(): | |
1560 | self._stdoutProxy.join() | |
1561 | except AttributeError: | |
1562 | # May not have gotten far enough in the __init__ to set | |
1563 | # self.stdout, etc. | |
1564 | pass | |
1565 | self.__log.info("[%s] ProcessProxy: closing stderr (%r)."\ | |
1566 | % (id(self), self.stderr)) | |
1567 | try: | |
1568 | self.stderr.close() | |
1569 | if self._stderrProxy is not threading.currentThread(): | |
1570 | self._stderrProxy.join() | |
1571 | except AttributeError: | |
1572 | # May not have gotten far enough in the __init__ to set | |
1573 | # self.stderr, etc. | |
1574 | pass | |
1575 | ||
1576 | self._closed = 1 | |
1577 | ||
1578 | def _forkAndExecChildOnUnix(self, fdChildStdinRd, fdChildStdoutWr, | |
1579 | fdChildStderrWr): | |
1580 | """Fork and start the child process. | |
1581 | ||
1582 | Sets self._pid as a side effect. | |
1583 | """ | |
1584 | pid = os.fork() | |
1585 | if pid == 0: # child | |
1586 | os.dup2(fdChildStdinRd, 0) | |
1587 | os.dup2(fdChildStdoutWr, 1) | |
1588 | os.dup2(fdChildStderrWr, 2) | |
1589 | self._runChildOnUnix() | |
1590 | # parent | |
1591 | self._pid = pid | |
1592 | ||
1593 | def _startOnUnix(self): | |
1594 | # Create pipes for std handles. | |
1595 | fdChildStdinRd, fdChildStdinWr = os.pipe() | |
1596 | fdChildStdoutRd, fdChildStdoutWr = os.pipe() | |
1597 | fdChildStderrRd, fdChildStderrWr = os.pipe() | |
1598 | ||
1599 | if self._cwd: | |
1600 | oldDir = os.getcwd() | |
1601 | try: | |
1602 | os.chdir(self._cwd) | |
1603 | except OSError, ex: | |
1604 | raise ProcessError(msg=str(ex), errno=ex.errno) | |
1605 | self._forkAndExecChildOnUnix(fdChildStdinRd, fdChildStdoutWr, | |
1606 | fdChildStderrWr) | |
1607 | if self._cwd: | |
1608 | os.chdir(oldDir) | |
1609 | ||
1610 | os.close(fdChildStdinRd) | |
1611 | os.close(fdChildStdoutWr) | |
1612 | os.close(fdChildStderrWr) | |
1613 | ||
1614 | childStdin = _FileWrapper(descriptor=fdChildStdinWr) | |
1615 | logres.info("[%s] ProcessProxy._start(): create child stdin: %r", | |
1616 | id(self), childStdin) | |
1617 | childStdout = _FileWrapper(descriptor=fdChildStdoutRd) | |
1618 | logres.info("[%s] ProcessProxy._start(): create child stdout: %r", | |
1619 | id(self), childStdout) | |
1620 | childStderr = _FileWrapper(descriptor=fdChildStderrRd) | |
1621 | logres.info("[%s] ProcessProxy._start(): create child stderr: %r", | |
1622 | id(self), childStderr) | |
1623 | ||
1624 | # Create proxy threads for the out pipes. | |
1625 | self._stdinProxy = _InFileProxy(self.stdin, childStdin, name='<stdin>') | |
1626 | self._stdinProxy.start() | |
1627 | # Clean up the parent's side of <stdin> when it is observed that | |
1628 | # the child has closed its side of <stdout> and <stderr>. (This | |
1629 | # is one way of determining when it is appropriate to clean up | |
1630 | # this pipe, with compromises. See the discussion at the top of | |
1631 | # this module.) | |
1632 | closer = _CountingCloser([self.stdin, childStdin, self], 2) | |
1633 | self._stdoutProxy = _OutFileProxy(childStdout, self.stdout, | |
1634 | [closer], | |
1635 | name='<stdout>') | |
1636 | self._stdoutProxy.start() | |
1637 | self._stderrProxy = _OutFileProxy(childStderr, self.stderr, | |
1638 | [closer], | |
1639 | name='<stderr>') | |
1640 | self._stderrProxy.start() | |
1641 | ||
1642 | def _startOnWindows(self): | |
1643 | if type(self._cmd) in (types.ListType, types.TupleType): | |
1644 | # An arg vector was passed in. | |
1645 | cmd = _joinArgv(self._cmd) | |
1646 | else: | |
1647 | cmd = self._cmd | |
1648 | ||
1649 | # Create pipes for std handles. | |
1650 | # (Set the bInheritHandle flag so pipe handles are inherited.) | |
1651 | saAttr = pywintypes.SECURITY_ATTRIBUTES() | |
1652 | saAttr.bInheritHandle = 1 | |
1653 | #XXX Should maybe try with os.pipe. Dunno what that does for | |
1654 | # inheritability though. | |
1655 | hChildStdinRd, hChildStdinWr = win32pipe.CreatePipe(saAttr, 0) | |
1656 | hChildStdoutRd, hChildStdoutWr = win32pipe.CreatePipe(saAttr, 0) | |
1657 | hChildStderrRd, hChildStderrWr = win32pipe.CreatePipe(saAttr, 0) | |
1658 | ||
1659 | try: | |
1660 | # Duplicate the parent ends of the pipes so they are not | |
1661 | # inherited. | |
1662 | hChildStdinWrDup = win32api.DuplicateHandle( | |
1663 | win32api.GetCurrentProcess(), | |
1664 | hChildStdinWr, | |
1665 | win32api.GetCurrentProcess(), | |
1666 | 0, | |
1667 | 0, # not inherited | |
1668 | DUPLICATE_SAME_ACCESS) | |
1669 | win32api.CloseHandle(hChildStdinWr) | |
1670 | self._hChildStdinWr = hChildStdinWrDup | |
1671 | hChildStdoutRdDup = win32api.DuplicateHandle( | |
1672 | win32api.GetCurrentProcess(), | |
1673 | hChildStdoutRd, | |
1674 | win32api.GetCurrentProcess(), | |
1675 | 0, | |
1676 | 0, # not inherited | |
1677 | DUPLICATE_SAME_ACCESS) | |
1678 | win32api.CloseHandle(hChildStdoutRd) | |
1679 | self._hChildStdoutRd = hChildStdoutRdDup | |
1680 | hChildStderrRdDup = win32api.DuplicateHandle( | |
1681 | win32api.GetCurrentProcess(), | |
1682 | hChildStderrRd, | |
1683 | win32api.GetCurrentProcess(), | |
1684 | 0, | |
1685 | 0, # not inherited | |
1686 | DUPLICATE_SAME_ACCESS) | |
1687 | win32api.CloseHandle(hChildStderrRd) | |
1688 | self._hChildStderrRd = hChildStderrRdDup | |
1689 | ||
1690 | # Set the translation mode. | |
1691 | if self._mode == 't': | |
1692 | flags = os.O_TEXT | |
1693 | mode = '' | |
1694 | else: | |
1695 | flags = 0 | |
1696 | mode = 'b' | |
1697 | fdChildStdinWr = msvcrt.open_osfhandle(self._hChildStdinWr, flags) | |
1698 | fdChildStdoutRd = msvcrt.open_osfhandle(self._hChildStdoutRd, flags) | |
1699 | fdChildStderrRd = msvcrt.open_osfhandle(self._hChildStderrRd, flags) | |
1700 | ||
1701 | childStdin = _FileWrapper(descriptor=fdChildStdinWr, | |
1702 | handle=self._hChildStdinWr) | |
1703 | logres.info("[%s] ProcessProxy._start(): create child stdin: %r", | |
1704 | id(self), childStdin) | |
1705 | childStdout = _FileWrapper(descriptor=fdChildStdoutRd, | |
1706 | handle=self._hChildStdoutRd) | |
1707 | logres.info("[%s] ProcessProxy._start(): create child stdout: %r", | |
1708 | id(self), childStdout) | |
1709 | childStderr = _FileWrapper(descriptor=fdChildStderrRd, | |
1710 | handle=self._hChildStderrRd) | |
1711 | logres.info("[%s] ProcessProxy._start(): create child stderr: %r", | |
1712 | id(self), childStderr) | |
1713 | ||
1714 | # Start the child process. | |
1715 | si = win32process.STARTUPINFO() | |
1716 | si.dwFlags = win32process.STARTF_USESHOWWINDOW | |
1717 | si.wShowWindow = 0 # SW_HIDE | |
1718 | si.hStdInput = hChildStdinRd | |
1719 | si.hStdOutput = hChildStdoutWr | |
1720 | si.hStdError = hChildStderrWr | |
1721 | si.dwFlags |= win32process.STARTF_USESTDHANDLES | |
1722 | ||
1723 | cmd = _fixupCommand(cmd, self._env) | |
1724 | log.debug("cmd = %r", cmd) | |
1725 | ||
1726 | creationFlags = win32process.CREATE_NEW_PROCESS_GROUP | |
1727 | try: | |
1728 | self._hProcess, hThread, self._processId, threadId\ | |
1729 | = _SaferCreateProcess( | |
1730 | None, # app name | |
1731 | cmd, # command line | |
1732 | None, # process security attributes | |
1733 | None, # primary thread security attributes | |
1734 | 1, # handles are inherited | |
1735 | creationFlags, # creation flags | |
1736 | self._env, # environment | |
1737 | self._cwd, # current working directory | |
1738 | si) # STARTUPINFO pointer | |
1739 | except win32api.error, ex: | |
1740 | raise ProcessError(msg=ex.args[2], errno=ex.args[0]) | |
1741 | win32api.CloseHandle(hThread) | |
1742 | ||
1743 | finally: | |
1744 | # Close child ends of pipes on the parent's side (the | |
1745 | # parent's ends of the pipe are closed in the _FileWrappers.) | |
1746 | win32file.CloseHandle(hChildStdinRd) | |
1747 | win32file.CloseHandle(hChildStdoutWr) | |
1748 | win32file.CloseHandle(hChildStderrWr) | |
1749 | ||
1750 | # Create proxy threads for the pipes. | |
1751 | self._stdinProxy = _InFileProxy(self.stdin, childStdin, name='<stdin>') | |
1752 | self._stdinProxy.start() | |
1753 | # Clean up the parent's side of <stdin> when it is observed that | |
1754 | # the child has closed its side of <stdout>. (This is one way of | |
1755 | # determining when it is appropriate to clean up this pipe, with | |
1756 | # compromises. See the discussion at the top of this module.) | |
1757 | self._stdoutProxy = _OutFileProxy(childStdout, self.stdout, | |
1758 | [self.stdin, childStdin, self], | |
1759 | name='<stdout>') | |
1760 | self._stdoutProxy.start() | |
1761 | self._stderrProxy = _OutFileProxy(childStderr, self.stderr, | |
1762 | name='<stderr>') | |
1763 | self._stderrProxy.start() | |
1764 | ||
1765 | def wait(self, timeout=None): | |
1766 | """Wait for the started process to complete. | |
1767 | ||
1768 | "timeout" (on Windows) is a floating point number of seconds after | |
1769 | which to timeout. Default is win32event.INFINITE. | |
1770 | "timeout" (on Unix) is akin to the os.waitpid() "options" argument | |
1771 | (os.WNOHANG may be used to return immediately if the process has | |
1772 | not exited). Default is 0, i.e. wait forever. | |
1773 | ||
1774 | If the wait time's out it will raise a ProcessError. Otherwise it | |
1775 | will return the child's exit value (on Windows) or the child's exit | |
1776 | status excoded as per os.waitpid() (on Linux): | |
1777 | "a 16-bit number, whose low byte is the signal number that killed | |
1778 | the process, and whose high byte is the exit status (if the | |
1779 | signal number is zero); the high bit of the low byte is set if a | |
1780 | core file was produced." | |
1781 | In the latter case, use the os.W*() methods to interpret the return | |
1782 | value. | |
1783 | """ | |
1784 | # XXX Or should returning the exit value be move out to another | |
1785 | # function as on Win32 process control? If so, then should | |
1786 | # perhaps not make WaitForSingleObject semantic transformation. | |
1787 | if sys.platform.startswith("win"): | |
1788 | if timeout is None: | |
1789 | timeout = win32event.INFINITE | |
1790 | else: | |
1791 | timeout = timeout * 1000.0 # Win32 API's timeout is in millisecs | |
1792 | ||
1793 | rc = win32event.WaitForSingleObject(self._hProcess, timeout) | |
1794 | if rc == win32event.WAIT_FAILED: | |
1795 | raise ProcessError("'WAIT_FAILED' when waiting for process to "\ | |
1796 | "terminate: %r" % self._cmd, rc) | |
1797 | elif rc == win32event.WAIT_TIMEOUT: | |
1798 | raise ProcessError("'WAIT_TIMEOUT' when waiting for process to "\ | |
1799 | "terminate: %r" % self._cmd, rc) | |
1800 | ||
1801 | retval = win32process.GetExitCodeProcess(self._hProcess) | |
1802 | else: | |
1803 | # os.waitpid() will raise: | |
1804 | # OSError: [Errno 10] No child processes | |
1805 | # on subsequent .wait() calls. Change these semantics to have | |
1806 | # subsequent .wait() calls return the exit status and return | |
1807 | # immediately without raising an exception. | |
1808 | # (XXX It would require synchronization code to handle the case | |
1809 | # of multiple simultaneous .wait() requests, however we can punt | |
1810 | # on that because it is moot while Linux still has the problem | |
1811 | # for which _ThreadFixer() exists.) | |
1812 | if self.__retvalCache is not None: | |
1813 | retval = self.__retvalCache | |
1814 | else: | |
1815 | if timeout is None: | |
1816 | timeout = 0 | |
1817 | pid, sts = os.waitpid(self._pid, timeout) | |
1818 | if pid == self._pid: | |
1819 | self.__retvalCache = retval = sts | |
1820 | else: | |
1821 | raise ProcessError("Wait for process timed out.", | |
1822 | self.WAIT_TIMEOUT) | |
1823 | _unregisterProcess(self) | |
1824 | return retval | |
1825 | ||
1826 | def kill(self, exitCode=0, gracePeriod=1.0, sig=None): | |
1827 | """Kill process. | |
1828 | ||
1829 | "exitCode" [deprecated, not supported] (Windows only) is the | |
1830 | code the terminated process should exit with. | |
1831 | "gracePeriod" (Windows only) is a number of seconds the process is | |
1832 | allowed to shutdown with a WM_CLOSE signal before a hard | |
1833 | terminate is called. | |
1834 | "sig" (Unix only) is the signal to use to kill the process. Defaults | |
1835 | to signal.SIGKILL. See os.kill() for more information. | |
1836 | ||
1837 | Windows: | |
1838 | Try for an orderly shutdown via WM_CLOSE. If still running | |
1839 | after gracePeriod (1 sec. default), terminate. | |
1840 | """ | |
1841 | if sys.platform.startswith("win"): | |
1842 | import win32gui | |
1843 | # Send WM_CLOSE to windows in this process group. | |
1844 | win32gui.EnumWindows(self._close_, 0) | |
1845 | ||
1846 | # Send Ctrl-Break signal to all processes attached to this | |
1847 | # console. This is supposed to trigger shutdown handlers in | |
1848 | # each of the processes. | |
1849 | try: | |
1850 | win32api.GenerateConsoleCtrlEvent(CTRL_BREAK_EVENT, | |
1851 | self._processId) | |
1852 | except AttributeError: | |
1853 | log.warn("The win32api module does not have "\ | |
1854 | "GenerateConsoleCtrlEvent(). This may mean that "\ | |
1855 | "parts of this process group have NOT been killed.") | |
1856 | except win32api.error, ex: | |
1857 | if ex.args[0] not in (6, 87): | |
1858 | # Ignore the following: | |
1859 | # api_error: (87, 'GenerateConsoleCtrlEvent', 'The parameter is incorrect.') | |
1860 | # api_error: (6, 'GenerateConsoleCtrlEvent', 'The handle is invalid.') | |
1861 | # Get error 6 if there is no console. | |
1862 | raise | |
1863 | ||
1864 | # Last resort: call TerminateProcess if it has not yet. | |
1865 | retval = 0 | |
1866 | try: | |
1867 | self.wait(gracePeriod) | |
1868 | except ProcessError, ex: | |
1869 | log.info("[%s] Process.kill: calling TerminateProcess", id(self)) | |
1870 | win32process.TerminateProcess(self._hProcess, -1) | |
1871 | win32api.Sleep(100) # wait for resources to be released | |
1872 | ||
1873 | else: | |
1874 | if sig is None: | |
1875 | sig = signal.SIGKILL | |
1876 | try: | |
1877 | os.kill(self._pid, sig) | |
1878 | except OSError, ex: | |
1879 | if ex.errno != 3: | |
1880 | # Ignore: OSError: [Errno 3] No such process | |
1881 | raise | |
1882 | ||
1883 | _unregisterProcess(self) | |
1884 | ||
1885 | def _close_(self, hwnd, dummy): | |
1886 | """Callback used by .kill() on Windows. | |
1887 | ||
1888 | EnumWindows callback - sends WM_CLOSE to any window owned by this | |
1889 | process. | |
1890 | """ | |
1891 | threadId, processId = win32process.GetWindowThreadProcessId(hwnd) | |
1892 | if processId == self._processId: | |
1893 | import win32gui | |
1894 | win32gui.PostMessage(hwnd, WM_CLOSE, 0, 0) | |
1895 | ||
1896 | ||
1897 | class IOBuffer: | |
1898 | """Want to be able to both read and write to this buffer from | |
1899 | difference threads and have the same read/write semantics as for a | |
1900 | std handler. | |
1901 | ||
1902 | This class is subclass-able. _doRead(), _doWrite(), _doReadline(), | |
1903 | _doClose(), _haveLine(), and _haveNumBytes() can be overridden for | |
1904 | specific functionality. The synchronization issues (block on read | |
1905 | until write provides the needed data, termination) are handled for | |
1906 | free. | |
1907 | ||
1908 | Cannot support: | |
1909 | .seek() # Because we are managing *two* positions (one each | |
1910 | .tell() # for reading and writing), these do not make | |
1911 | # sense. | |
1912 | """ | |
1913 | #TODO: | |
1914 | # - Is performance a problem? This will likely be slower that | |
1915 | # StringIO.StringIO(). | |
1916 | # | |
1917 | def __init__(self, mutex=None, stateChange=None, name=None): | |
1918 | """'name' can be set for debugging, it will be used in log messages.""" | |
1919 | if name is not None: | |
1920 | self._name = name | |
1921 | else: | |
1922 | self._name = id(self) | |
1923 | log.info("[%s] IOBuffer.__init__()" % self._name) | |
1924 | ||
1925 | self.__buf = '' | |
1926 | # A state change is defined as the buffer being closed or a | |
26ee3a06 | 1927 | # write occuring. |
1f780e48 RD |
1928 | if mutex is not None: |
1929 | self._mutex = mutex | |
1930 | else: | |
1931 | self._mutex = threading.Lock() | |
1932 | if stateChange is not None: | |
1933 | self._stateChange = stateChange | |
1934 | else: | |
1935 | self._stateChange = threading.Condition() | |
1936 | self._closed = 0 | |
1937 | ||
1938 | def _doWrite(self, s): | |
1939 | self.__buf += s # Append to buffer. | |
1940 | ||
1941 | def write(self, s): | |
1942 | log.info("[%s] IOBuffer.write(s=%r)", self._name, s) | |
1943 | # Silently drop writes after the buffer has been close()'d. | |
1944 | if self._closed: | |
1945 | return | |
1946 | # If empty write, close buffer (mimicking behaviour from | |
1947 | # koprocess.cpp.) | |
1948 | if not s: | |
1949 | self.close() | |
1950 | return | |
1951 | ||
1952 | self._mutex.acquire() | |
1953 | self._doWrite(s) | |
1954 | self._stateChange.acquire() | |
1955 | self._stateChange.notifyAll() # Notify of the write(). | |
1956 | self._stateChange.release() | |
1957 | self._mutex.release() | |
1958 | ||
1959 | def writelines(self, list): | |
1960 | self.write(''.join(list)) | |
1961 | ||
1962 | def _doRead(self, n): | |
1963 | """Pop 'n' bytes from the internal buffer and return them.""" | |
1964 | if n < 0: | |
1965 | idx = len(self.__buf) | |
1966 | else: | |
1967 | idx = min(n, len(self.__buf)) | |
1968 | retval, self.__buf = self.__buf[:idx], self.__buf[idx:] | |
1969 | return retval | |
1970 | ||
1971 | def read(self, n=-1): | |
1972 | log.info("[%s] IOBuffer.read(n=%r)" % (self._name, n)) | |
1973 | log.info("[%s] IOBuffer.read(): wait for data" % self._name) | |
1974 | if n < 0: | |
1975 | # Wait until the buffer is closed, i.e. no more writes will | |
1976 | # come. | |
1977 | while 1: | |
1978 | if self._closed: break | |
1979 | #log.debug("[%s] <<< IOBuffer.read: state change .wait()"\ | |
1980 | # % self._name) | |
1981 | self._stateChange.acquire() | |
1982 | self._stateChange.wait() | |
1983 | self._stateChange.release() | |
1984 | #log.debug("[%s] >>> IOBuffer.read: done change .wait()"\ | |
1985 | # % self._name) | |
1986 | else: | |
1987 | # Wait until there are the requested number of bytes to read | |
1988 | # (or until the buffer is closed, i.e. no more writes will | |
1989 | # come). | |
1990 | # XXX WARNING: I *think* there is a race condition around | |
1991 | # here whereby self.fparent.read() in _InFileProxy can | |
1992 | # hang. *Sometime* test_stdin::test_stdin_buffer() will | |
1993 | # hang. This was *before* I moved the | |
1994 | # _stateChange.acquire() and .release() calls out side | |
1995 | # of the 'while 1:' here. ...and now they are back | |
1996 | # inside. | |
1997 | while 1: | |
1998 | if self._closed: break | |
1999 | if self._haveNumBytes(n): break | |
2000 | #log.debug("[%s] <<< IOBuffer.read: state change .wait()"\ | |
2001 | # % self._name) | |
2002 | self._stateChange.acquire() | |
2003 | self._stateChange.wait() | |
2004 | self._stateChange.release() | |
2005 | #log.debug("[%s] >>> IOBuffer.read: done change .wait()"\ | |
2006 | # % self._name) | |
2007 | log.info("[%s] IOBuffer.read(): done waiting for data" % self._name) | |
2008 | ||
2009 | self._mutex.acquire() | |
2010 | retval = self._doRead(n) | |
2011 | self._mutex.release() | |
2012 | return retval | |
2013 | ||
2014 | def _doReadline(self, n): | |
2015 | """Pop the front line (or n bytes of it, whichever is less) from | |
2016 | the internal buffer and return it. | |
2017 | """ | |
2018 | idx = self.__buf.find('\n') | |
2019 | if idx == -1: | |
2020 | idx = len(self.__buf) | |
2021 | else: | |
2022 | idx += 1 # include the '\n' | |
2023 | if n is not None: | |
2024 | idx = min(idx, n) | |
2025 | retval, self.__buf = self.__buf[:idx], self.__buf[idx:] | |
2026 | return retval | |
2027 | ||
2028 | def _haveLine(self): | |
2029 | return self.__buf.find('\n') != -1 | |
2030 | ||
2031 | def _haveNumBytes(self, n=None): | |
2032 | return len(self.__buf) >= n | |
2033 | ||
2034 | def readline(self, n=None): | |
2035 | # Wait until there is a full line (or at least 'n' bytes) | |
2036 | # in the buffer or until the buffer is closed, i.e. no more | |
2037 | # writes will come. | |
2038 | log.info("[%s] IOBuffer.readline(n=%r)" % (self._name, n)) | |
2039 | ||
2040 | log.info("[%s] IOBuffer.readline(): wait for data" % self._name) | |
2041 | while 1: | |
2042 | if self._closed: break | |
2043 | if self._haveLine(): break | |
2044 | if n is not None and self._haveNumBytes(n): break | |
2045 | self._stateChange.acquire() | |
2046 | self._stateChange.wait() | |
2047 | self._stateChange.release() | |
2048 | log.info("[%s] IOBuffer.readline(): done waiting for data"\ | |
2049 | % self._name) | |
2050 | ||
2051 | self._mutex.acquire() | |
2052 | retval = self._doReadline(n) | |
2053 | self._mutex.release() | |
2054 | return retval | |
2055 | ||
2056 | def readlines(self): | |
2057 | lines = [] | |
2058 | while 1: | |
2059 | line = self.readline() | |
2060 | if line: | |
2061 | lines.append(line) | |
2062 | else: | |
2063 | break | |
2064 | return lines | |
2065 | ||
2066 | def _doClose(self): | |
2067 | pass | |
2068 | ||
2069 | def close(self): | |
2070 | if not self._closed: | |
2071 | log.info("[%s] IOBuffer.close()" % self._name) | |
2072 | self._doClose() | |
2073 | self._closed = 1 | |
2074 | self._stateChange.acquire() | |
2075 | self._stateChange.notifyAll() # Notify of the close(). | |
2076 | self._stateChange.release() | |
2077 | ||
2078 | def flush(self): | |
2079 | log.info("[%s] IOBuffer.flush()" % self._name) | |
2080 | #XXX Perhaps flush() should unwedged possible waiting .read() | |
2081 | # and .readline() calls that are waiting for more data??? | |
2082 | ||
2083 | ||
2084 | class _InFileProxy(threading.Thread): | |
2085 | """A thread to proxy stdin.write()'s from the parent to the child.""" | |
2086 | def __init__(self, fParent, fChild, name=None): | |
2087 | """ | |
2088 | "fParent" is a Python file-like object setup for writing. | |
2089 | "fChild" is a Win32 handle to the a child process' output pipe. | |
2090 | "name" can be set for debugging, it will be used in log messages. | |
2091 | """ | |
2092 | log.info("[%s, %s] _InFileProxy.__init__(fChild=%r, fParent=%r)", | |
2093 | name, id(self), fChild, fParent) | |
2094 | threading.Thread.__init__(self, name=name) | |
2095 | self.fChild = fChild | |
2096 | self.fParent = fParent | |
2097 | ||
2098 | def run(self): | |
2099 | log.info("[%s] _InFileProxy: start" % self.getName()) | |
2100 | try: | |
2101 | self._proxyFromParentToChild() | |
2102 | finally: | |
2103 | log.info("[%s] _InFileProxy: closing parent (%r)"\ | |
2104 | % (self.getName(), self.fParent)) | |
2105 | try: | |
2106 | self.fParent.close() | |
2107 | except IOError: | |
2108 | pass # Ignore: IOError: [Errno 4] Interrupted system call | |
2109 | log.info("[%s] _InFileProxy: done" % self.getName()) | |
2110 | ||
2111 | def _proxyFromParentToChild(self): | |
2112 | CHUNKSIZE = 4096 | |
2113 | # Read output from the child process, and (for now) just write | |
2114 | # it out. | |
2115 | while 1: | |
2116 | log.info("[%s] _InFileProxy: waiting for read on parent (%r)"\ | |
2117 | % (self.getName(), self.fParent)) | |
2118 | # XXX Get hangs here (!) even with | |
2119 | # self.stdin.close() in ProcessProxy' __del__() under this | |
2120 | # cond: | |
2121 | # p = ProcessProxy([...], stdin=sys.stdin) | |
2122 | # The user must manually send '\n' via <Enter> or EOF | |
2123 | # via <Ctrl-Z> to unlock this. How to get around that? | |
2124 | # See cleanOnTermination note in _OutFileProxy.run() | |
2125 | # below. | |
2126 | #log.debug("XXX -> start read on %r" % self.fParent) | |
2127 | try: | |
2128 | text = self.fParent.read(CHUNKSIZE) | |
2129 | except ValueError, ex: | |
2130 | # ValueError is raised with trying to write to a closed | |
2131 | # file/pipe. | |
2132 | text = None | |
2133 | #log.debug("XXX <- done read on %r" % self.fParent) | |
2134 | if not text: | |
2135 | # Empty text signifies that the pipe has been closed on | |
2136 | # the parent's end. | |
2137 | log.info("[%s] _InFileProxy: observed close of parent (%r)"\ | |
2138 | % (self.getName(), self.fParent)) | |
2139 | # Signal the child so it knows to stop listening. | |
2140 | try: | |
2141 | logres.info("[%s] _InFileProxy: closing child after "\ | |
2142 | "observing parent's close: %r", self.getName(), | |
2143 | self.fChild) | |
2144 | try: | |
2145 | self.fChild.close() | |
2146 | except IOError: | |
2147 | pass # Ignore: IOError: [Errno 4] Interrupted system call | |
2148 | except IOError, ex: | |
2149 | # Ignore: IOError: [Errno 9] Bad file descriptor | |
2150 | # XXX Do we *know* we want to do that? | |
2151 | pass | |
2152 | break | |
2153 | else: | |
2154 | log.info("[%s] _InFileProxy: read %d bytes from parent: %r"\ | |
2155 | % (self.getName(), len(text), text)) | |
2156 | ||
2157 | log.info("[%s, %s] _InFileProxy: writing %r to child (%r)", | |
2158 | self.getName(), id(self), text, self.fChild) | |
2159 | try: | |
2160 | self.fChild.write(text) | |
2161 | except (OSError, IOError), ex: | |
2162 | # Ignore errors for now. For example: | |
2163 | # - Get this on Win9x when writing multiple lines to "dir": | |
2164 | # OSError: [Errno 32] Broken pipe | |
2165 | #XXX There *may* be errors we don't want to avoid. | |
2166 | #XXX Should maybe just ignore EnvironmentError (base class). | |
2167 | log.info("[%s] _InFileProxy: error writing to child (%r), "\ | |
2168 | "closing: %s" % (self.getName(), self.fParent, ex)) | |
2169 | break | |
2170 | log.info("[%s] _InFileProxy: wrote %d bytes to child: %r"\ | |
2171 | % (self.getName(), len(text), text)) | |
2172 | ||
2173 | ||
2174 | class _OutFileProxy(threading.Thread): | |
2175 | """A thread to watch an "out" file from the spawned child process | |
2176 | and pass on write's to the parent. | |
2177 | """ | |
2178 | def __init__(self, fChild, fParent, toClose=[], name=None): | |
2179 | """ | |
2180 | "fChild" is a Win32 handle to the a child process' output pipe. | |
2181 | "fParent" is a Python file-like object setup for writing. | |
2182 | "toClose" is a list of objects on which to call .close when this | |
2183 | proxy is terminating. | |
2184 | "name" can be set for debugging, it will be used in log messages. | |
2185 | """ | |
2186 | log.info("[%s] _OutFileProxy.__init__(fChild=%r, fParent=%r, "\ | |
2187 | "toClose=%r)", name, fChild, fParent, toClose) | |
2188 | threading.Thread.__init__(self, name=name) | |
2189 | self.fChild = fChild | |
2190 | self.fParent = fParent | |
2191 | self.toClose = toClose | |
2192 | ||
2193 | def run(self): | |
2194 | log.info("[%s] _OutFileProxy: start" % self.getName()) | |
2195 | try: | |
2196 | self._proxyFromChildToParent() | |
2197 | finally: | |
2198 | logres.info("[%s] _OutFileProxy: terminating, close child (%r)", | |
2199 | self.getName(), self.fChild) | |
2200 | try: | |
2201 | self.fChild.close() | |
2202 | except IOError: | |
2203 | pass # Ignore: IOError: [Errno 4] Interrupted system call | |
2204 | log.info("[%s] _OutFileProxy: closing parent (%r)", | |
2205 | self.getName(), self.fParent) | |
2206 | try: | |
2207 | self.fParent.close() | |
2208 | except IOError: | |
2209 | pass # Ignore: IOError: [Errno 4] Interrupted system call | |
2210 | while self.toClose: | |
2211 | logres.info("[%s] _OutFileProxy: closing %r after "\ | |
2212 | "closing parent", self.getName(), self.toClose[0]) | |
2213 | try: | |
2214 | self.toClose[0].close() | |
2215 | except IOError: | |
2216 | pass # Ignore: IOError: [Errno 4] Interrupted system call | |
2217 | del self.toClose[0] | |
2218 | log.info("[%s] _OutFileProxy: done" % self.getName()) | |
2219 | ||
2220 | def _proxyFromChildToParent(self): | |
2221 | CHUNKSIZE = 4096 | |
2222 | # Read output from the child process, and (for now) just write | |
2223 | # it out. | |
2224 | while 1: | |
2225 | text = None | |
2226 | try: | |
2227 | log.info("[%s] _OutFileProxy: waiting for read on child (%r)"\ | |
2228 | % (self.getName(), self.fChild)) | |
2229 | text = self.fChild.read(CHUNKSIZE) | |
2230 | except IOError, ex: | |
2231 | # Ignore: IOError: [Errno 9] Bad file descriptor | |
2232 | # XXX Do we *know* we want to do that? | |
2233 | log.info("[%s] _OutFileProxy: error reading from child (%r), "\ | |
2234 | "shutting down: %s", self.getName(), self.fChild, ex) | |
2235 | break | |
2236 | if not text: | |
2237 | # Empty text signifies that the pipe has been closed on | |
2238 | # the child's end. | |
2239 | log.info("[%s] _OutFileProxy: observed close of child (%r)"\ | |
2240 | % (self.getName(), self.fChild)) | |
2241 | break | |
2242 | ||
2243 | log.info("[%s] _OutFileProxy: text(len=%d): %r", | |
2244 | self.getName(), len(text), text) | |
2245 | self.fParent.write(text) | |
2246 | ||
2247 | ||
2248 | ||
2249 | if sys.platform.startswith("linux"): | |
2250 | class _ThreadFixer: | |
2251 | """Mixin class for various classes in the Process hierarchy to | |
2252 | work around the known LinuxThreads bug where one cannot .wait() | |
2253 | on a created process from a subthread of the thread that created | |
2254 | the process. | |
2255 | ||
2256 | Usage: | |
2257 | class ProcessXXX(_ThreadFixer, BrokenProcessXXX): | |
2258 | _pclass = BrokenProcessXXX | |
2259 | ||
2260 | Details: | |
2261 | Because we must do all real os.wait() calls on the child | |
2262 | process from the thread that spawned it, we use a proxy | |
2263 | thread whose only responsibility is just that. The proxy | |
2264 | thread just starts the child and then immediately wait's for | |
2265 | the child to terminate. On termination is stores the exit | |
2266 | status (for use by the main thread) and notifies any thread | |
2267 | waiting for this termination (possibly the main thread). The | |
2268 | overriden .wait() uses this stored exit status and the | |
2269 | termination notification to simulate the .wait(). | |
2270 | """ | |
2271 | def __init__(self, *args, **kwargs): | |
2272 | # Keep a reference to 'log' ensure it is around for this object's | |
2273 | # destruction. | |
2274 | self.__log = log | |
2275 | self.__waiter = None | |
2276 | self.__hasTerminated = threading.Condition() | |
2277 | self.__terminationResult = None | |
2278 | self.__childStarted = threading.Condition() | |
2279 | self._pclass.__init__(self, *args, **kwargs) | |
2280 | ||
2281 | def _forkAndExecChildOnUnix(self, *args, **kwargs): | |
2282 | """Fork and start the child process do it in a special subthread | |
2283 | that will negotiate subsequent .wait()'s. | |
2284 | ||
2285 | Sets self._pid as a side effect. | |
2286 | """ | |
2287 | self.__waiter = threading.Thread( | |
2288 | target=self.__launchAndWait, args=args, kwargs=kwargs) | |
2289 | ||
2290 | # Start subthread that will launch child and wait until it | |
2291 | # *has* started. | |
2292 | self.__childStarted.acquire() | |
2293 | self.__waiter.start() | |
2294 | self.__childStarted.wait() | |
2295 | self.__childStarted.release() | |
2296 | ||
2297 | def __launchAndWait(self, *args, **kwargs): | |
2298 | """Launch the given command and wait for it to terminate. | |
2299 | ||
2300 | When the process has terminated then store its exit value | |
2301 | and finish. | |
2302 | """ | |
2303 | logfix.info("start child in thread %s", | |
2304 | threading.currentThread().getName()) | |
2305 | ||
2306 | # Spawn the child process and notify the main thread of | |
2307 | # this. | |
2308 | self.__childStarted.acquire() | |
2309 | self._pclass._forkAndExecChildOnUnix(self, *args, **kwargs) | |
2310 | self.__childStarted.notifyAll() | |
2311 | self.__childStarted.release() | |
2312 | ||
2313 | # Wait on the thread and store appropriate results when | |
2314 | # finished. | |
2315 | try: | |
2316 | waitResult = self._pclass.wait(self) | |
2317 | except ProcessError, ex: | |
2318 | waitResult = ex | |
2319 | self.__hasTerminated.acquire() | |
2320 | self.__terminationResult = waitResult | |
2321 | self.__hasTerminated.notifyAll() | |
2322 | self.__hasTerminated.release() | |
2323 | ||
2324 | self.__waiter = None # drop ref that would keep instance alive | |
2325 | ||
2326 | def wait(self, timeout=None): | |
2327 | # If the process __hasTerminated then return the exit | |
2328 | # status. Otherwise simulate the wait as appropriate. | |
2329 | # Note: | |
2330 | # - This class is only used on linux so 'timeout' has the | |
2331 | # Unix 'timeout' semantics. | |
2332 | self.__hasTerminated.acquire() | |
2333 | if self.__terminationResult is None: | |
2334 | if timeout == os.WNOHANG: # Poll. | |
2335 | self.__hasTerminated.wait(0) | |
2336 | else: # Block until process finishes. | |
2337 | self.__hasTerminated.wait() | |
2338 | terminationResult = self.__terminationResult | |
2339 | self.__hasTerminated.release() | |
2340 | ||
2341 | if terminationResult is None: | |
2342 | # process has not finished yet | |
2343 | raise ProcessError("Wait for process timed out.", | |
2344 | self.WAIT_TIMEOUT) | |
2345 | elif isinstance(terminationResult, Exception): | |
2346 | # some error waiting for process termination | |
2347 | raise terminationResult | |
2348 | else: | |
2349 | # the process terminated | |
2350 | return terminationResult | |
2351 | ||
2352 | _ThreadBrokenProcess = Process | |
2353 | class Process(_ThreadFixer, _ThreadBrokenProcess): | |
2354 | _pclass = _ThreadBrokenProcess | |
2355 | ||
2356 | _ThreadBrokenProcessOpen = ProcessOpen | |
2357 | class ProcessOpen(_ThreadFixer, _ThreadBrokenProcessOpen): | |
2358 | _pclass = _ThreadBrokenProcessOpen | |
2359 | ||
2360 | _ThreadBrokenProcessProxy = ProcessProxy | |
2361 | class ProcessProxy(_ThreadFixer, _ThreadBrokenProcessProxy): | |
2362 | _pclass = _ThreadBrokenProcessProxy | |
2363 | ||
2364 |