From 738c843e99b780e8b4457eef8c4f94d23e08c754 Mon Sep 17 00:00:00 2001 From: Simo Sorce Date: Tue, 6 Jul 2010 13:03:35 -0400 Subject: s3-rpc_server: Renamed rpc_ncacn_np_internal.c. Signed-off-by: Andreas Schneider --- source3/Makefile.in | 9 +- source3/rpc_server/rpc_ncacn_np.c | 577 +++++++++++++++++++++++++++++ source3/rpc_server/rpc_ncacn_np_internal.c | 577 ----------------------------- 3 files changed, 582 insertions(+), 581 deletions(-) create mode 100644 source3/rpc_server/rpc_ncacn_np.c delete mode 100644 source3/rpc_server/rpc_ncacn_np_internal.c (limited to 'source3') diff --git a/source3/Makefile.in b/source3/Makefile.in index 51d64bdd6f..863274575a 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -706,13 +706,13 @@ RPC_EVENTLOG_OBJ = rpc_server/srv_eventlog_nt.o \ NPA_TSTREAM_OBJ = ../libcli/named_pipe_auth/npa_tstream.o \ librpc/gen_ndr/ndr_named_pipe_auth.o -RPC_NCACN_NP_INTERNAL = rpc_server/srv_pipe_register.o rpc_server/rpc_ncacn_np_internal.o \ - rpc_server/rpc_handles.o +RPC_NCACN_NP = rpc_server/srv_pipe_register.o rpc_server/rpc_ncacn_np.o \ + rpc_server/rpc_handles.o RPC_SERVICE = rpc_server/rpc_server.o RPC_PIPE_OBJ = rpc_server/srv_pipe.o rpc_server/srv_pipe_hnd.o \ - $(RPC_NCACN_NP_INTERNAL) $(RPC_SERVICE) + $(RPC_NCACN_NP) $(RPC_SERVICE) RPC_RPCECHO_OBJ = rpc_server/srv_echo_nt.o librpc/gen_ndr/srv_echo.o @@ -1386,7 +1386,8 @@ WINBINDD_OBJ = \ $(PROFILE_OBJ) $(SLCACHE_OBJ) $(SMBLDAP_OBJ) \ $(LIBADS_OBJ) $(KRBCLIENT_OBJ) $(POPT_LIB_OBJ) \ $(DCUTIL_OBJ) $(IDMAP_OBJ) $(NSS_INFO_OBJ) \ - $(RPC_NCACN_NP_INTERNAL) $(RPC_SAMR_OBJ) $(RPC_LSARPC_OBJ) \ + $(RPC_NCACN_NP) $(RPC_SAMR_OBJ) $(RPC_LSARPC_OBJ) \ + $(NPA_TSTREAM_OBJ) \ $(AFS_OBJ) $(AFS_SETTOKEN_OBJ) \ $(LIBADS_SERVER_OBJ) \ $(TDB_VALIDATE_OBJ) \ diff --git a/source3/rpc_server/rpc_ncacn_np.c b/source3/rpc_server/rpc_ncacn_np.c new file mode 100644 index 0000000000..8253f65bb7 --- /dev/null +++ b/source3/rpc_server/rpc_ncacn_np.c @@ -0,0 +1,577 @@ +/* + * Unix SMB/CIFS implementation. + * RPC Pipe client / server routines + * Copyright (C) Andrew Tridgell 1992-1998, + * Largely re-written : 2005 + * Copyright (C) Jeremy Allison 1998 - 2005 + * + * 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 + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, see . + */ + +#include "includes.h" +#include "rpc_server/srv_pipe_internal.h" +#include "rpc_dce.h" + +#undef DBGC_CLASS +#define DBGC_CLASS DBGC_RPC_SRV + +static int pipes_open; + +static struct pipes_struct *InternalPipes; + +/* TODO + * the following prototypes are declared here to avoid + * code being moved about too much for a patch to be + * disrupted / less obvious. + * + * these functions, and associated functions that they + * call, should be moved behind a .so module-loading + * system _anyway_. so that's the next step... + */ + +/**************************************************************************** + Internal Pipe iterator functions. +****************************************************************************/ + +struct pipes_struct *get_first_internal_pipe(void) +{ + return InternalPipes; +} + +struct pipes_struct *get_next_internal_pipe(struct pipes_struct *p) +{ + return p->next; +} + +static void free_pipe_rpc_context_internal( PIPE_RPC_FNS *list ) +{ + PIPE_RPC_FNS *tmp = list; + PIPE_RPC_FNS *tmp2; + + while (tmp) { + tmp2 = tmp->next; + SAFE_FREE(tmp); + tmp = tmp2; + } + + return; +} + +bool check_open_pipes(void) +{ + struct pipes_struct *p; + + for (p = InternalPipes; p != NULL; p = p->next) { + if (num_pipe_handles(p) != 0) { + return true; + } + } + return false; +} + +/**************************************************************************** + Close an rpc pipe. +****************************************************************************/ + +int close_internal_rpc_pipe_hnd(struct pipes_struct *p) +{ + if (!p) { + DEBUG(0,("Invalid pipe in close_internal_rpc_pipe_hnd\n")); + return False; + } + + if (p->auth.auth_data_free_func) { + (*p->auth.auth_data_free_func)(&p->auth); + } + + free_pipe_rpc_context_internal( p->contexts ); + + /* Free the handles database. */ + close_policy_by_pipe(p); + + DLIST_REMOVE(InternalPipes, p); + + ZERO_STRUCTP(p); + + return 0; +} + +/**************************************************************************** + Make an internal namedpipes structure +****************************************************************************/ + +struct pipes_struct *make_internal_rpc_pipe_p(TALLOC_CTX *mem_ctx, + const struct ndr_syntax_id *syntax, + struct client_address *client_id, + const struct auth_serversupplied_info *server_info, + struct messaging_context *msg_ctx) +{ + struct pipes_struct *p; + + DEBUG(4,("Create pipe requested %s\n", + get_pipe_name_from_syntax(talloc_tos(), syntax))); + + p = TALLOC_ZERO_P(mem_ctx, struct pipes_struct); + + if (!p) { + DEBUG(0,("ERROR! no memory for pipes_struct!\n")); + return NULL; + } + + p->mem_ctx = talloc_named(p, 0, "pipe %s %p", + get_pipe_name_from_syntax(talloc_tos(), + syntax), p); + if (p->mem_ctx == NULL) { + DEBUG(0,("open_rpc_pipe_p: talloc_init failed.\n")); + TALLOC_FREE(p); + return NULL; + } + + if (!init_pipe_handles(p, syntax)) { + DEBUG(0,("open_rpc_pipe_p: init_pipe_handles failed.\n")); + TALLOC_FREE(p); + return NULL; + } + + p->server_info = copy_serverinfo(p, server_info); + if (p->server_info == NULL) { + DEBUG(0, ("open_rpc_pipe_p: copy_serverinfo failed\n")); + close_policy_by_pipe(p); + TALLOC_FREE(p); + return NULL; + } + + p->msg_ctx = msg_ctx; + + DLIST_ADD(InternalPipes, p); + + p->client_id = client_id; + + p->endian = RPC_LITTLE_ENDIAN; + + p->syntax = *syntax; + + DEBUG(4,("Created internal pipe %s (pipes_open=%d)\n", + get_pipe_name_from_syntax(talloc_tos(), syntax), pipes_open)); + + talloc_set_destructor(p, close_internal_rpc_pipe_hnd); + + return p; +} + +static NTSTATUS rpcint_dispatch(struct pipes_struct *p, + TALLOC_CTX *mem_ctx, + uint32_t opnum, + const DATA_BLOB *in_data, + DATA_BLOB *out_data) +{ + uint32_t num_cmds = rpc_srv_get_pipe_num_cmds(&p->syntax); + const struct api_struct *cmds = rpc_srv_get_pipe_cmds(&p->syntax); + uint32_t i; + bool ok; + + /* set opnum */ + p->opnum = opnum; + + for (i = 0; i < num_cmds; i++) { + if (cmds[i].opnum == opnum && cmds[i].fn != NULL) { + break; + } + } + + if (i == num_cmds) { + return NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE; + } + + p->in_data.data = *in_data; + p->out_data.rdata = data_blob_null; + + ok = cmds[i].fn(p); + p->in_data.data = data_blob_null; + if (!ok) { + data_blob_free(&p->out_data.rdata); + talloc_free_children(p->mem_ctx); + return NT_STATUS_RPC_CALL_FAILED; + } + + if (p->fault_state) { + p->fault_state = false; + data_blob_free(&p->out_data.rdata); + talloc_free_children(p->mem_ctx); + return NT_STATUS_RPC_CALL_FAILED; + } + + if (p->bad_handle_fault_state) { + p->bad_handle_fault_state = false; + data_blob_free(&p->out_data.rdata); + talloc_free_children(p->mem_ctx); + return NT_STATUS_RPC_SS_CONTEXT_MISMATCH; + } + + if (p->rng_fault_state) { + p->rng_fault_state = false; + data_blob_free(&p->out_data.rdata); + talloc_free_children(p->mem_ctx); + return NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE; + } + + *out_data = p->out_data.rdata; + talloc_steal(mem_ctx, out_data->data); + p->out_data.rdata = data_blob_null; + + talloc_free_children(p->mem_ctx); + return NT_STATUS_OK; +} + +struct rpcint_bh_state { + struct pipes_struct *p; +}; + +static bool rpcint_bh_is_connected(struct dcerpc_binding_handle *h) +{ + struct rpcint_bh_state *hs = dcerpc_binding_handle_data(h, + struct rpcint_bh_state); + + if (!hs->p) { + return false; + } + + return true; +} + +static uint32_t rpcint_bh_set_timeout(struct dcerpc_binding_handle *h, + uint32_t timeout) +{ + /* TODO: implement timeouts */ + return UINT32_MAX; +} + +struct rpcint_bh_raw_call_state { + DATA_BLOB in_data; + DATA_BLOB out_data; + uint32_t out_flags; +}; + +static struct tevent_req *rpcint_bh_raw_call_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct dcerpc_binding_handle *h, + const struct GUID *object, + uint32_t opnum, + uint32_t in_flags, + const uint8_t *in_data, + size_t in_length) +{ + struct rpcint_bh_state *hs = + dcerpc_binding_handle_data(h, + struct rpcint_bh_state); + struct tevent_req *req; + struct rpcint_bh_raw_call_state *state; + bool ok; + NTSTATUS status; + + req = tevent_req_create(mem_ctx, &state, + struct rpcint_bh_raw_call_state); + if (req == NULL) { + return NULL; + } + state->in_data.data = discard_const_p(uint8_t, in_data); + state->in_data.length = in_length; + + ok = rpcint_bh_is_connected(h); + if (!ok) { + tevent_req_nterror(req, NT_STATUS_INVALID_CONNECTION); + return tevent_req_post(req, ev); + } + + /* TODO: allow async */ + status = rpcint_dispatch(hs->p, state, opnum, + &state->in_data, + &state->out_data); + if (!NT_STATUS_IS_OK(status)) { + tevent_req_nterror(req, status); + return tevent_req_post(req, ev); + } + + tevent_req_done(req); + return tevent_req_post(req, ev); +} + +static NTSTATUS rpcint_bh_raw_call_recv(struct tevent_req *req, + TALLOC_CTX *mem_ctx, + uint8_t **out_data, + size_t *out_length, + uint32_t *out_flags) +{ + struct rpcint_bh_raw_call_state *state = + tevent_req_data(req, + struct rpcint_bh_raw_call_state); + NTSTATUS status; + + if (tevent_req_is_nterror(req, &status)) { + tevent_req_received(req); + return status; + } + + *out_data = talloc_move(mem_ctx, &state->out_data.data); + *out_length = state->out_data.length; + *out_flags = 0; + tevent_req_received(req); + return NT_STATUS_OK; +} + +struct rpcint_bh_disconnect_state { + uint8_t _dummy; +}; + +static struct tevent_req *rpcint_bh_disconnect_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct dcerpc_binding_handle *h) +{ + struct rpcint_bh_state *hs = dcerpc_binding_handle_data(h, + struct rpcint_bh_state); + struct tevent_req *req; + struct rpcint_bh_disconnect_state *state; + bool ok; + + req = tevent_req_create(mem_ctx, &state, + struct rpcint_bh_disconnect_state); + if (req == NULL) { + return NULL; + } + + ok = rpcint_bh_is_connected(h); + if (!ok) { + tevent_req_nterror(req, NT_STATUS_INVALID_CONNECTION); + return tevent_req_post(req, ev); + } + + /* + * TODO: do a real async disconnect ... + * + * For now the caller needs to free pipes_struct + */ + hs->p = NULL; + + tevent_req_done(req); + return tevent_req_post(req, ev); +} + +static NTSTATUS rpcint_bh_disconnect_recv(struct tevent_req *req) +{ + NTSTATUS status; + + if (tevent_req_is_nterror(req, &status)) { + tevent_req_received(req); + return status; + } + + tevent_req_received(req); + return NT_STATUS_OK; +} + +static bool rpcint_bh_ref_alloc(struct dcerpc_binding_handle *h) +{ + return true; +} + +static void rpcint_bh_do_ndr_print(struct dcerpc_binding_handle *h, + int ndr_flags, + const void *_struct_ptr, + const struct ndr_interface_call *call) +{ + void *struct_ptr = discard_const(_struct_ptr); + + if (DEBUGLEVEL < 10) { + return; + } + + if (ndr_flags & NDR_IN) { + ndr_print_function_debug(call->ndr_print, + call->name, + ndr_flags, + struct_ptr); + } + if (ndr_flags & NDR_OUT) { + ndr_print_function_debug(call->ndr_print, + call->name, + ndr_flags, + struct_ptr); + } +} + +static const struct dcerpc_binding_handle_ops rpcint_bh_ops = { + .name = "rpcint", + .is_connected = rpcint_bh_is_connected, + .set_timeout = rpcint_bh_set_timeout, + .raw_call_send = rpcint_bh_raw_call_send, + .raw_call_recv = rpcint_bh_raw_call_recv, + .disconnect_send = rpcint_bh_disconnect_send, + .disconnect_recv = rpcint_bh_disconnect_recv, + + .ref_alloc = rpcint_bh_ref_alloc, + .do_ndr_print = rpcint_bh_do_ndr_print, +}; + +static NTSTATUS rpcint_binding_handle_ex(TALLOC_CTX *mem_ctx, + const struct ndr_syntax_id *abstract_syntax, + const struct ndr_interface_table *ndr_table, + struct client_address *client_id, + const struct auth_serversupplied_info *server_info, + struct messaging_context *msg_ctx, + struct dcerpc_binding_handle **binding_handle) +{ + struct dcerpc_binding_handle *h; + struct rpcint_bh_state *hs; + + if (ndr_table) { + abstract_syntax = &ndr_table->syntax_id; + } + + h = dcerpc_binding_handle_create(mem_ctx, + &rpcint_bh_ops, + NULL, + ndr_table, + &hs, + struct rpcint_bh_state, + __location__); + if (h == NULL) { + return NT_STATUS_NO_MEMORY; + } + hs->p = make_internal_rpc_pipe_p(hs, + abstract_syntax, + client_id, + server_info, + msg_ctx); + if (hs->p == NULL) { + TALLOC_FREE(h); + return NT_STATUS_NO_MEMORY; + } + + *binding_handle = h; + return NT_STATUS_OK; +} +/** + * @brief Create a new DCERPC Binding Handle which uses a local dispatch function. + * + * @param[in] mem_ctx The memory context to use. + * + * @param[in] ndr_table Normally the ndr_table_. + * + * @param[in] client_id The info about the connected client. + * + * @param[in] serversupplied_info The server supplied authentication function. + * + * @param[in] msg_ctx The messaging context that can be used by the server + * + * @param[out] binding_handle A pointer to store the connected + * dcerpc_binding_handle + * + * @return NT_STATUS_OK on success, a corresponding NT status if an + * error occured. + * + * @code + * struct dcerpc_binding_handle *winreg_binding; + * NTSTATUS status; + * + * status = rpcint_binding_handle(tmp_ctx, + * &ndr_table_winreg, + * p->client_id, + * p->server_info, + * p->msg_ctx + * &winreg_binding); + * @endcode + */ +NTSTATUS rpcint_binding_handle(TALLOC_CTX *mem_ctx, + const struct ndr_interface_table *ndr_table, + struct client_address *client_id, + const struct auth_serversupplied_info *server_info, + struct messaging_context *msg_ctx, + struct dcerpc_binding_handle **binding_handle) +{ + return rpcint_binding_handle_ex(mem_ctx, NULL, ndr_table, client_id, + server_info, msg_ctx, binding_handle); +} + +/** + * @brief Create a new RPC client context which uses a local dispatch function. + * + * @param[in] mem_ctx The memory context to use. + * + * @param[in] abstract_syntax Normally the syntax_id of the autogenerated + * ndr_table_. + * + * @param[in] dispatch The corresponding autogenerated dispatch function + * rpc__dispatch. + * + * @param[in] serversupplied_info The server supplied authentication function. + * + * @param[out] presult A pointer to store the connected rpc client pipe. + * + * @return NT_STATUS_OK on success, a corresponding NT status if an + * error occured. + * + * @code + * struct rpc_pipe_client *winreg_pipe; + * NTSTATUS status; + * + * status = rpc_pipe_open_internal(tmp_ctx, + * &ndr_table_winreg.syntax_id, + * rpc_winreg_dispatch, + * p->server_info, + * &winreg_pipe); + * @endcode + */ +NTSTATUS rpc_pipe_open_internal(TALLOC_CTX *mem_ctx, + const struct ndr_syntax_id *abstract_syntax, + struct auth_serversupplied_info *serversupplied_info, + struct client_address *client_id, + struct messaging_context *msg_ctx, + struct rpc_pipe_client **presult) +{ + struct rpc_pipe_client *result; + NTSTATUS status; + + result = TALLOC_ZERO_P(mem_ctx, struct rpc_pipe_client); + if (result == NULL) { + return NT_STATUS_NO_MEMORY; + } + + result->abstract_syntax = *abstract_syntax; + result->transfer_syntax = ndr_transfer_syntax; + + if (client_id == NULL) { + static struct client_address unknown; + strlcpy(unknown.addr, "", sizeof(unknown.addr)); + unknown.name = ""; + client_id = &unknown; + } + + result->max_xmit_frag = -1; + result->max_recv_frag = -1; + + status = rpcint_binding_handle_ex(result, + abstract_syntax, + NULL, + client_id, + serversupplied_info, + msg_ctx, + &result->binding_handle); + if (!NT_STATUS_IS_OK(status)) { + TALLOC_FREE(result); + return status; + } + + *presult = result; + return NT_STATUS_OK; +} diff --git a/source3/rpc_server/rpc_ncacn_np_internal.c b/source3/rpc_server/rpc_ncacn_np_internal.c deleted file mode 100644 index 8253f65bb7..0000000000 --- a/source3/rpc_server/rpc_ncacn_np_internal.c +++ /dev/null @@ -1,577 +0,0 @@ -/* - * Unix SMB/CIFS implementation. - * RPC Pipe client / server routines - * Copyright (C) Andrew Tridgell 1992-1998, - * Largely re-written : 2005 - * Copyright (C) Jeremy Allison 1998 - 2005 - * - * 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 - * the Free Software Foundation; either version 3 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, see . - */ - -#include "includes.h" -#include "rpc_server/srv_pipe_internal.h" -#include "rpc_dce.h" - -#undef DBGC_CLASS -#define DBGC_CLASS DBGC_RPC_SRV - -static int pipes_open; - -static struct pipes_struct *InternalPipes; - -/* TODO - * the following prototypes are declared here to avoid - * code being moved about too much for a patch to be - * disrupted / less obvious. - * - * these functions, and associated functions that they - * call, should be moved behind a .so module-loading - * system _anyway_. so that's the next step... - */ - -/**************************************************************************** - Internal Pipe iterator functions. -****************************************************************************/ - -struct pipes_struct *get_first_internal_pipe(void) -{ - return InternalPipes; -} - -struct pipes_struct *get_next_internal_pipe(struct pipes_struct *p) -{ - return p->next; -} - -static void free_pipe_rpc_context_internal( PIPE_RPC_FNS *list ) -{ - PIPE_RPC_FNS *tmp = list; - PIPE_RPC_FNS *tmp2; - - while (tmp) { - tmp2 = tmp->next; - SAFE_FREE(tmp); - tmp = tmp2; - } - - return; -} - -bool check_open_pipes(void) -{ - struct pipes_struct *p; - - for (p = InternalPipes; p != NULL; p = p->next) { - if (num_pipe_handles(p) != 0) { - return true; - } - } - return false; -} - -/**************************************************************************** - Close an rpc pipe. -****************************************************************************/ - -int close_internal_rpc_pipe_hnd(struct pipes_struct *p) -{ - if (!p) { - DEBUG(0,("Invalid pipe in close_internal_rpc_pipe_hnd\n")); - return False; - } - - if (p->auth.auth_data_free_func) { - (*p->auth.auth_data_free_func)(&p->auth); - } - - free_pipe_rpc_context_internal( p->contexts ); - - /* Free the handles database. */ - close_policy_by_pipe(p); - - DLIST_REMOVE(InternalPipes, p); - - ZERO_STRUCTP(p); - - return 0; -} - -/**************************************************************************** - Make an internal namedpipes structure -****************************************************************************/ - -struct pipes_struct *make_internal_rpc_pipe_p(TALLOC_CTX *mem_ctx, - const struct ndr_syntax_id *syntax, - struct client_address *client_id, - const struct auth_serversupplied_info *server_info, - struct messaging_context *msg_ctx) -{ - struct pipes_struct *p; - - DEBUG(4,("Create pipe requested %s\n", - get_pipe_name_from_syntax(talloc_tos(), syntax))); - - p = TALLOC_ZERO_P(mem_ctx, struct pipes_struct); - - if (!p) { - DEBUG(0,("ERROR! no memory for pipes_struct!\n")); - return NULL; - } - - p->mem_ctx = talloc_named(p, 0, "pipe %s %p", - get_pipe_name_from_syntax(talloc_tos(), - syntax), p); - if (p->mem_ctx == NULL) { - DEBUG(0,("open_rpc_pipe_p: talloc_init failed.\n")); - TALLOC_FREE(p); - return NULL; - } - - if (!init_pipe_handles(p, syntax)) { - DEBUG(0,("open_rpc_pipe_p: init_pipe_handles failed.\n")); - TALLOC_FREE(p); - return NULL; - } - - p->server_info = copy_serverinfo(p, server_info); - if (p->server_info == NULL) { - DEBUG(0, ("open_rpc_pipe_p: copy_serverinfo failed\n")); - close_policy_by_pipe(p); - TALLOC_FREE(p); - return NULL; - } - - p->msg_ctx = msg_ctx; - - DLIST_ADD(InternalPipes, p); - - p->client_id = client_id; - - p->endian = RPC_LITTLE_ENDIAN; - - p->syntax = *syntax; - - DEBUG(4,("Created internal pipe %s (pipes_open=%d)\n", - get_pipe_name_from_syntax(talloc_tos(), syntax), pipes_open)); - - talloc_set_destructor(p, close_internal_rpc_pipe_hnd); - - return p; -} - -static NTSTATUS rpcint_dispatch(struct pipes_struct *p, - TALLOC_CTX *mem_ctx, - uint32_t opnum, - const DATA_BLOB *in_data, - DATA_BLOB *out_data) -{ - uint32_t num_cmds = rpc_srv_get_pipe_num_cmds(&p->syntax); - const struct api_struct *cmds = rpc_srv_get_pipe_cmds(&p->syntax); - uint32_t i; - bool ok; - - /* set opnum */ - p->opnum = opnum; - - for (i = 0; i < num_cmds; i++) { - if (cmds[i].opnum == opnum && cmds[i].fn != NULL) { - break; - } - } - - if (i == num_cmds) { - return NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE; - } - - p->in_data.data = *in_data; - p->out_data.rdata = data_blob_null; - - ok = cmds[i].fn(p); - p->in_data.data = data_blob_null; - if (!ok) { - data_blob_free(&p->out_data.rdata); - talloc_free_children(p->mem_ctx); - return NT_STATUS_RPC_CALL_FAILED; - } - - if (p->fault_state) { - p->fault_state = false; - data_blob_free(&p->out_data.rdata); - talloc_free_children(p->mem_ctx); - return NT_STATUS_RPC_CALL_FAILED; - } - - if (p->bad_handle_fault_state) { - p->bad_handle_fault_state = false; - data_blob_free(&p->out_data.rdata); - talloc_free_children(p->mem_ctx); - return NT_STATUS_RPC_SS_CONTEXT_MISMATCH; - } - - if (p->rng_fault_state) { - p->rng_fault_state = false; - data_blob_free(&p->out_data.rdata); - talloc_free_children(p->mem_ctx); - return NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE; - } - - *out_data = p->out_data.rdata; - talloc_steal(mem_ctx, out_data->data); - p->out_data.rdata = data_blob_null; - - talloc_free_children(p->mem_ctx); - return NT_STATUS_OK; -} - -struct rpcint_bh_state { - struct pipes_struct *p; -}; - -static bool rpcint_bh_is_connected(struct dcerpc_binding_handle *h) -{ - struct rpcint_bh_state *hs = dcerpc_binding_handle_data(h, - struct rpcint_bh_state); - - if (!hs->p) { - return false; - } - - return true; -} - -static uint32_t rpcint_bh_set_timeout(struct dcerpc_binding_handle *h, - uint32_t timeout) -{ - /* TODO: implement timeouts */ - return UINT32_MAX; -} - -struct rpcint_bh_raw_call_state { - DATA_BLOB in_data; - DATA_BLOB out_data; - uint32_t out_flags; -}; - -static struct tevent_req *rpcint_bh_raw_call_send(TALLOC_CTX *mem_ctx, - struct tevent_context *ev, - struct dcerpc_binding_handle *h, - const struct GUID *object, - uint32_t opnum, - uint32_t in_flags, - const uint8_t *in_data, - size_t in_length) -{ - struct rpcint_bh_state *hs = - dcerpc_binding_handle_data(h, - struct rpcint_bh_state); - struct tevent_req *req; - struct rpcint_bh_raw_call_state *state; - bool ok; - NTSTATUS status; - - req = tevent_req_create(mem_ctx, &state, - struct rpcint_bh_raw_call_state); - if (req == NULL) { - return NULL; - } - state->in_data.data = discard_const_p(uint8_t, in_data); - state->in_data.length = in_length; - - ok = rpcint_bh_is_connected(h); - if (!ok) { - tevent_req_nterror(req, NT_STATUS_INVALID_CONNECTION); - return tevent_req_post(req, ev); - } - - /* TODO: allow async */ - status = rpcint_dispatch(hs->p, state, opnum, - &state->in_data, - &state->out_data); - if (!NT_STATUS_IS_OK(status)) { - tevent_req_nterror(req, status); - return tevent_req_post(req, ev); - } - - tevent_req_done(req); - return tevent_req_post(req, ev); -} - -static NTSTATUS rpcint_bh_raw_call_recv(struct tevent_req *req, - TALLOC_CTX *mem_ctx, - uint8_t **out_data, - size_t *out_length, - uint32_t *out_flags) -{ - struct rpcint_bh_raw_call_state *state = - tevent_req_data(req, - struct rpcint_bh_raw_call_state); - NTSTATUS status; - - if (tevent_req_is_nterror(req, &status)) { - tevent_req_received(req); - return status; - } - - *out_data = talloc_move(mem_ctx, &state->out_data.data); - *out_length = state->out_data.length; - *out_flags = 0; - tevent_req_received(req); - return NT_STATUS_OK; -} - -struct rpcint_bh_disconnect_state { - uint8_t _dummy; -}; - -static struct tevent_req *rpcint_bh_disconnect_send(TALLOC_CTX *mem_ctx, - struct tevent_context *ev, - struct dcerpc_binding_handle *h) -{ - struct rpcint_bh_state *hs = dcerpc_binding_handle_data(h, - struct rpcint_bh_state); - struct tevent_req *req; - struct rpcint_bh_disconnect_state *state; - bool ok; - - req = tevent_req_create(mem_ctx, &state, - struct rpcint_bh_disconnect_state); - if (req == NULL) { - return NULL; - } - - ok = rpcint_bh_is_connected(h); - if (!ok) { - tevent_req_nterror(req, NT_STATUS_INVALID_CONNECTION); - return tevent_req_post(req, ev); - } - - /* - * TODO: do a real async disconnect ... - * - * For now the caller needs to free pipes_struct - */ - hs->p = NULL; - - tevent_req_done(req); - return tevent_req_post(req, ev); -} - -static NTSTATUS rpcint_bh_disconnect_recv(struct tevent_req *req) -{ - NTSTATUS status; - - if (tevent_req_is_nterror(req, &status)) { - tevent_req_received(req); - return status; - } - - tevent_req_received(req); - return NT_STATUS_OK; -} - -static bool rpcint_bh_ref_alloc(struct dcerpc_binding_handle *h) -{ - return true; -} - -static void rpcint_bh_do_ndr_print(struct dcerpc_binding_handle *h, - int ndr_flags, - const void *_struct_ptr, - const struct ndr_interface_call *call) -{ - void *struct_ptr = discard_const(_struct_ptr); - - if (DEBUGLEVEL < 10) { - return; - } - - if (ndr_flags & NDR_IN) { - ndr_print_function_debug(call->ndr_print, - call->name, - ndr_flags, - struct_ptr); - } - if (ndr_flags & NDR_OUT) { - ndr_print_function_debug(call->ndr_print, - call->name, - ndr_flags, - struct_ptr); - } -} - -static const struct dcerpc_binding_handle_ops rpcint_bh_ops = { - .name = "rpcint", - .is_connected = rpcint_bh_is_connected, - .set_timeout = rpcint_bh_set_timeout, - .raw_call_send = rpcint_bh_raw_call_send, - .raw_call_recv = rpcint_bh_raw_call_recv, - .disconnect_send = rpcint_bh_disconnect_send, - .disconnect_recv = rpcint_bh_disconnect_recv, - - .ref_alloc = rpcint_bh_ref_alloc, - .do_ndr_print = rpcint_bh_do_ndr_print, -}; - -static NTSTATUS rpcint_binding_handle_ex(TALLOC_CTX *mem_ctx, - const struct ndr_syntax_id *abstract_syntax, - const struct ndr_interface_table *ndr_table, - struct client_address *client_id, - const struct auth_serversupplied_info *server_info, - struct messaging_context *msg_ctx, - struct dcerpc_binding_handle **binding_handle) -{ - struct dcerpc_binding_handle *h; - struct rpcint_bh_state *hs; - - if (ndr_table) { - abstract_syntax = &ndr_table->syntax_id; - } - - h = dcerpc_binding_handle_create(mem_ctx, - &rpcint_bh_ops, - NULL, - ndr_table, - &hs, - struct rpcint_bh_state, - __location__); - if (h == NULL) { - return NT_STATUS_NO_MEMORY; - } - hs->p = make_internal_rpc_pipe_p(hs, - abstract_syntax, - client_id, - server_info, - msg_ctx); - if (hs->p == NULL) { - TALLOC_FREE(h); - return NT_STATUS_NO_MEMORY; - } - - *binding_handle = h; - return NT_STATUS_OK; -} -/** - * @brief Create a new DCERPC Binding Handle which uses a local dispatch function. - * - * @param[in] mem_ctx The memory context to use. - * - * @param[in] ndr_table Normally the ndr_table_. - * - * @param[in] client_id The info about the connected client. - * - * @param[in] serversupplied_info The server supplied authentication function. - * - * @param[in] msg_ctx The messaging context that can be used by the server - * - * @param[out] binding_handle A pointer to store the connected - * dcerpc_binding_handle - * - * @return NT_STATUS_OK on success, a corresponding NT status if an - * error occured. - * - * @code - * struct dcerpc_binding_handle *winreg_binding; - * NTSTATUS status; - * - * status = rpcint_binding_handle(tmp_ctx, - * &ndr_table_winreg, - * p->client_id, - * p->server_info, - * p->msg_ctx - * &winreg_binding); - * @endcode - */ -NTSTATUS rpcint_binding_handle(TALLOC_CTX *mem_ctx, - const struct ndr_interface_table *ndr_table, - struct client_address *client_id, - const struct auth_serversupplied_info *server_info, - struct messaging_context *msg_ctx, - struct dcerpc_binding_handle **binding_handle) -{ - return rpcint_binding_handle_ex(mem_ctx, NULL, ndr_table, client_id, - server_info, msg_ctx, binding_handle); -} - -/** - * @brief Create a new RPC client context which uses a local dispatch function. - * - * @param[in] mem_ctx The memory context to use. - * - * @param[in] abstract_syntax Normally the syntax_id of the autogenerated - * ndr_table_. - * - * @param[in] dispatch The corresponding autogenerated dispatch function - * rpc__dispatch. - * - * @param[in] serversupplied_info The server supplied authentication function. - * - * @param[out] presult A pointer to store the connected rpc client pipe. - * - * @return NT_STATUS_OK on success, a corresponding NT status if an - * error occured. - * - * @code - * struct rpc_pipe_client *winreg_pipe; - * NTSTATUS status; - * - * status = rpc_pipe_open_internal(tmp_ctx, - * &ndr_table_winreg.syntax_id, - * rpc_winreg_dispatch, - * p->server_info, - * &winreg_pipe); - * @endcode - */ -NTSTATUS rpc_pipe_open_internal(TALLOC_CTX *mem_ctx, - const struct ndr_syntax_id *abstract_syntax, - struct auth_serversupplied_info *serversupplied_info, - struct client_address *client_id, - struct messaging_context *msg_ctx, - struct rpc_pipe_client **presult) -{ - struct rpc_pipe_client *result; - NTSTATUS status; - - result = TALLOC_ZERO_P(mem_ctx, struct rpc_pipe_client); - if (result == NULL) { - return NT_STATUS_NO_MEMORY; - } - - result->abstract_syntax = *abstract_syntax; - result->transfer_syntax = ndr_transfer_syntax; - - if (client_id == NULL) { - static struct client_address unknown; - strlcpy(unknown.addr, "", sizeof(unknown.addr)); - unknown.name = ""; - client_id = &unknown; - } - - result->max_xmit_frag = -1; - result->max_recv_frag = -1; - - status = rpcint_binding_handle_ex(result, - abstract_syntax, - NULL, - client_id, - serversupplied_info, - msg_ctx, - &result->binding_handle); - if (!NT_STATUS_IS_OK(status)) { - TALLOC_FREE(result); - return status; - } - - *presult = result; - return NT_STATUS_OK; -} -- cgit