libcli:util: Update HRESULT definitions
[Samba.git] / source4 / client / client.c
blob1a4e538e6edd7c65963b41c9324704bcbe6ba724
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/cmdline.h"
37 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
38 #include "librpc/gen_ndr/ndr_lsa.h"
39 #include "librpc/gen_ndr/ndr_security.h"
40 #include "libcli/util/clilsa.h"
41 #include "system/dir.h"
42 #include "system/filesys.h"
43 #include "../lib/util/dlinklist.h"
44 #include "system/readline.h"
45 #include "auth/credentials/credentials.h"
46 #include "auth/gensec/gensec.h"
47 #include "system/time.h" /* needed by some systems for asctime() */
48 #include "libcli/resolve/resolve.h"
49 #include "libcli/security/security.h"
50 #include "../libcli/smbreadline/smbreadline.h"
51 #include "librpc/gen_ndr/ndr_nbt.h"
52 #include "param/param.h"
53 #include "libcli/raw/raw_proto.h"
55 /* the default pager to use for the client "more" command. Users can
56 * override this with the PAGER environment variable */
57 #ifndef DEFAULT_PAGER
58 #define DEFAULT_PAGER "more"
59 #endif
61 #undef strncasecmp
63 struct smbclient_context {
64 char *remote_cur_dir;
65 struct smbcli_state *cli;
66 char *fileselection;
67 time_t newer_than;
68 bool prompt;
69 bool recurse;
70 int archive_level;
71 bool lowercase;
72 int printmode;
73 bool translation;
74 int io_bufsize;
77 /* timing globals */
78 static uint64_t get_total_size = 0;
79 static unsigned int get_total_time_ms = 0;
80 static uint64_t put_total_size = 0;
81 static unsigned int put_total_time_ms = 0;
83 /* Unfortunately, there is no way to pass a context to the completion function as an argument */
84 static struct smbclient_context *rl_ctx;
86 /* totals globals */
87 static double dir_total;
89 /*******************************************************************
90 Reduce a file name, removing .. elements.
91 ********************************************************************/
92 static void dos_clean_name(char *s)
94 char *p=NULL,*r;
96 DEBUG(3,("dos_clean_name [%s]\n",s));
98 /* remove any double slashes */
99 all_string_sub(s, "\\\\", "\\", 0);
101 while ((p = strstr(s,"\\..\\")) != NULL) {
102 *p = '\0';
103 if ((r = strrchr(s,'\\')) != NULL)
104 memmove(r,p+3,strlen(p+3)+1);
107 trim_string(s,NULL,"\\..");
109 all_string_sub(s, "\\.\\", "\\", 0);
112 /****************************************************************************
113 write to a local file with CR/LF->LF translation if appropriate. return the
114 number taken from the buffer. This may not equal the number written.
115 ****************************************************************************/
116 static int writefile(int f, const void *_b, int n, bool translation)
118 const uint8_t *b = (const uint8_t *)_b;
119 int i;
121 if (!translation) {
122 return write(f,b,n);
125 i = 0;
126 while (i < n) {
127 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
128 b++;i++;
130 if (write(f, b, 1) != 1) {
131 break;
133 b++;
134 i++;
137 return(i);
140 /****************************************************************************
141 read from a file with LF->CR/LF translation if appropriate. return the
142 number read. read approx n bytes.
143 ****************************************************************************/
144 static int readfile(void *_b, int n, FILE *f, bool translation)
146 uint8_t *b = (uint8_t *)_b;
147 int i;
148 int c;
150 if (!translation)
151 return fread(b,1,n,f);
153 i = 0;
154 while (i < (n - 1)) {
155 if ((c = getc(f)) == EOF) {
156 break;
159 if (c == '\n') { /* change all LFs to CR/LF */
160 b[i++] = '\r';
163 b[i++] = c;
166 return(i);
170 /****************************************************************************
171 send a message
172 ****************************************************************************/
173 static void send_message(struct smbcli_state *cli, const char *desthost)
175 char msg[1600];
176 int total_len = 0;
177 int grp_id;
178 struct cli_credentials *creds = samba_cmdline_get_creds();
180 if (!smbcli_message_start(cli->tree,
181 desthost,
182 cli_credentials_get_username(creds),
183 &grp_id)) {
184 d_printf("message start: %s\n", smbcli_errstr(cli->tree));
185 return;
189 d_printf("Connected. Type your message, ending it with a Control-D\n");
191 while (!feof(stdin) && total_len < 1600) {
192 int maxlen = MIN(1600 - total_len,127);
193 int l=0;
194 int c;
196 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
197 if (c == '\n')
198 msg[l++] = '\r';
199 msg[l] = c;
202 if (!smbcli_message_text(cli->tree, msg, l, grp_id)) {
203 d_printf("SMBsendtxt failed (%s)\n",smbcli_errstr(cli->tree));
204 return;
207 total_len += l;
210 if (total_len >= 1600)
211 d_printf("the message was truncated to 1600 bytes\n");
212 else
213 d_printf("sent %d bytes\n",total_len);
215 if (!smbcli_message_end(cli->tree, grp_id)) {
216 d_printf("SMBsendend failed (%s)\n",smbcli_errstr(cli->tree));
217 return;
223 /****************************************************************************
224 check the space on a device
225 ****************************************************************************/
226 static int do_dskattr(struct smbclient_context *ctx)
228 uint32_t bsize;
229 uint64_t total, avail;
231 if (NT_STATUS_IS_ERR(smbcli_dskattr(ctx->cli->tree, &bsize, &total, &avail))) {
232 d_printf("Error in dskattr: %s\n",smbcli_errstr(ctx->cli->tree));
233 return 1;
236 d_printf("\n\t\t%llu blocks of size %u. %llu blocks available\n",
237 (unsigned long long)total,
238 (unsigned)bsize,
239 (unsigned long long)avail);
241 return 0;
244 /****************************************************************************
245 show cd/pwd
246 ****************************************************************************/
247 static int cmd_pwd(struct smbclient_context *ctx, const char **args)
249 d_printf("Current directory is %s\n", ctx->remote_cur_dir);
250 return 0;
254 convert a string to dos format
256 static void dos_format(char *s)
258 string_replace(s, '/', '\\');
261 /****************************************************************************
262 change directory - inner section
263 ****************************************************************************/
264 static int do_cd(struct smbclient_context *ctx, const char *newdir)
266 char *dname;
268 /* Save the current directory in case the
269 new directory is invalid */
270 if (newdir[0] == '\\')
271 dname = talloc_strdup(ctx, newdir);
272 else
273 dname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, newdir);
275 dos_format(dname);
277 if (*(dname+strlen(dname)-1) != '\\') {
278 dname = talloc_append_string(NULL, dname, "\\");
280 dos_clean_name(dname);
282 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, dname))) {
283 d_printf("cd %s: %s\n", dname, smbcli_errstr(ctx->cli->tree));
284 talloc_free(dname);
285 } else {
286 ctx->remote_cur_dir = dname;
289 return 0;
292 /****************************************************************************
293 change directory
294 ****************************************************************************/
295 static int cmd_cd(struct smbclient_context *ctx, const char **args)
297 int rc = 0;
299 if (args[1])
300 rc = do_cd(ctx, args[1]);
301 else
302 d_printf("Current directory is %s\n",ctx->remote_cur_dir);
304 return rc;
308 static bool mask_match(struct smbcli_state *c, const char *string,
309 const char *pattern, bool is_case_sensitive)
311 int ret;
313 if (ISDOTDOT(string))
314 string = ".";
315 if (ISDOT(pattern))
316 return false;
318 ret = ms_fnmatch_protocol(pattern, string,
319 c->transport->negotiate.protocol,
320 is_case_sensitive);
321 return (ret == 0);
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;
457 if (entry == NULL) {
458 entry = "";
461 new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
462 while (new_end > do_list_queue_size)
464 do_list_queue_size *= 2;
465 DEBUG(4,("enlarging do_list_queue to %d\n",
466 (int)do_list_queue_size));
467 dlq = realloc_p(do_list_queue, char, do_list_queue_size);
468 if (! dlq) {
469 d_printf("failure enlarging do_list_queue to %d bytes\n",
470 (int)do_list_queue_size);
471 reset_do_list_queue();
473 else
475 do_list_queue = dlq;
476 memset(do_list_queue + do_list_queue_size / 2,
477 0, do_list_queue_size / 2);
480 if (do_list_queue)
482 strlcpy(do_list_queue + do_list_queue_end, entry,
483 do_list_queue_size - do_list_queue_end);
484 do_list_queue_end = new_end;
485 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
486 entry, (int)do_list_queue_start, (int)do_list_queue_end));
490 static char *do_list_queue_head(void)
492 return do_list_queue + do_list_queue_start;
495 static void remove_do_list_queue_head(void)
497 if (do_list_queue_end > do_list_queue_start)
499 do_list_queue_start += strlen(do_list_queue_head()) + 1;
500 adjust_do_list_queue();
501 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
502 (int)do_list_queue_start, (int)do_list_queue_end));
506 static int do_list_queue_empty(void)
508 return (! (do_list_queue && *do_list_queue));
511 /****************************************************************************
512 a helper for do_list
513 ****************************************************************************/
514 static void do_list_helper(struct clilist_file_info *f, const char *mask, void *state)
516 struct smbclient_context *ctx = (struct smbclient_context *)state;
518 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY) {
519 if (do_list_dirs && do_this_one(ctx, f)) {
520 do_list_fn(ctx, f);
522 if (do_list_recurse &&
523 !ISDOT(f->name) &&
524 !ISDOTDOT(f->name)) {
525 char *mask2;
526 char *p;
528 mask2 = talloc_strdup(NULL, mask);
529 p = strrchr_m(mask2,'\\');
530 if (!p) return;
531 p[1] = 0;
532 mask2 = talloc_asprintf_append_buffer(mask2, "%s\\*", f->name);
533 add_to_do_list_queue(mask2);
535 return;
538 if (do_this_one(ctx, f)) {
539 do_list_fn(ctx, f);
544 /****************************************************************************
545 a wrapper around smbcli_list that adds recursion
546 ****************************************************************************/
547 static void do_list(struct smbclient_context *ctx, const char *mask,uint16_t attribute,
548 void (*fn)(struct smbclient_context *, struct clilist_file_info *),bool rec, bool dirs)
550 static int in_do_list = 0;
552 if (in_do_list && rec)
554 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
555 exit(1);
558 in_do_list = 1;
560 do_list_recurse = rec;
561 do_list_dirs = dirs;
562 do_list_fn = fn;
564 if (rec)
566 init_do_list_queue();
567 add_to_do_list_queue(mask);
569 while (! do_list_queue_empty())
572 * Need to copy head so that it doesn't become
573 * invalid inside the call to smbcli_list. This
574 * would happen if the list were expanded
575 * during the call.
576 * Fix from E. Jay Berkenbilt (ejb@ql.org)
578 char *head;
579 head = do_list_queue_head();
580 smbcli_list(ctx->cli->tree, head, attribute, do_list_helper, ctx);
581 remove_do_list_queue_head();
582 if ((! do_list_queue_empty()) && (fn == display_finfo))
584 char* next_file = do_list_queue_head();
585 char* save_ch = 0;
586 if ((strlen(next_file) >= 2) &&
587 (next_file[strlen(next_file) - 1] == '*') &&
588 (next_file[strlen(next_file) - 2] == '\\'))
590 save_ch = next_file +
591 strlen(next_file) - 2;
592 *save_ch = '\0';
594 d_printf("\n%s\n",next_file);
595 if (save_ch)
597 *save_ch = '\\';
602 else
604 if (smbcli_list(ctx->cli->tree, mask, attribute, do_list_helper, ctx) == -1)
606 d_printf("%s listing %s\n", smbcli_errstr(ctx->cli->tree), mask);
610 in_do_list = 0;
611 reset_do_list_queue();
614 /****************************************************************************
615 get a directory listing
616 ****************************************************************************/
617 static int cmd_dir(struct smbclient_context *ctx, const char **args)
619 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
620 char *mask;
621 int rc;
623 dir_total = 0;
625 mask = talloc_strdup(ctx, ctx->remote_cur_dir);
626 if(mask[strlen(mask)-1]!='\\')
627 mask = talloc_append_string(ctx, mask,"\\");
629 if (args[1]) {
630 mask = talloc_strdup(ctx, args[1]);
631 if (mask[0] != '\\')
632 mask = talloc_append_string(ctx, mask, "\\");
633 dos_format(mask);
635 else {
636 if (ctx->cli->tree->session->transport->negotiate.protocol <=
637 PROTOCOL_LANMAN1) {
638 mask = talloc_append_string(ctx, mask, "*.*");
639 } else {
640 mask = talloc_append_string(ctx, mask, "*");
644 do_list(ctx, mask, attribute, display_finfo, ctx->recurse, true);
646 rc = do_dskattr(ctx);
648 DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
650 return rc;
654 /****************************************************************************
655 get a directory listing
656 ****************************************************************************/
657 static int cmd_du(struct smbclient_context *ctx, const char **args)
659 uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
660 int rc;
661 char *mask;
663 dir_total = 0;
665 if (args[1]) {
666 if (args[1][0] == '\\')
667 mask = talloc_strdup(ctx, args[1]);
668 else
669 mask = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
670 dos_format(mask);
671 } else {
672 mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
675 do_list(ctx, mask, attribute, do_du, ctx->recurse, true);
677 talloc_free(mask);
679 rc = do_dskattr(ctx);
681 d_printf("Total number of bytes: %.0f\n", dir_total);
683 return rc;
687 /****************************************************************************
688 get a file from rname to lname
689 ****************************************************************************/
690 static int do_get(struct smbclient_context *ctx, char *rname, const char *p_lname, bool reget)
692 int handle = 0, fnum;
693 bool newhandle = false;
694 uint8_t *data;
695 struct timeval tp_start;
696 int read_size = ctx->io_bufsize;
697 uint16_t attr;
698 size_t size;
699 off_t start = 0;
700 off_t nread = 0;
701 int rc = 0;
702 char *lname;
705 GetTimeOfDay(&tp_start);
707 if (ctx->lowercase) {
708 lname = strlower_talloc(ctx, p_lname);
709 } else {
710 lname = talloc_strdup(ctx, p_lname);
713 fnum = smbcli_open(ctx->cli->tree, rname, O_RDONLY, DENY_NONE);
715 if (fnum == -1) {
716 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
717 return 1;
720 if(!strcmp(lname,"-")) {
721 handle = fileno(stdout);
722 } else {
723 if (reget) {
724 handle = open(lname, O_WRONLY|O_CREAT, 0644);
725 if (handle >= 0) {
726 start = lseek(handle, 0, SEEK_END);
727 if (start == -1) {
728 d_printf("Error seeking local file\n");
729 close(handle);
730 return 1;
733 } else {
734 handle = open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
736 newhandle = true;
738 if (handle < 0) {
739 d_printf("Error opening local file %s\n",lname);
740 return 1;
744 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum,
745 &attr, &size, NULL, NULL, NULL, NULL, NULL)) &&
746 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum,
747 &attr, &size, NULL, NULL, NULL))) {
748 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
749 if (newhandle) {
750 close(handle);
752 return 1;
755 DEBUG(2,("getting file %s of size %.0f as %s ",
756 rname, (double)size, lname));
758 if(!(data = (uint8_t *)malloc(read_size))) {
759 d_printf("malloc fail for size %d\n", read_size);
760 smbcli_close(ctx->cli->tree, fnum);
761 if (newhandle) {
762 close(handle);
764 return 1;
767 while (1) {
768 int n = smbcli_read(ctx->cli->tree, fnum, data, nread + start, read_size);
770 if (n <= 0) break;
772 if (writefile(handle,data, n, ctx->translation) != n) {
773 d_printf("Error writing local file\n");
774 rc = 1;
775 break;
778 nread += n;
781 if (nread + start < size) {
782 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
783 rname, (long)nread));
785 rc = 1;
788 SAFE_FREE(data);
790 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
791 d_printf("Error %s closing remote file\n",smbcli_errstr(ctx->cli->tree));
792 rc = 1;
795 if (newhandle) {
796 close(handle);
799 if (ctx->archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) {
800 smbcli_setatr(ctx->cli->tree, rname, attr & ~(uint16_t)FILE_ATTRIBUTE_ARCHIVE, 0);
804 struct timeval tp_end;
805 int this_time;
807 GetTimeOfDay(&tp_end);
808 this_time =
809 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
810 (tp_end.tv_usec - tp_start.tv_usec)/1000;
811 get_total_time_ms += this_time;
812 get_total_size += nread;
814 DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n",
815 nread / (1.024*this_time + 1.0e-4),
816 get_total_size / (1.024*get_total_time_ms)));
819 return rc;
823 /****************************************************************************
824 get a file
825 ****************************************************************************/
826 static int cmd_get(struct smbclient_context *ctx, const char **args)
828 const char *lname;
829 char *rname;
831 if (!args[1]) {
832 d_printf("get <filename>\n");
833 return 1;
836 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
838 if (args[2])
839 lname = args[2];
840 else
841 lname = args[1];
843 dos_clean_name(rname);
845 return do_get(ctx, rname, lname, false);
848 /****************************************************************************
849 Put up a yes/no prompt.
850 ****************************************************************************/
851 static bool yesno(char *p)
853 char ans[4];
854 printf("%s",p);
856 if (!fgets(ans,sizeof(ans)-1,stdin))
857 return(false);
859 if (*ans == 'y' || *ans == 'Y')
860 return(true);
862 return(false);
865 /****************************************************************************
866 do a mget operation on one file
867 ****************************************************************************/
868 static void do_mget(struct smbclient_context *ctx, struct clilist_file_info *finfo)
870 char *rname;
871 char *quest;
872 char *mget_mask;
873 char *saved_curdir;
874 char *l_fname;
875 int ret;
877 if (ISDOT(finfo->name) || ISDOTDOT(finfo->name))
878 return;
880 if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)
881 quest = talloc_asprintf(ctx, "Get directory %s? ",finfo->name);
882 else
883 quest = talloc_asprintf(ctx, "Get file %s? ",finfo->name);
885 if (ctx->prompt && !yesno(quest)) return;
887 talloc_free(quest);
889 if (!(finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)) {
890 rname = talloc_asprintf(ctx, "%s%s",ctx->remote_cur_dir,
891 finfo->name);
892 do_get(ctx, rname, finfo->name, false);
893 talloc_free(rname);
894 return;
897 /* handle directories */
898 saved_curdir = talloc_strdup(ctx, ctx->remote_cur_dir);
900 ctx->remote_cur_dir = talloc_asprintf_append_buffer(NULL, "%s\\", finfo->name);
902 if (ctx->lowercase) {
903 l_fname = strlower_talloc(ctx, finfo->name);
904 } else {
905 l_fname = talloc_strdup(ctx, finfo->name);
908 string_replace(l_fname, '\\', '/');
910 if (!directory_exist(l_fname) &&
911 mkdir(l_fname, 0777) != 0) {
912 d_printf("failed to create directory %s\n", l_fname);
913 return;
916 if (chdir(l_fname) != 0) {
917 d_printf("failed to chdir to directory %s\n", l_fname);
918 return;
921 mget_mask = talloc_asprintf(ctx, "%s*", ctx->remote_cur_dir);
923 do_list(ctx, mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,false, true);
924 ret = chdir("..");
925 if (ret == -1) {
926 d_printf("failed to chdir to '..': %s\n", strerror(errno));
927 return;
929 talloc_free(ctx->remote_cur_dir);
931 ctx->remote_cur_dir = saved_curdir;
935 /****************************************************************************
936 view the file using the pager
937 ****************************************************************************/
938 static int cmd_more(struct smbclient_context *ctx, const char **args)
940 char *rname;
941 char *pager_cmd;
942 char *lname;
943 char *pager;
944 int fd;
945 int rc = 0;
946 mode_t mask;
948 lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir());
949 mask = umask(S_IRWXO | S_IRWXG);
950 fd = mkstemp(lname);
951 umask(mask);
952 if (fd == -1) {
953 d_printf("failed to create temporary file for more\n");
954 return 1;
956 close(fd);
958 if (!args[1]) {
959 d_printf("more <filename>\n");
960 unlink(lname);
961 return 1;
963 rname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
964 dos_clean_name(rname);
966 rc = do_get(ctx, rname, lname, false);
968 pager=getenv("PAGER");
970 pager_cmd = talloc_asprintf(ctx, "%s %s",(pager? pager:DEFAULT_PAGER), lname);
971 rc = system(pager_cmd);
972 if (rc == -1) {
973 d_printf("failed to call pager command\n");
974 return 1;
976 unlink(lname);
978 return rc;
983 /****************************************************************************
984 do a mget command
985 ****************************************************************************/
986 static int cmd_mget(struct smbclient_context *ctx, const char **args)
988 uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
989 char *mget_mask = NULL;
990 int i;
992 if (ctx->recurse)
993 attribute |= FILE_ATTRIBUTE_DIRECTORY;
995 for (i = 1; args[i]; i++) {
996 mget_mask = talloc_strdup(ctx, ctx->remote_cur_dir);
997 if(mget_mask[strlen(mget_mask)-1]!='\\')
998 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
1000 mget_mask = talloc_strdup(ctx, args[i]);
1001 if (mget_mask[0] != '\\')
1002 mget_mask = talloc_append_string(ctx, mget_mask, "\\");
1003 do_list(ctx, mget_mask, attribute,do_mget,false,true);
1005 talloc_free(mget_mask);
1008 if (mget_mask == NULL) {
1009 mget_mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
1010 do_list(ctx, mget_mask, attribute,do_mget,false,true);
1011 talloc_free(mget_mask);
1014 return 0;
1018 /****************************************************************************
1019 make a directory of name "name"
1020 ****************************************************************************/
1021 static NTSTATUS do_mkdir(struct smbclient_context *ctx, char *name)
1023 NTSTATUS status;
1025 if (NT_STATUS_IS_ERR(status = smbcli_mkdir(ctx->cli->tree, name))) {
1026 d_printf("%s making remote directory %s\n",
1027 smbcli_errstr(ctx->cli->tree),name);
1028 return status;
1031 return status;
1035 /****************************************************************************
1036 Exit client.
1037 ****************************************************************************/
1038 static int cmd_quit(struct smbclient_context *ctx, const char **args)
1040 talloc_free(ctx);
1041 exit(0);
1042 /* NOTREACHED */
1043 return 0;
1047 /****************************************************************************
1048 make a directory
1049 ****************************************************************************/
1050 static int cmd_mkdir(struct smbclient_context *ctx, const char **args)
1052 char *mask, *p;
1054 if (!args[1]) {
1055 if (!ctx->recurse)
1056 d_printf("mkdir <dirname>\n");
1057 return 1;
1060 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir,args[1]);
1062 if (ctx->recurse) {
1063 dos_clean_name(mask);
1065 trim_string(mask,".",NULL);
1066 for (p = strtok(mask,"/\\"); p; p = strtok(p, "/\\")) {
1067 char *parent = talloc_strndup(ctx, mask, PTR_DIFF(p, mask));
1069 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, parent))) {
1070 do_mkdir(ctx, parent);
1073 talloc_free(parent);
1075 } else {
1076 do_mkdir(ctx, mask);
1079 return 0;
1082 /****************************************************************************
1083 show 8.3 name of a file
1084 ****************************************************************************/
1085 static int cmd_altname(struct smbclient_context *ctx, const char **args)
1087 const char *p;
1088 char *altname;
1089 char *name;
1091 if (!args[1]) {
1092 d_printf("altname <file>\n");
1093 return 1;
1096 name = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1098 if (!NT_STATUS_IS_OK(smbcli_qpathinfo_alt_name(ctx->cli->tree, name, &p))) {
1099 d_printf("%s getting alt name for %s\n",
1100 smbcli_errstr(ctx->cli->tree),name);
1101 return(false);
1103 altname = discard_const_p(char, p);
1104 d_printf("%s\n", altname);
1106 SAFE_FREE(altname);
1108 return 0;
1112 /****************************************************************************
1113 put a single file
1114 ****************************************************************************/
1115 static int do_put(struct smbclient_context *ctx, char *rname, char *lname, bool reput)
1117 int fnum;
1118 FILE *f;
1119 size_t start = 0;
1120 off_t nread = 0;
1121 uint8_t *buf = NULL;
1122 int maxwrite = ctx->io_bufsize;
1123 int rc = 0;
1125 struct timeval tp_start;
1126 GetTimeOfDay(&tp_start);
1128 if (reput) {
1129 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT, DENY_NONE);
1130 if (fnum >= 0) {
1131 if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL)) &&
1132 NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL))) {
1133 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
1134 return 1;
1137 } else {
1138 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT|O_TRUNC,
1139 DENY_NONE);
1142 if (fnum == -1) {
1143 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1144 return 1;
1147 /* allow files to be piped into smbclient
1148 jdblair 24.jun.98
1150 Note that in this case this function will exit(0) rather
1151 than returning. */
1152 if (!strcmp(lname, "-")) {
1153 f = stdin;
1154 /* size of file is not known */
1155 } else {
1156 f = fopen(lname, "r");
1157 if (f && reput) {
1158 if (fseek(f, start, SEEK_SET) == -1) {
1159 d_printf("Error seeking local file\n");
1160 fclose(f);
1161 return 1;
1166 if (!f) {
1167 d_printf("Error opening local file %s\n",lname);
1168 return 1;
1172 DEBUG(1,("putting file %s as %s ",lname,
1173 rname));
1175 buf = (uint8_t *)malloc(maxwrite);
1176 if (!buf) {
1177 d_printf("ERROR: Not enough memory!\n");
1178 fclose(f);
1179 return 1;
1181 while (!feof(f)) {
1182 int n = maxwrite;
1183 int ret;
1185 if ((n = readfile(buf,n,f,ctx->translation)) < 1) {
1186 if((n == 0) && feof(f))
1187 break; /* Empty local file. */
1189 d_printf("Error reading local file: %s\n", strerror(errno));
1190 rc = 1;
1191 break;
1194 ret = smbcli_write(ctx->cli->tree, fnum, 0, buf, nread + start, n);
1196 if (n != ret) {
1197 d_printf("Error writing file: %s\n", smbcli_errstr(ctx->cli->tree));
1198 rc = 1;
1199 break;
1202 nread += n;
1205 if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
1206 d_printf("%s closing remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1207 fclose(f);
1208 SAFE_FREE(buf);
1209 return 1;
1213 if (f != stdin) {
1214 fclose(f);
1217 SAFE_FREE(buf);
1220 struct timeval tp_end;
1221 int this_time;
1223 GetTimeOfDay(&tp_end);
1224 this_time =
1225 (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1226 (tp_end.tv_usec - tp_start.tv_usec)/1000;
1227 put_total_time_ms += this_time;
1228 put_total_size += nread;
1230 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1231 nread / (1.024*this_time + 1.0e-4),
1232 put_total_size / (1.024*put_total_time_ms)));
1235 if (f == stdin) {
1236 talloc_free(ctx);
1237 exit(0);
1240 return rc;
1245 /****************************************************************************
1246 put a file
1247 ****************************************************************************/
1248 static int cmd_put(struct smbclient_context *ctx, const char **args)
1250 char *lname;
1251 char *rname;
1253 if (!args[1]) {
1254 d_printf("put <filename> [<remotename>]\n");
1255 return 1;
1258 lname = talloc_strdup(ctx, args[1]);
1260 if (args[2]) {
1261 if (args[2][0]=='\\')
1262 rname = talloc_strdup(ctx, args[2]);
1263 else
1264 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[2]);
1265 } else {
1266 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, lname);
1269 dos_clean_name(rname);
1271 /* allow '-' to represent stdin
1272 jdblair, 24.jun.98 */
1273 if (!file_exist(lname) && (strcmp(lname,"-"))) {
1274 d_printf("%s does not exist\n",lname);
1275 return 1;
1278 return do_put(ctx, rname, lname, false);
1281 /*************************************
1282 File list structure
1283 *************************************/
1285 static struct file_list {
1286 struct file_list *prev, *next;
1287 char *file_path;
1288 bool isdir;
1289 } *file_list;
1291 /****************************************************************************
1292 Free a file_list structure
1293 ****************************************************************************/
1295 static void free_file_list (struct file_list * list)
1297 struct file_list *tmp;
1299 while (list)
1301 tmp = list;
1302 DLIST_REMOVE(list, list);
1303 SAFE_FREE(tmp->file_path);
1304 SAFE_FREE(tmp);
1308 /****************************************************************************
1309 seek in a directory/file list until you get something that doesn't start with
1310 the specified name
1311 ****************************************************************************/
1312 static bool seek_list(struct file_list *list, char *name)
1314 while (list) {
1315 trim_string(list->file_path,"./","\n");
1316 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1317 return(true);
1319 list = list->next;
1322 return(false);
1325 /****************************************************************************
1326 set the file selection mask
1327 ****************************************************************************/
1328 static int cmd_select(struct smbclient_context *ctx, const char **args)
1330 talloc_free(ctx->fileselection);
1331 ctx->fileselection = talloc_strdup(ctx, args[1]);
1333 return 0;
1336 /*******************************************************************
1337 A readdir wrapper which just returns the file name.
1338 ********************************************************************/
1339 static const char *readdirname(DIR *p)
1341 struct dirent *ptr;
1342 char *dname;
1344 if (!p)
1345 return(NULL);
1347 ptr = (struct dirent *)readdir(p);
1348 if (!ptr)
1349 return(NULL);
1351 dname = ptr->d_name;
1353 #ifdef HAVE_BROKEN_READDIR
1354 /* using /usr/ucb/cc is BAD */
1355 dname = dname - 2;
1356 #endif
1359 static char *buf;
1360 int len = NAMLEN(ptr);
1361 buf = talloc_strndup(NULL, dname, len);
1362 dname = buf;
1365 return(dname);
1368 /****************************************************************************
1369 Recursive file matching function act as find
1370 match must be always set to true when calling this function
1371 ****************************************************************************/
1372 static int file_find(struct smbclient_context *ctx, struct file_list **list, const char *directory,
1373 const char *expression, bool match)
1375 DIR *dir;
1376 struct file_list *entry;
1377 struct stat statbuf;
1378 int ret;
1379 char *path;
1380 bool isdir;
1381 const char *dname;
1383 dir = opendir(directory);
1384 if (!dir) return -1;
1386 while ((dname = readdirname(dir))) {
1387 if (ISDOT(dname) || ISDOTDOT(dname)) {
1388 continue;
1391 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1392 continue;
1395 isdir = false;
1396 if (!match || !gen_fnmatch(expression, dname)) {
1397 if (ctx->recurse) {
1398 ret = stat(path, &statbuf);
1399 if (ret == 0) {
1400 if (S_ISDIR(statbuf.st_mode)) {
1401 isdir = true;
1402 ret = file_find(ctx, list, path, expression, false);
1404 } else {
1405 d_printf("file_find: cannot stat file %s\n", path);
1408 if (ret == -1) {
1409 SAFE_FREE(path);
1410 closedir(dir);
1411 return -1;
1414 entry = malloc_p(struct file_list);
1415 if (!entry) {
1416 d_printf("Out of memory in file_find\n");
1417 closedir(dir);
1418 return -1;
1420 entry->file_path = path;
1421 entry->isdir = isdir;
1422 DLIST_ADD(*list, entry);
1423 } else {
1424 SAFE_FREE(path);
1428 closedir(dir);
1429 return 0;
1432 /****************************************************************************
1433 mput some files
1434 ****************************************************************************/
1435 static int cmd_mput(struct smbclient_context *ctx, const char **args)
1437 int i;
1439 for (i = 1; args[i]; i++) {
1440 int ret;
1441 struct file_list *temp_list;
1442 char *quest, *lname, *rname;
1444 printf("%s\n", args[i]);
1446 file_list = NULL;
1448 ret = file_find(ctx, &file_list, ".", args[i], true);
1449 if (ret) {
1450 free_file_list(file_list);
1451 continue;
1454 quest = NULL;
1455 lname = NULL;
1456 rname = NULL;
1458 for (temp_list = file_list; temp_list;
1459 temp_list = temp_list->next) {
1461 SAFE_FREE(lname);
1462 if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1463 continue;
1464 trim_string(lname, "./", "/");
1466 /* check if it's a directory */
1467 if (temp_list->isdir) {
1468 /* if (!recurse) continue; */
1470 SAFE_FREE(quest);
1471 if (asprintf(&quest, "Put directory %s? ", lname) < 0) break;
1472 if (ctx->prompt && !yesno(quest)) { /* No */
1473 /* Skip the directory */
1474 lname[strlen(lname)-1] = '/';
1475 if (!seek_list(temp_list, lname))
1476 break;
1477 } else { /* Yes */
1478 SAFE_FREE(rname);
1479 if(asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1480 dos_format(rname);
1481 if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, rname)) &&
1482 NT_STATUS_IS_ERR(do_mkdir(ctx, rname))) {
1483 DEBUG (0, ("Unable to make dir, skipping...\n"));
1484 /* Skip the directory */
1485 lname[strlen(lname)-1] = '/';
1486 if (!seek_list(temp_list, lname))
1487 break;
1490 continue;
1491 } else {
1492 SAFE_FREE(quest);
1493 if (asprintf(&quest,"Put file %s? ", lname) < 0) break;
1494 if (ctx->prompt && !yesno(quest)) /* No */
1495 continue;
1497 /* Yes */
1498 SAFE_FREE(rname);
1499 if (asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1502 dos_format(rname);
1504 do_put(ctx, rname, lname, false);
1506 free_file_list(file_list);
1507 SAFE_FREE(quest);
1508 SAFE_FREE(lname);
1509 SAFE_FREE(rname);
1512 return 0;
1516 /****************************************************************************
1517 print a file
1518 ****************************************************************************/
1519 static int cmd_print(struct smbclient_context *ctx, const char **args)
1521 char *lname, *rname;
1522 char *p;
1524 if (!args[1]) {
1525 d_printf("print <filename>\n");
1526 return 1;
1529 lname = talloc_strdup(ctx, args[1]);
1530 if (lname == NULL) {
1531 d_printf("Out of memory in cmd_print\n");
1532 return 1;
1535 if (strequal(lname, "-")) {
1536 rname = talloc_asprintf(ctx, "stdin-%d", (int)getpid());
1537 } else {
1538 p = strrchr_m(lname, '/');
1539 if (p) {
1540 rname = talloc_asprintf(ctx, "%s-%d", p + 1,
1541 (int)getpid());
1542 } else {
1543 rname = talloc_strdup(ctx, lname);
1547 if (rname == NULL) {
1548 d_printf("Out of memory in cmd_print (stdin)\n");
1549 return 1;
1552 return do_put(ctx, rname, lname, false);
1556 static int cmd_rewrite(struct smbclient_context *ctx, const char **args)
1558 d_printf("REWRITE: command not implemented (FIXME!)\n");
1560 return 0;
1563 /****************************************************************************
1564 delete some files
1565 ****************************************************************************/
1566 static int cmd_del(struct smbclient_context *ctx, const char **args)
1568 char *mask;
1570 if (!args[1]) {
1571 d_printf("del <filename>\n");
1572 return 1;
1574 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1576 if (NT_STATUS_IS_ERR(smbcli_unlink(ctx->cli->tree, mask))) {
1577 d_printf("%s deleting remote file %s\n",smbcli_errstr(ctx->cli->tree),mask);
1580 return 0;
1584 /****************************************************************************
1585 delete a whole directory tree
1586 ****************************************************************************/
1587 static int cmd_deltree(struct smbclient_context *ctx, const char **args)
1589 char *dname;
1590 int ret;
1592 if (!args[1]) {
1593 d_printf("deltree <dirname>\n");
1594 return 1;
1597 dname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1599 ret = smbcli_deltree(ctx->cli->tree, dname);
1601 if (ret == -1) {
1602 printf("Failed to delete tree %s - %s\n", dname, smbcli_errstr(ctx->cli->tree));
1603 return -1;
1606 printf("Deleted %d files in %s\n", ret, dname);
1608 return 0;
1611 typedef struct {
1612 const char *level_name;
1613 enum smb_fsinfo_level level;
1614 } fsinfo_level_t;
1616 fsinfo_level_t fsinfo_levels[] = {
1617 {"dskattr", RAW_QFS_DSKATTR},
1618 {"allocation", RAW_QFS_ALLOCATION},
1619 {"volume", RAW_QFS_VOLUME},
1620 {"volumeinfo", RAW_QFS_VOLUME_INFO},
1621 {"sizeinfo", RAW_QFS_SIZE_INFO},
1622 {"deviceinfo", RAW_QFS_DEVICE_INFO},
1623 {"attributeinfo", RAW_QFS_ATTRIBUTE_INFO},
1624 {"unixinfo", RAW_QFS_UNIX_INFO},
1625 {"volume-information", RAW_QFS_VOLUME_INFORMATION},
1626 {"size-information", RAW_QFS_SIZE_INFORMATION},
1627 {"device-information", RAW_QFS_DEVICE_INFORMATION},
1628 {"attribute-information", RAW_QFS_ATTRIBUTE_INFORMATION},
1629 {"quota-information", RAW_QFS_QUOTA_INFORMATION},
1630 {"fullsize-information", RAW_QFS_FULL_SIZE_INFORMATION},
1631 {"objectid", RAW_QFS_OBJECTID_INFORMATION},
1632 {"sector-size-info", RAW_QFS_SECTOR_SIZE_INFORMATION},
1633 {NULL, RAW_QFS_GENERIC}
1637 static int cmd_fsinfo(struct smbclient_context *ctx, const char **args)
1639 union smb_fsinfo fsinfo;
1640 NTSTATUS status;
1641 fsinfo_level_t *fsinfo_level;
1643 if (!args[1]) {
1644 d_printf("fsinfo <level>, where level is one of following:\n");
1645 fsinfo_level = fsinfo_levels;
1646 while(fsinfo_level->level_name) {
1647 d_printf("%s\n", fsinfo_level->level_name);
1648 fsinfo_level++;
1650 return 1;
1653 fsinfo_level = fsinfo_levels;
1654 while(fsinfo_level->level_name && !strequal(args[1],fsinfo_level->level_name)) {
1655 fsinfo_level++;
1658 if (!fsinfo_level->level_name) {
1659 d_printf("wrong level name!\n");
1660 return 1;
1663 fsinfo.generic.level = fsinfo_level->level;
1664 status = smb_raw_fsinfo(ctx->cli->tree, ctx, &fsinfo);
1665 if (!NT_STATUS_IS_OK(status)) {
1666 d_printf("fsinfo-level-%s - %s\n", fsinfo_level->level_name, nt_errstr(status));
1667 return 1;
1670 d_printf("fsinfo-level-%s:\n", fsinfo_level->level_name);
1671 switch(fsinfo.generic.level) {
1672 case RAW_QFS_DSKATTR:
1673 d_printf("\tunits_total: %hu\n",
1674 (unsigned short) fsinfo.dskattr.out.units_total);
1675 d_printf("\tblocks_per_unit: %hu\n",
1676 (unsigned short) fsinfo.dskattr.out.blocks_per_unit);
1677 d_printf("\tblocks_size: %hu\n",
1678 (unsigned short) fsinfo.dskattr.out.block_size);
1679 d_printf("\tunits_free: %hu\n",
1680 (unsigned short) fsinfo.dskattr.out.units_free);
1681 break;
1682 case RAW_QFS_ALLOCATION:
1683 d_printf("\tfs_id: %lu\n",
1684 (unsigned long) fsinfo.allocation.out.fs_id);
1685 d_printf("\tsectors_per_unit: %lu\n",
1686 (unsigned long) fsinfo.allocation.out.sectors_per_unit);
1687 d_printf("\ttotal_alloc_units: %lu\n",
1688 (unsigned long) fsinfo.allocation.out.total_alloc_units);
1689 d_printf("\tavail_alloc_units: %lu\n",
1690 (unsigned long) fsinfo.allocation.out.avail_alloc_units);
1691 d_printf("\tbytes_per_sector: %hu\n",
1692 (unsigned short) fsinfo.allocation.out.bytes_per_sector);
1693 break;
1694 case RAW_QFS_VOLUME:
1695 d_printf("\tserial_number: %lu\n",
1696 (unsigned long) fsinfo.volume.out.serial_number);
1697 d_printf("\tvolume_name: %s\n", fsinfo.volume.out.volume_name.s);
1698 break;
1699 case RAW_QFS_VOLUME_INFO:
1700 case RAW_QFS_VOLUME_INFORMATION:
1701 d_printf("\tcreate_time: %s\n",
1702 nt_time_string(ctx,fsinfo.volume_info.out.create_time));
1703 d_printf("\tserial_number: %lu\n",
1704 (unsigned long) fsinfo.volume_info.out.serial_number);
1705 d_printf("\tvolume_name: %s\n", fsinfo.volume_info.out.volume_name.s);
1706 break;
1707 case RAW_QFS_SIZE_INFO:
1708 case RAW_QFS_SIZE_INFORMATION:
1709 d_printf("\ttotal_alloc_units: %llu\n",
1710 (unsigned long long) fsinfo.size_info.out.total_alloc_units);
1711 d_printf("\tavail_alloc_units: %llu\n",
1712 (unsigned long long) fsinfo.size_info.out.avail_alloc_units);
1713 d_printf("\tsectors_per_unit: %lu\n",
1714 (unsigned long) fsinfo.size_info.out.sectors_per_unit);
1715 d_printf("\tbytes_per_sector: %lu\n",
1716 (unsigned long) fsinfo.size_info.out.bytes_per_sector);
1717 break;
1718 case RAW_QFS_DEVICE_INFO:
1719 case RAW_QFS_DEVICE_INFORMATION:
1720 d_printf("\tdevice_type: %lu\n",
1721 (unsigned long) fsinfo.device_info.out.device_type);
1722 d_printf("\tcharacteristics: 0x%lx\n",
1723 (unsigned long) fsinfo.device_info.out.characteristics);
1724 break;
1725 case RAW_QFS_ATTRIBUTE_INFORMATION:
1726 case RAW_QFS_ATTRIBUTE_INFO:
1727 d_printf("\tfs_attr: 0x%lx\n",
1728 (unsigned long) fsinfo.attribute_info.out.fs_attr);
1729 d_printf("\tmax_file_component_length: %lu\n",
1730 (unsigned long) fsinfo.attribute_info.out.max_file_component_length);
1731 d_printf("\tfs_type: %s\n", fsinfo.attribute_info.out.fs_type.s);
1732 break;
1733 case RAW_QFS_UNIX_INFO:
1734 d_printf("\tmajor_version: %hu\n",
1735 (unsigned short) fsinfo.unix_info.out.major_version);
1736 d_printf("\tminor_version: %hu\n",
1737 (unsigned short) fsinfo.unix_info.out.minor_version);
1738 d_printf("\tcapability: 0x%llx\n",
1739 (unsigned long long) fsinfo.unix_info.out.capability);
1740 break;
1741 case RAW_QFS_QUOTA_INFORMATION:
1742 d_printf("\tunknown[3]: [%llu,%llu,%llu]\n",
1743 (unsigned long long) fsinfo.quota_information.out.unknown[0],
1744 (unsigned long long) fsinfo.quota_information.out.unknown[1],
1745 (unsigned long long) fsinfo.quota_information.out.unknown[2]);
1746 d_printf("\tquota_soft: %llu\n",
1747 (unsigned long long) fsinfo.quota_information.out.quota_soft);
1748 d_printf("\tquota_hard: %llu\n",
1749 (unsigned long long) fsinfo.quota_information.out.quota_hard);
1750 d_printf("\tquota_flags: 0x%llx\n",
1751 (unsigned long long) fsinfo.quota_information.out.quota_flags);
1752 break;
1753 case RAW_QFS_FULL_SIZE_INFORMATION:
1754 d_printf("\ttotal_alloc_units: %llu\n",
1755 (unsigned long long) fsinfo.full_size_information.out.total_alloc_units);
1756 d_printf("\tcall_avail_alloc_units: %llu\n",
1757 (unsigned long long) fsinfo.full_size_information.out.call_avail_alloc_units);
1758 d_printf("\tactual_avail_alloc_units: %llu\n",
1759 (unsigned long long) fsinfo.full_size_information.out.actual_avail_alloc_units);
1760 d_printf("\tsectors_per_unit: %lu\n",
1761 (unsigned long) fsinfo.full_size_information.out.sectors_per_unit);
1762 d_printf("\tbytes_per_sector: %lu\n",
1763 (unsigned long) fsinfo.full_size_information.out.bytes_per_sector);
1764 break;
1765 case RAW_QFS_OBJECTID_INFORMATION:
1766 d_printf("\tGUID: %s\n",
1767 GUID_string(ctx,&fsinfo.objectid_information.out.guid));
1768 d_printf("\tunknown[6]: [%llu,%llu,%llu,%llu,%llu,%llu]\n",
1769 (unsigned long long) fsinfo.objectid_information.out.unknown[0],
1770 (unsigned long long) fsinfo.objectid_information.out.unknown[1],
1771 (unsigned long long) fsinfo.objectid_information.out.unknown[2],
1772 (unsigned long long) fsinfo.objectid_information.out.unknown[3],
1773 (unsigned long long) fsinfo.objectid_information.out.unknown[4],
1774 (unsigned long long) fsinfo.objectid_information.out.unknown[5] );
1775 break;
1776 case RAW_QFS_SECTOR_SIZE_INFORMATION:
1777 d_printf("\tlogical_bytes_per_sector: %u\n",
1778 (unsigned)fsinfo.sector_size_info.out.logical_bytes_per_sector);
1779 d_printf("\tphys_bytes_per_sector_atomic: %u\n",
1780 (unsigned)fsinfo.sector_size_info.out.phys_bytes_per_sector_atomic);
1781 d_printf("\tphys_bytes_per_sector_perf: %u\n",
1782 (unsigned)fsinfo.sector_size_info.out.phys_bytes_per_sector_perf);
1783 d_printf("\tfs_effective_phys_bytes_per_sector_atomic: %u\n",
1784 (unsigned)fsinfo.sector_size_info.out.fs_effective_phys_bytes_per_sector_atomic);
1785 d_printf("\tflags: 0x%x\n",
1786 (unsigned)fsinfo.sector_size_info.out.flags);
1787 d_printf("\tbyte_off_sector_align: %u\n",
1788 (unsigned)fsinfo.sector_size_info.out.byte_off_sector_align);
1789 d_printf("\tbyte_off_partition_align: %u\n",
1790 (unsigned)fsinfo.sector_size_info.out.byte_off_partition_align);
1791 break;
1792 case RAW_QFS_GENERIC:
1793 d_printf("\twrong level returned\n");
1794 break;
1797 return 0;
1800 /****************************************************************************
1801 show as much information as possible about a file
1802 ****************************************************************************/
1803 static int cmd_allinfo(struct smbclient_context *ctx, const char **args)
1805 char *fname;
1806 union smb_fileinfo finfo;
1807 NTSTATUS status;
1808 int fnum;
1810 if (!args[1]) {
1811 d_printf("allinfo <filename>\n");
1812 return 1;
1814 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1816 /* first a ALL_INFO QPATHINFO */
1817 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1818 finfo.generic.in.file.path = fname;
1819 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1820 if (!NT_STATUS_IS_OK(status)) {
1821 d_printf("%s - %s\n", fname, nt_errstr(status));
1822 return 1;
1825 d_printf("\tcreate_time: %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1826 d_printf("\taccess_time: %s\n", nt_time_string(ctx, finfo.all_info.out.access_time));
1827 d_printf("\twrite_time: %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1828 d_printf("\tchange_time: %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1829 d_printf("\tattrib: 0x%x\n", finfo.all_info.out.attrib);
1830 d_printf("\talloc_size: %lu\n", (unsigned long)finfo.all_info.out.alloc_size);
1831 d_printf("\tsize: %lu\n", (unsigned long)finfo.all_info.out.size);
1832 d_printf("\tnlink: %u\n", finfo.all_info.out.nlink);
1833 d_printf("\tdelete_pending: %u\n", finfo.all_info.out.delete_pending);
1834 d_printf("\tdirectory: %u\n", finfo.all_info.out.directory);
1835 d_printf("\tea_size: %u\n", finfo.all_info.out.ea_size);
1836 d_printf("\tfname: '%s'\n", finfo.all_info.out.fname.s);
1838 /* 8.3 name if any */
1839 finfo.generic.level = RAW_FILEINFO_ALT_NAME_INFO;
1840 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1841 if (NT_STATUS_IS_OK(status)) {
1842 d_printf("\talt_name: %s\n", finfo.alt_name_info.out.fname.s);
1845 /* file_id if available */
1846 finfo.generic.level = RAW_FILEINFO_INTERNAL_INFORMATION;
1847 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1848 if (NT_STATUS_IS_OK(status)) {
1849 d_printf("\tfile_id %.0f\n",
1850 (double)finfo.internal_information.out.file_id);
1853 /* the EAs, if any */
1854 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1855 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1856 if (NT_STATUS_IS_OK(status)) {
1857 int i;
1858 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1859 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1860 finfo.all_eas.out.eas[i].flags,
1861 (int)finfo.all_eas.out.eas[i].value.length,
1862 finfo.all_eas.out.eas[i].name.s);
1866 /* streams, if available */
1867 finfo.generic.level = RAW_FILEINFO_STREAM_INFO;
1868 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1869 if (NT_STATUS_IS_OK(status)) {
1870 int i;
1871 for (i=0;i<finfo.stream_info.out.num_streams;i++) {
1872 d_printf("\tstream %d:\n", i);
1873 d_printf("\t\tsize %ld\n",
1874 (long)finfo.stream_info.out.streams[i].size);
1875 d_printf("\t\talloc size %ld\n",
1876 (long)finfo.stream_info.out.streams[i].alloc_size);
1877 d_printf("\t\tname %s\n", finfo.stream_info.out.streams[i].stream_name.s);
1881 /* dev/inode if available */
1882 finfo.generic.level = RAW_FILEINFO_COMPRESSION_INFORMATION;
1883 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1884 if (NT_STATUS_IS_OK(status)) {
1885 d_printf("\tcompressed size %ld\n", (long)finfo.compression_info.out.compressed_size);
1886 d_printf("\tformat %ld\n", (long)finfo.compression_info.out.format);
1887 d_printf("\tunit_shift %ld\n", (long)finfo.compression_info.out.unit_shift);
1888 d_printf("\tchunk_shift %ld\n", (long)finfo.compression_info.out.chunk_shift);
1889 d_printf("\tcluster_shift %ld\n", (long)finfo.compression_info.out.cluster_shift);
1892 /* shadow copies if available */
1893 fnum = smbcli_open(ctx->cli->tree, fname, O_RDONLY, DENY_NONE);
1894 if (fnum != -1) {
1895 struct smb_shadow_copy info;
1896 int i;
1897 info.in.file.fnum = fnum;
1898 info.in.max_data = ~0;
1899 status = smb_raw_shadow_data(ctx->cli->tree, ctx, &info);
1900 if (NT_STATUS_IS_OK(status)) {
1901 d_printf("\tshadow_copy: %u volumes %u names\n",
1902 info.out.num_volumes, info.out.num_names);
1903 for (i=0;i<info.out.num_names;i++) {
1904 d_printf("\t%s\n", info.out.names[i]);
1905 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1906 finfo.generic.in.file.path = talloc_asprintf(ctx, "%s%s",
1907 info.out.names[i], fname);
1908 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1909 if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_PATH_NOT_FOUND) ||
1910 NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
1911 continue;
1913 if (!NT_STATUS_IS_OK(status)) {
1914 d_printf("%s - %s\n", finfo.generic.in.file.path,
1915 nt_errstr(status));
1916 return 1;
1919 d_printf("\t\tcreate_time: %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1920 d_printf("\t\twrite_time: %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1921 d_printf("\t\tchange_time: %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1922 d_printf("\t\tsize: %lu\n", (unsigned long)finfo.all_info.out.size);
1927 return 0;
1931 /****************************************************************************
1932 shows EA contents
1933 ****************************************************************************/
1934 static int cmd_eainfo(struct smbclient_context *ctx, const char **args)
1936 char *fname;
1937 union smb_fileinfo finfo;
1938 NTSTATUS status;
1939 int i;
1941 if (!args[1]) {
1942 d_printf("eainfo <filename>\n");
1943 return 1;
1945 fname = talloc_strdup(ctx, args[1]);
1947 finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1948 finfo.generic.in.file.path = fname;
1949 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1951 if (!NT_STATUS_IS_OK(status)) {
1952 d_printf("RAW_FILEINFO_ALL_EAS - %s\n", nt_errstr(status));
1953 return 1;
1956 d_printf("%s has %d EAs\n", fname, finfo.all_eas.out.num_eas);
1958 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1959 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1960 finfo.all_eas.out.eas[i].flags,
1961 (int)finfo.all_eas.out.eas[i].value.length,
1962 finfo.all_eas.out.eas[i].name.s);
1963 fflush(stdout);
1964 dump_data(0,
1965 finfo.all_eas.out.eas[i].value.data,
1966 finfo.all_eas.out.eas[i].value.length);
1969 return 0;
1973 /****************************************************************************
1974 show any ACL on a file
1975 ****************************************************************************/
1976 static int cmd_acl(struct smbclient_context *ctx, const char **args)
1978 char *fname;
1979 union smb_fileinfo query;
1980 NTSTATUS status;
1981 int fnum;
1983 if (!args[1]) {
1984 d_printf("acl <filename>\n");
1985 return 1;
1987 fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1989 fnum = smbcli_nt_create_full(ctx->cli->tree, fname, 0,
1990 SEC_STD_READ_CONTROL,
1992 NTCREATEX_SHARE_ACCESS_DELETE|
1993 NTCREATEX_SHARE_ACCESS_READ|
1994 NTCREATEX_SHARE_ACCESS_WRITE,
1995 NTCREATEX_DISP_OPEN,
1996 0, 0);
1997 if (fnum == -1) {
1998 d_printf("%s - %s\n", fname, smbcli_errstr(ctx->cli->tree));
1999 return -1;
2002 query.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
2003 query.query_secdesc.in.file.fnum = fnum;
2004 query.query_secdesc.in.secinfo_flags = 0x7;
2006 status = smb_raw_fileinfo(ctx->cli->tree, ctx, &query);
2007 if (!NT_STATUS_IS_OK(status)) {
2008 d_printf("%s - %s\n", fname, nt_errstr(status));
2009 return 1;
2012 NDR_PRINT_DEBUG(security_descriptor, query.query_secdesc.out.sd);
2014 return 0;
2017 /****************************************************************************
2018 lookup a name or sid
2019 ****************************************************************************/
2020 static int cmd_lookup(struct smbclient_context *ctx, const char **args)
2022 NTSTATUS status;
2023 struct dom_sid *sid;
2025 if (!args[1]) {
2026 d_printf("lookup <sid|name>\n");
2027 return 1;
2030 sid = dom_sid_parse_talloc(ctx, args[1]);
2031 if (sid == NULL) {
2032 const char *sidstr;
2033 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sidstr);
2034 if (!NT_STATUS_IS_OK(status)) {
2035 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2036 return 1;
2039 d_printf("%s\n", sidstr);
2040 } else {
2041 const char *name;
2042 status = smblsa_lookup_sid(ctx->cli, args[1], ctx, &name);
2043 if (!NT_STATUS_IS_OK(status)) {
2044 d_printf("lsa_LookupSids - %s\n", nt_errstr(status));
2045 return 1;
2048 d_printf("%s\n", name);
2051 return 0;
2054 /****************************************************************************
2055 show privileges for a user
2056 ****************************************************************************/
2057 static int cmd_privileges(struct smbclient_context *ctx, const char **args)
2059 NTSTATUS status;
2060 struct dom_sid *sid;
2061 struct lsa_RightSet rights;
2062 unsigned i;
2064 if (!args[1]) {
2065 d_printf("privileges <sid|name>\n");
2066 return 1;
2069 sid = dom_sid_parse_talloc(ctx, args[1]);
2070 if (sid == NULL) {
2071 const char *sid_str;
2072 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2073 if (!NT_STATUS_IS_OK(status)) {
2074 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2075 return 1;
2077 sid = dom_sid_parse_talloc(ctx, sid_str);
2080 status = smblsa_sid_privileges(ctx->cli, sid, ctx, &rights);
2081 if (!NT_STATUS_IS_OK(status)) {
2082 d_printf("lsa_EnumAccountRights - %s\n", nt_errstr(status));
2083 return 1;
2086 for (i=0;i<rights.count;i++) {
2087 d_printf("\t%s\n", rights.names[i].string);
2090 return 0;
2094 /****************************************************************************
2095 add privileges for a user
2096 ****************************************************************************/
2097 static int cmd_addprivileges(struct smbclient_context *ctx, const char **args)
2099 NTSTATUS status;
2100 struct dom_sid *sid;
2101 struct lsa_RightSet rights;
2102 int i;
2104 if (!args[1]) {
2105 d_printf("addprivileges <sid|name> <privilege...>\n");
2106 return 1;
2109 sid = dom_sid_parse_talloc(ctx, args[1]);
2110 if (sid == NULL) {
2111 const char *sid_str;
2112 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2113 if (!NT_STATUS_IS_OK(status)) {
2114 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2115 return 1;
2117 sid = dom_sid_parse_talloc(ctx, sid_str);
2120 ZERO_STRUCT(rights);
2121 for (i = 2; args[i]; i++) {
2122 rights.names = talloc_realloc(ctx, rights.names,
2123 struct lsa_StringLarge, rights.count+1);
2124 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2125 rights.count++;
2129 status = smblsa_sid_add_privileges(ctx->cli, sid, ctx, &rights);
2130 if (!NT_STATUS_IS_OK(status)) {
2131 d_printf("lsa_AddAccountRights - %s\n", nt_errstr(status));
2132 return 1;
2135 return 0;
2138 /****************************************************************************
2139 delete privileges for a user
2140 ****************************************************************************/
2141 static int cmd_delprivileges(struct smbclient_context *ctx, const char **args)
2143 NTSTATUS status;
2144 struct dom_sid *sid;
2145 struct lsa_RightSet rights;
2146 int i;
2148 if (!args[1]) {
2149 d_printf("delprivileges <sid|name> <privilege...>\n");
2150 return 1;
2153 sid = dom_sid_parse_talloc(ctx, args[1]);
2154 if (sid == NULL) {
2155 const char *sid_str;
2156 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2157 if (!NT_STATUS_IS_OK(status)) {
2158 d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2159 return 1;
2161 sid = dom_sid_parse_talloc(ctx, sid_str);
2164 ZERO_STRUCT(rights);
2165 for (i = 2; args[i]; i++) {
2166 rights.names = talloc_realloc(ctx, rights.names,
2167 struct lsa_StringLarge, rights.count+1);
2168 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2169 rights.count++;
2173 status = smblsa_sid_del_privileges(ctx->cli, sid, ctx, &rights);
2174 if (!NT_STATUS_IS_OK(status)) {
2175 d_printf("lsa_RemoveAccountRights - %s\n", nt_errstr(status));
2176 return 1;
2179 return 0;
2183 /****************************************************************************
2184 open a file
2185 ****************************************************************************/
2186 static int cmd_open(struct smbclient_context *ctx, const char **args)
2188 char *filename;
2189 union smb_open io;
2190 NTSTATUS status;
2191 TALLOC_CTX *tmp_ctx;
2193 if (!args[1]) {
2194 d_printf("open <filename>\n");
2195 return 1;
2197 tmp_ctx = talloc_new(ctx);
2199 filename = talloc_asprintf(tmp_ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2201 io.generic.level = RAW_OPEN_NTCREATEX;
2202 io.ntcreatex.in.root_fid.fnum = 0;
2203 io.ntcreatex.in.flags = 0;
2204 io.ntcreatex.in.access_mask = SEC_RIGHTS_FILE_ALL;
2205 io.ntcreatex.in.create_options = 0;
2206 io.ntcreatex.in.file_attr = FILE_ATTRIBUTE_NORMAL;
2207 io.ntcreatex.in.share_access = NTCREATEX_SHARE_ACCESS_READ;
2208 io.ntcreatex.in.alloc_size = 0;
2209 io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN_IF;
2210 io.ntcreatex.in.impersonation = NTCREATEX_IMPERSONATION_ANONYMOUS;
2211 io.ntcreatex.in.security_flags = 0;
2212 io.ntcreatex.in.fname = filename;
2214 status = smb_raw_open(ctx->cli->tree, tmp_ctx, &io);
2215 talloc_free(tmp_ctx);
2217 if (NT_STATUS_IS_OK(status)) {
2218 d_printf("Opened file with fnum %u\n", (unsigned)io.ntcreatex.out.file.fnum);
2219 } else {
2220 d_printf("Opened failed: %s\n", nt_errstr(status));
2223 return 0;
2226 /****************************************************************************
2227 close a file
2228 ****************************************************************************/
2229 static int cmd_close(struct smbclient_context *ctx, const char **args)
2231 union smb_close io;
2232 NTSTATUS status;
2233 uint16_t fnum;
2235 if (!args[1]) {
2236 d_printf("close <fnum>\n");
2237 return 1;
2240 fnum = atoi(args[1]);
2242 ZERO_STRUCT(io);
2243 io.generic.level = RAW_CLOSE_CLOSE;
2244 io.close.in.file.fnum = fnum;
2246 status = smb_raw_close(ctx->cli->tree, &io);
2248 if (NT_STATUS_IS_OK(status)) {
2249 d_printf("Closed file OK\n");
2250 } else {
2251 d_printf("Close failed: %s\n", nt_errstr(status));
2254 return 0;
2258 /****************************************************************************
2259 remove a directory
2260 ****************************************************************************/
2261 static int cmd_rmdir(struct smbclient_context *ctx, const char **args)
2263 char *mask;
2265 if (!args[1]) {
2266 d_printf("rmdir <dirname>\n");
2267 return 1;
2269 mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2271 if (NT_STATUS_IS_ERR(smbcli_rmdir(ctx->cli->tree, mask))) {
2272 d_printf("%s removing remote directory file %s\n",
2273 smbcli_errstr(ctx->cli->tree),mask);
2276 return 0;
2279 /****************************************************************************
2280 UNIX hardlink.
2281 ****************************************************************************/
2282 static int cmd_link(struct smbclient_context *ctx, const char **args)
2284 char *src,*dest;
2286 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2287 d_printf("Server doesn't support UNIX CIFS calls.\n");
2288 return 1;
2292 if (!args[1] || !args[2]) {
2293 d_printf("link <src> <dest>\n");
2294 return 1;
2297 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2298 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2300 if (NT_STATUS_IS_ERR(smbcli_unix_hardlink(ctx->cli->tree, src, dest))) {
2301 d_printf("%s linking files (%s -> %s)\n", smbcli_errstr(ctx->cli->tree), src, dest);
2302 return 1;
2305 return 0;
2308 /****************************************************************************
2309 UNIX symlink.
2310 ****************************************************************************/
2312 static int cmd_symlink(struct smbclient_context *ctx, const char **args)
2314 char *src,*dest;
2316 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2317 d_printf("Server doesn't support UNIX CIFS calls.\n");
2318 return 1;
2321 if (!args[1] || !args[2]) {
2322 d_printf("symlink <src> <dest>\n");
2323 return 1;
2326 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2327 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2329 if (NT_STATUS_IS_ERR(smbcli_unix_symlink(ctx->cli->tree, src, dest))) {
2330 d_printf("%s symlinking files (%s -> %s)\n",
2331 smbcli_errstr(ctx->cli->tree), src, dest);
2332 return 1;
2335 return 0;
2338 /****************************************************************************
2339 UNIX chmod.
2340 ****************************************************************************/
2342 static int cmd_chmod(struct smbclient_context *ctx, const char **args)
2344 char *src;
2345 mode_t mode;
2347 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2348 d_printf("Server doesn't support UNIX CIFS calls.\n");
2349 return 1;
2352 if (!args[1] || !args[2]) {
2353 d_printf("chmod mode file\n");
2354 return 1;
2357 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2359 mode = (mode_t)strtol(args[1], NULL, 8);
2361 if (NT_STATUS_IS_ERR(smbcli_unix_chmod(ctx->cli->tree, src, mode))) {
2362 d_printf("%s chmod file %s 0%o\n",
2363 smbcli_errstr(ctx->cli->tree), src, (unsigned)mode);
2364 return 1;
2367 return 0;
2370 /****************************************************************************
2371 UNIX chown.
2372 ****************************************************************************/
2374 static int cmd_chown(struct smbclient_context *ctx, const char **args)
2376 char *src;
2377 uid_t uid;
2378 gid_t gid;
2380 if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2381 d_printf("Server doesn't support UNIX CIFS calls.\n");
2382 return 1;
2385 if (!args[1] || !args[2] || !args[3]) {
2386 d_printf("chown uid gid file\n");
2387 return 1;
2390 uid = (uid_t)atoi(args[1]);
2391 gid = (gid_t)atoi(args[2]);
2392 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[3]);
2394 if (NT_STATUS_IS_ERR(smbcli_unix_chown(ctx->cli->tree, src, uid, gid))) {
2395 d_printf("%s chown file %s uid=%d, gid=%d\n",
2396 smbcli_errstr(ctx->cli->tree), src, (int)uid, (int)gid);
2397 return 1;
2400 return 0;
2403 /****************************************************************************
2404 rename some files
2405 ****************************************************************************/
2406 static int cmd_rename(struct smbclient_context *ctx, const char **args)
2408 char *src,*dest;
2410 if (!args[1] || !args[2]) {
2411 d_printf("rename <src> <dest>\n");
2412 return 1;
2415 src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2416 dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2418 if (NT_STATUS_IS_ERR(smbcli_rename(ctx->cli->tree, src, dest))) {
2419 d_printf("%s renaming files\n",smbcli_errstr(ctx->cli->tree));
2420 return 1;
2423 return 0;
2427 /****************************************************************************
2428 toggle the prompt flag
2429 ****************************************************************************/
2430 static int cmd_prompt(struct smbclient_context *ctx, const char **args)
2432 ctx->prompt = !ctx->prompt;
2433 DEBUG(2,("prompting is now %s\n",ctx->prompt?"on":"off"));
2435 return 1;
2439 /****************************************************************************
2440 set the newer than time
2441 ****************************************************************************/
2442 static int cmd_newer(struct smbclient_context *ctx, const char **args)
2444 struct stat sbuf;
2446 if (args[1] && (stat(args[1],&sbuf) == 0)) {
2447 ctx->newer_than = sbuf.st_mtime;
2448 DEBUG(1,("Getting files newer than %s",
2449 asctime(localtime(&ctx->newer_than))));
2450 } else {
2451 ctx->newer_than = 0;
2454 if (args[1] && ctx->newer_than == 0) {
2455 d_printf("Error setting newer-than time\n");
2456 return 1;
2459 return 0;
2462 /****************************************************************************
2463 set the archive level
2464 ****************************************************************************/
2465 static int cmd_archive(struct smbclient_context *ctx, const char **args)
2467 if (args[1]) {
2468 ctx->archive_level = atoi(args[1]);
2469 } else
2470 d_printf("Archive level is %d\n",ctx->archive_level);
2472 return 0;
2475 /****************************************************************************
2476 toggle the lowercaseflag
2477 ****************************************************************************/
2478 static int cmd_lowercase(struct smbclient_context *ctx, const char **args)
2480 ctx->lowercase = !ctx->lowercase;
2481 DEBUG(2,("filename lowercasing is now %s\n",ctx->lowercase?"on":"off"));
2483 return 0;
2489 /****************************************************************************
2490 toggle the recurse flag
2491 ****************************************************************************/
2492 static int cmd_recurse(struct smbclient_context *ctx, const char **args)
2494 ctx->recurse = !ctx->recurse;
2495 DEBUG(2,("directory recursion is now %s\n",ctx->recurse?"on":"off"));
2497 return 0;
2500 /****************************************************************************
2501 toggle the translate flag
2502 ****************************************************************************/
2503 static int cmd_translate(struct smbclient_context *ctx, const char **args)
2505 ctx->translation = !ctx->translation;
2506 DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
2507 ctx->translation?"on":"off"));
2509 return 0;
2513 /****************************************************************************
2514 do a printmode command
2515 ****************************************************************************/
2516 static int cmd_printmode(struct smbclient_context *ctx, const char **args)
2518 if (args[1]) {
2519 if (strequal(args[1],"text")) {
2520 ctx->printmode = 0;
2521 } else {
2522 if (strequal(args[1],"graphics"))
2523 ctx->printmode = 1;
2524 else
2525 ctx->printmode = atoi(args[1]);
2529 switch(ctx->printmode)
2531 case 0:
2532 DEBUG(2,("the printmode is now text\n"));
2533 break;
2534 case 1:
2535 DEBUG(2,("the printmode is now graphics\n"));
2536 break;
2537 default:
2538 DEBUG(2,("the printmode is now %d\n", ctx->printmode));
2539 break;
2542 return 0;
2545 /****************************************************************************
2546 do the lcd command
2547 ****************************************************************************/
2548 static int cmd_lcd(struct smbclient_context *ctx, const char **args)
2550 char d[PATH_MAX];
2552 if (args[1]) {
2553 int ret;
2555 ret = chdir(args[1]);
2556 if (ret == -1) {
2557 d_printf("failed to chdir to dir '%s': %s\n",
2558 args[1], strerror(errno));
2559 return 1;
2563 DEBUG(2,("the local directory is now %s\n",getcwd(d, PATH_MAX)));
2565 return 0;
2568 /****************************************************************************
2569 history
2570 ****************************************************************************/
2571 static int cmd_history(struct smbclient_context *ctx, const char **args)
2573 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
2574 HIST_ENTRY **hlist;
2575 int i;
2577 hlist = history_list();
2579 for (i = 0; hlist && hlist[i]; i++) {
2580 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
2582 #else
2583 DEBUG(0,("no history without readline support\n"));
2584 #endif
2586 return 0;
2589 /****************************************************************************
2590 get a file restarting at end of local file
2591 ****************************************************************************/
2592 static int cmd_reget(struct smbclient_context *ctx, const char **args)
2594 char *local_name;
2595 char *remote_name;
2597 if (!args[1]) {
2598 d_printf("reget <filename>\n");
2599 return 1;
2601 remote_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2602 dos_clean_name(remote_name);
2604 if (args[2])
2605 local_name = talloc_strdup(ctx, args[2]);
2606 else
2607 local_name = talloc_strdup(ctx, args[1]);
2609 return do_get(ctx, remote_name, local_name, true);
2612 /****************************************************************************
2613 put a file restarting at end of local file
2614 ****************************************************************************/
2615 static int cmd_reput(struct smbclient_context *ctx, const char **args)
2617 char *local_name;
2618 char *remote_name;
2620 if (!args[1]) {
2621 d_printf("reput <filename>\n");
2622 return 1;
2624 local_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2626 if (!file_exist(local_name)) {
2627 d_printf("%s does not exist\n", local_name);
2628 return 1;
2631 if (args[2])
2632 remote_name = talloc_strdup(ctx, args[2]);
2633 else
2634 remote_name = talloc_strdup(ctx, args[1]);
2636 dos_clean_name(remote_name);
2638 return do_put(ctx, remote_name, local_name, true);
2643 return a string representing a share type
2645 static const char *share_type_str(uint32_t type)
2647 switch (type & 0xF) {
2648 case STYPE_DISKTREE:
2649 return "Disk";
2650 case STYPE_PRINTQ:
2651 return "Printer";
2652 case STYPE_DEVICE:
2653 return "Device";
2654 case STYPE_IPC:
2655 return "IPC";
2656 default:
2657 return "Unknown";
2663 display a list of shares from a level 1 share enum
2665 static void display_share_result(struct srvsvc_NetShareCtr1 *ctr1)
2667 int i;
2669 for (i=0;i<ctr1->count;i++) {
2670 struct srvsvc_NetShareInfo1 *info = ctr1->array+i;
2672 printf("\t%-15s %-10.10s %s\n",
2673 info->name,
2674 share_type_str(info->type),
2675 info->comment);
2681 /****************************************************************************
2682 try and browse available shares on a host
2683 ****************************************************************************/
2684 static bool browse_host(struct loadparm_context *lp_ctx,
2685 struct tevent_context *ev_ctx,
2686 const char *query_host)
2688 struct dcerpc_pipe *p;
2689 char *binding;
2690 NTSTATUS status;
2691 struct srvsvc_NetShareEnumAll r;
2692 struct srvsvc_NetShareInfoCtr info_ctr;
2693 uint32_t resume_handle = 0;
2694 TALLOC_CTX *mem_ctx = talloc_init("browse_host");
2695 struct srvsvc_NetShareCtr1 ctr1;
2696 uint32_t totalentries = 0;
2697 struct cli_credentials *creds = samba_cmdline_get_creds();
2699 binding = talloc_asprintf(mem_ctx, "ncacn_np:%s", query_host);
2701 status = dcerpc_pipe_connect(mem_ctx, &p, binding,
2702 &ndr_table_srvsvc,
2703 creds,
2704 ev_ctx,
2705 lp_ctx);
2706 if (!NT_STATUS_IS_OK(status)) {
2707 d_printf("Failed to connect to %s - %s\n",
2708 binding, nt_errstr(status));
2709 talloc_free(mem_ctx);
2710 return false;
2713 info_ctr.level = 1;
2714 info_ctr.ctr.ctr1 = &ctr1;
2716 r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
2717 r.in.info_ctr = &info_ctr;
2718 r.in.max_buffer = ~0;
2719 r.in.resume_handle = &resume_handle;
2720 r.out.resume_handle = &resume_handle;
2721 r.out.totalentries = &totalentries;
2722 r.out.info_ctr = &info_ctr;
2724 d_printf("\n\tSharename Type Comment\n");
2725 d_printf("\t--------- ---- -------\n");
2727 do {
2728 ZERO_STRUCT(ctr1);
2729 status = dcerpc_srvsvc_NetShareEnumAll_r(p->binding_handle, mem_ctx, &r);
2731 if (NT_STATUS_IS_OK(status) &&
2732 (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA) ||
2733 W_ERROR_IS_OK(r.out.result)) &&
2734 r.out.info_ctr->ctr.ctr1) {
2735 display_share_result(r.out.info_ctr->ctr.ctr1);
2736 resume_handle += r.out.info_ctr->ctr.ctr1->count;
2738 } while (NT_STATUS_IS_OK(status) && W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA));
2740 talloc_free(mem_ctx);
2742 if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2743 d_printf("Failed NetShareEnumAll %s - %s/%s\n",
2744 binding, nt_errstr(status), win_errstr(r.out.result));
2745 return false;
2748 return false;
2751 /****************************************************************************
2752 try and browse available connections on a host
2753 ****************************************************************************/
2754 static bool list_servers(const char *wk_grp)
2756 d_printf("REWRITE: list servers not implemented\n");
2757 return false;
2760 /* Some constants for completing filename arguments */
2762 #define COMPL_NONE 0 /* No completions */
2763 #define COMPL_REMOTE 1 /* Complete remote filename */
2764 #define COMPL_LOCAL 2 /* Complete local filename */
2766 static int cmd_help(struct smbclient_context *ctx, const char **args);
2768 /* This defines the commands supported by this client.
2769 * NOTE: The "!" must be the last one in the list because it's fn pointer
2770 * field is NULL, and NULL in that field is used in process_tok()
2771 * (below) to indicate the end of the list. crh
2773 static struct
2775 const char *name;
2776 int (*fn)(struct smbclient_context *ctx, const char **args);
2777 const char *description;
2778 char compl_args[2]; /* Completion argument info */
2779 } commands[] =
2781 {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2782 {"addprivileges",cmd_addprivileges,"<sid|name> <privilege...> add privileges for a user",{COMPL_NONE,COMPL_NONE}},
2783 {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
2784 {"acl",cmd_acl,"<file> show file ACL",{COMPL_NONE,COMPL_NONE}},
2785 {"allinfo",cmd_allinfo,"<file> show all possible info about a file",{COMPL_NONE,COMPL_NONE}},
2786 {"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}},
2787 {"cancel",cmd_rewrite,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
2788 {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
2789 {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
2790 {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
2791 {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2792 {"delprivileges",cmd_delprivileges,"<sid|name> <privilege...> remove privileges for a user",{COMPL_NONE,COMPL_NONE}},
2793 {"deltree",cmd_deltree,"<dir> delete a whole directory tree",{COMPL_REMOTE,COMPL_NONE}},
2794 {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2795 {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2796 {"eainfo",cmd_eainfo,"<file> show EA contents for a file",{COMPL_NONE,COMPL_NONE}},
2797 {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2798 {"fsinfo",cmd_fsinfo,"query file system info",{COMPL_NONE,COMPL_NONE}},
2799 {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
2800 {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2801 {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
2802 {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
2803 {"link",cmd_link,"<src> <dest> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2804 {"lookup",cmd_lookup,"<sid|name> show SID for name or name for SID",{COMPL_NONE,COMPL_NONE}},
2805 {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},
2806 {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2807 {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
2808 {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2809 {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
2810 {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2811 {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},
2812 {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
2813 {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
2814 {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
2815 {"close",cmd_close,"<fnum> close a file",{COMPL_NONE,COMPL_NONE}},
2816 {"privileges",cmd_privileges,"<user> show privileges for a user",{COMPL_NONE,COMPL_NONE}},
2817 {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
2818 {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
2819 {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},
2820 {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
2821 {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
2822 {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2823 {"queue",cmd_rewrite,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2824 {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2825 {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2826 {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},
2827 {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
2828 {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2829 {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
2830 {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2831 {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2832 {"symlink",cmd_symlink,"<src> <dest> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2833 {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2835 /* Yes, this must be here, see crh's comment above. */
2836 {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2837 {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2841 /*******************************************************************
2842 lookup a command string in the list of commands, including
2843 abbreviations
2844 ******************************************************************/
2845 static int process_tok(const char *tok)
2847 size_t i = 0, matches = 0;
2848 size_t cmd=0;
2849 size_t tok_len = strlen(tok);
2851 while (commands[i].fn != NULL) {
2852 if (strequal(commands[i].name,tok)) {
2853 matches = 1;
2854 cmd = i;
2855 break;
2856 } else if (strncasecmp(commands[i].name, tok, tok_len) == 0) {
2857 matches++;
2858 cmd = i;
2860 i++;
2863 if (matches == 0)
2864 return(-1);
2865 else if (matches == 1)
2866 return(cmd);
2867 else
2868 return(-2);
2871 /****************************************************************************
2872 help
2873 ****************************************************************************/
2874 static int cmd_help(struct smbclient_context *ctx, const char **args)
2876 int i=0,j;
2878 if (args[1]) {
2879 if ((i = process_tok(args[1])) >= 0)
2880 d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2881 } else {
2882 while (commands[i].description) {
2883 for (j=0; commands[i].description && (j<5); j++) {
2884 d_printf("%-15s",commands[i].name);
2885 i++;
2887 d_printf("\n");
2890 return 0;
2893 static int process_line(struct smbclient_context *ctx, const char *cline);
2894 /****************************************************************************
2895 process a -c command string
2896 ****************************************************************************/
2897 static int process_command_string(struct smbclient_context *ctx, const char *cmd)
2899 char **lines;
2900 int i, rc = 0;
2902 lines = str_list_make(NULL, cmd, ";");
2903 for (i = 0; lines[i]; i++) {
2904 rc |= process_line(ctx, lines[i]);
2906 talloc_free(lines);
2908 return rc;
2911 #define MAX_COMPLETIONS 100
2913 typedef struct {
2914 char *dirmask;
2915 char **matches;
2916 int count, samelen;
2917 const char *text;
2918 int len;
2919 } completion_remote_t;
2921 static void completion_remote_filter(struct clilist_file_info *f, const char *mask, void *state)
2923 completion_remote_t *info = (completion_remote_t *)state;
2925 if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (!ISDOT(f->name)) && (!ISDOTDOT(f->name))) {
2926 if ((info->dirmask[0] == 0) && !(f->attrib & FILE_ATTRIBUTE_DIRECTORY))
2927 info->matches[info->count] = strdup(f->name);
2928 else {
2929 char *tmp;
2931 if (info->dirmask[0] != 0)
2932 tmp = talloc_asprintf(NULL, "%s/%s", info->dirmask, f->name);
2933 else
2934 tmp = talloc_strdup(NULL, f->name);
2936 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2937 tmp = talloc_append_string(NULL, tmp, "/");
2938 info->matches[info->count] = tmp;
2940 if (info->matches[info->count] == NULL)
2941 return;
2942 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2943 smb_readline_ca_char(0);
2945 if (info->count == 1)
2946 info->samelen = strlen(info->matches[info->count]);
2947 else
2948 while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0)
2949 info->samelen--;
2950 info->count++;
2954 static char **remote_completion(const char *text, int len)
2956 char *dirmask;
2957 int i, ret;
2958 completion_remote_t info;
2960 info.samelen = len;
2961 info.text = text;
2962 info.len = len;
2963 info.count = 0;
2965 if (len >= PATH_MAX)
2966 return(NULL);
2968 info.matches = malloc_array_p(char *, MAX_COMPLETIONS);
2969 if (!info.matches) return NULL;
2970 info.matches[0] = NULL;
2972 for (i = len-1; i >= 0; i--)
2973 if ((text[i] == '/') || (text[i] == '\\'))
2974 break;
2975 info.text = text+i+1;
2976 info.samelen = info.len = len-i-1;
2978 if (i > 0) {
2979 info.dirmask = talloc_strndup(NULL, text, i+1);
2980 info.dirmask[i+1] = 0;
2981 ret = asprintf(&dirmask, "%s%*s*", rl_ctx->remote_cur_dir, i-1,
2982 text);
2983 } else {
2984 ret = asprintf(&dirmask, "%s*", rl_ctx->remote_cur_dir);
2986 if (ret < 0) {
2987 goto cleanup;
2990 if (smbcli_list(rl_ctx->cli->tree, dirmask,
2991 FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN,
2992 completion_remote_filter, &info) < 0)
2993 goto cleanup;
2995 if (info.count == 2)
2996 info.matches[0] = strdup(info.matches[1]);
2997 else {
2998 info.matches[0] = malloc_array_p(char, info.samelen+1);
2999 if (!info.matches[0])
3000 goto cleanup;
3001 strncpy(info.matches[0], info.matches[1], info.samelen);
3002 info.matches[0][info.samelen] = 0;
3004 info.matches[info.count] = NULL;
3005 return info.matches;
3007 cleanup:
3008 for (i = 0; i < info.count; i++)
3009 free(info.matches[i]);
3010 free(info.matches);
3011 return NULL;
3014 static char **completion_fn(const char *text, int start, int end)
3016 smb_readline_ca_char(' ');
3018 if (start) {
3019 const char *buf, *sp;
3020 int i;
3021 char compl_type;
3023 buf = smb_readline_get_line_buffer();
3024 if (buf == NULL)
3025 return NULL;
3027 sp = strchr(buf, ' ');
3028 if (sp == NULL)
3029 return NULL;
3031 for (i = 0; commands[i].name; i++)
3032 if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0))
3033 break;
3034 if (commands[i].name == NULL)
3035 return NULL;
3037 while (*sp == ' ')
3038 sp++;
3040 if (sp == (buf + start))
3041 compl_type = commands[i].compl_args[0];
3042 else
3043 compl_type = commands[i].compl_args[1];
3045 if (compl_type == COMPL_REMOTE)
3046 return remote_completion(text, end - start);
3047 else /* fall back to local filename completion */
3048 return NULL;
3049 } else {
3050 char **matches;
3051 size_t i, len, samelen = 0, count=1;
3053 matches = malloc_array_p(char *, MAX_COMPLETIONS);
3054 if (!matches) return NULL;
3055 matches[0] = NULL;
3057 len = strlen(text);
3058 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
3059 if (strncmp(text, commands[i].name, len) == 0) {
3060 matches[count] = strdup(commands[i].name);
3061 if (!matches[count])
3062 goto cleanup;
3063 if (count == 1)
3064 samelen = strlen(matches[count]);
3065 else
3066 while (strncmp(matches[count], matches[count-1], samelen) != 0)
3067 samelen--;
3068 count++;
3072 switch (count) {
3073 case 0: /* should never happen */
3074 case 1:
3075 goto cleanup;
3076 case 2:
3077 matches[0] = strdup(matches[1]);
3078 break;
3079 default:
3080 matches[0] = malloc_array_p(char, samelen+1);
3081 if (!matches[0])
3082 goto cleanup;
3083 strncpy(matches[0], matches[1], samelen);
3084 matches[0][samelen] = 0;
3086 matches[count] = NULL;
3087 return matches;
3089 cleanup:
3090 for (i = 0; i < count; i++) {
3091 free(matches[i]);
3093 free(matches);
3094 return NULL;
3098 /****************************************************************************
3099 make sure we swallow keepalives during idle time
3100 ****************************************************************************/
3101 static void readline_callback(void)
3103 static time_t last_t;
3104 time_t t;
3106 t = time(NULL);
3108 if (t - last_t < 5) return;
3110 last_t = t;
3112 smbcli_transport_process(rl_ctx->cli->transport);
3114 if (rl_ctx->cli->tree) {
3115 smbcli_chkpath(rl_ctx->cli->tree, "\\");
3119 static int process_line(struct smbclient_context *ctx, const char *cline)
3121 char **args;
3122 int i;
3124 /* and get the first part of the command */
3125 args = str_list_make_shell(ctx, cline, NULL);
3126 if (!args || !args[0])
3127 return 0;
3129 if ((i = process_tok(args[0])) >= 0) {
3130 const char **a = discard_const_p(const char *, args);
3131 i = commands[i].fn(ctx, a);
3132 } else if (i == -2) {
3133 d_printf("%s: command abbreviation ambiguous\n",args[0]);
3134 } else {
3135 d_printf("%s: command not found\n",args[0]);
3138 talloc_free(args);
3140 return i;
3143 /****************************************************************************
3144 process commands on stdin
3145 ****************************************************************************/
3146 static int process_stdin(struct smbclient_context *ctx)
3148 int rc = 0;
3149 while (1) {
3150 /* display a prompt */
3151 char *the_prompt = talloc_asprintf(ctx, "smb: %s> ", ctx->remote_cur_dir);
3152 char *cline = smb_readline(the_prompt, readline_callback, completion_fn);
3153 talloc_free(the_prompt);
3155 if (!cline) break;
3157 /* special case - first char is ! */
3158 if (*cline == '!') {
3159 int ret;
3160 ret = system(cline + 1);
3161 free(cline);
3162 if (ret == -1) {
3163 rc |= ret;
3165 continue;
3168 rc |= process_command_string(ctx, cline);
3169 free(cline);
3173 return rc;
3177 /*****************************************************
3178 return a connection to a server
3179 *******************************************************/
3180 static bool do_connect(struct smbclient_context *ctx,
3181 struct tevent_context *ev_ctx,
3182 struct resolve_context *resolve_ctx,
3183 const char *specified_server, const char **ports,
3184 const char *specified_share,
3185 const char *socket_options,
3186 struct cli_credentials *cred,
3187 struct smbcli_options *options,
3188 struct smbcli_session_options *session_options,
3189 struct gensec_settings *gensec_settings)
3191 NTSTATUS status;
3192 char *server, *share;
3194 rl_ctx = ctx; /* Ugly hack */
3196 if (strncmp(specified_share, "\\\\", 2) == 0 ||
3197 strncmp(specified_share, "//", 2) == 0) {
3198 bool ok;
3200 ok = smbcli_parse_unc(specified_share, ctx, &server, &share);
3201 if (!ok) {
3202 d_printf("Failed to parse UNC\n");
3203 talloc_free(ctx);
3204 return false;
3206 } else {
3207 share = talloc_strdup(ctx, specified_share);
3208 server = talloc_strdup(ctx, specified_server);
3209 if (share == NULL || server == NULL) {
3210 d_printf("Failed to allocate memory for server and share\n");
3211 talloc_free(ctx);
3212 return false;
3216 ctx->remote_cur_dir = talloc_strdup(ctx, "\\");
3217 if (ctx->remote_cur_dir == NULL) {
3218 talloc_free(ctx);
3219 return false;
3222 status = smbcli_full_connection(ctx, &ctx->cli, server, ports,
3223 share, NULL,
3224 socket_options,
3225 cred, resolve_ctx,
3226 ev_ctx, options, session_options,
3227 gensec_settings);
3228 if (!NT_STATUS_IS_OK(status)) {
3229 d_printf("Connection to \\\\%s\\%s failed - %s\n",
3230 server, share, nt_errstr(status));
3231 talloc_free(ctx);
3232 return false;
3235 return true;
3238 /****************************************************************************
3239 handle a -L query
3240 ****************************************************************************/
3241 static int do_host_query(struct loadparm_context *lp_ctx,
3242 struct tevent_context *ev_ctx,
3243 const char *query_host,
3244 const char *workgroup)
3246 browse_host(lp_ctx, ev_ctx, query_host);
3247 list_servers(workgroup);
3248 return(0);
3252 /****************************************************************************
3253 handle a message operation
3254 ****************************************************************************/
3255 static int do_message_op(const char *netbios_name, const char *desthost,
3256 const char **destports, const char *destip,
3257 int name_type,
3258 struct tevent_context *ev_ctx,
3259 struct resolve_context *resolve_ctx,
3260 struct smbcli_options *options,
3261 const char *socket_options)
3263 struct nbt_name called, calling;
3264 const char *server_name;
3265 struct smbcli_state *cli;
3266 bool ok;
3268 make_nbt_name_client(&calling, netbios_name);
3270 nbt_choose_called_name(NULL, &called, desthost, name_type);
3272 server_name = destip ? destip : desthost;
3274 cli = smbcli_state_init(NULL);
3275 if (cli == NULL) {
3276 d_printf("smbcli_state_init() failed\n");
3277 return 1;
3280 ok = smbcli_socket_connect(cli, server_name, destports,
3281 ev_ctx, resolve_ctx, options,
3282 socket_options,
3283 &calling, &called);
3284 if (!ok) {
3285 d_printf("Connection to %s failed\n", server_name);
3286 return 1;
3289 send_message(cli, desthost);
3290 talloc_free(cli);
3292 return 0;
3296 /****************************************************************************
3297 main program
3298 ****************************************************************************/
3299 int main(int argc, char *argv[])
3301 const char **const_argv = discard_const_p(const char *, argv);
3302 char *base_directory = NULL;
3303 const char *dest_ip = NULL;
3304 int opt;
3305 const char *query_host = NULL;
3306 bool message = false;
3307 char *desthost = NULL;
3308 poptContext pc;
3309 const char *service = NULL;
3310 int port = 0;
3311 char *p;
3312 int rc = 0;
3313 int name_type = 0x20;
3314 TALLOC_CTX *mem_ctx;
3315 struct tevent_context *ev_ctx;
3316 struct smbclient_context *ctx;
3317 const char *cmdstr = NULL;
3318 struct smbcli_options smb_options;
3319 struct smbcli_session_options smb_session_options;
3320 struct cli_credentials *creds = NULL;
3321 struct loadparm_context *lp_ctx = NULL;
3322 bool ok;
3324 struct poptOption long_options[] = {
3325 POPT_AUTOHELP
3328 .longName = "message",
3329 .shortName = 'M',
3330 .argInfo = POPT_ARG_STRING,
3331 .arg = NULL,
3332 .val = 'M',
3333 .descrip = "Send message",
3334 .argDescrip = "HOST",
3337 .longName = "ip-address",
3338 .shortName = 'I',
3339 .argInfo = POPT_ARG_STRING,
3340 .arg = NULL,
3341 .val = 'I',
3342 .descrip = "Use this IP to connect to",
3343 .argDescrip = "IP",
3346 .longName = "stderr",
3347 .shortName = 'E',
3348 .argInfo = POPT_ARG_NONE,
3349 .arg = NULL,
3350 .val = 'E',
3351 .descrip = "Write messages to stderr instead of stdout",
3354 .longName = "list",
3355 .shortName = 'L',
3356 .argInfo = POPT_ARG_STRING,
3357 .arg = NULL,
3358 .val = 'L',
3359 .descrip = "Get a list of shares available on a host",
3360 .argDescrip = "HOST",
3363 .longName = "directory",
3364 .shortName = 'D',
3365 .argInfo = POPT_ARG_STRING,
3366 .arg = NULL,
3367 .val = 'D',
3368 .descrip = "Start from directory",
3369 .argDescrip = "DIR",
3372 .longName = "command",
3373 .shortName = 'c',
3374 .argInfo = POPT_ARG_STRING,
3375 .arg = &cmdstr,
3376 .val = 'c',
3377 .descrip = "Execute semicolon separated commands",
3380 .longName = "send-buffer",
3381 .shortName = 'b',
3382 .argInfo = POPT_ARG_INT,
3383 .arg = NULL,
3384 .val = 'b',
3385 .descrip = "Changes the transmit/send buffer",
3386 .argDescrip = "BYTES",
3389 .longName = "port",
3390 .shortName = 'p',
3391 .argInfo = POPT_ARG_INT,
3392 .arg = &port,
3393 .val = 'p',
3394 .descrip = "Port to connect to",
3395 .argDescrip = "PORT",
3397 POPT_COMMON_SAMBA
3398 POPT_COMMON_CONNECTION
3399 POPT_COMMON_CREDENTIALS
3400 POPT_LEGACY_S4
3401 POPT_COMMON_VERSION
3402 POPT_TABLEEND
3405 mem_ctx = talloc_init("client.c/main");
3406 if (!mem_ctx) {
3407 d_printf("\nclient.c: Not enough memory\n");
3408 exit(1);
3411 ctx = talloc_zero(mem_ctx, struct smbclient_context);
3412 ctx->io_bufsize = 64512;
3414 ok = samba_cmdline_init(mem_ctx,
3415 SAMBA_CMDLINE_CONFIG_CLIENT,
3416 false /* require_smbconf */);
3417 if (!ok) {
3418 DBG_ERR("Failed to init cmdline parser!\n");
3419 TALLOC_FREE(ctx);
3420 exit(1);
3423 pc = samba_popt_get_context(getprogname(),
3424 argc,
3425 const_argv,
3426 long_options,
3428 if (pc == NULL) {
3429 DBG_ERR("Failed to setup popt context!\n");
3430 TALLOC_FREE(ctx);
3431 exit(1);
3433 poptSetOtherOptionHelp(pc, "[OPTIONS] service <password>");
3435 while ((opt = poptGetNextOpt(pc)) != -1) {
3436 switch (opt) {
3437 case 'M':
3438 /* Messages are sent to NetBIOS name type 0x3
3439 * (Messenger Service). Make sure we default
3440 * to port 139 instead of port 445. srl,crh
3442 name_type = 0x03;
3443 desthost = strdup(poptGetOptArg(pc));
3444 if( 0 == port ) port = 139;
3445 message = true;
3446 break;
3447 case 'I':
3448 dest_ip = poptGetOptArg(pc);
3449 break;
3450 case 'L':
3451 query_host = strdup(poptGetOptArg(pc));
3452 break;
3453 case 'D':
3454 base_directory = strdup(poptGetOptArg(pc));
3455 break;
3456 case 'b':
3457 ctx->io_bufsize = MAX(1, atoi(poptGetOptArg(pc)));
3458 break;
3459 case POPT_ERROR_BADOPT:
3460 fprintf(stderr, "\nInvalid option %s: %s\n\n",
3461 poptBadOption(pc, 0), poptStrerror(opt));
3462 poptPrintUsage(pc, stderr, 0);
3463 exit(1);
3467 gensec_init();
3469 if(poptPeekArg(pc)) {
3470 char *s = strdup(poptGetArg(pc));
3472 /* Convert any '/' characters in the service name to '\' characters */
3473 string_replace(s, '/','\\');
3475 service = s;
3477 if (count_chars(s,'\\') < 3) {
3478 d_printf("\n%s: Not enough '\\' characters in service\n",s);
3479 poptPrintUsage(pc, stderr, 0);
3480 exit(1);
3484 creds = samba_cmdline_get_creds();
3485 lp_ctx = samba_cmdline_get_lp_ctx();
3487 if (poptPeekArg(pc)) {
3488 cli_credentials_set_password(creds,
3489 poptGetArg(pc), CRED_SPECIFIED);
3492 if (!query_host && !service && !message) {
3493 poptPrintUsage(pc, stderr, 0);
3494 exit(1);
3497 poptFreeContext(pc);
3498 samba_cmdline_burn(argc, argv);
3500 lpcfg_smbcli_options(lp_ctx, &smb_options);
3501 lpcfg_smbcli_session_options(lp_ctx, &smb_session_options);
3503 ev_ctx = s4_event_context_init(ctx);
3505 DEBUG( 3, ( "Client started (version %s).\n", SAMBA_VERSION_STRING ) );
3507 if (query_host && (p=strchr_m(query_host,'#'))) {
3508 *p = 0;
3509 p++;
3510 sscanf(p, "%x", &name_type);
3513 if (query_host) {
3514 rc = do_host_query(lp_ctx, ev_ctx, query_host,
3515 lpcfg_workgroup(lp_ctx));
3516 return rc;
3519 if (message) {
3520 rc = do_message_op(lpcfg_netbios_name(lp_ctx), desthost,
3521 lpcfg_smb_ports(lp_ctx), dest_ip,
3522 name_type, ev_ctx,
3523 lpcfg_resolve_context(lp_ctx),
3524 &smb_options,
3525 lpcfg_socket_options(lp_ctx));
3526 return rc;
3529 if (!do_connect(ctx, ev_ctx, lpcfg_resolve_context(lp_ctx),
3530 desthost, lpcfg_smb_ports(lp_ctx), service,
3531 lpcfg_socket_options(lp_ctx),
3532 creds,
3533 &smb_options, &smb_session_options,
3534 lpcfg_gensec_settings(ctx, lp_ctx)))
3535 return 1;
3537 if (base_directory) {
3538 do_cd(ctx, base_directory);
3539 free(base_directory);
3542 if (cmdstr) {
3543 rc = process_command_string(ctx, cmdstr);
3544 } else {
3545 rc = process_stdin(ctx);
3548 free(desthost);
3549 talloc_free(mem_ctx);
3551 return rc;