Logo Search packages:      
Sourcecode: samba-doc-ja version File versions

masktest.c

/* 
   Unix SMB/Netbios implementation.
   Version 2.0
   mask_match tester
   Copyright (C) Andrew Tridgell 1999
   
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.
   
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#define NO_SYSLOG

#include "includes.h"

static pstring password = "";
static pstring username = "";
static BOOL got_pass;
static int max_protocol = PROTOCOL_NT1;
static BOOL showall = False;
static BOOL old_list = False;
static char *maskchars = "<>\"?*abc.";
static char *filechars = "abcdefghijklm.";
static int verbose;
static int die_on_error;
static int NumLoops = 0;

#define is_effective_sjis(s)  (lp_client_code_page() == KANJI_CODEPAGE \
                        && is_shift_jis_str(s))

/* a test fn for LANMAN mask support */
int ms_fnmatch_lanman_core(char *pattern, char *string)
{
      char *p = pattern, *n = string;
      char c;

      if (strcmp(p,"?")==0 && strcmp(n,".")==0) goto match;

      while ((c = *p++)) {
            switch (c) {
            case '.':
                  /* if (! *n && ! *p) goto match; */
                  if (*n != '.') goto nomatch;
                  n++;
                  break;

            case '?':
                  if ((*n == '.' && n[1] != '.') || ! *n) goto next;
                  if (is_effective_sjis(n)) n++;
                  n++;
                  break;

            case '>':
                  if (n[0] == '.') {
                        if (! n[1] && ms_fnmatch_lanman_core(p, n+1) == 0) goto match;
                        if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
                        goto nomatch;
                  }
                  if (! *n) goto next;
                  if (is_effective_sjis(n)) n++;
                  n++;
                  break;

            case '*':
                  if (! *p) goto match;
                  for (; *n; n++) {
                        if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
                        if (is_effective_sjis(n)) n++;
                  }
                  break;

            case '<':
                  for (; *n; n++) {
                        if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
                        if (*n == '.' && !strchr(n+1,'.')) {
                              n++;
                              break;
                        }
                        if (is_effective_sjis(n)) n++;
                  }
                  break;

            case '"':
                  if (*n == 0 && ms_fnmatch_lanman_core(p, n) == 0) goto match;
                  if (*n != '.') goto nomatch;
                  n++;
                  break;

            default:
                  if (is_effective_sjis(n)) {
                        if (is_shift_jis_str(p-1)){
                              if (*n != c) {
                                    int w1, w2;
                                    w1 = ((int)(unsigned char)*n << 8) | (unsigned char)*(n+1);
                                    w1 = regularize_sjis(w1);
                                    w2 = ((int)(unsigned char)c << 8) | (unsigned char)*(p);
                                    w2 = regularize_sjis(w2);
                                    if (w1 != w2)
                                          goto nomatch;
                              } else {
                                    if (*(n+1) != *p)
                                          goto nomatch;
                              }
                              p++;
                              n++;
                        } else {
                              goto nomatch;
                        }
                  } else if (c != *n) {
                        goto nomatch;
                  }
                  n++;
                  break;
            }
      }
      
      if (! *n) goto match;
      
 nomatch:
      if (verbose) printf("NOMATCH pattern=[%s] string=[%s]\n", pattern, string);
      return -1;

next:
      if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
        goto nomatch;

 match:
      if (verbose) printf("MATCH   pattern=[%s] string=[%s]\n", pattern, string);
      return 0;
}

int ms_fnmatch_lanman(char *pattern, char *string)
{
      if (!strpbrk(pattern, "?*<>\"")) {
            if (strcmp(string,"..") == 0) string = ".";
            return strcmp(pattern, string);
      }

      if (strcmp(string,"..") == 0 || strcmp(string,".") == 0) {
            return ms_fnmatch_lanman_core(pattern, "..") &&
                  ms_fnmatch_lanman_core(pattern, ".");
      }

      return ms_fnmatch_lanman_core(pattern, string);
}

