testtools: Import new upstream snapshot.
[Samba/gebeck_regimport.git] / source4 / client / client.c
blobc30d17d3eb32ee733acd070f14638c58205f271e
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 "libcli/util/clilsa.h"
39 #include "system/dir.h"
40 #include "system/filesys.h"
41 #include "../lib/util/dlinklist.h"
42 #include "system/readline.h"
43 #include "auth/credentials/credentials.h"
44 #include "auth/gensec/gensec.h"
45 #include "system/time.h" /* needed by some systems for asctime() */
46 #include "libcli/resolve/resolve.h"
47 #include "libcli/security/security.h"
48 #include "../libcli/smbreadline/smbreadline.h"
49 #include "librpc/gen_ndr/ndr_nbt.h"
50 #include "param/param.h"
51 #include "libcli/raw/raw_proto.h"
53 /* the default pager to use for the client "more" command. Users can
54 * override this with the PAGER environment variable */
55 #ifndef DEFAULT_PAGER
56 #define DEFAULT_PAGER "more"
57 #endif
59 struct smbclient_context {
60 char *remote_cur_dir;
61 struct smbcli_state *cli;
62 char *fileselection;
63 time_t newer_than;
64 bool prompt;
65 bool recurse;
66 int archive_level;
67 bool lowercase;
68 int printmode;
69 bool translation;
70 int io_bufsize;
73 /* timing globals */
74 static uint64_t get_total_size = 0;
75 static unsigned int get_total_time_ms = 0;
76 static uint64_t put_total_size = 0;
77 static unsigned int put_total_time_ms = 0;
79 /* Unfortunately, there is no way to pass the a context to the completion function as an argument */
80 static struct smbclient_context *rl_ctx;
82 /* totals globals */
83 static double dir_total;
85 /*******************************************************************
86 Reduce a file name, removing .. elements.
87 ********************************************************************/
88 static void dos_clean_name(char *s)
90 char *p=NULL,*r;
92 DEBUG(3,("dos_clean_name [%s]\n",s));
94 /* remove any double slashes */
95 all_string_sub(s, "\\\\", "\\", 0);
97 while ((p = strstr(s,"\\..\\")) != NULL) {
98 *p = '\0';
99 if ((r = strrchr(s,'\\')) != NULL)
100 memmove(r,p+3,strlen(p+3)+1);
103 trim_string(s,NULL,"\\..");
105 all_string_sub(s, "\\.\\", "\\", 0);
108 /****************************************************************************
109 write to a local file with CR/LF->LF translation if appropriate. return the
110 number taken from the buffer. This may not equal the number written.
111 ****************************************************************************/
112 static int writefile(int f, const void *_b, int n, bool translation)
114 const uint8_t *b = (const uint8_t *)_b;
115 int i;
117 if (!translation) {
118 return write(f,b,n);
121 i = 0;
122 while (i < n) {
123 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
124 b++;i++;
126 if (write(f, b, 1) != 1) {
127 break;
129 b++;
130 i++;
133 return(i);
136 /****************************************************************************
137 read from a file with LF->CR/LF translation if appropriate. return the
138 number read. read approx n bytes.
139 ****************************************************************************/
140 static int readfile(void *_b, int n, XFILE *f, bool translation)
142 uint8_t *b = (uint8_t *)_b;
143 int i;
144 int c;
146 if (!translation)
147 return x_fread(b,1,n,f);
149 i = 0;
150 while (i < (n - 1)) {
151 if ((c = x_getc(f)) == EOF) {
152 break;
155 if (c == '\n') { /* change all LFs to CR/LF */
156 b[i++] = '\r';
159 b[i++] = c;
162 return(i);
166 /****************************************************************************
167 send a message
168 ****************************************************************************/
169 static void send_message(struct smbcli_state *cli, const char *desthost)
171 char msg[1600];
172 int total_len = 0;
173 int grp_id;
175 if (!smbcli_message_start(cli->tree, desthost, cli_credentials_get_username(cmdline_credentials), &grp_id)) {
176 d_printf("message start: %s\n", smbcli_errstr(cli->tree));
177 return;
181 d_printf("Connected. Type your message, ending it with a Control-D\n");
183 while (!feof(stdin) && total_len < 1600) {
184 int maxlen = MIN(1600 - total_len,127);
185 int l=0;
186 int c;
188 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
189 if (c == '\n')
190 msg[l++] = '\r';
191 msg[l] = c;
194 if (!smbcli_message_text(cli->tree, msg, l, grp_id)) {
195 d_printf("SMBsendtxt failed (%s)\n",smbcli_errstr(cli->tree));
196 return;
199 total_len += l;
202 if (total_len >= 1600)
203 d_printf("the message was truncated to 1600 bytes\n");
204 else
205 d_printf("sent %d bytes\n",total_len);
207 if (!smbcli_message_end(cli->tree, grp_id)) {
208 d_printf("SMBsendend failed (%s)\n",smbcli_errstr(cli->tree));
209 return;
215 /****************************************************************************
216 check the space on a device
217 ****************************************************************************/
218 static int do_dskattr(struct smbclient_context *ctx)
220 uint32_t bsize;
221 uint64_t total, avail;
223 if (NT_STATUS_IS_ERR(smbcli_dskattr(ctx->cli->tree, &bsize, &total, &avail))) {
224 d_printf("Error in dskattr: %s\n",smbcli_errstr(ctx->cli->tree));
225 return 1;
228 d_printf("\n\t\t%llu blocks of size %u. %llu blocks available\n",
229 (unsigned long long)total,
230 (unsigned)bsize,
231 (unsigned long long)avail);
233 return 0;
236 /****************************************************************************
237 show cd/pwd
238 ****************************************************************************/
239 static int cmd_pwd(struct smbclient_context *ctx, const char **args)
241 d_printf("Current directory is %s\n", ctx->remote_cur_dir);
242 return 0;
246 convert a string to dos format
248 static void dos_format(char *s)
250 string_replace(s, '/', '\\');
253 /****************************************************************************
254 change directory - inner section
255 ****************************************************************************/
256 static int do_cd(struct smbclient_context *ctx, const char *newdir)
258 char *dname;
260 /* Save the current directory in case the
261 new directory is invalid */
262 if (newdir[0] == '\\')
263 dname = talloc_strdup(ctx, newdir);
264 else
265 dname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, newdir);
267 dos_format(dname);
269 if (*(dname+strlen(dname)-1) != '\\') {
270 dname = talloc_append_string(NULL, dname, "\\");
272 dos_clean_name(dname);
274 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, dname))) {
275 d_printf("cd %s: %s\n", dname, smbcli_errstr(ctx->cli->tree));
276 talloc_free(dname);
277 } else {
278 ctx->remote_cur_dir = dname;
281 return 0;
284 /****************************************************************************
285 change directory
286 ****************************************************************************/
287 static int cmd_cd(struct smbclient_context *ctx, const char **args)
289 int rc = 0;
291 if (args[1])
292 rc = do_cd(ctx, args[1]);
293 else
294 d_printf("Current directory is %s\n",ctx->remote_cur_dir);
296 return rc;
300 static bool mask_match(struct smbcli_state *c, const char *string,
301 const char *pattern, bool is_case_sensitive)
303 char *p2, *s2;
304 bool ret;
306 if (ISDOTDOT(string))
307 string = ".";
308 if (ISDOT(pattern))
309 return false;
311 if (is_case_sensitive)
312 return ms_fnmatch_protocol(pattern, string,
313 c->transport->negotiate.protocol) == 0;
315 p2 = strlower_talloc(NULL, pattern);
316 s2 = strlower_talloc(NULL, string);
317 ret = ms_fnmatch_protocol(p2, s2, c->transport->negotiate.protocol) == 0;
318 talloc_free(p2);
319 talloc_free(s2);
321 return ret;
326 /*******************************************************************
327 decide if a file should be operated on
328 ********************************************************************/
329 static bool do_this_one(struct smbclient_context *ctx, struct clilist_file_info *finfo)
331 if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY) return(true);
333 if (ctx->fileselection &&
334 !mask_match(ctx->cli, finfo->name,ctx->fileselection,false)) {
335 DEBUG(3,("mask_match %s failed\n", finfo->name));
336 return false;
339 if (ctx->newer_than && finfo->mtime < ctx->newer_than) {
340 DEBUG(3,("newer_than %s failed\n", finfo->name));
341 return(false);
344 if ((ctx->archive_level==1 || ctx->archive_level==2) && !(finfo->attrib & FILE_ATTRIBUTE_ARCHIVE)) {
345 DEBUG(3,("archive %s failed\n", finfo->name));
346 return(false);
349 return(true);
352 /****************************************************************************
353 display info about a file
354 ****************************************************************************/
355 static void display_finfo(struct smbclient_context *ctx, struct clilist_file_info *finfo)
357 if (do_this_one(ctx, finfo)) {
358 time_t t = finfo->mtime; /* the time is assumed to be passed as GMT */
359 char *astr = attrib_string(NULL, finfo->attrib);
360 d_printf(" %-30s%7.7s %8.0f %s",
361 finfo->name,
362 astr,
363 (double)finfo->size,
364 asctime(localtime(&t)));
365 dir_total += finfo->size;
366 talloc_free(astr);
371 /****************************************************************************
372 accumulate size of a file
373 ****************************************************************************/
374 static void do_du(struct smbclient_context *ctx, struct clilist_file_info *finfo)
376 if (do_this_one(ctx, finfo)) {
377 dir_total += finfo->size;
381 static bool do_list_recurse;
382 static bool do_list_dirs;
383 static char *do_list_queue = 0;
384 static long do_list_queue_size = 0;
385 static long do_list_queue_start = 0;
386 static long do_list_queue_end = 0;
387 static void (*do_list_fn)(struct smbclient_context *, struct clilist_file_info *);
389 /****************************************************************************
390 functions for do_list_queue
391 ****************************************************************************/
394 * The do_list_queue is a NUL-separated list of strings stored in a
395 * char*. Since this is a FIFO, we keep track of the beginning and
396 * ending locations of the data in the queue. When we overflow, we
397 * double the size of the char*. When the start of the data passes
398 * the midpoint, we move everything back. This is logically more
399 * complex than a linked list, but easier from a memory management
400 * angle. In any memory error condition, do_list_queue is reset.
401 * Functions check to ensure that do_list_queue is non-NULL before
402 * accessing it.
404 static void reset_do_list_queue(void)
406 SAFE_FREE(do_list_queue);
407 do_list_queue_size = 0;
408 do_list_queue_start = 0;
409 do_list_queue_end = 0;
412 static void init_do_list_queue(void)
414 reset_do_list_queue();
415 do_list_queue_size = 1024;
416 do_list_queue = malloc_array_p(char, do_list_queue_size);
417 if (do_list_queue == 0) {
418 d_printf("malloc fail for size %d\n",
419 (int)do_list_queue_size);
420 reset_do_list_queue();
421 } else {
422 memset(do_list_queue, 0, do_list_queue_size);
426 static void adjust_do_list_queue(void)
428 if (do_list_queue == NULL) return;
431 * If the starting point of the queue is more than half way through,
432 * move everything toward the beginning.
434 if (do_list_queue_start == do_list_queue_end)
436 DEBUG(4,("do_list_queue is empty\n"));
437 do_list_queue_start = do_list_queue_end = 0;
438 *do_list_queue = '\0';
440 else if (do_list_queue_start > (do_list_queue_size / 2))
442 DEBUG(4,("sliding do_list_queue backward\n"));
443 memmove(do_list_queue,
444 do_list_queue + do_list_queue_start,
445 do_list_queue_end - do_list_queue_start);
446 do_list_queue_end -= do_list_queue_start;
447 do_list_queue_start = 0;
452 static void add_to_do_list_queue(const char* entry)
454 char *dlq;
455 long new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
456 while (new_end > do_list_queue_size)
458 do_list_queue_size *= 2;
459 DEBUG(4,("enlarging do_list_queue to %d\n",
460 (int)do_list_queue_size));
461 dlq = realloc_p(do_list_queue, char, do_list_queue_size);
462 if (! dlq) {
463 d_printf("failure enlarging do_list_queue to %d bytes\n",
464 (int)do_list_queue_size);
465 reset_do_list_queue();
467 else
469 do_list_queue = dlq;
470 memset(do_list_queue + do_list_queue_size / 2,
471 0, do_list_queue_size / 2);
474 if (do_list_queue)
476 strlcpy(do_list_queue + do_list_queue_end, entry ? entry : "",
477 do_list_queue_size - do_list_queue_end);
478 do_list_queue_end = new_end;
479 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
480 entry, (int)do_list_queue_start, (int)do_list_queue_end));
484 static char *do_list_queue_head(void)
486 return do_list_queue + do_list_queue_start;
489 static void remove_do_list_queue_head(void)
491 if (do_list_queue_end > do_list_queue_start)
493 do_list_queue_start += strlen(do_list_queue_head()) + 1;
494 adjust_do_list_queue();
495 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
496 (int)do_list_queue_start, (int)do_list_queue_end));
500 static int do_list_queue_empty(void)
502 return (! (do_list_queue && *do_list_queue));
505 /****************************************************************************
506 a helper for do_list
507 ****************************************************************************/
508 static void do_list_helper(struct clilist_file_info *f, const char *mask, void *state)
510 struct smbclient_context *ctx = (struct smbclient_context *)state;
512 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY) {
513 if (do_list_dirs && do_this_one(ctx, f)) {
514 do_list_fn(ctx, f);
516 if (do_list_recurse &&
517 !ISDOT(f->name) &&
518 !ISDOTDOT(f->name)) {
519 char *mask2;
520 char *p;
522 mask2 = talloc_strdup(NULL, mask);
523 p = strrchr_m(mask2,'\\');
524 if (!p) return;
525 p[1] = 0;
526 mask2 = talloc_asprintf_append_buffer(mask2, "%s\\*", f->name);
527 add_to_do_list_queue(mask2);
529 return;
532 if (do_this_one(ctx, f)) {
533 do_list_fn(ctx, f);
538 /****************************************************************************
539 a wrapper around smbcli_list that adds recursion
540 ****************************************************************************/
541 static void do_list(struct smbclient_context *ctx, const char *mask,uint16_t attribute,
542 void (*fn)(struct smbclient_context *, struct clilist_file_info *),bool rec, bool dirs)
544 static int in_do_list = 0;
546 if (in_do_list && rec)
548 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
549 exit(1);
552 in_do_list = 1;
554 do_list_recurse = rec;
555 do_list_dirs = dirs;
556 do_list_fn = fn;
558 if (rec)
560 init_do_list_queue();
561 add_to_do_list_queue(mask);
563 while (! do_list_queue_empty())
566 * Need to copy head so that it doesn't become
567 * invalid inside the call to smbcli_list. This
568 * would happen if the list were expanded
569 * during the call.
570 * Fix from E. Jay Berkenbilt (ejb@ql.org)
572 char *head;
573 head = do_list_queue_head();
574 smbcli_list(ctx->cli->tree, head, attribute, do_list_helper, ctx);
575 remove_do_list_queue_head();
576 if ((! do_list_queue_empty()) && (fn == display_finfo))
578 char* next_file = do_list_queue_head();
579 char* save_ch = 0;
580 if ((strlen(next_file) >= 2) &&
581 (next_file[strlen(next_file) - 1] == '*') &&
582 (next_file[strlen(next_file) - 2] == '\\'))
584 save_ch = next_file +
585 strlen(next_file) - 2;
586 *save_ch = '\0';
588 d_printf("\n%s\n",next_file);
589 if (save_ch)
591 *save_ch = '\\';
596 else
598 if (smbcli_list(ctx->cli->tree, mask, attribute, do_list_helper, ctx) == -1)
600 d_printf("%s listing %s\n", smbcli_errstr(ctx->cli->tree), mask);
604 in_do_list = 0;
605 reset_do_list_queue();
608 /****************************************************************************
609 get a directory listing
610 ****************************************************************************/
611 static int cmd_dir(struct smbclient_context *ctx, const char **args)
613 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
614 char *mask;
615 int rc;
617 dir_total = 0;
619 mask = talloc_strdup(ctx, ctx->remote_cur_dir);
620 if(mask[strlen(mask)-1]!='\\')
621 mask = talloc_append_string(ctx, mask,"\\");
623 if (args[1]) {
624 mask = talloc_strdup(ctx, args[1]);
625 if (mask[0] != '\\')
626 mask = talloc_append_string(ctx, mask, "\\");
627 dos_format(mask);
629 else {
630 if (ctx->cli->tree->session->transport->negotiate.protocol <=
631 PROTOCOL_LANMAN1) {
632 mask = talloc_append_string(ctx, mask, "*.*");
633 } else {
634 mask = talloc_append_string(ctx, mask, "*");
638 do_list(ctx, mask, attribute, display_finfo, ctx->recurse, true);
640 rc = do_dskattr(ctx);
642 DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
644 return rc;
648 /****************************************************************************
649 get a directory listing
650 ****************************************************************************/
651 static int cmd_du(struct smbclient_context *ctx, const char **args)
653 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
654 int rc;
655 char *mask;
657 dir_total = 0;
659 if (args[1]) {
660 if (args[1][0] == '\\')
661 mask = talloc_strdup(ctx, args[1]);
662 else
663 mask = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
664 dos_format(mask);
665 } else {
666 mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
669 do_list(ctx, mask, attribute, do_du, ctx->recurse, true);
671 talloc_free(mask);
673 rc = do_dskattr(ctx);
675 d_printf("Total number of bytes: %.0f\n", dir_total);
677 return rc;
681 /****************************************************************************
682 get a file from rname to lname
683 ****************************************************************************/
684 static int do_get(struct smbclient_context *ctx, char *rname, const char *p_lname, bool reget)
686 int handle = 0, fnum;
687 bool newhandle = false;
688 uint8_t *data;
689 struct timeval tp_start;
690 int read_size = ctx->io_bufsize;
691 uint16_t attr;
692 size_t size;
693 off_t start = 0;
694 off_t nread = 0;
695 int rc = 0;
696 char *lname;
699 GetTimeOfDay(&tp_start);
701 if (ctx->lowercase) {
702 lname = strlower_talloc(ctx, p_lname);
703 } else {
704 lname = talloc_strdup(ctx, p_lname);
707 fnum = smbcli_open(ctx->cli->tree, rname, O_RDONLY, DENY_NONE);
709 if (fnum == -1) {
710 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
711 return 1;
714 if(!strcmp(lname,"-")) {
715 handle = fileno(stdout);
716 } else {
717 if (reget) {
718 handle = open(lname, O_WRONLY|O_CREAT, 0644);
719 if (handle >= 0) {
720 start = lseek(handle, 0, SEEK_END);
721 if (start == -1) {
722 d_printf("Error seeking local file\n");
723 return 1;
726 } else {
727 handle = open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
729 newhandle = true;
731 if (handle < 0) {
732 d_printf("Error opening local file %s\n",lname);
733 return 1;
737 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum,
738 &attr, &size, NULL, NULL, NULL, NULL, NULL)) &&
739 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum,
740 &attr, &size, NULL, NULL, NULL))) {
741 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
742 return 1;
745 DEBUG(2,("getting file %s of size %.0f as %s ",
746 rname, (double)size, lname));
748 if(!(data = (uint8_t *)malloc(read_size))) {
749 d_printf("malloc fail for size %d\n", read_size);
750 smbcli_close(ctx->cli->tree, fnum);
751 return 1;
754 while (1) {
755 int n = smbcli_read(ctx->cli->tree, fnum, data, nread + start, read_size);
757 if (n <= 0) break;
759 if (writefile(handle,data, n, ctx->translation) != n) {
760 d_printf("Error writing local file\n");
761 rc = 1;
762 break;
765 nread += n;
768 if (nread + start < size) {
769 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
770 rname, (long)nread));
772 rc = 1;
775 SAFE_FREE(data);
777 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
778 d_printf("Error %s closing remote file\n",smbcli_errstr(ctx->cli->tree));
779 rc = 1;
782 if (newhandle) {
783 close(handle);
786 if (ctx->archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) {
787 smbcli_setatr(ctx->cli->tree, rname, attr & ~(uint16_t)FILE_ATTRIBUTE_ARCHIVE, 0);
791 struct timeval tp_end;
792 int this_time;
794 GetTimeOfDay(&tp_end);
795 this_time =
796 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
797 (tp_end.tv_usec - tp_start.tv_usec)/1000;
798 get_total_time_ms += this_time;
799 get_total_size += nread;
801 DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n",
802 nread / (1.024*this_time + 1.0e-4),
803 get_total_size / (1.024*get_total_time_ms)));
806 return rc;
810 /****************************************************************************
811 get a file
812 ****************************************************************************/
813 static int cmd_get(struct smbclient_context *ctx, const char **args)
815 const char *lname;
816 char *rname;
818 if (!args[1]) {
819 d_printf("get <filename>\n");
820 return 1;
823 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
825 if (args[2])
826 lname = args[2];
827 else
828 lname = args[1];
830 dos_clean_name(rname);
832 return do_get(ctx, rname, lname, false);
835 /****************************************************************************
836 Put up a yes/no prompt.
837 ****************************************************************************/
838 static bool yesno(char *p)
840 char ans[4];
841 printf("%s",p);
843 if (!fgets(ans,sizeof(ans)-1,stdin))
844 return(false);
846 if (*ans == 'y' || *ans == 'Y')
847 return(true);
849 return(false);
852 /****************************************************************************
853 do a mget operation on one file
854 ****************************************************************************/
855 static void do_mget(struct smbclient_context *ctx, struct clilist_file_info *finfo)
857 char *rname;
858 char *quest;
859 char *mget_mask;
860 char *saved_curdir;
861 char *l_fname;
863 if (ISDOT(finfo->name) || ISDOTDOT(finfo->name))
864 return;
866 if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)
867 quest = talloc_asprintf(ctx, "Get directory %s? ",finfo->name);
868 else
869 quest = talloc_asprintf(ctx, "Get file %s? ",finfo->name);
871 if (ctx->prompt && !yesno(quest)) return;
873 talloc_free(quest);
875 if (!(finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)) {
876 rname = talloc_asprintf(ctx, "%s%s",ctx->remote_cur_dir,
877 finfo->name);
878 do_get(ctx, rname, finfo->name, false);
879 talloc_free(rname);
880 return;
883 /* handle directories */
884 saved_curdir = talloc_strdup(ctx, ctx->remote_cur_dir);
886 ctx->remote_cur_dir = talloc_asprintf_append_buffer(NULL, "%s\\", finfo->name);
888 if (ctx->lowercase) {
889 l_fname = strlower_talloc(ctx, finfo->name);
890 } else {
891 l_fname = talloc_strdup(ctx, finfo->name);
894 string_replace(l_fname, '\\', '/');
896 if (!directory_exist(l_fname) &&
897 mkdir(l_fname, 0777) != 0) {
898 d_printf("failed to create directory %s\n", l_fname);
899 return;
902 if (chdir(l_fname) != 0) {
903 d_printf("failed to chdir to directory %s\n", l_fname);
904 return;
907 mget_mask = talloc_asprintf(ctx, "%s*", ctx->remote_cur_dir);
909 do_list(ctx, mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,false, true);
910 chdir("..");
911 talloc_free(ctx->remote_cur_dir);
913 ctx->remote_cur_dir = saved_curdir;
917 /****************************************************************************
918 view the file using the pager
919 ****************************************************************************/
920 static int cmd_more(struct smbclient_context *ctx, const char **args)
922 char *rname;
923 char *pager_cmd;
924 char *lname;
925 char *pager;
926 int fd;
927 int rc = 0;
929 lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir());
930 fd = mkstemp(lname);
931 if (fd == -1) {
932 d_printf("failed to create temporary file for more\n");
933 return 1;
935 close(fd);
937 if (!args[1]) {
938 d_printf("more <filename>\n");
939 unlink(lname);
940 return 1;
942 rname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
943 dos_clean_name(rname);
945 rc = do_get(ctx, rname, lname, false);
947 pager=getenv("PAGER");
949 pager_cmd = talloc_asprintf(ctx, "%s %s",(pager? pager:DEFAULT_PAGER), lname);
950 system(pager_cmd);
951 unlink(lname);
953 return rc;
958 /****************************************************************************
959 do a mget command
960 ****************************************************************************/
961 static int cmd_mget(struct smbclient_context *ctx, const char **args)
963 uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
964 char *mget_mask = NULL;
965 int i;
967 if (ctx->recurse)
968 attribute |= FILE_ATTRIBUTE_DIRECTORY;
970 for (i = 1; args[i]; i++) {
971 mget_mask = talloc_strdup(ctx, ctx->remote_cur_dir);
972 if(mget_mask[strlen(mget_mask)-1]!='\\')
973 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
975 mget_mask = talloc_strdup(ctx, args[i]);
976 if (mget_mask[0] != '\\')
977 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
978 do_list(ctx, mget_mask, attribute,do_mget,false,true);
980 talloc_free(mget_mask);
983 if (mget_mask == NULL) {
984 mget_mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
985 do_list(ctx, mget_mask, attribute,do_mget,false,true);
986 talloc_free(mget_mask);
989 return 0;
993 /****************************************************************************
994 make a directory of name "name"
995 ****************************************************************************/
996 static NTSTATUS do_mkdir(struct smbclient_context *ctx, char *name)
998 NTSTATUS status;
1000 if (NT_STATUS_IS_ERR(status = smbcli_mkdir(ctx->cli->tree, name))) {
1001 d_printf("%s making remote directory %s\n",
1002 smbcli_errstr(ctx->cli->tree),name);
1003 return status;
1006 return status;
1010 /****************************************************************************
1011 Exit client.
1012 ****************************************************************************/
1013 static int cmd_quit(struct smbclient_context *ctx, const char **args)
1015 talloc_free(ctx);
1016 exit(0);
1017 /* NOTREACHED */
1018 return 0;
1022 /****************************************************************************
1023 make a directory
1024 ****************************************************************************/
1025 static int cmd_mkdir(struct smbclient_context *ctx, const char **args)
1027 char *mask, *p;
1029 if (!args[1]) {
1030 if (!ctx->recurse)
1031 d_printf("mkdir <dirname>\n");
1032 return 1;
1035 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir,args[1]);
1037 if (ctx->recurse) {
1038 dos_clean_name(mask);
1040 trim_string(mask,".",NULL);
1041 for (p = strtok(mask,"/\\"); p; p = strtok(p, "/\\")) {
1042 char *parent = talloc_strndup(ctx, mask, PTR_DIFF(p, mask));
1044 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, parent))) {
1045 do_mkdir(ctx, parent);
1048 talloc_free(parent);
1050 } else {
1051 do_mkdir(ctx, mask);
1054 return 0;
1057 /****************************************************************************
1058 show 8.3 name of a file
1059 ****************************************************************************/
1060 static int cmd_altname(struct smbclient_context *ctx, const char **args)
1062 const char *altname;
1063 char *name;
1065 if (!args[1]) {
1066 d_printf("altname <file>\n");
1067 return 1;
1070 name = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1072 if (!NT_STATUS_IS_OK(smbcli_qpathinfo_alt_name(ctx->cli->tree, name, &altname))) {
1073 d_printf("%s getting alt name for %s\n",
1074 smbcli_errstr(ctx->cli->tree),name);
1075 return(false);
1077 d_printf("%s\n", altname);
1079 return 0;
1083 /****************************************************************************
1084 put a single file
1085 ****************************************************************************/
1086 static int do_put(struct smbclient_context *ctx, char *rname, char *lname, bool reput)
1088 int fnum;
1089 XFILE *f;
1090 size_t start = 0;
1091 off_t nread = 0;
1092 uint8_t *buf = NULL;
1093 int maxwrite = ctx->io_bufsize;
1094 int rc = 0;
1096 struct timeval tp_start;
1097 GetTimeOfDay(&tp_start);
1099 if (reput) {
1100 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT, DENY_NONE);
1101 if (fnum >= 0) {
1102 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL)) &&
1103 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL))) {
1104 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
1105 return 1;
1108 } else {
1109 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT|O_TRUNC,
1110 DENY_NONE);
1113 if (fnum == -1) {
1114 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1115 return 1;
1118 /* allow files to be piped into smbclient
1119 jdblair 24.jun.98
1121 Note that in this case this function will exit(0) rather
1122 than returning. */
1123 if (!strcmp(lname, "-")) {
1124 f = x_stdin;
1125 /* size of file is not known */
1126 } else {
1127 f = x_fopen(lname,O_RDONLY, 0);
1128 if (f && reput) {
1129 if (x_tseek(f, start, SEEK_SET) == -1) {
1130 d_printf("Error seeking local file\n");
1131 return 1;
1136 if (!f) {
1137 d_printf("Error opening local file %s\n",lname);
1138 return 1;
1142 DEBUG(1,("putting file %s as %s ",lname,
1143 rname));
1145 buf = (uint8_t *)malloc(maxwrite);
1146 if (!buf) {
1147 d_printf("ERROR: Not enough memory!\n");
1148 return 1;
1150 while (!x_feof(f)) {
1151 int n = maxwrite;
1152 int ret;
1154 if ((n = readfile(buf,n,f,ctx->translation)) < 1) {
1155 if((n == 0) && x_feof(f))
1156 break; /* Empty local file. */
1158 d_printf("Error reading local file: %s\n", strerror(errno));
1159 rc = 1;
1160 break;
1163 ret = smbcli_write(ctx->cli->tree, fnum, 0, buf, nread + start, n);
1165 if (n != ret) {
1166 d_printf("Error writing file: %s\n", smbcli_errstr(ctx->cli->tree));
1167 rc = 1;
1168 break;
1171 nread += n;
1174 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
1175 d_printf("%s closing remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1176 x_fclose(f);
1177 SAFE_FREE(buf);
1178 return 1;
1182 if (f != x_stdin) {
1183 x_fclose(f);
1186 SAFE_FREE(buf);
1189 struct timeval tp_end;
1190 int this_time;
1192 GetTimeOfDay(&tp_end);
1193 this_time =
1194 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1195 (tp_end.tv_usec - tp_start.tv_usec)/1000;
1196 put_total_time_ms += this_time;
1197 put_total_size += nread;
1199 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1200 nread / (1.024*this_time + 1.0e-4),
1201 put_total_size / (1.024*put_total_time_ms)));
1204 if (f == x_stdin) {
1205 talloc_free(ctx);
1206 exit(0);
1209 return rc;
1214 /****************************************************************************
1215 put a file
1216 ****************************************************************************/
1217 static int cmd_put(struct smbclient_context *ctx, const char **args)
1219 char *lname;
1220 char *rname;
1222 if (!args[1]) {
1223 d_printf("put <filename> [<remotename>]\n");
1224 return 1;
1227 lname = talloc_strdup(ctx, args[1]);
1229 if (args[2]) {
1230 if (args[2][0]=='\\')
1231 rname = talloc_strdup(ctx, args[2]);
1232 else
1233 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[2]);
1234 } else {
1235 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, lname);
1238 dos_clean_name(rname);
1240 /* allow '-' to represent stdin
1241 jdblair, 24.jun.98 */
1242 if (!file_exist(lname) && (strcmp(lname,"-"))) {
1243 d_printf("%s does not exist\n",lname);
1244 return 1;
1247 return do_put(ctx, rname, lname, false);
1250 /*************************************
1251 File list structure
1252 *************************************/
1254 static struct file_list {
1255 struct file_list *prev, *next;
1256 char *file_path;
1257 bool isdir;
1258 } *file_list;
1260 /****************************************************************************
1261 Free a file_list structure
1262 ****************************************************************************/
1264 static void free_file_list (struct file_list * list)
1266 struct file_list *tmp;
1268 while (list)
1270 tmp = list;
1271 DLIST_REMOVE(list, list);
1272 SAFE_FREE(tmp->file_path);
1273 SAFE_FREE(tmp);
1277 /****************************************************************************
1278 seek in a directory/file list until you get something that doesn't start with
1279 the specified name
1280 ****************************************************************************/
1281 static bool seek_list(struct file_list *list, char *name)
1283 while (list) {
1284 trim_string(list->file_path,"./","\n");
1285 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1286 return(true);
1288 list = list->next;
1291 return(false);
1294 /****************************************************************************
1295 set the file selection mask
1296 ****************************************************************************/
1297 static int cmd_select(struct smbclient_context *ctx, const char **args)
1299 talloc_free(ctx->fileselection);
1300 ctx->fileselection = talloc_strdup(ctx, args[1]);
1302 return 0;
1305 /*******************************************************************
1306 A readdir wrapper which just returns the file name.
1307 ********************************************************************/
1308 static const char *readdirname(DIR *p)
1310 struct dirent *ptr;
1311 char *dname;
1313 if (!p)
1314 return(NULL);
1316 ptr = (struct dirent *)readdir(p);
1317 if (!ptr)
1318 return(NULL);
1320 dname = ptr->d_name;
1322 #ifdef NEXT2
1323 if (telldir(p) < 0)
1324 return(NULL);
1325 #endif
1327 #ifdef HAVE_BROKEN_READDIR
1328 /* using /usr/ucb/cc is BAD */
1329 dname = dname - 2;
1330 #endif
1333 static char *buf;
1334 int len = NAMLEN(ptr);
1335 buf = talloc_strndup(NULL, dname, len);
1336 dname = buf;
1339 return(dname);
1342 /****************************************************************************
1343 Recursive file matching function act as find
1344 match must be always set to true when calling this function
1345 ****************************************************************************/
1346 static int file_find(struct smbclient_context *ctx, struct file_list **list, const char *directory,
1347 const char *expression, bool match)
1349 DIR *dir;
1350 struct file_list *entry;
1351 struct stat statbuf;
1352 int ret;
1353 char *path;
1354 bool isdir;
1355 const char *dname;
1357 dir = opendir(directory);
1358 if (!dir) return -1;
1360 while ((dname = readdirname(dir))) {
1361 if (ISDOT(dname) || ISDOTDOT(dname)) {
1362 continue;
1365 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1366 continue;
1369 isdir = false;
1370 if (!match || !gen_fnmatch(expression, dname)) {
1371 if (ctx->recurse) {
1372 ret = stat(path, &statbuf);
1373 if (ret == 0) {
1374 if (S_ISDIR(statbuf.st_mode)) {
1375 isdir = true;
1376 ret = file_find(ctx, list, path, expression, false);
1378 } else {
1379 d_printf("file_find: cannot stat file %s\n", path);
1382 if (ret == -1) {
1383 SAFE_FREE(path);
1384 closedir(dir);
1385 return -1;
1388 entry = malloc_p(struct file_list);
1389 if (!entry) {
1390 d_printf("Out of memory in file_find\n");
1391 closedir(dir);
1392 return -1;
1394 entry->file_path = path;
1395 entry->isdir = isdir;
1396 DLIST_ADD(*list, entry);
1397 } else {
1398 SAFE_FREE(path);
1402 closedir(dir);
1403 return 0;
1406 /****************************************************************************
1407 mput some files
1408 ****************************************************************************/
1409 static int cmd_mput(struct smbclient_context *ctx, const char **args)
1411 int i;
1413 for (i = 1; args[i]; i++) {
1414 int ret;
1415 struct file_list *temp_list;
1416 char *quest, *lname, *rname;
1418 printf("%s\n", args[i]);
1420 file_list = NULL;
1422 ret = file_find(ctx, &file_list, ".", args[i], true);
1423 if (ret) {
1424 free_file_list(file_list);
1425 continue;
1428 quest = NULL;
1429 lname = NULL;
1430 rname = NULL;
1432 for (temp_list = file_list; temp_list;
1433 temp_list = temp_list->next) {
1435 SAFE_FREE(lname);
1436 if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1437 continue;
1438 trim_string(lname, "./", "/");
1440 /* check if it's a directory */
1441 if (temp_list->isdir) {
1442 /* if (!recurse) continue; */
1444 SAFE_FREE(quest);
1445 if (asprintf(&quest, "Put directory %s? ", lname) < 0) break;
1446 if (ctx->prompt && !yesno(quest)) { /* No */
1447 /* Skip the directory */
1448 lname[strlen(lname)-1] = '/';
1449 if (!seek_list(temp_list, lname))
1450 break;
1451 } else { /* Yes */
1452 SAFE_FREE(rname);
1453 if(asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1454 dos_format(rname);
1455 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, rname)) &&
1456 NT_STATUS_IS_ERR(do_mkdir(ctx, rname))) {
1457 DEBUG (0, ("Unable to make dir, skipping..."));
1458 /* Skip the directory */
1459 lname[strlen(lname)-1] = '/';
1460 if (!seek_list(temp_list, lname))
1461 break;
1464 continue;
1465 } else {
1466 SAFE_FREE(quest);
1467 if (asprintf(&quest,"Put file %s? ", lname) < 0) break;
1468 if (ctx->prompt && !yesno(quest)) /* No */
1469 continue;
1471 /* Yes */
1472 SAFE_FREE(rname);
1473 if (asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1476 dos_format(rname);
1478 do_put(ctx, rname, lname, false);
1480 free_file_list(file_list);
1481 SAFE_FREE(quest);
1482 SAFE_FREE(lname);
1483 SAFE_FREE(rname);
1486 return 0;
1490 /****************************************************************************
1491 print a file
1492 ****************************************************************************/
1493 static int cmd_print(struct smbclient_context *ctx, const char **args)
1495 char *lname, *rname;
1496 char *p;
1498 if (!args[1]) {
1499 d_printf("print <filename>\n");
1500 return 1;
1503 lname = talloc_strdup(ctx, args[1]);
1505 rname = talloc_strdup(ctx, lname);
1506 p = strrchr_m(rname,'/');
1507 if (p) {
1508 slprintf(rname, sizeof(rname)-1, "%s-%d", p+1, (int)getpid());
1511 if (strequal(lname,"-")) {
1512 slprintf(rname, sizeof(rname)-1, "stdin-%d", (int)getpid());
1515 return do_put(ctx, rname, lname, false);
1519 static int cmd_rewrite(struct smbclient_context *ctx, const char **args)
1521 d_printf("REWRITE: command not implemented (FIXME!)\n");
1523 return 0;
1526 /****************************************************************************
1527 delete some files
1528 ****************************************************************************/
1529 static int cmd_del(struct smbclient_context *ctx, const char **args)
1531 char *mask;
1532 uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
1534 if (ctx->recurse)
1535 attribute |= FILE_ATTRIBUTE_DIRECTORY;
1537 if (!args[1]) {
1538 d_printf("del <filename>\n");
1539 return 1;
1541 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1543 if (NT_STATUS_IS_ERR(smbcli_unlink(ctx->cli->tree, mask))) {
1544 d_printf("%s deleting remote file %s\n",smbcli_errstr(ctx->cli->tree),mask);
1547 return 0;
1551 /****************************************************************************
1552 delete a whole directory tree
1553 ****************************************************************************/
1554 static int cmd_deltree(struct smbclient_context *ctx, const char **args)
1556 char *dname;
1557 int ret;
1559 if (!args[1]) {
1560 d_printf("deltree <dirname>\n");
1561 return 1;
1564 dname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1566 ret = smbcli_deltree(ctx->cli->tree, dname);
1568 if (ret == -1) {
1569 printf("Failed to delete tree %s - %s\n", dname, smbcli_errstr(ctx->cli->tree));
1570 return -1;
1573 printf("Deleted %d files in %s\n", ret, dname);
1575 return 0;
1578 typedef struct {
1579 const char *level_name;
1580 enum smb_fsinfo_level level;
1581 } fsinfo_level_t;
1583 fsinfo_level_t fsinfo_levels[] = {
1584 {"dskattr", RAW_QFS_DSKATTR},
1585 {"allocation", RAW_QFS_ALLOCATION},
1586 {"volume", RAW_QFS_VOLUME},
1587 {"volumeinfo", RAW_QFS_VOLUME_INFO},
1588 {"sizeinfo", RAW_QFS_SIZE_INFO},
1589 {"deviceinfo", RAW_QFS_DEVICE_INFO},
1590 {"attributeinfo", RAW_QFS_ATTRIBUTE_INFO},
1591 {"unixinfo", RAW_QFS_UNIX_INFO},
1592 {"volume-information", RAW_QFS_VOLUME_INFORMATION},
1593 {"size-information", RAW_QFS_SIZE_INFORMATION},
1594 {"device-information", RAW_QFS_DEVICE_INFORMATION},
1595 {"attribute-information", RAW_QFS_ATTRIBUTE_INFORMATION},
1596 {"quota-information", RAW_QFS_QUOTA_INFORMATION},
1597 {"fullsize-information", RAW_QFS_FULL_SIZE_INFORMATION},
1598 {"objectid", RAW_QFS_OBJECTID_INFORMATION},
1599 {NULL, RAW_QFS_GENERIC}
1603 static int cmd_fsinfo(struct smbclient_context *ctx, const char **args)
1605 union smb_fsinfo fsinfo;
1606 NTSTATUS status;
1607 fsinfo_level_t *fsinfo_level;
1609 if (!args[1]) {
1610 d_printf("fsinfo <level>, where level is one of following:\n");
1611 fsinfo_level = fsinfo_levels;
1612 while(fsinfo_level->level_name) {
1613 d_printf("%s\n", fsinfo_level->level_name);
1614 fsinfo_level++;
1616 return 1;
1619 fsinfo_level = fsinfo_levels;
1620 while(fsinfo_level->level_name && !strequal(args[1],fsinfo_level->level_name)) {
1621 fsinfo_level++;
1624 if (!fsinfo_level->level_name) {
1625 d_printf("wrong level name!\n");
1626 return 1;
1629 fsinfo.generic.level = fsinfo_level->level;
1630 status = smb_raw_fsinfo(ctx->cli->tree, ctx, &fsinfo);
1631 if (!NT_STATUS_IS_OK(status)) {
1632 d_printf("fsinfo-level-%s - %s\n", fsinfo_level->level_name, nt_errstr(status));
1633 return 1;
1636 d_printf("fsinfo-level-%s:\n", fsinfo_level->level_name);
1637 switch(fsinfo.generic.level) {
1638 case RAW_QFS_DSKATTR:
1639 d_printf("\tunits_total: %hu\n",
1640 (unsigned short) fsinfo.dskattr.out.units_total);
1641 d_printf("\tblocks_per_unit: %hu\n",
1642 (unsigned short) fsinfo.dskattr.out.blocks_per_unit);
1643 d_printf("\tblocks_size: %hu\n",
1644 (unsigned short) fsinfo.dskattr.out.block_size);
1645 d_printf("\tunits_free: %hu\n",
1646 (unsigned short) fsinfo.dskattr.out.units_free);
1647 break;
1648 case RAW_QFS_ALLOCATION:
1649 d_printf("\tfs_id: %lu\n",
1650 (unsigned long) fsinfo.allocation.out.fs_id);
1651 d_printf("\tsectors_per_unit: %lu\n",
1652 (unsigned long) fsinfo.allocation.out.sectors_per_unit);
1653 d_printf("\ttotal_alloc_units: %lu\n",
1654 (unsigned long) fsinfo.allocation.out.total_alloc_units);
1655 d_printf("\tavail_alloc_units: %lu\n",
1656 (unsigned long) fsinfo.allocation.out.avail_alloc_units);
1657 d_printf("\tbytes_per_sector: %hu\n",
1658 (unsigned short) fsinfo.allocation.out.bytes_per_sector);
1659 break;
1660 case RAW_QFS_VOLUME:
1661 d_printf("\tserial_number: %lu\n",
1662 (unsigned long) fsinfo.volume.out.serial_number);
1663 d_printf("\tvolume_name: %s\n", fsinfo.volume.out.volume_name.s);
1664 break;
1665 case RAW_QFS_VOLUME_INFO:
1666 case RAW_QFS_VOLUME_INFORMATION:
1667 d_printf("\tcreate_time: %s\n",
1668 nt_time_string(ctx,fsinfo.volume_info.out.create_time));
1669 d_printf("\tserial_number: %lu\n",
1670 (unsigned long) fsinfo.volume_info.out.serial_number);
1671 d_printf("\tvolume_name: %s\n", fsinfo.volume_info.out.volume_name.s);
1672 break;
1673 case RAW_QFS_SIZE_INFO:
1674 case RAW_QFS_SIZE_INFORMATION:
1675 d_printf("\ttotal_alloc_units: %llu\n",
1676 (unsigned long long) fsinfo.size_info.out.total_alloc_units);
1677 d_printf("\tavail_alloc_units: %llu\n",
1678 (unsigned long long) fsinfo.size_info.out.avail_alloc_units);
1679 d_printf("\tsectors_per_unit: %lu\n",
1680 (unsigned long) fsinfo.size_info.out.sectors_per_unit);
1681 d_printf("\tbytes_per_sector: %lu\n",
1682 (unsigned long) fsinfo.size_info.out.bytes_per_sector);
1683 break;
1684 case RAW_QFS_DEVICE_INFO:
1685 case RAW_QFS_DEVICE_INFORMATION:
1686 d_printf("\tdevice_type: %lu\n",
1687 (unsigned long) fsinfo.device_info.out.device_type);
1688 d_printf("\tcharacteristics: 0x%lx\n",
1689 (unsigned long) fsinfo.device_info.out.characteristics);
1690 break;
1691 case RAW_QFS_ATTRIBUTE_INFORMATION:
1692 case RAW_QFS_ATTRIBUTE_INFO:
1693 d_printf("\tfs_attr: 0x%lx\n",
1694 (unsigned long) fsinfo.attribute_info.out.fs_attr);
1695 d_printf("\tmax_file_component_length: %lu\n",
1696 (unsigned long) fsinfo.attribute_info.out.max_file_component_length);
1697 d_printf("\tfs_type: %s\n", fsinfo.attribute_info.out.fs_type.s);
1698 break;
1699 case RAW_QFS_UNIX_INFO:
1700 d_printf("\tmajor_version: %hu\n",
1701 (unsigned short) fsinfo.unix_info.out.major_version);
1702 d_printf("\tminor_version: %hu\n",
1703 (unsigned short) fsinfo.unix_info.out.minor_version);
1704 d_printf("\tcapability: 0x%llx\n",
1705 (unsigned long long) fsinfo.unix_info.out.capability);
1706 break;
1707 case RAW_QFS_QUOTA_INFORMATION:
1708 d_printf("\tunknown[3]: [%llu,%llu,%llu]\n",
1709 (unsigned long long) fsinfo.quota_information.out.unknown[0],
1710 (unsigned long long) fsinfo.quota_information.out.unknown[1],
1711 (unsigned long long) fsinfo.quota_information.out.unknown[2]);
1712 d_printf("\tquota_soft: %llu\n",
1713 (unsigned long long) fsinfo.quota_information.out.quota_soft);
1714 d_printf("\tquota_hard: %llu\n",
1715 (unsigned long long) fsinfo.quota_information.out.quota_hard);
1716 d_printf("\tquota_flags: 0x%llx\n",
1717 (unsigned long long) fsinfo.quota_information.out.quota_flags);
1718 break;
1719 case RAW_QFS_FULL_SIZE_INFORMATION:
1720 d_printf("\ttotal_alloc_units: %llu\n",
1721 (unsigned long long) fsinfo.full_size_information.out.total_alloc_units);
1722 d_printf("\tcall_avail_alloc_units: %llu\n",
1723 (unsigned long long) fsinfo.full_size_information.out.call_avail_alloc_units);
1724 d_printf("\tactual_avail_alloc_units: %llu\n",
1725 (unsigned long long) fsinfo.full_size_information.out.actual_avail_alloc_units);
1726 d_printf("\tsectors_per_unit: %lu\n",
1727 (unsigned long) fsinfo.full_size_information.out.sectors_per_unit);
1728 d_printf("\tbytes_per_sector: %lu\n",
1729 (unsigned long) fsinfo.full_size_information.out.bytes_per_sector);
1730 break;
1731 case RAW_QFS_OBJECTID_INFORMATION:
1732 d_printf("\tGUID: %s\n",
1733 GUID_string(ctx,&fsinfo.objectid_information.out.guid));
1734 d_printf("\tunknown[6]: [%llu,%llu,%llu,%llu,%llu,%llu]\n",
1735 (unsigned long long) fsinfo.objectid_information.out.unknown[0],
1736 (unsigned long long) fsinfo.objectid_information.out.unknown[1],
1737 (unsigned long long) fsinfo.objectid_information.out.unknown[2],
1738 (unsigned long long) fsinfo.objectid_information.out.unknown[3],
1739 (unsigned long long) fsinfo.objectid_information.out.unknown[4],
1740 (unsigned long long) fsinfo.objectid_information.out.unknown[5] );
1741 break;
1742 case RAW_QFS_GENERIC:
1743 d_printf("\twrong level returned\n");
1744 break;
1747 return 0;
1750 /****************************************************************************
1751 show as much information as possible about a file
1752 ****************************************************************************/
1753 static int cmd_allinfo(struct smbclient_context *ctx, const char **args)
1755 char *fname;
1756 union smb_fileinfo finfo;
1757 NTSTATUS status;
1758 int fnum;
1760 if (!args[1]) {
1761 d_printf("allinfo <filename>\n");
1762 return 1;
1764 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1766 /* first a ALL_INFO QPATHINFO */
1767 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1768 finfo.generic.in.file.path = fname;
1769 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1770 if (!NT_STATUS_IS_OK(status)) {
1771 d_printf("%s - %s\n", fname, nt_errstr(status));
1772 return 1;
1775 d_printf("\tcreate_time: %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1776 d_printf("\taccess_time: %s\n", nt_time_string(ctx, finfo.all_info.out.access_time));
1777 d_printf("\twrite_time: %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1778 d_printf("\tchange_time: %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1779 d_printf("\tattrib: 0x%x\n", finfo.all_info.out.attrib);
1780 d_printf("\talloc_size: %lu\n", (unsigned long)finfo.all_info.out.alloc_size);
1781 d_printf("\tsize: %lu\n", (unsigned long)finfo.all_info.out.size);
1782 d_printf("\tnlink: %u\n", finfo.all_info.out.nlink);
1783 d_printf("\tdelete_pending: %u\n", finfo.all_info.out.delete_pending);
1784 d_printf("\tdirectory: %u\n", finfo.all_info.out.directory);
1785 d_printf("\tea_size: %u\n", finfo.all_info.out.ea_size);
1786 d_printf("\tfname: '%s'\n", finfo.all_info.out.fname.s);
1788 /* 8.3 name if any */
1789 finfo.generic.level = RAW_FILEINFO_ALT_NAME_INFO;
1790 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1791 if (NT_STATUS_IS_OK(status)) {
1792 d_printf("\talt_name: %s\n", finfo.alt_name_info.out.fname.s);
1795 /* file_id if available */
1796 finfo.generic.level = RAW_FILEINFO_INTERNAL_INFORMATION;
1797 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1798 if (NT_STATUS_IS_OK(status)) {
1799 d_printf("\tfile_id %.0f\n",
1800 (double)finfo.internal_information.out.file_id);
1803 /* the EAs, if any */
1804 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1805 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1806 if (NT_STATUS_IS_OK(status)) {
1807 int i;
1808 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1809 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1810 finfo.all_eas.out.eas[i].flags,
1811 (int)finfo.all_eas.out.eas[i].value.length,
1812 finfo.all_eas.out.eas[i].name.s);
1816 /* streams, if available */
1817 finfo.generic.level = RAW_FILEINFO_STREAM_INFO;
1818 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1819 if (NT_STATUS_IS_OK(status)) {
1820 int i;
1821 for (i=0;i<finfo.stream_info.out.num_streams;i++) {
1822 d_printf("\tstream %d:\n", i);
1823 d_printf("\t\tsize %ld\n",
1824 (long)finfo.stream_info.out.streams[i].size);
1825 d_printf("\t\talloc size %ld\n",
1826 (long)finfo.stream_info.out.streams[i].alloc_size);
1827 d_printf("\t\tname %s\n", finfo.stream_info.out.streams[i].stream_name.s);
1831 /* dev/inode if available */
1832 finfo.generic.level = RAW_FILEINFO_COMPRESSION_INFORMATION;
1833 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1834 if (NT_STATUS_IS_OK(status)) {
1835 d_printf("\tcompressed size %ld\n", (long)finfo.compression_info.out.compressed_size);
1836 d_printf("\tformat %ld\n", (long)finfo.compression_info.out.format);
1837 d_printf("\tunit_shift %ld\n", (long)finfo.compression_info.out.unit_shift);
1838 d_printf("\tchunk_shift %ld\n", (long)finfo.compression_info.out.chunk_shift);
1839 d_printf("\tcluster_shift %ld\n", (long)finfo.compression_info.out.cluster_shift);
1842 /* shadow copies if available */
1843 fnum = smbcli_open(ctx->cli->tree, fname, O_RDONLY, DENY_NONE);
1844 if (fnum != -1) {
1845 struct smb_shadow_copy info;
1846 int i;
1847 info.in.file.fnum = fnum;
1848 info.in.max_data = ~0;
1849 status = smb_raw_shadow_data(ctx->cli->tree, ctx, &info);
1850 if (NT_STATUS_IS_OK(status)) {
1851 d_printf("\tshadow_copy: %u volumes %u names\n",
1852 info.out.num_volumes, info.out.num_names);
1853 for (i=0;i<info.out.num_names;i++) {
1854 d_printf("\t%s\n", info.out.names[i]);
1855 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1856 finfo.generic.in.file.path = talloc_asprintf(ctx, "%s%s",
1857 info.out.names[i], fname);
1858 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1859 if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_PATH_NOT_FOUND) ||
1860 NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
1861 continue;
1863 if (!NT_STATUS_IS_OK(status)) {
1864 d_printf("%s - %s\n", finfo.generic.in.file.path,
1865 nt_errstr(status));
1866 return 1;
1869 d_printf("\t\tcreate_time: %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1870 d_printf("\t\twrite_time: %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1871 d_printf("\t\tchange_time: %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1872 d_printf("\t\tsize: %lu\n", (unsigned long)finfo.all_info.out.size);
1877 return 0;
1881 /****************************************************************************
1882 shows EA contents
1883 ****************************************************************************/
1884 static int cmd_eainfo(struct smbclient_context *ctx, const char **args)
1886 char *fname;
1887 union smb_fileinfo finfo;
1888 NTSTATUS status;
1889 int i;
1891 if (!args[1]) {
1892 d_printf("eainfo <filename>\n");
1893 return 1;
1895 fname = talloc_strdup(ctx, args[1]);
1897 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1898 finfo.generic.in.file.path = fname;
1899 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1901 if (!NT_STATUS_IS_OK(status)) {
1902 d_printf("RAW_FILEINFO_ALL_EAS - %s\n", nt_errstr(status));
1903 return 1;
1906 d_printf("%s has %d EAs\n", fname, finfo.all_eas.out.num_eas);
1908 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1909 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1910 finfo.all_eas.out.eas[i].flags,
1911 (int)finfo.all_eas.out.eas[i].value.length,
1912 finfo.all_eas.out.eas[i].name.s);
1913 fflush(stdout);
1914 dump_data(0,
1915 finfo.all_eas.out.eas[i].value.data,
1916 finfo.all_eas.out.eas[i].value.length);
1919 return 0;
1923 /****************************************************************************
1924 show any ACL on a file
1925 ****************************************************************************/
1926 static int cmd_acl(struct smbclient_context *ctx, const char **args)
1928 char *fname;
1929 union smb_fileinfo query;
1930 NTSTATUS status;
1931 int fnum;
1933 if (!args[1]) {
1934 d_printf("acl <filename>\n");
1935 return 1;
1937 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1939 fnum = smbcli_nt_create_full(ctx->cli->tree, fname, 0,
1940 SEC_STD_READ_CONTROL,
1942 NTCREATEX_SHARE_ACCESS_DELETE|
1943 NTCREATEX_SHARE_ACCESS_READ|
1944 NTCREATEX_SHARE_ACCESS_WRITE,
1945 NTCREATEX_DISP_OPEN,
1946 0, 0);
1947 if (fnum == -1) {
1948 d_printf("%s - %s\n", fname, smbcli_errstr(ctx->cli->tree));
1949 return -1;
1952 query.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
1953 query.query_secdesc.in.file.fnum = fnum;
1954 query.query_secdesc.in.secinfo_flags = 0x7;
1956 status = smb_raw_fileinfo(ctx->cli->tree, ctx, &query);
1957 if (!NT_STATUS_IS_OK(status)) {
1958 d_printf("%s - %s\n", fname, nt_errstr(status));
1959 return 1;
1962 NDR_PRINT_DEBUG(security_descriptor, query.query_secdesc.out.sd);
1964 return 0;
1967 /****************************************************************************
1968 lookup a name or sid
1969 ****************************************************************************/
1970 static int cmd_lookup(struct smbclient_context *ctx, const char **args)
1972 NTSTATUS status;
1973 struct dom_sid *sid;
1975 if (!args[1]) {
1976 d_printf("lookup <sid|name>\n");
1977 return 1;
1980 sid = dom_sid_parse_talloc(ctx, args[1]);
1981 if (sid == NULL) {
1982 const char *sidstr;
1983 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sidstr);
1984 if (!NT_STATUS_IS_OK(status)) {
1985 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
1986 return 1;
1989 d_printf("%s\n", sidstr);
1990 } else {
1991 const char *name;
1992 status = smblsa_lookup_sid(ctx->cli, args[1], ctx, &name);
1993 if (!NT_STATUS_IS_OK(status)) {
1994 d_printf("lsa_LookupSids - %s\n", nt_errstr(status));
1995 return 1;
1998 d_printf("%s\n", name);
2001 return 0;
2004 /****************************************************************************
2005 show privileges for a user
2006 ****************************************************************************/
2007 static int cmd_privileges(struct smbclient_context *ctx, const char **args)
2009 NTSTATUS status;
2010 struct dom_sid *sid;
2011 struct lsa_RightSet rights;
2012 unsigned i;
2014 if (!args[1]) {
2015 d_printf("privileges <sid|name>\n");
2016 return 1;
2019 sid = dom_sid_parse_talloc(ctx, args[1]);
2020 if (sid == NULL) {
2021 const char *sid_str;
2022 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2023 if (!NT_STATUS_IS_OK(status)) {
2024 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2025 return 1;
2027 sid = dom_sid_parse_talloc(ctx, sid_str);
2030 status = smblsa_sid_privileges(ctx->cli, sid, ctx, &rights);
2031 if (!NT_STATUS_IS_OK(status)) {
2032 d_printf("lsa_EnumAccountRights - %s\n", nt_errstr(status));
2033 return 1;
2036 for (i=0;i<rights.count;i++) {
2037 d_printf("\t%s\n", rights.names[i].string);
2040 return 0;
2044 /****************************************************************************
2045 add privileges for a user
2046 ****************************************************************************/
2047 static int cmd_addprivileges(struct smbclient_context *ctx, const char **args)
2049 NTSTATUS status;
2050 struct dom_sid *sid;
2051 struct lsa_RightSet rights;
2052 int i;
2054 if (!args[1]) {
2055 d_printf("addprivileges <sid|name> <privilege...>\n");
2056 return 1;
2059 sid = dom_sid_parse_talloc(ctx, args[1]);
2060 if (sid == NULL) {
2061 const char *sid_str;
2062 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2063 if (!NT_STATUS_IS_OK(status)) {
2064 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2065 return 1;
2067 sid = dom_sid_parse_talloc(ctx, sid_str);
2070 ZERO_STRUCT(rights);
2071 for (i = 2; args[i]; i++) {
2072 rights.names = talloc_realloc(ctx, rights.names,
2073 struct lsa_StringLarge, rights.count+1);
2074 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2075 rights.count++;
2079 status = smblsa_sid_add_privileges(ctx->cli, sid, ctx, &rights);
2080 if (!NT_STATUS_IS_OK(status)) {
2081 d_printf("lsa_AddAccountRights - %s\n", nt_errstr(status));
2082 return 1;
2085 return 0;
2088 /****************************************************************************
2089 delete privileges for a user
2090 ****************************************************************************/
2091 static int cmd_delprivileges(struct smbclient_context *ctx, const char **args)
2093 NTSTATUS status;
2094 struct dom_sid *sid;
2095 struct lsa_RightSet rights;
2096 int i;
2098 if (!args[1]) {
2099 d_printf("delprivileges <sid|name> <privilege...>\n");
2100 return 1;
2103 sid = dom_sid_parse_talloc(ctx, args[1]);
2104 if (sid == NULL) {
2105 const char *sid_str;
2106 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2107 if (!NT_STATUS_IS_OK(status)) {
2108 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2109 return 1;
2111 sid = dom_sid_parse_talloc(ctx, sid_str);
2114 ZERO_STRUCT(rights);
2115 for (i = 2; args[i]; i++) {
2116 rights.names = talloc_realloc(ctx, rights.names,
2117 struct lsa_StringLarge, rights.count+1);
2118 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2119 rights.count++;
2123 status = smblsa_sid_del_privileges(ctx->cli, sid, ctx, &rights);
2124 if (!NT_STATUS_IS_OK(status)) {
2125 d_printf("lsa_RemoveAccountRights - %s\n", nt_errstr(status));
2126 return 1;
2129 return 0;
2133 /****************************************************************************
2134 open a file
2135 ****************************************************************************/
2136 static int cmd_open(struct smbclient_context *ctx, const char **args)
2138 char *filename;
2139 union smb_open io;
2140 NTSTATUS status;
2141 TALLOC_CTX *tmp_ctx;
2143 if (!args[1]) {
2144 d_printf("open <filename>\n");
2145 return 1;
2147 tmp_ctx = talloc_new(ctx);
2149 filename = talloc_asprintf(tmp_ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2151 io.generic.level = RAW_OPEN_NTCREATEX;
2152 io.ntcreatex.in.root_fid.fnum = 0;
2153 io.ntcreatex.in.flags = 0;
2154 io.ntcreatex.in.access_mask = SEC_RIGHTS_FILE_ALL;
2155 io.ntcreatex.in.create_options = 0;
2156 io.ntcreatex.in.file_attr = FILE_ATTRIBUTE_NORMAL;
2157 io.ntcreatex.in.share_access = NTCREATEX_SHARE_ACCESS_READ;
2158 io.ntcreatex.in.alloc_size = 0;
2159 io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN_IF;
2160 io.ntcreatex.in.impersonation = NTCREATEX_IMPERSONATION_ANONYMOUS;
2161 io.ntcreatex.in.security_flags = 0;
2162 io.ntcreatex.in.fname = filename;
2164 status = smb_raw_open(ctx->cli->tree, tmp_ctx, &io);
2165 talloc_free(tmp_ctx);
2167 if (NT_STATUS_IS_OK(status)) {
2168 d_printf("Opened file with fnum %u\n", (unsigned)io.ntcreatex.out.file.fnum);
2169 } else {
2170 d_printf("Opened failed: %s\n", nt_errstr(status));
2173 return 0;
2176 /****************************************************************************
2177 close a file
2178 ****************************************************************************/
2179 static int cmd_close(struct smbclient_context *ctx, const char **args)
2181 union smb_close io;
2182 NTSTATUS status;
2183 uint16_t fnum;
2185 if (!args[1]) {
2186 d_printf("close <fnum>\n");
2187 return 1;
2190 fnum = atoi(args[1]);
2192 ZERO_STRUCT(io);
2193 io.generic.level = RAW_CLOSE_CLOSE;
2194 io.close.in.file.fnum = fnum;
2196 status = smb_raw_close(ctx->cli->tree, &io);
2198 if (NT_STATUS_IS_OK(status)) {
2199 d_printf("Closed file OK\n");
2200 } else {
2201 d_printf("Close failed: %s\n", nt_errstr(status));
2204 return 0;
2208 /****************************************************************************
2209 remove a directory
2210 ****************************************************************************/
2211 static int cmd_rmdir(struct smbclient_context *ctx, const char **args)
2213 char *mask;
2215 if (!args[1]) {
2216 d_printf("rmdir <dirname>\n");
2217 return 1;
2219 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2221 if (NT_STATUS_IS_ERR(smbcli_rmdir(ctx->cli->tree, mask))) {
2222 d_printf("%s removing remote directory file %s\n",
2223 smbcli_errstr(ctx->cli->tree),mask);
2226 return 0;
2229 /****************************************************************************
2230 UNIX hardlink.
2231 ****************************************************************************/
2232 static int cmd_link(struct smbclient_context *ctx, const char **args)
2234 char *src,*dest;
2236 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2237 d_printf("Server doesn't support UNIX CIFS calls.\n");
2238 return 1;
2242 if (!args[1] || !args[2]) {
2243 d_printf("link <src> <dest>\n");
2244 return 1;
2247 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2248 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2250 if (NT_STATUS_IS_ERR(smbcli_unix_hardlink(ctx->cli->tree, src, dest))) {
2251 d_printf("%s linking files (%s -> %s)\n", smbcli_errstr(ctx->cli->tree), src, dest);
2252 return 1;
2255 return 0;
2258 /****************************************************************************
2259 UNIX symlink.
2260 ****************************************************************************/
2262 static int cmd_symlink(struct smbclient_context *ctx, const char **args)
2264 char *src,*dest;
2266 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2267 d_printf("Server doesn't support UNIX CIFS calls.\n");
2268 return 1;
2271 if (!args[1] || !args[2]) {
2272 d_printf("symlink <src> <dest>\n");
2273 return 1;
2276 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2277 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2279 if (NT_STATUS_IS_ERR(smbcli_unix_symlink(ctx->cli->tree, src, dest))) {
2280 d_printf("%s symlinking files (%s -> %s)\n",
2281 smbcli_errstr(ctx->cli->tree), src, dest);
2282 return 1;
2285 return 0;
2288 /****************************************************************************
2289 UNIX chmod.
2290 ****************************************************************************/
2292 static int cmd_chmod(struct smbclient_context *ctx, const char **args)
2294 char *src;
2295 mode_t mode;
2297 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2298 d_printf("Server doesn't support UNIX CIFS calls.\n");
2299 return 1;
2302 if (!args[1] || !args[2]) {
2303 d_printf("chmod mode file\n");
2304 return 1;
2307 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2309 mode = (mode_t)strtol(args[1], NULL, 8);
2311 if (NT_STATUS_IS_ERR(smbcli_unix_chmod(ctx->cli->tree, src, mode))) {
2312 d_printf("%s chmod file %s 0%o\n",
2313 smbcli_errstr(ctx->cli->tree), src, (unsigned)mode);
2314 return 1;
2317 return 0;
2320 /****************************************************************************
2321 UNIX chown.
2322 ****************************************************************************/
2324 static int cmd_chown(struct smbclient_context *ctx, const char **args)
2326 char *src;
2327 uid_t uid;
2328 gid_t gid;
2330 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2331 d_printf("Server doesn't support UNIX CIFS calls.\n");
2332 return 1;
2335 if (!args[1] || !args[2] || !args[3]) {
2336 d_printf("chown uid gid file\n");
2337 return 1;
2340 uid = (uid_t)atoi(args[1]);
2341 gid = (gid_t)atoi(args[2]);
2342 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[3]);
2344 if (NT_STATUS_IS_ERR(smbcli_unix_chown(ctx->cli->tree, src, uid, gid))) {
2345 d_printf("%s chown file %s uid=%d, gid=%d\n",
2346 smbcli_errstr(ctx->cli->tree), src, (int)uid, (int)gid);
2347 return 1;
2350 return 0;
2353 /****************************************************************************
2354 rename some files
2355 ****************************************************************************/
2356 static int cmd_rename(struct smbclient_context *ctx, const char **args)
2358 char *src,*dest;
2360 if (!args[1] || !args[2]) {
2361 d_printf("rename <src> <dest>\n");
2362 return 1;
2365 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2366 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2368 if (NT_STATUS_IS_ERR(smbcli_rename(ctx->cli->tree, src, dest))) {
2369 d_printf("%s renaming files\n",smbcli_errstr(ctx->cli->tree));
2370 return 1;
2373 return 0;
2377 /****************************************************************************
2378 toggle the prompt flag
2379 ****************************************************************************/
2380 static int cmd_prompt(struct smbclient_context *ctx, const char **args)
2382 ctx->prompt = !ctx->prompt;
2383 DEBUG(2,("prompting is now %s\n",ctx->prompt?"on":"off"));
2385 return 1;
2389 /****************************************************************************
2390 set the newer than time
2391 ****************************************************************************/
2392 static int cmd_newer(struct smbclient_context *ctx, const char **args)
2394 struct stat sbuf;
2396 if (args[1] && (stat(args[1],&sbuf) == 0)) {
2397 ctx->newer_than = sbuf.st_mtime;
2398 DEBUG(1,("Getting files newer than %s",
2399 asctime(localtime(&ctx->newer_than))));
2400 } else {
2401 ctx->newer_than = 0;
2404 if (args[1] && ctx->newer_than == 0) {
2405 d_printf("Error setting newer-than time\n");
2406 return 1;
2409 return 0;
2412 /****************************************************************************
2413 set the archive level
2414 ****************************************************************************/
2415 static int cmd_archive(struct smbclient_context *ctx, const char **args)
2417 if (args[1]) {
2418 ctx->archive_level = atoi(args[1]);
2419 } else
2420 d_printf("Archive level is %d\n",ctx->archive_level);
2422 return 0;
2425 /****************************************************************************
2426 toggle the lowercaseflag
2427 ****************************************************************************/
2428 static int cmd_lowercase(struct smbclient_context *ctx, const char **args)
2430 ctx->lowercase = !ctx->lowercase;
2431 DEBUG(2,("filename lowercasing is now %s\n",ctx->lowercase?"on":"off"));
2433 return 0;
2439 /****************************************************************************
2440 toggle the recurse flag
2441 ****************************************************************************/
2442 static int cmd_recurse(struct smbclient_context *ctx, const char **args)
2444 ctx->recurse = !ctx->recurse;
2445 DEBUG(2,("directory recursion is now %s\n",ctx->recurse?"on":"off"));
2447 return 0;
2450 /****************************************************************************
2451 toggle the translate flag
2452 ****************************************************************************/
2453 static int cmd_translate(struct smbclient_context *ctx, const char **args)
2455 ctx->translation = !ctx->translation;
2456 DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
2457 ctx->translation?"on":"off"));
2459 return 0;
2463 /****************************************************************************
2464 do a printmode command
2465 ****************************************************************************/
2466 static int cmd_printmode(struct smbclient_context *ctx, const char **args)
2468 if (args[1]) {
2469 if (strequal(args[1],"text")) {
2470 ctx->printmode = 0;
2471 } else {
2472 if (strequal(args[1],"graphics"))
2473 ctx->printmode = 1;
2474 else
2475 ctx->printmode = atoi(args[1]);
2479 switch(ctx->printmode)
2481 case 0:
2482 DEBUG(2,("the printmode is now text\n"));
2483 break;
2484 case 1:
2485 DEBUG(2,("the printmode is now graphics\n"));
2486 break;
2487 default:
2488 DEBUG(2,("the printmode is now %d\n", ctx->printmode));
2489 break;
2492 return 0;
2495 /****************************************************************************
2496 do the lcd command
2497 ****************************************************************************/
2498 static int cmd_lcd(struct smbclient_context *ctx, const char **args)
2500 char d[PATH_MAX];
2502 if (args[1])
2503 chdir(args[1]);
2504 DEBUG(2,("the local directory is now %s\n",getcwd(d, PATH_MAX)));
2506 return 0;
2509 /****************************************************************************
2510 history
2511 ****************************************************************************/
2512 static int cmd_history(struct smbclient_context *ctx, const char **args)
2514 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
2515 HIST_ENTRY **hlist;
2516 int i;
2518 hlist = history_list();
2520 for (i = 0; hlist && hlist[i]; i++) {
2521 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
2523 #else
2524 DEBUG(0,("no history without readline support\n"));
2525 #endif
2527 return 0;
2530 /****************************************************************************
2531 get a file restarting at end of local file
2532 ****************************************************************************/
2533 static int cmd_reget(struct smbclient_context *ctx, const char **args)
2535 char *local_name;
2536 char *remote_name;
2538 if (!args[1]) {
2539 d_printf("reget <filename>\n");
2540 return 1;
2542 remote_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2543 dos_clean_name(remote_name);
2545 if (args[2])
2546 local_name = talloc_strdup(ctx, args[2]);
2547 else
2548 local_name = talloc_strdup(ctx, args[1]);
2550 return do_get(ctx, remote_name, local_name, true);
2553 /****************************************************************************
2554 put a file restarting at end of local file
2555 ****************************************************************************/
2556 static int cmd_reput(struct smbclient_context *ctx, const char **args)
2558 char *local_name;
2559 char *remote_name;
2561 if (!args[1]) {
2562 d_printf("reput <filename>\n");
2563 return 1;
2565 local_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2567 if (!file_exist(local_name)) {
2568 d_printf("%s does not exist\n", local_name);
2569 return 1;
2572 if (args[2])
2573 remote_name = talloc_strdup(ctx, args[2]);
2574 else
2575 remote_name = talloc_strdup(ctx, args[1]);
2577 dos_clean_name(remote_name);
2579 return do_put(ctx, remote_name, local_name, true);
2584 return a string representing a share type
2586 static const char *share_type_str(uint32_t type)
2588 switch (type & 0xF) {
2589 case STYPE_DISKTREE:
2590 return "Disk";
2591 case STYPE_PRINTQ:
2592 return "Printer";
2593 case STYPE_DEVICE:
2594 return "Device";
2595 case STYPE_IPC:
2596 return "IPC";
2597 default:
2598 return "Unknown";
2604 display a list of shares from a level 1 share enum
2606 static void display_share_result(struct srvsvc_NetShareCtr1 *ctr1)
2608 int i;
2610 for (i=0;i<ctr1->count;i++) {
2611 struct srvsvc_NetShareInfo1 *info = ctr1->array+i;
2613 printf("\t%-15s %-10.10s %s\n",
2614 info->name,
2615 share_type_str(info->type),
2616 info->comment);
2622 /****************************************************************************
2623 try and browse available shares on a host
2624 ****************************************************************************/
2625 static bool browse_host(struct loadparm_context *lp_ctx,
2626 struct tevent_context *ev_ctx,
2627 const char *query_host)
2629 struct dcerpc_pipe *p;
2630 char *binding;
2631 NTSTATUS status;
2632 struct srvsvc_NetShareEnumAll r;
2633 struct srvsvc_NetShareInfoCtr info_ctr;
2634 uint32_t resume_handle = 0;
2635 TALLOC_CTX *mem_ctx = talloc_init("browse_host");
2636 struct srvsvc_NetShareCtr1 ctr1;
2637 uint32_t totalentries = 0;
2639 binding = talloc_asprintf(mem_ctx, "ncacn_np:%s", query_host);
2641 status = dcerpc_pipe_connect(mem_ctx, &p, binding,
2642 &ndr_table_srvsvc,
2643 cmdline_credentials, ev_ctx,
2644 lp_ctx);
2645 if (!NT_STATUS_IS_OK(status)) {
2646 d_printf("Failed to connect to %s - %s\n",
2647 binding, nt_errstr(status));
2648 talloc_free(mem_ctx);
2649 return false;
2652 info_ctr.level = 1;
2653 info_ctr.ctr.ctr1 = &ctr1;
2655 r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
2656 r.in.info_ctr = &info_ctr;
2657 r.in.max_buffer = ~0;
2658 r.in.resume_handle = &resume_handle;
2659 r.out.resume_handle = &resume_handle;
2660 r.out.totalentries = &totalentries;
2661 r.out.info_ctr = &info_ctr;
2663 d_printf("\n\tSharename Type Comment\n");
2664 d_printf("\t--------- ---- -------\n");
2666 do {
2667 ZERO_STRUCT(ctr1);
2668 status = dcerpc_srvsvc_NetShareEnumAll_r(p->binding_handle, mem_ctx, &r);
2670 if (NT_STATUS_IS_OK(status) &&
2671 (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA) ||
2672 W_ERROR_IS_OK(r.out.result)) &&
2673 r.out.info_ctr->ctr.ctr1) {
2674 display_share_result(r.out.info_ctr->ctr.ctr1);
2675 resume_handle += r.out.info_ctr->ctr.ctr1->count;
2677 } while (NT_STATUS_IS_OK(status) && W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA));
2679 talloc_free(mem_ctx);
2681 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2682 d_printf("Failed NetShareEnumAll %s - %s/%s\n",
2683 binding, nt_errstr(status), win_errstr(r.out.result));
2684 return false;
2687 return false;
2690 /****************************************************************************
2691 try and browse available connections on a host
2692 ****************************************************************************/
2693 static bool list_servers(const char *wk_grp)
2695 d_printf("REWRITE: list servers not implemented\n");
2696 return false;
2699 /* Some constants for completing filename arguments */
2701 #define COMPL_NONE 0 /* No completions */
2702 #define COMPL_REMOTE 1 /* Complete remote filename */
2703 #define COMPL_LOCAL 2 /* Complete local filename */
2705 static int cmd_help(struct smbclient_context *ctx, const char **args);
2707 /* This defines the commands supported by this client.
2708 * NOTE: The "!" must be the last one in the list because it's fn pointer
2709 * field is NULL, and NULL in that field is used in process_tok()
2710 * (below) to indicate the end of the list. crh
2712 static struct
2714 const char *name;
2715 int (*fn)(struct smbclient_context *ctx, const char **args);
2716 const char *description;
2717 char compl_args[2]; /* Completion argument info */
2718 } commands[] =
2720 {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2721 {"addprivileges",cmd_addprivileges,"<sid|name> <privilege...> add privileges for a user",{COMPL_NONE,COMPL_NONE}},
2722 {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
2723 {"acl",cmd_acl,"<file> show file ACL",{COMPL_NONE,COMPL_NONE}},
2724 {"allinfo",cmd_allinfo,"<file> show all possible info about a file",{COMPL_NONE,COMPL_NONE}},
2725 {"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}},
2726 {"cancel",cmd_rewrite,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
2727 {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
2728 {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
2729 {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
2730 {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2731 {"delprivileges",cmd_delprivileges,"<sid|name> <privilege...> remove privileges for a user",{COMPL_NONE,COMPL_NONE}},
2732 {"deltree",cmd_deltree,"<dir> delete a whole directory tree",{COMPL_REMOTE,COMPL_NONE}},
2733 {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2734 {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2735 {"eainfo",cmd_eainfo,"<file> show EA contents for a file",{COMPL_NONE,COMPL_NONE}},
2736 {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2737 {"fsinfo",cmd_fsinfo,"query file system info",{COMPL_NONE,COMPL_NONE}},
2738 {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
2739 {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2740 {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
2741 {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
2742 {"link",cmd_link,"<src> <dest> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2743 {"lookup",cmd_lookup,"<sid|name> show SID for name or name for SID",{COMPL_NONE,COMPL_NONE}},
2744 {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},
2745 {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2746 {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
2747 {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2748 {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
2749 {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2750 {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},
2751 {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
2752 {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
2753 {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
2754 {"close",cmd_close,"<fnum> close a file",{COMPL_NONE,COMPL_NONE}},
2755 {"privileges",cmd_privileges,"<user> show privileges for a user",{COMPL_NONE,COMPL_NONE}},
2756 {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
2757 {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
2758 {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},
2759 {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
2760 {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
2761 {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2762 {"queue",cmd_rewrite,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2763 {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2764 {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2765 {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},
2766 {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
2767 {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2768 {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
2769 {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2770 {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2771 {"symlink",cmd_symlink,"<src> <dest> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2772 {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2774 /* Yes, this must be here, see crh's comment above. */
2775 {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2776 {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2780 /*******************************************************************
2781 lookup a command string in the list of commands, including
2782 abbreviations
2783 ******************************************************************/
2784 static int process_tok(const char *tok)
2786 int i = 0, matches = 0;
2787 int cmd=0;
2788 int tok_len = strlen(tok);
2790 while (commands[i].fn != NULL) {
2791 if (strequal(commands[i].name,tok)) {
2792 matches = 1;
2793 cmd = i;
2794 break;
2795 } else if (strncasecmp(commands[i].name, tok, tok_len) == 0) {
2796 matches++;
2797 cmd = i;
2799 i++;
2802 if (matches == 0)
2803 return(-1);
2804 else if (matches == 1)
2805 return(cmd);
2806 else
2807 return(-2);
2810 /****************************************************************************
2811 help
2812 ****************************************************************************/
2813 static int cmd_help(struct smbclient_context *ctx, const char **args)
2815 int i=0,j;
2817 if (args[1]) {
2818 if ((i = process_tok(args[1])) >= 0)
2819 d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2820 } else {
2821 while (commands[i].description) {
2822 for (j=0; commands[i].description && (j<5); j++) {
2823 d_printf("%-15s",commands[i].name);
2824 i++;
2826 d_printf("\n");
2829 return 0;
2832 static int process_line(struct smbclient_context *ctx, const char *cline);
2833 /****************************************************************************
2834 process a -c command string
2835 ****************************************************************************/
2836 static int process_command_string(struct smbclient_context *ctx, const char *cmd)
2838 char **lines;
2839 int i, rc = 0;
2841 lines = str_list_make(NULL, cmd, ";");
2842 for (i = 0; lines[i]; i++) {
2843 rc |= process_line(ctx, lines[i]);
2845 talloc_free(lines);
2847 return rc;
2850 #define MAX_COMPLETIONS 100
2852 typedef struct {
2853 char *dirmask;
2854 char **matches;
2855 int count, samelen;
2856 const char *text;
2857 int len;
2858 } completion_remote_t;
2860 static void completion_remote_filter(struct clilist_file_info *f, const char *mask, void *state)
2862 completion_remote_t *info = (completion_remote_t *)state;
2864 if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (!ISDOT(f->name)) && (!ISDOTDOT(f->name))) {
2865 if ((info->dirmask[0] == 0) && !(f->attrib & FILE_ATTRIBUTE_DIRECTORY))
2866 info->matches[info->count] = strdup(f->name);
2867 else {
2868 char *tmp;
2870 if (info->dirmask[0] != 0)
2871 tmp = talloc_asprintf(NULL, "%s/%s", info->dirmask, f->name);
2872 else
2873 tmp = talloc_strdup(NULL, f->name);
2875 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2876 tmp = talloc_append_string(NULL, tmp, "/");
2877 info->matches[info->count] = tmp;
2879 if (info->matches[info->count] == NULL)
2880 return;
2881 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2882 smb_readline_ca_char(0);
2884 if (info->count == 1)
2885 info->samelen = strlen(info->matches[info->count]);
2886 else
2887 while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0)
2888 info->samelen--;
2889 info->count++;
2893 static char **remote_completion(const char *text, int len)
2895 char *dirmask;
2896 int i, ret;
2897 completion_remote_t info;
2899 info.samelen = len;
2900 info.text = text;
2901 info.len = len;
2903 if (len >= PATH_MAX)
2904 return(NULL);
2906 info.matches = malloc_array_p(char *, MAX_COMPLETIONS);
2907 if (!info.matches) return NULL;
2908 info.matches[0] = NULL;
2910 for (i = len-1; i >= 0; i--)
2911 if ((text[i] == '/') || (text[i] == '\\'))
2912 break;
2913 info.text = text+i+1;
2914 info.samelen = info.len = len-i-1;
2916 if (i > 0) {
2917 info.dirmask = talloc_strndup(NULL, text, i+1);
2918 info.dirmask[i+1] = 0;
2919 ret = asprintf(&dirmask, "%s%*s*", rl_ctx->remote_cur_dir, i-1,
2920 text);
2921 } else {
2922 ret = asprintf(&dirmask, "%s*", rl_ctx->remote_cur_dir);
2924 if (ret < 0) {
2925 goto cleanup;
2928 if (smbcli_list(rl_ctx->cli->tree, dirmask,
2929 FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN,
2930 completion_remote_filter, &info) < 0)
2931 goto cleanup;
2933 if (info.count == 2)
2934 info.matches[0] = strdup(info.matches[1]);
2935 else {
2936 info.matches[0] = malloc_array_p(char, info.samelen+1);
2937 if (!info.matches[0])
2938 goto cleanup;
2939 strncpy(info.matches[0], info.matches[1], info.samelen);
2940 info.matches[0][info.samelen] = 0;
2942 info.matches[info.count] = NULL;
2943 return info.matches;
2945 cleanup:
2946 for (i = 0; i < info.count; i++)
2947 free(info.matches[i]);
2948 free(info.matches);
2949 return NULL;
2952 static char **completion_fn(const char *text, int start, int end)
2954 smb_readline_ca_char(' ');
2956 if (start) {
2957 const char *buf, *sp;
2958 int i;
2959 char compl_type;
2961 buf = smb_readline_get_line_buffer();
2962 if (buf == NULL)
2963 return NULL;
2965 sp = strchr(buf, ' ');
2966 if (sp == NULL)
2967 return NULL;
2969 for (i = 0; commands[i].name; i++)
2970 if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0))
2971 break;
2972 if (commands[i].name == NULL)
2973 return NULL;
2975 while (*sp == ' ')
2976 sp++;
2978 if (sp == (buf + start))
2979 compl_type = commands[i].compl_args[0];
2980 else
2981 compl_type = commands[i].compl_args[1];
2983 if (compl_type == COMPL_REMOTE)
2984 return remote_completion(text, end - start);
2985 else /* fall back to local filename completion */
2986 return NULL;
2987 } else {
2988 char **matches;
2989 int i, len, samelen = 0, count=1;
2991 matches = malloc_array_p(char *, MAX_COMPLETIONS);
2992 if (!matches) return NULL;
2993 matches[0] = NULL;
2995 len = strlen(text);
2996 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
2997 if (strncmp(text, commands[i].name, len) == 0) {
2998 matches[count] = strdup(commands[i].name);
2999 if (!matches[count])
3000 goto cleanup;
3001 if (count == 1)
3002 samelen = strlen(matches[count]);
3003 else
3004 while (strncmp(matches[count], matches[count-1], samelen) != 0)
3005 samelen--;
3006 count++;
3010 switch (count) {
3011 case 0: /* should never happen */
3012 case 1:
3013 goto cleanup;
3014 case 2:
3015 matches[0] = strdup(matches[1]);
3016 break;
3017 default:
3018 matches[0] = malloc_array_p(char, samelen+1);
3019 if (!matches[0])
3020 goto cleanup;
3021 strncpy(matches[0], matches[1], samelen);
3022 matches[0][samelen] = 0;
3024 matches[count] = NULL;
3025 return matches;
3027 cleanup:
3028 count--;
3029 while (count >= 0) {
3030 free(matches[count]);
3031 count--;
3033 free(matches);
3034 return NULL;
3038 /****************************************************************************
3039 make sure we swallow keepalives during idle time
3040 ****************************************************************************/
3041 static void readline_callback(void)
3043 static time_t last_t;
3044 time_t t;
3046 t = time(NULL);
3048 if (t - last_t < 5) return;
3050 last_t = t;
3052 smbcli_transport_process(rl_ctx->cli->transport);
3054 if (rl_ctx->cli->tree) {
3055 smbcli_chkpath(rl_ctx->cli->tree, "\\");
3059 static int process_line(struct smbclient_context *ctx, const char *cline)
3061 const char **args;
3062 int i;
3064 /* and get the first part of the command */
3065 args = (const char **) str_list_make_shell(ctx, cline, NULL);
3066 if (!args || !args[0])
3067 return 0;
3069 if ((i = process_tok(args[0])) >= 0) {
3070 i = commands[i].fn(ctx, args);
3071 } else if (i == -2) {
3072 d_printf("%s: command abbreviation ambiguous\n",args[0]);
3073 } else {
3074 d_printf("%s: command not found\n",args[0]);
3077 talloc_free(args);
3079 return i;
3082 /****************************************************************************
3083 process commands on stdin
3084 ****************************************************************************/
3085 static int process_stdin(struct smbclient_context *ctx)
3087 int rc = 0;
3088 while (1) {
3089 /* display a prompt */
3090 char *the_prompt = talloc_asprintf(ctx, "smb: %s> ", ctx->remote_cur_dir);
3091 char *cline = smb_readline(the_prompt, readline_callback, completion_fn);
3092 talloc_free(the_prompt);
3094 if (!cline) break;
3096 /* special case - first char is ! */
3097 if (*cline == '!') {
3098 system(cline + 1);
3099 free(cline);
3100 continue;
3103 rc |= process_command_string(ctx, cline);
3104 free(cline);
3108 return rc;
3112 /*****************************************************
3113 return a connection to a server
3114 *******************************************************/
3115 static bool do_connect(struct smbclient_context *ctx,
3116 struct tevent_context *ev_ctx,
3117 struct resolve_context *resolve_ctx,
3118 const char *specified_server, const char **ports,
3119 const char *specified_share,
3120 const char *socket_options,
3121 struct cli_credentials *cred,
3122 struct smbcli_options *options,
3123 struct smbcli_session_options *session_options,
3124 struct gensec_settings *gensec_settings)
3126 NTSTATUS status;
3127 char *server, *share;
3129 rl_ctx = ctx; /* Ugly hack */
3131 if (strncmp(specified_share, "\\\\", 2) == 0 ||
3132 strncmp(specified_share, "//", 2) == 0) {
3133 smbcli_parse_unc(specified_share, ctx, &server, &share);
3134 } else {
3135 share = talloc_strdup(ctx, specified_share);
3136 server = talloc_strdup(ctx, specified_server);
3139 ctx->remote_cur_dir = talloc_strdup(ctx, "\\");
3141 status = smbcli_full_connection(ctx, &ctx->cli, server, ports,
3142 share, NULL,
3143 socket_options,
3144 cred, resolve_ctx,
3145 ev_ctx, options, session_options,
3146 gensec_settings);
3147 if (!NT_STATUS_IS_OK(status)) {
3148 d_printf("Connection to \\\\%s\\%s failed - %s\n",
3149 server, share, nt_errstr(status));
3150 talloc_free(ctx);
3151 return false;
3154 return true;
3157 /****************************************************************************
3158 handle a -L query
3159 ****************************************************************************/
3160 static int do_host_query(struct loadparm_context *lp_ctx,
3161 struct tevent_context *ev_ctx,
3162 const char *query_host,
3163 const char *workgroup)
3165 browse_host(lp_ctx, ev_ctx, query_host);
3166 list_servers(workgroup);
3167 return(0);
3171 /****************************************************************************
3172 handle a message operation
3173 ****************************************************************************/
3174 static int do_message_op(const char *netbios_name, const char *desthost,
3175 const char **destports, const char *destip,
3176 int name_type,
3177 struct tevent_context *ev_ctx,
3178 struct resolve_context *resolve_ctx,
3179 struct smbcli_options *options,
3180 const char *socket_options)
3182 struct nbt_name called, calling;
3183 const char *server_name;
3184 struct smbcli_state *cli;
3186 make_nbt_name_client(&calling, netbios_name);
3188 nbt_choose_called_name(NULL, &called, desthost, name_type);
3190 server_name = destip ? destip : desthost;
3192 if (!(cli = smbcli_state_init(NULL)) ||
3193 !smbcli_socket_connect(cli, server_name, destports,
3194 ev_ctx, resolve_ctx, options,
3195 socket_options)) {
3196 d_printf("Connection to %s failed\n", server_name);
3197 return 1;
3200 if (!smbcli_transport_establish(cli, &calling, &called)) {
3201 d_printf("session request failed\n");
3202 talloc_free(cli);
3203 return 1;
3206 send_message(cli, desthost);
3207 talloc_free(cli);
3209 return 0;
3213 /****************************************************************************
3214 main program
3215 ****************************************************************************/
3216 int main(int argc,char *argv[])
3218 char *base_directory = NULL;
3219 const char *dest_ip = NULL;
3220 int opt;
3221 const char *query_host = NULL;
3222 bool message = false;
3223 char *desthost = NULL;
3224 poptContext pc;
3225 const char *service = NULL;
3226 int port = 0;
3227 char *p;
3228 int rc = 0;
3229 int name_type = 0x20;
3230 TALLOC_CTX *mem_ctx;
3231 struct tevent_context *ev_ctx;
3232 struct smbclient_context *ctx;
3233 const char *cmdstr = NULL;
3234 struct smbcli_options smb_options;
3235 struct smbcli_session_options smb_session_options;
3237 struct poptOption long_options[] = {
3238 POPT_AUTOHELP
3240 { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" },
3241 { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" },
3242 { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" },
3243 { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" },
3244 { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" },
3245 { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" },
3246 { "send-buffer", 'b', POPT_ARG_INT, NULL, 'b', "Changes the transmit/send buffer", "BYTES" },
3247 { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" },
3248 POPT_COMMON_SAMBA
3249 POPT_COMMON_CONNECTION
3250 POPT_COMMON_CREDENTIALS
3251 POPT_COMMON_VERSION
3252 { NULL }
3255 mem_ctx = talloc_init("client.c/main");
3256 if (!mem_ctx) {
3257 d_printf("\nclient.c: Not enough memory\n");
3258 exit(1);
3261 ctx = talloc_zero(mem_ctx, struct smbclient_context);
3262 ctx->io_bufsize = 64512;
3264 pc = poptGetContext("smbclient", argc, (const char **) argv, long_options, 0);
3265 poptSetOtherOptionHelp(pc, "[OPTIONS] service <password>");
3267 while ((opt = poptGetNextOpt(pc)) != -1) {
3268 switch (opt) {
3269 case 'M':
3270 /* Messages are sent to NetBIOS name type 0x3
3271 * (Messenger Service). Make sure we default
3272 * to port 139 instead of port 445. srl,crh
3274 name_type = 0x03;
3275 desthost = strdup(poptGetOptArg(pc));
3276 if( 0 == port ) port = 139;
3277 message = true;
3278 break;
3279 case 'I':
3280 dest_ip = poptGetOptArg(pc);
3281 break;
3282 case 'L':
3283 query_host = strdup(poptGetOptArg(pc));
3284 break;
3285 case 'D':
3286 base_directory = strdup(poptGetOptArg(pc));
3287 break;
3288 case 'b':
3289 ctx->io_bufsize = MAX(1, atoi(poptGetOptArg(pc)));
3290 break;
3294 gensec_init();
3296 if(poptPeekArg(pc)) {
3297 char *s = strdup(poptGetArg(pc));
3299 /* Convert any '/' characters in the service name to '\' characters */
3300 string_replace(s, '/','\\');
3302 service = s;
3304 if (count_chars(s,'\\') < 3) {
3305 d_printf("\n%s: Not enough '\\' characters in service\n",s);
3306 poptPrintUsage(pc, stderr, 0);
3307 exit(1);
3311 if (poptPeekArg(pc)) {
3312 cli_credentials_set_password(cmdline_credentials, poptGetArg(pc), CRED_SPECIFIED);
3315 /*init_names(); */
3317 if (!query_host && !service && !message) {
3318 poptPrintUsage(pc, stderr, 0);
3319 exit(1);
3322 poptFreeContext(pc);
3324 lpcfg_smbcli_options(cmdline_lp_ctx, &smb_options);
3325 lpcfg_smbcli_session_options(cmdline_lp_ctx, &smb_session_options);
3327 ev_ctx = s4_event_context_init(talloc_autofree_context());
3329 DEBUG( 3, ( "Client started (version %s).\n", SAMBA_VERSION_STRING ) );
3331 if (query_host && (p=strchr_m(query_host,'#'))) {
3332 *p = 0;
3333 p++;
3334 sscanf(p, "%x", &name_type);
3337 if (query_host) {
3338 rc = do_host_query(cmdline_lp_ctx, ev_ctx, query_host,
3339 lpcfg_workgroup(cmdline_lp_ctx));
3340 return rc;
3343 if (message) {
3344 rc = do_message_op(lpcfg_netbios_name(cmdline_lp_ctx), desthost,
3345 lpcfg_smb_ports(cmdline_lp_ctx), dest_ip,
3346 name_type, ev_ctx,
3347 lpcfg_resolve_context(cmdline_lp_ctx),
3348 &smb_options,
3349 lpcfg_socket_options(cmdline_lp_ctx));
3350 return rc;
3353 if (!do_connect(ctx, ev_ctx, lpcfg_resolve_context(cmdline_lp_ctx),
3354 desthost, lpcfg_smb_ports(cmdline_lp_ctx), service,
3355 lpcfg_socket_options(cmdline_lp_ctx),
3356 cmdline_credentials, &smb_options, &smb_session_options,
3357 lpcfg_gensec_settings(ctx, cmdline_lp_ctx)))
3358 return 1;
3360 if (base_directory) {
3361 do_cd(ctx, base_directory);
3362 free(base_directory);
3365 if (cmdstr) {
3366 rc = process_command_string(ctx, cmdstr);
3367 } else {
3368 rc = process_stdin(ctx);
3371 free(desthost);
3372 talloc_free(mem_ctx);
3374 return rc;