source3/client/client.c

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