OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license | |
5 * that can be found in the LICENSE file in the root of the source | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 | |
11 #include "webrtc/base/posix.h" | |
12 | |
13 #include <sys/wait.h> | |
14 #include <errno.h> | |
15 #include <unistd.h> | |
16 | |
17 #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID) | |
18 #include "webrtc/base/linuxfdwalk.h" | |
19 #endif | |
20 #include "webrtc/base/logging.h" | |
21 | |
22 namespace rtc { | |
23 | |
24 #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID) | |
25 static void closefds(void *close_errors, int fd) { | |
26 if (fd <= 2) { | |
27 // We leave stdin/out/err open to the browser's terminal, if any. | |
28 return; | |
29 } | |
30 if (close(fd) < 0) { | |
31 *static_cast<bool *>(close_errors) = true; | |
32 } | |
33 } | |
34 #endif | |
35 | |
36 enum { | |
37 EXIT_FLAG_CHDIR_ERRORS = 1 << 0, | |
38 #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID) | |
39 EXIT_FLAG_FDWALK_ERRORS = 1 << 1, | |
40 EXIT_FLAG_CLOSE_ERRORS = 1 << 2, | |
41 #endif | |
42 EXIT_FLAG_SECOND_FORK_FAILED = 1 << 3, | |
43 }; | |
44 | |
45 bool RunAsDaemon(const char *file, const char *const argv[]) { | |
46 // Fork intermediate child to daemonize. | |
47 pid_t pid = fork(); | |
48 if (pid < 0) { | |
49 LOG_ERR(LS_ERROR) << "fork()"; | |
50 return false; | |
51 } else if (!pid) { | |
52 // Child. | |
53 | |
54 // We try to close all fds and change directory to /, but if that fails we | |
55 // keep going because it's not critical. | |
56 int exit_code = 0; | |
57 if (chdir("/") < 0) { | |
58 exit_code |= EXIT_FLAG_CHDIR_ERRORS; | |
59 } | |
60 #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID) | |
61 bool close_errors = false; | |
62 if (fdwalk(&closefds, &close_errors) < 0) { | |
63 exit_code |= EXIT_FLAG_FDWALK_ERRORS; | |
64 } | |
65 if (close_errors) { | |
66 exit_code |= EXIT_FLAG_CLOSE_ERRORS; | |
67 } | |
68 #endif | |
69 | |
70 // Fork again to become a daemon. | |
71 pid = fork(); | |
72 // It is important that everything here use _exit() and not exit(), because | |
73 // exit() would call the destructors of all global variables in the whole | |
74 // process, which is both unnecessary and unsafe. | |
75 if (pid < 0) { | |
76 exit_code |= EXIT_FLAG_SECOND_FORK_FAILED; | |
77 _exit(exit_code); // if second fork failed | |
78 } else if (!pid) { | |
79 // Child. | |
80 // Successfully daemonized. Run command. | |
81 // WEBRTC_POSIX requires the args to be typed as non-const for historical | |
82 // reasons, but it mandates that the actual implementation be const, so | |
83 // the cast is safe. | |
84 execvp(file, const_cast<char *const *>(argv)); | |
85 _exit(255); // if execvp failed | |
86 } | |
87 | |
88 // Parent. | |
89 // Successfully spawned process, but report any problems to the parent where | |
90 // we can log them. | |
91 _exit(exit_code); | |
92 } | |
93 | |
94 // Parent. Reap intermediate child. | |
95 int status; | |
96 pid_t child = waitpid(pid, &status, 0); | |
97 if (child < 0) { | |
98 LOG_ERR(LS_ERROR) << "Error in waitpid()"; | |
99 return false; | |
100 } | |
101 if (child != pid) { | |
102 // Should never happen (see man page). | |
103 LOG(LS_ERROR) << "waitpid() chose wrong child???"; | |
104 return false; | |
105 } | |
106 if (!WIFEXITED(status)) { | |
107 LOG(LS_ERROR) << "Intermediate child killed uncleanly"; // Probably crashed | |
108 return false; | |
109 } | |
110 | |
111 int exit_code = WEXITSTATUS(status); | |
112 if (exit_code & EXIT_FLAG_CHDIR_ERRORS) { | |
113 LOG(LS_WARNING) << "Child reported probles calling chdir()"; | |
114 } | |
115 #if defined(WEBRTC_LINUX) && !defined(WEBRTC_ANDROID) | |
116 if (exit_code & EXIT_FLAG_FDWALK_ERRORS) { | |
117 LOG(LS_WARNING) << "Child reported problems calling fdwalk()"; | |
118 } | |
119 if (exit_code & EXIT_FLAG_CLOSE_ERRORS) { | |
120 LOG(LS_WARNING) << "Child reported problems calling close()"; | |
121 } | |
122 #endif | |
123 if (exit_code & EXIT_FLAG_SECOND_FORK_FAILED) { | |
124 LOG(LS_ERROR) << "Failed to daemonize"; | |
125 // This means the command was not launched, so failure. | |
126 return false; | |
127 } | |
128 return true; | |
129 } | |
130 | |
131 } // namespace rtc | |
OLD | NEW |