diff options
Diffstat (limited to 'source4/winbind')
-rw-r--r-- | source4/winbind/wb_async_helpers.c | 323 | ||||
-rw-r--r-- | source4/winbind/wb_async_helpers.h | 9 | ||||
-rw-r--r-- | source4/winbind/wb_samba3_cmd.c | 67 |
3 files changed, 387 insertions, 12 deletions
diff --git a/source4/winbind/wb_async_helpers.c b/source4/winbind/wb_async_helpers.c index e03c6bccef..4899f232a2 100644 --- a/source4/winbind/wb_async_helpers.c +++ b/source4/winbind/wb_async_helpers.c @@ -241,6 +241,7 @@ struct composite_context *wb_get_schannel_creds_send(struct wb_get_schannel_cred result->private_data = state; state->io = io; + state->ctx = result; state->p = dcerpc_pipe_init(state, ev); if (state->p == NULL) goto failed; @@ -251,7 +252,6 @@ struct composite_context *wb_get_schannel_creds_send(struct wb_get_schannel_cred ctx->async.fn = get_schannel_creds_recv_pipe; ctx->async.private_data = state; - state->ctx = result; return result; failed: @@ -475,6 +475,7 @@ struct composite_context *wb_get_lsa_pipe_send(struct wb_get_lsa_pipe *io) result->private_data = state; state->io = io; + state->ctx = result; state->finddcs = talloc(state, struct wb_finddcs); if (state->finddcs == NULL) goto failed; @@ -487,7 +488,6 @@ struct composite_context *wb_get_lsa_pipe_send(struct wb_get_lsa_pipe *io) ctx->async.fn = get_lsa_pipe_recv_dcs; ctx->async.private_data = state; - state->ctx = result; return result; failed: @@ -729,10 +729,14 @@ NTSTATUS wb_get_lsa_pipe_recv(struct composite_context *c, struct get_lsa_pipe_state *state = talloc_get_type(c->private_data, struct get_lsa_pipe_state); - state->io->out.domain_sid = - talloc_steal(mem_ctx, state->queryinfo.out.info->domain.sid); - state->io->out.pipe = - talloc_steal(mem_ctx, state->lsa_pipe); + if (NT_STATUS_IS_OK(status)) { + state->io->out.domain_sid = + talloc_steal(mem_ctx, + state->queryinfo.out.info->domain.sid); + state->io->out.pipe = + talloc_steal(mem_ctx, + state->lsa_pipe); + } talloc_free(c); return status; } @@ -743,3 +747,310 @@ NTSTATUS wb_get_lsa_pipe(struct wb_get_lsa_pipe *io, struct composite_context *c = wb_get_lsa_pipe_send(io); return wb_get_lsa_pipe_recv(c, mem_ctx); } + +struct lsa_lookupnames_state { + struct composite_context *ctx; + uint32_t num_names; + struct lsa_LookupNames r; + struct lsa_TransSidArray sids; + uint32_t count; + struct wb_sid_object **result; +}; + +static void lsa_lookupnames_recv_sids(struct rpc_request *req); + +struct composite_context *wb_lsa_lookupnames_send(struct dcerpc_pipe *lsa_pipe, + struct policy_handle *handle, + int num_names, + const char **names) +{ + struct composite_context *result; + struct rpc_request *req; + struct lsa_lookupnames_state *state; + + struct lsa_String *lsa_names; + int i; + + result = talloc_zero(NULL, struct composite_context); + if (result == NULL) goto failed; + result->state = COMPOSITE_STATE_IN_PROGRESS; + result->event_ctx = lsa_pipe->conn->event_ctx; + + state = talloc(result, struct lsa_lookupnames_state); + if (state == NULL) goto failed; + result->private_data = state; + + state->sids.count = 0; + state->sids.sids = NULL; + state->num_names = num_names; + state->count = 0; + + lsa_names = talloc_array(state, struct lsa_String, num_names); + if (lsa_names == NULL) goto failed; + + for (i=0; i<num_names; i++) { + lsa_names[i].string = names[i]; + } + + state->r.in.handle = handle; + state->r.in.num_names = num_names; + state->r.in.names = lsa_names; + state->r.in.sids = &state->sids; + state->r.in.level = 1; + state->r.in.count = &state->count; + state->r.out.count = &state->count; + state->r.out.sids = &state->sids; + + req = dcerpc_lsa_LookupNames_send(lsa_pipe, state, &state->r); + if (req == NULL) goto failed; + + req->async.callback = lsa_lookupnames_recv_sids; + req->async.private = state; + state->ctx = result; + return result; + + failed: + talloc_free(result); + return NULL; +} + +static void lsa_lookupnames_recv_sids(struct rpc_request *req) +{ + struct lsa_lookupnames_state *state = + talloc_get_type(req->async.private, + struct lsa_lookupnames_state); + int i; + + state->ctx->status = dcerpc_ndr_request_recv(req); + if (!NT_STATUS_IS_OK(state->ctx->status)) goto done; + state->ctx->status = state->r.out.result; + if (!NT_STATUS_IS_OK(state->ctx->status) && + !NT_STATUS_EQUAL(state->ctx->status, STATUS_SOME_UNMAPPED)) { + goto done; + } + + state->result = talloc_array(state, struct wb_sid_object *, + state->num_names); + if (state->result == NULL) { + state->ctx->status = NT_STATUS_NO_MEMORY; + goto done; + } + + for (i=0; i<state->num_names; i++) { + struct lsa_TranslatedSid *sid = &state->r.out.sids->sids[i]; + struct lsa_TrustInformation *dom; + + state->result[i] = talloc_zero(state->result, + struct wb_sid_object); + if (state->result[i] == NULL) { + state->ctx->status = NT_STATUS_NO_MEMORY; + goto done; + } + + state->result[i]->type = sid->sid_type; + if (state->result[i]->type == SID_NAME_UNKNOWN) { + continue; + } + + if (sid->sid_index >= state->r.out.domains->count) { + state->ctx->status = NT_STATUS_INVALID_PARAMETER; + goto done; + } + + dom = &state->r.out.domains->domains[sid->sid_index]; + + state->result[i]->sid = dom_sid_add_rid(state->result[i], + dom->sid, sid->rid); + } + + state->ctx->state = COMPOSITE_STATE_DONE; + + done: + if (!NT_STATUS_IS_OK(state->ctx->status)) { + state->ctx->state = COMPOSITE_STATE_ERROR; + } + if ((state->ctx->state >= COMPOSITE_STATE_DONE) && + (state->ctx->async.fn != NULL)) { + state->ctx->async.fn(state->ctx); + } +} + +NTSTATUS wb_lsa_lookupnames_recv(struct composite_context *c, + TALLOC_CTX *mem_ctx, + struct wb_sid_object ***sids) +{ + NTSTATUS status = composite_wait(c); + if (NT_STATUS_IS_OK(status)) { + struct lsa_lookupnames_state *state = + talloc_get_type(c->private_data, + struct lsa_lookupnames_state); + *sids = talloc_steal(mem_ctx, state->result); + } + talloc_free(c); + return status; +} + +NTSTATUS wb_lsa_lookupnames(struct dcerpc_pipe *lsa_pipe, + struct policy_handle *handle, + int num_names, const char **names, + TALLOC_CTX *mem_ctx, + struct wb_sid_object ***sids) +{ + struct composite_context *c = + wb_lsa_lookupnames_send(lsa_pipe, handle, num_names, names); + return wb_lsa_lookupnames_recv(c, mem_ctx, sids); +} + +struct lsa_lookupname_state { + struct composite_context *ctx; + struct dcerpc_pipe *lsa_pipe; + const char *name; + struct wb_sid_object *sid; + + struct lsa_ObjectAttribute objectattr; + struct lsa_OpenPolicy2 openpolicy; + struct policy_handle policy_handle; + struct lsa_Close close; +}; + +static void lsa_lookupname_recv_open(struct rpc_request *req); +static void lsa_lookupname_recv_sids(struct composite_context *ctx); + +struct composite_context *wb_lsa_lookupname_send(struct dcerpc_pipe *lsa_pipe, + const char *name) +{ + struct composite_context *result; + struct rpc_request *req; + struct lsa_lookupname_state *state; + + result = talloc_zero(NULL, struct composite_context); + if (result == NULL) goto failed; + result->state = COMPOSITE_STATE_IN_PROGRESS; + result->event_ctx = lsa_pipe->conn->event_ctx; + + state = talloc(result, struct lsa_lookupname_state); + if (state == NULL) goto failed; + result->private_data = state; + + state->lsa_pipe = lsa_pipe; + state->name = talloc_strdup(state, name); + if (state->name == NULL) goto failed; + state->ctx = result; + + ZERO_STRUCT(state->openpolicy); + state->openpolicy.in.system_name = + talloc_asprintf(state, "\\\\%s", + dcerpc_server_name(state->lsa_pipe)); + ZERO_STRUCT(state->objectattr); + state->openpolicy.in.attr = &state->objectattr; + state->openpolicy.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED; + state->openpolicy.out.handle = &state->policy_handle; + + req = dcerpc_lsa_OpenPolicy2_send(state->lsa_pipe, state, + &state->openpolicy); + if (req == NULL) goto failed; + + req->async.callback = lsa_lookupname_recv_open; + req->async.private = state; + return result; + + failed: + talloc_free(result); + return NULL; +} + +static void lsa_lookupname_recv_open(struct rpc_request *req) +{ + struct lsa_lookupname_state *state = + talloc_get_type(req->async.private, + struct lsa_lookupname_state); + struct composite_context *ctx; + + state->ctx->status = dcerpc_ndr_request_recv(req); + if (!NT_STATUS_IS_OK(state->ctx->status)) goto done; + state->ctx->status = state->openpolicy.out.result; + if (!NT_STATUS_IS_OK(state->ctx->status)) goto done; + + ctx = wb_lsa_lookupnames_send(state->lsa_pipe, &state->policy_handle, + 1, &state->name); + if (ctx == NULL) { + state->ctx->status = NT_STATUS_NO_MEMORY; + goto done; + } + + ctx->async.fn = lsa_lookupname_recv_sids; + ctx->async.private_data = state; + return; + + done: + if (!NT_STATUS_IS_OK(state->ctx->status)) { + state->ctx->state = COMPOSITE_STATE_ERROR; + } + if ((state->ctx->state >= COMPOSITE_STATE_DONE) && + (state->ctx->async.fn != NULL)) { + state->ctx->async.fn(state->ctx); + } +} + +static void lsa_lookupname_recv_sids(struct composite_context *ctx) +{ + struct lsa_lookupname_state *state = + talloc_get_type(ctx->async.private_data, + struct lsa_lookupname_state); + struct rpc_request *req; + struct wb_sid_object **sids; + + state->ctx->status = wb_lsa_lookupnames_recv(ctx, state, &sids); + + if (NT_STATUS_IS_OK(state->ctx->status)) { + state->sid = NULL; + if (sids != NULL) { + state->sid = sids[0]; + } + } + + ZERO_STRUCT(state->close); + state->close.in.handle = &state->policy_handle; + state->close.out.handle = &state->policy_handle; + + req = dcerpc_lsa_Close_send(state->lsa_pipe, state, + &state->close); + if (req != NULL) { + req->async.callback = + (void(*)(struct rpc_request *))talloc_free; + } + + state->ctx->state = COMPOSITE_STATE_DONE; + + if (!NT_STATUS_IS_OK(state->ctx->status)) { + state->ctx->state = COMPOSITE_STATE_ERROR; + } + if ((state->ctx->state >= COMPOSITE_STATE_DONE) && + (state->ctx->async.fn != NULL)) { + state->ctx->async.fn(state->ctx); + } +} + +NTSTATUS wb_lsa_lookupname_recv(struct composite_context *c, + TALLOC_CTX *mem_ctx, + struct wb_sid_object **sid) +{ + NTSTATUS status = composite_wait(c); + if (NT_STATUS_IS_OK(status)) { + struct lsa_lookupname_state *state = + talloc_get_type(c->private_data, + struct lsa_lookupname_state); + *sid = talloc_steal(mem_ctx, state->sid); + } + talloc_free(c); + return status; +} + +NTSTATUS wb_lsa_lookupname(struct dcerpc_pipe *lsa_pipe, const char *name, + TALLOC_CTX *mem_ctx, struct wb_sid_object **sid) +{ + struct composite_context *c = + wb_lsa_lookupname_send(lsa_pipe, name); + return wb_lsa_lookupname_recv(c, mem_ctx, sid); +} + diff --git a/source4/winbind/wb_async_helpers.h b/source4/winbind/wb_async_helpers.h index e3de312590..473c2b772e 100644 --- a/source4/winbind/wb_async_helpers.h +++ b/source4/winbind/wb_async_helpers.h @@ -20,6 +20,8 @@ Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ +#include "librpc/gen_ndr/lsa.h" + struct wb_finddcs { struct { struct messaging_context *msg_ctx; @@ -56,3 +58,10 @@ struct wb_get_lsa_pipe { struct dcerpc_pipe *pipe; } out; }; + +struct wb_sid_object { + enum lsa_SidType type; + struct dom_sid *sid; + const char *domain; + const char *name; +}; diff --git a/source4/winbind/wb_samba3_cmd.c b/source4/winbind/wb_samba3_cmd.c index 60ad4eb164..c692adeaf1 100644 --- a/source4/winbind/wb_samba3_cmd.c +++ b/source4/winbind/wb_samba3_cmd.c @@ -307,7 +307,8 @@ struct lookupname_state { struct wb_get_lsa_pipe *getlsa; }; -static void lookupname_recv_lsa(struct composite_context *req); +static void lookupname_recv_lsa(struct composite_context *ctx); +static void lookupname_recv_sid(struct composite_context *ctx); NTSTATUS wbsrv_samba3_lookupname(struct wbsrv_samba3_call *s3call) { @@ -318,13 +319,20 @@ NTSTATUS wbsrv_samba3_lookupname(struct wbsrv_samba3_call *s3call) DEBUG(5, ("wbsrv_samba3_lookupname called\n")); - talloc_free(service->lsa_pipe); - service->lsa_pipe = NULL; - state = talloc(s3call, struct lookupname_state); NT_STATUS_HAVE_NO_MEMORY(state); - state->s3call = s3call; + + if (service->lsa_pipe != NULL) { + ctx = wb_lsa_lookupname_send(service->lsa_pipe, + s3call->request.data.name.name); + NT_STATUS_HAVE_NO_MEMORY(ctx); + ctx->async.fn = lookupname_recv_sid; + ctx->async.private_data = state; + s3call->call->flags |= WBSRV_CALL_FLAGS_REPLY_ASYNC; + return NT_STATUS_OK; + } + state->getlsa = talloc(s3call, struct wb_get_lsa_pipe); NT_STATUS_HAVE_NO_MEMORY(state->getlsa); @@ -356,7 +364,18 @@ static void lookupname_recv_lsa(struct composite_context *ctx) status = wb_get_lsa_pipe_recv(ctx, service); if (!NT_STATUS_IS_OK(status)) goto done; - service->lsa_pipe = state->getlsa->out.pipe; + service->lsa_pipe = talloc_steal(service, state->getlsa->out.pipe); + + ctx = wb_lsa_lookupname_send(service->lsa_pipe, + state->s3call->request.data.name.name); + if (ctx == NULL) { + status = NT_STATUS_NO_MEMORY; + goto done; + } + + ctx->async.fn = lookupname_recv_sid; + ctx->async.private_data = state; + return; done: if (!NT_STATUS_IS_OK(status)) { @@ -378,3 +397,39 @@ static void lookupname_recv_lsa(struct composite_context *ctx) } } + +static void lookupname_recv_sid(struct composite_context *ctx) +{ + struct lookupname_state *state = + talloc_get_type(ctx->async.private_data, + struct lookupname_state); + NTSTATUS status; + struct wb_sid_object *sid; + + status = wb_lsa_lookupname_recv(ctx, state, &sid); + if (!NT_STATUS_IS_OK(status)) goto done; + + state->s3call->response.result = WINBINDD_OK; + state->s3call->response.data.sid.type = sid->type; + WBSRV_SAMBA3_SET_STRING(state->s3call->response.data.sid.sid, + dom_sid_string(state, sid->sid)); + + done: + if (!NT_STATUS_IS_OK(status)) { + struct winbindd_response *resp = &state->s3call->response; + resp->result = WINBINDD_ERROR; + WBSRV_SAMBA3_SET_STRING(resp->data.auth.nt_status_string, + nt_errstr(status)); + WBSRV_SAMBA3_SET_STRING(resp->data.auth.error_string, + nt_errstr(status)); + resp->data.auth.pam_error = nt_status_to_pam(status); + + } + + status = wbsrv_send_reply(state->s3call->call); + if (!NT_STATUS_IS_OK(status)) { + wbsrv_terminate_connection(state->s3call->call->wbconn, + "wbsrv_queue_reply() failed"); + return; + } +} |