Rethink the whole line discipline architecture. Instead of having
[u/mdw/putty] / scp.c
diff --git a/scp.c b/scp.c
index 69ebac8..c13be5e 100644 (file)
--- a/scp.c
+++ b/scp.c
@@ -4,24 +4,47 @@
  *
  *  This is mainly based on ssh-1.2.26/scp.c by Timo Rinne & Tatu Ylonen.
  *  They, in turn, used stuff from BSD rcp.
+ *
+ *  Adaptations to enable connecting a GUI by L. Gunnarsson - Sept 2000
  */
 
 #include <windows.h>
+#ifndef AUTO_WINSOCK
+#ifdef WINSOCK_TWO
+#include <winsock2.h>
+#else
 #include <winsock.h>
+#endif
+#endif
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
 #include <time.h>
+/* GUI Adaptation - Sept 2000 */
+#include <winuser.h>
+#include <winbase.h>
 
 #define PUTTY_DO_GLOBALS
 #include "putty.h"
-#include "scp.h"
+#include "winstuff.h"
+#include "storage.h"
 
 #define TIME_POSIX_TO_WIN(t, ft) (*(LONGLONG*)&(ft) = \
        ((LONGLONG) (t) + (LONGLONG) 11644473600) * (LONGLONG) 10000000)
 #define TIME_WIN_TO_POSIX(ft, t) ((t) = (unsigned long) \
        ((*(LONGLONG*)&(ft)) / (LONGLONG) 10000000 - (LONGLONG) 11644473600))
 
+/* GUI Adaptation - Sept 2000 */
+#define   WM_APP_BASE          0x8000
+#define   WM_STD_OUT_CHAR      ( WM_APP_BASE+400 )
+#define   WM_STD_ERR_CHAR      ( WM_APP_BASE+401 )
+#define   WM_STATS_CHAR                ( WM_APP_BASE+402 )
+#define   WM_STATS_SIZE        ( WM_APP_BASE+403 )
+#define   WM_STATS_PERCENT     ( WM_APP_BASE+404 )
+#define   WM_STATS_ELAPSED     ( WM_APP_BASE+405 )
+#define   WM_RET_ERR_CNT       ( WM_APP_BASE+406 )
+#define   WM_LS_RET_ERR_CNT    ( WM_APP_BASE+407 )
+
 static int verbose = 0;
 static int recursive = 0;
 static int preserve = 0;
@@ -30,18 +53,161 @@ static int statistics = 1;
 static int portnumber = 0;
 static char *password = NULL;
 static int errs = 0;
-static int connection_open = 0;
+/* GUI Adaptation - Sept 2000 */
+#define NAME_STR_MAX 2048
+static char statname[NAME_STR_MAX+1];
+static unsigned long statsize = 0;
+static int statperct = 0;
+static unsigned long statelapsed = 0;
+static int gui_mode = 0;
+static char *gui_hwnd = NULL;
 
 static void source(char *src);
 static void rsource(char *src);
-static void sink(char *targ);
+static void sink(char *targ, char *src);
+/* GUI Adaptation - Sept 2000 */
+static void tell_char(FILE *stream, char c);
+static void tell_str(FILE *stream, char *str);
+static void tell_user(FILE *stream, char *fmt, ...);
+static void send_char_msg(unsigned int msg_id, char c);
+static void send_str_msg(unsigned int msg_id, char *str);
+static void gui_update_stats(char *name, unsigned long size,
+                             int percentage, unsigned long elapsed);
+
+void logevent(char *string) { }
+
+void verify_ssh_host_key(char *host, int port, char *keytype,
+                         char *keystr, char *fingerprint) {
+    int ret;
+
+    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 do not trust this host, enter \"n\" to abandon the\n"
+        "connection.\n"
+        "Continue connecting? (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 Yes to update PuTTY's cache and continue connecting.\n"
+        "If you want to carry on connecting but without updating\n"
+        "the cache, enter No.\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);
+        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);
+        }
+    }
+    if (ret == 1) {                    /* key was absent */
+        fprintf(stderr, absentmsg, fingerprint);
+        if (fgets(line, sizeof(line), stdin) &&
+            (line[0] == 'y' || line[0] == 'Y'))
+            store_host_key(host, port, keytype, keystr);
+        else {
+            fprintf(stderr, abandoned);
+            exit(0);
+        }
+    }
+}
 
