diff options
Diffstat (limited to 'source4/winbind/wb_connect_sam.c')
-rw-r--r-- | source4/winbind/wb_connect_sam.c | 210 |
1 files changed, 210 insertions, 0 deletions
diff --git a/source4/winbind/wb_connect_sam.c b/source4/winbind/wb_connect_sam.c new file mode 100644 index 0000000000..8693356c7a --- /dev/null +++ b/source4/winbind/wb_connect_sam.c @@ -0,0 +1,210 @@ +/* + Unix SMB/CIFS implementation. + + Connect to the SAMR pipe, given an smbcli_tree and possibly some + credentials. Try ntlmssp, schannel and anon in that order. + + Copyright (C) Volker Lendecke 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 2 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, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. +*/ + +#include "includes.h" +#include "libcli/composite/composite.h" + +#include "libcli/raw/libcliraw.h" +#include "librpc/gen_ndr/ndr_samr.h" + + +/* Helper to initialize SAMR with a specific auth methods. Verify by opening + * the SAM handle */ + +struct connect_samr_state { + struct composite_context *ctx; + uint8_t auth_type; + struct cli_credentials *creds; + struct dom_sid *sid; + + struct dcerpc_pipe *samr_pipe; + struct policy_handle *connect_handle; + struct policy_handle *domain_handle; + + struct samr_Connect2 c; + struct samr_OpenDomain o; +}; + +static void connect_samr_recv_pipe(struct composite_context *ctx); +static void connect_samr_recv_conn(struct rpc_request *req); +static void connect_samr_recv_open(struct rpc_request *req); + +struct composite_context *wb_connect_sam_send(struct smbcli_tree *tree, + uint8_t auth_type, + struct cli_credentials *creds, + const struct dom_sid *domain_sid) +{ + struct composite_context *result, *ctx; + struct connect_samr_state *state; + + result = talloc(NULL, struct composite_context); + if (result == NULL) goto failed; + result->state = COMPOSITE_STATE_IN_PROGRESS; + result->async.fn = NULL; + result->event_ctx = tree->session->transport->socket->event.ctx; + + state = talloc(result, struct connect_samr_state); + if (state == NULL) goto failed; + state->ctx = result; + result->private_data = state; + + state->auth_type = auth_type; + state->creds = creds; + state->sid = dom_sid_dup(state, domain_sid); + if (state->sid == NULL) goto failed; + + state->samr_pipe = dcerpc_pipe_init(state, result->event_ctx); + if (state->samr_pipe == NULL) goto failed; + + ctx = dcerpc_pipe_open_smb_send(state->samr_pipe->conn, tree, + "\\samr"); + ctx->async.fn = connect_samr_recv_pipe; + ctx->async.private_data = state; + return result; + + failed: + talloc_free(result); + return NULL; +} + +static void connect_samr_recv_pipe(struct composite_context *ctx) +{ + struct connect_samr_state *state = + talloc_get_type(ctx->async.private_data, + struct connect_samr_state); + struct rpc_request *req; + + state->ctx->status = dcerpc_pipe_open_smb_recv(ctx); + if (!composite_is_ok(state->ctx)) return; + + switch (state->auth_type) { + case DCERPC_AUTH_TYPE_NONE: + state->ctx->status = + dcerpc_bind_auth_none(state->samr_pipe, + DCERPC_SAMR_UUID, + DCERPC_SAMR_VERSION); + break; + case DCERPC_AUTH_TYPE_NTLMSSP: + case DCERPC_AUTH_TYPE_SCHANNEL: + if (state->creds == NULL) { + composite_error(state->ctx, NT_STATUS_INTERNAL_ERROR); + return; + } + state->samr_pipe->conn->flags |= (DCERPC_SIGN | DCERPC_SEAL); + state->ctx->status = + dcerpc_bind_auth_password(state->samr_pipe, + DCERPC_SAMR_UUID, + DCERPC_SAMR_VERSION, + state->creds, + state->auth_type, + NULL); + break; + default: + state->ctx->status = NT_STATUS_INTERNAL_ERROR; + + } + + state->connect_handle = talloc(state, struct policy_handle); + if (composite_nomem(state->connect_handle, state->ctx)) return; + + state->c.in.system_name = + talloc_asprintf(state, "\\\\%s", + dcerpc_server_name(state->samr_pipe)); + state->c.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED; + state->c.out.connect_handle = state->connect_handle; + + req = dcerpc_samr_Connect2_send(state->samr_pipe, state, &state->c); + composite_continue_rpc(state->ctx, req, connect_samr_recv_conn, state); +} + +static void connect_samr_recv_conn(struct rpc_request *req) +{ + struct connect_samr_state *state = + talloc_get_type(req->async.private, + struct connect_samr_state); + + state->ctx->status = dcerpc_ndr_request_recv(req); + if (!composite_is_ok(state->ctx)) return; + state->ctx->status = state->c.out.result; + if (!composite_is_ok(state->ctx)) return; + + state->domain_handle = talloc(state, struct policy_handle); + if (composite_nomem(state->domain_handle, state->ctx)) return; + + state->o.in.connect_handle = state->connect_handle; + state->o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED; + state->o.in.sid = state->sid; + state->o.out.domain_handle = state->domain_handle; + + req = dcerpc_samr_OpenDomain_send(state->samr_pipe, state, &state->o); + composite_continue_rpc(state->ctx, req, + connect_samr_recv_open, state); +} + +static void connect_samr_recv_open(struct rpc_request *req) +{ + struct connect_samr_state *state = + talloc_get_type(req->async.private, + struct connect_samr_state); + + state->ctx->status = dcerpc_ndr_request_recv(req); + if (!composite_is_ok(state->ctx)) return; + state->ctx->status = state->o.out.result; + if (!composite_is_ok(state->ctx)) return; + + composite_done(state->ctx); +} + +NTSTATUS wb_connect_sam_recv(struct composite_context *c, + TALLOC_CTX *mem_ctx, + struct dcerpc_pipe **samr_pipe, + struct policy_handle **connect_handle, + struct policy_handle **domain_handle) +{ + NTSTATUS status = composite_wait(c); + if (NT_STATUS_IS_OK(status)) { + struct connect_samr_state *state = + talloc_get_type(c->private_data, + struct connect_samr_state); + *samr_pipe = talloc_steal(mem_ctx, state->samr_pipe); + *connect_handle = talloc_steal(mem_ctx, state->connect_handle); + *domain_handle = talloc_steal(mem_ctx, state->domain_handle); + } + talloc_free(c); + return status; +} + +NTSTATUS wb_connect_sam(struct smbcli_tree *tree, + uint8_t auth_type, + struct cli_credentials *creds, + const struct dom_sid *domain_sid, + TALLOC_CTX *mem_ctx, + struct dcerpc_pipe **samr_pipe, + struct policy_handle **connect_handle, + struct policy_handle **domain_handle) +{ + struct composite_context *c = + wb_connect_sam_send(tree, auth_type, creds, domain_sid); + return wb_connect_sam_recv(c, mem_ctx, samr_pipe, connect_handle, + domain_handle); +} |