blob: ceb21d59599c4df518b02eb895dd5f860c603336 [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"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800130 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700131 "file transfer:\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700132 " push [--sync] [-z ALGORITHM] [-Z] LOCAL... REMOTE\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700133 " copy local files/directories to device\n"
Dan Albert06b0d6b2017-05-18 22:56:48 -0700134 " --sync: only push files that are newer on the host than the device\n"
Josh Gao5949fcc2020-03-30 23:25:16 -0700135 " -n: dry run: push files to device without storing to the filesystem\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700136 " -z: enable compression with a specified algorithm (any, none, brotli)\n"
Josh Gao939fc192020-03-04 19:34:08 -0800137 " -Z: disable compression\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700138 " pull [-a] [-z ALGORITHM] [-Z] REMOTE... LOCAL\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700139 " copy files/dirs from device\n"
140 " -a: preserve file timestamp and mode\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700141 " -z: enable compression with a specified algorithm (any, none, brotli)\n"
Josh Gao939fc192020-03-04 19:34:08 -0800142 " -Z: disable compression\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700143 " sync [-l] [-z ALGORITHM] [-Z] [all|data|odm|oem|product|system|system_ext|vendor]\n"
Elliott Hughes0754cb92017-05-01 11:04:31 -0700144 " sync a local build from $ANDROID_PRODUCT_OUT to the device (default all)\n"
Josh Gao5949fcc2020-03-30 23:25:16 -0700145 " -n: dry run: push files to device without storing to the filesystem\n"
Elliott Hughes4c8ea892019-07-30 10:32:53 -0700146 " -l: list files that would be copied, but don't copy them\n"
Josh Gao49ba5582020-03-26 19:33:25 -0700147 " -z: enable compression with a specified algorithm (any, none, brotli)\n"
Josh Gao939fc192020-03-04 19:34:08 -0800148 " -Z: disable compression\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800149 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700150 "shell:\n"
151 " shell [-e ESCAPE] [-n] [-Tt] [-x] [COMMAND...]\n"
152 " run remote shell command (interactive shell if no command given)\n"
153 " -e: choose escape character, or \"none\"; default '~'\n"
154 " -n: don't read from stdin\n"
Elliott Hughesdfe6c842019-12-13 16:43:10 -0800155 " -T: disable pty allocation\n"
156 " -t: allocate a pty if on a tty (-tt: force pty allocation)\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700157 " -x: disable remote exit codes and stdout/stderr separation\n"
158 " emu COMMAND run emulator console command\n"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800159 "\n"
Elliott Hughes0d4bf802019-01-11 13:50:05 -0800160 "app installation (see also `adb shell cmd package help`):\n"
Felipe Leme807be772018-05-08 18:40:18 -0700161 " install [-lrtsdg] [--instant] PACKAGE\n"
Patrick Baumanndc58b0a2018-10-09 10:45:03 -0700162 " push a single package to the device and install it\n"
Felipe Leme807be772018-05-08 18:40:18 -0700163 " install-multiple [-lrtsdpg] [--instant] PACKAGE...\n"
Patrick Baumanndc58b0a2018-10-09 10:45:03 -0700164 " push multiple APKs to the device for a single package and install them\n"
165 " install-multi-package [-lrtsdpg] [--instant] PACKAGE...\n"
166 " push one or more packages to the device and install them atomically\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700167 " -r: replace existing application\n"
168 " -t: allow test packages\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700169 " -d: allow version code downgrade (debuggable packages only)\n"
170 " -p: partial application install (install-multiple only)\n"
171 " -g: grant all runtime permissions\n"
Elliott Hughesc0ebd592019-08-02 17:09:30 -0700172 " --abi ABI: override platform's default ABI\n"
Felipe Leme807be772018-05-08 18:40:18 -0700173 " --instant: cause the app to be installed as an ephemeral install app\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000174 " --no-streaming: always push APK to device and invoke Package Manager as separate steps\n"
175 " --streaming: force streaming APK directly into Package Manager\n"
Henry Daitxee01c802018-12-12 10:40:57 +0000176 " --fastdeploy: use fast deploy\n"
177 " --no-fastdeploy: prevent use of fast deploy\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000178 " --force-agent: force update of deployment agent when using fast deploy\n"
179 " --date-check-agent: update deployment agent when local version is newer and using fast deploy\n"
180 " --version-check-agent: update deployment agent when local version has different version code and using fast deploy\n"
181#ifndef _WIN32
182 " --local-agent: locate agent files from local source build (instead of SDK location)\n"
183#endif
Elliott Hughesc0ebd592019-08-02 17:09:30 -0700184 " (See also `adb shell pm help` for more options.)\n"
Idries Hamadied409ea2018-01-29 16:30:36 +0000185 //TODO--installlog <filename>
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700186 " uninstall [-k] PACKAGE\n"
187 " remove this app package from the device\n"
188 " '-k': keep the data and cache directories\n"
189 "\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700190 "debugging:\n"
191 " bugreport [PATH]\n"
192 " write bugreport to given PATH [default=bugreport.zip];\n"
193 " if PATH is a directory, the bug report is saved in that directory.\n"
194 " devices that don't support zipped bug reports output to stdout.\n"
195 " jdwp list pids of processes hosting a JDWP transport\n"
196 " logcat show device log (logcat --help for more)\n"
197 "\n"
198 "security:\n"
199 " disable-verity disable dm-verity checking on userdebug builds\n"
200 " enable-verity re-enable dm-verity checking on userdebug builds\n"
201 " keygen FILE\n"
202 " generate adb public/private key; private key stored in FILE,\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700203 "\n"
204 "scripting:\n"
Elliott Hughesa52b4582020-03-10 16:57:47 -0700205 " wait-for[-TRANSPORT]-STATE...\n"
206 " wait for device to be in a given state\n"
Tao Bao55d407e2019-04-07 23:24:03 -0700207 " STATE: device, recovery, rescue, sideload, bootloader, or disconnect\n"
Josh Gao1e9e4712019-02-22 14:01:36 -0800208 " TRANSPORT: usb, local, or any [default=any]\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700209 " get-state print offline | bootloader | device\n"
210 " get-serialno print <serial-number>\n"
211 " get-devpath print <device-path>\n"
David Andersona5db1d72018-11-21 13:43:22 -0800212 " remount [-R]\n"
213 " remount partitions read-write. if a reboot is required, -R will\n"
214 " will automatically reboot the device.\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700215 " reboot [bootloader|recovery|sideload|sideload-auto-reboot]\n"
216 " reboot the device; defaults to booting system image but\n"
217 " supports bootloader and recovery too. sideload reboots\n"
218 " into recovery and automatically starts sideload mode,\n"
219 " sideload-auto-reboot is the same but reboots after sideloading.\n"
220 " sideload OTAPACKAGE sideload the given full OTA package\n"
221 " root restart adbd with root permissions\n"
222 " unroot restart adbd without root permissions\n"
bohuea0c3cb2018-10-24 10:38:33 -0700223 " usb restart adbd listening on USB\n"
224 " tcpip PORT restart adbd listening on TCP on PORT\n"
Timcd643152010-02-16 20:18:29 +0000225 "\n"
Yabin Cui1f4ec192016-04-05 13:50:44 -0700226 "internal debugging:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700227 " start-server ensure that there is a server running\n"
228 " kill-server kill the server if it is running\n"
229 " reconnect kick connection from host side to force reconnect\n"
230 " reconnect device kick connection from device side to force reconnect\n"
Yabin Cuib5e11412017-03-10 16:01:01 -0800231 " reconnect offline reset offline/unauthorized devices to force reconnect\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700232 "\n"
Elliott Hughes7e067cf2015-06-12 14:26:29 -0700233 "environment variables:\n"
Elliott Hughes4bf9bc12016-09-28 15:25:47 -0700234 " $ADB_TRACE\n"
235 " comma-separated list of debug info to log:\n"
236 " all,adb,sockets,packets,rwx,usb,sync,sysdeps,transport,jdwp\n"
237 " $ADB_VENDOR_KEYS colon-separated list of keys (files or directories)\n"
238 " $ANDROID_SERIAL serial number to connect to (see -s)\n"
Tim Baverstock25173c82019-02-08 16:27:16 +0000239 " $ANDROID_LOG_TAGS tags to be used by logcat (see logcat --help)\n"
240 " $ADB_LOCAL_TRANSPORT_MAX_PORT max emulator scan port (default 5585, 16 emus)\n"
241 );
Felipe Lemeb3239722016-07-29 17:57:00 -0700242 // clang-format on
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800243}
244
Yabin Cuid325e862014-11-17 14:48:25 -0800245#if defined(_WIN32)
246
Elliott Hughesa2f2e562015-04-16 16:47:02 -0700247// Implemented in sysdeps_win32.cpp.
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800248void stdin_raw_init();
249void stdin_raw_restore();
Yabin Cuid325e862014-11-17 14:48:25 -0800250
251#else
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100252static termios g_saved_terminal_state;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800253
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800254static void stdin_raw_init() {
255 if (tcgetattr(STDIN_FILENO, &g_saved_terminal_state)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800256
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100257 termios tio;
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800258 if (tcgetattr(STDIN_FILENO, &tio)) return;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800259
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100260 cfmakeraw(&tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800261
Alistair Buxtondfa09fd2013-03-01 22:16:41 +0100262 // No timeout but request at least one character per read.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800263 tio.c_cc[VTIME] = 0;
264 tio.c_cc[VMIN] = 1;
265
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800266 tcsetattr(STDIN_FILENO, TCSAFLUSH, &tio);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800267}
268
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800269static void stdin_raw_restore() {
270 tcsetattr(STDIN_FILENO, TCSAFLUSH, &g_saved_terminal_state);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800271}
272#endif
273
Alex Buynytskyy665f3ff2019-09-16 12:10:54 -0700274int read_and_dump(borrowed_fd fd, bool use_shell_protocol,
275 StandardStreamsCallbackInterface* callback) {
David Pursell606835a2015-09-08 17:17:02 -0700276 int exit_code = 0;
Felipe Lemed1dd1252016-06-06 16:05:36 -0700277 if (fd < 0) return exit_code;
278
David Pursell606835a2015-09-08 17:17:02 -0700279 std::unique_ptr<ShellProtocol> protocol;
280 int length = 0;
David Pursell606835a2015-09-08 17:17:02 -0700281
282 char raw_buffer[BUFSIZ];
283 char* buffer_ptr = raw_buffer;
284 if (use_shell_protocol) {
Josh Gao31b5be62018-03-07 16:51:08 -0800285 protocol = std::make_unique<ShellProtocol>(fd);
David Pursell606835a2015-09-08 17:17:02 -0700286 if (!protocol) {
287 LOG(ERROR) << "failed to allocate memory for ShellProtocol object";
288 return 1;
289 }
290 buffer_ptr = protocol->data();
291 }
292
Felipe Lemed1dd1252016-06-06 16:05:36 -0700293 while (true) {
David Pursell606835a2015-09-08 17:17:02 -0700294 if (use_shell_protocol) {
295 if (!protocol->Read()) {
296 break;
297 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700298 length = protocol->data_length();
David Pursell606835a2015-09-08 17:17:02 -0700299 switch (protocol->id()) {
300 case ShellProtocol::kIdStdout:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700301 callback->OnStdout(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700302 break;
303 case ShellProtocol::kIdStderr:
Felipe Leme0d4f0502016-07-26 12:14:39 -0700304 callback->OnStderr(buffer_ptr, length);
David Pursell606835a2015-09-08 17:17:02 -0700305 break;
306 case ShellProtocol::kIdExit:
Josh Gao1e1ae452019-03-15 14:49:25 -0700307 // data() returns a char* which doesn't have defined signedness.
308 // Cast to uint8_t to prevent 255 from being sign extended to INT_MIN,
309 // which doesn't get truncated on Windows.
310 exit_code = static_cast<uint8_t>(protocol->data()[0]);
David Pursell606835a2015-09-08 17:17:02 -0700311 continue;
312 default:
313 continue;
314 }
315 length = protocol->data_length();
316 } else {
Josh Gao27241a72019-04-25 14:04:57 -0700317 D("read_and_dump(): pre adb_read(fd=%d)", fd.get());
David Pursell606835a2015-09-08 17:17:02 -0700318 length = adb_read(fd, raw_buffer, sizeof(raw_buffer));
Josh Gao27241a72019-04-25 14:04:57 -0700319 D("read_and_dump(): post adb_read(fd=%d): length=%d", fd.get(), length);
David Pursell606835a2015-09-08 17:17:02 -0700320 if (length <= 0) {
321 break;
322 }
Felipe Leme0d4f0502016-07-26 12:14:39 -0700323 callback->OnStdout(buffer_ptr, length);
Felipe Leme44a42672016-04-01 17:43:27 -0700324 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800325 }
David Pursell606835a2015-09-08 17:17:02 -0700326
Felipe Leme0d4f0502016-07-26 12:14:39 -0700327 return callback->Done(exit_code);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800328}
329
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700330static void stdinout_raw_prologue(int inFd, int outFd, int& old_stdin_mode, int& old_stdout_mode) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700331 if (inFd == STDIN_FILENO) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800332 stdin_raw_init();
Spencer Lowb7dfb792015-05-22 16:48:31 -0700333#ifdef _WIN32
334 old_stdin_mode = _setmode(STDIN_FILENO, _O_BINARY);
335 if (old_stdin_mode == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700336 PLOG(FATAL) << "could not set stdin to binary";
Spencer Lowb7dfb792015-05-22 16:48:31 -0700337 }
338#endif
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700339 }
Yabin Cuid325e862014-11-17 14:48:25 -0800340
Spencer Lowb7dfb792015-05-22 16:48:31 -0700341#ifdef _WIN32
342 if (outFd == STDOUT_FILENO) {
343 old_stdout_mode = _setmode(STDOUT_FILENO, _O_BINARY);
344 if (old_stdout_mode == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700345 PLOG(FATAL) << "could not set stdout to binary";
Spencer Lowb7dfb792015-05-22 16:48:31 -0700346 }
347 }
348#endif
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700349}
350
351static void stdinout_raw_epilogue(int inFd, int outFd, int old_stdin_mode, int old_stdout_mode) {
352 if (inFd == STDIN_FILENO) {
353 stdin_raw_restore();
354#ifdef _WIN32
355 if (_setmode(STDIN_FILENO, old_stdin_mode) == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700356 PLOG(FATAL) << "could not restore stdin mode";
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700357 }
358#endif
359 }
360
361#ifdef _WIN32
362 if (outFd == STDOUT_FILENO) {
363 if (_setmode(STDOUT_FILENO, old_stdout_mode) == -1) {
Elliott Hughes4679a392018-10-19 13:59:44 -0700364 PLOG(FATAL) << "could not restore stdout mode";
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700365 }
366 }
367#endif
368}
369
Josh Gao3828ebc2019-12-03 16:05:54 -0800370bool copy_to_file(int inFd, int outFd) {
371 bool result = true;
Yurii Zubrytskyi6eca0e62019-06-27 13:47:34 -0700372 std::vector<char> buf(64 * 1024);
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700373 int len;
374 long total = 0;
375 int old_stdin_mode = -1;
376 int old_stdout_mode = -1;
377
378 D("copy_to_file(%d -> %d)", inFd, outFd);
379
380 stdinout_raw_prologue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700381
Elliott Hughesa7090b92015-04-17 17:03:59 -0700382 while (true) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700383 if (inFd == STDIN_FILENO) {
Elliott Hughes86ab9ff2018-09-05 12:13:11 -0700384 len = unix_read(inFd, buf.data(), buf.size());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700385 } else {
Elliott Hughes86ab9ff2018-09-05 12:13:11 -0700386 len = adb_read(inFd, buf.data(), buf.size());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700387 }
Christopher Tated2f54152011-04-21 12:53:28 -0700388 if (len == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700389 D("copy_to_file() : read 0 bytes; exiting");
Christopher Tated2f54152011-04-21 12:53:28 -0700390 break;
391 }
392 if (len < 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700393 D("copy_to_file(): read failed: %s", strerror(errno));
Josh Gao3828ebc2019-12-03 16:05:54 -0800394 result = false;
Christopher Tated2f54152011-04-21 12:53:28 -0700395 break;
396 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700397 if (outFd == STDOUT_FILENO) {
Luis Hector Chavez6150a372018-07-18 21:18:27 -0700398 fwrite(buf.data(), 1, len, stdout);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700399 fflush(stdout);
400 } else {
Luis Hector Chavez6150a372018-07-18 21:18:27 -0700401 adb_write(outFd, buf.data(), len);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -0700402 }
Christopher Tatec9cd3b92011-06-01 17:56:23 -0700403 total += len;
Christopher Tated2f54152011-04-21 12:53:28 -0700404 }
Yabin Cuid325e862014-11-17 14:48:25 -0800405
Christopher Tatec42f1bb2016-07-06 13:22:22 -0700406 stdinout_raw_epilogue(inFd, outFd, old_stdin_mode, old_stdout_mode);
Spencer Lowb7dfb792015-05-22 16:48:31 -0700407
Josh Gao3828ebc2019-12-03 16:05:54 -0800408 D("copy_to_file() finished with %s after %lu bytes", result ? "success" : "failure", total);
409 return result;
Christopher Tated2f54152011-04-21 12:53:28 -0700410}
411
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800412static void send_window_size_change(int fd, std::unique_ptr<ShellProtocol>& shell) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800413 // Old devices can't handle window size changes.
414 if (shell == nullptr) return;
415
Spencer Low2e02dc62015-11-07 17:34:39 -0800416#if defined(_WIN32)
417 struct winsize {
418 unsigned short ws_row;
419 unsigned short ws_col;
420 unsigned short ws_xpixel;
421 unsigned short ws_ypixel;
422 };
423#endif
424
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800425 winsize ws;
Spencer Low2e02dc62015-11-07 17:34:39 -0800426
427#if defined(_WIN32)
428 // If stdout is redirected to a non-console, we won't be able to get the
429 // console size, but that makes sense.
430 const intptr_t intptr_handle = _get_osfhandle(STDOUT_FILENO);
431 if (intptr_handle == -1) return;
432
433 const HANDLE handle = reinterpret_cast<const HANDLE>(intptr_handle);
434
435 CONSOLE_SCREEN_BUFFER_INFO info;
436 memset(&info, 0, sizeof(info));
437 if (!GetConsoleScreenBufferInfo(handle, &info)) return;
438
439 memset(&ws, 0, sizeof(ws));
440 // The number of visible rows, excluding offscreen scroll-back rows which are in info.dwSize.Y.
441 ws.ws_row = info.srWindow.Bottom - info.srWindow.Top + 1;
442 // If the user has disabled "Wrap text output on resize", they can make the screen buffer wider
443 // than the window, in which case we should use the width of the buffer.
444 ws.ws_col = info.dwSize.X;
445#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800446 if (ioctl(fd, TIOCGWINSZ, &ws) == -1) return;
Spencer Low2e02dc62015-11-07 17:34:39 -0800447#endif
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800448
449 // Send the new window size as human-readable ASCII for debugging convenience.
450 size_t l = snprintf(shell->data(), shell->data_capacity(), "%dx%d,%dx%d",
451 ws.ws_row, ws.ws_col, ws.ws_xpixel, ws.ws_ypixel);
452 shell->Write(ShellProtocol::kIdWindowSizeChange, l + 1);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800453}
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800454
David Pursell606835a2015-09-08 17:17:02 -0700455// Used to pass multiple values to the stdin read thread.
456struct StdinReadArgs {
457 int stdin_fd, write_fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700458 bool raw_stdin;
David Pursell606835a2015-09-08 17:17:02 -0700459 std::unique_ptr<ShellProtocol> protocol;
Elliott Hughes28416d62015-11-04 13:36:32 -0800460 char escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700461};
462
David Pursell606835a2015-09-08 17:17:02 -0700463// Loops to read from stdin and push the data to the given FD.
464// The argument should be a pointer to a StdinReadArgs object. This function
465// will take ownership of the object and delete it when finished.
Josh Gaod9db09c2016-02-12 14:31:15 -0800466static void stdin_read_thread_loop(void* x) {
David Pursell606835a2015-09-08 17:17:02 -0700467 std::unique_ptr<StdinReadArgs> args(reinterpret_cast<StdinReadArgs*>(x));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800468
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800469#if !defined(_WIN32)
470 // Mask SIGTTIN in case we're in a backgrounded process.
Josh Gao8dcdb572015-10-23 15:03:31 -0700471 sigset_t sigset;
472 sigemptyset(&sigset);
473 sigaddset(&sigset, SIGTTIN);
474 pthread_sigmask(SIG_BLOCK, &sigset, nullptr);
475#endif
476
Spencer Low2e02dc62015-11-07 17:34:39 -0800477#if defined(_WIN32)
478 // _get_interesting_input_record_uncached() causes unix_read_interruptible()
479 // to return -1 with errno == EINTR if the window size changes.
480#else
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800481 // Unblock SIGWINCH for this thread, so our read(2) below will be
482 // interrupted if the window size changes.
483 sigset_t mask;
484 sigemptyset(&mask);
485 sigaddset(&mask, SIGWINCH);
486 pthread_sigmask(SIG_UNBLOCK, &mask, nullptr);
487#endif
488
489 // Set up the initial window size.
490 send_window_size_change(args->stdin_fd, args->protocol);
491
Elliott Hughes28416d62015-11-04 13:36:32 -0800492 char raw_buffer[BUFSIZ];
David Pursell606835a2015-09-08 17:17:02 -0700493 char* buffer_ptr = raw_buffer;
494 size_t buffer_size = sizeof(raw_buffer);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800495 if (args->protocol != nullptr) {
David Pursell606835a2015-09-08 17:17:02 -0700496 buffer_ptr = args->protocol->data();
497 buffer_size = args->protocol->data_capacity();
498 }
499
Elliott Hughes28416d62015-11-04 13:36:32 -0800500 // If we need to parse escape sequences, make life easy.
501 if (args->raw_stdin && args->escape_char != '\0') {
502 buffer_size = 1;
503 }
504
505 enum EscapeState { kMidFlow, kStartOfLine, kInEscape };
506 EscapeState state = kStartOfLine;
507
Elliott Hughesaa245492015-08-03 10:38:08 -0700508 while (true) {
Spencer Low2e02dc62015-11-07 17:34:39 -0800509 // Use unix_read_interruptible() rather than adb_read() for stdin.
510 D("stdin_read_thread_loop(): pre unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
511 int r = unix_read_interruptible(args->stdin_fd, buffer_ptr,
512 buffer_size);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800513 if (r == -1 && errno == EINTR) {
514 send_window_size_change(args->stdin_fd, args->protocol);
515 continue;
516 }
Spencer Low2e02dc62015-11-07 17:34:39 -0800517 D("stdin_read_thread_loop(): post unix_read_interruptible(fdi=%d,...)", args->stdin_fd);
David Pursell1ed57f02015-10-06 15:30:03 -0700518 if (r <= 0) {
519 // Only devices using the shell protocol know to close subprocess
520 // stdin. For older devices we want to just leave the connection
521 // open, otherwise an unpredictable amount of return data could
522 // be lost due to the FD closing before all data has been received.
523 if (args->protocol) {
524 args->protocol->Write(ShellProtocol::kIdCloseStdin, 0);
525 }
526 break;
527 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800528 // If we made stdin raw, check input for escape sequences. In
David Pursell1ed57f02015-10-06 15:30:03 -0700529 // this situation signals like Ctrl+C are sent remotely rather than
530 // interpreted locally so this provides an emergency out if the remote
531 // process starts ignoring the signal. SSH also does this, see the
532 // "escape characters" section on the ssh man page for more info.
Elliott Hughes28416d62015-11-04 13:36:32 -0800533 if (args->raw_stdin && args->escape_char != '\0') {
534 char ch = buffer_ptr[0];
535 if (ch == args->escape_char) {
536 if (state == kStartOfLine) {
537 state = kInEscape;
538 // Swallow the escape character.
539 continue;
540 } else {
541 state = kMidFlow;
542 }
543 } else {
544 if (state == kInEscape) {
545 if (ch == '.') {
546 fprintf(stderr,"\r\n[ disconnected ]\r\n");
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800547 stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700548 exit(0);
Elliott Hughes28416d62015-11-04 13:36:32 -0800549 } else {
550 // We swallowed an escape character that wasn't part of
551 // a valid escape sequence; time to cough it up.
552 buffer_ptr[0] = args->escape_char;
553 buffer_ptr[1] = ch;
554 ++r;
David Pursell1ed57f02015-10-06 15:30:03 -0700555 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800556 }
Elliott Hughes28416d62015-11-04 13:36:32 -0800557 state = (ch == '\n' || ch == '\r') ? kStartOfLine : kMidFlow;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800558 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800559 }
David Pursell606835a2015-09-08 17:17:02 -0700560 if (args->protocol) {
561 if (!args->protocol->Write(ShellProtocol::kIdStdin, r)) {
562 break;
563 }
564 } else {
565 if (!WriteFdExactly(args->write_fd, buffer_ptr, r)) {
566 break;
567 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800568 }
569 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800570}
571
David Pursell1ed57f02015-10-06 15:30:03 -0700572// Returns a shell service string with the indicated arguments and command.
573static std::string ShellServiceString(bool use_shell_protocol,
574 const std::string& type_arg,
575 const std::string& command) {
576 std::vector<std::string> args;
577 if (use_shell_protocol) {
578 args.push_back(kShellServiceArgShellProtocol);
Elliott Hughesc2252df2015-11-18 12:45:48 -0800579
580 const char* terminal_type = getenv("TERM");
581 if (terminal_type != nullptr) {
582 args.push_back(std::string("TERM=") + terminal_type);
583 }
David Pursell1ed57f02015-10-06 15:30:03 -0700584 }
585 if (!type_arg.empty()) {
586 args.push_back(type_arg);
587 }
588
589 // Shell service string can look like: shell[,arg1,arg2,...]:[command].
590 return android::base::StringPrintf("shell%s%s:%s",
591 args.empty() ? "" : ",",
592 android::base::Join(args, ',').c_str(),
593 command.c_str());
594}
595
596// Connects to a shell on the device and read/writes data.
597//
598// Note: currently this function doesn't properly clean up resources; the
599// FD connected to the adb server is never closed and the stdin read thread
600// may never exit.
601//
602// On success returns the remote exit code if |use_shell_protocol| is true,
603// 0 otherwise. On failure returns 1.
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800604static int RemoteShell(bool use_shell_protocol, const std::string& type_arg, char escape_char,
605 bool empty_command, const std::string& service_string) {
Josh Gaoa019f782017-06-16 15:34:34 -0700606 // Old devices can't handle a service string that's longer than MAX_PAYLOAD_V1.
607 // Use |use_shell_protocol| to determine whether to allow a command longer than that.
608 if (service_string.size() > MAX_PAYLOAD_V1 && !use_shell_protocol) {
609 fprintf(stderr, "error: shell command too long\n");
610 return 1;
611 }
612
David Pursell1ed57f02015-10-06 15:30:03 -0700613 // Make local stdin raw if the device allocates a PTY, which happens if:
614 // 1. We are explicitly asking for a PTY shell, or
615 // 2. We don't specify shell type and are starting an interactive session.
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800616 bool raw_stdin = (type_arg == kShellServiceArgPty || (type_arg.empty() && empty_command));
David Pursell1ed57f02015-10-06 15:30:03 -0700617
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700618 std::string error;
David Pursell4e2fd362015-09-22 10:43:08 -0700619 int fd = adb_connect(service_string, &error);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700620 if (fd < 0) {
621 fprintf(stderr,"error: %s\n", error.c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800622 return 1;
623 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800624
David Pursell606835a2015-09-08 17:17:02 -0700625 StdinReadArgs* args = new StdinReadArgs;
626 if (!args) {
627 LOG(ERROR) << "couldn't allocate StdinReadArgs object";
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700628 return 1;
629 }
David Pursell1ed57f02015-10-06 15:30:03 -0700630 args->stdin_fd = STDIN_FILENO;
David Pursell606835a2015-09-08 17:17:02 -0700631 args->write_fd = fd;
David Pursell1ed57f02015-10-06 15:30:03 -0700632 args->raw_stdin = raw_stdin;
Elliott Hughes28416d62015-11-04 13:36:32 -0800633 args->escape_char = escape_char;
David Pursell606835a2015-09-08 17:17:02 -0700634 if (use_shell_protocol) {
Josh Gao31b5be62018-03-07 16:51:08 -0800635 args->protocol = std::make_unique<ShellProtocol>(args->write_fd);
David Pursell606835a2015-09-08 17:17:02 -0700636 }
Elliott Hughesdc3b4592015-04-21 19:39:52 -0700637
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800638 if (raw_stdin) stdin_raw_init();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800639
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800640#if !defined(_WIN32)
641 // Ensure our process is notified if the local window size changes.
642 // We use sigaction(2) to ensure that the SA_RESTART flag is not set,
643 // because the whole reason we're sending signals is to unblock the read(2)!
644 // That also means we don't need to do anything in the signal handler:
645 // the side effect of delivering the signal is all we need.
646 struct sigaction sa;
647 memset(&sa, 0, sizeof(sa));
648 sa.sa_handler = [](int) {};
649 sa.sa_flags = 0;
650 sigaction(SIGWINCH, &sa, nullptr);
651
652 // Now block SIGWINCH in this thread (the main thread) and all threads spawned
653 // from it. The stdin read thread will unblock this signal to ensure that it's
654 // the thread that receives the signal.
655 sigset_t mask;
656 sigemptyset(&mask);
657 sigaddset(&mask, SIGWINCH);
658 pthread_sigmask(SIG_BLOCK, &mask, nullptr);
659#endif
660
661 // TODO: combine read_and_dump with stdin_read_thread to make life simpler?
Josh Gaoe1dacfc2017-04-12 17:00:49 -0700662 std::thread(stdin_read_thread_loop, args).detach();
663 int exit_code = read_and_dump(fd, use_shell_protocol);
Elliott Hughes9b0f3542015-05-05 13:41:21 -0700664
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800665 // TODO: properly exit stdin_read_thread_loop and close |fd|.
David Pursell1ed57f02015-10-06 15:30:03 -0700666
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800667 // TODO: we should probably install signal handlers for this.
668 // TODO: can we use atexit? even on Windows?
669 if (raw_stdin) stdin_raw_restore();
David Pursell1ed57f02015-10-06 15:30:03 -0700670
David Pursell606835a2015-09-08 17:17:02 -0700671 return exit_code;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800672}
673
Josh Gaof3f6a1d2016-01-31 19:12:26 -0800674static int adb_shell(int argc, const char** argv) {
David Pursell0aacbbe2016-01-21 19:56:50 -0800675 FeatureSet features;
Elliott Hughes4679a392018-10-19 13:59:44 -0700676 std::string error_message;
677 if (!adb_get_feature_set(&features, &error_message)) {
678 fprintf(stderr, "error: %s\n", error_message.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -0800679 return 1;
680 }
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.
685 char escape_char = '~'; // -e
686 bool use_shell_protocol = CanUseFeature(features, kFeatureShell2); // -x
687 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",
762 !CanUseFeature(features, kFeatureShell2) ? "device" : "-x");
763 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) {
Alex Buynytskyy01a65ee2019-01-17 13:13:56 -0800782 FeatureSet features;
783 std::string error_message;
784 if (!adb_get_feature_set(&features, &error_message)) {
785 fprintf(stderr, "error: %s\n", error_message.c_str());
786 return 1;
787 }
788
789 if (!CanUseFeature(features, kFeatureAbb)) {
790 error_exit("abb is not supported by the device");
791 }
792
Alex Buynytskyycc2f1242019-06-27 10:33:54 -0700793 optind = 1; // argv[0] is always "abb", so set `optind` appropriately.
794
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800795 // Defaults.
796 constexpr char escape_char = '~'; // -e
797 constexpr bool use_shell_protocol = true;
798 constexpr auto shell_type_arg = kShellServiceArgRaw;
799 constexpr bool empty_command = false;
800
Alex Buynytskyycc2f1242019-06-27 10:33:54 -0700801 std::vector<const char*> args(argv + optind, argv + argc);
802 std::string service_string = "abb:" + android::base::Join(args, ABB_ARG_DELIMETER);
Alex Buynytskyy640407d2018-12-12 10:48:50 -0800803
804 D("abb -e 0x%x [%*.s]\n", escape_char, static_cast<int>(service_string.size()),
805 service_string.data());
806
807 return RemoteShell(use_shell_protocol, shell_type_arg, escape_char, empty_command,
808 service_string);
David Pursell71c83122015-09-14 15:33:50 -0700809}
810
Josh Gao9e756342020-01-08 12:39:52 -0800811static int adb_shell_noinput(int argc, const char** argv) {
812#if !defined(_WIN32)
813 unique_fd fd(adb_open("/dev/null", O_RDONLY));
814 CHECK_NE(STDIN_FILENO, fd.get());
815 dup2(fd.get(), STDIN_FILENO);
816#endif
817 return adb_shell(argc, argv);
818}
819
Elliott Hughes09e794c2017-05-02 14:31:59 -0700820static int adb_sideload_legacy(const char* filename, int in_fd, int size) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700821 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -0800822 unique_fd out_fd(adb_connect(android::base::StringPrintf("sideload:%d", size), &error));
Elliott Hughes09e794c2017-05-02 14:31:59 -0700823 if (out_fd < 0) {
824 fprintf(stderr, "adb: pre-KitKat sideload connection failed: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800825 return -1;
826 }
827
828 int opt = CHUNK_SIZE;
Elliott Hughes09e794c2017-05-02 14:31:59 -0700829 opt = adb_setsockopt(out_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
Doug Zongker447f0612012-01-09 14:54:53 -0800830
Elliott Hughes09e794c2017-05-02 14:31:59 -0700831 char buf[CHUNK_SIZE];
832 int total = size;
833 while (size > 0) {
834 unsigned xfer = (size > CHUNK_SIZE) ? CHUNK_SIZE : size;
835 if (!ReadFdExactly(in_fd, buf, xfer)) {
836 fprintf(stderr, "adb: failed to read data from %s: %s\n", filename, strerror(errno));
Doug Zongker447f0612012-01-09 14:54:53 -0800837 return -1;
838 }
Elliott Hughes09e794c2017-05-02 14:31:59 -0700839 if (!WriteFdExactly(out_fd, buf, xfer)) {
840 std::string error;
841 adb_status(out_fd, &error);
842 fprintf(stderr, "adb: failed to write data: %s\n", error.c_str());
Elliott Hughes09e794c2017-05-02 14:31:59 -0700843 return -1;
844 }
845 size -= xfer;
846 printf("sending: '%s' %4d%% \r", filename, (int)(100LL - ((100LL * size) / (total))));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700847 fflush(stdout);
Doug Zongker447f0612012-01-09 14:54:53 -0800848 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700849 printf("\n");
Doug Zongker447f0612012-01-09 14:54:53 -0800850
Elliott Hughes09e794c2017-05-02 14:31:59 -0700851 if (!adb_status(out_fd, &error)) {
852 fprintf(stderr, "adb: error response: %s\n", error.c_str());
Doug Zongker447f0612012-01-09 14:54:53 -0800853 return -1;
854 }
855
Doug Zongker447f0612012-01-09 14:54:53 -0800856 return 0;
857}
858
Doug Zongker71fe5842014-06-26 15:35:36 -0700859#define SIDELOAD_HOST_BLOCK_SIZE (CHUNK_SIZE)
860
Tao Bao11cf30b2019-04-07 23:24:03 -0700861// Connects to the sideload / rescue service on the device (served by minadbd) and sends over the
862// data in an OTA package.
863//
864// It uses a simple protocol as follows.
865//
866// - The connect message includes the total number of bytes in the file and a block size chosen by
867// us.
868//
869// - The other side sends the desired block number as eight decimal digits (e.g. "00000023" for
870// block 23). Blocks are numbered from zero.
871//
872// - We send back the data of the requested block. The last block is likely to be partial; when the
873// last block is requested we only send the part of the block that exists, it's not padded up to
874// the block size.
875//
876// - When the other side sends "DONEDONE" or "FAILFAIL" instead of a block number, we have done all
877// the data transfer.
878//
879static int adb_sideload_install(const char* filename, bool rescue_mode) {
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -0700880 // TODO: use a LinePrinter instead...
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700881 struct stat sb;
882 if (stat(filename, &sb) == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700883 fprintf(stderr, "adb: failed to stat file %s: %s\n", filename, strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700884 return -1;
885 }
886 unique_fd package_fd(adb_open(filename, O_RDONLY));
887 if (package_fd == -1) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700888 fprintf(stderr, "adb: failed to open file %s: %s\n", filename, strerror(errno));
Doug Zongker71fe5842014-06-26 15:35:36 -0700889 return -1;
890 }
891
Tao Bao11cf30b2019-04-07 23:24:03 -0700892 std::string service = android::base::StringPrintf(
893 "%s:%" PRId64 ":%d", rescue_mode ? "rescue-install" : "sideload-host",
894 static_cast<int64_t>(sb.st_size), SIDELOAD_HOST_BLOCK_SIZE);
Elliott Hughes078f0fc2015-04-29 08:35:59 -0700895 std::string error;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700896 unique_fd device_fd(adb_connect(service, &error));
897 if (device_fd < 0) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700898 fprintf(stderr, "adb: sideload connection failed: %s\n", error.c_str());
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700899
Tao Bao11cf30b2019-04-07 23:24:03 -0700900 if (rescue_mode) {
901 return -1;
902 }
903
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700904 // If this is a small enough package, maybe this is an older device that doesn't
905 // support sideload-host. Try falling back to the older (<= K) sideload method.
906 if (sb.st_size > INT_MAX) {
907 return -1;
908 }
Elliott Hughes09e794c2017-05-02 14:31:59 -0700909 fprintf(stderr, "adb: trying pre-KitKat sideload method...\n");
Josh Gao27241a72019-04-25 14:04:57 -0700910 return adb_sideload_legacy(filename, package_fd.get(), static_cast<int>(sb.st_size));
Doug Zongker71fe5842014-06-26 15:35:36 -0700911 }
912
Elliott Hughese8b663f2016-05-26 22:43:19 -0700913 int opt = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700914 adb_setsockopt(device_fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt));
915
916 char buf[SIDELOAD_HOST_BLOCK_SIZE];
Doug Zongker71fe5842014-06-26 15:35:36 -0700917
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700918 int64_t xfer = 0;
Elliott Hughese8b663f2016-05-26 22:43:19 -0700919 int last_percent = -1;
Elliott Hughesa7090b92015-04-17 17:03:59 -0700920 while (true) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700921 if (!ReadFdExactly(device_fd, buf, 8)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700922 fprintf(stderr, "adb: failed to read command: %s\n", strerror(errno));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700923 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700924 }
Elliott Hughes6452a892015-04-29 12:28:13 -0700925 buf[8] = '\0';
Doug Zongker71fe5842014-06-26 15:35:36 -0700926
xunchangc2265c52019-04-22 23:14:28 -0700927 if (strcmp(kMinadbdServicesExitSuccess, buf) == 0 ||
928 strcmp(kMinadbdServicesExitFailure, buf) == 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -0700929 printf("\rTotal xfer: %.2fx%*s\n",
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700930 static_cast<double>(xfer) / (sb.st_size ? sb.st_size : 1),
931 static_cast<int>(strlen(filename) + 10), "");
xunchangc2265c52019-04-22 23:14:28 -0700932 if (strcmp(kMinadbdServicesExitFailure, buf) == 0) {
Tao Bao11cf30b2019-04-07 23:24:03 -0700933 return 1;
934 }
Elliott Hughese8b663f2016-05-26 22:43:19 -0700935 return 0;
Doug Zongker71fe5842014-06-26 15:35:36 -0700936 }
937
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700938 int64_t block = strtoll(buf, nullptr, 10);
939 int64_t offset = block * SIDELOAD_HOST_BLOCK_SIZE;
940 if (offset >= static_cast<int64_t>(sb.st_size)) {
941 fprintf(stderr,
942 "adb: failed to read block %" PRId64 " at offset %" PRId64 ", past end %" PRId64
943 "\n",
944 block, offset, static_cast<int64_t>(sb.st_size));
Elliott Hughese8b663f2016-05-26 22:43:19 -0700945 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700946 }
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700947
Doug Zongker71fe5842014-06-26 15:35:36 -0700948 size_t to_write = SIDELOAD_HOST_BLOCK_SIZE;
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700949 if ((offset + SIDELOAD_HOST_BLOCK_SIZE) > static_cast<int64_t>(sb.st_size)) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700950 to_write = sb.st_size - offset;
Doug Zongker71fe5842014-06-26 15:35:36 -0700951 }
952
Elliott Hughescabfc3d2018-09-20 13:59:49 -0700953 if (adb_lseek(package_fd, offset, SEEK_SET) != offset) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700954 fprintf(stderr, "adb: failed to seek to package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700955 return -1;
956 }
957 if (!ReadFdExactly(package_fd, buf, to_write)) {
Elliott Hughes09e794c2017-05-02 14:31:59 -0700958 fprintf(stderr, "adb: failed to read package block: %s\n", strerror(errno));
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700959 return -1;
960 }
961
962 if (!WriteFdExactly(device_fd, buf, to_write)) {
963 adb_status(device_fd, &error);
Elliott Hughes09e794c2017-05-02 14:31:59 -0700964 fprintf(stderr, "adb: failed to write data '%s' *\n", error.c_str());
Elliott Hughese8b663f2016-05-26 22:43:19 -0700965 return -1;
Doug Zongker71fe5842014-06-26 15:35:36 -0700966 }
967 xfer += to_write;
968
969 // For normal OTA packages, we expect to transfer every byte
970 // twice, plus a bit of overhead (one read during
971 // verification, one read of each byte for installation, plus
972 // extra access to things like the zip central directory).
973 // This estimate of the completion becomes 100% when we've
974 // transferred ~2.13 (=100/47) times the package size.
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700975 int percent = static_cast<int>(xfer * 47LL / (sb.st_size ? sb.st_size : 1));
Doug Zongker71fe5842014-06-26 15:35:36 -0700976 if (percent != last_percent) {
Elliott Hughesee1ac7102017-03-21 17:14:18 -0700977 printf("\rserving: '%s' (~%d%%) ", filename, percent);
Doug Zongker71fe5842014-06-26 15:35:36 -0700978 fflush(stdout);
979 last_percent = percent;
980 }
981 }
Doug Zongker71fe5842014-06-26 15:35:36 -0700982}
983
xunchangc2265c52019-04-22 23:14:28 -0700984static int adb_wipe_devices() {
985 auto wipe_devices_message_size = strlen(kMinadbdServicesExitSuccess);
986 std::string error;
987 unique_fd fd(adb_connect(
988 android::base::StringPrintf("rescue-wipe:userdata:%zu", wipe_devices_message_size),
989 &error));
990 if (fd < 0) {
991 fprintf(stderr, "adb: wipe device connection failed: %s\n", error.c_str());
992 return 1;
993 }
994
995 std::string message(wipe_devices_message_size, '\0');
996 if (!ReadFdExactly(fd, message.data(), wipe_devices_message_size)) {
997 fprintf(stderr, "adb: failed to read wipe result: %s\n", strerror(errno));
998 return 1;
999 }
1000
1001 if (message == kMinadbdServicesExitSuccess) {
1002 return 0;
1003 }
1004
1005 if (message != kMinadbdServicesExitFailure) {
1006 fprintf(stderr, "adb: got unexpected message from rescue wipe %s\n", message.c_str());
1007 }
1008 return 1;
1009}
1010
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001011/**
1012 * Run ppp in "notty" mode against a resource listed as the first parameter
1013 * eg:
1014 *
1015 * ppp dev:/dev/omap_csmi_tty0 <ppp options>
1016 *
1017 */
Elliott Hughes58305772015-04-17 13:57:15 -07001018static int ppp(int argc, const char** argv) {
Yabin Cuie77b6a02014-11-11 09:24:11 -08001019#if defined(_WIN32)
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001020 error_exit("adb %s not implemented on Win32", argv[0]);
Elliott Hughes4679a392018-10-19 13:59:44 -07001021 __builtin_unreachable();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001022#else
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001023 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 -08001024
Dan Albertbac34742015-02-25 17:51:28 -08001025 const char* adb_service_name = argv[1];
Elliott Hughes4679a392018-10-19 13:59:44 -07001026 std::string error_message;
1027 int fd = adb_connect(adb_service_name, &error_message);
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001028 if (fd < 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001029 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 -08001030 }
1031
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001032 pid_t pid = fork();
Elliott Hughes4679a392018-10-19 13:59:44 -07001033 if (pid == -1) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001034 perror_exit("fork failed");
Elliott Hughes4679a392018-10-19 13:59:44 -07001035 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001036
Elliott Hughes4679a392018-10-19 13:59:44 -07001037 if (pid == 0) {
1038 // child side
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001039 int i;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001040
1041 // copy args
Elliott Hughes4679a392018-10-19 13:59:44 -07001042 const char** ppp_args = (const char**)alloca(sizeof(char*) * argc + 1);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001043 ppp_args[0] = "pppd";
1044 for (i = 2 ; i < argc ; i++) {
1045 //argv[2] and beyond become ppp_args[1] and beyond
1046 ppp_args[i - 1] = argv[i];
1047 }
Yi Kongaed415c2018-07-13 18:15:16 -07001048 ppp_args[i-1] = nullptr;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001049
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001050 dup2(fd, STDIN_FILENO);
1051 dup2(fd, STDOUT_FILENO);
1052 adb_close(STDERR_FILENO);
1053 adb_close(fd);
1054
Elliott Hughes4679a392018-10-19 13:59:44 -07001055 execvp("pppd", (char* const*)ppp_args);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001056 perror_exit("exec pppd failed");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001057 }
Elliott Hughes4679a392018-10-19 13:59:44 -07001058
1059 // parent side
1060 adb_close(fd);
1061 return 0;
Yabin Cuie77b6a02014-11-11 09:24:11 -08001062#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001063}
1064
Josh Gao3c99b0f2019-03-11 15:39:44 -07001065static bool wait_for_device(const char* service,
1066 std::optional<std::chrono::milliseconds> timeout = std::nullopt) {
Josh Gao08881e72016-04-13 12:14:16 -07001067 std::vector<std::string> components = android::base::Split(service, "-");
Elliott Hughesa52b4582020-03-10 16:57:47 -07001068 if (components.size() < 3) {
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001069 fprintf(stderr, "adb: couldn't parse 'wait-for' command: %s\n", service);
1070 return false;
1071 }
1072
Elliott Hughesa52b4582020-03-10 16:57:47 -07001073 // If the first thing after "wait-for-" wasn't a TRANSPORT, insert whatever
1074 // the current transport implies.
1075 if (components[2] != "usb" && components[2] != "local" && components[2] != "any") {
1076 TransportType t;
1077 adb_get_transport(&t, nullptr, nullptr);
Josh Gao08881e72016-04-13 12:14:16 -07001078 auto it = components.begin() + 2;
Elliott Hughes2b101112015-05-04 19:29:32 -07001079 if (t == kTransportUsb) {
Josh Gao08881e72016-04-13 12:14:16 -07001080 components.insert(it, "usb");
Elliott Hughes2b101112015-05-04 19:29:32 -07001081 } else if (t == kTransportLocal) {
Josh Gao08881e72016-04-13 12:14:16 -07001082 components.insert(it, "local");
Elliott Hughes2b101112015-05-04 19:29:32 -07001083 } else {
Josh Gao08881e72016-04-13 12:14:16 -07001084 components.insert(it, "any");
Elliott Hughes2b101112015-05-04 19:29:32 -07001085 }
Leo Sartre1fbc9db2015-11-27 18:56:48 +01001086 }
1087
Elliott Hughesa52b4582020-03-10 16:57:47 -07001088 // Stitch it back together and send it over...
Josh Gaob122b172017-08-16 16:57:01 -07001089 std::string cmd = format_host_command(android::base::Join(components, "-").c_str());
Josh Gao3c99b0f2019-03-11 15:39:44 -07001090 if (timeout) {
1091 std::thread([timeout]() {
1092 std::this_thread::sleep_for(*timeout);
1093 fprintf(stderr, "timeout expired while waiting for device\n");
1094 _exit(1);
1095 }).detach();
1096 }
Elliott Hughes424af022015-05-29 17:55:19 -07001097 return adb_command(cmd);
Elliott Hughes2b101112015-05-04 19:29:32 -07001098}
1099
Josh Gao06c73ae2016-03-04 15:16:18 -08001100static bool adb_root(const char* command) {
1101 std::string error;
Josh Gao06c73ae2016-03-04 15:16:18 -08001102
Josh Gao2020dd62019-02-22 15:20:27 -08001103 TransportId transport_id;
1104 unique_fd fd(adb_connect(&transport_id, android::base::StringPrintf("%s:", command), &error));
Elliott Hughese8b663f2016-05-26 22:43:19 -07001105 if (fd < 0) {
Josh Gao06c73ae2016-03-04 15:16:18 -08001106 fprintf(stderr, "adb: unable to connect for %s: %s\n", command, error.c_str());
1107 return false;
1108 }
1109
1110 // Figure out whether we actually did anything.
1111 char buf[256];
1112 char* cur = buf;
1113 ssize_t bytes_left = sizeof(buf);
1114 while (bytes_left > 0) {
Elliott Hughese8b663f2016-05-26 22:43:19 -07001115 ssize_t bytes_read = adb_read(fd, cur, bytes_left);
Josh Gao06c73ae2016-03-04 15:16:18 -08001116 if (bytes_read == 0) {
1117 break;
1118 } else if (bytes_read < 0) {
1119 fprintf(stderr, "adb: error while reading for %s: %s\n", command, strerror(errno));
1120 return false;
1121 }
1122 cur += bytes_read;
1123 bytes_left -= bytes_read;
1124 }
1125
1126 if (bytes_left == 0) {
1127 fprintf(stderr, "adb: unexpected output length for %s\n", command);
1128 return false;
1129 }
1130
Junyong Sun1ffccf52020-01-15 16:34:58 +08001131 fwrite(buf, 1, sizeof(buf) - bytes_left, stdout);
Josh Gao06c73ae2016-03-04 15:16:18 -08001132 fflush(stdout);
Josh Gao06c73ae2016-03-04 15:16:18 -08001133 if (cur != buf && strstr(buf, "restarting") == nullptr) {
1134 return true;
1135 }
1136
Josh Gao2020dd62019-02-22 15:20:27 -08001137 // Wait for the device to go away.
Josh Gao3c99b0f2019-03-11 15:39:44 -07001138 TransportType previous_type;
1139 const char* previous_serial;
1140 TransportId previous_id;
1141 adb_get_transport(&previous_type, &previous_serial, &previous_id);
1142
Josh Gao2020dd62019-02-22 15:20:27 -08001143 adb_set_transport(kTransportAny, nullptr, transport_id);
1144 wait_for_device("wait-for-disconnect");
Josh Gao3c99b0f2019-03-11 15:39:44 -07001145
1146 // Wait for the device to come back.
1147 // If we were using a specific transport ID, there's nothing we can wait for.
1148 if (previous_id == 0) {
1149 adb_set_transport(previous_type, previous_serial, 0);
Josh Gaoef4a9992019-05-01 18:51:46 -07001150 wait_for_device("wait-for-device", 6000ms);
Josh Gao3c99b0f2019-03-11 15:39:44 -07001151 }
1152
Josh Gao05824732016-06-16 14:00:09 -07001153 return true;
Josh Gao06c73ae2016-03-04 15:16:18 -08001154}
1155
Josh Gaob122b172017-08-16 16:57:01 -07001156int send_shell_command(const std::string& command, bool disable_shell_protocol,
1157 StandardStreamsCallbackInterface* callback) {
Josh Gao74ccdf92019-01-23 15:36:42 -08001158 unique_fd fd;
David Pursell0aacbbe2016-01-21 19:56:50 -08001159 bool use_shell_protocol = false;
1160
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001161 while (true) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001162 bool attempt_connection = true;
1163
Felipe Leme0d4f0502016-07-26 12:14:39 -07001164 // Use shell protocol if it's supported and the caller doesn't explicitly
1165 // disable it.
David Pursell0aacbbe2016-01-21 19:56:50 -08001166 if (!disable_shell_protocol) {
1167 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001168 std::string error;
1169 if (adb_get_feature_set(&features, &error)) {
David Pursell0aacbbe2016-01-21 19:56:50 -08001170 use_shell_protocol = CanUseFeature(features, kFeatureShell2);
1171 } else {
1172 // Device was unreachable.
1173 attempt_connection = false;
1174 }
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001175 }
David Pursell0aacbbe2016-01-21 19:56:50 -08001176
1177 if (attempt_connection) {
1178 std::string error;
1179 std::string service_string = ShellServiceString(use_shell_protocol, "", command);
1180
Josh Gao74ccdf92019-01-23 15:36:42 -08001181 fd.reset(adb_connect(service_string, &error));
David Pursell0aacbbe2016-01-21 19:56:50 -08001182 if (fd >= 0) {
1183 break;
1184 }
1185 }
1186
Felipe Leme0d4f0502016-07-26 12:14:39 -07001187 fprintf(stderr, "- waiting for device -\n");
Josh Gaob122b172017-08-16 16:57:01 -07001188 if (!wait_for_device("wait-for-device")) {
Josh Gaob6117c42016-02-24 15:16:17 -08001189 return 1;
1190 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001191 }
1192
Josh Gao74ccdf92019-01-23 15:36:42 -08001193 return read_and_dump(fd.get(), use_shell_protocol, callback);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001194}
1195
Josh Gaob122b172017-08-16 16:57:01 -07001196static int logcat(int argc, const char** argv) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001197 char* log_tags = getenv("ANDROID_LOG_TAGS");
1198 std::string quoted = escape_arg(log_tags == nullptr ? "" : log_tags);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001199
David Pursell70ef7b42015-09-30 13:35:42 -07001200 std::string cmd = "export ANDROID_LOG_TAGS=\"" + quoted + "\"; exec logcat";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001201
Jeff Sharkeyfd546e82014-06-10 11:31:24 -07001202 if (!strcmp(argv[0], "longcat")) {
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001203 cmd += " -v long";
Christopher Tatedb0a8802011-11-30 13:00:33 -08001204 }
1205
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001206 --argc;
1207 ++argv;
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001208 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001209 cmd += " " + escape_arg(*argv++);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001210 }
1211
Elliott Hughes31f0f1b2018-06-14 10:46:05 -07001212 return send_shell_command(cmd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001213}
1214
Josh Gao27241a72019-04-25 14:04:57 -07001215static void write_zeros(int bytes, borrowed_fd fd) {
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001216 int old_stdin_mode = -1;
1217 int old_stdout_mode = -1;
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001218 std::vector<char> buf(bytes);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001219
Josh Gao27241a72019-04-25 14:04:57 -07001220 D("write_zeros(%d) -> %d", bytes, fd.get());
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001221
Josh Gao27241a72019-04-25 14:04:57 -07001222 stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001223
1224 if (fd == STDOUT_FILENO) {
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001225 fwrite(buf.data(), 1, bytes, stdout);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001226 fflush(stdout);
1227 } else {
Luis Hector Chavez6150a372018-07-18 21:18:27 -07001228 adb_write(fd, buf.data(), bytes);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001229 }
1230
Josh Gao27241a72019-04-25 14:04:57 -07001231 stdinout_raw_prologue(-1, fd.get(), old_stdin_mode, old_stdout_mode);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001232
1233 D("write_zeros() finished");
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001234}
1235
Dan Albertbac34742015-02-25 17:51:28 -08001236static int backup(int argc, const char** argv) {
Al Sutton31d335f2019-05-09 16:14:05 +00001237 fprintf(stdout, "WARNING: adb backup is deprecated and may be removed in a future release\n");
1238
Elliott Hughes24f165f2015-08-21 20:31:31 -07001239 const char* filename = "backup.ab";
Christopher Tated2f54152011-04-21 12:53:28 -07001240
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001241 /* find, extract, and use any -f argument */
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001242 for (int i = 1; i < argc; i++) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001243 if (!strcmp("-f", argv[i])) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001244 if (i == argc - 1) error_exit("backup -f passed with no filename");
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001245 filename = argv[i+1];
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001246 for (int j = i+2; j <= argc; ) {
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001247 argv[i++] = argv[j++];
1248 }
1249 argc -= 2;
Yi Kongaed415c2018-07-13 18:15:16 -07001250 argv[argc] = nullptr;
Christopher Tatec9cd3b92011-06-01 17:56:23 -07001251 }
Christopher Tated2f54152011-04-21 12:53:28 -07001252 }
1253
Elliott Hughes56e68132015-11-13 11:04:10 -08001254 // Bare "adb backup" or "adb backup -f filename" are not valid invocations ---
1255 // a list of packages is required.
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001256 if (argc < 2) error_exit("backup either needs a list of packages or -all/-shared");
Christopher Tatebb86bc52011-08-22 17:12:08 -07001257
Christopher Tateb1dfffe2011-12-08 19:04:34 -08001258 adb_unlink(filename);
Josh Gao74ccdf92019-01-23 15:36:42 -08001259 unique_fd outFd(adb_creat(filename, 0640));
Christopher Tated2f54152011-04-21 12:53:28 -07001260 if (outFd < 0) {
Elliott Hughes56e68132015-11-13 11:04:10 -08001261 fprintf(stderr, "adb: backup unable to create file '%s': %s\n", filename, strerror(errno));
1262 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001263 }
1264
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001265 std::string cmd = "backup:";
1266 --argc;
1267 ++argv;
1268 while (argc-- > 0) {
1269 cmd += " " + escape_arg(*argv++);
Christopher Tated2f54152011-04-21 12:53:28 -07001270 }
1271
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001272 D("backup. filename=%s cmd=%s", filename, cmd.c_str());
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001273 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001274 unique_fd fd(adb_connect(cmd, &error));
Christopher Tated2f54152011-04-21 12:53:28 -07001275 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001276 fprintf(stderr, "adb: unable to connect for backup: %s\n", error.c_str());
Elliott Hughes56e68132015-11-13 11:04:10 -08001277 return EXIT_FAILURE;
Christopher Tated2f54152011-04-21 12:53:28 -07001278 }
1279
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001280 fprintf(stdout, "Now unlock your device and confirm the backup operation...\n");
Elliott Hughes56e68132015-11-13 11:04:10 -08001281 fflush(stdout);
1282
Josh Gao74ccdf92019-01-23 15:36:42 -08001283 copy_to_file(fd.get(), outFd.get());
Elliott Hughes56e68132015-11-13 11:04:10 -08001284 return EXIT_SUCCESS;
Christopher Tated2f54152011-04-21 12:53:28 -07001285}
1286
Dan Albertbac34742015-02-25 17:51:28 -08001287static int restore(int argc, const char** argv) {
Al Sutton31d335f2019-05-09 16:14:05 +00001288 fprintf(stdout, "WARNING: adb restore is deprecated and may be removed in a future release\n");
1289
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001290 if (argc != 2) error_exit("restore requires an argument");
Christopher Tate702967a2011-05-17 15:52:54 -07001291
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001292 const char* filename = argv[1];
Josh Gao74ccdf92019-01-23 15:36:42 -08001293 unique_fd tarFd(adb_open(filename, O_RDONLY));
Christopher Tate702967a2011-05-17 15:52:54 -07001294 if (tarFd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001295 fprintf(stderr, "adb: unable to open file %s: %s\n", filename, strerror(errno));
Christopher Tate702967a2011-05-17 15:52:54 -07001296 return -1;
1297 }
1298
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001299 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001300 unique_fd fd(adb_connect("restore:", &error));
Christopher Tate702967a2011-05-17 15:52:54 -07001301 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001302 fprintf(stderr, "adb: unable to connect for restore: %s\n", error.c_str());
Christopher Tate702967a2011-05-17 15:52:54 -07001303 return -1;
1304 }
1305
Elliott Hughes1fc8f6e2017-04-18 14:34:16 -07001306 fprintf(stdout, "Now unlock your device and confirm the restore operation.\n");
1307 fflush(stdout);
1308
Josh Gao74ccdf92019-01-23 15:36:42 -08001309 copy_to_file(tarFd.get(), fd.get());
Christopher Tate702967a2011-05-17 15:52:54 -07001310
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001311 // Provide an in-band EOD marker in case the archive file is malformed
Josh Gao74ccdf92019-01-23 15:36:42 -08001312 write_zeros(512 * 2, fd);
Christopher Tatec42f1bb2016-07-06 13:22:22 -07001313
Josh Gao5767d0d2016-03-04 15:51:03 -08001314 // Wait until the other side finishes, or it'll get sent SIGHUP.
Josh Gao74ccdf92019-01-23 15:36:42 -08001315 copy_to_file(fd.get(), STDOUT_FILENO);
Christopher Tate702967a2011-05-17 15:52:54 -07001316 return 0;
1317}
1318
Josh Gao49ba5582020-03-26 19:33:25 -07001319static CompressionType parse_compression_type(const std::string& str, bool allow_numbers) {
1320 if (allow_numbers) {
1321 if (str == "0") {
1322 return CompressionType::None;
1323 } else if (str == "1") {
1324 return CompressionType::Any;
1325 }
1326 }
1327
1328 if (str == "any") {
1329 return CompressionType::Any;
1330 } else if (str == "none") {
1331 return CompressionType::None;
1332 }
1333
1334 if (str == "brotli") {
1335 return CompressionType::Brotli;
Josh Gaoec44d352020-03-26 22:02:03 -07001336 } else if (str == "lz4") {
1337 return CompressionType::LZ4;
Josh Gao49ba5582020-03-26 19:33:25 -07001338 }
1339
1340 error_exit("unexpected compression type %s", str.c_str());
1341}
1342
Dan Albert06b0d6b2017-05-18 22:56:48 -07001343static void parse_push_pull_args(const char** arg, int narg, std::vector<const char*>* srcs,
Josh Gao49ba5582020-03-26 19:33:25 -07001344 const char** dst, bool* copy_attrs, bool* sync,
Josh Gao5949fcc2020-03-30 23:25:16 -07001345 CompressionType* compression, bool* dry_run) {
Josh Gao05786772015-10-30 16:57:19 -07001346 *copy_attrs = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001347 if (const char* adb_compression = getenv("ADB_COMPRESSION")) {
1348 *compression = parse_compression_type(adb_compression, true);
Josh Gao939fc192020-03-04 19:34:08 -08001349 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001350
Josh Gao05786772015-10-30 16:57:19 -07001351 srcs->clear();
1352 bool ignore_flags = false;
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001353 while (narg > 0) {
Josh Gao05786772015-10-30 16:57:19 -07001354 if (ignore_flags || *arg[0] != '-') {
1355 srcs->push_back(*arg);
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001356 } else {
Josh Gao05786772015-10-30 16:57:19 -07001357 if (!strcmp(*arg, "-p")) {
1358 // Silently ignore for backwards compatibility.
1359 } else if (!strcmp(*arg, "-a")) {
1360 *copy_attrs = true;
Josh Gao939fc192020-03-04 19:34:08 -08001361 } else if (!strcmp(*arg, "-z")) {
Josh Gao49ba5582020-03-26 19:33:25 -07001362 if (narg < 2) {
1363 error_exit("-z requires an argument");
Josh Gao939fc192020-03-04 19:34:08 -08001364 }
Josh Gao49ba5582020-03-26 19:33:25 -07001365 *compression = parse_compression_type(*++arg, false);
1366 --narg;
Josh Gao939fc192020-03-04 19:34:08 -08001367 } else if (!strcmp(*arg, "-Z")) {
Josh Gao49ba5582020-03-26 19:33:25 -07001368 *compression = CompressionType::None;
Josh Gao5949fcc2020-03-30 23:25:16 -07001369 } else if (dry_run && !strcmp(*arg, "-n")) {
1370 *dry_run = true;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001371 } else if (!strcmp(*arg, "--sync")) {
1372 if (sync != nullptr) {
1373 *sync = true;
1374 }
Josh Gao05786772015-10-30 16:57:19 -07001375 } else if (!strcmp(*arg, "--")) {
1376 ignore_flags = true;
1377 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001378 error_exit("unrecognized option '%s'", *arg);
Josh Gao05786772015-10-30 16:57:19 -07001379 }
Lajos Molnarde8ff4a2013-04-19 12:41:09 -07001380 }
Mark Lindner76f2a932014-03-11 17:55:59 -07001381 ++arg;
1382 --narg;
1383 }
1384
Josh Gao05786772015-10-30 16:57:19 -07001385 if (srcs->size() > 1) {
1386 *dst = srcs->back();
1387 srcs->pop_back();
Mark Lindner76f2a932014-03-11 17:55:59 -07001388 }
1389}
1390
Shukang Zhouf4ffae12020-02-13 17:01:39 -08001391static int adb_connect_command(const std::string& command, TransportId* transport,
1392 StandardStreamsCallbackInterface* callback) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001393 std::string error;
Josh Gaod10b7c42019-03-14 15:38:02 -07001394 unique_fd fd(adb_connect(transport, command, &error));
Elliott Hughes5677c232015-05-07 23:37:40 -07001395 if (fd < 0) {
1396 fprintf(stderr, "error: %s\n", error.c_str());
1397 return 1;
Tao Bao175b7bb2015-03-29 11:22:34 -07001398 }
Shukang Zhouf4ffae12020-02-13 17:01:39 -08001399 read_and_dump(fd, false, callback);
Elliott Hughes5677c232015-05-07 23:37:40 -07001400 return 0;
Tao Bao175b7bb2015-03-29 11:22:34 -07001401}
1402
Shukang Zhouf4ffae12020-02-13 17:01:39 -08001403static int adb_connect_command(const std::string& command, TransportId* transport = nullptr) {
1404 return adb_connect_command(command, transport, &DEFAULT_STANDARD_STREAMS_CALLBACK);
1405}
1406
1407// A class that prints out human readable form of the protobuf message for "track-app" service
1408// (received in binary format).
1409class TrackAppStreamsCallback : public DefaultStandardStreamsCallback {
1410 public:
1411 TrackAppStreamsCallback() : DefaultStandardStreamsCallback(nullptr, nullptr) {}
1412
1413 // Assume the buffer contains at least 4 bytes of valid data.
1414 void OnStdout(const char* buffer, int length) override {
1415 if (length < 4) return; // Unexpected length received. Do nothing.
1416
1417 adb::proto::AppProcesses binary_proto;
1418 // The first 4 bytes are the length of remaining content in hexadecimal format.
1419 binary_proto.ParseFromString(std::string(buffer + 4, length - 4));
1420 char summary[24]; // The following string includes digits and 16 fixed characters.
1421 int written = snprintf(summary, sizeof(summary), "Process count: %d\n",
1422 binary_proto.process_size());
1423 OnStream(nullptr, stdout, summary, written);
1424
1425 std::string string_proto;
1426 google::protobuf::TextFormat::PrintToString(binary_proto, &string_proto);
1427 OnStream(nullptr, stdout, string_proto.data(), string_proto.length());
1428 }
1429
1430 private:
1431 DISALLOW_COPY_AND_ASSIGN(TrackAppStreamsCallback);
1432};
1433
Josh Gaoe89a55d2018-11-16 14:47:59 -08001434static int adb_connect_command_bidirectional(const std::string& command) {
1435 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001436 unique_fd fd(adb_connect(command, &error));
Josh Gaoe89a55d2018-11-16 14:47:59 -08001437 if (fd < 0) {
1438 fprintf(stderr, "error: %s\n", error.c_str());
1439 return 1;
1440 }
1441
1442 static constexpr auto forward = [](int src, int sink, bool exit_on_end) {
1443 char buf[4096];
1444 while (true) {
1445 int rc = adb_read(src, buf, sizeof(buf));
1446 if (rc == 0) {
1447 if (exit_on_end) {
1448 exit(0);
1449 } else {
1450 adb_shutdown(sink, SHUT_WR);
1451 }
1452 return;
1453 } else if (rc < 0) {
1454 perror_exit("read failed");
1455 }
1456 if (!WriteFdExactly(sink, buf, rc)) {
1457 perror_exit("write failed");
1458 }
1459 }
1460 };
1461
Josh Gao74ccdf92019-01-23 15:36:42 -08001462 std::thread read(forward, fd.get(), STDOUT_FILENO, true);
1463 std::thread write(forward, STDIN_FILENO, fd.get(), false);
Josh Gaoe89a55d2018-11-16 14:47:59 -08001464 read.join();
1465 write.join();
Josh Gaoe89a55d2018-11-16 14:47:59 -08001466 return 0;
1467}
1468
Elliott Hughes6452a892015-04-29 12:28:13 -07001469static int adb_query_command(const std::string& command) {
1470 std::string result;
1471 std::string error;
1472 if (!adb_query(command, &result, &error)) {
1473 fprintf(stderr, "error: %s\n", error.c_str());
1474 return 1;
1475 }
1476 printf("%s\n", result.c_str());
1477 return 0;
1478}
1479
Spencer Lowa13df302015-09-07 16:20:13 -07001480// Disallow stdin, stdout, and stderr.
1481static bool _is_valid_ack_reply_fd(const int ack_reply_fd) {
1482#ifdef _WIN32
1483 const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd);
1484 return (GetStdHandle(STD_INPUT_HANDLE) != ack_reply_handle) &&
1485 (GetStdHandle(STD_OUTPUT_HANDLE) != ack_reply_handle) &&
1486 (GetStdHandle(STD_ERROR_HANDLE) != ack_reply_handle);
1487#else
1488 return ack_reply_fd > 2;
1489#endif
1490}
1491
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07001492static bool _is_valid_os_fd(int fd) {
Songchun Fan23bd5522020-03-09 11:33:44 -07001493 // Disallow invalid FDs and stdin/out/err as well.
1494 if (fd < 3) {
1495 return false;
1496 }
1497#ifdef _WIN32
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07001498 auto handle = (HANDLE)fd;
Songchun Fan23bd5522020-03-09 11:33:44 -07001499 DWORD info = 0;
1500 if (GetHandleInformation(handle, &info) == 0) {
1501 return false;
1502 }
1503#else
1504 int flags = fcntl(fd, F_GETFD);
1505 if (flags == -1) {
1506 return false;
1507 }
1508#endif
1509 return true;
1510}
1511
Elliott Hughescabfa112016-10-19 14:47:11 -07001512int adb_commandline(int argc, const char** argv) {
Josh Gaoe0323872018-08-10 14:23:43 -07001513 bool no_daemon = false;
1514 bool is_daemon = false;
1515 bool is_server = false;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001516 int r;
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001517 TransportType transport_type = kTransportAny;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001518 int ack_reply_fd = -1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001519
Elliott Hughes65433da2015-11-18 18:07:48 -08001520#if !defined(_WIN32)
1521 // We'd rather have EPIPE than SIGPIPE.
1522 signal(SIGPIPE, SIG_IGN);
1523#endif
1524
Josh Gao9c869b52016-08-25 16:00:22 -07001525 const char* server_host_str = nullptr;
1526 const char* server_port_str = nullptr;
1527 const char* server_socket_str = nullptr;
Stefan Hilzingera84a42e2010-04-19 12:21:12 +01001528
Elliott Hughes8d28e192015-10-07 14:55:10 -07001529 // We need to check for -d and -e before we look at $ANDROID_SERIAL.
1530 const char* serial = nullptr;
Josh Gaob122b172017-08-16 16:57:01 -07001531 TransportId transport_id = 0;
Elliott Hughes8d28e192015-10-07 14:55:10 -07001532
Riley Andrews98f58e82014-12-05 17:37:24 -08001533 while (argc > 0) {
Josh Gaod10b7c42019-03-14 15:38:02 -07001534 if (!strcmp(argv[0], "server")) {
Josh Gaoe0323872018-08-10 14:23:43 -07001535 is_server = true;
Josh Gaod10b7c42019-03-14 15:38:02 -07001536 } else if (!strcmp(argv[0], "nodaemon")) {
Josh Gaoe0323872018-08-10 14:23:43 -07001537 no_daemon = true;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001538 } else if (!strcmp(argv[0], "fork-server")) {
1539 /* this is a special flag used only when the ADB client launches the ADB Server */
Josh Gaoe0323872018-08-10 14:23:43 -07001540 is_daemon = true;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001541 } else if (!strcmp(argv[0], "--reply-fd")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001542 if (argc < 2) error_exit("--reply-fd requires an argument");
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001543 const char* reply_fd_str = argv[1];
1544 argc--;
1545 argv++;
1546 ack_reply_fd = strtol(reply_fd_str, nullptr, 10);
Spencer Lowa13df302015-09-07 16:20:13 -07001547 if (!_is_valid_ack_reply_fd(ack_reply_fd)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001548 fprintf(stderr, "adb: invalid reply fd \"%s\"\n", reply_fd_str);
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001549 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001550 }
Josh Gaob122b172017-08-16 16:57:01 -07001551 } else if (!strncmp(argv[0], "-s", 2)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001552 if (isdigit(argv[0][2])) {
1553 serial = argv[0] + 2;
1554 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001555 if (argc < 2 || argv[0][2] != '\0') error_exit("-s requires an argument");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001556 serial = argv[1];
1557 argc--;
1558 argv++;
1559 }
Josh Gaob122b172017-08-16 16:57:01 -07001560 } else if (!strncmp(argv[0], "-t", 2)) {
1561 const char* id;
1562 if (isdigit(argv[0][2])) {
1563 id = argv[0] + 2;
1564 } else {
1565 id = argv[1];
1566 argc--;
1567 argv++;
1568 }
1569 transport_id = strtoll(id, const_cast<char**>(&id), 10);
1570 if (*id != '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001571 error_exit("invalid transport id");
Josh Gaob122b172017-08-16 16:57:01 -07001572 }
Josh Gaod10b7c42019-03-14 15:38:02 -07001573 } else if (!strcmp(argv[0], "-d")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001574 transport_type = kTransportUsb;
Josh Gaod10b7c42019-03-14 15:38:02 -07001575 } else if (!strcmp(argv[0], "-e")) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001576 transport_type = kTransportLocal;
Josh Gaod10b7c42019-03-14 15:38:02 -07001577 } else if (!strcmp(argv[0], "-a")) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001578 gListenAll = 1;
Riley Andrews98f58e82014-12-05 17:37:24 -08001579 } else if (!strncmp(argv[0], "-H", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001580 if (argv[0][2] == '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001581 if (argc < 2) error_exit("-H requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001582 server_host_str = argv[1];
Matt Gumbeld7b33082012-11-14 10:16:17 -08001583 argc--;
1584 argv++;
1585 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001586 server_host_str = argv[0] + 2;
Matt Gumbeld7b33082012-11-14 10:16:17 -08001587 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001588 } else if (!strncmp(argv[0], "-P", 2)) {
Matt Gumbeld7b33082012-11-14 10:16:17 -08001589 if (argv[0][2] == '\0') {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001590 if (argc < 2) error_exit("-P requires an argument");
Matt Gumbeld7b33082012-11-14 10:16:17 -08001591 server_port_str = argv[1];
1592 argc--;
1593 argv++;
1594 } else {
1595 server_port_str = argv[0] + 2;
1596 }
Josh Gao9c869b52016-08-25 16:00:22 -07001597 } else if (!strcmp(argv[0], "-L")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001598 if (argc < 2) error_exit("-L requires an argument");
Josh Gao9c869b52016-08-25 16:00:22 -07001599 server_socket_str = argv[1];
1600 argc--;
1601 argv++;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001602 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001603 /* out of recognized modifiers and flags */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001604 break;
1605 }
1606 argc--;
1607 argv++;
1608 }
1609
Josh Gao9c869b52016-08-25 16:00:22 -07001610 if ((server_host_str || server_port_str) && server_socket_str) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001611 error_exit("-L is incompatible with -H or -P");
Josh Gao9c869b52016-08-25 16:00:22 -07001612 }
1613
1614 // If -L, -H, or -P are specified, ignore environment variables.
1615 // Otherwise, prefer ADB_SERVER_SOCKET over ANDROID_ADB_SERVER_ADDRESS/PORT.
Tao Wu135f4ab2016-09-16 13:01:24 -07001616 if (!server_host_str && !server_port_str && !server_socket_str) {
1617 server_socket_str = getenv("ADB_SERVER_SOCKET");
Josh Gao9c869b52016-08-25 16:00:22 -07001618 }
1619
1620 if (!server_socket_str) {
1621 // tcp:1234 and tcp:localhost:1234 are different with -a, so don't default to localhost
1622 server_host_str = server_host_str ? server_host_str : getenv("ANDROID_ADB_SERVER_ADDRESS");
1623
Tao Wu135f4ab2016-09-16 13:01:24 -07001624 int server_port = DEFAULT_ADB_PORT;
Josh Gao9c869b52016-08-25 16:00:22 -07001625 server_port_str = server_port_str ? server_port_str : getenv("ANDROID_ADB_SERVER_PORT");
Tao Wu135f4ab2016-09-16 13:01:24 -07001626 if (server_port_str && strlen(server_port_str) > 0) {
1627 if (!android::base::ParseInt(server_port_str, &server_port, 1, 65535)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001628 error_exit(
1629 "$ANDROID_ADB_SERVER_PORT must be a positive number less than 65535: "
1630 "got \"%s\"",
1631 server_port_str);
Tao Wu135f4ab2016-09-16 13:01:24 -07001632 }
1633 }
Josh Gao9c869b52016-08-25 16:00:22 -07001634
1635 int rc;
1636 char* temp;
1637 if (server_host_str) {
Tao Wu135f4ab2016-09-16 13:01:24 -07001638 rc = asprintf(&temp, "tcp:%s:%d", server_host_str, server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001639 } else {
Tao Wu135f4ab2016-09-16 13:01:24 -07001640 rc = asprintf(&temp, "tcp:%d", server_port);
Josh Gao9c869b52016-08-25 16:00:22 -07001641 }
1642 if (rc < 0) {
Elliott Hughes4679a392018-10-19 13:59:44 -07001643 LOG(FATAL) << "failed to allocate server socket specification";
Josh Gao9c869b52016-08-25 16:00:22 -07001644 }
1645 server_socket_str = temp;
1646 }
1647
1648 adb_set_socket_spec(server_socket_str);
1649
Elliott Hughes8d28e192015-10-07 14:55:10 -07001650 // If none of -d, -e, or -s were specified, try $ANDROID_SERIAL.
1651 if (transport_type == kTransportAny && serial == nullptr) {
1652 serial = getenv("ANDROID_SERIAL");
1653 }
1654
Josh Gaob122b172017-08-16 16:57:01 -07001655 adb_set_transport(transport_type, serial, transport_id);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001656
David 'Digit' Turner305b4b02011-01-31 14:23:56 +01001657 if (is_server) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001658 if (no_daemon || is_daemon) {
Spencer Low5c398d22015-08-08 15:07:07 -07001659 if (is_daemon && (ack_reply_fd == -1)) {
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001660 fprintf(stderr, "reply fd for adb server to client communication not specified.\n");
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001661 return 1;
Siva Velusamy9f2d1a92015-08-07 10:10:29 -07001662 }
Josh Gao9c869b52016-08-25 16:00:22 -07001663 r = adb_server_main(is_daemon, server_socket_str, ack_reply_fd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001664 } else {
Josh Gao9c869b52016-08-25 16:00:22 -07001665 r = launch_server(server_socket_str);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001666 }
Riley Andrews98f58e82014-12-05 17:37:24 -08001667 if (r) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001668 fprintf(stderr,"* could not start server *\n");
1669 }
1670 return r;
1671 }
1672
Riley Andrews98f58e82014-12-05 17:37:24 -08001673 if (argc == 0) {
Elliott Hughes2ec36b02017-02-06 16:20:30 -08001674 help();
1675 return 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001676 }
1677
Riley Andrewsc8514c82014-12-05 17:32:46 -08001678 /* handle wait-for-* prefix */
1679 if (!strncmp(argv[0], "wait-for-", strlen("wait-for-"))) {
Dan Albertbac34742015-02-25 17:51:28 -08001680 const char* service = argv[0];
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001681
Josh Gaob122b172017-08-16 16:57:01 -07001682 if (!wait_for_device(service)) {
Riley Andrewsc8514c82014-12-05 17:32:46 -08001683 return 1;
1684 }
1685
Elliott Hughes2b101112015-05-04 19:29:32 -07001686 // Allow a command to be run after wait-for-device,
1687 // e.g. 'adb wait-for-device shell'.
Riley Andrewsc8514c82014-12-05 17:32:46 -08001688 if (argc == 1) {
1689 return 0;
1690 }
1691
1692 /* Fall through */
1693 argc--;
1694 argv++;
1695 }
1696
1697 /* adb_connect() commands */
Riley Andrews98f58e82014-12-05 17:37:24 -08001698 if (!strcmp(argv[0], "devices")) {
Dan Albertbac34742015-02-25 17:51:28 -08001699 const char *listopt;
Elliott Hughes6452a892015-04-29 12:28:13 -07001700 if (argc < 2) {
Scott Andersone109d262012-04-20 11:21:14 -07001701 listopt = "";
Elliott Hughes6452a892015-04-29 12:28:13 -07001702 } else if (argc == 2 && !strcmp(argv[1], "-l")) {
Scott Andersone109d262012-04-20 11:21:14 -07001703 listopt = argv[1];
Elliott Hughes6452a892015-04-29 12:28:13 -07001704 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001705 error_exit("adb devices [-l]");
Scott Andersone109d262012-04-20 11:21:14 -07001706 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001707
1708 std::string query = android::base::StringPrintf("host:%s%s", argv[0], listopt);
Josh Gaod10b7c42019-03-14 15:38:02 -07001709 std::string error;
1710 if (!adb_check_server_version(&error)) {
1711 error_exit("failed to check server version: %s", error.c_str());
1712 }
Elliott Hughes6452a892015-04-29 12:28:13 -07001713 printf("List of devices attached\n");
1714 return adb_query_command(query);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001715 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001716 else if (!strcmp(argv[0], "connect")) {
Elliott Hughes65474fd2020-01-08 10:02:43 -08001717 if (argc != 2) error_exit("usage: adb connect HOST[:PORT]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001718
1719 std::string query = android::base::StringPrintf("host:connect:%s", argv[1]);
1720 return adb_query_command(query);
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -04001721 }
Riley Andrewsc8514c82014-12-05 17:32:46 -08001722 else if (!strcmp(argv[0], "disconnect")) {
Elliott Hughes573e67c2019-07-30 13:51:03 -07001723 if (argc > 2) error_exit("usage: adb disconnect [HOST[:PORT]]");
Elliott Hughes6452a892015-04-29 12:28:13 -07001724
1725 std::string query = android::base::StringPrintf("host:disconnect:%s",
1726 (argc == 2) ? argv[1] : "");
1727 return adb_query_command(query);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001728 } else if (!strcmp(argv[0], "abb")) {
1729 return adb_abb(argc, argv);
Joshua Duongd85f5c02019-11-20 14:18:43 -08001730 } else if (!strcmp(argv[0], "pair")) {
1731 if (argc != 2) error_exit("usage: adb pair <host>[:<port>]");
1732
1733 std::string password;
1734 printf("Enter pairing code: ");
1735 fflush(stdout);
1736 if (!std::getline(std::cin, password) || password.empty()) {
1737 error_exit("No pairing code provided");
1738 }
1739 std::string query =
1740 android::base::StringPrintf("host:pair:%s:%s", password.c_str(), argv[1]);
1741
1742 return adb_query_command(query);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001743 } else if (!strcmp(argv[0], "emu")) {
Spencer Low142ec752015-05-06 16:13:42 -07001744 return adb_send_emulator_command(argc, argv, serial);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001745 } else if (!strcmp(argv[0], "shell")) {
Josh Gaof3f6a1d2016-01-31 19:12:26 -08001746 return adb_shell(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001747 } else if (!strcmp(argv[0], "exec-in") || !strcmp(argv[0], "exec-out")) {
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001748 int exec_in = !strcmp(argv[0], "exec-in");
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001749
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001750 if (argc < 2) error_exit("usage: adb %s command", argv[0]);
Zach Riggleccafc792016-04-14 17:18:04 -04001751
Elliott Hughes2baae3a2015-04-17 10:59:34 -07001752 std::string cmd = "exec:";
1753 cmd += argv[1];
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001754 argc -= 2;
1755 argv += 2;
1756 while (argc-- > 0) {
Elliott Hughes6c34bba2015-04-17 20:11:08 -07001757 cmd += " " + escape_arg(*argv++);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001758 }
1759
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001760 std::string error;
Josh Gao74ccdf92019-01-23 15:36:42 -08001761 unique_fd fd(adb_connect(cmd, &error));
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001762 if (fd < 0) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07001763 fprintf(stderr, "error: %s\n", error.c_str());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001764 return -1;
1765 }
1766
1767 if (exec_in) {
Josh Gao74ccdf92019-01-23 15:36:42 -08001768 copy_to_file(STDIN_FILENO, fd.get());
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001769 } else {
Josh Gao74ccdf92019-01-23 15:36:42 -08001770 copy_to_file(fd.get(), STDOUT_FILENO);
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001771 }
Jeff Sharkey5d9d4342014-05-26 18:30:43 -07001772 return 0;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001773 } else if (!strcmp(argv[0], "kill-server")) {
Josh Gaob3c14ec2017-05-08 18:37:17 -07001774 return adb_kill_server() ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001775 } else if (!strcmp(argv[0], "sideload")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001776 if (argc != 2) error_exit("sideload requires an argument");
Tao Bao11cf30b2019-04-07 23:24:03 -07001777 if (adb_sideload_install(argv[1], false /* rescue_mode */)) {
Doug Zongker447f0612012-01-09 14:54:53 -08001778 return 1;
1779 } else {
1780 return 0;
1781 }
Tao Bao11cf30b2019-04-07 23:24:03 -07001782 } else if (!strcmp(argv[0], "rescue")) {
Tao Bao599eee12019-06-04 14:37:06 -07001783 // adb rescue getprop
Tao Bao11cf30b2019-04-07 23:24:03 -07001784 // adb rescue getprop <prop>
1785 // adb rescue install <filename>
xunchangc2265c52019-04-22 23:14:28 -07001786 // adb rescue wipe userdata
Tao Bao599eee12019-06-04 14:37:06 -07001787 if (argc < 2) error_exit("rescue requires at least one argument");
Tao Bao11cf30b2019-04-07 23:24:03 -07001788 if (!strcmp(argv[1], "getprop")) {
Tao Bao599eee12019-06-04 14:37:06 -07001789 if (argc == 2) {
1790 return adb_connect_command("rescue-getprop:");
1791 }
1792 if (argc == 3) {
1793 return adb_connect_command(
1794 android::base::StringPrintf("rescue-getprop:%s", argv[2]));
1795 }
1796 error_exit("invalid rescue getprop arguments");
Tao Bao11cf30b2019-04-07 23:24:03 -07001797 } else if (!strcmp(argv[1], "install")) {
Tao Bao599eee12019-06-04 14:37:06 -07001798 if (argc != 3) error_exit("rescue install requires two arguments");
Tao Bao11cf30b2019-04-07 23:24:03 -07001799 if (adb_sideload_install(argv[2], true /* rescue_mode */) != 0) {
1800 return 1;
1801 }
Tao Bao599eee12019-06-04 14:37:06 -07001802 } else if (!strcmp(argv[1], "wipe")) {
1803 if (argc != 3 || strcmp(argv[2], "userdata") != 0) {
1804 error_exit("invalid rescue wipe arguments");
1805 }
xunchangc2265c52019-04-22 23:14:28 -07001806 return adb_wipe_devices();
Tao Bao11cf30b2019-04-07 23:24:03 -07001807 } else {
1808 error_exit("invalid rescue argument");
1809 }
1810 return 0;
Elliott Hughese1632982017-08-23 15:42:28 -07001811 } else if (!strcmp(argv[0], "tcpip")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001812 if (argc != 2) error_exit("tcpip requires an argument");
Elliott Hughese1632982017-08-23 15:42:28 -07001813 int port;
1814 if (!android::base::ParseInt(argv[1], &port, 1, 65535)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001815 error_exit("tcpip: invalid port: %s", argv[1]);
Elliott Hughese1632982017-08-23 15:42:28 -07001816 }
1817 return adb_connect_command(android::base::StringPrintf("tcpip:%d", port));
Josh Gao8c2198c2019-07-11 14:15:32 -07001818 } else if (!strcmp(argv[0], "remount")) {
1819 FeatureSet features;
1820 std::string error;
1821 if (!adb_get_feature_set(&features, &error)) {
1822 fprintf(stderr, "error: %s\n", error.c_str());
1823 return 1;
1824 }
1825
1826 if (CanUseFeature(features, kFeatureRemountShell)) {
Mark Salyzyn6f908ce2019-09-17 08:01:15 -07001827 std::vector<const char*> args = {"shell"};
1828 args.insert(args.cend(), argv, argv + argc);
Josh Gao9e756342020-01-08 12:39:52 -08001829 return adb_shell_noinput(args.size(), args.data());
Mark Salyzyn6f908ce2019-09-17 08:01:15 -07001830 } else if (argc > 1) {
1831 auto command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
1832 return adb_connect_command(command);
Josh Gao8c2198c2019-07-11 14:15:32 -07001833 } else {
1834 return adb_connect_command("remount:");
1835 }
Elliott Hughes19d80b82015-07-21 16:13:40 -07001836 }
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001837 // clang-format off
Josh Gao8c2198c2019-07-11 14:15:32 -07001838 else if (!strcmp(argv[0], "reboot") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001839 !strcmp(argv[0], "reboot-bootloader") ||
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001840 !strcmp(argv[0], "reboot-fastboot") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001841 !strcmp(argv[0], "usb") ||
Riley Andrewsc8514c82014-12-05 17:32:46 -08001842 !strcmp(argv[0], "disable-verity") ||
1843 !strcmp(argv[0], "enable-verity")) {
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001844 // clang-format on
Elliott Hughes5677c232015-05-07 23:37:40 -07001845 std::string command;
Tao Bao175b7bb2015-03-29 11:22:34 -07001846 if (!strcmp(argv[0], "reboot-bootloader")) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001847 command = "reboot:bootloader";
Mark Salyzynd8e94bd2018-08-29 10:44:33 -07001848 } else if (!strcmp(argv[0], "reboot-fastboot")) {
1849 command = "reboot:fastboot";
Tao Bao175b7bb2015-03-29 11:22:34 -07001850 } else if (argc > 1) {
Elliott Hughes5677c232015-05-07 23:37:40 -07001851 command = android::base::StringPrintf("%s:%s", argv[0], argv[1]);
Tao Bao175b7bb2015-03-29 11:22:34 -07001852 } else {
Elliott Hughes5677c232015-05-07 23:37:40 -07001853 command = android::base::StringPrintf("%s:", argv[0]);
The Android Open Source Projecte037fd72009-03-13 13:04:37 -07001854 }
Tao Bao175b7bb2015-03-29 11:22:34 -07001855 return adb_connect_command(command);
Josh Gao06c73ae2016-03-04 15:16:18 -08001856 } else if (!strcmp(argv[0], "root") || !strcmp(argv[0], "unroot")) {
1857 return adb_root(argv[0]) ? 0 : 1;
1858 } else if (!strcmp(argv[0], "bugreport")) {
Felipe Leme698e0652016-07-19 17:07:22 -07001859 Bugreport bugreport;
Josh Gaob122b172017-08-16 16:57:01 -07001860 return bugreport.DoIt(argc, argv);
Felipe Leme44a42672016-04-01 17:43:27 -07001861 } else if (!strcmp(argv[0], "forward") || !strcmp(argv[0], "reverse")) {
Elliott Hughes424af022015-05-29 17:55:19 -07001862 bool reverse = !strcmp(argv[0], "reverse");
Elliott Hughes424af022015-05-29 17:55:19 -07001863 --argc;
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001864 if (argc < 1) error_exit("%s requires an argument", argv[0]);
Josh Gao258d4a92018-07-31 15:36:03 -07001865 ++argv;
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001866
1867 // Determine the <host-prefix> for this command.
Elliott Hughes424af022015-05-29 17:55:19 -07001868 std::string host_prefix;
David 'Digit' Turner25258692013-03-21 21:07:42 +01001869 if (reverse) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001870 host_prefix = "reverse:";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001871 } else {
Josh Gaoda9c7672019-08-06 16:43:34 -07001872 host_prefix = "host:";
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001873 }
1874
Elliott Hughes4679a392018-10-19 13:59:44 -07001875 std::string cmd, error_message;
Elliott Hughes424af022015-05-29 17:55:19 -07001876 if (strcmp(argv[0], "--list") == 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001877 if (argc != 1) error_exit("--list doesn't take any arguments");
Josh Gaoda9c7672019-08-06 16:43:34 -07001878 return adb_query_command(host_prefix + "list-forward");
Elliott Hughes424af022015-05-29 17:55:19 -07001879 } else if (strcmp(argv[0], "--remove-all") == 0) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001880 if (argc != 1) error_exit("--remove-all doesn't take any arguments");
Josh Gaoda9c7672019-08-06 16:43:34 -07001881 cmd = "killforward-all";
Elliott Hughes424af022015-05-29 17:55:19 -07001882 } else if (strcmp(argv[0], "--remove") == 0) {
1883 // forward --remove <local>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001884 if (argc != 2) error_exit("--remove requires an argument");
Josh Gaoda9c7672019-08-06 16:43:34 -07001885 cmd = std::string("killforward:") + argv[1];
Elliott Hughes424af022015-05-29 17:55:19 -07001886 } else if (strcmp(argv[0], "--no-rebind") == 0) {
1887 // forward --no-rebind <local> <remote>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001888 if (argc != 3) error_exit("--no-rebind takes two arguments");
Elliott Hughes4679a392018-10-19 13:59:44 -07001889 if (forward_targets_are_valid(argv[1], argv[2], &error_message)) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001890 cmd = std::string("forward:norebind:") + argv[1] + ";" + argv[2];
David Purselleaae97e2016-04-07 11:25:48 -07001891 }
Elliott Hughes424af022015-05-29 17:55:19 -07001892 } else {
1893 // forward <local> <remote>
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001894 if (argc != 2) error_exit("forward takes two arguments");
Elliott Hughes4679a392018-10-19 13:59:44 -07001895 if (forward_targets_are_valid(argv[0], argv[1], &error_message)) {
Josh Gaoda9c7672019-08-06 16:43:34 -07001896 cmd = std::string("forward:") + argv[0] + ";" + argv[1];
David Purselleaae97e2016-04-07 11:25:48 -07001897 }
David 'Digit' Turner0d82fbf2012-11-14 15:01:55 +01001898 }
1899
Elliott Hughes4679a392018-10-19 13:59:44 -07001900 if (!error_message.empty()) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001901 error_exit("error: %s", error_message.c_str());
David Purselleaae97e2016-04-07 11:25:48 -07001902 }
1903
Josh Gaoda9c7672019-08-06 16:43:34 -07001904 unique_fd fd(adb_connect(nullptr, host_prefix + cmd, &error_message, true));
Josh Gao74ccdf92019-01-23 15:36:42 -08001905 if (fd < 0 || !adb_status(fd.get(), &error_message)) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001906 error_exit("error: %s", error_message.c_str());
David Purselleaae97e2016-04-07 11:25:48 -07001907 }
1908
1909 // Server or device may optionally return a resolved TCP port number.
1910 std::string resolved_port;
Elliott Hughes4679a392018-10-19 13:59:44 -07001911 if (ReadProtocolString(fd, &resolved_port, &error_message) && !resolved_port.empty()) {
David Purselleaae97e2016-04-07 11:25:48 -07001912 printf("%s\n", resolved_port.c_str());
1913 }
1914
1915 ReadOrderlyShutdown(fd);
1916 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001917 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001918 /* do_sync_*() commands */
Riley Andrewsc8514c82014-12-05 17:32:46 -08001919 else if (!strcmp(argv[0], "ls")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001920 if (argc != 2) error_exit("ls requires an argument");
Elliott Hughesaa245492015-08-03 10:38:08 -07001921 return do_sync_ls(argv[1]) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001922 } else if (!strcmp(argv[0], "push")) {
Josh Gao05786772015-10-30 16:57:19 -07001923 bool copy_attrs = false;
Dan Albert06b0d6b2017-05-18 22:56:48 -07001924 bool sync = false;
Josh Gao5949fcc2020-03-30 23:25:16 -07001925 bool dry_run = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001926 CompressionType compression = CompressionType::Any;
Josh Gao05786772015-10-30 16:57:19 -07001927 std::vector<const char*> srcs;
1928 const char* dst = nullptr;
Mark Lindner76f2a932014-03-11 17:55:59 -07001929
Josh Gao5949fcc2020-03-30 23:25:16 -07001930 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, &sync, &compression,
1931 &dry_run);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001932 if (srcs.empty() || !dst) error_exit("push requires an argument");
Josh Gao5949fcc2020-03-30 23:25:16 -07001933 return do_sync_push(srcs, dst, sync, compression, dry_run) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001934 } else if (!strcmp(argv[0], "pull")) {
Josh Gao05786772015-10-30 16:57:19 -07001935 bool copy_attrs = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001936 CompressionType compression = CompressionType::Any;
Josh Gao05786772015-10-30 16:57:19 -07001937 std::vector<const char*> srcs;
1938 const char* dst = ".";
Mark Lindner76f2a932014-03-11 17:55:59 -07001939
Josh Gao5949fcc2020-03-30 23:25:16 -07001940 parse_push_pull_args(&argv[1], argc - 1, &srcs, &dst, &copy_attrs, nullptr, &compression,
1941 nullptr);
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001942 if (srcs.empty()) error_exit("pull requires an argument");
Josh Gao49ba5582020-03-26 19:33:25 -07001943 return do_sync_pull(srcs, dst, copy_attrs, compression) ? 0 : 1;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001944 } else if (!strcmp(argv[0], "install")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001945 if (argc < 2) error_exit("install requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001946 return install_app(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001947 } else if (!strcmp(argv[0], "install-multiple")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001948 if (argc < 2) error_exit("install-multiple requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001949 return install_multiple_app(argc, argv);
Patrick Baumanndc58b0a2018-10-09 10:45:03 -07001950 } else if (!strcmp(argv[0], "install-multi-package")) {
Mohammad Samiul Islamb581ce92019-05-29 15:00:00 +01001951 if (argc < 2) error_exit("install-multi-package requires an argument");
Patrick Baumanndc58b0a2018-10-09 10:45:03 -07001952 return install_multi_package(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001953 } else if (!strcmp(argv[0], "uninstall")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07001954 if (argc < 2) error_exit("uninstall requires an argument");
Josh Gaob122b172017-08-16 16:57:01 -07001955 return uninstall_app(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08001956 } else if (!strcmp(argv[0], "sync")) {
Elliott Hughesd236d072015-04-21 10:17:07 -07001957 std::string src;
Elliott Hughes58305772015-04-17 13:57:15 -07001958 bool list_only = false;
Josh Gao5949fcc2020-03-30 23:25:16 -07001959 bool dry_run = false;
Josh Gao49ba5582020-03-26 19:33:25 -07001960 CompressionType compression = CompressionType::Any;
Josh Gao939fc192020-03-04 19:34:08 -08001961
Josh Gao49ba5582020-03-26 19:33:25 -07001962 if (const char* adb_compression = getenv("ADB_COMPRESSION"); adb_compression) {
1963 compression = parse_compression_type(adb_compression, true);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001964 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001965
Josh Gao939fc192020-03-04 19:34:08 -08001966 int opt;
Josh Gao5949fcc2020-03-30 23:25:16 -07001967 while ((opt = getopt(argc, const_cast<char**>(argv), "lnz:Z")) != -1) {
Josh Gao939fc192020-03-04 19:34:08 -08001968 switch (opt) {
1969 case 'l':
1970 list_only = true;
1971 break;
Josh Gao5949fcc2020-03-30 23:25:16 -07001972 case 'n':
1973 dry_run = true;
1974 break;
Josh Gao939fc192020-03-04 19:34:08 -08001975 case 'z':
Josh Gao49ba5582020-03-26 19:33:25 -07001976 compression = parse_compression_type(optarg, false);
Josh Gao939fc192020-03-04 19:34:08 -08001977 break;
1978 case 'Z':
Josh Gao49ba5582020-03-26 19:33:25 -07001979 compression = CompressionType::None;
Josh Gao939fc192020-03-04 19:34:08 -08001980 break;
1981 default:
Josh Gao5949fcc2020-03-30 23:25:16 -07001982 error_exit("usage: adb sync [-l] [-n] [-z ALGORITHM] [-Z] [PARTITION]");
Josh Gao939fc192020-03-04 19:34:08 -08001983 }
1984 }
1985
1986 if (optind == argc) {
1987 src = "all";
1988 } else if (optind + 1 == argc) {
1989 src = argv[optind];
1990 } else {
Josh Gao5949fcc2020-03-30 23:25:16 -07001991 error_exit("usage: adb sync [-l] [-n] [-z ALGORITHM] [-Z] [PARTITION]");
Josh Gao939fc192020-03-04 19:34:08 -08001992 }
1993
Justin Yun5d0ac622019-06-26 09:18:03 +09001994 std::vector<std::string> partitions{"data", "odm", "oem", "product",
1995 "system", "system_ext", "vendor"};
Elliott Hughes57e97cb2018-04-03 10:54:39 -07001996 bool found = false;
1997 for (const auto& partition : partitions) {
1998 if (src == "all" || src == partition) {
1999 std::string src_dir{product_file(partition)};
2000 if (!directory_exists(src_dir)) continue;
2001 found = true;
Josh Gao5949fcc2020-03-30 23:25:16 -07002002 if (!do_sync_sync(src_dir, "/" + partition, list_only, compression, dry_run)) {
2003 return 1;
2004 }
Elliott Hughes57e97cb2018-04-03 10:54:39 -07002005 }
Elliott Hughes58305772015-04-17 13:57:15 -07002006 }
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002007 if (!found) error_exit("don't know how to sync %s partition", src.c_str());
Elliott Hughes4679a392018-10-19 13:59:44 -07002008 return 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002009 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002010 /* passthrough commands */
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002011 else if (!strcmp(argv[0], "get-state") || !strcmp(argv[0], "get-serialno") ||
2012 !strcmp(argv[0], "get-devpath")) {
Josh Gaob122b172017-08-16 16:57:01 -07002013 return adb_query_command(format_host_command(argv[0]));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002014 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002015 /* other commands */
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002016 else if (!strcmp(argv[0], "logcat") || !strcmp(argv[0], "lolcat") ||
2017 !strcmp(argv[0], "longcat")) {
Josh Gaob122b172017-08-16 16:57:01 -07002018 return logcat(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002019 } else if (!strcmp(argv[0], "ppp")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002020 return ppp(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002021 } else if (!strcmp(argv[0], "start-server")) {
Elliott Hughes078f0fc2015-04-29 08:35:59 -07002022 std::string error;
Spencer Lowf18fc082015-08-11 17:05:02 -07002023 const int result = adb_connect("host:start-server", &error);
2024 if (result < 0) {
2025 fprintf(stderr, "error: %s\n", error.c_str());
2026 }
2027 return result;
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002028 } else if (!strcmp(argv[0], "backup")) {
Christopher Tated2f54152011-04-21 12:53:28 -07002029 return backup(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002030 } else if (!strcmp(argv[0], "restore")) {
Christopher Tate702967a2011-05-17 15:52:54 -07002031 return restore(argc, argv);
Alex Buynytskyy640407d2018-12-12 10:48:50 -08002032 } else if (!strcmp(argv[0], "keygen")) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002033 if (argc != 2) error_exit("keygen requires an argument");
Yabin Cuiaed3c612015-09-22 15:52:57 -07002034 // Always print key generation information for keygen command.
2035 adb_trace_enable(AUTH);
Nick Kralevichbea3f9c2014-11-13 15:17:29 -08002036 return adb_auth_keygen(argv[1]);
Josh Gao2dc4cab2018-11-15 17:45:46 -08002037 } else if (!strcmp(argv[0], "pubkey")) {
2038 if (argc != 2) error_exit("pubkey requires an argument");
2039 return adb_auth_pubkey(argv[1]);
2040 } else if (!strcmp(argv[0], "jdwp")) {
Tao Bao175b7bb2015-03-29 11:22:34 -07002041 return adb_connect_command("jdwp");
Josh Gao2dc4cab2018-11-15 17:45:46 -08002042 } else if (!strcmp(argv[0], "track-jdwp")) {
Josh Gaob463baf2016-05-13 18:18:23 -07002043 return adb_connect_command("track-jdwp");
Shukang Zhouf4ffae12020-02-13 17:01:39 -08002044 } else if (!strcmp(argv[0], "track-app")) {
2045 FeatureSet features;
2046 std::string error;
2047 if (!adb_get_feature_set(&features, &error)) {
2048 fprintf(stderr, "error: %s\n", error.c_str());
2049 return 1;
2050 }
2051 if (!CanUseFeature(features, kFeatureTrackApp)) {
2052 error_exit("track-app is not supported by the device");
2053 }
2054 TrackAppStreamsCallback callback;
2055 return adb_connect_command("track-app", nullptr, &callback);
Josh Gao2dc4cab2018-11-15 17:45:46 -08002056 } else if (!strcmp(argv[0], "track-devices")) {
Elliott Hughesfc2f5f62019-07-31 14:13:57 -07002057 if (argc > 2 || (argc == 2 && strcmp(argv[1], "-l"))) {
2058 error_exit("usage: adb track-devices [-l]");
2059 }
2060 return adb_connect_command(argc == 2 ? "host:track-devices-l" : "host:track-devices");
Josh Gao172d0d42018-06-18 14:51:56 -07002061 } else if (!strcmp(argv[0], "raw")) {
2062 if (argc != 2) {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002063 error_exit("usage: adb raw SERVICE");
Josh Gao172d0d42018-06-18 14:51:56 -07002064 }
Josh Gaoe89a55d2018-11-16 14:47:59 -08002065 return adb_connect_command_bidirectional(argv[1]);
Josh Gaob463baf2016-05-13 18:18:23 -07002066 }
2067
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002068 /* "adb /?" is a common idiom under Windows */
Elliott Hughes2ec36b02017-02-06 16:20:30 -08002069 else if (!strcmp(argv[0], "--help") || !strcmp(argv[0], "help") || !strcmp(argv[0], "/?")) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002070 help();
2071 return 0;
Josh Gao2dc4cab2018-11-15 17:45:46 -08002072 } else if (!strcmp(argv[0], "--version") || !strcmp(argv[0], "version")) {
Elliott Hughes42ae2602015-08-12 08:32:10 -07002073 fprintf(stdout, "%s", adb_version().c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002074 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08002075 } else if (!strcmp(argv[0], "features")) {
David Pursell4e2fd362015-09-22 10:43:08 -07002076 // Only list the features common to both the adb client and the device.
David Pursell0aacbbe2016-01-21 19:56:50 -08002077 FeatureSet features;
Josh Gaof3f6a1d2016-01-31 19:12:26 -08002078 std::string error;
2079 if (!adb_get_feature_set(&features, &error)) {
2080 fprintf(stderr, "error: %s\n", error.c_str());
David Pursell0aacbbe2016-01-21 19:56:50 -08002081 return 1;
2082 }
2083
David Pursell4e2fd362015-09-22 10:43:08 -07002084 for (const std::string& name : features) {
David Pursell70ef7b42015-09-30 13:35:42 -07002085 if (CanUseFeature(features, name)) {
David Pursell4e2fd362015-09-22 10:43:08 -07002086 printf("%s\n", name.c_str());
2087 }
2088 }
2089 return 0;
Josh Gao5d1756c2017-02-22 17:07:01 -08002090 } else if (!strcmp(argv[0], "host-features")) {
2091 return adb_query_command("host:host-features");
Yabin Cui1f4ec192016-04-05 13:50:44 -07002092 } else if (!strcmp(argv[0], "reconnect")) {
2093 if (argc == 1) {
Josh Gaob122b172017-08-16 16:57:01 -07002094 return adb_query_command(format_host_command(argv[0]));
Josh Gao22d2b3e2016-10-27 14:01:08 -07002095 } else if (argc == 2) {
2096 if (!strcmp(argv[1], "device")) {
2097 std::string err;
2098 adb_connect("reconnect", &err);
2099 return 0;
2100 } else if (!strcmp(argv[1], "offline")) {
2101 std::string err;
2102 return adb_query_command("host:reconnect-offline");
2103 } else {
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002104 error_exit("usage: adb reconnect [device|offline]");
Josh Gao22d2b3e2016-10-27 14:01:08 -07002105 }
Yabin Cui1f4ec192016-04-05 13:50:44 -07002106 }
Alex Buynytskyy96ff54b2020-02-13 06:52:04 -08002107 } else if (!strcmp(argv[0], "inc-server")) {
Songchun Fan23bd5522020-03-09 11:33:44 -07002108 if (argc < 4) {
2109#ifdef _WIN32
2110 error_exit("usage: adb inc-server CONNECTION_HANDLE OUTPUT_HANDLE FILE1 FILE2 ...");
2111#else
2112 error_exit("usage: adb inc-server CONNECTION_FD OUTPUT_FD FILE1 FILE2 ...");
2113#endif
Alex Buynytskyy96ff54b2020-02-13 06:52:04 -08002114 }
Songchun Fan23bd5522020-03-09 11:33:44 -07002115 int connection_fd = atoi(argv[1]);
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07002116 if (!_is_valid_os_fd(connection_fd)) {
Songchun Fan23bd5522020-03-09 11:33:44 -07002117 error_exit("Invalid connection_fd number given: %d", connection_fd);
Alex Buynytskyy96ff54b2020-02-13 06:52:04 -08002118 }
Songchun Fan23bd5522020-03-09 11:33:44 -07002119
2120 connection_fd = adb_register_socket(connection_fd);
2121 close_on_exec(connection_fd);
2122
2123 int output_fd = atoi(argv[2]);
Yurii Zubrytskyi2488c212020-03-18 15:49:45 -07002124 if (!_is_valid_os_fd(output_fd)) {
Songchun Fan23bd5522020-03-09 11:33:44 -07002125 error_exit("Invalid output_fd number given: %d", output_fd);
2126 }
2127 output_fd = adb_register_socket(output_fd);
2128 close_on_exec(output_fd);
2129 return incremental::serve(connection_fd, output_fd, argc - 3, argv + 3);
Dan Albert90d4b732015-05-20 18:58:41 -07002130 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002131
Elliott Hughesd2aaede2018-10-22 17:02:51 -07002132 error_exit("unknown command %s", argv[0]);
Elliott Hughes4679a392018-10-19 13:59:44 -07002133 __builtin_unreachable();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08002134}