|
1 #!/bin/bash |
|
2 # |
|
3 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
|
4 # Use of this source code is governed by a BSD-style license that can be |
|
5 # found in the LICENSE file. |
|
6 # |
|
7 # A collection of functions useful for maintaining android devices |
|
8 |
|
9 |
|
10 # Run an adb command on all connected device in parallel. |
|
11 # Usage: adb_all command line to eval. Quoting is optional. |
|
12 # |
|
13 # Examples: |
|
14 # adb_all install Chrome.apk |
|
15 # adb_all 'shell cat /path/to/file' |
|
16 # |
|
17 adb_all() { |
|
18 if [[ $# == 0 ]]; then |
|
19 echo "Usage: adb_all <adb command>. Quoting is optional." |
|
20 echo "Example: adb_all install Chrome.apk" |
|
21 return 1 |
|
22 fi |
|
23 local DEVICES=$(adb_get_devices -b) |
|
24 local NUM_DEVICES=$(echo $DEVICES | wc -w) |
|
25 if (( $NUM_DEVICES > 1 )); then |
|
26 echo "Looping over $NUM_DEVICES devices" |
|
27 fi |
|
28 _adb_multi "$DEVICES" "$*" |
|
29 } |
|
30 |
|
31 |
|
32 # Run a command on each connected device. Quoting the command is suggested but |
|
33 # not required. The script setups up variable DEVICE to correspond to the |
|
34 # current serial number. Intended for complex one_liners that don't work in |
|
35 # adb_all |
|
36 # Usage: adb_device_loop 'command line to eval' |
|
37 adb_device_loop() { |
|
38 if [[ $# == 0 ]]; then |
|
39 echo "Intended for more complex one-liners that cannot be done with" \ |
|
40 "adb_all." |
|
41 echo 'Usage: adb_device_loop "echo $DEVICE: $(adb root &&' \ |
|
42 'adb shell cat /data/local.prop)"' |
|
43 return 1 |
|
44 fi |
|
45 local DEVICES=$(adb_get_devices) |
|
46 if [[ -z $DEVICES ]]; then |
|
47 return |
|
48 fi |
|
49 # Do not change DEVICE variable name - part of api |
|
50 for DEVICE in $DEVICES; do |
|
51 DEV_TYPE=$(adb -s $DEVICE shell getprop ro.product.device | sed 's/\r//') |
|
52 echo "Running on $DEVICE ($DEV_TYPE)" |
|
53 ANDROID_SERIAL=$DEVICE eval "$*" |
|
54 done |
|
55 } |
|
56 |
|
57 # Erases data from any devices visible on adb. To preserve a device, |
|
58 # disconnect it or: |
|
59 # 1) Reboot it into fastboot with 'adb reboot bootloader' |
|
60 # 2) Run wipe_all_devices to wipe remaining devices |
|
61 # 3) Restore device it with 'fastboot reboot' |
|
62 # |
|
63 # Usage: wipe_all_devices [-f] |
|
64 # |
|
65 wipe_all_devices() { |
|
66 if [[ -z $(which adb) || -z $(which fastboot) ]]; then |
|
67 echo "aborting: adb and fastboot not in path" |
|
68 return 1 |
|
69 elif ! $(groups | grep -q 'plugdev'); then |
|
70 echo "If fastboot fails, run: 'sudo adduser $(whoami) plugdev'" |
|
71 fi |
|
72 |
|
73 local DEVICES=$(adb_get_devices -b) |
|
74 |
|
75 if [[ $1 != '-f' ]]; then |
|
76 echo "This will ERASE ALL DATA from $(echo $DEVICES | wc -w) device." |
|
77 read -p "Hit enter to continue" |
|
78 fi |
|
79 |
|
80 _adb_multi "$DEVICES" "reboot bootloader" |
|
81 # Subshell to isolate job list |
|
82 ( |
|
83 for DEVICE in $DEVICES; do |
|
84 fastboot_erase $DEVICE & |
|
85 done |
|
86 wait |
|
87 ) |
|
88 |
|
89 # Reboot devices together |
|
90 for DEVICE in $DEVICES; do |
|
91 fastboot -s $DEVICE reboot |
|
92 done |
|
93 } |
|
94 |
|
95 # Wipe a device in fastboot. |
|
96 # Usage fastboot_erase [serial] |
|
97 fastboot_erase() { |
|
98 if [[ -n $1 ]]; then |
|
99 echo "Wiping $1" |
|
100 local SERIAL="-s $1" |
|
101 else |
|
102 if [ -z $(fastboot devices) ]; then |
|
103 echo "No devices in fastboot, aborting." |
|
104 echo "Check out wipe_all_devices to see if sufficient" |
|
105 echo "You can put a device in fastboot using adb reboot bootloader" |
|
106 return 1 |
|
107 fi |
|
108 local SERIAL="" |
|
109 fi |
|
110 fastboot $SERIAL erase cache |
|
111 fastboot $SERIAL erase userdata |
|
112 } |
|
113 |
|
114 # Get list of devices connected via adb |
|
115 # Args: -b block until adb detects a device |
|
116 adb_get_devices() { |
|
117 local DEVICES="$(adb devices | grep 'device$')" |
|
118 if [[ -z $DEVICES && $1 == '-b' ]]; then |
|
119 echo '- waiting for device -' >&2 |
|
120 local DEVICES="$(adb wait-for-device devices | grep 'device$')" |
|
121 fi |
|
122 echo "$DEVICES" | awk -vORS=' ' '{print $1}' | sed 's/ $/\n/' |
|
123 } |
|
124 |
|
125 ################################################### |
|
126 ## HELPER FUNCTIONS |
|
127 ################################################### |
|
128 |
|
129 # Run an adb command in parallel over a device list |
|
130 _adb_multi() { |
|
131 local DEVICES=$1 |
|
132 local ADB_ARGS=$2 |
|
133 ( |
|
134 for DEVICE in $DEVICES; do |
|
135 adb -s $DEVICE $ADB_ARGS & |
|
136 done |
|
137 wait |
|
138 ) |
|
139 } |