Chromium Code Reviews| Index: tools/testrunner/local/commands.py |
| diff --git a/tools/testrunner/local/commands.py b/tools/testrunner/local/commands.py |
| index d6445d0c7a67838f863cef5b6f2370f32f8790aa..1bbc460912d18d84815448880059a9d5b78cfa56 100644 |
| --- a/tools/testrunner/local/commands.py |
| +++ b/tools/testrunner/local/commands.py |
| @@ -27,27 +27,13 @@ |
| import os |
| -import signal |
| import subprocess |
| -import sys |
| -import tempfile |
| -import time |
| +from threading import Timer |
| from ..local import utils |
| from ..objects import output |
| -def KillProcessWithID(pid): |
| - if utils.IsWindows(): |
| - os.popen('taskkill /T /F /PID %d' % pid) |
| - else: |
| - os.kill(pid, signal.SIGTERM) |
| - |
| - |
| -MAX_SLEEP_TIME = 0.1 |
| -INITIAL_SLEEP_TIME = 0.0001 |
| -SLEEP_TIME_FACTOR = 1.25 |
| - |
| SEM_INVALID_VALUE = -1 |
| SEM_NOGPFAULTERRORBOX = 0x0002 # Microsoft Platform SDK WinBase.h |
| @@ -76,76 +62,33 @@ def RunProcess(verbose, timeout, args, **rest): |
| prev_error_mode = Win32SetErrorMode(error_mode) |
| Win32SetErrorMode(error_mode | prev_error_mode) |
| process = subprocess.Popen( |
| - shell=utils.IsWindows(), |
| args=popen_args, |
| + stdout=subprocess.PIPE, |
| + stderr=subprocess.PIPE, |
| **rest |
| ) |
| if (utils.IsWindows() and prev_error_mode != SEM_INVALID_VALUE): |
| Win32SetErrorMode(prev_error_mode) |
| - # Compute the end time - if the process crosses this limit we |
| - # consider it timed out. |
| - if timeout is None: end_time = None |
| - else: end_time = time.time() + timeout |
| - timed_out = False |
| - # Repeatedly check the exit code from the process in a |
| - # loop and keep track of whether or not it times out. |
| - exit_code = None |
| - sleep_time = INITIAL_SLEEP_TIME |
| - while exit_code is None: |
| - if (not end_time is None) and (time.time() >= end_time): |
| - # Kill the process and wait for it to exit. |
| - KillProcessWithID(process.pid) |
| - exit_code = process.wait() |
| - timed_out = True |
| - else: |
| - exit_code = process.poll() |
| - time.sleep(sleep_time) |
| - sleep_time = sleep_time * SLEEP_TIME_FACTOR |
| - if sleep_time > MAX_SLEEP_TIME: |
| - sleep_time = MAX_SLEEP_TIME |
| - return (exit_code, timed_out) |
| + def kill_process(process, timeout_result): |
| + timeout_result[0] = True |
| + process.kill() |
|
tandrii(chromium)
2015/05/08 15:47:26
This will raise exception if process is already de
Michael Achenbach
2015/05/11 08:13:00
Done.
|
| -def PrintError(string): |
| - sys.stderr.write(string) |
| - sys.stderr.write("\n") |
| + # Pseudo object to communicate with timer thread. |
| + timeout_result = [False] |
| - |
| -def CheckedUnlink(name): |
| - # On Windows, when run with -jN in parallel processes, |
| - # OS often fails to unlink the temp file. Not sure why. |
| - # Need to retry. |
| - # Idea from https://bugs.webkit.org/attachment.cgi?id=75982&action=prettypatch |
| - retry_count = 0 |
| - while retry_count < 30: |
| - try: |
| - os.unlink(name) |
| - return |
| - except OSError, e: |
| - retry_count += 1 |
| - time.sleep(retry_count * 0.1) |
| - PrintError("os.unlink() " + str(e)) |
| + timer = Timer(timeout, kill_process, [process, timeout_result]) |
| + timer.start() |
| + stdout, stderr = process.communicate() |
| + timer.cancel() |
| + return process.returncode, timeout_result[0], stdout, stderr |
| def Execute(args, verbose=False, timeout=None): |
| - try: |
| - args = [ c for c in args if c != "" ] |
| - (fd_out, outname) = tempfile.mkstemp() |
| - (fd_err, errname) = tempfile.mkstemp() |
| - (exit_code, timed_out) = RunProcess( |
| - verbose, |
| - timeout, |
| - args=args, |
| - stdout=fd_out, |
| - stderr=fd_err |
| - ) |
| - finally: |
| - # TODO(machenbach): A keyboard interrupt before the assignment to |
| - # fd_out|err can lead to reference errors here. |
| - os.close(fd_out) |
| - os.close(fd_err) |
| - out = file(outname).read() |
| - errors = file(errname).read() |
| - CheckedUnlink(outname) |
| - CheckedUnlink(errname) |
| - return output.Output(exit_code, timed_out, out, errors) |
| + args = [ c for c in args if c != "" ] |
| + exit_code, timed_out, stdout, stderr = RunProcess( |
| + verbose, |
| + timeout, |
| + args=args, |
| + ) |
| + return output.Output(exit_code, timed_out, stdout, stderr) |