From 6ed0840c3807569530d531c51581db650bfbc22d Mon Sep 17 00:00:00 2001 From: Andrew Tridgell Date: Sat, 12 Nov 2005 02:12:51 +0000 Subject: r11693: added a full async composite function for SMB2 that does: - name resolution - socket connect - negprot - multi-stage session setup - tcon --- source/libcli/smb2/connect.c | 280 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 280 insertions(+) create mode 100644 source/libcli/smb2/connect.c (limited to 'source/libcli/smb2/connect.c') diff --git a/source/libcli/smb2/connect.c b/source/libcli/smb2/connect.c new file mode 100644 index 00000000000..18f28539ea0 --- /dev/null +++ b/source/libcli/smb2/connect.c @@ -0,0 +1,280 @@ +/* + Unix SMB/CIFS implementation. + + SMB2 composite connection setup + + Copyright (C) Andrew Tridgell 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/raw/libcliraw.h" +#include "libcli/smb2/smb2.h" +#include "libcli/smb2/smb2_calls.h" +#include "libcli/composite/composite.h" + +struct smb2_connect_state { + struct smb2_request *req; + struct composite_context *creq; + struct cli_credentials *credentials; + const char *host; + const char *share; + struct smb2_negprot negprot; + struct smb2_tree_connect tcon; + struct smb2_session *session; + struct smb2_tree *tree; +}; + +/* + continue after tcon reply +*/ +static void continue_tcon(struct smb2_request *req) +{ + struct composite_context *c = talloc_get_type(req->async.private, + struct composite_context); + struct smb2_connect_state *state = talloc_get_type(c->private_data, + struct smb2_connect_state); + + c->status = smb2_tree_connect_recv(req, &state->tcon); + if (!NT_STATUS_IS_OK(c->status)) { + composite_error(c, c->status); + return; + } + + state->tree->tid = state->tcon.out.tid; + + composite_done(c); +} + +/* + continue after a session setup +*/ +static void continue_session(struct composite_context *creq) +{ + struct composite_context *c = talloc_get_type(creq->async.private_data, + struct composite_context); + struct smb2_connect_state *state = talloc_get_type(c->private_data, + struct smb2_connect_state); + + c->status = smb2_session_setup_spnego_recv(creq); + if (!NT_STATUS_IS_OK(c->status)) { + composite_error(c, c->status); + return; + } + + state->tree = smb2_tree_init(state->session, state, True); + if (state->tree == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + state->tcon.in.unknown1 = 0x09; + state->tcon.in.path = talloc_asprintf(state, "\\\\%s\\%s", + state->host, state->share); + if (state->tcon.in.path == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + state->req = smb2_tree_connect_send(state->tree, &state->tcon); + if (state->req == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + state->req->async.fn = continue_tcon; + state->req->async.private = c; +} + +/* + continue after negprot reply +*/ +static void continue_negprot(struct smb2_request *req) +{ + struct composite_context *c = talloc_get_type(req->async.private, + struct composite_context); + struct smb2_connect_state *state = talloc_get_type(c->private_data, + struct smb2_connect_state); + struct smb2_transport *transport = req->transport; + + c->status = smb2_negprot_recv(req, c, &state->negprot); + if (!NT_STATUS_IS_OK(c->status)) { + composite_error(c, c->status); + return; + } + + state->session = smb2_session_init(transport, state, True); + if (state->session == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + state->creq = smb2_session_setup_spnego_send(state->session, state->credentials); + if (state->creq == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + state->creq->async.fn = continue_session; + state->creq->async.private_data = c; +} + +/* + continue after a socket connect completes +*/ +static void continue_socket(struct composite_context *creq) +{ + struct composite_context *c = talloc_get_type(creq->async.private_data, + struct composite_context); + struct smb2_connect_state *state = talloc_get_type(c->private_data, + struct smb2_connect_state); + struct smbcli_socket *sock; + struct smb2_transport *transport; + + c->status = smbcli_sock_connect_recv(creq, state, &sock); + if (!NT_STATUS_IS_OK(c->status)) { + composite_error(c, c->status); + return; + } + + transport = smb2_transport_init(sock, state); + if (transport == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + ZERO_STRUCT(state->negprot); + state->negprot.in.unknown1 = 0x010024; + + state->req = smb2_negprot_send(transport, &state->negprot); + if (state->req == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + state->req->async.fn = continue_negprot; + state->req->async.private = c; +} + + +/* + continue after a resolve finishes +*/ +static void continue_resolve(struct composite_context *creq) +{ + struct composite_context *c = talloc_get_type(creq->async.private_data, + struct composite_context); + struct smb2_connect_state *state = talloc_get_type(c->private_data, + struct smb2_connect_state); + const char *addr; + + c->status = resolve_name_recv(creq, state, &addr); + if (!NT_STATUS_IS_OK(c->status)) { + composite_error(c, c->status); + return; + } + + state->creq = smbcli_sock_connect_send(state, addr, 445, state->host, c->event_ctx); + if (state->creq == NULL) { + composite_error(c, NT_STATUS_NO_MEMORY); + return; + } + + state->creq->async.private_data = c; + state->creq->async.fn = continue_socket; +} + +/* + a composite function that does a full negprot/sesssetup/tcon, returning + a connected smb2_tree + */ +struct composite_context *smb2_connect_send(TALLOC_CTX *mem_ctx, + const char *host, + const char *share, + struct cli_credentials *credentials, + struct event_context *ev) +{ + struct composite_context *c; + struct smb2_connect_state *state; + struct nbt_name name; + + c = talloc_zero(mem_ctx, struct composite_context); + if (c == NULL) return NULL; + + state = talloc(c, struct smb2_connect_state); + if (state == NULL) { + c->status = NT_STATUS_NO_MEMORY; + goto failed; + } + + c->state = COMPOSITE_STATE_IN_PROGRESS; + c->private_data = state; + c->event_ctx = ev; + + state->credentials = credentials; + state->host = talloc_strdup(c, host); + state->share = talloc_strdup(c, share); + if (state->host == NULL || state->share == NULL) { + c->status = NT_STATUS_NO_MEMORY; + goto failed; + } + + ZERO_STRUCT(name); + name.name = host; + + state->creq = resolve_name_send(&name, c->event_ctx, lp_name_resolve_order()); + if (state->creq == NULL) goto failed; + + state->creq->async.private_data = c; + state->creq->async.fn = continue_resolve; + + return c; + +failed: + composite_trigger_error(c); + return c; +} + +/* + receive a connect reply +*/ +NTSTATUS smb2_connect_recv(struct composite_context *c, TALLOC_CTX *mem_ctx, + struct smb2_tree **tree) +{ + NTSTATUS status; + struct smb2_connect_state *state = talloc_get_type(c->private_data, + struct smb2_connect_state); + status = composite_wait(c); + if (NT_STATUS_IS_OK(status)) { + *tree = talloc_steal(mem_ctx, state->tree); + } + talloc_free(c); + return status; +} + +/* + sync version of smb2_connect +*/ +NTSTATUS smb2_connect(TALLOC_CTX *mem_ctx, + const char *host, const char *share, + struct cli_credentials *credentials, + struct smb2_tree **tree, + struct event_context *ev) +{ + struct composite_context *c = smb2_connect_send(mem_ctx, host, share, + credentials, ev); + return smb2_connect_recv(c, mem_ctx, tree); +} -- cgit