r22882: It seems entirly reasonable to follow metze's suggestion and check for
[Samba.git] / source / client / client.c
blob557767022c9561a78362efced8e8f82a8e1a2c1b
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>
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 2 of the License, or
12 (at your option) any later version.
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.
19 You should have received a copy of the GNU General Public License
20 along with this program; if not, write to the Free Software
21 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 #include "includes.h"
25 #include "version.h"
26 #include "libcli/libcli.h"
27 #include "lib/cmdline/popt_common.h"
28 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
29 #include "librpc/gen_ndr/ndr_lsa.h"
30 #include "librpc/gen_ndr/ndr_security.h"
31 #include "libcli/raw/libcliraw.h"
32 #include "libcli/util/clilsa.h"
33 #include "system/dir.h"
34 #include "system/filesys.h"
35 #include "lib/util/dlinklist.h"
36 #include "system/readline.h"
37 #include "auth/credentials/credentials.h"
38 #include "auth/gensec/gensec.h"
39 #include "system/time.h" /* needed by some systems for asctime() */
40 #include "libcli/resolve/resolve.h"
41 #include "libcli/security/security.h"
42 #include "lib/smbreadline/smbreadline.h"
43 #include "librpc/gen_ndr/ndr_nbt.h"
45 static int io_bufsize = 64512;
47 struct smbclient_context {
48 char *remote_cur_dir;
49 struct smbcli_state *cli;
50 char *fileselection;
51 time_t newer_than;
52 BOOL prompt;
53 BOOL recurse;
54 int archive_level;
55 BOOL lowercase;
56 int printmode;
57 BOOL translation;
60 /* timing globals */
61 static uint64_t get_total_size = 0;
62 static uint_t get_total_time_ms = 0;
63 static uint64_t put_total_size = 0;
64 static uint_t put_total_time_ms = 0;
66 /* Unfortunately, there is no way to pass the a context to the completion function as an argument */
67 static struct smbclient_context *rl_ctx;
69 /* totals globals */
70 static double dir_total;
72 /*******************************************************************
73 Reduce a file name, removing .. elements.
74 ********************************************************************/
75 void dos_clean_name(char *s)
77 char *p=NULL,*r;
79 DEBUG(3,("dos_clean_name [%s]\n",s));
81 /* remove any double slashes */
82 all_string_sub(s, "\\\\", "\\", 0);
84 while ((p = strstr(s,"\\..\\")) != NULL) {
85 *p = '\0';
86 if ((r = strrchr(s,'\\')) != NULL)
87 memmove(r,p+3,strlen(p+3)+1);
90 trim_string(s,NULL,"\\..");
92 all_string_sub(s, "\\.\\", "\\", 0);
95 /****************************************************************************
96 write to a local file with CR/LF->LF translation if appropriate. return the
97 number taken from the buffer. This may not equal the number written.
98 ****************************************************************************/
99 static int writefile(int f, const void *_b, int n, BOOL translation)
101 const uint8_t *b = _b;
102 int i;
104 if (!translation) {
105 return write(f,b,n);
108 i = 0;
109 while (i < n) {
110 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
111 b++;i++;
113 if (write(f, b, 1) != 1) {
114 break;
116 b++;
117 i++;
120 return(i);
123 /****************************************************************************
124 read from a file with LF->CR/LF translation if appropriate. return the
125 number read. read approx n bytes.
126 ****************************************************************************/
127 static int readfile(void *_b, int n, XFILE *f, BOOL translation)
129 uint8_t *b = _b;
130 int i;
131 int c;
133 if (!translation)
134 return x_fread(b,1,n,f);
136 i = 0;
137 while (i < (n - 1)) {
138 if ((c = x_getc(f)) == EOF) {
139 break;
142 if (c == '\n') { /* change all LFs to CR/LF */
143 b[i++] = '\r';
146 b[i++] = c;
149 return(i);
153 /****************************************************************************
154 send a message
155 ****************************************************************************/
156 static void send_message(struct smbcli_state *cli, const char *desthost)
158 char msg[1600];
159 int total_len = 0;
160 int grp_id;
162 if (!smbcli_message_start(cli->tree, desthost, cli_credentials_get_username(cmdline_credentials), &grp_id)) {
163 d_printf("message start: %s\n", smbcli_errstr(cli->tree));
164 return;
168 d_printf("Connected. Type your message, ending it with a Control-D\n");
170 while (!feof(stdin) && total_len < 1600) {
171 int maxlen = MIN(1600 - total_len,127);
172 int l=0;
173 int c;
175 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
176 if (c == '\n')
177 msg[l++] = '\r';
178 msg[l] = c;
181 if (!smbcli_message_text(cli->tree, msg, l, grp_id)) {
182 d_printf("SMBsendtxt failed (%s)\n",smbcli_errstr(cli->tree));
183 return;
186 total_len += l;
189 if (total_len >= 1600)
190 d_printf("the message was truncated to 1600 bytes\n");
191 else
192 d_printf("sent %d bytes\n",total_len);
194 if (!smbcli_message_end(cli->tree, grp_id)) {
195 d_printf("SMBsendend failed (%s)\n",smbcli_errstr(cli->tree));
196 return;
202 /****************************************************************************
203 check the space on a device
204 ****************************************************************************/
205 static int do_dskattr(struct smbclient_context *ctx)
207 int total, bsize, avail;
209 if (NT_STATUS_IS_ERR(smbcli_dskattr(ctx->cli->tree, &bsize, &total, &avail))) {
210 d_printf("Error in dskattr: %s\n",smbcli_errstr(ctx->cli->tree));
211 return 1;
214 d_printf("\n\t\t%d blocks of size %d. %d blocks available\n",
215 total, bsize, avail);
217 return 0;
220 /****************************************************************************
221 show cd/pwd
222 ****************************************************************************/
223 static int cmd_pwd(struct smbclient_context *ctx, const char **args)
225 d_printf("Current directory is %s\n", ctx->remote_cur_dir);
226 return 0;
230 convert a string to dos format
232 static void dos_format(char *s)
234 string_replace(s, '/', '\\');
237 /****************************************************************************
238 change directory - inner section
239 ****************************************************************************/
240 static int do_cd(struct smbclient_context *ctx, const char *newdir)
242 char *dname;
244 /* Save the current directory in case the
245 new directory is invalid */
246 if (newdir[0] == '\\')
247 dname = talloc_strdup(NULL, newdir);
248 else
249 dname = talloc_asprintf(NULL, "%s\\%s", ctx->remote_cur_dir, newdir);
251 dos_format(dname);
253 if (*(dname+strlen(dname)-1) != '\\') {
254 dname = talloc_append_string(NULL, dname, "\\");
256 dos_clean_name(dname);
258 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, dname))) {
259 d_printf("cd %s: %s\n", dname, smbcli_errstr(ctx->cli->tree));
260 talloc_free(dname);
261 } else {
262 ctx->remote_cur_dir = dname;
265 return 0;
268 /****************************************************************************
269 change directory
270 ****************************************************************************/
271 static int cmd_cd(struct smbclient_context *ctx, const char **args)
273 int rc = 0;
275 if (args[1])
276 rc = do_cd(ctx, args[1]);
277 else
278 d_printf("Current directory is %s\n",ctx->remote_cur_dir);
280 return rc;
284 BOOL mask_match(struct smbcli_state *c, const char *string, const char *pattern,
285 BOOL is_case_sensitive)
287 char *p2, *s2;
288 BOOL ret;
290 if (ISDOTDOT(string))
291 string = ".";
292 if (ISDOT(pattern))
293 return False;
295 if (is_case_sensitive)
296 return ms_fnmatch(pattern, string,
297 c->transport->negotiate.protocol) == 0;
299 p2 = strlower_talloc(NULL, pattern);
300 s2 = strlower_talloc(NULL, string);
301 ret = ms_fnmatch(p2, s2, c->transport->negotiate.protocol) == 0;
302 talloc_free(p2);
303 talloc_free(s2);
305 return ret;
310 /*******************************************************************
311 decide if a file should be operated on
312 ********************************************************************/
313 static BOOL do_this_one(struct smbclient_context *ctx, struct clilist_file_info *finfo)
315 if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY) return(True);
317 if (ctx->fileselection &&
318 !mask_match(ctx->cli, finfo->name,ctx->fileselection,False)) {
319 DEBUG(3,("mask_match %s failed\n", finfo->name));
320 return False;
323 if (ctx->newer_than && finfo->mtime < ctx->newer_than) {
324 DEBUG(3,("newer_than %s failed\n", finfo->name));
325 return(False);
328 if ((ctx->archive_level==1 || ctx->archive_level==2) && !(finfo->attrib & FILE_ATTRIBUTE_ARCHIVE)) {
329 DEBUG(3,("archive %s failed\n", finfo->name));
330 return(False);
333 return(True);
336 /****************************************************************************
337 display info about a file
338 ****************************************************************************/
339 static void display_finfo(struct smbclient_context *ctx, struct clilist_file_info *finfo)
341 if (do_this_one(ctx, finfo)) {
342 time_t t = finfo->mtime; /* the time is assumed to be passed as GMT */
343 char *astr = attrib_string(NULL, finfo->attrib);
344 d_printf(" %-30s%7.7s %8.0f %s",
345 finfo->name,
346 astr,
347 (double)finfo->size,
348 asctime(localtime(&t)));
349 dir_total += finfo->size;
350 talloc_free(astr);
355 /****************************************************************************
356 accumulate size of a file
357 ****************************************************************************/
358 static void do_du(struct smbclient_context *ctx, struct clilist_file_info *finfo)
360 if (do_this_one(ctx, finfo)) {
361 dir_total += finfo->size;
365 static BOOL do_list_recurse;
366 static BOOL do_list_dirs;
367 static char *do_list_queue = 0;
368 static long do_list_queue_size = 0;
369 static long do_list_queue_start = 0;
370 static long do_list_queue_end = 0;
371 static void (*do_list_fn)(struct smbclient_context *, struct clilist_file_info *);
373 /****************************************************************************
374 functions for do_list_queue
375 ****************************************************************************/
378 * The do_list_queue is a NUL-separated list of strings stored in a
379 * char*. Since this is a FIFO, we keep track of the beginning and
380 * ending locations of the data in the queue. When we overflow, we
381 * double the size of the char*. When the start of the data passes
382 * the midpoint, we move everything back. This is logically more
383 * complex than a linked list, but easier from a memory management
384 * angle. In any memory error condition, do_list_queue is reset.
385 * Functions check to ensure that do_list_queue is non-NULL before
386 * accessing it.
388 static void reset_do_list_queue(void)
390 SAFE_FREE(do_list_queue);
391 do_list_queue_size = 0;
392 do_list_queue_start = 0;
393 do_list_queue_end = 0;
396 static void init_do_list_queue(void)
398 reset_do_list_queue();
399 do_list_queue_size = 1024;
400 do_list_queue = malloc(do_list_queue_size);
401 if (do_list_queue == 0) {
402 d_printf("malloc fail for size %d\n",
403 (int)do_list_queue_size);
404 reset_do_list_queue();
405 } else {
406 memset(do_list_queue, 0, do_list_queue_size);
410 static void adjust_do_list_queue(void)
412 if (do_list_queue == NULL) return;
415 * If the starting point of the queue is more than half way through,
416 * move everything toward the beginning.
418 if (do_list_queue_start == do_list_queue_end)
420 DEBUG(4,("do_list_queue is empty\n"));
421 do_list_queue_start = do_list_queue_end = 0;
422 *do_list_queue = '\0';
424 else if (do_list_queue_start > (do_list_queue_size / 2))
426 DEBUG(4,("sliding do_list_queue backward\n"));
427 memmove(do_list_queue,
428 do_list_queue + do_list_queue_start,
429 do_list_queue_end - do_list_queue_start);
430 do_list_queue_end -= do_list_queue_start;
431 do_list_queue_start = 0;
436 static void add_to_do_list_queue(const char* entry)
438 char *dlq;
439 long new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
440 while (new_end > do_list_queue_size)
442 do_list_queue_size *= 2;
443 DEBUG(4,("enlarging do_list_queue to %d\n",
444 (int)do_list_queue_size));
445 dlq = realloc_p(do_list_queue, char, do_list_queue_size);
446 if (! dlq) {
447 d_printf("failure enlarging do_list_queue to %d bytes\n",
448 (int)do_list_queue_size);
449 reset_do_list_queue();
451 else
453 do_list_queue = dlq;
454 memset(do_list_queue + do_list_queue_size / 2,
455 0, do_list_queue_size / 2);
458 if (do_list_queue)
460 safe_strcpy(do_list_queue + do_list_queue_end, entry,
461 do_list_queue_size - do_list_queue_end - 1);
462 do_list_queue_end = new_end;
463 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
464 entry, (int)do_list_queue_start, (int)do_list_queue_end));
468 static char *do_list_queue_head(void)
470 return do_list_queue + do_list_queue_start;
473 static void remove_do_list_queue_head(void)
475 if (do_list_queue_end > do_list_queue_start)
477 do_list_queue_start += strlen(do_list_queue_head()) + 1;
478 adjust_do_list_queue();
479 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
480 (int)do_list_queue_start, (int)do_list_queue_end));
484 static int do_list_queue_empty(void)
486 return (! (do_list_queue && *do_list_queue));
489 /****************************************************************************
490 a helper for do_list
491 ****************************************************************************/
492 static void do_list_helper(struct clilist_file_info *f, const char *mask, void *state)
494 struct smbclient_context *ctx = state;
496 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY) {
497 if (do_list_dirs && do_this_one(ctx, f)) {
498 do_list_fn(ctx, f);
500 if (do_list_recurse &&
501 !ISDOT(f->name) &&
502 !ISDOTDOT(f->name)) {
503 char *mask2;
504 char *p;
506 mask2 = talloc_strdup(NULL, mask);
507 p = strrchr_m(mask2,'\\');
508 if (!p) return;
509 p[1] = 0;
510 mask2 = talloc_asprintf_append(mask2, "%s\\*", f->name);
511 add_to_do_list_queue(mask2);
513 return;
516 if (do_this_one(ctx, f)) {
517 do_list_fn(ctx, f);
522 /****************************************************************************
523 a wrapper around smbcli_list that adds recursion
524 ****************************************************************************/
525 static void do_list(struct smbclient_context *ctx, const char *mask,uint16_t attribute,
526 void (*fn)(struct smbclient_context *, struct clilist_file_info *),BOOL rec, BOOL dirs)
528 static int in_do_list = 0;
530 if (in_do_list && rec)
532 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
533 exit(1);
536 in_do_list = 1;
538 do_list_recurse = rec;
539 do_list_dirs = dirs;
540 do_list_fn = fn;
542 if (rec)
544 init_do_list_queue();
545 add_to_do_list_queue(mask);
547 while (! do_list_queue_empty())
550 * Need to copy head so that it doesn't become
551 * invalid inside the call to smbcli_list. This
552 * would happen if the list were expanded
553 * during the call.
554 * Fix from E. Jay Berkenbilt (ejb@ql.org)
556 char *head;
557 head = do_list_queue_head();
558 smbcli_list(ctx->cli->tree, head, attribute, do_list_helper, ctx);
559 remove_do_list_queue_head();
560 if ((! do_list_queue_empty()) && (fn == display_finfo))
562 char* next_file = do_list_queue_head();
563 char* save_ch = 0;
564 if ((strlen(next_file) >= 2) &&
565 (next_file[strlen(next_file) - 1] == '*') &&
566 (next_file[strlen(next_file) - 2] == '\\'))
568 save_ch = next_file +
569 strlen(next_file) - 2;
570 *save_ch = '\0';
572 d_printf("\n%s\n",next_file);
573 if (save_ch)
575 *save_ch = '\\';
580 else
582 if (smbcli_list(ctx->cli->tree, mask, attribute, do_list_helper, ctx) == -1)
584 d_printf("%s listing %s\n", smbcli_errstr(ctx->cli->tree), mask);
588 in_do_list = 0;
589 reset_do_list_queue();
592 /****************************************************************************
593 get a directory listing
594 ****************************************************************************/
595 static int cmd_dir(struct smbclient_context *ctx, const char **args)
597 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
598 char *mask;
599 int rc;
601 dir_total = 0;
603 mask = talloc_strdup(ctx, ctx->remote_cur_dir);
604 if(mask[strlen(mask)-1]!='\\')
605 mask = talloc_append_string(ctx, mask,"\\");
607 if (args[1]) {
608 mask = talloc_strdup(ctx, args[1]);
609 if (mask[0] != '\\')
610 mask = talloc_append_string(ctx, mask, "\\");
611 dos_format(mask);
613 else {
614 if (ctx->cli->tree->session->transport->negotiate.protocol <=
615 PROTOCOL_LANMAN1) {
616 mask = talloc_append_string(ctx, mask, "*.*");
617 } else {
618 mask = talloc_append_string(ctx, mask, "*");
622 do_list(ctx, mask, attribute, display_finfo, ctx->recurse, True);
624 rc = do_dskattr(ctx);
626 DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
628 return rc;
632 /****************************************************************************
633 get a directory listing
634 ****************************************************************************/
635 static int cmd_du(struct smbclient_context *ctx, const char **args)
637 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
638 int rc;
639 char *mask;
641 dir_total = 0;
643 if (args[1]) {
644 if (args[1][0] == '\\')
645 mask = talloc_strdup(ctx, args[1]);
646 else
647 mask = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
648 dos_format(mask);
649 } else {
650 mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
653 do_list(ctx, mask, attribute, do_du, ctx->recurse, True);
655 talloc_free(mask);
657 rc = do_dskattr(ctx);
659 d_printf("Total number of bytes: %.0f\n", dir_total);
661 return rc;
665 /****************************************************************************
666 get a file from rname to lname
667 ****************************************************************************/
668 static int do_get(struct smbclient_context *ctx, char *rname, const char *lname, BOOL reget)
670 int handle = 0, fnum;
671 BOOL newhandle = False;
672 uint8_t *data;
673 struct timeval tp_start;
674 int read_size = io_bufsize;
675 uint16_t attr;
676 size_t size;
677 off_t start = 0;
678 off_t nread = 0;
679 int rc = 0;
681 GetTimeOfDay(&tp_start);
683 if (ctx->lowercase) {
684 strlower(discard_const_p(char, lname));
687 fnum = smbcli_open(ctx->cli->tree, rname, O_RDONLY, DENY_NONE);
689 if (fnum == -1) {
690 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
691 return 1;
694 if(!strcmp(lname,"-")) {
695 handle = fileno(stdout);
696 } else {
697 if (reget) {
698 handle = open(lname, O_WRONLY|O_CREAT, 0644);
699 if (handle >= 0) {
700 start = lseek(handle, 0, SEEK_END);
701 if (start == -1) {
702 d_printf("Error seeking local file\n");
703 return 1;
706 } else {
707 handle = open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
709 newhandle = True;
711 if (handle < 0) {
712 d_printf("Error opening local file %s\n",lname);
713 return 1;
717 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum,
718 &attr, &size, NULL, NULL, NULL, NULL, NULL)) &&
719 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum,
720 &attr, &size, NULL, NULL, NULL))) {
721 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
722 return 1;
725 DEBUG(2,("getting file %s of size %.0f as %s ",
726 rname, (double)size, lname));
728 if(!(data = (uint8_t *)malloc(read_size))) {
729 d_printf("malloc fail for size %d\n", read_size);
730 smbcli_close(ctx->cli->tree, fnum);
731 return 1;
734 while (1) {
735 int n = smbcli_read(ctx->cli->tree, fnum, data, nread + start, read_size);
737 if (n <= 0) break;
739 if (writefile(handle,data, n, ctx->translation) != n) {
740 d_printf("Error writing local file\n");
741 rc = 1;
742 break;
745 nread += n;
748 if (nread + start < size) {
749 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
750 rname, (long)nread));
752 rc = 1;
755 SAFE_FREE(data);
757 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
758 d_printf("Error %s closing remote file\n",smbcli_errstr(ctx->cli->tree));
759 rc = 1;
762 if (newhandle) {
763 close(handle);
766 if (ctx->archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) {
767 smbcli_setatr(ctx->cli->tree, rname, attr & ~(uint16_t)FILE_ATTRIBUTE_ARCHIVE, 0);
771 struct timeval tp_end;
772 int this_time;
774 GetTimeOfDay(&tp_end);
775 this_time =
776 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
777 (tp_end.tv_usec - tp_start.tv_usec)/1000;
778 get_total_time_ms += this_time;
779 get_total_size += nread;
781 DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n",
782 nread / (1.024*this_time + 1.0e-4),
783 get_total_size / (1.024*get_total_time_ms)));
786 return rc;
790 /****************************************************************************
791 get a file
792 ****************************************************************************/
793 static int cmd_get(struct smbclient_context *ctx, const char **args)
795 const char *lname;
796 char *rname;
798 if (!args[1]) {
799 d_printf("get <filename>\n");
800 return 1;
803 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
805 if (args[2])
806 lname = args[2];
807 else
808 lname = args[1];
810 dos_clean_name(rname);
812 return do_get(ctx, rname, lname, False);
815 /****************************************************************************
816 Put up a yes/no prompt.
817 ****************************************************************************/
818 static BOOL yesno(char *p)
820 char ans[4];
821 printf("%s",p);
823 if (!fgets(ans,sizeof(ans)-1,stdin))
824 return(False);
826 if (*ans == 'y' || *ans == 'Y')
827 return(True);
829 return(False);
832 /****************************************************************************
833 do a mget operation on one file
834 ****************************************************************************/
835 static void do_mget(struct smbclient_context *ctx, struct clilist_file_info *finfo)
837 char *rname;
838 char *quest;
839 char *mget_mask;
840 char *saved_curdir;
842 if (ISDOT(finfo->name) || ISDOTDOT(finfo->name))
843 return;
845 if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)
846 asprintf(&quest, "Get directory %s? ",finfo->name);
847 else
848 asprintf(&quest, "Get file %s? ",finfo->name);
850 if (ctx->prompt && !yesno(quest)) return;
852 SAFE_FREE(quest);
854 if (!(finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)) {
855 asprintf(&rname, "%s%s",ctx->remote_cur_dir,finfo->name);
856 do_get(ctx, rname, finfo->name, False);
857 SAFE_FREE(rname);
858 return;
861 /* handle directories */
862 saved_curdir = talloc_strdup(NULL, ctx->remote_cur_dir);
864 ctx->remote_cur_dir = talloc_asprintf_append(NULL, "%s\\", finfo->name);
866 string_replace(discard_const_p(char, finfo->name), '\\', '/');
867 if (ctx->lowercase) {
868 strlower(discard_const_p(char, finfo->name));
871 if (!directory_exist(finfo->name) &&
872 mkdir(finfo->name,0777) != 0) {
873 d_printf("failed to create directory %s\n",finfo->name);
874 return;
877 if (chdir(finfo->name) != 0) {
878 d_printf("failed to chdir to directory %s\n",finfo->name);
879 return;
882 mget_mask = talloc_asprintf(NULL, "%s*", ctx->remote_cur_dir);
884 do_list(ctx, mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,False, True);
885 chdir("..");
886 talloc_free(ctx->remote_cur_dir);
888 ctx->remote_cur_dir = saved_curdir;
892 /****************************************************************************
893 view the file using the pager
894 ****************************************************************************/
895 static int cmd_more(struct smbclient_context *ctx, const char **args)
897 char *rname;
898 char *pager_cmd;
899 char *lname;
900 char *pager;
901 int fd;
902 int rc = 0;
904 lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir());
905 fd = mkstemp(lname);
906 if (fd == -1) {
907 d_printf("failed to create temporary file for more\n");
908 return 1;
910 close(fd);
912 if (!args[1]) {
913 d_printf("more <filename>\n");
914 unlink(lname);
915 return 1;
917 rname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
918 dos_clean_name(rname);
920 rc = do_get(ctx, rname, lname, False);
922 pager=getenv("PAGER");
924 pager_cmd = talloc_asprintf(ctx, "%s %s",(pager? pager:PAGER), lname);
925 system(pager_cmd);
926 unlink(lname);
928 return rc;
933 /****************************************************************************
934 do a mget command
935 ****************************************************************************/
936 static int cmd_mget(struct smbclient_context *ctx, const char **args)
938 uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
939 char *mget_mask = NULL;
940 int i;
942 if (ctx->recurse)
943 attribute |= FILE_ATTRIBUTE_DIRECTORY;
945 for (i = 1; args[i]; i++) {
946 mget_mask = talloc_strdup(ctx,ctx->remote_cur_dir);
947 if(mget_mask[strlen(mget_mask)-1]!='\\')
948 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
950 mget_mask = talloc_strdup(ctx, args[i]);
951 if (mget_mask[0] != '\\')
952 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
953 do_list(ctx, mget_mask, attribute,do_mget,False,True);
955 talloc_free(mget_mask);
958 if (mget_mask == NULL) {
959 mget_mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
960 do_list(ctx, mget_mask, attribute,do_mget,False,True);
961 talloc_free(mget_mask);
964 return 0;
968 /****************************************************************************
969 make a directory of name "name"
970 ****************************************************************************/
971 static NTSTATUS do_mkdir(struct smbclient_context *ctx, char *name)
973 NTSTATUS status;
975 if (NT_STATUS_IS_ERR(status = smbcli_mkdir(ctx->cli->tree, name))) {
976 d_printf("%s making remote directory %s\n",
977 smbcli_errstr(ctx->cli->tree),name);
978 return status;
981 return status;
985 /****************************************************************************
986 Exit client.
987 ****************************************************************************/
988 static int cmd_quit(struct smbclient_context *ctx, const char **args)
990 talloc_free(ctx);
991 exit(0);
992 /* NOTREACHED */
993 return 0;
997 /****************************************************************************
998 make a directory
999 ****************************************************************************/
1000 static int cmd_mkdir(struct smbclient_context *ctx, const char **args)
1002 char *mask, *p;
1004 if (!args[1]) {
1005 if (!ctx->recurse)
1006 d_printf("mkdir <dirname>\n");
1007 return 1;
1010 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir,args[1]);
1012 if (ctx->recurse) {
1013 dos_clean_name(mask);
1015 trim_string(mask,".",NULL);
1016 for (p = strtok(mask,"/\\"); p; p = strtok(p, "/\\")) {
1017 char *parent = talloc_strndup(ctx, mask, PTR_DIFF(p, mask));
1019 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, parent))) {
1020 do_mkdir(ctx, parent);
1023 talloc_free(parent);
1025 } else {
1026 do_mkdir(ctx, mask);
1029 return 0;
1032 /****************************************************************************
1033 show 8.3 name of a file
1034 ****************************************************************************/
1035 static int cmd_altname(struct smbclient_context *ctx, const char **args)
1037 const char *altname;
1038 char *name;
1040 if (!args[1]) {
1041 d_printf("altname <file>\n");
1042 return 1;
1045 name = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1047 if (!NT_STATUS_IS_OK(smbcli_qpathinfo_alt_name(ctx->cli->tree, name, &altname))) {
1048 d_printf("%s getting alt name for %s\n",
1049 smbcli_errstr(ctx->cli->tree),name);
1050 return(False);
1052 d_printf("%s\n", altname);
1054 return 0;
1058 /****************************************************************************
1059 put a single file
1060 ****************************************************************************/
1061 static int do_put(struct smbclient_context *ctx, char *rname, char *lname, BOOL reput)
1063 int fnum;
1064 XFILE *f;
1065 size_t start = 0;
1066 off_t nread = 0;
1067 uint8_t *buf = NULL;
1068 int maxwrite = io_bufsize;
1069 int rc = 0;
1071 struct timeval tp_start;
1072 GetTimeOfDay(&tp_start);
1074 if (reput) {
1075 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT, DENY_NONE);
1076 if (fnum >= 0) {
1077 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL)) &&
1078 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL))) {
1079 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
1080 return 1;
1083 } else {
1084 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT|O_TRUNC,
1085 DENY_NONE);
1088 if (fnum == -1) {
1089 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1090 return 1;
1093 /* allow files to be piped into smbclient
1094 jdblair 24.jun.98
1096 Note that in this case this function will exit(0) rather
1097 than returning. */
1098 if (!strcmp(lname, "-")) {
1099 f = x_stdin;
1100 /* size of file is not known */
1101 } else {
1102 f = x_fopen(lname,O_RDONLY, 0);
1103 if (f && reput) {
1104 if (x_tseek(f, start, SEEK_SET) == -1) {
1105 d_printf("Error seeking local file\n");
1106 return 1;
1111 if (!f) {
1112 d_printf("Error opening local file %s\n",lname);
1113 return 1;
1117 DEBUG(1,("putting file %s as %s ",lname,
1118 rname));
1120 buf = (uint8_t *)malloc(maxwrite);
1121 if (!buf) {
1122 d_printf("ERROR: Not enough memory!\n");
1123 return 1;
1125 while (!x_feof(f)) {
1126 int n = maxwrite;
1127 int ret;
1129 if ((n = readfile(buf,n,f,ctx->translation)) < 1) {
1130 if((n == 0) && x_feof(f))
1131 break; /* Empty local file. */
1133 d_printf("Error reading local file: %s\n", strerror(errno));
1134 rc = 1;
1135 break;
1138 ret = smbcli_write(ctx->cli->tree, fnum, 0, buf, nread + start, n);
1140 if (n != ret) {
1141 d_printf("Error writing file: %s\n", smbcli_errstr(ctx->cli->tree));
1142 rc = 1;
1143 break;
1146 nread += n;
1149 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
1150 d_printf("%s closing remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1151 x_fclose(f);
1152 SAFE_FREE(buf);
1153 return 1;
1157 if (f != x_stdin) {
1158 x_fclose(f);
1161 SAFE_FREE(buf);
1164 struct timeval tp_end;
1165 int this_time;
1167 GetTimeOfDay(&tp_end);
1168 this_time =
1169 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1170 (tp_end.tv_usec - tp_start.tv_usec)/1000;
1171 put_total_time_ms += this_time;
1172 put_total_size += nread;
1174 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1175 nread / (1.024*this_time + 1.0e-4),
1176 put_total_size / (1.024*put_total_time_ms)));
1179 if (f == x_stdin) {
1180 talloc_free(ctx);
1181 exit(0);
1184 return rc;
1189 /****************************************************************************
1190 put a file
1191 ****************************************************************************/
1192 static int cmd_put(struct smbclient_context *ctx, const char **args)
1194 char *lname;
1195 char *rname;
1197 if (!args[1]) {
1198 d_printf("put <filename> [<remotename>]\n");
1199 return 1;
1202 lname = talloc_strdup(ctx, args[1]);
1204 if (args[2])
1205 rname = talloc_strdup(ctx, args[2]);
1206 else
1207 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, lname);
1209 dos_clean_name(rname);
1211 /* allow '-' to represent stdin
1212 jdblair, 24.jun.98 */
1213 if (!file_exist(lname) && (strcmp(lname,"-"))) {
1214 d_printf("%s does not exist\n",lname);
1215 return 1;
1218 return do_put(ctx, rname, lname, False);
1221 /*************************************
1222 File list structure
1223 *************************************/
1225 static struct file_list {
1226 struct file_list *prev, *next;
1227 char *file_path;
1228 BOOL isdir;
1229 } *file_list;
1231 /****************************************************************************
1232 Free a file_list structure
1233 ****************************************************************************/
1235 static void free_file_list (struct file_list * list)
1237 struct file_list *tmp;
1239 while (list)
1241 tmp = list;
1242 DLIST_REMOVE(list, list);
1243 SAFE_FREE(tmp->file_path);
1244 SAFE_FREE(tmp);
1248 /****************************************************************************
1249 seek in a directory/file list until you get something that doesn't start with
1250 the specified name
1251 ****************************************************************************/
1252 static BOOL seek_list(struct file_list *list, char *name)
1254 while (list) {
1255 trim_string(list->file_path,"./","\n");
1256 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1257 return(True);
1259 list = list->next;
1262 return(False);
1265 /****************************************************************************
1266 set the file selection mask
1267 ****************************************************************************/
1268 static int cmd_select(struct smbclient_context *ctx, const char **args)
1270 talloc_free(ctx->fileselection);
1271 ctx->fileselection = talloc_strdup(NULL, args[1]);
1273 return 0;
1276 /*******************************************************************
1277 A readdir wrapper which just returns the file name.
1278 ********************************************************************/
1279 static const char *readdirname(DIR *p)
1281 struct dirent *ptr;
1282 char *dname;
1284 if (!p)
1285 return(NULL);
1287 ptr = (struct dirent *)readdir(p);
1288 if (!ptr)
1289 return(NULL);
1291 dname = ptr->d_name;
1293 #ifdef NEXT2
1294 if (telldir(p) < 0)
1295 return(NULL);
1296 #endif
1298 #ifdef HAVE_BROKEN_READDIR
1299 /* using /usr/ucb/cc is BAD */
1300 dname = dname - 2;
1301 #endif
1304 static char *buf;
1305 int len = NAMLEN(ptr);
1306 buf = talloc_strndup(NULL, dname, len);
1307 dname = buf;
1310 return(dname);
1313 /****************************************************************************
1314 Recursive file matching function act as find
1315 match must be always set to True when calling this function
1316 ****************************************************************************/
1317 static int file_find(struct smbclient_context *ctx, struct file_list **list, const char *directory,
1318 const char *expression, BOOL match)
1320 DIR *dir;
1321 struct file_list *entry;
1322 struct stat statbuf;
1323 int ret;
1324 char *path;
1325 BOOL isdir;
1326 const char *dname;
1328 dir = opendir(directory);
1329 if (!dir) return -1;
1331 while ((dname = readdirname(dir))) {
1332 if (ISDOT(dname) || ISDOTDOT(dname)) {
1333 continue;
1336 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1337 continue;
1340 isdir = False;
1341 if (!match || !gen_fnmatch(expression, dname)) {
1342 if (ctx->recurse) {
1343 ret = stat(path, &statbuf);
1344 if (ret == 0) {
1345 if (S_ISDIR(statbuf.st_mode)) {
1346 isdir = True;
1347 ret = file_find(ctx, list, path, expression, False);
1349 } else {
1350 d_printf("file_find: cannot stat file %s\n", path);
1353 if (ret == -1) {
1354 SAFE_FREE(path);
1355 closedir(dir);
1356 return -1;
1359 entry = malloc_p(struct file_list);
1360 if (!entry) {
1361 d_printf("Out of memory in file_find\n");
1362 closedir(dir);
1363 return -1;
1365 entry->file_path = path;
1366 entry->isdir = isdir;
1367 DLIST_ADD(*list, entry);
1368 } else {
1369 SAFE_FREE(path);
1373 closedir(dir);
1374 return 0;
1377 /****************************************************************************
1378 mput some files
1379 ****************************************************************************/
1380 static int cmd_mput(struct smbclient_context *ctx, const char **args)
1382 int i;
1384 for (i = 1; args[i]; i++) {
1385 int ret;
1386 struct file_list *temp_list;
1387 char *quest, *lname, *rname;
1389 printf("%s\n", args[i]);
1391 file_list = NULL;
1393 ret = file_find(ctx, &file_list, ".", args[i], True);
1394 if (ret) {
1395 free_file_list(file_list);
1396 continue;
1399 quest = NULL;
1400 lname = NULL;
1401 rname = NULL;
1403 for (temp_list = file_list; temp_list;
1404 temp_list = temp_list->next) {
1406 SAFE_FREE(lname);
1407 if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1408 continue;
1409 trim_string(lname, "./", "/");
1411 /* check if it's a directory */
1412 if (temp_list->isdir) {
1413 /* if (!recurse) continue; */
1415 SAFE_FREE(quest);
1416 if (asprintf(&quest, "Put directory %s? ", lname) < 0) break;
1417 if (ctx->prompt && !yesno(quest)) { /* No */
1418 /* Skip the directory */
1419 lname[strlen(lname)-1] = '/';
1420 if (!seek_list(temp_list, lname))
1421 break;
1422 } else { /* Yes */
1423 SAFE_FREE(rname);
1424 if(asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1425 dos_format(rname);
1426 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, rname)) &&
1427 NT_STATUS_IS_ERR(do_mkdir(ctx, rname))) {
1428 DEBUG (0, ("Unable to make dir, skipping..."));
1429 /* Skip the directory */
1430 lname[strlen(lname)-1] = '/';
1431 if (!seek_list(temp_list, lname))
1432 break;
1435 continue;
1436 } else {
1437 SAFE_FREE(quest);
1438 if (asprintf(&quest,"Put file %s? ", lname) < 0) break;
1439 if (ctx->prompt && !yesno(quest)) /* No */
1440 continue;
1442 /* Yes */
1443 SAFE_FREE(rname);
1444 if (asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1447 dos_format(rname);
1449 do_put(ctx, rname, lname, False);
1451 free_file_list(file_list);
1452 SAFE_FREE(quest);
1453 SAFE_FREE(lname);
1454 SAFE_FREE(rname);
1457 return 0;
1461 /****************************************************************************
1462 print a file
1463 ****************************************************************************/
1464 static int cmd_print(struct smbclient_context *ctx, const char **args)
1466 char *lname, *rname;
1467 char *p;
1469 if (!args[1]) {
1470 d_printf("print <filename>\n");
1471 return 1;
1474 lname = talloc_strdup(ctx, args[1]);
1476 rname = talloc_strdup(ctx, lname);
1477 p = strrchr_m(rname,'/');
1478 if (p) {
1479 slprintf(rname, sizeof(rname)-1, "%s-%d", p+1, (int)getpid());
1482 if (strequal(lname,"-")) {
1483 slprintf(rname, sizeof(rname)-1, "stdin-%d", (int)getpid());
1486 return do_put(ctx, rname, lname, False);
1490 static int cmd_rewrite(struct smbclient_context *ctx, const char **args)
1492 d_printf("REWRITE: command not implemented (FIXME!)\n");
1494 return 0;
1497 /****************************************************************************
1498 delete some files
1499 ****************************************************************************/
1500 static int cmd_del(struct smbclient_context *ctx, const char **args)
1502 char *mask;
1503 uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
1505 if (ctx->recurse)
1506 attribute |= FILE_ATTRIBUTE_DIRECTORY;
1508 if (!args[1]) {
1509 d_printf("del <filename>\n");
1510 return 1;
1512 mask = talloc_asprintf(ctx,"%s%s", ctx->remote_cur_dir, args[1]);
1514 if (NT_STATUS_IS_ERR(smbcli_unlink(ctx->cli->tree, mask))) {
1515 d_printf("%s deleting remote file %s\n",smbcli_errstr(ctx->cli->tree),mask);
1518 return 0;
1522 /****************************************************************************
1523 delete a whole directory tree
1524 ****************************************************************************/
1525 static int cmd_deltree(struct smbclient_context *ctx, const char **args)
1527 char *dname;
1528 int ret;
1530 if (!args[1]) {
1531 d_printf("deltree <dirname>\n");
1532 return 1;
1535 dname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1537 ret = smbcli_deltree(ctx->cli->tree, dname);
1539 if (ret == -1) {
1540 printf("Failed to delete tree %s - %s\n", dname, smbcli_errstr(ctx->cli->tree));
1541 return -1;
1544 printf("Deleted %d files in %s\n", ret, dname);
1546 return 0;
1549 typedef struct {
1550 const char *level_name;
1551 enum smb_fsinfo_level level;
1552 } fsinfo_level_t;
1554 fsinfo_level_t fsinfo_levels[] = {
1555 {"dskattr", RAW_QFS_DSKATTR},
1556 {"allocation", RAW_QFS_ALLOCATION},
1557 {"volume", RAW_QFS_VOLUME},
1558 {"volumeinfo", RAW_QFS_VOLUME_INFO},
1559 {"sizeinfo", RAW_QFS_SIZE_INFO},
1560 {"deviceinfo", RAW_QFS_DEVICE_INFO},
1561 {"attributeinfo", RAW_QFS_ATTRIBUTE_INFO},
1562 {"unixinfo", RAW_QFS_UNIX_INFO},
1563 {"volume-information", RAW_QFS_VOLUME_INFORMATION},
1564 {"size-information", RAW_QFS_SIZE_INFORMATION},
1565 {"device-information", RAW_QFS_DEVICE_INFORMATION},
1566 {"attribute-information", RAW_QFS_ATTRIBUTE_INFORMATION},
1567 {"quota-information", RAW_QFS_QUOTA_INFORMATION},
1568 {"fullsize-information", RAW_QFS_FULL_SIZE_INFORMATION},
1569 {"objectid", RAW_QFS_OBJECTID_INFORMATION},
1570 {NULL, RAW_QFS_GENERIC}
1574 static int cmd_fsinfo(struct smbclient_context *ctx, const char **args)
1576 union smb_fsinfo fsinfo;
1577 NTSTATUS status;
1578 fsinfo_level_t *fsinfo_level;
1580 if (!args[1]) {
1581 d_printf("fsinfo <level>, where level is one of following:\n");
1582 fsinfo_level = fsinfo_levels;
1583 while(fsinfo_level->level_name) {
1584 d_printf("%s\n", fsinfo_level->level_name);
1585 fsinfo_level++;
1587 return 1;
1590 fsinfo_level = fsinfo_levels;
1591 while(fsinfo_level->level_name && !strequal(args[1],fsinfo_level->level_name)) {
1592 fsinfo_level++;
1595 if (!fsinfo_level->level_name) {
1596 d_printf("wrong level name!\n");
1597 return 1;
1600 fsinfo.generic.level = fsinfo_level->level;
1601 status = smb_raw_fsinfo(ctx->cli->tree, ctx, &fsinfo);
1602 if (!NT_STATUS_IS_OK(status)) {
1603 d_printf("fsinfo-level-%s - %s\n", fsinfo_level->level_name, nt_errstr(status));
1604 return 1;
1607 d_printf("fsinfo-level-%s:\n", fsinfo_level->level_name);
1608 switch(fsinfo.generic.level) {
1609 case RAW_QFS_DSKATTR:
1610 d_printf("\tunits_total: %hu\n",
1611 (unsigned short) fsinfo.dskattr.out.units_total);
1612 d_printf("\tblocks_per_unit: %hu\n",
1613 (unsigned short) fsinfo.dskattr.out.blocks_per_unit);
1614 d_printf("\tblocks_size: %hu\n",
1615 (unsigned short) fsinfo.dskattr.out.block_size);
1616 d_printf("\tunits_free: %hu\n",
1617 (unsigned short) fsinfo.dskattr.out.units_free);
1618 break;
1619 case RAW_QFS_ALLOCATION:
1620 d_printf("\tfs_id: %lu\n",
1621 (unsigned long) fsinfo.allocation.out.fs_id);
1622 d_printf("\tsectors_per_unit: %lu\n",
1623 (unsigned long) fsinfo.allocation.out.sectors_per_unit);
1624 d_printf("\ttotal_alloc_units: %lu\n",
1625 (unsigned long) fsinfo.allocation.out.total_alloc_units);
1626 d_printf("\tavail_alloc_units: %lu\n",
1627 (unsigned long) fsinfo.allocation.out.avail_alloc_units);
1628 d_printf("\tbytes_per_sector: %hu\n",
1629 (unsigned short) fsinfo.allocation.out.bytes_per_sector);
1630 break;
1631 case RAW_QFS_VOLUME:
1632 d_printf("\tserial_number: %lu\n",
1633 (unsigned long) fsinfo.volume.out.serial_number);
1634 d_printf("\tvolume_name: %s\n", fsinfo.volume.out.volume_name.s);
1635 break;
1636 case RAW_QFS_VOLUME_INFO:
1637 case RAW_QFS_VOLUME_INFORMATION:
1638 d_printf("\tcreate_time: %s\n",
1639 nt_time_string(ctx,fsinfo.volume_info.out.create_time));
1640 d_printf("\tserial_number: %lu\n",
1641 (unsigned long) fsinfo.volume_info.out.serial_number);
1642 d_printf("\tvolume_name: %s\n", fsinfo.volume_info.out.volume_name.s);
1643 break;
1644 case RAW_QFS_SIZE_INFO:
1645 case RAW_QFS_SIZE_INFORMATION:
1646 d_printf("\ttotal_alloc_units: %llu\n",
1647 (unsigned long long) fsinfo.size_info.out.total_alloc_units);
1648 d_printf("\tavail_alloc_units: %llu\n",
1649 (unsigned long long) fsinfo.size_info.out.avail_alloc_units);
1650 d_printf("\tsectors_per_unit: %lu\n",
1651 (unsigned long) fsinfo.size_info.out.sectors_per_unit);
1652 d_printf("\tbytes_per_sector: %lu\n",
1653 (unsigned long) fsinfo.size_info.out.bytes_per_sector);
1654 break;
1655 case RAW_QFS_DEVICE_INFO:
1656 case RAW_QFS_DEVICE_INFORMATION:
1657 d_printf("\tdevice_type: %lu\n",
1658 (unsigned long) fsinfo.device_info.out.device_type);
1659 d_printf("\tcharacteristics: 0x%lx\n",
1660 (unsigned long) fsinfo.device_info.out.characteristics);
1661 break;
1662 case RAW_QFS_ATTRIBUTE_INFORMATION:
1663 case RAW_QFS_ATTRIBUTE_INFO:
1664 d_printf("\tfs_attr: 0x%lx\n",
1665 (unsigned long) fsinfo.attribute_info.out.fs_attr);
1666 d_printf("\tmax_file_component_length: %lu\n",
1667 (unsigned long) fsinfo.attribute_info.out.max_file_component_length);
1668 d_printf("\tfs_type: %s\n", fsinfo.attribute_info.out.fs_type.s);
1669 break;
1670 case RAW_QFS_UNIX_INFO:
1671 d_printf("\tmajor_version: %hu\n",
1672 (unsigned short) fsinfo.unix_info.out.major_version);
1673 d_printf("\tminor_version: %hu\n",
1674 (unsigned short) fsinfo.unix_info.out.minor_version);
1675 d_printf("\tcapability: 0x%llx\n",
1676 (unsigned long long) fsinfo.unix_info.out.capability);
1677 break;
1678 case RAW_QFS_QUOTA_INFORMATION:
1679 d_printf("\tunknown[3]: [%llu,%llu,%llu]\n",
1680 (unsigned long long) fsinfo.quota_information.out.unknown[0],
1681 (unsigned long long) fsinfo.quota_information.out.unknown[1],
1682 (unsigned long long) fsinfo.quota_information.out.unknown[2]);
1683 d_printf("\tquota_soft: %llu\n",
1684 (unsigned long long) fsinfo.quota_information.out.quota_soft);
1685 d_printf("\tquota_hard: %llu\n",
1686 (unsigned long long) fsinfo.quota_information.out.quota_hard);
1687 d_printf("\tquota_flags: 0x%llx\n",
1688 (unsigned long long) fsinfo.quota_information.out.quota_flags);
1689 break;
1690 case RAW_QFS_FULL_SIZE_INFORMATION:
1691 d_printf("\ttotal_alloc_units: %llu\n",
1692 (unsigned long long) fsinfo.full_size_information.out.total_alloc_units);
1693 d_printf("\tcall_avail_alloc_units: %llu\n",
1694 (unsigned long long) fsinfo.full_size_information.out.call_avail_alloc_units);
1695 d_printf("\tactual_avail_alloc_units: %llu\n",
1696 (unsigned long long) fsinfo.full_size_information.out.actual_avail_alloc_units);
1697 d_printf("\tsectors_per_unit: %lu\n",
1698 (unsigned long) fsinfo.full_size_information.out.sectors_per_unit);
1699 d_printf("\tbytes_per_sector: %lu\n",
1700 (unsigned long) fsinfo.full_size_information.out.bytes_per_sector);
1701 break;
1702 case RAW_QFS_OBJECTID_INFORMATION:
1703 d_printf("\tGUID: %s\n",
1704 GUID_string(ctx,&fsinfo.objectid_information.out.guid));
1705 d_printf("\tunknown[6]: [%llu,%llu,%llu,%llu,%llu,%llu]\n",
1706 (unsigned long long) fsinfo.objectid_information.out.unknown[0],
1707 (unsigned long long) fsinfo.objectid_information.out.unknown[1],
1708 (unsigned long long) fsinfo.objectid_information.out.unknown[2],
1709 (unsigned long long) fsinfo.objectid_information.out.unknown[3],
1710 (unsigned long long) fsinfo.objectid_information.out.unknown[4],
1711 (unsigned long long) fsinfo.objectid_information.out.unknown[5] );
1712 break;
1713 case RAW_QFS_GENERIC:
1714 d_printf("\twrong level returned\n");
1715 break;
1718 return 0;
1721 /****************************************************************************
1722 show as much information as possible about a file
1723 ****************************************************************************/
1724 static int cmd_allinfo(struct smbclient_context *ctx, const char **args)
1726 char *fname;
1727 union smb_fileinfo finfo;
1728 NTSTATUS status;
1730 if (!args[1]) {
1731 d_printf("allinfo <filename>\n");
1732 return 1;
1734 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1736 /* first a ALL_INFO QPATHINFO */
1737 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1738 finfo.generic.in.file.path = fname;
1739 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1740 if (!NT_STATUS_IS_OK(status)) {
1741 d_printf("%s - %s\n", fname, nt_errstr(status));
1742 return 1;
1745 d_printf("\tcreate_time: %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1746 d_printf("\taccess_time: %s\n", nt_time_string(ctx, finfo.all_info.out.access_time));
1747 d_printf("\twrite_time: %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1748 d_printf("\tchange_time: %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1749 d_printf("\tattrib: 0x%x\n", finfo.all_info.out.attrib);
1750 d_printf("\talloc_size: %lu\n", (unsigned long)finfo.all_info.out.alloc_size);
1751 d_printf("\tsize: %lu\n", (unsigned long)finfo.all_info.out.size);
1752 d_printf("\tnlink: %u\n", finfo.all_info.out.nlink);
1753 d_printf("\tdelete_pending: %u\n", finfo.all_info.out.delete_pending);
1754 d_printf("\tdirectory: %u\n", finfo.all_info.out.directory);
1755 d_printf("\tea_size: %u\n", finfo.all_info.out.ea_size);
1756 d_printf("\tfname: '%s'\n", finfo.all_info.out.fname.s);
1758 /* 8.3 name if any */
1759 finfo.generic.level = RAW_FILEINFO_ALT_NAME_INFO;
1760 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1761 if (NT_STATUS_IS_OK(status)) {
1762 d_printf("\talt_name: %s\n", finfo.alt_name_info.out.fname.s);
1765 /* file_id if available */
1766 finfo.generic.level = RAW_FILEINFO_INTERNAL_INFORMATION;
1767 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1768 if (NT_STATUS_IS_OK(status)) {
1769 d_printf("\tfile_id %.0f\n",
1770 (double)finfo.internal_information.out.file_id);
1773 /* the EAs, if any */
1774 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1775 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1776 if (NT_STATUS_IS_OK(status)) {
1777 int i;
1778 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1779 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1780 finfo.all_eas.out.eas[i].flags,
1781 (int)finfo.all_eas.out.eas[i].value.length,
1782 finfo.all_eas.out.eas[i].name.s);
1786 /* streams, if available */
1787 finfo.generic.level = RAW_FILEINFO_STREAM_INFO;
1788 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1789 if (NT_STATUS_IS_OK(status)) {
1790 int i;
1791 for (i=0;i<finfo.stream_info.out.num_streams;i++) {
1792 d_printf("\tstream %d:\n", i);
1793 d_printf("\t\tsize %ld\n",
1794 (long)finfo.stream_info.out.streams[i].size);
1795 d_printf("\t\talloc size %ld\n",
1796 (long)finfo.stream_info.out.streams[i].alloc_size);
1797 d_printf("\t\tname %s\n", finfo.stream_info.out.streams[i].stream_name.s);
1801 /* dev/inode if available */
1802 finfo.generic.level = RAW_FILEINFO_COMPRESSION_INFORMATION;
1803 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1804 if (NT_STATUS_IS_OK(status)) {
1805 d_printf("\tcompressed size %ld\n", (long)finfo.compression_info.out.compressed_size);
1806 d_printf("\tformat %ld\n", (long)finfo.compression_info.out.format);
1807 d_printf("\tunit_shift %ld\n", (long)finfo.compression_info.out.unit_shift);
1808 d_printf("\tchunk_shift %ld\n", (long)finfo.compression_info.out.chunk_shift);
1809 d_printf("\tcluster_shift %ld\n", (long)finfo.compression_info.out.cluster_shift);
1812 return 0;
1816 /****************************************************************************
1817 shows EA contents
1818 ****************************************************************************/
1819 static int cmd_eainfo(struct smbclient_context *ctx, const char **args)
1821 char *fname;
1822 union smb_fileinfo finfo;
1823 NTSTATUS status;
1824 int i;
1826 if (!args[1]) {
1827 d_printf("eainfo <filename>\n");
1828 return 1;
1830 fname = talloc_strdup(ctx, args[1]);
1832 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1833 finfo.generic.in.file.path = fname;
1834 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1836 if (!NT_STATUS_IS_OK(status)) {
1837 d_printf("RAW_FILEINFO_ALL_EAS - %s\n", nt_errstr(status));
1838 return 1;
1841 d_printf("%s has %d EAs\n", fname, finfo.all_eas.out.num_eas);
1843 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1844 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1845 finfo.all_eas.out.eas[i].flags,
1846 (int)finfo.all_eas.out.eas[i].value.length,
1847 finfo.all_eas.out.eas[i].name.s);
1848 fflush(stdout);
1849 dump_data(0,
1850 finfo.all_eas.out.eas[i].value.data,
1851 finfo.all_eas.out.eas[i].value.length);
1854 return 0;
1858 /****************************************************************************
1859 show any ACL on a file
1860 ****************************************************************************/
1861 static int cmd_acl(struct smbclient_context *ctx, const char **args)
1863 char *fname;
1864 union smb_fileinfo query;
1865 NTSTATUS status;
1866 int fnum;
1868 if (!args[1]) {
1869 d_printf("acl <filename>\n");
1870 return 1;
1872 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1874 fnum = smbcli_nt_create_full(ctx->cli->tree, fname, 0,
1875 SEC_STD_READ_CONTROL,
1877 NTCREATEX_SHARE_ACCESS_DELETE|
1878 NTCREATEX_SHARE_ACCESS_READ|
1879 NTCREATEX_SHARE_ACCESS_WRITE,
1880 NTCREATEX_DISP_OPEN,
1881 0, 0);
1882 if (fnum == -1) {
1883 d_printf("%s - %s\n", fname, smbcli_errstr(ctx->cli->tree));
1884 return -1;
1887 query.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
1888 query.query_secdesc.in.file.fnum = fnum;
1889 query.query_secdesc.in.secinfo_flags = 0x7;
1891 status = smb_raw_fileinfo(ctx->cli->tree, ctx, &query);
1892 if (!NT_STATUS_IS_OK(status)) {
1893 d_printf("%s - %s\n", fname, nt_errstr(status));
1894 return 1;
1897 NDR_PRINT_DEBUG(security_descriptor, query.query_secdesc.out.sd);
1899 return 0;
1902 /****************************************************************************
1903 lookup a name or sid
1904 ****************************************************************************/
1905 static int cmd_lookup(struct smbclient_context *ctx, const char **args)
1907 NTSTATUS status;
1908 struct dom_sid *sid;
1910 if (!args[1]) {
1911 d_printf("lookup <sid|name>\n");
1912 return 1;
1915 sid = dom_sid_parse_talloc(ctx, args[1]);
1916 if (sid == NULL) {
1917 const char *sidstr;
1918 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sidstr);
1919 if (!NT_STATUS_IS_OK(status)) {
1920 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
1921 return 1;
1924 d_printf("%s\n", sidstr);
1925 } else {
1926 const char *name;
1927 status = smblsa_lookup_sid(ctx->cli, args[1], ctx, &name);
1928 if (!NT_STATUS_IS_OK(status)) {
1929 d_printf("lsa_LookupSids - %s\n", nt_errstr(status));
1930 return 1;
1933 d_printf("%s\n", name);
1936 return 0;
1939 /****************************************************************************
1940 show privileges for a user
1941 ****************************************************************************/
1942 static int cmd_privileges(struct smbclient_context *ctx, const char **args)
1944 NTSTATUS status;
1945 struct dom_sid *sid;
1946 struct lsa_RightSet rights;
1947 unsigned i;
1949 if (!args[1]) {
1950 d_printf("privileges <sid|name>\n");
1951 return 1;
1954 sid = dom_sid_parse_talloc(ctx, args[1]);
1955 if (sid == NULL) {
1956 const char *sid_str;
1957 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
1958 if (!NT_STATUS_IS_OK(status)) {
1959 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
1960 return 1;
1962 sid = dom_sid_parse_talloc(ctx, sid_str);
1965 status = smblsa_sid_privileges(ctx->cli, sid, ctx, &rights);
1966 if (!NT_STATUS_IS_OK(status)) {
1967 d_printf("lsa_EnumAccountRights - %s\n", nt_errstr(status));
1968 return 1;
1971 for (i=0;i<rights.count;i++) {
1972 d_printf("\t%s\n", rights.names[i].string);
1975 return 0;
1979 /****************************************************************************
1980 add privileges for a user
1981 ****************************************************************************/
1982 static int cmd_addprivileges(struct smbclient_context *ctx, const char **args)
1984 NTSTATUS status;
1985 struct dom_sid *sid;
1986 struct lsa_RightSet rights;
1987 int i;
1989 if (!args[1]) {
1990 d_printf("addprivileges <sid|name> <privilege...>\n");
1991 return 1;
1994 sid = dom_sid_parse_talloc(ctx, args[1]);
1995 if (sid == NULL) {
1996 const char *sid_str;
1997 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
1998 if (!NT_STATUS_IS_OK(status)) {
1999 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2000 return 1;
2002 sid = dom_sid_parse_talloc(ctx, sid_str);
2005 ZERO_STRUCT(rights);
2006 for (i = 2; args[i]; i++) {
2007 rights.names = talloc_realloc(ctx, rights.names,
2008 struct lsa_StringLarge, rights.count+1);
2009 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2010 rights.count++;
2014 status = smblsa_sid_add_privileges(ctx->cli, sid, ctx, &rights);
2015 if (!NT_STATUS_IS_OK(status)) {
2016 d_printf("lsa_AddAccountRights - %s\n", nt_errstr(status));
2017 return 1;
2020 return 0;
2023 /****************************************************************************
2024 delete privileges for a user
2025 ****************************************************************************/
2026 static int cmd_delprivileges(struct smbclient_context *ctx, const char **args)
2028 NTSTATUS status;
2029 struct dom_sid *sid;
2030 struct lsa_RightSet rights;
2031 int i;
2033 if (!args[1]) {
2034 d_printf("delprivileges <sid|name> <privilege...>\n");
2035 return 1;
2038 sid = dom_sid_parse_talloc(ctx, args[1]);
2039 if (sid == NULL) {
2040 const char *sid_str;
2041 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2042 if (!NT_STATUS_IS_OK(status)) {
2043 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2044 return 1;
2046 sid = dom_sid_parse_talloc(ctx, sid_str);
2049 ZERO_STRUCT(rights);
2050 for (i = 2; args[i]; i++) {
2051 rights.names = talloc_realloc(ctx, rights.names,
2052 struct lsa_StringLarge, rights.count+1);
2053 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2054 rights.count++;
2058 status = smblsa_sid_del_privileges(ctx->cli, sid, ctx, &rights);
2059 if (!NT_STATUS_IS_OK(status)) {
2060 d_printf("lsa_RemoveAccountRights - %s\n", nt_errstr(status));
2061 return 1;
2064 return 0;
2068 /****************************************************************************
2069 ****************************************************************************/
2070 static int cmd_open(struct smbclient_context *ctx, const char **args)
2072 char *mask;
2074 if (!args[1]) {
2075 d_printf("open <filename>\n");
2076 return 1;
2078 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2080 smbcli_open(ctx->cli->tree, mask, O_RDWR, DENY_ALL);
2082 return 0;
2086 /****************************************************************************
2087 remove a directory
2088 ****************************************************************************/
2089 static int cmd_rmdir(struct smbclient_context *ctx, const char **args)
2091 char *mask;
2093 if (!args[1]) {
2094 d_printf("rmdir <dirname>\n");
2095 return 1;
2097 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2099 if (NT_STATUS_IS_ERR(smbcli_rmdir(ctx->cli->tree, mask))) {
2100 d_printf("%s removing remote directory file %s\n",
2101 smbcli_errstr(ctx->cli->tree),mask);
2104 return 0;
2107 /****************************************************************************
2108 UNIX hardlink.
2109 ****************************************************************************/
2110 static int cmd_link(struct smbclient_context *ctx, const char **args)
2112 char *src,*dest;
2114 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2115 d_printf("Server doesn't support UNIX CIFS calls.\n");
2116 return 1;
2120 if (!args[1] || !args[2]) {
2121 d_printf("link <src> <dest>\n");
2122 return 1;
2125 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2126 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2128 if (NT_STATUS_IS_ERR(smbcli_unix_hardlink(ctx->cli->tree, src, dest))) {
2129 d_printf("%s linking files (%s -> %s)\n", smbcli_errstr(ctx->cli->tree), src, dest);
2130 return 1;
2133 return 0;
2136 /****************************************************************************
2137 UNIX symlink.
2138 ****************************************************************************/
2140 static int cmd_symlink(struct smbclient_context *ctx, const char **args)
2142 char *src,*dest;
2144 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2145 d_printf("Server doesn't support UNIX CIFS calls.\n");
2146 return 1;
2149 if (!args[1] || !args[2]) {
2150 d_printf("symlink <src> <dest>\n");
2151 return 1;
2154 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2155 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2157 if (NT_STATUS_IS_ERR(smbcli_unix_symlink(ctx->cli->tree, src, dest))) {
2158 d_printf("%s symlinking files (%s -> %s)\n",
2159 smbcli_errstr(ctx->cli->tree), src, dest);
2160 return 1;
2163 return 0;
2166 /****************************************************************************
2167 UNIX chmod.
2168 ****************************************************************************/
2170 static int cmd_chmod(struct smbclient_context *ctx, const char **args)
2172 char *src;
2173 mode_t mode;
2175 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2176 d_printf("Server doesn't support UNIX CIFS calls.\n");
2177 return 1;
2180 if (!args[1] || !args[2]) {
2181 d_printf("chmod mode file\n");
2182 return 1;
2185 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2187 mode = (mode_t)strtol(args[1], NULL, 8);
2189 if (NT_STATUS_IS_ERR(smbcli_unix_chmod(ctx->cli->tree, src, mode))) {
2190 d_printf("%s chmod file %s 0%o\n",
2191 smbcli_errstr(ctx->cli->tree), src, (mode_t)mode);
2192 return 1;
2195 return 0;
2198 /****************************************************************************
2199 UNIX chown.
2200 ****************************************************************************/
2202 static int cmd_chown(struct smbclient_context *ctx, const char **args)
2204 char *src;
2205 uid_t uid;
2206 gid_t gid;
2208 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2209 d_printf("Server doesn't support UNIX CIFS calls.\n");
2210 return 1;
2213 if (!args[1] || !args[2] || !args[3]) {
2214 d_printf("chown uid gid file\n");
2215 return 1;
2218 uid = (uid_t)atoi(args[1]);
2219 gid = (gid_t)atoi(args[2]);
2220 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[3]);
2222 if (NT_STATUS_IS_ERR(smbcli_unix_chown(ctx->cli->tree, src, uid, gid))) {
2223 d_printf("%s chown file %s uid=%d, gid=%d\n",
2224 smbcli_errstr(ctx->cli->tree), src, (int)uid, (int)gid);
2225 return 1;
2228 return 0;
2231 /****************************************************************************
2232 rename some files
2233 ****************************************************************************/
2234 static int cmd_rename(struct smbclient_context *ctx, const char **args)
2236 char *src,*dest;
2238 if (!args[1] || !args[2]) {
2239 d_printf("rename <src> <dest>\n");
2240 return 1;
2243 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2244 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2246 if (NT_STATUS_IS_ERR(smbcli_rename(ctx->cli->tree, src, dest))) {
2247 d_printf("%s renaming files\n",smbcli_errstr(ctx->cli->tree));
2248 return 1;
2251 return 0;
2255 /****************************************************************************
2256 toggle the prompt flag
2257 ****************************************************************************/
2258 static int cmd_prompt(struct smbclient_context *ctx, const char **args)
2260 ctx->prompt = !ctx->prompt;
2261 DEBUG(2,("prompting is now %s\n",ctx->prompt?"on":"off"));
2263 return 1;
2267 /****************************************************************************
2268 set the newer than time
2269 ****************************************************************************/
2270 static int cmd_newer(struct smbclient_context *ctx, const char **args)
2272 struct stat sbuf;
2274 if (args[1] && (stat(args[1],&sbuf) == 0)) {
2275 ctx->newer_than = sbuf.st_mtime;
2276 DEBUG(1,("Getting files newer than %s",
2277 asctime(localtime(&ctx->newer_than))));
2278 } else {
2279 ctx->newer_than = 0;
2282 if (args[1] && ctx->newer_than == 0) {
2283 d_printf("Error setting newer-than time\n");
2284 return 1;
2287 return 0;
2290 /****************************************************************************
2291 set the archive level
2292 ****************************************************************************/
2293 static int cmd_archive(struct smbclient_context *ctx, const char **args)
2295 if (args[1]) {
2296 ctx->archive_level = atoi(args[1]);
2297 } else
2298 d_printf("Archive level is %d\n",ctx->archive_level);
2300 return 0;
2303 /****************************************************************************
2304 toggle the lowercaseflag
2305 ****************************************************************************/
2306 static int cmd_lowercase(struct smbclient_context *ctx, const char **args)
2308 ctx->lowercase = !ctx->lowercase;
2309 DEBUG(2,("filename lowercasing is now %s\n",ctx->lowercase?"on":"off"));
2311 return 0;
2317 /****************************************************************************
2318 toggle the recurse flag
2319 ****************************************************************************/
2320 static int cmd_recurse(struct smbclient_context *ctx, const char **args)
2322 ctx->recurse = !ctx->recurse;
2323 DEBUG(2,("directory recursion is now %s\n",ctx->recurse?"on":"off"));
2325 return 0;
2328 /****************************************************************************
2329 toggle the translate flag
2330 ****************************************************************************/
2331 static int cmd_translate(struct smbclient_context *ctx, const char **args)
2333 ctx->translation = !ctx->translation;
2334 DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
2335 ctx->translation?"on":"off"));
2337 return 0;
2341 /****************************************************************************
2342 do a printmode command
2343 ****************************************************************************/
2344 static int cmd_printmode(struct smbclient_context *ctx, const char **args)
2346 if (args[1]) {
2347 if (strequal(args[1],"text")) {
2348 ctx->printmode = 0;
2349 } else {
2350 if (strequal(args[1],"graphics"))
2351 ctx->printmode = 1;
2352 else
2353 ctx->printmode = atoi(args[1]);
2357 switch(ctx->printmode)
2359 case 0:
2360 DEBUG(2,("the printmode is now text\n"));
2361 break;
2362 case 1:
2363 DEBUG(2,("the printmode is now graphics\n"));
2364 break;
2365 default:
2366 DEBUG(2,("the printmode is now %d\n", ctx->printmode));
2367 break;
2370 return 0;
2373 /****************************************************************************
2374 do the lcd command
2375 ****************************************************************************/
2376 static int cmd_lcd(struct smbclient_context *ctx, const char **args)
2378 char d[PATH_MAX];
2380 if (args[1])
2381 chdir(args[1]);
2382 DEBUG(2,("the local directory is now %s\n",getcwd(d, PATH_MAX)));
2384 return 0;
2387 /****************************************************************************
2388 history
2389 ****************************************************************************/
2390 static int cmd_history(struct smbclient_context *ctx, const char **args)
2392 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
2393 HIST_ENTRY **hlist;
2394 int i;
2396 hlist = history_list();
2398 for (i = 0; hlist && hlist[i]; i++) {
2399 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
2401 #else
2402 DEBUG(0,("no history without readline support\n"));
2403 #endif
2405 return 0;
2408 /****************************************************************************
2409 get a file restarting at end of local file
2410 ****************************************************************************/
2411 static int cmd_reget(struct smbclient_context *ctx, const char **args)
2413 char *local_name;
2414 char *remote_name;
2416 if (!args[1]) {
2417 d_printf("reget <filename>\n");
2418 return 1;
2420 remote_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2421 dos_clean_name(remote_name);
2423 if (args[2])
2424 local_name = talloc_strdup(ctx, args[2]);
2425 else
2426 local_name = talloc_strdup(ctx, args[1]);
2428 return do_get(ctx, remote_name, local_name, True);
2431 /****************************************************************************
2432 put a file restarting at end of local file
2433 ****************************************************************************/
2434 static int cmd_reput(struct smbclient_context *ctx, const char **args)
2436 char *local_name;
2437 char *remote_name;
2439 if (!args[1]) {
2440 d_printf("reput <filename>\n");
2441 return 1;
2443 local_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2445 if (!file_exist(local_name)) {
2446 d_printf("%s does not exist\n", local_name);
2447 return 1;
2450 if (args[2])
2451 remote_name = talloc_strdup(ctx, args[2]);
2452 else
2453 remote_name = talloc_strdup(ctx, args[1]);
2455 dos_clean_name(remote_name);
2457 return do_put(ctx, remote_name, local_name, True);
2462 return a string representing a share type
2464 static const char *share_type_str(uint32_t type)
2466 switch (type & 0xF) {
2467 case STYPE_DISKTREE:
2468 return "Disk";
2469 case STYPE_PRINTQ:
2470 return "Printer";
2471 case STYPE_DEVICE:
2472 return "Device";
2473 case STYPE_IPC:
2474 return "IPC";
2475 default:
2476 return "Unknown";
2482 display a list of shares from a level 1 share enum
2484 static void display_share_result(struct srvsvc_NetShareCtr1 *ctr1)
2486 int i;
2488 for (i=0;i<ctr1->count;i++) {
2489 struct srvsvc_NetShareInfo1 *info = ctr1->array+i;
2491 printf("\t%-15s %-10.10s %s\n",
2492 info->name,
2493 share_type_str(info->type),
2494 info->comment);
2500 /****************************************************************************
2501 try and browse available shares on a host
2502 ****************************************************************************/
2503 static BOOL browse_host(const char *query_host)
2505 struct dcerpc_pipe *p;
2506 char *binding;
2507 NTSTATUS status;
2508 struct srvsvc_NetShareEnumAll r;
2509 uint32_t resume_handle = 0;
2510 TALLOC_CTX *mem_ctx = talloc_init("browse_host");
2511 struct srvsvc_NetShareCtr1 ctr1;
2513 binding = talloc_asprintf(mem_ctx, "ncacn_np:%s", query_host);
2515 status = dcerpc_pipe_connect(mem_ctx, &p, binding,
2516 &dcerpc_table_srvsvc,
2517 cmdline_credentials, NULL);
2518 if (!NT_STATUS_IS_OK(status)) {
2519 d_printf("Failed to connect to %s - %s\n",
2520 binding, nt_errstr(status));
2521 talloc_free(mem_ctx);
2522 return False;
2525 r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
2526 r.in.level = 1;
2527 r.in.ctr.ctr1 = &ctr1;
2528 r.in.max_buffer = ~0;
2529 r.in.resume_handle = &resume_handle;
2531 d_printf("\n\tSharename Type Comment\n");
2532 d_printf("\t--------- ---- -------\n");
2534 do {
2535 ZERO_STRUCT(ctr1);
2536 status = dcerpc_srvsvc_NetShareEnumAll(p, mem_ctx, &r);
2538 if (NT_STATUS_IS_OK(status) &&
2539 (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA) ||
2540 W_ERROR_IS_OK(r.out.result)) &&
2541 r.out.ctr.ctr1) {
2542 display_share_result(r.out.ctr.ctr1);
2543 resume_handle += r.out.ctr.ctr1->count;
2545 } while (NT_STATUS_IS_OK(status) && W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA));
2547 talloc_free(mem_ctx);
2549 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2550 d_printf("Failed NetShareEnumAll %s - %s/%s\n",
2551 binding, nt_errstr(status), win_errstr(r.out.result));
2552 return False;
2555 return False;
2558 /****************************************************************************
2559 try and browse available connections on a host
2560 ****************************************************************************/
2561 static BOOL list_servers(const char *wk_grp)
2563 d_printf("REWRITE: list servers not implemented\n");
2564 return False;
2567 /* Some constants for completing filename arguments */
2569 #define COMPL_NONE 0 /* No completions */
2570 #define COMPL_REMOTE 1 /* Complete remote filename */
2571 #define COMPL_LOCAL 2 /* Complete local filename */
2573 static int cmd_help(struct smbclient_context *ctx, const char **args);
2575 /* This defines the commands supported by this client.
2576 * NOTE: The "!" must be the last one in the list because it's fn pointer
2577 * field is NULL, and NULL in that field is used in process_tok()
2578 * (below) to indicate the end of the list. crh
2580 static struct
2582 const char *name;
2583 int (*fn)(struct smbclient_context *ctx, const char **args);
2584 const char *description;
2585 char compl_args[2]; /* Completion argument info */
2586 } commands[] =
2588 {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2589 {"addprivileges",cmd_addprivileges,"<sid|name> <privilege...> add privileges for a user",{COMPL_NONE,COMPL_NONE}},
2590 {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
2591 {"acl",cmd_acl,"<file> show file ACL",{COMPL_NONE,COMPL_NONE}},
2592 {"allinfo",cmd_allinfo,"<file> show all possible info about a file",{COMPL_NONE,COMPL_NONE}},
2593 {"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}},
2594 {"cancel",cmd_rewrite,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
2595 {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
2596 {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
2597 {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
2598 {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2599 {"delprivileges",cmd_delprivileges,"<sid|name> <privilege...> remove privileges for a user",{COMPL_NONE,COMPL_NONE}},
2600 {"deltree",cmd_deltree,"<dir> delete a whole directory tree",{COMPL_REMOTE,COMPL_NONE}},
2601 {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2602 {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2603 {"eainfo",cmd_eainfo,"<file> show EA contents for a file",{COMPL_NONE,COMPL_NONE}},
2604 {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2605 {"fsinfo",cmd_fsinfo,"query file system info",{COMPL_NONE,COMPL_NONE}},
2606 {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
2607 {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2608 {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
2609 {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
2610 {"link",cmd_link,"<src> <dest> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2611 {"lookup",cmd_lookup,"<sid|name> show SID for name or name for SID",{COMPL_NONE,COMPL_NONE}},
2612 {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},
2613 {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2614 {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
2615 {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2616 {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
2617 {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2618 {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},
2619 {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
2620 {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
2621 {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
2622 {"privileges",cmd_privileges,"<user> show privileges for a user",{COMPL_NONE,COMPL_NONE}},
2623 {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
2624 {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
2625 {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},
2626 {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
2627 {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
2628 {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2629 {"queue",cmd_rewrite,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2630 {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2631 {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2632 {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},
2633 {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
2634 {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2635 {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
2636 {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2637 {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2638 {"symlink",cmd_symlink,"<src> <dest> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2639 {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2641 /* Yes, this must be here, see crh's comment above. */
2642 {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2643 {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2647 /*******************************************************************
2648 lookup a command string in the list of commands, including
2649 abbreviations
2650 ******************************************************************/
2651 static int process_tok(const char *tok)
2653 int i = 0, matches = 0;
2654 int cmd=0;
2655 int tok_len = strlen(tok);
2657 while (commands[i].fn != NULL) {
2658 if (strequal(commands[i].name,tok)) {
2659 matches = 1;
2660 cmd = i;
2661 break;
2662 } else if (strncasecmp(commands[i].name, tok, tok_len) == 0) {
2663 matches++;
2664 cmd = i;
2666 i++;
2669 if (matches == 0)
2670 return(-1);
2671 else if (matches == 1)
2672 return(cmd);
2673 else
2674 return(-2);
2677 /****************************************************************************
2678 help
2679 ****************************************************************************/
2680 static int cmd_help(struct smbclient_context *ctx, const char **args)
2682 int i=0,j;
2684 if (args[1]) {
2685 if ((i = process_tok(args[1])) >= 0)
2686 d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2687 } else {
2688 while (commands[i].description) {
2689 for (j=0; commands[i].description && (j<5); j++) {
2690 d_printf("%-15s",commands[i].name);
2691 i++;
2693 d_printf("\n");
2696 return 0;
2699 static int process_line(struct smbclient_context *ctx, const char *cline);
2700 /****************************************************************************
2701 process a -c command string
2702 ****************************************************************************/
2703 static int process_command_string(struct smbclient_context *ctx, const char *cmd)
2705 const char **lines;
2706 int i, rc = 0;
2708 lines = str_list_make(NULL, cmd, ";");
2709 for (i = 0; lines[i]; i++) {
2710 rc |= process_line(ctx, lines[i]);
2712 talloc_free(lines);
2714 return rc;
2717 #define MAX_COMPLETIONS 100
2719 typedef struct {
2720 char *dirmask;
2721 char **matches;
2722 int count, samelen;
2723 const char *text;
2724 int len;
2725 } completion_remote_t;
2727 static void completion_remote_filter(struct clilist_file_info *f, const char *mask, void *state)
2729 completion_remote_t *info = (completion_remote_t *)state;
2731 if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (!ISDOT(f->name)) && (!ISDOTDOT(f->name))) {
2732 if ((info->dirmask[0] == 0) && !(f->attrib & FILE_ATTRIBUTE_DIRECTORY))
2733 info->matches[info->count] = strdup(f->name);
2734 else {
2735 char *tmp;
2737 if (info->dirmask[0] != 0)
2738 tmp = talloc_asprintf(NULL, "%s/%s", info->dirmask, f->name);
2739 else
2740 tmp = talloc_strdup(NULL, f->name);
2742 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2743 tmp = talloc_append_string(NULL, tmp, "/");
2744 info->matches[info->count] = tmp;
2746 if (info->matches[info->count] == NULL)
2747 return;
2748 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2749 smb_readline_ca_char(0);
2751 if (info->count == 1)
2752 info->samelen = strlen(info->matches[info->count]);
2753 else
2754 while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0)
2755 info->samelen--;
2756 info->count++;
2760 static char **remote_completion(const char *text, int len)
2762 char *dirmask;
2763 int i;
2764 completion_remote_t info;
2766 info.samelen = len;
2767 info.text = text;
2768 info.len = len;
2770 if (len >= PATH_MAX)
2771 return(NULL);
2773 info.matches = malloc_array_p(char *, MAX_COMPLETIONS);
2774 if (!info.matches) return NULL;
2775 info.matches[0] = NULL;
2777 for (i = len-1; i >= 0; i--)
2778 if ((text[i] == '/') || (text[i] == '\\'))
2779 break;
2780 info.text = text+i+1;
2781 info.samelen = info.len = len-i-1;
2783 if (i > 0) {
2784 info.dirmask = talloc_strndup(NULL, text, i+1);
2785 info.dirmask[i+1] = 0;
2786 asprintf(&dirmask, "%s%*s*", rl_ctx->remote_cur_dir, i-1, text);
2787 } else
2788 asprintf(&dirmask, "%s*", rl_ctx->remote_cur_dir);
2790 if (smbcli_list(rl_ctx->cli->tree, dirmask,
2791 FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN,
2792 completion_remote_filter, &info) < 0)
2793 goto cleanup;
2795 if (info.count == 2)
2796 info.matches[0] = strdup(info.matches[1]);
2797 else {
2798 info.matches[0] = malloc(info.samelen+1);
2799 if (!info.matches[0])
2800 goto cleanup;
2801 strncpy(info.matches[0], info.matches[1], info.samelen);
2802 info.matches[0][info.samelen] = 0;
2804 info.matches[info.count] = NULL;
2805 return info.matches;
2807 cleanup:
2808 for (i = 0; i < info.count; i++)
2809 free(info.matches[i]);
2810 free(info.matches);
2811 return NULL;
2814 static char **completion_fn(const char *text, int start, int end)
2816 smb_readline_ca_char(' ');
2818 if (start) {
2819 const char *buf, *sp;
2820 int i;
2821 char compl_type;
2823 buf = smb_readline_get_line_buffer();
2824 if (buf == NULL)
2825 return NULL;
2827 sp = strchr(buf, ' ');
2828 if (sp == NULL)
2829 return NULL;
2831 for (i = 0; commands[i].name; i++)
2832 if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0))
2833 break;
2834 if (commands[i].name == NULL)
2835 return NULL;
2837 while (*sp == ' ')
2838 sp++;
2840 if (sp == (buf + start))
2841 compl_type = commands[i].compl_args[0];
2842 else
2843 compl_type = commands[i].compl_args[1];
2845 if (compl_type == COMPL_REMOTE)
2846 return remote_completion(text, end - start);
2847 else /* fall back to local filename completion */
2848 return NULL;
2849 } else {
2850 char **matches;
2851 int i, len, samelen = 0, count=1;
2853 matches = malloc_array_p(char *, MAX_COMPLETIONS);
2854 if (!matches) return NULL;
2855 matches[0] = NULL;
2857 len = strlen(text);
2858 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
2859 if (strncmp(text, commands[i].name, len) == 0) {
2860 matches[count] = strdup(commands[i].name);
2861 if (!matches[count])
2862 goto cleanup;
2863 if (count == 1)
2864 samelen = strlen(matches[count]);
2865 else
2866 while (strncmp(matches[count], matches[count-1], samelen) != 0)
2867 samelen--;
2868 count++;
2872 switch (count) {
2873 case 0: /* should never happen */
2874 case 1:
2875 goto cleanup;
2876 case 2:
2877 matches[0] = strdup(matches[1]);
2878 break;
2879 default:
2880 matches[0] = malloc(samelen+1);
2881 if (!matches[0])
2882 goto cleanup;
2883 strncpy(matches[0], matches[1], samelen);
2884 matches[0][samelen] = 0;
2886 matches[count] = NULL;
2887 return matches;
2889 cleanup:
2890 while (i >= 0) {
2891 free(matches[i]);
2892 i--;
2894 free(matches);
2895 return NULL;
2899 /****************************************************************************
2900 make sure we swallow keepalives during idle time
2901 ****************************************************************************/
2902 static void readline_callback(void)
2904 static time_t last_t;
2905 time_t t;
2907 t = time(NULL);
2909 if (t - last_t < 5) return;
2911 last_t = t;
2913 smbcli_transport_process(rl_ctx->cli->transport);
2915 if (rl_ctx->cli->tree) {
2916 smbcli_chkpath(rl_ctx->cli->tree, "\\");
2920 static int process_line(struct smbclient_context *ctx, const char *cline)
2922 const char **args;
2923 int i;
2925 /* and get the first part of the command */
2926 args = str_list_make_shell(ctx, cline, NULL);
2927 if (!args || !args[0])
2928 return 0;
2930 if ((i = process_tok(args[0])) >= 0) {
2931 i = commands[i].fn(ctx, args);
2932 } else if (i == -2) {
2933 d_printf("%s: command abbreviation ambiguous\n",args[0]);
2934 } else {
2935 d_printf("%s: command not found\n",args[0]);
2938 talloc_free(args);
2940 return i;
2943 /****************************************************************************
2944 process commands on stdin
2945 ****************************************************************************/
2946 static int process_stdin(struct smbclient_context *ctx)
2948 int rc = 0;
2949 while (1) {
2950 /* display a prompt */
2951 char *the_prompt = talloc_asprintf(ctx, "smb: %s> ", ctx->remote_cur_dir);
2952 char *cline = smb_readline(the_prompt, readline_callback, completion_fn);
2953 talloc_free(the_prompt);
2955 if (!cline) break;
2957 /* special case - first char is ! */
2958 if (*cline == '!') {
2959 system(cline + 1);
2960 continue;
2963 rc |= process_command_string(ctx, cline);
2966 return rc;
2970 /*****************************************************
2971 return a connection to a server
2972 *******************************************************/
2973 static struct smbclient_context *do_connect(TALLOC_CTX *mem_ctx,
2974 const char *specified_server, const char *specified_share, struct cli_credentials *cred)
2976 NTSTATUS status;
2977 struct smbclient_context *ctx = talloc_zero(mem_ctx, struct smbclient_context);
2978 char *server, *share;
2980 if (!ctx) {
2981 return NULL;
2984 rl_ctx = ctx; /* Ugly hack */
2986 if (strncmp(specified_share, "\\\\", 2) == 0 ||
2987 strncmp(specified_share, "//", 2) == 0) {
2988 smbcli_parse_unc(specified_share, ctx, &server, &share);
2989 } else {
2990 share = talloc_strdup(ctx, specified_share);
2991 server = talloc_strdup(ctx, specified_server);
2994 ctx->remote_cur_dir = talloc_strdup(ctx, "\\");
2996 status = smbcli_full_connection(ctx, &ctx->cli, server,
2997 share, NULL, cred, NULL);
2998 if (!NT_STATUS_IS_OK(status)) {
2999 d_printf("Connection to \\\\%s\\%s failed - %s\n",
3000 server, share, nt_errstr(status));
3001 talloc_free(ctx);
3002 return NULL;
3005 return ctx;
3008 /****************************************************************************
3009 handle a -L query
3010 ****************************************************************************/
3011 static int do_host_query(const char *query_host)
3013 browse_host(query_host);
3014 list_servers(lp_workgroup());
3015 return(0);
3019 /****************************************************************************
3020 handle a message operation
3021 ****************************************************************************/
3022 static int do_message_op(const char *desthost, const char *destip, int name_type)
3024 struct nbt_name called, calling;
3025 const char *server_name;
3026 struct smbcli_state *cli;
3028 make_nbt_name_client(&calling, lp_netbios_name());
3030 nbt_choose_called_name(NULL, &called, desthost, name_type);
3032 server_name = destip ? destip : desthost;
3034 if (!(cli=smbcli_state_init(NULL)) || !smbcli_socket_connect(cli, server_name)) {
3035 d_printf("Connection to %s failed\n", server_name);
3036 return 1;
3039 if (!smbcli_transport_establish(cli, &calling, &called)) {
3040 d_printf("session request failed\n");
3041 talloc_free(cli);
3042 return 1;
3045 send_message(cli, desthost);
3046 talloc_free(cli);
3048 return 0;
3052 /****************************************************************************
3053 main program
3054 ****************************************************************************/
3055 int main(int argc,char *argv[])
3057 const char *base_directory = NULL;
3058 const char *dest_ip = NULL;
3059 int opt;
3060 const char *query_host = NULL;
3061 BOOL message = False;
3062 const char *desthost = NULL;
3063 #ifdef KANJI
3064 const char *term_code = KANJI;
3065 #else
3066 const char *term_code = "";
3067 #endif /* KANJI */
3068 poptContext pc;
3069 const char *service = NULL;
3070 int port = 0;
3071 char *p;
3072 int rc = 0;
3073 int name_type = 0x20;
3074 TALLOC_CTX *mem_ctx;
3075 struct smbclient_context *ctx;
3076 const char *cmdstr = NULL;
3078 struct poptOption long_options[] = {
3079 POPT_AUTOHELP
3081 { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" },
3082 { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" },
3083 { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" },
3084 { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" },
3085 { "terminal", 't', POPT_ARG_STRING, NULL, 't', "Terminal I/O code {sjis|euc|jis7|jis8|junet|hex}", "CODE" },
3086 { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" },
3087 { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" },
3088 { "send-buffer", 'b', POPT_ARG_INT, NULL, 'b', "Changes the transmit/send buffer", "BYTES" },
3089 { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" },
3090 POPT_COMMON_SAMBA
3091 POPT_COMMON_CONNECTION
3092 POPT_COMMON_CREDENTIALS
3093 POPT_COMMON_VERSION
3094 { NULL }
3097 mem_ctx = talloc_init("client.c/main");
3098 if (!mem_ctx) {
3099 d_printf("\nclient.c: Not enough memory\n");
3100 exit(1);
3103 pc = poptGetContext("smbclient", argc, (const char **) argv, long_options, 0);
3104 poptSetOtherOptionHelp(pc, "[OPTIONS] service <password>");
3106 while ((opt = poptGetNextOpt(pc)) != -1) {
3107 switch (opt) {
3108 case 'M':
3109 /* Messages are sent to NetBIOS name type 0x3
3110 * (Messenger Service). Make sure we default
3111 * to port 139 instead of port 445. srl,crh
3113 name_type = 0x03;
3114 desthost = strdup(poptGetOptArg(pc));
3115 if( 0 == port ) port = 139;
3116 message = True;
3117 break;
3118 case 'I':
3119 dest_ip = poptGetOptArg(pc);
3120 break;
3121 case 'L':
3122 query_host = strdup(poptGetOptArg(pc));
3123 break;
3124 case 't':
3125 term_code = strdup(poptGetOptArg(pc));
3126 break;
3127 case 'D':
3128 base_directory = strdup(poptGetOptArg(pc));
3129 break;
3130 case 'b':
3131 io_bufsize = MAX(1, atoi(poptGetOptArg(pc)));
3132 break;
3136 gensec_init();
3138 if(poptPeekArg(pc)) {
3139 char *s = strdup(poptGetArg(pc));
3141 /* Convert any '/' characters in the service name to '\' characters */
3142 string_replace(s, '/','\\');
3144 service = s;
3146 if (count_chars(s,'\\') < 3) {
3147 d_printf("\n%s: Not enough '\\' characters in service\n",s);
3148 poptPrintUsage(pc, stderr, 0);
3149 exit(1);
3153 if (poptPeekArg(pc)) {
3154 cli_credentials_set_password(cmdline_credentials, poptGetArg(pc), CRED_SPECIFIED);
3157 /*init_names(); */
3159 if (!query_host && !service && !message) {
3160 poptPrintUsage(pc, stderr, 0);
3161 exit(1);
3164 poptFreeContext(pc);
3166 DEBUG( 3, ( "Client started (version %s).\n", SAMBA_VERSION_STRING ) );
3168 if (query_host && (p=strchr_m(query_host,'#'))) {
3169 *p = 0;
3170 p++;
3171 sscanf(p, "%x", &name_type);
3174 if (query_host) {
3175 return do_host_query(query_host);
3178 if (message) {
3179 return do_message_op(desthost, dest_ip, name_type);
3183 ctx = do_connect(mem_ctx, desthost, service, cmdline_credentials);
3184 if (!ctx)
3185 return 1;
3187 if (base_directory)
3188 do_cd(ctx, base_directory);
3190 if (cmdstr) {
3191 rc = process_command_string(ctx, cmdstr);
3192 } else {
3193 rc = process_stdin(ctx);
3196 talloc_free(mem_ctx);
3198 return rc;