source4/client/client.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    SMB client
   4    Copyright (C) Andrew Tridgell 1994-1998
   5    Copyright (C) Simo Sorce 2001-2002
   6    Copyright (C) Jelmer Vernooij 2003-2004
   7    Copyright (C) James J Myers   2003 <myersjj@samba.org>
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 /*
  24  * TODO: remove this ... and don't use talloc_append_string()
  25  *
  26  * NOTE: I'm not changing the code yet, because I assume there're
  27  *       some bugs in the existing code and I'm not sure how to fix
  28  *       them correctly.
  29  */
  30 #define TALLOC_DEPRECATED 1
  31
  32 #include "includes.h"
  33 #include "version.h"
  34 #include "libcli/libcli.h"
  35 #include "lib/events/events.h"
  36 #include "lib/cmdline/popt_common.h"
  37 #include "librpc/gen_ndr/ndr_srvsvc_c.h"
  38 #include "librpc/gen_ndr/ndr_lsa.h"
  39 #include "librpc/gen_ndr/ndr_security.h"
  40 #include "libcli/util/clilsa.h"
  41 #include "system/dir.h"
  42 #include "system/filesys.h"
  43 #include "../lib/util/dlinklist.h"
  44 #include "system/readline.h"
  45 #include "auth/credentials/credentials.h"
  46 #include "auth/gensec/gensec.h"
  47 #include "system/time.h" /* needed by some systems for asctime() */
  48 #include "libcli/resolve/resolve.h"
  49 #include "libcli/security/security.h"
  50 #include "../libcli/smbreadline/smbreadline.h"
  51 #include "librpc/gen_ndr/ndr_nbt.h"
  52 #include "param/param.h"
  53 #include "libcli/raw/raw_proto.h"
  54
  55 /* the default pager to use for the client "more" command. Users can
  56  *    override this with the PAGER environment variable */
  57 #ifndef DEFAULT_PAGER
  58 #define DEFAULT_PAGER "more"
  59 #endif
  60
  61 struct smbclient_context {
  62         char *remote_cur_dir;
  63         struct smbcli_state *cli;
  64         char *fileselection;
  65         time_t newer_than;
  66         bool prompt;
  67         bool recurse;
  68         int archive_level;
  69         bool lowercase;
  70         int printmode;
  71         bool translation;
  72         int io_bufsize;
  73 };
  74
  75 /* timing globals */
  76 static uint64_t get_total_size = 0;
  77 static unsigned int get_total_time_ms = 0;
  78 static uint64_t put_total_size = 0;
  79 static unsigned int put_total_time_ms = 0;
  80
  81 /* Unfortunately, there is no way to pass the a context to the completion function as an argument */
  82 static struct smbclient_context *rl_ctx;
  83
  84 /* totals globals */
  85 static double dir_total;
  86
  87 /*******************************************************************
  88  Reduce a file name, removing .. elements.
  89 ********************************************************************/
  90 static void dos_clean_name(char *s)
  91 {
  92         char *p=NULL,*r;
  93
  94         DEBUG(3,("dos_clean_name [%s]\n",s));
  95
  96         /* remove any double slashes */
  97         all_string_sub(s, "\\\\", "\\", 0);
  98
  99         while ((p = strstr(s,"\\..\\")) != NULL) {
 100                 *p = '\0';
 101                 if ((r = strrchr(s,'\\')) != NULL)
 102                         memmove(r,p+3,strlen(p+3)+1);
 103         }
 104
 105         trim_string(s,NULL,"\\..");
 106
 107         all_string_sub(s, "\\.\\", "\\", 0);
 108 }
 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, const void *_b, int n, bool translation)
 115 {
 116         const uint8_t *b = (const uint8_t *)_b;
 117         int i;
 118
 119         if (!translation) {
 120                 return write(f,b,n);
 121         }
 122
 123         i = 0;
 124         while (i < n) {
 125                 if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') {
 126                         b++;i++;
 127                 }
 128                 if (write(f, b, 1) != 1) {
 129                         break;
 130                 }
 131                 b++;
 132                 i++;
 133         }
 134
 135         return(i);
 136 }
 137
 138 /****************************************************************************
 139   read from a file with LF->CR/LF translation if appropriate. return the
 140   number read. read approx n bytes.
 141 ****************************************************************************/
 142 static int readfile(void *_b, int n, FILE *f, bool translation)
 143 {
 144         uint8_t *b = (uint8_t *)_b;
 145         int i;
 146         int c;
 147
 148         if (!translation)
 149                 return fread(b,1,n,f);
 150
 151         i = 0;
 152         while (i < (n - 1)) {
 153                 if ((c = getc(f)) == EOF) {
 154                         break;
 155                 }
 156
 157                 if (c == '\n') { /* change all LFs to CR/LF */
 158                         b[i++] = '\r';
 159                 }
 160
 161                 b[i++] = c;
 162         }
 163
 164         return(i);
 165 }
 166
 167
 168 /****************************************************************************
 169 send a message
 170 ****************************************************************************/
 171 static void send_message(struct smbcli_state *cli, const char *desthost)
 172 {
 173         char msg[1600];
 174         int total_len = 0;
 175         int grp_id;
 176
 177         if (!smbcli_message_start(cli->tree,
 178                         desthost,
 179                         cli_credentials_get_username(
 180                                 popt_get_cmdline_credentials()),
 181                         &grp_id)) {
 182                 d_printf("message start: %s\n", smbcli_errstr(cli->tree));
 183                 return;
 184         }
 185
 186
 187         d_printf("Connected. Type your message, ending it with a Control-D\n");
 188
 189         while (!feof(stdin) && total_len < 1600) {
 190                 int maxlen = MIN(1600 - total_len,127);
 191                 int l=0;
 192                 int c;
 193
 194                 for (l=0;l<maxlen && (c=fgetc(stdin))!=EOF;l++) {
 195                         if (c == '\n')
 196                                 msg[l++] = '\r';
 197                         msg[l] = c;
 198                 }
 199
 200                 if (!smbcli_message_text(cli->tree, msg, l, grp_id)) {
 201                         d_printf("SMBsendtxt failed (%s)\n",smbcli_errstr(cli->tree));
 202                         return;
 203                 }
 204
 205                 total_len += l;
 206         }
 207
 208         if (total_len >= 1600)
 209                 d_printf("the message was truncated to 1600 bytes\n");
 210         else
 211                 d_printf("sent %d bytes\n",total_len);
 212
 213         if (!smbcli_message_end(cli->tree, grp_id)) {
 214                 d_printf("SMBsendend failed (%s)\n",smbcli_errstr(cli->tree));
 215                 return;
 216         }
 217 }
 218
 219
 220
 221 /****************************************************************************
 222 check the space on a device
 223 ****************************************************************************/
 224 static int do_dskattr(struct smbclient_context *ctx)
 225 {
 226         uint32_t bsize;
 227         uint64_t total, avail;
 228
 229         if (NT_STATUS_IS_ERR(smbcli_dskattr(ctx->cli->tree, &bsize, &total, &avail))) {
 230                 d_printf("Error in dskattr: %s\n",smbcli_errstr(ctx->cli->tree));
 231                 return 1;
 232         }
 233
 234         d_printf("\n\t\t%llu blocks of size %u. %llu blocks available\n",
 235                  (unsigned long long)total,
 236                  (unsigned)bsize,
 237                  (unsigned long long)avail);
 238
 239         return 0;
 240 }
 241
 242 /****************************************************************************
 243 show cd/pwd
 244 ****************************************************************************/
 245 static int cmd_pwd(struct smbclient_context *ctx, const char **args)
 246 {
 247         d_printf("Current directory is %s\n", ctx->remote_cur_dir);
 248         return 0;
 249 }
 250
 251 /*
 252   convert a string to dos format
 253 */
 254 static void dos_format(char *s)
 255 {
 256         string_replace(s, '/', '\\');
 257 }
 258
 259 /****************************************************************************
 260 change directory - inner section
 261 ****************************************************************************/
 262 static int do_cd(struct smbclient_context *ctx, const char *newdir)
 263 {
 264         char *dname;
 265
 266         /* Save the current directory in case the
 267            new directory is invalid */
 268         if (newdir[0] == '\\')
 269                 dname = talloc_strdup(ctx, newdir);
 270         else
 271                 dname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, newdir);
 272
 273         dos_format(dname);
 274
 275         if (*(dname+strlen(dname)-1) != '\\') {
 276                 dname = talloc_append_string(NULL, dname, "\\");
 277         }
 278         dos_clean_name(dname);
 279
 280         if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, dname))) {
 281                 d_printf("cd %s: %s\n", dname, smbcli_errstr(ctx->cli->tree));
 282                 talloc_free(dname);
 283         } else {
 284                 ctx->remote_cur_dir = dname;
 285         }
 286
 287         return 0;
 288 }
 289
 290 /****************************************************************************
 291 change directory
 292 ****************************************************************************/
 293 static int cmd_cd(struct smbclient_context *ctx, const char **args)
 294 {
 295         int rc = 0;
 296
 297         if (args[1])
 298                 rc = do_cd(ctx, args[1]);
 299         else
 300                 d_printf("Current directory is %s\n",ctx->remote_cur_dir);
 301
 302         return rc;
 303 }
 304
 305
 306 static bool mask_match(struct smbcli_state *c, const char *string,
 307                 const char *pattern, bool is_case_sensitive)
 308 {
 309         int ret;
 310
 311         if (ISDOTDOT(string))
 312                 string = ".";
 313         if (ISDOT(pattern))
 314                 return false;
 315
 316         ret = ms_fnmatch_protocol(pattern, string,
 317                                   c->transport->negotiate.protocol,
 318                                   is_case_sensitive);
 319         return (ret == 0);
 320 }
 321
 322
 323
 324 /*******************************************************************
 325   decide if a file should be operated on
 326   ********************************************************************/
 327 static bool do_this_one(struct smbclient_context *ctx, struct clilist_file_info *finfo)
 328 {
 329         if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY) return(true);
 330
 331         if (ctx->fileselection &&
 332             !mask_match(ctx->cli, finfo->name,ctx->fileselection,false)) {
 333                 DEBUG(3,("mask_match %s failed\n", finfo->name));
 334                 return false;
 335         }
 336
 337         if (ctx->newer_than && finfo->mtime < ctx->newer_than) {
 338                 DEBUG(3,("newer_than %s failed\n", finfo->name));
 339                 return(false);
 340         }
 341
 342         if ((ctx->archive_level==1 || ctx->archive_level==2) && !(finfo->attrib & FILE_ATTRIBUTE_ARCHIVE)) {
 343                 DEBUG(3,("archive %s failed\n", finfo->name));
 344                 return(false);
 345         }
 346
 347         return(true);
 348 }
 349
 350 /****************************************************************************
 351   display info about a file
 352   ****************************************************************************/
 353 static void display_finfo(struct smbclient_context *ctx, struct clilist_file_info *finfo)
 354 {
 355         if (do_this_one(ctx, finfo)) {
 356                 time_t t = finfo->mtime; /* the time is assumed to be passed as GMT */
 357                 char *astr = attrib_string(NULL, finfo->attrib);
 358                 d_printf("  %-30s%7.7s %8.0f  %s",
 359                          finfo->name,
 360                          astr,
 361                          (double)finfo->size,
 362                          asctime(localtime(&t)));
 363                 dir_total += finfo->size;
 364                 talloc_free(astr);
 365         }
 366 }
 367
 368
 369 /****************************************************************************
 370    accumulate size of a file
 371   ****************************************************************************/
 372 static void do_du(struct smbclient_context *ctx, struct clilist_file_info *finfo)
 373 {
 374         if (do_this_one(ctx, finfo)) {
 375                 dir_total += finfo->size;
 376         }
 377 }
 378
 379 static bool do_list_recurse;
 380 static bool do_list_dirs;
 381 static char *do_list_queue = 0;
 382 static long do_list_queue_size = 0;
 383 static long do_list_queue_start = 0;
 384 static long do_list_queue_end = 0;
 385 static void (*do_list_fn)(struct smbclient_context *, struct clilist_file_info *);
 386
 387 /****************************************************************************
 388 functions for do_list_queue
 389   ****************************************************************************/
 390
 391 /*
 392  * The do_list_queue is a NUL-separated list of strings stored in a
 393  * char*.  Since this is a FIFO, we keep track of the beginning and
 394  * ending locations of the data in the queue.  When we overflow, we
 395  * double the size of the char*.  When the start of the data passes
 396  * the midpoint, we move everything back.  This is logically more
 397  * complex than a linked list, but easier from a memory management
 398  * angle.  In any memory error condition, do_list_queue is reset.
 399  * Functions check to ensure that do_list_queue is non-NULL before
 400  * accessing it.
 401  */
 402 static void reset_do_list_queue(void)
 403 {
 404         SAFE_FREE(do_list_queue);
 405         do_list_queue_size = 0;
 406         do_list_queue_start = 0;
 407         do_list_queue_end = 0;
 408 }
 409
 410 static void init_do_list_queue(void)
 411 {
 412         reset_do_list_queue();
 413         do_list_queue_size = 1024;
 414         do_list_queue = malloc_array_p(char, do_list_queue_size);
 415         if (do_list_queue == 0) {
 416                 d_printf("malloc fail for size %d\n",
 417                          (int)do_list_queue_size);
 418                 reset_do_list_queue();
 419         } else {
 420                 memset(do_list_queue, 0, do_list_queue_size);
 421         }
 422 }
 423
 424 static void adjust_do_list_queue(void)
 425 {
 426         if (do_list_queue == NULL) return;
 427
 428         /*
 429          * If the starting point of the queue is more than half way through,
 430          * move everything toward the beginning.
 431          */
 432         if (do_list_queue_start == do_list_queue_end)
 433         {
 434                 DEBUG(4,("do_list_queue is empty\n"));
 435                 do_list_queue_start = do_list_queue_end = 0;
 436                 *do_list_queue = '\0';
 437         }
 438         else if (do_list_queue_start > (do_list_queue_size / 2))
 439         {
 440                 DEBUG(4,("sliding do_list_queue backward\n"));
 441                 memmove(do_list_queue,
 442                         do_list_queue + do_list_queue_start,
 443                         do_list_queue_end - do_list_queue_start);
 444                 do_list_queue_end -= do_list_queue_start;
 445                 do_list_queue_start = 0;
 446         }
 447
 448 }
 449
 450 static void add_to_do_list_queue(const char* entry)
 451 {
 452         char *dlq;
 453         long new_end;
 454
 455         if (entry == NULL) {
 456                 entry = "";
 457         }
 458
 459         new_end = do_list_queue_end + ((long)strlen(entry)) + 1;
 460         while (new_end > do_list_queue_size)
 461         {
 462                 do_list_queue_size *= 2;
 463                 DEBUG(4,("enlarging do_list_queue to %d\n",
 464                          (int)do_list_queue_size));
 465                 dlq = realloc_p(do_list_queue, char, do_list_queue_size);
 466                 if (! dlq) {
 467                         d_printf("failure enlarging do_list_queue to %d bytes\n",
 468                                  (int)do_list_queue_size);
 469                         reset_do_list_queue();
 470                 }
 471                 else
 472                 {
 473                         do_list_queue = dlq;
 474                         memset(do_list_queue + do_list_queue_size / 2,
 475                                0, do_list_queue_size / 2);
 476                 }
 477         }
 478         if (do_list_queue)
 479         {
 480                 strlcpy(do_list_queue + do_list_queue_end, entry,
 481                             do_list_queue_size - do_list_queue_end);
 482                 do_list_queue_end = new_end;
 483                 DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n",
 484                          entry, (int)do_list_queue_start, (int)do_list_queue_end));
 485         }
 486 }
 487
 488 static char *do_list_queue_head(void)
 489 {
 490         return do_list_queue + do_list_queue_start;
 491 }
 492
 493 static void remove_do_list_queue_head(void)
 494 {
 495         if (do_list_queue_end > do_list_queue_start)
 496         {
 497                 do_list_queue_start += strlen(do_list_queue_head()) + 1;
 498                 adjust_do_list_queue();
 499                 DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n",
 500                          (int)do_list_queue_start, (int)do_list_queue_end));
 501         }
 502 }
 503
 504 static int do_list_queue_empty(void)
 505 {
 506         return (! (do_list_queue && *do_list_queue));
 507 }
 508
 509 /****************************************************************************
 510 a helper for do_list
 511   ****************************************************************************/
 512 static void do_list_helper(struct clilist_file_info *f, const char *mask, void *state)
 513 {
 514         struct smbclient_context *ctx = (struct smbclient_context *)state;
 515
 516         if (f->attrib & FILE_ATTRIBUTE_DIRECTORY) {
 517                 if (do_list_dirs && do_this_one(ctx, f)) {
 518                         do_list_fn(ctx, f);
 519                 }
 520                 if (do_list_recurse &&
 521                     !ISDOT(f->name) &&
 522                     !ISDOTDOT(f->name)) {
 523                         char *mask2;
 524                         char *p;
 525
 526                         mask2 = talloc_strdup(NULL, mask);
 527                         p = strrchr_m(mask2,'\\');
 528                         if (!p) return;
 529                         p[1] = 0;
 530                         mask2 = talloc_asprintf_append_buffer(mask2, "%s\\*", f->name);
 531                         add_to_do_list_queue(mask2);
 532                 }
 533                 return;
 534         }
 535
 536         if (do_this_one(ctx, f)) {
 537                 do_list_fn(ctx, f);
 538         }
 539 }
 540
 541
 542 /****************************************************************************
 543 a wrapper around smbcli_list that adds recursion
 544   ****************************************************************************/
 545 static void do_list(struct smbclient_context *ctx, const char *mask,uint16_t attribute,
 546              void (*fn)(struct smbclient_context *, struct clilist_file_info *),bool rec, bool dirs)
 547 {
 548         static int in_do_list = 0;
 549
 550         if (in_do_list && rec)
 551         {
 552                 fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n");
 553                 exit(1);
 554         }
 555
 556         in_do_list = 1;
 557
 558         do_list_recurse = rec;
 559         do_list_dirs = dirs;
 560         do_list_fn = fn;
 561
 562         if (rec)
 563         {
 564                 init_do_list_queue();
 565                 add_to_do_list_queue(mask);
 566
 567                 while (! do_list_queue_empty())
 568                 {
 569                         /*
 570                          * Need to copy head so that it doesn't become
 571                          * invalid inside the call to smbcli_list.  This
 572                          * would happen if the list were expanded
 573                          * during the call.
 574                          * Fix from E. Jay Berkenbilt (ejb@ql.org)
 575                          */
 576                         char *head;
 577                         head = do_list_queue_head();
 578                         smbcli_list(ctx->cli->tree, head, attribute, do_list_helper, ctx);
 579                         remove_do_list_queue_head();
 580                         if ((! do_list_queue_empty()) && (fn == display_finfo))
 581                         {
 582                                 char* next_file = do_list_queue_head();
 583                                 char* save_ch = 0;
 584                                 if ((strlen(next_file) >= 2) &&
 585                                     (next_file[strlen(next_file) - 1] == '*') &&
 586                                     (next_file[strlen(next_file) - 2] == '\\'))
 587                                 {
 588                                         save_ch = next_file +
 589                                                 strlen(next_file) - 2;
 590                                         *save_ch = '\0';
 591                                 }
 592                                 d_printf("\n%s\n",next_file);
 593                                 if (save_ch)
 594                                 {
 595                                         *save_ch = '\\';
 596                                 }
 597                         }
 598                 }
 599         }
 600         else
 601         {
 602                 if (smbcli_list(ctx->cli->tree, mask, attribute, do_list_helper, ctx) == -1)
 603                 {
 604                         d_printf("%s listing %s\n", smbcli_errstr(ctx->cli->tree), mask);
 605                 }
 606         }
 607
 608         in_do_list = 0;
 609         reset_do_list_queue();
 610 }
 611
 612 /****************************************************************************
 613   get a directory listing
 614   ****************************************************************************/
 615 static int cmd_dir(struct smbclient_context *ctx, const char **args)
 616 {
 617         uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
 618         char *mask;
 619         int rc;
 620
 621         dir_total = 0;
 622
 623         mask = talloc_strdup(ctx, ctx->remote_cur_dir);
 624         if(mask[strlen(mask)-1]!='\\')
 625                 mask = talloc_append_string(ctx, mask,"\\");
 626
 627         if (args[1]) {
 628                 mask = talloc_strdup(ctx, args[1]);
 629                 if (mask[0] != '\\')
 630                         mask = talloc_append_string(ctx, mask, "\\");
 631                 dos_format(mask);
 632         }
 633         else {
 634                 if (ctx->cli->tree->session->transport->negotiate.protocol <=
 635                     PROTOCOL_LANMAN1) {
 636                         mask = talloc_append_string(ctx, mask, "*.*");
 637                 } else {
 638                         mask = talloc_append_string(ctx, mask, "*");
 639                 }
 640         }
 641
 642         do_list(ctx, mask, attribute, display_finfo, ctx->recurse, true);
 643
 644         rc = do_dskattr(ctx);
 645
 646         DEBUG(3, ("Total bytes listed: %.0f\n", dir_total));
 647
 648         return rc;
 649 }
 650
 651
 652 /****************************************************************************
 653   get a directory listing
 654   ****************************************************************************/
 655 static int cmd_du(struct smbclient_context *ctx, const char **args)
 656 {
 657         uint16_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
 658         int rc;
 659         char *mask;
 660
 661         dir_total = 0;
 662
 663         if (args[1]) {
 664                 if (args[1][0] == '\\')
 665                         mask = talloc_strdup(ctx, args[1]);
 666                 else
 667                         mask = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
 668                 dos_format(mask);
 669         } else {
 670                 mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
 671         }
 672
 673         do_list(ctx, mask, attribute, do_du, ctx->recurse, true);
 674
 675         talloc_free(mask);
 676
 677         rc = do_dskattr(ctx);
 678
 679         d_printf("Total number of bytes: %.0f\n", dir_total);
 680
 681         return rc;
 682 }
 683
 684
 685 /****************************************************************************
 686   get a file from rname to lname
 687   ****************************************************************************/
 688 static int do_get(struct smbclient_context *ctx, char *rname, const char *p_lname, bool reget)
 689 {
 690         int handle = 0, fnum;
 691         bool newhandle = false;
 692         uint8_t *data;
 693         struct timeval tp_start;
 694         int read_size = ctx->io_bufsize;
 695         uint16_t attr;
 696         size_t size;
 697         off_t start = 0;
 698         off_t nread = 0;
 699         int rc = 0;
 700         char *lname;
 701
 702
 703         GetTimeOfDay(&tp_start);
 704
 705         if (ctx->lowercase) {
 706                 lname = strlower_talloc(ctx, p_lname);
 707         } else {
 708                 lname = talloc_strdup(ctx, p_lname);
 709         }
 710
 711         fnum = smbcli_open(ctx->cli->tree, rname, O_RDONLY, DENY_NONE);
 712
 713         if (fnum == -1) {
 714                 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
 715                 return 1;
 716         }
 717
 718         if(!strcmp(lname,"-")) {
 719                 handle = fileno(stdout);
 720         } else {
 721                 if (reget) {
 722                         handle = open(lname, O_WRONLY|O_CREAT, 0644);
 723                         if (handle >= 0) {
 724                                 start = lseek(handle, 0, SEEK_END);
 725                                 if (start == -1) {
 726                                         d_printf("Error seeking local file\n");
 727                                         close(handle);
 728                                         return 1;
 729                                 }
 730                         }
 731                 } else {
 732                         handle = open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
 733                 }
 734                 newhandle = true;
 735         }
 736         if (handle < 0) {
 737                 d_printf("Error opening local file %s\n",lname);
 738                 return 1;
 739         }
 740
 741
 742         if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum,
 743                            &attr, &size, NULL, NULL, NULL, NULL, NULL)) &&
 744             NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum,
 745                           &attr, &size, NULL, NULL, NULL))) {
 746                 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
 747                 if (newhandle) {
 748                         close(handle);
 749                 }
 750                 return 1;
 751         }
 752
 753         DEBUG(2,("getting file %s of size %.0f as %s ",
 754                  rname, (double)size, lname));
 755
 756         if(!(data = (uint8_t *)malloc(read_size))) {
 757                 d_printf("malloc fail for size %d\n", read_size);
 758                 smbcli_close(ctx->cli->tree, fnum);
 759                 if (newhandle) {
 760                         close(handle);
 761                 }
 762                 return 1;
 763         }
 764
 765         while (1) {
 766                 int n = smbcli_read(ctx->cli->tree, fnum, data, nread + start, read_size);
 767
 768                 if (n <= 0) break;
 769
 770                 if (writefile(handle,data, n, ctx->translation) != n) {
 771                         d_printf("Error writing local file\n");
 772                         rc = 1;
 773                         break;
 774                 }
 775
 776                 nread += n;
 777         }
 778
 779         if (nread + start < size) {
 780                 DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n",
 781                             rname, (long)nread));
 782
 783                 rc = 1;
 784         }
 785
 786         SAFE_FREE(data);
 787
 788         if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
 789                 d_printf("Error %s closing remote file\n",smbcli_errstr(ctx->cli->tree));
 790                 rc = 1;
 791         }
 792
 793         if (newhandle) {
 794                 close(handle);
 795         }
 796
 797         if (ctx->archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) {
 798                 smbcli_setatr(ctx->cli->tree, rname, attr & ~(uint16_t)FILE_ATTRIBUTE_ARCHIVE, 0);
 799         }
 800
 801         {
 802                 struct timeval tp_end;
 803                 int this_time;
 804
 805                 GetTimeOfDay(&tp_end);
 806                 this_time =
 807                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
 808                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
 809                 get_total_time_ms += this_time;
 810                 get_total_size += nread;
 811
 812                 DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n",
 813                          nread / (1.024*this_time + 1.0e-4),
 814                          get_total_size / (1.024*get_total_time_ms)));
 815         }
 816
 817         return rc;
 818 }
 819
 820
 821 /****************************************************************************
 822   get a file
 823   ****************************************************************************/
 824 static int cmd_get(struct smbclient_context *ctx, const char **args)
 825 {
 826         const char *lname;
 827         char *rname;
 828
 829         if (!args[1]) {
 830                 d_printf("get <filename>\n");
 831                 return 1;
 832         }
 833
 834         rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
 835
 836         if (args[2])
 837                 lname = args[2];
 838         else
 839                 lname = args[1];
 840
 841         dos_clean_name(rname);
 842
 843         return do_get(ctx, rname, lname, false);
 844 }
 845
 846 /****************************************************************************
 847  Put up a yes/no prompt.
 848 ****************************************************************************/
 849 static bool yesno(char *p)
 850 {
 851         char ans[4];
 852         printf("%s",p);
 853
 854         if (!fgets(ans,sizeof(ans)-1,stdin))
 855                 return(false);
 856
 857         if (*ans == 'y' || *ans == 'Y')
 858                 return(true);
 859
 860         return(false);
 861 }
 862
 863 /****************************************************************************
 864   do a mget operation on one file
 865   ****************************************************************************/
 866 static void do_mget(struct smbclient_context *ctx, struct clilist_file_info *finfo)
 867 {
 868         char *rname;
 869         char *quest;
 870         char *mget_mask;
 871         char *saved_curdir;
 872         char *l_fname;
 873         int ret;
 874
 875         if (ISDOT(finfo->name) || ISDOTDOT(finfo->name))
 876                 return;
 877
 878         if (finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)
 879                 quest = talloc_asprintf(ctx, "Get directory %s? ",finfo->name);
 880         else
 881                 quest = talloc_asprintf(ctx, "Get file %s? ",finfo->name);
 882
 883         if (ctx->prompt && !yesno(quest)) return;
 884
 885         talloc_free(quest);
 886
 887         if (!(finfo->attrib & FILE_ATTRIBUTE_DIRECTORY)) {
 888                 rname = talloc_asprintf(ctx, "%s%s",ctx->remote_cur_dir,
 889                                         finfo->name);
 890                 do_get(ctx, rname, finfo->name, false);
 891                 talloc_free(rname);
 892                 return;
 893         }
 894
 895         /* handle directories */
 896         saved_curdir = talloc_strdup(ctx, ctx->remote_cur_dir);
 897
 898         ctx->remote_cur_dir = talloc_asprintf_append_buffer(NULL, "%s\\", finfo->name);
 899
 900         if (ctx->lowercase) {
 901                 l_fname = strlower_talloc(ctx, finfo->name);
 902         } else {
 903                 l_fname = talloc_strdup(ctx, finfo->name);
 904         }
 905
 906         string_replace(l_fname, '\\', '/');
 907
 908         if (!directory_exist(l_fname) &&
 909             mkdir(l_fname, 0777) != 0) {
 910                 d_printf("failed to create directory %s\n", l_fname);
 911                 return;
 912         }
 913
 914         if (chdir(l_fname) != 0) {
 915                 d_printf("failed to chdir to directory %s\n", l_fname);
 916                 return;
 917         }
 918
 919         mget_mask = talloc_asprintf(ctx, "%s*", ctx->remote_cur_dir);
 920
 921         do_list(ctx, mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,false, true);
 922         ret = chdir("..");
 923         if (ret == -1) {
 924                 d_printf("failed to chdir to '..': %s\n", strerror(errno));
 925                 return;
 926         }
 927         talloc_free(ctx->remote_cur_dir);
 928
 929         ctx->remote_cur_dir = saved_curdir;
 930 }
 931
 932
 933 /****************************************************************************
 934 view the file using the pager
 935 ****************************************************************************/
 936 static int cmd_more(struct smbclient_context *ctx, const char **args)
 937 {
 938         char *rname;
 939         char *pager_cmd;
 940         char *lname;
 941         char *pager;
 942         int fd;
 943         int rc = 0;
 944         mode_t mask;
 945
 946         lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir());
 947         mask = umask(S_IRWXO | S_IRWXG);
 948         fd = mkstemp(lname);
 949         umask(mask);
 950         if (fd == -1) {
 951                 d_printf("failed to create temporary file for more\n");
 952                 return 1;
 953         }
 954         close(fd);
 955
 956         if (!args[1]) {
 957                 d_printf("more <filename>\n");
 958                 unlink(lname);
 959                 return 1;
 960         }
 961         rname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
 962         dos_clean_name(rname);
 963
 964         rc = do_get(ctx, rname, lname, false);
 965
 966         pager=getenv("PAGER");
 967
 968         pager_cmd = talloc_asprintf(ctx, "%s %s",(pager? pager:DEFAULT_PAGER), lname);
 969         rc = system(pager_cmd);
 970         if (rc == -1) {
 971                 d_printf("failed to call pager command\n");
 972                 return 1;
 973         }
 974         unlink(lname);
 975
 976         return rc;
 977 }
 978
 979
 980
 981 /****************************************************************************
 982 do a mget command
 983 ****************************************************************************/
 984 static int cmd_mget(struct smbclient_context *ctx, const char **args)
 985 {
 986         uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
 987         char *mget_mask = NULL;
 988         int i;
 989
 990         if (ctx->recurse)
 991                 attribute |= FILE_ATTRIBUTE_DIRECTORY;
 992
 993         for (i = 1; args[i]; i++) {
 994                 mget_mask = talloc_strdup(ctx, ctx->remote_cur_dir);
 995                 if(mget_mask[strlen(mget_mask)-1]!='\\')
 996                         mget_mask = talloc_append_string(ctx, mget_mask, "\\");
 997
 998                 mget_mask = talloc_strdup(ctx, args[i]);
 999                 if (mget_mask[0] != '\\')
1000                         mget_mask = talloc_append_string(ctx, mget_mask, "\\");
1001                 do_list(ctx, mget_mask, attribute,do_mget,false,true);
1002
1003                 talloc_free(mget_mask);
1004         }
1005
1006         if (mget_mask == NULL) {
1007                 mget_mask = talloc_asprintf(ctx, "%s\\*", ctx->remote_cur_dir);
1008                 do_list(ctx, mget_mask, attribute,do_mget,false,true);
1009                 talloc_free(mget_mask);
1010         }
1011
1012         return 0;
1013 }
1014
1015
1016 /****************************************************************************
1017 make a directory of name "name"
1018 ****************************************************************************/
1019 static NTSTATUS do_mkdir(struct smbclient_context *ctx, char *name)
1020 {
1021         NTSTATUS status;
1022
1023         if (NT_STATUS_IS_ERR(status = smbcli_mkdir(ctx->cli->tree, name))) {
1024                 d_printf("%s making remote directory %s\n",
1025                          smbcli_errstr(ctx->cli->tree),name);
1026                 return status;
1027         }
1028
1029         return status;
1030 }
1031
1032
1033 /****************************************************************************
1034  Exit client.
1035 ****************************************************************************/
1036 static int cmd_quit(struct smbclient_context *ctx, const char **args)
1037 {
1038         talloc_free(ctx);
1039         exit(0);
1040         /* NOTREACHED */
1041         return 0;
1042 }
1043
1044
1045 /****************************************************************************
1046   make a directory
1047   ****************************************************************************/
1048 static int cmd_mkdir(struct smbclient_context *ctx, const char **args)
1049 {
1050         char *mask, *p;
1051
1052         if (!args[1]) {
1053                 if (!ctx->recurse)
1054                         d_printf("mkdir <dirname>\n");
1055                 return 1;
1056         }
1057
1058         mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir,args[1]);
1059
1060         if (ctx->recurse) {
1061                 dos_clean_name(mask);
1062
1063                 trim_string(mask,".",NULL);
1064                 for (p = strtok(mask,"/\\"); p; p = strtok(p, "/\\")) {
1065                         char *parent = talloc_strndup(ctx, mask, PTR_DIFF(p, mask));
1066
1067                         if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, parent))) {
1068                                 do_mkdir(ctx, parent);
1069                         }
1070
1071                         talloc_free(parent);
1072                 }
1073         } else {
1074                 do_mkdir(ctx, mask);
1075         }
1076
1077         return 0;
1078 }
1079
1080 /****************************************************************************
1081 show 8.3 name of a file
1082 ****************************************************************************/
1083 static int cmd_altname(struct smbclient_context *ctx, const char **args)
1084 {
1085         const char *p;
1086         char *altname;
1087         char *name;
1088
1089         if (!args[1]) {
1090                 d_printf("altname <file>\n");
1091                 return 1;
1092         }
1093
1094         name = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1095
1096         if (!NT_STATUS_IS_OK(smbcli_qpathinfo_alt_name(ctx->cli->tree, name, &p))) {
1097                 d_printf("%s getting alt name for %s\n",
1098                          smbcli_errstr(ctx->cli->tree),name);
1099                 return(false);
1100         }
1101         altname = discard_const_p(char, p);
1102         d_printf("%s\n", altname);
1103
1104         SAFE_FREE(altname);
1105
1106         return 0;
1107 }
1108
1109
1110 /****************************************************************************
1111   put a single file
1112   ****************************************************************************/
1113 static int do_put(struct smbclient_context *ctx, char *rname, char *lname, bool reput)
1114 {
1115         int fnum;
1116         FILE *f;
1117         size_t start = 0;
1118         off_t nread = 0;
1119         uint8_t *buf = NULL;
1120         int maxwrite = ctx->io_bufsize;
1121         int rc = 0;
1122
1123         struct timeval tp_start;
1124         GetTimeOfDay(&tp_start);
1125
1126         if (reput) {
1127                 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT, DENY_NONE);
1128                 if (fnum >= 0) {
1129                         if (NT_STATUS_IS_ERR(smbcli_qfileinfo(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL)) &&
1130                             NT_STATUS_IS_ERR(smbcli_getattrE(ctx->cli->tree, fnum, NULL, &start, NULL, NULL, NULL))) {
1131                                 d_printf("getattrib: %s\n",smbcli_errstr(ctx->cli->tree));
1132                                 return 1;
1133                         }
1134                 }
1135         } else {
1136                 fnum = smbcli_open(ctx->cli->tree, rname, O_RDWR|O_CREAT|O_TRUNC,
1137                                 DENY_NONE);
1138         }
1139
1140         if (fnum == -1) {
1141                 d_printf("%s opening remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1142                 return 1;
1143         }
1144
1145         /* allow files to be piped into smbclient
1146            jdblair 24.jun.98
1147
1148            Note that in this case this function will exit(0) rather
1149            than returning. */
1150         if (!strcmp(lname, "-")) {
1151                 f = stdin;
1152                 /* size of file is not known */
1153         } else {
1154                 f = fopen(lname, "r");
1155                 if (f && reput) {
1156                         if (fseek(f, start, SEEK_SET) == -1) {
1157                                 d_printf("Error seeking local file\n");
1158                                 fclose(f);
1159                                 return 1;
1160                         }
1161                 }
1162         }
1163
1164         if (!f) {
1165                 d_printf("Error opening local file %s\n",lname);
1166                 return 1;
1167         }
1168
1169
1170         DEBUG(1,("putting file %s as %s ",lname,
1171                  rname));
1172
1173         buf = (uint8_t *)malloc(maxwrite);
1174         if (!buf) {
1175                 d_printf("ERROR: Not enough memory!\n");
1176                 fclose(f);
1177                 return 1;
1178         }
1179         while (!feof(f)) {
1180                 int n = maxwrite;
1181                 int ret;
1182
1183                 if ((n = readfile(buf,n,f,ctx->translation)) < 1) {
1184                         if((n == 0) && feof(f))
1185                                 break; /* Empty local file. */
1186
1187                         d_printf("Error reading local file: %s\n", strerror(errno));
1188                         rc = 1;
1189                         break;
1190                 }
1191
1192                 ret = smbcli_write(ctx->cli->tree, fnum, 0, buf, nread + start, n);
1193
1194                 if (n != ret) {
1195                         d_printf("Error writing file: %s\n", smbcli_errstr(ctx->cli->tree));
1196                         rc = 1;
1197                         break;
1198                 }
1199
1200                 nread += n;
1201         }
1202
1203         if (NT_STATUS_IS_ERR(smbcli_close(ctx->cli->tree, fnum))) {
1204                 d_printf("%s closing remote file %s\n",smbcli_errstr(ctx->cli->tree),rname);
1205                 fclose(f);
1206                 SAFE_FREE(buf);
1207                 return 1;
1208         }
1209
1210
1211         if (f != stdin) {
1212                 fclose(f);
1213         }
1214
1215         SAFE_FREE(buf);
1216
1217         {
1218                 struct timeval tp_end;
1219                 int this_time;
1220
1221                 GetTimeOfDay(&tp_end);
1222                 this_time =
1223                         (tp_end.tv_sec - tp_start.tv_sec)*1000 +
1224                         (tp_end.tv_usec - tp_start.tv_usec)/1000;
1225                 put_total_time_ms += this_time;
1226                 put_total_size += nread;
1227
1228                 DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n",
1229                          nread / (1.024*this_time + 1.0e-4),
1230                          put_total_size / (1.024*put_total_time_ms)));
1231         }
1232
1233         if (f == stdin) {
1234                 talloc_free(ctx);
1235                 exit(0);
1236         }
1237
1238         return rc;
1239 }
1240
1241
1242
1243 /****************************************************************************
1244   put a file
1245   ****************************************************************************/
1246 static int cmd_put(struct smbclient_context *ctx, const char **args)
1247 {
1248         char *lname;
1249         char *rname;
1250
1251         if (!args[1]) {
1252                 d_printf("put <filename> [<remotename>]\n");
1253                 return 1;
1254         }
1255
1256         lname = talloc_strdup(ctx, args[1]);
1257
1258         if (args[2]) {
1259                 if (args[2][0]=='\\')
1260                         rname = talloc_strdup(ctx, args[2]);
1261                 else
1262                         rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[2]);
1263         } else {
1264                 rname = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, lname);
1265         }
1266
1267         dos_clean_name(rname);
1268
1269         /* allow '-' to represent stdin
1270            jdblair, 24.jun.98 */
1271         if (!file_exist(lname) && (strcmp(lname,"-"))) {
1272                 d_printf("%s does not exist\n",lname);
1273                 return 1;
1274         }
1275
1276         return do_put(ctx, rname, lname, false);
1277 }
1278
1279 /*************************************
1280   File list structure
1281 *************************************/
1282
1283 static struct file_list {
1284         struct file_list *prev, *next;
1285         char *file_path;
1286         bool isdir;
1287 } *file_list;
1288
1289 /****************************************************************************
1290   Free a file_list structure
1291 ****************************************************************************/
1292
1293 static void free_file_list (struct file_list * list)
1294 {
1295         struct file_list *tmp;
1296
1297         while (list)
1298         {
1299                 tmp = list;
1300                 DLIST_REMOVE(list, list);
1301                 SAFE_FREE(tmp->file_path);
1302                 SAFE_FREE(tmp);
1303         }
1304 }
1305
1306 /****************************************************************************
1307   seek in a directory/file list until you get something that doesn't start with
1308   the specified name
1309   ****************************************************************************/
1310 static bool seek_list(struct file_list *list, char *name)
1311 {
1312         while (list) {
1313                 trim_string(list->file_path,"./","\n");
1314                 if (strncmp(list->file_path, name, strlen(name)) != 0) {
1315                         return(true);
1316                 }
1317                 list = list->next;
1318         }
1319
1320         return(false);
1321 }
1322
1323 /****************************************************************************
1324   set the file selection mask
1325   ****************************************************************************/
1326 static int cmd_select(struct smbclient_context *ctx, const char **args)
1327 {
1328         talloc_free(ctx->fileselection);
1329         ctx->fileselection = talloc_strdup(ctx, args[1]);
1330
1331         return 0;
1332 }
1333
1334 /*******************************************************************
1335   A readdir wrapper which just returns the file name.
1336  ********************************************************************/
1337 static const char *readdirname(DIR *p)
1338 {
1339         struct dirent *ptr;
1340         char *dname;
1341
1342         if (!p)
1343                 return(NULL);
1344
1345         ptr = (struct dirent *)readdir(p);
1346         if (!ptr)
1347                 return(NULL);
1348
1349         dname = ptr->d_name;
1350
1351 #ifdef NEXT2
1352         if (telldir(p) < 0)
1353                 return(NULL);
1354 #endif
1355
1356 #ifdef HAVE_BROKEN_READDIR
1357         /* using /usr/ucb/cc is BAD */
1358         dname = dname - 2;
1359 #endif
1360
1361         {
1362                 static char *buf;
1363                 int len = NAMLEN(ptr);
1364                 buf = talloc_strndup(NULL, dname, len);
1365                 dname = buf;
1366         }
1367
1368         return(dname);
1369 }
1370
1371 /****************************************************************************
1372   Recursive file matching function act as find
1373   match must be always set to true when calling this function
1374 ****************************************************************************/
1375 static int file_find(struct smbclient_context *ctx, struct file_list **list, const char *directory,
1376                       const char *expression, bool match)
1377 {
1378         DIR *dir;
1379         struct file_list *entry;
1380         struct stat statbuf;
1381         int ret;
1382         char *path;
1383         bool isdir;
1384         const char *dname;
1385
1386         dir = opendir(directory);
1387         if (!dir) return -1;
1388
1389         while ((dname = readdirname(dir))) {
1390                 if (ISDOT(dname) || ISDOTDOT(dname)) {
1391                         continue;
1392                 }
1393
1394                 if (asprintf(&path, "%s/%s", directory, dname) <= 0) {
1395                         continue;
1396                 }
1397
1398                 isdir = false;
1399                 if (!match || !gen_fnmatch(expression, dname)) {
1400                         if (ctx->recurse) {
1401                                 ret = stat(path, &statbuf);
1402                                 if (ret == 0) {
1403                                         if (S_ISDIR(statbuf.st_mode)) {
1404                                                 isdir = true;
1405                                                 ret = file_find(ctx, list, path, expression, false);
1406                                         }
1407                                 } else {
1408                                         d_printf("file_find: cannot stat file %s\n", path);
1409                                 }
1410
1411                                 if (ret == -1) {
1412                                         SAFE_FREE(path);
1413                                         closedir(dir);
1414                                         return -1;
1415                                 }
1416                         }
1417                         entry = malloc_p(struct file_list);
1418                         if (!entry) {
1419                                 d_printf("Out of memory in file_find\n");
1420                                 closedir(dir);
1421                                 return -1;
1422                         }
1423                         entry->file_path = path;
1424                         entry->isdir = isdir;
1425                         DLIST_ADD(*list, entry);
1426                 } else {
1427                         SAFE_FREE(path);
1428                 }
1429         }
1430
1431         closedir(dir);
1432         return 0;
1433 }
1434
1435 /****************************************************************************
1436   mput some files
1437   ****************************************************************************/
1438 static int cmd_mput(struct smbclient_context *ctx, const char **args)
1439 {
1440         int i;
1441
1442         for (i = 1; args[i]; i++) {
1443                 int ret;
1444                 struct file_list *temp_list;
1445                 char *quest, *lname, *rname;
1446
1447                 printf("%s\n", args[i]);
1448
1449                 file_list = NULL;
1450
1451                 ret = file_find(ctx, &file_list, ".", args[i], true);
1452                 if (ret) {
1453                         free_file_list(file_list);
1454                         continue;
1455                 }
1456
1457                 quest = NULL;
1458                 lname = NULL;
1459                 rname = NULL;
1460
1461                 for (temp_list = file_list; temp_list;
1462                      temp_list = temp_list->next) {
1463
1464                         SAFE_FREE(lname);
1465                         if (asprintf(&lname, "%s/", temp_list->file_path) <= 0)
1466                                 continue;
1467                         trim_string(lname, "./", "/");
1468
1469                         /* check if it's a directory */
1470                         if (temp_list->isdir) {
1471                                 /* if (!recurse) continue; */
1472
1473                                 SAFE_FREE(quest);
1474                                 if (asprintf(&quest, "Put directory %s? ", lname) < 0) break;
1475                                 if (ctx->prompt && !yesno(quest)) { /* No */
1476                                         /* Skip the directory */
1477                                         lname[strlen(lname)-1] = '/';
1478                                         if (!seek_list(temp_list, lname))
1479                                                 break;
1480                                 } else { /* Yes */
1481                                         SAFE_FREE(rname);
1482                                         if(asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1483                                         dos_format(rname);
1484                                         if (NT_STATUS_IS_ERR(smbcli_chkpath(ctx->cli->tree, rname)) &&
1485                                             NT_STATUS_IS_ERR(do_mkdir(ctx, rname))) {
1486                                                 DEBUG (0, ("Unable to make dir, skipping..."));
1487                                                 /* Skip the directory */
1488                                                 lname[strlen(lname)-1] = '/';
1489                                                 if (!seek_list(temp_list, lname))
1490                                                         break;
1491                                         }
1492                                 }
1493                                 continue;
1494                         } else {
1495                                 SAFE_FREE(quest);
1496                                 if (asprintf(&quest,"Put file %s? ", lname) < 0) break;
1497                                 if (ctx->prompt && !yesno(quest)) /* No */
1498                                         continue;
1499
1500                                 /* Yes */
1501                                 SAFE_FREE(rname);
1502                                 if (asprintf(&rname, "%s%s", ctx->remote_cur_dir, lname) < 0) break;
1503                         }
1504
1505                         dos_format(rname);
1506
1507                         do_put(ctx, rname, lname, false);
1508                 }
1509                 free_file_list(file_list);
1510                 SAFE_FREE(quest);
1511                 SAFE_FREE(lname);
1512                 SAFE_FREE(rname);
1513         }
1514
1515         return 0;
1516 }
1517
1518
1519 /****************************************************************************
1520   print a file
1521   ****************************************************************************/
1522 static int cmd_print(struct smbclient_context *ctx, const char **args)
1523 {
1524         char *lname, *rname;
1525         char *p;
1526
1527         if (!args[1]) {
1528                 d_printf("print <filename>\n");
1529                 return 1;
1530         }
1531
1532         lname = talloc_strdup(ctx, args[1]);
1533         if (lname == NULL) {
1534                 d_printf("Out of memory in cmd_print\n");
1535                 return 1;
1536         }
1537
1538         if (strequal(lname, "-")) {
1539                 rname = talloc_asprintf(ctx, "stdin-%d", (int)getpid());
1540         } else {
1541                 p = strrchr_m(lname, '/');
1542                 if (p) {
1543                         rname = talloc_asprintf(ctx, "%s-%d", p + 1,
1544                                                 (int)getpid());
1545                 } else {
1546                         rname = talloc_strdup(ctx, lname);
1547                 }
1548         }
1549
1550         if (rname == NULL) {
1551                 d_printf("Out of memory in cmd_print (stdin)\n");
1552                 return 1;
1553         }
1554
1555         return do_put(ctx, rname, lname, false);
1556 }
1557
1558
1559 static int cmd_rewrite(struct smbclient_context *ctx, const char **args)
1560 {
1561         d_printf("REWRITE: command not implemented (FIXME!)\n");
1562
1563         return 0;
1564 }
1565
1566 /****************************************************************************
1567 delete some files
1568 ****************************************************************************/
1569 static int cmd_del(struct smbclient_context *ctx, const char **args)
1570 {
1571         char *mask;
1572         uint16_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN;
1573
1574         if (ctx->recurse)
1575                 attribute |= FILE_ATTRIBUTE_DIRECTORY;
1576
1577         if (!args[1]) {
1578                 d_printf("del <filename>\n");
1579                 return 1;
1580         }
1581         mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1582
1583         if (NT_STATUS_IS_ERR(smbcli_unlink(ctx->cli->tree, mask))) {
1584                 d_printf("%s deleting remote file %s\n",smbcli_errstr(ctx->cli->tree),mask);
1585         }
1586
1587         return 0;
1588 }
1589
1590
1591 /****************************************************************************
1592 delete a whole directory tree
1593 ****************************************************************************/
1594 static int cmd_deltree(struct smbclient_context *ctx, const char **args)
1595 {
1596         char *dname;
1597         int ret;
1598
1599         if (!args[1]) {
1600                 d_printf("deltree <dirname>\n");
1601                 return 1;
1602         }
1603
1604         dname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1605
1606         ret = smbcli_deltree(ctx->cli->tree, dname);
1607
1608         if (ret == -1) {
1609                 printf("Failed to delete tree %s - %s\n", dname, smbcli_errstr(ctx->cli->tree));
1610                 return -1;
1611         }
1612
1613         printf("Deleted %d files in %s\n", ret, dname);
1614
1615         return 0;
1616 }
1617
1618 typedef struct {
1619         const char  *level_name;
1620         enum smb_fsinfo_level level;
1621 } fsinfo_level_t;
1622
1623 fsinfo_level_t fsinfo_levels[] = {
1624         {"dskattr", RAW_QFS_DSKATTR},
1625         {"allocation", RAW_QFS_ALLOCATION},
1626         {"volume", RAW_QFS_VOLUME},
1627         {"volumeinfo", RAW_QFS_VOLUME_INFO},
1628         {"sizeinfo", RAW_QFS_SIZE_INFO},
1629         {"deviceinfo", RAW_QFS_DEVICE_INFO},
1630         {"attributeinfo", RAW_QFS_ATTRIBUTE_INFO},
1631         {"unixinfo", RAW_QFS_UNIX_INFO},
1632         {"volume-information", RAW_QFS_VOLUME_INFORMATION},
1633         {"size-information", RAW_QFS_SIZE_INFORMATION},
1634         {"device-information", RAW_QFS_DEVICE_INFORMATION},
1635         {"attribute-information", RAW_QFS_ATTRIBUTE_INFORMATION},
1636         {"quota-information", RAW_QFS_QUOTA_INFORMATION},
1637         {"fullsize-information", RAW_QFS_FULL_SIZE_INFORMATION},
1638         {"objectid", RAW_QFS_OBJECTID_INFORMATION},
1639         {"sector-size-info", RAW_QFS_SECTOR_SIZE_INFORMATION},
1640         {NULL, RAW_QFS_GENERIC}
1641 };
1642
1643
1644 static int cmd_fsinfo(struct smbclient_context *ctx, const char **args)
1645 {
1646         union smb_fsinfo fsinfo;
1647         NTSTATUS status;
1648         fsinfo_level_t *fsinfo_level;
1649
1650         if (!args[1]) {
1651                 d_printf("fsinfo <level>, where level is one of following:\n");
1652                 fsinfo_level = fsinfo_levels;
1653                 while(fsinfo_level->level_name) {
1654                         d_printf("%s\n", fsinfo_level->level_name);
1655                         fsinfo_level++;
1656                 }
1657                 return 1;
1658         }
1659
1660         fsinfo_level = fsinfo_levels;
1661         while(fsinfo_level->level_name && !strequal(args[1],fsinfo_level->level_name)) {
1662                 fsinfo_level++;
1663         }
1664
1665         if (!fsinfo_level->level_name) {
1666                 d_printf("wrong level name!\n");
1667                 return 1;
1668         }
1669
1670         fsinfo.generic.level = fsinfo_level->level;
1671         status = smb_raw_fsinfo(ctx->cli->tree, ctx, &fsinfo);
1672         if (!NT_STATUS_IS_OK(status)) {
1673                 d_printf("fsinfo-level-%s - %s\n", fsinfo_level->level_name, nt_errstr(status));
1674                 return 1;
1675         }
1676
1677         d_printf("fsinfo-level-%s:\n", fsinfo_level->level_name);
1678         switch(fsinfo.generic.level) {
1679         case RAW_QFS_DSKATTR:
1680                 d_printf("\tunits_total:                %hu\n",
1681                          (unsigned short) fsinfo.dskattr.out.units_total);
1682                 d_printf("\tblocks_per_unit:            %hu\n",
1683                          (unsigned short) fsinfo.dskattr.out.blocks_per_unit);
1684                 d_printf("\tblocks_size:                %hu\n",
1685                          (unsigned short) fsinfo.dskattr.out.block_size);
1686                 d_printf("\tunits_free:                 %hu\n",
1687                          (unsigned short) fsinfo.dskattr.out.units_free);
1688                 break;
1689         case RAW_QFS_ALLOCATION:
1690                 d_printf("\tfs_id:                      %lu\n",
1691                          (unsigned long) fsinfo.allocation.out.fs_id);
1692                 d_printf("\tsectors_per_unit:           %lu\n",
1693                          (unsigned long) fsinfo.allocation.out.sectors_per_unit);
1694                 d_printf("\ttotal_alloc_units:          %lu\n",
1695                          (unsigned long) fsinfo.allocation.out.total_alloc_units);
1696                 d_printf("\tavail_alloc_units:          %lu\n",
1697                          (unsigned long) fsinfo.allocation.out.avail_alloc_units);
1698                 d_printf("\tbytes_per_sector:           %hu\n",
1699                          (unsigned short) fsinfo.allocation.out.bytes_per_sector);
1700                 break;
1701         case RAW_QFS_VOLUME:
1702                 d_printf("\tserial_number:              %lu\n",
1703                          (unsigned long) fsinfo.volume.out.serial_number);
1704                 d_printf("\tvolume_name:                %s\n", fsinfo.volume.out.volume_name.s);
1705                 break;
1706         case RAW_QFS_VOLUME_INFO:
1707         case RAW_QFS_VOLUME_INFORMATION:
1708                 d_printf("\tcreate_time:                %s\n",
1709                          nt_time_string(ctx,fsinfo.volume_info.out.create_time));
1710                 d_printf("\tserial_number:              %lu\n",
1711                          (unsigned long) fsinfo.volume_info.out.serial_number);
1712                 d_printf("\tvolume_name:                %s\n", fsinfo.volume_info.out.volume_name.s);
1713                 break;
1714         case RAW_QFS_SIZE_INFO:
1715         case RAW_QFS_SIZE_INFORMATION:
1716                 d_printf("\ttotal_alloc_units:          %llu\n",
1717                          (unsigned long long) fsinfo.size_info.out.total_alloc_units);
1718                 d_printf("\tavail_alloc_units:          %llu\n",
1719                          (unsigned long long) fsinfo.size_info.out.avail_alloc_units);
1720                 d_printf("\tsectors_per_unit:           %lu\n",
1721                          (unsigned long) fsinfo.size_info.out.sectors_per_unit);
1722                 d_printf("\tbytes_per_sector:           %lu\n",
1723                          (unsigned long) fsinfo.size_info.out.bytes_per_sector);
1724                 break;
1725         case RAW_QFS_DEVICE_INFO:
1726         case RAW_QFS_DEVICE_INFORMATION:
1727                 d_printf("\tdevice_type:                %lu\n",
1728                          (unsigned long) fsinfo.device_info.out.device_type);
1729                 d_printf("\tcharacteristics:            0x%lx\n",
1730                          (unsigned long) fsinfo.device_info.out.characteristics);
1731                 break;
1732         case RAW_QFS_ATTRIBUTE_INFORMATION:
1733         case RAW_QFS_ATTRIBUTE_INFO:
1734                 d_printf("\tfs_attr:                    0x%lx\n",
1735                          (unsigned long) fsinfo.attribute_info.out.fs_attr);
1736                 d_printf("\tmax_file_component_length:  %lu\n",
1737                          (unsigned long) fsinfo.attribute_info.out.max_file_component_length);
1738                 d_printf("\tfs_type:                    %s\n", fsinfo.attribute_info.out.fs_type.s);
1739                 break;
1740         case RAW_QFS_UNIX_INFO:
1741                 d_printf("\tmajor_version:              %hu\n",
1742                          (unsigned short) fsinfo.unix_info.out.major_version);
1743                 d_printf("\tminor_version:              %hu\n",
1744                          (unsigned short) fsinfo.unix_info.out.minor_version);
1745                 d_printf("\tcapability:                 0x%llx\n",
1746                          (unsigned long long) fsinfo.unix_info.out.capability);
1747                 break;
1748         case RAW_QFS_QUOTA_INFORMATION:
1749                 d_printf("\tunknown[3]:                 [%llu,%llu,%llu]\n",
1750                          (unsigned long long) fsinfo.quota_information.out.unknown[0],
1751                          (unsigned long long) fsinfo.quota_information.out.unknown[1],
1752                          (unsigned long long) fsinfo.quota_information.out.unknown[2]);
1753                 d_printf("\tquota_soft:                 %llu\n",
1754                          (unsigned long long) fsinfo.quota_information.out.quota_soft);
1755                 d_printf("\tquota_hard:                 %llu\n",
1756                          (unsigned long long) fsinfo.quota_information.out.quota_hard);
1757                 d_printf("\tquota_flags:                0x%llx\n",
1758                          (unsigned long long) fsinfo.quota_information.out.quota_flags);
1759                 break;
1760         case RAW_QFS_FULL_SIZE_INFORMATION:
1761                 d_printf("\ttotal_alloc_units:          %llu\n",
1762                          (unsigned long long) fsinfo.full_size_information.out.total_alloc_units);
1763                 d_printf("\tcall_avail_alloc_units:     %llu\n",
1764                          (unsigned long long) fsinfo.full_size_information.out.call_avail_alloc_units);
1765                 d_printf("\tactual_avail_alloc_units:   %llu\n",
1766                          (unsigned long long) fsinfo.full_size_information.out.actual_avail_alloc_units);
1767                 d_printf("\tsectors_per_unit:           %lu\n",
1768                          (unsigned long) fsinfo.full_size_information.out.sectors_per_unit);
1769                 d_printf("\tbytes_per_sector:           %lu\n",
1770                          (unsigned long) fsinfo.full_size_information.out.bytes_per_sector);
1771                 break;
1772         case RAW_QFS_OBJECTID_INFORMATION:
1773                 d_printf("\tGUID:                       %s\n",
1774                          GUID_string(ctx,&fsinfo.objectid_information.out.guid));
1775                 d_printf("\tunknown[6]:                 [%llu,%llu,%llu,%llu,%llu,%llu]\n",
1776                          (unsigned long long) fsinfo.objectid_information.out.unknown[0],
1777                          (unsigned long long) fsinfo.objectid_information.out.unknown[1],
1778                          (unsigned long long) fsinfo.objectid_information.out.unknown[2],
1779                          (unsigned long long) fsinfo.objectid_information.out.unknown[3],
1780                          (unsigned long long) fsinfo.objectid_information.out.unknown[4],
1781                          (unsigned long long) fsinfo.objectid_information.out.unknown[5] );
1782                 break;
1783         case RAW_QFS_SECTOR_SIZE_INFORMATION:
1784                 d_printf("\tlogical_bytes_per_sector:                   %u\n",
1785                          (unsigned)fsinfo.sector_size_info.out.logical_bytes_per_sector);
1786                 d_printf("\tphys_bytes_per_sector_atomic:               %u\n",
1787                          (unsigned)fsinfo.sector_size_info.out.phys_bytes_per_sector_atomic);
1788                 d_printf("\tphys_bytes_per_sector_perf:                 %u\n",
1789                          (unsigned)fsinfo.sector_size_info.out.phys_bytes_per_sector_perf);
1790                 d_printf("\tfs_effective_phys_bytes_per_sector_atomic:  %u\n",
1791                          (unsigned)fsinfo.sector_size_info.out.fs_effective_phys_bytes_per_sector_atomic);
1792                 d_printf("\tflags:                                      0x%x\n",
1793                          (unsigned)fsinfo.sector_size_info.out.flags);
1794                 d_printf("\tbyte_off_sector_align:                      %u\n",
1795                          (unsigned)fsinfo.sector_size_info.out.byte_off_sector_align);
1796                 d_printf("\tbyte_off_partition_align:                   %u\n",
1797                          (unsigned)fsinfo.sector_size_info.out.byte_off_partition_align);
1798                 break;
1799         case RAW_QFS_GENERIC:
1800                 d_printf("\twrong level returned\n");
1801                 break;
1802         }
1803
1804         return 0;
1805 }
1806
1807 /****************************************************************************
1808 show as much information as possible about a file
1809 ****************************************************************************/
1810 static int cmd_allinfo(struct smbclient_context *ctx, const char **args)
1811 {
1812         char *fname;
1813         union smb_fileinfo finfo;
1814         NTSTATUS status;
1815         int fnum;
1816
1817         if (!args[1]) {
1818                 d_printf("allinfo <filename>\n");
1819                 return 1;
1820         }
1821         fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1822
1823         /* first a ALL_INFO QPATHINFO */
1824         finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1825         finfo.generic.in.file.path = fname;
1826         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1827         if (!NT_STATUS_IS_OK(status)) {
1828                 d_printf("%s - %s\n", fname, nt_errstr(status));
1829                 return 1;
1830         }
1831
1832         d_printf("\tcreate_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1833         d_printf("\taccess_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.access_time));
1834         d_printf("\twrite_time:     %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1835         d_printf("\tchange_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1836         d_printf("\tattrib:         0x%x\n", finfo.all_info.out.attrib);
1837         d_printf("\talloc_size:     %lu\n", (unsigned long)finfo.all_info.out.alloc_size);
1838         d_printf("\tsize:           %lu\n", (unsigned long)finfo.all_info.out.size);
1839         d_printf("\tnlink:          %u\n", finfo.all_info.out.nlink);
1840         d_printf("\tdelete_pending: %u\n", finfo.all_info.out.delete_pending);
1841         d_printf("\tdirectory:      %u\n", finfo.all_info.out.directory);
1842         d_printf("\tea_size:        %u\n", finfo.all_info.out.ea_size);
1843         d_printf("\tfname:          '%s'\n", finfo.all_info.out.fname.s);
1844
1845         /* 8.3 name if any */
1846         finfo.generic.level = RAW_FILEINFO_ALT_NAME_INFO;
1847         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1848         if (NT_STATUS_IS_OK(status)) {
1849                 d_printf("\talt_name:       %s\n", finfo.alt_name_info.out.fname.s);
1850         }
1851
1852         /* file_id if available */
1853         finfo.generic.level = RAW_FILEINFO_INTERNAL_INFORMATION;
1854         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1855         if (NT_STATUS_IS_OK(status)) {
1856                 d_printf("\tfile_id         %.0f\n",
1857                          (double)finfo.internal_information.out.file_id);
1858         }
1859
1860         /* the EAs, if any */
1861         finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1862         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1863         if (NT_STATUS_IS_OK(status)) {
1864                 int i;
1865                 for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1866                         d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1867                                  finfo.all_eas.out.eas[i].flags,
1868                                  (int)finfo.all_eas.out.eas[i].value.length,
1869                                  finfo.all_eas.out.eas[i].name.s);
1870                 }
1871         }
1872
1873         /* streams, if available */
1874         finfo.generic.level = RAW_FILEINFO_STREAM_INFO;
1875         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1876         if (NT_STATUS_IS_OK(status)) {
1877                 int i;
1878                 for (i=0;i<finfo.stream_info.out.num_streams;i++) {
1879                         d_printf("\tstream %d:\n", i);
1880                         d_printf("\t\tsize       %ld\n",
1881                                  (long)finfo.stream_info.out.streams[i].size);
1882                         d_printf("\t\talloc size %ld\n",
1883                                  (long)finfo.stream_info.out.streams[i].alloc_size);
1884                         d_printf("\t\tname       %s\n", finfo.stream_info.out.streams[i].stream_name.s);
1885                 }
1886         }
1887
1888         /* dev/inode if available */
1889         finfo.generic.level = RAW_FILEINFO_COMPRESSION_INFORMATION;
1890         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1891         if (NT_STATUS_IS_OK(status)) {
1892                 d_printf("\tcompressed size %ld\n", (long)finfo.compression_info.out.compressed_size);
1893                 d_printf("\tformat          %ld\n", (long)finfo.compression_info.out.format);
1894                 d_printf("\tunit_shift      %ld\n", (long)finfo.compression_info.out.unit_shift);
1895                 d_printf("\tchunk_shift     %ld\n", (long)finfo.compression_info.out.chunk_shift);
1896                 d_printf("\tcluster_shift   %ld\n", (long)finfo.compression_info.out.cluster_shift);
1897         }
1898
1899         /* shadow copies if available */
1900         fnum = smbcli_open(ctx->cli->tree, fname, O_RDONLY, DENY_NONE);
1901         if (fnum != -1) {
1902                 struct smb_shadow_copy info;
1903                 int i;
1904                 info.in.file.fnum = fnum;
1905                 info.in.max_data = ~0;
1906                 status = smb_raw_shadow_data(ctx->cli->tree, ctx, &info);
1907                 if (NT_STATUS_IS_OK(status)) {
1908                         d_printf("\tshadow_copy: %u volumes  %u names\n",
1909                                  info.out.num_volumes, info.out.num_names);
1910                         for (i=0;i<info.out.num_names;i++) {
1911                                 d_printf("\t%s\n", info.out.names[i]);
1912                                 finfo.generic.level = RAW_FILEINFO_ALL_INFO;
1913                                 finfo.generic.in.file.path = talloc_asprintf(ctx, "%s%s",
1914                                                                              info.out.names[i], fname);
1915                                 status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1916                                 if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_PATH_NOT_FOUND) ||
1917                                     NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
1918                                         continue;
1919                                 }
1920                                 if (!NT_STATUS_IS_OK(status)) {
1921                                         d_printf("%s - %s\n", finfo.generic.in.file.path,
1922                                                  nt_errstr(status));
1923                                         return 1;
1924                                 }
1925
1926                                 d_printf("\t\tcreate_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.create_time));
1927                                 d_printf("\t\twrite_time:     %s\n", nt_time_string(ctx, finfo.all_info.out.write_time));
1928                                 d_printf("\t\tchange_time:    %s\n", nt_time_string(ctx, finfo.all_info.out.change_time));
1929                                 d_printf("\t\tsize:           %lu\n", (unsigned long)finfo.all_info.out.size);
1930                         }
1931                 }
1932         }
1933
1934         return 0;
1935 }
1936
1937
1938 /****************************************************************************
1939 shows EA contents
1940 ****************************************************************************/
1941 static int cmd_eainfo(struct smbclient_context *ctx, const char **args)
1942 {
1943         char *fname;
1944         union smb_fileinfo finfo;
1945         NTSTATUS status;
1946         int i;
1947
1948         if (!args[1]) {
1949                 d_printf("eainfo <filename>\n");
1950                 return 1;
1951         }
1952         fname = talloc_strdup(ctx, args[1]);
1953
1954         finfo.generic.level = RAW_FILEINFO_ALL_EAS;
1955         finfo.generic.in.file.path = fname;
1956         status = smb_raw_pathinfo(ctx->cli->tree, ctx, &finfo);
1957
1958         if (!NT_STATUS_IS_OK(status)) {
1959                 d_printf("RAW_FILEINFO_ALL_EAS - %s\n", nt_errstr(status));
1960                 return 1;
1961         }
1962
1963         d_printf("%s has %d EAs\n", fname, finfo.all_eas.out.num_eas);
1964
1965         for (i=0;i<finfo.all_eas.out.num_eas;i++) {
1966                 d_printf("\tEA[%d] flags=%d len=%d '%s'\n", i,
1967                          finfo.all_eas.out.eas[i].flags,
1968                          (int)finfo.all_eas.out.eas[i].value.length,
1969                          finfo.all_eas.out.eas[i].name.s);
1970                 fflush(stdout);
1971                 dump_data(0,
1972                           finfo.all_eas.out.eas[i].value.data,
1973                           finfo.all_eas.out.eas[i].value.length);
1974         }
1975
1976         return 0;
1977 }
1978
1979
1980 /****************************************************************************
1981 show any ACL on a file
1982 ****************************************************************************/
1983 static int cmd_acl(struct smbclient_context *ctx, const char **args)
1984 {
1985         char *fname;
1986         union smb_fileinfo query;
1987         NTSTATUS status;
1988         int fnum;
1989
1990         if (!args[1]) {
1991                 d_printf("acl <filename>\n");
1992                 return 1;
1993         }
1994         fname = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
1995
1996         fnum = smbcli_nt_create_full(ctx->cli->tree, fname, 0,
1997                                      SEC_STD_READ_CONTROL,
1998                                      0,
1999                                      NTCREATEX_SHARE_ACCESS_DELETE|
2000                                      NTCREATEX_SHARE_ACCESS_READ|
2001                                      NTCREATEX_SHARE_ACCESS_WRITE,
2002                                      NTCREATEX_DISP_OPEN,
2003                                      0, 0);
2004         if (fnum == -1) {
2005                 d_printf("%s - %s\n", fname, smbcli_errstr(ctx->cli->tree));
2006                 return -1;
2007         }
2008
2009         query.query_secdesc.level = RAW_FILEINFO_SEC_DESC;
2010         query.query_secdesc.in.file.fnum = fnum;
2011         query.query_secdesc.in.secinfo_flags = 0x7;
2012
2013         status = smb_raw_fileinfo(ctx->cli->tree, ctx, &query);
2014         if (!NT_STATUS_IS_OK(status)) {
2015                 d_printf("%s - %s\n", fname, nt_errstr(status));
2016                 return 1;
2017         }
2018
2019         NDR_PRINT_DEBUG(security_descriptor, query.query_secdesc.out.sd);
2020
2021         return 0;
2022 }
2023
2024 /****************************************************************************
2025 lookup a name or sid
2026 ****************************************************************************/
2027 static int cmd_lookup(struct smbclient_context *ctx, const char **args)
2028 {
2029         NTSTATUS status;
2030         struct dom_sid *sid;
2031
2032         if (!args[1]) {
2033                 d_printf("lookup <sid|name>\n");
2034                 return 1;
2035         }
2036
2037         sid = dom_sid_parse_talloc(ctx, args[1]);
2038         if (sid == NULL) {
2039                 const char *sidstr;
2040                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sidstr);
2041                 if (!NT_STATUS_IS_OK(status)) {
2042                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2043                         return 1;
2044                 }
2045
2046                 d_printf("%s\n", sidstr);
2047         } else {
2048                 const char *name;
2049                 status = smblsa_lookup_sid(ctx->cli, args[1], ctx, &name);
2050                 if (!NT_STATUS_IS_OK(status)) {
2051                         d_printf("lsa_LookupSids - %s\n", nt_errstr(status));
2052                         return 1;
2053                 }
2054
2055                 d_printf("%s\n", name);
2056         }
2057
2058         return 0;
2059 }
2060
2061 /****************************************************************************
2062 show privileges for a user
2063 ****************************************************************************/
2064 static int cmd_privileges(struct smbclient_context *ctx, const char **args)
2065 {
2066         NTSTATUS status;
2067         struct dom_sid *sid;
2068         struct lsa_RightSet rights;
2069         unsigned i;
2070
2071         if (!args[1]) {
2072                 d_printf("privileges <sid|name>\n");
2073                 return 1;
2074         }
2075
2076         sid = dom_sid_parse_talloc(ctx, args[1]);
2077         if (sid == NULL) {
2078                 const char *sid_str;
2079                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2080                 if (!NT_STATUS_IS_OK(status)) {
2081                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2082                         return 1;
2083                 }
2084                 sid = dom_sid_parse_talloc(ctx, sid_str);
2085         }
2086
2087         status = smblsa_sid_privileges(ctx->cli, sid, ctx, &rights);
2088         if (!NT_STATUS_IS_OK(status)) {
2089                 d_printf("lsa_EnumAccountRights - %s\n", nt_errstr(status));
2090                 return 1;
2091         }
2092
2093         for (i=0;i<rights.count;i++) {
2094                 d_printf("\t%s\n", rights.names[i].string);
2095         }
2096
2097         return 0;
2098 }
2099
2100
2101 /****************************************************************************
2102 add privileges for a user
2103 ****************************************************************************/
2104 static int cmd_addprivileges(struct smbclient_context *ctx, const char **args)
2105 {
2106         NTSTATUS status;
2107         struct dom_sid *sid;
2108         struct lsa_RightSet rights;
2109         int i;
2110
2111         if (!args[1]) {
2112                 d_printf("addprivileges <sid|name> <privilege...>\n");
2113                 return 1;
2114         }
2115
2116         sid = dom_sid_parse_talloc(ctx, args[1]);
2117         if (sid == NULL) {
2118                 const char *sid_str;
2119                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2120                 if (!NT_STATUS_IS_OK(status)) {
2121                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2122                         return 1;
2123                 }
2124                 sid = dom_sid_parse_talloc(ctx, sid_str);
2125         }
2126
2127         ZERO_STRUCT(rights);
2128         for (i = 2; args[i]; i++) {
2129                 rights.names = talloc_realloc(ctx, rights.names,
2130                                               struct lsa_StringLarge, rights.count+1);
2131                 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2132                 rights.count++;
2133         }
2134
2135
2136         status = smblsa_sid_add_privileges(ctx->cli, sid, ctx, &rights);
2137         if (!NT_STATUS_IS_OK(status)) {
2138                 d_printf("lsa_AddAccountRights - %s\n", nt_errstr(status));
2139                 return 1;
2140         }
2141
2142         return 0;
2143 }
2144
2145 /****************************************************************************
2146 delete privileges for a user
2147 ****************************************************************************/
2148 static int cmd_delprivileges(struct smbclient_context *ctx, const char **args)
2149 {
2150         NTSTATUS status;
2151         struct dom_sid *sid;
2152         struct lsa_RightSet rights;
2153         int i;
2154
2155         if (!args[1]) {
2156                 d_printf("delprivileges <sid|name> <privilege...>\n");
2157                 return 1;
2158         }
2159
2160         sid = dom_sid_parse_talloc(ctx, args[1]);
2161         if (sid == NULL) {
2162                 const char *sid_str;
2163                 status = smblsa_lookup_name(ctx->cli, args[1], ctx, &sid_str);
2164                 if (!NT_STATUS_IS_OK(status)) {
2165                         d_printf("lsa_LookupNames - %s\n", nt_errstr(status));
2166                         return 1;
2167                 }
2168                 sid = dom_sid_parse_talloc(ctx, sid_str);
2169         }
2170
2171         ZERO_STRUCT(rights);
2172         for (i = 2; args[i]; i++) {
2173                 rights.names = talloc_realloc(ctx, rights.names,
2174                                               struct lsa_StringLarge, rights.count+1);
2175                 rights.names[rights.count].string = talloc_strdup(ctx, args[i]);
2176                 rights.count++;
2177         }
2178
2179
2180         status = smblsa_sid_del_privileges(ctx->cli, sid, ctx, &rights);
2181         if (!NT_STATUS_IS_OK(status)) {
2182                 d_printf("lsa_RemoveAccountRights - %s\n", nt_errstr(status));
2183                 return 1;
2184         }
2185
2186         return 0;
2187 }
2188
2189
2190 /****************************************************************************
2191 open a file
2192 ****************************************************************************/
2193 static int cmd_open(struct smbclient_context *ctx, const char **args)
2194 {
2195         char *filename;
2196         union smb_open io;
2197         NTSTATUS status;
2198         TALLOC_CTX *tmp_ctx;
2199
2200         if (!args[1]) {
2201                 d_printf("open <filename>\n");
2202                 return 1;
2203         }
2204         tmp_ctx = talloc_new(ctx);
2205
2206         filename = talloc_asprintf(tmp_ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2207
2208         io.generic.level = RAW_OPEN_NTCREATEX;
2209         io.ntcreatex.in.root_fid.fnum = 0;
2210         io.ntcreatex.in.flags = 0;
2211         io.ntcreatex.in.access_mask = SEC_RIGHTS_FILE_ALL;
2212         io.ntcreatex.in.create_options = 0;
2213         io.ntcreatex.in.file_attr = FILE_ATTRIBUTE_NORMAL;
2214         io.ntcreatex.in.share_access = NTCREATEX_SHARE_ACCESS_READ;
2215         io.ntcreatex.in.alloc_size = 0;
2216         io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN_IF;
2217         io.ntcreatex.in.impersonation = NTCREATEX_IMPERSONATION_ANONYMOUS;
2218         io.ntcreatex.in.security_flags = 0;
2219         io.ntcreatex.in.fname = filename;
2220
2221         status = smb_raw_open(ctx->cli->tree, tmp_ctx, &io);
2222         talloc_free(tmp_ctx);
2223
2224         if (NT_STATUS_IS_OK(status)) {
2225                 d_printf("Opened file with fnum %u\n", (unsigned)io.ntcreatex.out.file.fnum);
2226         } else {
2227                 d_printf("Opened failed: %s\n", nt_errstr(status));
2228         }
2229
2230         return 0;
2231 }
2232
2233 /****************************************************************************
2234 close a file
2235 ****************************************************************************/
2236 static int cmd_close(struct smbclient_context *ctx, const char **args)
2237 {
2238         union smb_close io;
2239         NTSTATUS status;
2240         uint16_t fnum;
2241
2242         if (!args[1]) {
2243                 d_printf("close <fnum>\n");
2244                 return 1;
2245         }
2246
2247         fnum = atoi(args[1]);
2248
2249         ZERO_STRUCT(io);
2250         io.generic.level = RAW_CLOSE_CLOSE;
2251         io.close.in.file.fnum = fnum;
2252
2253         status = smb_raw_close(ctx->cli->tree, &io);
2254
2255         if (NT_STATUS_IS_OK(status)) {
2256                 d_printf("Closed file OK\n");
2257         } else {
2258                 d_printf("Close failed: %s\n", nt_errstr(status));
2259         }
2260
2261         return 0;
2262 }
2263
2264
2265 /****************************************************************************
2266 remove a directory
2267 ****************************************************************************/
2268 static int cmd_rmdir(struct smbclient_context *ctx, const char **args)
2269 {
2270         char *mask;
2271
2272         if (!args[1]) {
2273                 d_printf("rmdir <dirname>\n");
2274                 return 1;
2275         }
2276         mask = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2277
2278         if (NT_STATUS_IS_ERR(smbcli_rmdir(ctx->cli->tree, mask))) {
2279                 d_printf("%s removing remote directory file %s\n",
2280                          smbcli_errstr(ctx->cli->tree),mask);
2281         }
2282
2283         return 0;
2284 }
2285
2286 /****************************************************************************
2287  UNIX hardlink.
2288 ****************************************************************************/
2289 static int cmd_link(struct smbclient_context *ctx, const char **args)
2290 {
2291         char *src,*dest;
2292
2293         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2294                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2295                 return 1;
2296         }
2297
2298
2299         if (!args[1] || !args[2]) {
2300                 d_printf("link <src> <dest>\n");
2301                 return 1;
2302         }
2303
2304         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2305         dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2306
2307         if (NT_STATUS_IS_ERR(smbcli_unix_hardlink(ctx->cli->tree, src, dest))) {
2308                 d_printf("%s linking files (%s -> %s)\n", smbcli_errstr(ctx->cli->tree), src, dest);
2309                 return 1;
2310         }
2311
2312         return 0;
2313 }
2314
2315 /****************************************************************************
2316  UNIX symlink.
2317 ****************************************************************************/
2318
2319 static int cmd_symlink(struct smbclient_context *ctx, const char **args)
2320 {
2321         char *src,*dest;
2322
2323         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2324                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2325                 return 1;
2326         }
2327
2328         if (!args[1] || !args[2]) {
2329                 d_printf("symlink <src> <dest>\n");
2330                 return 1;
2331         }
2332
2333         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2334         dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2335
2336         if (NT_STATUS_IS_ERR(smbcli_unix_symlink(ctx->cli->tree, src, dest))) {
2337                 d_printf("%s symlinking files (%s -> %s)\n",
2338                         smbcli_errstr(ctx->cli->tree), src, dest);
2339                 return 1;
2340         }
2341
2342         return 0;
2343 }
2344
2345 /****************************************************************************
2346  UNIX chmod.
2347 ****************************************************************************/
2348
2349 static int cmd_chmod(struct smbclient_context *ctx, const char **args)
2350 {
2351         char *src;
2352         mode_t mode;
2353
2354         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2355                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2356                 return 1;
2357         }
2358
2359         if (!args[1] || !args[2]) {
2360                 d_printf("chmod mode file\n");
2361                 return 1;
2362         }
2363
2364         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2365
2366         mode = (mode_t)strtol(args[1], NULL, 8);
2367
2368         if (NT_STATUS_IS_ERR(smbcli_unix_chmod(ctx->cli->tree, src, mode))) {
2369                 d_printf("%s chmod file %s 0%o\n",
2370                         smbcli_errstr(ctx->cli->tree), src, (unsigned)mode);
2371                 return 1;
2372         }
2373
2374         return 0;
2375 }
2376
2377 /****************************************************************************
2378  UNIX chown.
2379 ****************************************************************************/
2380
2381 static int cmd_chown(struct smbclient_context *ctx, const char **args)
2382 {
2383         char *src;
2384         uid_t uid;
2385         gid_t gid;
2386
2387         if (!(ctx->cli->transport->negotiate.capabilities & CAP_UNIX)) {
2388                 d_printf("Server doesn't support UNIX CIFS calls.\n");
2389                 return 1;
2390         }
2391
2392         if (!args[1] || !args[2] || !args[3]) {
2393                 d_printf("chown uid gid file\n");
2394                 return 1;
2395         }
2396
2397         uid = (uid_t)atoi(args[1]);
2398         gid = (gid_t)atoi(args[2]);
2399         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[3]);
2400
2401         if (NT_STATUS_IS_ERR(smbcli_unix_chown(ctx->cli->tree, src, uid, gid))) {
2402                 d_printf("%s chown file %s uid=%d, gid=%d\n",
2403                         smbcli_errstr(ctx->cli->tree), src, (int)uid, (int)gid);
2404                 return 1;
2405         }
2406
2407         return 0;
2408 }
2409
2410 /****************************************************************************
2411 rename some files
2412 ****************************************************************************/
2413 static int cmd_rename(struct smbclient_context *ctx, const char **args)
2414 {
2415         char *src,*dest;
2416
2417         if (!args[1] || !args[2]) {
2418                 d_printf("rename <src> <dest>\n");
2419                 return 1;
2420         }
2421
2422         src = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[1]);
2423         dest = talloc_asprintf(ctx, "%s%s", ctx->remote_cur_dir, args[2]);
2424
2425         if (NT_STATUS_IS_ERR(smbcli_rename(ctx->cli->tree, src, dest))) {
2426                 d_printf("%s renaming files\n",smbcli_errstr(ctx->cli->tree));
2427                 return 1;
2428         }
2429
2430         return 0;
2431 }
2432
2433
2434 /****************************************************************************
2435 toggle the prompt flag
2436 ****************************************************************************/
2437 static int cmd_prompt(struct smbclient_context *ctx, const char **args)
2438 {
2439         ctx->prompt = !ctx->prompt;
2440         DEBUG(2,("prompting is now %s\n",ctx->prompt?"on":"off"));
2441
2442         return 1;
2443 }
2444
2445
2446 /****************************************************************************
2447 set the newer than time
2448 ****************************************************************************/
2449 static int cmd_newer(struct smbclient_context *ctx, const char **args)
2450 {
2451         struct stat sbuf;
2452
2453         if (args[1] && (stat(args[1],&sbuf) == 0)) {
2454                 ctx->newer_than = sbuf.st_mtime;
2455                 DEBUG(1,("Getting files newer than %s",
2456                          asctime(localtime(&ctx->newer_than))));
2457         } else {
2458                 ctx->newer_than = 0;
2459         }
2460
2461         if (args[1] && ctx->newer_than == 0) {
2462                 d_printf("Error setting newer-than time\n");
2463                 return 1;
2464         }
2465
2466         return 0;
2467 }
2468
2469 /****************************************************************************
2470 set the archive level
2471 ****************************************************************************/
2472 static int cmd_archive(struct smbclient_context *ctx, const char **args)
2473 {
2474         if (args[1]) {
2475                 ctx->archive_level = atoi(args[1]);
2476         } else
2477                 d_printf("Archive level is %d\n",ctx->archive_level);
2478
2479         return 0;
2480 }
2481
2482 /****************************************************************************
2483 toggle the lowercaseflag
2484 ****************************************************************************/
2485 static int cmd_lowercase(struct smbclient_context *ctx, const char **args)
2486 {
2487         ctx->lowercase = !ctx->lowercase;
2488         DEBUG(2,("filename lowercasing is now %s\n",ctx->lowercase?"on":"off"));
2489
2490         return 0;
2491 }
2492
2493
2494
2495
2496 /****************************************************************************
2497 toggle the recurse flag
2498 ****************************************************************************/
2499 static int cmd_recurse(struct smbclient_context *ctx, const char **args)
2500 {
2501         ctx->recurse = !ctx->recurse;
2502         DEBUG(2,("directory recursion is now %s\n",ctx->recurse?"on":"off"));
2503
2504         return 0;
2505 }
2506
2507 /****************************************************************************
2508 toggle the translate flag
2509 ****************************************************************************/
2510 static int cmd_translate(struct smbclient_context *ctx, const char **args)
2511 {
2512         ctx->translation = !ctx->translation;
2513         DEBUG(2,("CR/LF<->LF and print text translation now %s\n",
2514                  ctx->translation?"on":"off"));
2515
2516         return 0;
2517 }
2518
2519
2520 /****************************************************************************
2521 do a printmode command
2522 ****************************************************************************/
2523 static int cmd_printmode(struct smbclient_context *ctx, const char **args)
2524 {
2525         if (args[1]) {
2526                 if (strequal(args[1],"text")) {
2527                         ctx->printmode = 0;
2528                 } else {
2529                         if (strequal(args[1],"graphics"))
2530                                 ctx->printmode = 1;
2531                         else
2532                                 ctx->printmode = atoi(args[1]);
2533                 }
2534         }
2535
2536         switch(ctx->printmode)
2537         {
2538                 case 0:
2539                         DEBUG(2,("the printmode is now text\n"));
2540                         break;
2541                 case 1:
2542                         DEBUG(2,("the printmode is now graphics\n"));
2543                         break;
2544                 default:
2545                         DEBUG(2,("the printmode is now %d\n", ctx->printmode));
2546                         break;
2547         }
2548
2549         return 0;
2550 }
2551
2552 /****************************************************************************
2553  do the lcd command
2554  ****************************************************************************/
2555 static int cmd_lcd(struct smbclient_context *ctx, const char **args)
2556 {
2557         char d[PATH_MAX];
2558
2559         if (args[1]) {
2560                 int ret;
2561
2562                 ret = chdir(args[1]);
2563                 if (ret == -1) {
2564                         d_printf("failed to chdir to dir '%s': %s\n",
2565                                  args[1], strerror(errno));
2566                         return 1;
2567                 }
2568         }
2569
2570         DEBUG(2,("the local directory is now %s\n",getcwd(d, PATH_MAX)));
2571
2572         return 0;
2573 }
2574
2575 /****************************************************************************
2576 history
2577 ****************************************************************************/
2578 static int cmd_history(struct smbclient_context *ctx, const char **args)
2579 {
2580 #if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST)
2581         HIST_ENTRY **hlist;
2582         int i;
2583
2584         hlist = history_list();
2585
2586         for (i = 0; hlist && hlist[i]; i++) {
2587                 DEBUG(0, ("%d: %s\n", i, hlist[i]->line));
2588         }
2589 #else
2590         DEBUG(0,("no history without readline support\n"));
2591 #endif
2592
2593         return 0;
2594 }
2595
2596 /****************************************************************************
2597  get a file restarting at end of local file
2598  ****************************************************************************/
2599 static int cmd_reget(struct smbclient_context *ctx, const char **args)
2600 {
2601         char *local_name;
2602         char *remote_name;
2603
2604         if (!args[1]) {
2605                 d_printf("reget <filename>\n");
2606                 return 1;
2607         }
2608         remote_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2609         dos_clean_name(remote_name);
2610
2611         if (args[2])
2612                 local_name = talloc_strdup(ctx, args[2]);
2613         else
2614                 local_name = talloc_strdup(ctx, args[1]);
2615
2616         return do_get(ctx, remote_name, local_name, true);
2617 }
2618
2619 /****************************************************************************
2620  put a file restarting at end of local file
2621  ****************************************************************************/
2622 static int cmd_reput(struct smbclient_context *ctx, const char **args)
2623 {
2624         char *local_name;
2625         char *remote_name;
2626
2627         if (!args[1]) {
2628                 d_printf("reput <filename>\n");
2629                 return 1;
2630         }
2631         local_name = talloc_asprintf(ctx, "%s\\%s", ctx->remote_cur_dir, args[1]);
2632
2633         if (!file_exist(local_name)) {
2634                 d_printf("%s does not exist\n", local_name);
2635                 return 1;
2636         }
2637
2638         if (args[2])
2639                 remote_name = talloc_strdup(ctx, args[2]);
2640         else
2641                 remote_name = talloc_strdup(ctx, args[1]);
2642
2643         dos_clean_name(remote_name);
2644
2645         return do_put(ctx, remote_name, local_name, true);
2646 }
2647
2648
2649 /*
2650   return a string representing a share type
2651 */
2652 static const char *share_type_str(uint32_t type)
2653 {
2654         switch (type & 0xF) {
2655         case STYPE_DISKTREE:
2656                 return "Disk";
2657         case STYPE_PRINTQ:
2658                 return "Printer";
2659         case STYPE_DEVICE:
2660                 return "Device";
2661         case STYPE_IPC:
2662                 return "IPC";
2663         default:
2664                 return "Unknown";
2665         }
2666 }
2667
2668
2669 /*
2670   display a list of shares from a level 1 share enum
2671 */
2672 static void display_share_result(struct srvsvc_NetShareCtr1 *ctr1)
2673 {
2674         int i;
2675
2676         for (i=0;i<ctr1->count;i++) {
2677                 struct srvsvc_NetShareInfo1 *info = ctr1->array+i;
2678
2679                 printf("\t%-15s %-10.10s %s\n",
2680                        info->name,
2681                        share_type_str(info->type),
2682                        info->comment);
2683         }
2684 }
2685
2686
2687
2688 /****************************************************************************
2689 try and browse available shares on a host
2690 ****************************************************************************/
2691 static bool browse_host(struct loadparm_context *lp_ctx,
2692                         struct tevent_context *ev_ctx,
2693                         const char *query_host)
2694 {
2695         struct dcerpc_pipe *p;
2696         char *binding;
2697         NTSTATUS status;
2698         struct srvsvc_NetShareEnumAll r;
2699         struct srvsvc_NetShareInfoCtr info_ctr;
2700         uint32_t resume_handle = 0;
2701         TALLOC_CTX *mem_ctx = talloc_init("browse_host");
2702         struct srvsvc_NetShareCtr1 ctr1;
2703         uint32_t totalentries = 0;
2704
2705         binding = talloc_asprintf(mem_ctx, "ncacn_np:%s", query_host);
2706
2707         status = dcerpc_pipe_connect(mem_ctx, &p, binding,
2708                                          &ndr_table_srvsvc,
2709                                      popt_get_cmdline_credentials(), ev_ctx,
2710                                      lp_ctx);
2711         if (!NT_STATUS_IS_OK(status)) {
2712                 d_printf("Failed to connect to %s - %s\n",
2713                          binding, nt_errstr(status));
2714                 talloc_free(mem_ctx);
2715                 return false;
2716         }
2717
2718         info_ctr.level = 1;
2719         info_ctr.ctr.ctr1 = &ctr1;
2720
2721         r.in.server_unc = talloc_asprintf(mem_ctx,"\\\\%s",dcerpc_server_name(p));
2722         r.in.info_ctr = &info_ctr;
2723         r.in.max_buffer = ~0;
2724         r.in.resume_handle = &resume_handle;
2725         r.out.resume_handle = &resume_handle;
2726         r.out.totalentries = &totalentries;
2727         r.out.info_ctr = &info_ctr;
2728
2729         d_printf("\n\tSharename       Type       Comment\n");
2730         d_printf("\t---------       ----       -------\n");
2731
2732         do {
2733                 ZERO_STRUCT(ctr1);
2734                 status = dcerpc_srvsvc_NetShareEnumAll_r(p->binding_handle, mem_ctx, &r);
2735
2736                 if (NT_STATUS_IS_OK(status) &&
2737                     (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA) ||
2738                      W_ERROR_IS_OK(r.out.result)) &&
2739                     r.out.info_ctr->ctr.ctr1) {
2740                         display_share_result(r.out.info_ctr->ctr.ctr1);
2741                         resume_handle += r.out.info_ctr->ctr.ctr1->count;
2742                 }
2743         } while (NT_STATUS_IS_OK(status) && W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA));
2744
2745         talloc_free(mem_ctx);
2746
2747         if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(r.out.result)) {
2748                 d_printf("Failed NetShareEnumAll %s - %s/%s\n",
2749                          binding, nt_errstr(status), win_errstr(r.out.result));
2750                 return false;
2751         }
2752
2753         return false;
2754 }
2755
2756 /****************************************************************************
2757 try and browse available connections on a host
2758 ****************************************************************************/
2759 static bool list_servers(const char *wk_grp)
2760 {
2761         d_printf("REWRITE: list servers not implemented\n");
2762         return false;
2763 }
2764
2765 /* Some constants for completing filename arguments */
2766
2767 #define COMPL_NONE        0          /* No completions */
2768 #define COMPL_REMOTE      1          /* Complete remote filename */
2769 #define COMPL_LOCAL       2          /* Complete local filename */
2770
2771 static int cmd_help(struct smbclient_context *ctx, const char **args);
2772
2773 /* This defines the commands supported by this client.
2774  * NOTE: The "!" must be the last one in the list because it's fn pointer
2775  *       field is NULL, and NULL in that field is used in process_tok()
2776  *       (below) to indicate the end of the list.  crh
2777  */
2778 static struct
2779 {
2780   const char *name;
2781   int (*fn)(struct smbclient_context *ctx, const char **args);
2782   const char *description;
2783   char compl_args[2];      /* Completion argument info */
2784 } commands[] =
2785 {
2786   {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2787   {"addprivileges",cmd_addprivileges,"<sid|name> <privilege...> add privileges for a user",{COMPL_NONE,COMPL_NONE}},
2788   {"altname",cmd_altname,"<file> show alt name",{COMPL_NONE,COMPL_NONE}},
2789   {"acl",cmd_acl,"<file> show file ACL",{COMPL_NONE,COMPL_NONE}},
2790   {"allinfo",cmd_allinfo,"<file> show all possible info about a file",{COMPL_NONE,COMPL_NONE}},
2791   {"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}},
2792   {"cancel",cmd_rewrite,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}},
2793   {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}},
2794   {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}},
2795   {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}},
2796   {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2797   {"delprivileges",cmd_delprivileges,"<sid|name> <privilege...> remove privileges for a user",{COMPL_NONE,COMPL_NONE}},
2798   {"deltree",cmd_deltree,"<dir> delete a whole directory tree",{COMPL_REMOTE,COMPL_NONE}},
2799   {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2800   {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2801   {"eainfo",cmd_eainfo,"<file> show EA contents for a file",{COMPL_NONE,COMPL_NONE}},
2802   {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2803   {"fsinfo",cmd_fsinfo,"query file system info",{COMPL_NONE,COMPL_NONE}},
2804   {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}},
2805   {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}},
2806   {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}},
2807   {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}},
2808   {"link",cmd_link,"<src> <dest> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}},
2809   {"lookup",cmd_lookup,"<sid|name> show SID for name or name for SID",{COMPL_NONE,COMPL_NONE}},
2810   {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}},
2811   {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}},
2812   {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}},
2813   {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2814   {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}},
2815   {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}},
2816   {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}},
2817   {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}},
2818   {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}},
2819   {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}},
2820   {"close",cmd_close,"<fnum> close a file",{COMPL_NONE,COMPL_NONE}},
2821   {"privileges",cmd_privileges,"<user> show privileges for a user",{COMPL_NONE,COMPL_NONE}},
2822   {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}},
2823   {"printmode",cmd_printmode,"<graphics or text> set the print mode",{COMPL_NONE,COMPL_NONE}},
2824   {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}},
2825   {"put",cmd_put,"<local name> [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}},
2826   {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}},
2827   {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2828   {"queue",cmd_rewrite,"show the print queue",{COMPL_NONE,COMPL_NONE}},
2829   {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}},
2830   {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2831   {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}},
2832   {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}},
2833   {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}},
2834   {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}},
2835   {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}},
2836   {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}},
2837   {"symlink",cmd_symlink,"<src> <dest> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}},
2838   {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}},
2839
2840   /* Yes, this must be here, see crh's comment above. */
2841   {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}},
2842   {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}}
2843 };
2844
2845
2846 /*******************************************************************
2847   lookup a command string in the list of commands, including
2848   abbreviations
2849   ******************************************************************/
2850 static int process_tok(const char *tok)
2851 {
2852         int i = 0, matches = 0;
2853         int cmd=0;
2854         int tok_len = strlen(tok);
2855
2856         while (commands[i].fn != NULL) {
2857                 if (strequal(commands[i].name,tok)) {
2858                         matches = 1;
2859                         cmd = i;
2860                         break;
2861                 } else if (strncasecmp(commands[i].name, tok, tok_len) == 0) {
2862                         matches++;
2863                         cmd = i;
2864                 }
2865                 i++;
2866         }
2867
2868         if (matches == 0)
2869                 return(-1);
2870         else if (matches == 1)
2871                 return(cmd);
2872         else
2873                 return(-2);
2874 }
2875
2876 /****************************************************************************
2877 help
2878 ****************************************************************************/
2879 static int cmd_help(struct smbclient_context *ctx, const char **args)
2880 {
2881         int i=0,j;
2882
2883         if (args[1]) {
2884                 if ((i = process_tok(args[1])) >= 0)
2885                         d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description);
2886         } else {
2887                 while (commands[i].description) {
2888                         for (j=0; commands[i].description && (j<5); j++) {
2889                                 d_printf("%-15s",commands[i].name);
2890                                 i++;
2891                         }
2892                         d_printf("\n");
2893                 }
2894         }
2895         return 0;
2896 }
2897
2898 static int process_line(struct smbclient_context *ctx, const char *cline);
2899 /****************************************************************************
2900 process a -c command string
2901 ****************************************************************************/
2902 static int process_command_string(struct smbclient_context *ctx, const char *cmd)
2903 {
2904         char **lines;
2905         int i, rc = 0;
2906
2907         lines = str_list_make(NULL, cmd, ";");
2908         for (i = 0; lines[i]; i++) {
2909                 rc |= process_line(ctx, lines[i]);
2910         }
2911         talloc_free(lines);
2912
2913         return rc;
2914 }
2915
2916 #define MAX_COMPLETIONS 100
2917
2918 typedef struct {
2919         char *dirmask;
2920         char **matches;
2921         int count, samelen;
2922         const char *text;
2923         int len;
2924 } completion_remote_t;
2925
2926 static void completion_remote_filter(struct clilist_file_info *f, const char *mask, void *state)
2927 {
2928         completion_remote_t *info = (completion_remote_t *)state;
2929
2930         if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (!ISDOT(f->name)) && (!ISDOTDOT(f->name))) {
2931                 if ((info->dirmask[0] == 0) && !(f->attrib & FILE_ATTRIBUTE_DIRECTORY))
2932                         info->matches[info->count] = strdup(f->name);
2933                 else {
2934                         char *tmp;
2935
2936                         if (info->dirmask[0] != 0)
2937                                 tmp = talloc_asprintf(NULL, "%s/%s", info->dirmask, f->name);
2938                         else
2939                                 tmp = talloc_strdup(NULL, f->name);
2940
2941                         if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2942                                 tmp = talloc_append_string(NULL, tmp, "/");
2943                         info->matches[info->count] = tmp;
2944                 }
2945                 if (info->matches[info->count] == NULL)
2946                         return;
2947                 if (f->attrib & FILE_ATTRIBUTE_DIRECTORY)
2948                         smb_readline_ca_char(0);
2949
2950                 if (info->count == 1)
2951                         info->samelen = strlen(info->matches[info->count]);
2952                 else
2953                         while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0)
2954                                 info->samelen--;
2955                 info->count++;
2956         }
2957 }
2958
2959 static char **remote_completion(const char *text, int len)
2960 {
2961         char *dirmask;
2962         int i, ret;
2963         completion_remote_t info;
2964
2965         info.samelen = len;
2966         info.text = text;
2967         info.len = len;
2968         info.count = 0;
2969
2970         if (len >= PATH_MAX)
2971                 return(NULL);
2972
2973         info.matches = malloc_array_p(char *, MAX_COMPLETIONS);
2974         if (!info.matches) return NULL;
2975         info.matches[0] = NULL;
2976
2977         for (i = len-1; i >= 0; i--)
2978                 if ((text[i] == '/') || (text[i] == '\\'))
2979                         break;
2980         info.text = text+i+1;
2981         info.samelen = info.len = len-i-1;
2982
2983         if (i > 0) {
2984                 info.dirmask = talloc_strndup(NULL, text, i+1);
2985                 info.dirmask[i+1] = 0;
2986                 ret = asprintf(&dirmask, "%s%*s*", rl_ctx->remote_cur_dir, i-1,
2987                                text);
2988         } else {
2989                 ret = asprintf(&dirmask, "%s*", rl_ctx->remote_cur_dir);
2990         }
2991         if (ret < 0) {
2992                 goto cleanup;
2993         }
2994
2995         if (smbcli_list(rl_ctx->cli->tree, dirmask,
2996                      FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN,
2997                      completion_remote_filter, &info) < 0)
2998                 goto cleanup;
2999
3000         if (info.count == 2)
3001                 info.matches[0] = strdup(info.matches[1]);
3002         else {
3003                 info.matches[0] = malloc_array_p(char, info.samelen+1);
3004                 if (!info.matches[0])
3005                         goto cleanup;
3006                 strncpy(info.matches[0], info.matches[1], info.samelen);
3007                 info.matches[0][info.samelen] = 0;
3008         }
3009         info.matches[info.count] = NULL;
3010         return info.matches;
3011
3012 cleanup:
3013         for (i = 0; i < info.count; i++)
3014                 free(info.matches[i]);
3015         free(info.matches);
3016         return NULL;
3017 }
3018
3019 static char **completion_fn(const char *text, int start, int end)
3020 {
3021         smb_readline_ca_char(' ');
3022
3023         if (start) {
3024                 const char *buf, *sp;
3025                 int i;
3026                 char compl_type;
3027
3028                 buf = smb_readline_get_line_buffer();
3029                 if (buf == NULL)
3030                         return NULL;
3031
3032                 sp = strchr(buf, ' ');
3033                 if (sp == NULL)
3034                         return NULL;
3035
3036                 for (i = 0; commands[i].name; i++)
3037                         if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0))
3038                                 break;
3039                 if (commands[i].name == NULL)
3040                         return NULL;
3041
3042                 while (*sp == ' ')
3043                         sp++;
3044
3045                 if (sp == (buf + start))
3046                         compl_type = commands[i].compl_args[0];
3047                 else
3048                         compl_type = commands[i].compl_args[1];
3049
3050                 if (compl_type == COMPL_REMOTE)
3051                         return remote_completion(text, end - start);
3052                 else /* fall back to local filename completion */
3053                         return NULL;
3054         } else {
3055                 char **matches;
3056                 int i, len, samelen = 0, count=1;
3057
3058                 matches = malloc_array_p(char *, MAX_COMPLETIONS);
3059                 if (!matches) return NULL;
3060                 matches[0] = NULL;
3061
3062                 len = strlen(text);
3063                 for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) {
3064                         if (strncmp(text, commands[i].name, len) == 0) {
3065                                 matches[count] = strdup(commands[i].name);
3066                                 if (!matches[count])
3067                                         goto cleanup;
3068                                 if (count == 1)
3069                                         samelen = strlen(matches[count]);
3070                                 else
3071                                         while (strncmp(matches[count], matches[count-1], samelen) != 0)
3072                                                 samelen--;
3073                                 count++;
3074                         }
3075                 }
3076
3077                 switch (count) {
3078                 case 0: /* should never happen */
3079                 case 1:
3080                         goto cleanup;
3081                 case 2:
3082                         matches[0] = strdup(matches[1]);
3083                         break;
3084                 default:
3085                         matches[0] = malloc_array_p(char, samelen+1);
3086                         if (!matches[0])
3087                                 goto cleanup;
3088                         strncpy(matches[0], matches[1], samelen);
3089                         matches[0][samelen] = 0;
3090                 }
3091                 matches[count] = NULL;
3092                 return matches;
3093
3094 cleanup:
3095                 count--;
3096                 while (count >= 0) {
3097                         free(matches[count]);
3098                         count--;
3099                 }
3100                 free(matches);
3101                 return NULL;
3102         }
3103 }
3104
3105 /****************************************************************************
3106 make sure we swallow keepalives during idle time
3107 ****************************************************************************/
3108 static void readline_callback(void)
3109 {
3110         static time_t last_t;
3111         time_t t;
3112
3113         t = time(NULL);
3114
3115         if (t - last_t < 5) return;
3116
3117         last_t = t;
3118
3119         smbcli_transport_process(rl_ctx->cli->transport);
3120
3121         if (rl_ctx->cli->tree) {
3122                 smbcli_chkpath(rl_ctx->cli->tree, "\\");
3123         }
3124 }
3125
3126 static int process_line(struct smbclient_context *ctx, const char *cline)
3127 {
3128         char **args;
3129         int i;
3130
3131         /* and get the first part of the command */
3132         args = str_list_make_shell(ctx, cline, NULL);
3133         if (!args || !args[0])
3134                 return 0;
3135
3136         if ((i = process_tok(args[0])) >= 0) {
3137                 const char **a = discard_const_p(const char *, args);
3138                 i = commands[i].fn(ctx, a);
3139         } else if (i == -2) {
3140                 d_printf("%s: command abbreviation ambiguous\n",args[0]);
3141         } else {
3142                 d_printf("%s: command not found\n",args[0]);
3143         }
3144
3145         talloc_free(args);
3146
3147         return i;
3148 }
3149
3150 /****************************************************************************
3151 process commands on stdin
3152 ****************************************************************************/
3153 static int process_stdin(struct smbclient_context *ctx)
3154 {
3155         int rc = 0;
3156         while (1) {
3157                 /* display a prompt */
3158                 char *the_prompt = talloc_asprintf(ctx, "smb: %s> ", ctx->remote_cur_dir);
3159                 char *cline = smb_readline(the_prompt, readline_callback, completion_fn);
3160                 talloc_free(the_prompt);
3161
3162                 if (!cline) break;
3163
3164                 /* special case - first char is ! */
3165                 if (*cline == '!') {
3166                         int ret;
3167                         ret = system(cline + 1);
3168                         free(cline);
3169                         if (ret == -1) {
3170                                 rc |= ret;
3171                         }
3172                         continue;
3173                 }
3174
3175                 rc |= process_command_string(ctx, cline);
3176                 free(cline);
3177
3178         }
3179
3180         return rc;
3181 }
3182
3183
3184 /*****************************************************
3185 return a connection to a server
3186 *******************************************************/
3187 static bool do_connect(struct smbclient_context *ctx,
3188                        struct tevent_context *ev_ctx,
3189                        struct resolve_context *resolve_ctx,
3190                        const char *specified_server, const char **ports,
3191                        const char *specified_share,
3192                            const char *socket_options,
3193                        struct cli_credentials *cred,
3194                        struct smbcli_options *options,
3195                        struct smbcli_session_options *session_options,
3196                            struct gensec_settings *gensec_settings)
3197 {
3198         NTSTATUS status;
3199         char *server, *share;
3200
3201         rl_ctx = ctx; /* Ugly hack */
3202
3203         if (strncmp(specified_share, "\\\\", 2) == 0 ||
3204             strncmp(specified_share, "//", 2) == 0) {
3205                 bool ok;
3206
3207                 ok = smbcli_parse_unc(specified_share, ctx, &server, &share);
3208                 if (!ok) {
3209                         d_printf("Failed to parse UNC\n");
3210                         talloc_free(ctx);
3211                         return false;
3212                 }
3213         } else {
3214                 share = talloc_strdup(ctx, specified_share);
3215                 server = talloc_strdup(ctx, specified_server);
3216                 if (share == NULL || server == NULL) {
3217                         d_printf("Failed to allocate memory for server and share\n");
3218                         talloc_free(ctx);
3219                         return false;
3220                 }
3221         }
3222
3223         ctx->remote_cur_dir = talloc_strdup(ctx, "\\");
3224         if (ctx->remote_cur_dir == NULL) {
3225                 talloc_free(ctx);
3226                 return false;
3227         }
3228
3229         status = smbcli_full_connection(ctx, &ctx->cli, server, ports,
3230                                         share, NULL,
3231                                         socket_options,
3232                                         cred, resolve_ctx,
3233                                         ev_ctx, options, session_options,
3234                                         gensec_settings);
3235         if (!NT_STATUS_IS_OK(status)) {
3236                 d_printf("Connection to \\\\%s\\%s failed - %s\n",
3237                          server, share, nt_errstr(status));
3238                 talloc_free(ctx);
3239                 return false;
3240         }
3241
3242         return true;
3243 }
3244
3245 /****************************************************************************
3246 handle a -L query
3247 ****************************************************************************/
3248 static int do_host_query(struct loadparm_context *lp_ctx,
3249                          struct tevent_context *ev_ctx,
3250                          const char *query_host,
3251                          const char *workgroup)
3252 {
3253         browse_host(lp_ctx, ev_ctx, query_host);
3254         list_servers(workgroup);
3255         return(0);
3256 }
3257
3258
3259 /****************************************************************************
3260 handle a message operation
3261 ****************************************************************************/
3262 static int do_message_op(const char *netbios_name, const char *desthost,
3263                          const char **destports, const char *destip,
3264                          int name_type,
3265                          struct tevent_context *ev_ctx,
3266                          struct resolve_context *resolve_ctx,
3267                          struct smbcli_options *options,
3268              const char *socket_options)
3269 {
3270         struct nbt_name called, calling;
3271         const char *server_name;
3272         struct smbcli_state *cli;
3273         bool ok;
3274
3275         make_nbt_name_client(&calling, netbios_name);
3276
3277         nbt_choose_called_name(NULL, &called, desthost, name_type);
3278
3279         server_name = destip ? destip : desthost;
3280
3281         cli = smbcli_state_init(NULL);
3282         if (cli == NULL) {
3283                 d_printf("smbcli_state_init() failed\n");
3284                 return 1;
3285         }
3286
3287         ok = smbcli_socket_connect(cli, server_name, destports,
3288                                    ev_ctx, resolve_ctx, options,
3289                                    socket_options,
3290                                    &calling, &called);
3291         if (!ok) {
3292                 d_printf("Connection to %s failed\n", server_name);
3293                 return 1;
3294         }
3295
3296         send_message(cli, desthost);
3297         talloc_free(cli);
3298
3299         return 0;
3300 }
3301
3302
3303 /****************************************************************************
3304   main program
3305 ****************************************************************************/
3306  int main(int argc, const char *argv[])
3307 {
3308         char *base_directory = NULL;
3309         const char *dest_ip = NULL;
3310         int opt;
3311         const char *query_host = NULL;
3312         bool message = false;
3313         char *desthost = NULL;
3314         poptContext pc;
3315         const char *service = NULL;
3316         int port = 0;
3317         char *p;
3318         int rc = 0;
3319         int name_type = 0x20;
3320         TALLOC_CTX *mem_ctx;
3321         struct tevent_context *ev_ctx;
3322         struct smbclient_context *ctx;
3323         const char *cmdstr = NULL;
3324         struct smbcli_options smb_options;
3325         struct smbcli_session_options smb_session_options;
3326
3327         struct poptOption long_options[] = {
3328                 POPT_AUTOHELP
3329
3330                 { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" },
3331                 { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" },
3332                 { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" },
3333                 { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" },
3334                 { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" },
3335                 { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" },
3336                 { "send-buffer", 'b', POPT_ARG_INT, NULL, 'b', "Changes the transmit/send buffer", "BYTES" },
3337                 { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" },
3338                 POPT_COMMON_SAMBA
3339                 POPT_COMMON_CONNECTION
3340                 POPT_COMMON_CREDENTIALS
3341                 POPT_COMMON_VERSION
3342                 { NULL }
3343         };
3344
3345         mem_ctx = talloc_init("client.c/main");
3346         if (!mem_ctx) {
3347                 d_printf("\nclient.c: Not enough memory\n");
3348                 exit(1);
3349         }
3350
3351         ctx = talloc_zero(mem_ctx, struct smbclient_context);
3352         ctx->io_bufsize = 64512;
3353
3354         pc = poptGetContext("smbclient", argc, argv, long_options, 0);
3355         poptSetOtherOptionHelp(pc, "[OPTIONS] service <password>");
3356
3357         while ((opt = poptGetNextOpt(pc)) != -1) {
3358                 switch (opt) {
3359                 case 'M':
3360                         /* Messages are sent to NetBIOS name type 0x3
3361                          * (Messenger Service).  Make sure we default
3362                          * to port 139 instead of port 445. srl,crh
3363                          */
3364                         name_type = 0x03;
3365                         desthost = strdup(poptGetOptArg(pc));
3366                         if( 0 == port ) port = 139;
3367                         message = true;
3368                         break;
3369                 case 'I':
3370                         dest_ip = poptGetOptArg(pc);
3371                         break;
3372                 case 'L':
3373                         query_host = strdup(poptGetOptArg(pc));
3374                         break;
3375                 case 'D':
3376                         base_directory = strdup(poptGetOptArg(pc));
3377                         break;
3378                 case 'b':
3379                         ctx->io_bufsize = MAX(1, atoi(poptGetOptArg(pc)));
3380                         break;
3381                 }
3382         }
3383
3384         gensec_init();
3385
3386         if(poptPeekArg(pc)) {
3387                 char *s = strdup(poptGetArg(pc));
3388
3389                 /* Convert any '/' characters in the service name to '\' characters */
3390                 string_replace(s, '/','\\');
3391
3392                 service = s;
3393
3394                 if (count_chars(s,'\\') < 3) {
3395                         d_printf("\n%s: Not enough '\\' characters in service\n",s);
3396                         poptPrintUsage(pc, stderr, 0);
3397                         exit(1);
3398                 }
3399         }
3400
3401         if (poptPeekArg(pc)) {
3402                 cli_credentials_set_password(popt_get_cmdline_credentials(),
3403                         poptGetArg(pc), CRED_SPECIFIED);
3404         }
3405
3406         /*init_names(); */
3407
3408         if (!query_host && !service && !message) {
3409                 poptPrintUsage(pc, stderr, 0);
3410                 exit(1);
3411         }
3412
3413         poptFreeContext(pc);
3414
3415         lpcfg_smbcli_options(cmdline_lp_ctx, &smb_options);
3416         lpcfg_smbcli_session_options(cmdline_lp_ctx, &smb_session_options);
3417
3418         ev_ctx = s4_event_context_init(ctx);
3419
3420         DEBUG( 3, ( "Client started (version %s).\n", SAMBA_VERSION_STRING ) );
3421
3422         if (query_host && (p=strchr_m(query_host,'#'))) {
3423                 *p = 0;
3424                 p++;
3425                 sscanf(p, "%x", &name_type);
3426         }
3427
3428         if (query_host) {
3429                 rc = do_host_query(cmdline_lp_ctx, ev_ctx, query_host,
3430                                    lpcfg_workgroup(cmdline_lp_ctx));
3431                 return rc;
3432         }
3433
3434         if (message) {
3435                 rc = do_message_op(lpcfg_netbios_name(cmdline_lp_ctx), desthost,
3436                                    lpcfg_smb_ports(cmdline_lp_ctx), dest_ip,
3437                                    name_type, ev_ctx,
3438                                    lpcfg_resolve_context(cmdline_lp_ctx),
3439                                    &smb_options,
3440                    lpcfg_socket_options(cmdline_lp_ctx));
3441                 return rc;
3442         }
3443
3444         if (!do_connect(ctx, ev_ctx, lpcfg_resolve_context(cmdline_lp_ctx),
3445                         desthost, lpcfg_smb_ports(cmdline_lp_ctx), service,
3446                         lpcfg_socket_options(cmdline_lp_ctx),
3447                         popt_get_cmdline_credentials(),
3448                         &smb_options, &smb_session_options,
3449                         lpcfg_gensec_settings(ctx, cmdline_lp_ctx)))
3450                 return 1;
3451
3452         if (base_directory) {
3453                 do_cd(ctx, base_directory);
3454                 free(base_directory);
3455         }
3456
3457         if (cmdstr) {
3458                 rc = process_command_string(ctx, cmdstr);
3459         } else {
3460                 rc = process_stdin(ctx);
3461         }
3462
3463         free(desthost);
3464         talloc_free(mem_ctx);
3465
3466         return rc;
3467 }