blob: 54e254a1c2cc6b669f16347755396577eb873271 [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"
Felipe Leme698e0652016-07-19 17:07:22 -070057#include "bugreport.h"
58#include "commandline.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080059#include "file_sync_service.h"
David Pursell70ef7b42015-09-30 13:35:42 -070060#include "services.h"
David Pursell606835a2015-09-08 17:17:02 -070061#include "shell_service.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080062
Elliott Hughes3bd73c12015-05-05 13:10:43 -070063static int install_app(TransportType t, const char* serial, int argc, const char** argv);
64static int install_multiple_app(TransportType t, const char* serial, int argc, const char** argv);
65static int uninstall_app(TransportType t, const char* serial, int argc, const char** argv);
Todd Kennedy6fa848a2015-11-03 16:53:08 -080066static int install_app_legacy(TransportType t, const char* serial, int argc, const char** argv);
67static int uninstall_app_legacy(TransportType t, const char* serial, int argc, const char** argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080068
Josh Gaob7b1edf2015-11-11 17:56:12 -080069static auto& gProductOutPath = *new std::string();
Matt Gumbeld7b33082012-11-14 10:16:17 -080070extern int gListenAll;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080071
Felipe Leme0d4f0502016-07-26 12:14:39 -070072DefaultStandardStreamsCallback DEFAULT_STANDARD_STREAMS_CALLBACK(nullptr, nullptr);
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
Felipe Leme698e0652016-07-19 17:07:22 -0700232int 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 Leme0d4f0502016-07-26 12:14:39 -0700270// if |callback| is non-null, stdout/stderr output will be handled by it.
271int read_and_dump(int fd, bool use_shell_protocol = false,
272 StandardStreamsCallbackInterface* callback = &DEFAULT_STANDARD_STREAMS_CALLBACK) {
David Pursell606835a2015-09-08 17:17:02 -0700273 int exit_code = 0;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700274 if (fd < 0) return exit_code;
275
David Pursell606835a2015-09-08 17:17:02 -0700276 std::unique_ptr<ShellProtocol> protocol;
277 int length = 0;
David Pursell606835a2015-09-08 17:17:02 -0700278
279 char raw_buffer[BUFSIZ];
280 char* buffer_ptr = raw_buffer;
281 if (use_shell_protocol) {
282 protocol.reset(new ShellProtocol(fd));
283 if (!protocol) {
284 LOG(ERROR) << "failed to allocate memory for ShellProtocol object";
285 return 1;
286 }
287 buffer_ptr = protocol->data();
288 }
289
Felipe Lemed1dd1252016-06-06 16:05:36 -0700290 while (true) {
David Pursell606835a2015-09-08 17:17:02 -0700291 if (use_shell_protocol) {
292 if (!protocol->Read()) {
293 break;
294 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700295 length = protocol->data_length();
David Pursell606835a2015-09-08 17:17:02 -0700296 switch (protocol->id()) {
297 case ShellProtocol::kIdStdout:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700298 callback->OnStdout(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700299 break;
300 case ShellProtocol::kIdStderr:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700301 callback->OnStderr(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700302 break;
303 case ShellProtocol::kIdExit:
304 exit_code = protocol->data()[0];
305 continue;
306 default:
307 continue;
308 }
309 length = protocol->data_length();
310 } else {
311 D("read_and_dump(): pre adb_read(fd=%d)", fd);
312 length = adb_read(fd, raw_buffer, sizeof(raw_buffer));
313 D("read_and_dump(): post adb_read(fd=%d): length=%d", fd, length);
314 if (length <= 0) {
315 break;
316 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700317 callback->OnStdout(buffer_ptr, length);
Felipe Leme44a42672016-04-01 17:43:27 -0700318 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800319 }
David Pursell606835a2015-09-08 17:17:02 -0700320
Felipe Leme0d4f0502016-07-26 12:14:39 -0700321 return callback->Done(exit_code);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800322}
323
Jeff Sharkey960df972014-06-09 17:30:57 -0700324static void read_status_line(int fd, char* buf, size_t count)
325{
326 count--;
327 while (count > 0) {
328 int len = adb_read(fd, buf, count);
Elliott Hughes8fcd8bc2015-08-25 10:59:45 -0700329 if (len <= 0) {
Jeff Sharkey960df972014-06-09 17:30:57 -0700330 break;
331 }
332
333 buf += len;
334 count -= len;
335 }
336 *buf = '\0';
337}
338
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700339static void stdinout_raw_prologue(int inFd, int outFd, int& old_stdin_mode, int& old_stdout_mode) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700340 if (inFd == STDIN_FILENO) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800341 stdin_raw_init();
Spencer Lowb7dfb792015-05-22 16:48:31 -0700342#ifdef _WIN32
343 old_stdin_mode = _setmode(STDIN_FILENO, _O_BINARY);
344 if (old_stdin_mode == -1) {
345 fatal_errno("could not set stdin to binary");
346 }
347#endif
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700348 }
Yabin Cuid325e862014-11-17 14:48:25 -0800349
Spencer Lowb7dfb792015-05-22 16:48:31 -0700350#ifdef _WIN32
351 if (outFd == STDOUT_FILENO) {
352 old_stdout_mode = _setmode(STDOUT_FILENO, _O_BINARY);
353 if (old_stdout_mode == -1) {
354 fatal_errno("could not set stdout to binary");
355 }
356 }
357#endif
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700358}
359
360static void stdinout_raw_epilogue(int inFd, int outFd, int old_stdin_mode, int old_stdout_mode) {
361 if (inFd == STDIN_FILENO) {
362 stdin_raw_restore();
363#ifdef _WIN32
364 if (_setmode(STDIN_FILENO, old_stdin_mode) == -1) {
365 fatal_errno("could not restore stdin mode");
366 }
367#endif
368 }
369
370#ifdef _WIN32
371 if (outFd == STDOUT_FILENO) {
372 if (_setmode(STDOUT_FILENO, old_stdout_mode) == -1) {
373 fatal_errno("could not restore stdout mode");
374 }
375 }
376#endif
377}
378
379static void copy_to_file(int inFd, int outFd) {
380 const size_t BUFSIZE = 32 * 1024;
381 char* buf = (char*) malloc(BUFSIZE);
382 if (buf == nullptr) fatal("couldn't allocate buffer for copy_to_file");
383 int len;
384 long total = 0;
385 int old_stdin_mode = -1;
386 int old_stdout_mode = -1;
387
388 D("copy_to_file(%d -> %d)", inFd, outFd);
389
390 stdinout_raw_prologue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700391
Elliott Hughesa7090b92015-04-17 17:03:59 -0700392 while (true) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700393 if (inFd == STDIN_FILENO) {
394 len = unix_read(inFd, buf, BUFSIZE);
395 } else {
396 len = adb_read(inFd, buf, BUFSIZE);
397 }
Christopher Tated2f54152011-04-21 12:53:28 -0700398 if (len == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700399 D("copy_to_file() : read 0 bytes; exiting");
Christopher Tated2f54152011-04-21 12:53:28 -0700400 break;
401 }
402 if (len < 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700403 D("copy_to_file(): read failed: %s", strerror(errno));
Christopher Tated2f54152011-04-21 12:53:28 -0700404 break;
405 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700406 if (outFd == STDOUT_FILENO) {
407 fwrite(buf, 1, len, stdout);
408 fflush(stdout);
409 } else {
410 adb_write(outFd, buf, len);
411 }
Christopher Tatec9cd3b92011-06-01 17:56:23 -0700412 total += len;
Christopher Tated2f54152011-04-21 12:53:28 -0700413 }
Yabin Cuid325e862014-11-17 14:48:25 -0800414
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700415 stdinout_raw_epilogue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700416
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700417 D("copy_to_file() finished after %lu bytes", total);
Christopher Tate5b811fa2011-06-10 11:38:37 -0700418 free(buf);
Christopher Tated2f54152011-04-21 12:53:28 -0700419}
420
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800421static void send_window_size_change(int fd, std::unique_ptr<ShellProtocol>& shell) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800422 // Old devices can't handle window size changes.
423 if (shell == nullptr) return;
424
Spencer Low2e02dc62015-11-07 17:34:39 -0800425#if defined(_WIN32)
426 struct winsize {
427 unsigned short ws_row;
428 unsigned short ws_col;
429 unsigned short ws_xpixel;
430 unsigned short ws_ypixel;
431 };
432#endif
433
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800434 winsize ws;
Spencer Low2e02dc62015-11-07 17:34:39 -0800435
436#if defined(_WIN32)
437 // If stdout is redirected to a non-console, we won't be able to get the
438 // console size, but that makes sense.
439 const intptr_t intptr_handle = _get_osfhandle(STDOUT_FILENO);
440 if (intptr_handle == -1) return;
441
442 const HANDLE handle = reinterpret_cast<const HANDLE>(intptr_handle);
443
444 CONSOLE_SCREEN_BUFFER_INFO info;
445 memset(&info, 0, sizeof(info));
446 if (!GetConsoleScreenBufferInfo(handle, &info)) return;
447
448 memset(&ws, 0, sizeof(ws));
449 // The number of visible rows, excluding offscreen scroll-back rows which are in info.dwSize.Y.
450 ws.ws_row = info.srWindow.Bottom - info.srWindow.Top + 1;
451 // If the user has disabled "Wrap text output on resize", they can make the screen buffer wider
452 // than the window, in which case we should use the width of the buffer.
453 ws.ws_col = info.dwSize.X;
454#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800455 if (ioctl(fd, TIOCGWINSZ, &ws) == -1) return;
Spencer Low2e02dc62015-11-07 17:34:39 -0800456#endif
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800457
458 // Send the new window size as human-readable ASCII for debugging convenience.
459 size_t l = snprintf(shell->data(), shell->data_capacity(), "%dx%d,%dx%d",
460 ws.ws_row, ws.ws_col, ws.ws_xpixel, ws.ws_ypixel);
461 shell->Write(ShellProtocol::kIdWindowSizeChange, l + 1);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800462}
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800463
David Pursell606835a2015-09-08 17:17:02 -0700464// Used to pass multiple values to the stdin read thread.
465struct StdinReadArgs {
466 int stdin_fd, write_fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700467 bool raw_stdin;
David Pursell606835a2015-09-08 17:17:02 -0700468 std::unique_ptr<ShellProtocol> protocol;
Elliott Hughes28416d62015-11-04 13:36:32 -0800469 char escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700470};
471
David Pursell606835a2015-09-08 17:17:02 -0700472// Loops to read from stdin and push the data to the given FD.
473// The argument should be a pointer to a StdinReadArgs object. This function
474// will take ownership of the object and delete it when finished.
Josh Gaod9db09c2016-02-12 14:31:15 -0800475static void stdin_read_thread_loop(void* x) {
David Pursell606835a2015-09-08 17:17:02 -0700476 std::unique_ptr<StdinReadArgs> args(reinterpret_cast<StdinReadArgs*>(x));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800477
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800478#if !defined(_WIN32)
479 // Mask SIGTTIN in case we're in a backgrounded process.
Josh Gao8dcdb572015-10-23 15:03:31 -0700480 sigset_t sigset;
481 sigemptyset(&sigset);
482 sigaddset(&sigset, SIGTTIN);
483 pthread_sigmask(SIG_BLOCK, &sigset, nullptr);
484#endif
485
Spencer Low2e02dc62015-11-07 17:34:39 -0800486#if defined(_WIN32)
487 // _get_interesting_input_record_uncached() causes unix_read_interruptible()
488 // to return -1 with errno == EINTR if the window size changes.
489#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800490 // Unblock SIGWINCH for this thread, so our read(2) below will be
491 // interrupted if the window size changes.
492 sigset_t mask;
493 sigemptyset(&mask);
494 sigaddset(&mask, SIGWINCH);
495 pthread_sigmask(SIG_UNBLOCK, &mask, nullptr);
496#endif
497
498 // Set up the initial window size.
499 send_window_size_change(args->stdin_fd, args->protocol);
500
Elliott Hughes28416d62015-11-04 13:36:32 -0800501 char raw_buffer[BUFSIZ];
David Pursell606835a2015-09-08 17:17:02 -0700502 char* buffer_ptr = raw_buffer;
503 size_t buffer_size = sizeof(raw_buffer);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800504 if (args->protocol != nullptr) {
David Pursell606835a2015-09-08 17:17:02 -0700505 buffer_ptr = args->protocol->data();
506 buffer_size = args->protocol->data_capacity();
507 }
508
Elliott Hughes28416d62015-11-04 13:36:32 -0800509 // If we need to parse escape sequences, make life easy.
510 if (args->raw_stdin && args->escape_char != '\0') {
511 buffer_size = 1;
512 }
513
514 enum EscapeState { kMidFlow, kStartOfLine, kInEscape };
515 EscapeState state = kStartOfLine;
516
Elliott Hughesaa245492015-08-03 10:38:08 -0700517 while (true) {
Spencer Low2e02dc62015-11-07 17:34:39 -0800518 // Use unix_read_interruptible() rather than adb_read() for stdin.
519 D("stdin_read_thread_loop(): pre unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
520 int r = unix_read_interruptible(args->stdin_fd, buffer_ptr,
521 buffer_size);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800522 if (r == -1 && errno == EINTR) {
523 send_window_size_change(args->stdin_fd, args->protocol);
524 continue;
525 }
Spencer Low2e02dc62015-11-07 17:34:39 -0800526 D("stdin_read_thread_loop(): post unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
David Pursell1ed57f02015-10-06 15:30:03 -0700527 if (r <= 0) {
528 // Only devices using the shell protocol know to close subprocess
529 // stdin. For older devices we want to just leave the connection
530 // open, otherwise an unpredictable amount of return data could
531 // be lost due to the FD closing before all data has been received.
532 if (args->protocol) {
533 args->protocol->Write(ShellProtocol::kIdCloseStdin, 0);
534 }
535 break;
536 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800537 // If we made stdin raw, check input for escape sequences. In
David Pursell1ed57f02015-10-06 15:30:03 -0700538 // this situation signals like Ctrl+C are sent remotely rather than
539 // interpreted locally so this provides an emergency out if the remote
540 // process starts ignoring the signal. SSH also does this, see the
541 // "escape characters" section on the ssh man page for more info.
Elliott Hughes28416d62015-11-04 13:36:32 -0800542 if (args->raw_stdin && args->escape_char != '\0') {
543 char ch = buffer_ptr[0];
544 if (ch == args->escape_char) {
545 if (state == kStartOfLine) {
546 state = kInEscape;
547 // Swallow the escape character.
548 continue;
549 } else {
550 state = kMidFlow;
551 }
552 } else {
553 if (state == kInEscape) {
554 if (ch == '.') {
555 fprintf(stderr,"\r\n[ disconnected ]\r\n");
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800556 stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700557 exit(0);
Elliott Hughes28416d62015-11-04 13:36:32 -0800558 } else {
559 // We swallowed an escape character that wasn't part of
560 // a valid escape sequence; time to cough it up.
561 buffer_ptr[0] = args->escape_char;
562 buffer_ptr[1] = ch;
563 ++r;
David Pursell1ed57f02015-10-06 15:30:03 -0700564 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800565 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800566 state = (ch == '\n' || ch == '\r') ? kStartOfLine : kMidFlow;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800567 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800568 }
David Pursell606835a2015-09-08 17:17:02 -0700569 if (args->protocol) {
570 if (!args->protocol->Write(ShellProtocol::kIdStdin, r)) {
571 break;
572 }
573 } else {
574 if (!WriteFdExactly(args->write_fd, buffer_ptr, r)) {
575 break;
576 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800577 }
578 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800579}
580
David Pursell1ed57f02015-10-06 15:30:03 -0700581// Returns a shell service string with the indicated arguments and command.
582static std::string ShellServiceString(bool use_shell_protocol,
583 const std::string& type_arg,
584 const std::string& command) {
585 std::vector<std::string> args;
586 if (use_shell_protocol) {
587 args.push_back(kShellServiceArgShellProtocol);
Elliott Hughesc2252df2015-11-18 12:45:48 -0800588
589 const char* terminal_type = getenv("TERM");
590 if (terminal_type != nullptr) {
591 args.push_back(std::string("TERM=") + terminal_type);
592 }
David Pursell1ed57f02015-10-06 15:30:03 -0700593 }
594 if (!type_arg.empty()) {
595 args.push_back(type_arg);
596 }
597
598 // Shell service string can look like: shell[,arg1,arg2,...]:[command].
599 return android::base::StringPrintf("shell%s%s:%s",
600 args.empty() ? "" : ",",
601 android::base::Join(args, ',').c_str(),
602 command.c_str());
603}
604
605// Connects to a shell on the device and read/writes data.
606//
607// Note: currently this function doesn't properly clean up resources; the
608// FD connected to the adb server is never closed and the stdin read thread
609// may never exit.
610//
611// On success returns the remote exit code if |use_shell_protocol| is true,
612// 0 otherwise. On failure returns 1.
613static int RemoteShell(bool use_shell_protocol, const std::string& type_arg,
Elliott Hughes28416d62015-11-04 13:36:32 -0800614 char escape_char,
David Pursell1ed57f02015-10-06 15:30:03 -0700615 const std::string& command) {
616 std::string service_string = ShellServiceString(use_shell_protocol,
617 type_arg, command);
618
619 // Make local stdin raw if the device allocates a PTY, which happens if:
620 // 1. We are explicitly asking for a PTY shell, or
621 // 2. We don't specify shell type and are starting an interactive session.
622 bool raw_stdin = (type_arg == kShellServiceArgPty ||
623 (type_arg.empty() && command.empty()));
624
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700625 std::string error;
David Pursell4e2fd362015-09-22 10:43:08 -0700626 int fd = adb_connect(service_string, &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700627 if (fd < 0) {
628 fprintf(stderr,"error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800629 return 1;
630 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800631
David Pursell606835a2015-09-08 17:17:02 -0700632 StdinReadArgs* args = new StdinReadArgs;
633 if (!args) {
634 LOG(ERROR) << "couldn't allocate StdinReadArgs object";
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700635 return 1;
636 }
David Pursell1ed57f02015-10-06 15:30:03 -0700637 args->stdin_fd = STDIN_FILENO;
David Pursell606835a2015-09-08 17:17:02 -0700638 args->write_fd = fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700639 args->raw_stdin = raw_stdin;
Elliott Hughes28416d62015-11-04 13:36:32 -0800640 args->escape_char = escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700641 if (use_shell_protocol) {
642 args->protocol.reset(new ShellProtocol(args->write_fd));
643 }
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700644
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800645 if (raw_stdin) stdin_raw_init();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800646
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800647#if !defined(_WIN32)
648 // Ensure our process is notified if the local window size changes.
649 // We use sigaction(2) to ensure that the SA_RESTART flag is not set,
650 // because the whole reason we're sending signals is to unblock the read(2)!
651 // That also means we don't need to do anything in the signal handler:
652 // the side effect of delivering the signal is all we need.
653 struct sigaction sa;
654 memset(&sa, 0, sizeof(sa));
655 sa.sa_handler = [](int) {};
656 sa.sa_flags = 0;
657 sigaction(SIGWINCH, &sa, nullptr);
658
659 // Now block SIGWINCH in this thread (the main thread) and all threads spawned
660 // from it. The stdin read thread will unblock this signal to ensure that it's
661 // the thread that receives the signal.
662 sigset_t mask;
663 sigemptyset(&mask);
664 sigaddset(&mask, SIGWINCH);
665 pthread_sigmask(SIG_BLOCK, &mask, nullptr);
666#endif
667
668 // TODO: combine read_and_dump with stdin_read_thread to make life simpler?
669 int exit_code = 1;
670 if (!adb_thread_create(stdin_read_thread_loop, args)) {
David Pursell606835a2015-09-08 17:17:02 -0700671 PLOG(ERROR) << "error starting stdin read thread";
David Pursell606835a2015-09-08 17:17:02 -0700672 delete args;
673 } else {
674 exit_code = read_and_dump(fd, use_shell_protocol);
675 }
Elliott Hughes9b0f3542015-05-05 13:41:21 -0700676
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800677 // TODO: properly exit stdin_read_thread_loop and close |fd|.
David Pursell1ed57f02015-10-06 15:30:03 -0700678
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800679 // TODO: we should probably install signal handlers for this.
680 // TODO: can we use atexit? even on Windows?
681 if (raw_stdin) stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700682
David Pursell606835a2015-09-08 17:17:02 -0700683 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800684}
685
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800686static int adb_shell(int argc, const char** argv) {
David Pursell0aacbbe2016-01-21 19:56:50 -0800687 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800688 std::string error;
689
690 if (!adb_get_feature_set(&features, &error)) {
691 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -0800692 return 1;
693 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800694
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800695 bool use_shell_protocol = CanUseFeature(features, kFeatureShell2);
696 if (!use_shell_protocol) {
697 D("shell protocol not supported, using raw data transfer");
698 } else {
699 D("using shell protocol");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800700 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700701
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800702 // Parse shell-specific command-line options.
703 // argv[0] is always "shell".
704 --argc;
705 ++argv;
706 int t_arg_count = 0;
Elliott Hughes28416d62015-11-04 13:36:32 -0800707 char escape_char = '~';
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800708 while (argc) {
Elliott Hughes28416d62015-11-04 13:36:32 -0800709 if (!strcmp(argv[0], "-e")) {
710 if (argc < 2 || !(strlen(argv[1]) == 1 || strcmp(argv[1], "none") == 0)) {
711 fprintf(stderr, "error: -e requires a single-character argument or 'none'\n");
712 return 1;
713 }
714 escape_char = (strcmp(argv[1], "none") == 0) ? 0 : argv[1][0];
715 argc -= 2;
716 argv += 2;
717 } else if (!strcmp(argv[0], "-T") || !strcmp(argv[0], "-t")) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800718 // Like ssh, -t arguments are cumulative so that multiple -t's
719 // are needed to force a PTY.
720 if (argv[0][1] == 't') {
721 ++t_arg_count;
722 } else {
723 t_arg_count = -1;
724 }
725 --argc;
726 ++argv;
727 } else if (!strcmp(argv[0], "-x")) {
728 use_shell_protocol = false;
729 --argc;
730 ++argv;
Josh Gao7d586072015-11-20 15:37:31 -0800731 } else if (!strcmp(argv[0], "-n")) {
732 close_stdin();
733
734 --argc;
735 ++argv;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800736 } else {
737 break;
738 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700739 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800740
David Pursell57dd5ae2016-01-27 16:07:52 -0800741 // Legacy shell protocol requires a remote PTY to close the subprocess properly which creates
742 // some weird interactions with -tT.
743 if (!use_shell_protocol && t_arg_count != 0) {
744 if (!CanUseFeature(features, kFeatureShell2)) {
745 fprintf(stderr, "error: target doesn't support PTY args -Tt\n");
746 } else {
747 fprintf(stderr, "error: PTY args -Tt cannot be used with -x\n");
748 }
749 return 1;
750 }
751
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800752 std::string shell_type_arg;
753 if (CanUseFeature(features, kFeatureShell2)) {
754 if (t_arg_count < 0) {
755 shell_type_arg = kShellServiceArgRaw;
756 } else if (t_arg_count == 0) {
757 // If stdin isn't a TTY, default to a raw shell; this lets
758 // things like `adb shell < my_script.sh` work as expected.
759 // Otherwise leave |shell_type_arg| blank which uses PTY for
760 // interactive shells and raw for non-interactive.
761 if (!unix_isatty(STDIN_FILENO)) {
762 shell_type_arg = kShellServiceArgRaw;
763 }
764 } else if (t_arg_count == 1) {
765 // A single -t arg isn't enough to override implicit -T.
766 if (!unix_isatty(STDIN_FILENO)) {
767 fprintf(stderr,
768 "Remote PTY will not be allocated because stdin is not a terminal.\n"
769 "Use multiple -t options to force remote PTY allocation.\n");
770 shell_type_arg = kShellServiceArgRaw;
771 } else {
772 shell_type_arg = kShellServiceArgPty;
773 }
774 } else {
775 shell_type_arg = kShellServiceArgPty;
776 }
David Pursell71c83122015-09-14 15:33:50 -0700777 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800778
779 std::string command;
780 if (argc) {
781 // We don't escape here, just like ssh(1). http://b/20564385.
782 command = android::base::Join(std::vector<const char*>(argv, argv + argc), ' ');
783 }
784
Elliott Hughes28416d62015-11-04 13:36:32 -0800785 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, command);
David Pursell71c83122015-09-14 15:33:50 -0700786}
787
Elliott Hughes6452a892015-04-29 12:28:13 -0700788static int adb_download_buffer(const char *service, const char *fn, const void* data, unsigned sz,
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700789 bool show_progress)
Doug Zongker447f0612012-01-09 14:54:53 -0800790{
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700791 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -0700792 int fd = adb_connect(android::base::StringPrintf("%s:%d", service, sz), &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700793 if (fd < 0) {
794 fprintf(stderr,"error: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800795 return -1;
796 }
797
798 int opt = CHUNK_SIZE;
Spencer Lowf055c192015-01-25 14:40:16 -0800799 opt = adb_setsockopt(fd, SOL_SOCKET, SO_SNDBUF, (const void *) &opt, sizeof(opt));
Doug Zongker447f0612012-01-09 14:54:53 -0800800
Elliott Hughes6452a892015-04-29 12:28:13 -0700801 unsigned total = sz;
Dan Albertbac34742015-02-25 17:51:28 -0800802 const uint8_t* ptr = reinterpret_cast<const uint8_t*>(data);
Doug Zongker447f0612012-01-09 14:54:53 -0800803
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700804 if (show_progress) {
Elliott Hughes3e7048c2015-07-27 21:21:39 -0700805 const char* x = strrchr(service, ':');
806 if (x) service = x + 1;
Doug Zongker447f0612012-01-09 14:54:53 -0800807 }
808
Elliott Hughes6452a892015-04-29 12:28:13 -0700809 while (sz > 0) {
Doug Zongker447f0612012-01-09 14:54:53 -0800810 unsigned xfer = (sz > CHUNK_SIZE) ? CHUNK_SIZE : sz;
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700811 if (!WriteFdExactly(fd, ptr, xfer)) {
812 std::string error;
813 adb_status(fd, &error);
814 fprintf(stderr,"* failed to write data '%s' *\n", error.c_str());
Spencer Low351ecd12015-10-14 17:32:44 -0700815 adb_close(fd);
Doug Zongker447f0612012-01-09 14:54:53 -0800816 return -1;
817 }
818 sz -= xfer;
819 ptr += xfer;
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700820 if (show_progress) {
Magnus Eriksson86ae6d52013-03-05 07:37:32 +0100821 printf("sending: '%s' %4d%% \r", fn, (int)(100LL - ((100LL * sz) / (total))));
Doug Zongker447f0612012-01-09 14:54:53 -0800822 fflush(stdout);
823 }
824 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700825 if (show_progress) {
Doug Zongker447f0612012-01-09 14:54:53 -0800826 printf("\n");
827 }
828
Elliott Hughese67f1f82015-04-30 17:32:03 -0700829 if (!adb_status(fd, &error)) {
830 fprintf(stderr,"* error response '%s' *\n", error.c_str());
Spencer Low351ecd12015-10-14 17:32:44 -0700831 adb_close(fd);
Doug Zongker447f0612012-01-09 14:54:53 -0800832 return -1;
833 }
834
835 adb_close(fd);
836 return 0;
837}
838
Doug Zongker71fe5842014-06-26 15:35:36 -0700839#define SIDELOAD_HOST_BLOCK_SIZE (CHUNK_SIZE)
840
841/*
842 * The sideload-host protocol serves the data in a file (given on the
843 * command line) to the client, using a simple protocol:
844 *
845 * - The connect message includes the total number of bytes in the
846 * file and a block size chosen by us.
847 *
848 * - The other side sends the desired block number as eight decimal
849 * digits (eg "00000023" for block 23). Blocks are numbered from
850 * zero.
851 *
852 * - We send back the data of the requested block. The last block is
853 * likely to be partial; when the last block is requested we only
854 * send the part of the block that exists, it's not padded up to the
855 * block size.
856 *
857 * - When the other side sends "DONEDONE" instead of a block number,
858 * we hang up.
859 */
Elliott Hughes58305772015-04-17 13:57:15 -0700860static int adb_sideload_host(const char* fn) {
Elliott Hughes14415142016-06-15 14:45:34 -0700861 fprintf(stderr, "loading: '%s'...\n", fn);
Elliott Hughese8b663f2016-05-26 22:43:19 -0700862
863 std::string content;
864 if (!android::base::ReadFileToString(fn, &content)) {
Elliott Hughes14415142016-06-15 14:45:34 -0700865 fprintf(stderr, "failed: %s\n", strerror(errno));
Doug Zongker71fe5842014-06-26 15:35:36 -0700866 return -1;
867 }
868
Elliott Hughese8b663f2016-05-26 22:43:19 -0700869 const uint8_t* data = reinterpret_cast<const uint8_t*>(content.data());
870 unsigned sz = content.size();
871
Elliott Hughes14415142016-06-15 14:45:34 -0700872 fprintf(stderr, "connecting...\n");
Elliott Hughes6452a892015-04-29 12:28:13 -0700873 std::string service =
874 android::base::StringPrintf("sideload-host:%d:%d", sz, SIDELOAD_HOST_BLOCK_SIZE);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700875 std::string error;
Elliott Hughese8b663f2016-05-26 22:43:19 -0700876 unique_fd fd(adb_connect(service, &error));
Elliott Hughes356fd362016-06-13 15:01:13 -0700877 if (fd < 0) {
Doug Zongker71fe5842014-06-26 15:35:36 -0700878 // Try falling back to the older sideload method. Maybe this
879 // is an older device that doesn't support sideload-host.
Elliott Hughes14415142016-06-15 14:45:34 -0700880 fprintf(stderr, "falling back to older sideload method...\n");
Elliott Hughese8b663f2016-05-26 22:43:19 -0700881 return adb_download_buffer("sideload", fn, data, sz, true);
Doug Zongker71fe5842014-06-26 15:35:36 -0700882 }
883
Elliott Hughese8b663f2016-05-26 22:43:19 -0700884 int opt = SIDELOAD_HOST_BLOCK_SIZE;
885 adb_setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
Doug Zongker71fe5842014-06-26 15:35:36 -0700886
Elliott Hughese8b663f2016-05-26 22:43:19 -0700887 size_t xfer = 0;
888 int last_percent = -1;
Elliott Hughesa7090b92015-04-17 17:03:59 -0700889 while (true) {
Elliott Hughes6452a892015-04-29 12:28:13 -0700890 char buf[9];
Dan Albertcc731cc2015-02-24 21:26:58 -0800891 if (!ReadFdExactly(fd, buf, 8)) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700892 fprintf(stderr, "* failed to read command: %s\n", strerror(errno));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700893 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700894 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700895 buf[8] = '\0';
Doug Zongker71fe5842014-06-26 15:35:36 -0700896
Elliott Hughes6452a892015-04-29 12:28:13 -0700897 if (strcmp("DONEDONE", buf) == 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -0700898 printf("\rTotal xfer: %.2fx%*s\n",
899 (double)xfer / (sz ? sz : 1), (int)strlen(fn)+10, "");
900 return 0;
Doug Zongker71fe5842014-06-26 15:35:36 -0700901 }
902
Doug Zongker71fe5842014-06-26 15:35:36 -0700903 int block = strtol(buf, NULL, 10);
904
905 size_t offset = block * SIDELOAD_HOST_BLOCK_SIZE;
906 if (offset >= sz) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700907 fprintf(stderr, "* attempt to read block %d past end\n", block);
Elliott Hughese8b663f2016-05-26 22:43:19 -0700908 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700909 }
Elliott Hughese8b663f2016-05-26 22:43:19 -0700910 const uint8_t* start = data + offset;
Doug Zongker71fe5842014-06-26 15:35:36 -0700911 size_t offset_end = offset + SIDELOAD_HOST_BLOCK_SIZE;
912 size_t to_write = SIDELOAD_HOST_BLOCK_SIZE;
913 if (offset_end > sz) {
914 to_write = sz - offset;
915 }
916
Elliott Hughese8b663f2016-05-26 22:43:19 -0700917 if (!WriteFdExactly(fd, start, to_write)) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700918 adb_status(fd, &error);
919 fprintf(stderr,"* failed to write data '%s' *\n", error.c_str());
Elliott Hughese8b663f2016-05-26 22:43:19 -0700920 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700921 }
922 xfer += to_write;
923
924 // For normal OTA packages, we expect to transfer every byte
925 // twice, plus a bit of overhead (one read during
926 // verification, one read of each byte for installation, plus
927 // extra access to things like the zip central directory).
928 // This estimate of the completion becomes 100% when we've
929 // transferred ~2.13 (=100/47) times the package size.
930 int percent = (int)(xfer * 47LL / (sz ? sz : 1));
931 if (percent != last_percent) {
932 printf("\rserving: '%s' (~%d%%) ", fn, percent);
933 fflush(stdout);
934 last_percent = percent;
935 }
936 }
Doug Zongker71fe5842014-06-26 15:35:36 -0700937}
938
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800939/**
940 * Run ppp in "notty" mode against a resource listed as the first parameter
941 * eg:
942 *
943 * ppp dev:/dev/omap_csmi_tty0 <ppp options>
944 *
945 */
Elliott Hughes58305772015-04-17 13:57:15 -0700946static int ppp(int argc, const char** argv) {
Yabin Cuie77b6a02014-11-11 09:24:11 -0800947#if defined(_WIN32)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800948 fprintf(stderr, "error: adb %s not implemented on Win32\n", argv[0]);
949 return -1;
950#else
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800951 if (argc < 2) {
952 fprintf(stderr, "usage: adb %s <adb service name> [ppp opts]\n",
953 argv[0]);
954
955 return 1;
956 }
957
Dan Albertbac34742015-02-25 17:51:28 -0800958 const char* adb_service_name = argv[1];
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700959 std::string error;
960 int fd = adb_connect(adb_service_name, &error);
961 if (fd < 0) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800962 fprintf(stderr,"Error: Could not open adb service: %s. Error: %s\n",
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700963 adb_service_name, error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800964 return 1;
965 }
966
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700967 pid_t pid = fork();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800968
969 if (pid < 0) {
970 perror("from fork()");
971 return 1;
972 } else if (pid == 0) {
973 int err;
974 int i;
975 const char **ppp_args;
976
977 // copy args
978 ppp_args = (const char **) alloca(sizeof(char *) * argc + 1);
979 ppp_args[0] = "pppd";
980 for (i = 2 ; i < argc ; i++) {
981 //argv[2] and beyond become ppp_args[1] and beyond
982 ppp_args[i - 1] = argv[i];
983 }
984 ppp_args[i-1] = NULL;
985
986 // child side
987
988 dup2(fd, STDIN_FILENO);
989 dup2(fd, STDOUT_FILENO);
990 adb_close(STDERR_FILENO);
991 adb_close(fd);
992
993 err = execvp("pppd", (char * const *)ppp_args);
994
995 if (err < 0) {
996 perror("execing pppd");
997 }
998 exit(-1);
999 } else {
1000 // parent side
1001
1002 adb_close(fd);
1003 return 0;
1004 }
Yabin Cuie77b6a02014-11-11 09:24:11 -08001005#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001006}
1007
Josh Gao08881e72016-04-13 12:14:16 -07001008static bool wait_for_device(const char* service, TransportType t, const char* serial) {
1009 std::vector<std::string> components = android::base::Split(service, "-");
1010 if (components.size() < 3 || components.size() > 4) {
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001011 fprintf(stderr, "adb: couldn't parse 'wait-for' command: %s\n", service);
1012 return false;
1013 }
1014
Elliott Hughes2b101112015-05-04 19:29:32 -07001015 // Was the caller vague about what they'd like us to wait for?
1016 // If so, check they weren't more specific in their choice of transport type.
Josh Gao08881e72016-04-13 12:14:16 -07001017 if (components.size() == 3) {
1018 auto it = components.begin() + 2;
Elliott Hughes2b101112015-05-04 19:29:32 -07001019 if (t == kTransportUsb) {
Josh Gao08881e72016-04-13 12:14:16 -07001020 components.insert(it, "usb");
Elliott Hughes2b101112015-05-04 19:29:32 -07001021 } else if (t == kTransportLocal) {
Josh Gao08881e72016-04-13 12:14:16 -07001022 components.insert(it, "local");
Elliott Hughes2b101112015-05-04 19:29:32 -07001023 } else {
Josh Gao08881e72016-04-13 12:14:16 -07001024 components.insert(it, "any");
Elliott Hughes2b101112015-05-04 19:29:32 -07001025 }
Josh Gao08881e72016-04-13 12:14:16 -07001026 } else if (components[2] != "any" && components[2] != "local" && components[2] != "usb") {
1027 fprintf(stderr, "adb: unknown type %s; expected 'any', 'local', or 'usb'\n",
1028 components[2].c_str());
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001029 return false;
1030 }
1031
Josh Gao86441c32016-04-13 12:18:58 -07001032 if (components[3] != "any" && components[3] != "bootloader" && components[3] != "device" &&
1033 components[3] != "recovery" && components[3] != "sideload") {
Josh Gao08881e72016-04-13 12:14:16 -07001034 fprintf(stderr,
1035 "adb: unknown state %s; "
Josh Gao86441c32016-04-13 12:18:58 -07001036 "expected 'any', 'bootloader', 'device', 'recovery', or 'sideload'\n",
Josh Gao08881e72016-04-13 12:14:16 -07001037 components[3].c_str());
1038 return false;
1039 }
1040
1041 std::string cmd = format_host_command(android::base::Join(components, "-").c_str(), t, serial);
Elliott Hughes424af022015-05-29 17:55:19 -07001042 return adb_command(cmd);
Elliott Hughes2b101112015-05-04 19:29:32 -07001043}
1044
Josh Gao06c73ae2016-03-04 15:16:18 -08001045static bool adb_root(const char* command) {
1046 std::string error;
Josh Gao06c73ae2016-03-04 15:16:18 -08001047
Elliott Hughese8b663f2016-05-26 22:43:19 -07001048 unique_fd fd(adb_connect(android::base::StringPrintf("%s:", command), &error));
1049 if (fd < 0) {
Josh Gao06c73ae2016-03-04 15:16:18 -08001050 fprintf(stderr, "adb: unable to connect for %s: %s\n", command, error.c_str());
1051 return false;
1052 }
1053
1054 // Figure out whether we actually did anything.
1055 char buf[256];
1056 char* cur = buf;
1057 ssize_t bytes_left = sizeof(buf);
1058 while (bytes_left > 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -07001059 ssize_t bytes_read = adb_read(fd, cur, bytes_left);
Josh Gao06c73ae2016-03-04 15:16:18 -08001060 if (bytes_read == 0) {
1061 break;
1062 } else if (bytes_read < 0) {
1063 fprintf(stderr, "adb: error while reading for %s: %s\n", command, strerror(errno));
1064 return false;
1065 }
1066 cur += bytes_read;
1067 bytes_left -= bytes_read;
1068 }
1069
1070 if (bytes_left == 0) {
1071 fprintf(stderr, "adb: unexpected output length for %s\n", command);
1072 return false;
1073 }
1074
1075 fflush(stdout);
1076 WriteFdExactly(STDOUT_FILENO, buf, sizeof(buf) - bytes_left);
1077 if (cur != buf && strstr(buf, "restarting") == nullptr) {
1078 return true;
1079 }
1080
Josh Gao05824732016-06-16 14:00:09 -07001081 // Give adbd some time to kill itself and come back up.
1082 // We can't use wait-for-device because devices (e.g. adb over network) might not come back.
1083 adb_sleep_ms(3000);
1084 return true;
Josh Gao06c73ae2016-03-04 15:16:18 -08001085}
1086
Felipe Leme698e0652016-07-19 17:07:22 -07001087int send_shell_command(TransportType transport_type, const char* serial, const std::string& command,
Felipe Leme0d4f0502016-07-26 12:14:39 -07001088 bool disable_shell_protocol, StandardStreamsCallbackInterface* callback) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001089 int fd;
David Pursell0aacbbe2016-01-21 19:56:50 -08001090 bool use_shell_protocol = false;
1091
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001092 while (true) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001093 bool attempt_connection = true;
1094
Felipe Leme0d4f0502016-07-26 12:14:39 -07001095 // Use shell protocol if it's supported and the caller doesn't explicitly
1096 // disable it.
David Pursell0aacbbe2016-01-21 19:56:50 -08001097 if (!disable_shell_protocol) {
1098 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001099 std::string error;
1100 if (adb_get_feature_set(&features, &error)) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001101 use_shell_protocol = CanUseFeature(features, kFeatureShell2);
1102 } else {
1103 // Device was unreachable.
1104 attempt_connection = false;
1105 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001106 }
David Pursell0aacbbe2016-01-21 19:56:50 -08001107
1108 if (attempt_connection) {
1109 std::string error;
1110 std::string service_string = ShellServiceString(use_shell_protocol, "", command);
1111
1112 fd = adb_connect(service_string, &error);
1113 if (fd >= 0) {
1114 break;
1115 }
1116 }
1117
Felipe Leme0d4f0502016-07-26 12:14:39 -07001118 fprintf(stderr, "- waiting for device -\n");
Josh Gaob6117c42016-02-24 15:16:17 -08001119 if (!wait_for_device("wait-for-device", transport_type, serial)) {
1120 return 1;
1121 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001122 }
1123
Felipe Leme0d4f0502016-07-26 12:14:39 -07001124 int exit_code = read_and_dump(fd, use_shell_protocol, callback);
David Pursell71c83122015-09-14 15:33:50 -07001125
1126 if (adb_close(fd) < 0) {
1127 PLOG(ERROR) << "failure closing FD " << fd;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001128 }
David Pursell71c83122015-09-14 15:33:50 -07001129
1130 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001131}
1132
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001133static int logcat(TransportType transport, const char* serial, int argc, const char** argv) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001134 char* log_tags = getenv("ANDROID_LOG_TAGS");
1135 std::string quoted = escape_arg(log_tags == nullptr ? "" : log_tags);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001136
David Pursell70ef7b42015-09-30 13:35:42 -07001137 std::string cmd = "export ANDROID_LOG_TAGS=\"" + quoted + "\"; exec logcat";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001138
Jeff Sharkeyfd546e82014-06-10 11:31:24 -07001139 if (!strcmp(argv[0], "longcat")) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001140 cmd += " -v long";
Christopher Tatedb0a8802011-11-30 13:00:33 -08001141 }
1142
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001143 --argc;
1144 ++argv;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001145 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001146 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001147 }
1148
David Pursell70ef7b42015-09-30 13:35:42 -07001149 // No need for shell protocol with logcat, always disable for simplicity.
1150 return send_shell_command(transport, serial, cmd, true);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001151}
1152
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001153static void write_zeros(int bytes, int fd) {
1154 int old_stdin_mode = -1;
1155 int old_stdout_mode = -1;
1156 char* buf = (char*) calloc(1, bytes);
1157 if (buf == nullptr) fatal("couldn't allocate buffer for write_zeros");
1158
1159 D("write_zeros(%d) -> %d", bytes, fd);
1160
1161 stdinout_raw_prologue(-1, fd, old_stdin_mode, old_stdout_mode);
1162
1163 if (fd == STDOUT_FILENO) {
1164 fwrite(buf, 1, bytes, stdout);
1165 fflush(stdout);
1166 } else {
1167 adb_write(fd, buf, bytes);
1168 }
1169
1170 stdinout_raw_prologue(-1, fd, old_stdin_mode, old_stdout_mode);
1171
1172 D("write_zeros() finished");
1173 free(buf);
1174}
1175
Dan Albertbac34742015-02-25 17:51:28 -08001176static int backup(int argc, const char** argv) {
Elliott Hughes24f165f2015-08-21 20:31:31 -07001177 const char* filename = "backup.ab";
Christopher Tated2f54152011-04-21 12:53:28 -07001178
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001179 /* find, extract, and use any -f argument */
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001180 for (int i = 1; i < argc; i++) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001181 if (!strcmp("-f", argv[i])) {
1182 if (i == argc-1) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001183 fprintf(stderr, "adb: backup -f passed with no filename.\n");
1184 return EXIT_FAILURE;
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001185 }
1186 filename = argv[i+1];
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001187 for (int j = i+2; j <= argc; ) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001188 argv[i++] = argv[j++];
1189 }
1190 argc -= 2;
1191 argv[argc] = NULL;
1192 }
Christopher Tated2f54152011-04-21 12:53:28 -07001193 }
1194
Elliott Hughes56e68132015-11-13 11:04:10 -08001195 // Bare "adb backup" or "adb backup -f filename" are not valid invocations ---
1196 // a list of packages is required.
1197 if (argc < 2) {
1198 fprintf(stderr, "adb: backup either needs a list of packages or -all/-shared.\n");
1199 return EXIT_FAILURE;
1200 }
Christopher Tatebb86bc52011-08-22 17:12:08 -07001201
Christopher Tateb1dfffe2011-12-08 19:04:34 -08001202 adb_unlink(filename);
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001203 int outFd = adb_creat(filename, 0640);
Christopher Tated2f54152011-04-21 12:53:28 -07001204 if (outFd < 0) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001205 fprintf(stderr, "adb: backup unable to create file '%s': %s\n", filename, strerror(errno));
1206 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001207 }
1208
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001209 std::string cmd = "backup:";
1210 --argc;
1211 ++argv;
1212 while (argc-- > 0) {
1213 cmd += " " + escape_arg(*argv++);
Christopher Tated2f54152011-04-21 12:53:28 -07001214 }
1215
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001216 D("backup. filename=%s cmd=%s", filename, cmd.c_str());
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001217 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001218 int fd = adb_connect(cmd, &error);
Christopher Tated2f54152011-04-21 12:53:28 -07001219 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001220 fprintf(stderr, "adb: unable to connect for backup: %s\n", error.c_str());
Christopher Tated2f54152011-04-21 12:53:28 -07001221 adb_close(outFd);
Elliott Hughes56e68132015-11-13 11:04:10 -08001222 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001223 }
1224
Elliott Hughes56e68132015-11-13 11:04:10 -08001225 printf("Now unlock your device and confirm the backup operation...\n");
1226 fflush(stdout);
1227
Christopher Tated2f54152011-04-21 12:53:28 -07001228 copy_to_file(fd, outFd);
1229
1230 adb_close(fd);
1231 adb_close(outFd);
Elliott Hughes56e68132015-11-13 11:04:10 -08001232 return EXIT_SUCCESS;
Christopher Tated2f54152011-04-21 12:53:28 -07001233}
1234
Dan Albertbac34742015-02-25 17:51:28 -08001235static int restore(int argc, const char** argv) {
Christopher Tate702967a2011-05-17 15:52:54 -07001236 if (argc != 2) return usage();
1237
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001238 const char* filename = argv[1];
1239 int tarFd = adb_open(filename, O_RDONLY);
Christopher Tate702967a2011-05-17 15:52:54 -07001240 if (tarFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001241 fprintf(stderr, "adb: unable to open file %s: %s\n", filename, strerror(errno));
Christopher Tate702967a2011-05-17 15:52:54 -07001242 return -1;
1243 }
1244
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001245 std::string error;
1246 int fd = adb_connect("restore:", &error);
Christopher Tate702967a2011-05-17 15:52:54 -07001247 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001248 fprintf(stderr, "adb: unable to connect for restore: %s\n", error.c_str());
Christopher Tate702967a2011-05-17 15:52:54 -07001249 adb_close(tarFd);
1250 return -1;
1251 }
1252
Christopher Tatebffa4ca2012-01-06 15:43:03 -08001253 printf("Now unlock your device and confirm the restore operation.\n");
Christopher Tate702967a2011-05-17 15:52:54 -07001254 copy_to_file(tarFd, fd);
1255
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001256 // Provide an in-band EOD marker in case the archive file is malformed
1257 write_zeros(512*2, fd);
1258
Josh Gao5767d0d2016-03-04 15:51:03 -08001259 // Wait until the other side finishes, or it'll get sent SIGHUP.
1260 copy_to_file(fd, STDOUT_FILENO);
1261
Christopher Tate702967a2011-05-17 15:52:54 -07001262 adb_close(fd);
1263 adb_close(tarFd);
1264 return 0;
1265}
1266
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001267/* <hint> may be:
1268 * - A simple product name
1269 * e.g., "sooner"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001270 * - A relative path from the CWD to the ANDROID_PRODUCT_OUT dir
1271 * e.g., "out/target/product/sooner"
1272 * - An absolute path to the PRODUCT_OUT dir
1273 * e.g., "/src/device/out/target/product/sooner"
1274 *
1275 * Given <hint>, try to construct an absolute path to the
1276 * ANDROID_PRODUCT_OUT dir.
1277 */
Elliott Hughes5c742702015-07-30 17:42:01 -07001278static std::string find_product_out_path(const std::string& hint) {
1279 if (hint.empty()) {
Elliott Hughes58305772015-04-17 13:57:15 -07001280 return "";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001281 }
1282
Elliott Hughes58305772015-04-17 13:57:15 -07001283 // If it's already absolute, don't bother doing any work.
Elliott Hughes5c742702015-07-30 17:42:01 -07001284 if (adb_is_absolute_host_path(hint.c_str())) {
Elliott Hughes58305772015-04-17 13:57:15 -07001285 return hint;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001286 }
1287
Chih-Hung Hsieh8f7b9e32016-07-27 16:25:51 -07001288 // If any of the OS_PATH_SEPARATORS is found, assume it's a relative path;
Elliott Hughes58305772015-04-17 13:57:15 -07001289 // make it absolute.
Chih-Hung Hsieh8f7b9e32016-07-27 16:25:51 -07001290 // NOLINT: Do not complain if OS_PATH_SEPARATORS has only one character.
1291 if (hint.find_first_of(OS_PATH_SEPARATORS) != std::string::npos) { // NOLINT
Elliott Hughesa7090b92015-04-17 17:03:59 -07001292 std::string cwd;
1293 if (!getcwd(&cwd)) {
Felipe Lemeb3239722016-07-29 17:57:00 -07001294 perror("adb: getcwd failed");
Elliott Hughes58305772015-04-17 13:57:15 -07001295 return "";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001296 }
Elliott Hughes5c742702015-07-30 17:42:01 -07001297 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 -08001298 }
1299
Elliott Hughes58305772015-04-17 13:57:15 -07001300 // It's a string without any slashes. Try to do something with it.
1301 //
1302 // Try to find the root of the build tree, and build a PRODUCT_OUT
1303 // path from there.
Elliott Hughesa7090b92015-04-17 17:03:59 -07001304 char* top = getenv("ANDROID_BUILD_TOP");
Elliott Hughes58305772015-04-17 13:57:15 -07001305 if (top == nullptr) {
Elliott Hughesa7090b92015-04-17 17:03:59 -07001306 fprintf(stderr, "adb: ANDROID_BUILD_TOP not set!\n");
Elliott Hughes58305772015-04-17 13:57:15 -07001307 return "";
1308 }
Elliott Hughesa7090b92015-04-17 17:03:59 -07001309
1310 std::string path = top;
Elliott Hughes58305772015-04-17 13:57:15 -07001311 path += OS_PATH_SEPARATOR_STR;
1312 path += "out";
1313 path += OS_PATH_SEPARATOR_STR;
1314 path += "target";
1315 path += OS_PATH_SEPARATOR_STR;
1316 path += "product";
1317 path += OS_PATH_SEPARATOR_STR;
1318 path += hint;
1319 if (!directory_exists(path)) {
1320 fprintf(stderr, "adb: Couldn't find a product dir based on -p %s; "
Elliott Hughes5c742702015-07-30 17:42:01 -07001321 "\"%s\" doesn't exist\n", hint.c_str(), path.c_str());
Elliott Hughesa7090b92015-04-17 17:03:59 -07001322 return "";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001323 }
Elliott Hughes58305772015-04-17 13:57:15 -07001324 return path;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001325}
1326
Josh Gao05786772015-10-30 16:57:19 -07001327static void parse_push_pull_args(const char** arg, int narg,
1328 std::vector<const char*>* srcs,
1329 const char** dst, bool* copy_attrs) {
1330 *copy_attrs = false;
Mark Lindner76f2a932014-03-11 17:55:59 -07001331
Josh Gao05786772015-10-30 16:57:19 -07001332 srcs->clear();
1333 bool ignore_flags = false;
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001334 while (narg > 0) {
Josh Gao05786772015-10-30 16:57:19 -07001335 if (ignore_flags || *arg[0] != '-') {
1336 srcs->push_back(*arg);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001337 } else {
Josh Gao05786772015-10-30 16:57:19 -07001338 if (!strcmp(*arg, "-p")) {
1339 // Silently ignore for backwards compatibility.
1340 } else if (!strcmp(*arg, "-a")) {
1341 *copy_attrs = true;
1342 } else if (!strcmp(*arg, "--")) {
1343 ignore_flags = true;
1344 } else {
1345 fprintf(stderr, "adb: unrecognized option '%s'\n", *arg);
1346 exit(1);
1347 }
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001348 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001349 ++arg;
1350 --narg;
1351 }
1352
Josh Gao05786772015-10-30 16:57:19 -07001353 if (srcs->size() > 1) {
1354 *dst = srcs->back();
1355 srcs->pop_back();
Mark Lindner76f2a932014-03-11 17:55:59 -07001356 }
1357}
1358
Elliott Hughes6452a892015-04-29 12:28:13 -07001359static int adb_connect_command(const std::string& command) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001360 std::string error;
1361 int fd = adb_connect(command, &error);
Elliott Hughes5677c232015-05-07 23:37:40 -07001362 if (fd < 0) {
1363 fprintf(stderr, "error: %s\n", error.c_str());
1364 return 1;
Tao Bao175b7bb2015-03-29 11:22:34 -07001365 }
Elliott Hughes5677c232015-05-07 23:37:40 -07001366 read_and_dump(fd);
1367 adb_close(fd);
1368 return 0;
Tao Bao175b7bb2015-03-29 11:22:34 -07001369}
1370
Elliott Hughes6452a892015-04-29 12:28:13 -07001371static int adb_query_command(const std::string& command) {
1372 std::string result;
1373 std::string error;
1374 if (!adb_query(command, &result, &error)) {
1375 fprintf(stderr, "error: %s\n", error.c_str());
1376 return 1;
1377 }
1378 printf("%s\n", result.c_str());
1379 return 0;
1380}
1381
Spencer Lowa13df302015-09-07 16:20:13 -07001382// Disallow stdin, stdout, and stderr.
1383static bool _is_valid_ack_reply_fd(const int ack_reply_fd) {
1384#ifdef _WIN32
1385 const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd);
1386 return (GetStdHandle(STD_INPUT_HANDLE) != ack_reply_handle) &&
1387 (GetStdHandle(STD_OUTPUT_HANDLE) != ack_reply_handle) &&
1388 (GetStdHandle(STD_ERROR_HANDLE) != ack_reply_handle);
1389#else
1390 return ack_reply_fd > 2;
1391#endif
1392}
1393
Todd Kennedy248722e2016-07-20 16:22:37 -07001394static bool _use_legacy_install() {
1395 FeatureSet features;
1396 std::string error;
1397 if (!adb_get_feature_set(&features, &error)) {
1398 fprintf(stderr, "error: %s\n", error.c_str());
1399 return true;
1400 }
1401 return !CanUseFeature(features, kFeatureCmd);
1402}
1403
Elliott Hughesab52c182015-05-01 17:04:38 -07001404int adb_commandline(int argc, const char **argv) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001405 int no_daemon = 0;
1406 int is_daemon = 0;
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001407 int is_server = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001408 int r;
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001409 TransportType transport_type = kTransportAny;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001410 int ack_reply_fd = -1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001411
Elliott Hughes65433da2015-11-18 18:07:48 -08001412#if !defined(_WIN32)
1413 // We'd rather have EPIPE than SIGPIPE.
1414 signal(SIGPIPE, SIG_IGN);
1415#endif
1416
Elliott Hughes58305772015-04-17 13:57:15 -07001417 // If defined, this should be an absolute path to
1418 // the directory containing all of the various system images
1419 // for a particular product. If not defined, and the adb
1420 // command requires this information, then the user must
1421 // specify the path using "-p".
1422 char* ANDROID_PRODUCT_OUT = getenv("ANDROID_PRODUCT_OUT");
1423 if (ANDROID_PRODUCT_OUT != nullptr) {
1424 gProductOutPath = ANDROID_PRODUCT_OUT;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001425 }
1426 // TODO: also try TARGET_PRODUCT/TARGET_DEVICE as a hint
1427
Josh Gao9c869b52016-08-25 16:00:22 -07001428 const char* server_host_str = nullptr;
1429 const char* server_port_str = nullptr;
1430 const char* server_socket_str = nullptr;
Stefan Hilzingera84a42e2010-04-19 12:21:12 +01001431
Elliott Hughes8d28e192015-10-07 14:55:10 -07001432 // We need to check for -d and -e before we look at $ANDROID_SERIAL.
1433 const char* serial = nullptr;
1434
Riley Andrews98f58e82014-12-05 17:37:24 -08001435 while (argc > 0) {
1436 if (!strcmp(argv[0],"server")) {
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001437 is_server = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001438 } else if (!strcmp(argv[0],"nodaemon")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001439 no_daemon = 1;
1440 } else if (!strcmp(argv[0], "fork-server")) {
1441 /* this is a special flag used only when the ADB client launches the ADB Server */
1442 is_daemon = 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001443 } else if (!strcmp(argv[0], "--reply-fd")) {
1444 if (argc < 2) return usage();
1445 const char* reply_fd_str = argv[1];
1446 argc--;
1447 argv++;
1448 ack_reply_fd = strtol(reply_fd_str, nullptr, 10);
Spencer Lowa13df302015-09-07 16:20:13 -07001449 if (!_is_valid_ack_reply_fd(ack_reply_fd)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001450 fprintf(stderr, "adb: invalid reply fd \"%s\"\n", reply_fd_str);
1451 return usage();
1452 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001453 } else if (!strncmp(argv[0], "-p", 2)) {
Elliott Hughes048b27c2015-07-31 13:18:22 -07001454 const char* product = nullptr;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001455 if (argv[0][2] == '\0') {
1456 if (argc < 2) return usage();
1457 product = argv[1];
1458 argc--;
1459 argv++;
1460 } else {
Vairavan Srinivasan81273232012-08-04 16:40:50 -07001461 product = argv[0] + 2;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001462 }
Elliott Hughes048b27c2015-07-31 13:18:22 -07001463 gProductOutPath = find_product_out_path(product);
Elliott Hughes58305772015-04-17 13:57:15 -07001464 if (gProductOutPath.empty()) {
1465 fprintf(stderr, "adb: could not resolve \"-p %s\"\n", product);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001466 return usage();
1467 }
1468 } else if (argv[0][0]=='-' && argv[0][1]=='s') {
1469 if (isdigit(argv[0][2])) {
1470 serial = argv[0] + 2;
1471 } else {
Riley Andrews98f58e82014-12-05 17:37:24 -08001472 if (argc < 2 || argv[0][2] != '\0') return usage();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001473 serial = argv[1];
1474 argc--;
1475 argv++;
1476 }
1477 } else if (!strcmp(argv[0],"-d")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001478 transport_type = kTransportUsb;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001479 } else if (!strcmp(argv[0],"-e")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001480 transport_type = kTransportLocal;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001481 } else if (!strcmp(argv[0],"-a")) {
1482 gListenAll = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001483 } else if (!strncmp(argv[0], "-H", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001484 if (argv[0][2] == '\0') {
1485 if (argc < 2) return usage();
Josh Gao9c869b52016-08-25 16:00:22 -07001486 server_host_str = argv[1];
Matt Gumbeld7b33082012-11-14 10:16:17 -08001487 argc--;
1488 argv++;
1489 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001490 server_host_str = argv[0] + 2;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001491 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001492 } else if (!strncmp(argv[0], "-P", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001493 if (argv[0][2] == '\0') {
1494 if (argc < 2) return usage();
1495 server_port_str = argv[1];
1496 argc--;
1497 argv++;
1498 } else {
1499 server_port_str = argv[0] + 2;
1500 }
Josh Gao9c869b52016-08-25 16:00:22 -07001501 } else if (!strcmp(argv[0], "-L")) {
1502 if (argc < 2) return usage();
1503 server_socket_str = argv[1];
1504 argc--;
1505 argv++;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001506 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001507 /* out of recognized modifiers and flags */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001508 break;
1509 }
1510 argc--;
1511 argv++;
1512 }
1513
Josh Gao9c869b52016-08-25 16:00:22 -07001514 if ((server_host_str || server_port_str) && server_socket_str) {
1515 fprintf(stderr, "adb: -L is incompatible with -H or -P\n");
1516 exit(1);
1517 }
1518
1519 // If -L, -H, or -P are specified, ignore environment variables.
1520 // Otherwise, prefer ADB_SERVER_SOCKET over ANDROID_ADB_SERVER_ADDRESS/PORT.
Tao Wu135f4ab2016-09-16 13:01:24 -07001521 if (!server_host_str && !server_port_str && !server_socket_str) {
1522 server_socket_str = getenv("ADB_SERVER_SOCKET");
Josh Gao9c869b52016-08-25 16:00:22 -07001523 }
1524
1525 if (!server_socket_str) {
1526 // tcp:1234 and tcp:localhost:1234 are different with -a, so don't default to localhost
1527 server_host_str = server_host_str ? server_host_str : getenv("ANDROID_ADB_SERVER_ADDRESS");
1528
Tao Wu135f4ab2016-09-16 13:01:24 -07001529 int server_port = DEFAULT_ADB_PORT;
Josh Gao9c869b52016-08-25 16:00:22 -07001530 server_port_str = server_port_str ? server_port_str : getenv("ANDROID_ADB_SERVER_PORT");
Tao Wu135f4ab2016-09-16 13:01:24 -07001531 if (server_port_str && strlen(server_port_str) > 0) {
1532 if (!android::base::ParseInt(server_port_str, &server_port, 1, 65535)) {
1533 fprintf(stderr,
1534 "adb: Env var ANDROID_ADB_SERVER_PORT must be a positive"
1535 " number less than 65535. Got \"%s\"\n",
1536 server_port_str);
1537 exit(1);
1538 }
1539 }
Josh Gao9c869b52016-08-25 16:00:22 -07001540
1541 int rc;
1542 char* temp;
1543 if (server_host_str) {
Tao Wu135f4ab2016-09-16 13:01:24 -07001544 rc = asprintf(&temp, "tcp:%s:%d", server_host_str, server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001545 } else {
Tao Wu135f4ab2016-09-16 13:01:24 -07001546 rc = asprintf(&temp, "tcp:%d", server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001547 }
1548 if (rc < 0) {
1549 fatal("failed to allocate server socket specification");
1550 }
1551 server_socket_str = temp;
1552 }
1553
1554 adb_set_socket_spec(server_socket_str);
1555
Elliott Hughes8d28e192015-10-07 14:55:10 -07001556 // If none of -d, -e, or -s were specified, try $ANDROID_SERIAL.
1557 if (transport_type == kTransportAny && serial == nullptr) {
1558 serial = getenv("ANDROID_SERIAL");
1559 }
1560
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001561 adb_set_transport(transport_type, serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001562
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001563 if (is_server) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001564 if (no_daemon || is_daemon) {
Spencer Low5c398d22015-08-08 15:07:07 -07001565 if (is_daemon && (ack_reply_fd == -1)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001566 fprintf(stderr, "reply fd for adb server to client communication not specified.\n");
1567 return usage();
1568 }
Josh Gao9c869b52016-08-25 16:00:22 -07001569 r = adb_server_main(is_daemon, server_socket_str, ack_reply_fd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001570 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001571 r = launch_server(server_socket_str);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001572 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001573 if (r) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001574 fprintf(stderr,"* could not start server *\n");
1575 }
1576 return r;
1577 }
1578
Riley Andrews98f58e82014-12-05 17:37:24 -08001579 if (argc == 0) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001580 return usage();
1581 }
1582
Riley Andrewsc8514c82014-12-05 17:32:46 -08001583 /* handle wait-for-* prefix */
1584 if (!strncmp(argv[0], "wait-for-", strlen("wait-for-"))) {
Dan Albertbac34742015-02-25 17:51:28 -08001585 const char* service = argv[0];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001586
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001587 if (!wait_for_device(service, transport_type, serial)) {
Riley Andrewsc8514c82014-12-05 17:32:46 -08001588 return 1;
1589 }
1590
Elliott Hughes2b101112015-05-04 19:29:32 -07001591 // Allow a command to be run after wait-for-device,
1592 // e.g. 'adb wait-for-device shell'.
Riley Andrewsc8514c82014-12-05 17:32:46 -08001593 if (argc == 1) {
1594 return 0;
1595 }
1596
1597 /* Fall through */
1598 argc--;
1599 argv++;
1600 }
1601
1602 /* adb_connect() commands */
Riley Andrews98f58e82014-12-05 17:37:24 -08001603 if (!strcmp(argv[0], "devices")) {
Dan Albertbac34742015-02-25 17:51:28 -08001604 const char *listopt;
Elliott Hughes6452a892015-04-29 12:28:13 -07001605 if (argc < 2) {
Scott Andersone109d262012-04-20 11:21:14 -07001606 listopt = "";
Elliott Hughes6452a892015-04-29 12:28:13 -07001607 } else if (argc == 2 && !strcmp(argv[1], "-l")) {
Scott Andersone109d262012-04-20 11:21:14 -07001608 listopt = argv[1];
Elliott Hughes6452a892015-04-29 12:28:13 -07001609 } else {
Scott Andersone109d262012-04-20 11:21:14 -07001610 fprintf(stderr, "Usage: adb devices [-l]\n");
1611 return 1;
1612 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001613
1614 std::string query = android::base::StringPrintf("host:%s%s", argv[0], listopt);
1615 printf("List of devices attached\n");
1616 return adb_query_command(query);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001617 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001618 else if (!strcmp(argv[0], "connect")) {
Mike Lockwoodff196702009-08-24 15:58:40 -07001619 if (argc != 2) {
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001620 fprintf(stderr, "Usage: adb connect <host>[:<port>]\n");
Mike Lockwoodff196702009-08-24 15:58:40 -07001621 return 1;
1622 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001623
1624 std::string query = android::base::StringPrintf("host:connect:%s", argv[1]);
1625 return adb_query_command(query);
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001626 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001627 else if (!strcmp(argv[0], "disconnect")) {
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001628 if (argc > 2) {
1629 fprintf(stderr, "Usage: adb disconnect [<host>[:<port>]]\n");
1630 return 1;
1631 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001632
1633 std::string query = android::base::StringPrintf("host:disconnect:%s",
1634 (argc == 2) ? argv[1] : "");
1635 return adb_query_command(query);
Mike Lockwoodff196702009-08-24 15:58:40 -07001636 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001637 else if (!strcmp(argv[0], "emu")) {
Spencer Low142ec752015-05-06 16:13:42 -07001638 return adb_send_emulator_command(argc, argv, serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001639 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -08001640 else if (!strcmp(argv[0], "shell")) {
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001641 return adb_shell(argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001642 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001643 else if (!strcmp(argv[0], "exec-in") || !strcmp(argv[0], "exec-out")) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001644 int exec_in = !strcmp(argv[0], "exec-in");
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001645
Zach Riggleccafc792016-04-14 17:18:04 -04001646 if (argc < 2) {
1647 fprintf(stderr, "Usage: adb %s command\n", argv[0]);
1648 return 1;
1649 }
1650
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001651 std::string cmd = "exec:";
1652 cmd += argv[1];
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001653 argc -= 2;
1654 argv += 2;
1655 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001656 cmd += " " + escape_arg(*argv++);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001657 }
1658
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001659 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001660 int fd = adb_connect(cmd, &error);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001661 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001662 fprintf(stderr, "error: %s\n", error.c_str());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001663 return -1;
1664 }
1665
1666 if (exec_in) {
1667 copy_to_file(STDIN_FILENO, fd);
1668 } else {
1669 copy_to_file(fd, STDOUT_FILENO);
1670 }
1671
1672 adb_close(fd);
1673 return 0;
1674 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001675 else if (!strcmp(argv[0], "kill-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001676 std::string error;
1677 int fd = _adb_connect("host:kill", &error);
Spencer Lowf18fc082015-08-11 17:05:02 -07001678 if (fd == -2) {
1679 // Failed to make network connection to server. Don't output the
1680 // network error since that is expected.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001681 fprintf(stderr,"* server not running *\n");
Spencer Lowf18fc082015-08-11 17:05:02 -07001682 // Successful exit code because the server is already "killed".
1683 return 0;
1684 } else if (fd == -1) {
1685 // Some other error.
1686 fprintf(stderr, "error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001687 return 1;
Spencer Lowf18fc082015-08-11 17:05:02 -07001688 } else {
1689 // Successfully connected, kill command sent, okay status came back.
1690 // Server should exit() in a moment, if not already.
Spencer Low351ecd12015-10-14 17:32:44 -07001691 ReadOrderlyShutdown(fd);
Spencer Lowf18fc082015-08-11 17:05:02 -07001692 adb_close(fd);
1693 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001694 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001695 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001696 else if (!strcmp(argv[0], "sideload")) {
1697 if (argc != 2) return usage();
Doug Zongker71fe5842014-06-26 15:35:36 -07001698 if (adb_sideload_host(argv[1])) {
Doug Zongker447f0612012-01-09 14:54:53 -08001699 return 1;
1700 } else {
1701 return 0;
1702 }
1703 }
Elliott Hughes19d80b82015-07-21 16:13:40 -07001704 else if (!strcmp(argv[0], "tcpip") && argc > 1) {
1705 return adb_connect_command(android::base::StringPrintf("tcpip:%s", argv[1]));
1706 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001707 else if (!strcmp(argv[0], "remount") ||
1708 !strcmp(argv[0], "reboot") ||
1709 !strcmp(argv[0], "reboot-bootloader") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001710 !strcmp(argv[0], "usb") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001711 !strcmp(argv[0], "disable-verity") ||
1712 !strcmp(argv[0], "enable-verity")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001713 std::string command;
Tao Bao175b7bb2015-03-29 11:22:34 -07001714 if (!strcmp(argv[0], "reboot-bootloader")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001715 command = "reboot:bootloader";
Tao Bao175b7bb2015-03-29 11:22:34 -07001716 } else if (argc > 1) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001717 command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
Tao Bao175b7bb2015-03-29 11:22:34 -07001718 } else {
Elliott Hughes5677c232015-05-07 23:37:40 -07001719 command = android::base::StringPrintf("%s:", argv[0]);
The Android Open Source Projecte037fd72009-03-13 13:04:37 -07001720 }
Tao Bao175b7bb2015-03-29 11:22:34 -07001721 return adb_connect_command(command);
Josh Gao06c73ae2016-03-04 15:16:18 -08001722 } else if (!strcmp(argv[0], "root") || !strcmp(argv[0], "unroot")) {
1723 return adb_root(argv[0]) ? 0 : 1;
1724 } else if (!strcmp(argv[0], "bugreport")) {
Felipe Leme698e0652016-07-19 17:07:22 -07001725 Bugreport bugreport;
1726 return bugreport.DoIt(transport_type, serial, argc, argv);
Felipe Leme44a42672016-04-01 17:43:27 -07001727 } else if (!strcmp(argv[0], "forward") || !strcmp(argv[0], "reverse")) {
Elliott Hughes424af022015-05-29 17:55:19 -07001728 bool reverse = !strcmp(argv[0], "reverse");
1729 ++argv;
1730 --argc;
1731 if (argc < 1) return usage();
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001732
1733 // Determine the <host-prefix> for this command.
Elliott Hughes424af022015-05-29 17:55:19 -07001734 std::string host_prefix;
David 'Digit' Turner25258692013-03-21 21:07:42 +01001735 if (reverse) {
Elliott Hughes424af022015-05-29 17:55:19 -07001736 host_prefix = "reverse";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001737 } else {
David 'Digit' Turner25258692013-03-21 21:07:42 +01001738 if (serial) {
Elliott Hughes424af022015-05-29 17:55:19 -07001739 host_prefix = android::base::StringPrintf("host-serial:%s", serial);
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001740 } else if (transport_type == kTransportUsb) {
Elliott Hughes424af022015-05-29 17:55:19 -07001741 host_prefix = "host-usb";
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001742 } else if (transport_type == kTransportLocal) {
Elliott Hughes424af022015-05-29 17:55:19 -07001743 host_prefix = "host-local";
David 'Digit' Turner25258692013-03-21 21:07:42 +01001744 } else {
Elliott Hughes424af022015-05-29 17:55:19 -07001745 host_prefix = "host";
David 'Digit' Turner25258692013-03-21 21:07:42 +01001746 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001747 }
1748
David Purselleaae97e2016-04-07 11:25:48 -07001749 std::string cmd, error;
Elliott Hughes424af022015-05-29 17:55:19 -07001750 if (strcmp(argv[0], "--list") == 0) {
Elliott Hughesab52c182015-05-01 17:04:38 -07001751 if (argc != 1) return usage();
Elliott Hughes424af022015-05-29 17:55:19 -07001752 return adb_query_command(host_prefix + ":list-forward");
1753 } else if (strcmp(argv[0], "--remove-all") == 0) {
1754 if (argc != 1) return usage();
1755 cmd = host_prefix + ":killforward-all";
1756 } else if (strcmp(argv[0], "--remove") == 0) {
1757 // forward --remove <local>
Elliott Hughesab52c182015-05-01 17:04:38 -07001758 if (argc != 2) return usage();
Elliott Hughes424af022015-05-29 17:55:19 -07001759 cmd = host_prefix + ":killforward:" + argv[1];
1760 } else if (strcmp(argv[0], "--no-rebind") == 0) {
1761 // forward --no-rebind <local> <remote>
Elliott Hughesab52c182015-05-01 17:04:38 -07001762 if (argc != 3) return usage();
David Purselleaae97e2016-04-07 11:25:48 -07001763 if (forward_targets_are_valid(argv[1], argv[2], &error)) {
1764 cmd = host_prefix + ":forward:norebind:" + argv[1] + ";" + argv[2];
1765 }
Elliott Hughes424af022015-05-29 17:55:19 -07001766 } else {
1767 // forward <local> <remote>
1768 if (argc != 2) return usage();
David Purselleaae97e2016-04-07 11:25:48 -07001769 if (forward_targets_are_valid(argv[0], argv[1], &error)) {
1770 cmd = host_prefix + ":forward:" + argv[0] + ";" + argv[1];
1771 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001772 }
1773
David Purselleaae97e2016-04-07 11:25:48 -07001774 if (!error.empty()) {
1775 fprintf(stderr, "error: %s\n", error.c_str());
1776 return 1;
1777 }
1778
1779 int fd = adb_connect(cmd, &error);
1780 if (fd < 0 || !adb_status(fd, &error)) {
1781 adb_close(fd);
1782 fprintf(stderr, "error: %s\n", error.c_str());
1783 return 1;
1784 }
1785
1786 // Server or device may optionally return a resolved TCP port number.
1787 std::string resolved_port;
1788 if (ReadProtocolString(fd, &resolved_port, &error) && !resolved_port.empty()) {
1789 printf("%s\n", resolved_port.c_str());
1790 }
1791
1792 ReadOrderlyShutdown(fd);
1793 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001794 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001795 /* do_sync_*() commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001796 else if (!strcmp(argv[0], "ls")) {
1797 if (argc != 2) return usage();
Elliott Hughesaa245492015-08-03 10:38:08 -07001798 return do_sync_ls(argv[1]) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001799 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001800 else if (!strcmp(argv[0], "push")) {
Josh Gao05786772015-10-30 16:57:19 -07001801 bool copy_attrs = false;
1802 std::vector<const char*> srcs;
1803 const char* dst = nullptr;
Mark Lindner76f2a932014-03-11 17:55:59 -07001804
Josh Gao05786772015-10-30 16:57:19 -07001805 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs);
1806 if (srcs.empty() || !dst) return usage();
1807 return do_sync_push(srcs, dst) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001808 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001809 else if (!strcmp(argv[0], "pull")) {
Josh Gao05786772015-10-30 16:57:19 -07001810 bool copy_attrs = false;
1811 std::vector<const char*> srcs;
1812 const char* dst = ".";
Mark Lindner76f2a932014-03-11 17:55:59 -07001813
Josh Gao05786772015-10-30 16:57:19 -07001814 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs);
1815 if (srcs.empty()) return usage();
1816 return do_sync_pull(srcs, dst, copy_attrs) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001817 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001818 else if (!strcmp(argv[0], "install")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001819 if (argc < 2) return usage();
Todd Kennedy248722e2016-07-20 16:22:37 -07001820 if (_use_legacy_install()) {
1821 return install_app_legacy(transport_type, serial, argc, argv);
David Pursell0aacbbe2016-01-21 19:56:50 -08001822 }
Todd Kennedy248722e2016-07-20 16:22:37 -07001823 return install_app(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001824 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001825 else if (!strcmp(argv[0], "install-multiple")) {
Jeff Sharkey960df972014-06-09 17:30:57 -07001826 if (argc < 2) return usage();
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001827 return install_multiple_app(transport_type, serial, argc, argv);
Jeff Sharkey960df972014-06-09 17:30:57 -07001828 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001829 else if (!strcmp(argv[0], "uninstall")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001830 if (argc < 2) return usage();
Todd Kennedy248722e2016-07-20 16:22:37 -07001831 if (_use_legacy_install()) {
1832 return uninstall_app_legacy(transport_type, serial, argc, argv);
David Pursell0aacbbe2016-01-21 19:56:50 -08001833 }
Todd Kennedy248722e2016-07-20 16:22:37 -07001834 return uninstall_app(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001835 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001836 else if (!strcmp(argv[0], "sync")) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001837 std::string src;
Elliott Hughes58305772015-04-17 13:57:15 -07001838 bool list_only = false;
Riley Andrews98f58e82014-12-05 17:37:24 -08001839 if (argc < 2) {
Elliott Hughes58305772015-04-17 13:57:15 -07001840 // No local path was specified.
Elliott Hughesd236d072015-04-21 10:17:07 -07001841 src = "";
Anthony Newnam705c9442010-02-22 08:36:49 -06001842 } else if (argc >= 2 && strcmp(argv[1], "-l") == 0) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001843 list_only = true;
Anthony Newnam705c9442010-02-22 08:36:49 -06001844 if (argc == 3) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001845 src = argv[2];
Anthony Newnam705c9442010-02-22 08:36:49 -06001846 } else {
Elliott Hughesd236d072015-04-21 10:17:07 -07001847 src = "";
Anthony Newnam705c9442010-02-22 08:36:49 -06001848 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001849 } else if (argc == 2) {
Elliott Hughes58305772015-04-17 13:57:15 -07001850 // A local path or "android"/"data" arg was specified.
Elliott Hughesd236d072015-04-21 10:17:07 -07001851 src = argv[1];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001852 } else {
1853 return usage();
1854 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001855
Elliott Hughesd236d072015-04-21 10:17:07 -07001856 if (src != "" &&
1857 src != "system" && src != "data" && src != "vendor" && src != "oem") {
Elliott Hughes58305772015-04-17 13:57:15 -07001858 return usage();
1859 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001860
Elliott Hughes58305772015-04-17 13:57:15 -07001861 std::string system_src_path = product_file("system");
1862 std::string data_src_path = product_file("data");
1863 std::string vendor_src_path = product_file("vendor");
1864 std::string oem_src_path = product_file("oem");
Elliott Hughes58305772015-04-17 13:57:15 -07001865
Elliott Hughesaa245492015-08-03 10:38:08 -07001866 bool okay = true;
1867 if (okay && (src.empty() || src == "system")) {
1868 okay = do_sync_sync(system_src_path, "/system", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001869 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001870 if (okay && (src.empty() || src == "vendor") && directory_exists(vendor_src_path)) {
1871 okay = do_sync_sync(vendor_src_path, "/vendor", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001872 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001873 if (okay && (src.empty() || src == "oem") && directory_exists(oem_src_path)) {
1874 okay = do_sync_sync(oem_src_path, "/oem", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001875 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001876 if (okay && (src.empty() || src == "data")) {
1877 okay = do_sync_sync(data_src_path, "/data", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001878 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001879 return okay ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001880 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001881 /* passthrough commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001882 else if (!strcmp(argv[0],"get-state") ||
Scott Andersone109d262012-04-20 11:21:14 -07001883 !strcmp(argv[0],"get-serialno") ||
1884 !strcmp(argv[0],"get-devpath"))
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001885 {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001886 return adb_query_command(format_host_command(argv[0], transport_type, serial));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001887 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001888 /* other commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001889 else if (!strcmp(argv[0],"logcat") || !strcmp(argv[0],"lolcat") || !strcmp(argv[0],"longcat")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001890 return logcat(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001891 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001892 else if (!strcmp(argv[0],"ppp")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001893 return ppp(argc, argv);
1894 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001895 else if (!strcmp(argv[0], "start-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001896 std::string error;
Spencer Lowf18fc082015-08-11 17:05:02 -07001897 const int result = adb_connect("host:start-server", &error);
1898 if (result < 0) {
1899 fprintf(stderr, "error: %s\n", error.c_str());
1900 }
1901 return result;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001902 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001903 else if (!strcmp(argv[0], "backup")) {
Christopher Tated2f54152011-04-21 12:53:28 -07001904 return backup(argc, argv);
1905 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001906 else if (!strcmp(argv[0], "restore")) {
Christopher Tate702967a2011-05-17 15:52:54 -07001907 return restore(argc, argv);
1908 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001909 else if (!strcmp(argv[0], "keygen")) {
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08001910 if (argc < 2) return usage();
Yabin Cuiaed3c612015-09-22 15:52:57 -07001911 // Always print key generation information for keygen command.
1912 adb_trace_enable(AUTH);
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08001913 return adb_auth_keygen(argv[1]);
1914 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001915 else if (!strcmp(argv[0], "jdwp")) {
Tao Bao175b7bb2015-03-29 11:22:34 -07001916 return adb_connect_command("jdwp");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001917 }
Josh Gaob463baf2016-05-13 18:18:23 -07001918 else if (!strcmp(argv[0], "track-jdwp")) {
1919 return adb_connect_command("track-jdwp");
1920 }
1921 else if (!strcmp(argv[0], "track-devices")) {
1922 return adb_connect_command("host:track-devices");
1923 }
1924
1925
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001926 /* "adb /?" is a common idiom under Windows */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001927 else if (!strcmp(argv[0], "help") || !strcmp(argv[0], "/?")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001928 help();
1929 return 0;
1930 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001931 else if (!strcmp(argv[0], "version")) {
Elliott Hughes42ae2602015-08-12 08:32:10 -07001932 fprintf(stdout, "%s", adb_version().c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001933 return 0;
1934 }
Dan Albert90d4b732015-05-20 18:58:41 -07001935 else if (!strcmp(argv[0], "features")) {
David Pursell4e2fd362015-09-22 10:43:08 -07001936 // Only list the features common to both the adb client and the device.
David Pursell0aacbbe2016-01-21 19:56:50 -08001937 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001938 std::string error;
1939 if (!adb_get_feature_set(&features, &error)) {
1940 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08001941 return 1;
1942 }
1943
David Pursell4e2fd362015-09-22 10:43:08 -07001944 for (const std::string& name : features) {
David Pursell70ef7b42015-09-30 13:35:42 -07001945 if (CanUseFeature(features, name)) {
David Pursell4e2fd362015-09-22 10:43:08 -07001946 printf("%s\n", name.c_str());
1947 }
1948 }
1949 return 0;
Yabin Cui1f4ec192016-04-05 13:50:44 -07001950 } else if (!strcmp(argv[0], "reconnect")) {
1951 if (argc == 1) {
1952 return adb_query_command("host:reconnect");
1953 } else if (argc == 2 && !strcmp(argv[1], "device")) {
1954 std::string err;
1955 adb_connect("reconnect", &err);
1956 return 0;
1957 }
Dan Albert90d4b732015-05-20 18:58:41 -07001958 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001959
1960 usage();
1961 return 1;
1962}
1963
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001964static int uninstall_app(TransportType transport, const char* serial, int argc, const char** argv) {
1965 // 'adb uninstall' takes the same arguments as 'cmd package uninstall' on device
1966 std::string cmd = "cmd package";
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001967 while (argc-- > 0) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001968 // deny the '-k' option until the remaining data/cache can be removed with adb/UI
1969 if (strcmp(*argv, "-k") == 0) {
1970 printf(
1971 "The -k option uninstalls the application while retaining the data/cache.\n"
1972 "At the moment, there is no way to remove the remaining data.\n"
1973 "You will have to reinstall the application with the same signature, and fully uninstall it.\n"
1974 "If you truly wish to continue, execute 'adb shell cmd package uninstall -k'.\n");
1975 return EXIT_FAILURE;
1976 }
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001977 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001978 }
1979
David Pursell70ef7b42015-09-30 13:35:42 -07001980 return send_shell_command(transport, serial, cmd, false);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001981}
1982
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001983static int install_app(TransportType transport, const char* serial, int argc, const char** argv) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001984 // The last argument must be the APK file
1985 const char* file = argv[argc - 1];
1986 const char* dot = strrchr(file, '.');
1987 bool found_apk = false;
Jeff Sharkey960df972014-06-09 17:30:57 -07001988 struct stat sb;
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001989 if (dot && !strcasecmp(dot, ".apk")) {
1990 if (stat(file, &sb) == -1 || !S_ISREG(sb.st_mode)) {
1991 fprintf(stderr, "Invalid APK file: %s\n", file);
1992 return EXIT_FAILURE;
Kenny Root597ea5b2011-08-05 11:19:45 -07001993 }
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001994 found_apk = true;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001995 }
1996
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001997 if (!found_apk) {
Jeff Sharkey960df972014-06-09 17:30:57 -07001998 fprintf(stderr, "Missing APK file\n");
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001999 return EXIT_FAILURE;
Kenny Root597ea5b2011-08-05 11:19:45 -07002000 }
2001
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002002 int localFd = adb_open(file, O_RDONLY);
2003 if (localFd < 0) {
2004 fprintf(stderr, "Failed to open %s: %s\n", file, strerror(errno));
2005 return 1;
2006 }
Kenny Root597ea5b2011-08-05 11:19:45 -07002007
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002008 std::string error;
2009 std::string cmd = "exec:cmd package";
2010
2011 // don't copy the APK name, but, copy the rest of the arguments as-is
2012 while (argc-- > 1) {
2013 cmd += " " + escape_arg(std::string(*argv++));
2014 }
2015
2016 // add size parameter [required for streaming installs]
2017 // do last to override any user specified value
2018 cmd += " " + android::base::StringPrintf("-S %" PRIu64, static_cast<uint64_t>(sb.st_size));
2019
2020 int remoteFd = adb_connect(cmd, &error);
2021 if (remoteFd < 0) {
2022 fprintf(stderr, "Connect error for write: %s\n", error.c_str());
2023 adb_close(localFd);
2024 return 1;
2025 }
2026
2027 char buf[BUFSIZ];
2028 copy_to_file(localFd, remoteFd);
2029 read_status_line(remoteFd, buf, sizeof(buf));
2030
2031 adb_close(localFd);
2032 adb_close(remoteFd);
2033
2034 if (strncmp("Success", buf, 7)) {
Elliott Hughesfe018852015-12-17 14:04:38 -08002035 fprintf(stderr, "Failed to install %s: %s", file, buf);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002036 return 1;
2037 }
2038 fputs(buf, stderr);
2039 return 0;
Jeff Sharkey960df972014-06-09 17:30:57 -07002040}
2041
Elliott Hughes3bd73c12015-05-05 13:10:43 -07002042static int install_multiple_app(TransportType transport, const char* serial, int argc,
Elliott Hughes58305772015-04-17 13:57:15 -07002043 const char** argv)
Jeff Sharkey960df972014-06-09 17:30:57 -07002044{
Jeff Sharkey960df972014-06-09 17:30:57 -07002045 int i;
2046 struct stat sb;
Elliott Hughes2940ccf2015-04-17 14:07:52 -07002047 uint64_t total_size = 0;
Jeff Sharkey960df972014-06-09 17:30:57 -07002048 // Find all APK arguments starting at end.
2049 // All other arguments passed through verbatim.
2050 int first_apk = -1;
2051 for (i = argc - 1; i >= 0; i--) {
Dan Albertbac34742015-02-25 17:51:28 -08002052 const char* file = argv[i];
Elliott Hughes3e7048c2015-07-27 21:21:39 -07002053 const char* dot = strrchr(file, '.');
Jeff Sharkey960df972014-06-09 17:30:57 -07002054 if (dot && !strcasecmp(dot, ".apk")) {
2055 if (stat(file, &sb) == -1 || !S_ISREG(sb.st_mode)) {
2056 fprintf(stderr, "Invalid APK file: %s\n", file);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002057 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002058 }
2059
2060 total_size += sb.st_size;
2061 first_apk = i;
2062 } else {
2063 break;
2064 }
Kenny Root597ea5b2011-08-05 11:19:45 -07002065 }
2066
Jeff Sharkey960df972014-06-09 17:30:57 -07002067 if (first_apk == -1) {
2068 fprintf(stderr, "Missing APK file\n");
2069 return 1;
2070 }
Kenny Root597ea5b2011-08-05 11:19:45 -07002071
Todd Kennedy248722e2016-07-20 16:22:37 -07002072 std::string install_cmd;
2073 if (_use_legacy_install()) {
2074 install_cmd = "exec:pm";
2075 } else {
2076 install_cmd = "exec:cmd package";
2077 }
2078
2079 std::string cmd = android::base::StringPrintf("%s install-create -S %" PRIu64, install_cmd.c_str(), total_size);
Jeff Sharkey960df972014-06-09 17:30:57 -07002080 for (i = 1; i < first_apk; i++) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07002081 cmd += " " + escape_arg(argv[i]);
Jeff Sharkey960df972014-06-09 17:30:57 -07002082 }
2083
2084 // Create install session
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002085 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07002086 int fd = adb_connect(cmd, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07002087 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002088 fprintf(stderr, "Connect error for create: %s\n", error.c_str());
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002089 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002090 }
Elliott Hughes2baae3a2015-04-17 10:59:34 -07002091 char buf[BUFSIZ];
Jeff Sharkey960df972014-06-09 17:30:57 -07002092 read_status_line(fd, buf, sizeof(buf));
2093 adb_close(fd);
2094
2095 int session_id = -1;
2096 if (!strncmp("Success", buf, 7)) {
2097 char* start = strrchr(buf, '[');
2098 char* end = strrchr(buf, ']');
2099 if (start && end) {
2100 *end = '\0';
2101 session_id = strtol(start + 1, NULL, 10);
2102 }
2103 }
2104 if (session_id < 0) {
2105 fprintf(stderr, "Failed to create session\n");
Christopher Tate71bbc672014-07-14 16:45:13 -07002106 fputs(buf, stderr);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002107 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002108 }
2109
2110 // Valid session, now stream the APKs
2111 int success = 1;
2112 for (i = first_apk; i < argc; i++) {
Dan Albertbac34742015-02-25 17:51:28 -08002113 const char* file = argv[i];
Jeff Sharkey960df972014-06-09 17:30:57 -07002114 if (stat(file, &sb) == -1) {
2115 fprintf(stderr, "Failed to stat %s\n", file);
2116 success = 0;
2117 goto finalize_session;
2118 }
2119
Elliott Hughes2940ccf2015-04-17 14:07:52 -07002120 std::string cmd = android::base::StringPrintf(
Todd Kennedy248722e2016-07-20 16:22:37 -07002121 "%s install-write -S %" PRIu64 " %d %d_%s -",
2122 install_cmd.c_str(), static_cast<uint64_t>(sb.st_size), session_id, i, adb_basename(file).c_str());
Jeff Sharkey960df972014-06-09 17:30:57 -07002123
2124 int localFd = adb_open(file, O_RDONLY);
2125 if (localFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002126 fprintf(stderr, "Failed to open %s: %s\n", file, strerror(errno));
Jeff Sharkey960df972014-06-09 17:30:57 -07002127 success = 0;
2128 goto finalize_session;
2129 }
2130
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002131 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07002132 int remoteFd = adb_connect(cmd, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07002133 if (remoteFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002134 fprintf(stderr, "Connect error for write: %s\n", error.c_str());
Jeff Sharkey960df972014-06-09 17:30:57 -07002135 adb_close(localFd);
2136 success = 0;
2137 goto finalize_session;
2138 }
2139
2140 copy_to_file(localFd, remoteFd);
2141 read_status_line(remoteFd, buf, sizeof(buf));
2142
2143 adb_close(localFd);
2144 adb_close(remoteFd);
2145
2146 if (strncmp("Success", buf, 7)) {
2147 fprintf(stderr, "Failed to write %s\n", file);
Christopher Tate71bbc672014-07-14 16:45:13 -07002148 fputs(buf, stderr);
Jeff Sharkey960df972014-06-09 17:30:57 -07002149 success = 0;
2150 goto finalize_session;
2151 }
2152 }
2153
2154finalize_session:
Jeff Sharkeyac77e1f2014-07-25 09:58:25 -07002155 // Commit session if we streamed everything okay; otherwise abandon
Elliott Hughes6452a892015-04-29 12:28:13 -07002156 std::string service =
Todd Kennedy248722e2016-07-20 16:22:37 -07002157 android::base::StringPrintf("%s install-%s %d",
2158 install_cmd.c_str(), success ? "commit" : "abandon", session_id);
Elliott Hughes6452a892015-04-29 12:28:13 -07002159 fd = adb_connect(service, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07002160 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002161 fprintf(stderr, "Connect error for finalize: %s\n", error.c_str());
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002162 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002163 }
2164 read_status_line(fd, buf, sizeof(buf));
2165 adb_close(fd);
2166
2167 if (!strncmp("Success", buf, 7)) {
Christopher Tate71bbc672014-07-14 16:45:13 -07002168 fputs(buf, stderr);
Jeff Sharkey960df972014-06-09 17:30:57 -07002169 return 0;
2170 } else {
2171 fprintf(stderr, "Failed to finalize session\n");
Christopher Tate71bbc672014-07-14 16:45:13 -07002172 fputs(buf, stderr);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002173 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002174 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002175}
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002176
2177static int pm_command(TransportType transport, const char* serial, int argc, const char** argv) {
2178 std::string cmd = "pm";
2179
2180 while (argc-- > 0) {
2181 cmd += " " + escape_arg(*argv++);
2182 }
2183
2184 return send_shell_command(transport, serial, cmd, false);
2185}
2186
2187static int uninstall_app_legacy(TransportType transport, const char* serial, int argc, const char** argv) {
2188 /* if the user choose the -k option, we refuse to do it until devices are
2189 out with the option to uninstall the remaining data somehow (adb/ui) */
2190 int i;
2191 for (i = 1; i < argc; i++) {
2192 if (!strcmp(argv[i], "-k")) {
2193 printf(
2194 "The -k option uninstalls the application while retaining the data/cache.\n"
2195 "At the moment, there is no way to remove the remaining data.\n"
2196 "You will have to reinstall the application with the same signature, and fully uninstall it.\n"
2197 "If you truly wish to continue, execute 'adb shell pm uninstall -k'\n.");
2198 return EXIT_FAILURE;
2199 }
2200 }
2201
2202 /* 'adb uninstall' takes the same arguments as 'pm uninstall' on device */
2203 return pm_command(transport, serial, argc, argv);
2204}
2205
2206static int delete_file(TransportType transport, const char* serial, const std::string& filename) {
2207 std::string cmd = "rm -f " + escape_arg(filename);
2208 return send_shell_command(transport, serial, cmd, false);
2209}
2210
2211static int install_app_legacy(TransportType transport, const char* serial, int argc, const char** argv) {
2212 static const char *const DATA_DEST = "/data/local/tmp/%s";
2213 static const char *const SD_DEST = "/sdcard/tmp/%s";
2214 const char* where = DATA_DEST;
2215 int i;
2216 struct stat sb;
2217
2218 for (i = 1; i < argc; i++) {
2219 if (!strcmp(argv[i], "-s")) {
2220 where = SD_DEST;
2221 }
2222 }
2223
2224 // Find last APK argument.
2225 // All other arguments passed through verbatim.
2226 int last_apk = -1;
2227 for (i = argc - 1; i >= 0; i--) {
2228 const char* file = argv[i];
2229 const char* dot = strrchr(file, '.');
2230 if (dot && !strcasecmp(dot, ".apk")) {
2231 if (stat(file, &sb) == -1 || !S_ISREG(sb.st_mode)) {
2232 fprintf(stderr, "Invalid APK file: %s\n", file);
2233 return EXIT_FAILURE;
2234 }
2235
2236 last_apk = i;
2237 break;
2238 }
2239 }
2240
2241 if (last_apk == -1) {
2242 fprintf(stderr, "Missing APK file\n");
2243 return EXIT_FAILURE;
2244 }
2245
2246 int result = -1;
2247 std::vector<const char*> apk_file = {argv[last_apk]};
2248 std::string apk_dest = android::base::StringPrintf(
2249 where, adb_basename(argv[last_apk]).c_str());
2250 if (!do_sync_push(apk_file, apk_dest.c_str())) goto cleanup_apk;
2251 argv[last_apk] = apk_dest.c_str(); /* destination name, not source location */
2252 result = pm_command(transport, serial, argc, argv);
2253
2254cleanup_apk:
2255 delete_file(transport, serial, apk_dest);
2256 return result;
2257}