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 }