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