David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 1 | /* |
| 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 | |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 17 | // Functionality for launching and managing shell subprocesses. |
| 18 | // |
| 19 | // There are two types of subprocesses, PTY or raw. PTY is typically used for |
| 20 | // an interactive session, raw for non-interactive. There are also two methods |
| 21 | // of communication with the subprocess, passing raw data or using a simple |
| 22 | // protocol to wrap packets. The protocol allows separating stdout/stderr and |
| 23 | // passing the exit code back, but is not backwards compatible. |
| 24 | // ----------------+-------------------------------------- |
| 25 | // Type Protocol | Exit code? Separate stdout/stderr? |
| 26 | // ----------------+-------------------------------------- |
| 27 | // PTY No | No No |
| 28 | // Raw No | No No |
| 29 | // PTY Yes | Yes No |
| 30 | // Raw Yes | Yes Yes |
| 31 | // ----------------+-------------------------------------- |
| 32 | // |
| 33 | // Non-protocol subprocesses work by passing subprocess stdin/out/err through |
| 34 | // a single pipe which is registered with a local socket in adbd. The local |
| 35 | // socket uses the fdevent loop to pass raw data between this pipe and the |
| 36 | // transport, which then passes data back to the adb client. Cleanup is done by |
| 37 | // waiting in a separate thread for the subprocesses to exit and then signaling |
| 38 | // a separate fdevent to close out the local socket from the main loop. |
| 39 | // |
| 40 | // ------------------+-------------------------+------------------------------ |
| 41 | // Subprocess | adbd subprocess thread | adbd main fdevent loop |
| 42 | // ------------------+-------------------------+------------------------------ |
| 43 | // | | |
| 44 | // stdin/out/err <-----------------------------> LocalSocket |
| 45 | // | | | |
| 46 | // | | Block on exit | |
| 47 | // | | * | |
| 48 | // v | * | |
| 49 | // Exit ---> Unblock | |
| 50 | // | | | |
| 51 | // | v | |
| 52 | // | Notify shell exit FD ---> Close LocalSocket |
| 53 | // ------------------+-------------------------+------------------------------ |
| 54 | // |
| 55 | // The protocol requires the thread to intercept stdin/out/err in order to |
| 56 | // wrap/unwrap data with shell protocol packets. |
| 57 | // |
| 58 | // ------------------+-------------------------+------------------------------ |
| 59 | // Subprocess | adbd subprocess thread | adbd main fdevent loop |
| 60 | // ------------------+-------------------------+------------------------------ |
| 61 | // | | |
| 62 | // stdin/out <---> Protocol <---> LocalSocket |
| 63 | // stderr ---> Protocol ---> LocalSocket |
| 64 | // | | | |
| 65 | // v | | |
| 66 | // Exit ---> Exit code protocol ---> LocalSocket |
| 67 | // | | | |
| 68 | // | v | |
| 69 | // | Notify shell exit FD ---> Close LocalSocket |
| 70 | // ------------------+-------------------------+------------------------------ |
| 71 | // |
| 72 | // An alternate approach is to put the protocol wrapping/unwrapping in the main |
| 73 | // fdevent loop, which has the advantage of being able to re-use the existing |
| 74 | // select() code for handling data streams. However, implementation turned out |
| 75 | // to be more complex due to partial reads and non-blocking I/O so this model |
| 76 | // was chosen instead. |
| 77 | |
Yabin Cui | aed3c61 | 2015-09-22 15:52:57 -0700 | [diff] [blame] | 78 | #define TRACE_TAG SHELL |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 79 | |
Yabin Cui | 6dfef25 | 2015-10-06 15:10:05 -0700 | [diff] [blame] | 80 | #include "sysdeps.h" |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 81 | |
Yabin Cui | 6dfef25 | 2015-10-06 15:10:05 -0700 | [diff] [blame] | 82 | #include "shell_service.h" |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 83 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 84 | #include <errno.h> |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 85 | #include <pty.h> |
Elliott Hughes | fbe4332 | 2015-11-02 13:29:19 -0800 | [diff] [blame] | 86 | #include <pwd.h> |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 87 | #include <sys/select.h> |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 88 | #include <termios.h> |
| 89 | |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 90 | #include <memory> |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 91 | #include <string> |
| 92 | #include <unordered_map> |
| 93 | #include <vector> |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 94 | |
Elliott Hughes | 4f71319 | 2015-12-04 22:00:26 -0800 | [diff] [blame] | 95 | #include <android-base/logging.h> |
| 96 | #include <android-base/stringprintf.h> |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 97 | #include <paths.h> |
Rubin Xu | d61a25c | 2016-01-11 10:23:47 +0000 | [diff] [blame] | 98 | #include <log/log.h> |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 99 | |
| 100 | #include "adb.h" |
| 101 | #include "adb_io.h" |
| 102 | #include "adb_trace.h" |
Yabin Cui | 6dfef25 | 2015-10-06 15:10:05 -0700 | [diff] [blame] | 103 | #include "adb_utils.h" |
Rubin Xu | d61a25c | 2016-01-11 10:23:47 +0000 | [diff] [blame] | 104 | #include "security_log_tags.h" |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 105 | |
| 106 | namespace { |
| 107 | |
| 108 | void init_subproc_child() |
| 109 | { |
| 110 | setsid(); |
| 111 | |
| 112 | // Set OOM score adjustment to prevent killing |
| 113 | int fd = adb_open("/proc/self/oom_score_adj", O_WRONLY | O_CLOEXEC); |
| 114 | if (fd >= 0) { |
| 115 | adb_write(fd, "0", 1); |
| 116 | adb_close(fd); |
| 117 | } else { |
| 118 | D("adb: unable to update oom_score_adj"); |
| 119 | } |
| 120 | } |
| 121 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 122 | // Reads from |fd| until close or failure. |
| 123 | std::string ReadAll(int fd) { |
| 124 | char buffer[512]; |
| 125 | std::string received; |
| 126 | |
| 127 | while (1) { |
| 128 | int bytes = adb_read(fd, buffer, sizeof(buffer)); |
| 129 | if (bytes <= 0) { |
| 130 | break; |
| 131 | } |
| 132 | received.append(buffer, bytes); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 133 | } |
| 134 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 135 | return received; |
| 136 | } |
| 137 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 138 | // Creates a socketpair and saves the endpoints to |fd1| and |fd2|. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 139 | bool CreateSocketpair(unique_fd* fd1, unique_fd* fd2) { |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 140 | int sockets[2]; |
| 141 | if (adb_socketpair(sockets) < 0) { |
| 142 | PLOG(ERROR) << "cannot create socket pair"; |
| 143 | return false; |
| 144 | } |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 145 | fd1->reset(sockets[0]); |
| 146 | fd2->reset(sockets[1]); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 147 | return true; |
| 148 | } |
| 149 | |
| 150 | class Subprocess { |
| 151 | public: |
Elliott Hughes | 18ddf5c | 2015-11-16 10:55:34 -0800 | [diff] [blame] | 152 | Subprocess(const std::string& command, const char* terminal_type, |
| 153 | SubprocessType type, SubprocessProtocol protocol); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 154 | ~Subprocess(); |
| 155 | |
| 156 | const std::string& command() const { return command_; } |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 157 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 158 | int local_socket_fd() const { return local_socket_sfd_; } |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 159 | |
| 160 | pid_t pid() const { return pid_; } |
| 161 | |
| 162 | // Sets up FDs, forks a subprocess, starts the subprocess manager thread, |
| 163 | // and exec's the child. Returns false on failure. |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 164 | bool ForkAndExec(std::string* _Nonnull error); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 165 | |
| 166 | private: |
| 167 | // Opens the file at |pts_name|. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 168 | int OpenPtyChildFd(const char* pts_name, unique_fd* error_sfd); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 169 | |
Josh Gao | b5fea14 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 170 | static void ThreadHandler(void* userdata); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 171 | void PassDataStreams(); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 172 | void WaitForExit(); |
| 173 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 174 | unique_fd* SelectLoop(fd_set* master_read_set_ptr, |
| 175 | fd_set* master_write_set_ptr); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 176 | |
| 177 | // Input/output stream handlers. Success returns nullptr, failure returns |
| 178 | // a pointer to the failed FD. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 179 | unique_fd* PassInput(); |
| 180 | unique_fd* PassOutput(unique_fd* sfd, ShellProtocol::Id id); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 181 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 182 | const std::string command_; |
Elliott Hughes | 18ddf5c | 2015-11-16 10:55:34 -0800 | [diff] [blame] | 183 | const std::string terminal_type_; |
David Pursell | 57dd5ae | 2016-01-27 16:07:52 -0800 | [diff] [blame] | 184 | bool make_pty_raw_ = false; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 185 | SubprocessType type_; |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 186 | SubprocessProtocol protocol_; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 187 | pid_t pid_ = -1; |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 188 | unique_fd local_socket_sfd_; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 189 | |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 190 | // Shell protocol variables. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 191 | unique_fd stdinout_sfd_, stderr_sfd_, protocol_sfd_; |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 192 | std::unique_ptr<ShellProtocol> input_, output_; |
| 193 | size_t input_bytes_left_ = 0; |
| 194 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 195 | DISALLOW_COPY_AND_ASSIGN(Subprocess); |
| 196 | }; |
| 197 | |
Elliott Hughes | 18ddf5c | 2015-11-16 10:55:34 -0800 | [diff] [blame] | 198 | Subprocess::Subprocess(const std::string& command, const char* terminal_type, |
| 199 | SubprocessType type, SubprocessProtocol protocol) |
| 200 | : command_(command), |
| 201 | terminal_type_(terminal_type ? terminal_type : ""), |
| 202 | type_(type), |
| 203 | protocol_(protocol) { |
David Pursell | 57dd5ae | 2016-01-27 16:07:52 -0800 | [diff] [blame] | 204 | // If we aren't using the shell protocol we must allocate a PTY to properly close the |
| 205 | // subprocess. PTYs automatically send SIGHUP to the slave-side process when the master side |
| 206 | // of the PTY closes, which we rely on. If we use a raw pipe, processes that don't read/write, |
| 207 | // e.g. screenrecord, will never notice the broken pipe and terminate. |
| 208 | // The shell protocol doesn't require a PTY because it's always monitoring the local socket FD |
| 209 | // with select() and will send SIGHUP manually to the child process. |
| 210 | if (protocol_ == SubprocessProtocol::kNone && type_ == SubprocessType::kRaw) { |
| 211 | // Disable PTY input/output processing since the client is expecting raw data. |
| 212 | D("Can't create raw subprocess without shell protocol, using PTY in raw mode instead"); |
| 213 | type_ = SubprocessType::kPty; |
| 214 | make_pty_raw_ = true; |
| 215 | } |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 216 | } |
| 217 | |
| 218 | Subprocess::~Subprocess() { |
Josh Gao | c65fae9 | 2016-01-19 16:21:17 -0800 | [diff] [blame] | 219 | WaitForExit(); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 220 | } |
| 221 | |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 222 | bool Subprocess::ForkAndExec(std::string* error) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 223 | unique_fd child_stdinout_sfd, child_stderr_sfd; |
| 224 | unique_fd parent_error_sfd, child_error_sfd; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 225 | char pts_name[PATH_MAX]; |
| 226 | |
Rubin Xu | d61a25c | 2016-01-11 10:23:47 +0000 | [diff] [blame] | 227 | if (command_.empty()) { |
| 228 | __android_log_security_bswrite(SEC_TAG_ADB_SHELL_INTERACTIVE, ""); |
| 229 | } else { |
| 230 | __android_log_security_bswrite(SEC_TAG_ADB_SHELL_CMD, command_.c_str()); |
| 231 | } |
| 232 | |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 233 | // Create a socketpair for the fork() child to report any errors back to the parent. Since we |
| 234 | // use threads, logging directly from the child might deadlock due to locks held in another |
| 235 | // thread during the fork. |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 236 | if (!CreateSocketpair(&parent_error_sfd, &child_error_sfd)) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 237 | *error = android::base::StringPrintf( |
| 238 | "failed to create pipe for subprocess error reporting: %s", strerror(errno)); |
| 239 | return false; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 240 | } |
| 241 | |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 242 | // Construct the environment for the child before we fork. |
| 243 | passwd* pw = getpwuid(getuid()); |
| 244 | std::unordered_map<std::string, std::string> env; |
Josh Gao | e03c988 | 2015-12-11 15:49:12 -0800 | [diff] [blame] | 245 | if (environ) { |
| 246 | char** current = environ; |
| 247 | while (char* env_cstr = *current++) { |
| 248 | std::string env_string = env_cstr; |
Dan Austin | 80fd454 | 2016-03-28 14:37:01 -0700 | [diff] [blame] | 249 | char* delimiter = strchr(&env_string[0], '='); |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 250 | |
Josh Gao | e03c988 | 2015-12-11 15:49:12 -0800 | [diff] [blame] | 251 | // Drop any values that don't contain '='. |
| 252 | if (delimiter) { |
| 253 | *delimiter++ = '\0'; |
| 254 | env[env_string.c_str()] = delimiter; |
| 255 | } |
| 256 | } |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 257 | } |
| 258 | |
| 259 | if (pw != nullptr) { |
| 260 | // TODO: $HOSTNAME? Normally bash automatically sets that, but mksh doesn't. |
| 261 | env["HOME"] = pw->pw_dir; |
| 262 | env["LOGNAME"] = pw->pw_name; |
| 263 | env["USER"] = pw->pw_name; |
| 264 | env["SHELL"] = pw->pw_shell; |
| 265 | } |
| 266 | |
| 267 | if (!terminal_type_.empty()) { |
| 268 | env["TERM"] = terminal_type_; |
| 269 | } |
| 270 | |
| 271 | std::vector<std::string> joined_env; |
| 272 | for (auto it : env) { |
| 273 | const char* key = it.first.c_str(); |
| 274 | const char* value = it.second.c_str(); |
| 275 | joined_env.push_back(android::base::StringPrintf("%s=%s", key, value)); |
| 276 | } |
| 277 | |
| 278 | std::vector<const char*> cenv; |
| 279 | for (const std::string& str : joined_env) { |
| 280 | cenv.push_back(str.c_str()); |
| 281 | } |
| 282 | cenv.push_back(nullptr); |
| 283 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 284 | if (type_ == SubprocessType::kPty) { |
| 285 | int fd; |
| 286 | pid_ = forkpty(&fd, pts_name, nullptr, nullptr); |
Josh Gao | fcb063c | 2016-03-04 17:50:10 -0800 | [diff] [blame] | 287 | if (pid_ > 0) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 288 | stdinout_sfd_.reset(fd); |
Josh Gao | fcb063c | 2016-03-04 17:50:10 -0800 | [diff] [blame] | 289 | } |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 290 | } else { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 291 | if (!CreateSocketpair(&stdinout_sfd_, &child_stdinout_sfd)) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 292 | *error = android::base::StringPrintf("failed to create socketpair for stdin/out: %s", |
| 293 | strerror(errno)); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 294 | return false; |
| 295 | } |
| 296 | // Raw subprocess + shell protocol allows for splitting stderr. |
| 297 | if (protocol_ == SubprocessProtocol::kShell && |
| 298 | !CreateSocketpair(&stderr_sfd_, &child_stderr_sfd)) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 299 | *error = android::base::StringPrintf("failed to create socketpair for stderr: %s", |
| 300 | strerror(errno)); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 301 | return false; |
| 302 | } |
| 303 | pid_ = fork(); |
| 304 | } |
| 305 | |
| 306 | if (pid_ == -1) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 307 | *error = android::base::StringPrintf("fork failed: %s", strerror(errno)); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 308 | return false; |
| 309 | } |
| 310 | |
| 311 | if (pid_ == 0) { |
| 312 | // Subprocess child. |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 313 | init_subproc_child(); |
| 314 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 315 | if (type_ == SubprocessType::kPty) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 316 | child_stdinout_sfd.reset(OpenPtyChildFd(pts_name, &child_error_sfd)); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 317 | } |
| 318 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 319 | dup2(child_stdinout_sfd, STDIN_FILENO); |
| 320 | dup2(child_stdinout_sfd, STDOUT_FILENO); |
| 321 | dup2(child_stderr_sfd != -1 ? child_stderr_sfd : child_stdinout_sfd, STDERR_FILENO); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 322 | |
| 323 | // exec doesn't trigger destructors, close the FDs manually. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 324 | stdinout_sfd_.reset(-1); |
| 325 | stderr_sfd_.reset(-1); |
| 326 | child_stdinout_sfd.reset(-1); |
| 327 | child_stderr_sfd.reset(-1); |
| 328 | parent_error_sfd.reset(-1); |
| 329 | close_on_exec(child_error_sfd); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 330 | |
Josh Gao | b5028e4 | 2016-01-19 17:31:09 -0800 | [diff] [blame] | 331 | if (command_.empty()) { |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 332 | execle(_PATH_BSHELL, _PATH_BSHELL, "-", nullptr, cenv.data()); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 333 | } else { |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 334 | execle(_PATH_BSHELL, _PATH_BSHELL, "-c", command_.c_str(), nullptr, cenv.data()); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 335 | } |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 336 | WriteFdExactly(child_error_sfd, "exec '" _PATH_BSHELL "' failed: "); |
| 337 | WriteFdExactly(child_error_sfd, strerror(errno)); |
| 338 | child_error_sfd.reset(-1); |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 339 | _Exit(1); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 340 | } |
| 341 | |
| 342 | // Subprocess parent. |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 343 | D("subprocess parent: stdin/stdout FD = %d, stderr FD = %d", |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 344 | stdinout_sfd_.get(), stderr_sfd_.get()); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 345 | |
| 346 | // Wait to make sure the subprocess exec'd without error. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 347 | child_error_sfd.reset(-1); |
| 348 | std::string error_message = ReadAll(parent_error_sfd); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 349 | if (!error_message.empty()) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 350 | *error = error_message; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 351 | return false; |
| 352 | } |
| 353 | |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 354 | D("subprocess parent: exec completed"); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 355 | if (protocol_ == SubprocessProtocol::kNone) { |
| 356 | // No protocol: all streams pass through the stdinout FD and hook |
| 357 | // directly into the local socket for raw data transfer. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 358 | local_socket_sfd_.reset(stdinout_sfd_.release()); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 359 | } else { |
| 360 | // Shell protocol: create another socketpair to intercept data. |
| 361 | if (!CreateSocketpair(&protocol_sfd_, &local_socket_sfd_)) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 362 | *error = android::base::StringPrintf( |
| 363 | "failed to create socketpair to intercept data: %s", strerror(errno)); |
| 364 | kill(pid_, SIGKILL); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 365 | return false; |
| 366 | } |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 367 | D("protocol FD = %d", protocol_sfd_.get()); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 368 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 369 | input_.reset(new ShellProtocol(protocol_sfd_)); |
| 370 | output_.reset(new ShellProtocol(protocol_sfd_)); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 371 | if (!input_ || !output_) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 372 | *error = "failed to allocate shell protocol objects"; |
| 373 | kill(pid_, SIGKILL); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 374 | return false; |
| 375 | } |
| 376 | |
| 377 | // Don't let reads/writes to the subprocess block our thread. This isn't |
| 378 | // likely but could happen under unusual circumstances, such as if we |
| 379 | // write a ton of data to stdin but the subprocess never reads it and |
| 380 | // the pipe fills up. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 381 | for (int fd : {stdinout_sfd_.get(), stderr_sfd_.get()}) { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 382 | if (fd >= 0) { |
Yabin Cui | 6dfef25 | 2015-10-06 15:10:05 -0700 | [diff] [blame] | 383 | if (!set_file_block_mode(fd, false)) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 384 | *error = android::base::StringPrintf( |
| 385 | "failed to set non-blocking mode for fd %d", fd); |
| 386 | kill(pid_, SIGKILL); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 387 | return false; |
| 388 | } |
| 389 | } |
| 390 | } |
| 391 | } |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 392 | |
| 393 | if (!adb_thread_create(ThreadHandler, this)) { |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 394 | *error = |
| 395 | android::base::StringPrintf("failed to create subprocess thread: %s", strerror(errno)); |
| 396 | kill(pid_, SIGKILL); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 397 | return false; |
| 398 | } |
| 399 | |
Josh Gao | 9b3fd67 | 2015-12-11 10:52:55 -0800 | [diff] [blame] | 400 | D("subprocess parent: completed"); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 401 | return true; |
| 402 | } |
| 403 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 404 | int Subprocess::OpenPtyChildFd(const char* pts_name, unique_fd* error_sfd) { |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 405 | int child_fd = adb_open(pts_name, O_RDWR | O_CLOEXEC); |
| 406 | if (child_fd == -1) { |
| 407 | // Don't use WriteFdFmt; since we're in the fork() child we don't want |
| 408 | // to allocate any heap memory to avoid race conditions. |
| 409 | const char* messages[] = {"child failed to open pseudo-term slave ", |
| 410 | pts_name, ": ", strerror(errno)}; |
| 411 | for (const char* message : messages) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 412 | WriteFdExactly(*error_sfd, message); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 413 | } |
Josh Gao | 4abdeee | 2016-05-13 18:16:43 -0700 | [diff] [blame] | 414 | abort(); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 415 | } |
| 416 | |
David Pursell | 57dd5ae | 2016-01-27 16:07:52 -0800 | [diff] [blame] | 417 | if (make_pty_raw_) { |
| 418 | termios tattr; |
| 419 | if (tcgetattr(child_fd, &tattr) == -1) { |
| 420 | int saved_errno = errno; |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 421 | WriteFdExactly(*error_sfd, "tcgetattr failed: "); |
| 422 | WriteFdExactly(*error_sfd, strerror(saved_errno)); |
Josh Gao | 4abdeee | 2016-05-13 18:16:43 -0700 | [diff] [blame] | 423 | abort(); |
David Pursell | 57dd5ae | 2016-01-27 16:07:52 -0800 | [diff] [blame] | 424 | } |
| 425 | |
| 426 | cfmakeraw(&tattr); |
| 427 | if (tcsetattr(child_fd, TCSADRAIN, &tattr) == -1) { |
| 428 | int saved_errno = errno; |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 429 | WriteFdExactly(*error_sfd, "tcsetattr failed: "); |
| 430 | WriteFdExactly(*error_sfd, strerror(saved_errno)); |
Josh Gao | 4abdeee | 2016-05-13 18:16:43 -0700 | [diff] [blame] | 431 | abort(); |
David Pursell | 57dd5ae | 2016-01-27 16:07:52 -0800 | [diff] [blame] | 432 | } |
| 433 | } |
| 434 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 435 | return child_fd; |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 436 | } |
| 437 | |
Josh Gao | b5fea14 | 2016-02-12 14:31:15 -0800 | [diff] [blame] | 438 | void Subprocess::ThreadHandler(void* userdata) { |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 439 | Subprocess* subprocess = reinterpret_cast<Subprocess*>(userdata); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 440 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 441 | adb_thread_setname(android::base::StringPrintf( |
| 442 | "shell srvc %d", subprocess->local_socket_fd())); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 443 | |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 444 | subprocess->PassDataStreams(); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 445 | |
David Pursell | 1ed57f0 | 2015-10-06 15:30:03 -0700 | [diff] [blame] | 446 | D("deleting Subprocess for PID %d", subprocess->pid()); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 447 | delete subprocess; |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 448 | } |
| 449 | |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 450 | void Subprocess::PassDataStreams() { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 451 | if (protocol_sfd_ == -1) { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 452 | return; |
| 453 | } |
| 454 | |
| 455 | // Start by trying to read from the protocol FD, stdout, and stderr. |
| 456 | fd_set master_read_set, master_write_set; |
| 457 | FD_ZERO(&master_read_set); |
| 458 | FD_ZERO(&master_write_set); |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 459 | for (unique_fd* sfd : {&protocol_sfd_, &stdinout_sfd_, &stderr_sfd_}) { |
| 460 | if (*sfd != -1) { |
| 461 | FD_SET(*sfd, &master_read_set); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 462 | } |
| 463 | } |
| 464 | |
| 465 | // Pass data until the protocol FD or both the subprocess pipes die, at |
| 466 | // which point we can't pass any more data. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 467 | while (protocol_sfd_ != -1 && (stdinout_sfd_ != -1 || stderr_sfd_ != -1)) { |
| 468 | unique_fd* dead_sfd = SelectLoop(&master_read_set, &master_write_set); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 469 | if (dead_sfd) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 470 | D("closing FD %d", dead_sfd->get()); |
| 471 | FD_CLR(*dead_sfd, &master_read_set); |
| 472 | FD_CLR(*dead_sfd, &master_write_set); |
David Pursell | 544e795 | 2015-09-14 15:36:26 -0700 | [diff] [blame] | 473 | if (dead_sfd == &protocol_sfd_) { |
| 474 | // Using SIGHUP is a decent general way to indicate that the |
| 475 | // controlling process is going away. If specific signals are |
| 476 | // needed (e.g. SIGINT), pass those through the shell protocol |
| 477 | // and only fall back on this for unexpected closures. |
| 478 | D("protocol FD died, sending SIGHUP to pid %d", pid_); |
| 479 | kill(pid_, SIGHUP); |
| 480 | } |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 481 | dead_sfd->reset(-1); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 482 | } |
| 483 | } |
| 484 | } |
| 485 | |
| 486 | namespace { |
| 487 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 488 | inline bool ValidAndInSet(const unique_fd& sfd, fd_set* set) { |
| 489 | return sfd != -1 && FD_ISSET(sfd, set); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 490 | } |
| 491 | |
| 492 | } // namespace |
| 493 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 494 | unique_fd* Subprocess::SelectLoop(fd_set* master_read_set_ptr, |
| 495 | fd_set* master_write_set_ptr) { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 496 | fd_set read_set, write_set; |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 497 | int select_n = std::max(std::max(protocol_sfd_, stdinout_sfd_), stderr_sfd_) + 1; |
| 498 | unique_fd* dead_sfd = nullptr; |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 499 | |
| 500 | // Keep calling select() and passing data until an FD closes/errors. |
| 501 | while (!dead_sfd) { |
| 502 | memcpy(&read_set, master_read_set_ptr, sizeof(read_set)); |
| 503 | memcpy(&write_set, master_write_set_ptr, sizeof(write_set)); |
| 504 | if (select(select_n, &read_set, &write_set, nullptr, nullptr) < 0) { |
| 505 | if (errno == EINTR) { |
| 506 | continue; |
| 507 | } else { |
| 508 | PLOG(ERROR) << "select failed, closing subprocess pipes"; |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 509 | stdinout_sfd_.reset(-1); |
| 510 | stderr_sfd_.reset(-1); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 511 | return nullptr; |
| 512 | } |
| 513 | } |
| 514 | |
| 515 | // Read stdout, write to protocol FD. |
| 516 | if (ValidAndInSet(stdinout_sfd_, &read_set)) { |
| 517 | dead_sfd = PassOutput(&stdinout_sfd_, ShellProtocol::kIdStdout); |
| 518 | } |
| 519 | |
| 520 | // Read stderr, write to protocol FD. |
| 521 | if (!dead_sfd && ValidAndInSet(stderr_sfd_, &read_set)) { |
| 522 | dead_sfd = PassOutput(&stderr_sfd_, ShellProtocol::kIdStderr); |
| 523 | } |
| 524 | |
| 525 | // Read protocol FD, write to stdin. |
| 526 | if (!dead_sfd && ValidAndInSet(protocol_sfd_, &read_set)) { |
| 527 | dead_sfd = PassInput(); |
| 528 | // If we didn't finish writing, block on stdin write. |
| 529 | if (input_bytes_left_) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 530 | FD_CLR(protocol_sfd_, master_read_set_ptr); |
| 531 | FD_SET(stdinout_sfd_, master_write_set_ptr); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 532 | } |
| 533 | } |
| 534 | |
| 535 | // Continue writing to stdin; only happens if a previous write blocked. |
| 536 | if (!dead_sfd && ValidAndInSet(stdinout_sfd_, &write_set)) { |
| 537 | dead_sfd = PassInput(); |
| 538 | // If we finished writing, go back to blocking on protocol read. |
| 539 | if (!input_bytes_left_) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 540 | FD_SET(protocol_sfd_, master_read_set_ptr); |
| 541 | FD_CLR(stdinout_sfd_, master_write_set_ptr); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 542 | } |
| 543 | } |
| 544 | } // while (!dead_sfd) |
| 545 | |
| 546 | return dead_sfd; |
| 547 | } |
| 548 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 549 | unique_fd* Subprocess::PassInput() { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 550 | // Only read a new packet if we've finished writing the last one. |
| 551 | if (!input_bytes_left_) { |
| 552 | if (!input_->Read()) { |
| 553 | // Read() uses ReadFdExactly() which sets errno to 0 on EOF. |
| 554 | if (errno != 0) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 555 | PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_; |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 556 | } |
| 557 | return &protocol_sfd_; |
| 558 | } |
| 559 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 560 | if (stdinout_sfd_ != -1) { |
David Pursell | 1ed57f0 | 2015-10-06 15:30:03 -0700 | [diff] [blame] | 561 | switch (input_->id()) { |
Elliott Hughes | c15b17f | 2015-11-03 11:18:40 -0800 | [diff] [blame] | 562 | case ShellProtocol::kIdWindowSizeChange: |
| 563 | int rows, cols, x_pixels, y_pixels; |
| 564 | if (sscanf(input_->data(), "%dx%d,%dx%d", |
| 565 | &rows, &cols, &x_pixels, &y_pixels) == 4) { |
| 566 | winsize ws; |
| 567 | ws.ws_row = rows; |
| 568 | ws.ws_col = cols; |
| 569 | ws.ws_xpixel = x_pixels; |
| 570 | ws.ws_ypixel = y_pixels; |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 571 | ioctl(stdinout_sfd_, TIOCSWINSZ, &ws); |
Elliott Hughes | c15b17f | 2015-11-03 11:18:40 -0800 | [diff] [blame] | 572 | } |
| 573 | break; |
David Pursell | 1ed57f0 | 2015-10-06 15:30:03 -0700 | [diff] [blame] | 574 | case ShellProtocol::kIdStdin: |
| 575 | input_bytes_left_ = input_->data_length(); |
| 576 | break; |
| 577 | case ShellProtocol::kIdCloseStdin: |
| 578 | if (type_ == SubprocessType::kRaw) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 579 | if (adb_shutdown(stdinout_sfd_, SHUT_WR) == 0) { |
David Pursell | 1ed57f0 | 2015-10-06 15:30:03 -0700 | [diff] [blame] | 580 | return nullptr; |
| 581 | } |
| 582 | PLOG(ERROR) << "failed to shutdown writes to FD " |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 583 | << stdinout_sfd_; |
David Pursell | 1ed57f0 | 2015-10-06 15:30:03 -0700 | [diff] [blame] | 584 | return &stdinout_sfd_; |
| 585 | } else { |
| 586 | // PTYs can't close just input, so rather than close the |
| 587 | // FD and risk losing subprocess output, leave it open. |
| 588 | // This only happens if the client starts a PTY shell |
| 589 | // non-interactively which is rare and unsupported. |
| 590 | // If necessary, the client can manually close the shell |
| 591 | // with `exit` or by killing the adb client process. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 592 | D("can't close input for PTY FD %d", stdinout_sfd_.get()); |
David Pursell | 1ed57f0 | 2015-10-06 15:30:03 -0700 | [diff] [blame] | 593 | } |
| 594 | break; |
| 595 | } |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 596 | } |
| 597 | } |
| 598 | |
| 599 | if (input_bytes_left_ > 0) { |
| 600 | int index = input_->data_length() - input_bytes_left_; |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 601 | int bytes = adb_write(stdinout_sfd_, input_->data() + index, input_bytes_left_); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 602 | if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) { |
| 603 | if (bytes < 0) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 604 | PLOG(ERROR) << "error reading stdin FD " << stdinout_sfd_; |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 605 | } |
| 606 | // stdin is done, mark this packet as finished and we'll just start |
| 607 | // dumping any further data received from the protocol FD. |
| 608 | input_bytes_left_ = 0; |
| 609 | return &stdinout_sfd_; |
| 610 | } else if (bytes > 0) { |
| 611 | input_bytes_left_ -= bytes; |
| 612 | } |
| 613 | } |
| 614 | |
| 615 | return nullptr; |
| 616 | } |
| 617 | |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 618 | unique_fd* Subprocess::PassOutput(unique_fd* sfd, ShellProtocol::Id id) { |
| 619 | int bytes = adb_read(*sfd, output_->data(), output_->data_capacity()); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 620 | if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) { |
David Pursell | 1ed57f0 | 2015-10-06 15:30:03 -0700 | [diff] [blame] | 621 | // read() returns EIO if a PTY closes; don't report this as an error, |
| 622 | // it just means the subprocess completed. |
| 623 | if (bytes < 0 && !(type_ == SubprocessType::kPty && errno == EIO)) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 624 | PLOG(ERROR) << "error reading output FD " << *sfd; |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 625 | } |
| 626 | return sfd; |
| 627 | } |
| 628 | |
| 629 | if (bytes > 0 && !output_->Write(id, bytes)) { |
| 630 | if (errno != 0) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 631 | PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_; |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 632 | } |
| 633 | return &protocol_sfd_; |
| 634 | } |
| 635 | |
| 636 | return nullptr; |
| 637 | } |
| 638 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 639 | void Subprocess::WaitForExit() { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 640 | int exit_code = 1; |
| 641 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 642 | D("waiting for pid %d", pid_); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 643 | while (true) { |
| 644 | int status; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 645 | if (pid_ == waitpid(pid_, &status, 0)) { |
| 646 | D("post waitpid (pid=%d) status=%04x", pid_, status); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 647 | if (WIFSIGNALED(status)) { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 648 | exit_code = 0x80 | WTERMSIG(status); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 649 | D("subprocess killed by signal %d", WTERMSIG(status)); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 650 | break; |
| 651 | } else if (!WIFEXITED(status)) { |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 652 | D("subprocess didn't exit"); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 653 | break; |
| 654 | } else if (WEXITSTATUS(status) >= 0) { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 655 | exit_code = WEXITSTATUS(status); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 656 | D("subprocess exit code = %d", WEXITSTATUS(status)); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 657 | break; |
| 658 | } |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 659 | } |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 660 | } |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 661 | |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 662 | // If we have an open protocol FD send an exit packet. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 663 | if (protocol_sfd_ != -1) { |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 664 | output_->data()[0] = exit_code; |
| 665 | if (output_->Write(ShellProtocol::kIdExit, 1)) { |
| 666 | D("wrote the exit code packet: %d", exit_code); |
| 667 | } else { |
| 668 | PLOG(ERROR) << "failed to write the exit code packet"; |
| 669 | } |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 670 | protocol_sfd_.reset(-1); |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 671 | } |
| 672 | |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 673 | // Pass the local socket FD to the shell cleanup fdevent. |
| 674 | if (SHELL_EXIT_NOTIFY_FD >= 0) { |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 675 | int fd = local_socket_sfd_; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 676 | if (WriteFdExactly(SHELL_EXIT_NOTIFY_FD, &fd, sizeof(fd))) { |
| 677 | D("passed fd %d to SHELL_EXIT_NOTIFY_FD (%d) for pid %d", |
| 678 | fd, SHELL_EXIT_NOTIFY_FD, pid_); |
| 679 | // The shell exit fdevent now owns the FD and will close it once |
| 680 | // the last bit of data flushes through. |
Elliott Hughes | 2ce86e5 | 2016-05-27 17:51:24 -0700 | [diff] [blame^] | 681 | static_cast<void>(local_socket_sfd_.release()); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 682 | } else { |
| 683 | PLOG(ERROR) << "failed to write fd " << fd |
| 684 | << " to SHELL_EXIT_NOTIFY_FD (" << SHELL_EXIT_NOTIFY_FD |
| 685 | << ") for pid " << pid_; |
| 686 | } |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 687 | } |
| 688 | } |
| 689 | |
| 690 | } // namespace |
| 691 | |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 692 | // Create a pipe containing the error. |
| 693 | static int ReportError(SubprocessProtocol protocol, const std::string& message) { |
| 694 | int pipefd[2]; |
| 695 | if (pipe(pipefd) != 0) { |
| 696 | LOG(ERROR) << "failed to create pipe to report error"; |
| 697 | return -1; |
| 698 | } |
| 699 | |
| 700 | std::string buf = android::base::StringPrintf("error: %s\n", message.c_str()); |
| 701 | if (protocol == SubprocessProtocol::kShell) { |
| 702 | ShellProtocol::Id id = ShellProtocol::kIdStderr; |
| 703 | uint32_t length = buf.length(); |
| 704 | WriteFdExactly(pipefd[1], &id, sizeof(id)); |
| 705 | WriteFdExactly(pipefd[1], &length, sizeof(length)); |
| 706 | } |
| 707 | |
| 708 | WriteFdExactly(pipefd[1], buf.data(), buf.length()); |
| 709 | |
| 710 | if (protocol == SubprocessProtocol::kShell) { |
| 711 | ShellProtocol::Id id = ShellProtocol::kIdExit; |
| 712 | uint32_t length = 1; |
| 713 | char exit_code = 126; |
| 714 | WriteFdExactly(pipefd[1], &id, sizeof(id)); |
| 715 | WriteFdExactly(pipefd[1], &length, sizeof(length)); |
| 716 | WriteFdExactly(pipefd[1], &exit_code, sizeof(exit_code)); |
| 717 | } |
| 718 | |
| 719 | adb_close(pipefd[1]); |
| 720 | return pipefd[0]; |
| 721 | } |
| 722 | |
Elliott Hughes | 18ddf5c | 2015-11-16 10:55:34 -0800 | [diff] [blame] | 723 | int StartSubprocess(const char* name, const char* terminal_type, |
| 724 | SubprocessType type, SubprocessProtocol protocol) { |
| 725 | D("starting %s subprocess (protocol=%s, TERM=%s): '%s'", |
David Pursell | 0955c66 | 2015-08-31 10:42:13 -0700 | [diff] [blame] | 726 | type == SubprocessType::kRaw ? "raw" : "PTY", |
Elliott Hughes | 18ddf5c | 2015-11-16 10:55:34 -0800 | [diff] [blame] | 727 | protocol == SubprocessProtocol::kNone ? "none" : "shell", |
| 728 | terminal_type, name); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 729 | |
Elliott Hughes | 18ddf5c | 2015-11-16 10:55:34 -0800 | [diff] [blame] | 730 | Subprocess* subprocess = new Subprocess(name, terminal_type, type, protocol); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 731 | if (!subprocess) { |
| 732 | LOG(ERROR) << "failed to allocate new subprocess"; |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 733 | return ReportError(protocol, "failed to allocate new subprocess"); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 734 | } |
| 735 | |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 736 | std::string error; |
| 737 | if (!subprocess->ForkAndExec(&error)) { |
| 738 | LOG(ERROR) << "failed to start subprocess: " << error; |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 739 | delete subprocess; |
Josh Gao | 4323507 | 2016-01-25 17:11:43 -0800 | [diff] [blame] | 740 | return ReportError(protocol, error); |
David Pursell | a932058 | 2015-08-28 18:31:29 -0700 | [diff] [blame] | 741 | } |
| 742 | |
| 743 | D("subprocess creation successful: local_socket_fd=%d, pid=%d", |
| 744 | subprocess->local_socket_fd(), subprocess->pid()); |
| 745 | return subprocess->local_socket_fd(); |
David Pursell | 80f6702 | 2015-08-28 15:08:49 -0700 | [diff] [blame] | 746 | } |