s3-rpc_server: Make dcerpc_ncacn_accept() public.
[idra/samba.git] / source3 / rpc_server / rpc_server.c
index 7abb4598b444325bf9176000ec3252c5665ec1ba..2e6bfa53e23571d8232ae9e18c1c0313d7727f54 100644 (file)
@@ -2,6 +2,8 @@
    Unix SMB/Netbios implementation.
    Generic infrstructure for RPC Daemons
    Copyright (C) Simo Sorce 2010
+   Copyright (C) Andrew Bartlett 2011
+   Copyright (C) Andreas Schneider 2011
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 */
 
 #include "includes.h"
+#include "rpc_server/rpc_pipes.h"
 #include "rpc_server/rpc_server.h"
 #include "rpc_dce.h"
 #include "librpc/gen_ndr/netlogon.h"
-#include "registry/reg_parse_prs.h"
+#include "librpc/gen_ndr/auth.h"
 #include "lib/tsocket/tsocket.h"
 #include "libcli/named_pipe_auth/npa_tstream.h"
+#include "../auth/auth_sam_reply.h"
+#include "auth.h"
+#include "rpc_server/rpc_ncacn_np.h"
+#include "rpc_server/srv_pipe_hnd.h"
+#include "rpc_server/srv_pipe.h"
+
+#define SERVER_TCP_LOW_PORT  1024
+#define SERVER_TCP_HIGH_PORT 1300
+
+static NTSTATUS auth_anonymous_session_info(TALLOC_CTX *mem_ctx,
+                                           struct auth_session_info **session_info)
+{
+       NTSTATUS status;
+
+       status = make_session_info_guest(mem_ctx, session_info);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
+
+       return NT_STATUS_OK;
+}
 
 /* Creates a pipes_struct and initializes it with the information
  * sent from the client */
 static int make_server_pipes_struct(TALLOC_CTX *mem_ctx,
+                                   struct messaging_context *msg_ctx,
                                    const char *pipe_name,
-                                   const struct ndr_syntax_id id,
-                                   const char *client_address,
-                                   struct netr_SamInfo3 *info3,
+                                   enum dcerpc_transport_t transport,
+                                   bool ncalrpc_as_system,
+                                   const struct tsocket_address *local_address,
+                                   const struct tsocket_address *remote_address,
+                                   struct auth_session_info *session_info,
                                    struct pipes_struct **_p,
                                    int *perrno)
 {
        struct pipes_struct *p;
        NTSTATUS status;
-       bool ok;
-
-       p = talloc_zero(mem_ctx, struct pipes_struct);
-       if (!p) {
-               *perrno = ENOMEM;
-               return -1;
-       }
-       p->syntax = id;
+       int ret;
 
-       p->mem_ctx = talloc_named(p, 0, "pipe %s %p", pipe_name, p);
-       if (!p->mem_ctx) {
-               TALLOC_FREE(p);
-               *perrno = ENOMEM;
+       ret = make_base_pipes_struct(mem_ctx, msg_ctx, pipe_name,
+                                    transport, RPC_LITTLE_ENDIAN,
+                                    ncalrpc_as_system,
+                                    remote_address, local_address, &p);
+       if (ret) {
+               *perrno = ret;
                return -1;
        }
 
-       ok = init_pipe_handles(p, &id);
-       if (!ok) {
-               DEBUG(1, ("Failed to init handles\n"));
-               TALLOC_FREE(p);
-               *perrno = EINVAL;
-               return -1;
-       }
+       if (session_info->unix_token && session_info->unix_info && session_info->security_token) {
+               /* Don't call create_local_token(), we already have the full details here */
+               p->session_info = talloc_steal(p, session_info);
 
+       } else {
+               struct auth_user_info_dc *auth_user_info_dc;
+               struct auth_serversupplied_info *server_info;
+               struct netr_SamInfo3 *info3;
+
+               /* Fake up an auth_user_info_dc for now, to make an info3, to make the session_info structure */
+               auth_user_info_dc = talloc_zero(p, struct auth_user_info_dc);
+               if (!auth_user_info_dc) {
+                       TALLOC_FREE(p);
+                       *perrno = ENOMEM;
+                       return -1;
+               }
 
-       data_blob_free(&p->in_data.data);
-       data_blob_free(&p->in_data.pdu);
+               auth_user_info_dc->num_sids = session_info->security_token->num_sids;
+               auth_user_info_dc->sids = session_info->security_token->sids;
+               auth_user_info_dc->info = session_info->info;
+               auth_user_info_dc->user_session_key = session_info->session_key;
 
-       p->endian = RPC_LITTLE_ENDIAN;
+               /* This creates the input structure that make_server_info_info3 is looking for */
+               status = auth_convert_user_info_dc_saminfo3(p, auth_user_info_dc,
+                                                           &info3);
 
-       status = make_server_info_info3(p,
-                                       info3->base.account_name.string,
-                                       info3->base.domain.string,
-                                       &p->server_info, info3);
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to init server info\n"));
-               TALLOC_FREE(p);
-               *perrno = EINVAL;
-               return -1;
-       }
+               if (!NT_STATUS_IS_OK(status)) {
+                       DEBUG(1, ("Failed to convert auth_user_info_dc into netr_SamInfo3\n"));
+                       TALLOC_FREE(p);
+                       *perrno = EINVAL;
+                       return -1;
+               }
 
-       /*
-        * Some internal functions need a local token to determine access to
-        * resoutrces.
-        */
-       status = create_local_token(p->server_info);
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to init local auth token\n"));
-               TALLOC_FREE(p);
-               *perrno = EINVAL;
-               return -1;
-       }
+               status = make_server_info_info3(p,
+                                               info3->base.account_name.string,
+                                               info3->base.domain.string,
+                                               &server_info, info3);
+               if (!NT_STATUS_IS_OK(status)) {
+                       DEBUG(1, ("Failed to init server info\n"));
+                       TALLOC_FREE(p);
+                       *perrno = EINVAL;
+                       return -1;
+               }
 
-       p->client_id = talloc_zero(p, struct client_address);
-       if (!p->client_id) {
-               TALLOC_FREE(p);
-               *perrno = ENOMEM;
-               return -1;
+               /*
+                * Some internal functions need a local token to determine access to
+                * resources.
+                */
+               status = create_local_token(p, server_info, &session_info->session_key, info3->base.account_name.string,
+                                           &p->session_info);
+               talloc_free(server_info);
+               if (!NT_STATUS_IS_OK(status)) {
+                       DEBUG(1, ("Failed to init local auth token\n"));
+                       TALLOC_FREE(p);
+                       *perrno = EINVAL;
+                       return -1;
+               }
        }
-       strlcpy(p->client_id->addr,
-               client_address, sizeof(p->client_id->addr));
-
-       talloc_set_destructor(p, close_internal_rpc_pipe_hnd);
 
        *_p = p;
        return 0;
 }
 
