source3/client/client.c

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