2 Unix SMB/CIFS implementation.
3 RAW_SEARCH_* individual test suite
4 Copyright (C) Andrew Tridgell 2003
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 2 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, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 #define BASEDIR "\\testsearch"
27 callback function for single_search
29 static BOOL single_search_callback(void *private, union smb_search_data *file)
31 union smb_search_data *data = private;
39 do a single file (non-wildcard) search
41 static NTSTATUS single_search(struct cli_state *cli,
44 enum search_level level,
45 union smb_search_data *data)
47 union smb_search_first io;
50 io.generic.level = level;
51 if (level == RAW_SEARCH_SEARCH) {
52 io.search_first.in.max_count = 1;
53 io.search_first.in.search_attrib = 0;
54 io.search_first.in.pattern = pattern;
56 io.t2ffirst.in.search_attrib = 0;
57 io.t2ffirst.in.max_count = 1;
58 io.t2ffirst.in.flags = FLAG_TRANS2_FIND_CLOSE;
59 io.t2ffirst.in.storage_type = 0;
60 io.t2ffirst.in.pattern = pattern;
63 status = smb_raw_search_first(cli->tree, mem_ctx,
64 &io, (void *)data, single_search_callback);
72 enum search_level level;
73 uint32 capability_mask;
75 union smb_search_data data;
77 {"SEARCH", RAW_SEARCH_SEARCH, },
78 {"STANDARD", RAW_SEARCH_STANDARD, },
79 {"EA_SIZE", RAW_SEARCH_EA_SIZE, },
80 {"DIRECTORY_INFO", RAW_SEARCH_DIRECTORY_INFO, },
81 {"FULL_DIRECTORY_INFO", RAW_SEARCH_FULL_DIRECTORY_INFO, },
82 {"NAME_INFO", RAW_SEARCH_NAME_INFO, },
83 {"BOTH_DIRECTORY_INFO", RAW_SEARCH_BOTH_DIRECTORY_INFO, },
84 {"LEVEL_261", RAW_SEARCH_261, },
85 {"LEVEL_262", RAW_SEARCH_262, },
86 {"UNIX_INFO", RAW_SEARCH_UNIX_INFO, CAP_UNIX}
89 /* find a level in the table by name */
90 static union smb_search_data *find(const char *name)
93 for (i=0;i<ARRAY_SIZE(levels);i++) {
94 if (NT_STATUS_IS_OK(levels[i].status) &&
95 strcmp(levels[i].name, name) == 0) {
96 return &levels[i].data;
103 basic testing of all RAW_SEARCH_* calls using a single file
105 static BOOL test_one_file(struct cli_state *cli, TALLOC_CTX *mem_ctx)
109 const char *fname = "\\torture_search.txt";
112 union smb_fileinfo all_info, alt_info, name_info;
113 union smb_search_data *s;
115 printf("Testing one file searches\n");
117 fnum = create_complex_file(cli, mem_ctx, fname);
119 printf("ERROR: open of %s failed (%s)\n", fname, cli_errstr(cli));
124 /* call all the levels */
125 for (i=0;i<ARRAY_SIZE(levels);i++) {
126 uint32 cap = cli->transport->negotiate.capabilities;
128 levels[i].status = single_search(cli, mem_ctx, fname,
129 levels[i].level, &levels[i].data);
131 /* see if this server claims to support this level */
132 if ((cap & levels[i].capability_mask) != levels[i].capability_mask) {
136 printf("testing %s\n", levels[i].name);
138 if (!NT_STATUS_IS_OK(levels[i].status)) {
139 printf("search level %s(%d) failed - %s\n",
140 levels[i].name, (int)levels[i].level,
141 nt_errstr(levels[i].status));
146 /* get the all_info file into to check against */
147 all_info.generic.level = RAW_FILEINFO_ALL_INFO;
148 all_info.generic.in.fname = fname;
149 status = smb_raw_pathinfo(cli->tree, mem_ctx, &all_info);
150 if (!NT_STATUS_IS_OK(status)) {
151 printf("RAW_FILEINFO_ALL_INFO failed - %s\n", nt_errstr(status));
156 alt_info.generic.level = RAW_FILEINFO_ALT_NAME_INFO;
157 alt_info.generic.in.fname = fname;
158 status = smb_raw_pathinfo(cli->tree, mem_ctx, &alt_info);
159 if (!NT_STATUS_IS_OK(status)) {
160 printf("RAW_FILEINFO_ALT_NAME_INFO failed - %s\n", nt_errstr(status));
165 name_info.generic.level = RAW_FILEINFO_NAME_INFO;
166 name_info.generic.in.fname = fname;
167 status = smb_raw_pathinfo(cli->tree, mem_ctx, &name_info);
168 if (!NT_STATUS_IS_OK(status)) {
169 printf("RAW_FILEINFO_NAME_INFO failed - %s\n", nt_errstr(status));
174 #define CHECK_VAL(name, sname1, field1, v, sname2, field2) do { \
177 if (s->sname1.field1 != v.sname2.out.field2) { \
178 printf("(%d) %s/%s [%d] != %s/%s [%d]\n", \
180 #sname1, #field1, (int)s->sname1.field1, \
181 #sname2, #field2, (int)v.sname2.out.field2); \
186 #define CHECK_TIME(name, sname1, field1, v, sname2, field2) do { \
189 if (s->sname1.field1 != (~1 & nt_time_to_unix(&v.sname2.out.field2))) { \
190 printf("(%d) %s/%s [%s] != %s/%s [%s]\n", \
192 #sname1, #field1, time_string(mem_ctx, s->sname1.field1), \
193 #sname2, #field2, nt_time_string(mem_ctx, &v.sname2.out.field2)); \
198 #define CHECK_NTTIME(name, sname1, field1, v, sname2, field2) do { \
201 if (memcmp(&s->sname1.field1, &v.sname2.out.field2, sizeof(NTTIME))) { \
202 printf("(%d) %s/%s [%s] != %s/%s [%s]\n", \
204 #sname1, #field1, nt_time_string(mem_ctx, &s->sname1.field1), \
205 #sname2, #field2, nt_time_string(mem_ctx, &v.sname2.out.field2)); \
210 #define CHECK_STR(name, sname1, field1, v, sname2, field2) do { \
213 if (!s->sname1.field1 || strcmp(s->sname1.field1, v.sname2.out.field2.s)) { \
214 printf("(%d) %s/%s [%s] != %s/%s [%s]\n", \
216 #sname1, #field1, s->sname1.field1, \
217 #sname2, #field2, v.sname2.out.field2.s); \
222 #define CHECK_WSTR(name, sname1, field1, v, sname2, field2, flags) do { \
225 if (!s->sname1.field1.s || \
226 strcmp(s->sname1.field1.s, v.sname2.out.field2.s) || \
227 wire_bad_flags(&s->sname1.field1, flags)) { \
228 printf("(%d) %s/%s [%s] != %s/%s [%s]\n", \
230 #sname1, #field1, s->sname1.field1.s, \
231 #sname2, #field2, v.sname2.out.field2.s); \
236 #define CHECK_NAME(name, sname1, field1, fname, flags) do { \
239 if (!s->sname1.field1.s || \
240 strcmp(s->sname1.field1.s, fname) || \
241 wire_bad_flags(&s->sname1.field1, flags)) { \
242 printf("(%d) %s/%s [%s] != %s\n", \
244 #sname1, #field1, s->sname1.field1.s, \
250 /* check that all the results are as expected */
251 CHECK_VAL("SEARCH", search, attrib, all_info, all_info, attrib);
252 CHECK_VAL("STANDARD", standard, attrib, all_info, all_info, attrib);
253 CHECK_VAL("EA_SIZE", ea_size, attrib, all_info, all_info, attrib);
254 CHECK_VAL("DIRECTORY_INFO", directory_info, attrib, all_info, all_info, attrib);
255 CHECK_VAL("FULL_DIRECTORY_INFO", full_directory_info, attrib, all_info, all_info, attrib);
256 CHECK_VAL("BOTH_DIRECTORY_INFO", both_directory_info, attrib, all_info, all_info, attrib);
257 CHECK_VAL("LEVEL_261", level_261, attrib, all_info, all_info, attrib);
258 CHECK_VAL("LEVEL_262", level_262, attrib, all_info, all_info, attrib);
260 CHECK_TIME("SEARCH", search, write_time, all_info, all_info, write_time);
261 CHECK_TIME("STANDARD", standard, write_time, all_info, all_info, write_time);
262 CHECK_TIME("EA_SIZE", ea_size, write_time, all_info, all_info, write_time);
263 CHECK_TIME("STANDARD", standard, create_time, all_info, all_info, create_time);
264 CHECK_TIME("EA_SIZE", ea_size, create_time, all_info, all_info, create_time);
265 CHECK_TIME("STANDARD", standard, access_time, all_info, all_info, access_time);
266 CHECK_TIME("EA_SIZE", ea_size, access_time, all_info, all_info, access_time);
268 CHECK_NTTIME("DIRECTORY_INFO", directory_info, write_time, all_info, all_info, write_time);
269 CHECK_NTTIME("FULL_DIRECTORY_INFO", full_directory_info, write_time, all_info, all_info, write_time);
270 CHECK_NTTIME("BOTH_DIRECTORY_INFO", both_directory_info, write_time, all_info, all_info, write_time);
271 CHECK_NTTIME("LEVEL_261", level_261, write_time, all_info, all_info, write_time);
272 CHECK_NTTIME("LEVEL_262", level_262, write_time, all_info, all_info, write_time);
274 CHECK_NTTIME("DIRECTORY_INFO", directory_info, create_time, all_info, all_info, create_time);
275 CHECK_NTTIME("FULL_DIRECTORY_INFO", full_directory_info, create_time, all_info, all_info, create_time);
276 CHECK_NTTIME("BOTH_DIRECTORY_INFO", both_directory_info, create_time, all_info, all_info, create_time);
277 CHECK_NTTIME("LEVEL_261", level_261, create_time, all_info, all_info, create_time);
278 CHECK_NTTIME("LEVEL_262", level_262, create_time, all_info, all_info, create_time);
280 CHECK_NTTIME("DIRECTORY_INFO", directory_info, access_time, all_info, all_info, access_time);
281 CHECK_NTTIME("FULL_DIRECTORY_INFO", full_directory_info, access_time, all_info, all_info, access_time);
282 CHECK_NTTIME("BOTH_DIRECTORY_INFO", both_directory_info, access_time, all_info, all_info, access_time);
283 CHECK_NTTIME("LEVEL_261", level_261, access_time, all_info, all_info, access_time);
284 CHECK_NTTIME("LEVEL_262", level_262, access_time, all_info, all_info, access_time);
286 CHECK_NTTIME("DIRECTORY_INFO", directory_info, create_time, all_info, all_info, create_time);
287 CHECK_NTTIME("FULL_DIRECTORY_INFO", full_directory_info, create_time, all_info, all_info, create_time);
288 CHECK_NTTIME("BOTH_DIRECTORY_INFO", both_directory_info, create_time, all_info, all_info, create_time);
289 CHECK_NTTIME("LEVEL_261", level_261, create_time, all_info, all_info, create_time);
290 CHECK_NTTIME("LEVEL_262", level_262, create_time, all_info, all_info, create_time);
292 CHECK_VAL("SEARCH", search, size, all_info, all_info, size);
293 CHECK_VAL("STANDARD", standard, size, all_info, all_info, size);
294 CHECK_VAL("EA_SIZE", ea_size, size, all_info, all_info, size);
295 CHECK_VAL("DIRECTORY_INFO", directory_info, size, all_info, all_info, size);
296 CHECK_VAL("FULL_DIRECTORY_INFO", full_directory_info, size, all_info, all_info, size);
297 CHECK_VAL("BOTH_DIRECTORY_INFO", both_directory_info, size, all_info, all_info, size);
298 CHECK_VAL("LEVEL_261", level_261, size, all_info, all_info, size);
299 CHECK_VAL("LEVEL_262", level_262, size, all_info, all_info, size);
301 CHECK_VAL("STANDARD", standard, alloc_size, all_info, all_info, alloc_size);
302 CHECK_VAL("EA_SIZE", ea_size, alloc_size, all_info, all_info, alloc_size);
303 CHECK_VAL("DIRECTORY_INFO", directory_info, alloc_size, all_info, all_info, alloc_size);
304 CHECK_VAL("FULL_DIRECTORY_INFO", full_directory_info, alloc_size, all_info, all_info, alloc_size);
305 CHECK_VAL("BOTH_DIRECTORY_INFO", both_directory_info, alloc_size, all_info, all_info, alloc_size);
306 CHECK_VAL("LEVEL_261", level_261, alloc_size, all_info, all_info, alloc_size);
307 CHECK_VAL("LEVEL_262", level_262, alloc_size, all_info, all_info, alloc_size);
309 CHECK_VAL("EA_SIZE", ea_size, ea_size, all_info, all_info, ea_size);
310 CHECK_VAL("FULL_DIRECTORY_INFO", full_directory_info, ea_size, all_info, all_info, ea_size);
311 CHECK_VAL("BOTH_DIRECTORY_INFO", both_directory_info, ea_size, all_info, all_info, ea_size);
312 CHECK_VAL("LEVEL_261", level_261, ea_size, all_info, all_info, ea_size);
313 CHECK_VAL("LEVEL_262", level_262, ea_size, all_info, all_info, ea_size);
315 CHECK_STR("SEARCH", search, name, alt_info, alt_name_info, fname);
316 CHECK_WSTR("BOTH_DIRECTORY_INFO", both_directory_info, short_name, alt_info, alt_name_info, fname, STR_UNICODE);
318 CHECK_NAME("STANDARD", standard, name, fname+1, 0);
319 CHECK_NAME("EA_SIZE", ea_size, name, fname+1, 0);
320 CHECK_NAME("DIRECTORY_INFO", directory_info, name, fname+1, STR_TERMINATE_ASCII);
321 CHECK_NAME("FULL_DIRECTORY_INFO", full_directory_info, name, fname+1, STR_TERMINATE_ASCII);
322 CHECK_NAME("NAME_INFO", name_info, name, fname+1, STR_TERMINATE_ASCII);
323 CHECK_NAME("BOTH_DIRECTORY_INFO", both_directory_info, name, fname+1, STR_TERMINATE_ASCII);
324 CHECK_NAME("LEVEL_261", level_261, name, fname+1, STR_TERMINATE_ASCII);
325 CHECK_NAME("LEVEL_262", level_262, name, fname+1, STR_TERMINATE_ASCII);
328 smb_raw_exit(cli->session);
329 cli_unlink(cli, fname);
335 struct multiple_result {
338 union smb_search_data *list;
342 callback function for multiple_search
344 static BOOL multiple_search_callback(void *private, union smb_search_data *file)
346 struct multiple_result *data = private;
350 data->list = talloc_realloc(data->mem_ctx,
352 data->count * (sizeof(data->list[0])));
354 data->list[data->count-1] = *file;
359 enum continue_type {CONT_FLAGS, CONT_NAME, CONT_RESUME_KEY};
362 do a single file (non-wildcard) search
364 static NTSTATUS multiple_search(struct cli_state *cli,
367 enum search_level level,
368 enum continue_type cont_type,
371 union smb_search_first io;
372 union smb_search_next io2;
374 const int per_search = 300;
375 struct multiple_result *result = data;
377 io.generic.level = level;
378 if (level == RAW_SEARCH_SEARCH) {
379 io.search_first.in.max_count = per_search;
380 io.search_first.in.search_attrib = 0;
381 io.search_first.in.pattern = pattern;
383 io.t2ffirst.in.search_attrib = 0;
384 io.t2ffirst.in.max_count = per_search;
385 io.t2ffirst.in.flags = 0;
386 io.t2ffirst.in.storage_type = 0;
387 io.t2ffirst.in.pattern = pattern;
388 if (cont_type == CONT_RESUME_KEY) {
389 io.t2ffirst.in.flags = FLAG_TRANS2_FIND_REQUIRE_RESUME |
390 FLAG_TRANS2_FIND_BACKUP_INTENT;
394 status = smb_raw_search_first(cli->tree, mem_ctx,
395 &io, data, multiple_search_callback);
398 while (NT_STATUS_IS_OK(status)) {
399 io2.generic.level = level;
400 if (level == RAW_SEARCH_SEARCH) {
401 io2.search_next.in.max_count = per_search;
402 io2.search_next.in.search_attrib = 0;
403 io2.search_next.in.search_id = result->list[result->count-1].search.search_id;
405 io2.t2fnext.in.handle = io.t2ffirst.out.handle;
406 io2.t2fnext.in.max_count = per_search;
407 io2.t2fnext.in.resume_key = 0;
408 io2.t2fnext.in.flags = 0;
409 io2.t2fnext.in.last_name = "";
411 case CONT_RESUME_KEY:
412 if (level == RAW_SEARCH_STANDARD) {
413 io2.t2fnext.in.resume_key =
414 result->list[result->count-1].standard.resume_key;
416 io2.t2fnext.in.resume_key =
417 result->list[result->count-1].both_directory_info.file_index;
419 io2.t2fnext.in.flags = FLAG_TRANS2_FIND_REQUIRE_RESUME |
420 FLAG_TRANS2_FIND_BACKUP_INTENT;
423 if (level == RAW_SEARCH_STANDARD) {
424 io2.t2fnext.in.last_name =
425 result->list[result->count-1].standard.name.s;
427 io2.t2fnext.in.last_name =
428 result->list[result->count-1].both_directory_info.name.s;
432 io2.t2fnext.in.flags = FLAG_TRANS2_FIND_CONTINUE;
437 status = smb_raw_search_next(cli->tree, mem_ctx,
438 &io2, data, multiple_search_callback);
439 if (!NT_STATUS_IS_OK(status)) {
442 if (level == RAW_SEARCH_SEARCH) {
443 if (io2.search_next.out.count == 0) {
446 } else if (io2.t2fnext.out.count == 0 ||
447 io2.t2fnext.out.end_of_search) {
455 #define CHECK_STATUS(status, correct) do { \
456 if (!NT_STATUS_EQUAL(status, correct)) { \
457 printf("(%d) Incorrect status %s - should be %s\n", \
458 __LINE__, nt_errstr(status), nt_errstr(correct)); \
463 #define CHECK_VALUE(v, correct) do { \
464 if ((v) != (correct)) { \
465 printf("(%d) Incorrect value %s=%d - should be %d\n", \
466 __LINE__, #v, v, correct); \
471 static int search_both_compare(union smb_search_data *d1, union smb_search_data *d2)
473 return strcmp(d1->both_directory_info.name.s, d2->both_directory_info.name.s);
476 static int search_standard_compare(union smb_search_data *d1, union smb_search_data *d2)
478 return strcmp(d1->standard.name.s, d2->standard.name.s);
481 static int search_old_compare(union smb_search_data *d1, union smb_search_data *d2)
483 return strcmp(d1->search.name, d2->search.name);
488 basic testing of search calls using many files
490 static BOOL test_many_files(struct cli_state *cli, TALLOC_CTX *mem_ctx)
492 const int num_files = 700;
497 struct multiple_result result;
500 const char *cont_name;
501 enum search_level level;
502 enum continue_type cont_type;
504 {"BOTH_DIRECTORY_INFO", "NAME", RAW_SEARCH_BOTH_DIRECTORY_INFO, CONT_NAME},
505 {"BOTH_DIRECTORY_INFO", "FLAGS", RAW_SEARCH_BOTH_DIRECTORY_INFO, CONT_FLAGS},
506 {"BOTH_DIRECTORY_INFO", "KEY", RAW_SEARCH_BOTH_DIRECTORY_INFO, CONT_RESUME_KEY},
507 {"STANDARD", "FLAGS", RAW_SEARCH_STANDARD, CONT_FLAGS},
508 {"STANDARD", "KEY", RAW_SEARCH_STANDARD, CONT_RESUME_KEY},
509 {"STANDARD", "NAME", RAW_SEARCH_STANDARD, CONT_NAME},
510 {"SEARCH", "ID", RAW_SEARCH_SEARCH, CONT_RESUME_KEY}
513 if (cli_deltree(cli, BASEDIR) == -1 ||
514 !cli_mkdir(cli, BASEDIR)) {
515 printf("Failed to create " BASEDIR " - %s\n", cli_errstr(cli));
519 printf("Creating %d files\n", num_files);
521 for (i=0;i<num_files;i++) {
522 asprintf(&fname, BASEDIR "\\test%03d.txt", i);
523 fnum = cli_open(cli, fname, O_CREAT|O_RDWR, DENY_NONE);
525 printf("Failed to create %s - %s\n", fname, cli_errstr(cli));
530 cli_close(cli, fnum);
534 for (t=0;t<ARRAY_SIZE(search_types);t++) {
536 result.mem_ctx = mem_ctx;
538 printf("Continue %s via %s\n", search_types[t].name, search_types[t].cont_name);
540 status = multiple_search(cli, mem_ctx, BASEDIR "\\*.*",
541 search_types[t].level,
542 search_types[t].cont_type,
545 CHECK_STATUS(status, NT_STATUS_OK);
546 CHECK_VALUE(result.count, num_files);
548 if (search_types[t].level == RAW_SEARCH_BOTH_DIRECTORY_INFO) {
549 qsort(result.list, result.count, sizeof(result.list[0]), search_both_compare);
550 } else if (search_types[t].level == RAW_SEARCH_STANDARD) {
551 qsort(result.list, result.count, sizeof(result.list[0]), search_standard_compare);
553 qsort(result.list, result.count, sizeof(result.list[0]), search_old_compare);
556 for (i=0;i<num_files;i++) {
558 if (search_types[t].level == RAW_SEARCH_BOTH_DIRECTORY_INFO) {
559 s = result.list[i].both_directory_info.name.s;
560 } else if (search_types[t].level == RAW_SEARCH_STANDARD) {
561 s = result.list[i].standard.name.s;
563 s = result.list[i].search.name;
565 asprintf(&fname, "test%03d.txt", i);
566 if (strcmp(fname, s)) {
567 printf("Incorrect name %s at entry %d\n", s, i);
576 smb_raw_exit(cli->session);
577 cli_deltree(cli, BASEDIR);
584 basic testing of all RAW_SEARCH_* calls using a single file
586 BOOL torture_raw_search(int dummy)
588 struct cli_state *cli;
592 if (!torture_open_connection(&cli)) {
596 mem_ctx = talloc_init("torture_search");
598 if (!test_one_file(cli, mem_ctx)) {
602 if (!test_many_files(cli, mem_ctx)) {
606 torture_close_connection(cli);
607 talloc_destroy(mem_ctx);