The back ends now contain their own copies of the Config structure,
[u/mdw/putty] / psftp.c
diff --git a/psftp.c b/psftp.c
index 250a5fc..f8d1f79 100644 (file)
--- a/psftp.c
+++ b/psftp.c
  * send buffer.
  */
 
+static int psftp_connect(char *userhost, char *user, int portnumber);
+static int do_sftp_init(void);
+
 /* ----------------------------------------------------------------------
  * sftp client state.
  */
 
 char *pwd, *homedir;
+static Backend *back;
+static void *backhandle;
 
 /* ----------------------------------------------------------------------
  * Higher-level helper functions used in commands.
@@ -165,13 +170,13 @@ struct sftp_command {
 
 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]);
-    return 0;
+    return 0;                         /* failure */
 }
 
 int sftp_cmd_quit(struct sftp_command *cmd)
@@ -185,19 +190,24 @@ int sftp_cmd_quit(struct sftp_command *cmd)
  */
 static int sftp_ls_compare(const void *av, const void *bv)
 {
-    const struct fxp_name *a = (const struct fxp_name *) av;
-    const struct fxp_name *b = (const struct fxp_name *) bv;
-    return strcmp(a->filename, b->filename);
+    const struct fxp_name *const *a = (const struct fxp_name *const *) av;
+    const struct fxp_name *const *b = (const struct fxp_name *const *) bv;
+    return strcmp((*a)->filename, (*b)->filename);
 }
 int sftp_cmd_ls(struct sftp_command *cmd)
 {
     struct fxp_handle *dirh;
     struct fxp_names *names;
-    struct fxp_name *ournames;
+    struct fxp_name **ournames;
     int nnames, namesize;
     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
@@ -239,9 +249,8 @@ int sftp_cmd_ls(struct sftp_command *cmd)
            }
 
            for (i = 0; i < names->nnames; i++)
-               ournames[nnames++] = names->names[i];
+               ournames[nnames++] = fxp_dup_name(&names->names[i]);
 
-           names->nnames = 0;         /* prevent free_names */
            fxp_free_names(names);
        }
        fxp_close(dirh);
@@ -255,13 +264,16 @@ int sftp_cmd_ls(struct sftp_command *cmd)
        /*
         * And print them.
         */
-       for (i = 0; i < nnames; i++)
-           printf("%s\n", ournames[i].longname);
+       for (i = 0; i < nnames; i++) {
+           printf("%s\n", ournames[i]->longname);
+           fxp_free_name(ournames[i]);
+       }
+       sfree(ournames);
     }
 
     sfree(cdir);
 
-    return 0;
+    return 1;
 }
 
 /*
@@ -273,6 +285,11 @@ int sftp_cmd_cd(struct sftp_command *cmd)
     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
@@ -296,7 +313,7 @@ int sftp_cmd_cd(struct sftp_command *cmd)
     pwd = dir;
     printf("Remote directory is now %s\n", pwd);
 
-    return 0;
+    return 1;
 }
 
 /*
@@ -304,8 +321,13 @@ int sftp_cmd_cd(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 0;
+    return 1;
 }
 
 /*
@@ -320,6 +342,12 @@ int sftp_general_get(struct sftp_command *cmd, int restart)
     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");
@@ -370,6 +398,7 @@ int sftp_general_get(struct sftp_command *cmd, int restart)
      * 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;
@@ -380,6 +409,7 @@ int sftp_general_get(struct sftp_command *cmd, int restart)
            break;
        if (len == -1) {
            printf("error while reading: %s\n", fxp_error());
+           ret = 0;
            break;
        }
 
@@ -388,12 +418,15 @@ int sftp_general_get(struct sftp_command *cmd, int restart)
            wlen = fwrite(buffer, 1, len - wpos, fp);
            if (wlen <= 0) {
                printf("error while writing local file\n");
+               ret = 0;
                break;
            }
            wpos += wlen;
        }
-       if (wpos < len)                /* we had an error */
+       if (wpos < len) {              /* we had an error */
+           ret = 0;
            break;
+       }
        offset = uint64_add32(offset, len);
     }
 
@@ -401,7 +434,7 @@ int sftp_general_get(struct sftp_command *cmd, int restart)
     fxp_close(fh);
     sfree(fname);
 
