2 Unix SMB/CIFS implementation.
4 SMB2 composite connection setup
6 Copyright (C) Andrew Tridgell 2005
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #include "lib/util/tevent_ntstatus.h"
25 #include "libcli/raw/libcliraw.h"
26 #include "libcli/raw/raw_proto.h"
27 #include "libcli/smb2/smb2.h"
28 #include "libcli/smb2/smb2_calls.h"
29 #include "libcli/composite/composite.h"
30 #include "libcli/resolve/resolve.h"
31 #include "param/param.h"
32 #include "auth/credentials/credentials.h"
33 #include "../libcli/smb/smbXcli_base.h"
35 struct smb2_connect_state
{
36 struct tevent_context
*ev
;
37 struct cli_credentials
*credentials
;
38 uint64_t previous_session_id
;
39 struct resolve_context
*resolve_ctx
;
44 const char *socket_options
;
45 struct nbt_name calling
, called
;
46 struct gensec_settings
*gensec_settings
;
47 struct smbcli_options options
;
48 struct smb2_transport
*transport
;
49 struct smb2_session
*session
;
50 struct smb2_tree
*tree
;
53 static void smb2_connect_socket_done(struct composite_context
*creq
);
56 a composite function that does a full negprot/sesssetup/tcon, returning
59 struct tevent_req
*smb2_connect_send(TALLOC_CTX
*mem_ctx
,
60 struct tevent_context
*ev
,
64 struct resolve_context
*resolve_ctx
,
65 struct cli_credentials
*credentials
,
66 uint64_t previous_session_id
,
67 const struct smbcli_options
*options
,
68 const char *socket_options
,
69 struct gensec_settings
*gensec_settings
)
71 struct tevent_req
*req
;
72 struct smb2_connect_state
*state
;
73 struct composite_context
*creq
;
74 static const char *default_ports
[] = { "445", "139", NULL
};
76 req
= tevent_req_create(mem_ctx
, &state
,
77 struct smb2_connect_state
);
83 state
->credentials
= credentials
;
84 state
->previous_session_id
= previous_session_id
;
85 state
->options
= *options
;
89 state
->resolve_ctx
= resolve_ctx
;
90 state
->socket_options
= socket_options
;
91 state
->gensec_settings
= gensec_settings
;
93 if (state
->ports
== NULL
) {
94 state
->ports
= default_ports
;
97 make_nbt_name_client(&state
->calling
,
98 cli_credentials_get_workstation(credentials
));
100 nbt_choose_called_name(state
, &state
->called
,
101 host
, NBT_NAME_SERVER
);
103 state
->unc
= talloc_asprintf(state
, "\\\\%s\\%s",
104 state
->host
, state
->share
);
105 if (tevent_req_nomem(state
->unc
, req
)) {
106 return tevent_req_post(req
, ev
);
109 creq
= smbcli_sock_connect_send(state
, NULL
, state
->ports
,
110 state
->host
, state
->resolve_ctx
,
111 state
->ev
, state
->socket_options
,
114 if (tevent_req_nomem(creq
, req
)) {
115 return tevent_req_post(req
, ev
);
117 creq
->async
.fn
= smb2_connect_socket_done
;
118 creq
->async
.private_data
= req
;
123 static void smb2_connect_negprot_done(struct tevent_req
*subreq
);
125 static void smb2_connect_socket_done(struct composite_context
*creq
)
127 struct tevent_req
*req
=
128 talloc_get_type_abort(creq
->async
.private_data
,
130 struct smb2_connect_state
*state
=
132 struct smb2_connect_state
);
133 struct smbcli_socket
*sock
;
134 struct tevent_req
*subreq
;
136 uint32_t timeout_msec
;
137 enum protocol_types min_protocol
;
139 status
= smbcli_sock_connect_recv(creq
, state
, &sock
);
140 if (tevent_req_nterror(req
, status
)) {
144 state
->transport
= smb2_transport_init(sock
, state
, &state
->options
);
145 if (tevent_req_nomem(state
->transport
, req
)) {
149 timeout_msec
= state
->transport
->options
.request_timeout
* 1000;
150 min_protocol
= state
->transport
->options
.min_protocol
;
151 if (min_protocol
< PROTOCOL_SMB2_02
) {
152 min_protocol
= PROTOCOL_SMB2_02
;
155 subreq
= smbXcli_negprot_send(state
, state
->ev
,
156 state
->transport
->conn
, timeout_msec
,
158 state
->transport
->options
.max_protocol
,
159 state
->transport
->options
.max_credits
);
160 if (tevent_req_nomem(subreq
, req
)) {
163 tevent_req_set_callback(subreq
, smb2_connect_negprot_done
, req
);
166 static void smb2_connect_session_done(struct tevent_req
*subreq
);
168 static void smb2_connect_negprot_done(struct tevent_req
*subreq
)
170 struct tevent_req
*req
=
171 tevent_req_callback_data(subreq
,
173 struct smb2_connect_state
*state
=
175 struct smb2_connect_state
);
176 struct smb2_transport
*transport
= state
->transport
;
179 status
= smbXcli_negprot_recv(subreq
);
181 if (tevent_req_nterror(req
, status
)) {
185 state
->session
= smb2_session_init(transport
, state
->gensec_settings
, state
);
186 if (tevent_req_nomem(state
->session
, req
)) {
190 subreq
= smb2_session_setup_spnego_send(state
, state
->ev
,
193 state
->previous_session_id
);
194 if (tevent_req_nomem(subreq
, req
)) {
197 tevent_req_set_callback(subreq
, smb2_connect_session_done
, req
);
200 static void smb2_connect_tcon_done(struct tevent_req
*subreq
);
202 static void smb2_connect_session_done(struct tevent_req
*subreq
)
204 struct tevent_req
*req
=
205 tevent_req_callback_data(subreq
,
207 struct smb2_connect_state
*state
=
209 struct smb2_connect_state
);
211 uint32_t timeout_msec
;
213 status
= smb2_session_setup_spnego_recv(subreq
);
215 if (tevent_req_nterror(req
, status
)) {
219 state
->tree
= smb2_tree_init(state
->session
, state
, true);
220 if (tevent_req_nomem(state
->tree
, req
)) {
224 timeout_msec
= state
->transport
->options
.request_timeout
* 1000;
226 subreq
= smb2cli_tcon_send(state
, state
->ev
,
227 state
->transport
->conn
,
229 state
->session
->smbXcli
,
230 state
->tree
->smbXcli
,
233 if (tevent_req_nomem(subreq
, req
)) {
236 tevent_req_set_callback(subreq
, smb2_connect_tcon_done
, req
);
239 static void smb2_connect_tcon_done(struct tevent_req
*subreq
)
241 struct tevent_req
*req
=
242 tevent_req_callback_data(subreq
,
246 status
= smb2cli_tcon_recv(subreq
);
247 if (tevent_req_nterror(req
, status
)) {
251 tevent_req_done(req
);
254 NTSTATUS
smb2_connect_recv(struct tevent_req
*req
,
256 struct smb2_tree
**tree
)
258 struct smb2_connect_state
*state
=
260 struct smb2_connect_state
);
263 if (tevent_req_is_nterror(req
, &status
)) {
264 tevent_req_received(req
);
268 *tree
= talloc_move(mem_ctx
, &state
->tree
);
270 tevent_req_received(req
);
275 sync version of smb2_connect
277 NTSTATUS
smb2_connect_ext(TALLOC_CTX
*mem_ctx
,
281 struct resolve_context
*resolve_ctx
,
282 struct cli_credentials
*credentials
,
283 uint64_t previous_session_id
,
284 struct smb2_tree
**tree
,
285 struct tevent_context
*ev
,
286 const struct smbcli_options
*options
,
287 const char *socket_options
,
288 struct gensec_settings
*gensec_settings
)
290 struct tevent_req
*subreq
;
293 TALLOC_CTX
*frame
= talloc_stackframe();
296 return NT_STATUS_NO_MEMORY
;
299 subreq
= smb2_connect_send(frame
,
310 if (subreq
== NULL
) {
312 return NT_STATUS_NO_MEMORY
;
315 ok
= tevent_req_poll(subreq
, ev
);
317 status
= map_nt_error_from_unix_common(errno
);
322 status
= smb2_connect_recv(subreq
, mem_ctx
, tree
);
324 if (!NT_STATUS_IS_OK(status
)) {
333 NTSTATUS
smb2_connect(TALLOC_CTX
*mem_ctx
,
337 struct resolve_context
*resolve_ctx
,
338 struct cli_credentials
*credentials
,
339 struct smb2_tree
**tree
,
340 struct tevent_context
*ev
,
341 struct smbcli_options
*options
,
342 const char *socket_options
,
343 struct gensec_settings
*gensec_settings
)
347 status
= smb2_connect_ext(mem_ctx
, host
, ports
, share
, resolve_ctx
,
349 0, /* previous_session_id */
350 tree
, ev
, options
, socket_options
,