blob: f397da09fbd467f7160d851ed998ff474958211f [file] [log] [blame]
The Android Open Source Project9ca14dc2009-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 Cui19bec5b2015-09-22 15:52:57 -070017#define TRACE_TAG ADB
Dan Albertdb6fe642015-03-19 15:21:08 -070018
19#include "sysdeps.h"
20#include "adb.h"
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080021
Dan Albertecbd8742015-03-08 21:12:08 -070022#include <ctype.h>
23#include <errno.h>
24#include <stdarg.h>
25#include <stddef.h>
26#include <stdint.h>
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080027#include <stdio.h>
28#include <stdlib.h>
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080029#include <string.h>
Mike Lockwood4ac1e282009-05-25 18:17:55 -040030#include <sys/time.h>
Dan Albertecbd8742015-03-08 21:12:08 -070031#include <time.h>
Bradley Furmand8f12392022-04-11 16:36:33 +000032#include <unistd.h>
Dan Albertecbd8742015-03-08 21:12:08 -070033
Elliott Hughes73925982016-11-15 12:37:32 -080034#include <chrono>
Josh Gao1e3bf732017-05-03 22:37:10 -070035#include <condition_variable>
36#include <mutex>
Dan Albertecbd8742015-03-08 21:12:08 -070037#include <string>
Josh Gaobd775212020-02-26 16:39:20 -080038#include <string_view>
Elliott Hughes73925982016-11-15 12:37:32 -080039#include <thread>
Dan Albert27929b02015-03-19 13:25:27 -070040#include <vector>
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080041
David Pursellc573d522016-01-27 08:52:53 -080042#include <android-base/errors.h>
Elliott Hughes827a4a52016-08-31 15:07:18 -070043#include <android-base/file.h>
Elliott Hughesf55ead92015-12-04 22:00:26 -080044#include <android-base/logging.h>
45#include <android-base/macros.h>
David Pursella17d2722016-01-21 08:40:59 -080046#include <android-base/parsenetaddress.h>
Elliott Hughesf55ead92015-12-04 22:00:26 -080047#include <android-base/stringprintf.h>
48#include <android-base/strings.h>
Josh Gao6256d992021-05-20 18:28:13 -070049#include <diagnose_usb.h>
50
Josh Gao66766f82018-02-27 15:49:23 -080051#include <build/version.h>
Dan Willemsen18861892018-08-29 14:58:02 -070052#include <platform_tools_version.h>
Elliott Hughesa585cbd2015-04-20 08:09:20 -070053
Benoit Goby2cc19e42012-04-12 12:23:49 -070054#include "adb_auth.h"
Dan Albert66a91b02015-02-24 21:26:58 -080055#include "adb_io.h"
Dan Albert020292b2015-02-18 18:03:26 -080056#include "adb_listeners.h"
Joshua Duongf4ba8d72021-01-13 12:18:15 -080057#include "adb_mdns.h"
Elliott Hughescd61ae32017-06-15 08:35:24 -070058#include "adb_unique_fd.h"
Elliott Hughes09ccf1f2015-07-18 12:21:30 -070059#include "adb_utils.h"
Josh Gao1e3bf732017-05-03 22:37:10 -070060#include "sysdeps/chrono.h"
Dan Albertb302d122015-02-24 15:51:19 -080061#include "transport.h"
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080062
63#if !ADB_HOST
Nick Kralevich9de01bd2013-02-28 14:12:58 -080064#include <sys/capability.h>
Jeff Sharkey11987252012-08-14 21:00:22 -070065#include <sys/mount.h>
Elliott Hughes8b249d22016-09-23 15:40:03 -070066#include <android-base/properties.h>
Elliott Hughes73925982016-11-15 12:37:32 -080067using namespace std::chrono_literals;
Josh Gaobd775212020-02-26 16:39:20 -080068
69#include "daemon/logging.h"
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080070#endif
71
Josh Gao6b55e752020-03-27 18:09:56 -070072#if ADB_HOST
73#include "client/usb.h"
74#endif
75
Josh Gao2dc61962021-02-03 22:11:45 -080076#if !ADB_HOST && defined(__ANDROID__)
77#include "daemon/watchdog.h"
78
79static std::atomic<int> active_connections = 0;
80
81static void IncrementActiveConnections() {
82 if (active_connections++ == 0) {
83 watchdog::Stop();
84 }
85}
86
87static void DecrementActiveConnections() {
88 if (--active_connections == 0) {
89 watchdog::Start();
90 }
91}
92
93#endif
94
Elliott Hughesb9f01642015-08-12 08:32:10 -070095std::string adb_version() {
96 // Don't change the format of this --- it's parsed by ddmlib.
Elliott Hughes4e97fd62017-03-30 15:08:28 -070097 return android::base::StringPrintf(
98 "Android Debug Bridge version %d.%d.%d\n"
Dan Willemsen18861892018-08-29 14:58:02 -070099 "Version %s-%s\n"
Elliott Hughes4e97fd62017-03-30 15:08:28 -0700100 "Installed as %s\n",
Josh Gao361148b2018-01-02 12:01:43 -0800101 ADB_VERSION_MAJOR, ADB_VERSION_MINOR, ADB_SERVER_VERSION,
Dan Willemsen18861892018-08-29 14:58:02 -0700102 PLATFORM_TOOLS_VERSION, android::build::GetBuildNumber().c_str(),
103 android::base::GetExecutablePath().c_str());
Elliott Hughesb9f01642015-08-12 08:32:10 -0700104}
105
Josh Gao67ac3792016-10-06 13:31:44 -0700106uint32_t calculate_apacket_checksum(const apacket* p) {
Josh Gao67ac3792016-10-06 13:31:44 -0700107 uint32_t sum = 0;
Josh Gao839b9322018-02-05 18:49:10 -0800108 for (size_t i = 0; i < p->msg.data_length; ++i) {
109 sum += static_cast<uint8_t>(p->payload[i]);
Josh Gao67ac3792016-10-06 13:31:44 -0700110 }
Josh Gao67ac3792016-10-06 13:31:44 -0700111 return sum;
112}
113
Josh Gao6256d992021-05-20 18:28:13 -0700114std::string to_string(ConnectionState state) {
115 switch (state) {
116 case kCsOffline:
117 return "offline";
118 case kCsBootloader:
119 return "bootloader";
120 case kCsDevice:
121 return "device";
122 case kCsHost:
123 return "host";
124 case kCsRecovery:
125 return "recovery";
126 case kCsRescue:
127 return "rescue";
128 case kCsNoPerm:
129 return UsbNoPermissionsShortHelpText();
130 case kCsSideload:
131 return "sideload";
132 case kCsUnauthorized:
133 return "unauthorized";
134 case kCsAuthorizing:
135 return "authorizing";
136 case kCsConnecting:
137 return "connecting";
138 default:
139 return "unknown";
140 }
141}
142
143apacket* get_apacket(void) {
Josh Gao839b9322018-02-05 18:49:10 -0800144 apacket* p = new apacket();
Dan Albertf30d73c2015-02-25 17:51:28 -0800145 if (p == nullptr) {
Elliott Hughese64126b2018-10-19 13:59:44 -0700146 LOG(FATAL) << "failed to allocate an apacket";
Dan Albertf30d73c2015-02-25 17:51:28 -0800147 }
148
Josh Gao839b9322018-02-05 18:49:10 -0800149 memset(&p->msg, 0, sizeof(p->msg));
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800150 return p;
151}
152
153void put_apacket(apacket *p)
154{
Josh Gao839b9322018-02-05 18:49:10 -0800155 delete p;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800156}
157
Benoit Goby2cc19e42012-04-12 12:23:49 -0700158void handle_online(atransport *t)
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800159{
Yabin Cui815ad882015-09-02 17:44:28 -0700160 D("adb: online");
Benoit Goby2cc19e42012-04-12 12:23:49 -0700161 t->online = 1;
Josh Gao65d18e22020-04-22 20:57:26 -0700162#if ADB_HOST
Luis Hector Chavezda74b902018-04-17 14:25:04 -0700163 t->SetConnectionEstablished(true);
Josh Gao2dc61962021-02-03 22:11:45 -0800164#elif defined(__ANDROID__)
165 IncrementActiveConnections();
Josh Gao65d18e22020-04-22 20:57:26 -0700166#endif
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800167}
168
169void handle_offline(atransport *t)
170{
Josh Gao715fe602018-02-16 13:24:58 -0800171 if (t->GetConnectionState() == kCsOffline) {
172 LOG(INFO) << t->serial_name() << ": already offline";
173 return;
174 }
175
176 LOG(INFO) << t->serial_name() << ": offline";
177
Josh Gao2dc61962021-02-03 22:11:45 -0800178#if !ADB_HOST && defined(__ANDROID__)
179 DecrementActiveConnections();
180#endif
181
Josh Gao715fe602018-02-16 13:24:58 -0800182 t->SetConnectionState(kCsOffline);
183
184 // Close the associated usb
Benoit Goby2cc19e42012-04-12 12:23:49 -0700185 t->online = 0;
Yabin Cui10dedd82015-08-26 11:18:42 -0700186
Yabin Cui2d4c1982015-08-28 15:09:44 -0700187 // This is necessary to avoid a race condition that occurred when a transport closes
Yabin Cui10dedd82015-08-26 11:18:42 -0700188 // while a client socket is still active.
189 close_all_sockets(t);
190
Yabin Cui2d4c1982015-08-28 15:09:44 -0700191 t->RunDisconnects();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800192}
193
Benoit Goby2cc19e42012-04-12 12:23:49 -0700194#if DEBUG_PACKETS
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800195#define DUMPMAX 32
196void print_packet(const char *label, apacket *p)
197{
Josh Gao839b9322018-02-05 18:49:10 -0800198 const char* tag;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800199 unsigned count;
200
201 switch(p->msg.command){
202 case A_SYNC: tag = "SYNC"; break;
203 case A_CNXN: tag = "CNXN" ; break;
204 case A_OPEN: tag = "OPEN"; break;
205 case A_OKAY: tag = "OKAY"; break;
206 case A_CLSE: tag = "CLSE"; break;
207 case A_WRTE: tag = "WRTE"; break;
Benoit Goby2cc19e42012-04-12 12:23:49 -0700208 case A_AUTH: tag = "AUTH"; break;
Joshua Duong64fab752020-01-21 13:19:42 -0800209 case A_STLS:
Joshua Duong77b8ff32020-04-16 15:58:19 -0700210 tag = "STLS";
Joshua Duong64fab752020-01-21 13:19:42 -0800211 break;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800212 default: tag = "????"; break;
213 }
214
215 fprintf(stderr, "%s: %s %08x %08x %04x \"",
216 label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
217 count = p->msg.data_length;
Josh Gao839b9322018-02-05 18:49:10 -0800218 const char* x = p->payload.data();
219 if (count > DUMPMAX) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800220 count = DUMPMAX;
221 tag = "\n";
222 } else {
223 tag = "\"\n";
224 }
Josh Gao839b9322018-02-05 18:49:10 -0800225 while (count-- > 0) {
226 if ((*x >= ' ') && (*x < 127)) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800227 fputc(*x, stderr);
228 } else {
229 fputc('.', stderr);
230 }
231 x++;
232 }
Benoit Goby2cc19e42012-04-12 12:23:49 -0700233 fputs(tag, stderr);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800234}
235#endif
236
Josh Gaobd1f3002021-12-29 21:15:12 -0800237void send_ready(unsigned local, unsigned remote, atransport* t, uint32_t ack_bytes) {
Yabin Cui815ad882015-09-02 17:44:28 -0700238 D("Calling send_ready");
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800239 apacket *p = get_apacket();
240 p->msg.command = A_OKAY;
241 p->msg.arg0 = local;
242 p->msg.arg1 = remote;
Josh Gaobd1f3002021-12-29 21:15:12 -0800243 if (t->SupportsDelayedAck()) {
244 p->msg.data_length = sizeof(ack_bytes);
245 p->payload.resize(sizeof(ack_bytes));
246 memcpy(p->payload.data(), &ack_bytes, sizeof(ack_bytes));
247 }
248
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800249 send_packet(p, t);
250}
251
252static void send_close(unsigned local, unsigned remote, atransport *t)
253{
Yabin Cui815ad882015-09-02 17:44:28 -0700254 D("Calling send_close");
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800255 apacket *p = get_apacket();
256 p->msg.command = A_CLSE;
257 p->msg.arg0 = local;
258 p->msg.arg1 = remote;
259 send_packet(p, t);
260}
261
Dan Albertbe8e54b2015-05-18 13:06:53 -0700262std::string get_connection_string() {
263 std::vector<std::string> connection_properties;
264
265#if !ADB_HOST
266 static const char* cnxn_props[] = {
Scott Andersonfa020922012-05-25 14:10:02 -0700267 "ro.product.name",
268 "ro.product.model",
269 "ro.product.device",
270 };
Scott Andersonfa020922012-05-25 14:10:02 -0700271
Elliott Hughes8b249d22016-09-23 15:40:03 -0700272 for (const auto& prop : cnxn_props) {
273 std::string value = std::string(prop) + "=" + android::base::GetProperty(prop, "");
274 connection_properties.push_back(value);
Scott Andersonfa020922012-05-25 14:10:02 -0700275 }
Scott Andersonfa020922012-05-25 14:10:02 -0700276#endif
Dan Albertbe8e54b2015-05-18 13:06:53 -0700277
278 connection_properties.push_back(android::base::StringPrintf(
David Pursella07dbad2015-09-22 10:43:08 -0700279 "features=%s", FeatureSetToString(supported_features()).c_str()));
Dan Albertbe8e54b2015-05-18 13:06:53 -0700280
281 return android::base::StringPrintf(
282 "%s::%s", adb_device_banner,
283 android::base::Join(connection_properties, ';').c_str());
Scott Andersonfa020922012-05-25 14:10:02 -0700284}
285
Joshua Duong64fab752020-01-21 13:19:42 -0800286void send_tls_request(atransport* t) {
287 D("Calling send_tls_request");
288 apacket* p = get_apacket();
289 p->msg.command = A_STLS;
290 p->msg.arg0 = A_STLS_VERSION;
291 p->msg.data_length = 0;
292 send_packet(p, t);
293}
294
Dan Albertbe8e54b2015-05-18 13:06:53 -0700295void send_connect(atransport* t) {
Yabin Cui815ad882015-09-02 17:44:28 -0700296 D("Calling send_connect");
Dan Albertbe8e54b2015-05-18 13:06:53 -0700297 apacket* cp = get_apacket();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800298 cp->msg.command = A_CNXN;
Tim Murrayee7b44d2017-12-07 11:40:00 -0800299 // Send the max supported version, but because the transport is
300 // initialized to A_VERSION_MIN, this will be compatible with every
301 // device.
302 cp->msg.arg0 = A_VERSION;
Tamas Berghammera1c60c02015-07-13 19:12:28 +0100303 cp->msg.arg1 = t->get_max_payload();
Dan Albertbe8e54b2015-05-18 13:06:53 -0700304
305 std::string connection_str = get_connection_string();
306 // Connect and auth packets are limited to MAX_PAYLOAD_V1 because we don't
307 // yet know how much data the other size is willing to accept.
308 if (connection_str.length() > MAX_PAYLOAD_V1) {
309 LOG(FATAL) << "Connection banner is too long (length = "
310 << connection_str.length() << ")";
311 }
312
Josh Gaocd2a5292018-03-07 16:52:28 -0800313 cp->payload.assign(connection_str.begin(), connection_str.end());
Josh Gao839b9322018-02-05 18:49:10 -0800314 cp->msg.data_length = cp->payload.size();
Dan Albertbe8e54b2015-05-18 13:06:53 -0700315
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800316 send_packet(cp, t);
Benoit Goby2cc19e42012-04-12 12:23:49 -0700317}
318
Dan Albertbe8e54b2015-05-18 13:06:53 -0700319void parse_banner(const std::string& banner, atransport* t) {
Yabin Cui815ad882015-09-02 17:44:28 -0700320 D("parse_banner: %s", banner.c_str());
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700321
322 // The format is something like:
323 // "device::ro.product.name=x;ro.product.model=y;ro.product.device=z;".
324 std::vector<std::string> pieces = android::base::Split(banner, ":");
325
David Pursell3d9072b2015-09-25 13:04:21 -0700326 // Reset the features list or else if the server sends no features we may
327 // keep the existing feature set (http://b/24405971).
328 t->SetFeatures("");
329
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700330 if (pieces.size() > 2) {
331 const std::string& props = pieces[2];
Elliott Hughes85952832015-10-07 15:59:35 -0700332 for (const auto& prop : android::base::Split(props, ";")) {
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700333 // The list of properties was traditionally ;-terminated rather than ;-separated.
334 if (prop.empty()) continue;
335
336 std::vector<std::string> key_value = android::base::Split(prop, "=");
337 if (key_value.size() != 2) continue;
338
339 const std::string& key = key_value[0];
340 const std::string& value = key_value[1];
341 if (key == "ro.product.name") {
Luis Hector Chavezb4edbdf2018-07-18 21:18:27 -0700342 t->product = value;
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700343 } else if (key == "ro.product.model") {
Luis Hector Chavezb4edbdf2018-07-18 21:18:27 -0700344 t->model = value;
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700345 } else if (key == "ro.product.device") {
Luis Hector Chavezb4edbdf2018-07-18 21:18:27 -0700346 t->device = value;
Dan Albertbe8e54b2015-05-18 13:06:53 -0700347 } else if (key == "features") {
David Pursella07dbad2015-09-22 10:43:08 -0700348 t->SetFeatures(value);
Scott Andersonfa020922012-05-25 14:10:02 -0700349 }
350 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800351 }
352
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700353 const std::string& type = pieces[0];
354 if (type == "bootloader") {
Yabin Cui815ad882015-09-02 17:44:28 -0700355 D("setting connection_state to kCsBootloader");
Yabin Cui3cf1b362017-03-10 16:01:01 -0800356 t->SetConnectionState(kCsBootloader);
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700357 } else if (type == "device") {
Yabin Cui815ad882015-09-02 17:44:28 -0700358 D("setting connection_state to kCsDevice");
Yabin Cui3cf1b362017-03-10 16:01:01 -0800359 t->SetConnectionState(kCsDevice);
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700360 } else if (type == "recovery") {
Yabin Cui815ad882015-09-02 17:44:28 -0700361 D("setting connection_state to kCsRecovery");
Yabin Cui3cf1b362017-03-10 16:01:01 -0800362 t->SetConnectionState(kCsRecovery);
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700363 } else if (type == "sideload") {
Yabin Cui815ad882015-09-02 17:44:28 -0700364 D("setting connection_state to kCsSideload");
Yabin Cui3cf1b362017-03-10 16:01:01 -0800365 t->SetConnectionState(kCsSideload);
Tao Bao9d6eca52019-04-07 23:24:03 -0700366 } else if (type == "rescue") {
367 D("setting connection_state to kCsRescue");
368 t->SetConnectionState(kCsRescue);
Elliott Hughesb27e2412015-04-29 22:37:25 -0700369 } else {
Yabin Cui815ad882015-09-02 17:44:28 -0700370 D("setting connection_state to kCsHost");
Yabin Cui3cf1b362017-03-10 16:01:01 -0800371 t->SetConnectionState(kCsHost);
Doug Zongker6b217ed2012-01-09 14:54:53 -0800372 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800373}
374
Dan Albertbe8e54b2015-05-18 13:06:53 -0700375static void handle_new_connection(atransport* t, apacket* p) {
Josh Gao715fe602018-02-16 13:24:58 -0800376 handle_offline(t);
Dan Albertbe8e54b2015-05-18 13:06:53 -0700377
378 t->update_version(p->msg.arg0, p->msg.arg1);
Josh Gaocd2a5292018-03-07 16:52:28 -0800379 std::string banner(p->payload.begin(), p->payload.end());
380 parse_banner(banner, t);
Dan Albertbe8e54b2015-05-18 13:06:53 -0700381
382#if ADB_HOST
383 handle_online(t);
384#else
Josh Gaobd775212020-02-26 16:39:20 -0800385 ADB_LOG(Connection) << "received CNXN: version=" << p->msg.arg0 << ", maxdata = " << p->msg.arg1
386 << ", banner = '" << banner << "'";
387
Joshua Duong64fab752020-01-21 13:19:42 -0800388 if (t->use_tls) {
389 // We still handshake in TLS mode. If auth_required is disabled,
390 // we'll just not verify the client's certificate. This should be the
391 // first packet the client receives to indicate the new protocol.
392 send_tls_request(t);
393 } else if (!auth_required) {
Josh Gao7cac88a2019-10-22 12:30:39 -0700394 LOG(INFO) << "authentication not required";
Dan Albertbe8e54b2015-05-18 13:06:53 -0700395 handle_online(t);
396 send_connect(t);
397 } else {
398 send_auth_request(t);
399 }
400#endif
401}
402
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800403void handle_packet(apacket *p, atransport *t)
404{
Yabin Cui815ad882015-09-02 17:44:28 -0700405 D("handle_packet() %c%c%c%c", ((char*) (&(p->msg.command)))[0],
Viral Mehta841f9a72010-06-16 18:41:28 +0530406 ((char*) (&(p->msg.command)))[1],
407 ((char*) (&(p->msg.command)))[2],
408 ((char*) (&(p->msg.command)))[3]);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800409 print_packet("recv", p);
Josh Gao839b9322018-02-05 18:49:10 -0800410 CHECK_EQ(p->payload.size(), p->msg.data_length);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800411
412 switch(p->msg.command){
Dan Albertbe8e54b2015-05-18 13:06:53 -0700413 case A_CNXN: // CONNECT(version, maxdata, "system-id-string")
414 handle_new_connection(t, p);
Benoit Goby2cc19e42012-04-12 12:23:49 -0700415 break;
Joshua Duong64fab752020-01-21 13:19:42 -0800416 case A_STLS: // TLS(version, "")
417 t->use_tls = true;
418#if ADB_HOST
419 send_tls_request(t);
420 adb_auth_tls_handshake(t);
421#else
422 adbd_auth_tls_handshake(t);
423#endif
424 break;
Benoit Goby2cc19e42012-04-12 12:23:49 -0700425
426 case A_AUTH:
Joshua Duong64fab752020-01-21 13:19:42 -0800427 // All AUTH commands are ignored in TLS mode
428 if (t->use_tls) {
429 break;
430 }
Josh Gaoeac20582016-10-05 19:02:29 -0700431 switch (p->msg.arg0) {
432#if ADB_HOST
433 case ADB_AUTH_TOKEN:
Josh Gao7a7c5cb2018-05-04 16:04:49 -0700434 if (t->GetConnectionState() != kCsAuthorizing) {
435 t->SetConnectionState(kCsAuthorizing);
Yabin Cui3cf1b362017-03-10 16:01:01 -0800436 }
Josh Gao839b9322018-02-05 18:49:10 -0800437 send_auth_response(p->payload.data(), p->msg.data_length, t);
Josh Gaoeac20582016-10-05 19:02:29 -0700438 break;
439#else
Josh Gaocd2a5292018-03-07 16:52:28 -0800440 case ADB_AUTH_SIGNATURE: {
441 // TODO: Switch to string_view.
442 std::string signature(p->payload.begin(), p->payload.end());
Michael Groover02b74272019-04-25 18:33:35 -0700443 std::string auth_key;
444 if (adbd_auth_verify(t->token, sizeof(t->token), signature, &auth_key)) {
Josh Gaoeac20582016-10-05 19:02:29 -0700445 adbd_auth_verified(t);
446 t->failed_auth_attempts = 0;
Michael Groover02b74272019-04-25 18:33:35 -0700447 t->auth_key = auth_key;
448 adbd_notify_framework_connected_key(t);
Josh Gaoeac20582016-10-05 19:02:29 -0700449 } else {
Elliott Hughes73925982016-11-15 12:37:32 -0800450 if (t->failed_auth_attempts++ > 256) std::this_thread::sleep_for(1s);
Josh Gaoeac20582016-10-05 19:02:29 -0700451 send_auth_request(t);
452 }
453 break;
Josh Gaocd2a5292018-03-07 16:52:28 -0800454 }
Josh Gaoeac20582016-10-05 19:02:29 -0700455
456 case ADB_AUTH_RSAPUBLICKEY:
Michael Groover02b74272019-04-25 18:33:35 -0700457 t->auth_key = std::string(p->payload.data());
458 adbd_auth_confirm_key(t);
Josh Gaoeac20582016-10-05 19:02:29 -0700459 break;
460#endif
461 default:
Yabin Cui3cf1b362017-03-10 16:01:01 -0800462 t->SetConnectionState(kCsOffline);
Josh Gaoeac20582016-10-05 19:02:29 -0700463 handle_offline(t);
464 break;
Benoit Goby2cc19e42012-04-12 12:23:49 -0700465 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800466 break;
467
Josh Gaobd1f3002021-12-29 21:15:12 -0800468 case A_OPEN: {
469 /* OPEN(local-id, [send-buffer], "destination") */
470 if (!t->online || p->msg.arg0 == 0) {
471 break;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800472 }
Josh Gaobd1f3002021-12-29 21:15:12 -0800473
474 uint32_t send_bytes = static_cast<uint32_t>(p->msg.arg1);
475 if (t->SupportsDelayedAck() != static_cast<bool>(send_bytes)) {
476 LOG(ERROR) << "unexpected value of A_OPEN arg1: " << send_bytes
477 << " (delayed acks = " << t->SupportsDelayedAck() << ")";
478 send_close(0, p->msg.arg0, t);
479 break;
480 }
481
482 std::string_view address(p->payload.begin(), p->payload.size());
483
484 // Historically, we received service names as a char*, and stopped at the first NUL
485 // byte. The client sent strings with null termination, which post-string_view, start
486 // being interpreted as part of the string, unless we explicitly strip them.
487 address = StripTrailingNulls(address);
Shaju Mathew13508c12022-06-25 14:57:31 +0000488#if ADB_HOST
489 // The incoming address (from the payload) might be some other
490 // target (e.g tcp:<ip>:8000), however we do not allow *any*
491 // such requests - namely, those from (a potentially compromised)
492 // adbd (reverse:forward: source) port transport.
493 if (!t->IsReverseConfigured(address.data())) {
494 LOG(FATAL) << __func__ << " disallowed connect to " << address << " from "
495 << t->serial_name();
496 }
497#endif
Josh Gaobd1f3002021-12-29 21:15:12 -0800498 asocket* s = create_local_service_socket(address, t);
499 if (s == nullptr) {
500 send_close(0, p->msg.arg0, t);
501 break;
502 }
503
504 s->peer = create_remote_socket(p->msg.arg0, t);
505 s->peer->peer = s;
506
507 if (t->SupportsDelayedAck()) {
508 LOG(DEBUG) << "delayed ack available: send buffer = " << send_bytes;
509 s->available_send_bytes = send_bytes;
510
511 // TODO: Make this adjustable at connection time?
512 send_ready(s->id, s->peer->id, t, INITIAL_DELAYED_ACK_BYTES);
513 } else {
514 LOG(DEBUG) << "delayed ack unavailable";
515 send_ready(s->id, s->peer->id, t, 0);
516 }
517
518 s->ready(s);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800519 break;
Josh Gaobd1f3002021-12-29 21:15:12 -0800520 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800521
522 case A_OKAY: /* READY(local-id, remote-id, "") */
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100523 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
Elliott Hughesc3d1c112016-06-15 14:46:56 -0700524 asocket* s = find_local_socket(p->msg.arg1, 0);
525 if (s) {
Josh Gaobd1f3002021-12-29 21:15:12 -0800526 std::optional<int32_t> acked_bytes;
527 if (p->payload.size() == sizeof(int32_t)) {
528 int32_t value;
529 memcpy(&value, p->payload.data(), sizeof(value));
530 // acked_bytes can be negative!
531 //
532 // In the future, we can use this to preemptively supply backpressure, instead
533 // of waiting for the writer to hit its limit.
534 acked_bytes = value;
535 } else if (p->payload.size() != 0) {
536 LOG(ERROR) << "invalid A_OKAY payload size: " << p->payload.size();
537 return;
538 }
539
540 if (s->peer == nullptr) {
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100541 /* On first READY message, create the connection. */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800542 s->peer = create_remote_socket(p->msg.arg0, t);
543 s->peer->peer = s;
Josh Gaobd1f3002021-12-29 21:15:12 -0800544
545 local_socket_ack(s, acked_bytes);
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100546 s->ready(s);
547 } else if (s->peer->id == p->msg.arg0) {
548 /* Other READY messages must use the same local-id */
Josh Gaobd1f3002021-12-29 21:15:12 -0800549 local_socket_ack(s, acked_bytes);
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100550 } else {
Luis Hector Chavezb4edbdf2018-07-18 21:18:27 -0700551 D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s", p->msg.arg0,
552 p->msg.arg1, s->peer->id, p->msg.arg1, t->serial.c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800553 }
Yabin Cuiec2e7d82015-09-29 12:25:33 -0700554 } else {
555 // When receiving A_OKAY from device for A_OPEN request, the host server may
556 // have closed the local socket because of client disconnection. Then we need
557 // to send A_CLSE back to device to close the service on device.
558 send_close(p->msg.arg1, p->msg.arg0, t);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800559 }
560 }
561 break;
562
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100563 case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
564 if (t->online && p->msg.arg1 != 0) {
Elliott Hughesc3d1c112016-06-15 14:46:56 -0700565 asocket* s = find_local_socket(p->msg.arg1, p->msg.arg0);
566 if (s) {
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100567 /* According to protocol.txt, p->msg.arg0 might be 0 to indicate
568 * a failed OPEN only. However, due to a bug in previous ADB
569 * versions, CLOSE(0, remote-id, "") was also used for normal
570 * CLOSE() operations.
571 *
572 * This is bad because it means a compromised adbd could
573 * send packets to close connections between the host and
574 * other devices. To avoid this, only allow this if the local
575 * socket has a peer on the same transport.
576 */
577 if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
Luis Hector Chavezb4edbdf2018-07-18 21:18:27 -0700578 D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s", p->msg.arg1,
579 t->serial.c_str(), s->peer->transport->serial.c_str());
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100580 } else {
581 s->close(s);
582 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800583 }
584 }
585 break;
586
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100587 case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
588 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
Elliott Hughesc3d1c112016-06-15 14:46:56 -0700589 asocket* s = find_local_socket(p->msg.arg1, p->msg.arg0);
590 if (s) {
Josh Gaobd1f3002021-12-29 21:15:12 -0800591 s->enqueue(s, std::move(p->payload));
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800592 }
593 }
594 break;
595
596 default:
597 printf("handle_packet: what is %08x?!\n", p->msg.command);
598 }
599
600 put_apacket(p);
601}
602
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800603#if ADB_HOST
Nick Kralevich23b39f02013-02-15 10:22:08 -0800604
Spencer Low2bbb3a92015-08-26 18:46:09 -0700605#ifdef _WIN32
606
Spencer Lowb77fc722015-09-16 16:40:06 -0700607// Try to make a handle non-inheritable and if there is an error, don't output
608// any error info, but leave GetLastError() for the caller to read. This is
609// convenient if the caller is expecting that this may fail and they'd like to
610// ignore such a failure.
611static bool _try_make_handle_noninheritable(HANDLE h) {
Spencer Low2bbb3a92015-08-26 18:46:09 -0700612 if (h != INVALID_HANDLE_VALUE && h != NULL) {
Spencer Lowb77fc722015-09-16 16:40:06 -0700613 return SetHandleInformation(h, HANDLE_FLAG_INHERIT, 0) ? true : false;
614 }
615
616 return true;
617}
618
619// Try to make a handle non-inheritable with the expectation that this should
620// succeed, so if this fails, output error info.
621static bool _make_handle_noninheritable(HANDLE h) {
622 if (!_try_make_handle_noninheritable(h)) {
623 // Show the handle value to give us a clue in case we have problems
624 // with pseudo-handle values.
Elliott Hughes65bc2272017-04-18 14:34:16 -0700625 fprintf(stderr, "adb: cannot make handle 0x%p non-inheritable: %s\n", h,
626 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Lowb77fc722015-09-16 16:40:06 -0700627 return false;
Spencer Low2bbb3a92015-08-26 18:46:09 -0700628 }
629
630 return true;
631}
632
633// Create anonymous pipe, preventing inheritance of the read pipe and setting
634// security of the write pipe to sa.
635static bool _create_anonymous_pipe(unique_handle* pipe_read_out,
636 unique_handle* pipe_write_out,
637 SECURITY_ATTRIBUTES* sa) {
638 HANDLE pipe_read_raw = NULL;
639 HANDLE pipe_write_raw = NULL;
640 if (!CreatePipe(&pipe_read_raw, &pipe_write_raw, sa, 0)) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700641 fprintf(stderr, "adb: CreatePipe failed: %s\n",
David Pursellc573d522016-01-27 08:52:53 -0800642 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low2bbb3a92015-08-26 18:46:09 -0700643 return false;
644 }
645
646 unique_handle pipe_read(pipe_read_raw);
647 pipe_read_raw = NULL;
648 unique_handle pipe_write(pipe_write_raw);
649 pipe_write_raw = NULL;
650
651 if (!_make_handle_noninheritable(pipe_read.get())) {
652 return false;
653 }
654
655 *pipe_read_out = std::move(pipe_read);
656 *pipe_write_out = std::move(pipe_write);
657
658 return true;
659}
660
Spencer Low58216782015-11-21 17:23:51 -0800661// Read from a pipe (that we take ownership of) and write the result to stdout/stderr. Return on
662// error or when the pipe is closed. Internally makes inheritable handles, so this should not be
663// called if subprocesses may be started concurrently.
Spencer Low2bbb3a92015-08-26 18:46:09 -0700664static unsigned _redirect_pipe_thread(HANDLE h, DWORD nStdHandle) {
665 // Take ownership of the HANDLE and close when we're done.
666 unique_handle read_pipe(h);
Spencer Low58216782015-11-21 17:23:51 -0800667 const char* output_name = nStdHandle == STD_OUTPUT_HANDLE ? "stdout" : "stderr";
668 const int original_fd = fileno(nStdHandle == STD_OUTPUT_HANDLE ? stdout : stderr);
669 std::unique_ptr<FILE, decltype(&fclose)> stream(nullptr, fclose);
670
671 if (original_fd == -1) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700672 fprintf(stderr, "adb: failed to get file descriptor for %s: %s\n", output_name,
673 strerror(errno));
Spencer Low58216782015-11-21 17:23:51 -0800674 return EXIT_FAILURE;
675 }
676
677 // If fileno() is -2, stdout/stderr is not associated with an output stream, so we should read,
678 // but don't write. Otherwise, make a FILE* identical to stdout/stderr except that it is in
679 // binary mode with no CR/LR translation since we're reading raw.
680 if (original_fd >= 0) {
681 // This internally makes a duplicate file handle that is inheritable, so callers should not
682 // call this function if subprocesses may be started concurrently.
683 const int fd = dup(original_fd);
684 if (fd == -1) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700685 fprintf(stderr, "adb: failed to duplicate file descriptor for %s: %s\n", output_name,
Spencer Low58216782015-11-21 17:23:51 -0800686 strerror(errno));
687 return EXIT_FAILURE;
688 }
689
690 // Note that although we call fdopen() below with a binary flag, it may not adhere to that
691 // flag, so we have to set the mode manually.
692 if (_setmode(fd, _O_BINARY) == -1) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700693 fprintf(stderr, "adb: failed to set binary mode for duplicate of %s: %s\n", output_name,
Spencer Low58216782015-11-21 17:23:51 -0800694 strerror(errno));
695 unix_close(fd);
696 return EXIT_FAILURE;
697 }
698
699 stream.reset(fdopen(fd, "wb"));
700 if (stream.get() == nullptr) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700701 fprintf(stderr, "adb: failed to open duplicate stream for %s: %s\n", output_name,
Spencer Low58216782015-11-21 17:23:51 -0800702 strerror(errno));
703 unix_close(fd);
704 return EXIT_FAILURE;
705 }
706
707 // Unbuffer the stream because it will be buffered by default and we want subprocess output
708 // to be shown immediately.
709 if (setvbuf(stream.get(), NULL, _IONBF, 0) == -1) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700710 fprintf(stderr, "adb: failed to unbuffer %s: %s\n", output_name, strerror(errno));
Spencer Low58216782015-11-21 17:23:51 -0800711 return EXIT_FAILURE;
712 }
713
714 // fd will be closed when stream is closed.
715 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700716
717 while (true) {
718 char buf[64 * 1024];
719 DWORD bytes_read = 0;
720 if (!ReadFile(read_pipe.get(), buf, sizeof(buf), &bytes_read, NULL)) {
721 const DWORD err = GetLastError();
722 // ERROR_BROKEN_PIPE is expected when the subprocess closes
723 // the other end of the pipe.
724 if (err == ERROR_BROKEN_PIPE) {
725 return EXIT_SUCCESS;
726 } else {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700727 fprintf(stderr, "adb: failed to read from %s: %s\n", output_name,
David Pursellc573d522016-01-27 08:52:53 -0800728 android::base::SystemErrorCodeToString(err).c_str());
Spencer Low2bbb3a92015-08-26 18:46:09 -0700729 return EXIT_FAILURE;
730 }
731 }
732
Spencer Low58216782015-11-21 17:23:51 -0800733 // Don't try to write if our stdout/stderr was not setup by the parent process.
734 if (stream) {
735 // fwrite() actually calls adb_fwrite() which can write UTF-8 to the console.
736 const size_t bytes_written = fwrite(buf, 1, bytes_read, stream.get());
Spencer Low2bbb3a92015-08-26 18:46:09 -0700737 if (bytes_written != bytes_read) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700738 fprintf(stderr, "adb: error: only wrote %zu of %lu bytes to %s\n", bytes_written,
739 bytes_read, output_name);
Spencer Low2bbb3a92015-08-26 18:46:09 -0700740 return EXIT_FAILURE;
741 }
742 }
743 }
744}
745
746static unsigned __stdcall _redirect_stdout_thread(HANDLE h) {
Siva Velusamy2669cf92015-08-28 16:37:29 -0700747 adb_thread_setname("stdout redirect");
Spencer Low2bbb3a92015-08-26 18:46:09 -0700748 return _redirect_pipe_thread(h, STD_OUTPUT_HANDLE);
749}
750
751static unsigned __stdcall _redirect_stderr_thread(HANDLE h) {
Siva Velusamy2669cf92015-08-28 16:37:29 -0700752 adb_thread_setname("stderr redirect");
Spencer Low2bbb3a92015-08-26 18:46:09 -0700753 return _redirect_pipe_thread(h, STD_ERROR_HANDLE);
754}
755
756#endif
757
Elliott Hughescd61ae32017-06-15 08:35:24 -0700758static void ReportServerStartupFailure(pid_t pid) {
759 fprintf(stderr, "ADB server didn't ACK\n");
760 fprintf(stderr, "Full server startup log: %s\n", GetLogFilePath().c_str());
761 fprintf(stderr, "Server had pid: %d\n", pid);
762
Josh Gaof0c44032018-12-12 16:12:28 -0800763 android::base::unique_fd fd(unix_open(GetLogFilePath(), O_RDONLY));
Elliott Hughescd61ae32017-06-15 08:35:24 -0700764 if (fd == -1) return;
765
766 // Let's not show more than 128KiB of log...
Spencer Low04b575d2018-09-02 19:19:39 -0700767 unix_lseek(fd, -128 * 1024, SEEK_END);
Elliott Hughescd61ae32017-06-15 08:35:24 -0700768 std::string content;
769 if (!android::base::ReadFdToString(fd, &content)) return;
770
771 std::string header = android::base::StringPrintf("--- adb starting (pid %d) ---", pid);
772 std::vector<std::string> lines = android::base::Split(content, "\n");
773 int i = lines.size() - 1;
774 while (i >= 0 && lines[i] != header) --i;
775 while (static_cast<size_t>(i) < lines.size()) fprintf(stderr, "%s\n", lines[i++].c_str());
776}
777
Vince Harron5703eb32021-11-12 12:40:58 -0800778int launch_server(const std::string& socket_spec, const char* one_device) {
Yabin Cui2fa43212014-11-11 09:24:11 -0800779#if defined(_WIN32)
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800780 /* we need to start the server in the background */
781 /* we create a PIPE that will be used to wait for the server's "OK" */
782 /* message since the pipe handles must be inheritable, we use a */
783 /* security attribute */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800784 SECURITY_ATTRIBUTES sa;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800785 sa.nLength = sizeof(sa);
786 sa.lpSecurityDescriptor = NULL;
787 sa.bInheritHandle = TRUE;
788
Spencer Low2bbb3a92015-08-26 18:46:09 -0700789 // Redirect stdin to Windows /dev/null. If we instead pass an original
790 // stdin/stdout/stderr handle and it is a console handle, when the adb
791 // server starts up, the C Runtime will see a console handle for a process
792 // that isn't connected to a console and it will configure
793 // stdin/stdout/stderr to be closed. At that point, freopen() could be used
794 // to reopen stderr/out, but it would take more massaging to fixup the file
795 // descriptor number that freopen() uses. It's simplest to avoid all of this
796 // complexity by just redirecting stdin to `nul' and then the C Runtime acts
797 // as expected.
798 unique_handle nul_read(CreateFileW(L"nul", GENERIC_READ,
799 FILE_SHARE_READ | FILE_SHARE_WRITE, &sa, OPEN_EXISTING,
800 FILE_ATTRIBUTE_NORMAL, NULL));
801 if (nul_read.get() == INVALID_HANDLE_VALUE) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700802 fprintf(stderr, "adb: CreateFileW 'nul' failed: %s\n",
David Pursellc573d522016-01-27 08:52:53 -0800803 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low4911f4b2015-05-20 23:17:26 -0700804 return -1;
805 }
806
Spencer Low58216782015-11-21 17:23:51 -0800807 // Create pipes with non-inheritable read handle, inheritable write handle. We need to connect
808 // the subprocess to pipes instead of just letting the subprocess inherit our existing
809 // stdout/stderr handles because a DETACHED_PROCESS cannot write to a console that it is not
810 // attached to.
Spencer Low2bbb3a92015-08-26 18:46:09 -0700811 unique_handle ack_read, ack_write;
812 if (!_create_anonymous_pipe(&ack_read, &ack_write, &sa)) {
Spencer Low4911f4b2015-05-20 23:17:26 -0700813 return -1;
814 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700815 unique_handle stdout_read, stdout_write;
816 if (!_create_anonymous_pipe(&stdout_read, &stdout_write, &sa)) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800817 return -1;
818 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700819 unique_handle stderr_read, stderr_write;
820 if (!_create_anonymous_pipe(&stderr_read, &stderr_write, &sa)) {
821 return -1;
822 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800823
Ray Donnellyc0324702012-11-29 01:18:50 +0000824 /* Some programs want to launch an adb command and collect its output by
825 * calling CreateProcess with inheritable stdout/stderr handles, then
826 * using read() to get its output. When this happens, the stdout/stderr
827 * handles passed to the adb client process will also be inheritable.
828 * When starting the adb server here, care must be taken to reset them
829 * to non-inheritable.
830 * Otherwise, something bad happens: even if the adb command completes,
831 * the calling process is stuck while read()-ing from the stdout/stderr
832 * descriptors, because they're connected to corresponding handles in the
833 * adb server process (even if the latter never uses/writes to them).
Spencer Low2bbb3a92015-08-26 18:46:09 -0700834 * Note that even if we don't pass these handles in the STARTUPINFO struct,
835 * if they're marked inheritable, they're still inherited, requiring us to
836 * deal with this.
837 *
838 * If we're still having problems with inheriting random handles in the
839 * future, consider using PROC_THREAD_ATTRIBUTE_HANDLE_LIST to explicitly
840 * specify which handles should be inherited: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/16/10248328.aspx
Spencer Lowb77fc722015-09-16 16:40:06 -0700841 *
842 * Older versions of Windows return console pseudo-handles that cannot be
843 * made non-inheritable, so ignore those failures.
Ray Donnellyc0324702012-11-29 01:18:50 +0000844 */
Spencer Lowb77fc722015-09-16 16:40:06 -0700845 _try_make_handle_noninheritable(GetStdHandle(STD_INPUT_HANDLE));
846 _try_make_handle_noninheritable(GetStdHandle(STD_OUTPUT_HANDLE));
847 _try_make_handle_noninheritable(GetStdHandle(STD_ERROR_HANDLE));
Ray Donnellyc0324702012-11-29 01:18:50 +0000848
Spencer Low2bbb3a92015-08-26 18:46:09 -0700849 STARTUPINFOW startup;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800850 ZeroMemory( &startup, sizeof(startup) );
851 startup.cb = sizeof(startup);
Spencer Low2bbb3a92015-08-26 18:46:09 -0700852 startup.hStdInput = nul_read.get();
853 startup.hStdOutput = stdout_write.get();
854 startup.hStdError = stderr_write.get();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800855 startup.dwFlags = STARTF_USESTDHANDLES;
856
Spencer Low2bbb3a92015-08-26 18:46:09 -0700857 // Verify that the pipe_write handle value can be passed on the command line
858 // as %d and that the rest of adb code can pass it around in an int.
859 const int ack_write_as_int = cast_handle_to_int(ack_write.get());
860 if (cast_int_to_handle(ack_write_as_int) != ack_write.get()) {
861 // If this fires, either handle values are larger than 32-bits or else
862 // there is a bug in our casting.
863 // https://msdn.microsoft.com/en-us/library/windows/desktop/aa384203%28v=vs.85%29.aspx
Elliott Hughes65bc2272017-04-18 14:34:16 -0700864 fprintf(stderr, "adb: cannot fit pipe handle value into 32-bits: 0x%p\n", ack_write.get());
Spencer Low2bbb3a92015-08-26 18:46:09 -0700865 return -1;
866 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800867
Spencer Low2bbb3a92015-08-26 18:46:09 -0700868 // get path of current program
869 WCHAR program_path[MAX_PATH];
870 const DWORD module_result = GetModuleFileNameW(NULL, program_path,
871 arraysize(program_path));
872 if ((module_result >= arraysize(program_path)) || (module_result == 0)) {
Spencer Low6815c072015-05-11 01:08:48 -0700873 // String truncation or some other error.
Elliott Hughes65bc2272017-04-18 14:34:16 -0700874 fprintf(stderr, "adb: cannot get executable path: %s\n",
David Pursellc573d522016-01-27 08:52:53 -0800875 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low6815c072015-05-11 01:08:48 -0700876 return -1;
877 }
Spencer Lowb28812f2015-08-08 15:07:07 -0700878
Bradley Furmand8f12392022-04-11 16:36:33 +0000879 WCHAR args[4096];
Vince Harron5703eb32021-11-12 12:40:58 -0800880 if (one_device) {
881 snwprintf(args, arraysize(args),
882 L"adb -L %s fork-server server --reply-fd %d --one-device %s",
883 socket_spec.c_str(), ack_write_as_int, one_device);
884 } else {
885 snwprintf(args, arraysize(args), L"adb -L %s fork-server server --reply-fd %d",
886 socket_spec.c_str(), ack_write_as_int);
887 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700888
889 PROCESS_INFORMATION pinfo;
890 ZeroMemory(&pinfo, sizeof(pinfo));
891
892 if (!CreateProcessW(
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800893 program_path, /* program path */
Wenhao Lie0fa2c52013-11-13 16:23:37 +0800894 args,
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800895 /* the fork-server argument will set the
896 debug = 2 in the child */
897 NULL, /* process handle is not inheritable */
898 NULL, /* thread handle is not inheritable */
899 TRUE, /* yes, inherit some handles */
900 DETACHED_PROCESS, /* the new process doesn't have a console */
901 NULL, /* use parent's environment block */
902 NULL, /* use parent's starting directory */
903 &startup, /* startup info, i.e. std handles */
Spencer Low2bbb3a92015-08-26 18:46:09 -0700904 &pinfo )) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700905 fprintf(stderr, "adb: CreateProcessW failed: %s\n",
David Pursellc573d522016-01-27 08:52:53 -0800906 android::base::SystemErrorCodeToString(GetLastError()).c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800907 return -1;
908 }
909
Spencer Low2bbb3a92015-08-26 18:46:09 -0700910 unique_handle process_handle(pinfo.hProcess);
911 pinfo.hProcess = NULL;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800912
Spencer Low2bbb3a92015-08-26 18:46:09 -0700913 // Close handles that we no longer need to complete the rest.
914 CloseHandle(pinfo.hThread);
915 pinfo.hThread = NULL;
916
917 nul_read.reset();
918 ack_write.reset();
919 stdout_write.reset();
920 stderr_write.reset();
921
Spencer Low58216782015-11-21 17:23:51 -0800922 // Start threads to read from subprocess stdout/stderr and write to ours to make subprocess
923 // errors easier to diagnose. Note that the threads internally create inheritable handles, but
924 // that is ok because we've already spawned the subprocess.
Spencer Low2bbb3a92015-08-26 18:46:09 -0700925
926 // In the past, reading from a pipe before the child process's C Runtime
927 // started up and called GetFileType() caused a hang: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/02/10243553.aspx#10244216
928 // This is reportedly fixed in Windows Vista: https://support.microsoft.com/en-us/kb/2009703
929 // I was unable to reproduce the problem on Windows XP. It sounds like a
930 // Windows Update may have fixed this: https://www.duckware.com/tech/peeknamedpipe.html
931 unique_handle stdout_thread(reinterpret_cast<HANDLE>(
932 _beginthreadex(NULL, 0, _redirect_stdout_thread, stdout_read.get(),
933 0, NULL)));
934 if (stdout_thread.get() == nullptr) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700935 fprintf(stderr, "adb: cannot create thread: %s\n", strerror(errno));
Spencer Low2bbb3a92015-08-26 18:46:09 -0700936 return -1;
937 }
938 stdout_read.release(); // Transfer ownership to new thread
939
940 unique_handle stderr_thread(reinterpret_cast<HANDLE>(
941 _beginthreadex(NULL, 0, _redirect_stderr_thread, stderr_read.get(),
942 0, NULL)));
943 if (stderr_thread.get() == nullptr) {
Elliott Hughes65bc2272017-04-18 14:34:16 -0700944 fprintf(stderr, "adb: cannot create thread: %s\n", strerror(errno));
Spencer Low2bbb3a92015-08-26 18:46:09 -0700945 return -1;
946 }
947 stderr_read.release(); // Transfer ownership to new thread
948
949 bool got_ack = false;
950
951 // Wait for the "OK\n" message, for the pipe to be closed, or other error.
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800952 {
Spencer Low2bbb3a92015-08-26 18:46:09 -0700953 char temp[3];
954 DWORD count = 0;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800955
Spencer Low2bbb3a92015-08-26 18:46:09 -0700956 if (ReadFile(ack_read.get(), temp, sizeof(temp), &count, NULL)) {
957 const CHAR expected[] = "OK\n";
958 const DWORD expected_length = arraysize(expected) - 1;
959 if (count == expected_length &&
960 memcmp(temp, expected, expected_length) == 0) {
961 got_ack = true;
962 } else {
Spencer Low04b575d2018-09-02 19:19:39 -0700963 ReportServerStartupFailure(pinfo.dwProcessId);
Elliott Hughescd61ae32017-06-15 08:35:24 -0700964 return -1;
Spencer Low2bbb3a92015-08-26 18:46:09 -0700965 }
966 } else {
967 const DWORD err = GetLastError();
968 // If the ACK was not written and the process exited, GetLastError()
969 // is probably ERROR_BROKEN_PIPE, in which case that info is not
970 // useful to the user.
971 fprintf(stderr, "could not read ok from ADB Server%s\n",
972 err == ERROR_BROKEN_PIPE ? "" :
973 android::base::StringPrintf(": %s",
David Pursellc573d522016-01-27 08:52:53 -0800974 android::base::SystemErrorCodeToString(err).c_str()).c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800975 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700976 }
977
978 // Always try to wait a bit for threads reading stdout/stderr to finish.
979 // If the process started ok, it should close the pipes causing the threads
980 // to finish. If the process had an error, it should exit, also causing
981 // the pipes to be closed. In that case we want to read all of the output
982 // and write it out so that the user can diagnose failures.
983 const DWORD thread_timeout_ms = 15 * 1000;
984 const HANDLE threads[] = { stdout_thread.get(), stderr_thread.get() };
985 const DWORD wait_result = WaitForMultipleObjects(arraysize(threads),
986 threads, TRUE, thread_timeout_ms);
987 if (wait_result == WAIT_TIMEOUT) {
988 // Threads did not finish after waiting a little while. Perhaps the
989 // server didn't close pipes, or it is hung.
Elliott Hughes65bc2272017-04-18 14:34:16 -0700990 fprintf(stderr, "adb: timed out waiting for threads to finish reading from ADB server\n");
Spencer Low2bbb3a92015-08-26 18:46:09 -0700991 // Process handles are signaled when the process exits, so if we wait
992 // on the handle for 0 seconds and it returns 'timeout', that means that
993 // the process is still running.
994 if (WaitForSingleObject(process_handle.get(), 0) == WAIT_TIMEOUT) {
995 // We could TerminateProcess(), but that seems somewhat presumptive.
Elliott Hughes65bc2272017-04-18 14:34:16 -0700996 fprintf(stderr, "adb: server is running with process id %lu\n", pinfo.dwProcessId);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800997 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700998 return -1;
999 }
1000
1001 if (wait_result != WAIT_OBJECT_0) {
Elliott Hughes65bc2272017-04-18 14:34:16 -07001002 fprintf(stderr, "adb: unexpected result waiting for threads: %lu: %s\n", wait_result,
1003 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low2bbb3a92015-08-26 18:46:09 -07001004 return -1;
1005 }
1006
1007 // For now ignore the thread exit codes and assume they worked properly.
1008
1009 if (!got_ack) {
1010 return -1;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001011 }
Yabin Cui2fa43212014-11-11 09:24:11 -08001012#else /* !defined(_WIN32) */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001013 // set up a pipe so the child can tell us when it is ready.
Josh Gao810fedb2018-05-11 12:55:56 -07001014 unique_fd pipe_read, pipe_write;
1015 if (!Pipe(&pipe_read, &pipe_write)) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001016 fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
1017 return -1;
1018 }
Elliott Hughes827a4a52016-08-31 15:07:18 -07001019
1020 std::string path = android::base::GetExecutablePath();
1021
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001022 pid_t pid = fork();
Elliott Hughes827a4a52016-08-31 15:07:18 -07001023 if (pid < 0) return -1;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001024
1025 if (pid == 0) {
1026 // child side of the fork
Josh Gao810fedb2018-05-11 12:55:56 -07001027 pipe_read.reset();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001028
Josh Gao7d130c42018-08-06 18:38:47 -07001029 // android::base::Pipe unconditionally opens the pipe with O_CLOEXEC.
1030 // Undo this manually.
1031 fcntl(pipe_write.get(), F_SETFD, 0);
1032
Siva Velusamya55dbd82015-08-07 10:10:29 -07001033 char reply_fd[30];
Josh Gao810fedb2018-05-11 12:55:56 -07001034 snprintf(reply_fd, sizeof(reply_fd), "%d", pipe_write.get());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001035 // child process
Vince Harron5703eb32021-11-12 12:40:58 -08001036 std::vector<const char*> child_argv = {
1037 "adb", "-L", socket_spec.c_str(), "fork-server", "server", "--reply-fd", reply_fd};
1038 if (one_device) {
1039 child_argv.push_back("--one-device");
1040 child_argv.push_back(one_device);
Bradley Furmand8f12392022-04-11 16:36:33 +00001041 } else if (access("/etc/adb/one_device_required", F_OK) == 0) {
1042 fprintf(stderr,
1043 "adb: cannot start server: --one-device option is required for this system in "
1044 "order to start adb.\n");
1045 return -1;
Vince Harron5703eb32021-11-12 12:40:58 -08001046 }
1047 child_argv.push_back(nullptr);
1048 int result = execv(path.c_str(), const_cast<char* const*>(child_argv.data()));
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001049 // this should not return
Elliott Hughes65bc2272017-04-18 14:34:16 -07001050 fprintf(stderr, "adb: execl returned %d: %s\n", result, strerror(errno));
Elliott Hughescd61ae32017-06-15 08:35:24 -07001051 } else {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001052 // parent side of the fork
Elliott Hughescd61ae32017-06-15 08:35:24 -07001053 char temp[3] = {};
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001054 // wait for the "OK\n" message
Josh Gao810fedb2018-05-11 12:55:56 -07001055 pipe_write.reset();
1056 int ret = adb_read(pipe_read.get(), temp, 3);
JP Abgrall2e5dd6e2011-03-16 15:57:42 -07001057 int saved_errno = errno;
Josh Gao810fedb2018-05-11 12:55:56 -07001058 pipe_read.reset();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001059 if (ret < 0) {
JP Abgrall2e5dd6e2011-03-16 15:57:42 -07001060 fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", saved_errno);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001061 return -1;
1062 }
1063 if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
Elliott Hughescd61ae32017-06-15 08:35:24 -07001064 ReportServerStartupFailure(pid);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001065 return -1;
1066 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001067 }
Yabin Cui2fa43212014-11-11 09:24:11 -08001068#endif /* !defined(_WIN32) */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001069 return 0;
1070}
Yabin Cui2fa43212014-11-11 09:24:11 -08001071#endif /* ADB_HOST */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001072
Josh Gaodd655ec2018-07-30 18:49:03 -07001073bool handle_forward_request(const char* service, atransport* transport, int reply_fd) {
1074 return handle_forward_request(service, [transport](std::string*) { return transport; },
1075 reply_fd);
1076}
1077
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001078// Try to handle a network forwarding request.
Josh Gaodd655ec2018-07-30 18:49:03 -07001079bool handle_forward_request(const char* service,
1080 std::function<atransport*(std::string* error)> transport_acquirer,
1081 int reply_fd) {
1082 if (!strcmp(service, "list-forward")) {
1083 // Create the list of forward redirections.
1084 std::string listeners = format_listeners();
1085#if ADB_HOST
1086 SendOkay(reply_fd);
1087#endif
1088 SendProtocolString(reply_fd, listeners);
1089 return true;
1090 }
1091
1092 if (!strcmp(service, "killforward-all")) {
1093 remove_all_listeners();
1094#if ADB_HOST
1095 /* On the host: 1st OKAY is connect, 2nd OKAY is status */
1096 SendOkay(reply_fd);
1097#endif
1098 SendOkay(reply_fd);
1099 return true;
1100 }
1101
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001102 if (!strncmp(service, "forward:", 8) || !strncmp(service, "killforward:", 12)) {
1103 // killforward:local
1104 // forward:(norebind:)?local;remote
Josh Gaodd655ec2018-07-30 18:49:03 -07001105 std::string error;
1106 atransport* transport = transport_acquirer(&error);
1107 if (!transport) {
1108 SendFail(reply_fd, error);
1109 return true;
1110 }
1111
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001112 bool kill_forward = false;
1113 bool no_rebind = false;
1114 if (android::base::StartsWith(service, "killforward:")) {
1115 kill_forward = true;
1116 service += 12;
Spencer Low95365132015-07-29 17:21:21 -07001117 } else {
1118 service += 8; // skip past "forward:"
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001119 if (android::base::StartsWith(service, "norebind:")) {
1120 no_rebind = true;
1121 service += 9;
1122 }
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001123 }
1124
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001125 std::vector<std::string> pieces = android::base::Split(service, ";");
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001126
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001127 if (kill_forward) {
1128 // Check killforward: parameter format: '<local>'
1129 if (pieces.size() != 1 || pieces[0].empty()) {
1130 SendFail(reply_fd, android::base::StringPrintf("bad killforward: %s", service));
Josh Gaodd655ec2018-07-30 18:49:03 -07001131 return true;
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001132 }
1133 } else {
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001134 // Check forward: parameter format: '<local>;<remote>'
1135 if (pieces.size() != 2 || pieces[0].empty() || pieces[1].empty() || pieces[1][0] == '*') {
1136 SendFail(reply_fd, android::base::StringPrintf("bad forward: %s", service));
Josh Gaodd655ec2018-07-30 18:49:03 -07001137 return true;
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001138 }
1139 }
1140
Elliott Hughes3aec2ba2015-05-05 13:10:43 -07001141 InstallStatus r;
David Pursell19d0c232016-04-07 11:25:48 -07001142 int resolved_tcp_port = 0;
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001143 if (kill_forward) {
1144 r = remove_listener(pieces[0].c_str(), transport);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001145 } else {
Daniel Colascione232c39c2019-11-13 17:51:19 -08001146 int flags = 0;
1147 if (no_rebind) {
1148 flags |= INSTALL_LISTENER_NO_REBIND;
1149 }
1150 r = install_listener(pieces[0], pieces[1].c_str(), transport, flags, &resolved_tcp_port,
1151 &error);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001152 }
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001153 if (r == INSTALL_STATUS_OK) {
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001154#if ADB_HOST
David Pursell19d0c232016-04-07 11:25:48 -07001155 // On the host: 1st OKAY is connect, 2nd OKAY is status.
Elliott Hughes88b4c852015-04-30 17:32:03 -07001156 SendOkay(reply_fd);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001157#endif
Elliott Hughes88b4c852015-04-30 17:32:03 -07001158 SendOkay(reply_fd);
David Pursell19d0c232016-04-07 11:25:48 -07001159
1160 // If a TCP port was resolved, send the actual port number back.
1161 if (resolved_tcp_port != 0) {
1162 SendProtocolString(reply_fd, android::base::StringPrintf("%d", resolved_tcp_port));
1163 }
1164
Josh Gaodd655ec2018-07-30 18:49:03 -07001165 return true;
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001166 }
1167
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001168 std::string message;
1169 switch (r) {
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001170 case INSTALL_STATUS_OK: message = "success (!)"; break;
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001171 case INSTALL_STATUS_INTERNAL_ERROR: message = "internal error"; break;
1172 case INSTALL_STATUS_CANNOT_BIND:
Spencer Low32625852015-08-11 16:45:32 -07001173 message = android::base::StringPrintf("cannot bind listener: %s",
Spencer Low753d4852015-07-30 23:07:55 -07001174 error.c_str());
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001175 break;
1176 case INSTALL_STATUS_CANNOT_REBIND:
Spencer Low95365132015-07-29 17:21:21 -07001177 message = android::base::StringPrintf("cannot rebind existing socket");
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001178 break;
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001179 case INSTALL_STATUS_LISTENER_NOT_FOUND:
1180 message = android::base::StringPrintf("listener '%s' not found", service);
1181 break;
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001182 }
Elliott Hughes88b4c852015-04-30 17:32:03 -07001183 SendFail(reply_fd, message);
Josh Gaodd655ec2018-07-30 18:49:03 -07001184 return true;
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001185 }
Josh Gaodd655ec2018-07-30 18:49:03 -07001186
1187 return false;
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001188}
1189
Elliott Hughes24f52762015-06-23 13:00:32 -07001190#if ADB_HOST
1191static int SendOkay(int fd, const std::string& s) {
1192 SendOkay(fd);
1193 SendProtocolString(fd, s);
1194 return 0;
1195}
Elliott Hughes24f52762015-06-23 13:00:32 -07001196
Josh Gaobea8f3c2020-04-06 12:07:56 -07001197static bool g_reject_kill_server = false;
1198void adb_set_reject_kill_server(bool value) {
1199 g_reject_kill_server = value;
1200}
1201
Joshua Duonge22e1612020-03-31 10:58:50 -07001202static bool handle_mdns_request(std::string_view service, int reply_fd) {
1203 if (!android::base::ConsumePrefix(&service, "mdns:")) {
1204 return false;
1205 }
1206
1207 if (service == "check") {
1208 std::string check = mdns_check();
1209 SendOkay(reply_fd, check);
1210 return true;
1211 }
Joshua Duong5f63d112020-03-31 08:39:24 -07001212 if (service == "services") {
1213 std::string services_list = mdns_list_discovered_services();
1214 SendOkay(reply_fd, services_list);
1215 return true;
1216 }
Joshua Duonge22e1612020-03-31 10:58:50 -07001217
1218 return false;
1219}
1220
Josh Gaob13f3cd2019-02-20 20:37:26 -08001221HostRequestResult handle_host_request(std::string_view service, TransportType type,
1222 const char* serial, TransportId transport_id, int reply_fd,
1223 asocket* s) {
Josh Gaocd30bb32019-02-20 19:30:59 -08001224 if (service == "kill") {
Josh Gaobea8f3c2020-04-06 12:07:56 -07001225 if (g_reject_kill_server) {
1226 LOG(WARNING) << "adb server ignoring kill-server";
1227 SendFail(reply_fd, "kill-server rejected by remote server");
1228 } else {
1229 fprintf(stderr, "adb server killed by remote request\n");
1230 SendOkay(reply_fd);
Josh Gaoee356a72017-05-08 18:37:17 -07001231
Josh Gaobea8f3c2020-04-06 12:07:56 -07001232 // Rely on process exit to close the socket for us.
1233 exit(0);
1234 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001235 }
1236
Josh Gaob13f3cd2019-02-20 20:37:26 -08001237 LOG(DEBUG) << "handle_host_request(" << service << ")";
1238
1239 // Transport selection:
1240 if (service.starts_with("transport") || service.starts_with("tport:")) {
Elliott Hughes3aec2ba2015-05-05 13:10:43 -07001241 TransportType type = kTransportAny;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001242
Josh Gaocd30bb32019-02-20 19:30:59 -08001243 std::string serial_storage;
Josh Gaob13f3cd2019-02-20 20:37:26 -08001244 bool legacy = true;
Josh Gaocd30bb32019-02-20 19:30:59 -08001245
Josh Gaob13f3cd2019-02-20 20:37:26 -08001246 // New transport selection protocol:
1247 // This is essentially identical to the previous version, except it returns the selected
1248 // transport id to the caller as well.
Elliott Hughes0d1e8fd2019-05-03 09:02:45 -07001249 if (android::base::ConsumePrefix(&service, "tport:")) {
Josh Gaob13f3cd2019-02-20 20:37:26 -08001250 legacy = false;
Elliott Hughes0d1e8fd2019-05-03 09:02:45 -07001251 if (android::base::ConsumePrefix(&service, "serial:")) {
Josh Gaob13f3cd2019-02-20 20:37:26 -08001252 serial_storage = service;
1253 serial = serial_storage.c_str();
1254 } else if (service == "usb") {
1255 type = kTransportUsb;
1256 } else if (service == "local") {
1257 type = kTransportLocal;
1258 } else if (service == "any") {
1259 type = kTransportAny;
Josh Gaob39e4152017-08-16 16:57:01 -07001260 }
Josh Gaob13f3cd2019-02-20 20:37:26 -08001261
1262 // Selection by id is unimplemented, since you obviously already know the transport id
1263 // you're connecting to.
1264 } else {
Elliott Hughes0d1e8fd2019-05-03 09:02:45 -07001265 if (android::base::ConsumePrefix(&service, "transport-id:")) {
Josh Gaob13f3cd2019-02-20 20:37:26 -08001266 if (!ParseUint(&transport_id, service)) {
1267 SendFail(reply_fd, "invalid transport id");
1268 return HostRequestResult::Handled;
1269 }
1270 } else if (service == "transport-usb") {
1271 type = kTransportUsb;
1272 } else if (service == "transport-local") {
1273 type = kTransportLocal;
1274 } else if (service == "transport-any") {
1275 type = kTransportAny;
Elliott Hughes0d1e8fd2019-05-03 09:02:45 -07001276 } else if (android::base::ConsumePrefix(&service, "transport:")) {
Josh Gaob13f3cd2019-02-20 20:37:26 -08001277 serial_storage = service;
1278 serial = serial_storage.c_str();
1279 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001280 }
1281
Elliott Hughes67943d12015-10-07 14:55:10 -07001282 std::string error;
Josh Gaob39e4152017-08-16 16:57:01 -07001283 atransport* t = acquire_one_transport(type, serial, transport_id, nullptr, &error);
Elliott Hughes24f52762015-06-23 13:00:32 -07001284 if (t != nullptr) {
1285 s->transport = t;
Elliott Hughes88b4c852015-04-30 17:32:03 -07001286 SendOkay(reply_fd);
Josh Gaod3067472018-08-07 14:14:21 -07001287
Josh Gaob13f3cd2019-02-20 20:37:26 -08001288 if (!legacy) {
1289 // Nothing we can do if this fails.
1290 WriteFdExactly(reply_fd, &t->id, sizeof(t->id));
1291 }
1292
1293 return HostRequestResult::SwitchedTransport;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001294 } else {
Elliott Hughes67943d12015-10-07 14:55:10 -07001295 SendFail(reply_fd, error);
Josh Gaob13f3cd2019-02-20 20:37:26 -08001296 return HostRequestResult::Handled;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001297 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001298 }
1299
1300 // return a list of all connected devices
Josh Gaocd30bb32019-02-20 19:30:59 -08001301 if (service == "devices" || service == "devices-l") {
1302 bool long_listing = service == "devices-l";
1303 D("Getting device list...");
1304 std::string device_list = list_transports(long_listing);
1305 D("Sending device list...");
1306 SendOkay(reply_fd, device_list);
Josh Gaob13f3cd2019-02-20 20:37:26 -08001307 return HostRequestResult::Handled;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001308 }
1309
Josh Gaocd30bb32019-02-20 19:30:59 -08001310 if (service == "reconnect-offline") {
Josh Gao4e562502016-10-27 14:01:08 -07001311 std::string response;
1312 close_usb_devices([&response](const atransport* transport) {
Josh Gao7a7c5cb2018-05-04 16:04:49 -07001313 if (!ConnectionStateIsOnline(transport->GetConnectionState())) {
1314 response += "reconnecting " + transport->serial_name() + "\n";
1315 return true;
Josh Gao4e562502016-10-27 14:01:08 -07001316 }
Josh Gao7a7c5cb2018-05-04 16:04:49 -07001317 return false;
Josh Gao3a2172b2019-03-28 15:47:44 -07001318 }, true);
Josh Gao4e562502016-10-27 14:01:08 -07001319 if (!response.empty()) {
1320 response.resize(response.size() - 1);
1321 }
1322 SendOkay(reply_fd, response);
Josh Gaob13f3cd2019-02-20 20:37:26 -08001323 return HostRequestResult::Handled;
Josh Gao4e562502016-10-27 14:01:08 -07001324 }
1325
Josh Gaocd30bb32019-02-20 19:30:59 -08001326 if (service == "features") {
Elliott Hughes67943d12015-10-07 14:55:10 -07001327 std::string error;
Josh Gao07790752019-09-13 00:12:26 +08001328 atransport* t =
1329 s->transport ? s->transport
1330 : acquire_one_transport(type, serial, transport_id, nullptr, &error);
David Pursell2e9f0322015-09-04 16:40:30 -07001331 if (t != nullptr) {
David Pursella07dbad2015-09-22 10:43:08 -07001332 SendOkay(reply_fd, FeatureSetToString(t->features()));
David Pursell2e9f0322015-09-04 16:40:30 -07001333 } else {
Elliott Hughes67943d12015-10-07 14:55:10 -07001334 SendFail(reply_fd, error);
David Pursell2e9f0322015-09-04 16:40:30 -07001335 }
Josh Gaob13f3cd2019-02-20 20:37:26 -08001336 return HostRequestResult::Handled;
David Pursell2e9f0322015-09-04 16:40:30 -07001337 }
1338
Josh Gaocd30bb32019-02-20 19:30:59 -08001339 if (service == "host-features") {
Josh Gao210b63f2017-02-22 17:07:01 -08001340 FeatureSet features = supported_features();
1341 // Abuse features to report libusb status.
1342 if (should_use_libusb()) {
Yurii Zubrytskyi816c2d22020-03-26 18:19:28 -07001343 features.emplace_back(kFeatureLibusb);
Josh Gao210b63f2017-02-22 17:07:01 -08001344 }
Yurii Zubrytskyi816c2d22020-03-26 18:19:28 -07001345 features.emplace_back(kFeaturePushSync);
Josh Gao210b63f2017-02-22 17:07:01 -08001346 SendOkay(reply_fd, FeatureSetToString(features));
Josh Gaob13f3cd2019-02-20 20:37:26 -08001347 return HostRequestResult::Handled;
Josh Gao210b63f2017-02-22 17:07:01 -08001348 }
Josh Gao210b63f2017-02-22 17:07:01 -08001349
Mike Lockwood5f2c4a12009-10-11 23:04:18 -04001350 // remove TCP transport
Josh Gaocd30bb32019-02-20 19:30:59 -08001351 if (service.starts_with("disconnect:")) {
1352 std::string address(service.substr(11));
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001353 if (address.empty()) {
Yabin Cui4d64fd82015-08-27 12:03:11 -07001354 kick_all_tcp_devices();
Josh Gaod3067472018-08-07 14:14:21 -07001355 SendOkay(reply_fd, "disconnected everything");
Josh Gaob13f3cd2019-02-20 20:37:26 -08001356 return HostRequestResult::Handled;
Mike Lockwood5f2c4a12009-10-11 23:04:18 -04001357 }
1358
Joshua Duongc82a2802021-05-03 10:11:40 -07001359 // Mdns instance named device
1360 atransport* t = find_transport(address.c_str());
1361 if (t != nullptr) {
1362 kick_transport(t);
1363 SendOkay(reply_fd, android::base::StringPrintf("disconnected %s", address.c_str()));
1364 return HostRequestResult::Handled;
1365 }
1366
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001367 std::string serial;
1368 std::string host;
1369 int port = DEFAULT_ADB_LOCAL_TRANSPORT_PORT;
1370 std::string error;
Jason Jeremy Iman84613872019-07-19 12:44:39 +09001371 if (address.starts_with("vsock:") || address.starts_with("localfilesystem:")) {
Cody Schuffelen637aaf52019-01-04 18:51:11 -08001372 serial = address;
1373 } else if (!android::base::ParseNetAddress(address, &host, &port, &serial, &error)) {
Josh Gaod3067472018-08-07 14:14:21 -07001374 SendFail(reply_fd, android::base::StringPrintf("couldn't parse '%s': %s",
1375 address.c_str(), error.c_str()));
Josh Gaob13f3cd2019-02-20 20:37:26 -08001376 return HostRequestResult::Handled;
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001377 }
Joshua Duongc82a2802021-05-03 10:11:40 -07001378 t = find_transport(serial.c_str());
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001379 if (t == nullptr) {
Josh Gaod3067472018-08-07 14:14:21 -07001380 SendFail(reply_fd, android::base::StringPrintf("no such device '%s'", serial.c_str()));
Josh Gaob13f3cd2019-02-20 20:37:26 -08001381 return HostRequestResult::Handled;
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001382 }
Yabin Cui4d64fd82015-08-27 12:03:11 -07001383 kick_transport(t);
Josh Gaod3067472018-08-07 14:14:21 -07001384 SendOkay(reply_fd, android::base::StringPrintf("disconnected %s", address.c_str()));
Josh Gaob13f3cd2019-02-20 20:37:26 -08001385 return HostRequestResult::Handled;
Mike Lockwoodb51ae572009-08-24 15:58:40 -07001386 }
1387
Elliott Hughes67943d12015-10-07 14:55:10 -07001388 // Returns our value for ADB_SERVER_VERSION.
Josh Gaocd30bb32019-02-20 19:30:59 -08001389 if (service == "version") {
Josh Gaod3067472018-08-07 14:14:21 -07001390 SendOkay(reply_fd, android::base::StringPrintf("%04x", ADB_SERVER_VERSION));
Josh Gaob13f3cd2019-02-20 20:37:26 -08001391 return HostRequestResult::Handled;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001392 }
1393
Elliott Hughes24f52762015-06-23 13:00:32 -07001394 // These always report "unknown" rather than the actual error, for scripts.
Josh Gaocd30bb32019-02-20 19:30:59 -08001395 if (service == "get-serialno") {
Elliott Hughes67943d12015-10-07 14:55:10 -07001396 std::string error;
Josh Gao07790752019-09-13 00:12:26 +08001397 atransport* t =
1398 s->transport ? s->transport
1399 : acquire_one_transport(type, serial, transport_id, nullptr, &error);
Elliott Hughes67943d12015-10-07 14:55:10 -07001400 if (t) {
Josh Gaod3067472018-08-07 14:14:21 -07001401 SendOkay(reply_fd, !t->serial.empty() ? t->serial : "unknown");
Elliott Hughes67943d12015-10-07 14:55:10 -07001402 } else {
Josh Gaod3067472018-08-07 14:14:21 -07001403 SendFail(reply_fd, error);
Elliott Hughes67943d12015-10-07 14:55:10 -07001404 }
Josh Gaob13f3cd2019-02-20 20:37:26 -08001405 return HostRequestResult::Handled;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001406 }
Josh Gaocd30bb32019-02-20 19:30:59 -08001407 if (service == "get-devpath") {
Elliott Hughes67943d12015-10-07 14:55:10 -07001408 std::string error;
Josh Gao07790752019-09-13 00:12:26 +08001409 atransport* t =
1410 s->transport ? s->transport
1411 : acquire_one_transport(type, serial, transport_id, nullptr, &error);
Elliott Hughes67943d12015-10-07 14:55:10 -07001412 if (t) {
Josh Gaod3067472018-08-07 14:14:21 -07001413 SendOkay(reply_fd, !t->devpath.empty() ? t->devpath : "unknown");
Elliott Hughes67943d12015-10-07 14:55:10 -07001414 } else {
Josh Gaod3067472018-08-07 14:14:21 -07001415 SendFail(reply_fd, error);
Elliott Hughes67943d12015-10-07 14:55:10 -07001416 }
Josh Gaob13f3cd2019-02-20 20:37:26 -08001417 return HostRequestResult::Handled;
Scott Anderson6dfaf4b2012-04-20 11:21:14 -07001418 }
Josh Gaocd30bb32019-02-20 19:30:59 -08001419 if (service == "get-state") {
Elliott Hughes67943d12015-10-07 14:55:10 -07001420 std::string error;
Josh Gao07790752019-09-13 00:12:26 +08001421 atransport* t =
1422 s->transport ? s->transport
1423 : acquire_one_transport(type, serial, transport_id, nullptr, &error);
Elliott Hughes67943d12015-10-07 14:55:10 -07001424 if (t) {
Josh Gao6256d992021-05-20 18:28:13 -07001425 SendOkay(reply_fd, to_string(t->GetConnectionState()));
Elliott Hughes67943d12015-10-07 14:55:10 -07001426 } else {
Josh Gaod3067472018-08-07 14:14:21 -07001427 SendFail(reply_fd, error);
Elliott Hughes67943d12015-10-07 14:55:10 -07001428 }
Josh Gaob13f3cd2019-02-20 20:37:26 -08001429 return HostRequestResult::Handled;
Elliott Hughes24f52762015-06-23 13:00:32 -07001430 }
1431
Elliott Hughes67943d12015-10-07 14:55:10 -07001432 // Indicates a new emulator instance has started.
Elliott Hughes0d1e8fd2019-05-03 09:02:45 -07001433 if (android::base::ConsumePrefix(&service, "emulator:")) {
Josh Gaocd30bb32019-02-20 19:30:59 -08001434 unsigned int port;
1435 if (!ParseUint(&port, service)) {
1436 LOG(ERROR) << "received invalid port for emulator: " << service;
1437 } else {
1438 local_connect(port);
1439 }
1440
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001441 /* we don't even need to send a reply */
Josh Gaob13f3cd2019-02-20 20:37:26 -08001442 return HostRequestResult::Handled;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001443 }
Yabin Cuid78ed222016-04-05 13:50:44 -07001444
Josh Gaocd30bb32019-02-20 19:30:59 -08001445 if (service == "reconnect") {
Yabin Cui3cf1b362017-03-10 16:01:01 -08001446 std::string response;
Josh Gao07790752019-09-13 00:12:26 +08001447 atransport* t = s->transport ? s->transport
1448 : acquire_one_transport(type, serial, transport_id, nullptr,
1449 &response, true);
Yabin Cui3cf1b362017-03-10 16:01:01 -08001450 if (t != nullptr) {
Josh Gao3a2172b2019-03-28 15:47:44 -07001451 kick_transport(t, true);
Josh Gao6256d992021-05-20 18:28:13 -07001452 response = "reconnecting " + t->serial_name() + " [" +
1453 to_string(t->GetConnectionState()) + "]\n";
Yabin Cuid78ed222016-04-05 13:50:44 -07001454 }
Josh Gaod3067472018-08-07 14:14:21 -07001455 SendOkay(reply_fd, response);
Josh Gaob13f3cd2019-02-20 20:37:26 -08001456 return HostRequestResult::Handled;
Yabin Cuid78ed222016-04-05 13:50:44 -07001457 }
Simon Yee51c11e2014-07-14 17:23:06 -07001458
Josh Gao8e7c9722021-06-17 04:19:45 -07001459 if (service == "attach") {
1460 std::string error;
1461 atransport* t = s->transport ? s->transport
1462 : acquire_one_transport(type, serial, transport_id, nullptr,
1463 &error, true);
1464 if (!t) {
1465 SendFail(reply_fd, error);
1466 return HostRequestResult::Handled;
1467 }
1468
1469 if (t->Attach(&error)) {
1470 SendOkay(reply_fd,
1471 android::base::StringPrintf("%s attached", t->serial_name().c_str()));
1472 } else {
1473 SendFail(reply_fd, error);
1474 }
1475 return HostRequestResult::Handled;
1476 }
1477
1478 if (service == "detach") {
1479 std::string error;
1480 atransport* t = s->transport ? s->transport
1481 : acquire_one_transport(type, serial, transport_id, nullptr,
1482 &error, true);
1483 if (!t) {
1484 SendFail(reply_fd, error);
1485 return HostRequestResult::Handled;
1486 }
1487
1488 // HACK:
1489 // Detaching the transport will lead to all of its sockets being closed,
1490 // but we're handling one of those sockets right now!
1491 //
1492 // Mark the socket as not having a transport, knowing that it'll be cleaned up by the
1493 // function that called us.
1494 s->transport = nullptr;
1495
1496 if (t->Detach(&error)) {
1497 SendOkay(reply_fd,
1498 android::base::StringPrintf("%s detached", t->serial_name().c_str()));
1499 } else {
1500 SendFail(reply_fd, error);
1501 }
1502 return HostRequestResult::Handled;
1503 }
1504
Josh Gaocd30bb32019-02-20 19:30:59 -08001505 // TODO: Switch handle_forward_request to string_view.
1506 std::string service_str(service);
Josh Gao07790752019-09-13 00:12:26 +08001507 auto transport_acquirer = [=](std::string* error) {
1508 if (s->transport) {
1509 return s->transport;
1510 } else {
1511 std::string error;
1512 return acquire_one_transport(type, serial, transport_id, nullptr, &error);
1513 }
1514 };
1515 if (handle_forward_request(service_str.c_str(), transport_acquirer, reply_fd)) {
Josh Gaob13f3cd2019-02-20 20:37:26 -08001516 return HostRequestResult::Handled;
Josh Gao7c3b2622018-07-19 17:59:45 -07001517 }
1518
Joshua Duonge22e1612020-03-31 10:58:50 -07001519 if (handle_mdns_request(service, reply_fd)) {
1520 return HostRequestResult::Handled;
1521 }
1522
Josh Gaob13f3cd2019-02-20 20:37:26 -08001523 return HostRequestResult::Unhandled;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001524}
Josh Gao1e3bf732017-05-03 22:37:10 -07001525
1526static auto& init_mutex = *new std::mutex();
1527static auto& init_cv = *new std::condition_variable();
1528static bool device_scan_complete = false;
1529static bool transports_ready = false;
1530
1531void update_transport_status() {
1532 bool result = iterate_transports([](const atransport* t) {
1533 if (t->type == kTransportUsb && t->online != 1) {
1534 return false;
1535 }
1536 return true;
1537 });
1538
Josh Gao1e3bf732017-05-03 22:37:10 -07001539 bool ready;
Josh Gao1e3bf732017-05-03 22:37:10 -07001540 {
1541 std::lock_guard<std::mutex> lock(init_mutex);
1542 transports_ready = result;
1543 ready = transports_ready && device_scan_complete;
1544 }
1545
1546 if (ready) {
Josh Gao1e3bf732017-05-03 22:37:10 -07001547 init_cv.notify_all();
1548 }
1549}
1550
1551void adb_notify_device_scan_complete() {
Josh Gao1e3bf732017-05-03 22:37:10 -07001552 {
1553 std::lock_guard<std::mutex> lock(init_mutex);
Josh Gao4b640842017-05-31 11:54:56 -07001554 if (device_scan_complete) {
1555 return;
1556 }
1557
Josh Gao1e3bf732017-05-03 22:37:10 -07001558 device_scan_complete = true;
1559 }
1560
1561 update_transport_status();
1562}
1563
1564void adb_wait_for_device_initialization() {
1565 std::unique_lock<std::mutex> lock(init_mutex);
1566 init_cv.wait_for(lock, 3s, []() { return device_scan_complete && transports_ready; });
1567}
1568
Yabin Cui3cf1b362017-03-10 16:01:01 -08001569#endif // ADB_HOST