?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
docs: man smbtorture: Add missing meta data.
[Samba/gebeck_regimport.git] / source4 / torture / smbtorture.c
blob95f0719605a26912a4791c333f5f8544e8ac3c77
1 /*
2 Unix SMB/CIFS implementation.
3 SMB torture tester
4 Copyright (C) Andrew Tridgell 1997-2003
5 Copyright (C) Jelmer Vernooij 2006-2008
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 "../libcli/smbreadline/smbreadline.h"
28 #include "libcli/libcli.h"
29 #include "lib/events/events.h"
30
31 #include "torture/smbtorture.h"
32 #include "librpc/rpc/dcerpc.h"
33 #include "auth/gensec/gensec.h"
34 #include "param/param.h"
35 #include "lib/util/samba_modules.h"
36
37 #if HAVE_READLINE_HISTORY_H
38 #include <readline/history.h>
39 #endif
40
41 static char *prefix_name(TALLOC_CTX *mem_ctx, const char *prefix, const char *name)
42 {
43 if (prefix == NULL)
44 return talloc_strdup(mem_ctx, name);
45 else
46 return talloc_asprintf(mem_ctx, "%s.%s", prefix, name);
47 }
48
49 static void print_test_list(const struct torture_suite *suite, const char *prefix, const char *expr)
50 {
51 struct torture_suite *o;
52 struct torture_tcase *t;
53 struct torture_test *p;
54
55 for (o = suite->children; o; o = o->next) {
56 char *name = prefix_name(NULL, prefix, o->name);
57 print_test_list(o, name, expr);
58 talloc_free(name);
59 }
60
61 for (t = suite->testcases; t; t = t->next) {
62 for (p = t->tests; p; p = p->next) {
63 char *name = talloc_asprintf(NULL, "%s.%s.%s", prefix, t->name, p->name);
64 if (strncmp(name, expr, strlen(expr)) == 0) {
65 printf("%s\n", name);
66 }
67 talloc_free(name);
68 }
69 }
70 }
71
72 static bool run_matching(struct torture_context *torture,
73 const char *prefix,
74 const char *expr,
75 const char **restricted,
76 struct torture_suite *suite,
77 bool *matched)
78 {
79 bool ret = true;
80 struct torture_suite *o;
81 struct torture_tcase *t;
82 struct torture_test *p;
83
84 for (o = suite->children; o; o = o->next) {
85 char *name = NULL;
86 name = prefix_name(torture, prefix, o->name);
87 if (gen_fnmatch(expr, name) == 0) {
88 *matched = true;
89 reload_charcnv(torture->lp_ctx);
90 if (restricted != NULL)
91 ret &= torture_run_suite_restricted(torture, o, restricted);
92 else
93 ret &= torture_run_suite(torture, o);
94 }
95 ret &= run_matching(torture, name, expr, restricted, o, matched);
96 }
97
98 for (t = suite->testcases; t; t = t->next) {
99 char *name = talloc_asprintf(torture, "%s.%s", prefix, t->name);
100 if (gen_fnmatch(expr, name) == 0) {
101 *matched = true;
102 reload_charcnv(torture->lp_ctx);
103 ret &= torture_run_tcase_restricted(torture, t, restricted);
104 }
105 for (p = t->tests; p; p = p->next) {
106 name = talloc_asprintf(torture, "%s.%s.%s", prefix, t->name, p->name);
107 if (gen_fnmatch(expr, name) == 0) {
108 *matched = true;
109 reload_charcnv(torture->lp_ctx);
110 ret &= torture_run_test_restricted(torture, t, p, restricted);
111 }
112 }
113 }
114
115 return ret;
116 }
117
118 #define MAX_COLS 80 /* FIXME: Determine this at run-time */
119
120 /****************************************************************************
121 run a specified test or "ALL"
122 ****************************************************************************/
123 bool torture_run_named_tests(struct torture_context *torture, const char *name,
124 const char **restricted)
125 {
126 bool ret = true;
127 bool matched = false;
128 struct torture_suite *o;
129
130 torture_ui_report_time(torture);
131
132 if (strequal(name, "ALL")) {
133 if (restricted != NULL) {
134 printf("--load-list and ALL are incompatible\n");
135 return false;
136 }
137 for (o = torture_root->children; o; o = o->next) {
138 ret &= torture_run_suite(torture, o);
139 }
140 return ret;
141 }
142
143 ret = run_matching(torture, NULL, name, restricted, torture_root, &matched);
144
145 if (!matched) {
146 printf("Unknown torture operation '%s'\n", name);
147 return false;
148 }
149
150 return ret;
151 }
152
153 bool torture_parse_target(struct loadparm_context *lp_ctx, const char *target)
154 {
155 char *host = NULL, *share = NULL;
156 struct dcerpc_binding *binding_struct;
157 NTSTATUS status;
158
159 /* see if its a RPC transport specifier */
160 if (!smbcli_parse_unc(target, NULL, &host, &share)) {
161 status = dcerpc_parse_binding(talloc_autofree_context(), target, &binding_struct);
162 if (NT_STATUS_IS_ERR(status)) {
163 d_printf("Invalid option: %s is not a valid torture target (share or binding string)\n\n", target);
164 return false;
165 }
166 lpcfg_set_cmdline(lp_ctx, "torture:host", binding_struct->host);
167 if (lpcfg_parm_string(lp_ctx, NULL, "torture", "share") == NULL)
168 lpcfg_set_cmdline(lp_ctx, "torture:share", "IPC$");
169 lpcfg_set_cmdline(lp_ctx, "torture:binding", target);
170 } else {
171 lpcfg_set_cmdline(lp_ctx, "torture:host", host);
172 lpcfg_set_cmdline(lp_ctx, "torture:share", share);
173 lpcfg_set_cmdline(lp_ctx, "torture:binding", host);
174 }
175
176 return true;
177 }
178
179 static void parse_dns(struct loadparm_context *lp_ctx, const char *dns)
180 {
181 char *userdn, *basedn, *secret;
182 char *p, *d;
183
184 /* retrievieng the userdn */
185 p = strchr_m(dns, '#');
186 if (!p) {
187 lpcfg_set_cmdline(lp_ctx, "torture:ldap_userdn", "");
188 lpcfg_set_cmdline(lp_ctx, "torture:ldap_basedn", "");
189 lpcfg_set_cmdline(lp_ctx, "torture:ldap_secret", "");
190 return;
191 }
192 userdn = strndup(dns, p - dns);
193 lpcfg_set_cmdline(lp_ctx, "torture:ldap_userdn", userdn);
194
195 /* retrieve the basedn */
196 d = p + 1;
197 p = strchr_m(d, '#');
198 if (!p) {
199 lpcfg_set_cmdline(lp_ctx, "torture:ldap_basedn", "");
200 lpcfg_set_cmdline(lp_ctx, "torture:ldap_secret", "");
201 return;
202 }
203 basedn = strndup(d, p - d);
204 lpcfg_set_cmdline(lp_ctx, "torture:ldap_basedn", basedn);
205
206 /* retrieve the secret */
207 p = p + 1;
208 if (!p) {
209 lpcfg_set_cmdline(lp_ctx, "torture:ldap_secret", "");
210 return;
211 }
212 secret = strdup(p);
213 lpcfg_set_cmdline(lp_ctx, "torture:ldap_secret", secret);
214
215 printf ("%s - %s - %s\n", userdn, basedn, secret);
216
217 }
218
219 /* Print the full test list, formatted into separate labelled test
220 * groups.
221 */
222 static void print_structured_testsuite_list(void)
223 {
224 struct torture_suite *o;
225 struct torture_suite *s;
226 struct torture_tcase *t;
227 int i;
228
229 if (torture_root == NULL) {
230 printf("NO TESTS LOADED\n");
231 return;
232 }
233
234 for (o = torture_root->children; o; o = o->next) {
235 printf("\n%s (%s):\n ", o->description, o->name);
236
237 i = 0;
238 for (s = o->children; s; s = s->next) {
239 if (i + strlen(o->name) + strlen(s->name) >= (MAX_COLS - 3)) {
240 printf("\n ");
241 i = 0;
242 }
243 i+=printf("%s.%s ", o->name, s->name);
244 }
245
246 for (t = o->testcases; t; t = t->next) {
247 if (i + strlen(o->name) + strlen(t->name) >= (MAX_COLS - 3)) {
248 printf("\n ");
249 i = 0;
250 }
251 i+=printf("%s.%s ", o->name, t->name);
252 }
253
254 if (i) printf("\n");
255 }
256
257 printf("\nThe default test is ALL.\n");
258 }
259
260 static void print_testsuite_list(void)
261 {
262 struct torture_suite *o;
263 struct torture_suite *s;
264 struct torture_tcase *t;
265
266 if (torture_root == NULL)
267 return;
268
269 for (o = torture_root->children; o; o = o->next) {
270 for (s = o->children; s; s = s->next) {
271 printf("%s.%s\n", o->name, s->name);
272 }
273
274 for (t = o->testcases; t; t = t->next) {
275 printf("%s.%s\n", o->name, t->name);
276 }
277 }
278 }
279
280 void torture_print_testsuites(bool structured)
281 {
282 if (structured) {
283 print_structured_testsuite_list();
284 } else {
285 print_testsuite_list();
286 }
287 }
288
289 static void usage(poptContext pc)
290 {
291 poptPrintUsage(pc, stdout, 0);
292 printf("\n");
293
294 printf("The binding format is:\n\n");
295
296 printf(" TRANSPORT:host[flags]\n\n");
297
298 printf(" where TRANSPORT is either ncacn_np for SMB, ncacn_ip_tcp for RPC/TCP\n");
299 printf(" or ncalrpc for local connections.\n\n");
300
301 printf(" 'host' is an IP or hostname or netbios name. If the binding string\n");
302 printf(" identifies the server side of an endpoint, 'host' may be an empty\n");
303 printf(" string.\n\n");
304
305 printf(" 'flags' can include a SMB pipe name if using the ncacn_np transport or\n");
306 printf(" a TCP port number if using the ncacn_ip_tcp transport, otherwise they\n");
307 printf(" will be auto-determined.\n\n");
308
309 printf(" other recognised flags are:\n\n");
310
311 printf(" sign : enable ntlmssp signing\n");
312 printf(" seal : enable ntlmssp sealing\n");
313 printf(" connect : enable rpc connect level auth (auth, but no sign or seal)\n");
314 printf(" validate: enable the NDR validator\n");
315 printf(" print: enable debugging of the packets\n");
316 printf(" bigendian: use bigendian RPC\n");
317 printf(" padcheck: check reply data for non-zero pad bytes\n\n");
318
319 printf(" For example, these all connect to the samr pipe:\n\n");
320
321 printf(" ncacn_np:myserver\n");
322 printf(" ncacn_np:myserver[samr]\n");
323 printf(" ncacn_np:myserver[\\pipe\\samr]\n");
324 printf(" ncacn_np:myserver[/pipe/samr]\n");
325 printf(" ncacn_np:myserver[samr,sign,print]\n");
326 printf(" ncacn_np:myserver[\\pipe\\samr,sign,seal,bigendian]\n");
327 printf(" ncacn_np:myserver[/pipe/samr,seal,validate]\n");
328 printf(" ncacn_np:\n");
329 printf(" ncacn_np:[/pipe/samr]\n\n");
330
331 printf(" ncacn_ip_tcp:myserver\n");
332 printf(" ncacn_ip_tcp:myserver[1024]\n");
333 printf(" ncacn_ip_tcp:myserver[1024,sign,seal]\n\n");
334
335 printf(" ncalrpc:\n\n");
336
337 printf("The UNC format is:\n\n");
338
339 printf(" //server/share\n\n");
340
341 printf("Tests are:");
342
343 print_structured_testsuite_list();
344
345 }
346
347 _NORETURN_ static void max_runtime_handler(int sig)
348 {
349 DEBUG(0,("maximum runtime exceeded for smbtorture - terminating\n"));
350 exit(1);
351 }
352
353 /****************************************************************************
354 main program
355 ****************************************************************************/
356 int main(int argc,char *argv[])
357 {
358 int opt, i;
359 bool correct = true;
360 int max_runtime=0;
361 int argc_new;
362 struct torture_context *torture;
363 struct torture_results *results;
364 const struct torture_ui_ops *ui_ops;
365 char **argv_new;
366 poptContext pc;
367 static const char *target = "other";
368 NTSTATUS status;
369 int shell = false;
370 static const char *ui_ops_name = "subunit";
371 const char *basedir = NULL;
372 char *outputdir;
373 const char *extra_module = NULL;
374 static int list_tests = 0, list_testsuites = 0;
375 int num_extra_users = 0;
376 char **restricted = NULL;
377 int num_restricted = -1;
378 const char *load_list = NULL;
379 enum {OPT_LOADFILE=1000,OPT_UNCLIST,OPT_TIMELIMIT,OPT_DNS, OPT_LIST,
380 OPT_DANGEROUS,OPT_SMB_PORTS,OPT_ASYNC,OPT_NUMPROGS,
381 OPT_EXTRA_USER,};
382
383 struct poptOption long_options[] = {
384 POPT_AUTOHELP
385 {"format", 0, POPT_ARG_STRING, &ui_ops_name, 0, "Output format (one of: simple, subunit)", NULL },
386 {"smb-ports", 'p', POPT_ARG_STRING, NULL, OPT_SMB_PORTS, "SMB ports", NULL},
387 {"basedir", 0, POPT_ARG_STRING, &basedir, 0, "base directory", "BASEDIR" },
388 {"seed", 0, POPT_ARG_INT, &torture_seed, 0, "Seed to use for randomizer", NULL},
389 {"num-progs", 0, POPT_ARG_INT, NULL, OPT_NUMPROGS, "num progs", NULL},
390 {"num-ops", 0, POPT_ARG_INT, &torture_numops, 0, "num ops", NULL},
391 {"entries", 0, POPT_ARG_INT, &torture_entries, 0, "entries", NULL},
392 {"loadfile", 0, POPT_ARG_STRING, NULL, OPT_LOADFILE, "NBench load file to use", NULL},
393 {"list-suites", 0, POPT_ARG_NONE, &list_testsuites, 0, "List available testsuites and exit", NULL },
394 {"list", 0, POPT_ARG_NONE, &list_tests, 0, "List available tests in specified suites and exit", NULL },
395 {"unclist", 0, POPT_ARG_STRING, NULL, OPT_UNCLIST, "unclist", NULL},
396 {"timelimit", 't', POPT_ARG_INT, NULL, OPT_TIMELIMIT, "Set time limit (in seconds)", NULL},
397 {"failures", 'f', POPT_ARG_INT, &torture_failures, 0, "failures", NULL},
398 {"parse-dns", 'D', POPT_ARG_STRING, NULL, OPT_DNS, "parse-dns", NULL},
399 {"dangerous", 'X', POPT_ARG_NONE, NULL, OPT_DANGEROUS,
400 "run dangerous tests (eg. wiping out password database)", NULL},
401 {"load-module", 0, POPT_ARG_STRING, &extra_module, 0, "load tests from DSO file", "SOFILE"},
402 {"shell", 0, POPT_ARG_NONE, &shell, true, "Run shell", NULL},
403 {"target", 'T', POPT_ARG_STRING, &target, 0, "samba3|samba4|other", NULL},
404 {"async", 'a', POPT_ARG_NONE, NULL, OPT_ASYNC,
405 "run async tests", NULL},
406 {"num-async", 0, POPT_ARG_INT, &torture_numasync, 0,
407 "number of simultaneous async requests", NULL},
408 {"maximum-runtime", 0, POPT_ARG_INT, &max_runtime, 0,
409 "set maximum time for smbtorture to live", "seconds"},
410 {"extra-user", 0, POPT_ARG_STRING, NULL, OPT_EXTRA_USER,
411 "extra user credentials", NULL},
412 {"load-list", 0, POPT_ARG_STRING, &load_list, 0,
413 "load a test id list from a text file", NULL},
414 POPT_COMMON_SAMBA
415 POPT_COMMON_CONNECTION
416 POPT_COMMON_CREDENTIALS
417 POPT_COMMON_VERSION
418 { NULL }
419 };
420
421 setlinebuf(stdout);
422
423 /* we are never interested in SIGPIPE */
424 BlockSignals(true, SIGPIPE);
425
426 pc = poptGetContext("smbtorture", argc, (const char **) argv, long_options,
427 POPT_CONTEXT_KEEP_FIRST);
428
429 poptSetOtherOptionHelp(pc, "<binding>|<unc> TEST1 TEST2 ...");
430
431 while((opt = poptGetNextOpt(pc)) != -1) {
432 switch (opt) {
433 case OPT_LOADFILE:
434 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:loadfile", poptGetOptArg(pc));
435 break;
436 case OPT_UNCLIST:
437 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:unclist", poptGetOptArg(pc));
438 break;
439 case OPT_TIMELIMIT:
440 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:timelimit", poptGetOptArg(pc));
441 break;
442 case OPT_NUMPROGS:
443 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:nprocs", poptGetOptArg(pc));
444 break;
445 case OPT_DNS:
446 parse_dns(cmdline_lp_ctx, poptGetOptArg(pc));
447 break;
448 case OPT_DANGEROUS:
449 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:dangerous", "Yes");
450 break;
451 case OPT_ASYNC:
452 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:async", "Yes");
453 break;
454 case OPT_SMB_PORTS:
455 lpcfg_set_cmdline(cmdline_lp_ctx, "smb ports", poptGetOptArg(pc));
456 break;
457 case OPT_EXTRA_USER:
458 {
459 char *option = talloc_asprintf(NULL, "torture:extra_user%u",
460 ++num_extra_users);
461 const char *value = poptGetOptArg(pc);
462 lpcfg_set_cmdline(cmdline_lp_ctx, option, value);
463 talloc_free(option);
464 }
465 break;
466 default:
467 if (opt < 0) {
468 printf("bad command line option %d\n", opt);
469 exit(1);
470 }
471 }
472 }
473
474 if (load_list != NULL) {
475 restricted = file_lines_load(load_list, &num_restricted, 0,
476 talloc_autofree_context());
477 if (restricted == NULL) {
478 printf("Unable to read load list file '%s'\n", load_list);
479 exit(1);
480 }
481 }
482
483 if (strcmp(target, "samba3") == 0) {
484 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:samba3", "true");
485 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:resume_key_support", "false");
486 } else if (strcmp(target, "samba4") == 0) {
487 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:samba4", "true");
488 } else if (strcmp(target, "samba4-ntvfs") == 0) {
489 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:samba4", "true");
490 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:samba4-ntvfs", "true");
491 } else if (strcmp(target, "winxp") == 0) {
492 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:winxp", "true");
493 } else if (strcmp(target, "w2k3") == 0) {
494 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:w2k3", "true");
495 } else if (strcmp(target, "w2k8") == 0) {
496 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:w2k8", "true");
497 lpcfg_set_cmdline(cmdline_lp_ctx,
498 "torture:invalid_lock_range_support", "false");
499 } else if (strcmp(target, "win7") == 0) {
500 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:win7", "true");
501 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:cn_max_buffer_size",
502 "0x00010000");
503 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:resume_key_support", "false");
504 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:rewind_support", "false");
505
506 /* RAW-SEARCH for fails for inexplicable reasons against win7 */
507 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:search_ea_support", "false");
508
509 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:hide_on_access_denied",
510 "true");
511 } else if (strcmp(target, "onefs") == 0) {
512 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:onefs", "true");
513 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:openx_deny_dos_support",
514 "false");
515 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:range_not_locked_on_file_close", "false");
516 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:sacl_support", "false");
517 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:ea_support", "false");
518 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:smbexit_pdu_support",
519 "false");
520 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:smblock_pdu_support",
521 "false");
522 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:2_step_break_to_none",
523 "true");
524 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:deny_dos_support", "false");
525 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:deny_fcb_support", "false");
526 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:read_support", "false");
527 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:writeclose_support", "false");
528 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:resume_key_support", "false");
529 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:rewind_support", "false");
530 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:raw_search_search", "false");
531 lpcfg_set_cmdline(cmdline_lp_ctx, "torture:search_ea_size", "false");
532 }
533
534 if (max_runtime) {
535 /* this will only work if nobody else uses alarm(),
536 which means it won't work for some tests, but we
537 can't use the event context method we use for smbd
538 as so many tests create their own event
539 context. This will at least catch most cases. */
540 signal(SIGALRM, max_runtime_handler);
541 alarm(max_runtime);
542 }
543
544 if (extra_module != NULL) {
545 init_module_fn fn = load_module(poptGetOptArg(pc), false, NULL);
546
547 if (fn == NULL)
548 d_printf("Unable to load module from %s\n", poptGetOptArg(pc));
549 else {
550 status = fn();
551 if (NT_STATUS_IS_ERR(status)) {
552 d_printf("Error initializing module %s: %s\n",
553 poptGetOptArg(pc), nt_errstr(status));
554 }
555 }
556 } else {
557 torture_init();
558 }
559
560 if (list_testsuites) {
561 print_testsuite_list();
562 return 0;
563 }
564
565 argv_new = discard_const_p(char *, poptGetArgs(pc));
566
567 argc_new = argc;
568 for (i=0; i<argc; i++) {
569 if (argv_new[i] == NULL) {
570 argc_new = i;
571 break;
572 }
573 }
574
575 if (list_tests) {
576 if (argc_new == 1) {
577 print_test_list(torture_root, NULL, "");
578 } else {
579 for (i=1;i<argc_new;i++) {
580 print_test_list(torture_root, NULL, argv_new[i]);
581 }
582 }
583 return 0;
584 }
585
586 if (torture_seed == 0) {
587 torture_seed = time(NULL);
588 }
589 printf("Using seed %d\n", torture_seed);
590 srandom(torture_seed);
591
592 if (!strcmp(ui_ops_name, "simple")) {
593 ui_ops = &torture_simple_ui_ops;
594 } else if (!strcmp(ui_ops_name, "subunit")) {
595 ui_ops = &torture_subunit_ui_ops;
596 } else {
597 printf("Unknown output format '%s'\n", ui_ops_name);
598 exit(1);
599 }
600
601 results = torture_results_init(talloc_autofree_context(), ui_ops);
602
603 torture = torture_context_init(s4_event_context_init(talloc_autofree_context()),
604 results);
605 if (basedir != NULL) {
606 if (basedir[0] != '/') {
607 fprintf(stderr, "Please specify an absolute path to --basedir\n");
608 return 1;
609 }
610 outputdir = talloc_asprintf(torture, "%s/smbtortureXXXXXX", basedir);
611 } else {
612 char *pwd = talloc_size(torture, PATH_MAX);
613 if (!getcwd(pwd, PATH_MAX)) {
614 fprintf(stderr, "Unable to determine current working directory\n");
615 return 1;
616 }
617 outputdir = talloc_asprintf(torture, "%s/smbtortureXXXXXX", pwd);
618 }
619 if (!outputdir) {
620 fprintf(stderr, "Could not allocate per-run output dir\n");
621 return 1;
622 }
623 torture->outputdir = mkdtemp(outputdir);
624 if (!torture->outputdir) {
625 perror("Failed to make temp output dir");
626 }
627
628 torture->lp_ctx = cmdline_lp_ctx;
629
630 gensec_init();
631
632 if (shell) {
633 /* In shell mode, just ignore any remaining test names. */
634 torture_shell(torture);
635 } else {
636
637 /* At this point, we should just have a target string,
638 * followed by a series of test names. Unless we are in
639 * shell mode, in which case we don't need anythig more.
640 */
641
642 if (argc_new < 3) {
643 printf("You must specify a test to run, or 'ALL'\n");
644 usage(pc);
645 torture->results->returncode = 1;
646 } else if (!torture_parse_target(cmdline_lp_ctx, argv_new[1])) {
647 /* Take the target name or binding. */
648 usage(pc);
649 torture->results->returncode = 1;
650 } else {
651 for (i=2;i<argc_new;i++) {
652 if (!torture_run_named_tests(torture, argv_new[i],
653 (const char **)restricted)) {
654 correct = false;
655 }
656 }
657 }
658 }
659
660 /* Now delete the temp dir we created */
661 torture_deltree_outputdir(torture);
662
663 if (torture->results->returncode && correct) {
664 return(0);
665 } else {
666 return(1);
667 }
668 }
reg import