static BOOL reg_match_one(char *pattern, char *file)
{
      /* oh what a weird world this is */
      if (old_list && strcmp(pattern, "*.*") == 0) return True;

      if (strcmp(pattern,".") == 0) return False;

      if (max_protocol <= PROTOCOL_LANMAN2) {
            return ms_fnmatch_lanman(pattern, file)==0;
      }

      if (strcmp(file,"..") == 0) file = ".";

      return ms_fnmatch(pattern, file)==0;
}

static char *reg_test(char *pattern, char *long_name, char *short_name)
{
      static fstring ret;
      fstrcpy(ret, "---");

      pattern = 1+strrchr(pattern,'\\');

      if (reg_match_one(pattern, ".")) ret[0] = '+';
      if (reg_match_one(pattern, "..")) ret[1] = '+';
      if (reg_match_one(pattern, long_name) || 
          (*short_name && reg_match_one(pattern, short_name))) ret[2] = '+';
      return ret;
}


/***************************************************** 
return a connection to a server
*******************************************************/
struct cli_state *connect_one(char *share)
{
      struct cli_state *c;
      struct nmb_name called, calling;
      char *server_n;
      char *server;
      struct in_addr ip;

      server = share+2;
      share = strchr(server,'\\');
      if (!share) return NULL;
      *share = 0;
      share++;

      server_n = server;
      
      zero_ip(&ip);

      make_nmb_name(&calling, "masktest", 0x0);
      make_nmb_name(&called , server, 0x20);

 again:
      zero_ip(&ip);

      /* have to open a new connection */
      if (!(c=cli_initialise(NULL)) || (cli_set_port(c, 139) == 0) ||
          !cli_connect(c, server_n, &ip)) {
            DEBUG(0,("Connection to %s failed\n", server_n));
            return NULL;
      }

      c->protocol = max_protocol;

      if (!cli_session_request(c, &calling, &called)) {
            DEBUG(0,("session request to %s failed\n", called.name));
            cli_shutdown(c);
            if (strcmp(called.name, "*SMBSERVER")) {
                  make_nmb_name(&called , "*SMBSERVER", 0x20);
                  goto again;
            }
            return NULL;
      }

      DEBUG(4,(" session request ok\n"));

      if (!cli_negprot(c)) {
            DEBUG(0,("protocol negotiation failed\n"));
            cli_shutdown(c);
            return NULL;
      }

      if (!got_pass) {
            char *pass;

            pass = (char *)getpass("Password: ");
            if (pass) {
                  pstrcpy(password, pass);
            }
      }

      if (!cli_session_setup(c, username, 
                         password, strlen(password),
                         password, strlen(password),
                         lp_workgroup())) {
            DEBUG(0,("session setup failed: %s\n", cli_errstr(c)));
            return NULL;
      }

      /*
       * These next two lines are needed to emulate
       * old client behaviour for people who have
       * scripts based on client output.
       * QUESTION ? Do we want to have a 'client compatibility
       * mode to turn these on/off ? JRA.
       */

      if (*c->server_domain || *c->server_os || *c->server_type)
            DEBUG(1,("Domain=[%s] OS=[%s] Server=[%s]\n",
                  c->server_domain,c->server_os,c->server_type));
      
      DEBUG(4,(" session setup ok\n"));

      if (!cli_send_tconX(c, share, "?????",
                      password, strlen(password)+1)) {
            DEBUG(0,("tree connect failed: %s\n", cli_errstr(c)));
            cli_shutdown(c);
            return NULL;
      }

      DEBUG(4,(" tconx ok\n"));

      return c;
}

static char *resultp;
static file_info *finfo;

void listfn(file_info *f, const char *s, void *state)
{
      if (strcmp(f->name,".") == 0) {
            resultp[0] = '+';
      } else if (strcmp(f->name,"..") == 0) {
            resultp[1] = '+';       
      } else {
            resultp[2] = '+';
      }
      finfo = f;
}

