Add the `can we run under win3.1' question to the FAQ.
[sgt/putty] / psftp.c
diff --git a/psftp.c b/psftp.c
index fbf730a..ba740ce 100644 (file)
--- a/psftp.c
+++ b/psftp.c
@@ -8,6 +8,7 @@
 #include <stdlib.h>
 #include <stdarg.h>
 #include <assert.h>
 #include <stdlib.h>
 #include <stdarg.h>
 #include <assert.h>
+#include <limits.h>
 
 #define PUTTY_DO_GLOBALS
 #include "putty.h"
 
 #define PUTTY_DO_GLOBALS
 #include "putty.h"
 #include "sftp.h"
 #include "int64.h"
 
 #include "sftp.h"
 #include "int64.h"
 
-/* ----------------------------------------------------------------------
- * String handling routines.
+/*
+ * Since SFTP is a request-response oriented protocol, it requires
+ * no buffer management: when we send data, we stop and wait for an
+ * acknowledgement _anyway_, and so we can't possibly overfill our
+ * send buffer.
  */
 
  */
 
-char *dupstr(char *s)
-{
-    int len = strlen(s);
-    char *p = smalloc(len + 1);
-    strcpy(p, s);
-    return p;
-}
-
-/* Allocate the concatenation of N strings. Terminate arg list with NULL. */
-char *dupcat(char *s1, ...)
-{
-    int len;
-    char *p, *q, *sn;
-    va_list ap;
-
-    len = strlen(s1);
-    va_start(ap, s1);
-    while (1) {
-       sn = va_arg(ap, char *);
-       if (!sn)
-           break;
-       len += strlen(sn);
-    }
-    va_end(ap);
-
-    p = smalloc(len + 1);
-    strcpy(p, s1);
-    q = p + strlen(p);
-
-    va_start(ap, s1);
-    while (1) {
-       sn = va_arg(ap, char *);
-       if (!sn)
-           break;
-       strcpy(q, sn);
-       q += strlen(q);
-    }
-    va_end(ap);
-
-    return p;
-}
+static int psftp_connect(char *userhost, char *user, int portnumber);
+static void do_sftp_init(void);
 
 /* ----------------------------------------------------------------------
  * sftp client state.
 
 /* ----------------------------------------------------------------------
  * sftp client state.
@@ -168,6 +133,30 @@ char *canonify(char *name)
     }
 }
 
     }
 }
 
+/*
+ * Return a pointer to the portion of str that comes after the last
+ * slash (or backslash or colon, if `local' is TRUE).
+ */
+static char *stripslashes(char *str, int local)
+{
+    char *p;
+
+    if (local) {
+        p = strchr(str, ':');
+        if (p) str = p+1;
+    }
+
+    p = strrchr(str, '/');
+    if (p) str = p+1;
+
+    if (local) {
+       p = strrchr(str, '\\');
+       if (p) str = p+1;
+    }
+
+    return str;
+}
+
 /* ----------------------------------------------------------------------
  * Actual sftp commands.
  */
 /* ----------------------------------------------------------------------
  * Actual sftp commands.
  */
@@ -179,13 +168,13 @@ struct sftp_command {
 
 int sftp_cmd_null(struct sftp_command *cmd)
 {
 
 int sftp_cmd_null(struct sftp_command *cmd)
 {
-    return 0;
+    return 1;                         /* success */
 }
 
 int sftp_cmd_unknown(struct sftp_command *cmd)
 {
     printf("psftp: unknown command \"%s\"\n", cmd->words[0]);
 }
 
 int sftp_cmd_unknown(struct sftp_command *cmd)
 {
     printf("psftp: unknown command \"%s\"\n", cmd->words[0]);
-    return 0;
+    return 0;                         /* failure */
 }
 
 int sftp_cmd_quit(struct sftp_command *cmd)
 }
 
 int sftp_cmd_quit(struct sftp_command *cmd)
@@ -212,6 +201,11 @@ int sftp_cmd_ls(struct sftp_command *cmd)
     char *dir, *cdir;
     int i;
 
     char *dir, *cdir;
     int i;
 
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
     if (cmd->nwords < 2)
        dir = ".";
     else
     if (cmd->nwords < 2)
        dir = ".";
     else
@@ -275,7 +269,7 @@ int sftp_cmd_ls(struct sftp_command *cmd)
 
     sfree(cdir);
 
 
     sfree(cdir);
 
-    return 0;
+    return 1;
 }
 
 /*
 }
 
 /*
@@ -287,6 +281,11 @@ int sftp_cmd_cd(struct sftp_command *cmd)
     struct fxp_handle *dirh;
     char *dir;
 
     struct fxp_handle *dirh;
     char *dir;
 
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
     if (cmd->nwords < 2)
        dir = dupstr(homedir);
     else
     if (cmd->nwords < 2)
        dir = dupstr(homedir);
     else
@@ -310,18 +309,41 @@ int sftp_cmd_cd(struct sftp_command *cmd)
     pwd = dir;
     printf("Remote directory is now %s\n", pwd);
 
     pwd = dir;
     printf("Remote directory is now %s\n", pwd);
 
-    return 0;
+    return 1;
 }
 
 /*
 }
 
 /*
- * Get a file and save it at the local end.
+ * Print current directory. Easy as pie.
  */
  */
-int sftp_cmd_get(struct sftp_command *cmd)
+int sftp_cmd_pwd(struct sftp_command *cmd)
+{
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
+    printf("Remote directory is %s\n", pwd);
+    return 1;
+}
+
+/*
+ * Get a file and save it at the local end. We have two very
+ * similar commands here: `get' and `reget', which differ in that
+ * `reget' checks for the existence of the destination file and
+ * starts from where a previous aborted transfer left off.
+ */
+int sftp_general_get(struct sftp_command *cmd, int restart)
 {
     struct fxp_handle *fh;
     char *fname, *outfname;
     uint64 offset;
     FILE *fp;
 {
     struct fxp_handle *fh;
     char *fname, *outfname;
     uint64 offset;
     FILE *fp;
+    int ret;
+
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
 
     if (cmd->nwords < 2) {
        printf("get: expects a filename\n");
 
     if (cmd->nwords < 2) {
        printf("get: expects a filename\n");
@@ -333,7 +355,8 @@ int sftp_cmd_get(struct sftp_command *cmd)
        printf("%s: %s\n", cmd->words[1], fxp_error());
        return 0;
     }
        printf("%s: %s\n", cmd->words[1], fxp_error());
        return 0;
     }
-    outfname = (cmd->nwords == 2 ? cmd->words[1] : cmd->words[2]);
+    outfname = (cmd->nwords == 2 ?
+               stripslashes(cmd->words[1], 0) : cmd->words[2]);
 
     fh = fxp_open(fname, SSH_FXF_READ);
     if (!fh) {
 
     fh = fxp_open(fname, SSH_FXF_READ);
     if (!fh) {
@@ -341,7 +364,13 @@ int sftp_cmd_get(struct sftp_command *cmd)
        sfree(fname);
        return 0;
     }
        sfree(fname);
        return 0;
     }
-    fp = fopen(outfname, "wb");
+
+    if (restart) {
+       fp = fopen(outfname, "rb+");
+    } else {
+       fp = fopen(outfname, "wb");
+    }
+
     if (!fp) {
        printf("local: unable to open %s\n", outfname);
        fxp_close(fh);
     if (!fp) {
        printf("local: unable to open %s\n", outfname);
        fxp_close(fh);
@@ -349,14 +378,23 @@ int sftp_cmd_get(struct sftp_command *cmd)
        return 0;
     }
 
        return 0;
     }
 
-    printf("remote:%s => local:%s\n", fname, outfname);
+    if (restart) {
+       long posn;
+       fseek(fp, 0L, SEEK_END);
+       posn = ftell(fp);
+       printf("reget: restarting at file position %ld\n", posn);
+       offset = uint64_make(0, posn);
+    } else {
+       offset = uint64_make(0, 0);
+    }
 
 
-    offset = uint64_make(0, 0);
+    printf("remote:%s => local:%s\n", fname, outfname);
 
     /*
      * FIXME: we can use FXP_FSTAT here to get the file size, and
      * thus put up a progress bar.
      */
 
     /*
      * FIXME: we can use FXP_FSTAT here to get the file size, and
      * thus put up a progress bar.
      */
+    ret = 1;
     while (1) {
        char buffer[4096];
        int len;
     while (1) {
        char buffer[4096];
        int len;
@@ -367,6 +405,7 @@ int sftp_cmd_get(struct sftp_command *cmd)
            break;
        if (len == -1) {
            printf("error while reading: %s\n", fxp_error());
            break;
        if (len == -1) {
            printf("error while reading: %s\n", fxp_error());
+           ret = 0;
            break;
        }
 
            break;
        }
 
@@ -375,12 +414,15 @@ int sftp_cmd_get(struct sftp_command *cmd)
            wlen = fwrite(buffer, 1, len - wpos, fp);
            if (wlen <= 0) {
                printf("error while writing local file\n");
            wlen = fwrite(buffer, 1, len - wpos, fp);
            if (wlen <= 0) {
                printf("error while writing local file\n");
+               ret = 0;
                break;
            }
            wpos += wlen;
        }
                break;
            }
            wpos += wlen;
        }
-       if (wpos < len)                /* we had an error */
+       if (wpos < len) {              /* we had an error */
+           ret = 0;
            break;
            break;
+       }
        offset = uint64_add32(offset, len);
     }
 
        offset = uint64_add32(offset, len);
     }
 
