source4/client/client.c

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