static void get_real_name(struct cli_state *cli, 
                    pstring long_name, fstring short_name)
{
      /* nasty hack to force level 260 listings - tridge */
      cli->capabilities |= CAP_NT_SMBS;
      if (max_protocol <= PROTOCOL_LANMAN1) {
            cli_list_new(cli, "\\masktest\\*.*", aHIDDEN | aDIR, listfn, NULL);
      } else {
            cli_list_new(cli, "\\masktest\\*", aHIDDEN | aDIR, listfn, NULL);
      }
      if (finfo) {
            fstrcpy(short_name, finfo->short_name);
            strlower(short_name);
            pstrcpy(long_name, finfo->name);
            strlower(long_name);
      }

      if (*short_name == 0) {
            fstrcpy(short_name, long_name);
      }

#if 0
      if (!strchr(short_name,'.')) {
            fstrcat(short_name,".");
      }
#endif
}

static void testpair(struct cli_state *cli, char *mask, char *file)
{
      int fnum;
      fstring res1;
      char *res2;
      static int count;
      fstring short_name;
      pstring long_name;

      count++;

      unix_to_dos(mask, mask);
      unix_to_dos(file, file);
      fstrcpy(res1, "---");

      fnum = cli_open(cli, file, O_CREAT|O_TRUNC|O_RDWR, 0);
      if (fnum == -1) {
            DEBUG(0,("Can't create %s\n", file));
            return;
      }
      cli_close(cli, fnum);

      resultp = res1;
      fstrcpy(short_name, "");
      finfo = NULL;
      get_real_name(cli, long_name, short_name);
      finfo = NULL;
      fstrcpy(res1, "---");
      cli_list(cli, mask, aHIDDEN | aDIR, listfn, NULL);

      res2 = reg_test(mask, long_name, short_name);

      if (showall || strcmp(res1, res2)) {
            DEBUG(0,("%s %s %d mask=[%s] file=[%s] rfile=[%s/%s]\n",
                   res1, res2, count, mask, file, long_name, short_name));
            if (die_on_error) exit(1);
      }

      cli_unlink(cli, file);

      if (count % 100 == 0) DEBUG(0,("%d\n", count));
}

static void test_mask(int argc, char *argv[], 
                  struct cli_state *cli)
{
      pstring mask, file;
      int l1, l2, i, l;
      int mc_len = strlen(maskchars);
      int fc_len = strlen(filechars);

      cli_mkdir(cli, "\\masktest");

      cli_unlink(cli, "\\masktest\\*");

      if (argc >= 2) {
            while (argc >= 2) {
                  pstrcpy(mask,"\\masktest\\");
                  pstrcpy(file,"\\masktest\\");
                  pstrcat(mask, argv[0]);
                  pstrcat(file, argv[1]);
                  unix_to_dos(mask, mask);
                  unix_to_dos(file, file);
                  testpair(cli, mask, file);
                  argv += 2;
                  argc -= 2;
            }
            goto finished;
      }

      while (1) {
            l1 = 1 + random() % 20;
            l2 = 1 + random() % 20;
            pstrcpy(mask,"\\masktest\\");
            pstrcpy(file,"\\masktest\\");
            l = strlen(mask);
            for (i=0;i<l1;i++) {
                  mask[i+l] = maskchars[random() % mc_len];
            }
            mask[l+l1] = 0;

            for (i=0;i<l2;i++) {
                  file[i+l] = filechars[random() % fc_len];
            }
            file[l+l2] = 0;

            if (strcmp(file+l,".") == 0 || 
                strcmp(file+l,"..") == 0 ||
                strcmp(mask+l,"..") == 0) continue;

            if (strspn(file+l, ".") == strlen(file+l)) continue;

            testpair(cli, mask, file);
            if (NumLoops && (--NumLoops == 0))
            break;
      }

 finished:
      cli_rmdir(cli, "\\masktest");
}


