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