-/*
- *  This function is needed to link with ssh.c, but it never gets called.
- */
-void term_out(void)
+/* GUI Adaptation - Sept 2000 */
+static void send_msg(HWND h, UINT message, WPARAM wParam)
+{
+    while (!PostMessage( h, message, wParam, 0))
+        SleepEx(1000,TRUE);
+}
+
+static void tell_char(FILE *stream, char c)
 {
-    abort();
+    if (!gui_mode)
+       fputc(c, stream);
+    else
+    {
+       unsigned int msg_id = WM_STD_OUT_CHAR;
+       if (stream == stderr) msg_id = WM_STD_ERR_CHAR;
+       send_msg( (HWND)atoi(gui_hwnd), msg_id, (WPARAM)c );
+    }
+}
+
+static void tell_str(FILE *stream, char *str)
+{
+    unsigned int i;
+
+    for( i = 0; i < strlen(str); ++i )
+       tell_char(stream, str[i]);
+}
+
+static void tell_user(FILE *stream, char *fmt, ...)
+{
+    char str[0x100]; /* Make the size big enough */
+    va_list ap;
+    va_start(ap, fmt);
+    vsprintf(str, fmt, ap);
+    va_end(ap);
+    strcat(str, "\n");
+    tell_str(stream, str);
+}
+
+static void gui_update_stats(char *name, unsigned long size, int percentage, unsigned long elapsed)
+{
+    unsigned int i;
+
+    if (strcmp(name,statname) != 0)
+    {
+       for( i = 0; i < strlen(name); ++i )
+           send_msg( (HWND)atoi(gui_hwnd), WM_STATS_CHAR, (WPARAM)name[i]);
+       send_msg( (HWND)atoi(gui_hwnd), WM_STATS_CHAR, (WPARAM)'\n' );
+       strcpy(statname,name);
+    }
+    if (statsize != size)
+    {
+       send_msg( (HWND)atoi(gui_hwnd), WM_STATS_SIZE, (WPARAM)size );
+       statsize = size;
+    }
+    if (statelapsed != elapsed)
+    {
+       send_msg( (HWND)atoi(gui_hwnd), WM_STATS_ELAPSED, (WPARAM)elapsed );
+       statelapsed = elapsed;
+    }
+    if (statperct != percentage)
+    {
+       send_msg( (HWND)atoi(gui_hwnd), WM_STATS_PERCENT, (WPARAM)percentage );
+       statperct = percentage;
+    }
 }
 
 /*
@@ -49,64 +215,218 @@ void term_out(void)
  */
 void fatalbox(char *fmt, ...)
 {
+    char str[0x100]; /* Make the size big enough */
     va_list ap;
     va_start(ap, fmt);
-    fprintf(stderr, "Fatal: ");
-    vfprintf(stderr, fmt, ap);
-    fprintf(stderr, "\n");
+    strcpy(str, "Fatal:");
+    vsprintf(str+strlen(str), fmt, ap);
     va_end(ap);
+    strcat(str, "\n");
+    tell_str(stderr, str);
+
     exit(1);
 }
