?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
libcli: Avoid an explicit memset call
[Samba.git] / source3 / libsmb / clilist.c
blob3080fb883dd86b1579bdbaf8c035c431dedacd42
1 /*
2 Unix SMB/CIFS implementation.
3 client directory list routines
4 Copyright (C) Andrew Tridgell 1994-1998
5
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.
10
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.
15
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/>.
18 */
19
20 #include "includes.h"
21 #include "libsmb/libsmb.h"
22 #include "../lib/util/tevent_ntstatus.h"
23 #include "async_smb.h"
24 #include "trans2.h"
25 #include "../libcli/smb/smbXcli_base.h"
26
27 /****************************************************************************
28 Calculate a safe next_entry_offset.
29 ****************************************************************************/
30
31 static size_t calc_next_entry_offset(const char *base, const char *pdata_end)
32 {
33 size_t next_entry_offset = (size_t)IVAL(base,0);
34
35 if (next_entry_offset == 0 ||
36 base + next_entry_offset < base ||
37 base + next_entry_offset > pdata_end) {
38 next_entry_offset = pdata_end - base;
39 }
40 return next_entry_offset;
41 }
42
43 /****************************************************************************
44 Interpret a long filename structure - this is mostly guesses at the moment.
45 The length of the structure is returned
46 The structure of a long filename depends on the info level.
47 SMB_FIND_FILE_BOTH_DIRECTORY_INFO is used
48 by NT and SMB_FIND_EA_SIZE is used by OS/2
49 ****************************************************************************/
50
51 static size_t interpret_long_filename(TALLOC_CTX *ctx,
52 struct cli_state *cli,
53 int level,
54 const char *base_ptr,
55 uint16_t recv_flags2,
56 const char *p,
57 const char *pdata_end,
58 struct file_info *finfo,
59 uint32 *p_resume_key,
60 DATA_BLOB *p_last_name_raw)
61 {
62 int len;
63 size_t ret;
64 const char *base = p;
65
66 data_blob_free(p_last_name_raw);
67
68 if (p_resume_key) {
69 *p_resume_key = 0;
70 }
71 ZERO_STRUCTP(finfo);
72
73 switch (level) {
74 case SMB_FIND_INFO_STANDARD: /* OS/2 understands this */
75 /* these dates are converted to GMT by
76 make_unix_date */
77 if (pdata_end - base < 27) {
78 return pdata_end - base;
79 }
80 finfo->ctime_ts = convert_time_t_to_timespec(
81 make_unix_date2(p+4, smb1cli_conn_server_time_zone(cli->conn)));
82 finfo->atime_ts = convert_time_t_to_timespec(
83 make_unix_date2(p+8, smb1cli_conn_server_time_zone(cli->conn)));
84 finfo->mtime_ts = convert_time_t_to_timespec(
85 make_unix_date2(p+12, smb1cli_conn_server_time_zone(cli->conn)));
86 finfo->size = IVAL(p,16);
87 finfo->mode = CVAL(p,24);
88 len = CVAL(p, 26);
89 p += 27;
90 if (recv_flags2 & FLAGS2_UNICODE_STRINGS) {
91 p += ucs2_align(base_ptr, p, STR_UNICODE);
92 }
93
94 /* We can safely use len here (which is required by OS/2)
95 * and the NAS-BASIC server instead of +2 or +1 as the
96 * STR_TERMINATE flag below is
97 * actually used as the length calculation.
98 * The len is merely an upper bound.
99 * Due to the explicit 2 byte null termination
100 * in cli_receive_trans/cli_receive_nt_trans
101 * we know this is safe. JRA + kukks
102 */
103
104 if (p + len > pdata_end) {
105 return pdata_end - base;
106 }
107
108 /* the len+2 below looks strange but it is
109 important to cope with the differences
110 between win2000 and win9x for this call
111 (tridge) */
112 ret = clistr_pull_talloc(ctx,
113 base_ptr,
114 recv_flags2,
115 &finfo->name,
116 p,
117 len+2,
118 STR_TERMINATE);
119 if (ret == (size_t)-1) {
120 return pdata_end - base;
121 }
122 p += ret;
123 return PTR_DIFF(p, base);
124
125 case SMB_FIND_EA_SIZE: /* this is what OS/2 uses mostly */
126 /* these dates are converted to GMT by
127 make_unix_date */
128 if (pdata_end - base < 31) {
129 return pdata_end - base;
130 }
131 finfo->ctime_ts = convert_time_t_to_timespec(
132 make_unix_date2(p+4, smb1cli_conn_server_time_zone(cli->conn)));
133 finfo->atime_ts = convert_time_t_to_timespec(
134 make_unix_date2(p+8, smb1cli_conn_server_time_zone(cli->conn)));
135 finfo->mtime_ts = convert_time_t_to_timespec(
136 make_unix_date2(p+12, smb1cli_conn_server_time_zone(cli->conn)));
137 finfo->size = IVAL(p,16);
138 finfo->mode = CVAL(p,24);
139 len = CVAL(p, 30);
140 p += 31;
141 /* check for unisys! */
142 if (p + len + 1 > pdata_end) {
143 return pdata_end - base;
144 }
145 ret = clistr_pull_talloc(ctx,
146 base_ptr,
147 recv_flags2,
148 &finfo->name,
149 p,
150 len,
151 STR_NOALIGN);
152 if (ret == (size_t)-1) {
153 return pdata_end - base;
154 }
155 p += ret;
156 return PTR_DIFF(p, base) + 1;
157
158 case SMB_FIND_FILE_BOTH_DIRECTORY_INFO: /* NT uses this, but also accepts 2 */
159 {
160 size_t namelen, slen;
161
162 if (pdata_end - base < 94) {
163 return pdata_end - base;
164 }
165
166 p += 4; /* next entry offset */
167
168 if (p_resume_key) {
169 *p_resume_key = IVAL(p,0);
170 }
171 p += 4; /* fileindex */
172
173 /* Offset zero is "create time", not "change time". */
174 p += 8;
175 finfo->atime_ts = interpret_long_date(p);
176 p += 8;
177 finfo->mtime_ts = interpret_long_date(p);
178 p += 8;
179 finfo->ctime_ts = interpret_long_date(p);
180 p += 8;
181 finfo->size = IVAL2_TO_SMB_BIG_UINT(p,0);
182 p += 8;
183 p += 8; /* alloc size */
184 finfo->mode = CVAL(p,0);
185 p += 4;
186 namelen = IVAL(p,0);
187 p += 4;
188 p += 4; /* EA size */
189 slen = SVAL(p, 0);
190 if (slen > 24) {
191 /* Bad short name length. */
192 return pdata_end - base;
193 }
194 p += 2;
195 ret = clistr_pull_talloc(ctx,
196 base_ptr,
197 recv_flags2,
198 &finfo->short_name,
199 p,
200 slen,
201 STR_UNICODE);
202 if (ret == (size_t)-1) {
203 return pdata_end - base;
204 }
205 p += 24; /* short name? */
206 if (p + namelen < p || p + namelen > pdata_end) {
207 return pdata_end - base;
208 }
209 ret = clistr_pull_talloc(ctx,
210 base_ptr,
211 recv_flags2,
212 &finfo->name,
213 p,
214 namelen,
215 0);
216 if (ret == (size_t)-1) {
217 return pdata_end - base;
218 }
219
220 /* To be robust in the face of unicode conversion failures
221 we need to copy the raw bytes of the last name seen here.
222 Namelen doesn't include the terminating unicode null, so
223 copy it here. */
224
225 if (p_last_name_raw) {
226 *p_last_name_raw = data_blob(NULL, namelen+2);
227 memcpy(p_last_name_raw->data, p, namelen);
228 SSVAL(p_last_name_raw->data, namelen, 0);
229 }
230 return calc_next_entry_offset(base, pdata_end);
231 }
232 }
233
234 DEBUG(1,("Unknown long filename format %d\n",level));
235 return calc_next_entry_offset(base, pdata_end);
236 }
237
238 /****************************************************************************
239 Interpret a short filename structure.
240 The length of the structure is returned.
241 ****************************************************************************/
242
243 static bool interpret_short_filename(TALLOC_CTX *ctx,
244 struct cli_state *cli,
245 char *p,
246 struct file_info *finfo)
247 {
248 size_t ret;
249 ZERO_STRUCTP(finfo);
250
251 finfo->mode = CVAL(p,21);
252
253 /* this date is converted to GMT by make_unix_date */
254 finfo->ctime_ts.tv_sec = make_unix_date(p+22, smb1cli_conn_server_time_zone(cli->conn));
255 finfo->ctime_ts.tv_nsec = 0;
256 finfo->mtime_ts.tv_sec = finfo->atime_ts.tv_sec = finfo->ctime_ts.tv_sec;
257 finfo->mtime_ts.tv_nsec = finfo->atime_ts.tv_nsec = 0;
258 finfo->size = IVAL(p,26);
259 ret = clistr_pull_talloc(ctx,
260 NULL,
261 0,
262 &finfo->name,
263 p+30,
264 12,
265 STR_ASCII);
266 if (ret == (size_t)-1) {
267 return false;
268 }
269
270 if (finfo->name) {
271 finfo->short_name = talloc_strdup(ctx, finfo->name);
272 if (finfo->short_name == NULL) {
273 return false;
274 }
275 }
276 return true;
277 }
278
279 struct cli_list_old_state {
280 struct tevent_context *ev;
281 struct cli_state *cli;
282 uint16_t vwv[2];
283 char *mask;
284 int num_asked;
285 uint16_t attribute;
286 uint8_t search_status[23];
287 bool first;
288 bool done;
289 uint8_t *dirlist;
290 };
291
292 static void cli_list_old_done(struct tevent_req *subreq);
293
294 static struct tevent_req *cli_list_old_send(TALLOC_CTX *mem_ctx,
295 struct tevent_context *ev,
296 struct cli_state *cli,
297 const char *mask,
298 uint16_t attribute)
299 {
300 struct tevent_req *req, *subreq;
301 struct cli_list_old_state *state;
302 uint8_t *bytes;
303 static const uint16_t zero = 0;
304 uint32_t usable_space;
305
306 req = tevent_req_create(mem_ctx, &state, struct cli_list_old_state);
307 if (req == NULL) {
308 return NULL;
309 }
310 state->ev = ev;
311 state->cli = cli;
312 state->attribute = attribute;
313 state->first = true;
314 state->mask = talloc_strdup(state, mask);
315 if (tevent_req_nomem(state->mask, req)) {
316 return tevent_req_post(req, ev);
317 }
318 usable_space = cli_state_available_size(cli, 100);
319 state->num_asked = usable_space / DIR_STRUCT_SIZE;
320
321 SSVAL(state->vwv + 0, 0, state->num_asked);
322 SSVAL(state->vwv + 1, 0, state->attribute);
323
324 bytes = talloc_array(state, uint8_t, 1);
325 if (tevent_req_nomem(bytes, req)) {
326 return tevent_req_post(req, ev);
327 }
328 bytes[0] = 4;
329 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), mask,
330 strlen(mask)+1, NULL);
331
332 bytes = smb_bytes_push_bytes(bytes, 5, (const uint8_t *)&zero, 2);
333 if (tevent_req_nomem(bytes, req)) {
334 return tevent_req_post(req, ev);
335 }
336
337 subreq = cli_smb_send(state, state->ev, state->cli, SMBsearch,
338 0, 2, state->vwv, talloc_get_size(bytes), bytes);
339 if (tevent_req_nomem(subreq, req)) {
340 return tevent_req_post(req, ev);
341 }
342 tevent_req_set_callback(subreq, cli_list_old_done, req);
343 return req;
344 }
345
346 static void cli_list_old_done(struct tevent_req *subreq)
347 {
348 struct tevent_req *req = tevent_req_callback_data(
349 subreq, struct tevent_req);
350 struct cli_list_old_state *state = tevent_req_data(
351 req, struct cli_list_old_state);
352 NTSTATUS status;
353 uint8_t cmd;
354 uint8_t wct;
355 uint16_t *vwv;
356 uint32_t num_bytes;
357 uint8_t *bytes;
358 uint16_t received;
359 size_t dirlist_len;
360 uint8_t *tmp;
361
362 status = cli_smb_recv(subreq, state, NULL, 0, &wct, &vwv, &num_bytes,
363 &bytes);
364 if (!NT_STATUS_IS_OK(status)
365 && !NT_STATUS_EQUAL(status, NT_STATUS_DOS(ERRDOS, ERRnofiles))
366 && !NT_STATUS_EQUAL(status, STATUS_NO_MORE_FILES)) {
367 TALLOC_FREE(subreq);
368 tevent_req_nterror(req, status);
369 return;
370 }
371 if (NT_STATUS_EQUAL(status, NT_STATUS_DOS(ERRDOS, ERRnofiles))
372 || NT_STATUS_EQUAL(status, STATUS_NO_MORE_FILES)) {
373 received = 0;
374 } else {
375 if (wct < 1) {
376 TALLOC_FREE(subreq);
377 tevent_req_nterror(
378 req, NT_STATUS_INVALID_NETWORK_RESPONSE);
379 return;
380 }
381 received = SVAL(vwv + 0, 0);
382 }
383
384 if (received > 0) {
385 /*
386 * I don't think this can wrap. received is
387 * initialized from a 16-bit value.
388 */
389 if (num_bytes < (received * DIR_STRUCT_SIZE + 3)) {
390 TALLOC_FREE(subreq);
391 tevent_req_nterror(
392 req, NT_STATUS_INVALID_NETWORK_RESPONSE);
393 return;
394 }
395
396 dirlist_len = talloc_get_size(state->dirlist);
397
398 tmp = talloc_realloc(
399 state, state->dirlist, uint8_t,
400 dirlist_len + received * DIR_STRUCT_SIZE);
401 if (tevent_req_nomem(tmp, req)) {
402 return;
403 }
404 state->dirlist = tmp;
405 memcpy(state->dirlist + dirlist_len, bytes + 3,
406 received * DIR_STRUCT_SIZE);
407
408 SSVAL(state->search_status, 0, 21);
409 memcpy(state->search_status + 2,
410 bytes + 3 + (received-1)*DIR_STRUCT_SIZE, 21);
411 cmd = SMBsearch;
412 } else {
413 if (state->first || state->done) {
414 tevent_req_done(req);
415 return;
416 }
417 state->done = true;
418 state->num_asked = 0;
419 cmd = SMBfclose;
420 }
421 TALLOC_FREE(subreq);
422
423 state->first = false;
424
425 SSVAL(state->vwv + 0, 0, state->num_asked);
426 SSVAL(state->vwv + 1, 0, state->attribute);
427
428 bytes = talloc_array(state, uint8_t, 1);
429 if (tevent_req_nomem(bytes, req)) {
430 return;
431 }
432 bytes[0] = 4;
433 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(state->cli->conn), "",
434 1, NULL);
435 bytes = smb_bytes_push_bytes(bytes, 5, state->search_status,
436 sizeof(state->search_status));
437 if (tevent_req_nomem(bytes, req)) {
438 return;
439 }
440 subreq = cli_smb_send(state, state->ev, state->cli, cmd, 0,
441 2, state->vwv, talloc_get_size(bytes), bytes);
442 if (tevent_req_nomem(subreq, req)) {
443 return;
444 }
445 tevent_req_set_callback(subreq, cli_list_old_done, req);
446 }
447
448 static NTSTATUS cli_list_old_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
449 struct file_info **pfinfo)
450 {
451 struct cli_list_old_state *state = tevent_req_data(
452 req, struct cli_list_old_state);
453 NTSTATUS status;
454 size_t i, num_received;
455 struct file_info *finfo;
456
457 if (tevent_req_is_nterror(req, &status)) {
458 return status;
459 }
460
461 num_received = talloc_array_length(state->dirlist) / DIR_STRUCT_SIZE;
462
463 finfo = talloc_array(mem_ctx, struct file_info, num_received);
464 if (finfo == NULL) {
465 return NT_STATUS_NO_MEMORY;
466 }
467
468 for (i=0; i<num_received; i++) {
469 if (!interpret_short_filename(
470 finfo, state->cli,
471 (char *)state->dirlist + i * DIR_STRUCT_SIZE,
472 &finfo[i])) {
473 TALLOC_FREE(finfo);
474 return NT_STATUS_NO_MEMORY;
475 }
476 }
477 *pfinfo = finfo;
478 return NT_STATUS_OK;
479 }
480
481 NTSTATUS cli_list_old(struct cli_state *cli, const char *mask,
482 uint16 attribute,
483 NTSTATUS (*fn)(const char *, struct file_info *,
484 const char *, void *), void *state)
485 {
486 TALLOC_CTX *frame = talloc_stackframe();
487 struct tevent_context *ev;
488 struct tevent_req *req;
489 NTSTATUS status = NT_STATUS_NO_MEMORY;
490 struct file_info *finfo;
491 size_t i, num_finfo;
492
493 if (smbXcli_conn_has_async_calls(cli->conn)) {
494 /*
495 * Can't use sync call while an async call is in flight
496 */
497 status = NT_STATUS_INVALID_PARAMETER;
498 goto fail;
499 }
500 ev = samba_tevent_context_init(frame);
501 if (ev == NULL) {
502 goto fail;
503 }
504 req = cli_list_old_send(frame, ev, cli, mask, attribute);
505 if (req == NULL) {
506 goto fail;
507 }
508 if (!tevent_req_poll(req, ev)) {
509 status = map_nt_error_from_unix(errno);
510 goto fail;
511 }
512 status = cli_list_old_recv(req, frame, &finfo);
513 if (!NT_STATUS_IS_OK(status)) {
514 goto fail;
515 }
516 num_finfo = talloc_array_length(finfo);
517 for (i=0; i<num_finfo; i++) {
518 status = fn(cli->dfs_mountpoint, &finfo[i], mask, state);
519 if (!NT_STATUS_IS_OK(status)) {
520 goto fail;
521 }
522 }
523 fail:
524 TALLOC_FREE(frame);
525 return status;
526 }
527
528 struct cli_list_trans_state {
529 struct tevent_context *ev;
530 struct cli_state *cli;
531 char *mask;
532 uint16_t attribute;
533 uint16_t info_level;
534
535 int loop_count;
536 int total_received;
537 uint16_t max_matches;
538 bool first;
539
540 int ff_eos;
541 int ff_dir_handle;
542
543 uint16_t setup[1];
544 uint8_t *param;
545
546 struct file_info *finfo;
547 };
548
549 static void cli_list_trans_done(struct tevent_req *subreq);
550
551 static struct tevent_req *cli_list_trans_send(TALLOC_CTX *mem_ctx,
552 struct tevent_context *ev,
553 struct cli_state *cli,
554 const char *mask,
555 uint16_t attribute,
556 uint16_t info_level)
557 {
558 struct tevent_req *req, *subreq;
559 struct cli_list_trans_state *state;
560 size_t param_len;
561
562 req = tevent_req_create(mem_ctx, &state,
563 struct cli_list_trans_state);
564 if (req == NULL) {
565 return NULL;
566 }
567 state->ev = ev;
568 state->cli = cli;
569 state->mask = talloc_strdup(state, mask);
570 if (tevent_req_nomem(state->mask, req)) {
571 return tevent_req_post(req, ev);
572 }
573 state->attribute = attribute;
574 state->info_level = info_level;
575 state->loop_count = 0;
576 state->first = true;
577
578 state->max_matches = 1366; /* Match W2k */
579
580 SSVAL(&state->setup[0], 0, TRANSACT2_FINDFIRST);
581
582 state->param = talloc_array(state, uint8_t, 12);
583 if (tevent_req_nomem(state->param, req)) {
584 return tevent_req_post(req, ev);
585 }
586
587 SSVAL(state->param, 0, state->attribute);
588 SSVAL(state->param, 2, state->max_matches);
589 SSVAL(state->param, 4,
590 FLAG_TRANS2_FIND_REQUIRE_RESUME
591 |FLAG_TRANS2_FIND_CLOSE_IF_END
592 |(cli->backup_intent ? FLAG_TRANS2_FIND_BACKUP_INTENT : 0));
593 SSVAL(state->param, 6, state->info_level);
594 SIVAL(state->param, 8, 0);
595
596 state->param = trans2_bytes_push_str(state->param, smbXcli_conn_use_unicode(cli->conn),
597 state->mask, strlen(state->mask)+1,
598 NULL);
599 if (tevent_req_nomem(state->param, req)) {
600 return tevent_req_post(req, ev);
601 }
602 param_len = talloc_get_size(state->param);
603
604 subreq = cli_trans_send(state, state->ev, state->cli,
605 SMBtrans2, NULL, -1, 0, 0,
606 state->setup, 1, 0,
607 state->param, param_len, 10,
608 NULL, 0, CLI_BUFFER_SIZE);
609 if (tevent_req_nomem(subreq, req)) {
610 return tevent_req_post(req, ev);
611 }
612 tevent_req_set_callback(subreq, cli_list_trans_done, req);
613 return req;
614 }
615
616 static void cli_list_trans_done(struct tevent_req *subreq)
617 {
618 struct tevent_req *req = tevent_req_callback_data(
619 subreq, struct tevent_req);
620 struct cli_list_trans_state *state = tevent_req_data(
621 req, struct cli_list_trans_state);
622 NTSTATUS status;
623 uint8_t *param;
624 uint32_t num_param;
625 uint8_t *data;
626 char *data_end;
627 uint32_t num_data;
628 uint32_t min_param;
629 struct file_info *tmp;
630 size_t old_num_finfo;
631 uint16_t recv_flags2;
632 int ff_searchcount;
633 bool ff_eos;
634 char *p, *p2;
635 uint32_t resume_key = 0;
636 int i;
637 DATA_BLOB last_name_raw;
638 struct file_info *finfo = NULL;
639 size_t param_len;
640
641 min_param = (state->first ? 6 : 4);
642
643 status = cli_trans_recv(subreq, talloc_tos(), &recv_flags2,
644 NULL, 0, NULL,
645 &param, min_param, &num_param,
646 &data, 0, &num_data);
647 TALLOC_FREE(subreq);
648 if (!NT_STATUS_IS_OK(status)) {
649 /*
650 * TODO: retry, OS/2 nofiles
651 */
652 tevent_req_nterror(req, status);
653 return;
654 }
655
656 if (state->first) {
657 state->ff_dir_handle = SVAL(param, 0);
658 ff_searchcount = SVAL(param, 2);
659 ff_eos = SVAL(param, 4) != 0;
660 } else {
661 ff_searchcount = SVAL(param, 0);
662 ff_eos = SVAL(param, 2) != 0;
663 }
664
665 old_num_finfo = talloc_array_length(state->finfo);
666
667 tmp = talloc_realloc(state, state->finfo, struct file_info,
668 old_num_finfo + ff_searchcount);
669 if (tevent_req_nomem(tmp, req)) {
670 return;
671 }
672 state->finfo = tmp;
673
674 p2 = p = (char *)data;
675 data_end = (char *)data + num_data;
676 last_name_raw = data_blob_null;
677
678 for (i=0; i<ff_searchcount; i++) {
679 if (p2 >= data_end) {
680 ff_eos = true;
681 break;
682 }
683 if ((state->info_level == SMB_FIND_FILE_BOTH_DIRECTORY_INFO)
684 && (i == ff_searchcount-1)) {
685 /* Last entry - fixup the last offset length. */
686 SIVAL(p2, 0, PTR_DIFF((data + num_data), p2));
687 }
688
689 data_blob_free(&last_name_raw);
690
691 finfo = &state->finfo[old_num_finfo + i];
692
693 p2 += interpret_long_filename(
694 state->finfo, /* Stick fname to the array as such */
695 state->cli, state->info_level,
696 (char *)data, recv_flags2, p2,
697 data_end, finfo, &resume_key, &last_name_raw);
698
699 if (finfo->name == NULL) {
700 DEBUG(1, ("cli_list: Error: unable to parse name from "
701 "info level %d\n", state->info_level));
702 ff_eos = true;
703 break;
704 }
705 if (!state->first && (state->mask[0] != '\0') &&
706 strcsequal(finfo->name, state->mask)) {
707 DEBUG(1, ("Error: Looping in FIND_NEXT as name %s has "
708 "already been seen?\n", finfo->name));
709 ff_eos = true;
710 break;
711 }
712 }
713
714 if (ff_searchcount == 0) {
715 ff_eos = true;
716 }
717
718 TALLOC_FREE(param);
719 TALLOC_FREE(data);
720
721 /*
722 * Shrink state->finfo to the real length we received
723 */
724 tmp = talloc_realloc(state, state->finfo, struct file_info,
725 old_num_finfo + i);
726 if (tevent_req_nomem(tmp, req)) {
727 return;
728 }
729 state->finfo = tmp;
730
731 state->first = false;
732
733 if (ff_eos) {
734 data_blob_free(&last_name_raw);
735 tevent_req_done(req);
736 return;
737 }
738
739 TALLOC_FREE(state->mask);
740 state->mask = talloc_strdup(state, finfo->name);
741 if (tevent_req_nomem(state->mask, req)) {
742 return;
743 }
744
745 SSVAL(&state->setup[0], 0, TRANSACT2_FINDNEXT);
746
747 param = talloc_realloc(state, state->param, uint8_t, 12);
748 if (tevent_req_nomem(param, req)) {
749 return;
750 }
751 state->param = param;
752
753 SSVAL(param, 0, state->ff_dir_handle);
754 SSVAL(param, 2, state->max_matches); /* max count */
755 SSVAL(param, 4, state->info_level);
756 /*
757 * For W2K servers serving out FAT filesystems we *must* set
758 * the resume key. If it's not FAT then it's returned as zero.
759 */
760 SIVAL(param, 6, resume_key); /* ff_resume_key */
761 /*
762 * NB. *DON'T* use continue here. If you do it seems that W2K
763 * and bretheren can miss filenames. Use last filename
764 * continue instead. JRA
765 */
766 SSVAL(param, 10, (FLAG_TRANS2_FIND_REQUIRE_RESUME
767 |FLAG_TRANS2_FIND_CLOSE_IF_END
768 |(state->cli->backup_intent ? FLAG_TRANS2_FIND_BACKUP_INTENT : 0)));
769 if (last_name_raw.length) {
770 state->param = trans2_bytes_push_bytes(state->param,
771 last_name_raw.data,
772 last_name_raw.length);
773 if (tevent_req_nomem(state->param, req)) {
774 return;
775 }
776 data_blob_free(&last_name_raw);
777 } else {
778 state->param = trans2_bytes_push_str(state->param,
779 smbXcli_conn_use_unicode(state->cli->conn),
780 state->mask,
781 strlen(state->mask)+1,
782 NULL);
783 if (tevent_req_nomem(state->param, req)) {
784 return;
785 }
786 }
787 param_len = talloc_get_size(state->param);
788
789 subreq = cli_trans_send(state, state->ev, state->cli,
790 SMBtrans2, NULL, -1, 0, 0,
791 state->setup, 1, 0,
792 state->param, param_len, 10,
793 NULL, 0, CLI_BUFFER_SIZE);
794 if (tevent_req_nomem(subreq, req)) {
795 return;
796 }
797 tevent_req_set_callback(subreq, cli_list_trans_done, req);
798 }
799
800 static NTSTATUS cli_list_trans_recv(struct tevent_req *req,
801 TALLOC_CTX *mem_ctx,
802 struct file_info **finfo)
803 {
804 struct cli_list_trans_state *state = tevent_req_data(
805 req, struct cli_list_trans_state);
806 NTSTATUS status;
807
808 if (tevent_req_is_nterror(req, &status)) {
809 return status;
810 }
811 *finfo = talloc_move(mem_ctx, &state->finfo);
812 return NT_STATUS_OK;
813 }
814
815 NTSTATUS cli_list_trans(struct cli_state *cli, const char *mask,
816 uint16_t attribute, int info_level,
817 NTSTATUS (*fn)(const char *mnt, struct file_info *finfo,
818 const char *mask, void *private_data),
819 void *private_data)
820 {
821 TALLOC_CTX *frame = talloc_stackframe();
822 struct tevent_context *ev;
823 struct tevent_req *req;
824 int i, num_finfo;
825 struct file_info *finfo = NULL;
826 NTSTATUS status = NT_STATUS_NO_MEMORY;
827
828 if (smbXcli_conn_has_async_calls(cli->conn)) {
829 /*
830 * Can't use sync call while an async call is in flight
831 */
832 status = NT_STATUS_INVALID_PARAMETER;
833 goto fail;
834 }
835 ev = samba_tevent_context_init(frame);
836 if (ev == NULL) {
837 goto fail;
838 }
839 req = cli_list_trans_send(frame, ev, cli, mask, attribute, info_level);
840 if (req == NULL) {
841 goto fail;
842 }
843 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
844 goto fail;
845 }
846 status = cli_list_trans_recv(req, frame, &finfo);
847 if (!NT_STATUS_IS_OK(status)) {
848 goto fail;
849 }
850 num_finfo = talloc_array_length(finfo);
851 for (i=0; i<num_finfo; i++) {
852 status = fn(cli->dfs_mountpoint, &finfo[i], mask, private_data);
853 if (!NT_STATUS_IS_OK(status)) {
854 goto fail;
855 }
856 }
857 fail:
858 TALLOC_FREE(frame);
859 return status;
860 }
861
862 struct cli_list_state {
863 NTSTATUS (*recv_fn)(struct tevent_req *req, TALLOC_CTX *mem_ctx,
864 struct file_info **finfo);
865 struct file_info *finfo;
866 };
867
868 static void cli_list_done(struct tevent_req *subreq);
869
870 struct tevent_req *cli_list_send(TALLOC_CTX *mem_ctx,
871 struct tevent_context *ev,
872 struct cli_state *cli,
873 const char *mask,
874 uint16_t attribute,
875 uint16_t info_level)
876 {
877 struct tevent_req *req, *subreq;
878 struct cli_list_state *state;
879
880 req = tevent_req_create(mem_ctx, &state, struct cli_list_state);
881 if (req == NULL) {
882 return NULL;
883 }
884
885 if (smbXcli_conn_protocol(cli->conn) <= PROTOCOL_LANMAN1) {
886 subreq = cli_list_old_send(state, ev, cli, mask, attribute);
887 state->recv_fn = cli_list_old_recv;
888 } else {
889 subreq = cli_list_trans_send(state, ev, cli, mask, attribute,
890 info_level);
891 state->recv_fn = cli_list_trans_recv;
892 }
893 if (tevent_req_nomem(subreq, req)) {
894 return tevent_req_post(req, ev);
895 }
896 tevent_req_set_callback(subreq, cli_list_done, req);
897 return req;
898 }
899
900 static void cli_list_done(struct tevent_req *subreq)
901 {
902 struct tevent_req *req = tevent_req_callback_data(
903 subreq, struct tevent_req);
904 struct cli_list_state *state = tevent_req_data(
905 req, struct cli_list_state);
906 NTSTATUS status;
907
908 status = state->recv_fn(subreq, state, &state->finfo);
909 TALLOC_FREE(subreq);
910 if (!NT_STATUS_IS_OK(status)) {
911 tevent_req_nterror(req, status);
912 return;
913 }
914 tevent_req_done(req);
915 }
916
917 NTSTATUS cli_list_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
918 struct file_info **finfo, size_t *num_finfo)
919 {
920 struct cli_list_state *state = tevent_req_data(
921 req, struct cli_list_state);
922 NTSTATUS status;
923
924 if (tevent_req_is_nterror(req, &status)) {
925 return status;
926 }
927 *num_finfo = talloc_array_length(state->finfo);
928 *finfo = talloc_move(mem_ctx, &state->finfo);
929 return NT_STATUS_OK;
930 }
931
932 NTSTATUS cli_list(struct cli_state *cli, const char *mask, uint16 attribute,
933 NTSTATUS (*fn)(const char *, struct file_info *, const char *,
934 void *), void *state)
935 {
936 TALLOC_CTX *frame = NULL;
937 struct tevent_context *ev;
938 struct tevent_req *req;
939 NTSTATUS status = NT_STATUS_NO_MEMORY;
940 struct file_info *finfo;
941 size_t i, num_finfo;
942 uint16_t info_level;
943
944 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
945 return cli_smb2_list(cli, mask, attribute, fn, state);
946 }
947
948 frame = talloc_stackframe();
949
950 if (smbXcli_conn_has_async_calls(cli->conn)) {
951 /*
952 * Can't use sync call while an async call is in flight
953 */
954 status = NT_STATUS_INVALID_PARAMETER;
955 goto fail;
956 }
957 ev = samba_tevent_context_init(frame);
958 if (ev == NULL) {
959 goto fail;
960 }
961
962 info_level = (smb1cli_conn_capabilities(cli->conn) & CAP_NT_SMBS)
963 ? SMB_FIND_FILE_BOTH_DIRECTORY_INFO : SMB_FIND_INFO_STANDARD;
964
965 req = cli_list_send(frame, ev, cli, mask, attribute, info_level);
966 if (req == NULL) {
967 goto fail;
968 }
969 if (!tevent_req_poll(req, ev)) {
970 status = map_nt_error_from_unix(errno);
971 goto fail;
972 }
973
974 status = cli_list_recv(req, frame, &finfo, &num_finfo);
975 if (!NT_STATUS_IS_OK(status)) {
976 goto fail;
977 }
978
979 for (i=0; i<num_finfo; i++) {
980 status = fn(cli->dfs_mountpoint, &finfo[i], mask, state);
981 if (!NT_STATUS_IS_OK(status)) {
982 goto fail;
983 }
984 }
985 fail:
986 TALLOC_FREE(frame);
987 return status;
988 }
Samba GIT mirror