?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
s3:winbindd: use struct unixid instead of uint64 in Sids2Xids parent<->child
[Samba/gbeck.git] / source4 / torture / smb2 / util.c
blob71e87f3961e44c9a55ea1aac0c64d5de2d578c65
1 /*
2 Unix SMB/CIFS implementation.
3
4 helper functions for SMB2 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/security/security_descriptor.h"
24 #include "libcli/smb2/smb2.h"
25 #include "libcli/smb2/smb2_calls.h"
26 #include "../libcli/smb/smbXcli_base.h"
27 #include "lib/cmdline/popt_common.h"
28 #include "system/time.h"
29 #include "librpc/gen_ndr/ndr_security.h"
30 #include "param/param.h"
31 #include "libcli/resolve/resolve.h"
32
33 #include "torture/torture.h"
34 #include "torture/smb2/proto.h"
35
36
37 /*
38 write to a file on SMB2
39 */
40 NTSTATUS smb2_util_write(struct smb2_tree *tree,
41 struct smb2_handle handle,
42 const void *buf, off_t offset, size_t size)
43 {
44 struct smb2_write w;
45
46 ZERO_STRUCT(w);
47 w.in.file.handle = handle;
48 w.in.offset = offset;
49 w.in.data = data_blob_const(buf, size);
50
51 return smb2_write(tree, &w);
52 }
53
54 /*
55 create a complex file/dir using the SMB2 protocol
56 */
57 static NTSTATUS smb2_create_complex(struct smb2_tree *tree, const char *fname,
58 struct smb2_handle *handle, bool dir)
59 {
60 TALLOC_CTX *tmp_ctx = talloc_new(tree);
61 char buf[7] = "abc";
62 struct smb2_create io;
63 union smb_setfileinfo setfile;
64 union smb_fileinfo fileinfo;
65 time_t t = (time(NULL) & ~1);
66 NTSTATUS status;
67
68 smb2_util_unlink(tree, fname);
69 ZERO_STRUCT(io);
70 io.in.desired_access = SEC_FLAG_MAXIMUM_ALLOWED;
71 io.in.file_attributes = FILE_ATTRIBUTE_NORMAL;
72 io.in.create_disposition = NTCREATEX_DISP_OVERWRITE_IF;
73 io.in.share_access =
74 NTCREATEX_SHARE_ACCESS_DELETE|
75 NTCREATEX_SHARE_ACCESS_READ|
76 NTCREATEX_SHARE_ACCESS_WRITE;
77 io.in.create_options = 0;
78 io.in.fname = fname;
79 if (dir) {
80 io.in.create_options = NTCREATEX_OPTIONS_DIRECTORY;
81 io.in.share_access &= ~NTCREATEX_SHARE_ACCESS_DELETE;
82 io.in.file_attributes = FILE_ATTRIBUTE_DIRECTORY;
83 io.in.create_disposition = NTCREATEX_DISP_CREATE;
84 }
85
86 /* it seems vista is now fussier about alignment? */
87 if (strchr(fname, ':') == NULL) {
88 /* setup some EAs */
89 io.in.eas.num_eas = 2;
90 io.in.eas.eas = talloc_array(tmp_ctx, struct ea_struct, 2);
91 io.in.eas.eas[0].flags = 0;
92 io.in.eas.eas[0].name.s = "EAONE";
93 io.in.eas.eas[0].value = data_blob_talloc(tmp_ctx, "VALUE1", 6);
94 io.in.eas.eas[1].flags = 0;
95 io.in.eas.eas[1].name.s = "SECONDEA";
96 io.in.eas.eas[1].value = data_blob_talloc(tmp_ctx, "ValueTwo", 8);
97 }
98
99 status = smb2_create(tree, tmp_ctx, &io);
100 talloc_free(tmp_ctx);
101 NT_STATUS_NOT_OK_RETURN(status);
102
103 *handle = io.out.file.handle;
104
105 if (!dir) {
106 status = smb2_util_write(tree, *handle, buf, 0, sizeof(buf));
107 NT_STATUS_NOT_OK_RETURN(status);
108 }
109
110 /* make sure all the timestamps aren't the same, and are also
111 in different DST zones*/
112 setfile.generic.level = RAW_SFILEINFO_BASIC_INFORMATION;
113 setfile.generic.in.file.handle = *handle;
114
115 unix_to_nt_time(&setfile.basic_info.in.create_time, t + 9*30*24*60*60);
116 unix_to_nt_time(&setfile.basic_info.in.access_time, t + 6*30*24*60*60);
117 unix_to_nt_time(&setfile.basic_info.in.write_time, t + 3*30*24*60*60);
118 unix_to_nt_time(&setfile.basic_info.in.change_time, t + 1*30*24*60*60);
119 setfile.basic_info.in.attrib = FILE_ATTRIBUTE_NORMAL;
120
121 status = smb2_setinfo_file(tree, &setfile);
122 if (!NT_STATUS_IS_OK(status)) {
123 printf("Failed to setup file times - %s\n", nt_errstr(status));
124 return status;
125 }
126
127 /* make sure all the timestamps aren't the same */
128 fileinfo.generic.level = RAW_FILEINFO_SMB2_ALL_INFORMATION;
129 fileinfo.generic.in.file.handle = *handle;
130
131 status = smb2_getinfo_file(tree, tree, &fileinfo);
132 if (!NT_STATUS_IS_OK(status)) {
133 printf("Failed to query file times - %s\n", nt_errstr(status));
134 return status;
135
136 }
137
138 #define CHECK_TIME(field) do {\
139 if (setfile.basic_info.in.field != fileinfo.all_info2.out.field) { \
140 printf("(%s) " #field " not setup correctly: %s(%llu) => %s(%llu)\n", \
141 __location__, \
142 nt_time_string(tree, setfile.basic_info.in.field), \
143 (unsigned long long)setfile.basic_info.in.field, \
144 nt_time_string(tree, fileinfo.basic_info.out.field), \
145 (unsigned long long)fileinfo.basic_info.out.field); \
146 status = NT_STATUS_INVALID_PARAMETER; \
147 } \
148 } while (0)
149
150 CHECK_TIME(create_time);
151 CHECK_TIME(access_time);
152 CHECK_TIME(write_time);
153 CHECK_TIME(change_time);
154
155 return status;
156 }
157
158 /*
159 create a complex file using the SMB2 protocol
160 */
161 NTSTATUS smb2_create_complex_file(struct smb2_tree *tree, const char *fname,
162 struct smb2_handle *handle)
163 {
164 return smb2_create_complex(tree, fname, handle, false);
165 }
166
167 /*
168 create a complex dir using the SMB2 protocol
169 */
170 NTSTATUS smb2_create_complex_dir(struct smb2_tree *tree, const char *fname,
171 struct smb2_handle *handle)
172 {
173 return smb2_create_complex(tree, fname, handle, true);
174 }
175
176 /*
177 show lots of information about a file
178 */
179 void torture_smb2_all_info(struct smb2_tree *tree, struct smb2_handle handle)
180 {
181 NTSTATUS status;
182 TALLOC_CTX *tmp_ctx = talloc_new(tree);
183 union smb_fileinfo io;
184
185 io.generic.level = RAW_FILEINFO_SMB2_ALL_INFORMATION;
186 io.generic.in.file.handle = handle;
187
188 status = smb2_getinfo_file(tree, tmp_ctx, &io);
189 if (!NT_STATUS_IS_OK(status)) {
190 DEBUG(0,("getinfo failed - %s\n", nt_errstr(status)));
191 talloc_free(tmp_ctx);
192 return;
193 }
194
195 d_printf("all_info for '%s'\n", io.all_info2.out.fname.s);
196 d_printf("\tcreate_time: %s\n", nt_time_string(tmp_ctx, io.all_info2.out.create_time));
197 d_printf("\taccess_time: %s\n", nt_time_string(tmp_ctx, io.all_info2.out.access_time));
198 d_printf("\twrite_time: %s\n", nt_time_string(tmp_ctx, io.all_info2.out.write_time));
199 d_printf("\tchange_time: %s\n", nt_time_string(tmp_ctx, io.all_info2.out.change_time));
200 d_printf("\tattrib: 0x%x\n", io.all_info2.out.attrib);
201 d_printf("\tunknown1: 0x%x\n", io.all_info2.out.unknown1);
202 d_printf("\talloc_size: %llu\n", (long long)io.all_info2.out.alloc_size);
203 d_printf("\tsize: %llu\n", (long long)io.all_info2.out.size);
204 d_printf("\tnlink: %u\n", io.all_info2.out.nlink);
205 d_printf("\tdelete_pending: %u\n", io.all_info2.out.delete_pending);
206 d_printf("\tdirectory: %u\n", io.all_info2.out.directory);
207 d_printf("\tfile_id: %llu\n", (long long)io.all_info2.out.file_id);
208 d_printf("\tea_size: %u\n", io.all_info2.out.ea_size);
209 d_printf("\taccess_mask: 0x%08x\n", io.all_info2.out.access_mask);
210 d_printf("\tposition: 0x%llx\n", (long long)io.all_info2.out.position);
211 d_printf("\tmode: 0x%llx\n", (long long)io.all_info2.out.mode);
212
213 /* short name, if any */
214 io.generic.level = RAW_FILEINFO_ALT_NAME_INFORMATION;
215 status = smb2_getinfo_file(tree, tmp_ctx, &io);
216 if (NT_STATUS_IS_OK(status)) {
217 d_printf("\tshort name: '%s'\n", io.alt_name_info.out.fname.s);
218 }
219
220 /* the EAs, if any */
221 io.generic.level = RAW_FILEINFO_SMB2_ALL_EAS;
222 status = smb2_getinfo_file(tree, tmp_ctx, &io);
223 if (NT_STATUS_IS_OK(status)) {
224 int i;
225 for (i=0;i<io.all_eas.out.num_eas;i++) {
226 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
227 io.all_eas.out.eas[i].flags,
228 (int)io.all_eas.out.eas[i].value.length,
229 io.all_eas.out.eas[i].name.s);
230 }
231 }
232
233 /* streams, if available */
234 io.generic.level = RAW_FILEINFO_STREAM_INFORMATION;
235 status = smb2_getinfo_file(tree, tmp_ctx, &io);
236 if (NT_STATUS_IS_OK(status)) {
237 int i;
238 for (i=0;i<io.stream_info.out.num_streams;i++) {
239 d_printf("\tstream %d:\n", i);
240 d_printf("\t\tsize %ld\n",
241 (long)io.stream_info.out.streams[i].size);
242 d_printf("\t\talloc size %ld\n",
243 (long)io.stream_info.out.streams[i].alloc_size);
244 d_printf("\t\tname %s\n", io.stream_info.out.streams[i].stream_name.s);
245 }
246 }
247
248 if (DEBUGLVL(1)) {
249 /* the security descriptor */
250 io.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
251 io.query_secdesc.in.secinfo_flags =
252 SECINFO_OWNER|SECINFO_GROUP|
253 SECINFO_DACL;
254 status = smb2_getinfo_file(tree, tmp_ctx, &io);
255 if (NT_STATUS_IS_OK(status)) {
256 NDR_PRINT_DEBUG(security_descriptor, io.query_secdesc.out.sd);
257 }
258 }
259
260 talloc_free(tmp_ctx);
261 }
262
263 /**
264 * open a smb2 tree connect
265 */
266 bool torture_smb2_tree_connect(struct torture_context *tctx,
267 struct smb2_session *session,
268 TALLOC_CTX *mem_ctx,
269 struct smb2_tree **_tree)
270 {
271 NTSTATUS status;
272 const char *host = torture_setting_string(tctx, "host", NULL);
273 const char *share = torture_setting_string(tctx, "share", NULL);
274 struct smb2_tree_connect tcon;
275 struct smb2_tree *tree;
276
277 ZERO_STRUCT(tcon);
278 tcon.in.reserved = 0;
279 tcon.in.path = talloc_asprintf(tctx, "\\\\%s\\%s", host, share);
280 if (tcon.in.path == NULL) {
281 printf("talloc failed\n");
282 return false;
283 }
284
285 status = smb2_tree_connect(session, &tcon);
286 if (!NT_STATUS_IS_OK(status)) {
287 printf("Failed to tree_connect to SMB2 share \\\\%s\\%s - %s\n",
288 host, share, nt_errstr(status));
289 return false;
290 }
291
292 tree = smb2_tree_init(session, mem_ctx, true);
293 if (tree == NULL) {
294 printf("talloc failed\n");
295 return false;
296 }
297
298 smb2cli_tcon_set_values(tree->smbXcli,
299 tree->session->smbXcli,
300 tcon.out.tid,
301 tcon.out.share_type,
302 tcon.out.flags,
303 tcon.out.capabilities,
304 tcon.out.access_mask);
305
306 *_tree = tree;
307
308 return true;
309 }
310
311 /**
312 * do a smb2 session setup (without a tree connect)
313 */
314 bool torture_smb2_session_setup(struct torture_context *tctx,
315 struct smb2_transport *transport,
316 uint64_t previous_session_id,
317 TALLOC_CTX *mem_ctx,
318 struct smb2_session **_session)
319 {
320 NTSTATUS status;
321 struct smb2_session *session;
322 struct cli_credentials *credentials = cmdline_credentials;
323
324 session = smb2_session_init(transport,
325 lpcfg_gensec_settings(tctx, tctx->lp_ctx),
326 mem_ctx, true);
327
328 if (session == NULL) {
329 return false;
330 }
331
332 status = smb2_session_setup_spnego(session, credentials,
333 previous_session_id);
334 if (!NT_STATUS_IS_OK(status)) {
335 printf("session setup failed: %s\n", nt_errstr(status));
336 talloc_free(session);
337 return false;
338 }
339
340 *_session = session;
341
342 return true;
343 }
344
345 /*
346 open a smb2 connection
347 */
348 bool torture_smb2_connection_ext(struct torture_context *tctx,
349 uint64_t previous_session_id,
350 struct smb2_tree **tree)
351 {
352 NTSTATUS status;
353 const char *host = torture_setting_string(tctx, "host", NULL);
354 const char *share = torture_setting_string(tctx, "share", NULL);
355 struct cli_credentials *credentials = cmdline_credentials;
356 struct smbcli_options options;
357
358 lpcfg_smbcli_options(tctx->lp_ctx, &options);
359
360 status = smb2_connect_ext(tctx,
361 host,
362 lpcfg_smb_ports(tctx->lp_ctx),
363 share,
364 lpcfg_resolve_context(tctx->lp_ctx),
365 credentials,
366 previous_session_id,
367 tree,
368 tctx->ev,
369 &options,
370 lpcfg_socket_options(tctx->lp_ctx),
371 lpcfg_gensec_settings(tctx, tctx->lp_ctx)
372 );
373 if (!NT_STATUS_IS_OK(status)) {
374 printf("Failed to connect to SMB2 share \\\\%s\\%s - %s\n",
375 host, share, nt_errstr(status));
376 return false;
377 }
378 return true;
379 }
380
381 bool torture_smb2_connection(struct torture_context *tctx, struct smb2_tree **tree)
382 {
383 bool ret;
384
385 ret = torture_smb2_connection_ext(tctx, 0, tree);
386
387 return ret;
388 }
389
390
391 /*
392 create and return a handle to a test file
393 */
394 NTSTATUS torture_smb2_testfile(struct smb2_tree *tree, const char *fname,
395 struct smb2_handle *handle)
396 {
397 struct smb2_create io;
398 NTSTATUS status;
399
400 ZERO_STRUCT(io);
401 io.in.oplock_level = 0;
402 io.in.desired_access = SEC_RIGHTS_FILE_ALL;
403 io.in.file_attributes = FILE_ATTRIBUTE_NORMAL;
404 io.in.create_disposition = NTCREATEX_DISP_OPEN_IF;
405 io.in.share_access =
406 NTCREATEX_SHARE_ACCESS_DELETE|
407 NTCREATEX_SHARE_ACCESS_READ|
408 NTCREATEX_SHARE_ACCESS_WRITE;
409 io.in.create_options = 0;
410 io.in.fname = fname;
411
412 status = smb2_create(tree, tree, &io);
413 NT_STATUS_NOT_OK_RETURN(status);
414
415 *handle = io.out.file.handle;
416
417 return NT_STATUS_OK;
418 }
419
420 /*
421 create and return a handle to a test directory
422 */
423 NTSTATUS torture_smb2_testdir(struct smb2_tree *tree, const char *fname,
424 struct smb2_handle *handle)
425 {
426 struct smb2_create io;
427 NTSTATUS status;
428
429 ZERO_STRUCT(io);
430 io.in.oplock_level = 0;
431 io.in.desired_access = SEC_RIGHTS_DIR_ALL;
432 io.in.file_attributes = FILE_ATTRIBUTE_DIRECTORY;
433 io.in.create_disposition = NTCREATEX_DISP_OPEN_IF;
434 io.in.share_access = NTCREATEX_SHARE_ACCESS_READ|NTCREATEX_SHARE_ACCESS_WRITE|NTCREATEX_SHARE_ACCESS_DELETE;
435 io.in.create_options = NTCREATEX_OPTIONS_DIRECTORY;
436 io.in.fname = fname;
437
438 status = smb2_create(tree, tree, &io);
439 NT_STATUS_NOT_OK_RETURN(status);
440
441 *handle = io.out.file.handle;
442
443 return NT_STATUS_OK;
444 }
445
446
447 /*
448 create a complex file using SMB2, to make it easier to
449 find fields in SMB2 getinfo levels
450 */
451 NTSTATUS torture_setup_complex_file(struct smb2_tree *tree, const char *fname)
452 {
453 struct smb2_handle handle;
454 NTSTATUS status = smb2_create_complex_file(tree, fname, &handle);
455 NT_STATUS_NOT_OK_RETURN(status);
456 return smb2_util_close(tree, handle);
457 }
458
459
460 /*
461 create a complex dir using SMB2, to make it easier to
462 find fields in SMB2 getinfo levels
463 */
464 NTSTATUS torture_setup_complex_dir(struct smb2_tree *tree, const char *fname)
465 {
466 struct smb2_handle handle;
467 NTSTATUS status = smb2_create_complex_dir(tree, fname, &handle);
468 NT_STATUS_NOT_OK_RETURN(status);
469 return smb2_util_close(tree, handle);
470 }
471
472
473 /*
474 return a handle to the root of the share
475 */
476 NTSTATUS smb2_util_roothandle(struct smb2_tree *tree, struct smb2_handle *handle)
477 {
478 struct smb2_create io;
479 NTSTATUS status;
480
481 ZERO_STRUCT(io);
482 io.in.oplock_level = 0;
483 io.in.desired_access = SEC_STD_SYNCHRONIZE | SEC_DIR_READ_ATTRIBUTE | SEC_DIR_LIST;
484 io.in.file_attributes = 0;
485 io.in.create_disposition = NTCREATEX_DISP_OPEN;
486 io.in.share_access = NTCREATEX_SHARE_ACCESS_READ|NTCREATEX_SHARE_ACCESS_DELETE;
487 io.in.create_options = NTCREATEX_OPTIONS_ASYNC_ALERT;
488 io.in.fname = NULL;
489
490 status = smb2_create(tree, tree, &io);
491 NT_STATUS_NOT_OK_RETURN(status);
492
493 *handle = io.out.file.handle;
494
495 return NT_STATUS_OK;
496 }
497
498 /* Comparable to torture_setup_dir, but for SMB2. */
499 bool smb2_util_setup_dir(struct torture_context *tctx, struct smb2_tree *tree,
500 const char *dname)
501 {
502 NTSTATUS status;
503
504 /* XXX: smb_raw_exit equivalent?
505 smb_raw_exit(cli->session); */
506 if (smb2_deltree(tree, dname) == -1) {
507 torture_result(tctx, TORTURE_ERROR, "Unable to deltree when setting up %s.\n", dname);
508 return false;
509 }
510
511 status = smb2_util_mkdir(tree, dname);
512 if (NT_STATUS_IS_ERR(status)) {
513 torture_result(tctx, TORTURE_ERROR, "Unable to mkdir when setting up %s - %s\n", dname,
514 nt_errstr(status));
515 return false;
516 }
517
518 return true;
519 }
520
521 #define CHECK_STATUS(status, correct) do { \
522 if (!NT_STATUS_EQUAL(status, correct)) { \
523 torture_result(tctx, TORTURE_FAIL, "(%s) Incorrect status %s - should be %s\n", \
524 __location__, nt_errstr(status), nt_errstr(correct)); \
525 ret = false; \
526 goto done; \
527 }} while (0)
528
529 /*
530 * Helper function to verify a security descriptor, by querying
531 * and comparing against the passed in sd.
532 */
533 bool smb2_util_verify_sd(TALLOC_CTX *tctx, struct smb2_tree *tree,
534 struct smb2_handle handle, struct security_descriptor *sd)
535 {
536 NTSTATUS status;
537 bool ret = true;
538 union smb_fileinfo q = {};
539
540 q.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
541 q.query_secdesc.in.file.handle = handle;
542 q.query_secdesc.in.secinfo_flags =
543 SECINFO_OWNER |
544 SECINFO_GROUP |
545 SECINFO_DACL;
546 status = smb2_getinfo_file(tree, tctx, &q);
547 CHECK_STATUS(status, NT_STATUS_OK);
548
549 if (!security_acl_equal(
550 q.query_secdesc.out.sd->dacl, sd->dacl)) {
551 torture_warning(tctx, "%s: security descriptors don't match!\n",
552 __location__);
553 torture_warning(tctx, "got:\n");
554 NDR_PRINT_DEBUG(security_descriptor,
555 q.query_secdesc.out.sd);
556 torture_warning(tctx, "expected:\n");
557 NDR_PRINT_DEBUG(security_descriptor, sd);
558 ret = false;
559 }
560
561 done:
562 return ret;
563 }
564
565 /*
566 * Helper function to verify attributes, by querying
567 * and comparing against the passed in attrib.
568 */
569 bool smb2_util_verify_attrib(TALLOC_CTX *tctx, struct smb2_tree *tree,
570 struct smb2_handle handle, uint32_t attrib)
571 {
572 NTSTATUS status;
573 bool ret = true;
574 union smb_fileinfo q = {};
575
576 q.standard.level = RAW_FILEINFO_SMB2_ALL_INFORMATION;
577 q.standard.in.file.handle = handle;
578 status = smb2_getinfo_file(tree, tctx, &q);
579 CHECK_STATUS(status, NT_STATUS_OK);
580
581 q.all_info2.out.attrib &= ~(FILE_ATTRIBUTE_ARCHIVE | FILE_ATTRIBUTE_NONINDEXED);
582
583 if (q.all_info2.out.attrib != attrib) {
584 torture_warning(tctx, "%s: attributes don't match! "
585 "got %x, expected %x\n", __location__,
586 (uint32_t)q.standard.out.attrib,
587 (uint32_t)attrib);
588 ret = false;
589 }
590
591 done:
592 return ret;
593 }
594
595
596 uint32_t smb2_util_lease_state(const char *ls)
597 {
598 uint32_t val = 0;
599 int i;
600
601 for (i = 0; i < strlen(ls); i++) {
602 switch (ls[i]) {
603 case 'R':
604 val |= SMB2_LEASE_READ;
605 break;
606 case 'H':
607 val |= SMB2_LEASE_HANDLE;
608 break;
609 case 'W':
610 val |= SMB2_LEASE_WRITE;
611 break;
612 }
613 }
614
615 return val;
616 }
617
618
619 uint32_t smb2_util_share_access(const char *sharemode)
620 {
621 uint32_t val = NTCREATEX_SHARE_ACCESS_NONE; /* 0 */
622 int i;
623
624 for (i = 0; i < strlen(sharemode); i++) {
625 switch(sharemode[i]) {
626 case 'R':
627 val |= NTCREATEX_SHARE_ACCESS_READ;
628 break;
629 case 'W':
630 val |= NTCREATEX_SHARE_ACCESS_WRITE;
631 break;
632 case 'D':
633 val |= NTCREATEX_SHARE_ACCESS_DELETE;
634 break;
635 }
636 }
637
638 return val;
639 }
640
641 uint8_t smb2_util_oplock_level(const char *op)
642 {
643 uint8_t val = SMB2_OPLOCK_LEVEL_NONE;
644 int i;
645
646 for (i = 0; i < strlen(op); i++) {
647 switch (op[i]) {
648 case 's':
649 return SMB2_OPLOCK_LEVEL_II;
650 case 'x':
651 return SMB2_OPLOCK_LEVEL_EXCLUSIVE;
652 case 'b':
653 return SMB2_OPLOCK_LEVEL_BATCH;
654 default:
655 continue;
656 }
657 }
658
659 return val;
660 }
661
662 /**
663 * Helper functions to fill a smb2_create struct for several
664 * open scenarios.
665 */
666 void smb2_generic_create_share(struct smb2_create *io, struct smb2_lease *ls,
667 bool dir, const char *name, uint32_t disposition,
668 uint32_t share_access,
669 uint8_t oplock, uint64_t leasekey,
670 uint32_t leasestate)
671 {
672 ZERO_STRUCT(*io);
673 io->in.security_flags = 0x00;
674 io->in.oplock_level = oplock;
675 io->in.impersonation_level = NTCREATEX_IMPERSONATION_IMPERSONATION;
676 io->in.create_flags = 0x00000000;
677 io->in.reserved = 0x00000000;
678 io->in.desired_access = SEC_RIGHTS_FILE_ALL;
679 io->in.file_attributes = FILE_ATTRIBUTE_NORMAL;
680 io->in.share_access = share_access;
681 io->in.create_disposition = disposition;
682 io->in.create_options = NTCREATEX_OPTIONS_SEQUENTIAL_ONLY |
683 NTCREATEX_OPTIONS_ASYNC_ALERT |
684 NTCREATEX_OPTIONS_NON_DIRECTORY_FILE |
685 0x00200000;
686 io->in.fname = name;
687
688 if (dir) {
689 io->in.create_options = NTCREATEX_OPTIONS_DIRECTORY;
690 io->in.share_access &= ~NTCREATEX_SHARE_ACCESS_DELETE;
691 io->in.file_attributes = FILE_ATTRIBUTE_DIRECTORY;
692 io->in.create_disposition = NTCREATEX_DISP_CREATE;
693 }
694
695 if (ls) {
696 ZERO_STRUCT(*ls);
697 ls->lease_key.data[0] = leasekey;
698 ls->lease_key.data[1] = ~leasekey;
699 ls->lease_state = leasestate;
700 io->in.lease_request = ls;
701 }
702 }
703
704 void smb2_generic_create(struct smb2_create *io, struct smb2_lease *ls,
705 bool dir, const char *name, uint32_t disposition,
706 uint8_t oplock, uint64_t leasekey,
707 uint32_t leasestate)
708 {
709 smb2_generic_create_share(io, ls, dir, name, disposition,
710 smb2_util_share_access("RWD"),
711 oplock,
712 leasekey, leasestate);
713 }
714
715 void smb2_lease_create_share(struct smb2_create *io, struct smb2_lease *ls,
716 bool dir, const char *name, uint32_t share_access,
717 uint64_t leasekey, uint32_t leasestate)
718 {
719 smb2_generic_create_share(io, ls, dir, name, NTCREATEX_DISP_OPEN_IF,
720 share_access, SMB2_OPLOCK_LEVEL_LEASE,
721 leasekey, leasestate);
722 }
723
724 void smb2_lease_create(struct smb2_create *io, struct smb2_lease *ls,
725 bool dir, const char *name, uint64_t leasekey,
726 uint32_t leasestate)
727 {
728 smb2_lease_create_share(io, ls, dir, name,
729 smb2_util_share_access("RWD"),
730 leasekey, leasestate);
731 }
732
733 void smb2_oplock_create_share(struct smb2_create *io, const char *name,
734 uint32_t share_access, uint8_t oplock)
735 {
736 smb2_generic_create_share(io, NULL, false, name, NTCREATEX_DISP_OPEN_IF,
737 share_access, oplock, 0, 0);
738 }
739 void smb2_oplock_create(struct smb2_create *io, const char *name, uint8_t oplock)
740 {
741 smb2_oplock_create_share(io, name, smb2_util_share_access("RWD"),
742 oplock);
743 }
744
WIP