From 4030bc9b2df3f2a1c18b373a94c53e48ba80cfc1 Mon Sep 17 00:00:00 2001 From: Stefan Metzmacher Date: Thu, 5 Aug 2010 18:14:04 +0200 Subject: librpc/rpc: add dcerpc_binding_handle abstration metze --- librpc/rpc/binding_handle.c | 470 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 470 insertions(+) create mode 100644 librpc/rpc/binding_handle.c (limited to 'librpc') diff --git a/librpc/rpc/binding_handle.c b/librpc/rpc/binding_handle.c new file mode 100644 index 0000000000..161e7d7842 --- /dev/null +++ b/librpc/rpc/binding_handle.c @@ -0,0 +1,470 @@ +/* + Unix SMB/CIFS implementation. + + dcerpc binding handle functions + + Copyright (C) Stefan Metzmacher 2010 + + 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 +#include "../lib/util/tevent_ntstatus.h" +#include "librpc/rpc/dcerpc.h" + +static int dcerpc_binding_handle_destructor(struct dcerpc_binding_handle *b) +{ + return 0; +} + +struct dcerpc_binding_handle *_dcerpc_binding_handle_create(TALLOC_CTX *mem_ctx, + const struct dcerpc_binding_handle_ops *ops, + const struct GUID *object, + const struct ndr_interface_table *table, + void *pstate, + size_t psize, + const char *type, + const char *location) +{ + struct dcerpc_binding_handle *h; + void **ppstate = (void **)pstate; + void *state; + + h = talloc_zero(mem_ctx, struct dcerpc_binding_handle); + if (h == NULL) { + return NULL; + } + h->ops = ops; + h->location = location; + h->object = object; + h->table = table; + + state = talloc_zero_size(h, psize); + if (state == NULL) { + talloc_free(h); + return NULL; + } + talloc_set_name_const(state, type); + + h->private_data = state; + + talloc_set_destructor(h, dcerpc_binding_handle_destructor); + + *ppstate = state; + return h; +} + +void *_dcerpc_binding_handle_data(struct dcerpc_binding_handle *h) +{ + return h->private_data; +} + +void dcerpc_binding_handle_set_sync_ev(struct dcerpc_binding_handle *h, + struct tevent_context *ev) +{ + h->sync_ev = ev; +} + +bool dcerpc_binding_handle_is_connected(struct dcerpc_binding_handle *h) +{ + return h->ops->is_connected(h); +} + +struct dcerpc_binding_handle_raw_call_state { + const struct dcerpc_binding_handle_ops *ops; + uint8_t *out_data; + size_t out_length; + uint32_t out_flags; +}; + +static void dcerpc_binding_handle_raw_call_done(struct tevent_req *subreq); + +struct tevent_req *dcerpc_binding_handle_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 tevent_req *req; + struct dcerpc_binding_handle_raw_call_state *state; + struct tevent_req *subreq; + + req = tevent_req_create(mem_ctx, &state, + struct dcerpc_binding_handle_raw_call_state); + if (req == NULL) { + return NULL; + } + state->ops = h->ops; + state->out_data = NULL; + state->out_length = 0; + state->out_flags = 0; + + subreq = state->ops->raw_call_send(state, ev, h, + object, opnum, + in_flags, in_data, in_length); + if (tevent_req_nomem(subreq, req)) { + return tevent_req_post(req, ev); + } + tevent_req_set_callback(subreq, dcerpc_binding_handle_raw_call_done, req); + + return req; +} + +static void dcerpc_binding_handle_raw_call_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data(subreq, + struct tevent_req); + struct dcerpc_binding_handle_raw_call_state *state = + tevent_req_data(req, + struct dcerpc_binding_handle_raw_call_state); + NTSTATUS error; + + error = state->ops->raw_call_recv(subreq, state, + &state->out_data, + &state->out_length, + &state->out_flags); + TALLOC_FREE(subreq); + if (!NT_STATUS_IS_OK(error)) { + tevent_req_nterror(req, error); + return; + } + + tevent_req_done(req); +} + +NTSTATUS dcerpc_binding_handle_raw_call_recv(struct tevent_req *req, + TALLOC_CTX *mem_ctx, + uint8_t **out_data, + size_t *out_length, + uint32_t *out_flags) +{ + struct dcerpc_binding_handle_raw_call_state *state = + tevent_req_data(req, + struct dcerpc_binding_handle_raw_call_state); + NTSTATUS error; + + if (tevent_req_is_nterror(req, &error)) { + tevent_req_received(req); + return error; + } + + *out_data = talloc_move(mem_ctx, &state->out_data); + *out_length = state->out_length; + *out_flags = state->out_flags; + tevent_req_received(req); + return NT_STATUS_OK; +} + +struct dcerpc_binding_handle_disconnect_state { + const struct dcerpc_binding_handle_ops *ops; +}; + +static void dcerpc_binding_handle_disconnect_done(struct tevent_req *subreq); + +struct tevent_req *dcerpc_binding_handle_disconnect_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct dcerpc_binding_handle *h) +{ + struct tevent_req *req; + struct dcerpc_binding_handle_disconnect_state *state; + struct tevent_req *subreq; + + req = tevent_req_create(mem_ctx, &state, + struct dcerpc_binding_handle_disconnect_state); + if (req == NULL) { + return NULL; + } + + state->ops = h->ops; + + subreq = state->ops->disconnect_send(state, ev, h); + if (tevent_req_nomem(subreq, req)) { + return tevent_req_post(req, ev); + } + tevent_req_set_callback(subreq, dcerpc_binding_handle_disconnect_done, req); + + return req; +} + +static void dcerpc_binding_handle_disconnect_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data(subreq, + struct tevent_req); + struct dcerpc_binding_handle_disconnect_state *state = + tevent_req_data(req, + struct dcerpc_binding_handle_disconnect_state); + NTSTATUS error; + + error = state->ops->disconnect_recv(subreq); + TALLOC_FREE(subreq); + if (!NT_STATUS_IS_OK(error)) { + tevent_req_nterror(req, error); + return; + } + + tevent_req_done(req); +} + +NTSTATUS dcerpc_binding_handle_disconnect_recv(struct tevent_req *req) +{ + NTSTATUS error; + + if (tevent_req_is_nterror(req, &error)) { + tevent_req_received(req); + return error; + } + + tevent_req_received(req); + return NT_STATUS_OK; +} + +struct dcerpc_binding_handle_call_state { + struct dcerpc_binding_handle *h; + const struct ndr_interface_call *call; + TALLOC_CTX *r_mem; + void *r_ptr; + struct ndr_push *push; + DATA_BLOB request; + DATA_BLOB response; + struct ndr_pull *pull; +}; + +static void dcerpc_binding_handle_call_done(struct tevent_req *subreq); + +struct tevent_req *dcerpc_binding_handle_call_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev, + struct dcerpc_binding_handle *h, + const struct GUID *object, + const struct ndr_interface_table *table, + uint32_t opnum, + TALLOC_CTX *r_mem, + void *r_ptr) +{ + struct tevent_req *req; + struct dcerpc_binding_handle_call_state *state; + struct tevent_req *subreq; + enum ndr_err_code ndr_err; + + req = tevent_req_create(mem_ctx, &state, + struct dcerpc_binding_handle_call_state); + if (req == NULL) { + return NULL; + } + +#if 0 /* TODO: activate this when the callers are fixed */ + if (table != h->table) { + tevent_req_nterror(req, NT_STATUS_INVALID_HANDLE); + return tevent_req_post(req, ev); + } +#endif + + if (opnum >= table->num_calls) { + tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR); + return tevent_req_post(req, ev); + } + + state->h = h; + state->call = &table->calls[opnum]; + + state->r_mem = r_mem; + state->r_ptr = r_ptr; + + /* setup for a ndr_push_* call */ + state->push = ndr_push_init_ctx(state); + if (tevent_req_nomem(state->push, req)) { + return tevent_req_post(req, ev); + } + + if (h->ops->ref_alloc && h->ops->ref_alloc(h)) { + state->push->flags |= LIBNDR_FLAG_REF_ALLOC; + } + + if (h->ops->push_bigendian && h->ops->push_bigendian(h)) { + state->push->flags |= LIBNDR_FLAG_BIGENDIAN; + } + + if (h->ops->use_ndr64 && h->ops->use_ndr64(h)) { + state->push->flags |= LIBNDR_FLAG_NDR64; + } + + /* push the structure into a blob */ + ndr_err = state->call->ndr_push(state->push, NDR_IN, state->r_ptr); + if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) { + NTSTATUS error; + error = ndr_map_error2ntstatus(ndr_err); + if (h->ops->ndr_push_failed) { + h->ops->ndr_push_failed(h, error, + state->r_ptr, + state->call); + } + tevent_req_nterror(req, error); + return tevent_req_post(req, ev); + } + + /* retrieve the blob */ + state->request = ndr_push_blob(state->push); + + if (h->ops->ndr_validate_in) { + NTSTATUS error; + error = h->ops->ndr_validate_in(h, state, + &state->request, + state->call); + if (!NT_STATUS_IS_OK(error)) { + tevent_req_nterror(req, error); + return tevent_req_post(req, ev); + } + } + + subreq = dcerpc_binding_handle_raw_call_send(state, ev, + h, object, opnum, + state->push->flags, + state->request.data, + state->request.length); + if (tevent_req_nomem(subreq, req)) { + return tevent_req_post(req, ev); + } + tevent_req_set_callback(subreq, dcerpc_binding_handle_call_done, req); + + return req; +} + +static void dcerpc_binding_handle_call_done(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data(subreq, + struct tevent_req); + struct dcerpc_binding_handle_call_state *state = + tevent_req_data(req, + struct dcerpc_binding_handle_call_state); + struct dcerpc_binding_handle *h = state->h; + NTSTATUS error; + uint32_t out_flags = 0; + enum ndr_err_code ndr_err; + + error = dcerpc_binding_handle_raw_call_recv(subreq, state, + &state->response.data, + &state->response.length, + &out_flags); + TALLOC_FREE(subreq); + if (!NT_STATUS_IS_OK(error)) { + tevent_req_nterror(req, error); + return; + } + + state->pull = ndr_pull_init_blob(&state->response, state); + if (tevent_req_nomem(state->pull, req)) { + return; + } + state->pull->flags = state->push->flags; + + if (out_flags & LIBNDR_FLAG_BIGENDIAN) { + state->pull->flags |= LIBNDR_FLAG_BIGENDIAN; + } else { + state->pull->flags &= ~LIBNDR_FLAG_BIGENDIAN; + } + + state->pull->current_mem_ctx = state->r_mem; + + /* pull the structure from the blob */ + ndr_err = state->call->ndr_pull(state->pull, NDR_OUT, state->r_ptr); + if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) { + error = ndr_map_error2ntstatus(ndr_err); + if (h->ops->ndr_pull_failed) { + h->ops->ndr_pull_failed(h, error, + &state->response, + state->call); + } + tevent_req_nterror(req, error); + return; + } + + if (h->ops->ndr_validate_out) { + error = h->ops->ndr_validate_out(h, + state->pull, + state->r_ptr, + state->call); + if (!NT_STATUS_IS_OK(error)) { + tevent_req_nterror(req, error); + return; + } + } + + tevent_req_done(req); +} + +NTSTATUS dcerpc_binding_handle_call_recv(struct tevent_req *req) +{ + NTSTATUS error; + + if (tevent_req_is_nterror(req, &error)) { + tevent_req_received(req); + return error; + } + + tevent_req_received(req); + return NT_STATUS_OK; +} + +NTSTATUS dcerpc_binding_handle_call(struct dcerpc_binding_handle *h, + const struct GUID *object, + const struct ndr_interface_table *table, + uint32_t opnum, + TALLOC_CTX *r_mem, + void *r_ptr) +{ + TALLOC_CTX *frame = talloc_stackframe(); + struct tevent_context *ev; + struct tevent_req *subreq; + NTSTATUS status; + + /* + * TODO: allow only one sync call + */ + + if (h->sync_ev) { + ev = h->sync_ev; + } else { + ev = tevent_context_init(frame); + } + if (ev == NULL) { + talloc_free(frame); + return NT_STATUS_NO_MEMORY; + } + + subreq = dcerpc_binding_handle_call_send(frame, ev, + h, object, table, + opnum, r_mem, r_ptr); + if (subreq == NULL) { + talloc_free(frame); + return NT_STATUS_NO_MEMORY; + } + + if (!tevent_req_poll(subreq, ev)) { + status = map_nt_error_from_unix(errno); + talloc_free(frame); + return status; + } + + status = dcerpc_binding_handle_call_recv(subreq); + if (!NT_STATUS_IS_OK(status)) { + talloc_free(frame); + return status; + } + + TALLOC_FREE(frame); + return NT_STATUS_OK; +} -- cgit