main.cpp revision c89e0ccd40624e8b2ce30d425e8bed0264fae548
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); 113c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_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); 124c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_close(fd); 125c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fprintf(stderr, "--- adb starting (pid %d) ---\n", getpid()); 126c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 127c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 128c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint adb_main(int is_daemon, int server_port) { 129c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert HOST = 1; 130c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 131c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 132c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert SetConsoleCtrlHandler(ctrlc_handler, TRUE); 133c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 134c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert signal(SIGPIPE, SIG_IGN); 135c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 136c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 137c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert init_transport_registration(); 138c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 139c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (kWorkaroundBug6558362 && is_daemon) { 140c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_workaround_affinity(); 141c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 142c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 143c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert usb_init(); 144c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT); 145c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_auth_init(); 146c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 147c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert std::string local_name = android::base::StringPrintf("tcp:%d", server_port); 148c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (install_listener(local_name, "*smartsocket*", nullptr, 0)) { 149c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert LOG(FATAL) << "Could not install *smartsocket* listener"; 150c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 151c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 152c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (is_daemon) { 153c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // Inform our parent that we are up and running. 154c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // TODO(danalbert): Can't use SendOkay because we're sending "OK\n", not 155c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // "OKAY". 156c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // TODO(danalbert): Why do we use stdout for Windows? 157c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 158c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int reply_fd = STDOUT_FILENO; 159c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 160c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int reply_fd = STDERR_FILENO; 161c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 162c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert android::base::WriteStringToFd("OK\n", reply_fd); 163c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert close_stdin(); 164c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert setup_daemon_logging(); 165c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 166c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 167c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert D("Event loop starting\n"); 168c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fdevent_loop(); 169c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 170c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return 0; 171c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 172c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 173c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint main(int argc, char** argv) { 174c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_sysdeps_init(); 175c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 176c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert android::base::InitLogging(argv); 177c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_trace_init(); 178c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert D("Handling commandline()\n"); 179c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return adb_commandline(argc - 1, const_cast<const char**>(argv + 1)); 180c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 181