tools/buildstate.py

   1 #!/usr/bin/env vpython3
   2 # Copyright 2021 The Chromium Authors
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5 """
   6 This script checks to see what build commands are currently running by printing
   7 the command lines of any processes that are the children of ninja processes.
   8
   9 The idea is that if the build is serialized (not many build steps running) then
  10 you can run this to see what it is serialized on.
  11
  12 This uses python3 on Windows and vpython elsewhere (for psutil).
  13 """
  14
  15 import sys
  16
  17
  18 def main():
  19   parents = []
  20   processes = []
  21
  22   print('Gathering process data...')
  23   # Ninja's name on Linux is ninja-linux64, presumably different elsewhere, so
  24   # we look for a matching prefix.
  25   ninja_prefix = 'ninja.exe' if sys.platform in ['win32', 'cygwin'] else 'ninja'
  26
  27   if sys.platform in ['win32', 'cygwin']:
  28     # psutil handles short-lived ninja descendants poorly on Windows (it misses
  29     # most of them) so use wmic instead.
  30     import subprocess
  31     cmd = 'wmic process get Caption,ParentProcessId,ProcessId,CommandLine'
  32     lines = subprocess.check_output(cmd, universal_newlines=True).splitlines()
  33
  34     # Find the offsets for the various data columns by looking at the labels in
  35     # the first line of output.
  36     CAPTION_OFF = 0
  37     COMMAND_LINE_OFF = lines[0].find('CommandLine')
  38     PARENT_PID_OFF = lines[0].find('ParentProcessId')
  39     PID_OFF = lines[0].find(' ProcessId') + 1
  40
  41     for line in lines[1:]:
  42       # Ignore blank lines
  43       if not line.strip():
  44         continue
  45       command = line[:COMMAND_LINE_OFF].strip()
  46       command_line = line[COMMAND_LINE_OFF:PARENT_PID_OFF].strip()
  47       parent_pid = int(line[PARENT_PID_OFF:PID_OFF].strip())
  48       pid = int(line[PID_OFF:].strip())
  49       processes.append((command, command_line, parent_pid, pid))
  50
  51   else:
  52     # Portable process-collection code, but works badly on Windows.
  53     import psutil
  54     for proc in psutil.process_iter(['pid', 'ppid', 'name', 'cmdline']):
  55       try:
  56         cmdline = proc.cmdline()
  57         # Convert from list to a single string.
  58         cmdline = ' '.join(cmdline)
  59       except psutil.AccessDenied:
  60         cmdline = "Access denied"
  61       processes.append(
  62           (proc.name()[:], cmdline, int(proc.ppid()), int(proc.pid)))
  63
  64   # Scan the list of processes to find ninja.
  65   for process in processes:
  66     command, command_line, parent_pid, pid = process
  67     if command.startswith(ninja_prefix):
  68       parents.append(pid)
  69
  70   if not parents:
  71     print('No interesting parent processes found.')
  72     return 1
  73
  74   print('Tracking the children of these PIDs:')
  75   print(', '.join(map(lambda x: str(x), parents)))
  76
  77   print()
  78
  79   # Print all the processes that have parent-processes of interest.
  80   count = 0
  81   for process in processes:
  82     command, command_line, parent_pid, pid = process
  83     if parent_pid in parents:
  84       if not command_line:
  85         command_line = command
  86       print('%5d: %s' % (pid, command_line[:160]))
  87       count += 1
  88   print('Found %d children' % count)
  89   return 0
  90
  91
  92 if __name__ == '__main__':
  93   main()