blob: 0ffdbc28b765156a3169cc29c085503b15e6ae7a [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"
Idries Hamadied409ea2018-01-29 16:30:36 +000055#include "adb_install.h"
Dan Albertcc731cc2015-02-24 21:26:58 -080056#include "adb_io.h"
Josh Gao924d35a2016-08-30 15:39:25 -070057#include "adb_unique_fd.h"
Elliott Hughes58305772015-04-17 13:57:15 -070058#include "adb_utils.h"
Felipe Leme698e0652016-07-19 17:07:22 -070059#include "bugreport.h"
Josh Gaoedec0332018-07-25 15:55:25 -070060#include "client/file_sync_client.h"
Felipe Leme698e0652016-07-19 17:07:22 -070061#include "commandline.h"
Idries Hamadied409ea2018-01-29 16:30:36 +000062#include "fastdeploy.h"
David Pursell70ef7b42015-09-30 13:35:42 -070063#include "services.h"
Josh Gao86acb862018-07-25 16:07:26 -070064#include "shell_protocol.h"
Josh Gao4602adb2016-11-15 18:55:47 -080065#include "sysdeps/chrono.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080066
Matt Gumbeld7b33082012-11-14 10:16:17 -080067extern int gListenAll;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080068
Felipe Leme0d4f0502016-07-26 12:14:39 -070069DefaultStandardStreamsCallback DEFAULT_STANDARD_STREAMS_CALLBACK(nullptr, nullptr);
70
Elliott Hughes57e97cb2018-04-03 10:54:39 -070071static std::string product_file(const std::string& file) {
Elliott Hughes0754cb92017-05-01 11:04:31 -070072 const char* ANDROID_PRODUCT_OUT = getenv("ANDROID_PRODUCT_OUT");
73 if (ANDROID_PRODUCT_OUT == nullptr) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -070074 error_exit("product directory not specified; set $ANDROID_PRODUCT_OUT");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080075 }
Elliott Hughes57e97cb2018-04-03 10:54:39 -070076 return std::string{ANDROID_PRODUCT_OUT} + OS_PATH_SEPARATOR_STR + file;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080077}
78
Elliott Hughes58305772015-04-17 13:57:15 -070079static void help() {
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070080 fprintf(stdout, "%s\n", adb_version().c_str());
Felipe Lemeb3239722016-07-29 17:57:00 -070081 // clang-format off
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070082 fprintf(stdout,
83 "global options:\n"
84 " -a listen on all network interfaces, not just localhost\n"
85 " -d use USB device (error if multiple devices connected)\n"
Elliott Hughes83ab5c22017-01-13 16:58:25 -080086 " -e use TCP/IP device (error if multiple TCP/IP devices available)\n"
Elliott Hughes0754cb92017-05-01 11:04:31 -070087 " -s SERIAL use device with given serial (overrides $ANDROID_SERIAL)\n"
Josh Gaob122b172017-08-16 16:57:01 -070088 " -t ID use device with given transport id\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070089 " -H name of adb server host [default=localhost]\n"
90 " -P port of adb server [default=5037]\n"
91 " -L SOCKET listen on given socket for adb server [default=tcp:localhost:5037]\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080092 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070093 "general commands:\n"
94 " devices [-l] list connected devices (-l for long output)\n"
95 " help show this help message\n"
96 " version show version num\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -070097 "\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080098 "networking:\n"
Elliott Hughes573e67c2019-07-30 13:51:03 -070099 " connect HOST[:PORT] connect to a device via TCP/IP\n"
100 " disconnect [[HOST]:PORT] disconnect from given TCP/IP device, or all\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700101 " forward --list list all forward socket connections\n"
102 " forward [--no-rebind] LOCAL REMOTE\n"
103 " forward socket connection using:\n"
104 " tcp:<port> (<local> may be \"tcp:0\" to pick any open port)\n"
105 " localabstract:<unix domain socket name>\n"
106 " localreserved:<unix domain socket name>\n"
107 " localfilesystem:<unix domain socket name>\n"
108 " dev:<character device name>\n"
109 " jdwp:<process pid> (remote only)\n"
110 " forward --remove LOCAL remove specific forward socket connection\n"
111 " forward --remove-all remove all forward socket connections\n"
112 " ppp TTY [PARAMETER...] run PPP over USB\n"
113 " reverse --list list all reverse socket connections from device\n"
114 " reverse [--no-rebind] REMOTE LOCAL\n"
115 " reverse socket connection using:\n"
116 " tcp:<port> (<remote> may be \"tcp:0\" to pick any open port)\n"
117 " localabstract:<unix domain socket name>\n"
118 " localreserved:<unix domain socket name>\n"
119 " localfilesystem:<unix domain socket name>\n"
120 " reverse --remove REMOTE remove specific reverse socket connection\n"
121 " reverse --remove-all remove all reverse socket connections from device\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800122 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700123 "file transfer:\n"
Dan Albert06b0d6b2017-05-18 22:56:48 -0700124 " push [--sync] LOCAL... REMOTE\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700125 " copy local files/directories to device\n"
Dan Albert06b0d6b2017-05-18 22:56:48 -0700126 " --sync: only push files that are newer on the host than the device\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700127 " pull [-a] REMOTE... LOCAL\n"
128 " copy files/dirs from device\n"
129 " -a: preserve file timestamp and mode\n"
Justin Yun5d0ac622019-06-26 09:18:03 +0900130 " sync [all|data|odm|oem|product|system|system_ext|vendor]\n"
Elliott Hughes0754cb92017-05-01 11:04:31 -0700131 " sync a local build from $ANDROID_PRODUCT_OUT to the device (default all)\n"
Elliott Hughes4c8ea892019-07-30 10:32:53 -0700132 " -l: list files that would be copied, but don't copy them\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800133 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700134 "shell:\n"
135 " shell [-e ESCAPE] [-n] [-Tt] [-x] [COMMAND...]\n"
136 " run remote shell command (interactive shell if no command given)\n"
137 " -e: choose escape character, or \"none\"; default '~'\n"
138 " -n: don't read from stdin\n"
139 " -T: disable PTY allocation\n"
140 " -t: force PTY allocation\n"
141 " -x: disable remote exit codes and stdout/stderr separation\n"
142 " emu COMMAND run emulator console command\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800143 "\n"
Elliott Hughes0d4bf802019-01-11 13:50:05 -0800144 "app installation (see also `adb shell cmd package help`):\n"
Felipe Leme807be772018-05-08 18:40:18 -0700145 " install [-lrtsdg] [--instant] PACKAGE\n"
Patrick Baumanndc58b0a2018-10-09 10:45:03 -0700146 " push a single package to the device and install it\n"
Felipe Leme807be772018-05-08 18:40:18 -0700147 " install-multiple [-lrtsdpg] [--instant] PACKAGE...\n"
Patrick Baumanndc58b0a2018-10-09 10:45:03 -0700148 " push multiple APKs to the device for a single package and install them\n"
149 " install-multi-package [-lrtsdpg] [--instant] PACKAGE...\n"
150 " push one or more packages to the device and install them atomically\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700151 " -r: replace existing application\n"
152 " -t: allow test packages\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700153 " -d: allow version code downgrade (debuggable packages only)\n"
154 " -p: partial application install (install-multiple only)\n"
155 " -g: grant all runtime permissions\n"
Elliott Hughesc0ebd592019-08-02 17:09:30 -0700156 " --abi ABI: override platform's default ABI\n"
Felipe Leme807be772018-05-08 18:40:18 -0700157 " --instant: cause the app to be installed as an ephemeral install app\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000158 " --no-streaming: always push APK to device and invoke Package Manager as separate steps\n"
159 " --streaming: force streaming APK directly into Package Manager\n"
Henry Daitxee01c802018-12-12 10:40:57 +0000160 " --fastdeploy: use fast deploy\n"
161 " --no-fastdeploy: prevent use of fast deploy\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000162 " --force-agent: force update of deployment agent when using fast deploy\n"
163 " --date-check-agent: update deployment agent when local version is newer and using fast deploy\n"
164 " --version-check-agent: update deployment agent when local version has different version code and using fast deploy\n"
165#ifndef _WIN32
166 " --local-agent: locate agent files from local source build (instead of SDK location)\n"
167#endif
Elliott Hughesc0ebd592019-08-02 17:09:30 -0700168 " (See also `adb shell pm help` for more options.)\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000169 //TODO--installlog <filename>
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700170 " uninstall [-k] PACKAGE\n"
171 " remove this app package from the device\n"
172 " '-k': keep the data and cache directories\n"
173 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700174 "debugging:\n"
175 " bugreport [PATH]\n"
176 " write bugreport to given PATH [default=bugreport.zip];\n"
177 " if PATH is a directory, the bug report is saved in that directory.\n"
178 " devices that don't support zipped bug reports output to stdout.\n"
179 " jdwp list pids of processes hosting a JDWP transport\n"
180 " logcat show device log (logcat --help for more)\n"
181 "\n"
182 "security:\n"
183 " disable-verity disable dm-verity checking on userdebug builds\n"
184 " enable-verity re-enable dm-verity checking on userdebug builds\n"
185 " keygen FILE\n"
186 " generate adb public/private key; private key stored in FILE,\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700187 "\n"
188 "scripting:\n"
189 " wait-for[-TRANSPORT]-STATE\n"
190 " wait for device to be in the given state\n"
Tao Bao55d407e2019-04-07 23:24:03 -0700191 " STATE: device, recovery, rescue, sideload, bootloader, or disconnect\n"
Josh Gao1e9e4712019-02-22 14:01:36 -0800192 " TRANSPORT: usb, local, or any [default=any]\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700193 " get-state print offline | bootloader | device\n"
194 " get-serialno print <serial-number>\n"
195 " get-devpath print <device-path>\n"
David Andersona5db1d72018-11-21 13:43:22 -0800196 " remount [-R]\n"
197 " remount partitions read-write. if a reboot is required, -R will\n"
198 " will automatically reboot the device.\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700199 " reboot [bootloader|recovery|sideload|sideload-auto-reboot]\n"
200 " reboot the device; defaults to booting system image but\n"
201 " supports bootloader and recovery too. sideload reboots\n"
202 " into recovery and automatically starts sideload mode,\n"
203 " sideload-auto-reboot is the same but reboots after sideloading.\n"
204 " sideload OTAPACKAGE sideload the given full OTA package\n"
205 " root restart adbd with root permissions\n"
206 " unroot restart adbd without root permissions\n"
bohuea0c3cb2018-10-24 10:38:33 -0700207 " usb restart adbd listening on USB\n"
208 " tcpip PORT restart adbd listening on TCP on PORT\n"
Timcd643152010-02-16 20:18:29 +0000209 "\n"
Yabin Cui1f4ec192016-04-05 13:50:44 -0700210 "internal debugging:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700211 " start-server ensure that there is a server running\n"
212 " kill-server kill the server if it is running\n"
213 " reconnect kick connection from host side to force reconnect\n"
214 " reconnect device kick connection from device side to force reconnect\n"
Yabin Cuib5e11412017-03-10 16:01:01 -0800215 " reconnect offline reset offline/unauthorized devices to force reconnect\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700216 "\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700217 "environment variables:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700218 " $ADB_TRACE\n"
219 " comma-separated list of debug info to log:\n"
220 " all,adb,sockets,packets,rwx,usb,sync,sysdeps,transport,jdwp\n"
221 " $ADB_VENDOR_KEYS colon-separated list of keys (files or directories)\n"
222 " $ANDROID_SERIAL serial number to connect to (see -s)\n"
Tim Baverstock25173c82019-02-08 16:27:16 +0000223 " $ANDROID_LOG_TAGS tags to be used by logcat (see logcat --help)\n"
224 " $ADB_LOCAL_TRANSPORT_MAX_PORT max emulator scan port (default 5585, 16 emus)\n"
225 );
Felipe Lemeb3239722016-07-29 17:57:00 -0700226 // clang-format on
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800227}
228
Yabin Cuid325e862014-11-17 14:48:25 -0800229#if defined(_WIN32)
230
Elliott Hughesa2f2e562015-04-16 16:47:02 -0700231// Implemented in sysdeps_win32.cpp.
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800232void stdin_raw_init();
233void stdin_raw_restore();
Yabin Cuid325e862014-11-17 14:48:25 -0800234
235#else
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100236static termios g_saved_terminal_state;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800237
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800238static void stdin_raw_init() {
239 if (tcgetattr(STDIN_FILENO, &g_saved_terminal_state)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800240
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100241 termios tio;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800242 if (tcgetattr(STDIN_FILENO, &tio)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800243
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100244 cfmakeraw(&tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800245
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100246 // No timeout but request at least one character per read.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800247 tio.c_cc[VTIME] = 0;
248 tio.c_cc[VMIN] = 1;
249
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800250 tcsetattr(STDIN_FILENO, TCSAFLUSH, &tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800251}
252
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800253static void stdin_raw_restore() {
254 tcsetattr(STDIN_FILENO, TCSAFLUSH, &g_saved_terminal_state);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800255}
256#endif
257
Alex Buynytskyy665f3ff2019-09-16 12:10:54 -0700258int read_and_dump(borrowed_fd fd, bool use_shell_protocol,
259 StandardStreamsCallbackInterface* callback) {
David Pursell606835a2015-09-08 17:17:02 -0700260 int exit_code = 0;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700261 if (fd < 0) return exit_code;
262
David Pursell606835a2015-09-08 17:17:02 -0700263 std::unique_ptr<ShellProtocol> protocol;
264 int length = 0;
David Pursell606835a2015-09-08 17:17:02 -0700265
266 char raw_buffer[BUFSIZ];
267 char* buffer_ptr = raw_buffer;
268 if (use_shell_protocol) {
Josh Gao31b5be62018-03-07 16:51:08 -0800269 protocol = std::make_unique<ShellProtocol>(fd);
David Pursell606835a2015-09-08 17:17:02 -0700270 if (!protocol) {
271 LOG(ERROR) << "failed to allocate memory for ShellProtocol object";
272 return 1;
273 }
274 buffer_ptr = protocol->data();
275 }
276
Felipe Lemed1dd1252016-06-06 16:05:36 -0700277 while (true) {
David Pursell606835a2015-09-08 17:17:02 -0700278 if (use_shell_protocol) {
279 if (!protocol->Read()) {
280 break;
281 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700282 length = protocol->data_length();
David Pursell606835a2015-09-08 17:17:02 -0700283 switch (protocol->id()) {
284 case ShellProtocol::kIdStdout:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700285 callback->OnStdout(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700286 break;
287 case ShellProtocol::kIdStderr:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700288 callback->OnStderr(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700289 break;
290 case ShellProtocol::kIdExit:
Josh Gao1e1ae452019-03-15 14:49:25 -0700291 // data() returns a char* which doesn't have defined signedness.
292 // Cast to uint8_t to prevent 255 from being sign extended to INT_MIN,
293 // which doesn't get truncated on Windows.
294 exit_code = static_cast<uint8_t>(protocol->data()[0]);
David Pursell606835a2015-09-08 17:17:02 -0700295 continue;
296 default:
297 continue;
298 }
299 length = protocol->data_length();
300 } else {
Josh Gao27241a72019-04-25 14:04:57 -0700301 D("read_and_dump(): pre adb_read(fd=%d)", fd.get());
David Pursell606835a2015-09-08 17:17:02 -0700302 length = adb_read(fd, raw_buffer, sizeof(raw_buffer));
Josh Gao27241a72019-04-25 14:04:57 -0700303 D("read_and_dump(): post adb_read(fd=%d): length=%d", fd.get(), length);
David Pursell606835a2015-09-08 17:17:02 -0700304 if (length <= 0) {
305 break;
306 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700307 callback->OnStdout(buffer_ptr, length);
Felipe Leme44a42672016-04-01 17:43:27 -0700308 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800309 }
David Pursell606835a2015-09-08 17:17:02 -0700310
Felipe Leme0d4f0502016-07-26 12:14:39 -0700311 return callback->Done(exit_code);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800312}
313
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700314static void stdinout_raw_prologue(int inFd, int outFd, int& old_stdin_mode, int& old_stdout_mode) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700315 if (inFd == STDIN_FILENO) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800316 stdin_raw_init();
Spencer Lowb7dfb792015-05-22 16:48:31 -0700317#ifdef _WIN32
318 old_stdin_mode = _setmode(STDIN_FILENO, _O_BINARY);
319 if (old_stdin_mode == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700320 PLOG(FATAL) << "could not set stdin to binary";
Spencer Lowb7dfb792015-05-22 16:48:31 -0700321 }
322#endif
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700323 }
Yabin Cuid325e862014-11-17 14:48:25 -0800324
Spencer Lowb7dfb792015-05-22 16:48:31 -0700325#ifdef _WIN32
326 if (outFd == STDOUT_FILENO) {
327 old_stdout_mode = _setmode(STDOUT_FILENO, _O_BINARY);
328 if (old_stdout_mode == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700329 PLOG(FATAL) << "could not set stdout to binary";
Spencer Lowb7dfb792015-05-22 16:48:31 -0700330 }
331 }
332#endif
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700333}
334
335static void stdinout_raw_epilogue(int inFd, int outFd, int old_stdin_mode, int old_stdout_mode) {
336 if (inFd == STDIN_FILENO) {
337 stdin_raw_restore();
338#ifdef _WIN32
339 if (_setmode(STDIN_FILENO, old_stdin_mode) == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700340 PLOG(FATAL) << "could not restore stdin mode";
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700341 }
342#endif
343 }
344
345#ifdef _WIN32
346 if (outFd == STDOUT_FILENO) {
347 if (_setmode(STDOUT_FILENO, old_stdout_mode) == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700348 PLOG(FATAL) << "could not restore stdout mode";
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700349 }
350 }
351#endif
352}
353
Idries Hamadied409ea2018-01-29 16:30:36 +0000354void copy_to_file(int inFd, int outFd) {
Yurii Zubrytskyi6eca0e62019-06-27 13:47:34 -0700355 std::vector<char> buf(64 * 1024);
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700356 int len;
357 long total = 0;
358 int old_stdin_mode = -1;
359 int old_stdout_mode = -1;
360
361 D("copy_to_file(%d -> %d)", inFd, outFd);
362
363 stdinout_raw_prologue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700364
Elliott Hughesa7090b92015-04-17 17:03:59 -0700365 while (true) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700366 if (inFd == STDIN_FILENO) {
Elliott Hughes86ab9ff2018-09-05 12:13:11 -0700367 len = unix_read(inFd, buf.data(), buf.size());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700368 } else {
Elliott Hughes86ab9ff2018-09-05 12:13:11 -0700369 len = adb_read(inFd, buf.data(), buf.size());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700370 }
Christopher Tated2f54152011-04-21 12:53:28 -0700371 if (len == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700372 D("copy_to_file() : read 0 bytes; exiting");
Christopher Tated2f54152011-04-21 12:53:28 -0700373 break;
374 }
375 if (len < 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700376 D("copy_to_file(): read failed: %s", strerror(errno));
Christopher Tated2f54152011-04-21 12:53:28 -0700377 break;
378 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700379 if (outFd == STDOUT_FILENO) {
Luis Hector Chavez6150a372018-07-18 21:18:27 -0700380 fwrite(buf.data(), 1, len, stdout);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700381 fflush(stdout);
382 } else {
Luis Hector Chavez6150a372018-07-18 21:18:27 -0700383 adb_write(outFd, buf.data(), len);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700384 }
Christopher Tatec9cd3b92011-06-01 17:56:23 -0700385 total += len;
Christopher Tated2f54152011-04-21 12:53:28 -0700386 }
Yabin Cuid325e862014-11-17 14:48:25 -0800387
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700388 stdinout_raw_epilogue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700389
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700390 D("copy_to_file() finished after %lu bytes", total);
Christopher Tated2f54152011-04-21 12:53:28 -0700391}
392
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800393static void send_window_size_change(int fd, std::unique_ptr<ShellProtocol>& shell) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800394 // Old devices can't handle window size changes.
395 if (shell == nullptr) return;
396
Spencer Low2e02dc62015-11-07 17:34:39 -0800397#if defined(_WIN32)
398 struct winsize {
399 unsigned short ws_row;
400 unsigned short ws_col;
401 unsigned short ws_xpixel;
402 unsigned short ws_ypixel;
403 };
404#endif
405
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800406 winsize ws;
Spencer Low2e02dc62015-11-07 17:34:39 -0800407
408#if defined(_WIN32)
409 // If stdout is redirected to a non-console, we won't be able to get the
410 // console size, but that makes sense.
411 const intptr_t intptr_handle = _get_osfhandle(STDOUT_FILENO);
412 if (intptr_handle == -1) return;
413
414 const HANDLE handle = reinterpret_cast<const HANDLE>(intptr_handle);
415
416 CONSOLE_SCREEN_BUFFER_INFO info;
417 memset(&info, 0, sizeof(info));
418 if (!GetConsoleScreenBufferInfo(handle, &info)) return;
419
420 memset(&ws, 0, sizeof(ws));
421 // The number of visible rows, excluding offscreen scroll-back rows which are in info.dwSize.Y.
422 ws.ws_row = info.srWindow.Bottom - info.srWindow.Top + 1;
423 // If the user has disabled "Wrap text output on resize", they can make the screen buffer wider
424 // than the window, in which case we should use the width of the buffer.
425 ws.ws_col = info.dwSize.X;
426#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800427 if (ioctl(fd, TIOCGWINSZ, &ws) == -1) return;
Spencer Low2e02dc62015-11-07 17:34:39 -0800428#endif
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800429
430 // Send the new window size as human-readable ASCII for debugging convenience.
431 size_t l = snprintf(shell->data(), shell->data_capacity(), "%dx%d,%dx%d",
432 ws.ws_row, ws.ws_col, ws.ws_xpixel, ws.ws_ypixel);
433 shell->Write(ShellProtocol::kIdWindowSizeChange, l + 1);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800434}
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800435
David Pursell606835a2015-09-08 17:17:02 -0700436// Used to pass multiple values to the stdin read thread.
437struct StdinReadArgs {
438 int stdin_fd, write_fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700439 bool raw_stdin;
David Pursell606835a2015-09-08 17:17:02 -0700440 std::unique_ptr<ShellProtocol> protocol;
Elliott Hughes28416d62015-11-04 13:36:32 -0800441 char escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700442};
443
David Pursell606835a2015-09-08 17:17:02 -0700444// Loops to read from stdin and push the data to the given FD.
445// The argument should be a pointer to a StdinReadArgs object. This function
446// will take ownership of the object and delete it when finished.
Josh Gaod9db09c2016-02-12 14:31:15 -0800447static void stdin_read_thread_loop(void* x) {
David Pursell606835a2015-09-08 17:17:02 -0700448 std::unique_ptr<StdinReadArgs> args(reinterpret_cast<StdinReadArgs*>(x));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800449
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800450#if !defined(_WIN32)
451 // Mask SIGTTIN in case we're in a backgrounded process.
Josh Gao8dcdb572015-10-23 15:03:31 -0700452 sigset_t sigset;
453 sigemptyset(&sigset);
454 sigaddset(&sigset, SIGTTIN);
455 pthread_sigmask(SIG_BLOCK, &sigset, nullptr);
456#endif
457
Spencer Low2e02dc62015-11-07 17:34:39 -0800458#if defined(_WIN32)
459 // _get_interesting_input_record_uncached() causes unix_read_interruptible()
460 // to return -1 with errno == EINTR if the window size changes.
461#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800462 // Unblock SIGWINCH for this thread, so our read(2) below will be
463 // interrupted if the window size changes.
464 sigset_t mask;
465 sigemptyset(&mask);
466 sigaddset(&mask, SIGWINCH);
467 pthread_sigmask(SIG_UNBLOCK, &mask, nullptr);
468#endif
469
470 // Set up the initial window size.
471 send_window_size_change(args->stdin_fd, args->protocol);
472
Elliott Hughes28416d62015-11-04 13:36:32 -0800473 char raw_buffer[BUFSIZ];
David Pursell606835a2015-09-08 17:17:02 -0700474 char* buffer_ptr = raw_buffer;
475 size_t buffer_size = sizeof(raw_buffer);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800476 if (args->protocol != nullptr) {
David Pursell606835a2015-09-08 17:17:02 -0700477 buffer_ptr = args->protocol->data();
478 buffer_size = args->protocol->data_capacity();
479 }
480
Elliott Hughes28416d62015-11-04 13:36:32 -0800481 // If we need to parse escape sequences, make life easy.
482 if (args->raw_stdin && args->escape_char != '\0') {
483 buffer_size = 1;
484 }
485
486 enum EscapeState { kMidFlow, kStartOfLine, kInEscape };
487 EscapeState state = kStartOfLine;
488
Elliott Hughesaa245492015-08-03 10:38:08 -0700489 while (true) {
Spencer Low2e02dc62015-11-07 17:34:39 -0800490 // Use unix_read_interruptible() rather than adb_read() for stdin.
491 D("stdin_read_thread_loop(): pre unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
492 int r = unix_read_interruptible(args->stdin_fd, buffer_ptr,
493 buffer_size);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800494 if (r == -1 && errno == EINTR) {
495 send_window_size_change(args->stdin_fd, args->protocol);
496 continue;
497 }
Spencer Low2e02dc62015-11-07 17:34:39 -0800498 D("stdin_read_thread_loop(): post unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
David Pursell1ed57f02015-10-06 15:30:03 -0700499 if (r <= 0) {
500 // Only devices using the shell protocol know to close subprocess
501 // stdin. For older devices we want to just leave the connection
502 // open, otherwise an unpredictable amount of return data could
503 // be lost due to the FD closing before all data has been received.
504 if (args->protocol) {
505 args->protocol->Write(ShellProtocol::kIdCloseStdin, 0);
506 }
507 break;
508 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800509 // If we made stdin raw, check input for escape sequences. In
David Pursell1ed57f02015-10-06 15:30:03 -0700510 // this situation signals like Ctrl+C are sent remotely rather than
511 // interpreted locally so this provides an emergency out if the remote
512 // process starts ignoring the signal. SSH also does this, see the
513 // "escape characters" section on the ssh man page for more info.
Elliott Hughes28416d62015-11-04 13:36:32 -0800514 if (args->raw_stdin && args->escape_char != '\0') {
515 char ch = buffer_ptr[0];
516 if (ch == args->escape_char) {
517 if (state == kStartOfLine) {
518 state = kInEscape;
519 // Swallow the escape character.
520 continue;
521 } else {
522 state = kMidFlow;
523 }
524 } else {
525 if (state == kInEscape) {
526 if (ch == '.') {
527 fprintf(stderr,"\r\n[ disconnected ]\r\n");
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800528 stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700529 exit(0);
Elliott Hughes28416d62015-11-04 13:36:32 -0800530 } else {
531 // We swallowed an escape character that wasn't part of
532 // a valid escape sequence; time to cough it up.
533 buffer_ptr[0] = args->escape_char;
534 buffer_ptr[1] = ch;
535 ++r;
David Pursell1ed57f02015-10-06 15:30:03 -0700536 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800537 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800538 state = (ch == '\n' || ch == '\r') ? kStartOfLine : kMidFlow;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800539 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800540 }
David Pursell606835a2015-09-08 17:17:02 -0700541 if (args->protocol) {
542 if (!args->protocol->Write(ShellProtocol::kIdStdin, r)) {
543 break;
544 }
545 } else {
546 if (!WriteFdExactly(args->write_fd, buffer_ptr, r)) {
547 break;
548 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800549 }
550 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800551}
552
David Pursell1ed57f02015-10-06 15:30:03 -0700553// Returns a shell service string with the indicated arguments and command.
554static std::string ShellServiceString(bool use_shell_protocol,
555 const std::string& type_arg,
556 const std::string& command) {
557 std::vector<std::string> args;
558 if (use_shell_protocol) {
559 args.push_back(kShellServiceArgShellProtocol);
Elliott Hughesc2252df2015-11-18 12:45:48 -0800560
561 const char* terminal_type = getenv("TERM");
562 if (terminal_type != nullptr) {
563 args.push_back(std::string("TERM=") + terminal_type);
564 }
David Pursell1ed57f02015-10-06 15:30:03 -0700565 }
566 if (!type_arg.empty()) {
567 args.push_back(type_arg);
568 }
569
570 // Shell service string can look like: shell[,arg1,arg2,...]:[command].
571 return android::base::StringPrintf("shell%s%s:%s",
572 args.empty() ? "" : ",",
573 android::base::Join(args, ',').c_str(),
574 command.c_str());
575}
576
577// Connects to a shell on the device and read/writes data.
578//
579// Note: currently this function doesn't properly clean up resources; the
580// FD connected to the adb server is never closed and the stdin read thread
581// may never exit.
582//
583// On success returns the remote exit code if |use_shell_protocol| is true,
584// 0 otherwise. On failure returns 1.
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800585static int RemoteShell(bool use_shell_protocol, const std::string& type_arg, char escape_char,
586 bool empty_command, const std::string& service_string) {
Josh Gaoa019f782017-06-16 15:34:34 -0700587 // Old devices can't handle a service string that's longer than MAX_PAYLOAD_V1.
588 // Use |use_shell_protocol| to determine whether to allow a command longer than that.
589 if (service_string.size() > MAX_PAYLOAD_V1 && !use_shell_protocol) {
590 fprintf(stderr, "error: shell command too long\n");
591 return 1;
592 }
593
David Pursell1ed57f02015-10-06 15:30:03 -0700594 // Make local stdin raw if the device allocates a PTY, which happens if:
595 // 1. We are explicitly asking for a PTY shell, or
596 // 2. We don't specify shell type and are starting an interactive session.
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800597 bool raw_stdin = (type_arg == kShellServiceArgPty || (type_arg.empty() && empty_command));
David Pursell1ed57f02015-10-06 15:30:03 -0700598
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700599 std::string error;
David Pursell4e2fd362015-09-22 10:43:08 -0700600 int fd = adb_connect(service_string, &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700601 if (fd < 0) {
602 fprintf(stderr,"error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800603 return 1;
604 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800605
David Pursell606835a2015-09-08 17:17:02 -0700606 StdinReadArgs* args = new StdinReadArgs;
607 if (!args) {
608 LOG(ERROR) << "couldn't allocate StdinReadArgs object";
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700609 return 1;
610 }
David Pursell1ed57f02015-10-06 15:30:03 -0700611 args->stdin_fd = STDIN_FILENO;
David Pursell606835a2015-09-08 17:17:02 -0700612 args->write_fd = fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700613 args->raw_stdin = raw_stdin;
Elliott Hughes28416d62015-11-04 13:36:32 -0800614 args->escape_char = escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700615 if (use_shell_protocol) {
Josh Gao31b5be62018-03-07 16:51:08 -0800616 args->protocol = std::make_unique<ShellProtocol>(args->write_fd);
David Pursell606835a2015-09-08 17:17:02 -0700617 }
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700618
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800619 if (raw_stdin) stdin_raw_init();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800620
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800621#if !defined(_WIN32)
622 // Ensure our process is notified if the local window size changes.
623 // We use sigaction(2) to ensure that the SA_RESTART flag is not set,
624 // because the whole reason we're sending signals is to unblock the read(2)!
625 // That also means we don't need to do anything in the signal handler:
626 // the side effect of delivering the signal is all we need.
627 struct sigaction sa;
628 memset(&sa, 0, sizeof(sa));
629 sa.sa_handler = [](int) {};
630 sa.sa_flags = 0;
631 sigaction(SIGWINCH, &sa, nullptr);
632
633 // Now block SIGWINCH in this thread (the main thread) and all threads spawned
634 // from it. The stdin read thread will unblock this signal to ensure that it's
635 // the thread that receives the signal.
636 sigset_t mask;
637 sigemptyset(&mask);
638 sigaddset(&mask, SIGWINCH);
639 pthread_sigmask(SIG_BLOCK, &mask, nullptr);
640#endif
641
642 // TODO: combine read_and_dump with stdin_read_thread to make life simpler?
Josh Gaoe1dacfc2017-04-12 17:00:49 -0700643 std::thread(stdin_read_thread_loop, args).detach();
644 int exit_code = read_and_dump(fd, use_shell_protocol);
Elliott Hughes9b0f3542015-05-05 13:41:21 -0700645
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800646 // TODO: properly exit stdin_read_thread_loop and close |fd|.
David Pursell1ed57f02015-10-06 15:30:03 -0700647
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800648 // TODO: we should probably install signal handlers for this.
649 // TODO: can we use atexit? even on Windows?
650 if (raw_stdin) stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700651
David Pursell606835a2015-09-08 17:17:02 -0700652 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800653}
654
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800655static int adb_shell(int argc, const char** argv) {
David Pursell0aacbbe2016-01-21 19:56:50 -0800656 FeatureSet features;
Elliott Hughes4679a392018-10-19 13:59:44 -0700657 std::string error_message;
658 if (!adb_get_feature_set(&features, &error_message)) {
659 fprintf(stderr, "error: %s\n", error_message.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -0800660 return 1;
661 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800662
Elliott Hughescabfa112016-10-19 14:47:11 -0700663 enum PtyAllocationMode { kPtyAuto, kPtyNo, kPtyYes, kPtyDefinitely };
664
665 // Defaults.
666 char escape_char = '~'; // -e
667 bool use_shell_protocol = CanUseFeature(features, kFeatureShell2); // -x
668 PtyAllocationMode tty = use_shell_protocol ? kPtyAuto : kPtyDefinitely; // -t/-T
Elliott Hughes6452a892015-04-29 12:28:13 -0700669
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800670 // Parse shell-specific command-line options.
Elliott Hughescabfa112016-10-19 14:47:11 -0700671 argv[0] = "adb shell"; // So getopt(3) error messages start "adb shell".
Huihong Luoaf398852017-08-17 15:23:08 -0700672#ifdef _WIN32
673 // fixes "adb shell -l" crash on Windows, b/37284906
674 __argv = const_cast<char**>(argv);
675#endif
Elliott Hughescabfa112016-10-19 14:47:11 -0700676 optind = 1; // argv[0] is always "shell", so set `optind` appropriately.
677 int opt;
678 while ((opt = getopt(argc, const_cast<char**>(argv), "+e:ntTx")) != -1) {
679 switch (opt) {
680 case 'e':
681 if (!(strlen(optarg) == 1 || strcmp(optarg, "none") == 0)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -0700682 error_exit("-e requires a single-character argument or 'none'");
Elliott Hughescabfa112016-10-19 14:47:11 -0700683 }
684 escape_char = (strcmp(optarg, "none") == 0) ? 0 : optarg[0];
685 break;
686 case 'n':
687 close_stdin();
688 break;
689 case 'x':
690 // This option basically asks for historical behavior, so set options that
691 // correspond to the historical defaults. This is slightly weird in that -Tx
692 // is fine (because we'll undo the -T) but -xT isn't, but that does seem to
693 // be our least worst choice...
694 use_shell_protocol = false;
695 tty = kPtyDefinitely;
696 escape_char = '~';
697 break;
698 case 't':
699 // Like ssh, -t arguments are cumulative so that multiple -t's
700 // are needed to force a PTY.
701 tty = (tty >= kPtyYes) ? kPtyDefinitely : kPtyYes;
702 break;
703 case 'T':
704 tty = kPtyNo;
705 break;
706 default:
707 // getopt(3) already printed an error message for us.
Elliott Hughes28416d62015-11-04 13:36:32 -0800708 return 1;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800709 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700710 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800711
Elliott Hughescabfa112016-10-19 14:47:11 -0700712 bool is_interactive = (optind == argc);
David Pursell57dd5ae2016-01-27 16:07:52 -0800713
Elliott Hughescabfa112016-10-19 14:47:11 -0700714 std::string shell_type_arg = kShellServiceArgPty;
715 if (tty == kPtyNo) {
716 shell_type_arg = kShellServiceArgRaw;
717 } else if (tty == kPtyAuto) {
718 // If stdin isn't a TTY, default to a raw shell; this lets
719 // things like `adb shell < my_script.sh` work as expected.
720 // Non-interactive shells should also not have a pty.
721 if (!unix_isatty(STDIN_FILENO) || !is_interactive) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800722 shell_type_arg = kShellServiceArgRaw;
Elliott Hughescabfa112016-10-19 14:47:11 -0700723 }
724 } else if (tty == kPtyYes) {
725 // A single -t arg isn't enough to override implicit -T.
726 if (!unix_isatty(STDIN_FILENO)) {
727 fprintf(stderr,
728 "Remote PTY will not be allocated because stdin is not a terminal.\n"
729 "Use multiple -t options to force remote PTY allocation.\n");
730 shell_type_arg = kShellServiceArgRaw;
731 }
732 }
733
734 D("shell -e 0x%x t=%d use_shell_protocol=%s shell_type_arg=%s\n",
735 escape_char, tty,
736 use_shell_protocol ? "true" : "false",
737 (shell_type_arg == kShellServiceArgPty) ? "pty" : "raw");
738
739 // Raw mode is only supported when talking to a new device *and* using the shell protocol.
740 if (!use_shell_protocol) {
741 if (shell_type_arg != kShellServiceArgPty) {
742 fprintf(stderr, "error: %s only supports allocating a pty\n",
743 !CanUseFeature(features, kFeatureShell2) ? "device" : "-x");
744 return 1;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800745 } else {
Elliott Hughescabfa112016-10-19 14:47:11 -0700746 // If we're not using the shell protocol, the type argument must be empty.
747 shell_type_arg = "";
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800748 }
David Pursell71c83122015-09-14 15:33:50 -0700749 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800750
751 std::string command;
Elliott Hughescabfa112016-10-19 14:47:11 -0700752 if (optind < argc) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800753 // We don't escape here, just like ssh(1). http://b/20564385.
Elliott Hughescabfa112016-10-19 14:47:11 -0700754 command = android::base::Join(std::vector<const char*>(argv + optind, argv + argc), ' ');
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800755 }
756
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800757 std::string service_string = ShellServiceString(use_shell_protocol, shell_type_arg, command);
758 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, command.empty(),
759 service_string);
760}
761
762static int adb_abb(int argc, const char** argv) {
Alex Buynytskyy01a65ee2019-01-17 13:13:56 -0800763 FeatureSet features;
764 std::string error_message;
765 if (!adb_get_feature_set(&features, &error_message)) {
766 fprintf(stderr, "error: %s\n", error_message.c_str());
767 return 1;
768 }
769
770 if (!CanUseFeature(features, kFeatureAbb)) {
771 error_exit("abb is not supported by the device");
772 }
773
Alex Buynytskyycc2f1242019-06-27 10:33:54 -0700774 optind = 1; // argv[0] is always "abb", so set `optind` appropriately.
775
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800776 // Defaults.
777 constexpr char escape_char = '~'; // -e
778 constexpr bool use_shell_protocol = true;
779 constexpr auto shell_type_arg = kShellServiceArgRaw;
780 constexpr bool empty_command = false;
781
Alex Buynytskyycc2f1242019-06-27 10:33:54 -0700782 std::vector<const char*> args(argv + optind, argv + argc);
783 std::string service_string = "abb:" + android::base::Join(args, ABB_ARG_DELIMETER);
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800784
785 D("abb -e 0x%x [%*.s]\n", escape_char, static_cast<int>(service_string.size()),
786 service_string.data());
787
788 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, empty_command,
789 service_string);
David Pursell71c83122015-09-14 15:33:50 -0700790}
791
Elliott Hughes09e794c2017-05-02 14:31:59 -0700792static int adb_sideload_legacy(const char* filename, int in_fd, int size) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700793 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -0800794 unique_fd out_fd(adb_connect(android::base::StringPrintf("sideload:%d", size), &error));
Elliott Hughes09e794c2017-05-02 14:31:59 -0700795 if (out_fd < 0) {
796 fprintf(stderr, "adb: pre-KitKat sideload connection failed: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800797 return -1;
798 }
799
800 int opt = CHUNK_SIZE;
Elliott Hughes09e794c2017-05-02 14:31:59 -0700801 opt = adb_setsockopt(out_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
Doug Zongker447f0612012-01-09 14:54:53 -0800802
Elliott Hughes09e794c2017-05-02 14:31:59 -0700803 char buf[CHUNK_SIZE];
804 int total = size;
805 while (size > 0) {
806 unsigned xfer = (size > CHUNK_SIZE) ? CHUNK_SIZE : size;
807 if (!ReadFdExactly(in_fd, buf, xfer)) {
808 fprintf(stderr, "adb: failed to read data from %s: %s\n", filename, strerror(errno));
Doug Zongker447f0612012-01-09 14:54:53 -0800809 return -1;
810 }
Elliott Hughes09e794c2017-05-02 14:31:59 -0700811 if (!WriteFdExactly(out_fd, buf, xfer)) {
812 std::string error;
813 adb_status(out_fd, &error);
814 fprintf(stderr, "adb: failed to write data: %s\n", error.c_str());
Elliott Hughes09e794c2017-05-02 14:31:59 -0700815 return -1;
816 }
817 size -= xfer;
818 printf("sending: '%s' %4d%% \r", filename, (int)(100LL - ((100LL * size) / (total))));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700819 fflush(stdout);
Doug Zongker447f0612012-01-09 14:54:53 -0800820 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700821 printf("\n");
Doug Zongker447f0612012-01-09 14:54:53 -0800822
Elliott Hughes09e794c2017-05-02 14:31:59 -0700823 if (!adb_status(out_fd, &error)) {
824 fprintf(stderr, "adb: error response: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800825 return -1;
826 }
827
Doug Zongker447f0612012-01-09 14:54:53 -0800828 return 0;
829}
830
Doug Zongker71fe5842014-06-26 15:35:36 -0700831#define SIDELOAD_HOST_BLOCK_SIZE (CHUNK_SIZE)
832
Tao Bao11cf30b2019-04-07 23:24:03 -0700833// Connects to the sideload / rescue service on the device (served by minadbd) and sends over the
834// data in an OTA package.
835//
836// It uses a simple protocol as follows.
837//
838// - The connect message includes the total number of bytes in the file and a block size chosen by
839// us.
840//
841// - The other side sends the desired block number as eight decimal digits (e.g. "00000023" for
842// block 23). Blocks are numbered from zero.
843//
844// - We send back the data of the requested block. The last block is likely to be partial; when the
845// last block is requested we only send the part of the block that exists, it's not padded up to
846// the block size.
847//
848// - When the other side sends "DONEDONE" or "FAILFAIL" instead of a block number, we have done all
849// the data transfer.
850//
851static int adb_sideload_install(const char* filename, bool rescue_mode) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -0700852 // TODO: use a LinePrinter instead...
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700853 struct stat sb;
854 if (stat(filename, &sb) == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700855 fprintf(stderr, "adb: failed to stat file %s: %s\n", filename, strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700856 return -1;
857 }
858 unique_fd package_fd(adb_open(filename, O_RDONLY));
859 if (package_fd == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700860 fprintf(stderr, "adb: failed to open file %s: %s\n", filename, strerror(errno));
Doug Zongker71fe5842014-06-26 15:35:36 -0700861 return -1;
862 }
863
Tao Bao11cf30b2019-04-07 23:24:03 -0700864 std::string service = android::base::StringPrintf(
865 "%s:%" PRId64 ":%d", rescue_mode ? "rescue-install" : "sideload-host",
866 static_cast<int64_t>(sb.st_size), SIDELOAD_HOST_BLOCK_SIZE);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700867 std::string error;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700868 unique_fd device_fd(adb_connect(service, &error));
869 if (device_fd < 0) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700870 fprintf(stderr, "adb: sideload connection failed: %s\n", error.c_str());
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700871
Tao Bao11cf30b2019-04-07 23:24:03 -0700872 if (rescue_mode) {
873 return -1;
874 }
875
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700876 // If this is a small enough package, maybe this is an older device that doesn't
877 // support sideload-host. Try falling back to the older (<= K) sideload method.
878 if (sb.st_size > INT_MAX) {
879 return -1;
880 }
Elliott Hughes09e794c2017-05-02 14:31:59 -0700881 fprintf(stderr, "adb: trying pre-KitKat sideload method...\n");
Josh Gao27241a72019-04-25 14:04:57 -0700882 return adb_sideload_legacy(filename, package_fd.get(), static_cast<int>(sb.st_size));
Doug Zongker71fe5842014-06-26 15:35:36 -0700883 }
884
Elliott Hughese8b663f2016-05-26 22:43:19 -0700885 int opt = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700886 adb_setsockopt(device_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
887
888 char buf[SIDELOAD_HOST_BLOCK_SIZE];
Doug Zongker71fe5842014-06-26 15:35:36 -0700889
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700890 int64_t xfer = 0;
Elliott Hughese8b663f2016-05-26 22:43:19 -0700891 int last_percent = -1;
Elliott Hughesa7090b92015-04-17 17:03:59 -0700892 while (true) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700893 if (!ReadFdExactly(device_fd, buf, 8)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700894 fprintf(stderr, "adb: failed to read command: %s\n", strerror(errno));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700895 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700896 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700897 buf[8] = '\0';
Doug Zongker71fe5842014-06-26 15:35:36 -0700898
xunchangc2265c52019-04-22 23:14:28 -0700899 if (strcmp(kMinadbdServicesExitSuccess, buf) == 0 ||
900 strcmp(kMinadbdServicesExitFailure, buf) == 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -0700901 printf("\rTotal xfer: %.2fx%*s\n",
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700902 static_cast<double>(xfer) / (sb.st_size ? sb.st_size : 1),
903 static_cast<int>(strlen(filename) + 10), "");
xunchangc2265c52019-04-22 23:14:28 -0700904 if (strcmp(kMinadbdServicesExitFailure, buf) == 0) {
Tao Bao11cf30b2019-04-07 23:24:03 -0700905 return 1;
906 }
Elliott Hughese8b663f2016-05-26 22:43:19 -0700907 return 0;
Doug Zongker71fe5842014-06-26 15:35:36 -0700908 }
909
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700910 int64_t block = strtoll(buf, nullptr, 10);
911 int64_t offset = block * SIDELOAD_HOST_BLOCK_SIZE;
912 if (offset >= static_cast<int64_t>(sb.st_size)) {
913 fprintf(stderr,
914 "adb: failed to read block %" PRId64 " at offset %" PRId64 ", past end %" PRId64
915 "\n",
916 block, offset, static_cast<int64_t>(sb.st_size));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700917 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700918 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700919
Doug Zongker71fe5842014-06-26 15:35:36 -0700920 size_t to_write = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700921 if ((offset + SIDELOAD_HOST_BLOCK_SIZE) > static_cast<int64_t>(sb.st_size)) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700922 to_write = sb.st_size - offset;
Doug Zongker71fe5842014-06-26 15:35:36 -0700923 }
924
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700925 if (adb_lseek(package_fd, offset, SEEK_SET) != offset) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700926 fprintf(stderr, "adb: failed to seek to package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700927 return -1;
928 }
929 if (!ReadFdExactly(package_fd, buf, to_write)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700930 fprintf(stderr, "adb: failed to read package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700931 return -1;
932 }
933
934 if (!WriteFdExactly(device_fd, buf, to_write)) {
935 adb_status(device_fd, &error);
Elliott Hughes09e794c2017-05-02 14:31:59 -0700936 fprintf(stderr, "adb: failed to write data '%s' *\n", error.c_str());
Elliott Hughese8b663f2016-05-26 22:43:19 -0700937 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700938 }
939 xfer += to_write;
940
941 // For normal OTA packages, we expect to transfer every byte
942 // twice, plus a bit of overhead (one read during
943 // verification, one read of each byte for installation, plus
944 // extra access to things like the zip central directory).
945 // This estimate of the completion becomes 100% when we've
946 // transferred ~2.13 (=100/47) times the package size.
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700947 int percent = static_cast<int>(xfer * 47LL / (sb.st_size ? sb.st_size : 1));
Doug Zongker71fe5842014-06-26 15:35:36 -0700948 if (percent != last_percent) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700949 printf("\rserving: '%s' (~%d%%) ", filename, percent);
Doug Zongker71fe5842014-06-26 15:35:36 -0700950 fflush(stdout);
951 last_percent = percent;
952 }
953 }
Doug Zongker71fe5842014-06-26 15:35:36 -0700954}
955
xunchangc2265c52019-04-22 23:14:28 -0700956static int adb_wipe_devices() {
957 auto wipe_devices_message_size = strlen(kMinadbdServicesExitSuccess);
958 std::string error;
959 unique_fd fd(adb_connect(
960 android::base::StringPrintf("rescue-wipe:userdata:%zu", wipe_devices_message_size),
961 &error));
962 if (fd < 0) {
963 fprintf(stderr, "adb: wipe device connection failed: %s\n", error.c_str());
964 return 1;
965 }
966
967 std::string message(wipe_devices_message_size, '\0');
968 if (!ReadFdExactly(fd, message.data(), wipe_devices_message_size)) {
969 fprintf(stderr, "adb: failed to read wipe result: %s\n", strerror(errno));
970 return 1;
971 }
972
973 if (message == kMinadbdServicesExitSuccess) {
974 return 0;
975 }
976
977 if (message != kMinadbdServicesExitFailure) {
978 fprintf(stderr, "adb: got unexpected message from rescue wipe %s\n", message.c_str());
979 }
980 return 1;
981}
982
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800983/**
984 * Run ppp in "notty" mode against a resource listed as the first parameter
985 * eg:
986 *
987 * ppp dev:/dev/omap_csmi_tty0 <ppp options>
988 *
989 */
Elliott Hughes58305772015-04-17 13:57:15 -0700990static int ppp(int argc, const char** argv) {
Yabin Cuie77b6a02014-11-11 09:24:11 -0800991#if defined(_WIN32)
Elliott Hughesd2aaede2018-10-22 17:02:51 -0700992 error_exit("adb %s not implemented on Win32", argv[0]);
Elliott Hughes4679a392018-10-19 13:59:44 -0700993 __builtin_unreachable();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800994#else
Elliott Hughesd2aaede2018-10-22 17:02:51 -0700995 if (argc < 2) error_exit("usage: adb %s <adb service name> [ppp opts]", argv[0]);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800996
Dan Albertbac34742015-02-25 17:51:28 -0800997 const char* adb_service_name = argv[1];
Elliott Hughes4679a392018-10-19 13:59:44 -0700998 std::string error_message;
999 int fd = adb_connect(adb_service_name, &error_message);
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001000 if (fd < 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001001 error_exit("could not open adb service %s: %s", adb_service_name, error_message.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001002 }
1003
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001004 pid_t pid = fork();
Elliott Hughes4679a392018-10-19 13:59:44 -07001005 if (pid == -1) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001006 perror_exit("fork failed");
Elliott Hughes4679a392018-10-19 13:59:44 -07001007 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001008
Elliott Hughes4679a392018-10-19 13:59:44 -07001009 if (pid == 0) {
1010 // child side
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001011 int i;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001012
1013 // copy args
Elliott Hughes4679a392018-10-19 13:59:44 -07001014 const char** ppp_args = (const char**)alloca(sizeof(char*) * argc + 1);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001015 ppp_args[0] = "pppd";
1016 for (i = 2 ; i < argc ; i++) {
1017 //argv[2] and beyond become ppp_args[1] and beyond
1018 ppp_args[i - 1] = argv[i];
1019 }
Yi Kongaed415c2018-07-13 18:15:16 -07001020 ppp_args[i-1] = nullptr;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001021
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001022 dup2(fd, STDIN_FILENO);
1023 dup2(fd, STDOUT_FILENO);
1024 adb_close(STDERR_FILENO);
1025 adb_close(fd);
1026
Elliott Hughes4679a392018-10-19 13:59:44 -07001027 execvp("pppd", (char* const*)ppp_args);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001028 perror_exit("exec pppd failed");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001029 }
Elliott Hughes4679a392018-10-19 13:59:44 -07001030
1031 // parent side
1032 adb_close(fd);
1033 return 0;
Yabin Cuie77b6a02014-11-11 09:24:11 -08001034#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001035}
1036
Josh Gao3c99b0f2019-03-11 15:39:44 -07001037static bool wait_for_device(const char* service,
1038 std::optional<std::chrono::milliseconds> timeout = std::nullopt) {
Josh Gao08881e72016-04-13 12:14:16 -07001039 std::vector<std::string> components = android::base::Split(service, "-");
1040 if (components.size() < 3 || components.size() > 4) {
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001041 fprintf(stderr, "adb: couldn't parse 'wait-for' command: %s\n", service);
1042 return false;
1043 }
1044
Josh Gaob122b172017-08-16 16:57:01 -07001045 TransportType t;
1046 adb_get_transport(&t, nullptr, nullptr);
1047
Elliott Hughes2b101112015-05-04 19:29:32 -07001048 // Was the caller vague about what they'd like us to wait for?
1049 // If so, check they weren't more specific in their choice of transport type.
Josh Gao08881e72016-04-13 12:14:16 -07001050 if (components.size() == 3) {
1051 auto it = components.begin() + 2;
Elliott Hughes2b101112015-05-04 19:29:32 -07001052 if (t == kTransportUsb) {
Josh Gao08881e72016-04-13 12:14:16 -07001053 components.insert(it, "usb");
Elliott Hughes2b101112015-05-04 19:29:32 -07001054 } else if (t == kTransportLocal) {
Josh Gao08881e72016-04-13 12:14:16 -07001055 components.insert(it, "local");
Elliott Hughes2b101112015-05-04 19:29:32 -07001056 } else {
Josh Gao08881e72016-04-13 12:14:16 -07001057 components.insert(it, "any");
Elliott Hughes2b101112015-05-04 19:29:32 -07001058 }
Josh Gao08881e72016-04-13 12:14:16 -07001059 } else if (components[2] != "any" && components[2] != "local" && components[2] != "usb") {
1060 fprintf(stderr, "adb: unknown type %s; expected 'any', 'local', or 'usb'\n",
1061 components[2].c_str());
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001062 return false;
1063 }
1064
Josh Gao86441c32016-04-13 12:18:58 -07001065 if (components[3] != "any" && components[3] != "bootloader" && components[3] != "device" &&
Tao Bao55d407e2019-04-07 23:24:03 -07001066 components[3] != "recovery" && components[3] != "rescue" && components[3] != "sideload" &&
Josh Gao1e9e4712019-02-22 14:01:36 -08001067 components[3] != "disconnect") {
Josh Gao08881e72016-04-13 12:14:16 -07001068 fprintf(stderr,
1069 "adb: unknown state %s; "
Tao Bao55d407e2019-04-07 23:24:03 -07001070 "expected 'any', 'bootloader', 'device', 'recovery', 'rescue', 'sideload', or "
1071 "'disconnect'\n",
Josh Gao08881e72016-04-13 12:14:16 -07001072 components[3].c_str());
1073 return false;
1074 }
1075
Josh Gaob122b172017-08-16 16:57:01 -07001076 std::string cmd = format_host_command(android::base::Join(components, "-").c_str());
Josh Gao3c99b0f2019-03-11 15:39:44 -07001077 if (timeout) {
1078 std::thread([timeout]() {
1079 std::this_thread::sleep_for(*timeout);
1080 fprintf(stderr, "timeout expired while waiting for device\n");
1081 _exit(1);
1082 }).detach();
1083 }
Elliott Hughes424af022015-05-29 17:55:19 -07001084 return adb_command(cmd);
Elliott Hughes2b101112015-05-04 19:29:32 -07001085}
1086
Josh Gao06c73ae2016-03-04 15:16:18 -08001087static bool adb_root(const char* command) {
1088 std::string error;
Josh Gao06c73ae2016-03-04 15:16:18 -08001089
Josh Gao2020dd62019-02-22 15:20:27 -08001090 TransportId transport_id;
1091 unique_fd fd(adb_connect(&transport_id, android::base::StringPrintf("%s:", command), &error));
Elliott Hughese8b663f2016-05-26 22:43:19 -07001092 if (fd < 0) {
Josh Gao06c73ae2016-03-04 15:16:18 -08001093 fprintf(stderr, "adb: unable to connect for %s: %s\n", command, error.c_str());
1094 return false;
1095 }
1096
1097 // Figure out whether we actually did anything.
1098 char buf[256];
1099 char* cur = buf;
1100 ssize_t bytes_left = sizeof(buf);
1101 while (bytes_left > 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -07001102 ssize_t bytes_read = adb_read(fd, cur, bytes_left);
Josh Gao06c73ae2016-03-04 15:16:18 -08001103 if (bytes_read == 0) {
1104 break;
1105 } else if (bytes_read < 0) {
1106 fprintf(stderr, "adb: error while reading for %s: %s\n", command, strerror(errno));
1107 return false;
1108 }
1109 cur += bytes_read;
1110 bytes_left -= bytes_read;
1111 }
1112
1113 if (bytes_left == 0) {
1114 fprintf(stderr, "adb: unexpected output length for %s\n", command);
1115 return false;
1116 }
1117
1118 fflush(stdout);
1119 WriteFdExactly(STDOUT_FILENO, buf, sizeof(buf) - bytes_left);
1120 if (cur != buf && strstr(buf, "restarting") == nullptr) {
1121 return true;
1122 }
1123
Josh Gao2020dd62019-02-22 15:20:27 -08001124 // Wait for the device to go away.
Josh Gao3c99b0f2019-03-11 15:39:44 -07001125 TransportType previous_type;
1126 const char* previous_serial;
1127 TransportId previous_id;
1128 adb_get_transport(&previous_type, &previous_serial, &previous_id);
1129
Josh Gao2020dd62019-02-22 15:20:27 -08001130 adb_set_transport(kTransportAny, nullptr, transport_id);
1131 wait_for_device("wait-for-disconnect");
Josh Gao3c99b0f2019-03-11 15:39:44 -07001132
1133 // Wait for the device to come back.
1134 // If we were using a specific transport ID, there's nothing we can wait for.
1135 if (previous_id == 0) {
1136 adb_set_transport(previous_type, previous_serial, 0);
Josh Gaoef4a9992019-05-01 18:51:46 -07001137 wait_for_device("wait-for-device", 6000ms);
Josh Gao3c99b0f2019-03-11 15:39:44 -07001138 }
1139
Josh Gao05824732016-06-16 14:00:09 -07001140 return true;
Josh Gao06c73ae2016-03-04 15:16:18 -08001141}
1142
Josh Gaob122b172017-08-16 16:57:01 -07001143int send_shell_command(const std::string& command, bool disable_shell_protocol,
1144 StandardStreamsCallbackInterface* callback) {
Josh Gao74ccdf92019-01-23 15:36:42 -08001145 unique_fd fd;
David Pursell0aacbbe2016-01-21 19:56:50 -08001146 bool use_shell_protocol = false;
1147
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001148 while (true) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001149 bool attempt_connection = true;
1150
Felipe Leme0d4f0502016-07-26 12:14:39 -07001151 // Use shell protocol if it's supported and the caller doesn't explicitly
1152 // disable it.
David Pursell0aacbbe2016-01-21 19:56:50 -08001153 if (!disable_shell_protocol) {
1154 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001155 std::string error;
1156 if (adb_get_feature_set(&features, &error)) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001157 use_shell_protocol = CanUseFeature(features, kFeatureShell2);
1158 } else {
1159 // Device was unreachable.
1160 attempt_connection = false;
1161 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001162 }
David Pursell0aacbbe2016-01-21 19:56:50 -08001163
1164 if (attempt_connection) {
1165 std::string error;
1166 std::string service_string = ShellServiceString(use_shell_protocol, "", command);
1167
Josh Gao74ccdf92019-01-23 15:36:42 -08001168 fd.reset(adb_connect(service_string, &error));
David Pursell0aacbbe2016-01-21 19:56:50 -08001169 if (fd >= 0) {
1170 break;
1171 }
1172 }
1173
Felipe Leme0d4f0502016-07-26 12:14:39 -07001174 fprintf(stderr, "- waiting for device -\n");
Josh Gaob122b172017-08-16 16:57:01 -07001175 if (!wait_for_device("wait-for-device")) {
Josh Gaob6117c42016-02-24 15:16:17 -08001176 return 1;
1177 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001178 }
1179
Josh Gao74ccdf92019-01-23 15:36:42 -08001180 return read_and_dump(fd.get(), use_shell_protocol, callback);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001181}
1182
Josh Gaob122b172017-08-16 16:57:01 -07001183static int logcat(int argc, const char** argv) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001184 char* log_tags = getenv("ANDROID_LOG_TAGS");
1185 std::string quoted = escape_arg(log_tags == nullptr ? "" : log_tags);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001186
David Pursell70ef7b42015-09-30 13:35:42 -07001187 std::string cmd = "export ANDROID_LOG_TAGS=\"" + quoted + "\"; exec logcat";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001188
Jeff Sharkeyfd546e82014-06-10 11:31:24 -07001189 if (!strcmp(argv[0], "longcat")) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001190 cmd += " -v long";
Christopher Tatedb0a8802011-11-30 13:00:33 -08001191 }
1192
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001193 --argc;
1194 ++argv;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001195 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001196 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001197 }
1198
Elliott Hughes31f0f1b2018-06-14 10:46:05 -07001199 return send_shell_command(cmd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001200}
1201
Josh Gao27241a72019-04-25 14:04:57 -07001202static void write_zeros(int bytes, borrowed_fd fd) {
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001203 int old_stdin_mode = -1;
1204 int old_stdout_mode = -1;
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001205 std::vector<char> buf(bytes);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001206
Josh Gao27241a72019-04-25 14:04:57 -07001207 D("write_zeros(%d) -> %d", bytes, fd.get());
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001208
Josh Gao27241a72019-04-25 14:04:57 -07001209 stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001210
1211 if (fd == STDOUT_FILENO) {
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001212 fwrite(buf.data(), 1, bytes, stdout);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001213 fflush(stdout);
1214 } else {
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001215 adb_write(fd, buf.data(), bytes);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001216 }
1217
Josh Gao27241a72019-04-25 14:04:57 -07001218 stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001219
1220 D("write_zeros() finished");
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001221}
1222
Dan Albertbac34742015-02-25 17:51:28 -08001223static int backup(int argc, const char** argv) {
Al Sutton31d335f2019-05-09 16:14:05 +00001224 fprintf(stdout, "WARNING: adb backup is deprecated and may be removed in a future release\n");
1225
Elliott Hughes24f165f2015-08-21 20:31:31 -07001226 const char* filename = "backup.ab";
Christopher Tated2f54152011-04-21 12:53:28 -07001227
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001228 /* find, extract, and use any -f argument */
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001229 for (int i = 1; i < argc; i++) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001230 if (!strcmp("-f", argv[i])) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001231 if (i == argc - 1) error_exit("backup -f passed with no filename");
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001232 filename = argv[i+1];
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001233 for (int j = i+2; j <= argc; ) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001234 argv[i++] = argv[j++];
1235 }
1236 argc -= 2;
Yi Kongaed415c2018-07-13 18:15:16 -07001237 argv[argc] = nullptr;
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001238 }
Christopher Tated2f54152011-04-21 12:53:28 -07001239 }
1240
Elliott Hughes56e68132015-11-13 11:04:10 -08001241 // Bare "adb backup" or "adb backup -f filename" are not valid invocations ---
1242 // a list of packages is required.
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001243 if (argc < 2) error_exit("backup either needs a list of packages or -all/-shared");
Christopher Tatebb86bc52011-08-22 17:12:08 -07001244
Christopher Tateb1dfffe2011-12-08 19:04:34 -08001245 adb_unlink(filename);
Josh Gao74ccdf92019-01-23 15:36:42 -08001246 unique_fd outFd(adb_creat(filename, 0640));
Christopher Tated2f54152011-04-21 12:53:28 -07001247 if (outFd < 0) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001248 fprintf(stderr, "adb: backup unable to create file '%s': %s\n", filename, strerror(errno));
1249 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001250 }
1251
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001252 std::string cmd = "backup:";
1253 --argc;
1254 ++argv;
1255 while (argc-- > 0) {
1256 cmd += " " + escape_arg(*argv++);
Christopher Tated2f54152011-04-21 12:53:28 -07001257 }
1258
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001259 D("backup. filename=%s cmd=%s", filename, cmd.c_str());
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001260 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001261 unique_fd fd(adb_connect(cmd, &error));
Christopher Tated2f54152011-04-21 12:53:28 -07001262 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001263 fprintf(stderr, "adb: unable to connect for backup: %s\n", error.c_str());
Elliott Hughes56e68132015-11-13 11:04:10 -08001264 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001265 }
1266
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001267 fprintf(stdout, "Now unlock your device and confirm the backup operation...\n");
Elliott Hughes56e68132015-11-13 11:04:10 -08001268 fflush(stdout);
1269
Josh Gao74ccdf92019-01-23 15:36:42 -08001270 copy_to_file(fd.get(), outFd.get());
Elliott Hughes56e68132015-11-13 11:04:10 -08001271 return EXIT_SUCCESS;
Christopher Tated2f54152011-04-21 12:53:28 -07001272}
1273
Dan Albertbac34742015-02-25 17:51:28 -08001274static int restore(int argc, const char** argv) {
Al Sutton31d335f2019-05-09 16:14:05 +00001275 fprintf(stdout, "WARNING: adb restore is deprecated and may be removed in a future release\n");
1276
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001277 if (argc != 2) error_exit("restore requires an argument");
Christopher Tate702967a2011-05-17 15:52:54 -07001278
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001279 const char* filename = argv[1];
Josh Gao74ccdf92019-01-23 15:36:42 -08001280 unique_fd tarFd(adb_open(filename, O_RDONLY));
Christopher Tate702967a2011-05-17 15:52:54 -07001281 if (tarFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001282 fprintf(stderr, "adb: unable to open file %s: %s\n", filename, strerror(errno));
Christopher Tate702967a2011-05-17 15:52:54 -07001283 return -1;
1284 }
1285
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001286 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001287 unique_fd fd(adb_connect("restore:", &error));
Christopher Tate702967a2011-05-17 15:52:54 -07001288 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001289 fprintf(stderr, "adb: unable to connect for restore: %s\n", error.c_str());
Christopher Tate702967a2011-05-17 15:52:54 -07001290 return -1;
1291 }
1292
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001293 fprintf(stdout, "Now unlock your device and confirm the restore operation.\n");
1294 fflush(stdout);
1295
Josh Gao74ccdf92019-01-23 15:36:42 -08001296 copy_to_file(tarFd.get(), fd.get());
Christopher Tate702967a2011-05-17 15:52:54 -07001297
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001298 // Provide an in-band EOD marker in case the archive file is malformed
Josh Gao74ccdf92019-01-23 15:36:42 -08001299 write_zeros(512 * 2, fd);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001300
Josh Gao5767d0d2016-03-04 15:51:03 -08001301 // Wait until the other side finishes, or it'll get sent SIGHUP.
Josh Gao74ccdf92019-01-23 15:36:42 -08001302 copy_to_file(fd.get(), STDOUT_FILENO);
Christopher Tate702967a2011-05-17 15:52:54 -07001303 return 0;
1304}
1305
Dan Albert06b0d6b2017-05-18 22:56:48 -07001306static void parse_push_pull_args(const char** arg, int narg, std::vector<const char*>* srcs,
1307 const char** dst, bool* copy_attrs, bool* sync) {
Josh Gao05786772015-10-30 16:57:19 -07001308 *copy_attrs = false;
Mark Lindner76f2a932014-03-11 17:55:59 -07001309
Josh Gao05786772015-10-30 16:57:19 -07001310 srcs->clear();
1311 bool ignore_flags = false;
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001312 while (narg > 0) {
Josh Gao05786772015-10-30 16:57:19 -07001313 if (ignore_flags || *arg[0] != '-') {
1314 srcs->push_back(*arg);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001315 } else {
Josh Gao05786772015-10-30 16:57:19 -07001316 if (!strcmp(*arg, "-p")) {
1317 // Silently ignore for backwards compatibility.
1318 } else if (!strcmp(*arg, "-a")) {
1319 *copy_attrs = true;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001320 } else if (!strcmp(*arg, "--sync")) {
1321 if (sync != nullptr) {
1322 *sync = true;
1323 }
Josh Gao05786772015-10-30 16:57:19 -07001324 } else if (!strcmp(*arg, "--")) {
1325 ignore_flags = true;
1326 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001327 error_exit("unrecognized option '%s'", *arg);
Josh Gao05786772015-10-30 16:57:19 -07001328 }
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001329 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001330 ++arg;
1331 --narg;
1332 }
1333
Josh Gao05786772015-10-30 16:57:19 -07001334 if (srcs->size() > 1) {
1335 *dst = srcs->back();
1336 srcs->pop_back();
Mark Lindner76f2a932014-03-11 17:55:59 -07001337 }
1338}
1339
Josh Gaod10b7c42019-03-14 15:38:02 -07001340static int adb_connect_command(const std::string& command, TransportId* transport = nullptr) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001341 std::string error;
Josh Gaod10b7c42019-03-14 15:38:02 -07001342 unique_fd fd(adb_connect(transport, command, &error));
Elliott Hughes5677c232015-05-07 23:37:40 -07001343 if (fd < 0) {
1344 fprintf(stderr, "error: %s\n", error.c_str());
1345 return 1;
Tao Bao175b7bb2015-03-29 11:22:34 -07001346 }
Elliott Hughes5677c232015-05-07 23:37:40 -07001347 read_and_dump(fd);
Elliott Hughes5677c232015-05-07 23:37:40 -07001348 return 0;
Tao Bao175b7bb2015-03-29 11:22:34 -07001349}
1350
Josh Gaoe89a55d2018-11-16 14:47:59 -08001351static int adb_connect_command_bidirectional(const std::string& command) {
1352 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001353 unique_fd fd(adb_connect(command, &error));
Josh Gaoe89a55d2018-11-16 14:47:59 -08001354 if (fd < 0) {
1355 fprintf(stderr, "error: %s\n", error.c_str());
1356 return 1;
1357 }
1358
1359 static constexpr auto forward = [](int src, int sink, bool exit_on_end) {
1360 char buf[4096];
1361 while (true) {
1362 int rc = adb_read(src, buf, sizeof(buf));
1363 if (rc == 0) {
1364 if (exit_on_end) {
1365 exit(0);
1366 } else {
1367 adb_shutdown(sink, SHUT_WR);
1368 }
1369 return;
1370 } else if (rc < 0) {
1371 perror_exit("read failed");
1372 }
1373 if (!WriteFdExactly(sink, buf, rc)) {
1374 perror_exit("write failed");
1375 }
1376 }
1377 };
1378
Josh Gao74ccdf92019-01-23 15:36:42 -08001379 std::thread read(forward, fd.get(), STDOUT_FILENO, true);
1380 std::thread write(forward, STDIN_FILENO, fd.get(), false);
Josh Gaoe89a55d2018-11-16 14:47:59 -08001381 read.join();
1382 write.join();
Josh Gaoe89a55d2018-11-16 14:47:59 -08001383 return 0;
1384}
1385
Elliott Hughes6452a892015-04-29 12:28:13 -07001386static int adb_query_command(const std::string& command) {
1387 std::string result;
1388 std::string error;
1389 if (!adb_query(command, &result, &error)) {
1390 fprintf(stderr, "error: %s\n", error.c_str());
1391 return 1;
1392 }
1393 printf("%s\n", result.c_str());
1394 return 0;
1395}
1396
Spencer Lowa13df302015-09-07 16:20:13 -07001397// Disallow stdin, stdout, and stderr.
1398static bool _is_valid_ack_reply_fd(const int ack_reply_fd) {
1399#ifdef _WIN32
1400 const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd);
1401 return (GetStdHandle(STD_INPUT_HANDLE) != ack_reply_handle) &&
1402 (GetStdHandle(STD_OUTPUT_HANDLE) != ack_reply_handle) &&
1403 (GetStdHandle(STD_ERROR_HANDLE) != ack_reply_handle);
1404#else
1405 return ack_reply_fd > 2;
1406#endif
1407}
1408
Elliott Hughescabfa112016-10-19 14:47:11 -07001409int adb_commandline(int argc, const char** argv) {
Josh Gaoe0323872018-08-10 14:23:43 -07001410 bool no_daemon = false;
1411 bool is_daemon = false;
1412 bool is_server = false;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001413 int r;
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001414 TransportType transport_type = kTransportAny;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001415 int ack_reply_fd = -1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001416
Elliott Hughes65433da2015-11-18 18:07:48 -08001417#if !defined(_WIN32)
1418 // We'd rather have EPIPE than SIGPIPE.
1419 signal(SIGPIPE, SIG_IGN);
1420#endif
1421
Josh Gao9c869b52016-08-25 16:00:22 -07001422 const char* server_host_str = nullptr;
1423 const char* server_port_str = nullptr;
1424 const char* server_socket_str = nullptr;
Stefan Hilzingera84a42e2010-04-19 12:21:12 +01001425
Elliott Hughes8d28e192015-10-07 14:55:10 -07001426 // We need to check for -d and -e before we look at $ANDROID_SERIAL.
1427 const char* serial = nullptr;
Josh Gaob122b172017-08-16 16:57:01 -07001428 TransportId transport_id = 0;
Elliott Hughes8d28e192015-10-07 14:55:10 -07001429
Riley Andrews98f58e82014-12-05 17:37:24 -08001430 while (argc > 0) {
Josh Gaod10b7c42019-03-14 15:38:02 -07001431 if (!strcmp(argv[0], "server")) {
Josh Gaoe0323872018-08-10 14:23:43 -07001432 is_server = true;
Josh Gaod10b7c42019-03-14 15:38:02 -07001433 } else if (!strcmp(argv[0], "nodaemon")) {
Josh Gaoe0323872018-08-10 14:23:43 -07001434 no_daemon = true;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001435 } else if (!strcmp(argv[0], "fork-server")) {
1436 /* this is a special flag used only when the ADB client launches the ADB Server */
Josh Gaoe0323872018-08-10 14:23:43 -07001437 is_daemon = true;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001438 } else if (!strcmp(argv[0], "--reply-fd")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001439 if (argc < 2) error_exit("--reply-fd requires an argument");
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001440 const char* reply_fd_str = argv[1];
1441 argc--;
1442 argv++;
1443 ack_reply_fd = strtol(reply_fd_str, nullptr, 10);
Spencer Lowa13df302015-09-07 16:20:13 -07001444 if (!_is_valid_ack_reply_fd(ack_reply_fd)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001445 fprintf(stderr, "adb: invalid reply fd \"%s\"\n", reply_fd_str);
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001446 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001447 }
Josh Gaob122b172017-08-16 16:57:01 -07001448 } else if (!strncmp(argv[0], "-s", 2)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001449 if (isdigit(argv[0][2])) {
1450 serial = argv[0] + 2;
1451 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001452 if (argc < 2 || argv[0][2] != '\0') error_exit("-s requires an argument");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001453 serial = argv[1];
1454 argc--;
1455 argv++;
1456 }
Josh Gaob122b172017-08-16 16:57:01 -07001457 } else if (!strncmp(argv[0], "-t", 2)) {
1458 const char* id;
1459 if (isdigit(argv[0][2])) {
1460 id = argv[0] + 2;
1461 } else {
1462 id = argv[1];
1463 argc--;
1464 argv++;
1465 }
1466 transport_id = strtoll(id, const_cast<char**>(&id), 10);
1467 if (*id != '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001468 error_exit("invalid transport id");
Josh Gaob122b172017-08-16 16:57:01 -07001469 }
Josh Gaod10b7c42019-03-14 15:38:02 -07001470 } else if (!strcmp(argv[0], "-d")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001471 transport_type = kTransportUsb;
Josh Gaod10b7c42019-03-14 15:38:02 -07001472 } else if (!strcmp(argv[0], "-e")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001473 transport_type = kTransportLocal;
Josh Gaod10b7c42019-03-14 15:38:02 -07001474 } else if (!strcmp(argv[0], "-a")) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001475 gListenAll = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001476 } else if (!strncmp(argv[0], "-H", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001477 if (argv[0][2] == '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001478 if (argc < 2) error_exit("-H requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001479 server_host_str = argv[1];
Matt Gumbeld7b33082012-11-14 10:16:17 -08001480 argc--;
1481 argv++;
1482 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001483 server_host_str = argv[0] + 2;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001484 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001485 } else if (!strncmp(argv[0], "-P", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001486 if (argv[0][2] == '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001487 if (argc < 2) error_exit("-P requires an argument");
Matt Gumbeld7b33082012-11-14 10:16:17 -08001488 server_port_str = argv[1];
1489 argc--;
1490 argv++;
1491 } else {
1492 server_port_str = argv[0] + 2;
1493 }
Josh Gao9c869b52016-08-25 16:00:22 -07001494 } else if (!strcmp(argv[0], "-L")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001495 if (argc < 2) error_exit("-L requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001496 server_socket_str = argv[1];
1497 argc--;
1498 argv++;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001499 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001500 /* out of recognized modifiers and flags */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001501 break;
1502 }
1503 argc--;
1504 argv++;
1505 }
1506
Josh Gao9c869b52016-08-25 16:00:22 -07001507 if ((server_host_str || server_port_str) && server_socket_str) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001508 error_exit("-L is incompatible with -H or -P");
Josh Gao9c869b52016-08-25 16:00:22 -07001509 }
1510
1511 // If -L, -H, or -P are specified, ignore environment variables.
1512 // Otherwise, prefer ADB_SERVER_SOCKET over ANDROID_ADB_SERVER_ADDRESS/PORT.
Tao Wu135f4ab2016-09-16 13:01:24 -07001513 if (!server_host_str && !server_port_str && !server_socket_str) {
1514 server_socket_str = getenv("ADB_SERVER_SOCKET");
Josh Gao9c869b52016-08-25 16:00:22 -07001515 }
1516
1517 if (!server_socket_str) {
1518 // tcp:1234 and tcp:localhost:1234 are different with -a, so don't default to localhost
1519 server_host_str = server_host_str ? server_host_str : getenv("ANDROID_ADB_SERVER_ADDRESS");
1520
Tao Wu135f4ab2016-09-16 13:01:24 -07001521 int server_port = DEFAULT_ADB_PORT;
Josh Gao9c869b52016-08-25 16:00:22 -07001522 server_port_str = server_port_str ? server_port_str : getenv("ANDROID_ADB_SERVER_PORT");
Tao Wu135f4ab2016-09-16 13:01:24 -07001523 if (server_port_str && strlen(server_port_str) > 0) {
1524 if (!android::base::ParseInt(server_port_str, &server_port, 1, 65535)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001525 error_exit(
1526 "$ANDROID_ADB_SERVER_PORT must be a positive number less than 65535: "
1527 "got \"%s\"",
1528 server_port_str);
Tao Wu135f4ab2016-09-16 13:01:24 -07001529 }
1530 }
Josh Gao9c869b52016-08-25 16:00:22 -07001531
1532 int rc;
1533 char* temp;
1534 if (server_host_str) {
Tao Wu135f4ab2016-09-16 13:01:24 -07001535 rc = asprintf(&temp, "tcp:%s:%d", server_host_str, server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001536 } else {
Tao Wu135f4ab2016-09-16 13:01:24 -07001537 rc = asprintf(&temp, "tcp:%d", server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001538 }
1539 if (rc < 0) {
Elliott Hughes4679a392018-10-19 13:59:44 -07001540 LOG(FATAL) << "failed to allocate server socket specification";
Josh Gao9c869b52016-08-25 16:00:22 -07001541 }
1542 server_socket_str = temp;
1543 }
1544
1545 adb_set_socket_spec(server_socket_str);
1546
Elliott Hughes8d28e192015-10-07 14:55:10 -07001547 // If none of -d, -e, or -s were specified, try $ANDROID_SERIAL.
1548 if (transport_type == kTransportAny && serial == nullptr) {
1549 serial = getenv("ANDROID_SERIAL");
1550 }
1551
Josh Gaob122b172017-08-16 16:57:01 -07001552 adb_set_transport(transport_type, serial, transport_id);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001553
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001554 if (is_server) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001555 if (no_daemon || is_daemon) {
Spencer Low5c398d22015-08-08 15:07:07 -07001556 if (is_daemon && (ack_reply_fd == -1)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001557 fprintf(stderr, "reply fd for adb server to client communication not specified.\n");
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001558 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001559 }
Josh Gao9c869b52016-08-25 16:00:22 -07001560 r = adb_server_main(is_daemon, server_socket_str, ack_reply_fd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001561 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001562 r = launch_server(server_socket_str);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001563 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001564 if (r) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001565 fprintf(stderr,"* could not start server *\n");
1566 }
1567 return r;
1568 }
1569
Riley Andrews98f58e82014-12-05 17:37:24 -08001570 if (argc == 0) {
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001571 help();
1572 return 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001573 }
1574
Riley Andrewsc8514c82014-12-05 17:32:46 -08001575 /* handle wait-for-* prefix */
1576 if (!strncmp(argv[0], "wait-for-", strlen("wait-for-"))) {
Dan Albertbac34742015-02-25 17:51:28 -08001577 const char* service = argv[0];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001578
Josh Gaob122b172017-08-16 16:57:01 -07001579 if (!wait_for_device(service)) {
Riley Andrewsc8514c82014-12-05 17:32:46 -08001580 return 1;
1581 }
1582
Elliott Hughes2b101112015-05-04 19:29:32 -07001583 // Allow a command to be run after wait-for-device,
1584 // e.g. 'adb wait-for-device shell'.
Riley Andrewsc8514c82014-12-05 17:32:46 -08001585 if (argc == 1) {
1586 return 0;
1587 }
1588
1589 /* Fall through */
1590 argc--;
1591 argv++;
1592 }
1593
1594 /* adb_connect() commands */
Riley Andrews98f58e82014-12-05 17:37:24 -08001595 if (!strcmp(argv[0], "devices")) {
Dan Albertbac34742015-02-25 17:51:28 -08001596 const char *listopt;
Elliott Hughes6452a892015-04-29 12:28:13 -07001597 if (argc < 2) {
Scott Andersone109d262012-04-20 11:21:14 -07001598 listopt = "";
Elliott Hughes6452a892015-04-29 12:28:13 -07001599 } else if (argc == 2 && !strcmp(argv[1], "-l")) {
Scott Andersone109d262012-04-20 11:21:14 -07001600 listopt = argv[1];
Elliott Hughes6452a892015-04-29 12:28:13 -07001601 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001602 error_exit("adb devices [-l]");
Scott Andersone109d262012-04-20 11:21:14 -07001603 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001604
1605 std::string query = android::base::StringPrintf("host:%s%s", argv[0], listopt);
Josh Gaod10b7c42019-03-14 15:38:02 -07001606 std::string error;
1607 if (!adb_check_server_version(&error)) {
1608 error_exit("failed to check server version: %s", error.c_str());
1609 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001610 printf("List of devices attached\n");
1611 return adb_query_command(query);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001612 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001613 else if (!strcmp(argv[0], "connect")) {
Elliott Hughes573e67c2019-07-30 13:51:03 -07001614 if (argc != 2) error_exit("usage: adb connect HOST[:PORT>]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001615
1616 std::string query = android::base::StringPrintf("host:connect:%s", argv[1]);
1617 return adb_query_command(query);
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001618 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001619 else if (!strcmp(argv[0], "disconnect")) {
Elliott Hughes573e67c2019-07-30 13:51:03 -07001620 if (argc > 2) error_exit("usage: adb disconnect [HOST[:PORT]]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001621
1622 std::string query = android::base::StringPrintf("host:disconnect:%s",
1623 (argc == 2) ? argv[1] : "");
1624 return adb_query_command(query);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001625 } else if (!strcmp(argv[0], "abb")) {
1626 return adb_abb(argc, argv);
1627 } else if (!strcmp(argv[0], "emu")) {
Spencer Low142ec752015-05-06 16:13:42 -07001628 return adb_send_emulator_command(argc, argv, serial);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001629 } else if (!strcmp(argv[0], "shell")) {
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001630 return adb_shell(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001631 } else if (!strcmp(argv[0], "exec-in") || !strcmp(argv[0], "exec-out")) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001632 int exec_in = !strcmp(argv[0], "exec-in");
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001633
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001634 if (argc < 2) error_exit("usage: adb %s command", argv[0]);
Zach Riggleccafc792016-04-14 17:18:04 -04001635
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001636 std::string cmd = "exec:";
1637 cmd += argv[1];
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001638 argc -= 2;
1639 argv += 2;
1640 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001641 cmd += " " + escape_arg(*argv++);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001642 }
1643
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001644 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001645 unique_fd fd(adb_connect(cmd, &error));
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001646 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001647 fprintf(stderr, "error: %s\n", error.c_str());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001648 return -1;
1649 }
1650
1651 if (exec_in) {
Josh Gao74ccdf92019-01-23 15:36:42 -08001652 copy_to_file(STDIN_FILENO, fd.get());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001653 } else {
Josh Gao74ccdf92019-01-23 15:36:42 -08001654 copy_to_file(fd.get(), STDOUT_FILENO);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001655 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001656 return 0;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001657 } else if (!strcmp(argv[0], "kill-server")) {
Josh Gaob3c14ec2017-05-08 18:37:17 -07001658 return adb_kill_server() ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001659 } else if (!strcmp(argv[0], "sideload")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001660 if (argc != 2) error_exit("sideload requires an argument");
Tao Bao11cf30b2019-04-07 23:24:03 -07001661 if (adb_sideload_install(argv[1], false /* rescue_mode */)) {
Doug Zongker447f0612012-01-09 14:54:53 -08001662 return 1;
1663 } else {
1664 return 0;
1665 }
Tao Bao11cf30b2019-04-07 23:24:03 -07001666 } else if (!strcmp(argv[0], "rescue")) {
Tao Bao599eee12019-06-04 14:37:06 -07001667 // adb rescue getprop
Tao Bao11cf30b2019-04-07 23:24:03 -07001668 // adb rescue getprop <prop>
1669 // adb rescue install <filename>
xunchangc2265c52019-04-22 23:14:28 -07001670 // adb rescue wipe userdata
Tao Bao599eee12019-06-04 14:37:06 -07001671 if (argc < 2) error_exit("rescue requires at least one argument");
Tao Bao11cf30b2019-04-07 23:24:03 -07001672 if (!strcmp(argv[1], "getprop")) {
Tao Bao599eee12019-06-04 14:37:06 -07001673 if (argc == 2) {
1674 return adb_connect_command("rescue-getprop:");
1675 }
1676 if (argc == 3) {
1677 return adb_connect_command(
1678 android::base::StringPrintf("rescue-getprop:%s", argv[2]));
1679 }
1680 error_exit("invalid rescue getprop arguments");
Tao Bao11cf30b2019-04-07 23:24:03 -07001681 } else if (!strcmp(argv[1], "install")) {
Tao Bao599eee12019-06-04 14:37:06 -07001682 if (argc != 3) error_exit("rescue install requires two arguments");
Tao Bao11cf30b2019-04-07 23:24:03 -07001683 if (adb_sideload_install(argv[2], true /* rescue_mode */) != 0) {
1684 return 1;
1685 }
Tao Bao599eee12019-06-04 14:37:06 -07001686 } else if (!strcmp(argv[1], "wipe")) {
1687 if (argc != 3 || strcmp(argv[2], "userdata") != 0) {
1688 error_exit("invalid rescue wipe arguments");
1689 }
xunchangc2265c52019-04-22 23:14:28 -07001690 return adb_wipe_devices();
Tao Bao11cf30b2019-04-07 23:24:03 -07001691 } else {
1692 error_exit("invalid rescue argument");
1693 }
1694 return 0;
Elliott Hughese1632982017-08-23 15:42:28 -07001695 } else if (!strcmp(argv[0], "tcpip")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001696 if (argc != 2) error_exit("tcpip requires an argument");
Elliott Hughese1632982017-08-23 15:42:28 -07001697 int port;
1698 if (!android::base::ParseInt(argv[1], &port, 1, 65535)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001699 error_exit("tcpip: invalid port: %s", argv[1]);
Elliott Hughese1632982017-08-23 15:42:28 -07001700 }
1701 return adb_connect_command(android::base::StringPrintf("tcpip:%d", port));
Josh Gao8c2198c2019-07-11 14:15:32 -07001702 } else if (!strcmp(argv[0], "remount")) {
1703 FeatureSet features;
1704 std::string error;
1705 if (!adb_get_feature_set(&features, &error)) {
1706 fprintf(stderr, "error: %s\n", error.c_str());
1707 return 1;
1708 }
1709
1710 if (CanUseFeature(features, kFeatureRemountShell)) {
Mark Salyzyn6f908ce2019-09-17 08:01:15 -07001711 std::vector<const char*> args = {"shell"};
1712 args.insert(args.cend(), argv, argv + argc);
1713 return adb_shell(args.size(), args.data());
1714 } else if (argc > 1) {
1715 auto command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
1716 return adb_connect_command(command);
Josh Gao8c2198c2019-07-11 14:15:32 -07001717 } else {
1718 return adb_connect_command("remount:");
1719 }
Elliott Hughes19d80b82015-07-21 16:13:40 -07001720 }
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001721 // clang-format off
Josh Gao8c2198c2019-07-11 14:15:32 -07001722 else if (!strcmp(argv[0], "reboot") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001723 !strcmp(argv[0], "reboot-bootloader") ||
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001724 !strcmp(argv[0], "reboot-fastboot") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001725 !strcmp(argv[0], "usb") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001726 !strcmp(argv[0], "disable-verity") ||
1727 !strcmp(argv[0], "enable-verity")) {
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001728 // clang-format on
Elliott Hughes5677c232015-05-07 23:37:40 -07001729 std::string command;
Tao Bao175b7bb2015-03-29 11:22:34 -07001730 if (!strcmp(argv[0], "reboot-bootloader")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001731 command = "reboot:bootloader";
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001732 } else if (!strcmp(argv[0], "reboot-fastboot")) {
1733 command = "reboot:fastboot";
Tao Bao175b7bb2015-03-29 11:22:34 -07001734 } else if (argc > 1) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001735 command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
Tao Bao175b7bb2015-03-29 11:22:34 -07001736 } else {
Elliott Hughes5677c232015-05-07 23:37:40 -07001737 command = android::base::StringPrintf("%s:", argv[0]);
The Android Open Source Projecte037fd72009-03-13 13:04:37 -07001738 }
Tao Bao175b7bb2015-03-29 11:22:34 -07001739 return adb_connect_command(command);
Josh Gao06c73ae2016-03-04 15:16:18 -08001740 } else if (!strcmp(argv[0], "root") || !strcmp(argv[0], "unroot")) {
1741 return adb_root(argv[0]) ? 0 : 1;
1742 } else if (!strcmp(argv[0], "bugreport")) {
Felipe Leme698e0652016-07-19 17:07:22 -07001743 Bugreport bugreport;
Josh Gaob122b172017-08-16 16:57:01 -07001744 return bugreport.DoIt(argc, argv);
Felipe Leme44a42672016-04-01 17:43:27 -07001745 } else if (!strcmp(argv[0], "forward") || !strcmp(argv[0], "reverse")) {
Elliott Hughes424af022015-05-29 17:55:19 -07001746 bool reverse = !strcmp(argv[0], "reverse");
Elliott Hughes424af022015-05-29 17:55:19 -07001747 --argc;
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001748 if (argc < 1) error_exit("%s requires an argument", argv[0]);
Josh Gao258d4a92018-07-31 15:36:03 -07001749 ++argv;
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001750
1751 // Determine the <host-prefix> for this command.
Elliott Hughes424af022015-05-29 17:55:19 -07001752 std::string host_prefix;
David 'Digit' Turner25258692013-03-21 21:07:42 +01001753 if (reverse) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001754 host_prefix = "reverse:";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001755 } else {
Josh Gaoda9c7672019-08-06 16:43:34 -07001756 host_prefix = "host:";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001757 }
1758
Elliott Hughes4679a392018-10-19 13:59:44 -07001759 std::string cmd, error_message;
Elliott Hughes424af022015-05-29 17:55:19 -07001760 if (strcmp(argv[0], "--list") == 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001761 if (argc != 1) error_exit("--list doesn't take any arguments");
Josh Gaoda9c7672019-08-06 16:43:34 -07001762 return adb_query_command(host_prefix + "list-forward");
Elliott Hughes424af022015-05-29 17:55:19 -07001763 } else if (strcmp(argv[0], "--remove-all") == 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001764 if (argc != 1) error_exit("--remove-all doesn't take any arguments");
Josh Gaoda9c7672019-08-06 16:43:34 -07001765 cmd = "killforward-all";
Elliott Hughes424af022015-05-29 17:55:19 -07001766 } else if (strcmp(argv[0], "--remove") == 0) {
1767 // forward --remove <local>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001768 if (argc != 2) error_exit("--remove requires an argument");
Josh Gaoda9c7672019-08-06 16:43:34 -07001769 cmd = std::string("killforward:") + argv[1];
Elliott Hughes424af022015-05-29 17:55:19 -07001770 } else if (strcmp(argv[0], "--no-rebind") == 0) {
1771 // forward --no-rebind <local> <remote>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001772 if (argc != 3) error_exit("--no-rebind takes two arguments");
Elliott Hughes4679a392018-10-19 13:59:44 -07001773 if (forward_targets_are_valid(argv[1], argv[2], &error_message)) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001774 cmd = std::string("forward:norebind:") + argv[1] + ";" + argv[2];
David Purselleaae97e2016-04-07 11:25:48 -07001775 }
Elliott Hughes424af022015-05-29 17:55:19 -07001776 } else {
1777 // forward <local> <remote>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001778 if (argc != 2) error_exit("forward takes two arguments");
Elliott Hughes4679a392018-10-19 13:59:44 -07001779 if (forward_targets_are_valid(argv[0], argv[1], &error_message)) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001780 cmd = std::string("forward:") + argv[0] + ";" + argv[1];
David Purselleaae97e2016-04-07 11:25:48 -07001781 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001782 }
1783
Elliott Hughes4679a392018-10-19 13:59:44 -07001784 if (!error_message.empty()) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001785 error_exit("error: %s", error_message.c_str());
David Purselleaae97e2016-04-07 11:25:48 -07001786 }
1787
Josh Gaoda9c7672019-08-06 16:43:34 -07001788 unique_fd fd(adb_connect(nullptr, host_prefix + cmd, &error_message, true));
Josh Gao74ccdf92019-01-23 15:36:42 -08001789 if (fd < 0 || !adb_status(fd.get(), &error_message)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001790 error_exit("error: %s", error_message.c_str());
David Purselleaae97e2016-04-07 11:25:48 -07001791 }
1792
1793 // Server or device may optionally return a resolved TCP port number.
1794 std::string resolved_port;
Elliott Hughes4679a392018-10-19 13:59:44 -07001795 if (ReadProtocolString(fd, &resolved_port, &error_message) && !resolved_port.empty()) {
David Purselleaae97e2016-04-07 11:25:48 -07001796 printf("%s\n", resolved_port.c_str());
1797 }
1798
1799 ReadOrderlyShutdown(fd);
1800 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001801 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001802 /* do_sync_*() commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001803 else if (!strcmp(argv[0], "ls")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001804 if (argc != 2) error_exit("ls requires an argument");
Elliott Hughesaa245492015-08-03 10:38:08 -07001805 return do_sync_ls(argv[1]) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001806 } else if (!strcmp(argv[0], "push")) {
Josh Gao05786772015-10-30 16:57:19 -07001807 bool copy_attrs = false;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001808 bool sync = false;
Josh Gao05786772015-10-30 16:57:19 -07001809 std::vector<const char*> srcs;
1810 const char* dst = nullptr;
Mark Lindner76f2a932014-03-11 17:55:59 -07001811
Dan Albert06b0d6b2017-05-18 22:56:48 -07001812 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, &sync);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001813 if (srcs.empty() || !dst) error_exit("push requires an argument");
Dan Albert06b0d6b2017-05-18 22:56:48 -07001814 return do_sync_push(srcs, dst, sync) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001815 } else if (!strcmp(argv[0], "pull")) {
Josh Gao05786772015-10-30 16:57:19 -07001816 bool copy_attrs = false;
1817 std::vector<const char*> srcs;
1818 const char* dst = ".";
Mark Lindner76f2a932014-03-11 17:55:59 -07001819
Dan Albert06b0d6b2017-05-18 22:56:48 -07001820 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, nullptr);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001821 if (srcs.empty()) error_exit("pull requires an argument");
Josh Gao05786772015-10-30 16:57:19 -07001822 return do_sync_pull(srcs, dst, copy_attrs) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001823 } else if (!strcmp(argv[0], "install")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001824 if (argc < 2) error_exit("install requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001825 return install_app(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001826 } else if (!strcmp(argv[0], "install-multiple")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001827 if (argc < 2) error_exit("install-multiple requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001828 return install_multiple_app(argc, argv);
Patrick Baumanndc58b0a2018-10-09 10:45:03 -07001829 } else if (!strcmp(argv[0], "install-multi-package")) {
Mohammad Samiul Islamb581ce92019-05-29 15:00:00 +01001830 if (argc < 2) error_exit("install-multi-package requires an argument");
Patrick Baumanndc58b0a2018-10-09 10:45:03 -07001831 return install_multi_package(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001832 } else if (!strcmp(argv[0], "uninstall")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001833 if (argc < 2) error_exit("uninstall requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001834 return uninstall_app(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001835 } else if (!strcmp(argv[0], "sync")) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001836 std::string src;
Elliott Hughes58305772015-04-17 13:57:15 -07001837 bool list_only = false;
Riley Andrews98f58e82014-12-05 17:37:24 -08001838 if (argc < 2) {
Elliott Hughes57e97cb2018-04-03 10:54:39 -07001839 // No partition specified: sync all of them.
Anthony Newnam705c9442010-02-22 08:36:49 -06001840 } else if (argc >= 2 && strcmp(argv[1], "-l") == 0) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001841 list_only = true;
Elliott Hughes57e97cb2018-04-03 10:54:39 -07001842 if (argc == 3) src = argv[2];
Riley Andrews98f58e82014-12-05 17:37:24 -08001843 } else if (argc == 2) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001844 src = argv[1];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001845 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001846 error_exit("usage: adb sync [-l] [PARTITION]");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001847 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001848
Elliott Hughes57e97cb2018-04-03 10:54:39 -07001849 if (src.empty()) src = "all";
Justin Yun5d0ac622019-06-26 09:18:03 +09001850 std::vector<std::string> partitions{"data", "odm", "oem", "product",
1851 "system", "system_ext", "vendor"};
Elliott Hughes57e97cb2018-04-03 10:54:39 -07001852 bool found = false;
1853 for (const auto& partition : partitions) {
1854 if (src == "all" || src == partition) {
1855 std::string src_dir{product_file(partition)};
1856 if (!directory_exists(src_dir)) continue;
1857 found = true;
1858 if (!do_sync_sync(src_dir, "/" + partition, list_only)) return 1;
1859 }
Elliott Hughes58305772015-04-17 13:57:15 -07001860 }
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001861 if (!found) error_exit("don't know how to sync %s partition", src.c_str());
Elliott Hughes4679a392018-10-19 13:59:44 -07001862 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001863 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001864 /* passthrough commands */
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001865 else if (!strcmp(argv[0], "get-state") || !strcmp(argv[0], "get-serialno") ||
1866 !strcmp(argv[0], "get-devpath")) {
Josh Gaob122b172017-08-16 16:57:01 -07001867 return adb_query_command(format_host_command(argv[0]));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001868 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001869 /* other commands */
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001870 else if (!strcmp(argv[0], "logcat") || !strcmp(argv[0], "lolcat") ||
1871 !strcmp(argv[0], "longcat")) {
Josh Gaob122b172017-08-16 16:57:01 -07001872 return logcat(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001873 } else if (!strcmp(argv[0], "ppp")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001874 return ppp(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001875 } else if (!strcmp(argv[0], "start-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001876 std::string error;
Spencer Lowf18fc082015-08-11 17:05:02 -07001877 const int result = adb_connect("host:start-server", &error);
1878 if (result < 0) {
1879 fprintf(stderr, "error: %s\n", error.c_str());
1880 }
1881 return result;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001882 } else if (!strcmp(argv[0], "backup")) {
Christopher Tated2f54152011-04-21 12:53:28 -07001883 return backup(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001884 } else if (!strcmp(argv[0], "restore")) {
Christopher Tate702967a2011-05-17 15:52:54 -07001885 return restore(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001886 } else if (!strcmp(argv[0], "keygen")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001887 if (argc != 2) error_exit("keygen requires an argument");
Yabin Cuiaed3c612015-09-22 15:52:57 -07001888 // Always print key generation information for keygen command.
1889 adb_trace_enable(AUTH);
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08001890 return adb_auth_keygen(argv[1]);
Josh Gao2dc4cab2018-11-15 17:45:46 -08001891 } else if (!strcmp(argv[0], "pubkey")) {
1892 if (argc != 2) error_exit("pubkey requires an argument");
1893 return adb_auth_pubkey(argv[1]);
1894 } else if (!strcmp(argv[0], "jdwp")) {
Tao Bao175b7bb2015-03-29 11:22:34 -07001895 return adb_connect_command("jdwp");
Josh Gao2dc4cab2018-11-15 17:45:46 -08001896 } else if (!strcmp(argv[0], "track-jdwp")) {
Josh Gaob463baf2016-05-13 18:18:23 -07001897 return adb_connect_command("track-jdwp");
Josh Gao2dc4cab2018-11-15 17:45:46 -08001898 } else if (!strcmp(argv[0], "track-devices")) {
Elliott Hughesfc2f5f62019-07-31 14:13:57 -07001899 if (argc > 2 || (argc == 2 && strcmp(argv[1], "-l"))) {
1900 error_exit("usage: adb track-devices [-l]");
1901 }
1902 return adb_connect_command(argc == 2 ? "host:track-devices-l" : "host:track-devices");
Josh Gao172d0d42018-06-18 14:51:56 -07001903 } else if (!strcmp(argv[0], "raw")) {
1904 if (argc != 2) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001905 error_exit("usage: adb raw SERVICE");
Josh Gao172d0d42018-06-18 14:51:56 -07001906 }
Josh Gaoe89a55d2018-11-16 14:47:59 -08001907 return adb_connect_command_bidirectional(argv[1]);
Josh Gaob463baf2016-05-13 18:18:23 -07001908 }
1909
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001910 /* "adb /?" is a common idiom under Windows */
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001911 else if (!strcmp(argv[0], "--help") || !strcmp(argv[0], "help") || !strcmp(argv[0], "/?")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001912 help();
1913 return 0;
Josh Gao2dc4cab2018-11-15 17:45:46 -08001914 } else if (!strcmp(argv[0], "--version") || !strcmp(argv[0], "version")) {
Elliott Hughes42ae2602015-08-12 08:32:10 -07001915 fprintf(stdout, "%s", adb_version().c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001916 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08001917 } else if (!strcmp(argv[0], "features")) {
David Pursell4e2fd362015-09-22 10:43:08 -07001918 // Only list the features common to both the adb client and the device.
David Pursell0aacbbe2016-01-21 19:56:50 -08001919 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001920 std::string error;
1921 if (!adb_get_feature_set(&features, &error)) {
1922 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08001923 return 1;
1924 }
1925
David Pursell4e2fd362015-09-22 10:43:08 -07001926 for (const std::string& name : features) {
David Pursell70ef7b42015-09-30 13:35:42 -07001927 if (CanUseFeature(features, name)) {
David Pursell4e2fd362015-09-22 10:43:08 -07001928 printf("%s\n", name.c_str());
1929 }
1930 }
1931 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08001932 } else if (!strcmp(argv[0], "host-features")) {
1933 return adb_query_command("host:host-features");
Yabin Cui1f4ec192016-04-05 13:50:44 -07001934 } else if (!strcmp(argv[0], "reconnect")) {
1935 if (argc == 1) {
Josh Gaob122b172017-08-16 16:57:01 -07001936 return adb_query_command(format_host_command(argv[0]));
Josh Gao22d2b3e2016-10-27 14:01:08 -07001937 } else if (argc == 2) {
1938 if (!strcmp(argv[1], "device")) {
1939 std::string err;
1940 adb_connect("reconnect", &err);
1941 return 0;
1942 } else if (!strcmp(argv[1], "offline")) {
1943 std::string err;
1944 return adb_query_command("host:reconnect-offline");
1945 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001946 error_exit("usage: adb reconnect [device|offline]");
Josh Gao22d2b3e2016-10-27 14:01:08 -07001947 }
Yabin Cui1f4ec192016-04-05 13:50:44 -07001948 }
Dan Albert90d4b732015-05-20 18:58:41 -07001949 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001950
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001951 error_exit("unknown command %s", argv[0]);
Elliott Hughes4679a392018-10-19 13:59:44 -07001952 __builtin_unreachable();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001953}