source4/client/client.c

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