X-Git-Url: https://git.distorted.org.uk/u/mdw/putty/blobdiff_plain/d74d141c2daed084c8a62c5dc5b88e801b81ee5a..0622ea7168aa265db70be885fbffb050f29f0205:/plink.c

diff --git a/plink.c b/plink.c
index 4113a812..047414ad 100644
--- a/plink.c
+++ b/plink.c
@@ -15,6 +15,8 @@
 #include "storage.h"
 #include "tree234.h"
 
+#define MAX_STDIN_BACKLOG 4096
+
 void fatalbox(char *p, ...)
 {
     va_list ap;
@@ -40,105 +42,11 @@ void connection_fatal(char *p, ...)
 
 static char *password = NULL;
 
-void logevent(char *string)
-{
-}
-
-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);
-    }
-}
-
 HANDLE inhandle, outhandle, errhandle;
 DWORD orig_console_mode;
 
 WSAEVENT netevent;
 
-void from_backend(int is_stderr, char *data, int len)
-{
-    int pos;
-    DWORD ret;
-    HANDLE h = (is_stderr ? errhandle : outhandle);
-
-    pos = 0;
-    while (pos < len) {
-	if (!WriteFile(h, data + pos, len - pos, &ret, NULL))
-	    return;		       /* give up in panic */
-	pos += ret;
-    }
-}
-
 int term_ldisc(int mode)
 {
     return FALSE;
@@ -166,56 +74,6 @@ struct input_data {
     HANDLE event, eventback;
 };
 
-static int get_line(const char *prompt, char *str, int maxlen, int is_pw)
-{
-    HANDLE hin, hout;
-    DWORD savemode, newmode, i;
-
-    if (is_pw && 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");
-	return 0;
-    }
-
-    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;
-}
-
 static DWORD WINAPI stdin_read_thread(void *param)
 {
     struct input_data *idata = (struct input_data *) param;
@@ -235,6 +93,74 @@ static DWORD WINAPI stdin_read_thread(void *param)
     return 0;
 }
 
+struct output_data {
+    DWORD len, lenwritten;
+    int writeret;
+    char *buffer;
+    int is_stderr, done;
+    HANDLE event, eventback;
+    int busy;
+};
+
+static DWORD WINAPI stdout_write_thread(void *param)
+{
+    struct output_data *odata = (struct output_data *) param;
+    HANDLE outhandle, errhandle;
+
+    outhandle = GetStdHandle(STD_OUTPUT_HANDLE);
+    errhandle = GetStdHandle(STD_ERROR_HANDLE);
+
+    while (1) {
+	WaitForSingleObject(odata->eventback, INFINITE);
+	if (odata->done)
+	    break;
+	odata->writeret =
+	    WriteFile(odata->is_stderr ? errhandle : outhandle,
+		      odata->buffer, odata->len, &odata->lenwritten, NULL);
+	SetEvent(odata->event);
+    }
+
+    return 0;
+}
+
+bufchain stdout_data, stderr_data;
+struct output_data odata, edata;
+
+void try_output(int is_stderr)
+{
+    struct output_data *data = (is_stderr ? &edata : &odata);
+    void *senddata;
+    int sendlen;
+
+    if (!data->busy) {
+	bufchain_prefix(is_stderr ? &stderr_data : &stdout_data,
+			&senddata, &sendlen);
+	data->buffer = senddata;
+	data->len = sendlen;
+	SetEvent(data->eventback);
+	data->busy = 1;
+    }
+}
+
+int from_backend(int is_stderr, char *data, int len)
+{
+    HANDLE h = (is_stderr ? errhandle : outhandle);
+    int osize, esize;
+
+    if (is_stderr) {
+	bufchain_add(&stderr_data, data, len);
+	try_output(1);
+    } else {
+	bufchain_add(&stdout_data, data, len);
+	try_output(0);
+    }
+
+    osize = bufchain_size(&stdout_data);
+    esize = bufchain_size(&stderr_data);
+
+    return osize + esize;
+}
+
 /*
  *  Short description of parameters.
  */
@@ -243,12 +169,17 @@ static void usage(void)
     printf("PuTTY Link: command-line connection utility\n");
     printf("%s\n", ver);
     printf("Usage: plink [options] [user@]host [command]\n");
+    printf("       (\"host\" can also be a PuTTY saved session name)\n");
     printf("Options:\n");
     printf("  -v        show verbose messages\n");
     printf("  -ssh      force use of ssh protocol\n");
     printf("  -P port   connect to specified port\n");
     printf("  -pw passw login with specified password\n");
     printf("  -m file   read remote command(s) from file\n");
+    printf("  -L listen-port:host:port   Forward local port to "
+	   "remote address\n");
+    printf("  -R listen-port:host:port   Forward remote port to"
+	   " local address\n");
     exit(1);
 }
 
