[u/mdw/putty] / plink.c

/*
 * PLink - a command-line (stdin/stdout) variant of PuTTY.
 */

#ifndef AUTO_WINSOCK
#include <winsock2.h>
#endif
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <stdarg.h>

#define PUTTY_DO_GLOBALS	       /* actually _define_ globals */
#include "putty.h"
#include "storage.h"
#include "tree234.h"

#define MAX_STDIN_BACKLOG 4096

void fatalbox(char *p, ...)
{
    va_list ap;
    fprintf(stderr, "FATAL ERROR: ");
    va_start(ap, p);
    vfprintf(stderr, p, ap);
    va_end(ap);
    fputc('\n', stderr);
    WSACleanup();
    cleanup_exit(1);
}
void modalfatalbox(char *p, ...)
{
    va_list ap;
    fprintf(stderr, "FATAL ERROR: ");
    va_start(ap, p);
    vfprintf(stderr, p, ap);
    va_end(ap);
    fputc('\n', stderr);
    WSACleanup();
    cleanup_exit(1);
}
void connection_fatal(char *p, ...)
{
    va_list ap;
    fprintf(stderr, "FATAL ERROR: ");
    va_start(ap, p);
    vfprintf(stderr, p, ap);
    va_end(ap);
    fputc('\n', stderr);
    WSACleanup();
    cleanup_exit(1);
}
void cmdline_error(char *p, ...)
{
    va_list ap;
    fprintf(stderr, "plink: ");
    va_start(ap, p);
    vfprintf(stderr, p, ap);
    va_end(ap);
    fputc('\n', stderr);
    exit(1);
}

static char *password = NULL;

HANDLE inhandle, outhandle, errhandle;
DWORD orig_console_mode;

WSAEVENT netevent;

int term_ldisc(int mode)
{
    return FALSE;
}
void ldisc_update(int echo, int edit)
{
    /* Update stdin read mode to reflect changes in line discipline. */
    DWORD mode;

    mode = ENABLE_PROCESSED_INPUT;
    if (echo)
	mode = mode | ENABLE_ECHO_INPUT;
    else
	mode = mode & ~ENABLE_ECHO_INPUT;
    if (edit)
	mode = mode | ENABLE_LINE_INPUT;
    else
	mode = mode & ~ENABLE_LINE_INPUT;
    SetConsoleMode(inhandle, mode);
}

struct input_data {
    DWORD len;
    char buffer[4096];
    HANDLE event, eventback;
};

static DWORD WINAPI stdin_read_thread(void *param)
{
    struct input_data *idata = (struct input_data *) param;
    HANDLE inhandle;

    inhandle = GetStdHandle(STD_INPUT_HANDLE);

    while (ReadFile(inhandle, idata->buffer, sizeof(idata->buffer),
		    &idata->len, NULL) && idata->len > 0) {
	SetEvent(idata->event);
	WaitForSingleObject(idata->eventback, INFINITE);
    }

    idata->len = 0;
    SetEvent(idata->event);

    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;

    assert(len > 0);

    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.
 */
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("  -load sessname  Load settings from saved session\n");
    printf("  -ssh -telnet -rlogin -raw\n");
    printf("            force use of a particular protocol (default SSH)\n");
    printf("  -P port   connect to specified port\n");
    printf("  -l user   connect with specified username\n");
    printf("  -m file   read remote command(s) from file\n");
    printf("  -batch    disable all interactive prompts\n");
    printf("The following options only apply to SSH connections:\n");
    printf("  -pw passw login with specified password\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");
    printf("  -X -x     enable / disable X11 forwarding\n");
    printf("  -A -a     enable / disable agent forwarding\n");
    printf("  -t -T     enable / disable pty allocation\n");
    printf("  -1 -2     force use of particular protocol version\n");
    printf("  -C        enable compression\n");
    printf("  -i key    private key file for authentication\n");
    exit(1);
}

char *do_select(SOCKET skt, int startup)
{
    int events;
    if (startup) {
	events = (FD_CONNECT | FD_READ | FD_WRITE |
		  FD_OOB | FD_CLOSE | FD_ACCEPT);
    } else {
	events = 0;
    }
    if (WSAEventSelect(skt, netevent, events) == SOCKET_ERROR) {
	switch (WSAGetLastError()) {
	  case WSAENETDOWN:
	    return "Network is down";
	  default:
	    return "WSAAsyncSelect(): unknown error";
	}
    }
    return NULL;
}