+void connection_fatal(char *fmt, ...)
+{
+    char str[0x100]; /* Make the size big enough */
+    va_list ap;
+    va_start(ap, fmt);
+    strcpy(str, "Fatal:");
+    vsprintf(str+strlen(str), fmt, ap);
+    va_end(ap);
+    strcat(str, "\n");
+    tell_str(stderr, str);
+
+    exit(1);
+}
+
+/*
+ * Be told what socket we're supposed to be using.
+ */
+static SOCKET scp_ssh_socket;
+char *do_select(SOCKET skt, int startup) {
+    if (startup)
+       scp_ssh_socket = skt;
+    else
+       scp_ssh_socket = INVALID_SOCKET;
+    return NULL;
+}
+extern int select_result(WPARAM, LPARAM);
+
+/*
+ * Receive a block of data from the SSH link. Block until all data
+ * is available.
+ *
+ * To do this, we repeatedly call the SSH protocol module, with our
+ * own trap in from_backend() to catch the data that comes back. We
+ * do this until we have enough data.
+ */
+
+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 */
+void from_backend(int is_stderr, char *data, int datalen) {
+    unsigned char *p = (unsigned char *)data;
+    unsigned len = (unsigned)datalen;
+
+    /*
+     * stderr data is just spouted to local stderr and otherwise
+     * ignored.
+     */
+    if (is_stderr) {
+       fwrite(data, 1, len, stderr);
+       return;
+    }
+
+    inbuf_head = 0;
+
+    /*
+     * If this is before the real session begins, just return.
+     */
+    if (!outptr)
+        return;
+
+    if (outlen > 0) {
+        unsigned used = outlen;
+        if (used > len) used = len;
+        memcpy(outptr, p, used);
+        outptr += used; outlen -= used;
+        p += used; len -= used;
+    }
+
+    if (len > 0) {
+        if (pendsize < pendlen + len) {
+            pendsize = pendlen + len + 4096;
+            pending = (pending ? srealloc(pending, pendsize) :
+                       smalloc(pendsize));
+            if (!pending)
+                fatalbox("Out of memory");
+        }
+        memcpy(pending+pendlen, p, len);
+        pendlen += len;
+    }
+}
+static int ssh_scp_recv(unsigned char *buf, int len) {
+    outptr = buf;
+    outlen = len;
+
+    /*
+     * See if the pending-input block contains some of what we
+     * need.
+     */
+    if (pendlen > 0) {
+        unsigned pendused = pendlen;
+        if (pendused > outlen)
+            pendused = outlen;
+       memcpy(outptr, pending, pendused);
+        memmove(pending, pending+pendused, pendlen-pendused);
+       outptr += pendused;
+       outlen -= pendused;
+        pendlen -= pendused;
+        if (pendlen == 0) {
+            pendsize = 0;
+            sfree(pending);
+            pending = NULL;
+        }
+        if (outlen == 0)
+            return len;
+    }
+
+    while (outlen > 0) {
+        fd_set readfds;
+
+        FD_ZERO(&readfds);
+        FD_SET(scp_ssh_socket, &readfds);
+        if (select(1, &readfds, NULL, NULL, NULL) < 0)
+            return 0;                  /* doom */
+        select_result((WPARAM)scp_ssh_socket, (LPARAM)FD_READ);
+    }
+
+    return len;
+}
+
+/*
+ * Loop through the ssh connection and authentication process.
+ */
+static void ssh_scp_init(void) {
+    if (scp_ssh_socket == INVALID_SOCKET)
+       return;
+    while (!back->sendok()) {
+        fd_set readfds;
+        FD_ZERO(&readfds);
+        FD_SET(scp_ssh_socket, &readfds);
+        if (select(1, &readfds, NULL, NULL, NULL) < 0)
+            return;                    /* doom */
+        select_result((WPARAM)scp_ssh_socket, (LPARAM)FD_READ);
+    }
+}
 
 /*
  *  Print an error message and exit after closing the SSH link.
  */
 static void bump(char *fmt, ...)
 {
+    char str[0x100]; /* Make the size big enough */
     va_list ap;
     va_start(ap, fmt);
-    fprintf(stderr, "Fatal: ");
-    vfprintf(stderr, fmt, ap);
-    fprintf(stderr, "\n");
+    strcpy(str, "Fatal:");
+    vsprintf(str+strlen(str), fmt, ap);
     va_end(ap);
-    if (connection_open) {
+    strcat(str, "\n");
+    tell_str(stderr, str);
+
+    if (back != NULL && back->socket() != NULL) {
        char ch;
-       ssh_scp_send_eof();
+       back->special(TS_EOF);
        ssh_scp_recv(&ch, 1);
     }
     exit(1);
 }
 
-static void get_password(const char *prompt, char *str, int maxlen)
+static int get_password(const char *prompt, char *str, int maxlen)
 {
     HANDLE hin, hout;
     DWORD savemode, i;
 
     if (password) {
-       strncpy(str, password, maxlen);
-       str[maxlen-1] = '\0';
-       password = NULL;
-       return;
+        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)
-       bump("Cannot get standard input/output handles");
+    /* GUI Adaptation - Sept 2000 */
+    if (gui_mode) {
+       if (maxlen>0) str[0] = '\0';
+    } else {
+       hin = GetStdHandle(STD_INPUT_HANDLE);
+       hout = GetStdHandle(STD_OUTPUT_HANDLE);
+       if (hin == INVALID_HANDLE_VALUE || hout == INVALID_HANDLE_VALUE)
+           bump("Cannot get standard input/output handles");
+
+       GetConsoleMode(hin, &savemode);
+       SetConsoleMode(hin, (savemode & (~ENABLE_ECHO_INPUT)) |
+                      ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT);
 
-    GetConsoleMode(hin, &savemode);
-    SetConsoleMode(hin, (savemode & (~ENABLE_ECHO_INPUT)) |
-                  ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT);
+       WriteFile(hout, prompt, strlen(prompt), &i, NULL);
+       ReadFile(hin, str, maxlen-1, &i, NULL);
 
-    WriteFile(hout, prompt, strlen(prompt), &i, NULL);
-    ReadFile(hin, str, maxlen-1, &i, NULL);
+       SetConsoleMode(hin, savemode);
 
-    SetConsoleMode(hin, savemode);
+       if ((int)i > maxlen) i = maxlen-1; else i = i - 2;
+       str[i] = '\0';
 
-    if ((int)i > maxlen) i = maxlen-1; else i = i - 2;
-    str[i] = '\0';
+       WriteFile(hout, "\r\n", 2, &i, NULL);
+    }
 
-    WriteFile(hout, "\r\n", 2, &i, NULL);
+    return 1;
 }
 
 /*
@@ -115,14 +435,16 @@ static void get_password(const char *prompt, char *str, int maxlen)
 static void do_cmd(char *host, char *user, char *cmd)
 {
     char *err, *realhost;
+    DWORD namelen;
 
     if (host == NULL || host[0] == '\0')
        bump("Empty host name");
 
     /* Try to load settings for this host */
-    do_defaults(host);
+    do_defaults(host, &cfg);
     if (cfg.host[0] == '\0') {
        /* No settings for this host; use defaults */
+        do_defaults(NULL, &cfg);
        strncpy(cfg.host, host, sizeof(cfg.host)-1);
        cfg.host[sizeof(cfg.host)-1] = '\0';
        cfg.port = 22;
@@ -133,7 +455,15 @@ static void do_cmd(char *host, char *user, char *cmd)
        strncpy(cfg.username, user, sizeof(cfg.username)-1);
        cfg.username[sizeof(cfg.username)-1] = '\0';
     } else if (cfg.username[0] == '\0') {
-       bump("Empty user name");
+       namelen = 0;
+       if (GetUserName(user, &namelen) == FALSE)
+           bump("Empty user name");
+       user = smalloc(namelen * sizeof(char));
+       GetUserName(user, &namelen);
+       if (verbose) tell_user(stderr, "Guessing user name: %s", user);
+       strncpy(cfg.username, user, sizeof(cfg.username)-1);
+       cfg.username[sizeof(cfg.username)-1] = '\0';
+       free(user);
     }
 
     if (cfg.protocol != PROT_SSH)
@@ -142,13 +472,18 @@ static void do_cmd(char *host, char *user, char *cmd)
     if (portnumber)
        cfg.port = portnumber;
 
-    err = ssh_scp_init(cfg.host, cfg.port, cmd, &realhost);
+    strncpy(cfg.remote_cmd, cmd, sizeof(cfg.remote_cmd));
+    cfg.remote_cmd[sizeof(cfg.remote_cmd)-1] = '\0';
+    cfg.nopty = TRUE;
+
+    back = &ssh_backend;
+
+    err = back->init(cfg.host, cfg.port, &realhost);
     if (err != NULL)
        bump("ssh_init: %s", err);
+    ssh_scp_init();
     if (verbose && realhost != NULL)
-       fprintf(stderr, "Connected to %s\n", realhost);
-
-    connection_open = 1;
+       tell_user(stderr, "Connected to %s\n", realhost);
 }
 
 /*
@@ -162,26 +497,32 @@ static void print_stats(char *name, unsigned long size, unsigned long done,
     char etastr[10];
     int pct;
 
-    if (now > start)
-       ratebs = (float) done / (now - start);
-    else
-       ratebs = (float) done;
+    /* GUI Adaptation - Sept 2000 */
+    if (gui_mode)
+       gui_update_stats(name, size, (int)(100 * (done*1.0/size)),
+                         (unsigned long)difftime(now, start));
+    else {
+       if (now > start)
+           ratebs = (float) done / (now - start);
+       else
+           ratebs = (float) done;
 
-    if (ratebs < 1.0)
-       eta = size - done;
-    else
-       eta = (unsigned long) ((size - done) / ratebs);
-    sprintf(etastr, "%02ld:%02ld:%02ld",
-           eta / 3600, (eta % 3600) / 60, eta % 60);
+       if (ratebs < 1.0)
+           eta = size - done;
+       else
+           eta = (unsigned long) ((size - done) / ratebs);
+       sprintf(etastr, "%02ld:%02ld:%02ld",
+               eta / 3600, (eta % 3600) / 60, eta % 60);
 
-    pct = (int) (100.0 * (float) done / size);
+       pct = (int) (100.0 * (float) done / size);
 
-    printf("\r%-25.25s | %10ld kB | %5.1f kB/s | ETA: %8s | %3d%%",
-          name, done / 1024, ratebs / 1024.0,
-          etastr, pct);
+       printf("\r%-25.25s | %10ld kB | %5.1f kB/s | ETA: %8s | %3d%%",
+              name, done / 1024, ratebs / 1024.0,
+              etastr, pct);
 
-    if (done == size)
-       printf("\n");
+       if (done == size)
+           printf("\n");
+    }
 }
 
 /*
@@ -236,7 +577,7 @@ static int response(void)
        } while (p < sizeof(rbuf) && ch != '\n');
        rbuf[p-1] = '\0';
        if (resp == 1)
-           fprintf(stderr, "%s\n", rbuf);
+           tell_user(stderr, "%s\n", rbuf);
        else
            bump("%s", rbuf);
        errs++;
@@ -254,12 +595,11 @@ static void run_err(const char *fmt, ...)
     va_list ap;
     va_start(ap, fmt);
     errs++;
-    strcpy(str, "\01scp: ");
+    strcpy(str, "scp: ");
     vsprintf(str+strlen(str), fmt, ap);
     strcat(str, "\n");
-    ssh_scp_send(str, strlen(str));
-    vfprintf(stderr, fmt, ap);
-    fprintf(stderr, "\n");
+    back->send(str, strlen(str));
+    tell_user(stderr, "%s",str);
     va_end(ap);
 }
 
@@ -329,7 +669,7 @@ static void source(char *src)
        TIME_WIN_TO_POSIX(actime, atime);
        TIME_WIN_TO_POSIX(wrtime, mtime);
        sprintf(buf, "T%lu 0 %lu 0\n", mtime, atime);
-       ssh_scp_send(buf, strlen(buf));
+       back->send(buf, strlen(buf));
        if (response())
            return;
     }
@@ -337,8 +677,8 @@ static void source(char *src)
     size = GetFileSize(f, NULL);
     sprintf(buf, "C0644 %lu %s\n", size, last);
     if (verbose)
-       fprintf(stderr, "Sending file modes: %s", buf);
-    ssh_scp_send(buf, strlen(buf));
+       tell_user(stderr, "Sending file modes: %s", buf);
+    back->send(buf, strlen(buf));
     if (response())
        return;
 
@@ -356,7 +696,7 @@ static void source(char *src)
            if (statistics) printf("\n");
            bump("%s: Read error", src);
        }
-       ssh_scp_send(transbuf, k);
+       back->send(transbuf, k);
        if (statistics) {
            stat_bytes += k;
            if (time(NULL) != stat_lasttime ||
@@ -369,7 +709,7 @@ static void source(char *src)
     }
     CloseHandle(f);
 
-    ssh_scp_send("", 1);
+    back->send("", 1);
     (void) response();
 }
 
@@ -397,8 +737,8 @@ static void rsource(char *src)
 
     sprintf(buf, "D0755 0 %s\n", last);
     if (verbose)
-       fprintf(stderr, "Entering directory: %s", buf);
-    ssh_scp_send(buf, strlen(buf));
+       tell_user(stderr, "Entering directory: %s", buf);
+    back->send(buf, strlen(buf));
     if (response())
        return;
 
@@ -420,14 +760,14 @@ static void rsource(char *src)
     FindClose(dir);
 
     sprintf(buf, "E\n");
-    ssh_scp_send(buf, strlen(buf));
+    back->send(buf, strlen(buf));
     (void) response();
 }
 
 /*
  *  Execute the sink part of the SCP protocol.
  */
