s3-debug Remove 'AllowDebugChange' and use lp_set_cmdline() instead
[sfrench/samba-autobuild/.git] / source3 / torture / masktest.c
1 /* 
2    Unix SMB/CIFS implementation.
3    mask_match tester
4    Copyright (C) Andrew Tridgell 1999
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
22 static fstring password;
23 static fstring username;
24 static int got_pass;
25 static int max_protocol = PROTOCOL_NT1;
26 static bool showall = False;
27 static bool old_list = False;
28 static const char *maskchars = "<>\"?*abc.";
29 static const char *filechars = "abcdefghijklm.";
30 static int verbose;
31 static int die_on_error;
32 static int NumLoops = 0;
33 static int ignore_dot_errors = 0;
34
35 extern char *optarg;
36 extern int optind;
37
38 /* a test fn for LANMAN mask support */
39 static int ms_fnmatch_lanman_core(const char *pattern, const char *string)
40 {
41         const char *p = pattern, *n = string;
42         char c;
43
44         if (strcmp(p,"?")==0 && strcmp(n,".")==0) goto match;
45
46         while ((c = *p++)) {
47                 switch (c) {
48                 case '.':
49                         /* if (! *n && ! *p) goto match; */
50                         if (*n != '.') goto nomatch;
51                         n++;
52                         break;
53
54                 case '?':
55                         if ((*n == '.' && n[1] != '.') || ! *n) goto next;
56                         n++;
57                         break;
58
59                 case '>':
60                         if (n[0] == '.') {
61                                 if (! n[1] && ms_fnmatch_lanman_core(p, n+1) == 0) goto match;
62                                 if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
63                                 goto nomatch;
64                         }
65                         if (! *n) goto next;
66                         n++;
67                         break;
68
69                 case '*':
70                         if (! *p) goto match;
71                         for (; *n; n++) {
72                                 if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
73                         }
74                         break;
75
76                 case '<':
77                         for (; *n; n++) {
78                                 if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
79                                 if (*n == '.' && !strchr_m(n+1,'.')) {
80                                         n++;
81                                         break;
82                                 }
83                         }
84                         break;
85
86                 case '"':
87                         if (*n == 0 && ms_fnmatch_lanman_core(p, n) == 0) goto match;
88                         if (*n != '.') goto nomatch;
89                         n++;
90                         break;
91
92                 default:
93                         if (c != *n) goto nomatch;
94                         n++;
95                 }
96         }
97
98         if (! *n) goto match;
99
100  nomatch:
101         if (verbose) printf("NOMATCH pattern=[%s] string=[%s]\n", pattern, string);
102         return -1;
103
104 next:
105         if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
106         goto nomatch;
107
108  match:
109         if (verbose) printf("MATCH   pattern=[%s] string=[%s]\n", pattern, string);
110         return 0;
111 }
112
113 static int ms_fnmatch_lanman(const char *pattern, const char *string)
114 {
115         if (!strpbrk(pattern, "?*<>\"")) {
116                 if (strcmp(string,"..") == 0) 
117                         string = ".";
118
119                 return strcmp(pattern, string);
120         }
121
122         if (strcmp(string,"..") == 0 || strcmp(string,".") == 0) {
123                 return ms_fnmatch_lanman_core(pattern, "..") &&
124                         ms_fnmatch_lanman_core(pattern, ".");
125         }
126
127         return ms_fnmatch_lanman_core(pattern, string);
128 }
129
130 static bool reg_match_one(struct cli_state *cli, const char *pattern, const char *file)
131 {
132         /* oh what a weird world this is */
133         if (old_list && strcmp(pattern, "*.*") == 0) return True;
134
135         if (strcmp(pattern,".") == 0) return False;
136
137         if (max_protocol <= PROTOCOL_LANMAN2) {
138                 return ms_fnmatch_lanman(pattern, file)==0;
139         }
140
141         if (strcmp(file,"..") == 0) file = ".";
142
143         return ms_fnmatch(pattern, file, cli->protocol, False) == 0;
144 }
145
146 static char *reg_test(struct cli_state *cli, const char *pattern, const char *long_name, const char *short_name)
147 {
148         static fstring ret;
149         const char *new_pattern = 1+strrchr_m(pattern,'\\');
150
151         fstrcpy(ret, "---");
152         if (reg_match_one(cli, new_pattern, ".")) ret[0] = '+';
153         if (reg_match_one(cli, new_pattern, "..")) ret[1] = '+';
154         if (reg_match_one(cli, new_pattern, long_name) ||
155             (*short_name && reg_match_one(cli, new_pattern, short_name))) ret[2] = '+';
156         return ret;
157 }
158
159
160 /***************************************************** 
161 return a connection to a server
162 *******************************************************/
163 static struct cli_state *connect_one(char *share)
164 {
165         struct cli_state *c;
166         struct nmb_name called, calling;
167         char *server_n;
168         char *server;
169         struct sockaddr_storage ss;
170         NTSTATUS status;
171
172         server = share+2;
173         share = strchr_m(server,'\\');
174         if (!share) return NULL;
175         *share = 0;
176         share++;
177
178         server_n = server;
179
180         zero_sockaddr(&ss);
181
182         make_nmb_name(&calling, "masktest", 0x0);
183         make_nmb_name(&called , server, 0x20);
184
185  again:
186         zero_sockaddr(&ss);
187
188         /* have to open a new connection */
189         if (!(c=cli_initialise())) {
190                 DEBUG(0,("Connection to %s failed\n", server_n));
191                 return NULL;
192         }
193
194         status = cli_connect(c, server_n, &ss);
195         if (!NT_STATUS_IS_OK(status)) {
196                 DEBUG(0,("Connection to %s failed. Error %s\n", server_n, nt_errstr(status) ));
197                 return NULL;
198         }
199
200         c->protocol = max_protocol;
201
202         if (!cli_session_request(c, &calling, &called)) {
203                 DEBUG(0,("session request to %s failed\n", called.name));
204                 cli_shutdown(c);
205                 if (strcmp(called.name, "*SMBSERVER")) {
206                         make_nmb_name(&called , "*SMBSERVER", 0x20);
207                         goto again;
208                 }
209                 return NULL;
210         }
211
212         DEBUG(4,(" session request ok\n"));
213
214         status = cli_negprot(c);
215         if (!NT_STATUS_IS_OK(status)) {
216                 DEBUG(0, ("protocol negotiation failed: %s\n",
217                           nt_errstr(status)));
218                 cli_shutdown(c);
219                 return NULL;
220         }
221
222         if (!got_pass) {
223                 char *pass = getpass("Password: ");
224                 if (pass) {
225                         fstrcpy(password, pass);
226                 }
227         }
228
229         if (!NT_STATUS_IS_OK(cli_session_setup(c, username, 
230                                                password, strlen(password),
231                                                password, strlen(password),
232                                                lp_workgroup()))) {
233                 DEBUG(0,("session setup failed: %s\n", cli_errstr(c)));
234                 return NULL;
235         }
236
237         /*
238          * These next two lines are needed to emulate
239          * old client behaviour for people who have
240          * scripts based on client output.
241          * QUESTION ? Do we want to have a 'client compatibility
242          * mode to turn these on/off ? JRA.
243          */
244
245         if (*c->server_domain || *c->server_os || *c->server_type)
246                 DEBUG(1,("Domain=[%s] OS=[%s] Server=[%s]\n",
247                         c->server_domain,c->server_os,c->server_type));
248
249         DEBUG(4,(" session setup ok\n"));
250
251         status = cli_tcon_andx(c, share, "?????", password,
252                                strlen(password)+1);
253         if (!NT_STATUS_IS_OK(status)) {
254                 DEBUG(0,("tree connect failed: %s\n", nt_errstr(status)));
255                 cli_shutdown(c);
256                 return NULL;
257         }
258
259         DEBUG(4,(" tconx ok\n"));
260
261         return c;
262 }
263
264 static char *resultp;
265
266 struct rn_state {
267         char **pp_long_name;
268         char *short_name;
269 };
270
271 static NTSTATUS listfn(const char *mnt, struct file_info *f, const char *s,
272                    void *private_data)
273 {
274         struct rn_state *state = (struct rn_state *)private_data;
275         if (strcmp(f->name,".") == 0) {
276                 resultp[0] = '+';
277         } else if (strcmp(f->name,"..") == 0) {
278                 resultp[1] = '+';
279         } else {
280                 resultp[2] = '+';
281         }
282
283         if (state == NULL) {
284                 return NT_STATUS_INTERNAL_ERROR;
285         }
286
287         if (ISDOT(f->name) || ISDOTDOT(f->name))  {
288                 return NT_STATUS_OK;
289         }
290
291         fstrcpy(state->short_name, f->short_name);
292         strlower_m(state->short_name);
293         *state->pp_long_name = SMB_STRDUP(f->name);
294         if (!*state->pp_long_name) {
295                 return NT_STATUS_NO_MEMORY;
296         }
297         strlower_m(*state->pp_long_name);
298         return NT_STATUS_OK;
299 }
300
301 static void get_real_name(struct cli_state *cli,
302                           char **pp_long_name, fstring short_name)
303 {
304         struct rn_state state;
305
306         state.pp_long_name = pp_long_name;
307         state.short_name = short_name;
308
309         *pp_long_name = NULL;
310         /* nasty hack to force level 260 listings - tridge */
311         if (max_protocol <= PROTOCOL_LANMAN1) {
312                 cli_list_trans(cli, "\\masktest\\*.*", aHIDDEN | aDIR,
313                                SMB_FIND_FILE_BOTH_DIRECTORY_INFO, listfn,
314                                &state);
315         } else {
316                 cli_list_trans(cli, "\\masktest\\*", aHIDDEN | aDIR,
317                                SMB_FIND_FILE_BOTH_DIRECTORY_INFO,
318                                listfn, &state);
319         }
320
321         if (*short_name == 0) {
322                 fstrcpy(short_name, *pp_long_name);
323         }
324
325 #if 0
326         if (!strchr_m(short_name,'.')) {
327                 fstrcat(short_name,".");
328         }
329 #endif
330 }
331
332 static void testpair(struct cli_state *cli, const char *mask, const char *file)
333 {
334         uint16_t fnum;
335         fstring res1;
336         char *res2;
337         static int count;
338         fstring short_name;
339         char *long_name = NULL;
340
341         count++;
342
343         fstrcpy(res1, "---");
344
345         if (!NT_STATUS_IS_OK(cli_open(cli, file, O_CREAT|O_TRUNC|O_RDWR, 0, &fnum))) {
346                 DEBUG(0,("Can't create %s\n", file));
347                 return;
348         }
349         cli_close(cli, fnum);
350
351         resultp = res1;
352         fstrcpy(short_name, "");
353         get_real_name(cli, &long_name, short_name);
354         if (!long_name) {
355                 return;
356         }
357         fstrcpy(res1, "---");
358         cli_list(cli, mask, aHIDDEN | aDIR, listfn, NULL);
359
360         res2 = reg_test(cli, mask, long_name, short_name);
361
362         if (showall ||
363             ((strcmp(res1, res2) && !ignore_dot_errors) ||
364              (strcmp(res1+2, res2+2) && ignore_dot_errors))) {
365                 DEBUG(0,("%s %s %d mask=[%s] file=[%s] rfile=[%s/%s]\n",
366                          res1, res2, count, mask, file, long_name, short_name));
367                 if (die_on_error) exit(1);
368         }
369
370         cli_unlink(cli, file, aSYSTEM | aHIDDEN);
371
372         if (count % 100 == 0) DEBUG(0,("%d\n", count));
373         SAFE_FREE(long_name);
374 }
375
376 static void test_mask(int argc, char *argv[],
377                       struct cli_state *cli)
378 {
379         char *mask, *file;
380         int l1, l2, i, l;
381         int mc_len = strlen(maskchars);
382         int fc_len = strlen(filechars);
383         TALLOC_CTX *ctx = talloc_tos();
384
385         cli_mkdir(cli, "\\masktest");
386
387         cli_unlink(cli, "\\masktest\\*", aSYSTEM | aHIDDEN);
388
389         if (argc >= 2) {
390                 while (argc >= 2) {
391                         mask = talloc_asprintf(ctx,
392                                         "\\masktest\\%s",
393                                         argv[0]);
394                         file = talloc_asprintf(ctx,
395                                         "\\masktest\\%s",
396                                         argv[1]);
397                         if (!mask || !file) {
398                                 goto finished;
399                         }
400                         testpair(cli, mask, file);
401                         argv += 2;
402                         argc -= 2;
403                 }
404                 goto finished;
405         }
406
407         while (1) {
408                 l1 = 1 + random() % 20;
409                 l2 = 1 + random() % 20;
410                 mask = TALLOC_ARRAY(ctx, char, strlen("\\masktest\\")+1+22);
411                 file = TALLOC_ARRAY(ctx, char, strlen("\\masktest\\")+1+22);
412                 if (!mask || !file) {
413                         goto finished;
414                 }
415                 memcpy(mask,"\\masktest\\",strlen("\\masktest\\")+1);
416                 memcpy(file,"\\masktest\\",strlen("\\masktest\\")+1);
417                 l = strlen(mask);
418                 for (i=0;i<l1;i++) {
419                         mask[i+l] = maskchars[random() % mc_len];
420                 }
421                 mask[l+l1] = 0;
422
423                 for (i=0;i<l2;i++) {
424                         file[i+l] = filechars[random() % fc_len];
425                 }
426                 file[l+l2] = 0;
427
428                 if (strcmp(file+l,".") == 0 || 
429                     strcmp(file+l,"..") == 0 ||
430                     strcmp(mask+l,"..") == 0) continue;
431
432                 if (strspn(file+l, ".") == strlen(file+l)) continue;
433
434                 testpair(cli, mask, file);
435                 if (NumLoops && (--NumLoops == 0))
436                         break;
437                 TALLOC_FREE(mask);
438                 TALLOC_FREE(file);
439         }
440
441  finished:
442         cli_rmdir(cli, "\\masktest");
443 }
444
445
446 static void usage(void)
447 {
448         printf(
449 "Usage:\n\
450   masktest //server/share [options..]\n\
451   options:\n\
452         -d debuglevel\n\
453         -n numloops\n\
454         -W workgroup\n\
455         -U user%%pass\n\
456         -s seed\n\
457         -M max protocol\n\
458         -f filechars (default %s)\n\
459         -m maskchars (default %s)\n\
460         -v                             verbose mode\n\
461         -E                             die on error\n\
462         -a                             show all tests\n\
463         -i                             ignore . and .. errors\n\
464 \n\
465   This program tests wildcard matching between two servers. It generates\n\
466   random pairs of filenames/masks and tests that they match in the same\n\
467   way on the servers and internally\n\
468 ", 
469   filechars, maskchars);
470 }
471
472 /****************************************************************************
473   main program
474 ****************************************************************************/
475  int main(int argc,char *argv[])
476 {
477         char *share;
478         struct cli_state *cli;
479         int opt;
480         char *p;
481         int seed;
482         TALLOC_CTX *frame = talloc_stackframe();
483
484         setlinebuf(stdout);
485
486         lp_set_cmdline("log level", "0");
487
488         if (argc < 2 || argv[1][0] == '-') {
489                 usage();
490                 exit(1);
491         }
492
493         share = argv[1];
494
495         all_string_sub(share,"/","\\",0);
496
497         setup_logging(argv[0], DEBUG_STDERR);
498
499         argc -= 1;
500         argv += 1;
501
502         load_case_tables();
503         lp_load(get_dyn_CONFIGFILE(),True,False,False,True);
504         load_interfaces();
505
506         if (getenv("USER")) {
507                 fstrcpy(username,getenv("USER"));
508         }
509
510         seed = time(NULL);
511
512         while ((opt = getopt(argc, argv, "n:d:U:s:hm:f:aoW:M:vEi")) != EOF) {
513                 switch (opt) {
514                 case 'n':
515                         NumLoops = atoi(optarg);
516                         break;
517                 case 'd':
518                         DEBUGLEVEL = atoi(optarg);
519                         break;
520                 case 'E':
521                         die_on_error = 1;
522                         break;
523                 case 'i':
524                         ignore_dot_errors = 1;
525                         break;
526                 case 'v':
527                         verbose++;
528                         break;
529                 case 'M':
530                         max_protocol = interpret_protocol(optarg, max_protocol);
531                         break;
532                 case 'U':
533                         fstrcpy(username,optarg);
534                         p = strchr_m(username,'%');
535                         if (p) {
536                                 *p = 0;
537                                 fstrcpy(password, p+1);
538                                 got_pass = 1;
539                         }
540                         break;
541                 case 's':
542                         seed = atoi(optarg);
543                         break;
544                 case 'h':
545                         usage();
546                         exit(1);
547                 case 'm':
548                         maskchars = optarg;
549                         break;
550                 case 'f':
551                         filechars = optarg;
552                         break;
553                 case 'a':
554                         showall = 1;
555                         break;
556                 case 'o':
557                         old_list = True;
558                         break;
559                 default:
560                         printf("Unknown option %c (%d)\n", (char)opt, opt);
561                         exit(1);
562                 }
563         }
564
565         argc -= optind;
566         argv += optind;
567
568
569         cli = connect_one(share);
570         if (!cli) {
571                 DEBUG(0,("Failed to connect to %s\n", share));
572                 exit(1);
573         }
574
575         /* need to init seed after connect as clientgen uses random numbers */
576         DEBUG(0,("seed=%d\n", seed));
577         srandom(seed);
578
579         test_mask(argc, argv, cli);
580
581         TALLOC_FREE(frame);
582         return(0);
583 }