2 Unix SMB/CIFS implementation.
4 process model: prefork (n client connections per process)
6 Copyright (C) Andrew Tridgell 1992-2005
7 Copyright (C) James J Myers 2003 <myersjj@samba.org>
8 Copyright (C) Stefan (metze) Metzmacher 2004
9 Copyright (C) Andrew Bartlett 2008 <abartlet@samba.org>
10 Copyright (C) David Disseldorp 2008 <ddiss@sgi.com>
12 This program is free software; you can redistribute it and/or modify
13 it under the terms of the GNU General Public License as published by
14 the Free Software Foundation; either version 3 of the License, or
15 (at your option) any later version.
17 This program is distributed in the hope that it will be useful,
18 but WITHOUT ANY WARRANTY; without even the implied warranty of
19 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 GNU General Public License for more details.
22 You should have received a copy of the GNU General Public License
23 along with this program. If not, see <http://www.gnu.org/licenses/>.
27 #include "lib/events/events.h"
28 #include "../tdb/include/tdb.h"
29 #include "lib/socket/socket.h"
30 #include "smbd/process_model.h"
31 #include "param/secrets.h"
32 #include "system/filesys.h"
33 #include "cluster/cluster.h"
34 #include "param/param.h"
36 #ifdef HAVE_SETPROCTITLE
37 #ifdef HAVE_SETPROCTITLE_H
38 #include <setproctitle.h>
41 #define setproctitle none_setproctitle
42 static int none_setproctitle(const char *fmt, ...) PRINTF_ATTRIBUTE(1, 2);
43 static int none_setproctitle(const char *fmt, ...)
50 called when the process model is selected
52 static void prefork_model_init(struct tevent_context *ev)
54 signal(SIGCHLD, SIG_IGN);
57 static void prefork_reload_after_fork(void)
59 /* tdb needs special fork handling */
60 if (tdb_reopen_all(1) == -1) {
61 DEBUG(0,("prefork_reload_after_fork: tdb_reopen_all failed.\n"));
64 /* Ensure that the forked children do not expose identical random streams */
65 set_need_random_reseed();
69 called when a listening socket becomes readable.
71 static void prefork_accept_connection(struct tevent_context *ev,
72 struct loadparm_context *lp_ctx,
73 struct socket_context *listen_socket,
74 void (*new_conn)(struct tevent_context *,
75 struct loadparm_context *, struct socket_context *,
76 struct server_id , void *),
80 struct socket_context *connected_socket;
83 /* accept an incoming connection. */
84 status = socket_accept(listen_socket, &connected_socket);
85 if (!NT_STATUS_IS_OK(status)) {
89 talloc_steal(private_data, connected_socket);
91 new_conn(ev, lp_ctx, connected_socket, cluster_id(pid, socket_get_fd(connected_socket)), private_data);
95 called to create a new server task
97 static void prefork_new_task(struct tevent_context *ev,
98 struct loadparm_context *lp_ctx,
99 const char *service_name,
100 void (*new_task_fn)(struct tevent_context *, struct loadparm_context *lp_ctx, struct server_id , void *),
106 struct tevent_context *ev2, *ev_parent;
111 /* parent or error code ... go back to the event loop */
117 /* This is now the child code. We need a completely new event_context to work with */
118 ev2 = s4_event_context_init(NULL);
120 /* the service has given us a private pointer that
121 encapsulates the context it needs for this new connection -
122 everything else will be freed */
123 talloc_steal(ev2, private_data);
125 /* this will free all the listening sockets and all state that
126 is not associated with this new connection */
129 setproctitle("task %s server_id[%d]", service_name, pid);
131 prefork_reload_after_fork();
133 /* setup this new connection: process will bind to it's sockets etc */
134 new_task_fn(ev2, lp_ctx, cluster_id(pid, 0), private_data);
136 num_children = lp_parm_int(lp_ctx, NULL, "prefork children", service_name, 0);
137 if (num_children == 0) {
139 /* We don't want any kids hanging around for this one,
140 * let the parent do all the work */
141 event_loop_wait(ev2);
147 /* We are now free to spawn some child proccesses */
149 for (i=0; i < num_children; i++) {
154 } else if (pid == -1) {
158 setproctitle("task %s server_id[%d]", service_name, pid);
160 prefork_reload_after_fork();
162 /* we can't return to the top level here, as that event context is gone,
163 so we now process events in the new event context until there are no
165 event_loop_wait(ev2);
172 /* Don't listen on the sockets we just gave to the children */
175 /* But we need a events system to handle reaping children */
176 ev_parent = s4_event_context_init(NULL);
178 /* TODO: Handle some events... */
180 /* we can't return to the top level here, as that event context is gone,
181 so we now process events in the new event context until there are no
183 event_loop_wait(ev_parent);
185 talloc_free(ev_parent);
191 /* called when a task goes down */
192 _NORETURN_ static void prefork_terminate(struct tevent_context *ev, struct loadparm_context *lp_ctx, const char *reason)
194 DEBUG(2,("prefork_terminate: reason[%s]\n",reason));
197 /* called to set a title of a task or connection */
198 static void prefork_set_title(struct tevent_context *ev, const char *title)
201 setproctitle("%s", title);
207 static const struct model_ops prefork_ops = {
209 .model_init = prefork_model_init,
210 .accept_connection = prefork_accept_connection,
211 .new_task = prefork_new_task,
212 .terminate = prefork_terminate,
213 .set_title = prefork_set_title,
217 initialise the prefork process model, registering ourselves with the process model subsystem
219 NTSTATUS process_model_prefork_init(void)
221 return register_process_model(&prefork_ops);