-static void sink(char *targ)
+static void sink(char *targ, char *src)
 {
     char buf[2048];
     char namebuf[2048];
@@ -452,7 +792,7 @@ static void sink(char *targ)
     if (targetshouldbedirectory && !targisdir)
        bump("%s: Not a directory", targ);
 
-    ssh_scp_send("", 1);
+    back->send("", 1);
     while (1) {
        settime = 0;
        gottime:
@@ -470,19 +810,19 @@ static void sink(char *targ)
        buf[i-1] = '\0';
        switch (buf[0]) {
          case '\01':   /* error */
-           fprintf(stderr, "%s\n", buf+1);
+           tell_user(stderr, "%s\n", buf+1);
            errs++;
            continue;
          case '\02':   /* fatal error */
            bump("%s", buf+1);
          case 'E':
-           ssh_scp_send("", 1);
+           back->send("", 1);
            return;
          case 'T':
            if (sscanf(buf, "T%ld %*d %ld %*d",
                       &mtime, &atime) == 2) {
                settime = 1;
-               ssh_scp_send("", 1);
+               back->send("", 1);
                goto gottime;
            }
            bump("Protocol error: Illegal time format");
@@ -495,12 +835,17 @@ static void sink(char *targ)
 
        if (sscanf(buf+1, "%u %lu %[^\n]", &mode, &size, namebuf) != 3)
            bump("Protocol error: Illegal file descriptor format");
+       /* Security fix: ensure the file ends up where we asked for it. */
        if (targisdir) {
            char t[2048];
+           char *p;
            strcpy(t, targ);
            if (targ[0] != '\0')
                strcat(t, "/");
-           strcat(t, namebuf);
+           p = namebuf + strlen(namebuf);
+           while (p > namebuf && p[-1] != '/' && p[-1] != '\\')
+               p--;
+           strcat(t, p);
            strcpy(namebuf, t);
        } else {
            strcpy(namebuf, targ);
@@ -520,7 +865,7 @@ static void sink(char *targ)
                    continue;
                }
            }
-           sink(namebuf);
+           sink(namebuf, NULL);
            /* can we set the timestamp for directories ? */
            continue;
        }
@@ -532,7 +877,7 @@ static void sink(char *targ)
            continue;
        }
 
