py:dcerpc/raw_testcase: pass auth_context and stub_len to parse_auth() in order to...
[Samba.git] / source4 / client / client.c
blob284511fb58a825fa39bf9faa7ee29fc8b7ee13ac
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 3 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, see <http://www.gnu.org/licenses/>.
23 /*
24 * TODO: remove this ... and don't use talloc_append_string()
26 * NOTE: I'm not changing the code yet, because I assume there're
27 * some bugs in the existing code and I'm not sure how to fix
28 * them correctly.
30 #define TALLOC_DEPRECATED 1
32 #include "includes.h"
33 #include "version.h"
34 #include "libcli/libcli.h"
35 #include "lib/events/events.h"
36 #include "lib/cmdline/popt_common.h"
37 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
38 #include "librpc/gen_ndr/ndr_lsa.h"
39 #include "librpc/gen_ndr/ndr_security.h"
40 #include "libcli/util/clilsa.h"
41 #include "system/dir.h"
42 #include "system/filesys.h"
43 #include "../lib/util/dlinklist.h"
44 #include "system/readline.h"
45 #include "auth/credentials/credentials.h"
46 #include "auth/gensec/gensec.h"
47 #include "system/time.h" /* needed by some systems for asctime() */
48 #include "libcli/resolve/resolve.h"
49 #include "libcli/security/security.h"
50 #include "../libcli/smbreadline/smbreadline.h"
51 #include "librpc/gen_ndr/ndr_nbt.h"
52 #include "param/param.h"
53 #include "libcli/raw/raw_proto.h"
55 /* the default pager to use for the client "more" command. Users can
56 * override this with the PAGER environment variable */
57 #ifndef DEFAULT_PAGER
58 #define DEFAULT_PAGER "more"
59 #endif
61 struct smbclient_context {
62 char *remote_cur_dir;
63 struct smbcli_state *cli;
64 char *fileselection;
65 time_t newer_than;
66 bool prompt;
67 bool recurse;
68 int archive_level;
69 bool lowercase;
70 int printmode;
71 bool translation;
72 int io_bufsize;
75 /* timing globals */
76 static uint64_t get_total_size = 0;
77 static unsigned int get_total_time_ms = 0;
78 static uint64_t put_total_size = 0;
79 static unsigned int put_total_time_ms = 0;
81 /* Unfortunately, there is no way to pass the a context to the completion function as an argument */
82 static struct smbclient_context *rl_ctx;
84 /* totals globals */
85 static double dir_total;
87 /*******************************************************************
88 Reduce a file name, removing .. elements.
89 ********************************************************************/
90 static void dos_clean_name(char *s)
92 char *p=NULL,*r;
94 DEBUG(3,("dos_clean_name [%s]\n",s));
96 /* remove any double slashes */
97 all_string_sub(s, "\\\\", "\\", 0);
99 while ((p = strstr(s,"\\..\\")) != NULL) {
100 *p = '\0';
101 if ((r = strrchr(s,'\\')) != NULL)
102 memmove(r,p+3,strlen(p+3)+1);
105 trim_string(s,NULL,"\\..");
107 all_string_sub(s, "\\.\\", "\\", 0);
110 /****************************************************************************
111 write to a local file with CR/LF->LF translation if appropriate. return the
112 number taken from the buffer. This may not equal the number written.
113 ****************************************************************************/
114 static int writefile(int f, const void *_b, int n, bool translation)
116 const uint8_t *b = (const uint8_t *)_b;
117 int i;
119 if (!translation) {
120 return write(f,b,n);
123 i = 0;
124 while (i < n) {
125 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
126 b++;i++;
128 if (write(f, b, 1) != 1) {
129 break;
131 b++;
132 i++;
135 return(i);
138 /****************************************************************************
139 read from a file with LF->CR/LF translation if appropriate. return the
140 number read. read approx n bytes.
141 ****************************************************************************/
142 static int readfile(void *_b, int n, FILE *f, bool translation)
144 uint8_t *b = (uint8_t *)_b;
145 int i;
146 int c;
148 if (!translation)
149 return fread(b,1,n,f);
151 i = 0;
152 while (i < (n - 1)) {
153 if ((c = getc(f)) == EOF) {
154 break;
157 if (c == '\n') { /* change all LFs to CR/LF */
158 b[i++] = '\r';
161 b[i++] = c;
164 return(i);
168 /****************************************************************************
169 send a message
170 ****************************************************************************/
171 static void send_message(struct smbcli_state *cli, const char *desthost)
173 char msg[1600];
174 int total_len = 0;
175 int grp_id;
177 if (!smbcli_message_start(cli->tree,
178 desthost,
179 cli_credentials_get_username(
180 popt_get_cmdline_credentials()),
181 &grp_id)) {
182 d_printf("message start: %s\n", smbcli_errstr(cli->tree));
183 return;
187 d_printf("Connected. Type your message, ending it with a Control-D\n");
189 while (!feof(stdin) && total_len < 1600) {
190 int maxlen = MIN(1600 - total_len,127);
191 int l=0;
192 int c;
194 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
195 if (c == '\n')
196 msg[l++] = '\r';
197 msg[l] = c;
200 if (!smbcli_message_text(cli->tree, msg, l, grp_id)) {
201 d_printf("SMBsendtxt failed (%s)\n",smbcli_errstr(cli->tree));
202 return;
205 total_len += l;
208 if (total_len >= 1600)
209 d_printf("the message was truncated to 1600 bytes\n");
210 else
211 d_printf("sent %d bytes\n",total_len);
213 if (!smbcli_message_end(cli->tree, grp_id)) {
214 d_printf("SMBsendend failed (%s)\n",smbcli_errstr(cli->tree));
215 return;
221 /****************************************************************************
222 check the space on a device
223 ****************************************************************************/
224 static int do_dskattr(struct smbclient_context *ctx)
226 uint32_t bsize;
227 uint64_t total, avail;
229 if (NT_STATUS_IS_ERR(smbcli_dskattr(ctx->cli->tree, &bsize, &total, &avail))) {
230 d_printf("Error in dskattr: %s\n",smbcli_errstr(ctx->cli->tree));
231 return 1;
234 d_printf("\n\t\t%llu blocks of size %u. %llu blocks available\n",
235 (unsigned long long)total,
236 (unsigned)bsize,
237 (unsigned long long)avail);
239 return 0;
242 /****************************************************************************
243 show cd/pwd
244 ****************************************************************************/
245 static int cmd_pwd(struct smbclient_context *ctx, const char **args)
247 d_printf("Current directory is %s\n", ctx->remote_cur_dir);
248 return 0;
252 convert a string to dos format
254 static void dos_format(char *s)
256 string_replace(s, '/', '\\');
259 /****************************************************************************
260 change directory - inner section
261 ****************************************************************************/
262 static int do_cd(struct smbclient_context *ctx, const char *newdir)
264 char *dname;
266 /* Save the current directory in case the
267 new directory is invalid */
268 if (newdir[0] == '\\')
269 dname = talloc_strdup(ctx, newdir);
270 else
271 dname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, newdir);
273 dos_format(dname);
275 if (*(dname+strlen(dname)-1) != '\\') {
276 dname = talloc_append_string(NULL, dname, "\\");
278 dos_clean_name(dname);
280 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, dname))) {
281 d_printf("cd %s: %s\n", dname, smbcli_errstr(ctx->cli->tree));
282 talloc_free(dname);
283 } else {
284 ctx->remote_cur_dir = dname;
287 return 0;
290 /****************************************************************************
291 change directory
292 ****************************************************************************/
293 static int cmd_cd(struct smbclient_context *ctx, const char **args)
295 int rc = 0;
297 if (args[1])
298 rc = do_cd(ctx, args[1]);
299 else
300 d_printf("Current directory is %s\n",ctx->remote_cur_dir);
302 return rc;
306 static bool mask_match(struct smbcli_state *c, const char *string,
307 const char *pattern, bool is_case_sensitive)
309 int ret;
311 if (ISDOTDOT(string))
312 string = ".";
313 if (ISDOT(pattern))
314 return false;
316 ret = ms_fnmatch_protocol(pattern, string,
317 c->transport->negotiate.protocol,
318 is_case_sensitive);
319 return (ret == 0);
324 /*******************************************************************
325 decide if a file should be operated on
326 ********************************************************************/
327 static bool do_this_one(struct smbclient_context *ctx, struct clilist_file_info *finfo)
329 if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY) return(true);
331 if (ctx->fileselection &&
332 !mask_match(ctx->cli, finfo->name,ctx->fileselection,false)) {
333 DEBUG(3,("mask_match %s failed\n", finfo->name));
334 return false;
337 if (ctx->newer_than && finfo->mtime < ctx->newer_than) {
338 DEBUG(3,("newer_than %s failed\n", finfo->name));
339 return(false);
342 if ((ctx->archive_level==1 || ctx->archive_level==2) && !(finfo->attrib & FILE_ATTRIBUTE_ARCHIVE)) {
343 DEBUG(3,("archive %s failed\n", finfo->name));
344 return(false);
347 return(true);
350 /****************************************************************************
351 display info about a file
352 ****************************************************************************/
353 static void display_finfo(struct smbclient_context *ctx, struct clilist_file_info *finfo)
355 if (do_this_one(ctx, finfo)) {
356 time_t t = finfo->mtime; /* the time is assumed to be passed as GMT */
357 char *astr = attrib_string(NULL, finfo->attrib);
358 d_printf(" %-30s%7.7s %8.0f %s",
359 finfo->name,
360 astr,
361 (double)finfo->size,
362 asctime(localtime(&t)));
363 dir_total += finfo->size;
364 talloc_free(astr);
369 /****************************************************************************
370 accumulate size of a file
371 ****************************************************************************/
372 static void do_du(struct smbclient_context *ctx, struct clilist_file_info *finfo)
374 if (do_this_one(ctx, finfo)) {
375 dir_total += finfo->size;
379 static bool do_list_recurse;
380 static bool do_list_dirs;
381 static char *do_list_queue = 0;
382 static long do_list_queue_size = 0;
383 static long do_list_queue_start = 0;
384 static long do_list_queue_end = 0;
385 static void (*do_list_fn)(struct smbclient_context *, struct clilist_file_info *);
387 /****************************************************************************
388 functions for do_list_queue
389 ****************************************************************************/
392 * The do_list_queue is a NUL-separated list of strings stored in a
393 * char*. Since this is a FIFO, we keep track of the beginning and
394 * ending locations of the data in the queue. When we overflow, we
395 * double the size of the char*. When the start of the data passes
396 * the midpoint, we move everything back. This is logically more
397 * complex than a linked list, but easier from a memory management
398 * angle. In any memory error condition, do_list_queue is reset.
399 * Functions check to ensure that do_list_queue is non-NULL before
400 * accessing it.
402 static void reset_do_list_queue(void)
404 SAFE_FREE(do_list_queue);
405 do_list_queue_size = 0;
406 do_list_queue_start = 0;
407 do_list_queue_end = 0;
410 static void init_do_list_queue(void)
412 reset_do_list_queue();
413 do_list_queue_size = 1024;
414 do_list_queue = malloc_array_p(char, do_list_queue_size);
415 if (do_list_queue == 0) {
416 d_printf("malloc fail for size %d\n",
417 (int)do_list_queue_size);
418 reset_do_list_queue();
419 } else {
420 memset(do_list_queue, 0, do_list_queue_size);
424 static void adjust_do_list_queue(void)
426 if (do_list_queue == NULL) return;
429 * If the starting point of the queue is more than half way through,
430 * move everything toward the beginning.
432 if (do_list_queue_start == do_list_queue_end)
434 DEBUG(4,("do_list_queue is empty\n"));
435 do_list_queue_start = do_list_queue_end = 0;
436 *do_list_queue = '\0';
438 else if (do_list_queue_start > (do_list_queue_size / 2))
440 DEBUG(4,("sliding do_list_queue backward\n"));
441 memmove(do_list_queue,
442 do_list_queue + do_list_queue_start,
443 do_list_queue_end - do_list_queue_start);
444 do_list_queue_end -= do_list_queue_start;
445 do_list_queue_start = 0;
450 static void add_to_do_list_queue(const char* entry)
452 char *dlq;
453 long new_end;
455 if (entry == NULL) {
456 entry = "";
459 new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
460 while (new_end > do_list_queue_size)
462 do_list_queue_size *= 2;
463 DEBUG(4,("enlarging do_list_queue to %d\n",
464 (int)do_list_queue_size));
465 dlq = realloc_p(do_list_queue, char, do_list_queue_size);
466 if (! dlq) {
467 d_printf("failure enlarging do_list_queue to %d bytes\n",
468 (int)do_list_queue_size);
469 reset_do_list_queue();
471 else
473 do_list_queue = dlq;
474 memset(do_list_queue + do_list_queue_size / 2,
475 0, do_list_queue_size / 2);
478 if (do_list_queue)
480 strlcpy(do_list_queue + do_list_queue_end, entry,
481 do_list_queue_size - do_list_queue_end);
482 do_list_queue_end = new_end;
483 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
484 entry, (int)do_list_queue_start, (int)do_list_queue_end));
488 static char *do_list_queue_head(void)
490 return do_list_queue + do_list_queue_start;
493 static void remove_do_list_queue_head(void)
495 if (do_list_queue_end > do_list_queue_start)
497 do_list_queue_start += strlen(do_list_queue_head()) + 1;
498 adjust_do_list_queue();
499 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
500 (int)do_list_queue_start, (int)do_list_queue_end));
504 static int do_list_queue_empty(void)
506 return (! (do_list_queue && *do_list_queue));
509 /****************************************************************************
510 a helper for do_list
511 ****************************************************************************/
512 static void do_list_helper(struct clilist_file_info *f, const char *mask, void *state)
514 struct smbclient_context *ctx = (struct smbclient_context *)state;
516 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY) {
517 if (do_list_dirs && do_this_one(ctx, f)) {
518 do_list_fn(ctx, f);
520 if (do_list_recurse &&
521 !ISDOT(f->name) &&
522 !ISDOTDOT(f->name)) {
523 char *mask2;
524 char *p;
526 mask2 = talloc_strdup(NULL, mask);
527 p = strrchr_m(mask2,'\\');
528 if (!p) return;
529 p[1] = 0;
530 mask2 = talloc_asprintf_append_buffer(mask2, "%s\\*", f->name);
531 add_to_do_list_queue(mask2);
533 return;
536 if (do_this_one(ctx, f)) {
537 do_list_fn(ctx, f);
542 /****************************************************************************
543 a wrapper around smbcli_list that adds recursion
544 ****************************************************************************/
545 static void do_list(struct smbclient_context *ctx, const char *mask,uint16_t attribute,
546 void (*fn)(struct smbclient_context *, struct clilist_file_info *),bool rec, bool dirs)
548 static int in_do_list = 0;
550 if (in_do_list && rec)
552 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
553 exit(1);
556 in_do_list = 1;
558 do_list_recurse = rec;
559 do_list_dirs = dirs;
560 do_list_fn = fn;
562 if (rec)
564 init_do_list_queue();
565 add_to_do_list_queue(mask);
567 while (! do_list_queue_empty())
570 * Need to copy head so that it doesn't become
571 * invalid inside the call to smbcli_list. This
572 * would happen if the list were expanded
573 * during the call.
574 * Fix from E. Jay Berkenbilt (ejb@ql.org)
576 char *head;
577 head = do_list_queue_head();
578 smbcli_list(ctx->cli->tree, head, attribute, do_list_helper, ctx);
579 remove_do_list_queue_head();
580 if ((! do_list_queue_empty()) && (fn == display_finfo))
582 char* next_file = do_list_queue_head();
583 char* save_ch = 0;
584 if ((strlen(next_file) >= 2) &&
585 (next_file[strlen(next_file) - 1] == '*') &&
586 (next_file[strlen(next_file) - 2] == '\\'))
588 save_ch = next_file +
589 strlen(next_file) - 2;
590 *save_ch = '\0';
592 d_printf("\n%s\n",next_file);
593 if (save_ch)
595 *save_ch = '\\';
600 else
602 if (smbcli_list(ctx->cli->tree, mask, attribute, do_list_helper, ctx) == -1)
604 d_printf("%s listing %s\n", smbcli_errstr(ctx->cli->tree), mask);
608 in_do_list = 0;
609 reset_do_list_queue();
612 /****************************************************************************
613 get a directory listing
614 ****************************************************************************/
615 static int cmd_dir(struct smbclient_context *ctx, const char **args)
617 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
618 char *mask;
619 int rc;
621 dir_total = 0;
623 mask = talloc_strdup(ctx, ctx->remote_cur_dir);
624 if(mask[strlen(mask)-1]!='\\')
625 mask = talloc_append_string(ctx, mask,"\\");
627 if (args[1]) {
628 mask = talloc_strdup(ctx, args[1]);
629 if (mask[0] != '\\')
630 mask = talloc_append_string(ctx, mask, "\\");
631 dos_format(mask);
633 else {
634 if (ctx->cli->tree->session->transport->negotiate.protocol <=
635 PROTOCOL_LANMAN1) {
636 mask = talloc_append_string(ctx, mask, "*.*");
637 } else {
638 mask = talloc_append_string(ctx, mask, "*");
642 do_list(ctx, mask, attribute, display_finfo, ctx->recurse, true);
644 rc = do_dskattr(ctx);
646 DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
648 return rc;
652 /****************************************************************************
653 get a directory listing
654 ****************************************************************************/
655 static int cmd_du(struct smbclient_context *ctx, const char **args)
657 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
658 int rc;
659 char *mask;
661 dir_total = 0;
663 if (args[1]) {
664 if (args[1][0] == '\\')
665 mask = talloc_strdup(ctx, args[1]);
666 else
667 mask = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
668 dos_format(mask);
669 } else {
670 mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
673 do_list(ctx, mask, attribute, do_du, ctx->recurse, true);
675 talloc_free(mask);
677 rc = do_dskattr(ctx);
679 d_printf("Total number of bytes: %.0f\n", dir_total);
681 return rc;
685 /****************************************************************************
686 get a file from rname to lname
687 ****************************************************************************/
688 static int do_get(struct smbclient_context *ctx, char *rname, const char *p_lname, bool reget)
690 int handle = 0, fnum;
691 bool newhandle = false;
692 uint8_t *data;
693 struct timeval tp_start;
694 int read_size = ctx->io_bufsize;
695 uint16_t attr;
696 size_t size;
697 off_t start = 0;
698 off_t nread = 0;
699 int rc = 0;
700 char *lname;
703 GetTimeOfDay(&tp_start);
705 if (ctx->lowercase) {
706 lname = strlower_talloc(ctx, p_lname);
707 } else {
708 lname = talloc_strdup(ctx, p_lname);
711 fnum = smbcli_open(ctx->cli->tree, rname, O_RDONLY, DENY_NONE);
713 if (fnum == -1) {
714 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
715 return 1;
718 if(!strcmp(lname,"-")) {
719 handle = fileno(stdout);
720 } else {
721 if (reget) {
722 handle = open(lname, O_WRONLY|O_CREAT, 0644);
723 if (handle >= 0) {
724 start = lseek(handle, 0, SEEK_END);
725 if (start == -1) {
726 d_printf("Error seeking local file\n");
727 close(handle);
728 return 1;
731 } else {
732 handle = open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
734 newhandle = true;
736 if (handle < 0) {
737 d_printf("Error opening local file %s\n",lname);
738 return 1;
742 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum,
743 &attr, &size, NULL, NULL, NULL, NULL, NULL)) &&
744 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum,
745 &attr, &size, NULL, NULL, NULL))) {
746 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
747 if (newhandle) {
748 close(handle);
750 return 1;
753 DEBUG(2,("getting file %s of size %.0f as %s ",
754 rname, (double)size, lname));
756 if(!(data = (uint8_t *)malloc(read_size))) {
757 d_printf("malloc fail for size %d\n", read_size);
758 smbcli_close(ctx->cli->tree, fnum);
759 if (newhandle) {
760 close(handle);
762 return 1;
765 while (1) {
766 int n = smbcli_read(ctx->cli->tree, fnum, data, nread + start, read_size);
768 if (n <= 0) break;
770 if (writefile(handle,data, n, ctx->translation) != n) {
771 d_printf("Error writing local file\n");
772 rc = 1;
773 break;
776 nread += n;
779 if (nread + start < size) {
780 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
781 rname, (long)nread));
783 rc = 1;
786 SAFE_FREE(data);
788 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
789 d_printf("Error %s closing remote file\n",smbcli_errstr(ctx->cli->tree));
790 rc = 1;
793 if (newhandle) {
794 close(handle);
797 if (ctx->archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) {
798 smbcli_setatr(ctx->cli->tree, rname, attr & ~(uint16_t)FILE_ATTRIBUTE_ARCHIVE, 0);
802 struct timeval tp_end;
803 int this_time;
805 GetTimeOfDay(&tp_end);
806 this_time =
807 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
808 (tp_end.tv_usec - tp_start.tv_usec)/1000;
809 get_total_time_ms += this_time;
810 get_total_size += nread;
812 DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n",
813 nread / (1.024*this_time + 1.0e-4),
814 get_total_size / (1.024*get_total_time_ms)));
817 return rc;
821 /****************************************************************************
822 get a file
823 ****************************************************************************/
824 static int cmd_get(struct smbclient_context *ctx, const char **args)
826 const char *lname;
827 char *rname;
829 if (!args[1]) {
830 d_printf("get <filename>\n");
831 return 1;
834 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
836 if (args[2])
837 lname = args[2];
838 else
839 lname = args[1];
841 dos_clean_name(rname);
843 return do_get(ctx, rname, lname, false);
846 /****************************************************************************
847 Put up a yes/no prompt.
848 ****************************************************************************/
849 static bool yesno(char *p)
851 char ans[4];
852 printf("%s",p);
854 if (!fgets(ans,sizeof(ans)-1,stdin))
855 return(false);
857 if (*ans == 'y' || *ans == 'Y')
858 return(true);
860 return(false);
863 /****************************************************************************
864 do a mget operation on one file
865 ****************************************************************************/
866 static void do_mget(struct smbclient_context *ctx, struct clilist_file_info *finfo)
868 char *rname;
869 char *quest;
870 char *mget_mask;
871 char *saved_curdir;
872 char *l_fname;
873 int ret;
875 if (ISDOT(finfo->name) || ISDOTDOT(finfo->name))
876 return;
878 if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)
879 quest = talloc_asprintf(ctx, "Get directory %s? ",finfo->name);
880 else
881 quest = talloc_asprintf(ctx, "Get file %s? ",finfo->name);
883 if (ctx->prompt && !yesno(quest)) return;
885 talloc_free(quest);
887 if (!(finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)) {
888 rname = talloc_asprintf(ctx, "%s%s",ctx->remote_cur_dir,
889 finfo->name);
890 do_get(ctx, rname, finfo->name, false);
891 talloc_free(rname);
892 return;
895 /* handle directories */
896 saved_curdir = talloc_strdup(ctx, ctx->remote_cur_dir);
898 ctx->remote_cur_dir = talloc_asprintf_append_buffer(NULL, "%s\\", finfo->name);
900 if (ctx->lowercase) {
901 l_fname = strlower_talloc(ctx, finfo->name);
902 } else {
903 l_fname = talloc_strdup(ctx, finfo->name);
906 string_replace(l_fname, '\\', '/');
908 if (!directory_exist(l_fname) &&
909 mkdir(l_fname, 0777) != 0) {
910 d_printf("failed to create directory %s\n", l_fname);
911 return;
914 if (chdir(l_fname) != 0) {
915 d_printf("failed to chdir to directory %s\n", l_fname);
916 return;
919 mget_mask = talloc_asprintf(ctx, "%s*", ctx->remote_cur_dir);
921 do_list(ctx, mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,false, true);
922 ret = chdir("..");
923 if (ret == -1) {
924 d_printf("failed to chdir to '..': %s\n", strerror(errno));
925 return;
927 talloc_free(ctx->remote_cur_dir);
929 ctx->remote_cur_dir = saved_curdir;
933 /****************************************************************************
934 view the file using the pager
935 ****************************************************************************/
936 static int cmd_more(struct smbclient_context *ctx, const char **args)
938 char *rname;
939 char *pager_cmd;
940 char *lname;
941 char *pager;
942 int fd;
943 int rc = 0;
944 mode_t mask;
946 lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir());
947 mask = umask(S_IRWXO | S_IRWXG);
948 fd = mkstemp(lname);
949 umask(mask);
950 if (fd == -1) {
951 d_printf("failed to create temporary file for more\n");
952 return 1;
954 close(fd);
956 if (!args[1]) {
957 d_printf("more <filename>\n");
958 unlink(lname);
959 return 1;
961 rname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
962 dos_clean_name(rname);
964 rc = do_get(ctx, rname, lname, false);
966 pager=getenv("PAGER");
968 pager_cmd = talloc_asprintf(ctx, "%s %s",(pager? pager:DEFAULT_PAGER), lname);
969 rc = system(pager_cmd);
970 if (rc == -1) {
971 d_printf("failed to call pager command\n");
972 return 1;
974 unlink(lname);
976 return rc;
981 /****************************************************************************
982 do a mget command
983 ****************************************************************************/
984 static int cmd_mget(struct smbclient_context *ctx, const char **args)
986 uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
987 char *mget_mask = NULL;
988 int i;
990 if (ctx->recurse)
991 attribute |= FILE_ATTRIBUTE_DIRECTORY;
993 for (i = 1; args[i]; i++) {
994 mget_mask = talloc_strdup(ctx, ctx->remote_cur_dir);
995 if(mget_mask[strlen(mget_mask)-1]!='\\')
996 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
998 mget_mask = talloc_strdup(ctx, args[i]);
999 if (mget_mask[0] != '\\')
1000 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
1001 do_list(ctx, mget_mask, attribute,do_mget,false,true);
1003 talloc_free(mget_mask);
1006 if (mget_mask == NULL) {
1007 mget_mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
1008 do_list(ctx, mget_mask, attribute,do_mget,false,true);
1009 talloc_free(mget_mask);
1012 return 0;
1016 /****************************************************************************
1017 make a directory of name "name"
1018 ****************************************************************************/
1019 static NTSTATUS do_mkdir(struct smbclient_context *ctx, char *name)
1021 NTSTATUS status;
1023 if (NT_STATUS_IS_ERR(status = smbcli_mkdir(ctx->cli->tree, name))) {
1024 d_printf("%s making remote directory %s\n",
1025 smbcli_errstr(ctx->cli->tree),name);
1026 return status;
1029 return status;
1033 /****************************************************************************
1034 Exit client.
1035 ****************************************************************************/
1036 static int cmd_quit(struct smbclient_context *ctx, const char **args)
1038 talloc_free(ctx);
1039 exit(0);
1040 /* NOTREACHED */
1041 return 0;
1045 /****************************************************************************
1046 make a directory
1047 ****************************************************************************/
1048 static int cmd_mkdir(struct smbclient_context *ctx, const char **args)
1050 char *mask, *p;
1052 if (!args[1]) {
1053 if (!ctx->recurse)
1054 d_printf("mkdir <dirname>\n");
1055 return 1;
1058 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir,args[1]);
1060 if (ctx->recurse) {
1061 dos_clean_name(mask);
1063 trim_string(mask,".",NULL);
1064 for (p = strtok(mask,"/\\"); p; p = strtok(p, "/\\")) {
1065 char *parent = talloc_strndup(ctx, mask, PTR_DIFF(p, mask));
1067 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, parent))) {
1068 do_mkdir(ctx, parent);
1071 talloc_free(parent);
1073 } else {
1074 do_mkdir(ctx, mask);
1077 return 0;
1080 /****************************************************************************
1081 show 8.3 name of a file
1082 ****************************************************************************/
1083 static int cmd_altname(struct smbclient_context *ctx, const char **args)
1085 const char *p;
1086 char *altname;
1087 char *name;
1089 if (!args[1]) {
1090 d_printf("altname <file>\n");
1091 return 1;
1094 name = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1096 if (!NT_STATUS_IS_OK(smbcli_qpathinfo_alt_name(ctx->cli->tree, name, &p))) {
1097 d_printf("%s getting alt name for %s\n",
1098 smbcli_errstr(ctx->cli->tree),name);
1099 return(false);
1101 altname = discard_const_p(char, p);
1102 d_printf("%s\n", altname);
1104 SAFE_FREE(altname);
1106 return 0;
1110 /****************************************************************************
1111 put a single file
1112 ****************************************************************************/
1113 static int do_put(struct smbclient_context *ctx, char *rname, char *lname, bool reput)
1115 int fnum;
1116 FILE *f;
1117 size_t start = 0;
1118 off_t nread = 0;
1119 uint8_t *buf = NULL;
1120 int maxwrite = ctx->io_bufsize;
1121 int rc = 0;
1123 struct timeval tp_start;
1124 GetTimeOfDay(&tp_start);
1126 if (reput) {
1127 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT, DENY_NONE);
1128 if (fnum >= 0) {
1129 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL)) &&
1130 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL))) {
1131 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
1132 return 1;
1135 } else {
1136 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT|O_TRUNC,
1137 DENY_NONE);
1140 if (fnum == -1) {
1141 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1142 return 1;
1145 /* allow files to be piped into smbclient
1146 jdblair 24.jun.98
1148 Note that in this case this function will exit(0) rather
1149 than returning. */
1150 if (!strcmp(lname, "-")) {
1151 f = stdin;
1152 /* size of file is not known */
1153 } else {
1154 f = fopen(lname, "r");
1155 if (f && reput) {
1156 if (fseek(f, start, SEEK_SET) == -1) {
1157 d_printf("Error seeking local file\n");
1158 fclose(f);
1159 return 1;
1164 if (!f) {
1165 d_printf("Error opening local file %s\n",lname);
1166 return 1;
1170 DEBUG(1,("putting file %s as %s ",lname,
1171 rname));
1173 buf = (uint8_t *)malloc(maxwrite);
1174 if (!buf) {
1175 d_printf("ERROR: Not enough memory!\n");
1176 fclose(f);
1177 return 1;
1179 while (!feof(f)) {
1180 int n = maxwrite;
1181 int ret;
1183 if ((n = readfile(buf,n,f,ctx->translation)) < 1) {
1184 if((n == 0) && feof(f))
1185 break; /* Empty local file. */
1187 d_printf("Error reading local file: %s\n", strerror(errno));
1188 rc = 1;
1189 break;
1192 ret = smbcli_write(ctx->cli->tree, fnum, 0, buf, nread + start, n);
1194 if (n != ret) {
1195 d_printf("Error writing file: %s\n", smbcli_errstr(ctx->cli->tree));
1196 rc = 1;
1197 break;
1200 nread += n;
1203 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
1204 d_printf("%s closing remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1205 fclose(f);
1206 SAFE_FREE(buf);
1207 return 1;
1211 if (f != stdin) {
1212 fclose(f);
1215 SAFE_FREE(buf);
1218 struct timeval tp_end;
1219 int this_time;
1221 GetTimeOfDay(&tp_end);
1222 this_time =
1223 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1224 (tp_end.tv_usec - tp_start.tv_usec)/1000;
1225 put_total_time_ms += this_time;
1226 put_total_size += nread;
1228 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1229 nread / (1.024*this_time + 1.0e-4),
1230 put_total_size / (1.024*put_total_time_ms)));
1233 if (f == stdin) {
1234 talloc_free(ctx);
1235 exit(0);
1238 return rc;
1243 /****************************************************************************
1244 put a file
1245 ****************************************************************************/
1246 static int cmd_put(struct smbclient_context *ctx, const char **args)
1248 char *lname;
1249 char *rname;
1251 if (!args[1]) {
1252 d_printf("put <filename> [<remotename>]\n");
1253 return 1;
1256 lname = talloc_strdup(ctx, args[1]);
1258 if (args[2]) {
1259 if (args[2][0]=='\\')
1260 rname = talloc_strdup(ctx, args[2]);
1261 else
1262 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[2]);
1263 } else {
1264 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, lname);
1267 dos_clean_name(rname);
1269 /* allow '-' to represent stdin
1270 jdblair, 24.jun.98 */
1271 if (!file_exist(lname) && (strcmp(lname,"-"))) {
1272 d_printf("%s does not exist\n",lname);
1273 return 1;
1276 return do_put(ctx, rname, lname, false);
1279 /*************************************
1280 File list structure
1281 *************************************/
1283 static struct file_list {
1284 struct file_list *prev, *next;
1285 char *file_path;
1286 bool isdir;
1287 } *file_list;
1289 /****************************************************************************
1290 Free a file_list structure
1291 ****************************************************************************/
1293 static void free_file_list (struct file_list * list)
1295 struct file_list *tmp;
1297 while (list)
1299 tmp = list;
1300 DLIST_REMOVE(list, list);
1301 SAFE_FREE(tmp->file_path);
1302 SAFE_FREE(tmp);
1306 /****************************************************************************
1307 seek in a directory/file list until you get something that doesn't start with
1308 the specified name
1309 ****************************************************************************/
1310 static bool seek_list(struct file_list *list, char *name)
1312 while (list) {
1313 trim_string(list->file_path,"./","\n");
1314 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1315 return(true);
1317 list = list->next;
1320 return(false);
1323 /****************************************************************************
1324 set the file selection mask
1325 ****************************************************************************/
1326 static int cmd_select(struct smbclient_context *ctx, const char **args)
1328 talloc_free(ctx->fileselection);
1329 ctx->fileselection = talloc_strdup(ctx, args[1]);
1331 return 0;
1334 /*******************************************************************
1335 A readdir wrapper which just returns the file name.
1336 ********************************************************************/
1337 static const char *readdirname(DIR *p)
1339 struct dirent *ptr;
1340 char *dname;
1342 if (!p)
1343 return(NULL);
1345 ptr = (struct dirent *)readdir(p);
1346 if (!ptr)
1347 return(NULL);
1349 dname = ptr->d_name;
1351 #ifdef NEXT2
1352 if (telldir(p) < 0)
1353 return(NULL);
1354 #endif
1356 #ifdef HAVE_BROKEN_READDIR
1357 /* using /usr/ucb/cc is BAD */
1358 dname = dname - 2;
1359 #endif
1362 static char *buf;
1363 int len = NAMLEN(ptr);
1364 buf = talloc_strndup(NULL, dname, len);
1365 dname = buf;
1368 return(dname);
1371 /****************************************************************************
1372 Recursive file matching function act as find
1373 match must be always set to true when calling this function
1374 ****************************************************************************/
1375 static int file_find(struct smbclient_context *ctx, struct file_list **list, const char *directory,
1376 const char *expression, bool match)
1378 DIR *dir;
1379 struct file_list *entry;
1380 struct stat statbuf;
1381 int ret;
1382 char *path;
1383 bool isdir;
1384 const char *dname;
1386 dir = opendir(directory);
1387 if (!dir) return -1;
1389 while ((dname = readdirname(dir))) {
1390 if (ISDOT(dname) || ISDOTDOT(dname)) {
1391 continue;
1394 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1395 continue;
1398 isdir = false;
1399 if (!match || !gen_fnmatch(expression, dname)) {
1400 if (ctx->recurse) {
1401 ret = stat(path, &statbuf);
1402 if (ret == 0) {
1403 if (S_ISDIR(statbuf.st_mode)) {
1404 isdir = true;
1405 ret = file_find(ctx, list, path, expression, false);
1407 } else {
1408 d_printf("file_find: cannot stat file %s\n", path);
1411 if (ret == -1) {
1412 SAFE_FREE(path);
1413 closedir(dir);
1414 return -1;
1417 entry = malloc_p(struct file_list);
1418 if (!entry) {
1419 d_printf("Out of memory in file_find\n");
1420 closedir(dir);
1421 return -1;
1423 entry->file_path = path;
1424 entry->isdir = isdir;
1425 DLIST_ADD(*list, entry);
1426 } else {
1427 SAFE_FREE(path);
1431 closedir(dir);
1432 return 0;
1435 /****************************************************************************
1436 mput some files
1437 ****************************************************************************/
1438 static int cmd_mput(struct smbclient_context *ctx, const char **args)
1440 int i;
1442 for (i = 1; args[i]; i++) {
1443 int ret;
1444 struct file_list *temp_list;
1445 char *quest, *lname, *rname;
1447 printf("%s\n", args[i]);
1449 file_list = NULL;
1451 ret = file_find(ctx, &file_list, ".", args[i], true);
1452 if (ret) {
1453 free_file_list(file_list);
1454 continue;
1457 quest = NULL;
1458 lname = NULL;
1459 rname = NULL;
1461 for (temp_list = file_list; temp_list;
1462 temp_list = temp_list->next) {
1464 SAFE_FREE(lname);
1465 if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1466 continue;
1467 trim_string(lname, "./", "/");
1469 /* check if it's a directory */
1470 if (temp_list->isdir) {
1471 /* if (!recurse) continue; */
1473 SAFE_FREE(quest);
1474 if (asprintf(&quest, "Put directory %s? ", lname) < 0) break;
1475 if (ctx->prompt && !yesno(quest)) { /* No */
1476 /* Skip the directory */
1477 lname[strlen(lname)-1] = '/';
1478 if (!seek_list(temp_list, lname))
1479 break;
1480 } else { /* Yes */
1481 SAFE_FREE(rname);
1482 if(asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1483 dos_format(rname);
1484 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, rname)) &&
1485 NT_STATUS_IS_ERR(do_mkdir(ctx, rname))) {
1486 DEBUG (0, ("Unable to make dir, skipping..."));
1487 /* Skip the directory */
1488 lname[strlen(lname)-1] = '/';
1489 if (!seek_list(temp_list, lname))
1490 break;
1493 continue;
1494 } else {
1495 SAFE_FREE(quest);
1496 if (asprintf(&quest,"Put file %s? ", lname) < 0) break;
1497 if (ctx->prompt && !yesno(quest)) /* No */
1498 continue;
1500 /* Yes */
1501 SAFE_FREE(rname);
1502 if (asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1505 dos_format(rname);
1507 do_put(ctx, rname, lname, false);
1509 free_file_list(file_list);
1510 SAFE_FREE(quest);
1511 SAFE_FREE(lname);
1512 SAFE_FREE(rname);
1515 return 0;
1519 /****************************************************************************
1520 print a file
1521 ****************************************************************************/
1522 static int cmd_print(struct smbclient_context *ctx, const char **args)
1524 char *lname, *rname;
1525 char *p;
1527 if (!args[1]) {
1528 d_printf("print <filename>\n");
1529 return 1;
1532 lname = talloc_strdup(ctx, args[1]);
1533 if (lname == NULL) {
1534 d_printf("Out of memory in cmd_print\n");
1535 return 1;
1538 if (strequal(lname, "-")) {
1539 rname = talloc_asprintf(ctx, "stdin-%d", (int)getpid());
1540 } else {
1541 p = strrchr_m(lname, '/');
1542 if (p) {
1543 rname = talloc_asprintf(ctx, "%s-%d", p + 1,
1544 (int)getpid());
1545 } else {
1546 rname = talloc_strdup(ctx, lname);
1550 if (rname == NULL) {
1551 d_printf("Out of memory in cmd_print (stdin)\n");
1552 return 1;
1555 return do_put(ctx, rname, lname, false);
1559 static int cmd_rewrite(struct smbclient_context *ctx, const char **args)
1561 d_printf("REWRITE: command not implemented (FIXME!)\n");
1563 return 0;
1566 /****************************************************************************
1567 delete some files
1568 ****************************************************************************/
1569 static int cmd_del(struct smbclient_context *ctx, const char **args)
1571 char *mask;
1572 uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
1574 if (ctx->recurse)
1575 attribute |= FILE_ATTRIBUTE_DIRECTORY;
1577 if (!args[1]) {
1578 d_printf("del <filename>\n");
1579 return 1;
1581 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1583 if (NT_STATUS_IS_ERR(smbcli_unlink(ctx->cli->tree, mask))) {
1584 d_printf("%s deleting remote file %s\n",smbcli_errstr(ctx->cli->tree),mask);
1587 return 0;
1591 /****************************************************************************
1592 delete a whole directory tree
1593 ****************************************************************************/
1594 static int cmd_deltree(struct smbclient_context *ctx, const char **args)
1596 char *dname;
1597 int ret;
1599 if (!args[1]) {
1600 d_printf("deltree <dirname>\n");
1601 return 1;
1604 dname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1606 ret = smbcli_deltree(ctx->cli->tree, dname);
1608 if (ret == -1) {
1609 printf("Failed to delete tree %s - %s\n", dname, smbcli_errstr(ctx->cli->tree));
1610 return -1;
1613 printf("Deleted %d files in %s\n", ret, dname);
1615 return 0;
1618 typedef struct {
1619 const char *level_name;
1620 enum smb_fsinfo_level level;
1621 } fsinfo_level_t;
1623 fsinfo_level_t fsinfo_levels[] = {
1624 {"dskattr", RAW_QFS_DSKATTR},
1625 {"allocation", RAW_QFS_ALLOCATION},
1626 {"volume", RAW_QFS_VOLUME},
1627 {"volumeinfo", RAW_QFS_VOLUME_INFO},
1628 {"sizeinfo", RAW_QFS_SIZE_INFO},
1629 {"deviceinfo", RAW_QFS_DEVICE_INFO},
1630 {"attributeinfo", RAW_QFS_ATTRIBUTE_INFO},
1631 {"unixinfo", RAW_QFS_UNIX_INFO},
1632 {"volume-information", RAW_QFS_VOLUME_INFORMATION},
1633 {"size-information", RAW_QFS_SIZE_INFORMATION},
1634 {"device-information", RAW_QFS_DEVICE_INFORMATION},
1635 {"attribute-information", RAW_QFS_ATTRIBUTE_INFORMATION},
1636 {"quota-information", RAW_QFS_QUOTA_INFORMATION},
1637 {"fullsize-information", RAW_QFS_FULL_SIZE_INFORMATION},
1638 {"objectid", RAW_QFS_OBJECTID_INFORMATION},
1639 {"sector-size-info", RAW_QFS_SECTOR_SIZE_INFORMATION},
1640 {NULL, RAW_QFS_GENERIC}
1644 static int cmd_fsinfo(struct smbclient_context *ctx, const char **args)
1646 union smb_fsinfo fsinfo;
1647 NTSTATUS status;
1648 fsinfo_level_t *fsinfo_level;
1650 if (!args[1]) {
1651 d_printf("fsinfo <level>, where level is one of following:\n");
1652 fsinfo_level = fsinfo_levels;
1653 while(fsinfo_level->level_name) {
1654 d_printf("%s\n", fsinfo_level->level_name);
1655 fsinfo_level++;
1657 return 1;
1660 fsinfo_level = fsinfo_levels;
1661 while(fsinfo_level->level_name && !strequal(args[1],fsinfo_level->level_name)) {
1662 fsinfo_level++;
1665 if (!fsinfo_level->level_name) {
1666 d_printf("wrong level name!\n");
1667 return 1;
1670 fsinfo.generic.level = fsinfo_level->level;
1671 status = smb_raw_fsinfo(ctx->cli->tree, ctx, &fsinfo);
1672 if (!NT_STATUS_IS_OK(status)) {
1673 d_printf("fsinfo-level-%s - %s\n", fsinfo_level->level_name, nt_errstr(status));
1674 return 1;
1677 d_printf("fsinfo-level-%s:\n", fsinfo_level->level_name);
1678 switch(fsinfo.generic.level) {
1679 case RAW_QFS_DSKATTR:
1680 d_printf("\tunits_total: %hu\n",
1681 (unsigned short) fsinfo.dskattr.out.units_total);
1682 d_printf("\tblocks_per_unit: %hu\n",
1683 (unsigned short) fsinfo.dskattr.out.blocks_per_unit);
1684 d_printf("\tblocks_size: %hu\n",
1685 (unsigned short) fsinfo.dskattr.out.block_size);
1686 d_printf("\tunits_free: %hu\n",
1687 (unsigned short) fsinfo.dskattr.out.units_free);
1688 break;
1689 case RAW_QFS_ALLOCATION:
1690 d_printf("\tfs_id: %lu\n",
1691 (unsigned long) fsinfo.allocation.out.fs_id);
1692 d_printf("\tsectors_per_unit: %lu\n",
1693 (unsigned long) fsinfo.allocation.out.sectors_per_unit);
1694 d_printf("\ttotal_alloc_units: %lu\n",
1695 (unsigned long) fsinfo.allocation.out.total_alloc_units);
1696 d_printf("\tavail_alloc_units: %lu\n",
1697 (unsigned long) fsinfo.allocation.out.avail_alloc_units);
1698 d_printf("\tbytes_per_sector: %hu\n",
1699 (unsigned short) fsinfo.allocation.out.bytes_per_sector);
1700 break;
1701 case RAW_QFS_VOLUME:
1702 d_printf("\tserial_number: %lu\n",
1703 (unsigned long) fsinfo.volume.out.serial_number);
1704 d_printf("\tvolume_name: %s\n", fsinfo.volume.out.volume_name.s);
1705 break;
1706 case RAW_QFS_VOLUME_INFO:
1707 case RAW_QFS_VOLUME_INFORMATION:
1708 d_printf("\tcreate_time: %s\n",
1709 nt_time_string(ctx,fsinfo.volume_info.out.create_time));
1710 d_printf("\tserial_number: %lu\n",
1711 (unsigned long) fsinfo.volume_info.out.serial_number);
1712 d_printf("\tvolume_name: %s\n", fsinfo.volume_info.out.volume_name.s);
1713 break;
1714 case RAW_QFS_SIZE_INFO:
1715 case RAW_QFS_SIZE_INFORMATION:
1716 d_printf("\ttotal_alloc_units: %llu\n",
1717 (unsigned long long) fsinfo.size_info.out.total_alloc_units);
1718 d_printf("\tavail_alloc_units: %llu\n",
1719 (unsigned long long) fsinfo.size_info.out.avail_alloc_units);
1720 d_printf("\tsectors_per_unit: %lu\n",
1721 (unsigned long) fsinfo.size_info.out.sectors_per_unit);
1722 d_printf("\tbytes_per_sector: %lu\n",
1723 (unsigned long) fsinfo.size_info.out.bytes_per_sector);
1724 break;
1725 case RAW_QFS_DEVICE_INFO:
1726 case RAW_QFS_DEVICE_INFORMATION:
1727 d_printf("\tdevice_type: %lu\n",
1728 (unsigned long) fsinfo.device_info.out.device_type);
1729 d_printf("\tcharacteristics: 0x%lx\n",
1730 (unsigned long) fsinfo.device_info.out.characteristics);
1731 break;
1732 case RAW_QFS_ATTRIBUTE_INFORMATION:
1733 case RAW_QFS_ATTRIBUTE_INFO:
1734 d_printf("\tfs_attr: 0x%lx\n",
1735 (unsigned long) fsinfo.attribute_info.out.fs_attr);
1736 d_printf("\tmax_file_component_length: %lu\n",
1737 (unsigned long) fsinfo.attribute_info.out.max_file_component_length);
1738 d_printf("\tfs_type: %s\n", fsinfo.attribute_info.out.fs_type.s);
1739 break;
1740 case RAW_QFS_UNIX_INFO:
1741 d_printf("\tmajor_version: %hu\n",
1742 (unsigned short) fsinfo.unix_info.out.major_version);
1743 d_printf("\tminor_version: %hu\n",
1744 (unsigned short) fsinfo.unix_info.out.minor_version);
1745 d_printf("\tcapability: 0x%llx\n",
1746 (unsigned long long) fsinfo.unix_info.out.capability);
1747 break;
1748 case RAW_QFS_QUOTA_INFORMATION:
1749 d_printf("\tunknown[3]: [%llu,%llu,%llu]\n",
1750 (unsigned long long) fsinfo.quota_information.out.unknown[0],
1751 (unsigned long long) fsinfo.quota_information.out.unknown[1],
1752 (unsigned long long) fsinfo.quota_information.out.unknown[2]);
1753 d_printf("\tquota_soft: %llu\n",
1754 (unsigned long long) fsinfo.quota_information.out.quota_soft);
1755 d_printf("\tquota_hard: %llu\n",
1756 (unsigned long long) fsinfo.quota_information.out.quota_hard);
1757 d_printf("\tquota_flags: 0x%llx\n",
1758 (unsigned long long) fsinfo.quota_information.out.quota_flags);
1759 break;
1760 case RAW_QFS_FULL_SIZE_INFORMATION:
1761 d_printf("\ttotal_alloc_units: %llu\n",
1762 (unsigned long long) fsinfo.full_size_information.out.total_alloc_units);
1763 d_printf("\tcall_avail_alloc_units: %llu\n",
1764 (unsigned long long) fsinfo.full_size_information.out.call_avail_alloc_units);
1765 d_printf("\tactual_avail_alloc_units: %llu\n",
1766 (unsigned long long) fsinfo.full_size_information.out.actual_avail_alloc_units);
1767 d_printf("\tsectors_per_unit: %lu\n",
1768 (unsigned long) fsinfo.full_size_information.out.sectors_per_unit);
1769 d_printf("\tbytes_per_sector: %lu\n",
1770 (unsigned long) fsinfo.full_size_information.out.bytes_per_sector);
1771 break;
1772 case RAW_QFS_OBJECTID_INFORMATION:
1773 d_printf("\tGUID: %s\n",
1774 GUID_string(ctx,&fsinfo.objectid_information.out.guid));
1775 d_printf("\tunknown[6]: [%llu,%llu,%llu,%llu,%llu,%llu]\n",
1776 (unsigned long long) fsinfo.objectid_information.out.unknown[0],
1777 (unsigned long long) fsinfo.objectid_information.out.unknown[1],
1778 (unsigned long long) fsinfo.objectid_information.out.unknown[2],
1779 (unsigned long long) fsinfo.objectid_information.out.unknown[3],
1780 (unsigned long long) fsinfo.objectid_information.out.unknown[4],
1781 (unsigned long long) fsinfo.objectid_information.out.unknown[5] );
1782 break;
1783 case RAW_QFS_SECTOR_SIZE_INFORMATION:
1784 d_printf("\tlogical_bytes_per_sector: %u\n",
1785 (unsigned)fsinfo.sector_size_info.out.logical_bytes_per_sector);
1786 d_printf("\tphys_bytes_per_sector_atomic: %u\n",
1787 (unsigned)fsinfo.sector_size_info.out.phys_bytes_per_sector_atomic);
1788 d_printf("\tphys_bytes_per_sector_perf: %u\n",
1789 (unsigned)fsinfo.sector_size_info.out.phys_bytes_per_sector_perf);
1790 d_printf("\tfs_effective_phys_bytes_per_sector_atomic: %u\n",
1791 (unsigned)fsinfo.sector_size_info.out.fs_effective_phys_bytes_per_sector_atomic);
1792 d_printf("\tflags: 0x%x\n",
1793 (unsigned)fsinfo.sector_size_info.out.flags);
1794 d_printf("\tbyte_off_sector_align: %u\n",
1795 (unsigned)fsinfo.sector_size_info.out.byte_off_sector_align);
1796 d_printf("\tbyte_off_partition_align: %u\n",
1797 (unsigned)fsinfo.sector_size_info.out.byte_off_partition_align);
1798 break;
1799 case RAW_QFS_GENERIC:
1800 d_printf("\twrong level returned\n");
1801 break;
1804 return 0;
1807 /****************************************************************************
1808 show as much information as possible about a file
1809 ****************************************************************************/
1810 static int cmd_allinfo(struct smbclient_context *ctx, const char **args)
1812 char *fname;
1813 union smb_fileinfo finfo;
1814 NTSTATUS status;
1815 int fnum;
1817 if (!args[1]) {
1818 d_printf("allinfo <filename>\n");
1819 return 1;
1821 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1823 /* first a ALL_INFO QPATHINFO */
1824 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1825 finfo.generic.in.file.path = fname;
1826 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1827 if (!NT_STATUS_IS_OK(status)) {
1828 d_printf("%s - %s\n", fname, nt_errstr(status));
1829 return 1;
1832 d_printf("\tcreate_time: %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1833 d_printf("\taccess_time: %s\n", nt_time_string(ctx, finfo.all_info.out.access_time));
1834 d_printf("\twrite_time: %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1835 d_printf("\tchange_time: %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1836 d_printf("\tattrib: 0x%x\n", finfo.all_info.out.attrib);
1837 d_printf("\talloc_size: %lu\n", (unsigned long)finfo.all_info.out.alloc_size);
1838 d_printf("\tsize: %lu\n", (unsigned long)finfo.all_info.out.size);
1839 d_printf("\tnlink: %u\n", finfo.all_info.out.nlink);
1840 d_printf("\tdelete_pending: %u\n", finfo.all_info.out.delete_pending);
1841 d_printf("\tdirectory: %u\n", finfo.all_info.out.directory);
1842 d_printf("\tea_size: %u\n", finfo.all_info.out.ea_size);
1843 d_printf("\tfname: '%s'\n", finfo.all_info.out.fname.s);
1845 /* 8.3 name if any */
1846 finfo.generic.level = RAW_FILEINFO_ALT_NAME_INFO;
1847 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1848 if (NT_STATUS_IS_OK(status)) {
1849 d_printf("\talt_name: %s\n", finfo.alt_name_info.out.fname.s);
1852 /* file_id if available */
1853 finfo.generic.level = RAW_FILEINFO_INTERNAL_INFORMATION;
1854 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1855 if (NT_STATUS_IS_OK(status)) {
1856 d_printf("\tfile_id %.0f\n",
1857 (double)finfo.internal_information.out.file_id);
1860 /* the EAs, if any */
1861 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1862 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1863 if (NT_STATUS_IS_OK(status)) {
1864 int i;
1865 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1866 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1867 finfo.all_eas.out.eas[i].flags,
1868 (int)finfo.all_eas.out.eas[i].value.length,
1869 finfo.all_eas.out.eas[i].name.s);
1873 /* streams, if available */
1874 finfo.generic.level = RAW_FILEINFO_STREAM_INFO;
1875 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1876 if (NT_STATUS_IS_OK(status)) {
1877 int i;
1878 for (i=0;i<finfo.stream_info.out.num_streams;i++) {
1879 d_printf("\tstream %d:\n", i);
1880 d_printf("\t\tsize %ld\n",
1881 (long)finfo.stream_info.out.streams[i].size);
1882 d_printf("\t\talloc size %ld\n",
1883 (long)finfo.stream_info.out.streams[i].alloc_size);
1884 d_printf("\t\tname %s\n", finfo.stream_info.out.streams[i].stream_name.s);
1888 /* dev/inode if available */
1889 finfo.generic.level = RAW_FILEINFO_COMPRESSION_INFORMATION;
1890 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1891 if (NT_STATUS_IS_OK(status)) {
1892 d_printf("\tcompressed size %ld\n", (long)finfo.compression_info.out.compressed_size);
1893 d_printf("\tformat %ld\n", (long)finfo.compression_info.out.format);
1894 d_printf("\tunit_shift %ld\n", (long)finfo.compression_info.out.unit_shift);
1895 d_printf("\tchunk_shift %ld\n", (long)finfo.compression_info.out.chunk_shift);
1896 d_printf("\tcluster_shift %ld\n", (long)finfo.compression_info.out.cluster_shift);
1899 /* shadow copies if available */
1900 fnum = smbcli_open(ctx->cli->tree, fname, O_RDONLY, DENY_NONE);
1901 if (fnum != -1) {
1902 struct smb_shadow_copy info;
1903 int i;
1904 info.in.file.fnum = fnum;
1905 info.in.max_data = ~0;
1906 status = smb_raw_shadow_data(ctx->cli->tree, ctx, &info);
1907 if (NT_STATUS_IS_OK(status)) {
1908 d_printf("\tshadow_copy: %u volumes %u names\n",
1909 info.out.num_volumes, info.out.num_names);
1910 for (i=0;i<info.out.num_names;i++) {
1911 d_printf("\t%s\n", info.out.names[i]);
1912 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1913 finfo.generic.in.file.path = talloc_asprintf(ctx, "%s%s",
1914 info.out.names[i], fname);
1915 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1916 if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_PATH_NOT_FOUND) ||
1917 NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
1918 continue;
1920 if (!NT_STATUS_IS_OK(status)) {
1921 d_printf("%s - %s\n", finfo.generic.in.file.path,
1922 nt_errstr(status));
1923 return 1;
1926 d_printf("\t\tcreate_time: %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1927 d_printf("\t\twrite_time: %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1928 d_printf("\t\tchange_time: %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1929 d_printf("\t\tsize: %lu\n", (unsigned long)finfo.all_info.out.size);
1934 return 0;
1938 /****************************************************************************
1939 shows EA contents
1940 ****************************************************************************/
1941 static int cmd_eainfo(struct smbclient_context *ctx, const char **args)
1943 char *fname;
1944 union smb_fileinfo finfo;
1945 NTSTATUS status;
1946 int i;
1948 if (!args[1]) {
1949 d_printf("eainfo <filename>\n");
1950 return 1;
1952 fname = talloc_strdup(ctx, args[1]);
1954 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1955 finfo.generic.in.file.path = fname;
1956 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1958 if (!NT_STATUS_IS_OK(status)) {
1959 d_printf("RAW_FILEINFO_ALL_EAS - %s\n", nt_errstr(status));
1960 return 1;
1963 d_printf("%s has %d EAs\n", fname, finfo.all_eas.out.num_eas);
1965 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1966 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1967 finfo.all_eas.out.eas[i].flags,
1968 (int)finfo.all_eas.out.eas[i].value.length,
1969 finfo.all_eas.out.eas[i].name.s);
1970 fflush(stdout);
1971 dump_data(0,
1972 finfo.all_eas.out.eas[i].value.data,
1973 finfo.all_eas.out.eas[i].value.length);
1976 return 0;
1980 /****************************************************************************
1981 show any ACL on a file
1982 ****************************************************************************/
1983 static int cmd_acl(struct smbclient_context *ctx, const char **args)
1985 char *fname;
1986 union smb_fileinfo query;
1987 NTSTATUS status;
1988 int fnum;
1990 if (!args[1]) {
1991 d_printf("acl <filename>\n");
1992 return 1;
1994 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1996 fnum = smbcli_nt_create_full(ctx->cli->tree, fname, 0,
1997 SEC_STD_READ_CONTROL,
1999 NTCREATEX_SHARE_ACCESS_DELETE|
2000 NTCREATEX_SHARE_ACCESS_READ|
2001 NTCREATEX_SHARE_ACCESS_WRITE,
2002 NTCREATEX_DISP_OPEN,
2003 0, 0);
2004 if (fnum == -1) {
2005 d_printf("%s - %s\n", fname, smbcli_errstr(ctx->cli->tree));
2006 return -1;
2009 query.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
2010 query.query_secdesc.in.file.fnum = fnum;
2011 query.query_secdesc.in.secinfo_flags = 0x7;
2013 status = smb_raw_fileinfo(ctx->cli->tree, ctx, &query);
2014 if (!NT_STATUS_IS_OK(status)) {
2015 d_printf("%s - %s\n", fname, nt_errstr(status));
2016 return 1;
2019 NDR_PRINT_DEBUG(security_descriptor, query.query_secdesc.out.sd);
2021 return 0;
2024 /****************************************************************************
2025 lookup a name or sid
2026 ****************************************************************************/
2027 static int cmd_lookup(struct smbclient_context *ctx, const char **args)
2029 NTSTATUS status;
2030 struct dom_sid *sid;
2032 if (!args[1]) {
2033 d_printf("lookup <sid|name>\n");
2034 return 1;
2037 sid = dom_sid_parse_talloc(ctx, args[1]);
2038 if (sid == NULL) {
2039 const char *sidstr;
2040 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sidstr);
2041 if (!NT_STATUS_IS_OK(status)) {
2042 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2043 return 1;
2046 d_printf("%s\n", sidstr);
2047 } else {
2048 const char *name;
2049 status = smblsa_lookup_sid(ctx->cli, args[1], ctx, &name);
2050 if (!NT_STATUS_IS_OK(status)) {
2051 d_printf("lsa_LookupSids - %s\n", nt_errstr(status));
2052 return 1;
2055 d_printf("%s\n", name);
2058 return 0;
2061 /****************************************************************************
2062 show privileges for a user
2063 ****************************************************************************/
2064 static int cmd_privileges(struct smbclient_context *ctx, const char **args)
2066 NTSTATUS status;
2067 struct dom_sid *sid;
2068 struct lsa_RightSet rights;
2069 unsigned i;
2071 if (!args[1]) {
2072 d_printf("privileges <sid|name>\n");
2073 return 1;
2076 sid = dom_sid_parse_talloc(ctx, args[1]);
2077 if (sid == NULL) {
2078 const char *sid_str;
2079 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2080 if (!NT_STATUS_IS_OK(status)) {
2081 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2082 return 1;
2084 sid = dom_sid_parse_talloc(ctx, sid_str);
2087 status = smblsa_sid_privileges(ctx->cli, sid, ctx, &rights);
2088 if (!NT_STATUS_IS_OK(status)) {
2089 d_printf("lsa_EnumAccountRights - %s\n", nt_errstr(status));
2090 return 1;
2093 for (i=0;i<rights.count;i++) {
2094 d_printf("\t%s\n", rights.names[i].string);
2097 return 0;
2101 /****************************************************************************
2102 add privileges for a user
2103 ****************************************************************************/
2104 static int cmd_addprivileges(struct smbclient_context *ctx, const char **args)
2106 NTSTATUS status;
2107 struct dom_sid *sid;
2108 struct lsa_RightSet rights;
2109 int i;
2111 if (!args[1]) {
2112 d_printf("addprivileges <sid|name> <privilege...>\n");
2113 return 1;
2116 sid = dom_sid_parse_talloc(ctx, args[1]);
2117 if (sid == NULL) {
2118 const char *sid_str;
2119 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2120 if (!NT_STATUS_IS_OK(status)) {
2121 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2122 return 1;
2124 sid = dom_sid_parse_talloc(ctx, sid_str);
2127 ZERO_STRUCT(rights);
2128 for (i = 2; args[i]; i++) {
2129 rights.names = talloc_realloc(ctx, rights.names,
2130 struct lsa_StringLarge, rights.count+1);
2131 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2132 rights.count++;
2136 status = smblsa_sid_add_privileges(ctx->cli, sid, ctx, &rights);
2137 if (!NT_STATUS_IS_OK(status)) {
2138 d_printf("lsa_AddAccountRights - %s\n", nt_errstr(status));
2139 return 1;
2142 return 0;
2145 /****************************************************************************
2146 delete privileges for a user
2147 ****************************************************************************/
2148 static int cmd_delprivileges(struct smbclient_context *ctx, const char **args)
2150 NTSTATUS status;
2151 struct dom_sid *sid;
2152 struct lsa_RightSet rights;
2153 int i;
2155 if (!args[1]) {
2156 d_printf("delprivileges <sid|name> <privilege...>\n");
2157 return 1;
2160 sid = dom_sid_parse_talloc(ctx, args[1]);
2161 if (sid == NULL) {
2162 const char *sid_str;
2163 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2164 if (!NT_STATUS_IS_OK(status)) {
2165 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2166 return 1;
2168 sid = dom_sid_parse_talloc(ctx, sid_str);
2171 ZERO_STRUCT(rights);
2172 for (i = 2; args[i]; i++) {
2173 rights.names = talloc_realloc(ctx, rights.names,
2174 struct lsa_StringLarge, rights.count+1);
2175 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2176 rights.count++;
2180 status = smblsa_sid_del_privileges(ctx->cli, sid, ctx, &rights);
2181 if (!NT_STATUS_IS_OK(status)) {
2182 d_printf("lsa_RemoveAccountRights - %s\n", nt_errstr(status));
2183 return 1;
2186 return 0;
2190 /****************************************************************************
2191 open a file
2192 ****************************************************************************/
2193 static int cmd_open(struct smbclient_context *ctx, const char **args)
2195 char *filename;
2196 union smb_open io;
2197 NTSTATUS status;
2198 TALLOC_CTX *tmp_ctx;
2200 if (!args[1]) {
2201 d_printf("open <filename>\n");
2202 return 1;
2204 tmp_ctx = talloc_new(ctx);
2206 filename = talloc_asprintf(tmp_ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2208 io.generic.level = RAW_OPEN_NTCREATEX;
2209 io.ntcreatex.in.root_fid.fnum = 0;
2210 io.ntcreatex.in.flags = 0;
2211 io.ntcreatex.in.access_mask = SEC_RIGHTS_FILE_ALL;
2212 io.ntcreatex.in.create_options = 0;
2213 io.ntcreatex.in.file_attr = FILE_ATTRIBUTE_NORMAL;
2214 io.ntcreatex.in.share_access = NTCREATEX_SHARE_ACCESS_READ;
2215 io.ntcreatex.in.alloc_size = 0;
2216 io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN_IF;
2217 io.ntcreatex.in.impersonation = NTCREATEX_IMPERSONATION_ANONYMOUS;
2218 io.ntcreatex.in.security_flags = 0;
2219 io.ntcreatex.in.fname = filename;
2221 status = smb_raw_open(ctx->cli->tree, tmp_ctx, &io);
2222 talloc_free(tmp_ctx);
2224 if (NT_STATUS_IS_OK(status)) {
2225 d_printf("Opened file with fnum %u\n", (unsigned)io.ntcreatex.out.file.fnum);
2226 } else {
2227 d_printf("Opened failed: %s\n", nt_errstr(status));
2230 return 0;
2233 /****************************************************************************
2234 close a file
2235 ****************************************************************************/
2236 static int cmd_close(struct smbclient_context *ctx, const char **args)
2238 union smb_close io;
2239 NTSTATUS status;
2240 uint16_t fnum;
2242 if (!args[1]) {
2243 d_printf("close <fnum>\n");
2244 return 1;
2247 fnum = atoi(args[1]);
2249 ZERO_STRUCT(io);
2250 io.generic.level = RAW_CLOSE_CLOSE;
2251 io.close.in.file.fnum = fnum;
2253 status = smb_raw_close(ctx->cli->tree, &io);
2255 if (NT_STATUS_IS_OK(status)) {
2256 d_printf("Closed file OK\n");
2257 } else {
2258 d_printf("Close failed: %s\n", nt_errstr(status));
2261 return 0;
2265 /****************************************************************************
2266 remove a directory
2267 ****************************************************************************/
2268 static int cmd_rmdir(struct smbclient_context *ctx, const char **args)
2270 char *mask;
2272 if (!args[1]) {
2273 d_printf("rmdir <dirname>\n");
2274 return 1;
2276 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2278 if (NT_STATUS_IS_ERR(smbcli_rmdir(ctx->cli->tree, mask))) {
2279 d_printf("%s removing remote directory file %s\n",
2280 smbcli_errstr(ctx->cli->tree),mask);
2283 return 0;
2286 /****************************************************************************
2287 UNIX hardlink.
2288 ****************************************************************************/
2289 static int cmd_link(struct smbclient_context *ctx, const char **args)
2291 char *src,*dest;
2293 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2294 d_printf("Server doesn't support UNIX CIFS calls.\n");
2295 return 1;
2299 if (!args[1] || !args[2]) {
2300 d_printf("link <src> <dest>\n");
2301 return 1;
2304 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2305 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2307 if (NT_STATUS_IS_ERR(smbcli_unix_hardlink(ctx->cli->tree, src, dest))) {
2308 d_printf("%s linking files (%s -> %s)\n", smbcli_errstr(ctx->cli->tree), src, dest);
2309 return 1;
2312 return 0;
2315 /****************************************************************************
2316 UNIX symlink.
2317 ****************************************************************************/
2319 static int cmd_symlink(struct smbclient_context *ctx, const char **args)
2321 char *src,*dest;
2323 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2324 d_printf("Server doesn't support UNIX CIFS calls.\n");
2325 return 1;
2328 if (!args[1] || !args[2]) {
2329 d_printf("symlink <src> <dest>\n");
2330 return 1;
2333 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2334 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2336 if (NT_STATUS_IS_ERR(smbcli_unix_symlink(ctx->cli->tree, src, dest))) {
2337 d_printf("%s symlinking files (%s -> %s)\n",
2338 smbcli_errstr(ctx->cli->tree), src, dest);
2339 return 1;
2342 return 0;
2345 /****************************************************************************
2346 UNIX chmod.
2347 ****************************************************************************/
2349 static int cmd_chmod(struct smbclient_context *ctx, const char **args)
2351 char *src;
2352 mode_t mode;
2354 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2355 d_printf("Server doesn't support UNIX CIFS calls.\n");
2356 return 1;
2359 if (!args[1] || !args[2]) {
2360 d_printf("chmod mode file\n");
2361 return 1;
2364 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2366 mode = (mode_t)strtol(args[1], NULL, 8);
2368 if (NT_STATUS_IS_ERR(smbcli_unix_chmod(ctx->cli->tree, src, mode))) {
2369 d_printf("%s chmod file %s 0%o\n",
2370 smbcli_errstr(ctx->cli->tree), src, (unsigned)mode);
2371 return 1;
2374 return 0;
2377 /****************************************************************************
2378 UNIX chown.
2379 ****************************************************************************/
2381 static int cmd_chown(struct smbclient_context *ctx, const char **args)
2383 char *src;
2384 uid_t uid;
2385 gid_t gid;
2387 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2388 d_printf("Server doesn't support UNIX CIFS calls.\n");
2389 return 1;
2392 if (!args[1] || !args[2] || !args[3]) {
2393 d_printf("chown uid gid file\n");
2394 return 1;
2397 uid = (uid_t)atoi(args[1]);
2398 gid = (gid_t)atoi(args[2]);
2399 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[3]);
2401 if (NT_STATUS_IS_ERR(smbcli_unix_chown(ctx->cli->tree, src, uid, gid))) {
2402 d_printf("%s chown file %s uid=%d, gid=%d\n",
2403 smbcli_errstr(ctx->cli->tree), src, (int)uid, (int)gid);
2404 return 1;
2407 return 0;
2410 /****************************************************************************
2411 rename some files
2412 ****************************************************************************/
2413 static int cmd_rename(struct smbclient_context *ctx, const char **args)
2415 char *src,*dest;
2417 if (!args[1] || !args[2]) {
2418 d_printf("rename <src> <dest>\n");
2419 return 1;
2422 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2423 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2425 if (NT_STATUS_IS_ERR(smbcli_rename(ctx->cli->tree, src, dest))) {
2426 d_printf("%s renaming files\n",smbcli_errstr(ctx->cli->tree));
2427 return 1;
2430 return 0;
2434 /****************************************************************************
2435 toggle the prompt flag
2436 ****************************************************************************/
2437 static int cmd_prompt(struct smbclient_context *ctx, const char **args)
2439 ctx->prompt = !ctx->prompt;
2440 DEBUG(2,("prompting is now %s\n",ctx->prompt?"on":"off"));
2442 return 1;
2446 /****************************************************************************
2447 set the newer than time
2448 ****************************************************************************/
2449 static int cmd_newer(struct smbclient_context *ctx, const char **args)
2451 struct stat sbuf;
2453 if (args[1] && (stat(args[1],&sbuf) == 0)) {
2454 ctx->newer_than = sbuf.st_mtime;
2455 DEBUG(1,("Getting files newer than %s",
2456 asctime(localtime(&ctx->newer_than))));
2457 } else {
2458 ctx->newer_than = 0;
2461 if (args[1] && ctx->newer_than == 0) {
2462 d_printf("Error setting newer-than time\n");
2463 return 1;
2466 return 0;
2469 /****************************************************************************
2470 set the archive level
2471 ****************************************************************************/
2472 static int cmd_archive(struct smbclient_context *ctx, const char **args)
2474 if (args[1]) {
2475 ctx->archive_level = atoi(args[1]);
2476 } else
2477 d_printf("Archive level is %d\n",ctx->archive_level);
2479 return 0;
2482 /****************************************************************************
2483 toggle the lowercaseflag
2484 ****************************************************************************/
2485 static int cmd_lowercase(struct smbclient_context *ctx, const char **args)
2487 ctx->lowercase = !ctx->lowercase;
2488 DEBUG(2,("filename lowercasing is now %s\n",ctx->lowercase?"on":"off"));
2490 return 0;
2496 /****************************************************************************
2497 toggle the recurse flag
2498 ****************************************************************************/
2499 static int cmd_recurse(struct smbclient_context *ctx, const char **args)
2501 ctx->recurse = !ctx->recurse;
2502 DEBUG(2,("directory recursion is now %s\n",ctx->recurse?"on":"off"));
2504 return 0;
2507 /****************************************************************************
2508 toggle the translate flag
2509 ****************************************************************************/
2510 static int cmd_translate(struct smbclient_context *ctx, const char **args)
2512 ctx->translation = !ctx->translation;
2513 DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
2514 ctx->translation?"on":"off"));
2516 return 0;
2520 /****************************************************************************
2521 do a printmode command
2522 ****************************************************************************/
2523 static int cmd_printmode(struct smbclient_context *ctx, const char **args)
2525 if (args[1]) {
2526 if (strequal(args[1],"text")) {
2527 ctx->printmode = 0;
2528 } else {
2529 if (strequal(args[1],"graphics"))
2530 ctx->printmode = 1;
2531 else
2532 ctx->printmode = atoi(args[1]);
2536 switch(ctx->printmode)
2538 case 0:
2539 DEBUG(2,("the printmode is now text\n"));
2540 break;
2541 case 1:
2542 DEBUG(2,("the printmode is now graphics\n"));
2543 break;
2544 default:
2545 DEBUG(2,("the printmode is now %d\n", ctx->printmode));
2546 break;
2549 return 0;
2552 /****************************************************************************
2553 do the lcd command
2554 ****************************************************************************/
2555 static int cmd_lcd(struct smbclient_context *ctx, const char **args)
2557 char d[PATH_MAX];
2559 if (args[1]) {
2560 int ret;
2562 ret = chdir(args[1]);
2563 if (ret == -1) {
2564 d_printf("failed to chdir to dir '%s': %s\n",
2565 args[1], strerror(errno));
2566 return 1;
2570 DEBUG(2,("the local directory is now %s\n",getcwd(d, PATH_MAX)));
2572 return 0;
2575 /****************************************************************************
2576 history
2577 ****************************************************************************/
2578 static int cmd_history(struct smbclient_context *ctx, const char **args)
2580 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
2581 HIST_ENTRY **hlist;
2582 int i;
2584 hlist = history_list();
2586 for (i = 0; hlist && hlist[i]; i++) {
2587 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
2589 #else
2590 DEBUG(0,("no history without readline support\n"));
2591 #endif
2593 return 0;
2596 /****************************************************************************
2597 get a file restarting at end of local file
2598 ****************************************************************************/
2599 static int cmd_reget(struct smbclient_context *ctx, const char **args)
2601 char *local_name;
2602 char *remote_name;
2604 if (!args[1]) {
2605 d_printf("reget <filename>\n");
2606 return 1;
2608 remote_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2609 dos_clean_name(remote_name);
2611 if (args[2])
2612 local_name = talloc_strdup(ctx, args[2]);
2613 else
2614 local_name = talloc_strdup(ctx, args[1]);
2616 return do_get(ctx, remote_name, local_name, true);
2619 /****************************************************************************
2620 put a file restarting at end of local file
2621 ****************************************************************************/
2622 static int cmd_reput(struct smbclient_context *ctx, const char **args)
2624 char *local_name;
2625 char *remote_name;
2627 if (!args[1]) {
2628 d_printf("reput <filename>\n");
2629 return 1;
2631 local_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2633 if (!file_exist(local_name)) {
2634 d_printf("%s does not exist\n", local_name);
2635 return 1;
2638 if (args[2])
2639 remote_name = talloc_strdup(ctx, args[2]);
2640 else
2641 remote_name = talloc_strdup(ctx, args[1]);
2643 dos_clean_name(remote_name);
2645 return do_put(ctx, remote_name, local_name, true);
2650 return a string representing a share type
2652 static const char *share_type_str(uint32_t type)
2654 switch (type & 0xF) {
2655 case STYPE_DISKTREE:
2656 return "Disk";
2657 case STYPE_PRINTQ:
2658 return "Printer";
2659 case STYPE_DEVICE:
2660 return "Device";
2661 case STYPE_IPC:
2662 return "IPC";
2663 default:
2664 return "Unknown";
2670 display a list of shares from a level 1 share enum
2672 static void display_share_result(struct srvsvc_NetShareCtr1 *ctr1)
2674 int i;
2676 for (i=0;i<ctr1->count;i++) {
2677 struct srvsvc_NetShareInfo1 *info = ctr1->array+i;
2679 printf("\t%-15s %-10.10s %s\n",
2680 info->name,
2681 share_type_str(info->type),
2682 info->comment);
2688 /****************************************************************************
2689 try and browse available shares on a host
2690 ****************************************************************************/
2691 static bool browse_host(struct loadparm_context *lp_ctx,
2692 struct tevent_context *ev_ctx,
2693 const char *query_host)
2695 struct dcerpc_pipe *p;
2696 char *binding;
2697 NTSTATUS status;
2698 struct srvsvc_NetShareEnumAll r;
2699 struct srvsvc_NetShareInfoCtr info_ctr;
2700 uint32_t resume_handle = 0;
2701 TALLOC_CTX *mem_ctx = talloc_init("browse_host");
2702 struct srvsvc_NetShareCtr1 ctr1;
2703 uint32_t totalentries = 0;
2705 binding = talloc_asprintf(mem_ctx, "ncacn_np:%s", query_host);
2707 status = dcerpc_pipe_connect(mem_ctx, &p, binding,
2708 &ndr_table_srvsvc,
2709 popt_get_cmdline_credentials(), ev_ctx,
2710 lp_ctx);
2711 if (!NT_STATUS_IS_OK(status)) {
2712 d_printf("Failed to connect to %s - %s\n",
2713 binding, nt_errstr(status));
2714 talloc_free(mem_ctx);
2715 return false;
2718 info_ctr.level = 1;
2719 info_ctr.ctr.ctr1 = &ctr1;
2721 r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
2722 r.in.info_ctr = &info_ctr;
2723 r.in.max_buffer = ~0;
2724 r.in.resume_handle = &resume_handle;
2725 r.out.resume_handle = &resume_handle;
2726 r.out.totalentries = &totalentries;
2727 r.out.info_ctr = &info_ctr;
2729 d_printf("\n\tSharename Type Comment\n");
2730 d_printf("\t--------- ---- -------\n");
2732 do {
2733 ZERO_STRUCT(ctr1);
2734 status = dcerpc_srvsvc_NetShareEnumAll_r(p->binding_handle, mem_ctx, &r);
2736 if (NT_STATUS_IS_OK(status) &&
2737 (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA) ||
2738 W_ERROR_IS_OK(r.out.result)) &&
2739 r.out.info_ctr->ctr.ctr1) {
2740 display_share_result(r.out.info_ctr->ctr.ctr1);
2741 resume_handle += r.out.info_ctr->ctr.ctr1->count;
2743 } while (NT_STATUS_IS_OK(status) && W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA));
2745 talloc_free(mem_ctx);
2747 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2748 d_printf("Failed NetShareEnumAll %s - %s/%s\n",
2749 binding, nt_errstr(status), win_errstr(r.out.result));
2750 return false;
2753 return false;
2756 /****************************************************************************
2757 try and browse available connections on a host
2758 ****************************************************************************/
2759 static bool list_servers(const char *wk_grp)
2761 d_printf("REWRITE: list servers not implemented\n");
2762 return false;
2765 /* Some constants for completing filename arguments */
2767 #define COMPL_NONE 0 /* No completions */
2768 #define COMPL_REMOTE 1 /* Complete remote filename */
2769 #define COMPL_LOCAL 2 /* Complete local filename */
2771 static int cmd_help(struct smbclient_context *ctx, const char **args);
2773 /* This defines the commands supported by this client.
2774 * NOTE: The "!" must be the last one in the list because it's fn pointer
2775 * field is NULL, and NULL in that field is used in process_tok()
2776 * (below) to indicate the end of the list. crh
2778 static struct
2780 const char *name;
2781 int (*fn)(struct smbclient_context *ctx, const char **args);
2782 const char *description;
2783 char compl_args[2]; /* Completion argument info */
2784 } commands[] =
2786 {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2787 {"addprivileges",cmd_addprivileges,"<sid|name> <privilege...> add privileges for a user",{COMPL_NONE,COMPL_NONE}},
2788 {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
2789 {"acl",cmd_acl,"<file> show file ACL",{COMPL_NONE,COMPL_NONE}},
2790 {"allinfo",cmd_allinfo,"<file> show all possible info about a file",{COMPL_NONE,COMPL_NONE}},
2791 {"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}},
2792 {"cancel",cmd_rewrite,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
2793 {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
2794 {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
2795 {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
2796 {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2797 {"delprivileges",cmd_delprivileges,"<sid|name> <privilege...> remove privileges for a user",{COMPL_NONE,COMPL_NONE}},
2798 {"deltree",cmd_deltree,"<dir> delete a whole directory tree",{COMPL_REMOTE,COMPL_NONE}},
2799 {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2800 {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2801 {"eainfo",cmd_eainfo,"<file> show EA contents for a file",{COMPL_NONE,COMPL_NONE}},
2802 {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2803 {"fsinfo",cmd_fsinfo,"query file system info",{COMPL_NONE,COMPL_NONE}},
2804 {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
2805 {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2806 {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
2807 {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
2808 {"link",cmd_link,"<src> <dest> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2809 {"lookup",cmd_lookup,"<sid|name> show SID for name or name for SID",{COMPL_NONE,COMPL_NONE}},
2810 {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},
2811 {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2812 {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
2813 {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2814 {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
2815 {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2816 {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},
2817 {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
2818 {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
2819 {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
2820 {"close",cmd_close,"<fnum> close a file",{COMPL_NONE,COMPL_NONE}},
2821 {"privileges",cmd_privileges,"<user> show privileges for a user",{COMPL_NONE,COMPL_NONE}},
2822 {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
2823 {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
2824 {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},
2825 {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
2826 {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
2827 {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2828 {"queue",cmd_rewrite,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2829 {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2830 {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2831 {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},
2832 {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
2833 {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2834 {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
2835 {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2836 {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2837 {"symlink",cmd_symlink,"<src> <dest> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2838 {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2840 /* Yes, this must be here, see crh's comment above. */
2841 {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2842 {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2846 /*******************************************************************
2847 lookup a command string in the list of commands, including
2848 abbreviations
2849 ******************************************************************/
2850 static int process_tok(const char *tok)
2852 size_t i = 0, matches = 0;
2853 size_t cmd=0;
2854 size_t tok_len = strlen(tok);
2856 while (commands[i].fn != NULL) {
2857 if (strequal(commands[i].name,tok)) {
2858 matches = 1;
2859 cmd = i;
2860 break;
2861 } else if (strncasecmp(commands[i].name, tok, tok_len) == 0) {
2862 matches++;
2863 cmd = i;
2865 i++;
2868 if (matches == 0)
2869 return(-1);
2870 else if (matches == 1)
2871 return(cmd);
2872 else
2873 return(-2);
2876 /****************************************************************************
2877 help
2878 ****************************************************************************/
2879 static int cmd_help(struct smbclient_context *ctx, const char **args)
2881 int i=0,j;
2883 if (args[1]) {
2884 if ((i = process_tok(args[1])) >= 0)
2885 d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2886 } else {
2887 while (commands[i].description) {
2888 for (j=0; commands[i].description && (j<5); j++) {
2889 d_printf("%-15s",commands[i].name);
2890 i++;
2892 d_printf("\n");
2895 return 0;
2898 static int process_line(struct smbclient_context *ctx, const char *cline);
2899 /****************************************************************************
2900 process a -c command string
2901 ****************************************************************************/
2902 static int process_command_string(struct smbclient_context *ctx, const char *cmd)
2904 char **lines;
2905 int i, rc = 0;
2907 lines = str_list_make(NULL, cmd, ";");
2908 for (i = 0; lines[i]; i++) {
2909 rc |= process_line(ctx, lines[i]);
2911 talloc_free(lines);
2913 return rc;
2916 #define MAX_COMPLETIONS 100
2918 typedef struct {
2919 char *dirmask;
2920 char **matches;
2921 int count, samelen;
2922 const char *text;
2923 int len;
2924 } completion_remote_t;
2926 static void completion_remote_filter(struct clilist_file_info *f, const char *mask, void *state)
2928 completion_remote_t *info = (completion_remote_t *)state;
2930 if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (!ISDOT(f->name)) && (!ISDOTDOT(f->name))) {
2931 if ((info->dirmask[0] == 0) && !(f->attrib & FILE_ATTRIBUTE_DIRECTORY))
2932 info->matches[info->count] = strdup(f->name);
2933 else {
2934 char *tmp;
2936 if (info->dirmask[0] != 0)
2937 tmp = talloc_asprintf(NULL, "%s/%s", info->dirmask, f->name);
2938 else
2939 tmp = talloc_strdup(NULL, f->name);
2941 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2942 tmp = talloc_append_string(NULL, tmp, "/");
2943 info->matches[info->count] = tmp;
2945 if (info->matches[info->count] == NULL)
2946 return;
2947 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2948 smb_readline_ca_char(0);
2950 if (info->count == 1)
2951 info->samelen = strlen(info->matches[info->count]);
2952 else
2953 while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0)
2954 info->samelen--;
2955 info->count++;
2959 static char **remote_completion(const char *text, int len)
2961 char *dirmask;
2962 int i, ret;
2963 completion_remote_t info;
2965 info.samelen = len;
2966 info.text = text;
2967 info.len = len;
2968 info.count = 0;
2970 if (len >= PATH_MAX)
2971 return(NULL);
2973 info.matches = malloc_array_p(char *, MAX_COMPLETIONS);
2974 if (!info.matches) return NULL;
2975 info.matches[0] = NULL;
2977 for (i = len-1; i >= 0; i--)
2978 if ((text[i] == '/') || (text[i] == '\\'))
2979 break;
2980 info.text = text+i+1;
2981 info.samelen = info.len = len-i-1;
2983 if (i > 0) {
2984 info.dirmask = talloc_strndup(NULL, text, i+1);
2985 info.dirmask[i+1] = 0;
2986 ret = asprintf(&dirmask, "%s%*s*", rl_ctx->remote_cur_dir, i-1,
2987 text);
2988 } else {
2989 ret = asprintf(&dirmask, "%s*", rl_ctx->remote_cur_dir);
2991 if (ret < 0) {
2992 goto cleanup;
2995 if (smbcli_list(rl_ctx->cli->tree, dirmask,
2996 FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN,
2997 completion_remote_filter, &info) < 0)
2998 goto cleanup;
3000 if (info.count == 2)
3001 info.matches[0] = strdup(info.matches[1]);
3002 else {
3003 info.matches[0] = malloc_array_p(char, info.samelen+1);
3004 if (!info.matches[0])
3005 goto cleanup;
3006 strncpy(info.matches[0], info.matches[1], info.samelen);
3007 info.matches[0][info.samelen] = 0;
3009 info.matches[info.count] = NULL;
3010 return info.matches;
3012 cleanup:
3013 for (i = 0; i < info.count; i++)
3014 free(info.matches[i]);
3015 free(info.matches);
3016 return NULL;
3019 static char **completion_fn(const char *text, int start, int end)
3021 smb_readline_ca_char(' ');
3023 if (start) {
3024 const char *buf, *sp;
3025 int i;
3026 char compl_type;
3028 buf = smb_readline_get_line_buffer();
3029 if (buf == NULL)
3030 return NULL;
3032 sp = strchr(buf, ' ');
3033 if (sp == NULL)
3034 return NULL;
3036 for (i = 0; commands[i].name; i++)
3037 if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0))
3038 break;
3039 if (commands[i].name == NULL)
3040 return NULL;
3042 while (*sp == ' ')
3043 sp++;
3045 if (sp == (buf + start))
3046 compl_type = commands[i].compl_args[0];
3047 else
3048 compl_type = commands[i].compl_args[1];
3050 if (compl_type == COMPL_REMOTE)
3051 return remote_completion(text, end - start);
3052 else /* fall back to local filename completion */
3053 return NULL;
3054 } else {
3055 char **matches;
3056 size_t i, len, samelen = 0, count=1;
3058 matches = malloc_array_p(char *, MAX_COMPLETIONS);
3059 if (!matches) return NULL;
3060 matches[0] = NULL;
3062 len = strlen(text);
3063 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
3064 if (strncmp(text, commands[i].name, len) == 0) {
3065 matches[count] = strdup(commands[i].name);
3066 if (!matches[count])
3067 goto cleanup;
3068 if (count == 1)
3069 samelen = strlen(matches[count]);
3070 else
3071 while (strncmp(matches[count], matches[count-1], samelen) != 0)
3072 samelen--;
3073 count++;
3077 switch (count) {
3078 case 0: /* should never happen */
3079 case 1:
3080 goto cleanup;
3081 case 2:
3082 matches[0] = strdup(matches[1]);
3083 break;
3084 default:
3085 matches[0] = malloc_array_p(char, samelen+1);
3086 if (!matches[0])
3087 goto cleanup;
3088 strncpy(matches[0], matches[1], samelen);
3089 matches[0][samelen] = 0;
3091 matches[count] = NULL;
3092 return matches;
3094 cleanup:
3095 for (i = 0; i < count; i++) {
3096 free(matches[i]);
3098 free(matches);
3099 return NULL;
3103 /****************************************************************************
3104 make sure we swallow keepalives during idle time
3105 ****************************************************************************/
3106 static void readline_callback(void)
3108 static time_t last_t;
3109 time_t t;
3111 t = time(NULL);
3113 if (t - last_t < 5) return;
3115 last_t = t;
3117 smbcli_transport_process(rl_ctx->cli->transport);
3119 if (rl_ctx->cli->tree) {
3120 smbcli_chkpath(rl_ctx->cli->tree, "\\");
3124 static int process_line(struct smbclient_context *ctx, const char *cline)
3126 char **args;
3127 int i;
3129 /* and get the first part of the command */
3130 args = str_list_make_shell(ctx, cline, NULL);
3131 if (!args || !args[0])
3132 return 0;
3134 if ((i = process_tok(args[0])) >= 0) {
3135 const char **a = discard_const_p(const char *, args);
3136 i = commands[i].fn(ctx, a);
3137 } else if (i == -2) {
3138 d_printf("%s: command abbreviation ambiguous\n",args[0]);
3139 } else {
3140 d_printf("%s: command not found\n",args[0]);
3143 talloc_free(args);
3145 return i;
3148 /****************************************************************************
3149 process commands on stdin
3150 ****************************************************************************/
3151 static int process_stdin(struct smbclient_context *ctx)
3153 int rc = 0;
3154 while (1) {
3155 /* display a prompt */
3156 char *the_prompt = talloc_asprintf(ctx, "smb: %s> ", ctx->remote_cur_dir);
3157 char *cline = smb_readline(the_prompt, readline_callback, completion_fn);
3158 talloc_free(the_prompt);
3160 if (!cline) break;
3162 /* special case - first char is ! */
3163 if (*cline == '!') {
3164 int ret;
3165 ret = system(cline + 1);
3166 free(cline);
3167 if (ret == -1) {
3168 rc |= ret;
3170 continue;
3173 rc |= process_command_string(ctx, cline);
3174 free(cline);
3178 return rc;
3182 /*****************************************************
3183 return a connection to a server
3184 *******************************************************/
3185 static bool do_connect(struct smbclient_context *ctx,
3186 struct tevent_context *ev_ctx,
3187 struct resolve_context *resolve_ctx,
3188 const char *specified_server, const char **ports,
3189 const char *specified_share,
3190 const char *socket_options,
3191 struct cli_credentials *cred,
3192 struct smbcli_options *options,
3193 struct smbcli_session_options *session_options,
3194 struct gensec_settings *gensec_settings)
3196 NTSTATUS status;
3197 char *server, *share;
3199 rl_ctx = ctx; /* Ugly hack */
3201 if (strncmp(specified_share, "\\\\", 2) == 0 ||
3202 strncmp(specified_share, "//", 2) == 0) {
3203 bool ok;
3205 ok = smbcli_parse_unc(specified_share, ctx, &server, &share);
3206 if (!ok) {
3207 d_printf("Failed to parse UNC\n");
3208 talloc_free(ctx);
3209 return false;
3211 } else {
3212 share = talloc_strdup(ctx, specified_share);
3213 server = talloc_strdup(ctx, specified_server);
3214 if (share == NULL || server == NULL) {
3215 d_printf("Failed to allocate memory for server and share\n");
3216 talloc_free(ctx);
3217 return false;
3221 ctx->remote_cur_dir = talloc_strdup(ctx, "\\");
3222 if (ctx->remote_cur_dir == NULL) {
3223 talloc_free(ctx);
3224 return false;
3227 status = smbcli_full_connection(ctx, &ctx->cli, server, ports,
3228 share, NULL,
3229 socket_options,
3230 cred, resolve_ctx,
3231 ev_ctx, options, session_options,
3232 gensec_settings);
3233 if (!NT_STATUS_IS_OK(status)) {
3234 d_printf("Connection to \\\\%s\\%s failed - %s\n",
3235 server, share, nt_errstr(status));
3236 talloc_free(ctx);
3237 return false;
3240 return true;
3243 /****************************************************************************
3244 handle a -L query
3245 ****************************************************************************/
3246 static int do_host_query(struct loadparm_context *lp_ctx,
3247 struct tevent_context *ev_ctx,
3248 const char *query_host,
3249 const char *workgroup)
3251 browse_host(lp_ctx, ev_ctx, query_host);
3252 list_servers(workgroup);
3253 return(0);
3257 /****************************************************************************
3258 handle a message operation
3259 ****************************************************************************/
3260 static int do_message_op(const char *netbios_name, const char *desthost,
3261 const char **destports, const char *destip,
3262 int name_type,
3263 struct tevent_context *ev_ctx,
3264 struct resolve_context *resolve_ctx,
3265 struct smbcli_options *options,
3266 const char *socket_options)
3268 struct nbt_name called, calling;
3269 const char *server_name;
3270 struct smbcli_state *cli;
3271 bool ok;
3273 make_nbt_name_client(&calling, netbios_name);
3275 nbt_choose_called_name(NULL, &called, desthost, name_type);
3277 server_name = destip ? destip : desthost;
3279 cli = smbcli_state_init(NULL);
3280 if (cli == NULL) {
3281 d_printf("smbcli_state_init() failed\n");
3282 return 1;
3285 ok = smbcli_socket_connect(cli, server_name, destports,
3286 ev_ctx, resolve_ctx, options,
3287 socket_options,
3288 &calling, &called);
3289 if (!ok) {
3290 d_printf("Connection to %s failed\n", server_name);
3291 return 1;
3294 send_message(cli, desthost);
3295 talloc_free(cli);
3297 return 0;
3301 /****************************************************************************
3302 main program
3303 ****************************************************************************/
3304 int main(int argc, const char *argv[])
3306 char *base_directory = NULL;
3307 const char *dest_ip = NULL;
3308 int opt;
3309 const char *query_host = NULL;
3310 bool message = false;
3311 char *desthost = NULL;
3312 poptContext pc;
3313 const char *service = NULL;
3314 int port = 0;
3315 char *p;
3316 int rc = 0;
3317 int name_type = 0x20;
3318 TALLOC_CTX *mem_ctx;
3319 struct tevent_context *ev_ctx;
3320 struct smbclient_context *ctx;
3321 const char *cmdstr = NULL;
3322 struct smbcli_options smb_options;
3323 struct smbcli_session_options smb_session_options;
3325 struct poptOption long_options[] = {
3326 POPT_AUTOHELP
3328 { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" },
3329 { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" },
3330 { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" },
3331 { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" },
3332 { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" },
3333 { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" },
3334 { "send-buffer", 'b', POPT_ARG_INT, NULL, 'b', "Changes the transmit/send buffer", "BYTES" },
3335 { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" },
3336 POPT_COMMON_SAMBA
3337 POPT_COMMON_CONNECTION
3338 POPT_COMMON_CREDENTIALS
3339 POPT_COMMON_VERSION
3340 { NULL }
3343 mem_ctx = talloc_init("client.c/main");
3344 if (!mem_ctx) {
3345 d_printf("\nclient.c: Not enough memory\n");
3346 exit(1);
3349 ctx = talloc_zero(mem_ctx, struct smbclient_context);
3350 ctx->io_bufsize = 64512;
3352 pc = poptGetContext("smbclient", argc, argv, long_options, 0);
3353 poptSetOtherOptionHelp(pc, "[OPTIONS] service <password>");
3355 while ((opt = poptGetNextOpt(pc)) != -1) {
3356 switch (opt) {
3357 case 'M':
3358 /* Messages are sent to NetBIOS name type 0x3
3359 * (Messenger Service). Make sure we default
3360 * to port 139 instead of port 445. srl,crh
3362 name_type = 0x03;
3363 desthost = strdup(poptGetOptArg(pc));
3364 if( 0 == port ) port = 139;
3365 message = true;
3366 break;
3367 case 'I':
3368 dest_ip = poptGetOptArg(pc);
3369 break;
3370 case 'L':
3371 query_host = strdup(poptGetOptArg(pc));
3372 break;
3373 case 'D':
3374 base_directory = strdup(poptGetOptArg(pc));
3375 break;
3376 case 'b':
3377 ctx->io_bufsize = MAX(1, atoi(poptGetOptArg(pc)));
3378 break;
3382 gensec_init();
3384 if(poptPeekArg(pc)) {
3385 char *s = strdup(poptGetArg(pc));
3387 /* Convert any '/' characters in the service name to '\' characters */
3388 string_replace(s, '/','\\');
3390 service = s;
3392 if (count_chars(s,'\\') < 3) {
3393 d_printf("\n%s: Not enough '\\' characters in service\n",s);
3394 poptPrintUsage(pc, stderr, 0);
3395 exit(1);
3399 if (poptPeekArg(pc)) {
3400 cli_credentials_set_password(popt_get_cmdline_credentials(),
3401 poptGetArg(pc), CRED_SPECIFIED);
3404 /*init_names(); */
3406 if (!query_host && !service && !message) {
3407 poptPrintUsage(pc, stderr, 0);
3408 exit(1);
3411 poptFreeContext(pc);
3413 lpcfg_smbcli_options(cmdline_lp_ctx, &smb_options);
3414 lpcfg_smbcli_session_options(cmdline_lp_ctx, &smb_session_options);
3416 ev_ctx = s4_event_context_init(ctx);
3418 DEBUG( 3, ( "Client started (version %s).\n", SAMBA_VERSION_STRING ) );
3420 if (query_host && (p=strchr_m(query_host,'#'))) {
3421 *p = 0;
3422 p++;
3423 sscanf(p, "%x", &name_type);
3426 if (query_host) {
3427 rc = do_host_query(cmdline_lp_ctx, ev_ctx, query_host,
3428 lpcfg_workgroup(cmdline_lp_ctx));
3429 return rc;
3432 if (message) {
3433 rc = do_message_op(lpcfg_netbios_name(cmdline_lp_ctx), desthost,
3434 lpcfg_smb_ports(cmdline_lp_ctx), dest_ip,
3435 name_type, ev_ctx,
3436 lpcfg_resolve_context(cmdline_lp_ctx),
3437 &smb_options,
3438 lpcfg_socket_options(cmdline_lp_ctx));
3439 return rc;
3442 if (!do_connect(ctx, ev_ctx, lpcfg_resolve_context(cmdline_lp_ctx),
3443 desthost, lpcfg_smb_ports(cmdline_lp_ctx), service,
3444 lpcfg_socket_options(cmdline_lp_ctx),
3445 popt_get_cmdline_credentials(),
3446 &smb_options, &smb_session_options,
3447 lpcfg_gensec_settings(ctx, cmdline_lp_ctx)))
3448 return 1;
3450 if (base_directory) {
3451 do_cd(ctx, base_directory);
3452 free(base_directory);
3455 if (cmdstr) {
3456 rc = process_command_string(ctx, cmdstr);
3457 } else {
3458 rc = process_stdin(ctx);
3461 free(desthost);
3462 talloc_free(mem_ctx);
3464 return rc;