@@ -388,18 +430,35 @@ int sftp_cmd_get(struct sftp_command *cmd)
     fxp_close(fh);
     sfree(fname);
 
     fxp_close(fh);
     sfree(fname);
 
-    return 0;
+    return ret;
+}
+int sftp_cmd_get(struct sftp_command *cmd)
+{
+    return sftp_general_get(cmd, 0);
+}
+int sftp_cmd_reget(struct sftp_command *cmd)
+{
+    return sftp_general_get(cmd, 1);
 }
 
 /*
 }
 
 /*
- * Send a file and store it at the remote end.
+ * Send a file and store it at the remote end. We have two very
+ * similar commands here: `put' and `reput', which differ in that
+ * `reput' checks for the existence of the destination file and
+ * starts from where a previous aborted transfer left off.
  */
  */
-int sftp_cmd_put(struct sftp_command *cmd)
+int sftp_general_put(struct sftp_command *cmd, int restart)
 {
     struct fxp_handle *fh;
     char *fname, *origoutfname, *outfname;
     uint64 offset;
     FILE *fp;
 {
     struct fxp_handle *fh;
     char *fname, *origoutfname, *outfname;
     uint64 offset;
     FILE *fp;
+    int ret;
+
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
 
     if (cmd->nwords < 2) {
        printf("put: expects a filename\n");
 
     if (cmd->nwords < 2) {
        printf("put: expects a filename\n");
@@ -407,7 +466,8 @@ int sftp_cmd_put(struct sftp_command *cmd)
     }
 
     fname = cmd->words[1];
     }
 
     fname = cmd->words[1];
-    origoutfname = (cmd->nwords == 2 ? cmd->words[1] : cmd->words[2]);
+    origoutfname = (cmd->nwords == 2 ?
+                   stripslashes(cmd->words[1], 1) : cmd->words[2]);
     outfname = canonify(origoutfname);
     if (!outfname) {
        printf("%s: %s\n", origoutfname, fxp_error());
     outfname = canonify(origoutfname);
     if (!outfname) {
        printf("%s: %s\n", origoutfname, fxp_error());
@@ -420,21 +480,53 @@ int sftp_cmd_put(struct sftp_command *cmd)
        sfree(outfname);
        return 0;
     }
        sfree(outfname);
        return 0;
     }
-    fh = fxp_open(outfname, SSH_FXF_WRITE | SSH_FXF_CREAT | SSH_FXF_TRUNC);
+    if (restart) {
+       fh = fxp_open(outfname,
+                     SSH_FXF_WRITE);
+    } else {
+       fh = fxp_open(outfname,
+                     SSH_FXF_WRITE | SSH_FXF_CREAT | SSH_FXF_TRUNC);
+    }
     if (!fh) {
        printf("%s: %s\n", outfname, fxp_error());
        sfree(outfname);
        return 0;
     }
 
     if (!fh) {
        printf("%s: %s\n", outfname, fxp_error());
        sfree(outfname);
        return 0;
     }
 
-    printf("local:%s => remote:%s\n", fname, outfname);
+    if (restart) {
+       char decbuf[30];
+       struct fxp_attrs attrs;
+       if (!fxp_fstat(fh, &attrs)) {
+           printf("read size of %s: %s\n", outfname, fxp_error());
+           sfree(outfname);
+           return 0;
+       }
+       if (!(attrs.flags & SSH_FILEXFER_ATTR_SIZE)) {
+           printf("read size of %s: size was not given\n", outfname);
+           sfree(outfname);
+           return 0;
+       }
+       offset = attrs.size;
+       uint64_decimal(offset, decbuf);
+       printf("reput: restarting at file position %s\n", decbuf);
+       if (uint64_compare(offset, uint64_make(0, LONG_MAX)) > 0) {
+           printf("reput: remote file is larger than we can deal with\n");
+           sfree(outfname);
+           return 0;
+       }
+       if (fseek(fp, offset.lo, SEEK_SET) != 0)
+           fseek(fp, 0, SEEK_END);    /* *shrug* */
+    } else {
+       offset = uint64_make(0, 0);
+    }
 
 
-    offset = uint64_make(0, 0);
+    printf("local:%s => remote:%s\n", fname, outfname);
 
     /*
      * FIXME: we can use FXP_FSTAT here to get the file size, and
      * thus put up a progress bar.
      */
 
     /*
      * FIXME: we can use FXP_FSTAT here to get the file size, and
      * thus put up a progress bar.
      */
+    ret = 1;
     while (1) {
        char buffer[4096];
        int len;
     while (1) {
        char buffer[4096];
        int len;
@@ -442,12 +534,14 @@ int sftp_cmd_put(struct sftp_command *cmd)
        len = fread(buffer, 1, sizeof(buffer), fp);
        if (len == -1) {
            printf("error while reading local file\n");
        len = fread(buffer, 1, sizeof(buffer), fp);
        if (len == -1) {
            printf("error while reading local file\n");
+           ret = 0;
            break;
        } else if (len == 0) {
            break;
        }
        if (!fxp_write(fh, buffer, offset, len)) {
            printf("error while writing: %s\n", fxp_error());
            break;
        } else if (len == 0) {
            break;
        }
        if (!fxp_write(fh, buffer, offset, len)) {
            printf("error while writing: %s\n", fxp_error());
+           ret = 0;
            break;
        }
        offset = uint64_add32(offset, len);
            break;
        }
        offset = uint64_add32(offset, len);
@@ -457,11 +551,438 @@ int sftp_cmd_put(struct sftp_command *cmd)
     fclose(fp);
     sfree(outfname);
 
     fclose(fp);
     sfree(outfname);
 
-    return 0;
+    return ret;
+}
+int sftp_cmd_put(struct sftp_command *cmd)
+{
+    return sftp_general_put(cmd, 0);
+}
+int sftp_cmd_reput(struct sftp_command *cmd)
+{
+    return sftp_general_put(cmd, 1);
 }
 
 }
 
+int sftp_cmd_mkdir(struct sftp_command *cmd)
+{
+    char *dir;
+    int result;
+
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
+    if (cmd->nwords < 2) {
+       printf("mkdir: expects a directory\n");
+       return 0;
+    }
+
+    dir = canonify(cmd->words[1]);
+    if (!dir) {
+       printf("%s: %s\n", dir, fxp_error());
+       return 0;
+    }
+
+    result = fxp_mkdir(dir);
+    if (!result) {
+       printf("mkdir %s: %s\n", dir, fxp_error());
+       sfree(dir);
+       return 0;
+    }
+
+    sfree(dir);
+    return 1;
+}
+
+int sftp_cmd_rmdir(struct sftp_command *cmd)
+{
+    char *dir;
+    int result;
+
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
+    if (cmd->nwords < 2) {
+       printf("rmdir: expects a directory\n");
+       return 0;
+    }
+
+    dir = canonify(cmd->words[1]);
+    if (!dir) {
+       printf("%s: %s\n", dir, fxp_error());
+       return 0;
+    }
+
+    result = fxp_rmdir(dir);
+    if (!result) {
+       printf("rmdir %s: %s\n", dir, fxp_error());
+       sfree(dir);
+       return 0;
+    }
+
+    sfree(dir);
+    return 1;
+}
+
+int sftp_cmd_rm(struct sftp_command *cmd)
+{
+    char *fname;
+    int result;
+
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
+    if (cmd->nwords < 2) {
+       printf("rm: expects a filename\n");
+       return 0;
+    }
+
+    fname = canonify(cmd->words[1]);
+    if (!fname) {
+       printf("%s: %s\n", fname, fxp_error());
+       return 0;
+    }
+
+    result = fxp_remove(fname);
+    if (!result) {
+       printf("rm %s: %s\n", fname, fxp_error());
+       sfree(fname);
+       return 0;
+    }
+
+    sfree(fname);
+    return 1;
+}
+
+int sftp_cmd_mv(struct sftp_command *cmd)
+{
+    char *srcfname, *dstfname;
+    int result;
+
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
+    if (cmd->nwords < 3) {
+       printf("mv: expects two filenames\n");
+       return 0;
+    }
+    srcfname = canonify(cmd->words[1]);
+    if (!srcfname) {
+       printf("%s: %s\n", srcfname, fxp_error());
+       return 0;
+    }
+
+    dstfname = canonify(cmd->words[2]);
+    if (!dstfname) {
+       printf("%s: %s\n", dstfname, fxp_error());
+       return 0;
+    }
+
+    result = fxp_rename(srcfname, dstfname);
+    if (!result) {
+       char const *error = fxp_error();
+       struct fxp_attrs attrs;
+
+       /*
+        * The move might have failed because dstfname pointed at a
+        * directory. We check this possibility now: if dstfname
+        * _is_ a directory, we re-attempt the move by appending
+        * the basename of srcfname to dstfname.
+        */
+       result = fxp_stat(dstfname, &attrs);
+       if (result &&
+           (attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS) &&
+           (attrs.permissions & 0040000)) {
+           char *p;
+           char *newname, *newcanon;
+           printf("(destination %s is a directory)\n", dstfname);
+           p = srcfname + strlen(srcfname);
+           while (p > srcfname && p[-1] != '/') p--;
+           newname = dupcat(dstfname, "/", p, NULL);
+           newcanon = canonify(newname);
+           sfree(newname);
+           if (newcanon) {
+               sfree(dstfname);
+               dstfname = newcanon;
+               result = fxp_rename(srcfname, dstfname);
+               error = result ? NULL : fxp_error();
+           }
+       }
+       if (error) {
+           printf("mv %s %s: %s\n", srcfname, dstfname, error);
+           sfree(srcfname);
+           sfree(dstfname);
+           return 0;
+       }
+    }
+    printf("%s -> %s\n", srcfname, dstfname);
+
+    sfree(srcfname);
+    sfree(dstfname);
+    return 1;
+}
+
+int sftp_cmd_chmod(struct sftp_command *cmd)
+{
+    char *fname, *mode;
+    int result;
+    struct fxp_attrs attrs;
+    unsigned attrs_clr, attrs_xor, oldperms, newperms;
+
+    if (back == NULL) {
+       printf("psftp: not connected to a host; use \"open host.name\"\n");
+       return 0;
+    }
+
+    if (cmd->nwords < 3) {
+       printf("chmod: expects a mode specifier and a filename\n");
+       return 0;
+    }
+
+    /*
+     * Attempt to parse the mode specifier in cmd->words[1]. We
+     * don't support the full horror of Unix chmod; instead we
+     * support a much simpler syntax in which the user can either
+     * specify an octal number, or a comma-separated sequence of
+     * [ugoa]*[-+=][rwxst]+. (The initial [ugoa] sequence may
+     * _only_ be omitted if the only attribute mentioned is t,
+     * since all others require a user/group/other specification.
+     * Additionally, the s attribute may not be specified for any
+     * [ugoa] specifications other than exactly u or exactly g.
+     */
+    attrs_clr = attrs_xor = 0;
+    mode = cmd->words[1];
+    if (mode[0] >= '0' && mode[0] <= '9') {
+       if (mode[strspn(mode, "01234567")]) {
+           printf("chmod: numeric file modes should"
+                  " contain digits 0-7 only\n");
+           return 0;
+       }
+       attrs_clr = 07777;
+       sscanf(mode, "%o", &attrs_xor);
+       attrs_xor &= attrs_clr;
+    } else {
+       while (*mode) {
+           char *modebegin = mode;
+           unsigned subset, perms;
+           int action;
+
+           subset = 0;
+           while (*mode && *mode != ',' &&
+                  *mode != '+' && *mode != '-' && *mode != '=') {
+               switch (*mode) {
+                 case 'u': subset |= 04700; break; /* setuid, user perms */
+                 case 'g': subset |= 02070; break; /* setgid, group perms */
+                 case 'o': subset |= 00007; break; /* just other perms */
+                 case 'a': subset |= 06777; break; /* all of the above */
+                 default:
+                   printf("chmod: file mode '%.*s' contains unrecognised"
+                          " user/group/other specifier '%c'\n",
+                          strcspn(modebegin, ","), modebegin, *mode);
+                   return 0;
+               }
+               mode++;
+           }
+           if (!*mode || *mode == ',') {
+               printf("chmod: file mode '%.*s' is incomplete\n",
+                      strcspn(modebegin, ","), modebegin);
+               return 0;
+           }
+           action = *mode++;
+           if (!*mode || *mode == ',') {
+               printf("chmod: file mode '%.*s' is incomplete\n",
+                      strcspn(modebegin, ","), modebegin);
+               return 0;
+           }
+           perms = 0;
+           while (*mode && *mode != ',') {
+               switch (*mode) {
+                 case 'r': perms |= 00444; break;
+                 case 'w': perms |= 00222; break;
+                 case 'x': perms |= 00111; break;
+                 case 't': perms |= 01000; subset |= 01000; break;
+                 case 's':
+                   if ((subset & 06777) != 04700 &&
+                       (subset & 06777) != 02070) {
+                       printf("chmod: file mode '%.*s': set[ug]id bit should"
+                              " be used with exactly one of u or g only\n",
+                              strcspn(modebegin, ","), modebegin);
+                       return 0;
+                   }
+                   perms |= 06000;
+                   break;
+                 default:
+                   printf("chmod: file mode '%.*s' contains unrecognised"
+                          " permission specifier '%c'\n",
+                          strcspn(modebegin, ","), modebegin, *mode);
+                   return 0;
+               }
+               mode++;
+           }
+           if (!(subset & 06777) && (perms &~ subset)) {
+               printf("chmod: file mode '%.*s' contains no user/group/other"
+                      " specifier and permissions other than 't' \n",
+                      strcspn(modebegin, ","), modebegin, *mode);
+               return 0;
+           }
+           perms &= subset;
+           switch (action) {
+             case '+':
+               attrs_clr |= perms;
+               attrs_xor |= perms;
+               break;
+             case '-':
+               attrs_clr |= perms;
+               attrs_xor &= ~perms;
+               break;
+             case '=':
+               attrs_clr |= subset;
+               attrs_xor |= perms;
+               break;
+           }
+           if (*mode) mode++;         /* eat comma */
+       }
+    }
+
+    fname = canonify(cmd->words[2]);
+    if (!fname) {
+       printf("%s: %s\n", fname, fxp_error());
+       return 0;
+    }
+
+    result = fxp_stat(fname, &attrs);
+    if (!result || !(attrs.flags & SSH_FILEXFER_ATTR_PERMISSIONS)) {
+       printf("get attrs for %s: %s\n", fname,
+              result ? "file permissions not provided" : fxp_error());
+       sfree(fname);
+       return 0;
+    }
+
+    attrs.flags = SSH_FILEXFER_ATTR_PERMISSIONS;   /* perms _only_ */
+    oldperms = attrs.permissions & 07777;
+    attrs.permissions &= ~attrs_clr;
+    attrs.permissions ^= attrs_xor;
+    newperms = attrs.permissions & 07777;
+
+    result = fxp_setstat(fname, attrs);
+
+    if (!result) {
+       printf("set attrs for %s: %s\n", fname, fxp_error());
+       sfree(fname);
+       return 0;
+    }
+
+    printf("%s: %04o -> %04o\n", fname, oldperms, newperms);
+
+    sfree(fname);
+    return 1;
+}
+
+static int sftp_cmd_open(struct sftp_command *cmd)
+{
+    if (back != NULL) {
+       printf("psftp: already connected\n");
+       return 0;
+    }
+
+    if (cmd->nwords < 2) {
+       printf("open: expects a host name\n");
+       return 0;
+    }
+
+    if (psftp_connect(cmd->words[1], NULL, 0)) {
+       back = NULL;                   /* connection is already closed */
+       return -1;                     /* this is fatal */
+    }
+    do_sftp_init();
+    return 1;
+}
+
+static int sftp_cmd_lcd(struct sftp_command *cmd)
+{
+    char *currdir;
+    int len;
+
+    if (cmd->nwords < 2) {
+       printf("lcd: expects a local directory name\n");
+       return 0;
+    }
+
+    if (!SetCurrentDirectory(cmd->words[1])) {
+       LPVOID message;
+       int i;
+       FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
+                     FORMAT_MESSAGE_FROM_SYSTEM |
+                     FORMAT_MESSAGE_IGNORE_INSERTS,
+                     NULL, GetLastError(),
+                     MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+                     (LPTSTR)&message, 0, NULL);
+       i = strcspn((char *)message, "\n");
+       printf("lcd: unable to change directory: %.*s\n", i, (LPCTSTR)message);
+       LocalFree(message);
+       return 0;
+    }
+
+    currdir = smalloc(256);
+    len = GetCurrentDirectory(256, currdir);
+    if (len > 256)
+       currdir = srealloc(currdir, len);
+    GetCurrentDirectory(len, currdir);
+    printf("New local directory is %s\n", currdir);
+    sfree(currdir);
+
+    return 1;
+}
+
+static int sftp_cmd_lpwd(struct sftp_command *cmd)
+{
+    char *currdir;
+    int len;
+
+    currdir = smalloc(256);
+    len = GetCurrentDirectory(256, currdir);
+    if (len > 256)
+       currdir = srealloc(currdir, len);
+    GetCurrentDirectory(len, currdir);
+    printf("Current local directory is %s\n", currdir);
+    sfree(currdir);
+
+    return 1;
+}
+
+static int sftp_cmd_pling(struct sftp_command *cmd)
+{
+    int exitcode;
+
+    exitcode = system(cmd->words[1]);
+    return (exitcode == 0);
+}
+
+static int sftp_cmd_help(struct sftp_command *cmd);
+
 static struct sftp_cmd_lookup {
     char *name;
 static struct sftp_cmd_lookup {
     char *name;
+    /*
+     * For help purposes, there are two kinds of command:
+     * 
+     *  - primary commands, in which `longhelp' is non-NULL. In
+     *    this case `shorthelp' is descriptive text, and `longhelp'
+     *    is longer descriptive text intended to be printed after
+     *    the command name.
+     * 
+     *  - alias commands, in which `longhelp' is NULL. In this case
+     *    `shorthelp' is the name of a primary command, which
+     *    contains the help that should double up for this command.
+     */
+    int listed;                               /* do we list this in primary help? */
+    char *shorthelp;
+    char *longhelp;
     int (*obey) (struct sftp_command *);
 } sftp_lookup[] = {
     /*
     int (*obey) (struct sftp_command *);
 } sftp_lookup[] = {
     /*
@@ -469,19 +990,253 @@ static struct sftp_cmd_lookup {
      * in ASCII order.
      */
     {
      * in ASCII order.
      */
     {
-    "bye", sftp_cmd_quit}, {
-    "cd", sftp_cmd_cd}, {
-    "dir", sftp_cmd_ls}, {
-    "exit", sftp_cmd_quit}, {
-    "get", sftp_cmd_get}, {
-    "ls", sftp_cmd_ls}, {
-    "put", sftp_cmd_put}, {
-"quit", sftp_cmd_quit},};
+       "!", TRUE, "run a local Windows command",
+           "<command>\n"
+           "  Runs a local Windows command. For example, \"!del myfile\".\n",
+           sftp_cmd_pling
+    },
+    {
+       "bye", TRUE, "finish your SFTP session",
+           "\n"
+           "  Terminates your SFTP session and quits the PSFTP program.\n",
+           sftp_cmd_quit
+    },
+    {
+       "cd", TRUE, "change your remote working directory",
+           " [ <New working directory> ]\n"
+           "  Change the remote working directory for your SFTP session.\n"
+           "  If a new working directory is not supplied, you will be\n"
+           "  returned to your home directory.\n",
+           sftp_cmd_cd
+    },
+    {
+       "chmod", TRUE, "change file permissions and modes",
+           " ( <octal-digits> | <modifiers> ) <filename>\n"
+           "  Change the file permissions on a file or directory.\n"
+           "  <octal-digits> can be any octal Unix permission specifier.\n"
+           "  Alternatively, <modifiers> can include:\n"
+           "    u+r     make file readable by owning user\n"
+           "    u+w     make file writable by owning user\n"
+           "    u+x     make file executable by owning user\n"
+           "    u-r     make file not readable by owning user\n"
+           "    [also u-w, u-x]\n"
+           "    g+r     make file readable by members of owning group\n"
+           "    [also g+w, g+x, g-r, g-w, g-x]\n"
+           "    o+r     make file readable by all other users\n"
+           "    [also o+w, o+x, o-r, o-w, o-x]\n"
+           "    a+r     make file readable by absolutely everybody\n"
+           "    [also a+w, a+x, a-r, a-w, a-x]\n"
+           "    u+s     enable the Unix set-user-ID bit\n"
+           "    u-s     disable the Unix set-user-ID bit\n"
+           "    g+s     enable the Unix set-group-ID bit\n"
+           "    g-s     disable the Unix set-group-ID bit\n"
+           "    +t      enable the Unix \"sticky bit\"\n"
+           "  You can give more than one modifier for the same user (\"g-rwx\"), and\n"
+           "  more than one user for the same modifier (\"ug+w\"). You can\n"
+           "  use commas to separate different modifiers (\"u+rwx,g+s\").\n",
+           sftp_cmd_chmod
+    },
+    {
+       "del", TRUE, "delete a file",
+           " <filename>\n"
+           "  Delete a file.\n",
+           sftp_cmd_rm
+    },
+    {
+       "delete", FALSE, "del", NULL, sftp_cmd_rm
+    },
+    {
+       "dir", TRUE, "list contents of a remote directory",
+           " [ <directory-name> ]\n"
+           "  List the contents of a specified directory on the server.\n"
+           "  If <directory-name> is not given, the current working directory\n"
+           "  will be listed.\n",
+           sftp_cmd_ls
+    },
+    {
+       "exit", TRUE, "bye", NULL, sftp_cmd_quit
+    },
+    {
+       "get", TRUE, "download a file from the server to your local machine",
+           " <filename> [ <local-filename> ]\n"
+           "  Downloads a file on the server and stores it locally under\n"
+           "  the same name, or under a different one if you supply the\n"
+           "  argument <local-filename>.\n",
+           sftp_cmd_get
+    },
+    {
+       "help", TRUE, "give help",
+           " [ <command> [ <command> ... ] ]\n"
+           "  Give general help if no commands are specified.\n"
+           "  If one or more commands are specified, give specific help on\n"
+           "  those particular commands.\n",
+           sftp_cmd_help
+    },
+    {
+       "lcd", TRUE, "change local working directory",
+           " <local-directory-name>\n"
+           "  Change the local working directory of the PSFTP program (the\n"
+           "  default location where the \"get\" command will save files).\n",
+           sftp_cmd_lcd
+    },
+    {
+       "lpwd", TRUE, "print local working directory",
+           "\n"
+           "  Print the local working directory of the PSFTP program (the\n"
+           "  default location where the \"get\" command will save files).\n",
+           sftp_cmd_lpwd
+    },
+    {
+       "ls", TRUE, "dir", NULL,
+           sftp_cmd_ls
+    },
+    {
+       "mkdir", TRUE, "create a directory on the remote server",
+           " <directory-name>\n"
+           "  Creates a directory with the given name on the server.\n",
+           sftp_cmd_mkdir
+    },
+    {
+       "mv", TRUE, "move or rename a file on the remote server",
+           " <source-filename> <destination-filename>\n"
+           "  Moves or renames the file <source-filename> on the server,\n"
+           "  so that it is accessible under the name <destination-filename>.\n",
+           sftp_cmd_mv
+    },
+    {
+       "put", TRUE, "upload a file from your local machine to the server",
+           " <filename> [ <remote-filename> ]\n"
+           "  Uploads a file to the server and stores it there under\n"
+           "  the same name, or under a different one if you supply the\n"
+           "  argument <remote-filename>.\n",
+           sftp_cmd_put
+    },
+    {
+       "open", TRUE, "connect to a host",
+           " [<user>@]<hostname>\n"
+           "  Establishes an SFTP connection to a given host. Only usable\n"
+           "  when you did not already specify a host name on the command\n"
+           "  line.\n",
+           sftp_cmd_open
+    },
+    {
+       "pwd", TRUE, "print your remote working directory",
+           "\n"
+           "  Print the current remote working directory for your SFTP session.\n",
+           sftp_cmd_pwd
+    },
+    {
+       "quit", TRUE, "bye", NULL,
+           sftp_cmd_quit
+    },
+    {
+       "reget", TRUE, "continue downloading a file",
+           " <filename> [ <local-filename> ]\n"
+           "  Works exactly like the \"get\" command, but the local file\n"
+           "  must already exist. The download will begin at the end of the\n"
+           "  file. This is for resuming a download that was interrupted.\n",
+           sftp_cmd_reget
+    },
+    {
+       "ren", TRUE, "mv", NULL,
+           sftp_cmd_mv
+    },
+    {
+       "rename", FALSE, "mv", NULL,
+           sftp_cmd_mv
+    },
+    {
+       "reput", TRUE, "continue uploading a file",
+           " <filename> [ <remote-filename> ]\n"
+           "  Works exactly like the \"put\" command, but the remote file\n"
+           "  must already exist. The upload will begin at the end of the\n"
+           "  file. This is for resuming an upload that was interrupted.\n",
+           sftp_cmd_reput
+    },
+    {
+       "rm", TRUE, "del", NULL,
+           sftp_cmd_rm
+    },
+    {
+       "rmdir", TRUE, "remove a directory on the remote server",
+           " <directory-name>\n"
+           "  Removes the directory with the given name on the server.\n"
+           "  The directory will not be removed unless it is empty.\n",
+           sftp_cmd_rmdir
+    }
+};
+
+const struct sftp_cmd_lookup *lookup_command(char *name)
+{
+    int i, j, k, cmp;
+
+    i = -1;
+    j = sizeof(sftp_lookup) / sizeof(*sftp_lookup);
+    while (j - i > 1) {
+       k = (j + i) / 2;
+       cmp = strcmp(name, sftp_lookup[k].name);
+       if (cmp < 0)
+           j = k;
+       else if (cmp > 0)
+           i = k;
+       else {
+           return &sftp_lookup[k];
+       }
+    }
+    return NULL;
+}
+
+static int sftp_cmd_help(struct sftp_command *cmd)
+{
+    int i;
+    if (cmd->nwords == 1) {
+       /*
+        * Give short help on each command.
+        */
+       int maxlen;
+       maxlen = 0;
+       for (i = 0; i < sizeof(sftp_lookup) / sizeof(*sftp_lookup); i++) {
+           int len;
+           if (!sftp_lookup[i].listed)
+               continue;
+           len = strlen(sftp_lookup[i].name);
+           if (maxlen < len)
+               maxlen = len;
+       }
+       for (i = 0; i < sizeof(sftp_lookup) / sizeof(*sftp_lookup); i++) {
+           const struct sftp_cmd_lookup *lookup;
+           if (!sftp_lookup[i].listed)
+               continue;
+           lookup = &sftp_lookup[i];
+           printf("%-*s", maxlen+2, lookup->name);
+           if (lookup->longhelp == NULL)
+               lookup = lookup_command(lookup->shorthelp);
+           printf("%s\n", lookup->shorthelp);
+       }
+    } else {
+       /*
+        * Give long help on specific commands.
+        */
+       for (i = 1; i < cmd->nwords; i++) {
+           const struct sftp_cmd_lookup *lookup;
+           lookup = lookup_command(cmd->words[i]);
+           if (!lookup) {
+               printf("help: %s: command not found\n", cmd->words[i]);
+           } else {
+               printf("%s", lookup->name);
+               if (lookup->longhelp == NULL)
+                   lookup = lookup_command(lookup->shorthelp);
+               printf("%s", lookup->longhelp);
+           }
+       }
+    }
+    return 1;
+}
 
 /* ----------------------------------------------------------------------
  * Command line reading and parsing.
  */
 
 /* ----------------------------------------------------------------------
  * Command line reading and parsing.
  */
-struct sftp_command *sftp_getcmd(void)
+struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
 {
     char *line;
     int linelen, linesize;
 {
     char *line;
     int linelen, linesize;
@@ -489,7 +1244,9 @@ struct sftp_command *sftp_getcmd(void)
     char *p, *q, *r;
     int quoting;
 
     char *p, *q, *r;
     int quoting;
 
-    printf("psftp> ");
+    if ((mode == 0) || (modeflags & 1)) {
+       printf("psftp> ");
+    }
     fflush(stdout);
 
     cmd = smalloc(sizeof(struct sftp_command));
     fflush(stdout);
 
     cmd = smalloc(sizeof(struct sftp_command));
@@ -505,11 +1262,12 @@ struct sftp_command *sftp_getcmd(void)
 
        linesize += 512;
        line = srealloc(line, linesize);
 
        linesize += 512;
        line = srealloc(line, linesize);
-       ret = fgets(line + linelen, linesize - linelen, stdin);
+       ret = fgets(line + linelen, linesize - linelen, fp);
 
        if (!ret || (linelen == 0 && line[0] == '\0')) {
            cmd->obey = sftp_cmd_quit;
 
        if (!ret || (linelen == 0 && line[0] == '\0')) {
            cmd->obey = sftp_cmd_quit;
-           printf("quit\n");
+           if ((mode == 0) || (modeflags & 1))
+               printf("quit\n");
            return cmd;                /* eof */
        }
        len = linelen + strlen(line + linelen);
            return cmd;                /* eof */
        }
        len = linelen + strlen(line + linelen);
@@ -520,50 +1278,69 @@ struct sftp_command *sftp_getcmd(void)
            break;
        }
     }
            break;
        }
     }
