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