@@ -256,7 +187,8 @@ char *do_select(SOCKET skt, int startup)
 {
     int events;
     if (startup) {
-	events = FD_READ | FD_WRITE | FD_OOB | FD_CLOSE | FD_ACCEPT;
+	events = (FD_CONNECT | FD_READ | FD_WRITE |
+		  FD_OOB | FD_CLOSE | FD_ACCEPT);
     } else {
 	events = 0;
     }
@@ -275,17 +207,20 @@ int main(int argc, char **argv)
 {
     WSADATA wsadata;
     WORD winsock_ver;
-    WSAEVENT stdinevent;
-    HANDLE handles[2];
-    DWORD threadid;
+    WSAEVENT stdinevent, stdoutevent, stderrevent;
+    HANDLE handles[4];
+    DWORD in_threadid, out_threadid, err_threadid;
     struct input_data idata;
+    int reading;
     int sending;
     int portnumber = -1;
     SOCKET *sklist;
     int skcount, sksize;
     int connopen;
+    int exitcode;
+    char extra_portfwd[sizeof(cfg.portfwd)];
 
-    ssh_get_line = get_line;
+    ssh_get_line = console_get_line;
 
     sklist = NULL;
     skcount = sksize = 0;
@@ -329,19 +264,42 @@ int main(int argc, char **argv)
 	    } else if (!strcmp(p, "-telnet")) {
 		default_protocol = cfg.protocol = PROT_TELNET;
 		default_port = cfg.port = 23;
+	    } else if (!strcmp(p, "-rlogin")) {
+		default_protocol = cfg.protocol = PROT_RLOGIN;
+		default_port = cfg.port = 513;
 	    } else if (!strcmp(p, "-raw")) {
 		default_protocol = cfg.protocol = PROT_RAW;
+	    } else if (!strcmp(p, "-batch")) {
+		console_batch_mode = TRUE;
 	    } else if (!strcmp(p, "-v")) {
 		flags |= FLAG_VERBOSE;
 	    } else if (!strcmp(p, "-log")) {
 		logfile = "putty.log";
 	    } else if (!strcmp(p, "-pw") && argc > 1) {
-		--argc, password = *++argv;
+		--argc, console_password = *++argv;
 	    } else if (!strcmp(p, "-l") && argc > 1) {
 		char *username;
 		--argc, username = *++argv;
 		strncpy(cfg.username, username, sizeof(cfg.username));
 		cfg.username[sizeof(cfg.username) - 1] = '\0';
+	    } else if ((!strcmp(p, "-L") || !strcmp(p, "-R")) && argc > 1) {
+		char *fwd, *ptr, *q;
+		int i=0;
+		--argc, fwd = *++argv;
+		ptr = extra_portfwd;
+		/* if multiple forwards, find end of list */
+		if (ptr[0]=='R' || ptr[0]=='L') {
+		    for (i = 0; i < sizeof(extra_portfwd) - 2; i++)
+			if (ptr[i]=='\000' && ptr[i+1]=='\000')
+			    break;
+		    ptr = ptr + i + 1;  /* point to next forward slot */
+		}
+		ptr[0] = p[1];  /* insert a 'L' or 'R' at the start */
+		strncpy(ptr+1, fwd, sizeof(extra_portfwd) - i);
+		q = strchr(ptr, ':');
+		if (q) *q = '\t';      /* replace first : with \t */
+		ptr[strlen(ptr)+1] = '\000';    /* append two '\000' */
+		extra_portfwd[sizeof(extra_portfwd) - 1] = '\0';
 	    } else if (!strcmp(p, "-m") && argc > 1) {
 		char *filename, *command;
 		int cmdlen, cmdsize;
@@ -370,6 +328,7 @@ int main(int argc, char **argv)
 		    command[cmdlen++] = d;
 		} while (c != EOF);
 		cfg.remote_cmd_ptr = command;
+		cfg.remote_cmd_ptr2 = NULL;
 		cfg.nopty = TRUE;      /* command => no terminal */
 	    } else if (!strcmp(p, "-P") && argc > 1) {
 		--argc, portnumber = atoi(*++argv);
@@ -491,6 +450,32 @@ int main(int argc, char **argv)
 	usage();
     }
 