-    return 0;
+    return ret;
 }
 int sftp_cmd_get(struct sftp_command *cmd)
 {
@@ -424,6 +457,12 @@ int sftp_general_put(struct sftp_command *cmd, int restart)
     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");
@@ -491,6 +530,7 @@ int sftp_general_put(struct sftp_command *cmd, int restart)
      * 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;
@@ -498,12 +538,14 @@ int sftp_general_put(struct sftp_command *cmd, int restart)
        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());
+           ret = 0;
            break;
        }
        offset = uint64_add32(offset, len);
@@ -513,7 +555,7 @@ int sftp_general_put(struct sftp_command *cmd, int restart)
     fclose(fp);
     sfree(outfname);
 
-    return 0;
+    return ret;
 }
 int sftp_cmd_put(struct sftp_command *cmd)
 {
@@ -529,6 +571,10 @@ 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");
@@ -549,7 +595,7 @@ int sftp_cmd_mkdir(struct sftp_command *cmd)
     }
 
     sfree(dir);
-    return 0;
+    return 1;
 }
 
 int sftp_cmd_rmdir(struct sftp_command *cmd)
@@ -557,6 +603,10 @@ 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");
@@ -577,7 +627,7 @@ int sftp_cmd_rmdir(struct sftp_command *cmd)
     }
 
     sfree(dir);
-    return 0;
+    return 1;
 }
 
 int sftp_cmd_rm(struct sftp_command *cmd)
@@ -585,6 +635,11 @@ 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;
@@ -604,8 +659,7 @@ int sftp_cmd_rm(struct sftp_command *cmd)
     }
 
     sfree(fname);
-    return 0;
-
+    return 1;
 }
 
 int sftp_cmd_mv(struct sftp_command *cmd)
@@ -613,6 +667,11 @@ 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;
@@ -670,7 +729,7 @@ int sftp_cmd_mv(struct sftp_command *cmd)
 
     sfree(srcfname);
     sfree(dstfname);
-    return 0;
+    return 1;
 }
 
 int sftp_cmd_chmod(struct sftp_command *cmd)
@@ -680,6 +739,11 @@ int sftp_cmd_chmod(struct sftp_command *cmd)
     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;
@@ -768,7 +832,7 @@ int sftp_cmd_chmod(struct sftp_command *cmd)
            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);
+                      strcspn(modebegin, ","), modebegin);
                return 0;
            }
            perms &= subset;
@@ -821,7 +885,87 @@ int sftp_cmd_chmod(struct sftp_command *cmd)
     printf("%s: %04o -> %04o\n", fname, oldperms, newperms);
 
     sfree(fname);
