main.cpp revision 5200c6670f041550c23821fec8e8e49b30ef6d29
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 17c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#define TRACE_TAG TRACE_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> 24c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 25c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert// We only build the affinity WAR code for Linux. 26c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(__linux__) 27c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include <sched.h> 28c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 29c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 30c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "base/file.h" 31c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "base/logging.h" 32c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "base/stringprintf.h" 33c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 34c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "adb.h" 35c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "adb_auth.h" 36c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "adb_listeners.h" 37c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#include "transport.h" 38c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 39c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(WORKAROUND_BUG6558362) && defined(__linux__) 40c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic const bool kWorkaroundBug6558362 = true; 41c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 42c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic const bool kWorkaroundBug6558362 = false; 43c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 44c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 45c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic void adb_workaround_affinity(void) { 46c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(__linux__) 47c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert const char affinity_env[] = "ADB_CPU_AFFINITY_BUG6558362"; 48c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert const char* cpunum_str = getenv(affinity_env); 49c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (cpunum_str == nullptr || *cpunum_str == '\0') { 50c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return; 51c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 52c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 53c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert char* strtol_res; 54c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int cpu_num = strtol(cpunum_str, &strtol_res, 0); 55c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (*strtol_res != '\0') { 56c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fatal("bad number (%s) in env var %s. Expecting 0..n.\n", cpunum_str, 57c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert affinity_env); 58c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 59c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 60c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert cpu_set_t cpu_set; 61c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert sched_getaffinity(0, sizeof(cpu_set), &cpu_set); 62c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert D("orig cpu_set[0]=0x%08lx\n", cpu_set.__bits[0]); 63c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 64c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert CPU_ZERO(&cpu_set); 65c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert CPU_SET(cpu_num, &cpu_set); 66c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert sched_setaffinity(0, sizeof(cpu_set), &cpu_set); 67c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 68c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert sched_getaffinity(0, sizeof(cpu_set), &cpu_set); 69c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert D("new cpu_set[0]=0x%08lx\n", cpu_set.__bits[0]); 70c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 71c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // No workaround was ever implemented for the other platforms. 72c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 73c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 74c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 75c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 76c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic const char kNullFileName[] = "NUL"; 77c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 78c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic BOOL WINAPI ctrlc_handler(DWORD type) { 79c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert exit(STATUS_CONTROL_C_EXIT); 80c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return TRUE; 81c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 82c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 83c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic std::string GetLogFilePath() { 84c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert const char log_name[] = "adb.log"; 85c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert char temp_path[MAX_PATH - sizeof(log_name) + 1]; 86c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 87c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // https://msdn.microsoft.com/en-us/library/windows/desktop/aa364992%28v=vs.85%29.aspx 88c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert DWORD nchars = GetTempPath(sizeof(temp_path), temp_path); 89c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert CHECK_LE(nchars, sizeof(temp_path)); 90c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (nchars == 0) { 91c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // TODO(danalbert): Log the error message from FormatError(). 92c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // Windows unfortunately has two errnos, errno and GetLastError(), so 93c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // I'm not sure what to do about PLOG here. Probably better to just 94c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // ignore it and add a simplified version of FormatError() for use in 95c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // log messages. 96c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert LOG(ERROR) << "Error creating log file"; 97c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 98c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 99c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return std::string(temp_path) + log_name; 100c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 101c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 102c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic const char kNullFileName[] = "/dev/null"; 103c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 104c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic std::string GetLogFilePath() { 105c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return std::string("/tmp/adb.log"); 106c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 107c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 108c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 109c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic void close_stdin() { 110c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int fd = unix_open(kNullFileName, O_RDONLY); 111c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert CHECK_NE(fd, -1); 112c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert dup2(fd, STDIN_FILENO); 113d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low unix_close(fd); 114c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 115c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 116c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic void setup_daemon_logging(void) { 117c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int fd = unix_open(GetLogFilePath().c_str(), O_WRONLY | O_CREAT | O_APPEND, 118c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 0640); 119c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (fd == -1) { 120c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fd = unix_open(kNullFileName, O_WRONLY); 121c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 122c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert dup2(fd, STDOUT_FILENO); 123c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert dup2(fd, STDERR_FILENO); 124d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low unix_close(fd); 125d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low 126d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low#ifdef _WIN32 127d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // On Windows, stderr is buffered by default, so switch to non-buffered 128d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // to match Linux. 129d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low setvbuf(stderr, NULL, _IONBF, 0); 130d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low#endif 131c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fprintf(stderr, "--- adb starting (pid %d) ---\n", getpid()); 132c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 133c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 134c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint adb_main(int is_daemon, int server_port) { 135c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert HOST = 1; 136c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 137c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 138c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert SetConsoleCtrlHandler(ctrlc_handler, TRUE); 139c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 140c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert signal(SIGPIPE, SIG_IGN); 141c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 142c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 143c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert init_transport_registration(); 144c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 145c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (kWorkaroundBug6558362 && is_daemon) { 146c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_workaround_affinity(); 147c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 148c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 149c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert usb_init(); 150c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT); 151c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_auth_init(); 152c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 1535200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low std::string error; 154c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert std::string local_name = android::base::StringPrintf("tcp:%d", server_port); 1555200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low if (install_listener(local_name, "*smartsocket*", nullptr, 0, &error)) { 1565200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low LOG(FATAL) << "Could not install *smartsocket* listener: " << error; 157c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 158c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 159c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (is_daemon) { 160c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // Inform our parent that we are up and running. 161c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // TODO(danalbert): Can't use SendOkay because we're sending "OK\n", not 162c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // "OKAY". 163d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // TODO(danalbert): Why do we use stdout for Windows? There is a 164d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // comment in launch_server() that suggests that non-Windows uses 165d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // stderr because it is non-buffered. So perhaps the history is that 166d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // stdout was preferred for all platforms, but it was discovered that 167d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // non-Windows needed a non-buffered fd, so stderr was used there. 168d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // Note that using stderr on unix means that if you do 169d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // `ADB_TRACE=all adb start-server`, it will say "ADB server didn't ACK" 170d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // and "* failed to start daemon *" because the adb server will write 171d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low // logging to stderr, obscuring the OK\n output that is sent to stderr. 172c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 173c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int reply_fd = STDOUT_FILENO; 174d396dc93a7b2e2f55eb6a34f04fe796769e9eceaSpencer Low // Change stdout mode to binary so \n => \r\n translation does not 175d396dc93a7b2e2f55eb6a34f04fe796769e9eceaSpencer Low // occur. In a moment stdout will be reopened to the daemon log file 176d396dc93a7b2e2f55eb6a34f04fe796769e9eceaSpencer Low // anyway. 177d396dc93a7b2e2f55eb6a34f04fe796769e9eceaSpencer Low _setmode(reply_fd, _O_BINARY); 178c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 179c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int reply_fd = STDERR_FILENO; 180c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 181c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert android::base::WriteStringToFd("OK\n", reply_fd); 182c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert close_stdin(); 183c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert setup_daemon_logging(); 184c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 185c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 186c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert D("Event loop starting\n"); 187c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fdevent_loop(); 188c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 189c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return 0; 190c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 191c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 192c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint main(int argc, char** argv) { 193c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_sysdeps_init(); 1949313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert adb_trace_init(argv); 195c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert D("Handling commandline()\n"); 196c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return adb_commandline(argc - 1, const_cast<const char**>(argv + 1)); 197c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 198