int main(int argc, char **argv)
{
    WSADATA wsadata;
    WORD winsock_ver;
    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;

    ssh_get_line = console_get_line;

    sklist = NULL;
    skcount = sksize = 0;
    /*
     * Initialise port and protocol to sensible defaults. (These
     * will be overridden by more or less anything.)
     */
    default_protocol = PROT_SSH;
    default_port = 22;

    flags = FLAG_STDERR;
    /*
     * Process the command line.
     */
    do_defaults(NULL, &cfg);
    default_protocol = cfg.protocol;
    default_port = cfg.port;
    {
	/*
	 * Override the default protocol if PLINK_PROTOCOL is set.
	 */
	char *p = getenv("PLINK_PROTOCOL");
	int i;
	if (p) {
	    for (i = 0; backends[i].backend != NULL; i++) {
		if (!strcmp(backends[i].name, p)) {
		    default_protocol = cfg.protocol = backends[i].protocol;
		    default_port = cfg.port =
			backends[i].backend->default_port;
		    break;
		}
	    }
	}
    }
    while (--argc) {
	char *p = *++argv;
	if (*p == '-') {
	    int ret = cmdline_process_param(p, (argc > 1 ? argv[1] : NULL), 1);
	    if (ret == -2) {
		fprintf(stderr,
			"plink: option \"%s\" requires an argument\n", p);
	    } else if (ret == 2) {
		--argc, ++argv;
	    } else if (ret == 1) {
		continue;
	    } else if (!strcmp(p, "-batch")) {
		console_batch_mode = 1;
	    }
	} else if (*p) {
	    if (!*cfg.host) {
		char *q = p;
		/*
		 * If the hostname starts with "telnet:", set the
		 * protocol to Telnet and process the string as a
		 * Telnet URL.
		 */
		if (!strncmp(q, "telnet:", 7)) {
		    char c;

		    q += 7;
		    if (q[0] == '/' && q[1] == '/')
			q += 2;
		    cfg.protocol = PROT_TELNET;
		    p = q;
		    while (*p && *p != ':' && *p != '/')
			p++;
		    c = *p;
		    if (*p)
			*p++ = '\0';
		    if (c == ':')
			cfg.port = atoi(p);
		    else
			cfg.port = -1;
		    strncpy(cfg.host, q, sizeof(cfg.host) - 1);
		    cfg.host[sizeof(cfg.host) - 1] = '\0';
		} else {
		    char *r;
		    /*
		     * Before we process the [user@]host string, we
		     * first check for the presence of a protocol
		     * prefix (a protocol name followed by ",").
		     */
		    r = strchr(p, ',');
		    if (r) {
			int i, j;
			for (i = 0; backends[i].backend != NULL; i++) {
			    j = strlen(backends[i].name);
			    if (j == r - p &&
				!memcmp(backends[i].name, p, j)) {
				default_protocol = cfg.protocol =
				    backends[i].protocol;
				portnumber =
				    backends[i].backend->default_port;
				p = r + 1;
				break;
			    }
			}
		    }

		    /*
		     * Three cases. Either (a) there's a nonzero
		     * length string followed by an @, in which
		     * case that's user and the remainder is host.
		     * Or (b) there's only one string, not counting
		     * a potential initial @, and it exists in the
		     * saved-sessions database. Or (c) only one
		     * string and it _doesn't_ exist in the
		     * database.
		     */
		    r = strrchr(p, '@');
		    if (r == p)
			p++, r = NULL; /* discount initial @ */
		    if (r == NULL) {
			/*
			 * One string.
			 */
			Config cfg2;
			do_defaults(p, &cfg2);
			if (cfg2.host[0] == '\0') {
			    /* No settings for this host; use defaults */
			    strncpy(cfg.host, p, sizeof(cfg.host) - 1);
			    cfg.host[sizeof(cfg.host) - 1] = '\0';
			    cfg.port = default_port;
			} else {
			    cfg = cfg2;
			    cfg.remote_cmd_ptr = cfg.remote_cmd;
			}
		    } else {
			*r++ = '\0';
			strncpy(cfg.username, p, sizeof(cfg.username) - 1);
			cfg.username[sizeof(cfg.username) - 1] = '\0';
			strncpy(cfg.host, r, sizeof(cfg.host) - 1);
			cfg.host[sizeof(cfg.host) - 1] = '\0';
			cfg.port = default_port;
		    }
		}
	    } else {
		char *command;
		int cmdlen, cmdsize;
		cmdlen = cmdsize = 0;
		command = NULL;

		while (argc) {
		    while (*p) {
			if (cmdlen >= cmdsize) {
			    cmdsize = cmdlen + 512;
			    command = srealloc(command, cmdsize);
			}
			command[cmdlen++]=*p++;
		    }
		    if (cmdlen >= cmdsize) {
			cmdsize = cmdlen + 512;
			command = srealloc(command, cmdsize);
		    }
		    command[cmdlen++]=' '; /* always add trailing space */
		    if (--argc) p = *++argv;
		}
		if (cmdlen) command[--cmdlen]='\0';
				       /* change trailing blank to NUL */
		cfg.remote_cmd_ptr = command;
		cfg.remote_cmd_ptr2 = NULL;
		cfg.nopty = TRUE;      /* command => no terminal */

		break;		       /* done with cmdline */
	    }
	}
    }

    if (!*cfg.host) {
	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));
	}
    }

    /*
     * Perform command-line overrides on session configuration.
     */
    cmdline_run_saved();

    /*
     * 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;

    /*
     * Select protocol. This is farmed out into a table in a
     * separate file to enable an ssh-free variant.
     */
    {
	int i;
	back = NULL;
	for (i = 0; backends[i].backend != NULL; i++)
	    if (backends[i].protocol == cfg.protocol) {
		back = backends[i].backend;
		break;
	    }
	if (back == NULL) {
	    fprintf(stderr,
		    "Internal fault: Unsupported protocol found\n");
	    return 1;
	}
    }

    /*
     * Select port.
     */
    if (portnumber != -1)
	cfg.port = portnumber;

    /*
     * Initialise WinSock.
     */
    winsock_ver = MAKEWORD(2, 0);
    if (WSAStartup(winsock_ver, &wsadata)) {
	MessageBox(NULL, "Unable to initialise WinSock", "WinSock Error",
		   MB_OK | MB_ICONEXCLAMATION);
	return 1;
    }
    if (LOBYTE(wsadata.wVersion) != 2 || HIBYTE(wsadata.wVersion) != 0) {
	MessageBox(NULL, "WinSock version is incompatible with 2.0",
		   "WinSock Error", MB_OK | MB_ICONEXCLAMATION);
	WSACleanup();
	return 1;
    }
    sk_init();

    /*
     * Start up the connection.
     */
    netevent = CreateEvent(NULL, FALSE, FALSE, NULL);
    {
	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, nodelay);
	if (error) {
	    fprintf(stderr, "Unable to open connection:\n%s", error);
	    return 1;
	}
	sfree(realhost);
    }
    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);
    errhandle = GetStdHandle(STD_ERROR_HANDLE);
    GetConsoleMode(inhandle, &orig_console_mode);
    SetConsoleMode(inhandle, ENABLE_PROCESSED_INPUT);

    /*
     * Turn off ECHO and LINE input modes. We don't care if this
     * call fails, because we know we aren't necessarily running in
     * a console.
     */
    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");
	cleanup_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");
	cleanup_exit(1);
    }

    while (1) {
	int n;

	if (!sending && back->sendok()) {
	    /*
	     * Create a separate thread to read from stdin. This is
	     * a total pain, but I can't find another way to do it:
	     *
	     *  - an overlapped ReadFile or ReadFileEx just doesn't
	     *    happen; we get failure from ReadFileEx, and
	     *    ReadFile blocks despite being given an OVERLAPPED
	     *    structure. Perhaps we can't do overlapped reads
	     *    on consoles. WHY THE HELL NOT?
	     * 
	     *  - WaitForMultipleObjects(netevent, console) doesn't
	     *    work, because it signals the console when
	     *    _anything_ happens, including mouse motions and
	     *    other things that don't cause data to be readable
	     *    - so we're back to ReadFile blocking.
	     */
	    idata.event = stdinevent;
	    idata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
	    if (!CreateThread(NULL, 0, stdin_read_thread,
			      &idata, 0, &in_threadid)) {
		fprintf(stderr, "Unable to create input thread\n");
		cleanup_exit(1);
	    }
	    sending = TRUE;
	}

	n = WaitForMultipleObjects(4, handles, FALSE, INFINITE);
	if (n == 0) {
	    WSANETWORKEVENTS things;
	    SOCKET socket;
	    extern SOCKET first_socket(int *), next_socket(int *);
	    extern int select_result(WPARAM, LPARAM);
	    int i, socketstate;

	    /*
	     * We must not call select_result() for any socket
	     * until we have finished enumerating within the tree.
	     * This is because select_result() may close the socket
	     * and modify the tree.
	     */
	    /* Count the active sockets. */
	    i = 0;
	    for (socket = first_socket(&socketstate);
		 socket != INVALID_SOCKET;
		 socket = next_socket(&socketstate)) i++;

	    /* Expand the buffer if necessary. */
	    if (i > sksize) {
		sksize = i + 16;
		sklist = srealloc(sklist, sksize * sizeof(*sklist));
	    }

	    /* Retrieve the sockets into sklist. */
	    skcount = 0;
	    for (socket = first_socket(&socketstate);
		 socket != INVALID_SOCKET;
		 socket = next_socket(&socketstate)) {
		sklist[skcount++] = socket;
	    }

	    /* Now we're done enumerating; go through the list. */
	    for (i = 0; i < skcount; i++) {
		WPARAM wp;
		socket = sklist[i];
		wp = (WPARAM) socket;
		if (!WSAEnumNetworkEvents(socket, NULL, &things)) {
                    static const struct { int bit, mask; } eventtypes[] = {
                        {FD_CONNECT_BIT, FD_CONNECT},
                        {FD_READ_BIT, FD_READ},
                        {FD_CLOSE_BIT, FD_CLOSE},
                        {FD_OOB_BIT, FD_OOB},
                        {FD_WRITE_BIT, FD_WRITE},
                        {FD_ACCEPT_BIT, FD_ACCEPT},
                    };
                    int e;

		    noise_ultralight(socket);
		    noise_ultralight(things.lNetworkEvents);

                    for (e = 0; e < lenof(eventtypes); e++)
                        if (things.lNetworkEvents & eventtypes[e].mask) {
                            LPARAM lp;
                            int err = things.iErrorCode[eventtypes[e].bit];
                            lp = WSAMAKESELECTREPLY(eventtypes[e].mask, err);
                            connopen &= select_result(wp, lp);
                        }
		}
	    }
	} else if (n == 1) {
	    reading = 0;
	    noise_ultralight(idata.len);
	    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");
		cleanup_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");
		cleanup_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) &&
	    bufchain_size(&stdout_data) == 0 &&
	    bufchain_size(&stderr_data) == 0)
	    break;		       /* we closed the connection */
    }
    WSACleanup();
    exitcode = back->exitcode();
    if (exitcode < 0) {
	fprintf(stderr, "Remote process exit code unavailable\n");
	exitcode = 1;		       /* this is an error condition */
    }
    return exitcode;
}
Commit	Line	Data
12dc4ec0	1	/*
	2	* PLink - a command-line (stdin/stdout) variant of PuTTY.
	3	*/
	4