+    /*
+     * 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';
+
     if (!*cfg.remote_cmd_ptr)
 	flags |= FLAG_INTERACTIVE;
 
@@ -514,6 +499,30 @@ int main(int argc, char **argv)
     }
 
     /*
+     * Add extra port forwardings (accumulated on command line) to
+     * cfg.
+     */
+    {
+	int i;
+	char *p;
+	p = extra_portfwd;
+	i = 0;
+	while (cfg.portfwd[i])
+	    i += strlen(cfg.portfwd+i) + 1;
+	while (*p) {
+	    if (strlen(p)+2 > sizeof(cfg.portfwd)-i) {
+		fprintf(stderr, "Internal fault: not enough space for all"
+			" port forwardings\n");
+		break;
+	    }
+	    strncpy(cfg.portfwd+i, p, sizeof(cfg.portfwd)-i-1);
+	    i += strlen(cfg.portfwd+i) + 1;
+	    cfg.portfwd[i] = '\0';
+	    p += strlen(p)+1;
+	}
+    }
+
+    /*
      * Select port.
      */
     if (portnumber != -1)
@@ -543,8 +552,11 @@ int main(int argc, char **argv)
     {
 	char *error;
 	char *realhost;
+	/* nodelay is only useful if stdin is a character device (console) */
+	int nodelay = cfg.tcp_nodelay &&
+	    (GetFileType(GetStdHandle(STD_INPUT_HANDLE)) == FILE_TYPE_CHAR);
 
-	error = back->init(cfg.host, cfg.port, &realhost);
+	error = back->init(cfg.host, cfg.port, &realhost, nodelay);
 	if (error) {
 	    fprintf(stderr, "Unable to open connection:\n%s", error);
 	    return 1;
@@ -554,6 +566,8 @@ int main(int argc, char **argv)
     connopen = 1;
 
     stdinevent = CreateEvent(NULL, FALSE, FALSE, NULL);
+    stdoutevent = CreateEvent(NULL, FALSE, FALSE, NULL);
+    stderrevent = CreateEvent(NULL, FALSE, FALSE, NULL);
 
     inhandle = GetStdHandle(STD_INPUT_HANDLE);
     outhandle = GetStdHandle(STD_OUTPUT_HANDLE);
@@ -568,7 +582,33 @@ int main(int argc, char **argv)
      */
     handles[0] = netevent;
     handles[1] = stdinevent;
+    handles[2] = stdoutevent;
+    handles[3] = stderrevent;
     sending = FALSE;
+
+    /*
+     * Create spare threads to write to stdout and stderr, so we
+     * can arrange asynchronous writes.
+     */
+    odata.event = stdoutevent;
+    odata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
+    odata.is_stderr = 0;
+    odata.busy = odata.done = 0;
+    if (!CreateThread(NULL, 0, stdout_write_thread,
+		      &odata, 0, &out_threadid)) {
+	fprintf(stderr, "Unable to create output thread\n");
+	exit(1);
+    }
+    edata.event = stderrevent;
+    edata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
+    edata.is_stderr = 1;
+    edata.busy = edata.done = 0;
+    if (!CreateThread(NULL, 0, stdout_write_thread,
+		      &edata, 0, &err_threadid)) {
+	fprintf(stderr, "Unable to create error output thread\n");
+	exit(1);
+    }
+
     while (1) {
 	int n;
 
@@ -592,14 +632,14 @@ int main(int argc, char **argv)
 	    idata.event = stdinevent;
 	    idata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
 	    if (!CreateThread(NULL, 0, stdin_read_thread,
-			      &idata, 0, &threadid)) {
-		fprintf(stderr, "Unable to create second thread\n");
+			      &idata, 0, &in_threadid)) {
+		fprintf(stderr, "Unable to create input thread\n");
 		exit(1);
 	    }
 	    sending = TRUE;
 	}
 
-	n = WaitForMultipleObjects(2, handles, FALSE, INFINITE);
+	n = WaitForMultipleObjects(4, handles, FALSE, INFINITE);
 	if (n == 0) {
 	    WSANETWORKEVENTS things;
 	    SOCKET socket;
@@ -641,6 +681,8 @@ int main(int argc, char **argv)
 		if (!WSAEnumNetworkEvents(socket, NULL, &things)) {
 		    noise_ultralight(socket);
 		    noise_ultralight(things.lNetworkEvents);
+		    if (things.lNetworkEvents & FD_CONNECT)
+			connopen &= select_result(wp, (LPARAM) FD_CONNECT);
 		    if (things.lNetworkEvents & FD_READ)
 			connopen &= select_result(wp, (LPARAM) FD_READ);
 		    if (things.lNetworkEvents & FD_CLOSE)
@@ -655,17 +697,56 @@ int main(int argc, char **argv)
 		}
 	    }
 	} else if (n == 1) {
+	    reading = 0;
 	    noise_ultralight(idata.len);
-	    if (idata.len > 0) {
-		back->send(idata.buffer, idata.len);
-	    } else {
-		back->special(TS_EOF);
+	    if (connopen && back->socket() != NULL) {
+		if (idata.len > 0) {
+		    back->send(idata.buffer, idata.len);
+		} else {
+		    back->special(TS_EOF);
+		}
+	    }
+	} else if (n == 2) {
+	    odata.busy = 0;
+	    if (!odata.writeret) {
+		fprintf(stderr, "Unable to write to standard output\n");
+		exit(0);
 	    }
+	    bufchain_consume(&stdout_data, odata.lenwritten);
+	    if (bufchain_size(&stdout_data) > 0)
+		try_output(0);
+	    if (connopen && back->socket() != NULL) {
+		back->unthrottle(bufchain_size(&stdout_data) +
+				 bufchain_size(&stderr_data));
+	    }
+	} else if (n == 3) {
+	    edata.busy = 0;
+	    if (!edata.writeret) {
+		fprintf(stderr, "Unable to write to standard output\n");
+		exit(0);
+	    }
+	    bufchain_consume(&stderr_data, edata.lenwritten);
+	    if (bufchain_size(&stderr_data) > 0)
+		try_output(1);
+	    if (connopen && back->socket() != NULL) {
+		back->unthrottle(bufchain_size(&stdout_data) +
+				 bufchain_size(&stderr_data));
+	    }
+	}
+	if (!reading && back->sendbuffer() < MAX_STDIN_BACKLOG) {
 	    SetEvent(idata.eventback);
+	    reading = 1;
 	}
-	if (!connopen || back->socket() == NULL)
+	if ((!connopen || back->socket() == NULL) &&
+	    bufchain_size(&stdout_data) == 0 &&
+	    bufchain_size(&stderr_data) == 0)
 	    break;		       /* we closed the connection */
     }
     WSACleanup();
-    return 0;
+    exitcode = back->exitcode();
+    if (exitcode < 0) {
+	fprintf(stderr, "Remote process exit code unavailable\n");
+	exitcode = 1;		       /* this is an error condition */
+    }
+    return exitcode;
 }