static void usage(void)
{
      printf(
"Usage:\n\
  masktest //server/share [options..]\n\
  options:\n\
      -d debuglevel\n\
      -n numloops\n\
        -W workgroup\n\
        -U user%%pass\n\
        -s seed\n\
      -M max protocol\n\
        -f filechars (default %s)\n\
        -m maskchars (default %s)\n\
        -v                             verbose mode\n\
        -E                             die on error\n\
        -a                             show all tests\n\
\n\
  This program tests wildcard matching between two servers. It generates\n\
  random pairs of filenames/masks and tests that they match in the same\n\
  way on the servers and internally\n\
", 
  filechars, maskchars);
}

/****************************************************************************
  main program
****************************************************************************/
 int main(int argc,char *argv[])
{
      pstring share;
      struct cli_state *cli;  
      extern char *optarg;
      extern int optind;
      extern FILE *dbf;
      extern BOOL AllowDebugChange;
      int opt;
      char *p;
      int seed;
      static pstring servicesf = CONFIGFILE;
      extern int Protocol;

      setlinebuf(stdout);

      AllowDebugChange = False;
      DEBUGLEVEL = 0;
      dbf = stderr;

      if (argc < 2 || argv[1][0] == '-') {
            usage();
            exit(1);
      }

      pstrcpy(share, argv[1]);

      setup_logging(argv[0],True);

      argc -= 1;
      argv += 1;

      TimeInit();
      charset_initialise();

      if (getenv("USER")) {
            pstrcpy(username,getenv("USER"));

            if ((p=strchr(username,'%'))) {
                  *p = 0;
                  pstrcpy(password,p+1);
                  got_pass = True;
                  memset(strchr(getenv("USER"),'%')+1,'X',strlen(password));
            }
      }

      if (getenv("PASSWD")) {
            pstrcpy(password,getenv("PASSWD"));
            got_pass = True;
      }

      if (*username == 0 && getenv("LOGNAME")) {
            pstrcpy(username,getenv("LOGNAME"));
      }

      seed = time(NULL);

      while ((opt = getopt(argc, argv, "n:d:U:s:hm:f:aoW:M:vE")) != EOF) {
            switch (opt) {
            case 'n':
                  NumLoops = atoi(optarg);
                  break;
            case 'd':
                  DEBUGLEVEL = atoi(optarg);
                  break;
            case 'E':
                  die_on_error = 1;
                  break;
            case 'v':
                  verbose++;
                  break;
            case 'M':
                  max_protocol = interpret_protocol(optarg, max_protocol);
                  break;
            case 'U':
                  pstrcpy(username,optarg);
                  p = strchr(username,'%');
                  if (p) {
                        *p = 0;
                        pstrcpy(password, p+1);
                        got_pass = True;
                        memset(strchr(optarg,'%')+1,'X',strlen(password));
                  }
                  break;
            case 's':
                  seed = atoi(optarg);
                  break;
            case 'h':
                  usage();
                  exit(1);
            case 'm':
                  maskchars = optarg;
                  break;
            case 'f':
                  filechars = optarg;
                  break;
            case 'a':
                  showall = 1;
                  break;
            case 'o':
                  old_list = True;
                  break;
            default:
                  printf("Unknown option %c (%d)\n", (char)opt, opt);
                  exit(1);
            }
      }

      argc -= optind;
      argv += optind;

      lp_load(servicesf,True,False,False);
      load_interfaces();
      codepage_initialise(lp_client_code_page());

      unix_to_dos(username, username);
      strupper(username);
      unix_to_dos(share, share);
      strupper(share);

      all_string_sub(share,"/","\\",0);

      cli = connect_one(share);
      if (!cli) {
            DEBUG(0,("Failed to connect to %s\n", share));
            exit(1);
      }
      Protocol = cli->protocol;

      /* need to init seed after connect as clientgen uses random numbers */
      DEBUG(0,("seed=%d\n", seed));
      srandom(seed);

      test_mask(argc, argv, cli);

      return(0);
}

Generated by  Doxygen 1.6.0   Back to index