source3/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
   7    Copyright (C) Gerald (Jerry) Carter    2004
   8    Copyright (C) Jeremy Allison           1994-2007
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 3 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #include "includes.h"
  25 #include "system/filesys.h"
  26 #include "popt_common.h"
  27 #include "rpc_client/cli_pipe.h"
  28 #include "client/client_proto.h"
  29 #include "../librpc/gen_ndr/ndr_srvsvc_c.h"
  30 #include "../lib/util/select.h"
  31 #include "system/readline.h"
  32 #include "../libcli/smbreadline/smbreadline.h"
  33 #include "../libcli/security/security.h"
  34 #include "system/select.h"
  35 #include "libsmb/libsmb.h"
  36 #include "libsmb/clirap.h"
  37 #include "trans2.h"
  38 #include "libsmb/nmblib.h"
  39 #include "include/ntioctl.h"
  40
  41 #ifndef REGISTER
  42 #define REGISTER 0
  43 #endif
  44
  45 extern int do_smb_browse(void); /* mDNS browsing */
  46
  47 extern bool override_logfile;
  48 extern char tar_type;
  49
  50 static int port = 0;
  51 static char *service;
  52 static char *desthost;
  53 static bool grepable = false;
  54 static char *cmdstr = NULL;
  55 const char *cmd_ptr = NULL;
  56
  57 static int io_bufsize = 524288;
  58
  59 static int name_type = 0x20;
  60 static int max_protocol = PROTOCOL_NT1;
  61
  62 static int process_tok(char *tok);
  63 static int cmd_help(void);
  64
  65 #define CREATE_ACCESS_READ READ_CONTROL_ACCESS
  66
  67 /* 30 second timeout on most commands */
  68 #define CLIENT_TIMEOUT (30*1000)
  69 #define SHORT_TIMEOUT (5*1000)
  70
  71 /* value for unused fid field in trans2 secondary request */
  72 #define FID_UNUSED (0xFFFF)
  73
  74 time_t newer_than = 0;
  75 static int archive_level = 0;
  76
  77 static bool translation = false;
  78 static bool have_ip;
  79
  80 /* clitar bits insert */
  81 extern int blocksize;
  82 extern bool tar_inc;
  83 extern bool tar_reset;
  84 /* clitar bits end */
  85
  86 static bool prompt = true;
  87
  88 static bool recurse = false;
  89 static bool showacls = false;
  90 bool lowercase = false;
  91
  92 static struct sockaddr_storage dest_ss;
  93 static char dest_ss_str[INET6_ADDRSTRLEN];
  94
  95 #define SEPARATORS " \t\n\r"
  96
  97 static bool abort_mget = true;
  98
  99 /* timing globals */
 100 uint64_t get_total_size = 0;
 101 unsigned int get_total_time_ms = 0;
 102 static uint64_t put_total_size = 0;
 103 static unsigned int put_total_time_ms = 0;
 104
 105 /* totals globals */
 106 static double dir_total;
 107
 108 /* encrypted state. */
 109 static bool smb_encrypt;
 110
 111 /* root cli_state connection */
 112
 113 struct cli_state *cli;
 114
 115 static char CLI_DIRSEP_CHAR = '\\';
 116 static char CLI_DIRSEP_STR[] = { '\\', '\0' };
 117
 118 /* Authentication for client connections. */
 119 struct user_auth_info *auth_info;
 120
 121 /* Accessor functions for directory paths. */
 122 static char *fileselection;
 123 static const char *client_get_fileselection(void)
 124 {
 125         if (fileselection) {
 126                 return fileselection;
 127         }
 128         return "";
 129 }
 130
 131 static const char *client_set_fileselection(const char *new_fs)
 132 {
 133         SAFE_FREE(fileselection);
 134         if (new_fs) {
 135                 fileselection = SMB_STRDUP(new_fs);
 136         }
 137         return client_get_fileselection();
 138 }
 139
 140 static char *cwd;
 141 static const char *client_get_cwd(void)
 142 {
 143         if (cwd) {
 144                 return cwd;
 145         }
 146         return CLI_DIRSEP_STR;
 147 }
 148
 149 static const char *client_set_cwd(const char *new_cwd)
 150 {
 151         SAFE_FREE(cwd);
 152         if (new_cwd) {
 153                 cwd = SMB_STRDUP(new_cwd);
 154         }
 155         return client_get_cwd();
 156 }
 157
 158 static char *cur_dir;
 159 const char *client_get_cur_dir(void)
 160 {
 161         if (cur_dir) {
 162                 return cur_dir;
 163         }
 164         return CLI_DIRSEP_STR;
 165 }
 166
 167 const char *client_set_cur_dir(const char *newdir)
 168 {
 169         SAFE_FREE(cur_dir);
 170         if (newdir) {
 171                 cur_dir = SMB_STRDUP(newdir);
 172         }
 173         return client_get_cur_dir();
 174 }
 175
 176 /****************************************************************************
 177  Put up a yes/no prompt.
 178 ****************************************************************************/
 179
 180 static bool yesno(const char *p)
 181 {
 182         char ans[20];
 183         printf("%s",p);
 184
 185         if (!fgets(ans,sizeof(ans)-1,stdin))
 186                 return(False);
 187
 188         if (*ans == 'y' || *ans == 'Y')
 189                 return(True);
 190
 191         return(False);
 192 }
 193
 194 /****************************************************************************
 195  Write to a local file with CR/LF->LF translation if appropriate. Return the
 196  number taken from the buffer. This may not equal the number written.
 197 ****************************************************************************/
 198
 199 static int writefile(int f, char *b, int n)
 200 {
 201         int i;
 202
 203         if (!translation) {
 204                 return write(f,b,n);
 205         }
 206
 207         i = 0;
 208         while (i < n) {
 209                 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
 210                         b++;i++;
 211                 }
 212                 if (write(f, b, 1) != 1) {
 213                         break;
 214                 }
 215                 b++;
 216                 i++;
 217         }
 218
 219         return(i);
 220 }
 221
 222 /****************************************************************************
 223  Read from a file with LF->CR/LF translation if appropriate. Return the
 224  number read. read approx n bytes.
 225 ****************************************************************************/
 226
 227 static int readfile(uint8_t *b, int n, XFILE *f)
 228 {
 229         int i;
 230         int c;
 231
 232         if (!translation)
 233                 return x_fread(b,1,n,f);
 234
 235         i = 0;
 236         while (i < (n - 1) && (i < BUFFER_SIZE)) {
 237                 if ((c = x_getc(f)) == EOF) {
 238                         break;
 239                 }
 240
 241                 if (c == '\n') { /* change all LFs to CR/LF */
 242                         b[i++] = '\r';
 243                 }
 244
 245                 b[i++] = c;
 246         }
 247
 248         return(i);
 249 }
 250
 251 struct push_state {
 252         XFILE *f;
 253         SMB_OFF_T nread;
 254 };
 255
 256 static size_t push_source(uint8_t *buf, size_t n, void *priv)
 257 {
 258         struct push_state *state = (struct push_state *)priv;
 259         int result;
 260
 261         if (x_feof(state->f)) {
 262                 return 0;
 263         }
 264
 265         result = readfile(buf, n, state->f);
 266         state->nread += result;
 267         return result;
 268 }
 269
 270 /****************************************************************************
 271  Send a message.
 272 ****************************************************************************/
 273
 274 static void send_message(const char *username)
 275 {
 276         char buf[1600];
 277         NTSTATUS status;
 278         int i;
 279
 280         d_printf("Type your message, ending it with a Control-D\n");
 281
 282         i = 0;
 283         while (i<sizeof(buf)-2) {
 284                 int c = fgetc(stdin);
 285                 if (c == EOF) {
 286                         break;
 287                 }
 288                 if (c == '\n') {
 289                         buf[i++] = '\r';
 290                 }
 291                 buf[i++] = c;
 292         }
 293         buf[i] = '\0';
 294
 295         status = cli_message(cli, desthost, username, buf);
 296         if (!NT_STATUS_IS_OK(status)) {
 297                 d_fprintf(stderr, "cli_message returned %s\n",
 298                           nt_errstr(status));
 299         }
 300 }
 301
 302 /****************************************************************************
 303  Check the space on a device.
 304 ****************************************************************************/
 305
 306 static int do_dskattr(void)
 307 {
 308         int total, bsize, avail;
 309         struct cli_state *targetcli = NULL;
 310         char *targetpath = NULL;
 311         TALLOC_CTX *ctx = talloc_tos();
 312         NTSTATUS status;
 313
 314         status = cli_resolve_path(ctx, "", auth_info, cli,
 315                                   client_get_cur_dir(), &targetcli,
 316                                   &targetpath);
 317         if (!NT_STATUS_IS_OK(status)) {
 318                 d_printf("Error in dskattr: %s\n", cli_errstr(cli));
 319                 return 1;
 320         }
 321
 322         status = cli_dskattr(targetcli, &bsize, &total, &avail);
 323         if (!NT_STATUS_IS_OK(status)) {
 324                 d_printf("Error in dskattr: %s\n", nt_errstr(status));
 325                 return 1;
 326         }
 327
 328         d_printf("\n\t\t%d blocks of size %d. %d blocks available\n",
 329                  total, bsize, avail);
 330
 331         return 0;
 332 }
 333
 334 /****************************************************************************
 335  Show cd/pwd.
 336 ****************************************************************************/
 337
 338 static int cmd_pwd(void)
 339 {
 340         d_printf("Current directory is %s",service);
 341         d_printf("%s\n",client_get_cur_dir());
 342         return 0;
 343 }
 344
 345 /****************************************************************************
 346  Ensure name has correct directory separators.
 347 ****************************************************************************/
 348
 349 static void normalize_name(char *newdir)
 350 {
 351         if (!(cli->requested_posix_capabilities & CIFS_UNIX_POSIX_PATHNAMES_CAP)) {
 352                 string_replace(newdir,'/','\\');
 353         }
 354 }
 355
 356 /****************************************************************************
 357  Change directory - inner section.
 358 ****************************************************************************/
 359
 360 static int do_cd(const char *new_dir)
 361 {
 362         char *newdir = NULL;
 363         char *saved_dir = NULL;
 364         char *new_cd = NULL;
 365         char *targetpath = NULL;
 366         struct cli_state *targetcli = NULL;
 367         SMB_STRUCT_STAT sbuf;
 368         uint32 attributes;
 369         int ret = 1;
 370         TALLOC_CTX *ctx = talloc_stackframe();
 371         NTSTATUS status;
 372
 373         newdir = talloc_strdup(ctx, new_dir);
 374         if (!newdir) {
 375                 TALLOC_FREE(ctx);
 376                 return 1;
 377         }
 378
 379         normalize_name(newdir);
 380
 381         /* Save the current directory in case the new directory is invalid */
 382
 383         saved_dir = talloc_strdup(ctx, client_get_cur_dir());
 384         if (!saved_dir) {
 385                 TALLOC_FREE(ctx);
 386                 return 1;
 387         }
 388
 389         if (*newdir == CLI_DIRSEP_CHAR) {
 390                 client_set_cur_dir(newdir);
 391                 new_cd = newdir;
 392         } else {
 393                 new_cd = talloc_asprintf(ctx, "%s%s",
 394                                 client_get_cur_dir(),
 395                                 newdir);
 396                 if (!new_cd) {
 397                         goto out;
 398                 }
 399         }
 400
 401         /* Ensure cur_dir ends in a DIRSEP */
 402         if ((new_cd[0] != '\0') && (*(new_cd+strlen(new_cd)-1) != CLI_DIRSEP_CHAR)) {
 403                 new_cd = talloc_asprintf_append(new_cd, "%s", CLI_DIRSEP_STR);
 404                 if (!new_cd) {
 405                         goto out;
 406                 }
 407         }
 408         client_set_cur_dir(new_cd);
 409
 410         new_cd = clean_name(ctx, new_cd);
 411         client_set_cur_dir(new_cd);
 412
 413         status = cli_resolve_path(ctx, "", auth_info, cli, new_cd,
 414                                   &targetcli, &targetpath);
 415         if (!NT_STATUS_IS_OK(status)) {
 416                 d_printf("cd %s: %s\n", new_cd, cli_errstr(cli));
 417                 client_set_cur_dir(saved_dir);
 418                 goto out;
 419         }
 420
 421         if (strequal(targetpath,CLI_DIRSEP_STR )) {
 422                 TALLOC_FREE(ctx);
 423                 return 0;
 424         }
 425
 426         /* Use a trans2_qpathinfo to test directories for modern servers.
 427            Except Win9x doesn't support the qpathinfo_basic() call..... */
 428
 429         if (targetcli->protocol > PROTOCOL_LANMAN2 && !targetcli->win95) {
 430
 431                 status = cli_qpathinfo_basic(targetcli, targetpath, &sbuf,
 432                                              &attributes);
 433                 if (!NT_STATUS_IS_OK(status)) {
 434                         d_printf("cd %s: %s\n", new_cd, nt_errstr(status));
 435                         client_set_cur_dir(saved_dir);
 436                         goto out;
 437                 }
 438
 439                 if (!(attributes & FILE_ATTRIBUTE_DIRECTORY)) {
 440                         d_printf("cd %s: not a directory\n", new_cd);
 441                         client_set_cur_dir(saved_dir);
 442                         goto out;
 443                 }
 444         } else {
 445
 446                 targetpath = talloc_asprintf(ctx,
 447                                 "%s%s",
 448                                 targetpath,
 449                                 CLI_DIRSEP_STR );
 450                 if (!targetpath) {
 451                         client_set_cur_dir(saved_dir);
 452                         goto out;
 453                 }
 454                 targetpath = clean_name(ctx, targetpath);
 455                 if (!targetpath) {
 456                         client_set_cur_dir(saved_dir);
 457                         goto out;
 458                 }
 459
 460                 status = cli_chkpath(targetcli, targetpath);
 461                 if (!NT_STATUS_IS_OK(status)) {
 462                         d_printf("cd %s: %s\n", new_cd, nt_errstr(status));
 463                         client_set_cur_dir(saved_dir);
 464                         goto out;
 465                 }
 466         }
 467
 468         ret = 0;
 469
 470 out:
 471
 472         TALLOC_FREE(ctx);
 473         return ret;
 474 }
 475
 476 /****************************************************************************
 477  Change directory.
 478 ****************************************************************************/
 479
 480 static int cmd_cd(void)
 481 {
 482         char *buf = NULL;
 483         int rc = 0;
 484
 485         if (next_token_talloc(talloc_tos(), &cmd_ptr, &buf,NULL)) {
 486                 rc = do_cd(buf);
 487         } else {
 488                 d_printf("Current directory is %s\n",client_get_cur_dir());
 489         }
 490
 491         return rc;
 492 }
 493
 494 /****************************************************************************
 495  Change directory.
 496 ****************************************************************************/
 497
 498 static int cmd_cd_oneup(void)
 499 {
 500         return do_cd("..");
 501 }
 502
 503 /*******************************************************************
 504  Decide if a file should be operated on.
 505 ********************************************************************/
 506
 507 static bool do_this_one(struct file_info *finfo)
 508 {
 509         if (!finfo->name) {
 510                 return false;
 511         }
 512
 513         if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) {
 514                 return true;
 515         }
 516
 517         if (*client_get_fileselection() &&
 518             !mask_match(finfo->name,client_get_fileselection(),false)) {
 519                 DEBUG(3,("mask_match %s failed\n", finfo->name));
 520                 return false;
 521         }
 522
 523         if (newer_than && finfo->mtime_ts.tv_sec < newer_than) {
 524                 DEBUG(3,("newer_than %s failed\n", finfo->name));
 525                 return false;
 526         }
 527
 528         if ((archive_level==1 || archive_level==2) && !(finfo->mode & FILE_ATTRIBUTE_ARCHIVE)) {
 529                 DEBUG(3,("archive %s failed\n", finfo->name));
 530                 return false;
 531         }
 532
 533         return true;
 534 }
 535
 536 /****************************************************************************
 537  Display info about a file.
 538 ****************************************************************************/
 539
 540 static NTSTATUS display_finfo(struct cli_state *cli_state, struct file_info *finfo,
 541                           const char *dir)
 542 {
 543         time_t t;
 544         TALLOC_CTX *ctx = talloc_tos();
 545         NTSTATUS status = NT_STATUS_OK;
 546
 547         if (!do_this_one(finfo)) {
 548                 return NT_STATUS_OK;
 549         }
 550
 551         t = finfo->mtime_ts.tv_sec; /* the time is assumed to be passed as GMT */
 552         if (!showacls) {
 553                 d_printf("  %-30s%7.7s %8.0f  %s",
 554                          finfo->name,
 555                          attrib_string(talloc_tos(), finfo->mode),
 556                         (double)finfo->size,
 557                         time_to_asc(t));
 558                 dir_total += finfo->size;
 559         } else {
 560                 char *afname = NULL;
 561                 uint16_t fnum;
 562
 563                 /* skip if this is . or .. */
 564                 if ( strequal(finfo->name,"..") || strequal(finfo->name,".") )
 565                         return NT_STATUS_OK;
 566                 /* create absolute filename for cli_ntcreate() FIXME */
 567                 afname = talloc_asprintf(ctx,
 568                                         "%s%s%s",
 569                                         dir,
 570                                         CLI_DIRSEP_STR,
 571                                         finfo->name);
 572                 if (!afname) {
 573                         return NT_STATUS_NO_MEMORY;
 574                 }
 575                 /* print file meta date header */
 576                 d_printf( "FILENAME:%s\n", finfo->name);
 577                 d_printf( "MODE:%s\n", attrib_string(talloc_tos(), finfo->mode));
 578                 d_printf( "SIZE:%.0f\n", (double)finfo->size);
 579                 d_printf( "MTIME:%s", time_to_asc(t));
 580                 status = cli_ntcreate(cli_state, afname, 0,
 581                                       CREATE_ACCESS_READ, 0,
 582                                       FILE_SHARE_READ|FILE_SHARE_WRITE,
 583                                       FILE_OPEN, 0x0, 0x0, &fnum);
 584                 if (!NT_STATUS_IS_OK(status)) {
 585                         DEBUG( 0, ("display_finfo() Failed to open %s: %s\n",
 586                                    afname, nt_errstr(status)));
 587                 } else {
 588                         struct security_descriptor *sd = NULL;
 589                         sd = cli_query_secdesc(cli_state, fnum, ctx);
 590                         if (!sd) {
 591                                 DEBUG( 0, ("display_finfo() failed to "
 592                                         "get security descriptor: %s",
 593                                         cli_errstr(cli_state)));
 594                                 status = cli_nt_error(cli_state);
 595                         } else {
 596                                 display_sec_desc(sd);
 597                         }
 598                         TALLOC_FREE(sd);
 599                 }
 600                 TALLOC_FREE(afname);
 601         }
 602         return status;
 603 }
 604
 605 /****************************************************************************
 606  Accumulate size of a file.
 607 ****************************************************************************/
 608
 609 static NTSTATUS do_du(struct cli_state *cli_state, struct file_info *finfo,
 610                   const char *dir)
 611 {
 612         if (do_this_one(finfo)) {
 613                 dir_total += finfo->size;
 614         }
 615         return NT_STATUS_OK;
 616 }
 617
 618 static bool do_list_recurse;
 619 static bool do_list_dirs;
 620 static char *do_list_queue = 0;
 621 static long do_list_queue_size = 0;
 622 static long do_list_queue_start = 0;
 623 static long do_list_queue_end = 0;
 624 static NTSTATUS (*do_list_fn)(struct cli_state *cli_state, struct file_info *,
 625                           const char *dir);
 626
 627 /****************************************************************************
 628  Functions for do_list_queue.
 629 ****************************************************************************/
 630
 631 /*
 632  * The do_list_queue is a NUL-separated list of strings stored in a
 633  * char*.  Since this is a FIFO, we keep track of the beginning and
 634  * ending locations of the data in the queue.  When we overflow, we
 635  * double the size of the char*.  When the start of the data passes
 636  * the midpoint, we move everything back.  This is logically more
 637  * complex than a linked list, but easier from a memory management
 638  * angle.  In any memory error condition, do_list_queue is reset.
 639  * Functions check to ensure that do_list_queue is non-NULL before
 640  * accessing it.
 641  */
 642
 643 static void reset_do_list_queue(void)
 644 {
 645         SAFE_FREE(do_list_queue);
 646         do_list_queue_size = 0;
 647         do_list_queue_start = 0;
 648         do_list_queue_end = 0;
 649 }
 650
 651 static void init_do_list_queue(void)
 652 {
 653         reset_do_list_queue();
 654         do_list_queue_size = 1024;
 655         do_list_queue = (char *)SMB_MALLOC(do_list_queue_size);
 656         if (do_list_queue == 0) {
 657                 d_printf("malloc fail for size %d\n",
 658                          (int)do_list_queue_size);
 659                 reset_do_list_queue();
 660         } else {
 661                 memset(do_list_queue, 0, do_list_queue_size);
 662         }
 663 }
 664
 665 static void adjust_do_list_queue(void)
 666 {
 667         /*
 668          * If the starting point of the queue is more than half way through,
 669          * move everything toward the beginning.
 670          */
 671
 672         if (do_list_queue == NULL) {
 673                 DEBUG(4,("do_list_queue is empty\n"));
 674                 do_list_queue_start = do_list_queue_end = 0;
 675                 return;
 676         }
 677
 678         if (do_list_queue_start == do_list_queue_end) {
 679                 DEBUG(4,("do_list_queue is empty\n"));
 680                 do_list_queue_start = do_list_queue_end = 0;
 681                 *do_list_queue = '\0';
 682         } else if (do_list_queue_start > (do_list_queue_size / 2)) {
 683                 DEBUG(4,("sliding do_list_queue backward\n"));
 684                 memmove(do_list_queue,
 685                         do_list_queue + do_list_queue_start,
 686                         do_list_queue_end - do_list_queue_start);
 687                 do_list_queue_end -= do_list_queue_start;
 688                 do_list_queue_start = 0;
 689         }
 690 }
 691
 692 static void add_to_do_list_queue(const char *entry)
 693 {
 694         long new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
 695         while (new_end > do_list_queue_size) {
 696                 do_list_queue_size *= 2;
 697                 DEBUG(4,("enlarging do_list_queue to %d\n",
 698                          (int)do_list_queue_size));
 699                 do_list_queue = (char *)SMB_REALLOC(do_list_queue, do_list_queue_size);
 700                 if (! do_list_queue) {
 701                         d_printf("failure enlarging do_list_queue to %d bytes\n",
 702                                  (int)do_list_queue_size);
 703                         reset_do_list_queue();
 704                 } else {
 705                         memset(do_list_queue + do_list_queue_size / 2,
 706                                0, do_list_queue_size / 2);
 707                 }
 708         }
 709         if (do_list_queue) {
 710                 strlcpy_base(do_list_queue + do_list_queue_end,
 711                                  entry, do_list_queue, do_list_queue_size);
 712                 do_list_queue_end = new_end;
 713                 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
 714                          entry, (int)do_list_queue_start, (int)do_list_queue_end));
 715         }
 716 }
 717
 718 static char *do_list_queue_head(void)
 719 {
 720         return do_list_queue + do_list_queue_start;
 721 }
 722
 723 static void remove_do_list_queue_head(void)
 724 {
 725         if (do_list_queue_end > do_list_queue_start) {
 726                 do_list_queue_start += strlen(do_list_queue_head()) + 1;
 727                 adjust_do_list_queue();
 728                 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
 729                          (int)do_list_queue_start, (int)do_list_queue_end));
 730         }
 731 }
 732
 733 static int do_list_queue_empty(void)
 734 {
 735         return (! (do_list_queue && *do_list_queue));
 736 }
 737
 738 /****************************************************************************
 739  A helper for do_list.
 740 ****************************************************************************/
 741
 742 static NTSTATUS do_list_helper(const char *mntpoint, struct file_info *f,
 743                            const char *mask, void *state)
 744 {
 745         struct cli_state *cli_state = (struct cli_state *)state;
 746         TALLOC_CTX *ctx = talloc_tos();
 747         char *dir = NULL;
 748         char *dir_end = NULL;
 749         NTSTATUS status = NT_STATUS_OK;
 750
 751         /* Work out the directory. */
 752         dir = talloc_strdup(ctx, mask);
 753         if (!dir) {
 754                 return NT_STATUS_NO_MEMORY;
 755         }
 756         if ((dir_end = strrchr(dir, CLI_DIRSEP_CHAR)) != NULL) {
 757                 *dir_end = '\0';
 758         }
 759
 760         if (f->mode & FILE_ATTRIBUTE_DIRECTORY) {
 761                 if (do_list_dirs && do_this_one(f)) {
 762                         status = do_list_fn(cli_state, f, dir);
 763                         if (!NT_STATUS_IS_OK(status)) {
 764                                 return status;
 765                         }
 766                 }
 767                 if (do_list_recurse &&
 768                     f->name &&
 769                     !strequal(f->name,".") &&
 770                     !strequal(f->name,"..")) {
 771                         char *mask2 = NULL;
 772                         char *p = NULL;
 773
 774                         if (!f->name[0]) {
 775                                 d_printf("Empty dir name returned. Possible server misconfiguration.\n");
 776                                 TALLOC_FREE(dir);
 777                                 return NT_STATUS_UNSUCCESSFUL;
 778                         }
 779
 780                         mask2 = talloc_asprintf(ctx,
 781                                         "%s%s",
 782                                         mntpoint,
 783                                         mask);
 784                         if (!mask2) {
 785                                 TALLOC_FREE(dir);
 786                                 return NT_STATUS_NO_MEMORY;
 787                         }
 788                         p = strrchr_m(mask2,CLI_DIRSEP_CHAR);
 789                         if (p) {
 790                                 p[1] = 0;
 791                         } else {
 792                                 mask2[0] = '\0';
 793                         }
 794                         mask2 = talloc_asprintf_append(mask2,
 795                                         "%s%s*",
 796                                         f->name,
 797                                         CLI_DIRSEP_STR);
 798                         if (!mask2) {
 799                                 TALLOC_FREE(dir);
 800                                 return NT_STATUS_NO_MEMORY;
 801                         }
 802                         add_to_do_list_queue(mask2);
 803                         TALLOC_FREE(mask2);
 804                 }
 805                 TALLOC_FREE(dir);
 806                 return NT_STATUS_OK;
 807         }
 808
 809         if (do_this_one(f)) {
 810                 status = do_list_fn(cli_state, f, dir);
 811         }
 812         TALLOC_FREE(dir);
 813         return status;
 814 }
 815
 816 /****************************************************************************
 817  A wrapper around cli_list that adds recursion.
 818 ****************************************************************************/
 819
 820 NTSTATUS do_list(const char *mask,
 821                         uint16 attribute,
 822                         NTSTATUS (*fn)(struct cli_state *cli_state, struct file_info *,
 823                                    const char *dir),
 824                         bool rec,
 825                         bool dirs)
 826 {
 827         static int in_do_list = 0;
 828         TALLOC_CTX *ctx = talloc_tos();
 829         struct cli_state *targetcli = NULL;
 830         char *targetpath = NULL;
 831         NTSTATUS ret_status = NT_STATUS_OK;
 832         NTSTATUS status = NT_STATUS_OK;
 833
 834         if (in_do_list && rec) {
 835                 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
 836                 exit(1);
 837         }
 838
 839         in_do_list = 1;
 840
 841         do_list_recurse = rec;
 842         do_list_dirs = dirs;
 843         do_list_fn = fn;
 844
 845         if (rec) {
 846                 init_do_list_queue();
 847                 add_to_do_list_queue(mask);
 848
 849                 while (!do_list_queue_empty()) {
 850                         /*
 851                          * Need to copy head so that it doesn't become
 852                          * invalid inside the call to cli_list.  This
 853                          * would happen if the list were expanded
 854                          * during the call.
 855                          * Fix from E. Jay Berkenbilt (ejb@ql.org)
 856                          */
 857                         char *head = talloc_strdup(ctx, do_list_queue_head());
 858
 859                         if (!head) {
 860                                 return NT_STATUS_NO_MEMORY;
 861                         }
 862
 863                         /* check for dfs */
 864
 865                         status = cli_resolve_path(ctx, "", auth_info, cli,
 866                                                   head, &targetcli,
 867                                                   &targetpath);
 868                         if (!NT_STATUS_IS_OK(status)) {
 869                                 d_printf("do_list: [%s] %s\n", head, cli_errstr(cli));
 870                                 remove_do_list_queue_head();
 871                                 continue;
 872                         }
 873
 874                         status = cli_list(targetcli, targetpath, attribute,
 875                                  do_list_helper, targetcli);
 876                         if (!NT_STATUS_IS_OK(status)) {
 877                                 d_printf("%s listing %s\n",
 878                                          nt_errstr(status), targetpath);
 879                                 ret_status = status;
 880                         }
 881                         remove_do_list_queue_head();
 882                         if ((! do_list_queue_empty()) && (fn == display_finfo)) {
 883                                 char *next_file = do_list_queue_head();
 884                                 char *save_ch = 0;
 885                                 if ((strlen(next_file) >= 2) &&
 886                                     (next_file[strlen(next_file) - 1] == '*') &&
 887                                     (next_file[strlen(next_file) - 2] == CLI_DIRSEP_CHAR)) {
 888                                         save_ch = next_file +
 889                                                 strlen(next_file) - 2;
 890                                         *save_ch = '\0';
 891                                         if (showacls) {
 892                                                 /* cwd is only used if showacls is on */
 893                                                 client_set_cwd(next_file);
 894                                         }
 895                                 }
 896                                 if (!showacls) /* don't disturbe the showacls output */
 897                                         d_printf("\n%s\n",next_file);
 898                                 if (save_ch) {
 899                                         *save_ch = CLI_DIRSEP_CHAR;
 900                                 }
 901                         }
 902                         TALLOC_FREE(head);
 903                         TALLOC_FREE(targetpath);
 904                 }
 905         } else {
 906                 /* check for dfs */
 907                 status = cli_resolve_path(ctx, "", auth_info, cli, mask,
 908                                           &targetcli, &targetpath);
 909                 if (NT_STATUS_IS_OK(status)) {
 910                         status = cli_list(targetcli, targetpath, attribute,
 911                                           do_list_helper, targetcli);
 912                         if (!NT_STATUS_IS_OK(status)) {
 913                                 d_printf("%s listing %s\n",
 914                                          nt_errstr(status), targetpath);
 915                                 ret_status = status;
 916                         }
 917                         TALLOC_FREE(targetpath);
 918                 } else {
 919                         d_printf("do_list: [%s] %s\n", mask, cli_errstr(cli));
 920                         ret_status = cli_nt_error(cli);
 921                 }
 922         }
 923
 924         in_do_list = 0;
 925         reset_do_list_queue();
 926         return ret_status;
 927 }
 928
 929 /****************************************************************************
 930  Get a directory listing.
 931 ****************************************************************************/
 932
 933 static int cmd_dir(void)
 934 {
 935         TALLOC_CTX *ctx = talloc_tos();
 936         uint16 attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
 937         char *mask = NULL;
 938         char *buf = NULL;
 939         int rc = 1;
 940         NTSTATUS status;
 941
 942         dir_total = 0;
 943         mask = talloc_strdup(ctx, client_get_cur_dir());
 944         if (!mask) {
 945                 return 1;
 946         }
 947
 948         if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
 949                 normalize_name(buf);
 950                 if (*buf == CLI_DIRSEP_CHAR) {
 951                         mask = talloc_strdup(ctx, buf);
 952                 } else {
 953                         mask = talloc_asprintf_append(mask, "%s", buf);
 954                 }
 955         } else {
 956                 mask = talloc_asprintf_append(mask, "*");
 957         }
 958         if (!mask) {
 959                 return 1;
 960         }
 961
 962         if (showacls) {
 963                 /* cwd is only used if showacls is on */
 964                 client_set_cwd(client_get_cur_dir());
 965         }
 966
 967         status = do_list(mask, attribute, display_finfo, recurse, true);
 968         if (!NT_STATUS_IS_OK(status)) {
 969                 return 1;
 970         }
 971
 972         rc = do_dskattr();
 973
 974         DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
 975
 976         return rc;
 977 }
 978
 979 /****************************************************************************
 980  Get a directory listing.
 981 ****************************************************************************/
 982
 983 static int cmd_du(void)
 984 {
 985         TALLOC_CTX *ctx = talloc_tos();
 986         uint16 attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
 987         char *mask = NULL;
 988         char *buf = NULL;
 989         NTSTATUS status;
 990         int rc = 1;
 991
 992         dir_total = 0;
 993         mask = talloc_strdup(ctx, client_get_cur_dir());
 994         if (!mask) {
 995                 return 1;
 996         }
 997         if ((mask[0] != '\0') && (mask[strlen(mask)-1]!=CLI_DIRSEP_CHAR)) {
 998                 mask = talloc_asprintf_append(mask, "%s", CLI_DIRSEP_STR);
 999                 if (!mask) {
1000                         return 1;
1001                 }
1002         }
1003
1004         if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
1005                 normalize_name(buf);
1006                 if (*buf == CLI_DIRSEP_CHAR) {
1007                         mask = talloc_strdup(ctx, buf);
1008                 } else {
1009                         mask = talloc_asprintf_append(mask, "%s", buf);
1010                 }
1011         } else {
1012                 mask = talloc_strdup(ctx, "*");
1013         }
1014
1015         status = do_list(mask, attribute, do_du, recurse, true);
1016         if (!NT_STATUS_IS_OK(status)) {
1017                 return 1;
1018         }
1019
1020         rc = do_dskattr();
1021
1022         d_printf("Total number of bytes: %.0f\n", dir_total);
1023
1024         return rc;
1025 }
1026
1027 static int cmd_echo(void)
1028 {
1029         TALLOC_CTX *ctx = talloc_tos();
1030         char *num;
1031         char *data;
1032         NTSTATUS status;
1033
1034         if (!next_token_talloc(ctx, &cmd_ptr, &num, NULL)
1035             || !next_token_talloc(ctx, &cmd_ptr, &data, NULL)) {
1036                 d_printf("echo <num> <data>\n");
1037                 return 1;
1038         }
1039
1040         status = cli_echo(cli, atoi(num), data_blob_const(data, strlen(data)));
1041
1042         if (!NT_STATUS_IS_OK(status)) {
1043                 d_printf("echo failed: %s\n", nt_errstr(status));
1044                 return 1;
1045         }
1046
1047         return 0;
1048 }
1049
1050 /****************************************************************************
1051  Get a file from rname to lname
1052 ****************************************************************************/
1053
1054 static NTSTATUS writefile_sink(char *buf, size_t n, void *priv)
1055 {
1056         int *pfd = (int *)priv;
1057         if (writefile(*pfd, buf, n) == -1) {
1058                 return map_nt_error_from_unix(errno);
1059         }
1060         return NT_STATUS_OK;
1061 }
1062
1063 static int do_get(const char *rname, const char *lname_in, bool reget)
1064 {
1065         TALLOC_CTX *ctx = talloc_tos();
1066         int handle = 0;
1067         uint16_t fnum;
1068         bool newhandle = false;
1069         struct timespec tp_start;
1070         uint16 attr;
1071         SMB_OFF_T size;
1072         off_t start = 0;
1073         SMB_OFF_T nread = 0;
1074         int rc = 0;
1075         struct cli_state *targetcli = NULL;
1076         char *targetname = NULL;
1077         char *lname = NULL;
1078         NTSTATUS status;
1079
1080         lname = talloc_strdup(ctx, lname_in);
1081         if (!lname) {
1082                 return 1;
1083         }
1084
1085         if (lowercase) {
1086                 strlower_m(lname);
1087         }
1088
1089         status = cli_resolve_path(ctx, "", auth_info, cli, rname, &targetcli,
1090                                   &targetname);
1091         if (!NT_STATUS_IS_OK(status)) {
1092                 d_printf("Failed to open %s: %s\n", rname, cli_errstr(cli));
1093                 return 1;
1094         }
1095
1096         clock_gettime_mono(&tp_start);
1097
1098         status = cli_open(targetcli, targetname, O_RDONLY, DENY_NONE, &fnum);
1099         if (!NT_STATUS_IS_OK(status)) {
1100                 d_printf("%s opening remote file %s\n", nt_errstr(status),
1101                          rname);
1102                 return 1;
1103         }
1104
1105         if(!strcmp(lname,"-")) {
1106                 handle = fileno(stdout);
1107         } else {
1108                 if (reget) {
1109                         handle = sys_open(lname, O_WRONLY|O_CREAT, 0644);
1110                         if (handle >= 0) {
1111                                 start = sys_lseek(handle, 0, SEEK_END);
1112                                 if (start == -1) {
1113                                         d_printf("Error seeking local file\n");
1114                                         return 1;
1115                                 }
1116                         }
1117                 } else {
1118                         handle = sys_open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
1119                 }
1120                 newhandle = true;
1121         }
1122         if (handle < 0) {
1123                 d_printf("Error opening local file %s\n",lname);
1124                 return 1;
1125         }
1126
1127
1128         status = cli_qfileinfo_basic(targetcli, fnum, &attr, &size, NULL, NULL,
1129                                      NULL, NULL, NULL);
1130         if (!NT_STATUS_IS_OK(status)) {
1131                 status = cli_getattrE(targetcli, fnum, &attr, &size, NULL, NULL,
1132                                       NULL);
1133                 if(!NT_STATUS_IS_OK(status)) {
1134                         d_printf("getattrib: %s\n", nt_errstr(status));
1135                         return 1;
1136                 }
1137         }
1138
1139         DEBUG(1,("getting file %s of size %.0f as %s ",
1140                  rname, (double)size, lname));
1141
1142         status = cli_pull(targetcli, fnum, start, size, io_bufsize,
1143                           writefile_sink, (void *)&handle, &nread);
1144         if (!NT_STATUS_IS_OK(status)) {
1145                 d_fprintf(stderr, "parallel_read returned %s\n",
1146                           nt_errstr(status));
1147                 cli_close(targetcli, fnum);
1148                 return 1;
1149         }
1150
1151         status = cli_close(targetcli, fnum);
1152         if (!NT_STATUS_IS_OK(status)) {
1153                 d_printf("Error %s closing remote file\n", nt_errstr(status));
1154                 rc = 1;
1155         }
1156
1157         if (newhandle) {
1158                 close(handle);
1159         }
1160
1161         if (archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) {
1162                 cli_setatr(cli, rname, attr & ~(uint16)FILE_ATTRIBUTE_ARCHIVE, 0);
1163         }
1164
1165         {
1166                 struct timespec tp_end;
1167                 int this_time;
1168
1169                 clock_gettime_mono(&tp_end);
1170                 this_time = nsec_time_diff(&tp_end,&tp_start)/1000000;
1171                 get_total_time_ms += this_time;
1172                 get_total_size += nread;
1173
1174                 DEBUG(1,("(%3.1f KiloBytes/sec) (average %3.1f KiloBytes/sec)\n",
1175                          nread / (1.024*this_time + 1.0e-4),
1176                          get_total_size / (1.024*get_total_time_ms)));
1177         }
1178
1179         TALLOC_FREE(targetname);
1180         return rc;
1181 }
1182
1183 /****************************************************************************
1184  Get a file.
1185 ****************************************************************************/
1186
1187 static int cmd_get(void)
1188 {
1189         TALLOC_CTX *ctx = talloc_tos();
1190         char *lname = NULL;
1191         char *rname = NULL;
1192         char *fname = NULL;
1193
1194         rname = talloc_strdup(ctx, client_get_cur_dir());
1195         if (!rname) {
1196                 return 1;
1197         }
1198
1199         if (!next_token_talloc(ctx, &cmd_ptr,&fname,NULL)) {
1200                 d_printf("get <filename> [localname]\n");
1201                 return 1;
1202         }
1203         rname = talloc_asprintf_append(rname, "%s", fname);
1204         if (!rname) {
1205                 return 1;
1206         }
1207         rname = clean_name(ctx, rname);
1208         if (!rname) {
1209                 return 1;
1210         }
1211
1212         next_token_talloc(ctx, &cmd_ptr,&lname,NULL);
1213         if (!lname) {
1214                 lname = fname;
1215         }
1216
1217         return do_get(rname, lname, false);
1218 }
1219
1220 /****************************************************************************
1221  Do an mget operation on one file.
1222 ****************************************************************************/
1223
1224 static NTSTATUS do_mget(struct cli_state *cli_state, struct file_info *finfo,
1225                     const char *dir)
1226 {
1227         TALLOC_CTX *ctx = talloc_tos();
1228         NTSTATUS status = NT_STATUS_OK;
1229         char *rname = NULL;
1230         char *quest = NULL;
1231         char *saved_curdir = NULL;
1232         char *mget_mask = NULL;
1233         char *new_cd = NULL;
1234
1235         if (!finfo->name) {
1236                 return NT_STATUS_OK;
1237         }
1238
1239         if (strequal(finfo->name,".") || strequal(finfo->name,".."))
1240                 return NT_STATUS_OK;
1241
1242         if (abort_mget) {
1243                 d_printf("mget aborted\n");
1244                 return NT_STATUS_UNSUCCESSFUL;
1245         }
1246
1247         if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) {
1248                 if (asprintf(&quest,
1249                          "Get directory %s? ",finfo->name) < 0) {
1250                         return NT_STATUS_NO_MEMORY;
1251                 }
1252         } else {
1253                 if (asprintf(&quest,
1254                          "Get file %s? ",finfo->name) < 0) {
1255                         return NT_STATUS_NO_MEMORY;
1256                 }
1257         }
1258
1259         if (prompt && !yesno(quest)) {
1260                 SAFE_FREE(quest);
1261                 return NT_STATUS_OK;
1262         }
1263         SAFE_FREE(quest);
1264
1265         if (!(finfo->mode & FILE_ATTRIBUTE_DIRECTORY)) {
1266                 rname = talloc_asprintf(ctx,
1267                                 "%s%s",
1268                                 client_get_cur_dir(),
1269                                 finfo->name);
1270                 if (!rname) {
1271                         return NT_STATUS_NO_MEMORY;
1272                 }
1273                 do_get(rname, finfo->name, false);
1274                 TALLOC_FREE(rname);
1275                 return NT_STATUS_OK;
1276         }
1277
1278         /* handle directories */
1279         saved_curdir = talloc_strdup(ctx, client_get_cur_dir());
1280         if (!saved_curdir) {
1281                 return NT_STATUS_NO_MEMORY;
1282         }
1283
1284         new_cd = talloc_asprintf(ctx,
1285                                 "%s%s%s",
1286                                 client_get_cur_dir(),
1287                                 finfo->name,
1288                                 CLI_DIRSEP_STR);
1289         if (!new_cd) {
1290                 return NT_STATUS_NO_MEMORY;
1291         }
1292         client_set_cur_dir(new_cd);
1293
1294         string_replace(finfo->name,'\\','/');
1295         if (lowercase) {
1296                 strlower_m(finfo->name);
1297         }
1298
1299         if (!directory_exist(finfo->name) &&
1300             mkdir(finfo->name,0777) != 0) {
1301                 d_printf("failed to create directory %s\n",finfo->name);
1302                 client_set_cur_dir(saved_curdir);
1303                 return map_nt_error_from_unix(errno);
1304         }
1305
1306         if (chdir(finfo->name) != 0) {
1307                 d_printf("failed to chdir to directory %s\n",finfo->name);
1308                 client_set_cur_dir(saved_curdir);
1309                 return map_nt_error_from_unix(errno);
1310         }
1311
1312         mget_mask = talloc_asprintf(ctx,
1313                         "%s*",
1314                         client_get_cur_dir());
1315
1316         if (!mget_mask) {
1317                 return NT_STATUS_NO_MEMORY;
1318         }
1319
1320         status = do_list(mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,false, true);
1321         if (!NT_STATUS_IS_OK(status)) {
1322                 return status;
1323         }
1324
1325         if (chdir("..") == -1) {
1326                 d_printf("do_mget: failed to chdir to .. (error %s)\n",
1327                         strerror(errno) );
1328                 return map_nt_error_from_unix(errno);
1329         }
1330         client_set_cur_dir(saved_curdir);
1331         TALLOC_FREE(mget_mask);
1332         TALLOC_FREE(saved_curdir);
1333         TALLOC_FREE(new_cd);
1334         return NT_STATUS_OK;
1335 }
1336
1337 /****************************************************************************
1338  View the file using the pager.
1339 ****************************************************************************/
1340
1341 static int cmd_more(void)
1342 {
1343         TALLOC_CTX *ctx = talloc_tos();
1344         char *rname = NULL;
1345         char *fname = NULL;
1346         char *lname = NULL;
1347         char *pager_cmd = NULL;
1348         const char *pager;
1349         int fd;
1350         int rc = 0;
1351
1352         rname = talloc_strdup(ctx, client_get_cur_dir());
1353         if (!rname) {
1354                 return 1;
1355         }
1356
1357         lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir());
1358         if (!lname) {
1359                 return 1;
1360         }
1361         fd = mkstemp(lname);
1362         if (fd == -1) {
1363                 d_printf("failed to create temporary file for more\n");
1364                 return 1;
1365         }
1366         close(fd);
1367
1368         if (!next_token_talloc(ctx, &cmd_ptr,&fname,NULL)) {
1369                 d_printf("more <filename>\n");
1370                 unlink(lname);
1371                 return 1;
1372         }
1373         rname = talloc_asprintf_append(rname, "%s", fname);
1374         if (!rname) {
1375                 return 1;
1376         }
1377         rname = clean_name(ctx,rname);
1378         if (!rname) {
1379                 return 1;
1380         }
1381
1382         rc = do_get(rname, lname, false);
1383
1384         pager=getenv("PAGER");
1385
1386         pager_cmd = talloc_asprintf(ctx,
1387                                 "%s %s",
1388                                 (pager? pager:PAGER),
1389                                 lname);
1390         if (!pager_cmd) {
1391                 return 1;
1392         }
1393         if (system(pager_cmd) == -1) {
1394                 d_printf("system command '%s' returned -1\n",
1395                         pager_cmd);
1396         }
1397         unlink(lname);
1398
1399         return rc;
1400 }
1401
1402 /****************************************************************************
1403  Do a mget command.
1404 ****************************************************************************/
1405
1406 static int cmd_mget(void)
1407 {
1408         TALLOC_CTX *ctx = talloc_tos();
1409         uint16 attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
1410         char *mget_mask = NULL;
1411         char *buf = NULL;
1412         NTSTATUS status = NT_STATUS_OK;
1413
1414         if (recurse) {
1415                 attribute |= FILE_ATTRIBUTE_DIRECTORY;
1416         }
1417
1418         abort_mget = false;
1419
1420         while (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
1421
1422                 mget_mask = talloc_strdup(ctx, client_get_cur_dir());
1423                 if (!mget_mask) {
1424                         return 1;
1425                 }
1426                 if (*buf == CLI_DIRSEP_CHAR) {
1427                         mget_mask = talloc_strdup(ctx, buf);
1428                 } else {
1429                         mget_mask = talloc_asprintf_append(mget_mask,
1430                                                         "%s", buf);
1431                 }
1432                 if (!mget_mask) {
1433                         return 1;
1434                 }
1435                 status = do_list(mget_mask, attribute, do_mget, false, true);
1436                 if (!NT_STATUS_IS_OK(status)) {
1437                         return 1;
1438                 }
1439         }
1440
1441         if (mget_mask == NULL) {
1442                 d_printf("nothing to mget\n");
1443                 return 0;
1444         }
1445
1446         if (!*mget_mask) {
1447                 mget_mask = talloc_asprintf(ctx,
1448                                         "%s*",
1449                                         client_get_cur_dir());
1450                 if (!mget_mask) {
1451                         return 1;
1452                 }
1453                 status = do_list(mget_mask, attribute, do_mget, false, true);
1454                 if (!NT_STATUS_IS_OK(status)) {
1455                         return 1;
1456                 }
1457         }
1458
1459         return 0;
1460 }
1461
1462 /****************************************************************************
1463  Make a directory of name "name".
1464 ****************************************************************************/
1465
1466 static bool do_mkdir(const char *name)
1467 {
1468         TALLOC_CTX *ctx = talloc_tos();
1469         struct cli_state *targetcli;
1470         char *targetname = NULL;
1471         NTSTATUS status;
1472
1473         status = cli_resolve_path(ctx, "", auth_info, cli, name, &targetcli,
1474                                   &targetname);
1475         if (!NT_STATUS_IS_OK(status)) {
1476                 d_printf("mkdir %s: %s\n", name, cli_errstr(cli));
1477                 return false;
1478         }
1479
1480         status = cli_mkdir(targetcli, targetname);
1481         if (!NT_STATUS_IS_OK(status)) {
1482                 d_printf("%s making remote directory %s\n",
1483                          nt_errstr(status),name);
1484                 return false;
1485         }
1486
1487         return true;
1488 }
1489
1490 /****************************************************************************
1491  Show 8.3 name of a file.
1492 ****************************************************************************/
1493
1494 static bool do_altname(const char *name)
1495 {
1496         fstring altname;
1497         NTSTATUS status;
1498
1499         status = cli_qpathinfo_alt_name(cli, name, altname);
1500         if (!NT_STATUS_IS_OK(status)) {
1501                 d_printf("%s getting alt name for %s\n",
1502                          nt_errstr(status),name);
1503                 return false;
1504         }
1505         d_printf("%s\n", altname);
1506
1507         return true;
1508 }
1509
1510 /****************************************************************************
1511  Exit client.
1512 ****************************************************************************/
1513
1514 static int cmd_quit(void)
1515 {
1516         cli_shutdown(cli);
1517         exit(0);
1518         /* NOTREACHED */
1519         return 0;
1520 }
1521
1522 /****************************************************************************
1523  Make a directory.
1524 ****************************************************************************/
1525
1526 static int cmd_mkdir(void)
1527 {
1528         TALLOC_CTX *ctx = talloc_tos();
1529         char *mask = NULL;
1530         char *buf = NULL;
1531         NTSTATUS status;
1532
1533         mask = talloc_strdup(ctx, client_get_cur_dir());
1534         if (!mask) {
1535                 return 1;
1536         }
1537
1538         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
1539                 if (!recurse) {
1540                         d_printf("mkdir <dirname>\n");
1541                 }
1542                 return 1;
1543         }
1544         mask = talloc_asprintf_append(mask, "%s", buf);
1545         if (!mask) {
1546                 return 1;
1547         }
1548
1549         if (recurse) {
1550                 char *ddir = NULL;
1551                 char *ddir2 = NULL;
1552                 struct cli_state *targetcli;
1553                 char *targetname = NULL;
1554                 char *p = NULL;
1555                 char *saveptr;
1556
1557                 ddir2 = talloc_strdup(ctx, "");
1558                 if (!ddir2) {
1559                         return 1;
1560                 }
1561
1562                 status = cli_resolve_path(ctx, "", auth_info, cli, mask,
1563                                           &targetcli, &targetname);
1564                 if (!NT_STATUS_IS_OK(status)) {
1565                         return 1;
1566                 }
1567
1568                 ddir = talloc_strdup(ctx, targetname);
1569                 if (!ddir) {
1570                         return 1;
1571                 }
1572                 trim_char(ddir,'.','\0');
1573                 p = strtok_r(ddir, "/\\", &saveptr);
1574                 while (p) {
1575                         ddir2 = talloc_asprintf_append(ddir2, "%s", p);
1576                         if (!ddir2) {
1577                                 return 1;
1578                         }
1579                         if (!NT_STATUS_IS_OK(cli_chkpath(targetcli, ddir2))) {
1580                                 do_mkdir(ddir2);
1581                         }
1582                         ddir2 = talloc_asprintf_append(ddir2, "%s", CLI_DIRSEP_STR);
1583                         if (!ddir2) {
1584                                 return 1;
1585                         }
1586                         p = strtok_r(NULL, "/\\", &saveptr);
1587                 }
1588         } else {
1589                 do_mkdir(mask);
1590         }
1591
1592         return 0;
1593 }
1594
1595 /****************************************************************************
1596  Show alt name.
1597 ****************************************************************************/
1598
1599 static int cmd_altname(void)
1600 {
1601         TALLOC_CTX *ctx = talloc_tos();
1602         char *name;
1603         char *buf;
1604
1605         name = talloc_strdup(ctx, client_get_cur_dir());
1606         if (!name) {
1607                 return 1;
1608         }
1609
1610         if (!next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) {
1611                 d_printf("altname <file>\n");
1612                 return 1;
1613         }
1614         name = talloc_asprintf_append(name, "%s", buf);
1615         if (!name) {
1616                 return 1;
1617         }
1618         do_altname(name);
1619         return 0;
1620 }
1621
1622 static char *attr_str(TALLOC_CTX *mem_ctx, uint16_t mode)
1623 {
1624         char *attrs = talloc_zero_array(mem_ctx, char, 17);
1625         int i = 0;
1626
1627         if (!(mode & FILE_ATTRIBUTE_NORMAL)) {
1628                 if (mode & FILE_ATTRIBUTE_ENCRYPTED) {
1629                         attrs[i++] = 'E';
1630                 }
1631                 if (mode & FILE_ATTRIBUTE_NONINDEXED) {
1632                         attrs[i++] = 'N';
1633                 }
1634                 if (mode & FILE_ATTRIBUTE_OFFLINE) {
1635                         attrs[i++] = 'O';
1636                 }
1637                 if (mode & FILE_ATTRIBUTE_COMPRESSED) {
1638                         attrs[i++] = 'C';
1639                 }
1640                 if (mode & FILE_ATTRIBUTE_REPARSE_POINT) {
1641                         attrs[i++] = 'r';
1642                 }
1643                 if (mode & FILE_ATTRIBUTE_SPARSE) {
1644                         attrs[i++] = 's';
1645                 }
1646                 if (mode & FILE_ATTRIBUTE_TEMPORARY) {
1647                         attrs[i++] = 'T';
1648                 }
1649                 if (mode & FILE_ATTRIBUTE_NORMAL) {
1650                         attrs[i++] = 'N';
1651                 }
1652                 if (mode & FILE_ATTRIBUTE_READONLY) {
1653                         attrs[i++] = 'R';
1654                 }
1655                 if (mode & FILE_ATTRIBUTE_HIDDEN) {
1656                         attrs[i++] = 'H';
1657                 }
1658                 if (mode & FILE_ATTRIBUTE_SYSTEM) {
1659                         attrs[i++] = 'S';
1660                 }
1661                 if (mode & FILE_ATTRIBUTE_DIRECTORY) {
1662                         attrs[i++] = 'D';
1663                 }
1664                 if (mode & FILE_ATTRIBUTE_ARCHIVE) {
1665                         attrs[i++] = 'A';
1666                 }
1667         }
1668         return attrs;
1669 }
1670
1671 /****************************************************************************
1672  Show all info we can get
1673 ****************************************************************************/
1674
1675 static int do_allinfo(const char *name)
1676 {
1677         fstring altname;
1678         struct timespec b_time, a_time, m_time, c_time;
1679         SMB_OFF_T size;
1680         uint16_t mode;
1681         SMB_INO_T ino;
1682         NTTIME tmp;
1683         uint16_t fnum;
1684         unsigned int num_streams;
1685         struct stream_struct *streams;
1686         int num_snapshots;
1687         char **snapshots;
1688         unsigned int i;
1689         NTSTATUS status;
1690
1691         status = cli_qpathinfo_alt_name(cli, name, altname);
1692         if (!NT_STATUS_IS_OK(status)) {
1693                 d_printf("%s getting alt name for %s\n", nt_errstr(status),
1694                          name);
1695                 return false;
1696         }
1697         d_printf("altname: %s\n", altname);
1698
1699         status = cli_qpathinfo2(cli, name, &b_time, &a_time, &m_time, &c_time,
1700                                 &size, &mode, &ino);
1701         if (!NT_STATUS_IS_OK(status)) {
1702                 d_printf("%s getting pathinfo for %s\n", nt_errstr(status),
1703                          name);
1704                 return false;
1705         }
1706
1707         unix_timespec_to_nt_time(&tmp, b_time);
1708         d_printf("create_time:    %s\n", nt_time_string(talloc_tos(), tmp));
1709
1710         unix_timespec_to_nt_time(&tmp, a_time);
1711         d_printf("access_time:    %s\n", nt_time_string(talloc_tos(), tmp));
1712
1713         unix_timespec_to_nt_time(&tmp, m_time);
1714         d_printf("write_time:     %s\n", nt_time_string(talloc_tos(), tmp));
1715
1716         unix_timespec_to_nt_time(&tmp, c_time);
1717         d_printf("change_time:    %s\n", nt_time_string(talloc_tos(), tmp));
1718
1719         d_printf("attributes: %s (%x)\n", attr_str(talloc_tos(), mode), mode);
1720
1721         status = cli_qpathinfo_streams(cli, name, talloc_tos(), &num_streams,
1722                                        &streams);
1723         if (!NT_STATUS_IS_OK(status)) {
1724                 d_printf("%s getting streams for %s\n", nt_errstr(status),
1725                          name);
1726                 return false;
1727         }
1728
1729         for (i=0; i<num_streams; i++) {
1730                 d_printf("stream: [%s], %lld bytes\n", streams[i].name,
1731                          (unsigned long long)streams[i].size);
1732         }
1733
1734         if (mode & FILE_ATTRIBUTE_REPARSE_POINT) {
1735                 char *subst, *print;
1736                 uint32_t flags;
1737
1738                 status = cli_readlink(cli, name, talloc_tos(), &subst, &print,
1739                                       &flags);
1740                 if (!NT_STATUS_IS_OK(status)) {
1741                         d_fprintf(stderr, "cli_readlink returned %s\n",
1742                                   nt_errstr(status));
1743                 } else {
1744                         d_printf("symlink: subst=[%s], print=[%s], flags=%x\n",
1745                                  subst, print, flags);
1746                         TALLOC_FREE(subst);
1747                         TALLOC_FREE(print);
1748                 }
1749         }
1750
1751         status = cli_ntcreate(cli, name, 0,
1752                               CREATE_ACCESS_READ, 0,
1753                               FILE_SHARE_READ|FILE_SHARE_WRITE
1754                               |FILE_SHARE_DELETE,
1755                               FILE_OPEN, 0x0, 0x0, &fnum);
1756         if (!NT_STATUS_IS_OK(status)) {
1757                 /*
1758                  * Ignore failure, it does not hurt if we can't list
1759                  * snapshots
1760                  */
1761                 return 0;
1762         }
1763         status = cli_shadow_copy_data(talloc_tos(), cli, fnum,
1764                                       true, &snapshots, &num_snapshots);
1765         if (!NT_STATUS_IS_OK(status)) {
1766                 cli_close(cli, fnum);
1767                 return 0;
1768         }
1769
1770         for (i=0; i<num_snapshots; i++) {
1771                 char *snap_name;
1772
1773                 d_printf("%s\n", snapshots[i]);
1774                 snap_name = talloc_asprintf(talloc_tos(), "%s%s",
1775                                             snapshots[i], name);
1776                 status = cli_qpathinfo2(cli, snap_name, &b_time, &a_time,
1777                                         &m_time, &c_time, &size,
1778                                         NULL, NULL);
1779                 if (!NT_STATUS_IS_OK(status)) {
1780                         d_fprintf(stderr, "pathinfo(%s) failed: %s\n",
1781                                   snap_name, nt_errstr(status));
1782                         TALLOC_FREE(snap_name);
1783                         continue;
1784                 }
1785                 unix_timespec_to_nt_time(&tmp, b_time);
1786                 d_printf("create_time:    %s\n", nt_time_string(talloc_tos(), tmp));
1787                 unix_timespec_to_nt_time(&tmp, a_time);
1788                 d_printf("access_time:    %s\n", nt_time_string(talloc_tos(), tmp));
1789                 unix_timespec_to_nt_time(&tmp, m_time);
1790                 d_printf("write_time:     %s\n", nt_time_string(talloc_tos(), tmp));
1791                 unix_timespec_to_nt_time(&tmp, c_time);
1792                 d_printf("change_time:    %s\n", nt_time_string(talloc_tos(), tmp));
1793                 d_printf("size: %d\n", (int)size);
1794         }
1795
1796         TALLOC_FREE(snapshots);
1797
1798         return 0;
1799 }
1800
1801 /****************************************************************************
1802  Show all info we can get
1803 ****************************************************************************/
1804
1805 static int cmd_allinfo(void)
1806 {
1807         TALLOC_CTX *ctx = talloc_tos();
1808         char *name;
1809         char *buf;
1810
1811         name = talloc_strdup(ctx, client_get_cur_dir());
1812         if (!name) {
1813                 return 1;
1814         }
1815
1816         if (!next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) {
1817                 d_printf("allinfo <file>\n");
1818                 return 1;
1819         }
1820         name = talloc_asprintf_append(name, "%s", buf);
1821         if (!name) {
1822                 return 1;
1823         }
1824
1825         do_allinfo(name);
1826
1827         return 0;
1828 }
1829
1830 /****************************************************************************
1831  Put a single file.
1832 ****************************************************************************/
1833
1834 static int do_put(const char *rname, const char *lname, bool reput)
1835 {
1836         TALLOC_CTX *ctx = talloc_tos();
1837         uint16_t fnum;
1838         XFILE *f;
1839         SMB_OFF_T start = 0;
1840         int rc = 0;
1841         struct timespec tp_start;
1842         struct cli_state *targetcli;
1843         char *targetname = NULL;
1844         struct push_state state;
1845         NTSTATUS status;
1846
1847         status = cli_resolve_path(ctx, "", auth_info, cli, rname,
1848                                   &targetcli, &targetname);
1849         if (!NT_STATUS_IS_OK(status)) {
1850                 d_printf("Failed to open %s: %s\n", rname, cli_errstr(cli));
1851                 return 1;
1852         }
1853
1854         clock_gettime_mono(&tp_start);
1855
1856         if (reput) {
1857                 status = cli_open(targetcli, targetname, O_RDWR|O_CREAT, DENY_NONE, &fnum);
1858                 if (NT_STATUS_IS_OK(status)) {
1859                         if (!NT_STATUS_IS_OK(status = cli_qfileinfo_basic(
1860                                                      targetcli, fnum, NULL,
1861                                                      &start, NULL, NULL,
1862                                                      NULL, NULL, NULL)) &&
1863                             !NT_STATUS_IS_OK(status = cli_getattrE(
1864                                                      targetcli, fnum, NULL,
1865                                                      &start, NULL, NULL,
1866                                                      NULL))) {
1867                                 d_printf("getattrib: %s\n", nt_errstr(status));
1868                                 return 1;
1869                         }
1870                 }
1871         } else {
1872                 status = cli_open(targetcli, targetname, O_RDWR|O_CREAT|O_TRUNC, DENY_NONE, &fnum);
1873         }
1874
1875         if (!NT_STATUS_IS_OK(status)) {
1876                 d_printf("%s opening remote file %s\n", nt_errstr(status),
1877                          rname);
1878                 return 1;
1879         }
1880
1881         /* allow files to be piped into smbclient
1882            jdblair 24.jun.98
1883
1884            Note that in this case this function will exit(0) rather
1885            than returning. */
1886         if (!strcmp(lname, "-")) {
1887                 f = x_stdin;
1888                 /* size of file is not known */
1889         } else {
1890                 f = x_fopen(lname,O_RDONLY, 0);
1891                 if (f && reput) {
1892                         if (x_tseek(f, start, SEEK_SET) == -1) {
1893                                 d_printf("Error seeking local file\n");
1894                                 x_fclose(f);
1895                                 return 1;
1896                         }
1897                 }
1898         }
1899
1900         if (!f) {
1901                 d_printf("Error opening local file %s\n",lname);
1902                 return 1;
1903         }
1904
1905         DEBUG(1,("putting file %s as %s ",lname,
1906                  rname));
1907
1908         x_setvbuf(f, NULL, X_IOFBF, io_bufsize);
1909
1910         state.f = f;
1911         state.nread = 0;
1912
1913         status = cli_push(targetcli, fnum, 0, 0, io_bufsize, push_source,
1914                           &state);
1915         if (!NT_STATUS_IS_OK(status)) {
1916                 d_fprintf(stderr, "cli_push returned %s\n", nt_errstr(status));
1917                 rc = 1;
1918         }
1919
1920         status = cli_close(targetcli, fnum);
1921         if (!NT_STATUS_IS_OK(status)) {
1922                 d_printf("%s closing remote file %s\n", nt_errstr(status),
1923                          rname);
1924                 if (f != x_stdin) {
1925                         x_fclose(f);
1926                 }
1927                 return 1;
1928         }
1929
1930         if (f != x_stdin) {
1931                 x_fclose(f);
1932         }
1933
1934         {
1935                 struct timespec tp_end;
1936                 int this_time;
1937
1938                 clock_gettime_mono(&tp_end);
1939                 this_time = nsec_time_diff(&tp_end,&tp_start)/1000000;
1940                 put_total_time_ms += this_time;
1941                 put_total_size += state.nread;
1942
1943                 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1944                          state.nread / (1.024*this_time + 1.0e-4),
1945                          put_total_size / (1.024*put_total_time_ms)));
1946         }
1947
1948         if (f == x_stdin) {
1949                 cli_shutdown(cli);
1950                 exit(0);
1951         }
1952
1953         return rc;
1954 }
1955
1956 /****************************************************************************
1957  Put a file.
1958 ****************************************************************************/
1959
1960 static int cmd_put(void)
1961 {
1962         TALLOC_CTX *ctx = talloc_tos();
1963         char *lname;
1964         char *rname;
1965         char *buf;
1966
1967         rname = talloc_strdup(ctx, client_get_cur_dir());
1968         if (!rname) {
1969                 return 1;
1970         }
1971
1972         if (!next_token_talloc(ctx, &cmd_ptr,&lname,NULL)) {
1973                 d_printf("put <filename>\n");
1974                 return 1;
1975         }
1976
1977         if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
1978                 rname = talloc_asprintf_append(rname, "%s", buf);
1979         } else {
1980                 rname = talloc_asprintf_append(rname, "%s", lname);
1981         }
1982         if (!rname) {
1983                 return 1;
1984         }
1985
1986         rname = clean_name(ctx, rname);
1987         if (!rname) {
1988                 return 1;
1989         }
1990
1991         {
1992                 SMB_STRUCT_STAT st;
1993                 /* allow '-' to represent stdin
1994                    jdblair, 24.jun.98 */
1995                 if (!file_exist_stat(lname, &st, false) &&
1996                     (strcmp(lname,"-"))) {
1997                         d_printf("%s does not exist\n",lname);
1998                         return 1;
1999                 }
2000         }
2001
2002         return do_put(rname, lname, false);
2003 }
2004
2005 /*************************************
2006  File list structure.
2007 *************************************/
2008
2009 static struct file_list {
2010         struct file_list *prev, *next;
2011         char *file_path;
2012         bool isdir;
2013 } *file_list;
2014
2015 /****************************************************************************
2016  Free a file_list structure.
2017 ****************************************************************************/
2018
2019 static void free_file_list (struct file_list *l_head)
2020 {
2021         struct file_list *list, *next;
2022
2023         for (list = l_head; list; list = next) {
2024                 next = list->next;
2025                 DLIST_REMOVE(l_head, list);
2026                 SAFE_FREE(list->file_path);
2027                 SAFE_FREE(list);
2028         }
2029 }
2030
2031 /****************************************************************************
2032  Seek in a directory/file list until you get something that doesn't start with
2033  the specified name.
2034 ****************************************************************************/
2035
2036 static bool seek_list(struct file_list *list, char *name)
2037 {
2038         while (list) {
2039                 trim_string(list->file_path,"./","\n");
2040                 if (strncmp(list->file_path, name, strlen(name)) != 0) {
2041                         return true;
2042                 }
2043                 list = list->next;
2044         }
2045
2046         return false;
2047 }
2048
2049 /****************************************************************************
2050  Set the file selection mask.
2051 ****************************************************************************/
2052
2053 static int cmd_select(void)
2054 {
2055         TALLOC_CTX *ctx = talloc_tos();
2056         char *new_fs = NULL;
2057         next_token_talloc(ctx, &cmd_ptr,&new_fs,NULL)
2058                 ;
2059         if (new_fs) {
2060                 client_set_fileselection(new_fs);
2061         } else {
2062                 client_set_fileselection("");
2063         }
2064         return 0;
2065 }
2066
2067 /****************************************************************************
2068   Recursive file matching function act as find
2069   match must be always set to true when calling this function
2070 ****************************************************************************/
2071
2072 static int file_find(struct file_list **list, const char *directory,
2073                       const char *expression, bool match)
2074 {
2075         SMB_STRUCT_DIR *dir;
2076         struct file_list *entry;
2077         struct stat statbuf;
2078         int ret;
2079         char *path;
2080         bool isdir;
2081         const char *dname;
2082
2083         dir = sys_opendir(directory);
2084         if (!dir)
2085                 return -1;
2086
2087         while ((dname = readdirname(dir))) {
2088                 if (!strcmp("..", dname))
2089                         continue;
2090                 if (!strcmp(".", dname))
2091                         continue;
2092
2093                 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
2094                         continue;
2095                 }
2096
2097                 isdir = false;
2098                 if (!match || !gen_fnmatch(expression, dname)) {
2099                         if (recurse) {
2100                                 ret = stat(path, &statbuf);
2101                                 if (ret == 0) {
2102                                         if (S_ISDIR(statbuf.st_mode)) {
2103                                                 isdir = true;
2104                                                 ret = file_find(list, path, expression, false);
2105                                         }
2106                                 } else {
2107                                         d_printf("file_find: cannot stat file %s\n", path);
2108                                 }
2109
2110                                 if (ret == -1) {
2111                                         SAFE_FREE(path);
2112                                         sys_closedir(dir);
2113                                         return -1;
2114                                 }
2115                         }
2116                         entry = SMB_MALLOC_P(struct file_list);
2117                         if (!entry) {
2118                                 d_printf("Out of memory in file_find\n");
2119                                 sys_closedir(dir);
2120                                 return -1;
2121                         }
2122                         entry->file_path = path;
2123                         entry->isdir = isdir;
2124                         DLIST_ADD(*list, entry);
2125                 } else {
2126                         SAFE_FREE(path);
2127                 }
2128         }
2129
2130         sys_closedir(dir);
2131         return 0;
2132 }
2133
2134 /****************************************************************************
2135  mput some files.
2136 ****************************************************************************/
2137
2138 static int cmd_mput(void)
2139 {
2140         TALLOC_CTX *ctx = talloc_tos();
2141         char *p = NULL;
2142
2143         while (next_token_talloc(ctx, &cmd_ptr,&p,NULL)) {
2144                 int ret;
2145                 struct file_list *temp_list;
2146                 char *quest, *lname, *rname;
2147
2148                 file_list = NULL;
2149
2150                 ret = file_find(&file_list, ".", p, true);
2151                 if (ret) {
2152                         free_file_list(file_list);
2153                         continue;
2154                 }
2155
2156                 quest = NULL;
2157                 lname = NULL;
2158                 rname = NULL;
2159
2160                 for (temp_list = file_list; temp_list;
2161                      temp_list = temp_list->next) {
2162
2163                         SAFE_FREE(lname);
2164                         if (asprintf(&lname, "%s/", temp_list->file_path) <= 0) {
2165                                 continue;
2166                         }
2167                         trim_string(lname, "./", "/");
2168
2169                         /* check if it's a directory */
2170                         if (temp_list->isdir) {
2171                                 /* if (!recurse) continue; */
2172
2173                                 SAFE_FREE(quest);
2174                                 if (asprintf(&quest, "Put directory %s? ", lname) < 0) {
2175                                         break;
2176                                 }
2177                                 if (prompt && !yesno(quest)) { /* No */
2178                                         /* Skip the directory */
2179                                         lname[strlen(lname)-1] = '/';
2180                                         if (!seek_list(temp_list, lname))
2181                                                 break;
2182                                 } else { /* Yes */
2183                                         SAFE_FREE(rname);
2184                                         if(asprintf(&rname, "%s%s", client_get_cur_dir(), lname) < 0) {
2185                                                 break;
2186                                         }
2187                                         normalize_name(rname);
2188                                         if (!NT_STATUS_IS_OK(cli_chkpath(cli, rname)) &&
2189                                             !do_mkdir(rname)) {
2190                                                 DEBUG (0, ("Unable to make dir, skipping..."));
2191                                                 /* Skip the directory */
2192                                                 lname[strlen(lname)-1] = '/';
2193                                                 if (!seek_list(temp_list, lname)) {
2194                                                         break;
2195                                                 }
2196                                         }
2197                                 }
2198                                 continue;
2199                         } else {
2200                                 SAFE_FREE(quest);
2201                                 if (asprintf(&quest,"Put file %s? ", lname) < 0) {
2202                                         break;
2203                                 }
2204                                 if (prompt && !yesno(quest)) {
2205                                         /* No */
2206                                         continue;
2207                                 }
2208
2209                                 /* Yes */
2210                                 SAFE_FREE(rname);
2211                                 if (asprintf(&rname, "%s%s", client_get_cur_dir(), lname) < 0) {
2212                                         break;
2213                                 }
2214                         }
2215
2216                         normalize_name(rname);
2217
2218                         do_put(rname, lname, false);
2219                 }
2220                 free_file_list(file_list);
2221                 SAFE_FREE(quest);
2222                 SAFE_FREE(lname);
2223                 SAFE_FREE(rname);
2224         }
2225
2226         return 0;
2227 }
2228
2229 /****************************************************************************
2230  Cancel a print job.
2231 ****************************************************************************/
2232
2233 static int do_cancel(int job)
2234 {
2235         if (cli_printjob_del(cli, job)) {
2236                 d_printf("Job %d cancelled\n",job);
2237                 return 0;
2238         } else {
2239                 d_printf("Error cancelling job %d : %s\n",job,cli_errstr(cli));
2240                 return 1;
2241         }
2242 }
2243
2244 /****************************************************************************
2245  Cancel a print job.
2246 ****************************************************************************/
2247
2248 static int cmd_cancel(void)
2249 {
2250         TALLOC_CTX *ctx = talloc_tos();
2251         char *buf = NULL;
2252         int job;
2253
2254         if (!next_token_talloc(ctx, &cmd_ptr, &buf,NULL)) {
2255                 d_printf("cancel <jobid> ...\n");
2256                 return 1;
2257         }
2258         do {
2259                 job = atoi(buf);
2260                 do_cancel(job);
2261         } while (next_token_talloc(ctx, &cmd_ptr,&buf,NULL));
2262
2263         return 0;
2264 }
2265
2266 /****************************************************************************
2267  Print a file.
2268 ****************************************************************************/
2269
2270 static int cmd_print(void)
2271 {
2272         TALLOC_CTX *ctx = talloc_tos();
2273         char *lname = NULL;
2274         char *rname = NULL;
2275         char *p = NULL;
2276
2277         if (!next_token_talloc(ctx, &cmd_ptr, &lname,NULL)) {
2278                 d_printf("print <filename>\n");
2279                 return 1;
2280         }
2281
2282         rname = talloc_strdup(ctx, lname);
2283         if (!rname) {
2284                 return 1;
2285         }
2286         p = strrchr_m(rname,'/');
2287         if (p) {
2288                 rname = talloc_asprintf(ctx,
2289                                         "%s-%d",
2290                                         p+1,
2291                                         (int)sys_getpid());
2292         }
2293         if (strequal(lname,"-")) {
2294                 rname = talloc_asprintf(ctx,
2295                                 "stdin-%d",
2296                                 (int)sys_getpid());
2297         }
2298         if (!rname) {
2299                 return 1;
2300         }
2301
2302         return do_put(rname, lname, false);
2303 }
2304
2305 /****************************************************************************
2306  Show a print queue entry.
2307 ****************************************************************************/
2308
2309 static void queue_fn(struct print_job_info *p)
2310 {
2311         d_printf("%-6d   %-9d    %s\n", (int)p->id, (int)p->size, p->name);
2312 }
2313
2314 /****************************************************************************
2315  Show a print queue.
2316 ****************************************************************************/
2317
2318 static int cmd_queue(void)
2319 {
2320         cli_print_queue(cli, queue_fn);
2321         return 0;
2322 }
2323
2324 /****************************************************************************
2325  Delete some files.
2326 ****************************************************************************/
2327
2328 static NTSTATUS do_del(struct cli_state *cli_state, struct file_info *finfo,
2329                    const char *dir)
2330 {
2331         TALLOC_CTX *ctx = talloc_tos();
2332         char *mask = NULL;
2333         NTSTATUS status;
2334
2335         mask = talloc_asprintf(ctx,
2336                                 "%s%c%s",
2337                                 dir,
2338                                 CLI_DIRSEP_CHAR,
2339                                 finfo->name);
2340         if (!mask) {
2341                 return NT_STATUS_NO_MEMORY;
2342         }
2343
2344         if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) {
2345                 TALLOC_FREE(mask);
2346                 return NT_STATUS_OK;
2347         }
2348
2349         status = cli_unlink(cli_state, mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN);
2350         if (!NT_STATUS_IS_OK(status)) {
2351                 d_printf("%s deleting remote file %s\n",
2352                          nt_errstr(status), mask);
2353         }
2354         TALLOC_FREE(mask);
2355         return status;
2356 }
2357
2358 /****************************************************************************
2359  Delete some files.
2360 ****************************************************************************/
2361
2362 static int cmd_del(void)
2363 {
2364         TALLOC_CTX *ctx = talloc_tos();
2365         char *mask = NULL;
2366         char *buf = NULL;
2367         NTSTATUS status = NT_STATUS_OK;
2368         uint16 attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
2369
2370         if (recurse) {
2371                 attribute |= FILE_ATTRIBUTE_DIRECTORY;
2372         }
2373
2374         mask = talloc_strdup(ctx, client_get_cur_dir());
2375         if (!mask) {
2376                 return 1;
2377         }
2378         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2379                 d_printf("del <filename>\n");
2380                 return 1;
2381         }
2382         mask = talloc_asprintf_append(mask, "%s", buf);
2383         if (!mask) {
2384                 return 1;
2385         }
2386
2387         status = do_list(mask,attribute,do_del,false,false);
2388         if (!NT_STATUS_IS_OK(status)) {
2389                 return 1;
2390         }
2391         return 0;
2392 }
2393
2394 /****************************************************************************
2395  Wildcard delete some files.
2396 ****************************************************************************/
2397
2398 static int cmd_wdel(void)
2399 {
2400         TALLOC_CTX *ctx = talloc_tos();
2401         char *mask = NULL;
2402         char *buf = NULL;
2403         uint16 attribute;
2404         struct cli_state *targetcli;
2405         char *targetname = NULL;
2406         NTSTATUS status;
2407
2408         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2409                 d_printf("wdel 0x<attrib> <wcard>\n");
2410                 return 1;
2411         }
2412
2413         attribute = (uint16)strtol(buf, (char **)NULL, 16);
2414
2415         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2416                 d_printf("wdel 0x<attrib> <wcard>\n");
2417                 return 1;
2418         }
2419
2420         mask = talloc_asprintf(ctx, "%s%s",
2421                         client_get_cur_dir(),
2422                         buf);
2423         if (!mask) {
2424                 return 1;
2425         }
2426
2427         status = cli_resolve_path(ctx, "", auth_info, cli, mask, &targetcli,
2428                                   &targetname);
2429         if (!NT_STATUS_IS_OK(status)) {
2430                 d_printf("cmd_wdel %s: %s\n", mask, cli_errstr(cli));
2431                 return 1;
2432         }
2433
2434         status = cli_unlink(targetcli, targetname, attribute);
2435         if (!NT_STATUS_IS_OK(status)) {
2436                 d_printf("%s deleting remote files %s\n", nt_errstr(status),
2437                          targetname);
2438         }
2439         return 0;
2440 }
2441
2442 /****************************************************************************
2443 ****************************************************************************/
2444
2445 static int cmd_open(void)
2446 {
2447         TALLOC_CTX *ctx = talloc_tos();
2448         char *mask = NULL;
2449         char *buf = NULL;
2450         char *targetname = NULL;
2451         struct cli_state *targetcli;
2452         uint16_t fnum = (uint16_t)-1;
2453         NTSTATUS status;
2454
2455         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2456                 d_printf("open <filename>\n");
2457                 return 1;
2458         }
2459         mask = talloc_asprintf(ctx,
2460                         "%s%s",
2461                         client_get_cur_dir(),
2462                         buf);
2463         if (!mask) {
2464                 return 1;
2465         }
2466
2467         status = cli_resolve_path(ctx, "", auth_info, cli, mask, &targetcli,
2468                                   &targetname);
2469         if (!NT_STATUS_IS_OK(status)) {
2470                 d_printf("open %s: %s\n", mask, cli_errstr(cli));
2471                 return 1;
2472         }
2473
2474         if (!NT_STATUS_IS_OK(cli_ntcreate(targetcli, targetname, 0,
2475                         FILE_READ_DATA|FILE_WRITE_DATA, 0,
2476                         FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN, 0x0, 0x0, &fnum))) {
2477                 if (NT_STATUS_IS_OK(cli_ntcreate(targetcli, targetname, 0,
2478                                 FILE_READ_DATA, 0,
2479                                 FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN, 0x0, 0x0, &fnum))) {
2480                         d_printf("open file %s: for read/write fnum %d\n", targetname, fnum);
2481                 } else {
2482                         d_printf("Failed to open file %s. %s\n", targetname, cli_errstr(cli));
2483                 }
2484         } else {
2485                 d_printf("open file %s: for read/write fnum %d\n", targetname, fnum);
2486         }
2487         return 0;
2488 }
2489
2490 static int cmd_posix_encrypt(void)
2491 {
2492         TALLOC_CTX *ctx = talloc_tos();
2493         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
2494
2495         if (cli->use_kerberos) {
2496                 status = cli_gss_smb_encryption_start(cli);
2497         } else {
2498                 char *domain = NULL;
2499                 char *user = NULL;
2500                 char *password = NULL;
2501
2502                 if (!next_token_talloc(ctx, &cmd_ptr,&domain,NULL)) {
2503                         d_printf("posix_encrypt domain user password\n");
2504                         return 1;
2505                 }
2506
2507                 if (!next_token_talloc(ctx, &cmd_ptr,&user,NULL)) {
2508                         d_printf("posix_encrypt domain user password\n");
2509                         return 1;
2510                 }
2511
2512                 if (!next_token_talloc(ctx, &cmd_ptr,&password,NULL)) {
2513                         d_printf("posix_encrypt domain user password\n");
2514                         return 1;
2515                 }
2516
2517                 status = cli_raw_ntlm_smb_encryption_start(cli,
2518                                                         user,
2519                                                         password,
2520                                                         domain);
2521         }
2522
2523         if (!NT_STATUS_IS_OK(status)) {
2524                 d_printf("posix_encrypt failed with error %s\n", nt_errstr(status));
2525         } else {
2526                 d_printf("encryption on\n");
2527                 smb_encrypt = true;
2528         }
2529
2530         return 0;
2531 }
2532
2533 /****************************************************************************
2534 ****************************************************************************/
2535
2536 static int cmd_posix_open(void)
2537 {
2538         TALLOC_CTX *ctx = talloc_tos();
2539         char *mask = NULL;
2540         char *buf = NULL;
2541         char *targetname = NULL;
2542         struct cli_state *targetcli;
2543         mode_t mode;
2544         uint16_t fnum;
2545         NTSTATUS status;
2546
2547         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2548                 d_printf("posix_open <filename> 0<mode>\n");
2549                 return 1;
2550         }
2551         mask = talloc_asprintf(ctx,
2552                         "%s%s",
2553                         client_get_cur_dir(),
2554                         buf);
2555         if (!mask) {
2556                 return 1;
2557         }
2558
2559         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2560                 d_printf("posix_open <filename> 0<mode>\n");
2561                 return 1;
2562         }
2563         mode = (mode_t)strtol(buf, (char **)NULL, 8);
2564
2565         status = cli_resolve_path(ctx, "", auth_info, cli, mask, &targetcli,
2566                                   &targetname);
2567         if (!NT_STATUS_IS_OK(status)) {
2568                 d_printf("posix_open %s: %s\n", mask, cli_errstr(cli));
2569                 return 1;
2570         }
2571
2572         if (!NT_STATUS_IS_OK(cli_posix_open(targetcli, targetname, O_CREAT|O_RDWR, mode, &fnum))) {
2573                 if (!NT_STATUS_IS_OK(cli_posix_open(targetcli, targetname, O_CREAT|O_RDONLY, mode, &fnum))) {
2574                         d_printf("posix_open file %s: for read/write fnum %d\n", targetname, fnum);
2575                 } else {
2576                         d_printf("Failed to open file %s. %s\n", targetname, cli_errstr(cli));
2577                 }
2578         } else {
2579                 d_printf("posix_open file %s: for read/write fnum %d\n", targetname, fnum);
2580         }
2581
2582         return 0;
2583 }
2584
2585 static int cmd_posix_mkdir(void)
2586 {
2587         TALLOC_CTX *ctx = talloc_tos();
2588         char *mask = NULL;
2589         char *buf = NULL;
2590         char *targetname = NULL;
2591         struct cli_state *targetcli;
2592         mode_t mode;
2593         NTSTATUS status;
2594
2595         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2596                 d_printf("posix_mkdir <filename> 0<mode>\n");
2597                 return 1;
2598         }
2599         mask = talloc_asprintf(ctx,
2600                         "%s%s",
2601                         client_get_cur_dir(),
2602                         buf);
2603         if (!mask) {
2604                 return 1;
2605         }
2606
2607         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2608                 d_printf("posix_mkdir <filename> 0<mode>\n");
2609                 return 1;
2610         }
2611         mode = (mode_t)strtol(buf, (char **)NULL, 8);
2612
2613         status = cli_resolve_path(ctx, "", auth_info, cli, mask, &targetcli,
2614                                   &targetname);
2615         if (!NT_STATUS_IS_OK(status)) {
2616                 d_printf("posix_mkdir %s: %s\n", mask, cli_errstr(cli));
2617                 return 1;
2618         }
2619
2620         if (!NT_STATUS_IS_OK(cli_posix_mkdir(targetcli, targetname, mode))) {
2621                 d_printf("Failed to open file %s. %s\n", targetname, cli_errstr(cli));
2622         } else {
2623                 d_printf("posix_mkdir created directory %s\n", targetname);
2624         }
2625         return 0;
2626 }
2627
2628 static int cmd_posix_unlink(void)
2629 {
2630         TALLOC_CTX *ctx = talloc_tos();
2631         char *mask = NULL;
2632         char *buf = NULL;
2633         char *targetname = NULL;
2634         struct cli_state *targetcli;
2635         NTSTATUS status;
2636
2637         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2638                 d_printf("posix_unlink <filename>\n");
2639                 return 1;
2640         }
2641         mask = talloc_asprintf(ctx,
2642                         "%s%s",
2643                         client_get_cur_dir(),
2644                         buf);
2645         if (!mask) {
2646                 return 1;
2647         }
2648
2649         status = cli_resolve_path(ctx, "", auth_info, cli, mask, &targetcli,
2650                                   &targetname);
2651         if (!NT_STATUS_IS_OK(status)) {
2652                 d_printf("posix_unlink %s: %s\n", mask, cli_errstr(cli));
2653                 return 1;
2654         }
2655
2656         if (!NT_STATUS_IS_OK(cli_posix_unlink(targetcli, targetname))) {
2657                 d_printf("Failed to unlink file %s. %s\n", targetname, cli_errstr(cli));
2658         } else {
2659                 d_printf("posix_unlink deleted file %s\n", targetname);
2660         }
2661
2662         return 0;
2663 }
2664
2665 static int cmd_posix_rmdir(void)
2666 {
2667         TALLOC_CTX *ctx = talloc_tos();
2668         char *mask = NULL;
2669         char *buf = NULL;
2670         char *targetname = NULL;
2671         struct cli_state *targetcli;
2672         NTSTATUS status;
2673
2674         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2675                 d_printf("posix_rmdir <filename>\n");
2676                 return 1;
2677         }
2678         mask = talloc_asprintf(ctx,
2679                         "%s%s",
2680                         client_get_cur_dir(),
2681                         buf);
2682         if (!mask) {
2683                 return 1;
2684         }
2685
2686         status = cli_resolve_path(ctx, "", auth_info, cli, mask, &targetcli,
2687                                   &targetname);
2688         if (!NT_STATUS_IS_OK(status)) {
2689                 d_printf("posix_rmdir %s: %s\n", mask, cli_errstr(cli));
2690                 return 1;
2691         }
2692
2693         if (!NT_STATUS_IS_OK(cli_posix_rmdir(targetcli, targetname))) {
2694                 d_printf("Failed to unlink directory %s. %s\n", targetname, cli_errstr(cli));
2695         } else {
2696                 d_printf("posix_rmdir deleted directory %s\n", targetname);
2697         }
2698
2699         return 0;
2700 }
2701
2702 static int cmd_close(void)
2703 {
2704         TALLOC_CTX *ctx = talloc_tos();
2705         char *buf = NULL;
2706         int fnum;
2707
2708         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2709                 d_printf("close <fnum>\n");
2710                 return 1;
2711         }
2712
2713         fnum = atoi(buf);
2714         /* We really should use the targetcli here.... */
2715         if (!NT_STATUS_IS_OK(cli_close(cli, fnum))) {
2716                 d_printf("close %d: %s\n", fnum, cli_errstr(cli));
2717                 return 1;
2718         }
2719         return 0;
2720 }
2721
2722 static int cmd_posix(void)
2723 {
2724         TALLOC_CTX *ctx = talloc_tos();
2725         uint16 major, minor;
2726         uint32 caplow, caphigh;
2727         char *caps;
2728         NTSTATUS status;
2729
2730         if (!SERVER_HAS_UNIX_CIFS(cli)) {
2731                 d_printf("Server doesn't support UNIX CIFS extensions.\n");
2732                 return 1;
2733         }
2734
2735         status = cli_unix_extensions_version(cli, &major, &minor, &caplow,
2736                                              &caphigh);
2737         if (!NT_STATUS_IS_OK(status)) {
2738                 d_printf("Can't get UNIX CIFS extensions version from "
2739                          "server: %s\n", nt_errstr(status));
2740                 return 1;
2741         }
2742
2743         d_printf("Server supports CIFS extensions %u.%u\n", (unsigned int)major, (unsigned int)minor);
2744
2745         caps = talloc_strdup(ctx, "");
2746         if (!caps) {
2747                 return 1;
2748         }
2749         if (caplow & CIFS_UNIX_FCNTL_LOCKS_CAP) {
2750                 caps = talloc_asprintf_append(caps, "locks ");
2751                 if (!caps) {
2752                         return 1;
2753                 }
2754         }
2755         if (caplow & CIFS_UNIX_POSIX_ACLS_CAP) {
2756                 caps = talloc_asprintf_append(caps, "acls ");
2757                 if (!caps) {
2758                         return 1;
2759                 }
2760         }
2761         if (caplow & CIFS_UNIX_XATTTR_CAP) {
2762                 caps = talloc_asprintf_append(caps, "eas ");
2763                 if (!caps) {
2764                         return 1;
2765                 }
2766         }
2767         if (caplow & CIFS_UNIX_POSIX_PATHNAMES_CAP) {
2768                 caps = talloc_asprintf_append(caps, "pathnames ");
2769                 if (!caps) {
2770                         return 1;
2771                 }
2772         }
2773         if (caplow & CIFS_UNIX_POSIX_PATH_OPERATIONS_CAP) {
2774                 caps = talloc_asprintf_append(caps, "posix_path_operations ");
2775                 if (!caps) {
2776                         return 1;
2777                 }
2778         }
2779         if (caplow & CIFS_UNIX_LARGE_READ_CAP) {
2780                 caps = talloc_asprintf_append(caps, "large_read ");
2781                 if (!caps) {
2782                         return 1;
2783                 }
2784         }
2785         if (caplow & CIFS_UNIX_LARGE_WRITE_CAP) {
2786                 caps = talloc_asprintf_append(caps, "large_write ");
2787                 if (!caps) {
2788                         return 1;
2789                 }
2790         }
2791         if (caplow & CIFS_UNIX_TRANSPORT_ENCRYPTION_CAP) {
2792                 caps = talloc_asprintf_append(caps, "posix_encrypt ");
2793                 if (!caps) {
2794                         return 1;
2795                 }
2796         }
2797         if (caplow & CIFS_UNIX_TRANSPORT_ENCRYPTION_MANDATORY_CAP) {
2798                 caps = talloc_asprintf_append(caps, "mandatory_posix_encrypt ");
2799                 if (!caps) {
2800                         return 1;
2801                 }
2802         }
2803
2804         if (*caps && caps[strlen(caps)-1] == ' ') {
2805                 caps[strlen(caps)-1] = '\0';
2806         }
2807
2808         d_printf("Server supports CIFS capabilities %s\n", caps);
2809
2810         status = cli_set_unix_extensions_capabilities(cli, major, minor,
2811                                                       caplow, caphigh);
2812         if (!NT_STATUS_IS_OK(status)) {
2813                 d_printf("Can't set UNIX CIFS extensions capabilities. %s.\n",
2814                          nt_errstr(status));
2815                 return 1;
2816         }
2817
2818         if (caplow & CIFS_UNIX_POSIX_PATHNAMES_CAP) {
2819                 CLI_DIRSEP_CHAR = '/';
2820                 *CLI_DIRSEP_STR = '/';
2821                 client_set_cur_dir(CLI_DIRSEP_STR);
2822         }
2823
2824         return 0;
2825 }
2826
2827 static int cmd_lock(void)
2828 {
2829         TALLOC_CTX *ctx = talloc_tos();
2830         char *buf = NULL;
2831         uint64_t start, len;
2832         enum brl_type lock_type;
2833         int fnum;
2834
2835         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2836                 d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n");
2837                 return 1;
2838         }
2839         fnum = atoi(buf);
2840
2841         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2842                 d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n");
2843                 return 1;
2844         }
2845
2846         if (*buf == 'r' || *buf == 'R') {
2847                 lock_type = READ_LOCK;
2848         } else if (*buf == 'w' || *buf == 'W') {
2849                 lock_type = WRITE_LOCK;
2850         } else {
2851                 d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n");
2852                 return 1;
2853         }
2854
2855         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2856                 d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n");
2857                 return 1;
2858         }
2859
2860         start = (uint64_t)strtol(buf, (char **)NULL, 16);
2861
2862         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2863                 d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n");
2864                 return 1;
2865         }
2866
2867         len = (uint64_t)strtol(buf, (char **)NULL, 16);
2868
2869         if (!NT_STATUS_IS_OK(cli_posix_lock(cli, fnum, start, len, true, lock_type))) {
2870                 d_printf("lock failed %d: %s\n", fnum, cli_errstr(cli));
2871         }
2872
2873         return 0;
2874 }
2875
2876 static int cmd_unlock(void)
2877 {
2878         TALLOC_CTX *ctx = talloc_tos();
2879         char *buf = NULL;
2880         uint64_t start, len;
2881         int fnum;
2882
2883         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2884                 d_printf("unlock <fnum> <hex-start> <hex-len>\n");
2885                 return 1;
2886         }
2887         fnum = atoi(buf);
2888
2889         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2890                 d_printf("unlock <fnum> <hex-start> <hex-len>\n");
2891                 return 1;
2892         }
2893
2894         start = (uint64_t)strtol(buf, (char **)NULL, 16);
2895
2896         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2897                 d_printf("unlock <fnum> <hex-start> <hex-len>\n");
2898                 return 1;
2899         }
2900
2901         len = (uint64_t)strtol(buf, (char **)NULL, 16);
2902
2903         if (!NT_STATUS_IS_OK(cli_posix_unlock(cli, fnum, start, len))) {
2904                 d_printf("unlock failed %d: %s\n", fnum, cli_errstr(cli));
2905         }
2906
2907         return 0;
2908 }
2909
2910
2911 /****************************************************************************
2912  Remove a directory.
2913 ****************************************************************************/
2914
2915 static int cmd_rmdir(void)
2916 {
2917         TALLOC_CTX *ctx = talloc_tos();
2918         char *mask = NULL;
2919         char *buf = NULL;
2920         char *targetname = NULL;
2921         struct cli_state *targetcli;
2922         NTSTATUS status;
2923
2924         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
2925                 d_printf("rmdir <dirname>\n");
2926                 return 1;
2927         }
2928         mask = talloc_asprintf(ctx,
2929                         "%s%s",
2930                         client_get_cur_dir(),
2931                         buf);
2932         if (!mask) {
2933                 return 1;
2934         }
2935
2936         status = cli_resolve_path(ctx, "", auth_info, cli, mask, &targetcli,
2937                                   &targetname);
2938         if (!NT_STATUS_IS_OK(status)) {
2939                 d_printf("rmdir %s: %s\n", mask, cli_errstr(cli));
2940                 return 1;
2941         }
2942
2943         if (!NT_STATUS_IS_OK(cli_rmdir(targetcli, targetname))) {
2944                 d_printf("%s removing remote directory file %s\n",
2945                          cli_errstr(targetcli),mask);
2946         }
2947
2948         return 0;
2949 }
2950
2951 /****************************************************************************
2952  UNIX hardlink.
2953 ****************************************************************************/
2954
2955 static int cmd_link(void)
2956 {
2957         TALLOC_CTX *ctx = talloc_tos();
2958         char *oldname = NULL;
2959         char *newname = NULL;
2960         char *buf = NULL;
2961         char *buf2 = NULL;
2962         char *targetname = NULL;
2963         struct cli_state *targetcli;
2964         NTSTATUS status;
2965
2966         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) ||
2967             !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) {
2968                 d_printf("link <oldname> <newname>\n");
2969                 return 1;
2970         }
2971         oldname = talloc_asprintf(ctx,
2972                         "%s%s",
2973                         client_get_cur_dir(),
2974                         buf);
2975         if (!oldname) {
2976                 return 1;
2977         }
2978         newname = talloc_asprintf(ctx,
2979                         "%s%s",
2980                         client_get_cur_dir(),
2981                         buf2);
2982         if (!newname) {
2983                 return 1;
2984         }
2985
2986         status = cli_resolve_path(ctx, "", auth_info, cli, oldname, &targetcli,
2987                                   &targetname);
2988         if (!NT_STATUS_IS_OK(status)) {
2989                 d_printf("link %s: %s\n", oldname, cli_errstr(cli));
2990                 return 1;
2991         }
2992
2993         if (!SERVER_HAS_UNIX_CIFS(targetcli)) {
2994                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2995                 return 1;
2996         }
2997
2998         if (!NT_STATUS_IS_OK(cli_posix_hardlink(targetcli, targetname, newname))) {
2999                 d_printf("%s linking files (%s -> %s)\n", cli_errstr(targetcli), newname, oldname);
3000                 return 1;
3001         }
3002         return 0;
3003 }
3004
3005 /****************************************************************************
3006  UNIX readlink.
3007 ****************************************************************************/
3008
3009 static int cmd_readlink(void)
3010 {
3011         TALLOC_CTX *ctx = talloc_tos();
3012         char *name= NULL;
3013         char *buf = NULL;
3014         char *targetname = NULL;
3015         char linkname[PATH_MAX+1];
3016         struct cli_state *targetcli;
3017         NTSTATUS status;
3018
3019         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
3020                 d_printf("readlink <name>\n");
3021                 return 1;
3022         }
3023         name = talloc_asprintf(ctx,
3024                         "%s%s",
3025                         client_get_cur_dir(),
3026                         buf);
3027         if (!name) {
3028                 return 1;
3029         }
3030
3031         status = cli_resolve_path(ctx, "", auth_info, cli, name, &targetcli,
3032                                   &targetname);
3033         if (!NT_STATUS_IS_OK(status)) {
3034                 d_printf("readlink %s: %s\n", name, cli_errstr(cli));
3035                 return 1;
3036         }
3037
3038         if (!SERVER_HAS_UNIX_CIFS(targetcli)) {
3039                 d_printf("Server doesn't support UNIX CIFS calls.\n");
3040                 return 1;
3041         }
3042
3043         if (!NT_STATUS_IS_OK(cli_posix_readlink(targetcli, name,
3044                         linkname, PATH_MAX+1))) {
3045                 d_printf("%s readlink on file %s\n",
3046                         cli_errstr(targetcli), name);
3047                 return 1;
3048         }
3049
3050         d_printf("%s -> %s\n", name, linkname);
3051
3052         return 0;
3053 }
3054
3055
3056 /****************************************************************************
3057  UNIX symlink.
3058 ****************************************************************************/
3059
3060 static int cmd_symlink(void)
3061 {
3062         TALLOC_CTX *ctx = talloc_tos();
3063         char *oldname = NULL;
3064         char *newname = NULL;
3065         char *buf = NULL;
3066         char *buf2 = NULL;
3067         struct cli_state *newcli;
3068         NTSTATUS status;
3069
3070         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) ||
3071             !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) {
3072                 d_printf("symlink <oldname> <newname>\n");
3073                 return 1;
3074         }
3075         /* Oldname (link target) must be an untouched blob. */
3076         oldname = buf;
3077
3078         if (SERVER_HAS_UNIX_CIFS(cli)) {
3079                 newname = talloc_asprintf(ctx, "%s%s", client_get_cur_dir(),
3080                                           buf2);
3081                 if (!newname) {
3082                         return 1;
3083                 }
3084                 /* New name must be present in share namespace. */
3085                 status = cli_resolve_path(ctx, "", auth_info, cli, newname,
3086                                           &newcli, &newname);
3087                 if (!NT_STATUS_IS_OK(status)) {
3088                         d_printf("link %s: %s\n", oldname, cli_errstr(cli));
3089                         return 1;
3090                 }
3091                 status = cli_posix_symlink(newcli, oldname, newname);
3092         } else {
3093                 status = cli_symlink(
3094                         cli, oldname, buf2,
3095                         buf2[0] == '\\' ? 0 : SYMLINK_FLAG_RELATIVE);
3096         }
3097
3098         if (!NT_STATUS_IS_OK(status)) {
3099                 d_printf("%s symlinking files (%s -> %s)\n",
3100                          nt_errstr(status), oldname, newname);
3101                 return 1;
3102         }
3103
3104         return 0;
3105 }
3106
3107 /****************************************************************************
3108  UNIX chmod.
3109 ****************************************************************************/
3110
3111 static int cmd_chmod(void)
3112 {
3113         TALLOC_CTX *ctx = talloc_tos();
3114         char *src = NULL;
3115         char *buf = NULL;
3116         char *buf2 = NULL;
3117         char *targetname = NULL;
3118         struct cli_state *targetcli;
3119         mode_t mode;
3120         NTSTATUS status;
3121
3122         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) ||
3123             !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) {
3124                 d_printf("chmod mode file\n");
3125                 return 1;
3126         }
3127         src = talloc_asprintf(ctx,
3128                         "%s%s",
3129                         client_get_cur_dir(),
3130                         buf2);
3131         if (!src) {
3132                 return 1;
3133         }
3134
3135         mode = (mode_t)strtol(buf, NULL, 8);
3136
3137         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3138                                   &targetname);
3139         if (!NT_STATUS_IS_OK(status)) {
3140                 d_printf("chmod %s: %s\n", src, cli_errstr(cli));
3141                 return 1;
3142         }
3143
3144         if (!SERVER_HAS_UNIX_CIFS(targetcli)) {
3145                 d_printf("Server doesn't support UNIX CIFS calls.\n");
3146                 return 1;
3147         }
3148
3149         if (!NT_STATUS_IS_OK(cli_posix_chmod(targetcli, targetname, mode))) {
3150                 d_printf("%s chmod file %s 0%o\n",
3151                         cli_errstr(targetcli), src, (unsigned int)mode);
3152                 return 1;
3153         }
3154
3155         return 0;
3156 }
3157
3158 static const char *filetype_to_str(mode_t mode)
3159 {
3160         if (S_ISREG(mode)) {
3161                 return "regular file";
3162         } else if (S_ISDIR(mode)) {
3163                 return "directory";
3164         } else
3165 #ifdef S_ISCHR
3166         if (S_ISCHR(mode)) {
3167                 return "character device";
3168         } else
3169 #endif
3170 #ifdef S_ISBLK
3171         if (S_ISBLK(mode)) {
3172                 return "block device";
3173         } else
3174 #endif
3175 #ifdef S_ISFIFO
3176         if (S_ISFIFO(mode)) {
3177                 return "fifo";
3178         } else
3179 #endif
3180 #ifdef S_ISLNK
3181         if (S_ISLNK(mode)) {
3182                 return "symbolic link";
3183         } else
3184 #endif
3185 #ifdef S_ISSOCK
3186         if (S_ISSOCK(mode)) {
3187                 return "socket";
3188         } else
3189 #endif
3190         return "";
3191 }
3192
3193 static char rwx_to_str(mode_t m, mode_t bt, char ret)
3194 {
3195         if (m & bt) {
3196                 return ret;
3197         } else {
3198                 return '-';
3199         }
3200 }
3201
3202 static char *unix_mode_to_str(char *s, mode_t m)
3203 {
3204         char *p = s;
3205         const char *str = filetype_to_str(m);
3206
3207         switch(str[0]) {
3208                 case 'd':
3209                         *p++ = 'd';
3210                         break;
3211                 case 'c':
3212                         *p++ = 'c';
3213                         break;
3214                 case 'b':
3215                         *p++ = 'b';
3216                         break;
3217                 case 'f':
3218                         *p++ = 'p';
3219                         break;
3220                 case 's':
3221                         *p++ = str[1] == 'y' ? 'l' : 's';
3222                         break;
3223                 case 'r':
3224                 default:
3225                         *p++ = '-';
3226                         break;
3227         }
3228         *p++ = rwx_to_str(m, S_IRUSR, 'r');
3229         *p++ = rwx_to_str(m, S_IWUSR, 'w');
3230         *p++ = rwx_to_str(m, S_IXUSR, 'x');
3231         *p++ = rwx_to_str(m, S_IRGRP, 'r');
3232         *p++ = rwx_to_str(m, S_IWGRP, 'w');
3233         *p++ = rwx_to_str(m, S_IXGRP, 'x');
3234         *p++ = rwx_to_str(m, S_IROTH, 'r');
3235         *p++ = rwx_to_str(m, S_IWOTH, 'w');
3236         *p++ = rwx_to_str(m, S_IXOTH, 'x');
3237         *p++ = '\0';
3238         return s;
3239 }
3240
3241 /****************************************************************************
3242  Utility function for UNIX getfacl.
3243 ****************************************************************************/
3244
3245 static char *perms_to_string(fstring permstr, unsigned char perms)
3246 {
3247         fstrcpy(permstr, "---");
3248         if (perms & SMB_POSIX_ACL_READ) {
3249                 permstr[0] = 'r';
3250         }
3251         if (perms & SMB_POSIX_ACL_WRITE) {
3252                 permstr[1] = 'w';
3253         }
3254         if (perms & SMB_POSIX_ACL_EXECUTE) {
3255                 permstr[2] = 'x';
3256         }
3257         return permstr;
3258 }
3259
3260 /****************************************************************************
3261  UNIX getfacl.
3262 ****************************************************************************/
3263
3264 static int cmd_getfacl(void)
3265 {
3266         TALLOC_CTX *ctx = talloc_tos();
3267         char *src = NULL;
3268         char *name = NULL;
3269         char *targetname = NULL;
3270         struct cli_state *targetcli;
3271         uint16 major, minor;
3272         uint32 caplow, caphigh;
3273         char *retbuf = NULL;
3274         size_t rb_size = 0;
3275         SMB_STRUCT_STAT sbuf;
3276         uint16 num_file_acls = 0;
3277         uint16 num_dir_acls = 0;
3278         uint16 i;
3279         NTSTATUS status;
3280
3281         if (!next_token_talloc(ctx, &cmd_ptr,&name,NULL)) {
3282                 d_printf("getfacl filename\n");
3283                 return 1;
3284         }
3285         src = talloc_asprintf(ctx,
3286                         "%s%s",
3287                         client_get_cur_dir(),
3288                         name);
3289         if (!src) {
3290                 return 1;
3291         }
3292
3293         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3294                                   &targetname);
3295         if (!NT_STATUS_IS_OK(status)) {
3296                 d_printf("stat %s: %s\n", src, cli_errstr(cli));
3297                 return 1;
3298         }
3299
3300         if (!SERVER_HAS_UNIX_CIFS(targetcli)) {
3301                 d_printf("Server doesn't support UNIX CIFS calls.\n");
3302                 return 1;
3303         }
3304
3305         status = cli_unix_extensions_version(targetcli, &major, &minor,
3306                                              &caplow, &caphigh);
3307         if (!NT_STATUS_IS_OK(status)) {
3308                 d_printf("Can't get UNIX CIFS version from server: %s.\n",
3309                          nt_errstr(status));
3310                 return 1;
3311         }
3312
3313         if (!(caplow & CIFS_UNIX_POSIX_ACLS_CAP)) {
3314                 d_printf("This server supports UNIX extensions "
3315                         "but doesn't support POSIX ACLs.\n");
3316                 return 1;
3317         }
3318
3319         if (!NT_STATUS_IS_OK(cli_posix_stat(targetcli, targetname, &sbuf))) {
3320                 d_printf("%s getfacl doing a stat on file %s\n",
3321                         cli_errstr(targetcli), src);
3322                 return 1;
3323         }
3324
3325         if (!NT_STATUS_IS_OK(cli_posix_getfacl(targetcli, targetname, ctx, &rb_size, &retbuf))) {
3326                 d_printf("%s getfacl file %s\n",
3327                         cli_errstr(targetcli), src);
3328                 return 1;
3329         }
3330
3331         /* ToDo : Print out the ACL values. */
3332         if (rb_size < 6 || SVAL(retbuf,0) != SMB_POSIX_ACL_VERSION) {
3333                 d_printf("getfacl file %s, unknown POSIX acl version %u.\n",
3334                         src, (unsigned int)CVAL(retbuf,0) );
3335                 return 1;
3336         }
3337
3338         num_file_acls = SVAL(retbuf,2);
3339         num_dir_acls = SVAL(retbuf,4);
3340         if (rb_size != SMB_POSIX_ACL_HEADER_SIZE + SMB_POSIX_ACL_ENTRY_SIZE*(num_file_acls+num_dir_acls)) {
3341                 d_printf("getfacl file %s, incorrect POSIX acl buffer size (should be %u, was %u).\n",
3342                         src,
3343                         (unsigned int)(SMB_POSIX_ACL_HEADER_SIZE + SMB_POSIX_ACL_ENTRY_SIZE*(num_file_acls+num_dir_acls)),
3344                         (unsigned int)rb_size);
3345                 return 1;
3346         }
3347
3348         d_printf("# file: %s\n", src);
3349         d_printf("# owner: %u\n# group: %u\n", (unsigned int)sbuf.st_ex_uid, (unsigned int)sbuf.st_ex_gid);
3350
3351         if (num_file_acls == 0 && num_dir_acls == 0) {
3352                 d_printf("No acls found.\n");
3353         }
3354
3355         for (i = 0; i < num_file_acls; i++) {
3356                 uint32 uorg;
3357                 fstring permstring;
3358                 unsigned char tagtype = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE));
3359                 unsigned char perms = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+1);
3360
3361                 switch(tagtype) {
3362                         case SMB_POSIX_ACL_USER_OBJ:
3363                                 d_printf("user::");
3364                                 break;
3365                         case SMB_POSIX_ACL_USER:
3366                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+2);
3367                                 d_printf("user:%u:", uorg);
3368                                 break;
3369                         case SMB_POSIX_ACL_GROUP_OBJ:
3370                                 d_printf("group::");
3371                                 break;
3372                         case SMB_POSIX_ACL_GROUP:
3373                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+2);
3374                                 d_printf("group:%u:", uorg);
3375                                 break;
3376                         case SMB_POSIX_ACL_MASK:
3377                                 d_printf("mask::");
3378                                 break;
3379                         case SMB_POSIX_ACL_OTHER:
3380                                 d_printf("other::");
3381                                 break;
3382                         default:
3383                                 d_printf("getfacl file %s, incorrect POSIX acl tagtype (%u).\n",
3384                                         src, (unsigned int)tagtype );
3385                                 SAFE_FREE(retbuf);
3386                                 return 1;
3387                 }
3388
3389                 d_printf("%s\n", perms_to_string(permstring, perms));
3390         }
3391
3392         for (i = 0; i < num_dir_acls; i++) {
3393                 uint32 uorg;
3394                 fstring permstring;
3395                 unsigned char tagtype = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE));
3396                 unsigned char perms = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+1);
3397
3398                 switch(tagtype) {
3399                         case SMB_POSIX_ACL_USER_OBJ:
3400                                 d_printf("default:user::");
3401                                 break;
3402                         case SMB_POSIX_ACL_USER:
3403                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+2);
3404                                 d_printf("default:user:%u:", uorg);
3405                                 break;
3406                         case SMB_POSIX_ACL_GROUP_OBJ:
3407                                 d_printf("default:group::");
3408                                 break;
3409                         case SMB_POSIX_ACL_GROUP:
3410                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+2);
3411                                 d_printf("default:group:%u:", uorg);
3412                                 break;
3413                         case SMB_POSIX_ACL_MASK:
3414                                 d_printf("default:mask::");
3415                                 break;
3416                         case SMB_POSIX_ACL_OTHER:
3417                                 d_printf("default:other::");
3418                                 break;
3419                         default:
3420                                 d_printf("getfacl file %s, incorrect POSIX acl tagtype (%u).\n",
3421                                         src, (unsigned int)tagtype );
3422                                 SAFE_FREE(retbuf);
3423                                 return 1;
3424                 }
3425
3426                 d_printf("%s\n", perms_to_string(permstring, perms));
3427         }
3428
3429         return 0;
3430 }
3431
3432 static void printf_cb(const char *buf, void *private_data)
3433 {
3434         printf("%s", buf);
3435 }
3436
3437 /****************************************************************************
3438  Get the EA list of a file
3439 ****************************************************************************/
3440
3441 static int cmd_geteas(void)
3442 {
3443         TALLOC_CTX *ctx = talloc_tos();
3444         char *src = NULL;
3445         char *name = NULL;
3446         char *targetname = NULL;
3447         struct cli_state *targetcli;
3448         NTSTATUS status;
3449         size_t i, num_eas;
3450         struct ea_struct *eas;
3451
3452         if (!next_token_talloc(ctx, &cmd_ptr,&name,NULL)) {
3453                 d_printf("geteas filename\n");
3454                 return 1;
3455         }
3456         src = talloc_asprintf(ctx,
3457                         "%s%s",
3458                         client_get_cur_dir(),
3459                         name);
3460         if (!src) {
3461                 return 1;
3462         }
3463
3464         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3465                                   &targetname);
3466         if (!NT_STATUS_IS_OK(status)) {
3467                 d_printf("stat %s: %s\n", src, cli_errstr(cli));
3468                 return 1;
3469         }
3470
3471         status = cli_get_ea_list_path(targetcli, targetname, talloc_tos(),
3472                                       &num_eas, &eas);
3473         if (!NT_STATUS_IS_OK(status)) {
3474                 d_printf("cli_get_ea_list_path: %s\n", nt_errstr(status));
3475                 return 1;
3476         }
3477
3478         for (i=0; i<num_eas; i++) {
3479                 d_printf("%s (%d) =\n", eas[i].name, (int)eas[i].flags);
3480                 dump_data_cb(eas[i].value.data, eas[i].value.length, false,
3481                              printf_cb, NULL);
3482                 d_printf("\n");
3483         }
3484
3485         TALLOC_FREE(eas);
3486
3487         return 0;
3488 }
3489
3490 /****************************************************************************
3491  Set an EA of a file
3492 ****************************************************************************/
3493
3494 static int cmd_setea(void)
3495 {
3496         TALLOC_CTX *ctx = talloc_tos();
3497         char *src = NULL;
3498         char *name = NULL;
3499         char *eaname = NULL;
3500         char *eavalue = NULL;
3501         char *targetname = NULL;
3502         struct cli_state *targetcli;
3503         NTSTATUS status;
3504
3505         if (!next_token_talloc(ctx, &cmd_ptr, &name, NULL)
3506             || !next_token_talloc(ctx, &cmd_ptr, &eaname, NULL)) {
3507                 d_printf("setea filename eaname value\n");
3508                 return 1;
3509         }
3510         if (!next_token_talloc(ctx, &cmd_ptr, &eavalue, NULL)) {
3511                 eavalue = talloc_strdup(ctx, "");
3512         }
3513         src = talloc_asprintf(ctx,
3514                         "%s%s",
3515                         client_get_cur_dir(),
3516                         name);
3517         if (!src) {
3518                 return 1;
3519         }
3520
3521         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3522                                   &targetname);
3523         if (!NT_STATUS_IS_OK(status)) {
3524                 d_printf("stat %s: %s\n", src, cli_errstr(cli));
3525                 return 1;
3526         }
3527
3528         status =  cli_set_ea_path(targetcli, targetname, eaname, eavalue,
3529                                   strlen(eavalue));
3530         if (!NT_STATUS_IS_OK(status)) {
3531                 d_printf("set_ea %s: %s\n", src, nt_errstr(status));
3532                 return 1;
3533         }
3534
3535         return 0;
3536 }
3537
3538 /****************************************************************************
3539  UNIX stat.
3540 ****************************************************************************/
3541
3542 static int cmd_stat(void)
3543 {
3544         TALLOC_CTX *ctx = talloc_tos();
3545         char *src = NULL;
3546         char *name = NULL;
3547         char *targetname = NULL;
3548         struct cli_state *targetcli;
3549         fstring mode_str;
3550         SMB_STRUCT_STAT sbuf;
3551         struct tm *lt;
3552         time_t tmp_time;
3553         NTSTATUS status;
3554
3555         if (!next_token_talloc(ctx, &cmd_ptr,&name,NULL)) {
3556                 d_printf("stat file\n");
3557                 return 1;
3558         }
3559         src = talloc_asprintf(ctx,
3560                         "%s%s",
3561                         client_get_cur_dir(),
3562                         name);
3563         if (!src) {
3564                 return 1;
3565         }
3566
3567         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3568                                   &targetname);
3569         if (!NT_STATUS_IS_OK(status)) {
3570                 d_printf("stat %s: %s\n", src, cli_errstr(cli));
3571                 return 1;
3572         }
3573
3574         if (!SERVER_HAS_UNIX_CIFS(targetcli)) {
3575                 d_printf("Server doesn't support UNIX CIFS calls.\n");
3576                 return 1;
3577         }
3578
3579         if (!NT_STATUS_IS_OK(cli_posix_stat(targetcli, targetname, &sbuf))) {
3580                 d_printf("%s stat file %s\n",
3581                         cli_errstr(targetcli), src);
3582                 return 1;
3583         }
3584
3585         /* Print out the stat values. */
3586         d_printf("File: %s\n", src);
3587         d_printf("Size: %-12.0f\tBlocks: %u\t%s\n",
3588                 (double)sbuf.st_ex_size,
3589                 (unsigned int)sbuf.st_ex_blocks,
3590                 filetype_to_str(sbuf.st_ex_mode));
3591
3592 #if defined(S_ISCHR) && defined(S_ISBLK)
3593         if (S_ISCHR(sbuf.st_ex_mode) || S_ISBLK(sbuf.st_ex_mode)) {
3594                 d_printf("Inode: %.0f\tLinks: %u\tDevice type: %u,%u\n",
3595                         (double)sbuf.st_ex_ino,
3596                         (unsigned int)sbuf.st_ex_nlink,
3597                         unix_dev_major(sbuf.st_ex_rdev),
3598                         unix_dev_minor(sbuf.st_ex_rdev));
3599         } else
3600 #endif
3601                 d_printf("Inode: %.0f\tLinks: %u\n",
3602                         (double)sbuf.st_ex_ino,
3603                         (unsigned int)sbuf.st_ex_nlink);
3604
3605         d_printf("Access: (0%03o/%s)\tUid: %u\tGid: %u\n",
3606                 ((int)sbuf.st_ex_mode & 0777),
3607                 unix_mode_to_str(mode_str, sbuf.st_ex_mode),
3608                 (unsigned int)sbuf.st_ex_uid,
3609                 (unsigned int)sbuf.st_ex_gid);
3610
3611         tmp_time = convert_timespec_to_time_t(sbuf.st_ex_atime);
3612         lt = localtime(&tmp_time);
3613         if (lt) {
3614                 strftime(mode_str, sizeof(mode_str), "%Y-%m-%d %T %z", lt);
3615         } else {
3616                 fstrcpy(mode_str, "unknown");
3617         }
3618         d_printf("Access: %s\n", mode_str);
3619
3620         tmp_time = convert_timespec_to_time_t(sbuf.st_ex_mtime);
3621         lt = localtime(&tmp_time);
3622         if (lt) {
3623                 strftime(mode_str, sizeof(mode_str), "%Y-%m-%d %T %z", lt);
3624         } else {
3625                 fstrcpy(mode_str, "unknown");
3626         }
3627         d_printf("Modify: %s\n", mode_str);
3628
3629         tmp_time = convert_timespec_to_time_t(sbuf.st_ex_ctime);
3630         lt = localtime(&tmp_time);
3631         if (lt) {
3632                 strftime(mode_str, sizeof(mode_str), "%Y-%m-%d %T %z", lt);
3633         } else {
3634                 fstrcpy(mode_str, "unknown");
3635         }
3636         d_printf("Change: %s\n", mode_str);
3637
3638         return 0;
3639 }
3640
3641
3642 /****************************************************************************
3643  UNIX chown.
3644 ****************************************************************************/
3645
3646 static int cmd_chown(void)
3647 {
3648         TALLOC_CTX *ctx = talloc_tos();
3649         char *src = NULL;
3650         uid_t uid;
3651         gid_t gid;
3652         char *buf, *buf2, *buf3;
3653         struct cli_state *targetcli;
3654         char *targetname = NULL;
3655         NTSTATUS status;
3656
3657         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) ||
3658             !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL) ||
3659             !next_token_talloc(ctx, &cmd_ptr,&buf3,NULL)) {
3660                 d_printf("chown uid gid file\n");
3661                 return 1;
3662         }
3663
3664         uid = (uid_t)atoi(buf);
3665         gid = (gid_t)atoi(buf2);
3666
3667         src = talloc_asprintf(ctx,
3668                         "%s%s",
3669                         client_get_cur_dir(),
3670                         buf3);
3671         if (!src) {
3672                 return 1;
3673         }
3674         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3675                                   &targetname);
3676         if (!NT_STATUS_IS_OK(status)) {
3677                 d_printf("chown %s: %s\n", src, cli_errstr(cli));
3678                 return 1;
3679         }
3680
3681         if (!SERVER_HAS_UNIX_CIFS(targetcli)) {
3682                 d_printf("Server doesn't support UNIX CIFS calls.\n");
3683                 return 1;
3684         }
3685
3686         if (!NT_STATUS_IS_OK(cli_posix_chown(targetcli, targetname, uid, gid))) {
3687                 d_printf("%s chown file %s uid=%d, gid=%d\n",
3688                         cli_errstr(targetcli), src, (int)uid, (int)gid);
3689                 return 1;
3690         }
3691
3692         return 0;
3693 }
3694
3695 /****************************************************************************
3696  Rename some file.
3697 ****************************************************************************/
3698
3699 static int cmd_rename(void)
3700 {
3701         TALLOC_CTX *ctx = talloc_tos();
3702         char *src, *dest;
3703         char *buf, *buf2;
3704         struct cli_state *targetcli;
3705         char *targetsrc;
3706         char *targetdest;
3707         NTSTATUS status;
3708
3709         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) ||
3710             !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) {
3711                 d_printf("rename <src> <dest>\n");
3712                 return 1;
3713         }
3714
3715         src = talloc_asprintf(ctx,
3716                         "%s%s",
3717                         client_get_cur_dir(),
3718                         buf);
3719         if (!src) {
3720                 return 1;
3721         }
3722
3723         dest = talloc_asprintf(ctx,
3724                         "%s%s",
3725                         client_get_cur_dir(),
3726                         buf2);
3727         if (!dest) {
3728                 return 1;
3729         }
3730
3731         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3732                                   &targetsrc);
3733         if (!NT_STATUS_IS_OK(status)) {
3734                 d_printf("rename %s: %s\n", src, cli_errstr(cli));
3735                 return 1;
3736         }
3737
3738         status = cli_resolve_path(ctx, "", auth_info, cli, dest, &targetcli,
3739                                   &targetdest);
3740         if (!NT_STATUS_IS_OK(status)) {
3741                 d_printf("rename %s: %s\n", dest, cli_errstr(cli));
3742                 return 1;
3743         }
3744
3745         if (!NT_STATUS_IS_OK(cli_rename(targetcli, targetsrc, targetdest))) {
3746                 d_printf("%s renaming files %s -> %s \n",
3747                         cli_errstr(targetcli),
3748                         targetsrc,
3749                         targetdest);
3750                 return 1;
3751         }
3752
3753         return 0;
3754 }
3755
3756 /****************************************************************************
3757  Print the volume name.
3758 ****************************************************************************/
3759
3760 static int cmd_volume(void)
3761 {
3762         fstring volname;
3763         uint32 serial_num;
3764         time_t create_date;
3765         NTSTATUS status;
3766
3767         status = cli_get_fs_volume_info(cli, volname, &serial_num,
3768                                         &create_date);
3769         if (!NT_STATUS_IS_OK(status)) {
3770                 d_printf("Error %s getting volume info\n", nt_errstr(status));
3771                 return 1;
3772         }
3773
3774         d_printf("Volume: |%s| serial number 0x%x\n",
3775                         volname, (unsigned int)serial_num);
3776         return 0;
3777 }
3778
3779 /****************************************************************************
3780  Hard link files using the NT call.
3781 ****************************************************************************/
3782
3783 static int cmd_hardlink(void)
3784 {
3785         TALLOC_CTX *ctx = talloc_tos();
3786         char *src, *dest;
3787         char *buf, *buf2;
3788         struct cli_state *targetcli;
3789         char *targetname;
3790         NTSTATUS status;
3791
3792         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) ||
3793             !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) {
3794                 d_printf("hardlink <src> <dest>\n");
3795                 return 1;
3796         }
3797
3798         src = talloc_asprintf(ctx,
3799                         "%s%s",
3800                         client_get_cur_dir(),
3801                         buf);
3802         if (!src) {
3803                 return 1;
3804         }
3805
3806         dest = talloc_asprintf(ctx,
3807                         "%s%s",
3808                         client_get_cur_dir(),
3809                         buf2);
3810         if (!dest) {
3811                 return 1;
3812         }
3813
3814         status = cli_resolve_path(ctx, "", auth_info, cli, src, &targetcli,
3815                                   &targetname);
3816         if (!NT_STATUS_IS_OK(status)) {
3817                 d_printf("hardlink %s: %s\n", src, cli_errstr(cli));
3818                 return 1;
3819         }
3820
3821         if (!NT_STATUS_IS_OK(cli_nt_hardlink(targetcli, targetname, dest))) {
3822                 d_printf("%s doing an NT hard link of files\n",cli_errstr(targetcli));
3823                 return 1;
3824         }
3825
3826         return 0;
3827 }
3828
3829 /****************************************************************************
3830  Toggle the prompt flag.
3831 ****************************************************************************/
3832
3833 static int cmd_prompt(void)
3834 {
3835         prompt = !prompt;
3836         DEBUG(2,("prompting is now %s\n",prompt?"on":"off"));
3837         return 1;
3838 }
3839
3840 /****************************************************************************
3841  Set the newer than time.
3842 ****************************************************************************/
3843
3844 static int cmd_newer(void)
3845 {
3846         TALLOC_CTX *ctx = talloc_tos();
3847         char *buf;
3848         bool ok;
3849         SMB_STRUCT_STAT sbuf;
3850
3851         ok = next_token_talloc(ctx, &cmd_ptr,&buf,NULL);
3852         if (ok && (sys_stat(buf, &sbuf, false) == 0)) {
3853                 newer_than = convert_timespec_to_time_t(sbuf.st_ex_mtime);
3854                 DEBUG(1,("Getting files newer than %s",
3855                          time_to_asc(newer_than)));
3856         } else {
3857                 newer_than = 0;
3858         }
3859
3860         if (ok && newer_than == 0) {
3861                 d_printf("Error setting newer-than time\n");
3862                 return 1;
3863         }
3864
3865         return 0;
3866 }
3867
3868 /****************************************************************************
3869  Set the archive level.
3870 ****************************************************************************/
3871
3872 static int cmd_archive(void)
3873 {
3874         TALLOC_CTX *ctx = talloc_tos();
3875         char *buf;
3876
3877         if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
3878                 archive_level = atoi(buf);
3879         } else {
3880                 d_printf("Archive level is %d\n",archive_level);
3881         }
3882
3883         return 0;
3884 }
3885
3886 /****************************************************************************
3887  Toggle the lowercaseflag.
3888 ****************************************************************************/
3889
3890 static int cmd_lowercase(void)
3891 {
3892         lowercase = !lowercase;
3893         DEBUG(2,("filename lowercasing is now %s\n",lowercase?"on":"off"));
3894         return 0;
3895 }
3896
3897 /****************************************************************************
3898  Toggle the case sensitive flag.
3899 ****************************************************************************/
3900
3901 static int cmd_setcase(void)
3902 {
3903         bool orig_case_sensitive = cli_set_case_sensitive(cli, false);
3904
3905         cli_set_case_sensitive(cli, !orig_case_sensitive);
3906         DEBUG(2,("filename case sensitivity is now %s\n",!orig_case_sensitive ?
3907                 "on":"off"));
3908         return 0;
3909 }
3910
3911 /****************************************************************************
3912  Toggle the showacls flag.
3913 ****************************************************************************/
3914
3915 static int cmd_showacls(void)
3916 {
3917         showacls = !showacls;
3918         DEBUG(2,("showacls is now %s\n",showacls?"on":"off"));
3919         return 0;
3920 }
3921
3922
3923 /****************************************************************************
3924  Toggle the recurse flag.
3925 ****************************************************************************/
3926
3927 static int cmd_recurse(void)
3928 {
3929         recurse = !recurse;
3930         DEBUG(2,("directory recursion is now %s\n",recurse?"on":"off"));
3931         return 0;
3932 }
3933
3934 /****************************************************************************
3935  Toggle the translate flag.
3936 ****************************************************************************/
3937
3938 static int cmd_translate(void)
3939 {
3940         translation = !translation;
3941         DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
3942                  translation?"on":"off"));
3943         return 0;
3944 }
3945
3946 /****************************************************************************
3947  Do the lcd command.
3948  ****************************************************************************/
3949
3950 static int cmd_lcd(void)
3951 {
3952         TALLOC_CTX *ctx = talloc_tos();
3953         char *buf;
3954         char *d;
3955
3956         if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
3957                 if (chdir(buf) == -1) {
3958                         d_printf("chdir to %s failed (%s)\n",
3959                                 buf, strerror(errno));
3960                 }
3961         }
3962         d = sys_getwd();
3963         if (!d) {
3964                 return 1;
3965         }
3966         DEBUG(2,("the local directory is now %s\n",d));
3967         SAFE_FREE(d);
3968         return 0;
3969 }
3970
3971 /****************************************************************************
3972  Get a file restarting at end of local file.
3973  ****************************************************************************/
3974
3975 static int cmd_reget(void)
3976 {
3977         TALLOC_CTX *ctx = talloc_tos();
3978         char *local_name = NULL;
3979         char *remote_name = NULL;
3980         char *fname = NULL;
3981         char *p = NULL;
3982
3983         remote_name = talloc_strdup(ctx, client_get_cur_dir());
3984         if (!remote_name) {
3985                 return 1;
3986         }
3987
3988         if (!next_token_talloc(ctx, &cmd_ptr, &fname, NULL)) {
3989                 d_printf("reget <filename>\n");
3990                 return 1;
3991         }
3992         remote_name = talloc_asprintf_append(remote_name, "%s", fname);
3993         if (!remote_name) {
3994                 return 1;
3995         }
3996         remote_name = clean_name(ctx,remote_name);
3997         if (!remote_name) {
3998                 return 1;
3999         }
4000
4001         local_name = fname;
4002         next_token_talloc(ctx, &cmd_ptr, &p, NULL);
4003         if (p) {
4004                 local_name = p;
4005         }
4006
4007         return do_get(remote_name, local_name, true);
4008 }
4009
4010 /****************************************************************************
4011  Put a file restarting at end of local file.
4012  ****************************************************************************/
4013
4014 static int cmd_reput(void)
4015 {
4016         TALLOC_CTX *ctx = talloc_tos();
4017         char *local_name = NULL;
4018         char *remote_name = NULL;
4019         char *buf;
4020         SMB_STRUCT_STAT st;
4021
4022         remote_name = talloc_strdup(ctx, client_get_cur_dir());
4023         if (!remote_name) {
4024                 return 1;
4025         }
4026
4027         if (!next_token_talloc(ctx, &cmd_ptr, &local_name, NULL)) {
4028                 d_printf("reput <filename>\n");
4029                 return 1;
4030         }
4031
4032         if (!file_exist_stat(local_name, &st, false)) {
4033                 d_printf("%s does not exist\n", local_name);
4034                 return 1;
4035         }
4036
4037         if (next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) {
4038                 remote_name = talloc_asprintf_append(remote_name,
4039                                                 "%s", buf);
4040         } else {
4041                 remote_name = talloc_asprintf_append(remote_name,
4042                                                 "%s", local_name);
4043         }
4044         if (!remote_name) {
4045                 return 1;
4046         }
4047
4048         remote_name = clean_name(ctx, remote_name);
4049         if (!remote_name) {
4050                 return 1;
4051         }
4052
4053         return do_put(remote_name, local_name, true);
4054 }
4055
4056 /****************************************************************************
4057  List a share name.
4058  ****************************************************************************/
4059
4060 static void browse_fn(const char *name, uint32 m,
4061                       const char *comment, void *state)
4062 {
4063         const char *typestr = "";
4064
4065         switch (m & 7) {
4066         case STYPE_DISKTREE:
4067                 typestr = "Disk";
4068                 break;
4069         case STYPE_PRINTQ:
4070                 typestr = "Printer";
4071                 break;
4072         case STYPE_DEVICE:
4073                 typestr = "Device";
4074                 break;
4075         case STYPE_IPC:
4076                 typestr = "IPC";
4077                 break;
4078         }
4079         /* FIXME: If the remote machine returns non-ascii characters
4080            in any of these fields, they can corrupt the output.  We
4081            should remove them. */
4082         if (!grepable) {
4083                 d_printf("\t%-15s %-10.10s%s\n",
4084                         name,typestr,comment);
4085         } else {
4086                 d_printf ("%s|%s|%s\n",typestr,name,comment);
4087         }
4088 }
4089
4090 static bool browse_host_rpc(bool sort)
4091 {
4092         NTSTATUS status;
4093         struct rpc_pipe_client *pipe_hnd = NULL;
4094         TALLOC_CTX *frame = talloc_stackframe();
4095         WERROR werr;
4096         struct srvsvc_NetShareInfoCtr info_ctr;
4097         struct srvsvc_NetShareCtr1 ctr1;
4098         uint32_t resume_handle = 0;
4099         uint32_t total_entries = 0;
4100         int i;
4101         struct dcerpc_binding_handle *b;
4102
4103         status = cli_rpc_pipe_open_noauth(cli, &ndr_table_srvsvc.syntax_id,
4104                                           &pipe_hnd);
4105
4106         if (!NT_STATUS_IS_OK(status)) {
4107                 DEBUG(10, ("Could not connect to srvsvc pipe: %s\n",
4108                            nt_errstr(status)));
4109                 TALLOC_FREE(frame);
4110                 return false;
4111         }
4112
4113         b = pipe_hnd->binding_handle;
4114
4115         ZERO_STRUCT(info_ctr);
4116         ZERO_STRUCT(ctr1);
4117
4118         info_ctr.level = 1;
4119         info_ctr.ctr.ctr1 = &ctr1;
4120
4121         status = dcerpc_srvsvc_NetShareEnumAll(b, frame,
4122                                               pipe_hnd->desthost,
4123                                               &info_ctr,
4124                                               0xffffffff,
4125                                               &total_entries,
4126                                               &resume_handle,
4127                                               &werr);
4128
4129         if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(werr)) {
4130                 TALLOC_FREE(pipe_hnd);
4131                 TALLOC_FREE(frame);
4132                 return false;
4133         }
4134
4135         for (i=0; i < info_ctr.ctr.ctr1->count; i++) {
4136                 struct srvsvc_NetShareInfo1 info = info_ctr.ctr.ctr1->array[i];
4137                 browse_fn(info.name, info.type, info.comment, NULL);
4138         }
4139
4140         TALLOC_FREE(pipe_hnd);
4141         TALLOC_FREE(frame);
4142         return true;
4143 }
4144
4145 /****************************************************************************
4146  Try and browse available connections on a host.
4147 ****************************************************************************/
4148
4149 static bool browse_host(bool sort)
4150 {
4151         int ret;
4152         if (!grepable) {
4153                 d_printf("\n\tSharename       Type      Comment\n");
4154                 d_printf("\t---------       ----      -------\n");
4155         }
4156
4157         if (browse_host_rpc(sort)) {
4158                 return true;
4159         }
4160
4161         if((ret = cli_RNetShareEnum(cli, browse_fn, NULL)) == -1)
4162                 d_printf("Error returning browse list: %s\n", cli_errstr(cli));
4163
4164         return (ret != -1);
4165 }
4166
4167 /****************************************************************************
4168  List a server name.
4169 ****************************************************************************/
4170
4171 static void server_fn(const char *name, uint32 m,
4172                       const char *comment, void *state)
4173 {
4174
4175         if (!grepable){
4176                 d_printf("\t%-16s     %s\n", name, comment);
4177         } else {
4178                 d_printf("%s|%s|%s\n",(char *)state, name, comment);
4179         }
4180 }
4181
4182 /****************************************************************************
4183  Try and browse available connections on a host.
4184 ****************************************************************************/
4185
4186 static bool list_servers(const char *wk_grp)
4187 {
4188         fstring state;
4189
4190         if (!cli->server_domain)
4191                 return false;
4192
4193         if (!grepable) {
4194                 d_printf("\n\tServer               Comment\n");
4195                 d_printf("\t---------            -------\n");
4196         };
4197         fstrcpy( state, "Server" );
4198         cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_ALL, server_fn,
4199                           state);
4200
4201         if (!grepable) {
4202                 d_printf("\n\tWorkgroup            Master\n");
4203                 d_printf("\t---------            -------\n");
4204         };
4205
4206         fstrcpy( state, "Workgroup" );
4207         cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_DOMAIN_ENUM,
4208                           server_fn, state);
4209         return true;
4210 }
4211
4212 /****************************************************************************
4213  Print or set current VUID
4214 ****************************************************************************/
4215
4216 static int cmd_vuid(void)
4217 {
4218         TALLOC_CTX *ctx = talloc_tos();
4219         char *buf;
4220
4221         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
4222                 d_printf("Current VUID is %d\n", cli->vuid);
4223                 return 0;
4224         }
4225
4226         cli->vuid = atoi(buf);
4227         return 0;
4228 }
4229
4230 /****************************************************************************
4231  Setup a new VUID, by issuing a session setup
4232 ****************************************************************************/
4233
4234 static int cmd_logon(void)
4235 {
4236         TALLOC_CTX *ctx = talloc_tos();
4237         char *l_username, *l_password;
4238         NTSTATUS nt_status;
4239
4240         if (!next_token_talloc(ctx, &cmd_ptr,&l_username,NULL)) {
4241                 d_printf("logon <username> [<password>]\n");
4242                 return 0;
4243         }
4244
4245         if (!next_token_talloc(ctx, &cmd_ptr,&l_password,NULL)) {
4246                 char *pass = getpass("Password: ");
4247                 if (pass) {
4248                         l_password = talloc_strdup(ctx,pass);
4249                 }
4250         }
4251         if (!l_password) {
4252                 return 1;
4253         }
4254
4255         nt_status = cli_session_setup(cli, l_username,
4256                                       l_password, strlen(l_password),
4257                                       l_password, strlen(l_password),
4258                                       lp_workgroup());
4259         if (!NT_STATUS_IS_OK(nt_status)) {
4260                 d_printf("session setup failed: %s\n", nt_errstr(nt_status));
4261                 return -1;
4262         }
4263
4264         d_printf("Current VUID is %d\n", cli->vuid);
4265         return 0;
4266 }
4267
4268
4269 /****************************************************************************
4270  list active connections
4271 ****************************************************************************/
4272
4273 static int cmd_list_connect(void)
4274 {
4275         cli_cm_display(cli);
4276         return 0;
4277 }
4278
4279 /****************************************************************************
4280  display the current active client connection
4281 ****************************************************************************/
4282
4283 static int cmd_show_connect( void )
4284 {
4285         TALLOC_CTX *ctx = talloc_tos();
4286         struct cli_state *targetcli;
4287         char *targetpath;
4288         NTSTATUS status;
4289
4290         status = cli_resolve_path(ctx, "", auth_info, cli,
4291                                   client_get_cur_dir(), &targetcli,
4292                                   &targetpath);
4293         if (!NT_STATUS_IS_OK(status)) {
4294                 d_printf("showconnect %s: %s\n", cur_dir, cli_errstr(cli));
4295                 return 1;
4296         }
4297
4298         d_printf("//%s/%s\n", targetcli->desthost, targetcli->share);
4299         return 0;
4300 }
4301
4302 /****************************************************************************
4303  iosize command
4304 ***************************************************************************/
4305
4306 int cmd_iosize(void)
4307 {
4308         TALLOC_CTX *ctx = talloc_tos();
4309         char *buf;
4310         int iosize;
4311
4312         if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
4313                 if (!smb_encrypt) {
4314                         d_printf("iosize <n> or iosize 0x<n>. "
4315                                 "Minimum is 16384 (0x4000), "
4316                                 "max is 16776960 (0xFFFF00)\n");
4317                 } else {
4318                         d_printf("iosize <n> or iosize 0x<n>. "
4319                                 "(Encrypted connection) ,"
4320                                 "Minimum is 16384 (0x4000), "
4321                                 "max is 130048 (0x1FC00)\n");
4322                 }
4323                 return 1;
4324         }
4325
4326         iosize = strtol(buf,NULL,0);
4327         if (smb_encrypt && (iosize < 0x4000 || iosize > 0xFC00)) {
4328                 d_printf("iosize out of range for encrypted "
4329                         "connection (min = 16384 (0x4000), "
4330                         "max = 130048 (0x1FC00)");
4331                 return 1;
4332         } else if (!smb_encrypt && (iosize < 0x4000 || iosize > 0xFFFF00)) {
4333                 d_printf("iosize out of range (min = 16384 (0x4000), "
4334                         "max = 16776960 (0xFFFF00)");
4335                 return 1;
4336         }
4337
4338         io_bufsize = iosize;
4339         d_printf("iosize is now %d\n", io_bufsize);
4340         return 0;
4341 }
4342
4343 /****************************************************************************
4344 history
4345 ****************************************************************************/
4346 static int cmd_history(void)
4347 {
4348 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
4349         HIST_ENTRY **hlist;
4350         int i;
4351
4352         hlist = history_list();
4353
4354         for (i = 0; hlist && hlist[i]; i++) {
4355                 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
4356         }
4357 #else
4358         DEBUG(0,("no history without readline support\n"));
4359 #endif
4360
4361         return 0;
4362 }
4363
4364 /* Some constants for completing filename arguments */
4365
4366 #define COMPL_NONE        0          /* No completions */
4367 #define COMPL_REMOTE      1          /* Complete remote filename */
4368 #define COMPL_LOCAL       2          /* Complete local filename */
4369
4370 /* This defines the commands supported by this client.
4371  * NOTE: The "!" must be the last one in the list because it's fn pointer
4372  *       field is NULL, and NULL in that field is used in process_tok()
4373  *       (below) to indicate the end of the list.  crh
4374  */
4375 static struct {
4376         const char *name;
4377         int (*fn)(void);
4378         const char *description;
4379         char compl_args[2];      /* Completion argument info */
4380 } commands[] = {
4381   {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
4382   {"allinfo",cmd_allinfo,"<file> show all available info",
4383    {COMPL_NONE,COMPL_NONE}},
4384   {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
4385   {"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}},
4386   {"blocksize",cmd_block,"blocksize <number> (default 20)",{COMPL_NONE,COMPL_NONE}},
4387   {"cancel",cmd_cancel,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
4388   {"case_sensitive",cmd_setcase,"toggle the case sensitive flag to server",{COMPL_NONE,COMPL_NONE}},
4389   {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
4390   {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
4391   {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
4392   {"close",cmd_close,"<fid> close a file given a fid",{COMPL_REMOTE,COMPL_REMOTE}},
4393   {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
4394   {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
4395   {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
4396   {"echo",cmd_echo,"ping the server",{COMPL_NONE,COMPL_NONE}},
4397   {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
4398   {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
4399   {"getfacl",cmd_getfacl,"<file name> get the POSIX ACL on a file (UNIX extensions only)",{COMPL_REMOTE,COMPL_LOCAL}},
4400   {"geteas", cmd_geteas, "<file name> get the EA list of a file",
4401    {COMPL_REMOTE, COMPL_LOCAL}},
4402   {"hardlink",cmd_hardlink,"<src> <dest> create a Windows hard link",{COMPL_REMOTE,COMPL_REMOTE}},
4403   {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
4404   {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
4405   {"iosize",cmd_iosize,"iosize <number> (default 64512)",{COMPL_NONE,COMPL_NONE}},
4406   {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
4407   {"link",cmd_link,"<oldname> <newname> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
4408   {"lock",cmd_lock,"lock <fnum> [r|w] <hex-start> <hex-len> : set a POSIX lock",{COMPL_REMOTE,COMPL_REMOTE}},
4409   {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},
4410   {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
4411   {"l",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
4412   {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
4413   {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
4414   {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
4415   {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
4416   {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},
4417   {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
4418   {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
4419   {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
4420   {"posix", cmd_posix, "turn on all POSIX capabilities", {COMPL_REMOTE,COMPL_NONE}},
4421   {"posix_encrypt",cmd_posix_encrypt,"<domain> <user> <password> start up transport encryption",{COMPL_REMOTE,COMPL_NONE}},
4422   {"posix_open",cmd_posix_open,"<name> 0<mode> open_flags mode open a file using POSIX interface",{COMPL_REMOTE,COMPL_NONE}},
4423   {"posix_mkdir",cmd_posix_mkdir,"<name> 0<mode> creates a directory using POSIX interface",{COMPL_REMOTE,COMPL_NONE}},
4424   {"posix_rmdir",cmd_posix_rmdir,"<name> removes a directory using POSIX interface",{COMPL_REMOTE,COMPL_NONE}},
4425   {"posix_unlink",cmd_posix_unlink,"<name> removes a file using POSIX interface",{COMPL_REMOTE,COMPL_NONE}},
4426   {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
4427   {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},
4428   {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
4429   {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
4430   {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
4431   {"queue",cmd_queue,"show the print queue",{COMPL_NONE,COMPL_NONE}},
4432   {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
4433   {"readlink",cmd_readlink,"filename Do a UNIX extensions readlink call on a symlink",{COMPL_REMOTE,COMPL_REMOTE}},
4434   {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
4435   {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},
4436   {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
4437   {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
4438   {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
4439   {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
4440   {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
4441   {"showacls",cmd_showacls,"toggle if ACLs are shown or not",{COMPL_NONE,COMPL_NONE}},
4442   {"setea", cmd_setea, "<file name> <eaname> <eaval> Set an EA of a file",
4443    {COMPL_REMOTE, COMPL_LOCAL}},
4444   {"setmode",cmd_setmode,"filename <setmode string> change modes of file",{COMPL_REMOTE,COMPL_NONE}},
4445   {"stat",cmd_stat,"filename Do a UNIX extensions stat call on a file",{COMPL_REMOTE,COMPL_REMOTE}},
4446   {"symlink",cmd_symlink,"<oldname> <newname> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
4447   {"tar",cmd_tar,"tar <c|x>[IXFqbgNan] current directory to/from <file name>",{COMPL_NONE,COMPL_NONE}},
4448   {"tarmode",cmd_tarmode,"<full|inc|reset|noreset> tar's behaviour towards archive bits",{COMPL_NONE,COMPL_NONE}},
4449   {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
4450   {"unlock",cmd_unlock,"unlock <fnum> <hex-start> <hex-len> : remove a POSIX lock",{COMPL_REMOTE,COMPL_REMOTE}},
4451   {"volume",cmd_volume,"print the volume name",{COMPL_NONE,COMPL_NONE}},
4452   {"vuid",cmd_vuid,"change current vuid",{COMPL_NONE,COMPL_NONE}},
4453   {"wdel",cmd_wdel,"<attrib> <mask> wildcard delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
4454   {"logon",cmd_logon,"establish new logon",{COMPL_NONE,COMPL_NONE}},
4455   {"listconnect",cmd_list_connect,"list open connections",{COMPL_NONE,COMPL_NONE}},
4456   {"showconnect",cmd_show_connect,"display the current active connection",{COMPL_NONE,COMPL_NONE}},
4457   {"..",cmd_cd_oneup,"change the remote directory (up one level)",{COMPL_REMOTE,COMPL_NONE}},
4458
4459   /* Yes, this must be here, see crh's comment above. */
4460   {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
4461   {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
4462 };
4463
4464 /*******************************************************************
4465  Lookup a command string in the list of commands, including
4466  abbreviations.
4467 ******************************************************************/
4468
4469 static int process_tok(char *tok)
4470 {
4471         int i = 0, matches = 0;
4472         int cmd=0;
4473         int tok_len = strlen(tok);
4474
4475         while (commands[i].fn != NULL) {
4476                 if (strequal(commands[i].name,tok)) {
4477                         matches = 1;
4478                         cmd = i;
4479                         break;
4480                 } else if (strnequal(commands[i].name, tok, tok_len)) {
4481                         matches++;
4482                         cmd = i;
4483                 }
4484                 i++;
4485         }
4486
4487         if (matches == 0)
4488                 return(-1);
4489         else if (matches == 1)
4490                 return(cmd);
4491         else
4492                 return(-2);
4493 }
4494
4495 /****************************************************************************
4496  Help.
4497 ****************************************************************************/
4498
4499 static int cmd_help(void)
4500 {
4501         TALLOC_CTX *ctx = talloc_tos();
4502         int i=0,j;
4503         char *buf;
4504
4505         if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) {
4506                 if ((i = process_tok(buf)) >= 0)
4507                         d_printf("HELP %s:\n\t%s\n\n",
4508                                 commands[i].name,commands[i].description);
4509         } else {
4510                 while (commands[i].description) {
4511                         for (j=0; commands[i].description && (j<5); j++) {
4512                                 d_printf("%-15s",commands[i].name);
4513                                 i++;
4514                         }
4515                         d_printf("\n");
4516                 }
4517         }
4518         return 0;
4519 }
4520
4521 /****************************************************************************
4522  Process a -c command string.
4523 ****************************************************************************/
4524
4525 static int process_command_string(const char *cmd_in)
4526 {
4527         TALLOC_CTX *ctx = talloc_tos();
4528         char *cmd = talloc_strdup(ctx, cmd_in);
4529         int rc = 0;
4530
4531         if (!cmd) {
4532                 return 1;
4533         }
4534         /* establish the connection if not already */
4535
4536         if (!cli) {
4537                 NTSTATUS status;
4538
4539                 status = cli_cm_open(talloc_tos(), NULL,
4540                                      have_ip ? dest_ss_str : desthost,
4541                                      service, auth_info,
4542                                      true, smb_encrypt,
4543                                      max_protocol, port, name_type,
4544                                      &cli);
4545                 if (!NT_STATUS_IS_OK(status)) {
4546                         return 1;
4547                 }
4548         }
4549
4550         while (cmd[0] != '\0')    {
4551                 char *line;
4552                 char *p;
4553                 char *tok;
4554                 int i;
4555
4556                 if ((p = strchr_m(cmd, ';')) == 0) {
4557                         line = cmd;
4558                         cmd += strlen(cmd);
4559                 } else {
4560                         *p = '\0';
4561                         line = cmd;
4562                         cmd = p + 1;
4563                 }
4564
4565                 /* and get the first part of the command */
4566                 cmd_ptr = line;
4567                 if (!next_token_talloc(ctx, &cmd_ptr,&tok,NULL)) {
4568                         continue;
4569                 }
4570
4571                 if ((i = process_tok(tok)) >= 0) {
4572                         rc = commands[i].fn();
4573                 } else if (i == -2) {
4574                         d_printf("%s: command abbreviation ambiguous\n",tok);
4575                 } else {
4576                         d_printf("%s: command not found\n",tok);
4577                 }
4578         }
4579
4580         return rc;
4581 }
4582
4583 #define MAX_COMPLETIONS 100
4584
4585 struct completion_remote {
4586         char *dirmask;
4587         char **matches;
4588         int count, samelen;
4589         const char *text;
4590         int len;
4591 };
4592
4593 static NTSTATUS completion_remote_filter(const char *mnt,
4594                                 struct file_info *f,
4595                                 const char *mask,
4596                                 void *state)
4597 {
4598         struct completion_remote *info = (struct completion_remote *)state;
4599
4600         if (info->count >= MAX_COMPLETIONS - 1) {
4601                 return NT_STATUS_OK;
4602         }
4603         if (strncmp(info->text, f->name, info->len) != 0) {
4604                 return NT_STATUS_OK;
4605         }
4606         if (ISDOT(f->name) || ISDOTDOT(f->name)) {
4607                 return NT_STATUS_OK;
4608         }
4609
4610         if ((info->dirmask[0] == 0) && !(f->mode & FILE_ATTRIBUTE_DIRECTORY))
4611                 info->matches[info->count] = SMB_STRDUP(f->name);
4612         else {
4613                 TALLOC_CTX *ctx = talloc_stackframe();
4614                 char *tmp;
4615
4616                 tmp = talloc_strdup(ctx,info->dirmask);
4617                 if (!tmp) {
4618                         TALLOC_FREE(ctx);
4619                         return NT_STATUS_NO_MEMORY;
4620                 }
4621                 tmp = talloc_asprintf_append(tmp, "%s", f->name);
4622                 if (!tmp) {
4623                         TALLOC_FREE(ctx);
4624                         return NT_STATUS_NO_MEMORY;
4625                 }
4626                 if (f->mode & FILE_ATTRIBUTE_DIRECTORY) {
4627                         tmp = talloc_asprintf_append(tmp, "%s",
4628                                                      CLI_DIRSEP_STR);
4629                 }
4630                 if (!tmp) {
4631                         TALLOC_FREE(ctx);
4632                         return NT_STATUS_NO_MEMORY;
4633                 }
4634                 info->matches[info->count] = SMB_STRDUP(tmp);
4635                 TALLOC_FREE(ctx);
4636         }
4637         if (info->matches[info->count] == NULL) {
4638                 return NT_STATUS_OK;
4639         }
4640         if (f->mode & FILE_ATTRIBUTE_DIRECTORY) {
4641                 smb_readline_ca_char(0);
4642         }
4643         if (info->count == 1) {
4644                 info->samelen = strlen(info->matches[info->count]);
4645         } else {
4646                 while (strncmp(info->matches[info->count],
4647                                info->matches[info->count-1],
4648                                info->samelen) != 0) {
4649                         info->samelen--;
4650                 }
4651         }
4652         info->count++;
4653         return NT_STATUS_OK;
4654 }
4655
4656 static char **remote_completion(const char *text, int len)
4657 {
4658         TALLOC_CTX *ctx = talloc_stackframe();
4659         char *dirmask = NULL;
4660         char *targetpath = NULL;
4661         struct cli_state *targetcli = NULL;
4662         int i;
4663         struct completion_remote info = { NULL, NULL, 1, 0, NULL, 0 };
4664         NTSTATUS status;
4665
4666         /* can't have non-static initialisation on Sun CC, so do it
4667            at run time here */
4668         info.samelen = len;
4669         info.text = text;
4670         info.len = len;
4671
4672         info.matches = SMB_MALLOC_ARRAY(char *,MAX_COMPLETIONS);
4673         if (!info.matches) {
4674                 TALLOC_FREE(ctx);
4675                 return NULL;
4676         }
4677
4678         /*
4679          * We're leaving matches[0] free to fill it later with the text to
4680          * display: Either the one single match or the longest common subset
4681          * of the matches.
4682          */
4683         info.matches[0] = NULL;
4684         info.count = 1;
4685
4686         for (i = len-1; i >= 0; i--) {
4687                 if ((text[i] == '/') || (text[i] == CLI_DIRSEP_CHAR)) {
4688                         break;
4689                 }
4690         }
4691
4692         info.text = text+i+1;
4693         info.samelen = info.len = len-i-1;
4694
4695         if (i > 0) {
4696                 info.dirmask = SMB_MALLOC_ARRAY(char, i+2);
4697                 if (!info.dirmask) {
4698                         goto cleanup;
4699                 }
4700                 strncpy(info.dirmask, text, i+1);
4701                 info.dirmask[i+1] = 0;
4702                 dirmask = talloc_asprintf(ctx,
4703                                         "%s%*s*",
4704                                         client_get_cur_dir(),
4705                                         i-1,
4706                                         text);
4707         } else {
4708                 info.dirmask = SMB_STRDUP("");
4709                 if (!info.dirmask) {
4710                         goto cleanup;
4711                 }
4712                 dirmask = talloc_asprintf(ctx,
4713                                         "%s*",
4714                                         client_get_cur_dir());
4715         }
4716         if (!dirmask) {
4717                 goto cleanup;
4718         }
4719
4720         status = cli_resolve_path(ctx, "", auth_info, cli, dirmask, &targetcli,
4721                                   &targetpath);
4722         if (!NT_STATUS_IS_OK(status)) {
4723                 goto cleanup;
4724         }
4725         status = cli_list(targetcli, targetpath, FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN,
4726                           completion_remote_filter, (void *)&info);
4727         if (!NT_STATUS_IS_OK(status)) {
4728                 goto cleanup;
4729         }
4730
4731         if (info.count == 1) {
4732                 /*
4733                  * No matches at all, NULL indicates there is nothing
4734                  */
4735                 SAFE_FREE(info.matches[0]);
4736                 SAFE_FREE(info.matches);
4737                 TALLOC_FREE(ctx);
4738                 return NULL;
4739         }
4740
4741         if (info.count == 2) {
4742                 /*
4743                  * Exactly one match in matches[1], indicate this is the one
4744                  * in matches[0].
4745                  */
4746                 info.matches[0] = info.matches[1];
4747                 info.matches[1] = NULL;
4748                 info.count -= 1;
4749                 TALLOC_FREE(ctx);
4750                 return info.matches;
4751         }
4752
4753         /*
4754          * We got more than one possible match, set the result to the maximum
4755          * common subset
4756          */
4757
4758         info.matches[0] = SMB_STRNDUP(info.matches[1], info.samelen);
4759         info.matches[info.count] = NULL;
4760         return info.matches;
4761
4762 cleanup:
4763         for (i = 0; i < info.count; i++) {
4764                 SAFE_FREE(info.matches[i]);
4765         }
4766         SAFE_FREE(info.matches);
4767         SAFE_FREE(info.dirmask);
4768         TALLOC_FREE(ctx);
4769         return NULL;
4770 }
4771
4772 static char **completion_fn(const char *text, int start, int end)
4773 {
4774         smb_readline_ca_char(' ');
4775
4776         if (start) {
4777                 const char *buf, *sp;
4778                 int i;
4779                 char compl_type;
4780
4781                 buf = smb_readline_get_line_buffer();
4782                 if (buf == NULL)
4783                         return NULL;
4784
4785                 sp = strchr(buf, ' ');
4786                 if (sp == NULL)
4787                         return NULL;
4788
4789                 for (i = 0; commands[i].name; i++) {
4790                         if ((strncmp(commands[i].name, buf, sp - buf) == 0) &&
4791                             (commands[i].name[sp - buf] == 0)) {
4792                                 break;
4793                         }
4794                 }
4795                 if (commands[i].name == NULL)
4796                         return NULL;
4797
4798                 while (*sp == ' ')
4799                         sp++;
4800
4801                 if (sp == (buf + start))
4802                         compl_type = commands[i].compl_args[0];
4803                 else
4804                         compl_type = commands[i].compl_args[1];
4805
4806                 if (compl_type == COMPL_REMOTE)
4807                         return remote_completion(text, end - start);
4808                 else /* fall back to local filename completion */
4809                         return NULL;
4810         } else {
4811                 char **matches;
4812                 int i, len, samelen = 0, count=1;
4813
4814                 matches = SMB_MALLOC_ARRAY(char *, MAX_COMPLETIONS);
4815                 if (!matches) {
4816                         return NULL;
4817                 }
4818                 matches[0] = NULL;
4819
4820                 len = strlen(text);
4821                 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
4822                         if (strncmp(text, commands[i].name, len) == 0) {
4823                                 matches[count] = SMB_STRDUP(commands[i].name);
4824                                 if (!matches[count])
4825                                         goto cleanup;
4826                                 if (count == 1)
4827                                         samelen = strlen(matches[count]);
4828                                 else
4829                                         while (strncmp(matches[count], matches[count-1], samelen) != 0)
4830                                                 samelen--;
4831                                 count++;
4832                         }
4833                 }
4834
4835                 switch (count) {
4836                 case 0: /* should never happen */
4837                 case 1:
4838                         goto cleanup;
4839                 case 2:
4840                         matches[0] = SMB_STRDUP(matches[1]);
4841                         break;
4842                 default:
4843                         matches[0] = (char *)SMB_MALLOC(samelen+1);
4844                         if (!matches[0])
4845                                 goto cleanup;
4846                         strncpy(matches[0], matches[1], samelen);
4847                         matches[0][samelen] = 0;
4848                 }
4849                 matches[count] = NULL;
4850                 return matches;
4851
4852 cleanup:
4853                 for (i = 0; i < count; i++)
4854                         free(matches[i]);
4855
4856                 free(matches);
4857                 return NULL;
4858         }
4859 }
4860
4861 static bool finished;
4862
4863 /****************************************************************************
4864  Make sure we swallow keepalives during idle time.
4865 ****************************************************************************/
4866
4867 static void readline_callback(void)
4868 {
4869         static time_t last_t;
4870         struct timespec now;
4871         time_t t;
4872         int ret, revents;
4873
4874         clock_gettime_mono(&now);
4875         t = now.tv_sec;
4876
4877         if (t - last_t < 5)
4878                 return;
4879
4880         last_t = t;
4881
4882  again:
4883
4884         if (cli->fd == -1)
4885                 return;
4886
4887         /* We deliberately use receive_smb_raw instead of
4888            client_receive_smb as we want to receive
4889            session keepalives and then drop them here.
4890         */
4891
4892         ret = poll_intr_one_fd(cli->fd, POLLIN|POLLHUP, 0, &revents);
4893
4894         if ((ret > 0) && (revents & (POLLIN|POLLHUP|POLLERR))) {
4895                 NTSTATUS status;
4896                 size_t len;
4897
4898                 set_smb_read_error(&cli->smb_rw_error, SMB_READ_OK);
4899
4900                 status = receive_smb_raw(cli->fd, cli->inbuf, cli->bufsize, 0, 0, &len);
4901
4902                 if (!NT_STATUS_IS_OK(status)) {
4903                         DEBUG(0, ("Read from server failed, maybe it closed "
4904                                   "the connection\n"));
4905
4906                         finished = true;
4907                         smb_readline_done();
4908                         if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)) {
4909                                 set_smb_read_error(&cli->smb_rw_error,
4910                                                    SMB_READ_EOF);
4911                                 return;
4912                         }
4913
4914                         if (NT_STATUS_EQUAL(status, NT_STATUS_IO_TIMEOUT)) {
4915                                 set_smb_read_error(&cli->smb_rw_error,
4916                                                    SMB_READ_TIMEOUT);
4917                                 return;
4918                         }
4919
4920                         set_smb_read_error(&cli->smb_rw_error, SMB_READ_ERROR);
4921                         return;
4922                 }
4923                 if(CVAL(cli->inbuf,0) != SMBkeepalive) {
4924                         DEBUG(0, ("Read from server "
4925                                 "returned unexpected packet!\n"));
4926                         return;
4927                 }
4928
4929                 goto again;
4930         }
4931
4932         /* Ping the server to keep the connection alive using SMBecho. */
4933         {
4934                 NTSTATUS status;
4935                 unsigned char garbage[16];
4936                 memset(garbage, 0xf0, sizeof(garbage));
4937                 status = cli_echo(cli, 1, data_blob_const(garbage, sizeof(garbage)));
4938
4939                 if (!NT_STATUS_IS_OK(status)) {
4940                         DEBUG(0, ("SMBecho failed. Maybe server has closed "
4941                                 "the connection\n"));
4942                         finished = true;
4943                         smb_readline_done();
4944                 }
4945         }
4946 }
4947
4948 /****************************************************************************
4949  Process commands on stdin.
4950 ****************************************************************************/
4951
4952 static int process_stdin(void)
4953 {
4954         int rc = 0;
4955
4956         while (!finished) {
4957                 TALLOC_CTX *frame = talloc_stackframe();
4958                 char *tok = NULL;
4959                 char *the_prompt = NULL;
4960                 char *line = NULL;
4961                 int i;
4962
4963                 /* display a prompt */
4964                 if (asprintf(&the_prompt, "smb: %s> ", client_get_cur_dir()) < 0) {
4965                         TALLOC_FREE(frame);
4966                         break;
4967                 }
4968                 line = smb_readline(the_prompt, readline_callback, completion_fn);
4969                 SAFE_FREE(the_prompt);
4970                 if (!line) {
4971                         TALLOC_FREE(frame);
4972                         break;
4973                 }
4974
4975                 /* special case - first char is ! */
4976                 if (*line == '!') {
4977                         if (system(line + 1) == -1) {
4978                                 d_printf("system() command %s failed.\n",
4979                                         line+1);
4980                         }
4981                         SAFE_FREE(line);
4982                         TALLOC_FREE(frame);
4983                         continue;
4984                 }
4985
4986                 /* and get the first part of the command */
4987                 cmd_ptr = line;
4988                 if (!next_token_talloc(frame, &cmd_ptr,&tok,NULL)) {
4989                         TALLOC_FREE(frame);
4990                         SAFE_FREE(line);
4991                         continue;
4992                 }
4993
4994                 if ((i = process_tok(tok)) >= 0) {
4995                         rc = commands[i].fn();
4996                 } else if (i == -2) {
4997                         d_printf("%s: command abbreviation ambiguous\n",tok);
4998                 } else {
4999                         d_printf("%s: command not found\n",tok);
5000                 }
5001                 SAFE_FREE(line);
5002                 TALLOC_FREE(frame);
5003         }
5004         return rc;
5005 }
5006
5007 /****************************************************************************
5008  Process commands from the client.
5009 ****************************************************************************/
5010
5011 static int process(const char *base_directory)
5012 {
5013         int rc = 0;
5014         NTSTATUS status;
5015
5016         status = cli_cm_open(talloc_tos(), NULL,
5017                              have_ip ? dest_ss_str : desthost,
5018                              service, auth_info, true, smb_encrypt,
5019                              max_protocol, port, name_type, &cli);
5020         if (!NT_STATUS_IS_OK(status)) {
5021                 return 1;
5022         }
5023
5024         if (base_directory && *base_directory) {
5025                 rc = do_cd(base_directory);
5026                 if (rc) {
5027                         cli_shutdown(cli);
5028                         return rc;
5029                 }
5030         }
5031
5032         if (cmdstr) {
5033                 rc = process_command_string(cmdstr);
5034         } else {
5035                 process_stdin();
5036         }
5037
5038         cli_shutdown(cli);
5039         return rc;
5040 }
5041
5042 /****************************************************************************
5043  Handle a -L query.
5044 ****************************************************************************/
5045
5046 static int do_host_query(const char *query_host)
5047 {
5048         NTSTATUS status;
5049
5050         status = cli_cm_open(talloc_tos(), NULL,
5051                              have_ip ? dest_ss_str : query_host,
5052                              "IPC$", auth_info, true, smb_encrypt,
5053                              max_protocol, port, name_type, &cli);
5054         if (!NT_STATUS_IS_OK(status)) {
5055                 return 1;
5056         }
5057
5058         browse_host(true);
5059
5060         /* Ensure that the host can do IPv4 */
5061
5062         if (!interpret_addr(query_host)) {
5063                 struct sockaddr_storage ss;
5064                 if (interpret_string_addr(&ss, query_host, 0) &&
5065                                 (ss.ss_family != AF_INET)) {
5066                         d_printf("%s is an IPv6 address -- no workgroup available\n",
5067                                 query_host);
5068                         return 1;
5069                 }
5070         }
5071
5072         if (port != 139) {
5073
5074                 /* Workgroups simply don't make sense over anything
5075                    else but port 139... */
5076
5077                 cli_shutdown(cli);
5078                 status = cli_cm_open(talloc_tos(), NULL,
5079                                      have_ip ? dest_ss_str : query_host,
5080                                      "IPC$", auth_info, true, smb_encrypt,
5081                                      max_protocol, 139, name_type, &cli);
5082                 if (!NT_STATUS_IS_OK(status)) {
5083                         cli = NULL;
5084                 }
5085         }
5086
5087         if (cli == NULL) {
5088                 d_printf("NetBIOS over TCP disabled -- no workgroup available\n");
5089                 return 1;
5090         }
5091
5092         list_servers(lp_workgroup());
5093
5094         cli_shutdown(cli);
5095
5096         return(0);
5097 }
5098
5099 /****************************************************************************
5100  Handle a tar operation.
5101 ****************************************************************************/
5102
5103 static int do_tar_op(const char *base_directory)
5104 {
5105         int ret;
5106
5107         /* do we already have a connection? */
5108         if (!cli) {
5109                 NTSTATUS status;
5110
5111                 status = cli_cm_open(talloc_tos(), NULL,
5112                                      have_ip ? dest_ss_str : desthost,
5113                                      service, auth_info, true, smb_encrypt,
5114                                      max_protocol, port, name_type, &cli);
5115                 if (!NT_STATUS_IS_OK(status)) {
5116                         return 1;
5117                 }
5118         }
5119
5120         recurse=true;
5121
5122         if (base_directory && *base_directory)  {
5123                 ret = do_cd(base_directory);
5124                 if (ret) {
5125                         cli_shutdown(cli);
5126                         return ret;
5127                 }
5128         }
5129
5130         ret=process_tar();
5131
5132         cli_shutdown(cli);
5133
5134         return(ret);
5135 }
5136
5137 /****************************************************************************
5138  Handle a message operation.
5139 ****************************************************************************/
5140
5141 static int do_message_op(struct user_auth_info *a_info)
5142 {
5143         NTSTATUS status;
5144
5145         status = cli_connect_nb(desthost, have_ip ? &dest_ss : NULL,
5146                                 port ? port : 139, name_type,
5147                                 lp_netbios_name(), Undefined, &cli);
5148         if (!NT_STATUS_IS_OK(status)) {
5149                 d_printf("Connection to %s failed. Error %s\n", desthost, nt_errstr(status));
5150                 return 1;
5151         }
5152
5153         send_message(get_cmdline_auth_info_username(a_info));
5154         cli_shutdown(cli);
5155
5156         return 0;
5157 }
5158
5159 /****************************************************************************
5160   main program
5161 ****************************************************************************/
5162
5163  int main(int argc,char *argv[])
5164 {
5165         char *base_directory = NULL;
5166         int opt;
5167         char *query_host = NULL;
5168         bool message = false;
5169         static const char *new_name_resolve_order = NULL;
5170         poptContext pc;
5171         char *p;
5172         int rc = 0;
5173         bool tar_opt = false;
5174         bool service_opt = false;
5175         struct poptOption long_options[] = {
5176                 POPT_AUTOHELP
5177
5178                 { "name-resolve", 'R', POPT_ARG_STRING, &new_name_resolve_order, 'R', "Use these name resolution services only", "NAME-RESOLVE-ORDER" },
5179                 { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" },
5180                 { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" },
5181                 { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" },
5182                 { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" },
5183                 { "max-protocol", 'm', POPT_ARG_STRING, NULL, 'm', "Set the max protocol level", "LEVEL" },
5184                 { "tar", 'T', POPT_ARG_STRING, NULL, 'T', "Command line tar", "<c|x>IXFqgbNan" },
5185                 { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" },
5186                 { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" },
5187                 { "send-buffer", 'b', POPT_ARG_INT, &io_bufsize, 'b', "Changes the transmit/send buffer", "BYTES" },
5188                 { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" },
5189                 { "grepable", 'g', POPT_ARG_NONE, NULL, 'g', "Produce grepable output" },
5190                 { "browse", 'B', POPT_ARG_NONE, NULL, 'B', "Browse SMB servers using DNS" },
5191                 POPT_COMMON_SAMBA
5192                 POPT_COMMON_CONNECTION
5193                 POPT_COMMON_CREDENTIALS
5194                 POPT_TABLEEND
5195         };
5196         TALLOC_CTX *frame = talloc_stackframe();
5197
5198         if (!client_set_cur_dir("\\")) {
5199                 exit(ENOMEM);
5200         }
5201
5202         /* set default debug level to 1 regardless of what smb.conf sets */
5203         setup_logging( "smbclient", DEBUG_DEFAULT_STDERR );
5204         load_case_tables();
5205
5206         lp_set_cmdline("log level", "1");
5207
5208         auth_info = user_auth_info_init(frame);
5209         if (auth_info == NULL) {
5210                 exit(1);
5211         }
5212         popt_common_set_auth_info(auth_info);
5213
5214         /* skip argv(0) */
5215         pc = poptGetContext("smbclient", argc, (const char **) argv, long_options, 0);
5216         poptSetOtherOptionHelp(pc, "service <password>");
5217
5218         lp_set_in_client(true); /* Make sure that we tell lp_load we are */
5219
5220         while ((opt = poptGetNextOpt(pc)) != -1) {
5221
5222                 /* if the tar option has been called previouslt, now we need to eat out the leftovers */
5223                 /* I see no other way to keep things sane --SSS */
5224                 if (tar_opt == true) {
5225                         while (poptPeekArg(pc)) {
5226                                 poptGetArg(pc);
5227                         }
5228                         tar_opt = false;
5229                 }
5230
5231                 /* if the service has not yet been specified lets see if it is available in the popt stack */
5232                 if (!service_opt && poptPeekArg(pc)) {
5233                         service = talloc_strdup(frame, poptGetArg(pc));
5234                         if (!service) {
5235                                 exit(ENOMEM);
5236                         }
5237                         service_opt = true;
5238                 }
5239
5240                 /* if the service has already been retrieved then check if we have also a password */
5241                 if (service_opt
5242                     && (!get_cmdline_auth_info_got_pass(auth_info))
5243                     && poptPeekArg(pc)) {
5244                         set_cmdline_auth_info_password(auth_info,
5245                                                        poptGetArg(pc));
5246                 }
5247
5248                 switch (opt) {
5249                 case 'M':
5250                         /* Messages are sent to NetBIOS name type 0x3
5251                          * (Messenger Service).  Make sure we default
5252                          * to port 139 instead of port 445. srl,crh
5253                          */
5254                         name_type = 0x03;
5255                         desthost = talloc_strdup(frame,poptGetOptArg(pc));
5256                         if (!desthost) {
5257                                 exit(ENOMEM);
5258                         }
5259                         if( !port )
5260                                 port = 139;
5261                         message = true;
5262                         break;
5263                 case 'I':
5264                         {
5265                                 if (!interpret_string_addr(&dest_ss, poptGetOptArg(pc), 0)) {
5266                                         exit(1);
5267                                 }
5268                                 have_ip = true;
5269                                 print_sockaddr(dest_ss_str, sizeof(dest_ss_str), &dest_ss);
5270                         }
5271                         break;
5272                 case 'E':
5273                         setup_logging("smbclient", DEBUG_STDERR );
5274                         display_set_stderr();
5275                         break;
5276
5277                 case 'L':
5278                         query_host = talloc_strdup(frame, poptGetOptArg(pc));
5279                         if (!query_host) {
5280                                 exit(ENOMEM);
5281                         }
5282                         break;
5283                 case 'm':
5284                         max_protocol = interpret_protocol(poptGetOptArg(pc), max_protocol);
5285                         break;
5286                 case 'T':
5287                         /* We must use old option processing for this. Find the
5288                          * position of the -T option in the raw argv[]. */
5289                         {
5290                                 int i;
5291                                 for (i = 1; i < argc; i++) {
5292                                         if (strncmp("-T", argv[i],2)==0)
5293                                                 break;
5294                                 }
5295                                 i++;
5296                                 if (!tar_parseargs(argc, argv, poptGetOptArg(pc), i)) {
5297                                         poptPrintUsage(pc, stderr, 0);
5298                                         exit(1);
5299                                 }
5300                         }
5301                         /* this must be the last option, mark we have parsed it so that we know we have */
5302                         tar_opt = true;
5303                         break;
5304                 case 'D':
5305                         base_directory = talloc_strdup(frame, poptGetOptArg(pc));
5306                         if (!base_directory) {
5307                                 exit(ENOMEM);
5308                         }
5309                         break;
5310                 case 'g':
5311                         grepable=true;
5312                         break;
5313                 case 'e':
5314                         smb_encrypt=true;
5315                         break;
5316                 case 'B':
5317                         return(do_smb_browse());
5318
5319                 }
5320         }
5321
5322         /* We may still have some leftovers after the last popt option has been called */
5323         if (tar_opt == true) {
5324                 while (poptPeekArg(pc)) {
5325                         poptGetArg(pc);
5326                 }
5327                 tar_opt = false;
5328         }
5329
5330         /* if the service has not yet been specified lets see if it is available in the popt stack */
5331         if (!service_opt && poptPeekArg(pc)) {
5332                 service = talloc_strdup(frame,poptGetArg(pc));
5333                 if (!service) {
5334                         exit(ENOMEM);
5335                 }
5336                 service_opt = true;
5337         }
5338
5339         /* if the service has already been retrieved then check if we have also a password */
5340         if (service_opt
5341             && !get_cmdline_auth_info_got_pass(auth_info)
5342             && poptPeekArg(pc)) {
5343                 set_cmdline_auth_info_password(auth_info,
5344                                                poptGetArg(pc));
5345         }
5346
5347         if ( override_logfile )
5348                 setup_logging( lp_logfile(), DEBUG_FILE );
5349
5350         if (!lp_load(get_dyn_CONFIGFILE(),true,false,false,true)) {
5351                 fprintf(stderr, "%s: Can't load %s - run testparm to debug it\n",
5352                         argv[0], get_dyn_CONFIGFILE());
5353         }
5354
5355         if (get_cmdline_auth_info_use_machine_account(auth_info) &&
5356             !set_cmdline_auth_info_machine_account_creds(auth_info)) {
5357                 exit(-1);
5358         }
5359
5360         load_interfaces();
5361
5362         if (service_opt && service) {
5363                 size_t len;
5364
5365                 /* Convert any '/' characters in the service name to '\' characters */
5366                 string_replace(service, '/','\\');
5367                 if (count_chars(service,'\\') < 3) {
5368                         d_printf("\n%s: Not enough '\\' characters in service\n",service);
5369                         poptPrintUsage(pc, stderr, 0);
5370                         exit(1);
5371                 }
5372                 /* Remove trailing slashes */
5373                 len = strlen(service);
5374                 while(len > 0 && service[len - 1] == '\\') {
5375                         --len;
5376                         service[len] = '\0';
5377                 }
5378         }
5379
5380         smb_encrypt = get_cmdline_auth_info_smb_encrypt(auth_info);
5381         if (!init_names()) {
5382                 fprintf(stderr, "init_names() failed\n");
5383                 exit(1);
5384         }
5385
5386         if(new_name_resolve_order)
5387                 lp_set_name_resolve_order(new_name_resolve_order);
5388
5389         if (!tar_type && !query_host && !service && !message) {
5390                 poptPrintUsage(pc, stderr, 0);
5391                 exit(1);
5392         }
5393
5394         poptFreeContext(pc);
5395
5396         DEBUG(3,("Client started (version %s).\n", samba_version_string()));
5397
5398         /* Ensure we have a password (or equivalent). */
5399         set_cmdline_auth_info_getpass(auth_info);
5400
5401         if (tar_type) {
5402                 if (cmdstr)
5403                         process_command_string(cmdstr);
5404                 return do_tar_op(base_directory);
5405         }
5406
5407         if (query_host && *query_host) {
5408                 char *qhost = query_host;
5409                 char *slash;
5410
5411                 while (*qhost == '\\' || *qhost == '/')
5412                         qhost++;
5413
5414                 if ((slash = strchr_m(qhost, '/'))
5415                     || (slash = strchr_m(qhost, '\\'))) {
5416                         *slash = 0;
5417                 }
5418
5419                 if ((p=strchr_m(qhost, '#'))) {
5420                         *p = 0;
5421                         p++;
5422                         sscanf(p, "%x", &name_type);
5423                 }
5424
5425                 return do_host_query(qhost);
5426         }
5427
5428         if (message) {
5429                 return do_message_op(auth_info);
5430         }
5431
5432         if (process(base_directory)) {
5433                 return 1;
5434         }
5435
5436         TALLOC_FREE(frame);
5437         return rc;
5438 }