replaced stdio in many parts of samba with a XFILE. XFILE is a cut-down
[tprouty/samba.git] / source / torture / masktest.c
1 /* 
2    Unix SMB/Netbios implementation.
3    Version 2.0
4    mask_match tester
5    Copyright (C) Andrew Tridgell 1999
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #define NO_SYSLOG
23
24 #include "includes.h"
25
26 extern int DEBUGLEVEL;
27 static fstring password;
28 static fstring username;
29 static int got_pass;
30 static int max_protocol = PROTOCOL_NT1;
31 static BOOL showall = False;
32 static BOOL old_list = False;
33 static char *maskchars = "<>\"?*abc.";
34 static char *filechars = "abcdefghijklm.";
35 static int verbose;
36 static int die_on_error;
37
38 /* a test fn for LANMAN mask support */
39 int ms_fnmatch_lanman_core(char *pattern, char *string)
40 {
41         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 int ms_fnmatch_lanman(char *pattern, char *string)
114 {
115         if (!strpbrk(pattern, "?*<>\"")) {
116                 if (strcmp(string,"..") == 0) string = ".";
117                 return strcmp(pattern, string);
118         }
119
120         if (strcmp(string,"..") == 0 || strcmp(string,".") == 0) {
121                 return ms_fnmatch_lanman_core(pattern, "..") &&
122                         ms_fnmatch_lanman_core(pattern, ".");
123         }
124
125         return ms_fnmatch_lanman_core(pattern, string);
126 }
127
128 static BOOL reg_match_one(struct cli_state *cli, char *pattern, char *file)
129 {
130         /* oh what a weird world this is */
131         if (old_list && strcmp(pattern, "*.*") == 0) return True;
132
133         if (strcmp(pattern,".") == 0) return False;
134
135         if (max_protocol <= PROTOCOL_LANMAN2) {
136                 return ms_fnmatch_lanman(pattern, file)==0;
137         }
138
139         if (strcmp(file,"..") == 0) file = ".";
140
141         return ms_fnmatch(pattern, file, cli->protocol)==0;
142 }
143
144 static char *reg_test(struct cli_state *cli, char *pattern, char *long_name, char *short_name)
145 {
146         static fstring ret;
147         fstrcpy(ret, "---");
148
149         pattern = 1+strrchr_m(pattern,'\\');
150
151         if (reg_match_one(cli, pattern, ".")) ret[0] = '+';
152         if (reg_match_one(cli, pattern, "..")) ret[1] = '+';
153         if (reg_match_one(cli, pattern, long_name) || 
154             (*short_name && reg_match_one(cli, pattern, short_name))) ret[2] = '+';
155         return ret;
156 }
157
158
159 /***************************************************** 
160 return a connection to a server
161 *******************************************************/
162 struct cli_state *connect_one(char *share)
163 {
164         struct cli_state *c;
165         struct nmb_name called, calling;
166         char *server_n;
167         char *server;
168         struct in_addr ip;
169         extern struct in_addr ipzero;
170
171         server = share+2;
172         share = strchr_m(server,'\\');
173         if (!share) return NULL;
174         *share = 0;
175         share++;
176
177         server_n = server;
178         
179         ip = ipzero;
180
181         make_nmb_name(&calling, "masktest", 0x0);
182         make_nmb_name(&called , server, 0x20);
183
184  again:
185         ip = ipzero;
186
187         /* have to open a new connection */
188         if (!(c=cli_initialise(NULL)) || !cli_connect(c, server_n, &ip)) {
189                 DEBUG(0,("Connection to %s failed\n", server_n));
190                 return NULL;
191         }
192
193         c->protocol = max_protocol;
194
195         if (!cli_session_request(c, &calling, &called)) {
196                 DEBUG(0,("session request to %s failed\n", called.name));
197                 cli_shutdown(c);
198                 if (strcmp(called.name, "*SMBSERVER")) {
199                         make_nmb_name(&called , "*SMBSERVER", 0x20);
200                         goto again;
201                 }
202                 return NULL;
203         }
204
205         DEBUG(4,(" session request ok\n"));
206
207         if (!cli_negprot(c)) {
208                 DEBUG(0,("protocol negotiation failed\n"));
209                 cli_shutdown(c);
210                 return NULL;
211         }
212
213         if (!got_pass) {
214                 char *pass = getpass("Password: ");
215                 if (pass) {
216                         pstrcpy(password, pass);
217                 }
218         }
219
220         if (!cli_session_setup(c, username, 
221                                password, strlen(password),
222                                password, strlen(password),
223                                lp_workgroup())) {
224                 DEBUG(0,("session setup failed: %s\n", cli_errstr(c)));
225                 return NULL;
226         }
227
228         /*
229          * These next two lines are needed to emulate
230          * old client behaviour for people who have
231          * scripts based on client output.
232          * QUESTION ? Do we want to have a 'client compatibility
233          * mode to turn these on/off ? JRA.
234          */
235
236         if (*c->server_domain || *c->server_os || *c->server_type)
237                 DEBUG(1,("Domain=[%s] OS=[%s] Server=[%s]\n",
238                         c->server_domain,c->server_os,c->server_type));
239         
240         DEBUG(4,(" session setup ok\n"));
241
242         if (!cli_send_tconX(c, share, "?????",
243                             password, strlen(password)+1)) {
244                 DEBUG(0,("tree connect failed: %s\n", cli_errstr(c)));
245                 cli_shutdown(c);
246                 return NULL;
247         }
248
249         DEBUG(4,(" tconx ok\n"));
250
251         return c;
252 }
253
254 static char *resultp;
255 static file_info *f_info;
256
257 void listfn(file_info *f, const char *s, void *state)
258 {
259         if (strcmp(f->name,".") == 0) {
260                 resultp[0] = '+';
261         } else if (strcmp(f->name,"..") == 0) {
262                 resultp[1] = '+';               
263         } else {
264                 resultp[2] = '+';
265         }
266         f_info = f;
267 }
268
269 static void get_real_name(struct cli_state *cli, 
270                           pstring long_name, fstring short_name)
271 {
272         /* nasty hack to force level 260 listings - tridge */
273         cli->capabilities |= CAP_NT_SMBS;
274         if (max_protocol <= PROTOCOL_LANMAN1) {
275                 cli_list_new(cli, "\\masktest\\*.*", aHIDDEN | aDIR, listfn, NULL);
276         } else {
277                 cli_list_new(cli, "\\masktest\\*", aHIDDEN | aDIR, listfn, NULL);
278         }
279         if (f_info) {
280                 fstrcpy(short_name, f_info->short_name);
281                 strlower(short_name);
282                 pstrcpy(long_name, f_info->name);
283                 strlower(long_name);
284         }
285
286         if (*short_name == 0) {
287                 fstrcpy(short_name, long_name);
288         }
289
290 #if 0
291         if (!strchr_m(short_name,'.')) {
292                 fstrcat(short_name,".");
293         }
294 #endif
295 }
296
297 static void testpair(struct cli_state *cli, char *mask, char *file)
298 {
299         int fnum;
300         fstring res1;
301         char *res2;
302         static int count;
303         fstring short_name;
304         pstring long_name;
305
306         count++;
307
308         fstrcpy(res1, "---");
309
310         fnum = cli_open(cli, file, O_CREAT|O_TRUNC|O_RDWR, 0);
311         if (fnum == -1) {
312                 DEBUG(0,("Can't create %s\n", file));
313                 return;
314         }
315         cli_close(cli, fnum);
316
317         resultp = res1;
318         fstrcpy(short_name, "");
319         f_info = NULL;
320         get_real_name(cli, long_name, short_name);
321         f_info = NULL;
322         fstrcpy(res1, "---");
323         cli_list(cli, mask, aHIDDEN | aDIR, listfn, NULL);
324
325         res2 = reg_test(cli, mask, long_name, short_name);
326
327         if (showall || strcmp(res1, res2)) {
328                 DEBUG(0,("%s %s %d mask=[%s] file=[%s] rfile=[%s/%s]\n",
329                          res1, res2, count, mask, file, long_name, short_name));
330                 if (die_on_error) exit(1);
331         }
332
333         cli_unlink(cli, file);
334
335         if (count % 100 == 0) DEBUG(0,("%d\n", count));
336 }
337
338 static void test_mask(int argc, char *argv[], 
339                       struct cli_state *cli)
340 {
341         pstring mask, file;
342         int l1, l2, i, l;
343         int mc_len = strlen(maskchars);
344         int fc_len = strlen(filechars);
345
346         cli_mkdir(cli, "\\masktest");
347
348         cli_unlink(cli, "\\masktest\\*");
349
350         if (argc >= 2) {
351                 while (argc >= 2) {
352                         pstrcpy(mask,"\\masktest\\");
353                         pstrcpy(file,"\\masktest\\");
354                         pstrcat(mask, argv[0]);
355                         pstrcat(file, argv[1]);
356                         testpair(cli, mask, file);
357                         argv += 2;
358                         argc -= 2;
359                 }
360                 goto finished;
361         }
362
363         while (1) {
364                 l1 = 1 + random() % 20;
365                 l2 = 1 + random() % 20;
366                 pstrcpy(mask,"\\masktest\\");
367                 pstrcpy(file,"\\masktest\\");
368                 l = strlen(mask);
369                 for (i=0;i<l1;i++) {
370                         mask[i+l] = maskchars[random() % mc_len];
371                 }
372                 mask[l+l1] = 0;
373
374                 for (i=0;i<l2;i++) {
375                         file[i+l] = filechars[random() % fc_len];
376                 }
377                 file[l+l2] = 0;
378
379                 if (strcmp(file+l,".") == 0 || 
380                     strcmp(file+l,"..") == 0 ||
381                     strcmp(mask+l,"..") == 0) continue;
382
383                 if (strspn(file+l, ".") == strlen(file+l)) continue;
384
385                 testpair(cli, mask, file);
386         }
387
388  finished:
389         cli_rmdir(cli, "\\masktest");
390 }
391
392
393 static void usage(void)
394 {
395         printf(
396 "Usage:\n\
397   masktest //server/share [options..]\n\
398   options:\n\
399         -W workgroup\n\
400         -U user%%pass\n\
401         -s seed\n\
402         -M max protocol\n\
403         -f filechars (default %s)\n\
404         -m maskchars (default %s)\n\
405         -a                             show all tests\n\
406 \n\
407   This program tests wildcard matching between two servers. It generates\n\
408   random pairs of filenames/masks and tests that they match in the same\n\
409   way on the servers and internally\n\
410 ", 
411   filechars, maskchars);
412 }
413
414 /****************************************************************************
415   main program
416 ****************************************************************************/
417  int main(int argc,char *argv[])
418 {
419         char *share;
420         struct cli_state *cli;  
421         extern char *optarg;
422         extern int optind;
423         extern XFILE *dbf;
424         int opt;
425         char *p;
426         int seed;
427         static pstring servicesf = CONFIGFILE;
428
429         setlinebuf(stdout);
430
431         dbf = x_stderr;
432
433         if (argv[1][0] == '-' || argc < 2) {
434                 usage();
435                 exit(1);
436         }
437
438         share = argv[1];
439
440         all_string_sub(share,"/","\\",0);
441
442         setup_logging(argv[0],True);
443
444         argc -= 1;
445         argv += 1;
446
447         TimeInit();
448
449         lp_load(servicesf,True,False,False);
450         load_interfaces();
451
452         if (getenv("USER")) {
453                 pstrcpy(username,getenv("USER"));
454         }
455
456         seed = time(NULL);
457
458         while ((opt = getopt(argc, argv, "U:s:hm:f:aoW:M:vE")) != EOF) {
459                 switch (opt) {
460                 case 'E':
461                         die_on_error = 1;
462                         break;
463                 case 'v':
464                         verbose++;
465                         break;
466                 case 'M':
467                         max_protocol = interpret_protocol(optarg, max_protocol);
468                         break;
469                 case 'U':
470                         pstrcpy(username,optarg);
471                         p = strchr_m(username,'%');
472                         if (p) {
473                                 *p = 0;
474                                 pstrcpy(password, p+1);
475                                 got_pass = 1;
476                         }
477                         break;
478                 case 's':
479                         seed = atoi(optarg);
480                         break;
481                 case 'h':
482                         usage();
483                         exit(1);
484                 case 'm':
485                         maskchars = optarg;
486                         break;
487                 case 'f':
488                         filechars = optarg;
489                         break;
490                 case 'a':
491                         showall = 1;
492                         break;
493                 case 'o':
494                         old_list = True;
495                         break;
496                 default:
497                         printf("Unknown option %c (%d)\n", (char)opt, opt);
498                         exit(1);
499                 }
500         }
501
502         argc -= optind;
503         argv += optind;
504
505
506         cli = connect_one(share);
507         if (!cli) {
508                 DEBUG(0,("Failed to connect to %s\n", share));
509                 exit(1);
510         }
511
512         /* need to init seed after connect as clientgen uses random numbers */
513         DEBUG(0,("seed=%d\n", seed));
514         srandom(seed);
515
516         test_mask(argc, argv, cli);
517
518         return(0);
519 }