blob: 68ae4affe39a0640ab1e8cc82e71c943db1cdab6 [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>
Elliott Hughesdbe91ee2016-11-15 12:37:32 -080036#include <thread>
Josh Gao05786772015-10-30 16:57:19 -070037#include <vector>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070038
Elliott Hughese8b663f2016-05-26 22:43:19 -070039#include <android-base/file.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080040#include <android-base/logging.h>
Tao Wu135f4ab2016-09-16 13:01:24 -070041#include <android-base/parseint.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080042#include <android-base/stringprintf.h>
43#include <android-base/strings.h>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070044
Yabin Cuid325e862014-11-17 14:48:25 -080045#if !defined(_WIN32)
Josh Gao8dcdb572015-10-23 15:03:31 -070046#include <signal.h>
Elliott Hughes0fbf9612015-11-04 13:07:47 -080047#include <sys/ioctl.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080048#include <termios.h>
Dan Albert76649012015-02-24 15:51:19 -080049#include <unistd.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080050#endif
51
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080052#include "adb.h"
Nick Kralevichbea3f9c2014-11-13 15:17:29 -080053#include "adb_auth.h"
Dan Albertcc731cc2015-02-24 21:26:58 -080054#include "adb_client.h"
55#include "adb_io.h"
Josh Gao924d35a2016-08-30 15:39:25 -070056#include "adb_unique_fd.h"
Elliott Hughes58305772015-04-17 13:57:15 -070057#include "adb_utils.h"
Felipe Leme698e0652016-07-19 17:07:22 -070058#include "bugreport.h"
59#include "commandline.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080060#include "file_sync_service.h"
David Pursell70ef7b42015-09-30 13:35:42 -070061#include "services.h"
David Pursell606835a2015-09-08 17:17:02 -070062#include "shell_service.h"
Josh Gao4602adb2016-11-15 18:55:47 -080063#include "sysdeps/chrono.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080064
Elliott Hughes3bd73c12015-05-05 13:10:43 -070065static int install_app(TransportType t, const char* serial, int argc, const char** argv);
66static int install_multiple_app(TransportType t, const char* serial, int argc, const char** argv);
67static int uninstall_app(TransportType t, const char* serial, int argc, const char** argv);
Todd Kennedy6fa848a2015-11-03 16:53:08 -080068static int install_app_legacy(TransportType t, const char* serial, int argc, const char** argv);
69static int uninstall_app_legacy(TransportType t, const char* serial, int argc, const char** argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080070
Matt Gumbeld7b33082012-11-14 10:16:17 -080071extern int gListenAll;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080072
Felipe Leme0d4f0502016-07-26 12:14:39 -070073DefaultStandardStreamsCallback DEFAULT_STANDARD_STREAMS_CALLBACK(nullptr, nullptr);
74
Elliott Hughes0754cb92017-05-01 11:04:31 -070075static std::string product_file(const char* file) {
76 const char* ANDROID_PRODUCT_OUT = getenv("ANDROID_PRODUCT_OUT");
77 if (ANDROID_PRODUCT_OUT == nullptr) {
78 fprintf(stderr, "adb: product directory not specified; set $ANDROID_PRODUCT_OUT\n");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080079 exit(1);
80 }
Elliott Hughes0754cb92017-05-01 11:04:31 -070081 return android::base::StringPrintf("%s%s%s", ANDROID_PRODUCT_OUT, OS_PATH_SEPARATOR_STR, file);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080082}
83
Elliott Hughes58305772015-04-17 13:57:15 -070084static void help() {
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070085 fprintf(stdout, "%s\n", adb_version().c_str());
Felipe Lemeb3239722016-07-29 17:57:00 -070086 // clang-format off
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070087 fprintf(stdout,
88 "global options:\n"
89 " -a listen on all network interfaces, not just localhost\n"
90 " -d use USB device (error if multiple devices connected)\n"
Elliott Hughes83ab5c22017-01-13 16:58:25 -080091 " -e use TCP/IP device (error if multiple TCP/IP devices available)\n"
Elliott Hughes0754cb92017-05-01 11:04:31 -070092 " -s SERIAL use device with given serial (overrides $ANDROID_SERIAL)\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070093 " -H name of adb server host [default=localhost]\n"
94 " -P port of adb server [default=5037]\n"
95 " -L SOCKET listen on given socket for adb server [default=tcp:localhost:5037]\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080096 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070097 "general commands:\n"
98 " devices [-l] list connected devices (-l for long output)\n"
99 " help show this help message\n"
100 " version show version num\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700101 "\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800102 "networking:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700103 " connect HOST[:PORT] connect to a device via TCP/IP [default port=5555]\n"
104 " disconnect [HOST[:PORT]]\n"
105 " disconnect from given TCP/IP device [default port=5555], or all\n"
106 " forward --list list all forward socket connections\n"
107 " forward [--no-rebind] LOCAL REMOTE\n"
108 " forward socket connection using:\n"
109 " tcp:<port> (<local> may be \"tcp:0\" to pick any open port)\n"
110 " localabstract:<unix domain socket name>\n"
111 " localreserved:<unix domain socket name>\n"
112 " localfilesystem:<unix domain socket name>\n"
113 " dev:<character device name>\n"
114 " jdwp:<process pid> (remote only)\n"
115 " forward --remove LOCAL remove specific forward socket connection\n"
116 " forward --remove-all remove all forward socket connections\n"
117 " ppp TTY [PARAMETER...] run PPP over USB\n"
118 " reverse --list list all reverse socket connections from device\n"
119 " reverse [--no-rebind] REMOTE LOCAL\n"
120 " reverse socket connection using:\n"
121 " tcp:<port> (<remote> may be \"tcp:0\" to pick any open port)\n"
122 " localabstract:<unix domain socket name>\n"
123 " localreserved:<unix domain socket name>\n"
124 " localfilesystem:<unix domain socket name>\n"
125 " reverse --remove REMOTE remove specific reverse socket connection\n"
126 " reverse --remove-all remove all reverse socket connections from device\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800127 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700128 "file transfer:\n"
Dan Albert06b0d6b2017-05-18 22:56:48 -0700129 " push [--sync] LOCAL... REMOTE\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700130 " copy local files/directories to device\n"
Dan Albert06b0d6b2017-05-18 22:56:48 -0700131 " --sync: only push files that are newer on the host than the device\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700132 " pull [-a] REMOTE... LOCAL\n"
133 " copy files/dirs from device\n"
134 " -a: preserve file timestamp and mode\n"
Elliott Hughes0754cb92017-05-01 11:04:31 -0700135 " sync [system|vendor|oem|data|all]\n"
136 " sync a local build from $ANDROID_PRODUCT_OUT to the device (default all)\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700137 " -l: list but don't copy\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800138 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700139 "shell:\n"
140 " shell [-e ESCAPE] [-n] [-Tt] [-x] [COMMAND...]\n"
141 " run remote shell command (interactive shell if no command given)\n"
142 " -e: choose escape character, or \"none\"; default '~'\n"
143 " -n: don't read from stdin\n"
144 " -T: disable PTY allocation\n"
145 " -t: force PTY allocation\n"
146 " -x: disable remote exit codes and stdout/stderr separation\n"
147 " emu COMMAND run emulator console command\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800148 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700149 "app installation:\n"
150 " install [-lrtsdg] PACKAGE\n"
151 " install-multiple [-lrtsdpg] PACKAGE...\n"
152 " push package(s) to the device and install them\n"
153 " -l: forward lock application\n"
154 " -r: replace existing application\n"
155 " -t: allow test packages\n"
156 " -s: install application on sdcard\n"
157 " -d: allow version code downgrade (debuggable packages only)\n"
158 " -p: partial application install (install-multiple only)\n"
159 " -g: grant all runtime permissions\n"
160 " uninstall [-k] PACKAGE\n"
161 " remove this app package from the device\n"
162 " '-k': keep the data and cache directories\n"
163 "\n"
164 "backup/restore:\n"
Johan Toras Halsetheca1b3b2017-04-11 19:15:17 +0100165 " to show usage run \"adb shell bu help\"\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700166 "\n"
167 "debugging:\n"
168 " bugreport [PATH]\n"
169 " write bugreport to given PATH [default=bugreport.zip];\n"
170 " if PATH is a directory, the bug report is saved in that directory.\n"
171 " devices that don't support zipped bug reports output to stdout.\n"
172 " jdwp list pids of processes hosting a JDWP transport\n"
173 " logcat show device log (logcat --help for more)\n"
174 "\n"
175 "security:\n"
176 " disable-verity disable dm-verity checking on userdebug builds\n"
177 " enable-verity re-enable dm-verity checking on userdebug builds\n"
178 " keygen FILE\n"
179 " generate adb public/private key; private key stored in FILE,\n"
180 " public key stored in FILE.pub (existing files overwritten)\n"
181 "\n"
182 "scripting:\n"
183 " wait-for[-TRANSPORT]-STATE\n"
184 " wait for device to be in the given state\n"
185 " State: device, recovery, sideload, or bootloader\n"
186 " Transport: usb, local, or any [default=any]\n"
187 " get-state print offline | bootloader | device\n"
188 " get-serialno print <serial-number>\n"
189 " get-devpath print <device-path>\n"
190 " remount\n"
191 " remount /system, /vendor, and /oem partitions read-write\n"
192 " reboot [bootloader|recovery|sideload|sideload-auto-reboot]\n"
193 " reboot the device; defaults to booting system image but\n"
194 " supports bootloader and recovery too. sideload reboots\n"
195 " into recovery and automatically starts sideload mode,\n"
196 " sideload-auto-reboot is the same but reboots after sideloading.\n"
197 " sideload OTAPACKAGE sideload the given full OTA package\n"
198 " root restart adbd with root permissions\n"
199 " unroot restart adbd without root permissions\n"
200 " usb restart adb server listening on USB\n"
201 " tcpip PORT restart adb server listening on TCP on PORT\n"
Timcd643152010-02-16 20:18:29 +0000202 "\n"
Yabin Cui1f4ec192016-04-05 13:50:44 -0700203 "internal debugging:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700204 " start-server ensure that there is a server running\n"
205 " kill-server kill the server if it is running\n"
206 " reconnect kick connection from host side to force reconnect\n"
207 " reconnect device kick connection from device side to force reconnect\n"
Yabin Cuib5e11412017-03-10 16:01:01 -0800208 " reconnect offline reset offline/unauthorized devices to force reconnect\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700209 "\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700210 "environment variables:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700211 " $ADB_TRACE\n"
212 " comma-separated list of debug info to log:\n"
213 " all,adb,sockets,packets,rwx,usb,sync,sysdeps,transport,jdwp\n"
214 " $ADB_VENDOR_KEYS colon-separated list of keys (files or directories)\n"
215 " $ANDROID_SERIAL serial number to connect to (see -s)\n"
216 " $ANDROID_LOG_TAGS tags to be used by logcat (see logcat --help)\n");
Felipe Lemeb3239722016-07-29 17:57:00 -0700217 // clang-format on
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800218}
219
Yabin Cuid325e862014-11-17 14:48:25 -0800220#if defined(_WIN32)
221
Elliott Hughesa2f2e562015-04-16 16:47:02 -0700222// Implemented in sysdeps_win32.cpp.
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800223void stdin_raw_init();
224void stdin_raw_restore();
Yabin Cuid325e862014-11-17 14:48:25 -0800225
226#else
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100227static termios g_saved_terminal_state;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800228
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800229static void stdin_raw_init() {
230 if (tcgetattr(STDIN_FILENO, &g_saved_terminal_state)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800231
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100232 termios tio;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800233 if (tcgetattr(STDIN_FILENO, &tio)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800234
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100235 cfmakeraw(&tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800236
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100237 // No timeout but request at least one character per read.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800238 tio.c_cc[VTIME] = 0;
239 tio.c_cc[VMIN] = 1;
240
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800241 tcsetattr(STDIN_FILENO, TCSAFLUSH, &tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800242}
243
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800244static void stdin_raw_restore() {
245 tcsetattr(STDIN_FILENO, TCSAFLUSH, &g_saved_terminal_state);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800246}
247#endif
248
David Pursell606835a2015-09-08 17:17:02 -0700249// Reads from |fd| and prints received data. If |use_shell_protocol| is true
250// this expects that incoming data will use the shell protocol, in which case
251// stdout/stderr are routed independently and the remote exit code will be
252// returned.
Felipe Leme0d4f0502016-07-26 12:14:39 -0700253// if |callback| is non-null, stdout/stderr output will be handled by it.
254int read_and_dump(int fd, bool use_shell_protocol = false,
255 StandardStreamsCallbackInterface* callback = &DEFAULT_STANDARD_STREAMS_CALLBACK) {
David Pursell606835a2015-09-08 17:17:02 -0700256 int exit_code = 0;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700257 if (fd < 0) return exit_code;
258
David Pursell606835a2015-09-08 17:17:02 -0700259 std::unique_ptr<ShellProtocol> protocol;
260 int length = 0;
David Pursell606835a2015-09-08 17:17:02 -0700261
262 char raw_buffer[BUFSIZ];
263 char* buffer_ptr = raw_buffer;
264 if (use_shell_protocol) {
265 protocol.reset(new ShellProtocol(fd));
266 if (!protocol) {
267 LOG(ERROR) << "failed to allocate memory for ShellProtocol object";
268 return 1;
269 }
270 buffer_ptr = protocol->data();
271 }
272
Felipe Lemed1dd1252016-06-06 16:05:36 -0700273 while (true) {
David Pursell606835a2015-09-08 17:17:02 -0700274 if (use_shell_protocol) {
275 if (!protocol->Read()) {
276 break;
277 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700278 length = protocol->data_length();
David Pursell606835a2015-09-08 17:17:02 -0700279 switch (protocol->id()) {
280 case ShellProtocol::kIdStdout:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700281 callback->OnStdout(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700282 break;
283 case ShellProtocol::kIdStderr:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700284 callback->OnStderr(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700285 break;
286 case ShellProtocol::kIdExit:
287 exit_code = protocol->data()[0];
288 continue;
289 default:
290 continue;
291 }
292 length = protocol->data_length();
293 } else {
294 D("read_and_dump(): pre adb_read(fd=%d)", fd);
295 length = adb_read(fd, raw_buffer, sizeof(raw_buffer));
296 D("read_and_dump(): post adb_read(fd=%d): length=%d", fd, length);
297 if (length <= 0) {
298 break;
299 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700300 callback->OnStdout(buffer_ptr, length);
Felipe Leme44a42672016-04-01 17:43:27 -0700301 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800302 }
David Pursell606835a2015-09-08 17:17:02 -0700303
Felipe Leme0d4f0502016-07-26 12:14:39 -0700304 return callback->Done(exit_code);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800305}
306
Jeff Sharkey960df972014-06-09 17:30:57 -0700307static void read_status_line(int fd, char* buf, size_t count)
308{
309 count--;
310 while (count > 0) {
311 int len = adb_read(fd, buf, count);
Elliott Hughes8fcd8bc2015-08-25 10:59:45 -0700312 if (len <= 0) {
Jeff Sharkey960df972014-06-09 17:30:57 -0700313 break;
314 }
315
316 buf += len;
317 count -= len;
318 }
319 *buf = '\0';
320}
321
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700322static void stdinout_raw_prologue(int inFd, int outFd, int& old_stdin_mode, int& old_stdout_mode) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700323 if (inFd == STDIN_FILENO) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800324 stdin_raw_init();
Spencer Lowb7dfb792015-05-22 16:48:31 -0700325#ifdef _WIN32
326 old_stdin_mode = _setmode(STDIN_FILENO, _O_BINARY);
327 if (old_stdin_mode == -1) {
328 fatal_errno("could not set stdin to binary");
329 }
330#endif
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700331 }
Yabin Cuid325e862014-11-17 14:48:25 -0800332
Spencer Lowb7dfb792015-05-22 16:48:31 -0700333#ifdef _WIN32
334 if (outFd == STDOUT_FILENO) {
335 old_stdout_mode = _setmode(STDOUT_FILENO, _O_BINARY);
336 if (old_stdout_mode == -1) {
337 fatal_errno("could not set stdout to binary");
338 }
339 }
340#endif
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700341}
342
343static void stdinout_raw_epilogue(int inFd, int outFd, int old_stdin_mode, int old_stdout_mode) {
344 if (inFd == STDIN_FILENO) {
345 stdin_raw_restore();
346#ifdef _WIN32
347 if (_setmode(STDIN_FILENO, old_stdin_mode) == -1) {
348 fatal_errno("could not restore stdin mode");
349 }
350#endif
351 }
352
353#ifdef _WIN32
354 if (outFd == STDOUT_FILENO) {
355 if (_setmode(STDOUT_FILENO, old_stdout_mode) == -1) {
356 fatal_errno("could not restore stdout mode");
357 }
358 }
359#endif
360}
361
362static void copy_to_file(int inFd, int outFd) {
363 const size_t BUFSIZE = 32 * 1024;
364 char* buf = (char*) malloc(BUFSIZE);
365 if (buf == nullptr) fatal("couldn't allocate buffer for copy_to_file");
366 int len;
367 long total = 0;
368 int old_stdin_mode = -1;
369 int old_stdout_mode = -1;
370
371 D("copy_to_file(%d -> %d)", inFd, outFd);
372
373 stdinout_raw_prologue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700374
Elliott Hughesa7090b92015-04-17 17:03:59 -0700375 while (true) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700376 if (inFd == STDIN_FILENO) {
377 len = unix_read(inFd, buf, BUFSIZE);
378 } else {
379 len = adb_read(inFd, buf, BUFSIZE);
380 }
Christopher Tated2f54152011-04-21 12:53:28 -0700381 if (len == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700382 D("copy_to_file() : read 0 bytes; exiting");
Christopher Tated2f54152011-04-21 12:53:28 -0700383 break;
384 }
385 if (len < 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700386 D("copy_to_file(): read failed: %s", strerror(errno));
Christopher Tated2f54152011-04-21 12:53:28 -0700387 break;
388 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700389 if (outFd == STDOUT_FILENO) {
390 fwrite(buf, 1, len, stdout);
391 fflush(stdout);
392 } else {
393 adb_write(outFd, buf, len);
394 }
Christopher Tatec9cd3b92011-06-01 17:56:23 -0700395 total += len;
Christopher Tated2f54152011-04-21 12:53:28 -0700396 }
Yabin Cuid325e862014-11-17 14:48:25 -0800397
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700398 stdinout_raw_epilogue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700399
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700400 D("copy_to_file() finished after %lu bytes", total);
Christopher Tate5b811fa2011-06-10 11:38:37 -0700401 free(buf);
Christopher Tated2f54152011-04-21 12:53:28 -0700402}
403
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800404static void send_window_size_change(int fd, std::unique_ptr<ShellProtocol>& shell) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800405 // Old devices can't handle window size changes.
406 if (shell == nullptr) return;
407
Spencer Low2e02dc62015-11-07 17:34:39 -0800408#if defined(_WIN32)
409 struct winsize {
410 unsigned short ws_row;
411 unsigned short ws_col;
412 unsigned short ws_xpixel;
413 unsigned short ws_ypixel;
414 };
415#endif
416
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800417 winsize ws;
Spencer Low2e02dc62015-11-07 17:34:39 -0800418
419#if defined(_WIN32)
420 // If stdout is redirected to a non-console, we won't be able to get the
421 // console size, but that makes sense.
422 const intptr_t intptr_handle = _get_osfhandle(STDOUT_FILENO);
423 if (intptr_handle == -1) return;
424
425 const HANDLE handle = reinterpret_cast<const HANDLE>(intptr_handle);
426
427 CONSOLE_SCREEN_BUFFER_INFO info;
428 memset(&info, 0, sizeof(info));
429 if (!GetConsoleScreenBufferInfo(handle, &info)) return;
430
431 memset(&ws, 0, sizeof(ws));
432 // The number of visible rows, excluding offscreen scroll-back rows which are in info.dwSize.Y.
433 ws.ws_row = info.srWindow.Bottom - info.srWindow.Top + 1;
434 // If the user has disabled "Wrap text output on resize", they can make the screen buffer wider
435 // than the window, in which case we should use the width of the buffer.
436 ws.ws_col = info.dwSize.X;
437#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800438 if (ioctl(fd, TIOCGWINSZ, &ws) == -1) return;
Spencer Low2e02dc62015-11-07 17:34:39 -0800439#endif
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800440
441 // Send the new window size as human-readable ASCII for debugging convenience.
442 size_t l = snprintf(shell->data(), shell->data_capacity(), "%dx%d,%dx%d",
443 ws.ws_row, ws.ws_col, ws.ws_xpixel, ws.ws_ypixel);
444 shell->Write(ShellProtocol::kIdWindowSizeChange, l + 1);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800445}
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800446
David Pursell606835a2015-09-08 17:17:02 -0700447// Used to pass multiple values to the stdin read thread.
448struct StdinReadArgs {
449 int stdin_fd, write_fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700450 bool raw_stdin;
David Pursell606835a2015-09-08 17:17:02 -0700451 std::unique_ptr<ShellProtocol> protocol;
Elliott Hughes28416d62015-11-04 13:36:32 -0800452 char escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700453};
454
David Pursell606835a2015-09-08 17:17:02 -0700455// Loops to read from stdin and push the data to the given FD.
456// The argument should be a pointer to a StdinReadArgs object. This function
457// will take ownership of the object and delete it when finished.
Josh Gaod9db09c2016-02-12 14:31:15 -0800458static void stdin_read_thread_loop(void* x) {
David Pursell606835a2015-09-08 17:17:02 -0700459 std::unique_ptr<StdinReadArgs> args(reinterpret_cast<StdinReadArgs*>(x));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800460
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800461#if !defined(_WIN32)
462 // Mask SIGTTIN in case we're in a backgrounded process.
Josh Gao8dcdb572015-10-23 15:03:31 -0700463 sigset_t sigset;
464 sigemptyset(&sigset);
465 sigaddset(&sigset, SIGTTIN);
466 pthread_sigmask(SIG_BLOCK, &sigset, nullptr);
467#endif
468
Spencer Low2e02dc62015-11-07 17:34:39 -0800469#if defined(_WIN32)
470 // _get_interesting_input_record_uncached() causes unix_read_interruptible()
471 // to return -1 with errno == EINTR if the window size changes.
472#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800473 // Unblock SIGWINCH for this thread, so our read(2) below will be
474 // interrupted if the window size changes.
475 sigset_t mask;
476 sigemptyset(&mask);
477 sigaddset(&mask, SIGWINCH);
478 pthread_sigmask(SIG_UNBLOCK, &mask, nullptr);
479#endif
480
481 // Set up the initial window size.
482 send_window_size_change(args->stdin_fd, args->protocol);
483
Elliott Hughes28416d62015-11-04 13:36:32 -0800484 char raw_buffer[BUFSIZ];
David Pursell606835a2015-09-08 17:17:02 -0700485 char* buffer_ptr = raw_buffer;
486 size_t buffer_size = sizeof(raw_buffer);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800487 if (args->protocol != nullptr) {
David Pursell606835a2015-09-08 17:17:02 -0700488 buffer_ptr = args->protocol->data();
489 buffer_size = args->protocol->data_capacity();
490 }
491
Elliott Hughes28416d62015-11-04 13:36:32 -0800492 // If we need to parse escape sequences, make life easy.
493 if (args->raw_stdin && args->escape_char != '\0') {
494 buffer_size = 1;
495 }
496
497 enum EscapeState { kMidFlow, kStartOfLine, kInEscape };
498 EscapeState state = kStartOfLine;
499
Elliott Hughesaa245492015-08-03 10:38:08 -0700500 while (true) {
Spencer Low2e02dc62015-11-07 17:34:39 -0800501 // Use unix_read_interruptible() rather than adb_read() for stdin.
502 D("stdin_read_thread_loop(): pre unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
503 int r = unix_read_interruptible(args->stdin_fd, buffer_ptr,
504 buffer_size);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800505 if (r == -1 && errno == EINTR) {
506 send_window_size_change(args->stdin_fd, args->protocol);
507 continue;
508 }
Spencer Low2e02dc62015-11-07 17:34:39 -0800509 D("stdin_read_thread_loop(): post unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
David Pursell1ed57f02015-10-06 15:30:03 -0700510 if (r <= 0) {
511 // Only devices using the shell protocol know to close subprocess
512 // stdin. For older devices we want to just leave the connection
513 // open, otherwise an unpredictable amount of return data could
514 // be lost due to the FD closing before all data has been received.
515 if (args->protocol) {
516 args->protocol->Write(ShellProtocol::kIdCloseStdin, 0);
517 }
518 break;
519 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800520 // If we made stdin raw, check input for escape sequences. In
David Pursell1ed57f02015-10-06 15:30:03 -0700521 // this situation signals like Ctrl+C are sent remotely rather than
522 // interpreted locally so this provides an emergency out if the remote
523 // process starts ignoring the signal. SSH also does this, see the
524 // "escape characters" section on the ssh man page for more info.
Elliott Hughes28416d62015-11-04 13:36:32 -0800525 if (args->raw_stdin && args->escape_char != '\0') {
526 char ch = buffer_ptr[0];
527 if (ch == args->escape_char) {
528 if (state == kStartOfLine) {
529 state = kInEscape;
530 // Swallow the escape character.
531 continue;
532 } else {
533 state = kMidFlow;
534 }
535 } else {
536 if (state == kInEscape) {
537 if (ch == '.') {
538 fprintf(stderr,"\r\n[ disconnected ]\r\n");
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800539 stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700540 exit(0);
Elliott Hughes28416d62015-11-04 13:36:32 -0800541 } else {
542 // We swallowed an escape character that wasn't part of
543 // a valid escape sequence; time to cough it up.
544 buffer_ptr[0] = args->escape_char;
545 buffer_ptr[1] = ch;
546 ++r;
David Pursell1ed57f02015-10-06 15:30:03 -0700547 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800548 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800549 state = (ch == '\n' || ch == '\r') ? kStartOfLine : kMidFlow;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800550 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800551 }
David Pursell606835a2015-09-08 17:17:02 -0700552 if (args->protocol) {
553 if (!args->protocol->Write(ShellProtocol::kIdStdin, r)) {
554 break;
555 }
556 } else {
557 if (!WriteFdExactly(args->write_fd, buffer_ptr, r)) {
558 break;
559 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800560 }
561 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800562}
563
David Pursell1ed57f02015-10-06 15:30:03 -0700564// Returns a shell service string with the indicated arguments and command.
565static std::string ShellServiceString(bool use_shell_protocol,
566 const std::string& type_arg,
567 const std::string& command) {
568 std::vector<std::string> args;
569 if (use_shell_protocol) {
570 args.push_back(kShellServiceArgShellProtocol);
Elliott Hughesc2252df2015-11-18 12:45:48 -0800571
572 const char* terminal_type = getenv("TERM");
573 if (terminal_type != nullptr) {
574 args.push_back(std::string("TERM=") + terminal_type);
575 }
David Pursell1ed57f02015-10-06 15:30:03 -0700576 }
577 if (!type_arg.empty()) {
578 args.push_back(type_arg);
579 }
580
581 // Shell service string can look like: shell[,arg1,arg2,...]:[command].
582 return android::base::StringPrintf("shell%s%s:%s",
583 args.empty() ? "" : ",",
584 android::base::Join(args, ',').c_str(),
585 command.c_str());
586}
587
588// Connects to a shell on the device and read/writes data.
589//
590// Note: currently this function doesn't properly clean up resources; the
591// FD connected to the adb server is never closed and the stdin read thread
592// may never exit.
593//
594// On success returns the remote exit code if |use_shell_protocol| is true,
595// 0 otherwise. On failure returns 1.
596static int RemoteShell(bool use_shell_protocol, const std::string& type_arg,
Elliott Hughes28416d62015-11-04 13:36:32 -0800597 char escape_char,
David Pursell1ed57f02015-10-06 15:30:03 -0700598 const std::string& command) {
599 std::string service_string = ShellServiceString(use_shell_protocol,
600 type_arg, command);
601
602 // Make local stdin raw if the device allocates a PTY, which happens if:
603 // 1. We are explicitly asking for a PTY shell, or
604 // 2. We don't specify shell type and are starting an interactive session.
605 bool raw_stdin = (type_arg == kShellServiceArgPty ||
606 (type_arg.empty() && command.empty()));
607
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700608 std::string error;
David Pursell4e2fd362015-09-22 10:43:08 -0700609 int fd = adb_connect(service_string, &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700610 if (fd < 0) {
611 fprintf(stderr,"error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800612 return 1;
613 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800614
David Pursell606835a2015-09-08 17:17:02 -0700615 StdinReadArgs* args = new StdinReadArgs;
616 if (!args) {
617 LOG(ERROR) << "couldn't allocate StdinReadArgs object";
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700618 return 1;
619 }
David Pursell1ed57f02015-10-06 15:30:03 -0700620 args->stdin_fd = STDIN_FILENO;
David Pursell606835a2015-09-08 17:17:02 -0700621 args->write_fd = fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700622 args->raw_stdin = raw_stdin;
Elliott Hughes28416d62015-11-04 13:36:32 -0800623 args->escape_char = escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700624 if (use_shell_protocol) {
625 args->protocol.reset(new ShellProtocol(args->write_fd));
626 }
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700627
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800628 if (raw_stdin) stdin_raw_init();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800629
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800630#if !defined(_WIN32)
631 // Ensure our process is notified if the local window size changes.
632 // We use sigaction(2) to ensure that the SA_RESTART flag is not set,
633 // because the whole reason we're sending signals is to unblock the read(2)!
634 // That also means we don't need to do anything in the signal handler:
635 // the side effect of delivering the signal is all we need.
636 struct sigaction sa;
637 memset(&sa, 0, sizeof(sa));
638 sa.sa_handler = [](int) {};
639 sa.sa_flags = 0;
640 sigaction(SIGWINCH, &sa, nullptr);
641
642 // Now block SIGWINCH in this thread (the main thread) and all threads spawned
643 // from it. The stdin read thread will unblock this signal to ensure that it's
644 // the thread that receives the signal.
645 sigset_t mask;
646 sigemptyset(&mask);
647 sigaddset(&mask, SIGWINCH);
648 pthread_sigmask(SIG_BLOCK, &mask, nullptr);
649#endif
650
651 // TODO: combine read_and_dump with stdin_read_thread to make life simpler?
Josh Gaoe1dacfc2017-04-12 17:00:49 -0700652 std::thread(stdin_read_thread_loop, args).detach();
653 int exit_code = read_and_dump(fd, use_shell_protocol);
Elliott Hughes9b0f3542015-05-05 13:41:21 -0700654
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800655 // TODO: properly exit stdin_read_thread_loop and close |fd|.
David Pursell1ed57f02015-10-06 15:30:03 -0700656
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800657 // TODO: we should probably install signal handlers for this.
658 // TODO: can we use atexit? even on Windows?
659 if (raw_stdin) stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700660
David Pursell606835a2015-09-08 17:17:02 -0700661 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800662}
663
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800664static int adb_shell(int argc, const char** argv) {
David Pursell0aacbbe2016-01-21 19:56:50 -0800665 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800666 std::string error;
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800667 if (!adb_get_feature_set(&features, &error)) {
668 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -0800669 return 1;
670 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800671
Elliott Hughescabfa112016-10-19 14:47:11 -0700672 enum PtyAllocationMode { kPtyAuto, kPtyNo, kPtyYes, kPtyDefinitely };
673
674 // Defaults.
675 char escape_char = '~'; // -e
676 bool use_shell_protocol = CanUseFeature(features, kFeatureShell2); // -x
677 PtyAllocationMode tty = use_shell_protocol ? kPtyAuto : kPtyDefinitely; // -t/-T
Elliott Hughes6452a892015-04-29 12:28:13 -0700678
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800679 // Parse shell-specific command-line options.
Elliott Hughescabfa112016-10-19 14:47:11 -0700680 argv[0] = "adb shell"; // So getopt(3) error messages start "adb shell".
681 optind = 1; // argv[0] is always "shell", so set `optind` appropriately.
682 int opt;
683 while ((opt = getopt(argc, const_cast<char**>(argv), "+e:ntTx")) != -1) {
684 switch (opt) {
685 case 'e':
686 if (!(strlen(optarg) == 1 || strcmp(optarg, "none") == 0)) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -0700687 return syntax_error("-e requires a single-character argument or 'none'");
Elliott Hughescabfa112016-10-19 14:47:11 -0700688 }
689 escape_char = (strcmp(optarg, "none") == 0) ? 0 : optarg[0];
690 break;
691 case 'n':
692 close_stdin();
693 break;
694 case 'x':
695 // This option basically asks for historical behavior, so set options that
696 // correspond to the historical defaults. This is slightly weird in that -Tx
697 // is fine (because we'll undo the -T) but -xT isn't, but that does seem to
698 // be our least worst choice...
699 use_shell_protocol = false;
700 tty = kPtyDefinitely;
701 escape_char = '~';
702 break;
703 case 't':
704 // Like ssh, -t arguments are cumulative so that multiple -t's
705 // are needed to force a PTY.
706 tty = (tty >= kPtyYes) ? kPtyDefinitely : kPtyYes;
707 break;
708 case 'T':
709 tty = kPtyNo;
710 break;
711 default:
712 // getopt(3) already printed an error message for us.
Elliott Hughes28416d62015-11-04 13:36:32 -0800713 return 1;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800714 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700715 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800716
Elliott Hughescabfa112016-10-19 14:47:11 -0700717 bool is_interactive = (optind == argc);
David Pursell57dd5ae2016-01-27 16:07:52 -0800718
Elliott Hughescabfa112016-10-19 14:47:11 -0700719 std::string shell_type_arg = kShellServiceArgPty;
720 if (tty == kPtyNo) {
721 shell_type_arg = kShellServiceArgRaw;
722 } else if (tty == kPtyAuto) {
723 // If stdin isn't a TTY, default to a raw shell; this lets
724 // things like `adb shell < my_script.sh` work as expected.
725 // Non-interactive shells should also not have a pty.
726 if (!unix_isatty(STDIN_FILENO) || !is_interactive) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800727 shell_type_arg = kShellServiceArgRaw;
Elliott Hughescabfa112016-10-19 14:47:11 -0700728 }
729 } else if (tty == kPtyYes) {
730 // A single -t arg isn't enough to override implicit -T.
731 if (!unix_isatty(STDIN_FILENO)) {
732 fprintf(stderr,
733 "Remote PTY will not be allocated because stdin is not a terminal.\n"
734 "Use multiple -t options to force remote PTY allocation.\n");
735 shell_type_arg = kShellServiceArgRaw;
736 }
737 }
738
739 D("shell -e 0x%x t=%d use_shell_protocol=%s shell_type_arg=%s\n",
740 escape_char, tty,
741 use_shell_protocol ? "true" : "false",
742 (shell_type_arg == kShellServiceArgPty) ? "pty" : "raw");
743
744 // Raw mode is only supported when talking to a new device *and* using the shell protocol.
745 if (!use_shell_protocol) {
746 if (shell_type_arg != kShellServiceArgPty) {
747 fprintf(stderr, "error: %s only supports allocating a pty\n",
748 !CanUseFeature(features, kFeatureShell2) ? "device" : "-x");
749 return 1;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800750 } else {
Elliott Hughescabfa112016-10-19 14:47:11 -0700751 // If we're not using the shell protocol, the type argument must be empty.
752 shell_type_arg = "";
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800753 }
David Pursell71c83122015-09-14 15:33:50 -0700754 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800755
756 std::string command;
Elliott Hughescabfa112016-10-19 14:47:11 -0700757 if (optind < argc) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800758 // We don't escape here, just like ssh(1). http://b/20564385.
Elliott Hughescabfa112016-10-19 14:47:11 -0700759 command = android::base::Join(std::vector<const char*>(argv + optind, argv + argc), ' ');
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800760 }
761
Elliott Hughes28416d62015-11-04 13:36:32 -0800762 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, command);
David Pursell71c83122015-09-14 15:33:50 -0700763}
764
Elliott Hughes09e794c2017-05-02 14:31:59 -0700765static int adb_sideload_legacy(const char* filename, int in_fd, int size) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700766 std::string error;
Elliott Hughes09e794c2017-05-02 14:31:59 -0700767 int out_fd = adb_connect(android::base::StringPrintf("sideload:%d", size), &error);
768 if (out_fd < 0) {
769 fprintf(stderr, "adb: pre-KitKat sideload connection failed: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800770 return -1;
771 }
772
773 int opt = CHUNK_SIZE;
Elliott Hughes09e794c2017-05-02 14:31:59 -0700774 opt = adb_setsockopt(out_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
Doug Zongker447f0612012-01-09 14:54:53 -0800775
Elliott Hughes09e794c2017-05-02 14:31:59 -0700776 char buf[CHUNK_SIZE];
777 int total = size;
778 while (size > 0) {
779 unsigned xfer = (size > CHUNK_SIZE) ? CHUNK_SIZE : size;
780 if (!ReadFdExactly(in_fd, buf, xfer)) {
781 fprintf(stderr, "adb: failed to read data from %s: %s\n", filename, strerror(errno));
782 adb_close(out_fd);
Doug Zongker447f0612012-01-09 14:54:53 -0800783 return -1;
784 }
Elliott Hughes09e794c2017-05-02 14:31:59 -0700785 if (!WriteFdExactly(out_fd, buf, xfer)) {
786 std::string error;
787 adb_status(out_fd, &error);
788 fprintf(stderr, "adb: failed to write data: %s\n", error.c_str());
789 adb_close(out_fd);
790 return -1;
791 }
792 size -= xfer;
793 printf("sending: '%s' %4d%% \r", filename, (int)(100LL - ((100LL * size) / (total))));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700794 fflush(stdout);
Doug Zongker447f0612012-01-09 14:54:53 -0800795 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700796 printf("\n");
Doug Zongker447f0612012-01-09 14:54:53 -0800797
Elliott Hughes09e794c2017-05-02 14:31:59 -0700798 if (!adb_status(out_fd, &error)) {
799 fprintf(stderr, "adb: error response: %s\n", error.c_str());
800 adb_close(out_fd);
Doug Zongker447f0612012-01-09 14:54:53 -0800801 return -1;
802 }
803
Elliott Hughes09e794c2017-05-02 14:31:59 -0700804 adb_close(out_fd);
Doug Zongker447f0612012-01-09 14:54:53 -0800805 return 0;
806}
807
Doug Zongker71fe5842014-06-26 15:35:36 -0700808#define SIDELOAD_HOST_BLOCK_SIZE (CHUNK_SIZE)
809
810/*
811 * The sideload-host protocol serves the data in a file (given on the
812 * command line) to the client, using a simple protocol:
813 *
814 * - The connect message includes the total number of bytes in the
815 * file and a block size chosen by us.
816 *
817 * - The other side sends the desired block number as eight decimal
818 * digits (eg "00000023" for block 23). Blocks are numbered from
819 * zero.
820 *
821 * - We send back the data of the requested block. The last block is
822 * likely to be partial; when the last block is requested we only
823 * send the part of the block that exists, it's not padded up to the
824 * block size.
825 *
826 * - When the other side sends "DONEDONE" instead of a block number,
827 * we hang up.
828 */
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700829static int adb_sideload_host(const char* filename) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -0700830 // TODO: use a LinePrinter instead...
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700831 struct stat sb;
832 if (stat(filename, &sb) == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700833 fprintf(stderr, "adb: failed to stat file %s: %s\n", filename, strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700834 return -1;
835 }
836 unique_fd package_fd(adb_open(filename, O_RDONLY));
837 if (package_fd == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700838 fprintf(stderr, "adb: failed to open file %s: %s\n", filename, strerror(errno));
Doug Zongker71fe5842014-06-26 15:35:36 -0700839 return -1;
840 }
841
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700842 std::string service = android::base::StringPrintf(
843 "sideload-host:%d:%d", static_cast<int>(sb.st_size), SIDELOAD_HOST_BLOCK_SIZE);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700844 std::string error;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700845 unique_fd device_fd(adb_connect(service, &error));
846 if (device_fd < 0) {
847 // Try falling back to the older (<= K) sideload method. Maybe this
Doug Zongker71fe5842014-06-26 15:35:36 -0700848 // is an older device that doesn't support sideload-host.
Elliott Hughes09e794c2017-05-02 14:31:59 -0700849 fprintf(stderr, "adb: sideload connection failed: %s\n", error.c_str());
850 fprintf(stderr, "adb: trying pre-KitKat sideload method...\n");
851 return adb_sideload_legacy(filename, package_fd, static_cast<int>(sb.st_size));
Doug Zongker71fe5842014-06-26 15:35:36 -0700852 }
853
Elliott Hughese8b663f2016-05-26 22:43:19 -0700854 int opt = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700855 adb_setsockopt(device_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
856
857 char buf[SIDELOAD_HOST_BLOCK_SIZE];
Doug Zongker71fe5842014-06-26 15:35:36 -0700858
Elliott Hughese8b663f2016-05-26 22:43:19 -0700859 size_t xfer = 0;
860 int last_percent = -1;
Elliott Hughesa7090b92015-04-17 17:03:59 -0700861 while (true) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700862 if (!ReadFdExactly(device_fd, buf, 8)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700863 fprintf(stderr, "adb: failed to read command: %s\n", strerror(errno));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700864 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700865 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700866 buf[8] = '\0';
Doug Zongker71fe5842014-06-26 15:35:36 -0700867
Elliott Hughes6452a892015-04-29 12:28:13 -0700868 if (strcmp("DONEDONE", buf) == 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -0700869 printf("\rTotal xfer: %.2fx%*s\n",
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700870 static_cast<double>(xfer) / (sb.st_size ? sb.st_size : 1),
871 static_cast<int>(strlen(filename) + 10), "");
Elliott Hughese8b663f2016-05-26 22:43:19 -0700872 return 0;
Doug Zongker71fe5842014-06-26 15:35:36 -0700873 }
874
Doug Zongker71fe5842014-06-26 15:35:36 -0700875 int block = strtol(buf, NULL, 10);
876
877 size_t offset = block * SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700878 if (offset >= static_cast<size_t>(sb.st_size)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700879 fprintf(stderr, "adb: failed to read block %d past end\n", block);
Elliott Hughese8b663f2016-05-26 22:43:19 -0700880 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700881 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700882
Doug Zongker71fe5842014-06-26 15:35:36 -0700883 size_t to_write = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700884 if ((offset + SIDELOAD_HOST_BLOCK_SIZE) > static_cast<size_t>(sb.st_size)) {
885 to_write = sb.st_size - offset;
Doug Zongker71fe5842014-06-26 15:35:36 -0700886 }
887
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700888 if (adb_lseek(package_fd, offset, SEEK_SET) != static_cast<int>(offset)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700889 fprintf(stderr, "adb: failed to seek to package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700890 return -1;
891 }
892 if (!ReadFdExactly(package_fd, buf, to_write)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700893 fprintf(stderr, "adb: failed to read package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700894 return -1;
895 }
896
897 if (!WriteFdExactly(device_fd, buf, to_write)) {
898 adb_status(device_fd, &error);
Elliott Hughes09e794c2017-05-02 14:31:59 -0700899 fprintf(stderr, "adb: failed to write data '%s' *\n", error.c_str());
Elliott Hughese8b663f2016-05-26 22:43:19 -0700900 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700901 }
902 xfer += to_write;
903
904 // For normal OTA packages, we expect to transfer every byte
905 // twice, plus a bit of overhead (one read during
906 // verification, one read of each byte for installation, plus
907 // extra access to things like the zip central directory).
908 // This estimate of the completion becomes 100% when we've
909 // transferred ~2.13 (=100/47) times the package size.
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700910 int percent = static_cast<int>(xfer * 47LL / (sb.st_size ? sb.st_size : 1));
Doug Zongker71fe5842014-06-26 15:35:36 -0700911 if (percent != last_percent) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700912 printf("\rserving: '%s' (~%d%%) ", filename, percent);
Doug Zongker71fe5842014-06-26 15:35:36 -0700913 fflush(stdout);
914 last_percent = percent;
915 }
916 }
Doug Zongker71fe5842014-06-26 15:35:36 -0700917}
918
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800919/**
920 * Run ppp in "notty" mode against a resource listed as the first parameter
921 * eg:
922 *
923 * ppp dev:/dev/omap_csmi_tty0 <ppp options>
924 *
925 */
Elliott Hughes58305772015-04-17 13:57:15 -0700926static int ppp(int argc, const char** argv) {
Yabin Cuie77b6a02014-11-11 09:24:11 -0800927#if defined(_WIN32)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800928 fprintf(stderr, "error: adb %s not implemented on Win32\n", argv[0]);
929 return -1;
930#else
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -0700931 if (argc < 2) return syntax_error("adb %s <adb service name> [ppp opts]", argv[0]);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800932
Dan Albertbac34742015-02-25 17:51:28 -0800933 const char* adb_service_name = argv[1];
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700934 std::string error;
935 int fd = adb_connect(adb_service_name, &error);
936 if (fd < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -0700937 fprintf(stderr, "adb: could not open adb service %s: %s\n", adb_service_name, error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800938 return 1;
939 }
940
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700941 pid_t pid = fork();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800942
943 if (pid < 0) {
944 perror("from fork()");
945 return 1;
946 } else if (pid == 0) {
947 int err;
948 int i;
949 const char **ppp_args;
950
951 // copy args
952 ppp_args = (const char **) alloca(sizeof(char *) * argc + 1);
953 ppp_args[0] = "pppd";
954 for (i = 2 ; i < argc ; i++) {
955 //argv[2] and beyond become ppp_args[1] and beyond
956 ppp_args[i - 1] = argv[i];
957 }
958 ppp_args[i-1] = NULL;
959
960 // child side
961
962 dup2(fd, STDIN_FILENO);
963 dup2(fd, STDOUT_FILENO);
964 adb_close(STDERR_FILENO);
965 adb_close(fd);
966
967 err = execvp("pppd", (char * const *)ppp_args);
968
969 if (err < 0) {
970 perror("execing pppd");
971 }
972 exit(-1);
973 } else {
974 // parent side
975
976 adb_close(fd);
977 return 0;
978 }
Yabin Cuie77b6a02014-11-11 09:24:11 -0800979#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800980}
981
Josh Gao08881e72016-04-13 12:14:16 -0700982static bool wait_for_device(const char* service, TransportType t, const char* serial) {
983 std::vector<std::string> components = android::base::Split(service, "-");
984 if (components.size() < 3 || components.size() > 4) {
Leo Sartre1fbc9db2015-11-27 18:56:48 +0100985 fprintf(stderr, "adb: couldn't parse 'wait-for' command: %s\n", service);
986 return false;
987 }
988
Elliott Hughes2b101112015-05-04 19:29:32 -0700989 // Was the caller vague about what they'd like us to wait for?
990 // If so, check they weren't more specific in their choice of transport type.
Josh Gao08881e72016-04-13 12:14:16 -0700991 if (components.size() == 3) {
992 auto it = components.begin() + 2;
Elliott Hughes2b101112015-05-04 19:29:32 -0700993 if (t == kTransportUsb) {
Josh Gao08881e72016-04-13 12:14:16 -0700994 components.insert(it, "usb");
Elliott Hughes2b101112015-05-04 19:29:32 -0700995 } else if (t == kTransportLocal) {
Josh Gao08881e72016-04-13 12:14:16 -0700996 components.insert(it, "local");
Elliott Hughes2b101112015-05-04 19:29:32 -0700997 } else {
Josh Gao08881e72016-04-13 12:14:16 -0700998 components.insert(it, "any");
Elliott Hughes2b101112015-05-04 19:29:32 -0700999 }
Josh Gao08881e72016-04-13 12:14:16 -07001000 } else if (components[2] != "any" && components[2] != "local" && components[2] != "usb") {
1001 fprintf(stderr, "adb: unknown type %s; expected 'any', 'local', or 'usb'\n",
1002 components[2].c_str());
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001003 return false;
1004 }
1005
Josh Gao86441c32016-04-13 12:18:58 -07001006 if (components[3] != "any" && components[3] != "bootloader" && components[3] != "device" &&
1007 components[3] != "recovery" && components[3] != "sideload") {
Josh Gao08881e72016-04-13 12:14:16 -07001008 fprintf(stderr,
1009 "adb: unknown state %s; "
Josh Gao86441c32016-04-13 12:18:58 -07001010 "expected 'any', 'bootloader', 'device', 'recovery', or 'sideload'\n",
Josh Gao08881e72016-04-13 12:14:16 -07001011 components[3].c_str());
1012 return false;
1013 }
1014
1015 std::string cmd = format_host_command(android::base::Join(components, "-").c_str(), t, serial);
Elliott Hughes424af022015-05-29 17:55:19 -07001016 return adb_command(cmd);
Elliott Hughes2b101112015-05-04 19:29:32 -07001017}
1018
Josh Gao06c73ae2016-03-04 15:16:18 -08001019static bool adb_root(const char* command) {
1020 std::string error;
Josh Gao06c73ae2016-03-04 15:16:18 -08001021
Elliott Hughese8b663f2016-05-26 22:43:19 -07001022 unique_fd fd(adb_connect(android::base::StringPrintf("%s:", command), &error));
1023 if (fd < 0) {
Josh Gao06c73ae2016-03-04 15:16:18 -08001024 fprintf(stderr, "adb: unable to connect for %s: %s\n", command, error.c_str());
1025 return false;
1026 }
1027
1028 // Figure out whether we actually did anything.
1029 char buf[256];
1030 char* cur = buf;
1031 ssize_t bytes_left = sizeof(buf);
1032 while (bytes_left > 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -07001033 ssize_t bytes_read = adb_read(fd, cur, bytes_left);
Josh Gao06c73ae2016-03-04 15:16:18 -08001034 if (bytes_read == 0) {
1035 break;
1036 } else if (bytes_read < 0) {
1037 fprintf(stderr, "adb: error while reading for %s: %s\n", command, strerror(errno));
1038 return false;
1039 }
1040 cur += bytes_read;
1041 bytes_left -= bytes_read;
1042 }
1043
1044 if (bytes_left == 0) {
1045 fprintf(stderr, "adb: unexpected output length for %s\n", command);
1046 return false;
1047 }
1048
1049 fflush(stdout);
1050 WriteFdExactly(STDOUT_FILENO, buf, sizeof(buf) - bytes_left);
1051 if (cur != buf && strstr(buf, "restarting") == nullptr) {
1052 return true;
1053 }
1054
Josh Gao05824732016-06-16 14:00:09 -07001055 // Give adbd some time to kill itself and come back up.
1056 // We can't use wait-for-device because devices (e.g. adb over network) might not come back.
Josh Gao4602adb2016-11-15 18:55:47 -08001057 std::this_thread::sleep_for(3s);
Josh Gao05824732016-06-16 14:00:09 -07001058 return true;
Josh Gao06c73ae2016-03-04 15:16:18 -08001059}
1060
Felipe Leme698e0652016-07-19 17:07:22 -07001061int send_shell_command(TransportType transport_type, const char* serial, const std::string& command,
Felipe Leme0d4f0502016-07-26 12:14:39 -07001062 bool disable_shell_protocol, StandardStreamsCallbackInterface* callback) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001063 int fd;
David Pursell0aacbbe2016-01-21 19:56:50 -08001064 bool use_shell_protocol = false;
1065
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001066 while (true) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001067 bool attempt_connection = true;
1068
Felipe Leme0d4f0502016-07-26 12:14:39 -07001069 // Use shell protocol if it's supported and the caller doesn't explicitly
1070 // disable it.
David Pursell0aacbbe2016-01-21 19:56:50 -08001071 if (!disable_shell_protocol) {
1072 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001073 std::string error;
1074 if (adb_get_feature_set(&features, &error)) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001075 use_shell_protocol = CanUseFeature(features, kFeatureShell2);
1076 } else {
1077 // Device was unreachable.
1078 attempt_connection = false;
1079 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001080 }
David Pursell0aacbbe2016-01-21 19:56:50 -08001081
1082 if (attempt_connection) {
1083 std::string error;
1084 std::string service_string = ShellServiceString(use_shell_protocol, "", command);
1085
1086 fd = adb_connect(service_string, &error);
1087 if (fd >= 0) {
1088 break;
1089 }
1090 }
1091
Felipe Leme0d4f0502016-07-26 12:14:39 -07001092 fprintf(stderr, "- waiting for device -\n");
Josh Gaob6117c42016-02-24 15:16:17 -08001093 if (!wait_for_device("wait-for-device", transport_type, serial)) {
1094 return 1;
1095 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001096 }
1097
Felipe Leme0d4f0502016-07-26 12:14:39 -07001098 int exit_code = read_and_dump(fd, use_shell_protocol, callback);
David Pursell71c83122015-09-14 15:33:50 -07001099
1100 if (adb_close(fd) < 0) {
1101 PLOG(ERROR) << "failure closing FD " << fd;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001102 }
David Pursell71c83122015-09-14 15:33:50 -07001103
1104 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001105}
1106
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001107static int logcat(TransportType transport, const char* serial, int argc, const char** argv) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001108 char* log_tags = getenv("ANDROID_LOG_TAGS");
1109 std::string quoted = escape_arg(log_tags == nullptr ? "" : log_tags);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001110
David Pursell70ef7b42015-09-30 13:35:42 -07001111 std::string cmd = "export ANDROID_LOG_TAGS=\"" + quoted + "\"; exec logcat";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001112
Jeff Sharkeyfd546e82014-06-10 11:31:24 -07001113 if (!strcmp(argv[0], "longcat")) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001114 cmd += " -v long";
Christopher Tatedb0a8802011-11-30 13:00:33 -08001115 }
1116
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001117 --argc;
1118 ++argv;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001119 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001120 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001121 }
1122
David Pursell70ef7b42015-09-30 13:35:42 -07001123 // No need for shell protocol with logcat, always disable for simplicity.
1124 return send_shell_command(transport, serial, cmd, true);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001125}
1126
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001127static void write_zeros(int bytes, int fd) {
1128 int old_stdin_mode = -1;
1129 int old_stdout_mode = -1;
1130 char* buf = (char*) calloc(1, bytes);
1131 if (buf == nullptr) fatal("couldn't allocate buffer for write_zeros");
1132
1133 D("write_zeros(%d) -> %d", bytes, fd);
1134
1135 stdinout_raw_prologue(-1, fd, old_stdin_mode, old_stdout_mode);
1136
1137 if (fd == STDOUT_FILENO) {
1138 fwrite(buf, 1, bytes, stdout);
1139 fflush(stdout);
1140 } else {
1141 adb_write(fd, buf, bytes);
1142 }
1143
1144 stdinout_raw_prologue(-1, fd, old_stdin_mode, old_stdout_mode);
1145
1146 D("write_zeros() finished");
1147 free(buf);
1148}
1149
Dan Albertbac34742015-02-25 17:51:28 -08001150static int backup(int argc, const char** argv) {
Elliott Hughes24f165f2015-08-21 20:31:31 -07001151 const char* filename = "backup.ab";
Christopher Tated2f54152011-04-21 12:53:28 -07001152
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001153 /* find, extract, and use any -f argument */
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001154 for (int i = 1; i < argc; i++) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001155 if (!strcmp("-f", argv[i])) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001156 if (i == argc - 1) return syntax_error("backup -f passed with no filename");
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001157 filename = argv[i+1];
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001158 for (int j = i+2; j <= argc; ) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001159 argv[i++] = argv[j++];
1160 }
1161 argc -= 2;
1162 argv[argc] = NULL;
1163 }
Christopher Tated2f54152011-04-21 12:53:28 -07001164 }
1165
Elliott Hughes56e68132015-11-13 11:04:10 -08001166 // Bare "adb backup" or "adb backup -f filename" are not valid invocations ---
1167 // a list of packages is required.
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001168 if (argc < 2) return syntax_error("backup either needs a list of packages or -all/-shared");
Christopher Tatebb86bc52011-08-22 17:12:08 -07001169
Christopher Tateb1dfffe2011-12-08 19:04:34 -08001170 adb_unlink(filename);
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001171 int outFd = adb_creat(filename, 0640);
Christopher Tated2f54152011-04-21 12:53:28 -07001172 if (outFd < 0) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001173 fprintf(stderr, "adb: backup unable to create file '%s': %s\n", filename, strerror(errno));
1174 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001175 }
1176
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001177 std::string cmd = "backup:";
1178 --argc;
1179 ++argv;
1180 while (argc-- > 0) {
1181 cmd += " " + escape_arg(*argv++);
Christopher Tated2f54152011-04-21 12:53:28 -07001182 }
1183
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001184 D("backup. filename=%s cmd=%s", filename, cmd.c_str());
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001185 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001186 int fd = adb_connect(cmd, &error);
Christopher Tated2f54152011-04-21 12:53:28 -07001187 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001188 fprintf(stderr, "adb: unable to connect for backup: %s\n", error.c_str());
Christopher Tated2f54152011-04-21 12:53:28 -07001189 adb_close(outFd);
Elliott Hughes56e68132015-11-13 11:04:10 -08001190 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001191 }
1192
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001193 fprintf(stdout, "Now unlock your device and confirm the backup operation...\n");
Elliott Hughes56e68132015-11-13 11:04:10 -08001194 fflush(stdout);
1195
Christopher Tated2f54152011-04-21 12:53:28 -07001196 copy_to_file(fd, outFd);
1197
1198 adb_close(fd);
1199 adb_close(outFd);
Elliott Hughes56e68132015-11-13 11:04:10 -08001200 return EXIT_SUCCESS;
Christopher Tated2f54152011-04-21 12:53:28 -07001201}
1202
Dan Albertbac34742015-02-25 17:51:28 -08001203static int restore(int argc, const char** argv) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001204 if (argc != 2) return syntax_error("restore requires an argument");
Christopher Tate702967a2011-05-17 15:52:54 -07001205
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001206 const char* filename = argv[1];
1207 int tarFd = adb_open(filename, O_RDONLY);
Christopher Tate702967a2011-05-17 15:52:54 -07001208 if (tarFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001209 fprintf(stderr, "adb: unable to open file %s: %s\n", filename, strerror(errno));
Christopher Tate702967a2011-05-17 15:52:54 -07001210 return -1;
1211 }
1212
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001213 std::string error;
1214 int fd = adb_connect("restore:", &error);
Christopher Tate702967a2011-05-17 15:52:54 -07001215 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001216 fprintf(stderr, "adb: unable to connect for restore: %s\n", error.c_str());
Christopher Tate702967a2011-05-17 15:52:54 -07001217 adb_close(tarFd);
1218 return -1;
1219 }
1220
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001221 fprintf(stdout, "Now unlock your device and confirm the restore operation.\n");
1222 fflush(stdout);
1223
Christopher Tate702967a2011-05-17 15:52:54 -07001224 copy_to_file(tarFd, fd);
1225
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001226 // Provide an in-band EOD marker in case the archive file is malformed
1227 write_zeros(512*2, fd);
1228
Josh Gao5767d0d2016-03-04 15:51:03 -08001229 // Wait until the other side finishes, or it'll get sent SIGHUP.
1230 copy_to_file(fd, STDOUT_FILENO);
1231
Christopher Tate702967a2011-05-17 15:52:54 -07001232 adb_close(fd);
1233 adb_close(tarFd);
1234 return 0;
1235}
1236
Dan Albert06b0d6b2017-05-18 22:56:48 -07001237static void parse_push_pull_args(const char** arg, int narg, std::vector<const char*>* srcs,
1238 const char** dst, bool* copy_attrs, bool* sync) {
Josh Gao05786772015-10-30 16:57:19 -07001239 *copy_attrs = false;
Mark Lindner76f2a932014-03-11 17:55:59 -07001240
Josh Gao05786772015-10-30 16:57:19 -07001241 srcs->clear();
1242 bool ignore_flags = false;
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001243 while (narg > 0) {
Josh Gao05786772015-10-30 16:57:19 -07001244 if (ignore_flags || *arg[0] != '-') {
1245 srcs->push_back(*arg);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001246 } else {
Josh Gao05786772015-10-30 16:57:19 -07001247 if (!strcmp(*arg, "-p")) {
1248 // Silently ignore for backwards compatibility.
1249 } else if (!strcmp(*arg, "-a")) {
1250 *copy_attrs = true;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001251 } else if (!strcmp(*arg, "--sync")) {
1252 if (sync != nullptr) {
1253 *sync = true;
1254 }
Josh Gao05786772015-10-30 16:57:19 -07001255 } else if (!strcmp(*arg, "--")) {
1256 ignore_flags = true;
1257 } else {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001258 syntax_error("unrecognized option '%s'", *arg);
Josh Gao05786772015-10-30 16:57:19 -07001259 exit(1);
1260 }
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001261 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001262 ++arg;
1263 --narg;
1264 }
1265
Josh Gao05786772015-10-30 16:57:19 -07001266 if (srcs->size() > 1) {
1267 *dst = srcs->back();
1268 srcs->pop_back();
Mark Lindner76f2a932014-03-11 17:55:59 -07001269 }
1270}
1271
Elliott Hughes6452a892015-04-29 12:28:13 -07001272static int adb_connect_command(const std::string& command) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001273 std::string error;
1274 int fd = adb_connect(command, &error);
Elliott Hughes5677c232015-05-07 23:37:40 -07001275 if (fd < 0) {
1276 fprintf(stderr, "error: %s\n", error.c_str());
1277 return 1;
Tao Bao175b7bb2015-03-29 11:22:34 -07001278 }
Elliott Hughes5677c232015-05-07 23:37:40 -07001279 read_and_dump(fd);
1280 adb_close(fd);
1281 return 0;
Tao Bao175b7bb2015-03-29 11:22:34 -07001282}
1283
Elliott Hughes6452a892015-04-29 12:28:13 -07001284static int adb_query_command(const std::string& command) {
1285 std::string result;
1286 std::string error;
1287 if (!adb_query(command, &result, &error)) {
1288 fprintf(stderr, "error: %s\n", error.c_str());
1289 return 1;
1290 }
1291 printf("%s\n", result.c_str());
1292 return 0;
1293}
1294
Spencer Lowa13df302015-09-07 16:20:13 -07001295// Disallow stdin, stdout, and stderr.
1296static bool _is_valid_ack_reply_fd(const int ack_reply_fd) {
1297#ifdef _WIN32
1298 const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd);
1299 return (GetStdHandle(STD_INPUT_HANDLE) != ack_reply_handle) &&
1300 (GetStdHandle(STD_OUTPUT_HANDLE) != ack_reply_handle) &&
1301 (GetStdHandle(STD_ERROR_HANDLE) != ack_reply_handle);
1302#else
1303 return ack_reply_fd > 2;
1304#endif
1305}
1306
Todd Kennedy248722e2016-07-20 16:22:37 -07001307static bool _use_legacy_install() {
1308 FeatureSet features;
1309 std::string error;
1310 if (!adb_get_feature_set(&features, &error)) {
1311 fprintf(stderr, "error: %s\n", error.c_str());
1312 return true;
1313 }
1314 return !CanUseFeature(features, kFeatureCmd);
1315}
1316
Elliott Hughescabfa112016-10-19 14:47:11 -07001317int adb_commandline(int argc, const char** argv) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001318 int no_daemon = 0;
1319 int is_daemon = 0;
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001320 int is_server = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001321 int r;
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001322 TransportType transport_type = kTransportAny;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001323 int ack_reply_fd = -1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001324
Elliott Hughes65433da2015-11-18 18:07:48 -08001325#if !defined(_WIN32)
1326 // We'd rather have EPIPE than SIGPIPE.
1327 signal(SIGPIPE, SIG_IGN);
1328#endif
1329
Josh Gao9c869b52016-08-25 16:00:22 -07001330 const char* server_host_str = nullptr;
1331 const char* server_port_str = nullptr;
1332 const char* server_socket_str = nullptr;
Stefan Hilzingera84a42e2010-04-19 12:21:12 +01001333
Elliott Hughes8d28e192015-10-07 14:55:10 -07001334 // We need to check for -d and -e before we look at $ANDROID_SERIAL.
1335 const char* serial = nullptr;
1336
Riley Andrews98f58e82014-12-05 17:37:24 -08001337 while (argc > 0) {
1338 if (!strcmp(argv[0],"server")) {
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001339 is_server = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001340 } else if (!strcmp(argv[0],"nodaemon")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001341 no_daemon = 1;
1342 } else if (!strcmp(argv[0], "fork-server")) {
1343 /* this is a special flag used only when the ADB client launches the ADB Server */
1344 is_daemon = 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001345 } else if (!strcmp(argv[0], "--reply-fd")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001346 if (argc < 2) return syntax_error("--reply-fd requires an argument");
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001347 const char* reply_fd_str = argv[1];
1348 argc--;
1349 argv++;
1350 ack_reply_fd = strtol(reply_fd_str, nullptr, 10);
Spencer Lowa13df302015-09-07 16:20:13 -07001351 if (!_is_valid_ack_reply_fd(ack_reply_fd)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001352 fprintf(stderr, "adb: invalid reply fd \"%s\"\n", reply_fd_str);
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001353 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001354 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001355 } else if (argv[0][0]=='-' && argv[0][1]=='s') {
1356 if (isdigit(argv[0][2])) {
1357 serial = argv[0] + 2;
1358 } else {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001359 if (argc < 2 || argv[0][2] != '\0') return syntax_error("-s requires an argument");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001360 serial = argv[1];
1361 argc--;
1362 argv++;
1363 }
1364 } else if (!strcmp(argv[0],"-d")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001365 transport_type = kTransportUsb;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001366 } else if (!strcmp(argv[0],"-e")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001367 transport_type = kTransportLocal;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001368 } else if (!strcmp(argv[0],"-a")) {
1369 gListenAll = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001370 } else if (!strncmp(argv[0], "-H", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001371 if (argv[0][2] == '\0') {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001372 if (argc < 2) return syntax_error("-H requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001373 server_host_str = argv[1];
Matt Gumbeld7b33082012-11-14 10:16:17 -08001374 argc--;
1375 argv++;
1376 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001377 server_host_str = argv[0] + 2;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001378 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001379 } else if (!strncmp(argv[0], "-P", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001380 if (argv[0][2] == '\0') {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001381 if (argc < 2) return syntax_error("-P requires an argument");
Matt Gumbeld7b33082012-11-14 10:16:17 -08001382 server_port_str = argv[1];
1383 argc--;
1384 argv++;
1385 } else {
1386 server_port_str = argv[0] + 2;
1387 }
Josh Gao9c869b52016-08-25 16:00:22 -07001388 } else if (!strcmp(argv[0], "-L")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001389 if (argc < 2) return syntax_error("-L requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001390 server_socket_str = argv[1];
1391 argc--;
1392 argv++;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001393 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001394 /* out of recognized modifiers and flags */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001395 break;
1396 }
1397 argc--;
1398 argv++;
1399 }
1400
Josh Gao9c869b52016-08-25 16:00:22 -07001401 if ((server_host_str || server_port_str) && server_socket_str) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001402 return syntax_error("-L is incompatible with -H or -P");
Josh Gao9c869b52016-08-25 16:00:22 -07001403 }
1404
1405 // If -L, -H, or -P are specified, ignore environment variables.
1406 // Otherwise, prefer ADB_SERVER_SOCKET over ANDROID_ADB_SERVER_ADDRESS/PORT.
Tao Wu135f4ab2016-09-16 13:01:24 -07001407 if (!server_host_str && !server_port_str && !server_socket_str) {
1408 server_socket_str = getenv("ADB_SERVER_SOCKET");
Josh Gao9c869b52016-08-25 16:00:22 -07001409 }
1410
1411 if (!server_socket_str) {
1412 // tcp:1234 and tcp:localhost:1234 are different with -a, so don't default to localhost
1413 server_host_str = server_host_str ? server_host_str : getenv("ANDROID_ADB_SERVER_ADDRESS");
1414
Tao Wu135f4ab2016-09-16 13:01:24 -07001415 int server_port = DEFAULT_ADB_PORT;
Josh Gao9c869b52016-08-25 16:00:22 -07001416 server_port_str = server_port_str ? server_port_str : getenv("ANDROID_ADB_SERVER_PORT");
Tao Wu135f4ab2016-09-16 13:01:24 -07001417 if (server_port_str && strlen(server_port_str) > 0) {
1418 if (!android::base::ParseInt(server_port_str, &server_port, 1, 65535)) {
1419 fprintf(stderr,
1420 "adb: Env var ANDROID_ADB_SERVER_PORT must be a positive"
1421 " number less than 65535. Got \"%s\"\n",
1422 server_port_str);
1423 exit(1);
1424 }
1425 }
Josh Gao9c869b52016-08-25 16:00:22 -07001426
1427 int rc;
1428 char* temp;
1429 if (server_host_str) {
Tao Wu135f4ab2016-09-16 13:01:24 -07001430 rc = asprintf(&temp, "tcp:%s:%d", server_host_str, server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001431 } else {
Tao Wu135f4ab2016-09-16 13:01:24 -07001432 rc = asprintf(&temp, "tcp:%d", server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001433 }
1434 if (rc < 0) {
1435 fatal("failed to allocate server socket specification");
1436 }
1437 server_socket_str = temp;
1438 }
1439
1440 adb_set_socket_spec(server_socket_str);
1441
Elliott Hughes8d28e192015-10-07 14:55:10 -07001442 // If none of -d, -e, or -s were specified, try $ANDROID_SERIAL.
1443 if (transport_type == kTransportAny && serial == nullptr) {
1444 serial = getenv("ANDROID_SERIAL");
1445 }
1446
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001447 adb_set_transport(transport_type, serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001448
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001449 if (is_server) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001450 if (no_daemon || is_daemon) {
Spencer Low5c398d22015-08-08 15:07:07 -07001451 if (is_daemon && (ack_reply_fd == -1)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001452 fprintf(stderr, "reply fd for adb server to client communication not specified.\n");
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001453 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001454 }
Josh Gao9c869b52016-08-25 16:00:22 -07001455 r = adb_server_main(is_daemon, server_socket_str, ack_reply_fd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001456 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001457 r = launch_server(server_socket_str);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001458 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001459 if (r) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001460 fprintf(stderr,"* could not start server *\n");
1461 }
1462 return r;
1463 }
1464
Riley Andrews98f58e82014-12-05 17:37:24 -08001465 if (argc == 0) {
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001466 help();
1467 return 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001468 }
1469
Riley Andrewsc8514c82014-12-05 17:32:46 -08001470 /* handle wait-for-* prefix */
1471 if (!strncmp(argv[0], "wait-for-", strlen("wait-for-"))) {
Dan Albertbac34742015-02-25 17:51:28 -08001472 const char* service = argv[0];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001473
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001474 if (!wait_for_device(service, transport_type, serial)) {
Riley Andrewsc8514c82014-12-05 17:32:46 -08001475 return 1;
1476 }
1477
Elliott Hughes2b101112015-05-04 19:29:32 -07001478 // Allow a command to be run after wait-for-device,
1479 // e.g. 'adb wait-for-device shell'.
Riley Andrewsc8514c82014-12-05 17:32:46 -08001480 if (argc == 1) {
1481 return 0;
1482 }
1483
1484 /* Fall through */
1485 argc--;
1486 argv++;
1487 }
1488
1489 /* adb_connect() commands */
Riley Andrews98f58e82014-12-05 17:37:24 -08001490 if (!strcmp(argv[0], "devices")) {
Dan Albertbac34742015-02-25 17:51:28 -08001491 const char *listopt;
Elliott Hughes6452a892015-04-29 12:28:13 -07001492 if (argc < 2) {
Scott Andersone109d262012-04-20 11:21:14 -07001493 listopt = "";
Elliott Hughes6452a892015-04-29 12:28:13 -07001494 } else if (argc == 2 && !strcmp(argv[1], "-l")) {
Scott Andersone109d262012-04-20 11:21:14 -07001495 listopt = argv[1];
Elliott Hughes6452a892015-04-29 12:28:13 -07001496 } else {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001497 return syntax_error("adb devices [-l]");
Scott Andersone109d262012-04-20 11:21:14 -07001498 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001499
1500 std::string query = android::base::StringPrintf("host:%s%s", argv[0], listopt);
1501 printf("List of devices attached\n");
1502 return adb_query_command(query);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001503 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001504 else if (!strcmp(argv[0], "connect")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001505 if (argc != 2) return syntax_error("adb connect <host>[:<port>]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001506
1507 std::string query = android::base::StringPrintf("host:connect:%s", argv[1]);
1508 return adb_query_command(query);
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001509 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001510 else if (!strcmp(argv[0], "disconnect")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001511 if (argc > 2) return syntax_error("adb disconnect [<host>[:<port>]]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001512
1513 std::string query = android::base::StringPrintf("host:disconnect:%s",
1514 (argc == 2) ? argv[1] : "");
1515 return adb_query_command(query);
Mike Lockwoodff196702009-08-24 15:58:40 -07001516 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001517 else if (!strcmp(argv[0], "emu")) {
Spencer Low142ec752015-05-06 16:13:42 -07001518 return adb_send_emulator_command(argc, argv, serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001519 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -08001520 else if (!strcmp(argv[0], "shell")) {
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001521 return adb_shell(argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001522 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001523 else if (!strcmp(argv[0], "exec-in") || !strcmp(argv[0], "exec-out")) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001524 int exec_in = !strcmp(argv[0], "exec-in");
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001525
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001526 if (argc < 2) return syntax_error("adb %s command", argv[0]);
Zach Riggleccafc792016-04-14 17:18:04 -04001527
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001528 std::string cmd = "exec:";
1529 cmd += argv[1];
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001530 argc -= 2;
1531 argv += 2;
1532 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001533 cmd += " " + escape_arg(*argv++);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001534 }
1535
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001536 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001537 int fd = adb_connect(cmd, &error);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001538 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001539 fprintf(stderr, "error: %s\n", error.c_str());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001540 return -1;
1541 }
1542
1543 if (exec_in) {
1544 copy_to_file(STDIN_FILENO, fd);
1545 } else {
1546 copy_to_file(fd, STDOUT_FILENO);
1547 }
1548
1549 adb_close(fd);
1550 return 0;
1551 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001552 else if (!strcmp(argv[0], "kill-server")) {
Josh Gaob3c14ec2017-05-08 18:37:17 -07001553 return adb_kill_server() ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001554 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001555 else if (!strcmp(argv[0], "sideload")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001556 if (argc != 2) return syntax_error("sideload requires an argument");
Doug Zongker71fe5842014-06-26 15:35:36 -07001557 if (adb_sideload_host(argv[1])) {
Doug Zongker447f0612012-01-09 14:54:53 -08001558 return 1;
1559 } else {
1560 return 0;
1561 }
1562 }
Elliott Hughes19d80b82015-07-21 16:13:40 -07001563 else if (!strcmp(argv[0], "tcpip") && argc > 1) {
1564 return adb_connect_command(android::base::StringPrintf("tcpip:%s", argv[1]));
1565 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001566 else if (!strcmp(argv[0], "remount") ||
1567 !strcmp(argv[0], "reboot") ||
1568 !strcmp(argv[0], "reboot-bootloader") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001569 !strcmp(argv[0], "usb") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001570 !strcmp(argv[0], "disable-verity") ||
1571 !strcmp(argv[0], "enable-verity")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001572 std::string command;
Tao Bao175b7bb2015-03-29 11:22:34 -07001573 if (!strcmp(argv[0], "reboot-bootloader")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001574 command = "reboot:bootloader";
Tao Bao175b7bb2015-03-29 11:22:34 -07001575 } else if (argc > 1) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001576 command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
Tao Bao175b7bb2015-03-29 11:22:34 -07001577 } else {
Elliott Hughes5677c232015-05-07 23:37:40 -07001578 command = android::base::StringPrintf("%s:", argv[0]);
The Android Open Source Projecte037fd72009-03-13 13:04:37 -07001579 }
Tao Bao175b7bb2015-03-29 11:22:34 -07001580 return adb_connect_command(command);
Josh Gao06c73ae2016-03-04 15:16:18 -08001581 } else if (!strcmp(argv[0], "root") || !strcmp(argv[0], "unroot")) {
1582 return adb_root(argv[0]) ? 0 : 1;
1583 } else if (!strcmp(argv[0], "bugreport")) {
Felipe Leme698e0652016-07-19 17:07:22 -07001584 Bugreport bugreport;
1585 return bugreport.DoIt(transport_type, serial, argc, argv);
Felipe Leme44a42672016-04-01 17:43:27 -07001586 } else if (!strcmp(argv[0], "forward") || !strcmp(argv[0], "reverse")) {
Elliott Hughes424af022015-05-29 17:55:19 -07001587 bool reverse = !strcmp(argv[0], "reverse");
1588 ++argv;
1589 --argc;
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001590 if (argc < 1) return syntax_error("%s requires an argument", argv[0]);
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001591
1592 // Determine the <host-prefix> for this command.
Elliott Hughes424af022015-05-29 17:55:19 -07001593 std::string host_prefix;
David 'Digit' Turner25258692013-03-21 21:07:42 +01001594 if (reverse) {
Elliott Hughes424af022015-05-29 17:55:19 -07001595 host_prefix = "reverse";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001596 } else {
David 'Digit' Turner25258692013-03-21 21:07:42 +01001597 if (serial) {
Elliott Hughes424af022015-05-29 17:55:19 -07001598 host_prefix = android::base::StringPrintf("host-serial:%s", serial);
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001599 } else if (transport_type == kTransportUsb) {
Elliott Hughes424af022015-05-29 17:55:19 -07001600 host_prefix = "host-usb";
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001601 } else if (transport_type == kTransportLocal) {
Elliott Hughes424af022015-05-29 17:55:19 -07001602 host_prefix = "host-local";
David 'Digit' Turner25258692013-03-21 21:07:42 +01001603 } else {
Elliott Hughes424af022015-05-29 17:55:19 -07001604 host_prefix = "host";
David 'Digit' Turner25258692013-03-21 21:07:42 +01001605 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001606 }
1607
David Purselleaae97e2016-04-07 11:25:48 -07001608 std::string cmd, error;
Elliott Hughes424af022015-05-29 17:55:19 -07001609 if (strcmp(argv[0], "--list") == 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001610 if (argc != 1) return syntax_error("--list doesn't take any arguments");
Elliott Hughes424af022015-05-29 17:55:19 -07001611 return adb_query_command(host_prefix + ":list-forward");
1612 } else if (strcmp(argv[0], "--remove-all") == 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001613 if (argc != 1) return syntax_error("--remove-all doesn't take any arguments");
Elliott Hughes424af022015-05-29 17:55:19 -07001614 cmd = host_prefix + ":killforward-all";
1615 } else if (strcmp(argv[0], "--remove") == 0) {
1616 // forward --remove <local>
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001617 if (argc != 2) return syntax_error("--remove requires an argument");
Elliott Hughes424af022015-05-29 17:55:19 -07001618 cmd = host_prefix + ":killforward:" + argv[1];
1619 } else if (strcmp(argv[0], "--no-rebind") == 0) {
1620 // forward --no-rebind <local> <remote>
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001621 if (argc != 3) return syntax_error("--no-rebind takes two arguments");
David Purselleaae97e2016-04-07 11:25:48 -07001622 if (forward_targets_are_valid(argv[1], argv[2], &error)) {
1623 cmd = host_prefix + ":forward:norebind:" + argv[1] + ";" + argv[2];
1624 }
Elliott Hughes424af022015-05-29 17:55:19 -07001625 } else {
1626 // forward <local> <remote>
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001627 if (argc != 2) return syntax_error("forward takes two arguments");
David Purselleaae97e2016-04-07 11:25:48 -07001628 if (forward_targets_are_valid(argv[0], argv[1], &error)) {
1629 cmd = host_prefix + ":forward:" + argv[0] + ";" + argv[1];
1630 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001631 }
1632
David Purselleaae97e2016-04-07 11:25:48 -07001633 if (!error.empty()) {
1634 fprintf(stderr, "error: %s\n", error.c_str());
1635 return 1;
1636 }
1637
1638 int fd = adb_connect(cmd, &error);
1639 if (fd < 0 || !adb_status(fd, &error)) {
1640 adb_close(fd);
1641 fprintf(stderr, "error: %s\n", error.c_str());
1642 return 1;
1643 }
1644
1645 // Server or device may optionally return a resolved TCP port number.
1646 std::string resolved_port;
1647 if (ReadProtocolString(fd, &resolved_port, &error) && !resolved_port.empty()) {
1648 printf("%s\n", resolved_port.c_str());
1649 }
1650
1651 ReadOrderlyShutdown(fd);
1652 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001653 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001654 /* do_sync_*() commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001655 else if (!strcmp(argv[0], "ls")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001656 if (argc != 2) return syntax_error("ls requires an argument");
Elliott Hughesaa245492015-08-03 10:38:08 -07001657 return do_sync_ls(argv[1]) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001658 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001659 else if (!strcmp(argv[0], "push")) {
Josh Gao05786772015-10-30 16:57:19 -07001660 bool copy_attrs = false;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001661 bool sync = false;
Josh Gao05786772015-10-30 16:57:19 -07001662 std::vector<const char*> srcs;
1663 const char* dst = nullptr;
Mark Lindner76f2a932014-03-11 17:55:59 -07001664
Dan Albert06b0d6b2017-05-18 22:56:48 -07001665 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, &sync);
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001666 if (srcs.empty() || !dst) return syntax_error("push requires an argument");
Dan Albert06b0d6b2017-05-18 22:56:48 -07001667 return do_sync_push(srcs, dst, sync) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001668 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001669 else if (!strcmp(argv[0], "pull")) {
Josh Gao05786772015-10-30 16:57:19 -07001670 bool copy_attrs = false;
1671 std::vector<const char*> srcs;
1672 const char* dst = ".";
Mark Lindner76f2a932014-03-11 17:55:59 -07001673
Dan Albert06b0d6b2017-05-18 22:56:48 -07001674 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, nullptr);
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001675 if (srcs.empty()) return syntax_error("pull requires an argument");
Josh Gao05786772015-10-30 16:57:19 -07001676 return do_sync_pull(srcs, dst, copy_attrs) ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001677 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001678 else if (!strcmp(argv[0], "install")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001679 if (argc < 2) return syntax_error("install requires an argument");
Todd Kennedy248722e2016-07-20 16:22:37 -07001680 if (_use_legacy_install()) {
1681 return install_app_legacy(transport_type, serial, argc, argv);
David Pursell0aacbbe2016-01-21 19:56:50 -08001682 }
Todd Kennedy248722e2016-07-20 16:22:37 -07001683 return install_app(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001684 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001685 else if (!strcmp(argv[0], "install-multiple")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001686 if (argc < 2) return syntax_error("install-multiple requires an argument");
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001687 return install_multiple_app(transport_type, serial, argc, argv);
Jeff Sharkey960df972014-06-09 17:30:57 -07001688 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001689 else if (!strcmp(argv[0], "uninstall")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001690 if (argc < 2) return syntax_error("uninstall requires an argument");
Todd Kennedy248722e2016-07-20 16:22:37 -07001691 if (_use_legacy_install()) {
1692 return uninstall_app_legacy(transport_type, serial, argc, argv);
David Pursell0aacbbe2016-01-21 19:56:50 -08001693 }
Todd Kennedy248722e2016-07-20 16:22:37 -07001694 return uninstall_app(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001695 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001696 else if (!strcmp(argv[0], "sync")) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001697 std::string src;
Elliott Hughes58305772015-04-17 13:57:15 -07001698 bool list_only = false;
Riley Andrews98f58e82014-12-05 17:37:24 -08001699 if (argc < 2) {
Elliott Hughes58305772015-04-17 13:57:15 -07001700 // No local path was specified.
Elliott Hughesd236d072015-04-21 10:17:07 -07001701 src = "";
Anthony Newnam705c9442010-02-22 08:36:49 -06001702 } else if (argc >= 2 && strcmp(argv[1], "-l") == 0) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001703 list_only = true;
Anthony Newnam705c9442010-02-22 08:36:49 -06001704 if (argc == 3) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001705 src = argv[2];
Anthony Newnam705c9442010-02-22 08:36:49 -06001706 } else {
Elliott Hughesd236d072015-04-21 10:17:07 -07001707 src = "";
Anthony Newnam705c9442010-02-22 08:36:49 -06001708 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001709 } else if (argc == 2) {
Elliott Hughes58305772015-04-17 13:57:15 -07001710 // A local path or "android"/"data" arg was specified.
Elliott Hughesd236d072015-04-21 10:17:07 -07001711 src = argv[1];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001712 } else {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001713 return syntax_error("adb sync [-l] [PARTITION]");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001714 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001715
Elliott Hughes0754cb92017-05-01 11:04:31 -07001716 if (src == "all") src = "";
1717
Elliott Hughesd236d072015-04-21 10:17:07 -07001718 if (src != "" &&
1719 src != "system" && src != "data" && src != "vendor" && src != "oem") {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001720 return syntax_error("don't know how to sync %s partition", src.c_str());
Elliott Hughes58305772015-04-17 13:57:15 -07001721 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001722
Elliott Hughes58305772015-04-17 13:57:15 -07001723 std::string system_src_path = product_file("system");
1724 std::string data_src_path = product_file("data");
1725 std::string vendor_src_path = product_file("vendor");
1726 std::string oem_src_path = product_file("oem");
Elliott Hughes58305772015-04-17 13:57:15 -07001727
Elliott Hughesaa245492015-08-03 10:38:08 -07001728 bool okay = true;
1729 if (okay && (src.empty() || src == "system")) {
1730 okay = do_sync_sync(system_src_path, "/system", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001731 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001732 if (okay && (src.empty() || src == "vendor") && directory_exists(vendor_src_path)) {
1733 okay = do_sync_sync(vendor_src_path, "/vendor", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001734 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001735 if (okay && (src.empty() || src == "oem") && directory_exists(oem_src_path)) {
1736 okay = do_sync_sync(oem_src_path, "/oem", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001737 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001738 if (okay && (src.empty() || src == "data")) {
1739 okay = do_sync_sync(data_src_path, "/data", list_only);
Elliott Hughes58305772015-04-17 13:57:15 -07001740 }
Elliott Hughesaa245492015-08-03 10:38:08 -07001741 return okay ? 0 : 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001742 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001743 /* passthrough commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001744 else if (!strcmp(argv[0],"get-state") ||
Scott Andersone109d262012-04-20 11:21:14 -07001745 !strcmp(argv[0],"get-serialno") ||
1746 !strcmp(argv[0],"get-devpath"))
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001747 {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001748 return adb_query_command(format_host_command(argv[0], transport_type, serial));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001749 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001750 /* other commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001751 else if (!strcmp(argv[0],"logcat") || !strcmp(argv[0],"lolcat") || !strcmp(argv[0],"longcat")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001752 return logcat(transport_type, serial, argc, argv);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001753 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001754 else if (!strcmp(argv[0],"ppp")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001755 return ppp(argc, argv);
1756 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001757 else if (!strcmp(argv[0], "start-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001758 std::string error;
Spencer Lowf18fc082015-08-11 17:05:02 -07001759 const int result = adb_connect("host:start-server", &error);
1760 if (result < 0) {
1761 fprintf(stderr, "error: %s\n", error.c_str());
1762 }
1763 return result;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001764 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001765 else if (!strcmp(argv[0], "backup")) {
Christopher Tated2f54152011-04-21 12:53:28 -07001766 return backup(argc, argv);
1767 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001768 else if (!strcmp(argv[0], "restore")) {
Christopher Tate702967a2011-05-17 15:52:54 -07001769 return restore(argc, argv);
1770 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001771 else if (!strcmp(argv[0], "keygen")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001772 if (argc != 2) return syntax_error("keygen requires an argument");
Yabin Cuiaed3c612015-09-22 15:52:57 -07001773 // Always print key generation information for keygen command.
1774 adb_trace_enable(AUTH);
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08001775 return adb_auth_keygen(argv[1]);
1776 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001777 else if (!strcmp(argv[0], "jdwp")) {
Tao Bao175b7bb2015-03-29 11:22:34 -07001778 return adb_connect_command("jdwp");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001779 }
Josh Gaob463baf2016-05-13 18:18:23 -07001780 else if (!strcmp(argv[0], "track-jdwp")) {
1781 return adb_connect_command("track-jdwp");
1782 }
1783 else if (!strcmp(argv[0], "track-devices")) {
1784 return adb_connect_command("host:track-devices");
1785 }
1786
1787
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001788 /* "adb /?" is a common idiom under Windows */
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001789 else if (!strcmp(argv[0], "--help") || !strcmp(argv[0], "help") || !strcmp(argv[0], "/?")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001790 help();
1791 return 0;
1792 }
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001793 else if (!strcmp(argv[0], "--version") || !strcmp(argv[0], "version")) {
Elliott Hughes42ae2602015-08-12 08:32:10 -07001794 fprintf(stdout, "%s", adb_version().c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001795 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08001796 } else if (!strcmp(argv[0], "features")) {
David Pursell4e2fd362015-09-22 10:43:08 -07001797 // Only list the features common to both the adb client and the device.
David Pursell0aacbbe2016-01-21 19:56:50 -08001798 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001799 std::string error;
1800 if (!adb_get_feature_set(&features, &error)) {
1801 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08001802 return 1;
1803 }
1804
David Pursell4e2fd362015-09-22 10:43:08 -07001805 for (const std::string& name : features) {
David Pursell70ef7b42015-09-30 13:35:42 -07001806 if (CanUseFeature(features, name)) {
David Pursell4e2fd362015-09-22 10:43:08 -07001807 printf("%s\n", name.c_str());
1808 }
1809 }
1810 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08001811 } else if (!strcmp(argv[0], "host-features")) {
1812 return adb_query_command("host:host-features");
Yabin Cui1f4ec192016-04-05 13:50:44 -07001813 } else if (!strcmp(argv[0], "reconnect")) {
1814 if (argc == 1) {
Yabin Cuib5e11412017-03-10 16:01:01 -08001815 return adb_query_command(format_host_command(argv[0], transport_type, serial));
Josh Gao22d2b3e2016-10-27 14:01:08 -07001816 } else if (argc == 2) {
1817 if (!strcmp(argv[1], "device")) {
1818 std::string err;
1819 adb_connect("reconnect", &err);
1820 return 0;
1821 } else if (!strcmp(argv[1], "offline")) {
1822 std::string err;
1823 return adb_query_command("host:reconnect-offline");
1824 } else {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001825 return syntax_error("adb reconnect [device|offline]");
Josh Gao22d2b3e2016-10-27 14:01:08 -07001826 }
Yabin Cui1f4ec192016-04-05 13:50:44 -07001827 }
Dan Albert90d4b732015-05-20 18:58:41 -07001828 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001829
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001830 syntax_error("unknown command %s", argv[0]);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001831 return 1;
1832}
1833
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001834static int uninstall_app(TransportType transport, const char* serial, int argc, const char** argv) {
1835 // 'adb uninstall' takes the same arguments as 'cmd package uninstall' on device
1836 std::string cmd = "cmd package";
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001837 while (argc-- > 0) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001838 // deny the '-k' option until the remaining data/cache can be removed with adb/UI
1839 if (strcmp(*argv, "-k") == 0) {
1840 printf(
1841 "The -k option uninstalls the application while retaining the data/cache.\n"
1842 "At the moment, there is no way to remove the remaining data.\n"
1843 "You will have to reinstall the application with the same signature, and fully uninstall it.\n"
1844 "If you truly wish to continue, execute 'adb shell cmd package uninstall -k'.\n");
1845 return EXIT_FAILURE;
1846 }
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001847 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001848 }
1849
David Pursell70ef7b42015-09-30 13:35:42 -07001850 return send_shell_command(transport, serial, cmd, false);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001851}
1852
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001853static int install_app(TransportType transport, const char* serial, int argc, const char** argv) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001854 // The last argument must be the APK file
1855 const char* file = argv[argc - 1];
Elliott Hughes72884782016-10-24 18:29:21 -07001856 if (!android::base::EndsWithIgnoreCase(file, ".apk")) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001857 return syntax_error("filename doesn't end .apk: %s", file);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001858 }
1859
Elliott Hughes72884782016-10-24 18:29:21 -07001860 struct stat sb;
1861 if (stat(file, &sb) == -1) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001862 fprintf(stderr, "adb: failed to stat %s: %s\n", file, strerror(errno));
Elliott Hughes72884782016-10-24 18:29:21 -07001863 return 1;
Kenny Root597ea5b2011-08-05 11:19:45 -07001864 }
1865
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001866 int localFd = adb_open(file, O_RDONLY);
1867 if (localFd < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001868 fprintf(stderr, "adb: failed to open %s: %s\n", file, strerror(errno));
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001869 return 1;
1870 }
Kenny Root597ea5b2011-08-05 11:19:45 -07001871
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001872 std::string error;
1873 std::string cmd = "exec:cmd package";
1874
1875 // don't copy the APK name, but, copy the rest of the arguments as-is
1876 while (argc-- > 1) {
1877 cmd += " " + escape_arg(std::string(*argv++));
1878 }
1879
1880 // add size parameter [required for streaming installs]
1881 // do last to override any user specified value
1882 cmd += " " + android::base::StringPrintf("-S %" PRIu64, static_cast<uint64_t>(sb.st_size));
1883
1884 int remoteFd = adb_connect(cmd, &error);
1885 if (remoteFd < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001886 fprintf(stderr, "adb: connect error for write: %s\n", error.c_str());
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001887 adb_close(localFd);
1888 return 1;
1889 }
1890
1891 char buf[BUFSIZ];
1892 copy_to_file(localFd, remoteFd);
1893 read_status_line(remoteFd, buf, sizeof(buf));
1894
1895 adb_close(localFd);
1896 adb_close(remoteFd);
1897
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001898 if (!strncmp("Success", buf, 7)) {
1899 fputs(buf, stdout);
1900 return 0;
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001901 }
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001902 fprintf(stderr, "adb: failed to install %s: %s", file, buf);
1903 return 1;
Jeff Sharkey960df972014-06-09 17:30:57 -07001904}
1905
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001906static int install_multiple_app(TransportType transport, const char* serial, int argc,
Elliott Hughes58305772015-04-17 13:57:15 -07001907 const char** argv)
Jeff Sharkey960df972014-06-09 17:30:57 -07001908{
Jeff Sharkey960df972014-06-09 17:30:57 -07001909 // Find all APK arguments starting at end.
1910 // All other arguments passed through verbatim.
1911 int first_apk = -1;
Elliott Hughes72884782016-10-24 18:29:21 -07001912 uint64_t total_size = 0;
1913 for (int i = argc - 1; i >= 0; i--) {
Dan Albertbac34742015-02-25 17:51:28 -08001914 const char* file = argv[i];
Jeff Sharkey960df972014-06-09 17:30:57 -07001915
Elliott Hughes72884782016-10-24 18:29:21 -07001916 if (android::base::EndsWithIgnoreCase(file, ".apk")) {
1917 struct stat sb;
1918 if (stat(file, &sb) != -1) total_size += sb.st_size;
Jeff Sharkey960df972014-06-09 17:30:57 -07001919 first_apk = i;
1920 } else {
1921 break;
1922 }
Kenny Root597ea5b2011-08-05 11:19:45 -07001923 }
1924
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001925 if (first_apk == -1) return syntax_error("need APK file on command line");
Kenny Root597ea5b2011-08-05 11:19:45 -07001926
Todd Kennedy248722e2016-07-20 16:22:37 -07001927 std::string install_cmd;
1928 if (_use_legacy_install()) {
1929 install_cmd = "exec:pm";
1930 } else {
1931 install_cmd = "exec:cmd package";
1932 }
1933
1934 std::string cmd = android::base::StringPrintf("%s install-create -S %" PRIu64, install_cmd.c_str(), total_size);
Elliott Hughes72884782016-10-24 18:29:21 -07001935 for (int i = 1; i < first_apk; i++) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001936 cmd += " " + escape_arg(argv[i]);
Jeff Sharkey960df972014-06-09 17:30:57 -07001937 }
1938
1939 // Create install session
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001940 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001941 int fd = adb_connect(cmd, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07001942 if (fd < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001943 fprintf(stderr, "adb: connect error for create: %s\n", error.c_str());
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001944 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07001945 }
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001946 char buf[BUFSIZ];
Jeff Sharkey960df972014-06-09 17:30:57 -07001947 read_status_line(fd, buf, sizeof(buf));
1948 adb_close(fd);
1949
1950 int session_id = -1;
1951 if (!strncmp("Success", buf, 7)) {
1952 char* start = strrchr(buf, '[');
1953 char* end = strrchr(buf, ']');
1954 if (start && end) {
1955 *end = '\0';
1956 session_id = strtol(start + 1, NULL, 10);
1957 }
1958 }
1959 if (session_id < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001960 fprintf(stderr, "adb: failed to create session\n");
Christopher Tate71bbc672014-07-14 16:45:13 -07001961 fputs(buf, stderr);
Todd Kennedy6fa848a2015-11-03 16:53:08 -08001962 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07001963 }
1964
1965 // Valid session, now stream the APKs
1966 int success = 1;
Elliott Hughes72884782016-10-24 18:29:21 -07001967 for (int i = first_apk; i < argc; i++) {
Dan Albertbac34742015-02-25 17:51:28 -08001968 const char* file = argv[i];
Elliott Hughes72884782016-10-24 18:29:21 -07001969 struct stat sb;
Jeff Sharkey960df972014-06-09 17:30:57 -07001970 if (stat(file, &sb) == -1) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001971 fprintf(stderr, "adb: failed to stat %s: %s\n", file, strerror(errno));
Jeff Sharkey960df972014-06-09 17:30:57 -07001972 success = 0;
1973 goto finalize_session;
1974 }
1975
Elliott Hughes2940ccf2015-04-17 14:07:52 -07001976 std::string cmd = android::base::StringPrintf(
Todd Kennedy248722e2016-07-20 16:22:37 -07001977 "%s install-write -S %" PRIu64 " %d %d_%s -",
Colin Cross58021d12017-02-23 21:23:05 -08001978 install_cmd.c_str(), static_cast<uint64_t>(sb.st_size), session_id, i,
1979 android::base::Basename(file).c_str());
Jeff Sharkey960df972014-06-09 17:30:57 -07001980
1981 int localFd = adb_open(file, O_RDONLY);
1982 if (localFd < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001983 fprintf(stderr, "adb: failed to open %s: %s\n", file, strerror(errno));
Jeff Sharkey960df972014-06-09 17:30:57 -07001984 success = 0;
1985 goto finalize_session;
1986 }
1987
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001988 std::string error;
Elliott Hughes6452a892015-04-29 12:28:13 -07001989 int remoteFd = adb_connect(cmd, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07001990 if (remoteFd < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001991 fprintf(stderr, "adb: connect error for write: %s\n", error.c_str());
Jeff Sharkey960df972014-06-09 17:30:57 -07001992 adb_close(localFd);
1993 success = 0;
1994 goto finalize_session;
1995 }
1996
1997 copy_to_file(localFd, remoteFd);
1998 read_status_line(remoteFd, buf, sizeof(buf));
1999
2000 adb_close(localFd);
2001 adb_close(remoteFd);
2002
2003 if (strncmp("Success", buf, 7)) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07002004 fprintf(stderr, "adb: failed to write %s\n", file);
Christopher Tate71bbc672014-07-14 16:45:13 -07002005 fputs(buf, stderr);
Jeff Sharkey960df972014-06-09 17:30:57 -07002006 success = 0;
2007 goto finalize_session;
2008 }
2009 }
2010
2011finalize_session:
Jeff Sharkeyac77e1f2014-07-25 09:58:25 -07002012 // Commit session if we streamed everything okay; otherwise abandon
Elliott Hughes6452a892015-04-29 12:28:13 -07002013 std::string service =
Todd Kennedy248722e2016-07-20 16:22:37 -07002014 android::base::StringPrintf("%s install-%s %d",
2015 install_cmd.c_str(), success ? "commit" : "abandon", session_id);
Elliott Hughes6452a892015-04-29 12:28:13 -07002016 fd = adb_connect(service, &error);
Jeff Sharkey960df972014-06-09 17:30:57 -07002017 if (fd < 0) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07002018 fprintf(stderr, "adb: connect error for finalize: %s\n", error.c_str());
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002019 return EXIT_FAILURE;
Jeff Sharkey960df972014-06-09 17:30:57 -07002020 }
2021 read_status_line(fd, buf, sizeof(buf));
2022 adb_close(fd);
2023
2024 if (!strncmp("Success", buf, 7)) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07002025 fputs(buf, stdout);
Jeff Sharkey960df972014-06-09 17:30:57 -07002026 return 0;
Jeff Sharkey960df972014-06-09 17:30:57 -07002027 }
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07002028 fprintf(stderr, "adb: failed to finalize session\n");
2029 fputs(buf, stderr);
2030 return EXIT_FAILURE;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002031}
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002032
2033static int pm_command(TransportType transport, const char* serial, int argc, const char** argv) {
2034 std::string cmd = "pm";
2035
2036 while (argc-- > 0) {
2037 cmd += " " + escape_arg(*argv++);
2038 }
2039
2040 return send_shell_command(transport, serial, cmd, false);
2041}
2042
2043static int uninstall_app_legacy(TransportType transport, const char* serial, int argc, const char** argv) {
2044 /* if the user choose the -k option, we refuse to do it until devices are
2045 out with the option to uninstall the remaining data somehow (adb/ui) */
2046 int i;
2047 for (i = 1; i < argc; i++) {
2048 if (!strcmp(argv[i], "-k")) {
2049 printf(
2050 "The -k option uninstalls the application while retaining the data/cache.\n"
2051 "At the moment, there is no way to remove the remaining data.\n"
2052 "You will have to reinstall the application with the same signature, and fully uninstall it.\n"
2053 "If you truly wish to continue, execute 'adb shell pm uninstall -k'\n.");
2054 return EXIT_FAILURE;
2055 }
2056 }
2057
2058 /* 'adb uninstall' takes the same arguments as 'pm uninstall' on device */
2059 return pm_command(transport, serial, argc, argv);
2060}
2061
2062static int delete_file(TransportType transport, const char* serial, const std::string& filename) {
2063 std::string cmd = "rm -f " + escape_arg(filename);
2064 return send_shell_command(transport, serial, cmd, false);
2065}
2066
2067static int install_app_legacy(TransportType transport, const char* serial, int argc, const char** argv) {
2068 static const char *const DATA_DEST = "/data/local/tmp/%s";
2069 static const char *const SD_DEST = "/sdcard/tmp/%s";
2070 const char* where = DATA_DEST;
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002071
Elliott Hughes72884782016-10-24 18:29:21 -07002072 for (int i = 1; i < argc; i++) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002073 if (!strcmp(argv[i], "-s")) {
2074 where = SD_DEST;
2075 }
2076 }
2077
2078 // Find last APK argument.
2079 // All other arguments passed through verbatim.
2080 int last_apk = -1;
Elliott Hughes72884782016-10-24 18:29:21 -07002081 for (int i = argc - 1; i >= 0; i--) {
2082 if (android::base::EndsWithIgnoreCase(argv[i], ".apk")) {
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002083 last_apk = i;
2084 break;
2085 }
2086 }
2087
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07002088 if (last_apk == -1) return syntax_error("need APK file on command line");
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002089
2090 int result = -1;
2091 std::vector<const char*> apk_file = {argv[last_apk]};
2092 std::string apk_dest = android::base::StringPrintf(
Colin Cross58021d12017-02-23 21:23:05 -08002093 where, android::base::Basename(argv[last_apk]).c_str());
Dan Albert06b0d6b2017-05-18 22:56:48 -07002094 if (!do_sync_push(apk_file, apk_dest.c_str(), false)) goto cleanup_apk;
Todd Kennedy6fa848a2015-11-03 16:53:08 -08002095 argv[last_apk] = apk_dest.c_str(); /* destination name, not source location */
2096 result = pm_command(transport, serial, argc, argv);
2097
2098cleanup_apk:
2099 delete_file(transport, serial, apk_dest);
2100 return result;
2101}