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