r26605: masktest: Get rid of pstrings.
[ira/wip.git] / source / torture / masktest.c
index 4097c8c35a6603980cb7ea3fa06b024c93e0733c..201738626b3d3a25f0590b6104285a6935318554 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 BOOL showall = False;
-static BOOL old_list = False;
+#include "system/filesys.h"
+#include "system/dir.h"
+#include "libcli/libcli.h"
+#include "libcli/raw/libcliraw.h"
+#include "system/time.h"
+#include "pstring.h"
+#include "auth/credentials/credentials.h"
+#include "auth/gensec/gensec.h"
+#include "param/param.h"
+#include "dynconfig.h"
+
+#define MAX_STRING_LEN 1024
+
+static struct cli_credentials *credentials;
+static bool showall = false;
+static bool old_list = false;
 static const char *maskchars = "<>\"?*abc.";
 static const char *filechars = "abcdefghijklm.";
 static int verbose;
@@ -31,14 +41,14 @@ static int die_on_error;
 static int NumLoops = 0;
 static int max_length = 20;
 
-static BOOL reg_match_one(struct smbcli_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 (ISDOT(pattern)) return false;
 
-       if (strcmp(file,"..") == 0) file = ".";
+       if (ISDOTDOT(file)) file = ".";
 
        return ms_fnmatch(pattern, file, cli->transport->negotiate.protocol)==0;
 }
