• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2012 the V8 project authors. All rights reserved.
2# Redistribution and use in source and binary forms, with or without
3# modification, are permitted provided that the following conditions are
4# met:
5#
6#     * Redistributions of source code must retain the above copyright
7#       notice, this list of conditions and the following disclaimer.
8#     * Redistributions in binary form must reproduce the above
9#       copyright notice, this list of conditions and the following
10#       disclaimer in the documentation and/or other materials provided
11#       with the distribution.
12#     * Neither the name of Google Inc. nor the names of its
13#       contributors may be used to endorse or promote products derived
14#       from this software without specific prior written permission.
15#
16# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
20# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27
28
29import os
30import signal
31import subprocess
32import sys
33import tempfile
34import time
35
36from ..local import utils
37from ..objects import output
38
39
40def KillProcessWithID(pid):
41  if utils.IsWindows():
42    os.popen('taskkill /T /F /PID %d' % pid)
43  else:
44    os.kill(pid, signal.SIGTERM)
45
46
47MAX_SLEEP_TIME = 0.1
48INITIAL_SLEEP_TIME = 0.0001
49SLEEP_TIME_FACTOR = 1.25
50
51SEM_INVALID_VALUE = -1
52SEM_NOGPFAULTERRORBOX = 0x0002  # Microsoft Platform SDK WinBase.h
53
54
55def Win32SetErrorMode(mode):
56  prev_error_mode = SEM_INVALID_VALUE
57  try:
58    import ctypes
59    prev_error_mode = \
60        ctypes.windll.kernel32.SetErrorMode(mode)  #@UndefinedVariable
61  except ImportError:
62    pass
63  return prev_error_mode
64
65
66def RunProcess(verbose, timeout, args, **rest):
67  if verbose: print "#", " ".join(args)
68  popen_args = args
69  prev_error_mode = SEM_INVALID_VALUE
70  if utils.IsWindows():
71    popen_args = subprocess.list2cmdline(args)
72    # Try to change the error mode to avoid dialogs on fatal errors. Don't
73    # touch any existing error mode flags by merging the existing error mode.
74    # See http://blogs.msdn.com/oldnewthing/archive/2004/07/27/198410.aspx.
75    error_mode = SEM_NOGPFAULTERRORBOX
76    prev_error_mode = Win32SetErrorMode(error_mode)
77    Win32SetErrorMode(error_mode | prev_error_mode)
78  process = subprocess.Popen(
79    shell=utils.IsWindows(),
80    args=popen_args,
81    **rest
82  )
83  if (utils.IsWindows() and prev_error_mode != SEM_INVALID_VALUE):
84    Win32SetErrorMode(prev_error_mode)
85  # Compute the end time - if the process crosses this limit we
86  # consider it timed out.
87  if timeout is None: end_time = None
88  else: end_time = time.time() + timeout
89  timed_out = False
90  # Repeatedly check the exit code from the process in a
91  # loop and keep track of whether or not it times out.
92  exit_code = None
93  sleep_time = INITIAL_SLEEP_TIME
94  while exit_code is None:
95    if (not end_time is None) and (time.time() >= end_time):
96      # Kill the process and wait for it to exit.
97      KillProcessWithID(process.pid)
98      exit_code = process.wait()
99      timed_out = True
100    else:
101      exit_code = process.poll()
102      time.sleep(sleep_time)
103      sleep_time = sleep_time * SLEEP_TIME_FACTOR
104      if sleep_time > MAX_SLEEP_TIME:
105        sleep_time = MAX_SLEEP_TIME
106  return (exit_code, timed_out)
107
108
109def PrintError(string):
110  sys.stderr.write(string)
111  sys.stderr.write("\n")
112
113
114def CheckedUnlink(name):
115  # On Windows, when run with -jN in parallel processes,
116  # OS often fails to unlink the temp file. Not sure why.
117  # Need to retry.
118  # Idea from https://bugs.webkit.org/attachment.cgi?id=75982&action=prettypatch
119  retry_count = 0
120  while retry_count < 30:
121    try:
122      os.unlink(name)
123      return
124    except OSError, e:
125      retry_count += 1
126      time.sleep(retry_count * 0.1)
127  PrintError("os.unlink() " + str(e))
128
129
130def Execute(args, verbose=False, timeout=None):
131  try:
132    args = [ c for c in args if c != "" ]
133    (fd_out, outname) = tempfile.mkstemp()
134    (fd_err, errname) = tempfile.mkstemp()
135    (exit_code, timed_out) = RunProcess(
136      verbose,
137      timeout,
138      args=args,
139      stdout=fd_out,
140      stderr=fd_err
141    )
142  finally:
143    # TODO(machenbach): A keyboard interrupt before the assignment to
144    # fd_out|err can lead to reference errors here.
145    os.close(fd_out)
146    os.close(fd_err)
147    out = file(outname).read()
148    errors = file(errname).read()
149    CheckedUnlink(outname)
150    CheckedUnlink(errname)
151  return output.Output(exit_code, timed_out, out, errors)
152