4d331a77	5	#ifndef AUTO_WINSOCK
12dc4ec0	6	#include <winsock2.h>
4d331a77	7	#endif
12dc4ec0	8	#include <windows.h>
12dc4ec0	9	#include <stdio.h>
49bad831	10	#include <stdlib.h>
2b0c045b	11	#include <assert.h>
12dc4ec0	12	#include <stdarg.h>
12dc4ec0	13
32874aea	14	#define PUTTY_DO_GLOBALS /* actually _define_ globals */
12dc4ec0	15	#include "putty.h"
a9422f39	16	#include "storage.h"
8df7a775	17	#include "tree234.h"
12dc4ec0	18
5471d09a	19	#define MAX_STDIN_BACKLOG 4096
5471d09a	20
32874aea	21	void fatalbox(char *p, ...)
32874aea	22	{
12dc4ec0	23	va_list ap;
49bad831	24	fprintf(stderr, "FATAL ERROR: ");
1709795f	25	va_start(ap, p);
	26	vfprintf(stderr, p, ap);
	27	va_end(ap);
	28	fputc('\n', stderr);
	29	WSACleanup();
	30	cleanup_exit(1);
	31	}
	32	void modalfatalbox(char *p, ...)
	33	{
	34	va_list ap;
	35	fprintf(stderr, "FATAL ERROR: ");
12dc4ec0	36	va_start(ap, p);
	37	vfprintf(stderr, p, ap);
	38	va_end(ap);
	39	fputc('\n', stderr);
	40	WSACleanup();
93b581bd	41	cleanup_exit(1);
12dc4ec0	42	}
32874aea	43	void connection_fatal(char *p, ...)
32874aea	44	{
8d5de777	45	va_list ap;
49bad831	46	fprintf(stderr, "FATAL ERROR: ");
8d5de777	47	va_start(ap, p);
	48	vfprintf(stderr, p, ap);
	49	va_end(ap);
	50	fputc('\n', stderr);
	51	WSACleanup();
93b581bd	52	cleanup_exit(1);
8d5de777	53	}
c0a81592	54	void cmdline_error(char *p, ...)
	55	{
	56	va_list ap;
	57	fprintf(stderr, "plink: ");
	58	va_start(ap, p);
	59	vfprintf(stderr, p, ap);
	60	va_end(ap);
	61	fputc('\n', stderr);
	62	exit(1);
	63	}
