source/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
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21 */
  22
  23 #define NO_SYSLOG
  24
  25 #include "includes.h"
  26 #include "client/client_proto.h"
  27 #ifndef REGISTER
  28 #define REGISTER 0
  29 #endif
  30
  31 struct cli_state *cli;
  32 extern BOOL in_client;
  33 static int port = 0;
  34 pstring cur_dir = "\\";
  35 static pstring cd_path = "";
  36 static pstring service;
  37 static pstring desthost;
  38 static pstring username;
  39 static pstring password;
  40 static BOOL use_kerberos;
  41 static BOOL got_pass;
  42 static BOOL grepable=False;
  43 static char *cmdstr = NULL;
  44
  45 static int io_bufsize = 64512;
  46
  47 static int name_type = 0x20;
  48 static int max_protocol = PROTOCOL_NT1;
  49
  50 static int process_tok(pstring tok);
  51 static int cmd_help(void);
  52
  53 /* 30 second timeout on most commands */
  54 #define CLIENT_TIMEOUT (30*1000)
  55 #define SHORT_TIMEOUT (5*1000)
  56
  57 /* value for unused fid field in trans2 secondary request */
  58 #define FID_UNUSED (0xFFFF)
  59
  60 time_t newer_than = 0;
  61 static int archive_level = 0;
  62
  63 static BOOL translation = False;
  64
  65 static BOOL have_ip;
  66
  67 /* clitar bits insert */
  68 extern int blocksize;
  69 extern BOOL tar_inc;
  70 extern BOOL tar_reset;
  71 /* clitar bits end */
  72
  73
  74 static BOOL prompt = True;
  75
  76 static int printmode = 1;
  77
  78 static BOOL recurse = False;
  79 BOOL lowercase = False;
  80
  81 static struct in_addr dest_ip;
  82
  83 #define SEPARATORS " \t\n\r"
  84
  85 static BOOL abort_mget = True;
  86
  87 static pstring fileselection = "";
  88
  89 extern file_info def_finfo;
  90
  91 /* timing globals */
  92 SMB_BIG_UINT get_total_size = 0;
  93 unsigned int get_total_time_ms = 0;
  94 static SMB_BIG_UINT put_total_size = 0;
  95 static unsigned int put_total_time_ms = 0;
  96
  97 /* totals globals */
  98 static double dir_total;
  99
 100 #define USENMB
 101
 102 /* some forward declarations */
 103 static struct cli_state *do_connect(const char *server, const char *share);
 104
 105 /****************************************************************************
 106  Write to a local file with CR/LF->LF translation if appropriate. Return the
 107  number taken from the buffer. This may not equal the number written.
 108 ****************************************************************************/
 109
 110 static int writefile(int f, char *b, int n)
 111 {
 112         int i;
 113
 114         if (!translation) {
 115                 return write(f,b,n);
 116         }
 117
 118         i = 0;
 119         while (i < n) {
 120                 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
 121                         b++;i++;
 122                 }
 123                 if (write(f, b, 1) != 1) {
 124                         break;
 125                 }
 126                 b++;
 127                 i++;
 128         }
 129
 130         return(i);
 131 }
 132
 133 /****************************************************************************
 134  Read from a file with LF->CR/LF translation if appropriate. Return the
 135  number read. read approx n bytes.
 136 ****************************************************************************/
 137
 138 static int readfile(char *b, int n, XFILE *f)
 139 {
 140         int i;
 141         int c;
 142
 143         if (!translation)
 144                 return x_fread(b,1,n,f);
 145
 146         i = 0;
 147         while (i < (n - 1) && (i < BUFFER_SIZE)) {
 148                 if ((c = x_getc(f)) == EOF) {
 149                         break;
 150                 }
 151
 152                 if (c == '\n') { /* change all LFs to CR/LF */
 153                         b[i++] = '\r';
 154                 }
 155
 156                 b[i++] = c;
 157         }
 158
 159         return(i);
 160 }
 161
 162 /****************************************************************************
 163  Send a message.
 164 ****************************************************************************/
 165
 166 static void send_message(void)
 167 {
 168         int total_len = 0;
 169         int grp_id;
 170
 171         if (!cli_message_start(cli, desthost, username, &grp_id)) {
 172                 d_printf("message start: %s\n", cli_errstr(cli));
 173                 return;
 174         }
 175
 176
 177         d_printf("Connected. Type your message, ending it with a Control-D\n");
 178
 179         while (!feof(stdin) && total_len < 1600) {
 180                 int maxlen = MIN(1600 - total_len,127);
 181                 pstring msg;
 182                 int l=0;
 183                 int c;
 184
 185                 ZERO_ARRAY(msg);
 186
 187                 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
 188                         if (c == '\n')
 189                                 msg[l++] = '\r';
 190                         msg[l] = c;
 191                 }
 192
 193                 if (!cli_message_text(cli, msg, l, grp_id)) {
 194                         d_printf("SMBsendtxt failed (%s)\n",cli_errstr(cli));
 195                         return;
 196                 }
 197
 198                 total_len += l;
 199         }
 200
 201         if (total_len >= 1600)
 202                 d_printf("the message was truncated to 1600 bytes\n");
 203         else
 204                 d_printf("sent %d bytes\n",total_len);
 205
 206         if (!cli_message_end(cli, grp_id)) {
 207                 d_printf("SMBsendend failed (%s)\n",cli_errstr(cli));
 208                 return;
 209         }
 210 }
 211
 212 /****************************************************************************
 213  Check the space on a device.
 214 ****************************************************************************/
 215
 216 static int do_dskattr(void)
 217 {
 218         int total, bsize, avail;
 219
 220         if (!cli_dskattr(cli, &bsize, &total, &avail)) {
 221                 d_printf("Error in dskattr: %s\n",cli_errstr(cli));
 222                 return 1;
 223         }
 224
 225         d_printf("\n\t\t%d blocks of size %d. %d blocks available\n",
 226                  total, bsize, avail);
 227
 228         return 0;
 229 }
 230
 231 /****************************************************************************
 232  Show cd/pwd.
 233 ****************************************************************************/
 234
 235 static int cmd_pwd(void)
 236 {
 237         d_printf("Current directory is %s",service);
 238         d_printf("%s\n",cur_dir);
 239         return 0;
 240 }
 241
 242 /****************************************************************************
 243  Change directory - inner section.
 244 ****************************************************************************/
 245
 246 static int do_cd(char *newdir)
 247 {
 248         char *p = newdir;
 249         pstring saved_dir;
 250         pstring dname;
 251
 252         dos_format(newdir);
 253
 254         /* Save the current directory in case the
 255            new directory is invalid */
 256         pstrcpy(saved_dir, cur_dir);
 257         if (*p == '\\')
 258                 pstrcpy(cur_dir,p);
 259         else
 260                 pstrcat(cur_dir,p);
 261         if (*(cur_dir+strlen(cur_dir)-1) != '\\') {
 262                 pstrcat(cur_dir, "\\");
 263         }
 264         dos_clean_name(cur_dir);
 265         pstrcpy(dname,cur_dir);
 266         pstrcat(cur_dir,"\\");
 267         dos_clean_name(cur_dir);
 268
 269         if (!strequal(cur_dir,"\\")) {
 270                 if (!cli_chkpath(cli, dname)) {
 271                         d_printf("cd %s: %s\n", dname, cli_errstr(cli));
 272                         pstrcpy(cur_dir,saved_dir);
 273                 }
 274         }
 275
 276         pstrcpy(cd_path,cur_dir);
 277
 278         return 0;
 279 }
 280
 281 /****************************************************************************
 282  Change directory.
 283 ****************************************************************************/
 284
 285 static int cmd_cd(void)
 286 {
 287         pstring buf;
 288         int rc = 0;
 289
 290         if (next_token_nr(NULL,buf,NULL,sizeof(buf)))
 291                 rc = do_cd(buf);
 292         else
 293                 d_printf("Current directory is %s\n",cur_dir);
 294
 295         return rc;
 296 }
 297
 298 /*******************************************************************
 299  Decide if a file should be operated on.
 300 ********************************************************************/
 301
 302 static BOOL do_this_one(file_info *finfo)
 303 {
 304         if (finfo->mode & aDIR)
 305                 return(True);
 306
 307         if (*fileselection &&
 308             !mask_match(finfo->name,fileselection,False)) {
 309                 DEBUG(3,("mask_match %s failed\n", finfo->name));
 310                 return False;
 311         }
 312
 313         if (newer_than && finfo->mtime < newer_than) {
 314                 DEBUG(3,("newer_than %s failed\n", finfo->name));
 315                 return(False);
 316         }
 317
 318         if ((archive_level==1 || archive_level==2) && !(finfo->mode & aARCH)) {
 319                 DEBUG(3,("archive %s failed\n", finfo->name));
 320                 return(False);
 321         }
 322
 323         return(True);
 324 }
 325
 326 /****************************************************************************
 327  Display info about a file.
 328 ****************************************************************************/
 329
 330 static void display_finfo(file_info *finfo)
 331 {
 332         if (do_this_one(finfo)) {
 333                 time_t t = finfo->mtime; /* the time is assumed to be passed as GMT */
 334                 d_printf("  %-30s%7.7s %8.0f  %s",
 335                          finfo->name,
 336                          attrib_string(finfo->mode),
 337                          (double)finfo->size,
 338                          asctime(LocalTime(&t)));
 339                 dir_total += finfo->size;
 340         }
 341 }
 342
 343 /****************************************************************************
 344  Accumulate size of a file.
 345 ****************************************************************************/
 346
 347 static void do_du(file_info *finfo)
 348 {
 349         if (do_this_one(finfo)) {
 350                 dir_total += finfo->size;
 351         }
 352 }
 353
 354 static BOOL do_list_recurse;
 355 static BOOL do_list_dirs;
 356 static char *do_list_queue = 0;
 357 static long do_list_queue_size = 0;
 358 static long do_list_queue_start = 0;
 359 static long do_list_queue_end = 0;
 360 static void (*do_list_fn)(file_info *);
 361
 362 /****************************************************************************
 363  Functions for do_list_queue.
 364 ****************************************************************************/
 365
 366 /*
 367  * The do_list_queue is a NUL-separated list of strings stored in a
 368  * char*.  Since this is a FIFO, we keep track of the beginning and
 369  * ending locations of the data in the queue.  When we overflow, we
 370  * double the size of the char*.  When the start of the data passes
 371  * the midpoint, we move everything back.  This is logically more
 372  * complex than a linked list, but easier from a memory management
 373  * angle.  In any memory error condition, do_list_queue is reset.
 374  * Functions check to ensure that do_list_queue is non-NULL before
 375  * accessing it.
 376  */
 377
 378 static void reset_do_list_queue(void)
 379 {
 380         SAFE_FREE(do_list_queue);
 381         do_list_queue_size = 0;
 382         do_list_queue_start = 0;
 383         do_list_queue_end = 0;
 384 }
 385
 386 static void init_do_list_queue(void)
 387 {
 388         reset_do_list_queue();
 389         do_list_queue_size = 1024;
 390         do_list_queue = SMB_MALLOC(do_list_queue_size);
 391         if (do_list_queue == 0) {
 392                 d_printf("malloc fail for size %d\n",
 393                          (int)do_list_queue_size);
 394                 reset_do_list_queue();
 395         } else {
 396                 memset(do_list_queue, 0, do_list_queue_size);
 397         }
 398 }
 399
 400 static void adjust_do_list_queue(void)
 401 {
 402         /*
 403          * If the starting point of the queue is more than half way through,
 404          * move everything toward the beginning.
 405          */
 406         if (do_list_queue && (do_list_queue_start == do_list_queue_end)) {
 407                 DEBUG(4,("do_list_queue is empty\n"));
 408                 do_list_queue_start = do_list_queue_end = 0;
 409                 *do_list_queue = '\0';
 410         } else if (do_list_queue_start > (do_list_queue_size / 2)) {
 411                 DEBUG(4,("sliding do_list_queue backward\n"));
 412                 memmove(do_list_queue,
 413                         do_list_queue + do_list_queue_start,
 414                         do_list_queue_end - do_list_queue_start);
 415                 do_list_queue_end -= do_list_queue_start;
 416                 do_list_queue_start = 0;
 417         }
 418 }
 419
 420 static void add_to_do_list_queue(const char* entry)
 421 {
 422         char *dlq;
 423         long new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
 424         while (new_end > do_list_queue_size) {
 425                 do_list_queue_size *= 2;
 426                 DEBUG(4,("enlarging do_list_queue to %d\n",
 427                          (int)do_list_queue_size));
 428                 dlq = SMB_REALLOC(do_list_queue, do_list_queue_size);
 429                 if (! dlq) {
 430                         d_printf("failure enlarging do_list_queue to %d bytes\n",
 431                                  (int)do_list_queue_size);
 432                         reset_do_list_queue();
 433                 } else {
 434                         do_list_queue = dlq;
 435                         memset(do_list_queue + do_list_queue_size / 2,
 436                                0, do_list_queue_size / 2);
 437                 }
 438         }
 439         if (do_list_queue) {
 440                 safe_strcpy_base(do_list_queue + do_list_queue_end,
 441                                  entry, do_list_queue, do_list_queue_size);
 442                 do_list_queue_end = new_end;
 443                 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
 444                          entry, (int)do_list_queue_start, (int)do_list_queue_end));
 445         }
 446 }
 447
 448 static char *do_list_queue_head(void)
 449 {
 450         return do_list_queue + do_list_queue_start;
 451 }
 452
 453 static void remove_do_list_queue_head(void)
 454 {
 455         if (do_list_queue_end > do_list_queue_start) {
 456                 do_list_queue_start += strlen(do_list_queue_head()) + 1;
 457                 adjust_do_list_queue();
 458                 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
 459                          (int)do_list_queue_start, (int)do_list_queue_end));
 460         }
 461 }
 462
 463 static int do_list_queue_empty(void)
 464 {
 465         return (! (do_list_queue && *do_list_queue));
 466 }
 467
 468 /****************************************************************************
 469  A helper for do_list.
 470 ****************************************************************************/
 471
 472 static void do_list_helper(file_info *f, const char *mask, void *state)
 473 {
 474         if (f->mode & aDIR) {
 475                 if (do_list_dirs && do_this_one(f)) {
 476                         do_list_fn(f);
 477                 }
 478                 if (do_list_recurse &&
 479                     !strequal(f->name,".") &&
 480                     !strequal(f->name,"..")) {
 481                         pstring mask2;
 482                         char *p;
 483
 484                         if (!f->name[0]) {
 485                                 d_printf("Empty dir name returned. Possible server misconfiguration.\n");
 486                                 return;
 487                         }
 488
 489                         pstrcpy(mask2, mask);
 490                         p = strrchr_m(mask2,'\\');
 491                         if (!p)
 492                                 return;
 493                         p[1] = 0;
 494                         pstrcat(mask2, f->name);
 495                         pstrcat(mask2,"\\*");
 496                         add_to_do_list_queue(mask2);
 497                 }
 498                 return;
 499         }
 500
 501         if (do_this_one(f)) {
 502                 do_list_fn(f);
 503         }
 504 }
 505
 506 /****************************************************************************
 507  A wrapper around cli_list that adds recursion.
 508 ****************************************************************************/
 509
 510 void do_list(const char *mask,uint16 attribute,void (*fn)(file_info *),BOOL rec, BOOL dirs)
 511 {
 512         static int in_do_list = 0;
 513
 514         if (in_do_list && rec) {
 515                 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
 516                 exit(1);
 517         }
 518
 519         in_do_list = 1;
 520
 521         do_list_recurse = rec;
 522         do_list_dirs = dirs;
 523         do_list_fn = fn;
 524
 525         if (rec) {
 526                 init_do_list_queue();
 527                 add_to_do_list_queue(mask);
 528
 529                 while (! do_list_queue_empty()) {
 530                         /*
 531                          * Need to copy head so that it doesn't become
 532                          * invalid inside the call to cli_list.  This
 533                          * would happen if the list were expanded
 534                          * during the call.
 535                          * Fix from E. Jay Berkenbilt (ejb@ql.org)
 536                          */
 537                         pstring head;
 538                         pstrcpy(head, do_list_queue_head());
 539                         cli_list(cli, head, attribute, do_list_helper, NULL);
 540                         remove_do_list_queue_head();
 541                         if ((! do_list_queue_empty()) && (fn == display_finfo)) {
 542                                 char* next_file = do_list_queue_head();
 543                                 char* save_ch = 0;
 544                                 if ((strlen(next_file) >= 2) &&
 545                                     (next_file[strlen(next_file) - 1] == '*') &&
 546                                     (next_file[strlen(next_file) - 2] == '\\')) {
 547                                         save_ch = next_file +
 548                                                 strlen(next_file) - 2;
 549                                         *save_ch = '\0';
 550                                 }
 551                                 d_printf("\n%s\n",next_file);
 552                                 if (save_ch) {
 553                                         *save_ch = '\\';
 554                                 }
 555                         }
 556                 }
 557         } else {
 558                 if (cli_list(cli, mask, attribute, do_list_helper, NULL) == -1) {
 559                         d_printf("%s listing %s\n", cli_errstr(cli), mask);
 560                 }
 561         }
 562
 563         in_do_list = 0;
 564         reset_do_list_queue();
 565 }
 566
 567 /****************************************************************************
 568  Get a directory listing.
 569 ****************************************************************************/
 570
 571 static int cmd_dir(void)
 572 {
 573         uint16 attribute = aDIR | aSYSTEM | aHIDDEN;
 574         pstring mask;
 575         pstring buf;
 576         char *p=buf;
 577         int rc;
 578
 579         dir_total = 0;
 580         if (strcmp(cur_dir, "\\") != 0) {
 581                 pstrcpy(mask,cur_dir);
 582                 if(mask[strlen(mask)-1]!='\\')
 583                         pstrcat(mask,"\\");
 584         } else {
 585                 *mask = '\0';
 586         }
 587
 588         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
 589                 dos_format(p);
 590                 if (*p == '\\')
 591                         pstrcpy(mask,p);
 592                 else
 593                         pstrcat(mask,p);
 594         } else {
 595                 pstrcat(mask,"\\*");
 596         }
 597
 598         do_list(mask, attribute, display_finfo, recurse, True);
 599
 600         rc = do_dskattr();
 601
 602         DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
 603
 604         return rc;
 605 }
 606
 607 /****************************************************************************
 608  Get a directory listing.
 609 ****************************************************************************/
 610
 611 static int cmd_du(void)
 612 {
 613         uint16 attribute = aDIR | aSYSTEM | aHIDDEN;
 614         pstring mask;
 615         pstring buf;
 616         char *p=buf;
 617         int rc;
 618
 619         dir_total = 0;
 620         pstrcpy(mask,cur_dir);
 621         if(mask[strlen(mask)-1]!='\\')
 622                 pstrcat(mask,"\\");
 623
 624         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
 625                 dos_format(p);
 626                 if (*p == '\\')
 627                         pstrcpy(mask,p);
 628                 else
 629                         pstrcat(mask,p);
 630         } else {
 631                 pstrcat(mask,"*");
 632         }
 633
 634         do_list(mask, attribute, do_du, recurse, True);
 635
 636         rc = do_dskattr();
 637
 638         d_printf("Total number of bytes: %.0f\n", dir_total);
 639
 640         return rc;
 641 }
 642
 643 /****************************************************************************
 644  Get a file from rname to lname
 645 ****************************************************************************/
 646
 647 static int do_get(char *rname, char *lname, BOOL reget)
 648 {
 649         int handle = 0, fnum;
 650         BOOL newhandle = False;
 651         char *data;
 652         struct timeval tp_start;
 653         int read_size = io_bufsize;
 654         uint16 attr;
 655         size_t size;
 656         off_t start = 0;
 657         off_t nread = 0;
 658         int rc = 0;
 659
 660         GetTimeOfDay(&tp_start);
 661
 662         if (lowercase) {
 663                 strlower_m(lname);
 664         }
 665
 666         fnum = cli_open(cli, rname, O_RDONLY, DENY_NONE);
 667
 668         if (fnum == -1) {
 669                 d_printf("%s opening remote file %s\n",cli_errstr(cli),rname);
 670                 return 1;
 671         }
 672
 673         if(!strcmp(lname,"-")) {
 674                 handle = fileno(stdout);
 675         } else {
 676                 if (reget) {
 677                         handle = sys_open(lname, O_WRONLY|O_CREAT, 0644);
 678                         if (handle >= 0) {
 679                                 start = sys_lseek(handle, 0, SEEK_END);
 680                                 if (start == -1) {
 681                                         d_printf("Error seeking local file\n");
 682                                         return 1;
 683                                 }
 684                         }
 685                 } else {
 686                         handle = sys_open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
 687                 }
 688                 newhandle = True;
 689         }
 690         if (handle < 0) {
 691                 d_printf("Error opening local file %s\n",lname);
 692                 return 1;
 693         }
 694
 695
 696         if (!cli_qfileinfo(cli, fnum,
 697                            &attr, &size, NULL, NULL, NULL, NULL, NULL) &&
 698             !cli_getattrE(cli, fnum,
 699                           &attr, &size, NULL, NULL, NULL)) {
 700                 d_printf("getattrib: %s\n",cli_errstr(cli));
 701                 return 1;
 702         }
 703
 704         DEBUG(1,("getting file %s of size %.0f as %s ",
 705                  rname, (double)size, lname));
 706
 707         if(!(data = (char *)SMB_MALLOC(read_size))) {
 708                 d_printf("malloc fail for size %d\n", read_size);
 709                 cli_close(cli, fnum);
 710                 return 1;
 711         }
 712
 713         while (1) {
 714                 int n = cli_read(cli, fnum, data, nread + start, read_size);
 715
 716                 if (n <= 0)
 717                         break;
 718
 719                 if (writefile(handle,data, n) != n) {
 720                         d_printf("Error writing local file\n");
 721                         rc = 1;
 722                         break;
 723                 }
 724
 725                 nread += n;
 726         }
 727
 728         if (nread + start < size) {
 729                 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
 730                             rname, (long)nread));
 731
 732                 rc = 1;
 733         }
 734
 735         SAFE_FREE(data);
 736
 737         if (!cli_close(cli, fnum)) {
 738                 d_printf("Error %s closing remote file\n",cli_errstr(cli));
 739                 rc = 1;
 740         }
 741
 742         if (newhandle) {
 743                 close(handle);
 744         }
 745
 746         if (archive_level >= 2 && (attr & aARCH)) {
 747                 cli_setatr(cli, rname, attr & ~(uint16)aARCH, 0);
 748         }
 749
 750         {
 751                 struct timeval tp_end;
 752                 int this_time;
 753
 754                 GetTimeOfDay(&tp_end);
 755                 this_time =
 756                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
 757                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
 758                 get_total_time_ms += this_time;
 759                 get_total_size += nread;
 760
 761                 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
 762                          nread / (1.024*this_time + 1.0e-4),
 763                          get_total_size / (1.024*get_total_time_ms)));
 764         }
 765
 766         return rc;
 767 }
 768
 769 /****************************************************************************
 770  Get a file.
 771 ****************************************************************************/
 772
 773 static int cmd_get(void)
 774 {
 775         pstring lname;
 776         pstring rname;
 777         char *p;
 778
 779         pstrcpy(rname,cur_dir);
 780         pstrcat(rname,"\\");
 781
 782         p = rname + strlen(rname);
 783
 784         if (!next_token_nr(NULL,p,NULL,sizeof(rname)-strlen(rname))) {
 785                 d_printf("get <filename>\n");
 786                 return 1;
 787         }
 788         pstrcpy(lname,p);
 789         dos_clean_name(rname);
 790
 791         next_token_nr(NULL,lname,NULL,sizeof(lname));
 792
 793         return do_get(rname, lname, False);
 794 }
 795
 796 /****************************************************************************
 797  Do an mget operation on one file.
 798 ****************************************************************************/
 799
 800 static void do_mget(file_info *finfo)
 801 {
 802         pstring rname;
 803         pstring quest;
 804         pstring saved_curdir;
 805         pstring mget_mask;
 806
 807         if (strequal(finfo->name,".") || strequal(finfo->name,".."))
 808                 return;
 809
 810         if (abort_mget) {
 811                 d_printf("mget aborted\n");
 812                 return;
 813         }
 814
 815         if (finfo->mode & aDIR)
 816                 slprintf(quest,sizeof(pstring)-1,
 817                          "Get directory %s? ",finfo->name);
 818         else
 819                 slprintf(quest,sizeof(pstring)-1,
 820                          "Get file %s? ",finfo->name);
 821
 822         if (prompt && !yesno(quest))
 823                 return;
 824
 825         if (!(finfo->mode & aDIR)) {
 826                 pstrcpy(rname,cur_dir);
 827                 pstrcat(rname,finfo->name);
 828                 do_get(rname, finfo->name, False);
 829                 return;
 830         }
 831
 832         /* handle directories */
 833         pstrcpy(saved_curdir,cur_dir);
 834
 835         pstrcat(cur_dir,finfo->name);
 836         pstrcat(cur_dir,"\\");
 837
 838         unix_format(finfo->name);
 839         if (lowercase)
 840                 strlower_m(finfo->name);
 841
 842         if (!directory_exist(finfo->name,NULL) &&
 843             mkdir(finfo->name,0777) != 0) {
 844                 d_printf("failed to create directory %s\n",finfo->name);
 845                 pstrcpy(cur_dir,saved_curdir);
 846                 return;
 847         }
 848
 849         if (chdir(finfo->name) != 0) {
 850                 d_printf("failed to chdir to directory %s\n",finfo->name);
 851                 pstrcpy(cur_dir,saved_curdir);
 852                 return;
 853         }
 854
 855         pstrcpy(mget_mask,cur_dir);
 856         pstrcat(mget_mask,"*");
 857
 858         do_list(mget_mask, aSYSTEM | aHIDDEN | aDIR,do_mget,False, True);
 859         chdir("..");
 860         pstrcpy(cur_dir,saved_curdir);
 861 }
 862
 863 /****************************************************************************
 864  View the file using the pager.
 865 ****************************************************************************/
 866
 867 static int cmd_more(void)
 868 {
 869         pstring rname,lname,pager_cmd;
 870         char *pager;
 871         int fd;
 872         int rc = 0;
 873
 874         pstrcpy(rname,cur_dir);
 875         pstrcat(rname,"\\");
 876
 877         slprintf(lname,sizeof(lname)-1, "%s/smbmore.XXXXXX",tmpdir());
 878         fd = smb_mkstemp(lname);
 879         if (fd == -1) {
 880                 d_printf("failed to create temporary file for more\n");
 881                 return 1;
 882         }
 883         close(fd);
 884
 885         if (!next_token_nr(NULL,rname+strlen(rname),NULL,sizeof(rname)-strlen(rname))) {
 886                 d_printf("more <filename>\n");
 887                 unlink(lname);
 888                 return 1;
 889         }
 890         dos_clean_name(rname);
 891
 892         rc = do_get(rname, lname, False);
 893
 894         pager=getenv("PAGER");
 895
 896         slprintf(pager_cmd,sizeof(pager_cmd)-1,
 897                  "%s %s",(pager? pager:PAGER), lname);
 898         system(pager_cmd);
 899         unlink(lname);
 900
 901         return rc;
 902 }
 903
 904 /****************************************************************************
 905  Do a mget command.
 906 ****************************************************************************/
 907
 908 static int cmd_mget(void)
 909 {
 910         uint16 attribute = aSYSTEM | aHIDDEN;
 911         pstring mget_mask;
 912         pstring buf;
 913         char *p=buf;
 914
 915         *mget_mask = 0;
 916
 917         if (recurse)
 918                 attribute |= aDIR;
 919
 920         abort_mget = False;
 921
 922         while (next_token_nr(NULL,p,NULL,sizeof(buf))) {
 923                 pstrcpy(mget_mask,cur_dir);
 924                 if(mget_mask[strlen(mget_mask)-1]!='\\')
 925                         pstrcat(mget_mask,"\\");
 926
 927                 if (*p == '\\')
 928                         pstrcpy(mget_mask,p);
 929                 else
 930                         pstrcat(mget_mask,p);
 931                 do_list(mget_mask, attribute,do_mget,False,True);
 932         }
 933
 934         if (!*mget_mask) {
 935                 pstrcpy(mget_mask,cur_dir);
 936                 if(mget_mask[strlen(mget_mask)-1]!='\\')
 937                         pstrcat(mget_mask,"\\");
 938                 pstrcat(mget_mask,"*");
 939                 do_list(mget_mask, attribute,do_mget,False,True);
 940         }
 941
 942         return 0;
 943 }
 944
 945 /****************************************************************************
 946  Make a directory of name "name".
 947 ****************************************************************************/
 948
 949 static BOOL do_mkdir(char *name)
 950 {
 951         if (!cli_mkdir(cli, name)) {
 952                 d_printf("%s making remote directory %s\n",
 953                          cli_errstr(cli),name);
 954                 return(False);
 955         }
 956
 957         return(True);
 958 }
 959
 960 /****************************************************************************
 961  Show 8.3 name of a file.
 962 ****************************************************************************/
 963
 964 static BOOL do_altname(char *name)
 965 {
 966         pstring altname;
 967         if (!NT_STATUS_IS_OK(cli_qpathinfo_alt_name(cli, name, altname))) {
 968                 d_printf("%s getting alt name for %s\n",
 969                          cli_errstr(cli),name);
 970                 return(False);
 971         }
 972         d_printf("%s\n", altname);
 973
 974         return(True);
 975 }
 976
 977 /****************************************************************************
 978  Exit client.
 979 ****************************************************************************/
 980
 981 static int cmd_quit(void)
 982 {
 983         cli_shutdown(cli);
 984         exit(0);
 985         /* NOTREACHED */
 986         return 0;
 987 }
 988
 989 /****************************************************************************
 990  Make a directory.
 991 ****************************************************************************/
 992
 993 static int cmd_mkdir(void)
 994 {
 995         pstring mask;
 996         pstring buf;
 997         char *p=buf;
 998
 999         pstrcpy(mask,cur_dir);
1000
1001         if (!next_token_nr(NULL,p,NULL,sizeof(buf))) {
1002                 if (!recurse)
1003                         d_printf("mkdir <dirname>\n");
1004                 return 1;
1005         }
1006         pstrcat(mask,p);
1007
1008         if (recurse) {
1009                 pstring ddir;
1010                 pstring ddir2;
1011                 *ddir2 = 0;
1012
1013                 pstrcpy(ddir,mask);
1014                 trim_char(ddir,'.','\0');
1015                 p = strtok(ddir,"/\\");
1016                 while (p) {
1017                         pstrcat(ddir2,p);
1018                         if (!cli_chkpath(cli, ddir2)) {
1019                                 do_mkdir(ddir2);
1020                         }
1021                         pstrcat(ddir2,"\\");
1022                         p = strtok(NULL,"/\\");
1023                 }
1024         } else {
1025                 do_mkdir(mask);
1026         }
1027
1028         return 0;
1029 }
1030
1031 /****************************************************************************
1032  Show alt name.
1033 ****************************************************************************/
1034
1035 static int cmd_altname(void)
1036 {
1037         pstring name;
1038         pstring buf;
1039         char *p=buf;
1040
1041         pstrcpy(name,cur_dir);
1042
1043         if (!next_token_nr(NULL,p,NULL,sizeof(buf))) {
1044                 d_printf("altname <file>\n");
1045                 return 1;
1046         }
1047         pstrcat(name,p);
1048
1049         do_altname(name);
1050
1051         return 0;
1052 }
1053
1054 /****************************************************************************
1055  Put a single file.
1056 ****************************************************************************/
1057
1058 static int do_put(char *rname, char *lname, BOOL reput)
1059 {
1060         int fnum;
1061         XFILE *f;
1062         size_t start = 0;
1063         off_t nread = 0;
1064         char *buf = NULL;
1065         int maxwrite = io_bufsize;
1066         int rc = 0;
1067
1068         struct timeval tp_start;
1069         GetTimeOfDay(&tp_start);
1070
1071         if (reput) {
1072                 fnum = cli_open(cli, rname, O_RDWR|O_CREAT, DENY_NONE);
1073                 if (fnum >= 0) {
1074                         if (!cli_qfileinfo(cli, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL) &&
1075                             !cli_getattrE(cli, fnum, NULL, &start, NULL, NULL, NULL)) {
1076                                 d_printf("getattrib: %s\n",cli_errstr(cli));
1077                                 return 1;
1078                         }
1079                 }
1080         } else {
1081                 fnum = cli_open(cli, rname, O_RDWR|O_CREAT|O_TRUNC, DENY_NONE);
1082         }
1083
1084         if (fnum == -1) {
1085                 d_printf("%s opening remote file %s\n",cli_errstr(cli),rname);
1086                 return 1;
1087         }
1088
1089         /* allow files to be piped into smbclient
1090            jdblair 24.jun.98
1091
1092            Note that in this case this function will exit(0) rather
1093            than returning. */
1094         if (!strcmp(lname, "-")) {
1095                 f = x_stdin;
1096                 /* size of file is not known */
1097         } else {
1098                 f = x_fopen(lname,O_RDONLY, 0);
1099                 if (f && reput) {
1100                         if (x_tseek(f, start, SEEK_SET) == -1) {
1101                                 d_printf("Error seeking local file\n");
1102                                 return 1;
1103                         }
1104                 }
1105         }
1106
1107         if (!f) {
1108                 d_printf("Error opening local file %s\n",lname);
1109                 return 1;
1110         }
1111
1112         DEBUG(1,("putting file %s as %s ",lname,
1113                  rname));
1114
1115         buf = (char *)SMB_MALLOC(maxwrite);
1116         if (!buf) {
1117                 d_printf("ERROR: Not enough memory!\n");
1118                 return 1;
1119         }
1120         while (!x_feof(f)) {
1121                 int n = maxwrite;
1122                 int ret;
1123
1124                 if ((n = readfile(buf,n,f)) < 1) {
1125                         if((n == 0) && x_feof(f))
1126                                 break; /* Empty local file. */
1127
1128                         d_printf("Error reading local file: %s\n", strerror(errno));
1129                         rc = 1;
1130                         break;
1131                 }
1132
1133                 ret = cli_write(cli, fnum, 0, buf, nread + start, n);
1134
1135                 if (n != ret) {
1136                         d_printf("Error writing file: %s\n", cli_errstr(cli));
1137                         rc = 1;
1138                         break;
1139                 }
1140
1141                 nread += n;
1142         }
1143
1144         if (!cli_close(cli, fnum)) {
1145                 d_printf("%s closing remote file %s\n",cli_errstr(cli),rname);
1146                 x_fclose(f);
1147                 SAFE_FREE(buf);
1148                 return 1;
1149         }
1150
1151
1152         if (f != x_stdin) {
1153                 x_fclose(f);
1154         }
1155
1156         SAFE_FREE(buf);
1157
1158         {
1159                 struct timeval tp_end;
1160                 int this_time;
1161
1162                 GetTimeOfDay(&tp_end);
1163                 this_time =
1164                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1165                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
1166                 put_total_time_ms += this_time;
1167                 put_total_size += nread;
1168
1169                 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1170                          nread / (1.024*this_time + 1.0e-4),
1171                          put_total_size / (1.024*put_total_time_ms)));
1172         }
1173
1174         if (f == x_stdin) {
1175                 cli_shutdown(cli);
1176                 exit(0);
1177         }
1178
1179         return rc;
1180 }
1181
1182 /****************************************************************************
1183  Put a file.
1184 ****************************************************************************/
1185
1186 static int cmd_put(void)
1187 {
1188         pstring lname;
1189         pstring rname;
1190         pstring buf;
1191         char *p=buf;
1192
1193         pstrcpy(rname,cur_dir);
1194         pstrcat(rname,"\\");
1195
1196         if (!next_token_nr(NULL,p,NULL,sizeof(buf))) {
1197                 d_printf("put <filename>\n");
1198                 return 1;
1199         }
1200         pstrcpy(lname,p);
1201
1202         if (next_token_nr(NULL,p,NULL,sizeof(buf)))
1203                 pstrcat(rname,p);
1204         else
1205                 pstrcat(rname,lname);
1206
1207         dos_clean_name(rname);
1208
1209         {
1210                 SMB_STRUCT_STAT st;
1211                 /* allow '-' to represent stdin
1212                    jdblair, 24.jun.98 */
1213                 if (!file_exist(lname,&st) &&
1214                     (strcmp(lname,"-"))) {
1215                         d_printf("%s does not exist\n",lname);
1216                         return 1;
1217                 }
1218         }
1219
1220         return do_put(rname, lname, False);
1221 }
1222
1223 /*************************************
1224  File list structure.
1225 *************************************/
1226
1227 static struct file_list {
1228         struct file_list *prev, *next;
1229         char *file_path;
1230         BOOL isdir;
1231 } *file_list;
1232
1233 /****************************************************************************
1234  Free a file_list structure.
1235 ****************************************************************************/
1236
1237 static void free_file_list (struct file_list * list)
1238 {
1239         struct file_list *tmp;
1240
1241         while (list) {
1242                 tmp = list;
1243                 DLIST_REMOVE(list, list);
1244                 SAFE_FREE(tmp->file_path);
1245                 SAFE_FREE(tmp);
1246         }
1247 }
1248
1249 /****************************************************************************
1250  Seek in a directory/file list until you get something that doesn't start with
1251  the specified name.
1252 ****************************************************************************/
1253
1254 static BOOL seek_list(struct file_list *list, char *name)
1255 {
1256         while (list) {
1257                 trim_string(list->file_path,"./","\n");
1258                 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1259                         return(True);
1260                 }
1261                 list = list->next;
1262         }
1263
1264         return(False);
1265 }
1266
1267 /****************************************************************************
1268  Set the file selection mask.
1269 ****************************************************************************/
1270
1271 static int cmd_select(void)
1272 {
1273         pstrcpy(fileselection,"");
1274         next_token_nr(NULL,fileselection,NULL,sizeof(fileselection));
1275
1276         return 0;
1277 }
1278
1279 /****************************************************************************
1280   Recursive file matching function act as find
1281   match must be always set to True when calling this function
1282 ****************************************************************************/
1283
1284 static int file_find(struct file_list **list, const char *directory,
1285                       const char *expression, BOOL match)
1286 {
1287         DIR *dir;
1288         struct file_list *entry;
1289         struct stat statbuf;
1290         int ret;
1291         char *path;
1292         BOOL isdir;
1293         const char *dname;
1294
1295         dir = opendir(directory);
1296         if (!dir)
1297                 return -1;
1298
1299         while ((dname = readdirname(dir))) {
1300                 if (!strcmp("..", dname))
1301                         continue;
1302                 if (!strcmp(".", dname))
1303                         continue;
1304
1305                 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1306                         continue;
1307                 }
1308
1309                 isdir = False;
1310                 if (!match || !gen_fnmatch(expression, dname)) {
1311                         if (recurse) {
1312                                 ret = stat(path, &statbuf);
1313                                 if (ret == 0) {
1314                                         if (S_ISDIR(statbuf.st_mode)) {
1315                                                 isdir = True;
1316                                                 ret = file_find(list, path, expression, False);
1317                                         }
1318                                 } else {
1319                                         d_printf("file_find: cannot stat file %s\n", path);
1320                                 }
1321
1322                                 if (ret == -1) {
1323                                         SAFE_FREE(path);
1324                                         closedir(dir);
1325                                         return -1;
1326                                 }
1327                         }
1328                         entry = SMB_MALLOC_P(struct file_list);
1329                         if (!entry) {
1330                                 d_printf("Out of memory in file_find\n");
1331                                 closedir(dir);
1332                                 return -1;
1333                         }
1334                         entry->file_path = path;
1335                         entry->isdir = isdir;
1336                         DLIST_ADD(*list, entry);
1337                 } else {
1338                         SAFE_FREE(path);
1339                 }
1340         }
1341
1342         closedir(dir);
1343         return 0;
1344 }
1345
1346 /****************************************************************************
1347  mput some files.
1348 ****************************************************************************/
1349
1350 static int cmd_mput(void)
1351 {
1352         pstring buf;
1353         char *p=buf;
1354
1355         while (next_token_nr(NULL,p,NULL,sizeof(buf))) {
1356                 int ret;
1357                 struct file_list *temp_list;
1358                 char *quest, *lname, *rname;
1359
1360                 file_list = NULL;
1361
1362                 ret = file_find(&file_list, ".", p, True);
1363                 if (ret) {
1364                         free_file_list(file_list);
1365                         continue;
1366                 }
1367
1368                 quest = NULL;
1369                 lname = NULL;
1370                 rname = NULL;
1371
1372                 for (temp_list = file_list; temp_list;
1373                      temp_list = temp_list->next) {
1374
1375                         SAFE_FREE(lname);
1376                         if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1377                                 continue;
1378                         trim_string(lname, "./", "/");
1379
1380                         /* check if it's a directory */
1381                         if (temp_list->isdir) {
1382                                 /* if (!recurse) continue; */
1383
1384                                 SAFE_FREE(quest);
1385                                 if (asprintf(&quest, "Put directory %s? ", lname) < 0) break;
1386                                 if (prompt && !yesno(quest)) { /* No */
1387                                         /* Skip the directory */
1388                                         lname[strlen(lname)-1] = '/';
1389                                         if (!seek_list(temp_list, lname))
1390                                                 break;
1391                                 } else { /* Yes */
1392                                         SAFE_FREE(rname);
1393                                         if(asprintf(&rname, "%s%s", cur_dir, lname) < 0) break;
1394                                         dos_format(rname);
1395                                         if (!cli_chkpath(cli, rname) &&
1396                                             !do_mkdir(rname)) {
1397                                                 DEBUG (0, ("Unable to make dir, skipping..."));
1398                                                 /* Skip the directory */
1399                                                 lname[strlen(lname)-1] = '/';
1400                                                 if (!seek_list(temp_list, lname))
1401                                                         break;
1402                                         }
1403                                 }
1404                                 continue;
1405                         } else {
1406                                 SAFE_FREE(quest);
1407                                 if (asprintf(&quest,"Put file %s? ", lname) < 0) break;
1408                                 if (prompt && !yesno(quest)) /* No */
1409                                         continue;
1410
1411                                 /* Yes */
1412                                 SAFE_FREE(rname);
1413                                 if (asprintf(&rname, "%s%s", cur_dir, lname) < 0) break;
1414                         }
1415
1416                         dos_format(rname);
1417
1418                         do_put(rname, lname, False);
1419                 }
1420                 free_file_list(file_list);
1421                 SAFE_FREE(quest);
1422                 SAFE_FREE(lname);
1423                 SAFE_FREE(rname);
1424         }
1425
1426         return 0;
1427 }
1428
1429 /****************************************************************************
1430  Cancel a print job.
1431 ****************************************************************************/
1432
1433 static int do_cancel(int job)
1434 {
1435         if (cli_printjob_del(cli, job)) {
1436                 d_printf("Job %d cancelled\n",job);
1437                 return 0;
1438         } else {
1439                 d_printf("Error cancelling job %d : %s\n",job,cli_errstr(cli));
1440                 return 1;
1441         }
1442 }
1443
1444 /****************************************************************************
1445  Cancel a print job.
1446 ****************************************************************************/
1447
1448 static int cmd_cancel(void)
1449 {
1450         pstring buf;
1451         int job;
1452
1453         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1454                 d_printf("cancel <jobid> ...\n");
1455                 return 1;
1456         }
1457         do {
1458                 job = atoi(buf);
1459                 do_cancel(job);
1460         } while (next_token_nr(NULL,buf,NULL,sizeof(buf)));
1461
1462         return 0;
1463 }
1464
1465 /****************************************************************************
1466  Print a file.
1467 ****************************************************************************/
1468
1469 static int cmd_print(void)
1470 {
1471         pstring lname;
1472         pstring rname;
1473         char *p;
1474
1475         if (!next_token_nr(NULL,lname,NULL, sizeof(lname))) {
1476                 d_printf("print <filename>\n");
1477                 return 1;
1478         }
1479
1480         pstrcpy(rname,lname);
1481         p = strrchr_m(rname,'/');
1482         if (p) {
1483                 slprintf(rname, sizeof(rname)-1, "%s-%d", p+1, (int)sys_getpid());
1484         }
1485
1486         if (strequal(lname,"-")) {
1487                 slprintf(rname, sizeof(rname)-1, "stdin-%d", (int)sys_getpid());
1488         }
1489
1490         return do_put(rname, lname, False);
1491 }
1492
1493 /****************************************************************************
1494  Show a print queue entry.
1495 ****************************************************************************/
1496
1497 static void queue_fn(struct print_job_info *p)
1498 {
1499         d_printf("%-6d   %-9d    %s\n", (int)p->id, (int)p->size, p->name);
1500 }
1501
1502 /****************************************************************************
1503  Show a print queue.
1504 ****************************************************************************/
1505
1506 static int cmd_queue(void)
1507 {
1508         cli_print_queue(cli, queue_fn);
1509
1510         return 0;
1511 }
1512
1513 /****************************************************************************
1514  Delete some files.
1515 ****************************************************************************/
1516
1517 static void do_del(file_info *finfo)
1518 {
1519         pstring mask;
1520
1521         pstrcpy(mask,cur_dir);
1522         pstrcat(mask,finfo->name);
1523
1524         if (finfo->mode & aDIR)
1525                 return;
1526
1527         if (!cli_unlink(cli, mask)) {
1528                 d_printf("%s deleting remote file %s\n",cli_errstr(cli),mask);
1529         }
1530 }
1531
1532 /****************************************************************************
1533  Delete some files.
1534 ****************************************************************************/
1535
1536 static int cmd_del(void)
1537 {
1538         pstring mask;
1539         pstring buf;
1540         uint16 attribute = aSYSTEM | aHIDDEN;
1541
1542         if (recurse)
1543                 attribute |= aDIR;
1544
1545         pstrcpy(mask,cur_dir);
1546
1547         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1548                 d_printf("del <filename>\n");
1549                 return 1;
1550         }
1551         pstrcat(mask,buf);
1552
1553         do_list(mask, attribute,do_del,False,False);
1554
1555         return 0;
1556 }
1557
1558 /****************************************************************************
1559 ****************************************************************************/
1560
1561 static int cmd_open(void)
1562 {
1563         pstring mask;
1564         pstring buf;
1565
1566         pstrcpy(mask,cur_dir);
1567
1568         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1569                 d_printf("open <filename>\n");
1570                 return 1;
1571         }
1572         pstrcat(mask,buf);
1573
1574         cli_nt_create(cli, mask, FILE_READ_DATA);
1575
1576         return 0;
1577 }
1578
1579
1580 /****************************************************************************
1581  Remove a directory.
1582 ****************************************************************************/
1583
1584 static int cmd_rmdir(void)
1585 {
1586         pstring mask;
1587         pstring buf;
1588
1589         pstrcpy(mask,cur_dir);
1590
1591         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
1592                 d_printf("rmdir <dirname>\n");
1593                 return 1;
1594         }
1595         pstrcat(mask,buf);
1596
1597         if (!cli_rmdir(cli, mask)) {
1598                 d_printf("%s removing remote directory file %s\n",
1599                          cli_errstr(cli),mask);
1600         }
1601
1602         return 0;
1603 }
1604
1605 /****************************************************************************
1606  UNIX hardlink.
1607 ****************************************************************************/
1608
1609 static int cmd_link(void)
1610 {
1611         pstring oldname,newname;
1612         pstring buf,buf2;
1613
1614         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1615                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1616                 return 1;
1617         }
1618
1619         pstrcpy(oldname,cur_dir);
1620         pstrcpy(newname,cur_dir);
1621
1622         if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) ||
1623             !next_token_nr(NULL,buf2,NULL, sizeof(buf2))) {
1624                 d_printf("link <oldname> <newname>\n");
1625                 return 1;
1626         }
1627
1628         pstrcat(oldname,buf);
1629         pstrcat(newname,buf2);
1630
1631         if (!cli_unix_hardlink(cli, oldname, newname)) {
1632                 d_printf("%s linking files (%s -> %s)\n", cli_errstr(cli), newname, oldname);
1633                 return 1;
1634         }
1635
1636         return 0;
1637 }
1638
1639 /****************************************************************************
1640  UNIX symlink.
1641 ****************************************************************************/
1642
1643 static int cmd_symlink(void)
1644 {
1645         pstring oldname,newname;
1646         pstring buf,buf2;
1647
1648         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1649                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1650                 return 1;
1651         }
1652
1653         pstrcpy(newname,cur_dir);
1654
1655         if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) ||
1656             !next_token_nr(NULL,buf2,NULL, sizeof(buf2))) {
1657                 d_printf("symlink <oldname> <newname>\n");
1658                 return 1;
1659         }
1660
1661         pstrcpy(oldname,buf);
1662         pstrcat(newname,buf2);
1663
1664         if (!cli_unix_symlink(cli, oldname, newname)) {
1665                 d_printf("%s symlinking files (%s -> %s)\n",
1666                         cli_errstr(cli), newname, oldname);
1667                 return 1;
1668         }
1669
1670         return 0;
1671 }
1672
1673 /****************************************************************************
1674  UNIX chmod.
1675 ****************************************************************************/
1676
1677 static int cmd_chmod(void)
1678 {
1679         pstring src;
1680         mode_t mode;
1681         pstring buf, buf2;
1682
1683         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1684                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1685                 return 1;
1686         }
1687
1688         pstrcpy(src,cur_dir);
1689
1690         if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) ||
1691             !next_token_nr(NULL,buf2,NULL, sizeof(buf2))) {
1692                 d_printf("chmod mode file\n");
1693                 return 1;
1694         }
1695
1696         mode = (mode_t)strtol(buf, NULL, 8);
1697         pstrcat(src,buf2);
1698
1699         if (!cli_unix_chmod(cli, src, mode)) {
1700                 d_printf("%s chmod file %s 0%o\n",
1701                         cli_errstr(cli), src, (unsigned int)mode);
1702                 return 1;
1703         }
1704
1705         return 0;
1706 }
1707
1708 static const char *filetype_to_str(mode_t mode)
1709 {
1710         if (S_ISREG(mode)) {
1711                 return "regular file";
1712         } else if (S_ISDIR(mode)) {
1713                 return "directory";
1714         } else
1715 #ifdef S_ISCHR
1716         if (S_ISCHR(mode)) {
1717                 return "character device";
1718         } else
1719 #endif
1720 #ifdef S_ISBLK
1721         if (S_ISBLK(mode)) {
1722                 return "block device";
1723         } else
1724 #endif
1725 #ifdef S_ISFIFO
1726         if (S_ISFIFO(mode)) {
1727                 return "fifo";
1728         } else
1729 #endif
1730 #ifdef S_ISLNK
1731         if (S_ISLNK(mode)) {
1732                 return "symbolic link";
1733         } else
1734 #endif
1735 #ifdef S_ISSOCK
1736         if (S_ISSOCK(mode)) {
1737                 return "socket";
1738         } else
1739 #endif
1740         return "";
1741 }
1742
1743 static char rwx_to_str(mode_t m, mode_t bt, char ret)
1744 {
1745         if (m & bt) {
1746                 return ret;
1747         } else {
1748                 return '-';
1749         }
1750 }
1751
1752 static char *unix_mode_to_str(char *s, mode_t m)
1753 {
1754         char *p = s;
1755         const char *str = filetype_to_str(m);
1756
1757         switch(str[0]) {
1758                 case 'd':
1759                         *p++ = 'd';
1760                         break;
1761                 case 'c':
1762                         *p++ = 'c';
1763                         break;
1764                 case 'b':
1765                         *p++ = 'b';
1766                         break;
1767                 case 'f':
1768                         *p++ = 'p';
1769                         break;
1770                 case 's':
1771                         *p++ = str[1] == 'y' ? 'l' : 's';
1772                         break;
1773                 case 'r':
1774                 default:
1775                         *p++ = '-';
1776                         break;
1777         }
1778         *p++ = rwx_to_str(m, S_IRUSR, 'r');
1779         *p++ = rwx_to_str(m, S_IWUSR, 'w');
1780         *p++ = rwx_to_str(m, S_IXUSR, 'x');
1781         *p++ = rwx_to_str(m, S_IRGRP, 'r');
1782         *p++ = rwx_to_str(m, S_IWGRP, 'w');
1783         *p++ = rwx_to_str(m, S_IXGRP, 'x');
1784         *p++ = rwx_to_str(m, S_IROTH, 'r');
1785         *p++ = rwx_to_str(m, S_IWOTH, 'w');
1786         *p++ = rwx_to_str(m, S_IXOTH, 'x');
1787         *p++ = '\0';
1788         return s;
1789 }
1790
1791 /****************************************************************************
1792  Utility function for UNIX getfacl.
1793 ****************************************************************************/
1794
1795 static char *perms_to_string(fstring permstr, unsigned char perms)
1796 {
1797         fstrcpy(permstr, "---");
1798         if (perms & SMB_POSIX_ACL_READ) {
1799                 permstr[0] = 'r';
1800         }
1801         if (perms & SMB_POSIX_ACL_WRITE) {
1802                 permstr[1] = 'w';
1803         }
1804         if (perms & SMB_POSIX_ACL_EXECUTE) {
1805                 permstr[2] = 'x';
1806         }
1807         return permstr;
1808 }
1809
1810 /****************************************************************************
1811  UNIX getfacl.
1812 ****************************************************************************/
1813
1814 static int cmd_getfacl(void)
1815 {
1816         pstring src, name;
1817         uint16 major, minor;
1818         uint32 caplow, caphigh;
1819         char *retbuf = NULL;
1820         size_t rb_size = 0;
1821         SMB_STRUCT_STAT sbuf;
1822         uint16 num_file_acls = 0;
1823         uint16 num_dir_acls = 0;
1824         uint16 i;
1825
1826         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1827                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1828                 return 1;
1829         }
1830
1831         if (!cli_unix_extensions_version(cli, &major, &minor, &caplow, &caphigh)) {
1832                 d_printf("Can't get UNIX CIFS version from server.\n");
1833                 return 1;
1834         }
1835
1836         if (!(caplow & CIFS_UNIX_POSIX_ACLS_CAP)) {
1837                 d_printf("This server supports UNIX extensions but doesn't support POSIX ACLs.\n");
1838                 return 1;
1839         }
1840
1841         pstrcpy(src,cur_dir);
1842
1843         if (!next_token_nr(NULL,name,NULL,sizeof(name))) {
1844                 d_printf("stat file\n");
1845                 return 1;
1846         }
1847
1848         pstrcat(src,name);
1849
1850         if (!cli_unix_stat(cli, src, &sbuf)) {
1851                 d_printf("%s getfacl doing a stat on file %s\n",
1852                         cli_errstr(cli), src);
1853                 return 1;
1854         }
1855
1856         if (!cli_unix_getfacl(cli, src, &rb_size, &retbuf)) {
1857                 d_printf("%s getfacl file %s\n",
1858                         cli_errstr(cli), src);
1859                 return 1;
1860         }
1861
1862         /* ToDo : Print out the ACL values. */
1863         if (SVAL(retbuf,0) != SMB_POSIX_ACL_VERSION || rb_size < 6) {
1864                 d_printf("getfacl file %s, unknown POSIX acl version %u.\n",
1865                         src, (unsigned int)CVAL(retbuf,0) );
1866                 SAFE_FREE(retbuf);
1867                 return 1;
1868         }
1869
1870         num_file_acls = SVAL(retbuf,2);
1871         num_dir_acls = SVAL(retbuf,4);
1872         if (rb_size != SMB_POSIX_ACL_HEADER_SIZE + SMB_POSIX_ACL_ENTRY_SIZE*(num_file_acls+num_dir_acls)) {
1873                 d_printf("getfacl file %s, incorrect POSIX acl buffer size (should be %u, was %u).\n",
1874                         src,
1875                         (unsigned int)(SMB_POSIX_ACL_HEADER_SIZE + SMB_POSIX_ACL_ENTRY_SIZE*(num_file_acls+num_dir_acls)),
1876                         (unsigned int)rb_size);
1877
1878                 SAFE_FREE(retbuf);
1879                 return 1;
1880         }
1881
1882         d_printf("# file: %s\n", src);
1883         d_printf("# owner: %u\n# group: %u\n", (unsigned int)sbuf.st_uid, (unsigned int)sbuf.st_gid);
1884
1885         if (num_file_acls == 0 && num_dir_acls == 0) {
1886                 d_printf("No acls found.\n");
1887         }
1888
1889         for (i = 0; i < num_file_acls; i++) {
1890                 uint32 uorg;
1891                 fstring permstring;
1892                 unsigned char tagtype = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE));
1893                 unsigned char perms = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+1);
1894
1895                 switch(tagtype) {
1896                         case SMB_POSIX_ACL_USER_OBJ:
1897                                 d_printf("user::");
1898                                 break;
1899                         case SMB_POSIX_ACL_USER:
1900                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+2);
1901                                 d_printf("user:%u:", uorg);
1902                                 break;
1903                         case SMB_POSIX_ACL_GROUP_OBJ:
1904                                 d_printf("group::");
1905                                 break;
1906                         case SMB_POSIX_ACL_GROUP:
1907                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+2);
1908                                 d_printf("group:%u", uorg);
1909                                 break;
1910                         case SMB_POSIX_ACL_MASK:
1911                                 d_printf("mask::");
1912                                 break;
1913                         case SMB_POSIX_ACL_OTHER:
1914                                 d_printf("other::");
1915                                 break;
1916                         default:
1917                                 d_printf("getfacl file %s, incorrect POSIX acl tagtype (%u).\n",
1918                                         src, (unsigned int)tagtype );
1919                                 SAFE_FREE(retbuf);
1920                                 return 1;
1921                 }
1922
1923                 d_printf("%s\n", perms_to_string(permstring, perms));
1924         }
1925
1926         for (i = 0; i < num_dir_acls; i++) {
1927                 uint32 uorg;
1928                 fstring permstring;
1929                 unsigned char tagtype = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE));
1930                 unsigned char perms = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+1);
1931
1932                 switch(tagtype) {
1933                         case SMB_POSIX_ACL_USER_OBJ:
1934                                 d_printf("default:user::");
1935                                 break;
1936                         case SMB_POSIX_ACL_USER:
1937                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+2);
1938                                 d_printf("default:user:%u:", uorg);
1939                                 break;
1940                         case SMB_POSIX_ACL_GROUP_OBJ:
1941                                 d_printf("default:group::");
1942                                 break;
1943                         case SMB_POSIX_ACL_GROUP:
1944                                 uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+2);
1945                                 d_printf("default:group:%u", uorg);
1946                                 break;
1947                         case SMB_POSIX_ACL_MASK:
1948                                 d_printf("default:mask::");
1949                                 break;
1950                         case SMB_POSIX_ACL_OTHER:
1951                                 d_printf("default:other::");
1952                                 break;
1953                         default:
1954                                 d_printf("getfacl file %s, incorrect POSIX acl tagtype (%u).\n",
1955                                         src, (unsigned int)tagtype );
1956                                 SAFE_FREE(retbuf);
1957                                 return 1;
1958                 }
1959
1960                 d_printf("%s\n", perms_to_string(permstring, perms));
1961         }
1962
1963         SAFE_FREE(retbuf);
1964         return 0;
1965 }
1966
1967 /****************************************************************************
1968  UNIX stat.
1969 ****************************************************************************/
1970
1971 static int cmd_stat(void)
1972 {
1973         pstring src, name;
1974         fstring mode_str;
1975         SMB_STRUCT_STAT sbuf;
1976
1977         if (!SERVER_HAS_UNIX_CIFS(cli)) {
1978                 d_printf("Server doesn't support UNIX CIFS calls.\n");
1979                 return 1;
1980         }
1981
1982         pstrcpy(src,cur_dir);
1983
1984         if (!next_token_nr(NULL,name,NULL,sizeof(name))) {
1985                 d_printf("stat file\n");
1986                 return 1;
1987         }
1988
1989         pstrcat(src,name);
1990
1991         if (!cli_unix_stat(cli, src, &sbuf)) {
1992                 d_printf("%s stat file %s\n",
1993                         cli_errstr(cli), src);
1994                 return 1;
1995         }
1996
1997         /* Print out the stat values. */
1998         d_printf("File: %s\n", src);
1999         d_printf("Size: %-12.0f\tBlocks: %u\t%s\n",
2000                 (double)sbuf.st_size,
2001                 (unsigned int)sbuf.st_blocks,
2002                 filetype_to_str(sbuf.st_mode));
2003
2004 #if defined(S_ISCHR) && defined(S_ISBLK)
2005         if (S_ISCHR(sbuf.st_mode) || S_ISBLK(sbuf.st_mode)) {
2006                 d_printf("Inode: %.0f\tLinks: %u\tDevice type: %u,%u\n",
2007                         (double)sbuf.st_ino,
2008                         (unsigned int)sbuf.st_nlink,
2009                         unix_dev_major(sbuf.st_rdev),
2010                         unix_dev_minor(sbuf.st_rdev));
2011         } else
2012 #endif
2013                 d_printf("Inode: %.0f\tLinks: %u\n",
2014                         (double)sbuf.st_ino,
2015                         (unsigned int)sbuf.st_nlink);
2016
2017         d_printf("Access: (0%03o/%s)\tUid: %u\tGid: %u\n",
2018                 ((int)sbuf.st_mode & 0777),
2019                 unix_mode_to_str(mode_str, sbuf.st_mode),
2020                 (unsigned int)sbuf.st_uid,
2021                 (unsigned int)sbuf.st_gid);
2022
2023         strftime(mode_str, sizeof(mode_str), "%F %T %z", localtime(&sbuf.st_atime));
2024         d_printf("Access: %s\n", mode_str);
2025
2026         strftime(mode_str, sizeof(mode_str), "%F %T %z", localtime(&sbuf.st_mtime));
2027         d_printf("Modify: %s\n", mode_str);
2028
2029         strftime(mode_str, sizeof(mode_str), "%F %T %z", localtime(&sbuf.st_ctime));
2030         d_printf("Change: %s\n", mode_str);
2031
2032         return 0;
2033 }
2034
2035
2036 /****************************************************************************
2037  UNIX chown.
2038 ****************************************************************************/
2039
2040 static int cmd_chown(void)
2041 {
2042         pstring src;
2043         uid_t uid;
2044         gid_t gid;
2045         pstring buf, buf2, buf3;
2046
2047         if (!SERVER_HAS_UNIX_CIFS(cli)) {
2048                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2049                 return 1;
2050         }
2051
2052         pstrcpy(src,cur_dir);
2053
2054         if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) ||
2055             !next_token_nr(NULL,buf2,NULL, sizeof(buf2)) ||
2056             !next_token_nr(NULL,buf3,NULL, sizeof(buf3))) {
2057                 d_printf("chown uid gid file\n");
2058                 return 1;
2059         }
2060
2061         uid = (uid_t)atoi(buf);
2062         gid = (gid_t)atoi(buf2);
2063         pstrcat(src,buf3);
2064
2065         if (!cli_unix_chown(cli, src, uid, gid)) {
2066                 d_printf("%s chown file %s uid=%d, gid=%d\n",
2067                         cli_errstr(cli), src, (int)uid, (int)gid);
2068                 return 1;
2069         }
2070
2071         return 0;
2072 }
2073
2074 /****************************************************************************
2075  Rename some file.
2076 ****************************************************************************/
2077
2078 static int cmd_rename(void)
2079 {
2080         pstring src,dest;
2081         pstring buf,buf2;
2082
2083         pstrcpy(src,cur_dir);
2084         pstrcpy(dest,cur_dir);
2085
2086         if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) ||
2087             !next_token_nr(NULL,buf2,NULL, sizeof(buf2))) {
2088                 d_printf("rename <src> <dest>\n");
2089                 return 1;
2090         }
2091
2092         pstrcat(src,buf);
2093         pstrcat(dest,buf2);
2094
2095         if (!cli_rename(cli, src, dest)) {
2096                 d_printf("%s renaming files\n",cli_errstr(cli));
2097                 return 1;
2098         }
2099
2100         return 0;
2101 }
2102
2103 /****************************************************************************
2104  Hard link files using the NT call.
2105 ****************************************************************************/
2106
2107 static int cmd_hardlink(void)
2108 {
2109         pstring src,dest;
2110         pstring buf,buf2;
2111
2112         pstrcpy(src,cur_dir);
2113         pstrcpy(dest,cur_dir);
2114
2115         if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) ||
2116             !next_token_nr(NULL,buf2,NULL, sizeof(buf2))) {
2117                 d_printf("hardlink <src> <dest>\n");
2118                 return 1;
2119         }
2120
2121         pstrcat(src,buf);
2122         pstrcat(dest,buf2);
2123
2124         if (!cli_nt_hardlink(cli, src, dest)) {
2125                 d_printf("%s doing an NT hard link of files\n",cli_errstr(cli));
2126                 return 1;
2127         }
2128
2129         return 0;
2130 }
2131
2132 /****************************************************************************
2133  Toggle the prompt flag.
2134 ****************************************************************************/
2135
2136 static int cmd_prompt(void)
2137 {
2138         prompt = !prompt;
2139         DEBUG(2,("prompting is now %s\n",prompt?"on":"off"));
2140
2141         return 1;
2142 }
2143
2144 /****************************************************************************
2145  Set the newer than time.
2146 ****************************************************************************/
2147
2148 static int cmd_newer(void)
2149 {
2150         pstring buf;
2151         BOOL ok;
2152         SMB_STRUCT_STAT sbuf;
2153
2154         ok = next_token_nr(NULL,buf,NULL,sizeof(buf));
2155         if (ok && (sys_stat(buf,&sbuf) == 0)) {
2156                 newer_than = sbuf.st_mtime;
2157                 DEBUG(1,("Getting files newer than %s",
2158                          asctime(LocalTime(&newer_than))));
2159         } else {
2160                 newer_than = 0;
2161         }
2162
2163         if (ok && newer_than == 0) {
2164                 d_printf("Error setting newer-than time\n");
2165                 return 1;
2166         }
2167
2168         return 0;
2169 }
2170
2171 /****************************************************************************
2172  Set the archive level.
2173 ****************************************************************************/
2174
2175 static int cmd_archive(void)
2176 {
2177         pstring buf;
2178
2179         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
2180                 archive_level = atoi(buf);
2181         } else
2182                 d_printf("Archive level is %d\n",archive_level);
2183
2184         return 0;
2185 }
2186
2187 /****************************************************************************
2188  Toggle the lowercaseflag.
2189 ****************************************************************************/
2190
2191 static int cmd_lowercase(void)
2192 {
2193         lowercase = !lowercase;
2194         DEBUG(2,("filename lowercasing is now %s\n",lowercase?"on":"off"));
2195
2196         return 0;
2197 }
2198
2199 /****************************************************************************
2200  Toggle the case sensitive flag.
2201 ****************************************************************************/
2202
2203 static int cmd_setcase(void)
2204 {
2205         BOOL orig_case_sensitive = cli_set_case_sensitive(cli, False);
2206
2207         cli_set_case_sensitive(cli, !orig_case_sensitive);
2208         DEBUG(2,("filename case sensitivity is now %s\n",!orig_case_sensitive ?
2209                 "on":"off"));
2210
2211         return 0;
2212 }
2213
2214 /****************************************************************************
2215  Toggle the recurse flag.
2216 ****************************************************************************/
2217
2218 static int cmd_recurse(void)
2219 {
2220         recurse = !recurse;
2221         DEBUG(2,("directory recursion is now %s\n",recurse?"on":"off"));
2222
2223         return 0;
2224 }
2225
2226 /****************************************************************************
2227  Toggle the translate flag.
2228 ****************************************************************************/
2229
2230 static int cmd_translate(void)
2231 {
2232         translation = !translation;
2233         DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
2234                  translation?"on":"off"));
2235
2236         return 0;
2237 }
2238
2239 /****************************************************************************
2240  Do a printmode command.
2241 ****************************************************************************/
2242
2243 static int cmd_printmode(void)
2244 {
2245         fstring buf;
2246         fstring mode;
2247
2248         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
2249                 if (strequal(buf,"text")) {
2250                         printmode = 0;
2251                 } else {
2252                         if (strequal(buf,"graphics"))
2253                                 printmode = 1;
2254                         else
2255                                 printmode = atoi(buf);
2256                 }
2257         }
2258
2259         switch(printmode) {
2260                 case 0:
2261                         fstrcpy(mode,"text");
2262                         break;
2263                 case 1:
2264                         fstrcpy(mode,"graphics");
2265                         break;
2266                 default:
2267                         slprintf(mode,sizeof(mode)-1,"%d",printmode);
2268                         break;
2269         }
2270
2271         DEBUG(2,("the printmode is now %s\n",mode));
2272
2273         return 0;
2274 }
2275
2276 /****************************************************************************
2277  Do the lcd command.
2278  ****************************************************************************/
2279
2280 static int cmd_lcd(void)
2281 {
2282         pstring buf;
2283         pstring d;
2284
2285         if (next_token_nr(NULL,buf,NULL,sizeof(buf)))
2286                 chdir(buf);
2287         DEBUG(2,("the local directory is now %s\n",sys_getwd(d)));
2288
2289         return 0;
2290 }
2291
2292 /****************************************************************************
2293  Get a file restarting at end of local file.
2294  ****************************************************************************/
2295
2296 static int cmd_reget(void)
2297 {
2298         pstring local_name;
2299         pstring remote_name;
2300         char *p;
2301
2302         pstrcpy(remote_name, cur_dir);
2303         pstrcat(remote_name, "\\");
2304
2305         p = remote_name + strlen(remote_name);
2306
2307         if (!next_token_nr(NULL, p, NULL, sizeof(remote_name) - strlen(remote_name))) {
2308                 d_printf("reget <filename>\n");
2309                 return 1;
2310         }
2311         pstrcpy(local_name, p);
2312         dos_clean_name(remote_name);
2313
2314         next_token_nr(NULL, local_name, NULL, sizeof(local_name));
2315
2316         return do_get(remote_name, local_name, True);
2317 }
2318
2319 /****************************************************************************
2320  Put a file restarting at end of local file.
2321  ****************************************************************************/
2322
2323 static int cmd_reput(void)
2324 {
2325         pstring local_name;
2326         pstring remote_name;
2327         pstring buf;
2328         char *p = buf;
2329         SMB_STRUCT_STAT st;
2330
2331         pstrcpy(remote_name, cur_dir);
2332         pstrcat(remote_name, "\\");
2333
2334         if (!next_token_nr(NULL, p, NULL, sizeof(buf))) {
2335                 d_printf("reput <filename>\n");
2336                 return 1;
2337         }
2338         pstrcpy(local_name, p);
2339
2340         if (!file_exist(local_name, &st)) {
2341                 d_printf("%s does not exist\n", local_name);
2342                 return 1;
2343         }
2344
2345         if (next_token_nr(NULL, p, NULL, sizeof(buf)))
2346                 pstrcat(remote_name, p);
2347         else
2348                 pstrcat(remote_name, local_name);
2349
2350         dos_clean_name(remote_name);
2351
2352         return do_put(remote_name, local_name, True);
2353 }
2354
2355 /****************************************************************************
2356  List a share name.
2357  ****************************************************************************/
2358
2359 static void browse_fn(const char *name, uint32 m,
2360                       const char *comment, void *state)
2361 {
2362         fstring typestr;
2363
2364         *typestr=0;
2365
2366         switch (m)
2367         {
2368           case STYPE_DISKTREE:
2369             fstrcpy(typestr,"Disk"); break;
2370           case STYPE_PRINTQ:
2371             fstrcpy(typestr,"Printer"); break;
2372           case STYPE_DEVICE:
2373             fstrcpy(typestr,"Device"); break;
2374           case STYPE_IPC:
2375             fstrcpy(typestr,"IPC"); break;
2376         }
2377         /* FIXME: If the remote machine returns non-ascii characters
2378            in any of these fields, they can corrupt the output.  We
2379            should remove them. */
2380         if (!grepable) {
2381                 d_printf("\t%-15s %-10.10s%s\n",
2382                         name,typestr,comment);
2383         } else {
2384                 d_printf ("%s|%s|%s\n",typestr,name,comment);
2385         }
2386 }
2387
2388 /****************************************************************************
2389  Try and browse available connections on a host.
2390 ****************************************************************************/
2391
2392 static BOOL browse_host(BOOL sort)
2393 {
2394         int ret;
2395         if (!grepable) {
2396                 d_printf("\n\tSharename       Type      Comment\n");
2397                 d_printf("\t---------       ----      -------\n");
2398         }
2399
2400         if((ret = cli_RNetShareEnum(cli, browse_fn, NULL)) == -1)
2401                 d_printf("Error returning browse list: %s\n", cli_errstr(cli));
2402
2403         return (ret != -1);
2404 }
2405
2406 /****************************************************************************
2407  List a server name.
2408 ****************************************************************************/
2409
2410 static void server_fn(const char *name, uint32 m,
2411                       const char *comment, void *state)
2412 {
2413
2414         if (!grepable){
2415                 d_printf("\t%-16s     %s\n", name, comment);
2416         } else {
2417                 d_printf("%s|%s|%s\n",(char *)state, name, comment);
2418         }
2419 }
2420
2421 /****************************************************************************
2422  Try and browse available connections on a host.
2423 ****************************************************************************/
2424
2425 static BOOL list_servers(const char *wk_grp)
2426 {
2427         if (!cli->server_domain)
2428                 return False;
2429
2430         if (!grepable) {
2431                 d_printf("\n\tServer               Comment\n");
2432                 d_printf("\t---------            -------\n");
2433         };
2434         cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_ALL, server_fn,
2435                           "Server");
2436
2437         if (!grepable) {
2438                 d_printf("\n\tWorkgroup            Master\n");
2439                 d_printf("\t---------            -------\n");
2440         };
2441
2442         cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_DOMAIN_ENUM,
2443                           server_fn, "Workgroup");
2444         return True;
2445 }
2446
2447 /****************************************************************************
2448  Print or set current VUID
2449 ****************************************************************************/
2450
2451 static int cmd_vuid(void)
2452 {
2453         fstring buf;
2454
2455         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
2456                 d_printf("Current VUID is %d\n", cli->vuid);
2457                 return 0;
2458         }
2459
2460         cli->vuid = atoi(buf);
2461         return 0;
2462 }
2463
2464 /****************************************************************************
2465  Setup a new VUID, by issuing a session setup
2466 ****************************************************************************/
2467
2468 static int cmd_logon(void)
2469 {
2470         pstring l_username, l_password;
2471         pstring buf,buf2;
2472
2473         if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) {
2474                 d_printf("logon <username> [<password>]\n");
2475                 return 0;
2476         }
2477
2478         pstrcpy(l_username, buf);
2479
2480         if (!next_token_nr(NULL,buf2,NULL,sizeof(buf))) {
2481                 char *pass = getpass("Password: ");
2482                 if (pass) {
2483                         pstrcpy(l_password, pass);
2484                         got_pass = 1;
2485                 }
2486         } else {
2487                 pstrcpy(l_password, buf2);
2488         }
2489
2490         if (!cli_session_setup(cli, l_username,
2491                                l_password, strlen(l_password),
2492                                l_password, strlen(l_password),
2493                                lp_workgroup())) {
2494                 d_printf("session setup failed: %s\n", cli_errstr(cli));
2495                 return -1;
2496         }
2497
2498         d_printf("Current VUID is %d\n", cli->vuid);
2499         return 0;
2500 }
2501
2502 /* Some constants for completing filename arguments */
2503
2504 #define COMPL_NONE        0          /* No completions */
2505 #define COMPL_REMOTE      1          /* Complete remote filename */
2506 #define COMPL_LOCAL       2          /* Complete local filename */
2507
2508 /* This defines the commands supported by this client.
2509  * NOTE: The "!" must be the last one in the list because it's fn pointer
2510  *       field is NULL, and NULL in that field is used in process_tok()
2511  *       (below) to indicate the end of the list.  crh
2512  */
2513 static struct
2514 {
2515   const char *name;
2516   int (*fn)(void);
2517   const char *description;
2518   char compl_args[2];      /* Completion argument info */
2519 } commands[] = {
2520   {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2521   {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
2522   {"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}},
2523   {"blocksize",cmd_block,"blocksize <number> (default 20)",{COMPL_NONE,COMPL_NONE}},
2524   {"cancel",cmd_cancel,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
2525   {"case_sensitive",cmd_setcase,"toggle the case sensitive flag to server",{COMPL_NONE,COMPL_NONE}},
2526   {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
2527   {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
2528   {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
2529   {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2530   {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2531   {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2532   {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2533   {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
2534   {"getfacl",cmd_getfacl,"<file name> get the POSIX ACL on a file (UNIX extensions only)",{COMPL_REMOTE,COMPL_LOCAL}},
2535   {"hardlink",cmd_hardlink,"<src> <dest> create a Windows hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2536   {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2537   {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
2538   {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
2539   {"link",cmd_link,"<oldname> <newname> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2540   {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},
2541   {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2542   {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
2543   {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2544   {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
2545   {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2546   {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},
2547   {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
2548   {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
2549   {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
2550   {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
2551   {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
2552   {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},
2553   {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
2554   {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
2555   {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2556   {"queue",cmd_queue,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2557   {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2558   {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2559   {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},
2560   {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
2561   {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2562   {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
2563   {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2564   {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2565   {"setmode",cmd_setmode,"filename <setmode string> change modes of file",{COMPL_REMOTE,COMPL_NONE}},
2566   {"stat",cmd_stat,"filename Do a UNIX extensions stat call on a file",{COMPL_REMOTE,COMPL_REMOTE}},
2567   {"symlink",cmd_symlink,"<oldname> <newname> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2568   {"tar",cmd_tar,"tar <c|x>[IXFqbgNan] current directory to/from <file name>",{COMPL_NONE,COMPL_NONE}},
2569   {"tarmode",cmd_tarmode,"<full|inc|reset|noreset> tar's behaviour towards archive bits",{COMPL_NONE,COMPL_NONE}},
2570   {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2571   {"vuid",cmd_vuid,"change current vuid",{COMPL_NONE,COMPL_NONE}},
2572   {"logon",cmd_logon,"establish new logon",{COMPL_NONE,COMPL_NONE}},
2573
2574   /* Yes, this must be here, see crh's comment above. */
2575   {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2576   {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2577 };
2578
2579 /*******************************************************************
2580  Lookup a command string in the list of commands, including
2581  abbreviations.
2582 ******************************************************************/
2583
2584 static int process_tok(pstring tok)
2585 {
2586         int i = 0, matches = 0;
2587         int cmd=0;
2588         int tok_len = strlen(tok);
2589
2590         while (commands[i].fn != NULL) {
2591                 if (strequal(commands[i].name,tok)) {
2592                         matches = 1;
2593                         cmd = i;
2594                         break;
2595                 } else if (strnequal(commands[i].name, tok, tok_len)) {
2596                         matches++;
2597                         cmd = i;
2598                 }
2599                 i++;
2600         }
2601
2602         if (matches == 0)
2603                 return(-1);
2604         else if (matches == 1)
2605                 return(cmd);
2606         else
2607                 return(-2);
2608 }
2609
2610 /****************************************************************************
2611  Help.
2612 ****************************************************************************/
2613
2614 static int cmd_help(void)
2615 {
2616         int i=0,j;
2617         pstring buf;
2618
2619         if (next_token_nr(NULL,buf,NULL,sizeof(buf))) {
2620                 if ((i = process_tok(buf)) >= 0)
2621                         d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2622         } else {
2623                 while (commands[i].description) {
2624                         for (j=0; commands[i].description && (j<5); j++) {
2625                                 d_printf("%-15s",commands[i].name);
2626                                 i++;
2627                         }
2628                         d_printf("\n");
2629                 }
2630         }
2631         return 0;
2632 }
2633
2634 /****************************************************************************
2635  Process a -c command string.
2636 ****************************************************************************/
2637
2638 static int process_command_string(char *cmd)
2639 {
2640         pstring line;
2641         const char *ptr;
2642         int rc = 0;
2643
2644         /* establish the connection if not already */
2645
2646         if (!cli) {
2647                 cli = do_connect(desthost, service);
2648                 if (!cli)
2649                         return 0;
2650         }
2651
2652         while (cmd[0] != '\0')    {
2653                 char *p;
2654                 pstring tok;
2655                 int i;
2656
2657                 if ((p = strchr_m(cmd, ';')) == 0) {
2658                         strncpy(line, cmd, 999);
2659                         line[1000] = '\0';
2660                         cmd += strlen(cmd);
2661                 } else {
2662                         if (p - cmd > 999)
2663                                 p = cmd + 999;
2664                         strncpy(line, cmd, p - cmd);
2665                         line[p - cmd] = '\0';
2666                         cmd = p + 1;
2667                 }
2668
2669                 /* and get the first part of the command */
2670                 ptr = line;
2671                 if (!next_token_nr(&ptr,tok,NULL,sizeof(tok))) continue;
2672
2673                 if ((i = process_tok(tok)) >= 0) {
2674                         rc = commands[i].fn();
2675                 } else if (i == -2) {
2676                         d_printf("%s: command abbreviation ambiguous\n",tok);
2677                 } else {
2678                         d_printf("%s: command not found\n",tok);
2679                 }
2680         }
2681
2682         return rc;
2683 }
2684
2685 #define MAX_COMPLETIONS 100
2686
2687 typedef struct {
2688         pstring dirmask;
2689         char **matches;
2690         int count, samelen;
2691         const char *text;
2692         int len;
2693 } completion_remote_t;
2694
2695 static void completion_remote_filter(file_info *f, const char *mask, void *state)
2696 {
2697         completion_remote_t *info = (completion_remote_t *)state;
2698
2699         if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (strcmp(f->name, ".") != 0) && (strcmp(f->name, "..") != 0)) {
2700                 if ((info->dirmask[0] == 0) && !(f->mode & aDIR))
2701                         info->matches[info->count] = SMB_STRDUP(f->name);
2702                 else {
2703                         pstring tmp;
2704
2705                         if (info->dirmask[0] != 0)
2706                                 pstrcpy(tmp, info->dirmask);
2707                         else
2708                                 tmp[0] = 0;
2709                         pstrcat(tmp, f->name);
2710                         if (f->mode & aDIR)
2711                                 pstrcat(tmp, "/");
2712                         info->matches[info->count] = SMB_STRDUP(tmp);
2713                 }
2714                 if (info->matches[info->count] == NULL)
2715                         return;
2716                 if (f->mode & aDIR)
2717                         smb_readline_ca_char(0);
2718
2719                 if (info->count == 1)
2720                         info->samelen = strlen(info->matches[info->count]);
2721                 else
2722                         while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0)
2723                                 info->samelen--;
2724                 info->count++;
2725         }
2726 }
2727
2728 static char **remote_completion(const char *text, int len)
2729 {
2730         pstring dirmask;
2731         int i;
2732         completion_remote_t info = { "", NULL, 1, 0, NULL, 0 };
2733
2734         /* can't have non-static intialisation on Sun CC, so do it
2735            at run time here */
2736         info.samelen = len;
2737         info.text = text;
2738         info.len = len;
2739
2740         if (len >= PATH_MAX)
2741                 return(NULL);
2742
2743         info.matches = SMB_MALLOC_ARRAY(char *,MAX_COMPLETIONS);
2744         if (!info.matches) return NULL;
2745         info.matches[0] = NULL;
2746
2747         for (i = len-1; i >= 0; i--)
2748                 if ((text[i] == '/') || (text[i] == '\\'))
2749                         break;
2750         info.text = text+i+1;
2751         info.samelen = info.len = len-i-1;
2752
2753         if (i > 0) {
2754                 strncpy(info.dirmask, text, i+1);
2755                 info.dirmask[i+1] = 0;
2756                 pstr_sprintf(dirmask, "%s%*s*", cur_dir, i-1, text);
2757         } else
2758                 pstr_sprintf(dirmask, "%s*", cur_dir);
2759
2760         if (cli_list(cli, dirmask, aDIR | aSYSTEM | aHIDDEN, completion_remote_filter, &info) < 0)
2761                 goto cleanup;
2762
2763         if (info.count == 2)
2764                 info.matches[0] = SMB_STRDUP(info.matches[1]);
2765         else {
2766                 info.matches[0] = SMB_MALLOC(info.samelen+1);
2767                 if (!info.matches[0])
2768                         goto cleanup;
2769                 strncpy(info.matches[0], info.matches[1], info.samelen);
2770                 info.matches[0][info.samelen] = 0;
2771         }
2772         info.matches[info.count] = NULL;
2773         return info.matches;
2774
2775 cleanup:
2776         for (i = 0; i < info.count; i++)
2777                 free(info.matches[i]);
2778         free(info.matches);
2779         return NULL;
2780 }
2781
2782 static char **completion_fn(const char *text, int start, int end)
2783 {
2784         smb_readline_ca_char(' ');
2785
2786         if (start) {
2787                 const char *buf, *sp;
2788                 int i;
2789                 char compl_type;
2790
2791                 buf = smb_readline_get_line_buffer();
2792                 if (buf == NULL)
2793                         return NULL;
2794
2795                 sp = strchr(buf, ' ');
2796                 if (sp == NULL)
2797                         return NULL;
2798
2799                 for (i = 0; commands[i].name; i++)
2800                         if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0))
2801                                 break;
2802                 if (commands[i].name == NULL)
2803                         return NULL;
2804
2805                 while (*sp == ' ')
2806                         sp++;
2807
2808                 if (sp == (buf + start))
2809                         compl_type = commands[i].compl_args[0];
2810                 else
2811                         compl_type = commands[i].compl_args[1];
2812
2813                 if (compl_type == COMPL_REMOTE)
2814                         return remote_completion(text, end - start);
2815                 else /* fall back to local filename completion */
2816                         return NULL;
2817         } else {
2818                 char **matches;
2819                 int i, len, samelen = 0, count=1;
2820
2821                 matches = SMB_MALLOC_ARRAY(char *, MAX_COMPLETIONS);
2822                 if (!matches) {
2823                         return NULL;
2824                 }
2825                 matches[0] = NULL;
2826
2827                 len = strlen(text);
2828                 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
2829                         if (strncmp(text, commands[i].name, len) == 0) {
2830                                 matches[count] = SMB_STRDUP(commands[i].name);
2831                                 if (!matches[count])
2832                                         goto cleanup;
2833                                 if (count == 1)
2834                                         samelen = strlen(matches[count]);
2835                                 else
2836                                         while (strncmp(matches[count], matches[count-1], samelen) != 0)
2837                                                 samelen--;
2838                                 count++;
2839                         }
2840                 }
2841
2842                 switch (count) {
2843                 case 0: /* should never happen */
2844                 case 1:
2845                         goto cleanup;
2846                 case 2:
2847                         matches[0] = SMB_STRDUP(matches[1]);
2848                         break;
2849                 default:
2850                         matches[0] = SMB_MALLOC(samelen+1);
2851                         if (!matches[0])
2852                                 goto cleanup;
2853                         strncpy(matches[0], matches[1], samelen);
2854                         matches[0][samelen] = 0;
2855                 }
2856                 matches[count] = NULL;
2857                 return matches;
2858
2859 cleanup:
2860                 for (i = 0; i < count; i++)
2861                         free(matches[i]);
2862
2863                 free(matches);
2864                 return NULL;
2865         }
2866 }
2867
2868 /****************************************************************************
2869  Make sure we swallow keepalives during idle time.
2870 ****************************************************************************/
2871
2872 static void readline_callback(void)
2873 {
2874         fd_set fds;
2875         struct timeval timeout;
2876         static time_t last_t;
2877         time_t t;
2878
2879         t = time(NULL);
2880
2881         if (t - last_t < 5)
2882                 return;
2883
2884         last_t = t;
2885
2886  again:
2887
2888         if (cli->fd == -1)
2889                 return;
2890
2891         FD_ZERO(&fds);
2892         FD_SET(cli->fd,&fds);
2893
2894         timeout.tv_sec = 0;
2895         timeout.tv_usec = 0;
2896         sys_select_intr(cli->fd+1,&fds,NULL,NULL,&timeout);
2897
2898         /* We deliberately use receive_smb instead of
2899            client_receive_smb as we want to receive
2900            session keepalives and then drop them here.
2901         */
2902         if (FD_ISSET(cli->fd,&fds)) {
2903                 receive_smb(cli->fd,cli->inbuf,0);
2904                 goto again;
2905         }
2906
2907         cli_chkpath(cli, "\\");
2908 }
2909
2910 /****************************************************************************
2911  Process commands on stdin.
2912 ****************************************************************************/
2913
2914 static int process_stdin(void)
2915 {
2916         const char *ptr;
2917         int rc = 0;
2918
2919         while (1) {
2920                 pstring tok;
2921                 pstring the_prompt;
2922                 char *cline;
2923                 pstring line;
2924                 int i;
2925
2926                 /* display a prompt */
2927                 slprintf(the_prompt, sizeof(the_prompt)-1, "smb: %s> ", cur_dir);
2928                 cline = smb_readline(the_prompt, readline_callback, completion_fn);
2929
2930                 if (!cline) break;
2931
2932                 pstrcpy(line, cline);
2933
2934                 /* special case - first char is ! */
2935                 if (*line == '!') {
2936                         system(line + 1);
2937                         continue;
2938                 }
2939
2940                 /* and get the first part of the command */
2941                 ptr = line;
2942                 if (!next_token_nr(&ptr,tok,NULL,sizeof(tok))) continue;
2943
2944                 if ((i = process_tok(tok)) >= 0) {
2945                         rc = commands[i].fn();
2946                 } else if (i == -2) {
2947                         d_printf("%s: command abbreviation ambiguous\n",tok);
2948                 } else {
2949                         d_printf("%s: command not found\n",tok);
2950                 }
2951         }
2952         return rc;
2953 }
2954
2955 /*****************************************************
2956  Return a connection to a server.
2957 *******************************************************/
2958
2959 static struct cli_state *do_connect(const char *server, const char *share)
2960 {
2961         struct cli_state *c;
2962         struct nmb_name called, calling;
2963         const char *server_n;
2964         struct in_addr ip;
2965         pstring servicename;
2966         char *sharename;
2967
2968         /* make a copy so we don't modify the global string 'service' */
2969         pstrcpy(servicename, share);
2970         sharename = servicename;
2971         if (*sharename == '\\') {
2972                 server = sharename+2;
2973                 sharename = strchr_m(server,'\\');
2974                 if (!sharename) return NULL;
2975                 *sharename = 0;
2976                 sharename++;
2977         }
2978
2979         server_n = server;
2980
2981         zero_ip(&ip);
2982
2983         make_nmb_name(&calling, global_myname(), 0x0);
2984         make_nmb_name(&called , server, name_type);
2985
2986  again:
2987         zero_ip(&ip);
2988         if (have_ip) ip = dest_ip;
2989
2990         /* have to open a new connection */
2991         if (!(c=cli_initialise(NULL)) || (cli_set_port(c, port) != port) ||
2992             !cli_connect(c, server_n, &ip)) {
2993                 d_printf("Connection to %s failed\n", server_n);
2994                 return NULL;
2995         }
2996
2997         c->protocol = max_protocol;
2998         c->use_kerberos = use_kerberos;
2999         cli_setup_signing_state(c, cmdline_auth_info.signing_state);
3000
3001
3002         if (!cli_session_request(c, &calling, &called)) {
3003                 char *p;
3004                 d_printf("session request to %s failed (%s)\n",
3005                          called.name, cli_errstr(c));
3006                 cli_shutdown(c);
3007                 if ((p=strchr_m(called.name, '.'))) {
3008                         *p = 0;
3009                         goto again;
3010                 }
3011                 if (strcmp(called.name, "*SMBSERVER")) {
3012                         make_nmb_name(&called , "*SMBSERVER", 0x20);
3013                         goto again;
3014                 }
3015                 return NULL;
3016         }
3017
3018         DEBUG(4,(" session request ok\n"));
3019
3020         if (!cli_negprot(c)) {
3021                 d_printf("protocol negotiation failed\n");
3022                 cli_shutdown(c);
3023                 return NULL;
3024         }
3025
3026         if (!got_pass) {
3027                 char *pass = getpass("Password: ");
3028                 if (pass) {
3029                         pstrcpy(password, pass);
3030                         got_pass = 1;
3031                 }
3032         }
3033
3034         if (!cli_session_setup(c, username,
3035                                password, strlen(password),
3036                                password, strlen(password),
3037                                lp_workgroup())) {
3038                 /* if a password was not supplied then try again with a null username */
3039                 if (password[0] || !username[0] || use_kerberos ||
3040                     !cli_session_setup(c, "", "", 0, "", 0, lp_workgroup())) {
3041                         d_printf("session setup failed: %s\n", cli_errstr(c));
3042                         if (NT_STATUS_V(cli_nt_error(c)) ==
3043                             NT_STATUS_V(NT_STATUS_MORE_PROCESSING_REQUIRED))
3044                                 d_printf("did you forget to run kinit?\n");
3045                         cli_shutdown(c);
3046                         return NULL;
3047                 }
3048                 d_printf("Anonymous login successful\n");
3049         }
3050
3051         if (*c->server_domain) {
3052                 DEBUG(1,("Domain=[%s] OS=[%s] Server=[%s]\n",
3053                         c->server_domain,c->server_os,c->server_type));
3054         } else if (*c->server_os || *c->server_type){
3055                 DEBUG(1,("OS=[%s] Server=[%s]\n",
3056                          c->server_os,c->server_type));
3057         }
3058         DEBUG(4,(" session setup ok\n"));
3059
3060         if (!cli_send_tconX(c, sharename, "?????",
3061                             password, strlen(password)+1)) {
3062                 d_printf("tree connect failed: %s\n", cli_errstr(c));
3063                 cli_shutdown(c);
3064                 return NULL;
3065         }
3066
3067         DEBUG(4,(" tconx ok\n"));
3068
3069         return c;
3070 }
3071
3072 /****************************************************************************
3073  Process commands from the client.
3074 ****************************************************************************/
3075
3076 static int process(char *base_directory)
3077 {
3078         int rc = 0;
3079
3080         cli = do_connect(desthost, service);
3081         if (!cli) {
3082                 return 1;
3083         }
3084
3085         if (*base_directory) do_cd(base_directory);
3086
3087         if (cmdstr) {
3088                 rc = process_command_string(cmdstr);
3089         } else {
3090                 process_stdin();
3091         }
3092
3093         cli_shutdown(cli);
3094         return rc;
3095 }
3096
3097 /****************************************************************************
3098  Handle a -L query.
3099 ****************************************************************************/
3100
3101 static int do_host_query(char *query_host)
3102 {
3103         cli = do_connect(query_host, "IPC$");
3104         if (!cli)
3105                 return 1;
3106
3107         browse_host(True);
3108
3109         if (port != 139) {
3110
3111                 /* Workgroups simply don't make sense over anything
3112                    else but port 139... */
3113
3114                 cli_shutdown(cli);
3115                 port = 139;
3116                 cli = do_connect(query_host, "IPC$");
3117         }
3118
3119         if (cli == NULL) {
3120                 d_printf("NetBIOS over TCP disabled -- no workgroup available\n");
3121                 return 1;
3122         }
3123
3124         list_servers(lp_workgroup());
3125
3126         cli_shutdown(cli);
3127
3128         return(0);
3129 }
3130
3131
3132 /****************************************************************************
3133  Handle a tar operation.
3134 ****************************************************************************/
3135
3136 static int do_tar_op(char *base_directory)
3137 {
3138         int ret;
3139
3140         /* do we already have a connection? */
3141         if (!cli) {
3142                 cli = do_connect(desthost, service);
3143                 if (!cli)
3144                         return 1;
3145         }
3146
3147         recurse=True;
3148
3149         if (*base_directory) do_cd(base_directory);
3150
3151         ret=process_tar();
3152
3153         cli_shutdown(cli);
3154
3155         return(ret);
3156 }
3157
3158 /****************************************************************************
3159  Handle a message operation.
3160 ****************************************************************************/
3161
3162 static int do_message_op(void)
3163 {
3164         struct in_addr ip;
3165         struct nmb_name called, calling;
3166         fstring server_name;
3167         char name_type_hex[10];
3168
3169         make_nmb_name(&calling, global_myname(), 0x0);
3170         make_nmb_name(&called , desthost, name_type);
3171
3172         fstrcpy(server_name, desthost);
3173         snprintf(name_type_hex, sizeof(name_type_hex), "#%X", name_type);
3174         fstrcat(server_name, name_type_hex);
3175
3176         zero_ip(&ip);
3177         if (have_ip) ip = dest_ip;
3178
3179         if (!(cli=cli_initialise(NULL)) || (cli_set_port(cli, port) != port) ||
3180             !cli_connect(cli, server_name, &ip)) {
3181                 d_printf("Connection to %s failed\n", desthost);
3182                 return 1;
3183         }
3184
3185         if (!cli_session_request(cli, &calling, &called)) {
3186                 d_printf("session request failed\n");
3187                 cli_shutdown(cli);
3188                 return 1;
3189         }
3190
3191         send_message();
3192         cli_shutdown(cli);
3193
3194         return 0;
3195 }
3196
3197
3198 /****************************************************************************
3199   main program
3200 ****************************************************************************/
3201
3202  int main(int argc,char *argv[])
3203 {
3204         extern BOOL AllowDebugChange;
3205         extern BOOL override_logfile;
3206         pstring base_directory;
3207         int opt;
3208         pstring query_host;
3209         BOOL message = False;
3210         extern char tar_type;
3211         pstring term_code;
3212         static const char *new_name_resolve_order = NULL;
3213         poptContext pc;
3214         char *p;
3215         int rc = 0;
3216         fstring new_workgroup;
3217         struct poptOption long_options[] = {
3218                 POPT_AUTOHELP
3219
3220                 { "name-resolve", 'R', POPT_ARG_STRING, &new_name_resolve_order, 'R', "Use these name resolution services only", "NAME-RESOLVE-ORDER" },
3221                 { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" },
3222                 { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" },
3223                 { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" },
3224                 { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" },
3225                 { "terminal", 't', POPT_ARG_STRING, NULL, 't', "Terminal I/O code {sjis|euc|jis7|jis8|junet|hex}", "CODE" },
3226                 { "max-protocol", 'm', POPT_ARG_STRING, NULL, 'm', "Set the max protocol level", "LEVEL" },
3227                 { "tar", 'T', POPT_ARG_STRING, NULL, 'T', "Command line tar", "<c|x>IXFqgbNan" },
3228                 { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" },
3229                 { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" },
3230                 { "send-buffer", 'b', POPT_ARG_INT, &io_bufsize, 'b', "Changes the transmit/send buffer", "BYTES" },
3231                 { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" },
3232                 { "grepable", 'g', POPT_ARG_NONE, NULL, 'g', "Produce grepable output" },
3233                 POPT_COMMON_SAMBA
3234                 POPT_COMMON_CONNECTION
3235                 POPT_COMMON_CREDENTIALS
3236                 POPT_TABLEEND
3237         };
3238
3239
3240 #ifdef KANJI
3241         pstrcpy(term_code, KANJI);
3242 #else /* KANJI */
3243         *term_code = 0;
3244 #endif /* KANJI */
3245
3246         *query_host = 0;
3247         *base_directory = 0;
3248
3249         /* initialize the workgroup name so we can determine whether or
3250            not it was set by a command line option */
3251
3252         set_global_myworkgroup( "" );
3253
3254         /* set default debug level to 0 regardless of what smb.conf sets */
3255         setup_logging( "smbclient", True );
3256         DEBUGLEVEL_CLASS[DBGC_ALL] = 1;
3257         dbf = x_stderr;
3258         x_setbuf( x_stderr, NULL );
3259
3260         pc = poptGetContext("smbclient", argc, (const char **) argv, long_options,
3261                                 POPT_CONTEXT_KEEP_FIRST);
3262         poptSetOtherOptionHelp(pc, "service <password>");
3263
3264         in_client = True;   /* Make sure that we tell lp_load we are */
3265
3266         while ((opt = poptGetNextOpt(pc)) != -1) {
3267                 switch (opt) {
3268                 case 'M':
3269                         /* Messages are sent to NetBIOS name type 0x3
3270                          * (Messenger Service).  Make sure we default
3271                          * to port 139 instead of port 445. srl,crh
3272                          */
3273                         name_type = 0x03;
3274                         pstrcpy(desthost,poptGetOptArg(pc));
3275                         if( 0 == port )
3276                                 port = 139;
3277                         message = True;
3278                         break;
3279                 case 'I':
3280                         {
3281                                 dest_ip = *interpret_addr2(poptGetOptArg(pc));
3282                                 if (is_zero_ip(dest_ip))
3283                                         exit(1);
3284                                 have_ip = True;
3285                         }
3286                         break;
3287                 case 'E':
3288                         dbf = x_stderr;
3289                         display_set_stderr();
3290                         break;
3291
3292                 case 'L':
3293                         pstrcpy(query_host, poptGetOptArg(pc));
3294                         break;
3295                 case 't':
3296                         pstrcpy(term_code, poptGetOptArg(pc));
3297                         break;
3298                 case 'm':
3299                         max_protocol = interpret_protocol(poptGetOptArg(pc), max_protocol);
3300                         break;
3301                 case 'T':
3302                         /* We must use old option processing for this. Find the
3303                          * position of the -T option in the raw argv[]. */
3304                         {
3305                                 int i, optnum;
3306                                 for (i = 1; i < argc; i++) {
3307                                         if (strncmp("-T", argv[i],2)==0)
3308                                                 break;
3309                                 }
3310                                 i++;
3311                                 if (!(optnum = tar_parseargs(argc, argv, poptGetOptArg(pc), i))) {
3312                                         poptPrintUsage(pc, stderr, 0);
3313                                         exit(1);
3314                                 }
3315                                 /* Now we must eat (optnum - i) options - they have
3316                                  * been processed by tar_parseargs().
3317                                  */
3318                                 optnum -= i;
3319                                 for (i = 0; i < optnum; i++)
3320                                         poptGetOptArg(pc);
3321                         }
3322                         break;
3323                 case 'D':
3324                         pstrcpy(base_directory,poptGetOptArg(pc));
3325                         break;
3326                 case 'g':
3327                         grepable=True;
3328                         break;
3329                 }
3330         }
3331
3332         poptGetArg(pc);
3333
3334         /*
3335          * Don't load debug level from smb.conf. It should be
3336          * set by cmdline arg or remain default (0)
3337          */
3338         AllowDebugChange = False;
3339
3340         /* save the workgroup...
3341
3342            FIXME!! do we need to do this for other options as well
3343            (or maybe a generic way to keep lp_load() from overwriting
3344            everything)?  */
3345
3346         fstrcpy( new_workgroup, lp_workgroup() );
3347
3348         if ( override_logfile )
3349                 setup_logging( lp_logfile(), False );
3350
3351         if (!lp_load(dyn_CONFIGFILE,True,False,False)) {
3352                 fprintf(stderr, "%s: Can't load %s - run testparm to debug it\n",
3353                         argv[0], dyn_CONFIGFILE);
3354         }
3355
3356         load_interfaces();
3357
3358         if ( strlen(new_workgroup) != 0 )
3359                 set_global_myworkgroup( new_workgroup );
3360
3361         if(poptPeekArg(pc)) {
3362                 pstrcpy(service,poptGetArg(pc));
3363                 /* Convert any '/' characters in the service name to '\' characters */
3364                 string_replace(service, '/','\\');
3365
3366                 if (count_chars(service,'\\') < 3) {
3367                         d_printf("\n%s: Not enough '\\' characters in service\n",service);
3368                         poptPrintUsage(pc, stderr, 0);
3369                         exit(1);
3370                 }
3371         }
3372
3373         if (poptPeekArg(pc) && !cmdline_auth_info.got_pass) {
3374                 cmdline_auth_info.got_pass = True;
3375                 pstrcpy(cmdline_auth_info.password,poptGetArg(pc));
3376         }
3377
3378         init_names();
3379
3380         if(new_name_resolve_order)
3381                 lp_set_name_resolve_order(new_name_resolve_order);
3382
3383         if (!tar_type && !*query_host && !*service && !message) {
3384                 poptPrintUsage(pc, stderr, 0);
3385                 exit(1);
3386         }
3387
3388         poptFreeContext(pc);
3389
3390         pstrcpy(username, cmdline_auth_info.username);
3391         pstrcpy(password, cmdline_auth_info.password);
3392
3393         use_kerberos = cmdline_auth_info.use_kerberos;
3394         got_pass = cmdline_auth_info.got_pass;
3395
3396         DEBUG(3,("Client started (version %s).\n", SAMBA_VERSION_STRING));
3397
3398         if (tar_type) {
3399                 if (cmdstr)
3400                         process_command_string(cmdstr);
3401                 return do_tar_op(base_directory);
3402         }
3403
3404         if (*query_host) {
3405                 char *qhost = query_host;
3406                 char *slash;
3407
3408                 while (*qhost == '\\' || *qhost == '/')
3409                         qhost++;
3410
3411                 if ((slash = strchr_m(qhost, '/'))
3412                     || (slash = strchr_m(qhost, '\\'))) {
3413                         *slash = 0;
3414                 }
3415
3416                 if ((p=strchr_m(qhost, '#'))) {
3417                         *p = 0;
3418                         p++;
3419                         sscanf(p, "%x", &name_type);
3420                 }
3421
3422                 return do_host_query(qhost);
3423         }
3424
3425         if (message) {
3426                 return do_message_op();
3427         }
3428
3429         if (process(base_directory)) {
3430                 return 1;
3431         }
3432
3433         return rc;
3434 }