adb.c revision 899913f8168b54e00971c0e8d4ae16d06a4651fe
1/* 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 17#define TRACE_TAG TRACE_ADB 18 19#include <stdio.h> 20#include <stdlib.h> 21#include <ctype.h> 22#include <stdarg.h> 23#include <errno.h> 24#include <string.h> 25#include <time.h> 26#include <sys/time.h> 27 28#include "sysdeps.h" 29#include "adb.h" 30 31#if !ADB_HOST 32#include <private/android_filesystem_config.h> 33#include <linux/capability.h> 34#include <linux/prctl.h> 35#else 36#include "usb_vendors.h" 37#endif 38 39 40int HOST = 0; 41 42static const char *adb_device_banner = "device"; 43 44void fatal(const char *fmt, ...) 45{ 46 va_list ap; 47 va_start(ap, fmt); 48 fprintf(stderr, "error: "); 49 vfprintf(stderr, fmt, ap); 50 fprintf(stderr, "\n"); 51 va_end(ap); 52 exit(-1); 53} 54 55void fatal_errno(const char *fmt, ...) 56{ 57 va_list ap; 58 va_start(ap, fmt); 59 fprintf(stderr, "error: %s: ", strerror(errno)); 60 vfprintf(stderr, fmt, ap); 61 fprintf(stderr, "\n"); 62 va_end(ap); 63 exit(-1); 64} 65 66int adb_trace_mask; 67 68/* read a comma/space/colum/semi-column separated list of tags 69 * from the ADB_TRACE environment variable and build the trace 70 * mask from it. note that '1' and 'all' are special cases to 71 * enable all tracing 72 */ 73void adb_trace_init(void) 74{ 75 const char* p = getenv("ADB_TRACE"); 76 const char* q; 77 78 static const struct { 79 const char* tag; 80 int flag; 81 } tags[] = { 82 { "1", 0 }, 83 { "all", 0 }, 84 { "adb", TRACE_ADB }, 85 { "sockets", TRACE_SOCKETS }, 86 { "packets", TRACE_PACKETS }, 87 { "rwx", TRACE_RWX }, 88 { "usb", TRACE_USB }, 89 { "sync", TRACE_SYNC }, 90 { "sysdeps", TRACE_SYSDEPS }, 91 { "transport", TRACE_TRANSPORT }, 92 { "jdwp", TRACE_JDWP }, 93 { NULL, 0 } 94 }; 95 96 if (p == NULL) 97 return; 98 99 /* use a comma/column/semi-colum/space separated list */ 100 while (*p) { 101 int len, tagn; 102 103 q = strpbrk(p, " ,:;"); 104 if (q == NULL) { 105 q = p + strlen(p); 106 } 107 len = q - p; 108 109 for (tagn = 0; tags[tagn].tag != NULL; tagn++) 110 { 111 int taglen = strlen(tags[tagn].tag); 112 113 if (len == taglen && !memcmp(tags[tagn].tag, p, len) ) 114 { 115 int flag = tags[tagn].flag; 116 if (flag == 0) { 117 adb_trace_mask = ~0; 118 return; 119 } 120 adb_trace_mask |= (1 << flag); 121 break; 122 } 123 } 124 p = q; 125 if (*p) 126 p++; 127 } 128} 129 130 131apacket *get_apacket(void) 132{ 133 apacket *p = malloc(sizeof(apacket)); 134 if(p == 0) fatal("failed to allocate an apacket"); 135 memset(p, 0, sizeof(apacket) - MAX_PAYLOAD); 136 return p; 137} 138 139void put_apacket(apacket *p) 140{ 141 free(p); 142} 143 144void handle_online(void) 145{ 146 D("adb: online\n"); 147} 148 149void handle_offline(atransport *t) 150{ 151 D("adb: offline\n"); 152 //Close the associated usb 153 run_transport_disconnects(t); 154} 155 156#if TRACE_PACKETS 157#define DUMPMAX 32 158void print_packet(const char *label, apacket *p) 159{ 160 char *tag; 161 char *x; 162 unsigned count; 163 164 switch(p->msg.command){ 165 case A_SYNC: tag = "SYNC"; break; 166 case A_CNXN: tag = "CNXN" ; break; 167 case A_OPEN: tag = "OPEN"; break; 168 case A_OKAY: tag = "OKAY"; break; 169 case A_CLSE: tag = "CLSE"; break; 170 case A_WRTE: tag = "WRTE"; break; 171 default: tag = "????"; break; 172 } 173 174 fprintf(stderr, "%s: %s %08x %08x %04x \"", 175 label, tag, p->msg.arg0, p->msg.arg1, p->msg.data_length); 176 count = p->msg.data_length; 177 x = (char*) p->data; 178 if(count > DUMPMAX) { 179 count = DUMPMAX; 180 tag = "\n"; 181 } else { 182 tag = "\"\n"; 183 } 184 while(count-- > 0){ 185 if((*x >= ' ') && (*x < 127)) { 186 fputc(*x, stderr); 187 } else { 188 fputc('.', stderr); 189 } 190 x++; 191 } 192 fprintf(stderr, tag); 193} 194#endif 195 196static void send_ready(unsigned local, unsigned remote, atransport *t) 197{ 198 D("Calling send_ready \n"); 199 apacket *p = get_apacket(); 200 p->msg.command = A_OKAY; 201 p->msg.arg0 = local; 202 p->msg.arg1 = remote; 203 send_packet(p, t); 204} 205 206static void send_close(unsigned local, unsigned remote, atransport *t) 207{ 208 D("Calling send_close \n"); 209 apacket *p = get_apacket(); 210 p->msg.command = A_CLSE; 211 p->msg.arg0 = local; 212 p->msg.arg1 = remote; 213 send_packet(p, t); 214} 215 216static void send_connect(atransport *t) 217{ 218 D("Calling send_connect \n"); 219 apacket *cp = get_apacket(); 220 cp->msg.command = A_CNXN; 221 cp->msg.arg0 = A_VERSION; 222 cp->msg.arg1 = MAX_PAYLOAD; 223 snprintf((char*) cp->data, sizeof cp->data, "%s::", 224 HOST ? "host" : adb_device_banner); 225 cp->msg.data_length = strlen((char*) cp->data) + 1; 226 send_packet(cp, t); 227#if ADB_HOST 228 /* XXX why sleep here? */ 229 // allow the device some time to respond to the connect message 230 adb_sleep_ms(1000); 231#endif 232} 233 234static char *connection_state_name(atransport *t) 235{ 236 if (t == NULL) { 237 return "unknown"; 238 } 239 240 switch(t->connection_state) { 241 case CS_BOOTLOADER: 242 return "bootloader"; 243 case CS_DEVICE: 244 return "device"; 245 case CS_OFFLINE: 246 return "offline"; 247 default: 248 return "unknown"; 249 } 250} 251 252void parse_banner(char *banner, atransport *t) 253{ 254 char *type, *product, *end; 255 256 D("parse_banner: %s\n", banner); 257 type = banner; 258 product = strchr(type, ':'); 259 if(product) { 260 *product++ = 0; 261 } else { 262 product = ""; 263 } 264 265 /* remove trailing ':' */ 266 end = strchr(product, ':'); 267 if(end) *end = 0; 268 269 /* save product name in device structure */ 270 if (t->product == NULL) { 271 t->product = strdup(product); 272 } else if (strcmp(product, t->product) != 0) { 273 free(t->product); 274 t->product = strdup(product); 275 } 276 277 if(!strcmp(type, "bootloader")){ 278 D("setting connection_state to CS_BOOTLOADER\n"); 279 t->connection_state = CS_BOOTLOADER; 280 update_transports(); 281 return; 282 } 283 284 if(!strcmp(type, "device")) { 285 D("setting connection_state to CS_DEVICE\n"); 286 t->connection_state = CS_DEVICE; 287 update_transports(); 288 return; 289 } 290 291 if(!strcmp(type, "recovery")) { 292 D("setting connection_state to CS_RECOVERY\n"); 293 t->connection_state = CS_RECOVERY; 294 update_transports(); 295 return; 296 } 297 298 t->connection_state = CS_HOST; 299} 300 301void handle_packet(apacket *p, atransport *t) 302{ 303 asocket *s; 304 305 D("handle_packet() %c%c%c%c\n", ((char*) (&(p->msg.command)))[0], 306 ((char*) (&(p->msg.command)))[1], 307 ((char*) (&(p->msg.command)))[2], 308 ((char*) (&(p->msg.command)))[3]); 309 print_packet("recv", p); 310 311 switch(p->msg.command){ 312 case A_SYNC: 313 if(p->msg.arg0){ 314 send_packet(p, t); 315 if(HOST) send_connect(t); 316 } else { 317 t->connection_state = CS_OFFLINE; 318 handle_offline(t); 319 send_packet(p, t); 320 } 321 return; 322 323 case A_CNXN: /* CONNECT(version, maxdata, "system-id-string") */ 324 /* XXX verify version, etc */ 325 if(t->connection_state != CS_OFFLINE) { 326 t->connection_state = CS_OFFLINE; 327 handle_offline(t); 328 } 329 parse_banner((char*) p->data, t); 330 handle_online(); 331 if(!HOST) send_connect(t); 332 break; 333 334 case A_OPEN: /* OPEN(local-id, 0, "destination") */ 335 if(t->connection_state != CS_OFFLINE) { 336 char *name = (char*) p->data; 337 name[p->msg.data_length > 0 ? p->msg.data_length - 1 : 0] = 0; 338 s = create_local_service_socket(name); 339 if(s == 0) { 340 send_close(0, p->msg.arg0, t); 341 } else { 342 s->peer = create_remote_socket(p->msg.arg0, t); 343 s->peer->peer = s; 344 send_ready(s->id, s->peer->id, t); 345 s->ready(s); 346 } 347 } 348 break; 349 350 case A_OKAY: /* READY(local-id, remote-id, "") */ 351 if(t->connection_state != CS_OFFLINE) { 352 if((s = find_local_socket(p->msg.arg1))) { 353 if(s->peer == 0) { 354 s->peer = create_remote_socket(p->msg.arg0, t); 355 s->peer->peer = s; 356 } 357 s->ready(s); 358 } 359 } 360 break; 361 362 case A_CLSE: /* CLOSE(local-id, remote-id, "") */ 363 if(t->connection_state != CS_OFFLINE) { 364 if((s = find_local_socket(p->msg.arg1))) { 365 s->close(s); 366 } 367 } 368 break; 369 370 case A_WRTE: 371 if(t->connection_state != CS_OFFLINE) { 372 if((s = find_local_socket(p->msg.arg1))) { 373 unsigned rid = p->msg.arg0; 374 p->len = p->msg.data_length; 375 376 if(s->enqueue(s, p) == 0) { 377 D("Enqueue the socket\n"); 378 send_ready(s->id, rid, t); 379 } 380 return; 381 } 382 } 383 break; 384 385 default: 386 printf("handle_packet: what is %08x?!\n", p->msg.command); 387 } 388 389 put_apacket(p); 390} 391 392alistener listener_list = { 393 .next = &listener_list, 394 .prev = &listener_list, 395}; 396 397static void ss_listener_event_func(int _fd, unsigned ev, void *_l) 398{ 399 asocket *s; 400 401 if(ev & FDE_READ) { 402 struct sockaddr addr; 403 socklen_t alen; 404 int fd; 405 406 alen = sizeof(addr); 407 fd = adb_socket_accept(_fd, &addr, &alen); 408 if(fd < 0) return; 409 410 adb_socket_setbufsize(fd, CHUNK_SIZE); 411 412 s = create_local_socket(fd); 413 if(s) { 414 connect_to_smartsocket(s); 415 return; 416 } 417 418 adb_close(fd); 419 } 420} 421 422static void listener_event_func(int _fd, unsigned ev, void *_l) 423{ 424 alistener *l = _l; 425 asocket *s; 426 427 if(ev & FDE_READ) { 428 struct sockaddr addr; 429 socklen_t alen; 430 int fd; 431 432 alen = sizeof(addr); 433 fd = adb_socket_accept(_fd, &addr, &alen); 434 if(fd < 0) return; 435 436 s = create_local_socket(fd); 437 if(s) { 438 s->transport = l->transport; 439 connect_to_remote(s, l->connect_to); 440 return; 441 } 442 443 adb_close(fd); 444 } 445} 446 447static void free_listener(alistener* l) 448{ 449 if (l->next) { 450 l->next->prev = l->prev; 451 l->prev->next = l->next; 452 l->next = l->prev = l; 453 } 454 455 // closes the corresponding fd 456 fdevent_remove(&l->fde); 457 458 if (l->local_name) 459 free((char*)l->local_name); 460 461 if (l->connect_to) 462 free((char*)l->connect_to); 463 464 if (l->transport) { 465 remove_transport_disconnect(l->transport, &l->disconnect); 466 } 467 free(l); 468} 469 470static void listener_disconnect(void* _l, atransport* t) 471{ 472 alistener* l = _l; 473 474 free_listener(l); 475} 476 477int local_name_to_fd(const char *name) 478{ 479 int port; 480 481 if(!strncmp("tcp:", name, 4)){ 482 int ret; 483 port = atoi(name + 4); 484 ret = socket_loopback_server(port, SOCK_STREAM); 485 return ret; 486 } 487#ifndef HAVE_WIN32_IPC /* no Unix-domain sockets on Win32 */ 488 // It's non-sensical to support the "reserved" space on the adb host side 489 if(!strncmp(name, "local:", 6)) { 490 return socket_local_server(name + 6, 491 ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM); 492 } else if(!strncmp(name, "localabstract:", 14)) { 493 return socket_local_server(name + 14, 494 ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM); 495 } else if(!strncmp(name, "localfilesystem:", 16)) { 496 return socket_local_server(name + 16, 497 ANDROID_SOCKET_NAMESPACE_FILESYSTEM, SOCK_STREAM); 498 } 499 500#endif 501 printf("unknown local portname '%s'\n", name); 502 return -1; 503} 504 505static int remove_listener(const char *local_name, const char *connect_to, atransport* transport) 506{ 507 alistener *l; 508 509 for (l = listener_list.next; l != &listener_list; l = l->next) { 510 if (!strcmp(local_name, l->local_name) && 511 !strcmp(connect_to, l->connect_to) && 512 l->transport && l->transport == transport) { 513 514 listener_disconnect(l, transport); 515 return 0; 516 } 517 } 518 519 return -1; 520} 521 522static int install_listener(const char *local_name, const char *connect_to, atransport* transport) 523{ 524 alistener *l; 525 526 //printf("install_listener('%s','%s')\n", local_name, connect_to); 527 528 for(l = listener_list.next; l != &listener_list; l = l->next){ 529 if(strcmp(local_name, l->local_name) == 0) { 530 char *cto; 531 532 /* can't repurpose a smartsocket */ 533 if(l->connect_to[0] == '*') { 534 return -1; 535 } 536 537 cto = strdup(connect_to); 538 if(cto == 0) { 539 return -1; 540 } 541 542 //printf("rebinding '%s' to '%s'\n", local_name, connect_to); 543 free((void*) l->connect_to); 544 l->connect_to = cto; 545 if (l->transport != transport) { 546 remove_transport_disconnect(l->transport, &l->disconnect); 547 l->transport = transport; 548 add_transport_disconnect(l->transport, &l->disconnect); 549 } 550 return 0; 551 } 552 } 553 554 if((l = calloc(1, sizeof(alistener))) == 0) goto nomem; 555 if((l->local_name = strdup(local_name)) == 0) goto nomem; 556 if((l->connect_to = strdup(connect_to)) == 0) goto nomem; 557 558 559 l->fd = local_name_to_fd(local_name); 560 if(l->fd < 0) { 561 free((void*) l->local_name); 562 free((void*) l->connect_to); 563 free(l); 564 printf("cannot bind '%s'\n", local_name); 565 return -2; 566 } 567 568 close_on_exec(l->fd); 569 if(!strcmp(l->connect_to, "*smartsocket*")) { 570 fdevent_install(&l->fde, l->fd, ss_listener_event_func, l); 571 } else { 572 fdevent_install(&l->fde, l->fd, listener_event_func, l); 573 } 574 fdevent_set(&l->fde, FDE_READ); 575 576 l->next = &listener_list; 577 l->prev = listener_list.prev; 578 l->next->prev = l; 579 l->prev->next = l; 580 l->transport = transport; 581 582 if (transport) { 583 l->disconnect.opaque = l; 584 l->disconnect.func = listener_disconnect; 585 add_transport_disconnect(transport, &l->disconnect); 586 } 587 return 0; 588 589nomem: 590 fatal("cannot allocate listener"); 591 return 0; 592} 593 594#ifdef HAVE_FORKEXEC 595static void sigchld_handler(int n) 596{ 597 int status; 598 while(waitpid(-1, &status, WNOHANG) > 0) ; 599} 600#endif 601 602#ifdef HAVE_WIN32_PROC 603static BOOL WINAPI ctrlc_handler(DWORD type) 604{ 605 exit(STATUS_CONTROL_C_EXIT); 606 return TRUE; 607} 608#endif 609 610static void adb_cleanup(void) 611{ 612 usb_cleanup(); 613} 614 615void start_logging(void) 616{ 617#ifdef HAVE_WIN32_PROC 618 char temp[ MAX_PATH ]; 619 FILE* fnul; 620 FILE* flog; 621 622 GetTempPath( sizeof(temp) - 8, temp ); 623 strcat( temp, "adb.log" ); 624 625 /* Win32 specific redirections */ 626 fnul = fopen( "NUL", "rt" ); 627 if (fnul != NULL) 628 stdin[0] = fnul[0]; 629 630 flog = fopen( temp, "at" ); 631 if (flog == NULL) 632 flog = fnul; 633 634 setvbuf( flog, NULL, _IONBF, 0 ); 635 636 stdout[0] = flog[0]; 637 stderr[0] = flog[0]; 638 fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid()); 639#else 640 int fd; 641 642 fd = unix_open("/dev/null", O_RDONLY); 643 dup2(fd, 0); 644 645 fd = unix_open("/tmp/adb.log", O_WRONLY | O_CREAT | O_APPEND, 0640); 646 if(fd < 0) { 647 fd = unix_open("/dev/null", O_WRONLY); 648 } 649 dup2(fd, 1); 650 dup2(fd, 2); 651 fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid()); 652#endif 653} 654 655#if !ADB_HOST 656void start_device_log(void) 657{ 658 int fd; 659 char path[PATH_MAX]; 660 struct tm now; 661 time_t t; 662 char value[PROPERTY_VALUE_MAX]; 663 664 // read the trace mask from persistent property persist.adb.trace_mask 665 // give up if the property is not set or cannot be parsed 666 property_get("persist.adb.trace_mask", value, ""); 667 if (sscanf(value, "%x", &adb_trace_mask) != 1) 668 return; 669 670 adb_mkdir("/data/adb", 0775); 671 tzset(); 672 time(&t); 673 localtime_r(&t, &now); 674 strftime(path, sizeof(path), 675 "/data/adb/adb-%Y-%m-%d-%H-%M-%S.txt", 676 &now); 677 fd = unix_open(path, O_WRONLY | O_CREAT | O_TRUNC, 0640); 678 if (fd < 0) 679 return; 680 681 // redirect stdout and stderr to the log file 682 dup2(fd, 1); 683 dup2(fd, 2); 684 fprintf(stderr,"--- adb starting (pid %d) ---\n", getpid()); 685 686 fd = unix_open("/dev/null", O_RDONLY); 687 dup2(fd, 0); 688} 689#endif 690 691#if ADB_HOST 692int launch_server(int server_port) 693{ 694#ifdef HAVE_WIN32_PROC 695 /* we need to start the server in the background */ 696 /* we create a PIPE that will be used to wait for the server's "OK" */ 697 /* message since the pipe handles must be inheritable, we use a */ 698 /* security attribute */ 699 HANDLE pipe_read, pipe_write; 700 SECURITY_ATTRIBUTES sa; 701 STARTUPINFO startup; 702 PROCESS_INFORMATION pinfo; 703 char program_path[ MAX_PATH ]; 704 int ret; 705 706 sa.nLength = sizeof(sa); 707 sa.lpSecurityDescriptor = NULL; 708 sa.bInheritHandle = TRUE; 709 710 /* create pipe, and ensure its read handle isn't inheritable */ 711 ret = CreatePipe( &pipe_read, &pipe_write, &sa, 0 ); 712 if (!ret) { 713 fprintf(stderr, "CreatePipe() failure, error %ld\n", GetLastError() ); 714 return -1; 715 } 716 717 SetHandleInformation( pipe_read, HANDLE_FLAG_INHERIT, 0 ); 718 719 ZeroMemory( &startup, sizeof(startup) ); 720 startup.cb = sizeof(startup); 721 startup.hStdInput = GetStdHandle( STD_INPUT_HANDLE ); 722 startup.hStdOutput = pipe_write; 723 startup.hStdError = GetStdHandle( STD_ERROR_HANDLE ); 724 startup.dwFlags = STARTF_USESTDHANDLES; 725 726 ZeroMemory( &pinfo, sizeof(pinfo) ); 727 728 /* get path of current program */ 729 GetModuleFileName( NULL, program_path, sizeof(program_path) ); 730 731 ret = CreateProcess( 732 program_path, /* program path */ 733 "adb fork-server server", 734 /* the fork-server argument will set the 735 debug = 2 in the child */ 736 NULL, /* process handle is not inheritable */ 737 NULL, /* thread handle is not inheritable */ 738 TRUE, /* yes, inherit some handles */ 739 DETACHED_PROCESS, /* the new process doesn't have a console */ 740 NULL, /* use parent's environment block */ 741 NULL, /* use parent's starting directory */ 742 &startup, /* startup info, i.e. std handles */ 743 &pinfo ); 744 745 CloseHandle( pipe_write ); 746 747 if (!ret) { 748 fprintf(stderr, "CreateProcess failure, error %ld\n", GetLastError() ); 749 CloseHandle( pipe_read ); 750 return -1; 751 } 752 753 CloseHandle( pinfo.hProcess ); 754 CloseHandle( pinfo.hThread ); 755 756 /* wait for the "OK\n" message */ 757 { 758 char temp[3]; 759 DWORD count; 760 761 ret = ReadFile( pipe_read, temp, 3, &count, NULL ); 762 CloseHandle( pipe_read ); 763 if ( !ret ) { 764 fprintf(stderr, "could not read ok from ADB Server, error = %ld\n", GetLastError() ); 765 return -1; 766 } 767 if (count != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') { 768 fprintf(stderr, "ADB server didn't ACK\n" ); 769 return -1; 770 } 771 } 772#elif defined(HAVE_FORKEXEC) 773 char path[PATH_MAX]; 774 int fd[2]; 775 776 // set up a pipe so the child can tell us when it is ready. 777 // fd[0] will be parent's end, and fd[1] will get mapped to stderr in the child. 778 if (pipe(fd)) { 779 fprintf(stderr, "pipe failed in launch_server, errno: %d\n", errno); 780 return -1; 781 } 782 get_my_path(path, PATH_MAX); 783 pid_t pid = fork(); 784 if(pid < 0) return -1; 785 786 if (pid == 0) { 787 // child side of the fork 788 789 // redirect stderr to the pipe 790 // we use stderr instead of stdout due to stdout's buffering behavior. 791 adb_close(fd[0]); 792 dup2(fd[1], STDERR_FILENO); 793 adb_close(fd[1]); 794 795 // child process 796 int result = execl(path, "adb", "fork-server", "server", NULL); 797 // this should not return 798 fprintf(stderr, "OOPS! execl returned %d, errno: %d\n", result, errno); 799 } else { 800 // parent side of the fork 801 802 char temp[3]; 803 804 temp[0] = 'A'; temp[1] = 'B'; temp[2] = 'C'; 805 // wait for the "OK\n" message 806 adb_close(fd[1]); 807 int ret = adb_read(fd[0], temp, 3); 808 adb_close(fd[0]); 809 if (ret < 0) { 810 fprintf(stderr, "could not read ok from ADB Server, errno = %d\n", errno); 811 return -1; 812 } 813 if (ret != 3 || temp[0] != 'O' || temp[1] != 'K' || temp[2] != '\n') { 814 fprintf(stderr, "ADB server didn't ACK\n" ); 815 return -1; 816 } 817 818 setsid(); 819 } 820#else 821#error "cannot implement background server start on this platform" 822#endif 823 return 0; 824} 825#endif 826 827/* Constructs a local name of form tcp:port. 828 * target_str points to the target string, it's content will be overwritten. 829 * target_size is the capacity of the target string. 830 * server_port is the port number to use for the local name. 831 */ 832void build_local_name(char* target_str, size_t target_size, int server_port) 833{ 834 snprintf(target_str, target_size, "tcp:%d", server_port); 835} 836 837int adb_main(int is_daemon, int server_port) 838{ 839#if !ADB_HOST 840 int secure = 0; 841 int port; 842 char value[PROPERTY_VALUE_MAX]; 843#endif 844 845 atexit(adb_cleanup); 846#ifdef HAVE_WIN32_PROC 847 SetConsoleCtrlHandler( ctrlc_handler, TRUE ); 848#elif defined(HAVE_FORKEXEC) 849 signal(SIGCHLD, sigchld_handler); 850 signal(SIGPIPE, SIG_IGN); 851#endif 852 853 init_transport_registration(); 854 855 856#if ADB_HOST 857 HOST = 1; 858 usb_vendors_init(); 859 usb_init(); 860 local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT); 861 862 char local_name[30]; 863 build_local_name(local_name, sizeof(local_name), server_port); 864 if(install_listener(local_name, "*smartsocket*", NULL)) { 865 exit(1); 866 } 867#else 868 /* run adbd in secure mode if ro.secure is set and 869 ** we are not in the emulator 870 */ 871 property_get("ro.kernel.qemu", value, ""); 872 if (strcmp(value, "1") != 0) { 873 property_get("ro.secure", value, ""); 874 if (strcmp(value, "1") == 0) { 875 // don't run as root if ro.secure is set... 876 secure = 1; 877 878 // ... except we allow running as root in userdebug builds if the 879 // service.adb.root property has been set by the "adb root" command 880 property_get("ro.debuggable", value, ""); 881 if (strcmp(value, "1") == 0) { 882 property_get("service.adb.root", value, ""); 883 if (strcmp(value, "1") == 0) { 884 secure = 0; 885 } 886 } 887 } 888 } 889 890 /* don't listen on a port (default 5037) if running in secure mode */ 891 /* don't run as root if we are running in secure mode */ 892 if (secure) { 893 struct __user_cap_header_struct header; 894 struct __user_cap_data_struct cap; 895 896 prctl(PR_SET_KEEPCAPS, 1, 0, 0, 0); 897 898 /* add extra groups: 899 ** AID_ADB to access the USB driver 900 ** AID_LOG to read system logs (adb logcat) 901 ** AID_INPUT to diagnose input issues (getevent) 902 ** AID_INET to diagnose network issues (netcfg, ping) 903 ** AID_GRAPHICS to access the frame buffer 904 ** AID_NET_BT and AID_NET_BT_ADMIN to diagnose bluetooth (hcidump) 905 ** AID_SDCARD_RW to allow writing to the SD card 906 ** AID_MOUNT to allow unmounting the SD card before rebooting 907 */ 908 gid_t groups[] = { AID_ADB, AID_LOG, AID_INPUT, AID_INET, AID_GRAPHICS, 909 AID_NET_BT, AID_NET_BT_ADMIN, AID_SDCARD_RW, AID_MOUNT }; 910 setgroups(sizeof(groups)/sizeof(groups[0]), groups); 911 912 /* then switch user and group to "shell" */ 913 setgid(AID_SHELL); 914 setuid(AID_SHELL); 915 916 /* set CAP_SYS_BOOT capability, so "adb reboot" will succeed */ 917 header.version = _LINUX_CAPABILITY_VERSION; 918 header.pid = 0; 919 cap.effective = cap.permitted = (1 << CAP_SYS_BOOT); 920 cap.inheritable = 0; 921 capset(&header, &cap); 922 923 D("Local port disabled\n"); 924 } else { 925 char local_name[30]; 926 build_local_name(local_name, sizeof(local_name), server_port); 927 if(install_listener(local_name, "*smartsocket*", NULL)) { 928 exit(1); 929 } 930 } 931 932 /* for the device, start the usb transport if the 933 ** android usb device exists and the "service.adb.tcp.port" and 934 ** "persist.adb.tcp.port" properties are not set. 935 ** Otherwise start the network transport. 936 */ 937 property_get("service.adb.tcp.port", value, ""); 938 if (!value[0]) 939 property_get("persist.adb.tcp.port", value, ""); 940 if (sscanf(value, "%d", &port) == 1 && port > 0) { 941 // listen on TCP port specified by service.adb.tcp.port property 942 local_init(port); 943 } else if (access("/dev/android_adb", F_OK) == 0) { 944 // listen on USB 945 usb_init(); 946 } else { 947 // listen on default port 948 local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT); 949 } 950 init_jdwp(); 951#endif 952 953 if (is_daemon) 954 { 955 // inform our parent that we are up and running. 956#ifdef HAVE_WIN32_PROC 957 DWORD count; 958 WriteFile( GetStdHandle( STD_OUTPUT_HANDLE ), "OK\n", 3, &count, NULL ); 959#elif defined(HAVE_FORKEXEC) 960 fprintf(stderr, "OK\n"); 961#endif 962 start_logging(); 963 } 964 965 fdevent_loop(); 966 967 usb_cleanup(); 968 969 return 0; 970} 971 972#if ADB_HOST 973void connect_device(char* host, char* buffer, int buffer_size) 974{ 975 int port, fd; 976 char* portstr = strchr(host, ':'); 977 char hostbuf[100]; 978 char serial[100]; 979 980 strncpy(hostbuf, host, sizeof(hostbuf) - 1); 981 if (portstr) { 982 if (portstr - host >= sizeof(hostbuf)) { 983 snprintf(buffer, buffer_size, "bad host name %s", host); 984 return; 985 } 986 // zero terminate the host at the point we found the colon 987 hostbuf[portstr - host] = 0; 988 if (sscanf(portstr + 1, "%d", &port) == 0) { 989 snprintf(buffer, buffer_size, "bad port number %s", portstr); 990 return; 991 } 992 } else { 993 port = DEFAULT_ADB_LOCAL_TRANSPORT_PORT; 994 } 995 996 snprintf(serial, sizeof(serial), "%s:%d", hostbuf, port); 997 if (find_transport(serial)) { 998 snprintf(buffer, buffer_size, "already connected to %s", serial); 999 return; 1000 } 1001 1002 fd = socket_network_client(hostbuf, port, SOCK_STREAM); 1003 if (fd < 0) { 1004 snprintf(buffer, buffer_size, "unable to connect to %s:%d", host, port); 1005 return; 1006 } 1007 1008 D("client: connected on remote on fd %d\n", fd); 1009 close_on_exec(fd); 1010 disable_tcp_nagle(fd); 1011 register_socket_transport(fd, serial, port, 0); 1012 snprintf(buffer, buffer_size, "connected to %s", serial); 1013} 1014 1015void connect_emulator(char* port_spec, char* buffer, int buffer_size) 1016{ 1017 char* port_separator = strchr(port_spec, ','); 1018 if (!port_separator) { 1019 snprintf(buffer, buffer_size, 1020 "unable to parse '%s' as <console port>,<adb port>", 1021 port_spec); 1022 return; 1023 } 1024 1025 // Zero-terminate console port and make port_separator point to 2nd port. 1026 *port_separator++ = 0; 1027 int console_port = strtol(port_spec, NULL, 0); 1028 int adb_port = strtol(port_separator, NULL, 0); 1029 if (!(console_port > 0 && adb_port > 0)) { 1030 *(port_separator - 1) = ','; 1031 snprintf(buffer, buffer_size, 1032 "Invalid port numbers: Expected positive numbers, got '%s'", 1033 port_spec); 1034 return; 1035 } 1036 1037 /* Check if the emulator is already known. 1038 * Note: There's a small but harmless race condition here: An emulator not 1039 * present just yet could be registered by another invocation right 1040 * after doing this check here. However, local_connect protects 1041 * against double-registration too. From here, a better error message 1042 * can be produced. In the case of the race condition, the very specific 1043 * error message won't be shown, but the data doesn't get corrupted. */ 1044 atransport* known_emulator = find_emulator_transport_by_adb_port(adb_port); 1045 if (known_emulator != NULL) { 1046 snprintf(buffer, buffer_size, 1047 "Emulator on port %d already registered.", adb_port); 1048 return; 1049 } 1050 1051 /* Check if more emulators can be registered. Similar unproblematic 1052 * race condition as above. */ 1053 int candidate_slot = get_available_local_transport_index(); 1054 if (candidate_slot < 0) { 1055 snprintf(buffer, buffer_size, "Cannot accept more emulators."); 1056 return; 1057 } 1058 1059 /* Preconditions met, try to connect to the emulator. */ 1060 if (!local_connect_arbitrary_ports(console_port, adb_port)) { 1061 snprintf(buffer, buffer_size, 1062 "Connected to emulator on ports %d,%d", console_port, adb_port); 1063 } else { 1064 snprintf(buffer, buffer_size, 1065 "Could not connect to emulator on ports %d,%d", 1066 console_port, adb_port); 1067 } 1068} 1069#endif 1070 1071int handle_host_request(char *service, transport_type ttype, char* serial, int reply_fd, asocket *s) 1072{ 1073 atransport *transport = NULL; 1074 char buf[4096]; 1075 1076 if(!strcmp(service, "kill")) { 1077 fprintf(stderr,"adb server killed by remote request\n"); 1078 fflush(stdout); 1079 adb_write(reply_fd, "OKAY", 4); 1080 usb_cleanup(); 1081 exit(0); 1082 } 1083 1084#if ADB_HOST 1085 // "transport:" is used for switching transport with a specified serial number 1086 // "transport-usb:" is used for switching transport to the only USB transport 1087 // "transport-local:" is used for switching transport to the only local transport 1088 // "transport-any:" is used for switching transport to the only transport 1089 if (!strncmp(service, "transport", strlen("transport"))) { 1090 char* error_string = "unknown failure"; 1091 transport_type type = kTransportAny; 1092 1093 if (!strncmp(service, "transport-usb", strlen("transport-usb"))) { 1094 type = kTransportUsb; 1095 } else if (!strncmp(service, "transport-local", strlen("transport-local"))) { 1096 type = kTransportLocal; 1097 } else if (!strncmp(service, "transport-any", strlen("transport-any"))) { 1098 type = kTransportAny; 1099 } else if (!strncmp(service, "transport:", strlen("transport:"))) { 1100 service += strlen("transport:"); 1101 serial = strdup(service); 1102 } 1103 1104 transport = acquire_one_transport(CS_ANY, type, serial, &error_string); 1105 1106 if (transport) { 1107 s->transport = transport; 1108 adb_write(reply_fd, "OKAY", 4); 1109 } else { 1110 sendfailmsg(reply_fd, error_string); 1111 } 1112 return 1; 1113 } 1114 1115 // return a list of all connected devices 1116 if (!strcmp(service, "devices")) { 1117 char buffer[4096]; 1118 memset(buf, 0, sizeof(buf)); 1119 memset(buffer, 0, sizeof(buffer)); 1120 D("Getting device list \n"); 1121 list_transports(buffer, sizeof(buffer)); 1122 snprintf(buf, sizeof(buf), "OKAY%04x%s",(unsigned)strlen(buffer),buffer); 1123 D("Wrote device list \n"); 1124 writex(reply_fd, buf, strlen(buf)); 1125 return 0; 1126 } 1127 1128 // add a new TCP transport, device or emulator 1129 if (!strncmp(service, "connect:", 8)) { 1130 char buffer[4096]; 1131 char* host = service + 8; 1132 if (!strncmp(host, "emu:", 4)) { 1133 connect_emulator(host + 4, buffer, sizeof(buffer)); 1134 } else { 1135 connect_device(host, buffer, sizeof(buffer)); 1136 } 1137 // Send response for emulator and device 1138 snprintf(buf, sizeof(buf), "OKAY%04x%s",(unsigned)strlen(buffer), buffer); 1139 writex(reply_fd, buf, strlen(buf)); 1140 return 0; 1141 } 1142 1143 // remove TCP transport 1144 if (!strncmp(service, "disconnect:", 11)) { 1145 char buffer[4096]; 1146 memset(buffer, 0, sizeof(buffer)); 1147 char* serial = service + 11; 1148 if (serial[0] == 0) { 1149 // disconnect from all TCP devices 1150 unregister_all_tcp_transports(); 1151 } else { 1152 char hostbuf[100]; 1153 // assume port 5555 if no port is specified 1154 if (!strchr(serial, ':')) { 1155 snprintf(hostbuf, sizeof(hostbuf) - 1, "%s:5555", serial); 1156 serial = hostbuf; 1157 } 1158 atransport *t = find_transport(serial); 1159 1160 if (t) { 1161 unregister_transport(t); 1162 } else { 1163 snprintf(buffer, sizeof(buffer), "No such device %s", serial); 1164 } 1165 } 1166 1167 snprintf(buf, sizeof(buf), "OKAY%04x%s",(unsigned)strlen(buffer), buffer); 1168 writex(reply_fd, buf, strlen(buf)); 1169 return 0; 1170 } 1171 1172 // returns our value for ADB_SERVER_VERSION 1173 if (!strcmp(service, "version")) { 1174 char version[12]; 1175 snprintf(version, sizeof version, "%04x", ADB_SERVER_VERSION); 1176 snprintf(buf, sizeof buf, "OKAY%04x%s", (unsigned)strlen(version), version); 1177 writex(reply_fd, buf, strlen(buf)); 1178 return 0; 1179 } 1180 1181 if(!strncmp(service,"get-serialno",strlen("get-serialno"))) { 1182 char *out = "unknown"; 1183 transport = acquire_one_transport(CS_ANY, ttype, serial, NULL); 1184 if (transport && transport->serial) { 1185 out = transport->serial; 1186 } 1187 snprintf(buf, sizeof buf, "OKAY%04x%s",(unsigned)strlen(out),out); 1188 writex(reply_fd, buf, strlen(buf)); 1189 return 0; 1190 } 1191 // indicates a new emulator instance has started 1192 if (!strncmp(service,"emulator:",9)) { 1193 int port = atoi(service+9); 1194 local_connect(port); 1195 /* we don't even need to send a reply */ 1196 return 0; 1197 } 1198#endif // ADB_HOST 1199 1200 if(!strncmp(service,"forward:",8) || !strncmp(service,"killforward:",12)) { 1201 char *local, *remote, *err; 1202 int r; 1203 atransport *transport; 1204 1205 int createForward = strncmp(service,"kill",4); 1206 1207 local = service + (createForward ? 8 : 12); 1208 remote = strchr(local,';'); 1209 if(remote == 0) { 1210 sendfailmsg(reply_fd, "malformed forward spec"); 1211 return 0; 1212 } 1213 1214 *remote++ = 0; 1215 if((local[0] == 0) || (remote[0] == 0) || (remote[0] == '*')){ 1216 sendfailmsg(reply_fd, "malformed forward spec"); 1217 return 0; 1218 } 1219 1220 transport = acquire_one_transport(CS_ANY, ttype, serial, &err); 1221 if (!transport) { 1222 sendfailmsg(reply_fd, err); 1223 return 0; 1224 } 1225 1226 if (createForward) { 1227 r = install_listener(local, remote, transport); 1228 } else { 1229 r = remove_listener(local, remote, transport); 1230 } 1231 if(r == 0) { 1232 /* 1st OKAY is connect, 2nd OKAY is status */ 1233 writex(reply_fd, "OKAYOKAY", 8); 1234 return 0; 1235 } 1236 1237 if (createForward) { 1238 sendfailmsg(reply_fd, (r == -1) ? "cannot rebind smartsocket" : "cannot bind socket"); 1239 } else { 1240 sendfailmsg(reply_fd, "cannot remove listener"); 1241 } 1242 return 0; 1243 } 1244 1245 if(!strncmp(service,"get-state",strlen("get-state"))) { 1246 transport = acquire_one_transport(CS_ANY, ttype, serial, NULL); 1247 char *state = connection_state_name(transport); 1248 snprintf(buf, sizeof buf, "OKAY%04x%s",(unsigned)strlen(state),state); 1249 writex(reply_fd, buf, strlen(buf)); 1250 return 0; 1251 } 1252 return -1; 1253} 1254 1255#if !ADB_HOST 1256int recovery_mode = 0; 1257#endif 1258 1259int main(int argc, char **argv) 1260{ 1261 adb_trace_init(); 1262#if ADB_HOST 1263 adb_sysdeps_init(); 1264 return adb_commandline(argc - 1, argv + 1); 1265#else 1266 if((argc > 1) && (!strcmp(argv[1],"recovery"))) { 1267 adb_device_banner = "recovery"; 1268 recovery_mode = 1; 1269 } 1270 1271 start_device_log(); 1272 return adb_main(0, DEFAULT_ADB_PORT); 1273#endif 1274} 1275