s4-loadparm: 2nd half of lp_ to lpcfg_ conversion
[gd/samba-autobuild/.git] / source4 / torture / masktest.c
index 7906684169d624e60bb763e7d5bea43229df913a..9c66291087fb6d3f1f7e7366eb5bdcb6a719c4dc 100644 (file)
@@ -5,7 +5,7 @@
    
    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
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    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.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
-
-static fstring password;
-static fstring username;
-static int max_protocol = PROTOCOL_NT1;
-static BOOL showall = False;
-static BOOL old_list = False;
+#include "lib/cmdline/popt_common.h"
+#include "system/filesys.h"
+#include "system/dir.h"
+#include "libcli/libcli.h"
+#include "system/time.h"
+#include "auth/credentials/credentials.h"
+#include "auth/gensec/gensec.h"
+#include "param/param.h"
+#include "libcli/resolve/resolve.h"
+#include "lib/events/events.h"
+
+static bool showall = false;
+static bool old_list = false;
 static const char *maskchars = "<>\"?*abc.";
 static const char *filechars = "abcdefghijklm.";
-static int verbose;
 static int die_on_error;
 static int NumLoops = 0;
+static int max_length = 20;
+struct masktest_state {
+       TALLOC_CTX *mem_ctx;
+};
 
-/* a test fn for LANMAN mask support */
-static int ms_fnmatch_lanman_core(const char *pattern, const char *string)
-{
-       const 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;
-                       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;
-                       n++;
-                       break;
-
-               case '*':
-                       if (! *p) goto match;
-                       for (; *n; n++) {
-                               if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
-                       }
-                       break;
-
-               case '<':
-                       for (; *n; n++) {
-                               if (ms_fnmatch_lanman_core(p, n) == 0) goto match;
-                               if (*n == '.' && !strchr_m(n+1,'.')) {
-                                       n++;
-                                       break;
-                               }
-                       }
-                       break;
-
-               case '"':
-                       if (*n == 0 && ms_fnmatch_lanman_core(p, n) == 0) goto match;
-                       if (*n != '.') goto nomatch;
-                       n++;
-                       break;
-
-               default:
-                       if (c != *n) goto nomatch;
-                       n++;
-               }
-       }
-       
-       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;
-}
-
-static int ms_fnmatch_lanman(const char *pattern, const 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(struct cli_state *cli, const char *pattern, const char *file)
+static bool reg_match_one(struct smbcli_state *cli, const char *pattern, const char *file)
 {
        /* oh what a weird world this is */
-       if (old_list && strcmp(pattern, "*.*") == 0) return True;
+       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 (ISDOT(pattern)) return false;
 
-       if (strcmp(file,"..") == 0) file = ".";
+       if (ISDOTDOT(file)) file = ".";
 
        return ms_fnmatch(pattern, file, cli->transport->negotiate.protocol)==0;
 }
 
-static char *reg_test(struct cli_state *cli, char *pattern, char *long_name, char *short_name)
+static char *reg_test(struct smbcli_state *cli, TALLOC_CTX *mem_ctx, const char *pattern, const char *long_name, const char *short_name)
 {
-       static fstring ret;
-       fstrcpy(ret, "---");
+       char *ret;
+       ret = talloc_strdup(mem_ctx, "---");
 
        pattern = 1+strrchr_m(pattern,'\\');
 
@@ -157,24 +70,35 @@ static char *reg_test(struct cli_state *cli, char *pattern, char *long_name, cha
 /***************************************************** 
 return a connection to a server
 *******************************************************/
-static struct cli_state *connect_one(char *share)
+static struct smbcli_state *connect_one(struct resolve_context *resolve_ctx, 
+                                       struct tevent_context *ev,
+                                       TALLOC_CTX *mem_ctx,
+                                       char *share, const char **ports,
+                                       const char *socket_options,
+                                       struct smbcli_options *options,
+                                       struct smbcli_session_options *session_options,
+                                       struct gensec_settings *gensec_settings)
 {
-       struct cli_state *c;
-       fstring server;
-       uint_t flags = 0;
+       struct smbcli_state *c;
+       char *server;
        NTSTATUS status;
 
-       fstrcpy(server,share+2);
+       server = talloc_strdup(mem_ctx, share+2);
        share = strchr_m(server,'\\');
        if (!share) return NULL;
        *share = 0;
        share++;
 
-       status = cli_full_connection(&c, "masktest",
-                                    server, NULL, 
-                                    share, "?????", 
-                                    username, lp_workgroup(), 
-                                    password, flags, NULL);
+       cli_credentials_set_workstation(cmdline_credentials, "masktest", CRED_SPECIFIED);
+
+       status = smbcli_full_connection(NULL, &c,
+                                       server, 
+                                       ports,
+                                       share, NULL,
+                                       socket_options,
+                                       cmdline_credentials, resolve_ctx, ev,
+                                       options, session_options,
+                                       gensec_settings);
 
        if (!NT_STATUS_IS_OK(status)) {
                return NULL;
@@ -185,80 +109,95 @@ static struct cli_state *connect_one(char *share)
 
 static char *resultp;
 static struct {
-       pstring long_name;
-       pstring short_name;
+       char *long_name;
+       char *short_name;
 } last_hit;
-static BOOL f_info_hit;
+static bool f_info_hit;
 
-static void listfn(file_info *f, const char *s, void *state)
+static void listfn(struct clilist_file_info *f, const char *s, void *state)
 {
-       if (strcmp(f->name,".") == 0) {
+       struct masktest_state *m = (struct masktest_state *)state;
+
+       if (ISDOT(f->name)) {
                resultp[0] = '+';
-       } else if (strcmp(f->name,"..") == 0) {
-               resultp[1] = '+';               
+       } else if (ISDOTDOT(f->name)) {
+               resultp[1] = '+';
        } else {
                resultp[2] = '+';
        }
-       pstrcpy(last_hit.long_name, f->name);
-       pstrcpy(last_hit.short_name, f->short_name);
-       f_info_hit = True;
+
+       last_hit.long_name = talloc_strdup(m->mem_ctx, f->name);
+       last_hit.short_name = talloc_strdup(m->mem_ctx, f->short_name);
+       f_info_hit = true;
 }
 
-static void get_real_name(struct cli_state *cli, 
-                         pstring long_name, fstring short_name)
+static void get_real_name(TALLOC_CTX *mem_ctx, struct smbcli_state *cli,
+                         char **long_name, char **short_name)
 {
        const char *mask;
-       if (max_protocol <= PROTOCOL_LANMAN1) {
+       struct masktest_state state;
+
+       if (cli->transport->negotiate.protocol <= PROTOCOL_LANMAN1) {
                mask = "\\masktest\\*.*";
        } else {
                mask = "\\masktest\\*";
        }
 
-       f_info_hit = False;
+       f_info_hit = false;
 
-       cli_list_new(cli, mask, FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY, 
-                    listfn, NULL);
+       state.mem_ctx = mem_ctx;
+
+       smbcli_list_new(cli->tree, mask,
+                       FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,
+                       RAW_SEARCH_DATA_BOTH_DIRECTORY_INFO,
+                       listfn, &state);
 
        if (f_info_hit) {
-               fstrcpy(short_name, last_hit.short_name);
-               strlower(short_name);
-               pstrcpy(long_name, last_hit.long_name);
-               strlower(long_name);
+               *short_name = talloc_strdup(mem_ctx, last_hit.short_name);
+               strlower(*short_name);
+               *long_name = talloc_strdup(mem_ctx, last_hit.long_name);
+               strlower(*long_name);
        }
 
-       if (*short_name == 0) {
-               fstrcpy(short_name, long_name);
+       if (*short_name == '\0') {
+               *short_name = talloc_strdup(mem_ctx, *long_name);
        }
 }
 
-static void testpair(struct cli_state *cli, char *mask, char *file)
+static void testpair(TALLOC_CTX *mem_ctx, struct smbcli_state *cli, char *mask,
+               char *file)
 {
        int fnum;
-       fstring res1;
+       char res1[256];
        char *res2;
        static int count;
-       fstring short_name;
-       pstring long_name;
+       char *short_name = NULL;
+       char *long_name = NULL;
+       struct masktest_state state;
 
        count++;
 
-       fstrcpy(res1, "---");
+       safe_strcpy(res1, "---", sizeof(res1));
+
+       state.mem_ctx = mem_ctx;
 
-       fnum = cli_open(cli, file, O_CREAT|O_TRUNC|O_RDWR, 0);
+       fnum = smbcli_open(cli->tree, file, O_CREAT|O_TRUNC|O_RDWR, 0);
        if (fnum == -1) {
                DEBUG(0,("Can't create %s\n", file));
                return;
        }
-       cli_close(cli, fnum);
+       smbcli_close(cli->tree, fnum);
 
        resultp = res1;
-       fstrcpy(short_name, "");
-       get_real_name(cli, long_name, short_name);
-       fstrcpy(res1, "---");
-       cli_list(cli, mask, FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY, 
-                listfn, NULL);
+       short_name = talloc_strdup(mem_ctx, "");
+       get_real_name(mem_ctx, cli, &long_name, &short_name);
+       safe_strcpy(res1, "---", sizeof(res1));
+       smbcli_list_new(cli->tree, mask,
+                       FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,
+                       RAW_SEARCH_DATA_BOTH_DIRECTORY_INFO,
+                       listfn, &state);
 
-       res2 = reg_test(cli, mask, long_name, short_name);
+       res2 = reg_test(cli, mem_ctx, mask, long_name, short_name);
 
        if (showall || strcmp(res1, res2)) {
                d_printf("%s %s %d mask=[%s] file=[%s] rfile=[%s/%s]\n",
@@ -266,30 +205,33 @@ static void testpair(struct cli_state *cli, char *mask, char *file)
                if (die_on_error) exit(1);
        }
 
-       cli_unlink(cli, file);
+       smbcli_unlink(cli->tree, file);
 
        if (count % 100 == 0) DEBUG(0,("%d\n", count));
+
+       resultp = NULL;
 }
 
-static void test_mask(int argc, char *argv[], 
-                     struct cli_state *cli)
+static void test_mask(int argc, char *argv[],
+                                         TALLOC_CTX *mem_ctx,
+                     struct smbcli_state *cli)
 {
-       pstring mask, file;
+       char *mask, *file;
        int l1, l2, i, l;
        int mc_len = strlen(maskchars);
        int fc_len = strlen(filechars);
 
-       cli_mkdir(cli, "\\masktest");
+       smbcli_mkdir(cli->tree, "\\masktest");
 
-       cli_unlink(cli, "\\masktest\\*");
+       smbcli_unlink(cli->tree, "\\masktest\\*");
 
        if (argc >= 2) {
                while (argc >= 2) {
-                       pstrcpy(mask,"\\masktest\\");
-                       pstrcpy(file,"\\masktest\\");
-                       pstrcat(mask, argv[0]);
-                       pstrcat(file, argv[1]);
-                       testpair(cli, mask, file);
+                       mask = talloc_strdup(mem_ctx, "\\masktest\\");
+                       file = talloc_strdup(mem_ctx, "\\masktest\\");
+                       mask = talloc_strdup_append(mask, argv[0]);
+                       file = talloc_strdup_append(file, argv[1]);
+                       testpair(mem_ctx, cli, mask, file);
                        argv += 2;
                        argc -= 2;
                }
@@ -297,10 +239,12 @@ static void test_mask(int argc, char *argv[],
        }
 
        while (1) {
-               l1 = 1 + random() % 20;
-               l2 = 1 + random() % 20;
-               pstrcpy(mask,"\\masktest\\");
-               pstrcpy(file,"\\masktest\\");
+               l1 = 1 + random() % max_length;
+               l2 = 1 + random() % max_length;
+               mask = talloc_strdup(mem_ctx, "\\masktest\\");
+               file = talloc_strdup(mem_ctx, "\\masktest\\");
+               mask = talloc_realloc_size(mem_ctx, mask, strlen(mask)+l1+1);
+               file = talloc_realloc_size(mem_ctx, file, strlen(file)+l2+1);
                l = strlen(mask);
                for (i=0;i<l1;i++) {
                        mask[i+l] = maskchars[random() % mc_len];
@@ -312,45 +256,33 @@ static void test_mask(int argc, char *argv[],
                }
                file[l+l2] = 0;
 
-               if (strcmp(file+l,".") == 0 || 
-                   strcmp(file+l,"..") == 0 ||
-                   strcmp(mask+l,"..") == 0) continue;
+               if (ISDOT(file+l) || ISDOTDOT(file+l) || ISDOTDOT(mask+l)) {
+                       continue;
+               }
 
                if (strspn(file+l, ".") == strlen(file+l)) continue;
 
-               testpair(cli, mask, file);
+               testpair(mem_ctx, cli, mask, file);
                if (NumLoops && (--NumLoops == 0))
                        break;
        }
 
  finished:
-       cli_rmdir(cli, "\\masktest");
+       smbcli_rmdir(cli->tree, "\\masktest");
+       talloc_free(mem_ctx);
 }
 
 
-static void usage(void)
+static void usage(poptContext pc)
 {
        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);
+  way on the servers and internally\n");
+       poptPrintUsage(pc, stdout, 0);
 }
 
 /****************************************************************************
@@ -359,104 +291,96 @@ static void usage(void)
  int main(int argc,char *argv[])
 {
        char *share;
-       struct cli_state *cli;  
+       struct smbcli_state *cli;       
        int opt;
-       char *p;
        int seed;
+       struct tevent_context *ev;
+       struct loadparm_context *lp_ctx;
+       struct smbcli_options options;
+       struct smbcli_session_options session_options;
+       poptContext pc;
+       int argc_new, i;
+       char **argv_new;
+       TALLOC_CTX *mem_ctx;
+       enum {OPT_UNCLIST=1000};
+       struct poptOption long_options[] = {
+               POPT_AUTOHELP
+               {"seed",          0, POPT_ARG_INT,  &seed,      0,      "Seed to use for randomizer",   NULL},
+               {"num-ops",       0, POPT_ARG_INT,  &NumLoops,  0,      "num ops",      NULL},
+               {"maxlength",     0, POPT_ARG_INT,  &max_length,0,      "maximum length",       NULL},
+               {"dieonerror",    0, POPT_ARG_NONE, &die_on_error, 0,   "die on errors", NULL},
+               {"showall",       0, POPT_ARG_NONE, &showall,    0,      "display all operations", NULL},
+               {"oldlist",       0, POPT_ARG_NONE, &old_list,    0,     "use old list call", NULL},
+               {"maskchars",     0, POPT_ARG_STRING,   &maskchars,    0,"mask characters",     NULL},
+               {"filechars",     0, POPT_ARG_STRING,   &filechars,    0,"file characters",     NULL},
+               POPT_COMMON_SAMBA
+               POPT_COMMON_CONNECTION
+               POPT_COMMON_CREDENTIALS
+               POPT_COMMON_VERSION
+               { NULL }
+       };
 
        setlinebuf(stdout);
+       seed = time(NULL);
 
-       setup_logging("masktest", DEBUG_STDOUT);
+       pc = poptGetContext("locktest", argc, (const char **) argv, long_options, 
+                           POPT_CONTEXT_KEEP_FIRST);
 
-       lp_set_cmdline("log level", "0");
+       poptSetOtherOptionHelp(pc, "<unc>");
 
-       if (argc < 2 || argv[1][0] == '-') {
-               usage();
-               exit(1);
+       while((opt = poptGetNextOpt(pc)) != -1) {
+               switch (opt) {
+               case OPT_UNCLIST:
+                       lpcfg_set_cmdline(cmdline_lp_ctx, "torture:unclist", poptGetOptArg(pc));
+                       break;
+               }
        }
 
-       share = argv[1];
+       argv_new = discard_const_p(char *, poptGetArgs(pc));
+       argc_new = argc;
+       for (i=0; i<argc; i++) {
+               if (argv_new[i] == NULL) {
+                       argc_new = i;
+                       break;
+               }
+       }
 
-       all_string_sub(share,"/","\\",0);
+       if (!(argc_new >= 2)) {
+               usage(pc);
+               exit(1);
+       }
 
-       setup_logging(argv[0], DEBUG_STDOUT);
+       setup_logging("masktest", DEBUG_STDOUT);
 
-       argc -= 1;
-       argv += 1;
+       share = argv_new[1];
 
-       lp_load(dyn_CONFIGFILE,True,False,False);
-       load_interfaces();
+       all_string_sub(share,"/","\\",0);
 
-       if (getenv("USER")) {
-               fstrcpy(username,getenv("USER"));
-       }
+       lp_ctx = cmdline_lp_ctx;
 
-       seed = time(NULL);
+       mem_ctx = talloc_autofree_context();
 
-       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':
-                       fstrcpy(username,optarg);
-                       p = strchr_m(username,'%');
-                       if (p) {
-                               *p = 0;
-                               fstrcpy(password, p+1);
-                       }
-                       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);
-               }
-       }
+       ev = s4_event_context_init(mem_ctx);
 
-       argc -= optind;
-       argv += optind;
+       gensec_init(lp_ctx);
 
+       lpcfg_smbcli_options(lp_ctx, &options);
+       lpcfg_smbcli_session_options(lp_ctx, &session_options);
 
-       cli = connect_one(share);
+       cli = connect_one(lpcfg_resolve_context(lp_ctx), ev, mem_ctx, share,
+                         lpcfg_smb_ports(lp_ctx), lpcfg_socket_options(lp_ctx),
+                         &options, &session_options,
+                         lpcfg_gensec_settings(mem_ctx, lp_ctx));
        if (!cli) {
                DEBUG(0,("Failed to connect to %s\n", share));
                exit(1);
        }
 
        /* need to init seed after connect as clientgen uses random numbers */
-       DEBUG(0,("seed=%d\n", seed));
+       DEBUG(0,("seed=%d     format --- --- (server, correct)\n", seed));
        srandom(seed);
 
-       test_mask(argc, argv, cli);
+       test_mask(argc_new-1, argv_new+1, mem_ctx, cli);
 
        return(0);
 }