1 /*
2 Unix SMB/CIFS implementation.
3 client file operations
4 Copyright (C) Andrew Tridgell 1994-1998
5 Copyright (C) Jeremy Allison 2001-2009
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 "system/filesys.h"
23 #include "libsmb/libsmb.h"
24 #include "../lib/util/tevent_ntstatus.h"
25 #include "async_smb.h"
26 #include "libsmb/clirap.h"
27 #include "trans2.h"
28 #include "ntioctl.h"
29 #include "libcli/security/security.h"
30 #include "../libcli/smb/smbXcli_base.h"
31
32 struct cli_setpathinfo_state {
33 uint16_t setup;
34 uint8_t *param;
35 };
36
37 static void cli_setpathinfo_done(struct tevent_req *subreq);
38
cli_setpathinfo_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t level,const char * path,uint8_t * data,size_t data_len)39 struct tevent_req *cli_setpathinfo_send(TALLOC_CTX *mem_ctx,
40 struct tevent_context *ev,
41 struct cli_state *cli,
42 uint16_t level,
43 const char *path,
44 uint8_t *data,
45 size_t data_len)
46 {
47 struct tevent_req *req, *subreq;
48 struct cli_setpathinfo_state *state;
49 uint16_t additional_flags2 = 0;
50
51 req = tevent_req_create(mem_ctx, &state,
52 struct cli_setpathinfo_state);
53 if (req == NULL) {
54 return NULL;
55 }
56
57 /* Setup setup word. */
58 SSVAL(&state->setup, 0, TRANSACT2_SETPATHINFO);
59
60 /* Setup param array. */
61 state->param = talloc_zero_array(state, uint8_t, 6);
62 if (tevent_req_nomem(state->param, req)) {
63 return tevent_req_post(req, ev);
64 }
65 SSVAL(state->param, 0, level);
66
67 state->param = trans2_bytes_push_str(
68 state->param, smbXcli_conn_use_unicode(cli->conn), path, strlen(path)+1, NULL);
69 if (tevent_req_nomem(state->param, req)) {
70 return tevent_req_post(req, ev);
71 }
72
73 if (clistr_is_previous_version_path(path, NULL, NULL, NULL) &&
74 !INFO_LEVEL_IS_UNIX(level)) {
75 additional_flags2 = FLAGS2_REPARSE_PATH;
76 }
77
78 subreq = cli_trans_send(
79 state, /* mem ctx. */
80 ev, /* event ctx. */
81 cli, /* cli_state. */
82 additional_flags2, /* additional_flags2 */
83 SMBtrans2, /* cmd. */
84 NULL, /* pipe name. */
85 -1, /* fid. */
86 0, /* function. */
87 0, /* flags. */
88 &state->setup, /* setup. */
89 1, /* num setup uint16_t words. */
90 0, /* max returned setup. */
91 state->param, /* param. */
92 talloc_get_size(state->param), /* num param. */
93 2, /* max returned param. */
94 data, /* data. */
95 data_len, /* num data. */
96 0); /* max returned data. */
97
98 if (tevent_req_nomem(subreq, req)) {
99 return tevent_req_post(req, ev);
100 }
101 tevent_req_set_callback(subreq, cli_setpathinfo_done, req);
102 return req;
103 }
104
cli_setpathinfo_done(struct tevent_req * subreq)105 static void cli_setpathinfo_done(struct tevent_req *subreq)
106 {
107 NTSTATUS status = cli_trans_recv(subreq, NULL, NULL, NULL, 0, NULL,
108 NULL, 0, NULL, NULL, 0, NULL);
109 tevent_req_simple_finish_ntstatus(subreq, status);
110 }
111
cli_setpathinfo_recv(struct tevent_req * req)112 NTSTATUS cli_setpathinfo_recv(struct tevent_req *req)
113 {
114 return tevent_req_simple_recv_ntstatus(req);
115 }
116
cli_setpathinfo(struct cli_state * cli,uint16_t level,const char * path,uint8_t * data,size_t data_len)117 NTSTATUS cli_setpathinfo(struct cli_state *cli,
118 uint16_t level,
119 const char *path,
120 uint8_t *data,
121 size_t data_len)
122 {
123 TALLOC_CTX *frame = talloc_stackframe();
124 struct tevent_context *ev;
125 struct tevent_req *req;
126 NTSTATUS status = NT_STATUS_NO_MEMORY;
127
128 if (smbXcli_conn_has_async_calls(cli->conn)) {
129 /*
130 * Can't use sync call while an async call is in flight
131 */
132 status = NT_STATUS_INVALID_PARAMETER;
133 goto fail;
134 }
135 ev = samba_tevent_context_init(frame);
136 if (ev == NULL) {
137 goto fail;
138 }
139 req = cli_setpathinfo_send(ev, ev, cli, level, path, data, data_len);
140 if (req == NULL) {
141 goto fail;
142 }
143 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
144 goto fail;
145 }
146 status = cli_setpathinfo_recv(req);
147 fail:
148 TALLOC_FREE(frame);
149 return status;
150 }
151
152 /****************************************************************************
153 Hard/Symlink a file (UNIX extensions).
154 Creates new name (sym)linked to link_target.
155 ****************************************************************************/
156
157 struct cli_posix_link_internal_state {
158 uint8_t *data;
159 };
160
161 static void cli_posix_link_internal_done(struct tevent_req *subreq);
162
cli_posix_link_internal_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t level,const char * link_target,const char * newname)163 static struct tevent_req *cli_posix_link_internal_send(TALLOC_CTX *mem_ctx,
164 struct tevent_context *ev,
165 struct cli_state *cli,
166 uint16_t level,
167 const char *link_target,
168 const char *newname)
169 {
170 struct tevent_req *req = NULL, *subreq = NULL;
171 struct cli_posix_link_internal_state *state = NULL;
172
173 req = tevent_req_create(mem_ctx, &state,
174 struct cli_posix_link_internal_state);
175 if (req == NULL) {
176 return NULL;
177 }
178
179 /* Setup data array. */
180 state->data = talloc_array(state, uint8_t, 0);
181 if (tevent_req_nomem(state->data, req)) {
182 return tevent_req_post(req, ev);
183 }
184 state->data = trans2_bytes_push_str(
185 state->data, smbXcli_conn_use_unicode(cli->conn),
186 link_target, strlen(link_target)+1, NULL);
187
188 subreq = cli_setpathinfo_send(
189 state, ev, cli, level, newname,
190 state->data, talloc_get_size(state->data));
191 if (tevent_req_nomem(subreq, req)) {
192 return tevent_req_post(req, ev);
193 }
194 tevent_req_set_callback(subreq, cli_posix_link_internal_done, req);
195 return req;
196 }
197
cli_posix_link_internal_done(struct tevent_req * subreq)198 static void cli_posix_link_internal_done(struct tevent_req *subreq)
199 {
200 NTSTATUS status = cli_setpathinfo_recv(subreq);
201 tevent_req_simple_finish_ntstatus(subreq, status);
202 }
203
cli_posix_link_internal_recv(struct tevent_req * req)204 static NTSTATUS cli_posix_link_internal_recv(struct tevent_req *req)
205 {
206 return tevent_req_simple_recv_ntstatus(req);
207 }
208
209 /****************************************************************************
210 Symlink a file (UNIX extensions).
211 ****************************************************************************/
212
213 struct cli_posix_symlink_state {
214 uint8_t dummy;
215 };
216
217 static void cli_posix_symlink_done(struct tevent_req *subreq);
218
cli_posix_symlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * link_target,const char * newname)219 struct tevent_req *cli_posix_symlink_send(TALLOC_CTX *mem_ctx,
220 struct tevent_context *ev,
221 struct cli_state *cli,
222 const char *link_target,
223 const char *newname)
224 {
225 struct tevent_req *req = NULL, *subreq = NULL;
226 struct cli_posix_symlink_state *state = NULL;
227
228 req = tevent_req_create(
229 mem_ctx, &state, struct cli_posix_symlink_state);
230 if (req == NULL) {
231 return NULL;
232 }
233
234 subreq = cli_posix_link_internal_send(
235 mem_ctx, ev, cli, SMB_SET_FILE_UNIX_LINK, link_target, newname);
236 if (tevent_req_nomem(subreq, req)) {
237 return tevent_req_post(req, ev);
238 }
239 tevent_req_set_callback(subreq, cli_posix_symlink_done, req);
240 return req;
241 }
242
cli_posix_symlink_done(struct tevent_req * subreq)243 static void cli_posix_symlink_done(struct tevent_req *subreq)
244 {
245 NTSTATUS status = cli_posix_link_internal_recv(subreq);
246 tevent_req_simple_finish_ntstatus(subreq, status);
247 }
248
cli_posix_symlink_recv(struct tevent_req * req)249 NTSTATUS cli_posix_symlink_recv(struct tevent_req *req)
250 {
251 return tevent_req_simple_recv_ntstatus(req);
252 }
253
cli_posix_symlink(struct cli_state * cli,const char * link_target,const char * newname)254 NTSTATUS cli_posix_symlink(struct cli_state *cli,
255 const char *link_target,
256 const char *newname)
257 {
258 TALLOC_CTX *frame = talloc_stackframe();
259 struct tevent_context *ev = NULL;
260 struct tevent_req *req = NULL;
261 NTSTATUS status = NT_STATUS_OK;
262
263 if (smbXcli_conn_has_async_calls(cli->conn)) {
264 /*
265 * Can't use sync call while an async call is in flight
266 */
267 status = NT_STATUS_INVALID_PARAMETER;
268 goto fail;
269 }
270
271 ev = samba_tevent_context_init(frame);
272 if (ev == NULL) {
273 status = NT_STATUS_NO_MEMORY;
274 goto fail;
275 }
276
277 req = cli_posix_symlink_send(frame,
278 ev,
279 cli,
280 link_target,
281 newname);
282 if (req == NULL) {
283 status = NT_STATUS_NO_MEMORY;
284 goto fail;
285 }
286
287 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
288 goto fail;
289 }
290
291 status = cli_posix_symlink_recv(req);
292
293 fail:
294 TALLOC_FREE(frame);
295 return status;
296 }
297
298 /****************************************************************************
299 Read a POSIX symlink.
300 ****************************************************************************/
301
302 struct cli_posix_readlink_state {
303 struct cli_state *cli;
304 char *converted;
305 };
306
307 static void cli_posix_readlink_done(struct tevent_req *subreq);
308
cli_posix_readlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname)309 struct tevent_req *cli_posix_readlink_send(TALLOC_CTX *mem_ctx,
310 struct tevent_context *ev,
311 struct cli_state *cli,
312 const char *fname)
313 {
314 struct tevent_req *req = NULL, *subreq = NULL;
315 struct cli_posix_readlink_state *state = NULL;
316
317 req = tevent_req_create(
318 mem_ctx, &state, struct cli_posix_readlink_state);
319 if (req == NULL) {
320 return NULL;
321 }
322 state->cli = cli;
323
324 subreq = cli_qpathinfo_send(
325 state,
326 ev,
327 cli,
328 fname,
329 SMB_QUERY_FILE_UNIX_LINK,
330 1,
331 UINT16_MAX);
332 if (tevent_req_nomem(subreq, req)) {
333 return tevent_req_post(req, ev);
334 }
335 tevent_req_set_callback(subreq, cli_posix_readlink_done, req);
336 return req;
337 }
338
cli_posix_readlink_done(struct tevent_req * subreq)339 static void cli_posix_readlink_done(struct tevent_req *subreq)
340 {
341 struct tevent_req *req = tevent_req_callback_data(
342 subreq, struct tevent_req);
343 struct cli_posix_readlink_state *state = tevent_req_data(
344 req, struct cli_posix_readlink_state);
345 NTSTATUS status;
346 uint8_t *data = NULL;
347 uint32_t num_data = 0;
348 charset_t charset;
349 size_t converted_size;
350 bool ok;
351
352 status = cli_qpathinfo_recv(subreq, state, &data, &num_data);
353 TALLOC_FREE(subreq);
354 if (tevent_req_nterror(req, status)) {
355 return;
356 }
357 /*
358 * num_data is > 1, we've given 1 as minimum to cli_qpathinfo_send
359 */
360 if (data == NULL || data[num_data-1] != '\0') {
361 tevent_req_nterror(req, NT_STATUS_DATA_ERROR);
362 return;
363 }
364
365 charset = smbXcli_conn_use_unicode(state->cli->conn) ?
366 CH_UTF16LE : CH_DOS;
367
368 /* The returned data is a pushed string, not raw data. */
369 ok = convert_string_talloc(
370 state,
371 charset,
372 CH_UNIX,
373 data,
374 num_data,
375 &state->converted,
376 &converted_size);
377 if (!ok) {
378 tevent_req_oom(req);
379 return;
380 }
381 tevent_req_done(req);
382 }
383
cli_posix_readlink_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx,char ** target)384 NTSTATUS cli_posix_readlink_recv(
385 struct tevent_req *req, TALLOC_CTX *mem_ctx, char **target)
386 {
387 struct cli_posix_readlink_state *state = tevent_req_data(
388 req, struct cli_posix_readlink_state);
389 NTSTATUS status;
390
391 if (tevent_req_is_nterror(req, &status)) {
392 return status;
393 }
394 *target = talloc_move(mem_ctx, &state->converted);
395 return NT_STATUS_OK;
396 }
397
cli_posix_readlink(struct cli_state * cli,const char * fname,TALLOC_CTX * mem_ctx,char ** target)398 NTSTATUS cli_posix_readlink(
399 struct cli_state *cli,
400 const char *fname,
401 TALLOC_CTX *mem_ctx,
402 char **target)
403 {
404 TALLOC_CTX *frame = talloc_stackframe();
405 struct tevent_context *ev = NULL;
406 struct tevent_req *req = NULL;
407 NTSTATUS status = NT_STATUS_OK;
408
409 if (smbXcli_conn_has_async_calls(cli->conn)) {
410 /*
411 * Can't use sync call while an async call is in flight
412 */
413 status = NT_STATUS_INVALID_PARAMETER;
414 goto fail;
415 }
416
417 ev = samba_tevent_context_init(frame);
418 if (ev == NULL) {
419 status = NT_STATUS_NO_MEMORY;
420 goto fail;
421 }
422
423 req = cli_posix_readlink_send(frame, ev, cli, fname);
424 if (req == NULL) {
425 status = NT_STATUS_NO_MEMORY;
426 goto fail;
427 }
428
429 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
430 goto fail;
431 }
432
433 status = cli_posix_readlink_recv(req, mem_ctx, target);
434
435 fail:
436 TALLOC_FREE(frame);
437 return status;
438 }
439
440 /****************************************************************************
441 Hard link a file (UNIX extensions).
442 ****************************************************************************/
443
444 struct cli_posix_hardlink_state {
445 uint8_t dummy;
446 };
447
448 static void cli_posix_hardlink_done(struct tevent_req *subreq);
449
cli_posix_hardlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * oldname,const char * newname)450 struct tevent_req *cli_posix_hardlink_send(TALLOC_CTX *mem_ctx,
451 struct tevent_context *ev,
452 struct cli_state *cli,
453 const char *oldname,
454 const char *newname)
455 {
456 struct tevent_req *req = NULL, *subreq = NULL;
457 struct cli_posix_hardlink_state *state = NULL;
458
459 req = tevent_req_create(
460 mem_ctx, &state, struct cli_posix_hardlink_state);
461 if (req == NULL) {
462 return NULL;
463 }
464
465 subreq = cli_posix_link_internal_send(
466 state, ev, cli, SMB_SET_FILE_UNIX_HLINK, oldname, newname);
467 if (tevent_req_nomem(subreq, req)) {
468 return tevent_req_post(req, ev);
469 }
470 tevent_req_set_callback(subreq, cli_posix_hardlink_done, req);
471 return req;
472 }
473
cli_posix_hardlink_done(struct tevent_req * subreq)474 static void cli_posix_hardlink_done(struct tevent_req *subreq)
475 {
476 NTSTATUS status = cli_posix_link_internal_recv(subreq);
477 tevent_req_simple_finish_ntstatus(subreq, status);
478 }
479
cli_posix_hardlink_recv(struct tevent_req * req)480 NTSTATUS cli_posix_hardlink_recv(struct tevent_req *req)
481 {
482 return tevent_req_simple_recv_ntstatus(req);
483 }
484
cli_posix_hardlink(struct cli_state * cli,const char * oldname,const char * newname)485 NTSTATUS cli_posix_hardlink(struct cli_state *cli,
486 const char *oldname,
487 const char *newname)
488 {
489 TALLOC_CTX *frame = talloc_stackframe();
490 struct tevent_context *ev = NULL;
491 struct tevent_req *req = NULL;
492 NTSTATUS status = NT_STATUS_OK;
493
494 if (smbXcli_conn_has_async_calls(cli->conn)) {
495 /*
496 * Can't use sync call while an async call is in flight
497 */
498 status = NT_STATUS_INVALID_PARAMETER;
499 goto fail;
500 }
501
502 ev = samba_tevent_context_init(frame);
503 if (ev == NULL) {
504 status = NT_STATUS_NO_MEMORY;
505 goto fail;
506 }
507
508 req = cli_posix_hardlink_send(frame,
509 ev,
510 cli,
511 oldname,
512 newname);
513 if (req == NULL) {
514 status = NT_STATUS_NO_MEMORY;
515 goto fail;
516 }
517
518 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
519 goto fail;
520 }
521
522 status = cli_posix_hardlink_recv(req);
523
524 fail:
525 TALLOC_FREE(frame);
526 return status;
527 }
528
529 /****************************************************************************
530 Do a POSIX getacl - pathname based ACL get (UNIX extensions).
531 ****************************************************************************/
532
533 struct getacl_state {
534 uint32_t num_data;
535 uint8_t *data;
536 };
537
538 static void cli_posix_getacl_done(struct tevent_req *subreq);
539
cli_posix_getacl_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname)540 struct tevent_req *cli_posix_getacl_send(TALLOC_CTX *mem_ctx,
541 struct tevent_context *ev,
542 struct cli_state *cli,
543 const char *fname)
544 {
545 struct tevent_req *req = NULL, *subreq = NULL;
546 struct getacl_state *state = NULL;
547
548 req = tevent_req_create(mem_ctx, &state, struct getacl_state);
549 if (req == NULL) {
550 return NULL;
551 }
552 subreq = cli_qpathinfo_send(state, ev, cli, fname, SMB_QUERY_POSIX_ACL,
553 0, CLI_BUFFER_SIZE);
554 if (tevent_req_nomem(subreq, req)) {
555 return tevent_req_post(req, ev);
556 }
557 tevent_req_set_callback(subreq, cli_posix_getacl_done, req);
558 return req;
559 }
560
cli_posix_getacl_done(struct tevent_req * subreq)561 static void cli_posix_getacl_done(struct tevent_req *subreq)
562 {
563 struct tevent_req *req = tevent_req_callback_data(
564 subreq, struct tevent_req);
565 struct getacl_state *state = tevent_req_data(
566 req, struct getacl_state);
567 NTSTATUS status;
568
569 status = cli_qpathinfo_recv(subreq, state, &state->data,
570 &state->num_data);
571 TALLOC_FREE(subreq);
572 if (tevent_req_nterror(req, status)) {
573 return;
574 }
575 tevent_req_done(req);
576 }
577
cli_posix_getacl_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx,size_t * prb_size,char ** retbuf)578 NTSTATUS cli_posix_getacl_recv(struct tevent_req *req,
579 TALLOC_CTX *mem_ctx,
580 size_t *prb_size,
581 char **retbuf)
582 {
583 struct getacl_state *state = tevent_req_data(req, struct getacl_state);
584 NTSTATUS status;
585
586 if (tevent_req_is_nterror(req, &status)) {
587 return status;
588 }
589 *prb_size = (size_t)state->num_data;
590 *retbuf = (char *)talloc_move(mem_ctx, &state->data);
591 return NT_STATUS_OK;
592 }
593
cli_posix_getacl(struct cli_state * cli,const char * fname,TALLOC_CTX * mem_ctx,size_t * prb_size,char ** retbuf)594 NTSTATUS cli_posix_getacl(struct cli_state *cli,
595 const char *fname,
596 TALLOC_CTX *mem_ctx,
597 size_t *prb_size,
598 char **retbuf)
599 {
600 TALLOC_CTX *frame = talloc_stackframe();
601 struct tevent_context *ev = NULL;
602 struct tevent_req *req = NULL;
603 NTSTATUS status = NT_STATUS_OK;
604
605 if (smbXcli_conn_has_async_calls(cli->conn)) {
606 /*
607 * Can't use sync call while an async call is in flight
608 */
609 status = NT_STATUS_INVALID_PARAMETER;
610 goto fail;
611 }
612
613 ev = samba_tevent_context_init(frame);
614 if (ev == NULL) {
615 status = NT_STATUS_NO_MEMORY;
616 goto fail;
617 }
618
619 req = cli_posix_getacl_send(frame,
620 ev,
621 cli,
622 fname);
623 if (req == NULL) {
624 status = NT_STATUS_NO_MEMORY;
625 goto fail;
626 }
627
628 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
629 goto fail;
630 }
631
632 status = cli_posix_getacl_recv(req, mem_ctx, prb_size, retbuf);
633
634 fail:
635 TALLOC_FREE(frame);
636 return status;
637 }
638
639 /****************************************************************************
640 Do a POSIX setacl - pathname based ACL set (UNIX extensions).
641 ****************************************************************************/
642
643 struct setacl_state {
644 uint8_t *data;
645 };
646
647 static void cli_posix_setacl_done(struct tevent_req *subreq);
648
cli_posix_setacl_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,const void * data,size_t num_data)649 struct tevent_req *cli_posix_setacl_send(TALLOC_CTX *mem_ctx,
650 struct tevent_context *ev,
651 struct cli_state *cli,
652 const char *fname,
653 const void *data,
654 size_t num_data)
655 {
656 struct tevent_req *req = NULL, *subreq = NULL;
657 struct setacl_state *state = NULL;
658
659 req = tevent_req_create(mem_ctx, &state, struct setacl_state);
660 if (req == NULL) {
661 return NULL;
662 }
663 state->data = talloc_memdup(state, data, num_data);
664 if (tevent_req_nomem(state->data, req)) {
665 return tevent_req_post(req, ev);
666 }
667
668 subreq = cli_setpathinfo_send(state,
669 ev,
670 cli,
671 SMB_SET_POSIX_ACL,
672 fname,
673 state->data,
674 num_data);
675 if (tevent_req_nomem(subreq, req)) {
676 return tevent_req_post(req, ev);
677 }
678 tevent_req_set_callback(subreq, cli_posix_setacl_done, req);
679 return req;
680 }
681
cli_posix_setacl_done(struct tevent_req * subreq)682 static void cli_posix_setacl_done(struct tevent_req *subreq)
683 {
684 NTSTATUS status = cli_setpathinfo_recv(subreq);
685 tevent_req_simple_finish_ntstatus(subreq, status);
686 }
687
cli_posix_setacl_recv(struct tevent_req * req)688 NTSTATUS cli_posix_setacl_recv(struct tevent_req *req)
689 {
690 return tevent_req_simple_recv_ntstatus(req);
691 }
692
cli_posix_setacl(struct cli_state * cli,const char * fname,const void * acl_buf,size_t acl_buf_size)693 NTSTATUS cli_posix_setacl(struct cli_state *cli,
694 const char *fname,
695 const void *acl_buf,
696 size_t acl_buf_size)
697 {
698 TALLOC_CTX *frame = talloc_stackframe();
699 struct tevent_context *ev = NULL;
700 struct tevent_req *req = NULL;
701 NTSTATUS status = NT_STATUS_OK;
702
703 if (smbXcli_conn_has_async_calls(cli->conn)) {
704 /*
705 * Can't use sync call while an async call is in flight
706 */
707 status = NT_STATUS_INVALID_PARAMETER;
708 goto fail;
709 }
710
711 ev = samba_tevent_context_init(frame);
712 if (ev == NULL) {
713 status = NT_STATUS_NO_MEMORY;
714 goto fail;
715 }
716
717 req = cli_posix_setacl_send(frame,
718 ev,
719 cli,
720 fname,
721 acl_buf,
722 acl_buf_size);
723 if (req == NULL) {
724 status = NT_STATUS_NO_MEMORY;
725 goto fail;
726 }
727
728 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
729 goto fail;
730 }
731
732 status = cli_posix_setacl_recv(req);
733
734 fail:
735 TALLOC_FREE(frame);
736 return status;
737 }
738
739 /****************************************************************************
740 Stat a file (UNIX extensions).
741 ****************************************************************************/
742
743 struct stat_state {
744 SMB_STRUCT_STAT *sbuf;
745 };
746
747 static void cli_posix_stat_done(struct tevent_req *subreq);
748
cli_posix_stat_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,SMB_STRUCT_STAT * sbuf)749 struct tevent_req *cli_posix_stat_send(TALLOC_CTX *mem_ctx,
750 struct tevent_context *ev,
751 struct cli_state *cli,
752 const char *fname,
753 SMB_STRUCT_STAT *sbuf)
754 {
755 struct tevent_req *req = NULL, *subreq = NULL;
756 struct stat_state *state = NULL;
757
758 req = tevent_req_create(mem_ctx, &state, struct stat_state);
759 if (req == NULL) {
760 return NULL;
761 }
762 state->sbuf = sbuf;
763
764 subreq = cli_qpathinfo_send(state, ev, cli, fname,
765 SMB_QUERY_FILE_UNIX_BASIC, 100, 100);
766 if (tevent_req_nomem(subreq, req)) {
767 return tevent_req_post(req, ev);
768 }
769 tevent_req_set_callback(subreq, cli_posix_stat_done, req);
770 return req;
771 }
772
cli_posix_stat_done(struct tevent_req * subreq)773 static void cli_posix_stat_done(struct tevent_req *subreq)
774 {
775 struct tevent_req *req = tevent_req_callback_data(
776 subreq, struct tevent_req);
777 struct stat_state *state = tevent_req_data(req, struct stat_state);
778 SMB_STRUCT_STAT *sbuf = state->sbuf;
779 uint8_t *data;
780 uint32_t num_data = 0;
781 NTSTATUS status;
782
783 status = cli_qpathinfo_recv(subreq, state, &data, &num_data);
784 TALLOC_FREE(subreq);
785 if (tevent_req_nterror(req, status)) {
786 return;
787 }
788
789 if (num_data != 100) {
790 /*
791 * Paranoia, cli_qpathinfo should have guaranteed
792 * this, but you never know...
793 */
794 tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);
795 return;
796 }
797
798 *sbuf = (SMB_STRUCT_STAT) { 0 };
799
800 /* total size, in bytes */
801 sbuf->st_ex_size = IVAL2_TO_SMB_BIG_UINT(data, 0);
802
803 /* number of blocks allocated */
804 sbuf->st_ex_blocks = IVAL2_TO_SMB_BIG_UINT(data,8);
805 #if defined (HAVE_STAT_ST_BLOCKS) && defined(STAT_ST_BLOCKSIZE)
806 sbuf->st_ex_blocks /= STAT_ST_BLOCKSIZE;
807 #else
808 /* assume 512 byte blocks */
809 sbuf->st_ex_blocks /= 512;
810 #endif
811 /* time of last change */
812 sbuf->st_ex_ctime = interpret_long_date((char *)(data + 16));
813
814 /* time of last access */
815 sbuf->st_ex_atime = interpret_long_date((char *)(data + 24));
816
817 /* time of last modification */
818 sbuf->st_ex_mtime = interpret_long_date((char *)(data + 32));
819
820 sbuf->st_ex_uid = (uid_t) IVAL(data, 40); /* user ID of owner */
821 sbuf->st_ex_gid = (gid_t) IVAL(data, 48); /* group ID of owner */
822 sbuf->st_ex_mode = unix_filetype_from_wire(IVAL(data, 56));
823
824 #if defined(HAVE_MAKEDEV)
825 {
826 uint32_t dev_major = IVAL(data,60);
827 uint32_t dev_minor = IVAL(data,68);
828 sbuf->st_ex_rdev = makedev(dev_major, dev_minor);
829 }
830 #endif
831 /* inode */
832 sbuf->st_ex_ino = (SMB_INO_T)IVAL2_TO_SMB_BIG_UINT(data, 76);
833
834 /* protection */
835 sbuf->st_ex_mode |= wire_perms_to_unix(IVAL(data, 84));
836
837 /* number of hard links */
838 sbuf->st_ex_nlink = BIG_UINT(data, 92);
839
840 tevent_req_done(req);
841 }
842
cli_posix_stat_recv(struct tevent_req * req)843 NTSTATUS cli_posix_stat_recv(struct tevent_req *req)
844 {
845 return tevent_req_simple_recv_ntstatus(req);
846 }
847
cli_posix_stat(struct cli_state * cli,const char * fname,SMB_STRUCT_STAT * sbuf)848 NTSTATUS cli_posix_stat(struct cli_state *cli,
849 const char *fname,
850 SMB_STRUCT_STAT *sbuf)
851 {
852 TALLOC_CTX *frame = talloc_stackframe();
853 struct tevent_context *ev = NULL;
854 struct tevent_req *req = NULL;
855 NTSTATUS status = NT_STATUS_OK;
856
857 if (smbXcli_conn_has_async_calls(cli->conn)) {
858 /*
859 * Can't use sync call while an async call is in flight
860 */
861 status = NT_STATUS_INVALID_PARAMETER;
862 goto fail;
863 }
864
865 ev = samba_tevent_context_init(frame);
866 if (ev == NULL) {
867 status = NT_STATUS_NO_MEMORY;
868 goto fail;
869 }
870
871 req = cli_posix_stat_send(frame, ev, cli, fname, sbuf);
872 if (req == NULL) {
873 status = NT_STATUS_NO_MEMORY;
874 goto fail;
875 }
876
877 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
878 goto fail;
879 }
880
881 status = cli_posix_stat_recv(req);
882
883 fail:
884 TALLOC_FREE(frame);
885 return status;
886 }
887
888 /****************************************************************************
889 Chmod or chown a file internal (UNIX extensions).
890 ****************************************************************************/
891
892 struct cli_posix_chown_chmod_internal_state {
893 uint8_t data[100];
894 };
895
896 static void cli_posix_chown_chmod_internal_done(struct tevent_req *subreq);
897
cli_posix_chown_chmod_internal_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint32_t mode,uint32_t uid,uint32_t gid)898 static struct tevent_req *cli_posix_chown_chmod_internal_send(TALLOC_CTX *mem_ctx,
899 struct tevent_context *ev,
900 struct cli_state *cli,
901 const char *fname,
902 uint32_t mode,
903 uint32_t uid,
904 uint32_t gid)
905 {
906 struct tevent_req *req = NULL, *subreq = NULL;
907 struct cli_posix_chown_chmod_internal_state *state = NULL;
908
909 req = tevent_req_create(mem_ctx, &state,
910 struct cli_posix_chown_chmod_internal_state);
911 if (req == NULL) {
912 return NULL;
913 }
914
915 memset(state->data, 0xff, 40); /* Set all sizes/times to no change. */
916 memset(&state->data[40], '\0', 60);
917 SIVAL(state->data,40,uid);
918 SIVAL(state->data,48,gid);
919 SIVAL(state->data,84,mode);
920
921 subreq = cli_setpathinfo_send(state, ev, cli, SMB_SET_FILE_UNIX_BASIC,
922 fname, state->data, sizeof(state->data));
923 if (tevent_req_nomem(subreq, req)) {
924 return tevent_req_post(req, ev);
925 }
926 tevent_req_set_callback(subreq, cli_posix_chown_chmod_internal_done,
927 req);
928 return req;
929 }
930
cli_posix_chown_chmod_internal_done(struct tevent_req * subreq)931 static void cli_posix_chown_chmod_internal_done(struct tevent_req *subreq)
932 {
933 NTSTATUS status = cli_setpathinfo_recv(subreq);
934 tevent_req_simple_finish_ntstatus(subreq, status);
935 }
936
cli_posix_chown_chmod_internal_recv(struct tevent_req * req)937 static NTSTATUS cli_posix_chown_chmod_internal_recv(struct tevent_req *req)
938 {
939 return tevent_req_simple_recv_ntstatus(req);
940 }
941
942 /****************************************************************************
943 chmod a file (UNIX extensions).
944 ****************************************************************************/
945
946 struct cli_posix_chmod_state {
947 uint8_t dummy;
948 };
949
950 static void cli_posix_chmod_done(struct tevent_req *subreq);
951
cli_posix_chmod_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,mode_t mode)952 struct tevent_req *cli_posix_chmod_send(TALLOC_CTX *mem_ctx,
953 struct tevent_context *ev,
954 struct cli_state *cli,
955 const char *fname,
956 mode_t mode)
957 {
958 struct tevent_req *req = NULL, *subreq = NULL;
959 struct cli_posix_chmod_state *state = NULL;
960
961 req = tevent_req_create(mem_ctx, &state, struct cli_posix_chmod_state);
962 if (req == NULL) {
963 return NULL;
964 }
965
966 subreq = cli_posix_chown_chmod_internal_send(
967 state,
968 ev,
969 cli,
970 fname,
971 unix_perms_to_wire(mode),
972 SMB_UID_NO_CHANGE,
973 SMB_GID_NO_CHANGE);
974 if (tevent_req_nomem(subreq, req)) {
975 return tevent_req_post(req, ev);
976 }
977 tevent_req_set_callback(subreq, cli_posix_chmod_done, req);
978 return req;
979 }
980
cli_posix_chmod_done(struct tevent_req * subreq)981 static void cli_posix_chmod_done(struct tevent_req *subreq)
982 {
983 NTSTATUS status = cli_posix_chown_chmod_internal_recv(subreq);
984 tevent_req_simple_finish_ntstatus(subreq, status);
985 }
986
cli_posix_chmod_recv(struct tevent_req * req)987 NTSTATUS cli_posix_chmod_recv(struct tevent_req *req)
988 {
989 return tevent_req_simple_recv_ntstatus(req);
990 }
991
cli_posix_chmod(struct cli_state * cli,const char * fname,mode_t mode)992 NTSTATUS cli_posix_chmod(struct cli_state *cli, const char *fname, mode_t mode)
993 {
994 TALLOC_CTX *frame = talloc_stackframe();
995 struct tevent_context *ev = NULL;
996 struct tevent_req *req = NULL;
997 NTSTATUS status = NT_STATUS_OK;
998
999 if (smbXcli_conn_has_async_calls(cli->conn)) {
1000 /*
1001 * Can't use sync call while an async call is in flight
1002 */
1003 status = NT_STATUS_INVALID_PARAMETER;
1004 goto fail;
1005 }
1006
1007 ev = samba_tevent_context_init(frame);
1008 if (ev == NULL) {
1009 status = NT_STATUS_NO_MEMORY;
1010 goto fail;
1011 }
1012
1013 req = cli_posix_chmod_send(frame,
1014 ev,
1015 cli,
1016 fname,
1017 mode);
1018 if (req == NULL) {
1019 status = NT_STATUS_NO_MEMORY;
1020 goto fail;
1021 }
1022
1023 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1024 goto fail;
1025 }
1026
1027 status = cli_posix_chmod_recv(req);
1028
1029 fail:
1030 TALLOC_FREE(frame);
1031 return status;
1032 }
1033
1034 /****************************************************************************
1035 chown a file (UNIX extensions).
1036 ****************************************************************************/
1037
1038 struct cli_posix_chown_state {
1039 uint8_t dummy;
1040 };
1041
1042 static void cli_posix_chown_done(struct tevent_req *subreq);
1043
cli_posix_chown_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uid_t uid,gid_t gid)1044 struct tevent_req *cli_posix_chown_send(TALLOC_CTX *mem_ctx,
1045 struct tevent_context *ev,
1046 struct cli_state *cli,
1047 const char *fname,
1048 uid_t uid,
1049 gid_t gid)
1050 {
1051 struct tevent_req *req = NULL, *subreq = NULL;
1052 struct cli_posix_chown_state *state = NULL;
1053
1054 req = tevent_req_create(
1055 mem_ctx, &state, struct cli_posix_chown_state);
1056 if (req == NULL) {
1057 return NULL;
1058 }
1059
1060 subreq = cli_posix_chown_chmod_internal_send(
1061 state,
1062 ev,
1063 cli,
1064 fname,
1065 SMB_MODE_NO_CHANGE,
1066 (uint32_t)uid,
1067 (uint32_t)gid);
1068 if (tevent_req_nomem(subreq, req)) {
1069 return tevent_req_post(req, ev);
1070 }
1071 tevent_req_set_callback(subreq, cli_posix_chown_done, req);
1072 return req;
1073 }
1074
cli_posix_chown_done(struct tevent_req * subreq)1075 static void cli_posix_chown_done(struct tevent_req *subreq)
1076 {
1077 NTSTATUS status = cli_posix_chown_chmod_internal_recv(subreq);
1078 tevent_req_simple_finish_ntstatus(subreq, status);
1079 }
1080
cli_posix_chown_recv(struct tevent_req * req)1081 NTSTATUS cli_posix_chown_recv(struct tevent_req *req)
1082 {
1083 return tevent_req_simple_recv_ntstatus(req);
1084 }
1085
cli_posix_chown(struct cli_state * cli,const char * fname,uid_t uid,gid_t gid)1086 NTSTATUS cli_posix_chown(struct cli_state *cli,
1087 const char *fname,
1088 uid_t uid,
1089 gid_t gid)
1090 {
1091 TALLOC_CTX *frame = talloc_stackframe();
1092 struct tevent_context *ev = NULL;
1093 struct tevent_req *req = NULL;
1094 NTSTATUS status = NT_STATUS_OK;
1095
1096 if (smbXcli_conn_has_async_calls(cli->conn)) {
1097 /*
1098 * Can't use sync call while an async call is in flight
1099 */
1100 status = NT_STATUS_INVALID_PARAMETER;
1101 goto fail;
1102 }
1103
1104 ev = samba_tevent_context_init(frame);
1105 if (ev == NULL) {
1106 status = NT_STATUS_NO_MEMORY;
1107 goto fail;
1108 }
1109
1110 req = cli_posix_chown_send(frame,
1111 ev,
1112 cli,
1113 fname,
1114 uid,
1115 gid);
1116 if (req == NULL) {
1117 status = NT_STATUS_NO_MEMORY;
1118 goto fail;
1119 }
1120
1121 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1122 goto fail;
1123 }
1124
1125 status = cli_posix_chown_recv(req);
1126
1127 fail:
1128 TALLOC_FREE(frame);
1129 return status;
1130 }
1131
1132 /****************************************************************************
1133 Rename a file.
1134 ****************************************************************************/
1135
1136 static struct tevent_req *cli_cifs_rename_send(TALLOC_CTX *mem_ctx,
1137 struct tevent_context *ev,
1138 struct cli_state *cli,
1139 const char *fname_src,
1140 const char *fname_dst,
1141 bool replace);
1142
1143 static struct tevent_req *cli_smb1_rename_send(TALLOC_CTX *mem_ctx,
1144 struct tevent_context *ev,
1145 struct cli_state *cli,
1146 const char *fname_src,
1147 const char *fname_dst,
1148 bool replace);
1149
cli_rename_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst,bool replace)1150 struct tevent_req *cli_rename_send(TALLOC_CTX *mem_ctx,
1151 struct tevent_context *ev,
1152 struct cli_state *cli,
1153 const char *fname_src,
1154 const char *fname_dst,
1155 bool replace)
1156 {
1157 if (replace && smbXcli_conn_support_passthrough(cli->conn)) {
1158 return cli_smb1_rename_send(mem_ctx, ev, cli, fname_src,
1159 fname_dst, replace);
1160 } else {
1161 return cli_cifs_rename_send(mem_ctx, ev, cli, fname_src,
1162 fname_dst, replace);
1163 }
1164 }
1165
1166 struct cli_smb1_rename_state {
1167 uint8_t *data;
1168 };
1169
1170 static void cli_smb1_rename_done(struct tevent_req *subreq);
1171
cli_smb1_rename_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst,bool replace)1172 static struct tevent_req *cli_smb1_rename_send(TALLOC_CTX *mem_ctx,
1173 struct tevent_context *ev,
1174 struct cli_state *cli,
1175 const char *fname_src,
1176 const char *fname_dst,
1177 bool replace)
1178 {
1179 NTSTATUS status;
1180 struct tevent_req *req = NULL, *subreq = NULL;
1181 struct cli_smb1_rename_state *state = NULL;
1182 smb_ucs2_t *converted_str = NULL;
1183 size_t converted_size_bytes = 0;
1184
1185 req = tevent_req_create(mem_ctx, &state, struct cli_smb1_rename_state);
1186 if (req == NULL) {
1187 return NULL;
1188 }
1189
1190 if (!push_ucs2_talloc(talloc_tos(), &converted_str, fname_dst,
1191 &converted_size_bytes)) {
1192 status = NT_STATUS_INVALID_PARAMETER;
1193 goto fail;
1194 }
1195
1196 /* W2K8 insists the dest name is not null
1197 terminated. Remove the last 2 zero bytes
1198 and reduce the name length. */
1199
1200 if (converted_size_bytes < 2) {
1201 status = NT_STATUS_INVALID_PARAMETER;
1202 goto fail;
1203 }
1204 converted_size_bytes -= 2;
1205
1206 state->data =
1207 talloc_zero_array(state, uint8_t, 12 + converted_size_bytes);
1208 if (state->data == NULL) {
1209 status = NT_STATUS_NO_MEMORY;
1210 goto fail;
1211 }
1212
1213 if (replace) {
1214 SCVAL(state->data, 0, 1);
1215 }
1216
1217 SIVAL(state->data, 8, converted_size_bytes);
1218 memcpy(state->data + 12, converted_str, converted_size_bytes);
1219
1220 TALLOC_FREE(converted_str);
1221
1222 subreq = cli_setpathinfo_send(
1223 state, ev, cli, SMB_FILE_RENAME_INFORMATION, fname_src, state->data,
1224 talloc_get_size(state->data));
1225 if (tevent_req_nomem(subreq, req)) {
1226 status = NT_STATUS_NO_MEMORY;
1227 goto fail;
1228 }
1229 tevent_req_set_callback(subreq, cli_smb1_rename_done, req);
1230 return req;
1231
1232 fail:
1233 TALLOC_FREE(converted_str);
1234 tevent_req_nterror(req, status);
1235 return tevent_req_post(req, ev);
1236 }
1237
cli_smb1_rename_done(struct tevent_req * subreq)1238 static void cli_smb1_rename_done(struct tevent_req *subreq)
1239 {
1240 NTSTATUS status = cli_setpathinfo_recv(subreq);
1241 tevent_req_simple_finish_ntstatus(subreq, status);
1242 }
1243
1244 static void cli_cifs_rename_done(struct tevent_req *subreq);
1245
1246 struct cli_cifs_rename_state {
1247 uint16_t vwv[1];
1248 };
1249
cli_cifs_rename_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst,bool replace)1250 static struct tevent_req *cli_cifs_rename_send(TALLOC_CTX *mem_ctx,
1251 struct tevent_context *ev,
1252 struct cli_state *cli,
1253 const char *fname_src,
1254 const char *fname_dst,
1255 bool replace)
1256 {
1257 struct tevent_req *req = NULL, *subreq = NULL;
1258 struct cli_cifs_rename_state *state = NULL;
1259 uint8_t additional_flags = 0;
1260 uint16_t additional_flags2 = 0;
1261 uint8_t *bytes = NULL;
1262
1263 req = tevent_req_create(mem_ctx, &state, struct cli_cifs_rename_state);
1264 if (req == NULL) {
1265 return NULL;
1266 }
1267
1268 if (replace) {
1269 /*
1270 * CIFS doesn't support replace
1271 */
1272 tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
1273 return tevent_req_post(req, ev);
1274 }
1275
1276 SSVAL(state->vwv+0, 0, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY);
1277
1278 bytes = talloc_array(state, uint8_t, 1);
1279 if (tevent_req_nomem(bytes, req)) {
1280 return tevent_req_post(req, ev);
1281 }
1282 bytes[0] = 4;
1283 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname_src,
1284 strlen(fname_src)+1, NULL);
1285 if (tevent_req_nomem(bytes, req)) {
1286 return tevent_req_post(req, ev);
1287 }
1288
1289 if (clistr_is_previous_version_path(fname_src, NULL, NULL, NULL)) {
1290 additional_flags2 = FLAGS2_REPARSE_PATH;
1291 }
1292
1293 bytes = talloc_realloc(state, bytes, uint8_t,
1294 talloc_get_size(bytes)+1);
1295 if (tevent_req_nomem(bytes, req)) {
1296 return tevent_req_post(req, ev);
1297 }
1298
1299 bytes[talloc_get_size(bytes)-1] = 4;
1300 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname_dst,
1301 strlen(fname_dst)+1, NULL);
1302 if (tevent_req_nomem(bytes, req)) {
1303 return tevent_req_post(req, ev);
1304 }
1305
1306 subreq = cli_smb_send(state, ev, cli, SMBmv, additional_flags,
1307 additional_flags2,
1308 1, state->vwv, talloc_get_size(bytes), bytes);
1309 if (tevent_req_nomem(subreq, req)) {
1310 return tevent_req_post(req, ev);
1311 }
1312 tevent_req_set_callback(subreq, cli_cifs_rename_done, req);
1313 return req;
1314 }
1315
cli_cifs_rename_done(struct tevent_req * subreq)1316 static void cli_cifs_rename_done(struct tevent_req *subreq)
1317 {
1318 struct tevent_req *req = tevent_req_callback_data(
1319 subreq, struct tevent_req);
1320 NTSTATUS status;
1321
1322 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
1323 TALLOC_FREE(subreq);
1324 if (tevent_req_nterror(req, status)) {
1325 return;
1326 }
1327 tevent_req_done(req);
1328 }
1329
cli_rename_recv(struct tevent_req * req)1330 NTSTATUS cli_rename_recv(struct tevent_req *req)
1331 {
1332 return tevent_req_simple_recv_ntstatus(req);
1333 }
1334
cli_rename(struct cli_state * cli,const char * fname_src,const char * fname_dst,bool replace)1335 NTSTATUS cli_rename(struct cli_state *cli,
1336 const char *fname_src,
1337 const char *fname_dst,
1338 bool replace)
1339 {
1340 TALLOC_CTX *frame = NULL;
1341 struct tevent_context *ev;
1342 struct tevent_req *req;
1343 NTSTATUS status = NT_STATUS_OK;
1344
1345 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
1346 return cli_smb2_rename(cli, fname_src, fname_dst, replace);
1347 }
1348
1349 frame = talloc_stackframe();
1350
1351 if (smbXcli_conn_has_async_calls(cli->conn)) {
1352 /*
1353 * Can't use sync call while an async call is in flight
1354 */
1355 status = NT_STATUS_INVALID_PARAMETER;
1356 goto fail;
1357 }
1358
1359 ev = samba_tevent_context_init(frame);
1360 if (ev == NULL) {
1361 status = NT_STATUS_NO_MEMORY;
1362 goto fail;
1363 }
1364
1365 req = cli_rename_send(frame, ev, cli, fname_src, fname_dst, replace);
1366 if (req == NULL) {
1367 status = NT_STATUS_NO_MEMORY;
1368 goto fail;
1369 }
1370
1371 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1372 goto fail;
1373 }
1374
1375 status = cli_rename_recv(req);
1376
1377 fail:
1378 TALLOC_FREE(frame);
1379 return status;
1380 }
1381
1382 /****************************************************************************
1383 NT Rename a file.
1384 ****************************************************************************/
1385
1386 static void cli_ntrename_internal_done(struct tevent_req *subreq);
1387
1388 struct cli_ntrename_internal_state {
1389 uint16_t vwv[4];
1390 };
1391
cli_ntrename_internal_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst,uint16_t rename_flag)1392 static struct tevent_req *cli_ntrename_internal_send(TALLOC_CTX *mem_ctx,
1393 struct tevent_context *ev,
1394 struct cli_state *cli,
1395 const char *fname_src,
1396 const char *fname_dst,
1397 uint16_t rename_flag)
1398 {
1399 struct tevent_req *req = NULL, *subreq = NULL;
1400 struct cli_ntrename_internal_state *state = NULL;
1401 uint8_t additional_flags = 0;
1402 uint16_t additional_flags2 = 0;
1403 uint8_t *bytes = NULL;
1404
1405 req = tevent_req_create(mem_ctx, &state,
1406 struct cli_ntrename_internal_state);
1407 if (req == NULL) {
1408 return NULL;
1409 }
1410
1411 SSVAL(state->vwv+0, 0 ,FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY);
1412 SSVAL(state->vwv+1, 0, rename_flag);
1413
1414 bytes = talloc_array(state, uint8_t, 1);
1415 if (tevent_req_nomem(bytes, req)) {
1416 return tevent_req_post(req, ev);
1417 }
1418 bytes[0] = 4;
1419 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname_src,
1420 strlen(fname_src)+1, NULL);
1421 if (tevent_req_nomem(bytes, req)) {
1422 return tevent_req_post(req, ev);
1423 }
1424
1425 if (clistr_is_previous_version_path(fname_src, NULL, NULL, NULL)) {
1426 additional_flags2 = FLAGS2_REPARSE_PATH;
1427 }
1428
1429 bytes = talloc_realloc(state, bytes, uint8_t,
1430 talloc_get_size(bytes)+1);
1431 if (tevent_req_nomem(bytes, req)) {
1432 return tevent_req_post(req, ev);
1433 }
1434
1435 bytes[talloc_get_size(bytes)-1] = 4;
1436 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname_dst,
1437 strlen(fname_dst)+1, NULL);
1438 if (tevent_req_nomem(bytes, req)) {
1439 return tevent_req_post(req, ev);
1440 }
1441
1442 subreq = cli_smb_send(state, ev, cli, SMBntrename, additional_flags,
1443 additional_flags2,
1444 4, state->vwv, talloc_get_size(bytes), bytes);
1445 if (tevent_req_nomem(subreq, req)) {
1446 return tevent_req_post(req, ev);
1447 }
1448 tevent_req_set_callback(subreq, cli_ntrename_internal_done, req);
1449 return req;
1450 }
1451
cli_ntrename_internal_done(struct tevent_req * subreq)1452 static void cli_ntrename_internal_done(struct tevent_req *subreq)
1453 {
1454 NTSTATUS status = cli_smb_recv(
1455 subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
1456 tevent_req_simple_finish_ntstatus(subreq, status);
1457 }
1458
cli_ntrename_internal_recv(struct tevent_req * req)1459 static NTSTATUS cli_ntrename_internal_recv(struct tevent_req *req)
1460 {
1461 return tevent_req_simple_recv_ntstatus(req);
1462 }
1463
cli_ntrename_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst)1464 struct tevent_req *cli_ntrename_send(TALLOC_CTX *mem_ctx,
1465 struct tevent_context *ev,
1466 struct cli_state *cli,
1467 const char *fname_src,
1468 const char *fname_dst)
1469 {
1470 return cli_ntrename_internal_send(mem_ctx,
1471 ev,
1472 cli,
1473 fname_src,
1474 fname_dst,
1475 RENAME_FLAG_RENAME);
1476 }
1477
cli_ntrename_recv(struct tevent_req * req)1478 NTSTATUS cli_ntrename_recv(struct tevent_req *req)
1479 {
1480 return cli_ntrename_internal_recv(req);
1481 }
1482
cli_ntrename(struct cli_state * cli,const char * fname_src,const char * fname_dst)1483 NTSTATUS cli_ntrename(struct cli_state *cli, const char *fname_src, const char *fname_dst)
1484 {
1485 TALLOC_CTX *frame = talloc_stackframe();
1486 struct tevent_context *ev;
1487 struct tevent_req *req;
1488 NTSTATUS status = NT_STATUS_OK;
1489
1490 if (smbXcli_conn_has_async_calls(cli->conn)) {
1491 /*
1492 * Can't use sync call while an async call is in flight
1493 */
1494 status = NT_STATUS_INVALID_PARAMETER;
1495 goto fail;
1496 }
1497
1498 ev = samba_tevent_context_init(frame);
1499 if (ev == NULL) {
1500 status = NT_STATUS_NO_MEMORY;
1501 goto fail;
1502 }
1503
1504 req = cli_ntrename_send(frame, ev, cli, fname_src, fname_dst);
1505 if (req == NULL) {
1506 status = NT_STATUS_NO_MEMORY;
1507 goto fail;
1508 }
1509
1510 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1511 goto fail;
1512 }
1513
1514 status = cli_ntrename_recv(req);
1515
1516 fail:
1517 TALLOC_FREE(frame);
1518 return status;
1519 }
1520
1521 /****************************************************************************
1522 NT hardlink a file.
1523 ****************************************************************************/
1524
cli_nt_hardlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst)1525 static struct tevent_req *cli_nt_hardlink_send(TALLOC_CTX *mem_ctx,
1526 struct tevent_context *ev,
1527 struct cli_state *cli,
1528 const char *fname_src,
1529 const char *fname_dst)
1530 {
1531 return cli_ntrename_internal_send(mem_ctx,
1532 ev,
1533 cli,
1534 fname_src,
1535 fname_dst,
1536 RENAME_FLAG_HARD_LINK);
1537 }
1538
cli_nt_hardlink_recv(struct tevent_req * req)1539 static NTSTATUS cli_nt_hardlink_recv(struct tevent_req *req)
1540 {
1541 return cli_ntrename_internal_recv(req);
1542 }
1543
1544 struct cli_smb2_hardlink_state {
1545 struct tevent_context *ev;
1546 struct cli_state *cli;
1547 uint16_t fnum_src;
1548 const char *fname_dst;
1549 bool overwrite;
1550 NTSTATUS status;
1551 };
1552
1553 static void cli_smb2_hardlink_opened(struct tevent_req *subreq);
1554 static void cli_smb2_hardlink_info_set(struct tevent_req *subreq);
1555 static void cli_smb2_hardlink_closed(struct tevent_req *subreq);
1556
cli_smb2_hardlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst,bool overwrite,struct smb2_create_blobs * in_cblobs)1557 static struct tevent_req *cli_smb2_hardlink_send(
1558 TALLOC_CTX *mem_ctx,
1559 struct tevent_context *ev,
1560 struct cli_state *cli,
1561 const char *fname_src,
1562 const char *fname_dst,
1563 bool overwrite,
1564 struct smb2_create_blobs *in_cblobs)
1565 {
1566 struct tevent_req *req = NULL, *subreq = NULL;
1567 struct cli_smb2_hardlink_state *state = NULL;
1568
1569 req = tevent_req_create(
1570 mem_ctx, &state, struct cli_smb2_hardlink_state);
1571 if (req == NULL) {
1572 return NULL;
1573 }
1574 state->ev = ev;
1575 state->cli = cli;
1576 state->fname_dst = fname_dst;
1577 state->overwrite = overwrite;
1578
1579 subreq = cli_smb2_create_fnum_send(
1580 state,
1581 ev,
1582 cli,
1583 fname_src,
1584 0, /* create_flags */
1585 SMB2_IMPERSONATION_IMPERSONATION,
1586 FILE_WRITE_ATTRIBUTES,
1587 0, /* file attributes */
1588 FILE_SHARE_READ|
1589 FILE_SHARE_WRITE|
1590 FILE_SHARE_DELETE, /* share_access */
1591 FILE_OPEN, /* create_disposition */
1592 FILE_NON_DIRECTORY_FILE, /* no hardlinks on directories */
1593 in_cblobs);
1594 if (tevent_req_nomem(subreq, req)) {
1595 return tevent_req_post(req, ev);
1596 }
1597 tevent_req_set_callback(subreq, cli_smb2_hardlink_opened, req);
1598 return req;
1599 }
1600
cli_smb2_hardlink_opened(struct tevent_req * subreq)1601 static void cli_smb2_hardlink_opened(struct tevent_req *subreq)
1602 {
1603 struct tevent_req *req = tevent_req_callback_data(
1604 subreq, struct tevent_req);
1605 struct cli_smb2_hardlink_state *state = tevent_req_data(
1606 req, struct cli_smb2_hardlink_state);
1607 NTSTATUS status;
1608 smb_ucs2_t *ucs2_dst;
1609 size_t ucs2_len;
1610 DATA_BLOB inbuf;
1611 bool ok;
1612
1613 status = cli_smb2_create_fnum_recv(
1614 subreq, &state->fnum_src, NULL, NULL, NULL);
1615 TALLOC_FREE(subreq);
1616 if (tevent_req_nterror(req, status)) {
1617 return;
1618 }
1619
1620 ok = push_ucs2_talloc(state, &ucs2_dst, state->fname_dst, &ucs2_len);
1621 if (!ok || (ucs2_len < 2)) {
1622 tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
1623 return;
1624 }
1625 /* Don't 0-terminate the name */
1626 ucs2_len -= 2;
1627
1628 inbuf = data_blob_talloc_zero(state, ucs2_len + 20);
1629 if (tevent_req_nomem(inbuf.data, req)) {
1630 return;
1631 }
1632
1633 if (state->overwrite) {
1634 SCVAL(inbuf.data, 0, 1);
1635 }
1636 SIVAL(inbuf.data, 16, ucs2_len);
1637 memcpy(inbuf.data + 20, ucs2_dst, ucs2_len);
1638 TALLOC_FREE(ucs2_dst);
1639
1640 subreq = cli_smb2_set_info_fnum_send(
1641 state,
1642 state->ev,
1643 state->cli,
1644 state->fnum_src,
1645 1, /* in_info_type */
1646 SMB_FILE_LINK_INFORMATION - 1000, /* in_file_info_class */
1647 &inbuf,
1648 0); /* in_additional_info */
1649 if (tevent_req_nomem(subreq, req)) {
1650 return;
1651 }
1652 tevent_req_set_callback(subreq, cli_smb2_hardlink_info_set, req);
1653 }
1654
cli_smb2_hardlink_info_set(struct tevent_req * subreq)1655 static void cli_smb2_hardlink_info_set(struct tevent_req *subreq)
1656 {
1657 struct tevent_req *req = tevent_req_callback_data(
1658 subreq, struct tevent_req);
1659 struct cli_smb2_hardlink_state *state = tevent_req_data(
1660 req, struct cli_smb2_hardlink_state);
1661
1662 state->status = cli_smb2_set_info_fnum_recv(subreq);
1663 TALLOC_FREE(subreq);
1664
1665 /* ignore error here, we need to close the file */
1666
1667 subreq = cli_smb2_close_fnum_send(
1668 state, state->ev, state->cli, state->fnum_src);
1669 if (tevent_req_nomem(subreq, req)) {
1670 return;
1671 }
1672 tevent_req_set_callback(subreq, cli_smb2_hardlink_closed, req);
1673 }
1674
cli_smb2_hardlink_closed(struct tevent_req * subreq)1675 static void cli_smb2_hardlink_closed(struct tevent_req *subreq)
1676 {
1677 NTSTATUS status = cli_smb2_close_fnum_recv(subreq);
1678 tevent_req_simple_finish_ntstatus(subreq, status);
1679 }
1680
cli_smb2_hardlink_recv(struct tevent_req * req)1681 static NTSTATUS cli_smb2_hardlink_recv(struct tevent_req *req)
1682 {
1683 struct cli_smb2_hardlink_state *state = tevent_req_data(
1684 req, struct cli_smb2_hardlink_state);
1685 NTSTATUS status;
1686
1687 if (tevent_req_is_nterror(req, &status)) {
1688 return status;
1689 }
1690 return state->status;
1691 }
1692
1693 struct cli_hardlink_state {
1694 uint8_t dummy;
1695 };
1696
1697 static void cli_hardlink_done(struct tevent_req *subreq);
1698 static void cli_hardlink_done2(struct tevent_req *subreq);
1699
cli_hardlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname_src,const char * fname_dst)1700 struct tevent_req *cli_hardlink_send(
1701 TALLOC_CTX *mem_ctx,
1702 struct tevent_context *ev,
1703 struct cli_state *cli,
1704 const char *fname_src,
1705 const char *fname_dst)
1706 {
1707 struct tevent_req *req = NULL, *subreq = NULL;
1708 struct cli_hardlink_state *state;
1709
1710 req = tevent_req_create(mem_ctx, &state, struct cli_hardlink_state);
1711 if (req == NULL) {
1712 return NULL;
1713 }
1714
1715 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
1716 subreq = cli_smb2_hardlink_send(
1717 state, ev, cli, fname_src, fname_dst, false, NULL);
1718 if (tevent_req_nomem(subreq, req)) {
1719 return tevent_req_post(req, ev);
1720 }
1721 tevent_req_set_callback(subreq, cli_hardlink_done2, req);
1722 return req;
1723 }
1724
1725 subreq = cli_nt_hardlink_send(state, ev, cli, fname_src, fname_dst);
1726 if (tevent_req_nomem(subreq, req)) {
1727 return tevent_req_post(req, ev);
1728 }
1729 tevent_req_set_callback(subreq, cli_hardlink_done, req);
1730 return req;
1731 }
1732
cli_hardlink_done(struct tevent_req * subreq)1733 static void cli_hardlink_done(struct tevent_req *subreq)
1734 {
1735 NTSTATUS status = cli_nt_hardlink_recv(subreq);
1736 tevent_req_simple_finish_ntstatus(subreq, status);
1737 }
1738
cli_hardlink_done2(struct tevent_req * subreq)1739 static void cli_hardlink_done2(struct tevent_req *subreq)
1740 {
1741 NTSTATUS status = cli_smb2_hardlink_recv(subreq);
1742 tevent_req_simple_finish_ntstatus(subreq, status);
1743 }
1744
cli_hardlink_recv(struct tevent_req * req)1745 NTSTATUS cli_hardlink_recv(struct tevent_req *req)
1746 {
1747 return tevent_req_simple_recv_ntstatus(req);
1748 }
1749
cli_hardlink(struct cli_state * cli,const char * fname_src,const char * fname_dst)1750 NTSTATUS cli_hardlink(
1751 struct cli_state *cli, const char *fname_src, const char *fname_dst)
1752 {
1753 TALLOC_CTX *frame = talloc_stackframe();
1754 struct tevent_context *ev = NULL;
1755 struct tevent_req *req = NULL;
1756 NTSTATUS status = NT_STATUS_NO_MEMORY;
1757
1758 if (smbXcli_conn_has_async_calls(cli->conn)) {
1759 status = NT_STATUS_INVALID_PARAMETER;
1760 goto fail;
1761 }
1762 ev = samba_tevent_context_init(frame);
1763 if (ev == NULL) {
1764 goto fail;
1765 }
1766 req = cli_hardlink_send(frame, ev, cli, fname_src, fname_dst);
1767 if (req == NULL) {
1768 goto fail;
1769 }
1770 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1771 goto fail;
1772 }
1773 status = cli_hardlink_recv(req);
1774 fail:
1775 TALLOC_FREE(frame);
1776 return status;
1777 }
1778
1779 /****************************************************************************
1780 Delete a file.
1781 ****************************************************************************/
1782
1783 static void cli_unlink_done(struct tevent_req *subreq);
1784
1785 struct cli_unlink_state {
1786 uint16_t vwv[1];
1787 };
1788
cli_unlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint16_t mayhave_attrs)1789 struct tevent_req *cli_unlink_send(TALLOC_CTX *mem_ctx,
1790 struct tevent_context *ev,
1791 struct cli_state *cli,
1792 const char *fname,
1793 uint16_t mayhave_attrs)
1794 {
1795 struct tevent_req *req = NULL, *subreq = NULL;
1796 struct cli_unlink_state *state = NULL;
1797 uint8_t additional_flags = 0;
1798 uint16_t additional_flags2 = 0;
1799 uint8_t *bytes = NULL;
1800
1801 req = tevent_req_create(mem_ctx, &state, struct cli_unlink_state);
1802 if (req == NULL) {
1803 return NULL;
1804 }
1805
1806 SSVAL(state->vwv+0, 0, mayhave_attrs);
1807
1808 bytes = talloc_array(state, uint8_t, 1);
1809 if (tevent_req_nomem(bytes, req)) {
1810 return tevent_req_post(req, ev);
1811 }
1812 bytes[0] = 4;
1813 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname,
1814 strlen(fname)+1, NULL);
1815
1816 if (tevent_req_nomem(bytes, req)) {
1817 return tevent_req_post(req, ev);
1818 }
1819
1820 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL)) {
1821 additional_flags2 = FLAGS2_REPARSE_PATH;
1822 }
1823
1824 subreq = cli_smb_send(state, ev, cli, SMBunlink, additional_flags,
1825 additional_flags2,
1826 1, state->vwv, talloc_get_size(bytes), bytes);
1827 if (tevent_req_nomem(subreq, req)) {
1828 return tevent_req_post(req, ev);
1829 }
1830 tevent_req_set_callback(subreq, cli_unlink_done, req);
1831 return req;
1832 }
1833
cli_unlink_done(struct tevent_req * subreq)1834 static void cli_unlink_done(struct tevent_req *subreq)
1835 {
1836 struct tevent_req *req = tevent_req_callback_data(
1837 subreq, struct tevent_req);
1838 NTSTATUS status;
1839
1840 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
1841 TALLOC_FREE(subreq);
1842 if (tevent_req_nterror(req, status)) {
1843 return;
1844 }
1845 tevent_req_done(req);
1846 }
1847
cli_unlink_recv(struct tevent_req * req)1848 NTSTATUS cli_unlink_recv(struct tevent_req *req)
1849 {
1850 return tevent_req_simple_recv_ntstatus(req);
1851 }
1852
cli_unlink(struct cli_state * cli,const char * fname,uint16_t mayhave_attrs)1853 NTSTATUS cli_unlink(struct cli_state *cli, const char *fname, uint16_t mayhave_attrs)
1854 {
1855 TALLOC_CTX *frame = NULL;
1856 struct tevent_context *ev;
1857 struct tevent_req *req;
1858 NTSTATUS status = NT_STATUS_OK;
1859
1860 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
1861 return cli_smb2_unlink(cli, fname, NULL);
1862 }
1863
1864 frame = talloc_stackframe();
1865
1866 if (smbXcli_conn_has_async_calls(cli->conn)) {
1867 /*
1868 * Can't use sync call while an async call is in flight
1869 */
1870 status = NT_STATUS_INVALID_PARAMETER;
1871 goto fail;
1872 }
1873
1874 ev = samba_tevent_context_init(frame);
1875 if (ev == NULL) {
1876 status = NT_STATUS_NO_MEMORY;
1877 goto fail;
1878 }
1879
1880 req = cli_unlink_send(frame, ev, cli, fname, mayhave_attrs);
1881 if (req == NULL) {
1882 status = NT_STATUS_NO_MEMORY;
1883 goto fail;
1884 }
1885
1886 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1887 goto fail;
1888 }
1889
1890 status = cli_unlink_recv(req);
1891
1892 fail:
1893 TALLOC_FREE(frame);
1894 return status;
1895 }
1896
1897 /****************************************************************************
1898 Create a directory.
1899 ****************************************************************************/
1900
1901 static void cli_mkdir_done(struct tevent_req *subreq);
1902
1903 struct cli_mkdir_state {
1904 int dummy;
1905 };
1906
cli_mkdir_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * dname)1907 struct tevent_req *cli_mkdir_send(TALLOC_CTX *mem_ctx,
1908 struct tevent_context *ev,
1909 struct cli_state *cli,
1910 const char *dname)
1911 {
1912 struct tevent_req *req = NULL, *subreq = NULL;
1913 struct cli_mkdir_state *state = NULL;
1914 uint8_t additional_flags = 0;
1915 uint16_t additional_flags2 = 0;
1916 uint8_t *bytes = NULL;
1917
1918 req = tevent_req_create(mem_ctx, &state, struct cli_mkdir_state);
1919 if (req == NULL) {
1920 return NULL;
1921 }
1922
1923 bytes = talloc_array(state, uint8_t, 1);
1924 if (tevent_req_nomem(bytes, req)) {
1925 return tevent_req_post(req, ev);
1926 }
1927 bytes[0] = 4;
1928 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), dname,
1929 strlen(dname)+1, NULL);
1930
1931 if (tevent_req_nomem(bytes, req)) {
1932 return tevent_req_post(req, ev);
1933 }
1934
1935 if (clistr_is_previous_version_path(dname, NULL, NULL, NULL)) {
1936 additional_flags2 = FLAGS2_REPARSE_PATH;
1937 }
1938
1939 subreq = cli_smb_send(state, ev, cli, SMBmkdir, additional_flags,
1940 additional_flags2,
1941 0, NULL, talloc_get_size(bytes), bytes);
1942 if (tevent_req_nomem(subreq, req)) {
1943 return tevent_req_post(req, ev);
1944 }
1945 tevent_req_set_callback(subreq, cli_mkdir_done, req);
1946 return req;
1947 }
1948
cli_mkdir_done(struct tevent_req * subreq)1949 static void cli_mkdir_done(struct tevent_req *subreq)
1950 {
1951 struct tevent_req *req = tevent_req_callback_data(
1952 subreq, struct tevent_req);
1953 NTSTATUS status;
1954
1955 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
1956 TALLOC_FREE(subreq);
1957 if (tevent_req_nterror(req, status)) {
1958 return;
1959 }
1960 tevent_req_done(req);
1961 }
1962
cli_mkdir_recv(struct tevent_req * req)1963 NTSTATUS cli_mkdir_recv(struct tevent_req *req)
1964 {
1965 return tevent_req_simple_recv_ntstatus(req);
1966 }
1967
cli_mkdir(struct cli_state * cli,const char * dname)1968 NTSTATUS cli_mkdir(struct cli_state *cli, const char *dname)
1969 {
1970 TALLOC_CTX *frame = NULL;
1971 struct tevent_context *ev;
1972 struct tevent_req *req;
1973 NTSTATUS status = NT_STATUS_OK;
1974
1975 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
1976 return cli_smb2_mkdir(cli, dname);
1977 }
1978
1979 frame = talloc_stackframe();
1980
1981 if (smbXcli_conn_has_async_calls(cli->conn)) {
1982 /*
1983 * Can't use sync call while an async call is in flight
1984 */
1985 status = NT_STATUS_INVALID_PARAMETER;
1986 goto fail;
1987 }
1988
1989 ev = samba_tevent_context_init(frame);
1990 if (ev == NULL) {
1991 status = NT_STATUS_NO_MEMORY;
1992 goto fail;
1993 }
1994
1995 req = cli_mkdir_send(frame, ev, cli, dname);
1996 if (req == NULL) {
1997 status = NT_STATUS_NO_MEMORY;
1998 goto fail;
1999 }
2000
2001 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
2002 goto fail;
2003 }
2004
2005 status = cli_mkdir_recv(req);
2006
2007 fail:
2008 TALLOC_FREE(frame);
2009 return status;
2010 }
2011
2012 /****************************************************************************
2013 Remove a directory.
2014 ****************************************************************************/
2015
2016 static void cli_rmdir_done(struct tevent_req *subreq);
2017
2018 struct cli_rmdir_state {
2019 int dummy;
2020 };
2021
cli_rmdir_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * dname)2022 struct tevent_req *cli_rmdir_send(TALLOC_CTX *mem_ctx,
2023 struct tevent_context *ev,
2024 struct cli_state *cli,
2025 const char *dname)
2026 {
2027 struct tevent_req *req = NULL, *subreq = NULL;
2028 struct cli_rmdir_state *state = NULL;
2029 uint8_t additional_flags = 0;
2030 uint16_t additional_flags2 = 0;
2031 uint8_t *bytes = NULL;
2032
2033 req = tevent_req_create(mem_ctx, &state, struct cli_rmdir_state);
2034 if (req == NULL) {
2035 return NULL;
2036 }
2037
2038 bytes = talloc_array(state, uint8_t, 1);
2039 if (tevent_req_nomem(bytes, req)) {
2040 return tevent_req_post(req, ev);
2041 }
2042 bytes[0] = 4;
2043 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), dname,
2044 strlen(dname)+1, NULL);
2045
2046 if (tevent_req_nomem(bytes, req)) {
2047 return tevent_req_post(req, ev);
2048 }
2049
2050 if (clistr_is_previous_version_path(dname, NULL, NULL, NULL)) {
2051 additional_flags2 = FLAGS2_REPARSE_PATH;
2052 }
2053
2054 subreq = cli_smb_send(state, ev, cli, SMBrmdir, additional_flags,
2055 additional_flags2,
2056 0, NULL, talloc_get_size(bytes), bytes);
2057 if (tevent_req_nomem(subreq, req)) {
2058 return tevent_req_post(req, ev);
2059 }
2060 tevent_req_set_callback(subreq, cli_rmdir_done, req);
2061 return req;
2062 }
2063
cli_rmdir_done(struct tevent_req * subreq)2064 static void cli_rmdir_done(struct tevent_req *subreq)
2065 {
2066 struct tevent_req *req = tevent_req_callback_data(
2067 subreq, struct tevent_req);
2068 NTSTATUS status;
2069
2070 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
2071 TALLOC_FREE(subreq);
2072 if (tevent_req_nterror(req, status)) {
2073 return;
2074 }
2075 tevent_req_done(req);
2076 }
2077
cli_rmdir_recv(struct tevent_req * req)2078 NTSTATUS cli_rmdir_recv(struct tevent_req *req)
2079 {
2080 return tevent_req_simple_recv_ntstatus(req);
2081 }
2082
cli_rmdir(struct cli_state * cli,const char * dname)2083 NTSTATUS cli_rmdir(struct cli_state *cli, const char *dname)
2084 {
2085 TALLOC_CTX *frame = NULL;
2086 struct tevent_context *ev;
2087 struct tevent_req *req;
2088 NTSTATUS status = NT_STATUS_OK;
2089
2090 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
2091 return cli_smb2_rmdir(cli, dname, NULL);
2092 }
2093
2094 frame = talloc_stackframe();
2095
2096 if (smbXcli_conn_has_async_calls(cli->conn)) {
2097 /*
2098 * Can't use sync call while an async call is in flight
2099 */
2100 status = NT_STATUS_INVALID_PARAMETER;
2101 goto fail;
2102 }
2103
2104 ev = samba_tevent_context_init(frame);
2105 if (ev == NULL) {
2106 status = NT_STATUS_NO_MEMORY;
2107 goto fail;
2108 }
2109
2110 req = cli_rmdir_send(frame, ev, cli, dname);
2111 if (req == NULL) {
2112 status = NT_STATUS_NO_MEMORY;
2113 goto fail;
2114 }
2115
2116 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
2117 goto fail;
2118 }
2119
2120 status = cli_rmdir_recv(req);
2121
2122 fail:
2123 TALLOC_FREE(frame);
2124 return status;
2125 }
2126
2127 /****************************************************************************
2128 Set or clear the delete on close flag.
2129 ****************************************************************************/
2130
2131 struct doc_state {
2132 uint16_t setup;
2133 uint8_t param[6];
2134 uint8_t data[1];
2135 };
2136
2137 static void cli_nt_delete_on_close_smb1_done(struct tevent_req *subreq);
2138 static void cli_nt_delete_on_close_smb2_done(struct tevent_req *subreq);
2139
cli_nt_delete_on_close_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,bool flag)2140 struct tevent_req *cli_nt_delete_on_close_send(TALLOC_CTX *mem_ctx,
2141 struct tevent_context *ev,
2142 struct cli_state *cli,
2143 uint16_t fnum,
2144 bool flag)
2145 {
2146 struct tevent_req *req = NULL, *subreq = NULL;
2147 struct doc_state *state = NULL;
2148
2149 req = tevent_req_create(mem_ctx, &state, struct doc_state);
2150 if (req == NULL) {
2151 return NULL;
2152 }
2153
2154 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
2155 subreq = cli_smb2_delete_on_close_send(state, ev, cli,
2156 fnum, flag);
2157 if (tevent_req_nomem(subreq, req)) {
2158 return tevent_req_post(req, ev);
2159 }
2160 tevent_req_set_callback(subreq,
2161 cli_nt_delete_on_close_smb2_done,
2162 req);
2163 return req;
2164 }
2165
2166 /* Setup setup word. */
2167 SSVAL(&state->setup, 0, TRANSACT2_SETFILEINFO);
2168
2169 /* Setup param array. */
2170 SSVAL(state->param,0,fnum);
2171 SSVAL(state->param,2,SMB_SET_FILE_DISPOSITION_INFO);
2172
2173 /* Setup data array. */
2174 SCVAL(&state->data[0], 0, flag ? 1 : 0);
2175
2176 subreq = cli_trans_send(state, /* mem ctx. */
2177 ev, /* event ctx. */
2178 cli, /* cli_state. */
2179 0, /* additional_flags2 */
2180 SMBtrans2, /* cmd. */
2181 NULL, /* pipe name. */
2182 -1, /* fid. */
2183 0, /* function. */
2184 0, /* flags. */
2185 &state->setup, /* setup. */
2186 1, /* num setup uint16_t words. */
2187 0, /* max returned setup. */
2188 state->param, /* param. */
2189 6, /* num param. */
2190 2, /* max returned param. */
2191 state->data, /* data. */
2192 1, /* num data. */
2193 0); /* max returned data. */
2194
2195 if (tevent_req_nomem(subreq, req)) {
2196 return tevent_req_post(req, ev);
2197 }
2198 tevent_req_set_callback(subreq,
2199 cli_nt_delete_on_close_smb1_done,
2200 req);
2201 return req;
2202 }
2203
cli_nt_delete_on_close_smb1_done(struct tevent_req * subreq)2204 static void cli_nt_delete_on_close_smb1_done(struct tevent_req *subreq)
2205 {
2206 NTSTATUS status = cli_trans_recv(subreq, NULL, NULL, NULL, 0, NULL,
2207 NULL, 0, NULL, NULL, 0, NULL);
2208 tevent_req_simple_finish_ntstatus(subreq, status);
2209 }
2210
cli_nt_delete_on_close_smb2_done(struct tevent_req * subreq)2211 static void cli_nt_delete_on_close_smb2_done(struct tevent_req *subreq)
2212 {
2213 NTSTATUS status = cli_smb2_delete_on_close_recv(subreq);
2214 tevent_req_simple_finish_ntstatus(subreq, status);
2215 }
2216
cli_nt_delete_on_close_recv(struct tevent_req * req)2217 NTSTATUS cli_nt_delete_on_close_recv(struct tevent_req *req)
2218 {
2219 return tevent_req_simple_recv_ntstatus(req);
2220 }
2221
cli_nt_delete_on_close(struct cli_state * cli,uint16_t fnum,bool flag)2222 NTSTATUS cli_nt_delete_on_close(struct cli_state *cli, uint16_t fnum, bool flag)
2223 {
2224 TALLOC_CTX *frame = talloc_stackframe();
2225 struct tevent_context *ev = NULL;
2226 struct tevent_req *req = NULL;
2227 NTSTATUS status = NT_STATUS_OK;
2228
2229 if (smbXcli_conn_has_async_calls(cli->conn)) {
2230 /*
2231 * Can't use sync call while an async call is in flight
2232 */
2233 status = NT_STATUS_INVALID_PARAMETER;
2234 goto fail;
2235 }
2236
2237 ev = samba_tevent_context_init(frame);
2238 if (ev == NULL) {
2239 status = NT_STATUS_NO_MEMORY;
2240 goto fail;
2241 }
2242
2243 req = cli_nt_delete_on_close_send(frame,
2244 ev,
2245 cli,
2246 fnum,
2247 flag);
2248 if (req == NULL) {
2249 status = NT_STATUS_NO_MEMORY;
2250 goto fail;
2251 }
2252
2253 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
2254 goto fail;
2255 }
2256
2257 status = cli_nt_delete_on_close_recv(req);
2258
2259 fail:
2260 TALLOC_FREE(frame);
2261 return status;
2262 }
2263
2264 struct cli_ntcreate1_state {
2265 uint16_t vwv[24];
2266 uint16_t fnum;
2267 struct smb_create_returns cr;
2268 struct tevent_req *subreq;
2269 };
2270
2271 static void cli_ntcreate1_done(struct tevent_req *subreq);
2272 static bool cli_ntcreate1_cancel(struct tevent_req *req);
2273
cli_ntcreate1_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint32_t CreatFlags,uint32_t DesiredAccess,uint32_t FileAttributes,uint32_t ShareAccess,uint32_t CreateDisposition,uint32_t CreateOptions,uint32_t ImpersonationLevel,uint8_t SecurityFlags)2274 static struct tevent_req *cli_ntcreate1_send(TALLOC_CTX *mem_ctx,
2275 struct tevent_context *ev,
2276 struct cli_state *cli,
2277 const char *fname,
2278 uint32_t CreatFlags,
2279 uint32_t DesiredAccess,
2280 uint32_t FileAttributes,
2281 uint32_t ShareAccess,
2282 uint32_t CreateDisposition,
2283 uint32_t CreateOptions,
2284 uint32_t ImpersonationLevel,
2285 uint8_t SecurityFlags)
2286 {
2287 struct tevent_req *req, *subreq;
2288 struct cli_ntcreate1_state *state;
2289 uint16_t *vwv;
2290 uint8_t *bytes;
2291 size_t converted_len;
2292 uint16_t additional_flags2 = 0;
2293
2294 req = tevent_req_create(mem_ctx, &state, struct cli_ntcreate1_state);
2295 if (req == NULL) {
2296 return NULL;
2297 }
2298
2299 vwv = state->vwv;
2300
2301 SCVAL(vwv+0, 0, 0xFF);
2302 SCVAL(vwv+0, 1, 0);
2303 SSVAL(vwv+1, 0, 0);
2304 SCVAL(vwv+2, 0, 0);
2305
2306 if (cli->use_oplocks) {
2307 CreatFlags |= (REQUEST_OPLOCK|REQUEST_BATCH_OPLOCK);
2308 }
2309 SIVAL(vwv+3, 1, CreatFlags);
2310 SIVAL(vwv+5, 1, 0x0); /* RootDirectoryFid */
2311 SIVAL(vwv+7, 1, DesiredAccess);
2312 SIVAL(vwv+9, 1, 0x0); /* AllocationSize */
2313 SIVAL(vwv+11, 1, 0x0); /* AllocationSize */
2314 SIVAL(vwv+13, 1, FileAttributes);
2315 SIVAL(vwv+15, 1, ShareAccess);
2316 SIVAL(vwv+17, 1, CreateDisposition);
2317 SIVAL(vwv+19, 1, CreateOptions |
2318 (cli->backup_intent ? FILE_OPEN_FOR_BACKUP_INTENT : 0));
2319 SIVAL(vwv+21, 1, ImpersonationLevel);
2320 SCVAL(vwv+23, 1, SecurityFlags);
2321
2322 bytes = talloc_array(state, uint8_t, 0);
2323 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn),
2324 fname, strlen(fname)+1,
2325 &converted_len);
2326
2327 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL)) {
2328 additional_flags2 = FLAGS2_REPARSE_PATH;
2329 }
2330
2331 /* sigh. this copes with broken netapp filer behaviour */
2332 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), "", 1, NULL);
2333
2334 if (tevent_req_nomem(bytes, req)) {
2335 return tevent_req_post(req, ev);
2336 }
2337
2338 SSVAL(vwv+2, 1, converted_len);
2339
2340 subreq = cli_smb_send(state, ev, cli, SMBntcreateX, 0,
2341 additional_flags2, 24, vwv,
2342 talloc_get_size(bytes), bytes);
2343 if (tevent_req_nomem(subreq, req)) {
2344 return tevent_req_post(req, ev);
2345 }
2346 tevent_req_set_callback(subreq, cli_ntcreate1_done, req);
2347
2348 state->subreq = subreq;
2349 tevent_req_set_cancel_fn(req, cli_ntcreate1_cancel);
2350
2351 return req;
2352 }
2353
cli_ntcreate1_done(struct tevent_req * subreq)2354 static void cli_ntcreate1_done(struct tevent_req *subreq)
2355 {
2356 struct tevent_req *req = tevent_req_callback_data(
2357 subreq, struct tevent_req);
2358 struct cli_ntcreate1_state *state = tevent_req_data(
2359 req, struct cli_ntcreate1_state);
2360 uint8_t wct;
2361 uint16_t *vwv;
2362 uint32_t num_bytes;
2363 uint8_t *bytes;
2364 NTSTATUS status;
2365
2366 status = cli_smb_recv(subreq, state, NULL, 34, &wct, &vwv,
2367 &num_bytes, &bytes);
2368 TALLOC_FREE(subreq);
2369 if (tevent_req_nterror(req, status)) {
2370 return;
2371 }
2372 state->cr.oplock_level = CVAL(vwv+2, 0);
2373 state->fnum = SVAL(vwv+2, 1);
2374 state->cr.create_action = IVAL(vwv+3, 1);
2375 state->cr.creation_time = BVAL(vwv+5, 1);
2376 state->cr.last_access_time = BVAL(vwv+9, 1);
2377 state->cr.last_write_time = BVAL(vwv+13, 1);
2378 state->cr.change_time = BVAL(vwv+17, 1);
2379 state->cr.file_attributes = IVAL(vwv+21, 1);
2380 state->cr.allocation_size = BVAL(vwv+23, 1);
2381 state->cr.end_of_file = BVAL(vwv+27, 1);
2382
2383 tevent_req_done(req);
2384 }
2385
cli_ntcreate1_cancel(struct tevent_req * req)2386 static bool cli_ntcreate1_cancel(struct tevent_req *req)
2387 {
2388 struct cli_ntcreate1_state *state = tevent_req_data(
2389 req, struct cli_ntcreate1_state);
2390 return tevent_req_cancel(state->subreq);
2391 }
2392
cli_ntcreate1_recv(struct tevent_req * req,uint16_t * pfnum,struct smb_create_returns * cr)2393 static NTSTATUS cli_ntcreate1_recv(struct tevent_req *req,
2394 uint16_t *pfnum,
2395 struct smb_create_returns *cr)
2396 {
2397 struct cli_ntcreate1_state *state = tevent_req_data(
2398 req, struct cli_ntcreate1_state);
2399 NTSTATUS status;
2400
2401 if (tevent_req_is_nterror(req, &status)) {
2402 return status;
2403 }
2404 *pfnum = state->fnum;
2405 if (cr != NULL) {
2406 *cr = state->cr;
2407 }
2408 return NT_STATUS_OK;
2409 }
2410
2411 struct cli_ntcreate_state {
2412 struct smb_create_returns cr;
2413 uint16_t fnum;
2414 struct tevent_req *subreq;
2415 };
2416
2417 static void cli_ntcreate_done_nt1(struct tevent_req *subreq);
2418 static void cli_ntcreate_done_smb2(struct tevent_req *subreq);
2419 static bool cli_ntcreate_cancel(struct tevent_req *req);
2420
cli_ntcreate_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint32_t create_flags,uint32_t desired_access,uint32_t file_attributes,uint32_t share_access,uint32_t create_disposition,uint32_t create_options,uint32_t impersonation_level,uint8_t security_flags)2421 struct tevent_req *cli_ntcreate_send(TALLOC_CTX *mem_ctx,
2422 struct tevent_context *ev,
2423 struct cli_state *cli,
2424 const char *fname,
2425 uint32_t create_flags,
2426 uint32_t desired_access,
2427 uint32_t file_attributes,
2428 uint32_t share_access,
2429 uint32_t create_disposition,
2430 uint32_t create_options,
2431 uint32_t impersonation_level,
2432 uint8_t security_flags)
2433 {
2434 struct tevent_req *req, *subreq;
2435 struct cli_ntcreate_state *state;
2436
2437 req = tevent_req_create(mem_ctx, &state, struct cli_ntcreate_state);
2438 if (req == NULL) {
2439 return NULL;
2440 }
2441
2442 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
2443 if (cli->use_oplocks) {
2444 create_flags |= REQUEST_OPLOCK|REQUEST_BATCH_OPLOCK;
2445 }
2446
2447 subreq = cli_smb2_create_fnum_send(
2448 state,
2449 ev,
2450 cli,
2451 fname,
2452 create_flags,
2453 impersonation_level,
2454 desired_access,
2455 file_attributes,
2456 share_access,
2457 create_disposition,
2458 create_options,
2459 NULL);
2460 if (tevent_req_nomem(subreq, req)) {
2461 return tevent_req_post(req, ev);
2462 }
2463 tevent_req_set_callback(subreq, cli_ntcreate_done_smb2, req);
2464 } else {
2465 subreq = cli_ntcreate1_send(
2466 state, ev, cli, fname, create_flags, desired_access,
2467 file_attributes, share_access, create_disposition,
2468 create_options, impersonation_level, security_flags);
2469 if (tevent_req_nomem(subreq, req)) {
2470 return tevent_req_post(req, ev);
2471 }
2472 tevent_req_set_callback(subreq, cli_ntcreate_done_nt1, req);
2473 }
2474
2475 state->subreq = subreq;
2476 tevent_req_set_cancel_fn(req, cli_ntcreate_cancel);
2477
2478 return req;
2479 }
2480
cli_ntcreate_done_nt1(struct tevent_req * subreq)2481 static void cli_ntcreate_done_nt1(struct tevent_req *subreq)
2482 {
2483 struct tevent_req *req = tevent_req_callback_data(
2484 subreq, struct tevent_req);
2485 struct cli_ntcreate_state *state = tevent_req_data(
2486 req, struct cli_ntcreate_state);
2487 NTSTATUS status;
2488
2489 status = cli_ntcreate1_recv(subreq, &state->fnum, &state->cr);
2490 TALLOC_FREE(subreq);
2491 if (tevent_req_nterror(req, status)) {
2492 return;
2493 }
2494 tevent_req_done(req);
2495 }
2496
cli_ntcreate_done_smb2(struct tevent_req * subreq)2497 static void cli_ntcreate_done_smb2(struct tevent_req *subreq)
2498 {
2499 struct tevent_req *req = tevent_req_callback_data(
2500 subreq, struct tevent_req);
2501 struct cli_ntcreate_state *state = tevent_req_data(
2502 req, struct cli_ntcreate_state);
2503 NTSTATUS status;
2504
2505 status = cli_smb2_create_fnum_recv(
2506 subreq,
2507 &state->fnum,
2508 &state->cr,
2509 NULL,
2510 NULL);
2511 TALLOC_FREE(subreq);
2512 if (tevent_req_nterror(req, status)) {
2513 return;
2514 }
2515 tevent_req_done(req);
2516 }
2517
cli_ntcreate_cancel(struct tevent_req * req)2518 static bool cli_ntcreate_cancel(struct tevent_req *req)
2519 {
2520 struct cli_ntcreate_state *state = tevent_req_data(
2521 req, struct cli_ntcreate_state);
2522 return tevent_req_cancel(state->subreq);
2523 }
2524
cli_ntcreate_recv(struct tevent_req * req,uint16_t * fnum,struct smb_create_returns * cr)2525 NTSTATUS cli_ntcreate_recv(struct tevent_req *req, uint16_t *fnum,
2526 struct smb_create_returns *cr)
2527 {
2528 struct cli_ntcreate_state *state = tevent_req_data(
2529 req, struct cli_ntcreate_state);
2530 NTSTATUS status;
2531
2532 if (tevent_req_is_nterror(req, &status)) {
2533 return status;
2534 }
2535 if (fnum != NULL) {
2536 *fnum = state->fnum;
2537 }
2538 if (cr != NULL) {
2539 *cr = state->cr;
2540 }
2541 return NT_STATUS_OK;
2542 }
2543
cli_ntcreate(struct cli_state * cli,const char * fname,uint32_t CreatFlags,uint32_t DesiredAccess,uint32_t FileAttributes,uint32_t ShareAccess,uint32_t CreateDisposition,uint32_t CreateOptions,uint8_t SecurityFlags,uint16_t * pfid,struct smb_create_returns * cr)2544 NTSTATUS cli_ntcreate(struct cli_state *cli,
2545 const char *fname,
2546 uint32_t CreatFlags,
2547 uint32_t DesiredAccess,
2548 uint32_t FileAttributes,
2549 uint32_t ShareAccess,
2550 uint32_t CreateDisposition,
2551 uint32_t CreateOptions,
2552 uint8_t SecurityFlags,
2553 uint16_t *pfid,
2554 struct smb_create_returns *cr)
2555 {
2556 TALLOC_CTX *frame = talloc_stackframe();
2557 struct tevent_context *ev;
2558 struct tevent_req *req;
2559 uint32_t ImpersonationLevel = SMB2_IMPERSONATION_IMPERSONATION;
2560 NTSTATUS status = NT_STATUS_NO_MEMORY;
2561
2562 if (smbXcli_conn_has_async_calls(cli->conn)) {
2563 /*
2564 * Can't use sync call while an async call is in flight
2565 */
2566 status = NT_STATUS_INVALID_PARAMETER;
2567 goto fail;
2568 }
2569
2570 ev = samba_tevent_context_init(frame);
2571 if (ev == NULL) {
2572 goto fail;
2573 }
2574
2575 req = cli_ntcreate_send(frame, ev, cli, fname, CreatFlags,
2576 DesiredAccess, FileAttributes, ShareAccess,
2577 CreateDisposition, CreateOptions,
2578 ImpersonationLevel, SecurityFlags);
2579 if (req == NULL) {
2580 goto fail;
2581 }
2582
2583 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
2584 goto fail;
2585 }
2586
2587 status = cli_ntcreate_recv(req, pfid, cr);
2588 fail:
2589 TALLOC_FREE(frame);
2590 return status;
2591 }
2592
2593 struct cli_nttrans_create_state {
2594 uint16_t fnum;
2595 struct smb_create_returns cr;
2596 };
2597
2598 static void cli_nttrans_create_done(struct tevent_req *subreq);
2599
cli_nttrans_create_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint32_t CreatFlags,uint32_t DesiredAccess,uint32_t FileAttributes,uint32_t ShareAccess,uint32_t CreateDisposition,uint32_t CreateOptions,uint8_t SecurityFlags,struct security_descriptor * secdesc,struct ea_struct * eas,int num_eas)2600 struct tevent_req *cli_nttrans_create_send(TALLOC_CTX *mem_ctx,
2601 struct tevent_context *ev,
2602 struct cli_state *cli,
2603 const char *fname,
2604 uint32_t CreatFlags,
2605 uint32_t DesiredAccess,
2606 uint32_t FileAttributes,
2607 uint32_t ShareAccess,
2608 uint32_t CreateDisposition,
2609 uint32_t CreateOptions,
2610 uint8_t SecurityFlags,
2611 struct security_descriptor *secdesc,
2612 struct ea_struct *eas,
2613 int num_eas)
2614 {
2615 struct tevent_req *req, *subreq;
2616 struct cli_nttrans_create_state *state;
2617 uint8_t *param;
2618 uint8_t *secdesc_buf;
2619 size_t secdesc_len;
2620 NTSTATUS status;
2621 size_t converted_len;
2622 uint16_t additional_flags2 = 0;
2623
2624 req = tevent_req_create(mem_ctx,
2625 &state, struct cli_nttrans_create_state);
2626 if (req == NULL) {
2627 return NULL;
2628 }
2629
2630 if (secdesc != NULL) {
2631 status = marshall_sec_desc(talloc_tos(), secdesc,
2632 &secdesc_buf, &secdesc_len);
2633 if (tevent_req_nterror(req, status)) {
2634 DEBUG(10, ("marshall_sec_desc failed: %s\n",
2635 nt_errstr(status)));
2636 return tevent_req_post(req, ev);
2637 }
2638 } else {
2639 secdesc_buf = NULL;
2640 secdesc_len = 0;
2641 }
2642
2643 if (num_eas != 0) {
2644 /*
2645 * TODO ;-)
2646 */
2647 tevent_req_nterror(req, NT_STATUS_NOT_IMPLEMENTED);
2648 return tevent_req_post(req, ev);
2649 }
2650
2651 param = talloc_array(state, uint8_t, 53);
2652 if (tevent_req_nomem(param, req)) {
2653 return tevent_req_post(req, ev);
2654 }
2655
2656 param = trans2_bytes_push_str(param, smbXcli_conn_use_unicode(cli->conn),
2657 fname, strlen(fname),
2658 &converted_len);
2659 if (tevent_req_nomem(param, req)) {
2660 return tevent_req_post(req, ev);
2661 }
2662
2663 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL)) {
2664 additional_flags2 = FLAGS2_REPARSE_PATH;
2665 }
2666
2667 SIVAL(param, 0, CreatFlags);
2668 SIVAL(param, 4, 0x0); /* RootDirectoryFid */
2669 SIVAL(param, 8, DesiredAccess);
2670 SIVAL(param, 12, 0x0); /* AllocationSize */
2671 SIVAL(param, 16, 0x0); /* AllocationSize */
2672 SIVAL(param, 20, FileAttributes);
2673 SIVAL(param, 24, ShareAccess);
2674 SIVAL(param, 28, CreateDisposition);
2675 SIVAL(param, 32, CreateOptions |
2676 (cli->backup_intent ? FILE_OPEN_FOR_BACKUP_INTENT : 0));
2677 SIVAL(param, 36, secdesc_len);
2678 SIVAL(param, 40, 0); /* EA length*/
2679 SIVAL(param, 44, converted_len);
2680 SIVAL(param, 48, 0x02); /* ImpersonationLevel */
2681 SCVAL(param, 52, SecurityFlags);
2682
2683 subreq = cli_trans_send(state, ev, cli,
2684 additional_flags2, /* additional_flags2 */
2685 SMBnttrans,
2686 NULL, -1, /* name, fid */
2687 NT_TRANSACT_CREATE, 0,
2688 NULL, 0, 0, /* setup */
2689 param, talloc_get_size(param), 128, /* param */
2690 secdesc_buf, secdesc_len, 0); /* data */
2691 if (tevent_req_nomem(subreq, req)) {
2692 return tevent_req_post(req, ev);
2693 }
2694 tevent_req_set_callback(subreq, cli_nttrans_create_done, req);
2695 return req;
2696 }
2697
cli_nttrans_create_done(struct tevent_req * subreq)2698 static void cli_nttrans_create_done(struct tevent_req *subreq)
2699 {
2700 struct tevent_req *req = tevent_req_callback_data(
2701 subreq, struct tevent_req);
2702 struct cli_nttrans_create_state *state = tevent_req_data(
2703 req, struct cli_nttrans_create_state);
2704 uint8_t *param;
2705 uint32_t num_param;
2706 NTSTATUS status;
2707
2708 status = cli_trans_recv(subreq, talloc_tos(), NULL,
2709 NULL, 0, NULL, /* rsetup */
2710 ¶m, 69, &num_param,
2711 NULL, 0, NULL);
2712 if (tevent_req_nterror(req, status)) {
2713 return;
2714 }
2715 state->cr.oplock_level = CVAL(param, 0);
2716 state->fnum = SVAL(param, 2);
2717 state->cr.create_action = IVAL(param, 4);
2718 state->cr.creation_time = BVAL(param, 12);
2719 state->cr.last_access_time = BVAL(param, 20);
2720 state->cr.last_write_time = BVAL(param, 28);
2721 state->cr.change_time = BVAL(param, 36);
2722 state->cr.file_attributes = IVAL(param, 44);
2723 state->cr.allocation_size = BVAL(param, 48);
2724 state->cr.end_of_file = BVAL(param, 56);
2725
2726 TALLOC_FREE(param);
2727 tevent_req_done(req);
2728 }
2729
cli_nttrans_create_recv(struct tevent_req * req,uint16_t * fnum,struct smb_create_returns * cr)2730 NTSTATUS cli_nttrans_create_recv(struct tevent_req *req,
2731 uint16_t *fnum,
2732 struct smb_create_returns *cr)
2733 {
2734 struct cli_nttrans_create_state *state = tevent_req_data(
2735 req, struct cli_nttrans_create_state);
2736 NTSTATUS status;
2737
2738 if (tevent_req_is_nterror(req, &status)) {
2739 return status;
2740 }
2741 *fnum = state->fnum;
2742 if (cr != NULL) {
2743 *cr = state->cr;
2744 }
2745 return NT_STATUS_OK;
2746 }
2747
cli_nttrans_create(struct cli_state * cli,const char * fname,uint32_t CreatFlags,uint32_t DesiredAccess,uint32_t FileAttributes,uint32_t ShareAccess,uint32_t CreateDisposition,uint32_t CreateOptions,uint8_t SecurityFlags,struct security_descriptor * secdesc,struct ea_struct * eas,int num_eas,uint16_t * pfid,struct smb_create_returns * cr)2748 NTSTATUS cli_nttrans_create(struct cli_state *cli,
2749 const char *fname,
2750 uint32_t CreatFlags,
2751 uint32_t DesiredAccess,
2752 uint32_t FileAttributes,
2753 uint32_t ShareAccess,
2754 uint32_t CreateDisposition,
2755 uint32_t CreateOptions,
2756 uint8_t SecurityFlags,
2757 struct security_descriptor *secdesc,
2758 struct ea_struct *eas,
2759 int num_eas,
2760 uint16_t *pfid,
2761 struct smb_create_returns *cr)
2762 {
2763 TALLOC_CTX *frame = talloc_stackframe();
2764 struct tevent_context *ev;
2765 struct tevent_req *req;
2766 NTSTATUS status = NT_STATUS_NO_MEMORY;
2767
2768 if (smbXcli_conn_has_async_calls(cli->conn)) {
2769 /*
2770 * Can't use sync call while an async call is in flight
2771 */
2772 status = NT_STATUS_INVALID_PARAMETER;
2773 goto fail;
2774 }
2775 ev = samba_tevent_context_init(frame);
2776 if (ev == NULL) {
2777 goto fail;
2778 }
2779 req = cli_nttrans_create_send(frame, ev, cli, fname, CreatFlags,
2780 DesiredAccess, FileAttributes,
2781 ShareAccess, CreateDisposition,
2782 CreateOptions, SecurityFlags,
2783 secdesc, eas, num_eas);
2784 if (req == NULL) {
2785 goto fail;
2786 }
2787 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
2788 goto fail;
2789 }
2790 status = cli_nttrans_create_recv(req, pfid, cr);
2791 fail:
2792 TALLOC_FREE(frame);
2793 return status;
2794 }
2795
2796 /****************************************************************************
2797 Open a file
2798 WARNING: if you open with O_WRONLY then getattrE won't work!
2799 ****************************************************************************/
2800
2801 struct cli_openx_state {
2802 const char *fname;
2803 uint16_t vwv[15];
2804 uint16_t fnum;
2805 struct iovec bytes;
2806 };
2807
2808 static void cli_openx_done(struct tevent_req *subreq);
2809
cli_openx_create(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,int flags,int share_mode,struct tevent_req ** psmbreq)2810 struct tevent_req *cli_openx_create(TALLOC_CTX *mem_ctx,
2811 struct tevent_context *ev,
2812 struct cli_state *cli, const char *fname,
2813 int flags, int share_mode,
2814 struct tevent_req **psmbreq)
2815 {
2816 struct tevent_req *req, *subreq;
2817 struct cli_openx_state *state;
2818 unsigned openfn;
2819 unsigned accessmode;
2820 uint8_t additional_flags;
2821 uint16_t additional_flags2 = 0;
2822 uint8_t *bytes;
2823
2824 req = tevent_req_create(mem_ctx, &state, struct cli_openx_state);
2825 if (req == NULL) {
2826 return NULL;
2827 }
2828
2829 openfn = 0;
2830 if (flags & O_CREAT) {
2831 openfn |= (1<<4);
2832 }
2833 if (!(flags & O_EXCL)) {
2834 if (flags & O_TRUNC)
2835 openfn |= (1<<1);
2836 else
2837 openfn |= (1<<0);
2838 }
2839
2840 accessmode = (share_mode<<4);
2841
2842 if ((flags & O_ACCMODE) == O_RDWR) {
2843 accessmode |= 2;
2844 } else if ((flags & O_ACCMODE) == O_WRONLY) {
2845 accessmode |= 1;
2846 }
2847
2848 #if defined(O_SYNC)
2849 if ((flags & O_SYNC) == O_SYNC) {
2850 accessmode |= (1<<14);
2851 }
2852 #endif /* O_SYNC */
2853
2854 if (share_mode == DENY_FCB) {
2855 accessmode = 0xFF;
2856 }
2857
2858 SCVAL(state->vwv + 0, 0, 0xFF);
2859 SCVAL(state->vwv + 0, 1, 0);
2860 SSVAL(state->vwv + 1, 0, 0);
2861 SSVAL(state->vwv + 2, 0, 0); /* no additional info */
2862 SSVAL(state->vwv + 3, 0, accessmode);
2863 SSVAL(state->vwv + 4, 0, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);
2864 SSVAL(state->vwv + 5, 0, 0);
2865 SIVAL(state->vwv + 6, 0, 0);
2866 SSVAL(state->vwv + 8, 0, openfn);
2867 SIVAL(state->vwv + 9, 0, 0);
2868 SIVAL(state->vwv + 11, 0, 0);
2869 SIVAL(state->vwv + 13, 0, 0);
2870
2871 additional_flags = 0;
2872
2873 if (cli->use_oplocks) {
2874 /* if using oplocks then ask for a batch oplock via
2875 core and extended methods */
2876 additional_flags =
2877 FLAG_REQUEST_OPLOCK|FLAG_REQUEST_BATCH_OPLOCK;
2878 SSVAL(state->vwv+2, 0, SVAL(state->vwv+2, 0) | 6);
2879 }
2880
2881 bytes = talloc_array(state, uint8_t, 0);
2882 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname,
2883 strlen(fname)+1, NULL);
2884
2885 if (tevent_req_nomem(bytes, req)) {
2886 return tevent_req_post(req, ev);
2887 }
2888
2889 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL)) {
2890 additional_flags2 = FLAGS2_REPARSE_PATH;
2891 }
2892
2893 state->bytes.iov_base = (void *)bytes;
2894 state->bytes.iov_len = talloc_get_size(bytes);
2895
2896 subreq = cli_smb_req_create(state, ev, cli, SMBopenX, additional_flags,
2897 additional_flags2, 15, state->vwv, 1, &state->bytes);
2898 if (subreq == NULL) {
2899 TALLOC_FREE(req);
2900 return NULL;
2901 }
2902 tevent_req_set_callback(subreq, cli_openx_done, req);
2903 *psmbreq = subreq;
2904 return req;
2905 }
2906
cli_openx_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,int flags,int share_mode)2907 struct tevent_req *cli_openx_send(TALLOC_CTX *mem_ctx, struct tevent_context *ev,
2908 struct cli_state *cli, const char *fname,
2909 int flags, int share_mode)
2910 {
2911 struct tevent_req *req, *subreq;
2912 NTSTATUS status;
2913
2914 req = cli_openx_create(mem_ctx, ev, cli, fname, flags, share_mode,
2915 &subreq);
2916 if (req == NULL) {
2917 return NULL;
2918 }
2919
2920 status = smb1cli_req_chain_submit(&subreq, 1);
2921 if (tevent_req_nterror(req, status)) {
2922 return tevent_req_post(req, ev);
2923 }
2924 return req;
2925 }
2926
cli_openx_done(struct tevent_req * subreq)2927 static void cli_openx_done(struct tevent_req *subreq)
2928 {
2929 struct tevent_req *req = tevent_req_callback_data(
2930 subreq, struct tevent_req);
2931 struct cli_openx_state *state = tevent_req_data(
2932 req, struct cli_openx_state);
2933 uint8_t wct;
2934 uint16_t *vwv;
2935 NTSTATUS status;
2936
2937 status = cli_smb_recv(subreq, state, NULL, 3, &wct, &vwv, NULL,
2938 NULL);
2939 TALLOC_FREE(subreq);
2940 if (tevent_req_nterror(req, status)) {
2941 return;
2942 }
2943 state->fnum = SVAL(vwv+2, 0);
2944 tevent_req_done(req);
2945 }
2946
cli_openx_recv(struct tevent_req * req,uint16_t * pfnum)2947 NTSTATUS cli_openx_recv(struct tevent_req *req, uint16_t *pfnum)
2948 {
2949 struct cli_openx_state *state = tevent_req_data(
2950 req, struct cli_openx_state);
2951 NTSTATUS status;
2952
2953 if (tevent_req_is_nterror(req, &status)) {
2954 return status;
2955 }
2956 *pfnum = state->fnum;
2957 return NT_STATUS_OK;
2958 }
2959
cli_openx(struct cli_state * cli,const char * fname,int flags,int share_mode,uint16_t * pfnum)2960 NTSTATUS cli_openx(struct cli_state *cli, const char *fname, int flags,
2961 int share_mode, uint16_t *pfnum)
2962 {
2963 TALLOC_CTX *frame = talloc_stackframe();
2964 struct tevent_context *ev;
2965 struct tevent_req *req;
2966 NTSTATUS status = NT_STATUS_NO_MEMORY;
2967
2968 if (smbXcli_conn_has_async_calls(cli->conn)) {
2969 /*
2970 * Can't use sync call while an async call is in flight
2971 */
2972 status = NT_STATUS_INVALID_PARAMETER;
2973 goto fail;
2974 }
2975
2976 ev = samba_tevent_context_init(frame);
2977 if (ev == NULL) {
2978 goto fail;
2979 }
2980
2981 req = cli_openx_send(frame, ev, cli, fname, flags, share_mode);
2982 if (req == NULL) {
2983 goto fail;
2984 }
2985
2986 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
2987 goto fail;
2988 }
2989
2990 status = cli_openx_recv(req, pfnum);
2991 fail:
2992 TALLOC_FREE(frame);
2993 return status;
2994 }
2995 /****************************************************************************
2996 Synchronous wrapper function that does an NtCreateX open by preference
2997 and falls back to openX if this fails.
2998 ****************************************************************************/
2999
cli_open(struct cli_state * cli,const char * fname,int flags,int share_mode_in,uint16_t * pfnum)3000 NTSTATUS cli_open(struct cli_state *cli, const char *fname, int flags,
3001 int share_mode_in, uint16_t *pfnum)
3002 {
3003 NTSTATUS status;
3004 unsigned int openfn = 0;
3005 unsigned int dos_deny = 0;
3006 uint32_t access_mask, share_mode, create_disposition, create_options;
3007 struct smb_create_returns cr = {0};
3008
3009 /* Do the initial mapping into OpenX parameters. */
3010 if (flags & O_CREAT) {
3011 openfn |= (1<<4);
3012 }
3013 if (!(flags & O_EXCL)) {
3014 if (flags & O_TRUNC)
3015 openfn |= (1<<1);
3016 else
3017 openfn |= (1<<0);
3018 }
3019
3020 dos_deny = (share_mode_in<<4);
3021
3022 if ((flags & O_ACCMODE) == O_RDWR) {
3023 dos_deny |= 2;
3024 } else if ((flags & O_ACCMODE) == O_WRONLY) {
3025 dos_deny |= 1;
3026 }
3027
3028 #if defined(O_SYNC)
3029 if ((flags & O_SYNC) == O_SYNC) {
3030 dos_deny |= (1<<14);
3031 }
3032 #endif /* O_SYNC */
3033
3034 if (share_mode_in == DENY_FCB) {
3035 dos_deny = 0xFF;
3036 }
3037
3038 #if 0
3039 /* Hmmm. This is what I think the above code
3040 should look like if it's using the constants
3041 we #define. JRA. */
3042
3043 if (flags & O_CREAT) {
3044 openfn |= OPENX_FILE_CREATE_IF_NOT_EXIST;
3045 }
3046 if (!(flags & O_EXCL)) {
3047 if (flags & O_TRUNC)
3048 openfn |= OPENX_FILE_EXISTS_TRUNCATE;
3049 else
3050 openfn |= OPENX_FILE_EXISTS_OPEN;
3051 }
3052
3053 dos_deny = SET_DENY_MODE(share_mode_in);
3054
3055 if ((flags & O_ACCMODE) == O_RDWR) {
3056 dos_deny |= DOS_OPEN_RDWR;
3057 } else if ((flags & O_ACCMODE) == O_WRONLY) {
3058 dos_deny |= DOS_OPEN_WRONLY;
3059 }
3060
3061 #if defined(O_SYNC)
3062 if ((flags & O_SYNC) == O_SYNC) {
3063 dos_deny |= FILE_SYNC_OPENMODE;
3064 }
3065 #endif /* O_SYNC */
3066
3067 if (share_mode_in == DENY_FCB) {
3068 dos_deny = 0xFF;
3069 }
3070 #endif
3071
3072 if (!map_open_params_to_ntcreate(fname, dos_deny,
3073 openfn, &access_mask,
3074 &share_mode, &create_disposition,
3075 &create_options, NULL)) {
3076 goto try_openx;
3077 }
3078
3079 status = cli_ntcreate(cli,
3080 fname,
3081 0,
3082 access_mask,
3083 0,
3084 share_mode,
3085 create_disposition,
3086 create_options,
3087 0,
3088 pfnum,
3089 &cr);
3090
3091 /* Try and cope will all varients of "we don't do this call"
3092 and fall back to openX. */
3093
3094 if (NT_STATUS_EQUAL(status,NT_STATUS_NOT_IMPLEMENTED) ||
3095 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_INFO_CLASS) ||
3096 NT_STATUS_EQUAL(status,NT_STATUS_PROCEDURE_NOT_FOUND) ||
3097 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_LEVEL) ||
3098 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_PARAMETER) ||
3099 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_DEVICE_REQUEST) ||
3100 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_DEVICE_STATE) ||
3101 NT_STATUS_EQUAL(status,NT_STATUS_CTL_FILE_NOT_SUPPORTED) ||
3102 NT_STATUS_EQUAL(status,NT_STATUS_UNSUCCESSFUL)) {
3103 goto try_openx;
3104 }
3105
3106 if (NT_STATUS_IS_OK(status) &&
3107 (create_options & FILE_NON_DIRECTORY_FILE) &&
3108 (cr.file_attributes & FILE_ATTRIBUTE_DIRECTORY))
3109 {
3110 /*
3111 * Some (broken) servers return a valid handle
3112 * for directories even if FILE_NON_DIRECTORY_FILE
3113 * is set. Just close the handle and set the
3114 * error explicitly to NT_STATUS_FILE_IS_A_DIRECTORY.
3115 */
3116 status = cli_close(cli, *pfnum);
3117 if (!NT_STATUS_IS_OK(status)) {
3118 return status;
3119 }
3120 status = NT_STATUS_FILE_IS_A_DIRECTORY;
3121 /* Set this so libsmbclient can retrieve it. */
3122 cli->raw_status = status;
3123 }
3124
3125 return status;
3126
3127 try_openx:
3128
3129 return cli_openx(cli, fname, flags, share_mode_in, pfnum);
3130 }
3131
3132 /****************************************************************************
3133 Close a file.
3134 ****************************************************************************/
3135
3136 struct cli_smb1_close_state {
3137 uint16_t vwv[3];
3138 };
3139
3140 static void cli_smb1_close_done(struct tevent_req *subreq);
3141
cli_smb1_close_create(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,struct tevent_req ** psubreq)3142 struct tevent_req *cli_smb1_close_create(TALLOC_CTX *mem_ctx,
3143 struct tevent_context *ev,
3144 struct cli_state *cli,
3145 uint16_t fnum,
3146 struct tevent_req **psubreq)
3147 {
3148 struct tevent_req *req, *subreq;
3149 struct cli_smb1_close_state *state;
3150
3151 req = tevent_req_create(mem_ctx, &state, struct cli_smb1_close_state);
3152 if (req == NULL) {
3153 return NULL;
3154 }
3155
3156 SSVAL(state->vwv+0, 0, fnum);
3157 SIVALS(state->vwv+1, 0, -1);
3158
3159 subreq = cli_smb_req_create(state, ev, cli, SMBclose, 0, 0,
3160 3, state->vwv, 0, NULL);
3161 if (subreq == NULL) {
3162 TALLOC_FREE(req);
3163 return NULL;
3164 }
3165 tevent_req_set_callback(subreq, cli_smb1_close_done, req);
3166 *psubreq = subreq;
3167 return req;
3168 }
3169
cli_smb1_close_done(struct tevent_req * subreq)3170 static void cli_smb1_close_done(struct tevent_req *subreq)
3171 {
3172 struct tevent_req *req = tevent_req_callback_data(
3173 subreq, struct tevent_req);
3174 NTSTATUS status;
3175
3176 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
3177 TALLOC_FREE(subreq);
3178 if (tevent_req_nterror(req, status)) {
3179 return;
3180 }
3181 tevent_req_done(req);
3182 }
3183
3184 struct cli_close_state {
3185 int dummy;
3186 };
3187
3188 static void cli_close_done(struct tevent_req *subreq);
3189
cli_close_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum)3190 struct tevent_req *cli_close_send(TALLOC_CTX *mem_ctx,
3191 struct tevent_context *ev,
3192 struct cli_state *cli,
3193 uint16_t fnum)
3194 {
3195 struct tevent_req *req, *subreq;
3196 struct cli_close_state *state;
3197 NTSTATUS status;
3198
3199 req = tevent_req_create(mem_ctx, &state, struct cli_close_state);
3200 if (req == NULL) {
3201 return NULL;
3202 }
3203
3204 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
3205 subreq = cli_smb2_close_fnum_send(state,
3206 ev,
3207 cli,
3208 fnum);
3209 if (tevent_req_nomem(subreq, req)) {
3210 return tevent_req_post(req, ev);
3211 }
3212 } else {
3213 struct tevent_req *ch_req = NULL;
3214 subreq = cli_smb1_close_create(state, ev, cli, fnum, &ch_req);
3215 if (tevent_req_nomem(subreq, req)) {
3216 return tevent_req_post(req, ev);
3217 }
3218 status = smb1cli_req_chain_submit(&ch_req, 1);
3219 if (tevent_req_nterror(req, status)) {
3220 return tevent_req_post(req, ev);
3221 }
3222 }
3223
3224 tevent_req_set_callback(subreq, cli_close_done, req);
3225 return req;
3226 }
3227
cli_close_done(struct tevent_req * subreq)3228 static void cli_close_done(struct tevent_req *subreq)
3229 {
3230 struct tevent_req *req = tevent_req_callback_data(
3231 subreq, struct tevent_req);
3232 NTSTATUS status = NT_STATUS_OK;
3233 bool err = tevent_req_is_nterror(subreq, &status);
3234
3235 TALLOC_FREE(subreq);
3236 if (err) {
3237 tevent_req_nterror(req, status);
3238 return;
3239 }
3240 tevent_req_done(req);
3241 }
3242
cli_close_recv(struct tevent_req * req)3243 NTSTATUS cli_close_recv(struct tevent_req *req)
3244 {
3245 return tevent_req_simple_recv_ntstatus(req);
3246 }
3247
cli_close(struct cli_state * cli,uint16_t fnum)3248 NTSTATUS cli_close(struct cli_state *cli, uint16_t fnum)
3249 {
3250 TALLOC_CTX *frame = NULL;
3251 struct tevent_context *ev;
3252 struct tevent_req *req;
3253 NTSTATUS status = NT_STATUS_OK;
3254
3255 frame = talloc_stackframe();
3256
3257 if (smbXcli_conn_has_async_calls(cli->conn)) {
3258 /*
3259 * Can't use sync call while an async call is in flight
3260 */
3261 status = NT_STATUS_INVALID_PARAMETER;
3262 goto fail;
3263 }
3264
3265 ev = samba_tevent_context_init(frame);
3266 if (ev == NULL) {
3267 status = NT_STATUS_NO_MEMORY;
3268 goto fail;
3269 }
3270
3271 req = cli_close_send(frame, ev, cli, fnum);
3272 if (req == NULL) {
3273 status = NT_STATUS_NO_MEMORY;
3274 goto fail;
3275 }
3276
3277 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
3278 goto fail;
3279 }
3280
3281 status = cli_close_recv(req);
3282 fail:
3283 TALLOC_FREE(frame);
3284 return status;
3285 }
3286
3287 /****************************************************************************
3288 Truncate a file to a specified size
3289 ****************************************************************************/
3290
3291 struct ftrunc_state {
3292 uint16_t setup;
3293 uint8_t param[6];
3294 uint8_t data[8];
3295 };
3296
cli_ftruncate_done(struct tevent_req * subreq)3297 static void cli_ftruncate_done(struct tevent_req *subreq)
3298 {
3299 NTSTATUS status = cli_trans_recv(subreq, NULL, NULL, NULL, 0, NULL,
3300 NULL, 0, NULL, NULL, 0, NULL);
3301 tevent_req_simple_finish_ntstatus(subreq, status);
3302 }
3303
cli_ftruncate_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint64_t size)3304 struct tevent_req *cli_ftruncate_send(TALLOC_CTX *mem_ctx,
3305 struct tevent_context *ev,
3306 struct cli_state *cli,
3307 uint16_t fnum,
3308 uint64_t size)
3309 {
3310 struct tevent_req *req = NULL, *subreq = NULL;
3311 struct ftrunc_state *state = NULL;
3312
3313 req = tevent_req_create(mem_ctx, &state, struct ftrunc_state);
3314 if (req == NULL) {
3315 return NULL;
3316 }
3317
3318 /* Setup setup word. */
3319 SSVAL(&state->setup, 0, TRANSACT2_SETFILEINFO);
3320
3321 /* Setup param array. */
3322 SSVAL(state->param,0,fnum);
3323 SSVAL(state->param,2,SMB_SET_FILE_END_OF_FILE_INFO);
3324 SSVAL(state->param,4,0);
3325
3326 /* Setup data array. */
3327 SBVAL(state->data, 0, size);
3328
3329 subreq = cli_trans_send(state, /* mem ctx. */
3330 ev, /* event ctx. */
3331 cli, /* cli_state. */
3332 0, /* additional_flags2 */
3333 SMBtrans2, /* cmd. */
3334 NULL, /* pipe name. */
3335 -1, /* fid. */
3336 0, /* function. */
3337 0, /* flags. */
3338 &state->setup, /* setup. */
3339 1, /* num setup uint16_t words. */
3340 0, /* max returned setup. */
3341 state->param, /* param. */
3342 6, /* num param. */
3343 2, /* max returned param. */
3344 state->data, /* data. */
3345 8, /* num data. */
3346 0); /* max returned data. */
3347
3348 if (tevent_req_nomem(subreq, req)) {
3349 return tevent_req_post(req, ev);
3350 }
3351 tevent_req_set_callback(subreq, cli_ftruncate_done, req);
3352 return req;
3353 }
3354
cli_ftruncate_recv(struct tevent_req * req)3355 NTSTATUS cli_ftruncate_recv(struct tevent_req *req)
3356 {
3357 return tevent_req_simple_recv_ntstatus(req);
3358 }
3359
cli_ftruncate(struct cli_state * cli,uint16_t fnum,uint64_t size)3360 NTSTATUS cli_ftruncate(struct cli_state *cli, uint16_t fnum, uint64_t size)
3361 {
3362 TALLOC_CTX *frame = NULL;
3363 struct tevent_context *ev = NULL;
3364 struct tevent_req *req = NULL;
3365 NTSTATUS status = NT_STATUS_OK;
3366
3367 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
3368 return cli_smb2_ftruncate(cli, fnum, size);
3369 }
3370
3371 frame = talloc_stackframe();
3372
3373 if (smbXcli_conn_has_async_calls(cli->conn)) {
3374 /*
3375 * Can't use sync call while an async call is in flight
3376 */
3377 status = NT_STATUS_INVALID_PARAMETER;
3378 goto fail;
3379 }
3380
3381 ev = samba_tevent_context_init(frame);
3382 if (ev == NULL) {
3383 status = NT_STATUS_NO_MEMORY;
3384 goto fail;
3385 }
3386
3387 req = cli_ftruncate_send(frame,
3388 ev,
3389 cli,
3390 fnum,
3391 size);
3392 if (req == NULL) {
3393 status = NT_STATUS_NO_MEMORY;
3394 goto fail;
3395 }
3396
3397 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
3398 goto fail;
3399 }
3400
3401 status = cli_ftruncate_recv(req);
3402
3403 fail:
3404 TALLOC_FREE(frame);
3405 return status;
3406 }
3407
cli_lockingx_put_locks(uint8_t * buf,bool large,uint16_t num_locks,const struct smb1_lock_element * locks)3408 static uint8_t *cli_lockingx_put_locks(
3409 uint8_t *buf,
3410 bool large,
3411 uint16_t num_locks,
3412 const struct smb1_lock_element *locks)
3413 {
3414 uint16_t i;
3415
3416 for (i=0; i<num_locks; i++) {
3417 const struct smb1_lock_element *e = &locks[i];
3418 if (large) {
3419 SSVAL(buf, 0, e->pid);
3420 SSVAL(buf, 2, 0);
3421 SOFF_T_R(buf, 4, e->offset);
3422 SOFF_T_R(buf, 12, e->length);
3423 buf += 20;
3424 } else {
3425 SSVAL(buf, 0, e->pid);
3426 SIVAL(buf, 2, e->offset);
3427 SIVAL(buf, 6, e->length);
3428 buf += 10;
3429 }
3430 }
3431 return buf;
3432 }
3433
3434 struct cli_lockingx_state {
3435 uint16_t vwv[8];
3436 struct iovec bytes;
3437 struct tevent_req *subreq;
3438 };
3439
3440 static void cli_lockingx_done(struct tevent_req *subreq);
3441 static bool cli_lockingx_cancel(struct tevent_req *req);
3442
cli_lockingx_create(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint8_t typeoflock,uint8_t newoplocklevel,int32_t timeout,uint16_t num_unlocks,const struct smb1_lock_element * unlocks,uint16_t num_locks,const struct smb1_lock_element * locks,struct tevent_req ** psmbreq)3443 struct tevent_req *cli_lockingx_create(
3444 TALLOC_CTX *mem_ctx,
3445 struct tevent_context *ev,
3446 struct cli_state *cli,
3447 uint16_t fnum,
3448 uint8_t typeoflock,
3449 uint8_t newoplocklevel,
3450 int32_t timeout,
3451 uint16_t num_unlocks,
3452 const struct smb1_lock_element *unlocks,
3453 uint16_t num_locks,
3454 const struct smb1_lock_element *locks,
3455 struct tevent_req **psmbreq)
3456 {
3457 struct tevent_req *req = NULL, *subreq = NULL;
3458 struct cli_lockingx_state *state = NULL;
3459 uint16_t *vwv;
3460 uint8_t *p;
3461 const bool large = (typeoflock & LOCKING_ANDX_LARGE_FILES);
3462 const size_t element_len = large ? 20 : 10;
3463
3464 /* uint16->size_t, no overflow */
3465 const size_t num_elements = (size_t)num_locks + (size_t)num_unlocks;
3466
3467 /* at most 20*2*65535 = 2621400, no overflow */
3468 const size_t num_bytes = num_elements * element_len;
3469
3470 req = tevent_req_create(mem_ctx, &state, struct cli_lockingx_state);
3471 if (req == NULL) {
3472 return NULL;
3473 }
3474 vwv = state->vwv;
3475
3476 SCVAL(vwv + 0, 0, 0xFF);
3477 SCVAL(vwv + 0, 1, 0);
3478 SSVAL(vwv + 1, 0, 0);
3479 SSVAL(vwv + 2, 0, fnum);
3480 SCVAL(vwv + 3, 0, typeoflock);
3481 SCVAL(vwv + 3, 1, newoplocklevel);
3482 SIVALS(vwv + 4, 0, timeout);
3483 SSVAL(vwv + 6, 0, num_unlocks);
3484 SSVAL(vwv + 7, 0, num_locks);
3485
3486 state->bytes.iov_len = num_bytes;
3487 state->bytes.iov_base = talloc_array(state, uint8_t, num_bytes);
3488 if (tevent_req_nomem(state->bytes.iov_base, req)) {
3489 return tevent_req_post(req, ev);
3490 }
3491
3492 p = cli_lockingx_put_locks(
3493 state->bytes.iov_base, large, num_unlocks, unlocks);
3494 cli_lockingx_put_locks(p, large, num_locks, locks);
3495
3496 subreq = cli_smb_req_create(
3497 state, ev, cli, SMBlockingX, 0, 0, 8, vwv, 1, &state->bytes);
3498 if (tevent_req_nomem(subreq, req)) {
3499 return tevent_req_post(req, ev);
3500 }
3501 tevent_req_set_callback(subreq, cli_lockingx_done, req);
3502 *psmbreq = subreq;
3503 return req;
3504 }
3505
cli_lockingx_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint8_t typeoflock,uint8_t newoplocklevel,int32_t timeout,uint16_t num_unlocks,const struct smb1_lock_element * unlocks,uint16_t num_locks,const struct smb1_lock_element * locks)3506 struct tevent_req *cli_lockingx_send(
3507 TALLOC_CTX *mem_ctx,
3508 struct tevent_context *ev,
3509 struct cli_state *cli,
3510 uint16_t fnum,
3511 uint8_t typeoflock,
3512 uint8_t newoplocklevel,
3513 int32_t timeout,
3514 uint16_t num_unlocks,
3515 const struct smb1_lock_element *unlocks,
3516 uint16_t num_locks,
3517 const struct smb1_lock_element *locks)
3518 {
3519 struct tevent_req *req = NULL, *subreq = NULL;
3520 struct cli_lockingx_state *state = NULL;
3521 NTSTATUS status;
3522
3523 req = cli_lockingx_create(
3524 mem_ctx,
3525 ev,
3526 cli,
3527 fnum,
3528 typeoflock,
3529 newoplocklevel,
3530 timeout,
3531 num_unlocks,
3532 unlocks,
3533 num_locks,
3534 locks,
3535 &subreq);
3536 if (req == NULL) {
3537 return NULL;
3538 }
3539 state = tevent_req_data(req, struct cli_lockingx_state);
3540 state->subreq = subreq;
3541
3542 status = smb1cli_req_chain_submit(&subreq, 1);
3543 if (tevent_req_nterror(req, status)) {
3544 return tevent_req_post(req, ev);
3545 }
3546 tevent_req_set_cancel_fn(req, cli_lockingx_cancel);
3547 return req;
3548 }
3549
cli_lockingx_done(struct tevent_req * subreq)3550 static void cli_lockingx_done(struct tevent_req *subreq)
3551 {
3552 NTSTATUS status = cli_smb_recv(
3553 subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
3554 tevent_req_simple_finish_ntstatus(subreq, status);
3555 }
3556
cli_lockingx_cancel(struct tevent_req * req)3557 static bool cli_lockingx_cancel(struct tevent_req *req)
3558 {
3559 struct cli_lockingx_state *state = tevent_req_data(
3560 req, struct cli_lockingx_state);
3561 if (state->subreq == NULL) {
3562 return false;
3563 }
3564 return tevent_req_cancel(state->subreq);
3565 }
3566
cli_lockingx_recv(struct tevent_req * req)3567 NTSTATUS cli_lockingx_recv(struct tevent_req *req)
3568 {
3569 return tevent_req_simple_recv_ntstatus(req);
3570 }
3571
cli_lockingx(struct cli_state * cli,uint16_t fnum,uint8_t typeoflock,uint8_t newoplocklevel,int32_t timeout,uint16_t num_unlocks,const struct smb1_lock_element * unlocks,uint16_t num_locks,const struct smb1_lock_element * locks)3572 NTSTATUS cli_lockingx(
3573 struct cli_state *cli,
3574 uint16_t fnum,
3575 uint8_t typeoflock,
3576 uint8_t newoplocklevel,
3577 int32_t timeout,
3578 uint16_t num_unlocks,
3579 const struct smb1_lock_element *unlocks,
3580 uint16_t num_locks,
3581 const struct smb1_lock_element *locks)
3582 {
3583 TALLOC_CTX *frame = talloc_stackframe();
3584 struct tevent_context *ev = NULL;
3585 struct tevent_req *req = NULL;
3586 NTSTATUS status = NT_STATUS_NO_MEMORY;
3587 unsigned int set_timeout = 0;
3588 unsigned int saved_timeout = 0;
3589
3590 if (smbXcli_conn_has_async_calls(cli->conn)) {
3591 return NT_STATUS_INVALID_PARAMETER;
3592 }
3593 ev = samba_tevent_context_init(frame);
3594 if (ev == NULL) {
3595 goto fail;
3596 }
3597
3598 if (timeout != 0) {
3599 if (timeout == -1) {
3600 set_timeout = 0x7FFFFFFF;
3601 } else {
3602 set_timeout = timeout + 2*1000;
3603 }
3604 saved_timeout = cli_set_timeout(cli, set_timeout);
3605 }
3606
3607 req = cli_lockingx_send(
3608 frame,
3609 ev,
3610 cli,
3611 fnum,
3612 typeoflock,
3613 newoplocklevel,
3614 timeout,
3615 num_unlocks,
3616 unlocks,
3617 num_locks,
3618 locks);
3619 if (req == NULL) {
3620 goto fail;
3621 }
3622 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
3623 goto fail;
3624 }
3625 status = cli_lockingx_recv(req);
3626
3627 if (saved_timeout != 0) {
3628 cli_set_timeout(cli, saved_timeout);
3629 }
3630 fail:
3631 TALLOC_FREE(frame);
3632 return status;
3633 }
3634
3635 /****************************************************************************
3636 send a lock with a specified locktype
3637 this is used for testing LOCKING_ANDX_CANCEL_LOCK
3638 ****************************************************************************/
3639
cli_locktype(struct cli_state * cli,uint16_t fnum,uint32_t offset,uint32_t len,int timeout,unsigned char locktype)3640 NTSTATUS cli_locktype(struct cli_state *cli, uint16_t fnum,
3641 uint32_t offset, uint32_t len,
3642 int timeout, unsigned char locktype)
3643 {
3644 struct smb1_lock_element lck = {
3645 .pid = cli_getpid(cli),
3646 .offset = offset,
3647 .length = len,
3648 };
3649 NTSTATUS status;
3650
3651 status = cli_lockingx(
3652 cli, /* cli */
3653 fnum, /* fnum */
3654 locktype, /* typeoflock */
3655 0, /* newoplocklevel */
3656 timeout, /* timeout */
3657 0, /* num_unlocks */
3658 NULL, /* unlocks */
3659 1, /* num_locks */
3660 &lck); /* locks */
3661 return status;
3662 }
3663
3664 /****************************************************************************
3665 Lock a file.
3666 note that timeout is in units of 2 milliseconds
3667 ****************************************************************************/
3668
cli_lock32(struct cli_state * cli,uint16_t fnum,uint32_t offset,uint32_t len,int timeout,enum brl_type lock_type)3669 NTSTATUS cli_lock32(struct cli_state *cli, uint16_t fnum,
3670 uint32_t offset, uint32_t len, int timeout,
3671 enum brl_type lock_type)
3672 {
3673 NTSTATUS status;
3674
3675 status = cli_locktype(cli, fnum, offset, len, timeout,
3676 (lock_type == READ_LOCK? 1 : 0));
3677 return status;
3678 }
3679
3680 /****************************************************************************
3681 Unlock a file.
3682 ****************************************************************************/
3683
3684 struct cli_unlock_state {
3685 struct smb1_lock_element lck;
3686 };
3687
3688 static void cli_unlock_done(struct tevent_req *subreq);
3689
cli_unlock_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint64_t offset,uint64_t len)3690 struct tevent_req *cli_unlock_send(TALLOC_CTX *mem_ctx,
3691 struct tevent_context *ev,
3692 struct cli_state *cli,
3693 uint16_t fnum,
3694 uint64_t offset,
3695 uint64_t len)
3696
3697 {
3698 struct tevent_req *req = NULL, *subreq = NULL;
3699 struct cli_unlock_state *state = NULL;
3700
3701 req = tevent_req_create(mem_ctx, &state, struct cli_unlock_state);
3702 if (req == NULL) {
3703 return NULL;
3704 }
3705 state->lck = (struct smb1_lock_element) {
3706 .pid = cli_getpid(cli),
3707 .offset = offset,
3708 .length = len,
3709 };
3710
3711 subreq = cli_lockingx_send(
3712 state, /* mem_ctx */
3713 ev, /* tevent_context */
3714 cli, /* cli */
3715 fnum, /* fnum */
3716 0, /* typeoflock */
3717 0, /* newoplocklevel */
3718 0, /* timeout */
3719 1, /* num_unlocks */
3720 &state->lck, /* unlocks */
3721 0, /* num_locks */
3722 NULL); /* locks */
3723 if (tevent_req_nomem(subreq, req)) {
3724 return tevent_req_post(req, ev);
3725 }
3726 tevent_req_set_callback(subreq, cli_unlock_done, req);
3727 return req;
3728 }
3729
cli_unlock_done(struct tevent_req * subreq)3730 static void cli_unlock_done(struct tevent_req *subreq)
3731 {
3732 NTSTATUS status = cli_lockingx_recv(subreq);
3733 tevent_req_simple_finish_ntstatus(subreq, status);
3734 }
3735
cli_unlock_recv(struct tevent_req * req)3736 NTSTATUS cli_unlock_recv(struct tevent_req *req)
3737 {
3738 return tevent_req_simple_recv_ntstatus(req);
3739 }
3740
cli_unlock(struct cli_state * cli,uint16_t fnum,uint32_t offset,uint32_t len)3741 NTSTATUS cli_unlock(struct cli_state *cli,
3742 uint16_t fnum,
3743 uint32_t offset,
3744 uint32_t len)
3745 {
3746 TALLOC_CTX *frame = talloc_stackframe();
3747 struct tevent_context *ev;
3748 struct tevent_req *req;
3749 NTSTATUS status = NT_STATUS_OK;
3750
3751 if (smbXcli_conn_has_async_calls(cli->conn)) {
3752 /*
3753 * Can't use sync call while an async call is in flight
3754 */
3755 status = NT_STATUS_INVALID_PARAMETER;
3756 goto fail;
3757 }
3758
3759 ev = samba_tevent_context_init(frame);
3760 if (ev == NULL) {
3761 status = NT_STATUS_NO_MEMORY;
3762 goto fail;
3763 }
3764
3765 req = cli_unlock_send(frame, ev, cli,
3766 fnum, offset, len);
3767 if (req == NULL) {
3768 status = NT_STATUS_NO_MEMORY;
3769 goto fail;
3770 }
3771
3772 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
3773 goto fail;
3774 }
3775
3776 status = cli_unlock_recv(req);
3777
3778 fail:
3779 TALLOC_FREE(frame);
3780 return status;
3781 }
3782
3783 /****************************************************************************
3784 Get/unlock a POSIX lock on a file - internal function.
3785 ****************************************************************************/
3786
3787 struct posix_lock_state {
3788 uint16_t setup;
3789 uint8_t param[4];
3790 uint8_t data[POSIX_LOCK_DATA_SIZE];
3791 };
3792
cli_posix_unlock_internal_done(struct tevent_req * subreq)3793 static void cli_posix_unlock_internal_done(struct tevent_req *subreq)
3794 {
3795 NTSTATUS status = cli_trans_recv(subreq, NULL, NULL, NULL, 0, NULL,
3796 NULL, 0, NULL, NULL, 0, NULL);
3797 tevent_req_simple_finish_ntstatus(subreq, status);
3798 }
3799
cli_posix_lock_internal_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint64_t offset,uint64_t len,bool wait_lock,enum brl_type lock_type)3800 static struct tevent_req *cli_posix_lock_internal_send(TALLOC_CTX *mem_ctx,
3801 struct tevent_context *ev,
3802 struct cli_state *cli,
3803 uint16_t fnum,
3804 uint64_t offset,
3805 uint64_t len,
3806 bool wait_lock,
3807 enum brl_type lock_type)
3808 {
3809 struct tevent_req *req = NULL, *subreq = NULL;
3810 struct posix_lock_state *state = NULL;
3811
3812 req = tevent_req_create(mem_ctx, &state, struct posix_lock_state);
3813 if (req == NULL) {
3814 return NULL;
3815 }
3816
3817 /* Setup setup word. */
3818 SSVAL(&state->setup, 0, TRANSACT2_SETFILEINFO);
3819
3820 /* Setup param array. */
3821 SSVAL(&state->param, 0, fnum);
3822 SSVAL(&state->param, 2, SMB_SET_POSIX_LOCK);
3823
3824 /* Setup data array. */
3825 switch (lock_type) {
3826 case READ_LOCK:
3827 SSVAL(&state->data, POSIX_LOCK_TYPE_OFFSET,
3828 POSIX_LOCK_TYPE_READ);
3829 break;
3830 case WRITE_LOCK:
3831 SSVAL(&state->data, POSIX_LOCK_TYPE_OFFSET,
3832 POSIX_LOCK_TYPE_WRITE);
3833 break;
3834 case UNLOCK_LOCK:
3835 SSVAL(&state->data, POSIX_LOCK_TYPE_OFFSET,
3836 POSIX_LOCK_TYPE_UNLOCK);
3837 break;
3838 default:
3839 return NULL;
3840 }
3841
3842 if (wait_lock) {
3843 SSVAL(&state->data, POSIX_LOCK_FLAGS_OFFSET,
3844 POSIX_LOCK_FLAG_WAIT);
3845 } else {
3846 SSVAL(state->data, POSIX_LOCK_FLAGS_OFFSET,
3847 POSIX_LOCK_FLAG_NOWAIT);
3848 }
3849
3850 SIVAL(&state->data, POSIX_LOCK_PID_OFFSET, cli_getpid(cli));
3851 SOFF_T(&state->data, POSIX_LOCK_START_OFFSET, offset);
3852 SOFF_T(&state->data, POSIX_LOCK_LEN_OFFSET, len);
3853
3854 subreq = cli_trans_send(state, /* mem ctx. */
3855 ev, /* event ctx. */
3856 cli, /* cli_state. */
3857 0, /* additional_flags2 */
3858 SMBtrans2, /* cmd. */
3859 NULL, /* pipe name. */
3860 -1, /* fid. */
3861 0, /* function. */
3862 0, /* flags. */
3863 &state->setup, /* setup. */
3864 1, /* num setup uint16_t words. */
3865 0, /* max returned setup. */
3866 state->param, /* param. */
3867 4, /* num param. */
3868 2, /* max returned param. */
3869 state->data, /* data. */
3870 POSIX_LOCK_DATA_SIZE, /* num data. */
3871 0); /* max returned data. */
3872
3873 if (tevent_req_nomem(subreq, req)) {
3874 return tevent_req_post(req, ev);
3875 }
3876 tevent_req_set_callback(subreq, cli_posix_unlock_internal_done, req);
3877 return req;
3878 }
3879
3880 /****************************************************************************
3881 POSIX Lock a file.
3882 ****************************************************************************/
3883
cli_posix_lock_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint64_t offset,uint64_t len,bool wait_lock,enum brl_type lock_type)3884 struct tevent_req *cli_posix_lock_send(TALLOC_CTX *mem_ctx,
3885 struct tevent_context *ev,
3886 struct cli_state *cli,
3887 uint16_t fnum,
3888 uint64_t offset,
3889 uint64_t len,
3890 bool wait_lock,
3891 enum brl_type lock_type)
3892 {
3893 return cli_posix_lock_internal_send(mem_ctx, ev, cli, fnum, offset, len,
3894 wait_lock, lock_type);
3895 }
3896
cli_posix_lock_recv(struct tevent_req * req)3897 NTSTATUS cli_posix_lock_recv(struct tevent_req *req)
3898 {
3899 return tevent_req_simple_recv_ntstatus(req);
3900 }
3901
cli_posix_lock(struct cli_state * cli,uint16_t fnum,uint64_t offset,uint64_t len,bool wait_lock,enum brl_type lock_type)3902 NTSTATUS cli_posix_lock(struct cli_state *cli, uint16_t fnum,
3903 uint64_t offset, uint64_t len,
3904 bool wait_lock, enum brl_type lock_type)
3905 {
3906 TALLOC_CTX *frame = talloc_stackframe();
3907 struct tevent_context *ev = NULL;
3908 struct tevent_req *req = NULL;
3909 NTSTATUS status = NT_STATUS_OK;
3910
3911 if (smbXcli_conn_has_async_calls(cli->conn)) {
3912 /*
3913 * Can't use sync call while an async call is in flight
3914 */
3915 status = NT_STATUS_INVALID_PARAMETER;
3916 goto fail;
3917 }
3918
3919 if (lock_type != READ_LOCK && lock_type != WRITE_LOCK) {
3920 status = NT_STATUS_INVALID_PARAMETER;
3921 goto fail;
3922 }
3923
3924 ev = samba_tevent_context_init(frame);
3925 if (ev == NULL) {
3926 status = NT_STATUS_NO_MEMORY;
3927 goto fail;
3928 }
3929
3930 req = cli_posix_lock_send(frame,
3931 ev,
3932 cli,
3933 fnum,
3934 offset,
3935 len,
3936 wait_lock,
3937 lock_type);
3938 if (req == NULL) {
3939 status = NT_STATUS_NO_MEMORY;
3940 goto fail;
3941 }
3942
3943 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
3944 goto fail;
3945 }
3946
3947 status = cli_posix_lock_recv(req);
3948
3949 fail:
3950 TALLOC_FREE(frame);
3951 return status;
3952 }
3953
3954 /****************************************************************************
3955 POSIX Unlock a file.
3956 ****************************************************************************/
3957
cli_posix_unlock_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint64_t offset,uint64_t len)3958 struct tevent_req *cli_posix_unlock_send(TALLOC_CTX *mem_ctx,
3959 struct tevent_context *ev,
3960 struct cli_state *cli,
3961 uint16_t fnum,
3962 uint64_t offset,
3963 uint64_t len)
3964 {
3965 return cli_posix_lock_internal_send(mem_ctx, ev, cli, fnum, offset, len,
3966 false, UNLOCK_LOCK);
3967 }
3968
cli_posix_unlock_recv(struct tevent_req * req)3969 NTSTATUS cli_posix_unlock_recv(struct tevent_req *req)
3970 {
3971 return tevent_req_simple_recv_ntstatus(req);
3972 }
3973
cli_posix_unlock(struct cli_state * cli,uint16_t fnum,uint64_t offset,uint64_t len)3974 NTSTATUS cli_posix_unlock(struct cli_state *cli, uint16_t fnum, uint64_t offset, uint64_t len)
3975 {
3976 TALLOC_CTX *frame = talloc_stackframe();
3977 struct tevent_context *ev = NULL;
3978 struct tevent_req *req = NULL;
3979 NTSTATUS status = NT_STATUS_OK;
3980
3981 if (smbXcli_conn_has_async_calls(cli->conn)) {
3982 /*
3983 * Can't use sync call while an async call is in flight
3984 */
3985 status = NT_STATUS_INVALID_PARAMETER;
3986 goto fail;
3987 }
3988
3989 ev = samba_tevent_context_init(frame);
3990 if (ev == NULL) {
3991 status = NT_STATUS_NO_MEMORY;
3992 goto fail;
3993 }
3994
3995 req = cli_posix_unlock_send(frame,
3996 ev,
3997 cli,
3998 fnum,
3999 offset,
4000 len);
4001 if (req == NULL) {
4002 status = NT_STATUS_NO_MEMORY;
4003 goto fail;
4004 }
4005
4006 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
4007 goto fail;
4008 }
4009
4010 status = cli_posix_unlock_recv(req);
4011
4012 fail:
4013 TALLOC_FREE(frame);
4014 return status;
4015 }
4016
4017 /****************************************************************************
4018 Do a SMBgetattrE call.
4019 ****************************************************************************/
4020
4021 static void cli_getattrE_done(struct tevent_req *subreq);
4022
4023 struct cli_getattrE_state {
4024 uint16_t vwv[1];
4025 int zone_offset;
4026 uint16_t attr;
4027 off_t size;
4028 time_t change_time;
4029 time_t access_time;
4030 time_t write_time;
4031 };
4032
cli_getattrE_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum)4033 struct tevent_req *cli_getattrE_send(TALLOC_CTX *mem_ctx,
4034 struct tevent_context *ev,
4035 struct cli_state *cli,
4036 uint16_t fnum)
4037 {
4038 struct tevent_req *req = NULL, *subreq = NULL;
4039 struct cli_getattrE_state *state = NULL;
4040 uint8_t additional_flags = 0;
4041
4042 req = tevent_req_create(mem_ctx, &state, struct cli_getattrE_state);
4043 if (req == NULL) {
4044 return NULL;
4045 }
4046
4047 state->zone_offset = smb1cli_conn_server_time_zone(cli->conn);
4048 SSVAL(state->vwv+0,0,fnum);
4049
4050 subreq = cli_smb_send(state, ev, cli, SMBgetattrE, additional_flags, 0,
4051 1, state->vwv, 0, NULL);
4052 if (tevent_req_nomem(subreq, req)) {
4053 return tevent_req_post(req, ev);
4054 }
4055 tevent_req_set_callback(subreq, cli_getattrE_done, req);
4056 return req;
4057 }
4058
cli_getattrE_done(struct tevent_req * subreq)4059 static void cli_getattrE_done(struct tevent_req *subreq)
4060 {
4061 struct tevent_req *req = tevent_req_callback_data(
4062 subreq, struct tevent_req);
4063 struct cli_getattrE_state *state = tevent_req_data(
4064 req, struct cli_getattrE_state);
4065 uint8_t wct;
4066 uint16_t *vwv = NULL;
4067 NTSTATUS status;
4068
4069 status = cli_smb_recv(subreq, state, NULL, 11, &wct, &vwv,
4070 NULL, NULL);
4071 TALLOC_FREE(subreq);
4072 if (tevent_req_nterror(req, status)) {
4073 return;
4074 }
4075
4076 state->size = (off_t)IVAL(vwv+6,0);
4077 state->attr = SVAL(vwv+10,0);
4078 state->change_time = make_unix_date2(vwv+0, state->zone_offset);
4079 state->access_time = make_unix_date2(vwv+2, state->zone_offset);
4080 state->write_time = make_unix_date2(vwv+4, state->zone_offset);
4081
4082 tevent_req_done(req);
4083 }
4084
cli_getattrE_recv(struct tevent_req * req,uint16_t * attr,off_t * size,time_t * change_time,time_t * access_time,time_t * write_time)4085 NTSTATUS cli_getattrE_recv(struct tevent_req *req,
4086 uint16_t *attr,
4087 off_t *size,
4088 time_t *change_time,
4089 time_t *access_time,
4090 time_t *write_time)
4091 {
4092 struct cli_getattrE_state *state = tevent_req_data(
4093 req, struct cli_getattrE_state);
4094 NTSTATUS status;
4095
4096 if (tevent_req_is_nterror(req, &status)) {
4097 return status;
4098 }
4099 if (attr) {
4100 *attr = state->attr;
4101 }
4102 if (size) {
4103 *size = state->size;
4104 }
4105 if (change_time) {
4106 *change_time = state->change_time;
4107 }
4108 if (access_time) {
4109 *access_time = state->access_time;
4110 }
4111 if (write_time) {
4112 *write_time = state->write_time;
4113 }
4114 return NT_STATUS_OK;
4115 }
4116
cli_getattrE(struct cli_state * cli,uint16_t fnum,uint16_t * attr,off_t * size,time_t * change_time,time_t * access_time,time_t * write_time)4117 NTSTATUS cli_getattrE(struct cli_state *cli,
4118 uint16_t fnum,
4119 uint16_t *attr,
4120 off_t *size,
4121 time_t *change_time,
4122 time_t *access_time,
4123 time_t *write_time)
4124 {
4125 TALLOC_CTX *frame = NULL;
4126 struct tevent_context *ev = NULL;
4127 struct tevent_req *req = NULL;
4128 NTSTATUS status = NT_STATUS_OK;
4129
4130 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
4131 return cli_smb2_getattrE(cli,
4132 fnum,
4133 attr,
4134 size,
4135 change_time,
4136 access_time,
4137 write_time);
4138 }
4139
4140 frame = talloc_stackframe();
4141
4142 if (smbXcli_conn_has_async_calls(cli->conn)) {
4143 /*
4144 * Can't use sync call while an async call is in flight
4145 */
4146 status = NT_STATUS_INVALID_PARAMETER;
4147 goto fail;
4148 }
4149
4150 ev = samba_tevent_context_init(frame);
4151 if (ev == NULL) {
4152 status = NT_STATUS_NO_MEMORY;
4153 goto fail;
4154 }
4155
4156 req = cli_getattrE_send(frame, ev, cli, fnum);
4157 if (req == NULL) {
4158 status = NT_STATUS_NO_MEMORY;
4159 goto fail;
4160 }
4161
4162 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
4163 goto fail;
4164 }
4165
4166 status = cli_getattrE_recv(req,
4167 attr,
4168 size,
4169 change_time,
4170 access_time,
4171 write_time);
4172
4173 fail:
4174 TALLOC_FREE(frame);
4175 return status;
4176 }
4177
4178 /****************************************************************************
4179 Do a SMBgetatr call
4180 ****************************************************************************/
4181
4182 static void cli_getatr_done(struct tevent_req *subreq);
4183
4184 struct cli_getatr_state {
4185 int zone_offset;
4186 uint16_t attr;
4187 off_t size;
4188 time_t write_time;
4189 };
4190
cli_getatr_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname)4191 struct tevent_req *cli_getatr_send(TALLOC_CTX *mem_ctx,
4192 struct tevent_context *ev,
4193 struct cli_state *cli,
4194 const char *fname)
4195 {
4196 struct tevent_req *req = NULL, *subreq = NULL;
4197 struct cli_getatr_state *state = NULL;
4198 uint8_t additional_flags = 0;
4199 uint16_t additional_flags2 = 0;
4200 uint8_t *bytes = NULL;
4201
4202 req = tevent_req_create(mem_ctx, &state, struct cli_getatr_state);
4203 if (req == NULL) {
4204 return NULL;
4205 }
4206
4207 state->zone_offset = smb1cli_conn_server_time_zone(cli->conn);
4208
4209 bytes = talloc_array(state, uint8_t, 1);
4210 if (tevent_req_nomem(bytes, req)) {
4211 return tevent_req_post(req, ev);
4212 }
4213 bytes[0] = 4;
4214 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname,
4215 strlen(fname)+1, NULL);
4216
4217 if (tevent_req_nomem(bytes, req)) {
4218 return tevent_req_post(req, ev);
4219 }
4220
4221 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL)) {
4222 additional_flags2 = FLAGS2_REPARSE_PATH;
4223 }
4224
4225 subreq = cli_smb_send(state, ev, cli, SMBgetatr, additional_flags,
4226 additional_flags2,
4227 0, NULL, talloc_get_size(bytes), bytes);
4228 if (tevent_req_nomem(subreq, req)) {
4229 return tevent_req_post(req, ev);
4230 }
4231 tevent_req_set_callback(subreq, cli_getatr_done, req);
4232 return req;
4233 }
4234
cli_getatr_done(struct tevent_req * subreq)4235 static void cli_getatr_done(struct tevent_req *subreq)
4236 {
4237 struct tevent_req *req = tevent_req_callback_data(
4238 subreq, struct tevent_req);
4239 struct cli_getatr_state *state = tevent_req_data(
4240 req, struct cli_getatr_state);
4241 uint8_t wct;
4242 uint16_t *vwv = NULL;
4243 NTSTATUS status;
4244
4245 status = cli_smb_recv(subreq, state, NULL, 4, &wct, &vwv, NULL,
4246 NULL);
4247 TALLOC_FREE(subreq);
4248 if (tevent_req_nterror(req, status)) {
4249 return;
4250 }
4251
4252 state->attr = SVAL(vwv+0,0);
4253 state->size = (off_t)IVAL(vwv+3,0);
4254 state->write_time = make_unix_date3(vwv+1, state->zone_offset);
4255
4256 tevent_req_done(req);
4257 }
4258
cli_getatr_recv(struct tevent_req * req,uint16_t * attr,off_t * size,time_t * write_time)4259 NTSTATUS cli_getatr_recv(struct tevent_req *req,
4260 uint16_t *attr,
4261 off_t *size,
4262 time_t *write_time)
4263 {
4264 struct cli_getatr_state *state = tevent_req_data(
4265 req, struct cli_getatr_state);
4266 NTSTATUS status;
4267
4268 if (tevent_req_is_nterror(req, &status)) {
4269 return status;
4270 }
4271 if (attr) {
4272 *attr = state->attr;
4273 }
4274 if (size) {
4275 *size = state->size;
4276 }
4277 if (write_time) {
4278 *write_time = state->write_time;
4279 }
4280 return NT_STATUS_OK;
4281 }
4282
cli_getatr(struct cli_state * cli,const char * fname,uint16_t * attr,off_t * size,time_t * write_time)4283 NTSTATUS cli_getatr(struct cli_state *cli,
4284 const char *fname,
4285 uint16_t *attr,
4286 off_t *size,
4287 time_t *write_time)
4288 {
4289 TALLOC_CTX *frame = NULL;
4290 struct tevent_context *ev = NULL;
4291 struct tevent_req *req = NULL;
4292 NTSTATUS status = NT_STATUS_OK;
4293
4294 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
4295 return cli_smb2_getatr(cli,
4296 fname,
4297 attr,
4298 size,
4299 write_time);
4300 }
4301
4302 frame = talloc_stackframe();
4303
4304 if (smbXcli_conn_has_async_calls(cli->conn)) {
4305 /*
4306 * Can't use sync call while an async call is in flight
4307 */
4308 status = NT_STATUS_INVALID_PARAMETER;
4309 goto fail;
4310 }
4311
4312 ev = samba_tevent_context_init(frame);
4313 if (ev == NULL) {
4314 status = NT_STATUS_NO_MEMORY;
4315 goto fail;
4316 }
4317
4318 req = cli_getatr_send(frame, ev, cli, fname);
4319 if (req == NULL) {
4320 status = NT_STATUS_NO_MEMORY;
4321 goto fail;
4322 }
4323
4324 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
4325 goto fail;
4326 }
4327
4328 status = cli_getatr_recv(req,
4329 attr,
4330 size,
4331 write_time);
4332
4333 fail:
4334 TALLOC_FREE(frame);
4335 return status;
4336 }
4337
4338 /****************************************************************************
4339 Do a SMBsetattrE call.
4340 ****************************************************************************/
4341
4342 static void cli_setattrE_done(struct tevent_req *subreq);
4343
4344 struct cli_setattrE_state {
4345 uint16_t vwv[7];
4346 };
4347
cli_setattrE_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,time_t change_time,time_t access_time,time_t write_time)4348 struct tevent_req *cli_setattrE_send(TALLOC_CTX *mem_ctx,
4349 struct tevent_context *ev,
4350 struct cli_state *cli,
4351 uint16_t fnum,
4352 time_t change_time,
4353 time_t access_time,
4354 time_t write_time)
4355 {
4356 struct tevent_req *req = NULL, *subreq = NULL;
4357 struct cli_setattrE_state *state = NULL;
4358 uint8_t additional_flags = 0;
4359
4360 req = tevent_req_create(mem_ctx, &state, struct cli_setattrE_state);
4361 if (req == NULL) {
4362 return NULL;
4363 }
4364
4365 SSVAL(state->vwv+0, 0, fnum);
4366 push_dos_date2((uint8_t *)&state->vwv[1], 0, change_time,
4367 smb1cli_conn_server_time_zone(cli->conn));
4368 push_dos_date2((uint8_t *)&state->vwv[3], 0, access_time,
4369 smb1cli_conn_server_time_zone(cli->conn));
4370 push_dos_date2((uint8_t *)&state->vwv[5], 0, write_time,
4371 smb1cli_conn_server_time_zone(cli->conn));
4372
4373 subreq = cli_smb_send(state, ev, cli, SMBsetattrE, additional_flags, 0,
4374 7, state->vwv, 0, NULL);
4375 if (tevent_req_nomem(subreq, req)) {
4376 return tevent_req_post(req, ev);
4377 }
4378 tevent_req_set_callback(subreq, cli_setattrE_done, req);
4379 return req;
4380 }
4381
cli_setattrE_done(struct tevent_req * subreq)4382 static void cli_setattrE_done(struct tevent_req *subreq)
4383 {
4384 struct tevent_req *req = tevent_req_callback_data(
4385 subreq, struct tevent_req);
4386 NTSTATUS status;
4387
4388 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
4389 TALLOC_FREE(subreq);
4390 if (tevent_req_nterror(req, status)) {
4391 return;
4392 }
4393 tevent_req_done(req);
4394 }
4395
cli_setattrE_recv(struct tevent_req * req)4396 NTSTATUS cli_setattrE_recv(struct tevent_req *req)
4397 {
4398 return tevent_req_simple_recv_ntstatus(req);
4399 }
4400
cli_setattrE(struct cli_state * cli,uint16_t fnum,time_t change_time,time_t access_time,time_t write_time)4401 NTSTATUS cli_setattrE(struct cli_state *cli,
4402 uint16_t fnum,
4403 time_t change_time,
4404 time_t access_time,
4405 time_t write_time)
4406 {
4407 TALLOC_CTX *frame = NULL;
4408 struct tevent_context *ev = NULL;
4409 struct tevent_req *req = NULL;
4410 NTSTATUS status = NT_STATUS_OK;
4411
4412 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
4413 return cli_smb2_setattrE(cli,
4414 fnum,
4415 change_time,
4416 access_time,
4417 write_time);
4418 }
4419
4420 frame = talloc_stackframe();
4421
4422 if (smbXcli_conn_has_async_calls(cli->conn)) {
4423 /*
4424 * Can't use sync call while an async call is in flight
4425 */
4426 status = NT_STATUS_INVALID_PARAMETER;
4427 goto fail;
4428 }
4429
4430 ev = samba_tevent_context_init(frame);
4431 if (ev == NULL) {
4432 status = NT_STATUS_NO_MEMORY;
4433 goto fail;
4434 }
4435
4436 req = cli_setattrE_send(frame, ev,
4437 cli,
4438 fnum,
4439 change_time,
4440 access_time,
4441 write_time);
4442
4443 if (req == NULL) {
4444 status = NT_STATUS_NO_MEMORY;
4445 goto fail;
4446 }
4447
4448 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
4449 goto fail;
4450 }
4451
4452 status = cli_setattrE_recv(req);
4453
4454 fail:
4455 TALLOC_FREE(frame);
4456 return status;
4457 }
4458
4459 /****************************************************************************
4460 Do a SMBsetatr call.
4461 ****************************************************************************/
4462
4463 static void cli_setatr_done(struct tevent_req *subreq);
4464
4465 struct cli_setatr_state {
4466 uint16_t vwv[8];
4467 };
4468
cli_setatr_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint16_t attr,time_t mtime)4469 struct tevent_req *cli_setatr_send(TALLOC_CTX *mem_ctx,
4470 struct tevent_context *ev,
4471 struct cli_state *cli,
4472 const char *fname,
4473 uint16_t attr,
4474 time_t mtime)
4475 {
4476 struct tevent_req *req = NULL, *subreq = NULL;
4477 struct cli_setatr_state *state = NULL;
4478 uint8_t additional_flags = 0;
4479 uint16_t additional_flags2 = 0;
4480 uint8_t *bytes = NULL;
4481
4482 req = tevent_req_create(mem_ctx, &state, struct cli_setatr_state);
4483 if (req == NULL) {
4484 return NULL;
4485 }
4486
4487 SSVAL(state->vwv+0, 0, attr);
4488 push_dos_date3((uint8_t *)&state->vwv[1], 0, mtime, smb1cli_conn_server_time_zone(cli->conn));
4489
4490 bytes = talloc_array(state, uint8_t, 1);
4491 if (tevent_req_nomem(bytes, req)) {
4492 return tevent_req_post(req, ev);
4493 }
4494 bytes[0] = 4;
4495 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname,
4496 strlen(fname)+1, NULL);
4497 if (tevent_req_nomem(bytes, req)) {
4498 return tevent_req_post(req, ev);
4499 }
4500 bytes = talloc_realloc(state, bytes, uint8_t,
4501 talloc_get_size(bytes)+1);
4502 if (tevent_req_nomem(bytes, req)) {
4503 return tevent_req_post(req, ev);
4504 }
4505
4506 bytes[talloc_get_size(bytes)-1] = 4;
4507 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), "",
4508 1, NULL);
4509 if (tevent_req_nomem(bytes, req)) {
4510 return tevent_req_post(req, ev);
4511 }
4512
4513 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL)) {
4514 additional_flags2 = FLAGS2_REPARSE_PATH;
4515 }
4516
4517 subreq = cli_smb_send(state, ev, cli, SMBsetatr, additional_flags,
4518 additional_flags2,
4519 8, state->vwv, talloc_get_size(bytes), bytes);
4520 if (tevent_req_nomem(subreq, req)) {
4521 return tevent_req_post(req, ev);
4522 }
4523 tevent_req_set_callback(subreq, cli_setatr_done, req);
4524 return req;
4525 }
4526
cli_setatr_done(struct tevent_req * subreq)4527 static void cli_setatr_done(struct tevent_req *subreq)
4528 {
4529 struct tevent_req *req = tevent_req_callback_data(
4530 subreq, struct tevent_req);
4531 NTSTATUS status;
4532
4533 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
4534 TALLOC_FREE(subreq);
4535 if (tevent_req_nterror(req, status)) {
4536 return;
4537 }
4538 tevent_req_done(req);
4539 }
4540
cli_setatr_recv(struct tevent_req * req)4541 NTSTATUS cli_setatr_recv(struct tevent_req *req)
4542 {
4543 return tevent_req_simple_recv_ntstatus(req);
4544 }
4545
cli_setatr(struct cli_state * cli,const char * fname,uint16_t attr,time_t mtime)4546 NTSTATUS cli_setatr(struct cli_state *cli,
4547 const char *fname,
4548 uint16_t attr,
4549 time_t mtime)
4550 {
4551 TALLOC_CTX *frame = NULL;
4552 struct tevent_context *ev = NULL;
4553 struct tevent_req *req = NULL;
4554 NTSTATUS status = NT_STATUS_OK;
4555
4556 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
4557 return cli_smb2_setatr(cli,
4558 fname,
4559 attr,
4560 mtime);
4561 }
4562
4563 frame = talloc_stackframe();
4564
4565 if (smbXcli_conn_has_async_calls(cli->conn)) {
4566 /*
4567 * Can't use sync call while an async call is in flight
4568 */
4569 status = NT_STATUS_INVALID_PARAMETER;
4570 goto fail;
4571 }
4572
4573 ev = samba_tevent_context_init(frame);
4574 if (ev == NULL) {
4575 status = NT_STATUS_NO_MEMORY;
4576 goto fail;
4577 }
4578
4579 req = cli_setatr_send(frame, ev, cli, fname, attr, mtime);
4580 if (req == NULL) {
4581 status = NT_STATUS_NO_MEMORY;
4582 goto fail;
4583 }
4584
4585 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
4586 goto fail;
4587 }
4588
4589 status = cli_setatr_recv(req);
4590
4591 fail:
4592 TALLOC_FREE(frame);
4593 return status;
4594 }
4595
4596 /****************************************************************************
4597 Check for existence of a dir.
4598 ****************************************************************************/
4599
4600 static void cli_chkpath_done(struct tevent_req *subreq);
4601
4602 struct cli_chkpath_state {
4603 int dummy;
4604 };
4605
cli_chkpath_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname)4606 struct tevent_req *cli_chkpath_send(TALLOC_CTX *mem_ctx,
4607 struct tevent_context *ev,
4608 struct cli_state *cli,
4609 const char *fname)
4610 {
4611 struct tevent_req *req = NULL, *subreq = NULL;
4612 struct cli_chkpath_state *state = NULL;
4613 uint8_t additional_flags = 0;
4614 uint16_t additional_flags2 = 0;
4615 uint8_t *bytes = NULL;
4616
4617 req = tevent_req_create(mem_ctx, &state, struct cli_chkpath_state);
4618 if (req == NULL) {
4619 return NULL;
4620 }
4621
4622 bytes = talloc_array(state, uint8_t, 1);
4623 if (tevent_req_nomem(bytes, req)) {
4624 return tevent_req_post(req, ev);
4625 }
4626 bytes[0] = 4;
4627 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), fname,
4628 strlen(fname)+1, NULL);
4629
4630 if (tevent_req_nomem(bytes, req)) {
4631 return tevent_req_post(req, ev);
4632 }
4633
4634 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL)) {
4635 additional_flags2 = FLAGS2_REPARSE_PATH;
4636 }
4637
4638 subreq = cli_smb_send(state, ev, cli, SMBcheckpath, additional_flags,
4639 additional_flags2,
4640 0, NULL, talloc_get_size(bytes), bytes);
4641 if (tevent_req_nomem(subreq, req)) {
4642 return tevent_req_post(req, ev);
4643 }
4644 tevent_req_set_callback(subreq, cli_chkpath_done, req);
4645 return req;
4646 }
4647
cli_chkpath_done(struct tevent_req * subreq)4648 static void cli_chkpath_done(struct tevent_req *subreq)
4649 {
4650 struct tevent_req *req = tevent_req_callback_data(
4651 subreq, struct tevent_req);
4652 NTSTATUS status;
4653
4654 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
4655 TALLOC_FREE(subreq);
4656 if (tevent_req_nterror(req, status)) {
4657 return;
4658 }
4659 tevent_req_done(req);
4660 }
4661
cli_chkpath_recv(struct tevent_req * req)4662 NTSTATUS cli_chkpath_recv(struct tevent_req *req)
4663 {
4664 return tevent_req_simple_recv_ntstatus(req);
4665 }
4666
cli_chkpath(struct cli_state * cli,const char * path)4667 NTSTATUS cli_chkpath(struct cli_state *cli, const char *path)
4668 {
4669 TALLOC_CTX *frame = NULL;
4670 struct tevent_context *ev = NULL;
4671 struct tevent_req *req = NULL;
4672 char *path2 = NULL;
4673 NTSTATUS status = NT_STATUS_OK;
4674
4675 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
4676 return cli_smb2_chkpath(cli, path);
4677 }
4678
4679 frame = talloc_stackframe();
4680
4681 if (smbXcli_conn_has_async_calls(cli->conn)) {
4682 /*
4683 * Can't use sync call while an async call is in flight
4684 */
4685 status = NT_STATUS_INVALID_PARAMETER;
4686 goto fail;
4687 }
4688
4689 path2 = talloc_strdup(frame, path);
4690 if (!path2) {
4691 status = NT_STATUS_NO_MEMORY;
4692 goto fail;
4693 }
4694 trim_char(path2,'\0','\\');
4695 if (!*path2) {
4696 path2 = talloc_strdup(frame, "\\");
4697 if (!path2) {
4698 status = NT_STATUS_NO_MEMORY;
4699 goto fail;
4700 }
4701 }
4702
4703 ev = samba_tevent_context_init(frame);
4704 if (ev == NULL) {
4705 status = NT_STATUS_NO_MEMORY;
4706 goto fail;
4707 }
4708
4709 req = cli_chkpath_send(frame, ev, cli, path2);
4710 if (req == NULL) {
4711 status = NT_STATUS_NO_MEMORY;
4712 goto fail;
4713 }
4714
4715 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
4716 goto fail;
4717 }
4718
4719 status = cli_chkpath_recv(req);
4720
4721 fail:
4722 TALLOC_FREE(frame);
4723 return status;
4724 }
4725
4726 /****************************************************************************
4727 Query disk space.
4728 ****************************************************************************/
4729
4730 static void cli_dskattr_done(struct tevent_req *subreq);
4731
4732 struct cli_dskattr_state {
4733 int bsize;
4734 int total;
4735 int avail;
4736 };
4737
cli_dskattr_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli)4738 struct tevent_req *cli_dskattr_send(TALLOC_CTX *mem_ctx,
4739 struct tevent_context *ev,
4740 struct cli_state *cli)
4741 {
4742 struct tevent_req *req = NULL, *subreq = NULL;
4743 struct cli_dskattr_state *state = NULL;
4744 uint8_t additional_flags = 0;
4745
4746 req = tevent_req_create(mem_ctx, &state, struct cli_dskattr_state);
4747 if (req == NULL) {
4748 return NULL;
4749 }
4750
4751 subreq = cli_smb_send(state, ev, cli, SMBdskattr, additional_flags, 0,
4752 0, NULL, 0, NULL);
4753 if (tevent_req_nomem(subreq, req)) {
4754 return tevent_req_post(req, ev);
4755 }
4756 tevent_req_set_callback(subreq, cli_dskattr_done, req);
4757 return req;
4758 }
4759
cli_dskattr_done(struct tevent_req * subreq)4760 static void cli_dskattr_done(struct tevent_req *subreq)
4761 {
4762 struct tevent_req *req = tevent_req_callback_data(
4763 subreq, struct tevent_req);
4764 struct cli_dskattr_state *state = tevent_req_data(
4765 req, struct cli_dskattr_state);
4766 uint8_t wct;
4767 uint16_t *vwv = NULL;
4768 NTSTATUS status;
4769
4770 status = cli_smb_recv(subreq, state, NULL, 4, &wct, &vwv, NULL,
4771 NULL);
4772 TALLOC_FREE(subreq);
4773 if (tevent_req_nterror(req, status)) {
4774 return;
4775 }
4776 state->bsize = SVAL(vwv+1, 0)*SVAL(vwv+2,0);
4777 state->total = SVAL(vwv+0, 0);
4778 state->avail = SVAL(vwv+3, 0);
4779 tevent_req_done(req);
4780 }
4781
cli_dskattr_recv(struct tevent_req * req,int * bsize,int * total,int * avail)4782 NTSTATUS cli_dskattr_recv(struct tevent_req *req, int *bsize, int *total, int *avail)
4783 {
4784 struct cli_dskattr_state *state = tevent_req_data(
4785 req, struct cli_dskattr_state);
4786 NTSTATUS status;
4787
4788 if (tevent_req_is_nterror(req, &status)) {
4789 return status;
4790 }
4791 *bsize = state->bsize;
4792 *total = state->total;
4793 *avail = state->avail;
4794 return NT_STATUS_OK;
4795 }
4796
cli_dskattr(struct cli_state * cli,int * bsize,int * total,int * avail)4797 NTSTATUS cli_dskattr(struct cli_state *cli, int *bsize, int *total, int *avail)
4798 {
4799 TALLOC_CTX *frame = NULL;
4800 struct tevent_context *ev = NULL;
4801 struct tevent_req *req = NULL;
4802 NTSTATUS status = NT_STATUS_OK;
4803
4804 frame = talloc_stackframe();
4805
4806 if (smbXcli_conn_has_async_calls(cli->conn)) {
4807 /*
4808 * Can't use sync call while an async call is in flight
4809 */
4810 status = NT_STATUS_INVALID_PARAMETER;
4811 goto fail;
4812 }
4813
4814 ev = samba_tevent_context_init(frame);
4815 if (ev == NULL) {
4816 status = NT_STATUS_NO_MEMORY;
4817 goto fail;
4818 }
4819
4820 req = cli_dskattr_send(frame, ev, cli);
4821 if (req == NULL) {
4822 status = NT_STATUS_NO_MEMORY;
4823 goto fail;
4824 }
4825
4826 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
4827 goto fail;
4828 }
4829
4830 status = cli_dskattr_recv(req, bsize, total, avail);
4831
4832 fail:
4833 TALLOC_FREE(frame);
4834 return status;
4835 }
4836
cli_disk_size(struct cli_state * cli,const char * path,uint64_t * bsize,uint64_t * total,uint64_t * avail)4837 NTSTATUS cli_disk_size(struct cli_state *cli, const char *path, uint64_t *bsize,
4838 uint64_t *total, uint64_t *avail)
4839 {
4840 uint64_t sectors_per_block;
4841 uint64_t bytes_per_sector;
4842 int old_bsize = 0, old_total = 0, old_avail = 0;
4843 NTSTATUS status;
4844
4845 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
4846 return cli_smb2_dskattr(cli, path, bsize, total, avail);
4847 }
4848
4849 /*
4850 * Try the trans2 disk full size info call first.
4851 * We already use this in SMBC_fstatvfs_ctx().
4852 * Ignore 'actual_available_units' as we only
4853 * care about the quota for the caller.
4854 */
4855
4856 status = cli_get_fs_full_size_info(cli,
4857 total,
4858 avail,
4859 NULL,
4860 §ors_per_block,
4861 &bytes_per_sector);
4862
4863 /* Try and cope will all varients of "we don't do this call"
4864 and fall back to cli_dskattr. */
4865
4866 if (NT_STATUS_EQUAL(status,NT_STATUS_NOT_IMPLEMENTED) ||
4867 NT_STATUS_EQUAL(status,NT_STATUS_NOT_SUPPORTED) ||
4868 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_INFO_CLASS) ||
4869 NT_STATUS_EQUAL(status,NT_STATUS_PROCEDURE_NOT_FOUND) ||
4870 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_LEVEL) ||
4871 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_PARAMETER) ||
4872 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_DEVICE_REQUEST) ||
4873 NT_STATUS_EQUAL(status,NT_STATUS_INVALID_DEVICE_STATE) ||
4874 NT_STATUS_EQUAL(status,NT_STATUS_CTL_FILE_NOT_SUPPORTED) ||
4875 NT_STATUS_EQUAL(status,NT_STATUS_UNSUCCESSFUL)) {
4876 goto try_dskattr;
4877 }
4878
4879 if (!NT_STATUS_IS_OK(status)) {
4880 return status;
4881 }
4882
4883 if (bsize) {
4884 *bsize = sectors_per_block *
4885 bytes_per_sector;
4886 }
4887
4888 return NT_STATUS_OK;
4889
4890 try_dskattr:
4891
4892 /* Old SMB1 core protocol fallback. */
4893 status = cli_dskattr(cli, &old_bsize, &old_total, &old_avail);
4894 if (!NT_STATUS_IS_OK(status)) {
4895 return status;
4896 }
4897 if (bsize) {
4898 *bsize = (uint64_t)old_bsize;
4899 }
4900 if (total) {
4901 *total = (uint64_t)old_total;
4902 }
4903 if (avail) {
4904 *avail = (uint64_t)old_avail;
4905 }
4906 return NT_STATUS_OK;
4907 }
4908
4909 /****************************************************************************
4910 Create and open a temporary file.
4911 ****************************************************************************/
4912
4913 static void cli_ctemp_done(struct tevent_req *subreq);
4914
4915 struct ctemp_state {
4916 uint16_t vwv[3];
4917 char *ret_path;
4918 uint16_t fnum;
4919 };
4920
cli_ctemp_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * path)4921 struct tevent_req *cli_ctemp_send(TALLOC_CTX *mem_ctx,
4922 struct tevent_context *ev,
4923 struct cli_state *cli,
4924 const char *path)
4925 {
4926 struct tevent_req *req = NULL, *subreq = NULL;
4927 struct ctemp_state *state = NULL;
4928 uint8_t additional_flags = 0;
4929 uint16_t additional_flags2 = 0;
4930 uint8_t *bytes = NULL;
4931
4932 req = tevent_req_create(mem_ctx, &state, struct ctemp_state);
4933 if (req == NULL) {
4934 return NULL;
4935 }
4936
4937 SSVAL(state->vwv,0,0);
4938 SIVALS(state->vwv+1,0,-1);
4939
4940 bytes = talloc_array(state, uint8_t, 1);
4941 if (tevent_req_nomem(bytes, req)) {
4942 return tevent_req_post(req, ev);
4943 }
4944 bytes[0] = 4;
4945 bytes = smb_bytes_push_str(bytes, smbXcli_conn_use_unicode(cli->conn), path,
4946 strlen(path)+1, NULL);
4947 if (tevent_req_nomem(bytes, req)) {
4948 return tevent_req_post(req, ev);
4949 }
4950
4951 if (clistr_is_previous_version_path(path, NULL, NULL, NULL)) {
4952 additional_flags2 = FLAGS2_REPARSE_PATH;
4953 }
4954
4955 subreq = cli_smb_send(state, ev, cli, SMBctemp, additional_flags,
4956 additional_flags2,
4957 3, state->vwv, talloc_get_size(bytes), bytes);
4958 if (tevent_req_nomem(subreq, req)) {
4959 return tevent_req_post(req, ev);
4960 }
4961 tevent_req_set_callback(subreq, cli_ctemp_done, req);
4962 return req;
4963 }
4964
cli_ctemp_done(struct tevent_req * subreq)4965 static void cli_ctemp_done(struct tevent_req *subreq)
4966 {
4967 struct tevent_req *req = tevent_req_callback_data(
4968 subreq, struct tevent_req);
4969 struct ctemp_state *state = tevent_req_data(
4970 req, struct ctemp_state);
4971 NTSTATUS status;
4972 uint8_t wcnt;
4973 uint16_t *vwv;
4974 uint32_t num_bytes = 0;
4975 uint8_t *bytes = NULL;
4976
4977 status = cli_smb_recv(subreq, state, NULL, 1, &wcnt, &vwv,
4978 &num_bytes, &bytes);
4979 TALLOC_FREE(subreq);
4980 if (tevent_req_nterror(req, status)) {
4981 return;
4982 }
4983
4984 state->fnum = SVAL(vwv+0, 0);
4985
4986 /* From W2K3, the result is just the ASCII name */
4987 if (num_bytes < 2) {
4988 tevent_req_nterror(req, NT_STATUS_DATA_ERROR);
4989 return;
4990 }
4991
4992 if (pull_string_talloc(state,
4993 NULL,
4994 0,
4995 &state->ret_path,
4996 bytes,
4997 num_bytes,
4998 STR_ASCII) == 0) {
4999 tevent_req_nterror(req, NT_STATUS_NO_MEMORY);
5000 return;
5001 }
5002 tevent_req_done(req);
5003 }
5004
cli_ctemp_recv(struct tevent_req * req,TALLOC_CTX * ctx,uint16_t * pfnum,char ** outfile)5005 NTSTATUS cli_ctemp_recv(struct tevent_req *req,
5006 TALLOC_CTX *ctx,
5007 uint16_t *pfnum,
5008 char **outfile)
5009 {
5010 struct ctemp_state *state = tevent_req_data(req,
5011 struct ctemp_state);
5012 NTSTATUS status;
5013
5014 if (tevent_req_is_nterror(req, &status)) {
5015 return status;
5016 }
5017 *pfnum = state->fnum;
5018 *outfile = talloc_strdup(ctx, state->ret_path);
5019 if (!*outfile) {
5020 return NT_STATUS_NO_MEMORY;
5021 }
5022 return NT_STATUS_OK;
5023 }
5024
cli_ctemp(struct cli_state * cli,TALLOC_CTX * ctx,const char * path,uint16_t * pfnum,char ** out_path)5025 NTSTATUS cli_ctemp(struct cli_state *cli,
5026 TALLOC_CTX *ctx,
5027 const char *path,
5028 uint16_t *pfnum,
5029 char **out_path)
5030 {
5031 TALLOC_CTX *frame = talloc_stackframe();
5032 struct tevent_context *ev;
5033 struct tevent_req *req;
5034 NTSTATUS status = NT_STATUS_OK;
5035
5036 if (smbXcli_conn_has_async_calls(cli->conn)) {
5037 /*
5038 * Can't use sync call while an async call is in flight
5039 */
5040 status = NT_STATUS_INVALID_PARAMETER;
5041 goto fail;
5042 }
5043
5044 ev = samba_tevent_context_init(frame);
5045 if (ev == NULL) {
5046 status = NT_STATUS_NO_MEMORY;
5047 goto fail;
5048 }
5049
5050 req = cli_ctemp_send(frame, ev, cli, path);
5051 if (req == NULL) {
5052 status = NT_STATUS_NO_MEMORY;
5053 goto fail;
5054 }
5055
5056 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
5057 goto fail;
5058 }
5059
5060 status = cli_ctemp_recv(req, ctx, pfnum, out_path);
5061
5062 fail:
5063 TALLOC_FREE(frame);
5064 return status;
5065 }
5066
5067 /*
5068 send a raw ioctl - used by the torture code
5069 */
cli_raw_ioctl(struct cli_state * cli,uint16_t fnum,uint32_t code,DATA_BLOB * blob)5070 NTSTATUS cli_raw_ioctl(struct cli_state *cli, uint16_t fnum, uint32_t code, DATA_BLOB *blob)
5071 {
5072 uint16_t vwv[3];
5073 NTSTATUS status;
5074
5075 SSVAL(vwv+0, 0, fnum);
5076 SSVAL(vwv+1, 0, code>>16);
5077 SSVAL(vwv+2, 0, (code&0xFFFF));
5078
5079 status = cli_smb(talloc_tos(), cli, SMBioctl, 0, 3, vwv, 0, NULL,
5080 NULL, 0, NULL, NULL, NULL, NULL);
5081 if (!NT_STATUS_IS_OK(status)) {
5082 return status;
5083 }
5084 *blob = data_blob_null;
5085 return NT_STATUS_OK;
5086 }
5087
5088 /*********************************************************
5089 Set an extended attribute utility fn.
5090 *********************************************************/
5091
cli_set_ea(struct cli_state * cli,uint16_t setup_val,uint8_t * param,unsigned int param_len,const char * ea_name,const char * ea_val,size_t ea_len)5092 static NTSTATUS cli_set_ea(struct cli_state *cli, uint16_t setup_val,
5093 uint8_t *param, unsigned int param_len,
5094 const char *ea_name,
5095 const char *ea_val, size_t ea_len)
5096 {
5097 uint16_t setup[1];
5098 unsigned int data_len = 0;
5099 uint8_t *data = NULL;
5100 char *p;
5101 size_t ea_namelen = strlen(ea_name);
5102 NTSTATUS status;
5103
5104 SSVAL(setup, 0, setup_val);
5105
5106 if (ea_namelen == 0 && ea_len == 0) {
5107 data_len = 4;
5108 data = talloc_array(talloc_tos(),
5109 uint8_t,
5110 data_len);
5111 if (!data) {
5112 return NT_STATUS_NO_MEMORY;
5113 }
5114 p = (char *)data;
5115 SIVAL(p,0,data_len);
5116 } else {
5117 data_len = 4 + 4 + ea_namelen + 1 + ea_len;
5118 data = talloc_array(talloc_tos(),
5119 uint8_t,
5120 data_len);
5121 if (!data) {
5122 return NT_STATUS_NO_MEMORY;
5123 }
5124 p = (char *)data;
5125 SIVAL(p,0,data_len);
5126 p += 4;
5127 SCVAL(p, 0, 0); /* EA flags. */
5128 SCVAL(p, 1, ea_namelen);
5129 SSVAL(p, 2, ea_len);
5130 memcpy(p+4, ea_name, ea_namelen+1); /* Copy in the name. */
5131 memcpy(p+4+ea_namelen+1, ea_val, ea_len);
5132 }
5133
5134 /*
5135 * FIXME - if we want to do previous version path
5136 * processing on an EA set call we need to turn this
5137 * into calls to cli_trans_send()/cli_trans_recv()
5138 * with a temporary event context, as cli_trans_send()
5139 * have access to the additional_flags2 needed to
5140 * send @GMT- paths. JRA.
5141 */
5142
5143 status = cli_trans(talloc_tos(), cli, SMBtrans2, NULL, -1, 0, 0,
5144 setup, 1, 0,
5145 param, param_len, 2,
5146 data, data_len, 0,
5147 NULL,
5148 NULL, 0, NULL, /* rsetup */
5149 NULL, 0, NULL, /* rparam */
5150 NULL, 0, NULL); /* rdata */
5151 talloc_free(data);
5152 return status;
5153 }
5154
5155 /*********************************************************
5156 Set an extended attribute on a pathname.
5157 *********************************************************/
5158
cli_set_ea_path(struct cli_state * cli,const char * path,const char * ea_name,const char * ea_val,size_t ea_len)5159 NTSTATUS cli_set_ea_path(struct cli_state *cli, const char *path,
5160 const char *ea_name, const char *ea_val,
5161 size_t ea_len)
5162 {
5163 unsigned int param_len = 0;
5164 uint8_t *param;
5165 NTSTATUS status;
5166 TALLOC_CTX *frame = NULL;
5167
5168 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
5169 return cli_smb2_set_ea_path(cli,
5170 path,
5171 ea_name,
5172 ea_val,
5173 ea_len);
5174 }
5175
5176 frame = talloc_stackframe();
5177
5178 param = talloc_array(frame, uint8_t, 6);
5179 if (!param) {
5180 status = NT_STATUS_NO_MEMORY;
5181 goto fail;
5182 }
5183 SSVAL(param,0,SMB_INFO_SET_EA);
5184 SSVAL(param,2,0);
5185 SSVAL(param,4,0);
5186
5187 param = trans2_bytes_push_str(param, smbXcli_conn_use_unicode(cli->conn),
5188 path, strlen(path)+1,
5189 NULL);
5190 param_len = talloc_get_size(param);
5191
5192 status = cli_set_ea(cli, TRANSACT2_SETPATHINFO, param, param_len,
5193 ea_name, ea_val, ea_len);
5194
5195 fail:
5196
5197 TALLOC_FREE(frame);
5198 return status;
5199 }
5200
5201 /*********************************************************
5202 Set an extended attribute on an fnum.
5203 *********************************************************/
5204
cli_set_ea_fnum(struct cli_state * cli,uint16_t fnum,const char * ea_name,const char * ea_val,size_t ea_len)5205 NTSTATUS cli_set_ea_fnum(struct cli_state *cli, uint16_t fnum,
5206 const char *ea_name, const char *ea_val,
5207 size_t ea_len)
5208 {
5209 uint8_t param[6];
5210
5211 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
5212 return cli_smb2_set_ea_fnum(cli,
5213 fnum,
5214 ea_name,
5215 ea_val,
5216 ea_len);
5217 }
5218
5219 memset(param, 0, 6);
5220 SSVAL(param,0,fnum);
5221 SSVAL(param,2,SMB_INFO_SET_EA);
5222
5223 return cli_set_ea(cli, TRANSACT2_SETFILEINFO, param, 6,
5224 ea_name, ea_val, ea_len);
5225 }
5226
5227 /*********************************************************
5228 Get an extended attribute list utility fn.
5229 *********************************************************/
5230
parse_ea_blob(TALLOC_CTX * ctx,const uint8_t * rdata,size_t rdata_len,size_t * pnum_eas,struct ea_struct ** pea_list)5231 static bool parse_ea_blob(TALLOC_CTX *ctx, const uint8_t *rdata,
5232 size_t rdata_len,
5233 size_t *pnum_eas, struct ea_struct **pea_list)
5234 {
5235 struct ea_struct *ea_list = NULL;
5236 size_t num_eas;
5237 size_t ea_size;
5238 const uint8_t *p;
5239
5240 if (rdata_len < 4) {
5241 return false;
5242 }
5243
5244 ea_size = (size_t)IVAL(rdata,0);
5245 if (ea_size > rdata_len) {
5246 return false;
5247 }
5248
5249 if (ea_size == 0) {
5250 /* No EA's present. */
5251 *pnum_eas = 0;
5252 *pea_list = NULL;
5253 return true;
5254 }
5255
5256 p = rdata + 4;
5257 ea_size -= 4;
5258
5259 /* Validate the EA list and count it. */
5260 for (num_eas = 0; ea_size >= 4; num_eas++) {
5261 unsigned int ea_namelen = CVAL(p,1);
5262 unsigned int ea_valuelen = SVAL(p,2);
5263 if (ea_namelen == 0) {
5264 return false;
5265 }
5266 if (4 + ea_namelen + 1 + ea_valuelen > ea_size) {
5267 return false;
5268 }
5269 ea_size -= 4 + ea_namelen + 1 + ea_valuelen;
5270 p += 4 + ea_namelen + 1 + ea_valuelen;
5271 }
5272
5273 if (num_eas == 0) {
5274 *pnum_eas = 0;
5275 *pea_list = NULL;
5276 return true;
5277 }
5278
5279 *pnum_eas = num_eas;
5280 if (!pea_list) {
5281 /* Caller only wants number of EA's. */
5282 return true;
5283 }
5284
5285 ea_list = talloc_array(ctx, struct ea_struct, num_eas);
5286 if (!ea_list) {
5287 return false;
5288 }
5289
5290 p = rdata + 4;
5291
5292 for (num_eas = 0; num_eas < *pnum_eas; num_eas++ ) {
5293 struct ea_struct *ea = &ea_list[num_eas];
5294 fstring unix_ea_name;
5295 unsigned int ea_namelen = CVAL(p,1);
5296 unsigned int ea_valuelen = SVAL(p,2);
5297
5298 ea->flags = CVAL(p,0);
5299 unix_ea_name[0] = '\0';
5300 pull_ascii(unix_ea_name, p + 4, sizeof(unix_ea_name), rdata_len - PTR_DIFF(p+4, rdata), STR_TERMINATE);
5301 ea->name = talloc_strdup(ea_list, unix_ea_name);
5302 if (!ea->name) {
5303 goto fail;
5304 }
5305 /* Ensure the value is null terminated (in case it's a string). */
5306 ea->value = data_blob_talloc(ea_list, NULL, ea_valuelen + 1);
5307 if (!ea->value.data) {
5308 goto fail;
5309 }
5310 if (ea_valuelen) {
5311 memcpy(ea->value.data, p+4+ea_namelen+1, ea_valuelen);
5312 }
5313 ea->value.data[ea_valuelen] = 0;
5314 ea->value.length--;
5315 p += 4 + ea_namelen + 1 + ea_valuelen;
5316 }
5317
5318 *pea_list = ea_list;
5319 return true;
5320
5321 fail:
5322 TALLOC_FREE(ea_list);
5323 return false;
5324 }
5325
5326 /*********************************************************
5327 Get an extended attribute list from a pathname.
5328 *********************************************************/
5329
5330 struct cli_get_ea_list_path_state {
5331 uint32_t num_data;
5332 uint8_t *data;
5333 };
5334
5335 static void cli_get_ea_list_path_done(struct tevent_req *subreq);
5336
cli_get_ea_list_path_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname)5337 struct tevent_req *cli_get_ea_list_path_send(TALLOC_CTX *mem_ctx,
5338 struct tevent_context *ev,
5339 struct cli_state *cli,
5340 const char *fname)
5341 {
5342 struct tevent_req *req, *subreq;
5343 struct cli_get_ea_list_path_state *state;
5344
5345 req = tevent_req_create(mem_ctx, &state,
5346 struct cli_get_ea_list_path_state);
5347 if (req == NULL) {
5348 return NULL;
5349 }
5350 subreq = cli_qpathinfo_send(state, ev, cli, fname,
5351 SMB_INFO_QUERY_ALL_EAS, 4,
5352 CLI_BUFFER_SIZE);
5353 if (tevent_req_nomem(subreq, req)) {
5354 return tevent_req_post(req, ev);
5355 }
5356 tevent_req_set_callback(subreq, cli_get_ea_list_path_done, req);
5357 return req;
5358 }
5359
cli_get_ea_list_path_done(struct tevent_req * subreq)5360 static void cli_get_ea_list_path_done(struct tevent_req *subreq)
5361 {
5362 struct tevent_req *req = tevent_req_callback_data(
5363 subreq, struct tevent_req);
5364 struct cli_get_ea_list_path_state *state = tevent_req_data(
5365 req, struct cli_get_ea_list_path_state);
5366 NTSTATUS status;
5367
5368 status = cli_qpathinfo_recv(subreq, state, &state->data,
5369 &state->num_data);
5370 TALLOC_FREE(subreq);
5371 if (tevent_req_nterror(req, status)) {
5372 return;
5373 }
5374 tevent_req_done(req);
5375 }
5376
cli_get_ea_list_path_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx,size_t * pnum_eas,struct ea_struct ** peas)5377 NTSTATUS cli_get_ea_list_path_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
5378 size_t *pnum_eas, struct ea_struct **peas)
5379 {
5380 struct cli_get_ea_list_path_state *state = tevent_req_data(
5381 req, struct cli_get_ea_list_path_state);
5382 NTSTATUS status;
5383
5384 if (tevent_req_is_nterror(req, &status)) {
5385 return status;
5386 }
5387 if (!parse_ea_blob(mem_ctx, state->data, state->num_data,
5388 pnum_eas, peas)) {
5389 return NT_STATUS_INVALID_NETWORK_RESPONSE;
5390 }
5391 return NT_STATUS_OK;
5392 }
5393
cli_get_ea_list_path(struct cli_state * cli,const char * path,TALLOC_CTX * ctx,size_t * pnum_eas,struct ea_struct ** pea_list)5394 NTSTATUS cli_get_ea_list_path(struct cli_state *cli, const char *path,
5395 TALLOC_CTX *ctx,
5396 size_t *pnum_eas,
5397 struct ea_struct **pea_list)
5398 {
5399 TALLOC_CTX *frame = NULL;
5400 struct tevent_context *ev = NULL;
5401 struct tevent_req *req = NULL;
5402 NTSTATUS status = NT_STATUS_NO_MEMORY;
5403
5404 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
5405 return cli_smb2_get_ea_list_path(cli,
5406 path,
5407 ctx,
5408 pnum_eas,
5409 pea_list);
5410 }
5411
5412 frame = talloc_stackframe();
5413
5414 if (smbXcli_conn_has_async_calls(cli->conn)) {
5415 /*
5416 * Can't use sync call while an async call is in flight
5417 */
5418 status = NT_STATUS_INVALID_PARAMETER;
5419 goto fail;
5420 }
5421 ev = samba_tevent_context_init(frame);
5422 if (ev == NULL) {
5423 goto fail;
5424 }
5425 req = cli_get_ea_list_path_send(frame, ev, cli, path);
5426 if (req == NULL) {
5427 goto fail;
5428 }
5429 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
5430 goto fail;
5431 }
5432 status = cli_get_ea_list_path_recv(req, ctx, pnum_eas, pea_list);
5433 fail:
5434 TALLOC_FREE(frame);
5435 return status;
5436 }
5437
5438 /****************************************************************************
5439 Convert open "flags" arg to uint32_t on wire.
5440 ****************************************************************************/
5441
open_flags_to_wire(int flags)5442 static uint32_t open_flags_to_wire(int flags)
5443 {
5444 int open_mode = flags & O_ACCMODE;
5445 uint32_t ret = 0;
5446
5447 switch (open_mode) {
5448 case O_WRONLY:
5449 ret |= SMB_O_WRONLY;
5450 break;
5451 case O_RDWR:
5452 ret |= SMB_O_RDWR;
5453 break;
5454 default:
5455 case O_RDONLY:
5456 ret |= SMB_O_RDONLY;
5457 break;
5458 }
5459
5460 if (flags & O_CREAT) {
5461 ret |= SMB_O_CREAT;
5462 }
5463 if (flags & O_EXCL) {
5464 ret |= SMB_O_EXCL;
5465 }
5466 if (flags & O_TRUNC) {
5467 ret |= SMB_O_TRUNC;
5468 }
5469 #if defined(O_SYNC)
5470 if (flags & O_SYNC) {
5471 ret |= SMB_O_SYNC;
5472 }
5473 #endif /* O_SYNC */
5474 if (flags & O_APPEND) {
5475 ret |= SMB_O_APPEND;
5476 }
5477 #if defined(O_DIRECT)
5478 if (flags & O_DIRECT) {
5479 ret |= SMB_O_DIRECT;
5480 }
5481 #endif
5482 #if defined(O_DIRECTORY)
5483 if (flags & O_DIRECTORY) {
5484 ret |= SMB_O_DIRECTORY;
5485 }
5486 #endif
5487 return ret;
5488 }
5489
5490 /****************************************************************************
5491 Open a file - POSIX semantics. Returns fnum. Doesn't request oplock.
5492 ****************************************************************************/
5493
5494 struct cli_posix_open_internal_state {
5495 uint16_t setup;
5496 uint8_t *param;
5497 uint8_t data[18];
5498 uint16_t fnum; /* Out */
5499 };
5500
5501 static void cli_posix_open_internal_done(struct tevent_req *subreq);
5502
cli_posix_open_internal_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint32_t wire_flags,mode_t mode)5503 static struct tevent_req *cli_posix_open_internal_send(TALLOC_CTX *mem_ctx,
5504 struct tevent_context *ev,
5505 struct cli_state *cli,
5506 const char *fname,
5507 uint32_t wire_flags,
5508 mode_t mode)
5509 {
5510 struct tevent_req *req = NULL, *subreq = NULL;
5511 struct cli_posix_open_internal_state *state = NULL;
5512
5513 req = tevent_req_create(
5514 mem_ctx, &state, struct cli_posix_open_internal_state);
5515 if (req == NULL) {
5516 return NULL;
5517 }
5518
5519 /* Setup setup word. */
5520 SSVAL(&state->setup, 0, TRANSACT2_SETPATHINFO);
5521
5522 /* Setup param array. */
5523 state->param = talloc_zero_array(state, uint8_t, 6);
5524 if (tevent_req_nomem(state->param, req)) {
5525 return tevent_req_post(req, ev);
5526 }
5527 SSVAL(state->param, 0, SMB_POSIX_PATH_OPEN);
5528
5529 state->param = trans2_bytes_push_str(
5530 state->param,
5531 smbXcli_conn_use_unicode(cli->conn),
5532 fname,
5533 strlen(fname)+1,
5534 NULL);
5535
5536 if (tevent_req_nomem(state->param, req)) {
5537 return tevent_req_post(req, ev);
5538 }
5539
5540 SIVAL(state->data,0,0); /* No oplock. */
5541 SIVAL(state->data,4,wire_flags);
5542 SIVAL(state->data,8,unix_perms_to_wire(mode));
5543 SIVAL(state->data,12,0); /* Top bits of perms currently undefined. */
5544 SSVAL(state->data,16,SMB_NO_INFO_LEVEL_RETURNED); /* No info level returned. */
5545
5546 subreq = cli_trans_send(state, /* mem ctx. */
5547 ev, /* event ctx. */
5548 cli, /* cli_state. */
5549 0, /* additional_flags2 */
5550 SMBtrans2, /* cmd. */
5551 NULL, /* pipe name. */
5552 -1, /* fid. */
5553 0, /* function. */
5554 0, /* flags. */
5555 &state->setup, /* setup. */
5556 1, /* num setup uint16_t words. */
5557 0, /* max returned setup. */
5558 state->param, /* param. */
5559 talloc_get_size(state->param),/* num param. */
5560 2, /* max returned param. */
5561 state->data, /* data. */
5562 18, /* num data. */
5563 12); /* max returned data. */
5564
5565 if (tevent_req_nomem(subreq, req)) {
5566 return tevent_req_post(req, ev);
5567 }
5568 tevent_req_set_callback(subreq, cli_posix_open_internal_done, req);
5569 return req;
5570 }
5571
cli_posix_open_internal_done(struct tevent_req * subreq)5572 static void cli_posix_open_internal_done(struct tevent_req *subreq)
5573 {
5574 struct tevent_req *req = tevent_req_callback_data(
5575 subreq, struct tevent_req);
5576 struct cli_posix_open_internal_state *state = tevent_req_data(
5577 req, struct cli_posix_open_internal_state);
5578 NTSTATUS status;
5579 uint8_t *data;
5580 uint32_t num_data;
5581
5582 status = cli_trans_recv(
5583 subreq,
5584 state,
5585 NULL,
5586 NULL,
5587 0,
5588 NULL,
5589 NULL,
5590 0,
5591 NULL,
5592 &data,
5593 12,
5594 &num_data);
5595 TALLOC_FREE(subreq);
5596 if (tevent_req_nterror(req, status)) {
5597 return;
5598 }
5599 state->fnum = SVAL(data,2);
5600 tevent_req_done(req);
5601 }
5602
cli_posix_open_internal_recv(struct tevent_req * req,uint16_t * pfnum)5603 static NTSTATUS cli_posix_open_internal_recv(struct tevent_req *req,
5604 uint16_t *pfnum)
5605 {
5606 struct cli_posix_open_internal_state *state = tevent_req_data(
5607 req, struct cli_posix_open_internal_state);
5608 NTSTATUS status;
5609
5610 if (tevent_req_is_nterror(req, &status)) {
5611 return status;
5612 }
5613 *pfnum = state->fnum;
5614 return NT_STATUS_OK;
5615 }
5616
5617 struct cli_posix_open_state {
5618 uint16_t fnum;
5619 };
5620
5621 static void cli_posix_open_done(struct tevent_req *subreq);
5622
cli_posix_open_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,int flags,mode_t mode)5623 struct tevent_req *cli_posix_open_send(TALLOC_CTX *mem_ctx,
5624 struct tevent_context *ev,
5625 struct cli_state *cli,
5626 const char *fname,
5627 int flags,
5628 mode_t mode)
5629 {
5630 struct tevent_req *req = NULL, *subreq = NULL;
5631 struct cli_posix_open_state *state = NULL;
5632 uint32_t wire_flags;
5633
5634 req = tevent_req_create(mem_ctx, &state,
5635 struct cli_posix_open_state);
5636 if (req == NULL) {
5637 return NULL;
5638 }
5639
5640 wire_flags = open_flags_to_wire(flags);
5641
5642 subreq = cli_posix_open_internal_send(
5643 mem_ctx, ev, cli, fname, wire_flags, mode);
5644 if (tevent_req_nomem(subreq, req)) {
5645 return tevent_req_post(req, ev);
5646 }
5647 tevent_req_set_callback(subreq, cli_posix_open_done, req);
5648 return req;
5649 }
5650
cli_posix_open_done(struct tevent_req * subreq)5651 static void cli_posix_open_done(struct tevent_req *subreq)
5652 {
5653 struct tevent_req *req = tevent_req_callback_data(
5654 subreq, struct tevent_req);
5655 struct cli_posix_open_state *state = tevent_req_data(
5656 req, struct cli_posix_open_state);
5657 NTSTATUS status;
5658
5659 status = cli_posix_open_internal_recv(subreq, &state->fnum);
5660 tevent_req_simple_finish_ntstatus(subreq, status);
5661 }
5662
cli_posix_open_recv(struct tevent_req * req,uint16_t * pfnum)5663 NTSTATUS cli_posix_open_recv(struct tevent_req *req, uint16_t *pfnum)
5664 {
5665 struct cli_posix_open_state *state = tevent_req_data(
5666 req, struct cli_posix_open_state);
5667 NTSTATUS status;
5668
5669 if (tevent_req_is_nterror(req, &status)) {
5670 return status;
5671 }
5672 *pfnum = state->fnum;
5673 return NT_STATUS_OK;
5674 }
5675
5676 /****************************************************************************
5677 Open - POSIX semantics. Doesn't request oplock.
5678 ****************************************************************************/
5679
cli_posix_open(struct cli_state * cli,const char * fname,int flags,mode_t mode,uint16_t * pfnum)5680 NTSTATUS cli_posix_open(struct cli_state *cli, const char *fname,
5681 int flags, mode_t mode, uint16_t *pfnum)
5682 {
5683
5684 TALLOC_CTX *frame = talloc_stackframe();
5685 struct tevent_context *ev = NULL;
5686 struct tevent_req *req = NULL;
5687 NTSTATUS status = NT_STATUS_NO_MEMORY;
5688
5689 if (smbXcli_conn_has_async_calls(cli->conn)) {
5690 /*
5691 * Can't use sync call while an async call is in flight
5692 */
5693 status = NT_STATUS_INVALID_PARAMETER;
5694 goto fail;
5695 }
5696 ev = samba_tevent_context_init(frame);
5697 if (ev == NULL) {
5698 goto fail;
5699 }
5700 req = cli_posix_open_send(
5701 frame, ev, cli, fname, flags, mode);
5702 if (req == NULL) {
5703 goto fail;
5704 }
5705 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
5706 goto fail;
5707 }
5708 status = cli_posix_open_recv(req, pfnum);
5709 fail:
5710 TALLOC_FREE(frame);
5711 return status;
5712 }
5713
5714 struct cli_posix_mkdir_state {
5715 struct tevent_context *ev;
5716 struct cli_state *cli;
5717 };
5718
5719 static void cli_posix_mkdir_done(struct tevent_req *subreq);
5720
cli_posix_mkdir_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,mode_t mode)5721 struct tevent_req *cli_posix_mkdir_send(TALLOC_CTX *mem_ctx,
5722 struct tevent_context *ev,
5723 struct cli_state *cli,
5724 const char *fname,
5725 mode_t mode)
5726 {
5727 struct tevent_req *req = NULL, *subreq = NULL;
5728 struct cli_posix_mkdir_state *state = NULL;
5729 uint32_t wire_flags;
5730
5731 req = tevent_req_create(
5732 mem_ctx, &state, struct cli_posix_mkdir_state);
5733 if (req == NULL) {
5734 return NULL;
5735 }
5736 state->ev = ev;
5737 state->cli = cli;
5738
5739 wire_flags = SMB_O_CREAT | SMB_O_DIRECTORY;
5740
5741 subreq = cli_posix_open_internal_send(
5742 mem_ctx, ev, cli, fname, wire_flags, mode);
5743 if (tevent_req_nomem(subreq, req)) {
5744 return tevent_req_post(req, ev);
5745 }
5746 tevent_req_set_callback(subreq, cli_posix_mkdir_done, req);
5747 return req;
5748 }
5749
cli_posix_mkdir_done(struct tevent_req * subreq)5750 static void cli_posix_mkdir_done(struct tevent_req *subreq)
5751 {
5752 struct tevent_req *req = tevent_req_callback_data(
5753 subreq, struct tevent_req);
5754 NTSTATUS status;
5755 uint16_t fnum;
5756
5757 status = cli_posix_open_internal_recv(subreq, &fnum);
5758 TALLOC_FREE(subreq);
5759 if (tevent_req_nterror(req, status)) {
5760 return;
5761 }
5762 tevent_req_done(req);
5763 }
5764
cli_posix_mkdir_recv(struct tevent_req * req)5765 NTSTATUS cli_posix_mkdir_recv(struct tevent_req *req)
5766 {
5767 return tevent_req_simple_recv_ntstatus(req);
5768 }
5769
cli_posix_mkdir(struct cli_state * cli,const char * fname,mode_t mode)5770 NTSTATUS cli_posix_mkdir(struct cli_state *cli, const char *fname, mode_t mode)
5771 {
5772 TALLOC_CTX *frame = talloc_stackframe();
5773 struct tevent_context *ev = NULL;
5774 struct tevent_req *req = NULL;
5775 NTSTATUS status = NT_STATUS_NO_MEMORY;
5776
5777 if (smbXcli_conn_has_async_calls(cli->conn)) {
5778 /*
5779 * Can't use sync call while an async call is in flight
5780 */
5781 status = NT_STATUS_INVALID_PARAMETER;
5782 goto fail;
5783 }
5784
5785 ev = samba_tevent_context_init(frame);
5786 if (ev == NULL) {
5787 goto fail;
5788 }
5789 req = cli_posix_mkdir_send(
5790 frame, ev, cli, fname, mode);
5791 if (req == NULL) {
5792 goto fail;
5793 }
5794 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
5795 goto fail;
5796 }
5797 status = cli_posix_mkdir_recv(req);
5798 fail:
5799 TALLOC_FREE(frame);
5800 return status;
5801 }
5802
5803 /****************************************************************************
5804 unlink or rmdir - POSIX semantics.
5805 ****************************************************************************/
5806
5807 struct cli_posix_unlink_internal_state {
5808 uint8_t data[2];
5809 };
5810
5811 static void cli_posix_unlink_internal_done(struct tevent_req *subreq);
5812
cli_posix_unlink_internal_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint16_t level)5813 static struct tevent_req *cli_posix_unlink_internal_send(TALLOC_CTX *mem_ctx,
5814 struct tevent_context *ev,
5815 struct cli_state *cli,
5816 const char *fname,
5817 uint16_t level)
5818 {
5819 struct tevent_req *req = NULL, *subreq = NULL;
5820 struct cli_posix_unlink_internal_state *state = NULL;
5821
5822 req = tevent_req_create(mem_ctx, &state,
5823 struct cli_posix_unlink_internal_state);
5824 if (req == NULL) {
5825 return NULL;
5826 }
5827
5828 /* Setup data word. */
5829 SSVAL(state->data, 0, level);
5830
5831 subreq = cli_setpathinfo_send(state, ev, cli,
5832 SMB_POSIX_PATH_UNLINK,
5833 fname,
5834 state->data, sizeof(state->data));
5835 if (tevent_req_nomem(subreq, req)) {
5836 return tevent_req_post(req, ev);
5837 }
5838 tevent_req_set_callback(subreq, cli_posix_unlink_internal_done, req);
5839 return req;
5840 }
5841
cli_posix_unlink_internal_done(struct tevent_req * subreq)5842 static void cli_posix_unlink_internal_done(struct tevent_req *subreq)
5843 {
5844 NTSTATUS status = cli_setpathinfo_recv(subreq);
5845 tevent_req_simple_finish_ntstatus(subreq, status);
5846 }
5847
cli_posix_unlink_internal_recv(struct tevent_req * req)5848 static NTSTATUS cli_posix_unlink_internal_recv(struct tevent_req *req)
5849 {
5850 return tevent_req_simple_recv_ntstatus(req);
5851 }
5852
5853 struct cli_posix_unlink_state {
5854 uint8_t dummy;
5855 };
5856
5857 static void cli_posix_unlink_done(struct tevent_req *subreq);
5858
cli_posix_unlink_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname)5859 struct tevent_req *cli_posix_unlink_send(TALLOC_CTX *mem_ctx,
5860 struct tevent_context *ev,
5861 struct cli_state *cli,
5862 const char *fname)
5863 {
5864 struct tevent_req *req = NULL, *subreq = NULL;
5865 struct cli_posix_unlink_state *state;
5866
5867 req = tevent_req_create(
5868 mem_ctx, &state, struct cli_posix_unlink_state);
5869 if (req == NULL) {
5870 return NULL;
5871 }
5872 subreq = cli_posix_unlink_internal_send(
5873 mem_ctx, ev, cli, fname, SMB_POSIX_UNLINK_FILE_TARGET);
5874 if (tevent_req_nomem(subreq, req)) {
5875 return tevent_req_post(req, ev);
5876 }
5877 tevent_req_set_callback(subreq, cli_posix_unlink_done, req);
5878 return req;
5879 }
5880
cli_posix_unlink_done(struct tevent_req * subreq)5881 static void cli_posix_unlink_done(struct tevent_req *subreq)
5882 {
5883 NTSTATUS status = cli_posix_unlink_internal_recv(subreq);
5884 tevent_req_simple_finish_ntstatus(subreq, status);
5885 }
5886
cli_posix_unlink_recv(struct tevent_req * req)5887 NTSTATUS cli_posix_unlink_recv(struct tevent_req *req)
5888 {
5889 return tevent_req_simple_recv_ntstatus(req);
5890 }
5891
5892 /****************************************************************************
5893 unlink - POSIX semantics.
5894 ****************************************************************************/
5895
cli_posix_unlink(struct cli_state * cli,const char * fname)5896 NTSTATUS cli_posix_unlink(struct cli_state *cli, const char *fname)
5897 {
5898 TALLOC_CTX *frame = talloc_stackframe();
5899 struct tevent_context *ev = NULL;
5900 struct tevent_req *req = NULL;
5901 NTSTATUS status = NT_STATUS_OK;
5902
5903 if (smbXcli_conn_has_async_calls(cli->conn)) {
5904 /*
5905 * Can't use sync call while an async call is in flight
5906 */
5907 status = NT_STATUS_INVALID_PARAMETER;
5908 goto fail;
5909 }
5910
5911 ev = samba_tevent_context_init(frame);
5912 if (ev == NULL) {
5913 status = NT_STATUS_NO_MEMORY;
5914 goto fail;
5915 }
5916
5917 req = cli_posix_unlink_send(frame,
5918 ev,
5919 cli,
5920 fname);
5921 if (req == NULL) {
5922 status = NT_STATUS_NO_MEMORY;
5923 goto fail;
5924 }
5925
5926 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
5927 goto fail;
5928 }
5929
5930 status = cli_posix_unlink_recv(req);
5931
5932 fail:
5933 TALLOC_FREE(frame);
5934 return status;
5935 }
5936
5937 /****************************************************************************
5938 rmdir - POSIX semantics.
5939 ****************************************************************************/
5940
5941 struct cli_posix_rmdir_state {
5942 uint8_t dummy;
5943 };
5944
5945 static void cli_posix_rmdir_done(struct tevent_req *subreq);
5946
cli_posix_rmdir_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname)5947 struct tevent_req *cli_posix_rmdir_send(TALLOC_CTX *mem_ctx,
5948 struct tevent_context *ev,
5949 struct cli_state *cli,
5950 const char *fname)
5951 {
5952 struct tevent_req *req = NULL, *subreq = NULL;
5953 struct cli_posix_rmdir_state *state;
5954
5955 req = tevent_req_create(mem_ctx, &state, struct cli_posix_rmdir_state);
5956 if (req == NULL) {
5957 return NULL;
5958 }
5959 subreq = cli_posix_unlink_internal_send(
5960 mem_ctx, ev, cli, fname, SMB_POSIX_UNLINK_DIRECTORY_TARGET);
5961 if (tevent_req_nomem(subreq, req)) {
5962 return tevent_req_post(req, ev);
5963 }
5964 tevent_req_set_callback(subreq, cli_posix_rmdir_done, req);
5965 return req;
5966 }
5967
cli_posix_rmdir_done(struct tevent_req * subreq)5968 static void cli_posix_rmdir_done(struct tevent_req *subreq)
5969 {
5970 NTSTATUS status = cli_posix_unlink_internal_recv(subreq);
5971 tevent_req_simple_finish_ntstatus(subreq, status);
5972 }
5973
cli_posix_rmdir_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx)5974 NTSTATUS cli_posix_rmdir_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx)
5975 {
5976 return tevent_req_simple_recv_ntstatus(req);
5977 }
5978
cli_posix_rmdir(struct cli_state * cli,const char * fname)5979 NTSTATUS cli_posix_rmdir(struct cli_state *cli, const char *fname)
5980 {
5981 TALLOC_CTX *frame = talloc_stackframe();
5982 struct tevent_context *ev = NULL;
5983 struct tevent_req *req = NULL;
5984 NTSTATUS status = NT_STATUS_OK;
5985
5986 if (smbXcli_conn_has_async_calls(cli->conn)) {
5987 /*
5988 * Can't use sync call while an async call is in flight
5989 */
5990 status = NT_STATUS_INVALID_PARAMETER;
5991 goto fail;
5992 }
5993
5994 ev = samba_tevent_context_init(frame);
5995 if (ev == NULL) {
5996 status = NT_STATUS_NO_MEMORY;
5997 goto fail;
5998 }
5999
6000 req = cli_posix_rmdir_send(frame,
6001 ev,
6002 cli,
6003 fname);
6004 if (req == NULL) {
6005 status = NT_STATUS_NO_MEMORY;
6006 goto fail;
6007 }
6008
6009 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
6010 goto fail;
6011 }
6012
6013 status = cli_posix_rmdir_recv(req, frame);
6014
6015 fail:
6016 TALLOC_FREE(frame);
6017 return status;
6018 }
6019
6020 /****************************************************************************
6021 filechangenotify
6022 ****************************************************************************/
6023
6024 struct cli_notify_state {
6025 struct tevent_req *subreq;
6026 uint8_t setup[8];
6027 uint32_t num_changes;
6028 struct notify_change *changes;
6029 };
6030
6031 static void cli_notify_done(struct tevent_req *subreq);
6032 static void cli_notify_done_smb2(struct tevent_req *subreq);
6033 static bool cli_notify_cancel(struct tevent_req *req);
6034
cli_notify_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint32_t buffer_size,uint32_t completion_filter,bool recursive)6035 struct tevent_req *cli_notify_send(TALLOC_CTX *mem_ctx,
6036 struct tevent_context *ev,
6037 struct cli_state *cli, uint16_t fnum,
6038 uint32_t buffer_size,
6039 uint32_t completion_filter, bool recursive)
6040 {
6041 struct tevent_req *req;
6042 struct cli_notify_state *state;
6043 unsigned old_timeout;
6044
6045 req = tevent_req_create(mem_ctx, &state, struct cli_notify_state);
6046 if (req == NULL) {
6047 return NULL;
6048 }
6049
6050 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
6051 /*
6052 * Notifies should not time out
6053 */
6054 old_timeout = cli_set_timeout(cli, 0);
6055
6056 state->subreq = cli_smb2_notify_send(
6057 state,
6058 ev,
6059 cli,
6060 fnum,
6061 buffer_size,
6062 completion_filter,
6063 recursive);
6064
6065 cli_set_timeout(cli, old_timeout);
6066
6067 if (tevent_req_nomem(state->subreq, req)) {
6068 return tevent_req_post(req, ev);
6069 }
6070 tevent_req_set_callback(
6071 state->subreq, cli_notify_done_smb2, req);
6072 goto done;
6073 }
6074
6075 SIVAL(state->setup, 0, completion_filter);
6076 SSVAL(state->setup, 4, fnum);
6077 SSVAL(state->setup, 6, recursive);
6078
6079 /*
6080 * Notifies should not time out
6081 */
6082 old_timeout = cli_set_timeout(cli, 0);
6083
6084 state->subreq = cli_trans_send(
6085 state, /* mem ctx. */
6086 ev, /* event ctx. */
6087 cli, /* cli_state. */
6088 0, /* additional_flags2 */
6089 SMBnttrans, /* cmd. */
6090 NULL, /* pipe name. */
6091 -1, /* fid. */
6092 NT_TRANSACT_NOTIFY_CHANGE, /* function. */
6093 0, /* flags. */
6094 (uint16_t *)state->setup, /* setup. */
6095 4, /* num setup uint16_t words. */
6096 0, /* max returned setup. */
6097 NULL, /* param. */
6098 0, /* num param. */
6099 buffer_size, /* max returned param. */
6100 NULL, /* data. */
6101 0, /* num data. */
6102 0); /* max returned data. */
6103
6104 cli_set_timeout(cli, old_timeout);
6105
6106 if (tevent_req_nomem(state->subreq, req)) {
6107 return tevent_req_post(req, ev);
6108 }
6109 tevent_req_set_callback(state->subreq, cli_notify_done, req);
6110 done:
6111 tevent_req_set_cancel_fn(req, cli_notify_cancel);
6112 return req;
6113 }
6114
cli_notify_cancel(struct tevent_req * req)6115 static bool cli_notify_cancel(struct tevent_req *req)
6116 {
6117 struct cli_notify_state *state = tevent_req_data(
6118 req, struct cli_notify_state);
6119 bool ok;
6120
6121 ok = tevent_req_cancel(state->subreq);
6122 return ok;
6123 }
6124
cli_notify_done(struct tevent_req * subreq)6125 static void cli_notify_done(struct tevent_req *subreq)
6126 {
6127 struct tevent_req *req = tevent_req_callback_data(
6128 subreq, struct tevent_req);
6129 struct cli_notify_state *state = tevent_req_data(
6130 req, struct cli_notify_state);
6131 NTSTATUS status;
6132 uint8_t *params;
6133 uint32_t i, ofs, num_params;
6134 uint16_t flags2;
6135
6136 status = cli_trans_recv(subreq, talloc_tos(), &flags2, NULL, 0, NULL,
6137 ¶ms, 0, &num_params, NULL, 0, NULL);
6138 TALLOC_FREE(subreq);
6139 state->subreq = NULL;
6140 if (tevent_req_nterror(req, status)) {
6141 DEBUG(10, ("cli_trans_recv returned %s\n", nt_errstr(status)));
6142 return;
6143 }
6144
6145 state->num_changes = 0;
6146 ofs = 0;
6147
6148 while (num_params - ofs > 12) {
6149 uint32_t next = IVAL(params, ofs);
6150 state->num_changes += 1;
6151
6152 if ((next == 0) || (ofs+next >= num_params)) {
6153 break;
6154 }
6155 ofs += next;
6156 }
6157
6158 state->changes = talloc_array(state, struct notify_change,
6159 state->num_changes);
6160 if (tevent_req_nomem(state->changes, req)) {
6161 TALLOC_FREE(params);
6162 return;
6163 }
6164
6165 ofs = 0;
6166
6167 for (i=0; i<state->num_changes; i++) {
6168 uint32_t next = IVAL(params, ofs);
6169 uint32_t len = IVAL(params, ofs+8);
6170 ssize_t ret;
6171 char *name;
6172
6173 if (trans_oob(num_params, ofs + 12, len)) {
6174 TALLOC_FREE(params);
6175 tevent_req_nterror(
6176 req, NT_STATUS_INVALID_NETWORK_RESPONSE);
6177 return;
6178 }
6179
6180 state->changes[i].action = IVAL(params, ofs+4);
6181 ret = clistr_pull_talloc(state->changes, (char *)params, flags2,
6182 &name, params+ofs+12, len,
6183 STR_TERMINATE|STR_UNICODE);
6184 if (ret == -1) {
6185 TALLOC_FREE(params);
6186 tevent_req_nterror(req, NT_STATUS_INTERNAL_ERROR);
6187 return;
6188 }
6189 state->changes[i].name = name;
6190 ofs += next;
6191 }
6192
6193 TALLOC_FREE(params);
6194 tevent_req_done(req);
6195 }
6196
cli_notify_done_smb2(struct tevent_req * subreq)6197 static void cli_notify_done_smb2(struct tevent_req *subreq)
6198 {
6199 struct tevent_req *req = tevent_req_callback_data(
6200 subreq, struct tevent_req);
6201 struct cli_notify_state *state = tevent_req_data(
6202 req, struct cli_notify_state);
6203 NTSTATUS status;
6204
6205 status = cli_smb2_notify_recv(
6206 subreq,
6207 state,
6208 &state->changes,
6209 &state->num_changes);
6210 TALLOC_FREE(subreq);
6211 if (tevent_req_nterror(req, status)) {
6212 return;
6213 }
6214 tevent_req_done(req);
6215 }
6216
cli_notify_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx,uint32_t * pnum_changes,struct notify_change ** pchanges)6217 NTSTATUS cli_notify_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
6218 uint32_t *pnum_changes,
6219 struct notify_change **pchanges)
6220 {
6221 struct cli_notify_state *state = tevent_req_data(
6222 req, struct cli_notify_state);
6223 NTSTATUS status;
6224
6225 if (tevent_req_is_nterror(req, &status)) {
6226 return status;
6227 }
6228
6229 *pnum_changes = state->num_changes;
6230 *pchanges = talloc_move(mem_ctx, &state->changes);
6231 return NT_STATUS_OK;
6232 }
6233
cli_notify(struct cli_state * cli,uint16_t fnum,uint32_t buffer_size,uint32_t completion_filter,bool recursive,TALLOC_CTX * mem_ctx,uint32_t * pnum_changes,struct notify_change ** pchanges)6234 NTSTATUS cli_notify(struct cli_state *cli, uint16_t fnum, uint32_t buffer_size,
6235 uint32_t completion_filter, bool recursive,
6236 TALLOC_CTX *mem_ctx, uint32_t *pnum_changes,
6237 struct notify_change **pchanges)
6238 {
6239 TALLOC_CTX *frame;
6240 struct tevent_context *ev;
6241 struct tevent_req *req;
6242 NTSTATUS status = NT_STATUS_NO_MEMORY;
6243
6244 frame = talloc_stackframe();
6245
6246 if (smbXcli_conn_has_async_calls(cli->conn)) {
6247 /*
6248 * Can't use sync call while an async call is in flight
6249 */
6250 status = NT_STATUS_INVALID_PARAMETER;
6251 goto fail;
6252 }
6253 ev = samba_tevent_context_init(frame);
6254 if (ev == NULL) {
6255 goto fail;
6256 }
6257 req = cli_notify_send(ev, ev, cli, fnum, buffer_size,
6258 completion_filter, recursive);
6259 if (req == NULL) {
6260 goto fail;
6261 }
6262 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
6263 goto fail;
6264 }
6265 status = cli_notify_recv(req, mem_ctx, pnum_changes, pchanges);
6266 fail:
6267 TALLOC_FREE(frame);
6268 return status;
6269 }
6270
6271 struct cli_qpathinfo_state {
6272 uint8_t *param;
6273 uint8_t *data;
6274 uint16_t setup[1];
6275 uint32_t min_rdata;
6276 uint8_t *rdata;
6277 uint32_t num_rdata;
6278 };
6279
6280 static void cli_qpathinfo_done(struct tevent_req *subreq);
6281
cli_qpathinfo_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,const char * fname,uint16_t level,uint32_t min_rdata,uint32_t max_rdata)6282 struct tevent_req *cli_qpathinfo_send(TALLOC_CTX *mem_ctx,
6283 struct tevent_context *ev,
6284 struct cli_state *cli, const char *fname,
6285 uint16_t level, uint32_t min_rdata,
6286 uint32_t max_rdata)
6287 {
6288 struct tevent_req *req, *subreq;
6289 struct cli_qpathinfo_state *state;
6290 uint16_t additional_flags2 = 0;
6291
6292 req = tevent_req_create(mem_ctx, &state, struct cli_qpathinfo_state);
6293 if (req == NULL) {
6294 return NULL;
6295 }
6296 state->min_rdata = min_rdata;
6297 SSVAL(state->setup, 0, TRANSACT2_QPATHINFO);
6298
6299 state->param = talloc_zero_array(state, uint8_t, 6);
6300 if (tevent_req_nomem(state->param, req)) {
6301 return tevent_req_post(req, ev);
6302 }
6303 SSVAL(state->param, 0, level);
6304 state->param = trans2_bytes_push_str(
6305 state->param, smbXcli_conn_use_unicode(cli->conn), fname, strlen(fname)+1, NULL);
6306 if (tevent_req_nomem(state->param, req)) {
6307 return tevent_req_post(req, ev);
6308 }
6309
6310 if (clistr_is_previous_version_path(fname, NULL, NULL, NULL) &&
6311 !INFO_LEVEL_IS_UNIX(level)) {
6312 additional_flags2 = FLAGS2_REPARSE_PATH;
6313 }
6314
6315 subreq = cli_trans_send(
6316 state, /* mem ctx. */
6317 ev, /* event ctx. */
6318 cli, /* cli_state. */
6319 additional_flags2, /* additional_flags2 */
6320 SMBtrans2, /* cmd. */
6321 NULL, /* pipe name. */
6322 -1, /* fid. */
6323 0, /* function. */
6324 0, /* flags. */
6325 state->setup, /* setup. */
6326 1, /* num setup uint16_t words. */
6327 0, /* max returned setup. */
6328 state->param, /* param. */
6329 talloc_get_size(state->param), /* num param. */
6330 2, /* max returned param. */
6331 NULL, /* data. */
6332 0, /* num data. */
6333 max_rdata); /* max returned data. */
6334
6335 if (tevent_req_nomem(subreq, req)) {
6336 return tevent_req_post(req, ev);
6337 }
6338 tevent_req_set_callback(subreq, cli_qpathinfo_done, req);
6339 return req;
6340 }
6341
cli_qpathinfo_done(struct tevent_req * subreq)6342 static void cli_qpathinfo_done(struct tevent_req *subreq)
6343 {
6344 struct tevent_req *req = tevent_req_callback_data(
6345 subreq, struct tevent_req);
6346 struct cli_qpathinfo_state *state = tevent_req_data(
6347 req, struct cli_qpathinfo_state);
6348 NTSTATUS status;
6349
6350 status = cli_trans_recv(subreq, state, NULL, NULL, 0, NULL,
6351 NULL, 0, NULL,
6352 &state->rdata, state->min_rdata,
6353 &state->num_rdata);
6354 if (tevent_req_nterror(req, status)) {
6355 return;
6356 }
6357 tevent_req_done(req);
6358 }
6359
cli_qpathinfo_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx,uint8_t ** rdata,uint32_t * num_rdata)6360 NTSTATUS cli_qpathinfo_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
6361 uint8_t **rdata, uint32_t *num_rdata)
6362 {
6363 struct cli_qpathinfo_state *state = tevent_req_data(
6364 req, struct cli_qpathinfo_state);
6365 NTSTATUS status;
6366
6367 if (tevent_req_is_nterror(req, &status)) {
6368 return status;
6369 }
6370 if (rdata != NULL) {
6371 *rdata = talloc_move(mem_ctx, &state->rdata);
6372 } else {
6373 TALLOC_FREE(state->rdata);
6374 }
6375 if (num_rdata != NULL) {
6376 *num_rdata = state->num_rdata;
6377 }
6378 return NT_STATUS_OK;
6379 }
6380
cli_qpathinfo(TALLOC_CTX * mem_ctx,struct cli_state * cli,const char * fname,uint16_t level,uint32_t min_rdata,uint32_t max_rdata,uint8_t ** rdata,uint32_t * num_rdata)6381 NTSTATUS cli_qpathinfo(TALLOC_CTX *mem_ctx, struct cli_state *cli,
6382 const char *fname, uint16_t level, uint32_t min_rdata,
6383 uint32_t max_rdata,
6384 uint8_t **rdata, uint32_t *num_rdata)
6385 {
6386 TALLOC_CTX *frame = talloc_stackframe();
6387 struct tevent_context *ev;
6388 struct tevent_req *req;
6389 NTSTATUS status = NT_STATUS_NO_MEMORY;
6390
6391 if (smbXcli_conn_has_async_calls(cli->conn)) {
6392 /*
6393 * Can't use sync call while an async call is in flight
6394 */
6395 status = NT_STATUS_INVALID_PARAMETER;
6396 goto fail;
6397 }
6398 ev = samba_tevent_context_init(frame);
6399 if (ev == NULL) {
6400 goto fail;
6401 }
6402 req = cli_qpathinfo_send(frame, ev, cli, fname, level, min_rdata,
6403 max_rdata);
6404 if (req == NULL) {
6405 goto fail;
6406 }
6407 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
6408 goto fail;
6409 }
6410 status = cli_qpathinfo_recv(req, mem_ctx, rdata, num_rdata);
6411 fail:
6412 TALLOC_FREE(frame);
6413 return status;
6414 }
6415
6416 struct cli_qfileinfo_state {
6417 uint16_t setup[1];
6418 uint8_t param[4];
6419 uint8_t *data;
6420 uint16_t recv_flags2;
6421 uint32_t min_rdata;
6422 uint8_t *rdata;
6423 uint32_t num_rdata;
6424 };
6425
6426 static void cli_qfileinfo_done(struct tevent_req *subreq);
6427
cli_qfileinfo_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,uint16_t level,uint32_t min_rdata,uint32_t max_rdata)6428 struct tevent_req *cli_qfileinfo_send(TALLOC_CTX *mem_ctx,
6429 struct tevent_context *ev,
6430 struct cli_state *cli, uint16_t fnum,
6431 uint16_t level, uint32_t min_rdata,
6432 uint32_t max_rdata)
6433 {
6434 struct tevent_req *req, *subreq;
6435 struct cli_qfileinfo_state *state;
6436
6437 req = tevent_req_create(mem_ctx, &state, struct cli_qfileinfo_state);
6438 if (req == NULL) {
6439 return NULL;
6440 }
6441 state->min_rdata = min_rdata;
6442 SSVAL(state->param, 0, fnum);
6443 SSVAL(state->param, 2, level);
6444 SSVAL(state->setup, 0, TRANSACT2_QFILEINFO);
6445
6446 subreq = cli_trans_send(
6447 state, /* mem ctx. */
6448 ev, /* event ctx. */
6449 cli, /* cli_state. */
6450 0, /* additional_flags2 */
6451 SMBtrans2, /* cmd. */
6452 NULL, /* pipe name. */
6453 -1, /* fid. */
6454 0, /* function. */
6455 0, /* flags. */
6456 state->setup, /* setup. */
6457 1, /* num setup uint16_t words. */
6458 0, /* max returned setup. */
6459 state->param, /* param. */
6460 sizeof(state->param), /* num param. */
6461 2, /* max returned param. */
6462 NULL, /* data. */
6463 0, /* num data. */
6464 max_rdata); /* max returned data. */
6465
6466 if (tevent_req_nomem(subreq, req)) {
6467 return tevent_req_post(req, ev);
6468 }
6469 tevent_req_set_callback(subreq, cli_qfileinfo_done, req);
6470 return req;
6471 }
6472
cli_qfileinfo_done(struct tevent_req * subreq)6473 static void cli_qfileinfo_done(struct tevent_req *subreq)
6474 {
6475 struct tevent_req *req = tevent_req_callback_data(
6476 subreq, struct tevent_req);
6477 struct cli_qfileinfo_state *state = tevent_req_data(
6478 req, struct cli_qfileinfo_state);
6479 NTSTATUS status;
6480
6481 status = cli_trans_recv(subreq, state,
6482 &state->recv_flags2,
6483 NULL, 0, NULL,
6484 NULL, 0, NULL,
6485 &state->rdata, state->min_rdata,
6486 &state->num_rdata);
6487 if (tevent_req_nterror(req, status)) {
6488 return;
6489 }
6490 tevent_req_done(req);
6491 }
6492
cli_qfileinfo_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx,uint16_t * recv_flags2,uint8_t ** rdata,uint32_t * num_rdata)6493 NTSTATUS cli_qfileinfo_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
6494 uint16_t *recv_flags2,
6495 uint8_t **rdata, uint32_t *num_rdata)
6496 {
6497 struct cli_qfileinfo_state *state = tevent_req_data(
6498 req, struct cli_qfileinfo_state);
6499 NTSTATUS status;
6500
6501 if (tevent_req_is_nterror(req, &status)) {
6502 return status;
6503 }
6504
6505 if (recv_flags2 != NULL) {
6506 *recv_flags2 = state->recv_flags2;
6507 }
6508 if (rdata != NULL) {
6509 *rdata = talloc_move(mem_ctx, &state->rdata);
6510 } else {
6511 TALLOC_FREE(state->rdata);
6512 }
6513 if (num_rdata != NULL) {
6514 *num_rdata = state->num_rdata;
6515 }
6516 return NT_STATUS_OK;
6517 }
6518
cli_qfileinfo(TALLOC_CTX * mem_ctx,struct cli_state * cli,uint16_t fnum,uint16_t level,uint32_t min_rdata,uint32_t max_rdata,uint16_t * recv_flags2,uint8_t ** rdata,uint32_t * num_rdata)6519 NTSTATUS cli_qfileinfo(TALLOC_CTX *mem_ctx, struct cli_state *cli,
6520 uint16_t fnum, uint16_t level, uint32_t min_rdata,
6521 uint32_t max_rdata, uint16_t *recv_flags2,
6522 uint8_t **rdata, uint32_t *num_rdata)
6523 {
6524 TALLOC_CTX *frame = talloc_stackframe();
6525 struct tevent_context *ev;
6526 struct tevent_req *req;
6527 NTSTATUS status = NT_STATUS_NO_MEMORY;
6528
6529 if (smbXcli_conn_has_async_calls(cli->conn)) {
6530 /*
6531 * Can't use sync call while an async call is in flight
6532 */
6533 status = NT_STATUS_INVALID_PARAMETER;
6534 goto fail;
6535 }
6536 ev = samba_tevent_context_init(frame);
6537 if (ev == NULL) {
6538 goto fail;
6539 }
6540 req = cli_qfileinfo_send(frame, ev, cli, fnum, level, min_rdata,
6541 max_rdata);
6542 if (req == NULL) {
6543 goto fail;
6544 }
6545 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
6546 goto fail;
6547 }
6548 status = cli_qfileinfo_recv(req, mem_ctx, recv_flags2, rdata, num_rdata);
6549 fail:
6550 TALLOC_FREE(frame);
6551 return status;
6552 }
6553
6554 struct cli_flush_state {
6555 uint16_t vwv[1];
6556 };
6557
6558 static void cli_flush_done(struct tevent_req *subreq);
6559
cli_flush_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum)6560 struct tevent_req *cli_flush_send(TALLOC_CTX *mem_ctx,
6561 struct tevent_context *ev,
6562 struct cli_state *cli,
6563 uint16_t fnum)
6564 {
6565 struct tevent_req *req, *subreq;
6566 struct cli_flush_state *state;
6567
6568 req = tevent_req_create(mem_ctx, &state, struct cli_flush_state);
6569 if (req == NULL) {
6570 return NULL;
6571 }
6572 SSVAL(state->vwv + 0, 0, fnum);
6573
6574 subreq = cli_smb_send(state, ev, cli, SMBflush, 0, 0, 1, state->vwv,
6575 0, NULL);
6576 if (tevent_req_nomem(subreq, req)) {
6577 return tevent_req_post(req, ev);
6578 }
6579 tevent_req_set_callback(subreq, cli_flush_done, req);
6580 return req;
6581 }
6582
cli_flush_done(struct tevent_req * subreq)6583 static void cli_flush_done(struct tevent_req *subreq)
6584 {
6585 struct tevent_req *req = tevent_req_callback_data(
6586 subreq, struct tevent_req);
6587 NTSTATUS status;
6588
6589 status = cli_smb_recv(subreq, NULL, NULL, 0, NULL, NULL, NULL, NULL);
6590 TALLOC_FREE(subreq);
6591 if (tevent_req_nterror(req, status)) {
6592 return;
6593 }
6594 tevent_req_done(req);
6595 }
6596
cli_flush_recv(struct tevent_req * req)6597 NTSTATUS cli_flush_recv(struct tevent_req *req)
6598 {
6599 return tevent_req_simple_recv_ntstatus(req);
6600 }
6601
cli_flush(TALLOC_CTX * mem_ctx,struct cli_state * cli,uint16_t fnum)6602 NTSTATUS cli_flush(TALLOC_CTX *mem_ctx, struct cli_state *cli, uint16_t fnum)
6603 {
6604 TALLOC_CTX *frame = talloc_stackframe();
6605 struct tevent_context *ev;
6606 struct tevent_req *req;
6607 NTSTATUS status = NT_STATUS_NO_MEMORY;
6608
6609 if (smbXcli_conn_has_async_calls(cli->conn)) {
6610 /*
6611 * Can't use sync call while an async call is in flight
6612 */
6613 status = NT_STATUS_INVALID_PARAMETER;
6614 goto fail;
6615 }
6616 ev = samba_tevent_context_init(frame);
6617 if (ev == NULL) {
6618 goto fail;
6619 }
6620 req = cli_flush_send(frame, ev, cli, fnum);
6621 if (req == NULL) {
6622 goto fail;
6623 }
6624 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
6625 goto fail;
6626 }
6627 status = cli_flush_recv(req);
6628 fail:
6629 TALLOC_FREE(frame);
6630 return status;
6631 }
6632
6633 struct cli_shadow_copy_data_state {
6634 uint16_t setup[4];
6635 uint8_t *data;
6636 uint32_t num_data;
6637 bool get_names;
6638 };
6639
6640 static void cli_shadow_copy_data_done(struct tevent_req *subreq);
6641
cli_shadow_copy_data_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct cli_state * cli,uint16_t fnum,bool get_names)6642 struct tevent_req *cli_shadow_copy_data_send(TALLOC_CTX *mem_ctx,
6643 struct tevent_context *ev,
6644 struct cli_state *cli,
6645 uint16_t fnum,
6646 bool get_names)
6647 {
6648 struct tevent_req *req, *subreq;
6649 struct cli_shadow_copy_data_state *state;
6650 uint32_t ret_size;
6651
6652 req = tevent_req_create(mem_ctx, &state,
6653 struct cli_shadow_copy_data_state);
6654 if (req == NULL) {
6655 return NULL;
6656 }
6657 state->get_names = get_names;
6658 ret_size = get_names ? CLI_BUFFER_SIZE : 16;
6659
6660 SIVAL(state->setup + 0, 0, FSCTL_GET_SHADOW_COPY_DATA);
6661 SSVAL(state->setup + 2, 0, fnum);
6662 SCVAL(state->setup + 3, 0, 1); /* isFsctl */
6663 SCVAL(state->setup + 3, 1, 0); /* compfilter, isFlags (WSSP) */
6664
6665 subreq = cli_trans_send(
6666 state, ev, cli, 0, SMBnttrans, NULL, 0, NT_TRANSACT_IOCTL, 0,
6667 state->setup, ARRAY_SIZE(state->setup),
6668 ARRAY_SIZE(state->setup),
6669 NULL, 0, 0,
6670 NULL, 0, ret_size);
6671 if (tevent_req_nomem(subreq, req)) {
6672 return tevent_req_post(req, ev);
6673 }
6674 tevent_req_set_callback(subreq, cli_shadow_copy_data_done, req);
6675 return req;
6676 }
6677
cli_shadow_copy_data_done(struct tevent_req * subreq)6678 static void cli_shadow_copy_data_done(struct tevent_req *subreq)
6679 {
6680 struct tevent_req *req = tevent_req_callback_data(
6681 subreq, struct tevent_req);
6682 struct cli_shadow_copy_data_state *state = tevent_req_data(
6683 req, struct cli_shadow_copy_data_state);
6684 NTSTATUS status;
6685
6686 status = cli_trans_recv(subreq, state, NULL,
6687 NULL, 0, NULL, /* setup */
6688 NULL, 0, NULL, /* param */
6689 &state->data, 12, &state->num_data);
6690 TALLOC_FREE(subreq);
6691 if (tevent_req_nterror(req, status)) {
6692 return;
6693 }
6694 tevent_req_done(req);
6695 }
6696
cli_shadow_copy_data_recv(struct tevent_req * req,TALLOC_CTX * mem_ctx,char *** pnames,int * pnum_names)6697 NTSTATUS cli_shadow_copy_data_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
6698 char ***pnames, int *pnum_names)
6699 {
6700 struct cli_shadow_copy_data_state *state = tevent_req_data(
6701 req, struct cli_shadow_copy_data_state);
6702 char **names = NULL;
6703 uint32_t i, num_names;
6704 uint32_t dlength;
6705 uint8_t *endp = NULL;
6706 NTSTATUS status;
6707
6708 if (tevent_req_is_nterror(req, &status)) {
6709 return status;
6710 }
6711
6712 if (state->num_data < 16) {
6713 return NT_STATUS_INVALID_NETWORK_RESPONSE;
6714 }
6715
6716 num_names = IVAL(state->data, 4);
6717 dlength = IVAL(state->data, 8);
6718
6719 if (num_names > 0x7FFFFFFF) {
6720 return NT_STATUS_INVALID_NETWORK_RESPONSE;
6721 }
6722
6723 if (!state->get_names) {
6724 *pnum_names = (int)num_names;
6725 return NT_STATUS_OK;
6726 }
6727
6728 if (dlength + 12 < 12) {
6729 return NT_STATUS_INVALID_NETWORK_RESPONSE;
6730 }
6731 if (dlength + 12 > state->num_data) {
6732 return NT_STATUS_INVALID_NETWORK_RESPONSE;
6733 }
6734 if (state->num_data + (2 * sizeof(SHADOW_COPY_LABEL)) <
6735 state->num_data) {
6736 return NT_STATUS_INVALID_NETWORK_RESPONSE;
6737 }
6738
6739 names = talloc_array(mem_ctx, char *, num_names);
6740 if (names == NULL) {
6741 return NT_STATUS_NO_MEMORY;
6742 }
6743
6744 endp = state->data + state->num_data;
6745
6746 for (i=0; i<num_names; i++) {
6747 bool ret;
6748 uint8_t *src;
6749 size_t converted_size;
6750
6751 src = state->data + 12 + i * 2 * sizeof(SHADOW_COPY_LABEL);
6752
6753 if (src + (2 * sizeof(SHADOW_COPY_LABEL)) > endp) {
6754 return NT_STATUS_INVALID_NETWORK_RESPONSE;
6755 }
6756
6757 ret = convert_string_talloc(
6758 names, CH_UTF16LE, CH_UNIX,
6759 src, 2 * sizeof(SHADOW_COPY_LABEL),
6760 &names[i], &converted_size);
6761 if (!ret) {
6762 TALLOC_FREE(names);
6763 return NT_STATUS_INVALID_NETWORK_RESPONSE;
6764 }
6765 }
6766 *pnum_names = (int)num_names;
6767 *pnames = names;
6768 return NT_STATUS_OK;
6769 }
6770
cli_shadow_copy_data(TALLOC_CTX * mem_ctx,struct cli_state * cli,uint16_t fnum,bool get_names,char *** pnames,int * pnum_names)6771 NTSTATUS cli_shadow_copy_data(TALLOC_CTX *mem_ctx, struct cli_state *cli,
6772 uint16_t fnum, bool get_names,
6773 char ***pnames, int *pnum_names)
6774 {
6775 TALLOC_CTX *frame = NULL;
6776 struct tevent_context *ev;
6777 struct tevent_req *req;
6778 NTSTATUS status = NT_STATUS_NO_MEMORY;
6779
6780 if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
6781 return cli_smb2_shadow_copy_data(mem_ctx,
6782 cli,
6783 fnum,
6784 get_names,
6785 pnames,
6786 pnum_names);
6787 }
6788
6789 frame = talloc_stackframe();
6790
6791 if (smbXcli_conn_has_async_calls(cli->conn)) {
6792 /*
6793 * Can't use sync call while an async call is in flight
6794 */
6795 status = NT_STATUS_INVALID_PARAMETER;
6796 goto fail;
6797 }
6798 ev = samba_tevent_context_init(frame);
6799 if (ev == NULL) {
6800 goto fail;
6801 }
6802 req = cli_shadow_copy_data_send(frame, ev, cli, fnum, get_names);
6803 if (req == NULL) {
6804 goto fail;
6805 }
6806 if (!tevent_req_poll_ntstatus(req, ev, &status)) {
6807 goto fail;
6808 }
6809 status = cli_shadow_copy_data_recv(req, mem_ctx, pnames, pnum_names);
6810 fail:
6811 TALLOC_FREE(frame);
6812 return status;
6813 }
6814