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

Last change on this file was 740, checked in by Silvan Scherrer, 13 years ago

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