diff options
Diffstat (limited to 'source3/librpc/rpc/dcerpc_ep.c')
| -rw-r--r-- | source3/librpc/rpc/dcerpc_ep.c | 167 |
1 files changed, 167 insertions, 0 deletions
diff --git a/source3/librpc/rpc/dcerpc_ep.c b/source3/librpc/rpc/dcerpc_ep.c new file mode 100644 index 0000000000..bf193662aa --- /dev/null +++ b/source3/librpc/rpc/dcerpc_ep.c @@ -0,0 +1,167 @@ +/* + * Endpoint Mapper Functions + * DCERPC local endpoint mapper client routines + * Copyright (c) 2010 Andreas Schneider. + * + * 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 <http://www.gnu.org/licenses/>. + */ + +#include "includes.h" +#include "librpc/rpc/dcerpc.h" +#include "librpc/rpc/dcerpc_ep.h" +#include "librpc/gen_ndr/cli_epmapper.h" + +#define EPM_MAX_ANNOTATION_SIZE 64 + +static NTSTATUS ep_register(const struct ndr_interface_table *iface, + const struct dcerpc_binding_vector *bind_vec, + const struct GUID *object_guid, + const char *annotation, + uint32_t replace) +{ + struct dcerpc_binding_handle *h = NULL; + static struct client_address client_id; + struct epm_entry_t *entries; + uint32_t num_ents, i; + TALLOC_CTX *tmp_ctx; + NTSTATUS result = NT_STATUS_OK; + NTSTATUS status; + + if (iface == NULL) { + return NT_STATUS_INVALID_PARAMETER; + } + + if (bind_vec == NULL || bind_vec->count == 0) { + return NT_STATUS_INVALID_PARAMETER; + } + + tmp_ctx = talloc_stackframe(); + if (tmp_ctx == NULL) { + return NT_STATUS_NO_MEMORY; + } + +#if 0 + /* NOTE: Samba3 doesn't have a ncalrpc server component yet. As soon as + * this is supported, we should talk to the endpoint mapper over the + * local transport. + */ + + /* Connect to the endpoint mapper locally */ + ncalrpc_sock = talloc_asprintf(tmp_ctx, + "%s/%s", + get_dyn_NCALRPCDIR(), + "epmapper"); + if (ncalrpc_sock == NULL) { + status = NT_STATUS_NO_MEMORY; + goto done; + } + + status = rpc_pipe_open_ncalrpc(tmp_ctx, + ncalrpc_sock, + &ndr_table_epmapper.syntax_id, + &cli); + if (!NT_STATUS_IS_OK(status)) { + goto done; + } +#endif + + strlcpy(client_id.addr, "localhost", sizeof(client_id.addr)); + client_id.name = "localhost"; + + status = rpcint_binding_handle(tmp_ctx, + &ndr_table_epmapper, + &client_id, + get_server_info_system(), + server_messaging_context(), + &h); + if (!NT_STATUS_IS_OK(status)) { + DEBUG(0, ("dcerpc_ep_register: Could not connect to epmapper (%s)", + nt_errstr(status))); + goto done; + } + + num_ents = bind_vec->count; + entries = talloc_array(tmp_ctx, struct epm_entry_t, num_ents); + + for (i = 0; i < num_ents; i++) { + struct dcerpc_binding *map_binding = &bind_vec->bindings[i]; + struct epm_twr_t *map_tower; + + map_tower = talloc_zero(entries, struct epm_twr_t); + if (map_tower == NULL) { + goto done; + } + + status = dcerpc_binding_build_tower(entries, + map_binding, + &map_tower->tower); + if (!NT_STATUS_IS_OK(status)) { + goto done; + } + + entries[i].tower = map_tower; + entries[i].annotation = talloc_strndup(entries, annotation, + EPM_MAX_ANNOTATION_SIZE); + if (entries[i].annotation == NULL) { + status = NT_STATUS_NO_MEMORY; + goto done; + } + if (object_guid != NULL) { + entries[i].object = *object_guid; + } else { + entries[i].object = map_binding->object.uuid; + } + } + + status = dcerpc_epm_Insert(h, + tmp_ctx, + num_ents, + entries, + replace, + &result); + if (!NT_STATUS_IS_OK(status)) { + DEBUG(0, ("dcerpc_ep_register: Could not insert tower (%s)\n", + nt_errstr(status))); + goto done; + } + if (!NT_STATUS_IS_OK(result)) { + DEBUG(0, ("dcerpc_ep_register: Could not insert tower (%s)\n", + nt_errstr(result))); + status = result; + goto done; + } + +done: + talloc_free(tmp_ctx); + + return status; +} + +NTSTATUS dcerpc_ep_register(const struct ndr_interface_table *iface, + const struct dcerpc_binding_vector *bind_vec, + const struct GUID *object_guid, + const char *annotation) +{ + return ep_register(iface, bind_vec, object_guid, annotation, 1); +} + +NTSTATUS dcerpc_ep_register_noreplace(const struct ndr_interface_table *iface, + const struct dcerpc_binding_vector *bind_vec, + const struct GUID *object_guid, + const char *annotation) +{ + return ep_register(iface, bind_vec, object_guid, annotation, 0); +} + +/* vim: set ts=8 sw=8 noet cindent syntax=c.doxygen: */ |