r25026: Move param/param.h out of includes.h
[jelmer/samba4-debian.git] / source / torture / smbtorture.c
index 41ee1461d13b13cd13eadb9352ada8f7d82da48d..a148081fb3b9f8300464355eae588ef1531f8aa2 100644 (file)
@@ -6,7 +6,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,
@@ -15,8 +15,7 @@
    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"
@@ -35,6 +34,7 @@
 #include "build.h"
 #include "lib/util/dlinklist.h"
 #include "librpc/rpc/dcerpc.h"
+#include "param/param.h"
 
 static bool run_matching(struct torture_context *torture,
                                                 const char *prefix, 
@@ -64,9 +64,11 @@ static bool run_matching(struct torture_context *torture,
 
                for (c = suite->children; c; c = c->next) {
                        asprintf(&name, "%s-%s", prefix, c->name);
+
                        if (gen_fnmatch(expr, name) == 0) {
                                *matched = true;
                                init_iconv();
+                               torture->active_testname = talloc_strdup(torture, prefix);
                                ret &= torture_run_suite(torture, c);
                                free(name);
                                continue;
@@ -82,7 +84,9 @@ static bool run_matching(struct torture_context *torture,
                        if (gen_fnmatch(expr, name) == 0) {
                                *matched = true;
                                init_iconv();
+                               torture->active_testname = talloc_strdup(torture, prefix);
                                ret &= torture_run_tcase(torture, t);
+                               talloc_free(torture->active_testname);
                        }
                        free(name);
                }
@@ -159,7 +163,24 @@ static void parse_dns(const char *dns)
 
 }
 
-static void usage(poptContext pc)
+static void print_test_list(void)
+{
+       struct torture_suite *o;
+       struct torture_suite *s;
+       struct torture_tcase *t;
+
+       for (o = torture_root->children; o; o = o->next) {
+               for (s = o->children; s; s = s->next) {
+                       printf("%s-%s\n", o->name, s->name);
+               }
+
+               for (t = o->testcases; t; t = t->next) {
+                       printf("%s-%s\n", o->name, t->name);
+               }
+       }
+}
+
+_NORETURN_ static void usage(poptContext pc)
 {
        struct torture_suite *o;
        struct torture_suite *s;
@@ -218,6 +239,11 @@ static void usage(poptContext pc)
 
        printf("Tests are:");
 
+       if (torture_root == NULL) {
+           printf("NO TESTS LOADED\n");
+           exit(1);
+       }
+
        for (o = torture_root->children; o; o = o->next) {
                printf("\n%s (%s):\n  ", o->description, o->name);
 
@@ -246,19 +272,7 @@ static void usage(poptContext pc)
        exit(1);
 }
 
-static bool is_binding_string(const char *binding_string)
-{
-       TALLOC_CTX *mem_ctx = talloc_named_const(NULL, 0, "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)
+_NORETURN_ static void max_runtime_handler(int sig)
 {
        DEBUG(0,("maximum runtime exceeded for smbtorture - terminating\n"));
        exit(1);
@@ -307,13 +321,29 @@ static void simple_comment (struct torture_context *test,
        printf("%s", comment);
 }
 
+static void simple_warning(struct torture_context *test, 
+                                                  const char *comment)
+{
+       fprintf(stderr, "WARNING: %s\n", comment);
+}
+
 const static struct torture_ui_ops std_ui_ops = {
        .comment = simple_comment,
+       .warning = simple_warning,
        .suite_start = simple_suite_start,
        .suite_finish = simple_suite_finish,
        .test_result = simple_test_result
 };
 
+static void subunit_init(struct torture_context *ctx) 
+{
+       /* FIXME: register segv and bus handler */
+}
+
+static void subunit_suite_start(struct torture_context *ctx,
+                                                          struct torture_suite *suite)
+{
+}
 
 static void subunit_test_start (struct torture_context *ctx, 
                                                            struct torture_tcase *tcase,
@@ -340,7 +370,7 @@ static void subunit_test_result (struct torture_context *context,
                break;
        }
        if (reason)
-               printf(" [ %s ]", reason);
+               printf(" [\n%s\n]", reason);
        printf("\n");
 }
 
@@ -351,44 +381,11 @@ static void subunit_comment (struct torture_context *test,
 }
 
 const static struct torture_ui_ops subunit_ui_ops = {
+       .init = subunit_init,
        .comment = subunit_comment,
        .test_start = subunit_test_start,
-       .test_result = subunit_test_result
-};
-
-static void harness_test_start (struct torture_context *ctx, 
-                                                           struct torture_tcase *tcase,
-                                                               struct torture_test *test)
-{
-}
-
-static void harness_test_result (struct torture_context *context, 
-                                                                enum torture_result res, const char *reason)
-{
-       switch (res) {
-       case TORTURE_OK:
-               printf("ok %s - %s\n", context->active_test->name, reason);
-               break;
-       case TORTURE_FAIL:
-       case TORTURE_ERROR:
-               printf("not ok %s - %s\n", context->active_test->name, reason);
-               break;
-       case TORTURE_SKIP:
-               printf("skip %s - %s\n", context->active_test->name, reason);
-               break;
-       }
-}
-
-static void harness_comment (struct torture_context *test, 
-                                                        const char *comment)
-{
-       printf("# %s\n", comment);
-}
-
-const static struct torture_ui_ops harness_ui_ops = {
-       .comment = harness_comment,
-       .test_start = harness_test_start,
-       .test_result = harness_test_result
+       .test_result = subunit_test_result,
+       .suite_start = subunit_suite_start
 };
 
 static void quiet_suite_start(struct torture_context *ctx,
@@ -425,33 +422,6 @@ const static struct torture_ui_ops quiet_ui_ops = {
        .test_result = quiet_test_result
 };
 
-void run_recipe(struct torture_context *tctx, const char *recipe)
-{
-       int numlines, i, ret;
-       char **lines;
-
-       lines = file_lines_load(recipe, &numlines, NULL);
-       if (lines == NULL) {
-               fprintf(stderr, "Unable to load file %s\n", recipe);
-               return;
-       }
-
-       for (i = 0; i < numlines; i++) {
-               int argc;
-               const char **argv;
-
-               ret = poptParseArgvString(lines[i], &argc, &argv);
-               if (ret != 0) {
-                       fprintf(stderr, "Error parsing line\n");
-                       continue;
-               }
-
-               run_test(tctx, argv[0]);
-       }
-
-       talloc_free(lines);
-}
-
 void run_shell(struct torture_context *tctx)
 {
        char *cline;
@@ -511,28 +481,36 @@ int main(int argc,char *argv[])
        char **argv_new;
        poptContext pc;
        static const char *target = "other";
-       const char **subunit_dir;
+       struct dcerpc_binding *binding_struct;
+       NTSTATUS status;
        int shell = False;
        static const char *ui_ops_name = "simple";
-       enum {OPT_LOADFILE=1000,OPT_UNCLIST,OPT_TIMELIMIT,OPT_DNS,
-             OPT_DANGEROUS,OPT_SMB_PORTS,OPT_ASYNC};
+       const char *basedir = NULL;
+       const char *extra_module = NULL;
+       static int list_tests = 0;
+       char *host = NULL, *share = NULL;
+       enum {OPT_LOADFILE=1000,OPT_UNCLIST,OPT_TIMELIMIT,OPT_DNS, OPT_LIST,
+             OPT_DANGEROUS,OPT_SMB_PORTS,OPT_ASYNC,OPT_NUMPROGS};
        
        struct poptOption long_options[] = {
                POPT_AUTOHELP
-               {"format", 0, POPT_ARG_STRING, &ui_ops_name, 0, "Output format (one of: simple, subunit, harness)", NULL },
+               {"format", 0, POPT_ARG_STRING, &ui_ops_name, 0, "Output format (one of: simple, subunit)", NULL },
                {"smb-ports",   'p', POPT_ARG_STRING, NULL,     OPT_SMB_PORTS,  "SMB ports",    NULL},
+               {"basedir",       0, POPT_ARG_STRING, &basedir, 0, "base directory", "BASEDIR" },
                {"seed",          0, POPT_ARG_INT,  &torture_seed,      0,      "seed",         NULL},
+               {"num-progs",     0, POPT_ARG_INT,  NULL,       OPT_NUMPROGS,   "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},
-               {"show-all",      0, POPT_ARG_NONE, &torture_showall,   0,      "show all",     NULL},
                {"loadfile",      0, POPT_ARG_STRING,   NULL,   OPT_LOADFILE,   "loadfile",     NULL},
+               {"list",          0, POPT_ARG_NONE, &list_tests, 0, NULL, NULL },
                {"unclist",       0, POPT_ARG_STRING,   NULL,   OPT_UNCLIST,    "unclist",      NULL},
-               {"timelimit",   't', POPT_ARG_STRING,   NULL,   OPT_TIMELIMIT,  "timelimit",    NULL},
+               {"timelimit",   't', POPT_ARG_INT,      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,
                 "run dangerous tests (eg. wiping out password database)", NULL},
-               {"shell",               's', POPT_ARG_NONE, &shell, True, "Run shell", NULL},
+               {"load-module",  0,  POPT_ARG_STRING, &extra_module,     0, "load tests from DSO file",    "SOFILE"},
+               {"shell",               0, POPT_ARG_NONE, &shell, True, "Run shell", NULL},
                {"target",              'T', POPT_ARG_STRING, &target, 0, "samba3|samba4|other", NULL},
                {"async",       'a', POPT_ARG_NONE,     NULL,   OPT_ASYNC,
                 "run async tests", NULL},
@@ -568,6 +546,9 @@ int main(int argc,char *argv[])
                case OPT_TIMELIMIT:
                        lp_set_cmdline("torture:timelimit", poptGetOptArg(pc));
                        break;
+               case OPT_NUMPROGS:
+                       lp_set_cmdline("torture:nprocs", poptGetOptArg(pc));
+                       break;
                case OPT_DNS:
                        parse_dns(poptGetOptArg(pc));
                        break;
@@ -585,10 +566,8 @@ int main(int argc,char *argv[])
 
        if (strcmp(target, "samba3") == 0) {
                lp_set_cmdline("torture:samba3", "true");
-               lp_set_cmdline("torture:knownfail", "samba3-knownfail");
        } else if (strcmp(target, "samba4") == 0) {
                lp_set_cmdline("torture:samba4", "true");
-               lp_set_cmdline("torture:knownfail", "samba4-knownfail");
        }
 
        if (max_runtime) {
@@ -601,15 +580,27 @@ int main(int argc,char *argv[])
                alarm(max_runtime);
        }
 
-       torture_init();
        ldb_global_init();
 
-       subunit_dir = lp_parm_string_list(-1, "torture", "subunitdir", ":");
-       if (subunit_dir == NULL) 
-               torture_subunit_load_testsuites(dyn_TORTUREDIR, true, NULL);
-       else {
-               for (i = 0; subunit_dir[i]; i++)
-                       torture_subunit_load_testsuites(subunit_dir[i], true, NULL);
+       if (extra_module != NULL) {
+           init_module_fn fn = load_module(talloc_autofree_context(), poptGetOptArg(pc));
+
+           if (fn == NULL) 
+               d_printf("Unable to load module from %s\n", poptGetOptArg(pc));
+           else {
+               status = fn();
+               if (NT_STATUS_IS_ERR(status)) {
+                   d_printf("Error initializing module %s: %s\n", 
+                            poptGetOptArg(pc), nt_errstr(status));
+               }
+           }
+       } else { 
+               torture_init();
+       }
+
+       if (list_tests) {
+               print_test_list();
+               return 0;
        }
 
        if (torture_seed == 0) {
@@ -634,29 +625,26 @@ int main(int argc,char *argv[])
        }
 
        /* 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)) {
+       if (!smbcli_parse_unc(argv_new[1], NULL, &host, &share)) {
+               status = dcerpc_parse_binding(talloc_autofree_context(), argv_new[1], &binding_struct);
+               if (NT_STATUS_IS_ERR(status)) {
                        d_printf("Invalid option: %s is not a valid torture target (share or binding string)\n\n", argv_new[1]);
                        usage(pc);
+                       return false;
                }
-
+               lp_set_cmdline("torture:host", binding_struct->host);
+               lp_set_cmdline("torture:share", "IPC$");
+               lp_set_cmdline("torture:binding", argv_new[1]);
+       } else {
                lp_set_cmdline("torture:host", host);
                lp_set_cmdline("torture:share", share);
-               asprintf(&binding, "ncacn_np:%s", host);
-               lp_set_cmdline("torture:binding", binding);
+               lp_set_cmdline("torture:binding", host);
        }
 
        if (!strcmp(ui_ops_name, "simple")) {
                ui_ops = &std_ui_ops;
        } else if (!strcmp(ui_ops_name, "subunit")) {
                ui_ops = &subunit_ui_ops;
-       } else if (!strcmp(ui_ops_name, "harness")) {
-               ui_ops = &harness_ui_ops;
        } else if (!strcmp(ui_ops_name, "quiet")) {
                ui_ops = &quiet_ui_ops;
        } else {
@@ -664,67 +652,35 @@ int main(int argc,char *argv[])
                exit(1);
        }
 
-       torture = torture_context_init(talloc_autofree_context(), 
-                               lp_parm_string(-1, "torture", "knownfail"), ui_ops);
+       torture = torture_context_init(talloc_autofree_context(), ui_ops);
+       if (basedir != NULL) {
+               if (basedir[0] != '/') {
+                       fprintf(stderr, "Please specify an absolute path to --basedir\n");
+                       return 1;
+               }
+               torture->outputdir = basedir;
+       } else {
+               char *pwd = talloc_size(torture, PATH_MAX);
+               if (!getcwd(pwd, PATH_MAX)) {
+                       fprintf(stderr, "Unable to determine current working directory\n");
+                       return 1;
+               }
+               torture->outputdir = pwd;
+       }
 
        if (argc_new == 0) {
                printf("You must specify a test to run, or 'ALL'\n");
        } else if (shell) {
                run_shell(torture);
        } else {
-               int total;
-               double rate;
-               int unexpected_failures;
                for (i=2;i<argc_new;i++) {
-                       if (argv_new[i][0] == '@') {
-                               run_recipe(torture, argv_new[i]+1);
-                       } else if (!run_test(torture, argv_new[i])) {
+                       if (!run_test(torture, argv_new[i])) {
                                correct = false;
                        }
                }
-
-               unexpected_failures = str_list_length(torture->results.unexpected_failures);
-
-               total = torture->results.skipped+torture->results.success+torture->results.failed+torture->results.errors;
-               if (total == 0) {
-                       printf("No tests run.\n");
-               } else {
-                       rate = ((total - unexpected_failures - torture->results.errors) * (100.0 / total));
-               
-                       printf("Tests: %d, Failures: %d", total, torture->results.failed);
-                       if (torture->results.failed - unexpected_failures) {
-                               printf(" (%d expected)", torture->results.failed - unexpected_failures);
-                       }
-                       printf(", Errors: %d, Skipped: %d. Success rate: %.2f%%\n",
-                          torture->results.errors, torture->results.skipped, rate);
-               }
-
-               if (unexpected_failures) {
-                       printf("The following tests failed:\n");
-                       for (i = 0; torture->results.unexpected_failures[i]; i++) {
-                               printf("  %s\n", torture->results.unexpected_failures[i]);
-                       }
-                       printf("\n");
-               }
-
-               if (str_list_length(torture->results.unexpected_errors)) {
-                       printf("Errors occurred while running the following tests:\n");
-                       for (i = 0; torture->results.unexpected_errors[i]; i++) {
-                               printf("  %s\n", torture->results.unexpected_errors[i]);
-                       }
-                       printf("\n");
-               }
-
-               if (str_list_length(torture->results.unexpected_successes)) {
-                       printf("The following tests were expected to fail but succeeded:\n");
-                       for (i = 0; torture->results.unexpected_successes[i]; i++) {
-                               printf("  %s\n", torture->results.unexpected_successes[i]);
-                       }
-                       printf("\n");
-               }
        }
 
-       if (torture->results.returncode) {
+       if (torture->returncode && correct) {
                return(0);
        } else {
                return(1);