+    if (modeflags & 1) {
+       printf("%s\n", line);
+    }
 
 
-    /*
-     * Parse the command line into words. The syntax is:
-     *  - double quotes are removed, but cause spaces within to be
-     *    treated as non-separating.
-     *  - a double-doublequote pair is a literal double quote, inside
-     *    _or_ outside quotes. Like this:
-     * 
-     *      firstword "second word" "this has ""quotes"" in" sodoes""this""
-     * 
-     * becomes
-     * 
-     *      >firstword<
-     *      >second word<
-     *      >this has "quotes" in<
-     *      >sodoes"this"<
-     */
     p = line;
     p = line;
-    while (*p) {
-       /* skip whitespace */
-       while (*p && (*p == ' ' || *p == '\t'))
-           p++;
-       /* mark start of word */
-       q = r = p;                     /* q sits at start, r writes word */
-       quoting = 0;
+    while (*p && (*p == ' ' || *p == '\t'))
+       p++;
+
+    if (*p == '!') {
+       /*
+        * Special case: the ! command. This is always parsed as
+        * exactly two words: one containing the !, and the second
+        * containing everything else on the line.
+        */
+       cmd->nwords = cmd->wordssize = 2;
+       cmd->words = srealloc(cmd->words, cmd->wordssize * sizeof(char *));
+       cmd->words[0] = "!";
+       cmd->words[1] = p+1;
+    } else {
+
+       /*
+        * Parse the command line into words. The syntax is:
+        *  - double quotes are removed, but cause spaces within to be
+        *    treated as non-separating.
+        *  - a double-doublequote pair is a literal double quote, inside
+        *    _or_ outside quotes. Like this:
+        *
+        *      firstword "second word" "this has ""quotes"" in" and""this""
+        *
+        * becomes
+        *
+        *      >firstword<
+        *      >second word<
+        *      >this has "quotes" in<
+        *      >and"this"<
+        */
        while (*p) {
        while (*p) {
-           if (!quoting && (*p == ' ' || *p == '\t'))
-               break;                 /* reached end of word */
-           else if (*p == '"' && p[1] == '"')
-               p += 2, *r++ = '"';    /* a literal quote */
-           else if (*p == '"')
-               p++, quoting = !quoting;
-           else
-               *r++ = *p++;
-       }
-       if (*p)
-           p++;                       /* skip over the whitespace */
-       *r = '\0';
-       if (cmd->nwords >= cmd->wordssize) {
-           cmd->wordssize = cmd->nwords + 16;
-           cmd->words =
-               srealloc(cmd->words, cmd->wordssize * sizeof(char *));
+           /* skip whitespace */
+           while (*p && (*p == ' ' || *p == '\t'))
+               p++;
+           /* mark start of word */
+           q = r = p;                 /* q sits at start, r writes word */
+           quoting = 0;
+           while (*p) {
+               if (!quoting && (*p == ' ' || *p == '\t'))
+                   break;                     /* reached end of word */
+               else if (*p == '"' && p[1] == '"')
+                   p += 2, *r++ = '"';    /* a literal quote */
+               else if (*p == '"')
+                   p++, quoting = !quoting;
+               else
+                   *r++ = *p++;
+           }
+           if (*p)
+               p++;                   /* skip over the whitespace */
+           *r = '\0';
+           if (cmd->nwords >= cmd->wordssize) {
+               cmd->wordssize = cmd->nwords + 16;
+               cmd->words =
+                   srealloc(cmd->words, cmd->wordssize * sizeof(char *));
+           }
+           cmd->words[cmd->nwords++] = q;
        }
        }
-       cmd->words[cmd->nwords++] = q;
     }
 
     /*
     }
 
     /*
@@ -573,30 +1350,18 @@ struct sftp_command *sftp_getcmd(void)
     if (cmd->nwords == 0)
        cmd->obey = sftp_cmd_null;
     else {
     if (cmd->nwords == 0)
        cmd->obey = sftp_cmd_null;
     else {
-       int i, j, k, cmp;
-
-       cmd->obey = sftp_cmd_unknown;
-
-       i = -1;
-       j = sizeof(sftp_lookup) / sizeof(*sftp_lookup);
-       while (j - i > 1) {
-           k = (j + i) / 2;
-           cmp = strcmp(cmd->words[0], sftp_lookup[k].name);
-           if (cmp < 0)
-               j = k;
-           else if (cmp > 0)
-               i = k;
-           else {
-               cmd->obey = sftp_lookup[k].obey;
-               break;
-           }
-       }
+       const struct sftp_cmd_lookup *lookup;
+       lookup = lookup_command(cmd->words[0]);
+       if (!lookup)
+           cmd->obey = sftp_cmd_unknown;
+       else
+           cmd->obey = lookup->obey;
     }
 
     return cmd;
 }
 
     }
 
     return cmd;
 }
 
-void do_sftp(void)
+static void do_sftp_init(void)
 {
     /*
      * Do protocol initialisation. 
 {
     /*
      * Do protocol initialisation. 
@@ -620,17 +1385,50 @@ void do_sftp(void)
        printf("Remote working directory is %s\n", homedir);
     }
     pwd = dupstr(homedir);
        printf("Remote working directory is %s\n", homedir);
     }
     pwd = dupstr(homedir);
+}
+
+void do_sftp(int mode, int modeflags, char *batchfile)
+{
+    FILE *fp;
+    int ret;
 
 
-    /* ------------------------------------------------------------------
-     * Now we're ready to do Real Stuff.
+    /*
+     * Batch mode?
      */
      */
-    while (1) {
-       struct sftp_command *cmd;
-       cmd = sftp_getcmd();
-       if (!cmd)
-           break;
-       if (cmd->obey(cmd) < 0)
-           break;
+    if (mode == 0) {
+
+        /* ------------------------------------------------------------------
+         * Now we're ready to do Real Stuff.
+         */
+        while (1) {
+           struct sftp_command *cmd;
+           cmd = sftp_getcmd(stdin, 0, 0);
+           if (!cmd)
+               break;
+           if (cmd->obey(cmd) < 0)
+               break;
+       }
+    } else {
+        fp = fopen(batchfile, "r");
+        if (!fp) {
+           printf("Fatal: unable to open %s\n", batchfile);
+           return;
+        }
+        while (1) {
+           struct sftp_command *cmd;
+           cmd = sftp_getcmd(fp, mode, modeflags);
+           if (!cmd)
+               break;
+           ret = cmd->obey(cmd);
+           if (ret < 0)
+               break;
+           if (ret == 0) {
+               if (!(modeflags & 2))
+                   break;
+           }
+        }
+       fclose(fp);
+
     }
 }
 
     }
 }
 
