Move lp_*() calls a bit higher up the calls tack.
[kai/samba.git] / source4 / client / client.c
1 /* 
2    Unix SMB/CIFS implementation.
3    SMB client
4    Copyright (C) Andrew Tridgell 1994-1998
5    Copyright (C) Simo Sorce 2001-2002
6    Copyright (C) Jelmer Vernooij 2003-2004
7    Copyright (C) James J Myers   2003 <myersjj@samba.org>
8    
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 3 of the License, or
12    (at your option) any later version.
13    
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18    
19    You should have received a copy of the GNU General Public License
20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 */
22
23 /* 
24  * TODO: remove this ... and don't use talloc_append_string()
25  *
26  * NOTE: I'm not changing the code yet, because I assume there're
27  *       some bugs in the existing code and I'm not sure how to fix
28  *       them correctly.
29  */
30 #define TALLOC_DEPRECATED 1
31
32 #include "includes.h"
33 #include "version.h"
34 #include "libcli/libcli.h"
35 #include "lib/events/events.h"
36 #include "lib/cmdline/popt_common.h"
37 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
38 #include "librpc/gen_ndr/ndr_lsa.h"
39 #include "librpc/gen_ndr/ndr_security.h"
40 #include "libcli/raw/libcliraw.h"
41 #include "libcli/util/clilsa.h"
42 #include "system/dir.h"
43 #include "system/filesys.h"
44 #include "../lib/util/dlinklist.h"
45 #include "system/readline.h"
46 #include "auth/credentials/credentials.h"
47 #include "auth/gensec/gensec.h"
48 #include "system/time.h" /* needed by some systems for asctime() */
49 #include "libcli/resolve/resolve.h"
50 #include "libcli/security/security.h"
51 #include "lib/smbreadline/smbreadline.h"
52 #include "librpc/gen_ndr/ndr_nbt.h"
53 #include "param/param.h"
54 #include "librpc/rpc/dcerpc.h"
55 #include "libcli/raw/raw_proto.h"
56
57 /* the default pager to use for the client "more" command. Users can
58  *    override this with the PAGER environment variable */
59 #ifndef DEFAULT_PAGER
60 #define DEFAULT_PAGER "more"
61 #endif
62
63 struct smbclient_context {
64         char *remote_cur_dir;
65         struct smbcli_state *cli;
66         char *fileselection;
67         time_t newer_than;
68         bool prompt;
69         bool recurse;
70         int archive_level;
71         bool lowercase;
72         int printmode;
73         bool translation;
74         int io_bufsize;
75 };
76
77 /* timing globals */
78 static uint64_t get_total_size = 0;
79 static uint_t get_total_time_ms = 0;
80 static uint64_t put_total_size = 0;
81 static uint_t put_total_time_ms = 0;
82
83 /* Unfortunately, there is no way to pass the a context to the completion function as an argument */
84 static struct smbclient_context *rl_ctx; 
85
86 /* totals globals */
87 static double dir_total;
88
89 /*******************************************************************
90  Reduce a file name, removing .. elements.
91 ********************************************************************/
92 static void dos_clean_name(char *s)
93 {
94         char *p=NULL,*r;
95
96         DEBUG(3,("dos_clean_name [%s]\n",s));
97
98         /* remove any double slashes */
99         all_string_sub(s, "\\\\", "\\", 0);
100
101         while ((p = strstr(s,"\\..\\")) != NULL) {
102                 *p = '\0';
103                 if ((r = strrchr(s,'\\')) != NULL)
104                         memmove(r,p+3,strlen(p+3)+1);
105         }
106
107         trim_string(s,NULL,"\\..");
108
109         all_string_sub(s, "\\.\\", "\\", 0);
110 }
111
112 /****************************************************************************
113 write to a local file with CR/LF->LF translation if appropriate. return the 
114 number taken from the buffer. This may not equal the number written.
115 ****************************************************************************/
116 static int writefile(int f, const void *_b, int n, bool translation)
117 {
118         const uint8_t *b = (const uint8_t *)_b;
119         int i;
120
121         if (!translation) {
122                 return write(f,b,n);
123         }
124
125         i = 0;
126         while (i < n) {
127                 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
128                         b++;i++;
129                 }
130                 if (write(f, b, 1) != 1) {
131                         break;
132                 }
133                 b++;
134                 i++;
135         }
136   
137         return(i);
138 }
139
140 /****************************************************************************
141   read from a file with LF->CR/LF translation if appropriate. return the 
142   number read. read approx n bytes.
143 ****************************************************************************/
144 static int readfile(void *_b, int n, XFILE *f, bool translation)
145 {
146         uint8_t *b = (uint8_t *)_b;
147         int i;
148         int c;
149
150         if (!translation)
151                 return x_fread(b,1,n,f);
152   
153         i = 0;
154         while (i < (n - 1)) {
155                 if ((c = x_getc(f)) == EOF) {
156                         break;
157                 }
158       
159                 if (c == '\n') { /* change all LFs to CR/LF */
160                         b[i++] = '\r';
161                 }
162       
163                 b[i++] = c;
164         }
165   
166         return(i);
167 }
168  
169
170 /****************************************************************************
171 send a message
172 ****************************************************************************/
173 static void send_message(struct smbcli_state *cli, const char *desthost)
174 {
175         char msg[1600];
176         int total_len = 0;
177         int grp_id;
178
179         if (!smbcli_message_start(cli->tree, desthost, cli_credentials_get_username(cmdline_credentials), &grp_id)) {
180                 d_printf("message start: %s\n", smbcli_errstr(cli->tree));
181                 return;
182         }
183
184
185         d_printf("Connected. Type your message, ending it with a Control-D\n");
186
187         while (!feof(stdin) && total_len < 1600) {
188                 int maxlen = MIN(1600 - total_len,127);
189                 int l=0;
190                 int c;
191
192                 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
193                         if (c == '\n')
194                                 msg[l++] = '\r';
195                         msg[l] = c;   
196                 }
197
198                 if (!smbcli_message_text(cli->tree, msg, l, grp_id)) {
199                         d_printf("SMBsendtxt failed (%s)\n",smbcli_errstr(cli->tree));
200                         return;
201                 }      
202                 
203                 total_len += l;
204         }
205
206         if (total_len >= 1600)
207                 d_printf("the message was truncated to 1600 bytes\n");
208         else
209                 d_printf("sent %d bytes\n",total_len);
210
211         if (!smbcli_message_end(cli->tree, grp_id)) {
212                 d_printf("SMBsendend failed (%s)\n",smbcli_errstr(cli->tree));
213                 return;
214         }      
215 }
216
217
218
219 /****************************************************************************
220 check the space on a device
221 ****************************************************************************/
222 static int do_dskattr(struct smbclient_context *ctx)
223 {
224         uint32_t bsize;
225         uint64_t total, avail;
226
227         if (NT_STATUS_IS_ERR(smbcli_dskattr(ctx->cli->tree, &bsize, &total, &avail))) {
228                 d_printf("Error in dskattr: %s\n",smbcli_errstr(ctx->cli->tree)); 
229                 return 1;
230         }
231
232         d_printf("\n\t\t%llu blocks of size %u. %llu blocks available\n",
233                  (unsigned long long)total, 
234                  (unsigned)bsize, 
235                  (unsigned long long)avail);
236
237         return 0;
238 }
239
240 /****************************************************************************
241 show cd/pwd
242 ****************************************************************************/
243 static int cmd_pwd(struct smbclient_context *ctx, const char **args)
244 {
245         d_printf("Current directory is %s\n", ctx->remote_cur_dir);
246         return 0;
247 }
248
249 /*
250   convert a string to dos format
251 */
252 static void dos_format(char *s)
253 {
254         string_replace(s, '/', '\\');
255 }
256
257 /****************************************************************************
258 change directory - inner section
259 ****************************************************************************/
260 static int do_cd(struct smbclient_context *ctx, const char *newdir)
261 {
262         char *dname;
263       
264         /* Save the current directory in case the
265            new directory is invalid */
266         if (newdir[0] == '\\')
267                 dname = talloc_strdup(NULL, newdir);
268         else
269                 dname = talloc_asprintf(NULL, "%s\\%s", ctx->remote_cur_dir, newdir);
270
271         dos_format(dname);
272
273         if (*(dname+strlen(dname)-1) != '\\') {
274                 dname = talloc_append_string(NULL, dname, "\\");
275         }
276         dos_clean_name(dname);
277         
278         if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, dname))) {
279                 d_printf("cd %s: %s\n", dname, smbcli_errstr(ctx->cli->tree));
280                 talloc_free(dname);
281         } else {
282                 ctx->remote_cur_dir = dname;
283         }
284         
285         return 0;
286 }
287
288 /****************************************************************************
289 change directory
290 ****************************************************************************/
291 static int cmd_cd(struct smbclient_context *ctx, const char **args)
292 {
293         int rc = 0;
294
295         if (args[1]) 
296                 rc = do_cd(ctx, args[1]);
297         else
298                 d_printf("Current directory is %s\n",ctx->remote_cur_dir);
299
300         return rc;
301 }
302
303
304 static bool mask_match(struct smbcli_state *c, const char *string, 
305                 const char *pattern, bool is_case_sensitive)
306 {
307         char *p2, *s2;
308         bool ret;
309
310         if (ISDOTDOT(string))
311                 string = ".";
312         if (ISDOT(pattern))
313                 return false;
314         
315         if (is_case_sensitive)
316                 return ms_fnmatch(pattern, string, 
317                                   c->transport->negotiate.protocol) == 0;
318
319         p2 = strlower_talloc(NULL, pattern);
320         s2 = strlower_talloc(NULL, string);
321         ret = ms_fnmatch(p2, s2, c->transport->negotiate.protocol) == 0;
322         talloc_free(p2);
323         talloc_free(s2);
324
325         return ret;
326 }
327
328
329
330 /*******************************************************************
331   decide if a file should be operated on
332   ********************************************************************/
333 static bool do_this_one(struct smbclient_context *ctx, struct clilist_file_info *finfo)
334 {
335         if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY) return(true);
336
337         if (ctx->fileselection && 
338             !mask_match(ctx->cli, finfo->name,ctx->fileselection,false)) {
339                 DEBUG(3,("mask_match %s failed\n", finfo->name));
340                 return false;
341         }
342
343         if (ctx->newer_than && finfo->mtime < ctx->newer_than) {
344                 DEBUG(3,("newer_than %s failed\n", finfo->name));
345                 return(false);
346         }
347
348         if ((ctx->archive_level==1 || ctx->archive_level==2) && !(finfo->attrib & FILE_ATTRIBUTE_ARCHIVE)) {
349                 DEBUG(3,("archive %s failed\n", finfo->name));
350                 return(false);
351         }
352         
353         return(true);
354 }
355
356 /****************************************************************************
357   display info about a file
358   ****************************************************************************/
359 static void display_finfo(struct smbclient_context *ctx, struct clilist_file_info *finfo)
360 {
361         if (do_this_one(ctx, finfo)) {
362                 time_t t = finfo->mtime; /* the time is assumed to be passed as GMT */
363                 char *astr = attrib_string(NULL, finfo->attrib);
364                 d_printf("  %-30s%7.7s %8.0f  %s",
365                          finfo->name,
366                          astr,
367                          (double)finfo->size,
368                          asctime(localtime(&t)));
369                 dir_total += finfo->size;
370                 talloc_free(astr);
371         }
372 }
373
374
375 /****************************************************************************
376    accumulate size of a file
377   ****************************************************************************/
378 static void do_du(struct smbclient_context *ctx, struct clilist_file_info *finfo)
379 {
380         if (do_this_one(ctx, finfo)) {
381                 dir_total += finfo->size;
382         }
383 }
384
385 static bool do_list_recurse;
386 static bool do_list_dirs;
387 static char *do_list_queue = 0;
388 static long do_list_queue_size = 0;
389 static long do_list_queue_start = 0;
390 static long do_list_queue_end = 0;
391 static void (*do_list_fn)(struct smbclient_context *, struct clilist_file_info *);
392
393 /****************************************************************************
394 functions for do_list_queue
395   ****************************************************************************/
396
397 /*
398  * The do_list_queue is a NUL-separated list of strings stored in a
399  * char*.  Since this is a FIFO, we keep track of the beginning and
400  * ending locations of the data in the queue.  When we overflow, we
401  * double the size of the char*.  When the start of the data passes
402  * the midpoint, we move everything back.  This is logically more
403  * complex than a linked list, but easier from a memory management
404  * angle.  In any memory error condition, do_list_queue is reset.
405  * Functions check to ensure that do_list_queue is non-NULL before
406  * accessing it.
407  */
408 static void reset_do_list_queue(void)
409 {
410         SAFE_FREE(do_list_queue);
411         do_list_queue_size = 0;
412         do_list_queue_start = 0;
413         do_list_queue_end = 0;
414 }
415
416 static void init_do_list_queue(void)
417 {
418         reset_do_list_queue();
419         do_list_queue_size = 1024;
420         do_list_queue = malloc_array_p(char, do_list_queue_size);
421         if (do_list_queue == 0) { 
422                 d_printf("malloc fail for size %d\n",
423                          (int)do_list_queue_size);
424                 reset_do_list_queue();
425         } else {
426                 memset(do_list_queue, 0, do_list_queue_size);
427         }
428 }
429
430 static void adjust_do_list_queue(void)
431 {
432         if (do_list_queue == NULL) return;
433
434         /*
435          * If the starting point of the queue is more than half way through,
436          * move everything toward the beginning.
437          */
438         if (do_list_queue_start == do_list_queue_end)
439         {
440                 DEBUG(4,("do_list_queue is empty\n"));
441                 do_list_queue_start = do_list_queue_end = 0;
442                 *do_list_queue = '\0';
443         }
444         else if (do_list_queue_start > (do_list_queue_size / 2))
445         {
446                 DEBUG(4,("sliding do_list_queue backward\n"));
447                 memmove(do_list_queue,
448                         do_list_queue + do_list_queue_start,
449                         do_list_queue_end - do_list_queue_start);
450                 do_list_queue_end -= do_list_queue_start;
451                 do_list_queue_start = 0;
452         }
453            
454 }
455
456 static void add_to_do_list_queue(const char* entry)
457 {
458         char *dlq;
459         long new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
460         while (new_end > do_list_queue_size)
461         {
462                 do_list_queue_size *= 2;
463                 DEBUG(4,("enlarging do_list_queue to %d\n",
464                          (int)do_list_queue_size));
465                 dlq = realloc_p(do_list_queue, char, do_list_queue_size);
466                 if (! dlq) {
467                         d_printf("failure enlarging do_list_queue to %d bytes\n",
468                                  (int)do_list_queue_size);
469                         reset_do_list_queue();
470                 }
471                 else
472                 {
473                         do_list_queue = dlq;
474                         memset(do_list_queue + do_list_queue_size / 2,
475                                0, do_list_queue_size / 2);
476                 }
477         }
478         if (do_list_queue)
479         {
480                 safe_strcpy(do_list_queue + do_list_queue_end, entry, 
481                             do_list_queue_size - do_list_queue_end - 1);
482                 do_list_queue_end = new_end;
483                 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
484                          entry, (int)do_list_queue_start, (int)do_list_queue_end));
485         }
486 }
487
488 static char *do_list_queue_head(void)
489 {
490         return do_list_queue + do_list_queue_start;
491 }
492
493 static void remove_do_list_queue_head(void)
494 {
495         if (do_list_queue_end > do_list_queue_start)
496         {
497                 do_list_queue_start += strlen(do_list_queue_head()) + 1;
498                 adjust_do_list_queue();
499                 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
500                          (int)do_list_queue_start, (int)do_list_queue_end));
501         }
502 }
503
504 static int do_list_queue_empty(void)
505 {
506         return (! (do_list_queue && *do_list_queue));
507 }
508
509 /****************************************************************************
510 a helper for do_list
511   ****************************************************************************/
512 static void do_list_helper(struct clilist_file_info *f, const char *mask, void *state)
513 {
514         struct smbclient_context *ctx = (struct smbclient_context *)state;
515
516         if (f->attrib & FILE_ATTRIBUTE_DIRECTORY) {
517                 if (do_list_dirs && do_this_one(ctx, f)) {
518                         do_list_fn(ctx, f);
519                 }
520                 if (do_list_recurse && 
521                     !ISDOT(f->name) &&
522                     !ISDOTDOT(f->name)) {
523                         char *mask2;
524                         char *p;
525
526                         mask2 = talloc_strdup(NULL, mask);
527                         p = strrchr_m(mask2,'\\');
528                         if (!p) return;
529                         p[1] = 0;
530                         mask2 = talloc_asprintf_append_buffer(mask2, "%s\\*", f->name);
531                         add_to_do_list_queue(mask2);
532                 }
533                 return;
534         }
535
536         if (do_this_one(ctx, f)) {
537                 do_list_fn(ctx, f);
538         }
539 }
540
541
542 /****************************************************************************
543 a wrapper around smbcli_list that adds recursion
544   ****************************************************************************/
545 static void do_list(struct smbclient_context *ctx, const char *mask,uint16_t attribute,
546              void (*fn)(struct smbclient_context *, struct clilist_file_info *),bool rec, bool dirs)
547 {
548         static int in_do_list = 0;
549
550         if (in_do_list && rec)
551         {
552                 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
553                 exit(1);
554         }
555
556         in_do_list = 1;
557
558         do_list_recurse = rec;
559         do_list_dirs = dirs;
560         do_list_fn = fn;
561
562         if (rec)
563         {
564                 init_do_list_queue();
565                 add_to_do_list_queue(mask);
566                 
567                 while (! do_list_queue_empty())
568                 {
569                         /*
570                          * Need to copy head so that it doesn't become
571                          * invalid inside the call to smbcli_list.  This
572                          * would happen if the list were expanded
573                          * during the call.
574                          * Fix from E. Jay Berkenbilt (ejb@ql.org)
575                          */
576                         char *head;
577                         head = do_list_queue_head();
578                         smbcli_list(ctx->cli->tree, head, attribute, do_list_helper, ctx);
579                         remove_do_list_queue_head();
580                         if ((! do_list_queue_empty()) && (fn == display_finfo))
581                         {
582                                 char* next_file = do_list_queue_head();
583                                 char* save_ch = 0;
584                                 if ((strlen(next_file) >= 2) &&
585                                     (next_file[strlen(next_file) - 1] == '*') &&
586                                     (next_file[strlen(next_file) - 2] == '\\'))
587                                 {
588                                         save_ch = next_file +
589                                                 strlen(next_file) - 2;
590                                         *save_ch = '\0';
591                                 }
592                                 d_printf("\n%s\n",next_file);
593                                 if (save_ch)
594                                 {
595                                         *save_ch = '\\';
596                                 }
597                         }
598                 }
599         }
600         else
601         {
602                 if (smbcli_list(ctx->cli->tree, mask, attribute, do_list_helper, ctx) == -1)
603                 {
604                         d_printf("%s listing %s\n", smbcli_errstr(ctx->cli->tree), mask);
605                 }
606         }
607
608         in_do_list = 0;
609         reset_do_list_queue();
610 }
611
612 /****************************************************************************
613   get a directory listing
614   ****************************************************************************/
615 static int cmd_dir(struct smbclient_context *ctx, const char **args)
616 {
617         uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
618         char *mask;
619         int rc;
620         
621         dir_total = 0;
622         
623         mask = talloc_strdup(ctx, ctx->remote_cur_dir);
624         if(mask[strlen(mask)-1]!='\\')
625                 mask = talloc_append_string(ctx, mask,"\\");
626         
627         if (args[1]) {
628                 mask = talloc_strdup(ctx, args[1]);
629                 if (mask[0] != '\\')
630                         mask = talloc_append_string(ctx, mask, "\\");
631                 dos_format(mask);
632         }
633         else {
634                 if (ctx->cli->tree->session->transport->negotiate.protocol <= 
635                     PROTOCOL_LANMAN1) { 
636                         mask = talloc_append_string(ctx, mask, "*.*");
637                 } else {
638                         mask = talloc_append_string(ctx, mask, "*");
639                 }
640         }
641
642         do_list(ctx, mask, attribute, display_finfo, ctx->recurse, true);
643
644         rc = do_dskattr(ctx);
645
646         DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
647
648         return rc;
649 }
650
651
652 /****************************************************************************
653   get a directory listing
654   ****************************************************************************/
655 static int cmd_du(struct smbclient_context *ctx, const char **args)
656 {
657         uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
658         int rc;
659         char *mask;
660         
661         dir_total = 0;
662         
663         if (args[1]) {
664                 if (args[1][0] == '\\')
665                         mask = talloc_strdup(ctx, args[1]);
666                 else
667                         mask = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
668                 dos_format(mask);
669         } else {
670                 mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
671         }
672
673         do_list(ctx, mask, attribute, do_du, ctx->recurse, true);
674
675         talloc_free(mask);
676
677         rc = do_dskattr(ctx);
678
679         d_printf("Total number of bytes: %.0f\n", dir_total);
680
681         return rc;
682 }
683
684
685 /****************************************************************************
686   get a file from rname to lname
687   ****************************************************************************/
688 static int do_get(struct smbclient_context *ctx, char *rname, const char *lname, bool reget)
689 {  
690         int handle = 0, fnum;
691         bool newhandle = false;
692         uint8_t *data;
693         struct timeval tp_start;
694         int read_size = ctx->io_bufsize;
695         uint16_t attr;
696         size_t size;
697         off_t start = 0;
698         off_t nread = 0;
699         int rc = 0;
700
701         GetTimeOfDay(&tp_start);
702
703         if (ctx->lowercase) {
704                 strlower(discard_const_p(char, lname));
705         }
706
707         fnum = smbcli_open(ctx->cli->tree, rname, O_RDONLY, DENY_NONE);
708
709         if (fnum == -1) {
710                 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
711                 return 1;
712         }
713
714         if(!strcmp(lname,"-")) {
715                 handle = fileno(stdout);
716         } else {
717                 if (reget) {
718                         handle = open(lname, O_WRONLY|O_CREAT, 0644);
719                         if (handle >= 0) {
720                                 start = lseek(handle, 0, SEEK_END);
721                                 if (start == -1) {
722                                         d_printf("Error seeking local file\n");
723                                         return 1;
724                                 }
725                         }
726                 } else {
727                         handle = open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
728                 }
729                 newhandle = true;
730         }
731         if (handle < 0) {
732                 d_printf("Error opening local file %s\n",lname);
733                 return 1;
734         }
735
736
737         if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum, 
738                            &attr, &size, NULL, NULL, NULL, NULL, NULL)) &&
739             NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum, 
740                           &attr, &size, NULL, NULL, NULL))) {
741                 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
742                 return 1;
743         }
744
745         DEBUG(2,("getting file %s of size %.0f as %s ", 
746                  rname, (double)size, lname));
747
748         if(!(data = (uint8_t *)malloc(read_size))) { 
749                 d_printf("malloc fail for size %d\n", read_size);
750                 smbcli_close(ctx->cli->tree, fnum);
751                 return 1;
752         }
753
754         while (1) {
755                 int n = smbcli_read(ctx->cli->tree, fnum, data, nread + start, read_size);
756
757                 if (n <= 0) break;
758  
759                 if (writefile(handle,data, n, ctx->translation) != n) {
760                         d_printf("Error writing local file\n");
761                         rc = 1;
762                         break;
763                 }
764       
765                 nread += n;
766         }
767
768         if (nread + start < size) {
769                 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
770                             rname, (long)nread));
771
772                 rc = 1;
773         }
774
775         SAFE_FREE(data);
776         
777         if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
778                 d_printf("Error %s closing remote file\n",smbcli_errstr(ctx->cli->tree));
779                 rc = 1;
780         }
781
782         if (newhandle) {
783                 close(handle);
784         }
785
786         if (ctx->archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) {
787                 smbcli_setatr(ctx->cli->tree, rname, attr & ~(uint16_t)FILE_ATTRIBUTE_ARCHIVE, 0);
788         }
789
790         {
791                 struct timeval tp_end;
792                 int this_time;
793                 
794                 GetTimeOfDay(&tp_end);
795                 this_time = 
796                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
797                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
798                 get_total_time_ms += this_time;
799                 get_total_size += nread;
800                 
801                 DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n",
802                          nread / (1.024*this_time + 1.0e-4),
803                          get_total_size / (1.024*get_total_time_ms)));
804         }
805         
806         return rc;
807 }
808
809
810 /****************************************************************************
811   get a file
812   ****************************************************************************/
813 static int cmd_get(struct smbclient_context *ctx, const char **args)
814 {
815         const char *lname;
816         char *rname;
817
818         if (!args[1]) {
819                 d_printf("get <filename>\n");
820                 return 1;
821         }
822
823         rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
824
825         if (args[2]) 
826                 lname = args[2];
827         else 
828                 lname = args[1];
829         
830         dos_clean_name(rname);
831         
832         return do_get(ctx, rname, lname, false);
833 }
834
835 /****************************************************************************
836  Put up a yes/no prompt.
837 ****************************************************************************/
838 static bool yesno(char *p)
839 {
840         char ans[4];
841         printf("%s",p);
842
843         if (!fgets(ans,sizeof(ans)-1,stdin))
844                 return(false);
845
846         if (*ans == 'y' || *ans == 'Y')
847                 return(true);
848
849         return(false);
850 }
851
852 /****************************************************************************
853   do a mget operation on one file
854   ****************************************************************************/
855 static void do_mget(struct smbclient_context *ctx, struct clilist_file_info *finfo)
856 {
857         char *rname;
858         char *quest;
859         char *mget_mask;
860         char *saved_curdir;
861
862         if (ISDOT(finfo->name) || ISDOTDOT(finfo->name))
863                 return;
864
865         if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)
866                 asprintf(&quest, "Get directory %s? ",finfo->name);
867         else
868                 asprintf(&quest, "Get file %s? ",finfo->name);
869
870         if (ctx->prompt && !yesno(quest)) return;
871
872         SAFE_FREE(quest);
873
874         if (!(finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)) {
875                 asprintf(&rname, "%s%s",ctx->remote_cur_dir,finfo->name);
876                 do_get(ctx, rname, finfo->name, false);
877                 SAFE_FREE(rname);
878                 return;
879         }
880
881         /* handle directories */
882         saved_curdir = talloc_strdup(NULL, ctx->remote_cur_dir);
883
884         ctx->remote_cur_dir = talloc_asprintf_append_buffer(NULL, "%s\\", finfo->name);
885
886         string_replace(discard_const_p(char, finfo->name), '\\', '/');
887         if (ctx->lowercase) {
888                 strlower(discard_const_p(char, finfo->name));
889         }
890         
891         if (!directory_exist(finfo->name) && 
892             mkdir(finfo->name,0777) != 0) {
893                 d_printf("failed to create directory %s\n",finfo->name);
894                 return;
895         }
896         
897         if (chdir(finfo->name) != 0) {
898                 d_printf("failed to chdir to directory %s\n",finfo->name);
899                 return;
900         }
901
902         mget_mask = talloc_asprintf(NULL, "%s*", ctx->remote_cur_dir);
903         
904         do_list(ctx, mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,false, true);
905         chdir("..");
906         talloc_free(ctx->remote_cur_dir);
907
908         ctx->remote_cur_dir = saved_curdir;
909 }
910
911
912 /****************************************************************************
913 view the file using the pager
914 ****************************************************************************/
915 static int cmd_more(struct smbclient_context *ctx, const char **args)
916 {
917         char *rname;
918         char *pager_cmd;
919         char *lname;
920         char *pager;
921         int fd;
922         int rc = 0;
923
924         lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir());
925         fd = mkstemp(lname);
926         if (fd == -1) {
927                 d_printf("failed to create temporary file for more\n");
928                 return 1;
929         }
930         close(fd);
931
932         if (!args[1]) {
933                 d_printf("more <filename>\n");
934                 unlink(lname);
935                 return 1;
936         }
937         rname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
938         dos_clean_name(rname);
939
940         rc = do_get(ctx, rname, lname, false);
941
942         pager=getenv("PAGER");
943
944         pager_cmd = talloc_asprintf(ctx, "%s %s",(pager? pager:DEFAULT_PAGER), lname);
945         system(pager_cmd);
946         unlink(lname);
947         
948         return rc;
949 }
950
951
952
953 /****************************************************************************
954 do a mget command
955 ****************************************************************************/
956 static int cmd_mget(struct smbclient_context *ctx, const char **args)
957 {
958         uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
959         char *mget_mask = NULL;
960         int i;
961
962         if (ctx->recurse)
963                 attribute |= FILE_ATTRIBUTE_DIRECTORY;
964         
965         for (i = 1; args[i]; i++) {
966                 mget_mask = talloc_strdup(ctx,ctx->remote_cur_dir);
967                 if(mget_mask[strlen(mget_mask)-1]!='\\')
968                         mget_mask = talloc_append_string(ctx, mget_mask, "\\");
969                 
970                 mget_mask = talloc_strdup(ctx, args[i]);
971                 if (mget_mask[0] != '\\')
972                         mget_mask = talloc_append_string(ctx, mget_mask, "\\");
973                 do_list(ctx, mget_mask, attribute,do_mget,false,true);
974
975                 talloc_free(mget_mask);
976         }
977
978         if (mget_mask == NULL) {
979                 mget_mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
980                 do_list(ctx, mget_mask, attribute,do_mget,false,true);
981                 talloc_free(mget_mask);
982         }
983         
984         return 0;
985 }
986
987
988 /****************************************************************************
989 make a directory of name "name"
990 ****************************************************************************/
991 static NTSTATUS do_mkdir(struct smbclient_context *ctx, char *name)
992 {
993         NTSTATUS status;
994
995         if (NT_STATUS_IS_ERR(status = smbcli_mkdir(ctx->cli->tree, name))) {
996                 d_printf("%s making remote directory %s\n",
997                          smbcli_errstr(ctx->cli->tree),name);
998                 return status;
999         }
1000
1001         return status;
1002 }
1003
1004
1005 /****************************************************************************
1006  Exit client.
1007 ****************************************************************************/
1008 static int cmd_quit(struct smbclient_context *ctx, const char **args)
1009 {
1010         talloc_free(ctx);
1011         exit(0);
1012         /* NOTREACHED */
1013         return 0;
1014 }
1015
1016
1017 /****************************************************************************
1018   make a directory
1019   ****************************************************************************/
1020 static int cmd_mkdir(struct smbclient_context *ctx, const char **args)
1021 {
1022         char *mask, *p;
1023   
1024         if (!args[1]) {
1025                 if (!ctx->recurse)
1026                         d_printf("mkdir <dirname>\n");
1027                 return 1;
1028         }
1029
1030         mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir,args[1]);
1031
1032         if (ctx->recurse) {
1033                 dos_clean_name(mask);
1034
1035                 trim_string(mask,".",NULL);
1036                 for (p = strtok(mask,"/\\"); p; p = strtok(p, "/\\")) {
1037                         char *parent = talloc_strndup(ctx, mask, PTR_DIFF(p, mask));
1038                         
1039                         if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, parent))) { 
1040                                 do_mkdir(ctx, parent);
1041                         }
1042
1043                         talloc_free(parent);
1044                 }        
1045         } else {
1046                 do_mkdir(ctx, mask);
1047         }
1048         
1049         return 0;
1050 }
1051
1052 /****************************************************************************
1053 show 8.3 name of a file
1054 ****************************************************************************/
1055 static int cmd_altname(struct smbclient_context *ctx, const char **args)
1056 {
1057         const char *altname;
1058         char *name;
1059   
1060         if (!args[1]) {
1061                 d_printf("altname <file>\n");
1062                 return 1;
1063         }
1064
1065         name = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1066
1067         if (!NT_STATUS_IS_OK(smbcli_qpathinfo_alt_name(ctx->cli->tree, name, &altname))) {
1068                 d_printf("%s getting alt name for %s\n",
1069                          smbcli_errstr(ctx->cli->tree),name);
1070                 return(false);
1071         }
1072         d_printf("%s\n", altname);
1073
1074         return 0;
1075 }
1076
1077
1078 /****************************************************************************
1079   put a single file
1080   ****************************************************************************/
1081 static int do_put(struct smbclient_context *ctx, char *rname, char *lname, bool reput)
1082 {
1083         int fnum;
1084         XFILE *f;
1085         size_t start = 0;
1086         off_t nread = 0;
1087         uint8_t *buf = NULL;
1088         int maxwrite = ctx->io_bufsize;
1089         int rc = 0;
1090         
1091         struct timeval tp_start;
1092         GetTimeOfDay(&tp_start);
1093
1094         if (reput) {
1095                 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT, DENY_NONE);
1096                 if (fnum >= 0) {
1097                         if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL)) &&
1098                             NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL))) {
1099                                 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
1100                                 return 1;
1101                         }
1102                 }
1103         } else {
1104                 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT|O_TRUNC, 
1105                                 DENY_NONE);
1106         }
1107   
1108         if (fnum == -1) {
1109                 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1110                 return 1;
1111         }
1112
1113         /* allow files to be piped into smbclient
1114            jdblair 24.jun.98
1115
1116            Note that in this case this function will exit(0) rather
1117            than returning. */
1118         if (!strcmp(lname, "-")) {
1119                 f = x_stdin;
1120                 /* size of file is not known */
1121         } else {
1122                 f = x_fopen(lname,O_RDONLY, 0);
1123                 if (f && reput) {
1124                         if (x_tseek(f, start, SEEK_SET) == -1) {
1125                                 d_printf("Error seeking local file\n");
1126                                 return 1;
1127                         }
1128                 }
1129         }
1130
1131         if (!f) {
1132                 d_printf("Error opening local file %s\n",lname);
1133                 return 1;
1134         }
1135
1136   
1137         DEBUG(1,("putting file %s as %s ",lname,
1138                  rname));
1139   
1140         buf = (uint8_t *)malloc(maxwrite);
1141         if (!buf) {
1142                 d_printf("ERROR: Not enough memory!\n");
1143                 return 1;
1144         }
1145         while (!x_feof(f)) {
1146                 int n = maxwrite;
1147                 int ret;
1148
1149                 if ((n = readfile(buf,n,f,ctx->translation)) < 1) {
1150                         if((n == 0) && x_feof(f))
1151                                 break; /* Empty local file. */
1152
1153                         d_printf("Error reading local file: %s\n", strerror(errno));
1154                         rc = 1;
1155                         break;
1156                 }
1157
1158                 ret = smbcli_write(ctx->cli->tree, fnum, 0, buf, nread + start, n);
1159
1160                 if (n != ret) {
1161                         d_printf("Error writing file: %s\n", smbcli_errstr(ctx->cli->tree));
1162                         rc = 1;
1163                         break;
1164                 } 
1165
1166                 nread += n;
1167         }
1168
1169         if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
1170                 d_printf("%s closing remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1171                 x_fclose(f);
1172                 SAFE_FREE(buf);
1173                 return 1;
1174         }
1175
1176         
1177         if (f != x_stdin) {
1178                 x_fclose(f);
1179         }
1180
1181         SAFE_FREE(buf);
1182
1183         {
1184                 struct timeval tp_end;
1185                 int this_time;
1186                 
1187                 GetTimeOfDay(&tp_end);
1188                 this_time = 
1189                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1190                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
1191                 put_total_time_ms += this_time;
1192                 put_total_size += nread;
1193                 
1194                 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1195                          nread / (1.024*this_time + 1.0e-4),
1196                          put_total_size / (1.024*put_total_time_ms)));
1197         }
1198
1199         if (f == x_stdin) {
1200                 talloc_free(ctx);
1201                 exit(0);
1202         }
1203         
1204         return rc;
1205 }
1206
1207  
1208
1209 /****************************************************************************
1210   put a file
1211   ****************************************************************************/
1212 static int cmd_put(struct smbclient_context *ctx, const char **args)
1213 {
1214         char *lname;
1215         char *rname;
1216         
1217         if (!args[1]) {
1218                 d_printf("put <filename> [<remotename>]\n");
1219                 return 1;
1220         }
1221
1222         lname = talloc_strdup(ctx, args[1]);
1223   
1224         if (args[2])
1225                 rname = talloc_strdup(ctx, args[2]);
1226         else
1227                 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, lname);
1228         
1229         dos_clean_name(rname);
1230
1231         /* allow '-' to represent stdin
1232            jdblair, 24.jun.98 */
1233         if (!file_exist(lname) && (strcmp(lname,"-"))) {
1234                 d_printf("%s does not exist\n",lname);
1235                 return 1;
1236         }
1237
1238         return do_put(ctx, rname, lname, false);
1239 }
1240
1241 /*************************************
1242   File list structure
1243 *************************************/
1244
1245 static struct file_list {
1246         struct file_list *prev, *next;
1247         char *file_path;
1248         bool isdir;
1249 } *file_list;
1250
1251 /****************************************************************************
1252   Free a file_list structure
1253 ****************************************************************************/
1254
1255 static void free_file_list (struct file_list * list)
1256 {
1257         struct file_list *tmp;
1258         
1259         while (list)
1260         {
1261                 tmp = list;
1262                 DLIST_REMOVE(list, list);
1263                 SAFE_FREE(tmp->file_path);
1264                 SAFE_FREE(tmp);
1265         }
1266 }
1267
1268 /****************************************************************************
1269   seek in a directory/file list until you get something that doesn't start with
1270   the specified name
1271   ****************************************************************************/
1272 static bool seek_list(struct file_list *list, char *name)
1273 {
1274         while (list) {
1275                 trim_string(list->file_path,"./","\n");
1276                 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1277                         return(true);
1278                 }
1279                 list = list->next;
1280         }
1281       
1282         return(false);
1283 }
1284
1285 /****************************************************************************
1286   set the file selection mask
1287   ****************************************************************************/
1288 static int cmd_select(struct smbclient_context *ctx, const char **args)
1289 {
1290         talloc_free(ctx->fileselection);
1291         ctx->fileselection = talloc_strdup(NULL, args[1]);
1292
1293         return 0;
1294 }
1295
1296 /*******************************************************************
1297   A readdir wrapper which just returns the file name.
1298  ********************************************************************/
1299 static const char *readdirname(DIR *p)
1300 {
1301         struct dirent *ptr;
1302         char *dname;
1303
1304         if (!p)
1305                 return(NULL);
1306   
1307         ptr = (struct dirent *)readdir(p);
1308         if (!ptr)
1309                 return(NULL);
1310
1311         dname = ptr->d_name;
1312
1313 #ifdef NEXT2
1314         if (telldir(p) < 0)
1315                 return(NULL);
1316 #endif
1317
1318 #ifdef HAVE_BROKEN_READDIR
1319         /* using /usr/ucb/cc is BAD */
1320         dname = dname - 2;
1321 #endif
1322
1323         {
1324                 static char *buf;
1325                 int len = NAMLEN(ptr);
1326                 buf = talloc_strndup(NULL, dname, len);
1327                 dname = buf;
1328         }
1329
1330         return(dname);
1331 }
1332
1333 /****************************************************************************
1334   Recursive file matching function act as find
1335   match must be always set to true when calling this function
1336 ****************************************************************************/
1337 static int file_find(struct smbclient_context *ctx, struct file_list **list, const char *directory, 
1338                       const char *expression, bool match)
1339 {
1340         DIR *dir;
1341         struct file_list *entry;
1342         struct stat statbuf;
1343         int ret;
1344         char *path;
1345         bool isdir;
1346         const char *dname;
1347
1348         dir = opendir(directory);
1349         if (!dir) return -1;
1350         
1351         while ((dname = readdirname(dir))) {
1352                 if (ISDOT(dname) || ISDOTDOT(dname)) {
1353                         continue;
1354                 }
1355                 
1356                 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1357                         continue;
1358                 }
1359
1360                 isdir = false;
1361                 if (!match || !gen_fnmatch(expression, dname)) {
1362                         if (ctx->recurse) {
1363                                 ret = stat(path, &statbuf);
1364                                 if (ret == 0) {
1365                                         if (S_ISDIR(statbuf.st_mode)) {
1366                                                 isdir = true;
1367                                                 ret = file_find(ctx, list, path, expression, false);
1368                                         }
1369                                 } else {
1370                                         d_printf("file_find: cannot stat file %s\n", path);
1371                                 }
1372                                 
1373                                 if (ret == -1) {
1374                                         SAFE_FREE(path);
1375                                         closedir(dir);
1376                                         return -1;
1377                                 }
1378                         }
1379                         entry = malloc_p(struct file_list);
1380                         if (!entry) {
1381                                 d_printf("Out of memory in file_find\n");
1382                                 closedir(dir);
1383                                 return -1;
1384                         }
1385                         entry->file_path = path;
1386                         entry->isdir = isdir;
1387                         DLIST_ADD(*list, entry);
1388                 } else {
1389                         SAFE_FREE(path);
1390                 }
1391         }
1392
1393         closedir(dir);
1394         return 0;
1395 }
1396
1397 /****************************************************************************
1398   mput some files
1399   ****************************************************************************/
1400 static int cmd_mput(struct smbclient_context *ctx, const char **args)
1401 {
1402         int i;
1403         
1404         for (i = 1; args[i]; i++) {
1405                 int ret;
1406                 struct file_list *temp_list;
1407                 char *quest, *lname, *rname;
1408
1409                 printf("%s\n", args[i]);
1410         
1411                 file_list = NULL;
1412
1413                 ret = file_find(ctx, &file_list, ".", args[i], true);
1414                 if (ret) {
1415                         free_file_list(file_list);
1416                         continue;
1417                 }
1418                 
1419                 quest = NULL;
1420                 lname = NULL;
1421                 rname = NULL;
1422                                 
1423                 for (temp_list = file_list; temp_list; 
1424                      temp_list = temp_list->next) {
1425
1426                         SAFE_FREE(lname);
1427                         if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1428                                 continue;
1429                         trim_string(lname, "./", "/");
1430                         
1431                         /* check if it's a directory */
1432                         if (temp_list->isdir) {
1433                                 /* if (!recurse) continue; */
1434                                 
1435                                 SAFE_FREE(quest);
1436                                 if (asprintf(&quest, "Put directory %s? ", lname) < 0) break;
1437                                 if (ctx->prompt && !yesno(quest)) { /* No */
1438                                         /* Skip the directory */
1439                                         lname[strlen(lname)-1] = '/';
1440                                         if (!seek_list(temp_list, lname))
1441                                                 break;              
1442                                 } else { /* Yes */
1443                                         SAFE_FREE(rname);
1444                                         if(asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1445                                         dos_format(rname);
1446                                         if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, rname)) && 
1447                                             NT_STATUS_IS_ERR(do_mkdir(ctx, rname))) {
1448                                                 DEBUG (0, ("Unable to make dir, skipping..."));
1449                                                 /* Skip the directory */
1450                                                 lname[strlen(lname)-1] = '/';
1451                                                 if (!seek_list(temp_list, lname))
1452                                                         break;
1453                                         }
1454                                 }
1455                                 continue;
1456                         } else {
1457                                 SAFE_FREE(quest);
1458                                 if (asprintf(&quest,"Put file %s? ", lname) < 0) break;
1459                                 if (ctx->prompt && !yesno(quest)) /* No */
1460                                         continue;
1461                                 
1462                                 /* Yes */
1463                                 SAFE_FREE(rname);
1464                                 if (asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1465                         }
1466
1467                         dos_format(rname);
1468
1469                         do_put(ctx, rname, lname, false);
1470                 }
1471                 free_file_list(file_list);
1472                 SAFE_FREE(quest);
1473                 SAFE_FREE(lname);
1474                 SAFE_FREE(rname);
1475         }
1476
1477         return 0;
1478 }
1479
1480
1481 /****************************************************************************
1482   print a file
1483   ****************************************************************************/
1484 static int cmd_print(struct smbclient_context *ctx, const char **args)
1485 {
1486         char *lname, *rname;
1487         char *p;
1488
1489         if (!args[1]) {
1490                 d_printf("print <filename>\n");
1491                 return 1;
1492         }
1493
1494         lname = talloc_strdup(ctx, args[1]);
1495
1496         rname = talloc_strdup(ctx, lname);
1497         p = strrchr_m(rname,'/');
1498         if (p) {
1499                 slprintf(rname, sizeof(rname)-1, "%s-%d", p+1, (int)getpid());
1500         }
1501
1502         if (strequal(lname,"-")) {
1503                 slprintf(rname, sizeof(rname)-1, "stdin-%d", (int)getpid());
1504         }
1505
1506         return do_put(ctx, rname, lname, false);
1507 }
1508
1509
1510 static int cmd_rewrite(struct smbclient_context *ctx, const char **args)
1511 {
1512         d_printf("REWRITE: command not implemented (FIXME!)\n");
1513         
1514         return 0;
1515 }
1516
1517 /****************************************************************************
1518 delete some files
1519 ****************************************************************************/
1520 static int cmd_del(struct smbclient_context *ctx, const char **args)
1521 {
1522         char *mask;
1523         uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
1524
1525         if (ctx->recurse)
1526                 attribute |= FILE_ATTRIBUTE_DIRECTORY;
1527         
1528         if (!args[1]) {
1529                 d_printf("del <filename>\n");
1530                 return 1;
1531         }
1532         mask = talloc_asprintf(ctx,"%s%s", ctx->remote_cur_dir, args[1]);
1533
1534         if (NT_STATUS_IS_ERR(smbcli_unlink(ctx->cli->tree, mask))) {
1535                 d_printf("%s deleting remote file %s\n",smbcli_errstr(ctx->cli->tree),mask);
1536         }
1537         
1538         return 0;
1539 }
1540
1541
1542 /****************************************************************************
1543 delete a whole directory tree
1544 ****************************************************************************/
1545 static int cmd_deltree(struct smbclient_context *ctx, const char **args)
1546 {
1547         char *dname;
1548         int ret;
1549
1550         if (!args[1]) {
1551                 d_printf("deltree <dirname>\n");
1552                 return 1;
1553         }
1554
1555         dname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1556         
1557         ret = smbcli_deltree(ctx->cli->tree, dname);
1558
1559         if (ret == -1) {
1560                 printf("Failed to delete tree %s - %s\n", dname, smbcli_errstr(ctx->cli->tree));
1561                 return -1;
1562         }
1563
1564         printf("Deleted %d files in %s\n", ret, dname);
1565         
1566         return 0;
1567 }
1568
1569 typedef struct {
1570         const char  *level_name;
1571         enum smb_fsinfo_level level;
1572 } fsinfo_level_t;
1573
1574 fsinfo_level_t fsinfo_levels[] = {
1575         {"dskattr", RAW_QFS_DSKATTR},
1576         {"allocation", RAW_QFS_ALLOCATION},
1577         {"volume", RAW_QFS_VOLUME},
1578         {"volumeinfo", RAW_QFS_VOLUME_INFO},
1579         {"sizeinfo", RAW_QFS_SIZE_INFO},
1580         {"deviceinfo", RAW_QFS_DEVICE_INFO},
1581         {"attributeinfo", RAW_QFS_ATTRIBUTE_INFO},
1582         {"unixinfo", RAW_QFS_UNIX_INFO},
1583         {"volume-information", RAW_QFS_VOLUME_INFORMATION},
1584         {"size-information", RAW_QFS_SIZE_INFORMATION},
1585         {"device-information", RAW_QFS_DEVICE_INFORMATION},
1586         {"attribute-information", RAW_QFS_ATTRIBUTE_INFORMATION},
1587         {"quota-information", RAW_QFS_QUOTA_INFORMATION},
1588         {"fullsize-information", RAW_QFS_FULL_SIZE_INFORMATION},
1589         {"objectid", RAW_QFS_OBJECTID_INFORMATION},
1590         {NULL, RAW_QFS_GENERIC}
1591 };
1592
1593
1594 static int cmd_fsinfo(struct smbclient_context *ctx, const char **args)
1595 {
1596         union smb_fsinfo fsinfo;
1597         NTSTATUS status;
1598         fsinfo_level_t *fsinfo_level;
1599         
1600         if (!args[1]) {
1601                 d_printf("fsinfo <level>, where level is one of following:\n");
1602                 fsinfo_level = fsinfo_levels;
1603                 while(fsinfo_level->level_name) {
1604                         d_printf("%s\n", fsinfo_level->level_name);
1605                         fsinfo_level++;
1606                 }
1607                 return 1;
1608         }
1609         
1610         fsinfo_level = fsinfo_levels;
1611         while(fsinfo_level->level_name && !strequal(args[1],fsinfo_level->level_name)) {
1612                 fsinfo_level++;
1613         }
1614   
1615         if (!fsinfo_level->level_name) {
1616                 d_printf("wrong level name!\n");
1617                 return 1;
1618         }
1619   
1620         fsinfo.generic.level = fsinfo_level->level;
1621         status = smb_raw_fsinfo(ctx->cli->tree, ctx, &fsinfo);
1622         if (!NT_STATUS_IS_OK(status)) {
1623                 d_printf("fsinfo-level-%s - %s\n", fsinfo_level->level_name, nt_errstr(status));
1624                 return 1;
1625         }
1626
1627         d_printf("fsinfo-level-%s:\n", fsinfo_level->level_name);
1628         switch(fsinfo.generic.level) {
1629         case RAW_QFS_DSKATTR:
1630                 d_printf("\tunits_total:                %hu\n", 
1631                          (unsigned short) fsinfo.dskattr.out.units_total);
1632                 d_printf("\tblocks_per_unit:            %hu\n", 
1633                          (unsigned short) fsinfo.dskattr.out.blocks_per_unit);
1634                 d_printf("\tblocks_size:                %hu\n", 
1635                          (unsigned short) fsinfo.dskattr.out.block_size);
1636                 d_printf("\tunits_free:                 %hu\n", 
1637                          (unsigned short) fsinfo.dskattr.out.units_free);
1638                 break;
1639         case RAW_QFS_ALLOCATION:
1640                 d_printf("\tfs_id:                      %lu\n", 
1641                          (unsigned long) fsinfo.allocation.out.fs_id);
1642                 d_printf("\tsectors_per_unit:           %lu\n", 
1643                          (unsigned long) fsinfo.allocation.out.sectors_per_unit);
1644                 d_printf("\ttotal_alloc_units:          %lu\n", 
1645                          (unsigned long) fsinfo.allocation.out.total_alloc_units);
1646                 d_printf("\tavail_alloc_units:          %lu\n", 
1647                          (unsigned long) fsinfo.allocation.out.avail_alloc_units);
1648                 d_printf("\tbytes_per_sector:           %hu\n", 
1649                          (unsigned short) fsinfo.allocation.out.bytes_per_sector);
1650                 break;
1651         case RAW_QFS_VOLUME:
1652                 d_printf("\tserial_number:              %lu\n", 
1653                          (unsigned long) fsinfo.volume.out.serial_number);
1654                 d_printf("\tvolume_name:                %s\n", fsinfo.volume.out.volume_name.s);
1655                 break;
1656         case RAW_QFS_VOLUME_INFO:
1657         case RAW_QFS_VOLUME_INFORMATION:
1658                 d_printf("\tcreate_time:                %s\n",
1659                          nt_time_string(ctx,fsinfo.volume_info.out.create_time));
1660                 d_printf("\tserial_number:              %lu\n", 
1661                          (unsigned long) fsinfo.volume_info.out.serial_number);
1662                 d_printf("\tvolume_name:                %s\n", fsinfo.volume_info.out.volume_name.s);
1663                 break;
1664         case RAW_QFS_SIZE_INFO:
1665         case RAW_QFS_SIZE_INFORMATION:
1666                 d_printf("\ttotal_alloc_units:          %llu\n", 
1667                          (unsigned long long) fsinfo.size_info.out.total_alloc_units);
1668                 d_printf("\tavail_alloc_units:          %llu\n", 
1669                          (unsigned long long) fsinfo.size_info.out.avail_alloc_units);
1670                 d_printf("\tsectors_per_unit:           %lu\n", 
1671                          (unsigned long) fsinfo.size_info.out.sectors_per_unit);
1672                 d_printf("\tbytes_per_sector:           %lu\n", 
1673                          (unsigned long) fsinfo.size_info.out.bytes_per_sector);
1674                 break;
1675         case RAW_QFS_DEVICE_INFO:
1676         case RAW_QFS_DEVICE_INFORMATION:
1677                 d_printf("\tdevice_type:                %lu\n", 
1678                          (unsigned long) fsinfo.device_info.out.device_type);
1679                 d_printf("\tcharacteristics:            0x%lx\n", 
1680                          (unsigned long) fsinfo.device_info.out.characteristics);
1681                 break;
1682         case RAW_QFS_ATTRIBUTE_INFORMATION:
1683         case RAW_QFS_ATTRIBUTE_INFO:
1684                 d_printf("\tfs_attr:                    0x%lx\n", 
1685                          (unsigned long) fsinfo.attribute_info.out.fs_attr);
1686                 d_printf("\tmax_file_component_length:  %lu\n", 
1687                          (unsigned long) fsinfo.attribute_info.out.max_file_component_length);
1688                 d_printf("\tfs_type:                    %s\n", fsinfo.attribute_info.out.fs_type.s);
1689                 break;
1690         case RAW_QFS_UNIX_INFO:
1691                 d_printf("\tmajor_version:              %hu\n", 
1692                          (unsigned short) fsinfo.unix_info.out.major_version);
1693                 d_printf("\tminor_version:              %hu\n", 
1694                          (unsigned short) fsinfo.unix_info.out.minor_version);
1695                 d_printf("\tcapability:                 0x%llx\n", 
1696                          (unsigned long long) fsinfo.unix_info.out.capability);
1697                 break;
1698         case RAW_QFS_QUOTA_INFORMATION:
1699                 d_printf("\tunknown[3]:                 [%llu,%llu,%llu]\n", 
1700                          (unsigned long long) fsinfo.quota_information.out.unknown[0],
1701                          (unsigned long long) fsinfo.quota_information.out.unknown[1],
1702                          (unsigned long long) fsinfo.quota_information.out.unknown[2]);
1703                 d_printf("\tquota_soft:                 %llu\n", 
1704                          (unsigned long long) fsinfo.quota_information.out.quota_soft);
1705                 d_printf("\tquota_hard:                 %llu\n", 
1706                          (unsigned long long) fsinfo.quota_information.out.quota_hard);
1707                 d_printf("\tquota_flags:                0x%llx\n", 
1708                          (unsigned long long) fsinfo.quota_information.out.quota_flags);
1709                 break;
1710         case RAW_QFS_FULL_SIZE_INFORMATION:
1711                 d_printf("\ttotal_alloc_units:          %llu\n", 
1712                          (unsigned long long) fsinfo.full_size_information.out.total_alloc_units);
1713                 d_printf("\tcall_avail_alloc_units:     %llu\n", 
1714                          (unsigned long long) fsinfo.full_size_information.out.call_avail_alloc_units);
1715                 d_printf("\tactual_avail_alloc_units:   %llu\n", 
1716                          (unsigned long long) fsinfo.full_size_information.out.actual_avail_alloc_units);
1717                 d_printf("\tsectors_per_unit:           %lu\n", 
1718                          (unsigned long) fsinfo.full_size_information.out.sectors_per_unit);
1719                 d_printf("\tbytes_per_sector:           %lu\n", 
1720                          (unsigned long) fsinfo.full_size_information.out.bytes_per_sector);
1721                 break;
1722         case RAW_QFS_OBJECTID_INFORMATION:
1723                 d_printf("\tGUID:                       %s\n", 
1724                          GUID_string(ctx,&fsinfo.objectid_information.out.guid));
1725                 d_printf("\tunknown[6]:                 [%llu,%llu,%llu,%llu,%llu,%llu]\n", 
1726                          (unsigned long long) fsinfo.objectid_information.out.unknown[0],
1727                          (unsigned long long) fsinfo.objectid_information.out.unknown[1],
1728                          (unsigned long long) fsinfo.objectid_information.out.unknown[2],
1729                          (unsigned long long) fsinfo.objectid_information.out.unknown[3],
1730                          (unsigned long long) fsinfo.objectid_information.out.unknown[4],
1731                          (unsigned long long) fsinfo.objectid_information.out.unknown[5] );
1732                 break;
1733         case RAW_QFS_GENERIC:
1734                 d_printf("\twrong level returned\n");
1735                 break;
1736         }
1737   
1738         return 0;
1739 }
1740
1741 /****************************************************************************
1742 show as much information as possible about a file
1743 ****************************************************************************/
1744 static int cmd_allinfo(struct smbclient_context *ctx, const char **args)
1745 {
1746         char *fname;
1747         union smb_fileinfo finfo;
1748         NTSTATUS status;
1749         int fnum;
1750
1751         if (!args[1]) {
1752                 d_printf("allinfo <filename>\n");
1753                 return 1;
1754         }
1755         fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1756
1757         /* first a ALL_INFO QPATHINFO */
1758         finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1759         finfo.generic.in.file.path = fname;
1760         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1761         if (!NT_STATUS_IS_OK(status)) {
1762                 d_printf("%s - %s\n", fname, nt_errstr(status));
1763                 return 1;
1764         }
1765
1766         d_printf("\tcreate_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1767         d_printf("\taccess_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.access_time));
1768         d_printf("\twrite_time:     %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1769         d_printf("\tchange_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1770         d_printf("\tattrib:         0x%x\n", finfo.all_info.out.attrib);
1771         d_printf("\talloc_size:     %lu\n", (unsigned long)finfo.all_info.out.alloc_size);
1772         d_printf("\tsize:           %lu\n", (unsigned long)finfo.all_info.out.size);
1773         d_printf("\tnlink:          %u\n", finfo.all_info.out.nlink);
1774         d_printf("\tdelete_pending: %u\n", finfo.all_info.out.delete_pending);
1775         d_printf("\tdirectory:      %u\n", finfo.all_info.out.directory);
1776         d_printf("\tea_size:        %u\n", finfo.all_info.out.ea_size);
1777         d_printf("\tfname:          '%s'\n", finfo.all_info.out.fname.s);
1778
1779         /* 8.3 name if any */
1780         finfo.generic.level = RAW_FILEINFO_ALT_NAME_INFO;
1781         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1782         if (NT_STATUS_IS_OK(status)) {
1783                 d_printf("\talt_name:       %s\n", finfo.alt_name_info.out.fname.s);
1784         }
1785
1786         /* file_id if available */
1787         finfo.generic.level = RAW_FILEINFO_INTERNAL_INFORMATION;
1788         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1789         if (NT_STATUS_IS_OK(status)) {
1790                 d_printf("\tfile_id         %.0f\n", 
1791                          (double)finfo.internal_information.out.file_id);
1792         }
1793
1794         /* the EAs, if any */
1795         finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1796         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1797         if (NT_STATUS_IS_OK(status)) {
1798                 int i;
1799                 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1800                         d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1801                                  finfo.all_eas.out.eas[i].flags,
1802                                  (int)finfo.all_eas.out.eas[i].value.length,
1803                                  finfo.all_eas.out.eas[i].name.s);
1804                 }
1805         }
1806
1807         /* streams, if available */
1808         finfo.generic.level = RAW_FILEINFO_STREAM_INFO;
1809         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1810         if (NT_STATUS_IS_OK(status)) {
1811                 int i;
1812                 for (i=0;i<finfo.stream_info.out.num_streams;i++) {
1813                         d_printf("\tstream %d:\n", i);
1814                         d_printf("\t\tsize       %ld\n", 
1815                                  (long)finfo.stream_info.out.streams[i].size);
1816                         d_printf("\t\talloc size %ld\n", 
1817                                  (long)finfo.stream_info.out.streams[i].alloc_size);
1818                         d_printf("\t\tname       %s\n", finfo.stream_info.out.streams[i].stream_name.s);
1819                 }
1820         }       
1821
1822         /* dev/inode if available */
1823         finfo.generic.level = RAW_FILEINFO_COMPRESSION_INFORMATION;
1824         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1825         if (NT_STATUS_IS_OK(status)) {
1826                 d_printf("\tcompressed size %ld\n", (long)finfo.compression_info.out.compressed_size);
1827                 d_printf("\tformat          %ld\n", (long)finfo.compression_info.out.format);
1828                 d_printf("\tunit_shift      %ld\n", (long)finfo.compression_info.out.unit_shift);
1829                 d_printf("\tchunk_shift     %ld\n", (long)finfo.compression_info.out.chunk_shift);
1830                 d_printf("\tcluster_shift   %ld\n", (long)finfo.compression_info.out.cluster_shift);
1831         }
1832
1833         /* shadow copies if available */
1834         fnum = smbcli_open(ctx->cli->tree, fname, O_RDONLY, DENY_NONE);
1835         if (fnum != -1) {
1836                 struct smb_shadow_copy info;
1837                 int i;
1838                 info.in.file.fnum = fnum;
1839                 info.in.max_data = ~0;
1840                 status = smb_raw_shadow_data(ctx->cli->tree, ctx, &info);
1841                 if (NT_STATUS_IS_OK(status)) {
1842                         d_printf("\tshadow_copy: %u volumes  %u names\n",
1843                                  info.out.num_volumes, info.out.num_names);
1844                         for (i=0;i<info.out.num_names;i++) {
1845                                 d_printf("\t%s\n", info.out.names[i]);
1846                                 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1847                                 finfo.generic.in.file.path = talloc_asprintf(ctx, "%s%s", 
1848                                                                              info.out.names[i], fname); 
1849                                 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1850                                 if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_PATH_NOT_FOUND) ||
1851                                     NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
1852                                         continue;
1853                                 }
1854                                 if (!NT_STATUS_IS_OK(status)) {
1855                                         d_printf("%s - %s\n", finfo.generic.in.file.path, 
1856                                                  nt_errstr(status));
1857                                         return 1;
1858                                 }
1859                                 
1860                                 d_printf("\t\tcreate_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1861                                 d_printf("\t\twrite_time:     %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1862                                 d_printf("\t\tchange_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1863                                 d_printf("\t\tsize:           %lu\n", (unsigned long)finfo.all_info.out.size);
1864                         }
1865                 }
1866         }
1867         
1868         return 0;
1869 }
1870
1871
1872 /****************************************************************************
1873 shows EA contents
1874 ****************************************************************************/
1875 static int cmd_eainfo(struct smbclient_context *ctx, const char **args)
1876 {
1877         char *fname;
1878         union smb_fileinfo finfo;
1879         NTSTATUS status;
1880         int i;
1881
1882         if (!args[1]) {
1883                 d_printf("eainfo <filename>\n");
1884                 return 1;
1885         }
1886         fname = talloc_strdup(ctx, args[1]);
1887
1888         finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1889         finfo.generic.in.file.path = fname;
1890         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1891         
1892         if (!NT_STATUS_IS_OK(status)) {
1893                 d_printf("RAW_FILEINFO_ALL_EAS - %s\n", nt_errstr(status));
1894                 return 1;
1895         }
1896
1897         d_printf("%s has %d EAs\n", fname, finfo.all_eas.out.num_eas);
1898
1899         for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1900                 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1901                          finfo.all_eas.out.eas[i].flags,
1902                          (int)finfo.all_eas.out.eas[i].value.length,
1903                          finfo.all_eas.out.eas[i].name.s);
1904                 fflush(stdout);
1905                 dump_data(0, 
1906                           finfo.all_eas.out.eas[i].value.data,
1907                           finfo.all_eas.out.eas[i].value.length);
1908         }
1909
1910         return 0;
1911 }
1912
1913
1914 /****************************************************************************
1915 show any ACL on a file
1916 ****************************************************************************/
1917 static int cmd_acl(struct smbclient_context *ctx, const char **args)
1918 {
1919         char *fname;
1920         union smb_fileinfo query;
1921         NTSTATUS status;
1922         int fnum;
1923
1924         if (!args[1]) {
1925                 d_printf("acl <filename>\n");
1926                 return 1;
1927         }
1928         fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1929
1930         fnum = smbcli_nt_create_full(ctx->cli->tree, fname, 0, 
1931                                      SEC_STD_READ_CONTROL,
1932                                      0,
1933                                      NTCREATEX_SHARE_ACCESS_DELETE|
1934                                      NTCREATEX_SHARE_ACCESS_READ|
1935                                      NTCREATEX_SHARE_ACCESS_WRITE, 
1936                                      NTCREATEX_DISP_OPEN,
1937                                      0, 0);
1938         if (fnum == -1) {
1939                 d_printf("%s - %s\n", fname, smbcli_errstr(ctx->cli->tree));
1940                 return -1;
1941         }
1942
1943         query.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
1944         query.query_secdesc.in.file.fnum = fnum;
1945         query.query_secdesc.in.secinfo_flags = 0x7;
1946
1947         status = smb_raw_fileinfo(ctx->cli->tree, ctx, &query);
1948         if (!NT_STATUS_IS_OK(status)) {
1949                 d_printf("%s - %s\n", fname, nt_errstr(status));
1950                 return 1;
1951         }
1952
1953         NDR_PRINT_DEBUG(security_descriptor, query.query_secdesc.out.sd);
1954
1955         return 0;
1956 }
1957
1958 /****************************************************************************
1959 lookup a name or sid
1960 ****************************************************************************/
1961 static int cmd_lookup(struct smbclient_context *ctx, const char **args)
1962 {
1963         NTSTATUS status;
1964         struct dom_sid *sid;
1965
1966         if (!args[1]) {
1967                 d_printf("lookup <sid|name>\n");
1968                 return 1;
1969         }
1970
1971         sid = dom_sid_parse_talloc(ctx, args[1]);
1972         if (sid == NULL) {
1973                 const char *sidstr;
1974                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sidstr);
1975                 if (!NT_STATUS_IS_OK(status)) {
1976                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
1977                         return 1;
1978                 }
1979
1980                 d_printf("%s\n", sidstr);
1981         } else {
1982                 const char *name;
1983                 status = smblsa_lookup_sid(ctx->cli, args[1], ctx, &name);
1984                 if (!NT_STATUS_IS_OK(status)) {
1985                         d_printf("lsa_LookupSids - %s\n", nt_errstr(status));
1986                         return 1;
1987                 }
1988
1989                 d_printf("%s\n", name);
1990         }
1991
1992         return 0;
1993 }
1994
1995 /****************************************************************************
1996 show privileges for a user
1997 ****************************************************************************/
1998 static int cmd_privileges(struct smbclient_context *ctx, const char **args)
1999 {
2000         NTSTATUS status;
2001         struct dom_sid *sid;
2002         struct lsa_RightSet rights;
2003         unsigned i;
2004
2005         if (!args[1]) {
2006                 d_printf("privileges <sid|name>\n");
2007                 return 1;
2008         }
2009
2010         sid = dom_sid_parse_talloc(ctx, args[1]);
2011         if (sid == NULL) {
2012                 const char *sid_str;
2013                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2014                 if (!NT_STATUS_IS_OK(status)) {
2015                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2016                         return 1;
2017                 }
2018                 sid = dom_sid_parse_talloc(ctx, sid_str);
2019         }
2020
2021         status = smblsa_sid_privileges(ctx->cli, sid, ctx, &rights);
2022         if (!NT_STATUS_IS_OK(status)) {
2023                 d_printf("lsa_EnumAccountRights - %s\n", nt_errstr(status));
2024                 return 1;
2025         }
2026
2027         for (i=0;i<rights.count;i++) {
2028                 d_printf("\t%s\n", rights.names[i].string);
2029         }
2030
2031         return 0;
2032 }
2033
2034
2035 /****************************************************************************
2036 add privileges for a user
2037 ****************************************************************************/
2038 static int cmd_addprivileges(struct smbclient_context *ctx, const char **args)
2039 {
2040         NTSTATUS status;
2041         struct dom_sid *sid;
2042         struct lsa_RightSet rights;
2043         int i;
2044
2045         if (!args[1]) {
2046                 d_printf("addprivileges <sid|name> <privilege...>\n");
2047                 return 1;
2048         }
2049
2050         sid = dom_sid_parse_talloc(ctx, args[1]);
2051         if (sid == NULL) {
2052                 const char *sid_str;
2053                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2054                 if (!NT_STATUS_IS_OK(status)) {
2055                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2056                         return 1;
2057                 }
2058                 sid = dom_sid_parse_talloc(ctx, sid_str);
2059         }
2060
2061         ZERO_STRUCT(rights);
2062         for (i = 2; args[i]; i++) {
2063                 rights.names = talloc_realloc(ctx, rights.names, 
2064                                               struct lsa_StringLarge, rights.count+1);
2065                 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2066                 rights.count++;
2067         }
2068
2069
2070         status = smblsa_sid_add_privileges(ctx->cli, sid, ctx, &rights);
2071         if (!NT_STATUS_IS_OK(status)) {
2072                 d_printf("lsa_AddAccountRights - %s\n", nt_errstr(status));
2073                 return 1;
2074         }
2075
2076         return 0;
2077 }
2078
2079 /****************************************************************************
2080 delete privileges for a user
2081 ****************************************************************************/
2082 static int cmd_delprivileges(struct smbclient_context *ctx, const char **args)
2083 {
2084         NTSTATUS status;
2085         struct dom_sid *sid;
2086         struct lsa_RightSet rights;
2087         int i;
2088
2089         if (!args[1]) {
2090                 d_printf("delprivileges <sid|name> <privilege...>\n");
2091                 return 1;
2092         }
2093
2094         sid = dom_sid_parse_talloc(ctx, args[1]);
2095         if (sid == NULL) {
2096                 const char *sid_str;
2097                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2098                 if (!NT_STATUS_IS_OK(status)) {
2099                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2100                         return 1;
2101                 }
2102                 sid = dom_sid_parse_talloc(ctx, sid_str);
2103         }
2104
2105         ZERO_STRUCT(rights);
2106         for (i = 2; args[i]; i++) {
2107                 rights.names = talloc_realloc(ctx, rights.names, 
2108                                               struct lsa_StringLarge, rights.count+1);
2109                 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2110                 rights.count++;
2111         }
2112
2113
2114         status = smblsa_sid_del_privileges(ctx->cli, sid, ctx, &rights);
2115         if (!NT_STATUS_IS_OK(status)) {
2116                 d_printf("lsa_RemoveAccountRights - %s\n", nt_errstr(status));
2117                 return 1;
2118         }
2119
2120         return 0;
2121 }
2122
2123
2124 /****************************************************************************
2125 ****************************************************************************/
2126 static int cmd_open(struct smbclient_context *ctx, const char **args)
2127 {
2128         char *mask;
2129         
2130         if (!args[1]) {
2131                 d_printf("open <filename>\n");
2132                 return 1;
2133         }
2134         mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2135
2136         smbcli_open(ctx->cli->tree, mask, O_RDWR, DENY_ALL);
2137
2138         return 0;
2139 }
2140
2141
2142 /****************************************************************************
2143 remove a directory
2144 ****************************************************************************/
2145 static int cmd_rmdir(struct smbclient_context *ctx, const char **args)
2146 {
2147         char *mask;
2148   
2149         if (!args[1]) {
2150                 d_printf("rmdir <dirname>\n");
2151                 return 1;
2152         }
2153         mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2154
2155         if (NT_STATUS_IS_ERR(smbcli_rmdir(ctx->cli->tree, mask))) {
2156                 d_printf("%s removing remote directory file %s\n",
2157                          smbcli_errstr(ctx->cli->tree),mask);
2158         }
2159         
2160         return 0;
2161 }
2162
2163 /****************************************************************************
2164  UNIX hardlink.
2165 ****************************************************************************/
2166 static int cmd_link(struct smbclient_context *ctx, const char **args)
2167 {
2168         char *src,*dest;
2169   
2170         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2171                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2172                 return 1;
2173         }
2174
2175         
2176         if (!args[1] || !args[2]) {
2177                 d_printf("link <src> <dest>\n");
2178                 return 1;
2179         }
2180
2181         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2182         dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2183
2184         if (NT_STATUS_IS_ERR(smbcli_unix_hardlink(ctx->cli->tree, src, dest))) {
2185                 d_printf("%s linking files (%s -> %s)\n", smbcli_errstr(ctx->cli->tree), src, dest);
2186                 return 1;
2187         }  
2188
2189         return 0;
2190 }
2191
2192 /****************************************************************************
2193  UNIX symlink.
2194 ****************************************************************************/
2195
2196 static int cmd_symlink(struct smbclient_context *ctx, const char **args)
2197 {
2198         char *src,*dest;
2199   
2200         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2201                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2202                 return 1;
2203         }
2204
2205         if (!args[1] || !args[2]) {
2206                 d_printf("symlink <src> <dest>\n");
2207                 return 1;
2208         }
2209
2210         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2211         dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2212
2213         if (NT_STATUS_IS_ERR(smbcli_unix_symlink(ctx->cli->tree, src, dest))) {
2214                 d_printf("%s symlinking files (%s -> %s)\n",
2215                         smbcli_errstr(ctx->cli->tree), src, dest);
2216                 return 1;
2217         } 
2218
2219         return 0;
2220 }
2221
2222 /****************************************************************************
2223  UNIX chmod.
2224 ****************************************************************************/
2225
2226 static int cmd_chmod(struct smbclient_context *ctx, const char **args)
2227 {
2228         char *src;
2229         mode_t mode;
2230   
2231         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2232                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2233                 return 1;
2234         }
2235
2236         if (!args[1] || !args[2]) {
2237                 d_printf("chmod mode file\n");
2238                 return 1;
2239         }
2240
2241         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2242         
2243         mode = (mode_t)strtol(args[1], NULL, 8);
2244
2245         if (NT_STATUS_IS_ERR(smbcli_unix_chmod(ctx->cli->tree, src, mode))) {
2246                 d_printf("%s chmod file %s 0%o\n",
2247                         smbcli_errstr(ctx->cli->tree), src, (mode_t)mode);
2248                 return 1;
2249         } 
2250
2251         return 0;
2252 }
2253
2254 /****************************************************************************
2255  UNIX chown.
2256 ****************************************************************************/
2257
2258 static int cmd_chown(struct smbclient_context *ctx, const char **args)
2259 {
2260         char *src;
2261         uid_t uid;
2262         gid_t gid;
2263   
2264         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2265                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2266                 return 1;
2267         }
2268
2269         if (!args[1] || !args[2] || !args[3]) {
2270                 d_printf("chown uid gid file\n");
2271                 return 1;
2272         }
2273
2274         uid = (uid_t)atoi(args[1]);
2275         gid = (gid_t)atoi(args[2]);
2276         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[3]);
2277
2278         if (NT_STATUS_IS_ERR(smbcli_unix_chown(ctx->cli->tree, src, uid, gid))) {
2279                 d_printf("%s chown file %s uid=%d, gid=%d\n",
2280                         smbcli_errstr(ctx->cli->tree), src, (int)uid, (int)gid);
2281                 return 1;
2282         } 
2283
2284         return 0;
2285 }
2286
2287 /****************************************************************************
2288 rename some files
2289 ****************************************************************************/
2290 static int cmd_rename(struct smbclient_context *ctx, const char **args)
2291 {
2292         char *src,*dest;
2293   
2294         if (!args[1] || !args[2]) {
2295                 d_printf("rename <src> <dest>\n");
2296                 return 1;
2297         }
2298
2299         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2300         dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2301
2302         if (NT_STATUS_IS_ERR(smbcli_rename(ctx->cli->tree, src, dest))) {
2303                 d_printf("%s renaming files\n",smbcli_errstr(ctx->cli->tree));
2304                 return 1;
2305         }
2306         
2307         return 0;
2308 }
2309
2310
2311 /****************************************************************************
2312 toggle the prompt flag
2313 ****************************************************************************/
2314 static int cmd_prompt(struct smbclient_context *ctx, const char **args)
2315 {
2316         ctx->prompt = !ctx->prompt;
2317         DEBUG(2,("prompting is now %s\n",ctx->prompt?"on":"off"));
2318         
2319         return 1;
2320 }
2321
2322
2323 /****************************************************************************
2324 set the newer than time
2325 ****************************************************************************/
2326 static int cmd_newer(struct smbclient_context *ctx, const char **args)
2327 {
2328         struct stat sbuf;
2329
2330         if (args[1] && (stat(args[1],&sbuf) == 0)) {
2331                 ctx->newer_than = sbuf.st_mtime;
2332                 DEBUG(1,("Getting files newer than %s",
2333                          asctime(localtime(&ctx->newer_than))));
2334         } else {
2335                 ctx->newer_than = 0;
2336         }
2337
2338         if (args[1] && ctx->newer_than == 0) {
2339                 d_printf("Error setting newer-than time\n");
2340                 return 1;
2341         }
2342
2343         return 0;
2344 }
2345
2346 /****************************************************************************
2347 set the archive level
2348 ****************************************************************************/
2349 static int cmd_archive(struct smbclient_context *ctx, const char **args)
2350 {
2351         if (args[1]) {
2352                 ctx->archive_level = atoi(args[1]);
2353         } else
2354                 d_printf("Archive level is %d\n",ctx->archive_level);
2355
2356         return 0;
2357 }
2358
2359 /****************************************************************************
2360 toggle the lowercaseflag
2361 ****************************************************************************/
2362 static int cmd_lowercase(struct smbclient_context *ctx, const char **args)
2363 {
2364         ctx->lowercase = !ctx->lowercase;
2365         DEBUG(2,("filename lowercasing is now %s\n",ctx->lowercase?"on":"off"));
2366
2367         return 0;
2368 }
2369
2370
2371
2372
2373 /****************************************************************************
2374 toggle the recurse flag
2375 ****************************************************************************/
2376 static int cmd_recurse(struct smbclient_context *ctx, const char **args)
2377 {
2378         ctx->recurse = !ctx->recurse;
2379         DEBUG(2,("directory recursion is now %s\n",ctx->recurse?"on":"off"));
2380
2381         return 0;
2382 }
2383
2384 /****************************************************************************
2385 toggle the translate flag
2386 ****************************************************************************/
2387 static int cmd_translate(struct smbclient_context *ctx, const char **args)
2388 {
2389         ctx->translation = !ctx->translation;
2390         DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
2391                  ctx->translation?"on":"off"));
2392
2393         return 0;
2394 }
2395
2396
2397 /****************************************************************************
2398 do a printmode command
2399 ****************************************************************************/
2400 static int cmd_printmode(struct smbclient_context *ctx, const char **args)
2401 {
2402         if (args[1]) {
2403                 if (strequal(args[1],"text")) {
2404                         ctx->printmode = 0;      
2405                 } else {
2406                         if (strequal(args[1],"graphics"))
2407                                 ctx->printmode = 1;
2408                         else
2409                                 ctx->printmode = atoi(args[1]);
2410                 }
2411         }
2412
2413         switch(ctx->printmode)
2414         {
2415                 case 0: 
2416                         DEBUG(2,("the printmode is now text\n"));
2417                         break;
2418                 case 1: 
2419                         DEBUG(2,("the printmode is now graphics\n"));
2420                         break;
2421                 default: 
2422                         DEBUG(2,("the printmode is now %d\n", ctx->printmode));
2423                         break;
2424         }
2425         
2426         return 0;
2427 }
2428
2429 /****************************************************************************
2430  do the lcd command
2431  ****************************************************************************/
2432 static int cmd_lcd(struct smbclient_context *ctx, const char **args)
2433 {
2434         char d[PATH_MAX];
2435         
2436         if (args[1]) 
2437                 chdir(args[1]);
2438         DEBUG(2,("the local directory is now %s\n",getcwd(d, PATH_MAX)));
2439
2440         return 0;
2441 }
2442
2443 /****************************************************************************
2444 history
2445 ****************************************************************************/
2446 static int cmd_history(struct smbclient_context *ctx, const char **args)
2447 {
2448 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
2449         HIST_ENTRY **hlist;
2450         int i;
2451
2452         hlist = history_list();
2453         
2454         for (i = 0; hlist && hlist[i]; i++) {
2455                 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
2456         }
2457 #else
2458         DEBUG(0,("no history without readline support\n"));
2459 #endif
2460
2461         return 0;
2462 }
2463
2464 /****************************************************************************
2465  get a file restarting at end of local file
2466  ****************************************************************************/
2467 static int cmd_reget(struct smbclient_context *ctx, const char **args)
2468 {
2469         char *local_name;
2470         char *remote_name;
2471
2472         if (!args[1]) {
2473                 d_printf("reget <filename>\n");
2474                 return 1;
2475         }
2476         remote_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2477         dos_clean_name(remote_name);
2478         
2479         if (args[2]) 
2480                 local_name = talloc_strdup(ctx, args[2]);
2481         else
2482                 local_name = talloc_strdup(ctx, args[1]);
2483         
2484         return do_get(ctx, remote_name, local_name, true);
2485 }
2486
2487 /****************************************************************************
2488  put a file restarting at end of local file
2489  ****************************************************************************/
2490 static int cmd_reput(struct smbclient_context *ctx, const char **args)
2491 {
2492         char *local_name;
2493         char *remote_name;
2494         
2495         if (!args[1]) {
2496                 d_printf("reput <filename>\n");
2497                 return 1;
2498         }
2499         local_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2500   
2501         if (!file_exist(local_name)) {
2502                 d_printf("%s does not exist\n", local_name);
2503                 return 1;
2504         }
2505
2506         if (args[2]) 
2507                 remote_name = talloc_strdup(ctx, args[2]);
2508         else
2509                 remote_name = talloc_strdup(ctx, args[1]);
2510         
2511         dos_clean_name(remote_name);
2512
2513         return do_put(ctx, remote_name, local_name, true);
2514 }
2515
2516
2517 /*
2518   return a string representing a share type
2519 */
2520 static const char *share_type_str(uint32_t type)
2521 {
2522         switch (type & 0xF) {
2523         case STYPE_DISKTREE: 
2524                 return "Disk";
2525         case STYPE_PRINTQ: 
2526                 return "Printer";
2527         case STYPE_DEVICE: 
2528                 return "Device";
2529         case STYPE_IPC: 
2530                 return "IPC";
2531         default:
2532                 return "Unknown";
2533         }
2534 }
2535
2536
2537 /*
2538   display a list of shares from a level 1 share enum
2539 */
2540 static void display_share_result(struct srvsvc_NetShareCtr1 *ctr1)
2541 {
2542         int i;
2543
2544         for (i=0;i<ctr1->count;i++) {
2545                 struct srvsvc_NetShareInfo1 *info = ctr1->array+i;
2546
2547                 printf("\t%-15s %-10.10s %s\n", 
2548                        info->name, 
2549                        share_type_str(info->type), 
2550                        info->comment);
2551         }
2552 }
2553
2554
2555
2556 /****************************************************************************
2557 try and browse available shares on a host
2558 ****************************************************************************/
2559 static bool browse_host(struct loadparm_context *lp_ctx,
2560                         struct event_context *ev_ctx,
2561                         const char *query_host)
2562 {
2563         struct dcerpc_pipe *p;
2564         char *binding;
2565         NTSTATUS status;
2566         struct srvsvc_NetShareEnumAll r;
2567         struct srvsvc_NetShareInfoCtr info_ctr;
2568         uint32_t resume_handle = 0;
2569         TALLOC_CTX *mem_ctx = talloc_init("browse_host");
2570         struct srvsvc_NetShareCtr1 ctr1;
2571         uint32_t totalentries = 0;
2572
2573         binding = talloc_asprintf(mem_ctx, "ncacn_np:%s", query_host);
2574
2575         status = dcerpc_pipe_connect(mem_ctx, &p, binding, 
2576                                          &ndr_table_srvsvc,
2577                                      cmdline_credentials, ev_ctx,
2578                                      lp_ctx);
2579         if (!NT_STATUS_IS_OK(status)) {
2580                 d_printf("Failed to connect to %s - %s\n", 
2581                          binding, nt_errstr(status));
2582                 talloc_free(mem_ctx);
2583                 return false;
2584         }
2585
2586         info_ctr.level = 1;
2587         info_ctr.ctr.ctr1 = &ctr1;
2588
2589         r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
2590         r.in.info_ctr = &info_ctr;
2591         r.in.max_buffer = ~0;
2592         r.in.resume_handle = &resume_handle;
2593         r.out.resume_handle = &resume_handle;
2594         r.out.totalentries = &totalentries;
2595         r.out.info_ctr = &info_ctr;
2596
2597         d_printf("\n\tSharename       Type       Comment\n");
2598         d_printf("\t---------       ----       -------\n");
2599
2600         do {
2601                 ZERO_STRUCT(ctr1);
2602                 status = dcerpc_srvsvc_NetShareEnumAll(p, mem_ctx, &r);
2603
2604                 if (NT_STATUS_IS_OK(status) && 
2605                     (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA) ||
2606                      W_ERROR_IS_OK(r.out.result)) &&
2607                     r.out.info_ctr->ctr.ctr1) {
2608                         display_share_result(r.out.info_ctr->ctr.ctr1);
2609                         resume_handle += r.out.info_ctr->ctr.ctr1->count;
2610                 }
2611         } while (NT_STATUS_IS_OK(status) && W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA));
2612
2613         talloc_free(mem_ctx);
2614
2615         if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2616                 d_printf("Failed NetShareEnumAll %s - %s/%s\n", 
2617                          binding, nt_errstr(status), win_errstr(r.out.result));
2618                 return false;
2619         }
2620
2621         return false;
2622 }
2623
2624 /****************************************************************************
2625 try and browse available connections on a host
2626 ****************************************************************************/
2627 static bool list_servers(const char *wk_grp)
2628 {
2629         d_printf("REWRITE: list servers not implemented\n");
2630         return false;
2631 }
2632
2633 /* Some constants for completing filename arguments */
2634
2635 #define COMPL_NONE        0          /* No completions */
2636 #define COMPL_REMOTE      1          /* Complete remote filename */
2637 #define COMPL_LOCAL       2          /* Complete local filename */
2638
2639 static int cmd_help(struct smbclient_context *ctx, const char **args);
2640
2641 /* This defines the commands supported by this client.
2642  * NOTE: The "!" must be the last one in the list because it's fn pointer
2643  *       field is NULL, and NULL in that field is used in process_tok()
2644  *       (below) to indicate the end of the list.  crh
2645  */
2646 static struct
2647 {
2648   const char *name;
2649   int (*fn)(struct smbclient_context *ctx, const char **args);
2650   const char *description;
2651   char compl_args[2];      /* Completion argument info */
2652 } commands[] = 
2653 {
2654   {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2655   {"addprivileges",cmd_addprivileges,"<sid|name> <privilege...> add privileges for a user",{COMPL_NONE,COMPL_NONE}},
2656   {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
2657   {"acl",cmd_acl,"<file> show file ACL",{COMPL_NONE,COMPL_NONE}},
2658   {"allinfo",cmd_allinfo,"<file> show all possible info about a file",{COMPL_NONE,COMPL_NONE}},
2659   {"archive",cmd_archive,"<level>\n0=ignore archive bit\n1=only get archive files\n2=only get archive files and reset archive bit\n3=get all files and reset archive bit",{COMPL_NONE,COMPL_NONE}},
2660   {"cancel",cmd_rewrite,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
2661   {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
2662   {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
2663   {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
2664   {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2665   {"delprivileges",cmd_delprivileges,"<sid|name> <privilege...> remove privileges for a user",{COMPL_NONE,COMPL_NONE}},
2666   {"deltree",cmd_deltree,"<dir> delete a whole directory tree",{COMPL_REMOTE,COMPL_NONE}},
2667   {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2668   {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2669   {"eainfo",cmd_eainfo,"<file> show EA contents for a file",{COMPL_NONE,COMPL_NONE}},
2670   {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2671   {"fsinfo",cmd_fsinfo,"query file system info",{COMPL_NONE,COMPL_NONE}},
2672   {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
2673   {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2674   {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
2675   {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
2676   {"link",cmd_link,"<src> <dest> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2677   {"lookup",cmd_lookup,"<sid|name> show SID for name or name for SID",{COMPL_NONE,COMPL_NONE}},
2678   {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},  
2679   {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2680   {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
2681   {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2682   {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
2683   {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2684   {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},  
2685   {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
2686   {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
2687   {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
2688   {"privileges",cmd_privileges,"<user> show privileges for a user",{COMPL_NONE,COMPL_NONE}},
2689   {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
2690   {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
2691   {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},  
2692   {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
2693   {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
2694   {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2695   {"queue",cmd_rewrite,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2696   {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2697   {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2698   {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},  
2699   {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
2700   {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2701   {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
2702   {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2703   {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2704   {"symlink",cmd_symlink,"<src> <dest> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2705   {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2706   
2707   /* Yes, this must be here, see crh's comment above. */
2708   {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2709   {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2710 };
2711
2712
2713 /*******************************************************************
2714   lookup a command string in the list of commands, including 
2715   abbreviations
2716   ******************************************************************/
2717 static int process_tok(const char *tok)
2718 {
2719         int i = 0, matches = 0;
2720         int cmd=0;
2721         int tok_len = strlen(tok);
2722         
2723         while (commands[i].fn != NULL) {
2724                 if (strequal(commands[i].name,tok)) {
2725                         matches = 1;
2726                         cmd = i;
2727                         break;
2728                 } else if (strncasecmp(commands[i].name, tok, tok_len) == 0) {
2729                         matches++;
2730                         cmd = i;
2731                 }
2732                 i++;
2733         }
2734   
2735         if (matches == 0)
2736                 return(-1);
2737         else if (matches == 1)
2738                 return(cmd);
2739         else
2740                 return(-2);
2741 }
2742
2743 /****************************************************************************
2744 help
2745 ****************************************************************************/
2746 static int cmd_help(struct smbclient_context *ctx, const char **args)
2747 {
2748         int i=0,j;
2749         
2750         if (args[1]) {
2751                 if ((i = process_tok(args[1])) >= 0)
2752                         d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2753         } else {
2754                 while (commands[i].description) {
2755                         for (j=0; commands[i].description && (j<5); j++) {
2756                                 d_printf("%-15s",commands[i].name);
2757                                 i++;
2758                         }
2759                         d_printf("\n");
2760                 }
2761         }
2762         return 0;
2763 }
2764
2765 static int process_line(struct smbclient_context *ctx, const char *cline);
2766 /****************************************************************************
2767 process a -c command string
2768 ****************************************************************************/
2769 static int process_command_string(struct smbclient_context *ctx, const char *cmd)
2770 {
2771         char **lines;
2772         int i, rc = 0;
2773
2774         lines = str_list_make(NULL, cmd, ";");
2775         for (i = 0; lines[i]; i++) {
2776                 rc |= process_line(ctx, lines[i]);
2777         }
2778         talloc_free(lines);
2779
2780         return rc;
2781 }       
2782
2783 #define MAX_COMPLETIONS 100
2784
2785 typedef struct {
2786         char *dirmask;
2787         char **matches;
2788         int count, samelen;
2789         const char *text;
2790         int len;
2791 } completion_remote_t;
2792
2793 static void completion_remote_filter(struct clilist_file_info *f, const char *mask, void *state)
2794 {
2795         completion_remote_t *info = (completion_remote_t *)state;
2796
2797         if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (!ISDOT(f->name)) && (!ISDOTDOT(f->name))) {
2798                 if ((info->dirmask[0] == 0) && !(f->attrib & FILE_ATTRIBUTE_DIRECTORY))
2799                         info->matches[info->count] = strdup(f->name);
2800                 else {
2801                         char *tmp;
2802
2803                         if (info->dirmask[0] != 0)
2804                                 tmp = talloc_asprintf(NULL, "%s/%s", info->dirmask, f->name);
2805                         else
2806                                 tmp = talloc_strdup(NULL, f->name);
2807                         
2808                         if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2809                                 tmp = talloc_append_string(NULL, tmp, "/");
2810                         info->matches[info->count] = tmp;
2811                 }
2812                 if (info->matches[info->count] == NULL)
2813                         return;
2814                 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2815                         smb_readline_ca_char(0);
2816
2817                 if (info->count == 1)
2818                         info->samelen = strlen(info->matches[info->count]);
2819                 else
2820                         while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0)
2821                                 info->samelen--;
2822                 info->count++;
2823         }
2824 }
2825
2826 static char **remote_completion(const char *text, int len)
2827 {
2828         char *dirmask;
2829         int i;
2830         completion_remote_t info;
2831
2832         info.samelen = len;
2833         info.text = text;
2834         info.len = len;
2835  
2836         if (len >= PATH_MAX)
2837                 return(NULL);
2838
2839         info.matches = malloc_array_p(char *, MAX_COMPLETIONS);
2840         if (!info.matches) return NULL;
2841         info.matches[0] = NULL;
2842
2843         for (i = len-1; i >= 0; i--)
2844                 if ((text[i] == '/') || (text[i] == '\\'))
2845                         break;
2846         info.text = text+i+1;
2847         info.samelen = info.len = len-i-1;
2848
2849         if (i > 0) {
2850                 info.dirmask = talloc_strndup(NULL, text, i+1);
2851                 info.dirmask[i+1] = 0;
2852                 asprintf(&dirmask, "%s%*s*", rl_ctx->remote_cur_dir, i-1, text);
2853         } else
2854                 asprintf(&dirmask, "%s*", rl_ctx->remote_cur_dir);
2855
2856         if (smbcli_list(rl_ctx->cli->tree, dirmask, 
2857                      FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN, 
2858                      completion_remote_filter, &info) < 0)
2859                 goto cleanup;
2860
2861         if (info.count == 2)
2862                 info.matches[0] = strdup(info.matches[1]);
2863         else {
2864                 info.matches[0] = malloc_array_p(char, info.samelen+1);
2865                 if (!info.matches[0])
2866                         goto cleanup;
2867                 strncpy(info.matches[0], info.matches[1], info.samelen);
2868                 info.matches[0][info.samelen] = 0;
2869         }
2870         info.matches[info.count] = NULL;
2871         return info.matches;
2872
2873 cleanup:
2874         for (i = 0; i < info.count; i++)
2875                 free(info.matches[i]);
2876         free(info.matches);
2877         return NULL;
2878 }
2879
2880 static char **completion_fn(const char *text, int start, int end)
2881 {
2882         smb_readline_ca_char(' ');
2883
2884         if (start) {
2885                 const char *buf, *sp;
2886                 int i;
2887                 char compl_type;
2888
2889                 buf = smb_readline_get_line_buffer();
2890                 if (buf == NULL)
2891                         return NULL;
2892                 
2893                 sp = strchr(buf, ' ');
2894                 if (sp == NULL)
2895                         return NULL;
2896                 
2897                 for (i = 0; commands[i].name; i++)
2898                         if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0))
2899                                 break;
2900                 if (commands[i].name == NULL)
2901                         return NULL;
2902
2903                 while (*sp == ' ')
2904                         sp++;
2905
2906                 if (sp == (buf + start))
2907                         compl_type = commands[i].compl_args[0];
2908                 else
2909                         compl_type = commands[i].compl_args[1];
2910
2911                 if (compl_type == COMPL_REMOTE)
2912                         return remote_completion(text, end - start);
2913                 else /* fall back to local filename completion */
2914                         return NULL;
2915         } else {
2916                 char **matches;
2917                 int i, len, samelen = 0, count=1;
2918
2919                 matches = malloc_array_p(char *, MAX_COMPLETIONS);
2920                 if (!matches) return NULL;
2921                 matches[0] = NULL;
2922
2923                 len = strlen(text);
2924                 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
2925                         if (strncmp(text, commands[i].name, len) == 0) {
2926                                 matches[count] = strdup(commands[i].name);
2927                                 if (!matches[count])
2928                                         goto cleanup;
2929                                 if (count == 1)
2930                                         samelen = strlen(matches[count]);
2931                                 else
2932                                         while (strncmp(matches[count], matches[count-1], samelen) != 0)
2933                                                 samelen--;
2934                                 count++;
2935                         }
2936                 }
2937
2938                 switch (count) {
2939                 case 0: /* should never happen */
2940                 case 1:
2941                         goto cleanup;
2942                 case 2:
2943                         matches[0] = strdup(matches[1]);
2944                         break;
2945                 default:
2946                         matches[0] = malloc_array_p(char, samelen+1);
2947                         if (!matches[0])
2948                                 goto cleanup;
2949                         strncpy(matches[0], matches[1], samelen);
2950                         matches[0][samelen] = 0;
2951                 }
2952                 matches[count] = NULL;
2953                 return matches;
2954
2955 cleanup:
2956                 count--;
2957                 while (count >= 0) {
2958                         free(matches[count]);
2959                         count--;
2960                 }
2961                 free(matches);
2962                 return NULL;
2963         }
2964 }
2965
2966 /****************************************************************************
2967 make sure we swallow keepalives during idle time
2968 ****************************************************************************/
2969 static void readline_callback(void)
2970 {
2971         static time_t last_t;
2972         time_t t;
2973
2974         t = time(NULL);
2975
2976         if (t - last_t < 5) return;
2977
2978         last_t = t;
2979
2980         smbcli_transport_process(rl_ctx->cli->transport);
2981
2982         if (rl_ctx->cli->tree) {
2983                 smbcli_chkpath(rl_ctx->cli->tree, "\\");
2984         }
2985 }
2986
2987 static int process_line(struct smbclient_context *ctx, const char *cline)
2988 {
2989         const char **args;
2990         int i;
2991
2992         /* and get the first part of the command */
2993         args = str_list_make_shell(ctx, cline, NULL);
2994         if (!args || !args[0])
2995                 return 0;
2996
2997         if ((i = process_tok(args[0])) >= 0) {
2998                 i = commands[i].fn(ctx, args);
2999         } else if (i == -2) {
3000                 d_printf("%s: command abbreviation ambiguous\n",args[0]);
3001         } else {
3002                 d_printf("%s: command not found\n",args[0]);
3003         }
3004
3005         talloc_free(args);
3006
3007         return i;
3008 }
3009
3010 /****************************************************************************
3011 process commands on stdin
3012 ****************************************************************************/
3013 static int process_stdin(struct smbclient_context *ctx)
3014 {
3015         int rc = 0;
3016         while (1) {
3017                 /* display a prompt */
3018                 char *the_prompt = talloc_asprintf(ctx, "smb: %s> ", ctx->remote_cur_dir);
3019                 char *cline = smb_readline(the_prompt, readline_callback, completion_fn);
3020                 talloc_free(the_prompt);
3021
3022                 if (!cline) break;
3023
3024                 /* special case - first char is ! */
3025                 if (*cline == '!') {
3026                         system(cline + 1);
3027                         free(cline);
3028                         continue;
3029                 }
3030
3031                 rc |= process_command_string(ctx, cline);
3032                 free(cline);
3033
3034         }
3035
3036         return rc;
3037 }
3038
3039
3040 /***************************************************** 
3041 return a connection to a server
3042 *******************************************************/
3043 static bool do_connect(struct smbclient_context *ctx, 
3044                        struct event_context *ev_ctx,
3045                        struct resolve_context *resolve_ctx,
3046                        const char *specified_server, const char **ports, 
3047                        const char *specified_share, 
3048                        struct cli_credentials *cred, 
3049                        struct smbcli_options *options,
3050                        struct smbcli_session_options *session_options,
3051                            struct smb_iconv_convenience *iconv_convenience)
3052 {
3053         NTSTATUS status;
3054         char *server, *share;
3055
3056         rl_ctx = ctx; /* Ugly hack */
3057
3058         if (strncmp(specified_share, "\\\\", 2) == 0 ||
3059             strncmp(specified_share, "//", 2) == 0) {
3060                 smbcli_parse_unc(specified_share, ctx, &server, &share);
3061         } else {
3062                 share = talloc_strdup(ctx, specified_share);
3063                 server = talloc_strdup(ctx, specified_server);
3064         }
3065
3066         ctx->remote_cur_dir = talloc_strdup(ctx, "\\");
3067         
3068         status = smbcli_full_connection(ctx, &ctx->cli, server, ports,
3069                                         share, NULL, cred, resolve_ctx, 
3070                                         ev_ctx, options, session_options,
3071                                         iconv_convenience);
3072         if (!NT_STATUS_IS_OK(status)) {
3073                 d_printf("Connection to \\\\%s\\%s failed - %s\n", 
3074                          server, share, nt_errstr(status));
3075                 talloc_free(ctx);
3076                 return NULL;
3077         }
3078
3079         return ctx;
3080 }
3081
3082 /****************************************************************************
3083 handle a -L query
3084 ****************************************************************************/
3085 static int do_host_query(struct loadparm_context *lp_ctx,
3086                          struct event_context *ev_ctx,
3087                          const char *query_host,
3088                          const char *workgroup)
3089 {
3090         browse_host(lp_ctx, ev_ctx, query_host);
3091         list_servers(workgroup);
3092         return(0);
3093 }
3094
3095
3096 /****************************************************************************
3097 handle a message operation
3098 ****************************************************************************/
3099 static int do_message_op(const char *netbios_name, const char *desthost,
3100                          const char **destports, const char *destip,
3101                          int name_type,
3102                          struct event_context *ev_ctx,
3103                          struct resolve_context *resolve_ctx,
3104                          struct smbcli_options *options,
3105                          struct smb_iconv_convenience *iconv_convenience,
3106              const char *socket_options)
3107 {
3108         struct nbt_name called, calling;
3109         const char *server_name;
3110         struct smbcli_state *cli;
3111
3112         make_nbt_name_client(&calling, netbios_name);
3113
3114         nbt_choose_called_name(NULL, &called, desthost, name_type);
3115
3116         server_name = destip ? destip : desthost;
3117
3118         if (!(cli = smbcli_state_init(NULL)) ||
3119             !smbcli_socket_connect(cli, server_name, destports,
3120                                    ev_ctx, resolve_ctx, options,
3121                                    iconv_convenience,
3122                    socket_options)) {
3123                 d_printf("Connection to %s failed\n", server_name);
3124                 return 1;
3125         }
3126
3127         if (!smbcli_transport_establish(cli, &calling, &called)) {
3128                 d_printf("session request failed\n");
3129                 talloc_free(cli);
3130                 return 1;
3131         }
3132
3133         send_message(cli, desthost);
3134         talloc_free(cli);
3135
3136         return 0;
3137 }
3138
3139
3140 /****************************************************************************
3141   main program
3142 ****************************************************************************/
3143  int main(int argc,char *argv[])
3144 {
3145         const char *base_directory = NULL;
3146         const char *dest_ip = NULL;
3147         int opt;
3148         const char *query_host = NULL;
3149         bool message = false;
3150         const char *desthost = NULL;
3151         poptContext pc;
3152         const char *service = NULL;
3153         int port = 0;
3154         char *p;
3155         int rc = 0;
3156         int name_type = 0x20;
3157         TALLOC_CTX *mem_ctx;
3158         struct event_context *ev_ctx;
3159         struct smbclient_context *ctx;
3160         const char *cmdstr = NULL;
3161         struct smbcli_options smb_options;
3162         struct smbcli_session_options smb_session_options;
3163
3164         struct poptOption long_options[] = {
3165                 POPT_AUTOHELP
3166
3167                 { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" },
3168                 { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" },
3169                 { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" },
3170                 { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" },
3171                 { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" },
3172                 { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" }, 
3173                 { "send-buffer", 'b', POPT_ARG_INT, NULL, 'b', "Changes the transmit/send buffer", "BYTES" },
3174                 { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" },
3175                 POPT_COMMON_SAMBA
3176                 POPT_COMMON_CONNECTION
3177                 POPT_COMMON_CREDENTIALS
3178                 POPT_COMMON_VERSION
3179                 { NULL }
3180         };
3181         
3182         mem_ctx = talloc_init("client.c/main");
3183         if (!mem_ctx) {
3184                 d_printf("\nclient.c: Not enough memory\n");
3185                 exit(1);
3186         }
3187
3188         ctx = talloc(mem_ctx, struct smbclient_context);
3189         ctx->io_bufsize = 64512;
3190
3191         pc = poptGetContext("smbclient", argc, (const char **) argv, long_options, 0);
3192         poptSetOtherOptionHelp(pc, "[OPTIONS] service <password>");
3193
3194         while ((opt = poptGetNextOpt(pc)) != -1) {
3195                 switch (opt) {
3196                 case 'M':
3197                         /* Messages are sent to NetBIOS name type 0x3
3198                          * (Messenger Service).  Make sure we default
3199                          * to port 139 instead of port 445. srl,crh
3200                          */
3201                         name_type = 0x03; 
3202                         desthost = strdup(poptGetOptArg(pc));
3203                         if( 0 == port ) port = 139;
3204                         message = true;
3205                         break;
3206                 case 'I':
3207                         dest_ip = poptGetOptArg(pc);
3208                         break;
3209                 case 'L':
3210                         query_host = strdup(poptGetOptArg(pc));
3211                         break;
3212                 case 'D':
3213                         base_directory = strdup(poptGetOptArg(pc));
3214                         break;
3215                 case 'b':
3216                         ctx->io_bufsize = MAX(1, atoi(poptGetOptArg(pc)));
3217                         break;
3218                 }
3219         }
3220
3221         gensec_init(cmdline_lp_ctx);
3222
3223         if(poptPeekArg(pc)) {
3224                 char *s = strdup(poptGetArg(pc)); 
3225
3226                 /* Convert any '/' characters in the service name to '\' characters */
3227                 string_replace(s, '/','\\');
3228
3229                 service = s;
3230
3231                 if (count_chars(s,'\\') < 3) {
3232                         d_printf("\n%s: Not enough '\\' characters in service\n",s);
3233                         poptPrintUsage(pc, stderr, 0);
3234                         exit(1);
3235                 }
3236         }
3237
3238         if (poptPeekArg(pc)) { 
3239                 cli_credentials_set_password(cmdline_credentials, poptGetArg(pc), CRED_SPECIFIED);
3240         }
3241
3242         /*init_names(); */
3243
3244         if (!query_host && !service && !message) {
3245                 poptPrintUsage(pc, stderr, 0);
3246                 exit(1);
3247         }
3248
3249         poptFreeContext(pc);
3250
3251         lp_smbcli_options(cmdline_lp_ctx, &smb_options);
3252         lp_smbcli_session_options(cmdline_lp_ctx, &smb_session_options);
3253
3254         ev_ctx = s4_event_context_init(talloc_autofree_context());
3255
3256         DEBUG( 3, ( "Client started (version %s).\n", SAMBA_VERSION_STRING ) );
3257
3258         if (query_host && (p=strchr_m(query_host,'#'))) {
3259                 *p = 0;
3260                 p++;
3261                 sscanf(p, "%x", &name_type);
3262         }
3263   
3264         if (query_host) {
3265                 rc = do_host_query(cmdline_lp_ctx, ev_ctx, query_host,
3266                                    lp_workgroup(cmdline_lp_ctx));
3267                 return rc;
3268         }
3269
3270         if (message) {
3271                 rc = do_message_op(lp_netbios_name(cmdline_lp_ctx), desthost,
3272                                    lp_smb_ports(cmdline_lp_ctx), dest_ip,
3273                                    name_type, ev_ctx,
3274                                    lp_resolve_context(cmdline_lp_ctx),
3275                                    &smb_options, lp_iconv_convenience(cmdline_lp_ctx),
3276                    lp_socket_options(cmdline_lp_ctx));
3277                 return rc;
3278         }