-/* Add some helper functions to wrap the common ncacn packet reading functions
- * until we can share more dcerpc code */
-struct named_pipe_read_packet_state {
-       struct ncacn_packet *pkt;
-       DATA_BLOB buffer;
-};
-
-static void named_pipe_read_packet_done(struct tevent_req *subreq);
-
-static struct tevent_req *named_pipe_read_packet_send(TALLOC_CTX *mem_ctx,
-                                       struct tevent_context *ev,
-                                       struct tstream_context *tstream)
-{
-       struct named_pipe_read_packet_state *state;
-       struct tevent_req *req, *subreq;
-
-       req = tevent_req_create(mem_ctx, &state,
-                               struct named_pipe_read_packet_state);
-       if (!req) {
-               return NULL;
-       }
-       ZERO_STRUCTP(state);
-
-       subreq = dcerpc_read_ncacn_packet_send(state, ev, tstream);
-       if (!subreq) {
-               tevent_req_nterror(req, NT_STATUS_NO_MEMORY);
-               tevent_req_post(req, ev);
-               return req;
-       }
-       tevent_req_set_callback(subreq, named_pipe_read_packet_done, req);
-
-       return req;
-}
-
-static void named_pipe_read_packet_done(struct tevent_req *subreq)
-{
-       struct tevent_req *req =
-               tevent_req_callback_data(subreq, struct tevent_req);
-       struct named_pipe_read_packet_state *state =
-               tevent_req_data(req, struct named_pipe_read_packet_state);
-       NTSTATUS status;
-
-       status = dcerpc_read_ncacn_packet_recv(subreq, state,
-                                               &state->pkt,
-                                               &state->buffer);
-       TALLOC_FREE(subreq);
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(3, ("Failed to receive dceprc packet!\n"));
-               tevent_req_nterror(req, status);
-               return;
-       }
-
-       tevent_req_done(req);
-}
-
-static NTSTATUS named_pipe_read_packet_recv(struct tevent_req *req,
-                                               TALLOC_CTX *mem_ctx,
-                                               DATA_BLOB *buffer)
-{
-       struct named_pipe_read_packet_state *state =
-               tevent_req_data(req, struct named_pipe_read_packet_state);
-       NTSTATUS status;
-
-       if (tevent_req_is_nterror(req, &status)) {
-               tevent_req_received(req);
-               return status;
-       }
-
-       buffer->data = talloc_move(mem_ctx, &state->buffer.data);
-       buffer->length = state->buffer.length;
-
-       tevent_req_received(req);
-       return NT_STATUS_OK;
-}
-
-
-
 /* Start listening on the appropriate unix socket and setup all is needed to
  * dispatch requests to the pipes rpc implementation */
 