12dc4ec0	64
d8426c54	65	static char *password = NULL;
d8426c54	66
0965bee0	67	HANDLE inhandle, outhandle, errhandle;
6f34e365	68	DWORD orig_console_mode;
6f34e365	69
8df7a775	70	WSAEVENT netevent;
8df7a775	71
32874aea	72	int term_ldisc(int mode)
	73	{
	74	return FALSE;
	75	}
	76	void ldisc_update(int echo, int edit)
	77	{
0965bee0	78	/* Update stdin read mode to reflect changes in line discipline. */
	79	DWORD mode;
	80
	81	mode = ENABLE_PROCESSED_INPUT;
	82	if (echo)
32874aea	83	mode = mode \| ENABLE_ECHO_INPUT;
0965bee0	84	else
32874aea	85	mode = mode & ~ENABLE_ECHO_INPUT;
0965bee0	86	if (edit)
32874aea	87	mode = mode \| ENABLE_LINE_INPUT;
0965bee0	88	else
32874aea	89	mode = mode & ~ENABLE_LINE_INPUT;
0965bee0	90	SetConsoleMode(inhandle, mode);
	91	}
	92
5471d09a	93	struct input_data {
	94	DWORD len;
	95	char buffer[4096];
	96	HANDLE event, eventback;
	97	};
	98
32874aea	99	static DWORD WINAPI stdin_read_thread(void *param)
	100	{
	101	struct input_data idata = (struct input_data ) param;
12dc4ec0	102	HANDLE inhandle;
	103
	104	inhandle = GetStdHandle(STD_INPUT_HANDLE);
	105
	106	while (ReadFile(inhandle, idata->buffer, sizeof(idata->buffer),
32874aea	107	&idata->len, NULL) && idata->len > 0) {
	108	SetEvent(idata->event);
	109	WaitForSingleObject(idata->eventback, INFINITE);
12dc4ec0	110	}
	111
	112	idata->len = 0;
	113	SetEvent(idata->event);
	114
	115	return 0;
	116	}
	117
5471d09a	118	struct output_data {
	119	DWORD len, lenwritten;
	120	int writeret;
	121	char *buffer;
	122	int is_stderr, done;
	123	HANDLE event, eventback;
	124	int busy;
	125	};
	126
	127	static DWORD WINAPI stdout_write_thread(void *param)
	128	{
	129	struct output_data odata = (struct output_data ) param;
	130	HANDLE outhandle, errhandle;
	131
	132	outhandle = GetStdHandle(STD_OUTPUT_HANDLE);
	133	errhandle = GetStdHandle(STD_ERROR_HANDLE);
	134
	135	while (1) {
	136	WaitForSingleObject(odata->eventback, INFINITE);
	137	if (odata->done)
	138	break;
	139	odata->writeret =
	140	WriteFile(odata->is_stderr ? errhandle : outhandle,
	141	odata->buffer, odata->len, &odata->lenwritten, NULL);
	142	SetEvent(odata->event);
	143	}
	144
	145	return 0;
	146	}
	147
	148	bufchain stdout_data, stderr_data;
	149	struct output_data odata, edata;
	150
	151	void try_output(int is_stderr)
	152	{
	153	struct output_data *data = (is_stderr ? &edata : &odata);
	154	void *senddata;
	155	int sendlen;
	156
	157	if (!data->busy) {
	158	bufchain_prefix(is_stderr ? &stderr_data : &stdout_data,
	159	&senddata, &sendlen);
	160	data->buffer = senddata;
	161	data->len = sendlen;
	162	SetEvent(data->eventback);
	163	data->busy = 1;
	164	}
	165	}
	166
	167	int from_backend(int is_stderr, char *data, int len)
	168	{
5471d09a	169	HANDLE h = (is_stderr ? errhandle : outhandle);
5471d09a	170	int osize, esize;
5471d09a	171
2b0c045b	172	assert(len > 0);
2b0c045b	173
5471d09a	174	if (is_stderr) {
	175	bufchain_add(&stderr_data, data, len);
	176	try_output(1);
	177	} else {
	178	bufchain_add(&stdout_data, data, len);
	179	try_output(0);
	180	}
	181
	182	osize = bufchain_size(&stdout_data);
	183	esize = bufchain_size(&stderr_data);
	184
	185	return osize + esize;
	186	}
	187
