blob: 8ece28c03957910e0ddb046a8e293dc1763b3be3 [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
138// Helper to automatically close an FD when it goes out of scope.
139class ScopedFd {
140 public:
141 ScopedFd() {}
142 ~ScopedFd() { Reset(); }
143
144 void Reset(int fd=-1) {
145 if (fd != fd_) {
146 if (valid()) {
147 adb_close(fd_);
148 }
149 fd_ = fd;
150 }
151 }
152
153 int Release() {
154 int temp = fd_;
155 fd_ = -1;
156 return temp;
157 }
158
159 bool valid() const { return fd_ >= 0; }
160
161 int fd() const { return fd_; }
162
163 private:
164 int fd_ = -1;
165
166 DISALLOW_COPY_AND_ASSIGN(ScopedFd);
167};
168
169// Creates a socketpair and saves the endpoints to |fd1| and |fd2|.
170bool CreateSocketpair(ScopedFd* fd1, ScopedFd* fd2) {
171 int sockets[2];
172 if (adb_socketpair(sockets) < 0) {
173 PLOG(ERROR) << "cannot create socket pair";
174 return false;
175 }
176 fd1->Reset(sockets[0]);
177 fd2->Reset(sockets[1]);
178 return true;
179}
180
181class Subprocess {
182 public:
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800183 Subprocess(const std::string& command, const char* terminal_type,
184 SubprocessType type, SubprocessProtocol protocol);
David Pursella9320582015-08-28 18:31:29 -0700185 ~Subprocess();
186
187 const std::string& command() const { return command_; }
David Pursella9320582015-08-28 18:31:29 -0700188
189 int local_socket_fd() const { return local_socket_sfd_.fd(); }
190
191 pid_t pid() const { return pid_; }
192
193 // Sets up FDs, forks a subprocess, starts the subprocess manager thread,
194 // and exec's the child. Returns false on failure.
Josh Gao43235072016-01-25 17:11:43 -0800195 bool ForkAndExec(std::string* _Nonnull error);
David Pursella9320582015-08-28 18:31:29 -0700196
197 private:
198 // Opens the file at |pts_name|.
199 int OpenPtyChildFd(const char* pts_name, ScopedFd* error_sfd);
200
Josh Gaob5fea142016-02-12 14:31:15 -0800201 static void ThreadHandler(void* userdata);
David Pursell0955c662015-08-31 10:42:13 -0700202 void PassDataStreams();
David Pursella9320582015-08-28 18:31:29 -0700203 void WaitForExit();
204
David Pursell0955c662015-08-31 10:42:13 -0700205 ScopedFd* SelectLoop(fd_set* master_read_set_ptr,
206 fd_set* master_write_set_ptr);
207
208 // Input/output stream handlers. Success returns nullptr, failure returns
209 // a pointer to the failed FD.
210 ScopedFd* PassInput();
211 ScopedFd* PassOutput(ScopedFd* sfd, ShellProtocol::Id id);
212
David Pursella9320582015-08-28 18:31:29 -0700213 const std::string command_;
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800214 const std::string terminal_type_;
David Pursell57dd5ae2016-01-27 16:07:52 -0800215 bool make_pty_raw_ = false;
David Pursella9320582015-08-28 18:31:29 -0700216 SubprocessType type_;
David Pursell0955c662015-08-31 10:42:13 -0700217 SubprocessProtocol protocol_;
David Pursella9320582015-08-28 18:31:29 -0700218 pid_t pid_ = -1;
219 ScopedFd local_socket_sfd_;
220
David Pursell0955c662015-08-31 10:42:13 -0700221 // Shell protocol variables.
222 ScopedFd stdinout_sfd_, stderr_sfd_, protocol_sfd_;
223 std::unique_ptr<ShellProtocol> input_, output_;
224 size_t input_bytes_left_ = 0;
225
David Pursella9320582015-08-28 18:31:29 -0700226 DISALLOW_COPY_AND_ASSIGN(Subprocess);
227};
228
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800229Subprocess::Subprocess(const std::string& command, const char* terminal_type,
230 SubprocessType type, SubprocessProtocol protocol)
231 : command_(command),
232 terminal_type_(terminal_type ? terminal_type : ""),
233 type_(type),
234 protocol_(protocol) {
David Pursell57dd5ae2016-01-27 16:07:52 -0800235 // If we aren't using the shell protocol we must allocate a PTY to properly close the
236 // subprocess. PTYs automatically send SIGHUP to the slave-side process when the master side
237 // of the PTY closes, which we rely on. If we use a raw pipe, processes that don't read/write,
238 // e.g. screenrecord, will never notice the broken pipe and terminate.
239 // The shell protocol doesn't require a PTY because it's always monitoring the local socket FD
240 // with select() and will send SIGHUP manually to the child process.
241 if (protocol_ == SubprocessProtocol::kNone && type_ == SubprocessType::kRaw) {
242 // Disable PTY input/output processing since the client is expecting raw data.
243 D("Can't create raw subprocess without shell protocol, using PTY in raw mode instead");
244 type_ = SubprocessType::kPty;
245 make_pty_raw_ = true;
246 }
David Pursella9320582015-08-28 18:31:29 -0700247}
248
249Subprocess::~Subprocess() {
Josh Gaoc65fae92016-01-19 16:21:17 -0800250 WaitForExit();
David Pursella9320582015-08-28 18:31:29 -0700251}
252
Josh Gao43235072016-01-25 17:11:43 -0800253bool Subprocess::ForkAndExec(std::string* error) {
David Pursell0955c662015-08-31 10:42:13 -0700254 ScopedFd child_stdinout_sfd, child_stderr_sfd;
255 ScopedFd parent_error_sfd, child_error_sfd;
David Pursella9320582015-08-28 18:31:29 -0700256 char pts_name[PATH_MAX];
257
Rubin Xud61a25c2016-01-11 10:23:47 +0000258 if (command_.empty()) {
259 __android_log_security_bswrite(SEC_TAG_ADB_SHELL_INTERACTIVE, "");
260 } else {
261 __android_log_security_bswrite(SEC_TAG_ADB_SHELL_CMD, command_.c_str());
262 }
263
Josh Gao9b3fd672015-12-11 10:52:55 -0800264 // Create a socketpair for the fork() child to report any errors back to the parent. Since we
265 // use threads, logging directly from the child might deadlock due to locks held in another
266 // thread during the fork.
David Pursella9320582015-08-28 18:31:29 -0700267 if (!CreateSocketpair(&parent_error_sfd, &child_error_sfd)) {
Josh Gao43235072016-01-25 17:11:43 -0800268 *error = android::base::StringPrintf(
269 "failed to create pipe for subprocess error reporting: %s", strerror(errno));
270 return false;
David Pursella9320582015-08-28 18:31:29 -0700271 }
272
Josh Gao9b3fd672015-12-11 10:52:55 -0800273 // Construct the environment for the child before we fork.
274 passwd* pw = getpwuid(getuid());
275 std::unordered_map<std::string, std::string> env;
Josh Gaoe03c9882015-12-11 15:49:12 -0800276 if (environ) {
277 char** current = environ;
278 while (char* env_cstr = *current++) {
279 std::string env_string = env_cstr;
280 char* delimiter = strchr(env_string.c_str(), '=');
Josh Gao9b3fd672015-12-11 10:52:55 -0800281
Josh Gaoe03c9882015-12-11 15:49:12 -0800282 // Drop any values that don't contain '='.
283 if (delimiter) {
284 *delimiter++ = '\0';
285 env[env_string.c_str()] = delimiter;
286 }
287 }
Josh Gao9b3fd672015-12-11 10:52:55 -0800288 }
289
290 if (pw != nullptr) {
291 // TODO: $HOSTNAME? Normally bash automatically sets that, but mksh doesn't.
292 env["HOME"] = pw->pw_dir;
293 env["LOGNAME"] = pw->pw_name;
294 env["USER"] = pw->pw_name;
295 env["SHELL"] = pw->pw_shell;
296 }
297
298 if (!terminal_type_.empty()) {
299 env["TERM"] = terminal_type_;
300 }
301
302 std::vector<std::string> joined_env;
303 for (auto it : env) {
304 const char* key = it.first.c_str();
305 const char* value = it.second.c_str();
306 joined_env.push_back(android::base::StringPrintf("%s=%s", key, value));
307 }
308
309 std::vector<const char*> cenv;
310 for (const std::string& str : joined_env) {
311 cenv.push_back(str.c_str());
312 }
313 cenv.push_back(nullptr);
314
David Pursella9320582015-08-28 18:31:29 -0700315 if (type_ == SubprocessType::kPty) {
316 int fd;
317 pid_ = forkpty(&fd, pts_name, nullptr, nullptr);
Josh Gaofcb063c2016-03-04 17:50:10 -0800318 if (pid_ > 0) {
319 stdinout_sfd_.Reset(fd);
320 }
David Pursella9320582015-08-28 18:31:29 -0700321 } else {
David Pursell0955c662015-08-31 10:42:13 -0700322 if (!CreateSocketpair(&stdinout_sfd_, &child_stdinout_sfd)) {
Josh Gao43235072016-01-25 17:11:43 -0800323 *error = android::base::StringPrintf("failed to create socketpair for stdin/out: %s",
324 strerror(errno));
David Pursell0955c662015-08-31 10:42:13 -0700325 return false;
326 }
327 // Raw subprocess + shell protocol allows for splitting stderr.
328 if (protocol_ == SubprocessProtocol::kShell &&
329 !CreateSocketpair(&stderr_sfd_, &child_stderr_sfd)) {
Josh Gao43235072016-01-25 17:11:43 -0800330 *error = android::base::StringPrintf("failed to create socketpair for stderr: %s",
331 strerror(errno));
David Pursella9320582015-08-28 18:31:29 -0700332 return false;
333 }
334 pid_ = fork();
335 }
336
337 if (pid_ == -1) {
Josh Gao43235072016-01-25 17:11:43 -0800338 *error = android::base::StringPrintf("fork failed: %s", strerror(errno));
David Pursella9320582015-08-28 18:31:29 -0700339 return false;
340 }
341
342 if (pid_ == 0) {
343 // Subprocess child.
David Pursell80f67022015-08-28 15:08:49 -0700344 init_subproc_child();
345
David Pursella9320582015-08-28 18:31:29 -0700346 if (type_ == SubprocessType::kPty) {
David Pursell0955c662015-08-31 10:42:13 -0700347 child_stdinout_sfd.Reset(OpenPtyChildFd(pts_name, &child_error_sfd));
David Pursella9320582015-08-28 18:31:29 -0700348 }
349
David Pursell0955c662015-08-31 10:42:13 -0700350 dup2(child_stdinout_sfd.fd(), STDIN_FILENO);
351 dup2(child_stdinout_sfd.fd(), STDOUT_FILENO);
352 dup2(child_stderr_sfd.valid() ? child_stderr_sfd.fd() : child_stdinout_sfd.fd(),
353 STDERR_FILENO);
David Pursella9320582015-08-28 18:31:29 -0700354
355 // exec doesn't trigger destructors, close the FDs manually.
David Pursell0955c662015-08-31 10:42:13 -0700356 stdinout_sfd_.Reset();
357 stderr_sfd_.Reset();
358 child_stdinout_sfd.Reset();
359 child_stderr_sfd.Reset();
David Pursella9320582015-08-28 18:31:29 -0700360 parent_error_sfd.Reset();
361 close_on_exec(child_error_sfd.fd());
362
Josh Gaob5028e42016-01-19 17:31:09 -0800363 if (command_.empty()) {
Josh Gao9b3fd672015-12-11 10:52:55 -0800364 execle(_PATH_BSHELL, _PATH_BSHELL, "-", nullptr, cenv.data());
David Pursella9320582015-08-28 18:31:29 -0700365 } else {
Josh Gao9b3fd672015-12-11 10:52:55 -0800366 execle(_PATH_BSHELL, _PATH_BSHELL, "-c", command_.c_str(), nullptr, cenv.data());
David Pursella9320582015-08-28 18:31:29 -0700367 }
Josh Gao43235072016-01-25 17:11:43 -0800368 WriteFdExactly(child_error_sfd.fd(), "exec '" _PATH_BSHELL "' failed: ");
369 WriteFdExactly(child_error_sfd.fd(), strerror(errno));
David Pursella9320582015-08-28 18:31:29 -0700370 child_error_sfd.Reset();
Josh Gao9b3fd672015-12-11 10:52:55 -0800371 _Exit(1);
David Pursella9320582015-08-28 18:31:29 -0700372 }
373
374 // Subprocess parent.
David Pursell0955c662015-08-31 10:42:13 -0700375 D("subprocess parent: stdin/stdout FD = %d, stderr FD = %d",
376 stdinout_sfd_.fd(), stderr_sfd_.fd());
David Pursella9320582015-08-28 18:31:29 -0700377
378 // Wait to make sure the subprocess exec'd without error.
379 child_error_sfd.Reset();
380 std::string error_message = ReadAll(parent_error_sfd.fd());
381 if (!error_message.empty()) {
Josh Gao43235072016-01-25 17:11:43 -0800382 *error = error_message;
David Pursella9320582015-08-28 18:31:29 -0700383 return false;
384 }
385
Josh Gao9b3fd672015-12-11 10:52:55 -0800386 D("subprocess parent: exec completed");
David Pursell0955c662015-08-31 10:42:13 -0700387 if (protocol_ == SubprocessProtocol::kNone) {
388 // No protocol: all streams pass through the stdinout FD and hook
389 // directly into the local socket for raw data transfer.
390 local_socket_sfd_.Reset(stdinout_sfd_.Release());
391 } else {
392 // Shell protocol: create another socketpair to intercept data.
393 if (!CreateSocketpair(&protocol_sfd_, &local_socket_sfd_)) {
Josh Gao43235072016-01-25 17:11:43 -0800394 *error = android::base::StringPrintf(
395 "failed to create socketpair to intercept data: %s", strerror(errno));
396 kill(pid_, SIGKILL);
David Pursell0955c662015-08-31 10:42:13 -0700397 return false;
398 }
399 D("protocol FD = %d", protocol_sfd_.fd());
400
401 input_.reset(new ShellProtocol(protocol_sfd_.fd()));
402 output_.reset(new ShellProtocol(protocol_sfd_.fd()));
403 if (!input_ || !output_) {
Josh Gao43235072016-01-25 17:11:43 -0800404 *error = "failed to allocate shell protocol objects";
405 kill(pid_, SIGKILL);
David Pursell0955c662015-08-31 10:42:13 -0700406 return false;
407 }
408
409 // Don't let reads/writes to the subprocess block our thread. This isn't
410 // likely but could happen under unusual circumstances, such as if we
411 // write a ton of data to stdin but the subprocess never reads it and
412 // the pipe fills up.
413 for (int fd : {stdinout_sfd_.fd(), stderr_sfd_.fd()}) {
414 if (fd >= 0) {
Yabin Cui6dfef252015-10-06 15:10:05 -0700415 if (!set_file_block_mode(fd, false)) {
Josh Gao43235072016-01-25 17:11:43 -0800416 *error = android::base::StringPrintf(
417 "failed to set non-blocking mode for fd %d", fd);
418 kill(pid_, SIGKILL);
David Pursell0955c662015-08-31 10:42:13 -0700419 return false;
420 }
421 }
422 }
423 }
David Pursella9320582015-08-28 18:31:29 -0700424
425 if (!adb_thread_create(ThreadHandler, this)) {
Josh Gao43235072016-01-25 17:11:43 -0800426 *error =
427 android::base::StringPrintf("failed to create subprocess thread: %s", strerror(errno));
428 kill(pid_, SIGKILL);
David Pursella9320582015-08-28 18:31:29 -0700429 return false;
430 }
431
Josh Gao9b3fd672015-12-11 10:52:55 -0800432 D("subprocess parent: completed");
David Pursella9320582015-08-28 18:31:29 -0700433 return true;
434}
435
436int Subprocess::OpenPtyChildFd(const char* pts_name, ScopedFd* error_sfd) {
437 int child_fd = adb_open(pts_name, O_RDWR | O_CLOEXEC);
438 if (child_fd == -1) {
439 // Don't use WriteFdFmt; since we're in the fork() child we don't want
440 // to allocate any heap memory to avoid race conditions.
441 const char* messages[] = {"child failed to open pseudo-term slave ",
442 pts_name, ": ", strerror(errno)};
443 for (const char* message : messages) {
444 WriteFdExactly(error_sfd->fd(), message);
445 }
446 exit(-1);
447 }
448
David Pursell57dd5ae2016-01-27 16:07:52 -0800449 if (make_pty_raw_) {
450 termios tattr;
451 if (tcgetattr(child_fd, &tattr) == -1) {
452 int saved_errno = errno;
453 WriteFdExactly(error_sfd->fd(), "tcgetattr failed: ");
454 WriteFdExactly(error_sfd->fd(), strerror(saved_errno));
455 exit(-1);
456 }
457
458 cfmakeraw(&tattr);
459 if (tcsetattr(child_fd, TCSADRAIN, &tattr) == -1) {
460 int saved_errno = errno;
461 WriteFdExactly(error_sfd->fd(), "tcsetattr failed: ");
462 WriteFdExactly(error_sfd->fd(), strerror(saved_errno));
463 exit(-1);
464 }
465 }
466
David Pursella9320582015-08-28 18:31:29 -0700467 return child_fd;
David Pursell80f67022015-08-28 15:08:49 -0700468}
469
Josh Gaob5fea142016-02-12 14:31:15 -0800470void Subprocess::ThreadHandler(void* userdata) {
David Pursella9320582015-08-28 18:31:29 -0700471 Subprocess* subprocess = reinterpret_cast<Subprocess*>(userdata);
David Pursell80f67022015-08-28 15:08:49 -0700472
David Pursella9320582015-08-28 18:31:29 -0700473 adb_thread_setname(android::base::StringPrintf(
474 "shell srvc %d", subprocess->local_socket_fd()));
David Pursell80f67022015-08-28 15:08:49 -0700475
David Pursell0955c662015-08-31 10:42:13 -0700476 subprocess->PassDataStreams();
David Pursell80f67022015-08-28 15:08:49 -0700477
David Pursell1ed57f02015-10-06 15:30:03 -0700478 D("deleting Subprocess for PID %d", subprocess->pid());
David Pursella9320582015-08-28 18:31:29 -0700479 delete subprocess;
David Pursell80f67022015-08-28 15:08:49 -0700480}
481
David Pursell0955c662015-08-31 10:42:13 -0700482void Subprocess::PassDataStreams() {
483 if (!protocol_sfd_.valid()) {
484 return;
485 }
486
487 // Start by trying to read from the protocol FD, stdout, and stderr.
488 fd_set master_read_set, master_write_set;
489 FD_ZERO(&master_read_set);
490 FD_ZERO(&master_write_set);
491 for (ScopedFd* sfd : {&protocol_sfd_, &stdinout_sfd_, &stderr_sfd_}) {
492 if (sfd->valid()) {
493 FD_SET(sfd->fd(), &master_read_set);
494 }
495 }
496
497 // Pass data until the protocol FD or both the subprocess pipes die, at
498 // which point we can't pass any more data.
499 while (protocol_sfd_.valid() &&
500 (stdinout_sfd_.valid() || stderr_sfd_.valid())) {
501 ScopedFd* dead_sfd = SelectLoop(&master_read_set, &master_write_set);
502 if (dead_sfd) {
503 D("closing FD %d", dead_sfd->fd());
504 FD_CLR(dead_sfd->fd(), &master_read_set);
505 FD_CLR(dead_sfd->fd(), &master_write_set);
David Pursell544e7952015-09-14 15:36:26 -0700506 if (dead_sfd == &protocol_sfd_) {
507 // Using SIGHUP is a decent general way to indicate that the
508 // controlling process is going away. If specific signals are
509 // needed (e.g. SIGINT), pass those through the shell protocol
510 // and only fall back on this for unexpected closures.
511 D("protocol FD died, sending SIGHUP to pid %d", pid_);
512 kill(pid_, SIGHUP);
513 }
David Pursell0955c662015-08-31 10:42:13 -0700514 dead_sfd->Reset();
515 }
516 }
517}
518
519namespace {
520
521inline bool ValidAndInSet(const ScopedFd& sfd, fd_set* set) {
522 return sfd.valid() && FD_ISSET(sfd.fd(), set);
523}
524
525} // namespace
526
527ScopedFd* Subprocess::SelectLoop(fd_set* master_read_set_ptr,
528 fd_set* master_write_set_ptr) {
529 fd_set read_set, write_set;
530 int select_n = std::max(std::max(protocol_sfd_.fd(), stdinout_sfd_.fd()),
531 stderr_sfd_.fd()) + 1;
532 ScopedFd* dead_sfd = nullptr;
533
534 // Keep calling select() and passing data until an FD closes/errors.
535 while (!dead_sfd) {
536 memcpy(&read_set, master_read_set_ptr, sizeof(read_set));
537 memcpy(&write_set, master_write_set_ptr, sizeof(write_set));
538 if (select(select_n, &read_set, &write_set, nullptr, nullptr) < 0) {
539 if (errno == EINTR) {
540 continue;
541 } else {
542 PLOG(ERROR) << "select failed, closing subprocess pipes";
543 stdinout_sfd_.Reset();
544 stderr_sfd_.Reset();
545 return nullptr;
546 }
547 }
548
549 // Read stdout, write to protocol FD.
550 if (ValidAndInSet(stdinout_sfd_, &read_set)) {
551 dead_sfd = PassOutput(&stdinout_sfd_, ShellProtocol::kIdStdout);
552 }
553
554 // Read stderr, write to protocol FD.
555 if (!dead_sfd && ValidAndInSet(stderr_sfd_, &read_set)) {
556 dead_sfd = PassOutput(&stderr_sfd_, ShellProtocol::kIdStderr);
557 }
558
559 // Read protocol FD, write to stdin.
560 if (!dead_sfd && ValidAndInSet(protocol_sfd_, &read_set)) {
561 dead_sfd = PassInput();
562 // If we didn't finish writing, block on stdin write.
563 if (input_bytes_left_) {
564 FD_CLR(protocol_sfd_.fd(), master_read_set_ptr);
565 FD_SET(stdinout_sfd_.fd(), master_write_set_ptr);
566 }
567 }
568
569 // Continue writing to stdin; only happens if a previous write blocked.
570 if (!dead_sfd && ValidAndInSet(stdinout_sfd_, &write_set)) {
571 dead_sfd = PassInput();
572 // If we finished writing, go back to blocking on protocol read.
573 if (!input_bytes_left_) {
574 FD_SET(protocol_sfd_.fd(), master_read_set_ptr);
575 FD_CLR(stdinout_sfd_.fd(), master_write_set_ptr);
576 }
577 }
578 } // while (!dead_sfd)
579
580 return dead_sfd;
581}
582
583ScopedFd* Subprocess::PassInput() {
584 // Only read a new packet if we've finished writing the last one.
585 if (!input_bytes_left_) {
586 if (!input_->Read()) {
587 // Read() uses ReadFdExactly() which sets errno to 0 on EOF.
588 if (errno != 0) {
589 PLOG(ERROR) << "error reading protocol FD "
590 << protocol_sfd_.fd();
591 }
592 return &protocol_sfd_;
593 }
594
David Pursell1ed57f02015-10-06 15:30:03 -0700595 if (stdinout_sfd_.valid()) {
596 switch (input_->id()) {
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800597 case ShellProtocol::kIdWindowSizeChange:
598 int rows, cols, x_pixels, y_pixels;
599 if (sscanf(input_->data(), "%dx%d,%dx%d",
600 &rows, &cols, &x_pixels, &y_pixels) == 4) {
601 winsize ws;
602 ws.ws_row = rows;
603 ws.ws_col = cols;
604 ws.ws_xpixel = x_pixels;
605 ws.ws_ypixel = y_pixels;
606 ioctl(stdinout_sfd_.fd(), TIOCSWINSZ, &ws);
607 }
608 break;
David Pursell1ed57f02015-10-06 15:30:03 -0700609 case ShellProtocol::kIdStdin:
610 input_bytes_left_ = input_->data_length();
611 break;
612 case ShellProtocol::kIdCloseStdin:
613 if (type_ == SubprocessType::kRaw) {
614 if (adb_shutdown(stdinout_sfd_.fd(), SHUT_WR) == 0) {
615 return nullptr;
616 }
617 PLOG(ERROR) << "failed to shutdown writes to FD "
618 << stdinout_sfd_.fd();
619 return &stdinout_sfd_;
620 } else {
621 // PTYs can't close just input, so rather than close the
622 // FD and risk losing subprocess output, leave it open.
623 // This only happens if the client starts a PTY shell
624 // non-interactively which is rare and unsupported.
625 // If necessary, the client can manually close the shell
626 // with `exit` or by killing the adb client process.
Elliott Hughesc15b17f2015-11-03 11:18:40 -0800627 D("can't close input for PTY FD %d", stdinout_sfd_.fd());
David Pursell1ed57f02015-10-06 15:30:03 -0700628 }
629 break;
630 }
David Pursell0955c662015-08-31 10:42:13 -0700631 }
632 }
633
634 if (input_bytes_left_ > 0) {
635 int index = input_->data_length() - input_bytes_left_;
636 int bytes = adb_write(stdinout_sfd_.fd(), input_->data() + index,
637 input_bytes_left_);
638 if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) {
639 if (bytes < 0) {
640 PLOG(ERROR) << "error reading stdin FD " << stdinout_sfd_.fd();
641 }
642 // stdin is done, mark this packet as finished and we'll just start
643 // dumping any further data received from the protocol FD.
644 input_bytes_left_ = 0;
645 return &stdinout_sfd_;
646 } else if (bytes > 0) {
647 input_bytes_left_ -= bytes;
648 }
649 }
650
651 return nullptr;
652}
653
654ScopedFd* Subprocess::PassOutput(ScopedFd* sfd, ShellProtocol::Id id) {
655 int bytes = adb_read(sfd->fd(), output_->data(), output_->data_capacity());
656 if (bytes == 0 || (bytes < 0 && errno != EAGAIN)) {
David Pursell1ed57f02015-10-06 15:30:03 -0700657 // read() returns EIO if a PTY closes; don't report this as an error,
658 // it just means the subprocess completed.
659 if (bytes < 0 && !(type_ == SubprocessType::kPty && errno == EIO)) {
David Pursell0955c662015-08-31 10:42:13 -0700660 PLOG(ERROR) << "error reading output FD " << sfd->fd();
661 }
662 return sfd;
663 }
664
665 if (bytes > 0 && !output_->Write(id, bytes)) {
666 if (errno != 0) {
667 PLOG(ERROR) << "error reading protocol FD " << protocol_sfd_.fd();
668 }
669 return &protocol_sfd_;
670 }
671
672 return nullptr;
673}
674
David Pursella9320582015-08-28 18:31:29 -0700675void Subprocess::WaitForExit() {
David Pursell0955c662015-08-31 10:42:13 -0700676 int exit_code = 1;
677
David Pursella9320582015-08-28 18:31:29 -0700678 D("waiting for pid %d", pid_);
David Pursell80f67022015-08-28 15:08:49 -0700679 while (true) {
680 int status;
David Pursella9320582015-08-28 18:31:29 -0700681 if (pid_ == waitpid(pid_, &status, 0)) {
682 D("post waitpid (pid=%d) status=%04x", pid_, status);
David Pursell80f67022015-08-28 15:08:49 -0700683 if (WIFSIGNALED(status)) {
David Pursell0955c662015-08-31 10:42:13 -0700684 exit_code = 0x80 | WTERMSIG(status);
David Pursella9320582015-08-28 18:31:29 -0700685 D("subprocess killed by signal %d", WTERMSIG(status));
David Pursell80f67022015-08-28 15:08:49 -0700686 break;
687 } else if (!WIFEXITED(status)) {
David Pursella9320582015-08-28 18:31:29 -0700688 D("subprocess didn't exit");
David Pursell80f67022015-08-28 15:08:49 -0700689 break;
690 } else if (WEXITSTATUS(status) >= 0) {
David Pursell0955c662015-08-31 10:42:13 -0700691 exit_code = WEXITSTATUS(status);
David Pursella9320582015-08-28 18:31:29 -0700692 D("subprocess exit code = %d", WEXITSTATUS(status));
David Pursell80f67022015-08-28 15:08:49 -0700693 break;
694 }
David Pursella9320582015-08-28 18:31:29 -0700695 }
David Pursell80f67022015-08-28 15:08:49 -0700696 }
David Pursella9320582015-08-28 18:31:29 -0700697
David Pursell0955c662015-08-31 10:42:13 -0700698 // If we have an open protocol FD send an exit packet.
699 if (protocol_sfd_.valid()) {
700 output_->data()[0] = exit_code;
701 if (output_->Write(ShellProtocol::kIdExit, 1)) {
702 D("wrote the exit code packet: %d", exit_code);
703 } else {
704 PLOG(ERROR) << "failed to write the exit code packet";
705 }
706 protocol_sfd_.Reset();
707 }
708
David Pursella9320582015-08-28 18:31:29 -0700709 // Pass the local socket FD to the shell cleanup fdevent.
710 if (SHELL_EXIT_NOTIFY_FD >= 0) {
711 int fd = local_socket_sfd_.fd();
712 if (WriteFdExactly(SHELL_EXIT_NOTIFY_FD, &fd, sizeof(fd))) {
713 D("passed fd %d to SHELL_EXIT_NOTIFY_FD (%d) for pid %d",
714 fd, SHELL_EXIT_NOTIFY_FD, pid_);
715 // The shell exit fdevent now owns the FD and will close it once
716 // the last bit of data flushes through.
717 local_socket_sfd_.Release();
718 } else {
719 PLOG(ERROR) << "failed to write fd " << fd
720 << " to SHELL_EXIT_NOTIFY_FD (" << SHELL_EXIT_NOTIFY_FD
721 << ") for pid " << pid_;
722 }
David Pursell80f67022015-08-28 15:08:49 -0700723 }
724}
725
726} // namespace
727
Josh Gao43235072016-01-25 17:11:43 -0800728// Create a pipe containing the error.
729static int ReportError(SubprocessProtocol protocol, const std::string& message) {
730 int pipefd[2];
731 if (pipe(pipefd) != 0) {
732 LOG(ERROR) << "failed to create pipe to report error";
733 return -1;
734 }
735
736 std::string buf = android::base::StringPrintf("error: %s\n", message.c_str());
737 if (protocol == SubprocessProtocol::kShell) {
738 ShellProtocol::Id id = ShellProtocol::kIdStderr;
739 uint32_t length = buf.length();
740 WriteFdExactly(pipefd[1], &id, sizeof(id));
741 WriteFdExactly(pipefd[1], &length, sizeof(length));
742 }
743
744 WriteFdExactly(pipefd[1], buf.data(), buf.length());
745
746 if (protocol == SubprocessProtocol::kShell) {
747 ShellProtocol::Id id = ShellProtocol::kIdExit;
748 uint32_t length = 1;
749 char exit_code = 126;
750 WriteFdExactly(pipefd[1], &id, sizeof(id));
751 WriteFdExactly(pipefd[1], &length, sizeof(length));
752 WriteFdExactly(pipefd[1], &exit_code, sizeof(exit_code));
753 }
754
755 adb_close(pipefd[1]);
756 return pipefd[0];
757}
758
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800759int StartSubprocess(const char* name, const char* terminal_type,
760 SubprocessType type, SubprocessProtocol protocol) {
761 D("starting %s subprocess (protocol=%s, TERM=%s): '%s'",
David Pursell0955c662015-08-31 10:42:13 -0700762 type == SubprocessType::kRaw ? "raw" : "PTY",
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800763 protocol == SubprocessProtocol::kNone ? "none" : "shell",
764 terminal_type, name);
David Pursell80f67022015-08-28 15:08:49 -0700765
Elliott Hughes18ddf5c2015-11-16 10:55:34 -0800766 Subprocess* subprocess = new Subprocess(name, terminal_type, type, protocol);
David Pursella9320582015-08-28 18:31:29 -0700767 if (!subprocess) {
768 LOG(ERROR) << "failed to allocate new subprocess";
Josh Gao43235072016-01-25 17:11:43 -0800769 return ReportError(protocol, "failed to allocate new subprocess");
David Pursell80f67022015-08-28 15:08:49 -0700770 }
771
Josh Gao43235072016-01-25 17:11:43 -0800772 std::string error;
773 if (!subprocess->ForkAndExec(&error)) {
774 LOG(ERROR) << "failed to start subprocess: " << error;
David Pursella9320582015-08-28 18:31:29 -0700775 delete subprocess;
Josh Gao43235072016-01-25 17:11:43 -0800776 return ReportError(protocol, error);
David Pursella9320582015-08-28 18:31:29 -0700777 }
778
779 D("subprocess creation successful: local_socket_fd=%d, pid=%d",
780 subprocess->local_socket_fd(), subprocess->pid());
781 return subprocess->local_socket_fd();
David Pursell80f67022015-08-28 15:08:49 -0700782}