The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 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 | |
Yabin Cui | 19bec5b | 2015-09-22 15:52:57 -0700 | [diff] [blame] | 17 | #define TRACE_TAG TRANSPORT |
Dan Albert | db6fe64 | 2015-03-19 15:21:08 -0700 | [diff] [blame] | 18 | |
| 19 | #include "sysdeps.h" |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 20 | #include "sysdeps/condition_variable.h" |
| 21 | #include "sysdeps/mutex.h" |
Dan Albert | db6fe64 | 2015-03-19 15:21:08 -0700 | [diff] [blame] | 22 | #include "transport.h" |
| 23 | |
Dan Albert | b302d12 | 2015-02-24 15:51:19 -0800 | [diff] [blame] | 24 | #include <errno.h> |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 25 | #include <stdio.h> |
| 26 | #include <stdlib.h> |
| 27 | #include <string.h> |
Dan Albert | b302d12 | 2015-02-24 15:51:19 -0800 | [diff] [blame] | 28 | #include <sys/types.h> |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 29 | |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 30 | #include <vector> |
| 31 | |
Elliott Hughes | f55ead9 | 2015-12-04 22:00:26 -0800 | [diff] [blame] | 32 | #include <android-base/stringprintf.h> |
Elliott Hughes | 43df109 | 2015-07-23 17:12:58 -0700 | [diff] [blame] | 33 | #include <cutils/sockets.h> |
Elliott Hughes | fb59684 | 2015-05-01 17:04:38 -0700 | [diff] [blame] | 34 | |
Dan Albert | b302d12 | 2015-02-24 15:51:19 -0800 | [diff] [blame] | 35 | #if !ADB_HOST |
| 36 | #include "cutils/properties.h" |
| 37 | #endif |
Dan Albert | db6fe64 | 2015-03-19 15:21:08 -0700 | [diff] [blame] | 38 | |
| 39 | #include "adb.h" |
| 40 | #include "adb_io.h" |
Elliott Hughes | 43df109 | 2015-07-23 17:12:58 -0700 | [diff] [blame] | 41 | #include "adb_utils.h" |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 42 | |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 43 | #if ADB_HOST |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 44 | /* we keep a list of opened transports. The atransport struct knows to which |
| 45 | * local transport it is connected. The list is used to detect when we're |
| 46 | * trying to connect twice to a given local transport. |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 47 | */ |
David 'Digit' Turner | b0f0a46 | 2014-03-13 11:21:58 +0100 | [diff] [blame] | 48 | #define ADB_LOCAL_TRANSPORT_MAX 64 |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 49 | |
| 50 | ADB_MUTEX_DEFINE( local_transports_lock ); |
| 51 | |
| 52 | static atransport* local_transports[ ADB_LOCAL_TRANSPORT_MAX ]; |
| 53 | #endif /* ADB_HOST */ |
| 54 | |
| 55 | static int remote_read(apacket *p, atransport *t) |
| 56 | { |
Dan Albert | 66a91b0 | 2015-02-24 21:26:58 -0800 | [diff] [blame] | 57 | if(!ReadFdExactly(t->sfd, &p->msg, sizeof(amessage))){ |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 58 | D("remote local: read terminated (message)"); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 59 | return -1; |
| 60 | } |
| 61 | |
Tamas Berghammer | a1c60c0 | 2015-07-13 19:12:28 +0100 | [diff] [blame] | 62 | if(check_header(p, t)) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 63 | D("bad header: terminated (data)"); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 64 | return -1; |
| 65 | } |
| 66 | |
Dan Albert | 66a91b0 | 2015-02-24 21:26:58 -0800 | [diff] [blame] | 67 | if(!ReadFdExactly(t->sfd, p->data, p->msg.data_length)){ |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 68 | D("remote local: terminated (data)"); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 69 | return -1; |
| 70 | } |
| 71 | |
| 72 | if(check_data(p)) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 73 | D("bad data: terminated (data)"); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 74 | return -1; |
| 75 | } |
| 76 | |
| 77 | return 0; |
| 78 | } |
| 79 | |
| 80 | static int remote_write(apacket *p, atransport *t) |
| 81 | { |
| 82 | int length = p->msg.data_length; |
| 83 | |
Dan Albert | 66a91b0 | 2015-02-24 21:26:58 -0800 | [diff] [blame] | 84 | if(!WriteFdExactly(t->sfd, &p->msg, sizeof(amessage) + length)) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 85 | D("remote local: write terminated"); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 86 | return -1; |
| 87 | } |
| 88 | |
| 89 | return 0; |
| 90 | } |
| 91 | |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 92 | bool local_connect(int port) { |
Elliott Hughes | 43df109 | 2015-07-23 17:12:58 -0700 | [diff] [blame] | 93 | std::string dummy; |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 94 | return local_connect_arbitrary_ports(port-1, port, &dummy) == 0; |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 95 | } |
| 96 | |
Elliott Hughes | 43df109 | 2015-07-23 17:12:58 -0700 | [diff] [blame] | 97 | int local_connect_arbitrary_ports(int console_port, int adb_port, std::string* error) { |
| 98 | int fd = -1; |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 99 | |
| 100 | #if ADB_HOST |
Yabin Cui | d802bcf | 2015-07-31 14:10:54 -0700 | [diff] [blame] | 101 | if (find_emulator_transport_by_adb_port(adb_port) != nullptr) { |
| 102 | return -1; |
| 103 | } |
| 104 | |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 105 | const char *host = getenv("ADBHOST"); |
| 106 | if (host) { |
Elliott Hughes | 43df109 | 2015-07-23 17:12:58 -0700 | [diff] [blame] | 107 | fd = network_connect(host, adb_port, SOCK_STREAM, 0, error); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 108 | } |
| 109 | #endif |
| 110 | if (fd < 0) { |
Spencer Low | 753d485 | 2015-07-30 23:07:55 -0700 | [diff] [blame] | 111 | fd = network_loopback_client(adb_port, SOCK_STREAM, error); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 112 | } |
| 113 | |
| 114 | if (fd >= 0) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 115 | D("client: connected on remote on fd %d", fd); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 116 | close_on_exec(fd); |
| 117 | disable_tcp_nagle(fd); |
Elliott Hughes | fb59684 | 2015-05-01 17:04:38 -0700 | [diff] [blame] | 118 | std::string serial = android::base::StringPrintf("emulator-%d", console_port); |
Yabin Cui | d802bcf | 2015-07-31 14:10:54 -0700 | [diff] [blame] | 119 | if (register_socket_transport(fd, serial.c_str(), adb_port, 1) == 0) { |
| 120 | return 0; |
| 121 | } |
| 122 | adb_close(fd); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 123 | } |
| 124 | return -1; |
| 125 | } |
| 126 | |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 127 | #if ADB_HOST |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 128 | |
| 129 | static void PollAllLocalPortsForEmulator() { |
| 130 | int port = DEFAULT_ADB_LOCAL_TRANSPORT_PORT; |
| 131 | int count = ADB_LOCAL_TRANSPORT_MAX; |
| 132 | |
| 133 | // Try to connect to any number of running emulator instances. |
| 134 | for ( ; count > 0; count--, port += 2 ) { |
| 135 | local_connect(port); |
| 136 | } |
| 137 | } |
| 138 | |
| 139 | // Retry the disconnected local port for 60 times, and sleep 1 second between two retries. |
| 140 | constexpr uint32_t LOCAL_PORT_RETRY_COUNT = 60; |
| 141 | constexpr uint32_t LOCAL_PORT_RETRY_INTERVAL_IN_MS = 1000; |
| 142 | |
| 143 | struct RetryPort { |
| 144 | int port; |
| 145 | uint32_t retry_count; |
| 146 | }; |
| 147 | |
| 148 | // Retry emulators just kicked. |
| 149 | static std::vector<RetryPort>& retry_ports = *new std::vector<RetryPort>; |
| 150 | std::mutex &retry_ports_lock = *new std::mutex; |
| 151 | std::condition_variable &retry_ports_cond = *new std::condition_variable; |
| 152 | |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 153 | static void client_socket_thread(void* x) { |
Siva Velusamy | 2669cf9 | 2015-08-28 16:37:29 -0700 | [diff] [blame] | 154 | adb_thread_setname("client_socket_thread"); |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 155 | D("transport: client_socket_thread() starting"); |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 156 | PollAllLocalPortsForEmulator(); |
Yabin Cui | d802bcf | 2015-07-31 14:10:54 -0700 | [diff] [blame] | 157 | while (true) { |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 158 | std::vector<RetryPort> ports; |
| 159 | // Collect retry ports. |
| 160 | { |
| 161 | std::unique_lock<std::mutex> lock(retry_ports_lock); |
| 162 | while (retry_ports.empty()) { |
| 163 | retry_ports_cond.wait(lock); |
| 164 | } |
| 165 | retry_ports.swap(ports); |
Yabin Cui | d802bcf | 2015-07-31 14:10:54 -0700 | [diff] [blame] | 166 | } |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 167 | // Sleep here instead of the end of loop, because if we immediately try to reconnect |
| 168 | // the emulator just kicked, the adbd on the emulator may not have time to remove the |
| 169 | // just kicked transport. |
| 170 | adb_sleep_ms(LOCAL_PORT_RETRY_INTERVAL_IN_MS); |
| 171 | |
| 172 | // Try connecting retry ports. |
| 173 | std::vector<RetryPort> next_ports; |
| 174 | for (auto& port : ports) { |
| 175 | VLOG(TRANSPORT) << "retry port " << port.port << ", last retry_count " |
| 176 | << port.retry_count; |
| 177 | if (local_connect(port.port)) { |
| 178 | VLOG(TRANSPORT) << "retry port " << port.port << " successfully"; |
| 179 | continue; |
| 180 | } |
| 181 | if (--port.retry_count > 0) { |
| 182 | next_ports.push_back(port); |
| 183 | } else { |
| 184 | VLOG(TRANSPORT) << "stop retrying port " << port.port; |
| 185 | } |
| 186 | } |
| 187 | |
| 188 | // Copy back left retry ports. |
| 189 | { |
| 190 | std::unique_lock<std::mutex> lock(retry_ports_lock); |
| 191 | retry_ports.insert(retry_ports.end(), next_ports.begin(), next_ports.end()); |
| 192 | } |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 193 | } |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 194 | } |
| 195 | |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 196 | #else // ADB_HOST |
| 197 | |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 198 | static void server_socket_thread(void* arg) { |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 199 | int serverfd, fd; |
Erik Kline | 3c8d44d | 2015-12-01 17:27:59 +0900 | [diff] [blame] | 200 | sockaddr_storage ss; |
| 201 | sockaddr *addrp = reinterpret_cast<sockaddr*>(&ss); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 202 | socklen_t alen; |
Elliott Hughes | 9c0d940 | 2014-01-16 10:53:11 -0800 | [diff] [blame] | 203 | int port = (int) (uintptr_t) arg; |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 204 | |
Siva Velusamy | 2669cf9 | 2015-08-28 16:37:29 -0700 | [diff] [blame] | 205 | adb_thread_setname("server socket"); |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 206 | D("transport: server_socket_thread() starting"); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 207 | serverfd = -1; |
| 208 | for(;;) { |
| 209 | if(serverfd == -1) { |
Spencer Low | 753d485 | 2015-07-30 23:07:55 -0700 | [diff] [blame] | 210 | std::string error; |
| 211 | serverfd = network_inaddr_any_server(port, SOCK_STREAM, &error); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 212 | if(serverfd < 0) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 213 | D("server: cannot bind socket yet: %s", error.c_str()); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 214 | adb_sleep_ms(1000); |
| 215 | continue; |
| 216 | } |
| 217 | close_on_exec(serverfd); |
| 218 | } |
| 219 | |
Erik Kline | 3c8d44d | 2015-12-01 17:27:59 +0900 | [diff] [blame] | 220 | alen = sizeof(ss); |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 221 | D("server: trying to get new connection from %d", port); |
Erik Kline | 3c8d44d | 2015-12-01 17:27:59 +0900 | [diff] [blame] | 222 | fd = adb_socket_accept(serverfd, addrp, &alen); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 223 | if(fd >= 0) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 224 | D("server: new connection on fd %d", fd); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 225 | close_on_exec(fd); |
| 226 | disable_tcp_nagle(fd); |
Yabin Cui | c722178 | 2016-05-17 15:14:25 -0700 | [diff] [blame] | 227 | if (register_socket_transport(fd, "host", port, 1) != 0) { |
| 228 | adb_close(fd); |
| 229 | } |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 230 | } |
| 231 | } |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 232 | D("transport: server_socket_thread() exiting"); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 233 | } |
| 234 | |
Vladimir Chtchetkine | 4bdc776 | 2011-12-13 12:19:29 -0800 | [diff] [blame] | 235 | /* This is relevant only for ADB daemon running inside the emulator. */ |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 236 | /* |
| 237 | * Redefine open and write for qemu_pipe.h that contains inlined references |
| 238 | * to those routines. We will redifine them back after qemu_pipe.h inclusion. |
| 239 | */ |
| 240 | #undef open |
| 241 | #undef write |
| 242 | #define open adb_open |
| 243 | #define write adb_write |
| 244 | #include <hardware/qemu_pipe.h> |
| 245 | #undef open |
| 246 | #undef write |
| 247 | #define open ___xxx_open |
| 248 | #define write ___xxx_write |
| 249 | |
| 250 | /* A worker thread that monitors host connections, and registers a transport for |
| 251 | * every new host connection. This thread replaces server_socket_thread on |
| 252 | * condition that adbd daemon runs inside the emulator, and emulator uses QEMUD |
| 253 | * pipe to communicate with adbd daemon inside the guest. This is done in order |
| 254 | * to provide more robust communication channel between ADB host and guest. The |
| 255 | * main issue with server_socket_thread approach is that it runs on top of TCP, |
| 256 | * and thus is sensitive to network disruptions. For instance, the |
| 257 | * ConnectionManager may decide to reset all network connections, in which case |
| 258 | * the connection between ADB host and guest will be lost. To make ADB traffic |
| 259 | * independent from the network, we use here 'adb' QEMUD service to transfer data |
| 260 | * between the host, and the guest. See external/qemu/android/adb-*.* that |
| 261 | * implements the emulator's side of the protocol. Another advantage of using |
| 262 | * QEMUD approach is that ADB will be up much sooner, since it doesn't depend |
| 263 | * anymore on network being set up. |
| 264 | * The guest side of the protocol contains the following phases: |
| 265 | * - Connect with adb QEMUD service. In this phase a handle to 'adb' QEMUD service |
| 266 | * is opened, and it becomes clear whether or not emulator supports that |
| 267 | * protocol. |
| 268 | * - Wait for the ADB host to create connection with the guest. This is done by |
| 269 | * sending an 'accept' request to the adb QEMUD service, and waiting on |
| 270 | * response. |
| 271 | * - When new ADB host connection is accepted, the connection with adb QEMUD |
| 272 | * service is registered as the transport, and a 'start' request is sent to the |
| 273 | * adb QEMUD service, indicating that the guest is ready to receive messages. |
| 274 | * Note that the guest will ignore messages sent down from the emulator before |
| 275 | * the transport registration is completed. That's why we need to send the |
| 276 | * 'start' request after the transport is registered. |
| 277 | */ |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 278 | static void qemu_socket_thread(void* arg) { |
| 279 | /* 'accept' request to the adb QEMUD service. */ |
| 280 | static const char _accept_req[] = "accept"; |
| 281 | /* 'start' request to the adb QEMUD service. */ |
| 282 | static const char _start_req[] = "start"; |
| 283 | /* 'ok' reply from the adb QEMUD service. */ |
| 284 | static const char _ok_resp[] = "ok"; |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 285 | |
Elliott Hughes | 9c0d940 | 2014-01-16 10:53:11 -0800 | [diff] [blame] | 286 | const int port = (int) (uintptr_t) arg; |
bohu | 82a2834 | 2016-02-29 13:13:19 -0800 | [diff] [blame] | 287 | int fd; |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 288 | char tmp[256]; |
| 289 | char con_name[32]; |
| 290 | |
Siva Velusamy | 2669cf9 | 2015-08-28 16:37:29 -0700 | [diff] [blame] | 291 | adb_thread_setname("qemu socket"); |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 292 | D("transport: qemu_socket_thread() starting"); |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 293 | |
| 294 | /* adb QEMUD service connection request. */ |
| 295 | snprintf(con_name, sizeof(con_name), "qemud:adb:%d", port); |
| 296 | |
| 297 | /* Connect to the adb QEMUD service. */ |
| 298 | fd = qemu_pipe_open(con_name); |
| 299 | if (fd < 0) { |
| 300 | /* This could be an older version of the emulator, that doesn't |
| 301 | * implement adb QEMUD service. Fall back to the old TCP way. */ |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 302 | D("adb service is not available. Falling back to TCP socket."); |
Elliott Hughes | f251714 | 2015-05-05 13:41:21 -0700 | [diff] [blame] | 303 | adb_thread_create(server_socket_thread, arg); |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 304 | return; |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 305 | } |
| 306 | |
| 307 | for(;;) { |
| 308 | /* |
| 309 | * Wait till the host creates a new connection. |
| 310 | */ |
| 311 | |
| 312 | /* Send the 'accept' request. */ |
bohu | 82a2834 | 2016-02-29 13:13:19 -0800 | [diff] [blame] | 313 | if (WriteFdExactly(fd, _accept_req, strlen(_accept_req))) { |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 314 | /* Wait for the response. In the response we expect 'ok' on success, |
| 315 | * or 'ko' on failure. */ |
bohu | 82a2834 | 2016-02-29 13:13:19 -0800 | [diff] [blame] | 316 | if (!ReadFdExactly(fd, tmp, 2) || memcmp(tmp, _ok_resp, 2)) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 317 | D("Accepting ADB host connection has failed."); |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 318 | adb_close(fd); |
| 319 | } else { |
| 320 | /* Host is connected. Register the transport, and start the |
| 321 | * exchange. */ |
Prathmesh Prabhu | 07778a0 | 2016-03-09 13:51:30 -0800 | [diff] [blame] | 322 | std::string serial = android::base::StringPrintf("host-%d", fd); |
Yabin Cui | c722178 | 2016-05-17 15:14:25 -0700 | [diff] [blame] | 323 | if (register_socket_transport(fd, serial.c_str(), port, 1) != 0 || |
| 324 | !WriteFdExactly(fd, _start_req, strlen(_start_req))) { |
bohu | 82a2834 | 2016-02-29 13:13:19 -0800 | [diff] [blame] | 325 | adb_close(fd); |
| 326 | } |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 327 | } |
| 328 | |
| 329 | /* Prepare for accepting of the next ADB host connection. */ |
| 330 | fd = qemu_pipe_open(con_name); |
| 331 | if (fd < 0) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 332 | D("adb service become unavailable."); |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 333 | return; |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 334 | } |
| 335 | } else { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 336 | D("Unable to send the '%s' request to ADB service.", _accept_req); |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 337 | return; |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 338 | } |
| 339 | } |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 340 | D("transport: qemu_socket_thread() exiting"); |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 341 | return; |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 342 | } |
Vladimir Chtchetkine | 4bdc776 | 2011-12-13 12:19:29 -0800 | [diff] [blame] | 343 | #endif // !ADB_HOST |
Vladimir Chtchetkine | 7c9339d | 2011-12-09 15:49:47 -0800 | [diff] [blame] | 344 | |
Mike Lockwood | 26b88e3 | 2009-08-24 15:58:40 -0700 | [diff] [blame] | 345 | void local_init(int port) |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 346 | { |
Josh Gao | 7d40525 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 347 | adb_thread_func_t func; |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 348 | const char* debug_name = ""; |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 349 | |
Vladimir Chtchetkine | 4bdc776 | 2011-12-13 12:19:29 -0800 | [diff] [blame] | 350 | #if ADB_HOST |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 351 | func = client_socket_thread; |
| 352 | debug_name = "client"; |
Vladimir Chtchetkine | 4bdc776 | 2011-12-13 12:19:29 -0800 | [diff] [blame] | 353 | #else |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 354 | /* For the adbd daemon in the system image we need to distinguish |
| 355 | * between the device, and the emulator. */ |
| 356 | char is_qemu[PROPERTY_VALUE_MAX]; |
| 357 | property_get("ro.kernel.qemu", is_qemu, ""); |
| 358 | if (!strcmp(is_qemu, "1")) { |
| 359 | /* Running inside the emulator: use QEMUD pipe as the transport. */ |
| 360 | func = qemu_socket_thread; |
| 361 | } else { |
| 362 | /* Running inside the device: use TCP socket as the transport. */ |
| 363 | func = server_socket_thread; |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 364 | } |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 365 | debug_name = "server"; |
| 366 | #endif // !ADB_HOST |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 367 | |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 368 | D("transport: local %s init", debug_name); |
Elliott Hughes | f251714 | 2015-05-05 13:41:21 -0700 | [diff] [blame] | 369 | if (!adb_thread_create(func, (void *) (uintptr_t) port)) { |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 370 | fatal_errno("cannot create local socket %s thread", debug_name); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 371 | } |
| 372 | } |
| 373 | |
| 374 | static void remote_kick(atransport *t) |
| 375 | { |
| 376 | int fd = t->sfd; |
| 377 | t->sfd = -1; |
Mike Lockwood | 81ffe17 | 2009-10-11 23:04:18 -0400 | [diff] [blame] | 378 | adb_shutdown(fd); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 379 | adb_close(fd); |
| 380 | |
| 381 | #if ADB_HOST |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 382 | int nn; |
| 383 | adb_mutex_lock( &local_transports_lock ); |
| 384 | for (nn = 0; nn < ADB_LOCAL_TRANSPORT_MAX; nn++) { |
| 385 | if (local_transports[nn] == t) { |
| 386 | local_transports[nn] = NULL; |
| 387 | break; |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 388 | } |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 389 | } |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 390 | adb_mutex_unlock( &local_transports_lock ); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 391 | #endif |
| 392 | } |
| 393 | |
| 394 | static void remote_close(atransport *t) |
| 395 | { |
Spencer Low | 6f38e23 | 2015-06-09 12:32:17 -0700 | [diff] [blame] | 396 | int fd = t->sfd; |
| 397 | if (fd != -1) { |
| 398 | t->sfd = -1; |
| 399 | adb_close(fd); |
| 400 | } |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 401 | #if ADB_HOST |
| 402 | int local_port; |
| 403 | if (t->GetLocalPortForEmulator(&local_port)) { |
| 404 | VLOG(TRANSPORT) << "remote_close, local_port = " << local_port; |
| 405 | std::unique_lock<std::mutex> lock(retry_ports_lock); |
| 406 | RetryPort port; |
| 407 | port.port = local_port; |
| 408 | port.retry_count = LOCAL_PORT_RETRY_COUNT; |
| 409 | retry_ports.push_back(port); |
| 410 | retry_ports_cond.notify_one(); |
| 411 | } |
| 412 | #endif |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 413 | } |
| 414 | |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 415 | |
| 416 | #if ADB_HOST |
| 417 | /* Only call this function if you already hold local_transports_lock. */ |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 418 | static atransport* find_emulator_transport_by_adb_port_locked(int adb_port) |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 419 | { |
| 420 | int i; |
| 421 | for (i = 0; i < ADB_LOCAL_TRANSPORT_MAX; i++) { |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 422 | int local_port; |
| 423 | if (local_transports[i] && local_transports[i]->GetLocalPortForEmulator(&local_port)) { |
| 424 | if (local_port == adb_port) { |
| 425 | return local_transports[i]; |
| 426 | } |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 427 | } |
| 428 | } |
| 429 | return NULL; |
| 430 | } |
| 431 | |
| 432 | atransport* find_emulator_transport_by_adb_port(int adb_port) |
| 433 | { |
| 434 | adb_mutex_lock( &local_transports_lock ); |
| 435 | atransport* result = find_emulator_transport_by_adb_port_locked(adb_port); |
| 436 | adb_mutex_unlock( &local_transports_lock ); |
| 437 | return result; |
| 438 | } |
| 439 | |
| 440 | /* Only call this function if you already hold local_transports_lock. */ |
| 441 | int get_available_local_transport_index_locked() |
| 442 | { |
| 443 | int i; |
| 444 | for (i = 0; i < ADB_LOCAL_TRANSPORT_MAX; i++) { |
| 445 | if (local_transports[i] == NULL) { |
| 446 | return i; |
| 447 | } |
| 448 | } |
| 449 | return -1; |
| 450 | } |
| 451 | |
| 452 | int get_available_local_transport_index() |
| 453 | { |
| 454 | adb_mutex_lock( &local_transports_lock ); |
| 455 | int result = get_available_local_transport_index_locked(); |
| 456 | adb_mutex_unlock( &local_transports_lock ); |
| 457 | return result; |
| 458 | } |
| 459 | #endif |
| 460 | |
| 461 | int init_socket_transport(atransport *t, int s, int adb_port, int local) |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 462 | { |
| 463 | int fail = 0; |
| 464 | |
Yabin Cui | f2a9f9b | 2016-04-18 11:22:34 -0700 | [diff] [blame] | 465 | t->SetKickFunction(remote_kick); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 466 | t->close = remote_close; |
| 467 | t->read_from_remote = remote_read; |
| 468 | t->write_to_remote = remote_write; |
| 469 | t->sfd = s; |
| 470 | t->sync_token = 1; |
Dan Albert | 9a50f4c | 2015-05-18 16:43:57 -0700 | [diff] [blame] | 471 | t->connection_state = kCsOffline; |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 472 | t->type = kTransportLocal; |
| 473 | |
| 474 | #if ADB_HOST |
Yabin Cui | 5fe6d0d | 2015-08-11 13:40:42 -0700 | [diff] [blame] | 475 | if (local) { |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 476 | adb_mutex_lock( &local_transports_lock ); |
| 477 | { |
Yabin Cui | f401ead | 2016-04-29 16:53:52 -0700 | [diff] [blame^] | 478 | t->SetLocalPortForEmulator(adb_port); |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 479 | atransport* existing_transport = |
| 480 | find_emulator_transport_by_adb_port_locked(adb_port); |
| 481 | int index = get_available_local_transport_index_locked(); |
| 482 | if (existing_transport != NULL) { |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 483 | D("local transport for port %d already registered (%p)?", |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 484 | adb_port, existing_transport); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 485 | fail = -1; |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 486 | } else if (index < 0) { |
| 487 | // Too many emulators. |
Yabin Cui | 815ad88 | 2015-09-02 17:44:28 -0700 | [diff] [blame] | 488 | D("cannot register more emulators. Maximum is %d", |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 489 | ADB_LOCAL_TRANSPORT_MAX); |
| 490 | fail = -1; |
| 491 | } else { |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 492 | local_transports[index] = t; |
Stefan Hilzinger | 1ec0342 | 2010-04-26 10:17:43 +0100 | [diff] [blame] | 493 | } |
| 494 | } |
| 495 | adb_mutex_unlock( &local_transports_lock ); |
The Android Open Source Project | 9ca14dc | 2009-03-03 19:32:55 -0800 | [diff] [blame] | 496 | } |
| 497 | #endif |
| 498 | return fail; |
| 499 | } |