sftp-int.c

   1 /*
   2  * Copyright (c) 2001 Damien Miller.  All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  14  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  15  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  16  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  17  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  18  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  19  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  20  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  21  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  22  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  23  */
  24
  25 /* XXX: globbed ls */
  26 /* XXX: recursive operations */
  27
  28 #include "includes.h"
  29 RCSID("$OpenBSD: sftp-int.c,v 1.42 2002/02/05 00:00:46 djm Exp $");
  30
  31 #include "buffer.h"
  32 #include "xmalloc.h"
  33 #include "log.h"
  34 #include "pathnames.h"
  35
  36 #include "sftp.h"
  37 #include "sftp-common.h"
  38 #include "sftp-glob.h"
  39 #include "sftp-client.h"
  40 #include "sftp-int.h"
  41
  42 /* File to read commands from */
  43 extern FILE *infile;
  44
  45 /* Size of buffer used when copying files */
  46 extern size_t copy_buffer_len;
  47
  48 /* Version of server we are speaking to */
  49 int version;
  50
  51 /* Seperators for interactive commands */
  52 #define WHITESPACE " \t\r\n"
  53
  54 /* Commands for interactive mode */
  55 #define I_CHDIR         1
  56 #define I_CHGRP         2
  57 #define I_CHMOD         3
  58 #define I_CHOWN         4
  59 #define I_GET           5
  60 #define I_HELP          6
  61 #define I_LCHDIR        7
  62 #define I_LLS           8
  63 #define I_LMKDIR        9
  64 #define I_LPWD          10
  65 #define I_LS            11
  66 #define I_LUMASK        12
  67 #define I_MKDIR         13
  68 #define I_PUT           14
  69 #define I_PWD           15
  70 #define I_QUIT          16
  71 #define I_RENAME        17
  72 #define I_RM            18
  73 #define I_RMDIR         19
  74 #define I_SHELL         20
  75 #define I_SYMLINK       21
  76 #define I_VERSION       22
  77
  78 struct CMD {
  79         const char *c;
  80         const int n;
  81 };
  82
  83 const struct CMD cmds[] = {
  84         { "bye",        I_QUIT },
  85         { "cd",         I_CHDIR },
  86         { "chdir",      I_CHDIR },
  87         { "chgrp",      I_CHGRP },
  88         { "chmod",      I_CHMOD },
  89         { "chown",      I_CHOWN },
  90         { "dir",        I_LS },
  91         { "exit",       I_QUIT },
  92         { "get",        I_GET },
  93         { "mget",       I_GET },
  94         { "help",       I_HELP },
  95         { "lcd",        I_LCHDIR },
  96         { "lchdir",     I_LCHDIR },
  97         { "lls",        I_LLS },
  98         { "lmkdir",     I_LMKDIR },
  99         { "ln",         I_SYMLINK },
 100         { "lpwd",       I_LPWD },
 101         { "ls",         I_LS },
 102         { "lumask",     I_LUMASK },
 103         { "mkdir",      I_MKDIR },
 104         { "put",        I_PUT },
 105         { "mput",       I_PUT },
 106         { "pwd",        I_PWD },
 107         { "quit",       I_QUIT },
 108         { "rename",     I_RENAME },
 109         { "rm",         I_RM },
 110         { "rmdir",      I_RMDIR },
 111         { "symlink",    I_SYMLINK },
 112         { "version",    I_VERSION },
 113         { "!",          I_SHELL },
 114         { "?",          I_HELP },
 115         { NULL,                 -1}
 116 };
 117
 118 static void
 119 help(void)
 120 {
 121         printf("Available commands:\n");
 122         printf("cd path                       Change remote directory to 'path'\n");
 123         printf("lcd path                      Change local directory to 'path'\n");
 124         printf("chgrp grp path                Change group of file 'path' to 'grp'\n");
 125         printf("chmod mode path               Change permissions of file 'path' to 'mode'\n");
 126         printf("chown own path                Change owner of file 'path' to 'own'\n");
 127         printf("help                          Display this help text\n");
 128         printf("get remote-path [local-path]  Download file\n");
 129         printf("lls [ls-options [path]]       Display local directory listing\n");
 130         printf("ln oldpath newpath            Symlink remote file\n");
 131         printf("lmkdir path                   Create local directory\n");
 132         printf("lpwd                          Print local working directory\n");
 133         printf("ls [path]                     Display remote directory listing\n");
 134         printf("lumask umask                  Set local umask to 'umask'\n");
 135         printf("mkdir path                    Create remote directory\n");
 136         printf("put local-path [remote-path]  Upload file\n");
 137         printf("pwd                           Display remote working directory\n");
 138         printf("exit                          Quit sftp\n");
 139         printf("quit                          Quit sftp\n");
 140         printf("rename oldpath newpath        Rename remote file\n");
 141         printf("rmdir path                    Remove remote directory\n");
 142         printf("rm path                       Delete remote file\n");
 143         printf("symlink oldpath newpath       Symlink remote file\n");
 144         printf("version                       Show SFTP version\n");
 145         printf("!command                      Execute 'command' in local shell\n");
 146         printf("!                             Escape to local shell\n");
 147         printf("?                             Synonym for help\n");
 148 }
 149
 150 static void
 151 local_do_shell(const char *args)
 152 {
 153         int status;
 154         char *shell;
 155         pid_t pid;
 156
 157         if (!*args)
 158                 args = NULL;
 159
 160         if ((shell = getenv("SHELL")) == NULL)
 161                 shell = _PATH_BSHELL;
 162
 163         if ((pid = fork()) == -1)
 164                 fatal("Couldn't fork: %s", strerror(errno));
 165
 166         if (pid == 0) {
 167                 /* XXX: child has pipe fds to ssh subproc open - issue? */
 168                 if (args) {
 169                         debug3("Executing %s -c \"%s\"", shell, args);
 170                         execl(shell, shell, "-c", args, (char *)NULL);
 171                 } else {
 172                         debug3("Executing %s", shell);
 173                         execl(shell, shell, (char *)NULL);
 174                 }
 175                 fprintf(stderr, "Couldn't execute \"%s\": %s\n", shell,
 176                     strerror(errno));
 177                 _exit(1);
 178         }
 179         if (waitpid(pid, &status, 0) == -1)
 180                 fatal("Couldn't wait for child: %s", strerror(errno));
 181         if (!WIFEXITED(status))
 182                 error("Shell exited abormally");
 183         else if (WEXITSTATUS(status))
 184                 error("Shell exited with status %d", WEXITSTATUS(status));
 185 }
 186
 187 static void
 188 local_do_ls(const char *args)
 189 {
 190         if (!args || !*args)
 191                 local_do_shell(_PATH_LS);
 192         else {
 193                 int len = strlen(_PATH_LS " ") + strlen(args) + 1;
 194                 char *buf = xmalloc(len);
 195
 196                 /* XXX: quoting - rip quoting code from ftp? */
 197                 snprintf(buf, len, _PATH_LS " %s", args);
 198                 local_do_shell(buf);
 199                 xfree(buf);
 200         }
 201 }
 202
 203 static char *
 204 path_append(char *p1, char *p2)
 205 {
 206         char *ret;
 207         int len = strlen(p1) + strlen(p2) + 2;
 208
 209         ret = xmalloc(len);
 210         strlcpy(ret, p1, len);
 211         if (strcmp(p1, "/") != 0)
 212                 strlcat(ret, "/", len);
 213         strlcat(ret, p2, len);
 214
 215         return(ret);
 216 }
 217
 218 static char *
 219 make_absolute(char *p, char *pwd)
 220 {
 221         char *abs;
 222
 223         /* Derelativise */
 224         if (p && p[0] != '/') {
 225                 abs = path_append(pwd, p);
 226                 xfree(p);
 227                 return(abs);
 228         } else
 229                 return(p);
 230 }
 231
 232 static int
 233 infer_path(const char *p, char **ifp)
 234 {
 235         char *cp;
 236
 237         cp = strrchr(p, '/');
 238         if (cp == NULL) {
 239                 *ifp = xstrdup(p);
 240                 return(0);
 241         }
 242
 243         if (!cp[1]) {
 244                 error("Invalid path");
 245                 return(-1);
 246         }
 247
 248         *ifp = xstrdup(cp + 1);
 249         return(0);
 250 }
 251
 252 static int
 253 parse_getput_flags(const char **cpp, int *pflag)
 254 {
 255         const char *cp = *cpp;
 256
 257         /* Check for flags */
 258         if (cp[0] == '-' && cp[1] && strchr(WHITESPACE, cp[2])) {
 259                 switch (cp[1]) {
 260                 case 'p':
 261                 case 'P':
 262                         *pflag = 1;
 263                         break;
 264                 default:
 265                         error("Invalid flag -%c", cp[1]);
 266                         return(-1);
 267                 }
 268                 cp += 2;
 269                 *cpp = cp + strspn(cp, WHITESPACE);
 270         }
 271
 272         return(0);
 273 }
 274
 275 static int
 276 get_pathname(const char **cpp, char **path)
 277 {
 278         const char *cp = *cpp, *end;
 279         char quot;
 280         int i;
 281
 282         cp += strspn(cp, WHITESPACE);
 283         if (!*cp) {
 284                 *cpp = cp;
 285                 *path = NULL;
 286                 return (0);
 287         }
 288
 289         /* Check for quoted filenames */
 290         if (*cp == '\"' || *cp == '\'') {
 291                 quot = *cp++;
 292
 293                 end = strchr(cp, quot);
 294                 if (end == NULL) {
 295                         error("Unterminated quote");
 296                         goto fail;
 297                 }
 298                 if (cp == end) {
 299                         error("Empty quotes");
 300                         goto fail;
 301                 }
 302                 *cpp = end + 1 + strspn(end + 1, WHITESPACE);
 303         } else {
 304                 /* Read to end of filename */
 305                 end = strpbrk(cp, WHITESPACE);
 306                 if (end == NULL)
 307                         end = strchr(cp, '\0');
 308                 *cpp = end + strspn(end, WHITESPACE);
 309         }
 310
 311         i = end - cp;
 312
 313         *path = xmalloc(i + 1);
 314         memcpy(*path, cp, i);
 315         (*path)[i] = '\0';
 316         return(0);
 317
 318  fail:
 319         *path = NULL;
 320         return (-1);
 321 }
 322
 323 static int
 324 is_dir(char *path)
 325 {
 326         struct stat sb;
 327
 328         /* XXX: report errors? */
 329         if (stat(path, &sb) == -1)
 330                 return(0);
 331
 332         return(sb.st_mode & S_IFDIR);
 333 }
 334
 335 static int
 336 remote_is_dir(int in, int out, char *path)
 337 {
 338         Attrib *a;
 339
 340         /* XXX: report errors? */
 341         if ((a = do_stat(in, out, path, 1)) == NULL)
 342                 return(0);
 343         if (!(a->flags & SSH2_FILEXFER_ATTR_PERMISSIONS))
 344                 return(0);
 345         return(a->perm & S_IFDIR);
 346 }
 347
 348 static int
 349 process_get(int in, int out, char *src, char *dst, char *pwd, int pflag)
 350 {
 351         char *abs_src = NULL;
 352         char *abs_dst = NULL;
 353         char *tmp;
 354         glob_t g;
 355         int err = 0;
 356         int i;
 357
 358         abs_src = xstrdup(src);
 359         abs_src = make_absolute(abs_src, pwd);
 360
 361         memset(&g, 0, sizeof(g));
 362         debug3("Looking up %s", abs_src);
 363         if (remote_glob(in, out, abs_src, 0, NULL, &g)) {
 364                 error("File \"%s\" not found.", abs_src);
 365                 err = -1;
 366                 goto out;
 367         }
 368
 369         /* Only one match, dst may be file, directory or unspecified */
 370         if (g.gl_pathv[0] && g.gl_matchc == 1) {
 371                 if (dst) {
 372                         /* If directory specified, append filename */
 373                         if (is_dir(dst)) {
 374                                 if (infer_path(g.gl_pathv[0], &tmp)) {
 375                                         err = 1;
 376                                         goto out;
 377                                 }
 378                                 abs_dst = path_append(dst, tmp);
 379                                 xfree(tmp);
 380                         } else
 381                                 abs_dst = xstrdup(dst);
 382                 } else if (infer_path(g.gl_pathv[0], &abs_dst)) {
 383                         err = -1;
 384                         goto out;
 385                 }
 386                 printf("Fetching %s to %s\n", g.gl_pathv[0], abs_dst);
 387                 err = do_download(in, out, g.gl_pathv[0], abs_dst, pflag,
 388                     copy_buffer_len);
 389                 goto out;
 390         }
 391
 392         /* Multiple matches, dst may be directory or unspecified */
 393         if (dst && !is_dir(dst)) {
 394                 error("Multiple files match, but \"%s\" is not a directory",
 395                     dst);
 396                 err = -1;
 397                 goto out;
 398         }
 399
 400         for (i = 0; g.gl_pathv[i]; i++) {
 401                 if (infer_path(g.gl_pathv[i], &tmp)) {
 402                         err = -1;
 403                         goto out;
 404                 }
 405                 if (dst) {
 406                         abs_dst = path_append(dst, tmp);
 407                         xfree(tmp);
 408                 } else
 409                         abs_dst = tmp;
 410
 411                 printf("Fetching %s to %s\n", g.gl_pathv[i], abs_dst);
 412                 if (do_download(in, out, g.gl_pathv[i], abs_dst, pflag,
 413                     copy_buffer_len) == -1)
 414                         err = -1;
 415                 xfree(abs_dst);
 416                 abs_dst = NULL;
 417         }
 418
 419 out:
 420         xfree(abs_src);
 421         if (abs_dst)
 422                 xfree(abs_dst);
 423         globfree(&g);
 424         return(err);
 425 }
 426
 427 static int
 428 process_put(int in, int out, char *src, char *dst, char *pwd, int pflag)
 429 {
 430         char *tmp_dst = NULL;
 431         char *abs_dst = NULL;
 432         char *tmp;
 433         glob_t g;
 434         int err = 0;
 435         int i;
 436
 437         if (dst) {
 438                 tmp_dst = xstrdup(dst);
 439                 tmp_dst = make_absolute(tmp_dst, pwd);
 440         }
 441
 442         memset(&g, 0, sizeof(g));
 443         debug3("Looking up %s", src);
 444         if (glob(src, 0, NULL, &g)) {
 445                 error("File \"%s\" not found.", src);
 446                 err = -1;
 447                 goto out;
 448         }
 449
 450         /* Only one match, dst may be file, directory or unspecified */
 451         if (g.gl_pathv[0] && g.gl_matchc == 1) {
 452                 if (tmp_dst) {
 453                         /* If directory specified, append filename */
 454                         if (remote_is_dir(in, out, tmp_dst)) {
 455                                 if (infer_path(g.gl_pathv[0], &tmp)) {
 456                                         err = 1;
 457                                         goto out;
 458                                 }
 459                                 abs_dst = path_append(tmp_dst, tmp);
 460                                 xfree(tmp);
 461                         } else
 462                                 abs_dst = xstrdup(tmp_dst);
 463                 } else {
 464                         if (infer_path(g.gl_pathv[0], &abs_dst)) {
 465                                 err = -1;
 466                                 goto out;
 467                         }
 468                         abs_dst = make_absolute(abs_dst, pwd);
 469                 }
 470                 printf("Uploading %s to %s\n", g.gl_pathv[0], abs_dst);
 471                 err = do_upload(in, out, g.gl_pathv[0], abs_dst, pflag,
 472                     copy_buffer_len);
 473                 goto out;
 474         }
 475
 476         /* Multiple matches, dst may be directory or unspecified */
 477         if (tmp_dst && !remote_is_dir(in, out, tmp_dst)) {
 478                 error("Multiple files match, but \"%s\" is not a directory",
 479                     tmp_dst);
 480                 err = -1;
 481                 goto out;
 482         }
 483
 484         for (i = 0; g.gl_pathv[i]; i++) {
 485                 if (infer_path(g.gl_pathv[i], &tmp)) {
 486                         err = -1;
 487                         goto out;
 488                 }
 489                 if (tmp_dst) {
 490                         abs_dst = path_append(tmp_dst, tmp);
 491                         xfree(tmp);
 492                 } else
 493                         abs_dst = make_absolute(tmp, pwd);
 494
 495                 printf("Uploading %s to %s\n", g.gl_pathv[i], abs_dst);
 496                 if (do_upload(in, out, g.gl_pathv[i], abs_dst, pflag,
 497                     copy_buffer_len) == -1)
 498                         err = -1;
 499         }
 500
 501 out:
 502         if (abs_dst)
 503                 xfree(abs_dst);
 504         if (tmp_dst)
 505                 xfree(tmp_dst);
 506         return(err);
 507 }
 508
 509 static int
 510 parse_args(const char **cpp, int *pflag, unsigned long *n_arg,
 511     char **path1, char **path2)
 512 {
 513         const char *cmd, *cp = *cpp;
 514         char *cp2;
 515         int base = 0;
 516         long l;
 517         int i, cmdnum;
 518
 519         /* Skip leading whitespace */
 520         cp = cp + strspn(cp, WHITESPACE);
 521
 522         /* Ignore blank lines */
 523         if (!*cp)
 524                 return(-1);
 525
 526         /* Figure out which command we have */
 527         for (i = 0; cmds[i].c; i++) {
 528                 int cmdlen = strlen(cmds[i].c);
 529
 530                 /* Check for command followed by whitespace */
 531                 if (!strncasecmp(cp, cmds[i].c, cmdlen) &&
 532                     strchr(WHITESPACE, cp[cmdlen])) {
 533                         cp += cmdlen;
 534                         cp = cp + strspn(cp, WHITESPACE);
 535                         break;
 536                 }
 537         }
 538         cmdnum = cmds[i].n;
 539         cmd = cmds[i].c;
 540
 541         /* Special case */
 542         if (*cp == '!') {
 543                 cp++;
 544                 cmdnum = I_SHELL;
 545         } else if (cmdnum == -1) {
 546                 error("Invalid command.");
 547                 return(-1);
 548         }
 549
 550         /* Get arguments and parse flags */
 551         *pflag = *n_arg = 0;
 552         *path1 = *path2 = NULL;
 553         switch (cmdnum) {
 554         case I_GET:
 555         case I_PUT:
 556                 if (parse_getput_flags(&cp, pflag))
 557                         return(-1);
 558                 /* Get first pathname (mandatory) */
 559                 if (get_pathname(&cp, path1))
 560                         return(-1);
 561                 if (*path1 == NULL) {
 562                         error("You must specify at least one path after a "
 563                             "%s command.", cmd);
 564                         return(-1);
 565                 }
 566                 /* Try to get second pathname (optional) */
 567                 if (get_pathname(&cp, path2))
 568                         return(-1);
 569                 break;
 570         case I_RENAME:
 571         case I_SYMLINK:
 572                 if (get_pathname(&cp, path1))
 573                         return(-1);
 574                 if (get_pathname(&cp, path2))
 575                         return(-1);
 576                 if (!*path1 || !*path2) {
 577                         error("You must specify two paths after a %s "
 578                             "command.", cmd);
 579                         return(-1);
 580                 }
 581                 break;
 582         case I_RM:
 583         case I_MKDIR:
 584         case I_RMDIR:
 585         case I_CHDIR:
 586         case I_LCHDIR:
 587         case I_LMKDIR:
 588                 /* Get pathname (mandatory) */
 589                 if (get_pathname(&cp, path1))
 590                         return(-1);
 591                 if (*path1 == NULL) {
 592                         error("You must specify a path after a %s command.",
 593                             cmd);
 594                         return(-1);
 595                 }
 596                 break;
 597         case I_LS:
 598                 /* Path is optional */
 599                 if (get_pathname(&cp, path1))
 600                         return(-1);
 601                 break;
 602         case I_LLS:
 603         case I_SHELL:
 604                 /* Uses the rest of the line */
 605                 break;
 606         case I_LUMASK:
 607                 base = 8;
 608         case I_CHMOD:
 609                 base = 8;
 610         case I_CHOWN:
 611         case I_CHGRP:
 612                 /* Get numeric arg (mandatory) */
 613                 l = strtol(cp, &cp2, base);
 614                 if (cp2 == cp || ((l == LONG_MIN || l == LONG_MAX) &&
 615                     errno == ERANGE) || l < 0) {
 616                         error("You must supply a numeric argument "
 617                             "to the %s command.", cmd);
 618                         return(-1);
 619                 }
 620                 cp = cp2;
 621                 *n_arg = l;
 622                 if (cmdnum == I_LUMASK && strchr(WHITESPACE, *cp))
 623                         break;
 624                 if (cmdnum == I_LUMASK || !strchr(WHITESPACE, *cp)) {
 625                         error("You must supply a numeric argument "
 626                             "to the %s command.", cmd);
 627                         return(-1);
 628                 }
 629                 cp += strspn(cp, WHITESPACE);
 630
 631                 /* Get pathname (mandatory) */
 632                 if (get_pathname(&cp, path1))
 633                         return(-1);
 634                 if (*path1 == NULL) {
 635                         error("You must specify a path after a %s command.",
 636                             cmd);
 637                         return(-1);
 638                 }
 639                 break;
 640         case I_QUIT:
 641         case I_PWD:
 642         case I_LPWD:
 643         case I_HELP:
 644         case I_VERSION:
 645                 break;
 646         default:
 647                 fatal("Command not implemented");
 648         }
 649
 650         *cpp = cp;
 651         return(cmdnum);
 652 }
 653
 654 static int
 655 parse_dispatch_command(int in, int out, const char *cmd, char **pwd)
 656 {
 657         char *path1, *path2, *tmp;
 658         int pflag, cmdnum, i;
 659         unsigned long n_arg;
 660         Attrib a, *aa;
 661         char path_buf[MAXPATHLEN];
 662         int err = 0;
 663         glob_t g;
 664
 665         path1 = path2 = NULL;
 666         cmdnum = parse_args(&cmd, &pflag, &n_arg, &path1, &path2);
 667
 668         memset(&g, 0, sizeof(g));
 669
 670         /* Perform command */
 671         switch (cmdnum) {
 672         case -1:
 673                 break;
 674         case I_GET:
 675                 err = process_get(in, out, path1, path2, *pwd, pflag);
 676                 break;
 677         case I_PUT:
 678                 err = process_put(in, out, path1, path2, *pwd, pflag);
 679                 break;
 680         case I_RENAME:
 681                 path1 = make_absolute(path1, *pwd);
 682                 path2 = make_absolute(path2, *pwd);
 683                 err = do_rename(in, out, path1, path2);
 684                 break;
 685         case I_SYMLINK:
 686                 if (version < 3) {
 687                         error("The server (version %d) does not support "
 688                             "this operation", version);
 689                         err = -1;
 690                 } else {
 691                         path2 = make_absolute(path2, *pwd);
 692                         err = do_symlink(in, out, path1, path2);
 693                 }
 694                 break;
 695         case I_RM:
 696                 path1 = make_absolute(path1, *pwd);
 697                 remote_glob(in, out, path1, GLOB_NOCHECK, NULL, &g);
 698                 for (i = 0; g.gl_pathv[i]; i++) {
 699                         printf("Removing %s\n", g.gl_pathv[i]);
 700                         if (do_rm(in, out, g.gl_pathv[i]) == -1)
 701                                 err = -1;
 702                 }
 703                 break;
 704         case I_MKDIR:
 705                 path1 = make_absolute(path1, *pwd);
 706                 attrib_clear(&a);
 707                 a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
 708                 a.perm = 0777;
 709                 err = do_mkdir(in, out, path1, &a);
 710                 break;
 711         case I_RMDIR:
 712                 path1 = make_absolute(path1, *pwd);
 713                 err = do_rmdir(in, out, path1);
 714                 break;
 715         case I_CHDIR:
 716                 path1 = make_absolute(path1, *pwd);
 717                 if ((tmp = do_realpath(in, out, path1)) == NULL) {
 718                         err = 1;
 719                         break;
 720                 }
 721                 if ((aa = do_stat(in, out, tmp, 0)) == NULL) {
 722                         xfree(tmp);
 723                         err = 1;
 724                         break;
 725                 }
 726                 if (!(aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS)) {
 727                         error("Can't change directory: Can't check target");
 728                         xfree(tmp);
 729                         err = 1;
 730                         break;
 731                 }
 732                 if (!S_ISDIR(aa->perm)) {
 733                         error("Can't change directory: \"%s\" is not "
 734                             "a directory", tmp);
 735                         xfree(tmp);
 736                         err = 1;
 737                         break;
 738                 }
 739                 xfree(*pwd);
 740                 *pwd = tmp;
 741                 break;
 742         case I_LS:
 743                 if (!path1) {
 744                         do_ls(in, out, *pwd);
 745                         break;
 746                 }
 747                 path1 = make_absolute(path1, *pwd);
 748                 if ((tmp = do_realpath(in, out, path1)) == NULL)
 749                         break;
 750                 xfree(path1);
 751                 path1 = tmp;
 752                 if ((aa = do_stat(in, out, path1, 0)) == NULL)
 753                         break;
 754                 if ((aa->flags & SSH2_FILEXFER_ATTR_PERMISSIONS) &&
 755                     !S_ISDIR(aa->perm)) {
 756                         error("Can't ls: \"%s\" is not a directory", path1);
 757                         break;
 758                 }
 759                 do_ls(in, out, path1);
 760                 break;
 761         case I_LCHDIR:
 762                 if (chdir(path1) == -1) {
 763                         error("Couldn't change local directory to "
 764                             "\"%s\": %s", path1, strerror(errno));
 765                         err = 1;
 766                 }
 767                 break;
 768         case I_LMKDIR:
 769                 if (mkdir(path1, 0777) == -1) {
 770                         error("Couldn't create local directory "
 771                             "\"%s\": %s", path1, strerror(errno));
 772                         err = 1;
 773                 }
 774                 break;
 775         case I_LLS:
 776                 local_do_ls(cmd);
 777                 break;
 778         case I_SHELL:
 779                 local_do_shell(cmd);
 780                 break;
 781         case I_LUMASK:
 782                 umask(n_arg);
 783                 printf("Local umask: %03lo\n", n_arg);
 784                 break;
 785         case I_CHMOD:
 786                 path1 = make_absolute(path1, *pwd);
 787                 attrib_clear(&a);
 788                 a.flags |= SSH2_FILEXFER_ATTR_PERMISSIONS;
 789                 a.perm = n_arg;
 790                 remote_glob(in, out, path1, GLOB_NOCHECK, NULL, &g);
 791                 for (i = 0; g.gl_pathv[i]; i++) {
 792                         printf("Changing mode on %s\n", g.gl_pathv[i]);
 793                         do_setstat(in, out, g.gl_pathv[i], &a);
 794                 }
 795                 break;
 796         case I_CHOWN:
 797                 path1 = make_absolute(path1, *pwd);
 798                 remote_glob(in, out, path1, GLOB_NOCHECK, NULL, &g);
 799                 for (i = 0; g.gl_pathv[i]; i++) {
 800                         if (!(aa = do_stat(in, out, g.gl_pathv[i], 0)))
 801                                 continue;
 802                         if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
 803                                 error("Can't get current ownership of "
 804                                     "remote file \"%s\"", g.gl_pathv[i]);
 805                                 continue;
 806                         }
 807                         printf("Changing owner on %s\n", g.gl_pathv[i]);
 808                         aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
 809                         aa->uid = n_arg;
 810                         do_setstat(in, out, g.gl_pathv[i], aa);
 811                 }
 812                 break;
 813         case I_CHGRP:
 814                 path1 = make_absolute(path1, *pwd);
 815                 remote_glob(in, out, path1, GLOB_NOCHECK, NULL, &g);
 816                 for (i = 0; g.gl_pathv[i]; i++) {
 817                         if (!(aa = do_stat(in, out, g.gl_pathv[i], 0)))
 818                                 continue;
 819                         if (!(aa->flags & SSH2_FILEXFER_ATTR_UIDGID)) {
 820                                 error("Can't get current ownership of "
 821                                     "remote file \"%s\"", g.gl_pathv[i]);
 822                                 continue;
 823                         }
 824                         printf("Changing group on %s\n", g.gl_pathv[i]);
 825                         aa->flags &= SSH2_FILEXFER_ATTR_UIDGID;
 826                         aa->gid = n_arg;
 827                         do_setstat(in, out, g.gl_pathv[i], aa);
 828                 }
 829                 break;
 830         case I_PWD:
 831                 printf("Remote working directory: %s\n", *pwd);
 832                 break;
 833         case I_LPWD:
 834                 if (!getcwd(path_buf, sizeof(path_buf)))
 835                         error("Couldn't get local cwd: %s",
 836                             strerror(errno));
 837                 else
 838                         printf("Local working directory: %s\n",
 839                             path_buf);
 840                 break;
 841         case I_QUIT:
 842                 return(-1);
 843         case I_HELP:
 844                 help();
 845                 break;
 846         case I_VERSION:
 847                 printf("SFTP protocol version %d\n", version);
 848                 break;
 849         default:
 850                 fatal("%d is not implemented", cmdnum);
 851         }
 852
 853         if (g.gl_pathc)
 854                 globfree(&g);
 855         if (path1)
 856                 xfree(path1);
 857         if (path2)
 858                 xfree(path2);
 859
 860         /* If an error occurs in batch mode we should abort. */
 861         if (infile != stdin && err > 0)
 862                 return -1;
 863
 864         return(0);
 865 }
 866
 867 void
 868 interactive_loop(int fd_in, int fd_out, char *file1, char *file2)
 869 {
 870         char *pwd;
 871         char *dir = NULL;
 872         char cmd[2048];
 873
 874         version = do_init(fd_in, fd_out);
 875         if (version == -1)
 876                 fatal("Couldn't initialise connection to server");
 877
 878         pwd = do_realpath(fd_in, fd_out, ".");
 879         if (pwd == NULL)
 880                 fatal("Need cwd");
 881
 882         if (file1 != NULL) {
 883                 dir = xstrdup(file1);
 884                 dir = make_absolute(dir, pwd);
 885
 886                 if (remote_is_dir(fd_in, fd_out, dir) && file2 == NULL) {
 887                         printf("Changing to: %s\n", dir);
 888                         snprintf(cmd, sizeof cmd, "cd \"%s\"", dir);
 889                         parse_dispatch_command(fd_in, fd_out, cmd, &pwd);
 890                 } else {
 891                         if (file2 == NULL)
 892                                 snprintf(cmd, sizeof cmd, "get %s", dir);
 893                         else
 894                                 snprintf(cmd, sizeof cmd, "get %s %s", dir,
 895                                     file2);
 896
 897                         parse_dispatch_command(fd_in, fd_out, cmd, &pwd);
 898                         return;
 899                 }
 900         }
 901 #if HAVE_SETVBUF
 902         setvbuf(stdout, NULL, _IOLBF, 0);
 903         setvbuf(infile, NULL, _IOLBF, 0);
 904 #else
 905         setlinebuf(stdout);
 906         setlinebuf(infile);
 907 #endif
 908
 909         for (;;) {
 910                 char *cp;
 911
 912                 printf("sftp> ");
 913
 914                 /* XXX: use libedit */
 915                 if (fgets(cmd, sizeof(cmd), infile) == NULL) {
 916                         printf("\n");
 917                         break;
 918                 } else if (infile != stdin) /* Bluff typing */
 919                         printf("%s", cmd);
 920
 921                 cp = strrchr(cmd, '\n');
 922                 if (cp)
 923                         *cp = '\0';
 924
 925                 if (parse_dispatch_command(fd_in, fd_out, cmd, &pwd))
 926                         break;
 927         }
 928         xfree(pwd);
 929 }