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