source3/rpc_client/rpc_transport_np.c

   1 /*
   2  *  Unix SMB/CIFS implementation.
   3  *  RPC client transport over named pipes
   4  *  Copyright (C) Volker Lendecke 2009
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include "includes.h"
  21 #include "../lib/util/tevent_ntstatus.h"
  22 #include "rpc_client/rpc_transport.h"
  23 #include "libsmb/cli_np_tstream.h"
  24
  25 #undef DBGC_CLASS
  26 #define DBGC_CLASS DBGC_RPC_CLI
  27
  28 struct rpc_transport_np_init_state {
  29         struct rpc_cli_transport *transport;
  30 };
  31
  32 static void rpc_transport_np_init_pipe_open(struct tevent_req *subreq);
  33
  34 struct tevent_req *rpc_transport_np_init_send(TALLOC_CTX *mem_ctx,
  35                                               struct event_context *ev,
  36                                               struct cli_state *cli,
  37                                               const struct ndr_syntax_id *abstract_syntax)
  38 {
  39         struct tevent_req *req;
  40         struct rpc_transport_np_init_state *state;
  41         const char *pipe_name;
  42         struct tevent_req *subreq;
  43
  44         req = tevent_req_create(mem_ctx, &state,
  45                                 struct rpc_transport_np_init_state);
  46         if (req == NULL) {
  47                 return NULL;
  48         }
  49
  50         pipe_name = get_pipe_name_from_syntax(state, abstract_syntax);
  51         if (tevent_req_nomem(pipe_name, req)) {
  52                 return tevent_req_post(req, ev);
  53         }
  54
  55         subreq = tstream_cli_np_open_send(state, ev, cli, pipe_name);
  56         if (tevent_req_nomem(subreq, req)) {
  57                 return tevent_req_post(req, ev);
  58         }
  59         tevent_req_set_callback(subreq, rpc_transport_np_init_pipe_open, req);
  60
  61         return req;
  62 }
  63
  64 static void rpc_transport_np_init_pipe_open(struct tevent_req *subreq)
  65 {
  66         struct tevent_req *req = tevent_req_callback_data(
  67                 subreq, struct tevent_req);
  68         struct rpc_transport_np_init_state *state = tevent_req_data(
  69                 req, struct rpc_transport_np_init_state);
  70         NTSTATUS status;
  71         struct tstream_context *stream;
  72
  73         status = tstream_cli_np_open_recv(subreq, state, &stream);
  74         TALLOC_FREE(subreq);
  75         if (!NT_STATUS_IS_OK(status)) {
  76                 tevent_req_nterror(req, status);
  77                 return;
  78         }
  79
  80         status = rpc_transport_tstream_init(state,
  81                                             &stream,
  82                                             &state->transport);
  83         if (!NT_STATUS_IS_OK(status)) {
  84                 tevent_req_nterror(req, status);
  85                 return;
  86         }
  87
  88         tevent_req_done(req);
  89 }
  90
  91 NTSTATUS rpc_transport_np_init_recv(struct tevent_req *req,
  92                                     TALLOC_CTX *mem_ctx,
  93                                     struct rpc_cli_transport **presult)
  94 {
  95         struct rpc_transport_np_init_state *state = tevent_req_data(
  96                 req, struct rpc_transport_np_init_state);
  97         NTSTATUS status;
  98
  99         if (tevent_req_is_nterror(req, &status)) {
 100                 return status;
 101         }
 102
 103         *presult = talloc_move(mem_ctx, &state->transport);
 104         return NT_STATUS_OK;
 105 }
 106
 107 NTSTATUS rpc_transport_np_init(TALLOC_CTX *mem_ctx, struct cli_state *cli,
 108                                const struct ndr_syntax_id *abstract_syntax,
 109                                struct rpc_cli_transport **presult)
 110 {
 111         TALLOC_CTX *frame = talloc_stackframe();
 112         struct event_context *ev;
 113         struct tevent_req *req;
 114         NTSTATUS status = NT_STATUS_OK;
 115
 116         ev = event_context_init(frame);
 117         if (ev == NULL) {
 118                 status = NT_STATUS_NO_MEMORY;
 119                 goto fail;
 120         }
 121
 122         req = rpc_transport_np_init_send(frame, ev, cli, abstract_syntax);
 123         if (req == NULL) {
 124                 status = NT_STATUS_NO_MEMORY;
 125                 goto fail;
 126         }
 127
 128         if (!tevent_req_poll(req, ev)) {
 129                 status = map_nt_error_from_unix(errno);
 130                 goto fail;
 131         }
 132
 133         status = rpc_transport_np_init_recv(req, mem_ctx, presult);
 134  fail:
 135         TALLOC_FREE(frame);
 136         return status;
 137 }