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