usr/src/tools/onbld/Checks/ProcessCheck.py

   1 #
   2 # CDDL HEADER START
   3 #
   4 # The contents of this file are subject to the terms of the
   5 # Common Development and Distribution License (the "License").
   6 # You may not use this file except in compliance with the License.
   7 #
   8 # You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9 # or http://www.opensolaris.org/os/licensing.
  10 # See the License for the specific language governing permissions
  11 # and limitations under the License.
  12 #
  13 # When distributing Covered Code, include this CDDL HEADER in each
  14 # file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15 # If applicable, add the following below this CDDL HEADER, with the
  16 # fields enclosed by brackets "[]" replaced with your own identifying
  17 # information: Portions Copyright [yyyy] [name of copyright owner]
  18 #
  19 # CDDL HEADER END
  20 #
  21
  22 #
  23 # Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
  24 # Use is subject to license terms.
  25 #
  26 # ident "%Z%%M% %I%     %E% SMI"
  27 #
  28
  29 #
  30 # Wrap a command-line check tool in a pythonic API
  31 #
  32
  33 import subprocess
  34 import tempfile
  35
  36 def processcheck(command, args, inpt, output):
  37         '''Run a checking command, command, with arguments as args.
  38         Input is provided by inpt (an iterable), error output is
  39         written to output (a stream-like entity).
  40
  41         Return a tuple (error, handle), where handle is a file handle
  42         (you must close it), containing output from the command.'''
  43
  44         #
  45         # We use a tempfile for output, rather than a pipe, so we
  46         # don't deadlock with the child if both pipes fill.
  47         #
  48         try:
  49                 tmpfile = tempfile.TemporaryFile(prefix=command)
  50         except EnvironmentError, e:
  51                 output.write("Could not create temporary file: %s\n" % e)
  52                 return (3, None)
  53
  54         try:
  55                 p = subprocess.Popen([command] + args,
  56                                      stdin=subprocess.PIPE, stdout=tmpfile,
  57                                      stderr=subprocess.STDOUT, close_fds=False)
  58         except OSError, e:
  59                 output.write("Could not execute %s: %s\n" % (command, e))
  60                 return (3, None)
  61
  62         for line in inpt:
  63                 p.stdin.write(line)
  64
  65         p.stdin.close()
  66
  67         ret = p.wait()
  68         tmpfile.seek(0)
  69
  70         return (ret < 0 and 1 or ret, tmpfile)