1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
# Copyright 2021 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import subprocess
import logging
import psutil
import os
import signal
def TerminateProcess(process: subprocess.Popen):
"""Terminates `process` and ensures it's cleaned up before returning.
Raises:
RuntimeError: When failed to terminate the process after a timeout.
"""
logging.debug(f"Terminating PID:{process.pid}")
try:
process.terminate()
process.wait(2.0)
except (psutil.TimeoutExpired, psutil.AccessDenied, PermissionError) as e:
logging.info(f"Terminate failed, moving on to kill.")
except psutil.NoSuchProcess:
return
else:
return
try:
process.kill()
process.wait(2.0)
except (psutil.TimeoutExpired, psutil.AccessDenied, PermissionError) as e:
raise RuntimeError(f"Could not clean up PID:{process.pid}.") from e
else:
return
def SendSignalToRootProcess(process: subprocess.Popen, signal: signal.Signals):
os.system(f"sudo kill -{signal.value} {process.pid}")
def TerminateRootProcess(process: subprocess.Popen):
"""Kills elevated `process` and ensures it's cleaned up before returning.
Raises:
RuntimeError: When failed to terminate the process after a timeout.
"""
logging.debug(f"Terminating PID:{process.pid}")
try:
SendSignalToRootProcess(process, signal.SIGTERM)
process.wait(2.0)
except (psutil.TimeoutExpired, psutil.AccessDenied, PermissionError) as e:
raise RuntimeError(f"Could not clean up PID:{process.pid}.") from e
except psutil.NoSuchProcess:
return
else:
return
def FindProcess(process_name: str) -> psutil.Process:
"""Looks for the process associated with |browser|.
Args:
process_name: Name of the process to return.
Returns:
A psutil.process representation of the process matching `process_name`.
Raises:
RuntimeError: When no process is found for the browser.
"""
processes = filter(lambda p: p.name() == process_name, psutil.process_iter())
returned_process = None
for process in processes:
if not returned_process:
returned_process = process
else:
raise RuntimeError(
"Too many copies of the process running, this is wrong.")
return returned_process
def FindChromiumSrcDir(path: os.PathLike) -> os.PathLike:
path = os.path.abspath(path)
while path:
(head, tail) = os.path.split(path)
if tail == "src":
return path
path = head
|