d8426c54	188	/*
	189	* Short description of parameters.
	190	*/
	191	static void usage(void)
	192	{
	193	printf("PuTTY Link: command-line connection utility\n");
	194	printf("%s\n", ver);
	195	printf("Usage: plink [options] [user@]host [command]\n");
e672967c	196	printf(" (\"host\" can also be a PuTTY saved session name)\n");
d8426c54	197	printf("Options:\n");
d8426c54	198	printf(" -v show verbose messages\n");
e2a197cf	199	printf(" -load sessname Load settings from saved session\n");
	200	printf(" -ssh -telnet -rlogin -raw\n");
	201	printf(" force use of a particular protocol (default SSH)\n");
d8426c54	202	printf(" -P port connect to specified port\n");
e2a197cf	203	printf(" -l user connect with specified username\n");
96621a84	204	printf(" -m file read remote command(s) from file\n");
e2a197cf	205	printf(" -batch disable all interactive prompts\n");
	206	printf("The following options only apply to SSH connections:\n");
	207	printf(" -pw passw login with specified password\n");
e7aabca4	208	printf(" -L listen-port:host:port Forward local port to "
	209	"remote address\n");
	210	printf(" -R listen-port:host:port Forward remote port to"
	211	" local address\n");
e2a197cf	212	printf(" -X -x enable / disable X11 forwarding\n");
	213	printf(" -A -a enable / disable agent forwarding\n");
	214	printf(" -t -T enable / disable pty allocation\n");
	215	printf(" -1 -2 force use of particular protocol version\n");
	216	printf(" -C enable compression\n");
	217	printf(" -i key private key file for authentication\n");
d8426c54	218	exit(1);
	219	}
	220
32874aea	221	char *do_select(SOCKET skt, int startup)
32874aea	222	{
8df7a775	223	int events;
8df7a775	224	if (startup) {
3ad9d396	225	events = (FD_CONNECT \| FD_READ \| FD_WRITE \|
3ad9d396	226	FD_OOB \| FD_CLOSE \| FD_ACCEPT);
8df7a775	227	} else {
	228	events = 0;
	229	}
32874aea	230	if (WSAEventSelect(skt, netevent, events) == SOCKET_ERROR) {
	231	switch (WSAGetLastError()) {
	232	case WSAENETDOWN:
	233	return "Network is down";
	234	default:
	235	return "WSAAsyncSelect(): unknown error";
	236	}
8df7a775	237	}
	238	return NULL;
	239	}
	240