@@ -644,6 +1442,8 @@ void verify_ssh_host_key(char *host, int port, char *keytype,
                         char *keystr, char *fingerprint)
 {
     int ret;
                         char *keystr, char *fingerprint)
 {
     int ret;
+    HANDLE hin;
+    DWORD savemode, i;
 
     static const char absentmsg[] =
        "The server's host key is not cached in the registry. You\n"
 
     static const char absentmsg[] =
        "The server's host key is not cached in the registry. You\n"
@@ -653,8 +1453,11 @@ void verify_ssh_host_key(char *host, int port, char *keytype,
        "%s\n"
        "If you trust this host, enter \"y\" to add the key to\n"
        "PuTTY's cache and carry on connecting.\n"
        "%s\n"
        "If you trust this host, enter \"y\" to add the key to\n"
        "PuTTY's cache and carry on connecting.\n"
-       "If you do not trust this host, enter \"n\" to abandon the\n"
-       "connection.\n" "Continue connecting? (y/n) ";
+       "If you want to carry on connecting just once, without\n"
+       "adding the key to the cache, enter \"n\".\n"
+       "If you do not trust this host, press Return to abandon the\n"
+       "connection.\n"
+       "Store key in cache? (y/n) ";
 
     static const char wrongmsg[] =
        "WARNING - POTENTIAL SECURITY BREACH!\n"
 
     static const char wrongmsg[] =
        "WARNING - POTENTIAL SECURITY BREACH!\n"
@@ -666,9 +1469,9 @@ void verify_ssh_host_key(char *host, int port, char *keytype,
        "The new key fingerprint is:\n"
        "%s\n"
        "If you were expecting this change and trust the new key,\n"
        "The new key fingerprint is:\n"
        "%s\n"
        "If you were expecting this change and trust the new key,\n"
-       "enter Yes to update PuTTY's cache and continue connecting.\n"
+       "enter \"y\" to update PuTTY's cache and continue connecting.\n"
        "If you want to carry on connecting but without updating\n"
        "If you want to carry on connecting but without updating\n"
-       "the cache, enter No.\n"
+       "the cache, enter \"n\".\n"
        "If you want to abandon the connection completely, press\n"
        "Return to cancel. Pressing Return is the ONLY guaranteed\n"
        "safe choice.\n"
        "If you want to abandon the connection completely, press\n"
        "Return to cancel. Pressing Return is the ONLY guaranteed\n"
        "safe choice.\n"
@@ -685,30 +1488,130 @@ void verify_ssh_host_key(char *host, int port, char *keytype,
 
     if (ret == 0)                     /* success - key matched OK */
        return;
 
     if (ret == 0)                     /* success - key matched OK */
        return;
+
     if (ret == 2) {                   /* key was different */
        fprintf(stderr, wrongmsg, fingerprint);
     if (ret == 2) {                   /* key was different */
        fprintf(stderr, wrongmsg, fingerprint);
-       if (fgets(line, sizeof(line), stdin) &&
-           line[0] != '\0' && line[0] != '\n') {
-           if (line[0] == 'y' || line[0] == 'Y')
-               store_host_key(host, port, keytype, keystr);
-       } else {
-           fprintf(stderr, abandoned);
-           exit(0);
-       }
+       fflush(stderr);
     }
     if (ret == 1) {                   /* key was absent */
        fprintf(stderr, absentmsg, fingerprint);
     }
     if (ret == 1) {                   /* key was absent */
        fprintf(stderr, absentmsg, fingerprint);
-       if (fgets(line, sizeof(line), stdin) &&
-           (line[0] == 'y' || line[0] == 'Y'))
+       fflush(stderr);
+    }
+
+    hin = GetStdHandle(STD_INPUT_HANDLE);
+    GetConsoleMode(hin, &savemode);
+    SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT |
+                        ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT));
+    ReadFile(hin, line, sizeof(line) - 1, &i, NULL);
+    SetConsoleMode(hin, savemode);
+
+    if (line[0] != '\0' && line[0] != '\r' && line[0] != '\n') {
+       if (line[0] == 'y' || line[0] == 'Y')
            store_host_key(host, port, keytype, keystr);
            store_host_key(host, port, keytype, keystr);
-       else {
-           fprintf(stderr, abandoned);
-           exit(0);
-       }
+    } else {
+       fprintf(stderr, abandoned);
+       exit(0);
+    }
+}
+
+/*
+ * Ask whether the selected cipher is acceptable (since it was
+ * below the configured 'warn' threshold).
+ * cs: 0 = both ways, 1 = client->server, 2 = server->client
+ */
+void askcipher(char *ciphername, int cs)
+{
+    HANDLE hin;
+    DWORD savemode, i;
+
+    static const char msg[] =
+       "The first %scipher supported by the server is\n"
+       "%s, which is below the configured warning threshold.\n"
+       "Continue with connection? (y/n) ";
+    static const char abandoned[] = "Connection abandoned.\n";
+
+    char line[32];
+
+    fprintf(stderr, msg,
+           (cs == 0) ? "" :
+           (cs == 1) ? "client-to-server " :
+                       "server-to-client ",
+           ciphername);
+    fflush(stderr);
+
+    hin = GetStdHandle(STD_INPUT_HANDLE);
+    GetConsoleMode(hin, &savemode);
+    SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT |
+                        ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT));
+    ReadFile(hin, line, sizeof(line) - 1, &i, NULL);
+    SetConsoleMode(hin, savemode);
+
+    if (line[0] == 'y' || line[0] == 'Y') {
+       return;
+    } else {
+       fprintf(stderr, abandoned);
+       exit(0);
     }
 }
 
 /*
     }
 }
 
 /*
+ * Ask whether to wipe a session log file before writing to it.
+ * Returns 2 for wipe, 1 for append, 0 for cancel (don't log).
+ */
+int askappend(char *filename)
+{
+    HANDLE hin;
+    DWORD savemode, i;
+
+    static const char msgtemplate[] =
+       "The session log file \"%.*s\" already exists.\n"
+       "You can overwrite it with a new session log,\n"
+       "append your session log to the end of it,\n"
+       "or disable session logging for this session.\n"
+       "Enter \"y\" to wipe the file, \"n\" to append to it,\n"
+       "or just press Return to disable logging.\n"
+       "Wipe the log file? (y/n, Return cancels logging) ";
+
+    char line[32];
+
+    fprintf(stderr, msgtemplate, FILENAME_MAX, filename);
+    fflush(stderr);
+
+    hin = GetStdHandle(STD_INPUT_HANDLE);
+    GetConsoleMode(hin, &savemode);
+    SetConsoleMode(hin, (savemode | ENABLE_ECHO_INPUT |
+                        ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT));
+    ReadFile(hin, line, sizeof(line) - 1, &i, NULL);
+    SetConsoleMode(hin, savemode);
+
+    if (line[0] == 'y' || line[0] == 'Y')
+       return 2;
+    else if (line[0] == 'n' || line[0] == 'N')
+       return 1;
+    else
+       return 0;
+}
+
+/*
+ * Warn about the obsolescent key file format.
+ */
+void old_keyfile_warning(void)
+{
+    static const char message[] =
+       "You are loading an SSH 2 private key which has an\n"
+       "old version of the file format. This means your key\n"
+       "file is not fully tamperproof. Future versions of\n"
+       "PuTTY may stop supporting this private key format,\n"
+       "so we recommend you convert your key to the new\n"
+       "format.\n"
+       "\n"
+       "Once the key is loaded into PuTTYgen, you can perform\n"
+       "this conversion simply by saving it again.\n";
+
+    fputs(message, stderr);
+}
+
+/*
  *  Print an error message and perform a fatal exit.
  */
 void fatalbox(char *fmt, ...)
  *  Print an error message and perform a fatal exit.
  */
 void fatalbox(char *fmt, ...)
