blob: 5c8eee7f5aeaa77e94fadc0c5734ebaf6fe954b2 [file] [log] [blame]
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001/*
2 * Copyright (C) 2007 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
Yabin Cuiaed3c612015-09-22 15:52:57 -070017#define TRACE_TAG ADB
Dan Albert33134262015-03-19 15:21:08 -070018
19#include "sysdeps.h"
20
Dan Albert76649012015-02-24 15:51:19 -080021#include <assert.h>
22#include <ctype.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080023#include <errno.h>
Elliott Hughes2940ccf2015-04-17 14:07:52 -070024#include <inttypes.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080025#include <limits.h>
26#include <stdarg.h>
Dan Albert76649012015-02-24 15:51:19 -080027#include <stdint.h>
28#include <stdio.h>
29#include <stdlib.h>
30#include <string.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080031#include <sys/stat.h>
Dan Albert76649012015-02-24 15:51:19 -080032#include <sys/types.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080033
David Pursell606835a2015-09-08 17:17:02 -070034#include <memory>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070035#include <string>
Josh Gao05786772015-10-30 16:57:19 -070036#include <vector>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070037
Elliott Hughese8b663f2016-05-26 22:43:19 -070038#include <android-base/file.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080039#include <android-base/logging.h>
Tao Wu135f4ab2016-09-16 13:01:24 -070040#include <android-base/parseint.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080041#include <android-base/stringprintf.h>
42#include <android-base/strings.h>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070043
Yabin Cuid325e862014-11-17 14:48:25 -080044#if !defined(_WIN32)
Josh Gao8dcdb572015-10-23 15:03:31 -070045#include <signal.h>
Elliott Hughes0fbf9612015-11-04 13:07:47 -080046#include <sys/ioctl.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080047#include <termios.h>
Dan Albert76649012015-02-24 15:51:19 -080048#include <unistd.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080049#endif
50
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080051#include "adb.h"
Nick Kralevichbea3f9c2014-11-13 15:17:29 -080052#include "adb_auth.h"
Dan Albertcc731cc2015-02-24 21:26:58 -080053#include "adb_client.h"
54#include "adb_io.h"
Josh Gao924d35a2016-08-30 15:39:25 -070055#include "adb_unique_fd.h"
Elliott Hughes58305772015-04-17 13:57:15 -070056#include "adb_utils.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080057#include "file_sync_service.h"
David Pursell70ef7b42015-09-30 13:35:42 -070058#include "services.h"
David Pursell606835a2015-09-08 17:17:02 -070059#include "shell_service.h"
60#include "transport.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080061
Elliott Hughes3bd73c12015-05-05 13:10:43 -070062static int install_app(TransportType t, const char* serial, int argc, const char** argv);
63static int install_multiple_app(TransportType t, const char* serial, int argc, const char** argv);
64static int uninstall_app(TransportType t, const char* serial, int argc, const char** argv);
Todd Kennedy6fa848a2015-11-03 16:53:08 -080065static int install_app_legacy(TransportType t, const char* serial, int argc, const char** argv);
66static int uninstall_app_legacy(TransportType t, const char* serial, int argc, const char** argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080067
Josh Gaob7b1edf2015-11-11 17:56:12 -080068static auto& gProductOutPath = *new std::string();
Matt Gumbeld7b33082012-11-14 10:16:17 -080069extern int gListenAll;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080070
Felipe Leme44a42672016-04-01 17:43:27 -070071static constexpr char BUGZ_OK_PREFIX[] = "OK:";
72static constexpr char BUGZ_FAIL_PREFIX[] = "FAIL:";
73
Elliott Hughes58305772015-04-17 13:57:15 -070074static std::string product_file(const char *extra) {
75 if (gProductOutPath.empty()) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080076 fprintf(stderr, "adb: Product directory not specified; "
77 "use -p or define ANDROID_PRODUCT_OUT\n");
78 exit(1);
79 }
80
Elliott Hughes58305772015-04-17 13:57:15 -070081 return android::base::StringPrintf("%s%s%s",
82 gProductOutPath.c_str(), OS_PATH_SEPARATOR_STR, extra);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080083}
84
Elliott Hughes58305772015-04-17 13:57:15 -070085static void help() {
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070086 fprintf(stdout, "%s\n", adb_version().c_str());
Felipe Lemeb3239722016-07-29 17:57:00 -070087 // clang-format off
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070088 fprintf(stdout,
89 "global options:\n"
90 " -a listen on all network interfaces, not just localhost\n"
91 " -d use USB device (error if multiple devices connected)\n"
92 " -e use emulator (error if multiple emulators running)\n"
93 " -s SERIAL\n"
94 " use device/emulator with given serial number (overrides $ANDROID_SERIAL)\n"
95 " -p PRODUCT\n"
96 " name or path ('angler'/'out/target/product/angler');\n"
97 " default $ANDROID_PRODUCT_OUT\n"
98 " -H name of adb server host [default=localhost]\n"
99 " -P port of adb server [default=5037]\n"
100 " -L SOCKET listen on given socket for adb server [default=tcp:localhost:5037]\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800101 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700102 "general commands:\n"
103 " devices [-l] list connected devices (-l for long output)\n"
104 " help show this help message\n"
105 " version show version num\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700106 "\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800107 "networking:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700108 " connect HOST[:PORT] connect to a device via TCP/IP [default port=5555]\n"
109 " disconnect [HOST[:PORT]]\n"
110 " disconnect from given TCP/IP device [default port=5555], or all\n"
111 " forward --list list all forward socket connections\n"
112 " forward [--no-rebind] LOCAL REMOTE\n"
113 " forward socket connection using:\n"
114 " tcp:<port> (<local> may be \"tcp:0\" to pick any open port)\n"
115 " localabstract:<unix domain socket name>\n"
116 " localreserved:<unix domain socket name>\n"
117 " localfilesystem:<unix domain socket name>\n"
118 " dev:<character device name>\n"
119 " jdwp:<process pid> (remote only)\n"
120 " forward --remove LOCAL remove specific forward socket connection\n"
121 " forward --remove-all remove all forward socket connections\n"
122 " ppp TTY [PARAMETER...] run PPP over USB\n"
123 " reverse --list list all reverse socket connections from device\n"
124 " reverse [--no-rebind] REMOTE LOCAL\n"
125 " reverse socket connection using:\n"
126 " tcp:<port> (<remote> may be \"tcp:0\" to pick any open port)\n"
127 " localabstract:<unix domain socket name>\n"
128 " localreserved:<unix domain socket name>\n"
129 " localfilesystem:<unix domain socket name>\n"
130 " reverse --remove REMOTE remove specific reverse socket connection\n"
131 " reverse --remove-all remove all reverse socket connections from device\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800132 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700133 "file transfer:\n"
134 " push LOCAL... REMOTE\n"
135 " copy local files/directories to device\n"
136 " pull [-a] REMOTE... LOCAL\n"
137 " copy files/dirs from device\n"
138 " -a: preserve file timestamp and mode\n"
139 " sync [DIR]\n"
140 " copy all changed files to device; if DIR is \"system\", \"vendor\", \"oem\",\n"
141 " or \"data\", only sync that partition (default all)\n"
142 " -l: list but don't copy\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800143 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700144 "shell:\n"
145 " shell [-e ESCAPE] [-n] [-Tt] [-x] [COMMAND...]\n"
146 " run remote shell command (interactive shell if no command given)\n"
147 " -e: choose escape character, or \"none\"; default '~'\n"
148 " -n: don't read from stdin\n"
149 " -T: disable PTY allocation\n"
150 " -t: force PTY allocation\n"
151 " -x: disable remote exit codes and stdout/stderr separation\n"
152 " emu COMMAND run emulator console command\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800153 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700154 "app installation:\n"
155 " install [-lrtsdg] PACKAGE\n"
156 " install-multiple [-lrtsdpg] PACKAGE...\n"
157 " push package(s) to the device and install them\n"
158 " -l: forward lock application\n"
159 " -r: replace existing application\n"
160 " -t: allow test packages\n"
161 " -s: install application on sdcard\n"
162 " -d: allow version code downgrade (debuggable packages only)\n"
163 " -p: partial application install (install-multiple only)\n"
164 " -g: grant all runtime permissions\n"
165 " uninstall [-k] PACKAGE\n"
166 " remove this app package from the device\n"
167 " '-k': keep the data and cache directories\n"
168 "\n"
169 "backup/restore:\n"
170 " backup [-f FILE] [-apk|-noapk] [-obb|-noobb] [-shared|-noshared] [-all] [-system|-nosystem] [PACKAGE...]\n"
171 " write an archive of the device's data to FILE [default=backup.adb]\n"
172 " package list optional if -all/-shared are supplied\n"
173 " -apk/-noapk: do/don't back up .apk files (default -noapk)\n"
174 " -obb/-noobb: do/don't back up .obb files (default -noobb)\n"
175 " -shared|-noshared: do/don't back up shared storage (default -noshared)\n"
176 " -all: back up all installed applications\n"
177 " -system|-nosystem: include system apps in -all (default -system)\n"
178 " restore FILE restore device contents from FILE\n"
179 "\n"
180 "debugging:\n"
181 " bugreport [PATH]\n"
182 " write bugreport to given PATH [default=bugreport.zip];\n"
183 " if PATH is a directory, the bug report is saved in that directory.\n"
184 " devices that don't support zipped bug reports output to stdout.\n"
185 " jdwp list pids of processes hosting a JDWP transport\n"
186 " logcat show device log (logcat --help for more)\n"
187 "\n"
188 "security:\n"
189 " disable-verity disable dm-verity checking on userdebug builds\n"
190 " enable-verity re-enable dm-verity checking on userdebug builds\n"
191 " keygen FILE\n"
192 " generate adb public/private key; private key stored in FILE,\n"
193 " public key stored in FILE.pub (existing files overwritten)\n"
194 "\n"
195 "scripting:\n"
196 " wait-for[-TRANSPORT]-STATE\n"
197 " wait for device to be in the given state\n"
198 " State: device, recovery, sideload, or bootloader\n"
199 " Transport: usb, local, or any [default=any]\n"
200 " get-state print offline | bootloader | device\n"
201 " get-serialno print <serial-number>\n"
202 " get-devpath print <device-path>\n"
203 " remount\n"
204 " remount /system, /vendor, and /oem partitions read-write\n"
205 " reboot [bootloader|recovery|sideload|sideload-auto-reboot]\n"
206 " reboot the device; defaults to booting system image but\n"
207 " supports bootloader and recovery too. sideload reboots\n"
208 " into recovery and automatically starts sideload mode,\n"
209 " sideload-auto-reboot is the same but reboots after sideloading.\n"
210 " sideload OTAPACKAGE sideload the given full OTA package\n"
211 " root restart adbd with root permissions\n"
212 " unroot restart adbd without root permissions\n"
213 " usb restart adb server listening on USB\n"
214 " tcpip PORT restart adb server listening on TCP on PORT\n"
Timcd643152010-02-16 20:18:29 +0000215 "\n"
Yabin Cui1f4ec192016-04-05 13:50:44 -0700216 "internal debugging:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700217 " start-server ensure that there is a server running\n"
218 " kill-server kill the server if it is running\n"
219 " reconnect kick connection from host side to force reconnect\n"
220 " reconnect device kick connection from device side to force reconnect\n"
221 "\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700222 "environment variables:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700223 " $ADB_TRACE\n"
224 " comma-separated list of debug info to log:\n"
225 " all,adb,sockets,packets,rwx,usb,sync,sysdeps,transport,jdwp\n"
226 " $ADB_VENDOR_KEYS colon-separated list of keys (files or directories)\n"
227 " $ANDROID_SERIAL serial number to connect to (see -s)\n"
228 " $ANDROID_LOG_TAGS tags to be used by logcat (see logcat --help)\n");
Felipe Lemeb3239722016-07-29 17:57:00 -0700229 // clang-format on
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800230}
231
Elliott Hughes58305772015-04-17 13:57:15 -0700232static int usage() {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800233 help();
234 return 1;
235}
236
Yabin Cuid325e862014-11-17 14:48:25 -0800237#if defined(_WIN32)
238
Elliott Hughesa2f2e562015-04-16 16:47:02 -0700239// Implemented in sysdeps_win32.cpp.
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800240void stdin_raw_init();
241void stdin_raw_restore();
Yabin Cuid325e862014-11-17 14:48:25 -0800242
243#else
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100244static termios g_saved_terminal_state;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800245
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800246static void stdin_raw_init() {
247 if (tcgetattr(STDIN_FILENO, &g_saved_terminal_state)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800248
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100249 termios tio;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800250 if (tcgetattr(STDIN_FILENO, &tio)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800251
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100252 cfmakeraw(&tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800253
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100254 // No timeout but request at least one character per read.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800255 tio.c_cc[VTIME] = 0;
256 tio.c_cc[VMIN] = 1;
257
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800258 tcsetattr(STDIN_FILENO, TCSAFLUSH, &tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800259}
260
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800261static void stdin_raw_restore() {
262 tcsetattr(STDIN_FILENO, TCSAFLUSH, &g_saved_terminal_state);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800263}
264#endif
265
David Pursell606835a2015-09-08 17:17:02 -0700266// Reads from |fd| and prints received data. If |use_shell_protocol| is true
267// this expects that incoming data will use the shell protocol, in which case
268// stdout/stderr are routed independently and the remote exit code will be
269// returned.
Felipe Leme44a42672016-04-01 17:43:27 -0700270// if |output| is non-null, stdout will be appended to it instead.
Felipe Lemed1dd1252016-06-06 16:05:36 -0700271// if |err| is non-null, stderr will be appended to it instead.
272static int read_and_dump(int fd, bool use_shell_protocol=false, std::string* output=nullptr,
273 std::string* err=nullptr) {
David Pursell606835a2015-09-08 17:17:02 -0700274 int exit_code = 0;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700275 if (fd < 0) return exit_code;
276
David Pursell606835a2015-09-08 17:17:02 -0700277 std::unique_ptr<ShellProtocol> protocol;
278 int length = 0;
279 FILE* outfile = stdout;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700280 std::string* outstring = output;
David Pursell606835a2015-09-08 17:17:02 -0700281
282 char raw_buffer[BUFSIZ];
283 char* buffer_ptr = raw_buffer;
284 if (use_shell_protocol) {
285 protocol.reset(new ShellProtocol(fd));
286 if (!protocol) {
287 LOG(ERROR) << "failed to allocate memory for ShellProtocol object";
288 return 1;
289 }
290 buffer_ptr = protocol->data();
291 }
292
Felipe Lemed1dd1252016-06-06 16:05:36 -0700293 while (true) {
David Pursell606835a2015-09-08 17:17:02 -0700294 if (use_shell_protocol) {
295 if (!protocol->Read()) {
296 break;
297 }
298 switch (protocol->id()) {
299 case ShellProtocol::kIdStdout:
300 outfile = stdout;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700301 outstring = output;
David Pursell606835a2015-09-08 17:17:02 -0700302 break;
303 case ShellProtocol::kIdStderr:
304 outfile = stderr;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700305 outstring = err;
David Pursell606835a2015-09-08 17:17:02 -0700306 break;
307 case ShellProtocol::kIdExit:
308 exit_code = protocol->data()[0];
309 continue;
310 default:
311 continue;
312 }
313 length = protocol->data_length();
314 } else {
315 D("read_and_dump(): pre adb_read(fd=%d)", fd);
316 length = adb_read(fd, raw_buffer, sizeof(raw_buffer));
317 D("read_and_dump(): post adb_read(fd=%d): length=%d", fd, length);
318 if (length <= 0) {
319 break;
320 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800321 }
322
Felipe Lemed1dd1252016-06-06 16:05:36 -0700323 if (outstring == nullptr) {
Felipe Leme44a42672016-04-01 17:43:27 -0700324 fwrite(buffer_ptr, 1, length, outfile);
325 fflush(outfile);
326 } else {
Felipe Lemed1dd1252016-06-06 16:05:36 -0700327 outstring->append(buffer_ptr, length);
Felipe Leme44a42672016-04-01 17:43:27 -0700328 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800329 }
David Pursell606835a2015-09-08 17:17:02 -0700330
331 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800332}
333
Jeff Sharkey960df972014-06-09 17:30:57 -0700334static void read_status_line(int fd, char* buf, size_t count)
335{
336 count--;
337 while (count > 0) {
338 int len = adb_read(fd, buf, count);
Elliott Hughes8fcd8bc2015-08-25 10:59:45 -0700339 if (len <= 0) {
Jeff Sharkey960df972014-06-09 17:30:57 -0700340 break;
341 }
342
343 buf += len;
344 count -= len;
345 }
346 *buf = '\0';
347}
348
Christopher Tated2f54152011-04-21 12:53:28 -0700349static void copy_to_file(int inFd, int outFd) {
Christopher Tate5b811fa2011-06-10 11:38:37 -0700350 const size_t BUFSIZE = 32 * 1024;
351 char* buf = (char*) malloc(BUFSIZE);
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700352 if (buf == nullptr) fatal("couldn't allocate buffer for copy_to_file");
Christopher Tated2f54152011-04-21 12:53:28 -0700353 int len;
Christopher Tatec9cd3b92011-06-01 17:56:23 -0700354 long total = 0;
Spencer Lowb7dfb792015-05-22 16:48:31 -0700355#ifdef _WIN32
356 int old_stdin_mode = -1;
357 int old_stdout_mode = -1;
358#endif
Christopher Tated2f54152011-04-21 12:53:28 -0700359
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700360 D("copy_to_file(%d -> %d)", inFd, outFd);
Yabin Cuid325e862014-11-17 14:48:25 -0800361
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700362 if (inFd == STDIN_FILENO) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800363 stdin_raw_init();
Spencer Lowb7dfb792015-05-22 16:48:31 -0700364#ifdef _WIN32
365 old_stdin_mode = _setmode(STDIN_FILENO, _O_BINARY);
366 if (old_stdin_mode == -1) {
367 fatal_errno("could not set stdin to binary");
368 }
369#endif
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700370 }
Yabin Cuid325e862014-11-17 14:48:25 -0800371
Spencer Lowb7dfb792015-05-22 16:48:31 -0700372#ifdef _WIN32
373 if (outFd == STDOUT_FILENO) {
374 old_stdout_mode = _setmode(STDOUT_FILENO, _O_BINARY);
375 if (old_stdout_mode == -1) {
376 fatal_errno("could not set stdout to binary");
377 }
378 }
379#endif
380
Elliott Hughesa7090b92015-04-17 17:03:59 -0700381 while (true) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700382 if (inFd == STDIN_FILENO) {
383 len = unix_read(inFd, buf, BUFSIZE);
384 } else {
385 len = adb_read(inFd, buf, BUFSIZE);
386 }
Christopher Tated2f54152011-04-21 12:53:28 -0700387 if (len == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700388 D("copy_to_file() : read 0 bytes; exiting");
Christopher Tated2f54152011-04-21 12:53:28 -0700389 break;
390 }
391 if (len < 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700392 D("copy_to_file(): read failed: %s", strerror(errno));
Christopher Tated2f54152011-04-21 12:53:28 -0700393 break;
394 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700395 if (outFd == STDOUT_FILENO) {
396 fwrite(buf, 1, len, stdout);
397 fflush(stdout);
398 } else {
399 adb_write(outFd, buf, len);
400 }
Christopher Tatec9cd3b92011-06-01 17:56:23 -0700401 total += len;
Christopher Tated2f54152011-04-21 12:53:28 -0700402 }
Yabin Cuid325e862014-11-17 14:48:25 -0800403
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700404 if (inFd == STDIN_FILENO) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800405 stdin_raw_restore();
Spencer Lowb7dfb792015-05-22 16:48:31 -0700406#ifdef _WIN32
407 if (_setmode(STDIN_FILENO, old_stdin_mode) == -1) {
408 fatal_errno("could not restore stdin mode");
409 }
410#endif
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700411 }
Yabin Cuid325e862014-11-17 14:48:25 -0800412
Spencer Lowb7dfb792015-05-22 16:48:31 -0700413#ifdef _WIN32
414 if (outFd == STDOUT_FILENO) {
415 if (_setmode(STDOUT_FILENO, old_stdout_mode) == -1) {
416 fatal_errno("could not restore stdout mode");
417 }
418 }
419#endif
420
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700421 D("copy_to_file() finished after %lu bytes", total);
Christopher Tate5b811fa2011-06-10 11:38:37 -0700422 free(buf);
Christopher Tated2f54152011-04-21 12:53:28 -0700423}
424
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800425static void send_window_size_change(int fd, std::unique_ptr<ShellProtocol>& shell) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800426 // Old devices can't handle window size changes.
427 if (shell == nullptr) return;
428
Spencer Low2e02dc62015-11-07 17:34:39 -0800429#if defined(_WIN32)
430 struct winsize {
431 unsigned short ws_row;
432 unsigned short ws_col;
433 unsigned short ws_xpixel;
434 unsigned short ws_ypixel;
435 };
436#endif
437
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800438 winsize ws;
Spencer Low2e02dc62015-11-07 17:34:39 -0800439
440#if defined(_WIN32)
441 // If stdout is redirected to a non-console, we won't be able to get the
442 // console size, but that makes sense.
443 const intptr_t intptr_handle = _get_osfhandle(STDOUT_FILENO);
444 if (intptr_handle == -1) return;
445
446 const HANDLE handle = reinterpret_cast<const HANDLE>(intptr_handle);
447
448 CONSOLE_SCREEN_BUFFER_INFO info;
449 memset(&info, 0, sizeof(info));
450 if (!GetConsoleScreenBufferInfo(handle, &info)) return;
451
452 memset(&ws, 0, sizeof(ws));
453 // The number of visible rows, excluding offscreen scroll-back rows which are in info.dwSize.Y.
454 ws.ws_row = info.srWindow.Bottom - info.srWindow.Top + 1;
455 // If the user has disabled "Wrap text output on resize", they can make the screen buffer wider
456 // than the window, in which case we should use the width of the buffer.
457 ws.ws_col = info.dwSize.X;
458#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800459 if (ioctl(fd, TIOCGWINSZ, &ws) == -1) return;
Spencer Low2e02dc62015-11-07 17:34:39 -0800460#endif
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800461
462 // Send the new window size as human-readable ASCII for debugging convenience.
463 size_t l = snprintf(shell->data(), shell->data_capacity(), "%dx%d,%dx%d",
464 ws.ws_row, ws.ws_col, ws.ws_xpixel, ws.ws_ypixel);
465 shell->Write(ShellProtocol::kIdWindowSizeChange, l + 1);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800466}
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800467
David Pursell606835a2015-09-08 17:17:02 -0700468// Used to pass multiple values to the stdin read thread.
469struct StdinReadArgs {
470 int stdin_fd, write_fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700471 bool raw_stdin;
David Pursell606835a2015-09-08 17:17:02 -0700472 std::unique_ptr<ShellProtocol> protocol;
Elliott Hughes28416d62015-11-04 13:36:32 -0800473 char escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700474};
475
David Pursell606835a2015-09-08 17:17:02 -0700476// Loops to read from stdin and push the data to the given FD.
477// The argument should be a pointer to a StdinReadArgs object. This function
478// will take ownership of the object and delete it when finished.
Josh Gaod9db09c2016-02-12 14:31:15 -0800479static void stdin_read_thread_loop(void* x) {
David Pursell606835a2015-09-08 17:17:02 -0700480 std::unique_ptr<StdinReadArgs> args(reinterpret_cast<StdinReadArgs*>(x));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800481
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800482#if !defined(_WIN32)
483 // Mask SIGTTIN in case we're in a backgrounded process.
Josh Gao8dcdb572015-10-23 15:03:31 -0700484 sigset_t sigset;
485 sigemptyset(&sigset);
486 sigaddset(&sigset, SIGTTIN);
487 pthread_sigmask(SIG_BLOCK, &sigset, nullptr);
488#endif
489
Spencer Low2e02dc62015-11-07 17:34:39 -0800490#if defined(_WIN32)
491 // _get_interesting_input_record_uncached() causes unix_read_interruptible()
492 // to return -1 with errno == EINTR if the window size changes.
493#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800494 // Unblock SIGWINCH for this thread, so our read(2) below will be
495 // interrupted if the window size changes.
496 sigset_t mask;
497 sigemptyset(&mask);
498 sigaddset(&mask, SIGWINCH);
499 pthread_sigmask(SIG_UNBLOCK, &mask, nullptr);
500#endif
501
502 // Set up the initial window size.
503 send_window_size_change(args->stdin_fd, args->protocol);
504
Elliott Hughes28416d62015-11-04 13:36:32 -0800505 char raw_buffer[BUFSIZ];
David Pursell606835a2015-09-08 17:17:02 -0700506 char* buffer_ptr = raw_buffer;
507 size_t buffer_size = sizeof(raw_buffer);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800508 if (args->protocol != nullptr) {
David Pursell606835a2015-09-08 17:17:02 -0700509 buffer_ptr = args->protocol->data();
510 buffer_size = args->protocol->data_capacity();
511 }
512
Elliott Hughes28416d62015-11-04 13:36:32 -0800513 // If we need to parse escape sequences, make life easy.
514 if (args->raw_stdin && args->escape_char != '\0') {
515 buffer_size = 1;
516 }
517
518 enum EscapeState { kMidFlow, kStartOfLine, kInEscape };
519 EscapeState state = kStartOfLine;
520
Elliott Hughesaa245492015-08-03 10:38:08 -0700521 while (true) {
Spencer Low2e02dc62015-11-07 17:34:39 -0800522 // Use unix_read_interruptible() rather than adb_read() for stdin.
523 D("stdin_read_thread_loop(): pre unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
524 int r = unix_read_interruptible(args->stdin_fd, buffer_ptr,
525 buffer_size);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800526 if (r == -1 && errno == EINTR) {
527 send_window_size_change(args->stdin_fd, args->protocol);
528 continue;
529 }
Spencer Low2e02dc62015-11-07 17:34:39 -0800530 D("stdin_read_thread_loop(): post unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
David Pursell1ed57f02015-10-06 15:30:03 -0700531 if (r <= 0) {
532 // Only devices using the shell protocol know to close subprocess
533 // stdin. For older devices we want to just leave the connection
534 // open, otherwise an unpredictable amount of return data could
535 // be lost due to the FD closing before all data has been received.
536 if (args->protocol) {
537 args->protocol->Write(ShellProtocol::kIdCloseStdin, 0);
538 }
539 break;
540 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800541 // If we made stdin raw, check input for escape sequences. In
David Pursell1ed57f02015-10-06 15:30:03 -0700542 // this situation signals like Ctrl+C are sent remotely rather than
543 // interpreted locally so this provides an emergency out if the remote
544 // process starts ignoring the signal. SSH also does this, see the
545 // "escape characters" section on the ssh man page for more info.
Elliott Hughes28416d62015-11-04 13:36:32 -0800546 if (args->raw_stdin && args->escape_char != '\0') {
547 char ch = buffer_ptr[0];
548 if (ch == args->escape_char) {
549 if (state == kStartOfLine) {
550 state = kInEscape;
551 // Swallow the escape character.
552 continue;
553 } else {
554 state = kMidFlow;
555 }
556 } else {
557 if (state == kInEscape) {
558 if (ch == '.') {
559 fprintf(stderr,"\r\n[ disconnected ]\r\n");
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800560 stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700561 exit(0);
Elliott Hughes28416d62015-11-04 13:36:32 -0800562 } else {
563 // We swallowed an escape character that wasn't part of
564 // a valid escape sequence; time to cough it up.
565 buffer_ptr[0] = args->escape_char;
566 buffer_ptr[1] = ch;
567 ++r;
David Pursell1ed57f02015-10-06 15:30:03 -0700568 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800569 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800570 state = (ch == '\n' || ch == '\r') ? kStartOfLine : kMidFlow;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800571 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800572 }
David Pursell606835a2015-09-08 17:17:02 -0700573 if (args->protocol) {
574 if (!args->protocol->Write(ShellProtocol::kIdStdin, r)) {
575 break;
576 }
577 } else {
578 if (!WriteFdExactly(args->write_fd, buffer_ptr, r)) {
579 break;
580 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800581 }
582 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800583}
584
David Pursell1ed57f02015-10-06 15:30:03 -0700585// Returns a shell service string with the indicated arguments and command.
586static std::string ShellServiceString(bool use_shell_protocol,
587 const std::string& type_arg,
588 const std::string& command) {
589 std::vector<std::string> args;
590 if (use_shell_protocol) {
591 args.push_back(kShellServiceArgShellProtocol);
Elliott Hughesc2252df2015-11-18 12:45:48 -0800592
593 const char* terminal_type = getenv("TERM");
594 if (terminal_type != nullptr) {
595 args.push_back(std::string("TERM=") + terminal_type);
596 }
David Pursell1ed57f02015-10-06 15:30:03 -0700597 }
598 if (!type_arg.empty()) {
599 args.push_back(type_arg);
600 }
601
602 // Shell service string can look like: shell[,arg1,arg2,...]:[command].
603 return android::base::StringPrintf("shell%s%s:%s",
604 args.empty() ? "" : ",",
605 android::base::Join(args, ',').c_str(),
606 command.c_str());
607}
608
609// Connects to a shell on the device and read/writes data.
610//
611// Note: currently this function doesn't properly clean up resources; the
612// FD connected to the adb server is never closed and the stdin read thread
613// may never exit.
614//
615// On success returns the remote exit code if |use_shell_protocol| is true,
616// 0 otherwise. On failure returns 1.
617static int RemoteShell(bool use_shell_protocol, const std::string& type_arg,
Elliott Hughes28416d62015-11-04 13:36:32 -0800618 char escape_char,
David Pursell1ed57f02015-10-06 15:30:03 -0700619 const std::string& command) {
620 std::string service_string = ShellServiceString(use_shell_protocol,
621 type_arg, command);
622
623 // Make local stdin raw if the device allocates a PTY, which happens if:
624 // 1. We are explicitly asking for a PTY shell, or
625 // 2. We don't specify shell type and are starting an interactive session.
626 bool raw_stdin = (type_arg == kShellServiceArgPty ||
627 (type_arg.empty() && command.empty()));
628
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700629 std::string error;
David Pursell4e2fd362015-09-22 10:43:08 -0700630 int fd = adb_connect(service_string, &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700631 if (fd < 0) {
632 fprintf(stderr,"error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800633 return 1;
634 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800635
David Pursell606835a2015-09-08 17:17:02 -0700636 StdinReadArgs* args = new StdinReadArgs;
637 if (!args) {
638 LOG(ERROR) << "couldn't allocate StdinReadArgs object";
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700639 return 1;
640 }
David Pursell1ed57f02015-10-06 15:30:03 -0700641 args->stdin_fd = STDIN_FILENO;
David Pursell606835a2015-09-08 17:17:02 -0700642 args->write_fd = fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700643 args->raw_stdin = raw_stdin;
Elliott Hughes28416d62015-11-04 13:36:32 -0800644 args->escape_char = escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700645 if (use_shell_protocol) {
646 args->protocol.reset(new ShellProtocol(args->write_fd));
647 }
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700648
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800649 if (raw_stdin) stdin_raw_init();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800650
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800651#if !defined(_WIN32)
652 // Ensure our process is notified if the local window size changes.
653 // We use sigaction(2) to ensure that the SA_RESTART flag is not set,
654 // because the whole reason we're sending signals is to unblock the read(2)!
655 // That also means we don't need to do anything in the signal handler:
656 // the side effect of delivering the signal is all we need.
657 struct sigaction sa;
658 memset(&sa, 0, sizeof(sa));
659 sa.sa_handler = [](int) {};
660 sa.sa_flags = 0;
661 sigaction(SIGWINCH, &sa, nullptr);
662
663 // Now block SIGWINCH in this thread (the main thread) and all threads spawned
664 // from it. The stdin read thread will unblock this signal to ensure that it's
665 // the thread that receives the signal.
666 sigset_t mask;
667 sigemptyset(&mask);
668 sigaddset(&mask, SIGWINCH);
669 pthread_sigmask(SIG_BLOCK, &mask, nullptr);
670#endif
671
672 // TODO: combine read_and_dump with stdin_read_thread to make life simpler?
673 int exit_code = 1;
674 if (!adb_thread_create(stdin_read_thread_loop, args)) {
David Pursell606835a2015-09-08 17:17:02 -0700675 PLOG(ERROR) << "error starting stdin read thread";
David Pursell606835a2015-09-08 17:17:02 -0700676 delete args;
677 } else {
678 exit_code = read_and_dump(fd, use_shell_protocol);
679 }
Elliott Hughes9b0f3542015-05-05 13:41:21 -0700680
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800681 // TODO: properly exit stdin_read_thread_loop and close |fd|.
David Pursell1ed57f02015-10-06 15:30:03 -0700682
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800683 // TODO: we should probably install signal handlers for this.
684 // TODO: can we use atexit? even on Windows?
685 if (raw_stdin) stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700686
David Pursell606835a2015-09-08 17:17:02 -0700687 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800688}
689
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800690static int adb_shell(int argc, const char** argv) {
David Pursell0aacbbe2016-01-21 19:56:50 -0800691 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800692 std::string error;
693
694 if (!adb_get_feature_set(&features, &error)) {
695 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -0800696 return 1;
697 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800698
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800699 bool use_shell_protocol = CanUseFeature(features, kFeatureShell2);
700 if (!use_shell_protocol) {
701 D("shell protocol not supported, using raw data transfer");
702 } else {
703 D("using shell protocol");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800704 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700705
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800706 // Parse shell-specific command-line options.
707 // argv[0] is always "shell".
708 --argc;
709 ++argv;
710 int t_arg_count = 0;
Elliott Hughes28416d62015-11-04 13:36:32 -0800711 char escape_char = '~';
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800712 while (argc) {
Elliott Hughes28416d62015-11-04 13:36:32 -0800713 if (!strcmp(argv[0], "-e")) {
714 if (argc < 2 || !(strlen(argv[1]) == 1 || strcmp(argv[1], "none") == 0)) {
715 fprintf(stderr, "error: -e requires a single-character argument or 'none'\n");
716 return 1;
717 }
718 escape_char = (strcmp(argv[1], "none") == 0) ? 0 : argv[1][0];
719 argc -= 2;
720 argv += 2;
721 } else if (!strcmp(argv[0], "-T") || !strcmp(argv[0], "-t")) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800722 // Like ssh, -t arguments are cumulative so that multiple -t's
723 // are needed to force a PTY.
724 if (argv[0][1] == 't') {
725 ++t_arg_count;
726 } else {
727 t_arg_count = -1;
728 }
729 --argc;
730 ++argv;
731 } else if (!strcmp(argv[0], "-x")) {
732 use_shell_protocol = false;
733 --argc;
734 ++argv;
Josh Gao7d586072015-11-20 15:37:31 -0800735 } else if (!strcmp(argv[0], "-n")) {
736 close_stdin();
737
738 --argc;
739 ++argv;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800740 } else {
741 break;
742 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700743 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800744
David Pursell57dd5ae2016-01-27 16:07:52 -0800745 // Legacy shell protocol requires a remote PTY to close the subprocess properly which creates
746 // some weird interactions with -tT.
747 if (!use_shell_protocol && t_arg_count != 0) {
748 if (!CanUseFeature(features, kFeatureShell2)) {
749 fprintf(stderr, "error: target doesn't support PTY args -Tt\n");
750 } else {
751 fprintf(stderr, "error: PTY args -Tt cannot be used with -x\n");
752 }
753 return 1;
754 }
755
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800756 std::string shell_type_arg;
757 if (CanUseFeature(features, kFeatureShell2)) {
758 if (t_arg_count < 0) {
759 shell_type_arg = kShellServiceArgRaw;
760 } else if (t_arg_count == 0) {
761 // If stdin isn't a TTY, default to a raw shell; this lets
762 // things like `adb shell < my_script.sh` work as expected.
763 // Otherwise leave |shell_type_arg| blank which uses PTY for
764 // interactive shells and raw for non-interactive.
765 if (!unix_isatty(STDIN_FILENO)) {
766 shell_type_arg = kShellServiceArgRaw;
767 }
768 } else if (t_arg_count == 1) {
769 // A single -t arg isn't enough to override implicit -T.
770 if (!unix_isatty(STDIN_FILENO)) {
771 fprintf(stderr,
772 "Remote PTY will not be allocated because stdin is not a terminal.\n"
773 "Use multiple -t options to force remote PTY allocation.\n");
774 shell_type_arg = kShellServiceArgRaw;
775 } else {
776 shell_type_arg = kShellServiceArgPty;
777 }
778 } else {
779 shell_type_arg = kShellServiceArgPty;
780 }
David Pursell71c83122015-09-14 15:33:50 -0700781 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800782
783 std::string command;
784 if (argc) {
785 // We don't escape here, just like ssh(1). http://b/20564385.
786 command = android::base::Join(std::vector<const char*>(argv, argv + argc), ' ');
787 }
788
Elliott Hughes28416d62015-11-04 13:36:32 -0800789 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, command);
David Pursell71c83122015-09-14 15:33:50 -0700790}
791
Elliott Hughes6452a892015-04-29 12:28:13 -0700792static int adb_download_buffer(const char *service, const char *fn, const void* data, unsigned sz,
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700793 bool show_progress)
Doug Zongker447f0612012-01-09 14:54:53 -0800794{
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700795 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -0700796 int fd = adb_connect(android::base::StringPrintf("%s:%d", service, sz), &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700797 if (fd < 0) {
798 fprintf(stderr,"error: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800799 return -1;
800 }
801
802 int opt = CHUNK_SIZE;
Spencer Lowf055c192015-01-25 14:40:16 -0800803 opt = adb_setsockopt(fd, SOL_SOCKET, SO_SNDBUF, (const void *) &opt, sizeof(opt));
Doug Zongker447f0612012-01-09 14:54:53 -0800804
Elliott Hughes6452a892015-04-29 12:28:13 -0700805 unsigned total = sz;
Dan Albertbac34742015-02-25 17:51:28 -0800806 const uint8_t* ptr = reinterpret_cast<const uint8_t*>(data);
Doug Zongker447f0612012-01-09 14:54:53 -0800807
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700808 if (show_progress) {
Elliott Hughes3e7048c2015-07-27 21:21:39 -0700809 const char* x = strrchr(service, ':');
810 if (x) service = x + 1;
Doug Zongker447f0612012-01-09 14:54:53 -0800811 }
812
Elliott Hughes6452a892015-04-29 12:28:13 -0700813 while (sz > 0) {
Doug Zongker447f0612012-01-09 14:54:53 -0800814 unsigned xfer = (sz > CHUNK_SIZE) ? CHUNK_SIZE : sz;
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700815 if (!WriteFdExactly(fd, ptr, xfer)) {
816 std::string error;
817 adb_status(fd, &error);
818 fprintf(stderr,"* failed to write data '%s' *\n", error.c_str());
Spencer Low351ecd12015-10-14 17:32:44 -0700819 adb_close(fd);
Doug Zongker447f0612012-01-09 14:54:53 -0800820 return -1;
821 }
822 sz -= xfer;
823 ptr += xfer;
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700824 if (show_progress) {
Magnus Eriksson86ae6d52013-03-05 07:37:32 +0100825 printf("sending: '%s' %4d%% \r", fn, (int)(100LL - ((100LL * sz) / (total))));
Doug Zongker447f0612012-01-09 14:54:53 -0800826 fflush(stdout);
827 }
828 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700829 if (show_progress) {
Doug Zongker447f0612012-01-09 14:54:53 -0800830 printf("\n");
831 }
832
Elliott Hughese67f1f82015-04-30 17:32:03 -0700833 if (!adb_status(fd, &error)) {
834 fprintf(stderr,"* error response '%s' *\n", error.c_str());
Spencer Low351ecd12015-10-14 17:32:44 -0700835 adb_close(fd);
Doug Zongker447f0612012-01-09 14:54:53 -0800836 return -1;
837 }
838
839 adb_close(fd);
840 return 0;
841}
842
Doug Zongker71fe5842014-06-26 15:35:36 -0700843#define SIDELOAD_HOST_BLOCK_SIZE (CHUNK_SIZE)
844
845/*
846 * The sideload-host protocol serves the data in a file (given on the
847 * command line) to the client, using a simple protocol:
848 *
849 * - The connect message includes the total number of bytes in the
850 * file and a block size chosen by us.
851 *
852 * - The other side sends the desired block number as eight decimal
853 * digits (eg "00000023" for block 23). Blocks are numbered from
854 * zero.
855 *
856 * - We send back the data of the requested block. The last block is
857 * likely to be partial; when the last block is requested we only
858 * send the part of the block that exists, it's not padded up to the
859 * block size.
860 *
861 * - When the other side sends "DONEDONE" instead of a block number,
862 * we hang up.
863 */
Elliott Hughes58305772015-04-17 13:57:15 -0700864static int adb_sideload_host(const char* fn) {
Elliott Hughes14415142016-06-15 14:45:34 -0700865 fprintf(stderr, "loading: '%s'...\n", fn);
Elliott Hughese8b663f2016-05-26 22:43:19 -0700866
867 std::string content;
868 if (!android::base::ReadFileToString(fn, &content)) {
Elliott Hughes14415142016-06-15 14:45:34 -0700869 fprintf(stderr, "failed: %s\n", strerror(errno));
Doug Zongker71fe5842014-06-26 15:35:36 -0700870 return -1;
871 }
872
Elliott Hughese8b663f2016-05-26 22:43:19 -0700873 const uint8_t* data = reinterpret_cast<const uint8_t*>(content.data());
874 unsigned sz = content.size();
875
Elliott Hughes14415142016-06-15 14:45:34 -0700876 fprintf(stderr, "connecting...\n");
Elliott Hughes6452a892015-04-29 12:28:13 -0700877 std::string service =
878 android::base::StringPrintf("sideload-host:%d:%d", sz, SIDELOAD_HOST_BLOCK_SIZE);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700879 std::string error;
Elliott Hughese8b663f2016-05-26 22:43:19 -0700880 unique_fd fd(adb_connect(service, &error));
Elliott Hughes356fd362016-06-13 15:01:13 -0700881 if (fd < 0) {
Doug Zongker71fe5842014-06-26 15:35:36 -0700882 // Try falling back to the older sideload method. Maybe this
883 // is an older device that doesn't support sideload-host.
Elliott Hughes14415142016-06-15 14:45:34 -0700884 fprintf(stderr, "falling back to older sideload method...\n");
Elliott Hughese8b663f2016-05-26 22:43:19 -0700885 return adb_download_buffer("sideload", fn, data, sz, true);
Doug Zongker71fe5842014-06-26 15:35:36 -0700886 }
887
Elliott Hughese8b663f2016-05-26 22:43:19 -0700888 int opt = SIDELOAD_HOST_BLOCK_SIZE;
889 adb_setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
Doug Zongker71fe5842014-06-26 15:35:36 -0700890
Elliott Hughese8b663f2016-05-26 22:43:19 -0700891 size_t xfer = 0;
892 int last_percent = -1;
Elliott Hughesa7090b92015-04-17 17:03:59 -0700893 while (true) {
Elliott Hughes6452a892015-04-29 12:28:13 -0700894 char buf[9];
Dan Albertcc731cc2015-02-24 21:26:58 -0800895 if (!ReadFdExactly(fd, buf, 8)) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700896 fprintf(stderr, "* failed to read command: %s\n", strerror(errno));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700897 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700898 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700899 buf[8] = '\0';
Doug Zongker71fe5842014-06-26 15:35:36 -0700900
Elliott Hughes6452a892015-04-29 12:28:13 -0700901 if (strcmp("DONEDONE", buf) == 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -0700902 printf("\rTotal xfer: %.2fx%*s\n",
903 (double)xfer / (sz ? sz : 1), (int)strlen(fn)+10, "");
904 return 0;
Doug Zongker71fe5842014-06-26 15:35:36 -0700905 }
906
Doug Zongker71fe5842014-06-26 15:35:36 -0700907 int block = strtol(buf, NULL, 10);
908
909 size_t offset = block * SIDELOAD_HOST_BLOCK_SIZE;
910 if (offset >= sz) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700911 fprintf(stderr, "* attempt to read block %d past end\n", block);
Elliott Hughese8b663f2016-05-26 22:43:19 -0700912 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700913 }
Elliott Hughese8b663f2016-05-26 22:43:19 -0700914 const uint8_t* start = data + offset;
Doug Zongker71fe5842014-06-26 15:35:36 -0700915 size_t offset_end = offset + SIDELOAD_HOST_BLOCK_SIZE;
916 size_t to_write = SIDELOAD_HOST_BLOCK_SIZE;
917 if (offset_end > sz) {
918 to_write = sz - offset;
919 }
920
Elliott Hughese8b663f2016-05-26 22:43:19 -0700921 if (!WriteFdExactly(fd, start, to_write)) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700922 adb_status(fd, &error);
923 fprintf(stderr,"* failed to write data '%s' *\n", error.c_str());
Elliott Hughese8b663f2016-05-26 22:43:19 -0700924 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700925 }
926 xfer += to_write;
927
928 // For normal OTA packages, we expect to transfer every byte
929 // twice, plus a bit of overhead (one read during
930 // verification, one read of each byte for installation, plus
931 // extra access to things like the zip central directory).
932 // This estimate of the completion becomes 100% when we've
933 // transferred ~2.13 (=100/47) times the package size.
934 int percent = (int)(xfer * 47LL / (sz ? sz : 1));
935 if (percent != last_percent) {
936 printf("\rserving: '%s' (~%d%%) ", fn, percent);
937 fflush(stdout);
938 last_percent = percent;
939 }
940 }
Doug Zongker71fe5842014-06-26 15:35:36 -0700941}
942
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800943/**
944 * Run ppp in "notty" mode against a resource listed as the first parameter
945 * eg:
946 *
947 * ppp dev:/dev/omap_csmi_tty0 <ppp options>
948 *
949 */
Elliott Hughes58305772015-04-17 13:57:15 -0700950static int ppp(int argc, const char** argv) {
Yabin Cuie77b6a02014-11-11 09:24:11 -0800951#if defined(_WIN32)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800952 fprintf(stderr, "error: adb %s not implemented on Win32\n", argv[0]);
953 return -1;
954#else
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800955 if (argc < 2) {
956 fprintf(stderr, "usage: adb %s <adb service name> [ppp opts]\n",
957 argv[0]);
958
959 return 1;
960 }
961
Dan Albertbac34742015-02-25 17:51:28 -0800962 const char* adb_service_name = argv[1];
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700963 std::string error;
964 int fd = adb_connect(adb_service_name, &error);
965 if (fd < 0) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800966 fprintf(stderr,"Error: Could not open adb service: %s. Error: %s\n",
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700967 adb_service_name, error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800968 return 1;
969 }
970
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700971 pid_t pid = fork();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800972
973 if (pid < 0) {
974 perror("from fork()");
975 return 1;
976 } else if (pid == 0) {
977 int err;
978 int i;
979 const char **ppp_args;
980
981 // copy args
982 ppp_args = (const char **) alloca(sizeof(char *) * argc + 1);
983 ppp_args[0] = "pppd";
984 for (i = 2 ; i < argc ; i++) {
985 //argv[2] and beyond become ppp_args[1] and beyond
986 ppp_args[i - 1] = argv[i];
987 }
988 ppp_args[i-1] = NULL;
989
990 // child side
991
992 dup2(fd, STDIN_FILENO);
993 dup2(fd, STDOUT_FILENO);
994 adb_close(STDERR_FILENO);
995 adb_close(fd);
996
997 err = execvp("pppd", (char * const *)ppp_args);
998
999 if (err < 0) {
1000 perror("execing pppd");
1001 }
1002 exit(-1);
1003 } else {
1004 // parent side
1005
1006 adb_close(fd);
1007 return 0;
1008 }
Yabin Cuie77b6a02014-11-11 09:24:11 -08001009#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001010}
1011
Josh Gao08881e72016-04-13 12:14:16 -07001012static bool wait_for_device(const char* service, TransportType t, const char* serial) {
1013 std::vector<std::string> components = android::base::Split(service, "-");
1014 if (components.size() < 3 || components.size() > 4) {
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001015 fprintf(stderr, "adb: couldn't parse 'wait-for' command: %s\n", service);
1016 return false;
1017 }
1018
Elliott Hughes2b101112015-05-04 19:29:32 -07001019 // Was the caller vague about what they'd like us to wait for?
1020 // If so, check they weren't more specific in their choice of transport type.
Josh Gao08881e72016-04-13 12:14:16 -07001021 if (components.size() == 3) {
1022 auto it = components.begin() + 2;
Elliott Hughes2b101112015-05-04 19:29:32 -07001023 if (t == kTransportUsb) {
Josh Gao08881e72016-04-13 12:14:16 -07001024 components.insert(it, "usb");
Elliott Hughes2b101112015-05-04 19:29:32 -07001025 } else if (t == kTransportLocal) {
Josh Gao08881e72016-04-13 12:14:16 -07001026 components.insert(it, "local");
Elliott Hughes2b101112015-05-04 19:29:32 -07001027 } else {
Josh Gao08881e72016-04-13 12:14:16 -07001028 components.insert(it, "any");
Elliott Hughes2b101112015-05-04 19:29:32 -07001029 }
Josh Gao08881e72016-04-13 12:14:16 -07001030 } else if (components[2] != "any" && components[2] != "local" && components[2] != "usb") {
1031 fprintf(stderr, "adb: unknown type %s; expected 'any', 'local', or 'usb'\n",
1032 components[2].c_str());
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001033 return false;
1034 }
1035
Josh Gao86441c32016-04-13 12:18:58 -07001036 if (components[3] != "any" && components[3] != "bootloader" && components[3] != "device" &&
1037 components[3] != "recovery" && components[3] != "sideload") {
Josh Gao08881e72016-04-13 12:14:16 -07001038 fprintf(stderr,
1039 "adb: unknown state %s; "
Josh Gao86441c32016-04-13 12:18:58 -07001040 "expected 'any', 'bootloader', 'device', 'recovery', or 'sideload'\n",
Josh Gao08881e72016-04-13 12:14:16 -07001041 components[3].c_str());
1042 return false;
1043 }
1044
1045 std::string cmd = format_host_command(android::base::Join(components, "-").c_str(), t, serial);
Elliott Hughes424af022015-05-29 17:55:19 -07001046 return adb_command(cmd);
Elliott Hughes2b101112015-05-04 19:29:32 -07001047}
1048
Josh Gao06c73ae2016-03-04 15:16:18 -08001049static bool adb_root(const char* command) {
1050 std::string error;
Josh Gao06c73ae2016-03-04 15:16:18 -08001051
Elliott Hughese8b663f2016-05-26 22:43:19 -07001052 unique_fd fd(adb_connect(android::base::StringPrintf("%s:", command), &error));
1053 if (fd < 0) {
Josh Gao06c73ae2016-03-04 15:16:18 -08001054 fprintf(stderr, "adb: unable to connect for %s: %s\n", command, error.c_str());
1055 return false;
1056 }
1057
1058 // Figure out whether we actually did anything.
1059 char buf[256];
1060 char* cur = buf;
1061 ssize_t bytes_left = sizeof(buf);
1062 while (bytes_left > 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -07001063 ssize_t bytes_read = adb_read(fd, cur, bytes_left);
Josh Gao06c73ae2016-03-04 15:16:18 -08001064 if (bytes_read == 0) {
1065 break;
1066 } else if (bytes_read < 0) {
1067 fprintf(stderr, "adb: error while reading for %s: %s\n", command, strerror(errno));
1068 return false;
1069 }
1070 cur += bytes_read;
1071 bytes_left -= bytes_read;
1072 }
1073
1074 if (bytes_left == 0) {
1075 fprintf(stderr, "adb: unexpected output length for %s\n", command);
1076 return false;
1077 }
1078
1079 fflush(stdout);
1080 WriteFdExactly(STDOUT_FILENO, buf, sizeof(buf) - bytes_left);
1081 if (cur != buf && strstr(buf, "restarting") == nullptr) {
1082 return true;
1083 }
1084
Josh Gao05824732016-06-16 14:00:09 -07001085 // Give adbd some time to kill itself and come back up.
1086 // We can't use wait-for-device because devices (e.g. adb over network) might not come back.
1087 adb_sleep_ms(3000);
1088 return true;
Josh Gao06c73ae2016-03-04 15:16:18 -08001089}
1090
David Pursell70ef7b42015-09-30 13:35:42 -07001091// Connects to the device "shell" service with |command| and prints the
1092// resulting output.
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001093static int send_shell_command(TransportType transport_type, const char* serial,
David Pursell70ef7b42015-09-30 13:35:42 -07001094 const std::string& command,
Felipe Leme44a42672016-04-01 17:43:27 -07001095 bool disable_shell_protocol,
Felipe Lemed1dd1252016-06-06 16:05:36 -07001096 std::string* output=nullptr,
1097 std::string* err=nullptr) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001098 int fd;
David Pursell0aacbbe2016-01-21 19:56:50 -08001099 bool use_shell_protocol = false;
1100
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001101 while (true) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001102 bool attempt_connection = true;
1103
1104 // Use shell protocol if it's supported and the caller doesn't explicitly disable it.
1105 if (!disable_shell_protocol) {
1106 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001107 std::string error;
1108 if (adb_get_feature_set(&features, &error)) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001109 use_shell_protocol = CanUseFeature(features, kFeatureShell2);
1110 } else {
1111 // Device was unreachable.
1112 attempt_connection = false;
1113 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001114 }
David Pursell0aacbbe2016-01-21 19:56:50 -08001115
1116 if (attempt_connection) {
1117 std::string error;
1118 std::string service_string = ShellServiceString(use_shell_protocol, "", command);
1119
1120 fd = adb_connect(service_string, &error);
1121 if (fd >= 0) {
1122 break;
1123 }
1124 }
1125
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001126 fprintf(stderr,"- waiting for device -\n");
Josh Gaob6117c42016-02-24 15:16:17 -08001127 if (!wait_for_device("wait-for-device", transport_type, serial)) {
1128 return 1;
1129 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001130 }
1131
Felipe Lemed1dd1252016-06-06 16:05:36 -07001132 int exit_code = read_and_dump(fd, use_shell_protocol, output, err);
David Pursell71c83122015-09-14 15:33:50 -07001133
1134 if (adb_close(fd) < 0) {
1135 PLOG(ERROR) << "failure closing FD " << fd;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001136 }
David Pursell71c83122015-09-14 15:33:50 -07001137
1138 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001139}
1140
Felipe Leme44a42672016-04-01 17:43:27 -07001141static int bugreport(TransportType transport_type, const char* serial, int argc,
1142 const char** argv) {
Felipe Lemed1dd1252016-06-06 16:05:36 -07001143 if (argc == 1) return send_shell_command(transport_type, serial, "bugreport", false);
Felipe Leme44a42672016-04-01 17:43:27 -07001144 if (argc != 2) return usage();
1145
1146 // Zipped bugreport option - will call 'bugreportz', which prints the location of the generated
1147 // file, then pull it to the destination file provided by the user.
Felipe Leme644a36c2016-04-29 13:15:23 -07001148 std::string dest_file = argv[1];
1149 if (!android::base::EndsWith(argv[1], ".zip")) {
1150 // TODO: use a case-insensitive comparison (like EndsWithIgnoreCase
1151 dest_file += ".zip";
1152 }
Felipe Leme44a42672016-04-01 17:43:27 -07001153 std::string output;
1154
Felipe Leme42e45c72016-05-04 16:12:02 -07001155 fprintf(stderr, "Bugreport is in progress and it could take minutes to complete.\n"
1156 "Please be patient and do not cancel or disconnect your device until it completes.\n");
Felipe Lemed1dd1252016-06-06 16:05:36 -07001157 int status = send_shell_command(transport_type, serial, "bugreportz", false, &output, nullptr);
Felipe Leme44a42672016-04-01 17:43:27 -07001158 if (status != 0 || output.empty()) return status;
1159 output = android::base::Trim(output);
1160
1161 if (android::base::StartsWith(output, BUGZ_OK_PREFIX)) {
1162 const char* zip_file = &output[strlen(BUGZ_OK_PREFIX)];
1163 std::vector<const char*> srcs{zip_file};
Felipe Leme644a36c2016-04-29 13:15:23 -07001164 status = do_sync_pull(srcs, dest_file.c_str(), true, dest_file.c_str()) ? 0 : 1;
Felipe Leme44a42672016-04-01 17:43:27 -07001165 if (status != 0) {
Felipe Leme644a36c2016-04-29 13:15:23 -07001166 fprintf(stderr, "Could not copy file '%s' to '%s'\n", zip_file, dest_file.c_str());
Felipe Leme44a42672016-04-01 17:43:27 -07001167 }
1168 return status;
1169 }
1170 if (android::base::StartsWith(output, BUGZ_FAIL_PREFIX)) {
1171 const char* error_message = &output[strlen(BUGZ_FAIL_PREFIX)];
Felipe Lemed1dd1252016-06-06 16:05:36 -07001172 fprintf(stderr, "Device failed to take a zipped bugreport: %s\n", error_message);
Felipe Leme44a42672016-04-01 17:43:27 -07001173 return -1;
1174 }
Felipe Lemed1dd1252016-06-06 16:05:36 -07001175 fprintf(stderr, "Unexpected string (%s) returned by bugreportz, "
Felipe Leme44a42672016-04-01 17:43:27 -07001176 "device probably does not support -z option\n", output.c_str());
1177 return -1;
1178}
1179
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001180static int logcat(TransportType transport, const char* serial, int argc, const char** argv) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001181 char* log_tags = getenv("ANDROID_LOG_TAGS");
1182 std::string quoted = escape_arg(log_tags == nullptr ? "" : log_tags);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001183
David Pursell70ef7b42015-09-30 13:35:42 -07001184 std::string cmd = "export ANDROID_LOG_TAGS=\"" + quoted + "\"; exec logcat";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001185
Jeff Sharkeyfd546e82014-06-10 11:31:24 -07001186 if (!strcmp(argv[0], "longcat")) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001187 cmd += " -v long";
Christopher Tatedb0a8802011-11-30 13:00:33 -08001188 }
1189
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001190 --argc;
1191 ++argv;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001192 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001193 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001194 }
1195
David Pursell70ef7b42015-09-30 13:35:42 -07001196 // No need for shell protocol with logcat, always disable for simplicity.
1197 return send_shell_command(transport, serial, cmd, true);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001198}
1199
Dan Albertbac34742015-02-25 17:51:28 -08001200static int backup(int argc, const char** argv) {
Elliott Hughes24f165f2015-08-21 20:31:31 -07001201 const char* filename = "backup.ab";
Christopher Tated2f54152011-04-21 12:53:28 -07001202
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001203 /* find, extract, and use any -f argument */
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001204 for (int i = 1; i < argc; i++) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001205 if (!strcmp("-f", argv[i])) {
1206 if (i == argc-1) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001207 fprintf(stderr, "adb: backup -f passed with no filename.\n");
1208 return EXIT_FAILURE;
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001209 }
1210 filename = argv[i+1];
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001211 for (int j = i+2; j <= argc; ) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001212 argv[i++] = argv[j++];
1213 }
1214 argc -= 2;
1215 argv[argc] = NULL;
1216 }
Christopher Tated2f54152011-04-21 12:53:28 -07001217 }
1218
Elliott Hughes56e68132015-11-13 11:04:10 -08001219 // Bare "adb backup" or "adb backup -f filename" are not valid invocations ---
1220 // a list of packages is required.
1221 if (argc < 2) {
1222 fprintf(stderr, "adb: backup either needs a list of packages or -all/-shared.\n");
1223 return EXIT_FAILURE;
1224 }
Christopher Tatebb86bc52011-08-22 17:12:08 -07001225
Christopher Tateb1dfffe2011-12-08 19:04:34 -08001226 adb_unlink(filename);
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001227 int outFd = adb_creat(filename, 0640);
Christopher Tated2f54152011-04-21 12:53:28 -07001228 if (outFd < 0) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001229 fprintf(stderr, "adb: backup unable to create file '%s': %s\n", filename, strerror(errno));
1230 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001231 }
1232
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001233 std::string cmd = "backup:";
1234 --argc;
1235 ++argv;
1236 while (argc-- > 0) {
1237 cmd += " " + escape_arg(*argv++);
Christopher Tated2f54152011-04-21 12:53:28 -07001238 }
1239
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001240 D("backup. filename=%s cmd=%s", filename, cmd.c_str());
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001241 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001242 int fd = adb_connect(cmd, &error);
Christopher Tated2f54152011-04-21 12:53:28 -07001243 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001244 fprintf(stderr, "adb: unable to connect for backup: %s\n", error.c_str());
Christopher Tated2f54152011-04-21 12:53:28 -07001245 adb_close(outFd);
Elliott Hughes56e68132015-11-13 11:04:10 -08001246 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001247 }
1248
Elliott Hughes56e68132015-11-13 11:04:10 -08001249 printf("Now unlock your device and confirm the backup operation...\n");
1250 fflush(stdout);
1251
Christopher Tated2f54152011-04-21 12:53:28 -07001252 copy_to_file(fd, outFd);
1253
1254 adb_close(fd);
1255 adb_close(outFd);
Elliott Hughes56e68132015-11-13 11:04:10 -08001256 return EXIT_SUCCESS;
Christopher Tated2f54152011-04-21 12:53:28 -07001257}
1258
Dan Albertbac34742015-02-25 17:51:28 -08001259static int restore(int argc, const char** argv) {
Christopher Tate702967a2011-05-17 15:52:54 -07001260 if (argc != 2) return usage();
1261
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001262 const char* filename = argv[1];
1263 int tarFd = adb_open(filename, O_RDONLY);
Christopher Tate702967a2011-05-17 15:52:54 -07001264 if (tarFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001265 fprintf(stderr, "adb: unable to open file %s: %s\n", filename, strerror(errno));
Christopher Tate702967a2011-05-17 15:52:54 -07001266 return -1;
1267 }
1268
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001269 std::string error;
1270 int fd = adb_connect("restore:", &error);
Christopher Tate702967a2011-05-17 15:52:54 -07001271 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001272 fprintf(stderr, "adb: unable to connect for restore: %s\n", error.c_str());
Christopher Tate702967a2011-05-17 15:52:54 -07001273 adb_close(tarFd);
1274 return -1;
1275 }
1276
Christopher Tatebffa4ca2012-01-06 15:43:03 -08001277 printf("Now unlock your device and confirm the restore operation.\n");
Christopher Tate702967a2011-05-17 15:52:54 -07001278 copy_to_file(tarFd, fd);
1279
Josh Gao5767d0d2016-03-04 15:51:03 -08001280 // Wait until the other side finishes, or it'll get sent SIGHUP.
1281 copy_to_file(fd, STDOUT_FILENO);
1282
Christopher Tate702967a2011-05-17 15:52:54 -07001283 adb_close(fd);
1284 adb_close(tarFd);
1285 return 0;
1286}
1287
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001288/* <hint> may be:
1289 * - A simple product name
1290 * e.g., "sooner"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001291 * - A relative path from the CWD to the ANDROID_PRODUCT_OUT dir
1292 * e.g., "out/target/product/sooner"
1293 * - An absolute path to the PRODUCT_OUT dir
1294 * e.g., "/src/device/out/target/product/sooner"
1295 *
1296 * Given <hint>, try to construct an absolute path to the
1297 * ANDROID_PRODUCT_OUT dir.
1298 */
Elliott Hughes5c742702015-07-30 17:42:01 -07001299static std::string find_product_out_path(const std::string& hint) {
1300 if (hint.empty()) {
Elliott Hughes58305772015-04-17 13:57:15 -07001301 return "";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001302 }
1303
Elliott Hughes58305772015-04-17 13:57:15 -07001304 // If it's already absolute, don't bother doing any work.
Elliott Hughes5c742702015-07-30 17:42:01 -07001305 if (adb_is_absolute_host_path(hint.c_str())) {
Elliott Hughes58305772015-04-17 13:57:15 -07001306 return hint;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001307 }
1308
Chih-Hung Hsieh8f7b9e32016-07-27 16:25:51 -07001309 // If any of the OS_PATH_SEPARATORS is found, assume it's a relative path;
Elliott Hughes58305772015-04-17 13:57:15 -07001310 // make it absolute.
Chih-Hung Hsieh8f7b9e32016-07-27 16:25:51 -07001311 // NOLINT: Do not complain if OS_PATH_SEPARATORS has only one character.
1312 if (hint.find_first_of(OS_PATH_SEPARATORS) != std::string::npos) { // NOLINT
Elliott Hughesa7090b92015-04-17 17:03:59 -07001313 std::string cwd;
1314 if (!getcwd(&cwd)) {
1315 fprintf(stderr, "adb: getcwd failed: %s\n", strerror(errno));
Elliott Hughes58305772015-04-17 13:57:15 -07001316 return "";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001317 }
Elliott Hughes5c742702015-07-30 17:42:01 -07001318 return android::base::StringPrintf("%s%c%s", cwd.c_str(), OS_PATH_SEPARATOR, hint.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001319 }
1320
Elliott Hughes58305772015-04-17 13:57:15 -07001321 // It's a string without any slashes. Try to do something with it.
1322 //
1323 // Try to find the root of the build tree, and build a PRODUCT_OUT
1324 // path from there.
Elliott Hughesa7090b92015-04-17 17:03:59 -07001325 char* top = getenv("ANDROID_BUILD_TOP");
Elliott Hughes58305772015-04-17 13:57:15 -07001326 if (top == nullptr) {
Elliott Hughesa7090b92015-04-17 17:03:59 -07001327 fprintf(stderr, "adb: ANDROID_BUILD_TOP not set!\n");
Elliott Hughes58305772015-04-17 13:57:15 -07001328 return "";
1329 }
Elliott Hughesa7090b92015-04-17 17:03:59 -07001330
1331 std::string path = top;
Elliott Hughes58305772015-04-17 13:57:15 -07001332 path += OS_PATH_SEPARATOR_STR;
1333 path += "out";
1334 path += OS_PATH_SEPARATOR_STR;
1335 path += "target";
1336 path += OS_PATH_SEPARATOR_STR;
1337 path += "product";
1338 path += OS_PATH_SEPARATOR_STR;
1339 path += hint;
1340 if (!directory_exists(path)) {
1341 fprintf(stderr, "adb: Couldn't find a product dir based on -p %s; "
Elliott Hughes5c742702015-07-30 17:42:01 -07001342 "\"%s\" doesn't exist\n", hint.c_str(), path.c_str());
Elliott Hughesa7090b92015-04-17 17:03:59 -07001343 return "";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001344 }
Elliott Hughes58305772015-04-17 13:57:15 -07001345 return path;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001346}
1347
Josh Gao05786772015-10-30 16:57:19 -07001348static void parse_push_pull_args(const char** arg, int narg,
1349 std::vector<const char*>* srcs,
1350 const char** dst, bool* copy_attrs) {
1351 *copy_attrs = false;
Mark Lindner76f2a932014-03-11 17:55:59 -07001352
Josh Gao05786772015-10-30 16:57:19 -07001353 srcs->clear();
1354 bool ignore_flags = false;
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001355 while (narg > 0) {
Josh Gao05786772015-10-30 16:57:19 -07001356 if (ignore_flags || *arg[0] != '-') {
1357 srcs->push_back(*arg);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001358 } else {
Josh Gao05786772015-10-30 16:57:19 -07001359 if (!strcmp(*arg, "-p")) {
1360 // Silently ignore for backwards compatibility.
1361 } else if (!strcmp(*arg, "-a")) {
1362 *copy_attrs = true;
1363 } else if (!strcmp(*arg, "--")) {
1364 ignore_flags = true;
1365 } else {
1366 fprintf(stderr, "adb: unrecognized option '%s'\n", *arg);
1367 exit(1);
1368 }
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001369 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001370 ++arg;
1371 --narg;
1372 }
1373
Josh Gao05786772015-10-30 16:57:19 -07001374 if (srcs->size() > 1) {
1375 *dst = srcs->back();
1376 srcs->pop_back();
Mark Lindner76f2a932014-03-11 17:55:59 -07001377 }
1378}
1379
Elliott Hughes6452a892015-04-29 12:28:13 -07001380static int adb_connect_command(const std::string& command) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001381 std::string error;
1382 int fd = adb_connect(command, &error);
Elliott Hughes5677c232015-05-07 23:37:40 -07001383 if (fd < 0) {
1384 fprintf(stderr, "error: %s\n", error.c_str());
1385 return 1;
Tao Bao175b7bb2015-03-29 11:22:34 -07001386 }
Elliott Hughes5677c232015-05-07 23:37:40 -07001387 read_and_dump(fd);
1388 adb_close(fd);
1389 return 0;
Tao Bao175b7bb2015-03-29 11:22:34 -07001390}
1391
Elliott Hughes6452a892015-04-29 12:28:13 -07001392static int adb_query_command(const std::string& command) {
1393 std::string result;
1394 std::string error;
1395 if (!adb_query(command, &result, &error)) {
1396 fprintf(stderr, "error: %s\n", error.c_str());
1397 return 1;
1398 }
1399 printf("%s\n", result.c_str());
1400 return 0;
1401}
1402
Spencer Lowa13df302015-09-07 16:20:13 -07001403// Disallow stdin, stdout, and stderr.
1404static bool _is_valid_ack_reply_fd(const int ack_reply_fd) {
1405#ifdef _WIN32
1406 const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd);
1407 return (GetStdHandle(STD_INPUT_HANDLE) != ack_reply_handle) &&
1408 (GetStdHandle(STD_OUTPUT_HANDLE) != ack_reply_handle) &&
1409 (GetStdHandle(STD_ERROR_HANDLE) != ack_reply_handle);
1410#else
1411 return ack_reply_fd > 2;
1412#endif
1413}
1414
Elliott Hughesab52c182015-05-01 17:04:38 -07001415int adb_commandline(int argc, const char **argv) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001416 int no_daemon = 0;
1417 int is_daemon = 0;
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001418 int is_server = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001419 int r;
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001420 TransportType transport_type = kTransportAny;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001421 int ack_reply_fd = -1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001422
Elliott Hughes65433da2015-11-18 18:07:48 -08001423#if !defined(_WIN32)
1424 // We'd rather have EPIPE than SIGPIPE.
1425 signal(SIGPIPE, SIG_IGN);
1426#endif
1427
Elliott Hughes58305772015-04-17 13:57:15 -07001428 // If defined, this should be an absolute path to
1429 // the directory containing all of the various system images
1430 // for a particular product. If not defined, and the adb
1431 // command requires this information, then the user must
1432 // specify the path using "-p".
1433 char* ANDROID_PRODUCT_OUT = getenv("ANDROID_PRODUCT_OUT");
1434 if (ANDROID_PRODUCT_OUT != nullptr) {
1435 gProductOutPath = ANDROID_PRODUCT_OUT;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001436 }
1437 // TODO: also try TARGET_PRODUCT/TARGET_DEVICE as a hint
1438
Josh Gao9c869b52016-08-25 16:00:22 -07001439 const char* server_host_str = nullptr;
1440 const char* server_port_str = nullptr;
1441 const char* server_socket_str = nullptr;
Stefan Hilzingera84a42e2010-04-19 12:21:12 +01001442
Elliott Hughes8d28e192015-10-07 14:55:10 -07001443 // We need to check for -d and -e before we look at $ANDROID_SERIAL.
1444 const char* serial = nullptr;
1445
Riley Andrews98f58e82014-12-05 17:37:24 -08001446 while (argc > 0) {
1447 if (!strcmp(argv[0],"server")) {
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001448 is_server = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001449 } else if (!strcmp(argv[0],"nodaemon")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001450 no_daemon = 1;
1451 } else if (!strcmp(argv[0], "fork-server")) {
1452 /* this is a special flag used only when the ADB client launches the ADB Server */
1453 is_daemon = 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001454 } else if (!strcmp(argv[0], "--reply-fd")) {
1455 if (argc < 2) return usage();
1456 const char* reply_fd_str = argv[1];
1457 argc--;
1458 argv++;
1459 ack_reply_fd = strtol(reply_fd_str, nullptr, 10);
Spencer Lowa13df302015-09-07 16:20:13 -07001460 if (!_is_valid_ack_reply_fd(ack_reply_fd)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001461 fprintf(stderr, "adb: invalid reply fd \"%s\"\n", reply_fd_str);
1462 return usage();
1463 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001464 } else if (!strncmp(argv[0], "-p", 2)) {
Elliott Hughes048b27c2015-07-31 13:18:22 -07001465 const char* product = nullptr;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001466 if (argv[0][2] == '\0') {
1467 if (argc < 2) return usage();
1468 product = argv[1];
1469 argc--;
1470 argv++;
1471 } else {
Vairavan Srinivasan81273232012-08-04 16:40:50 -07001472 product = argv[0] + 2;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001473 }
Elliott Hughes048b27c2015-07-31 13:18:22 -07001474 gProductOutPath = find_product_out_path(product);
Elliott Hughes58305772015-04-17 13:57:15 -07001475 if (gProductOutPath.empty()) {
1476 fprintf(stderr, "adb: could not resolve \"-p %s\"\n", product);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001477 return usage();
1478 }
1479 } else if (argv[0][0]=='-' && argv[0][1]=='s') {
1480 if (isdigit(argv[0][2])) {
1481 serial = argv[0] + 2;
1482 } else {
Riley Andrews98f58e82014-12-05 17:37:24 -08001483 if (argc < 2 || argv[0][2] != '\0') return usage();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001484 serial = argv[1];
1485 argc--;
1486 argv++;
1487 }
1488 } else if (!strcmp(argv[0],"-d")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001489 transport_type = kTransportUsb;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001490 } else if (!strcmp(argv[0],"-e")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001491 transport_type = kTransportLocal;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001492 } else if (!strcmp(argv[0],"-a")) {
1493 gListenAll = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001494 } else if (!strncmp(argv[0], "-H", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001495 if (argv[0][2] == '\0') {
1496 if (argc < 2) return usage();
Josh Gao9c869b52016-08-25 16:00:22 -07001497 server_host_str = argv[1];
Matt Gumbeld7b33082012-11-14 10:16:17 -08001498 argc--;
1499 argv++;
1500 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001501 server_host_str = argv[0] + 2;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001502 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001503 } else if (!strncmp(argv[0], "-P", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001504 if (argv[0][2] == '\0') {
1505 if (argc < 2) return usage();
1506 server_port_str = argv[1];
1507 argc--;
1508 argv++;
1509 } else {
1510 server_port_str = argv[0] + 2;
1511 }
Josh Gao9c869b52016-08-25 16:00:22 -07001512 } else if (!strcmp(argv[0], "-L")) {
1513 if (argc < 2) return usage();
1514 server_socket_str = argv[1];
1515 argc--;
1516 argv++;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001517 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001518 /* out of recognized modifiers and flags */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001519 break;
1520 }
1521 argc--;
1522 argv++;
1523 }
1524
Josh Gao9c869b52016-08-25 16:00:22 -07001525 if ((server_host_str || server_port_str) && server_socket_str) {
1526 fprintf(stderr, "adb: -L is incompatible with -H or -P\n");
1527 exit(1);
1528 }
1529
1530 // If -L, -H, or -P are specified, ignore environment variables.
1531 // Otherwise, prefer ADB_SERVER_SOCKET over ANDROID_ADB_SERVER_ADDRESS/PORT.
Tao Wu135f4ab2016-09-16 13:01:24 -07001532 if (!server_host_str && !server_port_str && !server_socket_str) {
1533 server_socket_str = getenv("ADB_SERVER_SOCKET");
Josh Gao9c869b52016-08-25 16:00:22 -07001534 }
1535
1536 if (!server_socket_str) {
1537 // tcp:1234 and tcp:localhost:1234 are different with -a, so don't default to localhost
1538 server_host_str = server_host_str ? server_host_str : getenv("ANDROID_ADB_SERVER_ADDRESS");
1539
Tao Wu135f4ab2016-09-16 13:01:24 -07001540 int server_port = DEFAULT_ADB_PORT;
Josh Gao9c869b52016-08-25 16:00:22 -07001541 server_port_str = server_port_str ? server_port_str : getenv("ANDROID_ADB_SERVER_PORT");
Tao Wu135f4ab2016-09-16 13:01:24 -07001542 if (server_port_str && strlen(server_port_str) > 0) {
1543 if (!android::base::ParseInt(server_port_str, &server_port, 1, 65535)) {
1544 fprintf(stderr,
1545 "adb: Env var ANDROID_ADB_SERVER_PORT must be a positive"
1546 " number less than 65535. Got \"%s\"\n",
1547 server_port_str);
1548 exit(1);
1549 }
1550 }
Josh Gao9c869b52016-08-25 16:00:22 -07001551
1552 int rc;
1553 char* temp;
1554 if (server_host_str) {
Tao Wu135f4ab2016-09-16 13:01:24 -07001555 rc = asprintf(&temp, "tcp:%s:%d", server_host_str, server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001556 } else {
Tao Wu135f4ab2016-09-16 13:01:24 -07001557 rc = asprintf(&temp, "tcp:%d", server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001558 }
1559 if (rc < 0) {
1560 fatal("failed to allocate server socket specification");
1561 }
1562 server_socket_str = temp;
1563 }
1564
1565 adb_set_socket_spec(server_socket_str);
1566
Elliott Hughes8d28e192015-10-07 14:55:10 -07001567 // If none of -d, -e, or -s were specified, try $ANDROID_SERIAL.
1568 if (transport_type == kTransportAny && serial == nullptr) {
1569 serial = getenv("ANDROID_SERIAL");
1570 }
1571
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001572 adb_set_transport(transport_type, serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001573
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001574 if (is_server) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001575 if (no_daemon || is_daemon) {
Spencer Low5c398d22015-08-08 15:07:07 -07001576 if (is_daemon && (ack_reply_fd == -1)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001577 fprintf(stderr, "reply fd for adb server to client communication not specified.\n");
1578 return usage();
1579 }
Josh Gao9c869b52016-08-25 16:00:22 -07001580 r = adb_server_main(is_daemon, server_socket_str, ack_reply_fd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001581 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001582 r = launch_server(server_socket_str);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001583 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001584 if (r) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001585 fprintf(stderr,"* could not start server *\n");
1586 }
1587 return r;
1588 }
1589
Riley Andrews98f58e82014-12-05 17:37:24 -08001590 if (argc == 0) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001591 return usage();
1592 }
1593
Riley Andrewsc8514c82014-12-05 17:32:46 -08001594 /* handle wait-for-* prefix */
1595 if (!strncmp(argv[0], "wait-for-", strlen("wait-for-"))) {
Dan Albertbac34742015-02-25 17:51:28 -08001596 const char* service = argv[0];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001597
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001598 if (!wait_for_device(service, transport_type, serial)) {
Riley Andrewsc8514c82014-12-05 17:32:46 -08001599 return 1;
1600 }
1601
Elliott Hughes2b101112015-05-04 19:29:32 -07001602 // Allow a command to be run after wait-for-device,
1603 // e.g. 'adb wait-for-device shell'.
Riley Andrewsc8514c82014-12-05 17:32:46 -08001604 if (argc == 1) {
1605 return 0;
1606 }
1607
1608 /* Fall through */
1609 argc--;
1610 argv++;
1611 }
1612
1613 /* adb_connect() commands */
Riley Andrews98f58e82014-12-05 17:37:24 -08001614 if (!strcmp(argv[0], "devices")) {
Dan Albertbac34742015-02-25 17:51:28 -08001615 const char *listopt;
Elliott Hughes6452a892015-04-29 12:28:13 -07001616 if (argc < 2) {
Scott Andersone109d262012-04-20 11:21:14 -07001617 listopt = "";
Elliott Hughes6452a892015-04-29 12:28:13 -07001618 } else if (argc == 2 && !strcmp(argv[1], "-l")) {
Scott Andersone109d262012-04-20 11:21:14 -07001619 listopt = argv[1];
Elliott Hughes6452a892015-04-29 12:28:13 -07001620 } else {
Scott Andersone109d262012-04-20 11:21:14 -07001621 fprintf(stderr, "Usage: adb devices [-l]\n");
1622 return 1;
1623 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001624
1625 std::string query = android::base::StringPrintf("host:%s%s", argv[0], listopt);
1626 printf("List of devices attached\n");
1627 return adb_query_command(query);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001628 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001629 else if (!strcmp(argv[0], "connect")) {
Mike Lockwoodff196702009-08-24 15:58:40 -07001630 if (argc != 2) {
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001631 fprintf(stderr, "Usage: adb connect <host>[:<port>]\n");
Mike Lockwoodff196702009-08-24 15:58:40 -07001632 return 1;
1633 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001634
1635 std::string query = android::base::StringPrintf("host:connect:%s", argv[1]);
1636 return adb_query_command(query);
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001637 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001638 else if (!strcmp(argv[0], "disconnect")) {
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001639 if (argc > 2) {
1640 fprintf(stderr, "Usage: adb disconnect [<host>[:<port>]]\n");
1641 return 1;
1642 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001643
1644 std::string query = android::base::StringPrintf("host:disconnect:%s",
1645 (argc == 2) ? argv[1] : "");
1646 return adb_query_command(query);
Mike Lockwoodff196702009-08-24 15:58:40 -07001647 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001648 else if (!strcmp(argv[0], "emu")) {
Spencer Low142ec752015-05-06 16:13:42 -07001649 return adb_send_emulator_command(argc, argv, serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001650 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -08001651 else if (!strcmp(argv[0], "shell")) {
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001652 return adb_shell(argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001653 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001654 else if (!strcmp(argv[0], "exec-in") || !strcmp(argv[0], "exec-out")) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001655 int exec_in = !strcmp(argv[0], "exec-in");
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001656
Zach Riggleccafc792016-04-14 17:18:04 -04001657 if (argc < 2) {
1658 fprintf(stderr, "Usage: adb %s command\n", argv[0]);
1659 return 1;
1660 }
1661
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001662 std::string cmd = "exec:";
1663 cmd += argv[1];
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001664 argc -= 2;
1665 argv += 2;
1666 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001667 cmd += " " + escape_arg(*argv++);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001668 }
1669
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001670 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001671 int fd = adb_connect(cmd, &error);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001672 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001673 fprintf(stderr, "error: %s\n", error.c_str());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001674 return -1;
1675 }
1676
1677 if (exec_in) {
1678 copy_to_file(STDIN_FILENO, fd);
1679 } else {
1680 copy_to_file(fd, STDOUT_FILENO);
1681 }
1682
1683 adb_close(fd);
1684 return 0;
1685 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001686 else if (!strcmp(argv[0], "kill-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001687 std::string error;
1688 int fd = _adb_connect("host:kill", &error);
Spencer Lowf18fc082015-08-11 17:05:02 -07001689 if (fd == -2) {
1690 // Failed to make network connection to server. Don't output the
1691 // network error since that is expected.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001692 fprintf(stderr,"* server not running *\n");
Spencer Lowf18fc082015-08-11 17:05:02 -07001693 // Successful exit code because the server is already "killed".
1694 return 0;
1695 } else if (fd == -1) {
1696 // Some other error.
1697 fprintf(stderr, "error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001698 return 1;
Spencer Lowf18fc082015-08-11 17:05:02 -07001699 } else {
1700 // Successfully connected, kill command sent, okay status came back.
1701 // Server should exit() in a moment, if not already.
Spencer Low351ecd12015-10-14 17:32:44 -07001702 ReadOrderlyShutdown(fd);
Spencer Lowf18fc082015-08-11 17:05:02 -07001703 adb_close(fd);
1704 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001705 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001706 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001707 else if (!strcmp(argv[0], "sideload")) {
1708 if (argc != 2) return usage();
Doug Zongker71fe5842014-06-26 15:35:36 -07001709 if (adb_sideload_host(argv[1])) {
Doug Zongker447f0612012-01-09 14:54:53 -08001710 return 1;
1711 } else {
1712 return 0;
1713 }
1714 }
Elliott Hughes19d80b82015-07-21 16:13:40 -07001715 else if (!strcmp(argv[0], "tcpip") && argc > 1) {
1716 return adb_connect_command(android::base::StringPrintf("tcpip:%s", argv[1]));
1717 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001718 else if (!strcmp(argv[0], "remount") ||
1719 !strcmp(argv[0], "reboot") ||
1720 !strcmp(argv[0], "reboot-bootloader") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001721 !strcmp(argv[0], "usb") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001722 !strcmp(argv[0], "disable-verity") ||
1723 !strcmp(argv[0], "enable-verity")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001724 std::string command;
Tao Bao175b7bb2015-03-29 11:22:34 -07001725 if (!strcmp(argv[0], "reboot-bootloader")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001726 command = "reboot:bootloader";
Tao Bao175b7bb2015-03-29 11:22:34 -07001727 } else if (argc > 1) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001728 command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
Tao Bao175b7bb2015-03-29 11:22:34 -07001729 } else {
Elliott Hughes5677c232015-05-07 23:37:40 -07001730 command = android::base::StringPrintf("%s:", argv[0]);
The Android Open Source Projecte037fd72009-03-13 13:04:37 -07001731 }
Tao Bao175b7bb2015-03-29 11:22:34 -07001732 return adb_connect_command(command);
Josh Gao06c73ae2016-03-04 15:16:18 -08001733 } else if (!strcmp(argv[0], "root") || !strcmp(argv[0], "unroot")) {
1734 return adb_root(argv[0]) ? 0 : 1;
1735 } else if (!strcmp(argv[0], "bugreport")) {
Felipe Leme44a42672016-04-01 17:43:27 -07001736 return bugreport(transport_type, serial, argc, argv);
1737 } else if (!strcmp(argv[0], "forward") || !strcmp(argv[0], "reverse")) {
Elliott Hughes424af022015-05-29 17:55:19 -07001738 bool reverse = !strcmp(argv[0], "reverse");
1739 ++argv;
1740 --argc;
1741 if (argc < 1) return usage();
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001742
1743 // Determine the <host-prefix> for this command.
Elliott Hughes424af022015-05-29 17:55:19 -07001744 std::string host_prefix;
David 'Digit' Turner25258692013-03-21 21:07:42 +01001745 if (reverse) {
Elliott Hughes424af022015-05-29 17:55:19 -07001746 host_prefix = "reverse";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001747 } else {
David 'Digit' Turner25258692013-03-21 21:07:42 +01001748 if (serial) {
Elliott Hughes424af022015-05-29 17:55:19 -07001749 host_prefix = android::base::StringPrintf("host-serial:%s", serial);
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001750 } else if (transport_type == kTransportUsb) {
Elliott Hughes424af022015-05-29 17:55:19 -07001751 host_prefix = "host-usb";
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001752 } else if (transport_type == kTransportLocal) {
Elliott Hughes424af022015-05-29 17:55:19 -07001753 host_prefix = "host-local";
David 'Digit' Turner25258692013-03-21 21:07:42 +01001754 } else {
Elliott Hughes424af022015-05-29 17:55:19 -07001755 host_prefix = "host";
David 'Digit' Turner25258692013-03-21 21:07:42 +01001756 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001757 }
1758
David Purselleaae97e2016-04-07 11:25:48 -07001759 std::string cmd, error;
Elliott Hughes424af022015-05-29 17:55:19 -07001760 if (strcmp(argv[0], "--list") == 0) {
Elliott Hughesab52c182015-05-01 17:04:38 -07001761 if (argc != 1) return usage();
Elliott Hughes424af022015-05-29 17:55:19 -07001762 return adb_query_command(host_prefix + ":list-forward");
1763 } else if (strcmp(argv[0], "--remove-all") == 0) {
1764 if (argc != 1) return usage();
1765 cmd = host_prefix + ":killforward-all";
1766 } else if (strcmp(argv[0], "--remove") == 0) {
1767 // forward --remove <local>
Elliott Hughesab52c182015-05-01 17:04:38 -07001768 if (argc != 2) return usage();
Elliott Hughes424af022015-05-29 17:55:19 -07001769 cmd = host_prefix + ":killforward:" + argv[1];
1770 } else if (strcmp(argv[0], "--no-rebind") == 0) {
1771 // forward --no-rebind <local> <remote>
Elliott Hughesab52c182015-05-01 17:04:38 -07001772 if (argc != 3) return usage();
David Purselleaae97e2016-04-07 11:25:48 -07001773 if (forward_targets_are_valid(argv[1], argv[2], &error)) {
1774 cmd = host_prefix + ":forward:norebind:" + argv[1] + ";" + argv[2];
1775 }
Elliott Hughes424af022015-05-29 17:55:19 -07001776 } else {
1777 // forward <local> <remote>
1778 if (argc != 2) return usage();
David Purselleaae97e2016-04-07 11:25:48 -07001779 if (forward_targets_are_valid(argv[0], argv[1], &error)) {
1780 cmd = host_prefix + ":forward:" + argv[0] + ";" + argv[1];
1781 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001782 }
1783
David Purselleaae97e2016-04-07 11:25:48 -07001784 if (!error.empty()) {
1785 fprintf(stderr, "error: %s\n", error.c_str());
1786 return 1;
1787 }
1788
1789 int fd = adb_connect(cmd, &error);
1790 if (fd < 0 || !adb_status(fd, &error)) {
1791 adb_close(fd);
1792 fprintf(stderr, "error: %s\n", error.c_str());
1793 return 1;
1794 }
1795
1796 // Server or device may optionally return a resolved TCP port number.
1797 std::string resolved_port;
1798 if (ReadProtocolString(fd, &resolved_port, &error) && !resolved_port.empty()) {
1799 printf("%s\n", resolved_port.c_str());
1800 }
1801
1802 ReadOrderlyShutdown(fd);
1803 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001804 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001805 /* do_sync_*() commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001806 else if (!strcmp(argv[0], "ls")) {
1807 if (argc != 2) return usage();
Elliott Hughesaa245492015-08-03 10:38:08 -07001808 return do_sync_ls(argv[1]) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001809 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001810 else if (!strcmp(argv[0], "push")) {
Josh Gao05786772015-10-30 16:57:19 -07001811 bool copy_attrs = false;
1812 std::vector<const char*> srcs;
1813 const char* dst = nullptr;
Mark Lindner76f2a932014-03-11 17:55:59 -07001814
Josh Gao05786772015-10-30 16:57:19 -07001815 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs);
1816 if (srcs.empty() || !dst) return usage();
1817 return do_sync_push(srcs, dst) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001818 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001819 else if (!strcmp(argv[0], "pull")) {
Josh Gao05786772015-10-30 16:57:19 -07001820 bool copy_attrs = false;
1821 std::vector<const char*> srcs;
1822 const char* dst = ".";
Mark Lindner76f2a932014-03-11 17:55:59 -07001823
Josh Gao05786772015-10-30 16:57:19 -07001824 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs);
1825 if (srcs.empty()) return usage();
1826 return do_sync_pull(srcs, dst, copy_attrs) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001827 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001828 else if (!strcmp(argv[0], "install")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001829 if (argc < 2) return usage();
David Pursell0aacbbe2016-01-21 19:56:50 -08001830 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001831 std::string error;
1832 if (!adb_get_feature_set(&features, &error)) {
1833 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08001834 return 1;
1835 }
1836
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001837 if (CanUseFeature(features, kFeatureCmd)) {
1838 return install_app(transport_type, serial, argc, argv);
1839 }
1840 return install_app_legacy(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001841 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001842 else if (!strcmp(argv[0], "install-multiple")) {
Jeff Sharkey960df972014-06-09 17:30:57 -07001843 if (argc < 2) return usage();
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001844 return install_multiple_app(transport_type, serial, argc, argv);
Jeff Sharkey960df972014-06-09 17:30:57 -07001845 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001846 else if (!strcmp(argv[0], "uninstall")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001847 if (argc < 2) return usage();
David Pursell0aacbbe2016-01-21 19:56:50 -08001848 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001849 std::string error;
1850 if (!adb_get_feature_set(&features, &error)) {
1851 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08001852 return 1;
1853 }
1854
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001855 if (CanUseFeature(features, kFeatureCmd)) {
1856 return uninstall_app(transport_type, serial, argc, argv);
1857 }
1858 return uninstall_app_legacy(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001859 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001860 else if (!strcmp(argv[0], "sync")) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001861 std::string src;
Elliott Hughes58305772015-04-17 13:57:15 -07001862 bool list_only = false;
Riley Andrews98f58e82014-12-05 17:37:24 -08001863 if (argc < 2) {
Elliott Hughes58305772015-04-17 13:57:15 -07001864 // No local path was specified.
Elliott Hughesd236d072015-04-21 10:17:07 -07001865 src = "";
Anthony Newnam705c9442010-02-22 08:36:49 -06001866 } else if (argc >= 2 && strcmp(argv[1], "-l") == 0) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001867 list_only = true;
Anthony Newnam705c9442010-02-22 08:36:49 -06001868 if (argc == 3) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001869 src = argv[2];
Anthony Newnam705c9442010-02-22 08:36:49 -06001870 } else {
Elliott Hughesd236d072015-04-21 10:17:07 -07001871 src = "";
Anthony Newnam705c9442010-02-22 08:36:49 -06001872 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001873 } else if (argc == 2) {
Elliott Hughes58305772015-04-17 13:57:15 -07001874 // A local path or "android"/"data" arg was specified.
Elliott Hughesd236d072015-04-21 10:17:07 -07001875 src = argv[1];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001876 } else {
1877 return usage();
1878 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001879
Elliott Hughesd236d072015-04-21 10:17:07 -07001880 if (src != "" &&
1881 src != "system" && src != "data" && src != "vendor" && src != "oem") {
Elliott Hughes58305772015-04-17 13:57:15 -07001882 return usage();
1883 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001884
Elliott Hughes58305772015-04-17 13:57:15 -07001885 std::string system_src_path = product_file("system");
1886 std::string data_src_path = product_file("data");
1887 std::string vendor_src_path = product_file("vendor");
1888 std::string oem_src_path = product_file("oem");
Elliott Hughes58305772015-04-17 13:57:15 -07001889
Elliott Hughesaa245492015-08-03 10:38:08 -07001890 bool okay = true;
1891 if (okay && (src.empty() || src == "system")) {
1892 okay = do_sync_sync(system_src_path, "/system", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001893 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001894 if (okay && (src.empty() || src == "vendor") && directory_exists(vendor_src_path)) {
1895 okay = do_sync_sync(vendor_src_path, "/vendor", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001896 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001897 if (okay && (src.empty() || src == "oem") && directory_exists(oem_src_path)) {
1898 okay = do_sync_sync(oem_src_path, "/oem", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001899 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001900 if (okay && (src.empty() || src == "data")) {
1901 okay = do_sync_sync(data_src_path, "/data", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001902 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001903 return okay ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001904 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001905 /* passthrough commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001906 else if (!strcmp(argv[0],"get-state") ||
Scott Andersone109d262012-04-20 11:21:14 -07001907 !strcmp(argv[0],"get-serialno") ||
1908 !strcmp(argv[0],"get-devpath"))
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001909 {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001910 return adb_query_command(format_host_command(argv[0], transport_type, serial));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001911 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001912 /* other commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001913 else if (!strcmp(argv[0],"logcat") || !strcmp(argv[0],"lolcat") || !strcmp(argv[0],"longcat")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001914 return logcat(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001915 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001916 else if (!strcmp(argv[0],"ppp")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001917 return ppp(argc, argv);
1918 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001919 else if (!strcmp(argv[0], "start-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001920 std::string error;
Spencer Lowf18fc082015-08-11 17:05:02 -07001921 const int result = adb_connect("host:start-server", &error);
1922 if (result < 0) {
1923 fprintf(stderr, "error: %s\n", error.c_str());
1924 }
1925 return result;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001926 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001927 else if (!strcmp(argv[0], "backup")) {
Christopher Tated2f54152011-04-21 12:53:28 -07001928 return backup(argc, argv);
1929 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001930 else if (!strcmp(argv[0], "restore")) {
Christopher Tate702967a2011-05-17 15:52:54 -07001931 return restore(argc, argv);
1932 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001933 else if (!strcmp(argv[0], "keygen")) {
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08001934 if (argc < 2) return usage();
Yabin Cuiaed3c612015-09-22 15:52:57 -07001935 // Always print key generation information for keygen command.
1936 adb_trace_enable(AUTH);
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08001937 return adb_auth_keygen(argv[1]);
1938 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001939 else if (!strcmp(argv[0], "jdwp")) {
Tao Bao175b7bb2015-03-29 11:22:34 -07001940 return adb_connect_command("jdwp");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001941 }
Josh Gaob463baf2016-05-13 18:18:23 -07001942 else if (!strcmp(argv[0], "track-jdwp")) {
1943 return adb_connect_command("track-jdwp");
1944 }
1945 else if (!strcmp(argv[0], "track-devices")) {
1946 return adb_connect_command("host:track-devices");
1947 }
1948
1949
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001950 /* "adb /?" is a common idiom under Windows */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001951 else if (!strcmp(argv[0], "help") || !strcmp(argv[0], "/?")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001952 help();
1953 return 0;
1954 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001955 else if (!strcmp(argv[0], "version")) {
Elliott Hughes42ae2602015-08-12 08:32:10 -07001956 fprintf(stdout, "%s", adb_version().c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001957 return 0;
1958 }
Dan Albert90d4b732015-05-20 18:58:41 -07001959 else if (!strcmp(argv[0], "features")) {
David Pursell4e2fd362015-09-22 10:43:08 -07001960 // Only list the features common to both the adb client and the device.
David Pursell0aacbbe2016-01-21 19:56:50 -08001961 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001962 std::string error;
1963 if (!adb_get_feature_set(&features, &error)) {
1964 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08001965 return 1;
1966 }
1967
David Pursell4e2fd362015-09-22 10:43:08 -07001968 for (const std::string& name : features) {
David Pursell70ef7b42015-09-30 13:35:42 -07001969 if (CanUseFeature(features, name)) {
David Pursell4e2fd362015-09-22 10:43:08 -07001970 printf("%s\n", name.c_str());
1971 }
1972 }
1973 return 0;
Yabin Cui1f4ec192016-04-05 13:50:44 -07001974 } else if (!strcmp(argv[0], "reconnect")) {
1975 if (argc == 1) {
1976 return adb_query_command("host:reconnect");
1977 } else if (argc == 2 && !strcmp(argv[1], "device")) {
1978 std::string err;
1979 adb_connect("reconnect", &err);
1980 return 0;
1981 }
Dan Albert90d4b732015-05-20 18:58:41 -07001982 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001983
1984 usage();
1985 return 1;
1986}
1987
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001988static int uninstall_app(TransportType transport, const char* serial, int argc, const char** argv) {
1989 // 'adb uninstall' takes the same arguments as 'cmd package uninstall' on device
1990 std::string cmd = "cmd package";
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001991 while (argc-- > 0) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001992 // deny the '-k' option until the remaining data/cache can be removed with adb/UI
1993 if (strcmp(*argv, "-k") == 0) {
1994 printf(
1995 "The -k option uninstalls the application while retaining the data/cache.\n"
1996 "At the moment, there is no way to remove the remaining data.\n"
1997 "You will have to reinstall the application with the same signature, and fully uninstall it.\n"
1998 "If you truly wish to continue, execute 'adb shell cmd package uninstall -k'.\n");
1999 return EXIT_FAILURE;
2000 }
Elliott Hughes6c34bba2015-04-17 20:11:08 -07002001 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002002 }
2003
David Pursell70ef7b42015-09-30 13:35:42 -07002004 return send_shell_command(transport, serial, cmd, false);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002005}
2006
Elliott Hughes3bd73c12015-05-05 13:10:43 -07002007static int install_app(TransportType transport, const char* serial, int argc, const char** argv) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002008 // The last argument must be the APK file
2009 const char* file = argv[argc - 1];
2010 const char* dot = strrchr(file, '.');
2011 bool found_apk = false;
Jeff Sharkey960df972014-06-09 17:30:57 -07002012 struct stat sb;
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002013 if (dot && !strcasecmp(dot, ".apk")) {
2014 if (stat(file, &sb) == -1 || !S_ISREG(sb.st_mode)) {
2015 fprintf(stderr, "Invalid APK file: %s\n", file);
2016 return EXIT_FAILURE;
Kenny Root597ea5b2011-08-05 11:19:45 -07002017 }
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002018 found_apk = true;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002019 }
2020
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002021 if (!found_apk) {
Jeff Sharkey960df972014-06-09 17:30:57 -07002022 fprintf(stderr, "Missing APK file\n");
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002023 return EXIT_FAILURE;
Kenny Root597ea5b2011-08-05 11:19:45 -07002024 }
2025
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002026 int localFd = adb_open(file, O_RDONLY);
2027 if (localFd < 0) {
2028 fprintf(stderr, "Failed to open %s: %s\n", file, strerror(errno));
2029 return 1;
2030 }
Kenny Root597ea5b2011-08-05 11:19:45 -07002031
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002032 std::string error;
2033 std::string cmd = "exec:cmd package";
2034
2035 // don't copy the APK name, but, copy the rest of the arguments as-is
2036 while (argc-- > 1) {
2037 cmd += " " + escape_arg(std::string(*argv++));
2038 }
2039
2040 // add size parameter [required for streaming installs]
2041 // do last to override any user specified value
2042 cmd += " " + android::base::StringPrintf("-S %" PRIu64, static_cast<uint64_t>(sb.st_size));
2043
2044 int remoteFd = adb_connect(cmd, &error);
2045 if (remoteFd < 0) {
2046 fprintf(stderr, "Connect error for write: %s\n", error.c_str());
2047 adb_close(localFd);
2048 return 1;
2049 }
2050
2051 char buf[BUFSIZ];
2052 copy_to_file(localFd, remoteFd);
2053 read_status_line(remoteFd, buf, sizeof(buf));
2054
2055 adb_close(localFd);
2056 adb_close(remoteFd);
2057
2058 if (strncmp("Success", buf, 7)) {
Elliott Hughesfe018852015-12-17 14:04:38 -08002059 fprintf(stderr, "Failed to install %s: %s", file, buf);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002060 return 1;
2061 }
2062 fputs(buf, stderr);
2063 return 0;
Jeff Sharkey960df972014-06-09 17:30:57 -07002064}
2065
Elliott Hughes3bd73c12015-05-05 13:10:43 -07002066static int install_multiple_app(TransportType transport, const char* serial, int argc,
Elliott Hughes58305772015-04-17 13:57:15 -07002067 const char** argv)
Jeff Sharkey960df972014-06-09 17:30:57 -07002068{
Jeff Sharkey960df972014-06-09 17:30:57 -07002069 int i;
2070 struct stat sb;
Elliott Hughes2940ccf2015-04-17 14:07:52 -07002071 uint64_t total_size = 0;
Jeff Sharkey960df972014-06-09 17:30:57 -07002072
2073 // Find all APK arguments starting at end.
2074 // All other arguments passed through verbatim.
2075 int first_apk = -1;
2076 for (i = argc - 1; i >= 0; i--) {
Dan Albertbac34742015-02-25 17:51:28 -08002077 const char* file = argv[i];
Elliott Hughes3e7048c2015-07-27 21:21:39 -07002078 const char* dot = strrchr(file, '.');
Jeff Sharkey960df972014-06-09 17:30:57 -07002079 if (dot && !strcasecmp(dot, ".apk")) {
2080 if (stat(file, &sb) == -1 || !S_ISREG(sb.st_mode)) {
2081 fprintf(stderr, "Invalid APK file: %s\n", file);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002082 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002083 }
2084
2085 total_size += sb.st_size;
2086 first_apk = i;
2087 } else {
2088 break;
2089 }
Kenny Root597ea5b2011-08-05 11:19:45 -07002090 }
2091
Jeff Sharkey960df972014-06-09 17:30:57 -07002092 if (first_apk == -1) {
2093 fprintf(stderr, "Missing APK file\n");
2094 return 1;
2095 }
Kenny Root597ea5b2011-08-05 11:19:45 -07002096
Elliott Hughes2940ccf2015-04-17 14:07:52 -07002097 std::string cmd = android::base::StringPrintf("exec:pm install-create -S %" PRIu64, total_size);
Jeff Sharkey960df972014-06-09 17:30:57 -07002098 for (i = 1; i < first_apk; i++) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07002099 cmd += " " + escape_arg(argv[i]);
Jeff Sharkey960df972014-06-09 17:30:57 -07002100 }
2101
2102 // Create install session
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002103 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07002104 int fd = adb_connect(cmd, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07002105 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002106 fprintf(stderr, "Connect error for create: %s\n", error.c_str());
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002107 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002108 }
Elliott Hughes2baae3a2015-04-17 10:59:34 -07002109 char buf[BUFSIZ];
Jeff Sharkey960df972014-06-09 17:30:57 -07002110 read_status_line(fd, buf, sizeof(buf));
2111 adb_close(fd);
2112
2113 int session_id = -1;
2114 if (!strncmp("Success", buf, 7)) {
2115 char* start = strrchr(buf, '[');
2116 char* end = strrchr(buf, ']');
2117 if (start && end) {
2118 *end = '\0';
2119 session_id = strtol(start + 1, NULL, 10);
2120 }
2121 }
2122 if (session_id < 0) {
2123 fprintf(stderr, "Failed to create session\n");
Christopher Tate71bbc672014-07-14 16:45:13 -07002124 fputs(buf, stderr);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002125 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002126 }
2127
2128 // Valid session, now stream the APKs
2129 int success = 1;
2130 for (i = first_apk; i < argc; i++) {
Dan Albertbac34742015-02-25 17:51:28 -08002131 const char* file = argv[i];
Jeff Sharkey960df972014-06-09 17:30:57 -07002132 if (stat(file, &sb) == -1) {
2133 fprintf(stderr, "Failed to stat %s\n", file);
2134 success = 0;
2135 goto finalize_session;
2136 }
2137
Elliott Hughes2940ccf2015-04-17 14:07:52 -07002138 std::string cmd = android::base::StringPrintf(
2139 "exec:pm install-write -S %" PRIu64 " %d %d_%s -",
Elliott Hughes5c742702015-07-30 17:42:01 -07002140 static_cast<uint64_t>(sb.st_size), session_id, i, adb_basename(file).c_str());
Jeff Sharkey960df972014-06-09 17:30:57 -07002141
2142 int localFd = adb_open(file, O_RDONLY);
2143 if (localFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002144 fprintf(stderr, "Failed to open %s: %s\n", file, strerror(errno));
Jeff Sharkey960df972014-06-09 17:30:57 -07002145 success = 0;
2146 goto finalize_session;
2147 }
2148
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002149 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07002150 int remoteFd = adb_connect(cmd, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07002151 if (remoteFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002152 fprintf(stderr, "Connect error for write: %s\n", error.c_str());
Jeff Sharkey960df972014-06-09 17:30:57 -07002153 adb_close(localFd);
2154 success = 0;
2155 goto finalize_session;
2156 }
2157
2158 copy_to_file(localFd, remoteFd);
2159 read_status_line(remoteFd, buf, sizeof(buf));
2160
2161 adb_close(localFd);
2162 adb_close(remoteFd);
2163
2164 if (strncmp("Success", buf, 7)) {
2165 fprintf(stderr, "Failed to write %s\n", file);
Christopher Tate71bbc672014-07-14 16:45:13 -07002166 fputs(buf, stderr);
Jeff Sharkey960df972014-06-09 17:30:57 -07002167 success = 0;
2168 goto finalize_session;
2169 }
2170 }
2171
2172finalize_session:
Jeff Sharkeyac77e1f2014-07-25 09:58:25 -07002173 // Commit session if we streamed everything okay; otherwise abandon
Elliott Hughes6452a892015-04-29 12:28:13 -07002174 std::string service =
2175 android::base::StringPrintf("exec:pm install-%s %d",
2176 success ? "commit" : "abandon", session_id);
2177 fd = adb_connect(service, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07002178 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002179 fprintf(stderr, "Connect error for finalize: %s\n", error.c_str());
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002180 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002181 }
2182 read_status_line(fd, buf, sizeof(buf));
2183 adb_close(fd);
2184
2185 if (!strncmp("Success", buf, 7)) {
Christopher Tate71bbc672014-07-14 16:45:13 -07002186 fputs(buf, stderr);
Jeff Sharkey960df972014-06-09 17:30:57 -07002187 return 0;
2188 } else {
2189 fprintf(stderr, "Failed to finalize session\n");
Christopher Tate71bbc672014-07-14 16:45:13 -07002190 fputs(buf, stderr);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002191 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002192 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002193}
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002194
2195static int pm_command(TransportType transport, const char* serial, int argc, const char** argv) {
2196 std::string cmd = "pm";
2197
2198 while (argc-- > 0) {
2199 cmd += " " + escape_arg(*argv++);
2200 }
2201
2202 return send_shell_command(transport, serial, cmd, false);
2203}
2204
2205static int uninstall_app_legacy(TransportType transport, const char* serial, int argc, const char** argv) {
2206 /* if the user choose the -k option, we refuse to do it until devices are
2207 out with the option to uninstall the remaining data somehow (adb/ui) */
2208 int i;
2209 for (i = 1; i < argc; i++) {
2210 if (!strcmp(argv[i], "-k")) {
2211 printf(
2212 "The -k option uninstalls the application while retaining the data/cache.\n"
2213 "At the moment, there is no way to remove the remaining data.\n"
2214 "You will have to reinstall the application with the same signature, and fully uninstall it.\n"
2215 "If you truly wish to continue, execute 'adb shell pm uninstall -k'\n.");
2216 return EXIT_FAILURE;
2217 }
2218 }
2219
2220 /* 'adb uninstall' takes the same arguments as 'pm uninstall' on device */
2221 return pm_command(transport, serial, argc, argv);
2222}
2223
2224static int delete_file(TransportType transport, const char* serial, const std::string& filename) {
2225 std::string cmd = "rm -f " + escape_arg(filename);
2226 return send_shell_command(transport, serial, cmd, false);
2227}
2228
2229static int install_app_legacy(TransportType transport, const char* serial, int argc, const char** argv) {
2230 static const char *const DATA_DEST = "/data/local/tmp/%s";
2231 static const char *const SD_DEST = "/sdcard/tmp/%s";
2232 const char* where = DATA_DEST;
2233 int i;
2234 struct stat sb;
2235
2236 for (i = 1; i < argc; i++) {
2237 if (!strcmp(argv[i], "-s")) {
2238 where = SD_DEST;
2239 }
2240 }
2241
2242 // Find last APK argument.
2243 // All other arguments passed through verbatim.
2244 int last_apk = -1;
2245 for (i = argc - 1; i >= 0; i--) {
2246 const char* file = argv[i];
2247 const char* dot = strrchr(file, '.');
2248 if (dot && !strcasecmp(dot, ".apk")) {
2249 if (stat(file, &sb) == -1 || !S_ISREG(sb.st_mode)) {
2250 fprintf(stderr, "Invalid APK file: %s\n", file);
2251 return EXIT_FAILURE;
2252 }
2253
2254 last_apk = i;
2255 break;
2256 }
2257 }
2258
2259 if (last_apk == -1) {
2260 fprintf(stderr, "Missing APK file\n");
2261 return EXIT_FAILURE;
2262 }
2263
2264 int result = -1;
2265 std::vector<const char*> apk_file = {argv[last_apk]};
2266 std::string apk_dest = android::base::StringPrintf(
2267 where, adb_basename(argv[last_apk]).c_str());
2268 if (!do_sync_push(apk_file, apk_dest.c_str())) goto cleanup_apk;
2269 argv[last_apk] = apk_dest.c_str(); /* destination name, not source location */
2270 result = pm_command(transport, serial, argc, argv);
2271
2272cleanup_apk:
2273 delete_file(transport, serial, apk_dest);
2274 return result;
2275}