-       ssh_scp_send("", 1);
+       back->send("", 1);
 
        if (statistics) {
            stat_bytes = 0;
@@ -585,7 +930,7 @@ static void sink(char *targ)
            run_err("%s: Write error", namebuf);
            continue;
        }
-       ssh_scp_send("", 1);
+       back->send("", 1);
     }
 }
 
@@ -654,8 +999,7 @@ static void toremote(int argc, char *argv[])
        WIN32_FIND_DATA fdat;
        src = argv[i];
        if (colon(src) != NULL) {
-           fprintf(stderr,
-                   "%s: Remote to remote not supported\n", src);
+           tell_user(stderr, "%s: Remote to remote not supported\n", src);
            errs++;
            continue;
        }
@@ -667,9 +1011,30 @@ static void toremote(int argc, char *argv[])
        do {
            char *last;
            char namebuf[2048];
+           /*
+            * Ensure that . and .. are never matched by wildcards,
+            * but only by deliberate action.
+            */
+           if (!strcmp(fdat.cFileName, ".") ||
+               !strcmp(fdat.cFileName, "..")) {
+               /*
+                * Find*File has returned a special dir. We require
+                * that _either_ `src' ends in a backslash followed
+                * by that string, _or_ `src' is precisely that
+                * string.
+                */
+               int len = strlen(src), dlen = strlen(fdat.cFileName);
+               if (len == dlen && !strcmp(src, fdat.cFileName)) {
+                   /* ok */;
+               } else if (len > dlen+1 && src[len-dlen-1] == '\\' &&
+                          !strcmp(src+len-dlen, fdat.cFileName)) {
+                   /* ok */;
+               } else
+                   continue;          /* ignore this one */
+           }
            if (strlen(src) + strlen(fdat.cFileName) >=
                sizeof(namebuf)) {
-               fprintf(stderr, "%s: Name too long", src);
+               tell_user(stderr, "%s: Name too long", src);
                continue;
            }
            strcpy(namebuf, src);
@@ -734,7 +1099,7 @@ static void tolocal(int argc, char *argv[])
     do_cmd(host, user, cmd);
     sfree(cmd);
 
-    sink(targ);
+    sink(targ, src);
 }
 
 /*
@@ -782,12 +1147,12 @@ static void get_dir_list(int argc, char *argv[])
     }
     *p++ = '\'';
     *p = '\0';
-    
+
     do_cmd(host, user, cmd);
     sfree(cmd);
 
     while (ssh_scp_recv(&c, 1) > 0)
-       fputc(c, stdout);              /* thank heavens for buffered I/O */
+       tell_char(stdout, c);
 }
 
 /*
@@ -823,6 +1188,8 @@ static void usage(void)
     printf("  -v        show verbose messages\n");
     printf("  -P port   connect to specified port\n");
     printf("  -pw passw login with specified password\n");
+    /* GUI Adaptation - Sept 2000 */
+    printf("  -gui hWnd GUI mode with the windows handle for receiving messages\n");
     exit(1);
 }
 
