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