rpctorture command
authorLuke Leighton <lkcl@samba.org>
Thu, 22 Oct 1998 22:19:08 +0000 (22:19 +0000)
committerLuke Leighton <lkcl@samba.org>
Thu, 22 Oct 1998 22:19:08 +0000 (22:19 +0000)
(This used to be commit b4ae65e2d0582274d67d02ea190f6d3d83b48594)

source3/Makefile.in
source3/include/proto.h
source3/lib/util.c
source3/rpcclient/cmd_netlogon.c
source3/utils/rpctorture.c [new file with mode: 0644]

index ede81c2ea91ea27567e151e4e83eaa745c9dfe50..187f1c799b6ae2b85c1563d9b8d95ed4d1d5ea3a 100644 (file)
@@ -221,6 +221,16 @@ NMBLOOKUP_OBJ = utils/nmblookup.o $(PARAM_OBJ) $(UBIQX_OBJ) \
 SMBTORTURE_OBJ = utils/torture.o $(LIBSMB_OBJ) $(PARAM_OBJ) \
                  $(UBIQX_OBJ) $(LIB_OBJ)
 
+RPCTORTURE_OBJ = utils/rpctorture.o \
+             rpcclient/display.o \
+             rpcclient/cmd_lsarpc.o \
+             rpcclient/cmd_wkssvc.o \
+             rpcclient/cmd_samr.o \
+             rpcclient/cmd_srvsvc.o \
+             rpcclient/cmd_netlogon.o \
+             $(PARAM_OBJ) $(LIBSMB_OBJ) $(UBIQX_OBJ) $(LIB_OBJ) \
+             $(RPC_CLIENT_OBJ) $(RPC_PARSE_OBJ) $(PASSDB_OBJ)
+
 PROTO_OBJ = $(SMBD_OBJ) $(NMBD_OBJ) $(SWAT_OBJ) $(CLIENT_OBJ) \
            $(RPCCLIENT_OBJ) $(SMBWRAPPER_OBJ)
 
@@ -237,6 +247,8 @@ smbwrapper : CHECK bin/smbsh bin/smbwrapper.so @WRAP32@
 
 smbtorture : CHECK bin/smbtorture
 
+rpctorture : CHECK bin/rpctorture
+
 .SUFFIXES:
 .SUFFIXES: .c .o .po .po32
 
@@ -352,6 +364,10 @@ bin/smbtorture: $(SMBTORTURE_OBJ) bin/.dummy
        @echo Linking $@
        @$(CC) $(FLAGS) -o $@ $(SMBTORTURE_OBJ) $(LDFLAGS) $(LIBS)
 
+bin/rpctorture: $(RPCTORTURE_OBJ) bin/.dummy
+       @echo Linking $@
+       @$(CC) $(FLAGS) -o $@ $(RPCTORTURE_OBJ) $(LDFLAGS) $(LIBS)
+
 bin/smbwrapper.so: $(PICOBJS)
        @echo Linking shared library $@
        @$(LD) @LDSHFLAGS@ -o $@ $(PICOBJS) $(LIBS)
index d75832395d16a43c9e01dd8adffb4354120f3a11..e0d70a873c0912def968b68ed4515c094752c673 100644 (file)
@@ -231,6 +231,7 @@ BOOL user_in_list(char *user,char *list);
 
 char *tmpdir(void);
 BOOL is_a_socket(int fd);
+BOOL set_first_token(char *ptr);
 BOOL next_token(char **ptr,char *buff,char *sep, int bufsize);
 char **toktocliplist(int *ctok, char *sep);
 void set_socket_options(int fd, char *options);
index 58106acd46877fd31319781e26007d3221d6a439..d4f939e0813ea88719d8d3753868c7ce5f3ca8e9 100644 (file)
@@ -125,6 +125,11 @@ BOOL is_a_socket(int fd)
 
 static char *last_ptr=NULL;
 
