r25026: Move param/param.h out of includes.h
[gd/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, &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 NTSTATUS dcesrv_add_ep_unix(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
220                             struct event_context *event_ctx, const struct model_ops *model_ops)
221 {
222         struct dcesrv_socket_context *dcesrv_sock;
223         uint16_t port = 1;
224         NTSTATUS status;
225
226         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
227         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
228
229         /* remember the endpoint of this socket */
230         dcesrv_sock->endpoint           = e;
231         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
232
233         status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops, 
234                                      "unix", e->ep_description->endpoint, &port, 
235                                      dcesrv_sock);
236         if (!NT_STATUS_IS_OK(status)) {
237                 DEBUG(0,("service_setup_stream_socket(path=%s) failed - %s\n",
238                          e->ep_description->endpoint, nt_errstr(status)));
239         }
240
241         return status;
242 }
243
244 NTSTATUS dcesrv_add_ep_ncalrpc(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
245                                struct event_context *event_ctx, const struct model_ops *model_ops)
246 {
247         struct dcesrv_socket_context *dcesrv_sock;
248         uint16_t port = 1;
249         char *full_path;
250         NTSTATUS status;
251
252         if (!e->ep_description->endpoint) {
253                 /* No identifier specified: use DEFAULT. 
254                  * DO NOT hardcode this value anywhere else. Rather, specify 
255                  * no endpoint and let the epmapper worry about it. */
256                 e->ep_description->endpoint = talloc_strdup(dce_ctx, "DEFAULT");
257         }
258
259         full_path = talloc_asprintf(dce_ctx, "%s/%s", lp_ncalrpc_dir(), e->ep_description->endpoint);
260
261         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
262         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
263
264         /* remember the endpoint of this socket */
265         dcesrv_sock->endpoint           = e;
266         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
267
268         status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops, 
269                                      "unix", full_path, &port, dcesrv_sock);
270         if (!NT_STATUS_IS_OK(status)) {
271                 DEBUG(0,("service_setup_stream_socket(identifier=%s,path=%s) failed - %s\n",
272                          e->ep_description->endpoint, full_path, nt_errstr(status)));
273         }
274         return status;
275 }
276
277
278 /*
279   add a socket address to the list of events, one event per dcerpc endpoint
280 */
281 static NTSTATUS add_socket_rpc_pipe_iface(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
282                                          struct event_context *event_ctx, const struct model_ops *model_ops)
283 {
284         struct dcesrv_socket_context *dcesrv_sock;
285         NTSTATUS status;
286                         
287         if (e->ep_description->endpoint == NULL) {
288                 DEBUG(0, ("Endpoint mandatory for named pipes\n"));
289                 return NT_STATUS_INVALID_PARAMETER;
290         }
291
292         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
293         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
294
295         /* remember the endpoint of this socket */
296         dcesrv_sock->endpoint           = e;
297         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
298
299         status = NT_STATUS_OK;
300 #if 0
301
302         status = stream_setup_smb_pipe(event_ctx, model_ops, &dcesrv_stream_ops, 
303                                      e->ep_description->endpoint, dcesrv_sock);
304         if (!NT_STATUS_IS_OK(status)) {
305                 DEBUG(0,("service_setup_stream_socket(path=%s) failed - %s\n", 
306                          e->ep_description->endpoint, nt_errstr(status)));
307         }
308 #endif
309         return status;
310 }
311
312 NTSTATUS dcesrv_add_ep_np(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
313                                    struct event_context *event_ctx, const struct model_ops *model_ops)
314 {
315         NTSTATUS status;
316
317         status = add_socket_rpc_pipe_iface(dce_ctx, e, event_ctx, model_ops);
318         NT_STATUS_NOT_OK_RETURN(status);
319
320         return status;
321 }
322
323 /*
324   add a socket address to the list of events, one event per dcerpc endpoint
325 */
326 static NTSTATUS add_socket_rpc_tcp_iface(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
327                                          struct event_context *event_ctx, const struct model_ops *model_ops,
328                                          const char *address)
329 {
330         struct dcesrv_socket_context *dcesrv_sock;
331         uint16_t port = 0;
332         NTSTATUS status;
333                         
334         if (e->ep_description->endpoint) {
335                 port = atoi(e->ep_description->endpoint);
336         }
337
338         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
339         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
340
341         /* remember the endpoint of this socket */
342         dcesrv_sock->endpoint           = e;
343         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
344
345         status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops, 
346                                      "ipv4", address, &port, dcesrv_sock);
347         if (!NT_STATUS_IS_OK(status)) {
348                 DEBUG(0,("service_setup_stream_socket(address=%s,port=%u) failed - %s\n", 
349                          address, port, nt_errstr(status)));
350         }
351
352         if (e->ep_description->endpoint == NULL) {
353                 e->ep_description->endpoint = talloc_asprintf(dce_ctx, "%d", port);
354         }
355
356         return status;
357 }
358
359 NTSTATUS dcesrv_add_ep_tcp(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
360                                    struct event_context *event_ctx, const struct model_ops *model_ops)
361 {
362         NTSTATUS status;
363
364         /* Add TCP/IP sockets */
365         if (lp_interfaces() && lp_bind_interfaces_only()) {
366                 int num_interfaces = iface_count();
367                 int i;
368                 for(i = 0; i < num_interfaces; i++) {
369                         const char *address = iface_n_ip(i);
370                         status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, address);
371                         NT_STATUS_NOT_OK_RETURN(status);
372                 }
373         } else {
374                 status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, lp_socket_address());
375                 NT_STATUS_NOT_OK_RETURN(status);
376         }
377
378         return NT_STATUS_OK;
379 }
380
381
382 NTSTATUS dcesrv_add_ep(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
383                           struct event_context *event_ctx, const struct model_ops *model_ops)
384 {
385         switch (e->ep_description->transport) {
386         case NCACN_UNIX_STREAM:
387                 return dcesrv_add_ep_unix(dce_ctx, e, event_ctx, model_ops);
388
389         case NCALRPC:
390                 return dcesrv_add_ep_ncalrpc(dce_ctx, e, event_ctx, model_ops);
391
392         case NCACN_IP_TCP:
393                 return dcesrv_add_ep_tcp(dce_ctx, e, event_ctx, model_ops);
394
395         case NCACN_NP:
396                 return dcesrv_add_ep_np(dce_ctx, e, event_ctx, model_ops);
397
398         default:
399                 return NT_STATUS_NOT_SUPPORTED;
400         }
401 }
402
403 /*
404   open the dcerpc server sockets
405 */
406 static void dcesrv_task_init(struct task_server *task)
407 {
408         NTSTATUS status;
409         struct dcesrv_context *dce_ctx;
410         struct dcesrv_endpoint *e;
411
412         task_server_set_title(task, "task[dcesrv]");
413
414         status = dcesrv_init_context(task->event_ctx,
415                                      lp_dcerpc_endpoint_servers(),
416                                      &dce_ctx);
417         if (!NT_STATUS_IS_OK(status)) goto failed;
418
419         /* Make sure the directory for NCALRPC exists */
420         if (!directory_exist(lp_ncalrpc_dir())) {
421                 mkdir(lp_ncalrpc_dir(), 0755);
422         }
423
424         for (e=dce_ctx->endpoint_list;e;e=e->next) {
425                 status = dcesrv_add_ep(dce_ctx, e, task->event_ctx, task->model_ops);
426                 if (!NT_STATUS_IS_OK(status)) goto failed;
427         }
428
429         return;
430 failed:
431         task_server_terminate(task, "Failed to startup dcerpc server task");    
432 }
433
434 /*
435   called on startup of the smb server service It's job is to start
436   listening on all configured sockets
437 */
438 static NTSTATUS dcesrv_init(struct event_context *event_context, 
439                             const struct model_ops *model_ops)
440 {       
441         return task_server_startup(event_context, model_ops, dcesrv_task_init);
442 }
443
444 NTSTATUS server_service_rpc_init(void)
445 {
446         init_module_fn static_init[] = STATIC_dcerpc_server_MODULES;
447         init_module_fn *shared_init = load_samba_modules(NULL, "dcerpc_server");
448
449         run_init_functions(static_init);
450         run_init_functions(shared_init);
451
452         talloc_free(shared_init);
453         
454         return register_server_service("rpc", dcesrv_init);
455 }
456
457