main.cpp revision 65433da1cbd8d2869a60ce3bf18b6b440461741d
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> 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(_WIN32) 40c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic const char kNullFileName[] = "NUL"; 41c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 42c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic BOOL WINAPI ctrlc_handler(DWORD type) { 432122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // TODO: Consider trying to kill a starting up adb server (if we're in 442122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // launch_server) by calling GenerateConsoleCtrlEvent(). 45c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert exit(STATUS_CONTROL_C_EXIT); 46c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return TRUE; 47c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 48c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 49c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic std::string GetLogFilePath() { 50c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert const char log_name[] = "adb.log"; 51cf4ff64f2d3cdc4ffa2f06e594d18a965f591a82Spencer Low WCHAR temp_path[MAX_PATH]; 52c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 53c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert // https://msdn.microsoft.com/en-us/library/windows/desktop/aa364992%28v=vs.85%29.aspx 54cf4ff64f2d3cdc4ffa2f06e594d18a965f591a82Spencer Low DWORD nchars = GetTempPathW(arraysize(temp_path), temp_path); 55cf4ff64f2d3cdc4ffa2f06e594d18a965f591a82Spencer Low if ((nchars >= arraysize(temp_path)) || (nchars == 0)) { 56cf4ff64f2d3cdc4ffa2f06e594d18a965f591a82Spencer Low // If string truncation or some other error. 57155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot retrieve temporary file path: %s\n", 58155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low SystemErrorCodeToString(GetLastError()).c_str()); 59c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 60c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 61d21dc825bbecad6ce480c5e5c574cc77eadcd779Spencer Low std::string temp_path_utf8; 62d21dc825bbecad6ce480c5e5c574cc77eadcd779Spencer Low if (!android::base::WideToUTF8(temp_path, &temp_path_utf8)) { 63d21dc825bbecad6ce480c5e5c574cc77eadcd779Spencer Low fatal_errno("cannot convert temporary file path from UTF-16 to UTF-8"); 64d21dc825bbecad6ce480c5e5c574cc77eadcd779Spencer Low } 65d21dc825bbecad6ce480c5e5c574cc77eadcd779Spencer Low 66d21dc825bbecad6ce480c5e5c574cc77eadcd779Spencer Low return temp_path_utf8 + log_name; 67c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 68c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#else 69c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic const char kNullFileName[] = "/dev/null"; 70c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 71c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic std::string GetLogFilePath() { 72c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return std::string("/tmp/adb.log"); 73c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 74c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 75c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 76c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic void close_stdin() { 77c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert int fd = unix_open(kNullFileName, O_RDONLY); 78155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low if (fd == -1) { 79155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot open '%s': %s", kNullFileName, strerror(errno)); 80155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low } 81155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low if (dup2(fd, STDIN_FILENO) == -1) { 82155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot redirect stdin: %s", strerror(errno)); 83155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low } 84d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low unix_close(fd); 85c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 86c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 87c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertstatic void setup_daemon_logging(void) { 88155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low const std::string log_file_path(GetLogFilePath()); 8965433da1cbd8d2869a60ce3bf18b6b440461741dElliott Hughes int fd = unix_open(log_file_path.c_str(), O_WRONLY | O_CREAT | O_APPEND, 0640); 90c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (fd == -1) { 91155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot open '%s': %s", log_file_path.c_str(), strerror(errno)); 92155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low } 93155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low if (dup2(fd, STDOUT_FILENO) == -1) { 94155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot redirect stdout: %s", strerror(errno)); 95155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low } 96155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low if (dup2(fd, STDERR_FILENO) == -1) { 97155159c545ee1a6d8e3cbea866ea66096f4db5f3Spencer Low fatal("cannot redirect stderr: %s", strerror(errno)); 98c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 99d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low unix_close(fd); 100d0f66c361641dc30162877ee3e4b4690d6833442Spencer Low 101c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fprintf(stderr, "--- adb starting (pid %d) ---\n", getpid()); 102743883b570b200ff4d73d60798d63136f3e3d82eSiva Velusamy LOG(INFO) << adb_version(); 103c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 104c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 10565433da1cbd8d2869a60ce3bf18b6b440461741dElliott Hughesint adb_server_main(int is_daemon, int server_port, int ack_reply_fd) { 106c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 1072122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // adb start-server starts us up with stdout and stderr hooked up to 10865433da1cbd8d2869a60ce3bf18b6b440461741dElliott Hughes // anonymous pipes. When the C Runtime sees this, it makes stderr and 1092122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // stdout buffered, but to improve the chance that error output is seen, 1102122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // unbuffer stdout and stderr just like if we were run at the console. 1112122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // This also keeps stderr unbuffered when it is redirected to adb.log. 1122122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low if (is_daemon) { 1132122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low if (setvbuf(stdout, NULL, _IONBF, 0) == -1) { 1142122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low fatal("cannot make stdout unbuffered: %s", strerror(errno)); 1152122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low } 1162122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low if (setvbuf(stderr, NULL, _IONBF, 0) == -1) { 1172122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low fatal("cannot make stderr unbuffered: %s", strerror(errno)); 1182122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low } 1192122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low } 1202122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low 121c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert SetConsoleCtrlHandler(ctrlc_handler, TRUE); 122c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 123c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 124c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert init_transport_registration(); 125c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 126c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert usb_init(); 127c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert local_init(DEFAULT_ADB_LOCAL_TRANSPORT_PORT); 128c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_auth_init(); 129c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 1305200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low std::string error; 131c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert std::string local_name = android::base::StringPrintf("tcp:%d", server_port); 1325200c6670f041550c23821fec8e8e49b30ef6d29Spencer Low if (install_listener(local_name, "*smartsocket*", nullptr, 0, &error)) { 133bf7c605d87f87c03066c384cecb0f0c91aa31403Spencer Low fatal("could not install *smartsocket* listener: %s", error.c_str()); 134c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 135c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 1369f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy // Inform our parent that we are up and running. 137c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert if (is_daemon) { 1382122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low close_stdin(); 1392122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low setup_daemon_logging(); 1402122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low 1412122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // Any error output written to stderr now goes to adb.log. We could 1422122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // keep around a copy of the stderr fd and use that to write any errors 1432122c7a1483dc6050b33f58fad4a15d5acd79fddSpencer Low // encountered by the following code, but that is probably overkill. 144c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#if defined(_WIN32) 1455c398d2ce96150e94e596e959bfe2246b94de660Spencer Low const HANDLE ack_reply_handle = cast_int_to_handle(ack_reply_fd); 1465c398d2ce96150e94e596e959bfe2246b94de660Spencer Low const CHAR ack[] = "OK\n"; 1475c398d2ce96150e94e596e959bfe2246b94de660Spencer Low const DWORD bytes_to_write = arraysize(ack) - 1; 1485c398d2ce96150e94e596e959bfe2246b94de660Spencer Low DWORD written = 0; 1495c398d2ce96150e94e596e959bfe2246b94de660Spencer Low if (!WriteFile(ack_reply_handle, ack, bytes_to_write, &written, NULL)) { 1505c398d2ce96150e94e596e959bfe2246b94de660Spencer Low fatal("adb: cannot write ACK to handle 0x%p: %s", ack_reply_handle, 1515c398d2ce96150e94e596e959bfe2246b94de660Spencer Low SystemErrorCodeToString(GetLastError()).c_str()); 1525c398d2ce96150e94e596e959bfe2246b94de660Spencer Low } 1535c398d2ce96150e94e596e959bfe2246b94de660Spencer Low if (written != bytes_to_write) { 1545c398d2ce96150e94e596e959bfe2246b94de660Spencer Low fatal("adb: cannot write %lu bytes of ACK: only wrote %lu bytes", 1555c398d2ce96150e94e596e959bfe2246b94de660Spencer Low bytes_to_write, written); 1565c398d2ce96150e94e596e959bfe2246b94de660Spencer Low } 1575c398d2ce96150e94e596e959bfe2246b94de660Spencer Low CloseHandle(ack_reply_handle); 1585c398d2ce96150e94e596e959bfe2246b94de660Spencer Low#else 1599f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy // TODO(danalbert): Can't use SendOkay because we're sending "OK\n", not 1609f2d1a9cfc04e1d5970823da1878097288a9a9cdSiva Velusamy // "OKAY". 161f18fc0879be9cd506be6b1e051feb543b0746bdbSpencer Low if (!android::base::WriteStringToFd("OK\n", ack_reply_fd)) { 162f18fc0879be9cd506be6b1e051feb543b0746bdbSpencer Low fatal_errno("error writing ACK to fd %d", ack_reply_fd); 163f18fc0879be9cd506be6b1e051feb543b0746bdbSpencer Low } 1645c398d2ce96150e94e596e959bfe2246b94de660Spencer Low unix_close(ack_reply_fd); 165c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert#endif 166c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert } 167c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 1687a3f8d6691b3fbd8014a98de8455dbcfcc9629e4Yabin Cui D("Event loop starting"); 169c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert fdevent_loop(); 170c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 171c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return 0; 172c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 173c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert 174c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albertint main(int argc, char** argv) { 175c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert adb_sysdeps_init(); 1769313c0df20d7cc8ea6a074a3d53022c4b3b5ea05Dan Albert adb_trace_init(argv); 177c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert return adb_commandline(argc - 1, const_cast<const char**>(argv + 1)); 178c89e0ccd40624e8b2ce30d425e8bed0264fae548Dan Albert} 179