blob: 725aa9193d38df8989e7c4816d5488d560aaf266 [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#include "adb.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080021
Dan Albertea2175a2015-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 Projectdd7bc332009-03-03 19:32:55 -080027#include <stdio.h>
28#include <stdlib.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080029#include <string.h>
Mike Lockwood1f546e62009-05-25 18:17:55 -040030#include <sys/time.h>
Dan Albertea2175a2015-03-08 21:12:08 -070031#include <time.h>
32
33#include <string>
Dan Alberte2462192015-03-19 13:25:27 -070034#include <vector>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080035
David Pursell5f787ed2016-01-27 08:52:53 -080036#include <android-base/errors.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080037#include <android-base/logging.h>
38#include <android-base/macros.h>
David Pursell706955f2016-01-21 08:40:59 -080039#include <android-base/parsenetaddress.h>
Elliott Hughes4f713192015-12-04 22:00:26 -080040#include <android-base/stringprintf.h>
41#include <android-base/strings.h>
Elliott Hughes7b506092015-04-20 08:09:20 -070042
Benoit Gobyd5fcafa2012-04-12 12:23:49 -070043#include "adb_auth.h"
Dan Albertcc731cc2015-02-24 21:26:58 -080044#include "adb_io.h"
Dan Alberte9fca142015-02-18 18:03:26 -080045#include "adb_listeners.h"
Elliott Hughes3d5f60d2015-07-18 12:21:30 -070046#include "adb_utils.h"
Dan Albert76649012015-02-24 15:51:19 -080047#include "transport.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080048
49#if !ADB_HOST
Nick Kralevich893a4a42013-05-23 09:54:13 -070050#include <cutils/properties.h>
Nick Kraleviche2864bf2013-02-28 14:12:58 -080051#include <sys/capability.h>
Jeff Sharkey885342a2012-08-14 21:00:22 -070052#include <sys/mount.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080053#endif
54
Elliott Hughes42ae2602015-08-12 08:32:10 -070055std::string adb_version() {
56 // Don't change the format of this --- it's parsed by ddmlib.
57 return android::base::StringPrintf("Android Debug Bridge version %d.%d.%d\n"
58 "Revision %s\n",
59 ADB_VERSION_MAJOR, ADB_VERSION_MINOR, ADB_SERVER_VERSION,
60 ADB_REVISION);
61}
62
Dan Albert9313c0d2015-05-21 13:58:50 -070063void fatal(const char *fmt, ...) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080064 va_list ap;
65 va_start(ap, fmt);
Josh Gao4abdeee2016-05-13 18:16:43 -070066 char buf[1024];
67 vsnprintf(buf, sizeof(buf), fmt, ap);
68
69#if ADB_HOST
70 fprintf(stderr, "error: %s\n", buf);
71#else
72 LOG(ERROR) << "error: " << buf;
73#endif
74
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080075 va_end(ap);
Josh Gao4abdeee2016-05-13 18:16:43 -070076 abort();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080077}
78
Dan Albert9313c0d2015-05-21 13:58:50 -070079void fatal_errno(const char* fmt, ...) {
Josh Gao4abdeee2016-05-13 18:16:43 -070080 int err = errno;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080081 va_list ap;
82 va_start(ap, fmt);
Josh Gao4abdeee2016-05-13 18:16:43 -070083 char buf[1024];
84 vsnprintf(buf, sizeof(buf), fmt, ap);
85
86#if ADB_HOST
87 fprintf(stderr, "error: %s: %s\n", buf, strerror(err));
88#else
89 LOG(ERROR) << "error: " << buf << ": " << strerror(err);
90#endif
91
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080092 va_end(ap);
Josh Gao4abdeee2016-05-13 18:16:43 -070093 abort();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080094}
95
Dan Albertbac34742015-02-25 17:51:28 -080096apacket* get_apacket(void)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080097{
Dan Albertbac34742015-02-25 17:51:28 -080098 apacket* p = reinterpret_cast<apacket*>(malloc(sizeof(apacket)));
99 if (p == nullptr) {
100 fatal("failed to allocate an apacket");
101 }
102
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800103 memset(p, 0, sizeof(apacket) - MAX_PAYLOAD);
104 return p;
105}
106
107void put_apacket(apacket *p)
108{
109 free(p);
110}
111
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700112void handle_online(atransport *t)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800113{
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700114 D("adb: online");
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700115 t->online = 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800116}
117
118void handle_offline(atransport *t)
119{
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700120 D("adb: offline");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800121 //Close the associated usb
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700122 t->online = 0;
Yabin Cui34f45c52015-08-26 11:18:42 -0700123
Yabin Cuib3298242015-08-28 15:09:44 -0700124 // This is necessary to avoid a race condition that occurred when a transport closes
Yabin Cui34f45c52015-08-26 11:18:42 -0700125 // while a client socket is still active.
126 close_all_sockets(t);
127
Yabin Cuib3298242015-08-28 15:09:44 -0700128 t->RunDisconnects();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800129}
130
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700131#if DEBUG_PACKETS
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800132#define DUMPMAX 32
133void print_packet(const char *label, apacket *p)
134{
135 char *tag;
136 char *x;
137 unsigned count;
138
139 switch(p->msg.command){
140 case A_SYNC: tag = "SYNC"; break;
141 case A_CNXN: tag = "CNXN" ; break;
142 case A_OPEN: tag = "OPEN"; break;
143 case A_OKAY: tag = "OKAY"; break;
144 case A_CLSE: tag = "CLSE"; break;
145 case A_WRTE: tag = "WRTE"; break;
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700146 case A_AUTH: tag = "AUTH"; break;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800147 default: tag = "????"; break;
148 }
149
150 fprintf(stderr, "%s: %s %08x %08x %04x \"",
151 label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
152 count = p->msg.data_length;
153 x = (char*) p->data;
154 if(count > DUMPMAX) {
155 count = DUMPMAX;
156 tag = "\n";
157 } else {
158 tag = "\"\n";
159 }
160 while(count-- > 0){
161 if((*x >= ' ') && (*x < 127)) {
162 fputc(*x, stderr);
163 } else {
164 fputc('.', stderr);
165 }
166 x++;
167 }
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700168 fputs(tag, stderr);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800169}
170#endif
171
172static void send_ready(unsigned local, unsigned remote, atransport *t)
173{
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700174 D("Calling send_ready");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800175 apacket *p = get_apacket();
176 p->msg.command = A_OKAY;
177 p->msg.arg0 = local;
178 p->msg.arg1 = remote;
179 send_packet(p, t);
180}
181
182static void send_close(unsigned local, unsigned remote, atransport *t)
183{
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700184 D("Calling send_close");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800185 apacket *p = get_apacket();
186 p->msg.command = A_CLSE;
187 p->msg.arg0 = local;
188 p->msg.arg1 = remote;
189 send_packet(p, t);
190}
191
Dan Albert1792c232015-05-18 13:06:53 -0700192std::string get_connection_string() {
193 std::vector<std::string> connection_properties;
194
195#if !ADB_HOST
196 static const char* cnxn_props[] = {
Scott Andersone82c2db2012-05-25 14:10:02 -0700197 "ro.product.name",
198 "ro.product.model",
199 "ro.product.device",
200 };
Scott Andersone82c2db2012-05-25 14:10:02 -0700201
Dan Albert1792c232015-05-18 13:06:53 -0700202 for (const auto& prop_name : cnxn_props) {
Scott Andersone82c2db2012-05-25 14:10:02 -0700203 char value[PROPERTY_VALUE_MAX];
Dan Albert1792c232015-05-18 13:06:53 -0700204 property_get(prop_name, value, "");
205 connection_properties.push_back(
206 android::base::StringPrintf("%s=%s", prop_name, value));
Scott Andersone82c2db2012-05-25 14:10:02 -0700207 }
Scott Andersone82c2db2012-05-25 14:10:02 -0700208#endif
Dan Albert1792c232015-05-18 13:06:53 -0700209
210 connection_properties.push_back(android::base::StringPrintf(
David Pursell4e2fd362015-09-22 10:43:08 -0700211 "features=%s", FeatureSetToString(supported_features()).c_str()));
Dan Albert1792c232015-05-18 13:06:53 -0700212
213 return android::base::StringPrintf(
214 "%s::%s", adb_device_banner,
215 android::base::Join(connection_properties, ';').c_str());
Scott Andersone82c2db2012-05-25 14:10:02 -0700216}
217
Dan Albert1792c232015-05-18 13:06:53 -0700218void send_connect(atransport* t) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700219 D("Calling send_connect");
Dan Albert1792c232015-05-18 13:06:53 -0700220 apacket* cp = get_apacket();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800221 cp->msg.command = A_CNXN;
Tamas Berghammer3d2904c2015-07-13 19:12:28 +0100222 cp->msg.arg0 = t->get_protocol_version();
223 cp->msg.arg1 = t->get_max_payload();
Dan Albert1792c232015-05-18 13:06:53 -0700224
225 std::string connection_str = get_connection_string();
226 // Connect and auth packets are limited to MAX_PAYLOAD_V1 because we don't
227 // yet know how much data the other size is willing to accept.
228 if (connection_str.length() > MAX_PAYLOAD_V1) {
229 LOG(FATAL) << "Connection banner is too long (length = "
230 << connection_str.length() << ")";
231 }
232
233 memcpy(cp->data, connection_str.c_str(), connection_str.length());
234 cp->msg.data_length = connection_str.length();
235
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800236 send_packet(cp, t);
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700237}
238
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700239// qual_overwrite is used to overwrite a qualifier string. dst is a
240// pointer to a char pointer. It is assumed that if *dst is non-NULL, it
241// was malloc'ed and needs to freed. *dst will be set to a dup of src.
242// TODO: switch to std::string for these atransport fields instead.
243static void qual_overwrite(char** dst, const std::string& src) {
Scott Andersone82c2db2012-05-25 14:10:02 -0700244 free(*dst);
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700245 *dst = strdup(src.c_str());
Scott Andersone82c2db2012-05-25 14:10:02 -0700246}
247
Dan Albert1792c232015-05-18 13:06:53 -0700248void parse_banner(const std::string& banner, atransport* t) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700249 D("parse_banner: %s", banner.c_str());
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700250
251 // The format is something like:
252 // "device::ro.product.name=x;ro.product.model=y;ro.product.device=z;".
253 std::vector<std::string> pieces = android::base::Split(banner, ":");
254
David Purselld2b588e2015-09-25 13:04:21 -0700255 // Reset the features list or else if the server sends no features we may
256 // keep the existing feature set (http://b/24405971).
257 t->SetFeatures("");
258
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700259 if (pieces.size() > 2) {
260 const std::string& props = pieces[2];
Elliott Hughes65fe2512015-10-07 15:59:35 -0700261 for (const auto& prop : android::base::Split(props, ";")) {
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700262 // The list of properties was traditionally ;-terminated rather than ;-separated.
263 if (prop.empty()) continue;
264
265 std::vector<std::string> key_value = android::base::Split(prop, "=");
266 if (key_value.size() != 2) continue;
267
268 const std::string& key = key_value[0];
269 const std::string& value = key_value[1];
270 if (key == "ro.product.name") {
271 qual_overwrite(&t->product, value);
272 } else if (key == "ro.product.model") {
273 qual_overwrite(&t->model, value);
274 } else if (key == "ro.product.device") {
275 qual_overwrite(&t->device, value);
Dan Albert1792c232015-05-18 13:06:53 -0700276 } else if (key == "features") {
David Pursell4e2fd362015-09-22 10:43:08 -0700277 t->SetFeatures(value);
Scott Andersone82c2db2012-05-25 14:10:02 -0700278 }
279 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800280 }
281
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700282 const std::string& type = pieces[0];
283 if (type == "bootloader") {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700284 D("setting connection_state to kCsBootloader");
Dan Albertdcd78a12015-05-18 16:43:57 -0700285 t->connection_state = kCsBootloader;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800286 update_transports();
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700287 } else if (type == "device") {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700288 D("setting connection_state to kCsDevice");
Dan Albertdcd78a12015-05-18 16:43:57 -0700289 t->connection_state = kCsDevice;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800290 update_transports();
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700291 } else if (type == "recovery") {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700292 D("setting connection_state to kCsRecovery");
Dan Albertdcd78a12015-05-18 16:43:57 -0700293 t->connection_state = kCsRecovery;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800294 update_transports();
Elliott Hughes8d5fa6d2015-04-24 23:02:00 -0700295 } else if (type == "sideload") {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700296 D("setting connection_state to kCsSideload");
Dan Albertdcd78a12015-05-18 16:43:57 -0700297 t->connection_state = kCsSideload;
Doug Zongker447f0612012-01-09 14:54:53 -0800298 update_transports();
Elliott Hughes3ce95752015-04-29 22:37:25 -0700299 } else {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700300 D("setting connection_state to kCsHost");
Dan Albertdcd78a12015-05-18 16:43:57 -0700301 t->connection_state = kCsHost;
Doug Zongker447f0612012-01-09 14:54:53 -0800302 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800303}
304
Dan Albert1792c232015-05-18 13:06:53 -0700305static void handle_new_connection(atransport* t, apacket* p) {
306 if (t->connection_state != kCsOffline) {
307 t->connection_state = kCsOffline;
308 handle_offline(t);
309 }
310
311 t->update_version(p->msg.arg0, p->msg.arg1);
312 std::string banner(reinterpret_cast<const char*>(p->data),
313 p->msg.data_length);
314 parse_banner(banner, t);
315
316#if ADB_HOST
317 handle_online(t);
318#else
319 if (!auth_required) {
320 handle_online(t);
321 send_connect(t);
322 } else {
323 send_auth_request(t);
324 }
325#endif
326}
327
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800328void handle_packet(apacket *p, atransport *t)
329{
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700330 D("handle_packet() %c%c%c%c", ((char*) (&(p->msg.command)))[0],
Viral Mehta899913f2010-06-16 18:41:28 +0530331 ((char*) (&(p->msg.command)))[1],
332 ((char*) (&(p->msg.command)))[2],
333 ((char*) (&(p->msg.command)))[3]);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800334 print_packet("recv", p);
335
336 switch(p->msg.command){
337 case A_SYNC:
Elliott Hughesffc73a32016-06-15 14:46:56 -0700338 if (p->msg.arg0){
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800339 send_packet(p, t);
Yabin Cui661327e2015-08-11 13:40:42 -0700340#if ADB_HOST
341 send_connect(t);
342#endif
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800343 } else {
Dan Albertdcd78a12015-05-18 16:43:57 -0700344 t->connection_state = kCsOffline;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800345 handle_offline(t);
346 send_packet(p, t);
347 }
348 return;
349
Dan Albert1792c232015-05-18 13:06:53 -0700350 case A_CNXN: // CONNECT(version, maxdata, "system-id-string")
351 handle_new_connection(t, p);
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700352 break;
353
354 case A_AUTH:
355 if (p->msg.arg0 == ADB_AUTH_TOKEN) {
Dan Albertdcd78a12015-05-18 16:43:57 -0700356 t->connection_state = kCsUnauthorized;
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700357 t->key = adb_auth_nextkey(t->key);
358 if (t->key) {
359 send_auth_response(p->data, p->msg.data_length, t);
360 } else {
361 /* No more private keys to try, send the public key */
362 send_auth_publickey(t);
363 }
364 } else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
Mattias Nissler097b6bb2016-03-31 16:32:09 +0200365 if (adb_auth_verify(t->token, sizeof(t->token),
366 p->data, p->msg.data_length)) {
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700367 adb_auth_verified(t);
368 t->failed_auth_attempts = 0;
369 } else {
370 if (t->failed_auth_attempts++ > 10)
371 adb_sleep_ms(1000);
372 send_auth_request(t);
373 }
374 } else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
375 adb_auth_confirm_key(p->data, p->msg.data_length, t);
376 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800377 break;
378
379 case A_OPEN: /* OPEN(local-id, 0, "destination") */
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100380 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800381 char *name = (char*) p->data;
382 name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
Elliott Hughesffc73a32016-06-15 14:46:56 -0700383 asocket* s = create_local_service_socket(name, t);
384 if (s == nullptr) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800385 send_close(0, p->msg.arg0, t);
386 } else {
387 s->peer = create_remote_socket(p->msg.arg0, t);
388 s->peer->peer = s;
389 send_ready(s->id, s->peer->id, t);
390 s->ready(s);
391 }
392 }
393 break;
394
395 case A_OKAY: /* READY(local-id, remote-id, "") */
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100396 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
Elliott Hughesffc73a32016-06-15 14:46:56 -0700397 asocket* s = find_local_socket(p->msg.arg1, 0);
398 if (s) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800399 if(s->peer == 0) {
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100400 /* On first READY message, create the connection. */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800401 s->peer = create_remote_socket(p->msg.arg0, t);
402 s->peer->peer = s;
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100403 s->ready(s);
404 } else if (s->peer->id == p->msg.arg0) {
405 /* Other READY messages must use the same local-id */
406 s->ready(s);
407 } else {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700408 D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s",
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100409 p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800410 }
Yabin Cuiaa77e222015-09-29 12:25:33 -0700411 } else {
412 // When receiving A_OKAY from device for A_OPEN request, the host server may
413 // have closed the local socket because of client disconnection. Then we need
414 // to send A_CLSE back to device to close the service on device.
415 send_close(p->msg.arg1, p->msg.arg0, t);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800416 }
417 }
418 break;
419
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100420 case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
421 if (t->online && p->msg.arg1 != 0) {
Elliott Hughesffc73a32016-06-15 14:46:56 -0700422 asocket* s = find_local_socket(p->msg.arg1, p->msg.arg0);
423 if (s) {
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100424 /* According to protocol.txt, p->msg.arg0 might be 0 to indicate
425 * a failed OPEN only. However, due to a bug in previous ADB
426 * versions, CLOSE(0, remote-id, "") was also used for normal
427 * CLOSE() operations.
428 *
429 * This is bad because it means a compromised adbd could
430 * send packets to close connections between the host and
431 * other devices. To avoid this, only allow this if the local
432 * socket has a peer on the same transport.
433 */
434 if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700435 D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s",
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100436 p->msg.arg1, t->serial, s->peer->transport->serial);
437 } else {
438 s->close(s);
439 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800440 }
441 }
442 break;
443
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100444 case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
445 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
Elliott Hughesffc73a32016-06-15 14:46:56 -0700446 asocket* s = find_local_socket(p->msg.arg1, p->msg.arg0);
447 if (s) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800448 unsigned rid = p->msg.arg0;
449 p->len = p->msg.data_length;
450
Elliott Hughesffc73a32016-06-15 14:46:56 -0700451 if (s->enqueue(s, p) == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -0700452 D("Enqueue the socket");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800453 send_ready(s->id, rid, t);
454 }
455 return;
456 }
457 }
458 break;
459
460 default:
461 printf("handle_packet: what is %08x?!\n", p->msg.command);
462 }
463
464 put_apacket(p);
465}
466
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800467#if ADB_HOST
JP Abgrall571c1362012-12-06 18:18:12 -0800468
Spencer Low2122c7a2015-08-26 18:46:09 -0700469#ifdef _WIN32
470
Spencer Lowbc36bdb2015-09-16 16:40:06 -0700471// Try to make a handle non-inheritable and if there is an error, don't output
472// any error info, but leave GetLastError() for the caller to read. This is
473// convenient if the caller is expecting that this may fail and they'd like to
474// ignore such a failure.
475static bool _try_make_handle_noninheritable(HANDLE h) {
Spencer Low2122c7a2015-08-26 18:46:09 -0700476 if (h != INVALID_HANDLE_VALUE && h != NULL) {
Spencer Lowbc36bdb2015-09-16 16:40:06 -0700477 return SetHandleInformation(h, HANDLE_FLAG_INHERIT, 0) ? true : false;
478 }
479
480 return true;
481}
482
483// Try to make a handle non-inheritable with the expectation that this should
484// succeed, so if this fails, output error info.
485static bool _make_handle_noninheritable(HANDLE h) {
486 if (!_try_make_handle_noninheritable(h)) {
487 // Show the handle value to give us a clue in case we have problems
488 // with pseudo-handle values.
489 fprintf(stderr, "Cannot make handle 0x%p non-inheritable: %s\n",
David Pursell5f787ed2016-01-27 08:52:53 -0800490 h, android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Lowbc36bdb2015-09-16 16:40:06 -0700491 return false;
Spencer Low2122c7a2015-08-26 18:46:09 -0700492 }
493
494 return true;
495}
496
497// Create anonymous pipe, preventing inheritance of the read pipe and setting
498// security of the write pipe to sa.
499static bool _create_anonymous_pipe(unique_handle* pipe_read_out,
500 unique_handle* pipe_write_out,
501 SECURITY_ATTRIBUTES* sa) {
502 HANDLE pipe_read_raw = NULL;
503 HANDLE pipe_write_raw = NULL;
504 if (!CreatePipe(&pipe_read_raw, &pipe_write_raw, sa, 0)) {
505 fprintf(stderr, "Cannot create pipe: %s\n",
David Pursell5f787ed2016-01-27 08:52:53 -0800506 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low2122c7a2015-08-26 18:46:09 -0700507 return false;
508 }
509
510 unique_handle pipe_read(pipe_read_raw);
511 pipe_read_raw = NULL;
512 unique_handle pipe_write(pipe_write_raw);
513 pipe_write_raw = NULL;
514
515 if (!_make_handle_noninheritable(pipe_read.get())) {
516 return false;
517 }
518
519 *pipe_read_out = std::move(pipe_read);
520 *pipe_write_out = std::move(pipe_write);
521
522 return true;
523}
524
Spencer Low53a0a992015-11-21 17:23:51 -0800525// Read from a pipe (that we take ownership of) and write the result to stdout/stderr. Return on
526// error or when the pipe is closed. Internally makes inheritable handles, so this should not be
527// called if subprocesses may be started concurrently.
Spencer Low2122c7a2015-08-26 18:46:09 -0700528static unsigned _redirect_pipe_thread(HANDLE h, DWORD nStdHandle) {
529 // Take ownership of the HANDLE and close when we're done.
530 unique_handle read_pipe(h);
Spencer Low53a0a992015-11-21 17:23:51 -0800531 const char* output_name = nStdHandle == STD_OUTPUT_HANDLE ? "stdout" : "stderr";
532 const int original_fd = fileno(nStdHandle == STD_OUTPUT_HANDLE ? stdout : stderr);
533 std::unique_ptr<FILE, decltype(&fclose)> stream(nullptr, fclose);
534
535 if (original_fd == -1) {
536 fprintf(stderr, "Failed to get file descriptor for %s: %s\n", output_name, strerror(errno));
537 return EXIT_FAILURE;
538 }
539
540 // If fileno() is -2, stdout/stderr is not associated with an output stream, so we should read,
541 // but don't write. Otherwise, make a FILE* identical to stdout/stderr except that it is in
542 // binary mode with no CR/LR translation since we're reading raw.
543 if (original_fd >= 0) {
544 // This internally makes a duplicate file handle that is inheritable, so callers should not
545 // call this function if subprocesses may be started concurrently.
546 const int fd = dup(original_fd);
547 if (fd == -1) {
548 fprintf(stderr, "Failed to duplicate file descriptor for %s: %s\n", output_name,
549 strerror(errno));
550 return EXIT_FAILURE;
551 }
552
553 // Note that although we call fdopen() below with a binary flag, it may not adhere to that
554 // flag, so we have to set the mode manually.
555 if (_setmode(fd, _O_BINARY) == -1) {
556 fprintf(stderr, "Failed to set binary mode for duplicate of %s: %s\n", output_name,
557 strerror(errno));
558 unix_close(fd);
559 return EXIT_FAILURE;
560 }
561
562 stream.reset(fdopen(fd, "wb"));
563 if (stream.get() == nullptr) {
564 fprintf(stderr, "Failed to open duplicate stream for %s: %s\n", output_name,
565 strerror(errno));
566 unix_close(fd);
567 return EXIT_FAILURE;
568 }
569
570 // Unbuffer the stream because it will be buffered by default and we want subprocess output
571 // to be shown immediately.
572 if (setvbuf(stream.get(), NULL, _IONBF, 0) == -1) {
573 fprintf(stderr, "Failed to unbuffer %s: %s\n", output_name, strerror(errno));
574 return EXIT_FAILURE;
575 }
576
577 // fd will be closed when stream is closed.
578 }
Spencer Low2122c7a2015-08-26 18:46:09 -0700579
580 while (true) {
581 char buf[64 * 1024];
582 DWORD bytes_read = 0;
583 if (!ReadFile(read_pipe.get(), buf, sizeof(buf), &bytes_read, NULL)) {
584 const DWORD err = GetLastError();
585 // ERROR_BROKEN_PIPE is expected when the subprocess closes
586 // the other end of the pipe.
587 if (err == ERROR_BROKEN_PIPE) {
588 return EXIT_SUCCESS;
589 } else {
590 fprintf(stderr, "Failed to read from %s: %s\n", output_name,
David Pursell5f787ed2016-01-27 08:52:53 -0800591 android::base::SystemErrorCodeToString(err).c_str());
Spencer Low2122c7a2015-08-26 18:46:09 -0700592 return EXIT_FAILURE;
593 }
594 }
595
Spencer Low53a0a992015-11-21 17:23:51 -0800596 // Don't try to write if our stdout/stderr was not setup by the parent process.
597 if (stream) {
598 // fwrite() actually calls adb_fwrite() which can write UTF-8 to the console.
599 const size_t bytes_written = fwrite(buf, 1, bytes_read, stream.get());
Spencer Low2122c7a2015-08-26 18:46:09 -0700600 if (bytes_written != bytes_read) {
Spencer Low53a0a992015-11-21 17:23:51 -0800601 fprintf(stderr, "Only wrote %zu of %lu bytes to %s\n", bytes_written, bytes_read,
602 output_name);
Spencer Low2122c7a2015-08-26 18:46:09 -0700603 return EXIT_FAILURE;
604 }
605 }
606 }
607}
608
609static unsigned __stdcall _redirect_stdout_thread(HANDLE h) {
Siva Velusamy49ee7cf2015-08-28 16:37:29 -0700610 adb_thread_setname("stdout redirect");
Spencer Low2122c7a2015-08-26 18:46:09 -0700611 return _redirect_pipe_thread(h, STD_OUTPUT_HANDLE);
612}
613
614static unsigned __stdcall _redirect_stderr_thread(HANDLE h) {
Siva Velusamy49ee7cf2015-08-28 16:37:29 -0700615 adb_thread_setname("stderr redirect");
Spencer Low2122c7a2015-08-26 18:46:09 -0700616 return _redirect_pipe_thread(h, STD_ERROR_HANDLE);
617}
618
619#endif
620
Stefan Hilzingera84a42e2010-04-19 12:21:12 +0100621int launch_server(int server_port)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800622{
Yabin Cuie77b6a02014-11-11 09:24:11 -0800623#if defined(_WIN32)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800624 /* we need to start the server in the background */
625 /* we create a PIPE that will be used to wait for the server's "OK" */
626 /* message since the pipe handles must be inheritable, we use a */
627 /* security attribute */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800628 SECURITY_ATTRIBUTES sa;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800629 sa.nLength = sizeof(sa);
630 sa.lpSecurityDescriptor = NULL;
631 sa.bInheritHandle = TRUE;
632
Spencer Low2122c7a2015-08-26 18:46:09 -0700633 // Redirect stdin to Windows /dev/null. If we instead pass an original
634 // stdin/stdout/stderr handle and it is a console handle, when the adb
635 // server starts up, the C Runtime will see a console handle for a process
636 // that isn't connected to a console and it will configure
637 // stdin/stdout/stderr to be closed. At that point, freopen() could be used
638 // to reopen stderr/out, but it would take more massaging to fixup the file
639 // descriptor number that freopen() uses. It's simplest to avoid all of this
640 // complexity by just redirecting stdin to `nul' and then the C Runtime acts
641 // as expected.
642 unique_handle nul_read(CreateFileW(L"nul", GENERIC_READ,
643 FILE_SHARE_READ | FILE_SHARE_WRITE, &sa, OPEN_EXISTING,
644 FILE_ATTRIBUTE_NORMAL, NULL));
645 if (nul_read.get() == INVALID_HANDLE_VALUE) {
646 fprintf(stderr, "Cannot open 'nul': %s\n",
David Pursell5f787ed2016-01-27 08:52:53 -0800647 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Lowd0f66c32015-05-20 23:17:26 -0700648 return -1;
649 }
650
Spencer Low53a0a992015-11-21 17:23:51 -0800651 // Create pipes with non-inheritable read handle, inheritable write handle. We need to connect
652 // the subprocess to pipes instead of just letting the subprocess inherit our existing
653 // stdout/stderr handles because a DETACHED_PROCESS cannot write to a console that it is not
654 // attached to.
Spencer Low2122c7a2015-08-26 18:46:09 -0700655 unique_handle ack_read, ack_write;
656 if (!_create_anonymous_pipe(&ack_read, &ack_write, &sa)) {
Spencer Lowd0f66c32015-05-20 23:17:26 -0700657 return -1;
658 }
Spencer Low2122c7a2015-08-26 18:46:09 -0700659 unique_handle stdout_read, stdout_write;
660 if (!_create_anonymous_pipe(&stdout_read, &stdout_write, &sa)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800661 return -1;
662 }
Spencer Low2122c7a2015-08-26 18:46:09 -0700663 unique_handle stderr_read, stderr_write;
664 if (!_create_anonymous_pipe(&stderr_read, &stderr_write, &sa)) {
665 return -1;
666 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800667
Ray Donnelly267aa8b2012-11-29 01:18:50 +0000668 /* Some programs want to launch an adb command and collect its output by
669 * calling CreateProcess with inheritable stdout/stderr handles, then
670 * using read() to get its output. When this happens, the stdout/stderr
671 * handles passed to the adb client process will also be inheritable.
672 * When starting the adb server here, care must be taken to reset them
673 * to non-inheritable.
674 * Otherwise, something bad happens: even if the adb command completes,
675 * the calling process is stuck while read()-ing from the stdout/stderr
676 * descriptors, because they're connected to corresponding handles in the
677 * adb server process (even if the latter never uses/writes to them).
Spencer Low2122c7a2015-08-26 18:46:09 -0700678 * Note that even if we don't pass these handles in the STARTUPINFO struct,
679 * if they're marked inheritable, they're still inherited, requiring us to
680 * deal with this.
681 *
682 * If we're still having problems with inheriting random handles in the
683 * future, consider using PROC_THREAD_ATTRIBUTE_HANDLE_LIST to explicitly
684 * specify which handles should be inherited: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/16/10248328.aspx
Spencer Lowbc36bdb2015-09-16 16:40:06 -0700685 *
686 * Older versions of Windows return console pseudo-handles that cannot be
687 * made non-inheritable, so ignore those failures.
Ray Donnelly267aa8b2012-11-29 01:18:50 +0000688 */
Spencer Lowbc36bdb2015-09-16 16:40:06 -0700689 _try_make_handle_noninheritable(GetStdHandle(STD_INPUT_HANDLE));
690 _try_make_handle_noninheritable(GetStdHandle(STD_OUTPUT_HANDLE));
691 _try_make_handle_noninheritable(GetStdHandle(STD_ERROR_HANDLE));
Ray Donnelly267aa8b2012-11-29 01:18:50 +0000692
Spencer Low2122c7a2015-08-26 18:46:09 -0700693 STARTUPINFOW startup;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800694 ZeroMemory( &startup, sizeof(startup) );
695 startup.cb = sizeof(startup);
Spencer Low2122c7a2015-08-26 18:46:09 -0700696 startup.hStdInput = nul_read.get();
697 startup.hStdOutput = stdout_write.get();
698 startup.hStdError = stderr_write.get();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800699 startup.dwFlags = STARTF_USESTDHANDLES;
700
Spencer Low2122c7a2015-08-26 18:46:09 -0700701 // Verify that the pipe_write handle value can be passed on the command line
702 // as %d and that the rest of adb code can pass it around in an int.
703 const int ack_write_as_int = cast_handle_to_int(ack_write.get());
704 if (cast_int_to_handle(ack_write_as_int) != ack_write.get()) {
705 // If this fires, either handle values are larger than 32-bits or else
706 // there is a bug in our casting.
707 // https://msdn.microsoft.com/en-us/library/windows/desktop/aa384203%28v=vs.85%29.aspx
708 fprintf(stderr, "Cannot fit pipe handle value into 32-bits: 0x%p\n",
709 ack_write.get());
710 return -1;
711 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800712
Spencer Low2122c7a2015-08-26 18:46:09 -0700713 // get path of current program
714 WCHAR program_path[MAX_PATH];
715 const DWORD module_result = GetModuleFileNameW(NULL, program_path,
716 arraysize(program_path));
717 if ((module_result >= arraysize(program_path)) || (module_result == 0)) {
Spencer Lowcf4ff642015-05-11 01:08:48 -0700718 // String truncation or some other error.
Spencer Low2122c7a2015-08-26 18:46:09 -0700719 fprintf(stderr, "Cannot get executable path: %s\n",
David Pursell5f787ed2016-01-27 08:52:53 -0800720 android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Lowcf4ff642015-05-11 01:08:48 -0700721 return -1;
722 }
Spencer Low5c398d22015-08-08 15:07:07 -0700723
Spencer Low2122c7a2015-08-26 18:46:09 -0700724 WCHAR args[64];
Spencer Lowcf4ff642015-05-11 01:08:48 -0700725 snwprintf(args, arraysize(args),
Spencer Low5c398d22015-08-08 15:07:07 -0700726 L"adb -P %d fork-server server --reply-fd %d", server_port,
Spencer Low2122c7a2015-08-26 18:46:09 -0700727 ack_write_as_int);
728
729 PROCESS_INFORMATION pinfo;
730 ZeroMemory(&pinfo, sizeof(pinfo));
731
732 if (!CreateProcessW(
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800733 program_path, /* program path */
Wenhao Lia09558c2013-11-13 16:23:37 +0800734 args,
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800735 /* the fork-server argument will set the
736 debug = 2 in the child */
737 NULL, /* process handle is not inheritable */
738 NULL, /* thread handle is not inheritable */
739 TRUE, /* yes, inherit some handles */
740 DETACHED_PROCESS, /* the new process doesn't have a console */
741 NULL, /* use parent's environment block */
742 NULL, /* use parent's starting directory */
743 &startup, /* startup info, i.e. std handles */
Spencer Low2122c7a2015-08-26 18:46:09 -0700744 &pinfo )) {
745 fprintf(stderr, "Cannot create process: %s\n",
David Pursell5f787ed2016-01-27 08:52:53 -0800746 android::base::SystemErrorCodeToString(GetLastError()).c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800747 return -1;
748 }
749
Spencer Low2122c7a2015-08-26 18:46:09 -0700750 unique_handle process_handle(pinfo.hProcess);
751 pinfo.hProcess = NULL;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800752
Spencer Low2122c7a2015-08-26 18:46:09 -0700753 // Close handles that we no longer need to complete the rest.
754 CloseHandle(pinfo.hThread);
755 pinfo.hThread = NULL;
756
757 nul_read.reset();
758 ack_write.reset();
759 stdout_write.reset();
760 stderr_write.reset();
761
Spencer Low53a0a992015-11-21 17:23:51 -0800762 // Start threads to read from subprocess stdout/stderr and write to ours to make subprocess
763 // errors easier to diagnose. Note that the threads internally create inheritable handles, but
764 // that is ok because we've already spawned the subprocess.
Spencer Low2122c7a2015-08-26 18:46:09 -0700765
766 // In the past, reading from a pipe before the child process's C Runtime
767 // started up and called GetFileType() caused a hang: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/02/10243553.aspx#10244216
768 // This is reportedly fixed in Windows Vista: https://support.microsoft.com/en-us/kb/2009703
769 // I was unable to reproduce the problem on Windows XP. It sounds like a
770 // Windows Update may have fixed this: https://www.duckware.com/tech/peeknamedpipe.html
771 unique_handle stdout_thread(reinterpret_cast<HANDLE>(
772 _beginthreadex(NULL, 0, _redirect_stdout_thread, stdout_read.get(),
773 0, NULL)));
774 if (stdout_thread.get() == nullptr) {
775 fprintf(stderr, "Cannot create thread: %s\n", strerror(errno));
776 return -1;
777 }
778 stdout_read.release(); // Transfer ownership to new thread
779
780 unique_handle stderr_thread(reinterpret_cast<HANDLE>(
781 _beginthreadex(NULL, 0, _redirect_stderr_thread, stderr_read.get(),
782 0, NULL)));
783 if (stderr_thread.get() == nullptr) {
784 fprintf(stderr, "Cannot create thread: %s\n", strerror(errno));
785 return -1;
786 }
787 stderr_read.release(); // Transfer ownership to new thread
788
789 bool got_ack = false;
790
791 // Wait for the "OK\n" message, for the pipe to be closed, or other error.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800792 {
Spencer Low2122c7a2015-08-26 18:46:09 -0700793 char temp[3];
794 DWORD count = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800795
Spencer Low2122c7a2015-08-26 18:46:09 -0700796 if (ReadFile(ack_read.get(), temp, sizeof(temp), &count, NULL)) {
797 const CHAR expected[] = "OK\n";
798 const DWORD expected_length = arraysize(expected) - 1;
799 if (count == expected_length &&
800 memcmp(temp, expected, expected_length) == 0) {
801 got_ack = true;
802 } else {
803 fprintf(stderr, "ADB server didn't ACK\n");
804 }
805 } else {
806 const DWORD err = GetLastError();
807 // If the ACK was not written and the process exited, GetLastError()
808 // is probably ERROR_BROKEN_PIPE, in which case that info is not
809 // useful to the user.
810 fprintf(stderr, "could not read ok from ADB Server%s\n",
811 err == ERROR_BROKEN_PIPE ? "" :
812 android::base::StringPrintf(": %s",
David Pursell5f787ed2016-01-27 08:52:53 -0800813 android::base::SystemErrorCodeToString(err).c_str()).c_str());
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800814 }
Spencer Low2122c7a2015-08-26 18:46:09 -0700815 }
816
817 // Always try to wait a bit for threads reading stdout/stderr to finish.
818 // If the process started ok, it should close the pipes causing the threads
819 // to finish. If the process had an error, it should exit, also causing
820 // the pipes to be closed. In that case we want to read all of the output
821 // and write it out so that the user can diagnose failures.
822 const DWORD thread_timeout_ms = 15 * 1000;
823 const HANDLE threads[] = { stdout_thread.get(), stderr_thread.get() };
824 const DWORD wait_result = WaitForMultipleObjects(arraysize(threads),
825 threads, TRUE, thread_timeout_ms);
826 if (wait_result == WAIT_TIMEOUT) {
827 // Threads did not finish after waiting a little while. Perhaps the
828 // server didn't close pipes, or it is hung.
829 fprintf(stderr, "Timed-out waiting for threads to finish reading from "
830 "ADB Server\n");
831 // Process handles are signaled when the process exits, so if we wait
832 // on the handle for 0 seconds and it returns 'timeout', that means that
833 // the process is still running.
834 if (WaitForSingleObject(process_handle.get(), 0) == WAIT_TIMEOUT) {
835 // We could TerminateProcess(), but that seems somewhat presumptive.
836 fprintf(stderr, "ADB Server is running: process id %lu\n",
837 pinfo.dwProcessId);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800838 }
Spencer Low2122c7a2015-08-26 18:46:09 -0700839 return -1;
840 }
841
842 if (wait_result != WAIT_OBJECT_0) {
843 fprintf(stderr, "Unexpected result waiting for threads: %lu: %s\n",
David Pursell5f787ed2016-01-27 08:52:53 -0800844 wait_result, android::base::SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low2122c7a2015-08-26 18:46:09 -0700845 return -1;
846 }
847
848 // For now ignore the thread exit codes and assume they worked properly.
849
850 if (!got_ack) {
851 return -1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800852 }
Yabin Cuie77b6a02014-11-11 09:24:11 -0800853#else /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800854 char path[PATH_MAX];
855 int fd[2];
856
857 // set up a pipe so the child can tell us when it is ready.
Siva Velusamy9f2d1a92015-08-07 10:10:29 -0700858 // fd[0] will be parent's end, and the child will write on fd[1]
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800859 if (pipe(fd)) {
860 fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
861 return -1;
862 }
Alexey Tarasov31664102009-10-22 02:55:00 +1100863 get_my_path(path, PATH_MAX);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800864 pid_t pid = fork();
865 if(pid < 0) return -1;
866
867 if (pid == 0) {
868 // child side of the fork
869
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800870 adb_close(fd[0]);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800871
Matt Gumbeld7b33082012-11-14 10:16:17 -0800872 char str_port[30];
Siva Velusamy9f2d1a92015-08-07 10:10:29 -0700873 snprintf(str_port, sizeof(str_port), "%d", server_port);
874 char reply_fd[30];
875 snprintf(reply_fd, sizeof(reply_fd), "%d", fd[1]);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800876 // child process
Siva Velusamy9f2d1a92015-08-07 10:10:29 -0700877 int result = execl(path, "adb", "-P", str_port, "fork-server", "server", "--reply-fd", reply_fd, NULL);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800878 // this should not return
879 fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno);
880 } else {
881 // parent side of the fork
882
883 char temp[3];
884
885 temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C';
886 // wait for the "OK\n" message
887 adb_close(fd[1]);
888 int ret = adb_read(fd[0], temp, 3);
JP Abgrall408fa572011-03-16 15:57:42 -0700889 int saved_errno = errno;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800890 adb_close(fd[0]);
891 if (ret < 0) {
JP Abgrall408fa572011-03-16 15:57:42 -0700892 fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", saved_errno);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800893 return -1;
894 }
895 if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
896 fprintf(stderr, "ADB server didn't ACK\n" );
897 return -1;
898 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800899 }
Yabin Cuie77b6a02014-11-11 09:24:11 -0800900#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800901 return 0;
902}
Yabin Cuie77b6a02014-11-11 09:24:11 -0800903#endif /* ADB_HOST */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800904
David 'Digit' Turner25258692013-03-21 21:07:42 +0100905// Try to handle a network forwarding request.
906// This returns 1 on success, 0 on failure, and -1 to indicate this is not
907// a forwarding-related request.
Elliott Hughesaee80fb2015-05-07 21:38:41 -0700908int handle_forward_request(const char* service, TransportType type, const char* serial, int reply_fd)
David 'Digit' Turner25258692013-03-21 21:07:42 +0100909{
910 if (!strcmp(service, "list-forward")) {
911 // Create the list of forward redirections.
Elliott Hughese67f1f82015-04-30 17:32:03 -0700912 std::string listeners = format_listeners();
David 'Digit' Turner25258692013-03-21 21:07:42 +0100913#if ADB_HOST
Elliott Hughese67f1f82015-04-30 17:32:03 -0700914 SendOkay(reply_fd);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100915#endif
Josh Gao7e6683c2016-01-15 14:35:54 -0800916 return SendProtocolString(reply_fd, listeners);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100917 }
918
919 if (!strcmp(service, "killforward-all")) {
920 remove_all_listeners();
921#if ADB_HOST
922 /* On the host: 1st OKAY is connect, 2nd OKAY is status */
Elliott Hughese67f1f82015-04-30 17:32:03 -0700923 SendOkay(reply_fd);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100924#endif
Elliott Hughese67f1f82015-04-30 17:32:03 -0700925 SendOkay(reply_fd);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100926 return 1;
927 }
928
Elliott Hughes424af022015-05-29 17:55:19 -0700929 if (!strncmp(service, "forward:", 8) || !strncmp(service, "killforward:", 12)) {
930 // killforward:local
931 // forward:(norebind:)?local;remote
932 bool kill_forward = false;
933 bool no_rebind = false;
934 if (android::base::StartsWith(service, "killforward:")) {
935 kill_forward = true;
936 service += 12;
Spencer Lowa6b340a2015-07-29 17:21:21 -0700937 } else {
938 service += 8; // skip past "forward:"
Elliott Hughes424af022015-05-29 17:55:19 -0700939 if (android::base::StartsWith(service, "norebind:")) {
940 no_rebind = true;
941 service += 9;
942 }
David 'Digit' Turner25258692013-03-21 21:07:42 +0100943 }
944
Elliott Hughes424af022015-05-29 17:55:19 -0700945 std::vector<std::string> pieces = android::base::Split(service, ";");
David 'Digit' Turner25258692013-03-21 21:07:42 +0100946
Elliott Hughes424af022015-05-29 17:55:19 -0700947 if (kill_forward) {
948 // Check killforward: parameter format: '<local>'
949 if (pieces.size() != 1 || pieces[0].empty()) {
950 SendFail(reply_fd, android::base::StringPrintf("bad killforward: %s", service));
David 'Digit' Turner25258692013-03-21 21:07:42 +0100951 return 1;
952 }
953 } else {
Elliott Hughes424af022015-05-29 17:55:19 -0700954 // Check forward: parameter format: '<local>;<remote>'
955 if (pieces.size() != 2 || pieces[0].empty() || pieces[1].empty() || pieces[1][0] == '*') {
956 SendFail(reply_fd, android::base::StringPrintf("bad forward: %s", service));
David 'Digit' Turner25258692013-03-21 21:07:42 +0100957 return 1;
958 }
959 }
960
Elliott Hughes7be29c82015-04-16 22:54:44 -0700961 std::string error_msg;
Elliott Hughes8d28e192015-10-07 14:55:10 -0700962 atransport* transport = acquire_one_transport(type, serial, nullptr, &error_msg);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100963 if (!transport) {
Elliott Hughese67f1f82015-04-30 17:32:03 -0700964 SendFail(reply_fd, error_msg);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100965 return 1;
966 }
967
Spencer Low5200c662015-07-30 23:07:55 -0700968 std::string error;
Elliott Hughes3bd73c12015-05-05 13:10:43 -0700969 InstallStatus r;
David Purselleaae97e2016-04-07 11:25:48 -0700970 int resolved_tcp_port = 0;
Elliott Hughes424af022015-05-29 17:55:19 -0700971 if (kill_forward) {
972 r = remove_listener(pieces[0].c_str(), transport);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100973 } else {
David Purselleaae97e2016-04-07 11:25:48 -0700974 r = install_listener(pieces[0], pieces[1].c_str(), transport, no_rebind,
975 &resolved_tcp_port, &error);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100976 }
Elliott Hughes7b506092015-04-20 08:09:20 -0700977 if (r == INSTALL_STATUS_OK) {
David 'Digit' Turner25258692013-03-21 21:07:42 +0100978#if ADB_HOST
David Purselleaae97e2016-04-07 11:25:48 -0700979 // On the host: 1st OKAY is connect, 2nd OKAY is status.
Elliott Hughese67f1f82015-04-30 17:32:03 -0700980 SendOkay(reply_fd);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100981#endif
Elliott Hughese67f1f82015-04-30 17:32:03 -0700982 SendOkay(reply_fd);
David Purselleaae97e2016-04-07 11:25:48 -0700983
984 // If a TCP port was resolved, send the actual port number back.
985 if (resolved_tcp_port != 0) {
986 SendProtocolString(reply_fd, android::base::StringPrintf("%d", resolved_tcp_port));
987 }
988
David 'Digit' Turner25258692013-03-21 21:07:42 +0100989 return 1;
990 }
991
Elliott Hughes7b506092015-04-20 08:09:20 -0700992 std::string message;
993 switch (r) {
Elliott Hughes424af022015-05-29 17:55:19 -0700994 case INSTALL_STATUS_OK: message = "success (!)"; break;
Elliott Hughes7b506092015-04-20 08:09:20 -0700995 case INSTALL_STATUS_INTERNAL_ERROR: message = "internal error"; break;
996 case INSTALL_STATUS_CANNOT_BIND:
Spencer Lowbf7c6052015-08-11 16:45:32 -0700997 message = android::base::StringPrintf("cannot bind listener: %s",
Spencer Low5200c662015-07-30 23:07:55 -0700998 error.c_str());
Elliott Hughes7b506092015-04-20 08:09:20 -0700999 break;
1000 case INSTALL_STATUS_CANNOT_REBIND:
Spencer Lowa6b340a2015-07-29 17:21:21 -07001001 message = android::base::StringPrintf("cannot rebind existing socket");
Elliott Hughes7b506092015-04-20 08:09:20 -07001002 break;
Elliott Hughes424af022015-05-29 17:55:19 -07001003 case INSTALL_STATUS_LISTENER_NOT_FOUND:
1004 message = android::base::StringPrintf("listener '%s' not found", service);
1005 break;
David 'Digit' Turner25258692013-03-21 21:07:42 +01001006 }
Elliott Hughese67f1f82015-04-30 17:32:03 -07001007 SendFail(reply_fd, message);
David 'Digit' Turner25258692013-03-21 21:07:42 +01001008 return 1;
1009 }
1010 return 0;
1011}
1012
Elliott Hughese2d36772015-06-23 13:00:32 -07001013#if ADB_HOST
1014static int SendOkay(int fd, const std::string& s) {
1015 SendOkay(fd);
1016 SendProtocolString(fd, s);
1017 return 0;
1018}
1019#endif
1020
Dan Albertc89e0cc2015-05-08 16:13:53 -07001021int handle_host_request(const char* service, TransportType type,
1022 const char* serial, int reply_fd, asocket* s) {
1023 if (strcmp(service, "kill") == 0) {
1024 fprintf(stderr, "adb server killed by remote request\n");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001025 fflush(stdout);
Elliott Hughese67f1f82015-04-30 17:32:03 -07001026 SendOkay(reply_fd);
Spencer Lowf18fc082015-08-11 17:05:02 -07001027
Spencer Low351ecd12015-10-14 17:32:44 -07001028 // On Windows, if the process exits with open sockets that
1029 // shutdown(SD_SEND) has not been called on, TCP RST segments will be
1030 // sent to the peers which will cause their next recv() to error-out
1031 // with WSAECONNRESET. In the case of this code, that means the client
1032 // may not read the OKAY sent above.
Spencer Lowf18fc082015-08-11 17:05:02 -07001033 adb_shutdown(reply_fd);
1034
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001035 exit(0);
1036 }
1037
1038#if ADB_HOST
1039 // "transport:" is used for switching transport with a specified serial number
1040 // "transport-usb:" is used for switching transport to the only USB transport
1041 // "transport-local:" is used for switching transport to the only local transport
1042 // "transport-any:" is used for switching transport to the only transport
1043 if (!strncmp(service, "transport", strlen("transport"))) {
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001044 TransportType type = kTransportAny;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001045
1046 if (!strncmp(service, "transport-usb", strlen("transport-usb"))) {
1047 type = kTransportUsb;
1048 } else if (!strncmp(service, "transport-local", strlen("transport-local"))) {
1049 type = kTransportLocal;
1050 } else if (!strncmp(service, "transport-any", strlen("transport-any"))) {
1051 type = kTransportAny;
1052 } else if (!strncmp(service, "transport:", strlen("transport:"))) {
1053 service += strlen("transport:");
Tom Marlin3175c8e2011-07-27 12:56:14 -05001054 serial = service;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001055 }
1056
Elliott Hughes8d28e192015-10-07 14:55:10 -07001057 std::string error;
1058 atransport* t = acquire_one_transport(type, serial, nullptr, &error);
Elliott Hughese2d36772015-06-23 13:00:32 -07001059 if (t != nullptr) {
1060 s->transport = t;
Elliott Hughese67f1f82015-04-30 17:32:03 -07001061 SendOkay(reply_fd);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001062 } else {
Elliott Hughes8d28e192015-10-07 14:55:10 -07001063 SendFail(reply_fd, error);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001064 }
1065 return 1;
1066 }
1067
1068 // return a list of all connected devices
Scott Andersone109d262012-04-20 11:21:14 -07001069 if (!strncmp(service, "devices", 7)) {
Elliott Hughese67f1f82015-04-30 17:32:03 -07001070 bool long_listing = (strcmp(service+7, "-l") == 0);
1071 if (long_listing || service[7] == 0) {
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001072 D("Getting device list...");
Elliott Hughese67f1f82015-04-30 17:32:03 -07001073 std::string device_list = list_transports(long_listing);
Yabin Cui7a3f8d62015-09-02 17:44:28 -07001074 D("Sending device list...");
Elliott Hughese2d36772015-06-23 13:00:32 -07001075 return SendOkay(reply_fd, device_list);
Scott Andersone109d262012-04-20 11:21:14 -07001076 }
Elliott Hughese67f1f82015-04-30 17:32:03 -07001077 return 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001078 }
1079
Dan Albert90d4b732015-05-20 18:58:41 -07001080 if (!strcmp(service, "features")) {
Elliott Hughes8d28e192015-10-07 14:55:10 -07001081 std::string error;
1082 atransport* t = acquire_one_transport(type, serial, nullptr, &error);
David Pursell880be432015-09-04 16:40:30 -07001083 if (t != nullptr) {
David Pursell4e2fd362015-09-22 10:43:08 -07001084 SendOkay(reply_fd, FeatureSetToString(t->features()));
David Pursell880be432015-09-04 16:40:30 -07001085 } else {
Elliott Hughes8d28e192015-10-07 14:55:10 -07001086 SendFail(reply_fd, error);
David Pursell880be432015-09-04 16:40:30 -07001087 }
1088 return 0;
1089 }
1090
Mike Lockwood74d7ff82009-10-11 23:04:18 -04001091 // remove TCP transport
1092 if (!strncmp(service, "disconnect:", 11)) {
Elliott Hughes3d5f60d2015-07-18 12:21:30 -07001093 const std::string address(service + 11);
1094 if (address.empty()) {
Yabin Cuif4b99282015-08-27 12:03:11 -07001095 kick_all_tcp_devices();
Elliott Hughes3d5f60d2015-07-18 12:21:30 -07001096 return SendOkay(reply_fd, "disconnected everything");
Mike Lockwood74d7ff82009-10-11 23:04:18 -04001097 }
1098
Elliott Hughes3d5f60d2015-07-18 12:21:30 -07001099 std::string serial;
1100 std::string host;
1101 int port = DEFAULT_ADB_LOCAL_TRANSPORT_PORT;
1102 std::string error;
David Pursell706955f2016-01-21 08:40:59 -08001103 if (!android::base::ParseNetAddress(address, &host, &port, &serial, &error)) {
Elliott Hughes3d5f60d2015-07-18 12:21:30 -07001104 return SendFail(reply_fd, android::base::StringPrintf("couldn't parse '%s': %s",
1105 address.c_str(), error.c_str()));
1106 }
1107 atransport* t = find_transport(serial.c_str());
1108 if (t == nullptr) {
1109 return SendFail(reply_fd, android::base::StringPrintf("no such device '%s'",
1110 serial.c_str()));
1111 }
Yabin Cuif4b99282015-08-27 12:03:11 -07001112 kick_transport(t);
Elliott Hughes3d5f60d2015-07-18 12:21:30 -07001113 return SendOkay(reply_fd, android::base::StringPrintf("disconnected %s", address.c_str()));
Mike Lockwood2f38b692009-08-24 15:58:40 -07001114 }
1115
Elliott Hughes8d28e192015-10-07 14:55:10 -07001116 // Returns our value for ADB_SERVER_VERSION.
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001117 if (!strcmp(service, "version")) {
Elliott Hughese2d36772015-06-23 13:00:32 -07001118 return SendOkay(reply_fd, android::base::StringPrintf("%04x", ADB_SERVER_VERSION));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001119 }
1120
Elliott Hughese2d36772015-06-23 13:00:32 -07001121 // These always report "unknown" rather than the actual error, for scripts.
1122 if (!strcmp(service, "get-serialno")) {
Elliott Hughes8d28e192015-10-07 14:55:10 -07001123 std::string error;
1124 atransport* t = acquire_one_transport(type, serial, nullptr, &error);
1125 if (t) {
1126 return SendOkay(reply_fd, t->serial ? t->serial : "unknown");
1127 } else {
1128 return SendFail(reply_fd, error);
1129 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001130 }
Elliott Hughese2d36772015-06-23 13:00:32 -07001131 if (!strcmp(service, "get-devpath")) {
Elliott Hughes8d28e192015-10-07 14:55:10 -07001132 std::string error;
1133 atransport* t = acquire_one_transport(type, serial, nullptr, &error);
1134 if (t) {
1135 return SendOkay(reply_fd, t->devpath ? t->devpath : "unknown");
1136 } else {
1137 return SendFail(reply_fd, error);
1138 }
Scott Andersone109d262012-04-20 11:21:14 -07001139 }
Elliott Hughese2d36772015-06-23 13:00:32 -07001140 if (!strcmp(service, "get-state")) {
Elliott Hughes8d28e192015-10-07 14:55:10 -07001141 std::string error;
1142 atransport* t = acquire_one_transport(type, serial, nullptr, &error);
1143 if (t) {
1144 return SendOkay(reply_fd, t->connection_state_name());
1145 } else {
1146 return SendFail(reply_fd, error);
1147 }
Elliott Hughese2d36772015-06-23 13:00:32 -07001148 }
1149
Elliott Hughes8d28e192015-10-07 14:55:10 -07001150 // Indicates a new emulator instance has started.
Elliott Hughese2d36772015-06-23 13:00:32 -07001151 if (!strncmp(service, "emulator:", 9)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001152 int port = atoi(service+9);
1153 local_connect(port);
1154 /* we don't even need to send a reply */
1155 return 0;
1156 }
Yabin Cui1f4ec192016-04-05 13:50:44 -07001157
1158 if (!strcmp(service, "reconnect")) {
1159 if (s->transport != nullptr) {
1160 kick_transport(s->transport);
1161 }
1162 return SendOkay(reply_fd, "done");
1163 }
Simon Yedc22c3c2014-07-14 17:23:06 -07001164#endif // ADB_HOST
1165
Elliott Hughes3bd73c12015-05-05 13:10:43 -07001166 int ret = handle_forward_request(service, type, serial, reply_fd);
Simon Yedc22c3c2014-07-14 17:23:06 -07001167 if (ret >= 0)
1168 return ret - 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001169 return -1;
1170}