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