blob: e8dad58e7cb86ccbb00d6fe98b3c77a786262a12 [file] [log] [blame]
David Pursell80f67022015-08-28 15:08:49 -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
David Pursell0955c662015-08-31 10:42:13 -070017// 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 Cuiaed3c612015-09-22 15:52:57 -070078#define TRACE_TAG SHELL
David Pursell80f67022015-08-28 15:08:49 -070079
Yabin Cui6dfef252015-10-06 15:10:05 -070080#include "sysdeps.h"
David Pursell80f67022015-08-28 15:08:49 -070081
Yabin Cui6dfef252015-10-06 15:10:05 -070082#include "shell_service.h"
David Pursell80f67022015-08-28 15:08:49 -070083
David Pursella9320582015-08-28 18:31:29 -070084#include <errno.h>
David Pursell80f67022015-08-28 15:08:49 -070085#include <pty.h>
Elliott Hughesfbe43322015-11-02 13:29:19 -080086#include <pwd.h>
David Pursell0955c662015-08-31 10:42:13 -070087#include <sys/select.h>
David Pursell80f67022015-08-28 15:08:49 -070088#include <termios.h>
89
David Pursell0955c662015-08-31 10:42:13 -070090#include <memory>
Josh Gao9b3fd672015-12-11 10:52:55 -080091#include <string>
92#include <unordered_map>
93#include <vector>
David Pursell0955c662015-08-31 10:42:13 -070094
Elliott Hughes4f713192015-12-04 22:00:26 -080095#include <android-base/logging.h>
96#include <android-base/stringprintf.h>
David Pursell80f67022015-08-28 15:08:49 -070097#include <paths.h>
Rubin Xud61a25c2016-01-11 10:23:47 +000098#include <log/log.h>
David Pursell80f67022015-08-28 15:08:49 -070099
100#include "adb.h"
101#include "adb_io.h"
102#include "adb_trace.h"
Yabin Cui6dfef252015-10-06 15:10:05 -0700103#include "adb_utils.h"
Rubin Xud61a25c2016-01-11 10:23:47 +0000104#include "security_log_tags.h"
David Pursell80f67022015-08-28 15:08:49 -0700105
106namespace {
107
108void 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 Pursella9320582015-08-28 18:31:29 -0700122// Reads from |fd| until close or failure.
123std::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 Pursell80f67022015-08-28 15:08:49 -0700133 }
134
David Pursella9320582015-08-28 18:31:29 -0700135 return received;
136}
137
David Pursella9320582015-08-28 18:31:29 -0700138// Creates a socketpair and saves the endpoints to |fd1| and |fd2|.
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700139bool CreateSocketpair(unique_fd* fd1, unique_fd* fd2) {
David Pursella9320582015-08-28 18:31:29 -0700140 int sockets[2];
141 if (adb_socketpair(sockets) < 0) {
142 PLOG(ERROR) << "cannot create socket pair";
143 return false;
144 }
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700145 fd1->reset(sockets[0]);
146 fd2->reset(sockets[1]);
David Pursella9320582015-08-28 18:31:29 -0700147 return true;
148}
149
150class Subprocess {
151 public:
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800152 Subprocess(const std::string& command, const char* terminal_type,
153 SubprocessType type, SubprocessProtocol protocol);
David Pursella9320582015-08-28 18:31:29 -0700154 ~Subprocess();
155
156 const std::string& command() const { return command_; }
David Pursella9320582015-08-28 18:31:29 -0700157
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700158 int local_socket_fd() const { return local_socket_sfd_; }
David Pursella9320582015-08-28 18:31:29 -0700159
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 Gao43235072016-01-25 17:11:43 -0800164 bool ForkAndExec(std::string* _Nonnull error);
David Pursella9320582015-08-28 18:31:29 -0700165
166 private:
167 // Opens the file at |pts_name|.
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700168 int OpenPtyChildFd(const char* pts_name, unique_fd* error_sfd);
David Pursella9320582015-08-28 18:31:29 -0700169
Josh Gaob5fea142016-02-12 14:31:15 -0800170 static void ThreadHandler(void* userdata);
David Pursell0955c662015-08-31 10:42:13 -0700171 void PassDataStreams();
David Pursella9320582015-08-28 18:31:29 -0700172 void WaitForExit();
173
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700174 unique_fd* SelectLoop(fd_set* master_read_set_ptr,
175 fd_set* master_write_set_ptr);
David Pursell0955c662015-08-31 10:42:13 -0700176
177 // Input/output stream handlers. Success returns nullptr, failure returns
178 // a pointer to the failed FD.
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700179 unique_fd* PassInput();
180 unique_fd* PassOutput(unique_fd* sfd, ShellProtocol::Id id);
David Pursell0955c662015-08-31 10:42:13 -0700181
David Pursella9320582015-08-28 18:31:29 -0700182 const std::string command_;
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800183 const std::string terminal_type_;
David Pursell57dd5ae2016-01-27 16:07:52 -0800184 bool make_pty_raw_ = false;
David Pursella9320582015-08-28 18:31:29 -0700185 SubprocessType type_;
David Pursell0955c662015-08-31 10:42:13 -0700186 SubprocessProtocol protocol_;
David Pursella9320582015-08-28 18:31:29 -0700187 pid_t pid_ = -1;
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700188 unique_fd local_socket_sfd_;
David Pursella9320582015-08-28 18:31:29 -0700189
David Pursell0955c662015-08-31 10:42:13 -0700190 // Shell protocol variables.
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700191 unique_fd stdinout_sfd_, stderr_sfd_, protocol_sfd_;
David Pursell0955c662015-08-31 10:42:13 -0700192 std::unique_ptr<ShellProtocol> input_, output_;
193 size_t input_bytes_left_ = 0;
194
David Pursella9320582015-08-28 18:31:29 -0700195 DISALLOW_COPY_AND_ASSIGN(Subprocess);
196};
197
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800198Subprocess::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 Pursell57dd5ae2016-01-27 16:07:52 -0800204 // 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 Pursella9320582015-08-28 18:31:29 -0700216}
217
218Subprocess::~Subprocess() {
Josh Gaoc65fae92016-01-19 16:21:17 -0800219 WaitForExit();
David Pursella9320582015-08-28 18:31:29 -0700220}
221
Josh Gao43235072016-01-25 17:11:43 -0800222bool Subprocess::ForkAndExec(std::string* error) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700223 unique_fd child_stdinout_sfd, child_stderr_sfd;
224 unique_fd parent_error_sfd, child_error_sfd;
David Pursella9320582015-08-28 18:31:29 -0700225 char pts_name[PATH_MAX];
226
Rubin Xud61a25c2016-01-11 10:23:47 +0000227 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 Gao9b3fd672015-12-11 10:52:55 -0800233 // 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 Pursella9320582015-08-28 18:31:29 -0700236 if (!CreateSocketpair(&parent_error_sfd, &child_error_sfd)) {
Josh Gao43235072016-01-25 17:11:43 -0800237 *error = android::base::StringPrintf(
238 "failed to create pipe for subprocess error reporting: %s", strerror(errno));
239 return false;
David Pursella9320582015-08-28 18:31:29 -0700240 }
241
Josh Gao9b3fd672015-12-11 10:52:55 -0800242 // Construct the environment for the child before we fork.
243 passwd* pw = getpwuid(getuid());
244 std::unordered_map<std::string, std::string> env;
Josh Gaoe03c9882015-12-11 15:49:12 -0800245 if (environ) {
246 char** current = environ;
247 while (char* env_cstr = *current++) {
248 std::string env_string = env_cstr;
Dan Austin80fd4542016-03-28 14:37:01 -0700249 char* delimiter = strchr(&env_string[0], '=');
Josh Gao9b3fd672015-12-11 10:52:55 -0800250
Josh Gaoe03c9882015-12-11 15:49:12 -0800251 // Drop any values that don't contain '='.
252 if (delimiter) {
253 *delimiter++ = '\0';
254 env[env_string.c_str()] = delimiter;
255 }
256 }
Josh Gao9b3fd672015-12-11 10:52:55 -0800257 }
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 Pursella9320582015-08-28 18:31:29 -0700284 if (type_ == SubprocessType::kPty) {
285 int fd;
286 pid_ = forkpty(&fd, pts_name, nullptr, nullptr);
Josh Gaofcb063c2016-03-04 17:50:10 -0800287 if (pid_ > 0) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700288 stdinout_sfd_.reset(fd);
Josh Gaofcb063c2016-03-04 17:50:10 -0800289 }
David Pursella9320582015-08-28 18:31:29 -0700290 } else {
David Pursell0955c662015-08-31 10:42:13 -0700291 if (!CreateSocketpair(&stdinout_sfd_, &child_stdinout_sfd)) {
Josh Gao43235072016-01-25 17:11:43 -0800292 *error = android::base::StringPrintf("failed to create socketpair for stdin/out: %s",
293 strerror(errno));
David Pursell0955c662015-08-31 10:42:13 -0700294 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 Gao43235072016-01-25 17:11:43 -0800299 *error = android::base::StringPrintf("failed to create socketpair for stderr: %s",
300 strerror(errno));
David Pursella9320582015-08-28 18:31:29 -0700301 return false;
302 }
303 pid_ = fork();
304 }
305
306 if (pid_ == -1) {
Josh Gao43235072016-01-25 17:11:43 -0800307 *error = android::base::StringPrintf("fork failed: %s", strerror(errno));
David Pursella9320582015-08-28 18:31:29 -0700308 return false;
309 }
310
311 if (pid_ == 0) {
312 // Subprocess child.
David Pursell80f67022015-08-28 15:08:49 -0700313 init_subproc_child();
314
David Pursella9320582015-08-28 18:31:29 -0700315 if (type_ == SubprocessType::kPty) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700316 child_stdinout_sfd.reset(OpenPtyChildFd(pts_name, &child_error_sfd));
David Pursella9320582015-08-28 18:31:29 -0700317 }
318
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700319 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 Pursella9320582015-08-28 18:31:29 -0700322
323 // exec doesn't trigger destructors, close the FDs manually.
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700324 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 Pursella9320582015-08-28 18:31:29 -0700330
Josh Gaob5028e42016-01-19 17:31:09 -0800331 if (command_.empty()) {
Josh Gao9b3fd672015-12-11 10:52:55 -0800332 execle(_PATH_BSHELL, _PATH_BSHELL, "-", nullptr, cenv.data());
David Pursella9320582015-08-28 18:31:29 -0700333 } else {
Josh Gao9b3fd672015-12-11 10:52:55 -0800334 execle(_PATH_BSHELL, _PATH_BSHELL, "-c", command_.c_str(), nullptr, cenv.data());
David Pursella9320582015-08-28 18:31:29 -0700335 }
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700336 WriteFdExactly(child_error_sfd, "exec '" _PATH_BSHELL "' failed: ");
337 WriteFdExactly(child_error_sfd, strerror(errno));
338 child_error_sfd.reset(-1);
Josh Gao9b3fd672015-12-11 10:52:55 -0800339 _Exit(1);
David Pursella9320582015-08-28 18:31:29 -0700340 }
341
342 // Subprocess parent.
David Pursell0955c662015-08-31 10:42:13 -0700343 D("subprocess parent: stdin/stdout FD = %d, stderr FD = %d",
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700344 stdinout_sfd_.get(), stderr_sfd_.get());
David Pursella9320582015-08-28 18:31:29 -0700345
346 // Wait to make sure the subprocess exec'd without error.
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700347 child_error_sfd.reset(-1);
348 std::string error_message = ReadAll(parent_error_sfd);
David Pursella9320582015-08-28 18:31:29 -0700349 if (!error_message.empty()) {
Josh Gao43235072016-01-25 17:11:43 -0800350 *error = error_message;
David Pursella9320582015-08-28 18:31:29 -0700351 return false;
352 }
353
Josh Gao9b3fd672015-12-11 10:52:55 -0800354 D("subprocess parent: exec completed");
David Pursell0955c662015-08-31 10:42:13 -0700355 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 Hughes2ce86e52016-05-27 17:51:24 -0700358 local_socket_sfd_.reset(stdinout_sfd_.release());
David Pursell0955c662015-08-31 10:42:13 -0700359 } else {
360 // Shell protocol: create another socketpair to intercept data.
361 if (!CreateSocketpair(&protocol_sfd_, &local_socket_sfd_)) {
Josh Gao43235072016-01-25 17:11:43 -0800362 *error = android::base::StringPrintf(
363 "failed to create socketpair to intercept data: %s", strerror(errno));
364 kill(pid_, SIGKILL);
David Pursell0955c662015-08-31 10:42:13 -0700365 return false;
366 }
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700367 D("protocol FD = %d", protocol_sfd_.get());
David Pursell0955c662015-08-31 10:42:13 -0700368
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700369 input_.reset(new ShellProtocol(protocol_sfd_));
370 output_.reset(new ShellProtocol(protocol_sfd_));
David Pursell0955c662015-08-31 10:42:13 -0700371 if (!input_ || !output_) {
Josh Gao43235072016-01-25 17:11:43 -0800372 *error = "failed to allocate shell protocol objects";
373 kill(pid_, SIGKILL);
David Pursell0955c662015-08-31 10:42:13 -0700374 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 Hughes2ce86e52016-05-27 17:51:24 -0700381 for (int fd : {stdinout_sfd_.get(), stderr_sfd_.get()}) {
David Pursell0955c662015-08-31 10:42:13 -0700382 if (fd >= 0) {
Yabin Cui6dfef252015-10-06 15:10:05 -0700383 if (!set_file_block_mode(fd, false)) {
Josh Gao43235072016-01-25 17:11:43 -0800384 *error = android::base::StringPrintf(
385 "failed to set non-blocking mode for fd %d", fd);
386 kill(pid_, SIGKILL);
David Pursell0955c662015-08-31 10:42:13 -0700387 return false;
388 }
389 }
390 }
391 }
David Pursella9320582015-08-28 18:31:29 -0700392
393 if (!adb_thread_create(ThreadHandler, this)) {
Josh Gao43235072016-01-25 17:11:43 -0800394 *error =
395 android::base::StringPrintf("failed to create subprocess thread: %s", strerror(errno));
396 kill(pid_, SIGKILL);
David Pursella9320582015-08-28 18:31:29 -0700397 return false;
398 }
399
Josh Gao9b3fd672015-12-11 10:52:55 -0800400 D("subprocess parent: completed");
David Pursella9320582015-08-28 18:31:29 -0700401 return true;
402}
403
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700404int Subprocess::OpenPtyChildFd(const char* pts_name, unique_fd* error_sfd) {
David Pursella9320582015-08-28 18:31:29 -0700405 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 Hughes2ce86e52016-05-27 17:51:24 -0700412 WriteFdExactly(*error_sfd, message);
David Pursella9320582015-08-28 18:31:29 -0700413 }
Josh Gao4abdeee2016-05-13 18:16:43 -0700414 abort();
David Pursella9320582015-08-28 18:31:29 -0700415 }
416
David Pursell57dd5ae2016-01-27 16:07:52 -0800417 if (make_pty_raw_) {
418 termios tattr;
419 if (tcgetattr(child_fd, &tattr) == -1) {
420 int saved_errno = errno;
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700421 WriteFdExactly(*error_sfd, "tcgetattr failed: ");
422 WriteFdExactly(*error_sfd, strerror(saved_errno));
Josh Gao4abdeee2016-05-13 18:16:43 -0700423 abort();
David Pursell57dd5ae2016-01-27 16:07:52 -0800424 }
425
426 cfmakeraw(&tattr);
427 if (tcsetattr(child_fd, TCSADRAIN, &tattr) == -1) {
428 int saved_errno = errno;
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700429 WriteFdExactly(*error_sfd, "tcsetattr failed: ");
430 WriteFdExactly(*error_sfd, strerror(saved_errno));
Josh Gao4abdeee2016-05-13 18:16:43 -0700431 abort();
David Pursell57dd5ae2016-01-27 16:07:52 -0800432 }
433 }
434
David Pursella9320582015-08-28 18:31:29 -0700435 return child_fd;
David Pursell80f67022015-08-28 15:08:49 -0700436}
437
Josh Gaob5fea142016-02-12 14:31:15 -0800438void Subprocess::ThreadHandler(void* userdata) {
David Pursella9320582015-08-28 18:31:29 -0700439 Subprocess* subprocess = reinterpret_cast<Subprocess*>(userdata);
David Pursell80f67022015-08-28 15:08:49 -0700440
David Pursella9320582015-08-28 18:31:29 -0700441 adb_thread_setname(android::base::StringPrintf(
442 "shell srvc %d", subprocess->local_socket_fd()));
David Pursell80f67022015-08-28 15:08:49 -0700443
David Pursell0955c662015-08-31 10:42:13 -0700444 subprocess->PassDataStreams();
David Pursell80f67022015-08-28 15:08:49 -0700445
David Pursell1ed57f02015-10-06 15:30:03 -0700446 D("deleting Subprocess for PID %d", subprocess->pid());
David Pursella9320582015-08-28 18:31:29 -0700447 delete subprocess;
David Pursell80f67022015-08-28 15:08:49 -0700448}
449
David Pursell0955c662015-08-31 10:42:13 -0700450void Subprocess::PassDataStreams() {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700451 if (protocol_sfd_ == -1) {
David Pursell0955c662015-08-31 10:42:13 -0700452 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 Hughes2ce86e52016-05-27 17:51:24 -0700459 for (unique_fd* sfd : {&protocol_sfd_, &stdinout_sfd_, &stderr_sfd_}) {
460 if (*sfd != -1) {
461 FD_SET(*sfd, &master_read_set);
David Pursell0955c662015-08-31 10:42:13 -0700462 }
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 Hughes2ce86e52016-05-27 17:51:24 -0700467 while (protocol_sfd_ != -1 && (stdinout_sfd_ != -1 || stderr_sfd_ != -1)) {
468 unique_fd* dead_sfd = SelectLoop(&master_read_set, &master_write_set);
David Pursell0955c662015-08-31 10:42:13 -0700469 if (dead_sfd) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700470 D("closing FD %d", dead_sfd->get());
471 FD_CLR(*dead_sfd, &master_read_set);
472 FD_CLR(*dead_sfd, &master_write_set);
David Pursell544e7952015-09-14 15:36:26 -0700473 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 Hughes2ce86e52016-05-27 17:51:24 -0700481 dead_sfd->reset(-1);
David Pursell0955c662015-08-31 10:42:13 -0700482 }
483 }
484}
485
486namespace {
487
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700488inline bool ValidAndInSet(const unique_fd& sfd, fd_set* set) {
489 return sfd != -1 && FD_ISSET(sfd, set);
David Pursell0955c662015-08-31 10:42:13 -0700490}
491
492} // namespace
493
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700494unique_fd* Subprocess::SelectLoop(fd_set* master_read_set_ptr,
495 fd_set* master_write_set_ptr) {
David Pursell0955c662015-08-31 10:42:13 -0700496 fd_set read_set, write_set;
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700497 int select_n = std::max(std::max(protocol_sfd_, stdinout_sfd_), stderr_sfd_) + 1;
498 unique_fd* dead_sfd = nullptr;
David Pursell0955c662015-08-31 10:42:13 -0700499
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 Hughes2ce86e52016-05-27 17:51:24 -0700509 stdinout_sfd_.reset(-1);
510 stderr_sfd_.reset(-1);
David Pursell0955c662015-08-31 10:42:13 -0700511 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 Hughes2ce86e52016-05-27 17:51:24 -0700530 FD_CLR(protocol_sfd_, master_read_set_ptr);
531 FD_SET(stdinout_sfd_, master_write_set_ptr);
David Pursell0955c662015-08-31 10:42:13 -0700532 }
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 Hughes2ce86e52016-05-27 17:51:24 -0700540 FD_SET(protocol_sfd_, master_read_set_ptr);
541 FD_CLR(stdinout_sfd_, master_write_set_ptr);
David Pursell0955c662015-08-31 10:42:13 -0700542 }
543 }
544 } // while (!dead_sfd)
545
546 return dead_sfd;
547}
548
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700549unique_fd* Subprocess::PassInput() {
David Pursell0955c662015-08-31 10:42:13 -0700550 // 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 Hughes2ce86e52016-05-27 17:51:24 -0700555 PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_;
David Pursell0955c662015-08-31 10:42:13 -0700556 }
557 return &protocol_sfd_;
558 }
559
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700560 if (stdinout_sfd_ != -1) {
David Pursell1ed57f02015-10-06 15:30:03 -0700561 switch (input_->id()) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800562 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 Hughes2ce86e52016-05-27 17:51:24 -0700571 ioctl(stdinout_sfd_, TIOCSWINSZ, &ws);
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800572 }
573 break;
David Pursell1ed57f02015-10-06 15:30:03 -0700574 case ShellProtocol::kIdStdin:
575 input_bytes_left_ = input_->data_length();
576 break;
577 case ShellProtocol::kIdCloseStdin:
578 if (type_ == SubprocessType::kRaw) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700579 if (adb_shutdown(stdinout_sfd_, SHUT_WR) == 0) {
David Pursell1ed57f02015-10-06 15:30:03 -0700580 return nullptr;
581 }
582 PLOG(ERROR) << "failed to shutdown writes to FD "
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700583 << stdinout_sfd_;
David Pursell1ed57f02015-10-06 15:30:03 -0700584 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 Hughes2ce86e52016-05-27 17:51:24 -0700592 D("can't close input for PTY FD %d", stdinout_sfd_.get());
David Pursell1ed57f02015-10-06 15:30:03 -0700593 }
594 break;
595 }
David Pursell0955c662015-08-31 10:42:13 -0700596 }
597 }
598
599 if (input_bytes_left_ > 0) {
600 int index = input_->data_length() - input_bytes_left_;
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700601 int bytes = adb_write(stdinout_sfd_, input_->data() + index, input_bytes_left_);
David Pursell0955c662015-08-31 10:42:13 -0700602 if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) {
603 if (bytes < 0) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700604 PLOG(ERROR) << "error reading stdin FD " << stdinout_sfd_;
David Pursell0955c662015-08-31 10:42:13 -0700605 }
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 Hughes2ce86e52016-05-27 17:51:24 -0700618unique_fd* Subprocess::PassOutput(unique_fd* sfd, ShellProtocol::Id id) {
619 int bytes = adb_read(*sfd, output_->data(), output_->data_capacity());
David Pursell0955c662015-08-31 10:42:13 -0700620 if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) {
David Pursell1ed57f02015-10-06 15:30:03 -0700621 // 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 Hughes2ce86e52016-05-27 17:51:24 -0700624 PLOG(ERROR) << "error reading output FD " << *sfd;
David Pursell0955c662015-08-31 10:42:13 -0700625 }
626 return sfd;
627 }
628
629 if (bytes > 0 && !output_->Write(id, bytes)) {
630 if (errno != 0) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700631 PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_;
David Pursell0955c662015-08-31 10:42:13 -0700632 }
633 return &protocol_sfd_;
634 }
635
636 return nullptr;
637}
638
David Pursella9320582015-08-28 18:31:29 -0700639void Subprocess::WaitForExit() {
David Pursell0955c662015-08-31 10:42:13 -0700640 int exit_code = 1;
641
David Pursella9320582015-08-28 18:31:29 -0700642 D("waiting for pid %d", pid_);
David Pursell80f67022015-08-28 15:08:49 -0700643 while (true) {
644 int status;
David Pursella9320582015-08-28 18:31:29 -0700645 if (pid_ == waitpid(pid_, &status, 0)) {
646 D("post waitpid (pid=%d) status=%04x", pid_, status);
David Pursell80f67022015-08-28 15:08:49 -0700647 if (WIFSIGNALED(status)) {
David Pursell0955c662015-08-31 10:42:13 -0700648 exit_code = 0x80 | WTERMSIG(status);
David Pursella9320582015-08-28 18:31:29 -0700649 D("subprocess killed by signal %d", WTERMSIG(status));
David Pursell80f67022015-08-28 15:08:49 -0700650 break;
651 } else if (!WIFEXITED(status)) {
David Pursella9320582015-08-28 18:31:29 -0700652 D("subprocess didn't exit");
David Pursell80f67022015-08-28 15:08:49 -0700653 break;
654 } else if (WEXITSTATUS(status) >= 0) {
David Pursell0955c662015-08-31 10:42:13 -0700655 exit_code = WEXITSTATUS(status);
David Pursella9320582015-08-28 18:31:29 -0700656 D("subprocess exit code = %d", WEXITSTATUS(status));
David Pursell80f67022015-08-28 15:08:49 -0700657 break;
658 }
David Pursella9320582015-08-28 18:31:29 -0700659 }
David Pursell80f67022015-08-28 15:08:49 -0700660 }
David Pursella9320582015-08-28 18:31:29 -0700661
David Pursell0955c662015-08-31 10:42:13 -0700662 // If we have an open protocol FD send an exit packet.
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700663 if (protocol_sfd_ != -1) {
David Pursell0955c662015-08-31 10:42:13 -0700664 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 Hughes2ce86e52016-05-27 17:51:24 -0700670 protocol_sfd_.reset(-1);
David Pursell0955c662015-08-31 10:42:13 -0700671 }
672
David Pursella9320582015-08-28 18:31:29 -0700673 // Pass the local socket FD to the shell cleanup fdevent.
674 if (SHELL_EXIT_NOTIFY_FD >= 0) {
Elliott Hughes2ce86e52016-05-27 17:51:24 -0700675 int fd = local_socket_sfd_;
David Pursella9320582015-08-28 18:31:29 -0700676 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 Hughes2ce86e52016-05-27 17:51:24 -0700681 static_cast<void>(local_socket_sfd_.release());
David Pursella9320582015-08-28 18:31:29 -0700682 } 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 Pursell80f67022015-08-28 15:08:49 -0700687 }
688}
689
690} // namespace
691
Josh Gao43235072016-01-25 17:11:43 -0800692// Create a pipe containing the error.
693static 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 Hughes18ddf5c2015-11-16 10:55:34 -0800723int StartSubprocess(const char* name, const char* terminal_type,
724 SubprocessType type, SubprocessProtocol protocol) {
725 D("starting %s subprocess (protocol=%s, TERM=%s): '%s'",
David Pursell0955c662015-08-31 10:42:13 -0700726 type == SubprocessType::kRaw ? "raw" : "PTY",
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800727 protocol == SubprocessProtocol::kNone ? "none" : "shell",
728 terminal_type, name);
David Pursell80f67022015-08-28 15:08:49 -0700729
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800730 Subprocess* subprocess = new Subprocess(name, terminal_type, type, protocol);
David Pursella9320582015-08-28 18:31:29 -0700731 if (!subprocess) {
732 LOG(ERROR) << "failed to allocate new subprocess";
Josh Gao43235072016-01-25 17:11:43 -0800733 return ReportError(protocol, "failed to allocate new subprocess");
David Pursell80f67022015-08-28 15:08:49 -0700734 }
735
Josh Gao43235072016-01-25 17:11:43 -0800736 std::string error;
737 if (!subprocess->ForkAndExec(&error)) {
738 LOG(ERROR) << "failed to start subprocess: " << error;
David Pursella9320582015-08-28 18:31:29 -0700739 delete subprocess;
Josh Gao43235072016-01-25 17:11:43 -0800740 return ReportError(protocol, error);
David Pursella9320582015-08-28 18:31:29 -0700741 }
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 Pursell80f67022015-08-28 15:08:49 -0700746}