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