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