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