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