xref: /PHP-5.3/sapi/fpm/fpm/fpm_stdio.c (revision 0549e55d)
1 
2 	/* $Id: fpm_stdio.c,v 1.22.2.2 2008/12/13 03:32:24 anight Exp $ */
3 	/* (c) 2007,2008 Andrei Nigmatulin */
4 
5 #include "fpm_config.h"
6 
7 #include <sys/types.h>
8 #include <sys/stat.h>
9 #include <string.h>
10 #include <fcntl.h>
11 #include <unistd.h>
12 #include <errno.h>
13 
14 #include "php_syslog.h"
15 
16 #include "fpm.h"
17 #include "fpm_children.h"
18 #include "fpm_events.h"
19 #include "fpm_sockets.h"
20 #include "fpm_stdio.h"
21 #include "zlog.h"
22 
23 static int fd_stdout[2];
24 static int fd_stderr[2];
25 
fpm_stdio_init_main()26 int fpm_stdio_init_main() /* {{{ */
27 {
28 	int fd = open("/dev/null", O_RDWR);
29 
30 	if (0 > fd) {
31 		zlog(ZLOG_SYSERROR, "failed to init stdio: open(\"/dev/null\")");
32 		return -1;
33 	}
34 
35 	if (0 > dup2(fd, STDIN_FILENO) || 0 > dup2(fd, STDOUT_FILENO)) {
36 		zlog(ZLOG_SYSERROR, "failed to init stdio: dup2()");
37 		close(fd);
38 		return -1;
39 	}
40 	close(fd);
41 	return 0;
42 }
43 /* }}} */
44 
fpm_stdio_init_final()45 int fpm_stdio_init_final() /* {{{ */
46 {
47 	if (fpm_global_config.daemonize) {
48 		/* prevent duping if logging to syslog */
49 		if (fpm_globals.error_log_fd > 0 && fpm_globals.error_log_fd != STDERR_FILENO) {
50 
51 			/* there might be messages to stderr from other parts of the code, we need to log them all */
52 			if (0 > dup2(fpm_globals.error_log_fd, STDERR_FILENO)) {
53 				zlog(ZLOG_SYSERROR, "failed to init stdio: dup2()");
54 				return -1;
55 			}
56 		}
57 	}
58 	zlog_set_launched();
59 	return 0;
60 }
61 /* }}} */
62 
fpm_stdio_init_child(struct fpm_worker_pool_s * wp)63 int fpm_stdio_init_child(struct fpm_worker_pool_s *wp) /* {{{ */
64 {
65 #ifdef HAVE_SYSLOG_H
66 	if (fpm_globals.error_log_fd == ZLOG_SYSLOG) {
67 		closelog(); /* ensure to close syslog not to interrupt with PHP syslog code */
68 	} else
69 #endif
70 	if (fpm_globals.error_log_fd > 0) {
71 		close(fpm_globals.error_log_fd);
72 	}
73 	fpm_globals.error_log_fd = -1;
74 	zlog_set_fd(-1);
75 
76 	if (wp->listening_socket != STDIN_FILENO) {
77 		if (0 > dup2(wp->listening_socket, STDIN_FILENO)) {
78 			zlog(ZLOG_SYSERROR, "failed to init child stdio: dup2()");
79 			return -1;
80 		}
81 	}
82 	return 0;
83 }
84 /* }}} */
85 
fpm_stdio_child_said(struct fpm_event_s * ev,short which,void * arg)86 static void fpm_stdio_child_said(struct fpm_event_s *ev, short which, void *arg) /* {{{ */
87 {
88 	static const int max_buf_size = 1024;
89 	int fd = ev->fd;
90 	char buf[max_buf_size];
91 	struct fpm_child_s *child;
92 	int is_stdout;
93 	struct fpm_event_s *event;
94 	int fifo_in = 1, fifo_out = 1;
95 	int is_last_message = 0;
96 	int in_buf = 0;
97 	int res;
98 
99 	if (!arg) {
100 		return;
101 	}
102 	child = (struct fpm_child_s *)arg;
103 	is_stdout = (fd == child->fd_stdout);
104 	if (is_stdout) {
105 		event = &child->ev_stdout;
106 	} else {
107 		event = &child->ev_stderr;
108 	}
109 
110 	while (fifo_in || fifo_out) {
111 		if (fifo_in) {
112 			res = read(fd, buf + in_buf, max_buf_size - 1 - in_buf);
113 			if (res <= 0) { /* no data */
114 				fifo_in = 0;
115 				if (res < 0 && (errno == EAGAIN || errno == EWOULDBLOCK)) {
116 					/* just no more data ready */
117 				} else { /* error or pipe is closed */
118 
119 					if (res < 0) { /* error */
120 						zlog(ZLOG_SYSERROR, "unable to read what child say");
121 					}
122 
123 					fpm_event_del(event);
124 					is_last_message = 1;
125 
126 					if (is_stdout) {
127 						close(child->fd_stdout);
128 						child->fd_stdout = -1;
129 					} else {
130 						close(child->fd_stderr);
131 						child->fd_stderr = -1;
132 					}
133 				}
134 			} else {
135 				in_buf += res;
136 			}
137 		}
138 
139 		if (fifo_out) {
140 			if (in_buf == 0) {
141 				fifo_out = 0;
142 			} else {
143 				char *nl;
144 				int should_print = 0;
145 				buf[in_buf] = '\0';
146 
147 				/* FIXME: there might be binary data */
148 
149 				/* we should print if no more space in the buffer */
150 				if (in_buf == max_buf_size - 1) {
151 					should_print = 1;
152 				}
153 
154 				/* we should print if no more data to come */
155 				if (!fifo_in) {
156 					should_print = 1;
157 				}
158 
159 				nl = strchr(buf, '\n');
160 				if (nl || should_print) {
161 
162 					if (nl) {
163 						*nl = '\0';
164 					}
165 
166 					zlog(ZLOG_WARNING, "[pool %s] child %d said into %s: \"%s\"%s", child->wp->config->name,
167 					  (int) child->pid, is_stdout ? "stdout" : "stderr", buf, is_last_message ? ", pipe is closed" : "");
168 
169 					if (nl) {
170 						int out_buf = 1 + nl - buf;
171 						memmove(buf, buf + out_buf, in_buf - out_buf);
172 						in_buf -= out_buf;
173 					} else {
174 						in_buf = 0;
175 					}
176 				}
177 			}
178 		}
179 	}
180 }
181 /* }}} */
182 
fpm_stdio_prepare_pipes(struct fpm_child_s * child)183 int fpm_stdio_prepare_pipes(struct fpm_child_s *child) /* {{{ */
184 {
185 	if (0 == child->wp->config->catch_workers_output) { /* not required */
186 		return 0;
187 	}
188 
189 	if (0 > pipe(fd_stdout)) {
190 		zlog(ZLOG_SYSERROR, "failed to prepare the stdout pipe");
191 		return -1;
192 	}
193 
194 	if (0 > pipe(fd_stderr)) {
195 		zlog(ZLOG_SYSERROR, "failed to prepare the stderr pipe");
196 		close(fd_stdout[0]);
197 		close(fd_stdout[1]);
198 		return -1;
199 	}
200 
201 	if (0 > fd_set_blocked(fd_stdout[0], 0) || 0 > fd_set_blocked(fd_stderr[0], 0)) {
202 		zlog(ZLOG_SYSERROR, "failed to unblock pipes");
203 		close(fd_stdout[0]);
204 		close(fd_stdout[1]);
205 		close(fd_stderr[0]);
206 		close(fd_stderr[1]);
207 		return -1;
208 	}
209 	return 0;
210 }
211 /* }}} */
212 
fpm_stdio_parent_use_pipes(struct fpm_child_s * child)213 int fpm_stdio_parent_use_pipes(struct fpm_child_s *child) /* {{{ */
214 {
215 	if (0 == child->wp->config->catch_workers_output) { /* not required */
216 		return 0;
217 	}
218 
219 	close(fd_stdout[1]);
220 	close(fd_stderr[1]);
221 
222 	child->fd_stdout = fd_stdout[0];
223 	child->fd_stderr = fd_stderr[0];
224 
225 	fpm_event_set(&child->ev_stdout, child->fd_stdout, FPM_EV_READ, fpm_stdio_child_said, child);
226 	fpm_event_add(&child->ev_stdout, 0);
227 
228 	fpm_event_set(&child->ev_stderr, child->fd_stderr, FPM_EV_READ, fpm_stdio_child_said, child);
229 	fpm_event_add(&child->ev_stderr, 0);
230 	return 0;
231 }
232 /* }}} */
233 
fpm_stdio_discard_pipes(struct fpm_child_s * child)234 int fpm_stdio_discard_pipes(struct fpm_child_s *child) /* {{{ */
235 {
236 	if (0 == child->wp->config->catch_workers_output) { /* not required */
237 		return 0;
238 	}
239 
240 	close(fd_stdout[1]);
241 	close(fd_stderr[1]);
242 
243 	close(fd_stdout[0]);
244 	close(fd_stderr[0]);
245 	return 0;
246 }
247 /* }}} */
248 
fpm_stdio_child_use_pipes(struct fpm_child_s * child)249 void fpm_stdio_child_use_pipes(struct fpm_child_s *child) /* {{{ */
250 {
251 	if (child->wp->config->catch_workers_output) {
252 		dup2(fd_stdout[1], STDOUT_FILENO);
253 		dup2(fd_stderr[1], STDERR_FILENO);
254 		close(fd_stdout[0]); close(fd_stdout[1]);
255 		close(fd_stderr[0]); close(fd_stderr[1]);
256 	} else {
257 		/* stdout of parent is always /dev/null */
258 		dup2(STDOUT_FILENO, STDERR_FILENO);
259 	}
260 }
261 /* }}} */
262 
fpm_stdio_open_error_log(int reopen)263 int fpm_stdio_open_error_log(int reopen) /* {{{ */
264 {
265 	int fd;
266 
267 #ifdef HAVE_SYSLOG_H
268 	if (!strcasecmp(fpm_global_config.error_log, "syslog")) {
269 		openlog(fpm_global_config.syslog_ident, LOG_PID | LOG_CONS, fpm_global_config.syslog_facility);
270 		fpm_globals.error_log_fd = ZLOG_SYSLOG;
271 		if (fpm_global_config.daemonize) {
272 			zlog_set_fd(fpm_globals.error_log_fd);
273 		}
274 		return 0;
275 	}
276 #endif
277 
278 	fd = open(fpm_global_config.error_log, O_WRONLY | O_APPEND | O_CREAT, S_IRUSR | S_IWUSR);
279 	if (0 > fd) {
280 		zlog(ZLOG_SYSERROR, "failed to open error_log (%s)", fpm_global_config.error_log);
281 		return -1;
282 	}
283 
284 	if (reopen) {
285 		if (fpm_global_config.daemonize) {
286 			dup2(fd, STDERR_FILENO);
287 		}
288 
289 		dup2(fd, fpm_globals.error_log_fd);
290 		close(fd);
291 		fd = fpm_globals.error_log_fd; /* for FD_CLOSEXEC to work */
292 	} else {
293 		fpm_globals.error_log_fd = fd;
294 		if (fpm_global_config.daemonize) {
295 			zlog_set_fd(fpm_globals.error_log_fd);
296 		}
297 	}
298 	if (0 > fcntl(fd, F_SETFD, fcntl(fd, F_GETFD) | FD_CLOEXEC)) {
299 		zlog(ZLOG_WARNING, "failed to change attribute of error_log");
300 	}
301 	return 0;
302 }
303 /* }}} */
304 
305