2 * Unix SMB/CIFS implementation.
3 * RPC Pipe client / server routines
4 * Copyright (C) Andrew Tridgell 1992-1998,
5 * Largely re-written : 2005
6 * Copyright (C) Jeremy Allison 1998 - 2005
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/>.
23 #include "fake_file.h"
26 #include "rpc_server/rpc_ncacn_np.h"
27 #include "rpc_server/srv_pipe_hnd.h"
28 #include "rpc_server/srv_pipe.h"
29 #include "rpc_server/rpc_server.h"
30 #include "rpc_server/rpc_config.h"
31 #include "../lib/tsocket/tsocket.h"
32 #include "../lib/util/tevent_ntstatus.h"
33 #include "librpc/ndr/ndr_table.h"
36 #define DBGC_CLASS DBGC_RPC_SRV
38 bool fsp_is_np(struct files_struct *fsp)
40 enum FAKE_FILE_TYPE type;
42 if ((fsp == NULL) || (fsp->fake_file_handle == NULL)) {
46 type = fsp->fake_file_handle->type;
48 return (type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY);
51 NTSTATUS np_open(TALLOC_CTX *mem_ctx, const char *name,
52 const struct tsocket_address *local_address,
53 const struct tsocket_address *remote_address,
54 struct auth_session_info *session_info,
55 struct tevent_context *ev_ctx,
56 struct messaging_context *msg_ctx,
57 struct fake_file_handle **phandle)
59 enum rpc_service_mode_e pipe_mode;
60 const char **proxy_list;
61 struct fake_file_handle *handle;
62 struct ndr_syntax_id syntax;
63 struct npa_state *npa = NULL;
67 proxy_list = lp_parm_string_list(-1, "np", "proxy", NULL);
69 handle = talloc(mem_ctx, struct fake_file_handle);
71 return NT_STATUS_NO_MEMORY;
74 /* Check what is the server type for this pipe.
75 Defaults to "embedded" */
76 pipe_mode = rpc_service_mode(name);
78 /* Still support the old method for defining external servers */
79 if ((proxy_list != NULL) && str_list_check_ci(proxy_list, name)) {
80 pipe_mode = RPC_SERVICE_MODE_EXTERNAL;
84 case RPC_SERVICE_MODE_EXTERNAL:
85 status = make_external_rpc_pipe(handle,
91 if (!NT_STATUS_IS_OK(status)) {
96 handle->private_data = (void *)npa;
97 handle->type = FAKE_FILE_TYPE_NAMED_PIPE_PROXY;
100 case RPC_SERVICE_MODE_EMBEDDED:
101 /* Check if we this daemon handles this pipe */
102 ok = is_known_pipename(name, &syntax);
104 DEBUG(0, ("ERROR! '%s' is not a registred pipe!\n",
107 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
110 status = make_internal_rpc_pipe_socketpair(handle,
118 if (!NT_STATUS_IS_OK(status)) {
123 handle->private_data = (void *)npa;
124 handle->type = FAKE_FILE_TYPE_NAMED_PIPE_PROXY;
127 case RPC_SERVICE_MODE_DISABLED:
129 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
137 bool np_read_in_progress(struct fake_file_handle *handle)
139 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY) {
140 struct npa_state *p =
141 talloc_get_type_abort(handle->private_data,
145 read_count = tevent_queue_length(p->read_queue);
146 if (read_count > 0) {
156 struct np_write_state {
157 struct tevent_context *ev;
163 static void np_write_done(struct tevent_req *subreq);
165 struct tevent_req *np_write_send(TALLOC_CTX *mem_ctx, struct tevent_context *ev,
166 struct fake_file_handle *handle,
167 const uint8_t *data, size_t len)
169 struct tevent_req *req;
170 struct np_write_state *state;
173 DEBUG(6, ("np_write_send: len: %d\n", (int)len));
174 dump_data(50, data, len);
176 req = tevent_req_create(mem_ctx, &state, struct np_write_state);
183 status = NT_STATUS_OK;
187 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY) {
188 struct npa_state *p = talloc_get_type_abort(
189 handle->private_data, struct npa_state);
190 struct tevent_req *subreq;
194 state->iov.iov_base = discard_const_p(void, data);
195 state->iov.iov_len = len;
197 subreq = tstream_writev_queue_send(state, ev,
201 if (subreq == NULL) {
204 tevent_req_set_callback(subreq, np_write_done, req);
208 status = NT_STATUS_INVALID_HANDLE;
210 if (NT_STATUS_IS_OK(status)) {
211 tevent_req_done(req);
213 tevent_req_nterror(req, status);
215 return tevent_req_post(req, ev);
221 static void np_write_done(struct tevent_req *subreq)
223 struct tevent_req *req = tevent_req_callback_data(
224 subreq, struct tevent_req);
225 struct np_write_state *state = tevent_req_data(
226 req, struct np_write_state);
230 received = tstream_writev_queue_recv(subreq, &err);
232 tevent_req_nterror(req, map_nt_error_from_unix(err));
235 state->nwritten = received;
236 tevent_req_done(req);
239 NTSTATUS np_write_recv(struct tevent_req *req, ssize_t *pnwritten)
241 struct np_write_state *state = tevent_req_data(
242 req, struct np_write_state);
245 if (tevent_req_is_nterror(req, &status)) {
248 *pnwritten = state->nwritten;
252 struct np_ipc_readv_next_vector_state {
259 static void np_ipc_readv_next_vector_init(struct np_ipc_readv_next_vector_state *s,
260 uint8_t *buf, size_t len)
265 s->len = MIN(len, UINT16_MAX);
268 static int np_ipc_readv_next_vector(struct tstream_context *stream,
271 struct iovec **_vector,
274 struct np_ipc_readv_next_vector_state *state =
275 (struct np_ipc_readv_next_vector_state *)private_data;
276 struct iovec *vector;
280 if (state->ofs == state->len) {
286 pending = tstream_pending_bytes(stream);
291 if (pending == 0 && state->ofs != 0) {
292 /* return a short read */
299 /* we want at least one byte and recheck again */
302 size_t missing = state->len - state->ofs;
303 if (pending > missing) {
304 /* there's more available */
305 state->remaining = pending - missing;
308 /* read what we can get and recheck in the next cycle */
313 vector = talloc_array(mem_ctx, struct iovec, 1);
318 vector[0].iov_base = state->buf + state->ofs;
319 vector[0].iov_len = wanted;
321 state->ofs += wanted;
328 struct np_read_state {
330 struct np_ipc_readv_next_vector_state next_vector;
333 bool is_data_outstanding;
336 static void np_read_done(struct tevent_req *subreq);
338 struct tevent_req *np_read_send(TALLOC_CTX *mem_ctx, struct tevent_context *ev,
339 struct fake_file_handle *handle,
340 uint8_t *data, size_t len)
342 struct tevent_req *req;
343 struct np_read_state *state;
346 req = tevent_req_create(mem_ctx, &state, struct np_read_state);
351 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY) {
352 struct npa_state *p = talloc_get_type_abort(
353 handle->private_data, struct npa_state);
354 struct tevent_req *subreq;
356 np_ipc_readv_next_vector_init(&state->next_vector,
359 subreq = tstream_readv_pdu_queue_send(state,
363 np_ipc_readv_next_vector,
364 &state->next_vector);
365 if (subreq == NULL) {
366 status = NT_STATUS_NO_MEMORY;
369 tevent_req_set_callback(subreq, np_read_done, req);
373 status = NT_STATUS_INVALID_HANDLE;
375 if (NT_STATUS_IS_OK(status)) {
376 tevent_req_done(req);
378 tevent_req_nterror(req, status);
380 return tevent_req_post(req, ev);
383 static void np_read_done(struct tevent_req *subreq)
385 struct tevent_req *req = tevent_req_callback_data(
386 subreq, struct tevent_req);
387 struct np_read_state *state = tevent_req_data(
388 req, struct np_read_state);
392 ret = tstream_readv_pdu_queue_recv(subreq, &err);
395 tevent_req_nterror(req, map_nt_error_from_unix(err));
400 state->is_data_outstanding = (state->next_vector.remaining > 0);
402 tevent_req_done(req);
406 NTSTATUS np_read_recv(struct tevent_req *req, ssize_t *nread,
407 bool *is_data_outstanding)
409 struct np_read_state *state = tevent_req_data(
410 req, struct np_read_state);
413 if (tevent_req_is_nterror(req, &status)) {
417 DEBUG(10, ("Received %d bytes. There is %smore data outstanding\n",
418 (int)state->nread, state->is_data_outstanding?"":"no "));
420 *nread = state->nread;
421 *is_data_outstanding = state->is_data_outstanding;