@@ -836,15 +1203,16 @@ int main(int argc, char *argv[])
 
     default_protocol = PROT_TELNET;
 
-    scp_flags = SCP_FLAG;
+    flags = FLAG_STDERR;
     ssh_get_password = &get_password;
     init_winsock();
+    sk_init();
 
     for (i = 1; i < argc; i++) {
        if (argv[i][0] != '-')
            break;
        if (strcmp(argv[i], "-v") == 0)
-           verbose = 1, scp_flags |= SCP_VERBOSE;
+           verbose = 1, flags |= FLAG_VERBOSE;
        else if (strcmp(argv[i], "-r") == 0)
            recursive = 1;
        else if (strcmp(argv[i], "-p") == 0)
@@ -858,8 +1226,11 @@ int main(int argc, char *argv[])
            portnumber = atoi(argv[++i]);
        else if (strcmp(argv[i], "-pw") == 0 && i+1 < argc)
            password = argv[++i];
-       else if (strcmp(argv[i], "-ls") == 0)
-           list = 1;
+       else if (strcmp(argv[i], "-gui") == 0 && i+1 < argc) {
+           gui_hwnd = argv[++i];
+           gui_mode = 1;
+       } else if (strcmp(argv[i], "-ls") == 0)
+               list = 1;
        else if (strcmp(argv[i], "--") == 0)
        { i++; break; }
        else
@@ -867,6 +1238,7 @@ int main(int argc, char *argv[])
     }
     argc -= i;
     argv += i;
+    back = NULL;
 
     if (list) {
        if (argc != 1)
@@ -886,16 +1258,22 @@ int main(int argc, char *argv[])
            tolocal(argc, argv);
     }
 
-    if (connection_open) {
+    if (back != NULL && back->socket() != NULL) {
        char ch;
-       ssh_scp_send_eof();
+       back->special(TS_EOF);
        ssh_scp_recv(&ch, 1);
     }
     WSACleanup();
     random_save_seed();
 
+    /* GUI Adaptation - August 2000 */
+    if (gui_mode) {
+       unsigned int msg_id = WM_RET_ERR_CNT;
+       if (list) msg_id = WM_LS_RET_ERR_CNT;
+       while (!PostMessage( (HWND)atoi(gui_hwnd), msg_id, (WPARAM)errs, 0/*lParam*/ ) )
+           SleepEx(1000,TRUE);
+    }
     return (errs == 0 ? 0 : 1);
 }
 
 /* end */
-