r18994: - torture_register_suite should be public
[samba.git] / source4 / torture / torture.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 2 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, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23 #include "system/time.h"
24 #include "torture/torture.h"
25 #include "build.h"
26 #include "lib/util/dlinklist.h"
27
28 _PUBLIC_ int torture_nprocs=4;
29 _PUBLIC_ int torture_numops=10;
30 _PUBLIC_ int torture_entries=1000;
31 _PUBLIC_ int torture_failures=1;
32 _PUBLIC_ int torture_seed=0;
33 _PUBLIC_ int torture_numasync=100;
34 _PUBLIC_ BOOL use_oplocks;
35 _PUBLIC_ BOOL use_level_II_oplocks;
36 _PUBLIC_ BOOL torture_showall = False;
37
38 struct torture_suite_list *torture_suites = NULL;
39
40 _PUBLIC_ NTSTATUS torture_register_suite(struct torture_suite *suite)
41 {
42         struct torture_suite_list *p, *n;
43
44         if (!suite) {
45                 return NT_STATUS_OK;
46         }
47
48         n = talloc(talloc_autofree_context(), struct torture_suite_list);
49         n->suite = suite;
50
51         for (p = torture_suites; p; p = p->next) {
52                 if (strcmp(p->suite->name, suite->name) == 0) {
53                         /* Check for duplicates */
54                         DEBUG(0,("There already is a suite registered with the name %s!\n", suite->name));
55                         return NT_STATUS_OBJECT_NAME_COLLISION;
56                 }
57
58                 if (strcmp(p->suite->name, suite->name) < 0 && 
59                         (!p->next || strcmp(p->next->suite->name, suite->name) > 0)) {
60                         DLIST_ADD_AFTER(torture_suites, n, p);
61                         return NT_STATUS_OK;
62                 }
63         }
64
65         DLIST_ADD(torture_suites, n);
66
67         return NT_STATUS_OK;
68 }
69
70 static BOOL wrap_old_torture_fn(struct torture_context *torture, const void *_fn)
71 {
72         BOOL (*fn)(struct torture_context *) = _fn;
73         return fn(torture);
74 }
75
76
77 /* Backwards compatibility wrapper */
78 _PUBLIC_ NTSTATUS register_torture_op(const char *name, BOOL (*fn)(struct torture_context *))
79 {
80         struct torture_suite *suite;
81         suite = torture_suite_create(talloc_autofree_context(), name);
82
83         torture_suite_add_simple_tcase(suite, name, wrap_old_torture_fn, fn);
84         torture_register_suite(suite);
85
86         return NT_STATUS_OK;
87 }
88
89 int torture_init(void)
90 {
91         init_module_fn static_init[] = STATIC_torture_MODULES;
92         init_module_fn *shared_init = load_samba_modules(NULL, "torture");
93         
94         run_init_functions(static_init);
95         run_init_functions(shared_init);
96
97         talloc_free(shared_init);
98
99         return 0;
100 }