1 | /*
|
---|
2 | Unix SMB/CIFS implementation.
|
---|
3 |
|
---|
4 | SMB2 getinfo test suite
|
---|
5 |
|
---|
6 | Copyright (C) Andrew Tridgell 2005
|
---|
7 |
|
---|
8 | This program is free software; you can redistribute it and/or modify
|
---|
9 | it under the terms of the GNU General Public License as published by
|
---|
10 | the Free Software Foundation; either version 3 of the License, or
|
---|
11 | (at your option) any later version.
|
---|
12 |
|
---|
13 | This program is distributed in the hope that it will be useful,
|
---|
14 | but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
15 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
16 | GNU General Public License for more details.
|
---|
17 |
|
---|
18 | You should have received a copy of the GNU General Public License
|
---|
19 | along with this program. If not, see <http://www.gnu.org/licenses/>.
|
---|
20 | */
|
---|
21 |
|
---|
22 | #include "includes.h"
|
---|
23 | #include "libcli/smb2/smb2.h"
|
---|
24 | #include "libcli/smb2/smb2_calls.h"
|
---|
25 |
|
---|
26 | #include "torture/torture.h"
|
---|
27 | #include "torture/smb2/proto.h"
|
---|
28 |
|
---|
29 | static struct {
|
---|
30 | const char *name;
|
---|
31 | uint16_t level;
|
---|
32 | NTSTATUS fstatus;
|
---|
33 | NTSTATUS dstatus;
|
---|
34 | union smb_fileinfo finfo;
|
---|
35 | union smb_fileinfo dinfo;
|
---|
36 | } file_levels[] = {
|
---|
37 | #define LEVEL(x) #x, x
|
---|
38 | { LEVEL(RAW_FILEINFO_BASIC_INFORMATION) },
|
---|
39 | { LEVEL(RAW_FILEINFO_STANDARD_INFORMATION) },
|
---|
40 | { LEVEL(RAW_FILEINFO_INTERNAL_INFORMATION) },
|
---|
41 | { LEVEL(RAW_FILEINFO_EA_INFORMATION) },
|
---|
42 | { LEVEL(RAW_FILEINFO_ACCESS_INFORMATION) },
|
---|
43 | { LEVEL(RAW_FILEINFO_POSITION_INFORMATION) },
|
---|
44 | { LEVEL(RAW_FILEINFO_MODE_INFORMATION) },
|
---|
45 | { LEVEL(RAW_FILEINFO_ALIGNMENT_INFORMATION) },
|
---|
46 | { LEVEL(RAW_FILEINFO_ALL_INFORMATION) },
|
---|
47 | { LEVEL(RAW_FILEINFO_ALT_NAME_INFORMATION) },
|
---|
48 | { LEVEL(RAW_FILEINFO_STREAM_INFORMATION) },
|
---|
49 | { LEVEL(RAW_FILEINFO_COMPRESSION_INFORMATION) },
|
---|
50 | { LEVEL(RAW_FILEINFO_NETWORK_OPEN_INFORMATION) },
|
---|
51 | { LEVEL(RAW_FILEINFO_ATTRIBUTE_TAG_INFORMATION) },
|
---|
52 |
|
---|
53 | { LEVEL(RAW_FILEINFO_SMB2_ALL_EAS) },
|
---|
54 |
|
---|
55 | { LEVEL(RAW_FILEINFO_SMB2_ALL_INFORMATION) },
|
---|
56 | { LEVEL(RAW_FILEINFO_SEC_DESC) }
|
---|
57 | };
|
---|
58 |
|
---|
59 | static struct {
|
---|
60 | const char *name;
|
---|
61 | uint16_t level;
|
---|
62 | NTSTATUS status;
|
---|
63 | union smb_fsinfo info;
|
---|
64 | } fs_levels[] = {
|
---|
65 | { LEVEL(RAW_QFS_VOLUME_INFORMATION) },
|
---|
66 | { LEVEL(RAW_QFS_SIZE_INFORMATION) },
|
---|
67 | { LEVEL(RAW_QFS_DEVICE_INFORMATION) },
|
---|
68 | { LEVEL(RAW_QFS_ATTRIBUTE_INFORMATION) },
|
---|
69 | { LEVEL(RAW_QFS_QUOTA_INFORMATION) },
|
---|
70 | { LEVEL(RAW_QFS_FULL_SIZE_INFORMATION) },
|
---|
71 | { LEVEL(RAW_QFS_OBJECTID_INFORMATION) }
|
---|
72 | };
|
---|
73 |
|
---|
74 | #define FNAME "testsmb2_file.dat"
|
---|
75 | #define DNAME "testsmb2_dir"
|
---|
76 |
|
---|
77 | /*
|
---|
78 | test fileinfo levels
|
---|
79 | */
|
---|
80 | static bool torture_smb2_fileinfo(struct torture_context *tctx, struct smb2_tree *tree)
|
---|
81 | {
|
---|
82 | struct smb2_handle hfile, hdir;
|
---|
83 | NTSTATUS status;
|
---|
84 | int i;
|
---|
85 |
|
---|
86 | status = torture_smb2_testfile(tree, FNAME, &hfile);
|
---|
87 | if (!NT_STATUS_IS_OK(status)) {
|
---|
88 | printf(__location__ " Unable to create test file '%s' - %s\n", FNAME, nt_errstr(status));
|
---|
89 | goto failed;
|
---|
90 | }
|
---|
91 |
|
---|
92 | status = torture_smb2_testdir(tree, DNAME, &hdir);
|
---|
93 | if (!NT_STATUS_IS_OK(status)) {
|
---|
94 | printf(__location__ " Unable to create test directory '%s' - %s\n", DNAME, nt_errstr(status));
|
---|
95 | goto failed;
|
---|
96 | }
|
---|
97 |
|
---|
98 | printf("Testing file info levels\n");
|
---|
99 | torture_smb2_all_info(tree, hfile);
|
---|
100 | torture_smb2_all_info(tree, hdir);
|
---|
101 |
|
---|
102 | for (i=0;i<ARRAY_SIZE(file_levels);i++) {
|
---|
103 | if (file_levels[i].level == RAW_FILEINFO_SEC_DESC) {
|
---|
104 | file_levels[i].finfo.query_secdesc.in.secinfo_flags = 0x7;
|
---|
105 | file_levels[i].dinfo.query_secdesc.in.secinfo_flags = 0x7;
|
---|
106 | }
|
---|
107 | if (file_levels[i].level == RAW_FILEINFO_SMB2_ALL_EAS) {
|
---|
108 | file_levels[i].finfo.all_eas.in.continue_flags =
|
---|
109 | SMB2_CONTINUE_FLAG_RESTART;
|
---|
110 | file_levels[i].dinfo.all_eas.in.continue_flags =
|
---|
111 | SMB2_CONTINUE_FLAG_RESTART;
|
---|
112 | }
|
---|
113 | file_levels[i].finfo.generic.level = file_levels[i].level;
|
---|
114 | file_levels[i].finfo.generic.in.file.handle = hfile;
|
---|
115 | file_levels[i].fstatus = smb2_getinfo_file(tree, tree, &file_levels[i].finfo);
|
---|
116 | if (!NT_STATUS_IS_OK(file_levels[i].fstatus)) {
|
---|
117 | printf("(%s) %s failed on file - %s\n", __location__,
|
---|
118 | file_levels[i].name, nt_errstr(file_levels[i].fstatus));
|
---|
119 | goto failed;
|
---|
120 | }
|
---|
121 | file_levels[i].dinfo.generic.level = file_levels[i].level;
|
---|
122 | file_levels[i].dinfo.generic.in.file.handle = hdir;
|
---|
123 | file_levels[i].dstatus = smb2_getinfo_file(tree, tree, &file_levels[i].dinfo);
|
---|
124 | if (!NT_STATUS_IS_OK(file_levels[i].dstatus)) {
|
---|
125 | printf("(%s) %s failed on dir - %s\n", __location__,
|
---|
126 | file_levels[i].name, nt_errstr(file_levels[i].dstatus));
|
---|
127 | goto failed;
|
---|
128 | }
|
---|
129 | }
|
---|
130 |
|
---|
131 | return true;
|
---|
132 |
|
---|
133 | failed:
|
---|
134 | return false;
|
---|
135 | }
|
---|
136 |
|
---|
137 |
|
---|
138 | /*
|
---|
139 | test fsinfo levels
|
---|
140 | */
|
---|
141 | static bool torture_smb2_fsinfo(struct smb2_tree *tree)
|
---|
142 | {
|
---|
143 | int i;
|
---|
144 | NTSTATUS status;
|
---|
145 | struct smb2_handle handle;
|
---|
146 |
|
---|
147 | printf("Testing fsinfo levels\n");
|
---|
148 | status = smb2_util_roothandle(tree, &handle);
|
---|
149 | if (!NT_STATUS_IS_OK(status)) {
|
---|
150 | printf(__location__ " Unable to create root handle - %s\n", nt_errstr(status));
|
---|
151 | return false;
|
---|
152 | }
|
---|
153 |
|
---|
154 | for (i=0;i<ARRAY_SIZE(fs_levels);i++) {
|
---|
155 | fs_levels[i].info.generic.level = fs_levels[i].level;
|
---|
156 | fs_levels[i].info.generic.handle = handle;
|
---|
157 | fs_levels[i].status = smb2_getinfo_fs(tree, tree, &fs_levels[i].info);
|
---|
158 | if (!NT_STATUS_IS_OK(fs_levels[i].status)) {
|
---|
159 | printf("%s failed - %s\n", fs_levels[i].name, nt_errstr(fs_levels[i].status));
|
---|
160 | return false;
|
---|
161 | }
|
---|
162 | }
|
---|
163 |
|
---|
164 | return true;
|
---|
165 | }
|
---|
166 |
|
---|
167 |
|
---|
168 | /*
|
---|
169 | test for buffer size handling
|
---|
170 | */
|
---|
171 | static bool torture_smb2_buffercheck(struct smb2_tree *tree)
|
---|
172 | {
|
---|
173 | NTSTATUS status;
|
---|
174 | struct smb2_handle handle;
|
---|
175 | struct smb2_getinfo b;
|
---|
176 |
|
---|
177 | printf("Testing buffer size handling\n");
|
---|
178 | status = smb2_util_roothandle(tree, &handle);
|
---|
179 | if (!NT_STATUS_IS_OK(status)) {
|
---|
180 | printf(__location__ " Unable to create root handle - %s\n", nt_errstr(status));
|
---|
181 | return false;
|
---|
182 | }
|
---|
183 |
|
---|
184 | ZERO_STRUCT(b);
|
---|
185 | b.in.info_type = SMB2_GETINFO_FS;
|
---|
186 | b.in.info_class = 1;
|
---|
187 | b.in.output_buffer_length = 0x1;
|
---|
188 | b.in.input_buffer_length = 0;
|
---|
189 | b.in.file.handle = handle;
|
---|
190 |
|
---|
191 | status = smb2_getinfo(tree, tree, &b);
|
---|
192 | if (!NT_STATUS_EQUAL(status, NT_STATUS_INFO_LENGTH_MISMATCH)) {
|
---|
193 | printf(__location__ " Wrong error code for small buffer %s\n",
|
---|
194 | nt_errstr(status));
|
---|
195 | return false;
|
---|
196 | }
|
---|
197 |
|
---|
198 | return true;
|
---|
199 | }
|
---|
200 |
|
---|
201 |
|
---|
202 | /* basic testing of all SMB2 getinfo levels
|
---|
203 | */
|
---|
204 | bool torture_smb2_getinfo(struct torture_context *torture)
|
---|
205 | {
|
---|
206 | TALLOC_CTX *mem_ctx = talloc_new(NULL);
|
---|
207 | struct smb2_tree *tree;
|
---|
208 | bool ret = true;
|
---|
209 | NTSTATUS status;
|
---|
210 |
|
---|
211 | if (!torture_smb2_connection(torture, &tree)) {
|
---|
212 | return false;
|
---|
213 | }
|
---|
214 |
|
---|
215 | smb2_deltree(tree, FNAME);
|
---|
216 | smb2_deltree(tree, DNAME);
|
---|
217 |
|
---|
218 | status = torture_setup_complex_file(tree, FNAME);
|
---|
219 | if (!NT_STATUS_IS_OK(status)) {
|
---|
220 | return false;
|
---|
221 | }
|
---|
222 | torture_setup_complex_file(tree, FNAME ":streamtwo");
|
---|
223 | status = torture_setup_complex_dir(tree, DNAME);
|
---|
224 | if (!NT_STATUS_IS_OK(status)) {
|
---|
225 | return false;
|
---|
226 | }
|
---|
227 | torture_setup_complex_file(tree, DNAME ":streamtwo");
|
---|
228 |
|
---|
229 | ret &= torture_smb2_fileinfo(torture, tree);
|
---|
230 | ret &= torture_smb2_fsinfo(tree);
|
---|
231 | ret &= torture_smb2_buffercheck(tree);
|
---|
232 |
|
---|
233 | talloc_free(mem_ctx);
|
---|
234 |
|
---|
235 | return ret;
|
---|
236 | }
|
---|