s3:vfs_gpfs: Export disk_norm function
[Samba/bb.git] / source3 / rpc_server / rpc_server.c
blob89885b9230ef82912a9234ff8f6bf02524b6b3ce
1 /*
2 Unix SMB/Netbios implementation.
3 Generic infrstructure for RPC Daemons
4 Copyright (C) Simo Sorce 2010
5 Copyright (C) Andrew Bartlett 2011
6 Copyright (C) Andreas Schneider 2011
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "rpc_server/rpc_pipes.h"
24 #include "rpc_server/rpc_server.h"
25 #include "rpc_server/rpc_config.h"
26 #include "rpc_dce.h"
27 #include "librpc/gen_ndr/netlogon.h"
28 #include "librpc/gen_ndr/auth.h"
29 #include "lib/tsocket/tsocket.h"
30 #include "libcli/named_pipe_auth/npa_tstream.h"
31 #include "../auth/auth_sam_reply.h"
32 #include "auth.h"
33 #include "rpc_server/rpc_ncacn_np.h"
34 #include "rpc_server/srv_pipe_hnd.h"
35 #include "rpc_server/srv_pipe.h"
37 #define SERVER_TCP_LOW_PORT 1024
38 #define SERVER_TCP_HIGH_PORT 1300
40 static NTSTATUS auth_anonymous_session_info(TALLOC_CTX *mem_ctx,
41 struct auth_session_info **session_info)
43 NTSTATUS status;
45 status = make_session_info_guest(mem_ctx, session_info);
46 if (!NT_STATUS_IS_OK(status)) {
47 return status;
50 return NT_STATUS_OK;
53 /* Creates a pipes_struct and initializes it with the information
54 * sent from the client */
55 static int make_server_pipes_struct(TALLOC_CTX *mem_ctx,
56 struct messaging_context *msg_ctx,
57 const char *pipe_name,
58 enum dcerpc_transport_t transport,
59 bool ncalrpc_as_system,
60 const struct tsocket_address *local_address,
61 const struct tsocket_address *remote_address,
62 struct auth_session_info *session_info,
63 struct pipes_struct **_p,
64 int *perrno)
66 struct pipes_struct *p;
67 NTSTATUS status;
68 int ret;
70 ret = make_base_pipes_struct(mem_ctx, msg_ctx, pipe_name,
71 transport, RPC_LITTLE_ENDIAN,
72 ncalrpc_as_system,
73 remote_address, local_address, &p);
74 if (ret) {
75 *perrno = ret;
76 return -1;
79 if (session_info->unix_token && session_info->unix_info && session_info->security_token) {
80 /* Don't call create_local_token(), we already have the full details here */
81 p->session_info = talloc_steal(p, session_info);
83 } else {
84 struct auth_user_info_dc *auth_user_info_dc;
85 struct auth_serversupplied_info *server_info;
86 struct netr_SamInfo3 *info3;
88 /* Fake up an auth_user_info_dc for now, to make an info3, to make the session_info structure */
89 auth_user_info_dc = talloc_zero(p, struct auth_user_info_dc);
90 if (!auth_user_info_dc) {
91 TALLOC_FREE(p);
92 *perrno = ENOMEM;
93 return -1;
96 auth_user_info_dc->num_sids = session_info->security_token->num_sids;
97 auth_user_info_dc->sids = session_info->security_token->sids;
98 auth_user_info_dc->info = session_info->info;
99 auth_user_info_dc->user_session_key = session_info->session_key;
101 /* This creates the input structure that make_server_info_info3 is looking for */
102 status = auth_convert_user_info_dc_saminfo3(p, auth_user_info_dc,
103 &info3);
105 if (!NT_STATUS_IS_OK(status)) {
106 DEBUG(1, ("Failed to convert auth_user_info_dc into netr_SamInfo3\n"));
107 TALLOC_FREE(p);
108 *perrno = EINVAL;
109 return -1;
112 status = make_server_info_info3(p,
113 info3->base.account_name.string,
114 info3->base.logon_domain.string,
115 &server_info, info3);
116 if (!NT_STATUS_IS_OK(status)) {
117 DEBUG(1, ("Failed to init server info\n"));
118 TALLOC_FREE(p);
119 *perrno = EINVAL;
120 return -1;
124 * Some internal functions need a local token to determine access to
125 * resources.
127 status = create_local_token(p, server_info, &session_info->session_key, info3->base.account_name.string,
128 &p->session_info);
129 talloc_free(server_info);
130 if (!NT_STATUS_IS_OK(status)) {
131 DEBUG(1, ("Failed to init local auth token\n"));
132 TALLOC_FREE(p);
133 *perrno = EINVAL;
134 return -1;
138 *_p = p;
139 return 0;
142 /* Start listening on the appropriate unix socket and setup all is needed to
143 * dispatch requests to the pipes rpc implementation */
145 struct dcerpc_ncacn_listen_state {
146 struct ndr_syntax_id syntax_id;
148 int fd;
149 union {
150 char *name;
151 uint16_t port;
152 } ep;
154 struct tevent_context *ev_ctx;
155 struct messaging_context *msg_ctx;
156 dcerpc_ncacn_disconnect_fn disconnect_fn;
159 static void named_pipe_listener(struct tevent_context *ev,
160 struct tevent_fd *fde,
161 uint16_t flags,
162 void *private_data);
164 int create_named_pipe_socket(const char *pipe_name)
166 char *np_dir = NULL;
167 int fd = -1;
170 * As lp_ncalrpc_dir() should have 0755, but
171 * lp_ncalrpc_dir()/np should have 0700, we need to
172 * create lp_ncalrpc_dir() first.
174 if (!directory_create_or_exist(lp_ncalrpc_dir(), geteuid(), 0755)) {
175 DEBUG(0, ("Failed to create pipe directory %s - %s\n",
176 lp_ncalrpc_dir(), strerror(errno)));
177 goto out;
180 np_dir = talloc_asprintf(talloc_tos(), "%s/np", lp_ncalrpc_dir());
181 if (!np_dir) {
182 DEBUG(0, ("Out of memory\n"));
183 goto out;
186 if (!directory_create_or_exist(np_dir, geteuid(), 0700)) {
187 DEBUG(0, ("Failed to create pipe directory %s - %s\n",
188 np_dir, strerror(errno)));
189 goto out;
192 fd = create_pipe_sock(np_dir, pipe_name, 0700);
193 if (fd == -1) {
194 DEBUG(0, ("Failed to create pipe socket! [%s/%s]\n",
195 np_dir, pipe_name));
196 goto out;
199 DEBUG(10, ("Openened pipe socket fd %d for %s\n", fd, pipe_name));
201 out:
202 talloc_free(np_dir);
203 return fd;
206 bool setup_named_pipe_socket(const char *pipe_name,
207 struct tevent_context *ev_ctx,
208 struct messaging_context *msg_ctx)
210 struct dcerpc_ncacn_listen_state *state;
211 struct tevent_fd *fde;
212 int rc;
214 state = talloc(ev_ctx, struct dcerpc_ncacn_listen_state);
215 if (!state) {
216 DEBUG(0, ("Out of memory\n"));
217 return false;
219 state->ep.name = talloc_strdup(state, pipe_name);
220 if (state->ep.name == NULL) {
221 DEBUG(0, ("Out of memory\n"));
222 goto out;
224 state->fd = create_named_pipe_socket(pipe_name);
225 if (state->fd == -1) {
226 goto out;
229 rc = listen(state->fd, 5);
230 if (rc < 0) {
231 DEBUG(0, ("Failed to listen on pipe socket %s: %s\n",
232 pipe_name, strerror(errno)));
233 goto out;
236 state->ev_ctx = ev_ctx;
237 state->msg_ctx = msg_ctx;
239 DEBUG(10, ("Openened pipe socket fd %d for %s\n",
240 state->fd, pipe_name));
242 fde = tevent_add_fd(ev_ctx,
243 state, state->fd, TEVENT_FD_READ,
244 named_pipe_listener, state);
245 if (!fde) {
246 DEBUG(0, ("Failed to add event handler!\n"));
247 goto out;
250 tevent_fd_set_auto_close(fde);
251 return true;
253 out:
254 if (state->fd != -1) {
255 close(state->fd);
257 TALLOC_FREE(state);
258 return false;
261 static void named_pipe_listener(struct tevent_context *ev,
262 struct tevent_fd *fde,
263 uint16_t flags,
264 void *private_data)
266 struct dcerpc_ncacn_listen_state *state =
267 talloc_get_type_abort(private_data,
268 struct dcerpc_ncacn_listen_state);
269 struct sockaddr_un sunaddr;
270 socklen_t len;
271 int sd = -1;
273 /* TODO: should we have a limit to the number of clients ? */
275 len = sizeof(sunaddr);
277 sd = accept(state->fd,
278 (struct sockaddr *)(void *)&sunaddr, &len);
280 if (sd == -1) {
281 if (errno != EINTR) {
282 DEBUG(6, ("Failed to get a valid socket [%s]\n",
283 strerror(errno)));
285 return;
288 DEBUG(6, ("Accepted socket %d\n", sd));
290 named_pipe_accept_function(state->ev_ctx,
291 state->msg_ctx,
292 state->ep.name,
293 sd, NULL, 0);
297 /* This is the core of the rpc server.
298 * Accepts connections from clients and process requests using the appropriate
299 * dispatcher table. */
301 struct named_pipe_client {
302 const char *pipe_name;
304 struct tevent_context *ev;
305 struct messaging_context *msg_ctx;
307 uint16_t file_type;
308 uint16_t device_state;
309 uint64_t allocation_size;
311 struct tstream_context *tstream;
313 struct tsocket_address *client;
314 char *client_name;
315 struct tsocket_address *server;
316 char *server_name;
318 struct auth_session_info *session_info;
320 struct pipes_struct *p;
322 struct tevent_queue *write_queue;
324 struct iovec *iov;
325 size_t count;
327 named_pipe_termination_fn *term_fn;
328 void *private_data;
331 static int named_pipe_destructor(struct named_pipe_client *npc)
333 if (npc->term_fn) {
334 npc->term_fn(npc->private_data);
336 return 0;
339 static void named_pipe_accept_done(struct tevent_req *subreq);
341 void named_pipe_accept_function(struct tevent_context *ev_ctx,
342 struct messaging_context *msg_ctx,
343 const char *pipe_name, int fd,
344 named_pipe_termination_fn *term_fn,
345 void *private_data)
347 struct named_pipe_client *npc;
348 struct tstream_context *plain;
349 struct tevent_req *subreq;
350 int ret;
352 npc = talloc_zero(ev_ctx, struct named_pipe_client);
353 if (!npc) {
354 DEBUG(0, ("Out of memory!\n"));
355 close(fd);
356 return;
359 npc->pipe_name = talloc_strdup(npc, pipe_name);
360 if (npc->pipe_name == NULL) {
361 DEBUG(0, ("Out of memory!\n"));
362 TALLOC_FREE(npc);
363 close(fd);
364 return;
366 npc->ev = ev_ctx;
367 npc->msg_ctx = msg_ctx;
368 npc->term_fn = term_fn;
369 npc->private_data = private_data;
371 talloc_set_destructor(npc, named_pipe_destructor);
373 /* make sure socket is in NON blocking state */
374 ret = set_blocking(fd, false);
375 if (ret != 0) {
376 DEBUG(2, ("Failed to make socket non-blocking\n"));
377 TALLOC_FREE(npc);
378 close(fd);
379 return;
382 ret = tstream_bsd_existing_socket(npc, fd, &plain);
383 if (ret != 0) {
384 DEBUG(2, ("Failed to create tstream socket\n"));
385 TALLOC_FREE(npc);
386 close(fd);
387 return;
390 npc->file_type = FILE_TYPE_MESSAGE_MODE_PIPE;
391 npc->device_state = 0xff | 0x0400 | 0x0100;
392 npc->allocation_size = 4096;
394 subreq = tstream_npa_accept_existing_send(npc, npc->ev, plain,
395 npc->file_type,
396 npc->device_state,
397 npc->allocation_size);
398 if (!subreq) {
399 DEBUG(2, ("Failed to start async accept procedure\n"));
400 TALLOC_FREE(npc);
401 close(fd);
402 return;
404 tevent_req_set_callback(subreq, named_pipe_accept_done, npc);
407 static void named_pipe_packet_process(struct tevent_req *subreq);
408 static void named_pipe_packet_done(struct tevent_req *subreq);
410 static void named_pipe_accept_done(struct tevent_req *subreq)
412 struct auth_session_info_transport *session_info_transport;
413 struct named_pipe_client *npc =
414 tevent_req_callback_data(subreq, struct named_pipe_client);
415 int error;
416 int ret;
418 ret = tstream_npa_accept_existing_recv(subreq, &error, npc,
419 &npc->tstream,
420 &npc->client,
421 &npc->client_name,
422 &npc->server,
423 &npc->server_name,
424 &session_info_transport);
426 npc->session_info = talloc_move(npc, &session_info_transport->session_info);
428 TALLOC_FREE(subreq);
429 if (ret != 0) {
430 DEBUG(2, ("Failed to accept named pipe connection! (%s)\n",
431 strerror(error)));
432 TALLOC_FREE(npc);
433 return;
436 ret = make_server_pipes_struct(npc,
437 npc->msg_ctx,
438 npc->pipe_name, NCACN_NP,
439 false, npc->server, npc->client, npc->session_info,
440 &npc->p, &error);
441 if (ret != 0) {
442 DEBUG(2, ("Failed to create pipes_struct! (%s)\n",
443 strerror(error)));
444 goto fail;
447 npc->write_queue = tevent_queue_create(npc, "np_server_write_queue");
448 if (!npc->write_queue) {
449 DEBUG(2, ("Failed to set up write queue!\n"));
450 goto fail;
453 /* And now start receiving and processing packets */
454 subreq = dcerpc_read_ncacn_packet_send(npc, npc->ev, npc->tstream);
455 if (!subreq) {
456 DEBUG(2, ("Failed to start receving packets\n"));
457 goto fail;
459 tevent_req_set_callback(subreq, named_pipe_packet_process, npc);
460 return;
462 fail:
463 DEBUG(2, ("Fatal error. Terminating client(%s) connection!\n",
464 npc->client_name));
465 /* terminate client connection */
466 talloc_free(npc);
467 return;
470 static void named_pipe_packet_process(struct tevent_req *subreq)
472 struct named_pipe_client *npc =
473 tevent_req_callback_data(subreq, struct named_pipe_client);
474 struct _output_data *out = &npc->p->out_data;
475 DATA_BLOB recv_buffer = data_blob_null;
476 struct ncacn_packet *pkt;
477 NTSTATUS status;
478 ssize_t data_left;
479 ssize_t data_used;
480 char *data;
481 uint32_t to_send;
482 size_t i;
483 bool ok;
485 status = dcerpc_read_ncacn_packet_recv(subreq, npc, &pkt, &recv_buffer);
486 TALLOC_FREE(subreq);
487 if (!NT_STATUS_IS_OK(status)) {
488 goto fail;
491 data_left = recv_buffer.length;
492 data = (char *)recv_buffer.data;
494 while (data_left) {
496 data_used = process_incoming_data(npc->p, data, data_left);
497 if (data_used < 0) {
498 DEBUG(3, ("Failed to process dceprc request!\n"));
499 status = NT_STATUS_UNEXPECTED_IO_ERROR;
500 goto fail;
503 data_left -= data_used;
504 data += data_used;
507 /* Do not leak this buffer, npc is a long lived context */
508 talloc_free(recv_buffer.data);
509 talloc_free(pkt);
511 /* this is needed because of the way DCERPC Binds work in
512 * the RPC marshalling code */
513 to_send = out->frag.length - out->current_pdu_sent;
514 if (to_send > 0) {
516 npc->iov = talloc_zero(npc, struct iovec);
517 if (!npc->iov) {
518 status = NT_STATUS_NO_MEMORY;
519 goto fail;
521 npc->count = 1;
523 npc->iov[0].iov_base = out->frag.data
524 + out->current_pdu_sent;
525 npc->iov[0].iov_len = to_send;
527 out->current_pdu_sent += to_send;
530 /* this condition is false for bind packets, or when we haven't
531 * yet got a full request, and need to wait for more data from
532 * the client */
533 while (out->data_sent_length < out->rdata.length) {
535 ok = create_next_pdu(npc->p);
536 if (!ok) {
537 DEBUG(3, ("Failed to create next PDU!\n"));
538 status = NT_STATUS_UNEXPECTED_IO_ERROR;
539 goto fail;
542 npc->iov = talloc_realloc(npc, npc->iov,
543 struct iovec, npc->count + 1);
544 if (!npc->iov) {
545 status = NT_STATUS_NO_MEMORY;
546 goto fail;
549 npc->iov[npc->count].iov_base = out->frag.data;
550 npc->iov[npc->count].iov_len = out->frag.length;
552 npc->count++;
555 /* we still don't have a complete request, go back and wait for more
556 * data */
557 if (npc->count == 0) {
558 /* Wait for the next packet */
559 subreq = dcerpc_read_ncacn_packet_send(npc, npc->ev, npc->tstream);
560 if (!subreq) {
561 DEBUG(2, ("Failed to start receving packets\n"));
562 status = NT_STATUS_NO_MEMORY;
563 goto fail;
565 tevent_req_set_callback(subreq, named_pipe_packet_process, npc);
566 return;
569 DEBUG(10, ("Sending %u fragments in a total of %u bytes\n",
570 (unsigned int)npc->count,
571 (unsigned int)npc->p->out_data.data_sent_length));
573 for (i = 0; i < npc->count; i++) {
574 DEBUG(10, ("Sending PDU number: %d, PDU Length: %u\n",
575 (unsigned int)i,
576 (unsigned int)npc->iov[i].iov_len));
577 dump_data(11, (const uint8_t *)npc->iov[i].iov_base,
578 npc->iov[i].iov_len);
580 subreq = tstream_writev_queue_send(npc,
581 npc->ev,
582 npc->tstream,
583 npc->write_queue,
584 (npc->iov + i),
586 if (!subreq) {
587 DEBUG(2, ("Failed to send packet\n"));
588 status = NT_STATUS_NO_MEMORY;
589 goto fail;
591 tevent_req_set_callback(subreq, named_pipe_packet_done, npc);
594 return;
596 fail:
597 DEBUG(2, ("Fatal error(%s). "
598 "Terminating client(%s) connection!\n",
599 nt_errstr(status), npc->client_name));
600 /* terminate client connection */
601 talloc_free(npc);
602 return;
605 static void named_pipe_packet_done(struct tevent_req *subreq)
607 struct named_pipe_client *npc =
608 tevent_req_callback_data(subreq, struct named_pipe_client);
609 int sys_errno;
610 int ret;
612 ret = tstream_writev_queue_recv(subreq, &sys_errno);
613 TALLOC_FREE(subreq);
614 if (ret == -1) {
615 DEBUG(2, ("Writev failed!\n"));
616 goto fail;
619 if (tevent_queue_length(npc->write_queue) > 0) {
620 return;
623 /* clear out any data that may have been left around */
624 npc->count = 0;
625 TALLOC_FREE(npc->iov);
626 data_blob_free(&npc->p->in_data.data);
627 data_blob_free(&npc->p->out_data.frag);
628 data_blob_free(&npc->p->out_data.rdata);
630 talloc_free_children(npc->p->mem_ctx);
632 /* Wait for the next packet */
633 subreq = dcerpc_read_ncacn_packet_send(npc, npc->ev, npc->tstream);
634 if (!subreq) {
635 DEBUG(2, ("Failed to start receving packets\n"));
636 sys_errno = ENOMEM;
637 goto fail;
639 tevent_req_set_callback(subreq, named_pipe_packet_process, npc);
640 return;
642 fail:
643 DEBUG(2, ("Fatal error(%s). "
644 "Terminating client(%s) connection!\n",
645 strerror(sys_errno), npc->client_name));
646 /* terminate client connection */
647 talloc_free(npc);
648 return;
651 /********************************************************************
652 * Start listening on the tcp/ip socket
653 ********************************************************************/
655 static void dcerpc_ncacn_tcpip_listener(struct tevent_context *ev,
656 struct tevent_fd *fde,
657 uint16_t flags,
658 void *private_data);
660 int create_tcpip_socket(const struct sockaddr_storage *ifss, uint16_t *port)
662 int fd = -1;
664 if (*port == 0) {
665 uint16_t i;
667 for (i = SERVER_TCP_LOW_PORT; i <= SERVER_TCP_HIGH_PORT; i++) {
668 fd = open_socket_in(SOCK_STREAM,
671 ifss,
672 false);
673 if (fd > 0) {
674 *port = i;
675 break;
678 } else {
679 fd = open_socket_in(SOCK_STREAM,
680 *port,
682 ifss,
683 true);
685 if (fd == -1) {
686 DEBUG(0, ("Failed to create socket on port %u!\n", *port));
687 return -1;
690 DEBUG(10, ("Opened tcpip socket fd %d for port %u\n", fd, *port));
692 return fd;
695 uint16_t setup_dcerpc_ncacn_tcpip_socket(struct tevent_context *ev_ctx,
696 struct messaging_context *msg_ctx,
697 const struct sockaddr_storage *ifss,
698 uint16_t port)
700 struct dcerpc_ncacn_listen_state *state;
701 struct tevent_fd *fde;
702 int rc;
704 state = talloc(ev_ctx, struct dcerpc_ncacn_listen_state);
705 if (state == NULL) {
706 DEBUG(0, ("setup_dcerpc_ncacn_tcpip_socket: Out of memory\n"));
707 return 0;
710 state->fd = -1;
711 state->ep.port = port;
712 state->disconnect_fn = NULL;
714 state->fd = create_tcpip_socket(ifss, &state->ep.port);
715 if (state->fd == -1) {
716 goto out;
719 state->ev_ctx = ev_ctx;
720 state->msg_ctx = msg_ctx;
722 /* ready to listen */
723 set_socket_options(state->fd, "SO_KEEPALIVE");
724 set_socket_options(state->fd, lp_socket_options());
726 /* Set server socket to non-blocking for the accept. */
727 set_blocking(state->fd, false);
729 rc = listen(state->fd, SMBD_LISTEN_BACKLOG);
730 if (rc == -1) {
731 DEBUG(0,("setup_tcpip_socket: listen - %s\n", strerror(errno)));
732 goto out;
735 DEBUG(10, ("setup_tcpip_socket: openened socket fd %d for port %u\n",
736 state->fd, state->ep.port));
738 fde = tevent_add_fd(state->ev_ctx,
739 state,
740 state->fd,
741 TEVENT_FD_READ,
742 dcerpc_ncacn_tcpip_listener,
743 state);
744 if (fde == NULL) {
745 DEBUG(0, ("setup_tcpip_socket: Failed to add event handler!\n"));
746 goto out;
749 tevent_fd_set_auto_close(fde);
751 return state->ep.port;
752 out:
753 if (state->fd != -1) {
754 close(state->fd);
756 TALLOC_FREE(state);
758 return 0;
761 static void dcerpc_ncacn_tcpip_listener(struct tevent_context *ev,
762 struct tevent_fd *fde,
763 uint16_t flags,
764 void *private_data)
766 struct dcerpc_ncacn_listen_state *state =
767 talloc_get_type_abort(private_data,
768 struct dcerpc_ncacn_listen_state);
769 struct tsocket_address *cli_addr = NULL;
770 struct tsocket_address *srv_addr = NULL;
771 struct sockaddr_storage addr;
772 socklen_t in_addrlen = sizeof(addr);
773 int s = -1;
774 int rc;
776 s = accept(state->fd, (struct sockaddr *)(void *) &addr, &in_addrlen);
777 if (s == -1) {
778 if (errno != EINTR) {
779 DEBUG(0,("tcpip_listener accept: %s\n",
780 strerror(errno)));
782 return;
785 rc = tsocket_address_bsd_from_sockaddr(state,
786 (struct sockaddr *)(void *) &addr,
787 in_addrlen,
788 &cli_addr);
789 if (rc < 0) {
790 close(s);
791 return;
794 rc = getsockname(s, (struct sockaddr *)(void *) &addr, &in_addrlen);
795 if (rc < 0) {
796 close(s);
797 return;
800 rc = tsocket_address_bsd_from_sockaddr(state,
801 (struct sockaddr *)(void *) &addr,
802 in_addrlen,
803 &srv_addr);
804 if (rc < 0) {
805 close(s);
806 return;
809 DEBUG(6, ("tcpip_listener: Accepted socket %d\n", s));
811 dcerpc_ncacn_accept(state->ev_ctx,
812 state->msg_ctx,
813 NCACN_IP_TCP,
814 NULL,
815 cli_addr,
816 srv_addr,
818 NULL);
821 /********************************************************************
822 * Start listening on the ncalrpc socket
823 ********************************************************************/
825 static void dcerpc_ncalrpc_listener(struct tevent_context *ev,
826 struct tevent_fd *fde,
827 uint16_t flags,
828 void *private_data);
830 int create_dcerpc_ncalrpc_socket(const char *name)
832 int fd = -1;
834 if (name == NULL) {
835 name = "DEFAULT";
838 if (!directory_create_or_exist(lp_ncalrpc_dir(), geteuid(), 0755)) {
839 DEBUG(0, ("Failed to create ncalrpc directory %s - %s\n",
840 lp_ncalrpc_dir(), strerror(errno)));
841 return -1;
844 fd = create_pipe_sock(lp_ncalrpc_dir(), name, 0755);
845 if (fd == -1) {
846 DEBUG(0, ("Failed to create ncalrpc socket! [%s/%s]\n",
847 lp_ncalrpc_dir(), name));
848 return -1;
851 DEBUG(10, ("Openened ncalrpc socket fd %d for %s\n", fd, name));
853 return fd;
856 bool setup_dcerpc_ncalrpc_socket(struct tevent_context *ev_ctx,
857 struct messaging_context *msg_ctx,
858 const char *name,
859 dcerpc_ncacn_disconnect_fn fn)
861 struct dcerpc_ncacn_listen_state *state;
862 struct tevent_fd *fde;
863 int rc;
865 state = talloc(ev_ctx, struct dcerpc_ncacn_listen_state);
866 if (state == NULL) {
867 DEBUG(0, ("Out of memory\n"));
868 return false;
871 state->fd = -1;
872 state->disconnect_fn = fn;
874 if (name == NULL) {
875 name = "DEFAULT";
878 state->ep.name = talloc_strdup(state, name);
879 if (state->ep.name == NULL) {
880 DEBUG(0, ("Out of memory\n"));
881 talloc_free(state);
882 return false;
885 state->fd = create_dcerpc_ncalrpc_socket(name);
886 if (state->fd == -1) {
887 goto out;
890 rc = listen(state->fd, 5);
891 if (rc < 0) {
892 DEBUG(0, ("Failed to listen on ncalrpc socket %s: %s\n",
893 name, strerror(errno)));
894 goto out;
897 state->ev_ctx = ev_ctx;
898 state->msg_ctx = msg_ctx;
900 /* Set server socket to non-blocking for the accept. */
901 set_blocking(state->fd, false);
903 fde = tevent_add_fd(state->ev_ctx,
904 state,
905 state->fd,
906 TEVENT_FD_READ,
907 dcerpc_ncalrpc_listener,
908 state);
909 if (fde == NULL) {
910 DEBUG(0, ("Failed to add event handler for ncalrpc!\n"));
911 goto out;
914 tevent_fd_set_auto_close(fde);
916 return true;
917 out:
918 if (state->fd != -1) {
919 close(state->fd);
921 TALLOC_FREE(state);
923 return 0;
926 static void dcerpc_ncalrpc_listener(struct tevent_context *ev,
927 struct tevent_fd *fde,
928 uint16_t flags,
929 void *private_data)
931 struct dcerpc_ncacn_listen_state *state =
932 talloc_get_type_abort(private_data,
933 struct dcerpc_ncacn_listen_state);
934 struct tsocket_address *cli_addr = NULL;
935 struct sockaddr_un sunaddr;
936 struct sockaddr *addr = (struct sockaddr *)(void *)&sunaddr;
937 socklen_t len = sizeof(sunaddr);
938 int sd = -1;
939 int rc;
941 ZERO_STRUCT(sunaddr);
943 sd = accept(state->fd, addr, &len);
944 if (sd == -1) {
945 if (errno != EINTR) {
946 DEBUG(0, ("ncalrpc accept() failed: %s\n", strerror(errno)));
948 return;
951 rc = tsocket_address_bsd_from_sockaddr(state,
952 addr, len,
953 &cli_addr);
954 if (rc < 0) {
955 close(sd);
956 return;
959 DEBUG(10, ("Accepted ncalrpc socket %d\n", sd));
961 dcerpc_ncacn_accept(state->ev_ctx,
962 state->msg_ctx,
963 NCALRPC,
964 state->ep.name,
965 cli_addr, NULL, sd,
966 state->disconnect_fn);
969 struct dcerpc_ncacn_conn {
970 enum dcerpc_transport_t transport;
972 int sock;
974 struct pipes_struct *p;
975 dcerpc_ncacn_disconnect_fn disconnect_fn;
977 struct tevent_context *ev_ctx;
978 struct messaging_context *msg_ctx;
980 struct tstream_context *tstream;
981 struct tevent_queue *send_queue;
983 struct tsocket_address *client;
984 char *client_name;
985 struct tsocket_address *server;
986 char *server_name;
987 struct auth_session_info *session_info;
989 struct iovec *iov;
990 size_t count;
993 static void dcerpc_ncacn_packet_process(struct tevent_req *subreq);
994 static void dcerpc_ncacn_packet_done(struct tevent_req *subreq);
996 void dcerpc_ncacn_accept(struct tevent_context *ev_ctx,
997 struct messaging_context *msg_ctx,
998 enum dcerpc_transport_t transport,
999 const char *name,
1000 struct tsocket_address *cli_addr,
1001 struct tsocket_address *srv_addr,
1002 int s,
1003 dcerpc_ncacn_disconnect_fn fn) {
1004 struct dcerpc_ncacn_conn *ncacn_conn;
1005 struct tevent_req *subreq;
1006 bool system_user = false;
1007 char *pipe_name;
1008 NTSTATUS status;
1009 int sys_errno;
1010 uid_t uid;
1011 int rc;
1013 DEBUG(10, ("dcerpc_ncacn_accept\n"));
1015 ncacn_conn = talloc_zero(ev_ctx, struct dcerpc_ncacn_conn);
1016 if (ncacn_conn == NULL) {
1017 DEBUG(0, ("Out of memory!\n"));
1018 close(s);
1019 return;
1022 ncacn_conn->transport = transport;
1023 ncacn_conn->ev_ctx = ev_ctx;
1024 ncacn_conn->msg_ctx = msg_ctx;
1025 ncacn_conn->sock = s;
1026 ncacn_conn->disconnect_fn = fn;
1028 ncacn_conn->client = talloc_move(ncacn_conn, &cli_addr);
1029 if (tsocket_address_is_inet(ncacn_conn->client, "ip")) {
1030 ncacn_conn->client_name =
1031 tsocket_address_inet_addr_string(ncacn_conn->client,
1032 ncacn_conn);
1033 } else {
1034 ncacn_conn->client_name =
1035 tsocket_address_unix_path(ncacn_conn->client,
1036 ncacn_conn);
1038 if (ncacn_conn->client_name == NULL) {
1039 DEBUG(0, ("Out of memory!\n"));
1040 talloc_free(ncacn_conn);
1041 close(s);
1042 return;
1045 if (srv_addr != NULL) {
1046 ncacn_conn->server = talloc_move(ncacn_conn, &srv_addr);
1048 ncacn_conn->server_name =
1049 tsocket_address_inet_addr_string(ncacn_conn->server,
1050 ncacn_conn);
1051 if (ncacn_conn->server_name == NULL) {
1052 DEBUG(0, ("Out of memory!\n"));
1053 talloc_free(ncacn_conn);
1054 close(s);
1055 return;
1059 switch (transport) {
1060 case NCACN_IP_TCP:
1061 pipe_name = tsocket_address_string(ncacn_conn->client,
1062 ncacn_conn);
1063 if (pipe_name == NULL) {
1064 close(s);
1065 talloc_free(ncacn_conn);
1066 return;
1069 break;
1070 case NCALRPC:
1071 rc = sys_getpeereid(s, &uid);
1072 if (rc < 0) {
1073 DEBUG(2, ("Failed to get ncalrpc connecting "
1074 "uid - %s!\n", strerror(errno)));
1075 } else {
1076 if (uid == sec_initial_uid()) {
1077 system_user = true;
1080 case NCACN_NP:
1081 pipe_name = talloc_strdup(ncacn_conn,
1082 name);
1083 if (pipe_name == NULL) {
1084 close(s);
1085 talloc_free(ncacn_conn);
1086 return;
1088 break;
1089 default:
1090 DEBUG(0, ("unknown dcerpc transport: %u!\n",
1091 transport));
1092 talloc_free(ncacn_conn);
1093 close(s);
1094 return;
1097 rc = set_blocking(s, false);
1098 if (rc < 0) {
1099 DEBUG(2, ("Failed to set dcerpc socket to non-blocking\n"));
1100 talloc_free(ncacn_conn);
1101 close(s);
1102 return;
1106 * As soon as we have tstream_bsd_existing_socket set up it will
1107 * take care of closing the socket.
1109 rc = tstream_bsd_existing_socket(ncacn_conn, s, &ncacn_conn->tstream);
1110 if (rc < 0) {
1111 DEBUG(2, ("Failed to create tstream socket for dcerpc\n"));
1112 talloc_free(ncacn_conn);
1113 close(s);
1114 return;
1117 if (ncacn_conn->session_info == NULL) {
1118 status = auth_anonymous_session_info(ncacn_conn,
1119 &ncacn_conn->session_info);
1120 if (!NT_STATUS_IS_OK(status)) {
1121 DEBUG(2, ("Failed to create "
1122 "auth_anonymous_session_info - %s\n",
1123 nt_errstr(status)));
1124 talloc_free(ncacn_conn);
1125 return;
1129 rc = make_server_pipes_struct(ncacn_conn,
1130 ncacn_conn->msg_ctx,
1131 pipe_name,
1132 ncacn_conn->transport,
1133 system_user,
1134 ncacn_conn->server,
1135 ncacn_conn->client,
1136 ncacn_conn->session_info,
1137 &ncacn_conn->p,
1138 &sys_errno);
1139 if (rc < 0) {
1140 DEBUG(2, ("Failed to create pipe struct - %s",
1141 strerror(sys_errno)));
1142 talloc_free(ncacn_conn);
1143 return;
1146 ncacn_conn->send_queue = tevent_queue_create(ncacn_conn,
1147 "dcerpc send queue");
1148 if (ncacn_conn->send_queue == NULL) {
1149 DEBUG(0, ("Out of memory!\n"));
1150 talloc_free(ncacn_conn);
1151 return;
1154 subreq = dcerpc_read_ncacn_packet_send(ncacn_conn,
1155 ncacn_conn->ev_ctx,
1156 ncacn_conn->tstream);
1157 if (subreq == NULL) {
1158 DEBUG(2, ("Failed to send ncacn packet\n"));
1159 talloc_free(ncacn_conn);
1160 return;
1163 tevent_req_set_callback(subreq, dcerpc_ncacn_packet_process, ncacn_conn);
1165 DEBUG(10, ("dcerpc_ncacn_accept done\n"));
1167 return;
1170 static void dcerpc_ncacn_packet_process(struct tevent_req *subreq)
1172 struct dcerpc_ncacn_conn *ncacn_conn =
1173 tevent_req_callback_data(subreq, struct dcerpc_ncacn_conn);
1175 struct _output_data *out = &ncacn_conn->p->out_data;
1176 DATA_BLOB recv_buffer = data_blob_null;
1177 struct ncacn_packet *pkt;
1178 ssize_t data_left;
1179 ssize_t data_used;
1180 uint32_t to_send;
1181 char *data;
1182 NTSTATUS status;
1183 bool ok;
1185 status = dcerpc_read_ncacn_packet_recv(subreq, ncacn_conn, &pkt, &recv_buffer);
1186 TALLOC_FREE(subreq);
1187 if (!NT_STATUS_IS_OK(status)) {
1188 if (ncacn_conn->disconnect_fn != NULL) {
1189 ok = ncacn_conn->disconnect_fn(ncacn_conn->p);
1190 if (!ok) {
1191 DEBUG(3, ("Failed to call disconnect function\n"));
1194 goto fail;
1197 data_left = recv_buffer.length;
1198 data = (char *) recv_buffer.data;
1200 while (data_left) {
1201 data_used = process_incoming_data(ncacn_conn->p, data, data_left);
1202 if (data_used < 0) {
1203 DEBUG(3, ("Failed to process dcerpc request!\n"));
1204 status = NT_STATUS_UNEXPECTED_IO_ERROR;
1205 goto fail;
1208 data_left -= data_used;
1209 data += data_used;
1212 /* Do not leak this buffer */
1213 talloc_free(recv_buffer.data);
1214 talloc_free(pkt);
1217 * This is needed because of the way DCERPC binds work in the RPC
1218 * marshalling code
1220 to_send = out->frag.length - out->current_pdu_sent;
1221 if (to_send > 0) {
1223 DEBUG(10, ("Current_pdu_len = %u, "
1224 "current_pdu_sent = %u "
1225 "Returning %u bytes\n",
1226 (unsigned int)out->frag.length,
1227 (unsigned int)out->current_pdu_sent,
1228 (unsigned int)to_send));
1230 ncacn_conn->iov = talloc_zero(ncacn_conn, struct iovec);
1231 if (ncacn_conn->iov == NULL) {
1232 status = NT_STATUS_NO_MEMORY;
1233 DEBUG(3, ("Out of memory!\n"));
1234 goto fail;
1236 ncacn_conn->count = 1;
1238 ncacn_conn->iov[0].iov_base = out->frag.data
1239 + out->current_pdu_sent;
1240 ncacn_conn->iov[0].iov_len = to_send;
1242 out->current_pdu_sent += to_send;
1246 * This condition is false for bind packets, or when we haven't yet got
1247 * a full request, and need to wait for more data from the client
1249 while (out->data_sent_length < out->rdata.length) {
1250 ok = create_next_pdu(ncacn_conn->p);
1251 if (!ok) {
1252 DEBUG(3, ("Failed to create next PDU!\n"));
1253 status = NT_STATUS_UNEXPECTED_IO_ERROR;
1254 goto fail;
1257 ncacn_conn->iov = talloc_realloc(ncacn_conn,
1258 ncacn_conn->iov,
1259 struct iovec,
1260 ncacn_conn->count + 1);
1261 if (ncacn_conn->iov == NULL) {
1262 DEBUG(3, ("Out of memory!\n"));
1263 status = NT_STATUS_NO_MEMORY;
1264 goto fail;
1267 ncacn_conn->iov[ncacn_conn->count].iov_base = out->frag.data;
1268 ncacn_conn->iov[ncacn_conn->count].iov_len = out->frag.length;
1270 DEBUG(10, ("PDU number: %d, PDU Length: %u\n",
1271 (unsigned int) ncacn_conn->count,
1272 (unsigned int) ncacn_conn->iov[ncacn_conn->count].iov_len));
1273 dump_data(11, (const uint8_t *) ncacn_conn->iov[ncacn_conn->count].iov_base,
1274 ncacn_conn->iov[ncacn_conn->count].iov_len);
1275 ncacn_conn->count++;
1279 * We still don't have a complete request, go back and wait for more
1280 * data.
1282 if (ncacn_conn->count == 0) {
1283 /* Wait for the next packet */
1284 subreq = dcerpc_read_ncacn_packet_send(ncacn_conn,
1285 ncacn_conn->ev_ctx,
1286 ncacn_conn->tstream);
1287 if (subreq == NULL) {
1288 DEBUG(2, ("Failed to start receving packets\n"));
1289 status = NT_STATUS_NO_MEMORY;
1290 goto fail;
1292 tevent_req_set_callback(subreq, dcerpc_ncacn_packet_process, ncacn_conn);
1293 return;
1296 DEBUG(10, ("Sending a total of %u bytes\n",
1297 (unsigned int)ncacn_conn->p->out_data.data_sent_length));
1299 subreq = tstream_writev_queue_send(ncacn_conn,
1300 ncacn_conn->ev_ctx,
1301 ncacn_conn->tstream,
1302 ncacn_conn->send_queue,
1303 ncacn_conn->iov,
1304 ncacn_conn->count);
1305 if (subreq == NULL) {
1306 DEBUG(2, ("Failed to send packet\n"));
1307 status = NT_STATUS_NO_MEMORY;
1308 goto fail;
1311 tevent_req_set_callback(subreq, dcerpc_ncacn_packet_done, ncacn_conn);
1312 return;
1314 fail:
1315 DEBUG(3, ("Terminating client(%s) connection! - '%s'\n",
1316 ncacn_conn->client_name, nt_errstr(status)));
1318 /* Terminate client connection */
1319 talloc_free(ncacn_conn);
1320 return;
1323 static void dcerpc_ncacn_packet_done(struct tevent_req *subreq)
1325 struct dcerpc_ncacn_conn *ncacn_conn =
1326 tevent_req_callback_data(subreq, struct dcerpc_ncacn_conn);
1327 NTSTATUS status = NT_STATUS_OK;
1328 int sys_errno;
1329 int rc;
1331 rc = tstream_writev_queue_recv(subreq, &sys_errno);
1332 TALLOC_FREE(subreq);
1333 if (rc < 0) {
1334 DEBUG(2, ("Writev failed!\n"));
1335 status = map_nt_error_from_unix(sys_errno);
1336 goto fail;
1339 /* clear out any data that may have been left around */
1340 ncacn_conn->count = 0;
1341 TALLOC_FREE(ncacn_conn->iov);
1342 data_blob_free(&ncacn_conn->p->in_data.data);
1343 data_blob_free(&ncacn_conn->p->out_data.frag);
1344 data_blob_free(&ncacn_conn->p->out_data.rdata);
1346 talloc_free_children(ncacn_conn->p->mem_ctx);
1348 /* Wait for the next packet */
1349 subreq = dcerpc_read_ncacn_packet_send(ncacn_conn,
1350 ncacn_conn->ev_ctx,
1351 ncacn_conn->tstream);
1352 if (subreq == NULL) {
1353 DEBUG(2, ("Failed to start receving packets\n"));
1354 status = NT_STATUS_NO_MEMORY;
1355 goto fail;
1358 tevent_req_set_callback(subreq, dcerpc_ncacn_packet_process, ncacn_conn);
1359 return;
1361 fail:
1362 DEBUG(3, ("Terminating client(%s) connection! - '%s'\n",
1363 ncacn_conn->client_name, nt_errstr(status)));
1365 /* Terminate client connection */
1366 talloc_free(ncacn_conn);
1367 return;
1370 /* vim: set ts=8 sw=8 noet cindent syntax=c.doxygen: */