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