OLD | NEW |
| (Empty) |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | |
2 # Use of this source code is governed by a BSD-style license that can be | |
3 # found in the LICENSE file. | |
4 | |
5 import logging | |
6 import psutil | |
7 import signal | |
8 | |
9 from pylib.device import device_errors | |
10 from pylib.device import device_utils | |
11 | |
12 | |
13 def _KillWebServers(): | |
14 for s in [signal.SIGTERM, signal.SIGINT, signal.SIGQUIT, signal.SIGKILL]: | |
15 signalled = [] | |
16 for server in ['lighttpd', 'webpagereplay']: | |
17 for p in psutil.process_iter(): | |
18 try: | |
19 if not server in ' '.join(p.cmdline): | |
20 continue | |
21 logging.info('Killing %s %s %s', s, server, p.pid) | |
22 p.send_signal(s) | |
23 signalled.append(p) | |
24 except Exception as e: | |
25 logging.warning('Failed killing %s %s %s', server, p.pid, e) | |
26 for p in signalled: | |
27 try: | |
28 p.wait(1) | |
29 except Exception as e: | |
30 logging.warning('Failed waiting for %s to die. %s', p.pid, e) | |
31 | |
32 | |
33 def CleanupLeftoverProcesses(): | |
34 """Clean up the test environment, restarting fresh adb and HTTP daemons.""" | |
35 _KillWebServers() | |
36 device_utils.RestartServer() | |
37 | |
38 def cleanup_device(d): | |
39 d.old_interface.RestartAdbdOnDevice() | |
40 try: | |
41 d.EnableRoot() | |
42 except device_errors.CommandFailedError as e: | |
43 logging.error(str(e)) | |
44 d.WaitUntilFullyBooted() | |
45 | |
46 device_utils.DeviceUtils.parallel().pMap(cleanup_device) | |
47 | |
OLD | NEW |