Add tools to the repository.

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@17 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 156 insertions, 0 deletions

diff --git a/tools/python/google/process_utils.py b/tools/python/google/process_utils.py
new file mode 100644
index 0000000..966fdf6
--- /dev/null
+++ b/tools/python/google/process_utils.py
@@ -0,0 +1,156 @@
+# Copyright 2008, Google Inc.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
+#
+#    * Redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer.
+#    * Redistributions in binary form must reproduce the above
+# copyright notice, this list of conditions and the following disclaimer
+# in the documentation and/or other materials provided with the
+# distribution.
+#    * Neither the name of Google Inc. nor the names of its
+# contributors may be used to endorse or promote products derived from
+# this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+"""Shared process-related utility functions."""
+
+import errno
+import os
+import subprocess
+import sys
+
+class CommandNotFound(Exception): pass
+
+
+TASKKILL = os.path.join(os.environ['WINDIR'], 'system32', 'taskkill.exe')
+TASKKILL_PROCESS_NOT_FOUND_ERR = 128
+# On windows 2000 there is no taskkill.exe, we need to have pskill somewhere
+# in the path.
+PSKILL = 'pskill.exe'
+PSKILL_PROCESS_NOT_FOUND_ERR = -1
+
+def KillAll(executables):
+  """Tries to kill all copies of each process in the processes list.  Returns
+  an error if any running processes couldn't be killed.
+  """
+  result = 0
+  if os.path.exists(TASKKILL):
+    command = [TASKKILL, '/f', '/im']
+    process_not_found_err = TASKKILL_PROCESS_NOT_FOUND_ERR
+  else:
+    command = [PSKILL, '/t']
+    process_not_found_err = PSKILL_PROCESS_NOT_FOUND_ERR
+
+  for name in executables:
+    new_error = RunCommand(command + [name])
+    # Ignore "process not found" error.
+    if new_error != 0 and new_error != process_not_found_err:
+      result = new_error
+  return result
+
+def RunCommandFull(command, verbose=True, collect_output=False,
+                   print_output=True):
+  """Runs the command list.
+
+  Prints the given command (which should be a list of one or more strings).
+  If specified, prints its stderr (and optionally stdout) to stdout,
+  line-buffered, converting line endings to CRLF (see note below).  If
+  specified, collects the output as a list of lines and returns it.  Waits
+  for the command to terminate and returns its status.
+
+  Args:
+    command: the full command to run, as a list of one or more strings
+    verbose: if True, combines all output (stdout and stderr) into stdout.
+             Otherwise, prints only the command's stderr to stdout.
+    collect_output: if True, collects the output of the command as a list of
+                    lines and returns it
+    print_output: if True, prints the output of the command
+
+  Returns:
+    A tuple consisting of the process's exit status and output.  If
+    collect_output is False, the output will be [].
+
+  Raises:
+    CommandNotFound if the command executable could not be found.
+  """
+  print '\n' + subprocess.list2cmdline(command).replace('\\', '/') + '\n', ###
+
+  if verbose:
+    out = subprocess.PIPE
+    err = subprocess.STDOUT
+  else:
+    out = file(os.devnull, 'w')
+    err = subprocess.PIPE
+  try:
+    proc = subprocess.Popen(command, stdout=out, stderr=err, bufsize=1)
+  except OSError, e:
+    if e.errno == errno.ENOENT:
+      raise CommandNotFound('Unable to find "%s"' % command[0])
+    raise
+
+  output = []
+
+  if verbose:
+    read_from = proc.stdout
+  else:
+    read_from = proc.stderr
+  line = read_from.readline()
+  while line:
+    line = line.rstrip()
+
+    if collect_output:
+      output.append(line)
+
+    if print_output:
+      # Windows Python converts \n to \r\n automatically whenever it
+      # encounters it written to a text file (including stdout).  The only
+      # way around it is to write to a binary file, which isn't feasible for
+      # stdout. So we end up with \r\n here even though we explicitly write
+      # \n.  (We could write \r instead, which doesn't get converted to \r\n,
+      # but that's probably more troublesome for people trying to read the
+      # files.)
+      print line + '\n',
+
+      # Python on windows writes the buffer only when it reaches 4k. This is
+      # not fast enough for all purposes.
+      sys.stdout.flush()
+    line = read_from.readline()
+
+  if not verbose:
+    out.close()
+  return (proc.returncode, output)
+
+def RunCommand(command, verbose=True):
+  """Runs the command list, printing its output and returning its exit status.
+
+  Prints the given command (which should be a list of one or more strings),
+  then runs it and prints its stderr (and optionally stdout) to stdout,
+  line-buffered, converting line endings to CRLF.  Waits for the command to
+  terminate and returns its status.
+
+  Args:
+    command: the full command to run, as a list of one or more strings
+    verbose: if True, combines all output (stdout and stderr) into stdout.
+             Otherwise, prints only the command's stderr to stdout.
+
+  Returns:
+    The process's exit status.
+
+  Raises:
+    CommandNotFound if the command executable could not be found.
+  """
+  return RunCommandFull(command, verbose)[0]