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
|
#!/usr/bin/python
# Copyright (c) 2011 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import os
import signal
import subprocess
import time
class BrowserProcessBase(object):
def __init__(self, handle):
self.handle = handle
print 'PID', self.handle.pid
def GetReturnCode(self):
return self.handle.returncode
def IsRunning(self):
return self.handle.poll() is None
def Wait(self, wait_steps, sleep_time):
try:
self.term()
except Exception:
# Terminating the handle can raise an exception. There is likely no point
# in waiting if the termination didn't succeed.
return
i = 0
# subprocess.wait() doesn't have a timeout, unfortunately.
while self.IsRunning() and i < wait_steps:
time.sleep(sleep_time)
i += 1
def Kill(self):
if self.IsRunning():
print 'KILLING the browser'
try:
self.kill()
# If it doesn't die, we hang. Oh well.
self.handle.wait()
except Exception:
# If it is already dead, then it's ok.
# This may happen if the browser dies after the first poll, but
# before the kill.
if self.IsRunning():
raise
class BrowserProcess(BrowserProcessBase):
def term(self):
self.handle.terminate()
def kill(self):
self.handle.kill()
class BrowserProcessPosix(BrowserProcessBase):
""" This variant of BrowserProcess uses process groups to manage browser
life time. """
def term(self):
os.killpg(self.handle.pid, signal.SIGTERM)
def kill(self):
os.killpg(self.handle.pid, signal.SIGKILL)
def RunCommandWithSubprocess(cmd, env=None):
handle = subprocess.Popen(cmd, env=env)
return BrowserProcess(handle)
def RunCommandInProcessGroup(cmd, env=None):
def SetPGrp():
os.setpgrp()
print 'I\'M THE SESSION LEADER!'
handle = subprocess.Popen(cmd, env=env, preexec_fn=SetPGrp)
return BrowserProcessPosix(handle)
|
