1c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert/* 2c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * Copyright (C) 2015 The Android Open Source Project 3c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * 4c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * Licensed under the Apache License, Version 2.0 (the "License"); 5c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * you may not use this file except in compliance with the License. 6c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * You may obtain a copy of the License at 7c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * 8c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * http://www.apache.org/licenses/LICENSE-2.0 9c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * 10c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * Unless required by applicable law or agreed to in writing, software 11c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * distributed under the License is distributed on an "AS IS" BASIS, 12c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * See the License for the specific language governing permissions and 14c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert * limitations under the License. 15c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert */ 16c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 17aed3c61c4437ebb05eadfb3bf85d6962c30b9935Yabin Cui#define TRACE_TAG ADB 18c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 19c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "sysdeps.h" 20c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 21c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include <signal.h> 22c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include <stdio.h> 23c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include <stdlib.h> 24b72b3f8c92a0b2d4377d5bec15eb45869b8711acJosh Gao#include <unistd.h> 25c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 26f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao#include <thread> 27f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao 285f787ed2b3b9f6cc02aa5923b95d77e2a5865438David Pursell#include <android-base/errors.h> 294f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/file.h> 304f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/logging.h> 311c70e1bcbcced190b351d4fb418f32b4e428f496Josh Gao#include <android-base/quick_exit.h> 324f71319df011d796a60a43fc1bc68e16fbf7d321Elliott Hughes#include <android-base/stringprintf.h> 33c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 34c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "adb.h" 35c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "adb_auth.h" 36c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "adb_listeners.h" 377d586073609723cb2f6ed37de0ad1a7996e621aeJosh Gao#include "adb_utils.h" 38698e065e14840541f218a80241fa880703a7dbc5Felipe Leme#include "commandline.h" 39f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao#include "sysdeps/chrono.h" 40c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "transport.h" 41c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 426eadee860798bf4e0be0293450017fbc95ff41c8Elliott Hughesstatic void setup_daemon_logging() { 43155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low const std::string log_file_path(GetLogFilePath()); 4465433da1cbd8d2869a60ce3bf18b6b440461741dElliott Hughes int fd = unix_open(log_file_path.c_str(), O_WRONLY | O_CREAT | O_APPEND, 0640); 45c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (fd == -1) { 46155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot open '%s': %s", log_file_path.c_str(), strerror(errno)); 47155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low } 48155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low if (dup2(fd, STDOUT_FILENO) == -1) { 49155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot redirect stdout: %s", strerror(errno)); 50155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low } 51155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low if (dup2(fd, STDERR_FILENO) == -1) { 52155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot redirect stderr: %s", strerror(errno)); 53c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 54d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low unix_close(fd); 55d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low 56c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fprintf(stderr, "--- adb starting (pid %d) ---\n", getpid()); 57743883b570b200ff4d73d60798d63136f3e3d82eSiva Velusamy LOG(INFO) << adb_version(); 58c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 59c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 60d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughes#if defined(_WIN32) 61d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughesstatic BOOL WINAPI ctrlc_handler(DWORD type) { 62d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughes // TODO: Consider trying to kill a starting up adb server (if we're in 63d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughes // launch_server) by calling GenerateConsoleCtrlEvent(). 641c70e1bcbcced190b351d4fb418f32b4e428f496Josh Gao android::base::quick_exit(STATUS_CONTROL_C_EXIT); 65d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughes return TRUE; 66d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughes} 67d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughes#endif 68d89a6c2285b64320918b25b1c818da24dd4d02b4Elliott Hughes 6901b7bc43e90b951675cabe88313b96f57b2da712Josh Gaovoid adb_server_cleanup() { 7001b7bc43e90b951675cabe88313b96f57b2da712Josh Gao // Upon exit, we want to clean up in the following order: 7101b7bc43e90b951675cabe88313b96f57b2da712Josh Gao // 1. close_smartsockets, so that we don't get any new clients 7201b7bc43e90b951675cabe88313b96f57b2da712Josh Gao // 2. kick_all_transports, to avoid writing only part of a packet to a transport. 7301b7bc43e90b951675cabe88313b96f57b2da712Josh Gao // 3. usb_cleanup, to tear down the USB stack. 7401b7bc43e90b951675cabe88313b96f57b2da712Josh Gao close_smartsockets(); 7501b7bc43e90b951675cabe88313b96f57b2da712Josh Gao kick_all_transports(); 7601b7bc43e90b951675cabe88313b96f57b2da712Josh Gao usb_cleanup(); 7701b7bc43e90b951675cabe88313b96f57b2da712Josh Gao} 7801b7bc43e90b951675cabe88313b96f57b2da712Josh Gao 799c869b58a8cf4f7c3bc88931fbd27d3f5187b2dbJosh Gaoint adb_server_main(int is_daemon, const std::string& socket_spec, int ack_reply_fd) { 80c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 812122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // adb start-server starts us up with stdout and stderr hooked up to 8265433da1cbd8d2869a60ce3bf18b6b440461741dElliott Hughes // anonymous pipes. When the C Runtime sees this, it makes stderr and 832122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // stdout buffered, but to improve the chance that error output is seen, 842122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // unbuffer stdout and stderr just like if we were run at the console. 852122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // This also keeps stderr unbuffered when it is redirected to adb.log. 862122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low if (is_daemon) { 872122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low if (setvbuf(stdout, NULL, _IONBF, 0) == -1) { 882122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low fatal("cannot make stdout unbuffered: %s", strerror(errno)); 892122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low } 902122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low if (setvbuf(stderr, NULL, _IONBF, 0) == -1) { 912122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low fatal("cannot make stderr unbuffered: %s", strerror(errno)); 922122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low } 932122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low } 942122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low 95c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert SetConsoleCtrlHandler(ctrlc_handler, TRUE); 961c70e1bcbcced190b351d4fb418f32b4e428f496Josh Gao#else 971c70e1bcbcced190b351d4fb418f32b4e428f496Josh Gao signal(SIGINT, [](int) { 9801b7bc43e90b951675cabe88313b96f57b2da712Josh Gao fdevent_run_on_main_thread([]() { android::base::quick_exit(0); }); 991c70e1bcbcced190b351d4fb418f32b4e428f496Josh Gao }); 100c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 101c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 10216c5a133ab2b80af2b56f86e42de04a4399c60fcJosh Gao if (is_daemon) { 10316c5a133ab2b80af2b56f86e42de04a4399c60fcJosh Gao close_stdin(); 10416c5a133ab2b80af2b56f86e42de04a4399c60fcJosh Gao setup_daemon_logging(); 10516c5a133ab2b80af2b56f86e42de04a4399c60fcJosh Gao } 10616c5a133ab2b80af2b56f86e42de04a4399c60fcJosh Gao 10701b7bc43e90b951675cabe88313b96f57b2da712Josh Gao android::base::at_quick_exit(adb_server_cleanup); 108c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 10901b7bc43e90b951675cabe88313b96f57b2da712Josh Gao init_transport_registration(); 11013a269ea24a98111d2e84a7ffa3c05ab8a4e73a9Casey Dahlin init_mdns_transport_discovery(); 11113a269ea24a98111d2e84a7ffa3c05ab8a4e73a9Casey Dahlin 112c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert usb_init(); 113c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT); 114c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 1155200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low std::string error; 116f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao 117f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao auto start = std::chrono::steady_clock::now(); 118f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao 119f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao // If we told a previous adb server to quit because of version mismatch, we can get to this 120f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao // point before it's finished exiting. Retry for a while to give it some time. 121f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao while (install_listener(socket_spec, "*smartsocket*", nullptr, 0, nullptr, &error) != 122f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao INSTALL_STATUS_OK) { 123f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao if (std::chrono::steady_clock::now() - start > 0.5s) { 124f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao fatal("could not install *smartsocket* listener: %s", error.c_str()); 125f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao } 126f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao 127f0f854bc76458b10b6b3ed88655fd87cd156dc70Josh Gao std::this_thread::sleep_for(100ms); 128c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 129c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 1300aeb50500c76ea67d6f452907f5503d590e81a54Elliott Hughes adb_auth_init(); 1312122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low 1320aeb50500c76ea67d6f452907f5503d590e81a54Elliott Hughes if (is_daemon) { 133b72b3f8c92a0b2d4377d5bec15eb45869b8711acJosh Gao#if !defined(_WIN32) 1346bf323b97a11194d6186f8db2ee4a5eaca8d0141Yabin Cui // Start a new session for the daemon. Do this here instead of after the fork so 1356bf323b97a11194d6186f8db2ee4a5eaca8d0141Yabin Cui // that a ctrl-c between the "starting server" and "done starting server" messages 1366bf323b97a11194d6186f8db2ee4a5eaca8d0141Yabin Cui // gets a chance to terminate the server. 137c8fab89f1b5ae8c3752979e1a1004eda8b185687Tao Wu // setsid will fail with EPERM if it's already been a lead process of new session. 138c8fab89f1b5ae8c3752979e1a1004eda8b185687Tao Wu // Ignore such error. 139c8fab89f1b5ae8c3752979e1a1004eda8b185687Tao Wu if (setsid() == -1 && errno != EPERM) { 1406bf323b97a11194d6186f8db2ee4a5eaca8d0141Yabin Cui fatal("setsid() failed: %s", strerror(errno)); 1416bf323b97a11194d6186f8db2ee4a5eaca8d0141Yabin Cui } 142b72b3f8c92a0b2d4377d5bec15eb45869b8711acJosh Gao#endif 143b72b3f8c92a0b2d4377d5bec15eb45869b8711acJosh Gao 144fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao // Wait for the USB scan to complete before notifying the parent that we're up. 145fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao // We need to perform this in a thread, because we would otherwise block the event loop. 146fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao std::thread notify_thread([ack_reply_fd]() { 147fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao adb_wait_for_device_initialization(); 148fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao 149fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao // Any error output written to stderr now goes to adb.log. We could 150fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao // keep around a copy of the stderr fd and use that to write any errors 151fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao // encountered by the following code, but that is probably overkill. 152c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 153fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd); 154fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao const CHAR ack[] = "OK\n"; 155fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao const DWORD bytes_to_write = arraysize(ack) - 1; 156fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao DWORD written = 0; 157fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao if (!WriteFile(ack_reply_handle, ack, bytes_to_write, &written, NULL)) { 158fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao fatal("adb: cannot write ACK to handle 0x%p: %s", ack_reply_handle, 159fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao android::base::SystemErrorCodeToString(GetLastError()).c_str()); 160fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao } 161fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao if (written != bytes_to_write) { 162fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao fatal("adb: cannot write %lu bytes of ACK: only wrote %lu bytes", bytes_to_write, 163fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao written); 164fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao } 165fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao CloseHandle(ack_reply_handle); 1665c398d2ce96150e94e596e959bfe2246b94de660Spencer Low#else 167fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao // TODO(danalbert): Can't use SendOkay because we're sending "OK\n", not 168fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao // "OKAY". 169fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao if (!android::base::WriteStringToFd("OK\n", ack_reply_fd)) { 170fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao fatal_errno("error writing ACK to fd %d", ack_reply_fd); 171fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao } 172fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao unix_close(ack_reply_fd); 173c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 174fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao }); 175fd713e53e890f5b3bf26a1c5a980137cc4378c3aJosh Gao notify_thread.detach(); 176c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 177c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 1787a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui D("Event loop starting"); 179c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fdevent_loop(); 180c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 181c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return 0; 182c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 183c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 184c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint main(int argc, char** argv) { 1859313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert adb_trace_init(argv); 186c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return adb_commandline(argc - 1, const_cast<const char**>(argv + 1)); 187c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 188