?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Convert cli_close to tevent_req
[Samba.git] / source3 / libsmb / clifile.c
blobc492f2b01f2b557b2d8dadf6e860b0cdf4e56117
1 /*
2 Unix SMB/CIFS implementation.
3 client file operations
4 Copyright (C) Andrew Tridgell 1994-1998
5 Copyright (C) Jeremy Allison 2001-2002
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
23 /****************************************************************************
24 Hard/Symlink a file (UNIX extensions).
25 Creates new name (sym)linked to oldname.
26 ****************************************************************************/
27
28 static bool cli_link_internal(struct cli_state *cli, const char *oldname, const char *newname, bool hard_link)
29 {
30 unsigned int data_len = 0;
31 unsigned int param_len = 0;
32 uint16 setup = TRANSACT2_SETPATHINFO;
33 char *param;
34 char *data;
35 char *rparam=NULL, *rdata=NULL;
36 char *p;
37 size_t oldlen = 2*(strlen(oldname)+1);
38 size_t newlen = 2*(strlen(newname)+1);
39
40 param = SMB_MALLOC_ARRAY(char, 6+newlen+2);
41
42 if (!param) {
43 return false;
44 }
45
46 data = SMB_MALLOC_ARRAY(char, oldlen+2);
47
48 if (!data) {
49 SAFE_FREE(param);
50 return false;
51 }
52
53 SSVAL(param,0,hard_link ? SMB_SET_FILE_UNIX_HLINK : SMB_SET_FILE_UNIX_LINK);
54 SIVAL(param,2,0);
55 p = &param[6];
56
57 p += clistr_push(cli, p, newname, newlen, STR_TERMINATE);
58 param_len = PTR_DIFF(p, param);
59
60 p = data;
61 p += clistr_push(cli, p, oldname, oldlen, STR_TERMINATE);
62 data_len = PTR_DIFF(p, data);
63
64 if (!cli_send_trans(cli, SMBtrans2,
65 NULL, /* name */
66 -1, 0, /* fid, flags */
67 &setup, 1, 0, /* setup, length, max */
68 param, param_len, 2, /* param, length, max */
69 data, data_len, cli->max_xmit /* data, length, max */
70 )) {
71 SAFE_FREE(data);
72 SAFE_FREE(param);
73 return false;
74 }
75
76 SAFE_FREE(data);
77 SAFE_FREE(param);
78
79 if (!cli_receive_trans(cli, SMBtrans2,
80 &rparam, &param_len,
81 &rdata, &data_len)) {
82 return false;
83 }
84
85 SAFE_FREE(data);
86 SAFE_FREE(param);
87 SAFE_FREE(rdata);
88 SAFE_FREE(rparam);
89
90 return true;
91 }
92
93 /****************************************************************************
94 Map standard UNIX permissions onto wire representations.
95 ****************************************************************************/
96
97 uint32 unix_perms_to_wire(mode_t perms)
98 {
99 unsigned int ret = 0;
100
101 ret |= ((perms & S_IXOTH) ? UNIX_X_OTH : 0);
102 ret |= ((perms & S_IWOTH) ? UNIX_W_OTH : 0);
103 ret |= ((perms & S_IROTH) ? UNIX_R_OTH : 0);
104 ret |= ((perms & S_IXGRP) ? UNIX_X_GRP : 0);
105 ret |= ((perms & S_IWGRP) ? UNIX_W_GRP : 0);
106 ret |= ((perms & S_IRGRP) ? UNIX_R_GRP : 0);
107 ret |= ((perms & S_IXUSR) ? UNIX_X_USR : 0);
108 ret |= ((perms & S_IWUSR) ? UNIX_W_USR : 0);
109 ret |= ((perms & S_IRUSR) ? UNIX_R_USR : 0);
110 #ifdef S_ISVTX
111 ret |= ((perms & S_ISVTX) ? UNIX_STICKY : 0);
112 #endif
113 #ifdef S_ISGID
114 ret |= ((perms & S_ISGID) ? UNIX_SET_GID : 0);
115 #endif
116 #ifdef S_ISUID
117 ret |= ((perms & S_ISUID) ? UNIX_SET_UID : 0);
118 #endif
119 return ret;
120 }
121
122 /****************************************************************************
123 Map wire permissions to standard UNIX.
124 ****************************************************************************/
125
126 mode_t wire_perms_to_unix(uint32 perms)
127 {
128 mode_t ret = (mode_t)0;
129
130 ret |= ((perms & UNIX_X_OTH) ? S_IXOTH : 0);
131 ret |= ((perms & UNIX_W_OTH) ? S_IWOTH : 0);
132 ret |= ((perms & UNIX_R_OTH) ? S_IROTH : 0);
133 ret |= ((perms & UNIX_X_GRP) ? S_IXGRP : 0);
134 ret |= ((perms & UNIX_W_GRP) ? S_IWGRP : 0);
135 ret |= ((perms & UNIX_R_GRP) ? S_IRGRP : 0);
136 ret |= ((perms & UNIX_X_USR) ? S_IXUSR : 0);
137 ret |= ((perms & UNIX_W_USR) ? S_IWUSR : 0);
138 ret |= ((perms & UNIX_R_USR) ? S_IRUSR : 0);
139 #ifdef S_ISVTX
140 ret |= ((perms & UNIX_STICKY) ? S_ISVTX : 0);
141 #endif
142 #ifdef S_ISGID
143 ret |= ((perms & UNIX_SET_GID) ? S_ISGID : 0);
144 #endif
145 #ifdef S_ISUID
146 ret |= ((perms & UNIX_SET_UID) ? S_ISUID : 0);
147 #endif
148 return ret;
149 }
150
151 /****************************************************************************
152 Return the file type from the wire filetype for UNIX extensions.
153 ****************************************************************************/
154
155 static mode_t unix_filetype_from_wire(uint32 wire_type)
156 {
157 switch (wire_type) {
158 case UNIX_TYPE_FILE:
159 return S_IFREG;
160 case UNIX_TYPE_DIR:
161 return S_IFDIR;
162 #ifdef S_IFLNK
163 case UNIX_TYPE_SYMLINK:
164 return S_IFLNK;
165 #endif
166 #ifdef S_IFCHR
167 case UNIX_TYPE_CHARDEV:
168 return S_IFCHR;
169 #endif
170 #ifdef S_IFBLK
171 case UNIX_TYPE_BLKDEV:
172 return S_IFBLK;
173 #endif
174 #ifdef S_IFIFO
175 case UNIX_TYPE_FIFO:
176 return S_IFIFO;
177 #endif
178 #ifdef S_IFSOCK
179 case UNIX_TYPE_SOCKET:
180 return S_IFSOCK;
181 #endif
182 default:
183 return (mode_t)0;
184 }
185 }
186
187 /****************************************************************************
188 Do a POSIX getfacl (UNIX extensions).
189 ****************************************************************************/
190
191 bool cli_unix_getfacl(struct cli_state *cli, const char *name, size_t *prb_size, char **retbuf)
192 {
193 unsigned int param_len = 0;
194 unsigned int data_len = 0;
195 uint16 setup = TRANSACT2_QPATHINFO;
196 char *param;
197 size_t nlen = 2*(strlen(name)+1);
198 char *rparam=NULL, *rdata=NULL;
199 char *p;
200
201 param = SMB_MALLOC_ARRAY(char, 6+nlen+2);
202 if (!param) {
203 return false;
204 }
205
206 p = param;
207 memset(p, '\0', 6);
208 SSVAL(p, 0, SMB_QUERY_POSIX_ACL);
209 p += 6;
210 p += clistr_push(cli, p, name, nlen, STR_TERMINATE);
211 param_len = PTR_DIFF(p, param);
212
213 if (!cli_send_trans(cli, SMBtrans2,
214 NULL, /* name */
215 -1, 0, /* fid, flags */
216 &setup, 1, 0, /* setup, length, max */
217 param, param_len, 2, /* param, length, max */
218 NULL, 0, cli->max_xmit /* data, length, max */
219 )) {
220 SAFE_FREE(param);
221 return false;
222 }
223
224 SAFE_FREE(param);
225
226 if (!cli_receive_trans(cli, SMBtrans2,
227 &rparam, &param_len,
228 &rdata, &data_len)) {
229 return false;
230 }
231
232 if (data_len < 6) {
233 SAFE_FREE(rdata);
234 SAFE_FREE(rparam);
235 return false;
236 }
237
238 SAFE_FREE(rparam);
239 *retbuf = rdata;
240 *prb_size = (size_t)data_len;
241
242 return true;
243 }
244
245 /****************************************************************************
246 Stat a file (UNIX extensions).
247 ****************************************************************************/
248
249 bool cli_unix_stat(struct cli_state *cli, const char *name, SMB_STRUCT_STAT *sbuf)
250 {
251 unsigned int param_len = 0;
252 unsigned int data_len = 0;
253 uint16 setup = TRANSACT2_QPATHINFO;
254 char *param;
255 size_t nlen = 2*(strlen(name)+1);
256 char *rparam=NULL, *rdata=NULL;
257 char *p;
258
259 ZERO_STRUCTP(sbuf);
260
261 param = SMB_MALLOC_ARRAY(char, 6+nlen+2);
262 if (!param) {
263 return false;
264 }
265 p = param;
266 memset(p, '\0', 6);
267 SSVAL(p, 0, SMB_QUERY_FILE_UNIX_BASIC);
268 p += 6;
269 p += clistr_push(cli, p, name, nlen, STR_TERMINATE);
270 param_len = PTR_DIFF(p, param);
271
272 if (!cli_send_trans(cli, SMBtrans2,
273 NULL, /* name */
274 -1, 0, /* fid, flags */
275 &setup, 1, 0, /* setup, length, max */
276 param, param_len, 2, /* param, length, max */
277 NULL, 0, cli->max_xmit /* data, length, max */
278 )) {
279 SAFE_FREE(param);
280 return false;
281 }
282
283 SAFE_FREE(param);
284
285 if (!cli_receive_trans(cli, SMBtrans2,
286 &rparam, &param_len,
287 &rdata, &data_len)) {
288 return false;
289 }
290
291 if (data_len < 96) {
292 SAFE_FREE(rdata);
293 SAFE_FREE(rparam);
294 return false;
295 }
296
297 sbuf->st_size = IVAL2_TO_SMB_BIG_UINT(rdata,0); /* total size, in bytes */
298 sbuf->st_blocks = IVAL2_TO_SMB_BIG_UINT(rdata,8); /* number of blocks allocated */
299 #if defined (HAVE_STAT_ST_BLOCKS) && defined(STAT_ST_BLOCKSIZE)
300 sbuf->st_blocks /= STAT_ST_BLOCKSIZE;
301 #else
302 /* assume 512 byte blocks */
303 sbuf->st_blocks /= 512;
304 #endif
305 set_ctimespec(sbuf, interpret_long_date(rdata + 16)); /* time of last change */
306 set_atimespec(sbuf, interpret_long_date(rdata + 24)); /* time of last access */
307 set_mtimespec(sbuf, interpret_long_date(rdata + 32)); /* time of last modification */
308
309 sbuf->st_uid = (uid_t) IVAL(rdata,40); /* user ID of owner */
310 sbuf->st_gid = (gid_t) IVAL(rdata,48); /* group ID of owner */
311 sbuf->st_mode |= unix_filetype_from_wire(IVAL(rdata, 56));
312 #if defined(HAVE_MAKEDEV)
313 {
314 uint32 dev_major = IVAL(rdata,60);
315 uint32 dev_minor = IVAL(rdata,68);
316 sbuf->st_rdev = makedev(dev_major, dev_minor);
317 }
318 #endif
319 sbuf->st_ino = (SMB_INO_T)IVAL2_TO_SMB_BIG_UINT(rdata,76); /* inode */
320 sbuf->st_mode |= wire_perms_to_unix(IVAL(rdata,84)); /* protection */
321 sbuf->st_nlink = IVAL(rdata,92); /* number of hard links */
322
323 SAFE_FREE(rdata);
324 SAFE_FREE(rparam);
325
326 return true;
327 }
328
329 /****************************************************************************
330 Symlink a file (UNIX extensions).
331 ****************************************************************************/
332
333 bool cli_unix_symlink(struct cli_state *cli, const char *oldname, const char *newname)
334 {
335 return cli_link_internal(cli, oldname, newname, False);
336 }
337
338 /****************************************************************************
339 Hard a file (UNIX extensions).
340 ****************************************************************************/
341
342 bool cli_unix_hardlink(struct cli_state *cli, const char *oldname, const char *newname)
343 {
344 return cli_link_internal(cli, oldname, newname, True);
345 }
346
347 /****************************************************************************
348 Chmod or chown a file internal (UNIX extensions).
349 ****************************************************************************/
350
351 static bool cli_unix_chmod_chown_internal(struct cli_state *cli, const char *fname, uint32 mode, uint32 uid, uint32 gid)
352 {
353 unsigned int data_len = 0;
354 unsigned int param_len = 0;
355 uint16 setup = TRANSACT2_SETPATHINFO;
356 size_t nlen = 2*(strlen(fname)+1);
357 char *param;
358 char data[100];
359 char *rparam=NULL, *rdata=NULL;
360 char *p;
361
362 param = SMB_MALLOC_ARRAY(char, 6+nlen+2);
363 if (!param) {
364 return false;
365 }
366 memset(param, '\0', 6);
367 memset(data, 0, sizeof(data));
368
369 SSVAL(param,0,SMB_SET_FILE_UNIX_BASIC);
370 p = &param[6];
371
372 p += clistr_push(cli, p, fname, nlen, STR_TERMINATE);
373 param_len = PTR_DIFF(p, param);
374
375 memset(data, 0xff, 40); /* Set all sizes/times to no change. */
376
377 SIVAL(data,40,uid);
378 SIVAL(data,48,gid);
379 SIVAL(data,84,mode);
380
381 data_len = 100;
382
383 if (!cli_send_trans(cli, SMBtrans2,
384 NULL, /* name */
385 -1, 0, /* fid, flags */
386 &setup, 1, 0, /* setup, length, max */
387 param, param_len, 2, /* param, length, max */
388 (char *)&data, data_len, cli->max_xmit /* data, length, max */
389 )) {
390 SAFE_FREE(param);
391 return False;
392 }
393
394 SAFE_FREE(param);
395
396 if (!cli_receive_trans(cli, SMBtrans2,
397 &rparam, &param_len,
398 &rdata, &data_len)) {
399 return false;
400 }
401
402 SAFE_FREE(rdata);
403 SAFE_FREE(rparam);
404
405 return true;
406 }
407
408 /****************************************************************************
409 chmod a file (UNIX extensions).
410 ****************************************************************************/
411
412 bool cli_unix_chmod(struct cli_state *cli, const char *fname, mode_t mode)
413 {
414 return cli_unix_chmod_chown_internal(cli, fname,
415 unix_perms_to_wire(mode), SMB_UID_NO_CHANGE, SMB_GID_NO_CHANGE);
416 }
417
418 /****************************************************************************
419 chown a file (UNIX extensions).
420 ****************************************************************************/
421
422 bool cli_unix_chown(struct cli_state *cli, const char *fname, uid_t uid, gid_t gid)
423 {
424 return cli_unix_chmod_chown_internal(cli, fname,
425 SMB_MODE_NO_CHANGE, (uint32)uid, (uint32)gid);
426 }
427
428 /****************************************************************************
429 Rename a file.
430 ****************************************************************************/
431
432 bool cli_rename(struct cli_state *cli, const char *fname_src, const char *fname_dst)
433 {
434 char *p;
435
436 memset(cli->outbuf,'\0',smb_size);
437 memset(cli->inbuf,'\0',smb_size);
438
439 cli_set_message(cli->outbuf,1, 0, true);
440
441 SCVAL(cli->outbuf,smb_com,SMBmv);
442 SSVAL(cli->outbuf,smb_tid,cli->cnum);
443 cli_setup_packet(cli);
444
445 SSVAL(cli->outbuf,smb_vwv0,aSYSTEM | aHIDDEN | aDIR);
446
447 p = smb_buf(cli->outbuf);
448 *p++ = 4;
449 p += clistr_push(cli, p, fname_src,
450 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
451 *p++ = 4;
452 p += clistr_push(cli, p, fname_dst,
453 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
454
455 cli_setup_bcc(cli, p);
456
457 cli_send_smb(cli);
458 if (!cli_receive_smb(cli)) {
459 return false;
460 }
461
462 if (cli_is_error(cli)) {
463 return false;
464 }
465
466 return true;
467 }
468
469 /****************************************************************************
470 NT Rename a file.
471 ****************************************************************************/
472
473 bool cli_ntrename(struct cli_state *cli, const char *fname_src, const char *fname_dst)
474 {
475 char *p;
476
477 memset(cli->outbuf,'\0',smb_size);
478 memset(cli->inbuf,'\0',smb_size);
479
480 cli_set_message(cli->outbuf, 4, 0, true);
481
482 SCVAL(cli->outbuf,smb_com,SMBntrename);
483 SSVAL(cli->outbuf,smb_tid,cli->cnum);
484 cli_setup_packet(cli);
485
486 SSVAL(cli->outbuf,smb_vwv0,aSYSTEM | aHIDDEN | aDIR);
487 SSVAL(cli->outbuf,smb_vwv1, RENAME_FLAG_RENAME);
488
489 p = smb_buf(cli->outbuf);
490 *p++ = 4;
491 p += clistr_push(cli, p, fname_src,
492 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
493 *p++ = 4;
494 p += clistr_push(cli, p, fname_dst,
495 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
496
497 cli_setup_bcc(cli, p);
498
499 cli_send_smb(cli);
500 if (!cli_receive_smb(cli)) {
501 return false;
502 }
503
504 if (cli_is_error(cli)) {
505 return false;
506 }
507
508 return true;
509 }
510
511 /****************************************************************************
512 NT hardlink a file.
513 ****************************************************************************/
514
515 bool cli_nt_hardlink(struct cli_state *cli, const char *fname_src, const char *fname_dst)
516 {
517 char *p;
518
519 memset(cli->outbuf,'\0',smb_size);
520 memset(cli->inbuf,'\0',smb_size);
521
522 cli_set_message(cli->outbuf, 4, 0, true);
523
524 SCVAL(cli->outbuf,smb_com,SMBntrename);
525 SSVAL(cli->outbuf,smb_tid,cli->cnum);
526 cli_setup_packet(cli);
527
528 SSVAL(cli->outbuf,smb_vwv0,aSYSTEM | aHIDDEN | aDIR);
529 SSVAL(cli->outbuf,smb_vwv1, RENAME_FLAG_HARD_LINK);
530
531 p = smb_buf(cli->outbuf);
532 *p++ = 4;
533 p += clistr_push(cli, p, fname_src,
534 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
535 *p++ = 4;
536 p += clistr_push(cli, p, fname_dst,
537 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
538
539 cli_setup_bcc(cli, p);
540
541 cli_send_smb(cli);
542 if (!cli_receive_smb(cli)) {
543 return false;
544 }
545
546 if (cli_is_error(cli)) {
547 return false;
548 }
549
550 return true;
551 }
552
553 /****************************************************************************
554 Delete a file.
555 ****************************************************************************/
556
557 bool cli_unlink_full(struct cli_state *cli, const char *fname, uint16 attrs)
558 {
559 char *p;
560
561 memset(cli->outbuf,'\0',smb_size);
562 memset(cli->inbuf,'\0',smb_size);
563
564 cli_set_message(cli->outbuf,1, 0, true);
565
566 SCVAL(cli->outbuf,smb_com,SMBunlink);
567 SSVAL(cli->outbuf,smb_tid,cli->cnum);
568 cli_setup_packet(cli);
569
570 SSVAL(cli->outbuf,smb_vwv0, attrs);
571
572 p = smb_buf(cli->outbuf);
573 *p++ = 4;
574 p += clistr_push(cli, p, fname,
575 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
576
577 cli_setup_bcc(cli, p);
578 cli_send_smb(cli);
579 if (!cli_receive_smb(cli)) {
580 return false;
581 }
582
583 if (cli_is_error(cli)) {
584 return false;
585 }
586
587 return true;
588 }
589
590 /****************************************************************************
591 Delete a file.
592 ****************************************************************************/
593
594 bool cli_unlink(struct cli_state *cli, const char *fname)
595 {
596 return cli_unlink_full(cli, fname, aSYSTEM | aHIDDEN);
597 }
598
599 /****************************************************************************
600 Create a directory.
601 ****************************************************************************/
602
603 bool cli_mkdir(struct cli_state *cli, const char *dname)
604 {
605 char *p;
606
607 memset(cli->outbuf,'\0',smb_size);
608 memset(cli->inbuf,'\0',smb_size);
609
610 cli_set_message(cli->outbuf,0, 0, true);
611
612 SCVAL(cli->outbuf,smb_com,SMBmkdir);
613 SSVAL(cli->outbuf,smb_tid,cli->cnum);
614 cli_setup_packet(cli);
615
616 p = smb_buf(cli->outbuf);
617 *p++ = 4;
618 p += clistr_push(cli, p, dname,
619 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
620
621 cli_setup_bcc(cli, p);
622
623 cli_send_smb(cli);
624 if (!cli_receive_smb(cli)) {
625 return False;
626 }
627
628 if (cli_is_error(cli)) {
629 return False;
630 }
631
632 return True;
633 }
634
635 /****************************************************************************
636 Remove a directory.
637 ****************************************************************************/
638
639 bool cli_rmdir(struct cli_state *cli, const char *dname)
640 {
641 char *p;
642
643 memset(cli->outbuf,'\0',smb_size);
644 memset(cli->inbuf,'\0',smb_size);
645
646 cli_set_message(cli->outbuf,0, 0, true);
647
648 SCVAL(cli->outbuf,smb_com,SMBrmdir);
649 SSVAL(cli->outbuf,smb_tid,cli->cnum);
650 cli_setup_packet(cli);
651
652 p = smb_buf(cli->outbuf);
653 *p++ = 4;
654 p += clistr_push(cli, p, dname,
655 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
656
657 cli_setup_bcc(cli, p);
658
659 cli_send_smb(cli);
660 if (!cli_receive_smb(cli)) {
661 return false;
662 }
663
664 if (cli_is_error(cli)) {
665 return false;
666 }
667
668 return true;
669 }
670
671 /****************************************************************************
672 Set or clear the delete on close flag.
673 ****************************************************************************/
674
675 int cli_nt_delete_on_close(struct cli_state *cli, int fnum, bool flag)
676 {
677 unsigned int data_len = 1;
678 unsigned int param_len = 6;
679 uint16 setup = TRANSACT2_SETFILEINFO;
680 char param[6];
681 unsigned char data;
682 char *rparam=NULL, *rdata=NULL;
683
684 memset(param, 0, param_len);
685 SSVAL(param,0,fnum);
686 SSVAL(param,2,SMB_SET_FILE_DISPOSITION_INFO);
687
688 data = flag ? 1 : 0;
689
690 if (!cli_send_trans(cli, SMBtrans2,
691 NULL, /* name */
692 -1, 0, /* fid, flags */
693 &setup, 1, 0, /* setup, length, max */
694 param, param_len, 2, /* param, length, max */
695 (char *)&data, data_len, cli->max_xmit /* data, length, max */
696 )) {
697 return false;
698 }
699
700 if (!cli_receive_trans(cli, SMBtrans2,
701 &rparam, &param_len,
702 &rdata, &data_len)) {
703 return false;
704 }
705
706 SAFE_FREE(rdata);
707 SAFE_FREE(rparam);
708
709 return true;
710 }
711
712 /****************************************************************************
713 Open a file - exposing the full horror of the NT API :-).
714 Used in smbtorture.
715 ****************************************************************************/
716
717 int cli_nt_create_full(struct cli_state *cli, const char *fname,
718 uint32 CreatFlags, uint32 DesiredAccess,
719 uint32 FileAttributes, uint32 ShareAccess,
720 uint32 CreateDisposition, uint32 CreateOptions,
721 uint8 SecurityFlags)
722 {
723 char *p;
724 int len;
725
726 memset(cli->outbuf,'\0',smb_size);
727 memset(cli->inbuf,'\0',smb_size);
728
729 cli_set_message(cli->outbuf,24,0, true);
730
731 SCVAL(cli->outbuf,smb_com,SMBntcreateX);
732 SSVAL(cli->outbuf,smb_tid,cli->cnum);
733 cli_setup_packet(cli);
734
735 SSVAL(cli->outbuf,smb_vwv0,0xFF);
736 if (cli->use_oplocks)
737 CreatFlags |= (REQUEST_OPLOCK|REQUEST_BATCH_OPLOCK);
738
739 SIVAL(cli->outbuf,smb_ntcreate_Flags, CreatFlags);
740 SIVAL(cli->outbuf,smb_ntcreate_RootDirectoryFid, 0x0);
741 SIVAL(cli->outbuf,smb_ntcreate_DesiredAccess, DesiredAccess);
742 SIVAL(cli->outbuf,smb_ntcreate_FileAttributes, FileAttributes);
743 SIVAL(cli->outbuf,smb_ntcreate_ShareAccess, ShareAccess);
744 SIVAL(cli->outbuf,smb_ntcreate_CreateDisposition, CreateDisposition);
745 SIVAL(cli->outbuf,smb_ntcreate_CreateOptions, CreateOptions);
746 SIVAL(cli->outbuf,smb_ntcreate_ImpersonationLevel, 0x02);
747 SCVAL(cli->outbuf,smb_ntcreate_SecurityFlags, SecurityFlags);
748
749 p = smb_buf(cli->outbuf);
750 /* this alignment and termination is critical for netapp filers. Don't change */
751 p += clistr_align_out(cli, p, 0);
752 len = clistr_push(cli, p, fname,
753 cli->bufsize - PTR_DIFF(p,cli->outbuf), 0);
754 p += len;
755 SSVAL(cli->outbuf,smb_ntcreate_NameLength, len);
756 /* sigh. this copes with broken netapp filer behaviour */
757 p += clistr_push(cli, p, "",
758 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
759
760 cli_setup_bcc(cli, p);
761
762 cli_send_smb(cli);
763 if (!cli_receive_smb(cli)) {
764 return -1;
765 }
766
767 if (cli_is_error(cli)) {
768 return -1;
769 }
770
771 return SVAL(cli->inbuf,smb_vwv2 + 1);
772 }
773
774 struct cli_ntcreate_state {
775 uint16_t vwv[24];
776 uint16_t fnum;
777 };
778
779 static void cli_ntcreate_done(struct tevent_req *subreq);
780
781 struct tevent_req *cli_ntcreate_send(TALLOC_CTX *mem_ctx,
782 struct event_context *ev,
783 struct cli_state *cli,
784 const char *fname,
785 uint32_t CreatFlags,
786 uint32_t DesiredAccess,
787 uint32_t FileAttributes,
788 uint32_t ShareAccess,
789 uint32_t CreateDisposition,
790 uint32_t CreateOptions,
791 uint8_t SecurityFlags)
792 {
793 struct tevent_req *req, *subreq;
794 struct cli_ntcreate_state *state;
795 uint16_t *vwv;
796 uint8_t *bytes;
797 size_t converted_len;
798
799 req = tevent_req_create(mem_ctx, &state, struct cli_ntcreate_state);
800 if (req == NULL) {
801 return NULL;
802 }
803 vwv = state->vwv;
804
805 SCVAL(vwv+0, 0, 0xFF);
806 SCVAL(vwv+0, 1, 0);
807 SSVAL(vwv+1, 0, 0);
808 SCVAL(vwv+2, 0, 0);
809
810 if (cli->use_oplocks) {
811 CreatFlags |= (REQUEST_OPLOCK|REQUEST_BATCH_OPLOCK);
812 }
813 SIVAL(vwv+3, 1, CreatFlags);
814 SIVAL(vwv+5, 1, 0x0); /* RootDirectoryFid */
815 SIVAL(vwv+7, 1, DesiredAccess);
816 SIVAL(vwv+9, 1, 0x0); /* AllocationSize */
817 SIVAL(vwv+11, 1, 0x0); /* AllocationSize */
818 SIVAL(vwv+13, 1, FileAttributes);
819 SIVAL(vwv+15, 1, ShareAccess);
820 SIVAL(vwv+17, 1, CreateDisposition);
821 SIVAL(vwv+19, 1, CreateOptions);
822 SIVAL(vwv+21, 1, 0x02); /* ImpersonationLevel */
823 SCVAL(vwv+23, 1, SecurityFlags);
824
825 bytes = talloc_array(state, uint8_t, 0);
826 bytes = smb_bytes_push_str(bytes, cli_ucs2(cli),
827 fname, strlen(fname)+1,
828 &converted_len);
829
830 /* sigh. this copes with broken netapp filer behaviour */
831 bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), "", 1, NULL);
832
833 if (tevent_req_nomem(bytes, req)) {
834 return tevent_req_post(req, ev);
835 }
836
837 SIVAL(vwv+2, 1, converted_len);
838
839 subreq = cli_smb_send(state, ev, cli, SMBntcreateX, 0, 24, vwv,
840 talloc_get_size(bytes), bytes);
841 if (tevent_req_nomem(subreq, req)) {
842 return tevent_req_post(req, ev);
843 }
844 tevent_req_set_callback(subreq, cli_ntcreate_done, req);
845 return req;
846 }
847
848 static void cli_ntcreate_done(struct tevent_req *subreq)
849 {
850 struct tevent_req *req = tevent_req_callback_data(
851 subreq, struct tevent_req);
852 struct cli_ntcreate_state *state = tevent_req_data(
853 req, struct cli_ntcreate_state);
854 uint8_t wct;
855 uint16_t *vwv;
856 uint32_t num_bytes;
857 uint8_t *bytes;
858 NTSTATUS status;
859
860 status = cli_smb_recv(subreq, 3, &wct, &vwv, &num_bytes, &bytes);
861 if (!NT_STATUS_IS_OK(status)) {
862 TALLOC_FREE(subreq);
863 tevent_req_nterror(req, status);
864 return;
865 }
866 state->fnum = SVAL(vwv+2, 1);
867 tevent_req_done(req);
868 }
869
870 NTSTATUS cli_ntcreate_recv(struct tevent_req *req, uint16_t *pfnum)
871 {
872 struct cli_ntcreate_state *state = tevent_req_data(
873 req, struct cli_ntcreate_state);
874 NTSTATUS status;
875
876 if (tevent_req_is_nterror(req, &status)) {
877 return status;
878 }
879 *pfnum = state->fnum;
880 return NT_STATUS_OK;
881 }
882
883 NTSTATUS cli_ntcreate(struct cli_state *cli,
884 const char *fname,
885 uint32_t CreatFlags,
886 uint32_t DesiredAccess,
887 uint32_t FileAttributes,
888 uint32_t ShareAccess,
889 uint32_t CreateDisposition,
890 uint32_t CreateOptions,
891 uint8_t SecurityFlags,
892 uint16_t *pfid)
893 {
894 TALLOC_CTX *frame = talloc_stackframe();
895 struct event_context *ev;
896 struct tevent_req *req;
897 NTSTATUS status = NT_STATUS_OK;
898
899 if (cli_has_async_calls(cli)) {
900 /*
901 * Can't use sync call while an async call is in flight
902 */
903 status = NT_STATUS_INVALID_PARAMETER;
904 goto fail;
905 }
906
907 ev = event_context_init(frame);
908 if (ev == NULL) {
909 status = NT_STATUS_NO_MEMORY;
910 goto fail;
911 }
912
913 req = cli_ntcreate_send(frame, ev, cli, fname, CreatFlags,
914 DesiredAccess, FileAttributes, ShareAccess,
915 CreateDisposition, CreateOptions,
916 SecurityFlags);
917 if (req == NULL) {
918 status = NT_STATUS_NO_MEMORY;
919 goto fail;
920 }
921
922 if (!tevent_req_poll(req, ev)) {
923 status = map_nt_error_from_unix(errno);
924 goto fail;
925 }
926
927 status = cli_ntcreate_recv(req, pfid);
928 fail:
929 TALLOC_FREE(frame);
930 if (!NT_STATUS_IS_OK(status)) {
931 cli_set_error(cli, status);
932 }
933 return status;
934 }
935
936 /****************************************************************************
937 Open a file.
938 ****************************************************************************/
939
940 int cli_nt_create(struct cli_state *cli, const char *fname, uint32 DesiredAccess)
941 {
942 return cli_nt_create_full(cli, fname, 0, DesiredAccess, 0,
943 FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN, 0x0, 0x0);
944 }
945
946 uint8_t *smb_bytes_push_str(uint8_t *buf, bool ucs2,
947 const char *str, size_t str_len,
948 size_t *pconverted_size)
949 {
950 size_t buflen;
951 char *converted;
952 size_t converted_size;
953
954 if (buf == NULL) {
955 return NULL;
956 }
957
958 buflen = talloc_get_size(buf);
959 /*
960 * We're pushing into an SMB buffer, align odd
961 */
962 if (ucs2 && (buflen % 2 == 0)) {
963 buf = TALLOC_REALLOC_ARRAY(NULL, buf, uint8_t, buflen + 1);
964 if (buf == NULL) {
965 return NULL;
966 }
967 buf[buflen] = '\0';
968 buflen += 1;
969 }
970
971 if (!convert_string_allocate(talloc_tos(), CH_UNIX,
972 ucs2 ? CH_UTF16LE : CH_DOS,
973 str, str_len, &converted,
974 &converted_size, true)) {
975 return NULL;
976 }
977
978 buf = TALLOC_REALLOC_ARRAY(NULL, buf, uint8_t,
979 buflen + converted_size);
980 if (buf == NULL) {
981 TALLOC_FREE(converted);
982 return NULL;
983 }
984
985 memcpy(buf + buflen, converted, converted_size);
986
987 TALLOC_FREE(converted);
988
989 if (pconverted_size) {
990 *pconverted_size = converted_size;
991 }
992
993 return buf;
994 }
995
996 /****************************************************************************
997 Open a file
998 WARNING: if you open with O_WRONLY then getattrE won't work!
999 ****************************************************************************/
1000
1001 struct async_req *cli_open_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
1002 struct cli_state *cli,
1003 const char *fname, int flags, int share_mode)
1004 {
1005 unsigned openfn = 0;
1006 unsigned accessmode = 0;
1007 uint8_t additional_flags = 0;
1008 uint8_t *bytes;
1009 uint16_t vwv[15];
1010 struct async_req *result;
1011
1012 if (flags & O_CREAT) {
1013 openfn |= (1<<4);
1014 }
1015 if (!(flags & O_EXCL)) {
1016 if (flags & O_TRUNC)
1017 openfn |= (1<<1);
1018 else
1019 openfn |= (1<<0);
1020 }
1021
1022 accessmode = (share_mode<<4);
1023
1024 if ((flags & O_ACCMODE) == O_RDWR) {
1025 accessmode |= 2;
1026 } else if ((flags & O_ACCMODE) == O_WRONLY) {
1027 accessmode |= 1;
1028 }
1029
1030 #if defined(O_SYNC)
1031 if ((flags & O_SYNC) == O_SYNC) {
1032 accessmode |= (1<<14);
1033 }
1034 #endif /* O_SYNC */
1035
1036 if (share_mode == DENY_FCB) {
1037 accessmode = 0xFF;
1038 }
1039
1040 SCVAL(vwv + 0, 0, 0xFF);
1041 SCVAL(vwv + 0, 1, 0);
1042 SSVAL(vwv + 1, 0, 0);
1043 SSVAL(vwv + 2, 0, 0); /* no additional info */
1044 SSVAL(vwv + 3, 0, accessmode);
1045 SSVAL(vwv + 4, 0, aSYSTEM | aHIDDEN);
1046 SSVAL(vwv + 5, 0, 0);
1047 SIVAL(vwv + 6, 0, 0);
1048 SSVAL(vwv + 8, 0, openfn);
1049 SIVAL(vwv + 9, 0, 0);
1050 SIVAL(vwv + 11, 0, 0);
1051 SIVAL(vwv + 13, 0, 0);
1052
1053 if (cli->use_oplocks) {
1054 /* if using oplocks then ask for a batch oplock via
1055 core and extended methods */
1056 additional_flags =
1057 FLAG_REQUEST_OPLOCK|FLAG_REQUEST_BATCH_OPLOCK;
1058 SSVAL(vwv+2, 0, SVAL(vwv+2, 0) | 6);
1059 }
1060
1061 bytes = talloc_array(talloc_tos(), uint8_t, 0);
1062 bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), fname,
1063 strlen(fname)+1, NULL);
1064 if (bytes == NULL) {
1065 return NULL;
1066 }
1067
1068 result = cli_request_send(mem_ctx, ev, cli, SMBopenX, additional_flags,
1069 15, vwv, 0, talloc_get_size(bytes), bytes);
1070 TALLOC_FREE(bytes);
1071 return result;
1072 }
1073
1074 NTSTATUS cli_open_recv(struct async_req *req, int *fnum)
1075 {
1076 uint8_t wct;
1077 uint16_t *vwv;
1078 uint16_t num_bytes;
1079 uint8_t *bytes;
1080 NTSTATUS status;
1081
1082 if (async_req_is_nterror(req, &status)) {
1083 return status;
1084 }
1085
1086 status = cli_pull_reply(req, &wct, &vwv, &num_bytes, &bytes);
1087 if (!NT_STATUS_IS_OK(status)) {
1088 return status;
1089 }
1090
1091 if (wct < 3) {
1092 return NT_STATUS_INVALID_NETWORK_RESPONSE;
1093 }
1094
1095 *fnum = SVAL(vwv+2, 0);
1096
1097 return NT_STATUS_OK;
1098 }
1099
1100 int cli_open(struct cli_state *cli, const char *fname, int flags,
1101 int share_mode)
1102 {
1103 TALLOC_CTX *frame = talloc_stackframe();
1104 struct event_context *ev;
1105 struct async_req *req;
1106 int result = -1;
1107
1108 if (cli->fd_event != NULL) {
1109 /*
1110 * Can't use sync call while an async call is in flight
1111 */
1112 cli_set_error(cli, NT_STATUS_INVALID_PARAMETER);
1113 goto fail;
1114 }
1115
1116 ev = event_context_init(frame);
1117 if (ev == NULL) {
1118 goto fail;
1119 }
1120
1121 req = cli_open_send(frame, ev, cli, fname, flags, share_mode);
1122 if (req == NULL) {
1123 goto fail;
1124 }
1125
1126 while (req->state < ASYNC_REQ_DONE) {
1127 event_loop_once(ev);
1128 }
1129
1130 cli_open_recv(req, &result);
1131 fail:
1132 TALLOC_FREE(frame);
1133 return result;
1134 }
1135
1136 /****************************************************************************
1137 Close a file.
1138 ****************************************************************************/
1139
1140 struct cli_close_state {
1141 uint16_t vwv[3];
1142 };
1143
1144 static void cli_close_done(struct tevent_req *subreq);
1145
1146 struct tevent_req *cli_close_create(TALLOC_CTX *mem_ctx,
1147 struct event_context *ev,
1148 struct cli_state *cli, int fnum,
1149 struct tevent_req **psubreq)
1150 {
1151 struct tevent_req *req, *subreq;
1152 struct cli_close_state *state;
1153
1154 req = tevent_req_create(mem_ctx, &state, struct cli_close_state);
1155 if (req == NULL) {
1156 return NULL;
1157 }
1158 SSVAL(state->vwv+0, 0, fnum);
1159 SIVALS(state->vwv+1, 0, -1);
1160
1161 subreq = cli_smb_req_create(state, ev, cli, SMBclose, 0, 3, state->vwv,
1162 0, NULL);
1163 if (subreq == NULL) {
1164 TALLOC_FREE(req);
1165 return NULL;
1166 }
1167 tevent_req_set_callback(subreq, cli_close_done, req);
1168 *psubreq = subreq;
1169 return req;
1170 }
1171
1172 struct tevent_req *cli_close_send(TALLOC_CTX *mem_ctx,
1173 struct event_context *ev,
1174 struct cli_state *cli, int fnum)
1175 {
1176 struct tevent_req *req, *subreq;
1177
1178 req = cli_close_create(mem_ctx, ev, cli, fnum, &subreq);
1179 if ((req == NULL) || !cli_smb_req_send(subreq)) {
1180 TALLOC_FREE(req);
1181 return NULL;
1182 }
1183 return req;
1184 }
1185
1186 static void cli_close_done(struct tevent_req *subreq)
1187 {
1188 struct tevent_req *req = tevent_req_callback_data(
1189 subreq, struct tevent_req);
1190 NTSTATUS status;
1191
1192 status = cli_smb_recv(subreq, 0, NULL, NULL, NULL, NULL);
1193 TALLOC_FREE(subreq);
1194 if (!NT_STATUS_IS_OK(status)) {
1195 tevent_req_nterror(req, status);
1196 return;
1197 }
1198 tevent_req_done(req);
1199 }
1200
1201 NTSTATUS cli_close_recv(struct tevent_req *req)
1202 {
1203 return tevent_req_simple_recv_ntstatus(req);
1204 }
1205
1206 bool cli_close(struct cli_state *cli, int fnum)
1207 {
1208 TALLOC_CTX *frame = talloc_stackframe();
1209 struct event_context *ev;
1210 struct tevent_req *req;
1211 NTSTATUS status = NT_STATUS_OK;
1212 bool result = false;
1213
1214 if (cli_has_async_calls(cli)) {
1215 /*
1216 * Can't use sync call while an async call is in flight
1217 */
1218 status = NT_STATUS_INVALID_PARAMETER;
1219 goto fail;
1220 }
1221
1222 ev = event_context_init(frame);
1223 if (ev == NULL) {
1224 status = NT_STATUS_NO_MEMORY;
1225 goto fail;
1226 }
1227
1228 req = cli_close_send(frame, ev, cli, fnum);
1229 if (req == NULL) {
1230 status = NT_STATUS_NO_MEMORY;
1231 goto fail;
1232 }
1233
1234 if (!tevent_req_poll(req, ev)) {
1235 status = map_nt_error_from_unix(errno);
1236 goto fail;
1237 }
1238
1239 result = NT_STATUS_IS_OK(cli_close_recv(req));
1240 fail:
1241 TALLOC_FREE(frame);
1242 if (!NT_STATUS_IS_OK(status)) {
1243 cli_set_error(cli, status);
1244 }
1245 return result;
1246 }
1247
1248 /****************************************************************************
1249 Truncate a file to a specified size
1250 ****************************************************************************/
1251
1252 bool cli_ftruncate(struct cli_state *cli, int fnum, uint64_t size)
1253 {
1254 unsigned int param_len = 6;
1255 unsigned int data_len = 8;
1256 uint16 setup = TRANSACT2_SETFILEINFO;
1257 char param[6];
1258 unsigned char data[8];
1259 char *rparam=NULL, *rdata=NULL;
1260 int saved_timeout = cli->timeout;
1261
1262 SSVAL(param,0,fnum);
1263 SSVAL(param,2,SMB_SET_FILE_END_OF_FILE_INFO);
1264 SSVAL(param,4,0);
1265
1266 SBVAL(data, 0, size);
1267
1268 if (!cli_send_trans(cli, SMBtrans2,
1269 NULL, /* name */
1270 -1, 0, /* fid, flags */
1271 &setup, 1, 0, /* setup, length, max */
1272 param, param_len, 2, /* param, length, max */
1273 (char *)&data, data_len,/* data, length, ... */
1274 cli->max_xmit)) { /* ... max */
1275 cli->timeout = saved_timeout;
1276 return False;
1277 }
1278
1279 if (!cli_receive_trans(cli, SMBtrans2,
1280 &rparam, &param_len,
1281 &rdata, &data_len)) {
1282 cli->timeout = saved_timeout;
1283 SAFE_FREE(rdata);
1284 SAFE_FREE(rparam);
1285 return False;
1286 }
1287
1288 cli->timeout = saved_timeout;
1289
1290 SAFE_FREE(rdata);
1291 SAFE_FREE(rparam);
1292
1293 return True;
1294 }
1295
1296
1297 /****************************************************************************
1298 send a lock with a specified locktype
1299 this is used for testing LOCKING_ANDX_CANCEL_LOCK
1300 ****************************************************************************/
1301
1302 NTSTATUS cli_locktype(struct cli_state *cli, int fnum,
1303 uint32 offset, uint32 len,
1304 int timeout, unsigned char locktype)
1305 {
1306 char *p;
1307 int saved_timeout = cli->timeout;
1308
1309 memset(cli->outbuf,'\0',smb_size);
1310 memset(cli->inbuf,'\0', smb_size);
1311
1312 cli_set_message(cli->outbuf,8,0,True);
1313
1314 SCVAL(cli->outbuf,smb_com,SMBlockingX);
1315 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1316 cli_setup_packet(cli);
1317
1318 SCVAL(cli->outbuf,smb_vwv0,0xFF);
1319 SSVAL(cli->outbuf,smb_vwv2,fnum);
1320 SCVAL(cli->outbuf,smb_vwv3,locktype);
1321 SIVALS(cli->outbuf, smb_vwv4, timeout);
1322 SSVAL(cli->outbuf,smb_vwv6,0);
1323 SSVAL(cli->outbuf,smb_vwv7,1);
1324
1325 p = smb_buf(cli->outbuf);
1326 SSVAL(p, 0, cli->pid);
1327 SIVAL(p, 2, offset);
1328 SIVAL(p, 6, len);
1329
1330 p += 10;
1331
1332 cli_setup_bcc(cli, p);
1333
1334 cli_send_smb(cli);
1335
1336 if (timeout != 0) {
1337 cli->timeout = (timeout == -1) ? 0x7FFFFFFF : (timeout + 2*1000);
1338 }
1339
1340 if (!cli_receive_smb(cli)) {
1341 cli->timeout = saved_timeout;
1342 return NT_STATUS_UNSUCCESSFUL;
1343 }
1344
1345 cli->timeout = saved_timeout;
1346
1347 return cli_nt_error(cli);
1348 }
1349
1350 /****************************************************************************
1351 Lock a file.
1352 note that timeout is in units of 2 milliseconds
1353 ****************************************************************************/
1354
1355 bool cli_lock(struct cli_state *cli, int fnum,
1356 uint32 offset, uint32 len, int timeout, enum brl_type lock_type)
1357 {
1358 char *p;
1359 int saved_timeout = cli->timeout;
1360
1361 memset(cli->outbuf,'\0',smb_size);
1362 memset(cli->inbuf,'\0', smb_size);
1363
1364 cli_set_message(cli->outbuf,8,0,True);
1365
1366 SCVAL(cli->outbuf,smb_com,SMBlockingX);
1367 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1368 cli_setup_packet(cli);
1369
1370 SCVAL(cli->outbuf,smb_vwv0,0xFF);
1371 SSVAL(cli->outbuf,smb_vwv2,fnum);
1372 SCVAL(cli->outbuf,smb_vwv3,(lock_type == READ_LOCK? 1 : 0));
1373 SIVALS(cli->outbuf, smb_vwv4, timeout);
1374 SSVAL(cli->outbuf,smb_vwv6,0);
1375 SSVAL(cli->outbuf,smb_vwv7,1);
1376
1377 p = smb_buf(cli->outbuf);
1378 SSVAL(p, 0, cli->pid);
1379 SIVAL(p, 2, offset);
1380 SIVAL(p, 6, len);
1381
1382 p += 10;
1383
1384 cli_setup_bcc(cli, p);
1385
1386 cli_send_smb(cli);
1387
1388 if (timeout != 0) {
1389 cli->timeout = (timeout == -1) ? 0x7FFFFFFF : (timeout*2 + 5*1000);
1390 }
1391
1392 if (!cli_receive_smb(cli)) {
1393 cli->timeout = saved_timeout;
1394 return False;
1395 }
1396
1397 cli->timeout = saved_timeout;
1398
1399 if (cli_is_error(cli)) {
1400 return False;
1401 }
1402
1403 return True;
1404 }
1405
1406 /****************************************************************************
1407 Unlock a file.
1408 ****************************************************************************/
1409
1410 bool cli_unlock(struct cli_state *cli, int fnum, uint32 offset, uint32 len)
1411 {
1412 char *p;
1413
1414 memset(cli->outbuf,'\0',smb_size);
1415 memset(cli->inbuf,'\0',smb_size);
1416
1417 cli_set_message(cli->outbuf,8,0,True);
1418
1419 SCVAL(cli->outbuf,smb_com,SMBlockingX);
1420 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1421 cli_setup_packet(cli);
1422
1423 SCVAL(cli->outbuf,smb_vwv0,0xFF);
1424 SSVAL(cli->outbuf,smb_vwv2,fnum);
1425 SCVAL(cli->outbuf,smb_vwv3,0);
1426 SIVALS(cli->outbuf, smb_vwv4, 0);
1427 SSVAL(cli->outbuf,smb_vwv6,1);
1428 SSVAL(cli->outbuf,smb_vwv7,0);
1429
1430 p = smb_buf(cli->outbuf);
1431 SSVAL(p, 0, cli->pid);
1432 SIVAL(p, 2, offset);
1433 SIVAL(p, 6, len);
1434 p += 10;
1435 cli_setup_bcc(cli, p);
1436 cli_send_smb(cli);
1437 if (!cli_receive_smb(cli)) {
1438 return False;
1439 }
1440
1441 if (cli_is_error(cli)) {
1442 return False;
1443 }
1444
1445 return True;
1446 }
1447
1448 /****************************************************************************
1449 Lock a file with 64 bit offsets.
1450 ****************************************************************************/
1451
1452 bool cli_lock64(struct cli_state *cli, int fnum,
1453 uint64_t offset, uint64_t len, int timeout, enum brl_type lock_type)
1454 {
1455 char *p;
1456 int saved_timeout = cli->timeout;
1457 int ltype;
1458
1459 if (! (cli->capabilities & CAP_LARGE_FILES)) {
1460 return cli_lock(cli, fnum, offset, len, timeout, lock_type);
1461 }
1462
1463 ltype = (lock_type == READ_LOCK? 1 : 0);
1464 ltype |= LOCKING_ANDX_LARGE_FILES;
1465
1466 memset(cli->outbuf,'\0',smb_size);
1467 memset(cli->inbuf,'\0', smb_size);
1468
1469 cli_set_message(cli->outbuf,8,0,True);
1470
1471 SCVAL(cli->outbuf,smb_com,SMBlockingX);
1472 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1473 cli_setup_packet(cli);
1474
1475 SCVAL(cli->outbuf,smb_vwv0,0xFF);
1476 SSVAL(cli->outbuf,smb_vwv2,fnum);
1477 SCVAL(cli->outbuf,smb_vwv3,ltype);
1478 SIVALS(cli->outbuf, smb_vwv4, timeout);
1479 SSVAL(cli->outbuf,smb_vwv6,0);
1480 SSVAL(cli->outbuf,smb_vwv7,1);
1481
1482 p = smb_buf(cli->outbuf);
1483 SIVAL(p, 0, cli->pid);
1484 SOFF_T_R(p, 4, offset);
1485 SOFF_T_R(p, 12, len);
1486 p += 20;
1487
1488 cli_setup_bcc(cli, p);
1489 cli_send_smb(cli);
1490
1491 if (timeout != 0) {
1492 cli->timeout = (timeout == -1) ? 0x7FFFFFFF : (timeout + 5*1000);
1493 }
1494
1495 if (!cli_receive_smb(cli)) {
1496 cli->timeout = saved_timeout;
1497 return False;
1498 }
1499
1500 cli->timeout = saved_timeout;
1501
1502 if (cli_is_error(cli)) {
1503 return False;
1504 }
1505
1506 return True;
1507 }
1508
1509 /****************************************************************************
1510 Unlock a file with 64 bit offsets.
1511 ****************************************************************************/
1512
1513 bool cli_unlock64(struct cli_state *cli, int fnum, uint64_t offset, uint64_t len)
1514 {
1515 char *p;
1516
1517 if (! (cli->capabilities & CAP_LARGE_FILES)) {
1518 return cli_unlock(cli, fnum, offset, len);
1519 }
1520
1521 memset(cli->outbuf,'\0',smb_size);
1522 memset(cli->inbuf,'\0',smb_size);
1523
1524 cli_set_message(cli->outbuf,8,0,True);
1525
1526 SCVAL(cli->outbuf,smb_com,SMBlockingX);
1527 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1528 cli_setup_packet(cli);
1529
1530 SCVAL(cli->outbuf,smb_vwv0,0xFF);
1531 SSVAL(cli->outbuf,smb_vwv2,fnum);
1532 SCVAL(cli->outbuf,smb_vwv3,LOCKING_ANDX_LARGE_FILES);
1533 SIVALS(cli->outbuf, smb_vwv4, 0);
1534 SSVAL(cli->outbuf,smb_vwv6,1);
1535 SSVAL(cli->outbuf,smb_vwv7,0);
1536
1537 p = smb_buf(cli->outbuf);
1538 SIVAL(p, 0, cli->pid);
1539 SOFF_T_R(p, 4, offset);
1540 SOFF_T_R(p, 12, len);
1541 p += 20;
1542 cli_setup_bcc(cli, p);
1543 cli_send_smb(cli);
1544 if (!cli_receive_smb(cli)) {
1545 return False;
1546 }
1547
1548 if (cli_is_error(cli)) {
1549 return False;
1550 }
1551
1552 return True;
1553 }
1554
1555 /****************************************************************************
1556 Get/unlock a POSIX lock on a file - internal function.
1557 ****************************************************************************/
1558
1559 static bool cli_posix_lock_internal(struct cli_state *cli, int fnum,
1560 uint64_t offset, uint64_t len, bool wait_lock, enum brl_type lock_type)
1561 {
1562 unsigned int param_len = 4;
1563 unsigned int data_len = POSIX_LOCK_DATA_SIZE;
1564 uint16 setup = TRANSACT2_SETFILEINFO;
1565 char param[4];
1566 unsigned char data[POSIX_LOCK_DATA_SIZE];
1567 char *rparam=NULL, *rdata=NULL;
1568 int saved_timeout = cli->timeout;
1569
1570 SSVAL(param,0,fnum);
1571 SSVAL(param,2,SMB_SET_POSIX_LOCK);
1572
1573 switch (lock_type) {
1574 case READ_LOCK:
1575 SSVAL(data, POSIX_LOCK_TYPE_OFFSET, POSIX_LOCK_TYPE_READ);
1576 break;
1577 case WRITE_LOCK:
1578 SSVAL(data, POSIX_LOCK_TYPE_OFFSET, POSIX_LOCK_TYPE_WRITE);
1579 break;
1580 case UNLOCK_LOCK:
1581 SSVAL(data, POSIX_LOCK_TYPE_OFFSET, POSIX_LOCK_TYPE_UNLOCK);
1582 break;
1583 default:
1584 return False;
1585 }
1586
1587 if (wait_lock) {
1588 SSVAL(data, POSIX_LOCK_FLAGS_OFFSET, POSIX_LOCK_FLAG_WAIT);
1589 cli->timeout = 0x7FFFFFFF;
1590 } else {
1591 SSVAL(data, POSIX_LOCK_FLAGS_OFFSET, POSIX_LOCK_FLAG_NOWAIT);
1592 }
1593
1594 SIVAL(data, POSIX_LOCK_PID_OFFSET, cli->pid);
1595 SOFF_T(data, POSIX_LOCK_START_OFFSET, offset);
1596 SOFF_T(data, POSIX_LOCK_LEN_OFFSET, len);
1597
1598 if (!cli_send_trans(cli, SMBtrans2,
1599 NULL, /* name */
1600 -1, 0, /* fid, flags */
1601 &setup, 1, 0, /* setup, length, max */
1602 param, param_len, 2, /* param, length, max */
1603 (char *)&data, data_len, cli->max_xmit /* data, length, max */
1604 )) {
1605 cli->timeout = saved_timeout;
1606 return False;
1607 }
1608
1609 if (!cli_receive_trans(cli, SMBtrans2,
1610 &rparam, &param_len,
1611 &rdata, &data_len)) {
1612 cli->timeout = saved_timeout;
1613 SAFE_FREE(rdata);
1614 SAFE_FREE(rparam);
1615 return False;
1616 }
1617
1618 cli->timeout = saved_timeout;
1619
1620 SAFE_FREE(rdata);
1621 SAFE_FREE(rparam);
1622
1623 return True;
1624 }
1625
1626 /****************************************************************************
1627 POSIX Lock a file.
1628 ****************************************************************************/
1629
1630 bool cli_posix_lock(struct cli_state *cli, int fnum,
1631 uint64_t offset, uint64_t len,
1632 bool wait_lock, enum brl_type lock_type)
1633 {
1634 if (lock_type != READ_LOCK && lock_type != WRITE_LOCK) {
1635 return False;
1636 }
1637 return cli_posix_lock_internal(cli, fnum, offset, len, wait_lock, lock_type);
1638 }
1639
1640 /****************************************************************************
1641 POSIX Unlock a file.
1642 ****************************************************************************/
1643
1644 bool cli_posix_unlock(struct cli_state *cli, int fnum, uint64_t offset, uint64_t len)
1645 {
1646 return cli_posix_lock_internal(cli, fnum, offset, len, False, UNLOCK_LOCK);
1647 }
1648
1649 /****************************************************************************
1650 POSIX Get any lock covering a file.
1651 ****************************************************************************/
1652
1653 bool cli_posix_getlock(struct cli_state *cli, int fnum, uint64_t *poffset, uint64_t *plen)
1654 {
1655 return True;
1656 }
1657
1658 /****************************************************************************
1659 Do a SMBgetattrE call.
1660 ****************************************************************************/
1661
1662 bool cli_getattrE(struct cli_state *cli, int fd,
1663 uint16 *attr, SMB_OFF_T *size,
1664 time_t *change_time,
1665 time_t *access_time,
1666 time_t *write_time)
1667 {
1668 memset(cli->outbuf,'\0',smb_size);
1669 memset(cli->inbuf,'\0',smb_size);
1670
1671 cli_set_message(cli->outbuf,1,0,True);
1672
1673 SCVAL(cli->outbuf,smb_com,SMBgetattrE);
1674 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1675 cli_setup_packet(cli);
1676
1677 SSVAL(cli->outbuf,smb_vwv0,fd);
1678
1679 cli_send_smb(cli);
1680 if (!cli_receive_smb(cli)) {
1681 return False;
1682 }
1683
1684 if (cli_is_error(cli)) {
1685 return False;
1686 }
1687
1688 if (size) {
1689 *size = IVAL(cli->inbuf, smb_vwv6);
1690 }
1691
1692 if (attr) {
1693 *attr = SVAL(cli->inbuf,smb_vwv10);
1694 }
1695
1696 if (change_time) {
1697 *change_time = cli_make_unix_date2(cli, cli->inbuf+smb_vwv0);
1698 }
1699
1700 if (access_time) {
1701 *access_time = cli_make_unix_date2(cli, cli->inbuf+smb_vwv2);
1702 }
1703
1704 if (write_time) {
1705 *write_time = cli_make_unix_date2(cli, cli->inbuf+smb_vwv4);
1706 }
1707
1708 return True;
1709 }
1710
1711 /****************************************************************************
1712 Do a SMBgetatr call
1713 ****************************************************************************/
1714
1715 bool cli_getatr(struct cli_state *cli, const char *fname,
1716 uint16 *attr, SMB_OFF_T *size, time_t *write_time)
1717 {
1718 char *p;
1719
1720 memset(cli->outbuf,'\0',smb_size);
1721 memset(cli->inbuf,'\0',smb_size);
1722
1723 cli_set_message(cli->outbuf,0,0,True);
1724
1725 SCVAL(cli->outbuf,smb_com,SMBgetatr);
1726 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1727 cli_setup_packet(cli);
1728
1729 p = smb_buf(cli->outbuf);
1730 *p++ = 4;
1731 p += clistr_push(cli, p, fname,
1732 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
1733
1734 cli_setup_bcc(cli, p);
1735
1736 cli_send_smb(cli);
1737 if (!cli_receive_smb(cli)) {
1738 return False;
1739 }
1740
1741 if (cli_is_error(cli)) {
1742 return False;
1743 }
1744
1745 if (size) {
1746 *size = IVAL(cli->inbuf, smb_vwv3);
1747 }
1748
1749 if (write_time) {
1750 *write_time = cli_make_unix_date3(cli, cli->inbuf+smb_vwv1);
1751 }
1752
1753 if (attr) {
1754 *attr = SVAL(cli->inbuf,smb_vwv0);
1755 }
1756
1757 return True;
1758 }
1759
1760 /****************************************************************************
1761 Do a SMBsetattrE call.
1762 ****************************************************************************/
1763
1764 bool cli_setattrE(struct cli_state *cli, int fd,
1765 time_t change_time,
1766 time_t access_time,
1767 time_t write_time)
1768
1769 {
1770 char *p;
1771
1772 memset(cli->outbuf,'\0',smb_size);
1773 memset(cli->inbuf,'\0',smb_size);
1774
1775 cli_set_message(cli->outbuf,7,0,True);
1776
1777 SCVAL(cli->outbuf,smb_com,SMBsetattrE);
1778 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1779 cli_setup_packet(cli);
1780
1781 SSVAL(cli->outbuf,smb_vwv0, fd);
1782 cli_put_dos_date2(cli, cli->outbuf,smb_vwv1, change_time);
1783 cli_put_dos_date2(cli, cli->outbuf,smb_vwv3, access_time);
1784 cli_put_dos_date2(cli, cli->outbuf,smb_vwv5, write_time);
1785
1786 p = smb_buf(cli->outbuf);
1787 *p++ = 4;
1788
1789 cli_setup_bcc(cli, p);
1790
1791 cli_send_smb(cli);
1792 if (!cli_receive_smb(cli)) {
1793 return False;
1794 }
1795
1796 if (cli_is_error(cli)) {
1797 return False;
1798 }
1799
1800 return True;
1801 }
1802
1803 /****************************************************************************
1804 Do a SMBsetatr call.
1805 ****************************************************************************/
1806
1807 bool cli_setatr(struct cli_state *cli, const char *fname, uint16 attr, time_t t)
1808 {
1809 char *p;
1810
1811 memset(cli->outbuf,'\0',smb_size);
1812 memset(cli->inbuf,'\0',smb_size);
1813
1814 cli_set_message(cli->outbuf,8,0,True);
1815
1816 SCVAL(cli->outbuf,smb_com,SMBsetatr);
1817 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1818 cli_setup_packet(cli);
1819
1820 SSVAL(cli->outbuf,smb_vwv0, attr);
1821 cli_put_dos_date3(cli, cli->outbuf,smb_vwv1, t);
1822
1823 p = smb_buf(cli->outbuf);
1824 *p++ = 4;
1825 p += clistr_push(cli, p, fname,
1826 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
1827 *p++ = 4;
1828
1829 cli_setup_bcc(cli, p);
1830
1831 cli_send_smb(cli);
1832 if (!cli_receive_smb(cli)) {
1833 return False;
1834 }
1835
1836 if (cli_is_error(cli)) {
1837 return False;
1838 }
1839
1840 return True;
1841 }
1842
1843 /****************************************************************************
1844 Check for existance of a dir.
1845 ****************************************************************************/
1846
1847 bool cli_chkpath(struct cli_state *cli, const char *path)
1848 {
1849 char *path2 = NULL;
1850 char *p;
1851 TALLOC_CTX *frame = talloc_stackframe();
1852
1853 path2 = talloc_strdup(frame, path);
1854 if (!path2) {
1855 TALLOC_FREE(frame);
1856 return false;
1857 }
1858 trim_char(path2,'\0','\\');
1859 if (!*path2) {
1860 path2 = talloc_strdup(frame, "\\");
1861 if (!path2) {
1862 TALLOC_FREE(frame);
1863 return false;
1864 }
1865 }
1866
1867 memset(cli->outbuf,'\0',smb_size);
1868 cli_set_message(cli->outbuf,0,0,True);
1869 SCVAL(cli->outbuf,smb_com,SMBcheckpath);
1870 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1871 cli_setup_packet(cli);
1872 p = smb_buf(cli->outbuf);
1873 *p++ = 4;
1874 p += clistr_push(cli, p, path2,
1875 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
1876
1877 cli_setup_bcc(cli, p);
1878
1879 cli_send_smb(cli);
1880 if (!cli_receive_smb(cli)) {
1881 TALLOC_FREE(frame);
1882 return False;
1883 }
1884
1885 TALLOC_FREE(frame);
1886
1887 if (cli_is_error(cli)) return False;
1888
1889 return True;
1890 }
1891
1892 /****************************************************************************
1893 Query disk space.
1894 ****************************************************************************/
1895
1896 bool cli_dskattr(struct cli_state *cli, int *bsize, int *total, int *avail)
1897 {
1898 memset(cli->outbuf,'\0',smb_size);
1899 cli_set_message(cli->outbuf,0,0,True);
1900 SCVAL(cli->outbuf,smb_com,SMBdskattr);
1901 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1902 cli_setup_packet(cli);
1903
1904 cli_send_smb(cli);
1905 if (!cli_receive_smb(cli)) {
1906 return False;
1907 }
1908
1909 *bsize = SVAL(cli->inbuf,smb_vwv1)*SVAL(cli->inbuf,smb_vwv2);
1910 *total = SVAL(cli->inbuf,smb_vwv0);
1911 *avail = SVAL(cli->inbuf,smb_vwv3);
1912
1913 return True;
1914 }
1915
1916 /****************************************************************************
1917 Create and open a temporary file.
1918 ****************************************************************************/
1919
1920 int cli_ctemp(struct cli_state *cli, const char *path, char **tmp_path)
1921 {
1922 int len;
1923 char *p;
1924
1925 memset(cli->outbuf,'\0',smb_size);
1926 memset(cli->inbuf,'\0',smb_size);
1927
1928 cli_set_message(cli->outbuf,3,0,True);
1929
1930 SCVAL(cli->outbuf,smb_com,SMBctemp);
1931 SSVAL(cli->outbuf,smb_tid,cli->cnum);
1932 cli_setup_packet(cli);
1933
1934 SSVAL(cli->outbuf,smb_vwv0,0);
1935 SIVALS(cli->outbuf,smb_vwv1,-1);
1936
1937 p = smb_buf(cli->outbuf);
1938 *p++ = 4;
1939 p += clistr_push(cli, p, path,
1940 cli->bufsize - PTR_DIFF(p,cli->outbuf), STR_TERMINATE);
1941
1942 cli_setup_bcc(cli, p);
1943
1944 cli_send_smb(cli);
1945 if (!cli_receive_smb(cli)) {
1946 return -1;
1947 }
1948
1949 if (cli_is_error(cli)) {
1950 return -1;
1951 }
1952
1953 /* despite the spec, the result has a -1, followed by
1954 length, followed by name */
1955 p = smb_buf(cli->inbuf);
1956 p += 4;
1957 len = smb_buflen(cli->inbuf) - 4;
1958 if (len <= 0 || len > PATH_MAX) return -1;
1959
1960 if (tmp_path) {
1961 char *path2 = SMB_MALLOC_ARRAY(char, len+1);
1962 if (!path2) {
1963 return -1;
1964 }
1965 clistr_pull(cli->inbuf, path2, p,
1966 len+1, len, STR_ASCII);
1967 *tmp_path = path2;
1968 }
1969
1970 return SVAL(cli->inbuf,smb_vwv0);
1971 }
1972
1973 /*
1974 send a raw ioctl - used by the torture code
1975 */
1976 NTSTATUS cli_raw_ioctl(struct cli_state *cli, int fnum, uint32 code, DATA_BLOB *blob)
1977 {
1978 memset(cli->outbuf,'\0',smb_size);
1979 memset(cli->inbuf,'\0',smb_size);
1980
1981 cli_set_message(cli->outbuf, 3, 0, True);
1982 SCVAL(cli->outbuf,smb_com,SMBioctl);
1983 cli_setup_packet(cli);
1984
1985 SSVAL(cli->outbuf, smb_vwv0, fnum);
1986 SSVAL(cli->outbuf, smb_vwv1, code>>16);
1987 SSVAL(cli->outbuf, smb_vwv2, (code&0xFFFF));
1988
1989 cli_send_smb(cli);
1990 if (!cli_receive_smb(cli)) {
1991 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
1992 }
1993
1994 if (cli_is_error(cli)) {
1995 return cli_nt_error(cli);
1996 }
1997
1998 *blob = data_blob_null;
1999
2000 return NT_STATUS_OK;
2001 }
2002
2003 /*********************************************************
2004 Set an extended attribute utility fn.
2005 *********************************************************/
2006
2007 static bool cli_set_ea(struct cli_state *cli, uint16 setup, char *param, unsigned int param_len,
2008 const char *ea_name, const char *ea_val, size_t ea_len)
2009 {
2010 unsigned int data_len = 0;
2011 char *data = NULL;
2012 char *rparam=NULL, *rdata=NULL;
2013 char *p;
2014 size_t ea_namelen = strlen(ea_name);
2015
2016 if (ea_namelen == 0 && ea_len == 0) {
2017 data_len = 4;
2018 data = (char *)SMB_MALLOC(data_len);
2019 if (!data) {
2020 return False;
2021 }
2022 p = data;
2023 SIVAL(p,0,data_len);
2024 } else {
2025 data_len = 4 + 4 + ea_namelen + 1 + ea_len;
2026 data = (char *)SMB_MALLOC(data_len);
2027 if (!data) {
2028 return False;
2029 }
2030 p = data;
2031 SIVAL(p,0,data_len);
2032 p += 4;
2033 SCVAL(p, 0, 0); /* EA flags. */
2034 SCVAL(p, 1, ea_namelen);
2035 SSVAL(p, 2, ea_len);
2036 memcpy(p+4, ea_name, ea_namelen+1); /* Copy in the name. */
2037 memcpy(p+4+ea_namelen+1, ea_val, ea_len);
2038 }
2039
2040 if (!cli_send_trans(cli, SMBtrans2,
2041 NULL, /* name */
2042 -1, 0, /* fid, flags */
2043 &setup, 1, 0, /* setup, length, max */
2044 param, param_len, 2, /* param, length, max */
2045 data, data_len, cli->max_xmit /* data, length, max */
2046 )) {
2047 SAFE_FREE(data);
2048 return False;
2049 }
2050
2051 if (!cli_receive_trans(cli, SMBtrans2,
2052 &rparam, &param_len,
2053 &rdata, &data_len)) {
2054 SAFE_FREE(data);
2055 return false;
2056 }
2057
2058 SAFE_FREE(data);
2059 SAFE_FREE(rdata);
2060 SAFE_FREE(rparam);
2061
2062 return True;
2063 }
2064
2065 /*********************************************************
2066 Set an extended attribute on a pathname.
2067 *********************************************************/
2068
2069 bool cli_set_ea_path(struct cli_state *cli, const char *path, const char *ea_name, const char *ea_val, size_t ea_len)
2070 {
2071 uint16 setup = TRANSACT2_SETPATHINFO;
2072 unsigned int param_len = 0;
2073 char *param;
2074 size_t srclen = 2*(strlen(path)+1);
2075 char *p;
2076 bool ret;
2077
2078 param = SMB_MALLOC_ARRAY(char, 6+srclen+2);
2079 if (!param) {
2080 return false;
2081 }
2082 memset(param, '\0', 6);
2083 SSVAL(param,0,SMB_INFO_SET_EA);
2084 p = &param[6];
2085
2086 p += clistr_push(cli, p, path, srclen, STR_TERMINATE);
2087 param_len = PTR_DIFF(p, param);
2088
2089 ret = cli_set_ea(cli, setup, param, param_len, ea_name, ea_val, ea_len);
2090 SAFE_FREE(param);
2091 return ret;
2092 }
2093
2094 /*********************************************************
2095 Set an extended attribute on an fnum.
2096 *********************************************************/
2097
2098 bool cli_set_ea_fnum(struct cli_state *cli, int fnum, const char *ea_name, const char *ea_val, size_t ea_len)
2099 {
2100 char param[6];
2101 uint16 setup = TRANSACT2_SETFILEINFO;
2102
2103 memset(param, 0, 6);
2104 SSVAL(param,0,fnum);
2105 SSVAL(param,2,SMB_INFO_SET_EA);
2106
2107 return cli_set_ea(cli, setup, param, 6, ea_name, ea_val, ea_len);
2108 }
2109
2110 /*********************************************************
2111 Get an extended attribute list utility fn.
2112 *********************************************************/
2113
2114 static bool cli_get_ea_list(struct cli_state *cli,
2115 uint16 setup, char *param, unsigned int param_len,
2116 TALLOC_CTX *ctx,
2117 size_t *pnum_eas,
2118 struct ea_struct **pea_list)
2119 {
2120 unsigned int data_len = 0;
2121 unsigned int rparam_len, rdata_len;
2122 char *rparam=NULL, *rdata=NULL;
2123 char *p;
2124 size_t ea_size;
2125 size_t num_eas;
2126 bool ret = False;
2127 struct ea_struct *ea_list;
2128
2129 *pnum_eas = 0;
2130 if (pea_list) {
2131 *pea_list = NULL;
2132 }
2133
2134 if (!cli_send_trans(cli, SMBtrans2,
2135 NULL, /* Name */
2136 -1, 0, /* fid, flags */
2137 &setup, 1, 0, /* setup, length, max */
2138 param, param_len, 10, /* param, length, max */
2139 NULL, data_len, cli->max_xmit /* data, length, max */
2140 )) {
2141 return False;
2142 }
2143
2144 if (!cli_receive_trans(cli, SMBtrans2,
2145 &rparam, &rparam_len,
2146 &rdata, &rdata_len)) {
2147 return False;
2148 }
2149
2150 if (!rdata || rdata_len < 4) {
2151 goto out;
2152 }
2153
2154 ea_size = (size_t)IVAL(rdata,0);
2155 if (ea_size > rdata_len) {
2156 goto out;
2157 }
2158
2159 if (ea_size == 0) {
2160 /* No EA's present. */
2161 ret = True;
2162 goto out;
2163 }
2164
2165 p = rdata + 4;
2166 ea_size -= 4;
2167
2168 /* Validate the EA list and count it. */
2169 for (num_eas = 0; ea_size >= 4; num_eas++) {
2170 unsigned int ea_namelen = CVAL(p,1);
2171 unsigned int ea_valuelen = SVAL(p,2);
2172 if (ea_namelen == 0) {
2173 goto out;
2174 }
2175 if (4 + ea_namelen + 1 + ea_valuelen > ea_size) {
2176 goto out;
2177 }
2178 ea_size -= 4 + ea_namelen + 1 + ea_valuelen;
2179 p += 4 + ea_namelen + 1 + ea_valuelen;
2180 }
2181
2182 if (num_eas == 0) {
2183 ret = True;
2184 goto out;
2185 }
2186
2187 *pnum_eas = num_eas;
2188 if (!pea_list) {
2189 /* Caller only wants number of EA's. */
2190 ret = True;
2191 goto out;
2192 }
2193
2194 ea_list = TALLOC_ARRAY(ctx, struct ea_struct, num_eas);
2195 if (!ea_list) {
2196 goto out;
2197 }
2198
2199 ea_size = (size_t)IVAL(rdata,0);
2200 p = rdata + 4;
2201
2202 for (num_eas = 0; num_eas < *pnum_eas; num_eas++ ) {
2203 struct ea_struct *ea = &ea_list[num_eas];
2204 fstring unix_ea_name;
2205 unsigned int ea_namelen = CVAL(p,1);
2206 unsigned int ea_valuelen = SVAL(p,2);
2207
2208 ea->flags = CVAL(p,0);
2209 unix_ea_name[0] = '\0';
2210 pull_ascii_fstring(unix_ea_name, p + 4);
2211 ea->name = talloc_strdup(ctx, unix_ea_name);
2212 /* Ensure the value is null terminated (in case it's a string). */
2213 ea->value = data_blob_talloc(ctx, NULL, ea_valuelen + 1);
2214 if (!ea->value.data) {
2215 goto out;
2216 }
2217 if (ea_valuelen) {
2218 memcpy(ea->value.data, p+4+ea_namelen+1, ea_valuelen);
2219 }
2220 ea->value.data[ea_valuelen] = 0;
2221 ea->value.length--;
2222 p += 4 + ea_namelen + 1 + ea_valuelen;
2223 }
2224
2225 *pea_list = ea_list;
2226 ret = True;
2227
2228 out :
2229
2230 SAFE_FREE(rdata);
2231 SAFE_FREE(rparam);
2232 return ret;
2233 }
2234
2235 /*********************************************************
2236 Get an extended attribute list from a pathname.
2237 *********************************************************/
2238
2239 bool cli_get_ea_list_path(struct cli_state *cli, const char *path,
2240 TALLOC_CTX *ctx,
2241 size_t *pnum_eas,
2242 struct ea_struct **pea_list)
2243 {
2244 uint16 setup = TRANSACT2_QPATHINFO;
2245 unsigned int param_len = 0;
2246 char *param;
2247 char *p;
2248 size_t srclen = 2*(strlen(path)+1);
2249 bool ret;
2250
2251 param = SMB_MALLOC_ARRAY(char, 6+srclen+2);
2252 if (!param) {
2253 return false;
2254 }
2255 p = param;
2256 memset(p, 0, 6);
2257 SSVAL(p, 0, SMB_INFO_QUERY_ALL_EAS);
2258 p += 6;
2259 p += clistr_push(cli, p, path, srclen, STR_TERMINATE);
2260 param_len = PTR_DIFF(p, param);
2261
2262 ret = cli_get_ea_list(cli, setup, param, param_len, ctx, pnum_eas, pea_list);
2263 SAFE_FREE(param);
2264 return ret;
2265 }
2266
2267 /*********************************************************
2268 Get an extended attribute list from an fnum.
2269 *********************************************************/
2270
2271 bool cli_get_ea_list_fnum(struct cli_state *cli, int fnum,
2272 TALLOC_CTX *ctx,
2273 size_t *pnum_eas,
2274 struct ea_struct **pea_list)
2275 {
2276 uint16 setup = TRANSACT2_QFILEINFO;
2277 char param[6];
2278
2279 memset(param, 0, 6);
2280 SSVAL(param,0,fnum);
2281 SSVAL(param,2,SMB_INFO_SET_EA);
2282
2283 return cli_get_ea_list(cli, setup, param, 6, ctx, pnum_eas, pea_list);
2284 }
2285
2286 /****************************************************************************
2287 Convert open "flags" arg to uint32 on wire.
2288 ****************************************************************************/
2289
2290 static uint32 open_flags_to_wire(int flags)
2291 {
2292 int open_mode = flags & O_ACCMODE;
2293 uint32 ret = 0;
2294
2295 switch (open_mode) {
2296 case O_WRONLY:
2297 ret |= SMB_O_WRONLY;
2298 break;
2299 case O_RDWR:
2300 ret |= SMB_O_RDWR;
2301 break;
2302 default:
2303 case O_RDONLY:
2304 ret |= SMB_O_RDONLY;
2305 break;
2306 }
2307
2308 if (flags & O_CREAT) {
2309 ret |= SMB_O_CREAT;
2310 }
2311 if (flags & O_EXCL) {
2312 ret |= SMB_O_EXCL;
2313 }
2314 if (flags & O_TRUNC) {
2315 ret |= SMB_O_TRUNC;
2316 }
2317 #if defined(O_SYNC)
2318 if (flags & O_SYNC) {
2319 ret |= SMB_O_SYNC;
2320 }
2321 #endif /* O_SYNC */
2322 if (flags & O_APPEND) {
2323 ret |= SMB_O_APPEND;
2324 }
2325 #if defined(O_DIRECT)
2326 if (flags & O_DIRECT) {
2327 ret |= SMB_O_DIRECT;
2328 }
2329 #endif
2330 #if defined(O_DIRECTORY)
2331 if (flags & O_DIRECTORY) {
2332 ret &= ~(SMB_O_RDONLY|SMB_O_RDWR|SMB_O_WRONLY);
2333 ret |= SMB_O_DIRECTORY;
2334 }
2335 #endif
2336 return ret;
2337 }
2338
2339 /****************************************************************************
2340 Open a file - POSIX semantics. Returns fnum. Doesn't request oplock.
2341 ****************************************************************************/
2342
2343 static int cli_posix_open_internal(struct cli_state *cli, const char *fname, int flags, mode_t mode, bool is_dir)
2344 {
2345 unsigned int data_len = 0;
2346 unsigned int param_len = 0;
2347 uint16 setup = TRANSACT2_SETPATHINFO;
2348 char *param;
2349 char data[18];
2350 char *rparam=NULL, *rdata=NULL;
2351 char *p;
2352 int fnum = -1;
2353 uint32 wire_flags = open_flags_to_wire(flags);
2354 size_t srclen = 2*(strlen(fname)+1);
2355
2356 param = SMB_MALLOC_ARRAY(char, 6+srclen+2);
2357 if (!param) {
2358 return false;
2359 }
2360 memset(param, '\0', 6);
2361 SSVAL(param,0, SMB_POSIX_PATH_OPEN);
2362 p = &param[6];
2363
2364 p += clistr_push(cli, p, fname, srclen, STR_TERMINATE);
2365 param_len = PTR_DIFF(p, param);
2366
2367 if (is_dir) {
2368 wire_flags &= ~(SMB_O_RDONLY|SMB_O_RDWR|SMB_O_WRONLY);
2369 wire_flags |= SMB_O_DIRECTORY;
2370 }
2371
2372 p = data;
2373 SIVAL(p,0,0); /* No oplock. */
2374 SIVAL(p,4,wire_flags);
2375 SIVAL(p,8,unix_perms_to_wire(mode));
2376 SIVAL(p,12,0); /* Top bits of perms currently undefined. */
2377 SSVAL(p,16,SMB_NO_INFO_LEVEL_RETURNED); /* No info level returned. */
2378
2379 data_len = 18;
2380
2381 if (!cli_send_trans(cli, SMBtrans2,
2382 NULL, /* name */
2383 -1, 0, /* fid, flags */
2384 &setup, 1, 0, /* setup, length, max */
2385 param, param_len, 2, /* param, length, max */
2386 (char *)&data, data_len, cli->max_xmit /* data, length, max */
2387 )) {
2388 SAFE_FREE(param);
2389 return -1;
2390 }
2391
2392 SAFE_FREE(param);
2393
2394 if (!cli_receive_trans(cli, SMBtrans2,
2395 &rparam, &param_len,
2396 &rdata, &data_len)) {
2397 return -1;
2398 }
2399
2400 fnum = SVAL(rdata,2);
2401
2402 SAFE_FREE(rdata);
2403 SAFE_FREE(rparam);
2404
2405 return fnum;
2406 }
2407
2408 /****************************************************************************
2409 open - POSIX semantics.
2410 ****************************************************************************/
2411
2412 int cli_posix_open(struct cli_state *cli, const char *fname, int flags, mode_t mode)
2413 {
2414 return cli_posix_open_internal(cli, fname, flags, mode, False);
2415 }
2416
2417 /****************************************************************************
2418 mkdir - POSIX semantics.
2419 ****************************************************************************/
2420
2421 int cli_posix_mkdir(struct cli_state *cli, const char *fname, mode_t mode)
2422 {
2423 return (cli_posix_open_internal(cli, fname, O_CREAT, mode, True) == -1) ? -1 : 0;
2424 }
2425
2426 /****************************************************************************
2427 unlink or rmdir - POSIX semantics.
2428 ****************************************************************************/
2429
2430 static bool cli_posix_unlink_internal(struct cli_state *cli, const char *fname, bool is_dir)
2431 {
2432 unsigned int data_len = 0;
2433 unsigned int param_len = 0;
2434 uint16 setup = TRANSACT2_SETPATHINFO;
2435 char *param;
2436 char data[2];
2437 char *rparam=NULL, *rdata=NULL;
2438 char *p;
2439 size_t srclen = 2*(strlen(fname)+1);
2440
2441 param = SMB_MALLOC_ARRAY(char, 6+srclen+2);
2442 if (!param) {
2443 return false;
2444 }
2445 memset(param, '\0', 6);
2446 SSVAL(param,0, SMB_POSIX_PATH_UNLINK);
2447 p = &param[6];
2448
2449 p += clistr_push(cli, p, fname, srclen, STR_TERMINATE);
2450 param_len = PTR_DIFF(p, param);
2451
2452 SSVAL(data, 0, is_dir ? SMB_POSIX_UNLINK_DIRECTORY_TARGET :
2453 SMB_POSIX_UNLINK_FILE_TARGET);
2454 data_len = 2;
2455
2456 if (!cli_send_trans(cli, SMBtrans2,
2457 NULL, /* name */
2458 -1, 0, /* fid, flags */
2459 &setup, 1, 0, /* setup, length, max */
2460 param, param_len, 2, /* param, length, max */
2461 (char *)&data, data_len, cli->max_xmit /* data, length, max */
2462 )) {
2463 SAFE_FREE(param);
2464 return False;
2465 }
2466
2467 SAFE_FREE(param);
2468
2469 if (!cli_receive_trans(cli, SMBtrans2,
2470 &rparam, &param_len,
2471 &rdata, &data_len)) {
2472 return False;
2473 }
2474
2475 SAFE_FREE(rdata);
2476 SAFE_FREE(rparam);
2477
2478 return True;
2479 }
2480
2481 /****************************************************************************
2482 unlink - POSIX semantics.
2483 ****************************************************************************/
2484
2485 bool cli_posix_unlink(struct cli_state *cli, const char *fname)
2486 {
2487 return cli_posix_unlink_internal(cli, fname, False);
2488 }
2489
2490 /****************************************************************************
2491 rmdir - POSIX semantics.
2492 ****************************************************************************/
2493
2494 int cli_posix_rmdir(struct cli_state *cli, const char *fname)
2495 {
2496 return cli_posix_unlink_internal(cli, fname, True);
2497 }
Samba GIT mirror