2 Unix SMB/CIFS implementation.
4 Copyright (C) Jelmer Vernooij 2006
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "libcli/smb2/smb2.h"
23 #include "torture/smbtorture.h"
24 #include "torture/smb2/proto.h"
25 #include "../lib/util/dlinklist.h"
27 static bool wrap_simple_1smb2_test(struct torture_context
*torture_ctx
,
28 struct torture_tcase
*tcase
,
29 struct torture_test
*test
)
31 bool (*fn
) (struct torture_context
*, struct smb2_tree
*);
33 struct smb2_tree
*tree1
;
34 TALLOC_CTX
*mem_ctx
= talloc_new(torture_ctx
);
36 if (!torture_smb2_connection(torture_ctx
, &tree1
))
41 * The test might close the connection. If we steal the tree context
42 * before that and free the parent instead of tree directly, we avoid
43 * a double free error.
45 talloc_steal(mem_ctx
, tree1
);
49 ret
= fn(torture_ctx
, tree1
);
56 struct torture_test
*torture_suite_add_1smb2_test(struct torture_suite
*suite
,
58 bool (*run
)(struct torture_context
*,
61 struct torture_test
*test
;
62 struct torture_tcase
*tcase
;
64 tcase
= torture_suite_add_tcase(suite
, name
);
66 test
= talloc(tcase
, struct torture_test
);
68 test
->name
= talloc_strdup(test
, name
);
69 test
->description
= NULL
;
70 test
->run
= wrap_simple_1smb2_test
;
72 test
->dangerous
= false;
74 DLIST_ADD_END(tcase
->tests
, test
, struct torture_test
*);
80 static bool wrap_simple_2smb2_test(struct torture_context
*torture_ctx
,
81 struct torture_tcase
*tcase
,
82 struct torture_test
*test
)
84 bool (*fn
) (struct torture_context
*, struct smb2_tree
*, struct smb2_tree
*);
87 struct smb2_tree
*tree1
;
88 struct smb2_tree
*tree2
;
89 TALLOC_CTX
*mem_ctx
= talloc_new(torture_ctx
);
91 if (!torture_smb2_connection(torture_ctx
, &tree1
)) {
95 talloc_steal(mem_ctx
, tree1
);
97 if (!torture_smb2_connection(torture_ctx
, &tree2
)) {
101 talloc_steal(mem_ctx
, tree2
);
105 ret
= fn(torture_ctx
, tree1
, tree2
);
108 /* the test may already have closed some of the connections */
109 talloc_free(mem_ctx
);
115 struct torture_test
*torture_suite_add_2smb2_test(struct torture_suite
*suite
,
117 bool (*run
)(struct torture_context
*,
121 struct torture_test
*test
;
122 struct torture_tcase
*tcase
;
124 tcase
= torture_suite_add_tcase(suite
, name
);
126 test
= talloc(tcase
, struct torture_test
);
128 test
->name
= talloc_strdup(test
, name
);
129 test
->description
= NULL
;
130 test
->run
= wrap_simple_2smb2_test
;
132 test
->dangerous
= false;
134 DLIST_ADD_END(tcase
->tests
, test
, struct torture_test
*);
139 NTSTATUS
torture_smb2_init(void)
141 struct torture_suite
*suite
= torture_suite_create(talloc_autofree_context(), "smb2");
142 torture_suite_add_simple_test(suite
, "connect", torture_smb2_connect
);
143 torture_suite_add_suite(suite
, torture_smb2_scan_init());
144 torture_suite_add_simple_test(suite
, "getinfo", torture_smb2_getinfo
);
145 torture_suite_add_simple_test(suite
, "setinfo", torture_smb2_setinfo
);
146 torture_suite_add_suite(suite
, torture_smb2_lock_init());
147 torture_suite_add_suite(suite
, torture_smb2_read_init());
148 torture_suite_add_suite(suite
, torture_smb2_create_init());
149 torture_suite_add_suite(suite
, torture_smb2_acls_init());
150 torture_suite_add_suite(suite
, torture_smb2_notify_init());
151 torture_suite_add_suite(suite
, torture_smb2_durable_open_init());
152 torture_suite_add_suite(suite
, torture_smb2_durable_v2_open_init());
153 torture_suite_add_suite(suite
, torture_smb2_dir_init());
154 torture_suite_add_suite(suite
, torture_smb2_lease_init());
155 torture_suite_add_suite(suite
, torture_smb2_compound_init());
156 torture_suite_add_suite(suite
, torture_smb2_oplocks_init());
157 torture_suite_add_suite(suite
, torture_smb2_streams_init());
158 torture_suite_add_suite(suite
, torture_smb2_ioctl_init());
159 torture_suite_add_suite(suite
, torture_smb2_rename_init());
160 torture_suite_add_1smb2_test(suite
, "bench-oplock", test_smb2_bench_oplock
);
161 torture_suite_add_1smb2_test(suite
, "hold-oplock", test_smb2_hold_oplock
);
162 torture_suite_add_suite(suite
, torture_smb2_session_init());
164 suite
->description
= talloc_strdup(suite
, "SMB2-specific tests");
166 torture_register_suite(suite
);