@@ -61,11 +71,10 @@ static char *reg_test(struct smbcli_state *cli, char *pattern, char *long_name,
 /***************************************************** 
 return a connection to a server
 *******************************************************/
-static struct smbcli_state *connect_one(char *share)
+static struct smbcli_state *connect_one(char *share, const char **ports)
 {
        struct smbcli_state *c;
        fstring server;
-       uint_t flags = 0;
        NTSTATUS status;
 
        fstrcpy(server,share+2);
@@ -74,11 +83,13 @@ static struct smbcli_state *connect_one(char *share)
        *share = 0;
        share++;
 
-       status = smbcli_full_connection(NULL, &c, "masktest",
-                                    server, NULL, 
-                                    share, "?????", 
-                                    username, lp_workgroup(), 
-                                    password, flags, NULL);
+       cli_credentials_set_workstation(credentials, "masktest", CRED_SPECIFIED);
+
+       status = smbcli_full_connection(NULL, &c,
+                                       server, 
+                                       ports,
+                                       share, NULL,
+                                       credentials, NULL);
 
        if (!NT_STATUS_IS_OK(status)) {
                return NULL;
@@ -89,27 +100,27 @@ static struct smbcli_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) {
+       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;
+       safe_strcpy(last_hit.long_name, f->name, MAX_STRING_LEN);
+       safe_strcpy(last_hit.short_name, f->short_name, MAX_STRING_LEN);
+       f_info_hit = true;
 }
 
-static void get_real_name(struct smbcli_state *cli, 
-                         pstring long_name, fstring short_name)
+static void get_real_name(TALLOC_CTX *mem_ctx, struct smbcli_state *cli,
+                         char *long_name, fstring short_name)
 {
        const char *mask;
        if (cli->transport->negotiate.protocol <= PROTOCOL_LANMAN1) {
@@ -118,16 +129,19 @@ static void get_real_name(struct smbcli_state *cli,
                mask = "\\masktest\\*";
        }
 
-       f_info_hit = False;
+       f_info_hit = false;
+       last_hit.short_name = talloc_array(mem_ctx, char, MAX_STRING_LEN);
+       last_hit.long_name = talloc_array(mem_ctx, char, MAX_STRING_LEN);
 
-       smbcli_list_new(cli->tree, mask, 
-                    FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY, 
-                    listfn, NULL);
+       smbcli_list_new(cli->tree, mask,
+                       FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,
+                       RAW_SEARCH_DATA_BOTH_DIRECTORY_INFO,
+                       listfn, NULL);
 
        if (f_info_hit) {
                fstrcpy(short_name, last_hit.short_name);
                strlower(short_name);
-               pstrcpy(long_name, last_hit.long_name);
+               safe_strcpy(long_name, last_hit.long_name, MAX_STRING_LEN);
                strlower(long_name);
        }
 
@@ -136,15 +150,17 @@ static void get_real_name(struct smbcli_state *cli,
        }
 }
 
-static void testpair(struct smbcli_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 *res2;
        static int count;
        fstring short_name;
-       pstring long_name;
+       char *long_name;
 
+       long_name = talloc_array(mem_ctx, char, MAX_STRING_LEN);
        count++;
 
        fstrcpy(res1, "---");
@@ -158,11 +174,12 @@ static void testpair(struct smbcli_state *cli, char *mask, char *file)
 
        resultp = res1;
        fstrcpy(short_name, "");
-       get_real_name(cli, long_name, short_name);
+       get_real_name(mem_ctx, cli, long_name, short_name);
        fstrcpy(res1, "---");
-       smbcli_list(cli->tree, mask, 
-                FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY, 
-                listfn, NULL);
+       smbcli_list_new(cli->tree, mask, 
+                       FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY, 
+                       RAW_SEARCH_DATA_BOTH_DIRECTORY_INFO,
+                       listfn, NULL);
 
        res2 = reg_test(cli, mask, long_name, short_name);
 
@@ -175,27 +192,34 @@ static void testpair(struct smbcli_state *cli, char *mask, char *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[], 
+static void test_mask(int argc, char *argv[],
                      struct smbcli_state *cli)
 {
-       pstring mask, file;
+       TALLOC_CTX *mem_ctx;
+       char *mask, *file;
        int l1, l2, i, l;
        int mc_len = strlen(maskchars);
        int fc_len = strlen(filechars);
 
+       mem_ctx = talloc_init("test_mask");
+       mask = talloc_array(mem_ctx, char, MAX_STRING_LEN);
+       file = talloc_array(mem_ctx, char, MAX_STRING_LEN);
+
        smbcli_mkdir(cli->tree, "\\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);
+                       safe_strcpy(mask,"\\masktest\\", MAX_STRING_LEN);
+                       safe_strcpy(file,"\\masktest\\", MAX_STRING_LEN);
+                       safe_strcat(mask, argv[0], MAX_STRING_LEN);
+                       safe_strcat(file, argv[1], MAX_STRING_LEN);
+                       testpair(mem_ctx, cli, mask, file);
                        argv += 2;
                        argc -= 2;
                }
@@ -205,8 +229,8 @@ static void test_mask(int argc, char *argv[],
        while (1) {
                l1 = 1 + random() % max_length;
                l2 = 1 + random() % max_length;
-               pstrcpy(mask,"\\masktest\\");
-               pstrcpy(file,"\\masktest\\");
+               safe_strcpy(mask,"\\masktest\\", MAX_STRING_LEN);
+               safe_strcpy(file,"\\masktest\\", MAX_STRING_LEN);
                l = strlen(mask);
                for (i=0;i<l1;i++) {
                        mask[i+l] = maskchars[random() % mc_len];
@@ -218,19 +242,20 @@ 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:
        smbcli_rmdir(cli->tree, "\\masktest");
+       talloc_free(mem_ctx);
 }
 
 
@@ -268,15 +293,13 @@ static void usage(void)
        char *share;
        struct smbcli_state *cli;       
        int opt;
-       char *p;
        int seed;
+       struct loadparm_context *lp_ctx;
 
        setlinebuf(stdout);
 
        setup_logging("masktest", DEBUG_STDOUT);
 
-       lp_set_cmdline("log level", "0");
-
        if (argc < 2 || argv[1][0] == '-') {
                usage();
                exit(1);
@@ -291,12 +314,11 @@ static void usage(void)
        argc -= 1;
        argv += 1;
 
-       lp_load(dyn_CONFIGFILE,True,False,False);
-       load_interfaces();
+       lp_ctx = loadparm_init(talloc_autofree_context());
+       lp_load(lp_ctx, dyn_CONFIGFILE);
 
-       if (getenv("USER")) {
-               fstrcpy(username,getenv("USER"));
-       }
+       credentials = cli_credentials_init(talloc_autofree_context());
+       cli_credentials_guess(credentials, lp_ctx);
 
        seed = time(NULL);
 
@@ -315,15 +337,10 @@ static void usage(void)
                        verbose++;
                        break;
                case 'M':
-                       lp_set_cmdline("max protocol", optarg);
+                       lp_set_cmdline(lp_ctx, "max protocol", optarg);
                        break;
                case 'U':
-                       fstrcpy(username,optarg);
-                       p = strchr_m(username,'%');
-                       if (p) {
-                               *p = 0;
-                               fstrcpy(password, p+1);
-                       }
+                       cli_credentials_parse_string(credentials, optarg, CRED_SPECIFIED);
                        break;
                case 's':
                        seed = atoi(optarg);
@@ -344,7 +361,7 @@ static void usage(void)
                        showall = 1;
                        break;
                case 'o':
-                       old_list = True;
+                       old_list = true;
                        break;
                default:
                        printf("Unknown option %c (%d)\n", (char)opt, opt);
@@ -352,11 +369,12 @@ static void usage(void)
                }
        }
 
+       gensec_init(lp_ctx);
+
        argc -= optind;
        argv += optind;
 
-
-       cli = connect_one(share);
+       cli = connect_one(share, lp_smb_ports(lp_ctx));
        if (!cli) {
                DEBUG(0,("Failed to connect to %s\n", share));
                exit(1);