source4/client/client.c

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