blob: d37ca363a22a703563a20d265d24cf24a1f1ebae [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
Dan Albert33134262015-03-19 15:21:08 -070017#define TRACE_TAG TRACE_ADB
18
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>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080034
Benoit Gobyd5fcafa2012-04-12 12:23:49 -070035#include "adb_auth.h"
Dan Albertcc731cc2015-02-24 21:26:58 -080036#include "adb_io.h"
Dan Alberte9fca142015-02-18 18:03:26 -080037#include "adb_listeners.h"
Dan Albert76649012015-02-24 15:51:19 -080038#include "transport.h"
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080039
Scott Andersone82c2db2012-05-25 14:10:02 -070040#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
41
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080042#if !ADB_HOST
Nick Kralevich893a4a42013-05-23 09:54:13 -070043#include <cutils/properties.h>
Nick Kraleviche2864bf2013-02-28 14:12:58 -080044#include <sys/capability.h>
Jeff Sharkey885342a2012-08-14 21:00:22 -070045#include <sys/mount.h>
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080046#endif
47
JP Abgrall408fa572011-03-16 15:57:42 -070048#if ADB_TRACE
49ADB_MUTEX_DEFINE( D_lock );
50#endif
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080051
52int HOST = 0;
53
Scott Andersone82c2db2012-05-25 14:10:02 -070054#if !ADB_HOST
Dan Albertbd0b7502015-02-18 18:22:45 -080055const char *adb_device_banner = "device";
Scott Andersone82c2db2012-05-25 14:10:02 -070056#endif
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080057
58void fatal(const char *fmt, ...)
59{
60 va_list ap;
61 va_start(ap, fmt);
62 fprintf(stderr, "error: ");
63 vfprintf(stderr, fmt, ap);
64 fprintf(stderr, "\n");
65 va_end(ap);
66 exit(-1);
67}
68
69void fatal_errno(const char *fmt, ...)
70{
71 va_list ap;
72 va_start(ap, fmt);
73 fprintf(stderr, "error: %s: ", strerror(errno));
74 vfprintf(stderr, fmt, ap);
75 fprintf(stderr, "\n");
76 va_end(ap);
77 exit(-1);
78}
79
Dan Albertea2175a2015-03-08 21:12:08 -070080#if !ADB_HOST
81void start_device_log(void) {
82 adb_mkdir("/data/adb", 0775);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -080083
Dan Albertea2175a2015-03-08 21:12:08 -070084 struct tm now;
85 time_t t;
86 tzset();
87 time(&t);
88 localtime_r(&t, &now);
89
90 char path[PATH_MAX];
91 strftime(path, sizeof(path), "/data/adb/adb-%Y-%m-%d-%H-%M-%S.txt", &now);
92
93 int fd = unix_open(path, O_WRONLY | O_CREAT | O_TRUNC, 0640);
94 if (fd == -1) {
95 return;
96 }
97
98 // redirect stdout and stderr to the log file
99 dup2(fd, STDOUT_FILENO);
100 dup2(fd, STDERR_FILENO);
101 fprintf(stderr, "--- adb starting (pid %d) ---\n", getpid());
102 adb_close(fd);
103
104 fd = unix_open("/dev/null", O_RDONLY);
105 dup2(fd, 0);
106 adb_close(fd);
107}
108#endif
109
110int adb_trace_mask;
111
112std::string get_trace_setting_from_env() {
113 const char* setting = getenv("ADB_TRACE");
114 if (setting == nullptr) {
115 setting = "";
116 }
117
118 return std::string(setting);
119}
120
121#if !ADB_HOST
122std::string get_trace_setting_from_prop() {
123 char buf[PROPERTY_VALUE_MAX];
124 property_get("persist.adb.trace_mask", buf, "");
125 return std::string(buf);
126}
127#endif
128
129std::string get_trace_setting() {
130#if ADB_HOST
131 return get_trace_setting_from_env();
132#else
133 return get_trace_setting_from_prop();
134#endif
135}
136
137// Split the comma/space/colum/semi-column separated list of tags from the trace
138// setting and build the trace mask from it. note that '1' and 'all' are special
139// cases to enable all tracing.
140//
141// adb's trace setting comes from the ADB_TRACE environment variable, whereas
142// adbd's comes from the system property persist.adb.trace_mask.
143void adb_trace_init() {
144 const std::string trace_setting = get_trace_setting();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800145
146 static const struct {
147 const char* tag;
148 int flag;
149 } tags[] = {
150 { "1", 0 },
151 { "all", 0 },
152 { "adb", TRACE_ADB },
153 { "sockets", TRACE_SOCKETS },
154 { "packets", TRACE_PACKETS },
155 { "rwx", TRACE_RWX },
156 { "usb", TRACE_USB },
157 { "sync", TRACE_SYNC },
158 { "sysdeps", TRACE_SYSDEPS },
159 { "transport", TRACE_TRANSPORT },
160 { "jdwp", TRACE_JDWP },
JP Abgrall408fa572011-03-16 15:57:42 -0700161 { "services", TRACE_SERVICES },
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700162 { "auth", TRACE_AUTH },
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800163 { NULL, 0 }
164 };
165
Dan Albertea2175a2015-03-08 21:12:08 -0700166 if (trace_setting.empty()) {
167 return;
168 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800169
Dan Albertea2175a2015-03-08 21:12:08 -0700170 // Use a comma/colon/semi-colon/space separated list
171 const char* p = trace_setting.c_str();
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800172 while (*p) {
173 int len, tagn;
174
Dan Albertea2175a2015-03-08 21:12:08 -0700175 const char* q = strpbrk(p, " ,:;");
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800176 if (q == NULL) {
177 q = p + strlen(p);
178 }
179 len = q - p;
180
Dan Albertea2175a2015-03-08 21:12:08 -0700181 for (tagn = 0; tags[tagn].tag != NULL; tagn++) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800182 int taglen = strlen(tags[tagn].tag);
183
Dan Albertea2175a2015-03-08 21:12:08 -0700184 if (len == taglen && !memcmp(tags[tagn].tag, p, len)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800185 int flag = tags[tagn].flag;
186 if (flag == 0) {
187 adb_trace_mask = ~0;
188 return;
189 }
190 adb_trace_mask |= (1 << flag);
191 break;
192 }
193 }
194 p = q;
195 if (*p)
196 p++;
197 }
Dan Albertea2175a2015-03-08 21:12:08 -0700198
199#if !ADB_HOST
200 start_device_log();
201#endif
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800202}
203
Dan Albertbac34742015-02-25 17:51:28 -0800204apacket* get_apacket(void)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800205{
Dan Albertbac34742015-02-25 17:51:28 -0800206 apacket* p = reinterpret_cast<apacket*>(malloc(sizeof(apacket)));
207 if (p == nullptr) {
208 fatal("failed to allocate an apacket");
209 }
210
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800211 memset(p, 0, sizeof(apacket) - MAX_PAYLOAD);
212 return p;
213}
214
215void put_apacket(apacket *p)
216{
217 free(p);
218}
219
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700220void handle_online(atransport *t)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800221{
222 D("adb: online\n");
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700223 t->online = 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800224}
225
226void handle_offline(atransport *t)
227{
228 D("adb: offline\n");
229 //Close the associated usb
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700230 t->online = 0;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800231 run_transport_disconnects(t);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800232}
233
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700234#if DEBUG_PACKETS
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800235#define DUMPMAX 32
236void print_packet(const char *label, apacket *p)
237{
238 char *tag;
239 char *x;
240 unsigned count;
241
242 switch(p->msg.command){
243 case A_SYNC: tag = "SYNC"; break;
244 case A_CNXN: tag = "CNXN" ; break;
245 case A_OPEN: tag = "OPEN"; break;
246 case A_OKAY: tag = "OKAY"; break;
247 case A_CLSE: tag = "CLSE"; break;
248 case A_WRTE: tag = "WRTE"; break;
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700249 case A_AUTH: tag = "AUTH"; break;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800250 default: tag = "????"; break;
251 }
252
253 fprintf(stderr, "%s: %s %08x %08x %04x \"",
254 label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length);
255 count = p->msg.data_length;
256 x = (char*) p->data;
257 if(count > DUMPMAX) {
258 count = DUMPMAX;
259 tag = "\n";
260 } else {
261 tag = "\"\n";
262 }
263 while(count-- > 0){
264 if((*x >= ' ') && (*x < 127)) {
265 fputc(*x, stderr);
266 } else {
267 fputc('.', stderr);
268 }
269 x++;
270 }
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700271 fputs(tag, stderr);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800272}
273#endif
274
275static void send_ready(unsigned local, unsigned remote, atransport *t)
276{
277 D("Calling send_ready \n");
278 apacket *p = get_apacket();
279 p->msg.command = A_OKAY;
280 p->msg.arg0 = local;
281 p->msg.arg1 = remote;
282 send_packet(p, t);
283}
284
285static void send_close(unsigned local, unsigned remote, atransport *t)
286{
287 D("Calling send_close \n");
288 apacket *p = get_apacket();
289 p->msg.command = A_CLSE;
290 p->msg.arg0 = local;
291 p->msg.arg1 = remote;
292 send_packet(p, t);
293}
294
Scott Andersone82c2db2012-05-25 14:10:02 -0700295static size_t fill_connect_data(char *buf, size_t bufsize)
296{
297#if ADB_HOST
298 return snprintf(buf, bufsize, "host::") + 1;
299#else
300 static const char *cnxn_props[] = {
301 "ro.product.name",
302 "ro.product.model",
303 "ro.product.device",
304 };
305 static const int num_cnxn_props = ARRAY_SIZE(cnxn_props);
306 int i;
307 size_t remaining = bufsize;
308 size_t len;
309
310 len = snprintf(buf, remaining, "%s::", adb_device_banner);
311 remaining -= len;
312 buf += len;
313 for (i = 0; i < num_cnxn_props; i++) {
314 char value[PROPERTY_VALUE_MAX];
315 property_get(cnxn_props[i], value, "");
316 len = snprintf(buf, remaining, "%s=%s;", cnxn_props[i], value);
317 remaining -= len;
318 buf += len;
319 }
320
321 return bufsize - remaining + 1;
322#endif
323}
324
David 'Digit' Turner25258692013-03-21 21:07:42 +0100325#if !ADB_HOST
326static void send_msg_with_header(int fd, const char* msg, size_t msglen) {
327 char header[5];
328 if (msglen > 0xffff)
329 msglen = 0xffff;
330 snprintf(header, sizeof(header), "%04x", (unsigned)msglen);
Dan Albertcc731cc2015-02-24 21:26:58 -0800331 WriteFdExactly(fd, header, 4);
332 WriteFdExactly(fd, msg, msglen);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100333}
334#endif
335
Chih-Hung Hsiehf787b382014-09-05 15:38:15 -0700336#if ADB_HOST
David 'Digit' Turner25258692013-03-21 21:07:42 +0100337static void send_msg_with_okay(int fd, const char* msg, size_t msglen) {
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100338 char header[9];
339 if (msglen > 0xffff)
340 msglen = 0xffff;
341 snprintf(header, sizeof(header), "OKAY%04x", (unsigned)msglen);
Dan Albertcc731cc2015-02-24 21:26:58 -0800342 WriteFdExactly(fd, header, 8);
343 WriteFdExactly(fd, msg, msglen);
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100344}
Chih-Hung Hsiehf787b382014-09-05 15:38:15 -0700345#endif // ADB_HOST
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100346
Dan Albertba3a2512015-02-18 17:47:33 -0800347void send_connect(atransport *t)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800348{
349 D("Calling send_connect \n");
350 apacket *cp = get_apacket();
351 cp->msg.command = A_CNXN;
352 cp->msg.arg0 = A_VERSION;
353 cp->msg.arg1 = MAX_PAYLOAD;
Scott Andersone82c2db2012-05-25 14:10:02 -0700354 cp->msg.data_length = fill_connect_data((char *)cp->data,
355 sizeof(cp->data));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800356 send_packet(cp, t);
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700357}
358
Chih-Hung Hsiehf787b382014-09-05 15:38:15 -0700359#if ADB_HOST
Dan Albertbac34742015-02-25 17:51:28 -0800360static const char* connection_state_name(atransport *t)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800361{
362 if (t == NULL) {
363 return "unknown";
364 }
365
366 switch(t->connection_state) {
367 case CS_BOOTLOADER:
368 return "bootloader";
369 case CS_DEVICE:
370 return "device";
trevda5ad5392013-04-17 14:34:23 +0100371 case CS_RECOVERY:
372 return "recovery";
373 case CS_SIDELOAD:
374 return "sideload";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800375 case CS_OFFLINE:
376 return "offline";
Benoit Goby77e8e582013-01-15 12:36:47 -0800377 case CS_UNAUTHORIZED:
378 return "unauthorized";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800379 default:
380 return "unknown";
381 }
382}
Chih-Hung Hsiehf787b382014-09-05 15:38:15 -0700383#endif // ADB_HOST
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800384
Scott Andersone82c2db2012-05-25 14:10:02 -0700385/* qual_overwrite is used to overwrite a qualifier string. dst is a
386 * pointer to a char pointer. It is assumed that if *dst is non-NULL, it
Scott Anderson2ca3e6b2012-05-30 18:11:27 -0700387 * was malloc'ed and needs to freed. *dst will be set to a dup of src.
Scott Andersone82c2db2012-05-25 14:10:02 -0700388 */
389static void qual_overwrite(char **dst, const char *src)
390{
391 if (!dst)
392 return;
393
394 free(*dst);
395 *dst = NULL;
396
397 if (!src || !*src)
398 return;
399
400 *dst = strdup(src);
401}
402
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800403void parse_banner(char *banner, atransport *t)
404{
Scott Andersone82c2db2012-05-25 14:10:02 -0700405 static const char *prop_seps = ";";
406 static const char key_val_sep = '=';
Scott Anderson2ca3e6b2012-05-30 18:11:27 -0700407 char *cp;
408 char *type;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800409
410 D("parse_banner: %s\n", banner);
411 type = banner;
Scott Andersone82c2db2012-05-25 14:10:02 -0700412 cp = strchr(type, ':');
413 if (cp) {
414 *cp++ = 0;
415 /* Nothing is done with second field. */
416 cp = strchr(cp, ':');
417 if (cp) {
418 char *save;
419 char *key;
Scott Anderson1b7a7e82012-06-05 17:54:27 -0700420 key = adb_strtok_r(cp + 1, prop_seps, &save);
Scott Andersone82c2db2012-05-25 14:10:02 -0700421 while (key) {
422 cp = strchr(key, key_val_sep);
423 if (cp) {
424 *cp++ = '\0';
425 if (!strcmp(key, "ro.product.name"))
426 qual_overwrite(&t->product, cp);
427 else if (!strcmp(key, "ro.product.model"))
428 qual_overwrite(&t->model, cp);
429 else if (!strcmp(key, "ro.product.device"))
430 qual_overwrite(&t->device, cp);
431 }
Scott Anderson1b7a7e82012-06-05 17:54:27 -0700432 key = adb_strtok_r(NULL, prop_seps, &save);
Scott Andersone82c2db2012-05-25 14:10:02 -0700433 }
434 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800435 }
436
437 if(!strcmp(type, "bootloader")){
438 D("setting connection_state to CS_BOOTLOADER\n");
439 t->connection_state = CS_BOOTLOADER;
440 update_transports();
441 return;
442 }
443
444 if(!strcmp(type, "device")) {
445 D("setting connection_state to CS_DEVICE\n");
446 t->connection_state = CS_DEVICE;
447 update_transports();
448 return;
449 }
450
451 if(!strcmp(type, "recovery")) {
452 D("setting connection_state to CS_RECOVERY\n");
453 t->connection_state = CS_RECOVERY;
454 update_transports();
455 return;
456 }
457
Doug Zongker447f0612012-01-09 14:54:53 -0800458 if(!strcmp(type, "sideload")) {
459 D("setting connection_state to CS_SIDELOAD\n");
460 t->connection_state = CS_SIDELOAD;
461 update_transports();
462 return;
463 }
464
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800465 t->connection_state = CS_HOST;
466}
467
468void handle_packet(apacket *p, atransport *t)
469{
470 asocket *s;
471
Viral Mehta899913f2010-06-16 18:41:28 +0530472 D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0],
473 ((char*) (&(p->msg.command)))[1],
474 ((char*) (&(p->msg.command)))[2],
475 ((char*) (&(p->msg.command)))[3]);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800476 print_packet("recv", p);
477
478 switch(p->msg.command){
479 case A_SYNC:
480 if(p->msg.arg0){
481 send_packet(p, t);
482 if(HOST) send_connect(t);
483 } else {
484 t->connection_state = CS_OFFLINE;
485 handle_offline(t);
486 send_packet(p, t);
487 }
488 return;
489
490 case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */
491 /* XXX verify version, etc */
492 if(t->connection_state != CS_OFFLINE) {
493 t->connection_state = CS_OFFLINE;
494 handle_offline(t);
495 }
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700496
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800497 parse_banner((char*) p->data, t);
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700498
499 if (HOST || !auth_enabled) {
500 handle_online(t);
501 if(!HOST) send_connect(t);
502 } else {
503 send_auth_request(t);
504 }
505 break;
506
507 case A_AUTH:
508 if (p->msg.arg0 == ADB_AUTH_TOKEN) {
Benoit Goby77e8e582013-01-15 12:36:47 -0800509 t->connection_state = CS_UNAUTHORIZED;
Benoit Gobyd5fcafa2012-04-12 12:23:49 -0700510 t->key = adb_auth_nextkey(t->key);
511 if (t->key) {
512 send_auth_response(p->data, p->msg.data_length, t);
513 } else {
514 /* No more private keys to try, send the public key */
515 send_auth_publickey(t);
516 }
517 } else if (p->msg.arg0 == ADB_AUTH_SIGNATURE) {
518 if (adb_auth_verify(t->token, p->data, p->msg.data_length)) {
519 adb_auth_verified(t);
520 t->failed_auth_attempts = 0;
521 } else {
522 if (t->failed_auth_attempts++ > 10)
523 adb_sleep_ms(1000);
524 send_auth_request(t);
525 }
526 } else if (p->msg.arg0 == ADB_AUTH_RSAPUBLICKEY) {
527 adb_auth_confirm_key(p->data, p->msg.data_length, t);
528 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800529 break;
530
531 case A_OPEN: /* OPEN(local-id, 0, "destination") */
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100532 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 == 0) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800533 char *name = (char*) p->data;
534 name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0;
535 s = create_local_service_socket(name);
536 if(s == 0) {
537 send_close(0, p->msg.arg0, t);
538 } else {
539 s->peer = create_remote_socket(p->msg.arg0, t);
540 s->peer->peer = s;
541 send_ready(s->id, s->peer->id, t);
542 s->ready(s);
543 }
544 }
545 break;
546
547 case A_OKAY: /* READY(local-id, remote-id, "") */
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100548 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
549 if((s = find_local_socket(p->msg.arg1, 0))) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800550 if(s->peer == 0) {
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100551 /* On first READY message, create the connection. */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800552 s->peer = create_remote_socket(p->msg.arg0, t);
553 s->peer->peer = s;
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100554 s->ready(s);
555 } else if (s->peer->id == p->msg.arg0) {
556 /* Other READY messages must use the same local-id */
557 s->ready(s);
558 } else {
559 D("Invalid A_OKAY(%d,%d), expected A_OKAY(%d,%d) on transport %s\n",
560 p->msg.arg0, p->msg.arg1, s->peer->id, p->msg.arg1, t->serial);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800561 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800562 }
563 }
564 break;
565
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100566 case A_CLSE: /* CLOSE(local-id, remote-id, "") or CLOSE(0, remote-id, "") */
567 if (t->online && p->msg.arg1 != 0) {
568 if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
569 /* According to protocol.txt, p->msg.arg0 might be 0 to indicate
570 * a failed OPEN only. However, due to a bug in previous ADB
571 * versions, CLOSE(0, remote-id, "") was also used for normal
572 * CLOSE() operations.
573 *
574 * This is bad because it means a compromised adbd could
575 * send packets to close connections between the host and
576 * other devices. To avoid this, only allow this if the local
577 * socket has a peer on the same transport.
578 */
579 if (p->msg.arg0 == 0 && s->peer && s->peer->transport != t) {
580 D("Invalid A_CLSE(0, %u) from transport %s, expected transport %s\n",
581 p->msg.arg1, t->serial, s->peer->transport->serial);
582 } else {
583 s->close(s);
584 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800585 }
586 }
587 break;
588
David 'Digit' Turner818d6412013-12-13 14:09:44 +0100589 case A_WRTE: /* WRITE(local-id, remote-id, <data>) */
590 if (t->online && p->msg.arg0 != 0 && p->msg.arg1 != 0) {
591 if((s = find_local_socket(p->msg.arg1, p->msg.arg0))) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800592 unsigned rid = p->msg.arg0;
593 p->len = p->msg.data_length;
594
595 if(s->enqueue(s, p) == 0) {
596 D("Enqueue the socket\n");
597 send_ready(s->id, rid, t);
598 }
599 return;
600 }
601 }
602 break;
603
604 default:
605 printf("handle_packet: what is %08x?!\n", p->msg.command);
606 }
607
608 put_apacket(p);
609}
610
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800611#if ADB_HOST
JP Abgrall571c1362012-12-06 18:18:12 -0800612
Stefan Hilzingera84a42e2010-04-19 12:21:12 +0100613int launch_server(int server_port)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800614{
Yabin Cuie77b6a02014-11-11 09:24:11 -0800615#if defined(_WIN32)
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800616 /* we need to start the server in the background */
617 /* we create a PIPE that will be used to wait for the server's "OK" */
618 /* message since the pipe handles must be inheritable, we use a */
619 /* security attribute */
620 HANDLE pipe_read, pipe_write;
Ray Donnelly267aa8b2012-11-29 01:18:50 +0000621 HANDLE stdout_handle, stderr_handle;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800622 SECURITY_ATTRIBUTES sa;
623 STARTUPINFO startup;
624 PROCESS_INFORMATION pinfo;
625 char program_path[ MAX_PATH ];
626 int ret;
627
628 sa.nLength = sizeof(sa);
629 sa.lpSecurityDescriptor = NULL;
630 sa.bInheritHandle = TRUE;
631
632 /* create pipe, and ensure its read handle isn't inheritable */
633 ret = CreatePipe( &pipe_read, &pipe_write, &sa, 0 );
634 if (!ret) {
635 fprintf(stderr, "CreatePipe() failure, error %ld\n", GetLastError() );
636 return -1;
637 }
638
639 SetHandleInformation( pipe_read, HANDLE_FLAG_INHERIT, 0 );
640
Ray Donnelly267aa8b2012-11-29 01:18:50 +0000641 /* Some programs want to launch an adb command and collect its output by
642 * calling CreateProcess with inheritable stdout/stderr handles, then
643 * using read() to get its output. When this happens, the stdout/stderr
644 * handles passed to the adb client process will also be inheritable.
645 * When starting the adb server here, care must be taken to reset them
646 * to non-inheritable.
647 * Otherwise, something bad happens: even if the adb command completes,
648 * the calling process is stuck while read()-ing from the stdout/stderr
649 * descriptors, because they're connected to corresponding handles in the
650 * adb server process (even if the latter never uses/writes to them).
651 */
652 stdout_handle = GetStdHandle( STD_OUTPUT_HANDLE );
653 stderr_handle = GetStdHandle( STD_ERROR_HANDLE );
654 if (stdout_handle != INVALID_HANDLE_VALUE) {
655 SetHandleInformation( stdout_handle, HANDLE_FLAG_INHERIT, 0 );
656 }
657 if (stderr_handle != INVALID_HANDLE_VALUE) {
658 SetHandleInformation( stderr_handle, HANDLE_FLAG_INHERIT, 0 );
659 }
660
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800661 ZeroMemory( &startup, sizeof(startup) );
662 startup.cb = sizeof(startup);
663 startup.hStdInput = GetStdHandle( STD_INPUT_HANDLE );
664 startup.hStdOutput = pipe_write;
665 startup.hStdError = GetStdHandle( STD_ERROR_HANDLE );
666 startup.dwFlags = STARTF_USESTDHANDLES;
667
668 ZeroMemory( &pinfo, sizeof(pinfo) );
669
670 /* get path of current program */
671 GetModuleFileName( NULL, program_path, sizeof(program_path) );
Wenhao Lia09558c2013-11-13 16:23:37 +0800672 char args[64];
673 snprintf(args, sizeof(args), "adb -P %d fork-server server", server_port);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800674 ret = CreateProcess(
675 program_path, /* program path */
Wenhao Lia09558c2013-11-13 16:23:37 +0800676 args,
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800677 /* the fork-server argument will set the
678 debug = 2 in the child */
679 NULL, /* process handle is not inheritable */
680 NULL, /* thread handle is not inheritable */
681 TRUE, /* yes, inherit some handles */
682 DETACHED_PROCESS, /* the new process doesn't have a console */
683 NULL, /* use parent's environment block */
684 NULL, /* use parent's starting directory */
685 &startup, /* startup info, i.e. std handles */
686 &pinfo );
687
688 CloseHandle( pipe_write );
689
690 if (!ret) {
691 fprintf(stderr, "CreateProcess failure, error %ld\n", GetLastError() );
692 CloseHandle( pipe_read );
693 return -1;
694 }
695
696 CloseHandle( pinfo.hProcess );
697 CloseHandle( pinfo.hThread );
698
699 /* wait for the "OK\n" message */
700 {
701 char temp[3];
702 DWORD count;
703
704 ret = ReadFile( pipe_read, temp, 3, &count, NULL );
705 CloseHandle( pipe_read );
706 if ( !ret ) {
707 fprintf(stderr, "could not read ok from ADB Server, error = %ld\n", GetLastError() );
708 return -1;
709 }
710 if (count != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
711 fprintf(stderr, "ADB server didn't ACK\n" );
712 return -1;
713 }
714 }
Yabin Cuie77b6a02014-11-11 09:24:11 -0800715#else /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800716 char path[PATH_MAX];
717 int fd[2];
718
719 // set up a pipe so the child can tell us when it is ready.
720 // fd[0] will be parent's end, and fd[1] will get mapped to stderr in the child.
721 if (pipe(fd)) {
722 fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno);
723 return -1;
724 }
Alexey Tarasov31664102009-10-22 02:55:00 +1100725 get_my_path(path, PATH_MAX);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800726 pid_t pid = fork();
727 if(pid < 0) return -1;
728
729 if (pid == 0) {
730 // child side of the fork
731
732 // redirect stderr to the pipe
733 // we use stderr instead of stdout due to stdout's buffering behavior.
734 adb_close(fd[0]);
735 dup2(fd[1], STDERR_FILENO);
736 adb_close(fd[1]);
737
Matt Gumbeld7b33082012-11-14 10:16:17 -0800738 char str_port[30];
739 snprintf(str_port, sizeof(str_port), "%d", server_port);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800740 // child process
Matt Gumbeld7b33082012-11-14 10:16:17 -0800741 int result = execl(path, "adb", "-P", str_port, "fork-server", "server", NULL);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800742 // this should not return
743 fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno);
744 } else {
745 // parent side of the fork
746
747 char temp[3];
748
749 temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C';
750 // wait for the "OK\n" message
751 adb_close(fd[1]);
752 int ret = adb_read(fd[0], temp, 3);
JP Abgrall408fa572011-03-16 15:57:42 -0700753 int saved_errno = errno;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800754 adb_close(fd[0]);
755 if (ret < 0) {
JP Abgrall408fa572011-03-16 15:57:42 -0700756 fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", saved_errno);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800757 return -1;
758 }
759 if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') {
760 fprintf(stderr, "ADB server didn't ACK\n" );
761 return -1;
762 }
763
764 setsid();
765 }
Yabin Cuie77b6a02014-11-11 09:24:11 -0800766#endif /* !defined(_WIN32) */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800767 return 0;
768}
Yabin Cuie77b6a02014-11-11 09:24:11 -0800769#endif /* ADB_HOST */
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800770
David 'Digit' Turner25258692013-03-21 21:07:42 +0100771// Try to handle a network forwarding request.
772// This returns 1 on success, 0 on failure, and -1 to indicate this is not
773// a forwarding-related request.
774int handle_forward_request(const char* service, transport_type ttype, char* serial, int reply_fd)
775{
776 if (!strcmp(service, "list-forward")) {
777 // Create the list of forward redirections.
778 int buffer_size = format_listeners(NULL, 0);
779 // Add one byte for the trailing zero.
Dan Albertbac34742015-02-25 17:51:28 -0800780 char* buffer = reinterpret_cast<char*>(malloc(buffer_size + 1));
781 if (buffer == nullptr) {
David 'Digit' Turner25258692013-03-21 21:07:42 +0100782 sendfailmsg(reply_fd, "not enough memory");
783 return 1;
784 }
785 (void) format_listeners(buffer, buffer_size + 1);
786#if ADB_HOST
787 send_msg_with_okay(reply_fd, buffer, buffer_size);
788#else
789 send_msg_with_header(reply_fd, buffer, buffer_size);
790#endif
791 free(buffer);
792 return 1;
793 }
794
795 if (!strcmp(service, "killforward-all")) {
796 remove_all_listeners();
797#if ADB_HOST
798 /* On the host: 1st OKAY is connect, 2nd OKAY is status */
799 adb_write(reply_fd, "OKAY", 4);
800#endif
801 adb_write(reply_fd, "OKAY", 4);
802 return 1;
803 }
804
805 if (!strncmp(service, "forward:",8) ||
806 !strncmp(service, "killforward:",12)) {
Dan Albertbac34742015-02-25 17:51:28 -0800807 char *local, *remote;
David 'Digit' Turner25258692013-03-21 21:07:42 +0100808 int r;
809 atransport *transport;
810
811 int createForward = strncmp(service, "kill", 4);
812 int no_rebind = 0;
813
814 local = strchr(service, ':') + 1;
815
816 // Handle forward:norebind:<local>... here
817 if (createForward && !strncmp(local, "norebind:", 9)) {
818 no_rebind = 1;
819 local = strchr(local, ':') + 1;
820 }
821
822 remote = strchr(local,';');
823
824 if (createForward) {
825 // Check forward: parameter format: '<local>;<remote>'
826 if(remote == 0) {
827 sendfailmsg(reply_fd, "malformed forward spec");
828 return 1;
829 }
830
831 *remote++ = 0;
832 if((local[0] == 0) || (remote[0] == 0) || (remote[0] == '*')) {
833 sendfailmsg(reply_fd, "malformed forward spec");
834 return 1;
835 }
836 } else {
837 // Check killforward: parameter format: '<local>'
838 if (local[0] == 0) {
839 sendfailmsg(reply_fd, "malformed forward spec");
840 return 1;
841 }
842 }
843
Dan Albertbac34742015-02-25 17:51:28 -0800844 const char* err;
David 'Digit' Turner25258692013-03-21 21:07:42 +0100845 transport = acquire_one_transport(CS_ANY, ttype, serial, &err);
846 if (!transport) {
847 sendfailmsg(reply_fd, err);
848 return 1;
849 }
850
851 if (createForward) {
852 r = install_listener(local, remote, transport, no_rebind);
853 } else {
854 r = remove_listener(local, transport);
855 }
856 if(r == 0) {
857#if ADB_HOST
858 /* On the host: 1st OKAY is connect, 2nd OKAY is status */
Dan Albertcc731cc2015-02-24 21:26:58 -0800859 WriteFdExactly(reply_fd, "OKAY", 4);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100860#endif
Dan Albertcc731cc2015-02-24 21:26:58 -0800861 WriteFdExactly(reply_fd, "OKAY", 4);
David 'Digit' Turner25258692013-03-21 21:07:42 +0100862 return 1;
863 }
864
865 if (createForward) {
866 const char* message;
867 switch (r) {
868 case INSTALL_STATUS_CANNOT_BIND:
869 message = "cannot bind to socket";
870 break;
871 case INSTALL_STATUS_CANNOT_REBIND:
872 message = "cannot rebind existing socket";
873 break;
874 default:
875 message = "internal error";
876 }
877 sendfailmsg(reply_fd, message);
878 } else {
879 sendfailmsg(reply_fd, "cannot remove listener");
880 }
881 return 1;
882 }
883 return 0;
884}
885
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800886int handle_host_request(char *service, transport_type ttype, char* serial, int reply_fd, asocket *s)
887{
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800888 if(!strcmp(service, "kill")) {
889 fprintf(stderr,"adb server killed by remote request\n");
890 fflush(stdout);
891 adb_write(reply_fd, "OKAY", 4);
892 usb_cleanup();
893 exit(0);
894 }
895
896#if ADB_HOST
Chih-Hung Hsiehf787b382014-09-05 15:38:15 -0700897 atransport *transport = NULL;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800898 // "transport:" is used for switching transport with a specified serial number
899 // "transport-usb:" is used for switching transport to the only USB transport
900 // "transport-local:" is used for switching transport to the only local transport
901 // "transport-any:" is used for switching transport to the only transport
902 if (!strncmp(service, "transport", strlen("transport"))) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800903 transport_type type = kTransportAny;
904
905 if (!strncmp(service, "transport-usb", strlen("transport-usb"))) {
906 type = kTransportUsb;
907 } else if (!strncmp(service, "transport-local", strlen("transport-local"))) {
908 type = kTransportLocal;
909 } else if (!strncmp(service, "transport-any", strlen("transport-any"))) {
910 type = kTransportAny;
911 } else if (!strncmp(service, "transport:", strlen("transport:"))) {
912 service += strlen("transport:");
Tom Marlin3175c8e2011-07-27 12:56:14 -0500913 serial = service;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800914 }
915
Dan Albertbac34742015-02-25 17:51:28 -0800916 const char* error_string = "unknown failure";
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800917 transport = acquire_one_transport(CS_ANY, type, serial, &error_string);
918
919 if (transport) {
920 s->transport = transport;
921 adb_write(reply_fd, "OKAY", 4);
922 } else {
923 sendfailmsg(reply_fd, error_string);
924 }
925 return 1;
926 }
927
928 // return a list of all connected devices
Scott Andersone109d262012-04-20 11:21:14 -0700929 if (!strncmp(service, "devices", 7)) {
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800930 char buffer[4096];
Scott Andersone109d262012-04-20 11:21:14 -0700931 int use_long = !strcmp(service+7, "-l");
932 if (use_long || service[7] == 0) {
Scott Andersone109d262012-04-20 11:21:14 -0700933 memset(buffer, 0, sizeof(buffer));
934 D("Getting device list \n");
935 list_transports(buffer, sizeof(buffer), use_long);
Scott Andersone109d262012-04-20 11:21:14 -0700936 D("Wrote device list \n");
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100937 send_msg_with_okay(reply_fd, buffer, strlen(buffer));
Scott Andersone109d262012-04-20 11:21:14 -0700938 return 0;
939 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800940 }
941
Mike Lockwood74d7ff82009-10-11 23:04:18 -0400942 // remove TCP transport
943 if (!strncmp(service, "disconnect:", 11)) {
944 char buffer[4096];
945 memset(buffer, 0, sizeof(buffer));
946 char* serial = service + 11;
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -0400947 if (serial[0] == 0) {
948 // disconnect from all TCP devices
949 unregister_all_tcp_transports();
Mike Lockwood74d7ff82009-10-11 23:04:18 -0400950 } else {
Mike Lockwoodcbbe79a2010-05-24 10:44:35 -0400951 char hostbuf[100];
952 // assume port 5555 if no port is specified
953 if (!strchr(serial, ':')) {
954 snprintf(hostbuf, sizeof(hostbuf) - 1, "%s:5555", serial);
955 serial = hostbuf;
956 }
957 atransport *t = find_transport(serial);
958
959 if (t) {
960 unregister_transport(t);
961 } else {
962 snprintf(buffer, sizeof(buffer), "No such device %s", serial);
963 }
Mike Lockwood74d7ff82009-10-11 23:04:18 -0400964 }
965
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100966 send_msg_with_okay(reply_fd, buffer, strlen(buffer));
Mike Lockwood2f38b692009-08-24 15:58:40 -0700967 return 0;
968 }
969
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800970 // returns our value for ADB_SERVER_VERSION
971 if (!strcmp(service, "version")) {
972 char version[12];
973 snprintf(version, sizeof version, "%04x", ADB_SERVER_VERSION);
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100974 send_msg_with_okay(reply_fd, version, strlen(version));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800975 return 0;
976 }
977
978 if(!strncmp(service,"get-serialno",strlen("get-serialno"))) {
Dan Albertbac34742015-02-25 17:51:28 -0800979 const char *out = "unknown";
980 transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800981 if (transport && transport->serial) {
982 out = transport->serial;
983 }
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100984 send_msg_with_okay(reply_fd, out, strlen(out));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800985 return 0;
986 }
Scott Andersone109d262012-04-20 11:21:14 -0700987 if(!strncmp(service,"get-devpath",strlen("get-devpath"))) {
Dan Albertbac34742015-02-25 17:51:28 -0800988 const char *out = "unknown";
989 transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
Scott Andersone109d262012-04-20 11:21:14 -0700990 if (transport && transport->devpath) {
991 out = transport->devpath;
992 }
Snild Dolkow2264e7c2014-01-30 10:08:38 +0100993 send_msg_with_okay(reply_fd, out, strlen(out));
Scott Andersone109d262012-04-20 11:21:14 -0700994 return 0;
995 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -0800996 // indicates a new emulator instance has started
997 if (!strncmp(service,"emulator:",9)) {
998 int port = atoi(service+9);
999 local_connect(port);
1000 /* we don't even need to send a reply */
1001 return 0;
1002 }
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001003
1004 if(!strncmp(service,"get-state",strlen("get-state"))) {
1005 transport = acquire_one_transport(CS_ANY, ttype, serial, NULL);
Dan Albertbac34742015-02-25 17:51:28 -08001006 const char *state = connection_state_name(transport);
Snild Dolkow2264e7c2014-01-30 10:08:38 +01001007 send_msg_with_okay(reply_fd, state, strlen(state));
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001008 return 0;
1009 }
Simon Yedc22c3c2014-07-14 17:23:06 -07001010#endif // ADB_HOST
1011
1012 int ret = handle_forward_request(service, ttype, serial, reply_fd);
1013 if (ret >= 0)
1014 return ret - 1;
The Android Open Source Projectdd7bc332009-03-03 19:32:55 -08001015 return -1;
1016}