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