Line data Source code
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 "librpc/rpc/dcerpc_util.h"
23 : #include "rpc_client/rpc_transport.h"
24 : #include "librpc/ndr/ndr_table.h"
25 : #include "libcli/smb/smbXcli_base.h"
26 : #include "libcli/smb/tstream_smbXcli_np.h"
27 : #include "client.h"
28 :
29 : #undef DBGC_CLASS
30 : #define DBGC_CLASS DBGC_RPC_CLI
31 :
32 : struct rpc_transport_np_init_state {
33 : struct rpc_cli_transport *transport;
34 : int retries;
35 : struct tevent_context *ev;
36 : struct smbXcli_conn *conn;
37 : int timeout;
38 : struct timeval abs_timeout;
39 : const char *pipe_name;
40 : struct smbXcli_session *session;
41 : struct smbXcli_tcon *tcon;
42 : uint16_t pid;
43 : };
44 :
45 : static void rpc_transport_np_init_pipe_open(struct tevent_req *subreq);
46 :
47 10060 : struct tevent_req *rpc_transport_np_init_send(TALLOC_CTX *mem_ctx,
48 : struct tevent_context *ev,
49 : struct cli_state *cli,
50 : const struct ndr_interface_table *table)
51 : {
52 0 : struct tevent_req *req;
53 0 : struct rpc_transport_np_init_state *state;
54 0 : struct tevent_req *subreq;
55 :
56 10060 : req = tevent_req_create(mem_ctx, &state,
57 : struct rpc_transport_np_init_state);
58 10060 : if (req == NULL) {
59 0 : return NULL;
60 : }
61 :
62 10060 : if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
63 9916 : state->tcon = cli->smb2.tcon;
64 9916 : state->session = cli->smb2.session;
65 : } else {
66 144 : state->tcon = cli->smb1.tcon;
67 144 : state->session = cli->smb1.session;
68 144 : state->pid = cli->smb1.pid;
69 : }
70 :
71 10060 : state->ev = ev;
72 10060 : state->conn = cli->conn;
73 10060 : state->timeout = cli->timeout;
74 10060 : state->abs_timeout = timeval_current_ofs_msec(cli->timeout);
75 10060 : state->pipe_name = dcerpc_default_transport_endpoint(state, NCACN_NP,
76 : table);
77 10060 : if (tevent_req_nomem(state->pipe_name, req)) {
78 0 : return tevent_req_post(req, ev);
79 : }
80 :
81 10060 : while (state->pipe_name[0] == '\\') {
82 0 : state->pipe_name++;
83 : }
84 :
85 10060 : subreq = tstream_smbXcli_np_open_send(state, ev, state->conn,
86 10060 : state->session, state->tcon,
87 10060 : state->pid, state->timeout,
88 10060 : state->pipe_name);
89 10060 : if (tevent_req_nomem(subreq, req)) {
90 0 : return tevent_req_post(req, ev);
91 : }
92 10060 : tevent_req_set_callback(subreq, rpc_transport_np_init_pipe_open, req);
93 :
94 10060 : return req;
95 : }
96 :
97 0 : static void rpc_transport_np_init_pipe_open_retry(struct tevent_context *ev,
98 : struct tevent_timer *te,
99 : struct timeval t,
100 : void *priv_data)
101 : {
102 0 : struct tevent_req *subreq;
103 0 : struct tevent_req *req = talloc_get_type(priv_data, struct tevent_req);
104 0 : struct rpc_transport_np_init_state *state = tevent_req_data(
105 : req, struct rpc_transport_np_init_state);
106 :
107 0 : subreq = tstream_smbXcli_np_open_send(state, ev,
108 : state->conn,
109 : state->session,
110 : state->tcon,
111 0 : state->pid,
112 0 : state->timeout,
113 : state->pipe_name);
114 0 : if (tevent_req_nomem(subreq, req)) {
115 0 : return;
116 : }
117 0 : tevent_req_set_callback(subreq, rpc_transport_np_init_pipe_open, req);
118 0 : state->retries++;
119 : }
120 :
121 10060 : static void rpc_transport_np_init_pipe_open(struct tevent_req *subreq)
122 : {
123 10060 : struct tevent_req *req = tevent_req_callback_data(
124 : subreq, struct tevent_req);
125 10060 : struct rpc_transport_np_init_state *state = tevent_req_data(
126 : req, struct rpc_transport_np_init_state);
127 0 : NTSTATUS status;
128 0 : struct tstream_context *stream;
129 :
130 10060 : status = tstream_smbXcli_np_open_recv(subreq, state, &stream);
131 10060 : TALLOC_FREE(subreq);
132 10060 : if (NT_STATUS_EQUAL(status, NT_STATUS_PIPE_NOT_AVAILABLE)
133 0 : && (!timeval_expired(&state->abs_timeout))) {
134 0 : struct tevent_timer *te;
135 : /*
136 : * Retry on STATUS_PIPE_NOT_AVAILABLE, Windows starts some
137 : * servers (FssagentRpc) on demand.
138 : */
139 0 : DEBUG(2, ("RPC pipe %s not available, retry %d\n",
140 : state->pipe_name, state->retries));
141 0 : te = tevent_add_timer(state->ev, state,
142 : timeval_current_ofs_msec(100 * state->retries),
143 : rpc_transport_np_init_pipe_open_retry, req);
144 0 : if (tevent_req_nomem(te, req)) {
145 0 : DEBUG(2, ("Failed to create asynchronous "
146 : "tevent_timer\n"));
147 : }
148 0 : return;
149 : }
150 :
151 10060 : if (tevent_req_nterror(req, status)) {
152 0 : return;
153 : }
154 :
155 10060 : status = rpc_transport_tstream_init(state,
156 : &stream,
157 : &state->transport);
158 10060 : if (tevent_req_nterror(req, status)) {
159 0 : return;
160 : }
161 :
162 10060 : tevent_req_done(req);
163 : }
164 :
165 10060 : NTSTATUS rpc_transport_np_init_recv(struct tevent_req *req,
166 : TALLOC_CTX *mem_ctx,
167 : struct rpc_cli_transport **presult)
168 : {
169 10060 : struct rpc_transport_np_init_state *state = tevent_req_data(
170 : req, struct rpc_transport_np_init_state);
171 0 : NTSTATUS status;
172 :
173 10060 : if (tevent_req_is_nterror(req, &status)) {
174 0 : return status;
175 : }
176 :
177 10060 : *presult = talloc_move(mem_ctx, &state->transport);
178 10060 : return NT_STATUS_OK;
179 : }
|
