c88b4f704bc94cf071f579b7b5dcddca01d95bf3
[jelmer/samba4-debian.git] / source / 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 #include "system/filesys.h"
22 #include "system/dir.h"
23 #include "libcli/libcli.h"
24 #include "libcli/raw/libcliraw.h"
25 #include "system/time.h"
26 #include "pstring.h"
27 #include "auth/credentials/credentials.h"
28 #include "auth/gensec/gensec.h"
29 #include "param/param.h"
30 #include "dynconfig.h"
31
32 static struct cli_credentials *credentials;
33 static bool showall = false;
34 static bool old_list = false;
35 static const char *maskchars = "<>\"?*abc.";
36 static const char *filechars = "abcdefghijklm.";
37 static int verbose;
38 static int die_on_error;
39 static int NumLoops = 0;
40 static int max_length = 20;
41 struct masktest_state {
42         TALLOC_CTX *mem_ctx;
43 };
44
45 static bool reg_match_one(struct smbcli_state *cli, const char *pattern, const char *file)
46 {
47         /* oh what a weird world this is */
48         if (old_list && strcmp(pattern, "*.*") == 0) return true;
49
50         if (ISDOT(pattern)) return false;
51
52         if (ISDOTDOT(file)) file = ".";
53
54         return ms_fnmatch(pattern, file, cli->transport->negotiate.protocol)==0;
55 }
56
57 static char *reg_test(struct smbcli_state *cli, char *pattern, char *long_name, char *short_name)
58 {
59         static fstring ret;
60         fstrcpy(ret, "---");
61
62         pattern = 1+strrchr_m(pattern,'\\');
63
64         if (reg_match_one(cli, pattern, ".")) ret[0] = '+';
65         if (reg_match_one(cli, pattern, "..")) ret[1] = '+';
66         if (reg_match_one(cli, pattern, long_name) || 
67             (*short_name && reg_match_one(cli, pattern, short_name))) ret[2] = '+';
68         return ret;
69 }
70
71
72 /***************************************************** 
73 return a connection to a server
74 *******************************************************/
75 static struct smbcli_state *connect_one(char *share, const char **ports)
76 {
77         struct smbcli_state *c;
78         fstring server;
79         NTSTATUS status;
80
81         fstrcpy(server,share+2);
82         share = strchr_m(server,'\\');
83         if (!share) return NULL;
84         *share = 0;
85         share++;
86
87         cli_credentials_set_workstation(credentials, "masktest", CRED_SPECIFIED);
88
89         status = smbcli_full_connection(NULL, &c,
90                                         server, 
91                                         ports,
92                                         share, NULL,
93                                         credentials, NULL);
94
95         if (!NT_STATUS_IS_OK(status)) {
96                 return NULL;
97         }
98
99         return c;
100 }
101
102 static char *resultp;
103 static struct {
104         char *long_name;
105         char *short_name;
106 } last_hit;
107 static bool f_info_hit;
108
109 static void listfn(struct clilist_file_info *f, const char *s, void *state)
110 {
111         struct masktest_state *m = talloc_get_type(state,struct masktest_state);
112
113         if (ISDOT(f->name)) {
114                 resultp[0] = '+';
115         } else if (ISDOTDOT(f->name)) {
116                 resultp[1] = '+';
117         } else {
118                 resultp[2] = '+';
119         }
120
121         last_hit.long_name = talloc_strdup(m->mem_ctx, f->name);
122         last_hit.short_name = talloc_strdup(m->mem_ctx, f->short_name);
123         f_info_hit = true;
124 }
125
126 static void get_real_name(TALLOC_CTX *mem_ctx, struct smbcli_state *cli,
127                           char *long_name, fstring short_name)
128 {
129         const char *mask;
130         struct masktest_state state;
131
132         if (cli->transport->negotiate.protocol <= PROTOCOL_LANMAN1) {
133                 mask = "\\masktest\\*.*";
134         } else {
135                 mask = "\\masktest\\*";
136         }
137
138         f_info_hit = false;
139
140         state.mem_ctx = mem_ctx;
141
142         smbcli_list_new(cli->tree, mask,
143                         FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,
144                         RAW_SEARCH_DATA_BOTH_DIRECTORY_INFO,
145                         listfn, &state);
146
147         if (f_info_hit) {
148                 fstrcpy(short_name, last_hit.short_name);
149                 strlower(short_name);
150                 long_name = talloc_strdup(mem_ctx, last_hit.long_name);
151                 strlower(long_name);
152         }
153
154         if (*short_name == '\0') {
155                 fstrcpy(short_name, long_name);
156         }
157 }
158
159 static void testpair(TALLOC_CTX *mem_ctx, struct smbcli_state *cli, char *mask,
160                 char *file)
161 {
162         int fnum;
163         fstring res1;
164         char *res2;
165         static int count;
166         fstring short_name;
167         char *long_name;
168         struct masktest_state state;
169
170         count++;
171
172         fstrcpy(res1, "---");
173
174         state.mem_ctx = mem_ctx;
175
176         fnum = smbcli_open(cli->tree, file, O_CREAT|O_TRUNC|O_RDWR, 0);
177         if (fnum == -1) {
178                 DEBUG(0,("Can't create %s\n", file));
179                 return;
180         }
181         smbcli_close(cli->tree, fnum);
182
183         resultp = res1;
184         fstrcpy(short_name, "");
185         get_real_name(mem_ctx, cli, long_name, short_name);
186         fstrcpy(res1, "---");
187         smbcli_list_new(cli->tree, mask,
188                         FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,
189                         RAW_SEARCH_DATA_BOTH_DIRECTORY_INFO,
190                         listfn, &state);
191
192         res2 = reg_test(cli, mask, long_name, short_name);
193
194         if (showall || strcmp(res1, res2)) {
195                 d_printf("%s %s %d mask=[%s] file=[%s] rfile=[%s/%s]\n",
196                          res1, res2, count, mask, file, long_name, short_name);
197                 if (die_on_error) exit(1);
198         }
199
200         smbcli_unlink(cli->tree, file);
201
202         if (count % 100 == 0) DEBUG(0,("%d\n", count));
203
204         resultp = NULL;
205 }
206
207 static void test_mask(int argc, char *argv[],
208                       struct smbcli_state *cli)
209 {
210         TALLOC_CTX *mem_ctx;
211         char *mask, *file;
212         int l1, l2, i, l;
213         int mc_len = strlen(maskchars);
214         int fc_len = strlen(filechars);
215
216         mem_ctx = talloc_init("test_mask");
217
218         smbcli_mkdir(cli->tree, "\\masktest");
219
220         smbcli_unlink(cli->tree, "\\masktest\\*");
221
222         if (argc >= 2) {
223                 while (argc >= 2) {
224                         mask = talloc_strdup(mem_ctx, "\\masktest\\");
225                         file = talloc_strdup(mem_ctx, "\\masktest\\");
226                         mask = talloc_strdup_append(mask, argv[0]);
227                         file = talloc_strdup_append(file, argv[1]);
228                         testpair(mem_ctx, cli, mask, file);
229                         argv += 2;
230                         argc -= 2;
231                 }
232                 goto finished;
233         }
234
235         while (1) {
236                 l1 = 1 + random() % max_length;
237                 l2 = 1 + random() % max_length;
238                 mask = talloc_strdup(mem_ctx, "\\masktest\\");
239                 file = talloc_strdup(mem_ctx, "\\masktest\\");
240                 l = strlen(mask);
241                 for (i=0;i<l1;i++) {
242                         mask[i+l] = maskchars[random() % mc_len];
243                 }
244                 mask[l+l1] = 0;
245
246                 for (i=0;i<l2;i++) {
247                         file[i+l] = filechars[random() % fc_len];
248                 }
249                 file[l+l2] = 0;
250
251                 if (ISDOT(file+l) || ISDOTDOT(file+l) || ISDOTDOT(mask+l)) {
252                         continue;
253                 }
254
255                 if (strspn(file+l, ".") == strlen(file+l)) continue;
256
257                 testpair(mem_ctx, cli, mask, file);
258                 if (NumLoops && (--NumLoops == 0))
259                         break;
260         }
261
262  finished:
263         smbcli_rmdir(cli->tree, "\\masktest");
264         talloc_free(mem_ctx);
265 }
266
267
268 static void usage(void)
269 {
270         printf(
271 "Usage:\n\
272   masktest //server/share [options..]\n\
273   options:\n\
274         -d debuglevel\n\
275         -n numloops\n\
276         -W workgroup\n\
277         -U user%%pass\n\
278         -s seed\n\
279         -l max test length\n\
280         -M max protocol\n\
281         -f filechars (default %s)\n\
282         -m maskchars (default %s)\n\
283         -v                             verbose mode\n\
284         -E                             die on error\n\
285         -a                             show all tests\n\
286 \n\
287   This program tests wildcard matching between two servers. It generates\n\
288   random pairs of filenames/masks and tests that they match in the same\n\
289   way on the servers and internally\n\
290 ", 
291   filechars, maskchars);
292 }
293
294 /****************************************************************************
295   main program
296 ****************************************************************************/
297  int main(int argc,char *argv[])
298 {
299         char *share;
300         struct smbcli_state *cli;       
301         int opt;
302         int seed;
303         struct loadparm_context *lp_ctx;
304
305         setlinebuf(stdout);
306
307         setup_logging("masktest", DEBUG_STDOUT);
308
309         if (argc < 2 || argv[1][0] == '-') {
310                 usage();
311                 exit(1);
312         }
313
314         share = argv[1];
315
316         all_string_sub(share,"/","\\",0);
317
318         setup_logging(argv[0], DEBUG_STDOUT);
319
320         argc -= 1;
321         argv += 1;
322
323         lp_ctx = loadparm_init(talloc_autofree_context());
324         lp_load(lp_ctx, dyn_CONFIGFILE);
325
326         credentials = cli_credentials_init(talloc_autofree_context());
327         cli_credentials_guess(credentials, lp_ctx);
328
329         seed = time(NULL);
330
331         while ((opt = getopt(argc, argv, "n:d:U:s:hm:f:aoW:M:vEl:")) != EOF) {
332                 switch (opt) {
333                 case 'n':
334                         NumLoops = atoi(optarg);
335                         break;
336                 case 'd':
337                         DEBUGLEVEL = atoi(optarg);
338                         break;
339                 case 'E':
340                         die_on_error = 1;
341                         break;
342                 case 'v':
343                         verbose++;
344                         break;
345                 case 'M':
346                         lp_set_cmdline(lp_ctx, "max protocol", optarg);
347                         break;
348                 case 'U':
349                         cli_credentials_parse_string(credentials, optarg, CRED_SPECIFIED);
350                         break;
351                 case 's':
352                         seed = atoi(optarg);
353                         break;
354                 case 'h':
355                         usage();
356                         exit(1);
357                 case 'm':
358                         maskchars = optarg;
359                         break;
360                 case 'l':
361                         max_length = atoi(optarg);
362                         break;
363                 case 'f':
364                         filechars = optarg;
365                         break;
366                 case 'a':
367                         showall = 1;
368                         break;
369                 case 'o':
370                         old_list = true;
371                         break;
372                 default:
373                         printf("Unknown option %c (%d)\n", (char)opt, opt);
374                         exit(1);
375                 }
376         }
377
378         gensec_init(lp_ctx);
379
380         argc -= optind;
381         argv += optind;
382
383         cli = connect_one(share, lp_smb_ports(lp_ctx));
384         if (!cli) {
385                 DEBUG(0,("Failed to connect to %s\n", share));
386                 exit(1);
387         }
388
389         /* need to init seed after connect as clientgen uses random numbers */
390         DEBUG(0,("seed=%d     format --- --- (server, correct)\n", seed));
391         srandom(seed);
392
393         test_mask(argc, argv, cli);
394
395         return(0);
396 }