r17800: Start using struct timespec internally for file times
[idra/samba.git] / source3 / libsmb / clifile.c
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 2 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, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23
24 /****************************************************************************
25  Hard/Symlink a file (UNIX extensions).
26  Creates new name (sym)linked to oldname.
27 ****************************************************************************/
28
29 static BOOL cli_link_internal(struct cli_state *cli, const char *oldname, const char *newname, BOOL hard_link)
30 {
31         unsigned int data_len = 0;
32         unsigned int param_len = 0;
33         uint16 setup = TRANSACT2_SETPATHINFO;
34         char param[sizeof(pstring)+6];
35         pstring data;
36         char *rparam=NULL, *rdata=NULL;
37         char *p;
38         size_t oldlen = 2*(strlen(oldname)+1);
39         size_t newlen = 2*(strlen(newname)+1);
40
41         memset(param, 0, sizeof(param));
42         SSVAL(param,0,hard_link ? SMB_SET_FILE_UNIX_HLINK : SMB_SET_FILE_UNIX_LINK);
43         p = &param[6];
44
45         p += clistr_push(cli, p, newname, MIN(newlen, sizeof(param)-6), STR_TERMINATE);
46         param_len = PTR_DIFF(p, param);
47
48         p = data;
49         p += clistr_push(cli, p, oldname, MIN(oldlen,sizeof(data)), STR_TERMINATE);
50         data_len = PTR_DIFF(p, data);
51
52         if (!cli_send_trans(cli, SMBtrans2,
53                 NULL,                        /* name */
54                 -1, 0,                          /* fid, flags */
55                 &setup, 1, 0,                   /* setup, length, max */
56                 param, param_len, 2,            /* param, length, max */
57                 (char *)&data,  data_len, cli->max_xmit /* data, length, max */
58                 )) {
59                         return False;
60         }
61
62         if (!cli_receive_trans(cli, SMBtrans2,
63                 &rparam, &param_len,
64                 &rdata, &data_len)) {
65                         return False;
66         }
67
68         SAFE_FREE(rdata);
69         SAFE_FREE(rparam);
70
71         return True;
72 }
73
74 /****************************************************************************
75  Map standard UNIX permissions onto wire representations.
76 ****************************************************************************/
77
78 uint32 unix_perms_to_wire(mode_t perms)
79 {
80         unsigned int ret = 0;
81
82         ret |= ((perms & S_IXOTH) ?  UNIX_X_OTH : 0);
83         ret |= ((perms & S_IWOTH) ?  UNIX_W_OTH : 0);
84         ret |= ((perms & S_IROTH) ?  UNIX_R_OTH : 0);
85         ret |= ((perms & S_IXGRP) ?  UNIX_X_GRP : 0);
86         ret |= ((perms & S_IWGRP) ?  UNIX_W_GRP : 0);
87         ret |= ((perms & S_IRGRP) ?  UNIX_R_GRP : 0);
88         ret |= ((perms & S_IXUSR) ?  UNIX_X_USR : 0);
89         ret |= ((perms & S_IWUSR) ?  UNIX_W_USR : 0);
90         ret |= ((perms & S_IRUSR) ?  UNIX_R_USR : 0);
91 #ifdef S_ISVTX
92         ret |= ((perms & S_ISVTX) ?  UNIX_STICKY : 0);
93 #endif
94 #ifdef S_ISGID
95         ret |= ((perms & S_ISGID) ?  UNIX_SET_GID : 0);
96 #endif
97 #ifdef S_ISUID
98         ret |= ((perms & S_ISUID) ?  UNIX_SET_UID : 0);
99 #endif
100         return ret;
101 }
102
103 /****************************************************************************
104  Map wire permissions to standard UNIX.
105 ****************************************************************************/
106
107 mode_t wire_perms_to_unix(uint32 perms)
108 {
109         mode_t ret = (mode_t)0;
110
111         ret |= ((perms & UNIX_X_OTH) ? S_IXOTH : 0);
112         ret |= ((perms & UNIX_W_OTH) ? S_IWOTH : 0);
113         ret |= ((perms & UNIX_R_OTH) ? S_IROTH : 0);
114         ret |= ((perms & UNIX_X_GRP) ? S_IXGRP : 0);
115         ret |= ((perms & UNIX_W_GRP) ? S_IWGRP : 0);
116         ret |= ((perms & UNIX_R_GRP) ? S_IRGRP : 0);
117         ret |= ((perms & UNIX_X_USR) ? S_IXUSR : 0);
118         ret |= ((perms & UNIX_W_USR) ? S_IWUSR : 0);
119         ret |= ((perms & UNIX_R_USR) ? S_IRUSR : 0);
120 #ifdef S_ISVTX
121         ret |= ((perms & UNIX_STICKY) ? S_ISVTX : 0);
122 #endif
123 #ifdef S_ISGID
124         ret |= ((perms & UNIX_SET_GID) ? S_ISGID : 0);
125 #endif
126 #ifdef S_ISUID
127         ret |= ((perms & UNIX_SET_UID) ? S_ISUID : 0);
128 #endif
129         return ret;
130 }
131
132 /****************************************************************************
133  Return the file type from the wire filetype for UNIX extensions.
134 ****************************************************************************/
135                                                                                                                 
136 static mode_t unix_filetype_from_wire(uint32 wire_type)
137 {
138         switch (wire_type) {
139                 case UNIX_TYPE_FILE:
140                         return S_IFREG;
141                 case UNIX_TYPE_DIR:
142                         return S_IFDIR;
143 #ifdef S_IFLNK
144                 case UNIX_TYPE_SYMLINK:
145                         return S_IFLNK;
146 #endif
147 #ifdef S_IFCHR
148                 case UNIX_TYPE_CHARDEV:
149                         return S_IFCHR;
150 #endif
151 #ifdef S_IFBLK
152                 case UNIX_TYPE_BLKDEV:
153                         return S_IFBLK;
154 #endif
155 #ifdef S_IFIFO
156                 case UNIX_TYPE_FIFO:
157                         return S_IFIFO;
158 #endif
159 #ifdef S_IFSOCK
160                 case UNIX_TYPE_SOCKET:
161                         return S_IFSOCK;
162 #endif
163                 default:
164                         return (mode_t)0;
165         }
166 }
167
168 /****************************************************************************
169  Do a POSIX getfacl (UNIX extensions).
170 ****************************************************************************/
171
172 BOOL cli_unix_getfacl(struct cli_state *cli, const char *name, size_t *prb_size, char **retbuf)
173 {
174         unsigned int param_len = 0;
175         unsigned int data_len = 0;
176         uint16 setup = TRANSACT2_QPATHINFO;
177         char param[sizeof(pstring)+6];
178         char *rparam=NULL, *rdata=NULL;
179         char *p;
180
181         p = param;
182         memset(p, 0, 6);
183         SSVAL(p, 0, SMB_QUERY_POSIX_ACL);
184         p += 6;
185         p += clistr_push(cli, p, name, sizeof(pstring)-6, STR_TERMINATE);
186         param_len = PTR_DIFF(p, param);
187
188         if (!cli_send_trans(cli, SMBtrans2,
189                 NULL,                        /* name */
190                 -1, 0,                       /* fid, flags */
191                 &setup, 1, 0,                /* setup, length, max */
192                 param, param_len, 2,         /* param, length, max */
193                 NULL,  0, cli->max_xmit      /* data, length, max */
194                 )) {
195                         return False;
196         }
197
198         if (!cli_receive_trans(cli, SMBtrans2,
199                 &rparam, &param_len,
200                 &rdata, &data_len)) {
201                         return False;
202         }
203
204         if (data_len < 6) {
205                 SAFE_FREE(rdata);
206                 SAFE_FREE(rparam);
207                 return False;
208         }
209
210         SAFE_FREE(rparam);
211         *retbuf = rdata;
212         *prb_size = (size_t)data_len;
213
214         return True;
215 }
216
217 /****************************************************************************
218  Stat a file (UNIX extensions).
219 ****************************************************************************/
220
221 BOOL cli_unix_stat(struct cli_state *cli, const char *name, SMB_STRUCT_STAT *sbuf)
222 {
223         unsigned int param_len = 0;
224         unsigned int data_len = 0;
225         uint16 setup = TRANSACT2_QPATHINFO;
226         char param[sizeof(pstring)+6];
227         char *rparam=NULL, *rdata=NULL;
228         char *p;
229
230         ZERO_STRUCTP(sbuf);
231
232         p = param;
233         memset(p, 0, 6);
234         SSVAL(p, 0, SMB_QUERY_FILE_UNIX_BASIC);
235         p += 6;
236         p += clistr_push(cli, p, name, sizeof(pstring)-6, STR_TERMINATE);
237         param_len = PTR_DIFF(p, param);
238
239         if (!cli_send_trans(cli, SMBtrans2,
240                 NULL,                        /* name */
241                 -1, 0,                       /* fid, flags */
242                 &setup, 1, 0,                /* setup, length, max */
243                 param, param_len, 2,         /* param, length, max */
244                 NULL,  0, cli->max_xmit      /* data, length, max */
245                 )) {
246                         return False;
247         }
248
249         if (!cli_receive_trans(cli, SMBtrans2,
250                 &rparam, &param_len,
251                 &rdata, &data_len)) {
252                         return False;
253         }
254
255         if (data_len < 96) {
256                 SAFE_FREE(rdata);
257                 SAFE_FREE(rparam);
258                 return False;
259         }
260
261         sbuf->st_size = IVAL2_TO_SMB_BIG_UINT(rdata,0);     /* total size, in bytes */
262         sbuf->st_blocks = IVAL2_TO_SMB_BIG_UINT(rdata,8);   /* number of blocks allocated */
263 #if defined (HAVE_STAT_ST_BLOCKS) && defined(STAT_ST_BLOCKSIZE)
264         sbuf->st_blocks /= STAT_ST_BLOCKSIZE;
265 #else
266         /* assume 512 byte blocks */
267         sbuf->st_blocks /= 512;
268 #endif
269         set_ctimespec(sbuf, interpret_long_date(rdata + 16));    /* time of last change */
270         set_atimespec(sbuf, interpret_long_date(rdata + 24));    /* time of last access */
271         set_mtimespec(sbuf, interpret_long_date(rdata + 32));    /* time of last modification */
272
273         sbuf->st_uid = (uid_t) IVAL(rdata,40);      /* user ID of owner */
274         sbuf->st_gid = (gid_t) IVAL(rdata,48);      /* group ID of owner */
275         sbuf->st_mode |= unix_filetype_from_wire(IVAL(rdata, 56));
276 #if defined(HAVE_MAKEDEV)
277         {
278                 uint32 dev_major = IVAL(rdata,60);
279                 uint32 dev_minor = IVAL(rdata,68);
280                 sbuf->st_rdev = makedev(dev_major, dev_minor);
281         }
282 #endif
283         sbuf->st_ino = (SMB_INO_T)IVAL2_TO_SMB_BIG_UINT(rdata,76);      /* inode */
284         sbuf->st_mode |= wire_perms_to_unix(IVAL(rdata,84));     /* protection */
285         sbuf->st_nlink = IVAL(rdata,92);    /* number of hard links */
286
287         SAFE_FREE(rdata);
288         SAFE_FREE(rparam);
289
290         return True;
291 }
292
293 /****************************************************************************
294  Symlink a file (UNIX extensions).
295 ****************************************************************************/
296
297 BOOL cli_unix_symlink(struct cli_state *cli, const char *oldname, const char *newname)
298 {
299         return cli_link_internal(cli, oldname, newname, False);
300 }
301
302 /****************************************************************************
303  Hard a file (UNIX extensions).
304 ****************************************************************************/
305
306 BOOL cli_unix_hardlink(struct cli_state *cli, const char *oldname, const char *newname)
307 {
308         return cli_link_internal(cli, oldname, newname, True);
309 }
310
311 /****************************************************************************
312  Chmod or chown a file internal (UNIX extensions).
313 ****************************************************************************/
314
315 static BOOL cli_unix_chmod_chown_internal(struct cli_state *cli, const char *fname, uint32 mode, uint32 uid, uint32 gid)
316 {
317         unsigned int data_len = 0;
318         unsigned int param_len = 0;
319         uint16 setup = TRANSACT2_SETPATHINFO;
320         char param[sizeof(pstring)+6];
321         char data[100];
322         char *rparam=NULL, *rdata=NULL;
323         char *p;
324
325         memset(param, 0, sizeof(param));
326         memset(data, 0, sizeof(data));
327         SSVAL(param,0,SMB_SET_FILE_UNIX_BASIC);
328         p = &param[6];
329
330         p += clistr_push(cli, p, fname, -1, STR_TERMINATE);
331         param_len = PTR_DIFF(p, param);
332
333         SIVAL(data,40,uid);
334         SIVAL(data,48,gid);
335         SIVAL(data,84,mode);
336
337         data_len = 100;
338
339         if (!cli_send_trans(cli, SMBtrans2,
340                 NULL,                        /* name */
341                 -1, 0,                          /* fid, flags */
342                 &setup, 1, 0,                   /* setup, length, max */
343                 param, param_len, 2,            /* param, length, max */
344                 (char *)&data,  data_len, cli->max_xmit /* data, length, max */
345                 )) {
346                         return False;
347         }
348
349         if (!cli_receive_trans(cli, SMBtrans2,
350                 &rparam, &param_len,
351                 &rdata, &data_len)) {
352                         return False;
353         }
354
355         SAFE_FREE(rdata);
356         SAFE_FREE(rparam);
357
358         return True;
359 }
360
361 /****************************************************************************
362  chmod a file (UNIX extensions).
363 ****************************************************************************/
364
365 BOOL cli_unix_chmod(struct cli_state *cli, const char *fname, mode_t mode)
366 {
367         return cli_unix_chmod_chown_internal(cli, fname, 
368                 unix_perms_to_wire(mode), SMB_UID_NO_CHANGE, SMB_GID_NO_CHANGE);
369 }
370
371 /****************************************************************************
372  chown a file (UNIX extensions).
373 ****************************************************************************/
374
375 BOOL cli_unix_chown(struct cli_state *cli, const char *fname, uid_t uid, gid_t gid)
376 {
377         return cli_unix_chmod_chown_internal(cli, fname, SMB_MODE_NO_CHANGE, (uint32)uid, (uint32)gid);
378 }
379
380 /****************************************************************************
381  Rename a file.
382 ****************************************************************************/
383
384 BOOL cli_rename(struct cli_state *cli, const char *fname_src, const char *fname_dst)
385 {
386         char *p;
387
388         memset(cli->outbuf,'\0',smb_size);
389         memset(cli->inbuf,'\0',smb_size);
390
391         set_message(cli->outbuf,1, 0, True);
392
393         SCVAL(cli->outbuf,smb_com,SMBmv);
394         SSVAL(cli->outbuf,smb_tid,cli->cnum);
395         cli_setup_packet(cli);
396
397         SSVAL(cli->outbuf,smb_vwv0,aSYSTEM | aHIDDEN | aDIR);
398
399         p = smb_buf(cli->outbuf);
400         *p++ = 4;
401         p += clistr_push(cli, p, fname_src, -1, STR_TERMINATE);
402         *p++ = 4;
403         p += clistr_push(cli, p, fname_dst, -1, STR_TERMINATE);
404
405         cli_setup_bcc(cli, p);
406
407         cli_send_smb(cli);
408         if (!cli_receive_smb(cli))
409                 return False;
410
411         if (cli_is_error(cli))
412                 return False;
413
414         return True;
415 }
416
417 /****************************************************************************
418  NT Rename a file.
419 ****************************************************************************/
420
421 BOOL cli_ntrename(struct cli_state *cli, const char *fname_src, const char *fname_dst)
422 {
423         char *p;
424
425         memset(cli->outbuf,'\0',smb_size);
426         memset(cli->inbuf,'\0',smb_size);
427
428         set_message(cli->outbuf, 4, 0, True);
429
430         SCVAL(cli->outbuf,smb_com,SMBntrename);
431         SSVAL(cli->outbuf,smb_tid,cli->cnum);
432         cli_setup_packet(cli);
433
434         SSVAL(cli->outbuf,smb_vwv0,aSYSTEM | aHIDDEN | aDIR);
435         SSVAL(cli->outbuf,smb_vwv1, RENAME_FLAG_RENAME);
436
437         p = smb_buf(cli->outbuf);
438         *p++ = 4;
439         p += clistr_push(cli, p, fname_src, -1, STR_TERMINATE);
440         *p++ = 4;
441         p += clistr_push(cli, p, fname_dst, -1, STR_TERMINATE);
442
443         cli_setup_bcc(cli, p);
444
445         cli_send_smb(cli);
446         if (!cli_receive_smb(cli))
447                 return False;
448
449         if (cli_is_error(cli))
450                 return False;
451
452         return True;
453 }
454
455 /****************************************************************************
456  NT hardlink a file.
457 ****************************************************************************/
458
459 BOOL cli_nt_hardlink(struct cli_state *cli, const char *fname_src, const char *fname_dst)
460 {
461         char *p;
462
463         memset(cli->outbuf,'\0',smb_size);
464         memset(cli->inbuf,'\0',smb_size);
465
466         set_message(cli->outbuf, 4, 0, True);
467
468         SCVAL(cli->outbuf,smb_com,SMBntrename);
469         SSVAL(cli->outbuf,smb_tid,cli->cnum);
470         cli_setup_packet(cli);
471
472         SSVAL(cli->outbuf,smb_vwv0,aSYSTEM | aHIDDEN | aDIR);
473         SSVAL(cli->outbuf,smb_vwv1, RENAME_FLAG_HARD_LINK);
474
475         p = smb_buf(cli->outbuf);
476         *p++ = 4;
477         p += clistr_push(cli, p, fname_src, -1, STR_TERMINATE);
478         *p++ = 4;
479         p += clistr_push(cli, p, fname_dst, -1, STR_TERMINATE);
480
481         cli_setup_bcc(cli, p);
482
483         cli_send_smb(cli);
484         if (!cli_receive_smb(cli))
485                 return False;
486
487         if (cli_is_error(cli))
488                 return False;
489
490         return True;
491 }
492
493 /****************************************************************************
494  Delete a file.
495 ****************************************************************************/
496
497 BOOL cli_unlink(struct cli_state *cli, const char *fname)
498 {
499         char *p;
500
501         memset(cli->outbuf,'\0',smb_size);
502         memset(cli->inbuf,'\0',smb_size);
503
504         set_message(cli->outbuf,1, 0,True);
505
506         SCVAL(cli->outbuf,smb_com,SMBunlink);
507         SSVAL(cli->outbuf,smb_tid,cli->cnum);
508         cli_setup_packet(cli);
509
510         SSVAL(cli->outbuf,smb_vwv0,aSYSTEM | aHIDDEN);
511   
512         p = smb_buf(cli->outbuf);
513         *p++ = 4;      
514         p += clistr_push(cli, p, fname, -1, STR_TERMINATE);
515
516         cli_setup_bcc(cli, p);
517         cli_send_smb(cli);
518         if (!cli_receive_smb(cli)) {
519                 return False;
520         }
521
522         if (cli_is_error(cli)) {
523                 return False;
524         }
525
526         return True;
527 }
528
529 /****************************************************************************
530  Create a directory.
531 ****************************************************************************/
532
533 BOOL cli_mkdir(struct cli_state *cli, const char *dname)
534 {
535         char *p;
536
537         memset(cli->outbuf,'\0',smb_size);
538         memset(cli->inbuf,'\0',smb_size);
539
540         set_message(cli->outbuf,0, 0,True);
541
542         SCVAL(cli->outbuf,smb_com,SMBmkdir);
543         SSVAL(cli->outbuf,smb_tid,cli->cnum);
544         cli_setup_packet(cli);
545
546         p = smb_buf(cli->outbuf);
547         *p++ = 4;      
548         p += clistr_push(cli, p, dname, -1, STR_TERMINATE);
549
550         cli_setup_bcc(cli, p);
551
552         cli_send_smb(cli);
553         if (!cli_receive_smb(cli)) {
554                 return False;
555         }
556
557         if (cli_is_error(cli)) {
558                 return False;
559         }
560
561         return True;
562 }
563
564 /****************************************************************************
565  Remove a directory.
566 ****************************************************************************/
567
568 BOOL cli_rmdir(struct cli_state *cli, const char *dname)
569 {
570         char *p;
571
572         memset(cli->outbuf,'\0',smb_size);
573         memset(cli->inbuf,'\0',smb_size);
574
575         set_message(cli->outbuf,0, 0, True);
576
577         SCVAL(cli->outbuf,smb_com,SMBrmdir);
578         SSVAL(cli->outbuf,smb_tid,cli->cnum);
579         cli_setup_packet(cli);
580
581         p = smb_buf(cli->outbuf);
582         *p++ = 4;      
583         p += clistr_push(cli, p, dname, -1, STR_TERMINATE);
584
585         cli_setup_bcc(cli, p);
586
587         cli_send_smb(cli);
588         if (!cli_receive_smb(cli)) {
589                 return False;
590         }
591
592         if (cli_is_error(cli)) {
593                 return False;
594         }
595
596         return True;
597 }
598
599 /****************************************************************************
600  Set or clear the delete on close flag.
601 ****************************************************************************/
602
603 int cli_nt_delete_on_close(struct cli_state *cli, int fnum, BOOL flag)
604 {
605         unsigned int data_len = 1;
606         unsigned int param_len = 6;
607         uint16 setup = TRANSACT2_SETFILEINFO;
608         pstring param;
609         unsigned char data;
610         char *rparam=NULL, *rdata=NULL;
611
612         memset(param, 0, param_len);
613         SSVAL(param,0,fnum);
614         SSVAL(param,2,SMB_SET_FILE_DISPOSITION_INFO);
615
616         data = flag ? 1 : 0;
617
618         if (!cli_send_trans(cli, SMBtrans2,
619                                                 NULL,                        /* name */
620                                                 -1, 0,                          /* fid, flags */
621                                                 &setup, 1, 0,                   /* setup, length, max */
622                                                 param, param_len, 2,            /* param, length, max */
623                                                 (char *)&data,  data_len, cli->max_xmit /* data, length, max */
624                                                 )) {
625                 return False;
626         }
627
628         if (!cli_receive_trans(cli, SMBtrans2,
629                                                 &rparam, &param_len,
630                                                 &rdata, &data_len)) {
631                 return False;
632         }
633
634         SAFE_FREE(rdata);
635         SAFE_FREE(rparam);
636
637         return True;
638 }
639
640 /****************************************************************************
641  Open a file - exposing the full horror of the NT API :-).
642  Used in smbtorture.
643 ****************************************************************************/
644
645 int cli_nt_create_full(struct cli_state *cli, const char *fname, 
646                  uint32 CreatFlags, uint32 DesiredAccess,
647                  uint32 FileAttributes, uint32 ShareAccess,
648                  uint32 CreateDisposition, uint32 CreateOptions,
649                  uint8 SecuityFlags)
650 {
651         char *p;
652         int len;
653
654         memset(cli->outbuf,'\0',smb_size);
655         memset(cli->inbuf,'\0',smb_size);
656
657         set_message(cli->outbuf,24,0,True);
658
659         SCVAL(cli->outbuf,smb_com,SMBntcreateX);
660         SSVAL(cli->outbuf,smb_tid,cli->cnum);
661         cli_setup_packet(cli);
662
663         SSVAL(cli->outbuf,smb_vwv0,0xFF);
664         if (cli->use_oplocks)
665                 CreatFlags |= (REQUEST_OPLOCK|REQUEST_BATCH_OPLOCK);
666         
667         SIVAL(cli->outbuf,smb_ntcreate_Flags, CreatFlags);
668         SIVAL(cli->outbuf,smb_ntcreate_RootDirectoryFid, 0x0);
669         SIVAL(cli->outbuf,smb_ntcreate_DesiredAccess, DesiredAccess);
670         SIVAL(cli->outbuf,smb_ntcreate_FileAttributes, FileAttributes);
671         SIVAL(cli->outbuf,smb_ntcreate_ShareAccess, ShareAccess);
672         SIVAL(cli->outbuf,smb_ntcreate_CreateDisposition, CreateDisposition);
673         SIVAL(cli->outbuf,smb_ntcreate_CreateOptions, CreateOptions);
674         SIVAL(cli->outbuf,smb_ntcreate_ImpersonationLevel, 0x02);
675         SCVAL(cli->outbuf,smb_ntcreate_SecurityFlags, SecuityFlags);
676
677         p = smb_buf(cli->outbuf);
678         /* this alignment and termination is critical for netapp filers. Don't change */
679         p += clistr_align_out(cli, p, 0);
680         len = clistr_push(cli, p, fname, -1, 0);
681         p += len;
682         SSVAL(cli->outbuf,smb_ntcreate_NameLength, len);
683         /* sigh. this copes with broken netapp filer behaviour */
684         p += clistr_push(cli, p, "", -1, STR_TERMINATE);
685
686         cli_setup_bcc(cli, p);
687
688         cli_send_smb(cli);
689         if (!cli_receive_smb(cli)) {
690                 return -1;
691         }
692
693         if (cli_is_error(cli)) {
694                 return -1;
695         }
696
697         return SVAL(cli->inbuf,smb_vwv2 + 1);
698 }
699
700 /****************************************************************************
701  Open a file.
702 ****************************************************************************/
703
704 int cli_nt_create(struct cli_state *cli, const char *fname, uint32 DesiredAccess)
705 {
706         return cli_nt_create_full(cli, fname, 0, DesiredAccess, 0,
707                                 FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN, 0x0, 0x0);
708 }
709
710 /****************************************************************************
711  Open a file
712  WARNING: if you open with O_WRONLY then getattrE won't work!
713 ****************************************************************************/
714
715 int cli_open(struct cli_state *cli, const char *fname, int flags, int share_mode)
716 {
717         char *p;
718         unsigned openfn=0;
719         unsigned accessmode=0;
720
721         if (flags & O_CREAT)
722                 openfn |= (1<<4);
723         if (!(flags & O_EXCL)) {
724                 if (flags & O_TRUNC)
725                         openfn |= (1<<1);
726                 else
727                         openfn |= (1<<0);
728         }
729
730         accessmode = (share_mode<<4);
731
732         if ((flags & O_ACCMODE) == O_RDWR) {
733                 accessmode |= 2;
734         } else if ((flags & O_ACCMODE) == O_WRONLY) {
735                 accessmode |= 1;
736         } 
737
738 #if defined(O_SYNC)
739         if ((flags & O_SYNC) == O_SYNC) {
740                 accessmode |= (1<<14);
741         }
742 #endif /* O_SYNC */
743
744         if (share_mode == DENY_FCB) {
745                 accessmode = 0xFF;
746         }
747
748         memset(cli->outbuf,'\0',smb_size);
749         memset(cli->inbuf,'\0',smb_size);
750
751         set_message(cli->outbuf,15,0,True);
752
753         SCVAL(cli->outbuf,smb_com,SMBopenX);
754         SSVAL(cli->outbuf,smb_tid,cli->cnum);
755         cli_setup_packet(cli);
756
757         SSVAL(cli->outbuf,smb_vwv0,0xFF);
758         SSVAL(cli->outbuf,smb_vwv2,0);  /* no additional info */
759         SSVAL(cli->outbuf,smb_vwv3,accessmode);
760         SSVAL(cli->outbuf,smb_vwv4,aSYSTEM | aHIDDEN);
761         SSVAL(cli->outbuf,smb_vwv5,0);
762         SSVAL(cli->outbuf,smb_vwv8,openfn);
763
764         if (cli->use_oplocks) {
765                 /* if using oplocks then ask for a batch oplock via
766                    core and extended methods */
767                 SCVAL(cli->outbuf,smb_flg, CVAL(cli->outbuf,smb_flg)|
768                         FLAG_REQUEST_OPLOCK|FLAG_REQUEST_BATCH_OPLOCK);
769                 SSVAL(cli->outbuf,smb_vwv2,SVAL(cli->outbuf,smb_vwv2) | 6);
770         }
771   
772         p = smb_buf(cli->outbuf);
773         p += clistr_push(cli, p, fname, -1, STR_TERMINATE);
774
775         cli_setup_bcc(cli, p);
776
777         cli_send_smb(cli);
778         if (!cli_receive_smb(cli)) {
779                 return -1;
780         }
781
782         if (cli_is_error(cli)) {
783                 return -1;
784         }
785
786         return SVAL(cli->inbuf,smb_vwv2);
787 }
788
789 /****************************************************************************
790  Close a file.
791 ****************************************************************************/
792
793 BOOL cli_close(struct cli_state *cli, int fnum)
794 {
795         memset(cli->outbuf,'\0',smb_size);
796         memset(cli->inbuf,'\0',smb_size);
797
798         set_message(cli->outbuf,3,0,True);
799
800         SCVAL(cli->outbuf,smb_com,SMBclose);
801         SSVAL(cli->outbuf,smb_tid,cli->cnum);
802         cli_setup_packet(cli);
803
804         SSVAL(cli->outbuf,smb_vwv0,fnum);
805         SIVALS(cli->outbuf,smb_vwv1,-1);
806
807         cli_send_smb(cli);
808         if (!cli_receive_smb(cli)) {
809                 return False;
810         }
811
812         return !cli_is_error(cli);
813 }
814
815
816 /****************************************************************************
817  send a lock with a specified locktype 
818  this is used for testing LOCKING_ANDX_CANCEL_LOCK
819 ****************************************************************************/
820
821 NTSTATUS cli_locktype(struct cli_state *cli, int fnum, 
822                       uint32 offset, uint32 len, int timeout, unsigned char locktype)
823 {
824         char *p;
825         int saved_timeout = cli->timeout;
826
827         memset(cli->outbuf,'\0',smb_size);
828         memset(cli->inbuf,'\0', smb_size);
829
830         set_message(cli->outbuf,8,0,True);
831
832         SCVAL(cli->outbuf,smb_com,SMBlockingX);
833         SSVAL(cli->outbuf,smb_tid,cli->cnum);
834         cli_setup_packet(cli);
835
836         SCVAL(cli->outbuf,smb_vwv0,0xFF);
837         SSVAL(cli->outbuf,smb_vwv2,fnum);
838         SCVAL(cli->outbuf,smb_vwv3,locktype);
839         SIVALS(cli->outbuf, smb_vwv4, timeout);
840         SSVAL(cli->outbuf,smb_vwv6,0);
841         SSVAL(cli->outbuf,smb_vwv7,1);
842
843         p = smb_buf(cli->outbuf);
844         SSVAL(p, 0, cli->pid);
845         SIVAL(p, 2, offset);
846         SIVAL(p, 6, len);
847
848         p += 10;
849
850         cli_setup_bcc(cli, p);
851
852         cli_send_smb(cli);
853
854         if (timeout != 0) {
855                 cli->timeout = (timeout == -1) ? 0x7FFFFFFF : (timeout + 2*1000);
856         }
857
858         if (!cli_receive_smb(cli)) {
859                 cli->timeout = saved_timeout;
860                 return NT_STATUS_UNSUCCESSFUL;
861         }
862
863         cli->timeout = saved_timeout;
864
865         return cli_nt_error(cli);
866 }
867
868 /****************************************************************************
869  Lock a file.
870  note that timeout is in units of 2 milliseconds
871 ****************************************************************************/
872
873 BOOL cli_lock(struct cli_state *cli, int fnum, 
874               uint32 offset, uint32 len, int timeout, enum brl_type lock_type)
875 {
876         char *p;
877         int saved_timeout = cli->timeout;
878
879         memset(cli->outbuf,'\0',smb_size);
880         memset(cli->inbuf,'\0', smb_size);
881
882         set_message(cli->outbuf,8,0,True);
883
884         SCVAL(cli->outbuf,smb_com,SMBlockingX);
885         SSVAL(cli->outbuf,smb_tid,cli->cnum);
886         cli_setup_packet(cli);
887
888         SCVAL(cli->outbuf,smb_vwv0,0xFF);
889         SSVAL(cli->outbuf,smb_vwv2,fnum);
890         SCVAL(cli->outbuf,smb_vwv3,(lock_type == READ_LOCK? 1 : 0));
891         SIVALS(cli->outbuf, smb_vwv4, timeout);
892         SSVAL(cli->outbuf,smb_vwv6,0);
893         SSVAL(cli->outbuf,smb_vwv7,1);
894
895         p = smb_buf(cli->outbuf);
896         SSVAL(p, 0, cli->pid);
897         SIVAL(p, 2, offset);
898         SIVAL(p, 6, len);
899
900         p += 10;
901
902         cli_setup_bcc(cli, p);
903
904         cli_send_smb(cli);
905
906         if (timeout != 0) {
907                 cli->timeout = (timeout == -1) ? 0x7FFFFFFF : (timeout*2 + 5*1000);
908         }
909
910         if (!cli_receive_smb(cli)) {
911                 cli->timeout = saved_timeout;
912                 return False;
913         }
914
915         cli->timeout = saved_timeout;
916
917         if (cli_is_error(cli)) {
918                 return False;
919         }
920
921         return True;
922 }
923
924 /****************************************************************************
925  Unlock a file.
926 ****************************************************************************/
927
928 BOOL cli_unlock(struct cli_state *cli, int fnum, uint32 offset, uint32 len)
929 {
930         char *p;
931
932         memset(cli->outbuf,'\0',smb_size);
933         memset(cli->inbuf,'\0',smb_size);
934
935         set_message(cli->outbuf,8,0,True);
936
937         SCVAL(cli->outbuf,smb_com,SMBlockingX);
938         SSVAL(cli->outbuf,smb_tid,cli->cnum);
939         cli_setup_packet(cli);
940
941         SCVAL(cli->outbuf,smb_vwv0,0xFF);
942         SSVAL(cli->outbuf,smb_vwv2,fnum);
943         SCVAL(cli->outbuf,smb_vwv3,0);
944         SIVALS(cli->outbuf, smb_vwv4, 0);
945         SSVAL(cli->outbuf,smb_vwv6,1);
946         SSVAL(cli->outbuf,smb_vwv7,0);
947
948         p = smb_buf(cli->outbuf);
949         SSVAL(p, 0, cli->pid);
950         SIVAL(p, 2, offset);
951         SIVAL(p, 6, len);
952         p += 10;
953         cli_setup_bcc(cli, p);
954         cli_send_smb(cli);
955         if (!cli_receive_smb(cli)) {
956                 return False;
957         }
958
959         if (cli_is_error(cli)) {
960                 return False;
961         }
962
963         return True;
964 }
965
966 /****************************************************************************
967  Lock a file with 64 bit offsets.
968 ****************************************************************************/
969
970 BOOL cli_lock64(struct cli_state *cli, int fnum, 
971                 SMB_BIG_UINT offset, SMB_BIG_UINT len, int timeout, enum brl_type lock_type)
972 {
973         char *p;
974         int saved_timeout = cli->timeout;
975         int ltype;
976
977         if (! (cli->capabilities & CAP_LARGE_FILES)) {
978                 return cli_lock(cli, fnum, offset, len, timeout, lock_type);
979         }
980
981         ltype = (lock_type == READ_LOCK? 1 : 0);
982         ltype |= LOCKING_ANDX_LARGE_FILES;
983
984         memset(cli->outbuf,'\0',smb_size);
985         memset(cli->inbuf,'\0', smb_size);
986
987         set_message(cli->outbuf,8,0,True);
988
989         SCVAL(cli->outbuf,smb_com,SMBlockingX);
990         SSVAL(cli->outbuf,smb_tid,cli->cnum);
991         cli_setup_packet(cli);
992
993         SCVAL(cli->outbuf,smb_vwv0,0xFF);
994         SSVAL(cli->outbuf,smb_vwv2,fnum);
995         SCVAL(cli->outbuf,smb_vwv3,ltype);
996         SIVALS(cli->outbuf, smb_vwv4, timeout);
997         SSVAL(cli->outbuf,smb_vwv6,0);
998         SSVAL(cli->outbuf,smb_vwv7,1);
999
1000         p = smb_buf(cli->outbuf);
1001         SIVAL(p, 0, cli->pid);
1002         SOFF_T_R(p, 4, offset);
1003         SOFF_T_R(p, 12, len);
1004         p += 20;
1005
1006         cli_setup_bcc(cli, p);
1007         cli_send_smb(cli);
1008
1009         if (timeout != 0) {
1010                 cli->timeout = (timeout == -1) ? 0x7FFFFFFF : (timeout + 5*1000);
1011         }
1012
1013         if (!cli_receive_smb(cli)) {
1014                 cli->timeout = saved_timeout;
1015                 return False;
1016         }
1017
1018         cli->timeout = saved_timeout;
1019
1020         if (cli_is_error(cli)) {
1021                 return False;
1022         }
1023
1024         return True;
1025 }
1026
1027 /****************************************************************************
1028  Unlock a file with 64 bit offsets.
1029 ****************************************************************************/
1030
1031 BOOL cli_unlock64(struct cli_state *cli, int fnum, SMB_BIG_UINT offset, SMB_BIG_UINT len)
1032 {
1033         char *p;
1034
1035         if (! (cli->capabilities & CAP_LARGE_FILES)) {
1036                 return cli_unlock(cli, fnum, offset, len);
1037         }
1038
1039         memset(cli->outbuf,'\0',smb_size);
1040         memset(cli->inbuf,'\0',smb_size);
1041
1042         set_message(cli->outbuf,8,0,True);
1043
1044         SCVAL(cli->outbuf,smb_com,SMBlockingX);
1045         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1046         cli_setup_packet(cli);
1047
1048         SCVAL(cli->outbuf,smb_vwv0,0xFF);
1049         SSVAL(cli->outbuf,smb_vwv2,fnum);
1050         SCVAL(cli->outbuf,smb_vwv3,LOCKING_ANDX_LARGE_FILES);
1051         SIVALS(cli->outbuf, smb_vwv4, 0);
1052         SSVAL(cli->outbuf,smb_vwv6,1);
1053         SSVAL(cli->outbuf,smb_vwv7,0);
1054
1055         p = smb_buf(cli->outbuf);
1056         SIVAL(p, 0, cli->pid);
1057         SOFF_T_R(p, 4, offset);
1058         SOFF_T_R(p, 12, len);
1059         p += 20;
1060         cli_setup_bcc(cli, p);
1061         cli_send_smb(cli);
1062         if (!cli_receive_smb(cli)) {
1063                 return False;
1064         }
1065
1066         if (cli_is_error(cli)) {
1067                 return False;
1068         }
1069
1070         return True;
1071 }
1072
1073 /****************************************************************************
1074  Get/unlock a POSIX lock on a file - internal function.
1075 ****************************************************************************/
1076
1077 static BOOL cli_posix_lock_internal(struct cli_state *cli, int fnum, 
1078                 SMB_BIG_UINT offset, SMB_BIG_UINT len, BOOL wait_lock, enum brl_type lock_type)
1079 {
1080         unsigned int param_len = 4;
1081         unsigned int data_len = POSIX_LOCK_DATA_SIZE;
1082         uint16 setup = TRANSACT2_SETFILEINFO;
1083         char param[4];
1084         unsigned char data[POSIX_LOCK_DATA_SIZE];
1085         char *rparam=NULL, *rdata=NULL;
1086         int saved_timeout = cli->timeout;
1087
1088         SSVAL(param,0,fnum);
1089         SSVAL(param,2,SMB_SET_POSIX_LOCK);
1090
1091         switch (lock_type) {
1092                 case READ_LOCK:
1093                         SSVAL(data, POSIX_LOCK_TYPE_OFFSET, POSIX_LOCK_TYPE_READ);
1094                         break;
1095                 case WRITE_LOCK:
1096                         SSVAL(data, POSIX_LOCK_TYPE_OFFSET, POSIX_LOCK_TYPE_WRITE);
1097                         break;
1098                 case UNLOCK_LOCK:
1099                         SSVAL(data, POSIX_LOCK_TYPE_OFFSET, POSIX_LOCK_TYPE_UNLOCK);
1100                         break;
1101                 default:
1102                         return False;
1103         }
1104
1105         if (wait_lock) {
1106                 SSVAL(data, POSIX_LOCK_FLAGS_OFFSET, POSIX_LOCK_FLAG_WAIT);
1107                 cli->timeout = 0x7FFFFFFF;
1108         } else {
1109                 SSVAL(data, POSIX_LOCK_FLAGS_OFFSET, POSIX_LOCK_FLAG_NOWAIT);
1110         }
1111
1112         SIVAL(data, POSIX_LOCK_PID_OFFSET, cli->pid);
1113         SOFF_T(data, POSIX_LOCK_START_OFFSET, offset);
1114         SOFF_T(data, POSIX_LOCK_LEN_OFFSET, len);
1115
1116         if (!cli_send_trans(cli, SMBtrans2,
1117                                 NULL,                        /* name */
1118                                 -1, 0,                          /* fid, flags */
1119                                 &setup, 1, 0,                   /* setup, length, max */
1120                                 param, param_len, 2,            /* param, length, max */
1121                                 (char *)&data,  data_len, cli->max_xmit /* data, length, max */
1122                                 )) {
1123                 cli->timeout = saved_timeout;
1124                 return False;
1125         }
1126
1127         if (!cli_receive_trans(cli, SMBtrans2,
1128                                 &rparam, &param_len,
1129                                 &rdata, &data_len)) {
1130                 cli->timeout = saved_timeout;
1131                 SAFE_FREE(rdata);
1132                 SAFE_FREE(rparam);
1133                 return False;
1134         }
1135
1136         cli->timeout = saved_timeout;
1137
1138         SAFE_FREE(rdata);
1139         SAFE_FREE(rparam);
1140
1141         return True;
1142 }
1143
1144 /****************************************************************************
1145  POSIX Lock a file.
1146 ****************************************************************************/
1147
1148 BOOL cli_posix_lock(struct cli_state *cli, int fnum,
1149                         SMB_BIG_UINT offset, SMB_BIG_UINT len,
1150                         BOOL wait_lock, enum brl_type lock_type)
1151 {
1152         if (lock_type != READ_LOCK && lock_type != WRITE_LOCK) {
1153                 return False;
1154         }
1155         return cli_posix_lock_internal(cli, fnum, offset, len, wait_lock, lock_type);
1156 }
1157
1158 /****************************************************************************
1159  POSIX Unlock a file.
1160 ****************************************************************************/
1161
1162 BOOL cli_posix_unlock(struct cli_state *cli, int fnum, SMB_BIG_UINT offset, SMB_BIG_UINT len)
1163 {
1164         return cli_posix_lock_internal(cli, fnum, offset, len, False, UNLOCK_LOCK);
1165 }
1166
1167 /****************************************************************************
1168  POSIX Get any lock covering a file.
1169 ****************************************************************************/
1170
1171 BOOL cli_posix_getlock(struct cli_state *cli, int fnum, SMB_BIG_UINT *poffset, SMB_BIG_UINT *plen)
1172 {
1173         return True;
1174 }
1175
1176 /****************************************************************************
1177  Do a SMBgetattrE call.
1178 ****************************************************************************/
1179
1180 BOOL cli_getattrE(struct cli_state *cli, int fd, 
1181                   uint16 *attr, SMB_OFF_T *size, 
1182                   time_t *c_time, time_t *a_time, time_t *m_time)
1183 {
1184         memset(cli->outbuf,'\0',smb_size);
1185         memset(cli->inbuf,'\0',smb_size);
1186
1187         set_message(cli->outbuf,1,0,True);
1188
1189         SCVAL(cli->outbuf,smb_com,SMBgetattrE);
1190         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1191         cli_setup_packet(cli);
1192
1193         SSVAL(cli->outbuf,smb_vwv0,fd);
1194
1195         cli_send_smb(cli);
1196         if (!cli_receive_smb(cli)) {
1197                 return False;
1198         }
1199         
1200         if (cli_is_error(cli)) {
1201                 return False;
1202         }
1203
1204         if (size) {
1205                 *size = IVAL(cli->inbuf, smb_vwv6);
1206         }
1207
1208         if (attr) {
1209                 *attr = SVAL(cli->inbuf,smb_vwv10);
1210         }
1211
1212         if (c_time) {
1213                 *c_time = cli_make_unix_date2(cli, cli->inbuf+smb_vwv0);
1214         }
1215
1216         if (a_time) {
1217                 *a_time = cli_make_unix_date2(cli, cli->inbuf+smb_vwv2);
1218         }
1219
1220         if (m_time) {
1221                 *m_time = cli_make_unix_date2(cli, cli->inbuf+smb_vwv4);
1222         }
1223
1224         return True;
1225 }
1226
1227 /****************************************************************************
1228  Do a SMBgetatr call
1229 ****************************************************************************/
1230
1231 BOOL cli_getatr(struct cli_state *cli, const char *fname, 
1232                 uint16 *attr, SMB_OFF_T *size, time_t *t)
1233 {
1234         char *p;
1235
1236         memset(cli->outbuf,'\0',smb_size);
1237         memset(cli->inbuf,'\0',smb_size);
1238
1239         set_message(cli->outbuf,0,0,True);
1240
1241         SCVAL(cli->outbuf,smb_com,SMBgetatr);
1242         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1243         cli_setup_packet(cli);
1244
1245         p = smb_buf(cli->outbuf);
1246         *p++ = 4;
1247         p += clistr_push(cli, p, fname, -1, STR_TERMINATE);
1248
1249         cli_setup_bcc(cli, p);
1250
1251         cli_send_smb(cli);
1252         if (!cli_receive_smb(cli)) {
1253                 return False;
1254         }
1255         
1256         if (cli_is_error(cli)) {
1257                 return False;
1258         }
1259
1260         if (size) {
1261                 *size = IVAL(cli->inbuf, smb_vwv3);
1262         }
1263
1264         if (t) {
1265                 *t = cli_make_unix_date3(cli, cli->inbuf+smb_vwv1);
1266         }
1267
1268         if (attr) {
1269                 *attr = SVAL(cli->inbuf,smb_vwv0);
1270         }
1271
1272
1273         return True;
1274 }
1275
1276 /****************************************************************************
1277  Do a SMBsetattrE call.
1278 ****************************************************************************/
1279
1280 BOOL cli_setattrE(struct cli_state *cli, int fd,
1281                   time_t c_time, time_t a_time, time_t m_time)
1282
1283 {
1284         char *p;
1285
1286         memset(cli->outbuf,'\0',smb_size);
1287         memset(cli->inbuf,'\0',smb_size);
1288
1289         set_message(cli->outbuf,7,0,True);
1290
1291         SCVAL(cli->outbuf,smb_com,SMBsetattrE);
1292         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1293         cli_setup_packet(cli);
1294
1295         SSVAL(cli->outbuf,smb_vwv0, fd);
1296         cli_put_dos_date2(cli, cli->outbuf,smb_vwv1, c_time);
1297         cli_put_dos_date2(cli, cli->outbuf,smb_vwv3, a_time);
1298         cli_put_dos_date2(cli, cli->outbuf,smb_vwv5, m_time);
1299
1300         p = smb_buf(cli->outbuf);
1301         *p++ = 4;
1302
1303         cli_setup_bcc(cli, p);
1304
1305         cli_send_smb(cli);
1306         if (!cli_receive_smb(cli)) {
1307                 return False;
1308         }
1309         
1310         if (cli_is_error(cli)) {
1311                 return False;
1312         }
1313
1314         return True;
1315 }
1316
1317 /****************************************************************************
1318  Do a SMBsetatr call.
1319 ****************************************************************************/
1320
1321 BOOL cli_setatr(struct cli_state *cli, const char *fname, uint16 attr, time_t t)
1322 {
1323         char *p;
1324
1325         memset(cli->outbuf,'\0',smb_size);
1326         memset(cli->inbuf,'\0',smb_size);
1327
1328         set_message(cli->outbuf,8,0,True);
1329
1330         SCVAL(cli->outbuf,smb_com,SMBsetatr);
1331         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1332         cli_setup_packet(cli);
1333
1334         SSVAL(cli->outbuf,smb_vwv0, attr);
1335         cli_put_dos_date3(cli, cli->outbuf,smb_vwv1, t);
1336
1337         p = smb_buf(cli->outbuf);
1338         *p++ = 4;
1339         p += clistr_push(cli, p, fname, -1, STR_TERMINATE);
1340         *p++ = 4;
1341
1342         cli_setup_bcc(cli, p);
1343
1344         cli_send_smb(cli);
1345         if (!cli_receive_smb(cli)) {
1346                 return False;
1347         }
1348         
1349         if (cli_is_error(cli)) {
1350                 return False;
1351         }
1352
1353         return True;
1354 }
1355
1356 /****************************************************************************
1357  Check for existance of a dir.
1358 ****************************************************************************/
1359 BOOL cli_chkpath(struct cli_state *cli, const char *path)
1360 {
1361         pstring path2;
1362         char *p;
1363         
1364         pstrcpy(path2,path);
1365         trim_char(path2,'\0','\\');
1366         if (!*path2)
1367                 *path2 = '\\';
1368         
1369         memset(cli->outbuf,'\0',smb_size);
1370         set_message(cli->outbuf,0,0,True);
1371         SCVAL(cli->outbuf,smb_com,SMBchkpth);
1372         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1373         cli_setup_packet(cli);
1374         p = smb_buf(cli->outbuf);
1375         *p++ = 4;
1376         p += clistr_push(cli, p, path2, -1, STR_TERMINATE);
1377
1378         cli_setup_bcc(cli, p);
1379
1380         cli_send_smb(cli);
1381         if (!cli_receive_smb(cli)) {
1382                 return False;
1383         }
1384
1385         if (cli_is_error(cli)) return False;
1386
1387         return True;
1388 }
1389
1390 /****************************************************************************
1391  Query disk space.
1392 ****************************************************************************/
1393
1394 BOOL cli_dskattr(struct cli_state *cli, int *bsize, int *total, int *avail)
1395 {
1396         memset(cli->outbuf,'\0',smb_size);
1397         set_message(cli->outbuf,0,0,True);
1398         SCVAL(cli->outbuf,smb_com,SMBdskattr);
1399         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1400         cli_setup_packet(cli);
1401
1402         cli_send_smb(cli);
1403         if (!cli_receive_smb(cli)) {
1404                 return False;
1405         }
1406
1407         *bsize = SVAL(cli->inbuf,smb_vwv1)*SVAL(cli->inbuf,smb_vwv2);
1408         *total = SVAL(cli->inbuf,smb_vwv0);
1409         *avail = SVAL(cli->inbuf,smb_vwv3);
1410         
1411         return True;
1412 }
1413
1414 /****************************************************************************
1415  Create and open a temporary file.
1416 ****************************************************************************/
1417
1418 int cli_ctemp(struct cli_state *cli, const char *path, char **tmp_path)
1419 {
1420         int len;
1421         char *p;
1422
1423         memset(cli->outbuf,'\0',smb_size);
1424         memset(cli->inbuf,'\0',smb_size);
1425
1426         set_message(cli->outbuf,3,0,True);
1427
1428         SCVAL(cli->outbuf,smb_com,SMBctemp);
1429         SSVAL(cli->outbuf,smb_tid,cli->cnum);
1430         cli_setup_packet(cli);
1431
1432         SSVAL(cli->outbuf,smb_vwv0,0);
1433         SIVALS(cli->outbuf,smb_vwv1,-1);
1434
1435         p = smb_buf(cli->outbuf);
1436         *p++ = 4;
1437         p += clistr_push(cli, p, path, -1, STR_TERMINATE);
1438
1439         cli_setup_bcc(cli, p);
1440
1441         cli_send_smb(cli);
1442         if (!cli_receive_smb(cli)) {
1443                 return -1;
1444         }
1445
1446         if (cli_is_error(cli)) {
1447                 return -1;
1448         }
1449
1450         /* despite the spec, the result has a -1, followed by
1451            length, followed by name */
1452         p = smb_buf(cli->inbuf);
1453         p += 4;
1454         len = smb_buflen(cli->inbuf) - 4;
1455         if (len <= 0) return -1;
1456
1457         if (tmp_path) {
1458                 pstring path2;
1459                 clistr_pull(cli, path2, p, 
1460                             sizeof(path2), len, STR_ASCII);
1461                 *tmp_path = SMB_STRDUP(path2);
1462         }
1463
1464         return SVAL(cli->inbuf,smb_vwv0);
1465 }
1466
1467
1468 /* 
1469    send a raw ioctl - used by the torture code
1470 */
1471 NTSTATUS cli_raw_ioctl(struct cli_state *cli, int fnum, uint32 code, DATA_BLOB *blob)
1472 {
1473         memset(cli->outbuf,'\0',smb_size);
1474         memset(cli->inbuf,'\0',smb_size);
1475
1476         set_message(cli->outbuf, 3, 0, True);
1477         SCVAL(cli->outbuf,smb_com,SMBioctl);
1478         cli_setup_packet(cli);
1479
1480         SSVAL(cli->outbuf, smb_vwv0, fnum);
1481         SSVAL(cli->outbuf, smb_vwv1, code>>16);
1482         SSVAL(cli->outbuf, smb_vwv2, (code&0xFFFF));
1483
1484         cli_send_smb(cli);
1485         if (!cli_receive_smb(cli)) {
1486                 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
1487         }
1488
1489         if (cli_is_error(cli)) {
1490                 return cli_nt_error(cli);
1491         }
1492
1493         *blob = data_blob(NULL, 0);
1494
1495         return NT_STATUS_OK;
1496 }
1497
1498 /*********************************************************
1499  Set an extended attribute utility fn.
1500 *********************************************************/
1501
1502 static BOOL cli_set_ea(struct cli_state *cli, uint16 setup, char *param, unsigned int param_len,
1503                         const char *ea_name, const char *ea_val, size_t ea_len)
1504 {       
1505         unsigned int data_len = 0;
1506         char *data = NULL;
1507         char *rparam=NULL, *rdata=NULL;
1508         char *p;
1509         size_t ea_namelen = strlen(ea_name);
1510
1511         if (ea_namelen == 0 && ea_len == 0) {
1512                 data_len = 4;
1513                 data = (char *)SMB_MALLOC(data_len);
1514                 if (!data) {
1515                         return False;
1516                 }
1517                 p = data;
1518                 SIVAL(p,0,data_len);
1519         } else {
1520                 data_len = 4 + 4 + ea_namelen + 1 + ea_len;
1521                 data = (char *)SMB_MALLOC(data_len);
1522                 if (!data) {
1523                         return False;
1524                 }
1525                 p = data;
1526                 SIVAL(p,0,data_len);
1527                 p += 4;
1528                 SCVAL(p, 0, 0); /* EA flags. */
1529                 SCVAL(p, 1, ea_namelen);
1530                 SSVAL(p, 2, ea_len);
1531                 memcpy(p+4, ea_name, ea_namelen+1); /* Copy in the name. */
1532                 memcpy(p+4+ea_namelen+1, ea_val, ea_len);
1533         }
1534
1535         if (!cli_send_trans(cli, SMBtrans2,
1536                 NULL,                        /* name */
1537                 -1, 0,                          /* fid, flags */
1538                 &setup, 1, 0,                   /* setup, length, max */
1539                 param, param_len, 2,            /* param, length, max */
1540                 data,  data_len, cli->max_xmit /* data, length, max */
1541                 )) {
1542                         return False;
1543         }
1544
1545         if (!cli_receive_trans(cli, SMBtrans2,
1546                 &rparam, &param_len,
1547                 &rdata, &data_len)) {
1548                         return False;
1549         }
1550
1551         SAFE_FREE(data);
1552         SAFE_FREE(rdata);
1553         SAFE_FREE(rparam);
1554
1555         return True;
1556 }
1557
1558 /*********************************************************
1559  Set an extended attribute on a pathname.
1560 *********************************************************/
1561
1562 BOOL cli_set_ea_path(struct cli_state *cli, const char *path, const char *ea_name, const char *ea_val, size_t ea_len)
1563 {
1564         uint16 setup = TRANSACT2_SETPATHINFO;
1565         unsigned int param_len = 0;
1566         char param[sizeof(pstring)+6];
1567         size_t srclen = 2*(strlen(path)+1);
1568         char *p;
1569
1570         memset(param, 0, sizeof(param));
1571         SSVAL(param,0,SMB_INFO_SET_EA);
1572         p = &param[6];
1573
1574         p += clistr_push(cli, p, path, MIN(srclen, sizeof(param)-6), STR_TERMINATE);
1575         param_len = PTR_DIFF(p, param);
1576
1577         return cli_set_ea(cli, setup, param, param_len, ea_name, ea_val, ea_len);
1578 }
1579
1580 /*********************************************************
1581  Set an extended attribute on an fnum.
1582 *********************************************************/
1583
1584 BOOL cli_set_ea_fnum(struct cli_state *cli, int fnum, const char *ea_name, const char *ea_val, size_t ea_len)
1585 {
1586         char param[6];
1587         uint16 setup = TRANSACT2_SETFILEINFO;
1588
1589         memset(param, 0, 6);
1590         SSVAL(param,0,fnum);
1591         SSVAL(param,2,SMB_INFO_SET_EA);
1592
1593         return cli_set_ea(cli, setup, param, 6, ea_name, ea_val, ea_len);
1594 }
1595
1596 /*********************************************************
1597  Get an extended attribute list tility fn.
1598 *********************************************************/
1599
1600 static BOOL cli_get_ea_list(struct cli_state *cli,
1601                 uint16 setup, char *param, unsigned int param_len,
1602                 TALLOC_CTX *ctx,
1603                 size_t *pnum_eas,
1604                 struct ea_struct **pea_list)
1605 {
1606         unsigned int data_len = 0;
1607         unsigned int rparam_len, rdata_len;
1608         char *rparam=NULL, *rdata=NULL;
1609         char *p;
1610         size_t ea_size;
1611         size_t num_eas;
1612         BOOL ret = False;
1613         struct ea_struct *ea_list;
1614
1615         *pnum_eas = 0;
1616         *pea_list = NULL;
1617
1618         if (!cli_send_trans(cli, SMBtrans2,
1619                         NULL,           /* Name */
1620                         -1, 0,          /* fid, flags */
1621                         &setup, 1, 0,   /* setup, length, max */
1622                         param, param_len, 10, /* param, length, max */
1623                         NULL, data_len, cli->max_xmit /* data, length, max */
1624                                 )) {
1625                 return False;
1626         }
1627
1628         if (!cli_receive_trans(cli, SMBtrans2,
1629                         &rparam, &rparam_len,
1630                         &rdata, &rdata_len)) {
1631                 return False;
1632         }
1633
1634         if (!rdata || rdata_len < 4) {
1635                 goto out;
1636         }
1637
1638         ea_size = (size_t)IVAL(rdata,0);
1639         if (ea_size > rdata_len) {
1640                 goto out;
1641         }
1642
1643         if (ea_size == 0) {
1644                 /* No EA's present. */
1645                 ret = True;
1646                 goto out;
1647         }
1648
1649         p = rdata + 4;
1650         ea_size -= 4;
1651
1652         /* Validate the EA list and count it. */
1653         for (num_eas = 0; ea_size >= 4; num_eas++) {
1654                 unsigned int ea_namelen = CVAL(p,1);
1655                 unsigned int ea_valuelen = SVAL(p,2);
1656                 if (ea_namelen == 0) {
1657                         goto out;
1658                 }
1659                 if (4 + ea_namelen + 1 + ea_valuelen > ea_size) {
1660                         goto out;
1661                 }
1662                 ea_size -= 4 + ea_namelen + 1 + ea_valuelen;
1663                 p += 4 + ea_namelen + 1 + ea_valuelen;
1664         }
1665
1666         if (num_eas == 0) {
1667                 ret = True;
1668                 goto out;
1669         }
1670
1671         *pnum_eas = num_eas;
1672         if (!pea_list) {
1673                 /* Caller only wants number of EA's. */
1674                 ret = True;
1675                 goto out;
1676         }
1677
1678         ea_list = TALLOC_ARRAY(ctx, struct ea_struct, num_eas);
1679         if (!ea_list) {
1680                 goto out;
1681         }
1682
1683         ea_size = (size_t)IVAL(rdata,0);
1684         p = rdata + 4;
1685
1686         for (num_eas = 0; num_eas < *pnum_eas; num_eas++ ) {
1687                 struct ea_struct *ea = &ea_list[num_eas];
1688                 fstring unix_ea_name;
1689                 unsigned int ea_namelen = CVAL(p,1);
1690                 unsigned int ea_valuelen = SVAL(p,2);
1691
1692                 ea->flags = CVAL(p,0);
1693                 unix_ea_name[0] = '\0';
1694                 pull_ascii_fstring(unix_ea_name, p + 4);
1695                 ea->name = talloc_strdup(ctx, unix_ea_name);
1696                 /* Ensure the value is null terminated (in case it's a string). */
1697                 ea->value = data_blob_talloc(ctx, NULL, ea_valuelen + 1);
1698                 if (!ea->value.data) {
1699                         goto out;
1700                 }
1701                 if (ea_valuelen) {
1702                         memcpy(ea->value.data, p+4+ea_namelen+1, ea_valuelen);
1703                 }
1704                 ea->value.data[ea_valuelen] = 0;
1705                 ea->value.length--;
1706                 p += 4 + ea_namelen + 1 + ea_valuelen;
1707         }
1708
1709         *pea_list = ea_list;
1710         ret = True;
1711
1712  out :
1713
1714         SAFE_FREE(rdata);
1715         SAFE_FREE(rparam);
1716         return ret;
1717 }
1718
1719 /*********************************************************
1720  Get an extended attribute list from a pathname.
1721 *********************************************************/
1722
1723 BOOL cli_get_ea_list_path(struct cli_state *cli, const char *path,
1724                 TALLOC_CTX *ctx,
1725                 size_t *pnum_eas,
1726                 struct ea_struct **pea_list)
1727 {
1728         uint16 setup = TRANSACT2_QPATHINFO;
1729         unsigned int param_len = 0;
1730         char param[sizeof(pstring)+6];
1731         char *p;
1732
1733         p = param;
1734         memset(p, 0, 6);
1735         SSVAL(p, 0, SMB_INFO_QUERY_ALL_EAS);
1736         p += 6;
1737         p += clistr_push(cli, p, path, sizeof(pstring)-6, STR_TERMINATE);
1738         param_len = PTR_DIFF(p, param);
1739
1740         return cli_get_ea_list(cli, setup, param, param_len, ctx, pnum_eas, pea_list);
1741 }
1742
1743 /*********************************************************
1744  Get an extended attribute list from an fnum.
1745 *********************************************************/
1746
1747 BOOL cli_get_ea_list_fnum(struct cli_state *cli, int fnum,
1748                 TALLOC_CTX *ctx,
1749                 size_t *pnum_eas,
1750                 struct ea_struct **pea_list)
1751 {
1752         uint16 setup = TRANSACT2_QFILEINFO;
1753         char param[6];
1754
1755         memset(param, 0, 6);
1756         SSVAL(param,0,fnum);
1757         SSVAL(param,2,SMB_INFO_SET_EA);
1758
1759         return cli_get_ea_list(cli, setup, param, 6, ctx, pnum_eas, pea_list);
1760 }