s3-rpc_server: Added helper functions to read data from a ncacn socket.

Signed-off-by: Andreas Schneider <asn@cynapses.org>

diff --git a/source3/librpc/rpc/dcerpc.h b/source3/librpc/rpc/dcerpc.h
index 56d6d320fdeb7fa64a4d7544bee67ae52106e375..318364575f90fdf3e23446d0afd00dc1f6010503 100644 (file)
--- a/source3/librpc/rpc/dcerpc.h
+++ b/source3/librpc/rpc/dcerpc.h
@@ -168,6 +168,10 @@ struct ndr_pull;
 struct tevent_context;
 struct tstream_context;
 
+/* from ../librpc/rpc/dcerpc_util.c */
+void dcerpc_set_frag_length(DATA_BLOB *blob, uint16_t v);
+uint16_t dcerpc_get_frag_length(const DATA_BLOB *blob);
+void dcerpc_set_auth_length(DATA_BLOB *blob, uint16_t v);
 struct tevent_req *dcerpc_read_ncacn_packet_send(TALLOC_CTX *mem_ctx,
                                                 struct tevent_context *ev,
                                                 struct tstream_context *stream);

diff --git a/source3/rpc_server/rpc_server.c b/source3/rpc_server/rpc_server.c
index 63484bb1767f4ac569fb09472baca709098bd1e8..89e224714c4e96d7fbcb40c18f464d492d15978a 100644 (file)
--- a/source3/rpc_server/rpc_server.c
+++ b/source3/rpc_server/rpc_server.c
@@ -22,6 +22,7 @@
 #include "rpc_dce.h"
 #include "librpc/gen_ndr/netlogon.h"
 #include "registry/reg_parse_prs.h"
+#include "lib/tsocket/tsocket.h"
 
 /* Creates a pipes_struct and initializes it with the information
  * sent from the client */
@@ -118,6 +119,85 @@ static int make_server_pipes_struct(TALLOC_CTX *mem_ctx,
        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 {
        int fd;