Rework service init functions to pass down service name. This is
[nivanova/samba-autobuild/.git] / source4 / rpc_server / service_rpc.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    smbd-specific dcerpc server code
5
6    Copyright (C) Andrew Tridgell 2003-2005
7    Copyright (C) Stefan (metze) Metzmacher 2004-2005
8    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2004,2007
9    
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14    
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19    
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25 #include "librpc/gen_ndr/ndr_dcerpc.h"
26 #include "auth/auth.h"
27 #include "auth/gensec/gensec.h"
28 #include "lib/util/dlinklist.h"
29 #include "rpc_server/dcerpc_server.h"
30 #include "lib/events/events.h"
31 #include "smbd/service_task.h"
32 #include "smbd/service_stream.h"
33 #include "smbd/service.h"
34 #include "system/filesys.h"
35 #include "libcli/security/security.h"
36 #include "lib/socket/socket.h"
37 #include "lib/messaging/irpc.h"
38 #include "system/network.h"
39 #include "lib/socket/netif.h"
40 #include "build.h"
41 #include "param/param.h"
42
43 struct dcesrv_socket_context {
44         const struct dcesrv_endpoint *endpoint;
45         struct dcesrv_context *dcesrv_ctx;
46 };
47
48 /*
49   write_fn callback for dcesrv_output()
50 */
51 static NTSTATUS dcerpc_write_fn(void *private_data, DATA_BLOB *out, size_t *nwritten)
52 {
53         NTSTATUS status;
54         struct socket_context *sock = talloc_get_type(private_data, struct socket_context);
55         size_t sendlen;
56
57         status = socket_send(sock, out, &sendlen);
58         NT_STATUS_IS_ERR_RETURN(status);
59
60         *nwritten = sendlen;
61         return status;
62 }
63
64 static void dcesrv_terminate_connection(struct dcesrv_connection *dce_conn, const char *reason)
65 {
66         struct stream_connection *srv_conn;
67         srv_conn = talloc_get_type(dce_conn->transport.private_data,
68                                    struct stream_connection);
69
70         stream_terminate_connection(srv_conn, reason);
71 }
72
73 static void dcesrv_sock_report_output_data(struct dcesrv_connection *dcesrv_conn)
74 {
75         struct stream_connection *srv_conn;
76         srv_conn = talloc_get_type(dcesrv_conn->transport.private_data,
77                                    struct stream_connection);
78
79         if (srv_conn && srv_conn->event.fde) {
80                 EVENT_FD_WRITEABLE(srv_conn->event.fde);
81         }
82 }
83
84 static struct socket_address *dcesrv_sock_get_my_addr(struct dcesrv_connection *dcesrv_conn, TALLOC_CTX *mem_ctx)
85 {
86         struct stream_connection *srv_conn;
87         srv_conn = talloc_get_type(dcesrv_conn->transport.private_data,
88                                    struct stream_connection);
89
90         return socket_get_my_addr(srv_conn->socket, mem_ctx);
91 }
92
93 static struct socket_address *dcesrv_sock_get_peer_addr(struct dcesrv_connection *dcesrv_conn, TALLOC_CTX *mem_ctx)
94 {
95         struct stream_connection *srv_conn;
96         srv_conn = talloc_get_type(dcesrv_conn->transport.private_data,
97                                    struct stream_connection);
98
99         return socket_get_peer_addr(srv_conn->socket, mem_ctx);
100 }
101
102 static void dcesrv_sock_accept(struct stream_connection *srv_conn)
103 {
104         NTSTATUS status;
105         struct dcesrv_socket_context *dcesrv_sock = 
106                 talloc_get_type(srv_conn->private, struct dcesrv_socket_context);
107         struct dcesrv_connection *dcesrv_conn = NULL;
108         struct auth_session_info *session_info = NULL;
109
110         status = auth_anonymous_session_info(srv_conn, dcesrv_sock->dcesrv_ctx->lp_ctx, &session_info);
111         if (!NT_STATUS_IS_OK(status)) {
112                 DEBUG(0,("dcesrv_sock_accept: auth_anonymous_session_info failed: %s\n", 
113                         nt_errstr(status)));
114                 stream_terminate_connection(srv_conn, nt_errstr(status));
115                 return;
116         }
117
118         status = dcesrv_endpoint_connect(dcesrv_sock->dcesrv_ctx,
119                                          srv_conn,
120                                          dcesrv_sock->endpoint,
121                                          session_info,
122                                          srv_conn->event.ctx,
123                                          srv_conn->msg_ctx,
124                                          srv_conn->server_id,
125                                          DCESRV_CALL_STATE_FLAG_MAY_ASYNC,
126                                          &dcesrv_conn);
127         if (!NT_STATUS_IS_OK(status)) {
128                 DEBUG(0,("dcesrv_sock_accept: dcesrv_endpoint_connect failed: %s\n", 
129                         nt_errstr(status)));
130                 stream_terminate_connection(srv_conn, nt_errstr(status));
131                 return;
132         }
133
134         dcesrv_conn->transport.private_data             = srv_conn;
135         dcesrv_conn->transport.report_output_data       = dcesrv_sock_report_output_data;
136         dcesrv_conn->transport.get_my_addr              = dcesrv_sock_get_my_addr;
137         dcesrv_conn->transport.get_peer_addr            = dcesrv_sock_get_peer_addr;
138
139         srv_conn->private = dcesrv_conn;
140
141         irpc_add_name(srv_conn->msg_ctx, "rpc_server");
142
143         return; 
144 }
145
146 static void dcesrv_sock_recv(struct stream_connection *conn, uint16_t flags)
147 {
148         NTSTATUS status;
149         struct dcesrv_connection *dce_conn = talloc_get_type(conn->private, struct dcesrv_connection);
150         DATA_BLOB tmp_blob;
151         size_t nread;
152
153         if (dce_conn->processing) {
154                 EVENT_FD_NOT_READABLE(conn->event.fde);
155                 return;
156         }
157
158         tmp_blob = data_blob_talloc(conn->socket, NULL, 0x1000);
159         if (tmp_blob.data == NULL) {
160                 dcesrv_terminate_connection(dce_conn, "out of memory");
161                 return;
162         }
163
164         status = socket_recv(conn->socket, tmp_blob.data, tmp_blob.length, &nread);
165         if (NT_STATUS_IS_ERR(status)) {
166                 dcesrv_terminate_connection(dce_conn, nt_errstr(status));
167                 return;
168         }
169         if (nread == 0) {
170                 talloc_free(tmp_blob.data);
171                 return;
172         }
173
174         tmp_blob.length = nread;
175
176         dce_conn->processing = true;
177         status = dcesrv_input(dce_conn, &tmp_blob);
178         dce_conn->processing = false;
179         talloc_free(tmp_blob.data);
180
181         EVENT_FD_READABLE(conn->event.fde);
182
183         if (!NT_STATUS_IS_OK(status)) {
184                 dcesrv_terminate_connection(dce_conn, nt_errstr(status));
185                 return;
186         }
187
188         if (dce_conn->call_list && dce_conn->call_list->replies) {
189                 EVENT_FD_WRITEABLE(conn->event.fde);
190         }
191 }
192
193 static void dcesrv_sock_send(struct stream_connection *conn, uint16_t flags)
194 {
195         struct dcesrv_connection *dce_conn = talloc_get_type(conn->private, struct dcesrv_connection);
196         NTSTATUS status;
197
198         status = dcesrv_output(dce_conn, conn->socket, dcerpc_write_fn);
199         if (NT_STATUS_IS_ERR(status)) {
200                 dcesrv_terminate_connection(dce_conn, "eof on socket");
201                 return;
202         }
203
204         if (!dce_conn->call_list || !dce_conn->call_list->replies) {
205                 EVENT_FD_NOT_WRITEABLE(conn->event.fde);
206         }
207 }
208
209
210 static const struct stream_server_ops dcesrv_stream_ops = {
211         .name                   = "rpc",
212         .accept_connection      = dcesrv_sock_accept,
213         .recv_handler           = dcesrv_sock_recv,
214         .send_handler           = dcesrv_sock_send,
215 };
216
217
218
219 static NTSTATUS dcesrv_add_ep_unix(struct dcesrv_context *dce_ctx, 
220                                    struct loadparm_context *lp_ctx,
221                                    struct dcesrv_endpoint *e,
222                             struct event_context *event_ctx, const struct model_ops *model_ops)
223 {
224         struct dcesrv_socket_context *dcesrv_sock;
225         uint16_t port = 1;
226         NTSTATUS status;
227
228         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
229         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
230
231         /* remember the endpoint of this socket */
232         dcesrv_sock->endpoint           = e;
233         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
234
235         status = stream_setup_socket(event_ctx, lp_ctx,
236                                      model_ops, &dcesrv_stream_ops, 
237                                      "unix", e->ep_description->endpoint, &port, 
238                                      lp_socket_options(lp_ctx), 
239                                      dcesrv_sock);
240         if (!NT_STATUS_IS_OK(status)) {
241                 DEBUG(0,("service_setup_stream_socket(path=%s) failed - %s\n",
242                          e->ep_description->endpoint, nt_errstr(status)));
243         }
244
245         return status;
246 }
247
248 static NTSTATUS dcesrv_add_ep_ncalrpc(struct dcesrv_context *dce_ctx, 
249                                       struct loadparm_context *lp_ctx,
250                                       struct dcesrv_endpoint *e,
251                                       struct event_context *event_ctx, const struct model_ops *model_ops)
252 {
253         struct dcesrv_socket_context *dcesrv_sock;
254         uint16_t port = 1;
255         char *full_path;
256         NTSTATUS status;
257
258         if (!e->ep_description->endpoint) {
259                 /* No identifier specified: use DEFAULT. 
260                  * DO NOT hardcode this value anywhere else. Rather, specify 
261                  * no endpoint and let the epmapper worry about it. */
262                 e->ep_description->endpoint = talloc_strdup(dce_ctx, "DEFAULT");
263         }
264
265         full_path = talloc_asprintf(dce_ctx, "%s/%s", lp_ncalrpc_dir(lp_ctx), 
266                                     e->ep_description->endpoint);
267
268         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
269         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
270
271         /* remember the endpoint of this socket */
272         dcesrv_sock->endpoint           = e;
273         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
274
275         status = stream_setup_socket(event_ctx, lp_ctx,
276                                      model_ops, &dcesrv_stream_ops, 
277                                      "unix", full_path, &port, 
278                                      lp_socket_options(lp_ctx), 
279                                      dcesrv_sock);
280         if (!NT_STATUS_IS_OK(status)) {
281                 DEBUG(0,("service_setup_stream_socket(identifier=%s,path=%s) failed - %s\n",
282                          e->ep_description->endpoint, full_path, nt_errstr(status)));
283         }
284         return status;
285 }
286
287
288 /*
289   add a socket address to the list of events, one event per dcerpc endpoint
290 */
291 static NTSTATUS add_socket_rpc_pipe_iface(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
292                                          struct event_context *event_ctx, const struct model_ops *model_ops)
293 {
294         struct dcesrv_socket_context *dcesrv_sock;
295         NTSTATUS status;
296                         
297         if (e->ep_description->endpoint == NULL) {
298                 DEBUG(0, ("Endpoint mandatory for named pipes\n"));
299                 return NT_STATUS_INVALID_PARAMETER;
300         }
301
302         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
303         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
304
305         /* remember the endpoint of this socket */
306         dcesrv_sock->endpoint           = e;
307         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
308
309         status = NT_STATUS_OK;
310 #if 0
311
312         status = stream_setup_smb_pipe(event_ctx, model_ops, &dcesrv_stream_ops, 
313                                      e->ep_description->endpoint, dcesrv_sock);
314         if (!NT_STATUS_IS_OK(status)) {
315                 DEBUG(0,("service_setup_stream_socket(path=%s) failed - %s\n", 
316                          e->ep_description->endpoint, nt_errstr(status)));
317         }
318 #endif
319         return status;
320 }
321
322 static NTSTATUS dcesrv_add_ep_np(struct dcesrv_context *dce_ctx, 
323                                  struct loadparm_context *lp_ctx,
324                                  struct dcesrv_endpoint *e,
325                                  struct event_context *event_ctx, const struct model_ops *model_ops)
326 {
327         NTSTATUS status;
328
329         status = add_socket_rpc_pipe_iface(dce_ctx, e, event_ctx, model_ops);
330         NT_STATUS_NOT_OK_RETURN(status);
331
332         return status;
333 }
334
335 /*
336   add a socket address to the list of events, one event per dcerpc endpoint
337 */
338 static NTSTATUS add_socket_rpc_tcp_iface(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
339                                          struct event_context *event_ctx, const struct model_ops *model_ops,
340                                          const char *address)
341 {
342         struct dcesrv_socket_context *dcesrv_sock;
343         uint16_t port = 0;
344         NTSTATUS status;
345                         
346         if (e->ep_description->endpoint) {
347                 port = atoi(e->ep_description->endpoint);
348         }
349
350         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
351         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
352
353         /* remember the endpoint of this socket */
354         dcesrv_sock->endpoint           = e;
355         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
356
357         status = stream_setup_socket(event_ctx, dce_ctx->lp_ctx,
358                                      model_ops, &dcesrv_stream_ops, 
359                                      "ipv4", address, &port, 
360                                      lp_socket_options(dce_ctx->lp_ctx), 
361                                      dcesrv_sock);
362         if (!NT_STATUS_IS_OK(status)) {
363                 DEBUG(0,("service_setup_stream_socket(address=%s,port=%u) failed - %s\n", 
364                          address, port, nt_errstr(status)));
365         }
366
367         if (e->ep_description->endpoint == NULL) {
368                 e->ep_description->endpoint = talloc_asprintf(dce_ctx, "%d", port);
369         }
370
371         return status;
372 }
373
374 static NTSTATUS dcesrv_add_ep_tcp(struct dcesrv_context *dce_ctx, 
375                                   struct loadparm_context *lp_ctx,
376                                   struct dcesrv_endpoint *e,
377                                   struct event_context *event_ctx, const struct model_ops *model_ops)
378 {
379         NTSTATUS status;
380
381         /* Add TCP/IP sockets */
382         if (lp_interfaces(lp_ctx) && lp_bind_interfaces_only(lp_ctx)) {
383                 int num_interfaces;
384                 int i;
385                 struct interface *ifaces;
386
387                 load_interfaces(dce_ctx, lp_interfaces(lp_ctx), &ifaces);
388
389                 num_interfaces = iface_count(ifaces);
390                 for(i = 0; i < num_interfaces; i++) {
391                         const char *address = iface_n_ip(ifaces, i);
392                         status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, address);
393                         NT_STATUS_NOT_OK_RETURN(status);
394                 }
395         } else {
396                 status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, 
397                                                   lp_socket_address(lp_ctx));
398                 NT_STATUS_NOT_OK_RETURN(status);
399         }
400
401         return NT_STATUS_OK;
402 }
403
404
405 static NTSTATUS dcesrv_add_ep(struct dcesrv_context *dce_ctx, 
406                               struct loadparm_context *lp_ctx,
407                               struct dcesrv_endpoint *e,
408                           struct event_context *event_ctx, const struct model_ops *model_ops)
409 {
410         switch (e->ep_description->transport) {
411         case NCACN_UNIX_STREAM:
412                 return dcesrv_add_ep_unix(dce_ctx, lp_ctx, e, event_ctx, model_ops);
413
414         case NCALRPC:
415                 return dcesrv_add_ep_ncalrpc(dce_ctx, lp_ctx, e, event_ctx, model_ops);
416
417         case NCACN_IP_TCP:
418                 return dcesrv_add_ep_tcp(dce_ctx, lp_ctx, e, event_ctx, model_ops);
419
420         case NCACN_NP:
421                 return dcesrv_add_ep_np(dce_ctx, lp_ctx, e, event_ctx, model_ops);
422
423         default:
424                 return NT_STATUS_NOT_SUPPORTED;
425         }
426 }
427
428 /*
429   open the dcerpc server sockets
430 */
431 static void dcesrv_task_init(struct task_server *task)
432 {
433         NTSTATUS status;
434         struct dcesrv_context *dce_ctx;
435         struct dcesrv_endpoint *e;
436
437         task_server_set_title(task, "task[dcesrv]");
438
439         status = dcesrv_init_context(task->event_ctx,
440                                      task->lp_ctx,
441                                      lp_dcerpc_endpoint_servers(task->lp_ctx),
442                                      &dce_ctx);
443         if (!NT_STATUS_IS_OK(status)) goto failed;
444
445         /* Make sure the directory for NCALRPC exists */
446         if (!directory_exist(lp_ncalrpc_dir(task->lp_ctx))) {
447                 mkdir(lp_ncalrpc_dir(task->lp_ctx), 0755);
448         }
449
450         for (e=dce_ctx->endpoint_list;e;e=e->next) {
451                 status = dcesrv_add_ep(dce_ctx, task->lp_ctx, e, task->event_ctx, task->model_ops);
452                 if (!NT_STATUS_IS_OK(status)) goto failed;
453         }
454
455         return;
456 failed:
457         task_server_terminate(task, "Failed to startup dcerpc server task");    
458 }
459
460 /*
461   called on startup of the smb server service It's job is to start
462   listening on all configured sockets
463 */
464 static NTSTATUS dcesrv_init(struct event_context *event_context, 
465                             struct loadparm_context *lp_ctx,
466                             const struct model_ops *model_ops)
467 {       
468         return task_server_startup(event_context, lp_ctx, "rpc",
469                                    model_ops, dcesrv_task_init);
470 }
471
472 NTSTATUS server_service_rpc_init(void)
473 {
474         init_module_fn static_init[] = { STATIC_dcerpc_server_MODULES };
475         init_module_fn *shared_init = load_samba_modules(NULL, global_loadparm, "dcerpc_server");
476
477         run_init_functions(static_init);
478         run_init_functions(shared_init);
479
480         talloc_free(shared_init);
481         
482         return register_server_service("rpc", dcesrv_init);
483 }