source4/torture/smbtorture.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    SMB torture tester
   4    Copyright (C) Andrew Tridgell 1997-2003
   5    Copyright (C) Jelmer Vernooij 2006
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "lib/cmdline/popt_common.h"
  23 #include "system/time.h"
  24 #include "system/wait.h"
  25 #include "system/filesys.h"
  26 #include "system/readline.h"
  27 #include "lib/smbreadline/smbreadline.h"
  28 #include "libcli/libcli.h"
  29 #include "lib/ldb/include/ldb.h"
  30 #include "lib/events/events.h"
  31 #include "dynconfig.h"
  32
  33 #include "torture/torture.h"
  34 #include "build.h"
  35 #include "lib/util/dlinklist.h"
  36 #include "librpc/rpc/dcerpc.h"
  37
  38 static bool run_matching(struct torture_context *torture,
  39                                                  const char *prefix,
  40                                                  const char *expr,
  41                                                  struct torture_suite *suite,
  42                                                  bool *matched)
  43 {
  44         bool ret = true;
  45
  46         if (suite == NULL) {
  47                 struct torture_suite *o;
  48
  49                 for (o = torture_root->children; o; o = o->next) {
  50                         if (gen_fnmatch(expr, o->name) == 0) {
  51                                 *matched = true;
  52                                 init_iconv();
  53                                 ret &= torture_run_suite(torture, o);
  54                                 continue;
  55                         }
  56
  57                         ret &= run_matching(torture, o->name, expr, o, matched);
  58                 }
  59         } else {
  60                 char *name;
  61                 struct torture_suite *c;
  62                 struct torture_tcase *t;
  63
  64                 for (c = suite->children; c; c = c->next) {
  65                         asprintf(&name, "%s-%s", prefix, c->name);
  66
  67                         if (gen_fnmatch(expr, name) == 0) {
  68                                 *matched = true;
  69                                 init_iconv();
  70                                 torture->active_testname = talloc_strdup(torture, prefix);
  71                                 ret &= torture_run_suite(torture, c);
  72                                 free(name);
  73                                 continue;
  74                         }
  75
  76                         ret &= run_matching(torture, name, expr, c, matched);
  77
  78                         free(name);
  79                 }
  80
  81                 for (t = suite->testcases; t; t = t->next) {
  82                         asprintf(&name, "%s-%s", prefix, t->name);
  83                         if (gen_fnmatch(expr, name) == 0) {
  84                                 *matched = true;
  85                                 init_iconv();
  86                                 torture->active_testname = talloc_strdup(torture, prefix);
  87                                 ret &= torture_run_tcase(torture, t);
  88                                 talloc_free(torture->active_testname);
  89                         }
  90                         free(name);
  91                 }
  92         }
  93
  94         return ret;
  95 }
  96
  97 #define MAX_COLS 80 /* FIXME: Determine this at run-time */
  98
  99 /****************************************************************************
 100 run a specified test or "ALL"
 101 ****************************************************************************/
 102 static bool run_test(struct torture_context *torture, const char *name)
 103 {
 104         bool ret = true;
 105         bool matched = false;
 106         struct torture_suite *o;
 107
 108         if (strequal(name, "ALL")) {
 109                 for (o = torture_root->children; o; o = o->next) {
 110                         ret &= torture_run_suite(torture, o);
 111                 }
 112                 return ret;
 113         }
 114
 115         ret = run_matching(torture, NULL, name, NULL, &matched);
 116
 117         if (!matched) {
 118                 printf("Unknown torture operation '%s'\n", name);
 119                 return false;
 120         }
 121
 122         return ret;
 123 }
 124
 125 static void parse_dns(const char *dns)
 126 {
 127         char *userdn, *basedn, *secret;
 128         char *p, *d;
 129
 130         /* retrievieng the userdn */
 131         p = strchr_m(dns, '#');
 132         if (!p) {
 133                 lp_set_cmdline("torture:ldap_userdn", "");
 134                 lp_set_cmdline("torture:ldap_basedn", "");
 135                 lp_set_cmdline("torture:ldap_secret", "");
 136                 return;
 137         }
 138         userdn = strndup(dns, p - dns);
 139         lp_set_cmdline("torture:ldap_userdn", userdn);
 140
 141         /* retrieve the basedn */
 142         d = p + 1;
 143         p = strchr_m(d, '#');
 144         if (!p) {
 145                 lp_set_cmdline("torture:ldap_basedn", "");
 146                 lp_set_cmdline("torture:ldap_secret", "");
 147                 return;
 148         }
 149         basedn = strndup(d, p - d);
 150         lp_set_cmdline("torture:ldap_basedn", basedn);
 151
 152         /* retrieve the secret */
 153         p = p + 1;
 154         if (!p) {
 155                 lp_set_cmdline("torture:ldap_secret", "");
 156                 return;
 157         }
 158         secret = strdup(p);
 159         lp_set_cmdline("torture:ldap_secret", secret);
 160
 161         printf ("%s - %s - %s\n", userdn, basedn, secret);
 162
 163 }
 164
 165 static void print_test_list(void)
 166 {
 167         struct torture_suite *o;
 168         struct torture_suite *s;
 169         struct torture_tcase *t;
 170
 171         for (o = torture_root->children; o; o = o->next) {
 172                 for (s = o->children; s; s = s->next) {
 173                         printf("%s-%s\n", o->name, s->name);
 174                 }
 175
 176                 for (t = o->testcases; t; t = t->next) {
 177                         printf("%s-%s\n", o->name, t->name);
 178                 }
 179         }
 180 }
 181
 182 static void usage(poptContext pc)
 183 {
 184         struct torture_suite *o;
 185         struct torture_suite *s;
 186         struct torture_tcase *t;
 187         int i;
 188
 189         poptPrintUsage(pc, stdout, 0);
 190         printf("\n");
 191
 192         printf("The binding format is:\n\n");
 193
 194         printf("  TRANSPORT:host[flags]\n\n");
 195
 196         printf("  where TRANSPORT is either ncacn_np for SMB, ncacn_ip_tcp for RPC/TCP\n");
 197         printf("  or ncalrpc for local connections.\n\n");
 198
 199         printf("  'host' is an IP or hostname or netbios name. If the binding string\n");
 200         printf("  identifies the server side of an endpoint, 'host' may be an empty\n");
 201         printf("  string.\n\n");
 202
 203         printf("  'flags' can include a SMB pipe name if using the ncacn_np transport or\n");
 204         printf("  a TCP port number if using the ncacn_ip_tcp transport, otherwise they\n");
 205         printf("  will be auto-determined.\n\n");
 206
 207         printf("  other recognised flags are:\n\n");
 208
 209         printf("    sign : enable ntlmssp signing\n");
 210         printf("    seal : enable ntlmssp sealing\n");
 211         printf("    connect : enable rpc connect level auth (auth, but no sign or seal)\n");
 212         printf("    validate: enable the NDR validator\n");
 213         printf("    print: enable debugging of the packets\n");
 214         printf("    bigendian: use bigendian RPC\n");
 215         printf("    padcheck: check reply data for non-zero pad bytes\n\n");
 216
 217         printf("  For example, these all connect to the samr pipe:\n\n");
 218
 219         printf("    ncacn_np:myserver\n");
 220         printf("    ncacn_np:myserver[samr]\n");
 221         printf("    ncacn_np:myserver[\\pipe\\samr]\n");
 222         printf("    ncacn_np:myserver[/pipe/samr]\n");
 223         printf("    ncacn_np:myserver[samr,sign,print]\n");
 224         printf("    ncacn_np:myserver[\\pipe\\samr,sign,seal,bigendian]\n");
 225         printf("    ncacn_np:myserver[/pipe/samr,seal,validate]\n");
 226         printf("    ncacn_np:\n");
 227         printf("    ncacn_np:[/pipe/samr]\n\n");
 228
 229         printf("    ncacn_ip_tcp:myserver\n");
 230         printf("    ncacn_ip_tcp:myserver[1024]\n");
 231         printf("    ncacn_ip_tcp:myserver[1024,sign,seal]\n\n");
 232
 233         printf("    ncalrpc:\n\n");
 234
 235         printf("The UNC format is:\n\n");
 236
 237         printf("  //server/share\n\n");
 238
 239         printf("Tests are:");
 240
 241         for (o = torture_root->children; o; o = o->next) {
 242                 printf("\n%s (%s):\n  ", o->description, o->name);
 243
 244                 i = 0;
 245                 for (s = o->children; s; s = s->next) {
 246                         if (i + strlen(o->name) + strlen(s->name) >= (MAX_COLS - 3)) {
 247                                 printf("\n  ");
 248                                 i = 0;
 249                         }
 250                         i+=printf("%s-%s ", o->name, s->name);
 251                 }
 252
 253                 for (t = o->testcases; t; t = t->next) {
 254                         if (i + strlen(o->name) + strlen(t->name) >= (MAX_COLS - 3)) {
 255                                 printf("\n  ");
 256                                 i = 0;
 257                         }
 258                         i+=printf("%s-%s ", o->name, t->name);
 259                 }
 260
 261                 if (i) printf("\n");
 262         }
 263
 264         printf("\nThe default test is ALL.\n");
 265
 266         exit(1);
 267 }
 268
 269 static void max_runtime_handler(int sig)
 270 {
 271         DEBUG(0,("maximum runtime exceeded for smbtorture - terminating\n"));
 272         exit(1);
 273 }
 274
 275 struct timeval last_suite_started;
 276
 277 static void simple_suite_start(struct torture_context *ctx,
 278                                                            struct torture_suite *suite)
 279 {
 280         last_suite_started = timeval_current();
 281         printf("Running %s\n", suite->name);
 282 }
 283
 284 static void simple_suite_finish(struct torture_context *ctx,
 285                                                            struct torture_suite *suite)
 286 {
 287
 288         printf("%s took %g secs\n\n", suite->name,
 289                    timeval_elapsed(&last_suite_started));
 290 }
 291
 292 static void simple_test_result (struct torture_context *context,
 293                                                                 enum torture_result res, const char *reason)
 294 {
 295         switch (res) {
 296         case TORTURE_OK:
 297                 if (reason)
 298                         printf("OK: %s\n", reason);
 299                 break;
 300         case TORTURE_FAIL:
 301                 printf("TEST %s FAILED! - %s\n", context->active_test->name, reason);
 302                 break;
 303         case TORTURE_ERROR:
 304                 printf("ERROR IN TEST %s! - %s\n", context->active_test->name, reason);
 305                 break;
 306         case TORTURE_SKIP:
 307                 printf("SKIP: %s - %s\n", context->active_test->name, reason);
 308                 break;
 309         }
 310 }
 311
 312 static void simple_comment (struct torture_context *test,
 313                                                         const char *comment)
 314 {
 315         printf("%s", comment);
 316 }
 317
 318 static void simple_warning(struct torture_context *test,
 319                                                    const char *comment)
 320 {
 321         fprintf(stderr, "WARNING: %s\n", comment);
 322 }
 323
 324 const static struct torture_ui_ops std_ui_ops = {
 325         .comment = simple_comment,
 326         .warning = simple_warning,
 327         .suite_start = simple_suite_start,
 328         .suite_finish = simple_suite_finish,
 329         .test_result = simple_test_result
 330 };
 331
 332 static void subunit_init(struct torture_context *ctx)
 333 {
 334         /* FIXME: register segv and bus handler */
 335 }
 336
 337 static void subunit_suite_start(struct torture_context *ctx,
 338                                                            struct torture_suite *suite)
 339 {
 340 }
 341
 342 static void subunit_test_start (struct torture_context *ctx,
 343                                                             struct torture_tcase *tcase,
 344                                                                 struct torture_test *test)
 345 {
 346         printf("test: %s\n", test->name);
 347 }
 348
 349 static void subunit_test_result (struct torture_context *context,
 350                                                                  enum torture_result res, const char *reason)
 351 {
 352         switch (res) {
 353         case TORTURE_OK:
 354                 printf("success: %s", context->active_test->name);
 355                 break;
 356         case TORTURE_FAIL:
 357                 printf("failure: %s", context->active_test->name);
 358                 break;
 359         case TORTURE_ERROR:
 360                 printf("error: %s", context->active_test->name);
 361                 break;
 362         case TORTURE_SKIP:
 363                 printf("skip: %s", context->active_test->name);
 364                 break;
 365         }
 366         if (reason)
 367                 printf(" [\n%s\n]", reason);
 368         printf("\n");
 369 }
 370
 371 static void subunit_comment (struct torture_context *test,
 372                                                          const char *comment)
 373 {
 374         fprintf(stderr, "%s", comment);
 375 }
 376
 377 const static struct torture_ui_ops subunit_ui_ops = {
 378         .init = subunit_init,
 379         .comment = subunit_comment,
 380         .test_start = subunit_test_start,
 381         .test_result = subunit_test_result,
 382         .suite_start = subunit_suite_start
 383 };
 384
 385 static void harness_test_start (struct torture_context *ctx,
 386                                                             struct torture_tcase *tcase,
 387                                                                 struct torture_test *test)
 388 {
 389 }
 390
 391 static void harness_test_result (struct torture_context *context,
 392                                                                  enum torture_result res, const char *reason)
 393 {
 394         switch (res) {
 395         case TORTURE_OK:
 396                 printf("ok %s - %s\n", context->active_test->name, reason);
 397                 break;
 398         case TORTURE_FAIL:
 399         case TORTURE_ERROR:
 400                 printf("not ok %s - %s\n", context->active_test->name, reason);
 401                 break;
 402         case TORTURE_SKIP:
 403                 printf("skip %s - %s\n", context->active_test->name, reason);
 404                 break;
 405         }
 406 }
 407
 408 static void harness_comment (struct torture_context *test,
 409                                                          const char *comment)
 410 {
 411         printf("# %s\n", comment);
 412 }
 413
 414 const static struct torture_ui_ops harness_ui_ops = {
 415         .comment = harness_comment,
 416         .test_start = harness_test_start,
 417         .test_result = harness_test_result
 418 };
 419
 420 static void quiet_suite_start(struct torture_context *ctx,
 421                                                   struct torture_suite *suite)
 422 {
 423         int i;
 424         ctx->quiet = true;
 425         for (i = 1; i < ctx->level; i++) putchar('\t');
 426         printf("%s: ", suite->name);
 427         fflush(stdout);
 428 }
 429
 430 static void quiet_suite_finish(struct torture_context *ctx,
 431                                                   struct torture_suite *suite)
 432 {
 433         putchar('\n');
 434 }
 435
 436 static void quiet_test_result (struct torture_context *context,
 437                                                            enum torture_result res, const char *reason)
 438 {
 439         fflush(stdout);
 440         switch (res) {
 441         case TORTURE_OK: putchar('.'); break;
 442         case TORTURE_FAIL: putchar('F'); break;
 443         case TORTURE_ERROR: putchar('E'); break;
 444         case TORTURE_SKIP: putchar('I'); break;
 445         }
 446 }
 447
 448 const static struct torture_ui_ops quiet_ui_ops = {
 449         .suite_start = quiet_suite_start,
 450         .suite_finish = quiet_suite_finish,
 451         .test_result = quiet_test_result
 452 };
 453
 454 void run_shell(struct torture_context *tctx)
 455 {
 456         char *cline;
 457         int argc;
 458         const char **argv;
 459         int ret;
 460
 461         while (1) {
 462                 cline = smb_readline("torture> ", NULL, NULL);
 463
 464                 if (cline == NULL)
 465                         return;
 466
 467                 ret = poptParseArgvString(cline, &argc, &argv);
 468                 if (ret != 0) {
 469                         fprintf(stderr, "Error parsing line\n");
 470                         continue;
 471                 }
 472
 473                 if (!strcmp(argv[0], "quit")) {
 474                         return;
 475                 } else if (!strcmp(argv[0], "set")) {
 476                         if (argc < 3) {
 477                                 fprintf(stderr, "Usage: set <variable> <value>\n");
 478                         } else {
 479                                 char *name = talloc_asprintf(NULL, "torture:%s", argv[1]);
 480                                 lp_set_cmdline(name, argv[2]);
 481                                 talloc_free(name);
 482                         }
 483                 } else if (!strcmp(argv[0], "help")) {
 484                         fprintf(stderr, "Available commands:\n"
 485                                                         " help - This help command\n"
 486                                                         " run - Run test\n"
 487                                                         " set - Change variables\n"
 488                                                         "\n");
 489                 } else if (!strcmp(argv[0], "run")) {
 490                         if (argc < 2) {
 491                                 fprintf(stderr, "Usage: run TEST-NAME [OPTIONS...]\n");
 492                         } else {
 493                                 run_test(tctx, argv[1]);
 494                         }
 495                 }
 496         }
 497 }
 498
 499 /****************************************************************************
 500   main program
 501 ****************************************************************************/
 502 int main(int argc,char *argv[])
 503 {
 504         int opt, i;
 505         bool correct = true;
 506         int max_runtime=0;
 507         int argc_new;
 508         struct torture_context *torture;
 509         const struct torture_ui_ops *ui_ops;
 510         char **argv_new;
 511         poptContext pc;
 512         static const char *target = "other";
 513         struct dcerpc_binding *binding_struct;
 514         NTSTATUS status;
 515         int shell = False;
 516         static const char *ui_ops_name = "simple";
 517         static int list_tests = 0;
 518         enum {OPT_LOADFILE=1000,OPT_UNCLIST,OPT_TIMELIMIT,OPT_DNS, OPT_LIST,
 519               OPT_DANGEROUS,OPT_SMB_PORTS,OPT_ASYNC,OPT_NUMPROGS, OPT_BASEDIR};
 520
 521         struct poptOption long_options[] = {
 522                 POPT_AUTOHELP
 523                 {"format", 0, POPT_ARG_STRING, &ui_ops_name, 0, "Output format (one of: simple, subunit, harness)", NULL },
 524                 {"smb-ports",   'p', POPT_ARG_STRING, NULL,     OPT_SMB_PORTS,  "SMB ports",    NULL},
 525                 {"basedir",       0, POPT_ARG_STRING, NULL, OPT_BASEDIR, "base directory", "BSAEDIR" },
 526                 {"seed",          0, POPT_ARG_INT,  &torture_seed,      0,      "seed",         NULL},
 527                 {"num-progs",     0, POPT_ARG_INT,  NULL,       OPT_NUMPROGS,   "num progs",    NULL},
 528                 {"num-ops",       0, POPT_ARG_INT,  &torture_numops,    0,      "num ops",      NULL},
 529                 {"entries",       0, POPT_ARG_INT,  &torture_entries,   0,      "entries",      NULL},
 530                 {"loadfile",      0, POPT_ARG_STRING,   NULL,   OPT_LOADFILE,   "loadfile",     NULL},
 531                 {"list",          0, POPT_ARG_NONE, &list_tests, 0, NULL, NULL },
 532                 {"unclist",       0, POPT_ARG_STRING,   NULL,   OPT_UNCLIST,    "unclist",      NULL},
 533                 {"timelimit",   't', POPT_ARG_INT,      NULL,   OPT_TIMELIMIT,  "timelimit",    NULL},
 534                 {"failures",    'f', POPT_ARG_INT,  &torture_failures,  0,      "failures",     NULL},
 535                 {"parse-dns",   'D', POPT_ARG_STRING,   NULL,   OPT_DNS,        "parse-dns",    NULL},
 536                 {"dangerous",   'X', POPT_ARG_NONE,     NULL,   OPT_DANGEROUS,
 537                  "run dangerous tests (eg. wiping out password database)", NULL},
 538                 {"shell",               0, POPT_ARG_NONE, &shell, True, "Run shell", NULL},
 539                 {"target",              'T', POPT_ARG_STRING, &target, 0, "samba3|samba4|other", NULL},
 540                 {"async",       'a', POPT_ARG_NONE,     NULL,   OPT_ASYNC,
 541                  "run async tests", NULL},
 542                 {"num-async",    0, POPT_ARG_INT,  &torture_numasync,  0,
 543                  "number of simultaneous async requests", NULL},
 544                 {"maximum-runtime", 0, POPT_ARG_INT, &max_runtime, 0,
 545                  "set maximum time for smbtorture to live", "seconds"},
 546                 POPT_COMMON_SAMBA
 547                 POPT_COMMON_CONNECTION
 548                 POPT_COMMON_CREDENTIALS
 549                 POPT_COMMON_VERSION
 550                 { NULL }
 551         };
 552
 553         setlinebuf(stdout);
 554
 555         /* we are never interested in SIGPIPE */
 556         BlockSignals(true, SIGPIPE);
 557
 558         pc = poptGetContext("smbtorture", argc, (const char **) argv, long_options,
 559                             POPT_CONTEXT_KEEP_FIRST);
 560
 561         poptSetOtherOptionHelp(pc, "<binding>|<unc> TEST1 TEST2 ...");
 562
 563         while((opt = poptGetNextOpt(pc)) != -1) {
 564                 switch (opt) {
 565                 case OPT_LOADFILE:
 566                         lp_set_cmdline("torture:loadfile", poptGetOptArg(pc));
 567                         break;
 568                 case OPT_UNCLIST:
 569                         lp_set_cmdline("torture:unclist", poptGetOptArg(pc));
 570                         break;
 571                 case OPT_TIMELIMIT:
 572                         lp_set_cmdline("torture:timelimit", poptGetOptArg(pc));
 573                         break;
 574                 case OPT_NUMPROGS:
 575                         lp_set_cmdline("torture:nprocs", poptGetOptArg(pc));
 576                         break;
 577                 case OPT_BASEDIR:
 578                         lp_set_cmdline("torture:basedir", poptGetOptArg(pc));
 579                         break;
 580                 case OPT_DNS:
 581                         parse_dns(poptGetOptArg(pc));
 582                         break;
 583                 case OPT_DANGEROUS:
 584                         lp_set_cmdline("torture:dangerous", "Yes");
 585                         break;
 586                 case OPT_ASYNC:
 587                         lp_set_cmdline("torture:async", "Yes");
 588                         break;
 589                 case OPT_SMB_PORTS:
 590                         lp_set_cmdline("smb ports", poptGetOptArg(pc));
 591                         break;
 592                 }
 593         }
 594
 595         if (strcmp(target, "samba3") == 0) {
 596                 lp_set_cmdline("torture:samba3", "true");
 597         } else if (strcmp(target, "samba4") == 0) {
 598                 lp_set_cmdline("torture:samba4", "true");
 599         }
 600
 601         if (max_runtime) {
 602                 /* this will only work if nobody else uses alarm(),
 603                    which means it won't work for some tests, but we
 604                    can't use the event context method we use for smbd
 605                    as so many tests create their own event
 606                    context. This will at least catch most cases. */
 607                 signal(SIGALRM, max_runtime_handler);
 608                 alarm(max_runtime);
 609         }
 610
 611         torture_init();
 612         ldb_global_init();
 613
 614         if (list_tests) {
 615                 print_test_list();
 616                 return 0;
 617         }
 618
 619         if (torture_seed == 0) {
 620                 torture_seed = time(NULL);
 621         }
 622         printf("Using seed %d\n", torture_seed);
 623         srandom(torture_seed);
 624
 625         argv_new = discard_const_p(char *, poptGetArgs(pc));
 626
 627         argc_new = argc;
 628         for (i=0; i<argc; i++) {
 629                 if (argv_new[i] == NULL) {
 630                         argc_new = i;
 631                         break;
 632                 }
 633         }
 634
 635         if (!(argc_new >= 3 || (shell && argc_new >= 2))) {
 636                 usage(pc);
 637                 exit(1);
 638         }
 639
 640         /* see if its a RPC transport specifier */
 641         status = dcerpc_parse_binding(talloc_autofree_context(), argv_new[1], &binding_struct);
 642         if (NT_STATUS_IS_OK(status)) {
 643                 lp_set_cmdline("torture:host", binding_struct->host);
 644                 lp_set_cmdline("torture:share", "IPC$");
 645                 lp_set_cmdline("torture:binding", argv_new[1]);
 646         } else {
 647                 char *binding = NULL;
 648                 char *host = NULL, *share = NULL;
 649
 650                 if (!smbcli_parse_unc(argv_new[1], NULL, &host, &share)) {
 651                         d_printf("Invalid option: %s is not a valid torture target (share or binding string)\n\n", argv_new[1]);
 652                         usage(pc);
 653                 }
 654
 655                 lp_set_cmdline("torture:host", host);
 656                 lp_set_cmdline("torture:share", share);
 657                 asprintf(&binding, "ncacn_np:%s", host);
 658                 lp_set_cmdline("torture:binding", binding);
 659         }
 660
 661         if (!strcmp(ui_ops_name, "simple")) {
 662                 ui_ops = &std_ui_ops;
 663         } else if (!strcmp(ui_ops_name, "subunit")) {
 664                 ui_ops = &subunit_ui_ops;
 665         } else if (!strcmp(ui_ops_name, "harness")) {
 666                 ui_ops = &harness_ui_ops;
 667         } else if (!strcmp(ui_ops_name, "quiet")) {
 668                 ui_ops = &quiet_ui_ops;
 669         } else {
 670                 printf("Unknown output format '%s'\n", ui_ops_name);
 671                 exit(1);
 672         }
 673
 674         torture = torture_context_init(talloc_autofree_context(), ui_ops);
 675
 676         if (argc_new == 0) {
 677                 printf("You must specify a test to run, or 'ALL'\n");
 678         } else if (shell) {
 679                 run_shell(torture);
 680         } else {
 681                 for (i=2;i<argc_new;i++) {
 682                         if (!run_test(torture, argv_new[i])) {
 683                                 correct = false;
 684                         }
 685                 }
 686         }
 687
 688         if (torture->returncode) {
 689                 return(0);
 690         } else {
 691                 return(1);
 692         }
 693 }