source3/client/client.c

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