32874aea	241	int main(int argc, char **argv)
32874aea	242	{
12dc4ec0	243	WSADATA wsadata;
12dc4ec0	244	WORD winsock_ver;
5471d09a	245	WSAEVENT stdinevent, stdoutevent, stderrevent;
	246	HANDLE handles[4];
	247	DWORD in_threadid, out_threadid, err_threadid;
12dc4ec0	248	struct input_data idata;
5471d09a	249	int reading;
12dc4ec0	250	int sending;
d8426c54	251	int portnumber = -1;
8df7a775	252	SOCKET *sklist;
	253	int skcount, sksize;
	254	int connopen;
d8d6c7e5	255	int exitcode;
12dc4ec0	256
ff2ae367	257	ssh_get_line = console_get_line;
67779be7	258
32874aea	259	sklist = NULL;
32874aea	260	skcount = sksize = 0;
c9bdcd96	261	/*
	262	* Initialise port and protocol to sensible defaults. (These
	263	* will be overridden by more or less anything.)
	264	*/
	265	default_protocol = PROT_SSH;
	266	default_port = 22;
8df7a775	267
67779be7	268	flags = FLAG_STDERR;
12dc4ec0	269	/*
	270	* Process the command line.
	271	*/
a9422f39	272	do_defaults(NULL, &cfg);
e7a7383f	273	default_protocol = cfg.protocol;
e7a7383f	274	default_port = cfg.port;
8cb9c947	275	{
32874aea	276	/*
	277	* Override the default protocol if PLINK_PROTOCOL is set.
	278	*/
	279	char *p = getenv("PLINK_PROTOCOL");
	280	int i;
	281	if (p) {
	282	for (i = 0; backends[i].backend != NULL; i++) {
	283	if (!strcmp(backends[i].name, p)) {
	284	default_protocol = cfg.protocol = backends[i].protocol;
	285	default_port = cfg.port =
	286	backends[i].backend->default_port;
	287	break;
	288	}
	289	}
	290	}
8cb9c947	291	}
12dc4ec0	292	while (--argc) {
32874aea	293	char p = ++argv;
32874aea	294	if (*p == '-') {
c0a81592	295	int ret = cmdline_process_param(p, (argc > 1 ? argv[1] : NULL), 1);
	296	if (ret == -2) {
	297	fprintf(stderr,
	298	"plink: option \"%s\" requires an argument\n", p);
	299	} else if (ret == 2) {
	300	--argc, ++argv;
	301	} else if (ret == 1) {
	302	continue;
ff2ae367	303	} else if (!strcmp(p, "-batch")) {
c0a81592	304	console_batch_mode = 1;
32874aea	305	}
12dc4ec0	306	} else if (*p) {
32874aea	307	if (!*cfg.host) {
	308	char *q = p;
	309	/*
	310	* If the hostname starts with "telnet:", set the
	311	* protocol to Telnet and process the string as a
	312	* Telnet URL.
	313	*/
	314	if (!strncmp(q, "telnet:", 7)) {
	315	char c;
	316
	317	q += 7;
	318	if (q[0] == '/' && q[1] == '/')
	319	q += 2;
	320	cfg.protocol = PROT_TELNET;
	321	p = q;
	322	while (p && p != ':' && *p != '/')
	323	p++;
	324	c = *p;
	325	if (*p)
	326	*p++ = '\0';
	327	if (c == ':')
	328	cfg.port = atoi(p);
	329	else
	330	cfg.port = -1;
	331	strncpy(cfg.host, q, sizeof(cfg.host) - 1);
	332	cfg.host[sizeof(cfg.host) - 1] = '\0';
	333	} else {
	334	char *r;
	335	/*
	336	* Before we process the [user@]host string, we
	337	* first check for the presence of a protocol
	338	* prefix (a protocol name followed by ",").
	339	*/
	340	r = strchr(p, ',');
	341	if (r) {
	342	int i, j;
	343	for (i = 0; backends[i].backend != NULL; i++) {
	344	j = strlen(backends[i].name);
	345	if (j == r - p &&
	346	!memcmp(backends[i].name, p, j)) {
	347	default_protocol = cfg.protocol =
	348	backends[i].protocol;
	349	portnumber =
	350	backends[i].backend->default_port;
	351	p = r + 1;
	352	break;
	353	}
	354	}
	355	}
	356
	357	/*
	358	* Three cases. Either (a) there's a nonzero
	359	* length string followed by an @, in which
	360	* case that's user and the remainder is host.
	361	* Or (b) there's only one string, not counting
	362	* a potential initial @, and it exists in the
	363	* saved-sessions database. Or (c) only one
	364	* string and it _doesn't_ exist in the
	365	* database.
	366	*/
	367	r = strrchr(p, '@');
	368	if (r == p)
	369	p++, r = NULL; /* discount initial @ */
	370	if (r == NULL) {
371	/*
372	* One string.
373	*/
374	Config cfg2;
375	do_defaults(p, &cfg2);
376	if (cfg2.host[0] == '\0') {
377	/* No settings for this host; use defaults */
378	strncpy(cfg.host, p, sizeof(cfg.host) - 1);
379	cfg.host[sizeof(cfg.host) - 1] = '\0';
380	cfg.port = default_port;
381	} else {
382	cfg = cfg2;
383	cfg.remote_cmd_ptr = cfg.remote_cmd;
384	}
385	} else {
386	*r++ = '\0';
387	strncpy(cfg.username, p, sizeof(cfg.username) - 1);
388	cfg.username[sizeof(cfg.username) - 1] = '\0';
389	strncpy(cfg.host, r, sizeof(cfg.host) - 1);
390	cfg.host[sizeof(cfg.host) - 1] = '\0';
391	cfg.port = default_port;
392	}
393	}
394	} else {
385528da	395	char *command;
	396	int cmdlen, cmdsize;
	397	cmdlen = cmdsize = 0;
	398	command = NULL;
	399
	400	while (argc) {
	401	while (*p) {
	402	if (cmdlen >= cmdsize) {
	403	cmdsize = cmdlen + 512;
	404	command = srealloc(command, cmdsize);
	405	}
	406	command[cmdlen++]=*p++;
	407	}
	408	if (cmdlen >= cmdsize) {
	409	cmdsize = cmdlen + 512;
	410	command = srealloc(command, cmdsize);
	411	}
	412	command[cmdlen++]=' '; /* always add trailing space */
	413	if (--argc) p = *++argv;
32874aea	414	}
385528da	415	if (cmdlen) command[--cmdlen]='\0';
	416	/* change trailing blank to NUL */
	417	cfg.remote_cmd_ptr = command;
	418	cfg.remote_cmd_ptr2 = NULL;
32874aea	419	cfg.nopty = TRUE; /* command => no terminal */
385528da	420
32874aea	421	break; /* done with cmdline */
32874aea	422	}
12dc4ec0	423	}
	424	}
	425
d8426c54	426	if (!*cfg.host) {
32874aea	427	usage();
d8426c54	428	}
d8426c54	429
449925a6	430	/*
	431	* Trim leading whitespace off the hostname if it's there.
	432	*/
	433	{
	434	int space = strspn(cfg.host, " \t");
	435	memmove(cfg.host, cfg.host+space, 1+strlen(cfg.host)-space);
	436	}
	437
	438	/* See if host is of the form user@host */
	439	if (cfg.host[0] != '\0') {
	440	char *atsign = strchr(cfg.host, '@');
	441	/* Make sure we're not overflowing the user field */
	442	if (atsign) {
	443	if (atsign - cfg.host < sizeof cfg.username) {
	444	strncpy(cfg.username, cfg.host, atsign - cfg.host);
	445	cfg.username[atsign - cfg.host] = '\0';
	446	}
	447	memmove(cfg.host, atsign + 1, 1 + strlen(atsign + 1));
	448	}
	449	}
	450
	451	/*
c0a81592	452	* Perform command-line overrides on session configuration.
	453	*/
	454	cmdline_run_saved();
	455
	456	/*
449925a6	457	* Trim a colon suffix off the hostname if it's there.
	458	*/
	459	cfg.host[strcspn(cfg.host, ":")] = '\0';
	460
96621a84	461	if (!*cfg.remote_cmd_ptr)
32874aea	462	flags \|= FLAG_INTERACTIVE;
67779be7	463
12dc4ec0	464	/*
	465	* Select protocol. This is farmed out into a table in a
	466	* separate file to enable an ssh-free variant.
	467	*/
	468	{
32874aea	469	int i;
	470	back = NULL;
	471	for (i = 0; backends[i].backend != NULL; i++)
	472	if (backends[i].protocol == cfg.protocol) {
	473	back = backends[i].backend;
	474	break;
	475	}
	476	if (back == NULL) {
	477	fprintf(stderr,
	478	"Internal fault: Unsupported protocol found\n");
	479	return 1;
	480	}
12dc4ec0	481	}
	482
	483	/*
8cb9c947	484	* Select port.
	485	*/
	486	if (portnumber != -1)
32874aea	487	cfg.port = portnumber;
8cb9c947	488
8cb9c947	489	/*
12dc4ec0	490	* Initialise WinSock.
	491	*/
	492	winsock_ver = MAKEWORD(2, 0);
	493	if (WSAStartup(winsock_ver, &wsadata)) {
	494	MessageBox(NULL, "Unable to initialise WinSock", "WinSock Error",
	495	MB_OK \| MB_ICONEXCLAMATION);
	496	return 1;
	497	}
	498	if (LOBYTE(wsadata.wVersion) != 2 \|\| HIBYTE(wsadata.wVersion) != 0) {
	499	MessageBox(NULL, "WinSock version is incompatible with 2.0",
	500	"WinSock Error", MB_OK \| MB_ICONEXCLAMATION);
	501	WSACleanup();
	502	return 1;
	503	}
8df7a775	504	sk_init();
12dc4ec0	505
	506	/*
	507	* Start up the connection.
	508	*/
8df7a775	509	netevent = CreateEvent(NULL, FALSE, FALSE, NULL);
12dc4ec0	510	{
	511	char *error;
	512	char *realhost;
2184a5d9	513	/* nodelay is only useful if stdin is a character device (console) */
	514	int nodelay = cfg.tcp_nodelay &&
	515	(GetFileType(GetStdHandle(STD_INPUT_HANDLE)) == FILE_TYPE_CHAR);
12dc4ec0	516
2184a5d9	517	error = back->init(cfg.host, cfg.port, &realhost, nodelay);
12dc4ec0	518	if (error) {
	519	fprintf(stderr, "Unable to open connection:\n%s", error);
	520	return 1;
	521	}
6e1ebb76	522	sfree(realhost);
12dc4ec0	523	}
8df7a775	524	connopen = 1;
12dc4ec0	525
12dc4ec0	526	stdinevent = CreateEvent(NULL, FALSE, FALSE, NULL);
5471d09a	527	stdoutevent = CreateEvent(NULL, FALSE, FALSE, NULL);
5471d09a	528	stderrevent = CreateEvent(NULL, FALSE, FALSE, NULL);
12dc4ec0	529
0965bee0	530	inhandle = GetStdHandle(STD_INPUT_HANDLE);
12dc4ec0	531	outhandle = GetStdHandle(STD_OUTPUT_HANDLE);
fe50e814	532	errhandle = GetStdHandle(STD_ERROR_HANDLE);
0965bee0	533	GetConsoleMode(inhandle, &orig_console_mode);
0965bee0	534	SetConsoleMode(inhandle, ENABLE_PROCESSED_INPUT);
12dc4ec0	535
12dc4ec0	536	/*
12dc4ec0	537	* Turn off ECHO and LINE input modes. We don't care if this
	538	* call fails, because we know we aren't necessarily running in
	539	* a console.
	540	*/
12dc4ec0	541	handles[0] = netevent;
12dc4ec0	542	handles[1] = stdinevent;
5471d09a	543	handles[2] = stdoutevent;
5471d09a	544	handles[3] = stderrevent;
12dc4ec0	545	sending = FALSE;
5471d09a	546
	547	/*
	548	* Create spare threads to write to stdout and stderr, so we
	549	* can arrange asynchronous writes.
	550	*/
	551	odata.event = stdoutevent;
	552	odata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
	553	odata.is_stderr = 0;
	554	odata.busy = odata.done = 0;
	555	if (!CreateThread(NULL, 0, stdout_write_thread,
	556	&odata, 0, &out_threadid)) {
	557	fprintf(stderr, "Unable to create output thread\n");
93b581bd	558	cleanup_exit(1);
5471d09a	559	}
	560	edata.event = stderrevent;
	561	edata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
	562	edata.is_stderr = 1;
	563	edata.busy = edata.done = 0;
	564	if (!CreateThread(NULL, 0, stdout_write_thread,
	565	&edata, 0, &err_threadid)) {
	566	fprintf(stderr, "Unable to create error output thread\n");
93b581bd	567	cleanup_exit(1);
5471d09a	568	}
5471d09a	569
12dc4ec0	570	while (1) {
32874aea	571	int n;
	572
	573	if (!sending && back->sendok()) {
	574	/*
	575	* Create a separate thread to read from stdin. This is
	576	* a total pain, but I can't find another way to do it:
	577	*
	578	* - an overlapped ReadFile or ReadFileEx just doesn't
	579	* happen; we get failure from ReadFileEx, and
	580	* ReadFile blocks despite being given an OVERLAPPED
	581	* structure. Perhaps we can't do overlapped reads
	582	* on consoles. WHY THE HELL NOT?
	583	*
	584	* - WaitForMultipleObjects(netevent, console) doesn't
	585	* work, because it signals the console when
	586	* _anything_ happens, including mouse motions and
	587	* other things that don't cause data to be readable
	588	* - so we're back to ReadFile blocking.
	589	*/
	590	idata.event = stdinevent;
	591	idata.eventback = CreateEvent(NULL, FALSE, FALSE, NULL);
	592	if (!CreateThread(NULL, 0, stdin_read_thread,
5471d09a	593	&idata, 0, &in_threadid)) {
5471d09a	594	fprintf(stderr, "Unable to create input thread\n");
93b581bd	595	cleanup_exit(1);
32874aea	596	}
	597	sending = TRUE;
	598	}
	599
5471d09a	600	n = WaitForMultipleObjects(4, handles, FALSE, INFINITE);
32874aea	601	if (n == 0) {
32874aea	602	WSANETWORKEVENTS things;
8df7a775	603	SOCKET socket;
d2371c81	604	extern SOCKET first_socket(int ), next_socket(int );
8df7a775	605	extern int select_result(WPARAM, LPARAM);
32874aea	606	int i, socketstate;
	607
	608	/*
	609	* We must not call select_result() for any socket
	610	* until we have finished enumerating within the tree.
	611	* This is because select_result() may close the socket
	612	* and modify the tree.
	613	*/
	614	/* Count the active sockets. */
	615	i = 0;
	616	for (socket = first_socket(&socketstate);
	617	socket != INVALID_SOCKET;
	618	socket = next_socket(&socketstate)) i++;
	619
	620	/* Expand the buffer if necessary. */
	621	if (i > sksize) {
	622	sksize = i + 16;
	623	sklist = srealloc(sklist, sksize * sizeof(*sklist));
	624	}
	625
	626	/* Retrieve the sockets into sklist. */
	627	skcount = 0;
	628	for (socket = first_socket(&socketstate);
	629	socket != INVALID_SOCKET;
d2371c81	630	socket = next_socket(&socketstate)) {
32874aea	631	sklist[skcount++] = socket;
	632	}
	633
	634	/* Now we're done enumerating; go through the list. */
	635	for (i = 0; i < skcount; i++) {
	636	WPARAM wp;
	637	socket = sklist[i];
	638	wp = (WPARAM) socket;
ffb959c7	639	if (!WSAEnumNetworkEvents(socket, NULL, &things)) {
64cdd21b	640	static const struct { int bit, mask; } eventtypes[] = {
	641	{FD_CONNECT_BIT, FD_CONNECT},
	642	{FD_READ_BIT, FD_READ},
	643	{FD_CLOSE_BIT, FD_CLOSE},
	644	{FD_OOB_BIT, FD_OOB},
	645	{FD_WRITE_BIT, FD_WRITE},
	646	{FD_ACCEPT_BIT, FD_ACCEPT},
	647	};
	648	int e;
	649
32874aea	650	noise_ultralight(socket);
32874aea	651	noise_ultralight(things.lNetworkEvents);
d74d141c	652
64cdd21b	653	for (e = 0; e < lenof(eventtypes); e++)
	654	if (things.lNetworkEvents & eventtypes[e].mask) {
	655	LPARAM lp;
	656	int err = things.iErrorCode[eventtypes[e].bit];
	657	lp = WSAMAKESELECTREPLY(eventtypes[e].mask, err);
	658	connopen &= select_result(wp, lp);
	659	}
8df7a775	660	}
8df7a775	661	}
32874aea	662	} else if (n == 1) {
5471d09a	663	reading = 0;
32874aea	664	noise_ultralight(idata.len);
42856df4	665	if (connopen && back->socket() != NULL) {
	666	if (idata.len > 0) {
	667	back->send(idata.buffer, idata.len);
	668	} else {
	669	back->special(TS_EOF);
	670	}
32874aea	671	}
5471d09a	672	} else if (n == 2) {
	673	odata.busy = 0;
	674	if (!odata.writeret) {
	675	fprintf(stderr, "Unable to write to standard output\n");
93b581bd	676	cleanup_exit(0);
5471d09a	677	}
	678	bufchain_consume(&stdout_data, odata.lenwritten);
	679	if (bufchain_size(&stdout_data) > 0)
	680	try_output(0);
42856df4	681	if (connopen && back->socket() != NULL) {
	682	back->unthrottle(bufchain_size(&stdout_data) +
	683	bufchain_size(&stderr_data));
	684	}
5471d09a	685	} else if (n == 3) {
	686	edata.busy = 0;
	687	if (!edata.writeret) {
	688	fprintf(stderr, "Unable to write to standard output\n");
93b581bd	689	cleanup_exit(0);
5471d09a	690	}
	691	bufchain_consume(&stderr_data, edata.lenwritten);
	692	if (bufchain_size(&stderr_data) > 0)
	693	try_output(1);
42856df4	694	if (connopen && back->socket() != NULL) {
	695	back->unthrottle(bufchain_size(&stdout_data) +
	696	bufchain_size(&stderr_data));
	697	}
5471d09a	698	}
5471d09a	699	if (!reading && back->sendbuffer() < MAX_STDIN_BACKLOG) {
32874aea	700	SetEvent(idata.eventback);
5471d09a	701	reading = 1;
32874aea	702	}
42856df4	703	if ((!connopen \|\| back->socket() == NULL) &&
	704	bufchain_size(&stdout_data) == 0 &&
	705	bufchain_size(&stderr_data) == 0)
32874aea	706	break; /* we closed the connection */
12dc4ec0	707	}
12dc4ec0	708	WSACleanup();
d8d6c7e5	709	exitcode = back->exitcode();
	710	if (exitcode < 0) {
	711	fprintf(stderr, "Remote process exit code unavailable\n");
	712	exitcode = 1; /* this is an error condition */
	713	}
	714	return exitcode;
12dc4ec0	715	}