blob: 7b8b2d7702fae43c84f88ed010ee877203b13f33 [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>
Joshua Duongd85f5c02019-11-20 14:18:43 -080033#include <iostream>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080034
David Pursell606835a2015-09-08 17:17:02 -070035#include <memory>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070036#include <string>
Elliott Hughesdbe91ee2016-11-15 12:37:32 -080037#include <thread>
Josh Gao05786772015-10-30 16:57:19 -070038#include <vector>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070039
Elliott Hughese8b663f2016-05-26 22:43:19 -070040#include <android-base/file.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080041#include <android-base/logging.h>
Tao Wu135f4ab2016-09-16 13:01:24 -070042#include <android-base/parseint.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080043#include <android-base/stringprintf.h>
44#include <android-base/strings.h>
Elliott Hughes2baae3a2015-04-17 10:59:34 -070045
Yabin Cuid325e862014-11-17 14:48:25 -080046#if !defined(_WIN32)
Josh Gao8dcdb572015-10-23 15:03:31 -070047#include <signal.h>
Elliott Hughes0fbf9612015-11-04 13:07:47 -080048#include <sys/ioctl.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080049#include <termios.h>
Dan Albert76649012015-02-24 15:51:19 -080050#include <unistd.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080051#endif
52
Shukang Zhouf4ffae12020-02-13 17:01:39 -080053#include <google/protobuf/text_format.h>
54
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080055#include "adb.h"
Nick Kralevichbea3f9c2014-11-13 15:17:29 -080056#include "adb_auth.h"
Dan Albertcc731cc2015-02-24 21:26:58 -080057#include "adb_client.h"
Idries Hamadied409ea2018-01-29 16:30:36 +000058#include "adb_install.h"
Dan Albertcc731cc2015-02-24 21:26:58 -080059#include "adb_io.h"
Josh Gao924d35a2016-08-30 15:39:25 -070060#include "adb_unique_fd.h"
Elliott Hughes58305772015-04-17 13:57:15 -070061#include "adb_utils.h"
Shukang Zhouf4ffae12020-02-13 17:01:39 -080062#include "app_processes.pb.h"
Felipe Leme698e0652016-07-19 17:07:22 -070063#include "bugreport.h"
Josh Gaoedec0332018-07-25 15:55:25 -070064#include "client/file_sync_client.h"
Felipe Leme698e0652016-07-19 17:07:22 -070065#include "commandline.h"
Idries Hamadied409ea2018-01-29 16:30:36 +000066#include "fastdeploy.h"
Alex Buynytskyy96ff54b2020-02-13 06:52:04 -080067#include "incremental_server.h"
David Pursell70ef7b42015-09-30 13:35:42 -070068#include "services.h"
Josh Gao86acb862018-07-25 16:07:26 -070069#include "shell_protocol.h"
Josh Gao4602adb2016-11-15 18:55:47 -080070#include "sysdeps/chrono.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080071
Matt Gumbeld7b33082012-11-14 10:16:17 -080072extern int gListenAll;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080073
Felipe Leme0d4f0502016-07-26 12:14:39 -070074DefaultStandardStreamsCallback DEFAULT_STANDARD_STREAMS_CALLBACK(nullptr, nullptr);
75
Elliott Hughes57e97cb2018-04-03 10:54:39 -070076static std::string product_file(const std::string& file) {
Elliott Hughes0754cb92017-05-01 11:04:31 -070077 const char* ANDROID_PRODUCT_OUT = getenv("ANDROID_PRODUCT_OUT");
78 if (ANDROID_PRODUCT_OUT == nullptr) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -070079 error_exit("product directory not specified; set $ANDROID_PRODUCT_OUT");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080080 }
Elliott Hughes57e97cb2018-04-03 10:54:39 -070081 return std::string{ANDROID_PRODUCT_OUT} + OS_PATH_SEPARATOR_STR + file;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080082}
83
Elliott Hughes58305772015-04-17 13:57:15 -070084static void help() {
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070085 fprintf(stdout, "%s\n", adb_version().c_str());
Felipe Lemeb3239722016-07-29 17:57:00 -070086 // clang-format off
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070087 fprintf(stdout,
88 "global options:\n"
89 " -a listen on all network interfaces, not just localhost\n"
90 " -d use USB device (error if multiple devices connected)\n"
Elliott Hughes83ab5c22017-01-13 16:58:25 -080091 " -e use TCP/IP device (error if multiple TCP/IP devices available)\n"
Elliott Hughes0754cb92017-05-01 11:04:31 -070092 " -s SERIAL use device with given serial (overrides $ANDROID_SERIAL)\n"
Josh Gaob122b172017-08-16 16:57:01 -070093 " -t ID use device with given transport id\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070094 " -H name of adb server host [default=localhost]\n"
95 " -P port of adb server [default=5037]\n"
96 " -L SOCKET listen on given socket for adb server [default=tcp:localhost:5037]\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080097 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -070098 "general commands:\n"
99 " devices [-l] list connected devices (-l for long output)\n"
100 " help show this help message\n"
101 " version show version num\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700102 "\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800103 "networking:\n"
Joshua Duongd85f5c02019-11-20 14:18:43 -0800104 " connect HOST[:PORT] connect to a device via TCP/IP [default port=5555]\n"
105 " disconnect [HOST[:PORT]]\n"
106 " disconnect from given TCP/IP device [default port=5555], or all\n"
107 " pair HOST[:PORT] pair with a device for secure TCP/IP communication\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700108 " forward --list list all forward socket connections\n"
109 " forward [--no-rebind] LOCAL REMOTE\n"
110 " forward socket connection using:\n"
111 " tcp:<port> (<local> may be \"tcp:0\" to pick any open port)\n"
112 " localabstract:<unix domain socket name>\n"
113 " localreserved:<unix domain socket name>\n"
114 " localfilesystem:<unix domain socket name>\n"
115 " dev:<character device name>\n"
116 " jdwp:<process pid> (remote only)\n"
Daniel Colascione3ec7be72019-11-13 17:49:37 -0800117 " acceptfd:<fd> (listen only)\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700118 " forward --remove LOCAL remove specific forward socket connection\n"
119 " forward --remove-all remove all forward socket connections\n"
120 " ppp TTY [PARAMETER...] run PPP over USB\n"
121 " reverse --list list all reverse socket connections from device\n"
122 " reverse [--no-rebind] REMOTE LOCAL\n"
123 " reverse socket connection using:\n"
124 " tcp:<port> (<remote> may be \"tcp:0\" to pick any open port)\n"
125 " localabstract:<unix domain socket name>\n"
126 " localreserved:<unix domain socket name>\n"
127 " localfilesystem:<unix domain socket name>\n"
128 " reverse --remove REMOTE remove specific reverse socket connection\n"
129 " reverse --remove-all remove all reverse socket connections from device\n"
Joshua Duong504d3932020-03-31 10:58:50 -0700130 " mdns check check if mdns discovery is available\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800131 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700132 "file transfer:\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700133 " push [--sync] [-z ALGORITHM] [-Z] LOCAL... REMOTE\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700134 " copy local files/directories to device\n"
Dan Albert06b0d6b2017-05-18 22:56:48 -0700135 " --sync: only push files that are newer on the host than the device\n"
Josh Gao5949fcc2020-03-30 23:25:16 -0700136 " -n: dry run: push files to device without storing to the filesystem\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700137 " -z: enable compression with a specified algorithm (any, none, brotli)\n"
Josh Gao939fc192020-03-04 19:34:08 -0800138 " -Z: disable compression\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700139 " pull [-a] [-z ALGORITHM] [-Z] REMOTE... LOCAL\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700140 " copy files/dirs from device\n"
141 " -a: preserve file timestamp and mode\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700142 " -z: enable compression with a specified algorithm (any, none, brotli)\n"
Josh Gao939fc192020-03-04 19:34:08 -0800143 " -Z: disable compression\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700144 " sync [-l] [-z ALGORITHM] [-Z] [all|data|odm|oem|product|system|system_ext|vendor]\n"
Elliott Hughes0754cb92017-05-01 11:04:31 -0700145 " sync a local build from $ANDROID_PRODUCT_OUT to the device (default all)\n"
Josh Gao5949fcc2020-03-30 23:25:16 -0700146 " -n: dry run: push files to device without storing to the filesystem\n"
Elliott Hughes4c8ea892019-07-30 10:32:53 -0700147 " -l: list files that would be copied, but don't copy them\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700148 " -z: enable compression with a specified algorithm (any, none, brotli)\n"
Josh Gao939fc192020-03-04 19:34:08 -0800149 " -Z: disable compression\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800150 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700151 "shell:\n"
152 " shell [-e ESCAPE] [-n] [-Tt] [-x] [COMMAND...]\n"
153 " run remote shell command (interactive shell if no command given)\n"
154 " -e: choose escape character, or \"none\"; default '~'\n"
155 " -n: don't read from stdin\n"
Elliott Hughesdfe6c842019-12-13 16:43:10 -0800156 " -T: disable pty allocation\n"
157 " -t: allocate a pty if on a tty (-tt: force pty allocation)\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700158 " -x: disable remote exit codes and stdout/stderr separation\n"
159 " emu COMMAND run emulator console command\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800160 "\n"
Elliott Hughes0d4bf802019-01-11 13:50:05 -0800161 "app installation (see also `adb shell cmd package help`):\n"
Felipe Leme807be772018-05-08 18:40:18 -0700162 " install [-lrtsdg] [--instant] PACKAGE\n"
Patrick Baumanndc58b0a2018-10-09 10:45:03 -0700163 " push a single package to the device and install it\n"
Felipe Leme807be772018-05-08 18:40:18 -0700164 " install-multiple [-lrtsdpg] [--instant] PACKAGE...\n"
Patrick Baumanndc58b0a2018-10-09 10:45:03 -0700165 " push multiple APKs to the device for a single package and install them\n"
166 " install-multi-package [-lrtsdpg] [--instant] PACKAGE...\n"
167 " push one or more packages to the device and install them atomically\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700168 " -r: replace existing application\n"
169 " -t: allow test packages\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700170 " -d: allow version code downgrade (debuggable packages only)\n"
171 " -p: partial application install (install-multiple only)\n"
172 " -g: grant all runtime permissions\n"
Elliott Hughesc0ebd592019-08-02 17:09:30 -0700173 " --abi ABI: override platform's default ABI\n"
Felipe Leme807be772018-05-08 18:40:18 -0700174 " --instant: cause the app to be installed as an ephemeral install app\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000175 " --no-streaming: always push APK to device and invoke Package Manager as separate steps\n"
176 " --streaming: force streaming APK directly into Package Manager\n"
Henry Daitxee01c802018-12-12 10:40:57 +0000177 " --fastdeploy: use fast deploy\n"
178 " --no-fastdeploy: prevent use of fast deploy\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000179 " --force-agent: force update of deployment agent when using fast deploy\n"
180 " --date-check-agent: update deployment agent when local version is newer and using fast deploy\n"
181 " --version-check-agent: update deployment agent when local version has different version code and using fast deploy\n"
182#ifndef _WIN32
183 " --local-agent: locate agent files from local source build (instead of SDK location)\n"
184#endif
Elliott Hughesc0ebd592019-08-02 17:09:30 -0700185 " (See also `adb shell pm help` for more options.)\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000186 //TODO--installlog <filename>
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700187 " uninstall [-k] PACKAGE\n"
188 " remove this app package from the device\n"
189 " '-k': keep the data and cache directories\n"
190 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700191 "debugging:\n"
192 " bugreport [PATH]\n"
193 " write bugreport to given PATH [default=bugreport.zip];\n"
194 " if PATH is a directory, the bug report is saved in that directory.\n"
195 " devices that don't support zipped bug reports output to stdout.\n"
196 " jdwp list pids of processes hosting a JDWP transport\n"
197 " logcat show device log (logcat --help for more)\n"
198 "\n"
199 "security:\n"
200 " disable-verity disable dm-verity checking on userdebug builds\n"
201 " enable-verity re-enable dm-verity checking on userdebug builds\n"
202 " keygen FILE\n"
203 " generate adb public/private key; private key stored in FILE,\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700204 "\n"
205 "scripting:\n"
Elliott Hughesa52b4582020-03-10 16:57:47 -0700206 " wait-for[-TRANSPORT]-STATE...\n"
207 " wait for device to be in a given state\n"
Tao Bao55d407e2019-04-07 23:24:03 -0700208 " STATE: device, recovery, rescue, sideload, bootloader, or disconnect\n"
Josh Gao1e9e4712019-02-22 14:01:36 -0800209 " TRANSPORT: usb, local, or any [default=any]\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700210 " get-state print offline | bootloader | device\n"
211 " get-serialno print <serial-number>\n"
212 " get-devpath print <device-path>\n"
David Andersona5db1d72018-11-21 13:43:22 -0800213 " remount [-R]\n"
214 " remount partitions read-write. if a reboot is required, -R will\n"
215 " will automatically reboot the device.\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700216 " reboot [bootloader|recovery|sideload|sideload-auto-reboot]\n"
217 " reboot the device; defaults to booting system image but\n"
218 " supports bootloader and recovery too. sideload reboots\n"
219 " into recovery and automatically starts sideload mode,\n"
220 " sideload-auto-reboot is the same but reboots after sideloading.\n"
221 " sideload OTAPACKAGE sideload the given full OTA package\n"
222 " root restart adbd with root permissions\n"
223 " unroot restart adbd without root permissions\n"
bohuea0c3cb2018-10-24 10:38:33 -0700224 " usb restart adbd listening on USB\n"
225 " tcpip PORT restart adbd listening on TCP on PORT\n"
Timcd643152010-02-16 20:18:29 +0000226 "\n"
Yabin Cui1f4ec192016-04-05 13:50:44 -0700227 "internal debugging:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700228 " start-server ensure that there is a server running\n"
229 " kill-server kill the server if it is running\n"
230 " reconnect kick connection from host side to force reconnect\n"
231 " reconnect device kick connection from device side to force reconnect\n"
Yabin Cuib5e11412017-03-10 16:01:01 -0800232 " reconnect offline reset offline/unauthorized devices to force reconnect\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700233 "\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700234 "environment variables:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700235 " $ADB_TRACE\n"
236 " comma-separated list of debug info to log:\n"
237 " all,adb,sockets,packets,rwx,usb,sync,sysdeps,transport,jdwp\n"
238 " $ADB_VENDOR_KEYS colon-separated list of keys (files or directories)\n"
239 " $ANDROID_SERIAL serial number to connect to (see -s)\n"
Tim Baverstock25173c82019-02-08 16:27:16 +0000240 " $ANDROID_LOG_TAGS tags to be used by logcat (see logcat --help)\n"
241 " $ADB_LOCAL_TRANSPORT_MAX_PORT max emulator scan port (default 5585, 16 emus)\n"
242 );
Felipe Lemeb3239722016-07-29 17:57:00 -0700243 // clang-format on
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800244}
245
Yabin Cuid325e862014-11-17 14:48:25 -0800246#if defined(_WIN32)
247
Elliott Hughesa2f2e562015-04-16 16:47:02 -0700248// Implemented in sysdeps_win32.cpp.
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800249void stdin_raw_init();
250void stdin_raw_restore();
Yabin Cuid325e862014-11-17 14:48:25 -0800251
252#else
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100253static termios g_saved_terminal_state;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800254
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800255static void stdin_raw_init() {
256 if (tcgetattr(STDIN_FILENO, &g_saved_terminal_state)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800257
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100258 termios tio;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800259 if (tcgetattr(STDIN_FILENO, &tio)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800260
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100261 cfmakeraw(&tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800262
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100263 // No timeout but request at least one character per read.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800264 tio.c_cc[VTIME] = 0;
265 tio.c_cc[VMIN] = 1;
266
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800267 tcsetattr(STDIN_FILENO, TCSAFLUSH, &tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800268}
269
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800270static void stdin_raw_restore() {
271 tcsetattr(STDIN_FILENO, TCSAFLUSH, &g_saved_terminal_state);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800272}
273#endif
274
Alex Buynytskyy665f3ff2019-09-16 12:10:54 -0700275int read_and_dump(borrowed_fd fd, bool use_shell_protocol,
276 StandardStreamsCallbackInterface* callback) {
David Pursell606835a2015-09-08 17:17:02 -0700277 int exit_code = 0;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700278 if (fd < 0) return exit_code;
279
David Pursell606835a2015-09-08 17:17:02 -0700280 std::unique_ptr<ShellProtocol> protocol;
281 int length = 0;
David Pursell606835a2015-09-08 17:17:02 -0700282
283 char raw_buffer[BUFSIZ];
284 char* buffer_ptr = raw_buffer;
285 if (use_shell_protocol) {
Josh Gao31b5be62018-03-07 16:51:08 -0800286 protocol = std::make_unique<ShellProtocol>(fd);
David Pursell606835a2015-09-08 17:17:02 -0700287 if (!protocol) {
288 LOG(ERROR) << "failed to allocate memory for ShellProtocol object";
289 return 1;
290 }
291 buffer_ptr = protocol->data();
292 }
293
Felipe Lemed1dd1252016-06-06 16:05:36 -0700294 while (true) {
David Pursell606835a2015-09-08 17:17:02 -0700295 if (use_shell_protocol) {
296 if (!protocol->Read()) {
297 break;
298 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700299 length = protocol->data_length();
David Pursell606835a2015-09-08 17:17:02 -0700300 switch (protocol->id()) {
301 case ShellProtocol::kIdStdout:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700302 callback->OnStdout(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700303 break;
304 case ShellProtocol::kIdStderr:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700305 callback->OnStderr(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700306 break;
307 case ShellProtocol::kIdExit:
Josh Gao1e1ae452019-03-15 14:49:25 -0700308 // data() returns a char* which doesn't have defined signedness.
309 // Cast to uint8_t to prevent 255 from being sign extended to INT_MIN,
310 // which doesn't get truncated on Windows.
311 exit_code = static_cast<uint8_t>(protocol->data()[0]);
David Pursell606835a2015-09-08 17:17:02 -0700312 continue;
313 default:
314 continue;
315 }
316 length = protocol->data_length();
317 } else {
Josh Gao27241a72019-04-25 14:04:57 -0700318 D("read_and_dump(): pre adb_read(fd=%d)", fd.get());
David Pursell606835a2015-09-08 17:17:02 -0700319 length = adb_read(fd, raw_buffer, sizeof(raw_buffer));
Josh Gao27241a72019-04-25 14:04:57 -0700320 D("read_and_dump(): post adb_read(fd=%d): length=%d", fd.get(), length);
David Pursell606835a2015-09-08 17:17:02 -0700321 if (length <= 0) {
322 break;
323 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700324 callback->OnStdout(buffer_ptr, length);
Felipe Leme44a42672016-04-01 17:43:27 -0700325 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800326 }
David Pursell606835a2015-09-08 17:17:02 -0700327
Felipe Leme0d4f0502016-07-26 12:14:39 -0700328 return callback->Done(exit_code);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800329}
330
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700331static void stdinout_raw_prologue(int inFd, int outFd, int& old_stdin_mode, int& old_stdout_mode) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700332 if (inFd == STDIN_FILENO) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800333 stdin_raw_init();
Spencer Lowb7dfb792015-05-22 16:48:31 -0700334#ifdef _WIN32
335 old_stdin_mode = _setmode(STDIN_FILENO, _O_BINARY);
336 if (old_stdin_mode == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700337 PLOG(FATAL) << "could not set stdin to binary";
Spencer Lowb7dfb792015-05-22 16:48:31 -0700338 }
339#endif
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700340 }
Yabin Cuid325e862014-11-17 14:48:25 -0800341
Spencer Lowb7dfb792015-05-22 16:48:31 -0700342#ifdef _WIN32
343 if (outFd == STDOUT_FILENO) {
344 old_stdout_mode = _setmode(STDOUT_FILENO, _O_BINARY);
345 if (old_stdout_mode == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700346 PLOG(FATAL) << "could not set stdout to binary";
Spencer Lowb7dfb792015-05-22 16:48:31 -0700347 }
348 }
349#endif
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700350}
351
352static void stdinout_raw_epilogue(int inFd, int outFd, int old_stdin_mode, int old_stdout_mode) {
353 if (inFd == STDIN_FILENO) {
354 stdin_raw_restore();
355#ifdef _WIN32
356 if (_setmode(STDIN_FILENO, old_stdin_mode) == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700357 PLOG(FATAL) << "could not restore stdin mode";
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700358 }
359#endif
360 }
361
362#ifdef _WIN32
363 if (outFd == STDOUT_FILENO) {
364 if (_setmode(STDOUT_FILENO, old_stdout_mode) == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700365 PLOG(FATAL) << "could not restore stdout mode";
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700366 }
367 }
368#endif
369}
370
Josh Gao3828ebc2019-12-03 16:05:54 -0800371bool copy_to_file(int inFd, int outFd) {
372 bool result = true;
Yurii Zubrytskyi6eca0e62019-06-27 13:47:34 -0700373 std::vector<char> buf(64 * 1024);
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700374 int len;
375 long total = 0;
376 int old_stdin_mode = -1;
377 int old_stdout_mode = -1;
378
379 D("copy_to_file(%d -> %d)", inFd, outFd);
380
381 stdinout_raw_prologue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700382
Elliott Hughesa7090b92015-04-17 17:03:59 -0700383 while (true) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700384 if (inFd == STDIN_FILENO) {
Elliott Hughes86ab9ff2018-09-05 12:13:11 -0700385 len = unix_read(inFd, buf.data(), buf.size());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700386 } else {
Elliott Hughes86ab9ff2018-09-05 12:13:11 -0700387 len = adb_read(inFd, buf.data(), buf.size());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700388 }
Christopher Tated2f54152011-04-21 12:53:28 -0700389 if (len == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700390 D("copy_to_file() : read 0 bytes; exiting");
Christopher Tated2f54152011-04-21 12:53:28 -0700391 break;
392 }
393 if (len < 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700394 D("copy_to_file(): read failed: %s", strerror(errno));
Josh Gao3828ebc2019-12-03 16:05:54 -0800395 result = false;
Christopher Tated2f54152011-04-21 12:53:28 -0700396 break;
397 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700398 if (outFd == STDOUT_FILENO) {
Luis Hector Chavez6150a372018-07-18 21:18:27 -0700399 fwrite(buf.data(), 1, len, stdout);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700400 fflush(stdout);
401 } else {
Luis Hector Chavez6150a372018-07-18 21:18:27 -0700402 adb_write(outFd, buf.data(), len);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700403 }
Christopher Tatec9cd3b92011-06-01 17:56:23 -0700404 total += len;
Christopher Tated2f54152011-04-21 12:53:28 -0700405 }
Yabin Cuid325e862014-11-17 14:48:25 -0800406
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700407 stdinout_raw_epilogue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700408
Josh Gao3828ebc2019-12-03 16:05:54 -0800409 D("copy_to_file() finished with %s after %lu bytes", result ? "success" : "failure", total);
410 return result;
Christopher Tated2f54152011-04-21 12:53:28 -0700411}
412
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800413static void send_window_size_change(int fd, std::unique_ptr<ShellProtocol>& shell) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800414 // Old devices can't handle window size changes.
415 if (shell == nullptr) return;
416
Spencer Low2e02dc62015-11-07 17:34:39 -0800417#if defined(_WIN32)
418 struct winsize {
419 unsigned short ws_row;
420 unsigned short ws_col;
421 unsigned short ws_xpixel;
422 unsigned short ws_ypixel;
423 };
424#endif
425
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800426 winsize ws;
Spencer Low2e02dc62015-11-07 17:34:39 -0800427
428#if defined(_WIN32)
429 // If stdout is redirected to a non-console, we won't be able to get the
430 // console size, but that makes sense.
431 const intptr_t intptr_handle = _get_osfhandle(STDOUT_FILENO);
432 if (intptr_handle == -1) return;
433
434 const HANDLE handle = reinterpret_cast<const HANDLE>(intptr_handle);
435
436 CONSOLE_SCREEN_BUFFER_INFO info;
437 memset(&info, 0, sizeof(info));
438 if (!GetConsoleScreenBufferInfo(handle, &info)) return;
439
440 memset(&ws, 0, sizeof(ws));
441 // The number of visible rows, excluding offscreen scroll-back rows which are in info.dwSize.Y.
442 ws.ws_row = info.srWindow.Bottom - info.srWindow.Top + 1;
443 // If the user has disabled "Wrap text output on resize", they can make the screen buffer wider
444 // than the window, in which case we should use the width of the buffer.
445 ws.ws_col = info.dwSize.X;
446#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800447 if (ioctl(fd, TIOCGWINSZ, &ws) == -1) return;
Spencer Low2e02dc62015-11-07 17:34:39 -0800448#endif
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800449
450 // Send the new window size as human-readable ASCII for debugging convenience.
451 size_t l = snprintf(shell->data(), shell->data_capacity(), "%dx%d,%dx%d",
452 ws.ws_row, ws.ws_col, ws.ws_xpixel, ws.ws_ypixel);
453 shell->Write(ShellProtocol::kIdWindowSizeChange, l + 1);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800454}
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800455
David Pursell606835a2015-09-08 17:17:02 -0700456// Used to pass multiple values to the stdin read thread.
457struct StdinReadArgs {
458 int stdin_fd, write_fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700459 bool raw_stdin;
David Pursell606835a2015-09-08 17:17:02 -0700460 std::unique_ptr<ShellProtocol> protocol;
Elliott Hughes28416d62015-11-04 13:36:32 -0800461 char escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700462};
463
David Pursell606835a2015-09-08 17:17:02 -0700464// Loops to read from stdin and push the data to the given FD.
465// The argument should be a pointer to a StdinReadArgs object. This function
466// will take ownership of the object and delete it when finished.
Josh Gaod9db09c2016-02-12 14:31:15 -0800467static void stdin_read_thread_loop(void* x) {
David Pursell606835a2015-09-08 17:17:02 -0700468 std::unique_ptr<StdinReadArgs> args(reinterpret_cast<StdinReadArgs*>(x));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800469
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800470#if !defined(_WIN32)
471 // Mask SIGTTIN in case we're in a backgrounded process.
Josh Gao8dcdb572015-10-23 15:03:31 -0700472 sigset_t sigset;
473 sigemptyset(&sigset);
474 sigaddset(&sigset, SIGTTIN);
475 pthread_sigmask(SIG_BLOCK, &sigset, nullptr);
476#endif
477
Spencer Low2e02dc62015-11-07 17:34:39 -0800478#if defined(_WIN32)
479 // _get_interesting_input_record_uncached() causes unix_read_interruptible()
480 // to return -1 with errno == EINTR if the window size changes.
481#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800482 // Unblock SIGWINCH for this thread, so our read(2) below will be
483 // interrupted if the window size changes.
484 sigset_t mask;
485 sigemptyset(&mask);
486 sigaddset(&mask, SIGWINCH);
487 pthread_sigmask(SIG_UNBLOCK, &mask, nullptr);
488#endif
489
490 // Set up the initial window size.
491 send_window_size_change(args->stdin_fd, args->protocol);
492
Elliott Hughes28416d62015-11-04 13:36:32 -0800493 char raw_buffer[BUFSIZ];
David Pursell606835a2015-09-08 17:17:02 -0700494 char* buffer_ptr = raw_buffer;
495 size_t buffer_size = sizeof(raw_buffer);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800496 if (args->protocol != nullptr) {
David Pursell606835a2015-09-08 17:17:02 -0700497 buffer_ptr = args->protocol->data();
498 buffer_size = args->protocol->data_capacity();
499 }
500
Elliott Hughes28416d62015-11-04 13:36:32 -0800501 // If we need to parse escape sequences, make life easy.
502 if (args->raw_stdin && args->escape_char != '\0') {
503 buffer_size = 1;
504 }
505
506 enum EscapeState { kMidFlow, kStartOfLine, kInEscape };
507 EscapeState state = kStartOfLine;
508
Elliott Hughesaa245492015-08-03 10:38:08 -0700509 while (true) {
Spencer Low2e02dc62015-11-07 17:34:39 -0800510 // Use unix_read_interruptible() rather than adb_read() for stdin.
511 D("stdin_read_thread_loop(): pre unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
512 int r = unix_read_interruptible(args->stdin_fd, buffer_ptr,
513 buffer_size);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800514 if (r == -1 && errno == EINTR) {
515 send_window_size_change(args->stdin_fd, args->protocol);
516 continue;
517 }
Spencer Low2e02dc62015-11-07 17:34:39 -0800518 D("stdin_read_thread_loop(): post unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
David Pursell1ed57f02015-10-06 15:30:03 -0700519 if (r <= 0) {
520 // Only devices using the shell protocol know to close subprocess
521 // stdin. For older devices we want to just leave the connection
522 // open, otherwise an unpredictable amount of return data could
523 // be lost due to the FD closing before all data has been received.
524 if (args->protocol) {
525 args->protocol->Write(ShellProtocol::kIdCloseStdin, 0);
526 }
527 break;
528 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800529 // If we made stdin raw, check input for escape sequences. In
David Pursell1ed57f02015-10-06 15:30:03 -0700530 // this situation signals like Ctrl+C are sent remotely rather than
531 // interpreted locally so this provides an emergency out if the remote
532 // process starts ignoring the signal. SSH also does this, see the
533 // "escape characters" section on the ssh man page for more info.
Elliott Hughes28416d62015-11-04 13:36:32 -0800534 if (args->raw_stdin && args->escape_char != '\0') {
535 char ch = buffer_ptr[0];
536 if (ch == args->escape_char) {
537 if (state == kStartOfLine) {
538 state = kInEscape;
539 // Swallow the escape character.
540 continue;
541 } else {
542 state = kMidFlow;
543 }
544 } else {
545 if (state == kInEscape) {
546 if (ch == '.') {
547 fprintf(stderr,"\r\n[ disconnected ]\r\n");
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800548 stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700549 exit(0);
Elliott Hughes28416d62015-11-04 13:36:32 -0800550 } else {
551 // We swallowed an escape character that wasn't part of
552 // a valid escape sequence; time to cough it up.
553 buffer_ptr[0] = args->escape_char;
554 buffer_ptr[1] = ch;
555 ++r;
David Pursell1ed57f02015-10-06 15:30:03 -0700556 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800557 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800558 state = (ch == '\n' || ch == '\r') ? kStartOfLine : kMidFlow;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800559 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800560 }
David Pursell606835a2015-09-08 17:17:02 -0700561 if (args->protocol) {
562 if (!args->protocol->Write(ShellProtocol::kIdStdin, r)) {
563 break;
564 }
565 } else {
566 if (!WriteFdExactly(args->write_fd, buffer_ptr, r)) {
567 break;
568 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800569 }
570 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800571}
572
David Pursell1ed57f02015-10-06 15:30:03 -0700573// Returns a shell service string with the indicated arguments and command.
574static std::string ShellServiceString(bool use_shell_protocol,
575 const std::string& type_arg,
576 const std::string& command) {
577 std::vector<std::string> args;
578 if (use_shell_protocol) {
579 args.push_back(kShellServiceArgShellProtocol);
Elliott Hughesc2252df2015-11-18 12:45:48 -0800580
581 const char* terminal_type = getenv("TERM");
582 if (terminal_type != nullptr) {
583 args.push_back(std::string("TERM=") + terminal_type);
584 }
David Pursell1ed57f02015-10-06 15:30:03 -0700585 }
586 if (!type_arg.empty()) {
587 args.push_back(type_arg);
588 }
589
590 // Shell service string can look like: shell[,arg1,arg2,...]:[command].
591 return android::base::StringPrintf("shell%s%s:%s",
592 args.empty() ? "" : ",",
593 android::base::Join(args, ',').c_str(),
594 command.c_str());
595}
596
597// Connects to a shell on the device and read/writes data.
598//
599// Note: currently this function doesn't properly clean up resources; the
600// FD connected to the adb server is never closed and the stdin read thread
601// may never exit.
602//
603// On success returns the remote exit code if |use_shell_protocol| is true,
604// 0 otherwise. On failure returns 1.
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800605static int RemoteShell(bool use_shell_protocol, const std::string& type_arg, char escape_char,
606 bool empty_command, const std::string& service_string) {
Josh Gaoa019f782017-06-16 15:34:34 -0700607 // Old devices can't handle a service string that's longer than MAX_PAYLOAD_V1.
608 // Use |use_shell_protocol| to determine whether to allow a command longer than that.
609 if (service_string.size() > MAX_PAYLOAD_V1 && !use_shell_protocol) {
610 fprintf(stderr, "error: shell command too long\n");
611 return 1;
612 }
613
David Pursell1ed57f02015-10-06 15:30:03 -0700614 // Make local stdin raw if the device allocates a PTY, which happens if:
615 // 1. We are explicitly asking for a PTY shell, or
616 // 2. We don't specify shell type and are starting an interactive session.
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800617 bool raw_stdin = (type_arg == kShellServiceArgPty || (type_arg.empty() && empty_command));
David Pursell1ed57f02015-10-06 15:30:03 -0700618
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700619 std::string error;
David Pursell4e2fd362015-09-22 10:43:08 -0700620 int fd = adb_connect(service_string, &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700621 if (fd < 0) {
622 fprintf(stderr,"error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800623 return 1;
624 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800625
David Pursell606835a2015-09-08 17:17:02 -0700626 StdinReadArgs* args = new StdinReadArgs;
627 if (!args) {
628 LOG(ERROR) << "couldn't allocate StdinReadArgs object";
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700629 return 1;
630 }
David Pursell1ed57f02015-10-06 15:30:03 -0700631 args->stdin_fd = STDIN_FILENO;
David Pursell606835a2015-09-08 17:17:02 -0700632 args->write_fd = fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700633 args->raw_stdin = raw_stdin;
Elliott Hughes28416d62015-11-04 13:36:32 -0800634 args->escape_char = escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700635 if (use_shell_protocol) {
Josh Gao31b5be62018-03-07 16:51:08 -0800636 args->protocol = std::make_unique<ShellProtocol>(args->write_fd);
David Pursell606835a2015-09-08 17:17:02 -0700637 }
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700638
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800639 if (raw_stdin) stdin_raw_init();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800640
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800641#if !defined(_WIN32)
642 // Ensure our process is notified if the local window size changes.
643 // We use sigaction(2) to ensure that the SA_RESTART flag is not set,
644 // because the whole reason we're sending signals is to unblock the read(2)!
645 // That also means we don't need to do anything in the signal handler:
646 // the side effect of delivering the signal is all we need.
647 struct sigaction sa;
648 memset(&sa, 0, sizeof(sa));
649 sa.sa_handler = [](int) {};
650 sa.sa_flags = 0;
651 sigaction(SIGWINCH, &sa, nullptr);
652
653 // Now block SIGWINCH in this thread (the main thread) and all threads spawned
654 // from it. The stdin read thread will unblock this signal to ensure that it's
655 // the thread that receives the signal.
656 sigset_t mask;
657 sigemptyset(&mask);
658 sigaddset(&mask, SIGWINCH);
659 pthread_sigmask(SIG_BLOCK, &mask, nullptr);
660#endif
661
662 // TODO: combine read_and_dump with stdin_read_thread to make life simpler?
Josh Gaoe1dacfc2017-04-12 17:00:49 -0700663 std::thread(stdin_read_thread_loop, args).detach();
664 int exit_code = read_and_dump(fd, use_shell_protocol);
Elliott Hughes9b0f3542015-05-05 13:41:21 -0700665
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800666 // TODO: properly exit stdin_read_thread_loop and close |fd|.
David Pursell1ed57f02015-10-06 15:30:03 -0700667
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800668 // TODO: we should probably install signal handlers for this.
669 // TODO: can we use atexit? even on Windows?
670 if (raw_stdin) stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700671
David Pursell606835a2015-09-08 17:17:02 -0700672 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800673}
674
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800675static int adb_shell(int argc, const char** argv) {
Josh Gaofb4616b2020-04-16 19:34:43 -0700676 std::string error;
677 auto&& features = adb_get_feature_set(&error);
678 if (!features) {
679 error_exit("%s", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -0800680 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800681
Elliott Hughescabfa112016-10-19 14:47:11 -0700682 enum PtyAllocationMode { kPtyAuto, kPtyNo, kPtyYes, kPtyDefinitely };
683
684 // Defaults.
Josh Gaofb4616b2020-04-16 19:34:43 -0700685 char escape_char = '~'; // -e
686 bool use_shell_protocol = CanUseFeature(*features, kFeatureShell2); // -x
Elliott Hughescabfa112016-10-19 14:47:11 -0700687 PtyAllocationMode tty = use_shell_protocol ? kPtyAuto : kPtyDefinitely; // -t/-T
Elliott Hughes6452a892015-04-29 12:28:13 -0700688
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800689 // Parse shell-specific command-line options.
Elliott Hughescabfa112016-10-19 14:47:11 -0700690 argv[0] = "adb shell"; // So getopt(3) error messages start "adb shell".
Huihong Luoaf398852017-08-17 15:23:08 -0700691#ifdef _WIN32
692 // fixes "adb shell -l" crash on Windows, b/37284906
693 __argv = const_cast<char**>(argv);
694#endif
Elliott Hughescabfa112016-10-19 14:47:11 -0700695 optind = 1; // argv[0] is always "shell", so set `optind` appropriately.
696 int opt;
697 while ((opt = getopt(argc, const_cast<char**>(argv), "+e:ntTx")) != -1) {
698 switch (opt) {
699 case 'e':
700 if (!(strlen(optarg) == 1 || strcmp(optarg, "none") == 0)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -0700701 error_exit("-e requires a single-character argument or 'none'");
Elliott Hughescabfa112016-10-19 14:47:11 -0700702 }
703 escape_char = (strcmp(optarg, "none") == 0) ? 0 : optarg[0];
704 break;
705 case 'n':
706 close_stdin();
707 break;
708 case 'x':
709 // This option basically asks for historical behavior, so set options that
710 // correspond to the historical defaults. This is slightly weird in that -Tx
711 // is fine (because we'll undo the -T) but -xT isn't, but that does seem to
712 // be our least worst choice...
713 use_shell_protocol = false;
714 tty = kPtyDefinitely;
715 escape_char = '~';
716 break;
717 case 't':
718 // Like ssh, -t arguments are cumulative so that multiple -t's
719 // are needed to force a PTY.
720 tty = (tty >= kPtyYes) ? kPtyDefinitely : kPtyYes;
721 break;
722 case 'T':
723 tty = kPtyNo;
724 break;
725 default:
726 // getopt(3) already printed an error message for us.
Elliott Hughes28416d62015-11-04 13:36:32 -0800727 return 1;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800728 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700729 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800730
Elliott Hughescabfa112016-10-19 14:47:11 -0700731 bool is_interactive = (optind == argc);
David Pursell57dd5ae2016-01-27 16:07:52 -0800732
Elliott Hughescabfa112016-10-19 14:47:11 -0700733 std::string shell_type_arg = kShellServiceArgPty;
734 if (tty == kPtyNo) {
735 shell_type_arg = kShellServiceArgRaw;
736 } else if (tty == kPtyAuto) {
737 // If stdin isn't a TTY, default to a raw shell; this lets
738 // things like `adb shell < my_script.sh` work as expected.
739 // Non-interactive shells should also not have a pty.
740 if (!unix_isatty(STDIN_FILENO) || !is_interactive) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800741 shell_type_arg = kShellServiceArgRaw;
Elliott Hughescabfa112016-10-19 14:47:11 -0700742 }
743 } else if (tty == kPtyYes) {
744 // A single -t arg isn't enough to override implicit -T.
745 if (!unix_isatty(STDIN_FILENO)) {
746 fprintf(stderr,
747 "Remote PTY will not be allocated because stdin is not a terminal.\n"
748 "Use multiple -t options to force remote PTY allocation.\n");
749 shell_type_arg = kShellServiceArgRaw;
750 }
751 }
752
753 D("shell -e 0x%x t=%d use_shell_protocol=%s shell_type_arg=%s\n",
754 escape_char, tty,
755 use_shell_protocol ? "true" : "false",
756 (shell_type_arg == kShellServiceArgPty) ? "pty" : "raw");
757
758 // Raw mode is only supported when talking to a new device *and* using the shell protocol.
759 if (!use_shell_protocol) {
760 if (shell_type_arg != kShellServiceArgPty) {
761 fprintf(stderr, "error: %s only supports allocating a pty\n",
Josh Gaofb4616b2020-04-16 19:34:43 -0700762 !CanUseFeature(*features, kFeatureShell2) ? "device" : "-x");
Elliott Hughescabfa112016-10-19 14:47:11 -0700763 return 1;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800764 } else {
Elliott Hughescabfa112016-10-19 14:47:11 -0700765 // If we're not using the shell protocol, the type argument must be empty.
766 shell_type_arg = "";
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800767 }
David Pursell71c83122015-09-14 15:33:50 -0700768 }
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800769
770 std::string command;
Elliott Hughescabfa112016-10-19 14:47:11 -0700771 if (optind < argc) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800772 // We don't escape here, just like ssh(1). http://b/20564385.
Elliott Hughescabfa112016-10-19 14:47:11 -0700773 command = android::base::Join(std::vector<const char*>(argv + optind, argv + argc), ' ');
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800774 }
775
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800776 std::string service_string = ShellServiceString(use_shell_protocol, shell_type_arg, command);
777 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, command.empty(),
778 service_string);
779}
780
781static int adb_abb(int argc, const char** argv) {
Josh Gaofb4616b2020-04-16 19:34:43 -0700782 std::string error;
783 auto&& features = adb_get_feature_set(&error);
784 if (!features) {
785 error_exit("%s", error.c_str());
Alex Buynytskyy01a65ee2019-01-17 13:13:56 -0800786 return 1;
787 }
Josh Gaofb4616b2020-04-16 19:34:43 -0700788 if (!CanUseFeature(*features, kFeatureAbb)) {
Alex Buynytskyy01a65ee2019-01-17 13:13:56 -0800789 error_exit("abb is not supported by the device");
790 }
791
Alex Buynytskyycc2f1242019-06-27 10:33:54 -0700792 optind = 1; // argv[0] is always "abb", so set `optind` appropriately.
793
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800794 // Defaults.
795 constexpr char escape_char = '~'; // -e
796 constexpr bool use_shell_protocol = true;
797 constexpr auto shell_type_arg = kShellServiceArgRaw;
798 constexpr bool empty_command = false;
799
Alex Buynytskyycc2f1242019-06-27 10:33:54 -0700800 std::vector<const char*> args(argv + optind, argv + argc);
801 std::string service_string = "abb:" + android::base::Join(args, ABB_ARG_DELIMETER);
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800802
803 D("abb -e 0x%x [%*.s]\n", escape_char, static_cast<int>(service_string.size()),
804 service_string.data());
805
806 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, empty_command,
807 service_string);
David Pursell71c83122015-09-14 15:33:50 -0700808}
809
Josh Gao9e756342020-01-08 12:39:52 -0800810static int adb_shell_noinput(int argc, const char** argv) {
811#if !defined(_WIN32)
812 unique_fd fd(adb_open("/dev/null", O_RDONLY));
813 CHECK_NE(STDIN_FILENO, fd.get());
814 dup2(fd.get(), STDIN_FILENO);
815#endif
816 return adb_shell(argc, argv);
817}
818
Elliott Hughes09e794c2017-05-02 14:31:59 -0700819static int adb_sideload_legacy(const char* filename, int in_fd, int size) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700820 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -0800821 unique_fd out_fd(adb_connect(android::base::StringPrintf("sideload:%d", size), &error));
Elliott Hughes09e794c2017-05-02 14:31:59 -0700822 if (out_fd < 0) {
823 fprintf(stderr, "adb: pre-KitKat sideload connection failed: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800824 return -1;
825 }
826
827 int opt = CHUNK_SIZE;
Elliott Hughes09e794c2017-05-02 14:31:59 -0700828 opt = adb_setsockopt(out_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
Doug Zongker447f0612012-01-09 14:54:53 -0800829
Elliott Hughes09e794c2017-05-02 14:31:59 -0700830 char buf[CHUNK_SIZE];
831 int total = size;
832 while (size > 0) {
833 unsigned xfer = (size > CHUNK_SIZE) ? CHUNK_SIZE : size;
834 if (!ReadFdExactly(in_fd, buf, xfer)) {
835 fprintf(stderr, "adb: failed to read data from %s: %s\n", filename, strerror(errno));
Doug Zongker447f0612012-01-09 14:54:53 -0800836 return -1;
837 }
Elliott Hughes09e794c2017-05-02 14:31:59 -0700838 if (!WriteFdExactly(out_fd, buf, xfer)) {
839 std::string error;
840 adb_status(out_fd, &error);
841 fprintf(stderr, "adb: failed to write data: %s\n", error.c_str());
Elliott Hughes09e794c2017-05-02 14:31:59 -0700842 return -1;
843 }
844 size -= xfer;
845 printf("sending: '%s' %4d%% \r", filename, (int)(100LL - ((100LL * size) / (total))));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700846 fflush(stdout);
Doug Zongker447f0612012-01-09 14:54:53 -0800847 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700848 printf("\n");
Doug Zongker447f0612012-01-09 14:54:53 -0800849
Elliott Hughes09e794c2017-05-02 14:31:59 -0700850 if (!adb_status(out_fd, &error)) {
851 fprintf(stderr, "adb: error response: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800852 return -1;
853 }
854
Doug Zongker447f0612012-01-09 14:54:53 -0800855 return 0;
856}
857
Doug Zongker71fe5842014-06-26 15:35:36 -0700858#define SIDELOAD_HOST_BLOCK_SIZE (CHUNK_SIZE)
859
Tao Bao11cf30b2019-04-07 23:24:03 -0700860// Connects to the sideload / rescue service on the device (served by minadbd) and sends over the
861// data in an OTA package.
862//
863// It uses a simple protocol as follows.
864//
865// - The connect message includes the total number of bytes in the file and a block size chosen by
866// us.
867//
868// - The other side sends the desired block number as eight decimal digits (e.g. "00000023" for
869// block 23). Blocks are numbered from zero.
870//
871// - We send back the data of the requested block. The last block is likely to be partial; when the
872// last block is requested we only send the part of the block that exists, it's not padded up to
873// the block size.
874//
875// - When the other side sends "DONEDONE" or "FAILFAIL" instead of a block number, we have done all
876// the data transfer.
877//
878static int adb_sideload_install(const char* filename, bool rescue_mode) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -0700879 // TODO: use a LinePrinter instead...
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700880 struct stat sb;
881 if (stat(filename, &sb) == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700882 fprintf(stderr, "adb: failed to stat file %s: %s\n", filename, strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700883 return -1;
884 }
885 unique_fd package_fd(adb_open(filename, O_RDONLY));
886 if (package_fd == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700887 fprintf(stderr, "adb: failed to open file %s: %s\n", filename, strerror(errno));
Doug Zongker71fe5842014-06-26 15:35:36 -0700888 return -1;
889 }
890
Tao Bao11cf30b2019-04-07 23:24:03 -0700891 std::string service = android::base::StringPrintf(
892 "%s:%" PRId64 ":%d", rescue_mode ? "rescue-install" : "sideload-host",
893 static_cast<int64_t>(sb.st_size), SIDELOAD_HOST_BLOCK_SIZE);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700894 std::string error;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700895 unique_fd device_fd(adb_connect(service, &error));
896 if (device_fd < 0) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700897 fprintf(stderr, "adb: sideload connection failed: %s\n", error.c_str());
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700898
Tao Bao11cf30b2019-04-07 23:24:03 -0700899 if (rescue_mode) {
900 return -1;
901 }
902
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700903 // If this is a small enough package, maybe this is an older device that doesn't
904 // support sideload-host. Try falling back to the older (<= K) sideload method.
905 if (sb.st_size > INT_MAX) {
906 return -1;
907 }
Elliott Hughes09e794c2017-05-02 14:31:59 -0700908 fprintf(stderr, "adb: trying pre-KitKat sideload method...\n");
Josh Gao27241a72019-04-25 14:04:57 -0700909 return adb_sideload_legacy(filename, package_fd.get(), static_cast<int>(sb.st_size));
Doug Zongker71fe5842014-06-26 15:35:36 -0700910 }
911
Elliott Hughese8b663f2016-05-26 22:43:19 -0700912 int opt = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700913 adb_setsockopt(device_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
914
915 char buf[SIDELOAD_HOST_BLOCK_SIZE];
Doug Zongker71fe5842014-06-26 15:35:36 -0700916
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700917 int64_t xfer = 0;
Elliott Hughese8b663f2016-05-26 22:43:19 -0700918 int last_percent = -1;
Elliott Hughesa7090b92015-04-17 17:03:59 -0700919 while (true) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700920 if (!ReadFdExactly(device_fd, buf, 8)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700921 fprintf(stderr, "adb: failed to read command: %s\n", strerror(errno));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700922 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700923 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700924 buf[8] = '\0';
Doug Zongker71fe5842014-06-26 15:35:36 -0700925
xunchangc2265c52019-04-22 23:14:28 -0700926 if (strcmp(kMinadbdServicesExitSuccess, buf) == 0 ||
927 strcmp(kMinadbdServicesExitFailure, buf) == 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -0700928 printf("\rTotal xfer: %.2fx%*s\n",
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700929 static_cast<double>(xfer) / (sb.st_size ? sb.st_size : 1),
930 static_cast<int>(strlen(filename) + 10), "");
xunchangc2265c52019-04-22 23:14:28 -0700931 if (strcmp(kMinadbdServicesExitFailure, buf) == 0) {
Tao Bao11cf30b2019-04-07 23:24:03 -0700932 return 1;
933 }
Elliott Hughese8b663f2016-05-26 22:43:19 -0700934 return 0;
Doug Zongker71fe5842014-06-26 15:35:36 -0700935 }
936
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700937 int64_t block = strtoll(buf, nullptr, 10);
938 int64_t offset = block * SIDELOAD_HOST_BLOCK_SIZE;
939 if (offset >= static_cast<int64_t>(sb.st_size)) {
940 fprintf(stderr,
941 "adb: failed to read block %" PRId64 " at offset %" PRId64 ", past end %" PRId64
942 "\n",
943 block, offset, static_cast<int64_t>(sb.st_size));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700944 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700945 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700946
Doug Zongker71fe5842014-06-26 15:35:36 -0700947 size_t to_write = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700948 if ((offset + SIDELOAD_HOST_BLOCK_SIZE) > static_cast<int64_t>(sb.st_size)) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700949 to_write = sb.st_size - offset;
Doug Zongker71fe5842014-06-26 15:35:36 -0700950 }
951
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700952 if (adb_lseek(package_fd, offset, SEEK_SET) != offset) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700953 fprintf(stderr, "adb: failed to seek to package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700954 return -1;
955 }
956 if (!ReadFdExactly(package_fd, buf, to_write)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700957 fprintf(stderr, "adb: failed to read package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700958 return -1;
959 }
960
961 if (!WriteFdExactly(device_fd, buf, to_write)) {
962 adb_status(device_fd, &error);
Elliott Hughes09e794c2017-05-02 14:31:59 -0700963 fprintf(stderr, "adb: failed to write data '%s' *\n", error.c_str());
Elliott Hughese8b663f2016-05-26 22:43:19 -0700964 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700965 }
966 xfer += to_write;
967
968 // For normal OTA packages, we expect to transfer every byte
969 // twice, plus a bit of overhead (one read during
970 // verification, one read of each byte for installation, plus
971 // extra access to things like the zip central directory).
972 // This estimate of the completion becomes 100% when we've
973 // transferred ~2.13 (=100/47) times the package size.
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700974 int percent = static_cast<int>(xfer * 47LL / (sb.st_size ? sb.st_size : 1));
Doug Zongker71fe5842014-06-26 15:35:36 -0700975 if (percent != last_percent) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700976 printf("\rserving: '%s' (~%d%%) ", filename, percent);
Doug Zongker71fe5842014-06-26 15:35:36 -0700977 fflush(stdout);
978 last_percent = percent;
979 }
980 }
Doug Zongker71fe5842014-06-26 15:35:36 -0700981}
982
xunchangc2265c52019-04-22 23:14:28 -0700983static int adb_wipe_devices() {
984 auto wipe_devices_message_size = strlen(kMinadbdServicesExitSuccess);
985 std::string error;
986 unique_fd fd(adb_connect(
987 android::base::StringPrintf("rescue-wipe:userdata:%zu", wipe_devices_message_size),
988 &error));
989 if (fd < 0) {
990 fprintf(stderr, "adb: wipe device connection failed: %s\n", error.c_str());
991 return 1;
992 }
993
994 std::string message(wipe_devices_message_size, '\0');
995 if (!ReadFdExactly(fd, message.data(), wipe_devices_message_size)) {
996 fprintf(stderr, "adb: failed to read wipe result: %s\n", strerror(errno));
997 return 1;
998 }
999
1000 if (message == kMinadbdServicesExitSuccess) {
1001 return 0;
1002 }
1003
1004 if (message != kMinadbdServicesExitFailure) {
1005 fprintf(stderr, "adb: got unexpected message from rescue wipe %s\n", message.c_str());
1006 }
1007 return 1;
1008}
1009
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001010/**
1011 * Run ppp in "notty" mode against a resource listed as the first parameter
1012 * eg:
1013 *
1014 * ppp dev:/dev/omap_csmi_tty0 <ppp options>
1015 *
1016 */
Elliott Hughes58305772015-04-17 13:57:15 -07001017static int ppp(int argc, const char** argv) {
Yabin Cuie77b6a02014-11-11 09:24:11 -08001018#if defined(_WIN32)
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001019 error_exit("adb %s not implemented on Win32", argv[0]);
Elliott Hughes4679a392018-10-19 13:59:44 -07001020 __builtin_unreachable();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001021#else
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001022 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 -08001023
Dan Albertbac34742015-02-25 17:51:28 -08001024 const char* adb_service_name = argv[1];
Elliott Hughes4679a392018-10-19 13:59:44 -07001025 std::string error_message;
1026 int fd = adb_connect(adb_service_name, &error_message);
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001027 if (fd < 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001028 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 -08001029 }
1030
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001031 pid_t pid = fork();
Elliott Hughes4679a392018-10-19 13:59:44 -07001032 if (pid == -1) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001033 perror_exit("fork failed");
Elliott Hughes4679a392018-10-19 13:59:44 -07001034 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001035
Elliott Hughes4679a392018-10-19 13:59:44 -07001036 if (pid == 0) {
1037 // child side
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001038 int i;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001039
1040 // copy args
Elliott Hughes4679a392018-10-19 13:59:44 -07001041 const char** ppp_args = (const char**)alloca(sizeof(char*) * argc + 1);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001042 ppp_args[0] = "pppd";
1043 for (i = 2 ; i < argc ; i++) {
1044 //argv[2] and beyond become ppp_args[1] and beyond
1045 ppp_args[i - 1] = argv[i];
1046 }
Yi Kongaed415c2018-07-13 18:15:16 -07001047 ppp_args[i-1] = nullptr;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001048
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001049 dup2(fd, STDIN_FILENO);
1050 dup2(fd, STDOUT_FILENO);
1051 adb_close(STDERR_FILENO);
1052 adb_close(fd);
1053
Elliott Hughes4679a392018-10-19 13:59:44 -07001054 execvp("pppd", (char* const*)ppp_args);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001055 perror_exit("exec pppd failed");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001056 }
Elliott Hughes4679a392018-10-19 13:59:44 -07001057
1058 // parent side
1059 adb_close(fd);
1060 return 0;
Yabin Cuie77b6a02014-11-11 09:24:11 -08001061#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001062}
1063
Josh Gao3c99b0f2019-03-11 15:39:44 -07001064static bool wait_for_device(const char* service,
1065 std::optional<std::chrono::milliseconds> timeout = std::nullopt) {
Josh Gao08881e72016-04-13 12:14:16 -07001066 std::vector<std::string> components = android::base::Split(service, "-");
Elliott Hughesa52b4582020-03-10 16:57:47 -07001067 if (components.size() < 3) {
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001068 fprintf(stderr, "adb: couldn't parse 'wait-for' command: %s\n", service);
1069 return false;
1070 }
1071
Elliott Hughesa52b4582020-03-10 16:57:47 -07001072 // If the first thing after "wait-for-" wasn't a TRANSPORT, insert whatever
1073 // the current transport implies.
1074 if (components[2] != "usb" && components[2] != "local" && components[2] != "any") {
1075 TransportType t;
1076 adb_get_transport(&t, nullptr, nullptr);
Josh Gao08881e72016-04-13 12:14:16 -07001077 auto it = components.begin() + 2;
Elliott Hughes2b101112015-05-04 19:29:32 -07001078 if (t == kTransportUsb) {
Josh Gao08881e72016-04-13 12:14:16 -07001079 components.insert(it, "usb");
Elliott Hughes2b101112015-05-04 19:29:32 -07001080 } else if (t == kTransportLocal) {
Josh Gao08881e72016-04-13 12:14:16 -07001081 components.insert(it, "local");
Elliott Hughes2b101112015-05-04 19:29:32 -07001082 } else {
Josh Gao08881e72016-04-13 12:14:16 -07001083 components.insert(it, "any");
Elliott Hughes2b101112015-05-04 19:29:32 -07001084 }
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001085 }
1086
Elliott Hughesa52b4582020-03-10 16:57:47 -07001087 // Stitch it back together and send it over...
Josh Gaob122b172017-08-16 16:57:01 -07001088 std::string cmd = format_host_command(android::base::Join(components, "-").c_str());
Josh Gao3c99b0f2019-03-11 15:39:44 -07001089 if (timeout) {
1090 std::thread([timeout]() {
1091 std::this_thread::sleep_for(*timeout);
1092 fprintf(stderr, "timeout expired while waiting for device\n");
1093 _exit(1);
1094 }).detach();
1095 }
Elliott Hughes424af022015-05-29 17:55:19 -07001096 return adb_command(cmd);
Elliott Hughes2b101112015-05-04 19:29:32 -07001097}
1098
Josh Gao06c73ae2016-03-04 15:16:18 -08001099static bool adb_root(const char* command) {
1100 std::string error;
Josh Gao06c73ae2016-03-04 15:16:18 -08001101
Josh Gao2020dd62019-02-22 15:20:27 -08001102 TransportId transport_id;
1103 unique_fd fd(adb_connect(&transport_id, android::base::StringPrintf("%s:", command), &error));
Elliott Hughese8b663f2016-05-26 22:43:19 -07001104 if (fd < 0) {
Josh Gao06c73ae2016-03-04 15:16:18 -08001105 fprintf(stderr, "adb: unable to connect for %s: %s\n", command, error.c_str());
1106 return false;
1107 }
1108
1109 // Figure out whether we actually did anything.
1110 char buf[256];
1111 char* cur = buf;
1112 ssize_t bytes_left = sizeof(buf);
1113 while (bytes_left > 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -07001114 ssize_t bytes_read = adb_read(fd, cur, bytes_left);
Josh Gao06c73ae2016-03-04 15:16:18 -08001115 if (bytes_read == 0) {
1116 break;
1117 } else if (bytes_read < 0) {
1118 fprintf(stderr, "adb: error while reading for %s: %s\n", command, strerror(errno));
1119 return false;
1120 }
1121 cur += bytes_read;
1122 bytes_left -= bytes_read;
1123 }
1124
1125 if (bytes_left == 0) {
1126 fprintf(stderr, "adb: unexpected output length for %s\n", command);
1127 return false;
1128 }
1129
Junyong Sun1ffccf52020-01-15 16:34:58 +08001130 fwrite(buf, 1, sizeof(buf) - bytes_left, stdout);
Josh Gao06c73ae2016-03-04 15:16:18 -08001131 fflush(stdout);
Josh Gao06c73ae2016-03-04 15:16:18 -08001132 if (cur != buf && strstr(buf, "restarting") == nullptr) {
1133 return true;
1134 }
1135
Josh Gao2020dd62019-02-22 15:20:27 -08001136 // Wait for the device to go away.
Josh Gao3c99b0f2019-03-11 15:39:44 -07001137 TransportType previous_type;
1138 const char* previous_serial;
1139 TransportId previous_id;
1140 adb_get_transport(&previous_type, &previous_serial, &previous_id);
1141
Josh Gao2020dd62019-02-22 15:20:27 -08001142 adb_set_transport(kTransportAny, nullptr, transport_id);
1143 wait_for_device("wait-for-disconnect");
Josh Gao3c99b0f2019-03-11 15:39:44 -07001144
1145 // Wait for the device to come back.
1146 // If we were using a specific transport ID, there's nothing we can wait for.
1147 if (previous_id == 0) {
1148 adb_set_transport(previous_type, previous_serial, 0);
Josh Gaoef4a9992019-05-01 18:51:46 -07001149 wait_for_device("wait-for-device", 6000ms);
Josh Gao3c99b0f2019-03-11 15:39:44 -07001150 }
1151
Josh Gao05824732016-06-16 14:00:09 -07001152 return true;
Josh Gao06c73ae2016-03-04 15:16:18 -08001153}
1154
Josh Gaob122b172017-08-16 16:57:01 -07001155int send_shell_command(const std::string& command, bool disable_shell_protocol,
1156 StandardStreamsCallbackInterface* callback) {
Josh Gao74ccdf92019-01-23 15:36:42 -08001157 unique_fd fd;
David Pursell0aacbbe2016-01-21 19:56:50 -08001158 bool use_shell_protocol = false;
1159
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001160 while (true) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001161 bool attempt_connection = true;
1162
Felipe Leme0d4f0502016-07-26 12:14:39 -07001163 // Use shell protocol if it's supported and the caller doesn't explicitly
1164 // disable it.
David Pursell0aacbbe2016-01-21 19:56:50 -08001165 if (!disable_shell_protocol) {
Josh Gaofb4616b2020-04-16 19:34:43 -07001166 auto&& features = adb_get_feature_set(nullptr);
1167 if (features) {
1168 use_shell_protocol = CanUseFeature(*features, kFeatureShell2);
David Pursell0aacbbe2016-01-21 19:56:50 -08001169 } else {
1170 // Device was unreachable.
1171 attempt_connection = false;
1172 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001173 }
David Pursell0aacbbe2016-01-21 19:56:50 -08001174
1175 if (attempt_connection) {
1176 std::string error;
1177 std::string service_string = ShellServiceString(use_shell_protocol, "", command);
1178
Josh Gao74ccdf92019-01-23 15:36:42 -08001179 fd.reset(adb_connect(service_string, &error));
David Pursell0aacbbe2016-01-21 19:56:50 -08001180 if (fd >= 0) {
1181 break;
1182 }
1183 }
1184
Felipe Leme0d4f0502016-07-26 12:14:39 -07001185 fprintf(stderr, "- waiting for device -\n");
Josh Gaob122b172017-08-16 16:57:01 -07001186 if (!wait_for_device("wait-for-device")) {
Josh Gaob6117c42016-02-24 15:16:17 -08001187 return 1;
1188 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001189 }
1190
Josh Gao74ccdf92019-01-23 15:36:42 -08001191 return read_and_dump(fd.get(), use_shell_protocol, callback);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001192}
1193
Josh Gaob122b172017-08-16 16:57:01 -07001194static int logcat(int argc, const char** argv) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001195 char* log_tags = getenv("ANDROID_LOG_TAGS");
1196 std::string quoted = escape_arg(log_tags == nullptr ? "" : log_tags);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001197
David Pursell70ef7b42015-09-30 13:35:42 -07001198 std::string cmd = "export ANDROID_LOG_TAGS=\"" + quoted + "\"; exec logcat";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001199
Jeff Sharkeyfd546e82014-06-10 11:31:24 -07001200 if (!strcmp(argv[0], "longcat")) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001201 cmd += " -v long";
Christopher Tatedb0a8802011-11-30 13:00:33 -08001202 }
1203
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001204 --argc;
1205 ++argv;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001206 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001207 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001208 }
1209
Elliott Hughes31f0f1b2018-06-14 10:46:05 -07001210 return send_shell_command(cmd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001211}
1212
Josh Gao27241a72019-04-25 14:04:57 -07001213static void write_zeros(int bytes, borrowed_fd fd) {
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001214 int old_stdin_mode = -1;
1215 int old_stdout_mode = -1;
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001216 std::vector<char> buf(bytes);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001217
Josh Gao27241a72019-04-25 14:04:57 -07001218 D("write_zeros(%d) -> %d", bytes, fd.get());
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001219
Josh Gao27241a72019-04-25 14:04:57 -07001220 stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001221
1222 if (fd == STDOUT_FILENO) {
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001223 fwrite(buf.data(), 1, bytes, stdout);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001224 fflush(stdout);
1225 } else {
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001226 adb_write(fd, buf.data(), bytes);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001227 }
1228
Josh Gao27241a72019-04-25 14:04:57 -07001229 stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001230
1231 D("write_zeros() finished");
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001232}
1233
Dan Albertbac34742015-02-25 17:51:28 -08001234static int backup(int argc, const char** argv) {
Al Sutton31d335f2019-05-09 16:14:05 +00001235 fprintf(stdout, "WARNING: adb backup is deprecated and may be removed in a future release\n");
1236
Elliott Hughes24f165f2015-08-21 20:31:31 -07001237 const char* filename = "backup.ab";
Christopher Tated2f54152011-04-21 12:53:28 -07001238
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001239 /* find, extract, and use any -f argument */
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001240 for (int i = 1; i < argc; i++) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001241 if (!strcmp("-f", argv[i])) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001242 if (i == argc - 1) error_exit("backup -f passed with no filename");
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001243 filename = argv[i+1];
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001244 for (int j = i+2; j <= argc; ) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001245 argv[i++] = argv[j++];
1246 }
1247 argc -= 2;
Yi Kongaed415c2018-07-13 18:15:16 -07001248 argv[argc] = nullptr;
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001249 }
Christopher Tated2f54152011-04-21 12:53:28 -07001250 }
1251
Elliott Hughes56e68132015-11-13 11:04:10 -08001252 // Bare "adb backup" or "adb backup -f filename" are not valid invocations ---
1253 // a list of packages is required.
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001254 if (argc < 2) error_exit("backup either needs a list of packages or -all/-shared");
Christopher Tatebb86bc52011-08-22 17:12:08 -07001255
Christopher Tateb1dfffe2011-12-08 19:04:34 -08001256 adb_unlink(filename);
Josh Gao74ccdf92019-01-23 15:36:42 -08001257 unique_fd outFd(adb_creat(filename, 0640));
Christopher Tated2f54152011-04-21 12:53:28 -07001258 if (outFd < 0) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001259 fprintf(stderr, "adb: backup unable to create file '%s': %s\n", filename, strerror(errno));
1260 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001261 }
1262
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001263 std::string cmd = "backup:";
1264 --argc;
1265 ++argv;
1266 while (argc-- > 0) {
1267 cmd += " " + escape_arg(*argv++);
Christopher Tated2f54152011-04-21 12:53:28 -07001268 }
1269
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001270 D("backup. filename=%s cmd=%s", filename, cmd.c_str());
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001271 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001272 unique_fd fd(adb_connect(cmd, &error));
Christopher Tated2f54152011-04-21 12:53:28 -07001273 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001274 fprintf(stderr, "adb: unable to connect for backup: %s\n", error.c_str());
Elliott Hughes56e68132015-11-13 11:04:10 -08001275 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001276 }
1277
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001278 fprintf(stdout, "Now unlock your device and confirm the backup operation...\n");
Elliott Hughes56e68132015-11-13 11:04:10 -08001279 fflush(stdout);
1280
Josh Gao74ccdf92019-01-23 15:36:42 -08001281 copy_to_file(fd.get(), outFd.get());
Elliott Hughes56e68132015-11-13 11:04:10 -08001282 return EXIT_SUCCESS;
Christopher Tated2f54152011-04-21 12:53:28 -07001283}
1284
Dan Albertbac34742015-02-25 17:51:28 -08001285static int restore(int argc, const char** argv) {
Al Sutton31d335f2019-05-09 16:14:05 +00001286 fprintf(stdout, "WARNING: adb restore is deprecated and may be removed in a future release\n");
1287
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001288 if (argc != 2) error_exit("restore requires an argument");
Christopher Tate702967a2011-05-17 15:52:54 -07001289
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001290 const char* filename = argv[1];
Josh Gao74ccdf92019-01-23 15:36:42 -08001291 unique_fd tarFd(adb_open(filename, O_RDONLY));
Christopher Tate702967a2011-05-17 15:52:54 -07001292 if (tarFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001293 fprintf(stderr, "adb: unable to open file %s: %s\n", filename, strerror(errno));
Christopher Tate702967a2011-05-17 15:52:54 -07001294 return -1;
1295 }
1296
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001297 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001298 unique_fd fd(adb_connect("restore:", &error));
Christopher Tate702967a2011-05-17 15:52:54 -07001299 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001300 fprintf(stderr, "adb: unable to connect for restore: %s\n", error.c_str());
Christopher Tate702967a2011-05-17 15:52:54 -07001301 return -1;
1302 }
1303
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001304 fprintf(stdout, "Now unlock your device and confirm the restore operation.\n");
1305 fflush(stdout);
1306
Josh Gao74ccdf92019-01-23 15:36:42 -08001307 copy_to_file(tarFd.get(), fd.get());
Christopher Tate702967a2011-05-17 15:52:54 -07001308
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001309 // Provide an in-band EOD marker in case the archive file is malformed
Josh Gao74ccdf92019-01-23 15:36:42 -08001310 write_zeros(512 * 2, fd);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001311
Josh Gao5767d0d2016-03-04 15:51:03 -08001312 // Wait until the other side finishes, or it'll get sent SIGHUP.
Josh Gao74ccdf92019-01-23 15:36:42 -08001313 copy_to_file(fd.get(), STDOUT_FILENO);
Christopher Tate702967a2011-05-17 15:52:54 -07001314 return 0;
1315}
1316
Josh Gao49ba5582020-03-26 19:33:25 -07001317static CompressionType parse_compression_type(const std::string& str, bool allow_numbers) {
1318 if (allow_numbers) {
1319 if (str == "0") {
1320 return CompressionType::None;
1321 } else if (str == "1") {
1322 return CompressionType::Any;
1323 }
1324 }
1325
1326 if (str == "any") {
1327 return CompressionType::Any;
1328 } else if (str == "none") {
1329 return CompressionType::None;
1330 }
1331
1332 if (str == "brotli") {
1333 return CompressionType::Brotli;
Josh Gaoec44d352020-03-26 22:02:03 -07001334 } else if (str == "lz4") {
1335 return CompressionType::LZ4;
Josh Gao49ba5582020-03-26 19:33:25 -07001336 }
1337
1338 error_exit("unexpected compression type %s", str.c_str());
1339}
1340
Dan Albert06b0d6b2017-05-18 22:56:48 -07001341static void parse_push_pull_args(const char** arg, int narg, std::vector<const char*>* srcs,
Josh Gao49ba5582020-03-26 19:33:25 -07001342 const char** dst, bool* copy_attrs, bool* sync,
Josh Gao5949fcc2020-03-30 23:25:16 -07001343 CompressionType* compression, bool* dry_run) {
Josh Gao05786772015-10-30 16:57:19 -07001344 *copy_attrs = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001345 if (const char* adb_compression = getenv("ADB_COMPRESSION")) {
1346 *compression = parse_compression_type(adb_compression, true);
Josh Gao939fc192020-03-04 19:34:08 -08001347 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001348
Josh Gao05786772015-10-30 16:57:19 -07001349 srcs->clear();
1350 bool ignore_flags = false;
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001351 while (narg > 0) {
Josh Gao05786772015-10-30 16:57:19 -07001352 if (ignore_flags || *arg[0] != '-') {
1353 srcs->push_back(*arg);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001354 } else {
Josh Gao05786772015-10-30 16:57:19 -07001355 if (!strcmp(*arg, "-p")) {
1356 // Silently ignore for backwards compatibility.
1357 } else if (!strcmp(*arg, "-a")) {
1358 *copy_attrs = true;
Josh Gao939fc192020-03-04 19:34:08 -08001359 } else if (!strcmp(*arg, "-z")) {
Josh Gao49ba5582020-03-26 19:33:25 -07001360 if (narg < 2) {
1361 error_exit("-z requires an argument");
Josh Gao939fc192020-03-04 19:34:08 -08001362 }
Josh Gao49ba5582020-03-26 19:33:25 -07001363 *compression = parse_compression_type(*++arg, false);
1364 --narg;
Josh Gao939fc192020-03-04 19:34:08 -08001365 } else if (!strcmp(*arg, "-Z")) {
Josh Gao49ba5582020-03-26 19:33:25 -07001366 *compression = CompressionType::None;
Josh Gao5949fcc2020-03-30 23:25:16 -07001367 } else if (dry_run && !strcmp(*arg, "-n")) {
1368 *dry_run = true;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001369 } else if (!strcmp(*arg, "--sync")) {
1370 if (sync != nullptr) {
1371 *sync = true;
1372 }
Josh Gao05786772015-10-30 16:57:19 -07001373 } else if (!strcmp(*arg, "--")) {
1374 ignore_flags = true;
1375 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001376 error_exit("unrecognized option '%s'", *arg);
Josh Gao05786772015-10-30 16:57:19 -07001377 }
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001378 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001379 ++arg;
1380 --narg;
1381 }
1382
Josh Gao05786772015-10-30 16:57:19 -07001383 if (srcs->size() > 1) {
1384 *dst = srcs->back();
1385 srcs->pop_back();
Mark Lindner76f2a932014-03-11 17:55:59 -07001386 }
1387}
1388
Shukang Zhouf4ffae12020-02-13 17:01:39 -08001389static int adb_connect_command(const std::string& command, TransportId* transport,
1390 StandardStreamsCallbackInterface* callback) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001391 std::string error;
Josh Gaod10b7c42019-03-14 15:38:02 -07001392 unique_fd fd(adb_connect(transport, command, &error));
Elliott Hughes5677c232015-05-07 23:37:40 -07001393 if (fd < 0) {
1394 fprintf(stderr, "error: %s\n", error.c_str());
1395 return 1;
Tao Bao175b7bb2015-03-29 11:22:34 -07001396 }
Shukang Zhouf4ffae12020-02-13 17:01:39 -08001397 read_and_dump(fd, false, callback);
Elliott Hughes5677c232015-05-07 23:37:40 -07001398 return 0;
Tao Bao175b7bb2015-03-29 11:22:34 -07001399}
1400
Shukang Zhouf4ffae12020-02-13 17:01:39 -08001401static int adb_connect_command(const std::string& command, TransportId* transport = nullptr) {
1402 return adb_connect_command(command, transport, &DEFAULT_STANDARD_STREAMS_CALLBACK);
1403}
1404
1405// A class that prints out human readable form of the protobuf message for "track-app" service
1406// (received in binary format).
1407class TrackAppStreamsCallback : public DefaultStandardStreamsCallback {
1408 public:
1409 TrackAppStreamsCallback() : DefaultStandardStreamsCallback(nullptr, nullptr) {}
1410
1411 // Assume the buffer contains at least 4 bytes of valid data.
1412 void OnStdout(const char* buffer, int length) override {
1413 if (length < 4) return; // Unexpected length received. Do nothing.
1414
1415 adb::proto::AppProcesses binary_proto;
1416 // The first 4 bytes are the length of remaining content in hexadecimal format.
1417 binary_proto.ParseFromString(std::string(buffer + 4, length - 4));
1418 char summary[24]; // The following string includes digits and 16 fixed characters.
1419 int written = snprintf(summary, sizeof(summary), "Process count: %d\n",
1420 binary_proto.process_size());
1421 OnStream(nullptr, stdout, summary, written);
1422
1423 std::string string_proto;
1424 google::protobuf::TextFormat::PrintToString(binary_proto, &string_proto);
1425 OnStream(nullptr, stdout, string_proto.data(), string_proto.length());
1426 }
1427
1428 private:
1429 DISALLOW_COPY_AND_ASSIGN(TrackAppStreamsCallback);
1430};
1431
Josh Gaoe89a55d2018-11-16 14:47:59 -08001432static int adb_connect_command_bidirectional(const std::string& command) {
1433 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001434 unique_fd fd(adb_connect(command, &error));
Josh Gaoe89a55d2018-11-16 14:47:59 -08001435 if (fd < 0) {
1436 fprintf(stderr, "error: %s\n", error.c_str());
1437 return 1;
1438 }
1439
1440 static constexpr auto forward = [](int src, int sink, bool exit_on_end) {
1441 char buf[4096];
1442 while (true) {
1443 int rc = adb_read(src, buf, sizeof(buf));
1444 if (rc == 0) {
1445 if (exit_on_end) {
1446 exit(0);
1447 } else {
1448 adb_shutdown(sink, SHUT_WR);
1449 }
1450 return;
1451 } else if (rc < 0) {
1452 perror_exit("read failed");
1453 }
1454 if (!WriteFdExactly(sink, buf, rc)) {
1455 perror_exit("write failed");
1456 }
1457 }
1458 };
1459
Josh Gao74ccdf92019-01-23 15:36:42 -08001460 std::thread read(forward, fd.get(), STDOUT_FILENO, true);
1461 std::thread write(forward, STDIN_FILENO, fd.get(), false);
Josh Gaoe89a55d2018-11-16 14:47:59 -08001462 read.join();
1463 write.join();
Josh Gaoe89a55d2018-11-16 14:47:59 -08001464 return 0;
1465}
1466
Elliott Hughes6452a892015-04-29 12:28:13 -07001467static int adb_query_command(const std::string& command) {
1468 std::string result;
1469 std::string error;
1470 if (!adb_query(command, &result, &error)) {
1471 fprintf(stderr, "error: %s\n", error.c_str());
1472 return 1;
1473 }
1474 printf("%s\n", result.c_str());
1475 return 0;
1476}
1477
Spencer Lowa13df302015-09-07 16:20:13 -07001478// Disallow stdin, stdout, and stderr.
1479static bool _is_valid_ack_reply_fd(const int ack_reply_fd) {
1480#ifdef _WIN32
1481 const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd);
1482 return (GetStdHandle(STD_INPUT_HANDLE) != ack_reply_handle) &&
1483 (GetStdHandle(STD_OUTPUT_HANDLE) != ack_reply_handle) &&
1484 (GetStdHandle(STD_ERROR_HANDLE) != ack_reply_handle);
1485#else
1486 return ack_reply_fd > 2;
1487#endif
1488}
1489
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07001490static bool _is_valid_os_fd(int fd) {
Songchun Fan23bd5522020-03-09 11:33:44 -07001491 // Disallow invalid FDs and stdin/out/err as well.
1492 if (fd < 3) {
1493 return false;
1494 }
1495#ifdef _WIN32
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07001496 auto handle = (HANDLE)fd;
Songchun Fan23bd5522020-03-09 11:33:44 -07001497 DWORD info = 0;
1498 if (GetHandleInformation(handle, &info) == 0) {
1499 return false;
1500 }
1501#else
1502 int flags = fcntl(fd, F_GETFD);
1503 if (flags == -1) {
1504 return false;
1505 }
1506#endif
1507 return true;
1508}
1509
Elliott Hughescabfa112016-10-19 14:47:11 -07001510int adb_commandline(int argc, const char** argv) {
Josh Gaoe0323872018-08-10 14:23:43 -07001511 bool no_daemon = false;
1512 bool is_daemon = false;
1513 bool is_server = false;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001514 int r;
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001515 TransportType transport_type = kTransportAny;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001516 int ack_reply_fd = -1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001517
Elliott Hughes65433da2015-11-18 18:07:48 -08001518#if !defined(_WIN32)
1519 // We'd rather have EPIPE than SIGPIPE.
1520 signal(SIGPIPE, SIG_IGN);
1521#endif
1522
Josh Gao9c869b52016-08-25 16:00:22 -07001523 const char* server_host_str = nullptr;
1524 const char* server_port_str = nullptr;
1525 const char* server_socket_str = nullptr;
Stefan Hilzingera84a42e2010-04-19 12:21:12 +01001526
Elliott Hughes8d28e192015-10-07 14:55:10 -07001527 // We need to check for -d and -e before we look at $ANDROID_SERIAL.
1528 const char* serial = nullptr;
Josh Gaob122b172017-08-16 16:57:01 -07001529 TransportId transport_id = 0;
Elliott Hughes8d28e192015-10-07 14:55:10 -07001530
Riley Andrews98f58e82014-12-05 17:37:24 -08001531 while (argc > 0) {
Josh Gaod10b7c42019-03-14 15:38:02 -07001532 if (!strcmp(argv[0], "server")) {
Josh Gaoe0323872018-08-10 14:23:43 -07001533 is_server = true;
Josh Gaod10b7c42019-03-14 15:38:02 -07001534 } else if (!strcmp(argv[0], "nodaemon")) {
Josh Gaoe0323872018-08-10 14:23:43 -07001535 no_daemon = true;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001536 } else if (!strcmp(argv[0], "fork-server")) {
1537 /* this is a special flag used only when the ADB client launches the ADB Server */
Josh Gaoe0323872018-08-10 14:23:43 -07001538 is_daemon = true;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001539 } else if (!strcmp(argv[0], "--reply-fd")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001540 if (argc < 2) error_exit("--reply-fd requires an argument");
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001541 const char* reply_fd_str = argv[1];
1542 argc--;
1543 argv++;
1544 ack_reply_fd = strtol(reply_fd_str, nullptr, 10);
Spencer Lowa13df302015-09-07 16:20:13 -07001545 if (!_is_valid_ack_reply_fd(ack_reply_fd)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001546 fprintf(stderr, "adb: invalid reply fd \"%s\"\n", reply_fd_str);
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001547 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001548 }
Josh Gaob122b172017-08-16 16:57:01 -07001549 } else if (!strncmp(argv[0], "-s", 2)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001550 if (isdigit(argv[0][2])) {
1551 serial = argv[0] + 2;
1552 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001553 if (argc < 2 || argv[0][2] != '\0') error_exit("-s requires an argument");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001554 serial = argv[1];
1555 argc--;
1556 argv++;
1557 }
Josh Gaob122b172017-08-16 16:57:01 -07001558 } else if (!strncmp(argv[0], "-t", 2)) {
1559 const char* id;
1560 if (isdigit(argv[0][2])) {
1561 id = argv[0] + 2;
1562 } else {
1563 id = argv[1];
1564 argc--;
1565 argv++;
1566 }
1567 transport_id = strtoll(id, const_cast<char**>(&id), 10);
1568 if (*id != '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001569 error_exit("invalid transport id");
Josh Gaob122b172017-08-16 16:57:01 -07001570 }
Josh Gaod10b7c42019-03-14 15:38:02 -07001571 } else if (!strcmp(argv[0], "-d")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001572 transport_type = kTransportUsb;
Josh Gaod10b7c42019-03-14 15:38:02 -07001573 } else if (!strcmp(argv[0], "-e")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001574 transport_type = kTransportLocal;
Josh Gaod10b7c42019-03-14 15:38:02 -07001575 } else if (!strcmp(argv[0], "-a")) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001576 gListenAll = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001577 } else if (!strncmp(argv[0], "-H", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001578 if (argv[0][2] == '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001579 if (argc < 2) error_exit("-H requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001580 server_host_str = argv[1];
Matt Gumbeld7b33082012-11-14 10:16:17 -08001581 argc--;
1582 argv++;
1583 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001584 server_host_str = argv[0] + 2;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001585 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001586 } else if (!strncmp(argv[0], "-P", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001587 if (argv[0][2] == '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001588 if (argc < 2) error_exit("-P requires an argument");
Matt Gumbeld7b33082012-11-14 10:16:17 -08001589 server_port_str = argv[1];
1590 argc--;
1591 argv++;
1592 } else {
1593 server_port_str = argv[0] + 2;
1594 }
Josh Gao9c869b52016-08-25 16:00:22 -07001595 } else if (!strcmp(argv[0], "-L")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001596 if (argc < 2) error_exit("-L requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001597 server_socket_str = argv[1];
1598 argc--;
1599 argv++;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001600 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001601 /* out of recognized modifiers and flags */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001602 break;
1603 }
1604 argc--;
1605 argv++;
1606 }
1607
Josh Gao9c869b52016-08-25 16:00:22 -07001608 if ((server_host_str || server_port_str) && server_socket_str) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001609 error_exit("-L is incompatible with -H or -P");
Josh Gao9c869b52016-08-25 16:00:22 -07001610 }
1611
1612 // If -L, -H, or -P are specified, ignore environment variables.
1613 // Otherwise, prefer ADB_SERVER_SOCKET over ANDROID_ADB_SERVER_ADDRESS/PORT.
Tao Wu135f4ab2016-09-16 13:01:24 -07001614 if (!server_host_str && !server_port_str && !server_socket_str) {
1615 server_socket_str = getenv("ADB_SERVER_SOCKET");
Josh Gao9c869b52016-08-25 16:00:22 -07001616 }
1617
1618 if (!server_socket_str) {
1619 // tcp:1234 and tcp:localhost:1234 are different with -a, so don't default to localhost
1620 server_host_str = server_host_str ? server_host_str : getenv("ANDROID_ADB_SERVER_ADDRESS");
1621
Tao Wu135f4ab2016-09-16 13:01:24 -07001622 int server_port = DEFAULT_ADB_PORT;
Josh Gao9c869b52016-08-25 16:00:22 -07001623 server_port_str = server_port_str ? server_port_str : getenv("ANDROID_ADB_SERVER_PORT");
Tao Wu135f4ab2016-09-16 13:01:24 -07001624 if (server_port_str && strlen(server_port_str) > 0) {
1625 if (!android::base::ParseInt(server_port_str, &server_port, 1, 65535)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001626 error_exit(
1627 "$ANDROID_ADB_SERVER_PORT must be a positive number less than 65535: "
1628 "got \"%s\"",
1629 server_port_str);
Tao Wu135f4ab2016-09-16 13:01:24 -07001630 }
1631 }
Josh Gao9c869b52016-08-25 16:00:22 -07001632
1633 int rc;
1634 char* temp;
1635 if (server_host_str) {
Tao Wu135f4ab2016-09-16 13:01:24 -07001636 rc = asprintf(&temp, "tcp:%s:%d", server_host_str, server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001637 } else {
Tao Wu135f4ab2016-09-16 13:01:24 -07001638 rc = asprintf(&temp, "tcp:%d", server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001639 }
1640 if (rc < 0) {
Elliott Hughes4679a392018-10-19 13:59:44 -07001641 LOG(FATAL) << "failed to allocate server socket specification";
Josh Gao9c869b52016-08-25 16:00:22 -07001642 }
1643 server_socket_str = temp;
1644 }
1645
1646 adb_set_socket_spec(server_socket_str);
1647
Elliott Hughes8d28e192015-10-07 14:55:10 -07001648 // If none of -d, -e, or -s were specified, try $ANDROID_SERIAL.
1649 if (transport_type == kTransportAny && serial == nullptr) {
1650 serial = getenv("ANDROID_SERIAL");
1651 }
1652
Josh Gaob122b172017-08-16 16:57:01 -07001653 adb_set_transport(transport_type, serial, transport_id);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001654
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001655 if (is_server) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001656 if (no_daemon || is_daemon) {
Spencer Low5c398d22015-08-08 15:07:07 -07001657 if (is_daemon && (ack_reply_fd == -1)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001658 fprintf(stderr, "reply fd for adb server to client communication not specified.\n");
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001659 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001660 }
Josh Gao9c869b52016-08-25 16:00:22 -07001661 r = adb_server_main(is_daemon, server_socket_str, ack_reply_fd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001662 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001663 r = launch_server(server_socket_str);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001664 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001665 if (r) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001666 fprintf(stderr,"* could not start server *\n");
1667 }
1668 return r;
1669 }
1670
Riley Andrews98f58e82014-12-05 17:37:24 -08001671 if (argc == 0) {
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001672 help();
1673 return 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001674 }
1675
Riley Andrewsc8514c82014-12-05 17:32:46 -08001676 /* handle wait-for-* prefix */
1677 if (!strncmp(argv[0], "wait-for-", strlen("wait-for-"))) {
Dan Albertbac34742015-02-25 17:51:28 -08001678 const char* service = argv[0];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001679
Josh Gaob122b172017-08-16 16:57:01 -07001680 if (!wait_for_device(service)) {
Riley Andrewsc8514c82014-12-05 17:32:46 -08001681 return 1;
1682 }
1683
Elliott Hughes2b101112015-05-04 19:29:32 -07001684 // Allow a command to be run after wait-for-device,
1685 // e.g. 'adb wait-for-device shell'.
Riley Andrewsc8514c82014-12-05 17:32:46 -08001686 if (argc == 1) {
1687 return 0;
1688 }
1689
1690 /* Fall through */
1691 argc--;
1692 argv++;
1693 }
1694
1695 /* adb_connect() commands */
Riley Andrews98f58e82014-12-05 17:37:24 -08001696 if (!strcmp(argv[0], "devices")) {
Dan Albertbac34742015-02-25 17:51:28 -08001697 const char *listopt;
Elliott Hughes6452a892015-04-29 12:28:13 -07001698 if (argc < 2) {
Scott Andersone109d262012-04-20 11:21:14 -07001699 listopt = "";
Elliott Hughes6452a892015-04-29 12:28:13 -07001700 } else if (argc == 2 && !strcmp(argv[1], "-l")) {
Scott Andersone109d262012-04-20 11:21:14 -07001701 listopt = argv[1];
Elliott Hughes6452a892015-04-29 12:28:13 -07001702 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001703 error_exit("adb devices [-l]");
Scott Andersone109d262012-04-20 11:21:14 -07001704 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001705
1706 std::string query = android::base::StringPrintf("host:%s%s", argv[0], listopt);
Josh Gaod10b7c42019-03-14 15:38:02 -07001707 std::string error;
1708 if (!adb_check_server_version(&error)) {
1709 error_exit("failed to check server version: %s", error.c_str());
1710 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001711 printf("List of devices attached\n");
1712 return adb_query_command(query);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001713 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001714 else if (!strcmp(argv[0], "connect")) {
Elliott Hughes65474fd2020-01-08 10:02:43 -08001715 if (argc != 2) error_exit("usage: adb connect HOST[:PORT]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001716
1717 std::string query = android::base::StringPrintf("host:connect:%s", argv[1]);
1718 return adb_query_command(query);
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001719 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001720 else if (!strcmp(argv[0], "disconnect")) {
Elliott Hughes573e67c2019-07-30 13:51:03 -07001721 if (argc > 2) error_exit("usage: adb disconnect [HOST[:PORT]]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001722
1723 std::string query = android::base::StringPrintf("host:disconnect:%s",
1724 (argc == 2) ? argv[1] : "");
1725 return adb_query_command(query);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001726 } else if (!strcmp(argv[0], "abb")) {
1727 return adb_abb(argc, argv);
Joshua Duongd85f5c02019-11-20 14:18:43 -08001728 } else if (!strcmp(argv[0], "pair")) {
1729 if (argc != 2) error_exit("usage: adb pair <host>[:<port>]");
1730
1731 std::string password;
1732 printf("Enter pairing code: ");
1733 fflush(stdout);
1734 if (!std::getline(std::cin, password) || password.empty()) {
1735 error_exit("No pairing code provided");
1736 }
1737 std::string query =
1738 android::base::StringPrintf("host:pair:%s:%s", password.c_str(), argv[1]);
1739
1740 return adb_query_command(query);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001741 } else if (!strcmp(argv[0], "emu")) {
Spencer Low142ec752015-05-06 16:13:42 -07001742 return adb_send_emulator_command(argc, argv, serial);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001743 } else if (!strcmp(argv[0], "shell")) {
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001744 return adb_shell(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001745 } else if (!strcmp(argv[0], "exec-in") || !strcmp(argv[0], "exec-out")) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001746 int exec_in = !strcmp(argv[0], "exec-in");
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001747
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001748 if (argc < 2) error_exit("usage: adb %s command", argv[0]);
Zach Riggleccafc792016-04-14 17:18:04 -04001749
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001750 std::string cmd = "exec:";
1751 cmd += argv[1];
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001752 argc -= 2;
1753 argv += 2;
1754 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001755 cmd += " " + escape_arg(*argv++);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001756 }
1757
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001758 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001759 unique_fd fd(adb_connect(cmd, &error));
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001760 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001761 fprintf(stderr, "error: %s\n", error.c_str());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001762 return -1;
1763 }
1764
1765 if (exec_in) {
Josh Gao74ccdf92019-01-23 15:36:42 -08001766 copy_to_file(STDIN_FILENO, fd.get());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001767 } else {
Josh Gao74ccdf92019-01-23 15:36:42 -08001768 copy_to_file(fd.get(), STDOUT_FILENO);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001769 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001770 return 0;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001771 } else if (!strcmp(argv[0], "kill-server")) {
Josh Gaob3c14ec2017-05-08 18:37:17 -07001772 return adb_kill_server() ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001773 } else if (!strcmp(argv[0], "sideload")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001774 if (argc != 2) error_exit("sideload requires an argument");
Tao Bao11cf30b2019-04-07 23:24:03 -07001775 if (adb_sideload_install(argv[1], false /* rescue_mode */)) {
Doug Zongker447f0612012-01-09 14:54:53 -08001776 return 1;
1777 } else {
1778 return 0;
1779 }
Tao Bao11cf30b2019-04-07 23:24:03 -07001780 } else if (!strcmp(argv[0], "rescue")) {
Tao Bao599eee12019-06-04 14:37:06 -07001781 // adb rescue getprop
Tao Bao11cf30b2019-04-07 23:24:03 -07001782 // adb rescue getprop <prop>
1783 // adb rescue install <filename>
xunchangc2265c52019-04-22 23:14:28 -07001784 // adb rescue wipe userdata
Tao Bao599eee12019-06-04 14:37:06 -07001785 if (argc < 2) error_exit("rescue requires at least one argument");
Tao Bao11cf30b2019-04-07 23:24:03 -07001786 if (!strcmp(argv[1], "getprop")) {
Tao Bao599eee12019-06-04 14:37:06 -07001787 if (argc == 2) {
1788 return adb_connect_command("rescue-getprop:");
1789 }
1790 if (argc == 3) {
1791 return adb_connect_command(
1792 android::base::StringPrintf("rescue-getprop:%s", argv[2]));
1793 }
1794 error_exit("invalid rescue getprop arguments");
Tao Bao11cf30b2019-04-07 23:24:03 -07001795 } else if (!strcmp(argv[1], "install")) {
Tao Bao599eee12019-06-04 14:37:06 -07001796 if (argc != 3) error_exit("rescue install requires two arguments");
Tao Bao11cf30b2019-04-07 23:24:03 -07001797 if (adb_sideload_install(argv[2], true /* rescue_mode */) != 0) {
1798 return 1;
1799 }
Tao Bao599eee12019-06-04 14:37:06 -07001800 } else if (!strcmp(argv[1], "wipe")) {
1801 if (argc != 3 || strcmp(argv[2], "userdata") != 0) {
1802 error_exit("invalid rescue wipe arguments");
1803 }
xunchangc2265c52019-04-22 23:14:28 -07001804 return adb_wipe_devices();
Tao Bao11cf30b2019-04-07 23:24:03 -07001805 } else {
1806 error_exit("invalid rescue argument");
1807 }
1808 return 0;
Elliott Hughese1632982017-08-23 15:42:28 -07001809 } else if (!strcmp(argv[0], "tcpip")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001810 if (argc != 2) error_exit("tcpip requires an argument");
Elliott Hughese1632982017-08-23 15:42:28 -07001811 int port;
1812 if (!android::base::ParseInt(argv[1], &port, 1, 65535)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001813 error_exit("tcpip: invalid port: %s", argv[1]);
Elliott Hughese1632982017-08-23 15:42:28 -07001814 }
1815 return adb_connect_command(android::base::StringPrintf("tcpip:%d", port));
Josh Gao8c2198c2019-07-11 14:15:32 -07001816 } else if (!strcmp(argv[0], "remount")) {
Josh Gaofb4616b2020-04-16 19:34:43 -07001817 std::string error;
1818 auto&& features = adb_get_feature_set(&error);
1819 if (!features) {
1820 error_exit("%s", error.c_str());
Josh Gao8c2198c2019-07-11 14:15:32 -07001821 }
1822
Josh Gaofb4616b2020-04-16 19:34:43 -07001823 if (CanUseFeature(*features, kFeatureRemountShell)) {
Mark Salyzyn6f908ce2019-09-17 08:01:15 -07001824 std::vector<const char*> args = {"shell"};
1825 args.insert(args.cend(), argv, argv + argc);
Josh Gao9e756342020-01-08 12:39:52 -08001826 return adb_shell_noinput(args.size(), args.data());
Mark Salyzyn6f908ce2019-09-17 08:01:15 -07001827 } else if (argc > 1) {
1828 auto command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
1829 return adb_connect_command(command);
Josh Gao8c2198c2019-07-11 14:15:32 -07001830 } else {
1831 return adb_connect_command("remount:");
1832 }
Elliott Hughes19d80b82015-07-21 16:13:40 -07001833 }
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001834 // clang-format off
Josh Gao8c2198c2019-07-11 14:15:32 -07001835 else if (!strcmp(argv[0], "reboot") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001836 !strcmp(argv[0], "reboot-bootloader") ||
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001837 !strcmp(argv[0], "reboot-fastboot") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001838 !strcmp(argv[0], "usb") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001839 !strcmp(argv[0], "disable-verity") ||
1840 !strcmp(argv[0], "enable-verity")) {
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001841 // clang-format on
Elliott Hughes5677c232015-05-07 23:37:40 -07001842 std::string command;
Tao Bao175b7bb2015-03-29 11:22:34 -07001843 if (!strcmp(argv[0], "reboot-bootloader")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001844 command = "reboot:bootloader";
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001845 } else if (!strcmp(argv[0], "reboot-fastboot")) {
1846 command = "reboot:fastboot";
Tao Bao175b7bb2015-03-29 11:22:34 -07001847 } else if (argc > 1) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001848 command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
Tao Bao175b7bb2015-03-29 11:22:34 -07001849 } else {
Elliott Hughes5677c232015-05-07 23:37:40 -07001850 command = android::base::StringPrintf("%s:", argv[0]);
The Android Open Source Projecte037fd72009-03-13 13:04:37 -07001851 }
Tao Bao175b7bb2015-03-29 11:22:34 -07001852 return adb_connect_command(command);
Josh Gao06c73ae2016-03-04 15:16:18 -08001853 } else if (!strcmp(argv[0], "root") || !strcmp(argv[0], "unroot")) {
1854 return adb_root(argv[0]) ? 0 : 1;
1855 } else if (!strcmp(argv[0], "bugreport")) {
Felipe Leme698e0652016-07-19 17:07:22 -07001856 Bugreport bugreport;
Josh Gaob122b172017-08-16 16:57:01 -07001857 return bugreport.DoIt(argc, argv);
Felipe Leme44a42672016-04-01 17:43:27 -07001858 } else if (!strcmp(argv[0], "forward") || !strcmp(argv[0], "reverse")) {
Elliott Hughes424af022015-05-29 17:55:19 -07001859 bool reverse = !strcmp(argv[0], "reverse");
Elliott Hughes424af022015-05-29 17:55:19 -07001860 --argc;
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001861 if (argc < 1) error_exit("%s requires an argument", argv[0]);
Josh Gao258d4a92018-07-31 15:36:03 -07001862 ++argv;
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001863
1864 // Determine the <host-prefix> for this command.
Elliott Hughes424af022015-05-29 17:55:19 -07001865 std::string host_prefix;
David 'Digit' Turner25258692013-03-21 21:07:42 +01001866 if (reverse) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001867 host_prefix = "reverse:";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001868 } else {
Josh Gaoda9c7672019-08-06 16:43:34 -07001869 host_prefix = "host:";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001870 }
1871
Elliott Hughes4679a392018-10-19 13:59:44 -07001872 std::string cmd, error_message;
Elliott Hughes424af022015-05-29 17:55:19 -07001873 if (strcmp(argv[0], "--list") == 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001874 if (argc != 1) error_exit("--list doesn't take any arguments");
Josh Gaoda9c7672019-08-06 16:43:34 -07001875 return adb_query_command(host_prefix + "list-forward");
Elliott Hughes424af022015-05-29 17:55:19 -07001876 } else if (strcmp(argv[0], "--remove-all") == 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001877 if (argc != 1) error_exit("--remove-all doesn't take any arguments");
Josh Gaoda9c7672019-08-06 16:43:34 -07001878 cmd = "killforward-all";
Elliott Hughes424af022015-05-29 17:55:19 -07001879 } else if (strcmp(argv[0], "--remove") == 0) {
1880 // forward --remove <local>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001881 if (argc != 2) error_exit("--remove requires an argument");
Josh Gaoda9c7672019-08-06 16:43:34 -07001882 cmd = std::string("killforward:") + argv[1];
Elliott Hughes424af022015-05-29 17:55:19 -07001883 } else if (strcmp(argv[0], "--no-rebind") == 0) {
1884 // forward --no-rebind <local> <remote>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001885 if (argc != 3) error_exit("--no-rebind takes two arguments");
Elliott Hughes4679a392018-10-19 13:59:44 -07001886 if (forward_targets_are_valid(argv[1], argv[2], &error_message)) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001887 cmd = std::string("forward:norebind:") + argv[1] + ";" + argv[2];
David Purselleaae97e2016-04-07 11:25:48 -07001888 }
Elliott Hughes424af022015-05-29 17:55:19 -07001889 } else {
1890 // forward <local> <remote>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001891 if (argc != 2) error_exit("forward takes two arguments");
Elliott Hughes4679a392018-10-19 13:59:44 -07001892 if (forward_targets_are_valid(argv[0], argv[1], &error_message)) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001893 cmd = std::string("forward:") + argv[0] + ";" + argv[1];
David Purselleaae97e2016-04-07 11:25:48 -07001894 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001895 }
1896
Elliott Hughes4679a392018-10-19 13:59:44 -07001897 if (!error_message.empty()) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001898 error_exit("error: %s", error_message.c_str());
David Purselleaae97e2016-04-07 11:25:48 -07001899 }
1900
Josh Gaoda9c7672019-08-06 16:43:34 -07001901 unique_fd fd(adb_connect(nullptr, host_prefix + cmd, &error_message, true));
Josh Gao74ccdf92019-01-23 15:36:42 -08001902 if (fd < 0 || !adb_status(fd.get(), &error_message)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001903 error_exit("error: %s", error_message.c_str());
David Purselleaae97e2016-04-07 11:25:48 -07001904 }
1905
1906 // Server or device may optionally return a resolved TCP port number.
1907 std::string resolved_port;
Elliott Hughes4679a392018-10-19 13:59:44 -07001908 if (ReadProtocolString(fd, &resolved_port, &error_message) && !resolved_port.empty()) {
David Purselleaae97e2016-04-07 11:25:48 -07001909 printf("%s\n", resolved_port.c_str());
1910 }
1911
1912 ReadOrderlyShutdown(fd);
1913 return 0;
Joshua Duong504d3932020-03-31 10:58:50 -07001914 } else if (!strcmp(argv[0], "mdns")) {
1915 --argc;
1916 if (argc < 1) error_exit("mdns requires an argument");
1917 ++argv;
1918
1919 std::string error;
1920 if (!adb_check_server_version(&error)) {
1921 error_exit("failed to check server version: %s", error.c_str());
1922 }
1923
1924 std::string query = "host:mdns:";
1925 if (!strcmp(argv[0], "check")) {
1926 if (argc != 1) error_exit("mdns %s doesn't take any arguments", argv[0]);
1927 query += "check";
1928 } else {
1929 error_exit("unknown mdns command [%s]", argv[0]);
1930 }
1931
1932 return adb_query_command(query);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001933 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001934 /* do_sync_*() commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001935 else if (!strcmp(argv[0], "ls")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001936 if (argc != 2) error_exit("ls requires an argument");
Elliott Hughesaa245492015-08-03 10:38:08 -07001937 return do_sync_ls(argv[1]) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001938 } else if (!strcmp(argv[0], "push")) {
Josh Gao05786772015-10-30 16:57:19 -07001939 bool copy_attrs = false;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001940 bool sync = false;
Josh Gao5949fcc2020-03-30 23:25:16 -07001941 bool dry_run = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001942 CompressionType compression = CompressionType::Any;
Josh Gao05786772015-10-30 16:57:19 -07001943 std::vector<const char*> srcs;
1944 const char* dst = nullptr;
Mark Lindner76f2a932014-03-11 17:55:59 -07001945
Josh Gao5949fcc2020-03-30 23:25:16 -07001946 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, &sync, &compression,
1947 &dry_run);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001948 if (srcs.empty() || !dst) error_exit("push requires an argument");
Josh Gao5949fcc2020-03-30 23:25:16 -07001949 return do_sync_push(srcs, dst, sync, compression, dry_run) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001950 } else if (!strcmp(argv[0], "pull")) {
Josh Gao05786772015-10-30 16:57:19 -07001951 bool copy_attrs = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001952 CompressionType compression = CompressionType::Any;
Josh Gao05786772015-10-30 16:57:19 -07001953 std::vector<const char*> srcs;
1954 const char* dst = ".";
Mark Lindner76f2a932014-03-11 17:55:59 -07001955
Josh Gao5949fcc2020-03-30 23:25:16 -07001956 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, nullptr, &compression,
1957 nullptr);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001958 if (srcs.empty()) error_exit("pull requires an argument");
Josh Gao49ba5582020-03-26 19:33:25 -07001959 return do_sync_pull(srcs, dst, copy_attrs, compression) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001960 } else if (!strcmp(argv[0], "install")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001961 if (argc < 2) error_exit("install requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001962 return install_app(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001963 } else if (!strcmp(argv[0], "install-multiple")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001964 if (argc < 2) error_exit("install-multiple requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001965 return install_multiple_app(argc, argv);
Patrick Baumanndc58b0a2018-10-09 10:45:03 -07001966 } else if (!strcmp(argv[0], "install-multi-package")) {
Mohammad Samiul Islamb581ce92019-05-29 15:00:00 +01001967 if (argc < 2) error_exit("install-multi-package requires an argument");
Patrick Baumanndc58b0a2018-10-09 10:45:03 -07001968 return install_multi_package(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001969 } else if (!strcmp(argv[0], "uninstall")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001970 if (argc < 2) error_exit("uninstall requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001971 return uninstall_app(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001972 } else if (!strcmp(argv[0], "sync")) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001973 std::string src;
Elliott Hughes58305772015-04-17 13:57:15 -07001974 bool list_only = false;
Josh Gao5949fcc2020-03-30 23:25:16 -07001975 bool dry_run = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001976 CompressionType compression = CompressionType::Any;
Josh Gao939fc192020-03-04 19:34:08 -08001977
Josh Gao49ba5582020-03-26 19:33:25 -07001978 if (const char* adb_compression = getenv("ADB_COMPRESSION"); adb_compression) {
1979 compression = parse_compression_type(adb_compression, true);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001980 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001981
Josh Gao939fc192020-03-04 19:34:08 -08001982 int opt;
Josh Gao5949fcc2020-03-30 23:25:16 -07001983 while ((opt = getopt(argc, const_cast<char**>(argv), "lnz:Z")) != -1) {
Josh Gao939fc192020-03-04 19:34:08 -08001984 switch (opt) {
1985 case 'l':
1986 list_only = true;
1987 break;
Josh Gao5949fcc2020-03-30 23:25:16 -07001988 case 'n':
1989 dry_run = true;
1990 break;
Josh Gao939fc192020-03-04 19:34:08 -08001991 case 'z':
Josh Gao49ba5582020-03-26 19:33:25 -07001992 compression = parse_compression_type(optarg, false);
Josh Gao939fc192020-03-04 19:34:08 -08001993 break;
1994 case 'Z':
Josh Gao49ba5582020-03-26 19:33:25 -07001995 compression = CompressionType::None;
Josh Gao939fc192020-03-04 19:34:08 -08001996 break;
1997 default:
Josh Gao5949fcc2020-03-30 23:25:16 -07001998 error_exit("usage: adb sync [-l] [-n] [-z ALGORITHM] [-Z] [PARTITION]");
Josh Gao939fc192020-03-04 19:34:08 -08001999 }
2000 }
2001
2002 if (optind == argc) {
2003 src = "all";
2004 } else if (optind + 1 == argc) {
2005 src = argv[optind];
2006 } else {
Josh Gao5949fcc2020-03-30 23:25:16 -07002007 error_exit("usage: adb sync [-l] [-n] [-z ALGORITHM] [-Z] [PARTITION]");
Josh Gao939fc192020-03-04 19:34:08 -08002008 }
2009
Justin Yun5d0ac622019-06-26 09:18:03 +09002010 std::vector<std::string> partitions{"data", "odm", "oem", "product",
2011 "system", "system_ext", "vendor"};
Elliott Hughes57e97cb2018-04-03 10:54:39 -07002012 bool found = false;
2013 for (const auto& partition : partitions) {
2014 if (src == "all" || src == partition) {
2015 std::string src_dir{product_file(partition)};
2016 if (!directory_exists(src_dir)) continue;
2017 found = true;
Josh Gao5949fcc2020-03-30 23:25:16 -07002018 if (!do_sync_sync(src_dir, "/" + partition, list_only, compression, dry_run)) {
2019 return 1;
2020 }
Elliott Hughes57e97cb2018-04-03 10:54:39 -07002021 }
Elliott Hughes58305772015-04-17 13:57:15 -07002022 }
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002023 if (!found) error_exit("don't know how to sync %s partition", src.c_str());
Elliott Hughes4679a392018-10-19 13:59:44 -07002024 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002025 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002026 /* passthrough commands */
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002027 else if (!strcmp(argv[0], "get-state") || !strcmp(argv[0], "get-serialno") ||
2028 !strcmp(argv[0], "get-devpath")) {
Josh Gaob122b172017-08-16 16:57:01 -07002029 return adb_query_command(format_host_command(argv[0]));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002030 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002031 /* other commands */
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002032 else if (!strcmp(argv[0], "logcat") || !strcmp(argv[0], "lolcat") ||
2033 !strcmp(argv[0], "longcat")) {
Josh Gaob122b172017-08-16 16:57:01 -07002034 return logcat(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002035 } else if (!strcmp(argv[0], "ppp")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002036 return ppp(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002037 } else if (!strcmp(argv[0], "start-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002038 std::string error;
Spencer Lowf18fc082015-08-11 17:05:02 -07002039 const int result = adb_connect("host:start-server", &error);
2040 if (result < 0) {
2041 fprintf(stderr, "error: %s\n", error.c_str());
2042 }
2043 return result;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002044 } else if (!strcmp(argv[0], "backup")) {
Christopher Tated2f54152011-04-21 12:53:28 -07002045 return backup(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002046 } else if (!strcmp(argv[0], "restore")) {
Christopher Tate702967a2011-05-17 15:52:54 -07002047 return restore(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002048 } else if (!strcmp(argv[0], "keygen")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002049 if (argc != 2) error_exit("keygen requires an argument");
Yabin Cuiaed3c612015-09-22 15:52:57 -07002050 // Always print key generation information for keygen command.
2051 adb_trace_enable(AUTH);
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08002052 return adb_auth_keygen(argv[1]);
Josh Gao2dc4cab2018-11-15 17:45:46 -08002053 } else if (!strcmp(argv[0], "pubkey")) {
2054 if (argc != 2) error_exit("pubkey requires an argument");
2055 return adb_auth_pubkey(argv[1]);
2056 } else if (!strcmp(argv[0], "jdwp")) {
Tao Bao175b7bb2015-03-29 11:22:34 -07002057 return adb_connect_command("jdwp");
Josh Gao2dc4cab2018-11-15 17:45:46 -08002058 } else if (!strcmp(argv[0], "track-jdwp")) {
Josh Gaob463baf2016-05-13 18:18:23 -07002059 return adb_connect_command("track-jdwp");
Shukang Zhouf4ffae12020-02-13 17:01:39 -08002060 } else if (!strcmp(argv[0], "track-app")) {
Josh Gaofb4616b2020-04-16 19:34:43 -07002061 std::string error;
2062 auto&& features = adb_get_feature_set(&error);
2063 if (!features) {
2064 error_exit("%s", error.c_str());
Shukang Zhouf4ffae12020-02-13 17:01:39 -08002065 }
Josh Gaofb4616b2020-04-16 19:34:43 -07002066 if (!CanUseFeature(*features, kFeatureTrackApp)) {
Shukang Zhouf4ffae12020-02-13 17:01:39 -08002067 error_exit("track-app is not supported by the device");
2068 }
2069 TrackAppStreamsCallback callback;
2070 return adb_connect_command("track-app", nullptr, &callback);
Josh Gao2dc4cab2018-11-15 17:45:46 -08002071 } else if (!strcmp(argv[0], "track-devices")) {
Elliott Hughesfc2f5f62019-07-31 14:13:57 -07002072 if (argc > 2 || (argc == 2 && strcmp(argv[1], "-l"))) {
2073 error_exit("usage: adb track-devices [-l]");
2074 }
2075 return adb_connect_command(argc == 2 ? "host:track-devices-l" : "host:track-devices");
Josh Gao172d0d42018-06-18 14:51:56 -07002076 } else if (!strcmp(argv[0], "raw")) {
2077 if (argc != 2) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002078 error_exit("usage: adb raw SERVICE");
Josh Gao172d0d42018-06-18 14:51:56 -07002079 }
Josh Gaoe89a55d2018-11-16 14:47:59 -08002080 return adb_connect_command_bidirectional(argv[1]);
Josh Gaob463baf2016-05-13 18:18:23 -07002081 }
2082
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002083 /* "adb /?" is a common idiom under Windows */
Elliott Hughes2ec36b02017-02-06 16:20:30 -08002084 else if (!strcmp(argv[0], "--help") || !strcmp(argv[0], "help") || !strcmp(argv[0], "/?")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002085 help();
2086 return 0;
Josh Gao2dc4cab2018-11-15 17:45:46 -08002087 } else if (!strcmp(argv[0], "--version") || !strcmp(argv[0], "version")) {
Elliott Hughes42ae2602015-08-12 08:32:10 -07002088 fprintf(stdout, "%s", adb_version().c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002089 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08002090 } else if (!strcmp(argv[0], "features")) {
David Pursell4e2fd362015-09-22 10:43:08 -07002091 // Only list the features common to both the adb client and the device.
Josh Gaofb4616b2020-04-16 19:34:43 -07002092 std::string error;
2093 auto&& features = adb_get_feature_set(&error);
2094 if (!features) {
2095 error_exit("%s", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08002096 }
2097
Josh Gaofb4616b2020-04-16 19:34:43 -07002098 for (const std::string& name : *features) {
2099 if (CanUseFeature(*features, name)) {
David Pursell4e2fd362015-09-22 10:43:08 -07002100 printf("%s\n", name.c_str());
2101 }
2102 }
2103 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08002104 } else if (!strcmp(argv[0], "host-features")) {
2105 return adb_query_command("host:host-features");
Yabin Cui1f4ec192016-04-05 13:50:44 -07002106 } else if (!strcmp(argv[0], "reconnect")) {
2107 if (argc == 1) {
Josh Gaob122b172017-08-16 16:57:01 -07002108 return adb_query_command(format_host_command(argv[0]));
Josh Gao22d2b3e2016-10-27 14:01:08 -07002109 } else if (argc == 2) {
2110 if (!strcmp(argv[1], "device")) {
2111 std::string err;
2112 adb_connect("reconnect", &err);
2113 return 0;
2114 } else if (!strcmp(argv[1], "offline")) {
2115 std::string err;
2116 return adb_query_command("host:reconnect-offline");
2117 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002118 error_exit("usage: adb reconnect [device|offline]");
Josh Gao22d2b3e2016-10-27 14:01:08 -07002119 }
Yabin Cui1f4ec192016-04-05 13:50:44 -07002120 }
Alex Buynytskyy96ff54b2020-02-13 06:52:04 -08002121 } else if (!strcmp(argv[0], "inc-server")) {
Songchun Fan23bd5522020-03-09 11:33:44 -07002122 if (argc < 4) {
2123#ifdef _WIN32
2124 error_exit("usage: adb inc-server CONNECTION_HANDLE OUTPUT_HANDLE FILE1 FILE2 ...");
2125#else
2126 error_exit("usage: adb inc-server CONNECTION_FD OUTPUT_FD FILE1 FILE2 ...");
2127#endif
Alex Buynytskyy96ff54b2020-02-13 06:52:04 -08002128 }
Songchun Fan23bd5522020-03-09 11:33:44 -07002129 int connection_fd = atoi(argv[1]);
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07002130 if (!_is_valid_os_fd(connection_fd)) {
Songchun Fan23bd5522020-03-09 11:33:44 -07002131 error_exit("Invalid connection_fd number given: %d", connection_fd);
Alex Buynytskyy96ff54b2020-02-13 06:52:04 -08002132 }
Songchun Fan23bd5522020-03-09 11:33:44 -07002133
2134 connection_fd = adb_register_socket(connection_fd);
2135 close_on_exec(connection_fd);
2136
2137 int output_fd = atoi(argv[2]);
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07002138 if (!_is_valid_os_fd(output_fd)) {
Songchun Fan23bd5522020-03-09 11:33:44 -07002139 error_exit("Invalid output_fd number given: %d", output_fd);
2140 }
2141 output_fd = adb_register_socket(output_fd);
2142 close_on_exec(output_fd);
2143 return incremental::serve(connection_fd, output_fd, argc - 3, argv + 3);
Dan Albert90d4b732015-05-20 18:58:41 -07002144 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002145
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002146 error_exit("unknown command %s", argv[0]);
Elliott Hughes4679a392018-10-19 13:59:44 -07002147 __builtin_unreachable();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002148}