-struct named_pipe_listen_state {
+struct dcerpc_ncacn_listen_state {
+       struct ndr_syntax_id syntax_id;
+
        int fd;
-       char *name;
+       union {
+               char *name;
+               uint16_t port;
+       } ep;
+
+       struct tevent_context *ev_ctx;
+       struct messaging_context *msg_ctx;
+       dcerpc_ncacn_disconnect_fn disconnect_fn;
 };
 
 static void named_pipe_listener(struct tevent_context *ev,
@@ -195,26 +160,23 @@ static void named_pipe_listener(struct tevent_context *ev,
                                uint16_t flags,
                                void *private_data);
 
-bool setup_named_pipe_socket(const char *pipe_name,
-                            struct tevent_context *ev_ctx)
+int create_named_pipe_socket(const char *pipe_name)
 {
-       struct named_pipe_listen_state *state;
-       struct tevent_fd *fde;
-       char *np_dir;
+       char *np_dir = NULL;
+       int fd = -1;
 
-       state = talloc(ev_ctx, struct named_pipe_listen_state);
-       if (!state) {
-               DEBUG(0, ("Out of memory\n"));
-               return false;
-       }
-       state->name = talloc_strdup(state, pipe_name);
-       if (!state->name) {
-               DEBUG(0, ("Out of memory\n"));
+       /*
+        * As lp_ncalrpc_dir() should have 0755, but
+        * lp_ncalrpc_dir()/np should have 0700, we need to
+        * create lp_ncalrpc_dir() first.
+        */
+       if (!directory_create_or_exist(lp_ncalrpc_dir(), geteuid(), 0755)) {
+               DEBUG(0, ("Failed to create pipe directory %s - %s\n",
+                         lp_ncalrpc_dir(), strerror(errno)));
                goto out;
        }
-       state->fd = -1;
 
-       np_dir = talloc_asprintf(state, "%s/np", lp_ncalrpc_dir());
+       np_dir = talloc_asprintf(talloc_tos(), "%s/np", lp_ncalrpc_dir());
        if (!np_dir) {
                DEBUG(0, ("Out of memory\n"));
                goto out;
@@ -226,13 +188,45 @@ bool setup_named_pipe_socket(const char *pipe_name,
                goto out;
        }
 
-       state->fd = create_pipe_sock(np_dir, pipe_name, 0700);
-       if (state->fd == -1) {
+       fd = create_pipe_sock(np_dir, pipe_name, 0700);
+       if (fd == -1) {
                DEBUG(0, ("Failed to create pipe socket! [%s/%s]\n",
                          np_dir, pipe_name));
                goto out;
        }
 
+       DEBUG(10, ("Openened pipe socket fd %d for %s\n", fd, pipe_name));
+
+out:
+       talloc_free(np_dir);
+       return fd;
+}
+
+bool setup_named_pipe_socket(const char *pipe_name,
+                            struct tevent_context *ev_ctx,
+                            struct messaging_context *msg_ctx)
+{
+       struct dcerpc_ncacn_listen_state *state;
+       struct tevent_fd *fde;
+
+       state = talloc(ev_ctx, struct dcerpc_ncacn_listen_state);
+       if (!state) {
+               DEBUG(0, ("Out of memory\n"));
+               return false;
+       }
+       state->ep.name = talloc_strdup(state, pipe_name);
+       if (state->ep.name == NULL) {
+               DEBUG(0, ("Out of memory\n"));
+               goto out;
+       }
+       state->fd = create_named_pipe_socket(pipe_name);
+       if (state->fd == -1) {
+               goto out;
+       }
+
+       state->ev_ctx = ev_ctx;
+       state->msg_ctx = msg_ctx;
+
        DEBUG(10, ("Openened pipe socket fd %d for %s\n",
                   state->fd, pipe_name));
 
@@ -255,16 +249,14 @@ out:
        return false;
 }
 
-static void named_pipe_accept_function(const char *pipe_name, int fd);
-
 static void named_pipe_listener(struct tevent_context *ev,
                                struct tevent_fd *fde,
                                uint16_t flags,
                                void *private_data)
 {
-       struct named_pipe_listen_state *state =
+       struct dcerpc_ncacn_listen_state *state =
                        talloc_get_type_abort(private_data,
-                                             struct named_pipe_listen_state);
+                                             struct dcerpc_ncacn_listen_state);
        struct sockaddr_un sunaddr;
        socklen_t len;
        int sd = -1;
@@ -273,21 +265,23 @@ static void named_pipe_listener(struct tevent_context *ev,
 
        len = sizeof(sunaddr);
 
-       while (sd == -1) {
-               sd = accept(state->fd,
-                           (struct sockaddr *)(void *)&sunaddr, &len);
-               if (errno != EINTR) break;
-       }
+       sd = accept(state->fd,
+                   (struct sockaddr *)(void *)&sunaddr, &len);
 
        if (sd == -1) {
-               DEBUG(6, ("Failed to get a valid socket [%s]\n",
-                         strerror(errno)));
+               if (errno != EINTR) {
+                       DEBUG(6, ("Failed to get a valid socket [%s]\n",
+                                 strerror(errno)));
+               }
                return;
        }
 
        DEBUG(6, ("Accepted socket %d\n", sd));
 
-       named_pipe_accept_function(state->name, sd);
+       named_pipe_accept_function(state->ev_ctx,
+                                  state->msg_ctx,
+                                  state->ep.name,
+                                  sd, NULL, 0);
 }
 
 
@@ -297,9 +291,9 @@ static void named_pipe_listener(struct tevent_context *ev,
 
 struct named_pipe_client {
        const char *pipe_name;
-       struct ndr_syntax_id pipe_id;
 
        struct tevent_context *ev;
+       struct messaging_context *msg_ctx;
 
        uint16_t file_type;
        uint16_t device_state;
@@ -311,9 +305,8 @@ struct named_pipe_client {
        char *client_name;
        struct tsocket_address *server;
        char *server_name;
-       struct netr_SamInfo3 *info3;
-       DATA_BLOB session_key;
-       DATA_BLOB delegated_creds;
+
+       struct auth_session_info *session_info;
 
        struct pipes_struct *p;
 
@@ -321,35 +314,52 @@ struct named_pipe_client {
 
        struct iovec *iov;
        size_t count;
+
+       named_pipe_termination_fn *term_fn;
+       void *private_data;
 };
 
+static int named_pipe_destructor(struct named_pipe_client *npc)
+{
+       if (npc->term_fn) {
+               npc->term_fn(npc->private_data);
+       }
+       return 0;
+}
+
 static void named_pipe_accept_done(struct tevent_req *subreq);
 
-static void named_pipe_accept_function(const char *pipe_name, int fd)
+void named_pipe_accept_function(struct tevent_context *ev_ctx,
+                               struct messaging_context *msg_ctx,
+                               const char *pipe_name, int fd,
+                               named_pipe_termination_fn *term_fn,
+                               void *private_data)
 {
-       struct ndr_syntax_id syntax;
        struct named_pipe_client *npc;
        struct tstream_context *plain;
        struct tevent_req *subreq;
-       bool ok;
        int ret;
 
-       ok = is_known_pipename(pipe_name, &syntax);
-       if (!ok) {
-               DEBUG(1, ("Unknown pipe [%s]\n", pipe_name));
+       npc = talloc_zero(ev_ctx, struct named_pipe_client);
+       if (!npc) {
+               DEBUG(0, ("Out of memory!\n"));
                close(fd);
                return;
        }
 
-       npc = talloc_zero(NULL, struct named_pipe_client);
-       if (!npc) {
+       npc->pipe_name = talloc_strdup(npc, pipe_name);
+       if (npc->pipe_name == NULL) {
                DEBUG(0, ("Out of memory!\n"));
+               TALLOC_FREE(npc);
                close(fd);
                return;
        }
-       npc->pipe_name = pipe_name;
-       npc->pipe_id = syntax;
-       npc->ev = server_event_context();
+       npc->ev = ev_ctx;
+       npc->msg_ctx = msg_ctx;
+       npc->term_fn = term_fn;
+       npc->private_data = private_data;
+
+       talloc_set_destructor(npc, named_pipe_destructor);
 
        /* make sure socket is in NON blocking state */
        ret = set_blocking(fd, false);
@@ -390,9 +400,9 @@ static void named_pipe_packet_done(struct tevent_req *subreq);
 
 static void named_pipe_accept_done(struct tevent_req *subreq)
 {
+       struct auth_session_info_transport *session_info_transport;
        struct named_pipe_client *npc =
                tevent_req_callback_data(subreq, struct named_pipe_client);
-       const char *cli_addr;
        int error;
        int ret;
 
@@ -402,9 +412,10 @@ static void named_pipe_accept_done(struct tevent_req *subreq)
                                                &npc->client_name,
                                                &npc->server,
                                                &npc->server_name,
-                                               &npc->info3,
-                                               &npc->session_key,
-                                               &npc->delegated_creds);
+                                               &session_info_transport);
+
+       npc->session_info = talloc_move(npc, &session_info_transport->session_info);
+
        TALLOC_FREE(subreq);
        if (ret != 0) {
                DEBUG(2, ("Failed to accept named pipe connection! (%s)\n",
@@ -413,20 +424,10 @@ static void named_pipe_accept_done(struct tevent_req *subreq)
                return;
        }
 
-       if (tsocket_address_is_inet(npc->client, "ip")) {
-               cli_addr = tsocket_address_inet_addr_string(npc->client,
-                                                           subreq);
-               if (cli_addr == NULL) {
-                       TALLOC_FREE(npc);
-                       return;
-               }
-       } else {
-               cli_addr = "";
-       }
-
        ret = make_server_pipes_struct(npc,
-                                       npc->pipe_name, npc->pipe_id,
-                                       cli_addr, npc->info3,
+                                      npc->msg_ctx,
+                                      npc->pipe_name, NCACN_NP,
+                                       false, npc->server, npc->client, npc->session_info,
                                        &npc->p, &error);
        if (ret != 0) {
                DEBUG(2, ("Failed to create pipes_struct! (%s)\n",
@@ -440,8 +441,8 @@ static void named_pipe_accept_done(struct tevent_req *subreq)
                goto fail;
        }
 
-       /* And now start receaving and processing packets */
-       subreq = named_pipe_read_packet_send(npc, npc->ev, npc->tstream);
+       /* And now start receiving and processing packets */
+       subreq = dcerpc_read_ncacn_packet_send(npc, npc->ev, npc->tstream);
        if (!subreq) {
                DEBUG(2, ("Failed to start receving packets\n"));
                goto fail;
@@ -462,15 +463,17 @@ static void named_pipe_packet_process(struct tevent_req *subreq)
        struct named_pipe_client *npc =
                tevent_req_callback_data(subreq, struct named_pipe_client);
        struct _output_data *out = &npc->p->out_data;
-       DATA_BLOB recv_buffer;
+       DATA_BLOB recv_buffer = data_blob_null;
+       struct ncacn_packet *pkt;
        NTSTATUS status;
        ssize_t data_left;
        ssize_t data_used;
        char *data;
        uint32_t to_send;
+       size_t i;
        bool ok;
 
-       status = named_pipe_read_packet_recv(subreq, npc, &recv_buffer);
+       status = dcerpc_read_ncacn_packet_recv(subreq, npc, &pkt, &recv_buffer);
        TALLOC_FREE(subreq);
        if (!NT_STATUS_IS_OK(status)) {
                goto fail;
@@ -494,19 +497,13 @@ static void named_pipe_packet_process(struct tevent_req *subreq)
 
        /* Do not leak this buffer, npc is a long lived context */
        talloc_free(recv_buffer.data);
+       talloc_free(pkt);
 
        /* this is needed because of the way DCERPC Binds work in
         * the RPC marshalling code */
        to_send = out->frag.length - out->current_pdu_sent;
        if (to_send > 0) {
 
-               DEBUG(10, ("Current_pdu_len = %u, "
-                          "current_pdu_sent = %u "
-                          "Returning %u bytes\n",
-                          (unsigned int)out->frag.length,
-                          (unsigned int)out->current_pdu_sent,
-                          (unsigned int)to_send));
-
                npc->iov = talloc_zero(npc, struct iovec);
                if (!npc->iov) {
                        status = NT_STATUS_NO_MEMORY;
@@ -543,11 +540,6 @@ static void named_pipe_packet_process(struct tevent_req *subreq)
                npc->iov[npc->count].iov_base = out->frag.data;
                npc->iov[npc->count].iov_len = out->frag.length;
 
-               DEBUG(10, ("PDU number: %d, PDU Length: %u\n",
-                          (unsigned int)npc->count,
-                          (unsigned int)npc->iov[npc->count].iov_len));
-               dump_data(11, npc->iov[npc->count].iov_base,
-                               npc->iov[npc->count].iov_len);
                npc->count++;
        }
 
@@ -555,7 +547,7 @@ static void named_pipe_packet_process(struct tevent_req *subreq)
         * data */
        if (npc->count == 0) {
                /* Wait for the next packet */
-               subreq = named_pipe_read_packet_send(npc, npc->ev, npc->tstream);
+               subreq = dcerpc_read_ncacn_packet_send(npc, npc->ev, npc->tstream);
                if (!subreq) {
                        DEBUG(2, ("Failed to start receving packets\n"));
                        status = NT_STATUS_NO_MEMORY;
@@ -565,19 +557,31 @@ static void named_pipe_packet_process(struct tevent_req *subreq)
                return;
        }
 
-       DEBUG(10, ("Sending a total of %u bytes\n",
+       DEBUG(10, ("Sending %u fragments in a total of %u bytes\n",
+                  (unsigned int)npc->count,
                   (unsigned int)npc->p->out_data.data_sent_length));
 
-       subreq = tstream_writev_queue_send(npc, npc->ev,
-                                          npc->tstream,
-                                          npc->write_queue,
-                                          npc->iov, npc->count);
-       if (!subreq) {
-               DEBUG(2, ("Failed to send packet\n"));
-               status = NT_STATUS_NO_MEMORY;
-               goto fail;
+       for (i = 0; i < npc->count; i++) {
+               DEBUG(10, ("Sending PDU number: %d, PDU Length: %u\n",
+                         (unsigned int)i,
+                         (unsigned int)npc->iov[i].iov_len));
+               dump_data(11, (const uint8_t *)npc->iov[i].iov_base,
+                               npc->iov[i].iov_len);
+
+               subreq = tstream_writev_queue_send(npc,
+                                                  npc->ev,
+                                                  npc->tstream,
+                                                  npc->write_queue,
+                                                  (npc->iov + i),
+                                                  1);
+               if (!subreq) {
+                       DEBUG(2, ("Failed to send packet\n"));
+                       status = NT_STATUS_NO_MEMORY;
+                       goto fail;
+               }
+               tevent_req_set_callback(subreq, named_pipe_packet_done, npc);
        }
-       tevent_req_set_callback(subreq, named_pipe_packet_done, npc);
+
        return;
 
 fail:
@@ -603,6 +607,10 @@ static void named_pipe_packet_done(struct tevent_req *subreq)
                goto fail;
        }
 
+       if (tevent_queue_length(npc->write_queue) > 0) {
+               return;
+       }
+
        /* clear out any data that may have been left around */
        npc->count = 0;
        TALLOC_FREE(npc->iov);
@@ -610,8 +618,10 @@ static void named_pipe_packet_done(struct tevent_req *subreq)
        data_blob_free(&npc->p->out_data.frag);
        data_blob_free(&npc->p->out_data.rdata);
 
+       talloc_free_children(npc->p->mem_ctx);
+
        /* Wait for the next packet */
-       subreq = named_pipe_read_packet_send(npc, npc->ev, npc->tstream);
+       subreq = dcerpc_read_ncacn_packet_send(npc, npc->ev, npc->tstream);
        if (!subreq) {
                DEBUG(2, ("Failed to start receving packets\n"));
                sys_errno = ENOMEM;
@@ -627,4 +637,700 @@ fail:
        /* terminate client connection */
        talloc_free(npc);
        return;
- }
+}
+
+/********************************************************************
+ * Start listening on the tcp/ip socket
+ ********************************************************************/
+
+static void dcerpc_ncacn_tcpip_listener(struct tevent_context *ev,
+                                       struct tevent_fd *fde,
+                                       uint16_t flags,
+                                       void *private_data);
+
+int create_tcpip_socket(const struct sockaddr_storage *ifss, uint16_t *port)
+{
+       int fd = -1;
+
+       if (*port == 0) {
+               uint16_t i;
+
+               for (i = SERVER_TCP_LOW_PORT; i <= SERVER_TCP_HIGH_PORT; i++) {
+                       fd = open_socket_in(SOCK_STREAM,
+                                           i,
+                                           0,
+                                           ifss,
+                                           false);
+                       if (fd > 0) {
+                               *port = i;
+                               break;
+                       }
+               }
+       } else {
+               fd = open_socket_in(SOCK_STREAM,
+                                   *port,
+                                   0,
+                                   ifss,
+                                   true);
+       }
+       if (fd == -1) {
+               DEBUG(0, ("Failed to create socket on port %u!\n", *port));
+               return -1;
+       }
+
+       DEBUG(10, ("Opened tcpip socket fd %d for port %u\n", fd, *port));
+
+       return fd;
+}
+
+uint16_t setup_dcerpc_ncacn_tcpip_socket(struct tevent_context *ev_ctx,
+                                        struct messaging_context *msg_ctx,
+                                        const struct sockaddr_storage *ifss,
+                                        uint16_t port)
+{
+       struct dcerpc_ncacn_listen_state *state;
+       struct tevent_fd *fde;
+       int rc;
+
+       state = talloc(ev_ctx, struct dcerpc_ncacn_listen_state);
+       if (state == NULL) {
+               DEBUG(0, ("setup_dcerpc_ncacn_tcpip_socket: Out of memory\n"));
+               return 0;
+       }
+
+       state->fd = -1;
+       state->ep.port = port;
+       state->disconnect_fn = NULL;
+
+       state->fd = create_tcpip_socket(ifss, &state->ep.port);
+       if (state->fd == -1) {
+               goto out;
+       }
+
+       state->ev_ctx = ev_ctx;
+       state->msg_ctx = msg_ctx;
+
+       /* ready to listen */
+       set_socket_options(state->fd, "SO_KEEPALIVE");
+       set_socket_options(state->fd, lp_socket_options());
+
+       /* Set server socket to non-blocking for the accept. */
+       set_blocking(state->fd, false);
+
+       rc = listen(state->fd, SMBD_LISTEN_BACKLOG);
+       if (rc == -1) {
+               DEBUG(0,("setup_tcpip_socket: listen - %s\n", strerror(errno)));
+               goto out;
+       }
+
+       DEBUG(10, ("setup_tcpip_socket: openened socket fd %d for port %u\n",
+                  state->fd, state->ep.port));
+
+       fde = tevent_add_fd(state->ev_ctx,
+                           state,
+                           state->fd,
+                           TEVENT_FD_READ,
+                           dcerpc_ncacn_tcpip_listener,
+                           state);
+       if (fde == NULL) {
+               DEBUG(0, ("setup_tcpip_socket: Failed to add event handler!\n"));
+               goto out;
+       }
+
+       tevent_fd_set_auto_close(fde);
+
+       return state->ep.port;
+out:
+       if (state->fd != -1) {
+               close(state->fd);
+       }
+       TALLOC_FREE(state);
+
+       return 0;
+}
+
+static void dcerpc_ncacn_tcpip_listener(struct tevent_context *ev,
+                                       struct tevent_fd *fde,
+                                       uint16_t flags,
+                                       void *private_data)
+{
+       struct dcerpc_ncacn_listen_state *state =
+                       talloc_get_type_abort(private_data,
+                                             struct dcerpc_ncacn_listen_state);
+       struct tsocket_address *cli_addr = NULL;
+       struct tsocket_address *srv_addr = NULL;
+       struct sockaddr_storage addr;
+       socklen_t in_addrlen = sizeof(addr);
+       int s = -1;
+       int rc;
+
+       s = accept(state->fd, (struct sockaddr *)(void *) &addr, &in_addrlen);
+       if (s == -1) {
+               if (errno != EINTR) {
+                       DEBUG(0,("tcpip_listener accept: %s\n",
+                                strerror(errno)));
+               }
+               return;
+       }
+
+       rc = tsocket_address_bsd_from_sockaddr(state,
+                                              (struct sockaddr *)(void *) &addr,
+                                              in_addrlen,
+                                              &cli_addr);
+       if (rc < 0) {
+               close(s);
+               return;
+       }
+
+       rc = getsockname(s, (struct sockaddr *)(void *) &addr, &in_addrlen);
+       if (rc < 0) {
+               close(s);
+               return;
+       }
+
+       rc = tsocket_address_bsd_from_sockaddr(state,
+                                              (struct sockaddr *)(void *) &addr,
+                                              in_addrlen,
+                                              &srv_addr);
+       if (rc < 0) {
+               close(s);
+               return;
+       }
+
+       DEBUG(6, ("tcpip_listener: Accepted socket %d\n", s));
+
+       dcerpc_ncacn_accept(state->ev_ctx,
+                           state->msg_ctx,
+                           NCACN_IP_TCP,
+                           NULL,
+                           cli_addr,
+                           srv_addr,
+                           s,
+                           NULL);
+}
+
+/********************************************************************
+ * Start listening on the ncalrpc socket
+ ********************************************************************/
+
+static void dcerpc_ncalrpc_listener(struct tevent_context *ev,
+                                   struct tevent_fd *fde,
+                                   uint16_t flags,
+                                   void *private_data);
+
+bool setup_dcerpc_ncalrpc_socket(struct tevent_context *ev_ctx,
+                                struct messaging_context *msg_ctx,
+                                const char *name,
+                                dcerpc_ncacn_disconnect_fn fn)
+{
+       struct dcerpc_ncacn_listen_state *state;
+       struct tevent_fd *fde;
+
+       state = talloc(ev_ctx, struct dcerpc_ncacn_listen_state);
+       if (state == NULL) {
+               DEBUG(0, ("Out of memory\n"));
+               return false;
+       }
+
+       state->fd = -1;
+       state->disconnect_fn = fn;
+
+       if (name == NULL) {
+               name = "DEFAULT";
+       }
+       state->ep.name = talloc_strdup(state, name);
+
+       if (state->ep.name == NULL) {
+               DEBUG(0, ("Out of memory\n"));
+               talloc_free(state);
+               return false;
+       }
+
+       if (!directory_create_or_exist(lp_ncalrpc_dir(), geteuid(), 0755)) {
+               DEBUG(0, ("Failed to create pipe directory %s - %s\n",
+                         lp_ncalrpc_dir(), strerror(errno)));
+               goto out;
+       }
+
+       state->fd = create_pipe_sock(lp_ncalrpc_dir(), name, 0755);
+       if (state->fd == -1) {
+               DEBUG(0, ("Failed to create pipe socket! [%s/%s]\n",
+                         lp_ncalrpc_dir(), name));
+               goto out;
+       }
+
+       DEBUG(10, ("Openened pipe socket fd %d for %s\n", state->fd, name));
+
+       state->ev_ctx = ev_ctx;
+       state->msg_ctx = msg_ctx;
+
+       /* Set server socket to non-blocking for the accept. */
+       set_blocking(state->fd, false);
+
+       fde = tevent_add_fd(state->ev_ctx,
+                           state,
+                           state->fd,
+                           TEVENT_FD_READ,
+                           dcerpc_ncalrpc_listener,
+                           state);
+       if (fde == NULL) {
+               DEBUG(0, ("Failed to add event handler for ncalrpc!\n"));
+               goto out;
+       }
+
+       tevent_fd_set_auto_close(fde);
+
+       return true;
+out:
+       if (state->fd != -1) {
+               close(state->fd);
+       }
+       TALLOC_FREE(state);
+
+       return 0;
+}
+
+static void dcerpc_ncalrpc_listener(struct tevent_context *ev,
+                                       struct tevent_fd *fde,
+                                       uint16_t flags,
+                                       void *private_data)
+{
+       struct dcerpc_ncacn_listen_state *state =
+                       talloc_get_type_abort(private_data,
+                                             struct dcerpc_ncacn_listen_state);
+       struct tsocket_address *cli_addr = NULL;
+       struct sockaddr_un sunaddr;
+       struct sockaddr *addr = (struct sockaddr *)(void *)&sunaddr;
+       socklen_t len = sizeof(sunaddr);
+       int sd = -1;
+       int rc;
+
+       ZERO_STRUCT(sunaddr);
+
+       sd = accept(state->fd, addr, &len);
+       if (sd == -1) {
+               if (errno != EINTR) {
+                       DEBUG(0, ("ncalrpc accept() failed: %s\n", strerror(errno)));
+               }
+               return;
+       }
+
+       rc = tsocket_address_bsd_from_sockaddr(state,
+                                              addr, len,
+                                              &cli_addr);
+       if (rc < 0) {
+               close(sd);
+               return;
+       }
+
+       DEBUG(10, ("Accepted ncalrpc socket %d\n", sd));
+
+       dcerpc_ncacn_accept(state->ev_ctx,
+                           state->msg_ctx,
+                           NCALRPC,
+                           state->ep.name,
+                           cli_addr, NULL, sd,
+                           state->disconnect_fn);
+}
+
+struct dcerpc_ncacn_conn {
+       enum dcerpc_transport_t transport;
+
+       int sock;
+
+       struct pipes_struct *p;
+       dcerpc_ncacn_disconnect_fn disconnect_fn;
+
+       struct tevent_context *ev_ctx;
+       struct messaging_context *msg_ctx;
+
+       struct tstream_context *tstream;
+       struct tevent_queue *send_queue;
+
+       struct tsocket_address *client;
+       char *client_name;
+       struct tsocket_address *server;
+       char *server_name;
+       struct auth_session_info *session_info;
+
+       struct iovec *iov;
+       size_t count;
+};
+
+static void dcerpc_ncacn_packet_process(struct tevent_req *subreq);
+static void dcerpc_ncacn_packet_done(struct tevent_req *subreq);
+
+void dcerpc_ncacn_accept(struct tevent_context *ev_ctx,
+                        struct messaging_context *msg_ctx,
+                        enum dcerpc_transport_t transport,
+                        const char *name,
+                        struct tsocket_address *cli_addr,
+                        struct tsocket_address *srv_addr,
+                        int s,
+                        dcerpc_ncacn_disconnect_fn fn) {
+       struct dcerpc_ncacn_conn *ncacn_conn;
+       struct tevent_req *subreq;
+       bool system_user = false;
+       char *pipe_name;
+       NTSTATUS status;
+       int sys_errno;
+       uid_t uid;
+       int rc;
+
+       DEBUG(10, ("dcerpc_ncacn_accept\n"));
+
+       ncacn_conn = talloc_zero(ev_ctx, struct dcerpc_ncacn_conn);
+       if (ncacn_conn == NULL) {
+               DEBUG(0, ("Out of memory!\n"));
+               close(s);
+               return;
+       }
+
+       ncacn_conn->transport = transport;
+       ncacn_conn->ev_ctx = ev_ctx;
+       ncacn_conn->msg_ctx = msg_ctx;
+       ncacn_conn->sock = s;
+       ncacn_conn->disconnect_fn = fn;
+
+       ncacn_conn->client = talloc_move(ncacn_conn, &cli_addr);
+       if (tsocket_address_is_inet(ncacn_conn->client, "ip")) {
+               ncacn_conn->client_name =
+                       tsocket_address_inet_addr_string(ncacn_conn->client,
+                                                        ncacn_conn);
+       } else {
+               ncacn_conn->client_name =
+                       tsocket_address_unix_path(ncacn_conn->client,
+                                                 ncacn_conn);
+       }
+       if (ncacn_conn->client_name == NULL) {
+               DEBUG(0, ("Out of memory!\n"));
+               talloc_free(ncacn_conn);
+               close(s);
+               return;
+       }
+
+       if (srv_addr != NULL) {
+               ncacn_conn->server = talloc_move(ncacn_conn, &srv_addr);
+
+               ncacn_conn->server_name =
+                       tsocket_address_inet_addr_string(ncacn_conn->server,
+                                                        ncacn_conn);
+               if (ncacn_conn->server_name == NULL) {
+                       DEBUG(0, ("Out of memory!\n"));
+                       talloc_free(ncacn_conn);
+                       close(s);
+                       return;
+               }
+       }
+
+       switch (transport) {
+               case NCACN_IP_TCP:
+                       pipe_name = tsocket_address_string(ncacn_conn->client,
+                                                          ncacn_conn);
+                       if (pipe_name == NULL) {
+                               close(s);
+                               talloc_free(ncacn_conn);
+                               return;
+                       }
+
+                       break;
+               case NCALRPC:
+                       rc = sys_getpeereid(s, &uid);
+                       if (rc < 0) {
+                               DEBUG(2, ("Failed to get ncalrpc connecting uid!"));
+                       } else {
+                               if (uid == sec_initial_uid()) {
+                                       system_user = true;
+                               }
+                       }
+               case NCACN_NP:
+                       pipe_name = talloc_strdup(ncacn_conn,
+                                                 name);
+                       if (pipe_name == NULL) {
+                               close(s);
+                               talloc_free(ncacn_conn);
+                               return;
+                       }
+                       break;
+               default:
+                       DEBUG(0, ("unknown dcerpc transport: %u!\n",
+                                 transport));
+                       talloc_free(ncacn_conn);
+                       close(s);
+                       return;
+       }
+
+       rc = set_blocking(s, false);
+       if (rc < 0) {
+               DEBUG(2, ("Failed to set dcerpc socket to non-blocking\n"));
+               talloc_free(ncacn_conn);
+               close(s);
+               return;
+       }
+
+       /*
+        * As soon as we have tstream_bsd_existing_socket set up it will
+        * take care of closing the socket.
+        */
+       rc = tstream_bsd_existing_socket(ncacn_conn, s, &ncacn_conn->tstream);
+       if (rc < 0) {
+               DEBUG(2, ("Failed to create tstream socket for dcerpc\n"));
+               talloc_free(ncacn_conn);
+               close(s);
+               return;
+       }
+
+       if (ncacn_conn->session_info == NULL) {
+               status = auth_anonymous_session_info(ncacn_conn,
+                                                    &ncacn_conn->session_info);
+               if (!NT_STATUS_IS_OK(status)) {
+                       DEBUG(2, ("Failed to create "
+                                 "auth_anonymous_session_info - %s\n",
+                                 nt_errstr(status)));
+                       talloc_free(ncacn_conn);
+                       return;
+               }
+       }
+
+       rc = make_server_pipes_struct(ncacn_conn,
+                                     ncacn_conn->msg_ctx,
+                                     pipe_name,
+                                     ncacn_conn->transport,
+                                     system_user,
+                                     ncacn_conn->server,
+                                     ncacn_conn->client,
+                                     ncacn_conn->session_info,
+                                     &ncacn_conn->p,
+                                     &sys_errno);
+       if (rc < 0) {
+               DEBUG(2, ("Failed to create pipe struct - %s",
+                         strerror(sys_errno)));
+               talloc_free(ncacn_conn);
+               return;
+       }
+
+       ncacn_conn->send_queue = tevent_queue_create(ncacn_conn,
+                                                       "dcerpc send queue");
+       if (ncacn_conn->send_queue == NULL) {
+               DEBUG(0, ("Out of memory!\n"));
+               talloc_free(ncacn_conn);
+               return;
+       }
+
+       subreq = dcerpc_read_ncacn_packet_send(ncacn_conn,
+                                              ncacn_conn->ev_ctx,
+                                              ncacn_conn->tstream);
+       if (subreq == NULL) {
+               DEBUG(2, ("Failed to send ncacn packet\n"));
+               talloc_free(ncacn_conn);
+               return;
+       }
+
+       tevent_req_set_callback(subreq, dcerpc_ncacn_packet_process, ncacn_conn);
+
+       DEBUG(10, ("dcerpc_ncacn_accept done\n"));
+
+       return;
+}
+
+static void dcerpc_ncacn_packet_process(struct tevent_req *subreq)
+{
+       struct dcerpc_ncacn_conn *ncacn_conn =
+               tevent_req_callback_data(subreq, struct dcerpc_ncacn_conn);
+
+       struct _output_data *out = &ncacn_conn->p->out_data;
+       DATA_BLOB recv_buffer = data_blob_null;
+       struct ncacn_packet *pkt;
+       ssize_t data_left;
+       ssize_t data_used;
+       uint32_t to_send;
+       char *data;
+       NTSTATUS status;
+       bool ok;
+
+       status = dcerpc_read_ncacn_packet_recv(subreq, ncacn_conn, &pkt, &recv_buffer);
+       TALLOC_FREE(subreq);
+       if (!NT_STATUS_IS_OK(status)) {
+               if (ncacn_conn->disconnect_fn != NULL) {
+                       ok = ncacn_conn->disconnect_fn(ncacn_conn->p);
+                       if (!ok) {
+                               DEBUG(3, ("Failed to call disconnect function\n"));
+                       }
+               }
+               goto fail;
+       }
+
+       data_left = recv_buffer.length;
+       data = (char *) recv_buffer.data;
+
+       while (data_left) {
+               data_used = process_incoming_data(ncacn_conn->p, data, data_left);
+               if (data_used < 0) {
+                       DEBUG(3, ("Failed to process dcerpc request!\n"));
+                       status = NT_STATUS_UNEXPECTED_IO_ERROR;
+                       goto fail;
+               }
+
+               data_left -= data_used;
+               data += data_used;
+       }
+
+       /* Do not leak this buffer */
+       talloc_free(recv_buffer.data);
+       talloc_free(pkt);
+
+       /*
+        * This is needed because of the way DCERPC binds work in the RPC
+        * marshalling code
+        */
+       to_send = out->frag.length - out->current_pdu_sent;
+       if (to_send > 0) {
+
+               DEBUG(10, ("Current_pdu_len = %u, "
+                          "current_pdu_sent = %u "
+                          "Returning %u bytes\n",
+                          (unsigned int)out->frag.length,
+                          (unsigned int)out->current_pdu_sent,
+                          (unsigned int)to_send));
+
+               ncacn_conn->iov = talloc_zero(ncacn_conn, struct iovec);
+               if (ncacn_conn->iov == NULL) {
+                       status = NT_STATUS_NO_MEMORY;
+                       DEBUG(3, ("Out of memory!\n"));
+                       goto fail;
+               }
+               ncacn_conn->count = 1;
+
+               ncacn_conn->iov[0].iov_base = out->frag.data
+                                           + out->current_pdu_sent;
+               ncacn_conn->iov[0].iov_len = to_send;
+
+               out->current_pdu_sent += to_send;
+       }
+
+       /*
+        * This condition is false for bind packets, or when we haven't yet got
+        * a full request, and need to wait for more data from the client
+        */
+       while (out->data_sent_length < out->rdata.length) {
+               ok = create_next_pdu(ncacn_conn->p);
+               if (!ok) {
+                       DEBUG(3, ("Failed to create next PDU!\n"));
+                       status = NT_STATUS_UNEXPECTED_IO_ERROR;
+                       goto fail;
+               }
+
+               ncacn_conn->iov = talloc_realloc(ncacn_conn,
+                                                ncacn_conn->iov,
+                                                struct iovec,
+                                                ncacn_conn->count + 1);
+               if (ncacn_conn->iov == NULL) {
+                       DEBUG(3, ("Out of memory!\n"));
+                       status = NT_STATUS_NO_MEMORY;
+                       goto fail;
+               }
+
+               ncacn_conn->iov[ncacn_conn->count].iov_base = out->frag.data;
+               ncacn_conn->iov[ncacn_conn->count].iov_len = out->frag.length;
+
+               DEBUG(10, ("PDU number: %d, PDU Length: %u\n",
+                          (unsigned int) ncacn_conn->count,
+                          (unsigned int) ncacn_conn->iov[ncacn_conn->count].iov_len));
+               dump_data(11, (const uint8_t *) ncacn_conn->iov[ncacn_conn->count].iov_base,
+                             ncacn_conn->iov[ncacn_conn->count].iov_len);
+               ncacn_conn->count++;
+       }
+
+       /*
+        * We still don't have a complete request, go back and wait for more
+        * data.
+        */
+       if (ncacn_conn->count == 0) {
+               /* Wait for the next packet */
+               subreq = dcerpc_read_ncacn_packet_send(ncacn_conn,
+                                                      ncacn_conn->ev_ctx,
+                                                      ncacn_conn->tstream);
+               if (subreq == NULL) {
+                       DEBUG(2, ("Failed to start receving packets\n"));
+                       status = NT_STATUS_NO_MEMORY;
+                       goto fail;
+               }
+               tevent_req_set_callback(subreq, dcerpc_ncacn_packet_process, ncacn_conn);
+               return;
+       }
+
+       DEBUG(10, ("Sending a total of %u bytes\n",
+                  (unsigned int)ncacn_conn->p->out_data.data_sent_length));
+
+       subreq = tstream_writev_queue_send(ncacn_conn,
+                                          ncacn_conn->ev_ctx,
+                                          ncacn_conn->tstream,
+                                          ncacn_conn->send_queue,
+                                          ncacn_conn->iov,
+                                          ncacn_conn->count);
+       if (subreq == NULL) {
+               DEBUG(2, ("Failed to send packet\n"));
+               status = NT_STATUS_NO_MEMORY;
+               goto fail;
+       }
+
+       tevent_req_set_callback(subreq, dcerpc_ncacn_packet_done, ncacn_conn);
+       return;
+
+fail:
+       DEBUG(3, ("Terminating client(%s) connection! - '%s'\n",
+                 ncacn_conn->client_name, nt_errstr(status)));
+
+       /* Terminate client connection */
+       talloc_free(ncacn_conn);
+       return;
+}
+
+static void dcerpc_ncacn_packet_done(struct tevent_req *subreq)
+{
+       struct dcerpc_ncacn_conn *ncacn_conn =
+               tevent_req_callback_data(subreq, struct dcerpc_ncacn_conn);
+       NTSTATUS status = NT_STATUS_OK;
+       int sys_errno;
+       int rc;
+
+       rc = tstream_writev_queue_recv(subreq, &sys_errno);
+       TALLOC_FREE(subreq);
+       if (rc < 0) {
+               DEBUG(2, ("Writev failed!\n"));
+               status = map_nt_error_from_unix(sys_errno);
+               goto fail;
+       }
+
+       /* clear out any data that may have been left around */
+       ncacn_conn->count = 0;
+       TALLOC_FREE(ncacn_conn->iov);
+       data_blob_free(&ncacn_conn->p->in_data.data);
+       data_blob_free(&ncacn_conn->p->out_data.frag);
+       data_blob_free(&ncacn_conn->p->out_data.rdata);
+
+       talloc_free_children(ncacn_conn->p->mem_ctx);
+
+       /* Wait for the next packet */
+       subreq = dcerpc_read_ncacn_packet_send(ncacn_conn,
+                                              ncacn_conn->ev_ctx,
+                                              ncacn_conn->tstream);
+       if (subreq == NULL) {
+               DEBUG(2, ("Failed to start receving packets\n"));
+               status = NT_STATUS_NO_MEMORY;
+               goto fail;
+       }
+
+       tevent_req_set_callback(subreq, dcerpc_ncacn_packet_process, ncacn_conn);
+       return;
+
+fail:
+       DEBUG(3, ("Terminating client(%s) connection! - '%s'\n",
+                 ncacn_conn->client_name, nt_errstr(status)));
+
+       /* Terminate client connection */
+       talloc_free(ncacn_conn);
+       return;
+}
+
+/* vim: set ts=8 sw=8 noet cindent syntax=c.doxygen: */