8a71739d2a51107877c69070ae26fb9179e1c542
[samba.git] / source4 / smbd / process_standard.c
1 /* 
2    Unix SMB/CIFS implementation.
3    process model: standard (1 process per client connection)
4    Copyright (C) Andrew Tridgell 1992-2003
5    Copyright (C) James J Myers 2003 <myersjj@samba.org>
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23
24 /*
25   called when the process model is selected
26 */
27 static void model_startup(void)
28 {
29 }
30
31 /*
32   called when a listening socket becomes readable
33 */
34 static void accept_connection(struct event_context *ev, struct fd_event *fde, time_t t, uint16 flags)
35 {
36         int accepted_fd;
37         struct sockaddr addr;
38         socklen_t in_addrlen = sizeof(addr);
39         pid_t pid;
40         struct model_ops *model_ops = fde->private;
41
42         accepted_fd = accept(fde->fd,&addr,&in_addrlen);
43         if (accepted_fd == -1) {
44                 DEBUG(0,("accept_connection_standard: accept: %s\n",
45                          strerror(errno)));
46                 return;
47         }
48
49         pid = fork();
50
51         if (pid != 0) {
52                 /* parent or error code ... */
53
54                 close(accepted_fd);
55                 /* go back to the event loop */
56                 return;
57         }
58
59         /* Child code ... */
60
61         /* close all the listening sockets */
62         event_remove_fd_all_handler(ev, model_ops->accept_connection);
63         event_remove_fd_all_handler(ev, model_ops->accept_rpc_connection);
64                         
65         /* tdb needs special fork handling */
66         if (tdb_reopen_all() == -1) {
67                 DEBUG(0,("accept_connection_standard: tdb_reopen_all failed.\n"));
68         }
69
70         /* Load DSO's */
71         init_modules();
72                 
73         /* initialize new process */
74         smbd_process_init();
75                 
76         init_smbsession(ev, model_ops, accepted_fd, smbd_read_handler);
77
78         /* return to the event loop */
79 }
80
81 /*
82   called when a rpc listening socket becomes readable
83 */
84 static void accept_rpc_connection(struct event_context *ev, struct fd_event *fde, time_t t, uint16 flags)
85 {
86         int accepted_fd;
87         struct sockaddr addr;
88         socklen_t in_addrlen = sizeof(addr);
89         pid_t pid;
90
91         accepted_fd = accept(fde->fd,&addr,&in_addrlen);
92         if (accepted_fd == -1) {
93                 DEBUG(0,("accept_connection_standard: accept: %s\n",
94                          strerror(errno)));
95                 return;
96         }
97
98         pid = fork();
99
100         if (pid != 0) {
101                 /* parent or error code ... */
102                 close(accepted_fd);
103                 /* go back to the event loop */
104                 return;
105         }
106
107         /* Child code ... */
108
109         /* close all the listening sockets */
110         event_remove_fd_all_handler(ev, accept_connection);
111         event_remove_fd_all_handler(ev, accept_rpc_connection);
112                         
113         init_rpc_session(ev, fde->private, accepted_fd); 
114 }
115
116 /* called when a SMB connection goes down */
117 static void terminate_connection(struct server_context *server, const char *reason) 
118 {
119         server_terminate(server);
120         /* terminate this process */
121         exit(0);
122 }
123
124 /* called when a rpc connection goes down */
125 static void terminate_rpc_connection(void *r, const char *reason) 
126 {
127         rpc_server_terminate(r);
128         /* terminate this process */
129         exit(0);
130 }
131
132 static int get_id(struct request_context *req)
133 {
134         return (int)req->smb->pid;
135 }
136
137 static void standard_exit_server(struct server_context *smb, const char *reason)
138 {
139         DEBUG(1,("standard_exit_server: reason[%s]\n",reason));
140 }
141
142 /*
143   initialise the standard process model, registering ourselves with the process model subsystem
144  */
145 NTSTATUS process_model_standard_init(void)
146 {
147         NTSTATUS ret;
148         struct model_ops ops;
149
150         ZERO_STRUCT(ops);
151
152         /* fill in our name */
153         ops.name = "standard";
154
155         /* fill in all the operations */
156         ops.model_startup = model_startup;
157         ops.accept_connection = accept_connection;
158         ops.accept_rpc_connection = accept_rpc_connection;
159         ops.terminate_connection = terminate_connection;
160         ops.terminate_rpc_connection = terminate_rpc_connection;
161         ops.exit_server = standard_exit_server;
162         ops.get_id = get_id;
163
164         /* register ourselves with the PROCESS_MODEL subsystem. */
165         ret = register_backend("process_model", &ops);
166         if (!NT_STATUS_IS_OK(ret)) {
167                 DEBUG(0,("Failed to register process_model 'standard'!\n"));
168                 return ret;
169         }
170
171         return ret;
172 }