source/client/client.c

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