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