blob: 9563eacb3368a51e5be8494534ccd40f394ce0a5 [file] [log] [blame]
Spencer Low7dc759f2015-05-06 16:13:42 -07001/*
2 * Copyright (C) 2015 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
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080017#include "sysdeps.h"
Spencer Low7dc759f2015-05-06 16:13:42 -070018
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080019#include <stdio.h>
20
Elliott Hughesf55ead92015-12-04 22:00:26 -080021#include <android-base/file.h>
22#include <android-base/logging.h>
23#include <android-base/strings.h>
Elliott Hughes43df1092015-07-23 17:12:58 -070024#include <cutils/sockets.h>
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080025
Spencer Low7dc759f2015-05-06 16:13:42 -070026#include "adb.h"
27#include "adb_client.h"
Spencer Lowcc4a4b12015-10-14 17:32:44 -070028#include "adb_io.h"
Yurii Zubrytskyic0889ab2016-05-25 15:17:10 -070029#include "adb_utils.h"
30
31// Return the console authentication command for the emulator, if needed
32static std::string adb_construct_auth_command() {
33 static const char auth_token_filename[] = ".emulator_console_auth_token";
34
Josh Gao62ff9d42016-08-30 15:23:35 -070035 std::string auth_token_path = adb_get_homedir_path();
Yurii Zubrytskyic0889ab2016-05-25 15:17:10 -070036 auth_token_path += OS_PATH_SEPARATOR;
37 auth_token_path += auth_token_filename;
38
39 // read the token
40 std::string token;
41 if (!android::base::ReadFileToString(auth_token_path, &token)
42 || token.empty()) {
43 // we either can't read the file, or it doesn't exist, or it's empty -
44 // either way we won't add any authentication command.
45 return {};
46 }
47
48 // now construct and return the actual command: "auth <token>\n"
49 std::string command = "auth ";
50 command += token;
51 command += '\n';
52 return command;
53}
Spencer Low7dc759f2015-05-06 16:13:42 -070054
55// Return the console port of the currently connected emulator (if any) or -1 if
56// there is no emulator, and -2 if there is more than one.
57static int adb_get_emulator_console_port(const char* serial) {
58 if (serial) {
59 // The user specified a serial number; is it an emulator?
60 int port;
61 return (sscanf(serial, "emulator-%d", &port) == 1) ? port : -1;
62 }
63
64 // No specific device was given, so get the list of connected devices and
65 // search for emulators. If there's one, we'll take it. If there are more
66 // than one, that's an error.
67 std::string devices;
68 std::string error;
69 if (!adb_query("host:devices", &devices, &error)) {
70 fprintf(stderr, "error: no emulator connected: %s\n", error.c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -080071 return -1;
72 }
Spencer Low7dc759f2015-05-06 16:13:42 -070073
74 int port;
75 size_t emulator_count = 0;
76 for (const auto& device : android::base::Split(devices, "\n")) {
77 if (sscanf(device.c_str(), "emulator-%d", &port) == 1) {
78 if (++emulator_count > 1) {
79 fprintf(
80 stderr, "error: more than one emulator detected; use -s\n");
81 return -1;
82 }
83 }
84 }
85
86 if (emulator_count == 0) {
87 fprintf(stderr, "error: no emulator detected\n");
88 return -1;
89 }
90
91 return port;
92}
93
94static int connect_to_console(const char* serial) {
95 int port = adb_get_emulator_console_port(serial);
96 if (port == -1) {
97 return -1;
98 }
99
Spencer Low753d4852015-07-30 23:07:55 -0700100 std::string error;
101 int fd = network_loopback_client(port, SOCK_STREAM, &error);
Spencer Low7dc759f2015-05-06 16:13:42 -0700102 if (fd == -1) {
Spencer Low753d4852015-07-30 23:07:55 -0700103 fprintf(stderr, "error: could not connect to TCP port %d: %s\n", port,
104 error.c_str());
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800105 return -1;
106 }
Spencer Low7dc759f2015-05-06 16:13:42 -0700107 return fd;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800108}
109
Spencer Low7dc759f2015-05-06 16:13:42 -0700110int adb_send_emulator_command(int argc, const char** argv, const char* serial) {
111 int fd = connect_to_console(serial);
112 if (fd == -1) {
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800113 return 1;
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800114 }
Spencer Low7dc759f2015-05-06 16:13:42 -0700115
Yurii Zubrytskyic0889ab2016-05-25 15:17:10 -0700116 std::string commands = adb_construct_auth_command();
Spencer Lowcc4a4b12015-10-14 17:32:44 -0700117
Spencer Low7dc759f2015-05-06 16:13:42 -0700118 for (int i = 1; i < argc; i++) {
Spencer Lowcc4a4b12015-10-14 17:32:44 -0700119 commands.append(argv[i]);
Yurii Zubrytskyic0889ab2016-05-25 15:17:10 -0700120 commands.push_back(i == argc - 1 ? '\n' : ' ');
Spencer Low7dc759f2015-05-06 16:13:42 -0700121 }
122
Spencer Lowcc4a4b12015-10-14 17:32:44 -0700123 commands.append("quit\n");
124
125 if (!WriteFdExactly(fd, commands)) {
126 fprintf(stderr, "error: cannot write to emulator: %s\n",
127 strerror(errno));
128 adb_close(fd);
129 return 1;
Spencer Low7dc759f2015-05-06 16:13:42 -0700130 }
131
132 // Drain output that the emulator console has sent us to prevent a problem
133 // on Windows where if adb closes the socket without reading all the data,
134 // the emulator's next call to recv() will have an ECONNABORTED error,
135 // preventing the emulator from reading the command that adb has sent.
136 // https://code.google.com/p/android/issues/detail?id=21021
137 int result;
138 do {
139 char buf[BUFSIZ];
140 result = adb_read(fd, buf, sizeof(buf));
Spencer Lowcc4a4b12015-10-14 17:32:44 -0700141 // Keep reading until zero bytes (orderly/graceful shutdown) or an
142 // error. If 'adb emu kill' is executed, the emulator calls exit() with
143 // the socket open (and shutdown(SD_SEND) was not called), which causes
144 // Windows to send a TCP RST segment which causes adb to get ECONNRESET.
145 // Any other emu command is followed by the quit command that we
146 // appended above, and that causes the emulator to close the socket
147 // which should cause zero bytes (orderly/graceful shutdown) to be
148 // returned.
Spencer Low7dc759f2015-05-06 16:13:42 -0700149 } while (result > 0);
150
The Android Open Source Project9ca14dc2009-03-03 19:32:55 -0800151 adb_close(fd);
152
153 return 0;
154}