Context Navigation

← Previous Revision
Next Revision →
Blame
Revision Log

source: vendor/3.5.7/source4/client/client.c

Visit:

Last change on this file was 414, checked in by Herwig Bauernfeind, 16 years ago
Samba 3.5.0: Initial import
File size: 90.9 KB

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

Note: See TracBrowser for help on using the repository browser.

Download in other formats: