diff options
Diffstat (limited to 'source4/torture/torture.c')
-rw-r--r-- | source4/torture/torture.c | 344 |
1 files changed, 6 insertions, 338 deletions
diff --git a/source4/torture/torture.c b/source4/torture/torture.c index 4003b52c92..a54c760dc8 100644 --- a/source4/torture/torture.c +++ b/source4/torture/torture.c @@ -48,16 +48,15 @@ #include "build.h" #include "dlinklist.h" -int torture_nprocs=4; +_PUBLIC_ int torture_nprocs=4; _PUBLIC_ int torture_numops=10; _PUBLIC_ int torture_entries=1000; _PUBLIC_ int torture_failures=1; _PUBLIC_ int torture_seed=0; +_PUBLIC_ BOOL use_oplocks; static int procnum; /* records process count number when forking */ static struct smbcli_state *current_cli; -static BOOL use_oplocks; static BOOL use_level_II_oplocks; -#define MAX_COLS 80 /* FIXME: Determine this at run-time */ _PUBLIC_ BOOL torture_showall = False; @@ -761,12 +760,7 @@ static void register_builtin_ops(void) } -static struct torture_op { - const char *name; - BOOL (*fn)(void); - BOOL (*multi_fn)(struct smbcli_state *, int ); - struct torture_op *prev, *next; -}* torture_ops = NULL;; +struct torture_op *torture_ops = NULL; static struct torture_op *find_torture_op(const char *name) { @@ -800,231 +794,11 @@ _PUBLIC_ NTSTATUS register_torture_op(const char *name, BOOL (*fn)(void), BOOL ( return NT_STATUS_OK; } -/**************************************************************************** -run a specified test or "ALL" -****************************************************************************/ -static BOOL run_test(const char *name) -{ - BOOL ret = True; - struct torture_op *o; - BOOL matched = False; - - if (strequal(name,"ALL")) { - for (o = torture_ops; o; o = o->next) { - if (!run_test(o->name)) { - ret = False; - } - } - return ret; - } - - for (o = torture_ops; o; o = o->next) { - if (gen_fnmatch(name, o->name) == 0) { - double t; - matched = True; - init_iconv(); - printf("Running %s\n", o->name); - if (o->multi_fn) { - BOOL result = False; - t = torture_create_procs(o->multi_fn, - &result); - if (!result) { - ret = False; - printf("TEST %s FAILED!\n", o->name); - } - - } else { - struct timeval tv = timeval_current(); - if (!o->fn()) { - ret = False; - printf("TEST %s FAILED!\n", o->name); - } - t = timeval_elapsed(&tv); - } - printf("%s took %g secs\n\n", o->name, t); - } - } - - if (!matched) { - printf("Unknown torture operation '%s'\n", name); - } - - return ret; -} - - -static void parse_dns(const char *dns) -{ - char *userdn, *basedn, *secret; - char *p, *d; - - /* retrievieng the userdn */ - p = strchr_m(dns, '#'); - if (!p) { - lp_set_cmdline("torture:ldap_userdn", ""); - lp_set_cmdline("torture:ldap_basedn", ""); - lp_set_cmdline("torture:ldap_secret", ""); - return; - } - userdn = strndup(dns, p - dns); - lp_set_cmdline("torture:ldap_userdn", userdn); - - /* retrieve the basedn */ - d = p + 1; - p = strchr_m(d, '#'); - if (!p) { - lp_set_cmdline("torture:ldap_basedn", ""); - lp_set_cmdline("torture:ldap_secret", ""); - return; - } - basedn = strndup(d, p - d); - lp_set_cmdline("torture:ldap_basedn", basedn); - - /* retrieve the secret */ - p = p + 1; - if (!p) { - lp_set_cmdline("torture:ldap_secret", ""); - return; - } - secret = strdup(p); - lp_set_cmdline("torture:ldap_secret", secret); - - printf ("%s - %s - %s\n", userdn, basedn, secret); - -} - -static void usage(poptContext pc) -{ - struct torture_op *o; - int i; - - poptPrintUsage(pc, stdout, 0); - printf("\n"); - - printf("The binding format is:\n\n"); - - printf(" TRANSPORT:host[flags]\n\n"); - - printf(" where TRANSPORT is either ncacn_np for SMB or ncacn_ip_tcp for RPC/TCP\n\n"); - - printf(" 'host' is an IP or hostname or netbios name. If the binding string\n"); - printf(" identifies the server side of an endpoint, 'host' may be an empty\n"); - printf(" string.\n\n"); - - printf(" 'flags' can include a SMB pipe name if using the ncacn_np transport or\n"); - printf(" a TCP port number if using the ncacn_ip_tcp transport, otherwise they\n"); - printf(" will be auto-determined.\n\n"); - - printf(" other recognised flags are:\n\n"); - - printf(" sign : enable ntlmssp signing\n"); - printf(" seal : enable ntlmssp sealing\n"); - printf(" connect : enable rpc connect level auth (auth, but no sign or seal)\n"); - printf(" validate: enable the NDR validator\n"); - printf(" print: enable debugging of the packets\n"); - printf(" bigendian: use bigendian RPC\n"); - printf(" padcheck: check reply data for non-zero pad bytes\n\n"); - - printf(" For example, these all connect to the samr pipe:\n\n"); - - printf(" ncacn_np:myserver\n"); - printf(" ncacn_np:myserver[samr]\n"); - printf(" ncacn_np:myserver[\\pipe\\samr]\n"); - printf(" ncacn_np:myserver[/pipe/samr]\n"); - printf(" ncacn_np:myserver[samr,sign,print]\n"); - printf(" ncacn_np:myserver[\\pipe\\samr,sign,seal,bigendian]\n"); - printf(" ncacn_np:myserver[/pipe/samr,seal,validate]\n"); - printf(" ncacn_np:\n"); - printf(" ncacn_np:[/pipe/samr]\n\n"); - - printf(" ncacn_ip_tcp:myserver\n"); - printf(" ncacn_ip_tcp:myserver[1024]\n"); - printf(" ncacn_ip_tcp:myserver[1024,sign,seal]\n\n"); - - printf("The unc format is:\n\n"); - - printf(" //server/share\n\n"); - - printf("tests are:\n"); - - i = 0; - for (o = torture_ops; o; o = o->next) { - if (i + strlen(o->name) >= MAX_COLS) { - printf("\n"); - i = 0; - } - i+=printf("%s ", o->name); - } - printf("\n\n"); - - printf("default test is ALL\n"); - - exit(1); -} - -static BOOL is_binding_string(const char *binding_string) -{ - TALLOC_CTX *mem_ctx = talloc_init("is_binding_string"); - struct dcerpc_binding *binding_struct; - NTSTATUS status; - - status = dcerpc_parse_binding(mem_ctx, binding_string, &binding_struct); - - talloc_free(mem_ctx); - return NT_STATUS_IS_OK(status); -} - -static void max_runtime_handler(int sig) -{ - DEBUG(0,("maximum runtime exceeded for smbtorture - terminating\n")); - exit(1); -} - -/**************************************************************************** - main program -****************************************************************************/ - int main(int argc,char *argv[]) +int torture_init(void) { - int opt, i; - char *p; - BOOL correct = True; - int max_runtime=0; - int argc_new; - char **argv_new; - poptContext pc; - enum {OPT_LOADFILE=1000,OPT_UNCLIST,OPT_TIMELIMIT,OPT_DNS, - OPT_DANGEROUS,OPT_SMB_PORTS}; - init_module_fn static_init[] = STATIC_smbtorture_MODULES; + init_module_fn static_init[] = STATIC_torture_MODULES; init_module_fn *shared_init = load_samba_modules(NULL, "torture"); - struct poptOption long_options[] = { - POPT_AUTOHELP - {"smb-ports", 'p', POPT_ARG_STRING, NULL, OPT_SMB_PORTS, "SMB ports", NULL}, - {"seed", 0, POPT_ARG_INT, &torture_seed, 0, "seed", NULL}, - {"num-progs", 0, POPT_ARG_INT, &torture_nprocs, 0, "num progs", NULL}, - {"num-ops", 0, POPT_ARG_INT, &torture_numops, 0, "num ops", NULL}, - {"entries", 0, POPT_ARG_INT, &torture_entries, 0, "entries", NULL}, - {"use-oplocks", 'L', POPT_ARG_NONE, &use_oplocks, 0, "use oplocks", NULL}, - {"show-all", 0, POPT_ARG_NONE, &torture_showall, 0, "show all", NULL}, - {"loadfile", 0, POPT_ARG_STRING, NULL, OPT_LOADFILE, "loadfile", NULL}, - {"unclist", 0, POPT_ARG_STRING, NULL, OPT_UNCLIST, "unclist", NULL}, - {"timelimit", 't', POPT_ARG_STRING, NULL, OPT_TIMELIMIT, "timelimit", NULL}, - {"failures", 'f', POPT_ARG_INT, &torture_failures, 0, "failures", NULL}, - {"parse-dns", 'D', POPT_ARG_STRING, NULL, OPT_DNS, "parse-dns", NULL}, - {"dangerous", 'X', POPT_ARG_NONE, NULL, OPT_DANGEROUS, "dangerous", NULL}, - {"maximum-runtime", 0, POPT_ARG_INT, &max_runtime, 0, - "set maximum time for smbtorture to live", "seconds"}, - POPT_COMMON_SAMBA - POPT_COMMON_CONNECTION - POPT_COMMON_CREDENTIALS - POPT_COMMON_VERSION - POPT_TABLEEND - }; - -#ifdef HAVE_SETBUFFER - setbuffer(stdout, NULL, 0); -#endif - register_builtin_ops(); run_init_functions(static_init); @@ -1032,111 +806,5 @@ static void max_runtime_handler(int sig) talloc_free(shared_init); - /* we are never interested in SIGPIPE */ - BlockSignals(True,SIGPIPE); - - pc = poptGetContext("smbtorture", argc, (const char **) argv, long_options, - POPT_CONTEXT_KEEP_FIRST); - - poptSetOtherOptionHelp(pc, "<binding>|<unc> TEST1 TEST2 ..."); - - while((opt = poptGetNextOpt(pc)) != -1) { - switch (opt) { - case OPT_LOADFILE: - lp_set_cmdline("torture:loadfile", poptGetOptArg(pc)); - break; - case OPT_UNCLIST: - lp_set_cmdline("torture:unclist", poptGetOptArg(pc)); - break; - case OPT_TIMELIMIT: - lp_set_cmdline("torture:timelimit", poptGetOptArg(pc)); - break; - case OPT_DNS: - parse_dns(poptGetOptArg(pc)); - break; - case OPT_DANGEROUS: - lp_set_cmdline("torture:dangerous", "Yes"); - break; - case OPT_SMB_PORTS: - lp_set_cmdline("smb ports", poptGetOptArg(pc)); - break; - default: - d_printf("Invalid option %s: %s\n", - poptBadOption(pc, 0), poptStrerror(opt)); - usage(pc); - exit(1); - } - } - - if (max_runtime) { - /* this will only work if nobody else uses alarm(), - which means it won't work for some tests, but we - can't use the event context method we use for smbd - as so many tests create their own event - context. This will at least catch most cases. */ - signal(SIGALRM, max_runtime_handler); - alarm(max_runtime); - } - - ldb_global_init(); - - if (torture_seed == 0) { - torture_seed = time(NULL); - } - printf("Using seed %d\n", torture_seed); - srandom(torture_seed); - - 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; - } - } - - if (argc_new < 3) { - usage(pc); - exit(1); - } - - for(p = argv_new[1]; *p; p++) { - if(*p == '\\') - *p = '/'; - } - - /* see if its a RPC transport specifier */ - if (is_binding_string(argv_new[1])) { - lp_set_cmdline("torture:binding", argv_new[1]); - } else { - char *binding = NULL; - char *host = NULL, *share = NULL; - - if (!smbcli_parse_unc(argv_new[1], NULL, &host, &share)) { - d_printf("Invalid option: %s is not a valid torture target (share or binding string)\n\n", argv_new[1]); - usage(pc); - } - - lp_set_cmdline("torture:host", host); - lp_set_cmdline("torture:share", share); - asprintf(&binding, "ncacn_np:%s", host); - lp_set_cmdline("torture:binding", binding); - } - - if (argc_new == 0) { - printf("You must specify a test to run, or 'ALL'\n"); - } else { - for (i=2;i<argc_new;i++) { - if (!run_test(argv_new[i])) { - correct = False; - } - } - } - - if (correct) { - return(0); - } else { - return(1); - } + return 0; } |