blob: eb01da82bb2332279f04b7f5c7a63972151138ff [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
Dan Albertdb6fe642015-03-19 15:21:08 -070017#define TRACE_TAG TRACE_ADB
18
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>
32
33#include <string>
Dan Albert27929b02015-03-19 13:25:27 -070034#include <vector>
35#include <unordered_map>
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080036
Dan Albert08d552b2015-05-21 13:58:50 -070037#include <base/logging.h>
Spencer Low6815c072015-05-11 01:08:48 -070038#include <base/macros.h>
Elliott Hughesa585cbd2015-04-20 08:09:20 -070039#include <base/stringprintf.h>
Elliott Hughesfb2ba512015-04-24 23:02:00 -070040#include <base/strings.h>
Elliott Hughesa585cbd2015-04-20 08:09:20 -070041
Benoit Goby2cc19e42012-04-12 12:23:49 -070042#include "adb_auth.h"
Dan Albert66a91b02015-02-24 21:26:58 -080043#include "adb_io.h"
Dan Albert020292b2015-02-18 18:03:26 -080044#include "adb_listeners.h"
Elliott Hughes09ccf1f2015-07-18 12:21:30 -070045#include "adb_utils.h"
Dan Albertb302d122015-02-24 15:51:19 -080046#include "transport.h"
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080047
Scott Andersonfa020922012-05-25 14:10:02 -070048#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
49
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080050#if !ADB_HOST
Nick Kralevich7472d292013-05-23 09:54:13 -070051#include <cutils/properties.h>
Nick Kralevich9de01bd2013-02-28 14:12:58 -080052#include <sys/capability.h>
Jeff Sharkey11987252012-08-14 21:00:22 -070053#include <sys/mount.h>
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080054#endif
55
Dan Albert08d552b2015-05-21 13:58:50 -070056ADB_MUTEX_DEFINE(D_lock);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080057
Scott Andersonfa020922012-05-25 14:10:02 -070058#if !ADB_HOST
Dan Albert08d552b2015-05-21 13:58:50 -070059const char* adb_device_banner = "device";
60static android::base::LogdLogger gLogdLogger;
Dan Albertbe8e54b2015-05-18 13:06:53 -070061#else
62const char* adb_device_banner = "host";
Scott Andersonfa020922012-05-25 14:10:02 -070063#endif
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080064
Dan Albert08d552b2015-05-21 13:58:50 -070065void AdbLogger(android::base::LogId id, android::base::LogSeverity severity,
66 const char* tag, const char* file, unsigned int line,
67 const char* message) {
68 android::base::StderrLogger(id, severity, tag, file, line, message);
69#if !ADB_HOST
70 gLogdLogger(id, severity, tag, file, line, message);
71#endif
72}
73
Elliott Hughesb9f01642015-08-12 08:32:10 -070074std::string adb_version() {
75 // Don't change the format of this --- it's parsed by ddmlib.
76 return android::base::StringPrintf("Android Debug Bridge version %d.%d.%d\n"
77 "Revision %s\n",
78 ADB_VERSION_MAJOR, ADB_VERSION_MINOR, ADB_SERVER_VERSION,
79 ADB_REVISION);
80}
81
Dan Albert08d552b2015-05-21 13:58:50 -070082void fatal(const char *fmt, ...) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080083 va_list ap;
84 va_start(ap, fmt);
85 fprintf(stderr, "error: ");
86 vfprintf(stderr, fmt, ap);
87 fprintf(stderr, "\n");
88 va_end(ap);
89 exit(-1);
90}
91
Dan Albert08d552b2015-05-21 13:58:50 -070092void fatal_errno(const char* fmt, ...) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080093 va_list ap;
94 va_start(ap, fmt);
95 fprintf(stderr, "error: %s: ", strerror(errno));
96 vfprintf(stderr, fmt, ap);
97 fprintf(stderr, "\n");
98 va_end(ap);
99 exit(-1);
100}
101
Dan Albertecbd8742015-03-08 21:12:08 -0700102#if !ADB_HOST
Dan Albert08d552b2015-05-21 13:58:50 -0700103static std::string get_log_file_name() {
Dan Albertecbd8742015-03-08 21:12:08 -0700104 struct tm now;
105 time_t t;
106 tzset();
107 time(&t);
108 localtime_r(&t, &now);
109
Dan Albertdec2b242015-03-19 22:53:30 -0700110 char timestamp[PATH_MAX];
111 strftime(timestamp, sizeof(timestamp), "%Y-%m-%d-%H-%M-%S", &now);
Dan Albertecbd8742015-03-08 21:12:08 -0700112
Dan Albert08d552b2015-05-21 13:58:50 -0700113 return android::base::StringPrintf("/data/adb/adb-%s-%d", timestamp,
114 getpid());
115}
116
117void start_device_log(void) {
118 int fd = unix_open(get_log_file_name().c_str(),
119 O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC, 0640);
Dan Albertecbd8742015-03-08 21:12:08 -0700120 if (fd == -1) {
121 return;
122 }
123
Dan Albert08d552b2015-05-21 13:58:50 -0700124 // Redirect stdout and stderr to the log file.
Dan Albertecbd8742015-03-08 21:12:08 -0700125 dup2(fd, STDOUT_FILENO);
126 dup2(fd, STDERR_FILENO);
127 fprintf(stderr, "--- adb starting (pid %d) ---\n", getpid());
Spencer Low3a2421b2015-05-22 20:09:06 -0700128 unix_close(fd);
Dan Albertecbd8742015-03-08 21:12:08 -0700129}
130#endif
131
132int adb_trace_mask;
133
134std::string get_trace_setting_from_env() {
135 const char* setting = getenv("ADB_TRACE");
136 if (setting == nullptr) {
137 setting = "";
138 }
139
140 return std::string(setting);
141}
142
143#if !ADB_HOST
144std::string get_trace_setting_from_prop() {
145 char buf[PROPERTY_VALUE_MAX];
146 property_get("persist.adb.trace_mask", buf, "");
147 return std::string(buf);
148}
149#endif
150
151std::string get_trace_setting() {
152#if ADB_HOST
153 return get_trace_setting_from_env();
154#else
155 return get_trace_setting_from_prop();
156#endif
157}
158
Dan Albert27929b02015-03-19 13:25:27 -0700159// Split the space separated list of tags from the trace setting and build the
160// trace mask from it. note that '1' and 'all' are special cases to enable all
161// tracing.
Dan Albertecbd8742015-03-08 21:12:08 -0700162//
163// adb's trace setting comes from the ADB_TRACE environment variable, whereas
164// adbd's comes from the system property persist.adb.trace_mask.
Dan Albert08d552b2015-05-21 13:58:50 -0700165static void setup_trace_mask() {
Dan Albertecbd8742015-03-08 21:12:08 -0700166 const std::string trace_setting = get_trace_setting();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800167
Dan Albert27929b02015-03-19 13:25:27 -0700168 std::unordered_map<std::string, int> trace_flags = {
169 {"1", 0},
170 {"all", 0},
171 {"adb", TRACE_ADB},
172 {"sockets", TRACE_SOCKETS},
173 {"packets", TRACE_PACKETS},
174 {"rwx", TRACE_RWX},
175 {"usb", TRACE_USB},
176 {"sync", TRACE_SYNC},
177 {"sysdeps", TRACE_SYSDEPS},
178 {"transport", TRACE_TRANSPORT},
179 {"jdwp", TRACE_JDWP},
180 {"services", TRACE_SERVICES},
181 {"auth", TRACE_AUTH}};
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800182
Dan Albert27929b02015-03-19 13:25:27 -0700183 std::vector<std::string> elements = android::base::Split(trace_setting, " ");
184 for (const auto& elem : elements) {
185 const auto& flag = trace_flags.find(elem);
186 if (flag == trace_flags.end()) {
Spencer Low5c761bd2015-07-21 02:06:26 -0700187 D("Unknown trace flag: %s\n", flag->first.c_str());
Dan Albert27929b02015-03-19 13:25:27 -0700188 continue;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800189 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800190
Dan Albert27929b02015-03-19 13:25:27 -0700191 if (flag->second == 0) {
192 // 0 is used for the special values "1" and "all" that enable all
193 // tracing.
194 adb_trace_mask = ~0;
195 return;
196 } else {
197 adb_trace_mask |= 1 << flag->second;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800198 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800199 }
Dan Albert08d552b2015-05-21 13:58:50 -0700200}
Dan Albertecbd8742015-03-08 21:12:08 -0700201
Dan Albert08d552b2015-05-21 13:58:50 -0700202void adb_trace_init(char** argv) {
Spencer Low566d3e12015-08-11 15:48:15 -0700203#if !ADB_HOST
Paul Lawrencef4ca27d2015-06-05 16:00:07 -0700204 // Don't open log file if no tracing, since this will block
205 // the crypto unmount of /data
Spencer Low566d3e12015-08-11 15:48:15 -0700206 if (!get_trace_setting().empty()) {
207 if (isatty(STDOUT_FILENO) == 0) {
208 start_device_log();
209 }
Dan Albert08d552b2015-05-21 13:58:50 -0700210 }
Dan Albertecbd8742015-03-08 21:12:08 -0700211#endif
Dan Albert08d552b2015-05-21 13:58:50 -0700212
213 setup_trace_mask();
214 android::base::InitLogging(argv, AdbLogger);
Elliott Hughesb9f01642015-08-12 08:32:10 -0700215
216 D("%s", adb_version().c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800217}
218
Dan Albertf30d73c2015-02-25 17:51:28 -0800219apacket* get_apacket(void)
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800220{
Dan Albertf30d73c2015-02-25 17:51:28 -0800221 apacket* p = reinterpret_cast<apacket*>(malloc(sizeof(apacket)));
222 if (p == nullptr) {
223 fatal("failed to allocate an apacket");
224 }
225
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800226 memset(p, 0, sizeof(apacket) - MAX_PAYLOAD);
227 return p;
228}
229
230void put_apacket(apacket *p)
231{
232 free(p);
233}
234
Benoit Goby2cc19e42012-04-12 12:23:49 -0700235void handle_online(atransport *t)
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800236{
237 D("adb: online\n");
Benoit Goby2cc19e42012-04-12 12:23:49 -0700238 t->online = 1;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800239}
240
241void handle_offline(atransport *t)
242{
243 D("adb: offline\n");
244 //Close the associated usb
Benoit Goby2cc19e42012-04-12 12:23:49 -0700245 t->online = 0;
Yabin Cui10dedd82015-08-26 11:18:42 -0700246
Yabin Cui2d4c1982015-08-28 15:09:44 -0700247 // This is necessary to avoid a race condition that occurred when a transport closes
Yabin Cui10dedd82015-08-26 11:18:42 -0700248 // while a client socket is still active.
249 close_all_sockets(t);
250
Yabin Cui2d4c1982015-08-28 15:09:44 -0700251 t->RunDisconnects();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800252}
253
Benoit Goby2cc19e42012-04-12 12:23:49 -0700254#if DEBUG_PACKETS
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800255#define DUMPMAX 32
256void print_packet(const char *label, apacket *p)
257{
258 char *tag;
259 char *x;
260 unsigned count;
261
262 switch(p->msg.command){
263 case A_SYNC: tag = "SYNC"; break;
264 case A_CNXN: tag = "CNXN" ; break;
265 case A_OPEN: tag = "OPEN"; break;
266 case A_OKAY: tag = "OKAY"; break;
267 case A_CLSE: tag = "CLSE"; break;
268 case A_WRTE: tag = "WRTE"; break;
Benoit Goby2cc19e42012-04-12 12:23:49 -0700269 case A_AUTH: tag = "AUTH"; break;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800270 default: tag = "????"; break;
271 }
272
273 fprintf(stderr, "%s: %s %08x %08x %04x \"",
274 label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
275 count = p->msg.data_length;
276 x = (char*) p->data;
277 if(count > DUMPMAX) {
278 count = DUMPMAX;
279 tag = "\n";
280 } else {
281 tag = "\"\n";
282 }
283 while(count-- > 0){
284 if((*x >= ' ') && (*x < 127)) {
285 fputc(*x, stderr);
286 } else {
287 fputc('.', stderr);
288 }
289 x++;
290 }
Benoit Goby2cc19e42012-04-12 12:23:49 -0700291 fputs(tag, stderr);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800292}
293#endif
294
295static void send_ready(unsigned local, unsigned remote, atransport *t)
296{
297 D("Calling send_ready \n");
298 apacket *p = get_apacket();
299 p->msg.command = A_OKAY;
300 p->msg.arg0 = local;
301 p->msg.arg1 = remote;
302 send_packet(p, t);
303}
304
305static void send_close(unsigned local, unsigned remote, atransport *t)
306{
307 D("Calling send_close \n");
308 apacket *p = get_apacket();
309 p->msg.command = A_CLSE;
310 p->msg.arg0 = local;
311 p->msg.arg1 = remote;
312 send_packet(p, t);
313}
314
Dan Albertbe8e54b2015-05-18 13:06:53 -0700315std::string get_connection_string() {
316 std::vector<std::string> connection_properties;
317
318#if !ADB_HOST
319 static const char* cnxn_props[] = {
Scott Andersonfa020922012-05-25 14:10:02 -0700320 "ro.product.name",
321 "ro.product.model",
322 "ro.product.device",
323 };
Scott Andersonfa020922012-05-25 14:10:02 -0700324
Dan Albertbe8e54b2015-05-18 13:06:53 -0700325 for (const auto& prop_name : cnxn_props) {
Scott Andersonfa020922012-05-25 14:10:02 -0700326 char value[PROPERTY_VALUE_MAX];
Dan Albertbe8e54b2015-05-18 13:06:53 -0700327 property_get(prop_name, value, "");
328 connection_properties.push_back(
329 android::base::StringPrintf("%s=%s", prop_name, value));
Scott Andersonfa020922012-05-25 14:10:02 -0700330 }
Scott Andersonfa020922012-05-25 14:10:02 -0700331#endif
Dan Albertbe8e54b2015-05-18 13:06:53 -0700332
333 connection_properties.push_back(android::base::StringPrintf(
334 "features=%s", android::base::Join(supported_features(), ',').c_str()));
335
336 return android::base::StringPrintf(
337 "%s::%s", adb_device_banner,
338 android::base::Join(connection_properties, ';').c_str());
Scott Andersonfa020922012-05-25 14:10:02 -0700339}
340
Dan Albertbe8e54b2015-05-18 13:06:53 -0700341void send_connect(atransport* t) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800342 D("Calling send_connect \n");
Dan Albertbe8e54b2015-05-18 13:06:53 -0700343 apacket* cp = get_apacket();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800344 cp->msg.command = A_CNXN;
Tamas Berghammera1c60c02015-07-13 19:12:28 +0100345 cp->msg.arg0 = t->get_protocol_version();
346 cp->msg.arg1 = t->get_max_payload();
Dan Albertbe8e54b2015-05-18 13:06:53 -0700347
348 std::string connection_str = get_connection_string();
349 // Connect and auth packets are limited to MAX_PAYLOAD_V1 because we don't
350 // yet know how much data the other size is willing to accept.
351 if (connection_str.length() > MAX_PAYLOAD_V1) {
352 LOG(FATAL) << "Connection banner is too long (length = "
353 << connection_str.length() << ")";
354 }
355
356 memcpy(cp->data, connection_str.c_str(), connection_str.length());
357 cp->msg.data_length = connection_str.length();
358
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800359 send_packet(cp, t);
Benoit Goby2cc19e42012-04-12 12:23:49 -0700360}
361
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700362// qual_overwrite is used to overwrite a qualifier string. dst is a
363// pointer to a char pointer. It is assumed that if *dst is non-NULL, it
364// was malloc'ed and needs to freed. *dst will be set to a dup of src.
365// TODO: switch to std::string for these atransport fields instead.
366static void qual_overwrite(char** dst, const std::string& src) {
Scott Andersonfa020922012-05-25 14:10:02 -0700367 free(*dst);
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700368 *dst = strdup(src.c_str());
Scott Andersonfa020922012-05-25 14:10:02 -0700369}
370
Dan Albertbe8e54b2015-05-18 13:06:53 -0700371void parse_banner(const std::string& banner, atransport* t) {
372 D("parse_banner: %s\n", banner.c_str());
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700373
374 // The format is something like:
375 // "device::ro.product.name=x;ro.product.model=y;ro.product.device=z;".
376 std::vector<std::string> pieces = android::base::Split(banner, ":");
377
378 if (pieces.size() > 2) {
379 const std::string& props = pieces[2];
380 for (auto& prop : android::base::Split(props, ";")) {
381 // The list of properties was traditionally ;-terminated rather than ;-separated.
382 if (prop.empty()) continue;
383
384 std::vector<std::string> key_value = android::base::Split(prop, "=");
385 if (key_value.size() != 2) continue;
386
387 const std::string& key = key_value[0];
388 const std::string& value = key_value[1];
389 if (key == "ro.product.name") {
390 qual_overwrite(&t->product, value);
391 } else if (key == "ro.product.model") {
392 qual_overwrite(&t->model, value);
393 } else if (key == "ro.product.device") {
394 qual_overwrite(&t->device, value);
Dan Albertbe8e54b2015-05-18 13:06:53 -0700395 } else if (key == "features") {
396 for (const auto& feature : android::base::Split(value, ",")) {
397 t->add_feature(feature);
398 }
Scott Andersonfa020922012-05-25 14:10:02 -0700399 }
400 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800401 }
402
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700403 const std::string& type = pieces[0];
404 if (type == "bootloader") {
Dan Albert9a50f4c2015-05-18 16:43:57 -0700405 D("setting connection_state to kCsBootloader\n");
406 t->connection_state = kCsBootloader;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800407 update_transports();
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700408 } else if (type == "device") {
Dan Albert9a50f4c2015-05-18 16:43:57 -0700409 D("setting connection_state to kCsDevice\n");
410 t->connection_state = kCsDevice;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800411 update_transports();
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700412 } else if (type == "recovery") {
Dan Albert9a50f4c2015-05-18 16:43:57 -0700413 D("setting connection_state to kCsRecovery\n");
414 t->connection_state = kCsRecovery;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800415 update_transports();
Elliott Hughesfb2ba512015-04-24 23:02:00 -0700416 } else if (type == "sideload") {
Dan Albert9a50f4c2015-05-18 16:43:57 -0700417 D("setting connection_state to kCsSideload\n");
418 t->connection_state = kCsSideload;
Doug Zongker6b217ed2012-01-09 14:54:53 -0800419 update_transports();
Elliott Hughesb27e2412015-04-29 22:37:25 -0700420 } else {
Dan Albert9a50f4c2015-05-18 16:43:57 -0700421 D("setting connection_state to kCsHost\n");
422 t->connection_state = kCsHost;
Doug Zongker6b217ed2012-01-09 14:54:53 -0800423 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800424}
425
Dan Albertbe8e54b2015-05-18 13:06:53 -0700426static void handle_new_connection(atransport* t, apacket* p) {
427 if (t->connection_state != kCsOffline) {
428 t->connection_state = kCsOffline;
429 handle_offline(t);
430 }
431
432 t->update_version(p->msg.arg0, p->msg.arg1);
433 std::string banner(reinterpret_cast<const char*>(p->data),
434 p->msg.data_length);
435 parse_banner(banner, t);
436
437#if ADB_HOST
438 handle_online(t);
439#else
440 if (!auth_required) {
441 handle_online(t);
442 send_connect(t);
443 } else {
444 send_auth_request(t);
445 }
446#endif
447}
448
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800449void handle_packet(apacket *p, atransport *t)
450{
451 asocket *s;
452
Viral Mehta841f9a72010-06-16 18:41:28 +0530453 D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
454 ((char*) (&(p->msg.command)))[1],
455 ((char*) (&(p->msg.command)))[2],
456 ((char*) (&(p->msg.command)))[3]);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800457 print_packet("recv", p);
458
459 switch(p->msg.command){
460 case A_SYNC:
461 if(p->msg.arg0){
462 send_packet(p, t);
Yabin Cui5fe6d0d2015-08-11 13:40:42 -0700463#if ADB_HOST
464 send_connect(t);
465#endif
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800466 } else {
Dan Albert9a50f4c2015-05-18 16:43:57 -0700467 t->connection_state = kCsOffline;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800468 handle_offline(t);
469 send_packet(p, t);
470 }
471 return;
472
Dan Albertbe8e54b2015-05-18 13:06:53 -0700473 case A_CNXN: // CONNECT(version, maxdata, "system-id-string")
474 handle_new_connection(t, p);
Benoit Goby2cc19e42012-04-12 12:23:49 -0700475 break;
476
477 case A_AUTH:
478 if (p->msg.arg0 == ADB_AUTH_TOKEN) {
Dan Albert9a50f4c2015-05-18 16:43:57 -0700479 t->connection_state = kCsUnauthorized;
Benoit Goby2cc19e42012-04-12 12:23:49 -0700480 t->key = adb_auth_nextkey(t->key);
481 if (t->key) {
482 send_auth_response(p->data, p->msg.data_length, t);
483 } else {
484 /* No more private keys to try, send the public key */
485 send_auth_publickey(t);
486 }
487 } else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
488 if (adb_auth_verify(t->token, p->data, p->msg.data_length)) {
489 adb_auth_verified(t);
490 t->failed_auth_attempts = 0;
491 } else {
492 if (t->failed_auth_attempts++ > 10)
493 adb_sleep_ms(1000);
494 send_auth_request(t);
495 }
496 } else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
497 adb_auth_confirm_key(p->data, p->msg.data_length, t);
498 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800499 break;
500
501 case A_OPEN: /* OPEN(local-id, 0, "destination") */
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100502 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800503 char *name = (char*) p->data;
504 name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
505 s = create_local_service_socket(name);
506 if(s == 0) {
507 send_close(0, p->msg.arg0, t);
508 } else {
509 s->peer = create_remote_socket(p->msg.arg0, t);
510 s->peer->peer = s;
511 send_ready(s->id, s->peer->id, t);
512 s->ready(s);
513 }
514 }
515 break;
516
517 case A_OKAY: /* READY(local-id, remote-id, "") */
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100518 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
519 if((s = find_local_socket(p->msg.arg1, 0))) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800520 if(s->peer == 0) {
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100521 /* On first READY message, create the connection. */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800522 s->peer = create_remote_socket(p->msg.arg0, t);
523 s->peer->peer = s;
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100524 s->ready(s);
525 } else if (s->peer->id == p->msg.arg0) {
526 /* Other READY messages must use the same local-id */
527 s->ready(s);
528 } else {
529 D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s\n",
530 p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800531 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800532 }
533 }
534 break;
535
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100536 case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
537 if (t->online && p->msg.arg1 != 0) {
538 if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
539 /* According to protocol.txt, p->msg.arg0 might be 0 to indicate
540 * a failed OPEN only. However, due to a bug in previous ADB
541 * versions, CLOSE(0, remote-id, "") was also used for normal
542 * CLOSE() operations.
543 *
544 * This is bad because it means a compromised adbd could
545 * send packets to close connections between the host and
546 * other devices. To avoid this, only allow this if the local
547 * socket has a peer on the same transport.
548 */
549 if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
550 D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s\n",
551 p->msg.arg1, t->serial, s->peer->transport->serial);
552 } else {
553 s->close(s);
554 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800555 }
556 }
557 break;
558
David 'Digit' Turnere92344d2013-12-13 14:09:44 +0100559 case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
560 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
561 if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800562 unsigned rid = p->msg.arg0;
563 p->len = p->msg.data_length;
564
565 if(s->enqueue(s, p) == 0) {
566 D("Enqueue the socket\n");
567 send_ready(s->id, rid, t);
568 }
569 return;
570 }
571 }
572 break;
573
574 default:
575 printf("handle_packet: what is %08x?!\n", p->msg.command);
576 }
577
578 put_apacket(p);
579}
580
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800581#if ADB_HOST
Nick Kralevich23b39f02013-02-15 10:22:08 -0800582
Spencer Low2bbb3a92015-08-26 18:46:09 -0700583#ifdef _WIN32
584
585static bool _make_handle_noninheritable(HANDLE h) {
586 if (h != INVALID_HANDLE_VALUE && h != NULL) {
587 if (!SetHandleInformation(h, HANDLE_FLAG_INHERIT, 0)) {
588 // Show the handle value to give us a clue in case we have problems
589 // with pseudo-handle values.
590 fprintf(stderr, "Cannot make handle 0x%p non-inheritable: %s\n",
591 h, SystemErrorCodeToString(GetLastError()).c_str());
592 return false;
593 }
594 }
595
596 return true;
597}
598
599// Create anonymous pipe, preventing inheritance of the read pipe and setting
600// security of the write pipe to sa.
601static bool _create_anonymous_pipe(unique_handle* pipe_read_out,
602 unique_handle* pipe_write_out,
603 SECURITY_ATTRIBUTES* sa) {
604 HANDLE pipe_read_raw = NULL;
605 HANDLE pipe_write_raw = NULL;
606 if (!CreatePipe(&pipe_read_raw, &pipe_write_raw, sa, 0)) {
607 fprintf(stderr, "Cannot create pipe: %s\n",
608 SystemErrorCodeToString(GetLastError()).c_str());
609 return false;
610 }
611
612 unique_handle pipe_read(pipe_read_raw);
613 pipe_read_raw = NULL;
614 unique_handle pipe_write(pipe_write_raw);
615 pipe_write_raw = NULL;
616
617 if (!_make_handle_noninheritable(pipe_read.get())) {
618 return false;
619 }
620
621 *pipe_read_out = std::move(pipe_read);
622 *pipe_write_out = std::move(pipe_write);
623
624 return true;
625}
626
627// Read from a pipe (that we take ownership of) and write what is returned to
628// GetStdHandle(nStdHandle). Return on error or when the pipe is closed.
629static unsigned _redirect_pipe_thread(HANDLE h, DWORD nStdHandle) {
630 // Take ownership of the HANDLE and close when we're done.
631 unique_handle read_pipe(h);
632 const HANDLE write_handle = GetStdHandle(nStdHandle);
633 const char* output_name = nStdHandle == STD_OUTPUT_HANDLE ?
634 "stdout" : "stderr";
635
636 while (true) {
637 char buf[64 * 1024];
638 DWORD bytes_read = 0;
639 if (!ReadFile(read_pipe.get(), buf, sizeof(buf), &bytes_read, NULL)) {
640 const DWORD err = GetLastError();
641 // ERROR_BROKEN_PIPE is expected when the subprocess closes
642 // the other end of the pipe.
643 if (err == ERROR_BROKEN_PIPE) {
644 return EXIT_SUCCESS;
645 } else {
646 fprintf(stderr, "Failed to read from %s: %s\n", output_name,
647 SystemErrorCodeToString(err).c_str());
648 return EXIT_FAILURE;
649 }
650 }
651
652 // Don't try to write if our stdout/stderr was not setup by the
653 // parent process.
654 if (write_handle != NULL && write_handle != INVALID_HANDLE_VALUE) {
655 DWORD bytes_written = 0;
656 if (!WriteFile(write_handle, buf, bytes_read, &bytes_written,
657 NULL)) {
658 fprintf(stderr, "Failed to write to %s: %s\n", output_name,
659 SystemErrorCodeToString(GetLastError()).c_str());
660 return EXIT_FAILURE;
661 }
662
663 if (bytes_written != bytes_read) {
664 fprintf(stderr, "Only wrote %lu of %lu bytes to %s\n",
665 bytes_written, bytes_read, output_name);
666 return EXIT_FAILURE;
667 }
668 }
669 }
670}
671
672static unsigned __stdcall _redirect_stdout_thread(HANDLE h) {
Siva Velusamy2669cf92015-08-28 16:37:29 -0700673 adb_thread_setname("stdout redirect");
Spencer Low2bbb3a92015-08-26 18:46:09 -0700674 return _redirect_pipe_thread(h, STD_OUTPUT_HANDLE);
675}
676
677static unsigned __stdcall _redirect_stderr_thread(HANDLE h) {
Siva Velusamy2669cf92015-08-28 16:37:29 -0700678 adb_thread_setname("stderr redirect");
Spencer Low2bbb3a92015-08-26 18:46:09 -0700679 return _redirect_pipe_thread(h, STD_ERROR_HANDLE);
680}
681
682#endif
683
Stefan Hilzinger92ca4fa2010-04-19 12:21:12 +0100684int launch_server(int server_port)
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800685{
Yabin Cui2fa43212014-11-11 09:24:11 -0800686#if defined(_WIN32)
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800687 /* we need to start the server in the background */
688 /* we create a PIPE that will be used to wait for the server's "OK" */
689 /* message since the pipe handles must be inheritable, we use a */
690 /* security attribute */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800691 SECURITY_ATTRIBUTES sa;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800692 sa.nLength = sizeof(sa);
693 sa.lpSecurityDescriptor = NULL;
694 sa.bInheritHandle = TRUE;
695
Spencer Low2bbb3a92015-08-26 18:46:09 -0700696 // Redirect stdin to Windows /dev/null. If we instead pass an original
697 // stdin/stdout/stderr handle and it is a console handle, when the adb
698 // server starts up, the C Runtime will see a console handle for a process
699 // that isn't connected to a console and it will configure
700 // stdin/stdout/stderr to be closed. At that point, freopen() could be used
701 // to reopen stderr/out, but it would take more massaging to fixup the file
702 // descriptor number that freopen() uses. It's simplest to avoid all of this
703 // complexity by just redirecting stdin to `nul' and then the C Runtime acts
704 // as expected.
705 unique_handle nul_read(CreateFileW(L"nul", GENERIC_READ,
706 FILE_SHARE_READ | FILE_SHARE_WRITE, &sa, OPEN_EXISTING,
707 FILE_ATTRIBUTE_NORMAL, NULL));
708 if (nul_read.get() == INVALID_HANDLE_VALUE) {
709 fprintf(stderr, "Cannot open 'nul': %s\n",
Spencer Low1711e012015-08-02 18:50:17 -0700710 SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low4911f4b2015-05-20 23:17:26 -0700711 return -1;
712 }
713
Spencer Low2bbb3a92015-08-26 18:46:09 -0700714 // create pipes with non-inheritable read handle, inheritable write handle
715 unique_handle ack_read, ack_write;
716 if (!_create_anonymous_pipe(&ack_read, &ack_write, &sa)) {
Spencer Low4911f4b2015-05-20 23:17:26 -0700717 return -1;
718 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700719 unique_handle stdout_read, stdout_write;
720 if (!_create_anonymous_pipe(&stdout_read, &stdout_write, &sa)) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800721 return -1;
722 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700723 unique_handle stderr_read, stderr_write;
724 if (!_create_anonymous_pipe(&stderr_read, &stderr_write, &sa)) {
725 return -1;
726 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800727
Ray Donnellyc0324702012-11-29 01:18:50 +0000728 /* Some programs want to launch an adb command and collect its output by
729 * calling CreateProcess with inheritable stdout/stderr handles, then
730 * using read() to get its output. When this happens, the stdout/stderr
731 * handles passed to the adb client process will also be inheritable.
732 * When starting the adb server here, care must be taken to reset them
733 * to non-inheritable.
734 * Otherwise, something bad happens: even if the adb command completes,
735 * the calling process is stuck while read()-ing from the stdout/stderr
736 * descriptors, because they're connected to corresponding handles in the
737 * adb server process (even if the latter never uses/writes to them).
Spencer Low2bbb3a92015-08-26 18:46:09 -0700738 * Note that even if we don't pass these handles in the STARTUPINFO struct,
739 * if they're marked inheritable, they're still inherited, requiring us to
740 * deal with this.
741 *
742 * If we're still having problems with inheriting random handles in the
743 * future, consider using PROC_THREAD_ATTRIBUTE_HANDLE_LIST to explicitly
744 * specify which handles should be inherited: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/16/10248328.aspx
Ray Donnellyc0324702012-11-29 01:18:50 +0000745 */
Spencer Low2bbb3a92015-08-26 18:46:09 -0700746 if (!_make_handle_noninheritable(GetStdHandle(STD_INPUT_HANDLE))) {
747 return -1;
Ray Donnellyc0324702012-11-29 01:18:50 +0000748 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700749 if (!_make_handle_noninheritable(GetStdHandle(STD_OUTPUT_HANDLE))) {
750 return -1;
751 }
752 if (!_make_handle_noninheritable(GetStdHandle(STD_ERROR_HANDLE))) {
753 return -1;
Ray Donnellyc0324702012-11-29 01:18:50 +0000754 }
755
Spencer Low2bbb3a92015-08-26 18:46:09 -0700756 STARTUPINFOW startup;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800757 ZeroMemory( &startup, sizeof(startup) );
758 startup.cb = sizeof(startup);
Spencer Low2bbb3a92015-08-26 18:46:09 -0700759 startup.hStdInput = nul_read.get();
760 startup.hStdOutput = stdout_write.get();
761 startup.hStdError = stderr_write.get();
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800762 startup.dwFlags = STARTF_USESTDHANDLES;
763
Spencer Low2bbb3a92015-08-26 18:46:09 -0700764 // Verify that the pipe_write handle value can be passed on the command line
765 // as %d and that the rest of adb code can pass it around in an int.
766 const int ack_write_as_int = cast_handle_to_int(ack_write.get());
767 if (cast_int_to_handle(ack_write_as_int) != ack_write.get()) {
768 // If this fires, either handle values are larger than 32-bits or else
769 // there is a bug in our casting.
770 // https://msdn.microsoft.com/en-us/library/windows/desktop/aa384203%28v=vs.85%29.aspx
771 fprintf(stderr, "Cannot fit pipe handle value into 32-bits: 0x%p\n",
772 ack_write.get());
773 return -1;
774 }
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800775
Spencer Low2bbb3a92015-08-26 18:46:09 -0700776 // get path of current program
777 WCHAR program_path[MAX_PATH];
778 const DWORD module_result = GetModuleFileNameW(NULL, program_path,
779 arraysize(program_path));
780 if ((module_result >= arraysize(program_path)) || (module_result == 0)) {
Spencer Low6815c072015-05-11 01:08:48 -0700781 // String truncation or some other error.
Spencer Low2bbb3a92015-08-26 18:46:09 -0700782 fprintf(stderr, "Cannot get executable path: %s\n",
Spencer Low1711e012015-08-02 18:50:17 -0700783 SystemErrorCodeToString(GetLastError()).c_str());
Spencer Low6815c072015-05-11 01:08:48 -0700784 return -1;
785 }
Spencer Lowb28812f2015-08-08 15:07:07 -0700786
Spencer Low2bbb3a92015-08-26 18:46:09 -0700787 WCHAR args[64];
Spencer Low6815c072015-05-11 01:08:48 -0700788 snwprintf(args, arraysize(args),
Spencer Lowb28812f2015-08-08 15:07:07 -0700789 L"adb -P %d fork-server server --reply-fd %d", server_port,
Spencer Low2bbb3a92015-08-26 18:46:09 -0700790 ack_write_as_int);
791
792 PROCESS_INFORMATION pinfo;
793 ZeroMemory(&pinfo, sizeof(pinfo));
794
795 if (!CreateProcessW(
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800796 program_path, /* program path */
Wenhao Lie0fa2c52013-11-13 16:23:37 +0800797 args,
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800798 /* the fork-server argument will set the
799 debug = 2 in the child */
800 NULL, /* process handle is not inheritable */
801 NULL, /* thread handle is not inheritable */
802 TRUE, /* yes, inherit some handles */
803 DETACHED_PROCESS, /* the new process doesn't have a console */
804 NULL, /* use parent's environment block */
805 NULL, /* use parent's starting directory */
806 &startup, /* startup info, i.e. std handles */
Spencer Low2bbb3a92015-08-26 18:46:09 -0700807 &pinfo )) {
808 fprintf(stderr, "Cannot create process: %s\n",
Spencer Low1711e012015-08-02 18:50:17 -0700809 SystemErrorCodeToString(GetLastError()).c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800810 return -1;
811 }
812
Spencer Low2bbb3a92015-08-26 18:46:09 -0700813 unique_handle process_handle(pinfo.hProcess);
814 pinfo.hProcess = NULL;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800815
Spencer Low2bbb3a92015-08-26 18:46:09 -0700816 // Close handles that we no longer need to complete the rest.
817 CloseHandle(pinfo.hThread);
818 pinfo.hThread = NULL;
819
820 nul_read.reset();
821 ack_write.reset();
822 stdout_write.reset();
823 stderr_write.reset();
824
825 // Start threads to read from subprocess stdout/stderr and write to ours
826 // to make subprocess errors easier to diagnose.
827
828 // In the past, reading from a pipe before the child process's C Runtime
829 // started up and called GetFileType() caused a hang: http://blogs.msdn.com/b/oldnewthing/archive/2011/12/02/10243553.aspx#10244216
830 // This is reportedly fixed in Windows Vista: https://support.microsoft.com/en-us/kb/2009703
831 // I was unable to reproduce the problem on Windows XP. It sounds like a
832 // Windows Update may have fixed this: https://www.duckware.com/tech/peeknamedpipe.html
833 unique_handle stdout_thread(reinterpret_cast<HANDLE>(
834 _beginthreadex(NULL, 0, _redirect_stdout_thread, stdout_read.get(),
835 0, NULL)));
836 if (stdout_thread.get() == nullptr) {
837 fprintf(stderr, "Cannot create thread: %s\n", strerror(errno));
838 return -1;
839 }
840 stdout_read.release(); // Transfer ownership to new thread
841
842 unique_handle stderr_thread(reinterpret_cast<HANDLE>(
843 _beginthreadex(NULL, 0, _redirect_stderr_thread, stderr_read.get(),
844 0, NULL)));
845 if (stderr_thread.get() == nullptr) {
846 fprintf(stderr, "Cannot create thread: %s\n", strerror(errno));
847 return -1;
848 }
849 stderr_read.release(); // Transfer ownership to new thread
850
851 bool got_ack = false;
852
853 // 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 -0800854 {
Spencer Low2bbb3a92015-08-26 18:46:09 -0700855 char temp[3];
856 DWORD count = 0;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800857
Spencer Low2bbb3a92015-08-26 18:46:09 -0700858 if (ReadFile(ack_read.get(), temp, sizeof(temp), &count, NULL)) {
859 const CHAR expected[] = "OK\n";
860 const DWORD expected_length = arraysize(expected) - 1;
861 if (count == expected_length &&
862 memcmp(temp, expected, expected_length) == 0) {
863 got_ack = true;
864 } else {
865 fprintf(stderr, "ADB server didn't ACK\n");
866 }
867 } else {
868 const DWORD err = GetLastError();
869 // If the ACK was not written and the process exited, GetLastError()
870 // is probably ERROR_BROKEN_PIPE, in which case that info is not
871 // useful to the user.
872 fprintf(stderr, "could not read ok from ADB Server%s\n",
873 err == ERROR_BROKEN_PIPE ? "" :
874 android::base::StringPrintf(": %s",
875 SystemErrorCodeToString(err).c_str()).c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800876 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700877 }
878
879 // Always try to wait a bit for threads reading stdout/stderr to finish.
880 // If the process started ok, it should close the pipes causing the threads
881 // to finish. If the process had an error, it should exit, also causing
882 // the pipes to be closed. In that case we want to read all of the output
883 // and write it out so that the user can diagnose failures.
884 const DWORD thread_timeout_ms = 15 * 1000;
885 const HANDLE threads[] = { stdout_thread.get(), stderr_thread.get() };
886 const DWORD wait_result = WaitForMultipleObjects(arraysize(threads),
887 threads, TRUE, thread_timeout_ms);
888 if (wait_result == WAIT_TIMEOUT) {
889 // Threads did not finish after waiting a little while. Perhaps the
890 // server didn't close pipes, or it is hung.
891 fprintf(stderr, "Timed-out waiting for threads to finish reading from "
892 "ADB Server\n");
893 // Process handles are signaled when the process exits, so if we wait
894 // on the handle for 0 seconds and it returns 'timeout', that means that
895 // the process is still running.
896 if (WaitForSingleObject(process_handle.get(), 0) == WAIT_TIMEOUT) {
897 // We could TerminateProcess(), but that seems somewhat presumptive.
898 fprintf(stderr, "ADB Server is running: process id %lu\n",
899 pinfo.dwProcessId);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800900 }
Spencer Low2bbb3a92015-08-26 18:46:09 -0700901 return -1;
902 }
903
904 if (wait_result != WAIT_OBJECT_0) {
905 fprintf(stderr, "Unexpected result waiting for threads: %lu: %s\n",
906 wait_result, SystemErrorCodeToString(GetLastError()).c_str());
907 return -1;
908 }
909
910 // For now ignore the thread exit codes and assume they worked properly.
911
912 if (!got_ack) {
913 return -1;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800914 }
Yabin Cui2fa43212014-11-11 09:24:11 -0800915#else /* !defined(_WIN32) */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800916 char path[PATH_MAX];
917 int fd[2];
918
919 // set up a pipe so the child can tell us when it is ready.
Siva Velusamya55dbd82015-08-07 10:10:29 -0700920 // fd[0] will be parent's end, and the child will write on fd[1]
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800921 if (pipe(fd)) {
922 fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
923 return -1;
924 }
Alexey Tarasov857f17a2009-10-22 02:55:00 +1100925 get_my_path(path, PATH_MAX);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800926 pid_t pid = fork();
927 if(pid < 0) return -1;
928
929 if (pid == 0) {
930 // child side of the fork
931
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800932 adb_close(fd[0]);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800933
Matt Gumbel411775c2012-11-14 10:16:17 -0800934 char str_port[30];
Siva Velusamya55dbd82015-08-07 10:10:29 -0700935 snprintf(str_port, sizeof(str_port), "%d", server_port);
936 char reply_fd[30];
937 snprintf(reply_fd, sizeof(reply_fd), "%d", fd[1]);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800938 // child process
Siva Velusamya55dbd82015-08-07 10:10:29 -0700939 int result = execl(path, "adb", "-P", str_port, "fork-server", "server", "--reply-fd", reply_fd, NULL);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800940 // this should not return
941 fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno);
942 } else {
943 // parent side of the fork
944
945 char temp[3];
946
947 temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C';
948 // wait for the "OK\n" message
949 adb_close(fd[1]);
950 int ret = adb_read(fd[0], temp, 3);
JP Abgrall2e5dd6e2011-03-16 15:57:42 -0700951 int saved_errno = errno;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800952 adb_close(fd[0]);
953 if (ret < 0) {
JP Abgrall2e5dd6e2011-03-16 15:57:42 -0700954 fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", saved_errno);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800955 return -1;
956 }
957 if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
958 fprintf(stderr, "ADB server didn't ACK\n" );
959 return -1;
960 }
961
962 setsid();
963 }
Yabin Cui2fa43212014-11-11 09:24:11 -0800964#endif /* !defined(_WIN32) */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800965 return 0;
966}
Yabin Cui2fa43212014-11-11 09:24:11 -0800967#endif /* ADB_HOST */
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800968
David 'Digit' Turner963a4492013-03-21 21:07:42 +0100969// Try to handle a network forwarding request.
970// This returns 1 on success, 0 on failure, and -1 to indicate this is not
971// a forwarding-related request.
Elliott Hughes9d07fee2015-05-07 21:38:41 -0700972int handle_forward_request(const char* service, TransportType type, const char* serial, int reply_fd)
David 'Digit' Turner963a4492013-03-21 21:07:42 +0100973{
974 if (!strcmp(service, "list-forward")) {
975 // Create the list of forward redirections.
Elliott Hughes88b4c852015-04-30 17:32:03 -0700976 std::string listeners = format_listeners();
David 'Digit' Turner963a4492013-03-21 21:07:42 +0100977#if ADB_HOST
Elliott Hughes88b4c852015-04-30 17:32:03 -0700978 SendOkay(reply_fd);
David 'Digit' Turner963a4492013-03-21 21:07:42 +0100979#endif
Elliott Hughes88b4c852015-04-30 17:32:03 -0700980 SendProtocolString(reply_fd, listeners);
David 'Digit' Turner963a4492013-03-21 21:07:42 +0100981 return 1;
982 }
983
984 if (!strcmp(service, "killforward-all")) {
985 remove_all_listeners();
986#if ADB_HOST
987 /* On the host: 1st OKAY is connect, 2nd OKAY is status */
Elliott Hughes88b4c852015-04-30 17:32:03 -0700988 SendOkay(reply_fd);
David 'Digit' Turner963a4492013-03-21 21:07:42 +0100989#endif
Elliott Hughes88b4c852015-04-30 17:32:03 -0700990 SendOkay(reply_fd);
David 'Digit' Turner963a4492013-03-21 21:07:42 +0100991 return 1;
992 }
993
Elliott Hughesd98ca8a2015-05-29 17:55:19 -0700994 if (!strncmp(service, "forward:", 8) || !strncmp(service, "killforward:", 12)) {
995 // killforward:local
996 // forward:(norebind:)?local;remote
997 bool kill_forward = false;
998 bool no_rebind = false;
999 if (android::base::StartsWith(service, "killforward:")) {
1000 kill_forward = true;
1001 service += 12;
Spencer Low95365132015-07-29 17:21:21 -07001002 } else {
1003 service += 8; // skip past "forward:"
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001004 if (android::base::StartsWith(service, "norebind:")) {
1005 no_rebind = true;
1006 service += 9;
1007 }
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001008 }
1009
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001010 std::vector<std::string> pieces = android::base::Split(service, ";");
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001011
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001012 if (kill_forward) {
1013 // Check killforward: parameter format: '<local>'
1014 if (pieces.size() != 1 || pieces[0].empty()) {
1015 SendFail(reply_fd, android::base::StringPrintf("bad killforward: %s", service));
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001016 return 1;
1017 }
1018 } else {
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001019 // Check forward: parameter format: '<local>;<remote>'
1020 if (pieces.size() != 2 || pieces[0].empty() || pieces[1].empty() || pieces[1][0] == '*') {
1021 SendFail(reply_fd, android::base::StringPrintf("bad forward: %s", service));
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001022 return 1;
1023 }
1024 }
1025
Elliott Hughesab882422015-04-16 22:54:44 -07001026 std::string error_msg;
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001027 atransport* transport = acquire_one_transport(kCsAny, type, serial, &error_msg);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001028 if (!transport) {
Elliott Hughes88b4c852015-04-30 17:32:03 -07001029 SendFail(reply_fd, error_msg);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001030 return 1;
1031 }
1032
Spencer Low753d4852015-07-30 23:07:55 -07001033 std::string error;
Elliott Hughes3aec2ba2015-05-05 13:10:43 -07001034 InstallStatus r;
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001035 if (kill_forward) {
1036 r = remove_listener(pieces[0].c_str(), transport);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001037 } else {
Spencer Low753d4852015-07-30 23:07:55 -07001038 r = install_listener(pieces[0], pieces[1].c_str(), transport,
1039 no_rebind, &error);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001040 }
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001041 if (r == INSTALL_STATUS_OK) {
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001042#if ADB_HOST
1043 /* On the host: 1st OKAY is connect, 2nd OKAY is status */
Elliott Hughes88b4c852015-04-30 17:32:03 -07001044 SendOkay(reply_fd);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001045#endif
Elliott Hughes88b4c852015-04-30 17:32:03 -07001046 SendOkay(reply_fd);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001047 return 1;
1048 }
1049
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001050 std::string message;
1051 switch (r) {
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001052 case INSTALL_STATUS_OK: message = "success (!)"; break;
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001053 case INSTALL_STATUS_INTERNAL_ERROR: message = "internal error"; break;
1054 case INSTALL_STATUS_CANNOT_BIND:
Spencer Low32625852015-08-11 16:45:32 -07001055 message = android::base::StringPrintf("cannot bind listener: %s",
Spencer Low753d4852015-07-30 23:07:55 -07001056 error.c_str());
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001057 break;
1058 case INSTALL_STATUS_CANNOT_REBIND:
Spencer Low95365132015-07-29 17:21:21 -07001059 message = android::base::StringPrintf("cannot rebind existing socket");
Elliott Hughesa585cbd2015-04-20 08:09:20 -07001060 break;
Elliott Hughesd98ca8a2015-05-29 17:55:19 -07001061 case INSTALL_STATUS_LISTENER_NOT_FOUND:
1062 message = android::base::StringPrintf("listener '%s' not found", service);
1063 break;
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001064 }
Elliott Hughes88b4c852015-04-30 17:32:03 -07001065 SendFail(reply_fd, message);
David 'Digit' Turner963a4492013-03-21 21:07:42 +01001066 return 1;
1067 }
1068 return 0;
1069}
1070
Elliott Hughes24f52762015-06-23 13:00:32 -07001071#if ADB_HOST
1072static int SendOkay(int fd, const std::string& s) {
1073 SendOkay(fd);
1074 SendProtocolString(fd, s);
1075 return 0;
1076}
1077#endif
1078
Dan Albert53a37442015-05-08 16:13:53 -07001079int handle_host_request(const char* service, TransportType type,
1080 const char* serial, int reply_fd, asocket* s) {
1081 if (strcmp(service, "kill") == 0) {
1082 fprintf(stderr, "adb server killed by remote request\n");
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001083 fflush(stdout);
Elliott Hughes88b4c852015-04-30 17:32:03 -07001084 SendOkay(reply_fd);
Spencer Low85ee64b2015-08-11 17:05:02 -07001085
1086 // At least on Windows, if we exit() without shutdown(SD_SEND) or
1087 // closesocket(), the client's next recv() will error-out with
1088 // WSAECONNRESET and they'll never read the OKAY.
1089 adb_shutdown(reply_fd);
1090
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001091 exit(0);
1092 }
1093
1094#if ADB_HOST
1095 // "transport:" is used for switching transport with a specified serial number
1096 // "transport-usb:" is used for switching transport to the only USB transport
1097 // "transport-local:" is used for switching transport to the only local transport
1098 // "transport-any:" is used for switching transport to the only transport
1099 if (!strncmp(service, "transport", strlen("transport"))) {
Elliott Hughes3aec2ba2015-05-05 13:10:43 -07001100 TransportType type = kTransportAny;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001101
1102 if (!strncmp(service, "transport-usb", strlen("transport-usb"))) {
1103 type = kTransportUsb;
1104 } else if (!strncmp(service, "transport-local", strlen("transport-local"))) {
1105 type = kTransportLocal;
1106 } else if (!strncmp(service, "transport-any", strlen("transport-any"))) {
1107 type = kTransportAny;
1108 } else if (!strncmp(service, "transport:", strlen("transport:"))) {
1109 service += strlen("transport:");
Tom Marlin851ee3a2011-07-27 12:56:14 -05001110 serial = service;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001111 }
1112
Elliott Hughes24f52762015-06-23 13:00:32 -07001113 std::string error_msg;
1114 atransport* t = acquire_one_transport(kCsAny, type, serial, &error_msg);
1115 if (t != nullptr) {
1116 s->transport = t;
Elliott Hughes88b4c852015-04-30 17:32:03 -07001117 SendOkay(reply_fd);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001118 } else {
Elliott Hughes88b4c852015-04-30 17:32:03 -07001119 SendFail(reply_fd, error_msg);
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001120 }
1121 return 1;
1122 }
1123
1124 // return a list of all connected devices
Scott Anderson6dfaf4b2012-04-20 11:21:14 -07001125 if (!strncmp(service, "devices", 7)) {
Elliott Hughes88b4c852015-04-30 17:32:03 -07001126 bool long_listing = (strcmp(service+7, "-l") == 0);
1127 if (long_listing || service[7] == 0) {
1128 D("Getting device list...\n");
1129 std::string device_list = list_transports(long_listing);
1130 D("Sending device list...\n");
Elliott Hughes24f52762015-06-23 13:00:32 -07001131 return SendOkay(reply_fd, device_list);
Scott Anderson6dfaf4b2012-04-20 11:21:14 -07001132 }
Elliott Hughes88b4c852015-04-30 17:32:03 -07001133 return 1;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001134 }
1135
Dan Albert996c12e2015-05-20 18:58:41 -07001136 if (!strcmp(service, "features")) {
1137 SendOkay(reply_fd);
1138 SendProtocolString(
1139 reply_fd, android::base::Join(supported_features(), '\n'));
1140 return 0;
1141 }
1142
Mike Lockwood5f2c4a12009-10-11 23:04:18 -04001143 // remove TCP transport
1144 if (!strncmp(service, "disconnect:", 11)) {
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001145 const std::string address(service + 11);
1146 if (address.empty()) {
Yabin Cui4d64fd82015-08-27 12:03:11 -07001147 kick_all_tcp_devices();
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001148 return SendOkay(reply_fd, "disconnected everything");
Mike Lockwood5f2c4a12009-10-11 23:04:18 -04001149 }
1150
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001151 std::string serial;
1152 std::string host;
1153 int port = DEFAULT_ADB_LOCAL_TRANSPORT_PORT;
1154 std::string error;
1155 if (!parse_host_and_port(address, &serial, &host, &port, &error)) {
1156 return SendFail(reply_fd, android::base::StringPrintf("couldn't parse '%s': %s",
1157 address.c_str(), error.c_str()));
1158 }
1159 atransport* t = find_transport(serial.c_str());
1160 if (t == nullptr) {
1161 return SendFail(reply_fd, android::base::StringPrintf("no such device '%s'",
1162 serial.c_str()));
1163 }
Yabin Cui4d64fd82015-08-27 12:03:11 -07001164 kick_transport(t);
Elliott Hughes09ccf1f2015-07-18 12:21:30 -07001165 return SendOkay(reply_fd, android::base::StringPrintf("disconnected %s", address.c_str()));
Mike Lockwoodb51ae572009-08-24 15:58:40 -07001166 }
1167
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001168 // returns our value for ADB_SERVER_VERSION
1169 if (!strcmp(service, "version")) {
Elliott Hughes24f52762015-06-23 13:00:32 -07001170 return SendOkay(reply_fd, android::base::StringPrintf("%04x", ADB_SERVER_VERSION));
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001171 }
1172
Elliott Hughes24f52762015-06-23 13:00:32 -07001173 // These always report "unknown" rather than the actual error, for scripts.
1174 if (!strcmp(service, "get-serialno")) {
1175 std::string ignored;
1176 atransport* t = acquire_one_transport(kCsAny, type, serial, &ignored);
1177 return SendOkay(reply_fd, (t && t->serial) ? t->serial : "unknown");
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001178 }
Elliott Hughes24f52762015-06-23 13:00:32 -07001179 if (!strcmp(service, "get-devpath")) {
1180 std::string ignored;
1181 atransport* t = acquire_one_transport(kCsAny, type, serial, &ignored);
1182 return SendOkay(reply_fd, (t && t->devpath) ? t->devpath : "unknown");
Scott Anderson6dfaf4b2012-04-20 11:21:14 -07001183 }
Elliott Hughes24f52762015-06-23 13:00:32 -07001184 if (!strcmp(service, "get-state")) {
1185 std::string ignored;
1186 atransport* t = acquire_one_transport(kCsAny, type, serial, &ignored);
1187 return SendOkay(reply_fd, t ? t->connection_state_name() : "unknown");
1188 }
1189
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001190 // indicates a new emulator instance has started
Elliott Hughes24f52762015-06-23 13:00:32 -07001191 if (!strncmp(service, "emulator:", 9)) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001192 int port = atoi(service+9);
1193 local_connect(port);
1194 /* we don't even need to send a reply */
1195 return 0;
1196 }
Simon Yee51c11e2014-07-14 17:23:06 -07001197#endif // ADB_HOST
1198
Elliott Hughes3aec2ba2015-05-05 13:10:43 -07001199 int ret = handle_forward_request(service, type, serial, reply_fd);
Simon Yee51c11e2014-07-14 17:23:06 -07001200 if (ret >= 0)
1201 return ret - 1;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -08001202 return -1;
1203}