@@ -720,7 +1623,7 @@ void fatalbox(char *fmt, ...)
     vsprintf(str + strlen(str), fmt, ap);
     va_end(ap);
     strcat(str, "\n");
     vsprintf(str + strlen(str), fmt, ap);
     va_end(ap);
     strcat(str, "\n");
-    fprintf(stderr, str);
+    fputs(stderr, str);
 
     exit(1);
 }
 
     exit(1);
 }
@@ -733,7 +1636,7 @@ void connection_fatal(char *fmt, ...)
     vsprintf(str + strlen(str), fmt, ap);
     va_end(ap);
     strcat(str, "\n");
     vsprintf(str + strlen(str), fmt, ap);
     va_end(ap);
     strcat(str, "\n");
-    fprintf(stderr, str);
+    fputs(stderr, str);
 
     exit(1);
 }
 
     exit(1);
 }
@@ -742,7 +1645,7 @@ void logevent(char *string)
 {
 }
 
 {
 }
 
-void ldisc_send(char *buf, int len)
+void ldisc_send(char *buf, int len, int interactive)
 {
     /*
      * This is only here because of the calls to ldisc_send(NULL,
 {
     /*
      * This is only here because of the calls to ldisc_send(NULL,
@@ -780,7 +1683,7 @@ static unsigned char *outptr;             /* where to put the data */
 static unsigned outlen;                       /* how much data required */
 static unsigned char *pending = NULL;  /* any spare data */
 static unsigned pendlen = 0, pendsize = 0;     /* length and phys. size of buffer */
 static unsigned outlen;                       /* how much data required */
 static unsigned char *pending = NULL;  /* any spare data */
 static unsigned pendlen = 0, pendsize = 0;     /* length and phys. size of buffer */
-void from_backend(int is_stderr, char *data, int datalen)
+int from_backend(int is_stderr, char *data, int datalen)
 {
     unsigned char *p = (unsigned char *) data;
     unsigned len = (unsigned) datalen;
 {
     unsigned char *p = (unsigned char *) data;
     unsigned len = (unsigned) datalen;
@@ -791,14 +1694,14 @@ void from_backend(int is_stderr, char *data, int datalen)
      */
     if (is_stderr) {
        fwrite(data, 1, len, stderr);
      */
     if (is_stderr) {
        fwrite(data, 1, len, stderr);
-       return;
+       return 0;
     }
 
     /*
      * If this is before the real session begins, just return.
      */
     if (!outptr)
     }
 
     /*
      * If this is before the real session begins, just return.
      */
     if (!outptr)
-       return;
+       return 0;
 
     if (outlen > 0) {
        unsigned used = outlen;
 
     if (outlen > 0) {
        unsigned used = outlen;
@@ -822,6 +1725,8 @@ void from_backend(int is_stderr, char *data, int datalen)
        memcpy(pending + pendlen, p, len);
        pendlen += len;
     }
        memcpy(pending + pendlen, p, len);
        pendlen += len;
     }
+
+    return 0;
 }
 int sftp_recvdata(char *buf, int len)
 {
 }
 int sftp_recvdata(char *buf, int len)
 {
@@ -964,6 +1869,9 @@ static void usage(void)
     printf("%s\n", ver);
     printf("Usage: psftp [options] user@host\n");
     printf("Options:\n");
     printf("%s\n", ver);
     printf("Usage: psftp [options] user@host\n");
     printf("Options:\n");
+    printf("  -b file   use specified batchfile\n");
+    printf("  -bc       output batchfile commands\n");
+    printf("  -be       don't stop batchfile processing if errors\n");
     printf("  -v        show verbose messages\n");
     printf("  -P port   connect to specified port\n");
     printf("  -pw passw login with specified password\n");
     printf("  -v        show verbose messages\n");
     printf("  -P port   connect to specified port\n");
     printf("  -pw passw login with specified password\n");
@@ -971,53 +1879,13 @@ static void usage(void)
 }
 
 /*
 }
 
 /*
- * Main program. Parse arguments etc.
+ * Connect to a host.
  */
  */
-int main(int argc, char *argv[])
+static int psftp_connect(char *userhost, char *user, int portnumber)
 {
 {
-    int i;
-    int portnumber = 0;
-    char *user, *host, *userhost, *realhost;
+    char *host, *realhost;
     char *err;
 
     char *err;
 
-    flags = FLAG_STDERR;
-    ssh_get_line = &get_line;
-    init_winsock();
-    sk_init();
-
-    userhost = user = NULL;
-
-    for (i = 1; i < argc; i++) {
-       if (argv[i][0] != '-') {
-           if (userhost)
-               usage();
-           else
-               userhost = dupstr(argv[i]);
-       } else if (strcmp(argv[i], "-v") == 0) {
-           verbose = 1, flags |= FLAG_VERBOSE;
-       } else if (strcmp(argv[i], "-h") == 0 ||
-                  strcmp(argv[i], "-?") == 0) {
-           usage();
-       } else if (strcmp(argv[i], "-l") == 0 && i + 1 < argc) {
-           user = argv[++i];
-       } else if (strcmp(argv[i], "-P") == 0 && i + 1 < argc) {
-           portnumber = atoi(argv[++i]);
-       } else if (strcmp(argv[i], "-pw") == 0 && i + 1 < argc) {
-           password = argv[++i];
-       } else if (strcmp(argv[i], "--") == 0) {
-           i++;
-           break;
-       } else {
-           usage();
-       }
-    }
-    argc -= i;
-    argv += i;
-    back = NULL;
-
-    if (argc > 0 || !userhost)
-       usage();
-
     /* Separate host and username */
     host = userhost;
     host = strrchr(host, '@');
     /* Separate host and username */
     host = userhost;
     host = strrchr(host, '@');
@@ -1042,6 +1910,32 @@ int main(int argc, char *argv[])
        cfg.port = 22;
     }
 
        cfg.port = 22;
     }
 
+    /*
+     * Trim leading whitespace off the hostname if it's there.
+     */
+    {
+       int space = strspn(cfg.host, " \t");
+       memmove(cfg.host, cfg.host+space, 1+strlen(cfg.host)-space);
+    }
+
+    /* See if host is of the form user@host */
+    if (cfg.host[0] != '\0') {
+       char *atsign = strchr(cfg.host, '@');
+       /* Make sure we're not overflowing the user field */
+       if (atsign) {
+           if (atsign - cfg.host < sizeof cfg.username) {
+               strncpy(cfg.username, cfg.host, atsign - cfg.host);
+               cfg.username[atsign - cfg.host] = '\0';
+           }
+           memmove(cfg.host, atsign + 1, 1 + strlen(atsign + 1));
+       }
+    }
+
+    /*
+     * Trim a colon suffix off the hostname if it's there.
+     */
+    cfg.host[strcspn(cfg.host, ":")] = '\0';
+
     /* Set username */
     if (user != NULL && user[0] != '\0') {
        strncpy(cfg.username, user, sizeof(cfg.username) - 1);
     /* Set username */
     if (user != NULL && user[0] != '\0') {
        strncpy(cfg.username, user, sizeof(cfg.username) - 1);
@@ -1068,23 +1962,124 @@ int main(int argc, char *argv[])
     /* SFTP uses SSH2 by default always */
     cfg.sshprot = 2;
 
     /* SFTP uses SSH2 by default always */
     cfg.sshprot = 2;
 
-    /* Set up subsystem name. FIXME: fudge for SSH1. */
+    /*
+     * Disable scary things which shouldn't be enabled for simple
+     * things like SCP and SFTP: agent forwarding, port forwarding,
+     * X forwarding.
+     */
+    cfg.x11_forward = 0;
+    cfg.agentfwd = 0;
+    cfg.portfwd[0] = cfg.portfwd[1] = '\0';
+
+    /* Set up subsystem name. */
     strcpy(cfg.remote_cmd, "sftp");
     cfg.ssh_subsys = TRUE;
     cfg.nopty = TRUE;
 
     strcpy(cfg.remote_cmd, "sftp");
     cfg.ssh_subsys = TRUE;
     cfg.nopty = TRUE;
 
+    /*
+     * Set up fallback option, for SSH1 servers or servers with the
+     * sftp subsystem not enabled but the server binary installed
+     * in the usual place. We only support fallback on Unix
+     * systems, and we use a kludgy piece of shellery which should
+     * try to find sftp-server in various places (the obvious
+     * systemwide spots /usr/lib and /usr/local/lib, and then the
+     * user's PATH) and finally give up.
+     * 
+     *   test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server
+     *   test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server
+     *   exec sftp-server
+     * 
+     * the idea being that this will attempt to use either of the
+     * obvious pathnames and then give up, and when it does give up
+     * it will print the preferred pathname in the error messages.
+     */
+    cfg.remote_cmd_ptr2 =
+       "test -x /usr/lib/sftp-server && exec /usr/lib/sftp-server\n"
+       "test -x /usr/local/lib/sftp-server && exec /usr/local/lib/sftp-server\n"
+       "exec sftp-server";
+    cfg.ssh_subsys2 = FALSE;
+
     back = &ssh_backend;
 
     back = &ssh_backend;
 
-    err = back->init(cfg.host, cfg.port, &realhost);
+    err = back->init(cfg.host, cfg.port, &realhost, 0);
     if (err != NULL) {
     if (err != NULL) {
-       fprintf(stderr, "ssh_init: %s", err);
+       fprintf(stderr, "ssh_init: %s\n", err);
        return 1;
     }
     ssh_sftp_init();
     if (verbose && realhost != NULL)
        printf("Connected to %s\n", realhost);
        return 1;
     }
     ssh_sftp_init();
     if (verbose && realhost != NULL)
        printf("Connected to %s\n", realhost);
+    return 0;
+}
+
+/*
+ * Main program. Parse arguments etc.
+ */
+int main(int argc, char *argv[])
+{
+    int i;
+    int portnumber = 0;
+    char *userhost, *user;
+    int mode = 0;
+    int modeflags = 0;
+    char *batchfile = NULL;
+
+    flags = FLAG_STDERR | FLAG_INTERACTIVE;
+    ssh_get_line = &get_line;
+    init_winsock();
+    sk_init();
+
+    userhost = user = NULL;
+
+    for (i = 1; i < argc; i++) {
+       if (argv[i][0] != '-') {
+           if (userhost)
+               usage();
+           else
+               userhost = dupstr(argv[i]);
+       } else if (strcmp(argv[i], "-v") == 0) {
+           verbose = 1, flags |= FLAG_VERBOSE;
+       } else if (strcmp(argv[i], "-h") == 0 ||
+                  strcmp(argv[i], "-?") == 0) {
+           usage();
+       } else if (strcmp(argv[i], "-l") == 0 && i + 1 < argc) {
+           user = argv[++i];
+       } else if (strcmp(argv[i], "-P") == 0 && i + 1 < argc) {
+           portnumber = atoi(argv[++i]);
+       } else if (strcmp(argv[i], "-pw") == 0 && i + 1 < argc) {
+           password = argv[++i];
+       } else if (strcmp(argv[i], "-b") == 0 && i + 1 < argc) {
+           mode = 1;
+           batchfile = argv[++i];
+       } else if (strcmp(argv[i], "-bc") == 0) {
+           modeflags = modeflags | 1;
+       } else if (strcmp(argv[i], "-be") == 0) {
+           modeflags = modeflags | 2;
+       } else if (strcmp(argv[i], "--") == 0) {
+           i++;
+           break;
+       } else {
+           usage();
+       }
+    }
+    argc -= i;
+    argv += i;
+    back = NULL;
+
+    /*
+     * If a user@host string has already been provided, connect to
+     * it now.
+     */
+    if (userhost) {
+       if (psftp_connect(userhost, user, portnumber))
+           return 1;
+       do_sftp_init();
+    } else {
+       printf("psftp: no hostname specified; use \"open host.name\""
+           " to connect\n");
+    }
 
 
-    do_sftp();
+    do_sftp(mode, modeflags, batchfile);
 
     if (back != NULL && back->socket() != NULL) {
        char ch;
 
     if (back != NULL && back->socket() != NULL) {
        char ch;