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
96
97
98
99
|
#!/usr/bin/env python
#
# Copyright (c) 2013 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.
"""Provisions Android devices with settings required for bots.
Usage:
./provision_devices.py [-d <device serial number>]
"""
import optparse
import os
import re
import subprocess
import sys
import time
from pylib import android_commands
from pylib import constants
def KillHostHeartbeat():
ps = subprocess.Popen(['ps', 'aux'], stdout = subprocess.PIPE)
stdout, _ = ps.communicate()
matches = re.findall('\\n.*host_heartbeat.*', stdout)
for match in matches:
print 'An instance of host heart beart running... will kill'
pid = re.findall('(\d+)', match)[1]
subprocess.call(['kill', str(pid)])
def LaunchHostHeartbeat():
# Kill if existing host_heartbeat
KillHostHeartbeat()
# Launch a new host_heartbeat
print 'Spawning host heartbeat...'
subprocess.Popen([os.path.join(constants.DIR_SOURCE_ROOT,
'build/android/host_heartbeat.py')])
def PushAndLaunchAdbReboot(devices, target):
"""Pushes and launches the adb_reboot binary on the device.
Arguments:
devices: The list of serial numbers of the device to which the
adb_reboot binary should be pushed.
target : The build target (example, Debug or Release) which helps in
locating the adb_reboot binary.
"""
for device in devices:
print 'Will push and launch adb_reboot on %s' % device
android_cmd = android_commands.AndroidCommands(device)
# Kill if adb_reboot is already running.
android_cmd.KillAllBlocking('adb_reboot', 2)
# Push adb_reboot
print ' Pushing adb_reboot ...'
adb_reboot = os.path.join(constants.DIR_SOURCE_ROOT,
'out/%s/adb_reboot' % target)
android_cmd.PushIfNeeded(adb_reboot, '/data/local/tmp/')
# Launch adb_reboot
print ' Launching adb_reboot ...'
p = subprocess.Popen(['adb', '-s', device, 'shell'], stdin=subprocess.PIPE)
p.communicate('/data/local/tmp/adb_reboot; exit\n')
LaunchHostHeartbeat()
def ProvisionDevices(options):
if options.device is not None:
devices = [options.device]
else:
devices = android_commands.GetAttachedDevices()
for device in devices:
android_cmd = android_commands.AndroidCommands(device)
android_cmd.RunShellCommandWithSU('date -u %f' % time.time())
if options.auto_reconnect:
PushAndLaunchAdbReboot(devices, options.target)
def main(argv):
parser = optparse.OptionParser()
parser.add_option('-d', '--device',
help='The serial number of the device to be provisioned')
parser.add_option('-t', '--target', default='Debug', help='The build target')
parser.add_option(
'-r', '--auto-reconnect', action='store_true',
help='Push binary which will reboot the device on adb disconnections.')
options, args = parser.parse_args(argv[1:])
constants.SetBuildType(options.target)
if args:
print >> sys.stderr, 'Unused args %s' % args
return 1
ProvisionDevices(options)
if __name__ == '__main__':
sys.exit(main(sys.argv))
|
