Move the bang (!) command back to the bottom of the list, allowing smbclient
[ira/wip.git] / source3 / client / client.c
1 /* 
2    Unix SMB/Netbios implementation.
3    Version 3.0.
4    SMB client
5    Copyright (C) Andrew Tridgell 1994-1998
6    Copyright (C) Simo Sorce 2001
7    
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #define NO_SYSLOG
24
25 #include "includes.h"
26
27 #ifndef REGISTER
28 #define REGISTER 0
29 #endif
30
31 const char prog_name[] = "smbclient";
32
33 struct cli_state *cli;
34 extern BOOL in_client;
35 extern BOOL AllowDebugChange;
36 static int port = 0;
37 pstring cur_dir = "\\";
38 pstring cd_path = "";
39 static pstring service;
40 static pstring desthost;
41 extern pstring global_myname;
42 static pstring password;
43 static pstring username;
44 static pstring workgroup;
45 static char *cmdstr;
46 static BOOL got_pass;
47 static int io_bufsize = 64512;
48 static BOOL use_kerberos;
49
50 static int name_type = 0x20;
51 static int max_protocol = PROTOCOL_NT1;
52 extern pstring user_socket_options;
53
54 static int process_tok(fstring tok);
55 static int cmd_help(void);
56
57 /* 30 second timeout on most commands */
58 #define CLIENT_TIMEOUT (30*1000)
59 #define SHORT_TIMEOUT (5*1000)
60
61 /* value for unused fid field in trans2 secondary request */
62 #define FID_UNUSED (0xFFFF)
63
64 time_t newer_than = 0;
65 int archive_level = 0;
66
67 BOOL translation = False;
68
69 static BOOL have_ip;
70
71 /* clitar bits insert */
72 extern int blocksize;
73 extern BOOL tar_inc;
74 extern BOOL tar_reset;
75 /* clitar bits end */
76  
77
78 mode_t myumask = 0755;
79
80 BOOL prompt = True;
81
82 int printmode = 1;
83
84 static BOOL recurse = False;
85 BOOL lowercase = False;
86
87 struct in_addr dest_ip;
88
89 #define SEPARATORS " \t\n\r"
90
91 BOOL abort_mget = True;
92
93 pstring fileselection = "";
94
95 extern file_info def_finfo;
96
97 /* timing globals */
98 int get_total_size = 0;
99 int get_total_time_ms = 0;
100 int put_total_size = 0;
101 int put_total_time_ms = 0;
102
103 /* totals globals */
104 static double dir_total;
105
106 #define USENMB
107
108 /****************************************************************************
109 write to a local file with CR/LF->LF translation if appropriate. return the 
110 number taken from the buffer. This may not equal the number written.
111 ****************************************************************************/
112 static int writefile(int f, char *b, int n)
113 {
114         int i;
115
116         if (!translation) {
117                 return write(f,b,n);
118         }
119
120         i = 0;
121         while (i < n) {
122                 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
123                         b++;i++;
124                 }
125                 if (write(f, b, 1) != 1) {
126                         break;
127                 }
128                 b++;
129                 i++;
130         }
131   
132         return(i);
133 }
134
135 /****************************************************************************
136   read from a file with LF->CR/LF translation if appropriate. return the 
137   number read. read approx n bytes.
138 ****************************************************************************/
139 static int readfile(char *b, int n, XFILE *f)
140 {
141         int i;
142         int c;
143
144         if (!translation)
145                 return x_fread(b,1,n,f);
146   
147         i = 0;
148         while (i < (n - 1) && (i < BUFFER_SIZE)) {
149                 if ((c = x_getc(f)) == EOF) {
150                         break;
151                 }
152       
153                 if (c == '\n') { /* change all LFs to CR/LF */
154                         b[i++] = '\r';
155                 }
156       
157                 b[i++] = c;
158         }
159   
160         return(i);
161 }
162  
163
164 /****************************************************************************
165 send a message
166 ****************************************************************************/
167 static void send_message(void)
168 {
169         int total_len = 0;
170         int grp_id;
171
172         if (!cli_message_start(cli, desthost, username, &grp_id)) {
173                 d_printf("message start: %s\n", cli_errstr(cli));
174                 return;
175         }
176
177
178         d_printf("Connected. Type your message, ending it with a Control-D\n");
179
180         while (!feof(stdin) && total_len < 1600) {
181                 int maxlen = MIN(1600 - total_len,127);
182                 pstring msg;
183                 int l=0;
184                 int c;
185
186                 ZERO_ARRAY(msg);
187
188                 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
189                         if (c == '\n')
190                                 msg[l++] = '\r';
191                         msg[l] = c;   
192                 }
193
194                 if (!cli_message_text(cli, msg, l, grp_id)) {
195                         d_printf("SMBsendtxt failed (%s)\n",cli_errstr(cli));
196                         return;
197                 }      
198                 
199                 total_len += l;
200         }
201
202         if (total_len >= 1600)
203                 d_printf("the message was truncated to 1600 bytes\n");
204         else
205                 d_printf("sent %d bytes\n",total_len);
206
207         if (!cli_message_end(cli, grp_id)) {
208                 d_printf("SMBsendend failed (%s)\n",cli_errstr(cli));
209                 return;
210         }      
211 }
212
213
214
215 /****************************************************************************
216 check the space on a device
217 ****************************************************************************/
218 static int do_dskattr(void)
219 {
220         int total, bsize, avail;
221
222         if (!cli_dskattr(cli, &bsize, &total, &avail)) {
223                 d_printf("Error in dskattr: %s\n",cli_errstr(cli)); 
224                 return 1;
225         }
226
227         d_printf("\n\t\t%d blocks of size %d. %d blocks available\n",
228                  total, bsize, avail);
229
230         return 0;
231 }
232
233 /****************************************************************************
234 show cd/pwd
235 ****************************************************************************/
236 static int cmd_pwd(void)
237 {
238         d_printf("Current directory is %s",service);
239         d_printf("%s\n",cur_dir);
240         return 0;
241 }
242
243
244 /****************************************************************************
245 change directory - inner section
246 ****************************************************************************/
247 static int do_cd(char *newdir)
248 {
249         char *p = newdir;
250         pstring saved_dir;
251         pstring dname;
252       
253         dos_format(newdir);
254
255         /* Save the current directory in case the
256            new directory is invalid */
257         pstrcpy(saved_dir, cur_dir);
258         if (*p == '\\')
259                 pstrcpy(cur_dir,p);
260         else
261                 pstrcat(cur_dir,p);
262         if (*(cur_dir+strlen(cur_dir)-1) != '\\') {
263                 pstrcat(cur_dir, "\\");
264         }
265         dos_clean_name(cur_dir);
266         pstrcpy(dname,cur_dir);
267         pstrcat(cur_dir,"\\");
268         dos_clean_name(cur_dir);
269         
270         if (!strequal(cur_dir,"\\")) {
271                 if (!cli_chkpath(cli, dname)) {
272                         d_printf("cd %s: %s\n", dname, cli_errstr(cli));
273                         pstrcpy(cur_dir,saved_dir);
274                 }
275         }
276         
277         pstrcpy(cd_path,cur_dir);
278
279         return 0;
280 }
281
282 /****************************************************************************
283 change directory
284 ****************************************************************************/
285 static int cmd_cd(void)
286 {
287         fstring buf;
288         int rc = 0;
289
290         if (next_token_nr(NULL,buf,NULL,sizeof(buf)))
291                 rc = do_cd(buf);
292         else
293                 d_printf("Current directory is %s\n",cur_dir);
294
295         return rc;
296 }
297
298
299 /*******************************************************************
300   decide if a file should be operated on
301   ********************************************************************/
302 static BOOL do_this_one(file_info *finfo)
303 {
304         if (finfo->mode & aDIR) return(True);
305
306         if (*fileselection && 
307             !mask_match(finfo->name,fileselection,False)) {
308                 DEBUG(3,("match_match %s failed\n", finfo->name));
309                 return False;
310         }
311
312         if (newer_than && finfo->mtime < newer_than) {
313                 DEBUG(3,("newer_than %s failed\n", finfo->name));
314                 return(False);
315         }
316
317         if ((archive_level==1 || archive_level==2) && !(finfo->mode & aARCH)) {
318                 DEBUG(3,("archive %s failed\n", finfo->name));
319                 return(False);
320         }
321         
322         return(True);
323 }
324
325 /****************************************************************************
326   display info about a file
327   ****************************************************************************/
328 static void display_finfo(file_info *finfo)
329 {
330         if (do_this_one(finfo)) {
331                 time_t t = finfo->mtime; /* the time is assumed to be passed as GMT */
332                 d_printf("  %-30s%7.7s %8.0f  %s",
333                          finfo->name,
334                          attrib_string(finfo->mode),
335                          (double)finfo->size,
336                          asctime(LocalTime(&t)));
337                 dir_total += finfo->size;
338         }
339 }
340
341
342 /****************************************************************************
343    accumulate size of a file
344   ****************************************************************************/
345 static void do_du(file_info *finfo)
346 {
347         if (do_this_one(finfo)) {
348                 dir_total += finfo->size;
349         }
350 }
351
352 static BOOL do_list_recurse;
353 static BOOL do_list_dirs;
354 static char *do_list_queue = 0;
355 static long do_list_queue_size = 0;
356 static long do_list_queue_start = 0;
357 static long do_list_queue_end = 0;
358 static void (*do_list_fn)(file_info *);
359
360 /****************************************************************************
361 functions for do_list_queue
362   ****************************************************************************/
363
364 /*
365  * The do_list_queue is a NUL-separated list of strings stored in a
366  * char*.  Since this is a FIFO, we keep track of the beginning and
367  * ending locations of the data in the queue.  When we overflow, we
368  * double the size of the char*.  When the start of the data passes
369  * the midpoint, we move everything back.  This is logically more
370  * complex than a linked list, but easier from a memory management
371  * angle.  In any memory error condition, do_list_queue is reset.
372  * Functions check to ensure that do_list_queue is non-NULL before
373  * accessing it.
374  */
375 static void reset_do_list_queue(void)
376 {
377         SAFE_FREE(do_list_queue);
378         do_list_queue_size = 0;
379         do_list_queue_start = 0;
380         do_list_queue_end = 0;
381 }
382
383 static void init_do_list_queue(void)
384 {
385         reset_do_list_queue();
386         do_list_queue_size = 1024;
387         do_list_queue = malloc(do_list_queue_size);
388         if (do_list_queue == 0) { 
389                 d_printf("malloc fail for size %d\n",
390                          (int)do_list_queue_size);
391                 reset_do_list_queue();
392         } else {
393                 memset(do_list_queue, 0, do_list_queue_size);
394         }
395 }
396
397 static void adjust_do_list_queue(void)
398 {
399         /*
400          * If the starting point of the queue is more than half way through,
401          * move everything toward the beginning.
402          */
403         if (do_list_queue && (do_list_queue_start == do_list_queue_end))
404         {
405                 DEBUG(4,("do_list_queue is empty\n"));
406                 do_list_queue_start = do_list_queue_end = 0;
407                 *do_list_queue = '\0';
408         }
409         else if (do_list_queue_start > (do_list_queue_size / 2))
410         {
411                 DEBUG(4,("sliding do_list_queue backward\n"));
412                 memmove(do_list_queue,
413                         do_list_queue + do_list_queue_start,
414                         do_list_queue_end - do_list_queue_start);
415                 do_list_queue_end -= do_list_queue_start;
416                 do_list_queue_start = 0;
417         }
418            
419 }
420
421 static void add_to_do_list_queue(const char* entry)
422 {
423         char *dlq;
424         long new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
425         while (new_end > do_list_queue_size)
426         {
427                 do_list_queue_size *= 2;
428                 DEBUG(4,("enlarging do_list_queue to %d\n",
429                          (int)do_list_queue_size));
430                 dlq = Realloc(do_list_queue, do_list_queue_size);
431                 if (! dlq) {
432                         d_printf("failure enlarging do_list_queue to %d bytes\n",
433                                  (int)do_list_queue_size);
434                         reset_do_list_queue();
435                 }
436                 else
437                 {
438                         do_list_queue = dlq;
439                         memset(do_list_queue + do_list_queue_size / 2,
440                                0, do_list_queue_size / 2);
441                 }
442         }
443         if (do_list_queue)
444         {
445                 pstrcpy(do_list_queue + do_list_queue_end, entry);
446                 do_list_queue_end = new_end;
447                 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
448                          entry, (int)do_list_queue_start, (int)do_list_queue_end));
449         }
450 }
451
452 static char *do_list_queue_head(void)
453 {
454         return do_list_queue + do_list_queue_start;
455 }
456
457 static void remove_do_list_queue_head(void)
458 {
459         if (do_list_queue_end > do_list_queue_start)
460         {
461                 do_list_queue_start += strlen(do_list_queue_head()) + 1;
462                 adjust_do_list_queue();
463                 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
464                          (int)do_list_queue_start, (int)do_list_queue_end));
465         }
466 }
467
468 static int do_list_queue_empty(void)
469 {
470         return (! (do_list_queue && *do_list_queue));
471 }
472
473 /****************************************************************************
474 a helper for do_list
475   ****************************************************************************/
476 static void do_list_helper(file_info *f, const char *mask, void *state)
477 {
478         if (f->mode & aDIR) {
479                 if (do_list_dirs && do_this_one(f)) {
480                         do_list_fn(f);
481                 }
482                 if (do_list_recurse && 
483                     !strequal(f->name,".") && 
484                     !strequal(f->name,"..")) {
485                         pstring mask2;
486                         char *p;
487
488                         pstrcpy(mask2, mask);
489                         p = strrchr_m(mask2,'\\');
490                         if (!p) return;
491                         p[1] = 0;
492                         pstrcat(mask2, f->name);
493                         pstrcat(mask2,"\\*");
494                         add_to_do_list_queue(mask2);
495                 }
496                 return;
497         }
498
499         if (do_this_one(f)) {
500                 do_list_fn(f);
501         }
502 }
503
504
505 /****************************************************************************
506 a wrapper around cli_list that adds recursion
507   ****************************************************************************/
508 void do_list(const char *mask,uint16 attribute,void (*fn)(file_info *),BOOL rec, BOOL dirs)
509 {
510         static int in_do_list = 0;
511
512         if (in_do_list && rec)
513         {
514                 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
515                 exit(1);
516         }
517
518         in_do_list = 1;
519
520         do_list_recurse = rec;
521         do_list_dirs = dirs;
522         do_list_fn = fn;
523
524         if (rec)
525         {
526                 init_do_list_queue();
527                 add_to_do_list_queue(mask);
528                 
529                 while (! do_list_queue_empty())
530                 {
531                         /*
532                          * Need to copy head so that it doesn't become
533                          * invalid inside the call to cli_list.  This
534                          * would happen if the list were expanded
535                          * during the call.
536                          * Fix from E. Jay Berkenbilt (ejb@ql.org)
537                          */
538                         pstring head;
539                         pstrcpy(head, do_list_queue_head());
540                         cli_list(cli, head, attribute, do_list_helper, NULL);
541                         remove_do_list_queue_head();
542                         if ((! do_list_queue_empty()) && (fn == display_finfo))
543                         {
544                                 char* next_file = do_list_queue_head();
545                                 char* save_ch = 0;
546                                 if ((strlen(next_file) >= 2) &&
547                                     (next_file[strlen(next_file) - 1] == '*') &&
548                                     (next_file[strlen(next_file) - 2] == '\\'))
549                                 {
550                                         save_ch = next_file +
551                                                 strlen(next_file) - 2;
552                                         *save_ch = '\0';
553                                 }
554                                 d_printf("\n%s\n",next_file);
555                                 if (save_ch)
556                                 {
557                                         *save_ch = '\\';
558                                 }
559                         }
560                 }
561         }
562         else
563         {
564                 if (cli_list(cli, mask, attribute, do_list_helper, NULL) == -1)
565                 {
566                         d_printf("%s listing %s\n", cli_errstr(cli), mask);
567                 }
568         }
569
570         in_do_list = 0;
571         reset_do_list_queue();
572 }
573
574 /****************************************************************************
575   get a directory listing
576   ****************************************************************************/
577 static int cmd_dir(void)
578 {
579         uint16 attribute = aDIR | aSYSTEM | aHIDDEN;
580         pstring mask;
581         fstring buf;
582         char *p=buf;
583         int rc;
584         
585         dir_total = 0;
586         pstrcpy(mask,cur_dir);
587         if(mask[strlen(mask)-1]!='\\')
588                 pstrcat(mask,"\\");
589         
590         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
591                 dos_format(p);
592                 if (*p == '\\')
593                         pstrcpy(mask,p);
594                 else
595                         pstrcat(mask,p);
596         }
597         else {
598                 pstrcat(mask,"*");
599         }
600
601         do_list(mask, attribute, display_finfo, recurse, True);
602
603         rc = do_dskattr();
604
605         DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
606
607         return rc;
608 }
609
610
611 /****************************************************************************
612   get a directory listing
613   ****************************************************************************/
614 static int cmd_du(void)
615 {
616         uint16 attribute = aDIR | aSYSTEM | aHIDDEN;
617         pstring mask;
618         fstring buf;
619         char *p=buf;
620         int rc;
621         
622         dir_total = 0;
623         pstrcpy(mask,cur_dir);
624         if(mask[strlen(mask)-1]!='\\')
625                 pstrcat(mask,"\\");
626         
627         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
628                 dos_format(p);
629                 if (*p == '\\')
630                         pstrcpy(mask,p);
631                 else
632                         pstrcat(mask,p);
633         } else {
634                 pstrcat(mask,"*");
635         }
636
637         do_list(mask, attribute, do_du, recurse, True);
638
639         rc = do_dskattr();
640
641         d_printf("Total number of bytes: %.0f\n", dir_total);
642
643         return rc;
644 }
645
646
647 /****************************************************************************
648   get a file from rname to lname
649   ****************************************************************************/
650 static int do_get(char *rname,char *lname)
651 {  
652         int handle=0,fnum;
653         BOOL newhandle = False;
654         char *data;
655         struct timeval tp_start;
656         int read_size = io_bufsize;
657         uint16 attr;
658         size_t size;
659         off_t nread = 0;
660         int rc = 0;
661
662         GetTimeOfDay(&tp_start);
663
664         if (lowercase) {
665                 strlower(lname);
666         }
667
668         fnum = cli_open(cli, rname, O_RDONLY, DENY_NONE);
669
670         if (fnum == -1) {
671                 d_printf("%s opening remote file %s\n",cli_errstr(cli),rname);
672                 return 1;
673         }
674
675         if(!strcmp(lname,"-")) {
676                 handle = fileno(stdout);
677         } else {
678                 handle = sys_open(lname,O_WRONLY|O_CREAT|O_TRUNC,0644);
679                 newhandle = True;
680         }
681         if (handle < 0) {
682                 d_printf("Error opening local file %s\n",lname);
683                 return 1;
684         }
685
686
687         if (!cli_qfileinfo(cli, fnum, 
688                            &attr, &size, NULL, NULL, NULL, NULL, NULL) &&
689             !cli_getattrE(cli, fnum, 
690                           &attr, &size, NULL, NULL, NULL)) {
691                 d_printf("getattrib: %s\n",cli_errstr(cli));
692                 return 1;
693         }
694
695         DEBUG(2,("getting file %s of size %.0f as %s ", 
696                  lname, (double)size, lname));
697
698         if(!(data = (char *)malloc(read_size))) { 
699                 d_printf("malloc fail for size %d\n", read_size);
700                 cli_close(cli, fnum);
701                 return 1;
702         }
703
704         while (1) {
705                 int n = cli_read(cli, fnum, data, nread, read_size);
706
707                 if (n <= 0) break;
708  
709                 if (writefile(handle,data, n) != n) {
710                         d_printf("Error writing local file\n");
711                         rc = 1;
712                         break;
713                 }
714       
715                 nread += n;
716         }
717
718         if (nread < size) {
719                 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
720                             rname, (long)nread));
721
722                 rc = 1;
723         }
724
725         SAFE_FREE(data);
726         
727         if (!cli_close(cli, fnum)) {
728                 d_printf("Error %s closing remote file\n",cli_errstr(cli));
729                 rc = 1;
730         }
731
732         if (newhandle) {
733                 close(handle);
734         }
735
736         if (archive_level >= 2 && (attr & aARCH)) {
737                 cli_setatr(cli, rname, attr & ~(uint16)aARCH, 0);
738         }
739
740         {
741                 struct timeval tp_end;
742                 int this_time;
743                 
744                 GetTimeOfDay(&tp_end);
745                 this_time = 
746                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
747                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
748                 get_total_time_ms += this_time;
749                 get_total_size += nread;
750                 
751                 DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n",
752                          nread / (1.024*this_time + 1.0e-4),
753                          get_total_size / (1.024*get_total_time_ms)));
754         }
755         
756         return rc;
757 }
758
759
760 /****************************************************************************
761   get a file
762   ****************************************************************************/
763 static int cmd_get(void)
764 {
765         pstring lname;
766         pstring rname;
767         char *p;
768
769         pstrcpy(rname,cur_dir);
770         pstrcat(rname,"\\");
771         
772         p = rname + strlen(rname);
773         
774         if (!next_token_nr(NULL,p,NULL,sizeof(rname)-strlen(rname))) {
775                 d_printf("get <filename>\n");
776                 return 1;
777         }
778         pstrcpy(lname,p);
779         dos_clean_name(rname);
780         
781         next_token_nr(NULL,lname,NULL,sizeof(lname));
782         
783         return do_get(rname, lname);
784 }
785
786
787 /****************************************************************************
788   do a mget operation on one file
789   ****************************************************************************/
790 static void do_mget(file_info *finfo)
791 {
792         pstring rname;
793         pstring quest;
794         pstring saved_curdir;
795         pstring mget_mask;
796
797         if (strequal(finfo->name,".") || strequal(finfo->name,".."))
798                 return;
799
800         if (abort_mget) {
801                 d_printf("mget aborted\n");
802                 return;
803         }
804
805         if (finfo->mode & aDIR)
806                 slprintf(quest,sizeof(pstring)-1,
807                          "Get directory %s? ",finfo->name);
808         else
809                 slprintf(quest,sizeof(pstring)-1,
810                          "Get file %s? ",finfo->name);
811
812         if (prompt && !yesno(quest)) return;
813
814         if (!(finfo->mode & aDIR)) {
815                 pstrcpy(rname,cur_dir);
816                 pstrcat(rname,finfo->name);
817                 do_get(rname,finfo->name);
818                 return;
819         }
820
821         /* handle directories */
822         pstrcpy(saved_curdir,cur_dir);
823
824         pstrcat(cur_dir,finfo->name);
825         pstrcat(cur_dir,"\\");
826
827         unix_format(finfo->name);
828         if (lowercase)
829                 strlower(finfo->name);
830         
831         if (!directory_exist(finfo->name,NULL) && 
832             mkdir(finfo->name,0777) != 0) {
833                 d_printf("failed to create directory %s\n",finfo->name);
834                 pstrcpy(cur_dir,saved_curdir);
835                 return;
836         }
837         
838         if (chdir(finfo->name) != 0) {
839                 d_printf("failed to chdir to directory %s\n",finfo->name);
840                 pstrcpy(cur_dir,saved_curdir);
841                 return;
842         }
843
844         pstrcpy(mget_mask,cur_dir);
845         pstrcat(mget_mask,"*");
846         
847         do_list(mget_mask, aSYSTEM | aHIDDEN | aDIR,do_mget,False, True);
848         chdir("..");
849         pstrcpy(cur_dir,saved_curdir);
850 }
851
852
853 /****************************************************************************
854 view the file using the pager
855 ****************************************************************************/
856 static int cmd_more(void)
857 {
858         fstring rname,lname,pager_cmd;
859         char *pager;
860         int fd;
861         int rc = 0;
862
863         fstrcpy(rname,cur_dir);
864         fstrcat(rname,"\\");
865         
866         slprintf(lname,sizeof(lname)-1, "%s/smbmore.XXXXXX",tmpdir());
867         fd = smb_mkstemp(lname);
868         if (fd == -1) {
869                 d_printf("failed to create temporary file for more\n");
870                 return 1;
871         }
872         close(fd);
873
874         if (!next_token_nr(NULL,rname+strlen(rname),NULL,sizeof(rname)-strlen(rname))) {
875                 d_printf("more <filename>\n");
876                 unlink(lname);
877                 return 1;
878         }
879         dos_clean_name(rname);
880
881         rc = do_get(rname,lname);
882
883         pager=getenv("PAGER");
884
885         slprintf(pager_cmd,sizeof(pager_cmd)-1,
886                  "%s %s",(pager? pager:PAGER), lname);
887         system(pager_cmd);
888         unlink(lname);
889         
890         return rc;
891 }
892
893
894
895 /****************************************************************************
896 do a mget command
897 ****************************************************************************/
898 static int cmd_mget(void)
899 {
900         uint16 attribute = aSYSTEM | aHIDDEN;
901         pstring mget_mask;
902         fstring buf;
903         char *p=buf;
904
905         *mget_mask = 0;
906
907         if (recurse)
908                 attribute |= aDIR;
909         
910         abort_mget = False;
911
912         while (next_token_nr(NULL,p,NULL,sizeof(buf))) {
913                 pstrcpy(mget_mask,cur_dir);
914                 if(mget_mask[strlen(mget_mask)-1]!='\\')
915                         pstrcat(mget_mask,"\\");
916                 
917                 if (*p == '\\')
918                         pstrcpy(mget_mask,p);
919                 else
920                         pstrcat(mget_mask,p);
921                 do_list(mget_mask, attribute,do_mget,False,True);
922         }
923
924         if (!*mget_mask) {
925                 pstrcpy(mget_mask,cur_dir);
926                 if(mget_mask[strlen(mget_mask)-1]!='\\')
927                         pstrcat(mget_mask,"\\");
928                 pstrcat(mget_mask,"*");
929                 do_list(mget_mask, attribute,do_mget,False,True);
930         }
931         
932         return 0;
933 }
934
935
936 /****************************************************************************
937 make a directory of name "name"
938 ****************************************************************************/
939 static BOOL do_mkdir(char *name)
940 {
941         if (!cli_mkdir(cli, name)) {
942                 d_printf("%s making remote directory %s\n",
943                          cli_errstr(cli),name);
944                 return(False);
945         }
946
947         return(True);
948 }
949
950 /****************************************************************************
951 show 8.3 name of a file
952 ****************************************************************************/
953 static BOOL do_altname(char *name)
954 {
955         fstring altname;
956         if (!NT_STATUS_IS_OK(cli_qpathinfo_alt_name(cli, name, altname))) {
957                 d_printf("%s getting alt name for %s\n",
958                          cli_errstr(cli),name);
959                 return(False);
960         }
961         d_printf("%s\n", altname);
962
963         return(True);
964 }
965
966
967 /****************************************************************************
968  Exit client.
969 ****************************************************************************/
970 static int cmd_quit(void)
971 {
972         cli_shutdown(cli);
973         exit(0);
974         /* NOTREACHED */
975         return 0;
976 }
977
978
979 /****************************************************************************
980   make a directory
981   ****************************************************************************/
982 static int cmd_mkdir(void)
983 {
984         pstring mask;
985         fstring buf;
986         char *p=buf;
987   
988         pstrcpy(mask,cur_dir);
989
990         if (!next_token_nr(NULL,p,NULL,sizeof(buf))) {
991                 if (!recurse)
992                         d_printf("mkdir <dirname>\n");
993                 return 1;
994         }
995         pstrcat(mask,p);
996
997         if (recurse) {
998                 pstring ddir;
999                 pstring ddir2;
1000                 *ddir2 = 0;
1001                 
1002                 pstrcpy(ddir,mask);
1003                 trim_string(ddir,".",NULL);
1004                 p = strtok(ddir,"/\\");
1005                 while (p) {
1006                         pstrcat(ddir2,p);
1007                         if (!cli_chkpath(cli, ddir2)) { 
1008                                 do_mkdir(ddir2);
1009                         }
1010                         pstrcat(ddir2,"\\");
1011                         p = strtok(NULL,"/\\");
1012                 }        
1013         } else {
1014                 do_mkdir(mask);
1015         }
1016         
1017         return 0;
1018 }
1019
1020
1021 /****************************************************************************
1022   show alt name
1023   ****************************************************************************/
1024 static int cmd_altname(void)
1025 {
1026         pstring name;
1027         fstring buf;
1028         char *p=buf;
1029   
1030         pstrcpy(name,cur_dir);
1031
1032         if (!next_token_nr(NULL,p,NULL,sizeof(buf))) {
1033                 d_printf("altname <file>\n");
1034                 return 1;
1035         }
1036         pstrcat(name,p);
1037
1038         do_altname(name);
1039
1040         return 0;
1041 }
1042
1043
1044 /****************************************************************************
1045   put a single file
1046   ****************************************************************************/
1047 static int do_put(char *rname,char *lname)
1048 {
1049         int fnum;
1050         XFILE *f;
1051         int nread=0;
1052         char *buf=NULL;
1053         int maxwrite=io_bufsize;
1054         int rc = 0;
1055         
1056         struct timeval tp_start;
1057         GetTimeOfDay(&tp_start);
1058
1059         fnum = cli_open(cli, rname, O_RDWR|O_CREAT|O_TRUNC, DENY_NONE);
1060   
1061         if (fnum == -1) {
1062                 d_printf("%s opening remote file %s\n",cli_errstr(cli),rname);
1063                 return 1;
1064         }
1065
1066         /* allow files to be piped into smbclient
1067            jdblair 24.jun.98 */
1068         if (!strcmp(lname, "-")) {
1069                 f = x_stdin;
1070                 /* size of file is not known */
1071         } else {
1072                 f = x_fopen(lname,O_RDONLY, 0);
1073         }
1074
1075         if (!f) {
1076                 d_printf("Error opening local file %s\n",lname);
1077                 return 1;
1078         }
1079
1080   
1081         DEBUG(1,("putting file %s as %s ",lname,
1082                  rname));
1083   
1084         buf = (char *)malloc(maxwrite);
1085         if (!buf) {
1086                 d_printf("ERROR: Not enough memory!\n");
1087                 return 1;
1088         }
1089         while (!x_feof(f)) {
1090                 int n = maxwrite;
1091                 int ret;
1092
1093                 if ((n = readfile(buf,n,f)) < 1) {
1094                         if((n == 0) && x_feof(f))
1095                                 break; /* Empty local file. */
1096
1097                         d_printf("Error reading local file: %s\n", strerror(errno));
1098                         rc = 1;
1099                         break;
1100                 }
1101
1102                 ret = cli_write(cli, fnum, 0, buf, nread, n);
1103
1104                 if (n != ret) {
1105                         d_printf("Error writing file: %s\n", cli_errstr(cli));
1106                         rc = 1;
1107                         break;
1108                 } 
1109
1110                 nread += n;
1111         }
1112
1113         if (!cli_close(cli, fnum)) {
1114                 d_printf("%s closing remote file %s\n",cli_errstr(cli),rname);
1115                 x_fclose(f);
1116                 SAFE_FREE(buf);
1117                 return 1;
1118         }
1119
1120         
1121         x_fclose(f);
1122         SAFE_FREE(buf);
1123
1124         {
1125                 struct timeval tp_end;
1126                 int this_time;
1127                 
1128                 GetTimeOfDay(&tp_end);
1129                 this_time = 
1130                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1131                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
1132                 put_total_time_ms += this_time;
1133                 put_total_size += nread;
1134                 
1135                 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1136                          nread / (1.024*this_time + 1.0e-4),
1137                          put_total_size / (1.024*put_total_time_ms)));
1138         }
1139
1140         if (f == x_stdin) {
1141                 cli_shutdown(cli);
1142                 exit(0);
1143         }
1144         
1145         return rc;
1146 }
1147
1148  
1149
1150 /****************************************************************************
1151   put a file
1152   ****************************************************************************/
1153 static int cmd_put(void)
1154 {
1155         pstring lname;
1156         pstring rname;
1157         fstring buf;
1158         char *p=buf;
1159         
1160         pstrcpy(rname,cur_dir);
1161         pstrcat(rname,"\\");
1162   
1163         if (!next_token_nr(NULL,p,NULL,sizeof(buf))) {
1164                 d_printf("put <filename>\n");
1165                 return 1;
1166         }
1167         pstrcpy(lname,p);
1168   
1169         if (next_token_nr(NULL,p,NULL,sizeof(buf)))
1170                 pstrcat(rname,p);      
1171         else
1172                 pstrcat(rname,lname);
1173         
1174         dos_clean_name(rname);
1175
1176         {
1177                 SMB_STRUCT_STAT st;
1178                 /* allow '-' to represent stdin
1179                    jdblair, 24.jun.98 */
1180                 if (!file_exist(lname,&st) &&
1181                     (strcmp(lname,"-"))) {
1182                         d_printf("%s does not exist\n",lname);
1183                         return 1;
1184                 }
1185         }
1186
1187         return do_put(rname,lname);
1188 }
1189
1190 /*************************************
1191   File list structure
1192 *************************************/
1193
1194 static struct file_list {
1195         struct file_list *prev, *next;
1196         char *file_path;
1197         BOOL isdir;
1198 } *file_list;
1199
1200 /****************************************************************************
1201   Free a file_list structure
1202 ****************************************************************************/
1203
1204 static void free_file_list (struct file_list * list)
1205 {
1206         struct file_list *tmp;
1207         
1208         while (list)
1209         {
1210                 tmp = list;
1211                 DLIST_REMOVE(list, list);
1212                 SAFE_FREE(tmp->file_path);
1213                 SAFE_FREE(tmp);
1214         }
1215 }
1216
1217 /****************************************************************************
1218   seek in a directory/file list until you get something that doesn't start with
1219   the specified name
1220   ****************************************************************************/
1221 static BOOL seek_list(struct file_list *list, char *name)
1222 {
1223         while (list) {
1224                 trim_string(list->file_path,"./","\n");
1225                 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1226                         return(True);
1227                 }
1228                 list = list->next;
1229         }
1230       
1231         return(False);
1232 }
1233
1234 /****************************************************************************
1235   set the file selection mask
1236   ****************************************************************************/
1237 static int cmd_select(void)
1238 {
1239         pstrcpy(fileselection,"");
1240         next_token_nr(NULL,fileselection,NULL,sizeof(fileselection));
1241
1242         return 0;
1243 }
1244
1245 /****************************************************************************
1246   Recursive file matching function act as find
1247   match must be always set to True when calling this function
1248 ****************************************************************************/
1249 static int file_find(struct file_list **list, const char *directory, 
1250                       const char *expression, BOOL match)
1251 {
1252         DIR *dir;
1253         struct file_list *entry;
1254         struct stat statbuf;
1255         int ret;
1256         char *path;
1257         BOOL isdir;
1258         char *dname;
1259
1260         dir = opendir(directory);
1261         if (!dir) return -1;
1262         
1263         while ((dname = readdirname(dir))) {
1264                 if (!strcmp("..", dname)) continue;
1265                 if (!strcmp(".", dname)) continue;
1266                 
1267                 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1268                         continue;
1269                 }
1270
1271                 isdir = False;
1272                 if (!match || !gen_fnmatch(expression, dname)) {
1273                         if (recurse) {
1274                                 ret = stat(path, &statbuf);
1275                                 if (ret == 0) {
1276                                         if (S_ISDIR(statbuf.st_mode)) {
1277                                                 isdir = True;
1278                                                 ret = file_find(list, path, expression, False);
1279                                         }
1280                                 } else {
1281                                         d_printf("file_find: cannot stat file %s\n", path);
1282                                 }
1283                                 
1284                                 if (ret == -1) {
1285                                         SAFE_FREE(path);
1286                                         closedir(dir);
1287                                         return -1;
1288                                 }
1289                         }
1290                         entry = (struct file_list *) malloc(sizeof (struct file_list));
1291                         if (!entry) {
1292                                 d_printf("Out of memory in file_find\n");
1293                                 closedir(dir);
1294                                 return -1;
1295                         }
1296                         entry->file_path = path;
1297                         entry->isdir = isdir;
1298                         DLIST_ADD(*list, entry);
1299                 } else {
1300                         SAFE_FREE(path);
1301                 }
1302         }
1303
1304         closedir(dir);
1305         return 0;
1306 }
1307
1308 /****************************************************************************
1309   mput some files
1310   ****************************************************************************/
1311 static int cmd_mput(void)
1312 {
1313         fstring buf;
1314         char *p=buf;
1315         
1316         while (next_token_nr(NULL,p,NULL,sizeof(buf))) {
1317                 int ret;
1318                 struct file_list *temp_list;
1319                 char *quest, *lname, *rname;
1320         
1321                 file_list = NULL;
1322
1323                 ret = file_find(&file_list, ".", p, True);
1324                 if (ret) {
1325                         free_file_list(file_list);
1326                         continue;
1327                 }
1328                 
1329                 quest = NULL;
1330                 lname = NULL;
1331                 rname = NULL;
1332                                 
1333                 for (temp_list = file_list; temp_list; 
1334                      temp_list = temp_list->next) {
1335
1336                         SAFE_FREE(lname);
1337                         if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1338                                 continue;
1339                         trim_string(lname, "./", "/");
1340                         
1341                         /* check if it's a directory */
1342                         if (temp_list->isdir) {
1343                                 /* if (!recurse) continue; */
1344                                 
1345                                 SAFE_FREE(quest);
1346                                 asprintf(&quest, "Put directory %s? ", lname);
1347                                 if (prompt && !yesno(quest)) { /* No */
1348                                         /* Skip the directory */
1349                                         lname[strlen(lname)-1] = '/';
1350                                         if (!seek_list(temp_list, lname))
1351                                                 break;              
1352                                 } else { /* Yes */
1353                                         SAFE_FREE(rname);
1354                                         asprintf(&rname, "%s%s", cur_dir, lname);
1355                                         dos_format(rname);
1356                                         if (!cli_chkpath(cli, rname) && 
1357                                             !do_mkdir(rname)) {
1358                                                 DEBUG (0, ("Unable to make dir, skipping..."));
1359                                                 /* Skip the directory */
1360                                                 lname[strlen(lname)-1] = '/';
1361                                                 if (!seek_list(temp_list, lname))
1362                                                         break;
1363                                         }
1364                                 }
1365                                 continue;
1366                         } else {
1367                                 SAFE_FREE(quest);
1368                                 asprintf(&quest,"Put file %s? ", lname);
1369                                 if (prompt && !yesno(quest)) /* No */
1370                                         continue;
1371                                 
1372                                 /* Yes */
1373                                 SAFE_FREE(rname);
1374                                 asprintf(&rname, "%s%s", cur_dir, lname);
1375                         }
1376
1377                         dos_format(rname);
1378
1379                         do_put(rname, lname);
1380                 }
1381                 free_file_list(file_list);
1382                 SAFE_FREE(quest);
1383                 SAFE_FREE(lname);
1384                 SAFE_FREE(rname);
1385         }
1386
1387         return 0;
1388 }
1389
1390
1391 /****************************************************************************
1392   cancel a print job
1393   ****************************************************************************/
1394 static int do_cancel(int job)
1395 {
1396         if (cli_printjob_del(cli, job)) {
1397                 d_printf("Job %d cancelled\n",job);
1398                 return 0;
1399         } else {
1400                 d_printf("Error cancelling job %d : %s\n",job,cli_errstr(cli));
1401                 return 1;
1402         }
1403 }
1404
1405
1406 /****************************************************************************
1407   cancel a print job
1408   ****************************************************************************/
1409 static int cmd_cancel(void)
1410 {
1411         fstring buf;
1412         int job; 
1413
1414         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1415                 d_printf("cancel <jobid> ...\n");
1416                 return 1;
1417         }
1418         do {
1419                 job = atoi(buf);
1420                 do_cancel(job);
1421         } while (next_token_nr(NULL,buf,NULL,sizeof(buf)));
1422         
1423         return 0;
1424 }
1425
1426
1427 /****************************************************************************
1428   print a file
1429   ****************************************************************************/
1430 static int cmd_print(void)
1431 {
1432         pstring lname;
1433         pstring rname;
1434         char *p;
1435
1436         if (!next_token_nr(NULL,lname,NULL, sizeof(lname))) {
1437                 d_printf("print <filename>\n");
1438                 return 1;
1439         }
1440
1441         pstrcpy(rname,lname);
1442         p = strrchr_m(rname,'/');
1443         if (p) {
1444                 slprintf(rname, sizeof(rname)-1, "%s-%d", p+1, (int)sys_getpid());
1445         }
1446
1447         if (strequal(lname,"-")) {
1448                 slprintf(rname, sizeof(rname)-1, "stdin-%d", (int)sys_getpid());
1449         }
1450
1451         return do_put(rname, lname);
1452 }
1453
1454
1455 /****************************************************************************
1456  show a print queue entry
1457 ****************************************************************************/
1458 static void queue_fn(struct print_job_info *p)
1459 {
1460         d_printf("%-6d   %-9d    %s\n", (int)p->id, (int)p->size, p->name);
1461 }
1462
1463 /****************************************************************************
1464  show a print queue
1465 ****************************************************************************/
1466 static int cmd_queue(void)
1467 {
1468         cli_print_queue(cli, queue_fn);
1469         
1470         return 0;
1471 }
1472
1473 /****************************************************************************
1474 delete some files
1475 ****************************************************************************/
1476 static void do_del(file_info *finfo)
1477 {
1478         pstring mask;
1479
1480         pstrcpy(mask,cur_dir);
1481         pstrcat(mask,finfo->name);
1482
1483         if (finfo->mode & aDIR) 
1484                 return;
1485
1486         if (!cli_unlink(cli, mask)) {
1487                 d_printf("%s deleting remote file %s\n",cli_errstr(cli),mask);
1488         }
1489 }
1490
1491 /****************************************************************************
1492 delete some files
1493 ****************************************************************************/
1494 static int cmd_del(void)
1495 {
1496         pstring mask;
1497         fstring buf;
1498         uint16 attribute = aSYSTEM | aHIDDEN;
1499
1500         if (recurse)
1501                 attribute |= aDIR;
1502         
1503         pstrcpy(mask,cur_dir);
1504         
1505         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1506                 d_printf("del <filename>\n");
1507                 return 1;
1508         }
1509         pstrcat(mask,buf);
1510
1511         do_list(mask, attribute,do_del,False,False);
1512         
1513         return 0;
1514 }
1515
1516 /****************************************************************************
1517 ****************************************************************************/
1518 static int cmd_open(void)
1519 {
1520         pstring mask;
1521         fstring buf;
1522         
1523         pstrcpy(mask,cur_dir);
1524         
1525         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1526                 d_printf("open <filename>\n");
1527                 return 1;
1528         }
1529         pstrcat(mask,buf);
1530
1531         cli_open(cli, mask, O_RDWR, DENY_ALL);
1532
1533         return 0;
1534 }
1535
1536
1537 /****************************************************************************
1538 remove a directory
1539 ****************************************************************************/
1540 static int cmd_rmdir(void)
1541 {
1542         pstring mask;
1543         fstring buf;
1544   
1545         pstrcpy(mask,cur_dir);
1546         
1547         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1548                 d_printf("rmdir <dirname>\n");
1549                 return 1;
1550         }
1551         pstrcat(mask,buf);
1552
1553         if (!cli_rmdir(cli, mask)) {
1554                 d_printf("%s removing remote directory file %s\n",
1555                          cli_errstr(cli),mask);
1556         }
1557         
1558         return 0;
1559 }
1560
1561 /****************************************************************************
1562  UNIX hardlink.
1563 ****************************************************************************/
1564
1565 static int cmd_link(void)
1566 {
1567         pstring src,dest;
1568         fstring buf,buf2;
1569   
1570         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1571                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1572                 return 1;
1573         }
1574
1575         pstrcpy(src,cur_dir);
1576         pstrcpy(dest,cur_dir);
1577   
1578         if (!next_token(NULL,buf,NULL,sizeof(buf)) || 
1579             !next_token(NULL,buf2,NULL, sizeof(buf2))) {
1580                 d_printf("link <src> <dest>\n");
1581                 return 1;
1582         }
1583
1584         pstrcat(src,buf);
1585         pstrcat(dest,buf2);
1586
1587         if (!cli_unix_hardlink(cli, src, dest)) {
1588                 d_printf("%s linking files (%s -> %s)\n", cli_errstr(cli), src, dest);
1589                 return 1;
1590         }  
1591
1592         return 0;
1593 }
1594
1595 /****************************************************************************
1596  UNIX symlink.
1597 ****************************************************************************/
1598
1599 static int cmd_symlink(void)
1600 {
1601         pstring src,dest;
1602         fstring buf,buf2;
1603   
1604         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1605                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1606                 return 1;
1607         }
1608
1609         pstrcpy(src,cur_dir);
1610         pstrcpy(dest,cur_dir);
1611         
1612         if (!next_token(NULL,buf,NULL,sizeof(buf)) || 
1613             !next_token(NULL,buf2,NULL, sizeof(buf2))) {
1614                 d_printf("symlink <src> <dest>\n");
1615                 return 1;
1616         }
1617
1618         pstrcat(src,buf);
1619         pstrcat(dest,buf2);
1620
1621         if (!cli_unix_symlink(cli, src, dest)) {
1622                 d_printf("%s symlinking files (%s -> %s)\n",
1623                         cli_errstr(cli), src, dest);
1624                 return 1;
1625         } 
1626
1627         return 0;
1628 }
1629
1630 /****************************************************************************
1631  UNIX chmod.
1632 ****************************************************************************/
1633
1634 static int cmd_chmod(void)
1635 {
1636         pstring src;
1637         mode_t mode;
1638         fstring buf, buf2;
1639   
1640         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1641                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1642                 return 1;
1643         }
1644
1645         pstrcpy(src,cur_dir);
1646         
1647         if (!next_token(NULL,buf,NULL,sizeof(buf)) || 
1648             !next_token(NULL,buf2,NULL, sizeof(buf2))) {
1649                 d_printf("chmod mode file\n");
1650                 return 1;
1651         }
1652
1653         mode = (mode_t)strtol(buf, NULL, 8);
1654         pstrcat(src,buf2);
1655
1656         if (!cli_unix_chmod(cli, src, mode)) {
1657                 d_printf("%s chmod file %s 0%o\n",
1658                         cli_errstr(cli), src, (unsigned int)mode);
1659                 return 1;
1660         } 
1661
1662         return 0;
1663 }
1664
1665 /****************************************************************************
1666  UNIX chown.
1667 ****************************************************************************/
1668
1669 static int cmd_chown(void)
1670 {
1671         pstring src;
1672         uid_t uid;
1673         gid_t gid;
1674         fstring buf, buf2, buf3;
1675   
1676         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1677                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1678                 return 1;
1679         }
1680
1681         pstrcpy(src,cur_dir);
1682         
1683         if (!next_token(NULL,buf,NULL,sizeof(buf)) || 
1684             !next_token(NULL,buf2,NULL, sizeof(buf2)) ||
1685             !next_token(NULL,buf3,NULL, sizeof(buf3))) {
1686                 d_printf("chown uid gid file\n");
1687                 return 1;
1688         }
1689
1690         uid = (uid_t)atoi(buf);
1691         gid = (gid_t)atoi(buf2);
1692         pstrcat(src,buf3);
1693
1694         if (!cli_unix_chown(cli, src, uid, gid)) {
1695                 d_printf("%s chown file %s uid=%d, gid=%d\n",
1696                         cli_errstr(cli), src, (int)uid, (int)gid);
1697                 return 1;
1698         } 
1699
1700         return 0;
1701 }
1702
1703 /****************************************************************************
1704 rename some files
1705 ****************************************************************************/
1706 static int cmd_rename(void)
1707 {
1708         pstring src,dest;
1709         fstring buf,buf2;
1710   
1711         pstrcpy(src,cur_dir);
1712         pstrcpy(dest,cur_dir);
1713         
1714         if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) || 
1715             !next_token_nr(NULL,buf2,NULL, sizeof(buf2))) {
1716                 d_printf("rename <src> <dest>\n");
1717                 return 1;
1718         }
1719
1720         pstrcat(src,buf);
1721         pstrcat(dest,buf2);
1722
1723         if (!cli_rename(cli, src, dest)) {
1724                 d_printf("%s renaming files\n",cli_errstr(cli));
1725                 return 1;
1726         }
1727         
1728         return 0;
1729 }
1730
1731
1732 /****************************************************************************
1733 toggle the prompt flag
1734 ****************************************************************************/
1735 static int cmd_prompt(void)
1736 {
1737         prompt = !prompt;
1738         DEBUG(2,("prompting is now %s\n",prompt?"on":"off"));
1739         
1740         return 1;
1741 }
1742
1743
1744 /****************************************************************************
1745 set the newer than time
1746 ****************************************************************************/
1747 static int cmd_newer(void)
1748 {
1749         fstring buf;
1750         BOOL ok;
1751         SMB_STRUCT_STAT sbuf;
1752
1753         ok = next_token_nr(NULL,buf,NULL,sizeof(buf));
1754         if (ok && (sys_stat(buf,&sbuf) == 0)) {
1755                 newer_than = sbuf.st_mtime;
1756                 DEBUG(1,("Getting files newer than %s",
1757                          asctime(LocalTime(&newer_than))));
1758         } else {
1759                 newer_than = 0;
1760         }
1761
1762         if (ok && newer_than == 0) {
1763                 d_printf("Error setting newer-than time\n");
1764                 return 1;
1765         }
1766
1767         return 0;
1768 }
1769
1770 /****************************************************************************
1771 set the archive level
1772 ****************************************************************************/
1773 static int cmd_archive(void)
1774 {
1775         fstring buf;
1776
1777         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1778                 archive_level = atoi(buf);
1779         } else
1780                 d_printf("Archive level is %d\n",archive_level);
1781
1782         return 0;
1783 }
1784
1785 /****************************************************************************
1786 toggle the lowercaseflag
1787 ****************************************************************************/
1788 static int cmd_lowercase(void)
1789 {
1790         lowercase = !lowercase;
1791         DEBUG(2,("filename lowercasing is now %s\n",lowercase?"on":"off"));
1792
1793         return 0;
1794 }
1795
1796
1797
1798
1799 /****************************************************************************
1800 toggle the recurse flag
1801 ****************************************************************************/
1802 static int cmd_recurse(void)
1803 {
1804         recurse = !recurse;
1805         DEBUG(2,("directory recursion is now %s\n",recurse?"on":"off"));
1806
1807         return 0;
1808 }
1809
1810 /****************************************************************************
1811 toggle the translate flag
1812 ****************************************************************************/
1813 static int cmd_translate(void)
1814 {
1815         translation = !translation;
1816         DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
1817                  translation?"on":"off"));
1818
1819         return 0;
1820 }
1821
1822
1823 /****************************************************************************
1824 do a printmode command
1825 ****************************************************************************/
1826 static int cmd_printmode(void)
1827 {
1828         fstring buf;
1829         fstring mode;
1830
1831         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1832                 if (strequal(buf,"text")) {
1833                         printmode = 0;      
1834                 } else {
1835                         if (strequal(buf,"graphics"))
1836                                 printmode = 1;
1837                         else
1838                                 printmode = atoi(buf);
1839                 }
1840         }
1841
1842         switch(printmode)
1843                 {
1844                 case 0: 
1845                         fstrcpy(mode,"text");
1846                         break;
1847                 case 1: 
1848                         fstrcpy(mode,"graphics");
1849                         break;
1850                 default: 
1851                         slprintf(mode,sizeof(mode)-1,"%d",printmode);
1852                         break;
1853                 }
1854         
1855         DEBUG(2,("the printmode is now %s\n",mode));
1856
1857         return 0;
1858 }
1859
1860 /****************************************************************************
1861 do the lcd command
1862 ****************************************************************************/
1863 static int cmd_lcd(void)
1864 {
1865         fstring buf;
1866         pstring d;
1867         
1868         if (next_token_nr(NULL,buf,NULL,sizeof(buf)))
1869                 chdir(buf);
1870         DEBUG(2,("the local directory is now %s\n",sys_getwd(d)));
1871
1872         return 0;
1873 }
1874
1875 /****************************************************************************
1876 list a share name
1877 ****************************************************************************/
1878 static void browse_fn(const char *name, uint32 m, 
1879                       const char *comment, void *state)
1880 {
1881         fstring typestr;
1882
1883         *typestr=0;
1884
1885         switch (m)
1886         {
1887           case STYPE_DISKTREE:
1888             fstrcpy(typestr,"Disk"); break;
1889           case STYPE_PRINTQ:
1890             fstrcpy(typestr,"Printer"); break;
1891           case STYPE_DEVICE:
1892             fstrcpy(typestr,"Device"); break;
1893           case STYPE_IPC:
1894             fstrcpy(typestr,"IPC"); break;
1895         }
1896         /* FIXME: If the remote machine returns non-ascii characters
1897            in any of these fields, they can corrupt the output.  We
1898            should remove them. */
1899         d_printf("\t%-15.15s%-10.10s%s\n",
1900                name,typestr,comment);
1901 }
1902
1903
1904 /****************************************************************************
1905 try and browse available connections on a host
1906 ****************************************************************************/
1907 static BOOL browse_host(BOOL sort)
1908 {
1909         int ret;
1910
1911         d_printf("\n\tSharename      Type      Comment\n");
1912         d_printf("\t---------      ----      -------\n");
1913
1914         if((ret = cli_RNetShareEnum(cli, browse_fn, NULL)) == -1)
1915                 d_printf("Error returning browse list: %s\n", cli_errstr(cli));
1916
1917         return (ret != -1);
1918 }
1919
1920 /****************************************************************************
1921 list a server name
1922 ****************************************************************************/
1923 static void server_fn(const char *name, uint32 m, 
1924                       const char *comment, void *state)
1925 {
1926         d_printf("\t%-16.16s     %s\n", name, comment);
1927 }
1928
1929 /****************************************************************************
1930 try and browse available connections on a host
1931 ****************************************************************************/
1932 static BOOL list_servers(char *wk_grp)
1933 {
1934         if (!cli->server_domain) return False;
1935         
1936         d_printf("\n\tServer               Comment\n");
1937         d_printf("\t---------            -------\n");
1938
1939         cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_ALL, server_fn, NULL);
1940
1941         d_printf("\n\tWorkgroup            Master\n");
1942         d_printf("\t---------            -------\n");
1943
1944         cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_DOMAIN_ENUM, server_fn, NULL);
1945         return True;
1946 }
1947
1948 /* Some constants for completing filename arguments */
1949
1950 #define COMPL_NONE        0          /* No completions */
1951 #define COMPL_REMOTE      1          /* Complete remote filename */
1952 #define COMPL_LOCAL       2          /* Complete local filename */
1953
1954 /* This defines the commands supported by this client.
1955  * NOTE: The "!" must be the last one in the list because it's fn pointer
1956  *       field is NULL, and NULL in that field is used in process_tok()
1957  *       (below) to indicate the end of the list.  crh
1958  */
1959 struct
1960 {
1961   char *name;
1962   int (*fn)(void);
1963   char *description;
1964   char compl_args[2];      /* Completion argument info */
1965 } commands[] = 
1966 {
1967   {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
1968   {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
1969   {"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}},
1970   {"blocksize",cmd_block,"blocksize <number> (default 20)",{COMPL_NONE,COMPL_NONE}},
1971   {"cancel",cmd_cancel,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
1972   {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
1973   {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
1974   {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
1975   {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
1976   {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
1977   {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
1978   {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
1979   {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
1980   {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
1981   {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
1982   {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
1983   {"link",cmd_link,"<src> <dest> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
1984   {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},  
1985   {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
1986   {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
1987   {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
1988   {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
1989   {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
1990   {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},  
1991   {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
1992   {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
1993   {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
1994   {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
1995   {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
1996   {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},  
1997   {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
1998   {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
1999   {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2000   {"queue",cmd_queue,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2001   {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2002   {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2003   {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},  
2004   {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2005   {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2006   {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2007   {"setmode",cmd_setmode,"filename <setmode string> change modes of file",{COMPL_REMOTE,COMPL_NONE}},
2008   {"symlink",cmd_symlink,"<src> <dest> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2009   {"tar",cmd_tar,"tar <c|x>[IXFqbgNan] current directory to/from <file name>",{COMPL_NONE,COMPL_NONE}},
2010   {"tarmode",cmd_tarmode,"<full|inc|reset|noreset> tar's behaviour towards archive bits",{COMPL_NONE,COMPL_NONE}},
2011   {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2012   
2013   /* Yes, this must be here, see crh's comment above. */
2014   {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2015   {"",NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2016 };
2017
2018
2019 /*******************************************************************
2020   lookup a command string in the list of commands, including 
2021   abbreviations
2022   ******************************************************************/
2023 static int process_tok(fstring tok)
2024 {
2025         int i = 0, matches = 0;
2026         int cmd=0;
2027         int tok_len = strlen(tok);
2028         
2029         while (commands[i].fn != NULL) {
2030                 if (strequal(commands[i].name,tok)) {
2031                         matches = 1;
2032                         cmd = i;
2033                         break;
2034                 } else if (strnequal(commands[i].name, tok, tok_len)) {
2035                         matches++;
2036                         cmd = i;
2037                 }
2038                 i++;
2039         }
2040   
2041         if (matches == 0)
2042                 return(-1);
2043         else if (matches == 1)
2044                 return(cmd);
2045         else
2046                 return(-2);
2047 }
2048
2049 /****************************************************************************
2050 help
2051 ****************************************************************************/
2052 static int cmd_help(void)
2053 {
2054         int i=0,j;
2055         fstring buf;
2056         
2057         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
2058                 if ((i = process_tok(buf)) >= 0)
2059                         d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2060         } else {
2061                 while (commands[i].description) {
2062                         for (j=0; commands[i].description && (j<5); j++) {
2063                                 d_printf("%-15s",commands[i].name);
2064                                 i++;
2065                         }
2066                         d_printf("\n");
2067                 }
2068         }
2069         return 0;
2070 }
2071
2072 /****************************************************************************
2073 process a -c command string
2074 ****************************************************************************/
2075 static int process_command_string(char *cmd)
2076 {
2077         pstring line;
2078         char *ptr;
2079         int rc = 0;
2080
2081         while (cmd[0] != '\0')    {
2082                 char *p;
2083                 fstring tok;
2084                 int i;
2085                 
2086                 if ((p = strchr_m(cmd, ';')) == 0) {
2087                         strncpy(line, cmd, 999);
2088                         line[1000] = '\0';
2089                         cmd += strlen(cmd);
2090                 } else {
2091                         if (p - cmd > 999) p = cmd + 999;
2092                         strncpy(line, cmd, p - cmd);
2093                         line[p - cmd] = '\0';
2094                         cmd = p + 1;
2095                 }
2096                 
2097                 /* and get the first part of the command */
2098                 ptr = line;
2099                 if (!next_token_nr(&ptr,tok,NULL,sizeof(tok))) continue;
2100                 
2101                 if ((i = process_tok(tok)) >= 0) {
2102                         rc = commands[i].fn();
2103                 } else if (i == -2) {
2104                         d_printf("%s: command abbreviation ambiguous\n",tok);
2105                 } else {
2106                         d_printf("%s: command not found\n",tok);
2107                 }
2108         }
2109         
2110         return rc;
2111 }       
2112
2113 /****************************************************************************
2114 handle completion of commands for readline
2115 ****************************************************************************/
2116 static char **completion_fn(char *text, int start, int end)
2117 {
2118 #define MAX_COMPLETIONS 100
2119         char **matches;
2120         int i, count=0;
2121
2122         /* for words not at the start of the line fallback to filename completion */
2123         if (start) return NULL;
2124
2125         matches = (char **)malloc(sizeof(matches[0])*MAX_COMPLETIONS);
2126         if (!matches) return NULL;
2127
2128         matches[count++] = strdup(text);
2129         if (!matches[0]) return NULL;
2130
2131         for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
2132                 if (strncmp(text, commands[i].name, strlen(text)) == 0) {
2133                         matches[count] = strdup(commands[i].name);
2134                         if (!matches[count]) return NULL;
2135                         count++;
2136                 }
2137         }
2138
2139         if (count == 2) {
2140                 SAFE_FREE(matches[0]);
2141                 matches[0] = strdup(matches[1]);
2142         }
2143         matches[count] = NULL;
2144         return matches;
2145 }
2146
2147
2148 /****************************************************************************
2149 make sure we swallow keepalives during idle time
2150 ****************************************************************************/
2151 static void readline_callback(void)
2152 {
2153         fd_set fds;
2154         struct timeval timeout;
2155         static time_t last_t;
2156         time_t t;
2157
2158         t = time(NULL);
2159
2160         if (t - last_t < 5) return;
2161
2162         last_t = t;
2163
2164  again:
2165         FD_ZERO(&fds);
2166         FD_SET(cli->fd,&fds);
2167
2168         timeout.tv_sec = 0;
2169         timeout.tv_usec = 0;
2170         sys_select_intr(cli->fd+1,&fds,&timeout);
2171                 
2172         /* We deliberately use receive_smb instead of
2173            client_receive_smb as we want to receive
2174            session keepalives and then drop them here.
2175         */
2176         if (FD_ISSET(cli->fd,&fds)) {
2177                 receive_smb(cli->fd,cli->inbuf,0);
2178                 goto again;
2179         }
2180       
2181         cli_chkpath(cli, "\\");
2182 }
2183
2184
2185 /****************************************************************************
2186 process commands on stdin
2187 ****************************************************************************/
2188 static void process_stdin(void)
2189 {
2190         char *ptr;
2191
2192         while (1) {
2193                 fstring tok;
2194                 fstring the_prompt;
2195                 char *cline;
2196                 pstring line;
2197                 int i;
2198                 
2199                 /* display a prompt */
2200                 slprintf(the_prompt, sizeof(the_prompt)-1, "smb: %s> ", cur_dir);
2201                 cline = smb_readline(the_prompt, readline_callback, completion_fn);
2202                         
2203                 if (!cline) break;
2204                 
2205                 pstrcpy(line, cline);
2206
2207                 /* special case - first char is ! */
2208                 if (*line == '!') {
2209                         system(line + 1);
2210                         continue;
2211                 }
2212       
2213                 /* and get the first part of the command */
2214                 ptr = line;
2215                 if (!next_token_nr(&ptr,tok,NULL,sizeof(tok))) continue;
2216
2217                 if ((i = process_tok(tok)) >= 0) {
2218                         commands[i].fn();
2219                 } else if (i == -2) {
2220                         d_printf("%s: command abbreviation ambiguous\n",tok);
2221                 } else {
2222                         d_printf("%s: command not found\n",tok);
2223                 }
2224         }
2225 }
2226
2227
2228 /***************************************************** 
2229 return a connection to a server
2230 *******************************************************/
2231 struct cli_state *do_connect(const char *server, const char *share)
2232 {
2233         struct cli_state *c;
2234         struct nmb_name called, calling;
2235         const char *server_n;
2236         struct in_addr ip;
2237         fstring servicename;
2238         char *sharename;
2239         
2240         /* make a copy so we don't modify the global string 'service' */
2241         safe_strcpy(servicename, share, sizeof(servicename)-1);
2242         sharename = servicename;
2243         if (*sharename == '\\') {
2244                 server = sharename+2;
2245                 sharename = strchr_m(server,'\\');
2246                 if (!sharename) return NULL;
2247                 *sharename = 0;
2248                 sharename++;
2249         }
2250
2251         server_n = server;
2252         
2253         zero_ip(&ip);
2254
2255         make_nmb_name(&calling, global_myname, 0x0);
2256         make_nmb_name(&called , server, name_type);
2257
2258  again:
2259         zero_ip(&ip);
2260         if (have_ip) ip = dest_ip;
2261
2262         /* have to open a new connection */
2263         if (!(c=cli_initialise(NULL)) || (cli_set_port(c, port) != port) ||
2264             !cli_connect(c, server_n, &ip)) {
2265                 d_printf("Connection to %s failed\n", server_n);
2266                 return NULL;
2267         }
2268
2269         c->protocol = max_protocol;
2270         c->use_kerberos = use_kerberos;
2271
2272         if (!cli_session_request(c, &calling, &called)) {
2273                 char *p;
2274                 d_printf("session request to %s failed (%s)\n", 
2275                          called.name, cli_errstr(c));
2276                 cli_shutdown(c);
2277                 if ((p=strchr_m(called.name, '.'))) {
2278                         *p = 0;
2279                         goto again;
2280                 }
2281                 if (strcmp(called.name, "*SMBSERVER")) {
2282                         make_nmb_name(&called , "*SMBSERVER", 0x20);
2283                         goto again;
2284                 }
2285                 return NULL;
2286         }
2287
2288         DEBUG(4,(" session request ok\n"));
2289
2290         if (!cli_negprot(c)) {
2291                 d_printf("protocol negotiation failed\n");
2292                 cli_shutdown(c);
2293                 return NULL;
2294         }
2295
2296         if (!got_pass) {
2297                 char *pass = getpass("Password: ");
2298                 if (pass) {
2299                         pstrcpy(password, pass);
2300                 }
2301         }
2302
2303         if (!cli_session_setup(c, username, 
2304                                password, strlen(password),
2305                                password, strlen(password),
2306                                workgroup)) {
2307                 /* if a password was not supplied then try again with a null username */
2308                 if (password[0] || !username[0] || use_kerberos ||
2309                     !cli_session_setup(c, "", "", 0, "", 0, workgroup)) { 
2310                         d_printf("session setup failed: %s\n", cli_errstr(c));
2311                         cli_shutdown(c);
2312                         return NULL;
2313                 }
2314                 d_printf("Anonymous login successful\n");
2315         }
2316
2317         if (*c->server_domain) {
2318                 DEBUG(1,("Domain=[%s] OS=[%s] Server=[%s]\n",
2319                         c->server_domain,c->server_os,c->server_type));
2320         } else if (*c->server_os || *c->server_type){
2321                 DEBUG(1,("OS=[%s] Server=[%s]\n",
2322                          c->server_os,c->server_type));
2323         }               
2324         
2325         DEBUG(4,(" session setup ok\n"));
2326
2327         if (!cli_send_tconX(c, sharename, "?????",
2328                             password, strlen(password)+1)) {
2329                 d_printf("tree connect failed: %s\n", cli_errstr(c));
2330                 cli_shutdown(c);
2331                 return NULL;
2332         }
2333
2334         DEBUG(4,(" tconx ok\n"));
2335
2336         return c;
2337 }
2338
2339
2340 /****************************************************************************
2341   process commands from the client
2342 ****************************************************************************/
2343 static int process(char *base_directory)
2344 {
2345         int rc = 0;
2346
2347         cli = do_connect(desthost, service);
2348         if (!cli) {
2349                 return 1;
2350         }
2351
2352         if (*base_directory) do_cd(base_directory);
2353         
2354         if (cmdstr) {
2355                 rc = process_command_string(cmdstr);
2356         } else {
2357                 process_stdin();
2358         }
2359   
2360         cli_shutdown(cli);
2361         return rc;
2362 }
2363
2364 /****************************************************************************
2365 usage on the program
2366 ****************************************************************************/
2367 static void usage(char *pname)
2368 {
2369   d_printf("Usage: %s service <password> [options]", pname);
2370
2371   d_printf("\nVersion %s\n",VERSION);
2372   d_printf("\t-s smb.conf           pathname to smb.conf file\n");
2373   d_printf("\t-O socket_options     socket options to use\n");
2374   d_printf("\t-R name resolve order use these name resolution services only\n");
2375   d_printf("\t-M host               send a winpopup message to the host\n");
2376   d_printf("\t-i scope              use this NetBIOS scope\n");
2377   d_printf("\t-N                    don't ask for a password\n");
2378   d_printf("\t-n netbios name.      Use this name as my netbios name\n");
2379   d_printf("\t-d debuglevel         set the debuglevel\n");
2380   d_printf("\t-P                    connect to service as a printer\n");
2381   d_printf("\t-p port               connect to the specified port\n");
2382   d_printf("\t-l log basename.      Basename for log/debug files\n");
2383   d_printf("\t-h                    Print this help message.\n");
2384   d_printf("\t-I dest IP            use this IP to connect to\n");
2385   d_printf("\t-E                    write messages to stderr instead of stdout\n");
2386   d_printf("\t-k                    use kerberos (active directory) authentication\n");
2387   d_printf("\t-U username           set the network username\n");
2388   d_printf("\t-L host               get a list of shares available on a host\n");
2389   d_printf("\t-t terminal code      terminal i/o code {sjis|euc|jis7|jis8|junet|hex}\n");
2390   d_printf("\t-m max protocol       set the max protocol level\n");
2391   d_printf("\t-A filename           get the credentials from a file\n");
2392   d_printf("\t-W workgroup          set the workgroup name\n");
2393   d_printf("\t-T<c|x>IXFqgbNan      command line tar\n");
2394   d_printf("\t-D directory          start from directory\n");
2395   d_printf("\t-c command string     execute semicolon separated commands\n");
2396   d_printf("\t-b xmit/send buffer   changes the transmit/send buffer (default: 65520)\n");
2397   d_printf("\n");
2398 }
2399
2400
2401 /****************************************************************************
2402 get a password from a a file or file descriptor
2403 exit on failure
2404 ****************************************************************************/
2405 static void get_password_file(void)
2406 {
2407         int fd = -1;
2408         char *p;
2409         BOOL close_it = False;
2410         pstring spec;
2411         char pass[128];
2412                 
2413         if ((p = getenv("PASSWD_FD")) != NULL) {
2414                 pstrcpy(spec, "descriptor ");
2415                 pstrcat(spec, p);
2416                 sscanf(p, "%d", &fd);
2417                 close_it = False;
2418         } else if ((p = getenv("PASSWD_FILE")) != NULL) {
2419                 fd = sys_open(p, O_RDONLY, 0);
2420                 pstrcpy(spec, p);
2421                 if (fd < 0) {
2422                         fprintf(stderr, "Error opening PASSWD_FILE %s: %s\n",
2423                                 spec, strerror(errno));
2424                         exit(1);
2425                 }
2426                 close_it = True;
2427         }
2428
2429         for(p = pass, *p = '\0'; /* ensure that pass is null-terminated */
2430             p && p - pass < sizeof(pass);) {
2431                 switch (read(fd, p, 1)) {
2432                 case 1:
2433                         if (*p != '\n' && *p != '\0') {
2434                                 *++p = '\0'; /* advance p, and null-terminate pass */
2435                                 break;
2436                         }
2437                 case 0:
2438                         if (p - pass) {
2439                                 *p = '\0'; /* null-terminate it, just in case... */
2440                                 p = NULL; /* then force the loop condition to become false */
2441                                 break;
2442                         } else {
2443                                 fprintf(stderr, "Error reading password from file %s: %s\n",
2444                                         spec, "empty password\n");
2445                                 exit(1);
2446                         }
2447                         
2448                 default:
2449                         fprintf(stderr, "Error reading password from file %s: %s\n",
2450                                 spec, strerror(errno));
2451                         exit(1);
2452                 }
2453         }
2454         pstrcpy(password, pass);
2455         if (close_it)
2456                 close(fd);
2457 }       
2458
2459
2460
2461 /****************************************************************************
2462 handle a -L query
2463 ****************************************************************************/
2464 static int do_host_query(char *query_host)
2465 {
2466         cli = do_connect(query_host, "IPC$");
2467         if (!cli)
2468                 return 1;
2469
2470         browse_host(True);
2471         list_servers(workgroup);
2472
2473         cli_shutdown(cli);
2474         
2475         return(0);
2476 }
2477
2478
2479 /****************************************************************************
2480 handle a tar operation
2481 ****************************************************************************/
2482 static int do_tar_op(char *base_directory)
2483 {
2484         int ret;
2485         cli = do_connect(desthost, service);
2486         if (!cli)
2487                 return 1;
2488
2489         recurse=True;
2490
2491         if (*base_directory) do_cd(base_directory);
2492         
2493         ret=process_tar();
2494
2495         cli_shutdown(cli);
2496
2497         return(ret);
2498 }
2499
2500 /****************************************************************************
2501 handle a message operation
2502 ****************************************************************************/
2503 static int do_message_op(void)
2504 {
2505         struct in_addr ip;
2506         struct nmb_name called, calling;
2507
2508         zero_ip(&ip);
2509
2510         make_nmb_name(&calling, global_myname, 0x0);
2511         make_nmb_name(&called , desthost, name_type);
2512
2513         zero_ip(&ip);
2514         if (have_ip) ip = dest_ip;
2515
2516         if (!(cli=cli_initialise(NULL)) || (cli_set_port(cli, port) != port) || !cli_connect(cli, desthost, &ip)) {
2517                 d_printf("Connection to %s failed\n", desthost);
2518                 return 1;
2519         }
2520
2521         if (!cli_session_request(cli, &calling, &called)) {
2522                 d_printf("session request failed\n");
2523                 cli_shutdown(cli);
2524                 return 1;
2525         }
2526
2527         send_message();
2528         cli_shutdown(cli);
2529
2530         return 0;
2531 }
2532
2533
2534 /****************************************************************************
2535   main program
2536 ****************************************************************************/
2537  int main(int argc,char *argv[])
2538 {
2539         fstring base_directory;
2540         char *pname = argv[0];
2541         int opt;
2542         extern char *optarg;
2543         extern int optind;
2544         int old_debug;
2545         pstring query_host;
2546         BOOL message = False;
2547         extern char tar_type;
2548         pstring term_code;
2549         pstring new_name_resolve_order;
2550         pstring logfile;
2551         char *p;
2552         int rc = 0;
2553
2554 #ifdef KANJI
2555         pstrcpy(term_code, KANJI);
2556 #else /* KANJI */
2557         *term_code = 0;
2558 #endif /* KANJI */
2559
2560         *query_host = 0;
2561         *base_directory = 0;
2562
2563         *new_name_resolve_order = 0;
2564
2565         DEBUGLEVEL = 2;
2566         AllowDebugChange = False;
2567  
2568         setup_logging(pname,True);
2569
2570         /*
2571          * If the -E option is given, be careful not to clobber stdout
2572          * before processing the options.  28.Feb.99, richard@hacom.nl.
2573          * Also pre-parse the -s option to get the service file name.
2574          */
2575
2576         for (opt = 1; opt < argc; opt++) {
2577                 if (strcmp(argv[opt], "-E") == 0)
2578                         dbf = x_stderr;
2579                 else if(strncmp(argv[opt], "-s", 2) == 0) {
2580                         if(argv[opt][2] != '\0')
2581                                 pstrcpy(dyn_CONFIGFILE, &argv[opt][2]);
2582                         else if(argv[opt+1] != NULL) {
2583                                 /*
2584                                  * At least one more arg left.
2585                                  */
2586                                 pstrcpy(dyn_CONFIGFILE, argv[opt+1]);
2587                         } else {
2588                                 usage(pname);
2589                                 exit(1);
2590                         }
2591                 }
2592         }
2593
2594         in_client = True;   /* Make sure that we tell lp_load we are */
2595
2596         old_debug = DEBUGLEVEL;
2597         if (!lp_load(dyn_CONFIGFILE,True,False,False)) {
2598                 fprintf(stderr, "%s: Can't load %s - run testparm to debug it\n",
2599                         prog_name, dyn_CONFIGFILE);
2600         }
2601         DEBUGLEVEL = old_debug;
2602         
2603 #ifdef WITH_SSL
2604         sslutil_init(0);
2605 #endif
2606
2607         pstrcpy(workgroup,lp_workgroup());
2608
2609         load_interfaces();
2610         myumask = umask(0);
2611         umask(myumask);
2612
2613         if (getenv("USER")) {
2614                 pstrcpy(username,getenv("USER"));
2615
2616                 /* modification to support userid%passwd syntax in the USER var
2617                    25.Aug.97, jdblair@uab.edu */
2618
2619                 if ((p=strchr_m(username,'%'))) {
2620                         *p = 0;
2621                         pstrcpy(password,p+1);
2622                         got_pass = True;
2623                         memset(strchr_m(getenv("USER"),'%')+1,'X',strlen(password));
2624                 }
2625                 strupper(username);
2626         }
2627
2628         /* modification to support PASSWD environmental var
2629            25.Aug.97, jdblair@uab.edu */
2630         if (getenv("PASSWD")) {
2631                 pstrcpy(password,getenv("PASSWD"));
2632                 got_pass = True;
2633         }
2634
2635         if (getenv("PASSWD_FD") || getenv("PASSWD_FILE")) {
2636                 get_password_file();
2637                 got_pass = True;
2638         }
2639
2640         if (*username == 0 && getenv("LOGNAME")) {
2641                 pstrcpy(username,getenv("LOGNAME"));
2642                 strupper(username);
2643         }
2644
2645         if (*username == 0) {
2646                 pstrcpy(username,"GUEST");
2647         }
2648
2649         if (argc < 2) {
2650                 usage(pname);
2651                 exit(1);
2652         }
2653
2654         /* FIXME: At the moment, if the user should happen to give the
2655          * options ahead of the service name (in standard Unix
2656          * fashion) then smbclient just spits out the usage message
2657          * with no explanation of what in particular was wrong.  Is
2658          * there any reason we can't just parse out the service name
2659          * and password after running getopt?? -- mbp */
2660         if (*argv[1] != '-') {
2661                 pstrcpy(service,argv[1]);  
2662                 /* Convert any '/' characters in the service name to '\' characters */
2663                 string_replace( service, '/','\\');
2664                 argc--;
2665                 argv++;
2666                 
2667                 if (count_chars(service,'\\') < 3) {
2668                         usage(pname);
2669                         d_printf("\n%s: Not enough '\\' characters in service\n",service);
2670                         exit(1);
2671                 }
2672
2673                 if (argc > 1 && (*argv[1] != '-')) {
2674                         got_pass = True;
2675                         pstrcpy(password,argv[1]);  
2676                         memset(argv[1],'X',strlen(argv[1]));
2677                         argc--;
2678                         argv++;
2679                 }
2680         }
2681
2682         while ((opt = 
2683                 getopt(argc, argv,"s:O:R:M:i:Nn:d:Pp:l:hI:EU:L:t:m:W:T:D:c:b:A:k")) != EOF) {
2684                 switch (opt) {
2685                 case 's':
2686                         pstrcpy(dyn_CONFIGFILE, optarg);
2687                         break;
2688                 case 'O':
2689                         pstrcpy(user_socket_options,optarg);
2690                         break;  
2691                 case 'R':
2692                         pstrcpy(new_name_resolve_order, optarg);
2693                         break;
2694                 case 'M':
2695                         name_type = 0x03; /* messages are sent to NetBIOS name type 0x3 */
2696                         pstrcpy(desthost,optarg);
2697                         message = True;
2698                         break;
2699                 case 'i':
2700                         {
2701                                 extern pstring global_scope;
2702                                 pstrcpy(global_scope,optarg);
2703                                 strupper(global_scope);
2704                         }
2705                         break;
2706                 case 'N':
2707                         got_pass = True;
2708                         break;
2709                 case 'n':
2710                         pstrcpy(global_myname,optarg);
2711                         break;
2712                 case 'd':
2713                         if (*optarg == 'A')
2714                                 DEBUGLEVEL = 10000;
2715                         else
2716                                 DEBUGLEVEL = atoi(optarg);
2717                         break;
2718                 case 'P':
2719                         /* not needed anymore */
2720                         break;
2721                 case 'p':
2722                         port = atoi(optarg);
2723                         break;
2724                 case 'l':
2725                         slprintf(logfile,sizeof(logfile)-1, "%s.client",optarg);
2726                         lp_set_logfile(logfile);
2727                         break;
2728                 case 'h':
2729                         usage(pname);
2730                         exit(0);
2731                         break;
2732                 case 'I':
2733                         {
2734                                 dest_ip = *interpret_addr2(optarg);
2735                                 if (is_zero_ip(dest_ip))
2736                                         exit(1);
2737                                 have_ip = True;
2738                         }
2739                         break;
2740                 case 'E':
2741                         display_set_stderr();
2742                         dbf = x_stderr;
2743                         break;
2744                 case 'U':
2745                         {
2746                                 char *lp;
2747                                 pstrcpy(username,optarg);
2748                                 if ((lp=strchr_m(username,'%'))) {
2749                                         *lp = 0;
2750                                         pstrcpy(password,lp+1);
2751                                         got_pass = True;
2752                                         memset(strchr_m(optarg,'%')+1,'X',strlen(password));
2753                                 }
2754                         }
2755                         break;
2756
2757                 case 'A':
2758                         {
2759                                 XFILE *auth;
2760                                 fstring buf;
2761                                 uint16 len = 0;
2762                                 char *ptr, *val, *param;
2763                                
2764                                 if ((auth=x_fopen(optarg, O_RDONLY, 0)) == NULL)
2765                                 {
2766                                         /* fail if we can't open the credentials file */
2767                                         d_printf("ERROR: Unable to open credentials file!\n");
2768                                         exit (-1);
2769                                 }
2770                                 
2771                                 while (!x_feof(auth))
2772                                 {  
2773                                         /* get a line from the file */
2774                                         if (!x_fgets(buf, sizeof(buf), auth))
2775                                                 continue;
2776                                         len = strlen(buf);
2777                                         
2778                                         if ((len) && (buf[len-1]=='\n'))
2779                                         {
2780                                                 buf[len-1] = '\0';
2781                                                 len--;
2782                                         }       
2783                                         if (len == 0)
2784                                                 continue;
2785                                         
2786                                         /* break up the line into parameter & value.
2787                                            will need to eat a little whitespace possibly */
2788                                         param = buf;
2789                                         if (!(ptr = strchr_m (buf, '=')))
2790                                                 continue;
2791                                         val = ptr+1;
2792                                         *ptr = '\0';
2793                                         
2794                                         /* eat leading white space */
2795                                         while ((*val!='\0') && ((*val==' ') || (*val=='\t')))
2796                                                 val++;
2797                                         
2798                                         if (strwicmp("password", param) == 0)
2799                                         {
2800                                                 pstrcpy(password, val);
2801                                                 got_pass = True;
2802                                         }
2803                                         else if (strwicmp("username", param) == 0)
2804                                                 pstrcpy(username, val);
2805                                         else if (strwicmp("domain", param) == 0)
2806                                                 pstrcpy(workgroup,val);
2807                                         memset(buf, 0, sizeof(buf));
2808                                 }
2809                                 x_fclose(auth);
2810                         }
2811                         break;
2812
2813                 case 'L':
2814                         p = optarg;
2815                         while(*p == '\\' || *p == '/')
2816                                 p++;
2817                         pstrcpy(query_host,p);
2818                         break;
2819                 case 't':
2820                         pstrcpy(term_code, optarg);
2821                         break;
2822                 case 'm':
2823                         max_protocol = interpret_protocol(optarg, max_protocol);
2824                         break;
2825                 case 'W':
2826                         pstrcpy(workgroup,optarg);
2827                         break;
2828                 case 'T':
2829                         if (!tar_parseargs(argc, argv, optarg, optind)) {
2830                                 usage(pname);
2831                                 exit(1);
2832                         }
2833                         break;
2834                 case 'D':
2835                         pstrcpy(base_directory,optarg);
2836                         break;
2837                 case 'c':
2838                         cmdstr = optarg;
2839                         break;
2840                 case 'b':
2841                         io_bufsize = MAX(1, atoi(optarg));
2842                         break;
2843                 case 'k':
2844 #ifdef HAVE_KRB5
2845                         use_kerberos = True;
2846                         got_pass = True;
2847 #else
2848                         d_printf("No kerberos support compiled in\n");
2849                         exit(1);
2850 #endif
2851                         break;
2852                 default:
2853                         usage(pname);
2854                         exit(1);
2855                 }
2856         }
2857
2858         get_myname((*global_myname)?NULL:global_myname);  
2859
2860         if(*new_name_resolve_order)
2861                 lp_set_name_resolve_order(new_name_resolve_order);
2862
2863         if (!tar_type && !*query_host && !*service && !message) {
2864                 usage(pname);
2865                 exit(1);
2866         }
2867
2868         DEBUG( 3, ( "Client started (version %s).\n", VERSION ) );
2869
2870         if (tar_type) {
2871                 if (cmdstr)
2872                         process_command_string(cmdstr);
2873                 return do_tar_op(base_directory);
2874         }
2875
2876         if ((p=strchr_m(query_host,'#'))) {
2877                 *p = 0;
2878                 p++;
2879                 sscanf(p, "%x", &name_type);
2880         }
2881   
2882         if (*query_host) {
2883                 return do_host_query(query_host);
2884         }
2885
2886         if (message) {
2887                 return do_message_op();
2888         }
2889         
2890         if (process(base_directory)) {
2891                 return 1;
2892         }
2893
2894         return rc;
2895 }