-    return 0;
+    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);
@@ -840,6 +984,7 @@ static struct sftp_cmd_lookup {
      *    `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 *);
@@ -849,13 +994,19 @@ static struct sftp_cmd_lookup {
      * in ASCII order.
      */
     {
-       "bye", "finish your SFTP session",
+       "!", 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", "change your remote working directory",
+       "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"
@@ -863,7 +1014,7 @@ static struct sftp_cmd_lookup {
            sftp_cmd_cd
     },
     {
-       "chmod", "change file permissions and modes",
+       "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"
@@ -890,19 +1041,16 @@ static struct sftp_cmd_lookup {
            sftp_cmd_chmod
     },
     {
-       "del", "delete a file",
+       "del", TRUE, "delete a file",
            " <filename>\n"
            "  Delete a file.\n",
            sftp_cmd_rm
     },
     {
-       "delete", "delete a file",
-           "\n"
-           "  Delete a file.\n",
-           sftp_cmd_rm
+       "delete", FALSE, "del", NULL, sftp_cmd_rm
     },
     {
-       "dir", "list contents of a remote directory",
+       "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"
@@ -910,10 +1058,10 @@ static struct sftp_cmd_lookup {
            sftp_cmd_ls
     },
     {
-       "exit", "bye", NULL, sftp_cmd_quit
+       "exit", TRUE, "bye", NULL, sftp_cmd_quit
     },
     {
-       "get", "download a file from the server to your local machine",
+       "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"
@@ -921,7 +1069,7 @@ static struct sftp_cmd_lookup {
            sftp_cmd_get
     },
     {
-       "help", "give help",
+       "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"
@@ -929,24 +1077,46 @@ static struct sftp_cmd_lookup {
            sftp_cmd_help
     },
     {
-       "ls", "dir", NULL,
+       "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", "create a directory on the remote server",
+       "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", "move or rename a file on the remote server",
+       "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", "upload a file from your local machine to the server",
+       "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
+    },
+    {
+       "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"
@@ -954,17 +1124,17 @@ static struct sftp_cmd_lookup {
            sftp_cmd_put
     },
     {
-       "pwd", "print your remote working directory",
+       "pwd", TRUE, "print your remote working directory",
            "\n"
            "  Print the current remote working directory for your SFTP session.\n",
            sftp_cmd_pwd
     },
     {
-       "quit", "bye", NULL,
+       "quit", TRUE, "bye", NULL,
            sftp_cmd_quit
     },
     {
-       "reget", "continue downloading a file",
+       "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"
@@ -972,15 +1142,15 @@ static struct sftp_cmd_lookup {
            sftp_cmd_reget
     },
     {
-       "ren", "mv", NULL,
+       "ren", TRUE, "mv", NULL,
            sftp_cmd_mv
     },
     {
-       "rename", "mv", NULL,
+       "rename", FALSE, "mv", NULL,
            sftp_cmd_mv
     },
     {
-       "reput", "continue uploading a file",
+       "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"
@@ -988,11 +1158,11 @@ static struct sftp_cmd_lookup {
            sftp_cmd_reput
     },
     {
-       "rm", "del", NULL,
+       "rm", TRUE, "del", NULL,
            sftp_cmd_rm
     },
     {
-       "rmdir", "remove a directory on the remote server",
+       "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",
@@ -1030,12 +1200,17 @@ static int sftp_cmd_help(struct sftp_command *cmd)
        int maxlen;
        maxlen = 0;
        for (i = 0; i < sizeof(sftp_lookup) / sizeof(*sftp_lookup); i++) {
-           int len = strlen(sftp_lookup[i].name);
+           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)
@@ -1059,7 +1234,7 @@ static int sftp_cmd_help(struct sftp_command *cmd)
            }
        }
     }
-    return 0;
+    return 1;
 }
 
 /* ----------------------------------------------------------------------
@@ -1073,9 +1248,9 @@ struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
     char *p, *q, *r;
     int quoting;
 
-       if ((mode == 0) || (modeflags & 1)) {
-           printf("psftp> ");
-       }
+    if ((mode == 0) || (modeflags & 1)) {
+       printf("psftp> ");
+    }
     fflush(stdout);
 
     cmd = smalloc(sizeof(struct sftp_command));
@@ -1092,13 +1267,11 @@ struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
        linesize += 512;
        line = srealloc(line, linesize);
        ret = fgets(line + linelen, linesize - linelen, fp);
-       if (modeflags & 1) {
-               printf("%s", ret);
-       }
 
        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);
@@ -1109,50 +1282,69 @@ struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
            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;
-    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) {
-           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;
     }
 
     /*
@@ -1173,17 +1365,15 @@ struct sftp_command *sftp_getcmd(FILE *fp, int mode, int modeflags)
     return cmd;
 }
 
-void do_sftp(int mode, int modeflags, char *batchfile)
+static int do_sftp_init(void)
 {
-    FILE *fp;
-
     /*
      * Do protocol initialisation. 
      */
     if (!fxp_init()) {
        fprintf(stderr,
                "Fatal: unable to initialise SFTP: %s\n", fxp_error());
-       return;
+       return 1;                      /* failure */
     }
 
     /*
@@ -1199,6 +1389,13 @@ void do_sftp(int mode, int modeflags, char *batchfile)
        printf("Remote working directory is %s\n", homedir);
     }
     pwd = dupstr(homedir);
+    return 0;
+}
+
+void do_sftp(int mode, int modeflags, char *batchfile)
+{
+    FILE *fp;
+    int ret;
 
     /*
      * Batch mode?
@@ -1209,12 +1406,12 @@ void do_sftp(int mode, int modeflags, char *batchfile)
          * 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;
+           struct sftp_command *cmd;
+           cmd = sftp_getcmd(stdin, 0, 0);
+           if (!cmd)
+               break;
+           if (cmd->obey(cmd) < 0)
+               break;
        }
     } else {
         fp = fopen(batchfile, "r");
@@ -1227,9 +1424,10 @@ void do_sftp(int mode, int modeflags, char *batchfile)
            cmd = sftp_getcmd(fp, mode, modeflags);
            if (!cmd)
                break;
-           if (cmd->obey(cmd) < 0)
+           ret = cmd->obey(cmd);
+           if (ret < 0)
                break;
-           if (fxp_error() != NULL) {
+           if (ret == 0) {
                if (!(modeflags & 2))
                    break;
            }
@@ -1245,176 +1443,53 @@ void do_sftp(int mode, int modeflags, char *batchfile)
 
 static int verbose = 0;
 
-void verify_ssh_host_key(char *host, int port, char *keytype,
-                        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"
-       "have no guarantee that the server is the computer you\n"
-       "think it is.\n"
-       "The server's key fingerprint is:\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 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"
-       "The server's host key does not match the one PuTTY has\n"
-       "cached in the registry. This means that either the\n"
-       "server administrator has changed the host key, or you\n"
-       "have actually connected to another computer pretending\n"
-       "to be the server.\n"
-       "The new key fingerprint is:\n"
-       "%s\n"
-       "If you were expecting this change and trust the new key,\n"
-       "enter \"y\" to update PuTTY's cache and continue connecting.\n"
-       "If you want to carry on connecting but without updating\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"
-       "Update cached key? (y/n, Return cancels connection) ";
-
-    static const char abandoned[] = "Connection abandoned.\n";
-
-    char line[32];
-
-    /*
-     * Verify the key against the registry.
-     */
-    ret = verify_host_key(host, port, keytype, keystr);
-
-    if (ret == 0)                     /* success - key matched OK */
-       return;
-
-    if (ret == 2) {                   /* key was different */
-       fprintf(stderr, wrongmsg, fingerprint);
-       fflush(stderr);
-    }
-    if (ret == 1) {                   /* key was absent */
-       fprintf(stderr, absentmsg, fingerprint);
-       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);
-    } 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);
-    }
-}
-
-/*
- * 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, ...)
 {
-    char str[0x100];                  /* Make the size big enough */
+    char *str, *str2;
     va_list ap;
     va_start(ap, fmt);
-    strcpy(str, "Fatal:");
-    vsprintf(str + strlen(str), fmt, ap);
+    str = dupvprintf(fmt, ap);
+    str2 = dupcat("Fatal: ", str, "\n", NULL);
+    sfree(str);
     va_end(ap);
-    strcat(str, "\n");
-    fprintf(stderr, str);
+    fputs(str2, stderr);
+    sfree(str2);
 
-    exit(1);
+    cleanup_exit(1);
 }
-void connection_fatal(char *fmt, ...)
+void modalfatalbox(char *fmt, ...)
 {
-    char str[0x100];                  /* Make the size big enough */
+    char *str, *str2;
     va_list ap;
     va_start(ap, fmt);
-    strcpy(str, "Fatal:");
-    vsprintf(str + strlen(str), fmt, ap);
+    str = dupvprintf(fmt, ap);
+    str2 = dupcat("Fatal: ", str, "\n", NULL);
+    sfree(str);
     va_end(ap);
-    strcat(str, "\n");
-    fprintf(stderr, str);
+    fputs(str2, stderr);
+    sfree(str2);
 
-    exit(1);
+    cleanup_exit(1);
 }
-
-void logevent(char *string)
+void connection_fatal(void *frontend, char *fmt, ...)
 {
+    char *str, *str2;
+    va_list ap;
+    va_start(ap, fmt);
+    str = dupvprintf(fmt, ap);
+    str2 = dupcat("Fatal: ", str, "\n", NULL);
+    sfree(str);
+    va_end(ap);
+    fputs(str2, stderr);
+    sfree(str2);
+
+    cleanup_exit(1);
 }
 
-void ldisc_send(char *buf, int len, int interactive)
+void ldisc_send(void *handle, char *buf, int len, int interactive)
 {
     /*
      * This is only here because of the calls to ldisc_send(NULL,
@@ -1452,11 +1527,13 @@ 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 */
-int from_backend(int is_stderr, char *data, int datalen)
+int from_backend(void *frontend, int is_stderr, char *data, int datalen)
 {
     unsigned char *p = (unsigned char *) data;
     unsigned len = (unsigned) datalen;
 
+    assert(len > 0);
+
     /*
      * stderr data is just spouted to local stderr and otherwise
      * ignored.
@@ -1538,7 +1615,7 @@ int sftp_recvdata(char *buf, int len)
 }
 int sftp_senddata(char *buf, int len)
 {
-    back->send((unsigned char *) buf, len);
+    back->send(backhandle, (unsigned char *) buf, len);
     return 1;
 }
 
@@ -1549,7 +1626,7 @@ static void ssh_sftp_init(void)
 {
     if (sftp_ssh_socket == INVALID_SOCKET)
        return;
-    while (!back->sendok()) {
+    while (!back->sendok(backhandle)) {
        fd_set readfds;
        FD_ZERO(&readfds);
        FD_SET(sftp_ssh_socket, &readfds);
@@ -1559,57 +1636,6 @@ static void ssh_sftp_init(void)
     }
 }
 
-static char *password = NULL;
-static int get_line(const char *prompt, char *str, int maxlen, int is_pw)
-{
-    HANDLE hin, hout;
-    DWORD savemode, newmode, i;
-
-    if (password) {
-       static int tried_once = 0;
-
-       if (tried_once) {
-           return 0;
-       } else {
-           strncpy(str, password, maxlen);
-           str[maxlen - 1] = '\0';
-           tried_once = 1;
-           return 1;
-       }
-    }
-
-    hin = GetStdHandle(STD_INPUT_HANDLE);
-    hout = GetStdHandle(STD_OUTPUT_HANDLE);
-    if (hin == INVALID_HANDLE_VALUE || hout == INVALID_HANDLE_VALUE) {
-       fprintf(stderr, "Cannot get standard input/output handles\n");
-       exit(1);
-    }
-
-    GetConsoleMode(hin, &savemode);
-    newmode = savemode | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT;
-    if (is_pw)
-       newmode &= ~ENABLE_ECHO_INPUT;
-    else
-       newmode |= ENABLE_ECHO_INPUT;
-    SetConsoleMode(hin, newmode);
-
-    WriteFile(hout, prompt, strlen(prompt), &i, NULL);
-    ReadFile(hin, str, maxlen - 1, &i, NULL);
-
-    SetConsoleMode(hin, savemode);
-
-    if ((int) i > maxlen)
-       i = maxlen - 1;
-    else
-       i = i - 2;
-    str[i] = '\0';
-
-    if (is_pw)
-       WriteFile(hout, "\r\n", 2, &i, NULL);
-
-    return 1;
-}
-
 /*
  *  Initialize the Win$ock driver.
  */
@@ -1621,11 +1647,11 @@ static void init_winsock(void)
     winsock_ver = MAKEWORD(1, 1);
     if (WSAStartup(winsock_ver, &wsadata)) {
        fprintf(stderr, "Unable to initialise WinSock");
-       exit(1);
+       cleanup_exit(1);
     }
     if (LOBYTE(wsadata.wVersion) != 1 || HIBYTE(wsadata.wVersion) != 1) {
        fprintf(stderr, "WinSock version is incompatible with 1.1");
-       exit(1);
+       cleanup_exit(1);
     }
 }
 
@@ -1642,68 +1668,24 @@ static void usage(void)
     printf("  -bc       output batchfile commands\n");
     printf("  -be       don't stop batchfile processing if errors\n");
     printf("  -v        show verbose messages\n");
+    printf("  -load sessname  Load settings from saved session\n");
+    printf("  -l user   connect with specified username\n");
     printf("  -P port   connect to specified port\n");
     printf("  -pw passw login with specified password\n");
-    exit(1);
+    printf("  -1 -2     force use of particular SSH protocol version\n");
+    printf("  -C        enable compression\n");
+    printf("  -i key    private key file for authentication\n");
+    printf("  -batch    disable all interactive prompts\n");
+    cleanup_exit(1);
 }
 
 /*
- * 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;
-    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 && i + 1 < argc) {
-           modeflags = modeflags | 1;
-    } else if (strcmp(argv[i], "-be") == 0 && i + 1 < argc) {
-           modeflags = modeflags | 2;
-       } 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;
@@ -1726,10 +1708,23 @@ int main(int argc, char *argv[])
        do_defaults(NULL, &cfg);
        strncpy(cfg.host, host, sizeof(cfg.host) - 1);
        cfg.host[sizeof(cfg.host) - 1] = '\0';
-       cfg.port = 22;
     }
 
     /*
+     * Force use of SSH. (If they got the protocol wrong we assume the
+     * port is useless too.)
+     */
+    if (cfg.protocol != PROT_SSH) {
+        cfg.protocol = PROT_SSH;
+        cfg.port = 22;
+    }
+
+    /*
+     * Enact command-line overrides.
+     */
+    cmdline_run_saved(&cfg);
+
+    /*
      * Trim leading whitespace off the hostname if it's there.
      */
     {
@@ -1755,6 +1750,21 @@ int main(int argc, char *argv[])
      */
     cfg.host[strcspn(cfg.host, ":")] = '\0';
 
+    /*
+     * Remove any remaining whitespace from the hostname.
+     */
+    {
+       int p1 = 0, p2 = 0;
+       while (cfg.host[p2] != '\0') {
+           if (cfg.host[p2] != ' ' && cfg.host[p2] != '\t') {
+               cfg.host[p1] = cfg.host[p2];
+               p1++;
+           }
+           p2++;
+       }
+       cfg.host[p1] = '\0';
+    }
+
     /* Set username */
     if (user != NULL && user[0] != '\0') {
        strncpy(cfg.username, user, sizeof(cfg.username) - 1);
@@ -1762,9 +1772,10 @@ int main(int argc, char *argv[])
     }
     if (!cfg.username[0]) {
        printf("login as: ");
+       fflush(stdout);
        if (!fgets(cfg.username, sizeof(cfg.username), stdin)) {
            fprintf(stderr, "psftp: aborting\n");
-           exit(1);
+           cleanup_exit(1);
        } else {
            int len = strlen(cfg.username);
            if (cfg.username[len - 1] == '\n')
@@ -1772,9 +1783,6 @@ int main(int argc, char *argv[])
        }
     }
 
-    if (cfg.protocol != PROT_SSH)
-       cfg.port = 22;
-
     if (portnumber)
        cfg.port = portnumber;
 
@@ -1820,20 +1828,112 @@ int main(int argc, char *argv[])
 
     back = &ssh_backend;
 
-    err = back->init(cfg.host, cfg.port, &realhost, 0);
+    err = back->init(NULL, &backhandle, &cfg, cfg.host, cfg.port, &realhost,0);
     if (err != NULL) {
-       fprintf(stderr, "ssh_init: %s", err);
+       fprintf(stderr, "ssh_init: %s\n", err);
        return 1;
     }
+    logctx = log_init(NULL);
+    back->provide_logctx(backhandle, logctx);
     ssh_sftp_init();
     if (verbose && realhost != NULL)
        printf("Connected to %s\n", realhost);
+    return 0;
+}
+
+void cmdline_error(char *p, ...)
+{
+    va_list ap;
+    fprintf(stderr, "psftp: ");
+    va_start(ap, p);
+    vfprintf(stderr, p, ap);
+    va_end(ap);
+    fprintf(stderr, "\n       try typing \"psftp -h\" for help\n");
+    exit(1);
+}
+
+/*
+ * 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;
+    int errors = 0;
+
+    flags = FLAG_STDERR | FLAG_INTERACTIVE;
+    cmdline_tooltype = TOOLTYPE_FILETRANSFER;
+    ssh_get_line = &console_get_line;
+    init_winsock();
+    sk_init();
+
+    userhost = user = NULL;
+
+    errors = 0;
+    for (i = 1; i < argc; i++) {
+       int ret;
+       if (argv[i][0] != '-') {
+            if (userhost)
+                usage();
+            else
+                userhost = dupstr(argv[i]);
+           continue;
+       }
+       ret = cmdline_process_param(argv[i], i+1<argc?argv[i+1]:NULL, 1, &cfg);
+       if (ret == -2) {
+           cmdline_error("option \"%s\" requires an argument", argv[i]);
+       } else if (ret == 2) {
+           i++;               /* skip next argument */
+       } else if (ret == 1) {
+           /* We have our own verbosity in addition to `flags'. */
+           if (flags & FLAG_VERBOSE)
+               verbose = 1;
+       } else if (strcmp(argv[i], "-h") == 0 ||
+                  strcmp(argv[i], "-?") == 0) {
+           usage();
+       } else if (strcmp(argv[i], "-batch") == 0) {
+           console_batch_mode = 1;
+       } 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 {
+           cmdline_error("unknown option \"%s\"", argv[i]);
+       }
+    }
+    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;
+       if (do_sftp_init())
+           return 1;
+    } else {
+       printf("psftp: no hostname specified; use \"open host.name\""
+           " to connect\n");
+    }
 
     do_sftp(mode, modeflags, batchfile);
 
-    if (back != NULL && back->socket() != NULL) {
+    if (back != NULL && back->socket(backhandle) != NULL) {
        char ch;
-       back->special(TS_EOF);
+       back->special(backhandle, TS_EOF);
        sftp_recvdata(&ch, 1);
     }
     WSACleanup();