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