2 Unix SMB/CIFS implementation.
3 Connect to 445 and 139/nbsesssetup
4 Copyright (C) Volker Lendecke 2010
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "../lib/async_req/async_sock.h"
22 #include "async_smb.h"
24 struct nb_connect_state
{
25 struct tevent_context
*ev
;
26 const struct sockaddr_storage
*addr
;
27 const char *called_name
;
30 struct nmb_name called
;
31 struct nmb_name calling
;
34 static int nb_connect_state_destructor(struct nb_connect_state
*state
);
35 static void nb_connect_connected(struct tevent_req
*subreq
);
36 static void nb_connect_done(struct tevent_req
*subreq
);
38 static struct tevent_req
*nb_connect_send(TALLOC_CTX
*mem_ctx
,
39 struct tevent_context
*ev
,
40 const struct sockaddr_storage
*addr
,
41 const char *called_name
,
43 const char *calling_name
,
46 struct tevent_req
*req
, *subreq
;
47 struct nb_connect_state
*state
;
49 req
= tevent_req_create(mem_ctx
, &state
, struct nb_connect_state
);
54 state
->called_name
= called_name
;
58 make_nmb_name(&state
->called
, called_name
, called_type
);
59 make_nmb_name(&state
->calling
, calling_name
, calling_type
);
61 talloc_set_destructor(state
, nb_connect_state_destructor
);
63 subreq
= open_socket_out_send(state
, ev
, addr
, 139, 5000);
64 if (tevent_req_nomem(subreq
, req
)) {
65 return tevent_req_post(req
, ev
);
67 tevent_req_set_callback(subreq
, nb_connect_connected
, req
);
71 static int nb_connect_state_destructor(struct nb_connect_state
*state
)
73 if (state
->sock
!= -1) {
79 static void nb_connect_connected(struct tevent_req
*subreq
)
81 struct tevent_req
*req
= tevent_req_callback_data(
82 subreq
, struct tevent_req
);
83 struct nb_connect_state
*state
= tevent_req_data(
84 req
, struct nb_connect_state
);
87 status
= open_socket_out_recv(subreq
, &state
->sock
);
89 if (!NT_STATUS_IS_OK(status
)) {
90 tevent_req_nterror(req
, status
);
93 subreq
= cli_session_request_send(state
, state
->ev
, state
->sock
,
94 &state
->called
, &state
->calling
);
95 if (tevent_req_nomem(subreq
, req
)) {
98 tevent_req_set_callback(subreq
, nb_connect_done
, req
);
101 static void nb_connect_done(struct tevent_req
*subreq
)
103 struct tevent_req
*req
= tevent_req_callback_data(
104 subreq
, struct tevent_req
);
105 struct nb_connect_state
*state
= tevent_req_data(
106 req
, struct nb_connect_state
);
111 ret
= cli_session_request_recv(subreq
, &err
, &resp
);
114 tevent_req_nterror(req
, map_nt_error_from_unix(err
));
119 * RFC1002: 0x82 - POSITIVE SESSION RESPONSE
124 * The server did not like our session request
129 if (strequal(state
->called_name
, "*SMBSERVER")) {
131 * Here we could try a name status request and
132 * use the first 0x20 type name.
135 req
, NT_STATUS_RESOURCE_NAME_NOT_FOUND
);
140 * We could be subtle and distinguish between
141 * different failure modes, but what we do here
142 * instead is just retry with *SMBSERVER type 0x20.
144 state
->called_name
= "*SMBSERVER";
145 make_nmb_name(&state
->called
, state
->called_name
, 0x20);
147 subreq
= open_socket_out_send(state
, state
->ev
, state
->addr
,
149 if (tevent_req_nomem(subreq
, req
)) {
152 tevent_req_set_callback(subreq
, nb_connect_connected
, req
);
156 tevent_req_done(req
);
161 static NTSTATUS
nb_connect_recv(struct tevent_req
*req
, int *sock
)
163 struct nb_connect_state
*state
= tevent_req_data(
164 req
, struct nb_connect_state
);
167 if (tevent_req_is_nterror(req
, &status
)) {
175 struct smbsock_connect_state
{
176 struct tevent_context
*ev
;
177 const struct sockaddr_storage
*addr
;
178 const char *called_name
;
179 const char *calling_name
;
180 struct tevent_req
*req_139
;
181 struct tevent_req
*req_445
;
186 static int smbsock_connect_state_destructor(
187 struct smbsock_connect_state
*state
);
188 static void smbsock_connect_connected(struct tevent_req
*subreq
);
189 static void smbsock_connect_do_139(struct tevent_req
*subreq
);
191 struct tevent_req
*smbsock_connect_send(TALLOC_CTX
*mem_ctx
,
192 struct tevent_context
*ev
,
193 const struct sockaddr_storage
*addr
,
194 const char *called_name
,
195 const char *calling_name
)
197 struct tevent_req
*req
, *subreq
;
198 struct smbsock_connect_state
*state
;
200 req
= tevent_req_create(mem_ctx
, &state
, struct smbsock_connect_state
);
208 (called_name
!= NULL
) ? called_name
: "*SMBSERVER";
209 state
->calling_name
=
210 (calling_name
!= NULL
) ? calling_name
: global_myname();
212 talloc_set_destructor(state
, smbsock_connect_state_destructor
);
214 state
->req_445
= open_socket_out_send(state
, ev
, addr
, 445, 5000);
215 if (tevent_req_nomem(state
->req_445
, req
)) {
216 return tevent_req_post(req
, ev
);
218 tevent_req_set_callback(state
->req_445
, smbsock_connect_connected
,
222 * After 5 msecs, fire the 139 request
224 subreq
= tevent_wakeup_send(state
, ev
, timeval_current_ofs(0, 5000));
225 if (tevent_req_nomem(subreq
, req
)) {
226 TALLOC_FREE(state
->req_445
);
227 return tevent_req_post(req
, ev
);
229 tevent_req_set_callback(subreq
, smbsock_connect_do_139
, req
);
233 static int smbsock_connect_state_destructor(
234 struct smbsock_connect_state
*state
)
236 if (state
->sock
!= -1) {
242 static void smbsock_connect_do_139(struct tevent_req
*subreq
)
244 struct tevent_req
*req
= tevent_req_callback_data(
245 subreq
, struct tevent_req
);
246 struct smbsock_connect_state
*state
= tevent_req_data(
247 req
, struct smbsock_connect_state
);
250 ret
= tevent_wakeup_recv(subreq
);
253 tevent_req_nterror(req
, NT_STATUS_INTERNAL_ERROR
);
256 state
->req_139
= nb_connect_send(state
, state
->ev
, state
->addr
,
257 state
->called_name
, 0x20,
258 state
->calling_name
, 0x0);
259 if (tevent_req_nomem(state
->req_139
, req
)) {
262 tevent_req_set_callback(state
->req_139
, smbsock_connect_connected
,
266 static void smbsock_connect_connected(struct tevent_req
*subreq
)
268 struct tevent_req
*req
= tevent_req_callback_data(
269 subreq
, struct tevent_req
);
270 struct smbsock_connect_state
*state
= tevent_req_data(
271 req
, struct smbsock_connect_state
);
272 struct tevent_req
*unfinished_req
;
275 if (subreq
== state
->req_445
) {
277 status
= open_socket_out_recv(subreq
, &state
->sock
);
278 TALLOC_FREE(state
->req_445
);
279 unfinished_req
= state
->req_139
;
282 } else if (subreq
== state
->req_139
) {
284 status
= nb_connect_recv(subreq
, &state
->sock
);
285 TALLOC_FREE(state
->req_139
);
286 unfinished_req
= state
->req_445
;
290 tevent_req_nterror(req
, NT_STATUS_INTERNAL_ERROR
);
294 if (NT_STATUS_IS_OK(status
)) {
295 TALLOC_FREE(unfinished_req
);
296 state
->req_139
= NULL
;
297 state
->req_445
= NULL
;
298 tevent_req_done(req
);
301 if (unfinished_req
== NULL
) {
303 * Both requests failed
305 tevent_req_nterror(req
, status
);
309 * Do nothing, wait for the second request to come here.
313 NTSTATUS
smbsock_connect_recv(struct tevent_req
*req
, int *sock
,
316 struct smbsock_connect_state
*state
= tevent_req_data(
317 req
, struct smbsock_connect_state
);
320 if (tevent_req_is_nterror(req
, &status
)) {
331 NTSTATUS
smbsock_connect(const struct sockaddr_storage
*addr
,
332 const char *called_name
, const char *calling_name
,
333 int *pfd
, uint16_t *port
)
335 TALLOC_CTX
*frame
= talloc_stackframe();
336 struct event_context
*ev
;
337 struct tevent_req
*req
;
338 NTSTATUS status
= NT_STATUS_NO_MEMORY
;
340 ev
= event_context_init(frame
);
344 req
= smbsock_connect_send(frame
, ev
, addr
, called_name
, calling_name
);
348 if (!tevent_req_poll_ntstatus(req
, ev
, &status
)) {
351 status
= smbsock_connect_recv(req
, pfd
, port
);
357 struct smbsock_any_connect_state
{
358 struct tevent_context
*ev
;
359 const struct sockaddr_storage
*addrs
;
360 const char **called_names
;
363 struct tevent_req
**requests
;
372 static bool smbsock_any_connect_send_next(
373 struct tevent_req
*req
, struct smbsock_any_connect_state
*state
);
374 static void smbsock_any_connect_trynext(struct tevent_req
*subreq
);
375 static void smbsock_any_connect_connected(struct tevent_req
*subreq
);
377 struct tevent_req
*smbsock_any_connect_send(TALLOC_CTX
*mem_ctx
,
378 struct tevent_context
*ev
,
379 const struct sockaddr_storage
*addrs
,
380 const char **called_names
,
383 struct tevent_req
*req
, *subreq
;
384 struct smbsock_any_connect_state
*state
;
386 req
= tevent_req_create(mem_ctx
, &state
,
387 struct smbsock_any_connect_state
);
392 state
->addrs
= addrs
;
393 state
->num_addrs
= num_addrs
;
394 state
->called_names
= called_names
;
396 if (num_addrs
== 0) {
397 tevent_req_nterror(req
, NT_STATUS_INVALID_PARAMETER
);
398 return tevent_req_post(req
, ev
);
401 state
->requests
= talloc_zero_array(state
, struct tevent_req
*,
403 if (tevent_req_nomem(state
->requests
, req
)) {
404 return tevent_req_post(req
, ev
);
406 if (!smbsock_any_connect_send_next(req
, state
)) {
407 return tevent_req_post(req
, ev
);
409 if (state
->num_sent
>= state
->num_addrs
) {
412 subreq
= tevent_wakeup_send(state
, ev
, timeval_current_ofs(0, 10000));
413 if (tevent_req_nomem(subreq
, req
)) {
414 return tevent_req_post(req
, ev
);
416 tevent_req_set_callback(subreq
, smbsock_any_connect_trynext
, req
);
420 static void smbsock_any_connect_trynext(struct tevent_req
*subreq
)
422 struct tevent_req
*req
= tevent_req_callback_data(
423 subreq
, struct tevent_req
);
424 struct smbsock_any_connect_state
*state
= tevent_req_data(
425 req
, struct smbsock_any_connect_state
);
428 ret
= tevent_wakeup_recv(subreq
);
431 tevent_req_nterror(req
, NT_STATUS_INTERNAL_ERROR
);
434 if (!smbsock_any_connect_send_next(req
, state
)) {
437 if (state
->num_sent
>= state
->num_addrs
) {
440 subreq
= tevent_wakeup_send(state
, state
->ev
,
441 tevent_timeval_set(0, 10000));
442 if (tevent_req_nomem(subreq
, req
)) {
445 tevent_req_set_callback(subreq
, smbsock_any_connect_trynext
, req
);
448 static bool smbsock_any_connect_send_next(
449 struct tevent_req
*req
, struct smbsock_any_connect_state
*state
)
451 struct tevent_req
*subreq
;
453 if (state
->num_sent
>= state
->num_addrs
) {
454 tevent_req_nterror(req
, NT_STATUS_INTERNAL_ERROR
);
457 subreq
= smbsock_connect_send(
458 state
->requests
, state
->ev
, &state
->addrs
[state
->num_sent
],
459 (state
->called_names
!= NULL
)
460 ? state
->called_names
[state
->num_sent
] : NULL
,
462 if (tevent_req_nomem(subreq
, req
)) {
465 tevent_req_set_callback(subreq
, smbsock_any_connect_connected
, req
);
467 state
->requests
[state
->num_sent
] = subreq
;
468 state
->num_sent
+= 1;
473 static void smbsock_any_connect_connected(struct tevent_req
*subreq
)
475 struct tevent_req
*req
= tevent_req_callback_data(
476 subreq
, struct tevent_req
);
477 struct smbsock_any_connect_state
*state
= tevent_req_data(
478 req
, struct smbsock_any_connect_state
);
483 size_t chosen_index
= 0;
485 for (i
=0; i
<state
->num_sent
; i
++) {
486 if (state
->requests
[i
] == subreq
) {
491 if (i
== state
->num_sent
) {
492 tevent_req_nterror(req
, NT_STATUS_INTERNAL_ERROR
);
496 status
= smbsock_connect_recv(subreq
, &fd
, &port
);
499 state
->requests
[chosen_index
] = NULL
;
501 if (NT_STATUS_IS_OK(status
)) {
503 * This will kill all the other requests
505 TALLOC_FREE(state
->requests
);
508 state
->chosen_index
= chosen_index
;
509 tevent_req_done(req
);
513 state
->num_received
+= 1;
514 if (state
->num_received
<= state
->num_addrs
) {
516 * More addrs pending, wait for the others
522 * This is the last response, none succeeded.
524 tevent_req_nterror(req
, status
);
528 NTSTATUS
smbsock_any_connect_recv(struct tevent_req
*req
, int *pfd
,
529 size_t *chosen_index
, uint16_t *port
)
531 struct smbsock_any_connect_state
*state
= tevent_req_data(
532 req
, struct smbsock_any_connect_state
);
535 if (tevent_req_is_nterror(req
, &status
)) {
539 if (chosen_index
!= NULL
) {
540 *chosen_index
= state
->chosen_index
;
548 NTSTATUS
smbsock_any_connect(const struct sockaddr_storage
*addrs
,
549 const char **called_names
, size_t num_addrs
,
550 int *pfd
, size_t *chosen_index
, uint16_t *port
)
552 TALLOC_CTX
*frame
= talloc_stackframe();
553 struct event_context
*ev
;
554 struct tevent_req
*req
;
555 NTSTATUS status
= NT_STATUS_NO_MEMORY
;
557 ev
= event_context_init(frame
);
561 req
= smbsock_any_connect_send(frame
, ev
, addrs
, called_names
,
566 if (!tevent_req_poll_ntstatus(req
, ev
, &status
)) {
569 status
= smbsock_any_connect_recv(req
, pfd
, chosen_index
, port
);