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