+BOOL set_first_token(char *ptr)
+{
+       last_ptr = ptr;
+}
+
 /****************************************************************************
   Get the next token from a string, return False if none found
   handles double-quotes. 
index 8e65e9a7b56eeb3b617c4b0a3a5cdd4cff1cc8c5..2c8514b43e8def6fbf45b513d35e87033803ab67 100644 (file)
@@ -127,13 +127,7 @@ void cmd_netlogon_login_test(struct client_info *info)
        /* close the session */
        cli_nt_session_close(smb_cli);
 
-       if (res)
-       {
-               DEBUG(5,("cmd_nt_login: login test succeeded\n"));
-       }
-       else
-       {
-               DEBUG(5,("cmd_nt_login: login test failed\n"));
-       }
+       fprintf(out_hnd,"cmd_nt_login: login (%s) test succeeded: %s\n",
+               nt_user_name, BOOLSTR(res));
 }
 
diff --git a/source3/utils/rpctorture.c b/source3/utils/rpctorture.c
new file mode 100644 (file)
index 0000000..d1f0063
--- /dev/null
@@ -0,0 +1,520 @@
+/* 
+   Unix SMB/Netbios implementation.
+   Version 1.9.
+   SMB client
+   Copyright (C) Andrew Tridgell 1994-1998
+   
+   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.
+*/
+
+#ifdef SYSLOG
+#undef SYSLOG
+#endif
+
+#include "includes.h"
+
+#ifndef REGISTER
+#define REGISTER 0
+#endif
+
+extern pstring scope;
+extern pstring global_myname;
+
+extern pstring user_socket_options;
+
+
+extern pstring debugf;
+extern int DEBUGLEVEL;
+
+
+extern file_info def_finfo;
+
+#define CNV_LANG(s) dos2unix_format(s,False)
+#define CNV_INPUT(s) unix2dos_format(s,True)
+
+static struct cli_state smbcli;
+struct cli_state *smb_cli = &smbcli;
+
+FILE *out_hnd;
+
+static pstring password; /* local copy only, if one is entered */
+
+/****************************************************************************
+initialise smb client structure
+****************************************************************************/
+void rpcclient_init(void)
+{
+       bzero(smb_cli, sizeof(smb_cli));
+       cli_initialise(smb_cli);
+       smb_cli->capabilities |= CAP_NT_SMBS;
+}
+
+/****************************************************************************
+make smb client connection
+****************************************************************************/
+static BOOL rpcclient_connect(struct client_info *info)
+{
+       struct nmb_name calling;
+       struct nmb_name called;
+
+       make_nmb_name(&called , dns_to_netbios_name(info->dest_host ), info->name_type, scope);
+       make_nmb_name(&calling, dns_to_netbios_name(info->myhostname), 0x0            , scope);
+
+       if (!cli_establish_connection(smb_cli, 
+                                 info->dest_host, &info->dest_ip, 
+                                 &calling, &called,
+                                 info->share, info->svc_type,
+                                 False, True))
+       {
+               DEBUG(0,("rpcclient_connect: connection failed\n"));
+               cli_shutdown(smb_cli);
+               return False;
+       }
+
+       return True;
+}
+
+/****************************************************************************
+stop the smb connection(s?)
+****************************************************************************/
+static void rpcclient_stop(void)
+{
+       cli_shutdown(smb_cli);
+}
+
+/****************************************************************************
+  log in as an nt user, log out again. 
+****************************************************************************/
+void run_ntlogin_test(int num_ops, struct client_info *cli_info, struct cli_state *cli)
+{
+       pstring cmd;
+       int i;
+
+       /* establish connections.  nothing to stop these being re-established. */
+       rpcclient_connect(cli_info);
+
+       DEBUG(5,("rpcclient_connect: cli->fd:%d\n", cli->fd));
+       if (cli->fd <= 0)
+       {
+               fprintf(out_hnd, "warning: connection could not be established to %s<%02x>\n",
+                                cli_info->dest_host, cli_info->name_type);
+               return;
+       }
+       
+       for (i = 0; i < num_ops; i++)
+       {
+               slprintf(cmd, sizeof(cmd)-1, "%s %s", cli->user_name, password);
+               set_first_token(cmd);
+
+               cmd_netlogon_login_test(cli_info);
+       }
+
+       rpcclient_stop();
+
+}
+
+/****************************************************************************
+  runs n simultaneous functions.
+****************************************************************************/
+static void create_procs(int nprocs, int numops, 
+               struct client_info *cli_info, struct cli_state *cli,
+               void (*fn)(int, struct client_info *, struct cli_state *))
+{
+       int i, status;
+
+       for (i=0;i<nprocs;i++)
+       {
+               if (fork() == 0)
+               {
+                       int mypid = getpid();
+                       sys_srandom(mypid ^ time(NULL));
+                       fn(numops, cli_info, cli);
+                       fflush(out_hnd);
+                       _exit(0);
+               }
+       }
+
+       for (i=0;i<nprocs;i++)
+       {
+               waitpid(0, &status, 0);
+       }
+}
+/****************************************************************************
+usage on the program - OUT OF DATE!
+****************************************************************************/
+static void usage(char *pname)
+{
+  fprintf(out_hnd, "Usage: %s service <password> [-d debuglevel] [-l log] ",
+          pname);
+
+  fprintf(out_hnd, "\nVersion %s\n",VERSION);
+  fprintf(out_hnd, "\t-d debuglevel         set the debuglevel\n");
+  fprintf(out_hnd, "\t-l log basename.      Basename for log/debug files\n");
+  fprintf(out_hnd, "\t-n netbios name.      Use this name as my netbios name\n");
+  fprintf(out_hnd, "\t-m max protocol       set the max protocol level\n");
+  fprintf(out_hnd, "\t-I dest IP            use this IP to connect to\n");
+  fprintf(out_hnd, "\t-E                    write messages to stderr instead of stdout\n");
+  fprintf(out_hnd, "\t-U username           set the network username\n");
+  fprintf(out_hnd, "\t-W workgroup          set the workgroup name\n");
+  fprintf(out_hnd, "\t-t terminal code      terminal i/o code {sjis|euc|jis7|jis8|junet|hex}\n");
+  fprintf(out_hnd, "\n");
+}
+
+enum client_action
+{
+       CLIENT_NONE,
+       CLIENT_IPC,
+       CLIENT_SVC
+};
+
+/****************************************************************************
+  main program
+****************************************************************************/
+ int main(int argc,char *argv[])
+{
+       char *pname = argv[0];
+       int opt;
+       extern FILE *dbf;
+       extern char *optarg;
+       extern int optind;
+       static pstring servicesf = CONFIGFILE;
+       pstring term_code;
+       BOOL got_pass = False;
+       char *cmd_str="";
+       mode_t myumask = 0755;
+       enum client_action cli_action = CLIENT_NONE;
+       int nprocs = 1;
+       int numops = 100;
+
+       struct client_info cli_info;
+
+       out_hnd = stdout;
+
+       rpcclient_init();
+
+#ifdef KANJI
+       pstrcpy(term_code, KANJI);
+#else /* KANJI */
+       *term_code = 0;
+#endif /* KANJI */
+
+       if (!lp_load(servicesf,True, False, False))
+       {
+               fprintf(stderr, "Can't load %s - run testparm to debug it\n", servicesf);
+       }
+
+       codepage_initialise(lp_client_code_page());
+
+       DEBUGLEVEL = 0;
+
+       cli_info.put_total_size = 0;
+       cli_info.put_total_time_ms = 0;
+       cli_info.get_total_size = 0;
+       cli_info.get_total_time_ms = 0;
+
+       cli_info.dir_total = 0;
+       cli_info.newer_than = 0;
+       cli_info.archive_level = 0;
+       cli_info.print_mode = 1;
+
+       cli_info.translation = False;
+       cli_info.recurse_dir = False;
+       cli_info.lowercase = False;
+       cli_info.prompt = True;
+       cli_info.abort_mget = True;
+
+       cli_info.dest_ip.s_addr = 0;
+       cli_info.name_type = 0x20;
+
+       pstrcpy(cli_info.cur_dir , "\\");
+       pstrcpy(cli_info.file_sel, "");
+       pstrcpy(cli_info.base_dir, "");
+       pstrcpy(smb_cli->domain, "");
+       pstrcpy(smb_cli->user_name, "");
+       pstrcpy(cli_info.myhostname, "");
+       pstrcpy(cli_info.dest_host, "");
+
+       pstrcpy(cli_info.svc_type, "A:");
+       pstrcpy(cli_info.share, "");
+       pstrcpy(cli_info.service, "");
+
+       pstrcpy(cli_info.dom.level3_sid, "");
+       pstrcpy(cli_info.dom.level3_dom, "");
+       pstrcpy(cli_info.dom.level5_sid, "");
+       pstrcpy(cli_info.dom.level5_dom, "");
+
+       smb_cli->nt_pipe_fnum   = 0xffff;
+
+       setup_logging(pname, True);
+
+       TimeInit();
+       charset_initialise();
+
+       myumask = umask(0);
+       umask(myumask);
+
+       if (!get_myname(global_myname, NULL))
+       {
+               fprintf(stderr, "Failed to get my hostname.\n");
+       }
+
+       password[0] = 0;
+
+       if (argc < 2)
+       {
+               usage(pname);
+               exit(1);
+       }
+
+       if (*argv[1] != '-')
+       {
+               pstrcpy(cli_info.service, argv[1]);  
+               /* Convert any '/' characters in the service name to '\' characters */
+               string_replace( cli_info.service, '/','\\');
+               argc--;
+               argv++;
+
+               DEBUG(1,("service: %s\n", cli_info.service));
+
+               if (count_chars(cli_info.service,'\\') < 3)
+               {
+                       usage(pname);
+                       printf("\n%s: Not enough '\\' characters in service\n", cli_info.service);
+                       exit(1);
+               }
+
+               /*
+               if (count_chars(cli_info.service,'\\') > 3)
+               {
+                       usage(pname);
+                       printf("\n%s: Too many '\\' characters in service\n", cli_info.service);
+                       exit(1);
+               }
+               */
+
+               if (argc > 1 && (*argv[1] != '-'))
+               {
+                       got_pass = True;
+                       pstrcpy(password,argv[1]);  
+                       memset(argv[1],'X',strlen(argv[1]));
+                       argc--;
+                       argv++;
+               }
+
+               cli_action = CLIENT_SVC;
+       }
+
+       while ((opt = getopt(argc, argv,"s:B:O:M:S:i:N:o:n:d:l:hI:EB:U:L:t:m:W:T:D:c:")) != EOF)
+       {
+               switch (opt)
+               {
+                       case 'm':
+                       {
+                               /* FIXME ... max_protocol seems to be funny here */
+
+                               int max_protocol = 0;
+                               max_protocol = interpret_protocol(optarg,max_protocol);
+                               fprintf(stderr, "max protocol not currently supported\n");
+                               break;
+                       }
+
+                       case 'O':
+                       {
+                               pstrcpy(user_socket_options,optarg);
+                               break;  
+                       }
+
+                       case 'S':
+                       {
+                               pstrcpy(cli_info.dest_host,optarg);
+                               strupper(cli_info.dest_host);
+                               cli_action = CLIENT_IPC;
+                               break;
+                       }
+
+                       case 'B':
+                       {
+                               iface_set_default(NULL,optarg,NULL);
+                               break;
+                       }
+
+                       case 'i':
+                       {
+                               pstrcpy(scope, optarg);
+                               break;
+                       }
+
+                       case 'U':
+                       {
+                               char *lp;
+                               pstrcpy(smb_cli->user_name,optarg);
+                               if ((lp=strchr(smb_cli->user_name,'%')))
+                               {
+                                       *lp = 0;
+                                       pstrcpy(password,lp+1);
+                                       got_pass = True;
+                                       memset(strchr(optarg,'%')+1,'X',strlen(password));
+                               }
+                               break;
+                       }
+
+                       case 'W':
+                       {
+                               pstrcpy(smb_cli->domain,optarg);
+                               break;
+                       }
+
+                       case 'E':
+                       {
+                               dbf = stderr;
+                               break;
+                       }
+
+                       case 'I':
+                       {
+                               cli_info.dest_ip = *interpret_addr2(optarg);
+                               if (zero_ip(cli_info.dest_ip))
+                               {
+                                       exit(1);
+                               }
+                               break;
+                       }
+
+                       case 'N':
+                       {
+                               nprocs = atoi(optarg);
+                               break;
+                       }
+
+                       case 'o':
+                       {
+                               numops = atoi(optarg);
+                               break;
+                       }
+
+                       case 'n':
+                       {
+                               fstrcpy(global_myname, optarg);
+                               break;
+                       }
+
+                       case 'd':
+                       {
+                               if (*optarg == 'A')
+                                       DEBUGLEVEL = 10000;
+                               else
+                                       DEBUGLEVEL = atoi(optarg);
+                               break;
+                       }
+
+                       case 'l':
+                       {
+                               slprintf(debugf, sizeof(debugf)-1,
+                                        "%s.client",optarg);
+                               break;
+                       }
+
+                       case 'c':
+                       {
+                               cmd_str = optarg;
+                               got_pass = True;
+                               break;
+                       }
+
+                       case 'h':
+                       {
+                               usage(pname);
+                               exit(0);
+                               break;
+                       }
+
+                       case 's':
+                       {
+                               pstrcpy(servicesf, optarg);
+                               break;
+                       }
+
+                       case 't':
+                       {
+                               pstrcpy(term_code, optarg);
+                               break;
+                       }
+
+                       default:
+                       {
+                               usage(pname);
+                               exit(1);
+                               break;
+                       }
+               }
+       }
+
+       if (cli_action == CLIENT_NONE)
+       {
+               usage(pname);
+               exit(1);
+       }
+
+       strupper(global_myname);
+       fstrcpy(cli_info.myhostname, global_myname);
+
+       DEBUG(3,("%s client started (version %s)\n",timestring(),VERSION));
+
+       if (*smb_cli->domain == 0)
+       {
+               pstrcpy(smb_cli->domain,lp_workgroup());
+       }
+       strupper(smb_cli->domain);
+
+       load_interfaces();
+
+       if (cli_action == CLIENT_IPC)
+       {
+               pstrcpy(cli_info.share, "IPC$");
+               pstrcpy(cli_info.svc_type, "IPC");
+       }
+
+       fstrcpy(cli_info.mach_acct, cli_info.myhostname);
+       strupper(cli_info.mach_acct);
+       fstrcat(cli_info.mach_acct, "$");
+
+       /* set the password cache info */
+       if (got_pass)
+       {
+               if (password[0] == 0)
+               {
+                       pwd_set_nullpwd(&(smb_cli->pwd));
+               }
+               else
+               {
+                       pwd_make_lm_nt_16(&(smb_cli->pwd), password); /* generate 16 byte hashes */
+               }
+       }
+       else 
+       {
+               char *pwd = getpass("Enter Password:");
+               safe_strcpy(password, pwd, sizeof(password));
+               pwd_make_lm_nt_16(&(smb_cli->pwd), password); /* generate 16 byte hashes */
+       }
+
+       if (password[0] != 0)
+       {
+               create_procs(nprocs, numops, &cli_info, smb_cli, run_ntlogin_test);
+       }
+
+       fflush(out_hnd);
+
+       return(0);
+}