| Takuto Ikuta | e840a3d | 2022-08-03 01:07:56 | [diff] [blame] | 1 | #!/usr/bin/env vpython3 |
| Avi Drissman | dfd88085 | 2022-09-15 20:11:09 | [diff] [blame] | 2 | # Copyright 2021 The Chromium Authors |
| Bruce Dawson | df6fc16 | 2021-02-05 01:55:50 | [diff] [blame] | 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 | |
| Takuto Ikuta | e840a3d | 2022-08-03 01:07:56 | [diff] [blame] | 12 | This uses python3 on Windows and vpython elsewhere (for psutil). |
| Bruce Dawson | df6fc16 | 2021-02-05 01:55:50 | [diff] [blame] | 13 | """ |
| 14 | |
| Bruce Dawson | df6fc16 | 2021-02-05 01:55:50 | [diff] [blame] | 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.platformin['win32','cygwin']else'ninja' |
| 26 | |
| 27 | if sys.platformin['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 linein lines[1:]: |
| 42 | # Ignore blank lines |
| 43 | ifnot 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 procin 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 processin processes: |
| 66 | command, command_line, parent_pid, pid= process |
| 67 | if command.startswith(ninja_prefix): |
| 68 | parents.append(pid) |
| 69 | |
| 70 | ifnot parents: |
| 71 | print('No interesting parent processes found.') |
| 72 | return1 |
| 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 processin processes: |
| 82 | command, command_line, parent_pid, pid= process |
| 83 | if parent_pidin parents: |
| 84 | ifnot command_line: |
| 85 | command_line= command |
| 86 | print('%5d: %s'%(pid, command_line[:160])) |
| 87 | count+=1 |
| 88 | print('Found %d children'% count) |
| 89 | return0 |
| 90 | |
| 91 | |
| 92 | if __name__=='__main__': |
| 93 | main() |
